From b45147a6d4876bb076a35489b60987570ef8fc5e Mon Sep 17 00:00:00 2001
From: Hyunjun Ko <zzoon@igalia.com>
Date: Tue, 13 Feb 2018 12:37:47 -0900
Subject: msdk: adds new utility functions for conversion from gstreamer to
 libmfx

https://bugzilla.gnome.org/show_bug.cgi?id=790752
---
 sys/msdk/msdk.c | 71 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 sys/msdk/msdk.h |  6 +++++
 2 files changed, 77 insertions(+)

diff --git a/sys/msdk/msdk.c b/sys/msdk/msdk.c
index bfc66e056..2c621b2b1 100644
--- a/sys/msdk/msdk.c
+++ b/sys/msdk/msdk.c
@@ -37,6 +37,26 @@ GST_DEBUG_CATEGORY_EXTERN (gst_msdkenc_debug);
 #define INVALID_INDEX         ((guint) -1)
 #define GST_MSDK_ALIGNMENT_PADDING(num) (32 - ((num) & 31))
 
+struct map
+{
+  GstVideoFormat format;
+  mfxU16 mfx_chroma_format;
+  mfxU32 mfx_fourcc;
+};
+
+#define GST_VIDEO_INFO_TO_MFX_MAP(FORMAT, CHROMA, FOURCC) \
+    { GST_VIDEO_FORMAT_##FORMAT, MFX_CHROMAFORMAT_##CHROMA, MFX_FOURCC_##FOURCC }
+
+static const struct map gst_msdk_video_format_to_mfx_map[] = {
+  GST_VIDEO_INFO_TO_MFX_MAP (NV12, YUV420, NV12),
+  GST_VIDEO_INFO_TO_MFX_MAP (YV12, YUV420, YV12),
+  GST_VIDEO_INFO_TO_MFX_MAP (I420, YUV420, YV12),
+  GST_VIDEO_INFO_TO_MFX_MAP (YUY2, YUV422, YUY2),
+  GST_VIDEO_INFO_TO_MFX_MAP (UYVY, YUV422, UYVY),
+  GST_VIDEO_INFO_TO_MFX_MAP (BGRA, YUV444, RGB4),
+  {0, 0, 0}
+};
+
 static inline guint
 msdk_get_free_surface_index (mfxFrameSurface1 * surfaces, guint size)
 {
@@ -382,3 +402,54 @@ gst_msdk_set_video_alignment (GstVideoInfo * info,
   if (height & 31)
     alignment->padding_bottom = GST_MSDK_ALIGNMENT_PADDING (height);
 }
+
+static const struct map *
+_map_lookup_format (GstVideoFormat format)
+{
+  const struct map *m = gst_msdk_video_format_to_mfx_map;
+
+  for (; m->format != 0; m++) {
+    if (m->format == format)
+      return m;
+  }
+  return NULL;
+}
+
+gint
+gst_msdk_get_mfx_chroma_from_format (GstVideoFormat format)
+{
+  const struct map *const m = _map_lookup_format (format);
+
+  return m ? m->mfx_chroma_format : -1;
+}
+
+gint
+gst_msdk_get_mfx_fourcc_from_format (GstVideoFormat format)
+{
+  const struct map *const m = _map_lookup_format (format);
+
+  return m ? m->mfx_fourcc : -1;
+}
+
+void
+gst_msdk_set_mfx_frame_info_from_video_info (mfxFrameInfo * mfx_info,
+    GstVideoInfo * info)
+{
+  g_return_if_fail (info && mfx_info);
+
+  mfx_info->Width = GST_ROUND_UP_32 (GST_VIDEO_INFO_WIDTH (info));
+  mfx_info->Height = GST_ROUND_UP_32 (GST_VIDEO_INFO_HEIGHT (info));
+  mfx_info->CropW = GST_VIDEO_INFO_WIDTH (info);
+  mfx_info->CropH = GST_VIDEO_INFO_HEIGHT (info);
+  mfx_info->FrameRateExtN = GST_VIDEO_INFO_FPS_N (info);
+  mfx_info->FrameRateExtD = GST_VIDEO_INFO_FPS_D (info);
+  mfx_info->AspectRatioW = GST_VIDEO_INFO_PAR_N (info);
+  mfx_info->AspectRatioH = GST_VIDEO_INFO_PAR_D (info);
+  mfx_info->PicStruct = MFX_PICSTRUCT_PROGRESSIVE;      /* this is by default */
+  mfx_info->FourCC =
+      gst_msdk_get_mfx_fourcc_from_format (GST_VIDEO_INFO_FORMAT (info));
+  mfx_info->ChromaFormat =
+      gst_msdk_get_mfx_chroma_from_format (GST_VIDEO_INFO_FORMAT (info));
+
+  return;
+}
diff --git a/sys/msdk/msdk.h b/sys/msdk/msdk.h
index 13db9fff8..d2e560211 100644
--- a/sys/msdk/msdk.h
+++ b/sys/msdk/msdk.h
@@ -69,6 +69,12 @@ const gchar *msdk_status_to_string (mfxStatus status);
 void gst_msdk_set_video_alignment (GstVideoInfo * info,
     GstVideoAlignment * alignment);
 
+/* Conversion from Gstreamer to libmfx */
+gint gst_msdk_get_mfx_chroma_from_format (GstVideoFormat format);
+gint gst_msdk_get_mfx_fourcc_from_format (GstVideoFormat format);
+void gst_msdk_set_mfx_frame_info_from_video_info (mfxFrameInfo * mfx_info,
+    GstVideoInfo * info);
+
 G_END_DECLS
 
 #endif /* __MSDK_H__ */
-- 
cgit v1.2.1