使用SWIG在Python中公开std :: list成员

时间:2015-11-02 18:23:42

标签: python c++ swig

在输出中使用std :: list或在C ++中输入函数时,我能够使用以下类型映射。这些类型映射允许我将列表对象用作Python中的标准列表。

但是,我还没有找到用于std :: list的typemap,它是我的C ++类的公共成员。

MyWork.h

class MyWork
{
public:

   // Functions

   void myFunc1(std::list<MyClass> my_list); // OK

   std::list<MyClass> myFunc2(); // OK

   // Properties

   std::list<MyClass> MyList; // ????
};

SWIG Typemaps

%typemap(out) std::list<MyClass>
{
PyObject* outList = PyList_New(0);

int error;

std::list<MyClass>::iterator it;
for ( it=$1.begin() ; it != $1.end(); it++ )
{
    PyObject* pyMyClass = SWIG_NewPointerObj(new MyClass(*it), SWIGTYPE_p_MyClass, SWIG_POINTER_OWN );

    error = PyList_Append(outList, pyMyClass);
    Py_DECREF(pyMyClass);
    if (error) SWIG_fail;       
}

$result = outList;
}

%typemap(in) std::list<MyClass>
{
//$input is the PyObject
//$1 is the parameter

if (PyList_Check($input))
{
    std::list<MyClass> listTemp;

    for(int i = 0; i<PyList_Size($input); i++)
    {
        PyObject* pyListItem = PyList_GetItem($input, i);

        MyClass* arg2 = (MyClass*) 0 ;

        int res1 = 0;
        void *argp1;

        res1 = SWIG_ConvertPtr(pyListItem, &argp1, SWIGTYPE_p_MyClass,  0  | 0);
        if (!SWIG_IsOK(res1))
        {
            PyErr_SetString(PyExc_TypeError,"List must only contain MyClassobjects");
            return NULL;
        }  
        if (!argp1)
        {
            PyErr_SetString(PyExc_TypeError,"Invalid null reference for object MyClass");
            return NULL;
        }

        arg2 = reinterpret_cast< MyClass* >(argp1);
        listTemp.push_back(*arg2);
    }

    $1 = listTemp;
}
else
{
    PyErr_SetString(PyExc_TypeError,"Wrong argument type, list expected");
    return NULL;
}
}

1 个答案:

答案 0 :(得分:0)

您可以使用SWIG提供的类型地图。添加到SWIG界面文件:

%include "std_list.i"
%include "MyClass.h" //(or declaration of MyClass)
%template(MyClassList) std::list<MyClass>;
%include "MyWork.h"

完成!

您可以在SWIG文档中找到实际问题的答案(访问结构成员变量需要哪些类型映射){/ 3}}:

  

为类生成访问器的包装器代码来自指针类型映射。

即,您需要定义

%typemap(in) std::list<MyClass> *
%typemap(out) std::list<MyClass> *

但是,如同一部分所述,使用%naturalvar指令可能是有意义的,然后访问者将使用const &类型映射,即,您需要定义

%typemap(in) const std::list<MyClass> &
%typemap(out) const std::list<MyClass> &