--- /dev/null
+use io::bitreader::*;
+use io::codebook::*;
+use formats;
+use super::super::*;
+use super::*;
+use super::decoder::*;
+use super::data::*;
+
+#[allow(dead_code)]
+struct Tables {
+ intra_mcbpc_cb: Codebook<u8>,
+ inter_mcbpc_cb: Codebook<u8>,
+ cbpy_cb: Codebook<u8>,
+ rl_cb: Codebook<H263RLSym>,
+ aic_rl_cb: Codebook<H263RLSym>,
+ mv_cb: Codebook<u8>,
+ luma_dc_cb: Codebook<u8>,
+ chroma_dc_cb: Codebook<u8>,
+}
+
+struct RealVideo10Decoder {
+ info: Rc<NACodecInfo>,
+ dec: H263BaseDecoder,
+ tables: Tables,
+ w: usize,
+ h: usize,
+ new_ver: bool,
+}
+
+struct RealVideo10BR<'a> {
+ br: BitReader<'a>,
+ tables: &'a Tables,
+ num_slices: usize,
+ slice_no: usize,
+ slice_off: Vec<u32>,
+ w: usize,
+ h: usize,
+ mb_w: usize,
+ mb_h: usize,
+ mb_x: usize,
+ mb_y: usize,
+ mb_count: usize,
+ mb_end: usize,
+ new_ver: bool,
+ dc_coded: [bool; 3],
+ last_dc: [i16; 3],
+ is_intra: bool,
+}
+
+struct RV10SliceInfo {
+ is_p: bool,
+ qscale: u8,
+ mb_x: usize,
+ mb_y: usize,
+ mb_c: usize,
+}
+
+impl RV10SliceInfo {
+ fn new(is_p: bool, qscale: u8, mb_x: usize, mb_y: usize, mb_c: usize) -> Self {
+ RV10SliceInfo { is_p: is_p, qscale: qscale, mb_x: mb_x, mb_y: mb_y, mb_c: mb_c }
+ }
+}
+
+impl<'a> RealVideo10BR<'a> {
+ fn new(src: &'a [u8], tables: &'a Tables, width: usize, height: usize, new_ver: bool) -> Self {
+ let nslices = (src[0] as usize) + 1;
+ let mut slice_offs = Vec::with_capacity(nslices);
+ {
+ let offs = &src[1..][..nslices * 8];
+ let mut br = BitReader::new(offs, offs.len(), BitReaderMode::BE);
+ for _ in 0..nslices {
+ br.skip(32).unwrap();
+ let off = br.read(32).unwrap();
+ slice_offs.push(off);
+ }
+ }
+ let soff = nslices * 8 + 1;
+ RealVideo10BR {
+ br: BitReader::new(&src[soff..], src.len() - soff, BitReaderMode::BE),
+ tables: tables,
+ num_slices: nslices,
+ slice_no: 0,
+ slice_off: slice_offs,
+ w: width,
+ h: height,
+ mb_w: (width + 15) >> 4,
+ mb_h: (height + 15) >> 4,
+ mb_x: 0,
+ mb_y: 0,
+ mb_count: 0,
+ mb_end: 0,
+ new_ver: new_ver,
+ dc_coded: [false; 3],
+ last_dc: [0; 3],
+ is_intra: false,
+ }
+ }
+
+ fn decode_block(&mut self, quant: u8, intra: bool, coded: bool, blk: &mut [i16; 64], plane_no: usize) -> DecoderResult<()> {
+ let mut br = &mut self.br;
+ let mut idx = 0;
+ if intra {
+ let mut dc;
+//println!("dc prev {} @ {} of {},{} / {}x{}", self.last_dc[plane_no], br.tell() - (self.slice_off[self.slice_no - 1] as usize) * 8, self.mb_x, self.mb_y, self.mb_w, self.mb_h);
+ if !self.new_ver || !self.is_intra {
+ dc = br.read(8)? as i16;
+ if dc == 255 { dc = 128; }
+ } else {
+ if self.dc_coded[plane_no] {
+ let diff;
+ let bits = br.peek(14);
+ let ret = if plane_no == 0 {
+ br.read_cb(&self.tables.luma_dc_cb)
+ } else {
+ br.read_cb(&self.tables.chroma_dc_cb)
+ };
+ if ret.is_err() {
+ println!("Illegal {} code {:X}", if plane_no==0{"luma"}else{"chroma"},bits);
+ }
+ let val = ret.unwrap() as i16;
+
+ if val != 0 {
+ diff = val - 128;
+ } else {
+println!("escape!!!");
+ let code = br.read(2)?;
+ match code {
+ 0x0 => { diff = ((br.read(7)? + 1) as i8) as i16; },
+ 0x1 => { diff = (br.read(7)? as i16) - 128; },
+ 0x2 => {
+ if plane_no == 0 {
+ if br.read_bool()? {
+ diff = ((br.read(8)? + 1) as i8) as i16;
+ } else {
+ diff = (br.read(8)? as i8) as i16;
+ }
+ } else {
+ br.skip(9)?;
+ diff = 1;
+ }
+ },
+ _ => {
+ if plane_no == 0 {
+ br.skip(4)?;
+ diff = 1;
+ } else {
+println!("!!! wrong chroma esc");
+ return Err(DecoderError::InvalidData);
+ }
+ },
+ };
+println!("!!! diff = {} @ {}", diff, br.tell());
+ }
+ dc = (self.last_dc[plane_no] - diff) & 0xFF;
+ self.last_dc[plane_no] = dc;
+ } else {
+ self.dc_coded[plane_no] = true;
+ dc = self.last_dc[plane_no];
+ }
+ }
+//println!("dc new {} @ {}", self.last_dc[plane_no], br.tell());
+ blk[0] = dc << 3;
+ idx = 1;
+ }
+ if !coded { return Ok(()); }
+
+ let rl_cb = &self.tables.rl_cb; // could be aic too
+ let q_add = if quant == 0 { 0i16 } else { ((quant - 1) | 1) as i16 };
+ let q = (quant * 2) as i16;
+ while idx < 64 {
+ let code = br.read_cb(rl_cb)?;
+ let run;
+ let mut level;
+ let last;
+ if !code.is_escape() {
+ run = code.get_run();
+ level = code.get_level();
+ last = code.is_last();
+ if br.read_bool()? { level = -level; }
+ level = (level * q) + q_add;
+ } else {
+ last = br.read_bool()?;
+ run = br.read(6)? as u8;
+ level = br.read_s(8)? as i16;
+ if level == -128 {
+ let low = br.read(5)? as i16;
+ let top = br.read_s(6)? as i16;
+ level = (top << 5) | low;
+ }
+ level = (level * q) + q_add;
+ if level < -2048 { level = -2048; }
+ if level > 2047 { level = 2047; }
+ }
+ idx += run;
+ validate!(idx < 64);
+ let oidx = H263_ZIGZAG[idx as usize];
+ blk[oidx] = level;
+ idx += 1;
+ if last { break; }
+ }
+ Ok(())
+ }
+}
+
+fn decode_mv_component(br: &mut BitReader, mv_cb: &Codebook<u8>) -> DecoderResult<i16> {
+ let code = br.read_cb(mv_cb)? as i16;
+ if code == 0 { return Ok(0) }
+ if !br.read_bool()? {
+ Ok(code)
+ } else {
+ Ok(-code)
+ }
+}
+
+fn decode_mv(br: &mut BitReader, mv_cb: &Codebook<u8>) -> DecoderResult<MV> {
+ let xval = decode_mv_component(br, mv_cb)?;
+ let yval = decode_mv_component(br, mv_cb)?;
+//println!(" MV {},{} @ {}", xval, yval, br.tell());
+ Ok(MV::new(xval, yval))
+}
+
+impl<'a> BlockDecoder for RealVideo10BR<'a> {
+
+#[allow(unused_variables)]
+ fn decode_pichdr(&mut self) -> DecoderResult<PicInfo> {
+ self.slice_no = 0;
+println!("decoding picture header size {}", if self.num_slices > 1 { self.slice_off[1] } else { ((self.br.tell() as u32) + (self.br.left() as u32))/8 });
+ self.mb_x = 0;
+ self.mb_y = 0;
+ self.mb_count = self.mb_w * self.mb_h;
+ let shdr = self.read_slice_header()?;
+ self.slice_no += 1;
+ validate!((shdr.mb_x == 0) && (shdr.mb_y == 0));
+
+ self.mb_x = shdr.mb_x;
+ self.mb_y = shdr.mb_y;
+ self.mb_count = shdr.mb_c;
+ self.mb_end = self.mb_x + self.mb_y * self.mb_w + self.mb_count;
+ self.is_intra = !shdr.is_p;
+
+ let ftype = if !shdr.is_p { Type::I } else { Type::P };
+ let picinfo = PicInfo::new(self.w, self.h, ftype, shdr.qscale, false, MVMode::Old, 0, None, true);
+ Ok(picinfo)
+ }
+
+ #[allow(unused_variables)]
+ fn decode_slice_header(&mut self, info: &PicInfo) -> DecoderResult<Slice> {
+//println!("read slice {} header", self.slice_no);
+ let shdr = self.read_slice_header()?;
+ let ret = Slice::new(shdr.mb_x, shdr.mb_y, shdr.qscale);
+ self.slice_no += 1;
+ self.mb_x = shdr.mb_x;
+ self.mb_y = shdr.mb_y;
+ self.mb_count = shdr.mb_c;
+ self.mb_end = self.mb_x + self.mb_y * self.mb_w + self.mb_count;
+
+ Ok(ret)
+ }
+
+ fn decode_block_header(&mut self, info: &PicInfo, slice: &Slice) -> DecoderResult<BlockInfo> {
+ let mut br = &mut self.br;
+ let mut q = slice.get_quant();
+ match info.get_mode() {
+ Type::I => {
+ let mut cbpc = br.read_cb(&self.tables.intra_mcbpc_cb)?;
+ while cbpc == 8 { cbpc = br.read_cb(&self.tables.intra_mcbpc_cb)?; }
+ let cbpy = br.read_cb(&self.tables.cbpy_cb)?;
+ let cbp = (cbpy << 2) | (cbpc & 3);
+ let dquant = (cbpc & 4) != 0;
+ if dquant {
+ let idx = br.read(2)? as usize;
+ q = ((q as i16) + (H263_DQUANT_TAB[idx] as i16)) as u8;
+ }
+ self.mb_x += 1;
+ if self.mb_x == self.mb_w {
+ self.mb_x = 0;
+ self.mb_y += 1;
+ }
+ Ok(BlockInfo::new(Type::I, cbp, q))
+ },
+ Type::P => {
+ if br.read_bool()? {
+self.mb_x += 1;
+if self.mb_x == self.mb_w {
+ self.mb_x = 0;
+ self.mb_y += 1;
+}
+ return Ok(BlockInfo::new(Type::Skip, 0, info.get_quant()));
+ }
+ let mut cbpc = br.read_cb(&self.tables.inter_mcbpc_cb)?;
+ while cbpc == 20 { cbpc = br.read_cb(&self.tables.inter_mcbpc_cb)?; }
+ let is_intra = (cbpc & 0x04) != 0;
+ let dquant = (cbpc & 0x08) != 0;
+ let is_4x4 = (cbpc & 0x10) != 0;
+ if is_intra {
+ let cbpy = br.read_cb(&self.tables.cbpy_cb)?;
+ let cbp = (cbpy << 2) | (cbpc & 3);
+ if dquant {
+ let idx = br.read(2)? as usize;
+ q = ((q as i16) + (H263_DQUANT_TAB[idx] as i16)) as u8;
+ }
+ let binfo = BlockInfo::new(Type::I, cbp, q);
+ self.mb_x += 1;
+ if self.mb_x == self.mb_w {
+ self.mb_x = 0;
+ self.mb_y += 1;
+ }
+ return Ok(binfo);
+ }
+
+ let mut cbpy = br.read_cb(&self.tables.cbpy_cb)?;
+// if /* !aiv && */(cbpc & 3) != 3 {
+ cbpy ^= 0xF;
+// }
+ let cbp = (cbpy << 2) | (cbpc & 3);
+ if dquant {
+ let idx = br.read(2)? as usize;
+ q = ((q as i16) + (H263_DQUANT_TAB[idx] as i16)) as u8;
+ }
+println!(" MB {}.{} cbp = {:X}", self.mb_x, self.mb_y, cbp);
+ let mut binfo = BlockInfo::new(Type::P, cbp, q);
+ if !is_4x4 {
+ let mvec: [MV; 1] = [decode_mv(br, &self.tables.mv_cb)?];
+ binfo.set_mv(&mvec);
+ } else {
+ let mvec: [MV; 4] = [
+ decode_mv(br, &self.tables.mv_cb)?,
+ decode_mv(br, &self.tables.mv_cb)?,
+ decode_mv(br, &self.tables.mv_cb)?,
+ decode_mv(br, &self.tables.mv_cb)?
+ ];
+ binfo.set_mv(&mvec);
+ }
+ self.mb_x += 1;
+ if self.mb_x == self.mb_w {
+ self.mb_x = 0;
+ self.mb_y += 1;
+ }
+ Ok(binfo)
+ },
+ _ => { println!("wrong info mode"); Err(DecoderError::InvalidData) },
+ }
+ }
+
+ #[allow(unused_variables)]
+ fn decode_block_intra(&mut self, info: &BlockInfo, quant: u8, no: usize, coded: bool, blk: &mut [i16; 64]) -> DecoderResult<()> {
+ self.decode_block(quant, true, coded, blk, if no < 4 { 0 } else { no - 3 })
+ }
+
+ #[allow(unused_variables)]
+ fn decode_block_inter(&mut self, info: &BlockInfo, quant: u8, no: usize, coded: bool, blk: &mut [i16; 64]) -> DecoderResult<()> {
+ self.decode_block(quant, false, coded, blk, if no < 4 { 0 } else { no - 3 })
+ }
+
+ fn is_slice_end(&mut self) -> bool { self.mb_x + self.mb_y * self.mb_w >= self.mb_end }
+
+ fn filter_row(&mut self, buf: &mut NAVideoBuffer<u8>, mb_y: usize, mb_w: usize, cbpi: &CBPInfo) {
+ let stride = buf.get_stride(0);
+ let mut off = buf.get_offset(0) + mb_y * 16 * stride;
+ for mb_x in 0..mb_w {
+ let coff = off;
+ let coded0 = cbpi.is_coded(mb_x, 0);
+ let coded1 = cbpi.is_coded(mb_x, 1);
+ let q = cbpi.get_q(mb_w + mb_x);
+ if mb_y != 0 {
+ if coded0 && cbpi.is_coded_top(mb_x, 0) { deblock_hor(buf, 0, q, coff); }
+ if coded1 && cbpi.is_coded_top(mb_x, 1) { deblock_hor(buf, 0, q, coff + 8); }
+ }
+ let coff = off + 8 * stride;
+ if cbpi.is_coded(mb_x, 2) && coded0 { deblock_hor(buf, 0, q, coff); }
+ if cbpi.is_coded(mb_x, 3) && coded1 { deblock_hor(buf, 0, q, coff + 8); }
+ off += 16;
+ }
+ let mut leftt = false;
+ let mut leftc = false;
+ let mut off = buf.get_offset(0) + mb_y * 16 * stride;
+ for mb_x in 0..mb_w {
+ let ctop0 = cbpi.is_coded_top(mb_x, 0);
+ let ctop1 = cbpi.is_coded_top(mb_x, 0);
+ let ccur0 = cbpi.is_coded(mb_x, 0);
+ let ccur1 = cbpi.is_coded(mb_x, 1);
+ let q = cbpi.get_q(mb_w + mb_x);
+ if mb_y != 0 {
+ let coff = off - 8 * stride;
+ let qtop = cbpi.get_q(mb_x);
+ if leftt && ctop0 { deblock_ver(buf, 0, qtop, coff); }
+ if ctop0 && ctop1 { deblock_ver(buf, 0, qtop, coff + 8); }
+ }
+ if leftc && ccur0 { deblock_ver(buf, 0, q, off); }
+ if ccur0 && ccur1 { deblock_ver(buf, 0, q, off + 8); }
+ leftt = ctop1;
+ leftc = ccur1;
+ off += 16;
+ }
+ let strideu = buf.get_stride(1);
+ let stridev = buf.get_stride(2);
+ let offu = buf.get_offset(1) + mb_y * 8 * strideu;
+ let offv = buf.get_offset(2) + mb_y * 8 * stridev;
+ if mb_y != 0 {
+ for mb_x in 0..mb_w {
+ let ctu = cbpi.is_coded_top(mb_x, 4);
+ let ccu = cbpi.is_coded(mb_x, 4);
+ let ctv = cbpi.is_coded_top(mb_x, 5);
+ let ccv = cbpi.is_coded(mb_x, 5);
+ let q = cbpi.get_q(mb_w + mb_x);
+ if ctu && ccu { deblock_hor(buf, 1, q, offu + mb_x * 8); }
+ if ctv && ccv { deblock_hor(buf, 2, q, offv + mb_x * 8); }
+ }
+ let mut leftu = false;
+ let mut leftv = false;
+ let offu = buf.get_offset(1) + (mb_y - 1) * 8 * strideu;
+ let offv = buf.get_offset(2) + (mb_y - 1) * 8 * stridev;
+ for mb_x in 0..mb_w {
+ let ctu = cbpi.is_coded_top(mb_x, 4);
+ let ctv = cbpi.is_coded_top(mb_x, 5);
+ let qt = cbpi.get_q(mb_x);
+ if leftu && ctu { deblock_ver(buf, 1, qt, offu + mb_x * 8); }
+ if leftv && ctv { deblock_ver(buf, 2, qt, offv + mb_x * 8); }
+ leftu = ctu;
+ leftv = ctv;
+ }
+ }
+ }
+}
+
+fn deblock_hor(buf: &mut NAVideoBuffer<u8>, comp: usize, q: u8, off: usize) {
+ let stride = buf.get_stride(comp);
+ let mut dptr = buf.get_data_mut();
+ let mut buf = dptr.as_mut_slice();
+ for x in 0..8 {
+ let a = buf[off - 2 * stride + x] as i16;
+ let b = buf[off - 1 * stride + x] as i16;
+ let c = buf[off + 0 * stride + x] as i16;
+ let d = buf[off + 1 * stride + x] as i16;
+ let diff = ((a - d) * 3 + (c - b) * 8) >> 4;
+ if (diff != 0) && (diff >= -32) && (diff < 32) {
+ let d0 = diff.abs() * 2 - (q as i16);
+ let d1 = if d0 < 0 { 0 } else { d0 };
+ let d2 = diff.abs() - d1;
+ let d3 = if d2 < 0 { 0 } else { d2 };
+
+ let delta = if diff < 0 { -d3 } else { d3 };
+
+ let b1 = b + delta;
+ if b1 < 0 { buf[off - 1 * stride + x] = 0; }
+ else if b1 > 255 { buf[off - 1 * stride + x] = 0xFF; }
+ else { buf[off - 1 * stride + x] = b1 as u8; }
+ let c1 = c - delta;
+ if c1 < 0 { buf[off + x] = 0; }
+ else if c1 > 255 { buf[off + x] = 0xFF; }
+ else { buf[off + x] = c1 as u8; }
+ }
+ }
+}
+
+impl<'a> RealVideo10BR<'a> {
+ fn read_slice_header(&mut self) -> DecoderResult<RV10SliceInfo> {
+ validate!(self.slice_no < self.num_slices);
+
+ let mut br = &mut self.br;
+ br.seek(self.slice_off[self.slice_no] * 8)?;
+//println!(" slice at off {}", br.tell());
+
+ let marker = br.read(1)?;
+ validate!(marker == 1);
+ let is_p = br.read_bool()?;
+ let pb_frame = br.read_bool()?;
+ validate!(!pb_frame);
+ let qscale = br.read(5)? as u8;
+ validate!(qscale > 0);
+
+ if !is_p && self.new_ver {
+ self.last_dc[0] = br.read(8)? as i16;
+ self.last_dc[1] = br.read(8)? as i16;
+ self.last_dc[2] = br.read(8)? as i16;
+ } else {
+ self.last_dc[0] = 0;
+ self.last_dc[1] = 0;
+ self.last_dc[2] = 0;
+ }
+ self.dc_coded[0] = false;
+ self.dc_coded[1] = false;
+ self.dc_coded[2] = false;
+
+ let mb_x;
+ let mb_y;
+ let mb_count;
+ let mb_pos = self.mb_x + self.mb_y * self.mb_w;
+ if (br.peek(12) == 0) || (mb_pos > 0) {
+ mb_x = br.read(6)? as usize;
+ mb_y = br.read(6)? as usize;
+ mb_count = br.read(12)? as usize;
+ } else {
+ mb_x = self.mb_x;
+ mb_y = self.mb_y;
+ mb_count = self.mb_count;
+ }
+ br.skip(3)?;
+println!("slice q {} mb {},{} {}", qscale, mb_x, mb_y, mb_count);
+ validate!(mb_x + mb_y * self.mb_w + mb_count <= self.mb_w * self.mb_h);
+
+ Ok(RV10SliceInfo::new(is_p, qscale, mb_x, mb_y, mb_count))
+ }
+}
+
+fn deblock_ver(buf: &mut NAVideoBuffer<u8>, comp: usize, q: u8, off: usize) {
+ let stride = buf.get_stride(comp);
+ let mut dptr = buf.get_data_mut();
+ let mut buf = dptr.as_mut_slice();
+ for y in 0..8 {
+ let a = buf[off - 2 + y * stride] as i16;
+ let b = buf[off - 1 + y * stride] as i16;
+ let c = buf[off + 0 + y * stride] as i16;
+ let d = buf[off + 1 + y * stride] as i16;
+ let diff = ((a - d) * 3 + (c - b) * 8) >> 4;
+ if (diff != 0) && (diff >= -32) && (diff < 32) {
+ let d0 = diff.abs() * 2 - (q as i16);
+ let d1 = if d0 < 0 { 0 } else { d0 };
+ let d2 = diff.abs() - d1;
+ let d3 = if d2 < 0 { 0 } else { d2 };
+
+ let delta = if diff < 0 { -d3 } else { d3 };
+
+ let b1 = b + delta;
+ if b1 < 0 { buf[off - 1 + y * stride] = 0; }
+ else if b1 > 255 { buf[off - 1 + y * stride] = 0xFF; }
+ else { buf[off - 1 + y * stride] = b1 as u8; }
+ let c1 = c - delta;
+ if c1 < 0 { buf[off + y * stride] = 0; }
+ else if c1 > 255 { buf[off + y * stride] = 0xFF; }
+ else { buf[off + y * stride] = c1 as u8; }
+ }
+ }
+}
+
+impl RealVideo10Decoder {
+ fn new() -> Self {
+ let mut coderead = H263ShortCodeReader::new(H263_INTRA_MCBPC);
+ let intra_mcbpc_cb = Codebook::new(&mut coderead, CodebookMode::MSB).unwrap();
+ let mut coderead = H263ShortCodeReader::new(H263_INTER_MCBPC);
+ let inter_mcbpc_cb = Codebook::new(&mut coderead, CodebookMode::MSB).unwrap();
+ let mut coderead = H263ShortCodeReader::new(H263_CBPY);
+ let cbpy_cb = Codebook::new(&mut coderead, CodebookMode::MSB).unwrap();
+ let mut coderead = H263RLCodeReader::new(H263_RL_CODES);
+ let rl_cb = Codebook::new(&mut coderead, CodebookMode::MSB).unwrap();
+ let mut coderead = H263RLCodeReader::new(H263_RL_CODES_AIC);
+ let aic_rl_cb = Codebook::new(&mut coderead, CodebookMode::MSB).unwrap();
+ let mut coderead = H263ShortCodeReader::new(H263_MV);
+ let mv_cb = Codebook::new(&mut coderead, CodebookMode::MSB).unwrap();
+ let mut coderead = CodeReader::new(RV10_LUMA_DC_CODES, RV10_LUMA_DC_BITS);
+ let luma_dc_cb = Codebook::new(&mut coderead, CodebookMode::MSB).unwrap();
+ let mut coderead = CodeReader::new(RV10_CHROMA_DC_CODES, RV10_CHROMA_DC_BITS);
+ let chroma_dc_cb = Codebook::new(&mut coderead, CodebookMode::MSB).unwrap();
+
+ let tables = Tables {
+ intra_mcbpc_cb: intra_mcbpc_cb,
+ inter_mcbpc_cb: inter_mcbpc_cb,
+ cbpy_cb: cbpy_cb,
+ rl_cb: rl_cb,
+ aic_rl_cb: aic_rl_cb,
+ mv_cb: mv_cb,
+ luma_dc_cb: luma_dc_cb,
+ chroma_dc_cb: chroma_dc_cb,
+ };
+
+ RealVideo10Decoder{
+ info: Rc::new(DUMMY_CODEC_INFO),
+ dec: H263BaseDecoder::new(),
+ tables: tables,
+ w: 0,
+ h: 0,
+ new_ver: false,
+ }
+ }
+}
+
+impl NADecoder for RealVideo10Decoder {
+ fn init(&mut self, info: Rc<NACodecInfo>) -> DecoderResult<()> {
+ if let NACodecTypeInfo::Video(vinfo) = info.get_properties() {
+ let w = vinfo.get_width();
+ let h = vinfo.get_height();
+ let fmt = formats::YUV420_FORMAT;
+ let myinfo = NACodecTypeInfo::Video(NAVideoInfo::new(w, h, false, fmt));
+ self.info = Rc::new(NACodecInfo::new_ref(info.get_name(), myinfo, info.get_extradata()));
+ self.w = w;
+ self.h = h;
+
+ let edata = info.get_extradata().unwrap();
+ let src: &[u8] = &edata;
+ let ver = ((src[4] as u32) << 12) | ((src[5] as u32) << 4) | ((src[6] as u32) >> 4);
+ let maj_ver = ver >> 16;
+ let mic_ver = ver & 0xFF;
+println!("ver {:06X} -> {}", ver, mic_ver);
+ validate!(maj_ver == 1);
+ self.new_ver = mic_ver > 1;
+{
+let mut br = BitReader::new(src, src.len(), BitReaderMode::BE);
+println!("edata:");
+println!("{:08X}", br.read(32).unwrap());
+println!("{:08X}", br.read(32).unwrap());
+}
+ Ok(())
+ } else {
+println!("???");
+ Err(DecoderError::InvalidData)
+ }
+ }
+ fn decode(&mut self, pkt: &NAPacket) -> DecoderResult<NAFrameRef> {
+ let src = pkt.get_buffer();
+
+println!(" decode frame size {}, {} slices", src.len(), src[0]+1);
+ let mut ibr = RealVideo10BR::new(&src, &self.tables, self.w, self.h, self.new_ver);
+
+ let bufinfo = self.dec.parse_frame(&mut ibr)?;
+
+ let mut frm = NAFrame::new_from_pkt(pkt, self.info.clone(), bufinfo);
+ frm.set_keyframe(self.dec.is_intra());
+ frm.set_frame_type(if self.dec.is_intra() { FrameType::I } else { FrameType::P });
+ Ok(Rc::new(RefCell::new(frm)))
+ }
+}
+
+
+pub fn get_decoder() -> Box<NADecoder> {
+ Box::new(RealVideo10Decoder::new())
+}
+
+#[cfg(test)]
+mod test {
+ use test::dec_video::test_file_decoding;
+ #[test]
+ fn test_rv10() {
+ test_file_decoding("realmedia", "assets/RV/rv10_dnet_640x352_realvideo_encoder_4.0.rm", Some(1000), true, false, Some("rv10"));
+ }
+}
+
+pub struct CodeReader { codes: &'static [u16], bits: &'static [u8] }
+
+impl CodeReader {
+ pub fn new(codes: &'static [u16], bits: &'static [u8]) -> Self {
+ CodeReader { codes: codes, bits: bits }
+ }
+}
+
+impl CodebookDescReader<u8> for CodeReader {
+ fn bits(&mut self, idx: usize) -> u8 { self.bits[idx] }
+ fn code(&mut self, idx: usize) -> u32 { self.codes[idx] as u32 }
+ fn sym (&mut self, idx: usize) -> u8 { idx as u8 }
+ fn len(&mut self) -> usize { self.bits.len() }
+}
+
+const RV10_LUMA_DC_CODES: &[u16] = &[
+ 0x001f, 0x0f00, 0x0f01, 0x0f02, 0x0f03, 0x0f04, 0x0f05, 0x0f06,
+ 0x0f07, 0x0f08, 0x0f09, 0x0f0a, 0x0f0b, 0x0f0c, 0x0f0d, 0x0f0e,
+ 0x0f0f, 0x0f10, 0x0f11, 0x0f12, 0x0f13, 0x0f14, 0x0f15, 0x0f16,
+ 0x0f17, 0x0f18, 0x0f19, 0x0f1a, 0x0f1b, 0x0f1c, 0x0f1d, 0x0f1e,
+ 0x0f1f, 0x0f20, 0x0f21, 0x0f22, 0x0f23, 0x0f24, 0x0f25, 0x0f26,
+ 0x0f27, 0x0f28, 0x0f29, 0x0f2a, 0x0f2b, 0x0f2c, 0x0f2d, 0x0f2e,
+ 0x0f2f, 0x0f30, 0x0f31, 0x0f32, 0x0f33, 0x0f34, 0x0f35, 0x0f36,
+ 0x0f37, 0x0f38, 0x0f39, 0x0f3a, 0x0f3b, 0x0f3c, 0x0f3d, 0x0f3e,
+ 0x0f3f, 0x0380, 0x0381, 0x0382, 0x0383, 0x0384, 0x0385, 0x0386,
+ 0x0387, 0x0388, 0x0389, 0x038a, 0x038b, 0x038c, 0x038d, 0x038e,
+ 0x038f, 0x0390, 0x0391, 0x0392, 0x0393, 0x0394, 0x0395, 0x0396,
+ 0x0397, 0x0398, 0x0399, 0x039a, 0x039b, 0x039c, 0x039d, 0x039e,
+ 0x039f, 0x00c0, 0x00c1, 0x00c2, 0x00c3, 0x00c4, 0x00c5, 0x00c6,
+ 0x00c7, 0x00c8, 0x00c9, 0x00ca, 0x00cb, 0x00cc, 0x00cd, 0x00ce,
+ 0x00cf, 0x0050, 0x0051, 0x0052, 0x0053, 0x0054, 0x0055, 0x0056,
+ 0x0057, 0x0020, 0x0021, 0x0022, 0x0023, 0x000c, 0x000d, 0x0004,
+ 0x0000, 0x0005, 0x000e, 0x000f, 0x0024, 0x0025, 0x0026, 0x0027,
+ 0x0058, 0x0059, 0x005a, 0x005b, 0x005c, 0x005d, 0x005e, 0x005f,
+ 0x00d0, 0x00d1, 0x00d2, 0x00d3, 0x00d4, 0x00d5, 0x00d6, 0x00d7,
+ 0x00d8, 0x00d9, 0x00da, 0x00db, 0x00dc, 0x00dd, 0x00de, 0x00df,
+ 0x03a0, 0x03a1, 0x03a2, 0x03a3, 0x03a4, 0x03a5, 0x03a6, 0x03a7,
+ 0x03a8, 0x03a9, 0x03aa, 0x03ab, 0x03ac, 0x03ad, 0x03ae, 0x03af,
+ 0x03b0, 0x03b1, 0x03b2, 0x03b3, 0x03b4, 0x03b5, 0x03b6, 0x03b7,
+ 0x03b8, 0x03b9, 0x03ba, 0x03bb, 0x03bc, 0x03bd, 0x03be, 0x03bf,
+ 0x0f40, 0x0f41, 0x0f42, 0x0f43, 0x0f44, 0x0f45, 0x0f46, 0x0f47,
+ 0x0f48, 0x0f49, 0x0f4a, 0x0f4b, 0x0f4c, 0x0f4d, 0x0f4e, 0x0f4f,
+ 0x0f50, 0x0f51, 0x0f52, 0x0f53, 0x0f54, 0x0f55, 0x0f56, 0x0f57,
+ 0x0f58, 0x0f59, 0x0f5a, 0x0f5b, 0x0f5c, 0x0f5d, 0x0f5e, 0x0f5f,
+ 0x0f60, 0x0f61, 0x0f62, 0x0f63, 0x0f64, 0x0f65, 0x0f66, 0x0f67,
+ 0x0f68, 0x0f69, 0x0f6a, 0x0f6b, 0x0f6c, 0x0f6d, 0x0f6e, 0x0f6f,
+ 0x0f70, 0x0f71, 0x0f72, 0x0f73, 0x0f74, 0x0f75, 0x0f76, 0x0f77,
+ 0x0f78, 0x0f79, 0x0f7a, 0x0f7b, 0x0f7c, 0x0f7d, 0x0f7e, 0x0f7f,
+];
+
+const RV10_LUMA_DC_BITS: &[u8] = &[
+ 5, 12, 12, 12, 12, 12, 12, 12,
+ 12, 12, 12, 12, 12, 12, 12, 12,
+ 12, 12, 12, 12, 12, 12, 12, 12,
+ 12, 12, 12, 12, 12, 12, 12, 12,
+ 12, 12, 12, 12, 12, 12, 12, 12,
+ 12, 12, 12, 12, 12, 12, 12, 12,
+ 12, 12, 12, 12, 12, 12, 12, 12,
+ 12, 12, 12, 12, 12, 12, 12, 12,
+ 12, 10, 10, 10, 10, 10, 10, 10,
+ 10, 10, 10, 10, 10, 10, 10, 10,
+ 10, 10, 10, 10, 10, 10, 10, 10,
+ 10, 10, 10, 10, 10, 10, 10, 10,
+ 10, 8, 8, 8, 8, 8, 8, 8,
+ 8, 8, 8, 8, 8, 8, 8, 8,
+ 8, 7, 7, 7, 7, 7, 7, 7,
+ 7, 6, 6, 6, 6, 5, 5, 4,
+ 2, 4, 5, 5, 6, 6, 6, 6,
+ 7, 7, 7, 7, 7, 7, 7, 7,
+ 8, 8, 8, 8, 8, 8, 8, 8,
+ 8, 8, 8, 8, 8, 8, 8, 8,
+ 10, 10, 10, 10, 10, 10, 10, 10,
+ 10, 10, 10, 10, 10, 10, 10, 10,
+ 10, 10, 10, 10, 10, 10, 10, 10,
+ 10, 10, 10, 10, 10, 10, 10, 10,
+ 12, 12, 12, 12, 12, 12, 12, 12,
+ 12, 12, 12, 12, 12, 12, 12, 12,
+ 12, 12, 12, 12, 12, 12, 12, 12,
+ 12, 12, 12, 12, 12, 12, 12, 12,
+ 12, 12, 12, 12, 12, 12, 12, 12,
+ 12, 12, 12, 12, 12, 12, 12, 12,
+ 12, 12, 12, 12, 12, 12, 12, 12,
+ 12, 12, 12, 12, 12, 12, 12, 12,
+];
+
+const RV10_CHROMA_DC_CODES: &[u16] = &[
+ 0x003f, 0x3f00, 0x3f01, 0x3f02, 0x3f03, 0x3f04, 0x3f05, 0x3f06,
+ 0x3f07, 0x3f08, 0x3f09, 0x3f0a, 0x3f0b, 0x3f0c, 0x3f0d, 0x3f0e,
+ 0x3f0f, 0x3f10, 0x3f11, 0x3f12, 0x3f13, 0x3f14, 0x3f15, 0x3f16,
+ 0x3f17, 0x3f18, 0x3f19, 0x3f1a, 0x3f1b, 0x3f1c, 0x3f1d, 0x3f1e,
+ 0x3f1f, 0x3f20, 0x3f21, 0x3f22, 0x3f23, 0x3f24, 0x3f25, 0x3f26,
+ 0x3f27, 0x3f28, 0x3f29, 0x3f2a, 0x3f2b, 0x3f2c, 0x3f2d, 0x3f2e,
+ 0x3f2f, 0x3f30, 0x3f31, 0x3f32, 0x3f33, 0x3f34, 0x3f35, 0x3f36,
+ 0x3f37, 0x3f38, 0x3f39, 0x3f3a, 0x3f3b, 0x3f3c, 0x3f3d, 0x3f3e,
+ 0x3f3f, 0x0f80, 0x0f81, 0x0f82, 0x0f83, 0x0f84, 0x0f85, 0x0f86,
+ 0x0f87, 0x0f88, 0x0f89, 0x0f8a, 0x0f8b, 0x0f8c, 0x0f8d, 0x0f8e,
+ 0x0f8f, 0x0f90, 0x0f91, 0x0f92, 0x0f93, 0x0f94, 0x0f95, 0x0f96,
+ 0x0f97, 0x0f98, 0x0f99, 0x0f9a, 0x0f9b, 0x0f9c, 0x0f9d, 0x0f9e,
+ 0x0f9f, 0x03c0, 0x03c1, 0x03c2, 0x03c3, 0x03c4, 0x03c5, 0x03c6,
+ 0x03c7, 0x03c8, 0x03c9, 0x03ca, 0x03cb, 0x03cc, 0x03cd, 0x03ce,
+ 0x03cf, 0x00e0, 0x00e1, 0x00e2, 0x00e3, 0x00e4, 0x00e5, 0x00e6,
+ 0x00e7, 0x0030, 0x0031, 0x0032, 0x0033, 0x0008, 0x0009, 0x0002,
+ 0x0000, 0x0003, 0x000a, 0x000b, 0x0034, 0x0035, 0x0036, 0x0037,
+ 0x00e8, 0x00e9, 0x00ea, 0x00eb, 0x00ec, 0x00ed, 0x00ee, 0x00ef,
+ 0x03d0, 0x03d1, 0x03d2, 0x03d3, 0x03d4, 0x03d5, 0x03d6, 0x03d7,
+ 0x03d8, 0x03d9, 0x03da, 0x03db, 0x03dc, 0x03dd, 0x03de, 0x03df,
+ 0x0fa0, 0x0fa1, 0x0fa2, 0x0fa3, 0x0fa4, 0x0fa5, 0x0fa6, 0x0fa7,
+ 0x0fa8, 0x0fa9, 0x0faa, 0x0fab, 0x0fac, 0x0fad, 0x0fae, 0x0faf,
+ 0x0fb0, 0x0fb1, 0x0fb2, 0x0fb3, 0x0fb4, 0x0fb5, 0x0fb6, 0x0fb7,
+ 0x0fb8, 0x0fb9, 0x0fba, 0x0fbb, 0x0fbc, 0x0fbd, 0x0fbe, 0x0fbf,
+ 0x3f40, 0x3f41, 0x3f42, 0x3f43, 0x3f44, 0x3f45, 0x3f46, 0x3f47,
+ 0x3f48, 0x3f49, 0x3f4a, 0x3f4b, 0x3f4c, 0x3f4d, 0x3f4e, 0x3f4f,
+ 0x3f50, 0x3f51, 0x3f52, 0x3f53, 0x3f54, 0x3f55, 0x3f56, 0x3f57,
+ 0x3f58, 0x3f59, 0x3f5a, 0x3f5b, 0x3f5c, 0x3f5d, 0x3f5e, 0x3f5f,
+ 0x3f60, 0x3f61, 0x3f62, 0x3f63, 0x3f64, 0x3f65, 0x3f66, 0x3f67,
+ 0x3f68, 0x3f69, 0x3f6a, 0x3f6b, 0x3f6c, 0x3f6d, 0x3f6e, 0x3f6f,
+ 0x3f70, 0x3f71, 0x3f72, 0x3f73, 0x3f74, 0x3f75, 0x3f76, 0x3f77,
+ 0x3f78, 0x3f79, 0x3f7a, 0x3f7b, 0x3f7c, 0x3f7d, 0x3f7e, 0x3f7f,
+];
+
+const RV10_CHROMA_DC_BITS: &[u8] = &[
+ 7, 14, 14, 14, 14, 14, 14, 14,
+ 14, 14, 14, 14, 14, 14, 14, 14,
+ 14, 14, 14, 14, 14, 14, 14, 14,
+ 14, 14, 14, 14, 14, 14, 14, 14,
+ 14, 14, 14, 14, 14, 14, 14, 14,
+ 14, 14, 14, 14, 14, 14, 14, 14,
+ 14, 14, 14, 14, 14, 14, 14, 14,
+ 14, 14, 14, 14, 14, 14, 14, 14,
+ 14, 12, 12, 12, 12, 12, 12, 12,
+ 12, 12, 12, 12, 12, 12, 12, 12,
+ 12, 12, 12, 12, 12, 12, 12, 12,
+ 12, 12, 12, 12, 12, 12, 12, 12,
+ 12, 10, 10, 10, 10, 10, 10, 10,
+ 10, 10, 10, 10, 10, 10, 10, 10,
+ 10, 8, 8, 8, 8, 8, 8, 8,
+ 8, 6, 6, 6, 6, 4, 4, 3,
+ 2, 3, 4, 4, 6, 6, 6, 6,
+ 8, 8, 8, 8, 8, 8, 8, 8,
+ 10, 10, 10, 10, 10, 10, 10, 10,
+ 10, 10, 10, 10, 10, 10, 10, 10,
+ 12, 12, 12, 12, 12, 12, 12, 12,
+ 12, 12, 12, 12, 12, 12, 12, 12,
+ 12, 12, 12, 12, 12, 12, 12, 12,
+ 12, 12, 12, 12, 12, 12, 12, 12,
+ 14, 14, 14, 14, 14, 14, 14, 14,
+ 14, 14, 14, 14, 14, 14, 14, 14,
+ 14, 14, 14, 14, 14, 14, 14, 14,
+ 14, 14, 14, 14, 14, 14, 14, 14,
+ 14, 14, 14, 14, 14, 14, 14, 14,
+ 14, 14, 14, 14, 14, 14, 14, 14,
+ 14, 14, 14, 14, 14, 14, 14, 14,
+ 14, 14, 14, 14, 14, 14, 14, 14,
+];
--- /dev/null
+use io::bitreader::*;
+use io::codebook::*;
+use formats;
+use super::super::*;
+use super::*;
+use super::decoder::*;
+use super::data::*;
+
+#[allow(dead_code)]
+struct Tables {
+ intra_mcbpc_cb: Codebook<u8>,
+ inter_mcbpc_cb: Codebook<u8>,
+ cbpy_cb: Codebook<u8>,
+ rl_cb: Codebook<H263RLSym>,
+ aic_rl_cb: Codebook<H263RLSym>,
+ mv_cb: Codebook<u8>,
+}
+
+#[derive(Clone,Copy)]
+struct RPRInfo {
+ present: bool,
+ bits: u8,
+ widths: [usize; 8],
+ heights: [usize; 8],
+}
+
+struct RealVideo20Decoder {
+ info: Rc<NACodecInfo>,
+ dec: H263BaseDecoder,
+ tables: Tables,
+ w: usize,
+ h: usize,
+ minor_ver: u8,
+ rpr: RPRInfo,
+}
+
+struct RealVideo20BR<'a> {
+ br: BitReader<'a>,
+ tables: &'a Tables,
+ num_slices: usize,
+ slice_no: usize,
+ slice_off: Vec<u32>,
+ w: usize,
+ h: usize,
+ mb_w: usize,
+ mb_h: usize,
+ mb_x: usize,
+ mb_y: usize,
+ mb_pos_bits: u8,
+ mb_count: usize,
+ mb_end: usize,
+ minor_ver: u8,
+ rpr: RPRInfo,
+ is_intra: bool,
+}
+
+struct RV20SliceInfo {
+ ftype: Type,
+ qscale: u8,
+ mb_x: usize,
+ mb_y: usize,
+ mb_pos: usize,
+ w: usize,
+ h: usize,
+}
+
+impl RV20SliceInfo {
+ fn new(ftype: Type, qscale: u8, mb_x: usize, mb_y: usize, mb_pos: usize, w: usize, h: usize) -> Self {
+ RV20SliceInfo { ftype: ftype, qscale: qscale, mb_x: mb_x, mb_y: mb_y, mb_pos: mb_pos, w: w, h: h }
+ }
+}
+
+impl<'a> RealVideo20BR<'a> {
+ fn new(src: &'a [u8], tables: &'a Tables, width: usize, height: usize, minor_ver: u8, rpr: RPRInfo) -> Self {
+ let nslices = (src[0] as usize) + 1;
+ let mut slice_offs = Vec::with_capacity(nslices);
+ {
+ let offs = &src[1..][..nslices * 8];
+ let mut br = BitReader::new(offs, offs.len(), BitReaderMode::BE);
+ for _ in 0..nslices {
+ br.skip(32).unwrap();
+ let off = br.read(32).unwrap();
+ slice_offs.push(off);
+ }
+ }
+ let soff = nslices * 8 + 1;
+ let mb_w = (width + 15) >> 4;
+ let mb_h = (height + 15) >> 4;
+ let max_pos = mb_w * mb_h - 1;
+ let mut mbpb = 0;
+ for i in 0..H263_MBB.len() {
+ if max_pos <= H263_MBB[i].blocks {
+ mbpb = H263_MBB[i].bits;
+ break;
+ }
+ }
+ RealVideo20BR {
+ br: BitReader::new(&src[soff..], src.len() - soff, BitReaderMode::BE),
+ tables: tables,
+ num_slices: nslices,
+ slice_no: 0,
+ slice_off: slice_offs,
+ w: width,
+ h: height,
+ mb_w: mb_w,
+ mb_h: mb_h,
+ mb_pos_bits: mbpb,
+ mb_x: 0,
+ mb_y: 0,
+ mb_count: 0,
+ mb_end: 0,
+ minor_ver: minor_ver,
+ rpr: rpr,
+ is_intra: false,
+ }
+ }
+
+ fn decode_block(&mut self, quant: u8, intra: bool, coded: bool, blk: &mut [i16; 64], plane_no: usize) -> DecoderResult<()> {
+ let mut br = &mut self.br;
+ let mut idx = 0;
+ if !self.is_intra && intra {
+ let mut dc = br.read(8).unwrap() as i16;
+ if dc == 255 { dc = 128; }
+ blk[0] = dc << 3;
+ idx = 1;
+ }
+ if !coded { return Ok(()); }
+
+ let rl_cb = if self.is_intra { &self.tables.aic_rl_cb } else { &self.tables.rl_cb };
+ let q_add = if quant == 0 { 0i16 } else { ((quant - 1) | 1) as i16 };
+ let q = (quant * 2) as i16;
+ while idx < 64 {
+ let code = br.read_cb(rl_cb).unwrap();
+ let run;
+ let mut level;
+ let last;
+ if !code.is_escape() {
+ run = code.get_run();
+ level = code.get_level();
+ last = code.is_last();
+ if br.read_bool().unwrap() { level = -level; }
+ level = (level * q) + q_add;
+ } else {
+ last = br.read_bool().unwrap();
+ run = br.read(6).unwrap() as u8;
+ level = br.read_s(8).unwrap() as i16;
+ if level == -128 {
+ let low = br.read(5).unwrap() as i16;
+ let top = br.read_s(6).unwrap() as i16;
+ level = (top << 5) | low;
+ }
+ level = (level * q) + q_add;
+ if level < -2048 { level = -2048; }
+ if level > 2047 { level = 2047; }
+ }
+ idx += run;
+ validate!(idx < 64);
+ let oidx = H263_ZIGZAG[idx as usize];
+ blk[oidx] = level;
+ idx += 1;
+ if last { break; }
+ }
+ Ok(())
+ }
+}
+
+fn decode_mv_component(br: &mut BitReader, mv_cb: &Codebook<u8>) -> DecoderResult<i16> {
+ let code = br.read_cb(mv_cb).unwrap() as i16;
+ if code == 0 { return Ok(0) }
+ if !br.read_bool().unwrap() {
+ Ok(code)
+ } else {
+ Ok(-code)
+ }
+}
+
+fn decode_mv(br: &mut BitReader, mv_cb: &Codebook<u8>) -> DecoderResult<MV> {
+ let xval = decode_mv_component(br, mv_cb).unwrap();
+ let yval = decode_mv_component(br, mv_cb).unwrap();
+//println!(" MV {},{} @ {}", xval, yval, br.tell());
+ Ok(MV::new(xval, yval))
+}
+
+impl<'a> BlockDecoder for RealVideo20BR<'a> {
+
+#[allow(unused_variables)]
+ fn decode_pichdr(&mut self) -> DecoderResult<PicInfo> {
+ self.slice_no = 0;
+println!("decoding picture header size {}", if self.num_slices > 1 { self.slice_off[1] } else { ((self.br.tell() as u32) + (self.br.left() as u32))/8 });
+ self.mb_x = 0;
+ self.mb_y = 0;
+ self.mb_count = self.mb_w * self.mb_h;
+ let shdr = self.read_slice_header().unwrap();
+println!("slice ends @ {}\n", self.br.tell());
+ self.slice_no += 1;
+ validate!((shdr.mb_x == 0) && (shdr.mb_y == 0));
+ let mb_count;
+ if self.slice_no < self.num_slices {
+ let pos = self.br.tell();
+ let shdr2 = self.read_slice_header().unwrap();
+ self.br.seek(pos as u32)?;
+ mb_count = shdr2.mb_pos - shdr.mb_pos;
+ } else {
+ mb_count = self.mb_w * self.mb_h;
+ }
+
+ self.mb_x = shdr.mb_x;
+ self.mb_y = shdr.mb_y;
+ self.mb_count = mb_count;
+ self.mb_end = shdr.mb_pos + mb_count;
+ self.is_intra = shdr.ftype == Type::I;
+
+ let picinfo = PicInfo::new(shdr.w, shdr.h, shdr.ftype, shdr.qscale, false, MVMode::Old, 0, None, true);
+ Ok(picinfo)
+ }
+
+ #[allow(unused_variables)]
+ fn decode_slice_header(&mut self, info: &PicInfo) -> DecoderResult<Slice> {
+//println!("read slice {} header", self.slice_no);
+ let shdr = self.read_slice_header().unwrap();
+ self.slice_no += 1;
+ let mb_count;
+ if self.slice_no < self.num_slices {
+ let shdr2 = self.read_slice_header().unwrap();
+ mb_count = shdr2.mb_pos - shdr.mb_pos;
+ } else {
+ mb_count = self.mb_w * self.mb_h - shdr.mb_pos;
+ }
+ let ret = Slice::new(shdr.mb_x, shdr.mb_y, shdr.qscale);
+ self.mb_x = shdr.mb_x;
+ self.mb_y = shdr.mb_y;
+ self.mb_count = mb_count;
+ self.mb_end = shdr.mb_pos + mb_count;
+
+ Ok(ret)
+ }
+
+ fn decode_block_header(&mut self, info: &PicInfo, slice: &Slice) -> DecoderResult<BlockInfo> {
+ let mut br = &mut self.br;
+ let mut q = slice.get_quant();
+ match info.get_mode() {
+ Type::I => {
+ let mut cbpc = br.read_cb(&self.tables.intra_mcbpc_cb).unwrap();
+ while cbpc == 8 { cbpc = br.read_cb(&self.tables.intra_mcbpc_cb).unwrap(); }
+if self.is_intra {
+let acpred = br.read_bool()?;
+println!(" acp {} @ {}", acpred, br.tell());
+if acpred {
+ br.skip(1)?;//pred direction
+}
+}
+ let cbpy = br.read_cb(&self.tables.cbpy_cb).unwrap();
+ let cbp = (cbpy << 2) | (cbpc & 3);
+ let dquant = (cbpc & 4) != 0;
+ if dquant {
+ let idx = br.read(2).unwrap() as usize;
+ q = ((q as i16) + (H263_DQUANT_TAB[idx] as i16)) as u8;
+ }
+println!(" MB {},{} CBP {:X} @ {}", self.mb_x, self.mb_y, cbp, br.tell());
+ self.mb_x += 1;
+ if self.mb_x == self.mb_w {
+ self.mb_x = 0;
+ self.mb_y += 1;
+ }
+ Ok(BlockInfo::new(Type::I, cbp, q))
+ },
+ Type::P => {
+ if br.read_bool().unwrap() {
+self.mb_x += 1;
+if self.mb_x == self.mb_w {
+ self.mb_x = 0;
+ self.mb_y += 1;
+}
+ return Ok(BlockInfo::new(Type::Skip, 0, info.get_quant()));
+ }
+ let mut cbpc = br.read_cb(&self.tables.inter_mcbpc_cb).unwrap();
+ while cbpc == 20 { cbpc = br.read_cb(&self.tables.inter_mcbpc_cb).unwrap(); }
+ let is_intra = (cbpc & 0x04) != 0;
+ let dquant = (cbpc & 0x08) != 0;
+ let is_4x4 = (cbpc & 0x10) != 0;
+ if is_intra {
+ let cbpy = br.read_cb(&self.tables.cbpy_cb).unwrap();
+ let cbp = (cbpy << 2) | (cbpc & 3);
+ if dquant {
+ let idx = br.read(2).unwrap() as usize;
+ q = ((q as i16) + (H263_DQUANT_TAB[idx] as i16)) as u8;
+ }
+ let binfo = BlockInfo::new(Type::I, cbp, q);
+ self.mb_x += 1;
+ if self.mb_x == self.mb_w {
+ self.mb_x = 0;
+ self.mb_y += 1;
+ }
+ return Ok(binfo);
+ }
+
+ let mut cbpy = br.read_cb(&self.tables.cbpy_cb).unwrap();
+// if /* !aiv && */(cbpc & 3) != 3 {
+ cbpy ^= 0xF;
+// }
+ let cbp = (cbpy << 2) | (cbpc & 3);
+ if dquant {
+ let idx = br.read(2).unwrap() as usize;
+ q = ((q as i16) + (H263_DQUANT_TAB[idx] as i16)) as u8;
+ }
+println!(" MB {}.{} cbp = {:X}", self.mb_x, self.mb_y, cbp);
+ let mut binfo = BlockInfo::new(Type::P, cbp, q);
+ if !is_4x4 {
+ let mvec: [MV; 1] = [decode_mv(br, &self.tables.mv_cb).unwrap()];
+ binfo.set_mv(&mvec);
+ } else {
+ let mvec: [MV; 4] = [
+ decode_mv(br, &self.tables.mv_cb).unwrap(),
+ decode_mv(br, &self.tables.mv_cb).unwrap(),
+ decode_mv(br, &self.tables.mv_cb).unwrap(),
+ decode_mv(br, &self.tables.mv_cb).unwrap()
+ ];
+ binfo.set_mv(&mvec);
+ }
+ self.mb_x += 1;
+ if self.mb_x == self.mb_w {
+ self.mb_x = 0;
+ self.mb_y += 1;
+ }
+ Ok(binfo)
+ },
+ _ => { println!("wrong info mode"); Err(DecoderError::InvalidData) },
+ }
+ }
+
+ #[allow(unused_variables)]
+ fn decode_block_intra(&mut self, info: &BlockInfo, quant: u8, no: usize, coded: bool, blk: &mut [i16; 64]) -> DecoderResult<()> {
+ self.decode_block(quant, true, coded, blk, if no < 4 { 0 } else { no - 3 })
+ }
+
+ #[allow(unused_variables)]
+ fn decode_block_inter(&mut self, info: &BlockInfo, quant: u8, no: usize, coded: bool, blk: &mut [i16; 64]) -> DecoderResult<()> {
+ self.decode_block(quant, false, coded, blk, if no < 4 { 0 } else { no - 3 })
+ }
+
+ fn is_slice_end(&mut self) -> bool { self.mb_x + self.mb_y * self.mb_w >= self.mb_end }
+
+ fn filter_row(&mut self, buf: &mut NAVideoBuffer<u8>, mb_y: usize, mb_w: usize, cbpi: &CBPInfo) {
+ let stride = buf.get_stride(0);
+ let mut off = buf.get_offset(0) + mb_y * 16 * stride;
+ for mb_x in 0..mb_w {
+ let coff = off;
+ let coded0 = cbpi.is_coded(mb_x, 0);
+ let coded1 = cbpi.is_coded(mb_x, 1);
+ let q = cbpi.get_q(mb_w + mb_x);
+ if mb_y != 0 {
+ if coded0 && cbpi.is_coded_top(mb_x, 0) { deblock_hor(buf, 0, q, coff); }
+ if coded1 && cbpi.is_coded_top(mb_x, 1) { deblock_hor(buf, 0, q, coff + 8); }
+ }
+ let coff = off + 8 * stride;
+ if cbpi.is_coded(mb_x, 2) && coded0 { deblock_hor(buf, 0, q, coff); }
+ if cbpi.is_coded(mb_x, 3) && coded1 { deblock_hor(buf, 0, q, coff + 8); }
+ off += 16;
+ }
+ let mut leftt = false;
+ let mut leftc = false;
+ let mut off = buf.get_offset(0) + mb_y * 16 * stride;
+ for mb_x in 0..mb_w {
+ let ctop0 = cbpi.is_coded_top(mb_x, 0);
+ let ctop1 = cbpi.is_coded_top(mb_x, 0);
+ let ccur0 = cbpi.is_coded(mb_x, 0);
+ let ccur1 = cbpi.is_coded(mb_x, 1);
+ let q = cbpi.get_q(mb_w + mb_x);
+ if mb_y != 0 {
+ let coff = off - 8 * stride;
+ let qtop = cbpi.get_q(mb_x);
+ if leftt && ctop0 { deblock_ver(buf, 0, qtop, coff); }
+ if ctop0 && ctop1 { deblock_ver(buf, 0, qtop, coff + 8); }
+ }
+ if leftc && ccur0 { deblock_ver(buf, 0, q, off); }
+ if ccur0 && ccur1 { deblock_ver(buf, 0, q, off + 8); }
+ leftt = ctop1;
+ leftc = ccur1;
+ off += 16;
+ }
+ let strideu = buf.get_stride(1);
+ let stridev = buf.get_stride(2);
+ let offu = buf.get_offset(1) + mb_y * 8 * strideu;
+ let offv = buf.get_offset(2) + mb_y * 8 * stridev;
+ if mb_y != 0 {
+ for mb_x in 0..mb_w {
+ let ctu = cbpi.is_coded_top(mb_x, 4);
+ let ccu = cbpi.is_coded(mb_x, 4);
+ let ctv = cbpi.is_coded_top(mb_x, 5);
+ let ccv = cbpi.is_coded(mb_x, 5);
+ let q = cbpi.get_q(mb_w + mb_x);
+ if ctu && ccu { deblock_hor(buf, 1, q, offu + mb_x * 8); }
+ if ctv && ccv { deblock_hor(buf, 2, q, offv + mb_x * 8); }
+ }
+ let mut leftu = false;
+ let mut leftv = false;
+ let offu = buf.get_offset(1) + (mb_y - 1) * 8 * strideu;
+ let offv = buf.get_offset(2) + (mb_y - 1) * 8 * stridev;
+ for mb_x in 0..mb_w {
+ let ctu = cbpi.is_coded_top(mb_x, 4);
+ let ctv = cbpi.is_coded_top(mb_x, 5);
+ let qt = cbpi.get_q(mb_x);
+ if leftu && ctu { deblock_ver(buf, 1, qt, offu + mb_x * 8); }
+ if leftv && ctv { deblock_ver(buf, 2, qt, offv + mb_x * 8); }
+ leftu = ctu;
+ leftv = ctv;
+ }
+ }
+ }
+}
+
+fn deblock_hor(buf: &mut NAVideoBuffer<u8>, comp: usize, q: u8, off: usize) {
+ let stride = buf.get_stride(comp);
+ let mut dptr = buf.get_data_mut();
+ let mut buf = dptr.as_mut_slice();
+ for x in 0..8 {
+ let a = buf[off - 2 * stride + x] as i16;
+ let b = buf[off - 1 * stride + x] as i16;
+ let c = buf[off + 0 * stride + x] as i16;
+ let d = buf[off + 1 * stride + x] as i16;
+ let diff = ((a - d) * 3 + (c - b) * 8) >> 4;
+ if (diff != 0) && (diff >= -32) && (diff < 32) {
+ let d0 = diff.abs() * 2 - (q as i16);
+ let d1 = if d0 < 0 { 0 } else { d0 };
+ let d2 = diff.abs() - d1;
+ let d3 = if d2 < 0 { 0 } else { d2 };
+
+ let delta = if diff < 0 { -d3 } else { d3 };
+
+ let b1 = b + delta;
+ if b1 < 0 { buf[off - 1 * stride + x] = 0; }
+ else if b1 > 255 { buf[off - 1 * stride + x] = 0xFF; }
+ else { buf[off - 1 * stride + x] = b1 as u8; }
+ let c1 = c - delta;
+ if c1 < 0 { buf[off + x] = 0; }
+ else if c1 > 255 { buf[off + x] = 0xFF; }
+ else { buf[off + x] = c1 as u8; }
+ }
+ }
+}
+
+impl<'a> RealVideo20BR<'a> {
+ fn read_slice_header(&mut self) -> DecoderResult<RV20SliceInfo> {
+ validate!(self.slice_no < self.num_slices);
+
+ let mut br = &mut self.br;
+ br.seek(self.slice_off[self.slice_no] * 8).unwrap();
+//println!(" slice at off {}", br.tell());
+
+ let frm_type = br.read(2).unwrap();
+ let ftype = match frm_type {
+ 0 | 1 => { Type::I },
+ 2 => { Type::P },
+ _ => { Type::Skip },
+ };
+
+ let marker = br.read(1).unwrap();
+ validate!(marker == 0);
+ let qscale = br.read(5).unwrap() as u8;
+ validate!(qscale > 0);
+ if self.minor_ver >= 2 {
+ br.skip(1).unwrap(); // loop filter
+ }
+ let seq = if self.minor_ver <= 1 {
+ br.read(8).unwrap() << 7
+ } else {
+ br.read(13).unwrap() << 2
+ };
+ let w;
+ let h;
+ if self.rpr.present {
+ let rpr = br.read(self.rpr.bits).unwrap() as usize;
+ if rpr == 0 {
+ w = self.w;
+ h = self.h;
+ } else {
+ w = self.rpr.widths[rpr];
+ h = self.rpr.heights[rpr];
+ validate!((w != 0) && (h != 0));
+ }
+ } else {
+ w = self.w;
+ h = self.h;
+ }
+
+ let mb_pos = br.read(self.mb_pos_bits).unwrap() as usize;
+ let mb_x = mb_pos % self.mb_w;
+ let mb_y = mb_pos / self.mb_w;
+
+ br.skip(1).unwrap(); // no rounding
+
+ if (self.minor_ver <= 1) && (frm_type == 3) {
+ br.skip(5).unwrap();
+ }
+println!("slice q {} mb {},{}", qscale, mb_x, mb_y);
+
+ Ok(RV20SliceInfo::new(ftype, qscale, mb_x, mb_y, mb_pos, w, h))
+ }
+}
+
+fn deblock_ver(buf: &mut NAVideoBuffer<u8>, comp: usize, q: u8, off: usize) {
+ let stride = buf.get_stride(comp);
+ let mut dptr = buf.get_data_mut();
+ let mut buf = dptr.as_mut_slice();
+ for y in 0..8 {
+ let a = buf[off - 2 + y * stride] as i16;
+ let b = buf[off - 1 + y * stride] as i16;
+ let c = buf[off + 0 + y * stride] as i16;
+ let d = buf[off + 1 + y * stride] as i16;
+ let diff = ((a - d) * 3 + (c - b) * 8) >> 4;
+ if (diff != 0) && (diff >= -32) && (diff < 32) {
+ let d0 = diff.abs() * 2 - (q as i16);
+ let d1 = if d0 < 0 { 0 } else { d0 };
+ let d2 = diff.abs() - d1;
+ let d3 = if d2 < 0 { 0 } else { d2 };
+
+ let delta = if diff < 0 { -d3 } else { d3 };
+
+ let b1 = b + delta;
+ if b1 < 0 { buf[off - 1 + y * stride] = 0; }
+ else if b1 > 255 { buf[off - 1 + y * stride] = 0xFF; }
+ else { buf[off - 1 + y * stride] = b1 as u8; }
+ let c1 = c - delta;
+ if c1 < 0 { buf[off + y * stride] = 0; }
+ else if c1 > 255 { buf[off + y * stride] = 0xFF; }
+ else { buf[off + y * stride] = c1 as u8; }
+ }
+ }
+}
+
+impl RealVideo20Decoder {
+ fn new() -> Self {
+ let mut coderead = H263ShortCodeReader::new(H263_INTRA_MCBPC);
+ let intra_mcbpc_cb = Codebook::new(&mut coderead, CodebookMode::MSB).unwrap();
+ let mut coderead = H263ShortCodeReader::new(H263_INTER_MCBPC);
+ let inter_mcbpc_cb = Codebook::new(&mut coderead, CodebookMode::MSB).unwrap();
+ let mut coderead = H263ShortCodeReader::new(H263_CBPY);
+ let cbpy_cb = Codebook::new(&mut coderead, CodebookMode::MSB).unwrap();
+ let mut coderead = H263RLCodeReader::new(H263_RL_CODES);
+ let rl_cb = Codebook::new(&mut coderead, CodebookMode::MSB).unwrap();
+ let mut coderead = H263RLCodeReader::new(H263_RL_CODES_AIC);
+ let aic_rl_cb = Codebook::new(&mut coderead, CodebookMode::MSB).unwrap();
+ let mut coderead = H263ShortCodeReader::new(H263_MV);
+ let mv_cb = Codebook::new(&mut coderead, CodebookMode::MSB).unwrap();
+
+ let tables = Tables {
+ intra_mcbpc_cb: intra_mcbpc_cb,
+ inter_mcbpc_cb: inter_mcbpc_cb,
+ cbpy_cb: cbpy_cb,
+ rl_cb: rl_cb,
+ aic_rl_cb: aic_rl_cb,
+ mv_cb: mv_cb,
+ };
+
+ RealVideo20Decoder{
+ info: Rc::new(DUMMY_CODEC_INFO),
+ dec: H263BaseDecoder::new(),
+ tables: tables,
+ w: 0,
+ h: 0,
+ minor_ver: 0,
+ rpr: RPRInfo { present: false, bits: 0, widths: [0; 8], heights: [0; 8] },
+ }
+ }
+}
+
+impl NADecoder for RealVideo20Decoder {
+ fn init(&mut self, info: Rc<NACodecInfo>) -> DecoderResult<()> {
+ if let NACodecTypeInfo::Video(vinfo) = info.get_properties() {
+ let w = vinfo.get_width();
+ let h = vinfo.get_height();
+ let fmt = formats::YUV420_FORMAT;
+ let myinfo = NACodecTypeInfo::Video(NAVideoInfo::new(w, h, false, fmt));
+ self.info = Rc::new(NACodecInfo::new_ref(info.get_name(), myinfo, info.get_extradata()));
+ self.w = w;
+ self.h = h;
+
+ let edata = info.get_extradata().unwrap();
+ let src: &[u8] = &edata;
+ let ver = ((src[4] as u32) << 12) | ((src[5] as u32) << 4) | ((src[6] as u32) >> 4);
+ let maj_ver = ver >> 16;
+ let min_ver = (ver >> 8) & 0xFF;
+ let mic_ver = ver & 0xFF;
+println!("ver {:06X}", ver);
+ validate!(maj_ver == 2);
+ self.minor_ver = min_ver as u8;
+ let rprb = src[1] & 7;
+ if rprb == 0 {
+ self.rpr.present = false;
+ } else {
+ self.rpr.present = true;
+ self.rpr.bits = rprb as u8;
+ for i in 4..(src.len()/2) {
+ self.rpr.widths [i - 4] = (src[i * 2] as usize) * 4;
+ self.rpr.heights[i - 4] = (src[i * 2 + 1] as usize) * 4;
+ }
+ }
+ Ok(())
+ } else {
+println!(".unwrap().unwrap().unwrap()");
+ Err(DecoderError::InvalidData)
+ }
+ }
+ fn decode(&mut self, pkt: &NAPacket) -> DecoderResult<NAFrameRef> {
+ let src = pkt.get_buffer();
+
+println!(" decode frame size {}, {} slices", src.len(), src[0]+1);
+ let mut ibr = RealVideo20BR::new(&src, &self.tables, self.w, self.h, self.minor_ver, self.rpr);
+
+ let bufinfo = self.dec.parse_frame(&mut ibr).unwrap();
+
+ let mut frm = NAFrame::new_from_pkt(pkt, self.info.clone(), bufinfo);
+ frm.set_keyframe(self.dec.is_intra());
+ frm.set_frame_type(if self.dec.is_intra() { FrameType::I } else { FrameType::P });
+ Ok(Rc::new(RefCell::new(frm)))
+ }
+}
+
+struct MBB { blocks: usize, bits: u8 }
+const H263_MBB: &[MBB; 7] = &[
+ MBB{ blocks: 47, bits: 6 },
+ MBB{ blocks: 98, bits: 7 },
+ MBB{ blocks: 395, bits: 9 },
+ MBB{ blocks: 1583, bits: 11 },
+ MBB{ blocks: 6335, bits: 13 },
+ MBB{ blocks: 9215, bits: 14 },
+ MBB{ blocks: 65536, bits: 14 },
+];
+
+pub fn get_decoder() -> Box<NADecoder> {
+ Box::new(RealVideo20Decoder::new())
+}
+
+#[cfg(test)]
+mod test {
+ use test::dec_video::test_file_decoding;
+ #[test]
+ fn test_rv20() {
+ test_file_decoding("realmedia", "assets/RV/rv20_cook_640x352_realproducer_plus_8.51.rm", None/*Some(160)*/, true, false, Some("rv20"));
+ }
+}
projects / nihav.git / commitdiff