abs.rs

use polars_core::prelude::*;

/// Convert numerical values to their absolute value.
pub fn abs(s: &Series) -> PolarsResult<Series> {
    use DataType::*;
    let out = match s.dtype() {
        #[cfg(feature = "dtype-i8")]
        Int8 => s.i8().unwrap().wrapping_abs().into_series(),
        #[cfg(feature = "dtype-i16")]
        Int16 => s.i16().unwrap().wrapping_abs().into_series(),
        Int32 => s.i32().unwrap().wrapping_abs().into_series(),
        Int64 => s.i64().unwrap().wrapping_abs().into_series(),
        #[cfg(feature = "dtype-i128")]
        Int128 => s.i128().unwrap().wrapping_abs().into_series(),
        #[cfg(feature = "dtype-f16")]
        Float16 => s.f16().unwrap().wrapping_abs().into_series(),
        Float32 => s.f32().unwrap().wrapping_abs().into_series(),
        Float64 => s.f64().unwrap().wrapping_abs().into_series(),
        #[cfg(feature = "dtype-decimal")]
        Decimal(_, _) => {
            let ca = s.decimal().unwrap();
            let precision = ca.precision();
            let scale = ca.scale();

            let out = ca.physical().wrapping_abs();
            out.into_decimal_unchecked(precision, scale).into_series()
        },
        #[cfg(feature = "dtype-duration")]
        Duration(_) => {
            let physical = s.to_physical_repr();
            let ca = physical.i64().unwrap();
            let out = ca.wrapping_abs().into_series();
            out.cast(s.dtype())?
        },
        dt if dt.is_unsigned_integer() => s.clone(),
        dt => polars_bail!(opq = abs, dt),
    };
    Ok(out)
}