FEI: plugin: Add fei video meta

GstVaapiFeiVideoMeta holds the fei codec objects:
  GstVaapiEncFeiMbCode,
  GstVaapiEncFeiMv,
  GstVaapiEncFeiMvPredictor,
  GstVaapiEncFeiMbControl,
  GstVaapiEncFeiQp,
  GstVaapiEncFeiDistortion

Signed-off-by: Sreerenj Balachandran <sreerenj.balachandran@intel.com>

3 files changed, 304 insertions, 0 deletions

diff --git a/gst/vaapi/Makefile.am b/gst/vaapi/Makefile.am
index 1f29516e..cfdc2f42 100644
--- a/gst/vaapi/Makefile.am
+++ b/gst/vaapi/Makefile.am
@@ -122,6 +122,18 @@ libgstvaapi_source_c += $(libgstvaapi_vp9enc_source_c)
 libgstvaapi_source_h += $(libgstvaapi_vp9enc_source_h)
 endif
 
+libgstvaapi_h264feienc_source_c = 	   \
+		 gstvaapifeivideometa.c    \
+		 $(NULL)
+libgstvaapi_h264feienc_source_h =	   \
+		 gstvaapifeivideometa.h    \
+                 $(NULL)
+
+if USE_H264_FEI_ENCODER
+libgstvaapi_source_c += $(libgstvaapi_h264feienc_source_c)
+libgstvaapi_source_h += $(libgstvaapi_h264feienc_source_h)
+endif
+
 libgstvaapi_la_SOURCES		= $(libgstvaapi_source_c)
 noinst_HEADERS			= $(libgstvaapi_source_h)
 
@@ -161,6 +173,8 @@ EXTRA_DIST = \
 	$(libgstvaapi_h265enc_source_h)  \
 	$(libgstvaapi_vp9enc_source_c)   \
 	$(libgstvaapi_vp9enc_source_h)   \
+	$(libgstvaapi_h264feienc_source_c) \
+	$(libgstvaapi_h264feienc_source_h) \
 	$(libgstvaapi_egl_source_c)	\
 	$(libgstvaapi_egl_source_h)	\
 	$(libgstvaapi_1_2p_source_c)	 \
