aac: clear M/S flags
[nihav.git] / nihav-duck / src / codecs / vp6.rs
index 803ce5ee417ead96dfddfb7253187c03eba9ac34..916711cddbc00b5392b75c4f1c7f6b28ba257f7e 100644 (file)
@@ -1,15 +1,10 @@
 use nihav_core::codecs::*;
 use nihav_core::io::bitreader::*;
-use nihav_core::codecs::blockdsp::edge_emu;
+use nihav_codec_support::codecs::{MV, ZIGZAG};
 use super::vpcommon::*;
 use super::vp56::*;
-
-const VERSION_VP60: u8 = 6;
-//const VERSION_VP61: u8 = 7;
-const VERSION_VP62: u8 = 8;
-
-const VP6_SIMPLE_PROFILE: u8    = 0;
-const VP6_ADVANCED_PROFILE: u8  = 3;
+use super::vp6data::*;
+use super::vp6dsp::*;
 
 #[derive(Default)]
 struct VP6BR {
@@ -36,7 +31,7 @@ impl VP56Parser for VP6BR {
         let mut hdr = VP56Header::default();
 // horrible hack to match VP6 header parsing
         let src = bc.src;
-        let mut br = BitReader::new(src, src.len(), BitReaderMode::BE);
+        let mut br = BitReader::new(src, BitReaderMode::BE);
 
         hdr.is_intra                            = !br.read_bool()?;
         hdr.is_golden = hdr.is_intra;
@@ -51,13 +46,14 @@ impl VP56Parser for VP6BR {
         } else {
             hdr.version = self.vpversion;
             hdr.profile = self.profile;
+            hdr.interlaced = self.interlaced;
         }
         if hdr.multistream || (hdr.profile == VP6_SIMPLE_PROFILE) {
             hdr.offset                          = br.read(16)? as u16;
             validate!(hdr.offset > if hdr.is_intra { 6 } else { 2 });
+            hdr.multistream = true;
         }
         let bytes = br.tell() >> 3;
-        std::mem::drop(br);
         bc.skip_bytes(bytes);
         self.loop_mode = 0;
         if hdr.is_intra {
@@ -108,8 +104,6 @@ impl VP56Parser for VP6BR {
         Ok(hdr)
     }
     fn decode_mv(&self, bc: &mut BoolCoder, model: &VP56MVModel) -> i16 {
-        const LONG_VECTOR_ORDER: [usize; 7] = [ 0, 1, 2, 7, 6, 5, 4 ];
-
         let val = if !bc.read_prob(model.nz_prob) { // short vector
                 vp_tree!(bc, model.tree_probs[0],
                          vp_tree!(bc, model.tree_probs[1],
@@ -137,20 +131,6 @@ impl VP56Parser for VP6BR {
         }
     }
     fn reset_models(&self, models: &mut VP56Models) {
-        const NZ_PROBS: [u8; 2] = [ 162, 164 ];
-        const RAW_PROBS: [[u8; 8]; 2] = [
-            [ 247, 210, 135, 68, 138, 220, 239, 246 ],
-            [ 244, 184, 201, 44, 173, 221, 239, 253 ]
-        ];
-        const TREE_PROBS: [[u8; 7]; 2] = [
-            [ 225, 146, 172, 147, 214,  39, 156 ],
-            [ 204, 170, 119, 235, 140, 230, 228 ]
-        ];
-        const ZERO_RUN_PROBS: [[u8; 14]; 2] = [
-            [ 198, 197, 196, 146, 198, 204, 169, 142, 130, 136, 149, 149, 191, 249 ],
-            [ 135, 201, 181, 154,  98, 117, 132, 126, 146, 169, 184, 240, 246, 254 ]
-        ];
-
         for (i, mdl) in models.mv_models.iter_mut().enumerate() {
             mdl.nz_prob         = NZ_PROBS[i];
             mdl.sign_prob       = 128;
@@ -161,17 +141,6 @@ impl VP56Parser for VP6BR {
         reset_scan(&mut models.vp6models, self.interlaced);
     }
     fn decode_mv_models(&self, bc: &mut BoolCoder, models: &mut [VP56MVModel; 2]) -> DecoderResult<()> {
-        const HAS_NZ_PROB: [u8; 2] = [ 237, 231 ];
-        const HAS_SIGN_PROB: [u8; 2] = [ 246, 243 ];
-        const HAS_TREE_PROB: [[u8; 7]; 2] = [
-            [ 253, 253, 254, 254, 254, 254, 254 ],
-            [ 245, 253, 254, 254, 254, 254, 254 ]
-        ];
-        const HAS_RAW_PROB: [[u8; 8]; 2] = [
-            [ 254, 254, 254, 254, 254, 250, 250, 252 ],
-            [ 254, 254, 254, 254, 254, 251, 251, 254 ]
-        ];
-
         for comp in 0..2 {
             if bc.read_prob(HAS_NZ_PROB[comp]) {
                 models[comp].nz_prob            = bc.read_probability();
@@ -197,29 +166,10 @@ impl VP56Parser for VP6BR {
         Ok(())
     }
     fn decode_coeff_models(&self, bc: &mut BoolCoder, models: &mut VP56Models, is_intra: bool) -> DecoderResult<()> {
-        const COEF_PROBS: [[u8; 11]; 2] = [
-            [ 146, 255, 181, 207, 232, 243, 238, 251, 244, 250, 249 ],
-            [ 179, 255, 214, 240, 250, 255, 244, 255, 255, 255, 255 ]
-        ];
-        const SCAN_UPD_PROBS: [u8; 64] = [
-              0, 132, 132, 159, 153, 151, 161, 170,
-            164, 162, 136, 110, 103, 114, 129, 118,
-            124, 125, 132, 136, 114, 110, 142, 135,
-            134, 123, 143, 126, 153, 183, 166, 161,
-            171, 180, 179, 164, 203, 218, 225, 217,
-            215, 206, 203, 217, 229, 241, 248, 243,
-            253, 255, 253, 255, 255, 255, 255, 255,
-            255, 255, 255, 255, 255, 255, 255, 255
-        ];
-        const ZERO_RUN_PROBS: [[u8; 14]; 2] = [
-            [ 219, 246, 238, 249, 232, 239, 249, 255, 248, 253, 239, 244, 241, 248 ],
-            [ 198, 232, 251, 253, 219, 241, 253, 255, 248, 249, 244, 238, 251, 255 ]
-        ];
-
         let mut def_prob = [128u8; 11];
         for plane in 0..2 {
             for i in 0..11 {
-                if bc.read_prob(COEF_PROBS[plane][i]) {
+                if bc.read_prob(HAS_COEF_PROBS[plane][i]) {
                     def_prob[i]                 = bc.read_probability();
                     models.coeff_models[plane].dc_value_probs[i] = def_prob[i];
                 } else if is_intra {
@@ -230,7 +180,7 @@ impl VP56Parser for VP6BR {
 
         if bc.read_bool() {
             for i in 1..64 {
-                if bc.read_prob(SCAN_UPD_PROBS[i]) {
+                if bc.read_prob(HAS_SCAN_UPD_PROBS[i]) {
                     models.vp6models.scan_order[i]  = bc.read_bits(4) as usize;
                 }
             }
@@ -241,7 +191,7 @@ impl VP56Parser for VP6BR {
 
         for comp in 0..2 {
             for i in 0..14 {
-                if bc.read_prob(ZERO_RUN_PROBS[comp][i]) {
+                if bc.read_prob(HAS_ZERO_RUN_PROBS[comp][i]) {
                     models.vp6models.zero_run_probs[comp][i] = bc.read_probability();
                 }
             }
@@ -351,38 +301,51 @@ impl VP56Parser for VP6BR {
 
         Ok(())
     }
-    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) {
+    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) {
         let is_luma = (plane != 1) && (plane != 2);
-        let (sx, sy, mx, my) = if is_luma {
-                (mv.x / 4, mv.y / 4, mv.x & 3, mv.y & 3)
+        let (sx, sy, mx, my, msx, msy) = if is_luma {
+                (mv.x >> 2, mv.y >> 2, (mv.x & 3) << 1, (mv.y & 3) << 1, mv.x / 4, mv.y / 4)
             } else {
-                (mv.x / 8, mv.y / 8, mv.x & 7, mv.y & 7)
+                (mv.x >> 3, mv.y >> 3, mv.x & 7, mv.y & 7, mv.x / 8, mv.y / 8)
             };
         let tmp_blk = mc_buf.get_data_mut().unwrap();
-        get_block(tmp_blk, 16, src.clone(), plane, x, y, sx, sy);
-        // todo filtering
-        let mut bicubic = self.bicubic;
-        if is_luma && (self.profile == VP6_ADVANCED_PROFILE) {
+        get_block(tmp_blk, 16, src, plane, x, y, sx, sy);
+        if (msx & 7) != 0 {
+            let foff = (8 - (sx & 7)) as usize;
+            let off = 2 + foff;
+            vp31_loop_filter(tmp_blk, off, 1, 16, 12, loop_str);
+        }
+        if (msy & 7) != 0 {
+            let foff = (8 - (sy & 7)) as usize;
+            let off = (2 + foff) * 16;
+            vp31_loop_filter(tmp_blk, off, 16, 1, 12, loop_str);
+        }
+        let copy_mode = (mx == 0) && (my == 0);
+        let mut bicubic = !copy_mode && is_luma && self.bicubic;
+        if is_luma && !copy_mode && (self.profile == VP6_ADVANCED_PROFILE) {
             if !self.autosel_pm {
                 bicubic = true;
             } else {
                 let mv_limit = 1 << (self.mv_thresh + 1);
                 if (mv.x.abs() <= mv_limit) && (mv.y.abs() <= mv_limit) {
-                    let var = calc_variance(&tmp_blk[16 * 2 + 2..], 16);
-                    if var > self.var_thresh {
-                        bicubic = false;
+                    let mut var_off = 16 * 2 + 2;
+                    if mv.x < 0 { var_off += 1; }
+                    if mv.y < 0 { var_off += 16; }
+                    let var = calc_variance(&tmp_blk[var_off..], 16);
+                    if var >= self.var_thresh {
+                        bicubic = true;
                     }
                 }
             }
         }
         let dstride = dst.stride[plane];
         let dbuf = &mut dst.data[dst.offset[plane] + x + y * dstride..];
-        if mx == 0 && my == 0 {
+        if copy_mode {
             let src = &tmp_blk[2 * 16 + 2..];
             for (dline, sline) in dbuf.chunks_mut(dst.stride[plane]).zip(src.chunks(16)).take(8) {
-                for i in 0..8 { dline[i] = sline[i]; }
+                dline[..8].copy_from_slice(&sline[..8]);
             }
-        } else if is_luma && bicubic {
+        } else if bicubic {
             let coeff_h = &VP6_BICUBIC_COEFFS[self.filter_alpha][mx as usize];
             let coeff_v = &VP6_BICUBIC_COEFFS[self.filter_alpha][my as usize];
             mc_bicubic(dbuf, dstride, tmp_blk, 16 * 2 + 2, 16, coeff_h, coeff_v);
@@ -408,27 +371,6 @@ fn update_scan(model: &mut VP6Models) {
 }
 
 fn reset_scan(model: &mut VP6Models, interlaced: bool) {
-    const VP6_DEFAULT_SCAN_ORDER: [usize; 64] = [
-         0,  0,  1,  1,  1,  2,  2,  2,
-         2,  2,  2,  3,  3,  4,  4,  4,
-         5,  5,  5,  5,  6,  6,  7,  7,
-         7,  7,  7,  8,  8,  9,  9,  9,
-         9,  9,  9, 10, 10, 11, 11, 11,
-        11, 11, 11, 12, 12, 12, 12, 12,
-        12, 13, 13, 13, 13, 13, 14, 14,
-        14, 14, 15, 15, 15, 15, 15, 15
-    ];
-    const VP6_INTERLACED_SCAN_ORDER: [usize; 64] = [
-         0,  1,  0,  1,  1,  2,  5,  3,
-         2,  2,  2,  2,  4,  7,  8, 10,
-         9,  7,  5,  4,  2,  3,  5,  6,
-         8,  9, 11, 12, 13, 12, 11, 10,
-         9,  7,  5,  4,  6,  7,  9, 11,
-        12, 12, 13, 13, 14, 12, 11,  9,
-         7,  9, 11, 12, 14, 14, 14, 15,
-        13, 11, 13, 15, 15, 15, 15, 15
-    ];
-
     if !interlaced {
         model.scan_order.copy_from_slice(&VP6_DEFAULT_SCAN_ORDER);
     } else {
@@ -478,20 +420,19 @@ fn decode_zero_run_bc(bc: &mut BoolCoder, probs: &[u8; 14]) -> usize {
 }
 
 fn decode_token_huff(br: &mut BitReader, huff: &VP6Huff) -> DecoderResult<(i16, bool)> {
-    const COEF_ADD_BITS: [u8; 6] = [ 1, 2, 3, 4, 5, 11 ];
     let tok                                     = br.read_huff(huff)?;
     match tok {
         0   => Ok((0, false)),
         1 | 2 | 3 | 4 => {
             if !br.read_bool()? {
-                Ok((tok as i16, false))
+                Ok((i16::from(tok), false))
             } else {
-                Ok((-(tok as i16), false))
+                Ok((-i16::from(tok), false))
             }
         },
         5 | 6 | 7 | 8 | 9 | 10 => {
             let base = (tok - 5) as usize;
-            let add_bits                        = br.read(COEF_ADD_BITS[base])? as i16;
+            let add_bits                        = br.read(VP6_COEF_ADD_BITS[base])? as i16;
             let val = VP56_COEF_BASE[base] + add_bits;
             if !br.read_bool()? {
                 Ok((val, false))
@@ -531,139 +472,22 @@ fn decode_zero_run_huff(br: &mut BitReader, huff: &VP6Huff) -> DecoderResult<usi
     }
 }
 
-
-fn get_block(dst: &mut [u8], dstride: usize, src: NAVideoBufferRef<u8>, comp: usize,
-             dx: usize, dy: usize, mv_x: i16, mv_y: i16)
-{
-    let (w, h) = src.get_dimensions(comp);
-    let sx = (dx as isize) + (mv_x as isize);
-    let sy = (dy as isize) + (mv_y as isize);
-
-    if (sx - 2 < 0) || (sx + 8 + 2 > (w as isize)) ||
-       (sy - 2 < 0) || (sy + 8 + 2 > (h as isize)) {
-        edge_emu(&src, sx - 2, sy - 2, 8 + 2 + 2, 8 + 2 + 2,
-                 dst, dstride, comp);
-    } else {
-        let sstride = src.get_stride(comp);
-        let soff    = src.get_offset(comp);
-        let sdta    = src.get_data();
-        let sbuf: &[u8] = sdta.as_slice();
-        let saddr = soff + ((sx - 2) as usize) + ((sy - 2) as usize) * sstride;
-        let src = &sbuf[saddr..];
-        for (dline, sline) in dst.chunks_mut(dstride).zip(src.chunks(sstride)).take(12) {
-            for i in 0..12 {
-                dline[i] = sline[i];
-            }
-        }
-    }
-}
-
-fn calc_variance(src: &[u8], stride: usize) -> u16 {
-    let mut sum = 0;
-    let mut ssum = 0;
-    for line in src.chunks(stride * 2).take(8) {
-        for el in line.iter().take(8).step_by(2) {
-            let pix = *el as u32;
-            sum += pix;
-            ssum += pix * pix;
-        }
-    }
-    ((ssum * 16 - sum * sum) >> 8) as u16
-}
-
-macro_rules! mc_filter {
-    (bilinear; $a: expr, $b: expr, $c: expr) => {
-        ((($a as u16) * (8 - $c) + ($b as u16) * $c + 3) >> 3) as u8
-    };
-    (bicubic; $src: expr, $off: expr, $step: expr, $coeffs: expr) => {
-        ((($src[$off - $step]     as i32) * ($coeffs[0] as i32) +
-          ($src[$off]             as i32) * ($coeffs[1] as i32) +
-          ($src[$off + $step]     as i32) * ($coeffs[2] as i32) +
-          ($src[$off + $step * 2] as i32) * ($coeffs[3] as i32) + 64) >> 7).min(255).max(0) as u8
-    }
-}
-
-//#[allow(snake_case)]
-fn mc_bilinear(dst: &mut [u8], dstride: usize, src: &[u8], mut soff: usize, sstride: usize, mx: u16, my: u16) {
-    if my == 0 {
-        for dline in dst.chunks_mut(dstride).take(8) {
-            for i in 0..8 {
-                dline[i] = mc_filter!(bilinear; src[soff + i], src[soff + i + 1], mx);
-            }
-            soff += sstride;
-        }
-    } else if mx == 0 {
-        for dline in dst.chunks_mut(dstride).take(8) {
-            for i in 0..8 {
-                dline[i] = mc_filter!(bilinear; src[soff + i], src[soff + i + sstride], my);
-            }
-            soff += sstride;
-        }
-    } else {
-        let mut tmp = [0u8; 8];
-        for i in 0..8 {
-            tmp[i] = mc_filter!(bilinear; src[soff + i], src[soff + i + 1], mx);
-        }
-        soff += sstride;
-        for dline in dst.chunks_mut(dstride).take(8) {
-            for i in 0..8 {
-                let cur = mc_filter!(bilinear; src[soff + i], src[soff + i + 1], mx);
-                dline[i] = mc_filter!(bilinear; tmp[i], cur, my);
-                tmp[i] = cur;
-            }
-            soff += sstride;
-        }
-    }
-}
-
-fn mc_bicubic(dst: &mut [u8], dstride: usize, src: &[u8], mut soff: usize, sstride: usize, coeffs_w: &[i16; 4], coeffs_h: &[i16; 4]) {
-    if coeffs_h[1] == 128 {
-        for dline in dst.chunks_mut(dstride).take(8) {
-            for i in 0..8 {
-                dline[i] = mc_filter!(bicubic; src, soff + i, 1, coeffs_w);
-            }
-            soff += sstride;
-        }
-    } else if coeffs_w[1] == 128 { // horizontal-only interpolation
-        for dline in dst.chunks_mut(dstride).take(8) {
-            for i in 0..8 {
-                dline[i] = mc_filter!(bicubic; src, soff + i, sstride, coeffs_h);
-            }
-            soff += sstride;
-        }
-    } else {
-        let mut buf = [0u8; 16 * 11];
-        soff -= sstride;
-        for dline in buf.chunks_mut(16) {
-            for i in 0..8 {
-                dline[i] = mc_filter!(bicubic; src, soff + i, 1, coeffs_w);
-            }
-            soff += sstride;
-        }
-        let mut soff = 16;
-        for dline in dst.chunks_mut(dstride).take(8) {
-            for i in 0..8 {
-                dline[i] = mc_filter!(bicubic; buf, soff + i, 16, coeffs_h);
-            }
-            soff += 16;
-        }
-    }
-}
-
 struct VP6Decoder {
     dec:        VP56Decoder,
     info:       NACodecInfoRef,
     br:         VP6BR,
     has_alpha:  bool,
+    flipped:    bool,
 }
 
 impl VP6Decoder {
-    fn new(has_alpha: bool) -> Self {
+    fn new(flipped: bool, has_alpha: bool) -> Self {
         Self {
-            dec:        VP56Decoder::new(6, has_alpha, true),
+            dec:        VP56Decoder::new(6, has_alpha, flipped),
             info:       NACodecInfoRef::default(),
             br:         VP6BR::new(),
             has_alpha,
+            flipped,
         }
     }
 }
@@ -674,16 +498,18 @@ impl NADecoder for VP6Decoder {
             let fmt = if !self.has_alpha {
                     YUV420_FORMAT
                 } else {
-                    NAPixelFormaton::new(ColorModel::YUV(YUVSubmodel::YUVJ),
-                                         Some(NAPixelChromaton::new(0, 0, false, 8, 0, 0, 1)),
-                                         Some(NAPixelChromaton::new(1, 1, false, 8, 0, 1, 1)),
-                                         Some(NAPixelChromaton::new(1, 1, false, 8, 0, 2, 1)),
-                                         Some(NAPixelChromaton::new(0, 0, false, 8, 0, 3, 1)),
-                                         None,
-                                         0, 4)
+                    VP_YUVA420_FORMAT
                 };
-            let myvinfo = NAVideoInfo::new(vinfo.get_width(), vinfo.get_height(), false, fmt);
-            let myinfo = NACodecTypeInfo::Video(myvinfo.clone());
+            let mut width  = vinfo.get_width();
+            let mut height = vinfo.get_height();
+            if let (false, Some(edta)) = (self.flipped, info.get_extradata()) {
+                if (width & 0xF) == 0 && (height & 0xF) == 0 {
+                    width  -= usize::from(edta[0] >> 4);
+                    height -= usize::from(edta[0] & 0xF);
+                }
+            }
+            let myvinfo = NAVideoInfo::new(width, height, self.flipped, fmt);
+            let myinfo = NACodecTypeInfo::Video(myvinfo);
             self.info = NACodecInfo::new_ref(info.get_name(), myinfo, info.get_extradata()).into_ref();
             self.dec.init(supp, myvinfo)?;
             Ok(())
@@ -706,265 +532,95 @@ impl NADecoder for VP6Decoder {
     }
 }
 
-pub fn get_decoder_vp6() -> Box<NADecoder> {
-    Box::new(VP6Decoder::new(false))
+impl NAOptionHandler for VP6Decoder {
+    fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] }
+    fn set_options(&mut self, _options: &[NAOption]) { }
+    fn query_option_value(&self, _name: &str) -> Option<NAValue> { None }
 }
 
-pub fn get_decoder_vp6_alpha() -> Box<NADecoder> {
-    Box::new(VP6Decoder::new(true))
+pub fn get_decoder_vp6() -> Box<dyn NADecoder + Send> {
+    Box::new(VP6Decoder::new(true, false))
+}
+
+pub fn get_decoder_vp6f() -> Box<dyn NADecoder + Send> {
+    Box::new(VP6Decoder::new(false, false))
+}
+
+pub fn get_decoder_vp6_alpha() -> Box<dyn NADecoder + Send> {
+    Box::new(VP6Decoder::new(false, true))
 }
 
 #[cfg(test)]
 mod test {
     use nihav_core::codecs::RegisteredDecoders;
     use nihav_core::demuxers::RegisteredDemuxers;
-    use nihav_core::test::dec_video::*;
-    use crate::codecs::duck_register_all_codecs;
-    use nihav_commonfmt::demuxers::generic_register_all_demuxers;
+    use nihav_codec_support::test::dec_video::*;
+    use crate::duck_register_all_decoders;
+    use nihav_commonfmt::generic_register_all_demuxers;
 
     #[test]
     fn test_vp6() {
         let mut dmx_reg = RegisteredDemuxers::new();
         generic_register_all_demuxers(&mut dmx_reg);
         let mut dec_reg = RegisteredDecoders::new();
-        duck_register_all_codecs(&mut dec_reg);
-
-        //let file = "assets/Duck/predator2_vp60.avi";
-        //let file = "assets/Duck/predator2_vp61.avi";
-        //let file = "assets/Duck/vp6_crash.avi";
-        let file = "assets/Duck/vp6_interlaced.avi";
-        //let file = "assets/Duck/vp6_vid.avi";
-        //let file = "assets/Duck/selection_720x576_300kBit_vp60i.avi";
-        //let file = "assets/Duck/selection_720x576_300kBit_flipped_vp60i.avi";
-        test_file_decoding("avi", file, Some(17), true, false, None/*Some("vp6")*/, &dmx_reg, &dec_reg);
-//panic!("end");
+        duck_register_all_decoders(&mut dec_reg);
+
+        // sample from a private collection
+        test_decoding("avi", "vp6", "assets/Duck/selection_720x576_300kBit_vp60i.avi", Some(16),
+                      &dmx_reg, &dec_reg,
+                      ExpectedTestResult::MD5([0x042c3e96, 0x8a9b26a2, 0x4dcbaf66, 0x1b788d03]));
+    }
+    #[test]
+    fn test_vp6_huff() {
+        let mut dmx_reg = RegisteredDemuxers::new();
+        generic_register_all_demuxers(&mut dmx_reg);
+        let mut dec_reg = RegisteredDecoders::new();
+        duck_register_all_decoders(&mut dec_reg);
+
+        // sample: https://samples.mplayerhq.hu/V-codecs/VP6/vp6_crash.avi
+        test_decoding("avi", "vp6", "assets/Duck/vp6_crash.avi", Some(4),
+                      &dmx_reg, &dec_reg, ExpectedTestResult::MD5Frames(vec![
+                            [0xdcd70fa0, 0x0d075ce2, 0xc9e65077, 0xb003a92e],
+                            [0x334abf96, 0x3a004c7a, 0x5781cd5c, 0x25c3ae5c],
+                            [0x6164b851, 0x528cd8de, 0xecab7328, 0x4b49708a],
+                            [0x11b048ac, 0xedb3e471, 0xd04e9399, 0x64e623e3],
+                            [0x182871b1, 0x2146893a, 0x2912210e, 0x6dd592e8]]));
+    }
+    #[test]
+    fn test_vp6_alpha() {
+        let mut dmx_reg = RegisteredDemuxers::new();
+        generic_register_all_demuxers(&mut dmx_reg);
+        let mut dec_reg = RegisteredDecoders::new();
+        duck_register_all_decoders(&mut dec_reg);
+
+        // sample created by remuxing some VP6A in FLV
+        test_decoding("avi", "vp6a", "assets/Duck/vp6a.avi", Some(25),
+                      &dmx_reg, &dec_reg, ExpectedTestResult::MD5Frames(vec![
+                            [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
+                            [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
+                            [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
+                            [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
+                            [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
+                            [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
+                            [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
+                            [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
+                            [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
+                            [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
+                            [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
+                            [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
+                            [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
+                            [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
+                            [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
+                            [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
+                            [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
+                            [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
+                            [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
+                            [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
+                            [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
+                            [0xaf903d79, 0x17ddb3c7, 0xf0a381e8, 0x26b36a7d],
+                            [0xd3801a96, 0x1b5384ff, 0x422b228c, 0x9c4582c4],
+                            [0x8ddb0dfe, 0x302eb1ed, 0x10e64e22, 0x5a5a62b9],
+                            [0x79338328, 0x06113781, 0x8b116d18, 0xde56707e],
+                            [0xdb58433b, 0x1de4ce67, 0x15fcbcee, 0x1df9de61]]));
     }
 }
-
-const VP6_AC_PROBS: [[[[u8; 11]; 6]; 2]; 3] = [
-  [
-    [
-      [ 227, 246, 230, 247, 244, 255, 255, 255, 255, 255, 255 ],
-      [ 255, 255, 209, 231, 231, 249, 249, 253, 255, 255, 255 ],
-      [ 255, 255, 225, 242, 241, 251, 253, 255, 255, 255, 255 ],
-      [ 255, 255, 241, 253, 252, 255, 255, 255, 255, 255, 255 ],
-      [ 255, 255, 248, 255, 255, 255, 255, 255, 255, 255, 255 ],
-      [ 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 ]
-    ], [
-      [ 240, 255, 248, 255, 255, 255, 255, 255, 255, 255, 255 ],
-      [ 255, 255, 240, 253, 255, 255, 255, 255, 255, 255, 255 ],
-      [ 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 ],
-      [ 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 ],
-      [ 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 ],
-      [ 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 ]
-    ]
-  ], [
-    [
-      [ 206, 203, 227, 239, 247, 255, 253, 255, 255, 255, 255 ],
-      [ 207, 199, 220, 236, 243, 252, 252, 255, 255, 255, 255 ],
-      [ 212, 219, 230, 243, 244, 253, 252, 255, 255, 255, 255 ],
-      [ 236, 237, 247, 252, 253, 255, 255, 255, 255, 255, 255 ],
-      [ 240, 240, 248, 255, 255, 255, 255, 255, 255, 255, 255 ],
-      [ 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 ]
-    ], [
-      [ 230, 233, 249, 255, 255, 255, 255, 255, 255, 255, 255 ],
-      [ 238, 238, 250, 255, 255, 255, 255, 255, 255, 255, 255 ],
-      [ 248, 251, 255, 255, 255, 255, 255, 255, 255, 255, 255 ],
-      [ 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 ],
-      [ 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 ],
-      [ 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 ]
-    ]
-  ], [
-    [
-      [ 225, 239, 227, 231, 244, 253, 243, 255, 255, 253, 255 ],
-      [ 232, 234, 224, 228, 242, 249, 242, 252, 251, 251, 255 ],
-      [ 235, 249, 238, 240, 251, 255, 249, 255, 253, 253, 255 ],
-      [ 249, 253, 251, 250, 255, 255, 255, 255, 255, 255, 255 ],
-      [ 251, 250, 249, 255, 255, 255, 255, 255, 255, 255, 255 ],
-      [ 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 ]
-    ], [
-      [ 243, 244, 250, 250, 255, 255, 255, 255, 255, 255, 255 ],
-      [ 249, 248, 250, 253, 255, 255, 255, 255, 255, 255, 255 ],
-      [ 253, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 ],
-      [ 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 ],
-      [ 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 ],
-      [ 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 ]
-    ]
-  ]
-];
-
-const VP6_DC_WEIGHTS: [[[i16; 2]; 3]; 5] = [
-    [ [ 122, 133 ], [ 133,  51 ], [ 142, -16 ] ],
-    [ [   0,   1 ], [   0,   1 ], [   0,   1 ] ],
-    [ [  78, 171 ], [ 169,  71 ], [ 221, -30 ] ],
-    [ [ 139, 117 ], [ 214,  44 ], [ 246,  -3 ] ],
-    [ [ 168,  79 ], [ 210,  38 ], [ 203,  17 ] ]
-];
-
-const VP6_IDX_TO_AC_BAND: [usize; 64] = [
-    0, 0, 1, 1, 1, 2, 2, 2,
-    2, 2, 2, 3, 3, 3, 3, 3,
-    3, 3, 3, 3, 3, 3, 4, 4,
-    4, 4, 4, 4, 4, 4, 4, 4,
-    4, 4, 4, 4, 4, 5, 5, 5,
-    5, 5, 5, 5, 5, 5, 5, 5,
-    5, 5, 5, 5, 5, 5, 5, 5,
-    5, 5, 5, 5, 5, 5, 5, 5
-];
-
-const VP6_BICUBIC_COEFFS: [[[i16; 4]; 8]; 17] = [
-  [
-    [   0, 128,   0,   0 ],
-    [  -3, 122,   9,   0 ],
-    [  -4, 109,  24,  -1 ],
-    [  -5,  91,  45,  -3 ],
-    [  -4,  68,  68,  -4 ],
-    [  -3,  45,  91,  -5 ],
-    [  -1,  24, 109,  -4 ],
-    [   0,   9, 122,  -3 ]
-  ], [
-    [   0, 128,   0,   0 ],
-    [  -4, 124,   9,  -1 ],
-    [  -5, 110,  25,  -2 ],
-    [  -6,  91,  46,  -3 ],
-    [  -5,  69,  69,  -5 ],
-    [  -3,  46,  91,  -6 ],
-    [  -2,  25, 110,  -5 ],
-    [  -1,   9, 124,  -4 ]
-  ], [
-    [   0, 128,   0,   0 ],
-    [  -4, 123,  10,  -1 ],
-    [  -6, 110,  26,  -2 ],
-    [  -7,  92,  47,  -4 ],
-    [  -6,  70,  70,  -6 ],
-    [  -4,  47,  92,  -7 ],
-    [  -2,  26, 110,  -6 ],
-    [  -1,  10, 123,  -4 ]
-  ], [
-    [   0, 128,   0,   0 ],
-    [  -5, 124,  10,  -1 ],
-    [  -7, 110,  27,  -2 ],
-    [  -7,  91,  48,  -4 ],
-    [  -6,  70,  70,  -6 ],
-    [  -4,  48,  92,  -8 ],
-    [  -2,  27, 110,  -7 ],
-    [  -1,  10, 124,  -5 ]
-  ], [
-    [   0, 128,   0,   0 ],
-    [  -6, 124,  11,  -1 ],
-    [  -8, 111,  28,  -3 ],
-    [  -8,  92,  49,  -5 ],
-    [  -7,  71,  71,  -7 ],
-    [  -5,  49,  92,  -8 ],
-    [  -3,  28, 111,  -8 ],
-    [  -1,  11, 124,  -6 ]
-  ], [
-    [   0, 128,   0,   0 ],
-    [  -6, 123,  12,  -1 ],
-    [  -9, 111,  29,  -3 ],
-    [  -9,  93,  50,  -6 ],
-    [  -8,  72,  72,  -8 ],
-    [  -6,  50,  93,  -9 ],
-    [  -3,  29, 111,  -9 ],
-    [  -1,  12, 123,  -6 ]
-  ], [
-    [   0, 128,   0,   0 ],
-    [  -7, 124,  12,  -1 ],
-    [ -10, 111,  30,  -3 ],
-    [ -10,  93,  51,  -6 ],
-    [  -9,  73,  73,  -9 ],
-    [  -6,  51,  93, -10 ],
-    [  -3,  30, 111, -10 ],
-    [  -1,  12, 124,  -7 ]
-  ], [
-    [   0, 128,   0,   0 ],
-    [  -7, 123,  13,  -1 ],
-    [ -11, 112,  31,  -4 ],
-    [ -11,  94,  52,  -7 ],
-    [ -10,  74,  74, -10 ],
-    [  -7,  52,  94, -11 ],
-    [  -4,  31, 112, -11 ],
-    [  -1,  13, 123,  -7 ]
-  ], [
-    [   0, 128,   0,   0 ],
-    [  -8, 124,  13,  -1 ],
-    [ -12, 112,  32,  -4 ],
-    [ -12,  94,  53,  -7 ],
-    [ -10,  74,  74, -10 ],
-    [  -7,  53,  94, -12 ],
-    [  -4,  32, 112, -12 ],
-    [  -1,  13, 124,  -8 ]
-  ], [
-    [   0, 128,   0,   0 ],
-    [  -9, 124,  14,  -1 ],
-    [ -13, 112,  33,  -4 ],
-    [ -13,  95,  54,  -8 ],
-    [ -11,  75,  75, -11 ],
-    [  -8,  54,  95, -13 ],
-    [  -4,  33, 112, -13 ],
-    [  -1,  14, 124,  -9 ]
-  ], [
-    [   0, 128,   0,   0 ],
-    [  -9, 123,  15,  -1 ],
-    [ -14, 113,  34,  -5 ],
-    [ -14,  95,  55,  -8 ],
-    [ -12,  76,  76, -12 ],
-    [  -8,  55,  95, -14 ],
-    [  -5,  34, 112, -13 ],
-    [  -1,  15, 123,  -9 ]
-  ], [
-    [   0, 128,   0,   0 ],
-    [ -10, 124,  15,  -1 ],
-    [ -14, 113,  34,  -5 ],
-    [ -15,  96,  56,  -9 ],
-    [ -13,  77,  77, -13 ],
-    [  -9,  56,  96, -15 ],
-    [  -5,  34, 113, -14 ],
-    [  -1,  15, 124, -10 ]
-  ], [
-    [   0, 128,   0,   0 ],
-    [ -10, 123,  16,  -1 ],
-    [ -15, 113,  35,  -5 ],
-    [ -16,  98,  56, -10 ],
-    [ -14,  78,  78, -14 ],
-    [ -10,  56,  98, -16 ],
-    [  -5,  35, 113, -15 ],
-    [  -1,  16, 123, -10 ]
-  ], [
-    [   0, 128,   0,   0 ],
-    [ -11, 124,  17,  -2 ],
-    [ -16, 113,  36,  -5 ],
-    [ -17,  98,  57, -10 ],
-    [ -14,  78,  78, -14 ],
-    [ -10,  57,  98, -17 ],
-    [  -5,  36, 113, -16 ],
-    [  -2,  17, 124, -11 ]
-  ], [
-    [   0, 128,   0,   0 ],
-    [ -12, 125,  17,  -2 ],
-    [ -17, 114,  37,  -6 ],
-    [ -18,  99,  58, -11 ],
-    [ -15,  79,  79, -15 ],
-    [ -11,  58,  99, -18 ],
-    [  -6,  37, 114, -17 ],
-    [  -2,  17, 125, -12 ]
-  ], [
-    [   0, 128,   0,   0 ],
-    [ -12, 124,  18,  -2 ],
-    [ -18, 114,  38,  -6 ],
-    [ -19,  99,  59, -11 ],
-    [ -16,  80,  80, -16 ],
-    [ -11,  59,  99, -19 ],
-    [  -6,  38, 114, -18 ],
-    [  -2,  18, 124, -12 ]
-  ], [
-    [   0, 128,   0,   0 ],
-    [  -4, 118,  16,  -2 ],
-    [  -7, 106,  34,  -5 ],
-    [  -8,  90,  53,  -7 ],
-    [  -8,  72,  72,  -8 ],
-    [  -7,  53,  90,  -8 ],
-    [  -5,  34, 106,  -7 ],
-    [  -2,  16, 118,  -4 ]
-  ]
-];