s2json-core 1.51.1

use crate::*;
use alloc::vec::Vec;
use serde::{Deserialize, Serialize};

/// Enum to represent specific vector geometry types as strings
#[derive(Serialize, Deserialize, Copy, Clone, Debug, PartialEq, Default)]
pub enum VectorGeometryType {
    /// Point
    #[default]
    Point,
    /// MultiPoint
    MultiPoint,
    /// LineString
    LineString,
    /// MultiLineString
    MultiLineString,
    /// Polygon
    Polygon,
    /// MultiPolygon
    MultiPolygon,
}
impl From<&str> for VectorGeometryType {
    fn from(s: &str) -> Self {
        match s {
            "Point" => VectorGeometryType::Point,
            "MultiPoint" => VectorGeometryType::MultiPoint,
            "LineString" => VectorGeometryType::LineString,
            "MultiLineString" => VectorGeometryType::MultiLineString,
            "Polygon" => VectorGeometryType::Polygon,
            "MultiPolygon" => VectorGeometryType::MultiPolygon,
            _ => panic!("Invalid vector geometry type: {}", s),
        }
    }
}

/// Definition of a Vector MultiPoint
pub type VectorMultiPoint<M = MValue> = Vec<VectorPoint<M>>;
/// Definition of a Vector LineString
pub type VectorLineString<M = MValue> = Vec<VectorPoint<M>>;
/// Definition of a Vector MultiLineString
pub type VectorMultiLineString<M = MValue> = Vec<VectorLineString<M>>;
/// Definition of a Vector Polygon
pub type VectorPolygon<M = MValue> = Vec<VectorLineString<M>>;
/// Definition of a Vector MultiPolygon
pub type VectorMultiPolygon<M = MValue> = Vec<VectorPolygon<M>>;

/// # Vector Geometry
///
/// ## Description
/// All possible geometry shapes. Builds ontop of [`VectorBaseGeometry`].
///
/// ## Usage
/// - [`VectorGeometry::bbox`]: Get the bbox of the geometry
/// - [`VectorGeometry::vec_bbox`]: Get the internal 0-1 clipping style vector bbox
/// - [`VectorGeometry::point`]: Get the geometry point
/// - [`VectorGeometry::new_point`]: Create a new point
/// - [`VectorGeometry::multipoint`]: Get the geometry multi point
/// - [`VectorGeometry::new_multipoint`]: Create a new multipoint
/// - [`VectorGeometry::linestring`]: Get the geometry line string
/// - [`VectorGeometry::new_linestring`]: Create a new linestring
/// - [`VectorGeometry::multilinestring`]: Get the geometry multi line string
/// - [`VectorGeometry::new_multilinestring`]: Create a new multi line string
/// - [`VectorGeometry::polygon`]: Get the geometry polygon
/// - [`VectorGeometry::new_polygon`]: Create a new polygon
/// - [`VectorGeometry::multipolygon`]: Get the geometry multi polygon
/// - [`VectorGeometry::new_multipolygon`]: Create a new multi polygon
/// - [`VectorGeometry::set_tess`]: Set the tessellation of the geometry (polygon and multipolygon only)
/// - [`VectorGeometry::set_indices`]: Set the indices of the geometry (polygon and multipolygon only)
/// - [`VectorGeometry::to_m_geometry`]: Convert the geometry so that all m-values are MValue rather then user defined
#[derive(Clone, Serialize, Debug, PartialEq)]
#[serde(untagged)]
pub enum VectorGeometry<M: Clone + Default = MValue> {
    /// Point Shape
    Point(VectorPointGeometry<M>),
    /// MultiPoint Shape
    MultiPoint(VectorMultiPointGeometry<M>),
    /// LineString Shape
    LineString(VectorLineStringGeometry<M>),
    /// MultiLineString Shape
    MultiLineString(VectorMultiLineStringGeometry<M>),
    /// Polygon Shape
    Polygon(VectorPolygonGeometry<M>),
    /// MultiPolygon Shape
    MultiPolygon(VectorMultiPolygonGeometry<M>),
}
impl<M: Clone + Default> VectorGeometry<M> {
    /// Get the bbox of the geometry
    pub fn bbox(&self) -> &Option<BBox3D> {
        match self {
            VectorGeometry::Point(g) => &g.bbox,
            VectorGeometry::MultiPoint(g) => &g.bbox,
            VectorGeometry::LineString(g) => &g.bbox,
            VectorGeometry::MultiLineString(g) => &g.bbox,
            VectorGeometry::Polygon(g) => &g.bbox,
            VectorGeometry::MultiPolygon(g) => &g.bbox,
        }
    }

