Files
subconverter/src/utils/rapidjson_extra.h
Tindy X 2f78eafa2c Enhancements
Add better error handling.
Add support for version field of Snell nodes.
Optimize codes.
2022-01-23 01:04:16 +08:00

76 lines
1.9 KiB
C++

#ifndef RAPIDJSON_EXTRA_H_INCLUDED
#define RAPIDJSON_EXTRA_H_INCLUDED
#include <stdexcept>
template <typename T> void exception_thrower(T e)
{
if(!e)
throw std::runtime_error("rapidjson assertion failed");
}
#ifdef RAPIDJSON_ASSERT
#undef RAPIDJSON_ASSERT
#endif // RAPIDJSON_ASSERT
#define RAPIDJSON_ASSERT(x) exception_thrower(x)
#include <rapidjson/document.h>
#include <rapidjson/writer.h>
#include <rapidjson/error/en.h>
#include <string>
inline void operator >> (const rapidjson::Value& value, std::string& i)
{
if(value.IsNull())
i = std::string();
else if(value.IsString())
i = std::string(value.GetString());
else if(value.IsInt64())
i = std::to_string(value.GetInt64());
else if(value.IsBool())
i = value.GetBool() ? "true" : "false";
else if(value.IsDouble())
i = std::to_string(value.GetDouble());
else
i = std::string();
}
inline void operator >> (const rapidjson::Value& value, int& i)
{
if(value.IsNull())
i = 0;
else if(value.IsInt())
i = value.GetInt();
else if(value.IsString())
i = std::stoi(value.GetString());
else if(value.IsBool())
i = value.GetBool() ? 1 : 0;
else
i = 0;
}
inline std::string GetMember(const rapidjson::Value& value, const std::string &member)
{
std::string retStr;
if(value.IsObject() && value.HasMember(member.data()))
value[member.data()] >> retStr;
return retStr;
}
inline void GetMember(const rapidjson::Value& value, const std::string &member, std::string& target)
{
std::string retStr = GetMember(value, member);
if(retStr.size())
target.assign(retStr);
}
inline std::string SerializeObject(const rapidjson::Value& value)
{
rapidjson::StringBuffer sb;
rapidjson::Writer<rapidjson::StringBuffer> writer_json(sb);
value.Accept(writer_json);
return sb.GetString();
}
#endif // RAPIDJSON_EXTRA_H_INCLUDED