如何将裸指针传递给Boost.Python？

Question

我正在尝试使用Boost.Python作为C++函数的包装器，该函数接收指针、修改数据（在Python端以numpy数组的形式管理），并返回修改后的结果。如何让Python numpy和Boost.Python协同工作，并在函数内获取原始指针？

#include <boost/python.hpp>
namespace
{
  char const *greet(double *p)
  {
    *p = 2.;
    return "hello world";
  }
}
BOOST_PYTHON_MODULE(module)
{
  boost::python::def("greet", &greet);
}

在Python中，当我尝试时，

import numpy as np
a = np.empty([2], dtype=np.double)
raw_ptr = a.data
print cmod.greet(raw_ptr)

我遇到了错误，

Boost.Python.ArgumentError: Python argument types in
<...>.module.greet(buffer)
did not match C++ signature:
greet(double*)

- rych

Answer 1

有一种看起来行之有效的方法，由Andreas Kloeckner建议，欢迎评论和提供替代方案。greet()函数修改如下：

char const *greet(boost::python::object obj) {
    PyObject* pobj = obj.ptr();
    Py_buffer pybuf;
    if(PyObject_GetBuffer(pobj, &pybuf, PyBUF_SIMPLE)!=-1)
    {
        void *buf = pybuf.buf;
        double *p = (double*)buf;
        *p = 2.;
        *(p+1) = 3;
        PyBuffer_Release(&pybuf);
    }
    return "hello world";
    }

在Python中，只需使用：

print cmod.greet(a)

- rych

Answer 2

您可能需要使用numpy ctypes接口来获取存储的原始指针，然后创建一个ctypes指向double的指针以传递到调用中。ndarray.data是缓冲区类型，而不是指针。

我没有使用boost.python的经验，但我怀疑类似于以下内容：

In [28]: x=np.array([1,2,3,4],dtype=np.double)

In [29]: p=x.ctypes.data_as(ctypes.POINTER(ctypes.c_double))

In [30]: type(p)
Out[30]: <class 'ctypes.LP_c_double'>

如果传递给一个期望指针作为参数的封装的C++函数，它将起作用。

- talonmies

---

是的，我也尝试过了。然后错误信息显示为： Boost.Python.ArgumentError: Python参数类型在 <...>.module.greet(LP_c_double) 与C++签名不匹配： greet(double*) - rych