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
|
#pragma once
#include "search/cancel_exception.hpp"
#include "search/geocoder.hpp"
#include "search/intermediate_result.hpp"
#include "search/keyword_lang_matcher.hpp"
#include "search/locality_finder.hpp"
#include "search/mode.hpp"
#include "search/result.hpp"
#include "search/reverse_geocoder.hpp"
#include "search/search_params.hpp"
#include "search/suggest.hpp"
#include "indexer/categories_holder.hpp"
#include "indexer/feature_decl.hpp"
#include "geometry/point2d.hpp"
#include "geometry/rect2d.hpp"
#include "base/string_utils.hpp"
#include "std/set.hpp"
#include "std/string.hpp"
#include "std/utility.hpp"
#include "std/vector.hpp"
class CategoriesHolder;
class Index;
namespace storage
{
class CountryInfoGetter;
} // namespace storage
namespace search
{
class PreResult2Maker;
class Ranker
{
public:
struct Params
{
using TLocales = buffer_vector<int8_t, 3>;
int8_t m_currentLocaleCode = CategoriesHolder::kEnglishCode;
m2::RectD m_viewport;
m2::PointD m_position;
string m_pivotRegion;
set<uint32_t> m_preferredTypes;
bool m_suggestsEnabled = false;
bool m_viewportSearch = false;
string m_query;
buffer_vector<strings::UniString, 32> m_tokens;
// Prefix of the last token in the query.
// We need it here to make suggestions.
strings::UniString m_prefix;
m2::PointD m_accuratePivotCenter = m2::PointD(0, 0);
TLocales m_categoryLocales;
size_t m_limit = 0;
SearchParams::TOnResults m_onResults;
};
static size_t const kBatchSize;
Ranker(Index const & index, storage::CountryInfoGetter const & infoGetter,
CategoriesHolder const & categories, vector<Suggest> const & suggests,
my::Cancellable const & cancellable);
virtual ~Ranker() = default;
void Init(Params const & params, Geocoder::Params const & geocoderParams);
bool IsResultExists(PreResult2 const & p, vector<IndexedValue> const & values);
void MakePreResult2(Geocoder::Params const & params, vector<IndexedValue> & cont,
vector<FeatureID> & streets);
Result MakeResult(PreResult2 const & r) const;
void MakeResultHighlight(Result & res) const;
void GetSuggestion(string const & name, string & suggest) const;
void SuggestStrings(Results & res);
void MatchForSuggestions(strings::UniString const & token, int8_t locale, string const & prolog,
Results & res);
void GetBestMatchName(FeatureType const & f, string & name) const;
void ProcessSuggestions(vector<IndexedValue> & vec, Results & res) const;
virtual void SetPreResults1(vector<PreResult1> && preResults1) { m_preResults1 = move(preResults1); }
virtual void UpdateResults(bool lastUpdate);
inline Results & GetResults() { return m_results; }
inline void FlushResults() { UpdateResults(true /* lastUpdate */); }
void ClearCaches();
inline void SetLocalityFinderLanguage(int8_t code) { m_locality.SetLanguage(code); }
inline void SetLanguage(pair<int, int> const & ind, int8_t lang)
{
m_keywordsScorer.SetLanguage(ind, lang);
}
inline int8_t GetLanguage(pair<int, int> const & ind) const
{
return m_keywordsScorer.GetLanguage(ind);
}
inline void SetLanguages(vector<vector<int8_t>> const & languagePriorities)
{
m_keywordsScorer.SetLanguages(languagePriorities);
}
inline void SetKeywords(KeywordMatcher::StringT const * keywords, size_t count,
KeywordMatcher::StringT const & prefix)
{
m_keywordsScorer.SetKeywords(keywords, count, prefix);
}
inline void BailIfCancelled() { ::search::BailIfCancelled(m_cancellable); }
private:
friend class PreResult2Maker;
Params m_params;
Geocoder::Params m_geocoderParams;
ReverseGeocoder const m_reverseGeocoder;
my::Cancellable const & m_cancellable;
KeywordLangMatcher m_keywordsScorer;
mutable LocalityFinder m_locality;
Index const & m_index;
storage::CountryInfoGetter const & m_infoGetter;
CategoriesHolder const & m_categories;
vector<Suggest> const & m_suggests;
vector<PreResult1> m_preResults1;
vector<IndexedValue> m_tentativeResults;
Results m_results;
};
} // namespace search
|