如何使用SWIG枚舉枚舉成員

Question

我可以將C++ enum作爲真實實體而不是一組常量公開給SWIG，以便我可以在Python代碼中枚舉它們？

Answer 1

我面臨同樣的問題。我希望SWIG很快支持C++ 11的enum class。

下面是說服夜風把枚舉結構中的一個黑客：

#ifdef SWIG 
%rename(MyEnum) MyEnumNS; 
#endif 

struct MyEnumNS 
{ 
    enum Value { Value1, Value2, Value3 }; 
}; 
typedef MyEnumNS::Value MyEnum; 

在.cpp代碼，你現在必須使用MyEnum::Value1和Python代碼是MyEnum.Value1。雖然令人費解，但typedef可以防止必須更改在任何地方使用枚舉的現有代碼，並且SWIG％重命名使枚舉在SWIG包裝器中具有相同的名稱。

在Python中，你可以用一點代碼枚舉值：

def values(enum): 
    return [(k,v) for k,v in vars(enum).items() if isinstance(v,int)] 

這不是漂亮，我喜歡看到一個更好的解決方案。

Answer 2

我敢肯定，你要找的是...
typemaps：由於類型的處理是如此重要的包裝代碼生成，SWIG使得它能夠被用戶完全定義（或重定義）。爲此，使用特殊的％typemap指令。 （SWIG Doc2.0）

對於您可能需要的有關typemaps的所有信息，請參閱SWIG文檔的鏈接。 http://www.swig.org/Doc2.0/Typemaps.html#Typemaps_nn2

Typemaps應該允許您告訴SWIG將C++枚舉轉換爲您想要的python對象。

Answer 3

我們可以做一些讓你在Python中枚舉的東西，相對較少入侵它所包裝的C++頭文件。例如，如果我們有一個頭文件：

#ifndef PYTHON_ENUM 
#define PYTHON_ENUM(x) enum x 
#endif 

PYTHON_ENUM(TestName) { 
    foo=1, 
    bar=2 
}; 

PYTHON_ENUM(SomeOtherName) { 
    woof, 
    moo 
}; 

它擴展到只是在C++中常規的枚舉，但足以作爲一個頭文件暴露在Python枚舉成員。

使用%typemap(constcode)我們可以在我們的Python模塊中爲enum注入一些額外的東西，但我們需要知道枚舉的名稱來完成此操作; SWIG typeinfo對象就好像它是一個int。因此，我們在我們的PYTHON_ENUM宏中使用了一些黑客技術來將枚舉的名稱存儲在自定義的類型映射中。

%module test 
%{ 
#include "test.h" 
%} 

%typemap(constcode) int { 
    PyObject *val = PyInt_FromLong(($type)($value)); 
    SWIG_Python_SetConstant(d, "$1", val); 
    const char *name = "$typemap(enum_realname,$1_type)"; 
    PyObject *e = PyDict_GetItemString(d, name); 
    if (!e) PyDict_SetItemString(d, name, e = PyDict_New()); 
    PyDict_SetItemString(e, "$value", val); 
} 
#define PYTHON_ENUM(x) \ 
     %typemap(enum_realname) int "x"; \ 
     %pythoncode %{ \ 
     x = _test.x\ 
     %} \ 
     enum x 

%include "test.h" 

這將在每個具有鍵/值對的枚舉的中間模塊中創建一個PyDict。還有一些%pythoncode粘合劑將中間模塊中的PyDict綁定到暴露的模塊中。 （我不知道如何通過名稱引用中間模塊，除了硬編碼爲_test - 根據需要更改）。

這是足夠的，我就可以使用它作爲：

Python 2.7.3 (default, Aug 1 2012, 05:16:07) 
[GCC 4.6.3] on linux2 
Type "help", "copyright", "credits" or "license" for more information. 
>>> import test 
>>> print test.SomeOtherName 
{'woof': 0, 'moo': 1} 
>>> print test.TestName 
{'foo': 1, 'bar': 2} 
>>>