use geo_types::{Point, point};
use geoarrow_schema::{CoordType, Dimension, PointType};
use geoarrow_test::raw;
use crate::array::PointArray;
use crate::builder::PointBuilder;
pub(crate) fn p0() -> Point {
point!(
x: 0., y: 1.
)
}
pub(crate) fn p1() -> Point {
point!(
x: 1., y: 2.
)
}
pub(crate) fn p2() -> Point {
point!(
x: 2., y: 3.
)
}
pub(crate) fn point_array(coord_type: CoordType) -> PointArray {
let geoms = [Some(p0()), Some(p1()), None, Some(p2())];
let typ = PointType::new(Dimension::XY, Default::default()).with_coord_type(coord_type);
PointBuilder::from_nullable_points(geoms.iter().map(|x| x.as_ref()), typ).finish()
}
pub fn array(coord_type: CoordType, dim: Dimension) -> PointArray {
let typ = PointType::new(dim, Default::default()).with_coord_type(coord_type);
let geoms = match dim {
Dimension::XY => raw::point::xy::geoms(),
Dimension::XYZ => raw::point::xyz::geoms(),
Dimension::XYM => raw::point::xym::geoms(),
Dimension::XYZM => raw::point::xyzm::geoms(),
};
PointBuilder::from_nullable_points(geoms.iter().map(|x| x.as_ref()), typ).finish()
}