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
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
|
#pragma once
#include "search/pre_ranking_info.hpp"
#include "search/ranking_info.hpp"
#include "search/ranking_utils.hpp"
#include "search/result.hpp"
#include "storage/storage_defines.hpp"
#include "indexer/feature_data.hpp"
#include <cstdint>
#include <string>
#include <utility>
#include <vector>
class FeatureType;
class CategoriesHolder;
namespace storage
{
class CountryInfoGetter;
struct CountryInfo;
}
namespace search
{
class ReverseGeocoder;
// First pass results class. Objects are created during search in trie.
// Works fast because it does not load features.
class PreRankerResult
{
public:
PreRankerResult(FeatureID const & id, PreRankingInfo const & info,
std::vector<ResultTracer::Branch> const & provenance);
static bool LessRankAndPopularity(PreRankerResult const & r1, PreRankerResult const & r2);
static bool LessDistance(PreRankerResult const & r1, PreRankerResult const & r2);
struct CategoriesComparator
{
bool operator()(PreRankerResult const & lhs, PreRankerResult const & rhs) const;
m2::RectD m_viewport;
bool m_positionIsInsideViewport = false;
bool m_detailedScale = false;
};
FeatureID const & GetId() const { return m_id; }
double GetDistance() const { return m_info.m_distanceToPivot; }
uint8_t GetRank() const { return m_info.m_rank; }
uint8_t GetPopularity() const { return m_info.m_popularity; }
std::pair<uint8_t, float> GetRating() const { return m_info.m_rating; }
PreRankingInfo & GetInfo() { return m_info; }
PreRankingInfo const & GetInfo() const { return m_info; }
std::vector<ResultTracer::Branch> const & GetProvenance() const { return m_provenance; }
private:
friend class RankerResult;
FeatureID m_id;
PreRankingInfo m_info;
// The call path in the Geocoder that leads to this result.
std::vector<ResultTracer::Branch> m_provenance;
};
// Second result class. Objects are created during reading of features.
// Read and fill needed info for ranking and getting final results.
class RankerResult
{
public:
enum Type
{
TYPE_LATLON,
TYPE_FEATURE,
TYPE_BUILDING //!< Buildings are not filtered out in duplicates filter.
};
/// For RESULT_FEATURE and RESULT_BUILDING.
RankerResult(FeatureType & f, m2::PointD const & center, m2::PointD const & pivot,
std::string const & displayName, std::string const & fileName);
/// For RESULT_LATLON.
RankerResult(double lat, double lon);
bool IsStreet() const;
search::RankingInfo const & GetRankingInfo() const { return m_info; }
template <typename Info>
inline void SetRankingInfo(Info && info)
{
m_info = std::forward<Info>(info);
}
FeatureID const & GetID() const { return m_id; }
std::string const & GetName() const { return m_str; }
feature::TypesHolder const & GetTypes() const { return m_types; }
Type const & GetResultType() const { return m_resultType; }
m2::PointD GetCenter() const { return m_region.m_point; }
double GetDistance() const { return m_distance; }
feature::GeomType GetGeomType() const { return m_geomType; }
Result::Metadata GetMetadata() const { return m_metadata; }
double GetDistanceToPivot() const { return m_info.m_distanceToPivot; }
double GetLinearModelRank() const { return m_info.GetLinearModelRank(); }
bool GetCountryId(storage::CountryInfoGetter const & infoGetter, uint32_t ftype,
storage::CountryId & countryId) const;
bool IsEqualCommon(RankerResult const & r) const;
uint32_t GetBestType(std::vector<uint32_t> const & preferredTypes = {}) const;
std::vector<ResultTracer::Branch> const & GetProvenance() const { return m_provenance; }
private:
friend class RankerResultMaker;
struct RegionInfo
{
storage::CountryId m_countryId;
m2::PointD m_point;
void SetParams(storage::CountryId const & countryId, m2::PointD const & point)
{
m_countryId = countryId;
m_point = point;
}
bool GetCountryId(storage::CountryInfoGetter const & infoGetter,
storage::CountryId & countryId) const;
};
RegionInfo m_region;
FeatureID m_id;
feature::TypesHolder m_types;
std::string m_str;
double m_distance;
Type m_resultType;
RankingInfo m_info;
feature::GeomType m_geomType;
Result::Metadata m_metadata;
// The call path in the Geocoder that leads to this result.
std::vector<ResultTracer::Branch> m_provenance;
};
void ProcessMetadata(FeatureType & ft, Result::Metadata & meta);
std::string DebugPrint(RankerResult const & r);
} // namespace search
|
