main.rs

mod communication;
mod data;
mod gui;
mod signal_process;

use anyhow::{Context, Result};
use communication::usb_comm::{connection, receive_data};
use gui::gui_config::GuiConfig;
use std::sync::atomic::AtomicBool;
use std::sync::Arc;
use tokio::{self, sync::mpsc, task};

/// Handles USB communication and data streaming
async fn usb_task(tx: mpsc::Sender<f32>) -> Result<()> {
    let mut usb_device_interface = connection().context("Failed to establish USB connection")?;

    // Start data receiving task
    let tx_clone = tx.clone();

    let _handle = tokio::spawn(async move {
        receive_data(&mut usb_device_interface, tx_clone.clone()).await;
    });

    Ok(())
}

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    // Create channel for data communication
    let (tx, rx) = mpsc::channel(100); // Buffer size of 100

    // Create shutdown signal
    let shutdown = Arc::new(AtomicBool::new(false));
    let _shutdown_clone = shutdown.clone();

    // Spawn USB connection task
    let _usb_handle = task::spawn(usb_task(tx));

    // let _debug_comm_handle = task::spawn(debug_data(tx.clone()));

    // Configure application window
    let native_options = eframe::NativeOptions {
        viewport: egui::ViewportBuilder::default()
            .with_inner_size(GuiConfig::default().get_window_size()),
        ..Default::default()
    };

    // Run the GUI application
    eframe::run_native(
        "Biosensor Virtualization Panel",
        native_options,
        Box::new(|_cc| Ok(Box::new(gui::ui::VirtualizationApp::new(rx)))),
    )?;

    Ok(())
}