    /// Get the vec_bbox of the geometry
    pub fn vec_bbox(&self) -> &Option<BBox3D> {
        match self {
            VectorGeometry::Point(g) => &g.vec_bbox,
            VectorGeometry::MultiPoint(g) => &g.vec_bbox,
            VectorGeometry::LineString(g) => &g.vec_bbox,
            VectorGeometry::MultiLineString(g) => &g.vec_bbox,
            VectorGeometry::Polygon(g) => &g.vec_bbox,
            VectorGeometry::MultiPolygon(g) => &g.vec_bbox,
        }
    }

    /// Get the geometry point
    pub fn point(&self) -> Option<&VectorPoint<M>> {
        match self {
            VectorGeometry::Point(g) => Some(&g.coordinates),
            _ => None,
        }
    }

    /// Create a new point
    pub fn new_point(coordinates: VectorPoint<M>, bbox: Option<BBox3D>) -> Self {
        VectorGeometry::Point(VectorPointGeometry {
            _type: VectorGeometryType::Point,
            is_3d: coordinates.z.is_some(),
            coordinates,
            bbox,
            ..Default::default()
        })
    }

    /// Get the geometry multi point
    pub fn multipoint(&self) -> Option<&VectorMultiPoint<M>> {
        match self {
            VectorGeometry::MultiPoint(g) => Some(&g.coordinates),
            _ => None,
        }
    }

    /// Create a new multipoint
    pub fn new_multipoint(coordinates: VectorMultiPoint<M>, bbox: Option<BBox3D>) -> Self {
        VectorGeometry::MultiPoint(VectorMultiPointGeometry {
            _type: VectorGeometryType::MultiPoint,
            is_3d: coordinates.iter().any(|point| point.z.is_some()),
            coordinates,
            bbox,
            ..Default::default()
        })
    }

    /// Get the geometry linestring
    pub fn linestring(&self) -> Option<&VectorLineString<M>> {
        match self {
            VectorGeometry::LineString(g) => Some(&g.coordinates),
            _ => None,
        }
    }

    /// Create a new linestring
    pub fn new_linestring(coordinates: VectorLineString<M>, bbox: Option<BBox3D>) -> Self {
        VectorGeometry::LineString(VectorLineStringGeometry {
            _type: VectorGeometryType::LineString,
            is_3d: coordinates.iter().any(|point| point.z.is_some()),
            coordinates,
            bbox,
            ..Default::default()
        })
    }

    /// Get the geometry multilinestring
    pub fn multilinestring(&self) -> Option<&VectorMultiLineString<M>> {
        match self {
            VectorGeometry::MultiLineString(g) => Some(&g.coordinates),
            _ => None,
        }
    }

    /// Create a new multilinestring
    pub fn new_multilinestring(
        coordinates: VectorMultiLineString<M>,
        bbox: Option<BBox3D>,
    ) -> Self {
        VectorGeometry::MultiLineString(VectorMultiLineStringGeometry {
            _type: VectorGeometryType::MultiLineString,
            is_3d: coordinates.iter().any(|line| line.iter().any(|point| point.z.is_some())),
            coordinates,
            bbox,
            ..Default::default()
        })
    }

    /// Get the geometry polygon
    pub fn polygon(&self) -> Option<&VectorPolygon<M>> {
        match self {
            VectorGeometry::Polygon(g) => Some(&g.coordinates),
            _ => None,
        }
    }

    /// Create a new polygon
    pub fn new_polygon(coordinates: VectorPolygon<M>, bbox: Option<BBox3D>) -> Self {
        VectorGeometry::Polygon(VectorPolygonGeometry {
            _type: VectorGeometryType::Polygon,
            is_3d: coordinates.iter().any(|ring| ring.iter().any(|point| point.z.is_some())),
            coordinates,
            bbox,
            ..Default::default()
        })
    }

