FEI: Add codec objects for fei usecase

There are 6 new va buffer types, each defined as a specific codec object.
Borrowed the code from gstvaapicodecobject , but made a clear separation
to avoid any possible mess-up. Because unlike the other gstvaaicodecobjects,
feicodecobjects can be shared between elements and also can be accessed
from different thread.

Unlike the other fei codecs object, VAEncMiscParameterTypeFEIFrameControl
object is not shared between elements.So we utilize the already
existing gst_vaapi_enc_misc_param_new(), but still keeping the code
in gstvaapfei_objects_priv.h in order to have a better
code readability.

Fixme:
-- Probably we need _locked_map() and _unlocked_map()
-- Context can be associated with PreEnc(not just Enoder)
once we have the proper support inplace, but for now we don't have
PreEnc support, so should be safe enough to use GstVaapiEncoder.

https://bugzilla.gnome.org/show_bug.cgi?id=785712
https://bugzilla.gnome.org/show_bug.cgi?id=784667

1 files changed, 386 insertions, 0 deletions

diff --git a/gst-libs/gst/vaapi/gstvaapifei_objects.c b/gst-libs/gst/vaapi/gstvaapifei_objects.c
new file mode 100644
index 00000000..fe71dcfb
--- /dev/null
+++ b/gst-libs/gst/vaapi/gstvaapifei_objects.c
@@ -0,0 +1,386 @@
+/*
+ *  gstvaapifei_objects.c - VA FEI objects abstraction
+ *
+ *  Copyright (C) 2017-2018 Intel Corporation
+ *    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
+ */
+
+#include "sysdeps.h"
+#include "gstvaapiencoder_priv.h"
+#include "gstvaapicompat.h"
+#include "gstvaapiutils.h"
+#include "gstvaapifei_objects.h"
+#include "gstvaapifei_objects_priv.h"
+#define DEBUG 1
+#include "gstvaapidebug.h"
+
+/* ------------------------------------------------------------------------- */
+/* --- Base Codec Object                                                 --- */
+/* ------------------------------------------------------------------------- */
+
+#define GST_VAAPI_FEI_CODEC_OBJECT_GET_CLASS(object) \
+    gst_vaapi_fei_codec_object_get_class(object)
+
+const GstVaapiFeiCodecObjectClass *
+gst_vaapi_fei_codec_object_get_class (GstVaapiFeiCodecObject * object)
+{
+  return (const GstVaapiFeiCodecObjectClass *)
+      GST_VAAPI_MINI_OBJECT_GET_CLASS (object);
+}
+
+static gboolean
+gst_vaapi_fei_codec_object_create (GstVaapiFeiCodecObject * object,
+    const GstVaapiFeiCodecObjectConstructorArgs * args)
+{
+  const GstVaapiFeiCodecObjectClass *klass;
+
+  g_return_val_if_fail (args->param_size > 0, FALSE);
+
+  if (GST_VAAPI_MINI_OBJECT_FLAG_IS_SET (object,
+          GST_VAAPI_FEI_CODEC_OBJECT_FLAG_CONSTRUCTED))
+    return TRUE;
+
+  klass = GST_VAAPI_FEI_CODEC_OBJECT_GET_CLASS (object);
+  if (!klass->create || !klass->create (object, args))
+    return FALSE;
+
+  GST_VAAPI_MINI_OBJECT_FLAG_SET (object,
+      GST_VAAPI_FEI_CODEC_OBJECT_FLAG_CONSTRUCTED);
+  return TRUE;
+}
+
+GstVaapiFeiCodecObject *
+gst_vaapi_fei_codec_object_new (const GstVaapiFeiCodecObjectClass *
+    object_class, GstVaapiFeiCodecBase * codec, gconstpointer param,
+    guint param_size, gconstpointer data, guint data_size, guint flags)
+{
+  GstVaapiFeiCodecObject *obj;
+  GstVaapiFeiCodecObjectConstructorArgs args;
+
+  obj = (GstVaapiFeiCodecObject *)
+      gst_vaapi_mini_object_new0 (GST_VAAPI_MINI_OBJECT_CLASS (object_class));
+  if (!obj)
+    return NULL;
+
+  obj = GST_VAAPI_FEI_CODEC_OBJECT (obj);
+  obj->codec = codec;
+
+  args.param = param;
+  args.param_size = param_size;
+  args.data = data;
+  args.data_size = data_size;
+  args.flags = flags;
+
+  if (gst_vaapi_fei_codec_object_create (obj, &args))
+    return obj;
+
+  gst_vaapi_fei_codec_object_unref (obj);
+  return NULL;
+}
+
+GstVaapiFeiCodecObject *
+gst_vaapi_fei_codec_object_ref (GstVaapiFeiCodecObject * object)
+{
+  return ((GstVaapiFeiCodecObject *)
+      gst_vaapi_mini_object_ref (GST_VAAPI_MINI_OBJECT (object)));
+}
+
+void
+gst_vaapi_fei_codec_object_unref (GstVaapiFeiCodecObject * object)
+{
+  gst_vaapi_mini_object_unref (GST_VAAPI_MINI_OBJECT (object));
+}
+
+void
+gst_vaapi_fei_codec_object_replace (GstVaapiFeiCodecObject ** old_object_ptr,
+    GstVaapiFeiCodecObject * new_object)
+{
+  gst_vaapi_mini_object_replace ((GstVaapiMiniObject **) (old_object_ptr),
+      GST_VAAPI_MINI_OBJECT (new_object));
+}
+
+/* FeiFixme: map_unlocked and map_lock could be needed */
+gboolean
+gst_vaapi_fei_codec_object_map (GstVaapiFeiCodecObject * object,
+    gpointer * data, guint * size)
+{
+  g_return_val_if_fail (object != NULL, FALSE);
+
+  /*FeiFixme: explicit map if not yet mapped */
+  *data = object->param;
+  *size = object->param_size;
+
+  return TRUE;
+}
+
+void
+gst_vaapi_fei_codec_object_unmap (GstVaapiFeiCodecObject * object)
+{
+  g_return_if_fail (object != NULL);
+  vaapi_unmap_buffer (GST_VAAPI_ENCODER_CAST (object->codec)->va_display,
+      object->param_id, &object->param);
+}
+
+#define GET_ENCODER(obj)    GST_VAAPI_ENCODER_CAST((obj)->parent_instance.codec)
+#define GET_VA_DISPLAY(obj) GET_ENCODER(obj)->va_display
+#define GET_VA_CONTEXT(obj) GET_ENCODER(obj)->va_context
+
+/* ------------------------------------------------------------------------- */
+/* ---  FEI Mb Code buffer                                               --- */
+/* ------------------------------------------------------------------------- */
+
+GST_VAAPI_FEI_CODEC_DEFINE_TYPE (GstVaapiEncFeiMbCode,
+    gst_vaapi_enc_fei_mb_code);
+
+void
+gst_vaapi_enc_fei_mb_code_destroy (GstVaapiEncFeiMbCode * fei_mb_code)
+{
+  GstVaapiFeiCodecObject *object = &fei_mb_code->parent_instance;
+  vaapi_destroy_buffer (GET_VA_DISPLAY (fei_mb_code), &object->param_id);
+  object->param = NULL;
+}
+
+gboolean
+gst_vaapi_enc_fei_mb_code_create (GstVaapiEncFeiMbCode *
+    fei_mb_code, const GstVaapiFeiCodecObjectConstructorArgs * args)
+{
+  GstVaapiFeiCodecObject *object = &fei_mb_code->parent_instance;
+  object->param_id = VA_INVALID_ID;
+  return vaapi_create_buffer (GET_VA_DISPLAY (fei_mb_code),
+      GET_VA_CONTEXT (fei_mb_code),
+      (VABufferType) VAEncFEIMBCodeBufferType, args->param_size,
+      args->param, &object->param_id, &object->param);
+}
+
+GstVaapiEncFeiMbCode *
+gst_vaapi_enc_fei_mb_code_new (GstVaapiEncoder * encoder,
+    gconstpointer param, guint param_size)
+{
+  GstVaapiFeiCodecObject *object;
+
+  object = gst_vaapi_fei_codec_object_new (&GstVaapiEncFeiMbCodeClass,
+      GST_VAAPI_FEI_CODEC_BASE (encoder), param, param_size, NULL, 0, 0);
+  if (!object)
+    return NULL;
+  object->param_size = param_size;
+  return GST_VAAPI_ENC_FEI_MB_CODE_CAST (object);
+}
+
+/* ------------------------------------------------------------------------- */
+/* ---  FEI MV buffer                                                    --- */
+/* ------------------------------------------------------------------------- */
+
+GST_VAAPI_FEI_CODEC_DEFINE_TYPE (GstVaapiEncFeiMv, gst_vaapi_enc_fei_mv);
+
+void
+gst_vaapi_enc_fei_mv_destroy (GstVaapiEncFeiMv * fei_mv)
+{
+  GstVaapiFeiCodecObject *object = &fei_mv->parent_instance;
+  vaapi_destroy_buffer (GET_VA_DISPLAY (fei_mv), &object->param_id);
+  object->param = NULL;
+}
+
+gboolean
+gst_vaapi_enc_fei_mv_create (GstVaapiEncFeiMv *
+    fei_mv, const GstVaapiFeiCodecObjectConstructorArgs * args)
+{
+  GstVaapiFeiCodecObject *object = &fei_mv->parent_instance;
+  object->param_id = VA_INVALID_ID;
+  return vaapi_create_buffer (GET_VA_DISPLAY (fei_mv),
+      GET_VA_CONTEXT (fei_mv),
+      (VABufferType) VAEncFEIMVBufferType, args->param_size,
+      args->param, &object->param_id, &object->param);
+}
+
+GstVaapiEncFeiMv *
+gst_vaapi_enc_fei_mv_new (GstVaapiEncoder * encoder,
+    gconstpointer param, guint param_size)
+{
+  GstVaapiFeiCodecObject *object;
+
+  object = gst_vaapi_fei_codec_object_new (&GstVaapiEncFeiMvClass,
+      GST_VAAPI_FEI_CODEC_BASE (encoder), param, param_size, NULL, 0, 0);
+  if (!object)
+    return NULL;
+  object->param_size = param_size;
+  return GST_VAAPI_ENC_FEI_NEW_MV_CAST (object);
+}
+
+/* ------------------------------------------------------------------------- */
+/* ---  FEI Mv predictor buffer                                          --- */
+/* ------------------------------------------------------------------------- */
+
+GST_VAAPI_FEI_CODEC_DEFINE_TYPE (GstVaapiEncFeiMvPredictor,
+    gst_vaapi_enc_fei_mv_predictor);
+
+void
+gst_vaapi_enc_fei_mv_predictor_destroy (GstVaapiEncFeiMvPredictor *
+    fei_mv_predictor)
+{
+  GstVaapiFeiCodecObject *object = &fei_mv_predictor->parent_instance;
+  vaapi_destroy_buffer (GET_VA_DISPLAY (fei_mv_predictor), &object->param_id);
+  object->param = NULL;
+}
+
+gboolean
+gst_vaapi_enc_fei_mv_predictor_create (GstVaapiEncFeiMvPredictor *
+    fei_mv_predictor, const GstVaapiFeiCodecObjectConstructorArgs * args)
+{
+  GstVaapiFeiCodecObject *object = &fei_mv_predictor->parent_instance;
+  object->param_id = VA_INVALID_ID;
+  return vaapi_create_buffer (GET_VA_DISPLAY (fei_mv_predictor),
+      GET_VA_CONTEXT (fei_mv_predictor),
+      (VABufferType) VAEncFEIMVPredictorBufferType, args->param_size,
+      args->param, &object->param_id, &object->param);
+}
+
+GstVaapiEncFeiMvPredictor *
+gst_vaapi_enc_fei_mv_predictor_new (GstVaapiEncoder * encoder,
+    gconstpointer param, guint param_size)
+{
+  GstVaapiFeiCodecObject *object;
+
+  object = gst_vaapi_fei_codec_object_new (&GstVaapiEncFeiMvPredictorClass,
+      GST_VAAPI_FEI_CODEC_BASE (encoder), param, param_size, NULL, 0, 0);
+  if (!object)
+    return NULL;
+  object->param_size = param_size;
+  return GST_VAAPI_ENC_FEI_NEW_MV_PREDICTOR_CAST (object);
+}
+
+/* ------------------------------------------------------------------------- */
+/* ---  FEI Mb Control  buffer                                           --- */
+/* ------------------------------------------------------------------------- */
+
+GST_VAAPI_FEI_CODEC_DEFINE_TYPE (GstVaapiEncFeiMbControl,
+    gst_vaapi_enc_fei_mb_control);
+
+void
+gst_vaapi_enc_fei_mb_control_destroy (GstVaapiEncFeiMbControl * fei_mb_control)
+{
+  GstVaapiFeiCodecObject *object = &fei_mb_control->parent_instance;
+  vaapi_destroy_buffer (GET_VA_DISPLAY (fei_mb_control), &object->param_id);
+  object->param = NULL;
+}
+
+gboolean
+gst_vaapi_enc_fei_mb_control_create (GstVaapiEncFeiMbControl *
+    fei_mb_control, const GstVaapiFeiCodecObjectConstructorArgs * args)
+{
+  GstVaapiFeiCodecObject *object = &fei_mb_control->parent_instance;
+  object->param_id = VA_INVALID_ID;
+  return vaapi_create_buffer (GET_VA_DISPLAY (fei_mb_control),
+      GET_VA_CONTEXT (fei_mb_control),
+      (VABufferType) VAEncFEIMBControlBufferType, args->param_size,
+      args->param, &object->param_id, &object->param);
+}
+
+GstVaapiEncFeiMbControl *
+gst_vaapi_enc_fei_mb_control_new (GstVaapiEncoder * encoder,
+    gconstpointer param, guint param_size)
+{
+  GstVaapiFeiCodecObject *object;
+
+  object = gst_vaapi_fei_codec_object_new (&GstVaapiEncFeiMbControlClass,
+      GST_VAAPI_FEI_CODEC_BASE (encoder), param, param_size, NULL, 0, 0);
+  if (!object)
+    return NULL;
+  object->param_size = param_size;
+  return GST_VAAPI_ENC_FEI_NEW_MB_CONTROL_CAST (object);
+}
+
+/* ------------------------------------------------------------------------- */
+/* ---  FEI qp buffer                                                    --- */
+/* ------------------------------------------------------------------------- */
+
+GST_VAAPI_FEI_CODEC_DEFINE_TYPE (GstVaapiEncFeiQp, gst_vaapi_enc_fei_qp);
+
+void
+gst_vaapi_enc_fei_qp_destroy (GstVaapiEncFeiQp * fei_qp)
+{
+  GstVaapiFeiCodecObject *object = &fei_qp->parent_instance;
+  vaapi_destroy_buffer (GET_VA_DISPLAY (fei_qp), &object->param_id);
+  object->param = NULL;
+}
+
+gboolean
+gst_vaapi_enc_fei_qp_create (GstVaapiEncFeiQp * fei_qp,
+    const GstVaapiFeiCodecObjectConstructorArgs * args)
+{
+  GstVaapiFeiCodecObject *object = &fei_qp->parent_instance;
+  object->param_id = VA_INVALID_ID;
+  return vaapi_create_buffer (GET_VA_DISPLAY (fei_qp),
+      GET_VA_CONTEXT (fei_qp),
+      (VABufferType) VAEncQPBufferType, args->param_size,
+      args->param, &object->param_id, &object->param);
+}
+
+GstVaapiEncFeiQp *
+gst_vaapi_enc_fei_qp_new (GstVaapiEncoder * encoder,
+    gconstpointer param, guint param_size)
+{
+  GstVaapiFeiCodecObject *object;
+
+  object = gst_vaapi_fei_codec_object_new (&GstVaapiEncFeiQpClass,
+      GST_VAAPI_FEI_CODEC_BASE (encoder), param, param_size, NULL, 0, 0);
+  if (!object)
+    return NULL;
+  object->param_size = param_size;
+  return GST_VAAPI_ENC_FEI_NEW_QP_CAST (object);
+}
+
+/* ------------------------------------------------------------------------- */
+/* ---  FEI Distortion buffer                                            --- */
+/* ------------------------------------------------------------------------- */
+
+GST_VAAPI_FEI_CODEC_DEFINE_TYPE (GstVaapiEncFeiDistortion,
+    gst_vaapi_enc_fei_distortion);
+
+void
+gst_vaapi_enc_fei_distortion_destroy (GstVaapiEncFeiDistortion * fei_dist)
+{
+  GstVaapiFeiCodecObject *object = &fei_dist->parent_instance;
+  vaapi_destroy_buffer (GET_VA_DISPLAY (fei_dist), &object->param_id);
+  object->param = NULL;
+}
+
+gboolean
+gst_vaapi_enc_fei_distortion_create (GstVaapiEncFeiDistortion * fei_dist,
+    const GstVaapiFeiCodecObjectConstructorArgs * args)
+{
+  GstVaapiFeiCodecObject *object = &fei_dist->parent_instance;
+  object->param_id = VA_INVALID_ID;
+  return vaapi_create_buffer (GET_VA_DISPLAY (fei_dist),
+      GET_VA_CONTEXT (fei_dist),
+      (VABufferType) VAEncFEIDistortionBufferType, args->param_size,
+      args->param, &object->param_id, &object->param);
+}
+
+GstVaapiEncFeiDistortion *
+gst_vaapi_enc_fei_distortion_new (GstVaapiEncoder * encoder,
+    gconstpointer param, guint param_size)
+{
+  GstVaapiFeiCodecObject *object;
+
+  object = gst_vaapi_fei_codec_object_new (&GstVaapiEncFeiDistortionClass,
+      GST_VAAPI_FEI_CODEC_BASE (encoder), param, param_size, NULL, 0, 0);
+  if (!object)
+    return NULL;
+  object->param_size = param_size;
+  return GST_VAAPI_ENC_FEI_NEW_DISTORTION_CAST (object);
+}