From 6c57d06ee1905cae936eabbfa0b32faf104f0c39 Mon Sep 17 00:00:00 2001
From: Guillaume Desmottes <guillaume.desmottes@collabora.co.uk>
Date: Tue, 7 Nov 2017 15:09:35 +0100
Subject: [PATCH] omxvideoenc: expose chroma format and bit depth in output
 caps

As we added in the parser (bgo#792039) expose the chroma and bit
depth information in output caps.

https://bugzilla.gnome.org/show_bug.cgi?id=792040
---
 omx/gstomxvideoenc.c | 56 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 56 insertions(+)

diff --git a/omx/gstomxvideoenc.c b/omx/gstomxvideoenc.c
index 80f9562c28..035eb89b76 100644
--- a/omx/gstomxvideoenc.c
+++ b/omx/gstomxvideoenc.c
@@ -549,14 +549,70 @@ gst_omx_video_enc_change_state (GstElement * element, GstStateChange transition)
   return ret;
 }
 
+static gboolean
+get_chroma_info_from_input (GstOMXVideoEnc * self, const gchar ** chroma_format,
+    guint * bit_depth_luma, guint * bit_depth_chroma)
+{
+  switch (self->input_state->info.finfo->format) {
+    case GST_VIDEO_FORMAT_GRAY8:
+      *chroma_format = "4:0:0";
+      *bit_depth_luma = 8;
+      *bit_depth_chroma = 0;
+      break;
+    case GST_VIDEO_FORMAT_I420:
+    case GST_VIDEO_FORMAT_NV12:
+      *chroma_format = "4:2:0";
+      *bit_depth_luma = *bit_depth_chroma = 8;
+      break;
+    case GST_VIDEO_FORMAT_NV16:
+    case GST_VIDEO_FORMAT_YUY2:
+    case GST_VIDEO_FORMAT_YVYU:
+    case GST_VIDEO_FORMAT_UYVY:
+      *chroma_format = "4:2:2";
+      *bit_depth_luma = *bit_depth_chroma = 8;
+      break;
+    case GST_VIDEO_FORMAT_GRAY10_LE32:
+      *chroma_format = "4:0:0";
+      *bit_depth_luma = 10;
+      *bit_depth_chroma = 0;
+      break;
+    case GST_VIDEO_FORMAT_NV12_10LE32:
+      *chroma_format = "4:2:0";
+      *bit_depth_luma = *bit_depth_chroma = 10;
+      break;
+    case GST_VIDEO_FORMAT_NV16_10LE32:
+      *chroma_format = "4:2:2";
+      *bit_depth_luma = *bit_depth_chroma = 10;
+      break;
+    default:
+      return FALSE;
+  }
+
+  return TRUE;
+}
+
 static GstCaps *
 get_output_caps (GstOMXVideoEnc * self)
 {
   GstOMXVideoEncClass *klass = GST_OMX_VIDEO_ENC_GET_CLASS (self);
   GstCaps *caps;
+  const gchar *chroma_format;
+  guint bit_depth_luma, bit_depth_chroma;
 
   caps = klass->get_caps (self, self->enc_out_port, self->input_state);
 
+  /* Add chroma info about the encoded stream inferred from the format of the input */
+  if (get_chroma_info_from_input (self, &chroma_format, &bit_depth_luma,
+          &bit_depth_chroma)) {
+    GST_DEBUG_OBJECT (self,
+        "adding chroma info to output caps: %s (luma %d bits) (chroma %d bits)",
+        chroma_format, bit_depth_luma, bit_depth_chroma);
+
+    gst_caps_set_simple (caps, "chroma-format", G_TYPE_STRING, chroma_format,
+        "bit-depth-luma", G_TYPE_UINT, bit_depth_luma,
+        "bit-depth-chroma", G_TYPE_UINT, bit_depth_chroma, NULL);
+  }
+
   return caps;
 }