thermalcam_decoder/examples/live.rs

83 lines
2.4 KiB
Rust
Raw Normal View History

2024-02-16 00:41:39 +02:00
use indicatif::ProgressBar;
use std::fs::File;
use std::io::Write;
use thermaldecoder::{write_raw_frame, 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<()> {
// get the default Device
let device = pcap::Device::lookup()
.expect("device lookup failed")
.expect("no device available");
println!("Using device {}", device.name);
// Setup Capture
let mut cap = pcap::Capture::from_device(device)
.unwrap()
.immediate_mode(true)
.open()
.unwrap();
// get a packet and print its bytes
let mut parts = vec![];
let pb = ProgressBar::new(1024);
let mut out = File::create("output.raw")?;
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 && parts.len() > 0 {
let frame = Frame {
header,
raw: parts.concat(),
};
if frame.raw.len() == 384 * 288 * 2 {
out.write_all(&frame.raw)?;
pb.inc(1);
}
parts.clear();
}
parts.push(data.to_vec());
}
Ok(())
}