Jak wyliczyć członków enum za pomocą SWIG

Question

Czy mogę wystawić C++ enum na SWIG jako rzeczywistej jednostki zamiast zestawu stałych, więc mogę wyliczyć nad nimi w kodzie Pythona?

Answer 1

Napotkałem ten sam problem. Mam nadzieję, że SWIG wkrótce obsługuje C++ 11: enum class.

Oto hack, który przekonuje SWIG umieścić teksty stałe w strukturze:

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

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

W .cpp kod, który teraz musi używać MyEnum::Value1 w kodzie Pythona to MyEnum.Value1. Chociaż zawikłany, typedef uniemożliwia zmianę istniejącego kodu, który używa enum wszędzie, a zmiana nazwy SWIG% sprawia, że ​​enum mają tę samą nazwę w opakowaniu SWIG.

W Pythonie można wyliczyć wartości z odrobiną kodu:

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

To nie jest ładna, i chciałbym zobaczyć lepsze rozwiązanie.

Answer 2

jestem całkiem pewny szukasz są ...
typemaps: Ponieważ rodzaj manipulacja jest tak centralny do generowania kodu wrapper, SWIG pozwala mu być całkowicie zdefiniowana (lub nowo) przez użytkownika. Aby to zrobić, używana jest specjalna dyrektywa% typemap. (SWIG Doc2.0)

Aby uzyskać wszystkie informacje, jakich można potrzebować w przypadku map typów, tutaj znajduje się link do dokumentacji SWIG na ten temat. http://www.swig.org/Doc2.0/Typemaps.html#Typemaps_nn2

Typemaps powinno umożliwić informowanie SWIG o konwersji wyrażeń w języku C++ na żądane obiekty Pythona.

Answer 3

Możemy stworzyć coś, co pozwoli ci wyliczyć w Pythonie, przy stosunkowo niewielkiej ingerencji w nagłówki C++, które zawija. Na przykład, jeśli mamy plik nagłówka:

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

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

PYTHON_ENUM(SomeOtherName) { 
    woof, 
    moo 
}; 

Rozszerza się być zwykłym enum w C++, ale jest wystarczający jako plik nagłówka narazić członków enum w Pythonie.

Używając %typemap(constcode) możemy wprowadzić dodatkowe elementy do naszego modułu Pythona dla wyliczenia, ale musimy znać nazwę enum, aby to zrobić; Obiekt typu swiss dla SWIG jest taki, jak gdyby był to int. Dlatego używamy odrobiny hackowania w naszym makrze PYTHON_ENUM, aby zapisać nazwę wyliczenia w niestandardowej mapie typów.

%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" 

Spowoduje to utworzenie PyDict w module pośrednim dla każdego wyliczenia z parami klucz/wartość. Jest tam również pewna ilość kleju %pythoncode, aby związać PyDict w module pośrednim z eksponowanym modułem. (Nie jestem pewien, jak odnieść się do modułu pośredniego po nazwie w nim, inne niż na stałe oznaczony jako _test - zmiana w razie potrzeby).

Wystarcza to, że mogę użyć go jako:

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} 
>>>