geoarrow_array/geozero/export/array/
linestring.rs

1use geozero::{GeomProcessor, GeozeroGeometry};
2
3use crate::array::LineStringArray;
4use crate::geozero::export::scalar::process_line_string;
5use crate::{GeoArrowArray, GeoArrowArrayAccessor};
6
7impl GeozeroGeometry for LineStringArray {
8    fn process_geom<P: GeomProcessor>(&self, processor: &mut P) -> geozero::error::Result<()>
9    where
10        Self: Sized,
11    {
12        let num_geometries = self.len();
13        processor.geometrycollection_begin(num_geometries, 0)?;
14
15        for geom_idx in 0..num_geometries {
16            process_line_string(&self.value(geom_idx).unwrap(), geom_idx, processor)?;
17        }
18
19        processor.geometrycollection_end(num_geometries - 1)?;
20        Ok(())
21    }
22}
23
24#[cfg(test)]
25mod test {
26    use geoarrow_schema::CoordType;
27    use geozero::ToWkt;
28
29    use crate::test::linestring::ls_array;
30
31    #[test]
32    fn geozero_process_geom() -> geozero::error::Result<()> {
33        let arr = ls_array(CoordType::Interleaved);
34        let wkt = ToWkt::to_wkt(&arr)?;
35        let expected = "GEOMETRYCOLLECTION(LINESTRING(0 1,1 2),LINESTRING EMPTY,LINESTRING(3 4,5 6),LINESTRING EMPTY)";
36        assert_eq!(wkt, expected);
37        Ok(())
38    }
39}