元组由SWIG为C++向量生成的Python包装器返回

2022-05-28 00:00:00 python swig c++

对于以下C++API:

std::vector<float> get_sweep_points()
{
    return program->sweep_points;
}

Swig生成一个包装器,它返回一个元组(),而不是一个列表[]。为什么?如何强制Swig将列表返回到python。


解决方案

如果您使用std_vector.i,您将获得由std_vector.i实现的类型映射。如果你不喜欢,你必须写你自己的类型映射。

下面是一个类型映射,用于覆盖默认行为(无错误检查)并返回列表而不是元组:

%typemap(out) std::vector<int> (PyObject* obj) %{
    obj = PyList_New($1.size());
    for(auto i = 0; i < $1.size(); ++i)
        PyList_SET_ITEM(obj, i, PyLong_FromLong($1[i]));
    $result = SWIG_Python_AppendOutput($result, obj);
%}

当然您也可以只做v = list(get_sweep_points()),现在它是一个列表:^)

相关文章