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
|
#include "generator/regions/collector_region_info.hpp"
#include "generator/feature_builder.hpp"
#include "generator/osm_element.hpp"
#include "coding/file_writer.hpp"
#include "base/assert.hpp"
#include "base/logging.hpp"
#include "base/macros.hpp"
#include <map>
namespace generator
{
namespace regions
{
std::string const CollectorRegionInfo::kDefaultExt = ".regions.bin";
uint8_t const CollectorRegionInfo::kVersion = 0;
PlaceType EncodePlaceType(std::string const & place)
{
static std::map<std::string, PlaceType> const m = {
{"city", PlaceType::City},
{"town", PlaceType::Town},
{"village", PlaceType::Village},
{"suburb", PlaceType::Suburb},
{"neighbourhood", PlaceType::Neighbourhood},
{"hamlet", PlaceType::Hamlet},
{"locality", PlaceType::Locality},
{"isolated_dwelling", PlaceType::IsolatedDwelling}
};
auto const it = m.find(place);
return it == m.end() ? PlaceType::Unknown : it->second;
}
CollectorRegionInfo::CollectorRegionInfo(std::string const & filename) : m_filename(filename) {}
void CollectorRegionInfo::CollectFeature(const FeatureBuilder1 &, OsmElement const & el)
{
base::GeoObjectId const osmId = GetGeoObjectId(el);
RegionData regionData;
FillRegionData(osmId, el, regionData);
m_mapRegionData.emplace(osmId, regionData);
// If the region is a country.
if (regionData.m_adminLevel == AdminLevel::Two)
{
IsoCode isoCode;
FillIsoCode(osmId, el, isoCode);
m_mapIsoCode.emplace(osmId, isoCode);
}
}
void CollectorRegionInfo::Save()
{
FileWriter writer(m_filename);
WriteToSink(writer, kVersion);
WriteMap(writer, m_mapRegionData);
WriteMap(writer, m_mapIsoCode);
}
void CollectorRegionInfo::FillRegionData(base::GeoObjectId const & osmId, OsmElement const & el,
RegionData & rd)
{
rd.m_osmId = osmId;
rd.m_place = EncodePlaceType(el.GetTag("place"));
auto const al = el.GetTag("admin_level");
if (al.empty())
return;
try
{
auto const adminLevel = std::stoi(al);
// Administrative level is in the range [1 ... 12].
// https://wiki.openstreetmap.org/wiki/Tag:boundary=administrative
rd.m_adminLevel = (adminLevel >= 1 && adminLevel <= 12) ?
static_cast<AdminLevel>(adminLevel) : AdminLevel::Unknown;
}
catch (std::exception const & e) // std::invalid_argument, std::out_of_range
{
LOG(::base::LWARNING, (e.what()));
rd.m_adminLevel = AdminLevel::Unknown;
}
}
void CollectorRegionInfo::FillIsoCode(base::GeoObjectId const & osmId, OsmElement const & el,
IsoCode & rd)
{
rd.m_osmId = osmId;
rd.SetAlpha2(el.GetTag("ISO3166-1:alpha2"));
rd.SetAlpha3(el.GetTag("ISO3166-1:alpha3"));
rd.SetNumeric(el.GetTag("ISO3166-1:numeric"));
}
void IsoCode::SetAlpha2(std::string const & alpha2)
{
CHECK_LESS_OR_EQUAL(alpha2.size() + 1, ARRAY_SIZE(m_alpha2), ());
std::strcpy(m_alpha2, alpha2.data());
}
void IsoCode::SetAlpha3(std::string const & alpha3)
{
CHECK_LESS_OR_EQUAL(alpha3.size() + 1, ARRAY_SIZE(m_alpha3), ());
std::strcpy(m_alpha3, alpha3.data());
}
void IsoCode::SetNumeric(std::string const & numeric)
{
CHECK_LESS_OR_EQUAL(numeric.size() + 1, ARRAY_SIZE(m_numeric), ());
std::strcpy(m_numeric, numeric.data());
}
} // namespace regions
} // namespace generator
|
