Skip to main content

embedded_nano_mesh/
embedded_nano_mesh.rs

1use std::time::Instant;
2
3use embedded_nano_mesh::{ms, ExactAddressType, Node, NodeConfig, NodeString};
4use proto_lab::{NetworkSimulator, WirelessModemFake};
5
6const NODE_1_LISTEN_PERIOD: ms = 1;
7const NODE_2_LISTEN_PERIOD: ms = 1;
8
9fn main() {
10    /* Create simulator, ether, and devices registered in that ether. */
11    let mut simulator = NetworkSimulator::new(1);
12
13    simulator.create_ether("1");
14
15    let mut driver_1 = WirelessModemFake::new("1");
16    let mut driver_2 = WirelessModemFake::new("2");
17
18    {
19        let mut ether_1 = simulator.get_ether("1").expect("Failed to find ether 1");
20        ether_1.register_driver(driver_1.clone());
21        ether_1.register_driver(driver_2.clone());
22    }
23
24    /* Create tested nodes. */
25    let mut mesh_node_1 = Node::new(NodeConfig {
26        device_address: ExactAddressType::try_from(1).expect("1 equals to 0"),
27        listen_period: NODE_1_LISTEN_PERIOD,
28    });
29
30    let mut mesh_node_2 = Node::new(NodeConfig {
31        device_address: ExactAddressType::try_from(2).expect("2 equals to 0"),
32        listen_period: NODE_2_LISTEN_PERIOD,
33    });
34
35    /* Do testing scenario */
36    let _ = mesh_node_1.send_to_exact(
37        NodeString::try_from("Message from node 1")
38            .expect("Fail to pack message")
39            .into_bytes(),
40        ExactAddressType::try_from(2).expect("2 is 0"),
41        2.into(),
42        false,
43    );
44
45    simulator.start_simulation_thread();
46
47    let start_time = Instant::now();
48    loop {
49        let current_time = Instant::now().duration_since(start_time).as_millis() as ms;
50
51        let _ = mesh_node_1.update(&mut driver_1, current_time);
52        let _ = mesh_node_2.update(&mut driver_2, current_time);
53
54        if current_time >= 200 as ms {
55            panic!("Simulation timeout");
56        }
57
58        if let Some(_packet) = mesh_node_2.receive() {
59            break;
60        }
61    }
62
63    simulator.stop_simulation_thread();
64
65    println!("Simulation done");
66}