use std::fs::File;
use std::io::Write;

use opossum::{
    error::OpossumError,
    lightdata::{DataEnergy, LightData},
    nodes::{Detector, Dummy, NodeGroup, Source},
    optic_node::OpticNode,
    spectrum::{create_he_ne_spectrum, Spectrum},
    OpticScenery, analyzer::AnalyzerEnergy,
};
use uom::si::length::{Length, nanometer};

fn main() -> Result<(), OpossumError> {
    let mut scenery = OpticScenery::new();
    scenery.set_description("Inverse Group test".into());

    let i_s = scenery.add_element(
        "Source",
        Source::new(LightData::Energy(DataEnergy {
            spectrum: create_he_ne_spectrum(1.0),
        })),
    );

    // let i_s = scenery.add_element(
    //     "Source",
    //     Source::new(LightData::Energy(DataEnergy {
    //         spectrum: Spectrum::new(
    //             Length::new::<nanometer>(500.0)..Length::new::<nanometer>(503.0),
    //             Length::new::<nanometer>(1.0)).unwrap()}),
    //     ),
    // );

    let mut group = NodeGroup::new();
    group.expand_view(true);
    let g_n1 = group.add_node(OpticNode::new("A", Dummy::default()));
    let g_n2 = group.add_node(OpticNode::new("B", Dummy::default()));

    group.connect_nodes(g_n1, "rear", g_n2, "front")?;
    group.map_input_port(g_n1, "front", "in1")?;
    group.map_output_port(g_n2, "rear", "out1")?;
    let mut groupnode=OpticNode::new("group", group);
    groupnode.set_inverted(true);

    let i_g = scenery.add_node(groupnode);

    let i_d = scenery.add_element("Detector", Detector::default());

    scenery.connect_nodes(i_s, "out1", i_g, "out1")?;
    scenery.connect_nodes(i_g, "in1", i_d, "in1")?;

    let path = "group_reverse.dot";
    let mut output = File::create(path).unwrap();
    write!(output, "{}", scenery.to_dot()?).unwrap();

    let mut analyzer = AnalyzerEnergy::new(&scenery);
    analyzer.analyze()?;
    scenery.report();

    Ok(())
}