[nihav.git] / nihav-realmedia / src / codecs / rv30.rs

use nihav_core::formats;
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::*;

struct RealVideo30BR {
    rpr_bits:       u8,
    width:          usize,
    height:         usize,
    widths:         Vec<usize>,
    heights:        Vec<usize>,
}

impl RealVideo30BR {
    fn new() -> Self {
        RealVideo30BR {
            rpr_bits:       0,
            width:          0,
            height:         0,
            widths:         Vec::new(),
            heights:        Vec::new(),
        }
    }
}

impl RV34BitstreamDecoder for RealVideo30BR {
    fn decode_slice_header(&mut self, br: &mut BitReader, _old_w: usize, _old_h: usize) -> DecoderResult<RV34SliceHeader> {
        if br.read(3)? != 0 { return Err(DecoderError::InvalidData); }
        let ft_idx              = br.read(2)?;
        let ftype = match ft_idx {
                0|1 => FrameType::I,
                2   => FrameType::P,
                _   => FrameType::B,
            };
        if br.read(1)? != 0 { return Err(DecoderError::InvalidData); }
        let q                   = br.read(5)? as u8;
        let deblock             = !br.read_bool()?;
        let pts                 = br.read(13)? as u16;
        let rpr                 = br.read(self.rpr_bits)? as usize;
        let (w, h) = if rpr != 0 {
                validate!(rpr < self.widths.len());
                (self.widths[rpr], self.heights[rpr])
            } else {
                (self.width, self.height)
            };
        let start               = br.read(get_slice_start_offset_bits(w, h))? as usize;
                                  br.skip(1)?;

        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 {
            for x in 0..2 {
                let code = br.read_code(UintCodeType::Gamma)? as usize;
                validate!(code < 81);
                for k in 0..2 {
                    let new  = RV30_ITYPE_MAP[code * 2 + k] as usize;
                    let top  = (types[pos + x * 2 + k - tstride] + 1) as usize;
                    let left = (types[pos + x * 2 + k - 1] + 1) as usize;
                    types[pos + x * 2 + k] = RV30_ITYPE[top * 90 + left * 9 + new];
                    validate!(types[pos + x * 2 + k] != 9);
                }
            }
            pos += tstride;
        }
        Ok(())
    }
    fn decode_inter_mb_hdr(&mut self, br: &mut BitReader, ftype: FrameType, _mbtype: MBType) -> DecoderResult<MBInfo> {
        let mut code = br.read_code(UintCodeType::Gamma)? as usize;
        let mut dq = false;
        validate!(code < 11);
        if code > 5 {
            code -= 6;
            dq = true;
        }
        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: &[RV34MBInfo]) -> (MV, MV) {
        let mb_x = sstate.mb_x;
        let mb_y = sstate.mb_y;
        let mv_f;
        let mv_b;
        match mbtype {
            MBType::MBForward | MBType::MBBackward => {
                    let mv_pred = mvi.pred_mb_mv(mb_x, mb_y, true, sstate.has_top, sstate.has_left, sstate.has_tr, sstate.has_tl);
                    mv_f = mv_pred + mvs[0];
                    mv_b = mv_f;
                },
            _   => {
                    mv_f = ZERO_MV;
                    mv_b = ZERO_MV;
                },
        };
        (mv_f, mv_b)
    }
    fn quant_dc(&self, _is_intra: bool, q: u8) -> u8 { RV30_QUANT_DC[q as usize] }
}

struct RealVideo30Decoder {
    bd:         RealVideo30BR,
    info:       NACodecInfoRef,
    dec:        RV34Decoder,
}

impl RealVideo30Decoder {
    fn new() -> Self {
        RealVideo30Decoder{
            bd:         RealVideo30BR::new(),
            info:       NACodecInfoRef::default(),
            dec:        RV34Decoder::new(true, Box::new(RV30DSP::new())),
        }
    }
}

impl NADecoder for RealVideo30Decoder {
    fn init(&mut self, supp: &mut NADecoderSupport, info: NACodecInfoRef) -> DecoderResult<()> {
        if let NACodecTypeInfo::Video(vinfo) = info.get_properties() {
            let fmt = formats::YUV420_FORMAT;
            let myinfo = NACodecTypeInfo::Video(NAVideoInfo::new(0, 0, false, fmt));
            self.info = NACodecInfo::new_ref(info.get_name(), myinfo, info.get_extradata()).into_ref();

            let edata = info.get_extradata().unwrap();
            let src: &[u8] = &edata;

            if src.len() < 2 { return Err(DecoderError::InvalidData); }
            let num_rpr     = (src[1] & 7) as usize;
            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 {
                self.bd.widths.push ((src[6 + i * 2] as usize) << 2);
                self.bd.heights.push((src[7 + i * 2] as usize) << 2);
            }

            self.bd.width  = vinfo.get_width();
            self.bd.height = vinfo.get_height();

            supp.pool_u8.set_dec_bufs(3);
            supp.pool_u8.prealloc_video(NAVideoInfo::new(self.bd.width, self.bd.height, false, fmt), 4)?;

            Ok(())
        } else {
println!("???");
            Err(DecoderError::InvalidData)
        }
    }
    fn decode(&mut self, supp: &mut NADecoderSupport, pkt: &NAPacket) -> DecoderResult<NAFrameRef> {
        let src = pkt.get_buffer();

        let (bufinfo, ftype, pts) = self.dec.parse_frame(supp, src.as_slice(), &mut self.bd)?;

        let mut frm = NAFrame::new_from_pkt(pkt, self.info.clone(), bufinfo);
        frm.set_keyframe(ftype == FrameType::I);
        frm.set_pts(Some(pts));
        frm.set_frame_type(ftype);//if ftype == FrameType::B { FrameType::Skip } else { ftype } );
        Ok(frm.into_ref())
    }
    fn flush(&mut self) {
        self.dec.flush();
    }
}

