use std::time::Instant;

use crseo::{
    wavefrontsensor::{PistonSensor, SegmentCalibration},
    Atmosphere, Builder, FromBuilder, Gmt, Propagation, SegmentWiseSensor,
    SegmentWiseSensorBuilder, Source, WavefrontSensor,
};

fn main() -> anyhow::Result<()> {
    let n_mode = 1;

    let builder = PistonSensor::builder().pupil_sampling(401);
    let mut wfs = builder.clone().build().unwrap();
    let src_builder = Source::builder().pupil_sampling(92);

    dbg!(&builder);

    /*     let now = Instant::now();
    let mut slopes_mat = builder.calibrate(
        SegmentCalibration::modes("Karhunen-Loeve", 0..n_mode, "M2"),
        src_builder.clone(),
    );
    println!(
        "M2 {}modes/segment calibrated in {}s",
        n_mode,
        now.elapsed().as_secs()
    );
    slopes_mat.pseudo_inverse().unwrap(); */

    let mut gmt = Gmt::builder().m2("Karhunen-Loeve", n_mode).build().unwrap();
    let mut src = src_builder.build().unwrap();

    let mut atm = Atmosphere::builder().build()?;

    let mut buffer = vec![0f64; 7 * n_mode];
    let gain = 0.5;

    for i in 0..10 {
        wfs.reset();
        src.through(&mut gmt).xpupil().through(&mut atm);
        wfs.propagate(&mut src);
        src.through(&mut gmt).xpupil().through(&mut atm);
        wfs.propagate(&mut src);

        println!(
            "#{:03}: WFE RMS [nm]: {:4.0?} {:4.0?} {:4.0?}",
            i,
            src.wfe_rms_10e(-9),
            src.segment_wfe_rms_10e(-9),
            src.segment_piston_10e(-9)
        );

        /*         let coefs = (&slopes_mat * &wfs).unwrap();
        // dbg!(&coefs);

        buffer
            .iter_mut()
            .zip(coefs)
            .for_each(|(b, c)| *b -= gain * c as f64);
        // dbg!(&buffer);
        gmt.m2_modes(&buffer); */
    }

    /*     let n = src.pupil_sampling();
    let _: complot::Heatmap = (
        (src.phase().as_slice(), (n, n)),
        Some(complot::Config::new().filename("opd.png")),
    )
        .into(); */

    Ok(())
}