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
|
#pragma once
#include "search/bookmarks/processor.hpp"
#include "search/bookmarks/types.hpp"
#include "search/categories_cache.hpp"
#include "search/categories_set.hpp"
#include "search/cities_boundaries_table.hpp"
#include "search/common.hpp"
#include "search/emitter.hpp"
#include "search/geocoder.hpp"
#include "search/pre_ranker.hpp"
#include "search/rank_table_cache.hpp"
#include "search/ranker.hpp"
#include "search/search_params.hpp"
#include "search/search_trie.hpp"
#include "search/suggest.hpp"
#include "search/token_slice.hpp"
#include "search/utils.hpp"
#include "indexer/string_slice.hpp"
#include "coding/multilang_utf8_string.hpp"
#include "geometry/rect2d.hpp"
#include "base/cancellable.hpp"
#include "base/string_utils.hpp"
#include <cstddef>
#include <cstdint>
#include <set>
#include <string>
#include <utility>
#include <vector>
class FeatureType;
class CategoriesHolder;
class DataSourceBase;
namespace coding
{
class CompressedBitVector;
}
namespace storage
{
class CountryInfoGetter;
}
namespace search
{
class Geocoder;
class QueryParams;
class Ranker;
class ReverseGeocoder;
class Processor : public ::base::Cancellable
{
public:
// Maximum result candidates count for each viewport/criteria.
static size_t const kPreResultsCount;
static double const kMinViewportRadiusM;
static double const kMaxViewportRadiusM;
Processor(DataSourceBase const & dataSource, CategoriesHolder const & categories,
std::vector<Suggest> const & suggests, storage::CountryInfoGetter const & infoGetter);
void SetViewport(m2::RectD const & viewport);
void SetPreferredLocale(std::string const & locale);
void SetInputLocale(std::string const & locale);
void SetQuery(std::string const & query);
inline void SetPosition(m2::PointD const & position) { m_position = position; }
inline std::string const & GetPivotRegion() const { return m_region; }
inline m2::PointD const & GetPosition() const { return m_position; }
inline bool IsEmptyQuery() const { return m_prefix.empty() && m_tokens.empty(); }
void Search(SearchParams const & params);
// Tries to generate a (lat, lon) result from |m_query|.
void SearchCoordinates();
// Tries to parse a plus code from |m_query| and generate a (lat, lon) result.
void SearchPlusCode();
void SearchBookmarks() const;
void InitParams(QueryParams & params) const;
void InitGeocoder(Geocoder::Params &geocoderParams,
SearchParams const &searchParams);
void InitPreRanker(Geocoder::Params const &geocoderParams,
SearchParams const &searchParams);
void InitRanker(Geocoder::Params const & geocoderParams, SearchParams const & searchParams);
void InitEmitter(SearchParams const & searchParams);
void ClearCaches();
void LoadCitiesBoundaries();
void LoadCountriesTree();
void OnBookmarksCreated(std::vector<std::pair<bookmarks::Id, bookmarks::Doc>> const & marks);
void OnBookmarksUpdated(std::vector<std::pair<bookmarks::Id, bookmarks::Doc>> const & marks);
void OnBookmarksDeleted(std::vector<bookmarks::Id> const & marks);
protected:
Locales GetCategoryLocales() const;
template <typename ToDo>
void ForEachCategoryType(StringSliceBase const & slice, ToDo && toDo) const;
template <typename ToDo>
void ForEachCategoryTypeFuzzy(StringSliceBase const & slice, ToDo && toDo) const;
m2::PointD GetPivotPoint(bool viewportSearch) const;
m2::RectD GetPivotRect(bool viewportSearch) const;
m2::RectD const & GetViewport() const;
CategoriesHolder const & m_categories;
storage::CountryInfoGetter const & m_infoGetter;
std::string m_region;
std::string m_query;
QueryTokens m_tokens;
strings::UniString m_prefix;
std::set<uint32_t> m_preferredTypes;
m2::RectD m_viewport;
m2::PointD m_position;
// Suggestions language code, not the same as we use in mwm data
int8_t m_inputLocaleCode = StringUtf8Multilang::kUnsupportedLanguageCode;
int8_t m_currentLocaleCode = StringUtf8Multilang::kUnsupportedLanguageCode;
VillagesCache m_villagesCache;
CitiesBoundariesTable m_citiesBoundaries;
KeywordLangMatcher m_keywordsScorer;
Emitter m_emitter;
Ranker m_ranker;
PreRanker m_preRanker;
Geocoder m_geocoder;
bookmarks::Processor m_bookmarksProcessor;
};
} // namespace search
|