impl NAOptionHandler for RealVideo30Decoder {
    fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] }
    fn set_options(&mut self, _options: &[NAOption]) { }
    fn query_option_value(&self, _name: &str) -> Option<NAValue> { None }
}

pub fn get_decoder() -> Box<dyn NADecoder + Send> {
    Box::new(RealVideo30Decoder::new())
}

#[cfg(test)]
mod test {
    use nihav_core::codecs::RegisteredDecoders;
    use nihav_core::demuxers::RegisteredDemuxers;
    use nihav_codec_support::test::dec_video::*;
    use crate::realmedia_register_all_decoders;
    use crate::realmedia_register_all_demuxers;
    #[test]
    fn test_rv30() {
        let mut dmx_reg = RegisteredDemuxers::new();
        realmedia_register_all_demuxers(&mut dmx_reg);
        let mut dec_reg = RegisteredDecoders::new();
        realmedia_register_all_decoders(&mut dec_reg);

        test_decoding("realmedia", "realvideo3", "assets/RV/rv30_weighted_mc.rm", Some(700),
                      &dmx_reg, &dec_reg, ExpectedTestResult::MD5Frames(vec![
                            [0x2a4d13bf, 0x2f21f3c9, 0xcbd601be, 0x61a6405c],
                            [0x17ea48c7, 0x68334ff5, 0x6fb9729b, 0x9a93ed12],
                            [0xce42a48c, 0x0b5b7f0d, 0x3f66c4a1, 0x261f08e2],
                            [0x91ca8f5b, 0x1f578a93, 0x44e533f2, 0x83beec8a],
                            [0x8cb256a7, 0xb3889afd, 0x28806114, 0x9bbd5287],
                            [0x694570e2, 0x4b2df948, 0xc7d2e36d, 0xa5eb66b2],
                            [0xb9b68059, 0x0d420917, 0x4e0f33d4, 0x8d3a6b0b],
                            [0xb9d6bfa6, 0x04442a8e, 0x6fafc34e, 0xb418a23e],
                            [0xb94e226d, 0xbf8a5fc5, 0x6d9a03c6, 0x4a0d1a50],
                            [0xa2e76d33, 0x1b6996e4, 0xb6a26052, 0x3f5f6145],
                            [0x3b509515, 0x4aa2f4f9, 0x12a0c73b, 0x5b9b20d1],
                            [0x976e0e06, 0xf6194e6f, 0xe0fefc31, 0xf7587bd3],
                            [0x7b38660e, 0xa46f4080, 0xa493f422, 0x36eaaa3b],
                            [0x6375934a, 0xf2a23087, 0x367f9738, 0xf2251e09],
                            [0x54bcefe7, 0xbbc91dc7, 0x0acec7d7, 0x95cf6d02]]));
        test_decoding("realmedia", "realvideo3", "assets/RV/simpsons-clip.rm", Some(1337),
                      &dmx_reg, &dec_reg,
                      ExpectedTestResult::MD5([0x36604117, 0x415f95cc, 0xec38e776, 0x9818d3be]));
    }
}

const RV30_QUANT_DC: [u8; 32] = [
     0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, 15,
    16, 17, 18, 19, 20, 21, 22, 22, 22, 23, 23, 23, 24, 24, 25, 25
];

const RV30_MB_TYPES: [[MBType; 6]; 2] = [
    [ MBType::MBSkip, MBType::MBP16x16, MBType::MBP8x8,    MBType::Invalid,    MBType::MBIntra, MBType::MBIntra16 ],
    [ MBType::MBSkip, MBType::MBDirect, MBType::MBForward, MBType::MBBackward, MBType::MBIntra, MBType::MBIntra16 ],
];

