handle feature collections

rafaqz · Nov 24, 2023 · 159f8b2 · 159f8b2
1 parent 8f571bf
commit 159f8b2
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/methods/extract.jl b/src/methods/extract.jl
@@ -79,6 +79,10 @@ end
 function _extract(A::RasterStackOrArray, ::GI.AbstractFeatureTrait, feature; kw...)
     _extract(A, GI.geometry(feature); kw...)
 end
+function _extract(A::RasterStackOrArray, ::GI.FeatureCollection, fc; kw...)
+    # Fall back to the Array/iterator method for feature collections
+    _extract(A, [GI.geometry(f) for f in GI.getfeature(fc)]; kw...)
+end
 function _extract(A::RasterStackOrArray, ::GI.AbstractMultiPointTrait, geom; skipmissing=false, kw...)
     rows = (_extract_point(A, g; kw...) for g in GI.getpoint(geom))
     return skipmissing ? collect(_skip_missing_rows(rows)) : collect(rows)