X-Git-Url: https://git.nihav.org/?a=blobdiff_plain;f=nihav-realmedia%2Fsrc%2Fcodecs%2Frv30.rs;h=552e7c6fdc5788f1d6603ba8afce316f28b52c4e;hb=ce742854b2912b880fb3d3e330042b049dac8504;hp=4f7028030befeda29a02f968441b8bf1fc08e00e;hpb=6011e20199143f519881660144a4ca95ba77fd2d;p=nihav.git diff --git a/nihav-realmedia/src/codecs/rv30.rs b/nihav-realmedia/src/codecs/rv30.rs index 4f70280..552e7c6 100644 --- a/nihav-realmedia/src/codecs/rv30.rs +++ b/nihav-realmedia/src/codecs/rv30.rs @@ -3,6 +3,7 @@ use nihav_core::io::bitreader::*; use nihav_core::io::intcode::*; use nihav_core::frame::*; use nihav_core::codecs::*; +use nihav_codec_support::codecs::{MV, ZERO_MV}; use super::rv3040::*; use super::rv30dsp::*; @@ -49,7 +50,7 @@ impl RV34BitstreamDecoder for RealVideo30BR { let start = br.read(get_slice_start_offset_bits(w, h))? as usize; br.skip(1)?; - Ok(RV34SliceHeader{ ftype: ftype, quant: q, deblock: deblock, pts: pts, width: w, height: h, start: start, end: 0, set_idx: 0 }) + Ok(RV34SliceHeader{ ftype, quant: q, deblock, pts, width: w, height: h, start, end: 0, set_idx: 0 }) } fn decode_intra_pred(&mut self, br: &mut BitReader, types: &mut [i8], mut pos: usize, tstride: usize, _has_top: bool) -> DecoderResult<()> { for _ in 0..4 { @@ -79,7 +80,7 @@ impl RV34BitstreamDecoder for RealVideo30BR { let idx = if ftype == FrameType::P { 0 } else { 1 }; Ok(MBInfo { mbtype: RV30_MB_TYPES[idx][code], skip_run: 0, dquant: dq }) } - fn predict_b_mv(&self, sstate: &SState, mvi: &MVInfo, mbtype: MBType, mvs: &[MV], _mbinfo: &Vec) -> (MV, MV) { + fn predict_b_mv(&self, sstate: &SState, mvi: &MVInfo, mbtype: MBType, mvs: &[MV], _mbinfo: &[RV34MBInfo]) -> (MV, MV) { let mb_x = sstate.mb_x; let mb_y = sstate.mb_y; let mv_f; @@ -131,7 +132,7 @@ impl NADecoder for RealVideo30Decoder { if src.len() < num_rpr * 2 + 8 { return Err(DecoderError::ShortData); } self.bd.rpr_bits = ((num_rpr >> 1) + 1) as u8; if self.bd.rpr_bits > 3 { self.bd.rpr_bits = 3; } - for i in 0..num_rpr+1 { + for i in 0..=num_rpr { self.bd.widths.push ((src[6 + i * 2] as usize) << 2); self.bd.heights.push((src[7 + i * 2] as usize) << 2); } @@ -159,9 +160,12 @@ println!("???"); frm.set_frame_type(ftype);//if ftype == FrameType::B { FrameType::Skip } else { ftype } ); Ok(frm.into_ref()) } + fn flush(&mut self) { + self.dec.flush(); + } } -pub fn get_decoder() -> Box { +pub fn get_decoder() -> Box { Box::new(RealVideo30Decoder::new()) } @@ -169,9 +173,9 @@ pub fn get_decoder() -> Box { mod test { use nihav_core::codecs::RegisteredDecoders; use nihav_core::demuxers::RegisteredDemuxers; - use nihav_core::test::dec_video::*; - use crate::codecs::realmedia_register_all_codecs; - use crate::demuxers::realmedia_register_all_demuxers; + use nihav_codec_support::test::dec_video::*; + use crate::realmedia_register_all_codecs; + use crate::realmedia_register_all_demuxers; #[test] fn test_rv30() { let mut dmx_reg = RegisteredDemuxers::new();