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

时间:2011-06-04 19:34:19

标签: c++ python swig operator-keyword subscript

我有一个包含std::vector<Foo>的类,其中Foo是一个包含键,值,注释等的类。请注意,我有使用向量而不是a的原因字典。

我在C ++中重载了下标运算符,以便foos["Key Name"]将在向量中搜索具有键匹配“Key Name”的Foo对象(其中foosstd::vector<Foo>)。

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

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

2 个答案:

答案 0 :(得分:5)

在直接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 ++处理这个问题,我最好的猜测(不,我还没有验证过这个)就是你要在C ++中创建__getitem____setitem__。您可以直接在C ++代码中执行此操作,也可以使用SWIG中的%extend指令调用C ++ []运算符。

答案 1 :(得分:2)

Python索引器是__getitem____setitem__方法。

请参阅here了解如何实施它们。

或者您可以在更现代的SWIG中使用%extend clause