Question

我在numpyarray和C ++中有二维数据 - 我想对这些数据执行一些操作的代码。使用swig和distutils以及numpy.i我设法将所有内容编译成python扩展“goldstein”，它提供了一个函数unwrap2d。我用它来测试它

import goldstein

data = np.ascontiguousarray(data_temp, dtype='double')
mask = np.ascontiguousarray(mask_temp, dtype='uint16')
outdata = np.ones_like(data)

goldstein.unwrap2d(data,mask,outdata)

我得到TypeError: array cannot be safely cast to required type。任何人都可以指出我如何以正确的方式传递这些数组？

供参考：为了创建模块，我使用了interface-file

%define DOCSTRING
"Wrapper for c++-code"
%enddef

%module(docstring=DOCSTRING) goldstein

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

/*include numpy typemaps*/
%include "numpy.i"
/*initialize module*/
%init %{
    import_array();
%}

%rename(unwrap2d) phase_unwrapping_func;

/* typemaps for the arrays*/
%apply (int DIM1,int DIM2,float* IN_ARRAY2) {(int xsize,int ysize,float* in)};
%apply (int DIM1,int DIM2,unsigned short* IN_ARRAY2) {(int x1,int y1,unsigned short* mask)};
%apply (int DIM1,int DIM2,float* INPLACE_ARRAY2) {(int x2,int y2,float* out)};

/*wrapper function calling the original phase_unwrapping using only the needed parameters*/
%inline %{
void phase_unwrapping_func(int xsize,int ysize,float* in,int x1,int y1,unsigned short* mask,int x2,int y2,float* out) {
    phase_unwrapping(xsize, ysize, in, mask, out);
    }
%}

Answer 1

@Jaimie当然是对的。像'double'这样的'float'在numpy中是64位，这就是为什么我没有再次检查，我只记得它并不重要。但是c ++中的float显然需要32位，其中numpy是'float32'。谢谢！

将2d numpy数组传递给C ++时的TypeError

1 个答案: