如何使用swig Python重载下标运算符

Question

我有一个包含一个类std::vector<Foo>,其中Foo是包含键,值,注释等,请注意,是有原因的,为什么我使用的是矢量,而不是一本字典类.

我已经在C++中重载了下标运算符,这样foos["Key Name"]就可以在向量中搜索一个Foo对象,其键与"Key Name"匹配(其中foos是a std::vector<Foo>).

我使用SWIG为我的库创建一个Python包装器,我真的希望这个下标操作符扩展到Python.换句话说,我希望能够foos["Key Name"]在Python中使用for来查找向量中的对象.

有关如何使SWIG识别下标运算符并在Python中重载它的任何提示？我有点惊讶,我找不到人们在网上这样做的例子.我猜大多数人只是使用了一个std::map并将SWIG转换为Python dict.

Answer 1

在直接Python中,如果要重载下标运算符,则需要创建一个类__getitem__和__setitem__类方法.举个简单的例子:

class MyClass(object):
    def __init__(self):
        self.storage = {}

    def __getitem__(self, key):
        return self.storage[key]

    def __setitem__(self, key, value):
        self.storage[key] = value

所以,如果你想拥有C++处理这个问题,我最好的猜测(不,我还没有证实这一点)是您将创建一个__getitem__和__setitem__C++中.您可以直接在C++代码中执行此操作,也可以使用%extendSWIG中的指令调用C++ []运算符.