githubmirror/omim

metrics/eye_serdes.cpp
178 строк5 KB
Arsentiy Milchakov
[eye] tests fix
14 ноя 2018, 17:56
14 ноя 2018, 17:566bb6aa7
#include "metrics/eye_serdes.hpp"

#include "coding/reader.hpp"
#include "coding/serdes_json.hpp"
#include "coding/write_to_sink.hpp"
#include "coding/writer.hpp"

#include "geometry/point2d.hpp"

#include "base/logging.hpp"

#include <sstream>
#include <string>
#include <type_traits>

namespace
{
struct MapObjectEvent
{
  DECLARE_VISITOR(visitor(m_bestPoiType, "best_type"), visitor(m_poiPos, "pos"),
                  visitor(m_readableName, "readable_name"), visitor(m_defaultName, "default_name"),
                  visitor(m_event, "event"));

  std::string m_bestPoiType;
  m2::PointD m_poiPos;
  std::string m_defaultName;
  std::string m_readableName;
  eye::MapObject::Event m_event;
};
}  // namespace

namespace eye
{
// static
void Serdes::SerializeInfo(Info const & info, std::vector<int8_t> & result)
{
  result.clear();
  using Sink = MemWriter<std::vector<int8_t>>;
  Sink writer(result);

  WriteToSink(writer, static_cast<int8_t>(Info::GetVersion()));

  coding::SerializerJson<Sink> ser(writer);
  ser(info);
}

// static
void Serdes::DeserializeInfo(std::vector<int8_t> const & bytes, Info & result)
{
  MemReader reader(bytes.data(), bytes.size());
  NonOwningReaderSource source(reader);

  auto version = static_cast<int8_t>(Version::Unknown);
  ReadPrimitiveFromSource(source, version);
  if (version == static_cast<int8_t>(Version::V0))
  {
    try
    {
      // TODO: Use temporary object InfoV0 and implement method to convert it to Info,
      // TODO: when InfoV1 will be implemented.
      coding::DeserializerJson des(source);
      des(result);
    }
    catch (base::Json::Exception & ex)
    {
      LOG(LERROR, ("Cannot deserialize eye file. Exception:", ex.Msg(), "Version:", version,
                   "File content:", std::string(bytes.begin(), bytes.end())));
    }
    return;
  }

  MYTHROW(UnknownVersion, ("Unknown data version:", static_cast<int>(version)));
}

// static
void Serdes::SerializeMapObjects(MapObjects const & mapObjects, std::vector<int8_t> & result)
{
  result.clear();
  using Sink = MemWriter<std::vector<int8_t>>;

  Sink writer(result);
  std::string const nextLine = "\n";
  MapObjectEvent event;

  mapObjects.ForEach([&writer, &event, &nextLine](MapObject const & item)
  {
    for (auto const & poiEvent : item.GetEvents())
    {
      // Additional scope is added because of the coding::SerializerJson dumps result at destruction.
      {
        coding::SerializerJson<Sink> ser(writer);
        event.m_bestPoiType = item.GetBestType();
        event.m_poiPos = item.GetPos();
        event.m_defaultName = item.GetDefaultName();
        event.m_readableName = item.GetReadableName();
        event.m_event = poiEvent;
        ser(event);
      }
      writer.Write(nextLine.data(), nextLine.size());
    }
  });
}

// static
void Serdes::DeserializeMapObjects(std::vector<int8_t> const & bytes, MapObjects & result)
{
  MemReader reader(bytes.data(), bytes.size());
  NonOwningReaderSource source(reader);

  std::string tmp(bytes.begin(), bytes.end());
  std::istringstream is(tmp);

  std::string eventString;
  MapObjectEvent event;
  MapObject poi;

  try
  {
    while (getline(is, eventString))
    {
      if (eventString.empty())
        return;

      coding::DeserializerJson des(eventString);
      des(event);
      poi.SetBestType(event.m_bestPoiType);
      poi.SetPos(event.m_poiPos);
      poi.SetDefaultName(event.m_defaultName);
      poi.SetReadableName(event.m_readableName);

      bool found = false;
      result.ForEachInRect(poi.GetLimitRect(), [&found, &poi, &event](MapObject const & item)
      {
        if (item != poi)
          return;

        if (!found)
          found = true;

        item.GetEditableEvents().push_back(event.m_event);
      });

      if (!found)
      {
        poi.GetEditableEvents().push_back(event.m_event);
        result.Add(poi);
      }
    }
  }
  catch (base::Json::Exception & ex)
  {
    LOG(LERROR, ("Cannot deserialize map objects. Exception:", ex.Msg(), ". Event string:",
                 eventString, ". Content:", std::string(bytes.begin(), bytes.end())));
  }
}

// static
void Serdes::SerializeMapObjectEvent(MapObject const & poi, MapObject::Event const & poiEvent,
                                     std::vector<int8_t> & result)
{
  result.clear();

  using Sink = MemWriter<std::vector<int8_t>>;

  Sink writer(result);
  coding::SerializerJson<Sink> ser(writer);
  std::string const nextLine = "\n";

  MapObjectEvent event;
  event.m_bestPoiType = poi.GetBestType();
  event.m_poiPos = poi.GetPos();
  event.m_defaultName = poi.GetDefaultName();
  event.m_readableName = poi.GetReadableName();
  event.m_event = poiEvent;
  ser(event);
  writer.Write(nextLine.data(), nextLine.size());
}
}  // namespace eye