用 swig 包装 boost::ublas

Question

我正在尝试在 numpy 和 boost::ublas 层之间传递数据。我 写了一个超薄包装器，因为 swig 无法解析 ublas' 标题正确。代码如下所示，

#include <boost/numeric/ublas/vector.hpp>
#include <boost/numeric/ublas/matrix.hpp>
#include <boost/lexical_cast.hpp>
#include <algorithm>
#include <sstream>
#include <string>

using std::copy;
using namespace boost;

typedef boost::numeric::ublas::matrix<double> dm;
typedef boost::numeric::ublas::vector<double> dv;

class dvector : public dv{
 public:
 dvector(const int rhs):dv(rhs){;};
 dvector();
 dvector(const int size, double* ptr):dv(size){
   copy(ptr, ptr+sizeof(double)*size, &(dv::data()[0]));
 }
 ~dvector(){}
};

带有 SWIG 界面，看起来就像

%apply(int DIM1, double* INPLACE_ARRAY1) {(const int size, double* ptr)}
class dvector{
 public:
 dvector(const int rhs);
 dvector();
 dvector(const int size, double* ptr);
       %newobject toString;
 char* toString();
       ~dvector();
};

我已经通过 gcc 4.3 和 vc++9.0 成功编译了它们。然而 当我简单地运行时，

a = dvector(array([1.,2.,3.]))

它会给我一个段错误。这是我第一次将 sigh 与 numpy 一起使用 并且对数据转换和之间没有充分的了解 内存缓冲区传递。有谁看到我有明显的东西 错过了？我尝试使用调试器进行跟踪，但它在 python.exe 的组件中崩溃了。我不知道这是一个问题还是我的简单包装的问题。任何事情都会受到赞赏。

Answer 1

您可能有兴趣查看 pyublas 模块。它无缝地在 numpy 数组和 ublas 数据类型之间进行转换，并且无需复制。

Answer 2

您可能想要替换

copy(ptr, ptr+sizeof(double)*size, &(dv::data()[0]));

通过

复制(ptr, ptr+size, &(dv::data()[0]));

请记住，在 C/C++ 中，对指针进行加法或减法会将其移动其所指向的数据类型大小的倍数。

最好的，