1 use nihav_core::codecs::*;
2 use nihav_core::io::byteio::*;
3 use nihav_core::io::bitreader::*;
4 use nihav_core::io::codebook::*;
5 use nihav_codec_support::codecs::{IPShuffler, HAMShuffler};
7 use super::binkviddata::*;
9 const SKIP_BLOCK: u8 = 0;
10 const SCALED_BLOCK: u8 = 1;
11 const MOTION_BLOCK: u8 = 2;
12 const RUN_BLOCK: u8 = 3;
13 const RESIDUE_BLOCK: u8 = 4;
14 const INTRA_BLOCK: u8 = 5;
15 const FILL_BLOCK: u8 = 6;
16 const INTER_BLOCK: u8 = 7;
17 const PATTERN_BLOCK: u8 = 8;
18 const RAW_BLOCK: u8 = 9;
20 #[derive(Default, Clone,Copy)]
27 fn read_desc(&mut self, br: &mut BitReader) -> DecoderResult<()> {
28 self.id = br.read(4)? as usize;
30 for i in 0..16 { self.syms[i] = i as u8; }
33 let len = br.read(3)? as usize;
34 let mut present: [bool; 16] = [false; 16];
36 self.syms[i] = br.read(4)? as u8;
37 present[self.syms[i] as usize] = true;
39 let mut idx = len + 1;
41 if present[i] { continue; }
42 self.syms[idx] = i as u8;
46 let len = br.read(2)? as usize;
47 let mut syms: [u8; 16] = [ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15];
48 let mut tmp: [u8; 16] = [0; 16];
51 for arr in syms.chunks_mut(size * 2) {
55 while (ptr0 < size) && (ptr1 < size * 2) {
57 tmp[optr] = arr[ptr0];
60 tmp[optr] = arr[ptr1];
66 tmp[optr] = arr[ptr0];
70 while ptr1 < size * 2 {
71 tmp[optr] = arr[ptr1];
75 arr.copy_from_slice(&tmp[0..size * 2]);
83 fn read_sym(&self, br: &mut BitReader, trees: &BinkTrees) -> DecoderResult<u8> {
84 let idx = br.read_cb(&trees.cb[self.id])?;
85 Ok(self.syms[idx as usize])
90 struct Bundle<T: Copy> {
98 impl<T:Copy> Bundle<T> {
99 fn binkb_reset(&mut self, bits: u8) {
104 fn reset(&mut self) {
108 fn read_desc(&mut self, br: &mut BitReader) -> DecoderResult<()> {
111 self.tree.read_desc(br)?;
114 fn read_len(&mut self, br: &mut BitReader) -> DecoderResult<usize> {
115 if self.read_pos < self.dec_pos { return Ok(0); }
116 let len = br.read(self.bits)? as usize;
118 self.dec_pos = self.data.len();
119 self.read_pos = self.data.len() - 1;
123 fn read_len_binkb(&mut self, br: &mut BitReader) -> DecoderResult<usize> {
124 if self.read_pos < self.dec_pos { return Ok(0); }
125 let len = br.read(13)? as usize;
127 self.dec_pos = self.data.len();
128 self.read_pos = self.data.len() - 1;
132 fn get_val(&mut self) -> DecoderResult<T> {
133 validate!(self.read_pos < self.dec_pos);
134 let val = self.data[self.read_pos];
140 const BLOCK_TYPE_RUNS: [usize; 4] = [ 4, 8, 12, 32 ];
142 fn read_binkb(&mut self, br: &mut BitReader) -> DecoderResult<()> {
143 let len = self.read_len_binkb(br)?;
144 if len == 0 { return Ok(()); }
145 let end = self.dec_pos + len;
146 validate!(end <= self.data.len());
148 self.data[self.dec_pos + i] = br.read(self.bits)? as u8;
153 fn read_runs(&mut self, br: &mut BitReader, trees: &BinkTrees) -> DecoderResult<()> {
154 let len = self.read_len(br)?;
155 if len == 0 { return Ok(()); }
156 let end = self.dec_pos + len;
157 validate!(end <= self.data.len());
159 let val = br.read(4)? as u8;
160 for i in 0..len { self.data[self.dec_pos + i] = val; }
163 while self.dec_pos < end {
164 self.data[self.dec_pos] = self.tree.read_sym(br, trees)?;
170 fn read_block_types(&mut self, br: &mut BitReader, trees: &BinkTrees) -> DecoderResult<()> {
171 let len = self.read_len(br)?;
172 if len == 0 { return Ok(()); }
173 let end = self.dec_pos + len;
174 validate!(end <= self.data.len());
176 let val = br.read(4)? as u8;
177 for i in 0..len { self.data[self.dec_pos + i] = val; }
181 while self.dec_pos < end {
182 let val = self.tree.read_sym(br, trees)?;
184 self.data[self.dec_pos] = val;
188 let run = BLOCK_TYPE_RUNS[(val - 12) as usize];
189 validate!(self.dec_pos + run <= end);
191 self.data[self.dec_pos + i] = last;
199 fn read_patterns(&mut self, br: &mut BitReader, trees: &BinkTrees) -> DecoderResult<()> {
200 let len = self.read_len(br)?;
201 if len == 0 { return Ok(()); }
202 let end = self.dec_pos + len;
203 validate!(end <= self.data.len());
205 let pat_lo = self.tree.read_sym(br, trees)?;
206 let pat_hi = self.tree.read_sym(br, trees)?;
207 self.data[self.dec_pos + i] = pat_lo | (pat_hi << 4);
212 fn cvt_color(lo: u8, hi: u8, new_bink: bool) -> u8 {
213 let val = lo | (hi << 4);
215 let sign = ((val as i8) >> 7) as u8;
216 ((val & 0x7F) ^ sign).wrapping_sub(sign) ^ 0x80
221 fn read_colors(&mut self, br: &mut BitReader, trees: &BinkTrees, col_hi: &[Tree; 16], col_last: &mut u8, new_bink: bool) -> DecoderResult<()> {
222 let len = self.read_len(br)?;
223 if len == 0 { return Ok(()); }
224 let end = self.dec_pos + len;
225 validate!(end <= self.data.len());
226 let mut last = *col_last;
228 last = col_hi[last as usize].read_sym(br, trees)?;
229 let lo = self.tree.read_sym(br, trees)?;
230 let val = Self::cvt_color(lo, last, new_bink);
231 for i in 0..len { self.data[self.dec_pos + i] = val; }
234 while self.dec_pos < end {
235 last = col_hi[last as usize].read_sym(br, trees)?;
236 let lo = self.tree.read_sym(br, trees)?;
237 let val = Self::cvt_color(lo, last, new_bink);
238 self.data[self.dec_pos] = val;
248 fn read_binkb(&mut self, br: &mut BitReader) -> DecoderResult<()> {
249 let len = self.read_len_binkb(br)?;
250 if len == 0 { return Ok(()); }
251 let end = self.dec_pos + len;
252 validate!(end <= self.data.len());
253 let bias = 1 << (self.bits - 1);
255 self.data[self.dec_pos + i] = (br.read(self.bits)? as i8) - bias;
260 fn read_motion_values(&mut self, br: &mut BitReader, trees: &BinkTrees) -> DecoderResult<()> {
261 let len = self.read_len(br)?;
262 if len == 0 { return Ok(()); }
263 let end = self.dec_pos + len;
264 validate!(end <= self.data.len());
266 let mut val = br.read(4)? as i8;
267 if val != 0 && br.read_bool()? { val = -val; }
268 for i in 0..len { self.data[self.dec_pos + i] = val; }
271 while self.dec_pos < end {
272 self.data[self.dec_pos] = self.tree.read_sym(br, trees)? as i8;
273 if self.data[self.dec_pos] != 0 && br.read_bool()? {
274 self.data[self.dec_pos] = -self.data[self.dec_pos];
283 const DC_START_BITS: u8 = 11;
285 fn read_binkb(&mut self, br: &mut BitReader) -> DecoderResult<()> {
286 let len = self.read_len_binkb(br)?;
287 if len == 0 { return Ok(()); }
288 let end = self.dec_pos + len;
289 validate!(end <= self.data.len());
291 self.data[self.dec_pos + i] = br.read(self.bits)? as u16;
296 fn read_dcs(&mut self, br: &mut BitReader, start_bits: u8) -> DecoderResult<()> {
297 let len = self.read_len(br)?;
298 if len == 0 { return Ok(()); }
299 let end = self.dec_pos + len;
300 validate!(end <= self.data.len());
301 let mut val = br.read(start_bits)? as u16;
302 self.data[self.dec_pos] = val;
304 for i in (1..len).step_by(8) {
305 let seg_len = (len - i).min(8);
306 let bits = br.read(4)? as u8;
308 for _ in 0..seg_len {
309 let diff = br.read(bits)? as u16;
310 let res = if diff != 0 && br.read_bool()? {
311 val.checked_sub(diff)
313 val.checked_add(diff)
315 validate!(res.is_some());
317 self.data[self.dec_pos] = val;
321 for _ in 0..seg_len {
322 self.data[self.dec_pos] = val;
332 fn read_binkb(&mut self, br: &mut BitReader) -> DecoderResult<()> {
333 let len = self.read_len_binkb(br)?;
334 if len == 0 { return Ok(()); }
335 let end = self.dec_pos + len;
336 validate!(end <= self.data.len());
337 let bias = 1 << (self.bits - 1);
339 self.data[self.dec_pos + i] = (br.read(self.bits)? as i16) - bias;
344 fn read_dcs(&mut self, br: &mut BitReader, start_bits: u8) -> DecoderResult<()> {
345 let len = self.read_len(br)?;
346 if len == 0 { return Ok(()); }
347 let end = self.dec_pos + len;
348 validate!(end <= self.data.len());
349 let mut val = br.read(start_bits - 1)? as i16;
350 if val != 0 && br.read_bool()? {
353 self.data[self.dec_pos] = val;
355 for i in (1..len).step_by(8) {
356 let seg_len = (len - i).min(8);
357 let bits = br.read(4)? as u8;
359 for _ in 0..seg_len {
360 let mut diff = br.read(bits)? as i16;
361 if diff != 0 && br.read_bool()? {
364 let res = val.checked_add(diff);
365 validate!(res.is_some());
367 self.data[self.dec_pos] = val;
371 for _ in 0..seg_len {
372 self.data[self.dec_pos] = val;
382 cb: [Codebook<u8>; 16],
385 fn map_u8(idx: usize) -> u8 { idx as u8 }
387 impl Default for BinkTrees {
388 fn default() -> Self {
390 let mut ucb: std::mem::MaybeUninit::<[Codebook<u8>; 16]> = std::mem::MaybeUninit::uninit();
392 let mut cr = TableCodebookDescReader::new(&BINK_TREE_CODES[i], &BINK_TREE_BITS[i], map_u8);
393 std::ptr::write(&mut (*ucb.as_mut_ptr())[i], Codebook::new(&mut cr, CodebookMode::LSB).unwrap());
401 const A1: i32 = 2896;
402 const A2: i32 = 2217;
403 const A3: i32 = 3784;
404 const A4: i32 = -5352;
407 ($src: expr, $sstep: expr, $dst: expr, $dstep: expr, $off: expr, $bias: expr, $shift: expr) => {
408 let a0 = $src[$off + 0 * $sstep] + $src[$off + 4 * $sstep];
409 let a1 = $src[$off + 0 * $sstep] - $src[$off + 4 * $sstep];
410 let a2 = $src[$off + 2 * $sstep] + $src[$off + 6 * $sstep];
411 let a3 = A1.wrapping_mul($src[$off + 2 * $sstep] - $src[$off + 6 * $sstep]) >> 11;
412 let a4 = $src[$off + 5 * $sstep] + $src[$off + 3 * $sstep];
413 let a5 = $src[$off + 5 * $sstep] - $src[$off + 3 * $sstep];
414 let a6 = $src[$off + 1 * $sstep] + $src[$off + 7 * $sstep];
415 let a7 = $src[$off + 1 * $sstep] - $src[$off + 7 * $sstep];
417 let b1 = A3.wrapping_mul(a5 + a7) >> 11;
418 let b2 = (A4.wrapping_mul(a5) >> 11) - b0 + b1;
419 let b3 = (A1.wrapping_mul(a6 - a4) >> 11) - b2;
420 let b4 = (A2.wrapping_mul(a7) >> 11) + b3 - b1;
423 let c2 = a1 + (a3 - a2);
424 let c3 = a1 - (a3 - a2);
426 $dst[$off + 0 * $dstep] = (c0 + b0 + $bias) >> $shift;
427 $dst[$off + 1 * $dstep] = (c2 + b2 + $bias) >> $shift;
428 $dst[$off + 2 * $dstep] = (c3 + b3 + $bias) >> $shift;
429 $dst[$off + 3 * $dstep] = (c1 - b4 + $bias) >> $shift;
430 $dst[$off + 4 * $dstep] = (c1 + b4 + $bias) >> $shift;
431 $dst[$off + 5 * $dstep] = (c3 - b3 + $bias) >> $shift;
432 $dst[$off + 6 * $dstep] = (c2 - b2 + $bias) >> $shift;
433 $dst[$off + 7 * $dstep] = (c0 - b0 + $bias) >> $shift;
439 info: NACodecInfoRef,
441 hams: HAMShuffler<u8>,
454 colhi_tree: [Tree; 16],
463 intradc: Bundle<u16>,
464 interdc: Bundle<i16>,
467 nresidues: Bundle<u8>,
475 fn calc_len(size: usize) -> u8 {
476 (32 - ((size + 511) as u32).leading_zeros()) as u8
483 fn init_bundle_bufs(&mut self, bw: usize, bh: usize) {
484 let size = bw * bh * 64;
485 self.btype.data.resize(size, 0);
486 self.sbtype.data.resize(size, 0);
487 self.colors.data.resize(size, 0);
488 self.pattern.data.resize(size, 0);
489 self.xoff.data.resize(size, 0);
490 self.yoff.data.resize(size, 0);
491 self.intradc.data.resize(size, 0);
492 self.interdc.data.resize(size, 0);
493 self.intraq.data.resize(size, 0);
494 self.interq.data.resize(size, 0);
495 self.nresidues.data.resize(size, 0);
496 self.run.data.resize(size, 0);
498 fn init_bundle_lengths(&mut self, w: usize, bw: usize) {
499 let w = (w + 7) & !7;
500 self.btype.bits = calc_len(w >> 3);
501 self.sbtype.bits = calc_len(w >> 4);
502 self.colors.bits = calc_len(bw * 64);
503 self.pattern.bits = calc_len(bw * 8);
504 self.xoff.bits = calc_len(w >> 3);
505 self.yoff.bits = calc_len(w >> 3);
506 self.intradc.bits = calc_len(w >> 3);
507 self.interdc.bits = calc_len(w >> 3);
508 self.run.bits = calc_len(bw * 48);
510 fn init_bundle_lengths_binkb(&mut self) {
511 self.btype.binkb_reset(4);
512 self.colors.binkb_reset(8);
513 self.pattern.binkb_reset(8);
514 self.xoff.binkb_reset(5);
515 self.yoff.binkb_reset(5);
516 self.intradc.binkb_reset(11);
517 self.interdc.binkb_reset(11);
518 self.intraq.binkb_reset(4);
519 self.interq.binkb_reset(4);
520 self.nresidues.binkb_reset(7);
522 fn read_bundles_desc(&mut self, br: &mut BitReader) -> DecoderResult<()> {
523 self.btype.read_desc(br)?;
524 self.sbtype.read_desc(br)?;
525 for el in &mut self.colhi_tree {
529 self.colors.read_desc(br)?;
530 self.pattern.read_desc(br)?;
531 self.xoff.read_desc(br)?;
532 self.yoff.read_desc(br)?;
533 self.intradc.reset();
534 self.interdc.reset();
535 self.run.read_desc(br)?;
538 fn read_bundles_binkb(&mut self, br: &mut BitReader) -> DecoderResult<()> {
539 self.btype.read_binkb(br)?;
540 self.colors.read_binkb(br)?;
541 self.pattern.read_binkb(br)?;
542 self.xoff.read_binkb(br)?;
543 self.yoff.read_binkb(br)?;
544 self.intradc.read_binkb(br)?;
545 self.interdc.read_binkb(br)?;
546 self.intraq.read_binkb(br)?;
547 self.interq.read_binkb(br)?;
548 self.nresidues.read_binkb(br)?;
551 fn read_bundles(&mut self, br: &mut BitReader) -> DecoderResult<()> {
552 self.btype.read_block_types(br, &self.trees)?;
553 self.sbtype.read_block_types(br, &self.trees)?;
554 self.colors.read_colors(br, &self.trees, &self.colhi_tree, &mut self.col_last, self.is_ver_i)?;
555 self.pattern.read_patterns(br, &self.trees)?;
556 self.xoff.read_motion_values(br, &self.trees)?;
557 self.yoff.read_motion_values(br, &self.trees)?;
558 self.intradc.read_dcs(br, DC_START_BITS)?;
559 self.interdc.read_dcs(br, DC_START_BITS)?;
560 self.run.read_runs(br, &self.trees)?;
564 fn put_block(&self, block: &[u8; 64], dst: &mut [u8], mut off: usize, stride: usize, scaled: bool) {
566 for src in block.chunks_exact(8) {
567 let out = &mut dst[off..][..8];
568 out.copy_from_slice(src);
572 for src in block.chunks_exact(8) {
574 dst[off + i * 2 + 0] = src[i];
575 dst[off + i * 2 + 1] = src[i];
579 dst[off + i * 2 + 0] = src[i];
580 dst[off + i * 2 + 1] = src[i];
586 fn copy_block(&mut self, dst: &mut [u8], mut off: usize, stride: usize, bx: usize, by: usize, xoff: i8, yoff: i8) -> DecoderResult<()> {
587 if let Some(prev_buf) = self.ips.get_ref() {
588 let xoff = ((bx * 8) as isize) + (xoff as isize);
589 let yoff = ((by * 8) as isize) + (yoff as isize);
590 validate!((xoff >= 0) && (xoff + 8 <= (self.cur_w as isize)));
591 validate!((yoff >= 0) && (yoff + 8 <= (self.cur_h as isize)));
592 let pstride = prev_buf.get_stride(self.cur_plane);
593 let mut poff = prev_buf.get_offset(self.cur_plane) + (xoff as usize) + (yoff as usize) * pstride;
594 let pdata = prev_buf.get_data();
595 let ppix = pdata.as_slice();
597 let src = &ppix[poff..][..8];
598 let out = &mut dst[off..][..8];
599 out.copy_from_slice(src);
605 Err(DecoderError::MissingReference)
608 fn copy_overlapped(&mut self, dst: &mut [u8], mut off: usize, stride: usize, bx: usize, by: usize, xoff: i8, yoff1: i8) -> DecoderResult<()> {
609 let ybias = if self.key_frame { -15 } else { 0 };
610 let yoff = yoff1 + ybias;
612 let xpos = ((bx * 8) as isize) + (xoff as isize);
613 let ypos = ((by * 8) as isize) + (yoff as isize);
614 validate!((xpos >= 0) && (xpos + 8 <= (self.cur_w as isize)));
615 validate!((ypos >= 0) && (ypos + 8 <= (self.cur_h as isize)));
617 let mut block: [u8; 64] = [0; 64];
618 let mut ref_off = ((off as isize) + (xoff as isize) + (yoff as isize) * (stride as isize)) as usize;
619 for row in block.chunks_exact_mut(8) {
620 row.copy_from_slice(&dst[ref_off..][..8]);
623 for row in block.chunks_exact(8) {
624 let out = &mut dst[off..][..8];
625 out.copy_from_slice(row);
631 fn add_block(&self, coeffs: &[i32; 64], dst: &mut [u8], mut off: usize, stride: usize) {
632 for src in coeffs.chunks_exact(8) {
634 let v = (dst[off + i] as i32) + src[i];
635 dst[off + i] = v as u8;
640 fn idct_put(&self, coeffs: &[i32; 64], dst: &mut [u8], mut off: usize, stride: usize) {
641 let mut tmp: [i32; 64] = [0; 64];
642 let mut row: [i32; 8] = [0; 8];
644 idct!(coeffs, 8, tmp, 8, i, 0, 0);
646 for srow in tmp.chunks_exact(8) {
647 idct!(srow, 1, row, 1, 0, 0x7F, 8);
649 dst[off + i] = row[i] as u8;
654 fn idct_add(&self, coeffs: &[i32; 64], dst: &mut [u8], mut off: usize, stride: usize) {
655 let mut tmp: [i32; 64] = [0; 64];
656 let mut row: [i32; 8] = [0; 8];
658 idct!(coeffs, 8, tmp, 8, i, 0, 0);
660 for srow in tmp.chunks_exact(8) {
661 idct!(srow, 1, row, 1, 0, 0x7F, 8);
663 let v = (dst[off + i] as i32) + row[i];
664 dst[off + i] = v as u8;
670 fn decode_plane_binkb(&mut self, br: &mut BitReader, plane_no: usize, buf: &mut NAVideoBuffer<u8>) -> DecoderResult<()> {
671 let stride = buf.get_stride(plane_no);
672 let mut off = buf.get_offset(plane_no);
673 let (width, height) = buf.get_dimensions(plane_no);
674 let data = buf.get_data_mut().unwrap();
675 let dst = data.as_mut_slice();
676 let bw = (width + 7) >> 3;
677 let bh = (height + 7) >> 3;
678 self.cur_w = (width + 7) & !7;
679 self.cur_h = (height + 7) & !7;
680 self.cur_plane = plane_no;
681 self.init_bundle_lengths_binkb();
683 self.read_bundles_binkb(br)?;
685 let mut coeffs: [i32; 64] = [0; 64];
686 let btype = self.btype.get_val()?;
691 let scan = BINK_PATTERNS[br.read(4)? as usize];
694 let run = br.read_bool()?;
695 let len = (br.read(BINKB_RUN_BITS[idx])? as usize) + 1;
696 validate!(idx + len <= 64);
698 let val = self.colors.get_val()?;
700 let pos = scan[idx + j] as usize;
701 dst[off + (pos >> 3) * stride + (pos & 7)] = val;
706 let pos = scan[idx] as usize;
707 dst[off + (pos >> 3) * stride + (pos & 7)] = self.colors.get_val()?;
713 let pos = scan[idx] as usize;
714 dst[off + (pos >> 3) * stride + (pos & 7)] = self.colors.get_val()?;
718 coeffs[0] = self.intradc.get_val()? as i32;
719 let q = self.intraq.get_val()? as usize;
720 read_dct_coefficients(br, &mut coeffs, &BINK_SCAN, &self.qmat_b.intra_qmat, Some(q))?;
721 self.idct_put(&coeffs, dst, off, stride);
724 let xoff = self.xoff.get_val()?;
725 let yoff = self.yoff.get_val()?;
726 self.copy_overlapped(dst, off, stride, bx, by, xoff, yoff)?;
727 let nmasks = self.nresidues.get_val()? as usize;
728 read_residue(br, &mut coeffs, nmasks)?;
729 self.add_block(&coeffs, dst, off, stride);
732 let xoff = self.xoff.get_val()?;
733 let yoff = self.yoff.get_val()?;
734 self.copy_overlapped(dst, off, stride, bx, by, xoff, yoff)?;
735 coeffs[0] = self.interdc.get_val()? as i32;
736 let q = self.interq.get_val()? as usize;
737 read_dct_coefficients(br, &mut coeffs, &BINK_SCAN, &self.qmat_b.inter_qmat, Some(q))?;
738 self.idct_add(&coeffs, dst, off, stride);
741 let fill = self.colors.get_val()?;
743 for j in 0..8 { dst[off + i * stride + j] = fill; }
747 let clr: [u8; 2] = [ self.colors.get_val()?, self.colors.get_val()? ];
749 let pattern = self.pattern.get_val()? as usize;
751 dst[off + i * stride + j] = clr[(pattern >> j) & 1];
755 7 => { // motion block
756 let xoff = self.xoff.get_val()?;
757 let yoff = self.yoff.get_val()?;
758 self.copy_overlapped(dst, off, stride, bx, by, xoff, yoff)?;
763 dst[off + i * stride + j] = self.colors.get_val()?;
767 _ => { return Err(DecoderError::InvalidData); },
771 off += stride * 8 - bw * 8;
773 if (br.tell() & 0x1F) != 0 {
774 let skip = (32 - (br.tell() & 0x1F)) as u32;
779 fn handle_block(&mut self, br: &mut BitReader, bx: usize, by: usize,
780 dst: &mut [u8], off: usize, stride: usize, btype: u8, scaled: bool) -> DecoderResult<()> {
781 let mut oblock: [u8; 64] = [0; 64];
782 let mut coeffs: [i32; 64] = [0; 64];
786 self.copy_block(dst, off, stride, bx, by, 0, 0)?;
790 let sbtype = self.sbtype.get_val()?;
791 self.handle_block(br, bx, by, dst, off, stride, sbtype, true)?;
795 let xoff = self.xoff.get_val()?;
796 let yoff = self.yoff.get_val()?;
797 self.copy_block(dst, off, stride, bx, by, xoff, yoff)?;
800 let scan = BINK_PATTERNS[br.read(4)? as usize];
803 let run = (self.run.get_val()? as usize) + 1;
804 validate!(idx + run <= 64);
806 let val = self.colors.get_val()?;
808 oblock[scan[idx + j] as usize] = val;
813 oblock[scan[idx] as usize] = self.colors.get_val()?;
818 if idx == 63 { oblock[scan[63] as usize] = self.colors.get_val()?; }
819 self.put_block(&oblock, dst, off, stride, scaled);
823 let xoff = self.xoff.get_val()?;
824 let yoff = self.yoff.get_val()?;
825 self.copy_block(dst, off, stride, bx, by, xoff, yoff)?;
826 let nmasks = br.read(7)? as usize;
827 read_residue(br, &mut coeffs, nmasks)?;
828 self.add_block(&coeffs, dst, off, stride);
831 coeffs[0] = self.intradc.get_val()? as i32;
832 read_dct_coefficients(br, &mut coeffs, &BINK_SCAN, BINK_INTRA_QUANT, None)?;
834 self.idct_put(&coeffs, dst, off, stride);
836 self.idct_put(&coeffs, &mut oblock, 0, 8);
837 self.put_block(&oblock, dst, off, stride, scaled);
841 let fill = self.colors.get_val()?;
843 self.put_block(&oblock, dst, off, stride, scaled);
847 let xoff = self.xoff.get_val()?;
848 let yoff = self.yoff.get_val()?;
849 self.copy_block(dst, off, stride, bx, by, xoff, yoff)?;
850 coeffs[0] = self.interdc.get_val()? as i32;
851 read_dct_coefficients(br, &mut coeffs, &BINK_SCAN, BINK_INTER_QUANT, None)?;
852 self.idct_add(&coeffs, dst, off, stride);
855 let clr: [u8; 2] = [ self.colors.get_val()?, self.colors.get_val()? ];
857 let pattern = self.pattern.get_val()? as usize;
859 oblock[i * 8 + j] = clr[(pattern >> j) & 1];
862 self.put_block(&oblock, dst, off, stride, scaled);
867 oblock[i * 8 + j] = self.colors.get_val()?;
870 self.put_block(&oblock, dst, off, stride, scaled);
872 _ => { return Err(DecoderError::InvalidData); },
876 fn decode_plane(&mut self, br: &mut BitReader, plane_no: usize, buf: &mut NAVideoBuffer<u8>) -> DecoderResult<()> {
877 let stride = buf.get_stride(plane_no);
878 let mut off = buf.get_offset(plane_no);
879 let (width, height) = buf.get_dimensions(plane_no);
880 let data = buf.get_data_mut().unwrap();
881 let dst = data.as_mut_slice();
882 let bw = (width + 7) >> 3;
883 let bh = (height + 7) >> 3;
884 self.cur_w = (width + 7) & !7;
885 self.cur_h = (height + 7) & !7;
886 self.cur_plane = plane_no;
887 self.init_bundle_lengths(width.max(8), bw);
888 self.read_bundles_desc(br)?;
890 self.read_bundles(br)?;
893 let btype = self.btype.get_val()?;
894 if btype == SCALED_BLOCK && (by & 1) == 1 { // already decoded scaled block, skip
898 self.handle_block(br, bx, by, dst, off + bx * 8, stride, btype, false)?;
899 if btype == SCALED_BLOCK {
906 if (br.tell() & 0x1F) != 0 {
907 let skip = (32 - (br.tell() & 0x1F)) as u32;
914 fn get_coef(br: &mut BitReader, bits1: u8) -> DecoderResult<i32> {
917 t = if br.read_bool()? { -1 } else { 1 };
919 let bits = bits1 - 1;
920 let val = (br.read(bits)? as i32) | (1 << bits);
930 fn read_dct_coefficients(br: &mut BitReader, block: &mut [i32; 64], scan: &[usize; 64],
931 quant_matrices: &[[i32; 64]; 16], q: Option<usize>) -> DecoderResult<()> {
932 let mut coef_list: [i32; 128] = [0; 128];
933 let mut mode_list: [u8; 128] = [0; 128];
934 let mut list_start = 64;
935 let mut list_end = 64;
936 let mut coef_idx: [usize; 64] = [0; 64];
937 let mut coef_count = 0;
939 coef_list[list_end] = 4; mode_list[list_end] = 0; list_end += 1;
940 coef_list[list_end] = 24; mode_list[list_end] = 0; list_end += 1;
941 coef_list[list_end] = 44; mode_list[list_end] = 0; list_end += 1;
942 coef_list[list_end] = 1; mode_list[list_end] = 3; list_end += 1;
943 coef_list[list_end] = 2; mode_list[list_end] = 3; list_end += 1;
944 coef_list[list_end] = 3; mode_list[list_end] = 3; list_end += 1;
946 let mut bits1 = br.read(4)? as u8;
948 let mut list_pos = list_start;
949 while list_pos < list_end {
950 let ccoef = coef_list[list_pos];
951 let mode = mode_list[list_pos];
952 if (mode == 0 && ccoef == 0) || !br.read_bool()? {
959 coef_list[list_pos] = ccoef + 4;
960 mode_list[list_pos] = 1;
962 coef_list[list_pos] = 0;
963 mode_list[list_pos] = 0;
969 coef_list[list_start] = ccoef + i;
970 mode_list[list_start] = 3;
972 let idx = (ccoef + i) as usize;
973 block[scan[idx]] = get_coef(br, bits1)?;
974 coef_idx[coef_count] = idx;
980 mode_list[list_pos] = 2;
982 coef_list[list_end] = ccoef + i * 4 + 4;
983 mode_list[list_end] = 2;
988 let idx = ccoef as usize;
989 block[scan[idx]] = get_coef(br, bits1)?;
990 coef_idx[coef_count] = idx;
992 coef_list[list_pos] = 0;
993 mode_list[list_pos] = 0;
1002 let q_index = if let Some(qidx) = q { qidx } else { br.read(4)? as usize };
1003 let qmat = &quant_matrices[q_index];
1004 block[0] = block[0].wrapping_mul(qmat[0]) >> 11;
1005 for idx in coef_idx.iter().take(coef_count) {
1006 block[scan[*idx]] = block[scan[*idx]].wrapping_mul(qmat[*idx]) >> 11;
1012 fn read_residue(br: &mut BitReader, block: &mut [i32; 64], mut masks_count: usize) -> DecoderResult<()> {
1013 let mut coef_list: [i32; 128] = [0; 128];
1014 let mut mode_list: [u8; 128] = [0; 128];
1015 let mut list_start = 64;
1016 let mut list_end = 64;
1017 let mut nz_coef_idx: [usize; 64] = [0; 64];
1018 let mut nz_coef_count = 0;
1020 coef_list[list_end] = 4; mode_list[list_end] = 0; list_end += 1;
1021 coef_list[list_end] = 24; mode_list[list_end] = 0; list_end += 1;
1022 coef_list[list_end] = 44; mode_list[list_end] = 0; list_end += 1;
1023 coef_list[list_end] = 0; mode_list[list_end] = 2; list_end += 1;
1025 let mut mask = 1 << br.read(3)?;
1027 for i in 0..nz_coef_count {
1028 if !br.read_bool()? { continue; }
1029 let idx = nz_coef_idx[i];
1035 if masks_count == 0 {
1040 let mut list_pos = list_start;
1041 while list_pos < list_end {
1042 let ccoef = coef_list[list_pos];
1043 let mode = mode_list[list_pos];
1044 if (mode == 0 && ccoef == 0) || !br.read_bool()? {
1051 coef_list[list_pos] = ccoef + 4;
1052 mode_list[list_pos] = 1;
1054 coef_list[list_pos] = 0;
1055 mode_list[list_pos] = 0;
1059 if br.read_bool()? {
1061 coef_list[list_start] = ccoef + i;
1062 mode_list[list_start] = 3;
1064 let idx = (ccoef + i) as usize;
1065 nz_coef_idx[nz_coef_count] = BINK_SCAN[idx];
1067 block[BINK_SCAN[idx]] = if br.read_bool()? { -mask } else { mask };
1068 if masks_count == 0 {
1076 mode_list[list_pos] = 2;
1078 coef_list[list_end] = ccoef + i * 4 + 4;
1079 mode_list[list_end] = 2;
1084 let idx = ccoef as usize;
1085 nz_coef_idx[nz_coef_count] = BINK_SCAN[idx];
1087 block[BINK_SCAN[idx]] = if br.read_bool()? { -mask } else { mask };
1088 coef_list[list_pos] = 0;
1089 mode_list[list_pos] = 0;
1091 if masks_count == 0 {
1096 _ => unreachable!(),
1105 const BINK_FLAG_ALPHA: u32 = 0x00100000;
1106 const BINK_FLAG_GRAY: u32 = 0x00020000;
1108 impl NADecoder for BinkDecoder {
1109 fn init(&mut self, _supp: &mut NADecoderSupport, info: NACodecInfoRef) -> DecoderResult<()> {
1110 if let NACodecTypeInfo::Video(vinfo) = info.get_properties() {
1111 let w = vinfo.get_width();
1112 let h = vinfo.get_height();
1114 let edata = info.get_extradata().unwrap();
1115 validate!(edata.len() >= 8);
1117 let mut mr = MemoryReader::new_read(&edata);
1118 let mut br = ByteReader::new(&mut mr);
1119 let magic = br.read_u32be()?;
1120 let flags = br.read_u32le()?;
1122 self.is_ver_b = (magic & 0xFF) == (b'b' as u32);
1123 self.is_ver_i = (magic & 0xFF) >= (b'i' as u32);
1124 self.has_alpha = (flags & BINK_FLAG_ALPHA) != 0;
1125 self.is_gray = (flags & BINK_FLAG_GRAY) != 0;
1126 self.swap_uv = (magic & 0xFF) >= (b'h' as u32);
1127 if self.has_alpha && self.is_gray { return Err(DecoderError::NotImplemented); }
1129 let aplane = if self.has_alpha { Some(NAPixelChromaton::new(0, 0, false, 8, 0, 3, 1)) } else { None };
1132 fmt = NAPixelFormaton::new(ColorModel::YUV(YUVSubmodel::YUVJ),
1133 Some(NAPixelChromaton::new(0, 0, false, 8, 0, 0, 1)),
1134 Some(NAPixelChromaton::new(1, 1, false, 8, 0, 1, 1)),
1135 Some(NAPixelChromaton::new(1, 1, false, 8, 0, 2, 1)),
1137 0, if self.has_alpha { 4 } else { 3 } );
1139 fmt = NAPixelFormaton::new(ColorModel::YUV(YUVSubmodel::YUVJ),
1140 Some(NAPixelChromaton::new(0, 0, false, 8, 0, 0, 1)),
1141 None, None, None, None, 0, 1);
1143 let myinfo = NACodecTypeInfo::Video(NAVideoInfo::new(w, h, false, fmt));
1144 self.info = NACodecInfo::new_ref(info.get_name(), myinfo, info.get_extradata()).into_ref();
1146 //self.init_bundle_lengths(w.max(8), (w + 7) >> 3);
1147 self.init_bundle_bufs((w + 7) >> 3, (h + 7) >> 3);
1150 self.qmat_b.calc_binkb_quants();
1155 Err(DecoderError::InvalidData)
1158 fn decode(&mut self, _supp: &mut NADecoderSupport, pkt: &NAPacket) -> DecoderResult<NAFrameRef> {
1159 let src = pkt.get_buffer();
1161 let mut br = BitReader::new(&src, BitReaderMode::LE);
1164 self.key_frame = pkt.is_keyframe();
1166 let bufret = self.hams.clone_ref();
1167 if let Some(bbuf) = bufret {
1170 let bufinfo = alloc_video_buffer(self.info.get_properties().get_video_info().unwrap(), 4)?;
1171 buf = bufinfo.get_vbuf().unwrap();
1172 self.key_frame = true;
1173 self.hams.add_frame(buf);
1174 buf = self.hams.get_output_frame().unwrap();
1177 let bufinfo = alloc_video_buffer(self.info.get_properties().get_video_info().unwrap(), 4)?;
1178 buf = bufinfo.get_vbuf().unwrap();
1181 let nplanes = if self.is_gray { 1 } else { 3 };
1183 validate!(!self.is_ver_b);
1187 self.decode_plane(&mut br, nplanes, &mut buf)?;
1192 for plane in 0..nplanes {
1194 self.decode_plane_binkb(&mut br, plane, &mut buf)?;
1196 let plane_idx = if plane > 0 && self.swap_uv { plane ^ 3 } else { plane };
1197 self.decode_plane(&mut br, plane_idx, &mut buf)?;
1200 let bufinfo = NABufferType::Video(buf);
1202 self.ips.add_frame(bufinfo.get_vbuf().unwrap());
1205 let mut frm = NAFrame::new_from_pkt(pkt, self.info.clone(), bufinfo);
1206 frm.set_frame_type(FrameType::P);
1209 fn flush(&mut self) {
1214 impl NAOptionHandler for BinkDecoder {
1215 fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] }
1216 fn set_options(&mut self, _options: &[NAOption]) { }
1217 fn query_option_value(&self, _name: &str) -> Option<NAValue> { None }
1220 pub fn get_decoder() -> Box<dyn NADecoder + Send> {
1221 Box::new(BinkDecoder::new())
1226 use nihav_core::codecs::RegisteredDecoders;
1227 use nihav_core::demuxers::RegisteredDemuxers;
1228 use nihav_codec_support::test::dec_video::*;
1229 use crate::rad_register_all_decoders;
1230 use crate::rad_register_all_demuxers;
1232 fn test_binkvid_b() {
1233 let mut dmx_reg = RegisteredDemuxers::new();
1234 rad_register_all_demuxers(&mut dmx_reg);
1235 let mut dec_reg = RegisteredDecoders::new();
1236 rad_register_all_decoders(&mut dec_reg);
1238 // sample: https://samples.mplayerhq.hu/game-formats/bink/bikb/NEW.BIK
1239 test_decoding("bink", "bink-video", "assets/RAD/NEW.BIK", Some(16), &dmx_reg, &dec_reg,
1240 ExpectedTestResult::MD5Frames(vec![
1241 [0x00edef7e, 0x7efad3b1, 0x7e8bdd74, 0x3f6b00ba],
1242 [0xbc40683f, 0xbeb1c5e4, 0x934777b5, 0x8a8a350d],
1243 [0x68b78627, 0x28ceb63d, 0xfdb1171a, 0x23e69d90],
1244 [0xc8d907a0, 0xb8d44079, 0x0286336b, 0x996479f3],
1245 [0x57bbe4ec, 0xdb8bb9c2, 0x0e6f1fd6, 0xe180125e],
1246 [0xd43c2ae0, 0x4010007f, 0x2a6360a1, 0xb5203a05],
1247 [0xa883acf6, 0x25843f92, 0x4ced9a46, 0x6d513ad9],
1248 [0x959e843f, 0x8d8182b9, 0x3f12d29b, 0x2af8d39f],
1249 [0x93840946, 0x1188c6d1, 0xd5499833, 0x62aac0c6],
1250 [0x4e5a56a6, 0x21517d9a, 0xbe1f270d, 0xe5621945],
1251 [0x1b133742, 0x1eb1bf0a, 0x68cab2e3, 0x92176b5d],
1252 [0x0cf78c43, 0x4bc15549, 0x3dd94323, 0x737eaaae],
1253 [0xdd731c4a, 0x801453b3, 0xa38bef3e, 0x285cfdfe],
1254 [0xe1fec4ee, 0x46737abc, 0x8c452209, 0xc8c6addd],
1255 [0x2978aa50, 0x5f1e6d5a, 0x1f5b0fba, 0xb8e32196],
1256 [0x2e1e95ab, 0x8e31a0b0, 0xfe998512, 0xea9397b6],
1257 [0xf7f6c0d8, 0x893e77a7, 0xdfe0309f, 0xf5e644e2]]));
1261 let mut dmx_reg = RegisteredDemuxers::new();
1262 rad_register_all_demuxers(&mut dmx_reg);
1263 let mut dec_reg = RegisteredDecoders::new();
1264 rad_register_all_decoders(&mut dec_reg);
1266 // sample: https://samples.mplayerhq.hu/game-formats/bink/ActivisionLogo.bik
1267 test_decoding("bink", "bink-video", "assets/RAD/ActivisionLogo.bik", Some(42),
1269 ExpectedTestResult::MD5([0x41128884, 0x73a8c710, 0x5072ea4a, 0x8caca428]));