Make MultiPolygons storable in rstar

georust · Nov 3, 2024 · d6f0d5a · d6f0d5a
1 parent ab9856f
commit d6f0d5a
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 0 deletions.
diff --git a/geo-types/CHANGES.md b/geo-types/CHANGES.md
@@ -1,5 +1,8 @@
 # Changes
 
+## Unreleased
+* Add rstar compatibility for MultiPolygon
+
 ## 0.7.13
 
 * POSSIBLY BREAKING: Minimum supported version of Rust (MSRV) is now 1.70

diff --git a/geo-types/src/geometry/multi_polygon.rs b/geo-types/src/geometry/multi_polygon.rs
@@ -170,6 +170,40 @@ where
     }
 }
 
+#[cfg(any(
+    feature = "rstar_0_8",
+    feature = "rstar_0_9",
+    feature = "rstar_0_10",
+    feature = "rstar_0_11",
+    feature = "rstar_0_12"
+))]
+macro_rules! impl_rstar_multi_polygon {
+    ($rstar:ident) => {
+        impl<T> $rstar::RTreeObject for MultiPolygon<T>
+        where
+            T: ::num_traits::Float + ::$rstar::RTreeNum,
+        {
+            type Envelope = ::$rstar::AABB<$crate::Point<T>>;
+            fn envelope(&self) -> Self::Envelope {
+                use ::$rstar::Envelope;
+                self.iter()
+                    .map(|p| p.envelope())
+                    .fold(::$rstar::AABB::new_empty(), |a, b| a.merged(&b))
+            }
+        }
+    };
+}
+#[cfg(feature = "rstar_0_8")]
+impl_rstar_multi_polygon!(rstar_0_8);
+#[cfg(feature = "rstar_0_9")]
+impl_rstar_multi_polygon!(rstar_0_9);
+#[cfg(feature = "rstar_0_10")]
+impl_rstar_multi_polygon!(rstar_0_10);
+#[cfg(feature = "rstar_0_11")]
+impl_rstar_multi_polygon!(rstar_0_11);
+#[cfg(feature = "rstar_0_12")]
+impl_rstar_multi_polygon!(rstar_0_12);
+
 #[cfg(test)]
 mod test {
     use super::*;