use eframe::egui; use egui_plot::{Line, Plot, PlotPoints, MarkerShape, LineStyle, PlotBounds, VLine}; use std::collections::{HashMap, VecDeque}; use std::net::{TcpStream, UdpSocket}; use std::io::{Write, BufReader, BufRead}; use std::sync::{Arc, Mutex}; use std::thread; use serde::{Deserialize, Serialize}; use chrono::Local; use crossbeam_channel::{unbounded, Receiver, Sender}; use socket2::{Socket, Domain, Type, Protocol}; use regex::Regex; use once_cell::sync::Lazy; static APP_START_TIME: Lazy = Lazy::new(std::time::Instant::now); // --- Models --- #[derive(Serialize, Deserialize, Clone, Debug)] struct Signal { name: String, id: u32, #[serde(rename = "type")] sig_type: String, } #[derive(Deserialize)] struct DiscoverResponse { #[serde(rename = "Signals")] signals: Vec, } #[derive(Serialize, Deserialize, Clone, Debug)] struct TreeItem { #[serde(rename = "Name")] name: String, #[serde(rename = "Class")] class: String, #[serde(rename = "Children")] children: Option>, #[serde(rename = "Type")] sig_type: Option, #[serde(rename = "Dimensions")] dimensions: Option, #[serde(rename = "Elements")] elements: Option, } #[derive(Clone)] struct LogEntry { time: String, level: String, message: String, } struct TraceData { values: VecDeque<[f64; 2]>, last_value: f64, } struct SignalMetadata { names: Vec, sig_type: String, } #[derive(Clone)] struct ConnectionConfig { ip: String, tcp_port: String, udp_port: String, log_port: String, version: u64, } #[derive(Clone, Copy, PartialEq, Debug)] enum PlotType { Normal, LogicAnalyzer, } #[derive(Clone, Copy, PartialEq, Debug)] enum AcquisitionMode { FreeRun, Triggered, } #[derive(Clone, Copy, PartialEq, Debug)] enum TriggerEdge { Rising, Falling, Both, } #[derive(Clone, Copy, PartialEq, Debug)] enum TriggerType { Single, Continuous, } #[derive(Clone, Copy, PartialEq, Debug)] enum MarkerType { None, Circle, Square, } impl MarkerType { fn to_shape(&self) -> Option { match self { MarkerType::None => None, MarkerType::Circle => Some(MarkerShape::Circle), MarkerType::Square => Some(MarkerShape::Square), } } } #[derive(Clone)] struct SignalPlotConfig { source_name: String, label: String, unit: String, color: egui::Color32, line_style: LineStyle, marker_type: MarkerType, gain: f64, offset: f64, } struct PlotInstance { id: String, plot_type: PlotType, signals: Vec, auto_bounds: bool, } enum InternalEvent { Log(LogEntry), Discovery(Vec), Tree(TreeItem), CommandResponse(String), NodeInfo(String), Connected, Disconnected, InternalLog(String), TraceRequested(String), ClearTrace(String), UdpStats(u64), UdpDropped(u32), } // --- App State --- struct ForcingDialog { signal_path: String, value: String, } struct LogFilters { show_debug: bool, show_info: bool, show_warning: bool, show_error: bool, paused: bool, content_regex: String, } struct ScopeSettings { enabled: bool, window_ms: f64, mode: AcquisitionMode, paused: bool, // Trigger Settings trigger_type: TriggerType, trigger_source: String, trigger_edge: TriggerEdge, trigger_threshold: f64, pre_trigger_percent: f64, // Internal State trigger_active: bool, last_trigger_time: f64, is_armed: bool, } struct MarteDebugApp { connected: bool, is_breaking: bool, config: ConnectionConfig, shared_config: Arc>, app_tree: Option, traced_signals: Arc>>, id_to_meta: Arc>>, plots: Vec, forced_signals: HashMap, logs: VecDeque, log_filters: LogFilters, show_left_panel: bool, show_right_panel: bool, show_bottom_panel: bool, selected_node: String, node_info: String, udp_packets: u64, udp_dropped: u64, forcing_dialog: Option, style_editor: Option<(usize, usize)>, // plot_idx, signal_idx tx_cmd: Sender, rx_events: Receiver, internal_tx: Sender, shared_x_range: Option<[f64; 2]>, scope: ScopeSettings, } impl MarteDebugApp { fn new(_cc: &eframe::CreationContext<'_>) -> Self { let (tx_cmd, rx_cmd_internal) = unbounded::(); let (tx_events, rx_events) = unbounded::(); let internal_tx = tx_events.clone(); let config = ConnectionConfig { ip: "127.0.0.1".to_string(), tcp_port: "8080".to_string(), udp_port: "8081".to_string(), log_port: "8082".to_string(), version: 0, }; let shared_config = Arc::new(Mutex::new(config.clone())); let id_to_meta = Arc::new(Mutex::new(HashMap::new())); let traced_signals = Arc::new(Mutex::new(HashMap::new())); let id_to_meta_clone = id_to_meta.clone(); let traced_signals_clone = traced_signals.clone(); let shared_config_cmd = shared_config.clone(); let shared_config_log = shared_config.clone(); let shared_config_udp = shared_config.clone(); let tx_events_c = tx_events.clone(); thread::spawn(move || { tcp_command_worker(shared_config_cmd, rx_cmd_internal, tx_events_c); }); let tx_events_log = tx_events.clone(); thread::spawn(move || { tcp_log_worker(shared_config_log, tx_events_log); }); let tx_events_udp = tx_events.clone(); thread::spawn(move || { udp_worker(shared_config_udp, id_to_meta_clone, traced_signals_clone, tx_events_udp); }); Self { connected: false, is_breaking: false, config, shared_config, app_tree: None, id_to_meta, traced_signals, plots: vec![PlotInstance { id: "Plot 1".to_string(), plot_type: PlotType::Normal, signals: Vec::new(), auto_bounds: true, }], forced_signals: HashMap::new(), logs: VecDeque::with_capacity(2000), log_filters: LogFilters { show_debug: true, show_info: true, show_warning: true, show_error: true, paused: false, content_regex: "".to_string(), }, show_left_panel: true, show_right_panel: true, show_bottom_panel: true, selected_node: "".to_string(), node_info: "".to_string(), udp_packets: 0, udp_dropped: 0, forcing_dialog: None, style_editor: None, tx_cmd, rx_events, internal_tx, shared_x_range: None, scope: ScopeSettings { enabled: false, window_ms: 1000.0, mode: AcquisitionMode::FreeRun, paused: false, trigger_type: TriggerType::Continuous, trigger_source: "".to_string(), trigger_edge: TriggerEdge::Rising, trigger_threshold: 0.0, pre_trigger_percent: 25.0, trigger_active: false, last_trigger_time: 0.0, is_armed: true, }, } } fn next_color(idx: usize) -> egui::Color32 { let colors = [ egui::Color32::from_rgb(100, 200, 255), egui::Color32::from_rgb(255, 100, 100), egui::Color32::from_rgb(100, 255, 100), egui::Color32::from_rgb(255, 200, 100), egui::Color32::from_rgb(255, 100, 255), egui::Color32::from_rgb(100, 255, 255), egui::Color32::from_rgb(200, 255, 100), egui::Color32::WHITE, ]; colors[idx % colors.len()] } fn render_tree(&mut self, ui: &mut egui::Ui, item: &TreeItem, path: String) { let current_path = if path.is_empty() { if item.name == "Root" { "".to_string() } else { item.name.clone() } } else { if path.is_empty() { item.name.clone() } else { format!("{}.{}", path, item.name) } }; let label = if item.class == "Signal" { format!("📈 {}", item.name) } else { item.name.clone() }; if let Some(children) = &item.children { let header = egui::CollapsingHeader::new(format!("{} [{}]", label, item.class)).id_salt(¤t_path); header.show(ui, |ui| { ui.horizontal(|ui| { if !current_path.is_empty() { if ui.selectable_label(self.selected_node == current_path, "ℹ Info").clicked() { self.selected_node = current_path.clone(); let _ = self.tx_cmd.send(format!("INFO {}", current_path)); } } }); for child in children { self.render_tree(ui, child, current_path.clone()); } }); } else { ui.horizontal(|ui| { if ui.selectable_label(self.selected_node == current_path, format!("{} [{}]", label, item.class)).clicked() { self.selected_node = current_path.clone(); let _ = self.tx_cmd.send(format!("INFO {}", current_path)); } if item.class.contains("Signal") { if ui.button("Trace").clicked() { let _ = self.tx_cmd.send(format!("TRACE {} 1", current_path)); let _ = self.internal_tx.send(InternalEvent::TraceRequested(current_path.clone())); } if ui.button("⚡ Force").clicked() { self.forcing_dialog = Some(ForcingDialog { signal_path: current_path.clone(), value: "".to_string() }); } } }); } } fn apply_trigger_logic(&mut self) { if self.scope.mode != AcquisitionMode::Triggered || !self.scope.is_armed { return; } if self.scope.trigger_source.is_empty() { return; } let data_map = self.traced_signals.lock().unwrap(); if let Some(data) = data_map.get(&self.scope.trigger_source) { if data.values.len() < 2 { return; } // Search for edge crossing in the last 100 samples let start_idx = if data.values.len() > 100 { data.values.len() - 100 } else { 0 }; for i in (start_idx + 1..data.values.len()).rev() { let v_prev = data.values[i-1][1]; let v_curr = data.values[i][1]; let t_curr = data.values[i][0]; // Avoid re-triggering on the same exact sample if t_curr <= self.scope.last_trigger_time { continue; } let triggered = match self.scope.trigger_edge { TriggerEdge::Rising => v_prev < self.scope.trigger_threshold && v_curr >= self.scope.trigger_threshold, TriggerEdge::Falling => v_prev > self.scope.trigger_threshold && v_curr <= self.scope.trigger_threshold, TriggerEdge::Both => (v_prev < self.scope.trigger_threshold && v_curr >= self.scope.trigger_threshold) || (v_prev > self.scope.trigger_threshold && v_curr <= self.scope.trigger_threshold), }; if triggered { self.scope.last_trigger_time = t_curr; self.scope.trigger_active = true; if self.scope.trigger_type == TriggerType::Single { self.scope.is_armed = false; } break; } } } } } fn tcp_command_worker(shared_config: Arc>, rx_cmd: Receiver, tx_events: Sender) { let mut current_version = 0; let mut current_addr = String::new(); loop { { let config = shared_config.lock().unwrap(); if config.version != current_version { current_version = config.version; current_addr = format!("{}:{}", config.ip, config.tcp_port); } } if current_addr.is_empty() || current_addr.starts_with(":") { thread::sleep(std::time::Duration::from_secs(1)); continue; } if let Ok(mut stream) = TcpStream::connect(¤t_addr) { let _ = stream.set_nodelay(true); let mut reader = BufReader::new(stream.try_clone().unwrap()); let _ = tx_events.send(InternalEvent::Connected); let stop_flag = Arc::new(Mutex::new(false)); let stop_flag_reader = stop_flag.clone(); let tx_events_inner = tx_events.clone(); thread::spawn(move || { let mut line = String::new(); let mut json_acc = String::new(); let mut in_json = false; while reader.read_line(&mut line).is_ok() { if *stop_flag_reader.lock().unwrap() { break; } let trimmed = line.trim(); if trimmed.is_empty() { line.clear(); continue; } if !in_json && trimmed.starts_with("{") { in_json = true; json_acc.clear(); } if in_json { json_acc.push_str(trimmed); if trimmed == "OK DISCOVER" { in_json = false; let json_clean = json_acc.trim_end_matches("OK DISCOVER").trim(); if let Ok(resp) = serde_json::from_str::(json_clean) { let _ = tx_events_inner.send(InternalEvent::Discovery(resp.signals)); } json_acc.clear(); } else if trimmed == "OK TREE" { in_json = false; let json_clean = json_acc.trim_end_matches("OK TREE").trim(); if let Ok(resp) = serde_json::from_str::(json_clean) { let _ = tx_events_inner.send(InternalEvent::Tree(resp)); } json_acc.clear(); } else if trimmed == "OK INFO" { in_json = false; let json_clean = json_acc.trim_end_matches("OK INFO").trim(); let _ = tx_events_inner.send(InternalEvent::NodeInfo(json_clean.to_string())); json_acc.clear(); } } else { let _ = tx_events_inner.send(InternalEvent::CommandResponse(trimmed.to_string())); } line.clear(); } }); while let Ok(cmd) = rx_cmd.recv() { { let config = shared_config.lock().unwrap(); if config.version != current_version { *stop_flag.lock().unwrap() = true; break; } } if stream.write_all(format!("{}\n", cmd).as_bytes()).is_err() { break; } } let _ = tx_events.send(InternalEvent::Disconnected); } thread::sleep(std::time::Duration::from_secs(2)); } } fn tcp_log_worker(shared_config: Arc>, tx_events: Sender) { let mut current_version = 0; let mut current_addr = String::new(); loop { { let config = shared_config.lock().unwrap(); if config.version != current_version { current_version = config.version; current_addr = format!("{}:{}", config.ip, config.log_port); } } if current_addr.is_empty() || current_addr.starts_with(":") { thread::sleep(std::time::Duration::from_secs(1)); continue; } if let Ok(stream) = TcpStream::connect(¤t_addr) { let mut reader = BufReader::new(stream); let mut line = String::new(); while reader.read_line(&mut line).is_ok() { if shared_config.lock().unwrap().version != current_version { break; } let trimmed = line.trim(); if trimmed.starts_with("LOG ") { let parts: Vec<&str> = trimmed[4..].splitn(2, ' ').collect(); if parts.len() == 2 { let _ = tx_events.send(InternalEvent::Log(LogEntry { time: Local::now().format("%H:%M:%S%.3f").to_string(), level: parts[0].to_string(), message: parts[1].to_string() })); } } line.clear(); } } thread::sleep(std::time::Duration::from_secs(2)); } } fn udp_worker(shared_config: Arc>, id_to_meta: Arc>>, traced_data: Arc>>, tx_events: Sender) { let mut current_version = 0; let mut socket: Option = None; let mut last_seq: Option = None; loop { let (ver, port) = { let config = shared_config.lock().unwrap(); (config.version, config.udp_port.clone()) }; if ver != current_version || socket.is_none() { current_version = ver; if port.is_empty() { socket = None; continue; } let port_num: u16 = port.parse().unwrap_or(8081); let s = Socket::new(Domain::IPV4, Type::DGRAM, Some(Protocol::UDP)).ok(); let mut bound = false; if let Some(sock) = s { let _ = sock.set_reuse_address(true); #[cfg(all(unix, not(target_os = "solaris"), not(target_os = "illumos")))] let _ = sock.set_reuse_port(true); let _ = sock.set_recv_buffer_size(10 * 1024 * 1024); let addr = format!("0.0.0.0:{}", port_num).parse::().unwrap(); if sock.bind(&addr.into()).is_ok() { socket = Some(sock.into()); bound = true; } } if !bound { thread::sleep(std::time::Duration::from_secs(5)); continue; } let _ = socket.as_ref().unwrap().set_read_timeout(Some(std::time::Duration::from_millis(500))); last_seq = None; } let s = if let Some(sock) = socket.as_ref() { sock } else { thread::sleep(std::time::Duration::from_secs(1)); continue; }; let mut buf = [0u8; 4096]; let mut total_packets = 0u64; loop { if shared_config.lock().unwrap().version != current_version { break; } if let Ok(n) = s.recv(&mut buf) { total_packets += 1; if (total_packets % 500) == 0 { let _ = tx_events.send(InternalEvent::UdpStats(total_packets)); } if n < 20 { continue; } let mut magic_buf = [0u8; 4]; magic_buf.copy_from_slice(&buf[0..4]); if u32::from_le_bytes(magic_buf) != 0xDA7A57AD { continue; } let mut seq_buf = [0u8; 4]; seq_buf.copy_from_slice(&buf[4..8]); let seq = u32::from_le_bytes(seq_buf); if let Some(last) = last_seq { if seq != last + 1 && seq > last { let _ = tx_events.send(InternalEvent::UdpDropped(seq - last - 1)); } } last_seq = Some(seq); let count = u32::from_le_bytes(buf[16..20].try_into().unwrap()); let now = APP_START_TIME.elapsed().as_secs_f64(); let mut offset = 20; let mut local_updates: HashMap> = HashMap::new(); let mut last_values: HashMap = HashMap::new(); let metas = id_to_meta.lock().unwrap(); for _ in 0..count { if offset + 8 > n { break; } let id = u32::from_le_bytes(buf[offset..offset+4].try_into().unwrap()); let size = u32::from_le_bytes(buf[offset+4..offset+8].try_into().unwrap()); offset += 8; if offset + size as usize > n { break; } let data_slice = &buf[offset..offset + size as usize]; if let Some(meta) = metas.get(&id) { let t = meta.sig_type.as_str(); let val = match size { 1 => { if t.contains('u') { data_slice[0] as f64 } else { (data_slice[0] as i8) as f64 } }, 2 => { let b = data_slice[0..2].try_into().unwrap(); if t.contains('u') { u16::from_le_bytes(b) as f64 } else { i16::from_le_bytes(b) as f64 } }, 4 => { let b = data_slice[0..4].try_into().unwrap(); if t.contains("float") { f32::from_le_bytes(b) as f64 } else if t.contains('u') { u32::from_le_bytes(b) as f64 } else { i32::from_le_bytes(b) as f64 } }, 8 => { let b = data_slice[0..8].try_into().unwrap(); if t.contains("float") { f64::from_le_bytes(b) } else if t.contains('u') { u64::from_le_bytes(b) as f64 } else { i64::from_le_bytes(b) as f64 } }, _ => 0.0, }; for name in &meta.names { local_updates.entry(name.clone()).or_default().push([now, val]); last_values.insert(name.clone(), val); } } offset += size as usize; } drop(metas); if !local_updates.is_empty() { let mut data_map = traced_data.lock().unwrap(); for (name, new_points) in local_updates { if let Some(entry) = data_map.get_mut(&name) { for point in new_points { entry.values.push_back(point); } if let Some(lv) = last_values.get(&name) { entry.last_value = *lv; } while entry.values.len() > 100000 { entry.values.pop_front(); } } } } } } } } impl eframe::App for MarteDebugApp { fn update(&mut self, ctx: &egui::Context, _frame: &mut eframe::Frame) { while let Ok(event) = self.rx_events.try_recv() { match event { InternalEvent::Log(log) => { if !self.log_filters.paused { self.logs.push_back(log); if self.logs.len() > 2000 { self.logs.pop_front(); } } } InternalEvent::Discovery(signals) => { let mut metas = self.id_to_meta.lock().unwrap(); metas.clear(); for s in &signals { let meta = metas.entry(s.id).or_insert_with(|| SignalMetadata { names: Vec::new(), sig_type: s.sig_type.clone() }); if !meta.names.contains(&s.name) { meta.names.push(s.name.clone()); } } } InternalEvent::Tree(tree) => { self.app_tree = Some(tree); } InternalEvent::NodeInfo(info) => { self.node_info = info; } InternalEvent::TraceRequested(name) => { let mut data_map = self.traced_signals.lock().unwrap(); data_map.entry(name).or_insert_with(|| TraceData { values: VecDeque::with_capacity(10000), last_value: 0.0 }); } InternalEvent::ClearTrace(name) => { let mut data_map = self.traced_signals.lock().unwrap(); data_map.remove(&name); for plot in &mut self.plots { plot.signals.retain(|s| s.source_name != name); } } InternalEvent::UdpStats(count) => { self.udp_packets = count; } InternalEvent::UdpDropped(dropped) => { self.udp_dropped += dropped as u64; } InternalEvent::Connected => { self.connected = true; let _ = self.tx_cmd.send("TREE".to_string()); let _ = self.tx_cmd.send("DISCOVER".to_string()); } InternalEvent::Disconnected => { self.connected = false; } InternalEvent::InternalLog(msg) => { self.logs.push_back(LogEntry { time: Local::now().format("%H:%M:%S").to_string(), level: "GUI_ERROR".to_string(), message: msg }); } InternalEvent::CommandResponse(resp) => { self.logs.push_back(LogEntry { time: Local::now().format("%H:%M:%S").to_string(), level: "CMD_RESP".to_string(), message: resp }); } } } if self.scope.enabled { self.apply_trigger_logic(); } if let Some(dragged_name) = ctx.data_mut(|d| d.get_temp::(egui::Id::new("drag_signal"))) { egui::Area::new(egui::Id::new("drag_ghost")).fixed_pos(ctx.input(|i| i.pointer.hover_pos().unwrap_or(egui::Pos2::ZERO))).order(egui::Order::Tooltip).show(ctx, |ui| { ui.group(|ui| { ui.label(format!("📈 {}", dragged_name)); }); }); } if let Some(dialog) = &mut self.forcing_dialog { let mut close = false; egui::Window::new("Force Signal").show(ctx, |ui| { ui.label(&dialog.signal_path); ui.text_edit_singleline(&mut dialog.value); ui.horizontal(|ui| { if ui.button("Apply").clicked() { let _ = self.tx_cmd.send(format!("FORCE {} {}", dialog.signal_path, dialog.value)); self.forced_signals.insert(dialog.signal_path.clone(), dialog.value.clone()); close = true; } if ui.button("Cancel").clicked() { close = true; } }); }); if close { self.forcing_dialog = None; } } if let Some((p_idx, s_idx)) = self.style_editor { let mut close = false; egui::Window::new("Signal Style").show(ctx, |ui| { if let Some(plot) = self.plots.get_mut(p_idx) { if let Some(sig) = plot.signals.get_mut(s_idx) { ui.horizontal(|ui| { ui.label("Label:"); ui.text_edit_singleline(&mut sig.label); }); ui.horizontal(|ui| { ui.label("Unit:"); ui.text_edit_singleline(&mut sig.unit); }); ui.horizontal(|ui| { ui.label("Color:"); let mut color = sig.color.to_array(); if ui.color_edit_button_srgba_unmultiplied(&mut color).changed() { sig.color = egui::Color32::from_rgba_unmultiplied(color[0], color[1], color[2], color[3]); } }); ui.horizontal(|ui| { ui.label("Gain:"); ui.add(egui::DragValue::new(&mut sig.gain).speed(0.1)); }); ui.horizontal(|ui| { ui.label("Offset:"); ui.add(egui::DragValue::new(&mut sig.offset).speed(1.0)); }); if ui.button("Close").clicked() { close = true; } } } }); if close { self.style_editor = None; } } egui::TopBottomPanel::top("top").show(ctx, |ui| { ui.horizontal(|ui| { ui.toggle_value(&mut self.show_left_panel, "🗂 Tree"); ui.toggle_value(&mut self.show_right_panel, "📊 Signals"); ui.toggle_value(&mut self.show_bottom_panel, "📜 Logs"); ui.separator(); if ui.button("➕ Plot").clicked() { self.plots.push(PlotInstance { id: format!("Plot {}", self.plots.len()+1), plot_type: PlotType::Normal, signals: Vec::new(), auto_bounds: true }); } ui.separator(); ui.checkbox(&mut self.scope.enabled, "🔭 Scope"); if self.scope.enabled { egui::ComboBox::from_id_salt("window_size").selected_text(format!("{}ms", self.scope.window_ms)).show_ui(ui, |ui| { for ms in [10.0, 20.0, 50.0, 100.0, 200.0, 500.0, 1000.0, 2000.0, 5000.0, 10000.0] { ui.selectable_value(&mut self.scope.window_ms, ms, format!("{}ms", ms)); } }); ui.selectable_value(&mut self.scope.mode, AcquisitionMode::FreeRun, "Free"); ui.selectable_value(&mut self.scope.mode, AcquisitionMode::Triggered, "Trig"); if self.scope.mode == AcquisitionMode::FreeRun { if ui.button(if self.scope.paused { "▶ Resume" } else { "⏸ Pause" }).clicked() { self.scope.paused = !self.scope.paused; } } else { if ui.button(if self.scope.is_armed { "🔴 Armed" } else { "⚪ Single" }).clicked() { self.scope.is_armed = true; self.scope.trigger_active = false; } ui.menu_button("⚙ Trigger", |ui| { egui::Grid::new("trig_grid").num_columns(2).show(ui, |ui| { ui.label("Source:"); ui.text_edit_singleline(&mut self.scope.trigger_source); ui.end_row(); ui.label("Edge:"); egui::ComboBox::from_id_salt("edge").selected_text(format!("{:?}", self.scope.trigger_edge)).show_ui(ui, |ui| { ui.selectable_value(&mut self.scope.trigger_edge, TriggerEdge::Rising, "Rising"); ui.selectable_value(&mut self.scope.trigger_edge, TriggerEdge::Falling, "Falling"); ui.selectable_value(&mut self.scope.trigger_edge, TriggerEdge::Both, "Both"); }); ui.end_row(); ui.label("Threshold:"); ui.add(egui::DragValue::new(&mut self.scope.trigger_threshold).speed(0.1)); ui.end_row(); ui.label("Pre-trig %:"); ui.add(egui::Slider::new(&mut self.scope.pre_trigger_percent, 0.0..=100.0)); ui.end_row(); ui.label("Mode:"); ui.selectable_value(&mut self.scope.trigger_type, TriggerType::Single, "Single"); ui.selectable_value(&mut self.scope.trigger_type, TriggerType::Continuous, "Cont"); ui.end_row(); }); }); } } ui.separator(); let (btn_text, btn_color) = if self.is_breaking { ("▶ Resume App", egui::Color32::GREEN) } else { ("⏸ Pause App", egui::Color32::YELLOW) }; if ui.button(egui::RichText::new(btn_text).color(btn_color)).clicked() { self.is_breaking = !self.is_breaking; let _ = self.tx_cmd.send(if self.is_breaking { "PAUSE".to_string() } else { "RESUME".to_string() }); } ui.separator(); ui.menu_button("🔌 Conn", |ui| { egui::Grid::new("conn_grid").num_columns(2).show(ui, |ui| { ui.label("IP:"); ui.text_edit_singleline(&mut self.config.ip); ui.end_row(); ui.label("Control:"); ui.text_edit_singleline(&mut self.config.tcp_port); ui.end_row(); }); if ui.button("🔄 Apply").clicked() { self.config.version += 1; *self.shared_config.lock().unwrap() = self.config.clone(); ui.close_menu(); } if ui.button("❌ Off").clicked() { self.config.version += 1; let mut cfg = self.config.clone(); cfg.ip = "".to_string(); *self.shared_config.lock().unwrap() = cfg; ui.close_menu(); } }); ui.with_layout(egui::Layout::right_to_left(egui::Align::Center), |ui| { ui.label(format!("UDP: OK[{}] DROP[{}]", self.udp_packets, self.udp_dropped)); }); }); }); if self.show_left_panel { egui::SidePanel::left("left").resizable(true).width_range(200.0..=500.0).show(ctx, |ui| { egui::ScrollArea::vertical().show(ui, |ui| { if let Some(tree) = self.app_tree.clone() { self.render_tree(ui, &tree, "".to_string()); } }); }); } if self.show_right_panel { egui::SidePanel::right("right").resizable(true).width_range(250.0..=400.0).show(ctx, |ui| { ui.heading("Traced Signals"); let mut names: Vec<_> = { let data_map = self.traced_signals.lock().unwrap(); data_map.keys().cloned().collect() }; names.sort(); egui::ScrollArea::vertical().id_salt("traced_scroll").show(ui, |ui| { for key in names { let last_val = { self.traced_signals.lock().unwrap().get(&key).map(|d| d.last_value).unwrap_or(0.0) }; ui.horizontal(|ui| { let response = ui.add(egui::Label::new(format!("{}: {:.2}", key, last_val)).sense(egui::Sense::drag())); if response.drag_started() { ctx.data_mut(|d| d.insert_temp(egui::Id::new("drag_signal"), key.clone())); } if ui.button("❌").clicked() { let _ = self.tx_cmd.send(format!("TRACE {} 0", key)); let _ = self.internal_tx.send(InternalEvent::ClearTrace(key)); } }); } }); ui.separator(); ui.heading("Forced Signals"); for (path, val) in &self.forced_signals { ui.horizontal(|ui| { ui.label(format!("{}: {}", path, val)); if ui.button("❌").clicked() { let _ = self.tx_cmd.send(format!("UNFORCE {}", path)); } }); } }); } if self.show_bottom_panel { egui::TopBottomPanel::bottom("log_panel").resizable(true).default_height(150.0).show(ctx, |ui| { ui.horizontal(|ui| { ui.heading("Logs"); ui.separator(); ui.checkbox(&mut self.log_filters.show_debug, "Debug"); ui.checkbox(&mut self.log_filters.show_info, "Info"); ui.checkbox(&mut self.log_filters.show_warning, "Warn"); ui.checkbox(&mut self.log_filters.show_error, "Error"); ui.separator(); ui.label("Filter:"); ui.text_edit_singleline(&mut self.log_filters.content_regex); if ui.button("🗑 Clear").clicked() { self.logs.clear(); } }); ui.separator(); let regex = if !self.log_filters.content_regex.is_empty() { Regex::new(&self.log_filters.content_regex).ok() } else { None }; egui::ScrollArea::vertical().stick_to_bottom(true).auto_shrink([false, false]).show(ui, |ui| { for log in &self.logs { let show = match log.level.as_str() { "Debug" => self.log_filters.show_debug, "Information" => self.log_filters.show_info, "Warning" => self.log_filters.show_warning, "FatalError" | "OSError" | "ParametersError" => self.log_filters.show_error, _ => true }; if !show { continue; } if let Some(re) = ®ex { if !re.is_match(&log.message) && !re.is_match(&log.level) { continue; } } let color = match log.level.as_str() { "FatalError" | "OSError" | "ParametersError" => egui::Color32::from_rgb(255, 100, 100), "Warning" => egui::Color32::from_rgb(255, 255, 100), "Information" => egui::Color32::from_rgb(100, 255, 100), "Debug" => egui::Color32::from_rgb(100, 100, 255), "GUI_ERROR" => egui::Color32::from_rgb(255, 50, 255), "CMD_RESP" => egui::Color32::from_rgb(255, 255, 255), _ => egui::Color32::WHITE }; ui.horizontal_wrapped(|ui| { ui.label(egui::RichText::new(&log.time).color(egui::Color32::GRAY).monospace()); ui.label(egui::RichText::new(format!("[{}]", log.level)).color(color).strong()); ui.add(egui::Label::new(&log.message).wrap()); }); } }); }); } egui::CentralPanel::default().show(ctx, |ui| { let n_plots = self.plots.len(); if n_plots > 0 { let plot_height = ui.available_height() / n_plots as f32; let mut to_remove = None; let mut current_range = None; for (p_idx, plot_inst) in self.plots.iter_mut().enumerate() { ui.group(|ui| { ui.horizontal(|ui| { ui.label(egui::RichText::new(&plot_inst.id).strong()); ui.selectable_value(&mut plot_inst.plot_type, PlotType::Normal, "Series"); ui.selectable_value(&mut plot_inst.plot_type, PlotType::LogicAnalyzer, "Logic"); if ui.button("🗑").clicked() { to_remove = Some(p_idx); } }); let mut plot = Plot::new(&plot_inst.id).height(plot_height - 40.0).show_axes([true, true]); // SCOPE SYNC LOGIC if self.scope.enabled { let window_s = self.scope.window_ms / 1000.0; let center_t = if self.scope.mode == AcquisitionMode::Triggered { if self.scope.trigger_active { self.scope.last_trigger_time } else { APP_START_TIME.elapsed().as_secs_f64() } } else { APP_START_TIME.elapsed().as_secs_f64() }; let pre_trigger_s = (self.scope.pre_trigger_percent / 100.0) * window_s; let x_min = center_t - pre_trigger_s; let x_max = x_min + window_s; plot = plot.include_x(x_min).include_x(x_max); if !self.scope.paused { plot = plot.auto_bounds(egui::Vec2b::new(true, true)); } } else { if let Some(range) = self.shared_x_range { if !plot_inst.auto_bounds { plot = plot.include_x(range[0]).include_x(range[1]); } } if plot_inst.auto_bounds { plot = plot.auto_bounds(egui::Vec2b::new(true, true)); } } let plot_resp = plot.show(ui, |plot_ui| { if !self.scope.enabled && !plot_inst.auto_bounds { if let Some(range) = self.shared_x_range { let bounds = plot_ui.plot_bounds(); plot_ui.set_plot_bounds(PlotBounds::from_min_max([range[0], bounds.min()[1]], [range[1], bounds.max()[1]])); } } // Trigger Line if self.scope.enabled && self.scope.mode == AcquisitionMode::Triggered && self.scope.trigger_active { plot_ui.vline(VLine::new(self.scope.last_trigger_time).color(egui::Color32::YELLOW).style(LineStyle::Dashed { length: 5.0 })); } let data_map = self.traced_signals.lock().unwrap(); for (s_idx, sig_cfg) in plot_inst.signals.iter().enumerate() { if let Some(data) = data_map.get(&sig_cfg.source_name) { let mut points_vec = Vec::new(); for [t, v] in &data.values { let mut final_v = *v * sig_cfg.gain + sig_cfg.offset; if plot_inst.plot_type == PlotType::LogicAnalyzer { final_v = (s_idx as f64 * 1.5) + (if final_v > 0.5 { 1.0 } else { 0.0 }); } points_vec.push([*t, final_v]); } plot_ui.line(Line::new(PlotPoints::from(points_vec)).name(&sig_cfg.label).color(sig_cfg.color)); } } if p_idx == 0 || current_range.is_none() { let b = plot_ui.plot_bounds(); current_range = Some([b.min()[0], b.max()[0]]); } }); if plot_resp.response.hovered() && ctx.input(|i| i.pointer.any_released()) { if let Some(dropped) = ctx.data_mut(|d| d.get_temp::(egui::Id::new("drag_signal"))) { let color = Self::next_color(plot_inst.signals.len()); plot_inst.signals.push(SignalPlotConfig { source_name: dropped.clone(), label: dropped.clone(), unit: "".to_string(), color, line_style: LineStyle::Solid, marker_type: MarkerType::None, gain: 1.0, offset: 0.0 }); ctx.data_mut(|d| d.remove_temp::(egui::Id::new("drag_signal"))); } } if plot_resp.response.dragged() || ctx.input(|i| i.pointer.any_click() || i.smooth_scroll_delta.y != 0.0) { if plot_resp.response.hovered() { plot_inst.auto_bounds = false; let b = plot_resp.transform.bounds(); self.shared_x_range = Some([b.min()[0], b.max()[0]]); } } plot_resp.response.context_menu(|ui| { if ui.button("🔍 Fit View").clicked() { plot_inst.auto_bounds = true; self.shared_x_range = None; ui.close_menu(); } ui.separator(); let mut sig_to_remove = None; for (s_idx, sig) in plot_inst.signals.iter().enumerate() { ui.horizontal(|ui| { ui.label(&sig.label); if ui.button("🎨").clicked() { self.style_editor = Some((p_idx, s_idx)); ui.close_menu(); } if ui.button("❌").clicked() { sig_to_remove = Some(s_idx); ui.close_menu(); } }); } if let Some(idx) = sig_to_remove { plot_inst.signals.remove(idx); } }); }); } if let Some(idx) = to_remove { self.plots.remove(idx); } if !self.scope.enabled { if let Some(range) = current_range { if self.shared_x_range.is_none() { self.shared_x_range = Some(range); } } } } else { ui.centered_and_justified(|ui| { ui.label("Add a plot panel to begin analysis"); }); } }); ctx.request_repaint_after(std::time::Duration::from_millis(16)); } } fn main() -> Result<(), eframe::Error> { let options = eframe::NativeOptions { viewport: egui::ViewportBuilder::default().with_inner_size([1280.0, 800.0]), ..Default::default() }; eframe::run_native("MARTe2 Debug Explorer", options, Box::new(|cc| Ok(Box::new(MarteDebugApp::new(cc))))) }