const RV30_ITYPE_MAP: [i8; 9*9*2] = [
    0, 0, 0, 1, 1, 0, 1, 1, 0, 2, 2, 0, 0, 3, 3, 0, 1, 2,
    2, 1, 0, 4, 4, 0, 3, 1, 1, 3, 0, 5, 5, 0, 2, 2, 1, 4,
    4, 1, 0, 6, 3, 2, 1, 5, 2, 3, 5, 1, 6, 0, 0, 7, 4, 2,
    2, 4, 3, 3, 6, 1, 1, 6, 7, 0, 0, 8, 5, 2, 4, 3, 2, 5,
    3, 4, 1, 7, 4, 4, 7, 1, 8, 0, 6, 2, 3, 5, 5, 3, 2, 6,
    1, 8, 2, 7, 7, 2, 8, 1, 5, 4, 4, 5, 3, 6, 6, 3, 8, 2,
    4, 6, 5, 5, 6, 4, 2, 8, 7, 3, 3, 7, 6, 5, 5, 6, 7, 4,
    4, 7, 8, 3, 3, 8, 7, 5, 8, 4, 5, 7, 4, 8, 6, 6, 7, 6,
    5, 8, 8, 5, 6, 7, 8, 6, 7, 7, 6, 8, 8, 7, 7, 8, 8, 8,
];

const RV30_ITYPE: [i8; 10*10*9] = [
    0, 9, 9, 9, 9, 9, 9, 9, 9,
    0, 2, 9, 9, 9, 9, 9, 9, 9,
    9, 9, 9, 9, 9, 9, 9, 9, 9,
    2, 0, 9, 9, 9, 9, 9, 9, 9,
    9, 9, 9, 9, 9, 9, 9, 9, 9,
    9, 9, 9, 9, 9, 9, 9, 9, 9,
    9, 9, 9, 9, 9, 9, 9, 9, 9,
    9, 9, 9, 9, 9, 9, 9, 9, 9,
    9, 9, 9, 9, 9, 9, 9, 9, 9,
    9, 9, 9, 9, 9, 9, 9, 9, 9,

    0, 1, 9, 9, 9, 9, 9, 9, 9,
    0, 2, 1, 6, 4, 8, 5, 7, 3,
    1, 0, 2, 6, 5, 4, 3, 8, 7,
    2, 8, 0, 1, 7, 4, 3, 6, 5,
    2, 0, 1, 3, 8, 5, 4, 7, 6,
    2, 0, 1, 4, 6, 7, 8, 3, 5,
    0, 1, 5, 2, 6, 3, 8, 4, 7,
    0, 1, 6, 2, 4, 7, 5, 8, 3,
    2, 7, 0, 1, 4, 8, 6, 3, 5,
    2, 8, 0, 1, 7, 3, 4, 5, 6,

    1, 0, 9, 9, 9, 9, 9, 9, 9,
    1, 2, 5, 6, 3, 0, 4, 8, 7,
    1, 6, 2, 5, 3, 0, 4, 8, 7,
    2, 1, 7, 6, 8, 3, 5, 0, 4,
    1, 2, 5, 3, 6, 8, 4, 7, 0,
    1, 6, 2, 0, 4, 5, 8, 7, 3,
    1, 5, 2, 6, 3, 8, 4, 0, 7,
    1, 6, 0, 2, 4, 5, 7, 3, 8,
    2, 1, 7, 6, 0, 8, 5, 4, 3,
    1, 2, 7, 8, 3, 4, 5, 6, 0,

    9, 9, 9, 9, 9, 9, 9, 9, 9,
    0, 2, 1, 8, 7, 6, 5, 4, 3,
    1, 2, 0, 6, 5, 7, 4, 8, 3,
    2, 8, 7, 1, 0, 6, 4, 3, 5,
    2, 0, 8, 1, 3, 7, 5, 4, 6,
    2, 0, 4, 1, 7, 8, 6, 3, 5,
    2, 0, 1, 5, 8, 4, 6, 7, 3,
    2, 0, 6, 1, 4, 7, 8, 5, 3,
    2, 7, 8, 1, 0, 5, 4, 6, 3,
    2, 8, 7, 1, 0, 4, 3, 6, 5,

    9, 9, 9, 9, 9, 9, 9, 9, 9,
    0, 2, 1, 3, 5, 8, 6, 4, 7,
    1, 0, 2, 5, 3, 6, 4, 8, 7,
    2, 8, 1, 0, 3, 5, 7, 6, 4,
    3, 2, 5, 8, 1, 4, 6, 7, 0,
    4, 2, 0, 6, 1, 5, 8, 3, 7,
    5, 3, 1, 2, 8, 6, 4, 0, 7,
    1, 6, 0, 2, 4, 5, 8, 3, 7,
    2, 7, 0, 1, 5, 4, 8, 6, 3,
    2, 8, 3, 5, 1, 0, 7, 6, 4,

    9, 9, 9, 9, 9, 9, 9, 9, 9,
    2, 0, 6, 1, 4, 7, 5, 8, 3,
    1, 6, 2, 0, 4, 5, 3, 7, 8,
    2, 8, 7, 6, 4, 0, 1, 5, 3,
    4, 2, 1, 0, 6, 8, 3, 5, 7,
    4, 2, 6, 0, 1, 5, 7, 8, 3,
    1, 2, 5, 0, 6, 3, 4, 7, 8,
    6, 4, 0, 1, 2, 7, 5, 3, 8,
    2, 7, 4, 6, 0, 1, 8, 5, 3,
    2, 8, 7, 4, 6, 1, 3, 5, 0,

    9, 9, 9, 9, 9, 9, 9, 9, 9,
    5, 1, 2, 3, 6, 8, 0, 4, 7,
    1, 5, 6, 3, 2, 0, 4, 8, 7,
    2, 1, 5, 3, 6, 8, 7, 4, 0,
    5, 3, 1, 2, 6, 8, 4, 7, 0,
    1, 6, 2, 4, 5, 8, 0, 3, 7,
    5, 1, 3, 6, 2, 0, 8, 4, 7,
    1, 6, 5, 2, 0, 4, 3, 7, 8,
    2, 7, 1, 6, 5, 0, 8, 3, 4,
    2, 5, 1, 3, 6, 8, 4, 0, 7,

    9, 9, 9, 9, 9, 9, 9, 9, 9,
    1, 6, 2, 0, 5, 4, 3, 7, 8,
    1, 6, 5, 4, 2, 3, 0, 7, 8,
    2, 1, 6, 7, 4, 8, 5, 3, 0,
    2, 1, 6, 5, 8, 4, 3, 0, 7,
    6, 4, 1, 2, 0, 5, 7, 8, 3,
    1, 6, 5, 2, 3, 0, 4, 8, 7,
    6, 1, 4, 0, 2, 7, 5, 3, 8,
    2, 7, 4, 6, 1, 5, 0, 8, 3,
    2, 1, 6, 8, 4, 7, 3, 5, 0,

    9, 9, 9, 9, 9, 9, 9, 9, 9,
    2, 0, 4, 7, 6, 1, 8, 5, 3,
    6, 1, 2, 0, 4, 7, 5, 8, 3,
    2, 7, 8, 0, 1, 6, 4, 3, 5,
    2, 4, 0, 8, 3, 1, 7, 6, 5,
    4, 2, 7, 0, 6, 1, 8, 5, 3,
    2, 1, 0, 8, 5, 6, 7, 4, 3,
    2, 6, 4, 1, 7, 0, 5, 8, 3,
    2, 7, 4, 0, 8, 6, 1, 5, 3,
    2, 8, 7, 4, 1, 0, 3, 6, 5,

    9, 9, 9, 9, 9, 9, 9, 9, 9,
    2, 0, 8, 1, 3, 4, 6, 5, 7,
    1, 2, 0, 6, 8, 5, 7, 3, 4,
    2, 8, 7, 1, 0, 3, 6, 5, 4,
    8, 3, 2, 5, 1, 0, 4, 7, 6,
    2, 0, 4, 8, 5, 1, 7, 6, 3,
    2, 1, 0, 8, 5, 3, 6, 4, 7,
    2, 1, 6, 0, 8, 4, 5, 7, 3,
    2, 7, 8, 4, 0, 6, 1, 5, 3,
    2, 8, 3, 0, 7, 4, 1, 6, 5,
];
Commit	Line	Data
5641dccf KS	1	use nihav_core::formats;
	2	use nihav_core::io::bitreader::*;
	3	use nihav_core::io::intcode::*;
	4	use nihav_core::frame::*;
	5	use nihav_core::codecs::*;
