mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-12 03:16:33 +00:00
387 lines
13 KiB
C
387 lines
13 KiB
C
|
/*
|
||
|
* 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);
|
||
|
}
|