geoarrow_array/geozero/export/array/
multilinestring.rs

1use geozero::{GeomProcessor, GeozeroGeometry};
2
3use crate::array::MultiLineStringArray;
4use crate::geozero::export::scalar::process_multi_line_string;
5use crate::{GeoArrowArray, GeoArrowArrayAccessor};
6
7impl GeozeroGeometry for MultiLineStringArray {
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_multi_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::{Dimension, MultiLineStringType};
27    use geozero::ToWkt;
28
29    use crate::builder::MultiLineStringBuilder;
30    use crate::test::multilinestring::{ml0, ml1};
31
32    #[test]
33    fn geozero_process_geom() -> geozero::error::Result<()> {
34        let typ = MultiLineStringType::new(Dimension::XY, Default::default());
35        let geo_arr =
36            MultiLineStringBuilder::from_multi_line_strings(&[&ml0(), &ml1()], typ).finish();
37        let wkt = ToWkt::to_wkt(&geo_arr)?;
38        let expected = "GEOMETRYCOLLECTION(MULTILINESTRING((-111 45,-111 41,-104 41,-104 45)),MULTILINESTRING((-111 45,-111 41,-104 41,-104 45),(-110 44,-110 42,-105 42,-105 44)))";
39        assert_eq!(wkt, expected);
40        Ok(())
41    }
42}