Implement conversions to GEOS objects (#108)

* Start implementing to GEOS

* Finish implementing to geos

* WIP convert from geos geometries to geoarrow arrays

* lint

Cargo.toml

@@ -21,7 +21,7 @@ byteorder = "1"
 gdal = { version = "0.15", optional = true }
 geo = "0.26"
 geodesy = { version = "0.10", optional = true }
-geos = { version = "8.3", features = ["v3_8_0", "geo"], optional = true }
+geos = { version = "8.3", features = ["v3_10_0", "geo"], optional = true }
 geozero = { version = "0.9.4", features = ["with-wkb"], optional = true }
 itertools = "0.11"
 # Note: geo has a hard dependency on rstar, so there's no point in feature flagging it

src/array/coord/combined/geos.rs

@@ -0,0 +1,14 @@
+use crate::array::CoordBuffer;
+use crate::error::{GeoArrowError, Result};
+use geos::CoordSeq;
+
+impl TryFrom<CoordBuffer> for CoordSeq<'_> {
+    type Error = GeoArrowError;
+
+    fn try_from(value: CoordBuffer) -> Result<Self> {
+        match value {
+            CoordBuffer::Separated(cb) => cb.try_into(),
+            CoordBuffer::Interleaved(cb) => cb.try_into(),
+        }
+    }
+}

src/array/coord/combined/mod.rs

@@ -1,4 +1,6 @@
 mod array;
+#[cfg(feature = "geos")]
+mod geos;
 mod mutable;
 
 pub use array::CoordBuffer;

src/array/coord/interleaved/geos.rs

@@ -0,0 +1,17 @@
+use crate::array::InterleavedCoordBuffer;
+use crate::error::{GeoArrowError, Result};
+use crate::GeometryArrayTrait;
+use geos::CoordSeq;
+
+impl TryFrom<InterleavedCoordBuffer> for CoordSeq<'_> {
+    type Error = GeoArrowError;
+
+    fn try_from(value: InterleavedCoordBuffer) -> Result<Self> {
+        Ok(CoordSeq::new_from_buffer(
+            value.coords.as_slice(),
+            value.len(),
+            false,
+            false,
+        )?)
+    }
+}

src/array/coord/interleaved/mod.rs

@@ -1,4 +1,6 @@
 mod array;
+#[cfg(feature = "geos")]
+mod geos;
 mod mutable;
 
 pub use array::InterleavedCoordBuffer;

src/array/coord/separated/geos.rs

@@ -0,0 +1,16 @@
+use crate::array::SeparatedCoordBuffer;
+use crate::error::{GeoArrowError, Result};
+use geos::CoordSeq;
+
+impl TryFrom<SeparatedCoordBuffer> for CoordSeq<'_> {
+    type Error = GeoArrowError;
+
+    fn try_from(value: SeparatedCoordBuffer) -> Result<Self> {
+        Ok(CoordSeq::new_from_arrays(
+            value.x.as_slice(),
+            value.y.as_slice(),
+            None,
+            None,
+        )?)
+    }
+}

src/array/coord/separated/mod.rs

@@ -1,4 +1,6 @@
 mod array;
+#[cfg(feature = "geos")]
+mod geos;
 mod mutable;
 
 pub use array::SeparatedCoordBuffer;

src/array/linestring/array.rs

@@ -239,7 +239,7 @@ impl<O: Offset> LineStringArray<O> {
     /// Returns the value at slot `i` as a GEOS geometry.
     #[cfg(feature = "geos")]
     pub fn value_as_geos(&self, i: usize) -> geos::Geometry {
-        (&self.value_as_geo(i)).try_into().unwrap()
+        self.value(i).try_into().unwrap()
     }
 
     /// Gets the value at slot `i` as a GEOS geometry, additionally checking the validity bitmap
@@ -249,7 +249,7 @@ impl<O: Offset> LineStringArray<O> {
             return None;
         }
 
-        self.get_as_geo(i).as_ref().map(|g| g.try_into().unwrap())
+        Some(self.value_as_geos(i))
     }
 
     /// Iterator over GEOS geometry objects

src/array/linestring/mutable.rs

@@ -262,6 +262,29 @@ impl<O: Offset> TryFrom<WKBArray<O>> for MutableLineStringArray<O> {
     }
 }
 
+// #[cfg(feature = "geos")]
+// impl<O: Offset> TryFrom<Vec<Option<geos::Geometry<'_>>>> for MutableLineStringArray<O> {
+//     type Error = GeoArrowError;
+//     fn try_from(value: Vec<Option<geos::Geometry>>) -> std::result::Result<Self, Self::Error> {
+//         let length = value.len();
+//         let geos_linestring_objects: Vec<Option<GEOSLineString>> = value
+//             .iter()
+//             .map(|geom| {
+//                 geom.map(|geom| GEOSLineString::new_unchecked(std::borrow::Cow::Owned(geom)))
+//             })
+//             .collect();
+//         let (coord_capacity, geom_capacity) = first_pass(
+//             geos_linestring_objects.iter().map(|item| item.as_ref()),
+//             length,
+//         );
+//         Ok(second_pass(
+//             geos_linestring_objects.iter().map(|item| item.as_ref()),
+//             coord_capacity,
+//             geom_capacity,
+//         ))
+//     }
+// }
+
 /// LineString and MultiPoint have the same layout, so enable conversions between the two to change
 /// the semantic type
 impl<O: Offset> From<MutableLineStringArray<O>> for MutableMultiPointArray<O> {