rv6: fix some typos in loop filtering

[nihav.git] / nihav-realmedia / src / codecs / rv60.rs

diff --git a/nihav-realmedia/src/codecs/rv60.rs b/nihav-realmedia/src/codecs/rv60.rs
index bedebc9ac2161641ddb6b9f41692cc64c1a40267..3f4ec9e263a20ade54db810397c844e3f6973265 100644 (file)
--- a/nihav-realmedia/src/codecs/rv60.rs
+++ b/nihav-realmedia/src/codecs/rv60.rs
@@ -49,7 +49,9 @@ struct FrameHeader {
     osvquant:           u8,
     ts:                 u32,
     width:              usize,
+    awidth:             usize,
     height:             usize,
+    aheight:            usize,
     two_f_refs:         bool,
     qp_off_type:        u8,
     deblock:            bool,
@@ -78,6 +80,8 @@ impl FrameHeader {
         let width                                       = ((br.read(11)? as usize) + 1) * 4;
         let height                                      = ((br.read(11)? as usize) + 0) * 4;
         validate!(height > 0);
+        let awidth  = (width  + 15) & !15;
+        let aheight = (height + 15) & !15;
         let _some_flag                                  = br.read_bool()?;
         let two_f_refs;
         if ftype == FrameType::I {
@@ -110,8 +114,8 @@ impl FrameHeader {
         }
 
         Ok(FrameHeader {
-                profile, ftype, qp, osvquant, ts, width, height, two_f_refs, qp_off_type,
-                deblock, deblock_chroma,
+                profile, ftype, qp, osvquant, ts, width, height, awidth, aheight,
+                two_f_refs, qp_off_type, deblock, deblock_chroma,
             })
     }
     fn parse_slice_sizes(&self, br: &mut BitReader, sizes: &mut Vec<usize>) -> DecoderResult<()> {
@@ -716,10 +720,10 @@ println!(" left {} bits", br.left());
     }
     #[allow(clippy::cognitive_complexity)]
     fn decode_cb_tree(&mut self, buf: &mut NASimpleVideoFrame<u8>, hdr: &FrameHeader, br: &mut BitReader, xpos: usize, ypos: usize, log_size: u8) -> DecoderResult<()> {
-        if (xpos >= hdr.width) || (ypos >= hdr.height) { return Ok(()); }
+        if (xpos >= hdr.awidth) || (ypos >= hdr.aheight) { return Ok(()); }
 
         let size = 1 << log_size;
-        let split = (xpos + size > hdr.width) || (ypos + size > hdr.height) || (size > 8 && br.read_bool()?);
+        let split = (xpos + size > hdr.awidth) || (ypos + size > hdr.aheight) || (size > 8 && br.read_bool()?);
         self.cu_splits.push(split);
         if split {
             let hsize = size >> 1;
@@ -1486,7 +1490,7 @@ println!("???");
         self.blk_info.clear();
         self.blk_info.resize(self.blk_stride * (cu_h << 4), BlockInfo::default());
         if hdr.deblock {
-            self.dblk.reinit(hdr.width, hdr.height);
+            self.dblk.reinit(hdr.awidth, hdr.aheight);
         }
         let mut off = hsize + ((br.tell() >> 3) as usize);
         let mut dframe = NASimpleVideoFrame::from_video_buf(&mut buf).unwrap();
@@ -1579,17 +1583,17 @@ mod test {
         test_decoding("realmedia", "realvideo6", "assets/RV/RV60.rmhd", Some(1000), &dmx_reg, &dec_reg,
                       ExpectedTestResult::MD5Frames(vec![
                             [0x2b1f1807, 0x09edef33, 0x0e6c78c1, 0x3b3c8179],
-                            [0xea406850, 0x400802b8, 0xac106fb6, 0xe1e2e766],
+                            [0x9b8feb58, 0x161b1b44, 0xbca532e5, 0xcb4ac211],
                             [0x2b1f1807, 0x09edef33, 0x0e6c78c1, 0x3b3c8179],
-                            [0xb04e2626, 0x976e16f5, 0xc41a7a78, 0x2d8765da],
-                            [0xf4f30d97, 0x7f2876eb, 0x265ffad4, 0x3542a7c4],
-                            [0xa5082524, 0x38a86952, 0x35bf1fee, 0xfc830d3f],
-                            [0x75eab1a2, 0x62e2222f, 0xe96a20d9, 0x652140b4],
-                            [0x7590fa49, 0x78c83490, 0x239eeff9, 0x64282ac7],
-                            [0x70b19e9f, 0x66c1f866, 0xb8d7142a, 0xf3e424b2],
-                            [0xc2934123, 0x3bf72fc4, 0x12d8d123, 0x1f39525b],
-                            [0x13344919, 0xecd01190, 0x2f69079b, 0xbf4d7026],
-                            [0xcefb3284, 0xa9b36d4d, 0xf1aa6752, 0xaae17d44],
-                            [0x57f01275, 0xf8e883ea, 0x4865752e, 0xc760a777]]));
+                            [0x5dc8b7d1, 0xef6e6840, 0x7b634afb, 0x645711a3],
+                            [0x64e802be, 0xd80dc046, 0x98309de0, 0xe9f2ee48],
+                            [0x64d1fa65, 0xfe30eccf, 0x08e07dd5, 0xb7f079ec],
+                            [0xfae26181, 0x63e190ec, 0xfa0ae49c, 0x095abb3c],
+                            [0xc6491458, 0xf3a1e979, 0x6f4e64ab, 0xe77046cb],
+                            [0x4ec5e66d, 0x5a99a40d, 0x871b16ab, 0xf63b2890],
+                            [0x2d22683b, 0x4022a36b, 0xdfcd3259, 0xa7f43f77],
+                            [0xff6a1c72, 0x2dd8b7ca, 0xe3f2f575, 0x2d7aa001],
+                            [0xf95f1898, 0x1660a50d, 0x2b36aacc, 0x921b5402],
+                            [0x8115fafe, 0x39ac33b2, 0x9d65b3cc, 0x5ec452fe]]));
     }
 }