generic int - >可以使用模板进行枚举转换吗?

时间:2016-03-10 16:54:14

标签: c++ templates enums casting int

我使用这样的函数将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

1 个答案:

答案 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;
}