Files
lief-project-LIEF/api/python/src/enums_wrapper.hpp
T
Romain Thomas bc6a8514e3 compilation: enable precompiled headers (PCH)
Add precompiled headers to the main library, Python bindings,
and unit tests to improve compilation times.
2026-03-28 17:24:44 +01:00

61 lines
2.1 KiB
C++

/* Copyright 2017 - 2026 R. Thomas
* Copyright 2017 - 2026 Quarkslab
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef PY_LIEF_ENUMS_WRAPPER_H
#define PY_LIEF_ENUMS_WRAPPER_H
#include <nanobind/nanobind.h>
#include <nanobind/stl/string.h>
#include "LIEF/visibility.h"
namespace nb = nanobind;
namespace LIEF {
template <class Type>
class LIEF_LOCAL enum_ : public nanobind::enum_<Type> {
public:
using Base = typename nanobind::enum_<Type>::Base;
using Underlying = typename nanobind::enum_<Type>::Underlying;
using nanobind::enum_<Type>::def_prop_ro;
using nanobind::enum_<Type>::def_static;
using nanobind::enum_<Type>::def;
template <typename... Extra>
NB_INLINE enum_(nanobind::handle scope, const char *name, const Extra &... extra) :
nanobind::enum_<Type>{scope, name, extra...}
{
def_static("from_value", [] (Underlying i) -> Type {
nb::object out = nb::cast(static_cast<Type>(i));
if (nb::isinstance<nb::int_>(out)) {
throw nb::value_error("Conversion error");
}
return static_cast<Type>(i);
});
def("__eq__", [](const Type &value, Underlying value2) { return (Underlying) value == value2; });
def("__eq__", [](const Type &value, Type value2) { return (Underlying) value == (Underlying) value2; });
def("__ne__", [](const Type &value, Underlying value2) { return (Underlying) value != value2; });
def("__ne__", [](const Type &value, Type value2) { return (Underlying) value != (Underlying) value2; });
def("__int__", [](const Type &value) { return (Underlying) value; });
}
};
}
#endif