thermalcam_decoder/examples/live.rs

use dotenv::dotenv;
use std::fs::File;
use std::io::Write;
use thermaldecoder::{Frame, Header, HDR_SIZE};

// fn main() -> Result<(), eframe::Error> {
//     env_logger::init(); // Log to stderr (if you run with `RUST_LOG=debug`).
//     let options = eframe::NativeOptions {
//         viewport: egui::ViewportBuilder::default().with_inner_size([600.0, 800.0]),
//         ..Default::default()
//     };
//     let (sender, receiver) = crossbeam::channel::unbounded();
//     eframe::run_native(
//         "Image Viewer",
//         options,
//         Box::new(|cc| {
//             // This gives us image support:
//             egui_extras::install_image_loaders(&cc.egui_ctx);
//             Box::<MyApp>::default()
//         }),
//     )
// }

// #[derive(Default)]
// struct MyApp {}

// impl eframe::App for MyApp {
//     fn update(&mut self, ctx: &egui::Context, _frame: &mut eframe::Frame) {
//         egui::CentralPanel::default().show(ctx, |ui| {
//             egui::ScrollArea::both().show(ui, |ui| {
//                 ui.add(egui::Image::new().rounding(10.0));

//                 ui.image(egui::include_image!("ferris.svg"));
//             });
//         });
//     }
// }

fn main() -> anyhow::Result<()> {
    dotenv().ok();
    let device = match std::env::var("THERMALCAM_IFACE=enp1s0f0") {
        Ok(d) => {
            let device = pcap::Device::list()
                .expect("device list failed")
                .into_iter()
                .find(|x| x.name == d)
                .expect(&format!("could not find device {}", d));
            device
        }
        Err(_) => pcap::Device::lookup()
            .expect("device lookup failed")
            .expect("no device available"),
    };
    // get the default Device

    println!("Using device {}", device.name);
    let output = std::env::args()
        .nth(1)
        .expect("required output file argument");
    println!("Using output {}", output);

    // Setup Capture
    let mut cap = pcap::Capture::from_device(device)
        .unwrap()
        .immediate_mode(true)
        .open()
        .unwrap();

    // get a packet and print its bytes
    const PACKET_LEN: usize = 6972;
    const FRAME_LEN: usize = 288 * 384 * 2;
    let mut frame = [0u8; FRAME_LEN];
    let mut len = 0;
    let mut out = File::create(&output)?;
    while let Ok(p) = cap.next_packet() {
        let data = p.data;
        if data.len() != 6972 {
            continue;
        }
        let data = &data[0x2a..];
        let header = match Header::read(data) {
            Ok(header) => header,
            Err(_) => continue,
        };
        let data = &data[HDR_SIZE..];
        if (header.part == 0 && len > 0)
            // do not write out of bounds - would panic, instead just skip
            || (data.len() + len > FRAME_LEN)
        {
            if len == FRAME_LEN {
                out.write_all(&frame[..])?;
            }
            len = 0;
        }
        frame[len..len + data.len()].copy_from_slice(data);
        len += data.len();
    }
    Ok(())
}