X-Git-Url: https://git.nihav.org/?p=nihav-tool.git;a=blobdiff_plain;f=src%2Fmain.rs;h=cba0b054e9d17ebe38fb9bf0b3afcaacc4aabf0c;hp=1be4e2a0a7e36a42ebb004e69251cc84006bd64a;hb=1e83105a93051ff19b417f18675eb7dafe55a1c3;hpb=69524874857bc441acc89c768e6cc7095f615a17 diff --git a/src/main.rs b/src/main.rs index 1be4e2a..cba0b05 100644 --- a/src/main.rs +++ b/src/main.rs @@ -77,14 +77,14 @@ fn main() { println!("usage: nihav-tool [-noout] [-vn] [-an] input [lastpts]"); return; } - let mut lastpts: Option = None; + let mut lastpts = NATimePoint::None; let mut cur_arg: usize = 1; let mut noout = false; let mut decode_video = true; let mut decode_audio = true; let mut nmode = NumberMode::FrmPTS; let mut smode = FrameSkipMode::None; - let mut seek_time = 0u64; + let mut seek_time = NATimePoint::None; let mut vpfx: Option = None; let mut apfx: Option<&str> = None; let mut ignore_errors = false; @@ -107,7 +107,7 @@ fn main() { println!("seek time missing"); return; } - let ret = args[cur_arg].parse::(); + let ret = args[cur_arg].parse::(); if ret.is_err() { println!("wrong seek time"); return; @@ -139,7 +139,12 @@ fn main() { let name = args[cur_arg].as_str(); cur_arg += 1; if cur_arg < args.len() { - lastpts = Some(u64::from_str_radix(args[cur_arg].as_str(), 10).unwrap()); + let ret = args[cur_arg].parse::(); + if ret.is_err() { + println!("cannot parse end time"); + return; + } + lastpts = ret.unwrap(); } let path = Path::new(name); @@ -158,13 +163,13 @@ println!("trying demuxer {} on {}", dmx_name, name); let mut dmx_reg = RegisteredDemuxers::new(); nihav_register_all_demuxers(&mut dmx_reg); let mut dec_reg = RegisteredDecoders::new(); - nihav_register_all_codecs(&mut dec_reg); + nihav_register_all_decoders(&mut dec_reg); dmx_fact = dmx_reg.find_demuxer(dmx_name).unwrap(); br.seek(SeekFrom::Start(0)).unwrap(); let mut dmx = create_demuxer(dmx_fact, &mut br).unwrap(); - if seek_time > 0 { + if seek_time != NATimePoint::None { let ret = dmx.seek(seek_time); if ret.is_err() { println!(" seek error {:?}", ret.err().unwrap()); @@ -225,7 +230,6 @@ println!("stream {} - {} {}", i, s, info.get_name()); } } else { decs.push(None); -panic!("decoder {} not found", info.get_name()); } if !has_out { writers.push(Outputter::None); @@ -259,7 +263,7 @@ panic!("decoder {} not found", info.get_name()); }; } }, - Err(DecoderError::MissingReference) if seek_time > 0 => { + Err(DecoderError::MissingReference) if seek_time != NATimePoint::None => { println!("ignoring missing ref"); }, Err(reason) => { @@ -269,7 +273,7 @@ panic!("decoder {} not found", info.get_name()); } }, }; - if pkt.get_pts() != None && lastpts.is_some() && pkt.get_pts() >= lastpts { break; } + if pkt.get_pts() != None && lastpts != NATimePoint::None && !pkt.ts.less_than(lastpts) { break; } } } //panic!("end");