fix: Add missing output format in AV1 Argon Profile 0 test suite vectors

Author: mcesariniflu

README.md

@@ -471,7 +471,7 @@ AAC
 - [AAC](https://standards.iso.org/ittf/PubliclyAvailableStandards/ISO_IEC_14496-26_2010_Bitstreams/DVD1/mpeg4audio-conformance/).
 - [AV1](https://storage.googleapis.com/aom-test-data/).
 - [AV1](https://source.chromium.org/chromiumos/chromiumos/codesearch/+/main:src/platform/tast-tests/src/chromiumos/tast/local/bundles/cros/video/data/test_vectors/av1/).
-- [AV1](https://storage.googleapis.com/downloads.aomedia.org/assets/zip/).
+- [AV1](https://aom-cwg-av1-argon-streams-public.s3.us-east-1.amazonaws.com/).
 - [H.264/AVC](https://www.itu.int/wftp3/av-arch/jvt-site/draft_conformance/).
 - [H.265/HEVC](https://www.itu.int/wftp3/av-arch/jctvc-site/bitstream_exchange/draft_conformance/).
 - [H.266/VVC](https://www.itu.int/wftp3/av-arch/jvet-site/bitstream_exchange/VVC/draft_conformance/).

scripts/gen_av1_argon.py

@@ -154,7 +154,24 @@ def generate(self, download: bool) -> None:
                     output_format = OutputFormat[pix_fmt.upper()]
                     if output_format == OutputFormat.UNKNOWN:
                         raise KeyError
-                except (subprocess.CalledProcessError, KeyError) as error:
+                except (subprocess.CalledProcessError, KeyError):
+                    exceptions_profile0_output_format = {
+                        # Exceptions for profile 0 core (annex b) vectors
+                        "test54": OutputFormat.YUV420P10LE,
+                        "test11142": OutputFormat.YUV420P10LE,
+                        "test11079_11110": OutputFormat.YUV444P12LE,
+                        "test11145": OutputFormat.GRAY10LE,
+                        "test11023": OutputFormat.YUV420P10LE,
+                        "test11174": OutputFormat.YUV420P10LE,
+                        # Exceptions for profile 0 non-annex b vectors
+                        "test12293_12304": OutputFormat.GRAY10LE,
+                        # Exceptions for profile 0 stress annex b vectors
+                        "test50688": OutputFormat.YUV420P10LE,
+                        "test23792": OutputFormat.YUV420P10LE,
+                        "test79399": OutputFormat.YUV420P10LE,
+                        "test29600": OutputFormat.YUV420P10LE,
+                    }
+
                     exceptions_profile1_output_format = {
                         # Exceptions for profile 1 core (annex b) vectors
                         "test54": OutputFormat.GBRP10LE,
@@ -194,12 +211,12 @@ def generate(self, download: bool) -> None:
                         "test85818": OutputFormat.YUV422P10LE,
                     }
 
-                    if tv_filename in exceptions_profile1_output_format.keys():
+                    if tv_filename in exceptions_profile0_output_format.keys():
+                        output_format = exceptions_profile0_output_format[tv_filename]
+                    elif tv_filename in exceptions_profile1_output_format.keys():
                         output_format = exceptions_profile1_output_format[tv_filename]
                     elif tv_filename in exceptions_profile2_output_format.keys():
                         output_format = exceptions_profile2_output_format[tv_filename]
-                    else:
-                        raise error
 
             temp_output_ref = f"{os.path.splitext(tv_abs_path)[0]}.out"
             try:

test_suites/av1/AV1-ARGON-PROFILE0-CORE-ANNEX-B.json

@@ -72,7 +72,7 @@
             "source": "https://aom-cwg-av1-argon-streams-public.s3.us-east-1.amazonaws.com/argon_coveragetool_av1_base_and_extended_profiles_v2.1.1.zip",
             "source_checksum": "5dd3554c3f73c80f06c7fceeaa1f5dd9",
             "input_file": "argon_coveragetool_av1_base_and_extended_profiles_v2.1/profile0_core_special/streams/test54.obu",
-            "output_format": "None",
+            "output_format": "yuv420p10le",
             "result": "139317f4dc2134bcef8d36a8fb117cf4"
         },
         {
@@ -824,7 +824,7 @@
             "source": "https://aom-cwg-av1-argon-streams-public.s3.us-east-1.amazonaws.com/argon_coveragetool_av1_base_and_extended_profiles_v2.1.1.zip",
             "source_checksum": "5dd3554c3f73c80f06c7fceeaa1f5dd9",
             "input_file": "argon_coveragetool_av1_base_and_extended_profiles_v2.1/profile0_core/streams/test11142.obu",
-            "output_format": "None",
+            "output_format": "yuv420p10le",
             "result": "513f5dd6a0a96db2ca82be66d5bf13dd"
         },
         {
@@ -1592,7 +1592,7 @@
             "source": "https://aom-cwg-av1-argon-streams-public.s3.us-east-1.amazonaws.com/argon_coveragetool_av1_base_and_extended_profiles_v2.1.1.zip",
             "source_checksum": "5dd3554c3f73c80f06c7fceeaa1f5dd9",
             "input_file": "argon_coveragetool_av1_base_and_extended_profiles_v2.1/profile0_core/streams/test11079_11110.obu",
-            "output_format": "None",
+            "output_format": "yuv444p12le",
             "result": "2c292c849264f017d40caf9970df7040"
         },
         {
@@ -1848,7 +1848,7 @@
             "source": "https://aom-cwg-av1-argon-streams-public.s3.us-east-1.amazonaws.com/argon_coveragetool_av1_base_and_extended_profiles_v2.1.1.zip",
             "source_checksum": "5dd3554c3f73c80f06c7fceeaa1f5dd9",
             "input_file": "argon_coveragetool_av1_base_and_extended_profiles_v2.1/profile0_core/streams/test11145.obu",
-            "output_format": "None",
+            "output_format": "gray10le",
             "result": "c5401b78b8e5e473cad61ea61b3c9241"
         },
         {
@@ -2312,7 +2312,7 @@
             "source": "https://aom-cwg-av1-argon-streams-public.s3.us-east-1.amazonaws.com/argon_coveragetool_av1_base_and_extended_profiles_v2.1.1.zip",
             "source_checksum": "5dd3554c3f73c80f06c7fceeaa1f5dd9",
             "input_file": "argon_coveragetool_av1_base_and_extended_profiles_v2.1/profile0_core/streams/test11023.obu",
-            "output_format": "None",
+            "output_format": "yuv420p10le",
             "result": "6661cb3843a0e4ebae39c024973e314f"
         },
         {
@@ -4128,7 +4128,7 @@
             "source": "https://aom-cwg-av1-argon-streams-public.s3.us-east-1.amazonaws.com/argon_coveragetool_av1_base_and_extended_profiles_v2.1.1.zip",
             "source_checksum": "5dd3554c3f73c80f06c7fceeaa1f5dd9",
             "input_file": "argon_coveragetool_av1_base_and_extended_profiles_v2.1/profile0_core/streams/test11174.obu",
-            "output_format": "None",
+            "output_format": "yuv420p10le",
             "result": "6aa590195fd910008eeb70c6f0583fef"
         },
         {

test_suites/av1/AV1-ARGON-PROFILE0-NON-ANNEX-B.json

@@ -280,7 +280,7 @@
             "source": "https://aom-cwg-av1-argon-streams-public.s3.us-east-1.amazonaws.com/argon_coveragetool_av1_base_and_extended_profiles_v2.1.1.zip",
             "source_checksum": "5dd3554c3f73c80f06c7fceeaa1f5dd9",
             "input_file": "argon_coveragetool_av1_base_and_extended_profiles_v2.1/profile0_error/streams/test50.obu",
-            "output_format": "Unknown",
+            "output_format": "gbrp10le",
             "result": ""
         },
         {
@@ -440,7 +440,7 @@
             "source": "https://aom-cwg-av1-argon-streams-public.s3.us-east-1.amazonaws.com/argon_coveragetool_av1_base_and_extended_profiles_v2.1.1.zip",
             "source_checksum": "5dd3554c3f73c80f06c7fceeaa1f5dd9",
             "input_file": "argon_coveragetool_av1_base_and_extended_profiles_v2.1/profile0_error/streams/test12.obu",
-            "output_format": "None",
+            "output_format": "Unknown",
             "result": ""
         },
         {
@@ -472,7 +472,7 @@
             "source": "https://aom-cwg-av1-argon-streams-public.s3.us-east-1.amazonaws.com/argon_coveragetool_av1_base_and_extended_profiles_v2.1.1.zip",
             "source_checksum": "5dd3554c3f73c80f06c7fceeaa1f5dd9",
             "input_file": "argon_coveragetool_av1_base_and_extended_profiles_v2.1/profile0_not_annexb/streams/test12293_12304.obu",
-            "output_format": "None",
+            "output_format": "gray10le",
             "result": "266b24ac245fc94cf2ff96e07cc22f7d"
         },
         {
@@ -656,7 +656,7 @@
             "source": "https://aom-cwg-av1-argon-streams-public.s3.us-east-1.amazonaws.com/argon_coveragetool_av1_base_and_extended_profiles_v2.1.1.zip",
             "source_checksum": "5dd3554c3f73c80f06c7fceeaa1f5dd9",
             "input_file": "argon_coveragetool_av1_base_and_extended_profiles_v2.1/profile0_error/streams/test242.obu",
-            "output_format": "None",
+            "output_format": "Unknown",
             "result": ""
         },
         {

test_suites/av1/AV1-ARGON-PROFILE0-STRESS-ANNEX-B.json

@@ -48,7 +48,7 @@
             "source": "https://aom-cwg-av1-argon-streams-public.s3.us-east-1.amazonaws.com/argon_coveragetool_av1_base_and_extended_profiles_v2.1.1.zip",
             "source_checksum": "5dd3554c3f73c80f06c7fceeaa1f5dd9",
             "input_file": "argon_coveragetool_av1_base_and_extended_profiles_v2.1/profile0_stress/streams/test50688.obu",
-            "output_format": "Unknown",
+            "output_format": "yuv420p10le",
             "result": "600e32ac4e4d93d25a81a92a0e0627ca"
         },
         {
@@ -168,7 +168,7 @@
             "source": "https://aom-cwg-av1-argon-streams-public.s3.us-east-1.amazonaws.com/argon_coveragetool_av1_base_and_extended_profiles_v2.1.1.zip",
             "source_checksum": "5dd3554c3f73c80f06c7fceeaa1f5dd9",
             "input_file": "argon_coveragetool_av1_base_and_extended_profiles_v2.1/profile0_stress/streams/test23792.obu",
-            "output_format": "Unknown",
+            "output_format": "yuv420p10le",
             "result": "04499b24157642121d89398adca47144"
         },
         {
@@ -200,7 +200,7 @@
             "source": "https://aom-cwg-av1-argon-streams-public.s3.us-east-1.amazonaws.com/argon_coveragetool_av1_base_and_extended_profiles_v2.1.1.zip",
             "source_checksum": "5dd3554c3f73c80f06c7fceeaa1f5dd9",
             "input_file": "argon_coveragetool_av1_base_and_extended_profiles_v2.1/profile0_stress/streams/test79399.obu",
-            "output_format": "Unknown",
+            "output_format": "yuv420p10le",
             "result": "a7c0ef2cfe9a2b2bdacc22861a6e29e9"
         },
         {
@@ -344,7 +344,7 @@
             "source": "https://aom-cwg-av1-argon-streams-public.s3.us-east-1.amazonaws.com/argon_coveragetool_av1_base_and_extended_profiles_v2.1.1.zip",
             "source_checksum": "5dd3554c3f73c80f06c7fceeaa1f5dd9",
             "input_file": "argon_coveragetool_av1_base_and_extended_profiles_v2.1/profile0_stress/streams/test29600.obu",
-            "output_format": "Unknown",
+            "output_format": "yuv420p10le",
             "result": "82591ecfd78b22df254031b6a76fc56f"
         },
         {