pybind11 эмулирует поведение Python Enum - PullRequest
Новая
       11

pybind11 эмулирует поведение Python Enum

0 голосов
/ 22 сентября 2018

Я использую pybind11 для предоставления интерфейса Python для моей библиотеки C ++.Моя библиотека содержит перечисления, для которых я предоставил вспомогательные функции, позволяющие перебирать значения перечислений и преобразовывать их в строку;в качестве примера 

template <typename Enum>
struct iter_enum
{
    struct iterator
    {
        using value_type = Enum;
        using difference_type = ptrdiff_t;
        using reference = const Enum&;
        using pointer = const Enum*;
        using iterator_category = std::input_iterator_tag;

        iterator(Enum value) : cur(value) {}

        reference operator * () { return cur; }
        pointer operator -> () { return &cur; }
        bool operator == (const iterator& other) { return cur == other.cur; }
        bool operator != (const iterator&other) { return!(*this == other); }
        iterator& operator ++ () { if (cur != Enum::Unknown) cur = static_cast<Enum>(static_cast<std::underlying_type_t<Enum>>(cur) + 1); return *this; }
        iterator operator ++ (int) { iterator other = *this; ++(*this); return other; }
    private:
        Enum cur;
    };

    iterator begin()
    {
        return iterator(static_cast<Enum>(0));
    }

    iterator end()
    {
        return iterator(Enum::Unknown);
    }
};

enum class Colour : char
{
    Black = 0,
    Blue,
    Red,
    Yellow,
    Unknown
};

const char* to_string(Colour colour)
{
    switch (colour) {
    case Colour::Black: return "Black";
    case Colour::Blue: return "Blue";
    case Colour::Red: return "Red";
    case Colour::Yellow: return "Yellow";
    default: return "Unknown";
    }
}

int main()
{
    for (auto colour : iter_enum<Colour>())
        std::cout << to_string(colour) << '\n';
}

При переносе такого кода с использованием pybind11 в настоящее время я определяю итератор как статическую функцию-член, 

 pybind11::enum_<Colour>(m, "Colour")
    .value("Black", Colour::Black)
    // add other values and __str__
    .def_static("iter", []() { 
        iter_enum<Colour> i; 
        return pybind11::make_iterator(i.begin(), i.end(); 
    });

, однако я хотел бы иметь возможность работать на Pythonside напишите что-то вроде 

for c in Colour:
    print("Colour")

, как я бы сделал для перечисления, определенного с помощью Python enum.Enum.Я знаю, что мог бы реализовать это в Python с использованием метакласса, но я не могу понять из документации по pybind11, которую я мог найти для метаклассов здесь и примера для github (строка 334)как написать такой метакласс на конце C ++.Из того, что я могу сказать, похоже, что метакласс должен быть представлен с использованием типа C API PyObject.

Любая информация относительно того, нахожусь ли я на правильном пути, и если да, то как написать такойmetaclass, который использует мой класс iter_enum, был бы очень признателен.

...