Update src/codecs/avif/yuv.rs

Co-authored-by: Andreas Molzer <[email protected]>
image-rs · Nov 2, 2024 · fa0bf3b · fa0bf3b
1 parent b2e7a73
commit fa0bf3b
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/codecs/avif/yuv.rs b/src/codecs/avif/yuv.rs
@@ -727,9 +727,15 @@ where
             let cb_value = u_src.as_() - bias_uv;
             let cr_value = v_src.as_() - bias_uv;
 
-            let r = ((y_value + cr_coef * cr_value + ROUNDING) >> PRECISION).clamp(0, max_value);
-            let b = ((y_value + cb_coef * cb_value + ROUNDING) >> PRECISION).clamp(0, max_value);
-            let g = ((y_value - g_coef_1 * cr_value - g_coef_2 * cb_value + ROUNDING) >> PRECISION)
+#[inline(always)]
+fn round<const PRECISION: i32, const MAX: i32>(val: i32) -> i32 {
+    let ROUNDING: i32 = 1 << (PRECISION - 1);
+    ((val + ROUNDING) >> PRECISION).clamp(0, MAX)
+}
+
+            let r = round::<PRECISION, MAX>(y_value + cr_coef * cr_value);
+            let b = round::<PRECISION, MAX>(y_value + cb_coef * cb_value);
+            let g = round::<PRECISION, MAX>(y_value - g_coef_1 * cr_value - g_coef_2 * cb_value);
                 .clamp(0, max_value);
 
             if CHANNELS == 4 {