1 files changed, 55 insertions, 0 deletions
diff --git a/search/locality_scorer.hpp b/search/locality_scorer.hpp
new file mode 100644
index 0000000000..102350d0ee
--- /dev/null
+++ b/search/locality_scorer.hpp
@@ -0,0 +1,55 @@
+#pragma once
+
+#include "search/geocoder.hpp"
+#include "search/ranking_utils.hpp"
+
+#include "std/string.hpp"
+#include "std/vector.hpp"
+
+namespace search
+{
+struct QueryParams;
+
+class LocalityScorer
+{
+public:
+  static size_t const kDefaultReadLimit;
+
+  class Delegate
+  {
+  public:
+    virtual ~Delegate() = default;
+
+    virtual void GetNames(uint32_t featureId, vector<string> & names) const = 0;
+    virtual uint8_t GetRank(uint32_t featureId) const = 0;
+  };
+
+  LocalityScorer(QueryParams const & params, Delegate const & delegate);
+
+  // Leaves at most |limit| elements of |localities|, ordered by some
+  // combination of ranks and number of matched tokens.
+  void GetTopLocalities(size_t limit, vector<Geocoder::Locality> & localities) const;
+
+private:
+  struct ExLocality
+  {
+    ExLocality();
+    explicit ExLocality(Geocoder::Locality const & locality);
+
+    inline uint32_t GetId() const { return m_locality.m_featureId; }
+
+    Geocoder::Locality m_locality;
+    size_t m_numTokens;
+    uint8_t m_rank;
+    NameScore m_nameScore;
+  };
+
+  void RemoveDuplicates(vector<ExLocality> & ls) const;
+  void LeaveTopByRank(size_t limit, vector<ExLocality> & ls) const;
+  void SortByName(vector<ExLocality> & ls) const;
+
+  QueryParams const & m_params;
+  Delegate const & m_delegate;
+};
+
+}  // namespace search