在没有std_vector.i的情况下实现对std :: vector的支持

时间:2012-01-23 08:44:12

标签: swig stdvector

好的,我已经问了两个关于我的问题的问题,尽管回复真的很有用,但我无法找到解决问题的最佳方案。让我现在解释一下我的主要目标/问题。

由于一些限制,我不能在swig接口中使用std_vector.i,但是我需要在Python中使用C ++对象(字符串向量)vector<vector<string>>。我实现了一个解决方案,我将整个vector<vector<string> >转换为Python“列表列表”,其中我正在进行以下转换: 使用PyString_FromString()将每个C ++字符串转换为Python String 每个vector<string>到Python列表l1,l2,l3,l4 ...... 最后vector<vector<string> >到包含l1,l2,l3,l4 ..作为元素的Python列表。

虽然,上面的解决方案工作正常,但我能够在Python中访问字符串值,但这个解决方案对我来说并不是最佳选择。

我更喜欢一个类(不使用std_vector.i),其对象我可以作为函数参数传递以填充值,并且从函数返回后我应该能够使用ob[0][0]等访问值通过这种方式,对于__getitem__中访问的每个值,我只需要进行一次转换(C ++字符串到python字符串)。但我不知道如何在不使用vector<vector<string> >的情况下在Python中定义代表%template的类。

1 个答案:

答案 0 :(得分:1)

我已经为std::vector<std::vector<std::string > >编写了一个最小包装器的示例,它不包含任何额外的SWIG文件(例如std_vector.i和std_string.i)。

我还整理了一个小的头文件来测试我的实现:

#include <vector>
#include <string>
#include <algorithm>
#include <iterator>
#include <iostream>

inline void print_vec(const std::vector<std::string>& v) {
  std::copy(v.begin(),v.end(), std::ostream_iterator<std::string>(std::cout, "\n"));
}

inline void print_vec_vec(const std::vector<std::vector<std::string> >& v) {
  std::for_each(v.begin(),v.end(),print_vec);
}

std::vector<std::vector<std::string> > make() {
  static std::vector<std::string> test1;
  static std::vector<std::string> test2;

  static std::vector<std::vector<std::string> > ret;
  test1.push_back("hello");
  test2.push_back("world");
  test2.push_back("another");
  ret.push_back(test1);
  ret.push_back(test2);
  return ret;
}

这是我能想到的最小的实现,它可以有效地运用生成的界面。

我写的SWIG接口提供了std::vector的骨架定义 - 足以说服SWIG实际包装东西。我们还针对我们关心的两个案例进行了扩展,以提供__getitem__的实现,这是您希望能够使用的obj[x][y]语法的最低要求。

%module Test

%{
#include "test.hh"
%}

namespace std {
template <typename T>
class vector {
};
}

%extend std::vector<std::vector<std::string> > {
  std::vector<std::string> __getitem__(unsigned i) throw(std::out_of_range) {
    return $self->at(i);
  }
}

%extend std::vector<std::string> {
  const char * __getitem__(unsigned i) throw(std::out_of_range) {
    return $self->at(i).c_str();
  }
}

%template (VecString) std::vector<std::string>;
%template (VecVecString) std::vector<std::vector<std::string> >;

%include "test.hh"

有一个c_str()的技巧可以避免包含std_string.i。这个界面允许我在Python中做这样的事情:

Python 2.7.1+ (r271:86832, Apr 11 2011, 18:05:24) 
[GCC 4.5.2] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> import Test
>>> t=Test.make()
>>> print t[0][0]
hello
>>>

它目前没有在__getitem__中引发正确类型的Python异常。您可以使用%include "exception.i"%exception进行此操作,并在try周围编写自己的catch / $action

您可能还希望提供__setitem__的类似实现,以使其有用。

这可能不比std_vector.i快,或者你的家庭brew类型映射直接转换为Python列表。总的来说,虽然我不认为这样做是个好主意 - 使用现有的std_vector.i实现而不是重新发明轮子似乎更合乎逻辑。