From 24e78e6cd073f03a676ac59686a1f5f62572719e Mon Sep 17 00:00:00 2001
From: Pete Gadomski <pete.gadomski@gmail.com>
Date: Mon, 23 Oct 2023 14:26:35 -0400
Subject: [PATCH] feat: add geometry::rect

---
 stac/src/geometry.rs | 51 ++++++++++++++++++++++++++++++++++++++------
 1 file changed, 44 insertions(+), 7 deletions(-)

diff --git a/stac/src/geometry.rs b/stac/src/geometry.rs
index d597e8b8..c981f83f 100644
--- a/stac/src/geometry.rs
+++ b/stac/src/geometry.rs
@@ -1,5 +1,5 @@
 use serde::{Deserialize, Serialize};
-use serde_json::{Map, Value};
+use serde_json::{json, Map, Value};
 
 #[cfg(feature = "geo")]
 use crate::{Error, Result};
@@ -31,8 +31,6 @@ impl Geometry {
     /// let geometry = Geometry::point(-108.0, 42.0);
     /// ```
     pub fn point(x: f64, y: f64) -> Geometry {
-        use serde_json::json;
-
         let mut attributes = Map::new();
         let _ = attributes.insert("coordinates".to_string(), json!([x, y]));
         Geometry {
@@ -40,6 +38,32 @@ impl Geometry {
             attributes,
         }
     }
+
+    /// Creates a geometry.
+    ///
+    /// # Examples
+    ///
+    /// ```
+    /// use stac::Geometry;
+    /// let geometry = Geometry::rect(-108.0, 42.0, -107.0, 43.0);
+    /// ```
+    pub fn rect(xmin: f64, ymin: f64, xmax: f64, ymax: f64) -> Geometry {
+        let mut attributes = Map::new();
+        let _ = attributes.insert(
+            "coordinates".to_string(),
+            json!([[
+                [xmin, ymin],
+                [xmax, ymin],
+                [xmax, ymax],
+                [xmin, ymax],
+                [xmin, ymin]
+            ]]),
+        );
+        Geometry {
+            r#type: "Polygon".to_string(),
+            attributes,
+        }
+    }
 }
 
 #[cfg(feature = "geo")]
@@ -52,16 +76,29 @@ impl TryFrom<Geometry> for geo::Geometry {
     }
 }
 
-#[cfg(test)]
+#[cfg(all(test, feature = "geo"))]
 mod tests {
+    use geo::{algorithm::orient::Direction, Orient, Point, Polygon, Rect};
+
     #[test]
-    #[cfg(feature = "geo")]
     fn point() {
         let point = super::Geometry::point(-108.0, 42.0);
         let geometry: geo::Geometry = point.try_into().unwrap();
+        assert_eq!(geometry, geo::Geometry::Point(Point::new(-108.0, 42.0)));
+    }
+
+    #[test]
+    fn rect() {
+        let rect = super::Geometry::rect(-108.0, 42.0, -107.0, 43.0);
+        let geometry: geo::Geometry = rect.try_into().unwrap();
         assert_eq!(
-            geometry,
-            geo::Geometry::Point(geo::Point::new(-108.0, 42.0))
+            Polygon::try_from(geometry).unwrap(),
+            Rect::new(
+                geo::coord! { x: -108.0, y: 42.0 },
+                geo::coord! { x: -107.0, y: 43.0 },
+            )
+            .to_polygon()
+            .orient(Direction::Default)
         );
     }
 }