ellip-dev-utils 1.1.0

/*
 * Ellip is licensed under The 3-Clause BSD, see LICENSE.
 * Copyright 2025 Sira Pornsiriprasert <code@psira.me>
 */

use num_traits::Float;
use std::fmt::Debug;
use tabled::{Table, Tabled, settings::Style};

use crate::stats::Stats;

/// Calculates error in unit of epsilon
pub fn err_func<T: Float>(a: T, b: T) -> f64 {
    let abs_err = (a - b).abs();
    if abs_err < T::epsilon() {
        return 0.0;
    }

    let a = a.abs();
    let b = b.abs();

    (abs_err / a.max(b) / T::epsilon())
        .to_f64()
        .expect("Cannot convert to f64")
}

#[derive(Debug, Clone)]
pub struct Case<T: Float> {
    pub inputs: Vec<T>,
    pub expected: T,
}

pub fn compute_errors_from_cases<T: Float + Debug>(
    func: &dyn Fn(&Vec<T>) -> T,
    cases: Vec<Case<T>>,
) -> Vec<f64> {
    cases
        .iter()
        .map(|case| {
            if case.expected.is_finite() {
                let res = func(&case.inputs);
                let error = err_func(res, case.expected);
                // if e > 20.0 {
                //     println!(
                //         "Using parameters: {:?}, got={:?}, actual={:?} (error={:.2})",
                //         &case.inputs, res, case.expected, e
                //     );
                // }
                error
            } else {
                f64::NAN
            }
        })
        .collect()
}

pub fn format_float(value: &f64) -> String {
    if value.is_nan() {
        "NAN".to_string()
    } else if *value >= 1e3 {
        format!("{:.2e}", value)
    } else {
        format!("{:.2}", value)
    }
}

fn format_mu(value: &u64) -> String {
    if *value >= 10000 {
        format!("{:e}", value)
    } else {
        format!("{}", value)
    }
}

#[derive(Tabled)]
pub struct ErrorEntry<'a> {
    #[tabled(rename = "Function")]
    name: &'a str,
    #[tabled(rename = "Mean (ε)", display = "format_float")]
    mean: f64,
    #[tabled(rename = "Median (ε)", display = "format_float")]
    median: f64,
    #[tabled(rename = "P99 (ε)", display = "format_float")]
    p99: f64,
    #[tabled(rename = "Max (ε)", display = "format_float")]
    max: f64,
    #[tabled(rename = "Variance (ε²)", display = "format_float")]
    variance: f64,
    #[tabled(rename = "μ (ε)", display = "format_mu")]
    mu: u64,
}

pub fn generate_error_entry_from_file<T: Float + Debug>(
    file_path: &str,
    func: &dyn Fn(&Vec<T>) -> T,
) -> Stats {
    let result = crate::parser::read_wolfram_data(file_path);
    match result {
        Ok(cases) => Stats::from_vec(&compute_errors_from_cases(func, cases)),
        Err(_) => Stats::nan(),
    }
}

pub fn generate_error_table(entries: &[(&str, u64, Stats, Stats)]) -> [String; 2] {
    let f64_rows: Vec<ErrorEntry> = entries
        .iter()
        .map(|(name, mu, f64_stats, _)| ErrorEntry {
            name,
            mean: f64_stats.mean,
            median: f64_stats.median,
            p99: f64_stats.p99,
            max: f64_stats.max,
            variance: f64_stats.variance,
            mu: *mu,
        })
        .collect();

    let f32_rows: Vec<ErrorEntry> = entries
        .iter()
        .map(|(name, mu, _, f32_stats)| ErrorEntry {
            name,
            mean: f32_stats.mean,
            median: f32_stats.median,
            p99: f32_stats.p99,
            max: f32_stats.max,
            variance: f32_stats.variance,
            mu: *mu,
        })
        .collect();

    [
        Table::new(f64_rows).with(Style::markdown()).to_string(),
        Table::new(f32_rows).with(Style::markdown()).to_string(),
    ]
}

#[macro_export]
macro_rules! get_entry {
    ($file_name: expr, $name: expr, $func: expr, $arg_count: tt, $mu: expr) => {{
        let f64_entries = {
            let file_path_f64 = concat!["tests/data/", $file_name, ".csv"];
            ellip_dev_utils::func_wrapper!($func, f64, $arg_count);
            ellip_dev_utils::test_report::generate_error_entry_from_file(
                &file_path_f64,
                &wrapped_func,
            )
        };
        let f32_entries = {
            let file_path_f32 = concat!["tests/data/f32/", $file_name, ".csv"];
            ellip_dev_utils::func_wrapper!($func, f32, $arg_count);
            ellip_dev_utils::test_report::generate_error_entry_from_file(
                &file_path_f32,
                &wrapped_func,
            )
        };

        ($name, $mu, f64_entries, f32_entries)
    }};
}

