效果
ENUM_DEFINE ( Color,
Red,
Blue,
)
EnumHelper(Color::Red) -> "Red"
EnumHelper(Color::Red, std::toupper) -> "RED"
关键技术
-
__VA_ARGS__
__VA_ARGS__
实现了可变参数的宏。#define XXX(type, ...) enum class type { __VA_ARGS__ };
XXX(Color, Red, Blue)
等价于:enum class Color { Red, Blue };
-
#__VA_ARGS__
#__VA_ARGS__
可将宏的可变参数转为字符串。#define XXX(type, ...) #__VA_ARGS__
XXX(Color, Red, Blue)
等价于:"Red, Blue"
-
在函数外执行代码的能力
在函数体外,可以通过定义全局变量来执行一个函数。需要注意的是,头文件中正常是不能进行变量初始化的,除非加上
static
或者const
。const int temp = initialize();
另外,如果多个代码文件
#include
了该头文件,会产生多个变量,即在不同代码文件取得的temp
变量不是同一个。与之对应,initialize
函数也会调用多次。 -
模板函数的静态变量
函数的静态变量可以用于存放枚举值到枚举字符串的映射,而将枚举类型作为模板参数的模板函数,则可以直接为每种枚举提供了一个映射容器。
-
关键代码
template<typename T> std::string EnumHelper(T key, const std::function<char(char)> processor = nullptr, const char* pszName = NULL) { static_assert(std::is_enum_v<T>, __FUNCTION__ "'s key need a enum"); static std::map<T, std::string> s_mapName; if (nullptr != pszName) { s_mapName[key] = pszName; } std::string res = ""; auto it = s_mapName.find(key); if (it != s_mapName.end()) { res = it->second; } if (nullptr != processor) { std::transform(res.begin(), res.end(), res.begin(), processor); } return res; } template <class T> size_t ParseEnum(T enumClass, const char* pszNames) { static_assert(std::is_enum_v<T>, __FUNCTION__ "'s enumClass need a enum"); static size_t s_sizeOfEnum = 0; // 避免重复调用的开销 if (s_sizeOfEnum != 0) { return s_sizeOfEnum; } if (nullptr != pszNames) { const std::vector<std::string>& vecName = split(pszNames, ","); for (int i = 0; i < vecName.size(); ++i) { if (vecName.at(i).size() > 0) { EnumHelper((T)(i + 1), nullptr, vecName.at(i).c_str() + (i == 0 ? 0 : 1) ); } } s_sizeOfEnum = vecName.size(); return s_sizeOfEnum; } return 0; } #define ENUM_DEFINE(type, ...) enum class type { placeholder, __VA_ARGS__ }; const size_t g_uEnumSizeOf##type = ParseEnum(type::placeholder, #__VA_ARGS__);
-
源码地址