use nalgebra::DMatrix;

use dyna3::engine::{Q, realize_gram, sphere, PartialMatrix};

fn main() {
    let gram = {
        let mut gram_to_be = PartialMatrix::new();
        for j in 0..3 {
            for k in j..3 {
                gram_to_be.push_sym(j, k, if j == k { 1.0 } else { -1.0 });
            }
        }
        gram_to_be
    };
    let guess = {
        let a: f64 = 0.75_f64.sqrt();
        DMatrix::from_columns(&[
            sphere(1.0, 0.0, 0.0, 1.0),
            sphere(-0.5, a, 0.0, 1.0),
            sphere(-0.5, -a, 0.0, 1.0)
        ])
    };
    println!();
    let (config, success, history) = realize_gram(
        &gram, guess, &[],
        1.0e-12, 0.5, 0.9, 1.1, 200, 110
    );
    print!("\nCompleted Gram matrix:{}", config.tr_mul(&*Q) * &config);
    if success {
        println!("Target accuracy achieved!");
    } else {
        println!("Failed to reach target accuracy");
    }
    println!("Steps: {}", history.scaled_loss.len() - 1);
    println!("Loss: {}", history.scaled_loss.last().unwrap());
    println!("\nStep │ Loss\n─────┼────────────────────────────────");
    for (step, scaled_loss) in history.scaled_loss.into_iter().enumerate() {
        println!("{:<4} │ {}", step, scaled_loss);
    }
}