Question

我正在尝试使用SWIG和Numpy类型映射在python中创建一个小C函数接口

此功能定义如下

void nw(int* D, int Dx, int Dy, int* mat, int mx, int my, char *xstr, int xL,char *ystr, int yL);

我的界面文件如下

%module nw

%{
    #define SWIG_FILE_WITH_INIT
    #include "nw.h"
%}

%include "numpy.i"

%init %{
  import_array();
%}
/*type maps for input arrays and strings*/
%apply (int* INPLACE_ARRAY2, int DIM1, int DIM2) {(int* D, int Dx, int Dy)}
%apply (int* IN_ARRAY2, int DIM1, int DIM2) {(int* mat, int mx, int my)}
%apply (char* IN_ARRAY, int DIM1){(char *xstr, int xL),(char *ystr, int yL)}

%include "nw.h"

为了测试它，我使用了以下输入

D = numpy.zeros((5,5),numpy.int)
mat = numpy.array([[1, 0, 0, 0, 0, 0],
                   [0, 1, 0, 0, 0, 0],
                   [0, 0, 1, 0, 0, 0],
                   [0, 0, 0, 1, 0, 0],
                   [0, 0, 0, 0, 1, 0],
                   [0, 0, 0, 0, 0, 1]],numpy.int)
x = numpy.array(list("ABCD"))
y = numpy.array(list("ABCD"))
import nw
nw.nw(D,mat,x,y)

但是当我运行它时，我得到以下

TypeError: nw() takes exactly 6 arguments (4 given)

我对如何定义这些参数感到困惑。这里有没有人知道为什么有6个参数以及这些参数是什么？谢谢！

Answer 1

好吧，我想我已经找到了问题。

事实证明，SWIG 真的不喜欢我为cstrings做的apply指令。

我应该是以下指令。

%apply (char *STRING, int LENGTH) {(char *xstr, int xL),(char *ystr, int yL)}

应该更仔细地遵循食谱了哈哈

swig numpy多个矩阵和数组输入

1 个答案: