spenso 0.5.5

A tensor (n-dim array) network, iterating, and contraction (using automatic abstract index matching) library.
Documentation
use duplicate::duplicate;
use std::borrow::Borrow;

#[cfg(feature = "shadowing")]
use symbolica::{
    atom::Atom,
    domains::{float::Real, integer::Integer},
};

#[cfg(feature = "shadowing")]
use crate::shadowing::symbolica_utils::SerializableAtom;

pub trait One {
    fn one() -> Self;
}

pub trait Zero {
    fn zero() -> Self;
}

pub trait IsZero {
    fn is_zero(&self) -> bool;

    fn is_non_zero(&self) -> bool {
        !self.is_zero()
    }
}

impl<T: RefZero + PartialEq> IsZero for T {
    fn is_zero(&self) -> bool {
        self.ref_zero() == *self
    }
}

pub trait RefZero<T = Self>: Borrow<T> {
    fn ref_zero(&self) -> T;
}

pub trait RefOne {
    fn ref_one(&self) -> Self;
}

// #[cfg(feature = "shadowing")]
// impl<T: RefZero + NumericalFloatLike> RefZero for T {
//     fn ref_zero(&self) -> Self {
//         self.zero()
//     }
// } fu

#[cfg(feature = "shadowing")]
impl RefZero for Atom {
    fn ref_zero(&self) -> Self {
        Atom::num(0)
    }
}

#[cfg(feature = "shadowing")]
impl RefOne for Atom {
    fn ref_one(&self) -> Self {
        Atom::num(1)
    }
}

#[cfg(feature = "shadowing")]
impl RefZero for Integer {
    fn ref_zero(&self) -> Self {
        Integer::zero()
    }
}

#[cfg(feature = "shadowing")]
impl RefZero for SerializableAtom {
    fn ref_zero(&self) -> Self {
        Atom::num(0).into()
    }
}

#[cfg(feature = "shadowing")]
impl<T: RefOne + Real + RefZero> RefOne for symbolica::domains::float::Complex<T> {
    fn ref_one(&self) -> Self {
        Self::new(self.re.ref_one(), self.im.ref_zero())
    }
}

#[cfg(feature = "shadowing")]
impl<T: Real + RefZero> RefZero for symbolica::domains::float::Complex<T> {
    fn ref_zero(&self) -> Self {
        Self::new(self.re.ref_zero(), self.im.ref_zero())
    }
}

// impl<T: num::Zero> RefZero for T { future impls Grrr
//     fn zero(&self) -> Self {
//         T::zero()
//     }
// }

duplicate! {
    [types zero one;
        [f32] [0.0] [1.];
        [f64] [0.0] [1.];
        [i8] [0] [1];
        [i16] [0] [1] ;
        [i32] [0] [1];
        [i64] [0] [1];
        [i128] [0] [1];
        [u8] [0] [1];
        [u16] [0] [1];
        [u32] [0] [1];
        [u64] [0] [1];
        [u128] [0] [1];
        ]

    impl RefZero for types{
        fn ref_zero(&self)-> Self{
            zero
        }
    }

    impl RefZero<types> for &types{
        fn ref_zero(&self)-> types{
            zero
        }
    }

    impl RefOne for types{
        fn ref_one(&self)-> Self{
            one
        }
    }
}