Add a way to simulate messages (#61)

* cangen simulation structure

* moved simulate into a separate binary

* added sim code genreation

* improved new value generation

* bug fixes

* added struct for param passing, removed format and signed

* fixed value initialization for boolean values

* added rounding for nicer numbers and boolean cases

* updated submodule pointer

* ok cargo clippy

* fixed NaN values

* limit retry attempts; refactoring and cleanup

* update embedded-base pointer

* bump submodule

---------

Co-authored-by: Jack Rubacha <[email protected]>

.gitignore

@@ -29,6 +29,8 @@ decode_data.rs
 encode_data.rs
 encode_master_mapping.rs
 format_data.rs
+simulate_data.rs
 
 # my dumb script
 ./format
+/privatetest/

Cargo.toml

@@ -3,6 +3,7 @@ name = "calypso"
 version = "0.1.0"
 edition = "2021"
 rust-version = "1.79"
+default-run = "calypso"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
@@ -13,6 +14,7 @@ protobuf-codegen = "3.5.1"
 protobuf = "3.5.1"
 bitstream-io = "2.5.3"
 clap = { version = "4.5.20", features = ["derive", "env"] }
+rand = "0.8"
 
 
 [build-dependencies]

README.md

@@ -39,6 +39,12 @@ Ex. `cansend vcan0 702#01010101FFFFFFFF`
 Now view calypso interpret the can message and broadcast it on `mqttui`
 
 
+### Simulation Mode
+- Same setup as above, then use the entry point `simulate` instead of `main`
+- ```cargo run --bin simulate```
+- ```cargo run --bin simulate  -- -u localhost:1883```
+
+
 
 ### Generate Proto
 

calypsogen.py

@@ -23,6 +23,8 @@
 encode_data = open("./src/encode_data.rs", "w")
 encode_master_mapping = open("./src/encode_master_mapping.rs", "w")
 
+simulate_data = open("./src/simulate_data.rs", "w")
+
 format_data = open("./src/format_data.rs", "w")
 
 
@@ -56,6 +58,9 @@
 encode_master_mapping.write(result.encode_master_mapping)
 encode_master_mapping.close()
 
+simulate_data.write(result.simulate_data)
+simulate_data.close()
+
 format_data.write(result.format_data)
 format_data.close()
 

src/bin/simulate.rs

@@ -0,0 +1,69 @@
+use std::{
+    thread::{self},
+    time::Duration,
+};
+
+use calypso::{
+    mqtt::MqttClient, serverdata, simulatable_message::SimulatedComponent, simulate_data::create_simulated_components
+};
+use clap::Parser;
+
+
+/// Calypso command line arguments
+#[derive(Parser, Debug)]
+#[command(version)]
+struct CalypsoArgs {
+    /// The host url of the siren, including port and excluding protocol prefix
+    #[arg(
+        short = 'u',
+        long,
+        env = "CALYPSO_SIREN_HOST_URL",
+        default_value = "localhost:1883"
+    )]
+    siren_host_url: String,
+}
+
+
+fn simulate_out(pub_path: &str) {
+    let mut client = MqttClient::new(pub_path, "calypso-simulator");
+    let _ = client.connect(); // todo: add error handling
+    let sleep_time = Duration::from_millis(10);
+
+    // todo: a way to turn individual components on and off
+    let mut simulated_components: Vec<SimulatedComponent> = create_simulated_components();
+
+    // loop through the simulated components, if they should update, update them and publish the data
+    loop {
+        for component in simulated_components.iter_mut() {
+            if component.should_update() {
+                component.update();
+                let data: calypso::data::DecodeData = component.get_data();
+                let mut payload = serverdata::ServerData::new();
+                payload.unit = data.unit.to_string();
+                payload.value = data.value.iter().map(|x| x.to_string()).collect();
+
+                client
+                    .publish(
+                        data.topic.to_string(),
+                        protobuf::Message::write_to_bytes(&payload).unwrap_or_else(|e| {
+                            format!("failed to serialize {}", e).as_bytes().to_vec()
+                        }),
+                    )
+                    .expect("Could not publish!");
+            }
+        }
+        // sleep for a bit
+        thread::sleep(sleep_time);
+    }
+}
+
+
+
+/**
+ * Main Function
+ * Calls the `simulate_out` function with the siren host URL from the command line arguments.
+ */
+fn main() {
+    let cli = CalypsoArgs::parse();
+    simulate_out(&cli.siren_host_url);
+}

src/lib.rs

@@ -9,3 +9,5 @@ pub mod encode_master_mapping;
 pub mod format_data;
 pub mod mqtt;
 pub mod serverdata;
+pub mod simulate_data;
+pub mod simulatable_message;

src/main.rs

@@ -283,4 +283,4 @@ fn main() {
 
     can_handle.join().expect("Decoder failed with ");
     println!("Decoder ended");
-}
+}