graphrecords-query 0.5.0

mod absolute;
mod ceil;
mod clip;
mod cube_root;
mod exponential;
mod floor;
mod logarithm;
mod negate;
mod round;
mod sign;
mod square_root;

use crate::{
    QueryResult, ValueDomain,
    element::{BarePipeline, IndexedValuePipeline, Pipeline, Preserving},
    index::IndexDomain,
    registry::OperationManifest,
};
pub use absolute::AbsoluteOperation;
pub use ceil::CeilOperation;
pub use clip::ClipOperation;
pub use cube_root::CubeRootOperation;
pub use exponential::ExponentialOperation;
pub use floor::FloorOperation;
pub use logarithm::LogarithmOperation;
pub use negate::NegateOperation;
pub use round::RoundOperation;
pub use sign::SignOperation;
pub use square_root::SquareRootOperation;

pub(super) fn operation_manifests() -> Vec<OperationManifest> {
    vec![
        absolute::operation_manifest(),
        ceil::operation_manifest(),
        clip::operation_manifest(),
        cube_root::operation_manifest(),
        exponential::operation_manifest(),
        floor::operation_manifest(),
        logarithm::operation_manifest(),
        negate::operation_manifest(),
        round::operation_manifest(),
        sign::operation_manifest(),
        square_root::operation_manifest(),
    ]
}

fn numeric_indexed<'a, I, V, F>(
    label: &'static str,
    operation: F,
) -> IndexedValuePipeline<'a, I, V, V, Preserving>
where
    I: IndexDomain,
    V: ValueDomain,
    F: Fn(&'static str, V::Value<'a>) -> QueryResult<V::Value<'a>> + 'a,
{
    Pipeline::keyed(move |index, item: QueryResult<_>| {
        item.and_then(|value| operation(label, value).map_err(|failure| failure.at::<I>(&index)))
    })
}

fn numeric_bare<'a, V, F>(label: &'static str, operation: F) -> BarePipeline<'a, V, V, Preserving>
where
    V: ValueDomain,
    F: Fn(&'static str, V::Value<'a>) -> QueryResult<V::Value<'a>> + 'a,
{
    Pipeline::new(move |item: QueryResult<_>| item.and_then(|value| operation(label, value)))
}