]> git.nihav.org Git - nihav-player.git/blame_incremental - videoplayer/src/main.rs
hwdec-vaapi: handle copying data into kodaed chroma planes
[nihav-player.git] / videoplayer / src / main.rs
... / ...
CommitLineData
1extern crate sdl2;
2extern crate nihav_core;
3extern crate nihav_registry;
4extern crate nihav_allstuff;
5
6use std::env;
7use std::fs::File;
8use std::io::Write;
9use std::path::Path;
10use std::time::{Duration, Instant};
11use std::thread;
12use std::sync::atomic::{AtomicU8, Ordering};
13
14use sdl2::event::{Event, WindowEvent};
15use sdl2::keyboard::{Keycode, Mod};
16use sdl2::mouse::{MouseButton, MouseWheelDirection};
17use sdl2::render::{Canvas, Texture, TextureCreator};
18use sdl2::pixels::PixelFormatEnum;
19use sdl2::video::{Window, WindowContext};
20
21use nihav_registry::detect;
22use nihav_core::frame::*;
23use nihav_core::io::byteio::{FileReader, ByteReader};
24use nihav_core::reorder::*;
25use nihav_core::codecs::*;
26use nihav_core::demuxers::*;
27use nihav_registry::register::*;
28use nihav_allstuff::*;
29
30#[cfg(feature="hwaccel")]
31use hwdec_vaapi::*;
32
33mod audiodec;
34use audiodec::*;
35mod videodec;
36use videodec::*;
37mod osd;
38use osd::*;
39
40#[repr(u8)]
41#[derive(Clone,Copy,Debug,PartialEq,Default)]
42enum DecodingState {
43 #[default]
44 Normal,
45 Waiting,
46 Flush,
47 Prefetch,
48 Error,
49 End,
50}
51
52impl From<u8> for DecodingState {
53 fn from(val: u8) -> Self {
54 match val {
55 0 => DecodingState::Normal,
56 1 => DecodingState::Waiting,
57 2 => DecodingState::Flush,
58 3 => DecodingState::Prefetch,
59 4 => DecodingState::End,
60 _ => DecodingState::Error,
61 }
62 }
63}
64
65struct DecoderState {
66 state: AtomicU8
67}
68
69impl DecoderState {
70 const fn new() -> Self {
71 Self {
72 state: AtomicU8::new(DecodingState::Normal as u8)
73 }
74 }
75 fn set_state(&self, state: DecodingState) {
76 self.state.store(state as u8, Ordering::Release);
77 }
78 fn get_state(&self) -> DecodingState {
79 self.state.load(Ordering::Acquire).into()
80 }
81 fn is_flushing(&self) -> bool {
82 matches!(self.get_state(), DecodingState::Flush | DecodingState::Error)
83 }
84}
85
86#[cfg(feature="debug")]
87macro_rules! debug_log {
88 ($log: expr; $blk: block) => {
89 $log.logfile.write($blk.as_bytes()).unwrap();
90 $log.logfile.write(b"\n").unwrap();
91 };
92}
93#[cfg(not(feature="debug"))]
94macro_rules! debug_log {
95 ($log: expr; $blk: block) => {};
96}
97
98enum ScaleSize {
99 Auto,
100 Times(f32),
101 Fixed(usize, usize)
102}
103
104impl FromStr for ScaleSize {
105 type Err = ();
106 fn from_str(s: &str) -> Result<Self, Self::Err> {
107 if matches!(s, "" | "auto") {
108 Ok(ScaleSize::Auto)
109 } else if s.ends_with('x') || s.ends_with('X') {
110 let factor = s[..s.len() - 1].parse::<f32>().map_err(|_| ())?;
111 if factor > 0.0 {
112 Ok(ScaleSize::Times(factor))
113 } else {
114 Err(())
115 }
116 } else if s.contains('x') | s.contains('X') {
117 let mut dims = if s.contains('x') { s.split('x') } else { s.split('X') };
118 let w = dims.next().unwrap();
119 let h = dims.next().unwrap();
120 let width = w.parse::<usize>().map_err(|_| ())?;
121 let height = h.parse::<usize>().map_err(|_| ())?;
122 if width > 0 && height > 0 {
123 Ok(ScaleSize::Fixed(width, height))
124 } else {
125 Err(())
126 }
127 } else {
128 Err(())
129 }
130 }
131}
132
133pub enum PktSendEvent {
134 Packet(NAPacket),
135 GetFrames,
136 Flush,
137 End,
138 ImmediateEnd,
139 HurryUp,
140}
141
142pub enum DecoderType {
143 Audio(Box<dyn NADecoder + Send>),
144 Video(Box<dyn NADecoder + Send>, Box<dyn FrameReorderer + Send>),
145 VideoMT(Box<dyn NADecoderMT + Send>, MTFrameReorderer),
146 #[cfg(feature="hwaccel")]
147 VideoHW(Box<dyn HWDecoder + Send>),
148}
149
150pub struct DecoderStuff {
151 pub dsupp: Box<NADecoderSupport>,
152 pub dec: DecoderType,
153}
154
155fn format_time(ms: u64) -> String {
156 let s = ms / 1000;
157 let ds = (ms % 1000) / 100;
158 let (min, s) = (s / 60, s % 60);
159 let (h, min) = (min / 60, min % 60);
160 if h == 0 {
161 if min == 0 {
162 format!("{}.{}", s, ds)
163 } else {
164 format!("{}:{:02}.{}", min, s, ds)
165 }
166 } else {
167 format!("{}:{:02}:{:02}.{}", h, min, s, ds)
168 }
169}
170
171const FRAME_QUEUE_LEN: usize = 25;
172const MAX_VOLUME: usize = 200;
173
174pub type FrameRecord = (NABufferType, u64);
175
176pub struct TimeKeep {
177 ref_time: Instant,
178 ref_ts: u64,
179}
180
181impl TimeKeep {
182 fn new() -> Self {
183 Self {
184 ref_time: Instant::now(),
185 ref_ts: 0,
186 }
187 }
188 pub fn get_cur_time(&self) -> u64 {
189 let add = self.ref_time.elapsed().as_millis() as u64;
190 self.ref_ts + add
191 }
192 fn reset_ts(&mut self) {
193 self.ref_ts = 0;
194 }
195 fn reset_all(&mut self, ts: u64) {
196 self.ref_time = Instant::now();
197 self.ref_ts = ts;
198 }
199 fn set_ts(&mut self) {
200 self.ref_ts = self.get_cur_time();
201 }
202 fn set_time(&mut self) {
203 self.ref_time = Instant::now();
204 }
205}
206
207pub struct DispFrame<'a> {
208 pub ts: u64,
209 pub is_yuv: bool,
210 pub valid: bool,
211 pub rgb_tex: Texture<'a>,
212 pub yuv_tex: Texture<'a>,
213}
214
215pub struct DispQueue<'a> {
216 pub pool: Vec<DispFrame<'a>>,
217 pub first_ts: u64,
218 pub last_ts: u64,
219 pub start: usize,
220 pub end: usize,
221 pub len: usize,
222 pub width: usize,
223 pub height: usize,
224}
225
226impl<'a> DispQueue<'a> {
227 fn new(texture_creator: &'a TextureCreator<WindowContext>, width: usize, height: usize, len: usize) -> Self {
228 let mut pool = Vec::with_capacity(len);
229 for _ in 0..len + 1 {
230 let rgb_tex = texture_creator.create_texture_streaming(PixelFormatEnum::RGB24, width as u32, height as u32).expect("failed to create RGB texture");
231 let yuv_tex = texture_creator.create_texture_streaming(PixelFormatEnum::IYUV, ((width + 1) & !1) as u32, ((height + 1) & !1) as u32).expect("failed to create YUV texture");
232 pool.push(DispFrame{ ts: 0, is_yuv: false, valid: false, rgb_tex, yuv_tex });
233 }
234 pool[len].is_yuv = false;
235 pool[len].rgb_tex.with_lock(None, |buffer: &mut [u8], _pitch: usize| {
236 for el in buffer.iter_mut() { *el = 0; }
237 }).expect("RGB texture could not be locked");
238
239 Self { pool, first_ts: 0, last_ts: 0, start: 0, end: 0, len, width, height }
240 }
241
242 fn flush(&mut self) {
243 self.start = 0;
244 self.end = 0;
245 self.first_ts = 0;
246 self.last_ts = 0;
247 for frm in self.pool.iter_mut() {
248 frm.valid = false;
249 }
250 }
251
252 fn get_last_texture(&mut self, osd: &OSD) -> &Texture<'a> {
253 if self.pool[self.len].is_yuv {
254 if osd.is_active() {
255 self.pool[self.len].yuv_tex.with_lock(None, |buffer: &mut [u8], pitch: usize| {
256 osd.draw_yuv(buffer, pitch);
257 }).expect("YUV texture locking failure");
258 }
259 &self.pool[self.len].yuv_tex
260 } else {
261 if osd.is_active() {
262 self.pool[self.len].rgb_tex.with_lock(None, |buffer: &mut [u8], pitch: usize| {
263 osd.draw_rgb(buffer, pitch);
264 }).expect("RGB texture locking failure");
265 }
266 &self.pool[self.len].rgb_tex
267 }
268 }
269 pub fn is_empty(&self) -> bool { self.start == self.end }
270 pub fn is_full(&self) -> bool { self.len == 0 || self.start == (self.end + 1) % self.len }
271 pub fn move_end(&mut self) {
272 self.end += 1;
273 if self.end >= self.len {
274 self.end -= self.len;
275 }
276 }
277 pub fn move_start(&mut self) {
278 self.pool.swap(self.start, self.len);
279 self.start += 1;
280 if self.start >= self.len {
281 self.start -= self.len;
282 }
283 if !self.is_empty() {
284 self.first_ts = self.pool[self.start].ts;
285 }
286 }
287}
288
289fn try_display(disp_queue: &mut DispQueue, canvas: &mut Canvas<Window>, osd: &mut OSD, ctime: &TimeKeep) -> Option<u64> {
290 while !disp_queue.is_empty() {
291 let disp_time = disp_queue.first_ts;
292 let ctime = ctime.get_cur_time();
293 if disp_time > ctime + 10 {
294 return Some(disp_time - ctime);
295 } else if disp_time + 10 < ctime {
296 disp_queue.move_start();
297 } else {
298 if osd.is_active() {
299 osd.prepare(ctime);
300 }
301 let frm = &mut disp_queue.pool[disp_queue.start];
302 let texture = if frm.is_yuv {
303 if osd.is_active() {
304 frm.yuv_tex.with_lock(None, |buffer: &mut [u8], pitch: usize| {
305 osd.draw_yuv(buffer, pitch);
306 }).expect("YUV texture locking failure");
307 }
308 &frm.yuv_tex
309 } else {
310 if osd.is_active() {
311 frm.rgb_tex.with_lock(None, |buffer: &mut [u8], pitch: usize| {
312 osd.draw_rgb(buffer, pitch);
313 }).expect("RGB texture locking failure");
314 }
315 &frm.rgb_tex
316 };
317 canvas.clear();
318 canvas.copy(texture, None, None).expect("canvas blit failure");
319 canvas.present();
320
321 disp_queue.move_start();
322 if !disp_queue.is_empty() {
323 return Some(disp_queue.first_ts.saturating_sub(ctime).saturating_sub(2));
324 } else {
325 return None;
326 }
327 }
328 }
329 None
330}
331
332struct Player {
333 sdl_context: sdl2::Sdl,
334 vsystem: sdl2::VideoSubsystem,
335 asystem: sdl2::AudioSubsystem,
336
337 acontrol: AudioControl,
338 vcontrol: VideoControl,
339
340 play_video: bool,
341 play_audio: bool,
342 has_video: bool,
343 has_audio: bool,
344 video_str: u32,
345 audio_str: u32,
346 sc_size: ScaleSize,
347
348 vthreads: usize,
349 use_mt: bool,
350 #[cfg(feature="hwaccel")]
351 use_hwaccel: bool,
352
353 paused: bool,
354 mute: bool,
355 volume: usize,
356 end: bool,
357
358 tkeep: TimeKeep,
359
360 debug: bool,
361 osd: OSD,
362
363 #[cfg(feature="debug")]
364 logfile: File,
365}
366
367impl Player {
368 fn new() -> Self {
369 let sdl_context = sdl2::init().expect("SDL2 init failure");
370 let vsystem = sdl_context.video().expect("video subsystem init failure");
371 let asystem = sdl_context.audio().expect("audio subsystem init failure");
372 vsystem.disable_screen_saver();
373 let acontrol = AudioControl::new(None, None, false, &asystem);
374 let vcontrol = VideoControl::new(None, 0, 0, 0, 0);
375 Self {
376 sdl_context, asystem, vsystem,
377
378 acontrol, vcontrol,
379
380 play_video: true,
381 play_audio: true,
382 has_video: false,
383 has_audio: false,
384 video_str: 0,
385 audio_str: 0,
386 sc_size: ScaleSize::Auto,
387
388 vthreads: 3,
389 use_mt: true,
390 #[cfg(feature="hwaccel")]
391 use_hwaccel: true,
392
393 paused: false,
394 mute: false,
395 volume: 100,
396 end: false,
397
398 tkeep: TimeKeep::new(),
399
400 debug: false,
401 osd: OSD::new(),
402
403 #[cfg(feature="debug")]
404 logfile: File::create("debug.log").expect("'debug.log' should be available for writing"),
405 }
406 }
407 fn seek(&mut self, off: u64, fwd: bool, dmx: &mut Demuxer, disp_queue: &mut DispQueue) -> Result<(), ()> {
408 let cur_time = self.tkeep.get_cur_time();
409 let seektime = if fwd { cur_time + off * 1000 } else {
410 cur_time.saturating_sub(off * 1000) };
411 debug_log!(self; {format!(" seek to {}", seektime)});
412
413 let ret = dmx.seek(NATimePoint::Milliseconds(seektime));
414 if ret.is_err() {
415 println!(" seek error");
416 return Ok(()); //TODO: not ignore some of seek errors?
417 }
418
419 self.acontrol.flush();
420 self.vcontrol.flush();
421 disp_queue.flush();
422
423 self.tkeep.reset_ts();
424 self.prefill(dmx, disp_queue)?;
425 if !disp_queue.is_empty() {
426 self.tkeep.reset_all(disp_queue.first_ts);
427 } else {
428 let mut iterations = 0;
429 let mut time = self.acontrol.get_time();
430 while time.is_none() {
431 iterations += 1;
432 std::thread::yield_now();
433 if iterations > 1000000 { println!(" still no time set?!"); break; }
434 time = self.acontrol.get_time();
435 }
436 if let Some(time) = time {
437 self.tkeep.reset_all(time);
438 }
439 }
440 if !self.paused {
441 self.acontrol.resume();
442 }
443 Ok(())
444 }
445 fn prefill(&mut self, dmx: &mut Demuxer, disp_queue: &mut DispQueue) -> Result<(), ()> {
446 debug_log!(self; {" prefilling"});
447 while self.vcontrol.get_queue_size() < FRAME_QUEUE_LEN {
448 let mut try_send = self.acontrol.get_queue_size() < FRAME_QUEUE_LEN && (!self.has_video || (!self.vcontrol.is_filled(FRAME_QUEUE_LEN) && !disp_queue.is_full()));
449
450 if !self.vcontrol.try_send_queued() && self.vcontrol.get_queue_size() > FRAME_QUEUE_LEN / 2 {
451 try_send = false;
452 }
453 if !self.acontrol.try_send_queued() && self.acontrol.get_queue_size() > FRAME_QUEUE_LEN / 2 {
454 try_send = false;
455 }
456 if try_send {
457 match dmx.get_frame() {
458 Err(DemuxerError::EOF) => break,
459 Err(_) => break,
460 Ok(pkt) => {
461 let streamno = pkt.get_stream().get_id();
462 if self.has_video && streamno == self.video_str {
463 self.vcontrol.try_send_video(PktSendEvent::Packet(pkt));
464 } else if self.has_audio && streamno == self.audio_str {
465 self.acontrol.try_send_audio(PktSendEvent::Packet(pkt));
466 }
467 }
468 };
469 }
470 self.vcontrol.fill(disp_queue);
471
472 if !try_send {
473 break;
474 }
475 }
476 if self.has_video {
477 while self.vcontrol.get_queue_size() > 0 && !disp_queue.is_full() {
478 self.vcontrol.try_send_queued();
479 self.vcontrol.fill(disp_queue);
480 std::thread::sleep(Duration::from_millis(10));
481 }
482 self.vcontrol.wait_for_frames()?;
483 self.vcontrol.fill(disp_queue);
484 }
485 debug_log!(self; {format!(" prefilling done, frames {}-{} audio {}", disp_queue.start, disp_queue.end, self.acontrol.get_fill())});
486 Ok(())
487 }
488 fn toggle_pause(&mut self) {
489 self.paused = !self.paused;
490 if self.paused {
491 self.vsystem.enable_screen_saver();
492 self.tkeep.set_ts();
493 } else {
494 self.vsystem.disable_screen_saver();
495 self.tkeep.set_time();
496 }
497 if self.paused {
498 self.acontrol.pause();
499 } else {
500 self.acontrol.resume();
501 }
502 }
503 fn handle_events(&mut self, event_pump: &mut sdl2::EventPump, canvas: &mut Canvas<Window>, dmx: &mut Demuxer, disp_queue: &mut DispQueue) -> Result<bool, ()> {
504 for event in event_pump.poll_iter() {
505 if let Event::Quit {..} = event {
506 self.end = true;
507 println!();
508 return Ok(true);
509 }
510 if let Event::Window {win_event: WindowEvent::Exposed, ..} = event {
511 canvas.clear();
512 canvas.copy(disp_queue.get_last_texture(&self.osd), None, None).expect("blitting failure");
513 canvas.present();
514 }
515 if let Event::MouseButtonDown {mouse_btn, ..} = event {
516 match mouse_btn {
517 MouseButton::Right => self.toggle_pause(),
518 MouseButton::Middle => self.osd.toggle(),
519 _ => {},
520 };
521 }
522 if let Event::MouseWheel {direction: MouseWheelDirection::Normal, x: 0, y, ..} = event {
523 self.seek(10, y > 0, dmx, disp_queue)?;
524 }
525 if let Event::KeyDown {keycode: Some(keycode), keymod, ..} = event {
526 match keycode {
527 Keycode::Escape => {
528 self.end = true;
529 println!();
530 return Ok(true);
531 },
532 Keycode::Q if keymod.contains(Mod::RSHIFTMOD) || keymod.contains(Mod::LSHIFTMOD) => {
533 self.end = true;
534 println!();
535 return Ok(true);
536 },
537 Keycode::Return | Keycode::KpEnter => return Ok(true),
538 Keycode::R => { self.seek(0, true, dmx, disp_queue)?; },
539 Keycode::Right | Keycode::Kp6 => { self.seek(10, true, dmx, disp_queue)?; },
540 Keycode::Left | Keycode::Kp4 => { self.seek(10, false, dmx, disp_queue)?; },
541 Keycode::Up | Keycode::Kp8 => { self.seek(60, true, dmx, disp_queue)?; },
542 Keycode::Down | Keycode::Kp2 => { self.seek(60, false, dmx, disp_queue)?; },
543 Keycode::PageUp | Keycode::Kp9 => { self.seek(600, true, dmx, disp_queue)?; },
544 Keycode::PageDown | Keycode::Kp3 => { self.seek(600, false, dmx, disp_queue)?; },
545 Keycode::Space => { self.toggle_pause(); },
546 Keycode::Plus | Keycode::KpPlus => {
547 self.volume = (self.volume + 10).min(MAX_VOLUME);
548 if !self.mute {
549 self.acontrol.set_volume(self.volume);
550 }
551 },
552 Keycode::Equals if keymod.contains(Mod::RSHIFTMOD) || keymod.contains(Mod::LSHIFTMOD) => {
553 self.volume = (self.volume + 10).min(MAX_VOLUME);
554 if !self.mute {
555 self.acontrol.set_volume(self.volume);
556 }
557 },
558 Keycode::Minus | Keycode::KpMinus => {
559 self.volume = self.volume.saturating_sub(10);
560 if !self.mute {
561 self.acontrol.set_volume(self.volume);
562 }
563 },
564 Keycode::D => {
565 self.debug = !self.debug;
566 },
567 Keycode::M => {
568 self.mute = !self.mute;
569 if self.mute {
570 self.acontrol.set_volume(0);
571 } else {
572 self.acontrol.set_volume(self.volume);
573 }
574 },
575 Keycode::H => {
576 self.vcontrol.try_send_video(PktSendEvent::HurryUp);
577 },
578 Keycode::O => {
579 if keymod.contains(Mod::RSHIFTMOD) || keymod.contains(Mod::LSHIFTMOD) {
580 self.osd.toggle_perm();
581 } else {
582 self.osd.toggle();
583 }
584 },
585 _ => {},
586 };
587 if !self.paused {
588 print!("{:60}\r", ' ');
589 std::io::stdout().flush().unwrap();
590 }
591 }
592 }
593 Ok(false)
594 }
595 fn play(&mut self, mut window: Window, name: &str, start_time: NATimePoint) -> Window {
596 debug_log!(self; {format!("Playing {}", name)});
597
598 // prepare data source
599 let path = Path::new(name);
600 let mut file = if let Ok(handle) = File::open(path) {
601 if let Ok(meta) = handle.metadata() {
602 if meta.is_dir() {
603 return window;
604 }
605 }
606 handle
607 } else {
608 println!("failed to open {}", name);
609 return window;
610 };
611 let mut fr = FileReader::new_read(&mut file);
612 let mut br = ByteReader::new(&mut fr);
613 let res = detect::detect_format(name, &mut br);
614 if res.is_none() {
615 println!("cannot detect format for {}", name);
616 return window;
617 }
618 let (dmx_name, _score) = res.unwrap();
619 debug_log!(self; {format!(" found demuxer {} with score {:?}", dmx_name, _score)});
620 println!("trying demuxer {} on {}", dmx_name, name);
621
622 let mut dmx_reg = RegisteredDemuxers::new();
623 nihav_register_all_demuxers(&mut dmx_reg);
624 let mut dec_reg = RegisteredDecoders::new();
625 nihav_register_all_decoders(&mut dec_reg);
626 let mut mtdec_reg = RegisteredMTDecoders::new();
627 if self.use_mt {
628 nihav_register_all_mt_decoders(&mut mtdec_reg);
629 }
630
631 let ret = dmx_reg.find_demuxer(dmx_name);
632 if ret.is_none() {
633 println!("error finding {} demuxer", dmx_name);
634 return window;
635 }
636 let dmx_fact = ret.unwrap();
637 br.seek(SeekFrom::Start(0)).expect("should be able to seek to the start");
638 let ret = create_demuxer(dmx_fact, &mut br);
639 if ret.is_err() {
640 println!("error creating demuxer");
641 return window;
642 }
643 let mut dmx = ret.unwrap();
644 if start_time != NATimePoint::None {
645 debug_log!(self; {format!(" start seek to {}", start_time)});
646 if dmx.seek(start_time).is_err() {
647 println!("initial seek failed");
648 }
649 }
650
651 let mut width = 640;
652 let mut height = 480;
653 let mut tb_num = 0;
654 let mut tb_den = 0;
655 let mut ainfo: Option<NAAudioInfo> = None;
656 let mut sbr_hack = false;
657
658 let mut video_dec: Option<DecoderStuff> = None;
659 let mut audio_dec: Option<DecoderStuff> = None;
660
661 let duration = dmx.get_duration();
662 if duration != 0 {
663 println!(" total duration {}", format_time(duration));
664 }
665 self.has_video = false;
666 self.has_audio = false;
667 self.osd.reset();
668 self.osd.set_duration(duration);
669 for i in 0..dmx.get_num_streams() {
670 let s = dmx.get_stream(i).unwrap();
671 let info = s.get_info();
672 let decfunc = dec_reg.find_decoder(info.get_name());
673 let decfunc_mt = mtdec_reg.find_decoder(info.get_name());
674 println!("stream {} - {} {}", i, s, info.get_name());
675 debug_log!(self; {format!(" stream {} - {} {}", i, s, info.get_name())});
676 let str_id = s.get_id();
677 if info.is_video() {
678 if video_dec.is_none() && self.play_video {
679 #[cfg(feature="hwaccel")]
680 if info.get_name() == "h264" && self.use_hwaccel {
681 let mut dec = new_h264_hwdec();
682 let dsupp = Box::new(NADecoderSupport::new());
683 let props = info.get_properties().get_video_info().unwrap();
684 if props.get_width() != 0 {
685 width = props.get_width();
686 height = props.get_height();
687 }
688 if dec.init(info.clone()).is_err() {
689 println!("failed to initialise hwaccel video decoder");
690 } else {
691 video_dec = Some(DecoderStuff{ dsupp, dec: DecoderType::VideoHW(dec) });
692 self.video_str = str_id;
693 let (tbn, tbd) = s.get_timebase();
694 tb_num = tbn;
695 tb_den = tbd;
696 self.has_video = true;
697 println!(" using hardware-accelerated decoding");
698 continue;
699 }
700 }
701 if let Some(decfunc) = decfunc_mt {
702 let mut dec = (decfunc)();
703 let mut dsupp = Box::new(NADecoderSupport::new());
704 let props = info.get_properties().get_video_info().unwrap();
705 if props.get_width() != 0 {
706 width = props.get_width();
707 height = props.get_height();
708 }
709 if dec.init(&mut dsupp, info.clone(), self.vthreads).is_ok() {
710 video_dec = Some(DecoderStuff{ dsupp, dec: DecoderType::VideoMT(dec, MTFrameReorderer::new()) });
711 self.video_str = str_id;
712 let (tbn, tbd) = s.get_timebase();
713 tb_num = tbn;
714 tb_den = tbd;
715 self.has_video = true;
716 continue;
717 } else {
718 println!("failed to create multi-threaded decoder, falling back");
719 }
720 }
721 if let Some(decfunc) = decfunc {
722 let mut dec = (decfunc)();
723 let mut dsupp = Box::new(NADecoderSupport::new());
724 let props = info.get_properties().get_video_info().unwrap();
725 if props.get_width() != 0 {
726 width = props.get_width();
727 height = props.get_height();
728 }
729 let desc = get_codec_description(info.get_name());
730 let (reorder_depth, reord) = if desc.is_none() || (desc.unwrap().caps & CODEC_CAP_COMPLEX_REORDER) == 0 {
731 let reord: Box<dyn FrameReorderer + Send> = Box::new(IPBReorderer::new());
732 (3, reord)
733 } else {
734 let reord: Box<dyn FrameReorderer + Send> = Box::new(ComplexReorderer::new());
735 (16, reord)
736 };
737 dsupp.pool_u8 = NAVideoBufferPool::new(reorder_depth);
738 dsupp.pool_u16 = NAVideoBufferPool::new(reorder_depth);
739 dsupp.pool_u32 = NAVideoBufferPool::new(reorder_depth);
740 if dec.init(&mut dsupp, info).is_err() {
741 println!("failed to initialise video decoder");
742 return window;
743 }
744 video_dec = Some(DecoderStuff{ dsupp, dec: DecoderType::Video(dec, reord) });
745 self.video_str = str_id;
746 let (tbn, tbd) = s.get_timebase();
747 tb_num = tbn;
748 tb_den = tbd;
749 self.has_video = true;
750 } else {
751 println!("no video decoder for {} found!", info.get_name());
752 }
753 }
754 } else if info.is_audio() {
755 if audio_dec.is_none() && self.play_audio {
756 if let Some(decfunc) = decfunc {
757 let mut dec = (decfunc)();
758 let mut dsupp = Box::new(NADecoderSupport::new());
759 ainfo = info.get_properties().get_audio_info();
760 if let (true, Some(ref ai)) = (info.get_name() == "aac", ainfo) {
761 if ai.sample_rate < 32000 {
762 sbr_hack = true;
763 }
764 }
765 if dec.init(&mut dsupp, info).is_err() {
766 println!("failed to initialise audio decoder");
767 return window;
768 }
769 audio_dec = Some(DecoderStuff{ dsupp, dec: DecoderType::Audio(dec) });
770 self.audio_str = str_id;
771 self.has_audio = true;
772 } else {
773 println!("no audio decoder for {} found!", info.get_name());
774 }
775 }
776 } else {
777 println!("decoder {} not found", info.get_name());
778 }
779 }
780 if !self.has_video && !self.has_audio {
781 println!("No playable streams found.");
782 return window;
783 }
784
785 match self.sc_size {
786 ScaleSize::Auto => {
787 while (width <= 384) && (height <= 288) {
788 width <<= 1;
789 height <<= 1;
790 }
791 },
792 ScaleSize::Times(factor) => {
793 let nw = ((width as f32) * factor).ceil() as usize;
794 let nh = ((height as f32) * factor).ceil() as usize;
795 if nw > 0 && nh > 0 {
796 width = nw;
797 height = nh;
798 }
799 },
800 ScaleSize::Fixed(w, h) => {
801 width = w;
802 height = h;
803 },
804 };
805
806 // prepare playback structure
807 let mut new_vcontrol = VideoControl::new(video_dec, width, height, tb_num, tb_den);
808 std::mem::swap(&mut self.vcontrol, &mut new_vcontrol);
809
810 let mut new_acontrol = AudioControl::new(audio_dec, ainfo, sbr_hack, &self.asystem);
811 std::mem::swap(&mut self.acontrol, &mut new_acontrol);
812
813 if self.mute {
814 self.acontrol.set_volume(0);
815 } else {
816 self.acontrol.set_volume(self.volume);
817 }
818
819 let fname = path.file_name();
820 let wname = if let Some(fname) = fname {
821 // workaround for libSDL2 workaround for non-UTF8 windowing systems
822 // see https://github.com/libsdl-org/SDL/pull/4290 for detais
823 let nname = fname.to_str().expect("should be able to set window title").replace('\u{2013}', "-").replace('\u{2014}', "-");
824 "NihAV player - ".to_owned() + &nname
825 } else {
826 "NihAV player".to_owned()
827 };
828 window.set_title(&wname).expect("set window title");
829 if window.size() != (width as u32, height as u32) {
830 window.set_size(width as u32, height as u32).expect("resize window");
831 }
832 window.show();
833 let mut canvas = window.into_canvas().build().expect("should be able to build canvas");
834 let texture_creator = canvas.texture_creator();
835 let mut disp_q = DispQueue::new(&texture_creator, width, height, if self.has_video { FRAME_QUEUE_LEN } else { 0 });
836 if !self.has_video {
837 canvas.clear();
838 canvas.copy(disp_q.get_last_texture(&self.osd), None, None).expect("blit failure");
839 canvas.present();
840 }
841
842 self.has_audio = self.acontrol.has_audio();
843 if !self.has_video && !self.has_audio {
844 println!("No playable streams.");
845 return canvas.into_window();
846 }
847
848 // play
849 if self.prefill(&mut dmx, &mut disp_q).is_err() {
850 std::mem::swap(&mut self.vcontrol, &mut new_vcontrol);
851 new_vcontrol.finish();
852 std::mem::swap(&mut self.acontrol, &mut new_acontrol);
853 new_acontrol.finish();
854 return canvas.into_window();
855 }
856 self.tkeep.reset_all(if !disp_q.is_empty() { disp_q.first_ts } else { 0 });
857 if !self.paused {
858 self.acontrol.resume();
859 }
860 let mut event_pump = self.sdl_context.event_pump().expect("should be able to create event pump");
861 let mut last_disp = Instant::now();
862 let mut has_data = true;
863 'main: loop {
864 let ret = self.handle_events(&mut event_pump, &mut canvas, &mut dmx, &mut disp_q);
865 if matches!(ret, Ok(true) | Err(_)) {
866 println!();
867 break 'main;
868 }
869 if !self.paused {
870 let mut try_send = self.acontrol.get_queue_size() < FRAME_QUEUE_LEN && self.vcontrol.get_queue_size() < FRAME_QUEUE_LEN;
871 if !self.vcontrol.try_send_queued() && self.vcontrol.is_filled(FRAME_QUEUE_LEN) {
872 try_send = false;
873 }
874 if !self.acontrol.try_send_queued() {
875 try_send = false;
876 }
877 while has_data && try_send {
878 match dmx.get_frame() {
879 Err(DemuxerError::EOF) => {
880 self.vcontrol.try_send_video(PktSendEvent::End);
881 self.acontrol.try_send_audio(PktSendEvent::End);
882 has_data = false;
883 },
884 Err(err) => {
885 println!("demuxer error {:?}", err);
886 if err == DemuxerError::IOError {
887 self.vcontrol.try_send_video(PktSendEvent::End);
888 self.acontrol.try_send_audio(PktSendEvent::End);
889 has_data = false;
890 }
891 },
892 Ok(pkt) => {
893 let streamno = pkt.get_stream().get_id();
894 if self.has_video && streamno == self.video_str {
895 debug_log!(self; {" sending video packet"});
896 self.vcontrol.try_send_video(PktSendEvent::Packet(pkt));
897 if self.vcontrol.is_filled(FRAME_QUEUE_LEN) {
898 try_send = false;
899 }
900 } else if self.has_audio && streamno == self.audio_str {
901 debug_log!(self; {" sending audio packet"});
902 self.acontrol.try_send_audio(PktSendEvent::Packet(pkt));
903 if self.acontrol.get_queue_size() >= FRAME_QUEUE_LEN {
904 try_send = false;
905 }
906 }
907 }
908 };
909 }
910 self.vcontrol.fill(&mut disp_q);
911 let mut sleep_time = 25;
912 debug_log!(self; {format!(" time {}", self.tkeep.get_cur_time())});
913 if self.has_video {
914 debug_log!(self; {format!(" disp queue {}-{}, {}-{} vqueue fill {}", disp_q.first_ts, disp_q.last_ts, disp_q.start, disp_q.end, self.vcontrol.get_queue_size())});
915 let ret = try_display(&mut disp_q, &mut canvas, &mut self.osd, &self.tkeep);
916 if let Some(next_time) = ret {
917 sleep_time = sleep_time.min(next_time);
918 }
919 }
920 if self.has_audio {
921 let time_left = self.acontrol.get_time_left();
922 debug_log!(self; {format!(" audio left {}", time_left)});
923 sleep_time = sleep_time.min(time_left);
924 }
925 debug_log!(self; {format!(" sleep {}ms", sleep_time)});
926 if last_disp.elapsed().as_millis() >= 10 {
927 let c_time = self.tkeep.get_cur_time();
928
929 if !self.debug {
930 print!(" {} {}% \r", format_time(c_time), self.acontrol.get_volume());
931 } else {
932 print!(" {} {} {}% {:3} {:6}\r", format_time(c_time), if self.vcontrol.is_yuv() { 'Y' } else { 'R' }, self.acontrol.get_volume(), (disp_q.end + disp_q.len - disp_q.start) % disp_q.len, self.acontrol.get_fill());
933 }
934 std::io::stdout().flush().unwrap();
935 last_disp = Instant::now();
936 }
937 let mut end = true;
938 if self.has_video && !self.vcontrol.is_video_end() {
939 end = false;
940 }
941 if self.has_audio && !self.acontrol.is_audio_end() {
942 end = false;
943 }
944 if end {
945 break;
946 }
947 thread::sleep(Duration::from_millis(sleep_time));
948 } else {
949 thread::sleep(Duration::from_millis(20));
950 }
951 }
952 println!();
953 std::mem::swap(&mut self.vcontrol, &mut new_vcontrol);
954 new_vcontrol.finish();
955 std::mem::swap(&mut self.acontrol, &mut new_acontrol);
956 new_acontrol.finish();
957 canvas.into_window()
958 }
959}
960
961fn main() {
962 let args: Vec<String> = env::args().collect();
963
964 if args.len() == 1 {
965 println!("usage: nihav-player file1 file2 ...");
966 return;
967 }
968
969 let mut player = Player::new();
970 let mut builder = player.vsystem.window("NihAV Player", 640, 480);
971 let mut window = builder.position_centered().hidden().build().expect("should be able to centre window");
972
973 let mut aiter = args.iter().skip(1);
974 let mut seek_time = NATimePoint::None;
975 while let Some(arg) = aiter.next() {
976 match arg.as_str() {
977 "-an" => { player.play_audio = false; },
978 "-ae" => { player.play_audio = true; },
979 "-vn" => { player.play_video = false; },
980 "-ve" => { player.play_video = true; },
981 "-seek" | "-start" => {
982 if let Some(arg) = aiter.next() {
983 if let Ok(time) = arg.parse::<NATimePoint>() {
984 seek_time = time;
985 } else {
986 println!("wrong seek time");
987 seek_time = NATimePoint::None;
988 }
989 }
990 },
991 "-vol" => {
992 if let Some(arg) = aiter.next() {
993 if let Ok(vol) = arg.parse::<usize>() {
994 player.volume = vol.min(MAX_VOLUME);
995 } else {
996 println!("wrong volume");
997 }
998 }
999 },
1000 "-debug" => {
1001 player.debug = true;
1002 },
1003 "-nodebug" => {
1004 player.debug = false;
1005 },
1006 "-mt" => {
1007 player.use_mt = true;
1008 },
1009 "-nomt" => {
1010 player.use_mt = false;
1011 },
1012 #[cfg(feature="hwaccel")]
1013 "-hwaccel" => {
1014 player.use_hwaccel = true;
1015 },
1016 #[cfg(feature="hwaccel")]
1017 "-nohwaccel" => {
1018 player.use_hwaccel = false;
1019 },
1020 "-threads" => {
1021 if let Some(arg) = aiter.next() {
1022 if let Ok(val) = arg.parse::<usize>() {
1023 player.vthreads = val.max(1);
1024 } else {
1025 println!("wrong number of threads");
1026 }
1027 }
1028 },
1029 "-scale" => {
1030 if let Some(arg) = aiter.next() {
1031 if let Ok(ssize) = arg.parse::<ScaleSize>() {
1032 player.sc_size = ssize;
1033 } else {
1034 println!("invalid scale size");
1035 }
1036 }
1037 },
1038 _ => {
1039 window = player.play(window, arg, seek_time);
1040 if player.end { break; }
1041 seek_time = NATimePoint::None;
1042 },
1043 };
1044 }
1045}