Repo created

libs/search/retrieval.hpp

@@ -0,0 +1,114 @@
+#pragma once
+
+#include "search/cbv.hpp"
+#include "search/feature_offset_match.hpp"
+#include "search/query_params.hpp"
+
+#include "platform/mwm_traits.hpp"
+
+#include "coding/reader.hpp"
+
+#include "geometry/rect2d.hpp"
+
+#include "base/cancellable.hpp"
+#include "base/checked_cast.hpp"
+#include "base/dfa_helpers.hpp"
+#include "base/levenshtein_dfa.hpp"
+
+#include <cstdint>
+#include <functional>
+#include <memory>
+#include <utility>
+
+class MwmValue;
+
+namespace search
+{
+class MwmContext;
+class TokenSlice;
+
+class Retrieval
+{
+public:
+  template <typename Value>
+  using TrieRoot = trie::Iterator<ValueList<Value>>;
+  using Features = search::CBV;
+
+  struct ExtendedFeatures
+  {
+    ExtendedFeatures() = default;
+    ExtendedFeatures(ExtendedFeatures const &) = default;
+    ExtendedFeatures(ExtendedFeatures &&) = default;
+
+    explicit ExtendedFeatures(Features const & cbv) : m_features(cbv), m_exactMatchingFeatures(cbv) {}
+
+    ExtendedFeatures(Features && features, Features && exactMatchingFeatures)
+      : m_features(std::move(features))
+      , m_exactMatchingFeatures(std::move(exactMatchingFeatures))
+    {}
+
+    ExtendedFeatures & operator=(ExtendedFeatures const &) = default;
+    ExtendedFeatures & operator=(ExtendedFeatures &&) = default;
+
+    ExtendedFeatures Intersect(ExtendedFeatures const & rhs) const
+    {
+      ExtendedFeatures result;
+      result.m_features = m_features.Intersect(rhs.m_features);
+      result.m_exactMatchingFeatures = m_exactMatchingFeatures.Intersect(rhs.m_exactMatchingFeatures);
+      return result;
+    }
+
+    ExtendedFeatures Intersect(Features const & cbv) const
+    {
+      ExtendedFeatures result;
+      result.m_features = m_features.Intersect(cbv);
+      result.m_exactMatchingFeatures = m_exactMatchingFeatures.Intersect(cbv);
+      return result;
+    }
+
+    void SetFull()
+    {
+      m_features.SetFull();
+      m_exactMatchingFeatures.SetFull();
+    }
+
+    void ForEach(std::function<void(uint32_t, bool)> const & f) const
+    {
+      m_features.ForEach([&](uint64_t id)
+      { f(base::asserted_cast<uint32_t>(id), m_exactMatchingFeatures.HasBit(id)); });
+    }
+
+    Features m_features;
+    Features m_exactMatchingFeatures;
+  };
+
+  Retrieval(MwmContext const & context, base::Cancellable const & cancellable);
+
+  // Following functions retrieve all features matching to |request| from the search index.
+  ExtendedFeatures RetrieveAddressFeatures(SearchTrieRequest<strings::UniStringDFA> const & request) const;
+
+  ExtendedFeatures RetrieveAddressFeatures(
+      SearchTrieRequest<strings::PrefixDFAModifier<strings::UniStringDFA>> const & request) const;
+
+  ExtendedFeatures RetrieveAddressFeatures(SearchTrieRequest<strings::LevenshteinDFA> const & request) const;
+
+  ExtendedFeatures RetrieveAddressFeatures(
+      SearchTrieRequest<strings::PrefixDFAModifier<strings::LevenshteinDFA>> const & request) const;
+
+  // Retrieves all postcodes matching to |slice| from the search index.
+  Features RetrievePostcodeFeatures(TokenSlice const & slice) const;
+
+  // Retrieves all features belonging to |rect| from the geometry index.
+  Features RetrieveGeometryFeatures(m2::RectD const & rect, int scale) const;
+
+private:
+  template <template <typename> class R, typename... Args>
+  ExtendedFeatures Retrieve(Args &&... args) const;
+
+  MwmContext const & m_context;
+  base::Cancellable const & m_cancellable;
+  ModelReaderPtr m_reader;
+
+  std::unique_ptr<TrieRoot<Uint64IndexValue>> m_root;
+};
+}  // namespace search