swig numpy 多个矩阵和数组输入

Posted 2023-02-16

【中文标题】swig numpy 多个矩阵和数组输入

【英文标题】：swig numpy multiple matrix and array inputs

【发布时间】：2013-08-03 00:15:45

【问题描述】：

我正在尝试使用 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);

而我的接口文件如下

%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 个论点以及这些论点是什么？谢谢！

【问题讨论】：

只是猜测。 nw 接受 10 个参数，而你使用 %apply 指令接受 4 个参数。所以需要提供6个参数。

我不确定我是否遵循...您是在暗示 %apply 指令不会将我的原始参数减少到 4 个参数？

【参考方案1】：

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

事实证明，SWIG 真的不喜欢我为 cstrings 制作的应用指令。

我应该改为以下指令。

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

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