X-Git-Url: https://git.nihav.org/?a=blobdiff_plain;f=nihav-duck%2Fsrc%2Fcodecs%2Fvp3.rs;h=2e6137f6ba0ee25bbf2c6dd8b5d8c216e8be3959;hb=56e5e5455bffbe9c180896aec15cda1396fe7e95;hp=1d85a7915d85dc0961b240b9536dcc408cf471e3;hpb=5b24175d582e438c4b8ccb685e2eb5913178fc3e;p=nihav.git diff --git a/nihav-duck/src/codecs/vp3.rs b/nihav-duck/src/codecs/vp3.rs index 1d85a79..2e6137f 100644 --- a/nihav-duck/src/codecs/vp3.rs +++ b/nihav-duck/src/codecs/vp3.rs @@ -14,7 +14,49 @@ enum SBState { Uncoded, } -struct VP3Codes { +fn map_idx(idx: usize) -> u8 { + idx as u8 +} + +struct VP30Codes { + dc_cb: [Codebook; 5], + ac_i_cb: [Codebook; 5], + ac_p_cb: [Codebook; 5], + mbtype_cb: Codebook, +} + +fn map_mbt(idx: usize) -> VPMBType { + VP30_MBTYPE_SYMS[idx] +} + +impl VP30Codes { + fn new() -> Self { + let mut dc_cb: [Codebook; 5]; + let mut ac_i_cb: [Codebook; 5]; + let mut ac_p_cb: [Codebook; 5]; + let mut cr = TableCodebookDescReader::new(&VP30_MBTYPE_CODES, &VP30_MBTYPE_BITS, map_mbt); + let mbtype_cb = Codebook::new(&mut cr, CodebookMode::MSB).unwrap(); + unsafe { + dc_cb = mem::uninitialized(); + ac_i_cb = mem::uninitialized(); + ac_p_cb = mem::uninitialized(); + for i in 0..5 { + let mut cr = TableCodebookDescReader::new(&VP30_DC_CODES[i], &VP30_DC_BITS[i], map_idx); + let cb = Codebook::new(&mut cr, CodebookMode::MSB).unwrap(); + ptr::write(&mut dc_cb[i], cb); + let mut cr = TableCodebookDescReader::new(&VP30_AC_INTRA_CODES[i], &VP30_AC_INTRA_BITS[i], map_idx); + let cb = Codebook::new(&mut cr, CodebookMode::MSB).unwrap(); + ptr::write(&mut ac_i_cb[i], cb); + let mut cr = TableCodebookDescReader::new(&VP30_AC_INTER_CODES[i], &VP30_AC_INTER_BITS[i], map_idx); + let cb = Codebook::new(&mut cr, CodebookMode::MSB).unwrap(); + ptr::write(&mut ac_p_cb[i], cb); + } + } + Self { dc_cb, ac_i_cb, ac_p_cb, mbtype_cb } + } +} + +struct VP31Codes { dc_cb: [Codebook; 16], ac0_cb: [Codebook; 16], ac1_cb: [Codebook; 16], @@ -22,11 +64,7 @@ struct VP3Codes { ac3_cb: [Codebook; 16], } -fn map_idx(idx: usize) -> u8 { - idx as u8 -} - -impl VP3Codes { +impl VP31Codes { fn new() -> Self { let mut dc_cb: [Codebook; 16]; let mut ac0_cb: [Codebook; 16]; @@ -40,20 +78,20 @@ impl VP3Codes { ac2_cb = mem::uninitialized(); ac3_cb = mem::uninitialized(); for i in 0..16 { - let mut cr = TableCodebookDescReader::new(&VP3_DC_CODES[i], &VP3_DC_BITS[i], map_idx); + let mut cr = TableCodebookDescReader::new(&VP31_DC_CODES[i], &VP31_DC_BITS[i], map_idx); let cb = Codebook::new(&mut cr, CodebookMode::MSB).unwrap(); ptr::write(&mut dc_cb[i], cb); - let mut cr = TableCodebookDescReader::new(&VP3_AC_CAT0_CODES[i], &VP3_AC_CAT0_BITS[i], map_idx); + let mut cr = TableCodebookDescReader::new(&VP31_AC_CAT0_CODES[i], &VP31_AC_CAT0_BITS[i], map_idx); let cb = Codebook::new(&mut cr, CodebookMode::MSB).unwrap(); ptr::write(&mut ac0_cb[i], cb); - let mut cr = TableCodebookDescReader::new(&VP3_AC_CAT1_CODES[i], &VP3_AC_CAT1_BITS[i], map_idx); + let mut cr = TableCodebookDescReader::new(&VP31_AC_CAT1_CODES[i], &VP31_AC_CAT1_BITS[i], map_idx); let cb = Codebook::new(&mut cr, CodebookMode::MSB).unwrap(); ptr::write(&mut ac1_cb[i], cb); - let mut cr = TableCodebookDescReader::new(&VP3_AC_CAT2_CODES[i], &VP3_AC_CAT2_BITS[i], map_idx); + let mut cr = TableCodebookDescReader::new(&VP31_AC_CAT2_CODES[i], &VP31_AC_CAT2_BITS[i], map_idx); let cb = Codebook::new(&mut cr, CodebookMode::MSB).unwrap(); ptr::write(&mut ac2_cb[i], cb); - let mut cr = TableCodebookDescReader::new(&VP3_AC_CAT3_CODES[i], &VP3_AC_CAT3_BITS[i], map_idx); + let mut cr = TableCodebookDescReader::new(&VP31_AC_CAT3_CODES[i], &VP31_AC_CAT3_BITS[i], map_idx); let cb = Codebook::new(&mut cr, CodebookMode::MSB).unwrap(); ptr::write(&mut ac3_cb[i], cb); } @@ -62,6 +100,12 @@ impl VP3Codes { } } +enum Codes { + None, + VP30(VP30Codes), + VP31(VP31Codes), +} + #[derive(Clone)] struct Block { btype: VPMBType, @@ -87,19 +131,19 @@ impl Block { type ReadRunFunc = fn (&mut BitReader) -> DecoderResult; -const VP3_LONG_RUN_BASE: [usize; 7] = [ 1, 2, 4, 6, 10, 18, 34 ]; -const VP3_LONG_RUN_BITS: [u8; 7] = [ 0, 1, 1, 2, 3, 4, 12 ]; +const VP31_LONG_RUN_BASE: [usize; 7] = [ 1, 2, 4, 6, 10, 18, 34 ]; +const VP31_LONG_RUN_BITS: [u8; 7] = [ 0, 1, 1, 2, 3, 4, 12 ]; fn read_long_run(br: &mut BitReader) -> DecoderResult { let pfx = br.read_code(UintCodeType::LimitedUnary(6, 0))? as usize; if pfx == 0 { return Ok(1); } - Ok(VP3_LONG_RUN_BASE[pfx] + (br.read(VP3_LONG_RUN_BITS[pfx])? as usize)) + Ok(VP31_LONG_RUN_BASE[pfx] + (br.read(VP31_LONG_RUN_BITS[pfx])? as usize)) } -const VP3_SHORT_RUN_BASE: [usize; 6] = [ 1, 3, 5, 7, 11, 15 ]; -const VP3_SHORT_RUN_BITS: [u8; 6] = [ 1, 1, 1, 2, 2, 4 ]; +const VP31_SHORT_RUN_BASE: [usize; 6] = [ 1, 3, 5, 7, 11, 15 ]; +const VP31_SHORT_RUN_BITS: [u8; 6] = [ 1, 1, 1, 2, 2, 4 ]; fn read_short_run(br: &mut BitReader) -> DecoderResult { let pfx = br.read_code(UintCodeType::LimitedUnary(5, 0))? as usize; - Ok(VP3_SHORT_RUN_BASE[pfx] + (br.read(VP3_SHORT_RUN_BITS[pfx])? as usize)) + Ok(VP31_SHORT_RUN_BASE[pfx] + (br.read(VP31_SHORT_RUN_BITS[pfx])? as usize)) } struct BitRunDecoder { @@ -123,6 +167,63 @@ impl BitRunDecoder { } } +const VP30_NE0_BITS: [u8; 5] = [ 2, 2, 3, 4, 8 ]; +const VP30_NE0_BASE: [usize; 5] = [ 1, 5, 9, 17, 33 ]; +fn vp30_read_ne_run0(br: &mut BitReader) -> DecoderResult { + let len = br.read_code(UintCodeType::LimitedUnary(4, 0))? as usize; + Ok(VP30_NE0_BASE[len] + (br.read(VP30_NE0_BITS[len])? as usize)) +} +fn vp30_read_ne_run1(br: &mut BitReader) -> DecoderResult { + let len = br.read_code(UintCodeType::LimitedUnary(6, 0))? as usize; + if len == 0 { + Ok((br.read(1)? as usize) + 1) + } else if len < 6 { + Ok(len + 2) + } else { + Ok((br.read(8)? as usize) + 8) + } +} +fn vp30_read_coded_run0(br: &mut BitReader) -> DecoderResult { + let len = br.read_code(UintCodeType::LimitedUnary(5, 0))? as usize; + Ok(len + 1) +} +/* + 0 - 1 + 11 - 2 + 1000 - 3 + 1010 - 4 + 10011 - 5 + 10111 - 6 + 10010 - 7 + get_bits(3) + 101100 - 15 + get_bits(5) + 1011010 - 47 + get_bits(8) + 1011011 - 303 + get_bits(16) + */ +const VP30_CRUN1_LUT: [u8; 32] = [ + 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, + 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, + 0x34, 0x34, 0x75, 0x55, 0x44, 0x44, 0x85, 0x65, + 0x22, 0x22, 0x22, 0x22, 0x22, 0x22, 0x22, 0x22 +]; +fn vp30_read_coded_run1(br: &mut BitReader) -> DecoderResult { + let idx = br.peek(5) as usize; + let sym = (VP30_CRUN1_LUT[idx] >> 4) as usize; + let bits = VP30_CRUN1_LUT[idx] & 0xF; + br.skip(bits as u32)?; + if sym < 7 { + Ok(sym) + } else if sym == 7 { + Ok(7 + (br.read(3)? as usize)) + } else { + let len = br.read_code(UintCodeType::Unary012)?; + match len { + 0 => Ok(15 + (br.read(5)? as usize)), + 1 => Ok(47 + (br.read(8)? as usize)), + _ => Ok(303 + (br.read(16)? as usize)), + } + } +} + struct VP34Decoder { info: NACodecInfoRef, width: usize, @@ -133,10 +234,11 @@ struct VP34Decoder { is_intra: bool, quant: usize, shuf: VPShuffler, - codes: VP3Codes, + codes: Codes, loop_str: i16, blocks: Vec, + mb_coded: Vec, y_blocks: usize, y_sbs: usize, qmat_y: [i16; 64], @@ -149,6 +251,30 @@ struct VP34Decoder { blk_addr: Vec, sb_info: Vec, sb_blocks: Vec, + sb_mbs: Vec, + mb_blocks: Vec, +} + +fn vp30_read_mv_comp(br: &mut BitReader) -> DecoderResult { + let mode = br.read(2)?; + if mode == 0 { return Ok(0); } + let sign = br.read_bool()?; + let val = match mode - 1 { + 0 => 1, + 1 => 2 + (br.read(2)? as i16), + _ => br.read(5)? as i16, + }; + if !sign { + Ok(val) + } else { + Ok(-val) + } +} + +fn vp30_read_mv(br: &mut BitReader) -> DecoderResult { + let x = vp30_read_mv_comp(br)?; + let y = vp30_read_mv_comp(br)?; + Ok(MV{ x, y }) } fn read_mv_comp_packed(br: &mut BitReader) -> DecoderResult { @@ -207,11 +333,119 @@ fn read_mv_raw(br: &mut BitReader) -> DecoderResult { Ok(MV{ x, y }) } -fn rescale_qmat(dst_qmat: &mut [i16; 64], base_qmat: &[i16; 64], dc_quant: i16, ac_quant: i16) { +fn rescale_qmat(dst_qmat: &mut [i16; 64], base_qmat: &[i16; 64], dc_quant: i16, ac_quant: i16, minval: i16) { for (dst, src) in dst_qmat.iter_mut().zip(base_qmat.iter()) { - *dst = (src.wrapping_mul(ac_quant) / 100).max(2) << 2; + *dst = (src.wrapping_mul(ac_quant) / 100).max(minval) << 2; } - dst_qmat[0] = (base_qmat[0] * dc_quant / 100).max(4) << 2; + dst_qmat[0] = (base_qmat[0] * dc_quant / 100).max(minval * 2) << 2; +} + +fn expand_token(blk: &mut Block, br: &mut BitReader, eob_run: &mut usize, coef_no: usize, token: u8) -> DecoderResult<()> { + match token { + // EOBs + 0 | 1 | 2 => { *eob_run = (token as usize) + 1; }, + 3 | 4 | 5 => { + let bits = token - 1; + *eob_run = (br.read(bits)? as usize) + (1 << bits); + }, + 6 => { *eob_run = br.read(12)? as usize; }, + // zero runs + 7 | 8 => { + let bits = if token == 7 { 3 } else { 6 }; + let run = (br.read(bits)? as usize) + 1; + blk.idx += run; + validate!(blk.idx <= 64); + }, + // single coefficients + 9 | 10 | 11 | 12 => { + let val = (i16::from(token) - 7) >> 1; + if (token & 1) == 1 { + blk.coeffs[ZIGZAG[blk.idx]] = val; + } else { + blk.coeffs[ZIGZAG[blk.idx]] = -val; + } + blk.idx += 1; + }, + 13 | 14 | 15 | 16 => { + let val = i16::from(token) - 10; + if !br.read_bool()? { + blk.coeffs[ZIGZAG[blk.idx]] = val; + } else { + blk.coeffs[ZIGZAG[blk.idx]] = -val; + } + blk.idx += 1; + }, + 17 | 18 | 19 | 20 | 21 | 22 => { + let add_bits = if token == 22 { 9 } else { token - 16 }; + let sign = br.read_bool()?; + let val = (br.read(add_bits)? as i16) + VP3_LITERAL_BASE[(token - 17) as usize]; + if !sign { + blk.coeffs[ZIGZAG[blk.idx]] = val; + } else { + blk.coeffs[ZIGZAG[blk.idx]] = -val; + } + blk.idx += 1; + } + // zero run plus coefficient + 23 | 24 | 25 | 26 | 27 => { + blk.idx += (token - 22) as usize; + validate!(blk.idx < 64); + if !br.read_bool()? { + blk.coeffs[ZIGZAG[blk.idx]] = 1; + } else { + blk.coeffs[ZIGZAG[blk.idx]] = -1; + } + blk.idx += 1; + }, + 28 | 29 => { + let run_bits = token - 26; + if token == 28 { + blk.idx += 6; + } else { + blk.idx += 10; + } + let sign = br.read_bool()?; + blk.idx += br.read(run_bits)? as usize; + validate!(blk.idx < 64); + if !sign { + blk.coeffs[ZIGZAG[blk.idx]] = 1; + } else { + blk.coeffs[ZIGZAG[blk.idx]] = -1; + } + blk.idx += 1; + }, + 30 => { + blk.idx += 1; + validate!(blk.idx < 64); + let sign = br.read_bool()?; + let val = (br.read(1)? as i16) + 2; + if !sign { + blk.coeffs[ZIGZAG[blk.idx]] = val; + } else { + blk.coeffs[ZIGZAG[blk.idx]] = -val; + } + blk.idx += 1; + }, + _ => { + let sign = br.read_bool()?; + let val = (br.read(1)? as i16) + 2; + blk.idx += (br.read(1)? as usize) + 2; + validate!(blk.idx < 64); + if !sign { + blk.coeffs[ZIGZAG[blk.idx]] = val; + } else { + blk.coeffs[ZIGZAG[blk.idx]] = -val; + } + blk.idx += 1; + }, + }; + if *eob_run > 0 { + blk.idx = 64; + *eob_run -= 1; + } else if coef_no > 0 { + blk.has_ac = true; + } + Ok(()) } macro_rules! fill_dc_pred { @@ -272,39 +506,14 @@ fn vp3_interp11(dst: &mut [u8], dstride: usize, src: &[u8], sstride: usize, bw: } } -fn vp3_loop_filter(data: &mut [u8], mut off: usize, step: usize, stride: usize, loop_str: i16) { - for _ in 0..8 { - let a = data[off - step * 2] as i16; - let b = data[off - step] as i16; - let c = data[off] as i16; - let d = data[off + step] as i16; - let mut diff = ((a - d) + 3 * (c - b) + 4) >> 3; - if diff.abs() >= 2 * loop_str { - diff = 0; - } else if diff.abs() >= loop_str { - if diff < 0 { - diff = -diff - 2 * loop_str; - } else { - diff = -diff + 2 * loop_str; - } - } - if diff != 0 { - data[off - step] = (b + diff).max(0).min(255) as u8; - data[off] = (c - diff).max(0).min(255) as u8; - } - - off += stride; - } -} - -fn vp3_loop_filter_v(frm: &mut NASimpleVideoFrame, x: usize, y: usize, plane: usize, loop_str: i16) { +fn vp31_loop_filter_v(frm: &mut NASimpleVideoFrame, x: usize, y: usize, plane: usize, loop_str: i16) { let off = frm.offset[plane] + x + y * frm.stride[plane]; - vp3_loop_filter(frm.data, off, 1, frm.stride[plane], loop_str); + vp31_loop_filter(frm.data, off, 1, frm.stride[plane], 8, loop_str); } -fn vp3_loop_filter_h(frm: &mut NASimpleVideoFrame, x: usize, y: usize, plane: usize, loop_str: i16) { +fn vp31_loop_filter_h(frm: &mut NASimpleVideoFrame, x: usize, y: usize, plane: usize, loop_str: i16) { let off = frm.offset[plane] + x + y * frm.stride[plane]; - vp3_loop_filter(frm.data, off, frm.stride[plane], 1, loop_str); + vp31_loop_filter(frm.data, off, frm.stride[plane], 1, 8, loop_str); } pub const VP3_INTERP_FUNCS: &[blockdsp::BlkInterpFunc] = &[ vp3_interp00, vp3_interp01, vp3_interp10, vp3_interp11 ]; @@ -321,10 +530,11 @@ impl VP34Decoder { is_intra: true, quant: 0, shuf: VPShuffler::new(), - codes: VP3Codes::new(), + codes: Codes::None, loop_str: 0, blocks: Vec::new(), + mb_coded: Vec::new(), y_blocks: 0, y_sbs: 0, @@ -338,28 +548,231 @@ impl VP34Decoder { blk_addr: Vec::new(), sb_info: Vec::new(), sb_blocks: Vec::new(), + sb_mbs: Vec::new(), + mb_blocks: Vec::new(), } } fn parse_header(&mut self, br: &mut BitReader) -> DecoderResult<()> { self.is_intra = !br.read_bool()?; br.skip(1)?; self.quant = br.read(6)? as usize; - self.loop_str = VP3_LOOP_STRENGTH[self.quant]; -println!("quant = {}", self.quant); + self.loop_str = VP31_LOOP_STRENGTH[self.quant]; if self.is_intra { if br.peek(8) != 0 { - unimplemented!(); - } - let version = br.read(13)?; + validate!(self.version == 3 || self.version == 30); + let mb_w = br.read(8)? as usize; + let mb_h = br.read(8)? as usize; +println!(" VP30 {}x{} ({}x{})", mb_w, mb_h, self.mb_w, self.mb_h); + validate!(mb_w == self.mb_w && mb_h == self.mb_h); + if self.version == 3 { + self.version = 30; + self.codes = Codes::VP30(VP30Codes::new()); + } + } else { + let version = br.read(13)?; println!("intra, ver {} (self {})", version, self.version); - validate!((self.version == 3 && version == 1) || (self.version == 4 && version == 3)); - let coding_type = br.read(1)?; - validate!(coding_type == 0); + let coding_type = br.read(1)?; + validate!(coding_type == 0); br.skip(2)?; + if version == 1 { + validate!(self.version == 3 || self.version == 31); + if self.version == 3 { + self.version = 31; + self.codes = Codes::VP31(VP31Codes::new()); + } + } else if version == 2 { + validate!(self.version == 4); + unimplemented!(); + } else { + return Err(DecoderError::InvalidData); + } + } } +//println!("decode frame({},{},{})", self.is_intra as u8, self.is_intra as u8, self.quant); Ok(()) } - fn unpack_sb_info(&mut self, br: &mut BitReader) -> DecoderResult<()> { + fn vp30_unpack_sb_info(&mut self, br: &mut BitReader) -> DecoderResult<()> { + let mut has_nonempty = false; + { + let mut bit = !br.read_bool()?; + let mut run = 0; + for sb in self.sb_info.iter_mut() { + if run == 0 { + bit = !bit; + run = if bit { vp30_read_ne_run1(br)? } else { vp30_read_ne_run0(br)? }; + } + *sb = if bit { has_nonempty = true; SBState::Partial } else { SBState::Uncoded }; + run -= 1; + } + validate!(run == 0); + } + if has_nonempty { + for el in self.mb_coded.iter_mut() { *el = false; } + let mut bit = !br.read_bool()?; + let mut run = 0; + let mut mbiter = self.mb_coded.iter_mut(); + for (sb, nmb) in self.sb_info.iter_mut().zip(self.sb_mbs.iter()) { + let nmbs = *nmb as usize; + if *sb == SBState::Partial { + for _ in 0..nmbs { + if run == 0 { + bit = !bit; + run = if bit { vp30_read_coded_run1(br)? } else { vp30_read_coded_run0(br)? }; + } + run -= 1; + *mbiter.next().unwrap() = bit; + } + } else { + for _ in 0..nmbs { + mbiter.next().unwrap(); + } + } + } + validate!(run == 0); + let mut bit = !br.read_bool()?; + let mut run = 0; + let mut cur_blk = 0; + for (coded, nblk) in self.mb_coded.iter().zip(self.mb_blocks.iter()) { + let nblks = *nblk as usize; + if *coded { + let mut cb = [false; 4]; + for j in 0..nblks { + if run == 0 { + bit = !bit; + run = if bit { vp30_read_coded_run1(br)? } else { vp30_read_coded_run0(br)? }; + } + run -= 1; + cb[j] = bit; + } + for j in 0..nblks { + let addr = self.blk_addr[cur_blk + j] >> 2; + self.blocks[addr].coded = cb[j]; + } + } + cur_blk += nblks; + } + validate!(run == 0); + } + Ok(()) + } + fn vp30_unpack_mb_info(&mut self, br: &mut BitReader) -> DecoderResult<()> { + let mut cur_blk = 0; + if let Codes::VP30(ref codes) = self.codes { + for (sb, nblk) in self.sb_info.iter_mut().zip(self.sb_blocks.iter()).take(self.y_sbs) { + let nblks = *nblk as usize; + if *sb == SBState::Uncoded { + for _ in 0..nblks { + self.blocks[self.blk_addr[cur_blk] >> 2].btype = VPMBType::InterNoMV; + cur_blk += 1; + } + } else { + for _ in 0..nblks/4 { + let mut coded = *sb == SBState::Coded; + if !coded { + for blk in 0..4 { + if self.blocks[self.blk_addr[cur_blk + blk] >> 2].coded { + coded = true; + break; + } + } + } + let mode = if !coded { + VPMBType::InterNoMV + } else { + br.read_cb(&codes.mbtype_cb)? + }; + for _ in 0..4 { + self.blocks[self.blk_addr[cur_blk] >> 2].btype = mode; + cur_blk += 1; + } + } + } + } + } else { + return Err(DecoderError::Bug); + } + // replicate types for chroma + let mut off_y = 0; + let mut off_u = self.y_blocks; + let mut off_v = off_u + self.mb_w * self.mb_h; + for _blk_y in 0..self.mb_h { + for blk_x in 0..self.mb_w { + let btype = self.blocks[off_y + blk_x * 2].btype; + self.blocks[off_u + blk_x].btype = btype; + self.blocks[off_v + blk_x].btype = btype; + } + off_y += self.mb_w * 2 * 2; + off_u += self.mb_w; + off_v += self.mb_w; + } + Ok(()) + } + fn vp30_unpack_mv_info(&mut self, br: &mut BitReader) -> DecoderResult<()> { + let mut last_mv = ZERO_MV; + + let mut cur_blk = 0; + for _ in 0..self.y_blocks/4 { + let baddr = self.blk_addr[cur_blk] >> 2; + if self.blocks[baddr].btype == VPMBType::InterFourMV { + let saddr = baddr.min(self.blk_addr[cur_blk + 1] >> 2).min(self.blk_addr[cur_blk + 2] >> 2).min(self.blk_addr[cur_blk + 3] >> 2); + for i in 0..4 { + let blk = &mut self.blocks[saddr + (i & 1) + (i & 2) * self.mb_w]; + if blk.coded { + blk.mv = vp30_read_mv(br)?; + } + cur_blk += 1; + } + } else { + let cur_mv; + match self.blocks[baddr].btype { + VPMBType::Intra | VPMBType::InterNoMV | VPMBType::GoldenNoMV => { + cur_mv = ZERO_MV; + }, + VPMBType::InterMV => { + cur_mv = vp30_read_mv(br)?; + last_mv = cur_mv; + }, + VPMBType::InterNearest => { + cur_mv = last_mv; + }, + _ => { // GoldenMV + cur_mv = vp30_read_mv(br)?; + }, + }; + for _ in 0..4 { + self.blocks[self.blk_addr[cur_blk] >> 2].mv = cur_mv; + cur_blk += 1; + } + } + } + Ok(()) + } + fn vp30_unpack_coeffs(&mut self, br: &mut BitReader, coef_no: usize, table: usize) -> DecoderResult<()> { + if let Codes::VP30(ref codes) = self.codes { + for blkaddr in self.blk_addr.iter() { + let blk: &mut Block = &mut self.blocks[blkaddr >> 2]; + if !blk.coded || blk.idx != coef_no { continue; } + if self.eob_run > 0 { + blk.idx = 64; + self.eob_run -= 1; + continue; + } + let cb = if coef_no == 0 { + &codes.dc_cb[table] + } else if blk.btype.is_intra() { + &codes.ac_i_cb[table] + } else { + &codes.ac_p_cb[table] + }; + let token = br.read_cb(cb)?; + expand_token(blk, br, &mut self.eob_run, coef_no, token)?; + } + Ok(()) + } else { + Err(DecoderError::Bug) + } + } + fn vp31_unpack_sb_info(&mut self, br: &mut BitReader) -> DecoderResult<()> { let mut has_uncoded = false; let mut has_partial = false; { @@ -413,16 +826,16 @@ println!("intra, ver {} (self {})", version, self.version); } Ok(()) } - fn unpack_mb_info(&mut self, br: &mut BitReader) -> DecoderResult<()> { + fn vp31_unpack_mb_info(&mut self, br: &mut BitReader) -> DecoderResult<()> { let mut modes = [VPMBType::InterNoMV; 8]; let alphabet = br.read(3)? as usize; let raw_modes = alphabet >= 7; if alphabet == 0 { - for mode in VP3_DEFAULT_MB_MODES.iter() { + for mode in VP31_DEFAULT_MB_MODES.iter() { modes[br.read(3)? as usize] = *mode; } } else if alphabet < 7 { - modes.copy_from_slice(&VP3_MB_MODES[alphabet - 1]); + modes.copy_from_slice(&VP31_MB_MODES[alphabet - 1]); } let mut cur_blk = 0; @@ -450,7 +863,7 @@ println!("intra, ver {} (self {})", version, self.version); let code = br.read_code(UintCodeType::LimitedUnary(7, 0))?; modes[code as usize] } else { - VP3_DEFAULT_MB_MODES[br.read(3)? as usize] + VP31_DEFAULT_MB_MODES[br.read(3)? as usize] }; for _ in 0..4 { self.blocks[self.blk_addr[cur_blk] >> 2].btype = mode; @@ -475,7 +888,7 @@ println!("intra, ver {} (self {})", version, self.version); } Ok(()) } - fn unpack_mv_info(&mut self, br: &mut BitReader) -> DecoderResult<()> { + fn vp31_unpack_mv_info(&mut self, br: &mut BitReader) -> DecoderResult<()> { let mut last_mv = ZERO_MV; let mut last2_mv = ZERO_MV; let read_mv = if br.read_bool()? { read_mv_raw } else { read_mv_packed }; @@ -522,136 +935,113 @@ println!("intra, ver {} (self {})", version, self.version); } Ok(()) } - fn unpack_coeffs(&mut self, br: &mut BitReader, coef_no: usize, table_y: usize, table_c: usize) -> DecoderResult<()> { - let cbs = if coef_no == 0 { - [&self.codes.dc_cb[table_y], &self.codes.dc_cb[table_c]] - } else if coef_no < 6 { - [&self.codes.ac0_cb[table_y], &self.codes.ac0_cb[table_c]] - } else if coef_no < 15 { - [&self.codes.ac1_cb[table_y], &self.codes.ac1_cb[table_c]] - } else if coef_no < 28 { - [&self.codes.ac2_cb[table_y], &self.codes.ac2_cb[table_c]] + fn vp31_unpack_coeffs(&mut self, br: &mut BitReader, coef_no: usize, table_y: usize, table_c: usize) -> DecoderResult<()> { + if let Codes::VP31(ref codes) = self.codes { + let cbs = if coef_no == 0 { + [&codes.dc_cb[table_y], &codes.dc_cb[table_c]] + } else if coef_no < 6 { + [&codes.ac0_cb[table_y], &codes.ac0_cb[table_c]] + } else if coef_no < 15 { + [&codes.ac1_cb[table_y], &codes.ac1_cb[table_c]] + } else if coef_no < 28 { + [&codes.ac2_cb[table_y], &codes.ac2_cb[table_c]] + } else { + [&codes.ac3_cb[table_y], &codes.ac3_cb[table_c]] + }; + for blkaddr in self.blk_addr.iter() { + let blk: &mut Block = &mut self.blocks[blkaddr >> 2]; + if !blk.coded || blk.idx != coef_no { continue; } + if self.eob_run > 0 { + blk.idx = 64; + self.eob_run -= 1; + continue; + } + let cb = if (blkaddr & 3) == 0 { cbs[0] } else { cbs[1] }; + let token = br.read_cb(cb)?; + expand_token(blk, br, &mut self.eob_run, coef_no, token)?; + } + Ok(()) + } else { + Err(DecoderError::Bug) + } + } + fn decode_vp30(&mut self, br: &mut BitReader, frm: &mut NASimpleVideoFrame) -> DecoderResult<()> { + for blk in self.blocks.iter_mut() { + blk.coeffs = [0; 64]; + blk.idx = 0; + blk.coded = false; + blk.has_ac = false; + } + if self.is_intra { + for sb in self.sb_info.iter_mut() { *sb = SBState::Coded; } + for blk in self.blocks.iter_mut() { + blk.btype = VPMBType::Intra; + blk.coded = true; + } + } else { + if self.shuf.get_last().is_none() || self.shuf.get_golden().is_none() { + return Err(DecoderError::MissingReference); + } + self.vp30_unpack_sb_info(br)?; + self.vp30_unpack_mb_info(br)?; + self.vp30_unpack_mv_info(br)?; + } + let dc_quant = VP30_DC_SCALES[self.quant] * 10; + let ac_quant = VP30_AC_SCALES[self.quant]; + rescale_qmat(&mut self.qmat_y, VP3_QMAT_Y, dc_quant, ac_quant, 2); + rescale_qmat(&mut self.qmat_c, VP3_QMAT_C, dc_quant, ac_quant, 2); + rescale_qmat(&mut self.qmat_inter, VP3_QMAT_INTER, dc_quant, ac_quant, 4); + if self.quant == 10 { + self.qmat_y[29] = 980; + self.qmat_y[58] = 1636; + self.qmat_y[59] = 1964; + } else if self.quant == 31 { + self.qmat_y[58] = 456; + } else if self.quant == 44 { + self.qmat_y[58] = 224; + } + + let table = if ac_quant <= 50 { + 0 + } else if ac_quant <= 150 { + 1 + } else if ac_quant <= 300 { + 2 + } else if ac_quant <= 600 { + 3 } else { - [&self.codes.ac3_cb[table_y], &self.codes.ac3_cb[table_c]] + 4 }; + + self.eob_run = 0; + self.vp30_unpack_coeffs(br, 0, table)?; + let mut last_dc_i = 0; + let mut last_dc_p = 0; for blkaddr in self.blk_addr.iter() { let blk: &mut Block = &mut self.blocks[blkaddr >> 2]; - if !blk.coded || blk.idx != coef_no { continue; } - if self.eob_run > 0 { - blk.idx = 64; - self.eob_run -= 1; - continue; - } - let cb = if (blkaddr & 3) == 0 { cbs[0] } else { cbs[1] }; - let token = br.read_cb(cb)?; - match token { - // EOBs - 0 | 1 | 2 => { self.eob_run = (token as usize) + 1; }, - 3 | 4 | 5 => { - let bits = token - 1; - self.eob_run = (br.read(bits)? as usize) + (1 << bits); - }, - 6 => { self.eob_run = br.read(12)? as usize; }, - // zero runs - 7 | 8 => { - let bits = if token == 7 { 3 } else { 6 }; - let run = (br.read(bits)? as usize) + 1; - blk.idx += run; - validate!(blk.idx <= 64); - }, - // single coefficients - 9 | 10 | 11 | 12 => { - let val = (i16::from(token) - 7) >> 1; - if (token & 1) == 1 { - blk.coeffs[ZIGZAG[blk.idx]] = val; - } else { - blk.coeffs[ZIGZAG[blk.idx]] = -val; - } - blk.idx += 1; - }, - 13 | 14 | 15 | 16 => { - let val = i16::from(token) - 10; - if !br.read_bool()? { - blk.coeffs[ZIGZAG[blk.idx]] = val; - } else { - blk.coeffs[ZIGZAG[blk.idx]] = -val; - } - blk.idx += 1; - }, - 17 | 18 | 19 | 20 | 21 | 22 => { - let add_bits = if token == 22 { 9 } else { token - 16 }; - let sign = br.read_bool()?; - let val = (br.read(add_bits)? as i16) + VP3_LITERAL_BASE[(token - 17) as usize]; - if !sign { - blk.coeffs[ZIGZAG[blk.idx]] = val; - } else { - blk.coeffs[ZIGZAG[blk.idx]] = -val; - } - blk.idx += 1; - } - // zero run plus coefficient - 23 | 24 | 25 | 26 | 27 => { - blk.idx += (token - 22) as usize; - validate!(blk.idx < 64); - if !br.read_bool()? { - blk.coeffs[ZIGZAG[blk.idx]] = 1; - } else { - blk.coeffs[ZIGZAG[blk.idx]] = -1; - } - blk.idx += 1; - }, - 28 | 29 => { - let run_bits = token - 26; - if token == 28 { - blk.idx += 6; - } else { - blk.idx += 10; - } - let sign = br.read_bool()?; - blk.idx += br.read(run_bits)? as usize; - validate!(blk.idx < 64); - if !sign { - blk.coeffs[ZIGZAG[blk.idx]] = 1; - } else { - blk.coeffs[ZIGZAG[blk.idx]] = -1; - } - blk.idx += 1; - }, - 30 => { - blk.idx += 1; - validate!(blk.idx < 64); - let sign = br.read_bool()?; - let val = (br.read(1)? as i16) + 2; - if !sign { - blk.coeffs[ZIGZAG[blk.idx]] = val; - } else { - blk.coeffs[ZIGZAG[blk.idx]] = -val; - } - blk.idx += 1; - }, - _ => { - let sign = br.read_bool()?; - let val = (br.read(1)? as i16) + 2; - blk.idx += (br.read(1)? as usize) + 2; - validate!(blk.idx < 64); - if !sign { - blk.coeffs[ZIGZAG[blk.idx]] = val; - } else { - blk.coeffs[ZIGZAG[blk.idx]] = -val; - } - blk.idx += 1; - }, - }; - if self.eob_run > 0 { - blk.idx = 64; - self.eob_run -= 1; - } else if coef_no > 0 { - blk.has_ac = true; + if !blk.coded { continue; } + if blk.btype.is_intra() { + blk.coeffs[0] += last_dc_i; + last_dc_i = blk.coeffs[0]; + } else { + blk.coeffs[0] += last_dc_p; + last_dc_p = blk.coeffs[0]; } } + + for coef_no in 1..64 { + self.vp30_unpack_coeffs(br, coef_no, table)?; + } + + if self.is_intra { + self.output_blocks_intra(frm); + } else { + self.output_blocks_inter(frm); + } + Ok(()) } - fn decode_vp3(&mut self, br: &mut BitReader, frm: &mut NASimpleVideoFrame) -> DecoderResult<()> { + fn decode_vp31(&mut self, br: &mut BitReader, frm: &mut NASimpleVideoFrame) -> DecoderResult<()> { for blk in self.blocks.iter_mut() { blk.coeffs = [0; 64]; blk.idx = 0; @@ -668,26 +1058,26 @@ println!("intra, ver {} (self {})", version, self.version); if self.shuf.get_last().is_none() || self.shuf.get_golden().is_none() { return Err(DecoderError::MissingReference); } - self.unpack_sb_info(br)?; - self.unpack_mb_info(br)?; - self.unpack_mv_info(br)?; + self.vp31_unpack_sb_info(br)?; + self.vp31_unpack_mb_info(br)?; + self.vp31_unpack_mv_info(br)?; } - let dc_quant = VP3_DC_SCALES[self.quant]; - let ac_quant = VP3_AC_SCALES[self.quant]; - rescale_qmat(&mut self.qmat_y, VP3_QMAT_Y, dc_quant, ac_quant); - rescale_qmat(&mut self.qmat_c, VP3_QMAT_C, dc_quant, ac_quant); - rescale_qmat(&mut self.qmat_inter, VP3_QMAT_INTER, dc_quant, ac_quant); + let dc_quant = VP31_DC_SCALES[self.quant]; + let ac_quant = VP31_AC_SCALES[self.quant]; + rescale_qmat(&mut self.qmat_y, VP3_QMAT_Y, dc_quant, ac_quant, 2); + rescale_qmat(&mut self.qmat_c, VP3_QMAT_C, dc_quant, ac_quant, 2); + rescale_qmat(&mut self.qmat_inter, VP3_QMAT_INTER, dc_quant, ac_quant, 4); self.eob_run = 0; let dc_table_y = br.read(4)? as usize; let dc_table_c = br.read(4)? as usize; - self.unpack_coeffs(br, 0, dc_table_y, dc_table_c)?; + self.vp31_unpack_coeffs(br, 0, dc_table_y, dc_table_c)?; self.restore_dcs(); let ac_table_y = br.read(4)? as usize; let ac_table_c = br.read(4)? as usize; for coef_no in 1..64 { - self.unpack_coeffs(br, coef_no, ac_table_y, ac_table_c)?; + self.vp31_unpack_coeffs(br, coef_no, ac_table_y, ac_table_c)?; } if self.is_intra { @@ -696,7 +1086,7 @@ println!("intra, ver {} (self {})", version, self.version); self.output_blocks_inter(frm); } if self.loop_str > 0 { - self.loop_filter(frm); + self.vp31_loop_filter(frm); } Ok(()) @@ -725,10 +1115,10 @@ unimplemented!(); let mut pred = 0i32; for i in 0..4 { if (pp & (1 << i)) != 0 { - pred += (preds[i] as i32) * (VP3_DC_WEIGHTS[pp][i] as i32); + pred += (preds[i] as i32) * (VP31_DC_WEIGHTS[pp][i] as i32); } } - pred /= VP3_DC_WEIGHTS[pp][4] as i32; + pred /= VP31_DC_WEIGHTS[pp][4] as i32; if (pp & 7) == 7 { if (pred - preds[2]).abs() > 128 { return preds[2] as i16; } if (pred - preds[0]).abs() > 128 { return preds[0] as i16; } @@ -826,7 +1216,7 @@ unimplemented!(); copy_block(frm, src.clone(), 1, bx * 4, by * 4, mx >> 1, my >> 1, 8, 8, 0, 1, mode, VP3_INTERP_FUNCS); copy_block(frm, src.clone(), 2, bx * 4, by * 4, mx >> 1, my >> 1, 8, 8, 0, 1, mode, VP3_INTERP_FUNCS); } - blk_idx += bstride; + blk_idx += bstride * 2; } let mut biter = self.blocks.iter_mut(); @@ -898,23 +1288,23 @@ unimplemented!(); } } } - fn loop_filter(&mut self, frm: &mut NASimpleVideoFrame) { + fn vp31_loop_filter(&mut self, frm: &mut NASimpleVideoFrame) { let mut blk_idx = 0; let blk_w = self.mb_w * 2; for by in 0..self.mb_h*2 { for bx in 0..blk_w { let blk = &self.blocks[blk_idx + bx]; if (bx > 0) && blk.coded { - vp3_loop_filter_v(frm, bx * 8, by * 8, 0, self.loop_str); + vp31_loop_filter_v(frm, bx * 8, by * 8, 0, self.loop_str); } if (by > 0) && blk.coded { - vp3_loop_filter_h(frm, bx * 8, by * 8, 0, self.loop_str); + vp31_loop_filter_h(frm, bx * 8, by * 8, 0, self.loop_str); } if (bx < blk_w - 1) && !self.blocks[blk_idx + bx + 1].coded { - vp3_loop_filter_v(frm, bx * 8 + 8, by * 8, 0, self.loop_str); + vp31_loop_filter_v(frm, bx * 8 + 8, by * 8, 0, self.loop_str); } if (by < self.mb_h * 2 - 1) && !self.blocks[blk_idx + bx + blk_w].coded { - vp3_loop_filter_h(frm, bx * 8, by * 8 + 8, 0, self.loop_str); + vp31_loop_filter_h(frm, bx * 8, by * 8 + 8, 0, self.loop_str); } } blk_idx += blk_w; @@ -933,11 +1323,14 @@ unimplemented!(); let sb_h_c = ((self.height >> 1) + 31) >> 5; self.y_sbs = sb_w_y * sb_h_y; let tot_sb = sb_w_y * sb_h_y + 2 * sb_w_c * sb_h_c; + let tot_mb = self.mb_w * self.mb_h * 2 + ((self.mb_w + 1) & !1) * ((self.mb_h + 1) & !1) * 2; let bw = self.width >> 3; let bh = self.height >> 3; let tot_blk = bw * bh * 3 / 2; self.sb_info.resize(tot_sb, SBState::Uncoded); self.sb_blocks = Vec::with_capacity(tot_sb); + self.mb_blocks = Vec::with_capacity(tot_mb); + self.sb_mbs = Vec::with_capacity(tot_sb); self.blk_addr = Vec::with_capacity(tot_blk); self.y_blocks = bw * bh; let mut base_idx = 0; @@ -950,6 +1343,23 @@ unimplemented!(); let blk_h = h >> 3; for y in 0..sb_h { for x in 0..sb_w { + let mut nmbs = 0; + for mb_no in 0..4 { + let bx = x * 4 + HILBERT_ORDER[mb_no * 4][0]; + let by = y * 4 + HILBERT_ORDER[mb_no * 4][1]; + if (bx >= blk_w) || (by >= blk_h) { continue; } + let mut nblocks = 0; + for blk_no in 0..4 { + let bx = x * 4 + HILBERT_ORDER[mb_no * 4 + blk_no][0]; + let by = y * 4 + HILBERT_ORDER[mb_no * 4 + blk_no][1]; + if (bx >= blk_w) || (by >= blk_h) { continue; } + nblocks += 1; + } + self.mb_blocks.push(nblocks); + nmbs += 1; + } + self.sb_mbs.push(nmbs); + let mut nblocks = 0; for blk_no in 0..16 { let bx = x * 4 + HILBERT_ORDER[blk_no][0]; @@ -965,6 +1375,7 @@ unimplemented!(); base_idx += blk_w * blk_h; } self.blocks.resize(tot_blk, Block::new()); + self.mb_coded.resize(tot_mb, false); } } @@ -1006,10 +1417,11 @@ impl NADecoder for VP34Decoder { } let mut buf = ret.unwrap(); let mut dframe = NASimpleVideoFrame::from_video_buf(&mut buf).unwrap(); - if self.version == 3 { - self.decode_vp3(&mut br, &mut dframe)?; - } else { - self.decode_vp4()?; + match self.version { + 30 => self.decode_vp30(&mut br, &mut dframe)?, + 31 => self.decode_vp31(&mut br, &mut dframe)?, + 4 => self.decode_vp4()?, + _ => return Err(DecoderError::Bug), } if self.is_intra { @@ -1041,17 +1453,27 @@ mod test { use nihav_commonfmt::demuxers::generic_register_all_demuxers; #[test] - fn test_vp3() { + fn test_vp30() { + let mut dmx_reg = RegisteredDemuxers::new(); + generic_register_all_demuxers(&mut dmx_reg); + let mut dec_reg = RegisteredDecoders::new(); + duck_register_all_codecs(&mut dec_reg); + + let file = "assets/Duck/vp30-logo.avi"; + test_file_decoding("avi", file, Some(23), true, false, Some("vp30"), &dmx_reg, &dec_reg); + } + + #[test] + fn test_vp31() { let mut dmx_reg = RegisteredDemuxers::new(); generic_register_all_demuxers(&mut dmx_reg); let mut dec_reg = RegisteredDecoders::new(); duck_register_all_codecs(&mut dec_reg); -// let file = "assets/Duck/vp30-logo.avi"; let file = "assets/Duck/vp31.avi"; // let file = "assets/Duck/vp31_crash.avi"; // let file = "assets/Duck/01-vp31-0500.avi"; - test_file_decoding("avi", file, Some(3), true, false, Some("vp3"), &dmx_reg, &dec_reg); + test_file_decoding("avi", file, Some(3), true, false, Some("vp31"), &dmx_reg, &dec_reg); //panic!("end"); } @@ -1074,7 +1496,7 @@ const HILBERT_ORDER: [[usize; 2]; 16] = [ [ 3, 1 ], [ 2, 1 ], [ 2, 0 ], [ 3, 0 ] ]; -const VP3_LOOP_STRENGTH: [i16; 64] = [ +const VP31_LOOP_STRENGTH: [i16; 64] = [ 30, 25, 20, 20, 15, 15, 14, 14, 13, 13, 12, 12, 11, 11, 10, 10, 9, 9, 8, 8, 7, 7, 7, 7, @@ -1085,12 +1507,12 @@ const VP3_LOOP_STRENGTH: [i16; 64] = [ 0, 0, 0, 0, 0, 0, 0, 0 ]; -const VP3_DEFAULT_MB_MODES: [VPMBType; 8] = [ +const VP31_DEFAULT_MB_MODES: [VPMBType; 8] = [ VPMBType::InterNoMV, VPMBType::Intra, VPMBType::InterMV, VPMBType::InterNearest, VPMBType::InterNear, VPMBType::GoldenNoMV, VPMBType::GoldenMV, VPMBType::InterFourMV ]; -const VP3_MB_MODES: [[VPMBType; 8]; 6] = [ +const VP31_MB_MODES: [[VPMBType; 8]; 6] = [ [ VPMBType::InterNearest, VPMBType::InterNear, VPMBType::InterMV, VPMBType::InterNoMV, VPMBType::Intra, VPMBType::GoldenNoMV, VPMBType::GoldenMV, VPMBType::InterFourMV @@ -1114,7 +1536,7 @@ const VP3_MB_MODES: [[VPMBType; 8]; 6] = [ const VP3_LITERAL_BASE: [i16; 6] = [ 7, 9, 13, 21, 37, 69 ]; -const VP3_AC_SCALES: [i16; 64] = [ +const VP31_AC_SCALES: [i16; 64] = [ 500, 450, 400, 370, 340, 310, 285, 265, 245, 225, 210, 195, 185, 180, 170, 160, 150, 145, 135, 130, 125, 115, 110, 107, @@ -1125,7 +1547,7 @@ const VP3_AC_SCALES: [i16; 64] = [ 21, 19, 18, 17, 15, 13, 12, 10 ]; -const VP3_DC_SCALES: [i16; 64] = [ +const VP31_DC_SCALES: [i16; 64] = [ 220, 200, 190, 180, 170, 170, 160, 160, 150, 150, 140, 140, 130, 130, 120, 120, 110, 110, 100, 100, 90, 90, 90, 80, @@ -1180,7 +1602,7 @@ const ZIGZAG: [usize; 64] = [ 53, 60, 61, 54, 47, 55, 62, 63 ]; -const VP3_DC_CODES: [[u16; 32]; 16] = [ +const VP31_DC_CODES: [[u16; 32]; 16] = [ [ 0x002D, 0x0026, 0x0166, 0x004E, 0x02CE, 0x059E, 0x027D, 0x0008, 0x04F9, 0x000F, 0x000E, 0x001B, 0x0006, 0x0008, 0x0005, 0x001A, @@ -1264,7 +1686,7 @@ const VP3_DC_CODES: [[u16; 32]; 16] = [ ] ]; -const VP3_DC_BITS: [[u8; 32]; 16] = [ +const VP31_DC_BITS: [[u8; 32]; 16] = [ [ 6, 7, 9, 8, 10, 11, 11, 5, 12, 4, 4, 5, 4, 4, 4, 5, 5, 4, 4, 3, 3, 4, 5, 6, 6, 8, 12, 11, 9, 10, 6, 7, @@ -1316,7 +1738,7 @@ const VP3_DC_BITS: [[u8; 32]; 16] = [ ] ]; -const VP3_AC_CAT0_CODES: [[u16; 32]; 16] = [ +const VP31_AC_CAT0_CODES: [[u16; 32]; 16] = [ [ 0x0008, 0x0025, 0x017A, 0x02F7, 0x0BDB, 0x17B4, 0x2F6B, 0x001D, 0x2F6A, 0x0008, 0x0007, 0x0001, 0x0002, 0x000A, 0x0006, 0x0000, @@ -1400,7 +1822,7 @@ const VP3_AC_CAT0_CODES: [[u16; 32]; 16] = [ ] ]; -const VP3_AC_CAT0_BITS: [[u8; 32]; 16] = [ +const VP31_AC_CAT0_BITS: [[u8; 32]; 16] = [ [ 5, 7, 9, 10, 12, 13, 14, 5, 14, 4, 4, 4, 4, 4, 4, 4, 5, 4, 4, 4, 4, 4, 5, 5, 6, 7, 7, 8, 7, 11, 5, 7, @@ -1452,7 +1874,7 @@ const VP3_AC_CAT0_BITS: [[u8; 32]; 16] = [ ] ]; -const VP3_AC_CAT1_CODES: [[u16; 32]; 16] = [ +const VP31_AC_CAT1_CODES: [[u16; 32]; 16] = [ [ 0x000B, 0x002B, 0x0054, 0x01B7, 0x06D9, 0x0DB1, 0x0DB0, 0x0002, 0x00AB, 0x0009, 0x000A, 0x0007, 0x0008, 0x000F, 0x000C, 0x0003, @@ -1536,7 +1958,7 @@ const VP3_AC_CAT1_CODES: [[u16; 32]; 16] = [ ] ]; -const VP3_AC_CAT1_BITS: [[u8; 32]; 16] = [ +const VP31_AC_CAT1_BITS: [[u8; 32]; 16] = [ [ 5, 7, 8, 9, 11, 12, 12, 4, 9, 4, 4, 4, 4, 4, 4, 4, 5, 4, 4, 4, 5, 6, 9, 4, 5, 6, 7, 8, 6, 10, 5, 6, @@ -1588,7 +2010,7 @@ const VP3_AC_CAT1_BITS: [[u8; 32]; 16] = [ ] ]; -const VP3_AC_CAT2_CODES: [[u16; 32]; 16] = [ +const VP31_AC_CAT2_CODES: [[u16; 32]; 16] = [ [ 0x0003, 0x0009, 0x00D0, 0x01A3, 0x0344, 0x0D14, 0x1A2B, 0x0004, 0x0015, 0x0000, 0x000F, 0x000B, 0x000C, 0x000E, 0x0009, 0x001B, @@ -1672,7 +2094,7 @@ const VP3_AC_CAT2_CODES: [[u16; 32]; 16] = [ ] ]; -const VP3_AC_CAT2_BITS: [[u8; 32]; 16] = [ +const VP31_AC_CAT2_BITS: [[u8; 32]; 16] = [ [ 4, 6, 8, 9, 10, 12, 13, 4, 7, 3, 4, 4, 4, 4, 4, 5, 5, 5, 5, 6, 7, 11, 13, 4, 5, 6, 6, 7, 6, 6, 4, 5, @@ -1724,7 +2146,7 @@ const VP3_AC_CAT2_BITS: [[u8; 32]; 16] = [ ] ]; -const VP3_AC_CAT3_CODES: [[u16; 32]; 16] = [ +const VP31_AC_CAT3_CODES: [[u16; 32]; 16] = [ [ 0x0000, 0x0010, 0x0072, 0x0071, 0x0154, 0x0AAB, 0x0AA8, 0x0014, 0x0070, 0x0002, 0x0003, 0x000C, 0x000B, 0x0003, 0x0011, 0x0073, @@ -1808,7 +2230,7 @@ const VP3_AC_CAT3_CODES: [[u16; 32]; 16] = [ ] ]; -const VP3_AC_CAT3_BITS: [[u8; 32]; 16] = [ +const VP31_AC_CAT3_BITS: [[u8; 32]; 16] = [ [ 3, 5, 7, 7, 9, 12, 12, 5, 7, 3, 3, 4, 4, 4, 5, 7, 7, 8, 10, 13, 13, 13, 13, 4, 5, 5, 6, 6, 4, 6, 5, 5, @@ -1860,7 +2282,7 @@ const VP3_AC_CAT3_BITS: [[u8; 32]; 16] = [ ] ]; -const VP3_DC_WEIGHTS: [[i16; 5]; 16] = [ +const VP31_DC_WEIGHTS: [[i16; 5]; 16] = [ [ 0, 0, 0, 0, 0 ], [ 1, 0, 0, 0, 1 ], [ 0, 1, 0, 0, 1 ], @@ -1881,3 +2303,170 @@ const VP3_DC_WEIGHTS: [[i16; 5]; 16] = [ [ 0, 3, 10, 3, 16 ], [ 29, -26, 29, 0, 32 ], ]; + +const VP30_DC_SCALES: [i16; 64] = [ + 24, 20, 20, 20, 20, 20, 20, 20, + 19, 19, 19, 19, 18, 18, 18, 18, + 17, 17, 17, 17, 16, 16, 15, 15, + 14, 14, 13, 13, 12, 12, 11, 11, + 10, 10, 9, 9, 8, 8, 7, 7, + 6, 6, 6, 6, 5, 5, 5, 5, + 4, 4, 4, 4, 3, 3, 3, 3, + 2, 2, 2, 2, 1, 1, 1, 1 +]; + +const VP30_AC_SCALES: [i16; 64] = [ + 3000, 2500, 2000, 1500, 1200, 1000, 900, 800, + 750, 700, 650, 630, 600, 550, 500, 450, + 410, 380, 350, 320, 290, 260, 240, 220, + 200, 180, 165, 150, 140, 130, 120, 115, + 110, 100, 95, 90, 85, 80, 75, 70, + 67, 65, 63, 61, 57, 55, 53, 50, + 49, 46, 44, 42, 39, 36, 33, 30, + 27, 24, 21, 19, 17, 15, 12, 10 +]; + +const VP30_DC_CODES: [[u16; 32]; 5] = [ + [ + 0x0005, 0x002D, 0x0004, 0x0009, 0x0088, 0x0225, 0x0224, 0x0005, + 0x0011, 0x0007, 0x0006, 0x0009, 0x000A, 0x0007, 0x0017, 0x000C, + 0x002C, 0x0005, 0x0008, 0x0003, 0x0012, 0x0010, 0x0113, 0x0003, + 0x0010, 0x0000, 0x0013, 0x001A, 0x0023, 0x0045, 0x0001, 0x001B + ], [ + 0x000B, 0x0012, 0x0029, 0x0010, 0x000D, 0x00A2, 0x0020, 0x0009, + 0x0050, 0x0007, 0x0006, 0x0006, 0x0005, 0x0002, 0x0008, 0x0027, + 0x0005, 0x0022, 0x0023, 0x0057, 0x00A3, 0x0011, 0x0021, 0x0007, + 0x0000, 0x0009, 0x002A, 0x0003, 0x0007, 0x0026, 0x000C, 0x0056 + ], [ + 0x000D, 0x0018, 0x0009, 0x0017, 0x0033, 0x0056, 0x00F7, 0x00F1, + 0x007A, 0x0000, 0x0007, 0x0009, 0x0008, 0x0005, 0x000D, 0x002D, + 0x0010, 0x001D, 0x001C, 0x0057, 0x00CB, 0x00F6, 0x00F0, 0x0014, + 0x000C, 0x002C, 0x0011, 0x001F, 0x002A, 0x0064, 0x00CA, 0x0079 + ], [ + 0x000F, 0x001A, 0x0013, 0x001B, 0x003B, 0x0072, 0x01D3, 0x0707, + 0x0E0D, 0x0001, 0x0000, 0x000C, 0x000B, 0x0008, 0x0012, 0x002A, + 0x0073, 0x0028, 0x0075, 0x0056, 0x0052, 0x01C0, 0x0E0C, 0x0071, + 0x0057, 0x00E1, 0x00A6, 0x00E8, 0x00A7, 0x03A5, 0x03A4, 0x0382 + ], [ + 0x000F, 0x001B, 0x0014, 0x001D, 0x0010, 0x0073, 0x00E2, 0x023C, + 0x11C9, 0x0001, 0x0000, 0x000C, 0x000B, 0x0009, 0x0015, 0x0035, + 0x0072, 0x0034, 0x0022, 0x0070, 0x0046, 0x011D, 0x11C8, 0x01C7, + 0x01C6, 0x0238, 0x047E, 0x023E, 0x0473, 0x08E5, 0x023D, 0x047F + ] +]; +const VP30_DC_BITS: [[u8; 32]; 5] = [ + [ + 4, 6, 6, 6, 8, 10, 10, 6, 7, 3, 3, 4, 4, 4, 5, 5, + 6, 5, 5, 5, 6, 7, 9, 4, 5, 5, 6, 6, 6, 7, 5, 6 + ], [ + 4, 5, 6, 5, 6, 8, 9, 7, 7, 3, 3, 4, 4, 4, 5, 6, + 6, 6, 6, 7, 8, 8, 9, 4, 4, 5, 6, 5, 5, 6, 6, 7 + ], [ + 4, 5, 5, 5, 6, 7, 9, 9, 8, 2, 3, 4, 4, 4, 5, 6, + 6, 6, 6, 7, 8, 9, 9, 5, 5, 6, 6, 6, 6, 7, 8, 8 + ], [ + 4, 5, 5, 5, 6, 7, 9, 11, 12, 2, 2, 4, 4, 4, 5, 6, + 7, 6, 7, 7, 7, 9, 12, 7, 7, 8, 8, 8, 8, 10, 10, 10 + ], [ + 4, 5, 5, 5, 5, 7, 8, 10, 13, 2, 2, 4, 4, 4, 5, 6, + 7, 6, 6, 7, 7, 9, 13, 9, 9, 10, 11, 10, 11, 12, 10, 11 + ] +]; +const VP30_AC_INTRA_CODES: [[u16; 32]; 5] = [ + [ + 0x0008, 0x0033, 0x0008, 0x004B, 0x0089, 0x0221, 0x0220, 0x001F, + 0x0045, 0x0000, 0x000E, 0x000B, 0x000A, 0x000D, 0x0006, 0x001E, + 0x000A, 0x0018, 0x0013, 0x0005, 0x0009, 0x0046, 0x0111, 0x0007, + 0x000B, 0x0032, 0x0010, 0x004A, 0x0024, 0x0047, 0x0003, 0x0009 + ], [ + 0x000E, 0x000E, 0x007B, 0x001E, 0x007E, 0x03EF, 0x07DD, 0x0018, + 0x00FA, 0x0002, 0x0000, 0x000A, 0x0008, 0x000B, 0x0003, 0x0012, + 0x0033, 0x000C, 0x003C, 0x001A, 0x007F, 0x01F6, 0x07DC, 0x000D, + 0x0013, 0x0004, 0x001B, 0x007A, 0x0032, 0x007C, 0x001F, 0x0005 + ], [ + 0x0000, 0x0018, 0x0034, 0x000C, 0x006A, 0x01F9, 0x07EA, 0x0016, + 0x0FD7, 0x0002, 0x0001, 0x000A, 0x0009, 0x0007, 0x001B, 0x003E, + 0x0020, 0x0021, 0x006B, 0x01FB, 0x03F4, 0x1FAD, 0x1FAC, 0x000E, + 0x0019, 0x0011, 0x002F, 0x007F, 0x002E, 0x01F8, 0x001E, 0x000D + ], [ + 0x000E, 0x0016, 0x002E, 0x0003, 0x006E, 0x008B, 0x0113, 0x0018, + 0x0221, 0x0001, 0x0002, 0x000A, 0x0009, 0x0007, 0x001A, 0x0002, + 0x001B, 0x0023, 0x006F, 0x008A, 0x0111, 0x0441, 0x0440, 0x000F, + 0x0019, 0x0010, 0x0036, 0x001A, 0x002F, 0x0112, 0x0000, 0x000C + ], [ + 0x000E, 0x000F, 0x001B, 0x0033, 0x005A, 0x00B6, 0x0008, 0x001A, + 0x004D, 0x0001, 0x0002, 0x000A, 0x0009, 0x0008, 0x001B, 0x0003, + 0x002C, 0x002E, 0x0005, 0x00B7, 0x0027, 0x0099, 0x0098, 0x000F, + 0x0018, 0x000E, 0x002F, 0x001A, 0x0032, 0x0012, 0x0000, 0x000C + ] +]; +const VP30_AC_INTRA_BITS: [[u8; 32]; 5] = [ + [ + 4, 6, 6, 7, 9, 11, 11, 5, 8, 3, 4, 4, 4, 4, 4, 5, + 5, 5, 5, 5, 6, 8, 10, 4, 5, 6, 6, 7, 6, 8, 4, 5 + ], [ + 4, 5, 7, 6, 8, 11, 12, 5, 9, 3, 3, 4, 4, 4, 4, 5, + 6, 5, 6, 6, 8, 10, 12, 4, 5, 5, 6, 7, 6, 8, 5, 5 + ], [ + 3, 5, 6, 5, 7, 9, 11, 5, 12, 3, 3, 4, 4, 4, 5, 6, + 6, 6, 7, 9, 10, 13, 13, 4, 5, 5, 6, 7, 6, 9, 5, 5 + ], [ + 4, 5, 6, 5, 7, 8, 9, 5, 10, 3, 3, 4, 4, 4, 5, 5, + 6, 6, 7, 8, 9, 11, 11, 4, 5, 5, 6, 6, 6, 9, 4, 5 + ], [ + 4, 5, 6, 6, 7, 8, 7, 5, 10, 3, 3, 4, 4, 4, 5, 5, + 6, 6, 6, 8, 9, 11, 11, 4, 5, 5, 6, 6, 6, 8, 4, 5 + ] +]; +const VP30_AC_INTER_CODES: [[u16; 32]; 5] = [ + [ + 0x000D, 0x0038, 0x0061, 0x0060, 0x0393, 0x1C95, 0x1C94, 0x0014, + 0x0073, 0x0001, 0x0000, 0x000B, 0x0009, 0x001D, 0x000E, 0x0022, + 0x0046, 0x0047, 0x00E5, 0x01C8, 0x0724, 0x1C97, 0x1C96, 0x000F, + 0x0006, 0x0019, 0x000F, 0x0031, 0x0004, 0x0010, 0x0005, 0x0015 + ], [ + 0x0004, 0x001B, 0x0030, 0x0034, 0x00D5, 0x06B3, 0x3595, 0x0031, + 0x001A, 0x0002, 0x0001, 0x001F, 0x001E, 0x000C, 0x001B, 0x00D7, + 0x00D4, 0x01AD, 0x0358, 0x0D64, 0x3594, 0x3597, 0x3596, 0x0000, + 0x000A, 0x001D, 0x0017, 0x0039, 0x0007, 0x0019, 0x0016, 0x0038 + ], [ + 0x0005, 0x0009, 0x001A, 0x001E, 0x001F, 0x00E2, 0x038E, 0x0070, + 0x003B, 0x0001, 0x0000, 0x0019, 0x0018, 0x001E, 0x003A, 0x01C6, + 0x071E, 0x0E3E, 0x1C7E, 0x71FD, 0x71FC, 0x71FF, 0x71FE, 0x0002, + 0x0008, 0x001D, 0x001B, 0x0039, 0x001F, 0x000D, 0x000C, 0x001C + ], [ + 0x0003, 0x000B, 0x001C, 0x000D, 0x0004, 0x000A, 0x0076, 0x00E8, + 0x01DC, 0x0001, 0x0000, 0x0033, 0x0032, 0x00E9, 0x03BB, 0x0774, + 0x1DD5, 0x3BAD, 0x3BAC, 0x3BAF, 0x3BAE, 0x3BA9, 0x3BA8, 0x0004, + 0x000A, 0x001F, 0x001E, 0x0030, 0x000B, 0x0075, 0x0031, 0x00EF + ], [ + 0x0009, 0x001E, 0x000F, 0x000E, 0x000C, 0x0008, 0x0001, 0x00E3, + 0x00E2, 0x0002, 0x0000, 0x003A, 0x0039, 0x0070, 0x01DC, 0x0776, + 0x0775, 0x0EEF, 0x0EE8, 0x1DD3, 0x1DD2, 0x1DDD, 0x1DDC, 0x0005, + 0x001F, 0x001B, 0x0006, 0x006A, 0x0034, 0x0076, 0x006B, 0x00EF + ] +]; +const VP30_AC_INTER_BITS: [[u8; 32]; 5] = [ + [ + 4, 6, 7, 7, 10, 13, 13, 5, 7, 3, 3, 4, 4, 5, 5, 6, + 7, 7, 8, 9, 11, 13, 13, 4, 4, 5, 5, 6, 4, 5, 4, 5 + ], [ + 3, 5, 6, 6, 8, 11, 14, 6, 6, 3, 3, 5, 5, 5, 6, 8, + 8, 9, 10, 12, 14, 14, 14, 3, 4, 5, 5, 6, 4, 5, 5, 6 + ], [ + 3, 4, 5, 5, 6, 8, 10, 7, 7, 3, 3, 5, 5, 6, 7, 9, + 11, 12, 13, 15, 15, 15, 15, 3, 4, 5, 5, 6, 5, 5, 5, 6 + ], [ + 3, 4, 5, 4, 4, 5, 7, 8, 9, 3, 3, 6, 6, 8, 10, 11, + 13, 14, 14, 14, 14, 14, 14, 3, 4, 5, 5, 6, 5, 7, 6, 8 + ], [ + 4, 5, 5, 4, 4, 4, 3, 9, 9, 3, 3, 7, 7, 8, 10, 12, + 12, 13, 13, 14, 14, 14, 14, 3, 5, 5, 4, 7, 6, 8, 7, 9 + ] +]; +const VP30_MBTYPE_CODES: [u8; 7] = [ 0x00, 0x08, 0x0A, 0x03, 0x0B, 0x13, 0x12 ]; +const VP30_MBTYPE_BITS: [u8; 7] = [ 1, 4, 4, 2, 4, 5, 5 ]; +const VP30_MBTYPE_SYMS: [VPMBType; 7] = [ + VPMBType::InterNoMV, VPMBType::Intra, VPMBType::InterMV, VPMBType::InterNearest, + VPMBType::GoldenNoMV, VPMBType::GoldenMV, VPMBType::InterFourMV +];