如何处理我的 C++ 中的切片__getitem__ 函数（由 SWIG 使用）

Question

我正在为 C++ 类开发 Python 绑定，该类扩展了 std::vector 实例化。为了让 Python 下标运算符适用于此类，我添加了 __getitem__ 函数，如下所示（我删除了不相关的代码和错误处理）：

class Column;

typedef vector<Column*> MetaDataBase;

class MetaData : public MetaDataBase {
public:

#ifdef SWIGPYTHON
Column* __getitem__(int i) { return (*this)[i]; }
#endif
};

上面的代码在 Python 中可以正常访问单个元素，但它不能为切片工作。

好的，所以我知道我需要将函数参数的类型更改为 PyObject * 并使用 PySlice_Check 来查看函数是否应该返回 PyList.

这很好，没有问题。但是因为有时我必须从函数返回 PyList，所以 __getitem__ 返回值的类型也必须是 PyObject* 并且我不能依靠 SWIG 将我的 C++ 类型 (Column *) 转换为包装类。另外，在创建切片时，我需要“手动”将 Column* 转换为 PyObject*，然后再将其插入 PyList。

我该怎么做呢？

Answer 1

我认为对于带有 SWIG+Python 的 std::vector 来说，有一个更简单的解决方案。 SWIG 的 Python 代码生成已经支持很好地包装一些 STL 容器。

如果您在模块接口的开头添加：

%include "pyabc.i"
%include "std_vector.i" // Assuming you don't already

以及某处，如果您没有这样做：

%template(MetaDataBase) std::vector<Column*>;

那么这将导致包装的 std::vector 满足 Python 的 MutableSequence。 （我认为这应该足以实现您在 Python 方面寻找的目标，您可能还需要使用 -extranative 调用 SWIG）。

另外值得注意的是 - 对于您当前的 __getitem__ ，您可以使用类似以下内容在 SWIG 接口文件中声明和定义它：

%extend MetaData {
   Column* __getitem__(int i) { return (*self)[i]; }
};

这允许您在不使用 SWIG+“污染”您的“正常”头文件的情况下执行此操作Python特定代码。