use std::time::Instant;

use crseo::{Builder, FromBuilder, Gmt, Source};

#[test]
fn ray_tracing() {
    let mut src = Source::builder().build().unwrap();
    let mut gmt = Gmt::builder().build().unwrap();
    let now = Instant::now();
    src.through(&mut gmt).xpupil();
    println!("Ray tracing in {:?}", now.elapsed());
    let wfe = src.wfe_rms_10e(-9)[0];
    println!("WFE RMS: {}", wfe);
    assert!((wfe - 0.7966555).abs() < 1e-3);

    /*
    Ray tracing in:
     * g6e.4xlarge: 2.07499ms
    */
}

#[test]
fn amplitude() {
    let mut src = Source::builder().build().unwrap();
    let mut gmt = Gmt::builder().build().unwrap();
    let now = Instant::now();
    src.through(&mut gmt).xpupil();

    let _: complot::Heatmap = ((src.amplitude().as_slice(), (512, 512)), None).into();
}