b4d5b851	6	use nihav_codec_support::codecs::{MV, ZERO_MV};
47527732 KS	7	use super::rv3040::*;
	8	use super::rv30dsp::*;
	9
	10	struct RealVideo30BR {
	11	rpr_bits: u8,
	12	width: usize,
	13	height: usize,
	14	widths: Vec<usize>,
	15	heights: Vec<usize>,
	16	}
	17
	18	impl RealVideo30BR {
	19	fn new() -> Self {
	20	RealVideo30BR {
	21	rpr_bits: 0,
	22	width: 0,
	23	height: 0,
	24	widths: Vec::new(),
	25	heights: Vec::new(),
	26	}
	27	}
	28	}
	29
	30	impl RV34BitstreamDecoder for RealVideo30BR {
	31	fn decode_slice_header(&mut self, br: &mut BitReader, _old_w: usize, _old_h: usize) -> DecoderResult<RV34SliceHeader> {
	32	if br.read(3)? != 0 { return Err(DecoderError::InvalidData); }
	33	let ft_idx = br.read(2)?;
	34	let ftype = match ft_idx {
	35	0\|1 => FrameType::I,
	36	2 => FrameType::P,
	37	_ => FrameType::B,
	38	};
	39	if br.read(1)? != 0 { return Err(DecoderError::InvalidData); }
	40	let q = br.read(5)? as u8;
	41	let deblock = !br.read_bool()?;
	42	let pts = br.read(13)? as u16;
	43	let rpr = br.read(self.rpr_bits)? as usize;
	44	let (w, h) = if rpr != 0 {
	45	validate!(rpr < self.widths.len());
	46	(self.widths[rpr], self.heights[rpr])
	47	} else {
	48	(self.width, self.height)
	49	};
	50	let start = br.read(get_slice_start_offset_bits(w, h))? as usize;
	51	br.skip(1)?;
	52
e07387c7	53	Ok(RV34SliceHeader{ ftype, quant: q, deblock, pts, width: w, height: h, start, end: 0, set_idx: 0 })
47527732 KS	54	}
	55	fn decode_intra_pred(&mut self, br: &mut BitReader, types: &mut [i8], mut pos: usize, tstride: usize, _has_top: bool) -> DecoderResult<()> {
	56	for _ in 0..4 {
	57	for x in 0..2 {
	58	let code = br.read_code(UintCodeType::Gamma)? as usize;
	59	validate!(code < 81);
	60	for k in 0..2 {
	61	let new = RV30_ITYPE_MAP[code * 2 + k] as usize;
	62	let top = (types[pos + x * 2 + k - tstride] + 1) as usize;
	63	let left = (types[pos + x * 2 + k - 1] + 1) as usize;
	64	types[pos + x * 2 + k] = RV30_ITYPE[top * 90 + left * 9 + new];
	65	validate!(types[pos + x * 2 + k] != 9);
	66	}
	67	}
	68	pos += tstride;
	69	}
	70	Ok(())
	71	}
	72	fn decode_inter_mb_hdr(&mut self, br: &mut BitReader, ftype: FrameType, _mbtype: MBType) -> DecoderResult<MBInfo> {
	73	let mut code = br.read_code(UintCodeType::Gamma)? as usize;
	74	let mut dq = false;
	75	validate!(code < 11);
	76	if code > 5 {
	77	code -= 6;
	78	dq = true;
	79	}
	80	let idx = if ftype == FrameType::P { 0 } else { 1 };
	81	Ok(MBInfo { mbtype: RV30_MB_TYPES[idx][code], skip_run: 0, dquant: dq })
	82	}
e07387c7	83	fn predict_b_mv(&self, sstate: &SState, mvi: &MVInfo, mbtype: MBType, mvs: &[MV], _mbinfo: &[RV34MBInfo]) -> (MV, MV) {
47527732 KS	84	let mb_x = sstate.mb_x;
	85	let mb_y = sstate.mb_y;
	86	let mv_f;
	87	let mv_b;
	88	match mbtype {
	89	MBType::MBForward \| MBType::MBBackward => {
	90	let mv_pred = mvi.pred_mb_mv(mb_x, mb_y, true, sstate.has_top, sstate.has_left, sstate.has_tr, sstate.has_tl);
	91	mv_f = mv_pred + mvs[0];
	92	mv_b = mv_f;
	93	},
	94	_ => {
	95	mv_f = ZERO_MV;
	96	mv_b = ZERO_MV;
	97	},
	98	};
	99	(mv_f, mv_b)
	100	}
	101	fn quant_dc(&self, _is_intra: bool, q: u8) -> u8 { RV30_QUANT_DC[q as usize] }
	102	}
	103
	104	struct RealVideo30Decoder {
	105	bd: RealVideo30BR,
2422d969	106	info: NACodecInfoRef,
47527732 KS	107	dec: RV34Decoder,
	108	}
	109
	110	impl RealVideo30Decoder {
	111	fn new() -> Self {
	112	RealVideo30Decoder{
	113	bd: RealVideo30BR::new(),
2422d969	114	info: NACodecInfoRef::default(),
47527732 KS	115	dec: RV34Decoder::new(true, Box::new(RV30DSP::new())),
	116	}
	117	}
	118	}
	119
	120	impl NADecoder for RealVideo30Decoder {
3c69ce1b	121	fn init(&mut self, supp: &mut NADecoderSupport, info: NACodecInfoRef) -> DecoderResult<()> {
47527732 KS	122	if let NACodecTypeInfo::Video(vinfo) = info.get_properties() {
	123	let fmt = formats::YUV420_FORMAT;
	124	let myinfo = NACodecTypeInfo::Video(NAVideoInfo::new(0, 0, false, fmt));
2422d969	125	self.info = NACodecInfo::new_ref(info.get_name(), myinfo, info.get_extradata()).into_ref();
47527732 KS	126
	127	let edata = info.get_extradata().unwrap();
	128	let src: &[u8] = &edata;
	129
	130	if src.len() < 2 { return Err(DecoderError::InvalidData); }
	131	let num_rpr = (src[1] & 7) as usize;
	132	if src.len() < num_rpr * 2 + 8 { return Err(DecoderError::ShortData); }
	133	self.bd.rpr_bits = ((num_rpr >> 1) + 1) as u8;
	134	if self.bd.rpr_bits > 3 { self.bd.rpr_bits = 3; }
e07387c7	135	for i in 0..=num_rpr {
47527732 KS	136	self.bd.widths.push ((src[6 + i * 2] as usize) << 2);
	137	self.bd.heights.push((src[7 + i * 2] as usize) << 2);
	138	}
	139
	140	self.bd.width = vinfo.get_width();
	141	self.bd.height = vinfo.get_height();
3c69ce1b KS	142
	143	supp.pool_u8.set_dec_bufs(3);
	144	supp.pool_u8.prealloc_video(NAVideoInfo::new(self.bd.width, self.bd.height, false, fmt), 4)?;
	145
47527732 KS	146	Ok(())
	147	} else {
	148	println!("???");
	149	Err(DecoderError::InvalidData)
	150	}
	151	}
3c69ce1b	152	fn decode(&mut self, supp: &mut NADecoderSupport, pkt: &NAPacket) -> DecoderResult<NAFrameRef> {
47527732 KS	153	let src = pkt.get_buffer();
47527732 KS	154
3c69ce1b	155	let (bufinfo, ftype, pts) = self.dec.parse_frame(supp, src.as_slice(), &mut self.bd)?;
47527732 KS	156
	157	let mut frm = NAFrame::new_from_pkt(pkt, self.info.clone(), bufinfo);
	158	frm.set_keyframe(ftype == FrameType::I);
	159	frm.set_pts(Some(pts));
	160	frm.set_frame_type(ftype);//if ftype == FrameType::B { FrameType::Skip } else { ftype } );
171860fc	161	Ok(frm.into_ref())
47527732	162	}
f9be4e75 KS	163	fn flush(&mut self) {
	164	self.dec.flush();
	165	}
47527732 KS	166	}
47527732 KS	167
7d57ae2f KS	168	impl NAOptionHandler for RealVideo30Decoder {
	169	fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] }
	170	fn set_options(&mut self, _options: &[NAOption]) { }
	171	fn query_option_value(&self, _name: &str) -> Option<NAValue> { None }
	172	}
	173
