Enum.h

Go to the documentation of this file.

#ifndef ENUM_1E2YM5S5
#define ENUM_1E2YM5S5

/*
* @file Tools/Enum.h
* Defines a macro that declares an enum and provides
* a function to access the names of its elements.
*
* @author Thomas Röfer
* @author Samuel Barrett
*/

#include <string>
#include <vector>
#include <string.h>
#include <bwi_tools/common/Params.h>

class EnumName {
private:
  std::vector<std::string> names; 
  static std::string trim(const std::string& s);
  
public:
  EnumName(const std::string& enums, size_t numOfEnums);

  const char* getName(size_t e) {return e >= names.size() ? 0 : names[e].c_str();}
};

#define ENUM(Enum, ...) \
  namespace Enum { \
    enum type {__VA_ARGS__, NUM}; \
  } \
  typedef Enum::type Enum##_t; \
  inline static const char* getName(Enum::type e) {static EnumName en(#__VA_ARGS__, (size_t) Enum::NUM); return en.getName((size_t) e);} \
  namespace Enum { \
    inline type fromName(const char* name, bool ignoreUnknown = false) { \
      for (size_t i = 0; i < NUM; i++) { \
        if (strcmp(getName((type)i),name) == 0) \
          return (type)i; \
      } \
      if (ignoreUnknown) \
        return NUM; \
      std::cerr << "Problem converting " << name << " to type " << #Enum << std::endl; \
      exit(13); \
    } \
    inline type fromName(const std::string &name, bool ignoreUnknown = false) {return fromName(name.c_str(),ignoreUnknown);} \
  } \
  SET_FROM_JSON_ENUM(Enum)

#endif /* end of include guard: ENUM_1E2YM5S5 */