use opossum::error::OpossumError;
use opossum::nodes::{Dummy, NodeGroup, BeamSplitter};
use opossum::OpticScenery;
use std::fs::File;
use std::io::Write;
use opossum::optic_node::OpticNode;

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

    let mut group1 = NodeGroup::new();
    group1.expand_view(true);
    let g1_n1 = group1.add_node(OpticNode::new("TFP1_g1", Dummy::default()));
    let g1_n2 = group1.add_node(OpticNode::new("TFP2_g1", BeamSplitter::default()));
    group1.map_output_port(g1_n2, "out1_trans1_refl2", "out1")?;
    group1.connect_nodes(g1_n1, "rear", g1_n2, "input1")?;

    let mut nested_group = NodeGroup::new(); 
    let nested_g_n1 = nested_group.add_node(OpticNode::new("TFP1_g_nested_1", Dummy::default()));
    let nested_g_n2 = nested_group.add_node(OpticNode::new("TFP1_g_nested_2", Dummy::default()));
    nested_group.expand_view(false);

    nested_group.connect_nodes(nested_g_n1, "rear", nested_g_n2, "front")?;
    nested_group.map_input_port(nested_g_n1, "front", "in1")?;
    nested_group.map_output_port(nested_g_n2, "rear", "out1")?;
    
    let nested_group_index = group1.add_node(OpticNode::new("nested_group", nested_group));
    group1.connect_nodes(nested_group_index, "out1", g1_n1, "front")?;

    
    let mut group2: NodeGroup = NodeGroup::new();
    group2.expand_view(true);
    let g2_n1 = group2.add_node(OpticNode::new("TFP1_g2", Dummy::default()));
    let g2_n2 = group2.add_node(OpticNode::new("TFP2_g2", Dummy::default()));
    group2.map_input_port(g2_n1, "front", "in1")?;

    group2.connect_nodes(g2_n1, "rear", g2_n2, "front")?;


    let scene_g1 = scenery.add_node(OpticNode::new("Group1_TFPs", group1));
    let scene_g2 = scenery.add_node(OpticNode::new("Group2_TFPs", group2));
    // set_output_port
    scenery.connect_nodes(scene_g1, "out1", scene_g2, "in1")?;
    let path = "graph_group.dot";
    let mut output = File::create(path).unwrap();
    write!(output, "{}", scenery.to_dot("LR")?).unwrap();

    Ok(())
}