pub fn format_performance(value: &f64) -> String {
    if value.is_nan() {
        "NAN".to_string()
    } else if *value < 1000.0 {
        format!("{:.1} ns", value)
    } else if *value < 1_000_000.0 {
        format!("{:.1} μs", value / 1000.0)
    } else {
        format!("{:.1} ms", value / 1_000_000.0)
    }
}

#[derive(Tabled)]
pub struct SummaryEntry<'a> {
    #[tabled(rename = "Function")]
    name: &'a str,
    #[tabled(rename = "Median Error (ε)", display = "format_float")]
    median_error: f64,
    #[tabled(rename = "Max Error (ε)", display = "format_float")]
    max_error: f64,
    #[tabled(rename = "Mean Performance", display = "format_performance")]
    mean_performance: f64,
}

pub fn generate_summary_table(entries: &[(&str, Stats, f64)]) -> String {
    let rows: Vec<SummaryEntry> = entries
        .iter()
        .map(|(name, stats, perf)| SummaryEntry {
            name,
            median_error: stats.median,
            max_error: stats.max,
            mean_performance: *perf / stats.n as f64,
        })
        .collect();

    Table::new(rows).with(Style::markdown()).to_string()
}

#[macro_export]
macro_rules! get_summary_entry {
    ($group:expr, $name:expr, $func:expr, $arg_count:tt, $test_file_name:expr) => {{
        use ellip_dev_utils::{
            benchmark, file, parser, stats,
            test_report::{self, Case},
        };
        use std::path::Path;

        ellip_dev_utils::func_wrapper!($func, f64, $arg_count);

        let test_paths = file::find_test_files($test_file_name, "wolfram");
        let cases = test_paths
            .iter()
            .flat_map(|test_path| parser::read_wolfram_data(test_path.to_str().unwrap()).unwrap())
            .collect::<Vec<Case<f64>>>();
        let stats = stats::Stats::from_vec(&test_report::compute_errors_from_cases(
            &wrapped_func,
            cases,
        ));

        // Criterion directory structure: target/criterion/<group>/<func>/new/estimates.json
        let estimates_path_buf = Path::new("target/criterion")
            .join($group)
            .join(stringify!($func))
            .join("new")
            .join("estimates.json");
        let perf = benchmark::extract_criterion_mean(&estimates_path_buf).unwrap_or(f64::NAN);

        ($name, stats, perf)
    }};
    ($group:expr, $name:expr, $func:expr, $arg_count:tt) => {{
        get_summary_entry! {$group, $name, $func, $arg_count, stringify!($func)}
    }};
}

pub fn format_exp(value: &f64) -> String {
    if value.is_nan() {
        "NAN".to_string()
    } else if *value >= 1e3 {
        format!("${:.2e}$", value).replace("e", "*10^")
    } else {
        format!("${:.2}$", value)
    }
}

#[derive(Tabled)]
pub struct AccuracyEntry<'a> {
    #[tabled(rename = "**Function**")]
    name: &'a str,
    #[tabled(rename = "**Median (ε)**", display = "format_exp")]
    median: f64,
    #[tabled(rename = "**Max (ε)**", display = "format_exp")]
    max: f64,
    #[tabled(rename = "**Variance (ε²)**", display = "format_exp")]
    variance: f64,
}

pub fn generate_accuracy_summary_table(entries: &[(&str, Stats)]) -> String {
    let rows: Vec<AccuracyEntry> = entries
        .iter()
        .map(|(name, stats)| AccuracyEntry {
            name,
            median: stats.median,
            max: stats.max,
            variance: stats.variance,
        })
        .collect();

    Table::new(rows).with(Style::ascii()).to_string()
}

#[macro_export]
macro_rules! get_accuracy_entry {
    ($group:expr, $name:expr, $func:expr, $arg_count:tt, $test_file_name:expr) => {{
        use ellip_dev_utils::{
            file, parser, stats,
            test_report::{self, Case},
        };
        use std::path::Path;

        ellip_dev_utils::func_wrapper!($func, f64, $arg_count);

        let test_paths = file::find_test_files($test_file_name, "wolfram");
        let cases = test_paths
            .iter()
            .flat_map(|test_path| parser::read_wolfram_data(test_path.to_str().unwrap()).unwrap())
            .collect::<Vec<Case<f64>>>();
        let stats = stats::Stats::from_vec(&test_report::compute_errors_from_cases(
            &wrapped_func,
            cases,
        ));

        ($name, stats)
    }};
    ($group:expr, $name:expr, $func:expr, $arg_count:tt) => {{
        get_accuracy_entry! {$group, $name, $func, $arg_count, stringify!($func)}
    }};
}