Repo created

libs/search/street_vicinity_loader.hpp

@@ -0,0 +1,89 @@
+#pragma once
+
+#include "search/mwm_context.hpp"
+#include "search/stats_cache.hpp"
+
+#include "geometry/rect2d.hpp"
+
+#include "base/macros.hpp"
+
+#include <vector>
+
+namespace search
+{
+class MwmContext;
+
+// This class is able to load features in a street's vicinity.
+//
+// NOTE: this class *IS NOT* thread-safe.
+class StreetVicinityLoader
+{
+public:
+  static bool IsStreet(FeatureType & ft);
+
+  struct Street
+  {
+    inline bool IsEmpty() const { return m_rect.IsEmptyInterior(); }
+
+    std::vector<uint32_t> m_features;
+    m2::RectD m_rect;
+    // std::unique_ptr<ProjectionOnStreetCalculator> m_calculator;
+
+    /// @todo Cache GetProjection results for features here, because
+    /// feature::GetCenter and ProjectionOnStreetCalculator::GetProjection are not so fast.
+  };
+
+  StreetVicinityLoader(int scale, double offsetMeters);
+  void SetContext(MwmContext * context);
+
+  // Calls |fn| on each index in |sortedIds| where sortedIds[index]
+  // belongs to the street's vicinity.
+  /*
+  template <typename TFn>
+  void ForEachInVicinity(uint32_t streetId, std::vector<uint32_t> const & sortedIds,
+                         double offsetMeters, TFn const & fn)
+  {
+    // Passed offset param should be less than the cached one, or the cache is invalid otherwise.
+    ASSERT_LESS_OR_EQUAL(offsetMeters, m_offsetMeters, ());
+
+    Street const & street = GetStreet(streetId);
+    if (street.IsEmpty())
+      return;
+
+    ProjectionOnStreetCalculator const & calculator = *street.m_calculator;
+    ProjectionOnStreet proj;
+    for (uint32_t id : street.m_features)
+    {
+      // Load center and check projection only when |id| is in |sortedIds|.
+      if (!std::binary_search(sortedIds.begin(), sortedIds.end(), id))
+        continue;
+
+      auto ft = m_context->GetFeature(id);
+      if (!ft)
+        continue;  // Feature was deleted.
+
+      if (calculator.GetProjection(feature::GetCenter(*ft, FeatureType::WORST_GEOMETRY), proj) &&
+          proj.m_distMeters <= offsetMeters)
+      {
+        fn(id);
+      }
+    }
+  }
+  */
+
+  void OnQueryFinished();
+
+  Street const & GetStreet(uint32_t featureId, bool ignoreEditedStatus = false);
+
+private:
+  void LoadStreet(uint32_t featureId, Street & street, bool ignoreEditedStatus = false);
+
+  MwmContext * m_context;
+  int m_scale;
+  double const m_offsetMeters;
+
+  Cache<uint32_t, Street> m_cache;
+
+  DISALLOW_COPY_AND_MOVE(StreetVicinityLoader);
+};
+}  // namespace search