example/
example.rs

1use std::{
2    collections::HashMap,
3    fs::File,
4    io::{self, BufRead, BufReader, Read, Write},
5    path::Path,
6    process::ExitCode,
7    str::FromStr,
8};
9
10use event_simulation::{CallState, PlayDirection, Simulation, SimulationState};
11use petri_net_simulation::{MultiPetriNetSimulation, PetriNetSimulationBorrowMut};
12use pns::{Net, Tid};
13
14fn main() -> ExitCode {
15    let mut save = false;
16    let mut load = false;
17
18    let args = std::env::args().skip(1);
19    for arg in args {
20        match arg.as_ref() {
21            "save" => save = true,
22            "load" => load = true,
23            _ => eprintln!("Ignore invalid argument '{arg}'"),
24        }
25    }
26
27    let Ok(net) = Net::load("examples/example.pn".as_ref()) else {
28        eprintln!("Failed to load example net");
29        return ExitCode::FAILURE;
30    };
31    let mut net = MultiPetriNetSimulation::new(net);
32
33    let mut names = HashMap::new();
34    let Ok(file) = File::open("examples/example.pnk") else {
35        eprintln!("Failed to load example keys");
36        return ExitCode::FAILURE;
37    };
38    for (tid, line) in net.transition_ids().zip(BufReader::new(file).lines()) {
39        let Ok(line) = line else {
40            eprintln!("Failed to read key");
41            return ExitCode::FAILURE;
42        };
43        names.insert(tid, line);
44    }
45
46    if save && net.save("examples/example_copy.pns".as_ref()).is_err() {
47        eprintln!("Failed to save example net");
48    }
49
50    if load {
51        fn read_values(filename: &Path) -> std::io::Result<Vec<u32>> {
52            let size = std::fs::metadata(filename)?.len().div_ceil(4);
53
54            let mut file = File::open(filename)?;
55
56            (0..size)
57                .map(|_| {
58                    let mut value = [0; 4];
59                    file.read_exact(&mut value)?;
60                    Ok(u32::from_le_bytes(value))
61                })
62                .collect()
63        }
64
65        let Ok(data) = read_values("examples/example.pns".as_ref()) else {
66            eprintln!("Reading state data failed");
67            return ExitCode::FAILURE;
68        };
69
70        if net.add_simulation_from_data(data).is_err() {
71            eprintln!("State initialization failed");
72            return ExitCode::FAILURE;
73        }
74    } else {
75        net.add_simulation();
76    }
77
78    enum StepAction {
79        Continue,
80        Reverse,
81        Save,
82        Quit,
83    }
84
85    let mut forward = true;
86    for mut simulation in &mut net {
87        fn step<D: PlayDirection>(
88            fire: CallState<PetriNetSimulationBorrowMut, D>,
89            names: &HashMap<Tid, String>,
90        ) -> StepAction {
91            if fire.callables.is_empty() {
92                return StepAction::Reverse;
93            }
94
95            println!("Choose a transition:");
96            for (i, tid) in fire.callables.iter().enumerate() {
97                println!("{}: {}", i + 1, names[tid]);
98            }
99            print!("> ");
100            let _ = io::stdout().flush();
101            let stdin = io::stdin();
102            let mut string = String::new();
103            let Ok(size) = stdin.read_line(&mut string) else {
104                eprintln!("Input error");
105                return StepAction::Continue;
106            };
107            println!();
108            if size <= 1 {
109                return StepAction::Continue;
110            }
111            match unsafe { string.chars().next().unwrap_unchecked() } {
112                'r' => return StepAction::Reverse,
113                'q' => return StepAction::Quit,
114                's' => return StepAction::Save,
115                _ => (),
116            }
117            match usize::from_str(&string[..(string.len() - 1)]) {
118                Ok(num) if num != 0 && num <= fire.callables.len() => {
119                    fire.call(num - 1);
120                }
121                _ => {
122                    println!(
123                        "You have to input a valid number from 1 to {}",
124                        fire.callables.len()
125                    );
126                    println!();
127                    println!("Other options:");
128                    println!("q: Quit");
129                    println!("r: Reverse play direction");
130                    println!("s: Save");
131                    println!();
132                    return StepAction::Continue;
133                }
134            }
135
136            StepAction::Continue
137        }
138
139        loop {
140            let step_action = if forward {
141                step(simulation.prepare_call(), &names)
142            } else {
143                step(simulation.prepare_revert(), &names)
144            };
145
146            use StepAction::*;
147            match step_action {
148                Continue => (),
149                Reverse => {
150                    println!("Reverse play direction!");
151                    println!();
152                    forward = !forward;
153                }
154                Save => {
155                    fn save(data: &[usize], filename: &Path) -> std::io::Result<()> {
156                        let mut file = File::create(filename)?;
157                        for &count in data {
158                            file.write_all(&(count as u32).to_le_bytes())?;
159                        }
160
161                        Ok(())
162                    }
163
164                    let data = simulation.data();
165
166                    let result = if save(data, "examples/example.pns".as_ref()).is_ok() {
167                        "successful"
168                    } else {
169                        "failed"
170                    };
171
172                    println!("Saving state {result}");
173                    println!();
174                }
175                Quit => break,
176            }
177        }
178    }
179
180    ExitCode::SUCCESS
181}
example/example.rs

example/
example.rs
