use opossum::error::OpossumError;
use opossum::nodes::{BeamSplitter, Dummy};
use opossum::optic_node::OpticNode;
use opossum::OpticScenery;

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

fn main() -> Result<(), OpossumError> {
    let mut scenery = OpticScenery::new();
    scenery.set_description("Fancy Graph with Ports");

    let in1 = scenery.add_node(OpticNode::new("Input", Dummy::default()));
    let out1 = scenery.add_node(OpticNode::new("Output", Dummy::default()));
    let bs1 = scenery.add_node(OpticNode::new("Beamsplitter 1", BeamSplitter::default()));
    let bs2 = scenery.add_node(OpticNode::new("Beamsplitter 2", BeamSplitter::default()));
    let m1 = scenery.add_node(OpticNode::new("Mirror 1", Dummy::default()));
    let m2 = scenery.add_node(OpticNode::new("Mirror 2", Dummy::default()));

    scenery.connect_nodes(in1, "rear", bs1, "input1")?;
    scenery.connect_nodes(bs1, "out1_trans1_refl2", m1, "front")?;
    scenery.connect_nodes(bs1, "out2_trans2_refl1", m2, "front")?;

    scenery.connect_nodes(m1, "rear", bs2, "input1")?;
    scenery.connect_nodes(m2, "rear", bs2, "input2")?;
    scenery.connect_nodes(bs2, "out1_trans1_refl2", out1, "front")?;
    scenery.connect_nodes(bs2, "out2_trans2_refl1", out1, "front")?;

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

    Ok(())
}