使用C++11模板SFINAE实现enum class与整数的比较
在C++11中添加了enum class类型,与enum类型不同的是,使用enum class中的枚举值时必须加上类型限定符,而原来的enum枚举是可以不用加类型限定符的,它们都会被当作全局(如果没有在命名空间内)整数来对待。这样就会导致不同的模块,不能定义相同的枚举值。
比如: 头文件a.h中定义了一个枚举:
1enum ETest1
2{
3 A,
4 B,
5};
头文件b.h需要包含a.h,如果此时在b.h中定义:
1enum ETest2
2{
3 A,
4 B,
5};
则会导致编译不过,提示A与B重定义。
所以C++11添加了enum class,如果把前面的enum改为enum class则是没有问题的。此时的枚举值在使用的时候必须加类型限定符,且不会被自动当作整数来对待。
如果有下面的代码:
1if (1 == ETest1::A || ETest1::A == 1)
2{
3}
则会报错:
此时需要把整数强制转换成枚举类型或者把枚举类型强制转换成整数类型才能进行比较。
1if (1 == (int)ETest1::A || ETest1::B == (ETest1)2)
2{
3}
这就要我们每次在比较的时候都需要多写一些代码来进行强制转换,这也是C++11设计enum class时故意要求这么做的。
如果想省事,有没有办法做到与普通的enum一样可以直接与整数进行比较呢?
掌握C++11模板SFINAE特性的话就可以很方便地做到,可以看看笔者的上篇文章 使用C++11模板SFINAE特性重写序列化与反序列化,有介绍与使用C++11模板SFINAE特性。
SFINAE是Substitution Failure Is Not An Error的缩写,直译为中文是:替换失败不是错误,cppreference中的定义为:
在函数模板的重载决议中:为模板形参替换推导类型失败时,从重载集抛弃特化,而非导致编译错误。
我的理解就是编译器在匹配重载函数时,通过类型的推导可能匹配到类型不一致的重载函数而导致匹配失败,但是只要最后能正确匹配到一个重载函数,编译器就不会报错。使用过Erlang语言的读者会对函数匹配有比较深刻的理解。
主要是使用std::enable_if_t或者enable_if模板函数来让编译器对函数进行匹配。
std::enable_if 或者enable_if_t可以使用在以下三个方面:
- 作为函数参数
- 作为模板参数
- 作为返回类型
这里附上实现enum class与整数比较的源码,包含了等于、不等于、大于、大于等于、小于、小于等于:
1template<typename A, typename B,
2 std::enable_if_t<std::is_integral<A>::value || std::is_enum<A>::value> * = nullptr,
3 std::enable_if_t<std::is_enum<B>::value || std::is_integral<B>::value> * = nullptr>
4bool operator==(A a, B b)
5{
6 return a == (A)b;
7}
8
9template<typename A, typename B,
10 std::enable_if_t<std::is_integral<A>::value || std::is_enum<A>::value> * = nullptr,
11 std::enable_if_t<std::is_enum<B>::value || std::is_integral<B>::value> * = nullptr>
12bool operator>(A a, B b)
13{
14 return a > (A)b;
15}
16
17template<typename A, typename B,
18 std::enable_if_t<std::is_integral<A>::value || std::is_enum<A>::value> * = nullptr,
19 std::enable_if_t<std::is_enum<B>::value || std::is_integral<B>::value> * = nullptr>
20bool operator>=(A a, B b)
21{
22 return a >= (A)b;
23}
24
25template<typename A, typename B,
26 std::enable_if_t<std::is_integral<A>::value || std::is_enum<A>::value> * = nullptr,
27 std::enable_if_t<std::is_enum<B>::value || std::is_integral<B>::value> * = nullptr>
28bool operator<(A a, B b)
29{
30 return a < (A)b;
31}
32
33template<typename A, typename B,
34 std::enable_if_t<std::is_integral<A>::value || std::is_enum<A>::value> * = nullptr,
35 std::enable_if_t<std::is_enum<B>::value || std::is_integral<B>::value> * = nullptr>
36bool operator<=(A a, B b)
37{
38 return a <= (A)b;
39}
40
41template<typename A, typename B,
42 std::enable_if_t<std::is_integral<A>::value || std::is_enum<A>::value> * = nullptr,
43 std::enable_if_t<std::is_enum<B>::value || std::is_integral<B>::value> * = nullptr>
44bool operator!=(A a, B b)
45{
46 return a != (A)b;
47}
这样就可以直接与整数比较了。
- 原文作者:Witton
- 原文链接:https://wittonbell.github.io/posts/2020/2020-12-08-使用C++11模板SFINAE实现enum-class与整数的比较/
- 版权声明:本作品采用知识共享署名-非商业性使用-禁止演绎 4.0 国际许可协议. 进行许可,非商业转载请注明出处(作者,原文链接),商业转载请联系作者获得授权。