Added logger to GUI.

This commit is contained in:
2025-10-15 18:11:10 +02:00
parent cb7bc2f025
commit 7229d4cd33
10 changed files with 269 additions and 94 deletions

1
.gitignore vendored
View File

@@ -1,2 +1,3 @@
/target /target
.DS_Store .DS_Store
log_*

69
Cargo.lock generated
View File

@@ -172,6 +172,15 @@ version = "0.2.2"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "fc7eb209b1518d6bb87b283c20095f5228ecda460da70b44f0802523dea6da04" checksum = "fc7eb209b1518d6bb87b283c20095f5228ecda460da70b44f0802523dea6da04"
[[package]]
name = "android_system_properties"
version = "0.1.5"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "819e7219dbd41043ac279b19830f2efc897156490d7fd6ea916720117ee66311"
dependencies = [
"libc",
]
[[package]] [[package]]
name = "arboard" name = "arboard"
version = "3.6.0" version = "3.6.0"
@@ -439,6 +448,7 @@ version = "0.1.0"
dependencies = [ dependencies = [
"atomic_float", "atomic_float",
"bioz-icd-rs", "bioz-icd-rs",
"chrono",
"defmt", "defmt",
"eframe", "eframe",
"egui_dock", "egui_dock",
@@ -622,6 +632,19 @@ dependencies = [
"libc", "libc",
] ]
[[package]]
name = "chrono"
version = "0.4.42"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "145052bdd345b87320e369255277e3fb5152762ad123a901ef5c262dd38fe8d2"
dependencies = [
"iana-time-zone",
"js-sys",
"num-traits",
"wasm-bindgen",
"windows-link 0.2.1",
]
[[package]] [[package]]
name = "clipboard-win" name = "clipboard-win"
version = "5.4.1" version = "5.4.1"
@@ -1587,6 +1610,30 @@ version = "0.2.1"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "dfa686283ad6dd069f105e5ab091b04c62850d3e4cf5d67debad1933f55023df" checksum = "dfa686283ad6dd069f105e5ab091b04c62850d3e4cf5d67debad1933f55023df"
[[package]]
name = "iana-time-zone"
version = "0.1.64"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "33e57f83510bb73707521ebaffa789ec8caf86f9657cad665b092b581d40e9fb"
dependencies = [
"android_system_properties",
"core-foundation-sys",
"iana-time-zone-haiku",
"js-sys",
"log",
"wasm-bindgen",
"windows-core",
]
[[package]]
name = "iana-time-zone-haiku"
version = "0.1.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "f31827a206f56af32e590ba56d5d2d085f558508192593743f16b2306495269f"
dependencies = [
"cc",
]
[[package]] [[package]]
name = "icu_collections" name = "icu_collections"
version = "2.0.0" version = "2.0.0"
@@ -4035,7 +4082,7 @@ dependencies = [
"windows-collections", "windows-collections",
"windows-core", "windows-core",
"windows-future", "windows-future",
"windows-link", "windows-link 0.1.3",
"windows-numerics", "windows-numerics",
] ]
@@ -4056,7 +4103,7 @@ checksum = "c0fdd3ddb90610c7638aa2b3a3ab2904fb9e5cdbecc643ddb3647212781c4ae3"
dependencies = [ dependencies = [
"windows-implement", "windows-implement",
"windows-interface", "windows-interface",
"windows-link", "windows-link 0.1.3",
"windows-result", "windows-result",
"windows-strings", "windows-strings",
] ]
@@ -4068,7 +4115,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "fc6a41e98427b19fe4b73c550f060b59fa592d7d686537eebf9385621bfbad8e" checksum = "fc6a41e98427b19fe4b73c550f060b59fa592d7d686537eebf9385621bfbad8e"
dependencies = [ dependencies = [
"windows-core", "windows-core",
"windows-link", "windows-link 0.1.3",
"windows-threading", "windows-threading",
] ]
@@ -4100,6 +4147,12 @@ version = "0.1.3"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "5e6ad25900d524eaabdbbb96d20b4311e1e7ae1699af4fb28c17ae66c80d798a" checksum = "5e6ad25900d524eaabdbbb96d20b4311e1e7ae1699af4fb28c17ae66c80d798a"
[[package]]
name = "windows-link"
version = "0.2.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "f0805222e57f7521d6a62e36fa9163bc891acd422f971defe97d64e70d0a4fe5"
[[package]] [[package]]
name = "windows-numerics" name = "windows-numerics"
version = "0.2.0" version = "0.2.0"
@@ -4107,7 +4160,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "9150af68066c4c5c07ddc0ce30421554771e528bde427614c61038bc2c92c2b1" checksum = "9150af68066c4c5c07ddc0ce30421554771e528bde427614c61038bc2c92c2b1"
dependencies = [ dependencies = [
"windows-core", "windows-core",
"windows-link", "windows-link 0.1.3",
] ]
[[package]] [[package]]
@@ -4116,7 +4169,7 @@ version = "0.3.4"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "56f42bd332cc6c8eac5af113fc0c1fd6a8fd2aa08a0119358686e5160d0586c6" checksum = "56f42bd332cc6c8eac5af113fc0c1fd6a8fd2aa08a0119358686e5160d0586c6"
dependencies = [ dependencies = [
"windows-link", "windows-link 0.1.3",
] ]
[[package]] [[package]]
@@ -4125,7 +4178,7 @@ version = "0.4.2"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "56e6c93f3a0c3b36176cb1327a4958a0353d5d166c2a35cb268ace15e91d3b57" checksum = "56e6c93f3a0c3b36176cb1327a4958a0353d5d166c2a35cb268ace15e91d3b57"
dependencies = [ dependencies = [
"windows-link", "windows-link 0.1.3",
] ]
[[package]] [[package]]
@@ -4225,7 +4278,7 @@ version = "0.53.3"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "d5fe6031c4041849d7c496a8ded650796e7b6ecc19df1a431c1a363342e5dc91" checksum = "d5fe6031c4041849d7c496a8ded650796e7b6ecc19df1a431c1a363342e5dc91"
dependencies = [ dependencies = [
"windows-link", "windows-link 0.1.3",
"windows_aarch64_gnullvm 0.53.0", "windows_aarch64_gnullvm 0.53.0",
"windows_aarch64_msvc 0.53.0", "windows_aarch64_msvc 0.53.0",
"windows_i686_gnu 0.53.0", "windows_i686_gnu 0.53.0",
@@ -4242,7 +4295,7 @@ version = "0.1.0"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "b66463ad2e0ea3bbf808b7f1d371311c80e115c0b71d60efc142cafbcfb057a6" checksum = "b66463ad2e0ea3bbf808b7f1d371311c80e115c0b71d60efc142cafbcfb057a6"
dependencies = [ dependencies = [
"windows-link", "windows-link 0.1.3",
] ]
[[package]] [[package]]

View File

@@ -12,6 +12,7 @@ egui_extras = "0.32.3"
log = "0.4.27" log = "0.4.27"
simple_logger = "5.0.0" simple_logger = "5.0.0"
atomic_float = "1.1.0" atomic_float = "1.1.0"
chrono = { version = "0.4.42" }
[dependencies.bioz-icd-rs] [dependencies.bioz-icd-rs]
path = "../bioz-icd-rs" path = "../bioz-icd-rs"
@@ -35,6 +36,7 @@ features = [
"rt-multi-thread", "rt-multi-thread",
"macros", "macros",
"time", "time",
"fs",
] ]
[dependencies.tokio-serial] [dependencies.tokio-serial]

View File

@@ -1,21 +1,25 @@
use log::info; use log::{info, error};
use core::f32;
use std::f32::consts::PI; use std::f32::consts::PI;
use std::ops::RangeInclusive; use std::ops::RangeInclusive;
use std::sync::{Arc, Mutex}; use std::sync::{Arc, Mutex};
use std::sync::atomic::{AtomicBool, Ordering}; use std::sync::atomic::{AtomicBool, Ordering};
use chrono::Local;
use atomic_float::AtomicF32; use atomic_float::AtomicF32;
use tokio::{sync::mpsc::{Sender}}; use tokio::{sync::mpsc::{Sender}};
use eframe::egui::{self, Button, CollapsingHeader, Color32, ComboBox, DragValue, Key, Label, Layout, Modifiers}; use eframe::egui::{self, Button, CollapsingHeader, Color32, ComboBox, DragValue, Id, Key, Label, Layout, Modifiers, TextEdit, Widget};
use egui_plot::{Corner, GridInput, GridMark, Legend, Line, Plot, PlotPoint, Points}; use egui_plot::{Corner, GridInput, GridMark, Legend, Line, Plot, PlotPoint, Points};
use egui_dock::{DockArea, DockState, Style}; use egui_dock::{DockArea, DockState, Style};
use egui_extras::{TableBuilder, Column}; use egui_extras::{TableBuilder, Column};
use crate::plot::{TimeSeriesPlot, BodePlot}; use crate::plot::{TimeSeriesPlot, BodePlot};
use crate::signals::StartStopSignal; use crate::signals::{LoggingSignal, StartStopSignal};
use crate::icd::{IcdDftNum, MeasurementPointSet}; use crate::icd::{IcdDftNum, MeasurementPointSet};
@@ -42,6 +46,7 @@ pub struct App {
tree: DockState<String>, tree: DockState<String>,
tab_viewer: TabViewer, tab_viewer: TabViewer,
run_impedancemeter_tx: Sender<StartStopSignal>, run_impedancemeter_tx: Sender<StartStopSignal>,
log_tx: Sender<LoggingSignal>,
pub magnitude: Arc<Mutex<f32>>, pub magnitude: Arc<Mutex<f32>>,
pub phase: Arc<Mutex<f32>>, pub phase: Arc<Mutex<f32>>,
pub magnitude_series: Arc<Mutex<TimeSeriesPlot>>, pub magnitude_series: Arc<Mutex<TimeSeriesPlot>>,
@@ -55,8 +60,9 @@ pub struct App {
pub dft_num: Arc<Mutex<IcdDftNum>>, pub dft_num: Arc<Mutex<IcdDftNum>>,
pub measurement_points: Arc<Mutex<MeasurementPointSet>>, pub measurement_points: Arc<Mutex<MeasurementPointSet>>,
pub periods_per_dft: Arc<Mutex<Option<f32>>>, pub periods_per_dft: Arc<Mutex<Option<f32>>>,
pub periods_per_dft_multi: Arc<Mutex<(Vec<f32>, Option<Vec<f32>>)>>, pub periods_per_dft_multi: Arc<Mutex<(Vec<u32>, Option<Vec<f32>>)>>,
pub gui_logging_enabled: Arc<AtomicBool>, pub gui_logging_enabled: Arc<AtomicBool>,
log_filename: String,
} }
struct TabViewer { struct TabViewer {
@@ -70,7 +76,7 @@ struct TabViewer {
dft_num: Arc<Mutex<IcdDftNum>>, dft_num: Arc<Mutex<IcdDftNum>>,
measurement_points: Arc<Mutex<MeasurementPointSet>>, measurement_points: Arc<Mutex<MeasurementPointSet>>,
periods_per_dft: Arc<Mutex<Option<f32>>>, periods_per_dft: Arc<Mutex<Option<f32>>>,
periods_per_dft_multi: Arc<Mutex<(Vec<f32>, Option<Vec<f32>>)>>, periods_per_dft_multi: Arc<Mutex<(Vec<u32>, Option<Vec<f32>>)>>,
show_settings: bool, show_settings: bool,
show_settings_toggle: Option<bool>, show_settings_toggle: Option<bool>,
} }
@@ -200,17 +206,17 @@ impl TabViewer {
ui.add_enabled_ui(*on, |ui| { ui.add_enabled_ui(*on, |ui| {
let (freq, periods_per_dft_vec) = self.periods_per_dft_multi.lock().unwrap().clone(); let (freq, periods_per_dft_vec) = self.periods_per_dft_multi.lock().unwrap().clone();
fn format_frequency(freq: f32) -> String { fn format_frequency(freq: u32) -> String {
if freq >= 1_000.0 { if freq >= 1_000 {
// kHz // kHz
if freq % 1_000.0 == 0.0 { if freq % 1_000 == 0 {
format!("{}k", (freq / 1_000.0) as u64) format!("{}k", (freq / 1_000) as u64)
} else { } else {
format!("{:.1}k", freq / 1_000.0) format!("{:.1}k", freq / 1_000)
} }
} else { } else {
// Hz // Hz
if freq % 1.0 == 0.0 { if freq % 1 == 0 {
format!("{}", freq as u64) format!("{}", freq as u64)
} else { } else {
format!("{:.1}", freq) format!("{:.1}", freq)
@@ -413,7 +419,7 @@ impl egui_dock::TabViewer for TabViewer {
} }
impl App { impl App {
pub fn new(run_impedancemeter_tx: Sender<StartStopSignal>) -> Self { pub fn new(run_impedancemeter_tx: Sender<StartStopSignal>, log_tx: Sender<LoggingSignal>) -> Self {
// Step 1: Initialize shared fields first // Step 1: Initialize shared fields first
let magnitude = Arc::new(Mutex::new(0.0)); let magnitude = Arc::new(Mutex::new(0.0));
let phase = Arc::new(Mutex::new(0.0)); let phase = Arc::new(Mutex::new(0.0));
@@ -450,6 +456,7 @@ impl App {
tree: DockState::new(vec!["Single".to_string(), "Multi".to_string(), "Shortcuts".to_string()]), tree: DockState::new(vec!["Single".to_string(), "Multi".to_string(), "Shortcuts".to_string()]),
tab_viewer, tab_viewer,
run_impedancemeter_tx, run_impedancemeter_tx,
log_tx,
magnitude, magnitude,
phase, phase,
magnitude_series, magnitude_series,
@@ -465,6 +472,7 @@ impl App {
periods_per_dft, periods_per_dft,
periods_per_dft_multi, periods_per_dft_multi,
gui_logging_enabled: Arc::new(AtomicBool::new(false)), gui_logging_enabled: Arc::new(AtomicBool::new(false)),
log_filename: format!("log_{}.csv", Local::now().format("%Y%m%d"))
}; };
// For testing purposes, populate the Bode plot with a sample low-pass filter response // For testing purposes, populate the Bode plot with a sample low-pass filter response
@@ -493,17 +501,17 @@ impl App {
match (self.tab_active, *self.on.lock().unwrap()) { match (self.tab_active, *self.on.lock().unwrap()) {
(TabActive::Single, true) => { (TabActive::Single, true) => {
if let Err(e) = self.run_impedancemeter_tx.try_send(StartStopSignal::StartSingle(*self.single_frequency.lock().unwrap(), *self.dft_num.lock().unwrap())) { if let Err(e) = self.run_impedancemeter_tx.try_send(StartStopSignal::StartSingle(*self.single_frequency.lock().unwrap(), *self.dft_num.lock().unwrap())) {
eprintln!("Failed to send start command: {:?}", e); error!("Failed to send start command: {:?}", e);
} }
}, },
(TabActive::Multi, true) => { (TabActive::Multi, true) => {
if let Err(e) = self.run_impedancemeter_tx.try_send(StartStopSignal::StartMulti(*self.measurement_points.lock().unwrap())) { if let Err(e) = self.run_impedancemeter_tx.try_send(StartStopSignal::StartMulti(*self.measurement_points.lock().unwrap())) {
eprintln!("Failed to send start command: {:?}", e); error!("Failed to send start command: {:?}", e);
} }
}, },
(_, false) => { (_, false) => {
if let Err(e) = self.run_impedancemeter_tx.try_send(StartStopSignal::Stop) { if let Err(e) = self.run_impedancemeter_tx.try_send(StartStopSignal::Stop) {
eprintln!("Failed to send stop command: {:?}", e); error!("Failed to send stop command: {:?}", e);
} }
}, },
(_, _) => {} (_, _) => {}
@@ -545,8 +553,24 @@ impl eframe::App for App {
let mut gui_logging_enabled = self.gui_logging_enabled.load(Ordering::Relaxed); let mut gui_logging_enabled = self.gui_logging_enabled.load(Ordering::Relaxed);
if ui.add(egui::Checkbox::new(&mut gui_logging_enabled, "Logging")).changed() { if ui.add(egui::Checkbox::new(&mut gui_logging_enabled, "Logging")).changed() {
self.gui_logging_enabled.store(gui_logging_enabled, Ordering::Relaxed); self.gui_logging_enabled.store(gui_logging_enabled, Ordering::Relaxed);
if gui_logging_enabled {
if let Err(e) = self.log_tx.try_send(LoggingSignal::StartFileLogging(self.log_filename.clone())) {
error!("Failed to send logging signal: {:?}", e);
}
} else {
if let Err(e) = self.log_tx.try_send(LoggingSignal::StopFileLogging) {
error!("Failed to send logging signal: {:?}", e);
}
}
} }
ui.separator();
ui.add_enabled_ui(!gui_logging_enabled, |ui| {
ui.label("Log filename:");
TextEdit::singleline(&mut self.log_filename).desired_width(125.0).id(Id::new("file_name_field")).ui(ui);
});
// Spacer to push the LED to the right // Spacer to push the LED to the right
ui.with_layout(egui::Layout::right_to_left(egui::Align::Center), |ui| { ui.with_layout(egui::Layout::right_to_left(egui::Align::Center), |ui| {
ui.scope(|ui| { ui.scope(|ui| {
@@ -620,39 +644,54 @@ impl eframe::App for App {
ctx.send_viewport_cmd(egui::ViewportCommand::Close); ctx.send_viewport_cmd(egui::ViewportCommand::Close);
} }
// Space to start/stop measurement // Check if the file name field is focused
if ctx.input(|i| i.key_pressed(Key::Space)) // If it is, we don't want to trigger shortcuts
{ let file_name_field_is_focused = ctx.memory(|memory| memory.has_focus(Id::new("file_name_field")));
let value = *self.on.lock().unwrap(); if !file_name_field_is_focused {
*self.on.lock().unwrap() = !value;
self.update_start_stop();
}
// Send stop command when the window is closed // Space to start/stop measurement
if ctx.input(|i| i.viewport().close_requested()) { if ctx.input(|i| i.key_pressed(Key::Space))
*self.on.lock().unwrap() = false; {
self.update_start_stop(); let value = *self.on.lock().unwrap();
} *self.on.lock().unwrap() = !value;
self.update_start_stop();
}
// Reset view // Send stop command when the window is closed
if ctx.input(|i| i.key_pressed(Key::R)) { if ctx.input(|i| i.viewport().close_requested()) {
self.reset_view(); *self.on.lock().unwrap() = false;
} self.update_start_stop();
}
// Enable/disable GUI logging // Reset view
if ctx.input(|i| i.key_pressed(egui::Key::L)) { if ctx.input(|i| i.key_pressed(Key::R)) {
let mut gui_logging_enabled = self.gui_logging_enabled.load(Ordering::Relaxed); self.reset_view();
gui_logging_enabled = !gui_logging_enabled; }
self.gui_logging_enabled.store(gui_logging_enabled, Ordering::Relaxed);
}
// Toggle setttings view // Enable/disable GUI logging
if ctx.input(|i| i.key_pressed(egui::Key::S)) { if ctx.input(|i| i.key_pressed(egui::Key::L)) {
self.tab_viewer.show_settings = !self.tab_viewer.show_settings; let mut gui_logging_enabled = self.gui_logging_enabled.load(Ordering::Relaxed);
if self.tab_viewer.show_settings { gui_logging_enabled = !gui_logging_enabled;
self.tab_viewer.show_settings_toggle = Some(true); self.gui_logging_enabled.store(gui_logging_enabled, Ordering::Relaxed);
} else { if gui_logging_enabled {
self.tab_viewer.show_settings_toggle = Some(false); if let Err(e) = self.log_tx.try_send(LoggingSignal::StartFileLogging(self.log_filename.clone())) {
error!("Failed to send logging signal: {:?}", e);
}
} else {
if let Err(e) = self.log_tx.try_send(LoggingSignal::StopFileLogging) {
error!("Failed to send logging signal: {:?}", e);
}
}
}
// Toggle setttings view
if ctx.input(|i| i.key_pressed(egui::Key::S)) {
self.tab_viewer.show_settings = !self.tab_viewer.show_settings;
if self.tab_viewer.show_settings {
self.tab_viewer.show_settings_toggle = Some(true);
} else {
self.tab_viewer.show_settings_toggle = Some(false);
}
} }
} }

View File

@@ -9,6 +9,7 @@ use bioz_host_rs::communication::communicate_with_hardware;
use tokio::sync::mpsc::{self}; use tokio::sync::mpsc::{self};
use bioz_host_rs::signals::StartStopSignal; use bioz_host_rs::signals::StartStopSignal;
use bioz_host_rs::logging::log_data;
#[tokio::main] #[tokio::main]
async fn main() { async fn main() {
@@ -26,9 +27,10 @@ async fn main() {
let run_impedancemeter_tx_clone = run_impedancemeter_tx.clone(); let run_impedancemeter_tx_clone = run_impedancemeter_tx.clone();
// Logging // Logging
let (log_tx, mut log_rx) = mpsc::channel::<LoggingSignal>(10); let (log_tx, log_rx) = mpsc::channel::<LoggingSignal>(10);
let log_tx_clone = log_tx.clone();
let app = App::new(run_impedancemeter_tx); let app = App::new(run_impedancemeter_tx, log_tx);
let magnitude_clone = app.magnitude.clone(); let magnitude_clone = app.magnitude.clone();
let phase_clone = app.phase.clone(); let phase_clone = app.phase.clone();
let magnitude_series_clone = app.magnitude_series.clone(); let magnitude_series_clone = app.magnitude_series.clone();
@@ -44,26 +46,7 @@ async fn main() {
let gui_logging_enabled = app.gui_logging_enabled.clone(); let gui_logging_enabled = app.gui_logging_enabled.clone();
// Log thread // Log thread
tokio::spawn(async move { tokio::spawn(async move { log_data(log_rx).await });
loop {
match log_rx.recv().await {
Some(signal) => {
match signal {
LoggingSignal::SingleImpedance(magnitude, phase) => {
info!("Single Impedance - Magnitude: {:.3}, Phase: {:.3}", magnitude, phase);
}
LoggingSignal::MultiImpedance(magnitudes, phases) => {
info!("Multi Impedance - Magnitudes: {:?}, Phases: {:?}", magnitudes, phases);
}
}
}
None => {
// Channel closed
break;
}
}
}
});
// Execute the runtime in its own thread. // Execute the runtime in its own thread.
std::thread::spawn(move || { std::thread::spawn(move || {
@@ -80,7 +63,7 @@ async fn main() {
periods_per_dft, periods_per_dft,
periods_per_dft_multi, periods_per_dft_multi,
gui_logging_enabled, gui_logging_enabled,
log_tx, log_tx_clone,
)); ));
}); });

View File

@@ -1,7 +1,9 @@
use std::time::SystemTime;
use log::{error, info}; use log::{error, info};
use tokio::select; use tokio::select;
use tokio::sync::mpsc::{Sender, Receiver}; use tokio::sync::mpsc::{Receiver, Sender};
use std::sync::atomic::{AtomicBool, AtomicU32, Ordering}; use std::sync::atomic::{AtomicBool, AtomicU32, Ordering};
use atomic_float::AtomicF32; use atomic_float::AtomicF32;
@@ -28,7 +30,7 @@ pub async fn communicate_with_hardware(
connected: Arc<AtomicBool>, connected: Arc<AtomicBool>,
data_frequency: Arc<AtomicF32>, data_frequency: Arc<AtomicF32>,
periods_per_dft: Arc<Mutex<Option<f32>>>, periods_per_dft: Arc<Mutex<Option<f32>>>,
periods_per_dft_multi: Arc<Mutex<(Vec<f32>, Option<Vec<f32>>)>>, periods_per_dft_multi: Arc<Mutex<(Vec<u32>, Option<Vec<f32>>)>>,
gui_logging_enabled: Arc<AtomicBool>, gui_logging_enabled: Arc<AtomicBool>,
log_tx: Sender<LoggingSignal>, log_tx: Sender<LoggingSignal>,
) { ) {
@@ -42,17 +44,17 @@ pub async fn communicate_with_hardware(
} }
}); });
#[derive(Default)] #[derive(Default, Clone, Copy)]
struct Settings { struct Settings {
frequency: Option<StartStopSignal>, frequency: Option<StartStopSignal>,
} }
let mut settings = Settings::default(); let settings = Arc::new(Mutex::new(Settings::default()));
loop { loop {
let workbook_client = match WorkbookClient::new() { let workbook_client = match WorkbookClient::new() {
Ok(client) => { Ok(client) => {
info!("Connected to hardware successfully."); info!("Connected to hardware successfully.");
if let Some(frequency) = settings.frequency { if let Some(frequency) = settings.lock().unwrap().frequency {
run_impedancemeter_tx.send(frequency).await.unwrap(); run_impedancemeter_tx.send(frequency).await.unwrap();
} }
connected.store(true, Ordering::Relaxed); connected.store(true, Ordering::Relaxed);
@@ -79,6 +81,8 @@ pub async fn communicate_with_hardware(
let gui_logging_enabled_clone = gui_logging_enabled.clone(); let gui_logging_enabled_clone = gui_logging_enabled.clone();
let log_tx_clone = log_tx.clone(); let log_tx_clone = log_tx.clone();
let settings_clone = settings.clone();
tokio::spawn(async move { tokio::spawn(async move {
while let Ok(val) = single_impedance_sub.recv().await { while let Ok(val) = single_impedance_sub.recv().await {
{ {
@@ -97,9 +101,20 @@ pub async fn communicate_with_hardware(
} }
// Send logging signal // Send logging signal
if gui_logging_enabled_clone.load(Ordering::Relaxed) { if gui_logging_enabled_clone.load(Ordering::Relaxed) {
if let Err(e) = log_tx_clone.try_send(LoggingSignal::SingleImpedance(val.magnitude, val.phase)) { let settings = *settings_clone.lock().unwrap();
error!("Failed to send logging signal: {:?}", e); match settings.frequency {
Some(StartStopSignal::StartSingle(freq, _)) => {
if let Err(e) = log_tx_clone.send(LoggingSignal::SingleImpedance(SystemTime::now(), freq, val.magnitude, val.phase)).await {
error!("Failed to send logging signal: {:?}", e);
}
},
_ => {
error!("Frequency not set for single impedance logging.");
},
} }
// if let Err(e) = log_tx_clone.send(LoggingSignal::SingleImpedance(SystemTime::now(), settings.frequency, val.magnitude, val.phase)).await {
// error!("Failed to send logging signal: {:?}", e);
// }
} }
} }
info!("SingleImpedanceOutputTopic subscription ended."); info!("SingleImpedanceOutputTopic subscription ended.");
@@ -133,7 +148,7 @@ pub async fn communicate_with_hardware(
bode_plot.update_phases(MeasurementPointSet::Eight, phases.clone()); bode_plot.update_phases(MeasurementPointSet::Eight, phases.clone());
} }
if gui_logging_enabled_clone.load(Ordering::Relaxed) { if gui_logging_enabled_clone.load(Ordering::Relaxed) {
if let Err(e) = log_tx_clone.try_send(LoggingSignal::MultiImpedance(magnitudes.clone(), phases.clone())) { if let Err(e) = log_tx_clone.send(LoggingSignal::MultiImpedance(SystemTime::now(), MeasurementPointSet::Eight.values().to_vec(), magnitudes.clone(), phases.clone())).await {
error!("Failed to send logging signal: {:?}", e); error!("Failed to send logging signal: {:?}", e);
} }
} }
@@ -147,7 +162,7 @@ pub async fn communicate_with_hardware(
bode_plot.update_phases(MeasurementPointSet::Eighteen, phases.clone()); bode_plot.update_phases(MeasurementPointSet::Eighteen, phases.clone());
} }
if gui_logging_enabled_clone.load(Ordering::Relaxed) { if gui_logging_enabled_clone.load(Ordering::Relaxed) {
if let Err(e) = log_tx_clone.try_send(LoggingSignal::MultiImpedance(magnitudes.clone(), phases.clone())) { if let Err(e) = log_tx_clone.send(LoggingSignal::MultiImpedance(SystemTime::now(), MeasurementPointSet::Eighteen.values().to_vec(), magnitudes.clone(), phases.clone())).await {
error!("Failed to send logging signal: {:?}", e); error!("Failed to send logging signal: {:?}", e);
} }
} }
@@ -166,7 +181,7 @@ pub async fn communicate_with_hardware(
match workbook_client.start_impedancemeter_single(freq, dft_num).await { match workbook_client.start_impedancemeter_single(freq, dft_num).await {
Ok(Ok(periods)) => { Ok(Ok(periods)) => {
info!("Impedance meter started at frequency: {} with periods per DFT: {}", freq, periods); info!("Impedance meter started at frequency: {} with periods per DFT: {}", freq, periods);
settings.frequency = Some(StartStopSignal::StartSingle(freq, dft_num)); settings.lock().unwrap().frequency = Some(StartStopSignal::StartSingle(freq, dft_num));
*periods_per_dft.lock().unwrap() = Some(periods); *periods_per_dft.lock().unwrap() = Some(periods);
}, },
Ok(Err(e)) => { Ok(Err(e)) => {
@@ -182,7 +197,7 @@ pub async fn communicate_with_hardware(
StartStopSignal::StartMulti(num_points) => { StartStopSignal::StartMulti(num_points) => {
match workbook_client.start_impedancemeter_multi(num_points).await { match workbook_client.start_impedancemeter_multi(num_points).await {
Ok(Ok(periods)) => { Ok(Ok(periods)) => {
settings.frequency = Some(StartStopSignal::StartMulti(num_points)); settings.lock().unwrap().frequency = Some(StartStopSignal::StartMulti(num_points));
info!("Multi-point Impedancemeter started."); info!("Multi-point Impedancemeter started.");
match num_points { match num_points {
MeasurementPointSet::Eight => { MeasurementPointSet::Eight => {
@@ -207,7 +222,7 @@ pub async fn communicate_with_hardware(
if let Err(e) = workbook_client.stop_impedancemeter().await { if let Err(e) = workbook_client.stop_impedancemeter().await {
error!("Failed to stop impedancemeter: {:?}", e); error!("Failed to stop impedancemeter: {:?}", e);
} else { } else {
settings.frequency = Some(StartStopSignal::Stop); settings.lock().unwrap().frequency = Some(StartStopSignal::Stop);
*periods_per_dft.lock().unwrap() = None; *periods_per_dft.lock().unwrap() = None;
let (freq, _) = periods_per_dft_multi.lock().unwrap().clone(); let (freq, _) = periods_per_dft_multi.lock().unwrap().clone();
*periods_per_dft_multi.lock().unwrap() = (freq, None); *periods_per_dft_multi.lock().unwrap() = (freq, None);

View File

@@ -5,6 +5,7 @@ pub mod app;
pub mod communication; pub mod communication;
pub mod plot; pub mod plot;
pub mod signals; pub mod signals;
pub mod logging;
pub use bioz_icd_rs as icd; pub use bioz_icd_rs as icd;
pub async fn read_line() -> String { pub async fn read_line() -> String {

77
src/logging.rs Normal file
View File

@@ -0,0 +1,77 @@
use log::info;
use tokio::sync::mpsc::Receiver;
use tokio::fs::File;
use tokio::io::AsyncWriteExt;
use std::path::Path;
use std::time::UNIX_EPOCH;
use crate::signals::LoggingSignal;
pub async fn log_data(mut data: Receiver<LoggingSignal>) {
let mut file: Option<File> = None;
loop {
match data.recv().await {
Some(signal) => {
match signal {
LoggingSignal::SingleImpedance(timestamp, frequency, magnitude, phase) => {
if let Some(f) = file.as_mut() {
let _ = f
.write_all(format!("{},{},{:.3},{:.3}\n", timestamp.duration_since(UNIX_EPOCH).unwrap().as_millis(), frequency, magnitude, phase).as_bytes())
.await;
}
}
LoggingSignal::MultiImpedance(timestamp, frequencies, magnitudes, phases) => {
if let Some(f) = file.as_mut() {
for i in 0..frequencies.len() {
let _ = f.write_all(
format!("{},{},{},{},{}\n",
timestamp.duration_since(UNIX_EPOCH).unwrap().as_millis(),
frequencies[i],
magnitudes[i],
phases[i],
i // optional index
).as_bytes()
).await;
}
}
}
LoggingSignal::StartFileLogging(filename) => {
let path = Path::new(&filename);
let base = path.file_stem().and_then(|s| s.to_str()).unwrap_or("log");
let ext = path.extension().and_then(|s| s.to_str()).unwrap_or("");
let mut counter = 0;
let mut new_filename = filename.clone();
while Path::new(&new_filename).exists() {
counter += 1;
new_filename = if ext.is_empty() {
format!("{}_{}", base, counter)
} else {
format!("{}_{}.{}", base, counter, ext)
};
}
match File::create(&new_filename).await {
Ok(f) => {
info!("Started file logging to: {}", new_filename);
file = Some(f);
}
Err(e) => {
info!("Failed to create log file '{}': {}", new_filename, e);
}
}
}
LoggingSignal::StopFileLogging => {
if file.is_some() {
info!("Stopped file logging");
file = None;
}
}
}
}
None => break, // Channel closed
}
}
}

View File

@@ -66,12 +66,12 @@ impl BodePlot {
pub fn update_magnitudes(&mut self, points: MeasurementPointSet, magnitudes: Vec<f32>) { pub fn update_magnitudes(&mut self, points: MeasurementPointSet, magnitudes: Vec<f32>) {
let freqs = points.values().to_vec(); let freqs = points.values().to_vec();
// self.magnitudes = freqs.into_iter().zip(magnitudes.into_iter()).map(|(f, m)| PlotPoint::new(f.log10(), 20.0 * m.log10() as f32)).collect(); // self.magnitudes = freqs.into_iter().zip(magnitudes.into_iter()).map(|(f, m)| PlotPoint::new(f.log10(), 20.0 * m.log10() as f32)).collect();
self.magnitudes = freqs.into_iter().zip(magnitudes.into_iter()).map(|(f, m)| PlotPoint::new(f.log10(), m as f32)).collect(); self.magnitudes = freqs.into_iter().zip(magnitudes.into_iter()).map(|(f, m)| PlotPoint::new((f as f32).log10(), m)).collect(); // Convert to f32 first due to rouding errors
} }
pub fn update_phases(&mut self, points: MeasurementPointSet, phases: Vec<f32>) { pub fn update_phases(&mut self, points: MeasurementPointSet, phases: Vec<f32>) {
let freqs = points.values().to_vec(); let freqs = points.values().to_vec();
self.phases = freqs.into_iter().zip(phases.into_iter()).map(|(f, p)| PlotPoint::new(f.log10(), p as f64)).collect(); self.phases = freqs.into_iter().zip(phases.into_iter()).map(|(f, p)| PlotPoint::new((f as f32).log10(), p)).collect(); // Convert to f32 first due to rouding errors
} }
pub fn plot_magnitudes(&self) -> PlotPoints { pub fn plot_magnitudes(&self) -> PlotPoints {

View File

@@ -1,6 +1,8 @@
use std::time::SystemTime;
use crate::icd::{IcdDftNum, MeasurementPointSet}; use crate::icd::{IcdDftNum, MeasurementPointSet};
#[derive(Copy, Clone)] #[derive(Copy, Clone, Debug)]
pub enum StartStopSignal { pub enum StartStopSignal {
StartSingle(u32, IcdDftNum), // frequency in Hz, DFT number StartSingle(u32, IcdDftNum), // frequency in Hz, DFT number
StartMulti(MeasurementPointSet), // DFT number, number of points per measurement StartMulti(MeasurementPointSet), // DFT number, number of points per measurement
@@ -8,6 +10,8 @@ pub enum StartStopSignal {
} }
pub enum LoggingSignal { pub enum LoggingSignal {
SingleImpedance(f32, f32), // magnitude, phase SingleImpedance(SystemTime, u32, f32, f32), // frequency, magnitude, phase
MultiImpedance(Vec<f32>, Vec<f32>), // magnitude, phase MultiImpedance(SystemTime, Vec<u32>, Vec<f32>, Vec<f32>), // frequency, magnitude, phase
StartFileLogging(String), // e.g. filename
StopFileLogging,
} }