08a1fab7	174	pub fn get_decoder() -> Box<dyn NADecoder + Send> {
47527732 KS	175	Box::new(RealVideo30Decoder::new())
	176	}
	177
	178	#[cfg(test)]
	179	mod test {
3167c45c KS	180	use nihav_core::codecs::RegisteredDecoders;
3167c45c KS	181	use nihav_core::demuxers::RegisteredDemuxers;
ce742854	182	use nihav_codec_support::test::dec_video::*;
78fb6560	183	use crate::realmedia_register_all_decoders;
e64739f8	184	use crate::realmedia_register_all_demuxers;
47527732 KS	185	#[test]
47527732 KS	186	fn test_rv30() {
3167c45c KS	187	let mut dmx_reg = RegisteredDemuxers::new();
	188	realmedia_register_all_demuxers(&mut dmx_reg);
	189	let mut dec_reg = RegisteredDecoders::new();
78fb6560	190	realmedia_register_all_decoders(&mut dec_reg);
3167c45c	191
70860638 KS	192	test_decoding("realmedia", "realvideo3", "assets/RV/rv30_weighted_mc.rm", Some(700),
	193	&dmx_reg, &dec_reg, ExpectedTestResult::MD5Frames(vec![
	194	[0x2a4d13bf, 0x2f21f3c9, 0xcbd601be, 0x61a6405c],
	195	[0x17ea48c7, 0x68334ff5, 0x6fb9729b, 0x9a93ed12],
	196	[0xce42a48c, 0x0b5b7f0d, 0x3f66c4a1, 0x261f08e2],
	197	[0x91ca8f5b, 0x1f578a93, 0x44e533f2, 0x83beec8a],
	198	[0x8cb256a7, 0xb3889afd, 0x28806114, 0x9bbd5287],
	199	[0x694570e2, 0x4b2df948, 0xc7d2e36d, 0xa5eb66b2],
	200	[0xb9b68059, 0x0d420917, 0x4e0f33d4, 0x8d3a6b0b],
	201	[0xb9d6bfa6, 0x04442a8e, 0x6fafc34e, 0xb418a23e],
	202	[0xb94e226d, 0xbf8a5fc5, 0x6d9a03c6, 0x4a0d1a50],
	203	[0xa2e76d33, 0x1b6996e4, 0xb6a26052, 0x3f5f6145],
	204	[0x3b509515, 0x4aa2f4f9, 0x12a0c73b, 0x5b9b20d1],
	205	[0x976e0e06, 0xf6194e6f, 0xe0fefc31, 0xf7587bd3],
	206	[0x7b38660e, 0xa46f4080, 0xa493f422, 0x36eaaa3b],
	207	[0x6375934a, 0xf2a23087, 0x367f9738, 0xf2251e09],
	208	[0x54bcefe7, 0xbbc91dc7, 0x0acec7d7, 0x95cf6d02]]));
	209	test_decoding("realmedia", "realvideo3", "assets/RV/simpsons-clip.rm", Some(1337),
	210	&dmx_reg, &dec_reg,
	211	ExpectedTestResult::MD5([0x36604117, 0x415f95cc, 0xec38e776, 0x9818d3be]));
47527732 KS	212	}
	213	}
	214
	215	const RV30_QUANT_DC: [u8; 32] = [
	216	0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
	217	16, 17, 18, 19, 20, 21, 22, 22, 22, 23, 23, 23, 24, 24, 25, 25
	218	];
	219
	220	const RV30_MB_TYPES: [[MBType; 6]; 2] = [
	221	[ MBType::MBSkip, MBType::MBP16x16, MBType::MBP8x8, MBType::Invalid, MBType::MBIntra, MBType::MBIntra16 ],
	222	[ MBType::MBSkip, MBType::MBDirect, MBType::MBForward, MBType::MBBackward, MBType::MBIntra, MBType::MBIntra16 ],
	223	];
	224
	225	const RV30_ITYPE_MAP: [i8; 992] = [
	226	0, 0, 0, 1, 1, 0, 1, 1, 0, 2, 2, 0, 0, 3, 3, 0, 1, 2,
	227	2, 1, 0, 4, 4, 0, 3, 1, 1, 3, 0, 5, 5, 0, 2, 2, 1, 4,
	228	4, 1, 0, 6, 3, 2, 1, 5, 2, 3, 5, 1, 6, 0, 0, 7, 4, 2,
	229	2, 4, 3, 3, 6, 1, 1, 6, 7, 0, 0, 8, 5, 2, 4, 3, 2, 5,
	230	3, 4, 1, 7, 4, 4, 7, 1, 8, 0, 6, 2, 3, 5, 5, 3, 2, 6,
	231	1, 8, 2, 7, 7, 2, 8, 1, 5, 4, 4, 5, 3, 6, 6, 3, 8, 2,
	232	4, 6, 5, 5, 6, 4, 2, 8, 7, 3, 3, 7, 6, 5, 5, 6, 7, 4,
	233	4, 7, 8, 3, 3, 8, 7, 5, 8, 4, 5, 7, 4, 8, 6, 6, 7, 6,
	234	5, 8, 8, 5, 6, 7, 8, 6, 7, 7, 6, 8, 8, 7, 7, 8, 8, 8,
	235	];
	236
	237	const RV30_ITYPE: [i8; 10109] = [
	238	0, 9, 9, 9, 9, 9, 9, 9, 9,
	239	0, 2, 9, 9, 9, 9, 9, 9, 9,
	240	9, 9, 9, 9, 9, 9, 9, 9, 9,
	241	2, 0, 9, 9, 9, 9, 9, 9, 9,
	242	9, 9, 9, 9, 9, 9, 9, 9, 9,
	243	9, 9, 9, 9, 9, 9, 9, 9, 9,
	244	9, 9, 9, 9, 9, 9, 9, 9, 9,
	245	9, 9, 9, 9, 9, 9, 9, 9, 9,
	246	9, 9, 9, 9, 9, 9, 9, 9, 9,
	247	9, 9, 9, 9, 9, 9, 9, 9, 9,
	248
	249	0, 1, 9, 9, 9, 9, 9, 9, 9,
	250	0, 2, 1, 6, 4, 8, 5, 7, 3,
	251	1, 0, 2, 6, 5, 4, 3, 8, 7,
	252	2, 8, 0, 1, 7, 4, 3, 6, 5,
	253	2, 0, 1, 3, 8, 5, 4, 7, 6,
	254	2, 0, 1, 4, 6, 7, 8, 3, 5,
	255	0, 1, 5, 2, 6, 3, 8, 4, 7,
	256	0, 1, 6, 2, 4, 7, 5, 8, 3,
	257	2, 7, 0, 1, 4, 8, 6, 3, 5,
	258	2, 8, 0, 1, 7, 3, 4, 5, 6,
	259
	260	1, 0, 9, 9, 9, 9, 9, 9, 9,
	261	1, 2, 5, 6, 3, 0, 4, 8, 7,
	262	1, 6, 2, 5, 3, 0, 4, 8, 7,
	263	2, 1, 7, 6, 8, 3, 5, 0, 4,
	264	1, 2, 5, 3, 6, 8, 4, 7, 0,
	265	1, 6, 2, 0, 4, 5, 8, 7, 3,
	266	1, 5, 2, 6, 3, 8, 4, 0, 7,
	267	1, 6, 0, 2, 4, 5, 7, 3, 8,
	268	2, 1, 7, 6, 0, 8, 5, 4, 3,
	269	1, 2, 7, 8, 3, 4, 5, 6, 0,
	270
	271	9, 9, 9, 9, 9, 9, 9, 9, 9,
	272	0, 2, 1, 8, 7, 6, 5, 4, 3,
	273	1, 2, 0, 6, 5, 7, 4, 8, 3,
	274	2, 8, 7, 1, 0, 6, 4, 3, 5,
	275	2, 0, 8, 1, 3, 7, 5, 4, 6,
276	2, 0, 4, 1, 7, 8, 6, 3, 5,
277	2, 0, 1, 5, 8, 4, 6, 7, 3,
278	2, 0, 6, 1, 4, 7, 8, 5, 3,
279	2, 7, 8, 1, 0, 5, 4, 6, 3,
280	2, 8, 7, 1, 0, 4, 3, 6, 5,
281
282	9, 9, 9, 9, 9, 9, 9, 9, 9,
283	0, 2, 1, 3, 5, 8, 6, 4, 7,
284	1, 0, 2, 5, 3, 6, 4, 8, 7,
285	2, 8, 1, 0, 3, 5, 7, 6, 4,
286	3, 2, 5, 8, 1, 4, 6, 7, 0,
287	4, 2, 0, 6, 1, 5, 8, 3, 7,
288	5, 3, 1, 2, 8, 6, 4, 0, 7,
289	1, 6, 0, 2, 4, 5, 8, 3, 7,
290	2, 7, 0, 1, 5, 4, 8, 6, 3,
291	2, 8, 3, 5, 1, 0, 7, 6, 4,
292
293	9, 9, 9, 9, 9, 9, 9, 9, 9,
294	2, 0, 6, 1, 4, 7, 5, 8, 3,
295	1, 6, 2, 0, 4, 5, 3, 7, 8,
296	2, 8, 7, 6, 4, 0, 1, 5, 3,
297	4, 2, 1, 0, 6, 8, 3, 5, 7,
298	4, 2, 6, 0, 1, 5, 7, 8, 3,
299	1, 2, 5, 0, 6, 3, 4, 7, 8,
300	6, 4, 0, 1, 2, 7, 5, 3, 8,
301	2, 7, 4, 6, 0, 1, 8, 5, 3,
302	2, 8, 7, 4, 6, 1, 3, 5, 0,
303
304	9, 9, 9, 9, 9, 9, 9, 9, 9,
305	5, 1, 2, 3, 6, 8, 0, 4, 7,
306	1, 5, 6, 3, 2, 0, 4, 8, 7,
307	2, 1, 5, 3, 6, 8, 7, 4, 0,
308	5, 3, 1, 2, 6, 8, 4, 7, 0,
309	1, 6, 2, 4, 5, 8, 0, 3, 7,
310	5, 1, 3, 6, 2, 0, 8, 4, 7,
311	1, 6, 5, 2, 0, 4, 3, 7, 8,
312	2, 7, 1, 6, 5, 0, 8, 3, 4,
313	2, 5, 1, 3, 6, 8, 4, 0, 7,
314
315	9, 9, 9, 9, 9, 9, 9, 9, 9,
316	1, 6, 2, 0, 5, 4, 3, 7, 8,
317	1, 6, 5, 4, 2, 3, 0, 7, 8,
318	2, 1, 6, 7, 4, 8, 5, 3, 0,
319	2, 1, 6, 5, 8, 4, 3, 0, 7,
320	6, 4, 1, 2, 0, 5, 7, 8, 3,
321	1, 6, 5, 2, 3, 0, 4, 8, 7,
322	6, 1, 4, 0, 2, 7, 5, 3, 8,
323	2, 7, 4, 6, 1, 5, 0, 8, 3,
324	2, 1, 6, 8, 4, 7, 3, 5, 0,
325
326	9, 9, 9, 9, 9, 9, 9, 9, 9,
327	2, 0, 4, 7, 6, 1, 8, 5, 3,
328	6, 1, 2, 0, 4, 7, 5, 8, 3,
329	2, 7, 8, 0, 1, 6, 4, 3, 5,
330	2, 4, 0, 8, 3, 1, 7, 6, 5,
331	4, 2, 7, 0, 6, 1, 8, 5, 3,
332	2, 1, 0, 8, 5, 6, 7, 4, 3,
333	2, 6, 4, 1, 7, 0, 5, 8, 3,
334	2, 7, 4, 0, 8, 6, 1, 5, 3,
335	2, 8, 7, 4, 1, 0, 3, 6, 5,
336
337	9, 9, 9, 9, 9, 9, 9, 9, 9,
338	2, 0, 8, 1, 3, 4, 6, 5, 7,
339	1, 2, 0, 6, 8, 5, 7, 3, 4,
340	2, 8, 7, 1, 0, 3, 6, 5, 4,
341	8, 3, 2, 5, 1, 0, 4, 7, 6,
342	2, 0, 4, 8, 5, 1, 7, 6, 3,
343	2, 1, 0, 8, 5, 3, 6, 4, 7,
344	2, 1, 6, 0, 8, 4, 5, 7, 3,
345	2, 7, 8, 4, 0, 6, 1, 5, 3,
346	2, 8, 3, 0, 7, 4, 1, 6, 5,
347	];