indexer/caching_rank_table_loader.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54

#include "indexer/caching_rank_table_loader.hpp"

#include "search/dummy_rank_table.hpp"

#include "indexer/data_source.hpp"
#include "indexer/feature.hpp"

#include "defines.hpp"

namespace
{
uint8_t const kNoRank = 0;
}  // namespace

CachingRankTableLoader::CachingRankTableLoader(DataSource const & dataSource,
                                               std::string const & sectionName)
  : m_dataSource(dataSource), m_sectionName(sectionName)
{
}

uint8_t CachingRankTableLoader::Get(FeatureID const & featureId) const
{
  auto const handle = m_dataSource.GetMwmHandleById(featureId.m_mwmId);

  if (!handle.IsAlive())
    return kNoRank;

  auto it = m_deserializers.find(featureId.m_mwmId);

  if (it == m_deserializers.end())
  {
    auto rankTable = search::RankTable::Load(handle.GetValue<MwmValue>()->m_cont, m_sectionName);

    if (!rankTable)
      rankTable = std::make_unique<search::DummyRankTable>();

    auto const result = m_deserializers.emplace(featureId.m_mwmId, std::move(rankTable));
    it = result.first;
  }

  return it->second->Get(featureId.m_index);
}

void CachingRankTableLoader::OnMwmDeregistered(platform::LocalCountryFile const & localFile)
{
  for (auto it = m_deserializers.begin(); it != m_deserializers.end(); ++it)
  {
    if (it->first.IsDeregistered(localFile))
    {
      m_deserializers.erase(it);
      return;
    }
  }
}