githubmirror/omim

indexer/data_header.cpp
159 строк4 KB
Maxim Pimenov
Review fixes.
23 ноя 2018, 16:59
23 ноя 2018, 16:595ecc13b
#include "indexer/data_header.hpp"
#include "indexer/scales.hpp"

#include "platform/platform.hpp"

#include "coding/file_container.hpp"
#include "coding/file_writer.hpp"
#include "coding/point_coding.hpp"
#include "coding/varint.hpp"
#include "coding/write_to_sink.hpp"

#include "defines.hpp"


namespace feature
{
  DataHeader::DataHeader(string const & fileName)
    : DataHeader((FilesContainerR(GetPlatform().GetReader(fileName))))
  {
  }

  DataHeader::DataHeader(FilesContainerR const & cont)
  {
    Load(cont);
  }

  serial::GeometryCodingParams DataHeader::GetGeometryCodingParams(int scaleIndex) const
  {
    return serial::GeometryCodingParams(
        m_codingParams.GetCoordBits() - (m_scales.back() - m_scales[scaleIndex]) / 2,
        m_codingParams.GetBasePointUint64());
  }

  m2::RectD const DataHeader::GetBounds() const
  {
    return Int64ToRectObsolete(m_bounds, m_codingParams.GetCoordBits());
  }

  void DataHeader::SetBounds(m2::RectD const & r)
  {
    m_bounds = RectToInt64Obsolete(r, m_codingParams.GetCoordBits());
  }

  pair<int, int> DataHeader::GetScaleRange() const
  {
    using namespace scales;

    int const low = 0;
    int const high = GetUpperScale();
    int const worldH = GetUpperWorldScale();
    MapType const type = GetType();

    switch (type)
    {
    case world: return make_pair(low, worldH);
    case worldcoasts: return make_pair(low, high);
    default:
      ASSERT_EQUAL(type, country, ());
      return make_pair(worldH + 1, high);

      // Uncomment this to test countries drawing in all scales.
      //return make_pair(1, high);
    }
  }

  namespace
  {
    template <class TSink, class TCont>
    void SaveBytes(TSink & sink, TCont const & cont)
    {
      static_assert(sizeof(typename TCont::value_type) == 1, "");

      uint32_t const count = static_cast<uint32_t>(cont.size());
      WriteVarUint(sink, count);
      if (count > 0)
        sink.Write(&cont[0], count);
    }

    template <class TSource, class TCont>
    void LoadBytes(TSource & src, TCont & cont)
    {
      static_assert(sizeof(typename TCont::value_type) == 1, "");
      ASSERT ( cont.empty(), () );

      uint32_t const count = ReadVarUint<uint32_t>(src);
      if (count > 0)
      {
        cont.resize(count);
        src.Read(&cont[0], count);
      }
    }
  }

  void DataHeader::Save(FileWriter & w) const
  {
    m_codingParams.Save(w);

    WriteVarInt(w, m_bounds.first);
    WriteVarInt(w, m_bounds.second);

    SaveBytes(w, m_scales);
    SaveBytes(w, m_langs);

    WriteVarInt(w, static_cast<int32_t>(m_type));
  }

  void DataHeader::Load(FilesContainerR const & cont)
  {
    ModelReaderPtr headerReader = cont.GetReader(HEADER_FILE_TAG);
    version::MwmVersion version;

    if (version::ReadVersion(cont, version))
      Load(headerReader, version.GetFormat());
    else
      LoadV1(headerReader);
  }

  void DataHeader::Load(ModelReaderPtr const & r, version::Format format)
  {
    ReaderSource<ModelReaderPtr> src(r);
    m_codingParams.Load(src);

    m_bounds.first = ReadVarInt<int64_t>(src);
    m_bounds.second = ReadVarInt<int64_t>(src);

    LoadBytes(src, m_scales);
    LoadBytes(src, m_langs);

    m_type = static_cast<MapType>(ReadVarInt<int32_t>(src));
    m_format = format;

    if (!IsMWMSuitable())
    {
      // Actually, old versions of the app should read mwm header correct!
      // This condition is also checked in adding mwm to the model.
      return;
    }

    // Place all new serializable staff here.
  }

  void DataHeader::LoadV1(ModelReaderPtr const & r)
  {
    ReaderSource<ModelReaderPtr> src(r);
    int64_t const base = ReadPrimitiveFromSource<int64_t>(src);
    m_codingParams = serial::GeometryCodingParams(kPointCoordBits, base);

    m_bounds.first = ReadVarInt<int64_t>(src) + base;
    m_bounds.second = ReadVarInt<int64_t>(src) + base;

    uint32_t const count = 4;
    m_scales.resize(count);
    src.Read(m_scales.data(), count);

    m_type = country;

    m_format = version::Format::v1;
  }
}