use ffav::easy::{AudioDesc, OpenOptions, VideoDesc};
use std::convert::TryInto;
use std::sync::atomic::{AtomicBool, Ordering};
use std::sync::Arc;
use std::time::Instant;
fn main() -> Result<(), Box<dyn std::error::Error>> {
let early_exit = Arc::new(AtomicBool::new(false));
let early_exit_cloned = Arc::clone(&early_exit);
let early_exit_thread = std::thread::spawn(move || {
let mut buffer = String::new();
std::io::stdin().read_line(&mut buffer).unwrap();
early_exit_cloned.store(true, Ordering::SeqCst);
});
let a_desc = AudioDesc::new();
let v_desc = VideoDesc::with_h264(352, 288, 4000, 1000000);
let example_bytes = include_bytes!("envivio-352x288.264.framed");
for n in 0..100000 {
if early_exit.load(Ordering::SeqCst) {
break;
}
let start = Instant::now();
let mut mp4_writer = OpenOptions::new()
.media(a_desc)
.media(v_desc)
.format_options("movflags=frag_keyframe")
.open("/tmp/envivio-352x288.264.mp4")?;
let mut ts_writer = OpenOptions::new()
.media(a_desc)
.media(v_desc)
.format("mpegts")
.format_options("mpegts_copyts=1")
.open("/tmp/envivio-352x288.264.ts")?;
let mut offset: usize = 0;
let mut pts = 0;
while offset + 4 < example_bytes.len() {
let size_bytes = &example_bytes[offset..offset + 4];
let frame_size = i32::from_be_bytes(size_bytes.try_into().unwrap()) as usize;
offset += 4;
let frame_bytes = &example_bytes[offset..offset + frame_size];
offset += frame_size;
mp4_writer.write_bytes(frame_bytes, pts, 40000, false, 0)?;
ts_writer.write_bytes(frame_bytes, pts, 40000, false, 0)?;
pts += 40000;
}
let duration = start.elapsed();
println!(
"#{:6} Time elapsed {:?} to processing {:?} bytes!",
n,
duration,
example_bytes.len()
);
}
early_exit_thread.join().unwrap();
Ok(())
}