diff options
Diffstat (limited to 'host')
| -rw-r--r-- | host/Cargo.toml | 22 | ||||
| -rw-r--r-- | host/src/bin/debug_downlink.rs | 97 | ||||
| -rw-r--r-- | host/src/lib.rs | 1 |
3 files changed, 120 insertions, 0 deletions
diff --git a/host/Cargo.toml b/host/Cargo.toml new file mode 100644 index 0000000..886962c --- /dev/null +++ b/host/Cargo.toml @@ -0,0 +1,22 @@ +[package] +name = "ocularium_host" +version = "0.1.0" +edition = "2021" + +[dependencies] +clap = { version = "4.5.18", features = ["derive"] } +tokio = { version = "1.40.0", features = ["full"] } +chrono = "0.4.38" +tokio-serial = "5.4" +postcard = "1.0" +heapless = { version = "0.8", features = ["serde"] } +eyre = "0.6" +serde = "1.0" +ssmarshal = { version = "1.0", features = ["std"] } + +embassy-executor = { git = "https://github.com/embassy-rs/embassy", features = ["arch-std"] } +embassy-time = { git = "https://github.com/embassy-rs/embassy", features = ["std", "generic-queue"] } + +molybdos_lib = { git = "https://pub.npry.dev/molybdos", features = ["postcard", "serde", "std"], default-features = false } + +hound = "3.5"
\ No newline at end of file diff --git a/host/src/bin/debug_downlink.rs b/host/src/bin/debug_downlink.rs new file mode 100644 index 0000000..a9744ac --- /dev/null +++ b/host/src/bin/debug_downlink.rs @@ -0,0 +1,97 @@ +use clap::Parser; +use tokio::{ + io::AsyncReadExt, + sync::Mutex, +}; +use tokio_serial::SerialPortBuilderExt; + +#[derive(Debug, Clone, PartialEq, PartialOrd, serde::Serialize, serde::Deserialize)] +pub enum Uplink { + Ping, +} + +#[derive(Debug, Clone, PartialEq, PartialOrd, serde::Serialize, serde::Deserialize)] +pub enum Downlink { + Pong, + PDM(heapless::Vec<u8, 128>), +} + +#[derive(clap::Parser, Debug, Clone, PartialEq, Eq, Hash, PartialOrd, Ord)] +pub struct Args { + #[clap(short, long)] + #[cfg_attr(target_os = "linux", clap(default_value = "/dev/ttyACM0"))] + pub serial_port: String, +} + +#[tokio::main] +async fn main() -> eyre::Result<()> { + let args: Args = Args::parse(); + + let mut port = tokio_serial::new(args.serial_port, 115200).open_native_async()?; + let mut cobs_acc = postcard::accumulator::CobsAccumulator::<256>::new(); + + let spec = hound::WavSpec { + channels: 1, + sample_rate: 48000 * 8, + bits_per_sample: 16, + sample_format: hound::SampleFormat::Int, + }; + + let writer = + tokio::task::spawn_blocking(move || hound::WavWriter::create("data.wav", spec)).await??; + + let writer = std::sync::Arc::new(Mutex::new(Some(writer))); + + tokio::task::spawn({ + let writer = writer.clone(); + + async move { + tokio::signal::ctrl_c().await.unwrap(); + + let mut writer = writer.lock().await; + let writer = writer.take().unwrap(); + writer.finalize().unwrap(); + } + }); + + loop { + let mut bytes = [0u8; 1024]; + + let n = port.read(&mut bytes).await?; + if n == 0 { + eyre::bail!("serial port closed"); + } + + for msg in + molybdos_lib::util::cobs::do_deserialize::<Downlink, 256>(&mut cobs_acc, &bytes[..n]) + { + match msg { + Downlink::PDM(data) => { + let writer = writer.clone(); + + tokio::task::spawn_blocking(move || { + let mut writer = writer.try_lock().unwrap(); + let writer = writer.as_mut().unwrap(); + + for b in data { + for bit in 0..8 { + let sample = (b >> bit) & 1; + writer + .write_sample(if sample == 1 { + i16::MAX + } else { + i16::MIN + }) + .unwrap(); + } + } + }) + .await?; + }, + _ => { + println!("{msg:?}"); + }, + } + } + } +} diff --git a/host/src/lib.rs b/host/src/lib.rs new file mode 100644 index 0000000..8b13789 --- /dev/null +++ b/host/src/lib.rs @@ -0,0 +1 @@ + |