我使用这样的函数将int
(从旧版API)转换为enum
:
TestEnum to_test_enum(int value) {
TestEnum converted(static_cast<TestEnum>(value));
# pragma GCC diagnostic push
# pragma GCC diagnostic error "-Wswitch"
switch (converted) {
case TestEnum::A:
case TestEnum::B:
return converted;
}
# pragma GCC diagnostic pop
throw std::runtime_error("wrong value");
}
如果将无效值传递给to_test_enum
,则 to_enum
会抛出异常,否则返回相应的TestEnum
值。 pragma
确保在合法值丢失的情况下我会收到编译器错误。
这些行现在将转换一个整数并在运行时进行有效性检查:
enum class TestEnum {
A = 1,
B = 2,
};
auto v1 = to_test_enum(2);
auto v2 = to_test_enum(3); // will throw
问题:我想知道是否可以使用某种模板魔法将其转换为函数模板:
template<class E>
E to_enum(int value) {
E converted(static_cast<E>(value));
switch (converted) {
EACH ELEMENT e IN E: case e: <--- here goes some unrolling magic
return converted;
}
throw std::runtime_error("wrong value");
}
然后将使用这个函数:
auto v1 = to_enum<TestEnum>(2);
auto v2 = to_enum<TestEnum>(3); // will throw
答案 0 :(得分:0)
我不确定我是否明白你要做的事情。见下面的想法。您必须定义自己的map_enum< your_enum >()
才能将整数映射到枚举。
#include <map>
#include <iostream>
template< typename E > const std::map< int, E >& map_enum()
{
return 0;
}
template < typename E > E to_enum( int ai )
{
auto m = map_enum< E >();
auto const i = m.find( ai );
return i == m.end() ? throw -1 : i->second;
}
// { old api
#define RED 0
#define GREEN 1
#define BLUE 2
int old_api_fun()
{
return 2;
}
// } old api
// { your code
enum color_e
{
red = RED,
green = GREEN,
blue = BLUE
};
template<> const std::map< int, color_e >& map_enum()
{
static std::map< int, color_e > m ={ { RED, red }, { GREEN, green }, { BLUE, blue } };
return m;
}
// } your code
int main()
{
try
{
auto r = to_enum< color_e >( old_api_fun() );
std::cout << "success";
}
catch ( int )
{
std::cout << "failed";
}
return 0;
}