Skip to content
Snippets Groups Projects
group_test.rs 1.84 KiB
Newer Older
Udo Eisenbarth's avatar
Udo Eisenbarth committed
use opossum::nodes::{BeamSplitter, Dummy, NodeGroup};
y.zobus's avatar
y.zobus committed
use opossum::OpticScenery;
use std::fs::File;
use std::io::Write;
fn main() -> Result<(), OpossumError> {
    let mut scenery = OpticScenery::new();
    scenery.set_description("Node Group test section".into());

Udo Eisenbarth's avatar
Udo Eisenbarth committed
    let mut group1 = NodeGroup::default();
    let g1_n1 = group1.add_node(Dummy::new("node1"));
    let g1_n2 = group1.add_node(BeamSplitter::default());
y.zobus's avatar
y.zobus committed
    group1.map_output_port(g1_n2, "out1_trans1_refl2", "out1")?;
    group1.connect_nodes(g1_n1, "rear", g1_n2, "input1")?;
Udo Eisenbarth's avatar
Udo Eisenbarth committed
    let mut nested_group = NodeGroup::default();
    let nested_g_n1 = nested_group.add_node(Dummy::new("node1_1"));
    let nested_g_n2 = nested_group.add_node(Dummy::new("node1_2"));
y.zobus's avatar
y.zobus committed
    nested_group.expand_view(true);
y.zobus's avatar
y.zobus committed
    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")?;
Udo Eisenbarth's avatar
Udo Eisenbarth committed

    let nested_group_index = group1.add_node(nested_group);
y.zobus's avatar
y.zobus committed
    group1.connect_nodes(nested_group_index, "out1", g1_n1, "front")?;
Udo Eisenbarth's avatar
Udo Eisenbarth committed
    let mut group2: NodeGroup = NodeGroup::default();
    let g2_n1 = group2.add_node(Dummy::new("node2_1"));
    let g2_n2 = group2.add_node(Dummy::new("node2_2"));
y.zobus's avatar
y.zobus committed
    group2.map_input_port(g2_n1, "front", "in1")?;
y.zobus's avatar
y.zobus committed

y.zobus's avatar
y.zobus committed
    group2.connect_nodes(g2_n1, "rear", g2_n2, "front")?;
    let scene_g1 = scenery.add_node(group1);
    let scene_g2 = scenery.add_node(group2);
y.zobus's avatar
y.zobus committed
    scenery.connect_nodes(scene_g1, "out1", scene_g2, "in1")?;
    println!("{}", serde_yaml::to_string(&scenery).unwrap());
    let path = "graph_group.dot";
    let mut output = File::create(path).unwrap();
    write!(output, "{}", scenery.to_dot("LR")?).unwrap();