Làm cách nào để hiển thị bộ đệm byte thô bằng Boost :: Python?

Question

Tôi đã có thư viện C++ của bên thứ ba trong đó một số phương thức lớp sử dụng bộ đệm byte thô. Tôi không hoàn toàn chắc chắn làm thế nào để đối phó trong Boost :: Python với nó.

Thư viện C++ tiêu đề là cái gì đó như:

class CSomeClass 
{ 
    public: 
     int load(unsigned char *& pInBufferData, int & iInBufferSize); 
     int save(unsigned char *& pOutBufferData, int & iOutBufferSize); 
} 

Trong mắc kẹt với Boost :: mã Python ...

class_<CSomeClass>("CSomeClass", init<>()) 
    .def("load", &CSomeClass::load, (args(/* what do I put here??? */))) 
    .def("save", &CSomeClass::save, (args(/* what do I put here??? */))) 

Làm thế nào để bọc các bộ đệm thô để lộ chúng như liệu chuỗi trong Python?

Answer 1

Bạn phải tự viết các hàm trên đối tượng Py_buffer từ dữ liệu đó, cho phép bạn chỉ đọc (sử dụng PyBuffer_FromMemory) hoặc đọc-ghi (sử dụng PyBuffer_FromReadWriteMemory) C được phân bổ trước của bạn Bộ nhớ C++ từ Python.

Đây là cách nó sẽ trông giống như (phản hồi hoan nghênh nhất):

#include <boost/python.hpp> 

using namespace boost::python; 

//I'm assuming your buffer data is allocated from CSomeClass::load() 
//it should return the allocated size in the second argument 
static object csomeclass_load(CSomeClass& self) { 
    unsigned char* buffer; 
    int size; 
    self.load(buffer, size); 

    //now you wrap that as buffer 
    PyObject* py_buf = PyBuffer_FromReadWriteMemory(buffer, size); 
    object retval = object(handle<>(py_buf)); 
    return retval; 
} 

static int csomeclass_save(CSomeClass& self, object buffer) { 
    PyObject* py_buffer = buffer.ptr(); 
    if (!PyBuffer_Check(py_buffer)) { 
    //raise TypeError using standard boost::python mechanisms 
    } 

    //you can also write checks here for length, verify the 
    //buffer is memory-contiguous, etc. 
    unsigned char* cxx_buf = (unsigned char*)py_buffer.buf; 
    int size = (int)py_buffer.len; 
    return self.save(cxx_buf, size); 
} 

Sau đó, khi bạn gắn CSomeClass, sử dụng các chức năng tĩnh ở trên thay vì các phương pháp load và save:

//I think that you should use boost::python::arg instead of boost::python::args 
// -- it gives you better control on the documentation 
class_<CSomeClass>("CSomeClass", init<>()) 
    .def("load", &csomeclass_load, (arg("self")), "doc for load - returns a buffer") 
    .def("save", &csomeclass_save, (arg("self"), arg("buffer")), "doc for save - requires a buffer") 
    ; 

Điều này sẽ trông đủ sâu đối với tôi.