SWIG包装C ++ for Python:将字符串列表转换为STL字符串的STL向量

时间:2011-12-12 01:56:30

标签: c++ python string vector swig

我想用SWIG包装一个C ++函数,它接受一个STL字符串向量作为输入参数:

#include <iostream>
#include <string>
#include <vector>

using namespace std;

void print_function(vector<string> strs) {
  for (unsigned int i=0; i < strs.size(); i++)
  cout << strs[i] << endl;
}

我想将它包装成一个名为`mymod'的模块中的Python函数:

/*mymod.i*/
%module mymod
%include "typemaps.i"
%include "std_string.i"
%include "std_vector.i"

%{
 #include "mymod.hpp"
%}

%include "mymod.hpp"

当我使用

构建此扩展时
from distutils.core import setup, Extension

setup(name='mymod',
  version='0.1.0',
  description='test module',
  author='Craig',
  author_email='balh.org',
  packages=['mymod'],
  ext_modules=[Extension('mymod._mymod',
                         ['mymod/mymod.i'],
                         language='c++',
                         swig_opts=['-c++']),
                         ],
  )

然后导入它并尝试运行它,我收到此错误:

Python 2.7.2 (default, Sep 19 2011, 11:18:13) 
[GCC 4.1.2 20080704 (Red Hat 4.1.2-48)] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> import mymod
>>> mymod.print_function("hello is seymour butts available".split())
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: in method 'print_function', argument 1 of type 'std::vector<  std::string,std::allocator< std::string > >'
>>> 

我猜这是说SWIG没有提供默认的类型映射,用于在Python字符串的Python列表和STL字符串的C ++ STL向量之间进行转换。我觉得这是他们默认提供的东西,但也许我不知道要包含的正确文件。那我怎么能让它发挥作用呢?

提前致谢!

2 个答案:

答案 0 :(得分:17)

您需要告诉SWIG您需要矢量字符串类型映射。它并没有神奇地猜测可能存在的所有不同的矢量类型。

这是由Schollii提供的链接:

//To wrap with SWIG, you might write the following:

%module example
%{
#include "example.h"
%}

%include "std_vector.i"
%include "std_string.i"

// Instantiate templates used by example
namespace std {
   %template(IntVector) vector<int>;
   %template(DoubleVector) vector<double>;
   %template(StringVector) vector<string>;
   %template(ConstCharVector) vector<const char*>;
}

// Include the header file with above prototypes
%include "example.h"

答案 1 :(得分:2)

SWIG确实支持将列表传递给以矢量作为值或const矢量引用的函数。 http://www.swig.org/Doc2.0/Library.html#Library_std_vector上的示例显示了这一点,我发布的内容无法显示任何错误。别的错了; python发现的DLL不是最新版本,标题中的using namespace std会混淆执行类型检查的SWIG包装器代码(注意.hpp中的“using namespace”语句通常是禁止的,因为它会提取所有内容从std到全局命名空间)等。