    /// Get the geometry multipolygon
    pub fn multipolygon(&self) -> Option<&VectorMultiPolygon<M>> {
        match self {
            VectorGeometry::MultiPolygon(g) => Some(&g.coordinates),
            _ => None,
        }
    }

    /// Create a new multipolygon
    pub fn new_multipolygon(coordinates: VectorMultiPolygon<M>, bbox: Option<BBox3D>) -> Self {
        VectorGeometry::MultiPolygon(VectorMultiPolygonGeometry {
            _type: VectorGeometryType::MultiPolygon,
            is_3d: coordinates.iter().any(|polygon| {
                polygon.iter().any(|ring| ring.iter().any(|point| point.z.is_some()))
            }),
            coordinates,
            bbox,
            ..Default::default()
        })
    }

    /// set the tessellation of the geometry (polygon and multipolygon only)
    pub fn set_tess(&mut self, tessellation: Vec<f64>) {
        match self {
            VectorGeometry::Polygon(g) => g.tessellation = Some(tessellation),
            VectorGeometry::MultiPolygon(g) => g.tessellation = Some(tessellation),
            _ => {}
        }
    }

    /// set the indices of the geometry (polygon and multipolygon only)
    pub fn set_indices(&mut self, indices: Vec<u32>) {
        match self {
            VectorGeometry::Polygon(g) => g.indices = Some(indices),
            VectorGeometry::MultiPolygon(g) => g.indices = Some(indices),
            _ => {}
        }
    }

    /// Convert the geometry so that all m-values are MValue rather then user defined
    pub fn to_m_geometry(&self) -> VectorGeometry<MValue>
    where
        M: MValueCompatible,
    {
        match self {
            VectorGeometry::Point(g) => VectorGeometry::Point(VectorPointGeometry {
                _type: g._type,
                is_3d: g.is_3d,
                coordinates: g.coordinates.to_m_value(),
                offset: g.offset.clone(),
                bbox: g.bbox,
                vec_bbox: g.vec_bbox,
                ..Default::default()
            }),
            VectorGeometry::MultiPoint(g) => VectorGeometry::MultiPoint(VectorMultiPointGeometry {
                _type: g._type,
                is_3d: g.is_3d,
                coordinates: g.coordinates.iter().map(|point| point.to_m_value()).collect(),
                offset: g.offset,
                bbox: g.bbox,
                vec_bbox: g.vec_bbox,
                ..Default::default()
            }),
            VectorGeometry::LineString(g) => VectorGeometry::LineString(VectorLineStringGeometry {
                _type: g._type,
                is_3d: g.is_3d,
                coordinates: g.coordinates.iter().map(|point| point.to_m_value()).collect(),
                offset: g.offset,
                bbox: g.bbox,
                vec_bbox: g.vec_bbox,
                ..Default::default()
            }),
            VectorGeometry::MultiLineString(g) => {
                VectorGeometry::MultiLineString(VectorMultiLineStringGeometry {
                    _type: g._type,
                    is_3d: g.is_3d,
                    coordinates: g
                        .coordinates
                        .iter()
                        .map(|line| line.iter().map(|point| point.to_m_value()).collect())
                        .collect(),
                    offset: g.offset.clone(),
                    bbox: g.bbox,
                    vec_bbox: g.vec_bbox,
                    ..Default::default()
                })
            }
            VectorGeometry::Polygon(g) => VectorGeometry::Polygon(VectorPolygonGeometry {
                _type: g._type,
                is_3d: g.is_3d,
                coordinates: g
                    .coordinates
                    .iter()
                    .map(|ring| ring.iter().map(|point| point.to_m_value()).collect())
                    .collect(),
                offset: g.offset.clone(),
                bbox: g.bbox,
                vec_bbox: g.vec_bbox,
                ..Default::default()
            }),
            VectorGeometry::MultiPolygon(g) => {
                VectorGeometry::MultiPolygon(VectorMultiPolygonGeometry {
                    _type: g._type,
                    is_3d: g.is_3d,
                    coordinates: g
                        .coordinates
                        .iter()
                        .map(|polygon| {
                            polygon
                                .iter()
                                .map(|ring| ring.iter().map(|point| point.to_m_value()).collect())
                                .collect()
                        })
                        .collect(),
                    offset: g.offset.clone(),
                    bbox: g.bbox,
                    vec_bbox: g.vec_bbox,
                    ..Default::default()
                })
            }
        }
    }
}
impl<M: Clone + Default> Default for VectorGeometry<M> {
    fn default() -> Self {
        VectorGeometry::Point(VectorPointGeometry::default())
    }
}

