tests: pbutils: Add test parsing H265 profiles

diff --git a/tests/check/libs/pbutils.c b/tests/check/libs/pbutils.c
index ac5aeb5..bf61c08 100644 (file)
--- a/tests/check/libs/pbutils.c
+++ b/tests/check/libs/pbutils.c
@@ -867,6 +867,299 @@ GST_START_TEST (test_pb_utils_aac_get_profile)
 
 GST_END_TEST;
 
+#define PROFILE_TIER_LEVEL_LEN 11
+
+static void
+fill_h265_profile (guint8 * profile_tier_level,
+    guint8 profile_idc, guint8 max_14bit_flag, guint8 max_12bit_flag,
+    guint8 max_10bit_flag, guint8 max_8bit_flag, guint8 max_422_flag,
+    guint8 max_420_flag, guint8 max_mono_flag, guint8 intra_flag,
+    guint8 one_pic_flag, guint8 lower_bit_rate_flag)
+{
+  /* Bit 0:1   - general_profile_space
+   * Bit 2     - general_tier_flag
+   * Bit 3:7   - general_profile_idc
+   * Bit 8:39  - gernal_profile_compatibility_flags
+   * Bit 40    - general_progressive_source_flag
+   * Bit 41    - general_interlaced_source_flag
+   * Bit 42    - general_non_packed_constraint_flag
+   * Bit 43    - general_frame_only_constraint_flag
+   */
+
+  memset (profile_tier_level, 0x0, PROFILE_TIER_LEVEL_LEN);
+
+  profile_tier_level[0] = profile_idc;
+
+  if (profile_idc < 4)
+    return;
+
+  profile_tier_level[5] |= (max_12bit_flag << 3);
+  profile_tier_level[5] |= (max_10bit_flag << 2);
+  profile_tier_level[5] |= (max_8bit_flag << 1);
+  profile_tier_level[5] |= max_422_flag;
+  profile_tier_level[6] |= (max_420_flag << 7);
+  profile_tier_level[6] |= (max_mono_flag << 6);
+  profile_tier_level[6] |= (intra_flag << 5);
+  profile_tier_level[6] |= (one_pic_flag << 4);
+  profile_tier_level[6] |= (lower_bit_rate_flag << 3);
+  profile_tier_level[6] |= (max_14bit_flag << 2);
+}
+
+GST_START_TEST (test_pb_utils_h265_profiles)
+{
+  guint8 profile_tier_level[PROFILE_TIER_LEVEL_LEN] = { 0, };
+  const gchar *profile;
+
+  fill_h265_profile (profile_tier_level, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "main");
+
+  fill_h265_profile (profile_tier_level, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "main-10");
+
+  fill_h265_profile (profile_tier_level, 3, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "main-still-picture");
+
+  /* Format range extensions profiles */
+  fill_h265_profile (profile_tier_level, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless (profile == NULL);
+
+  fill_h265_profile (profile_tier_level, 4, 0, 1, 1, 1, 1, 1, 1, 0, 0, 1);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "monochrome");
+
+  fill_h265_profile (profile_tier_level, 4, 0, 1, 1, 0, 1, 1, 1, 0, 0, 1);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "monochrome-10");
+
+  fill_h265_profile (profile_tier_level, 4, 0, 1, 0, 0, 1, 1, 1, 0, 0, 1);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "monochrome-12");
+
+  fill_h265_profile (profile_tier_level, 4, 0, 0, 0, 0, 1, 1, 1, 0, 0, 1);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "monochrome-16");
+
+  fill_h265_profile (profile_tier_level, 4, 0, 1, 0, 0, 1, 1, 0, 0, 0, 1);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "main-12");
+
+  fill_h265_profile (profile_tier_level, 4, 0, 1, 1, 0, 1, 0, 0, 0, 0, 1);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "main-422-10");
+
+  fill_h265_profile (profile_tier_level, 4, 0, 1, 0, 0, 1, 0, 0, 0, 0, 1);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "main-422-12");
+
+  fill_h265_profile (profile_tier_level, 4, 0, 1, 1, 1, 0, 0, 0, 0, 0, 1);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "main-444");
+
+  fill_h265_profile (profile_tier_level, 4, 0, 1, 1, 0, 0, 0, 0, 0, 0, 1);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "main-444-10");
+
+  fill_h265_profile (profile_tier_level, 4, 0, 1, 0, 0, 0, 0, 0, 0, 0, 1);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "main-444-12");
+
+  fill_h265_profile (profile_tier_level, 4, 0, 1, 1, 1, 1, 1, 0, 1, 0, 0);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "main-intra");
+
+  fill_h265_profile (profile_tier_level, 4, 0, 1, 1, 0, 1, 1, 0, 1, 0, 0);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "main-10-intra");
+
+  fill_h265_profile (profile_tier_level, 4, 0, 1, 0, 0, 1, 1, 0, 1, 0, 0);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "main-12-intra");
+
+  fill_h265_profile (profile_tier_level, 4, 0, 1, 1, 0, 1, 0, 0, 1, 0, 0);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "main-422-10-intra");
+
+  fill_h265_profile (profile_tier_level, 4, 0, 1, 0, 0, 1, 0, 0, 1, 0, 0);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "main-422-12-intra");
+
+  fill_h265_profile (profile_tier_level, 4, 0, 1, 1, 1, 0, 0, 0, 1, 0, 0);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "main-444-intra");
+
+  fill_h265_profile (profile_tier_level, 4, 0, 1, 1, 0, 0, 0, 0, 1, 0, 0);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "main-444-10-intra");
+
+  fill_h265_profile (profile_tier_level, 4, 0, 1, 0, 0, 0, 0, 0, 1, 0, 0);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "main-444-12-intra");
+
+  fill_h265_profile (profile_tier_level, 4, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "main-444-16-intra");
+
+  fill_h265_profile (profile_tier_level, 4, 0, 1, 1, 1, 0, 0, 0, 1, 1, 0);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "main-444-still-picture");
+
+  fill_h265_profile (profile_tier_level, 4, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "main-444-16-still-picture");
+
+  /* High Throughput profiles */
+  fill_h265_profile (profile_tier_level, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless (profile == NULL);
+
+  fill_h265_profile (profile_tier_level, 5, 1, 1, 1, 1, 0, 0, 0, 0, 0, 1);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "high-throughput-444");
+
+  fill_h265_profile (profile_tier_level, 5, 1, 1, 1, 0, 0, 0, 0, 0, 0, 1);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "high-throughput-444-10");
+
+  fill_h265_profile (profile_tier_level, 5, 1, 0, 0, 0, 0, 0, 0, 0, 0, 1);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "high-throughput-444-14");
+
+  fill_h265_profile (profile_tier_level, 5, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "high-throughput-444-16-intra");
+
+  /* Multiview Main profile */
+  fill_h265_profile (profile_tier_level, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless (profile == NULL);
+
+  fill_h265_profile (profile_tier_level, 6, 0, 1, 1, 1, 1, 1, 0, 0, 0, 1);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "multiview-main");
+
+  /* Scalable Main profiles */
+  fill_h265_profile (profile_tier_level, 7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless (profile == NULL);
+
+  fill_h265_profile (profile_tier_level, 7, 0, 1, 1, 1, 1, 1, 0, 0, 0, 1);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "scalable-main");
+
+  fill_h265_profile (profile_tier_level, 7, 0, 1, 1, 0, 1, 1, 0, 0, 0, 1);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "scalable-main-10");
+
+  /* 3D Main profile */
+  fill_h265_profile (profile_tier_level, 8, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless (profile == NULL);
+
+  fill_h265_profile (profile_tier_level, 8, 0, 1, 1, 1, 1, 1, 0, 0, 0, 1);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "3d-main");
+
+  /* Screen content coding extensions profiles */
+  fill_h265_profile (profile_tier_level, 9, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless (profile == NULL);
+
+  fill_h265_profile (profile_tier_level, 9, 1, 1, 1, 1, 1, 1, 0, 0, 0, 1);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "screen-extended-main");
+
+  fill_h265_profile (profile_tier_level, 9, 1, 1, 1, 0, 1, 1, 0, 0, 0, 1);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "screen-extended-main-10");
+
+  fill_h265_profile (profile_tier_level, 9, 1, 1, 1, 1, 0, 0, 0, 0, 0, 1);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "screen-extended-main-444");
+
+  fill_h265_profile (profile_tier_level, 9, 1, 1, 1, 0, 0, 0, 0, 0, 0, 1);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "screen-extended-main-444-10");
+
+  fill_h265_profile (profile_tier_level, 9, 1, 0, 0, 0, 0, 0, 0, 0, 0, 1);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "screen-extended-high-throughput-444-14");
+
+  /* Scalable format range extensions profiles */
+  fill_h265_profile (profile_tier_level, 10, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless (profile == NULL);
+
+  fill_h265_profile (profile_tier_level, 10, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "scalable-monochrome");
+
+  fill_h265_profile (profile_tier_level, 10, 1, 1, 0, 0, 1, 1, 1, 0, 0, 1);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "scalable-monochrome-12");
+
+  fill_h265_profile (profile_tier_level, 10, 0, 0, 0, 0, 1, 1, 1, 0, 0, 1);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "scalable-monochrome-16");
+
+  fill_h265_profile (profile_tier_level, 10, 1, 1, 1, 1, 0, 0, 0, 0, 0, 1);
+  profile = gst_codec_utils_h265_get_profile (profile_tier_level,
+      sizeof (profile_tier_level));
+  fail_unless_equals_string (profile, "scalable-main-444");
+}
+
+GST_END_TEST;
+
 static Suite *
 libgstpbutils_suite (void)
 {
@@ -882,6 +1175,7 @@ libgstpbutils_suite (void)
   tcase_add_test (tc_chain, test_pb_utils_installer_details);
   tcase_add_test (tc_chain, test_pb_utils_versions);
   tcase_add_test (tc_chain, test_pb_utils_aac_get_profile);
+  tcase_add_test (tc_chain, test_pb_utils_h265_profiles);
   return s;
 }