githubmirror/omim

indexer/data_header.cpp
172 строки4 KB
Maxim Pimenov
[coding] Changed the filenames of the FilesContainers to better reflect the class names.
20 сен 2019, 18:29
20 сен 2019, 18:29ab23306
#include "indexer/data_header.hpp"
#include "indexer/scales.hpp"

#include "platform/platform.hpp"

#include "coding/files_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"

using namespace std;

namespace
{
  template <class Sink, class Cont>
  void SaveBytes(Sink & sink, Cont const & cont)
  {
    static_assert(sizeof(typename Cont::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 Source, class Cont>
  void LoadBytes(Source & src, Cont & cont)
  {
    static_assert(sizeof(typename Cont::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);
    }
  }
}  // namespace

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 MapType::World: return make_pair(low, worldH);
    case MapType::WorldCoasts: return make_pair(low, high);
    default:
      ASSERT_EQUAL(type, MapType::Country, ());
      return make_pair(worldH + 1, high);

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

  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 = MapType::Country;

    m_format = version::Format::v1;
  }

  string DebugPrint(DataHeader::MapType type)
  {
    switch (type)
    {
    case DataHeader::MapType::World: return "World";
    case DataHeader::MapType::WorldCoasts: return "WorldCoasts";
    case DataHeader::MapType::Country: return "Country";
    }

    UNREACHABLE();
  }
}  // namespace feature