[core] implement queryRenderedFeatures

1 files changed, 100 insertions, 0 deletions

diff --git a/src/mbgl/geometry/feature_index.hpp b/src/mbgl/geometry/feature_index.hpp
new file mode 100644
index 0000000000..dc64ea9cee
--- /dev/null
+++ b/src/mbgl/geometry/feature_index.hpp
@@ -0,0 +1,100 @@
+#ifndef MBGL_GEOMETRY_FEATURE_INDEX
+#define MBGL_GEOMETRY_FEATURE_INDEX
+
+#include <mbgl/tile/geometry_tile.hpp>
+
+#include <vector>
+#include <string>
+#include <unordered_map>
+
+#pragma GCC diagnostic push
+#pragma GCC diagnostic ignored "-Wunused-function"
+#pragma GCC diagnostic ignored "-Wunused-parameter"
+#pragma GCC diagnostic ignored "-Wunused-variable"
+#pragma GCC diagnostic ignored "-Wshadow"
+#ifdef __clang__
+#pragma GCC diagnostic ignored "-Wunknown-pragmas"
+#endif
+#pragma GCC diagnostic ignored "-Wpragmas"
+#pragma GCC diagnostic ignored "-Wdeprecated-register"
+#pragma GCC diagnostic ignored "-Wshorten-64-to-32"
+#pragma GCC diagnostic ignored "-Wunused-local-typedefs"
+#pragma GCC diagnostic ignored "-Wmaybe-uninitialized"
+#include <boost/geometry.hpp>
+#include <boost/geometry/geometries/point.hpp>
+#include <boost/geometry/geometries/box.hpp>
+#include <boost/geometry/index/rtree.hpp>
+#pragma GCC diagnostic pop
+
+namespace mbgl {
+
+class Style;
+class CollisionTile;
+enum class TranslateAnchorType : bool;
+
+class IndexedSubfeature {
+    public:
+        std::size_t index;
+        std::string sourceLayerName;
+        std::string bucketName;
+        size_t sortIndex;
+};
+
+namespace bg = boost::geometry;
+namespace bgm = bg::model;
+namespace bgi = bg::index;
+typedef bgm::point<float, 2, bg::cs::cartesian> TreePoint;
+typedef bgm::box<TreePoint> TreeBox;
+typedef std::pair<TreeBox, IndexedSubfeature> FeatureTreeBox;
+typedef bgi::rtree<FeatureTreeBox, bgi::linear<16, 4>> FeatureTree;
+
+class FeatureIndex {
+    public:
+        FeatureIndex();
+
+        void insert(const GeometryCollection&, std::size_t index, const std::string& sourceLayerName, const std::string& bucketName);
+        void loadTree();
+
+        void query(
+                std::unordered_map<std::string, std::vector<std::string>>& result,
+                const GeometryCollection& queryGeometry,
+                const float bearing,
+                const double tileSize,
+                const double scale,
+                const optional<std::vector<std::string>>& layerIDs,
+                const GeometryTile& geometryTile,
+                const Style&) const;
+
+        static optional<GeometryCollection> translateQueryGeometry(
+                const GeometryCollection& queryGeometry,
+                const std::array<float, 2>& translate,
+                const TranslateAnchorType,
+                const float bearing,
+                const float pixelsToTileUnits);
+
+        void addBucketLayerName(const std::string &bucketName, const std::string &layerName);
+
+        void setCollisionTile(std::unique_ptr<CollisionTile>);
+
+    private:
+
+        void addFeature(
+                std::unordered_map<std::string, std::vector<std::string>>& result,
+                const IndexedSubfeature &indexedFeature,
+                const GeometryCollection& queryGeometry,
+                const optional<std::vector<std::string>>& filterLayerIDs,
+                const GeometryTile& geometryTile,
+                const Style& style,
+                const float bearing,
+                const float pixelsToTileUnits) const;
+
+        std::unique_ptr<CollisionTile> collisionTile;
+        std::vector<FeatureTreeBox> treeBoxes;
+        FeatureTree tree;
+
+        std::unordered_map<std::string,std::vector<std::string>> bucketLayerIDs;
+
+};
+}
+
+#endif