//! Debug: print nuclear attraction integrals for a CH system in STO-3G.
//!
//! Category: debug

use sci_form::hf::basis::{build_sto3g_basis, ANG_TO_BOHR};
use sci_form::hf::nuclear::compute_nuclear_matrix;

fn main() {
    let elements = [6, 1];
    let pos_angstrom = [[0.0, 0.0, 0.0], [1.0, 0.0, 0.0]];
    let basis = build_sto3g_basis(&elements, &pos_angstrom);
    let mut pos_bohr = Vec::new();
    for p in &pos_angstrom {
        pos_bohr.push([p[0] * ANG_TO_BOHR, p[1] * ANG_TO_BOHR, p[2] * ANG_TO_BOHR]);
    }

    let v = compute_nuclear_matrix(&basis, &elements, &pos_bohr);

    println!("V[0,0]: {}", v[(0, 0)]);
    println!("V[2,2]: {}", v[(2, 2)]);
    println!("V[0,5]: {}", v[(0, 5)]);
    println!("V[2,5]: {}", v[(2, 5)]);
}
