2

I want to create a python wrapper for my C++ library. It would be cool, if there is a automatic conversion of std::vector to python lists and the other way round. Unfortunatly if I add this code to my Interface-file I still get errors in run-time.

%typemap(in) std::vector<float> value (std::vector<float> vIn) {
    int iLen = PySequence_Length($input); 
    for(unsigned int i = 0; i < iLen; i++) {
        PyObject *o = PySequence_GetItem($input, i);
        if (PyNumber_Check(o)) {
            vIn.push_back((float)PyFloat_AsDouble(o) );
        }
    }
    $1 = vIn;
}
%typemap(out) std::vector<float> {
    std::vector<float> vOut = $1;
    int iLen = vOut.size();
    $result = PyList_New(iLen);
    for(unsigned int i = 0; i < iLen; i++) {
        double fVal = vOut.at(i);
        PyObject *o = PyFloat_FromDouble((double) fVal);
        PyList_SetItem($result, i, o);
    }
}

Class header:

class TrainingSet {
    private:
        std::vector<std::vector<float> > m_vInputList;
        std::vector<std::vector<float> > m_vOutputList;
    public:
        void AddInput(const std::vector<float> &vIn);
    // ..

Python code:

trainSet = TrainingSet()
trainSet.AddInput([0.5, 0.5, 0.5])

Error:

File "runSOMNet.py", line 9, in <module>
trainSet.AddInput([0.5, 0.5, 0.5])
File "/home/dgrat/annetgpgpu/build/ANNet/ANPyNetCPU.py", line 674, in AddInput
def AddInput(self, *args): return _ANPyNetCPU.TrainingSet_AddInput(self, *args)
NotImplementedError: Wrong number or type of arguments for overloaded function      'TrainingSet_AddInput'.
Possible C/C++ prototypes are:
ANN::TrainingSet::AddInput(std::vector< float,std::allocator< float > > const &)
ANN::TrainingSet::AddInput(float *,unsigned int const &)
4

2 回答 2

3

The std_vector.i library in SWIG provides support for std::vector.

http://www.swig.org/Doc2.0/Library.html#Library_stl_cpp_library

You just need to tell SWIG about the template instantiations you want it to know about:

%include "std_vector.i"

namespace std {
    %template(FloatVector) vector<float>;
}

Note that the following Python code will work, but will incur an array copy:

for x in range(0, 3):
    list[x] = x
myModule.myFunction(list)

To do the same thing without incurring a copy, construct the list using the SWIG-generated proxy object constructor:

list = myModule.FloatVector()
for x in range(0, 3):
    list[x] = x
myModule.myFunction(list)
于 2013-05-13T18:22:17.393 回答
1

If you're going to define the %typemaps manually, you will also need a %typemap(check), something like:

%typemap(typecheck) std::vector<float>& {
    $1 = PySequence_Check($input) ? 1 : 0;
}

I believe the rule of thumb is that if you define a %typemap(in), you should also define a %typemap(check) --- otherwise, the generated code never gets to where it's put your %typemap(in).

于 2014-02-13T05:50:11.693 回答