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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
#include "search/postcode_points.hpp"
#include "indexer/trie_reader.hpp"
#include "platform/mwm_traits.hpp"
#include "coding/reader_wrapper.hpp"
#include "geometry/mercator.hpp"
#include "base/checked_cast.hpp"
#include "base/string_utils.hpp"
#include <algorithm>
#include <iterator>
#include "defines.hpp"
using namespace std;
namespace search
{
void PostcodePoints::Header::Read(Reader & reader)
{
NonOwningReaderSource source(reader);
CHECK_EQUAL(static_cast<uint8_t>(m_version), static_cast<uint8_t>(Version::V0), ());
m_version = static_cast<Version>(ReadPrimitiveFromSource<uint8_t>(source));
CHECK_EQUAL(static_cast<uint8_t>(m_version), static_cast<uint8_t>(Version::V0), ());
m_trieOffset = ReadPrimitiveFromSource<uint32_t>(source);
m_trieSize = ReadPrimitiveFromSource<uint32_t>(source);
m_pointsOffset = ReadPrimitiveFromSource<uint32_t>(source);
m_pointsSize = ReadPrimitiveFromSource<uint32_t>(source);
}
PostcodePoints::PostcodePoints(MwmValue const & value)
{
auto reader = value.m_cont.GetReader(POSTCODE_POINTS_FILE_TAG);
m_header.Read(*reader.GetPtr());
m_trieSubReader = reader.GetPtr()->CreateSubReader(m_header.m_trieOffset, m_header.m_trieSize);
m_root = trie::ReadTrie<SubReaderWrapper<Reader>, SingleUint64Value>(
SubReaderWrapper<Reader>(m_trieSubReader.get()), SingleValueSerializer<Uint64IndexValue>());
CHECK(m_root, ());
version::MwmTraits traits(value.GetMwmVersion());
auto const format = traits.GetCentersTableFormat();
CHECK_EQUAL(format, version::MwmTraits::CentersTableFormat::EliasFanoMapWithHeader,
("Unexpected format."));
m_pointsSubReader =
reader.GetPtr()->CreateSubReader(m_header.m_pointsOffset, m_header.m_pointsSize);
m_points = CentersTable::LoadV1(*m_pointsSubReader);
CHECK(m_points, ());
auto const kPostcodeRadiusMultiplier = 5.0;
auto const area = value.GetHeader().GetBounds().Area();
auto const count = static_cast<double>(m_points->Count());
CHECK_NOT_EQUAL(
count, 0.0,
("Zero postcodes should not be serialized to", POSTCODE_POINTS_FILE_TAG, "section"));
m_radius = kPostcodeRadiusMultiplier * 0.5 * sqrt(area / count);
}
void PostcodePoints::Get(strings::UniString const & postcode, bool recursive,
vector<m2::PointD> & points) const
{
if (!m_root || !m_points || !m_trieSubReader || !m_pointsSubReader || postcode.empty())
return;
auto postcodeIt = postcode.begin();
auto trieIt = m_root->Clone();
while (postcodeIt != postcode.end())
{
auto it = find_if(trieIt->m_edges.begin(), trieIt->m_edges.end(), [&](auto const & edge) {
return strings::StartsWith(postcodeIt, postcode.end(), edge.m_label.begin(),
edge.m_label.end());
});
if (it == trieIt->m_edges.end())
return;
trieIt = trieIt->GoToEdge(distance(trieIt->m_edges.begin(), it));
postcodeIt += it->m_label.size();
}
if (postcodeIt != postcode.end())
return;
vector<uint32_t> indexes;
trieIt->m_values.ForEach([&indexes](auto const & v) {
indexes.push_back(base::asserted_cast<uint32_t>(v.m_featureId));
});
if (recursive)
{
trie::ForEachRef(
*trieIt,
[&indexes](auto const & /* s */, auto const & v) {
indexes.push_back(base::asserted_cast<uint32_t>(v.m_featureId));
},
strings::UniString{});
}
points.resize(indexes.size());
for (size_t i = 0; i < indexes.size(); ++i)
CHECK(m_points->Get(indexes[i], points[i]), ());
}
void PostcodePoints::Get(strings::UniString const & postcode, vector<m2::PointD> & points) const
{
points.clear();
Get(postcode, false /* recursive */, points);
if (!points.empty())
return;
auto static const space = strings::MakeUniString(" ");
Get(postcode + space, true /* recursive */, points);
}
PostcodePoints & PostcodePointsCache::Get(MwmContext const & context)
{
auto const mwmId = context.GetId();
auto const it = m_entries.find(mwmId);
if (it != m_entries.end())
return *it->second;
auto const emplaceRes = m_entries.emplace(mwmId, make_unique<PostcodePoints>(context.m_value));
ASSERT(emplaceRes.second, ("Failed to load postcode points for", mwmId));
return *(emplaceRes.first)->second;
}
} // namespace search
|
