X-Git-Url: https://git.nihav.org/?p=nihav.git;a=blobdiff_plain;f=nihav-rad%2Fsrc%2Fdemuxers%2Fbink.rs;h=83da6acaebcb95643070ae21d027b969dc2ded1b;hp=d0ae143ac9b7d7e827a5105a95e4aa0b3723261d;hb=a480a0de101483d802a11e72d758dae00fa4860a;hpb=4cfb5dd496118a94b7b59b806f0b81a17e47630c diff --git a/nihav-rad/src/demuxers/bink.rs b/nihav-rad/src/demuxers/bink.rs index d0ae143..83da6ac 100644 --- a/nihav-rad/src/demuxers/bink.rs +++ b/nihav-rad/src/demuxers/bink.rs @@ -9,7 +9,7 @@ const BINK_AUD_FLAG_DCT: u8 = 0x10; const BINK_AUD_FLAG_STEREO: u8 = 0x20; impl AudioTrack { - fn new(strmgr: &mut StreamManager, srate: u32, flags: u8, str_id: usize, magic: &[u8; 4]) -> DemuxerResult { + fn new(strmgr: &mut StreamManager, srate: u32, flags: u8, str_id: usize, magic: [u8; 4]) -> DemuxerResult { let channels = if (flags & BINK_AUD_FLAG_STEREO) != 0 { 2 } else { 1 }; let codecname = if (flags & BINK_AUD_FLAG_DCT) != 0 { "bink-audio-dct" @@ -18,9 +18,9 @@ impl AudioTrack { }; let ahdr = NAAudioInfo::new(srate, channels, SND_F32P_FORMAT, 1); let mut edata: Vec = Vec::with_capacity(4); - edata.extend_from_slice(magic); + edata.extend_from_slice(&magic); let ainfo = NACodecInfo::new(codecname, NACodecTypeInfo::Audio(ahdr), Some(edata)); - let res = strmgr.add_stream(NAStream::new(StreamType::Audio, (str_id + 1) as u32, ainfo, 1, srate)); + let res = strmgr.add_stream(NAStream::new(StreamType::Audio, (str_id + 1) as u32, ainfo, 1, srate, 0)); validate!(res.is_some()); let id = res.unwrap(); Ok(Self{ id }) @@ -66,6 +66,8 @@ impl<'a> DemuxCore<'a> for BinkDemuxer<'a> { let tb_num = src.read_u32le()?; validate!((width > 0) && (height > 0) && (width <= 7680) && (height <= 4800)); validate!((self.frames > 0) && (tb_num > 0) && (tb_den > 0) && (max_size < fsize)); + self.tb_num = tb_num; + self.tb_den = tb_den; let mut flags: [u8; 4] = [0; 4]; src.read_buf(&mut flags)?; let mut edata: Vec = vec![0; 8]; @@ -76,7 +78,7 @@ impl<'a> DemuxCore<'a> for BinkDemuxer<'a> { let vhdr = NAVideoInfo::new(width, height, false, YUV420_FORMAT); let codec = if magic[0] == b'K' && magic[1] == b'B' && magic[2] == b'2' { "bink2-video" } else { "bink-video" }; let vinfo = NACodecInfo::new(codec, NACodecTypeInfo::Video(vhdr), Some(edata)); - let res = strmgr.add_stream(NAStream::new(StreamType::Video, 0, vinfo, self.tb_num, self.tb_den)); + let res = strmgr.add_stream(NAStream::new(StreamType::Video, 0, vinfo, self.tb_num, self.tb_den, self.frames as u64)); validate!(res.is_some()); self.video_id = res.unwrap(); @@ -91,7 +93,7 @@ impl<'a> DemuxCore<'a> for BinkDemuxer<'a> { let srate = src.read_u24le()?; let flags = src.read_byte()?; validate!(srate > 0); - self.ainfo.push(AudioTrack::new(strmgr, srate, flags, i, &magic)?); + self.ainfo.push(AudioTrack::new(strmgr, srate, flags, i, magic)?); } for _ in 0..num_audio { let _trk_id = src.read_u32le()?; @@ -112,7 +114,7 @@ impl<'a> DemuxCore<'a> for BinkDemuxer<'a> { seek_idx.seek_info[0].filled = true; self.cur_frame = 0; - + Ok(()) } fn get_frame(&mut self, strmgr: &mut StreamManager) -> DemuxerResult { @@ -150,7 +152,7 @@ impl<'a> DemuxCore<'a> for BinkDemuxer<'a> { Ok(pkt) } - fn seek(&mut self, time: u64, seek_idx: &SeekIndex) -> DemuxerResult<()> { + fn seek(&mut self, time: NATimePoint, seek_idx: &SeekIndex) -> DemuxerResult<()> { let ret = seek_idx.find_pos(time); if ret.is_none() { return Err(DemuxerError::SeekError); @@ -161,6 +163,13 @@ impl<'a> DemuxCore<'a> for BinkDemuxer<'a> { self.cur_frame = seek_info.pts as usize; Ok(()) } + fn get_duration(&self) -> u64 { 0 } +} + +impl<'a> NAOptionHandler for BinkDemuxer<'a> { + fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] } + fn set_options(&mut self, _options: &[NAOption]) { } + fn query_option_value(&self, _name: &str) -> Option { None } } impl<'a> BinkDemuxer<'a> { @@ -205,7 +214,8 @@ mod test { let mut br = ByteReader::new(&mut fr); let mut dmx = BinkDemuxer::new(&mut br); let mut sm = StreamManager::new(); - dmx.open(&mut sm).unwrap(); + let mut si = SeekIndex::new(); + dmx.open(&mut sm, &mut si).unwrap(); loop { let pktres = dmx.get_frame(&mut sm); if let Err(e) = pktres {