python - swig numpy 多矩阵和数组输入

Question

我正在尝试使用 SWIG 和 Numpy 类型映射将一个小型 C 函数连接到 python 中

这个函数定义如下

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

而我的接口(interface)文件如下

%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 真的不喜欢我为 cstring 制作的应用指令。

我应该改为使用以下指令。

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

应该更仔细地看食谱哈哈