c++11 實現枚舉值到枚舉名的轉換

效果

ENUM_DEFINE ( Color,
    Red,
    Blue,
)

EnumHelper(Color::Red) -> "Red"
EnumHelper(Color::Red, std::toupper) -> "RED"

關鍵技術

1. __VA_ARGS__

   __VA_ARGS__ 實現了可變參數的宏。

   #define XXX(type, ...) enum class type { __VA_ARGS__ };
   

   XXX(Color, Red, Blue) 等價於：

   enum class Color
   {
       Red,
       Blue
   };
   

2. #__VA_ARGS__

   #__VA_ARGS__ 可將宏的可變參數轉為字符串。

   #define XXX(type, ...) #__VA_ARGS__
   

   XXX(Color, Red, Blue) 等價於："Red, Blue"

3. 在函數外執行代碼的能力

   在函數體外，可以通過定義全局變量來執行一個函數。需要注意的是，頭文件中正常是不能進行變量初始化的，除非加上 static 或者 const。

   const int temp = initialize();
   

   另外，如果多個代碼文件 #include 了該頭文件，會產生多個變量，即在不同代碼文件取得的 temp 變量不是同一個。與之對應，initialize 函數也會調用多次。

4. 模板函數的靜態變量

   函數的靜態變量可以用於存放枚舉值到枚舉字符串的映射，而將枚舉類型作為模板參數的模板函數，則可以直接為每種枚舉提供了一個映射容器。

5. 關鍵代碼

   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__);
   

6. 源碼地址