nihav-duck/src/codecs/vp6.rs

   1 use nihav_core::codecs::*;
   2 use nihav_core::io::bitreader::*;
   3 use nihav_codec_support::codecs::{MV, ZIGZAG};
   4 use nihav_codec_support::codecs::blockdsp::edge_emu;
   5 use super::vpcommon::*;
   6 use super::vp56::*;
   7 use super::vp6data::*;
   8
   9 #[derive(Default)]
  10 struct VP6BR {
  11     vpversion:      u8,
  12     profile:        u8,
  13     interlaced:     bool,
  14     do_pm:          bool,
  15     loop_mode:      u8,
  16     autosel_pm:     bool,
  17     var_thresh:     u16,
  18     mv_thresh:      u8,
  19     bicubic:        bool,
  20     filter_alpha:   usize,
  21 }
  22
  23 impl VP6BR {
  24     fn new() -> Self {
  25         Self::default()
  26     }
  27 }
  28
  29 impl VP56Parser for VP6BR {
  30     fn parse_header(&mut self, bc: &mut BoolCoder) -> DecoderResult<VP56Header> {
  31         let mut hdr = VP56Header::default();
  32 // horrible hack to match VP6 header parsing
  33         let src = bc.src;
  34         let mut br = BitReader::new(src, BitReaderMode::BE);
  35
  36         hdr.is_intra                            = !br.read_bool()?;
  37         hdr.is_golden = hdr.is_intra;
  38         hdr.quant                               = br.read(6)? as u8;
  39         hdr.multistream                         = br.read_bool()?;
  40         if hdr.is_intra {
  41             hdr.version                         = br.read(5)? as u8;
  42             validate!((hdr.version >= VERSION_VP60) && (hdr.version <= VERSION_VP62));
  43             hdr.profile                         = br.read(2)? as u8;
  44             validate!((hdr.profile == VP6_SIMPLE_PROFILE) || (hdr.profile == VP6_ADVANCED_PROFILE));
  45             hdr.interlaced                      = br.read_bool()?;
  46         } else {
  47             hdr.version = self.vpversion;
  48             hdr.profile = self.profile;
  49             hdr.interlaced = self.interlaced;
  50         }
  51         if hdr.multistream || (hdr.profile == VP6_SIMPLE_PROFILE) {
  52             hdr.offset                          = br.read(16)? as u16;
  53             validate!(hdr.offset > if hdr.is_intra { 6 } else { 2 });
  54             hdr.multistream = true;
  55         }
  56         let bytes = br.tell() >> 3;
  57         std::mem::drop(br);
  58         bc.skip_bytes(bytes);
  59         self.loop_mode = 0;
  60         if hdr.is_intra {
  61             hdr.mb_h                            = bc.read_bits(8) as u8;
  62             hdr.mb_w                            = bc.read_bits(8) as u8;
  63             hdr.disp_h                          = bc.read_bits(8) as u8;
  64             hdr.disp_w                          = bc.read_bits(8) as u8;
  65             validate!((hdr.mb_h > 0) && (hdr.mb_w > 0) && (hdr.disp_w > 0) && (hdr.disp_h > 0));
  66             validate!((hdr.disp_w <= hdr.mb_w) && (hdr.disp_h <= hdr.mb_h));
  67             hdr.scale                           = bc.read_bits(2) as u8;
  68         } else {
  69             hdr.is_golden                       = bc.read_bool();
  70             if hdr.profile == VP6_ADVANCED_PROFILE {
  71                 self.loop_mode                  = bc.read_bool() as u8;
  72                 if self.loop_mode != 0 {
  73                     self.loop_mode             += bc.read_bool() as u8;
  74                     validate!(self.loop_mode <= 1);
  75                 }
  76                 if hdr.version == VERSION_VP62 {
  77                     self.do_pm                  = bc.read_bool();
  78                 }
  79             }
  80         }
  81
  82         if (hdr.profile == VP6_ADVANCED_PROFILE) && (hdr.is_intra || self.do_pm) {
  83             self.autosel_pm                     = bc.read_bool();
  84             if self.autosel_pm {
  85                 self.var_thresh                 = bc.read_bits(5) as u16;
  86                 if hdr.version != VERSION_VP62 {
  87                     self.var_thresh <<= 5;
  88                 }
  89                 self.mv_thresh                  = bc.read_bits(3) as u8;
  90             } else {
  91                 self.bicubic                    = bc.read_bool();
  92             }
  93             if hdr.version == VERSION_VP62 {
  94                 self.filter_alpha               = bc.read_bits(4) as usize;
  95             } else {
  96                 self.filter_alpha = 16;
  97             }
  98         }
  99
 100         hdr.use_huffman                         = bc.read_bool();
 101
 102         self.vpversion  = hdr.version;
 103         self.profile    = hdr.profile;
 104         self.interlaced = hdr.interlaced;
 105         Ok(hdr)
 106     }
 107     fn decode_mv(&self, bc: &mut BoolCoder, model: &VP56MVModel) -> i16 {
 108         let val = if !bc.read_prob(model.nz_prob) { // short vector
 109                 vp_tree!(bc, model.tree_probs[0],
 110                          vp_tree!(bc, model.tree_probs[1],
 111                                   vp_tree!(bc, model.tree_probs[2], 0, 1),
 112                                   vp_tree!(bc, model.tree_probs[3], 2, 3)),
 113                          vp_tree!(bc, model.tree_probs[4],
 114                                   vp_tree!(bc, model.tree_probs[5], 4, 5),
 115                                   vp_tree!(bc, model.tree_probs[6], 6, 7)))
 116             } else {
 117                 let mut raw = 0;
 118                 for ord in LONG_VECTOR_ORDER.iter() {
 119                     raw                         |= (bc.read_prob(model.raw_probs[*ord]) as i16) << *ord;
 120                 }
 121                 if (raw & 0xF0) != 0 {
 122                     raw                         |= (bc.read_prob(model.raw_probs[3]) as i16) << 3;
 123                 } else {
 124                     raw |= 1 << 3;
 125                 }
 126                 raw
 127             };
 128         if (val != 0) && bc.read_prob(model.sign_prob) {
 129             -val
 130         } else {
 131             val
 132         }
 133     }
 134     fn reset_models(&self, models: &mut VP56Models) {
 135         for (i, mdl) in models.mv_models.iter_mut().enumerate() {
 136             mdl.nz_prob         = NZ_PROBS[i];
 137             mdl.sign_prob       = 128;
 138             mdl.raw_probs.copy_from_slice(&RAW_PROBS[i]);
 139             mdl.tree_probs.copy_from_slice(&TREE_PROBS[i]);
 140         }
 141         models.vp6models.zero_run_probs.copy_from_slice(&ZERO_RUN_PROBS);
 142         reset_scan(&mut models.vp6models, self.interlaced);
 143     }
 144     fn decode_mv_models(&self, bc: &mut BoolCoder, models: &mut [VP56MVModel; 2]) -> DecoderResult<()> {
 145         for comp in 0..2 {
 146             if bc.read_prob(HAS_NZ_PROB[comp]) {
 147                 models[comp].nz_prob            = bc.read_probability();
 148             }
 149             if bc.read_prob(HAS_SIGN_PROB[comp]) {
 150                 models[comp].sign_prob          = bc.read_probability();
 151             }
 152         }
 153         for comp in 0..2 {
 154             for (i, prob) in HAS_TREE_PROB[comp].iter().enumerate() {
 155                 if bc.read_prob(*prob) {
 156                     models[comp].tree_probs[i]  = bc.read_probability();
 157                 }
 158             }
 159         }
 160         for comp in 0..2 {
 161             for (i, prob) in HAS_RAW_PROB[comp].iter().enumerate() {
 162                 if bc.read_prob(*prob) {
 163                     models[comp].raw_probs[i]   = bc.read_probability();
 164                 }
 165             }
 166         }
 167         Ok(())
 168     }
 169     fn decode_coeff_models(&self, bc: &mut BoolCoder, models: &mut VP56Models, is_intra: bool) -> DecoderResult<()> {
 170         let mut def_prob = [128u8; 11];
 171         for plane in 0..2 {
 172             for i in 0..11 {
 173                 if bc.read_prob(HAS_COEF_PROBS[plane][i]) {
 174                     def_prob[i]                 = bc.read_probability();
 175                     models.coeff_models[plane].dc_value_probs[i] = def_prob[i];
 176                 } else if is_intra {
 177                     models.coeff_models[plane].dc_value_probs[i] = def_prob[i];
 178                 }
 179             }
 180         }
 181
 182         if bc.read_bool() {
 183             for i in 1..64 {
 184                 if bc.read_prob(HAS_SCAN_UPD_PROBS[i]) {
 185                     models.vp6models.scan_order[i]  = bc.read_bits(4) as usize;
 186                 }
 187             }
 188             update_scan(&mut models.vp6models);
 189         } else {
 190             reset_scan(&mut models.vp6models, self.interlaced);
 191         }
 192
 193         for comp in 0..2 {
 194             for i in 0..14 {
 195                 if bc.read_prob(HAS_ZERO_RUN_PROBS[comp][i]) {
 196                     models.vp6models.zero_run_probs[comp][i] = bc.read_probability();
 197                 }
 198             }
 199         }
 200
 201         for ctype in 0..3 {
 202             for plane in 0..2 {
 203                 for group in 0..6 {
 204                     for i in 0..11 {
 205                         if bc.read_prob(VP6_AC_PROBS[ctype][plane][group][i]) {
 206                             def_prob[i]         = bc.read_probability();
 207                             models.coeff_models[plane].ac_val_probs[ctype][group][i] = def_prob[i];
 208                         } else if is_intra {
 209                             models.coeff_models[plane].ac_val_probs[ctype][group][i] = def_prob[i];
 210                         }
 211                     }
 212                 }
 213             }
 214         }
 215         for plane in 0..2 {
 216             let mdl = &mut models.coeff_models[plane];
 217             for i in 0..3 {
 218                 for k in 0..5 {
 219                     mdl.dc_token_probs[0][i][k] = rescale_prob(mdl.dc_value_probs[k], &VP6_DC_WEIGHTS[k][i], 255);
 220                 }
 221             }
 222         }
 223         Ok(())
 224     }
 225     fn decode_block(&self, bc: &mut BoolCoder, coeffs: &mut [i16; 64], model: &VP56CoeffModel, vp6model: &VP6Models, fstate: &mut FrameState) -> DecoderResult<()> {
 226         let left_ctx = fstate.coeff_cat[fstate.ctx_idx][0] as usize;
 227         let top_ctx = fstate.top_ctx as usize;
 228         let dc_mode = top_ctx + left_ctx;
 229         let token = decode_token_bc(bc, &model.dc_token_probs[0][dc_mode], model.dc_value_probs[5], true, true);
 230         let val = expand_token_bc(bc, &model.dc_value_probs, token, 6);
 231         coeffs[0] = val;
 232         fstate.last_idx[fstate.ctx_idx] = 0;
 233
 234         let mut idx = 1;
 235         let mut last_val = val;
 236         while idx < 64 {
 237             let ac_band = VP6_IDX_TO_AC_BAND[idx];
 238             let ac_mode = last_val.abs().min(2) as usize;
 239             let has_nnz = (idx == 1) || (last_val != 0);
 240             let token = decode_token_bc(bc, &model.ac_val_probs[ac_mode][ac_band], model.ac_val_probs[ac_mode][ac_band][5], false, has_nnz);
 241             if token == 42 { break; }
 242             let val = expand_token_bc(bc, &model.ac_val_probs[ac_mode][ac_band], token, 6);
 243             coeffs[vp6model.zigzag[idx]] = val.wrapping_mul(fstate.ac_quant);
 244             idx += 1;
 245             last_val = val;
 246             if val == 0 {
 247                 idx += decode_zero_run_bc(bc, &vp6model.zero_run_probs[if idx >= 7 { 1 } else { 0 }]);
 248                 validate!(idx <= 64);
 249             }
 250         }
 251         fstate.coeff_cat[fstate.ctx_idx][0] = if coeffs[0] != 0 { 1 } else { 0 };
 252         fstate.top_ctx = fstate.coeff_cat[fstate.ctx_idx][0];
 253         fstate.last_idx[fstate.ctx_idx] = idx;
 254         Ok(())
 255     }
 256     fn decode_block_huff(&self, br: &mut BitReader, coeffs: &mut [i16; 64], vp6model: &VP6Models, model: &VP6HuffModels, fstate: &mut FrameState) -> DecoderResult<()> {
 257         let plane = if (fstate.plane == 0) || (fstate.plane == 3) { 0 } else { 1 };
 258         let mut last_val;
 259
 260         if fstate.dc_zero_run[plane] == 0 {
 261             let (val, eob) = decode_token_huff(br, &model.dc_token_tree[plane])?;
 262             if eob {
 263                 return Ok(());
 264             }
 265             last_val = val;
 266             coeffs[0] = val;
 267             if val == 0 {
 268                 fstate.dc_zero_run[plane] = decode_eob_run_huff(br)?;
 269             }
 270         } else {
 271             last_val = 0;
 272             fstate.dc_zero_run[plane] -= 1;
 273         }
 274
 275         if fstate.ac_zero_run[plane] > 0 {
 276             fstate.ac_zero_run[plane] -= 1;
 277             fstate.last_idx[fstate.ctx_idx] = 0;
 278             return Ok(());
 279         }
 280
 281         let mut idx = 1;
 282         while idx < 64 {
 283             let ac_band = VP6_IDX_TO_AC_BAND[idx].min(3);
 284             let ac_mode = last_val.abs().min(2) as usize;
 285             let (val, eob) = decode_token_huff(br, &model.ac_token_tree[plane][ac_mode][ac_band])?;
 286             if eob {
 287                 if idx == 1 {
 288                     fstate.ac_zero_run[plane] = decode_eob_run_huff(br)?;
 289                 }
 290                 break;
 291             }
 292             coeffs[vp6model.zigzag[idx]] = val.wrapping_mul(fstate.ac_quant);
 293             idx += 1;
 294             last_val = val;
 295             if val == 0 {
 296                 idx += decode_zero_run_huff(br, &model.zero_run_tree[if idx >= 7 { 1 } else { 0 }])?;
 297                 validate!(idx <= 64);
 298             }
 299         }
 300
 301         fstate.last_idx[fstate.ctx_idx] = idx;
 302
 303         Ok(())
 304     }
 305     fn mc_block(&self, dst: &mut NASimpleVideoFrame<u8>, mut mc_buf: NAVideoBufferRef<u8>, src: NAVideoBufferRef<u8>, plane: usize, x: usize, y: usize, mv: MV, loop_str: i16) {
 306         let is_luma = (plane != 1) && (plane != 2);
 307         let (sx, sy, mx, my, msx, msy) = if is_luma {
 308                 (mv.x >> 2, mv.y >> 2, (mv.x & 3) << 1, (mv.y & 3) << 1, mv.x / 4, mv.y / 4)
 309             } else {
 310                 (mv.x >> 3, mv.y >> 3, mv.x & 7, mv.y & 7, mv.x / 8, mv.y / 8)
 311             };
 312         let tmp_blk = mc_buf.get_data_mut().unwrap();
 313         get_block(tmp_blk, 16, src, plane, x, y, sx, sy);
 314         if (msx & 7) != 0 {
 315             let foff = (8 - (sx & 7)) as usize;
 316             let off = 2 + foff;
 317             vp31_loop_filter(tmp_blk, off, 1, 16, 12, loop_str);
 318         }
 319         if (msy & 7) != 0 {
 320             let foff = (8 - (sy & 7)) as usize;
 321             let off = (2 + foff) * 16;
 322             vp31_loop_filter(tmp_blk, off, 16, 1, 12, loop_str);
 323         }
 324         let copy_mode = (mx == 0) && (my == 0);
 325         let mut bicubic = !copy_mode && is_luma && self.bicubic;
 326         if is_luma && !copy_mode && (self.profile == VP6_ADVANCED_PROFILE) {
 327             if !self.autosel_pm {
 328                 bicubic = true;
 329             } else {
 330                 let mv_limit = 1 << (self.mv_thresh + 1);
 331                 if (mv.x.abs() <= mv_limit) && (mv.y.abs() <= mv_limit) {
 332                     let mut var_off = 16 * 2 + 2;
 333                     if mv.x < 0 { var_off += 1; }
 334                     if mv.y < 0 { var_off += 16; }
 335                     let var = calc_variance(&tmp_blk[var_off..], 16);
 336                     if var >= self.var_thresh {
 337                         bicubic = true;
 338                     }
 339                 }
 340             }
 341         }
 342         let dstride = dst.stride[plane];
 343         let dbuf = &mut dst.data[dst.offset[plane] + x + y * dstride..];
 344         if copy_mode {
 345             let src = &tmp_blk[2 * 16 + 2..];
 346             for (dline, sline) in dbuf.chunks_mut(dst.stride[plane]).zip(src.chunks(16)).take(8) {
 347                 dline[..8].copy_from_slice(&sline[..8]);
 348             }
 349         } else if bicubic {
 350             let coeff_h = &VP6_BICUBIC_COEFFS[self.filter_alpha][mx as usize];
 351             let coeff_v = &VP6_BICUBIC_COEFFS[self.filter_alpha][my as usize];
 352             mc_bicubic(dbuf, dstride, tmp_blk, 16 * 2 + 2, 16, coeff_h, coeff_v);
 353         } else {
 354             mc_bilinear(dbuf, dstride, tmp_blk, 16 * 2 + 2, 16, mx as u16, my as u16);
 355         }
 356     }
 357 }
 358
 359 fn update_scan(model: &mut VP6Models) {
 360     let mut idx = 1;
 361     for band in 0..16 {
 362         for i in 1..64 {
 363             if model.scan_order[i] == band {
 364                 model.scan[idx] = i;
 365                 idx += 1;
 366             }
 367         }
 368     }
 369     for i in 1..64 {
 370         model.zigzag[i] = ZIGZAG[model.scan[i]];
 371     }
 372 }
 373
 374 fn reset_scan(model: &mut VP6Models, interlaced: bool) {
 375     if !interlaced {
 376         model.scan_order.copy_from_slice(&VP6_DEFAULT_SCAN_ORDER);
 377     } else {
 378         model.scan_order.copy_from_slice(&VP6_INTERLACED_SCAN_ORDER);
 379     }
 380     for i in 0..64 { model.scan[i] = i; }
 381     model.zigzag.copy_from_slice(&ZIGZAG);
 382 }
 383
 384 fn decode_token_bc(bc: &mut BoolCoder, probs: &[u8], prob34: u8, is_dc: bool, has_nnz: bool) -> u8 {
 385     if has_nnz && !bc.read_prob(probs[0]) {
 386         if is_dc || bc.read_prob(probs[1]) {
 387             0
 388         } else {
 389             TOKEN_EOB
 390         }
 391     } else {
 392         vp_tree!(bc, probs[2],
 393                  1,
 394                  vp_tree!(bc, probs[3],
 395                           vp_tree!(bc, probs[4],
 396                                    2,
 397                                    vp_tree!(bc, prob34, 3, 4)),
 398                           TOKEN_LARGE))
 399     }
 400 }
 401
 402 fn decode_zero_run_bc(bc: &mut BoolCoder, probs: &[u8; 14]) -> usize {
 403     let val = vp_tree!(bc, probs[0],
 404                     vp_tree!(bc, probs[1],
 405                         vp_tree!(bc, probs[2], 0, 1),
 406                         vp_tree!(bc, probs[3], 2, 3)),
 407                     vp_tree!(bc, probs[4],
 408                         vp_tree!(bc, probs[5],
 409                             vp_tree!(bc, probs[6], 4, 5),
 410                             vp_tree!(bc, probs[7], 6, 7)),
 411                         42));
 412     if val != 42 {
 413         val
 414     } else {
 415         let mut nval = 8;
 416         for i in 0..6 {
 417             nval                                += (bc.read_prob(probs[i + 8]) as usize) << i;
 418         }
 419         nval
 420     }
 421 }
 422
 423 fn decode_token_huff(br: &mut BitReader, huff: &VP6Huff) -> DecoderResult<(i16, bool)> {
 424     let tok                                     = br.read_huff(huff)?;
 425     match tok {
 426         0   => Ok((0, false)),
 427         1 | 2 | 3 | 4 => {
 428             if !br.read_bool()? {
 429                 Ok((i16::from(tok), false))
 430             } else {
 431                 Ok((-i16::from(tok), false))
 432             }
 433         },
 434         5 | 6 | 7 | 8 | 9 | 10 => {
 435             let base = (tok - 5) as usize;
 436             let add_bits                        = br.read(VP6_COEF_ADD_BITS[base])? as i16;
 437             let val = VP56_COEF_BASE[base] + add_bits;
 438             if !br.read_bool()? {
 439                 Ok((val, false))
 440             } else {
 441                 Ok((-val, false))
 442             }
 443         },
 444         _   => Ok((0, true)),
 445     }
 446 }
 447
 448 fn decode_eob_run_huff(br: &mut BitReader) -> DecoderResult<usize> {
 449     let val                                     = br.read(2)?;
 450     match val {
 451         0 => Ok(0),
 452         1 => Ok(1),
 453         2 => {
 454             let val                             = br.read(2)?;
 455             Ok((val as usize) + 2)
 456         },
 457         _ => {
 458             if br.read_bool()? {
 459                 Ok((br.read(6)? as usize) + 10)
 460             } else {
 461                 Ok((br.read(2)? as usize) + 6)
 462             }
 463         },
 464     }
 465 }
 466
 467 fn decode_zero_run_huff(br: &mut BitReader, huff: &VP6Huff) -> DecoderResult<usize> {
 468     let val                                     = br.read_huff(huff)?;
 469     if val < 8 {
 470         Ok(val as usize)
 471     } else {
 472         Ok((br.read(6)? as usize) + 8)
 473     }
 474 }
 475
 476 #[allow(clippy::too_many_arguments)]
 477 fn get_block(dst: &mut [u8], dstride: usize, src: NAVideoBufferRef<u8>, comp: usize,
 478              dx: usize, dy: usize, mv_x: i16, mv_y: i16)
 479 {
 480     let (w, h) = src.get_dimensions(comp);
 481     let sx = (dx as isize) + (mv_x as isize);
 482     let sy = (dy as isize) + (mv_y as isize);
 483
 484     if (sx - 2 < 0) || (sx + 8 + 2 > (w as isize)) ||
 485        (sy - 2 < 0) || (sy + 8 + 2 > (h as isize)) {
 486         edge_emu(&src, sx - 2, sy - 2, 8 + 2 + 2, 8 + 2 + 2,
 487                  dst, dstride, comp, 0);
 488     } else {
 489         let sstride = src.get_stride(comp);
 490         let soff    = src.get_offset(comp);
 491         let sdta    = src.get_data();
 492         let sbuf: &[u8] = sdta.as_slice();
 493         let saddr = soff + ((sx - 2) as usize) + ((sy - 2) as usize) * sstride;
 494         let src = &sbuf[saddr..];
 495         for (dline, sline) in dst.chunks_mut(dstride).zip(src.chunks(sstride)).take(12) {
 496             dline[..12].copy_from_slice(&sline[..12]);
 497         }
 498     }
 499 }
 500
 501 fn calc_variance(src: &[u8], stride: usize) -> u16 {
 502     let mut sum = 0;
 503     let mut ssum = 0;
 504     for line in src.chunks(stride * 2).take(4) {
 505         for el in line.iter().take(8).step_by(2) {
 506             let pix = u32::from(*el);
 507             sum += pix;
 508             ssum += pix * pix;
 509         }
 510     }
 511     ((ssum * 16 - sum * sum) >> 8) as u16
 512 }
 513
 514 macro_rules! mc_filter {
 515     (bilinear; $a: expr, $b: expr, $c: expr) => {
 516         ((u16::from($a) * (8 - $c) + u16::from($b) * $c + 4) >> 3) as u8
 517     };
 518     (bicubic; $src: expr, $off: expr, $step: expr, $coeffs: expr) => {
 519         ((i32::from($src[$off - $step]    ) * i32::from($coeffs[0]) +
 520           i32::from($src[$off]            ) * i32::from($coeffs[1]) +
 521           i32::from($src[$off + $step]    ) * i32::from($coeffs[2]) +
 522           i32::from($src[$off + $step * 2]) * i32::from($coeffs[3]) + 64) >> 7).min(255).max(0) as u8
 523     }
 524 }
 525
 526 //#[allow(snake_case)]
 527 fn mc_bilinear(dst: &mut [u8], dstride: usize, src: &[u8], mut soff: usize, sstride: usize, mx: u16, my: u16) {
 528     if my == 0 {
 529         for dline in dst.chunks_mut(dstride).take(8) {
 530             for i in 0..8 {
 531                 dline[i] = mc_filter!(bilinear; src[soff + i], src[soff + i + 1], mx);
 532             }
 533             soff += sstride;
 534         }
 535     } else if mx == 0 {
 536         for dline in dst.chunks_mut(dstride).take(8) {
 537             for i in 0..8 {
 538                 dline[i] = mc_filter!(bilinear; src[soff + i], src[soff + i + sstride], my);
 539             }
 540             soff += sstride;
 541         }
 542     } else {
 543         let mut tmp = [0u8; 8];
 544         for i in 0..8 {
 545             tmp[i] = mc_filter!(bilinear; src[soff + i], src[soff + i + 1], mx);
 546         }
 547         soff += sstride;
 548         for dline in dst.chunks_mut(dstride).take(8) {
 549             for i in 0..8 {
 550                 let cur = mc_filter!(bilinear; src[soff + i], src[soff + i + 1], mx);
 551                 dline[i] = mc_filter!(bilinear; tmp[i], cur, my);
 552                 tmp[i] = cur;
 553             }
 554             soff += sstride;
 555         }
 556     }
 557 }
 558
 559 #[allow(clippy::trivially_copy_pass_by_ref)]
 560 fn mc_bicubic(dst: &mut [u8], dstride: usize, src: &[u8], mut soff: usize, sstride: usize, coeffs_w: &[i16; 4], coeffs_h: &[i16; 4]) {
 561     if coeffs_h[1] == 128 {
 562         for dline in dst.chunks_mut(dstride).take(8) {
 563             for i in 0..8 {
 564                 dline[i] = mc_filter!(bicubic; src, soff + i, 1, coeffs_w);
 565             }
 566             soff += sstride;
 567         }
 568     } else if coeffs_w[1] == 128 { // horizontal-only interpolation
 569         for dline in dst.chunks_mut(dstride).take(8) {
 570             for i in 0..8 {
 571                 dline[i] = mc_filter!(bicubic; src, soff + i, sstride, coeffs_h);
 572             }
 573             soff += sstride;
 574         }
 575     } else {
 576         let mut buf = [0u8; 16 * 11];
 577         soff -= sstride;
 578         for dline in buf.chunks_mut(16) {
 579             for i in 0..8 {
 580                 dline[i] = mc_filter!(bicubic; src, soff + i, 1, coeffs_w);
 581             }
 582             soff += sstride;
 583         }
 584         let mut soff = 16;
 585         for dline in dst.chunks_mut(dstride).take(8) {
 586             for i in 0..8 {
 587                 dline[i] = mc_filter!(bicubic; buf, soff + i, 16, coeffs_h);
 588             }
 589             soff += 16;
 590         }
 591     }
 592 }
 593
 594 struct VP6Decoder {
 595     dec:        VP56Decoder,
 596     info:       NACodecInfoRef,
 597     br:         VP6BR,
 598     has_alpha:  bool,
 599 }
 600
 601 impl VP6Decoder {
 602     fn new(has_alpha: bool) -> Self {
 603         Self {
 604             dec:        VP56Decoder::new(6, has_alpha, true),
 605             info:       NACodecInfoRef::default(),
 606             br:         VP6BR::new(),
 607             has_alpha,
 608         }
 609     }
 610 }
 611
 612 impl NADecoder for VP6Decoder {
 613     fn init(&mut self, supp: &mut NADecoderSupport, info: NACodecInfoRef) -> DecoderResult<()> {
 614         if let NACodecTypeInfo::Video(vinfo) = info.get_properties() {
 615             let fmt = if !self.has_alpha {
 616                     YUV420_FORMAT
 617                 } else {
 618                     VP_YUVA420_FORMAT
 619                 };
 620             let myvinfo = NAVideoInfo::new(vinfo.get_width(), vinfo.get_height(), false, fmt);
 621             let myinfo = NACodecTypeInfo::Video(myvinfo);
 622             self.info = NACodecInfo::new_ref(info.get_name(), myinfo, info.get_extradata()).into_ref();
 623             self.dec.init(supp, myvinfo)?;
 624             Ok(())
 625         } else {
 626             Err(DecoderError::InvalidData)
 627         }
 628     }
 629     fn decode(&mut self, supp: &mut NADecoderSupport, pkt: &NAPacket) -> DecoderResult<NAFrameRef> {
 630         let src = pkt.get_buffer();
 631
 632         let (bufinfo, ftype) = self.dec.decode_frame(supp, src.as_slice(), &mut self.br)?;
 633
 634         let mut frm = NAFrame::new_from_pkt(pkt, self.info.clone(), bufinfo);
 635         frm.set_keyframe(ftype == FrameType::I);
 636         frm.set_frame_type(ftype);
 637         Ok(frm.into_ref())
 638     }
 639     fn flush(&mut self) {
 640         self.dec.flush();
 641     }
 642 }
 643
 644 impl NAOptionHandler for VP6Decoder {
 645     fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] }
 646     fn set_options(&mut self, _options: &[NAOption]) { }
 647     fn query_option_value(&self, _name: &str) -> Option<NAValue> { None }
 648 }
 649
 650 pub fn get_decoder_vp6() -> Box<dyn NADecoder + Send> {
 651     Box::new(VP6Decoder::new(false))
 652 }
 653
 654 pub fn get_decoder_vp6_alpha() -> Box<dyn NADecoder + Send> {
 655     Box::new(VP6Decoder::new(true))
 656 }
 657
 658 #[cfg(test)]
 659 mod test {
 660     use nihav_core::codecs::RegisteredDecoders;
 661     use nihav_core::demuxers::RegisteredDemuxers;
 662     use nihav_codec_support::test::dec_video::*;
 663     use crate::duck_register_all_decoders;
 664     use nihav_commonfmt::generic_register_all_demuxers;
 665
 666     #[test]
 667     fn test_vp6() {
 668         let mut dmx_reg = RegisteredDemuxers::new();
 669         generic_register_all_demuxers(&mut dmx_reg);
 670         let mut dec_reg = RegisteredDecoders::new();
 671         duck_register_all_decoders(&mut dec_reg);
 672
 673         test_decoding("avi", "vp6", "assets/Duck/selection_720x576_300kBit_vp60i.avi", Some(16),
 674                       &dmx_reg, &dec_reg,
 675                       ExpectedTestResult::MD5([0x042c3e96, 0x8a9b26a2, 0x4dcbaf66, 0x1b788d03]));
 676     }
 677     #[test]
 678     fn test_vp6_huff() {
 679         let mut dmx_reg = RegisteredDemuxers::new();
 680         generic_register_all_demuxers(&mut dmx_reg);
 681         let mut dec_reg = RegisteredDecoders::new();
 682         duck_register_all_decoders(&mut dec_reg);
 683
 684         test_decoding("avi", "vp6", "assets/Duck/vp6_crash.avi", Some(4),
 685                       &dmx_reg, &dec_reg, ExpectedTestResult::MD5Frames(vec![
 686                             [0xdcd70fa0, 0x0d075ce2, 0xc9e65077, 0xb003a92e],
 687                             [0x334abf96, 0x3a004c7a, 0x5781cd5c, 0x25c3ae5c],
 688                             [0x6164b851, 0x528cd8de, 0xecab7328, 0x4b49708a],
 689                             [0x11b048ac, 0xedb3e471, 0xd04e9399, 0x64e623e3],
 690                             [0x182871b1, 0x2146893a, 0x2912210e, 0x6dd592e8]]));
 691     }
 692     #[test]
 693     fn test_vp6_alpha() {
 694         let mut dmx_reg = RegisteredDemuxers::new();
 695         generic_register_all_demuxers(&mut dmx_reg);
 696         let mut dec_reg = RegisteredDecoders::new();
 697         duck_register_all_decoders(&mut dec_reg);
 698
 699         test_decoding("avi", "vp6a", "assets/Duck/vp6a.avi", Some(25),
 700                       &dmx_reg, &dec_reg, ExpectedTestResult::MD5Frames(vec![
 701                             [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
 702                             [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
 703                             [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
 704                             [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
 705                             [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
 706                             [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
 707                             [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
 708                             [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
 709                             [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
 710                             [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
 711                             [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
 712                             [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
 713                             [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
 714                             [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
 715                             [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
 716                             [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
 717                             [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
 718                             [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
 719                             [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
 720                             [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
 721                             [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
 722                             [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
 723                             [0xd3801a96, 0x1b5384ff, 0x422b228c, 0x9c4582c4],
 724                             [0x8ddb0dfe, 0x302eb1ed, 0x10e64e22, 0x5a5a62b9],
 725                             [0x79338328, 0x06113781, 0x8b116d18, 0xde56707e],
 726                             [0x671fa1b3, 0x0b3e41e9, 0xeb3a494c, 0xcdd24b1b]]));
 727     }
 728 }