mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-18 06:16:36 +00:00
a119a940e4
The AV1 codec needs to support the film grain feature. When the film grain feature is enabled, we need two surfaces as the output of the decoded picture, one without film grain effect and the other one with it. The first one acts as the reference and is needed for later pictures' reconstruction, and the second one is the real display output. So we need to attach another aux surface to the gst buffer/mem and make that aux surface as the target of vaBeginPicture. Part-of: <https://gitlab.freedesktop.org/gstreamer/gst-plugins-bad/-/merge_requests/1636>
780 lines
21 KiB
C
780 lines
21 KiB
C
/* GStreamer
|
|
* Copyright (C) 2020 Igalia, S.L.
|
|
* Author: Víctor Jáquez <vjaquez@igalia.com>
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Library General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2 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
|
|
* Library General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Library General Public
|
|
* License along with this library; if not, write to the
|
|
* Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
|
|
* Boston, MA 02110-1301, USA.
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include "config.h"
|
|
#endif
|
|
|
|
#include "gstvadecoder.h"
|
|
|
|
#include "gstvaallocator.h"
|
|
#include "gstvacaps.h"
|
|
#include "gstvadisplay_wrapped.h"
|
|
#include "gstvavideoformat.h"
|
|
|
|
struct _GstVaDecoder
|
|
{
|
|
GstObject parent;
|
|
|
|
GArray *available_profiles;
|
|
GstCaps *srcpad_caps;
|
|
GstCaps *sinkpad_caps;
|
|
GstVaDisplay *display;
|
|
VAConfigID config;
|
|
VAContextID context;
|
|
VAProfile profile;
|
|
guint rt_format;
|
|
gint coded_width;
|
|
gint coded_height;
|
|
};
|
|
|
|
GST_DEBUG_CATEGORY_STATIC (gst_va_decoder_debug);
|
|
#define GST_CAT_DEFAULT gst_va_decoder_debug
|
|
|
|
#define gst_va_decoder_parent_class parent_class
|
|
G_DEFINE_TYPE_WITH_CODE (GstVaDecoder, gst_va_decoder, GST_TYPE_OBJECT,
|
|
GST_DEBUG_CATEGORY_INIT (gst_va_decoder_debug, "vadecoder", 0,
|
|
"VA Decoder"));
|
|
|
|
enum
|
|
{
|
|
PROP_DISPLAY = 1,
|
|
PROP_PROFILE,
|
|
PROP_WIDTH,
|
|
PROP_HEIGHT,
|
|
PROP_CHROMA,
|
|
N_PROPERTIES
|
|
};
|
|
|
|
static GParamSpec *g_properties[N_PROPERTIES];
|
|
|
|
static gboolean _destroy_buffers (GstVaDecodePicture * pic);
|
|
|
|
static void
|
|
gst_va_decoder_set_property (GObject * object, guint prop_id,
|
|
const GValue * value, GParamSpec * pspec)
|
|
{
|
|
GstVaDecoder *self = GST_VA_DECODER (object);
|
|
|
|
switch (prop_id) {
|
|
case PROP_DISPLAY:{
|
|
g_assert (!self->display);
|
|
self->display = g_value_dup_object (value);
|
|
break;
|
|
}
|
|
default:
|
|
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
|
break;
|
|
}
|
|
}
|
|
|
|
static void
|
|
gst_va_decoder_get_property (GObject * object, guint prop_id, GValue * value,
|
|
GParamSpec * pspec)
|
|
{
|
|
GstVaDecoder *self = GST_VA_DECODER (object);
|
|
|
|
switch (prop_id) {
|
|
case PROP_DISPLAY:
|
|
g_value_set_object (value, self->display);
|
|
break;
|
|
case PROP_PROFILE:
|
|
g_value_set_int (value, self->profile);
|
|
break;
|
|
case PROP_CHROMA:
|
|
g_value_set_uint (value, self->rt_format);
|
|
break;
|
|
case PROP_WIDTH:
|
|
g_value_set_int (value, self->coded_width);
|
|
break;
|
|
case PROP_HEIGHT:
|
|
g_value_set_int (value, self->coded_height);
|
|
break;
|
|
default:
|
|
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
|
break;
|
|
}
|
|
}
|
|
|
|
static void
|
|
gst_va_decoder_dispose (GObject * object)
|
|
{
|
|
GstVaDecoder *self = GST_VA_DECODER (object);
|
|
|
|
gst_va_decoder_close (self);
|
|
|
|
g_clear_pointer (&self->available_profiles, g_array_unref);
|
|
gst_clear_object (&self->display);
|
|
|
|
G_OBJECT_CLASS (parent_class)->dispose (object);
|
|
}
|
|
|
|
static void
|
|
gst_va_decoder_class_init (GstVaDecoderClass * klass)
|
|
{
|
|
GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
|
|
|
|
gobject_class->set_property = gst_va_decoder_set_property;
|
|
gobject_class->get_property = gst_va_decoder_get_property;
|
|
gobject_class->dispose = gst_va_decoder_dispose;
|
|
|
|
g_properties[PROP_DISPLAY] =
|
|
g_param_spec_object ("display", "GstVaDisplay", "GstVADisplay object",
|
|
GST_TYPE_VA_DISPLAY,
|
|
G_PARAM_READWRITE | G_PARAM_CONSTRUCT_ONLY | G_PARAM_STATIC_STRINGS);
|
|
|
|
g_properties[PROP_PROFILE] =
|
|
g_param_spec_int ("va-profile", "VAProfile", "VA Profile",
|
|
VAProfileNone, 50, VAProfileNone,
|
|
G_PARAM_READABLE | G_PARAM_STATIC_STRINGS);
|
|
|
|
g_properties[PROP_CHROMA] =
|
|
g_param_spec_uint ("va-rt-format", "VARTFormat", "VA RT Fromat or chroma",
|
|
VA_RT_FORMAT_YUV420, VA_RT_FORMAT_PROTECTED, VA_RT_FORMAT_YUV420,
|
|
G_PARAM_READABLE | G_PARAM_STATIC_STRINGS);
|
|
|
|
g_properties[PROP_WIDTH] =
|
|
g_param_spec_int ("coded-width", "coded-picture-width",
|
|
"coded picture width", 0, G_MAXINT, 0,
|
|
G_PARAM_READABLE | G_PARAM_STATIC_STRINGS);
|
|
|
|
g_properties[PROP_HEIGHT] =
|
|
g_param_spec_int ("coded-height", "coded-picture-height",
|
|
"coded picture height", 0, G_MAXINT, 0,
|
|
G_PARAM_READABLE | G_PARAM_STATIC_STRINGS);
|
|
|
|
g_object_class_install_properties (gobject_class, N_PROPERTIES, g_properties);
|
|
}
|
|
|
|
static void
|
|
gst_va_decoder_init (GstVaDecoder * self)
|
|
{
|
|
self->profile = VAProfileNone;
|
|
self->config = VA_INVALID_ID;
|
|
self->context = VA_INVALID_ID;
|
|
self->rt_format = 0;
|
|
self->coded_width = 0;
|
|
self->coded_height = 0;
|
|
}
|
|
|
|
static gboolean
|
|
gst_va_decoder_initialize (GstVaDecoder * self, guint32 codec)
|
|
{
|
|
if (self->available_profiles)
|
|
return FALSE;
|
|
|
|
self->available_profiles = gst_va_display_get_profiles (self->display, codec,
|
|
VAEntrypointVLD);
|
|
|
|
return (self->available_profiles != NULL);
|
|
}
|
|
|
|
GstVaDecoder *
|
|
gst_va_decoder_new (GstVaDisplay * display, guint32 codec)
|
|
{
|
|
GstVaDecoder *self;
|
|
|
|
g_return_val_if_fail (GST_IS_VA_DISPLAY (display), NULL);
|
|
|
|
self = g_object_new (GST_TYPE_VA_DECODER, "display", display, NULL);
|
|
if (!gst_va_decoder_initialize (self, codec))
|
|
gst_clear_object (&self);
|
|
|
|
return self;
|
|
}
|
|
|
|
gboolean
|
|
gst_va_decoder_is_open (GstVaDecoder * self)
|
|
{
|
|
gboolean ret;
|
|
|
|
g_return_val_if_fail (GST_IS_VA_DECODER (self), FALSE);
|
|
|
|
GST_OBJECT_LOCK (self);
|
|
ret = (self->config != VA_INVALID_ID && self->profile != VAProfileNone);
|
|
GST_OBJECT_UNLOCK (self);
|
|
return ret;
|
|
}
|
|
|
|
gboolean
|
|
gst_va_decoder_open (GstVaDecoder * self, VAProfile profile, guint rt_format)
|
|
{
|
|
VAConfigAttrib attrib = {
|
|
.type = VAConfigAttribRTFormat,
|
|
.value = rt_format,
|
|
};
|
|
VAConfigID config;
|
|
VADisplay dpy;
|
|
VAStatus status;
|
|
|
|
g_return_val_if_fail (GST_IS_VA_DECODER (self), FALSE);
|
|
|
|
if (gst_va_decoder_is_open (self))
|
|
return TRUE;
|
|
|
|
if (!gst_va_decoder_has_profile (self, profile)) {
|
|
GST_ERROR_OBJECT (self, "Unsupported profile: %d", profile);
|
|
return FALSE;
|
|
}
|
|
|
|
dpy = gst_va_display_get_va_dpy (self->display);
|
|
gst_va_display_lock (self->display);
|
|
status = vaCreateConfig (dpy, profile, VAEntrypointVLD, &attrib, 1, &config);
|
|
gst_va_display_unlock (self->display);
|
|
if (status != VA_STATUS_SUCCESS) {
|
|
GST_ERROR_OBJECT (self, "vaCreateConfig: %s", vaErrorStr (status));
|
|
return FALSE;
|
|
}
|
|
|
|
GST_OBJECT_LOCK (self);
|
|
self->config = config;
|
|
self->profile = profile;
|
|
self->rt_format = rt_format;
|
|
GST_OBJECT_UNLOCK (self);
|
|
|
|
/* now we should return now only this profile's caps */
|
|
gst_caps_replace (&self->srcpad_caps, NULL);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
gboolean
|
|
gst_va_decoder_close (GstVaDecoder * self)
|
|
{
|
|
VADisplay dpy;
|
|
VAStatus status;
|
|
|
|
g_return_val_if_fail (GST_IS_VA_DECODER (self), FALSE);
|
|
|
|
if (!gst_va_decoder_is_open (self))
|
|
return TRUE;
|
|
|
|
dpy = gst_va_display_get_va_dpy (self->display);
|
|
|
|
if (self->context != VA_INVALID_ID) {
|
|
gst_va_display_lock (self->display);
|
|
status = vaDestroyContext (dpy, self->context);
|
|
gst_va_display_unlock (self->display);
|
|
if (status != VA_STATUS_SUCCESS)
|
|
GST_ERROR_OBJECT (self, "vaDestroyContext: %s", vaErrorStr (status));
|
|
}
|
|
|
|
gst_va_display_lock (self->display);
|
|
status = vaDestroyConfig (dpy, self->config);
|
|
gst_va_display_unlock (self->display);
|
|
if (status != VA_STATUS_SUCCESS) {
|
|
GST_ERROR_OBJECT (self, "vaDestroyConfig: %s", vaErrorStr (status));
|
|
return FALSE;
|
|
}
|
|
|
|
GST_OBJECT_LOCK (self);
|
|
gst_va_decoder_init (self);
|
|
GST_OBJECT_UNLOCK (self);
|
|
|
|
gst_caps_replace (&self->srcpad_caps, NULL);
|
|
gst_caps_replace (&self->sinkpad_caps, NULL);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
gboolean
|
|
gst_va_decoder_set_format (GstVaDecoder * self, gint coded_width,
|
|
gint coded_height, GArray * surfaces)
|
|
{
|
|
VAContextID context;
|
|
VADisplay dpy;
|
|
VAStatus status;
|
|
VASurfaceID *render_targets = NULL;
|
|
gint num_render_targets = 0;
|
|
|
|
g_return_val_if_fail (GST_IS_VA_DECODER (self), FALSE);
|
|
|
|
GST_OBJECT_LOCK (self);
|
|
if (self->context != VA_INVALID_ID) {
|
|
GST_OBJECT_UNLOCK (self);
|
|
GST_INFO_OBJECT (self, "decoder already has a format");
|
|
return TRUE;
|
|
}
|
|
GST_OBJECT_UNLOCK (self);
|
|
|
|
if (!gst_va_decoder_is_open (self)) {
|
|
GST_ERROR_OBJECT (self, "decoder has not been opened yet");
|
|
return FALSE;
|
|
}
|
|
|
|
if (surfaces) {
|
|
num_render_targets = surfaces->len;
|
|
render_targets = (VASurfaceID *) surfaces->data;
|
|
}
|
|
|
|
dpy = gst_va_display_get_va_dpy (self->display);
|
|
|
|
gst_va_display_lock (self->display);
|
|
status = vaCreateContext (dpy, self->config, coded_width, coded_height,
|
|
VA_PROGRESSIVE, render_targets, num_render_targets, &context);
|
|
gst_va_display_unlock (self->display);
|
|
|
|
if (status != VA_STATUS_SUCCESS) {
|
|
GST_ERROR_OBJECT (self, "vaDestroyConfig: %s", vaErrorStr (status));
|
|
return FALSE;
|
|
}
|
|
|
|
GST_OBJECT_LOCK (self);
|
|
self->context = context;
|
|
self->coded_width = coded_width;
|
|
self->coded_height = coded_height;
|
|
GST_OBJECT_UNLOCK (self);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
static gboolean
|
|
_get_codec_caps (GstVaDecoder * self)
|
|
{
|
|
GstCaps *sinkpad_caps = NULL, *srcpad_caps = NULL;
|
|
|
|
if (!gst_va_decoder_is_open (self)
|
|
&& GST_IS_VA_DISPLAY_WRAPPED (self->display)) {
|
|
if (gst_va_caps_from_profiles (self->display, self->available_profiles,
|
|
VAEntrypointVLD, &sinkpad_caps, &srcpad_caps)) {
|
|
gst_caps_replace (&self->sinkpad_caps, sinkpad_caps);
|
|
gst_caps_replace (&self->srcpad_caps, srcpad_caps);
|
|
gst_caps_unref (srcpad_caps);
|
|
gst_caps_unref (sinkpad_caps);
|
|
|
|
return TRUE;
|
|
}
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
GstCaps *
|
|
gst_va_decoder_get_srcpad_caps (GstVaDecoder * self)
|
|
{
|
|
GstCaps *srcpad_caps = NULL;
|
|
|
|
g_return_val_if_fail (GST_IS_VA_DECODER (self), FALSE);
|
|
|
|
if (g_atomic_pointer_get (&self->srcpad_caps))
|
|
return gst_caps_ref (self->srcpad_caps);
|
|
|
|
if (_get_codec_caps (self))
|
|
return gst_caps_ref (self->srcpad_caps);
|
|
|
|
if (gst_va_decoder_is_open (self)) {
|
|
srcpad_caps = gst_va_create_raw_caps_from_config (self->display,
|
|
self->config);
|
|
gst_caps_replace (&self->srcpad_caps, srcpad_caps);
|
|
gst_caps_unref (srcpad_caps);
|
|
|
|
return gst_caps_ref (self->srcpad_caps);
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
GstCaps *
|
|
gst_va_decoder_get_sinkpad_caps (GstVaDecoder * self)
|
|
{
|
|
g_return_val_if_fail (GST_IS_VA_DECODER (self), FALSE);
|
|
|
|
if (g_atomic_pointer_get (&self->sinkpad_caps))
|
|
return gst_caps_ref (self->sinkpad_caps);
|
|
|
|
if (_get_codec_caps (self))
|
|
return gst_caps_ref (self->sinkpad_caps);
|
|
|
|
return NULL;
|
|
}
|
|
|
|
gboolean
|
|
gst_va_decoder_has_profile (GstVaDecoder * self, VAProfile profile)
|
|
{
|
|
gint i;
|
|
|
|
g_return_val_if_fail (GST_IS_VA_DECODER (self), FALSE);
|
|
|
|
if (profile == VAProfileNone)
|
|
return FALSE;
|
|
|
|
for (i = 0; i < self->available_profiles->len; i++) {
|
|
if (g_array_index (self->available_profiles, VAProfile, i) == profile)
|
|
return TRUE;
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
gint
|
|
gst_va_decoder_get_mem_types (GstVaDecoder * self)
|
|
{
|
|
VASurfaceAttrib *attribs;
|
|
guint i, attrib_count;
|
|
gint ret = 0;
|
|
|
|
g_return_val_if_fail (GST_IS_VA_DECODER (self), 0);
|
|
|
|
if (!gst_va_decoder_is_open (self))
|
|
return 0;
|
|
|
|
attribs = gst_va_get_surface_attribs (self->display, self->config,
|
|
&attrib_count);
|
|
if (!attribs)
|
|
return 0;
|
|
|
|
for (i = 0; i < attrib_count; i++) {
|
|
if (attribs[i].value.type != VAGenericValueTypeInteger)
|
|
continue;
|
|
switch (attribs[i].type) {
|
|
case VASurfaceAttribMemoryType:
|
|
ret = attribs[i].value.value.i;
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
|
|
g_free (attribs);
|
|
|
|
return ret;
|
|
}
|
|
|
|
GArray *
|
|
gst_va_decoder_get_surface_formats (GstVaDecoder * self)
|
|
{
|
|
GArray *formats;
|
|
GstVideoFormat format;
|
|
VASurfaceAttrib *attribs;
|
|
guint i, attrib_count;
|
|
|
|
g_return_val_if_fail (GST_IS_VA_DECODER (self), NULL);
|
|
|
|
if (!gst_va_decoder_is_open (self))
|
|
return NULL;
|
|
|
|
attribs = gst_va_get_surface_attribs (self->display, self->config,
|
|
&attrib_count);
|
|
if (!attribs)
|
|
return NULL;
|
|
|
|
formats = g_array_new (FALSE, FALSE, sizeof (GstVideoFormat));
|
|
|
|
for (i = 0; i < attrib_count; i++) {
|
|
if (attribs[i].value.type != VAGenericValueTypeInteger)
|
|
continue;
|
|
switch (attribs[i].type) {
|
|
case VASurfaceAttribPixelFormat:
|
|
format = gst_va_video_format_from_va_fourcc (attribs[i].value.value.i);
|
|
if (format != GST_VIDEO_FORMAT_UNKNOWN)
|
|
g_array_append_val (formats, format);
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
|
|
g_free (attribs);
|
|
|
|
if (formats->len == 0) {
|
|
g_array_unref (formats);
|
|
return NULL;
|
|
}
|
|
|
|
return formats;
|
|
}
|
|
|
|
gboolean
|
|
gst_va_decoder_add_param_buffer (GstVaDecoder * self, GstVaDecodePicture * pic,
|
|
gint type, gpointer data, gsize size)
|
|
{
|
|
VABufferID buffer;
|
|
VADisplay dpy;
|
|
VAStatus status;
|
|
|
|
g_return_val_if_fail (GST_IS_VA_DECODER (self), FALSE);
|
|
g_return_val_if_fail (self->context != VA_INVALID_ID, FALSE);
|
|
g_return_val_if_fail (pic && data && size > 0, FALSE);
|
|
|
|
dpy = gst_va_display_get_va_dpy (self->display);
|
|
gst_va_display_lock (self->display);
|
|
status = vaCreateBuffer (dpy, self->context, type, size, 1, data, &buffer);
|
|
gst_va_display_unlock (self->display);
|
|
if (status != VA_STATUS_SUCCESS) {
|
|
GST_ERROR_OBJECT (self, "vaCreateBuffer: %s", vaErrorStr (status));
|
|
return FALSE;
|
|
}
|
|
|
|
g_array_append_val (pic->buffers, buffer);
|
|
return TRUE;
|
|
}
|
|
|
|
gboolean
|
|
gst_va_decoder_add_slice_buffer_with_n_params (GstVaDecoder * self,
|
|
GstVaDecodePicture * pic, gpointer params_data, gsize params_size,
|
|
guint params_num, gpointer slice_data, gsize slice_size)
|
|
{
|
|
VABufferID params_buffer, slice_buffer;
|
|
VADisplay dpy;
|
|
VAStatus status;
|
|
|
|
g_return_val_if_fail (GST_IS_VA_DECODER (self), FALSE);
|
|
g_return_val_if_fail (self->context != VA_INVALID_ID, FALSE);
|
|
g_return_val_if_fail (pic && slice_data && slice_size > 0
|
|
&& params_data && params_size > 0, FALSE);
|
|
|
|
dpy = gst_va_display_get_va_dpy (self->display);
|
|
gst_va_display_lock (self->display);
|
|
status = vaCreateBuffer (dpy, self->context, VASliceParameterBufferType,
|
|
params_size, params_num, params_data, ¶ms_buffer);
|
|
gst_va_display_unlock (self->display);
|
|
if (status != VA_STATUS_SUCCESS) {
|
|
GST_ERROR_OBJECT (self, "vaCreateBuffer: %s", vaErrorStr (status));
|
|
return FALSE;
|
|
}
|
|
|
|
gst_va_display_lock (self->display);
|
|
status = vaCreateBuffer (dpy, self->context, VASliceDataBufferType,
|
|
slice_size, 1, slice_data, &slice_buffer);
|
|
gst_va_display_unlock (self->display);
|
|
if (status != VA_STATUS_SUCCESS) {
|
|
GST_ERROR_OBJECT (self, "vaCreateBuffer: %s", vaErrorStr (status));
|
|
return FALSE;
|
|
}
|
|
|
|
g_array_append_val (pic->slices, params_buffer);
|
|
g_array_append_val (pic->slices, slice_buffer);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
gboolean
|
|
gst_va_decoder_add_slice_buffer (GstVaDecoder * self, GstVaDecodePicture * pic,
|
|
gpointer params_data, gsize params_size, gpointer slice_data,
|
|
gsize slice_size)
|
|
{
|
|
return gst_va_decoder_add_slice_buffer_with_n_params (self, pic, params_data,
|
|
params_size, 1, slice_data, slice_size);
|
|
}
|
|
|
|
gboolean
|
|
gst_va_decoder_decode_with_aux_surface (GstVaDecoder * self,
|
|
GstVaDecodePicture * pic, gboolean use_aux)
|
|
{
|
|
VADisplay dpy;
|
|
VAStatus status;
|
|
VASurfaceID surface = VA_INVALID_ID;
|
|
gboolean ret = FALSE;
|
|
|
|
g_return_val_if_fail (GST_IS_VA_DECODER (self), FALSE);
|
|
g_return_val_if_fail (self->context != VA_INVALID_ID, FALSE);
|
|
g_return_val_if_fail (pic, FALSE);
|
|
|
|
if (use_aux) {
|
|
surface = gst_va_decode_picture_get_aux_surface (pic);
|
|
} else {
|
|
surface = gst_va_decode_picture_get_surface (pic);
|
|
}
|
|
if (surface == VA_INVALID_ID) {
|
|
GST_ERROR_OBJECT (self, "Decode picture without VASurfaceID");
|
|
return FALSE;
|
|
}
|
|
|
|
GST_TRACE_OBJECT (self, "Decode to surface %#x", surface);
|
|
|
|
dpy = gst_va_display_get_va_dpy (self->display);
|
|
|
|
gst_va_display_lock (self->display);
|
|
status = vaBeginPicture (dpy, self->context, surface);
|
|
gst_va_display_unlock (self->display);
|
|
if (status != VA_STATUS_SUCCESS) {
|
|
GST_WARNING_OBJECT (self, "vaBeginPicture: %s", vaErrorStr (status));
|
|
goto fail_end_pic;
|
|
}
|
|
|
|
if (pic->buffers->len > 0) {
|
|
gst_va_display_lock (self->display);
|
|
status = vaRenderPicture (dpy, self->context,
|
|
(VABufferID *) pic->buffers->data, pic->buffers->len);
|
|
gst_va_display_unlock (self->display);
|
|
if (status != VA_STATUS_SUCCESS) {
|
|
GST_WARNING_OBJECT (self, "vaRenderPicture: %s", vaErrorStr (status));
|
|
goto fail_end_pic;
|
|
}
|
|
}
|
|
|
|
if (pic->slices->len > 0) {
|
|
gst_va_display_lock (self->display);
|
|
status = vaRenderPicture (dpy, self->context,
|
|
(VABufferID *) pic->slices->data, pic->slices->len);
|
|
gst_va_display_unlock (self->display);
|
|
if (status != VA_STATUS_SUCCESS) {
|
|
GST_WARNING_OBJECT (self, "vaRenderPicture: %s", vaErrorStr (status));
|
|
goto fail_end_pic;
|
|
}
|
|
}
|
|
|
|
gst_va_display_lock (self->display);
|
|
status = vaEndPicture (dpy, self->context);
|
|
gst_va_display_unlock (self->display);
|
|
if (status != VA_STATUS_SUCCESS)
|
|
GST_WARNING_OBJECT (self, "vaEndPicture: %s", vaErrorStr (status));
|
|
else
|
|
ret = TRUE;
|
|
|
|
bail:
|
|
_destroy_buffers (pic);
|
|
|
|
return ret;
|
|
|
|
fail_end_pic:
|
|
{
|
|
gst_va_display_lock (self->display);
|
|
status = vaEndPicture (dpy, self->context);
|
|
gst_va_display_unlock (self->display);
|
|
goto bail;
|
|
}
|
|
}
|
|
|
|
gboolean
|
|
gst_va_decoder_decode (GstVaDecoder * self, GstVaDecodePicture * pic)
|
|
{
|
|
return gst_va_decoder_decode_with_aux_surface (self, pic, FALSE);
|
|
}
|
|
|
|
static gboolean
|
|
_destroy_buffers (GstVaDecodePicture * pic)
|
|
{
|
|
VABufferID buffer;
|
|
VADisplay dpy;
|
|
VAStatus status;
|
|
guint i;
|
|
gboolean ret = TRUE;
|
|
|
|
g_return_val_if_fail (GST_IS_VA_DISPLAY (pic->display), FALSE);
|
|
|
|
dpy = gst_va_display_get_va_dpy (pic->display);
|
|
|
|
for (i = 0; i < pic->buffers->len; i++) {
|
|
buffer = g_array_index (pic->buffers, VABufferID, i);
|
|
gst_va_display_lock (pic->display);
|
|
status = vaDestroyBuffer (dpy, buffer);
|
|
gst_va_display_unlock (pic->display);
|
|
if (status != VA_STATUS_SUCCESS) {
|
|
ret = FALSE;
|
|
GST_WARNING ("Failed to destroy parameter buffer: %s",
|
|
vaErrorStr (status));
|
|
}
|
|
}
|
|
|
|
for (i = 0; i < pic->slices->len; i++) {
|
|
buffer = g_array_index (pic->slices, VABufferID, i);
|
|
gst_va_display_lock (pic->display);
|
|
status = vaDestroyBuffer (dpy, buffer);
|
|
gst_va_display_unlock (pic->display);
|
|
if (status != VA_STATUS_SUCCESS) {
|
|
ret = FALSE;
|
|
GST_WARNING ("Failed to destroy slice buffer: %s", vaErrorStr (status));
|
|
}
|
|
}
|
|
|
|
pic->buffers = g_array_set_size (pic->buffers, 0);
|
|
pic->slices = g_array_set_size (pic->slices, 0);
|
|
|
|
return ret;
|
|
}
|
|
|
|
GstVaDecodePicture *
|
|
gst_va_decode_picture_new (GstVaDecoder * self, GstBuffer * buffer)
|
|
{
|
|
GstVaDecodePicture *pic;
|
|
|
|
g_return_val_if_fail (buffer && GST_IS_BUFFER (buffer), NULL);
|
|
g_return_val_if_fail (self && GST_IS_VA_DECODER (self), NULL);
|
|
|
|
pic = g_slice_new (GstVaDecodePicture);
|
|
pic->gstbuffer = gst_buffer_ref (buffer);
|
|
pic->buffers = g_array_sized_new (FALSE, FALSE, sizeof (VABufferID), 16);
|
|
pic->slices = g_array_sized_new (FALSE, FALSE, sizeof (VABufferID), 64);
|
|
pic->display = gst_object_ref (self->display);
|
|
|
|
return pic;
|
|
}
|
|
|
|
VASurfaceID
|
|
gst_va_decode_picture_get_surface (GstVaDecodePicture * pic)
|
|
{
|
|
g_return_val_if_fail (pic, VA_INVALID_ID);
|
|
g_return_val_if_fail (pic->gstbuffer, VA_INVALID_ID);
|
|
|
|
return gst_va_buffer_get_surface (pic->gstbuffer);
|
|
}
|
|
|
|
VASurfaceID
|
|
gst_va_decode_picture_get_aux_surface (GstVaDecodePicture * pic)
|
|
{
|
|
g_return_val_if_fail (pic, VA_INVALID_ID);
|
|
g_return_val_if_fail (pic->gstbuffer, VA_INVALID_ID);
|
|
|
|
return gst_va_buffer_get_aux_surface (pic->gstbuffer);
|
|
}
|
|
|
|
void
|
|
gst_va_decode_picture_free (GstVaDecodePicture * pic)
|
|
{
|
|
g_return_if_fail (pic);
|
|
|
|
/* only if add_param_buffer() or add_slice_buffer() failed */
|
|
if ((pic->buffers != NULL && pic->buffers->len > 0)
|
|
|| (pic->slices != NULL && pic->slices->len > 0))
|
|
_destroy_buffers (pic);
|
|
|
|
gst_buffer_unref (pic->gstbuffer);
|
|
g_clear_pointer (&pic->buffers, g_array_unref);
|
|
g_clear_pointer (&pic->slices, g_array_unref);
|
|
gst_clear_object (&pic->display);
|
|
|
|
g_slice_free (GstVaDecodePicture, pic);
|
|
}
|
|
|
|
GstVaDecodePicture *
|
|
gst_va_decode_picture_dup (GstVaDecodePicture * pic)
|
|
{
|
|
GstVaDecodePicture *dup;
|
|
|
|
g_return_val_if_fail (pic, NULL);
|
|
|
|
dup = g_slice_new0 (GstVaDecodePicture);
|
|
|
|
/* dups only need gstbuffer */
|
|
dup->gstbuffer = gst_buffer_ref (pic->gstbuffer);
|
|
return dup;
|
|
}
|
|
|
|
gboolean
|
|
gst_va_decoder_format_changed (GstVaDecoder * decoder, VAProfile new_profile,
|
|
guint new_rtformat, gint new_width, gint new_height)
|
|
{
|
|
/* @TODO: Check if current buffers are large enough, and reuse
|
|
* them */
|
|
return !(decoder->profile == new_profile &&
|
|
decoder->rt_format == new_rtformat &&
|
|
decoder->coded_width == new_width && decoder->coded_height == new_height);
|
|
}
|