用宏定义获取枚举值的字符串

最新推荐文章于 2025-05-10 06:29:49 发布

原创最新推荐文章于 2025-05-10 06:29:49 发布 · 1.1k 阅读

5 ·

CC 4.0 BY-SA版权

c++学习记录专栏收录该内容

7 篇文章

订阅专栏

本文介绍了如何使用宏定义在C/C++中实现枚举类型枚举值到字符串以及字符串到枚举值的转换。通过DECLARE_ENUM和DEFINE_ENUM宏，可以方便地创建获取枚举值对应字符串的函数，以及通过字符串获取枚举值的函数，避免了手动维护映射数组的繁琐。这种方法既高效又简洁。

最近在学习有限状态机的时候，遇到了一个问题，如何获取枚举类型的枚举值对应的字符串？第一反应就是就是创建一个字符串的映射数组。这样做确实能够达到效果，但是始终觉得不太好看。最后终于在StackOverFlow上找到了一个类似的问题，下面我介绍一下最佳答案，代码如下：

代码段一: 作为头文件引入公共宏的部分

/// enumFactory.h
// expansion macro for enum value definition
#define ENUM_VALUE(name,assign) name assign,

// expansion macro for enum to string conversion
#define ENUM_CASE(name,assign) case name: return #name;

// expansion macro for string to enum conversion
#define ENUM_STRCMP(name,assign) if (!strcmp(str,#name)) return name;

/// declare the access function and define enum values
#define DECLARE_ENUM(EnumType,ENUM_DEF) \
  enum EnumType { \
    ENUM_DEF(ENUM_VALUE) \
  }; \
  const char *GetString(EnumType dummy); \
  EnumType Get##EnumType##Value(const char *string); \

/// define the access function names
#define DEFINE_ENUM(EnumType,ENUM_DEF) \
  const char *GetString(EnumType value) \
  { \
    switch(value) \
    { \
      ENUM_DEF(ENUM_CASE) \
      default: return ""; /* handle input error */ \
    } \
  } \
  EnumType Get##EnumType##Value(const char *str) \
  { \
    ENUM_DEF(ENUM_STRCMP) \
    return (EnumType)0; /* handle input error */ \
  } \

代码段二: 在使用处的定义

/// 放在引入的文件中
#define SOME_ENUM(XX) \
    XX(FirstValue,) \
    XX(SecondValue,) \
    XX(SomeOtherValue,=50) \
    XX(OneMoreValue,=100) \

DECLARE_ENUM(SomeEnum, SOME_ENUM)
DEFINE_ENUM(SomeEnum, SOME_ENUM)

我们在需要使用的文件中引入enumFactory.h文件，并在文件中添加对应的代码段二，就能够在代码中通过枚举值获取对应的字符串或者通过字符串获取对应的枚举值。
下面我们来看一下这段宏是如何实现的：

首先，假定我们希望定义一个宏enum SomeEnum{FirstValue, SecondValue, SomeOtherValue=50, OneMoreValue=100}，
大家会发现这个和上面的#define SOME_ENUM(XX)的内容很相似。实际上我们把SOME_ENUM和ENUM_VALUE这两个宏结合起来SOME_ENUM(ENUM_VALUE)，这个时候#define SOME_ENUM(XX)就展开成了{FirstValue, SecondValue, SomeOtherValue=50, OneMoreValue=100}再结合DECLARE_ENUM的定义我们就得到了我们最开始想定义的宏了。DEFINE_ENUM就是定义了两个函数，一个是GetString(),另一个是GetSomeEnumValue()。这两个函数就实现了通过枚举值获取字符串和通过字符串获取枚举值的功能。