swig，从列表到非标准向量

swig, from list to non-standard vector

本文关键字：非标准 向量 列表 swig      更新时间：2023-10-16

我想设置typemap，这样我们就可以传递Python列表，而不是非标准向量。

在C++中，我有

template<typename T>
class mm_vector
{
void set_mm_vector(const mm_vector * copy);
}

我希望能够将Python列表作为参数传递，所以我在.I文件中指定：

// python list into vec_of_ints: from Python to C++
%typemap(in) AMM::mm_vector<int>*
{
    int i;
    if (!PyList_Check($input))
    {
      PyErr_SetString(PyExc_ValueError, "Expecting a list");
      return NULL;
    }
    Py_ssize_t size = PyList_Size($input); //get size of the list
    for (i = 0; i < size; i++)
    {
      PyObject *s = PyList_GetItem($input,i);
      if (!PyInt_Check(s))
        {
         PyErr_SetString(PyExc_ValueError, "List items must be integers");
         return NULL;
        }
      $1->push_back((int)PyInt_AS_LONG(s)); //put the value into the array
    }
}

当我试着运行这些线路

l=[0,1]
v = mm.vec_of_ints()
v.set_mm_vector(l)

我有以下错误：

File "...", line 1295, in set_mm_vector
def set_mm_vector(self, *args): return _pyamt.vec_of_ints_set_mm_vector(self, *args)
ValueError: Expecting a list

如果有任何建议，我将不胜感激！！！

SWIG内置了对向量和模板的支持，因此您不必从头开始实现它。这里有一个简短的例子：

%module vec
// Include the built-in support for std::vector
%include <std_vector.i>
// Tell SWIG about the templates you will use.
%template() std::vector<int>;
// %inline adds the following code to the wrapper and exposes its interface via SWIG.
%inline %{
class Test {
    std::vector<int> m_v;
public:
    void set(const std::vector<int>& v) { m_v = v;}
    std::vector<int> get() const {return m_v;}
};
%}

输出：

>>> import vec
>>> t=vec.Test()
>>> t.set([1,2,3])
>>> t.get()
(1, 2, 3)
相关文章：