Скопируйте данные из PyObject в плавающее *, используя boost - PullRequest
Новая
       28

Скопируйте данные из PyObject в плавающее *, используя boost

0 голосов
/ 23 октября 2018

Думаю, это должно быть легко, но я не могу заставить его работать.Я хочу позвонить C++-function с питона.В качестве аргумента у меня есть np-array.C++ выглядит следующим образом: 

void foo_python(PyObject *p, const int m, const int n){
    float *data = new float[m*n];
    memcpy(data,PyArray_DATA(p),m*n*sizeof(float));
    delete [] data;
}

Когда я вызываю это из python, используя 

hello_ext.foo_python(A,A.shape[0], A.shape[1])

, я получаю 

  error: cannot convert ‘PyObject* {aka _object*}’ to ‘PyArrayObject* {aka tagPyArrayObject*}’ for argument ‘1’ to ‘void* PyArray_DATA(PyArrayObject*)’

Это работало, но пришлокогда я включил #include <boost/python/numpy.hpp> , который необходим в этой функции, где я возвращаю ndarrays в python. 

np::ndarray test() {
    int *data = new int[5];
    for (int i = 0; i < 5; ++i){
        data[i] = i+1;
    }
    p::tuple shape = p::make_tuple(5);
    p::tuple stride = p::make_tuple(sizeof(int));
    p::object own;
    np::dtype dt = np::dtype::get_builtin<int>();
    std::cout << " Here " << std::endl;
    np::ndarray array = np::from_data(data, dt, shape, stride, own);
    cout<<"Selective Multidimensional array :: "<<std::endl << 
    p::extract<char const *>(p::str(array)) << std::endl;
    // delete [] data;
    return array;
}

Это модуль: 

BOOST_PYTHON_MODULE(libhello_ext)
{
    using namespace boost::python;
    Py_Initialize();
    boost::python::numpy::initialize();
    def("greet", greet);
    def("sum", sum);
    def("foo_python", foo_python);
    def("test", test);
}

1 Ответ

0 голосов
/ 23 октября 2018

Решение, которое я придумал, состояло в том, чтобы заменить PyObject *p на np::ndarray &p.Затем я могу создать указатель float *p = reinterpret_cast<float *>(p.get_data()).

p, затем можно использовать в Map для создания Eigen::Matrix, используя Map<FMatrix > A(data, m, n); с

typedef Eigen::Matrix<float, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor> FMatrix;

...