78 lines
2.2 KiB
Rust
78 lines
2.2 KiB
Rust
use ffmpeg_rs_raw::{Decoder, Demuxer};
|
|
use ffmpeg_sys_the_third::{av_frame_free, av_packet_free};
|
|
use std::collections::HashMap;
|
|
use std::env::args;
|
|
use std::fs::File;
|
|
use std::io::{Cursor, Read};
|
|
use std::path::PathBuf;
|
|
|
|
fn main() {
|
|
let name = args().next().unwrap_or("main".to_string());
|
|
let path = if let Some(path) = args().skip(1).next() {
|
|
PathBuf::from(path)
|
|
} else {
|
|
eprintln!("Usage: {} <path>", name);
|
|
std::process::exit(1);
|
|
};
|
|
|
|
let cd = read_as_custom_io(path.clone());
|
|
scan_input(cd);
|
|
|
|
let fd = read_as_file(path.clone());
|
|
scan_input(fd);
|
|
|
|
let mut opt = HashMap::new();
|
|
opt.insert("analyzeduration".to_string(), "999".to_string());
|
|
let svg = include_bytes!("./cashu.svg");
|
|
let mut dx = Demuxer::new_custom_io(svg.as_slice(), Some("cashu.svg".to_string()));
|
|
dx.set_opt(opt.clone()).unwrap();
|
|
unsafe {
|
|
let mut decoder = Decoder::new();
|
|
let info = dx.probe_input().expect("probe failed");
|
|
for chan in &info.channels {
|
|
decoder.setup_decoder(chan, None).expect("setup failed");
|
|
}
|
|
loop_decoder(dx, decoder);
|
|
}
|
|
}
|
|
|
|
fn read_as_custom_io(path: PathBuf) -> Demuxer {
|
|
let mut data: Vec<u8> = Vec::new();
|
|
File::open(path).unwrap().read_to_end(&mut data).unwrap();
|
|
let reader = Cursor::new(data);
|
|
Demuxer::new_custom_io(reader, None)
|
|
}
|
|
|
|
fn read_as_file(path_buf: PathBuf) -> Demuxer {
|
|
Demuxer::new(path_buf.to_str().unwrap())
|
|
}
|
|
|
|
fn scan_input(mut demuxer: Demuxer) {
|
|
unsafe {
|
|
let info = demuxer.probe_input().expect("demuxer failed");
|
|
println!("{}", info);
|
|
decode_input(demuxer);
|
|
}
|
|
}
|
|
|
|
unsafe fn decode_input(demuxer: Demuxer) {
|
|
let decoder = Decoder::new();
|
|
loop_decoder(demuxer, decoder);
|
|
}
|
|
|
|
unsafe fn loop_decoder(mut demuxer: Demuxer, mut decoder: Decoder) {
|
|
loop {
|
|
let (mut pkt, stream) = demuxer.get_packet().expect("demuxer failed");
|
|
if pkt.is_null() {
|
|
break; // EOF
|
|
}
|
|
if let Ok(frames) = decoder.decode_pkt(pkt, stream) {
|
|
for (mut frame, _stream) in frames {
|
|
// do nothing but decode entire stream
|
|
av_frame_free(&mut frame);
|
|
}
|
|
}
|
|
av_packet_free(&mut pkt);
|
|
}
|
|
}
|