Merge pull request #150 from yamt/overflow

yuyv444_to_rgb: avoid highlight overflows
l1npengtul · Oct 17, 2023 · f0bbbf3 · f0bbbf3
2 parents ae0d818 + aa4d755
commit f0bbbf3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/nokhwa-core/src/types.rs b/nokhwa-core/src/types.rs
@@ -1617,9 +1617,9 @@ pub fn yuyv444_to_rgb(y: i32, u: i32, v: i32) -> [u8; 3] {
     let c298 = (y - 16) * 298;
     let d = u - 128;
     let e = v - 128;
-    let r = ((c298 + 409 * e + 128) >> 8) as u8;
-    let g = ((c298 - 100 * d - 208 * e + 128) >> 8) as u8;
-    let b = ((c298 + 516 * d + 128) >> 8) as u8;
+    let r = ((c298 + 409 * e + 128) >> 8).clamp(0, 255) as u8;
+    let g = ((c298 - 100 * d - 208 * e + 128) >> 8).clamp(0, 255) as u8;
+    let b = ((c298 + 516 * d + 128) >> 8).clamp(0, 255) as u8;
     [r, g, b]
 }