diff --git a/src/mbgl/style/style.cpp b/src/mbgl/style/style.cpp index 50de887db..522965b95 100644 --- a/src/mbgl/style/style.cpp +++ b/src/mbgl/style/style.cpp @@ -13,6 +13,7 @@ #include #include #include +#include #include #include #include @@ -393,10 +394,23 @@ RenderData Style::getRenderData(MapDebugOptions debugOptions) const { } std::vector Style::queryRenderedFeatures(const QueryParameters& parameters) const { + std::unordered_set sourceFilter; + + if (parameters.layerIDs) { + for (const auto& layerID : *parameters.layerIDs) { + auto layer = getLayer(layerID); + if (layer) sourceFilter.emplace(layer->baseImpl->source); + } + } + std::vector result; std::unordered_map> resultsByLayer; for (const auto& source : sources) { + if (!sourceFilter.empty() && sourceFilter.find(source->getID()) == sourceFilter.end()) { + continue; + } + auto sourceResults = source->baseImpl->queryRenderedFeatures(parameters); std::move(sourceResults.begin(), sourceResults.end(), std::inserter(resultsByLayer, resultsByLayer.begin())); }