add huffman to test2

1 files changed, 3 insertions, 3 deletions

diff --git a/test2/src/encode.rs b/test2/src/encode.rs
index 7eb82e0..53ca8a9 100644
--- a/test2/src/encode.rs
+++ b/test2/src/encode.rs
@@ -1,4 +1,4 @@
-use crate::{BLOCK_SIZE, Frame};
+use crate::{BLOCK_SIZE, Frame, huffman::encode_huff};
 use framework::BitstreamFilter;
 use glam::{IVec2, ivec2};
 
@@ -79,7 +79,7 @@ impl BitstreamFilter for Enc {
                     if neg {
                         d = (-(d as i8)) as u8;
                     }
-                    d &= 0xff << (6 - d.leading_zeros()); // only keep 3 msb
+                    d &= 0xff << (7 - d.leading_zeros()); // only keep 3 msb
                     if neg {
                         d = (-(d as i8)) as u8;
                     }
@@ -95,6 +95,6 @@ impl BitstreamFilter for Enc {
         }
 
         self.last = frame.clone();
-        out
+        encode_huff(&out)
     }
 }