feat: add support for test vector profile as optional parameter

mdimopoulos · mdimopoulos · commit f807d7d6ad8c · 2025-07-17T11:53:00.000+02:00
- create profile enumerator class for H.264 profiles
- modify test vector class structure
diff --git a/fluster/codec.py b/fluster/codec.py
@@ -60,3 +60,24 @@ class OutputFormat(Enum):
     GRAY16LE = "gray16le"
     UNKNOWN = "Unknown"
     FLTP = "fltp"
+
+
+class Profile(Enum):
+    """Profile"""
+
+    NONE = "None"
+
+    # H.264
+    CONSTRAINED_BASELINE = "Constrained Baseline"
+    BASELINE = "Baseline"
+    EXTENDED = "Extended"
+    MAIN = "Main"
+    HIGH = "High"
+    HIGH_10 = "High 10"
+    HIGH_10_INTRA = "High 10 Intra"
+    HIGH_4_2_2 = "High 4:2:2"
+    HIGH_4_2_2_INTRA = "High 4:2:2 Intra"
+    HIGH_4_4_4_INTRA = "High 4:4:4 Intra"
+    HIGH_4_4_4_PREDICTIVE = "High 4:4:4 Predictive"
+    CAVLC_4_4_4 = "CAVLC 4:4:4"
+    CAVLC_4_4_4_INTRA = "CAVLC 4:4:4 Intra"
diff --git a/fluster/test_vector.py b/fluster/test_vector.py
@@ -16,9 +16,9 @@
 # License along with this library. If not, see <https://www.gnu.org/licenses/>.
 
 from enum import Enum
-from typing import Any, Dict, List, Type
+from typing import Any, Dict, List, Optional, Type
 
-from fluster.codec import OutputFormat
+from fluster.codec import OutputFormat, Profile
 
 
 class TestVectorResult(Enum):
@@ -43,12 +43,14 @@ def __init__(
         input_file: str,
         output_format: OutputFormat,
         result: str,
+        profile: Optional[Profile] = None,
     ):
         # JSON members
         self.name = name
         self.source = source
         self.source_checksum = source_checksum
         self.input_file = input_file
+        self.profile = profile
         self.output_format = output_format
         self.result = result
 
@@ -64,6 +66,11 @@ def from_json(cls: Type["TestVector"], data: Any) -> Any:
             data["output_format"] = OutputFormat(data["output_format"])
         else:
             data["output_format"] = OutputFormat.NONE
+
+        # We only define profile if the paramter is found in .json of test suite
+        if "profile" in data:
+            data["profile"] = Profile(data["profile"])
+
         return (data["name"], cls(**data))
 
     def data_to_serialize(self) -> Dict[str, object]:
@@ -73,13 +80,18 @@ def data_to_serialize(self) -> Dict[str, object]:
         data.pop("errors")
         data.pop("test_time")
         data["output_format"] = str(self.output_format.value)
+        if self.profile is not None:
+            data["profile"] = str(self.profile.value)
+        else:
+            data.pop("profile")
         return data
 
     def __str__(self) -> str:
         ret = (
             f"        {self.name}\n"
             f"            Source: {self.source}\n"
             f"            Input: {self.input_file}\n"
+            f"            Profile: {self.profile}\n"
             f"            Result: {self.result}"
         )
         return ret