diff --git a/gst/vaapi/gstvaapifeivideometa.c b/gst/vaapi/gstvaapifeivideometa.c
new file mode 100644
index 00000000..5d12f083
--- /dev/null
+++ b/gst/vaapi/gstvaapifeivideometa.c
@@ -0,0 +1,221 @@
+/*
+ *  gstvaapifeivideometa.c - Gst VA FEI video meta
+ *
+ *  Copyright (C) 2016-2017 Intel Corporation
+ *    Author: Yi A Wang <yi.a.wang@intel.com>
+ *    Author: Sreerenj Balachandran <sreerenj.balachandran@intel.com>*
+ *
+ *  This library is free software; you can redistribute it and/or
+ *  modify it under the terms of the GNU Lesser General Public License
+ *  as published by the Free Software Foundation; either version 2.1
+ *  of the License, or (at your option) any later version.
+ *
+ *  This library is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ *  Lesser General Public License for more details.
+ *
+ *  You should have received a copy of the GNU Lesser General Public
+ *  License along with this library; if not, write to the Free
+ *  Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
+ *  Boston, MA 02110-1301 USA
+ */
+
+/**
+ * SECTION:gstvaapifeivideometa
+ * @short_description: VA FEI video meta for GStreamer
+ */
+#include "gstcompat.h"
+#include "gstvaapifeivideometa.h"
+
+#define GST_VAAPI_FEI_VIDEO_META(obj) \
+  ((GstVaapiFeiVideoMeta *) (obj))
+#define GST_VAAPI_IS_FEI_VIDEO_META(obj) \
+  (GST_VAAPI_FEI_VIDEO_META (obj) != NULL)
+
+struct _GstVaapiFeiVideoMetaHolder
+{
+  GstMeta base;
+  GstVaapiFeiVideoMeta *meta;
+};
+
+static void
+gst_vaapi_fei_video_meta_finalize (GstVaapiFeiVideoMeta * meta)
+{
+  if (meta->mbcode)
+    gst_vaapi_fei_codec_object_unref (GST_VAAPI_FEI_CODEC_OBJECT
+        (meta->mbcode));
+  if (meta->mv)
+    gst_vaapi_fei_codec_object_unref (GST_VAAPI_FEI_CODEC_OBJECT (meta->mv));
+  if (meta->mvpred)
+    gst_vaapi_fei_codec_object_unref (GST_VAAPI_FEI_CODEC_OBJECT
+        (meta->mvpred));
+  if (meta->mbcntrl)
+    gst_vaapi_fei_codec_object_unref (GST_VAAPI_FEI_CODEC_OBJECT
+        (meta->mbcntrl));
+  if (meta->qp)
+    gst_vaapi_fei_codec_object_unref (GST_VAAPI_FEI_CODEC_OBJECT (meta->qp));
+  if (meta->dist)
+    gst_vaapi_fei_codec_object_unref (GST_VAAPI_FEI_CODEC_OBJECT (meta->dist));
+}
+
+static void
+gst_vaapi_fei_video_meta_init (GstVaapiFeiVideoMeta * meta)
+{
+}
+
+static inline GstVaapiFeiVideoMeta *
+_gst_vaapi_fei_video_meta_create (void)
+{
+  return g_slice_new0 (GstVaapiFeiVideoMeta);
+}
+
+static inline void
+_gst_vaapi_fei_video_meta_destroy (GstVaapiFeiVideoMeta * meta)
+{
+  g_slice_free1 (sizeof (*meta), meta);
+}
+
+GstVaapiFeiVideoMeta *
+gst_vaapi_fei_video_meta_new (void)
+{
+  GstVaapiFeiVideoMeta *meta;
+
+  meta = _gst_vaapi_fei_video_meta_create ();
+  if (!meta)
+    return NULL;
+  gst_vaapi_fei_video_meta_init (meta);
+  return meta;
+}
+
+static inline void
+_gst_vaapi_fei_video_meta_free (GstVaapiFeiVideoMeta * meta)
+{
+  g_atomic_int_inc (&meta->ref_count);
+
+  gst_vaapi_fei_video_meta_finalize (meta);
+
+  if (G_LIKELY (g_atomic_int_dec_and_test (&meta->ref_count)))
+    _gst_vaapi_fei_video_meta_destroy (meta);
+}
+
+/**
+ * gst_vaapi_fei_video_meta_ref:
+ * @meta: a #GstVaapiFeiVideoMeta
+ *
+ * Atomically increases the reference count of the given @meta by one.
+ *
+ * Returns: The same @meta argument
+ */
+GstVaapiFeiVideoMeta *
+gst_vaapi_fei_video_meta_ref (GstVaapiFeiVideoMeta * meta)
+{
+  g_return_val_if_fail (meta != NULL, NULL);
+
+  g_atomic_int_inc (&meta->ref_count);
+  return meta;
+}
+
+/**
+ * gst_vaapi_fei_video_meta_unref:
+ * @meta: a #GstVaapiFeiVideoMeta
+ *
+ * Atomically decreases the reference count of the @meta by one. If
+ * the reference count reaches zero, the object will be free'd.
+ */
+void
+gst_vaapi_fei_video_meta_unref (GstVaapiFeiVideoMeta * meta)
+{
+  g_return_if_fail (meta != NULL);
+  g_return_if_fail (meta->ref_count > 0);
+  if (g_atomic_int_dec_and_test (&meta->ref_count))
+    _gst_vaapi_fei_video_meta_free (meta);
+}
+
+
+GType
+gst_vaapi_fei_video_meta_api_get_type (void)
+{
+  static gsize g_type;
+  static const gchar *tags[] = { "memory", NULL };
+
+  if (g_once_init_enter (&g_type)) {
+    GType type = gst_meta_api_type_register ("GstVaapiFeiVideoMetaAPI", tags);
+    g_once_init_leave (&g_type, type);
+  }
+  return g_type;
+}
+
+
+#define GST_VAAPI_FEI_VIDEO_META_HOLDER(meta) \
+  ((GstVaapiFeiVideoMetaHolder *) (meta))
+
+static gboolean
+gst_vaapi_fei_video_meta_holder_init (GstVaapiFeiVideoMetaHolder * meta,
+    gpointer params, GstBuffer * buffer)
+{
+  meta->meta = NULL;
+  return TRUE;
+}
+
+static void
+gst_vaapi_fei_video_meta_holder_free (GstVaapiFeiVideoMetaHolder * meta,
+    GstBuffer * buffer)
+{
+  if (meta->meta)
+    gst_vaapi_fei_video_meta_unref (meta->meta);
+}
+
+
+#define GST_VAAPI_FEI_VIDEO_META_INFO gst_vaapi_fei_video_meta_info_get ()
+static const GstMetaInfo *
+gst_vaapi_fei_video_meta_info_get (void)
+{
+  static gsize g_meta_info;
+
+  if (g_once_init_enter (&g_meta_info)) {
+    gsize meta_info =
+        GPOINTER_TO_SIZE (gst_meta_register (GST_VAAPI_FEI_VIDEO_META_API_TYPE,
+            "GstVaapiFeiVideoMeta", sizeof (GstVaapiFeiVideoMetaHolder),
+            (GstMetaInitFunction) gst_vaapi_fei_video_meta_holder_init,
+            (GstMetaFreeFunction) gst_vaapi_fei_video_meta_holder_free,
+            NULL));
+    g_once_init_leave (&g_meta_info, meta_info);
+  }
+  return GSIZE_TO_POINTER (g_meta_info);
+}
+
+GstVaapiFeiVideoMeta *
+gst_buffer_get_vaapi_fei_video_meta (GstBuffer * buffer)
+{
+  GstVaapiFeiVideoMeta *meta;
+  GstMeta *m;
+
+  g_return_val_if_fail (GST_IS_BUFFER (buffer), NULL);
+
+  m = gst_buffer_get_meta (buffer, GST_VAAPI_FEI_VIDEO_META_API_TYPE);
+  if (!m)
+    return NULL;
+
+  meta = GST_VAAPI_FEI_VIDEO_META_HOLDER (m)->meta;
+  if (meta)
+    meta->buffer = buffer;
+  return meta;
+}
+
+void
+gst_buffer_set_vaapi_fei_video_meta (GstBuffer * buffer,
+    GstVaapiFeiVideoMeta * meta)
+{
+  GstMeta *m = NULL;
+
+  g_return_if_fail (GST_IS_BUFFER (buffer));
+  g_return_if_fail (GST_VAAPI_IS_FEI_VIDEO_META (meta));
+
+  m = gst_buffer_add_meta (buffer, GST_VAAPI_FEI_VIDEO_META_INFO, NULL);
+
+  if (m)
+    GST_VAAPI_FEI_VIDEO_META_HOLDER (m)->meta =
+        gst_vaapi_fei_video_meta_ref (meta);
+  return;
+}
diff --git a/gst/vaapi/gstvaapifeivideometa.h b/gst/vaapi/gstvaapifeivideometa.h
new file mode 100644
index 00000000..d26f9c22
--- /dev/null
+++ b/gst/vaapi/gstvaapifeivideometa.h
@@ -0,0 +1,69 @@
+/*
+ *  gstvaapifeivideometa.h - Gstreamer/VA video meta
+ *
+ *  Copyright (C) 2016-2017 Intel Corporation
+ *    Author: Yi A Wang <yi.a.wang@intel.com>
+ *    Author: Sreerenj Balachandran <sreerenj.balachandran@intel.com>
+ *
+ *  This library is free software; you can redistribute it and/or
+ *  modify it under the terms of the GNU Lesser General Public License
+ *  as published by the Free Software Foundation; either version 2.1
+ *  of the License, or (at your option) any later version.
+ *
+ *  This library is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ *  Lesser General Public License for more details.
+ *
+ *  You should have received a copy of the GNU Lesser General Public
+ *  License along with this library; if not, write to the Free
+ *  Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
+ *  Boston, MA 02110-1301 USA
+ */
+
+#ifndef GST_VAAPI_FEI_VIDEO_META_H
+#define GST_VAAPI_FEI_VIDEO_META_H
+
+#include <va/va.h>
+#include <gst/vaapi/gstvaapifei_objects.h>
+G_BEGIN_DECLS
+
+typedef struct _GstVaapiFeiVideoMeta GstVaapiFeiVideoMeta;
+typedef struct _GstVaapiFeiVideoMetaHolder GstVaapiFeiVideoMetaHolder;
+
+struct _GstVaapiFeiVideoMeta {
+  GstVaapiEncFeiMbCode *mbcode;
+  GstVaapiEncFeiMv *mv;
+  GstVaapiEncFeiMvPredictor *mvpred;
+  GstVaapiEncFeiMbControl *mbcntrl;
+  GstVaapiEncFeiQp *qp;
+  GstVaapiEncFeiDistortion *dist;
+
+  GstBuffer *buffer;
+  gint ref_count;
+};
+
+#define GST_VAAPI_FEI_VIDEO_META_API_TYPE \
+  gst_vaapi_fei_video_meta_api_get_type ()
+
+GType
+gst_vaapi_fei_video_meta_api_get_type (void) G_GNUC_CONST;
+
+GstVaapiFeiVideoMeta *
+gst_vaapi_fei_video_meta_new (void);
+
+GstVaapiFeiVideoMeta *
+gst_vaapi_fei_video_meta_ref (GstVaapiFeiVideoMeta * meta);
+
+void
+gst_vaapi_fei_video_meta_unref (GstVaapiFeiVideoMeta * meta);
+
+GstVaapiFeiVideoMeta *
+gst_buffer_get_vaapi_fei_video_meta (GstBuffer * buffer);
+
+void
+gst_buffer_set_vaapi_fei_video_meta (GstBuffer * buffer, GstVaapiFeiVideoMeta * meta);
+
+G_END_DECLS
+
+#endif /* GST_VAAPI_FEI_VIDEO_META_H */