使用SWIG导向器将python中的numpy函数传递给C ++

Question

我的代码主要是在C ++中，但我想使用SWIG director选项在python中重新定义C ++虚函数，然后将该函数传递回C ++。 以下是一个简化示例。

这是我在C ++中的基类，具有虚拟功能的fun 。 fun接受(double* x, int n) ，它允许我将numpy数组传递给它。

class base{
public:
    virtual double fun(double* x, int n){return 0.0;}
};

我有另一个接受这个基类的类：

class main{
public:
    base* b_;

    main(base& b){
        b_ = &b; 
    }
};

这个类有一个构造函数，它接受一个基类并存储一个指向它的指针（ b_ ）。

我可以成功地将swig接口编译成一个名为mymodule的模块，确保启用mymodule （参见下面的SWIG接口文件），并且根据SWIG导向器文档，我继承了基类并重新定义了虚函数的fun ，如下所示：

import mymodule
class base(mymodule.base):
    def __init__(self):
        super(base,self).__init__()
    def fun(self,x):
        return x[0]

然后我把它传递到main类：

b = base()
m = mymodule.main(b)

但是，当我尝试m.b_.fun(array([1.,2.]))我得到一个TypeError ：

TypeError: 'SwigPyObject' object has no attribute '__getitem__'

有关如何使其工作的任何想法？ 我认为当我将b传递给mymodule.main时，C ++会将x视为double *指针，并且不知道如何处理它。

顺便说一下，这是我的swig界面文件：

%module(directors="1") mymodule

%{
#define SWIG_FILE_WITH_INIT
/* Includes the header in the wrapper code */
#include "myclasses.h"
%}

%feature("director") base;

%init %{
import_array();
%}

%apply (double* IN_ARRAY1, int DIM1) {(double* x, int n)};

/* Parse the header file to generate wrappers */
%include "myclasses.h"

Answer 1

它看起来像Python的所有权问题。

提供给SWIG的提示：

%new double *fun(double* IN_ARRAY1, int DIM1) {(double* x, int n)};

注意指向函数本身的指针。 当使用％new时，Python模块将为对象分配适当的所有权。 这就是为什么你可以在调用中使用值而不是变量。

Answer 2

导演似乎没有认识到你的Python实现的fun(double* x, int n) ; 它没有进行该功能的重载。

我怀疑原因是因为你的Python实现使用numpy作为参数。 请注意，导演将调用抽象函数或实现函数。

另一方面， %apply (double* IN_ARRAY1, int DIM1) {(double* x, int n)}仅声明用于在Python中包装函数的typemap逻辑，而不是Director。 请注意，这就是为什么b.fun(array([1.,2.])有效的原因！说，你应该向director声明typemap逻辑。

在这里你可以找到如何做到这一点：

%typemap(directorin,numinputs=1) (double *x, int n)
{
    npy_intp dim = $2;
    $input = PyArray_SimpleNewFromData(1, &dim, NPY_DOUBLE, (void *)$1);
}

希望这能解决你的问题和其他未来问题:)。 我遇到了类似的问题，并花了几天时间试图找出答案。