FEI: plugin: Add fei specific video meta

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

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

Signed-off-by: Sreerenj Balachandran <sreerenj.balachandran@intel.com>
This commit is contained in:
Yi A Wang 2017-08-09 18:26:57 -07:00 committed by Víctor Manuel Jáquez Leal
parent 7e0d5b934b
commit 0fdef0e349
3 changed files with 304 additions and 0 deletions

View file

@ -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) \

View file

@ -0,0 +1,210 @@
/*
* 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"
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;
}

View file

@ -0,0 +1,80 @@
/*
* 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;
#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;
};
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 */