Question

I have searched on the web and didn’t get success. I’m wrapping the sample code below to Python (using SWIG):

class atomo {
public:
    int i;
    atomo(int a) {
        i = a;
    };      
};

class funa {
public:
    atomo *lista[3];

    funa() {
        lista[0] = new atomo(1);
        lista[1] = new atomo(2);
        lista[2] = new atomo(3);
    };
};

But Python can’t iterate over or access lista using the comands

>>> test = myModule.funa()
>>> test.lista[0]
      Traceback (most recent call last):
      File "<stdin>", line 1, in <module>
      File "<stdin>", line 6, in __iter__
      TypeError: 'SwigPyObject' object is not subscriptable

>>> for i in test.lista:
>>>     print(i)
      Traceback (most recent call last):
      File "<stdin>", line 1, in <module>
      File "<stdin>", line 6, in __iter__
      TypeError: 'SwigPyObject' object is not subscriptable

How can I make lista iterable? There is a way to use Python lists instead of C++ arrays?

My Python version is 3.2 and I’m using SWIG 2.0.4 with g++ 4.6.1

Thanks

Answer 1

It’s a little unclear from your question if you want to use std::vector or an array of your own types.

For std::vector, given some C++ like:

#include <vector>
#include <string>

struct foo {
  std::string name;
};

inline std::vector<foo> test() {
  std::vector<foo> ret;
  foo instance;
  instance.name = "one";
  ret.push_back(instance);
  instance.name = "two";
  ret.push_back(instance);
  return ret;
}

You can wrap it with %template, pyabc.i and std_vector.i e.g.:

%module test

%{
#include "test.h"
%}

%include "pyabc.i"
%include "std_vector.i"

%include "test.h"

%template (FooVector) std::vector<foo>;

which will behave intuitively on the Python type. You’ll need to call SWIG with something like:

swig -python -c++ -py3 -extranative test.i

If the idea is to wrap a “custom” container to behave intuitively on the Python side I gave a detailed example in a previous answer.