Реализовать поддержку std :: vector без std_vector.i - PullRequest
Новая
       26

Реализовать поддержку std :: vector без std_vector.i

0 голосов
/ 23 января 2012

Хорошо, я уже задал 2 вопроса о своей проблеме, и, несмотря на то, что ответы были действительно полезными, я не могу найти оптимальное решение для моей проблемы. Позвольте мне объяснить мою главную цель / проблему сейчас.

Из-за некоторых ограничений я не могу использовать std_vector.i в своем интерфейсе swig, но мне нужно использовать объект C ++ (вектор векторов строки) vector<vector<string>> в Python. Я реализовал решение, в котором я конвертирую все vector<vector<string> > в Python «Список списков», в котором я делаю следующие преобразования: каждая строка C ++ в строку Python, используя PyString_FromString() каждый vector<string> в списки Python l1, l2, l3, l4 ... и, наконец, vector<vector<string> > в список Python, содержащий элементы l1, l2, l3, l4 .. в качестве элементов.

Хотя вышеупомянутое решение работает нормально, и я могу получить доступ к строковым значениям в Python, но это решение не выглядит для меня оптимальным.

Я бы предпочел класс (без использования std_vector.i), чей объект я могу передать в качестве аргумента функции для заполнения значениями, и после возвращения из функции я смогу получить доступ к значениям, используя ob[0][0] и т. Д. В таким образом мне придется сделать только одно преобразование (строка C ++ в строку python) для каждого доступного значения в __getitem__. Но я не знаю, как определить класс, представляющий vector<vector<string> > в Python, без использования %template.

1 Ответ

1 голос
/ 24 января 2012

Я собрал пример минимальной оболочки для 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
>>>

В настоящее время не вызывается правильный тип исключения Python в __getitem__. Вы можете сделать это либо с помощью %include "exception.i", либо с помощью %exception и написать свои try / catch вокруг $action.

Возможно, вы захотите предоставить аналогичную реализацию __setitem__, чтобы сделать это полезным.

Это, вероятно, не быстрее, чем std_vector.i, или ваша домашняя карта типа brew, которая напрямую преобразуется в список списков Python. В целом, хотя я не думаю, что делать это так - хорошая идея - использование существующей реализации std_vector.i вместо изобретения колеса кажется гораздо более логичным.

...