123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192 |
- #ifndef NEARGYE_MAGIC_ENUM_SWITCH_HPP
- #define NEARGYE_MAGIC_ENUM_SWITCH_HPP
- #include "magic_enum.hpp"
- namespace magic_enum {
- namespace detail {
- struct default_result_type {};
- template <typename T>
- struct identity {
- using type = T;
- };
- struct nonesuch {};
- template <typename F, typename V, bool = std::is_invocable_v<F, V>>
- struct invoke_result : identity<nonesuch> {};
- template <typename F, typename V>
- struct invoke_result<F, V, true> : std::invoke_result<F, V> {};
- template <typename F, typename V>
- using invoke_result_t = typename invoke_result<F, V>::type;
- template <typename E, typename F, std::size_t... I>
- constexpr auto common_invocable(std::index_sequence<I...>) noexcept {
- static_assert(is_enum_v<E>, "magic_enum::detail::invocable_index requires enum type.");
- if constexpr (count_v<E> == 0) {
- return identity<nonesuch>{};
- } else {
- return std::common_type<invoke_result_t<F, enum_constant<values_v<E>[I]>>...>{};
- }
- }
- template <typename E, typename Result, typename F>
- constexpr auto result_type() noexcept {
- static_assert(is_enum_v<E>, "magic_enum::detail::result_type requires enum type.");
- constexpr auto seq = std::make_index_sequence<detail::count_v<E>>{};
- using R = typename decltype(common_invocable<E, F>(seq))::type;
- if constexpr (std::is_same_v<Result, default_result_type>) {
- if constexpr (std::is_same_v<R, nonesuch>) {
- return identity<void>{};
- } else {
- return identity<R>{};
- }
- } else {
- if constexpr (std::is_convertible_v<R, Result>) {
- return identity<Result>{};
- } else if constexpr (std::is_convertible_v<Result, R>) {
- return identity<R>{};
- } else {
- return identity<nonesuch>{};
- }
- }
- }
- template <typename T, typename Result, typename F, typename D = std::decay_t<T>, typename R = typename decltype(result_type<D, Result, F>())::type>
- using result_t = std::enable_if_t<std::is_enum_v<D> && !std::is_same_v<R, nonesuch>, R>;
- #if !defined(MAGIC_ENUM_ENABLE_HASH)
- template <typename T = void>
- inline constexpr auto default_result_type_lambda = []() noexcept(std::is_nothrow_default_constructible_v<T>) { return T{}; };
- template <>
- inline constexpr auto default_result_type_lambda<void> = []() noexcept {};
- template <typename R, typename F, typename... Args>
- constexpr R invoke_r(F&& f, Args&&... args) noexcept(std::is_nothrow_invocable_r_v<R, F, Args...>) {
- if constexpr (std::is_void_v<R>) {
- std::forward<F>(f)(std::forward<Args>(args)...);
- } else {
- return static_cast<R>(std::forward<F>(f)(std::forward<Args>(args)...));
- }
- }
- template <std::size_t I, std::size_t End, typename R, typename E, typename F, typename Def>
- constexpr decltype(auto) constexpr_switch_impl(F&& f, E value, Def&& def) {
- if constexpr(I < End) {
- constexpr auto v = enum_constant<enum_value<E, I>()>{};
- if (value == v) {
- if constexpr (std::is_invocable_r_v<R, F, decltype(v)>) {
- return invoke_r<R>(std::forward<F>(f), v);
- } else {
- return def();
- }
- } else {
- return constexpr_switch_impl<I + 1, End, R>(std::forward<F>(f), value, std::forward<Def>(def));
- }
- } else {
- return def();
- }
- }
- template <typename R, typename E, typename F, typename Def>
- constexpr decltype(auto) constexpr_switch(F&& f, E value, Def&& def) {
- static_assert(is_enum_v<E>, "magic_enum::detail::constexpr_switch requires enum type.");
- if constexpr (count_v<E> == 0) {
- return def();
- } else {
- return constexpr_switch_impl<0, count_v<E>, R>(std::forward<F>(f), value, std::forward<Def>(def));
- }
- }
- #endif
- }
- template <typename Result = detail::default_result_type, typename E, typename F, typename R = detail::result_t<E, Result, F>>
- constexpr decltype(auto) enum_switch(F&& f, E value) {
- using D = std::decay_t<E>;
- static_assert(std::is_enum_v<D>, "magic_enum::enum_switch requires enum type.");
- #if defined(MAGIC_ENUM_ENABLE_HASH)
- return detail::constexpr_switch<&detail::values_v<D>, detail::case_call_t::value>(
- std::forward<F>(f),
- value,
- detail::default_result_type_lambda<R>);
- #else
- return detail::constexpr_switch<R, D>(
- std::forward<F>(f),
- value,
- detail::default_result_type_lambda<R>);
- #endif
- }
- template <typename Result, typename E, typename F, typename R = detail::result_t<E, Result, F>>
- constexpr decltype(auto) enum_switch(F&& f, E value, Result&& result) {
- using D = std::decay_t<E>;
- static_assert(std::is_enum_v<D>, "magic_enum::enum_switch requires enum type.");
- #if defined(MAGIC_ENUM_ENABLE_HASH)
- return detail::constexpr_switch<&detail::values_v<D>, detail::case_call_t::value>(
- std::forward<F>(f),
- value,
- [&result]() -> R { return std::forward<Result>(result); });
- #else
- return detail::constexpr_switch<R, D>(
- std::forward<F>(f),
- value,
- [&result]() -> R { return std::forward<Result>(result); });
- #endif
- }
- }
- template <>
- struct std::common_type<magic_enum::detail::nonesuch, magic_enum::detail::nonesuch> : magic_enum::detail::identity<magic_enum::detail::nonesuch> {};
- template <typename T>
- struct std::common_type<T, magic_enum::detail::nonesuch> : magic_enum::detail::identity<T> {};
- template <typename T>
- struct std::common_type<magic_enum::detail::nonesuch, T> : magic_enum::detail::identity<T> {};
- #endif
|