如何使用谷物序列化枚举类型?

时间:2016-10-11 02:27:14

标签: c++ serialization enums cereal

例如

enum Color {RED, BLUE, YELLOW};

我想用谷物序列化这种类型

Color c = RED;
JSONOutputArchive archive(std::cout);
archive(c);

输出看起来像

"value0" : "RED"

"value0" : RED

2 个答案:

答案 0 :(得分:3)

您想要的输出是没有序列化库能够自动执行的,因为您为枚举值提供的名称只有编译器知道。

通常在这种情况下,您需要提供可以在字符串表示和枚举的数值之间进行转换的代码(例如,Enum to String C++),或使用自动执行此操作的library对你而言。

我们假设您拥有此功能。然后,您需要编写专门用于枚举的ta对最小序列化函数。

这是一个完整的例子。你如何选择从枚举到字符串取决于你,我选择了两张地图,因为它不需要我真正的努力,但你可以很容易地隐藏所有这些,这将为你生成所有必需的代码:

#include <cereal/archives/json.hpp>
#include <iostream>
#include <sstream>
#include <map>

enum Color {RED, BLUE, GREEN};
enum AnotherEnum {HELLO_WORLD};

std::map<Color, std::string> ColorMapForward = {{RED, "RED"}, {BLUE, "BLUE"}, {GREEN, "GREEN"}};
std::map<std::string, Color> ColorMapReverse = {{"RED", RED}, {"BLUE", BLUE}, {"GREEN", GREEN}};

std::string Color_tostring( Color c )
{
  return ColorMapForward[c];
}

Color Color_fromstring( std::string const & s )
{
  return ColorMapReverse[s];
}

namespace cereal
{
  template <class Archive> inline
  std::string save_minimal( Archive const &, Color const & t )
  {
    return Color_tostring( t );
  }

  template <class Archive> inline
  void load_minimal( Archive const &, Color & t, std::string const & value )
  {
    t = Color_fromstring( value );
  }
}

int main()
{
  std::stringstream ss;

  {
    cereal::JSONOutputArchive ar(ss);
    ar( RED );
    ar( BLUE );
    ar( GREEN );
    ar( HELLO_WORLD ); // uses standard cereal implementation
  }

  std::cout << ss.str() << std::endl;
  std::stringstream ss2;

  {
    cereal::JSONInputArchive ar(ss);
    cereal::JSONOutputArchive ar2(ss2);

    Color r, b, g;
    AnotherEnum a;

    ar( r, b, g, a );
    ar2( r, b, g, a );
  }

  std::cout << ss2.str() << std::endl;
}

作为输出:

{
    "value0": "RED",
    "value1": "BLUE",
    "value2": "GREEN",
    "value3": 0
}
{
    "value0": "RED",
    "value1": "BLUE",
    "value2": "GREEN",
    "value3": 0
}

答案 1 :(得分:0)

您可以将枚举转换为int和back,如下所示:

struct Camera {
    enum Mode { MODE_ORTHOGRAPHIC, MODE_PERSPECTIVE, MODE_COUNT };

    template<class Archive>
    void serialize( Archive &archive )
    {
        auto mode = static_cast<int>( mMode );    
        archive( cereal::make_nvp( "mode", mode ) );    
        mMode = static_cast<Mode>( mode );
    }

    Mode mMode = MODE_PERSPECTIVE;
};