对c++结构体中的std::vector成员使用SWIG



我正在尝试使用SWIG来包装包含vector成员的c++结构体,并提供与Python的接口。我有一个使用以下三个文件(Ubuntu-16.04)的小测试用例:

一个像这样的Makefile:

FLAGS = -fPIC
PYTHONI = -I /usr/include/python3.5/
PYTHONL = -Xlinker -export-dynamic
all:
swig -Wall -c++ -python -o test.cxx test.i
g++ $(FLAGS) $(PYTHONI) -c test.cxx -o test_wrap.o
g++ $(PYTHONL) $(LIBFLAGS) -shared -o _test.so test_wrap.o -lstdc++

一个c++测试头文件,在这里定义了一个结构体(test.h):

#ifndef TEST_H
#define    TEST_H
#include <vector>
using std :: vector;
struct myStruct {

float var1; // m
vector<float> arr1; // m
vector<float> arr2; // C
int var2;
};
#endif

和一个SWIG接口文件(test.i):

%module test
%include "std_vector.i"
typedef std::vector<float> FloatVector;
namespace std {
%template(FloatVector) vector<float>;
};
%naturalvar myStruct::arr1;
%{
#include <vector>
#include "test.h"
%}
%include "test.h"

运行make all会创建一个test.py文件,该文件可以导入并创建myStruct类:

import test
foo = test.myStruct()

可以像这样设置var1和var2成员:

foo.var1 = 1.23
foo.var2 = 4

但是,不能对arr1或arr2成员进行编辑。这些都是类型'SwigPyObject',即使我试图在arr1上使用%naturalvar。我想做的是创建一个结构,用Python会话中的数据填充它,然后在对c++函数的其他调用中使用该结构。

在头文件中使用using是一个坏主意。SwigPyObjecttype表示类型类型映射不匹配。例如,可以这样做:

test.i:

%module test
%include <std_vector.i>
%template(FloatVector) std::vector<float>;
%naturalvar myStruct::arr1;
%naturalvar myStruct::arr2;
%inline %{
#include <vector>
struct myStruct {
float var1;
std::vector<float> arr1;
std::vector<float> arr2;
int var2;
};
%}

演示:

>>> import test
>>> foo = test.myStruct()
>>> foo.arr1
()
>>> foo.var1=1.23
>>> foo.var2=5
>>> foo.arr1=1,2,3
>>> foo.arr2=1.1,2.2
>>> foo
<test.myStruct; proxy of <Swig Object of type 'myStruct *' at 0x000001F94E7487E0> >
>>> foo.var1
1.2300000190734863
>>> foo.var2
5
>>> foo.arr1
(1.0, 2.0, 3.0)
>>> foo.arr2
(1.100000023841858, 2.200000047683716)

如果没有%naturalvar,值仍然可以分配,但需要:

>>> foo.arr1 = test.FloatVector([1,2,3])

最新更新