OnlyVideo/OnlyAudio muxer capabilities mean several streams are accepted
[nihav-encoder.git] / src / main.rs
CommitLineData
6dc69f51
KS
1extern crate nihav_core;
2extern crate nihav_codec_support;
3extern crate nihav_registry;
4extern crate nihav_allstuff;
5
6use std::fs::File;
5ec8115f 7use std::io::{BufReader, Write};
4a835a2a 8use std::pin::Pin;
6dc69f51
KS
9use nihav_core::io::byteio::{FileReader, ByteReader};
10use nihav_core::frame::*;
11use nihav_core::options::*;
12use nihav_core::codecs::*;
13use nihav_core::demuxers::*;
14use nihav_core::muxers::*;
d9fe2b71 15use nihav_core::reorder::*;
6dc69f51 16use nihav_core::scale::*;
6dc69f51 17use nihav_registry::detect;
e6cb09af 18use nihav_registry::register;
6dc69f51 19use std::env;
5ec8115f 20use std::time::{Duration, Instant};
6dc69f51 21
91a15e39
KS
22mod demux;
23use crate::demux::*;
93521506 24mod null;
b0481c9e
KS
25mod acvt;
26use crate::acvt::*;
90683bc3
KS
27mod imgseq;
28use crate::imgseq::*;
93521506 29
4a835a2a
KS
30#[derive(Clone,Copy,Default,PartialEq)]
31enum RegisterResult {
32 #[default]
33 Ok,
34 Ignored,
35 Failed,
36}
37
38pub struct SelfBorrow<T, U> {
39 bval: T,
40 dst: Option<U>,
41}
42
43impl<T: Unpin, U: Unpin> SelfBorrow<T, U> {
44 pub fn new<F>(src: T, create: F) -> Pin<Box<Self>>
45 where F: Fn(*mut T) -> U {
46 let mut obj = Box::pin(Self{
47 bval: src,
48 dst: None,
49 });
50 let ptr = &mut obj.bval as *mut T;
51 obj.dst = Some(create(ptr));
52 obj
53 }
54 fn get_object(&self) -> &U {
55 if let Some(ref dst) = self.dst {
56 dst
57 } else {
58 unreachable!()
59 }
60 }
61 fn get_object_mut(&mut self) -> &mut U {
62 if let Some(ref mut dst) = self.dst {
63 dst
64 } else {
65 unreachable!()
66 }
67 }
68}
69
70impl<T, U> Drop for SelfBorrow<T, U> {
71 fn drop(&mut self) {
72 self.dst = None;
73 }
74}
75
76pub type SBBox<T, U> = Pin<Box<SelfBorrow<T, U>>>;
77
6f6ba7bf
KS
78const SND_NO_FORMAT: NASoniton = NASoniton { bits: 0, be: false, packed: false, planar: false, float: false, signed: false };
79
5ec8115f
KS
80fn format_time(ms: u64) -> String {
81 let s = ms / 1000;
82 let ds = (ms % 1000) / 100;
83 let (min, s) = (s / 60, s % 60);
84 let (h, min) = (min / 60, min % 60);
85 if h == 0 {
86 if min == 0 {
87 format!("{}.{}", s, ds)
88 } else {
89 format!("{}:{:02}.{}", min, s, ds)
90 }
91 } else {
92 format!("{}:{:02}:{:02}.{}", h, min, s, ds)
93 }
94}
95
6dc69f51
KS
96fn print_options(name: &str, options: &[NAOptionDefinition]) {
97 if options.is_empty() {
98 println!("No custom options.");
99 } else {
100 println!("Options for '{}'", name);
101 for opt in options.iter() {
102 println!(" {}", opt);
103 }
104 }
105}
106
30611a63
KS
107fn parse_bitrate(strval: &str) -> Result<u32, ()> {
108 let mut val = 0;
109 let mut has_suffix = false;
110 for ch in strval.chars() {
111 match ch {
112 _ if has_suffix => return Err(()),
113 '0'..='9' => {
114 if val >= std::u32::MAX / 100 {
115 return Err(());
116 }
117 val = val * 10 + ch.to_digit(10).unwrap_or(0);
118 },
119 'k' | 'K' => {
120 if val >= std::u32::MAX / 1000 {
121 return Err(());
122 }
123 val *= 1000;
124 has_suffix = true;
125 },
126 'm' | 'M' => {
127 if val >= std::u32::MAX / 1000000 {
128 return Err(());
129 }
130 val *= 1000000;
131 has_suffix = true;
132 },
133 _ => return Err(()),
134 };
135 }
136 Ok(val)
137}
138
6dc69f51
KS
139struct OptionArgs {
140 name: String,
141 value: Option<String>,
142}
143
144struct InputStreamOptions {
145 id: u32,
146 drop: bool,
147 dec_opts: Vec<OptionArgs>,
148}
149
150struct OutputStreamOptions {
151 id: u32,
152 enc_params: EncodeParameters,
153 enc_name: String,
154 enc_opts: Vec<OptionArgs>,
155}
156
157enum OutputConvert {
158 Video(NAScale, NABufferType),
b0481c9e 159 Audio(AudioConverter),
6dc69f51
KS
160 None,
161}
162
86c54d88 163#[allow(clippy::large_enum_variant)]
6dc69f51
KS
164enum OutputMode {
165 Drop,
166 Copy(u32),
167 Encode(u32, Box<dyn NAEncoder>, OutputConvert),
168}
169
170#[derive(Default)]
86c54d88 171#[allow(clippy::type_complexity)]
6dc69f51 172struct Transcoder {
4a835a2a
KS
173 input_name: [Option<String>; 16],
174 input_fmt: [Option<String>; 16],
6dc69f51
KS
175 output_name: String,
176 output_fmt: Option<String>,
4a835a2a 177 demux_opts: [Vec<OptionArgs>; 16],
6dc69f51
KS
178 mux_opts: Vec<OptionArgs>,
179 istr_opts: Vec<InputStreamOptions>,
180 ostr_opts: Vec<OutputStreamOptions>,
df37d3b1 181 scale_opts: Vec<(String, String)>,
d9fe2b71 182 decoders: Vec<Option<(Box<NADecoderSupport>, Box<dyn NADecoder>, Box<dyn FrameReorderer>)>>,
6dc69f51
KS
183 encoders: Vec<OutputMode>,
184 no_video: bool,
185 no_audio: bool,
951916e8
KS
186 start: NATimePoint,
187 end: NATimePoint,
908ba48c 188 verbose: u8,
79eb70eb
KS
189
190 calc_len: bool,
191 nframes: Vec<usize>,
e47b2114 192 global_tb: (u32, u32),
6dc69f51
KS
193}
194
195macro_rules! parse_and_apply_options {
196 ($obj: expr, $in_opts: expr, $name: expr) => {
197 let mut opts = Vec::with_capacity($in_opts.len());
198 let opt_def = $obj.get_supported_options();
199 for opt in $in_opts.iter() {
200 let mut found = false;
201 for opt_def in opt_def.iter() {
3e2a4e88
KS
202 let mut matches = opt.name == opt_def.name;
203 if !matches && opt.name.starts_with("no") {
204 let (_, name) = opt.name.split_at(2);
205 matches = name == opt_def.name;
206 }
207 if matches {
fd03a232 208 let arg = if let Some(ref strval) = opt.value { Some(strval) } else { None };
6dc69f51 209 let ret = opt_def.parse(&opt.name, arg);
86c54d88 210 if let Ok((val, _)) = ret {
6dc69f51 211 opts.push(val);
86c54d88
KS
212 } else {
213 println!("invalid option {} for {}", opt.name, $name);
6dc69f51
KS
214 }
215 found = true;
216 }
217 }
218 if !found {
219 println!(" ignoring option '{}' for {}", opt.name, $name);
220 }
221 }
222 $obj.set_options(opts.as_slice());
223 }
224}
225
226impl Transcoder {
227 fn new() -> Self { Self::default() }
228 fn parse_istream_options(&mut self, opt0: &str, opt1: &str) -> bool {
229 let (_, strno) = opt0.split_at(9);
230 let ret = strno.parse::<u32>();
231 if ret.is_err() { return false; }
232 let streamno = ret.unwrap();
233
fd03a232 234 let sidx = if let Some(idx) = self.istr_opts.iter().position(|el| el.id == streamno) {
6dc69f51
KS
235 idx
236 } else {
237 self.istr_opts.push(InputStreamOptions {id: streamno, drop: false, dec_opts: Vec::new() });
238 self.istr_opts.len() - 1
239 };
240 let istr = &mut self.istr_opts[sidx];
830c03a1 241
6dc69f51
KS
242 for opt in opt1.split(',') {
243 let oval: Vec<_> = opt.split('=').collect();
244 if oval.len() == 1 {
245 match oval[0] {
246 "drop" => { istr.drop = true; },
247 _ => {
248 istr.dec_opts.push(OptionArgs{ name: oval[0].to_string(), value: None });
249 },
250 };
251 } else if oval.len() == 2 {
252 istr.dec_opts.push(OptionArgs{ name: oval[0].to_string(), value: Some(oval[1].to_string()) });
253 } else {
254 println!("unrecognized option '{}'", opt);
255 }
256 }
257 true
258 }
259 fn parse_ostream_options(&mut self, opt0: &str, opt1: &str, enc_reg: &RegisteredEncoders) -> bool {
260 let (_, strno) = opt0.split_at(9);
261 let ret = strno.parse::<u32>();
262 if ret.is_err() { return false; }
263 let streamno = ret.unwrap();
264
fd03a232 265 let sidx = if let Some(idx) = self.ostr_opts.iter().position(|el| el.id == streamno) {
6dc69f51
KS
266 idx
267 } else {
268 self.ostr_opts.push(OutputStreamOptions {id: streamno, enc_name: String::new(), enc_params: EncodeParameters::default(), enc_opts: Vec::new() });
269 self.ostr_opts.len() - 1
270 };
271 let ostr = &mut self.ostr_opts[sidx];
830c03a1 272
6dc69f51
KS
273 for opt in opt1.split(',') {
274 let oval: Vec<_> = opt.split('=').collect();
275 if oval.len() == 1 {
276 match oval[0] {
277 "flip" => {
278 if ostr.enc_params.format == NACodecTypeInfo::None {
279 ostr.enc_params.format = NACodecTypeInfo::Video(NAVideoInfo::new(0, 0, false, YUV420_FORMAT));
280 }
281 if let NACodecTypeInfo::Video(ref mut vinfo) = ostr.enc_params.format {
282 vinfo.flipped = true;
283 } else {
284 println!("video option for audio stream");
285 }
286 },
287 "noflip" => {
288 if ostr.enc_params.format == NACodecTypeInfo::None {
289 ostr.enc_params.format = NACodecTypeInfo::Video(NAVideoInfo::new(0, 0, false, YUV420_FORMAT));
290 }
291 if let NACodecTypeInfo::Video(ref mut vinfo) = ostr.enc_params.format {
292 vinfo.flipped = false;
293 } else {
294 println!("video option for audio stream");
295 }
296 },
297 _ => {
298 ostr.enc_opts.push(OptionArgs{ name: oval[0].to_string(), value: None });
299 },
300 };
301 } else if oval.len() == 2 {
302//todo parse encoder options, store, init later
303 match oval[0] {
ca3b31d7
KS
304 "timebase" => {
305 let mut parts = oval[1].split('/');
306 let num = parts.next().unwrap();
307 let den = parts.next();
308 if let Some(den) = den {
309 let rnum = num.parse::<u32>();
310 let rden = den.parse::<u32>();
311 if let (Ok(num), Ok(den)) = (rnum, rden) {
312 ostr.enc_params.tb_num = num;
313 ostr.enc_params.tb_den = den;
314 } else {
315 println!("invalid timebase value");
316 }
317 } else {
318 println!("invalid timebase format (should be num/den)");
319 }
320 },
6dc69f51
KS
321 "encoder" => {
322 if enc_reg.find_encoder(oval[1]).is_some() {
323 ostr.enc_name = oval[1].to_string();
324 } else {
325 println!("unknown encoder '{}'", oval[1]);
c0052668 326 ostr.enc_name = oval[1].to_string();
6dc69f51
KS
327 }
328 },
329 "width" => {
330 if ostr.enc_params.format == NACodecTypeInfo::None {
331 ostr.enc_params.format = NACodecTypeInfo::Video(NAVideoInfo::new(0, 0, false, YUV420_FORMAT));
332 }
333 if let NACodecTypeInfo::Video(ref mut vinfo) = ostr.enc_params.format {
334 let ret = oval[1].parse::<usize>();
335 if let Ok(val) = ret {
336 vinfo.width = val;
337 } else {
338 println!("invalid width");
339 }
340 } else {
341 println!("video option for audio stream");
342 }
343 },
344 "height" => {
345 if ostr.enc_params.format == NACodecTypeInfo::None {
346 ostr.enc_params.format = NACodecTypeInfo::Video(NAVideoInfo::new(0, 0, false, YUV420_FORMAT));
347 }
348 if let NACodecTypeInfo::Video(ref mut vinfo) = ostr.enc_params.format {
349 let ret = oval[1].parse::<usize>();
350 if let Ok(val) = ret {
351 vinfo.height = val;
352 } else {
353 println!("invalid height");
354 }
355 } else {
356 println!("video option for audio stream");
357 }
358 },
e176bfee
KS
359 "pixfmt" => {
360 if ostr.enc_params.format == NACodecTypeInfo::None {
361 ostr.enc_params.format = NACodecTypeInfo::Video(NAVideoInfo::new(0, 0, false, YUV420_FORMAT));
362 }
363 if let NACodecTypeInfo::Video(ref mut vinfo) = ostr.enc_params.format {
364 let ret = oval[1].parse::<NAPixelFormaton>();
365 if let Ok(val) = ret {
366 vinfo.format = val;
367 } else {
368 println!("invalid pixel format");
369 }
370 } else {
371 println!("video option for audio stream");
372 }
373 },
6dc69f51
KS
374 "srate" => {
375 if ostr.enc_params.format == NACodecTypeInfo::None {
6f6ba7bf 376 ostr.enc_params.format = NACodecTypeInfo::Audio(NAAudioInfo::new(0, 0, SND_NO_FORMAT, 0));
6dc69f51
KS
377 }
378 if let NACodecTypeInfo::Audio(ref mut ainfo) = ostr.enc_params.format {
379 let ret = oval[1].parse::<u32>();
380 if let Ok(val) = ret {
381 ainfo.sample_rate = val;
382 } else {
383 println!("invalid sampling rate");
384 }
385 } else {
386 println!("audio option for video stream");
387 }
388 },
389 "channels" => {
390 if ostr.enc_params.format == NACodecTypeInfo::None {
6f6ba7bf 391 ostr.enc_params.format = NACodecTypeInfo::Audio(NAAudioInfo::new(0, 0, SND_NO_FORMAT, 0));
6dc69f51
KS
392 }
393 if let NACodecTypeInfo::Audio(ref mut ainfo) = ostr.enc_params.format {
394 let ret = oval[1].parse::<u8>();
395 if let Ok(val) = ret {
396 ainfo.channels = val;
397 } else {
398 println!("invalid number of channels");
399 }
400 } else {
401 println!("audio option for video stream");
402 }
403 },
404 "block_len" => {
405 if ostr.enc_params.format == NACodecTypeInfo::None {
6f6ba7bf 406 ostr.enc_params.format = NACodecTypeInfo::Audio(NAAudioInfo::new(0, 0, SND_NO_FORMAT, 0));
6dc69f51
KS
407 }
408 if let NACodecTypeInfo::Audio(ref mut ainfo) = ostr.enc_params.format {
409 let ret = oval[1].parse::<usize>();
410 if let Ok(val) = ret {
411 ainfo.block_len = val;
412 } else {
413 println!("invalid block_length");
414 }
415 } else {
416 println!("audio option for video stream");
417 }
418 },
e176bfee
KS
419 "sfmt" => {
420 if ostr.enc_params.format == NACodecTypeInfo::None {
6f6ba7bf 421 ostr.enc_params.format = NACodecTypeInfo::Audio(NAAudioInfo::new(0, 0, SND_NO_FORMAT, 0));
e176bfee
KS
422 }
423 if let NACodecTypeInfo::Audio(ref mut ainfo) = ostr.enc_params.format {
424 let ret = oval[1].parse::<NASoniton>();
425 if let Ok(val) = ret {
426 ainfo.format = val;
427 } else {
428 println!("invalid audio format");
429 }
430 } else {
431 println!("audio option for video stream");
432 }
433 },
434// todo channel map negotiation
435 /*"chmap" => {
436 if ostr.enc_params.format == NACodecTypeInfo::None {
6f6ba7bf 437 ostr.enc_params.format = NACodecTypeInfo::Audio(NAAudioInfo::new(0, 0, SND_NO_FORMAT, 0));
e176bfee
KS
438 }
439 if let NACodecTypeInfo::Audio(ref mut ainfo) = ostr.enc_params.format {
440 let ret = oval[1].parse::<NAChannelMap>();
441 if let Ok(val) = ret {
442 ainfo.chmap = val;
443 } else {
444 println!("invalid channel map");
445 }
446 } else {
447 println!("audio option for video stream");
448 }
449 },*/
6dc69f51 450 "bitrate" => {
30611a63 451 let ret = parse_bitrate(oval[1]);
6dc69f51
KS
452 if let Ok(val) = ret {
453 ostr.enc_params.bitrate = val;
454 } else {
455 println!("invalid bitrate value");
456 }
457 },
458 "quality" => {
459 let ret = oval[1].parse::<u8>();
460 if let Ok(val) = ret {
461 ostr.enc_params.quality = val;
462 } else {
463 println!("invalid quality value");
464 }
465 },
466 _ => {
467 ostr.enc_opts.push(OptionArgs{ name: oval[0].to_string(), value: Some(oval[1].to_string()) });
468 },
469 }
470 } else {
471 println!("unrecognized option '{}'", opt);
472 }
473 }
474 true
475 }
4a835a2a 476 fn parse_demuxer_options(&mut self, opts: &str, dmx_reg: &RegisteredDemuxers, input_no: usize) -> bool {
6dc69f51
KS
477 for opt in opts.split(',') {
478 let oval: Vec<_> = opt.split('=').collect();
479 if oval.len() == 1 {
4a835a2a 480 self.demux_opts[input_no].push(OptionArgs{ name: oval[0].to_string(), value: None });
6dc69f51
KS
481 } else if oval.len() == 2 {
482 if oval[0] == "format" {
483 if dmx_reg.find_demuxer(oval[1]).is_some() {
4a835a2a 484 self.input_fmt[input_no] = Some(oval[1].to_string());
6dc69f51
KS
485 } else {
486 println!("unknown demuxer format '{}'", oval[1]);
487 }
488 } else {
4a835a2a 489 self.demux_opts[input_no].push(OptionArgs{ name: oval[0].to_string(), value: Some(oval[1].to_string()) });
6dc69f51
KS
490 }
491 } else {
492 println!("unrecognized option '{}'", opt);
493 }
494 }
495 true
496 }
497 fn parse_muxer_options(&mut self, opts: &str, mux_reg: &RegisteredMuxers) -> bool {
498 for opt in opts.split(',') {
499 let oval: Vec<_> = opt.split('=').collect();
500 if oval.len() == 1 {
501 self.mux_opts.push(OptionArgs{ name: oval[0].to_string(), value: None });
502 } else if oval.len() == 2 {
503 if oval[0] == "format" {
504 if mux_reg.find_muxer(oval[1]).is_some() {
505 self.output_fmt = Some(oval[1].to_string());
506 } else {
507 println!("unknown muxer format '{}'", oval[1]);
508 }
509 } else {
510 self.mux_opts.push(OptionArgs{ name: oval[0].to_string(), value: Some(oval[1].to_string()) });
511 }
512 } else {
513 println!("unrecognized option '{}'", opt);
514 }
515 }
516 true
517 }
df37d3b1
KS
518 fn parse_scale_options(&mut self, opts: &str) -> bool {
519 for opt in opts.split(',') {
520 let oval: Vec<_> = opt.split('=').collect();
521 if oval.len() == 1 {
522 self.scale_opts.push((oval[0].to_string(), "".to_string()));
523 } else if oval.len() == 2 {
524 self.scale_opts.push((oval[0].to_string(), oval[1].to_string()));
525 } else {
526 println!("unrecognized option '{}'", opt);
527 return false;
528 }
529 }
530 true
531 }
73f889bf 532 fn apply_decoder_options(&self, dec: &mut dyn NADecoder, str_id: u32) {
fd03a232 533 if let Some(str_idx) = self.istr_opts.iter().position(|el| el.id == str_id) {
6dc69f51
KS
534 let dec_opts = dec.get_supported_options();
535 if dec_opts.is_empty() { return; }
536 let name = format!("input stream {}", str_id);
537 parse_and_apply_options!(dec, &self.istr_opts[str_idx].dec_opts, name);
538 }
539 }
79eb70eb 540 fn register_output_stream(&mut self, cname: &str, istr: NAStreamRef, iidx: usize, out_sm: &mut StreamManager, enc_reg: &RegisteredEncoders) -> RegisterResult {
6dc69f51 541 let out_id = out_sm.get_num_streams() as u32;
fd03a232 542 if let Some(str_idx) = self.istr_opts.iter().position(|el| el.id == (istr.get_num() as u32)) {
6dc69f51
KS
543 if self.istr_opts[str_idx].drop {
544 self.encoders.push(OutputMode::Drop);
4a835a2a 545 return RegisterResult::Ignored;
6dc69f51
KS
546 }
547 }
548
fd03a232 549 if let Some(str_idx) = self.ostr_opts.iter().position(|el| el.id == out_id) {
6dc69f51 550 let oopts = &mut self.ostr_opts[str_idx];
a7b5f008 551 if oopts.enc_name.as_str() == "copy" && (cname == "any" || istr.get_info().get_name() == cname) {
4a835a2a 552 out_sm.add_stream((*istr).clone());
6dc69f51
KS
553 self.encoders.push(OutputMode::Copy(out_id));
554 } else if cname == "any" || oopts.enc_name.as_str() == cname {
555 let enc_create = enc_reg.find_encoder(oopts.enc_name.as_str());
556 if enc_create.is_none() {
557 println!("encoder '{}' not found", oopts.enc_name.as_str());
4a835a2a 558 return RegisterResult::Failed;
6dc69f51
KS
559 }
560 let mut encoder = (enc_create.unwrap())();
eb563e3d 561 let forced_out = oopts.enc_params.format != NACodecTypeInfo::None;
6f6ba7bf 562 let iformat = istr.get_info().get_properties();
6dc69f51
KS
563 if oopts.enc_params.format == NACodecTypeInfo::None {
564 oopts.enc_params.format = istr.get_info().get_properties();
6f6ba7bf
KS
565 } else {
566 match (&iformat, &mut oopts.enc_params.format) {
567 (NACodecTypeInfo::Video(svinfo), NACodecTypeInfo::Video(ref mut dvinfo)) => {
568 if dvinfo.width == 0 {
569 dvinfo.width = svinfo.width;
570 }
571 if dvinfo.height == 0 {
572 dvinfo.height = svinfo.height;
573 }
574 },
575 (NACodecTypeInfo::Audio(sainfo), NACodecTypeInfo::Audio(ref mut dainfo)) => {
576 if dainfo.sample_rate == 0 {
577 dainfo.sample_rate = sainfo.sample_rate;
578 }
579 if dainfo.format == SND_NO_FORMAT {
580 dainfo.format = sainfo.format;
581 }
582 if dainfo.channels == 0 {
583 dainfo.channels = sainfo.channels;
584 }
585 if dainfo.block_len == 0 {
586 dainfo.block_len = sainfo.block_len;
587 }
588 },
589 _ => {},
590 };
6dc69f51 591 }
e47b2114
KS
592 if self.global_tb != (0, 0) {
593 oopts.enc_params.tb_num = self.global_tb.0;
594 oopts.enc_params.tb_den = self.global_tb.1;
595 }
6dc69f51
KS
596 if oopts.enc_params.tb_num == 0 {
597 oopts.enc_params.tb_num = istr.tb_num;
598 oopts.enc_params.tb_den = istr.tb_den;
599 }
600 let ret_eparams = encoder.negotiate_format(&oopts.enc_params);
601 if ret_eparams.is_err() {
602 println!("cannot negotiate encoding parameters");
4a835a2a 603 return RegisterResult::Failed;
6dc69f51
KS
604 }
605 let ret_eparams = ret_eparams.unwrap();
606
607//todo check for params mismatch
6f6ba7bf 608 let cvt = match (&iformat, &ret_eparams.format) {
6dc69f51 609 (NACodecTypeInfo::Video(svinfo), NACodecTypeInfo::Video(dvinfo)) => {
eb563e3d 610 if svinfo == dvinfo && !forced_out {
6dc69f51
KS
611 OutputConvert::None
612 } else {
613 let ofmt = ScaleInfo { fmt: dvinfo.format, width: dvinfo.width, height: dvinfo.height };
df37d3b1 614 let ret = NAScale::new_with_options(ofmt, ofmt, &self.scale_opts);
6dc69f51
KS
615 if ret.is_err() {
616 println!("cannot create scaler");
4a835a2a 617 return RegisterResult::Failed;
6dc69f51
KS
618 }
619 let scaler = ret.unwrap();
620 let ret = alloc_video_buffer(*dvinfo, 4);
621 if ret.is_err() {
622 println!("cannot create scaler buffer");
4a835a2a 623 return RegisterResult::Failed;
6dc69f51
KS
624 }
625 let cvt_buf = ret.unwrap();
626 OutputConvert::Video(scaler, cvt_buf)
627 }
628 },
629 (NACodecTypeInfo::Audio(sainfo), NACodecTypeInfo::Audio(dainfo)) => {
630 if sainfo == dainfo {
631 OutputConvert::None
632 } else {
633 let dchmap = match dainfo.channels {
634 1 => NAChannelMap::from_ms_mapping(0x4),
635 2 => NAChannelMap::from_ms_mapping(0x3),
636 _ => {
637println!("can't generate default channel map for {} channels", dainfo.channels);
4a835a2a 638 return RegisterResult::Failed;
6dc69f51
KS
639 },
640 };
b0481c9e 641 let acvt = AudioConverter::new(sainfo, dainfo, dchmap);
6dc69f51 642//todo channelmap
b0481c9e 643 OutputConvert::Audio(acvt)
6dc69f51
KS
644 }
645 },
646 _ => OutputConvert::None,
647 };
720f1a09
KS
648 let name = format!("output stream {}", out_id);
649 parse_and_apply_options!(encoder, &oopts.enc_opts, name);
650
79eb70eb
KS
651 if self.calc_len && self.nframes.len() > iidx {
652 encoder.set_options(&[NAOption{name: "nframes", value: NAValue::Int(self.nframes[iidx] as i64)}]);
653 }
654
6dc69f51
KS
655 let ret = encoder.init(out_id, ret_eparams);
656 if ret.is_err() {
657 println!("error initialising encoder");
4a835a2a 658 return RegisterResult::Failed;
6dc69f51
KS
659 }
660 out_sm.add_stream_ref(ret.unwrap());
661
6dc69f51
KS
662 parse_and_apply_options!(encoder, &oopts.enc_opts, name);
663
664 self.encoders.push(OutputMode::Encode(out_id, encoder, cvt));
665 } else {
666println!("encoder {} is not supported by output (expected {})", istr.id, istr.get_info().get_name());
4a835a2a 667 return RegisterResult::Failed;
6dc69f51 668 }
86c54d88 669 } else if cname == "any" || istr.get_info().get_name() == cname {
4a835a2a 670 out_sm.add_stream((*istr).clone());
86c54d88 671 self.encoders.push(OutputMode::Copy(out_id));
6dc69f51 672 } else {
00b4c46f
KS
673 let mut oopts = OutputStreamOptions {id: out_id, enc_name: cname.to_owned(), enc_params: EncodeParameters::default(), enc_opts: Vec::new() };
674
675 let enc_create = enc_reg.find_encoder(cname);
676 if enc_create.is_none() {
677 println!("encoder '{}' not found", oopts.enc_name.as_str());
4a835a2a 678 return RegisterResult::Failed;
00b4c46f
KS
679 }
680 let mut encoder = (enc_create.unwrap())();
681 oopts.enc_params.format = istr.get_info().get_properties();
e47b2114
KS
682 if self.global_tb != (0, 0) {
683 oopts.enc_params.tb_num = self.global_tb.0;
684 oopts.enc_params.tb_den = self.global_tb.1;
685 } else {
686 oopts.enc_params.tb_num = istr.tb_num;
687 oopts.enc_params.tb_den = istr.tb_den;
688 }
00b4c46f
KS
689 let ret_eparams = encoder.negotiate_format(&oopts.enc_params);
690 if ret_eparams.is_err() {
691 println!("cannot negotiate encoding parameters");
4a835a2a 692 return RegisterResult::Failed;
00b4c46f
KS
693 }
694 let ret_eparams = ret_eparams.unwrap();
695
696//todo check for params mismatch
697 let cvt = match (&oopts.enc_params.format, &ret_eparams.format) {
698 (NACodecTypeInfo::Video(svinfo), NACodecTypeInfo::Video(dvinfo)) => {
699 if svinfo == dvinfo {
700 OutputConvert::None
701 } else {
702 let ofmt = ScaleInfo { fmt: dvinfo.format, width: dvinfo.width, height: dvinfo.height };
703 let ret = NAScale::new_with_options(ofmt, ofmt, &self.scale_opts);
704 if ret.is_err() {
705 println!("cannot create scaler");
4a835a2a 706 return RegisterResult::Failed;
00b4c46f
KS
707 }
708 let scaler = ret.unwrap();
709 let ret = alloc_video_buffer(*dvinfo, 4);
710 if ret.is_err() {
711 println!("cannot create scaler buffer");
4a835a2a 712 return RegisterResult::Failed;
00b4c46f
KS
713 }
714 let cvt_buf = ret.unwrap();
715 OutputConvert::Video(scaler, cvt_buf)
716 }
717 },
718 (NACodecTypeInfo::Audio(sainfo), NACodecTypeInfo::Audio(dainfo)) => {
719 if sainfo == dainfo {
720 OutputConvert::None
721 } else {
722 let dchmap = match dainfo.channels {
723 1 => NAChannelMap::from_ms_mapping(0x4),
724 2 => NAChannelMap::from_ms_mapping(0x3),
725 _ => {
726println!("can't generate default channel map for {} channels", dainfo.channels);
4a835a2a 727 return RegisterResult::Failed;
00b4c46f
KS
728 },
729 };
730//todo channelmap
b0481c9e
KS
731 let acvt = AudioConverter::new(sainfo, dainfo, dchmap);
732 OutputConvert::Audio(acvt)
00b4c46f
KS
733 }
734 },
735 _ => OutputConvert::None,
736 };
737 let ret = encoder.init(out_id, ret_eparams);
738 if ret.is_err() {
739 println!("error initialising encoder");
4a835a2a 740 return RegisterResult::Failed;
00b4c46f
KS
741 }
742 out_sm.add_stream_ref(ret.unwrap());
743 self.encoders.push(OutputMode::Encode(out_id, encoder, cvt));
744 self.ostr_opts.push(oopts);
6dc69f51 745 }
4a835a2a 746 RegisterResult::Ok
6dc69f51
KS
747 }
748 fn map_single(&mut self, cname: &str, ctype: StreamType, src_sm: &StreamManager, out_sm: &mut StreamManager, enc_reg: &RegisteredEncoders) -> bool {
749 let mut found_stream = false;
79eb70eb 750 for (iidx, istr) in src_sm.iter().enumerate() {
6dc69f51
KS
751 if istr.get_media_type() != ctype || found_stream {
752 self.encoders.push(OutputMode::Drop);
753 } else {
79eb70eb 754 match self.register_output_stream(cname, istr, iidx, out_sm, enc_reg) {
4a835a2a
KS
755 RegisterResult::Ok => found_stream = true,
756 RegisterResult::Failed => return false,
757 RegisterResult::Ignored => {},
758 };
6dc69f51
KS
759 }
760 }
761 found_stream
762 }
763 fn negotiate_stream_map(&mut self, src_sm: &StreamManager, mux_caps: MuxerCapabilities, out_sm: &mut StreamManager, enc_reg: &RegisteredEncoders) -> bool {
764 match mux_caps {
765 MuxerCapabilities::SingleVideo(cname) => {
766 if self.no_video { return false; }
767 self.map_single(cname, StreamType::Video, src_sm, out_sm, enc_reg)
768 },
769 MuxerCapabilities::SingleAudio(cname) => {
770 if self.no_audio { return false; }
771 self.map_single(cname, StreamType::Audio, src_sm, out_sm, enc_reg)
772 },
773 MuxerCapabilities::SingleVideoAndAudio(vname, aname) => {
774 let mut found_vid = false;
775 let mut found_aud = false;
79eb70eb 776 for (iidx, istr) in src_sm.iter().enumerate() {
6dc69f51 777 if istr.get_media_type() == StreamType::Video && !found_vid && !self.no_video {
79eb70eb 778 match self.register_output_stream(vname, istr, iidx, out_sm, enc_reg) {
4a835a2a
KS
779 RegisterResult::Ok => found_vid = true,
780 RegisterResult::Failed => return false,
781 RegisterResult::Ignored => {},
782 };
6dc69f51 783 } else if istr.get_media_type() == StreamType::Audio && !found_aud && !self.no_audio {
79eb70eb 784 match self.register_output_stream(aname, istr, iidx, out_sm, enc_reg) {
4a835a2a
KS
785 RegisterResult::Ok => found_aud = true,
786 RegisterResult::Failed => return false,
787 RegisterResult::Ignored => {},
788 };
6dc69f51
KS
789 } else {
790 self.encoders.push(OutputMode::Drop);
791 }
792 }
793 found_vid | found_aud
794 },
795 MuxerCapabilities::OnlyVideo => {
796 if self.no_video { return false; }
797
798 let mut found_vid = false;
79eb70eb 799 for (iidx, istr) in src_sm.iter().enumerate() {
429e677e 800 if istr.get_media_type() == StreamType::Video {
79eb70eb 801 match self.register_output_stream("any", istr, iidx, out_sm, enc_reg) {
4a835a2a
KS
802 RegisterResult::Ok => found_vid = true,
803 RegisterResult::Failed => return false,
804 RegisterResult::Ignored => {},
805 };
6dc69f51
KS
806 } else {
807 self.encoders.push(OutputMode::Drop);
808 }
809 }
810 found_vid
811 },
812 MuxerCapabilities::OnlyAudio => {
813 if self.no_audio { return false; }
814
815 let mut found_aud = false;
79eb70eb 816 for (iidx, istr) in src_sm.iter().enumerate() {
429e677e 817 if istr.get_media_type() == StreamType::Audio {
79eb70eb 818 match self.register_output_stream("any", istr, iidx, out_sm, enc_reg) {
4a835a2a
KS
819 RegisterResult::Ok => found_aud = true,
820 RegisterResult::Failed => return false,
821 RegisterResult::Ignored => {},
822 };
6dc69f51
KS
823 } else {
824 self.encoders.push(OutputMode::Drop);
825 }
826 }
827 found_aud
828 },
829 MuxerCapabilities::Universal => {
79eb70eb 830 for (iidx, istr) in src_sm.iter().enumerate() {
6dc69f51
KS
831 if (istr.get_media_type() == StreamType::Video && self.no_video) ||
832 (istr.get_media_type() == StreamType::Audio && self.no_audio) {
833 self.encoders.push(OutputMode::Drop);
834 continue;
835 }
79eb70eb 836 if self.register_output_stream("any", istr, iidx, out_sm, enc_reg) == RegisterResult::Failed {
6dc69f51
KS
837 return false;
838 }
839 }
840 true
841 },
842 }
843 }
79eb70eb
KS
844 fn create_demuxers(&mut self, demuxers: &mut Vec<(DemuxerObject, bool)>, full_reg: &FullRegister, print_info: bool) -> bool {
845 let mut isn_start = 0;
846 for (i, (iname, ifmt)) in self.input_name.iter().zip(
847 self.input_fmt.iter()).enumerate() {
848 match (iname, ifmt.as_ref().map(|s| s.as_str())) {
849 (Some(name), Some("imgseq")) => {
850 println!("trying image sequence {}", name);
851 let mut isdc = ImgSeqDemuxerCreator::new(name.as_str());
852 parse_and_apply_options!(isdc, &self.demux_opts[i], "input");
853 let isd = if let Ok(ctx) = isdc.open() {
854 ctx
855 } else {
856 println!("failed to create image sequence demuxer!");
857 return false;
858 };
859 let dmx = DemuxerObject::create_imgseq(isd);
860 if print_info {
861 for i in 0..dmx.get_num_streams() {
862 let s = dmx.get_stream(i).unwrap();
863 let info = s.get_info();
864 println!(" stream {}({}) - {} {}", i, i + isn_start, s, info.get_name());
865 }
866 }
867 isn_start += dmx.get_num_streams();
868 demuxers.push((dmx, false))
869 },
870 (Some(name), _) => {
871 let res = File::open(name);
872 if res.is_err() {
873 println!("error opening input");
874 return false;
875 }
876 let file = res.unwrap();
877 let file = BufReader::new(file);
878 let mut fr = FileReader::new_read(file);
879 let mut br = ByteReader::new(&mut fr);
880 let (is_raw, start, end) = if ifmt.is_none() {
881 detect_tags(&mut br)
882 } else {
883 (false, 0, None)
884 };
885
886 let nfr: Box<dyn ByteIO> = if start != 0 || end.is_some() {
887 let file = fr.finish();
888 Box::new(BoundedFileReader::new_read(file, start, end).unwrap())
889 } else {
890 Box::new(fr)
891 };
892 let sb = SelfBorrow::new(nfr, |rd| {
893 unsafe {
894 ByteReader::new(rd.as_mut().unwrap().as_mut())
895 }
896 });
897
898 let mut dmx = DemuxerObject::create(sb, full_reg, name, ifmt, is_raw, print_info);
899 if dmx.is_none() {
900 println!("cannot find demuxer for '{}'", name);
901 return false;
902 }
903 parse_and_apply_options!(dmx, &self.demux_opts[i], "input");
904 if print_info {
905 for i in 0..dmx.get_num_streams() {
906 let s = dmx.get_stream(i).unwrap();
907 let info = s.get_info();
908 println!(" stream {}({}) - {} {}", i, i + isn_start, s, info.get_name());
909 }
910 }
911 isn_start += dmx.get_num_streams();
912 demuxers.push((dmx, false));
913 },
914 _ => {},
915 };
916 }
917 true
918 }
6dc69f51
KS
919}
920
df37d3b1 921fn encode_frame(dst_id: u32, encoder: &mut Box<dyn NAEncoder>, cvt: &mut OutputConvert, frm: NAFrameRef, scale_opts: &[(String, String)]) -> bool {
d9fe2b71 922 let buf = frm.get_buffer();
637e4d45 923 let cbuf = if let NABufferType::None = buf {
343a59ec 924 if (encoder.get_capabilities() & ENC_CAPS_SKIPFRAME) == 0 {
b76448d9
KS
925 match cvt {
926 OutputConvert::Video(_, ref mut dbuf) => dbuf.clone(),
927 _ => {
928 println!("encoder does not support skip frames, skipping");
929 return true;
930 },
931 }
932 } else {
933 buf
343a59ec 934 }
637e4d45
KS
935 } else {
936 match cvt {
d9fe2b71
KS
937 OutputConvert::None => buf,
938 OutputConvert::Video(ref mut scaler, ref mut dbuf) => {
939 let cur_ifmt = get_scale_fmt_from_pic(&buf);
940 let last_ifmt = scaler.get_in_fmt();
941 if cur_ifmt != last_ifmt {
942 let ofmt = scaler.get_out_fmt();
df37d3b1 943 let ret = NAScale::new_with_options(cur_ifmt, ofmt, scale_opts);
d9fe2b71
KS
944 if ret.is_err() {
945 println!("error re-initialising scaler for {} -> {}", cur_ifmt, ofmt);
946 return false;
947 }
948 *scaler = ret.unwrap();
949 }
950 let ret = scaler.convert(&buf, dbuf);
951 if ret.is_err() {
952 println!("error converting frame for encoding");
953 return false;
954 }
955 dbuf.clone()
956 },
b0481c9e
KS
957 OutputConvert::Audio(ref mut acvt) => {
958 if !acvt.queue_frame(buf, frm.get_time_information()) {
d9fe2b71
KS
959 println!("error converting audio for stream {}", dst_id);
960 return false;
961 }
b0481c9e 962 return true;
d9fe2b71 963 },
637e4d45
KS
964 }
965 };
d9fe2b71
KS
966 let cfrm = NAFrame::new(frm.get_time_information(), frm.frame_type, frm.key, frm.get_info(), cbuf);
967 encoder.encode(&cfrm).unwrap();
968 true
969}
970
6dc69f51
KS
971macro_rules! next_arg {
972 ($args: expr, $arg_idx: expr) => {
973 if $arg_idx + 1 >= $args.len() {
974 println!("codec name is required");
975 }
976 $arg_idx += 1;
977 }
978}
979
4a835a2a
KS
980macro_rules! parse_id {
981 ($val: expr, $stype: expr, $maxval: expr) => {
982 if $val.is_empty() {
983 0
984 } else if let Ok(val) = $val.parse::<usize>() {
985 if val < $maxval {
986 val
987 } else {
988 println!("{} number should be below {}", $stype, $maxval);
989 return;
990 }
991 } else {
992 println!("invalid {} number '{}'", $stype, $val);
993 return;
994 }
995 }
996}
997
86c54d88 998#[allow(clippy::single_match)]
6dc69f51
KS
999fn main() {
1000 let args: Vec<_> = env::args().collect();
1001
1002 if args.len() == 1 {
071d353e
KS
1003 println!("usage: nihav-encoder [options] --input inputfile --output outputfile");
1004 println!(" use nihav-encoder --help to list all available options");
1005 return;
1006 }
1007 if args.len() == 2 && (args[1] == "--help" || args[1] == "-h") {
1008 println!("usage: nihav-encoder [options] --input inputfile --output outputfile");
1009 println!(" query options:");
1010 println!(" --list-{{decoders,encoders,demuxers,muxers}} - lists all available decoders/encoders/demuxers/muxers");
1011 println!(" --query-{{decoder,encoder,demuxer,muxer}}-options name - lists all options recognized by that decoder/encoder/demuxer/muxer");
1012 println!(" processing options:");
3e82df4a 1013 println!(" --verbose - show time for the currently processed input");
071d353e
KS
1014 println!(" --input inputfile - set input file");
1015 println!(" --input-format fmt - force input format");
1016 println!(" --demuxer-options options - set input demuxer options");
df37d3b1 1017 println!(" --scale-options options - set scaler options");
071d353e
KS
1018 println!(" --output outputfile - set output file");
1019 println!(" --output-format fmt - force output format");
1020 println!(" --muxer-options options - set output muxer options");
1021 println!(" --no-audio - do not decode audio streams");
1022 println!(" --no-video - do not decode video streams");
1023 println!(" --start starttime - start decoding from given position");
1024 println!(" --end endtime - end decoding at given position");
1025 println!(" --istreamX options - set options for input stream X");
1026 println!(" --ostreamX options - set options for output stream X");
1027 println!();
1028 println!(" (de)muxer and stream options are passed as comma-separated list e.g. --ostream0 width=320,height=240,flip");
6dc69f51
KS
1029 return;
1030 }
1031
91a15e39 1032 let full_reg = FullRegister::new();
6dc69f51
KS
1033
1034 let mut transcoder = Transcoder::new();
e47b2114 1035 let mut use_video_tb = false;
6dc69f51
KS
1036
1037 let mut arg_idx = 1;
5a8a7cdb 1038 let mut printed_info = false;
6dc69f51
KS
1039 while arg_idx < args.len() {
1040 match args[arg_idx].as_str() {
e6cb09af 1041 "--list-decoders" => {
91a15e39 1042 if full_reg.dec_reg.iter().len() > 0 {
e6cb09af 1043 println!("Registered decoders:");
91a15e39 1044 for dec in full_reg.dec_reg.iter() {
e6cb09af
KS
1045 let cdesc = register::get_codec_description(dec.name);
1046 let full_name = if let Some(cd) = cdesc { cd.get_full_name() } else { "???" };
1047 println!(" {} ({})", dec.name, full_name);
1048 }
1049 } else {
1050 println!("No registered decoders.");
1051 }
5a8a7cdb 1052 printed_info = true;
e6cb09af
KS
1053 },
1054 "--list-encoders" => {
91a15e39 1055 if full_reg.enc_reg.iter().len() > 0 {
e6cb09af 1056 println!("Registered encoders:");
91a15e39 1057 for enc in full_reg.enc_reg.iter() {
e6cb09af
KS
1058 let cdesc = register::get_codec_description(enc.name);
1059 let full_name = if let Some(cd) = cdesc { cd.get_full_name() } else { "???" };
1060 println!(" {} ({})", enc.name, full_name);
1061 }
1062 } else {
1063 println!("No registered encoders.");
1064 }
5a8a7cdb 1065 printed_info = true;
e6cb09af
KS
1066 },
1067 "--list-demuxers" => {
1068 print!("Registered demuxers:");
91a15e39 1069 for dmx in full_reg.dmx_reg.iter() {
e6cb09af
KS
1070 print!(" {}", dmx.get_name());
1071 }
1072 println!();
5a8a7cdb 1073 printed_info = true;
e6cb09af
KS
1074 },
1075 "--list-muxers" => {
1076 print!("Registered muxers:");
91a15e39 1077 for mux in full_reg.mux_reg.iter() {
e6cb09af
KS
1078 print!(" {}", mux.get_name());
1079 }
1080 println!();
5a8a7cdb 1081 printed_info = true;
e6cb09af 1082 },
6dc69f51
KS
1083 "--query-decoder-options" => {
1084 next_arg!(args, arg_idx);
1085 let cname = args[arg_idx].as_str();
91a15e39 1086 if let Some(decfunc) = full_reg.dec_reg.find_decoder(cname) {
6dc69f51
KS
1087 let dec = (decfunc)();
1088 let opts = dec.get_supported_options();
1089 print_options(cname, opts);
1090 } else {
1091 println!("codec {} is not found", cname);
1092 }
5a8a7cdb 1093 printed_info = true;
6dc69f51
KS
1094 },
1095 "--query-demuxer-options" => {
1096 next_arg!(args, arg_idx);
1097 let dname = args[arg_idx].as_str();
1098 let mut mr = MemoryReader::new_read(&[]);
1099 let mut br = ByteReader::new(&mut mr);
91a15e39 1100 if let Some(dmx_creator) = full_reg.dmx_reg.find_demuxer(dname) {
6dc69f51
KS
1101 let dmx = dmx_creator.new_demuxer(&mut br);
1102 let opts = dmx.get_supported_options();
1103 print_options(dname, opts);
1104 } else {
1105 println!("demuxer {} is not found", dname);
1106 }
5a8a7cdb 1107 printed_info = true;
6dc69f51
KS
1108 },
1109 "--query-encoder-options" => {
1110 next_arg!(args, arg_idx);
1111 let cname = args[arg_idx].as_str();
91a15e39 1112 if let Some(encfunc) = full_reg.enc_reg.find_encoder(cname) {
6dc69f51
KS
1113 let enc = (encfunc)();
1114 let opts = enc.get_supported_options();
1115 print_options(cname, opts);
1116 } else {
1117 println!("codec {} is not found", cname);
1118 }
5a8a7cdb 1119 printed_info = true;
6dc69f51
KS
1120 },
1121 "--query-muxer-options" => {
1122 next_arg!(args, arg_idx);
1123 let name = args[arg_idx].as_str();
1124 let mut data = [];
1125 let mut mw = MemoryWriter::new_write(&mut data);
1126 let mut bw = ByteWriter::new(&mut mw);
91a15e39 1127 if let Some(mux_creator) = full_reg.mux_reg.find_muxer(name) {
6dc69f51
KS
1128 let mux = mux_creator.new_muxer(&mut bw);
1129 let opts = mux.get_supported_options();
1130 print_options(name, opts);
1131 } else {
1132 println!("muxer {} is not found", name);
1133 }
5a8a7cdb 1134 printed_info = true;
6dc69f51 1135 },
4b5c61e2 1136 "--output" | "-o" => {
6dc69f51
KS
1137 next_arg!(args, arg_idx);
1138 transcoder.output_name = args[arg_idx].clone();
1139 },
1140 "--output-format" => {
1141 next_arg!(args, arg_idx);
1142 transcoder.output_fmt = Some(args[arg_idx].clone());
1143 },
df37d3b1
KS
1144 "--scale-options" => {
1145 next_arg!(args, arg_idx);
1146 if !transcoder.parse_scale_options(&args[arg_idx]) {
1147 println!("invalid scale option syntax");
1148 return;
1149 }
1150 },
4b5c61e2 1151 "--no-video" | "-vn" => {
6dc69f51
KS
1152 transcoder.no_video = true;
1153 },
4b5c61e2 1154 "--no-audio" | "-an" => {
6dc69f51
KS
1155 transcoder.no_audio = true;
1156 },
951916e8
KS
1157 "--start" => {
1158 next_arg!(args, arg_idx);
1159 let ret = args[arg_idx].parse::<NATimePoint>();
1160 if let Ok(val) = ret {
1161 transcoder.start = val;
1162 } else {
1163 println!("invalid start time");
1164 return;
1165 }
1166 },
1167 "--end" => {
1168 next_arg!(args, arg_idx);
1169 let ret = args[arg_idx].parse::<NATimePoint>();
1170 if let Ok(val) = ret {
1171 transcoder.end = val;
1172 } else {
b0c51548 1173 println!("invalid end time");
951916e8
KS
1174 return;
1175 }
1176 },
6dc69f51
KS
1177 "--muxer-options" => {
1178 next_arg!(args, arg_idx);
91a15e39 1179 if !transcoder.parse_muxer_options(&args[arg_idx], &full_reg.mux_reg) {
6dc69f51
KS
1180 println!("invalid muxer option syntax");
1181 return;
1182 }
1183 },
79eb70eb
KS
1184 "--calc-len" => {
1185 transcoder.calc_len = true;
1186 },
e47b2114
KS
1187 "--use-video-tb" => {
1188 use_video_tb = true;
1189 },
908ba48c
KS
1190 "--verbose" | "-v" => transcoder.verbose = 1,
1191 "-vv" => transcoder.verbose = 2,
1192 "-v-" => transcoder.verbose = 0,
6dc69f51
KS
1193 _ => {
1194 if args[arg_idx].starts_with("--istream") {
1195 let opt0 = &args[arg_idx];
1196 next_arg!(args, arg_idx);
1197 if !transcoder.parse_istream_options(opt0, &args[arg_idx]) {
1198 println!("invalid input stream option syntax");
1199 return;
1200 }
1201 } else if args[arg_idx].starts_with("--ostream") {
1202 let opt0 = &args[arg_idx];
1203 next_arg!(args, arg_idx);
91a15e39 1204 if !transcoder.parse_ostream_options(opt0, &args[arg_idx], &full_reg.enc_reg) {
6dc69f51
KS
1205 println!("invalid output stream option syntax");
1206 return;
1207 }
4a835a2a
KS
1208 } else if args[arg_idx].starts_with("--input-format") {
1209 let id = parse_id!(&args[arg_idx][14..], "input format", transcoder.input_fmt.len());
1210 next_arg!(args, arg_idx);
1211 transcoder.input_fmt[id] = Some(args[arg_idx].clone());
1212 } else if args[arg_idx].starts_with("--input") { // should be after --input-format
1213 let id = parse_id!(&args[arg_idx][7..], "input", transcoder.input_name.len());
1214 next_arg!(args, arg_idx);
1215 transcoder.input_name[id] = Some(args[arg_idx].clone());
1216 } else if args[arg_idx].starts_with("-i") {
1217 let id = parse_id!(&args[arg_idx][2..], "input", transcoder.input_name.len());
1218 next_arg!(args, arg_idx);
1219 transcoder.input_name[id] = Some(args[arg_idx].clone());
1220 } else if args[arg_idx].starts_with("--demuxer-options") {
1221 let id = parse_id!(&args[arg_idx][17..], "input options", transcoder.demux_opts.len());
1222 next_arg!(args, arg_idx);
1223 if !transcoder.parse_demuxer_options(&args[arg_idx], &full_reg.dmx_reg, id) {
1224 println!("invalid demuxer option syntax");
1225 return;
1226 }
6dc69f51 1227 } else if args[arg_idx].starts_with("--") {
bafd0754 1228 println!("unknown option '{}'", args[arg_idx]);
6dc69f51 1229 } else {
bafd0754 1230 println!("unrecognized argument '{}'", args[arg_idx]);
6dc69f51
KS
1231 }
1232 },
1233 };
1234 arg_idx += 1;
1235 }
1236
5a8a7cdb
KS
1237 if printed_info {
1238 return;
1239 }
1240
4a835a2a
KS
1241 if transcoder.input_name.iter().flatten().count() == 0 {
1242 println!("no input name(s) provided");
6dc69f51
KS
1243 return;
1244 }
86c54d88 1245 if transcoder.output_name.is_empty() {
6dc69f51
KS
1246 println!("no output name provided");
1247 return;
1248 }
1249
4a835a2a 1250 let mut demuxers = Vec::with_capacity(1);
79eb70eb
KS
1251 if !transcoder.create_demuxers(&mut demuxers, &full_reg, true) {
1252 return;
6dc69f51 1253 }
90683bc3 1254
4a835a2a
KS
1255 let duration = demuxers.iter().fold(0u64, |mindur, (dmx, _)| {
1256 let dur = dmx.get_duration();
1257 if dur > 0 {
1258 mindur.min(dur)
1259 } else {
1260 mindur
1261 }
1262 });
5ec8115f
KS
1263 let duration_string = if duration != 0 { format_time(duration) } else { String::new() };
1264
4a835a2a
KS
1265 let mut ism = StreamManager::new();
1266 let mut is_offset = Vec::with_capacity(demuxers.len());
1267 let mut start = 0;
1268 let mut nstreams = 0;
1269 for (dmx, _) in demuxers.iter() {
1270 is_offset.push(nstreams);
1271 let sm = dmx.get_stream_manager();
1272 let max_id = sm.iter().fold(0u32, |id, strm| id.max(strm.id));
1273 for stream in sm.iter() {
1274 let mut newstream = (*stream).clone();
e47b2114
KS
1275 if use_video_tb && transcoder.global_tb == (0, 0) && newstream.get_media_type() == StreamType::Video {
1276 transcoder.global_tb = newstream.get_timebase();
1277 }
4a835a2a
KS
1278 newstream.id += start;
1279 ism.add_stream(newstream);
6dc69f51 1280 }
4a835a2a
KS
1281 start += max_id + 1;
1282 nstreams += sm.get_num_streams();
6dc69f51 1283 }
4a835a2a
KS
1284
1285 for (&is_off, (dmx, _)) in is_offset.iter().zip(demuxers.iter_mut()) {
1286 for i in 0..dmx.get_num_streams() {
1287 let s = dmx.get_stream(i).unwrap();
1288 let info = s.get_info();
1289 let decfunc = full_reg.dec_reg.find_decoder(info.get_name());
1290 let str_id = (s.get_num() + is_off) as u32;
1291 if let Some(create_dec) = decfunc {
1292 let mut dec = (create_dec)();
1293 let mut dsupp = Box::new(NADecoderSupport::new());
1294 let ret = dec.init(&mut dsupp, info.clone());
1295 if ret.is_err() {
1296 println!("Error initialising decoder '{}' for stream {}", info.get_name(), str_id);
1297 return;
1298 }
1299 transcoder.apply_decoder_options(dec.as_mut(), str_id);
1300 let desc = register::get_codec_description(info.get_name());
1301 let has_b = if let Some(desc) = desc {
1302 desc.has_reorder()
1303 } else {
1304 println!("No codec description found, using B-frame reorderer.");
1305 true
1306 };
1307 let reord: Box<dyn FrameReorderer> = if has_b { Box::new(IPBReorderer::new()) } else { Box::new(NoReorderer::new()) };
1308 transcoder.decoders.push(Some((dsupp, dec, reord)));
1309 } else {
1310 println!("No decoder for stream {} ({}) is found", str_id, info.get_name());
1311 transcoder.decoders.push(None);
1312 }
1313 }
1314 if transcoder.start != NATimePoint::None {
1315 let ret = dmx.seek(transcoder.start);
1316 if ret.is_err() {
1317 println!(" failed to seek to {} error {:?}", transcoder.start, ret.err().unwrap());
1318 }
951916e8
KS
1319 }
1320 }
6dc69f51 1321
fd03a232
KS
1322 let output_fmt = if let Some(ref fmtname) = transcoder.output_fmt {
1323 fmtname
c23d0c3a
KS
1324 } else if transcoder.output_name.as_str() == "/dev/null" {
1325 "null"
fd03a232
KS
1326 } else if let Some(fmtname) = detect::detect_format_by_name(transcoder.output_name.as_str()) {
1327 fmtname
6dc69f51 1328 } else {
86c54d88
KS
1329 println!("Cannot guess muxer for output");
1330 return;
6dc69f51 1331 };
91a15e39 1332 let ret = full_reg.mux_reg.find_muxer(output_fmt);
6dc69f51
KS
1333 let ofmt = output_fmt.to_string();
1334
1335 if ret.is_none() {
1336 println!("cannot find muxer '{}'", output_fmt);
becae00f 1337 return;
6dc69f51
KS
1338 }
1339 let mux_creator = ret.unwrap();
1340
79eb70eb
KS
1341 if transcoder.calc_len {
1342 let mut sids = Vec::new();
1343 transcoder.nframes.clear();
1344 for (dmx, _) in demuxers.iter_mut() {
1345 let sstart = transcoder.nframes.len();
1346 let sm = dmx.get_stream_manager();
1347 sids.clear();
1348 for stream in sm.iter() {
1349 transcoder.nframes.push(0);
1350 sids.push(stream.get_id());
1351 }
1352
1353 while let Ok(pkt) = dmx.get_frame() {
1354 let stream = pkt.get_stream();
1355 let pos = sstart + sids.iter().position(|&x| x == stream.get_id()).unwrap();
1356 transcoder.nframes[pos] += 1;
1357 }
1358 }
1359 // this is necessary since not all demuxers allow to seek even back to the start
1360 demuxers.clear();
1361 if !transcoder.create_demuxers(&mut demuxers, &full_reg, false) {
1362 println!("failed to re-create demuxer(s)");
1363 return;
1364 }
1365 }
1366
6dc69f51
KS
1367 let mux_caps = mux_creator.get_capabilities();
1368 let mut out_sm = StreamManager::new();
4a835a2a 1369 if !transcoder.negotiate_stream_map(&ism, mux_caps, &mut out_sm, &full_reg.enc_reg) {
6dc69f51
KS
1370 println!("cannot determine stream map");
1371 return;
1372 }
1373
1374 let ret = File::create(transcoder.output_name.as_str());
1375 if ret.is_err() {
1376 println!("cannot open output file");
1377 return;
1378 }
1379 let mut fw = FileWriter::new_write(ret.unwrap());
1380 let mut bw = ByteWriter::new(&mut fw);
1381 let ret = create_muxer(mux_creator, out_sm, &mut bw);
1382 if let Err(err) = ret {
1383 println!("cannot create muxer instance {:?}", err);
1384 return;
1385 }
1386 let mut mux = ret.unwrap();
1387 parse_and_apply_options!(mux, &transcoder.mux_opts, "output");
1388
1389 println!("Output {} muxer {}", transcoder.output_name, ofmt);
1390 for ostr in mux.get_streams() {
1391 println!(" #{}: {} {}", ostr.get_num(), ostr, ostr.get_info().get_name());
1392 }
1393
5ec8115f
KS
1394 let mut time = Instant::now();
1395 let show_interval = Duration::from_millis(100);
908ba48c
KS
1396 let mut adata_size = 0;
1397 let mut vdata_size = 0;
4a835a2a 1398 let mut cur_dmx = 0;
951916e8 1399 'main_loop: loop {
4a835a2a
KS
1400 let mut pktres = Err(DemuxerError::EOF);
1401 let mut src_dmx = 0;
1402 loop {
1403 if !demuxers.iter().any(|(_, eof)| !eof) {
1404 break;
1405 }
1406 let mut got_res = false;
1407 if !demuxers[cur_dmx].1 {
1408 pktres = demuxers[cur_dmx].0.get_frame();
1409 got_res = true;
1410 src_dmx = cur_dmx;
1411 }
1412 cur_dmx += 1;
1413 if cur_dmx >= demuxers.len() {
1414 cur_dmx = 0;
1415 }
1416 if got_res {
1417 break;
1418 }
1419 }
1420
6dc69f51
KS
1421 if let Err(DemuxerError::EOF) = pktres { break; }
1422 if pktres.is_err() {
1423 println!("demuxing error");
1424 break;
1425 }
1426 let mut pkt = pktres.unwrap();
f88b8fb4 1427 if transcoder.start != NATimePoint::None && pkt.ts.less_than(transcoder.start) { continue; }
4a835a2a
KS
1428 let src_id = pkt.get_stream().get_num() + is_offset[src_dmx];
1429 let ts = pkt.ts;
1430 let newstream = ism.get_stream(src_id).unwrap();
1431 pkt.reassign(newstream, ts);
1432
908ba48c 1433 if transcoder.verbose > 0 && time.elapsed() >= show_interval {
5ec8115f
KS
1434 if let Some(pts) = pkt.get_pts() {
1435 let cur_time = format_time(NATimeInfo::ts_to_time(pts, 1000, pkt.ts.tb_num, pkt.ts.tb_den));
1436 print!(" {}", cur_time);
1437 } else {
1438 print!(" ???");
1439 }
1440 if !duration_string.is_empty() {
1441 print!(" / {}", duration_string);
1442 }
908ba48c
KS
1443 if transcoder.verbose > 1 {
1444 print!(" data sizes V: {} A: {}", vdata_size, adata_size);
1445 }
5ec8115f
KS
1446 print!("\r");
1447 std::io::stdout().flush().unwrap();
1448 time = Instant::now();
1449 }
6dc69f51
KS
1450 match transcoder.encoders[src_id] {
1451 OutputMode::Drop => {},
1452 OutputMode::Copy(dst_id) => {
1453 let dstr = mux.get_stream(dst_id as usize).unwrap();
1454 pkt.reassign(dstr, pkt.get_time_information());
951916e8 1455 if transcoder.end != NATimePoint::None && !pkt.ts.less_than(transcoder.end) { break 'main_loop; }
908ba48c
KS
1456 let pkt_size = pkt.get_buffer().len();
1457 match pkt.get_stream().get_media_type() {
1458 StreamType::Video => { vdata_size += pkt_size; },
1459 StreamType::Audio => { adata_size += pkt_size; },
1460 _ => {},
1461 };
6dc69f51
KS
1462 if mux.mux_frame(pkt).is_err() {
1463 println!("error muxing packet");
1464 break;
1465 }
1466 },
1467 OutputMode::Encode(dst_id, ref mut encoder, ref mut cvt) => {
d9fe2b71 1468 if let Some((ref mut dsupp, ref mut decoder, ref mut reorderer)) = transcoder.decoders[src_id] {
6dc69f51 1469 let ret = decoder.decode(dsupp, &pkt);
f88b8fb4
KS
1470 if let (true, Err(DecoderError::MissingReference)) = (transcoder.start != NATimePoint::None, &ret) {
1471 continue;
1472 }
6dc69f51
KS
1473 if ret.is_err() {
1474 println!("error decoding stream {}", src_id);
1475 break;
1476 }
1477 let frm = ret.unwrap();
b0481c9e 1478 let tinfo = frm.get_info();
d9fe2b71
KS
1479 reorderer.add_frame(frm);
1480 while let Some(frm) = reorderer.get_frame() {
df37d3b1 1481 if !encode_frame(dst_id, encoder, cvt, frm, &transcoder.scale_opts) {
d9fe2b71
KS
1482 break;
1483 }
1484 while let Ok(Some(pkt)) = encoder.get_packet() {
1485 if transcoder.end != NATimePoint::None && !pkt.ts.less_than(transcoder.end) { break 'main_loop; }
908ba48c
KS
1486 let pkt_size = pkt.get_buffer().len();
1487 match pkt.get_stream().get_media_type() {
1488 StreamType::Video => { vdata_size += pkt_size; },
1489 StreamType::Audio => { adata_size += pkt_size; },
1490 _ => {},
1491 };
d9fe2b71
KS
1492 mux.mux_frame(pkt).unwrap();
1493 }
6dc69f51 1494 }
b0481c9e
KS
1495 if let OutputConvert::Audio(ref mut acvt) = cvt {
1496 while let Some(ofrm) = acvt.get_frame(tinfo.clone()) {
1497 if encoder.encode(&ofrm).is_err() {
1498 break;
1499 }
1500 while let Ok(Some(pkt)) = encoder.get_packet() {
1501 if transcoder.end != NATimePoint::None && !pkt.ts.less_than(transcoder.end) { break 'main_loop; }
1502 let pkt_size = pkt.get_buffer().len();
1503 adata_size += pkt_size;
1504 mux.mux_frame(pkt).unwrap();
1505 }
1506 }
1507 }
6dc69f51
KS
1508 } else {
1509 println!("no decoder for stream {}", src_id);
1510 break;
1511 }
1512 },
1513 };
1514 }
4a835a2a 1515 'reord_flush_loop: for stream in ism.iter() {
fd03a232 1516 let src_id = stream.get_num();
d9fe2b71
KS
1517 if let OutputMode::Encode(dst_id, ref mut encoder, ref mut cvt) = transcoder.encoders[src_id] {
1518 if let Some((_, _, ref mut reorderer)) = transcoder.decoders[src_id] {
1519 while let Some(frm) = reorderer.get_last_frames() {
df37d3b1 1520 if !encode_frame(dst_id, encoder, cvt, frm, &transcoder.scale_opts) {
d9fe2b71
KS
1521 break;
1522 }
1523 while let Ok(Some(pkt)) = encoder.get_packet() {
1524 if transcoder.end != NATimePoint::None && !pkt.ts.less_than(transcoder.end) { break 'reord_flush_loop; }
1525 mux.mux_frame(pkt).unwrap();
1526 }
1527 }
1528 }
1529 }
1530 }
6dc69f51
KS
1531 'flush_loop: for enc in transcoder.encoders.iter_mut() {
1532 match enc {
1533 OutputMode::Encode(str_id, ref mut encoder, _) => {
1534 let ret = encoder.flush();
1535 if ret.is_err() {
1536 println!("error flushing encoder for stream {}", str_id);
1537 break;
1538 } else {
1539 while let Ok(Some(pkt)) = encoder.get_packet() {
1540 if mux.mux_frame(pkt).is_err() {
1541 println!("error muxing packet");
1542 break 'flush_loop;
1543 }
1544 }
1545 }
1546 },
1547 _ => {},
1548 };
1549 }
908ba48c 1550 if transcoder.verbose > 0 {
5ec8115f
KS
1551 println!();
1552 }
6dc69f51
KS
1553
1554 let ret = mux.end();
1555 if ret.is_err() {
1556 println!("error at finalising muxing");
1557 }
1558}