X-Git-Url: https://git.nihav.org/?a=blobdiff_plain;ds=sidebyside;f=src%2Fcodecs%2Fpcm.rs;h=288930b83b46ec807d5e05f65e4af9344e16a94f;hb=5a3ac1b7ad4ea52f43b0464ecedffdad9db643fe;hp=c5c3b5b1e118919dded9a1fe3e8c82db903586d9;hpb=3234da61c291704c7e9d928c4c50dd41f537e0c5;p=nihav.git diff --git a/src/codecs/pcm.rs b/src/codecs/pcm.rs index c5c3b5b..288930b 100644 --- a/src/codecs/pcm.rs +++ b/src/codecs/pcm.rs @@ -22,21 +22,19 @@ fn get_default_chmap(nch: u8) -> NAChannelMap { chmap } -fn get_duration(ainfo: &NAAudioInfo, duration: Option, data_size: usize) -> usize { -println!("pcm in {:?}, {}", duration, data_size); +fn get_duration(ainfo: &NAAudioInfo, duration: Option, data_size: usize) -> u64 { if duration == None { - let size_bits = data_size * 8; - let blk_size = (ainfo.get_channels() as usize) * (ainfo.get_format().get_bits() as usize); + let size_bits = (data_size as u64) * 8; + let blk_size = (ainfo.get_channels() as u64) * (ainfo.get_format().get_bits() as u64); size_bits / blk_size } else { - duration.unwrap() as usize + duration.unwrap() as u64 } } impl NADecoder for PCMDecoder { fn init(&mut self, info: Rc) -> DecoderResult<()> { if let NACodecTypeInfo::Audio(ainfo) = info.get_properties() { -println!("got info {}", ainfo); self.chmap = get_default_chmap(ainfo.get_channels()); if self.chmap.num_channels() == 0 { return Err(DecoderError::InvalidData); } Ok(()) @@ -48,12 +46,12 @@ println!("got info {}", ainfo); let info = pkt.get_stream().get_info(); if let NACodecTypeInfo::Audio(ainfo) = info.get_properties() { let duration = get_duration(&ainfo, pkt.get_duration(), pkt.get_buffer().len()); -println!("duration = {}", duration); let pktbuf = pkt.get_buffer(); let mut buf: Vec = Vec::with_capacity(pktbuf.len()); buf.clone_from(&pktbuf); let abuf = NAAudioBuffer::new_from_buf(ainfo, Rc::new(RefCell::new(buf)), self.chmap.clone()); let mut frm = NAFrame::new_from_pkt(pkt, info, NABufferType::AudioPacked(abuf)); + frm.set_duration(Some(duration)); frm.set_keyframe(true); Ok(Rc::new(RefCell::new(frm))) } else {