/// BaseGeometry is the a generic geometry type
#[derive(Serialize, Deserialize, Clone, Debug, PartialEq, Default)]
pub struct VectorBaseGeometry<G = VectorGeometry, O = VectorOffsets> {
    /// The geometry type
    #[serde(rename = "type")]
    pub _type: VectorGeometryType,
    /// Specifies if the geometry is 3D or 2D
    #[serde(rename = "is3D", default)]
    pub is_3d: bool,
    /// The geometry shape
    pub coordinates: G,
    /// The geometry offsets if applicable
    #[serde(skip_serializing_if = "Option::is_none")]
    pub offset: Option<O>,
    /// The BBox shape - always in lon-lat
    #[serde(skip_serializing_if = "Option::is_none")]
    pub bbox: Option<BBox3D>,
    /// temporary bbox to track 0->1 clipping
    #[serde(skip)]
    pub vec_bbox: Option<BBox3D>,
    /// Polygon and MultiPolygon specific property
    #[serde(skip_serializing_if = "Option::is_none")]
    pub indices: Option<Vec<u32>>,
    /// Polygon and MultiPolygon specific property
    #[serde(skip_serializing_if = "Option::is_none")]
    pub tessellation: Option<Vec<f64>>,
}

/// All possible geometry offsets
#[derive(Serialize, Deserialize, Clone, Debug, PartialEq)]
pub enum VectorOffsets {
    /// LineString offset
    LineOffset(VectorLineOffset),
    /// MultiLineString offset
    MultiLineOffset(VectorMultiLineOffset),
    /// Polygon offset
    PolygonOffset(VectorPolygonOffset),
    /// MultiPolygon offset
    MultiPolygonOffset(VectorMultiPolygonOffset),
}
impl Default for VectorOffsets {
    fn default() -> Self {
        VectorOffsets::LineOffset(0.0)
    }
}
/// An offset defines how far the starting line is from the original starting point pre-slice
pub type VectorLineOffset = f64;
/// A collection of offsets
pub type VectorMultiLineOffset = Vec<VectorLineOffset>;
/// A collection of offsets
pub type VectorPolygonOffset = VectorMultiLineOffset;
/// A collection of collections of offsets
pub type VectorMultiPolygonOffset = Vec<VectorPolygonOffset>;

/// PointGeometry is a point
pub type VectorPointGeometry<M = MValue> = VectorBaseGeometry<VectorPoint<M>>;
/// MultiPointGeometry contains multiple points
pub type VectorMultiPointGeometry<M = MValue> =
    VectorBaseGeometry<VectorMultiPoint<M>, VectorLineOffset>;
/// LineStringGeometry is a line
pub type VectorLineStringGeometry<M = MValue> =
    VectorBaseGeometry<VectorLineString<M>, VectorLineOffset>;
/// MultiLineStringGeometry contains multiple lines
pub type VectorMultiLineStringGeometry<M = MValue> =
    VectorBaseGeometry<VectorMultiLineString<M>, VectorMultiLineOffset>;
/// PolygonGeometry is a polygon with potential holes
pub type VectorPolygonGeometry<M = MValue> =
    VectorBaseGeometry<VectorPolygon<M>, VectorPolygonOffset>;
/// MultiPolygonGeometry is a polygon with multiple polygons with their own potential holes
pub type VectorMultiPolygonGeometry<M = MValue> =
    VectorBaseGeometry<VectorMultiPolygon<M>, VectorMultiPolygonOffset>;