geoarrow_array/geozero/export/array/
multipoint.rs

1use geozero::{GeomProcessor, GeozeroGeometry};
2
3use crate::array::MultiPointArray;
4use crate::geozero::export::scalar::process_multi_point;
5use crate::{GeoArrowArray, GeoArrowArrayAccessor};
6
7impl GeozeroGeometry for MultiPointArray {
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_point(&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, MultiPointType};
27    use geozero::ToWkt;
28    use geozero::error::Result;
29
30    use crate::builder::MultiPointBuilder;
31    use crate::test::multipoint::{mp0, mp1};
32
33    #[test]
34    fn geozero_process_geom() -> Result<()> {
35        let typ = MultiPointType::new(Dimension::XY, Default::default());
36        let geo_arr = MultiPointBuilder::from_multi_points(&[&mp0(), &mp1()], typ).finish();
37        let wkt = ToWkt::to_wkt(&geo_arr)?;
38        let expected = "GEOMETRYCOLLECTION(MULTIPOINT(0 1,1 2),MULTIPOINT(3 4,5 6))";
39        assert_eq!(wkt, expected);
40        Ok(())
41    }
42}