wipupstream/12044-android-query-test

1 files changed, 23 insertions, 4 deletions

diff --git a/platform/default/mbgl/map/map_snapshotter.cpp b/platform/default/mbgl/map/map_snapshotter.cpp
index a909e3fe9b..9358d76a18 100644
--- a/platform/default/mbgl/map/map_snapshotter.cpp
+++ b/platform/default/mbgl/map/map_snapshotter.cpp
@@ -2,6 +2,7 @@
 
 #include <mbgl/actor/actor_ref.hpp>
 #include <mbgl/gl/headless_frontend.hpp>
+#include <mbgl/renderer/renderer.hpp>
 #include <mbgl/map/map.hpp>
 #include <mbgl/map/transform_state.hpp>
 #include <mbgl/storage/file_source.hpp>
@@ -37,7 +38,9 @@ public:
     void setRegion(LatLngBounds);
     LatLngBounds getRegion() const;
 
-    void snapshot(ActorRef<MapSnapshotter::Callback>);
+    void snapshot(ActorRef<MapSnapshotter::SnapshotCallback>);
+
+    void queryFeatures(ActorRef<MapSnapshotter::QueryFeaturesCallback>, const ScreenBox&, const RenderedQueryOptions&);
 
 private:
     HeadlessFrontend frontend;
@@ -71,7 +74,7 @@ MapSnapshotter::Impl::Impl(FileSource& fileSource,
     }
 }
 
-void MapSnapshotter::Impl::snapshot(ActorRef<MapSnapshotter::Callback> callback) {
+void MapSnapshotter::Impl::snapshot(ActorRef<MapSnapshotter::SnapshotCallback> callback) {
     map.renderStill([this, callback = std::move(callback)] (std::exception_ptr error) mutable {
 
         // Create lambda that captures the current transform state
@@ -96,7 +99,7 @@ void MapSnapshotter::Impl::snapshot(ActorRef<MapSnapshotter::Callback> callback)
 
         // Invoke callback
         callback.invoke(
-                &MapSnapshotter::Callback::operator(),
+                &MapSnapshotter::SnapshotCallback::operator(),
                 error,
                 error ? PremultipliedImage() : frontend.readStillImage(),
                 std::move(attributions),
@@ -105,6 +108,17 @@ void MapSnapshotter::Impl::snapshot(ActorRef<MapSnapshotter::Callback> callback)
     });
 }
 
+void MapSnapshotter::Impl::queryFeatures(ActorRef<MapSnapshotter::QueryFeaturesCallback> callback,
+                                         const ScreenBox& box, const RenderedQueryOptions& options) {
+    map.renderStill([this, callback = std::move(callback), box, options] (std::exception_ptr error) mutable {
+        callback.invoke(
+                &MapSnapshotter::QueryFeaturesCallback::operator(),
+                error,
+                error ? std::vector<Feature>() : frontend.getRenderer()->queryRenderedFeatures(box, options)
+        );
+    });
+}
+
 void MapSnapshotter::Impl::setStyleURL(std::string styleURL) {
     map.getStyle().loadURL(styleURL);
 }
@@ -162,10 +176,15 @@ MapSnapshotter::MapSnapshotter(FileSource& fileSource,
 
 MapSnapshotter::~MapSnapshotter() = default;
 
-void MapSnapshotter::snapshot(ActorRef<MapSnapshotter::Callback> callback) {
+void MapSnapshotter::snapshot(ActorRef<MapSnapshotter::SnapshotCallback> callback) {
     impl->actor().invoke(&Impl::snapshot, std::move(callback));
 }
 
+void MapSnapshotter::queryFeatures(ActorRef<MapSnapshotter::QueryFeaturesCallback> callback,
+                                   const ScreenBox& box, const RenderedQueryOptions& options) {
+    impl->actor().invoke(&Impl::queryFeatures, std::move(callback), box, options);
+}
+
 void MapSnapshotter::setStyleURL(const std::string& styleURL) {
     impl->actor().invoke(&Impl::setStyleURL, styleURL);
 }