Numpy.i 多个不同类型数组的类型映射

Question

我有一个 C 函数，除其他参数外，它接受 3 个不同类型的数组： void create_signal(double ztab[], ..., int *pdata, ..., char ranging[], ...); 我想在使用 SWIG 的 Python 模块中使用这个函数。

我已经弄清楚如何使用 C 函数，该函数接受带有 numpy.i 类型映射的数组： %apply (double* INPLACE_ARRAY1, int DIM1) {(double xtab[], int length)} 例如，

甚至 2 个相同类型的数组： apply (double* INPLACE_ARRAY1, int DIM1) {(double xtab[], int length_x), (double ytab[], int length_y)}

但我不知道如果数组的类型不同（在我的例子中，int、double 和 char 是否可以这样做），或者我是否应该重写我自己的类型映射。

感谢您的帮助！

Answer 1

无论如何，这里有一个例子

test.h

#pragma once

void multipleInputs(const float* data, const int nData,
                    const float* pos, const int nPositions, const int nDim);

test.cpp

#include "test.h"

#include <cstdio>

void multipleInputs(const float* data, const int nData,
                    const float* pos, const int nPositions, const int nDim)
{
  if (nData > 0) {
    printf("%d'th element is: %f\n", (nData-1), data[nData-1]);
  }
  if ((nDim > 0) && (nPositions > 0)){
    printf("%d'th dimension of the %d'th element is: %f\n",
           (nDim-1), (nPositions-1), data[nData-1]);
  }
}

test.i

%module example
%{
  #define SWIG_FILE_WITH_INIT
  #include "test.h"
%}

%include "numpy.i"

%init {
  import_array();
}

%apply (float* IN_ARRAY1, int DIM1) \
{(const float* data, const int nData)}

%apply (float* IN_ARRAY2, int DIM1, int DIM2) \
{(const float* pos, const int nPositions, const int nDim)};

%include "test.h"

setup.py

import numpy

from setuptools import setup, Extension

setup(name="example",
      py_modules=["example"],
      ext_modules=[Extension("_example",
                             ["test.i","test.cpp"],
                             include_dirs = [numpy.get_include(), '.'],
                             swig_opts=['-c++', '-I.'],
      )]
)

test.py

import example
example.multipleInputs(np.ones((10),dtype=np.float32), 
                       np.ones((10,10),dtype=np.float32))

Numpy.i 多个不同类型数组的类型映射

Numpy.i typemap for multiple arrays of different types

c

python

arrays

swig

typemaps