mirror of
https://github.com/pybind/pybind11.git
synced 2024-11-22 21:25:13 +00:00
Fix scoped enums and add scoped enum example
PR #309 broke scoped enums, which failed to compile because the added: value == value2 comparison isn't valid for a scoped enum (they aren't implicitly convertible to the underlying type). This commit fixes it by explicitly converting the enum value to its underlying type before doing the comparison. It also adds a scoped enum example to the constants-and-functions example that triggers the problem fixed in this commit.
This commit is contained in:
parent
39ff2d0140
commit
613541947a
@ -14,6 +14,11 @@ enum EMyEnumeration {
|
||||
ESecondEntry
|
||||
};
|
||||
|
||||
enum class ECMyEnum {
|
||||
Two = 2,
|
||||
Three
|
||||
};
|
||||
|
||||
class ExampleWithEnum {
|
||||
public:
|
||||
enum EMode {
|
||||
@ -41,6 +46,10 @@ float test_function3(int i) {
|
||||
return (float) i / 2.f;
|
||||
}
|
||||
|
||||
void test_ecenum(ECMyEnum z) {
|
||||
std::cout << "test_ecenum(ECMyEnum::" << (z == ECMyEnum::Two ? "Two" : "Three") << ")" << std::endl;
|
||||
}
|
||||
|
||||
py::bytes return_bytes() {
|
||||
const char *data = "\x01\x00\x02\x00";
|
||||
return std::string(data, 4);
|
||||
@ -56,6 +65,7 @@ void init_ex_constants_and_functions(py::module &m) {
|
||||
m.def("test_function", &test_function1);
|
||||
m.def("test_function", &test_function2);
|
||||
m.def("test_function", &test_function3);
|
||||
m.def("test_ecenum", &test_ecenum);
|
||||
m.attr("some_constant") = py::int_(14);
|
||||
|
||||
py::enum_<EMyEnumeration>(m, "EMyEnumeration")
|
||||
@ -63,6 +73,11 @@ void init_ex_constants_and_functions(py::module &m) {
|
||||
.value("ESecondEntry", ESecondEntry)
|
||||
.export_values();
|
||||
|
||||
py::enum_<ECMyEnum>(m, "ECMyEnum")
|
||||
.value("Two", ECMyEnum::Two)
|
||||
.value("Three", ECMyEnum::Three)
|
||||
;
|
||||
|
||||
py::class_<ExampleWithEnum> exenum_class(m, "ExampleWithEnum");
|
||||
exenum_class.def_static("test_function", &ExampleWithEnum::test_function);
|
||||
py::enum_<ExampleWithEnum::EMode>(exenum_class, "EMode")
|
||||
|
@ -6,6 +6,7 @@ sys.path.append('.')
|
||||
from example import test_function
|
||||
from example import some_constant
|
||||
from example import EMyEnumeration
|
||||
from example import ECMyEnum, test_ecenum
|
||||
from example import EFirstEntry
|
||||
from example import ExampleWithEnum
|
||||
from example import return_bytes
|
||||
@ -20,6 +21,8 @@ print(test_function())
|
||||
print(test_function(7))
|
||||
print(test_function(EMyEnumeration.EFirstEntry))
|
||||
print(test_function(EMyEnumeration.ESecondEntry))
|
||||
test_ecenum(ECMyEnum.Three)
|
||||
test_ecenum(ECMyEnum.Two)
|
||||
print("enum->integer = %i" % int(EMyEnumeration.ESecondEntry))
|
||||
print("integer->enum = %s" % str(EMyEnumeration(2)))
|
||||
|
||||
|
@ -10,6 +10,8 @@ test_function(enum=1)
|
||||
None
|
||||
test_function(enum=2)
|
||||
None
|
||||
test_ecenum(ECMyEnum::Three)
|
||||
test_ecenum(ECMyEnum::Two)
|
||||
enum->integer = 2
|
||||
integer->enum = EMyEnumeration.ESecondEntry
|
||||
A constant = 14
|
||||
|
@ -1019,9 +1019,9 @@ public:
|
||||
this->def("__init__", [](Type& value, UnderlyingType i) { new (&value) Type((Type) i); });
|
||||
this->def("__int__", [](Type value) { return (UnderlyingType) value; });
|
||||
this->def("__eq__", [](const Type &value, Type *value2) { return value2 && value == *value2; });
|
||||
this->def("__eq__", [](const Type &value, UnderlyingType value2) { return value == value2; });
|
||||
this->def("__eq__", [](const Type &value, UnderlyingType value2) { return (UnderlyingType) value == value2; });
|
||||
this->def("__ne__", [](const Type &value, Type *value2) { return !value2 || value != *value2; });
|
||||
this->def("__ne__", [](const Type &value, UnderlyingType value2) { return value != value2; });
|
||||
this->def("__ne__", [](const Type &value, UnderlyingType value2) { return (UnderlyingType) value != value2; });
|
||||
this->def("__hash__", [](const Type &value) { return (UnderlyingType) value; });
|
||||
m_entries = entries;
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user