如何使用 SWIG 让 python 切片与我的 c++ 数组类一起使用

Question

我有一个数组类，Array1D，用 c++ 定义，它基本上包装了 STL 向量类。我扩展了这个类，以便可以显示数组向量的各个元素。这是我的 SWIG 接口文件中的相关代码：

namespace std{
    %template(dblVector) vector<double>;
}

%extend Array1D{
    double __getitem__(int index) {
        return (*self)[index];
    }
}

这允许我在 python 中访问数组的各个元素：

>>> a = Array1D(10) # creates a c++ vector of length 10 with zeros
>>> a[0]
>>> 0

例如，我希望能够调用 a[1:3]，但是，当我尝试这个时，我得到了一个 TypeError：

TypeError: in method 'Array1D___getitem__', argument 2 of type 'int'

Answer 1

问题在于，python 在调用 getitem 的切片变体时传递了一个 Slice 对象，而您的函数定义需要一个 int。您需要编写一个以 PyObject* 作为参数的 getitem 版本，然后您必须在那里实现向量的切片。

我在写这篇文章时并没有进行实际测试，所以请谨慎对待。但我会做类似以下的事情。

%extend Array1D
{
    Array1D* __getitem__(PyObject *param) 
    {
        if (PySlice_Check(param))
        {
            /* Py_ssize_t might be needed here instead of ints */
            int len = 0, start = 0, stop = 0, step = 0, slicelength = 0;

            len = this->size(); /* Or however you get the size of a vector */

            PySlice_GetIndicesEx((PySliceObject*)param, len, &start, &stop, &step, &slicelength);

            /* Here do stuff in order to return an Array1D that is the proper slice
               given the start/stop/step defined above */
        }

        /* Unexpected parameter, probably should throw an exception here */
    }
}