fn invert: Calculate y_off var with multiplication.

memorysafety · Sep 18, 2023 · 7ff1214 · 7ff1214
1 parent b747b59
commit 7ff1214
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/wedge.rs b/src/wedge.rs
@@ -196,12 +196,11 @@ const fn hflip(src: &[u8; 64 * 64]) -> [u8; 64 * 64] {
 }
 
 fn invert(dst: &mut [u8], src: &[u8], w: usize, h: usize) {
-    let mut y_off = 0;
-    for _ in 0..h {
+    for y in 0..h {
+        let y_off = y * w;
         for x in 0..w {
             dst[y_off + x] = 64 - src[y_off + x];
         }
-        y_off += w;
     }
 }