pub use aflak_cake::*;
use variant_name::VariantName;

#[derive(Clone, PartialEq, Debug, VariantName, Serialize, Deserialize)]
pub enum AlgoIO {
    Integer(u64),
    Image2d(Vec<Vec<f64>>),
}

/// `never` type representing an impossible error (similar to ! in rust nightly)
#[derive(Clone, PartialEq, Debug)]
pub enum E {}

pub fn get_plus1_transform() -> Transformation<AlgoIO, E> {
    cake_transform!("Add 1", plus1<AlgoIO, E>(i: Integer = 0) -> Integer {
        vec![Ok(AlgoIO::Integer(i + 1))]
    })
}

pub fn get_minus1_transform() -> Transformation<AlgoIO, E> {
    cake_transform!("Substract 1", minus1<AlgoIO, E>(i: Integer) -> Integer {
        vec![Ok(AlgoIO::Integer(i - 1))]
    })
}

pub fn get_get1_transform() -> Transformation<AlgoIO, E> {
    cake_constant!(get1, AlgoIO::Integer(1))
}

pub fn get_get_image_transform() -> Transformation<AlgoIO, E> {
    cake_constant!(image, AlgoIO::Image2d(vec![vec![10.0; 10000]; 10000]))
}

lazy_static! {
    pub static ref TRANSFORMATIONS: Vec<Transformation<AlgoIO, E>> = {
        vec![
            get_plus1_transform(),
            get_minus1_transform(),
            get_get1_transform(),
            get_get_image_transform(),
        ]
    };
}

impl NamedAlgorithms<E> for AlgoIO {
    fn get_transform(s: &str) -> Option<&'static Transformation<AlgoIO, E>> {
        for t in TRANSFORMATIONS.iter() {
            if t.name == s {
                return Some(t);
            }
        }
        None
    }
}