问题可能与PYBIND11_MAKE_OPAQUE对于ArrayComplex4数据结构的处理有关,虽然我现在没有办法让它工作,但我会在有时间的时候再仔细研究一下。
以下代码是我目前为止最接近您设计的代码。我使用了一个额外的用户定义数据结构来包装std::元素:
#include <pybind11/pybind11.h>
#include <pybind11/complex.h>
#include <pybind11/stl.h>
#include <memory>
#include <complex>
#include <array>
#include <cmath>
namespace py = pybind11;
typedef std::array<std::complex<double>, 4> ArrayComplex4;
struct ArrayComplex4Holder
{
ArrayComplex4 data;
ArrayComplex4 getData() { return data; }
};
class MyClass {
public:
MyClass() { }
std::unique_ptr<ArrayComplex4Holder> my_function(double x)
{
std::unique_ptr<ArrayComplex4Holder> ph( new ArrayComplex4Holder());
ph->data[0] = std::complex<double>(x);
return ph;
}
std::unique_ptr<ArrayComplex4Holder> my_function(double x, double y)
{
std::unique_ptr<ArrayComplex4Holder> ph( new ArrayComplex4Holder());
ph->data[0] = std::complex<double>(x);
ph->data[1] = std::complex<double>(y);
return ph;
}
};
PYBIND11_MODULE(example, m) {
m.doc() = "pybind11 example";
py::class_<ArrayComplex4Holder>(m, "ArrayComplex4Holder")
.def(py::init<>())
.def("getData", &ArrayComplex4Holder::getData);
py::class_<MyClass>(m, "MyClass")
.def(py::init<>())
.def("my_function", (std::unique_ptr<ArrayComplex4Holder> (MyClass::*)(double)) &MyClass::my_function)
.def("my_function", (std::unique_ptr<ArrayComplex4Holder> (MyClass::*)(double, double)) &MyClass::my_function);
}
我发现需要添加额外的pybind #includes才能使python转换工作。
简单的python代码如下:
import sys
sys.path.append('/Volumes/RAID 1/Projects/workspace/Project CPP 1')
import example
p = example.MyClass()
print (p.my_function(1.2345).getData())
print (p.my_function(1.2345, 6.7890).getData())
这将产生以下输出,看起来是正确的:
[(1.2345+0j), 0j, 0j, 0j]
[(1.2345+0j), (6.789+0j), 0j, 0j]
我希望这可以为你提供一个开始工作的起点。如果你找到更好的替代方案,我会感兴趣知道...敬礼,AS