Repo created

libs/search/nested_rects_cache.hpp

@@ -0,0 +1,52 @@
+#pragma once
+
+#include "indexer/feature_decl.hpp"
+
+#include "geometry/point2d.hpp"
+
+#include <cstdint>
+#include <map>
+#include <vector>
+
+class DataSource;
+
+namespace search
+{
+class NestedRectsCache
+{
+public:
+  explicit NestedRectsCache(DataSource const & dataSource);
+
+  void SetPosition(m2::PointD const & position, int scale);
+
+  double GetDistanceToFeatureMeters(FeatureID const & id) const;
+
+  void Clear();
+
+private:
+  enum RectScale
+  {
+    RECT_SCALE_TINY,
+    RECT_SCALE_SMALL,
+    RECT_SCALE_MEDIUM,
+    RECT_SCALE_LARGE,
+
+    RECT_SCALE_COUNT
+  };
+
+  static double GetRadiusMeters(RectScale scale);
+
+  void Update();
+
+  DataSource const & m_dataSource;
+  int m_scale;
+  m2::PointD m_position;
+  bool m_valid;
+
+  using Features = std::vector<uint32_t>;
+  using Bucket = std::map<MwmSet::MwmId, Features>;
+
+  // Sorted lists of features.
+  Bucket m_buckets[RECT_SCALE_COUNT];
+};
+}  // namespace search