Question

我在使用SWIG包装使用Eigen（线性代数包）的小项目时遇到了问题。我得到了一个我不理解的python错误，并且在网上找不到多少 - 但我怀疑某处有一些C ++内存损坏。我把它归结为一个玩具的例子......但不幸的是，它仍然是合理的：

--- testfunc.cxx ----

#include "Eigen/Dense" 

Eigen::VectorXd test(Eigen::MatrixXd data){ 
        Eigen::VectorXd temp; 
        return temp; 
}

--- testswig.i -----

%module testswig 

%{ 
#define SWIG_FILE_WITH_INIT 
#include "Eigen/Core" 
#include <Python.h>
#include <numpy/arrayobject.h>
#include "testfunc.cxx" 

%} 

%init 
%{ 
  import_array(); 
%} 

%include "numpy.i" 

%typemap(out) Eigen::VectorXd 
{ 
    npy_intp dims[1] = {$1.size()}; 
    PyObject* array = PyArray_SimpleNew(1, dims, NPY_DOUBLE); 
    double* data = ((double *)PyArray_DATA( array )); 
    for (int i = 0; i != dims[0]; ++i){ 
        *data++ = $1.data()[i]; 
    } 
    $result = array; 
} 

%typemap(in) Eigen::MatrixXd (Eigen::MatrixXd TEMP) 
{ 

  int rows = 0; 
  int cols = 0; 

  rows = PyArray_DIM($input,0); 
  cols = PyArray_DIM($input,1); 

  PyArrayObject* temp; 
  PyArg_ParseTuple($input, "O", &temp);   

  TEMP.resize(rows,cols); 
  TEMP.fill(0); 

  double *  values = ((double *) PyArray_DATA($input)); 
  for (long int i = 0; i != rows; ++i){ 
      for(long int j = 0; j != cols; ++j){ 
          // std::cout << "data " << data[i] << std::endl; 
          TEMP(i,j) = values[i*rows+j]; 
      } 
  }   

} 

%include "testfunc.cxx"

--- setup.py ----

from distutils.core import setup, Extension 
import numpy 
numpyinclude = numpy.__file__[:-12] + 'core/include/' 
testswig = Extension('_testswig', 
                     sources=['testswig_wrap.cxx'], 
                     include_dirs=['../', numpyinclude]) 

setup (name = 'testswig', 
       version = '0.1', 
       author      = "NoName", 
       description = """ """, 
       ext_modules = [testswig], 
       py_modules = ["testswig"])

-----建筑------

我正在一个包含所有3个文件和文件夹的文件夹中构建这个文件＆＃39; Eigen＆＃39;包含Eigen标头。命令是：

swig -c++ -python -I./ testswig.i 
python setup.py install

-------错误----------

然后我运行一个包含

的python文件

import testswig 
import numpy as np 
print testswig.test(np.array([[2,3],[4,5]]))

这给出了错误＆＃34; SystemError：新样式的getargs格式，但参数不是元组＆＃34;。

请注意以下几点： 1）相同的命令直接从python解释器运行。 2）如果函数没有返回Eigen :: VectorXd，或者没有采用Eigen：MatrixXd，它可以正常工作。

感谢您的时间。

Answer 1

在您的in typemap中，您有：

PyArrayObject *temp;
PyArg_ParseTuple($input, "O", &temp);

这是不正确的 - $input是一个PyObject，它已经从这个阶段的参数中提取出来，但不是一个元组，所以你想要：

PyArrayObject *temp=NULL;
if (PyArray_Check($input))
    temp = (PyArrayObject*)$input;

验证它是否为正确的类型，然后进行强制转换。

使用Swig用Python包装Eigen / C ++时出错

1 个答案: