为cython包装枚举类

时间:2015-06-23 11:45:25

标签: c++ c++11 cython enum-class

我正在尝试将一个枚举类包装在一个c ++头文件中,以便在cython项目中使用。我已经google了一下,无法找到如何实现这一点 - 是否支持?

4 个答案:

答案 0 :(得分:6)

CPP类

enum class Color {red, green = 20, blue}; 

类型

的定义
cdef extern from "colors.h":
  cdef cppclass Color:
    pass

颜色类型的定义

cdef extern from "colors.h" namespace "Color":
  cdef Color red
  cdef Color green
  cdef Color blue

Python实现

cdef class PyColor:
  cdef Color thisobj
  def __cinit__(self, int val):
    self.thisobj = <Color> val

  def get_color_type(self):
    cdef c = {<int>red : "red", <int> green : "green", <int> blue : "blue"}
    return c[<int>self.thisobj]

答案 1 :(得分:2)

这是一个替代解决方案,它使用能够更改cython和C ++标识符的名称。

<强> header.hpp

namespace foo {
enum class Bar : uint32_t {
    BAZ,
    QUUX
};
}

<强> header.pxd

cdef extern from "header.hpp" namespace "foo::Bar":
    cdef enum Bar "foo::Bar":
        BAZ,
        QUUX

<强> main.pyx

from header cimport *
cdef void doit(Bar b):
    pass

doit(BAZ) # Not Bar.BAZ, which would have been nicer.

它有效告诉cython存在名为&#34; foo :: Bar&#34;的命名空间,并在其中放入C风格的枚举。为了抵消酒吧本来会成为&#34; foo :: Bar :: Bar&#34;这也被赋予了一个被覆盖的名称。这确实意味着Bar :: BAZ在cython中被称为BAZ,而不是Bar.BAZ,这将是enum类的更惯用的表示,但它似乎足够接近。

答案 2 :(得分:2)

另一种允许使用PEP-435 Enums中提到的Cython docs的替代方法如下:

foo.h

namespace foo {
enum class Bar : uint32_t {
    Zero = 0,
    One = 1
};
}

foo.pxd

from libc.stdint cimport uint32_t

cdef extern from "foo.h" namespace 'foo':

    cdef enum _Bar 'foo::Bar':
        _Zero 'foo::Bar::Zero'
        _One  'foo::Bar::One'


cpdef enum Bar:
    Zero = <uint32_t> _Zero
    One  = <uint32_t> _One

main.pyx

from foo cimport Bar

print(Bar.Zero)
print(Bar.One)

# or iterate over elements
for value in Bar:
    print(value)

答案 3 :(得分:2)

最新的 cython (3.x) 直接支持 c++ enum class,这里记录: https://cython.readthedocs.io/en/latest/src/userguide/wrapping_CPlusPlus.html#scoped-enumerations

这是一个例子:

// cpp header
enum class State: int
{
    Good,
    Bad,
    Unknown,
};


const char* foo(State s){
    switch (s){
        case State::Good:
            return "Good";
        case State::Bad:
            return "Bad";
        case State::Unknown:
            return "Unknown";
    }
}

Cython 的一面

cdef extern from "test.h":
    cpdef enum class State(int):
        Good,
        Bad,
        Unknown,

    const char* foo(State s)
    
def py_foo(State s):
    return foo(s)

调用 py_foo(State.Good) 返回 b'Good'