parse xpsnr rgb output

alexheretic · Dec 22, 2024 · ddc2945 · ddc2945
1 parent 941c0a2
commit ddc2945
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/xpsnr.rs b/src/xpsnr.rs
@@ -87,7 +87,7 @@ impl XpsnrOut {
 fn score_from_line(line: &str) -> Option<f32> {
     const MIN_PREFIX: &str = "minimum: ";
 
-    if !line.contains("XPSNR  y: ") {
+    if !line.contains("XPSNR") {
         return None;
     }
 
@@ -105,6 +105,14 @@ fn score_from_line(line: &str) -> Option<f32> {
 mod test {
     use super::*;
 
+    #[test]
+    fn parse_rgb_line() {
+        let score = score_from_line(
+            "XPSNR average, 1 frames  r: 40.6130  g: 41.0275  b: 40.6961  (minimum: 40.6130)",
+        );
+        assert_eq!(score, Some(40.6130));
+    }
+
     #[test]
     fn parse_xpsnr_score() {
         // Note: some lines omitted for brevity