feldera-sqllib 0.294.0

// I cannot use the standard geopoint object because it doesn't implement Ord

use crate::{some_function2, some_polymorphic_function2};
use ::serde::{Deserialize, Serialize};
use dbsp::algebra::F64;
use dbsp::num_entries_scalar;
use feldera_macros::IsNone;
use feldera_types::serde_with_context::SerializeWithContext;
use geo::EuclideanDistance;
use geo::Point;
use serde::ser::Error;
use size_of::*;

#[doc(hidden)]
#[derive(
    Default,
    Eq,
    Ord,
    Clone,
    Hash,
    PartialEq,
    PartialOrd,
    SizeOf,
    Serialize,
    Deserialize,
    Debug,
    rkyv::Archive,
    rkyv::Serialize,
    rkyv::Deserialize,
    IsNone,
)]
#[archive_attr(derive(Clone, Ord, Eq, PartialEq, PartialOrd))]
#[archive(compare(PartialEq, PartialOrd))]
pub struct GeoPoint(F64, F64);

#[doc(hidden)]
impl<C> SerializeWithContext<C> for GeoPoint {
    fn serialize_with_context<S>(&self, _serializer: S, _context: &C) -> Result<S::Ok, S::Error>
    where
        S: serde::Serializer,
    {
        Err(S::Error::custom(
            "serialization is not implemented for the GEORGRAPHY type",
        ))
    }
}

num_entries_scalar! {
    GeoPoint,
}

impl GeoPoint {
    #[doc(hidden)]
    pub fn new<T, S>(left: T, right: S) -> Self
    where
        F64: From<T>,
        F64: From<S>,
    {
        Self(F64::from(left), F64::from(right))
    }

    #[doc(hidden)]
    pub fn to_point(&self) -> Point {
        Point::new(self.0.into_inner(), self.1.into_inner())
    }

    #[doc(hidden)]
    pub fn distance(&self, other: &GeoPoint) -> F64 {
        let left = self.to_point();
        let right = other.to_point();
        F64::from(left.euclidean_distance(&right))
    }
}

#[doc(hidden)]
pub fn make_geopoint__(left: F64, right: F64) -> GeoPoint {
    GeoPoint::new(left, right)
}

some_function2!(make_geopoint, F64, F64, GeoPoint);

#[doc(hidden)]
pub fn st_distance_geopoint_geopoint(left: GeoPoint, right: GeoPoint) -> F64 {
    left.distance(&right)
}

some_polymorphic_function2!(st_distance, geopoint, GeoPoint, geopoint, GeoPoint, F64);