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
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
|
#pragma once
#include "search/keyword_lang_matcher.hpp"
#include "search/mode.hpp"
#include "search/pre_ranker.hpp"
#include "search/reverse_geocoder.hpp"
#include "search/search_trie.hpp"
#include "search/suggest.hpp"
#include "search/v2/geocoder.hpp"
#include "search/v2/rank_table_cache.hpp"
#include "search/v2/token_slice.hpp"
#include "indexer/ftypes_matcher.hpp"
#include "indexer/index.hpp"
#include "indexer/rank_table.hpp"
#include "indexer/string_slice.hpp"
#include "geometry/rect2d.hpp"
#include "base/buffer_vector.hpp"
#include "base/cancellable.hpp"
#include "base/limited_priority_queue.hpp"
#include "base/string_utils.hpp"
#include "std/map.hpp"
#include "std/string.hpp"
#include "std/unordered_set.hpp"
#include "std/vector.hpp"
#define FIND_LOCALITY_TEST
#ifdef FIND_LOCALITY_TEST
#include "search/locality_finder.hpp"
#endif
class FeatureType;
class CategoriesHolder;
namespace coding
{
class CompressedBitVector;
}
namespace storage
{
class CountryInfoGetter;
}
namespace search
{
struct Locality;
struct Region;
struct QueryParams;
class ReverseGeocoder;
namespace v2
{
class Geocoder;
} // namespace search::v2
namespace impl
{
class FeatureLoader;
class BestNameFinder;
class PreResult2Maker;
class DoFindLocality;
class HouseCompFactory;
}
class Processor : public my::Cancellable
{
public:
// Maximum result candidates count for each viewport/criteria.
static size_t const kPreResultsCount = 200;
static double const kMinViewportRadiusM;
static double const kMaxViewportRadiusM;
Processor(Index & index, CategoriesHolder const & categories, vector<Suggest> const & suggests,
storage::CountryInfoGetter const & infoGetter);
inline void SupportOldFormat(bool b) { m_supportOldFormat = b; }
void Init(bool viewportSearch);
/// @param[in] forceUpdate Pass true (default) to recache feature's ids even
/// if viewport is a part of the old cached rect.
void SetViewport(m2::RectD const & viewport, bool forceUpdate);
void SetPreferredLocale(string const & locale);
void SetInputLocale(string const & locale);
void SetQuery(string const & query);
// TODO (@y): this function must be removed.
void SetRankPivot(m2::PointD const & pivot);
inline void SetMode(Mode mode) { m_mode = mode; }
inline void SetSearchInWorld(bool b) { m_worldSearch = b; }
inline void SetSuggestsEnabled(bool enabled) { m_suggestsEnabled = enabled; }
inline void SetPosition(m2::PointD const & position) { m_position = position; }
inline string const & GetPivotRegion() const { return m_region; }
inline m2::PointD const & GetPosition() const { return m_position; }
/// Suggestions language code, not the same as we use in mwm data
int8_t m_inputLocaleCode, m_currentLocaleCode;
inline bool IsEmptyQuery() const { return (m_prefix.empty() && m_tokens.empty()); }
/// @name Various search functions.
//@{
void Search(Results & results, size_t limit);
void SearchViewportPoints(Results & results);
// Tries to generate a (lat, lon) result from |m_query|.
void SearchCoordinates(Results & res) const;
//@}
struct CancelException
{
};
void InitParams(QueryParams & params);
void ClearCaches();
// my::Cancellable overrides:
void Reset() override;
void Cancel() override;
protected:
enum ViewportID
{
DEFAULT_V = -1,
CURRENT_V = 0,
LOCALITY_V = 1,
COUNT_V = 2 // Should always be the last
};
friend string DebugPrint(ViewportID viewportId);
friend class impl::FeatureLoader;
friend class impl::BestNameFinder;
friend class impl::PreResult2Maker;
friend class impl::DoFindLocality;
friend class impl::HouseCompFactory;
int GetCategoryLocales(int8_t(&arr)[3]) const;
template <class ToDo>
void ForEachCategoryTypes(StringSliceBase const & slice, ToDo toDo) const;
template <class ToDo>
void ProcessEmojiIfNeeded(strings::UniString const & token, size_t ind, ToDo & toDo) const;
using TMWMVector = vector<shared_ptr<MwmInfo>>;
using TOffsetsVector = map<MwmSet::MwmId, vector<uint32_t>>;
using TFHeader = feature::DataHeader;
m2::PointD GetPivotPoint() const;
m2::RectD GetPivotRect() const;
void SetViewportByIndex(TMWMVector const & mwmsInfo, m2::RectD const & viewport, size_t idx,
bool forceUpdate);
void ClearCache(size_t ind);
template <class T>
void MakePreResult2(v2::Geocoder::Params const & params, vector<T> & cont,
vector<FeatureID> & streets);
void FlushResults(v2::Geocoder::Params const & params, Results & res, size_t resCount);
void FlushViewportResults(v2::Geocoder::Params const & params, Results & res);
void RemoveStringPrefix(string const & str, string & res) const;
void GetSuggestion(string const & name, string & suggest) const;
template <class T>
void ProcessSuggestions(vector<T> & vec, Results & res) const;
void SuggestStrings(Results & res);
void MatchForSuggestionsImpl(strings::UniString const & token, int8_t locale,
string const & prolog, Results & res);
void GetBestMatchName(FeatureType const & f, string & name) const;
Result MakeResult(impl::PreResult2 const & r) const;
void MakeResultHighlight(Result & res) const;
Index & m_index;
CategoriesHolder const & m_categories;
vector<Suggest> const & m_suggests;
storage::CountryInfoGetter const & m_infoGetter;
string m_region;
string m_query;
buffer_vector<strings::UniString, 32> m_tokens;
strings::UniString m_prefix;
set<uint32_t> m_prefferedTypes;
#ifdef FIND_LOCALITY_TEST
mutable LocalityFinder m_locality;
#endif
m2::RectD m_viewport[COUNT_V];
m2::PointD m_pivot;
m2::PointD m_position;
Mode m_mode;
bool m_worldSearch;
bool m_suggestsEnabled;
/// @name Get ranking params.
//@{
/// @return Rect for viewport-distance calculation.
m2::RectD const & GetViewport(ViewportID vID = DEFAULT_V) const;
//@}
void SetLanguage(int id, int8_t lang);
int8_t GetLanguage(int id) const;
KeywordLangMatcher m_keywordsScorer;
bool m_supportOldFormat;
protected:
bool m_viewportSearch;
bool m_keepHouseNumberInQuery;
PreRanker m_preRanker;
v2::Geocoder m_geocoder;
ReverseGeocoder const m_reverseGeocoder;
};
} // namespace search
|
