2010-04-29 17:55:58 +00:00
|
|
|
/*
|
|
|
|
* gstvaapidecode.c - VA-API video decoder
|
|
|
|
*
|
2012-01-16 09:41:10 +00:00
|
|
|
* Copyright (C) 2010-2011 Splitted-Desktop Systems
|
2013-01-29 13:14:45 +00:00
|
|
|
* Copyright (C) 2011-2013 Intel Corporation
|
2010-04-29 17:55:58 +00:00
|
|
|
*
|
2011-10-18 07:06:52 +00:00
|
|
|
* 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.
|
2010-04-29 17:55:58 +00:00
|
|
|
*
|
2011-10-18 07:06:52 +00:00
|
|
|
* This library is distributed in the hope that it will be useful,
|
2010-04-29 17:55:58 +00:00
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
2011-10-18 07:06:52 +00:00
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
|
|
* Lesser General Public License for more details.
|
2010-04-29 17:55:58 +00:00
|
|
|
*
|
2011-10-18 07:06:52 +00:00
|
|
|
* 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
|
|
|
|
*/
|
2010-04-29 17:55:58 +00:00
|
|
|
|
|
|
|
/**
|
|
|
|
* SECTION:gstvaapidecode
|
|
|
|
* @short_description: A VA-API based video decoder
|
|
|
|
*
|
|
|
|
* vaapidecode decodes from raw bitstreams to surfaces suitable for
|
|
|
|
* the vaapisink element.
|
|
|
|
*/
|
|
|
|
|
2012-09-12 11:41:47 +00:00
|
|
|
#include "gst/vaapi/sysdeps.h"
|
2011-11-04 20:50:15 +00:00
|
|
|
#include <gst/vaapi/gstvaapidisplay.h>
|
2013-01-05 16:55:47 +00:00
|
|
|
#include <gst/vaapi/gstvaapivideometa.h>
|
2011-11-04 20:50:15 +00:00
|
|
|
#include <gst/video/videocontext.h>
|
|
|
|
|
|
|
|
#include "gstvaapidecode.h"
|
|
|
|
#include "gstvaapipluginutil.h"
|
2012-07-24 12:31:25 +00:00
|
|
|
#include "gstvaapipluginbuffer.h"
|
2010-04-29 17:55:58 +00:00
|
|
|
|
2012-07-19 15:16:28 +00:00
|
|
|
#include <gst/vaapi/gstvaapidecoder_h264.h>
|
|
|
|
#include <gst/vaapi/gstvaapidecoder_jpeg.h>
|
|
|
|
#include <gst/vaapi/gstvaapidecoder_mpeg2.h>
|
|
|
|
#include <gst/vaapi/gstvaapidecoder_mpeg4.h>
|
|
|
|
#include <gst/vaapi/gstvaapidecoder_vc1.h>
|
2011-08-12 08:21:19 +00:00
|
|
|
|
2010-04-29 17:55:58 +00:00
|
|
|
#define GST_PLUGIN_NAME "vaapidecode"
|
|
|
|
#define GST_PLUGIN_DESC "A VA-API based video decoder"
|
|
|
|
|
|
|
|
GST_DEBUG_CATEGORY_STATIC(gst_debug_vaapidecode);
|
|
|
|
#define GST_CAT_DEFAULT gst_debug_vaapidecode
|
|
|
|
|
|
|
|
/* ElementFactory information */
|
|
|
|
static const GstElementDetails gst_vaapidecode_details =
|
|
|
|
GST_ELEMENT_DETAILS(
|
2010-05-10 09:32:47 +00:00
|
|
|
"VA-API decoder",
|
2010-04-29 17:55:58 +00:00
|
|
|
"Codec/Decoder/Video",
|
|
|
|
GST_PLUGIN_DESC,
|
2011-10-19 12:43:56 +00:00
|
|
|
"Gwenole Beauchesne <gwenole.beauchesne@intel.com>");
|
2010-04-29 17:55:58 +00:00
|
|
|
|
|
|
|
/* Default templates */
|
2011-09-14 11:07:18 +00:00
|
|
|
#define GST_CAPS_CODEC(CODEC) CODEC "; "
|
2010-04-29 17:55:58 +00:00
|
|
|
|
|
|
|
static const char gst_vaapidecode_sink_caps_str[] =
|
2011-09-14 09:34:05 +00:00
|
|
|
GST_CAPS_CODEC("video/mpeg, mpegversion=2, systemstream=(boolean)false")
|
2010-04-29 17:55:58 +00:00
|
|
|
GST_CAPS_CODEC("video/mpeg, mpegversion=4")
|
2010-05-03 17:36:01 +00:00
|
|
|
GST_CAPS_CODEC("video/x-divx")
|
|
|
|
GST_CAPS_CODEC("video/x-xvid")
|
2010-04-29 17:55:58 +00:00
|
|
|
GST_CAPS_CODEC("video/x-h263")
|
|
|
|
GST_CAPS_CODEC("video/x-h264")
|
2010-05-03 11:44:23 +00:00
|
|
|
GST_CAPS_CODEC("video/x-wmv")
|
2012-02-09 16:21:04 +00:00
|
|
|
GST_CAPS_CODEC("image/jpeg")
|
2010-04-29 17:55:58 +00:00
|
|
|
;
|
|
|
|
|
|
|
|
static const char gst_vaapidecode_src_caps_str[] =
|
2010-05-07 06:35:31 +00:00
|
|
|
GST_VAAPI_SURFACE_CAPS;
|
2010-04-29 17:55:58 +00:00
|
|
|
|
|
|
|
static GstStaticPadTemplate gst_vaapidecode_sink_factory =
|
|
|
|
GST_STATIC_PAD_TEMPLATE(
|
|
|
|
"sink",
|
|
|
|
GST_PAD_SINK,
|
|
|
|
GST_PAD_ALWAYS,
|
|
|
|
GST_STATIC_CAPS(gst_vaapidecode_sink_caps_str));
|
|
|
|
|
|
|
|
static GstStaticPadTemplate gst_vaapidecode_src_factory =
|
|
|
|
GST_STATIC_PAD_TEMPLATE(
|
|
|
|
"src",
|
|
|
|
GST_PAD_SRC,
|
|
|
|
GST_PAD_ALWAYS,
|
|
|
|
GST_STATIC_CAPS(gst_vaapidecode_src_caps_str));
|
|
|
|
|
2012-07-19 09:43:03 +00:00
|
|
|
static void
|
|
|
|
gst_vaapidecode_implements_iface_init(GstImplementsInterfaceClass *iface);
|
|
|
|
|
2012-06-30 17:58:36 +00:00
|
|
|
static void
|
|
|
|
gst_video_context_interface_init(GstVideoContextInterface *iface);
|
|
|
|
|
2011-11-04 20:50:15 +00:00
|
|
|
#define GstVideoContextClass GstVideoContextInterface
|
2012-06-30 17:58:36 +00:00
|
|
|
G_DEFINE_TYPE_WITH_CODE(
|
2010-04-29 17:55:58 +00:00
|
|
|
GstVaapiDecode,
|
|
|
|
gst_vaapidecode,
|
2012-12-04 13:45:29 +00:00
|
|
|
GST_TYPE_VIDEO_DECODER,
|
2012-07-19 09:43:03 +00:00
|
|
|
G_IMPLEMENT_INTERFACE(GST_TYPE_IMPLEMENTS_INTERFACE,
|
|
|
|
gst_vaapidecode_implements_iface_init);
|
2012-06-30 17:58:36 +00:00
|
|
|
G_IMPLEMENT_INTERFACE(GST_TYPE_VIDEO_CONTEXT,
|
2012-09-07 14:11:12 +00:00
|
|
|
gst_video_context_interface_init))
|
2010-04-29 17:55:58 +00:00
|
|
|
|
2011-07-22 13:39:51 +00:00
|
|
|
static gboolean
|
2012-12-12 14:22:32 +00:00
|
|
|
gst_vaapidecode_update_src_caps(GstVaapiDecode *decode,
|
|
|
|
GstVideoCodecState *ref_state);
|
2011-07-22 13:39:51 +00:00
|
|
|
|
|
|
|
static void
|
|
|
|
gst_vaapi_decoder_notify_caps(GObject *obj, GParamSpec *pspec, void *user_data)
|
|
|
|
{
|
|
|
|
GstVaapiDecode * const decode = GST_VAAPIDECODE(user_data);
|
2012-12-12 14:22:32 +00:00
|
|
|
GstVideoCodecState *codec_state;
|
2011-07-22 13:39:51 +00:00
|
|
|
|
|
|
|
g_assert(decode->decoder == GST_VAAPI_DECODER(obj));
|
|
|
|
|
2012-12-12 14:22:32 +00:00
|
|
|
codec_state = gst_vaapi_decoder_get_codec_state(decode->decoder);
|
|
|
|
gst_vaapidecode_update_src_caps(decode, codec_state);
|
|
|
|
gst_video_codec_state_unref(codec_state);
|
2011-07-22 13:39:51 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
static inline gboolean
|
|
|
|
gst_vaapidecode_update_sink_caps(GstVaapiDecode *decode, GstCaps *caps)
|
|
|
|
{
|
2012-12-04 13:53:15 +00:00
|
|
|
gst_caps_replace(&decode->sinkpad_caps, caps);
|
2011-07-22 13:39:51 +00:00
|
|
|
return TRUE;
|
|
|
|
}
|
|
|
|
|
|
|
|
static gboolean
|
2012-12-12 14:22:32 +00:00
|
|
|
gst_vaapidecode_update_src_caps(GstVaapiDecode *decode,
|
|
|
|
GstVideoCodecState *ref_state)
|
2011-07-22 13:39:51 +00:00
|
|
|
{
|
2012-12-04 13:45:29 +00:00
|
|
|
GstVideoDecoder * const vdec = GST_VIDEO_DECODER(decode);
|
2012-12-12 14:22:32 +00:00
|
|
|
GstVideoCodecState *state;
|
|
|
|
GstVideoInfo *vi;
|
2012-12-04 13:45:29 +00:00
|
|
|
|
|
|
|
state = gst_video_decoder_set_output_state(vdec,
|
2012-12-12 14:22:32 +00:00
|
|
|
GST_VIDEO_INFO_FORMAT(&ref_state->info),
|
|
|
|
ref_state->info.width, ref_state->info.height, ref_state);
|
2012-12-04 13:45:29 +00:00
|
|
|
if (!state)
|
|
|
|
return FALSE;
|
|
|
|
|
2012-12-12 14:22:32 +00:00
|
|
|
vi = &state->info;
|
2012-12-04 13:45:29 +00:00
|
|
|
gst_video_codec_state_unref(state);
|
|
|
|
|
|
|
|
/* XXX: gst_video_info_to_caps() from GStreamer 0.10 does not
|
|
|
|
reconstruct suitable caps for "encoded" video formats */
|
|
|
|
state->caps = gst_caps_from_string(GST_VAAPI_SURFACE_CAPS_NAME);
|
|
|
|
if (!state->caps)
|
|
|
|
return FALSE;
|
|
|
|
|
|
|
|
gst_caps_set_simple(state->caps,
|
|
|
|
"type", G_TYPE_STRING, "vaapi",
|
|
|
|
"opengl", G_TYPE_BOOLEAN, USE_GLX,
|
2012-12-12 14:22:32 +00:00
|
|
|
"width", G_TYPE_INT, vi->width,
|
|
|
|
"height", G_TYPE_INT, vi->height,
|
|
|
|
"framerate", GST_TYPE_FRACTION, vi->fps_n, vi->fps_d,
|
|
|
|
"pixel-aspect-ratio", GST_TYPE_FRACTION, vi->par_n, vi->par_d,
|
2012-12-04 13:45:29 +00:00
|
|
|
NULL);
|
|
|
|
|
2012-12-12 14:22:32 +00:00
|
|
|
if (GST_VIDEO_INFO_IS_INTERLACED(vi))
|
2012-12-04 13:45:29 +00:00
|
|
|
gst_caps_set_simple(state->caps, "interlaced", G_TYPE_BOOLEAN,
|
|
|
|
TRUE, NULL);
|
|
|
|
|
|
|
|
gst_caps_replace(&decode->srcpad_caps, state->caps);
|
|
|
|
return TRUE;
|
2011-07-22 13:39:51 +00:00
|
|
|
}
|
|
|
|
|
2010-05-16 21:04:32 +00:00
|
|
|
static void
|
2012-12-03 12:46:28 +00:00
|
|
|
gst_vaapidecode_release(GstVaapiDecode *decode)
|
2010-05-16 21:04:32 +00:00
|
|
|
{
|
2012-10-04 16:39:53 +00:00
|
|
|
g_mutex_lock(&decode->decoder_mutex);
|
|
|
|
g_cond_signal(&decode->decoder_ready);
|
|
|
|
g_mutex_unlock(&decode->decoder_mutex);
|
2010-05-16 21:04:32 +00:00
|
|
|
}
|
|
|
|
|
2010-04-30 15:37:28 +00:00
|
|
|
static GstFlowReturn
|
2013-01-17 17:19:14 +00:00
|
|
|
gst_vaapidecode_decode_frame(GstVideoDecoder *vdec, GstVideoCodecFrame *frame)
|
2010-04-29 17:55:58 +00:00
|
|
|
{
|
2012-12-04 13:45:29 +00:00
|
|
|
GstVaapiDecode * const decode = GST_VAAPIDECODE(vdec);
|
2010-04-30 15:37:28 +00:00
|
|
|
GstVaapiDecoderStatus status;
|
2010-04-29 17:55:58 +00:00
|
|
|
GstFlowReturn ret;
|
2012-09-11 13:54:20 +00:00
|
|
|
gint64 end_time;
|
2010-04-29 17:55:58 +00:00
|
|
|
|
2013-01-03 12:10:33 +00:00
|
|
|
if (!decode->render_time_base)
|
|
|
|
decode->render_time_base = g_get_monotonic_time();
|
|
|
|
end_time = decode->render_time_base;
|
|
|
|
end_time += GST_TIME_AS_USECONDS(decode->last_buffer_time);
|
|
|
|
end_time += G_TIME_SPAN_SECOND;
|
|
|
|
|
2012-12-13 13:30:18 +00:00
|
|
|
/* Decode current frame */
|
2010-04-30 15:50:19 +00:00
|
|
|
for (;;) {
|
2012-12-13 13:30:18 +00:00
|
|
|
status = gst_vaapi_decoder_decode(decode->decoder, frame);
|
2012-12-04 13:45:29 +00:00
|
|
|
if (status == GST_VAAPI_DECODER_STATUS_ERROR_NO_SURFACE) {
|
|
|
|
gboolean was_signalled;
|
|
|
|
g_mutex_lock(&decode->decoder_mutex);
|
|
|
|
was_signalled = g_cond_wait_until(
|
|
|
|
&decode->decoder_ready,
|
|
|
|
&decode->decoder_mutex,
|
|
|
|
end_time
|
|
|
|
);
|
|
|
|
g_mutex_unlock(&decode->decoder_mutex);
|
|
|
|
if (was_signalled)
|
|
|
|
continue;
|
|
|
|
goto error_decode_timeout;
|
2010-04-30 15:50:19 +00:00
|
|
|
}
|
2012-12-04 13:45:29 +00:00
|
|
|
if (status != GST_VAAPI_DECODER_STATUS_SUCCESS)
|
|
|
|
goto error_decode;
|
2012-12-13 13:30:18 +00:00
|
|
|
break;
|
|
|
|
}
|
2013-01-17 17:19:14 +00:00
|
|
|
return GST_FLOW_OK;
|
|
|
|
|
|
|
|
/* ERRORS */
|
|
|
|
error_decode_timeout:
|
|
|
|
{
|
|
|
|
GST_WARNING("decode timeout. Decoder required a VA surface but none "
|
|
|
|
"got available within one second");
|
|
|
|
return GST_FLOW_UNEXPECTED;
|
|
|
|
}
|
|
|
|
error_decode:
|
|
|
|
{
|
|
|
|
GST_ERROR("decode error %d", status);
|
|
|
|
switch (status) {
|
|
|
|
case GST_VAAPI_DECODER_STATUS_ERROR_UNSUPPORTED_CODEC:
|
|
|
|
case GST_VAAPI_DECODER_STATUS_ERROR_UNSUPPORTED_PROFILE:
|
|
|
|
case GST_VAAPI_DECODER_STATUS_ERROR_UNSUPPORTED_CHROMA_FORMAT:
|
|
|
|
ret = GST_FLOW_NOT_SUPPORTED;
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
ret = GST_FLOW_UNEXPECTED;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
gst_video_decoder_drop_frame(vdec, frame);
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static GstFlowReturn
|
|
|
|
gst_vaapidecode_push_decoded_frames(GstVideoDecoder *vdec)
|
|
|
|
{
|
|
|
|
GstVaapiDecode * const decode = GST_VAAPIDECODE(vdec);
|
|
|
|
GstVaapiSurfaceProxy *proxy;
|
|
|
|
GstVaapiDecoderStatus status;
|
|
|
|
GstVideoCodecFrame *out_frame;
|
|
|
|
GstFlowReturn ret;
|
2012-12-04 13:45:29 +00:00
|
|
|
|
2012-12-13 13:30:18 +00:00
|
|
|
/* Output all decoded frames */
|
|
|
|
for (;;) {
|
|
|
|
status = gst_vaapi_decoder_get_frame(decode->decoder, &out_frame);
|
|
|
|
if (status != GST_VAAPI_DECODER_STATUS_SUCCESS)
|
|
|
|
return GST_FLOW_OK;
|
2012-12-04 13:45:29 +00:00
|
|
|
|
2013-01-30 15:26:07 +00:00
|
|
|
if (!GST_VIDEO_CODEC_FRAME_IS_DECODE_ONLY(out_frame)) {
|
|
|
|
proxy = gst_video_codec_frame_get_user_data(out_frame);
|
2010-04-30 15:50:19 +00:00
|
|
|
|
2013-01-30 15:26:07 +00:00
|
|
|
gst_vaapi_surface_proxy_set_user_data(proxy,
|
|
|
|
decode, (GDestroyNotify)gst_vaapidecode_release);
|
2010-05-16 21:04:32 +00:00
|
|
|
|
2013-01-30 15:26:07 +00:00
|
|
|
out_frame->output_buffer =
|
|
|
|
gst_vaapi_video_buffer_new_with_surface_proxy(proxy);
|
|
|
|
if (!out_frame->output_buffer)
|
|
|
|
goto error_create_buffer;
|
|
|
|
}
|
2010-04-30 15:50:19 +00:00
|
|
|
|
2012-12-04 13:45:29 +00:00
|
|
|
ret = gst_video_decoder_finish_frame(vdec, out_frame);
|
2010-04-30 15:50:19 +00:00
|
|
|
if (ret != GST_FLOW_OK)
|
|
|
|
goto error_commit_buffer;
|
2013-01-03 12:10:33 +00:00
|
|
|
|
2013-01-30 15:26:07 +00:00
|
|
|
if (GST_CLOCK_TIME_IS_VALID(out_frame->pts))
|
|
|
|
decode->last_buffer_time = out_frame->pts;
|
2012-12-13 14:34:10 +00:00
|
|
|
gst_video_codec_frame_unref(out_frame);
|
2012-12-13 13:30:18 +00:00
|
|
|
};
|
2010-04-30 15:37:28 +00:00
|
|
|
return GST_FLOW_OK;
|
2010-04-29 17:55:58 +00:00
|
|
|
|
|
|
|
/* ERRORS */
|
|
|
|
error_create_buffer:
|
|
|
|
{
|
|
|
|
const GstVaapiID surface_id =
|
|
|
|
gst_vaapi_surface_get_id(GST_VAAPI_SURFACE_PROXY_SURFACE(proxy));
|
|
|
|
|
2012-12-27 09:35:45 +00:00
|
|
|
GST_ERROR("video sink failed to create video buffer for proxy'ed "
|
2012-10-17 12:52:35 +00:00
|
|
|
"surface %" GST_VAAPI_ID_FORMAT,
|
|
|
|
GST_VAAPI_ID_ARGS(surface_id));
|
2012-12-04 13:45:29 +00:00
|
|
|
gst_video_decoder_drop_frame(vdec, out_frame);
|
2012-12-13 14:34:10 +00:00
|
|
|
gst_video_codec_frame_unref(out_frame);
|
2010-04-30 15:37:28 +00:00
|
|
|
return GST_FLOW_UNEXPECTED;
|
2010-04-29 17:55:58 +00:00
|
|
|
}
|
|
|
|
error_commit_buffer:
|
|
|
|
{
|
|
|
|
GST_DEBUG("video sink rejected the video buffer (error %d)", ret);
|
2012-12-13 14:34:10 +00:00
|
|
|
gst_video_codec_frame_unref(out_frame);
|
2010-04-30 15:37:28 +00:00
|
|
|
return GST_FLOW_UNEXPECTED;
|
2010-04-29 17:55:58 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-01-17 17:19:14 +00:00
|
|
|
static GstFlowReturn
|
|
|
|
gst_vaapidecode_handle_frame(GstVideoDecoder *vdec, GstVideoCodecFrame *frame)
|
|
|
|
{
|
|
|
|
GstFlowReturn ret;
|
|
|
|
|
|
|
|
ret = gst_vaapidecode_decode_frame(vdec, frame);
|
|
|
|
if (ret != GST_FLOW_OK)
|
|
|
|
return ret;
|
|
|
|
return gst_vaapidecode_push_decoded_frames(vdec);
|
|
|
|
}
|
|
|
|
|
2013-01-17 17:22:49 +00:00
|
|
|
static GstFlowReturn
|
|
|
|
gst_vaapidecode_finish(GstVideoDecoder *vdec)
|
|
|
|
{
|
|
|
|
GstVaapiDecode * const decode = GST_VAAPIDECODE(vdec);
|
|
|
|
GstVaapiDecoderStatus status;
|
|
|
|
|
|
|
|
status = gst_vaapi_decoder_flush(decode->decoder);
|
|
|
|
if (status != GST_VAAPI_DECODER_STATUS_SUCCESS)
|
|
|
|
goto error_flush;
|
|
|
|
return gst_vaapidecode_push_decoded_frames(vdec);
|
|
|
|
|
|
|
|
/* ERRORS */
|
|
|
|
error_flush:
|
|
|
|
{
|
|
|
|
GST_ERROR("failed to flush decoder (status %d)", status);
|
|
|
|
return GST_FLOW_UNEXPECTED;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-07-25 08:02:29 +00:00
|
|
|
static inline gboolean
|
|
|
|
gst_vaapidecode_ensure_display(GstVaapiDecode *decode)
|
|
|
|
{
|
|
|
|
return gst_vaapi_ensure_display(decode, GST_VAAPI_DISPLAY_TYPE_ANY,
|
|
|
|
&decode->display);
|
|
|
|
}
|
|
|
|
|
2012-09-11 14:41:32 +00:00
|
|
|
static inline guint
|
|
|
|
gst_vaapi_codec_from_caps(GstCaps *caps)
|
|
|
|
{
|
|
|
|
return gst_vaapi_profile_get_codec(gst_vaapi_profile_from_caps(caps));
|
|
|
|
}
|
|
|
|
|
2010-05-05 15:36:25 +00:00
|
|
|
static gboolean
|
2010-05-12 08:32:34 +00:00
|
|
|
gst_vaapidecode_create(GstVaapiDecode *decode, GstCaps *caps)
|
2010-05-05 15:36:25 +00:00
|
|
|
{
|
2012-04-10 11:29:10 +00:00
|
|
|
GstVaapiDisplay *dpy;
|
2011-08-12 08:21:19 +00:00
|
|
|
|
2012-07-25 08:02:29 +00:00
|
|
|
if (!gst_vaapidecode_ensure_display(decode))
|
2010-05-05 15:36:25 +00:00
|
|
|
return FALSE;
|
2012-07-19 15:16:28 +00:00
|
|
|
dpy = decode->display;
|
2010-04-29 17:55:58 +00:00
|
|
|
|
2012-09-11 14:41:32 +00:00
|
|
|
switch (gst_vaapi_codec_from_caps(caps)) {
|
|
|
|
case GST_VAAPI_CODEC_MPEG2:
|
|
|
|
decode->decoder = gst_vaapi_decoder_mpeg2_new(dpy, caps);
|
|
|
|
break;
|
|
|
|
case GST_VAAPI_CODEC_MPEG4:
|
|
|
|
case GST_VAAPI_CODEC_H263:
|
|
|
|
decode->decoder = gst_vaapi_decoder_mpeg4_new(dpy, caps);
|
|
|
|
break;
|
|
|
|
case GST_VAAPI_CODEC_H264:
|
2012-07-19 15:16:28 +00:00
|
|
|
decode->decoder = gst_vaapi_decoder_h264_new(dpy, caps);
|
2012-09-11 14:41:32 +00:00
|
|
|
break;
|
|
|
|
case GST_VAAPI_CODEC_WMV3:
|
|
|
|
case GST_VAAPI_CODEC_VC1:
|
2012-07-19 15:16:28 +00:00
|
|
|
decode->decoder = gst_vaapi_decoder_vc1_new(dpy, caps);
|
2012-09-11 14:41:32 +00:00
|
|
|
break;
|
2012-02-09 16:21:04 +00:00
|
|
|
#if USE_JPEG_DECODER
|
2012-09-11 14:41:32 +00:00
|
|
|
case GST_VAAPI_CODEC_JPEG:
|
2012-07-19 15:16:28 +00:00
|
|
|
decode->decoder = gst_vaapi_decoder_jpeg_new(dpy, caps);
|
2012-09-11 14:41:32 +00:00
|
|
|
break;
|
2011-08-12 08:21:19 +00:00
|
|
|
#endif
|
2012-09-11 14:41:32 +00:00
|
|
|
default:
|
|
|
|
decode->decoder = NULL;
|
|
|
|
break;
|
|
|
|
}
|
2010-05-12 08:32:34 +00:00
|
|
|
if (!decode->decoder)
|
|
|
|
return FALSE;
|
|
|
|
|
2011-07-22 13:39:51 +00:00
|
|
|
g_signal_connect(
|
|
|
|
G_OBJECT(decode->decoder),
|
|
|
|
"notify::caps",
|
|
|
|
G_CALLBACK(gst_vaapi_decoder_notify_caps),
|
|
|
|
decode
|
|
|
|
);
|
|
|
|
|
2010-05-12 08:32:34 +00:00
|
|
|
decode->decoder_caps = gst_caps_ref(caps);
|
|
|
|
return TRUE;
|
2010-04-29 17:55:58 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
gst_vaapidecode_destroy(GstVaapiDecode *decode)
|
|
|
|
{
|
2012-12-04 13:53:15 +00:00
|
|
|
g_clear_object(&decode->decoder);
|
|
|
|
gst_caps_replace(&decode->decoder_caps, NULL);
|
2012-12-03 12:46:28 +00:00
|
|
|
gst_vaapidecode_release(decode);
|
2010-04-29 17:55:58 +00:00
|
|
|
}
|
|
|
|
|
2010-05-12 08:32:34 +00:00
|
|
|
static gboolean
|
2013-01-23 22:49:17 +00:00
|
|
|
gst_vaapidecode_reset_full(GstVaapiDecode *decode, GstCaps *caps, gboolean hard)
|
2010-05-12 08:32:34 +00:00
|
|
|
{
|
2012-09-11 15:03:33 +00:00
|
|
|
GstVaapiCodec codec;
|
|
|
|
|
2013-01-23 22:49:17 +00:00
|
|
|
/* Reset timers if hard reset was requested (e.g. seek) */
|
|
|
|
if (hard) {
|
|
|
|
decode->render_time_base = 0;
|
|
|
|
decode->last_buffer_time = 0;
|
|
|
|
}
|
|
|
|
|
2012-09-11 15:03:33 +00:00
|
|
|
/* Only reset decoder if codec type changed */
|
2013-01-23 22:49:17 +00:00
|
|
|
else if (decode->decoder && decode->decoder_caps) {
|
2012-09-11 15:03:33 +00:00
|
|
|
if (gst_caps_is_always_compatible(caps, decode->decoder_caps))
|
|
|
|
return TRUE;
|
|
|
|
codec = gst_vaapi_codec_from_caps(caps);
|
|
|
|
if (codec == gst_vaapi_decoder_get_codec(decode->decoder))
|
|
|
|
return TRUE;
|
|
|
|
}
|
2010-05-12 08:32:34 +00:00
|
|
|
|
|
|
|
gst_vaapidecode_destroy(decode);
|
|
|
|
return gst_vaapidecode_create(decode, caps);
|
|
|
|
}
|
|
|
|
|
2012-07-19 09:43:03 +00:00
|
|
|
/* GstImplementsInterface interface */
|
|
|
|
|
|
|
|
static gboolean
|
|
|
|
gst_vaapidecode_implements_interface_supported(
|
|
|
|
GstImplementsInterface *iface,
|
|
|
|
GType type
|
|
|
|
)
|
|
|
|
{
|
|
|
|
return (type == GST_TYPE_VIDEO_CONTEXT);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
gst_vaapidecode_implements_iface_init(GstImplementsInterfaceClass *iface)
|
|
|
|
{
|
|
|
|
iface->supported = gst_vaapidecode_implements_interface_supported;
|
|
|
|
}
|
|
|
|
|
2011-11-04 20:50:15 +00:00
|
|
|
/* GstVideoContext interface */
|
|
|
|
|
|
|
|
static void
|
|
|
|
gst_vaapidecode_set_video_context(GstVideoContext *context, const gchar *type,
|
|
|
|
const GValue *value)
|
|
|
|
{
|
|
|
|
GstVaapiDecode *decode = GST_VAAPIDECODE (context);
|
|
|
|
gst_vaapi_set_display (type, value, &decode->display);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
gst_video_context_interface_init(GstVideoContextInterface *iface)
|
|
|
|
{
|
|
|
|
iface->set_context = gst_vaapidecode_set_video_context;
|
|
|
|
}
|
|
|
|
|
2010-04-29 17:55:58 +00:00
|
|
|
static void
|
|
|
|
gst_vaapidecode_finalize(GObject *object)
|
|
|
|
{
|
2010-05-12 08:32:34 +00:00
|
|
|
GstVaapiDecode * const decode = GST_VAAPIDECODE(object);
|
|
|
|
|
|
|
|
gst_vaapidecode_destroy(decode);
|
|
|
|
|
2012-12-04 13:53:15 +00:00
|
|
|
gst_caps_replace(&decode->sinkpad_caps, NULL);
|
|
|
|
gst_caps_replace(&decode->srcpad_caps, NULL);
|
|
|
|
gst_caps_replace(&decode->allowed_caps, NULL);
|
2011-07-22 13:39:51 +00:00
|
|
|
|
2012-06-30 20:50:17 +00:00
|
|
|
g_clear_object(&decode->display);
|
2010-04-29 17:55:58 +00:00
|
|
|
|
2012-10-04 16:39:53 +00:00
|
|
|
g_cond_clear(&decode->decoder_ready);
|
|
|
|
g_mutex_clear(&decode->decoder_mutex);
|
|
|
|
|
2012-06-30 17:58:36 +00:00
|
|
|
G_OBJECT_CLASS(gst_vaapidecode_parent_class)->finalize(object);
|
2010-04-29 17:55:58 +00:00
|
|
|
}
|
|
|
|
|
2012-12-04 13:45:29 +00:00
|
|
|
static gboolean
|
|
|
|
gst_vaapidecode_open(GstVideoDecoder *vdec)
|
2010-04-29 17:55:58 +00:00
|
|
|
{
|
2012-12-04 13:45:29 +00:00
|
|
|
GstVaapiDecode * const decode = GST_VAAPIDECODE(vdec);
|
2010-04-29 17:55:58 +00:00
|
|
|
|
2012-12-04 13:45:29 +00:00
|
|
|
decode->is_ready = TRUE;
|
|
|
|
return TRUE;
|
|
|
|
}
|
2010-04-29 17:55:58 +00:00
|
|
|
|
2012-12-04 13:45:29 +00:00
|
|
|
static gboolean
|
|
|
|
gst_vaapidecode_close(GstVideoDecoder *vdec)
|
|
|
|
{
|
|
|
|
GstVaapiDecode * const decode = GST_VAAPIDECODE(vdec);
|
|
|
|
|
|
|
|
gst_vaapidecode_destroy(decode);
|
|
|
|
g_clear_object(&decode->display);
|
|
|
|
decode->is_ready = FALSE;
|
|
|
|
return TRUE;
|
|
|
|
}
|
|
|
|
|
2013-01-23 22:49:17 +00:00
|
|
|
static gboolean
|
|
|
|
gst_vaapidecode_reset(GstVideoDecoder *vdec, gboolean hard)
|
|
|
|
{
|
|
|
|
GstVaapiDecode * const decode = GST_VAAPIDECODE(vdec);
|
|
|
|
|
|
|
|
return gst_vaapidecode_reset_full(decode, decode->sinkpad_caps, hard);
|
|
|
|
}
|
|
|
|
|
2012-12-04 13:45:29 +00:00
|
|
|
static gboolean
|
|
|
|
gst_vaapidecode_set_format(GstVideoDecoder *vdec, GstVideoCodecState *state)
|
|
|
|
{
|
|
|
|
GstVaapiDecode * const decode = GST_VAAPIDECODE(vdec);
|
|
|
|
|
2012-12-12 14:22:32 +00:00
|
|
|
if (!gst_vaapidecode_update_sink_caps(decode, state->caps))
|
2012-12-04 13:45:29 +00:00
|
|
|
return FALSE;
|
2012-12-12 14:22:32 +00:00
|
|
|
if (!gst_vaapidecode_update_src_caps(decode, state))
|
2012-12-04 13:45:29 +00:00
|
|
|
return FALSE;
|
2013-01-23 22:49:17 +00:00
|
|
|
if (!gst_vaapidecode_reset_full(decode, decode->sinkpad_caps, FALSE))
|
2012-12-04 13:45:29 +00:00
|
|
|
return FALSE;
|
|
|
|
return TRUE;
|
|
|
|
}
|
2010-04-29 17:55:58 +00:00
|
|
|
|
2012-12-04 13:45:29 +00:00
|
|
|
static GstFlowReturn
|
|
|
|
gst_vaapidecode_parse(GstVideoDecoder *vdec,
|
|
|
|
GstVideoCodecFrame *frame, GstAdapter *adapter, gboolean at_eos)
|
|
|
|
{
|
|
|
|
GstVaapiDecode * const decode = GST_VAAPIDECODE(vdec);
|
|
|
|
GstVaapiDecoderStatus status;
|
|
|
|
GstFlowReturn ret;
|
|
|
|
guint got_unit_size;
|
|
|
|
gboolean got_frame;
|
|
|
|
|
|
|
|
status = gst_vaapi_decoder_parse(decode->decoder, frame,
|
|
|
|
adapter, at_eos, &got_unit_size, &got_frame);
|
|
|
|
|
|
|
|
switch (status) {
|
|
|
|
case GST_VAAPI_DECODER_STATUS_SUCCESS:
|
|
|
|
if (got_unit_size > 0)
|
|
|
|
gst_video_decoder_add_to_frame(vdec, got_unit_size);
|
|
|
|
if (got_frame)
|
|
|
|
ret = gst_video_decoder_have_frame(vdec);
|
|
|
|
else
|
|
|
|
ret = GST_FLOW_OK;
|
2010-04-29 17:55:58 +00:00
|
|
|
break;
|
2012-12-04 13:45:29 +00:00
|
|
|
case GST_VAAPI_DECODER_STATUS_ERROR_NO_DATA:
|
|
|
|
ret = GST_VIDEO_DECODER_FLOW_NEED_DATA;
|
2010-04-29 17:55:58 +00:00
|
|
|
break;
|
2012-12-04 13:45:29 +00:00
|
|
|
case GST_VAAPI_DECODER_STATUS_ERROR_UNSUPPORTED_CODEC:
|
|
|
|
case GST_VAAPI_DECODER_STATUS_ERROR_UNSUPPORTED_PROFILE:
|
|
|
|
case GST_VAAPI_DECODER_STATUS_ERROR_UNSUPPORTED_CHROMA_FORMAT:
|
|
|
|
GST_WARNING("parse error %d", status);
|
|
|
|
ret = GST_FLOW_NOT_SUPPORTED;
|
2011-12-08 10:54:59 +00:00
|
|
|
break;
|
2010-04-29 17:55:58 +00:00
|
|
|
default:
|
2012-12-04 13:45:29 +00:00
|
|
|
GST_ERROR("parse error %d", status);
|
|
|
|
ret = GST_FLOW_UNEXPECTED;
|
2010-04-29 17:55:58 +00:00
|
|
|
break;
|
|
|
|
}
|
2012-12-04 13:45:29 +00:00
|
|
|
return ret;
|
2010-04-29 17:55:58 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
gst_vaapidecode_class_init(GstVaapiDecodeClass *klass)
|
|
|
|
{
|
|
|
|
GObjectClass * const object_class = G_OBJECT_CLASS(klass);
|
|
|
|
GstElementClass * const element_class = GST_ELEMENT_CLASS(klass);
|
2012-12-04 13:45:29 +00:00
|
|
|
GstVideoDecoderClass * const vdec_class = GST_VIDEO_DECODER_CLASS(klass);
|
2012-06-30 17:58:36 +00:00
|
|
|
GstPadTemplate *pad_template;
|
2010-04-29 17:55:58 +00:00
|
|
|
|
2011-09-14 19:12:41 +00:00
|
|
|
GST_DEBUG_CATEGORY_INIT(gst_debug_vaapidecode,
|
|
|
|
GST_PLUGIN_NAME, 0, GST_PLUGIN_DESC);
|
|
|
|
|
2012-12-04 13:45:29 +00:00
|
|
|
object_class->finalize = gst_vaapidecode_finalize;
|
2010-04-29 17:55:58 +00:00
|
|
|
|
2012-12-04 13:45:29 +00:00
|
|
|
vdec_class->open = GST_DEBUG_FUNCPTR(gst_vaapidecode_open);
|
|
|
|
vdec_class->close = GST_DEBUG_FUNCPTR(gst_vaapidecode_close);
|
|
|
|
vdec_class->set_format = GST_DEBUG_FUNCPTR(gst_vaapidecode_set_format);
|
2013-01-23 22:49:17 +00:00
|
|
|
vdec_class->reset = GST_DEBUG_FUNCPTR(gst_vaapidecode_reset);
|
2012-12-04 13:45:29 +00:00
|
|
|
vdec_class->parse = GST_DEBUG_FUNCPTR(gst_vaapidecode_parse);
|
|
|
|
vdec_class->handle_frame = GST_DEBUG_FUNCPTR(gst_vaapidecode_handle_frame);
|
2013-01-17 17:22:49 +00:00
|
|
|
vdec_class->finish = GST_DEBUG_FUNCPTR(gst_vaapidecode_finish);
|
2010-04-29 17:55:58 +00:00
|
|
|
|
2012-06-30 17:58:36 +00:00
|
|
|
gst_element_class_set_details_simple(
|
|
|
|
element_class,
|
|
|
|
gst_vaapidecode_details.longname,
|
|
|
|
gst_vaapidecode_details.klass,
|
|
|
|
gst_vaapidecode_details.description,
|
|
|
|
gst_vaapidecode_details.author
|
|
|
|
);
|
|
|
|
|
|
|
|
/* sink pad */
|
|
|
|
pad_template = gst_static_pad_template_get(&gst_vaapidecode_sink_factory);
|
|
|
|
gst_element_class_add_pad_template(element_class, pad_template);
|
|
|
|
gst_object_unref(pad_template);
|
|
|
|
|
|
|
|
/* src pad */
|
|
|
|
pad_template = gst_static_pad_template_get(&gst_vaapidecode_src_factory);
|
|
|
|
gst_element_class_add_pad_template(element_class, pad_template);
|
|
|
|
gst_object_unref(pad_template);
|
2010-04-29 17:55:58 +00:00
|
|
|
}
|
|
|
|
|
2010-05-12 09:22:49 +00:00
|
|
|
static gboolean
|
|
|
|
gst_vaapidecode_ensure_allowed_caps(GstVaapiDecode *decode)
|
|
|
|
{
|
|
|
|
GstCaps *decode_caps;
|
|
|
|
guint i, n_decode_caps;
|
|
|
|
|
|
|
|
if (decode->allowed_caps)
|
|
|
|
return TRUE;
|
|
|
|
|
2012-07-25 08:02:29 +00:00
|
|
|
if (!gst_vaapidecode_ensure_display(decode))
|
2011-11-04 20:50:15 +00:00
|
|
|
goto error_no_display;
|
2010-05-12 09:22:49 +00:00
|
|
|
|
2011-11-04 20:50:15 +00:00
|
|
|
decode_caps = gst_vaapi_display_get_decode_caps(decode->display);
|
2010-05-12 09:22:49 +00:00
|
|
|
if (!decode_caps)
|
|
|
|
goto error_no_decode_caps;
|
|
|
|
n_decode_caps = gst_caps_get_size(decode_caps);
|
|
|
|
|
|
|
|
decode->allowed_caps = gst_caps_new_empty();
|
|
|
|
if (!decode->allowed_caps)
|
|
|
|
goto error_no_memory;
|
|
|
|
|
|
|
|
for (i = 0; i < n_decode_caps; i++) {
|
|
|
|
GstStructure *structure;
|
|
|
|
structure = gst_caps_get_structure(decode_caps, i);
|
|
|
|
if (!structure)
|
|
|
|
continue;
|
|
|
|
structure = gst_structure_copy(structure);
|
|
|
|
if (!structure)
|
|
|
|
continue;
|
|
|
|
gst_structure_remove_field(structure, "profile");
|
|
|
|
gst_structure_set(
|
|
|
|
structure,
|
|
|
|
"width", GST_TYPE_INT_RANGE, 1, G_MAXINT,
|
|
|
|
"height", GST_TYPE_INT_RANGE, 1, G_MAXINT,
|
|
|
|
NULL
|
|
|
|
);
|
|
|
|
gst_caps_merge_structure(decode->allowed_caps, structure);
|
|
|
|
}
|
|
|
|
|
|
|
|
gst_caps_unref(decode_caps);
|
|
|
|
return TRUE;
|
|
|
|
|
|
|
|
/* ERRORS */
|
|
|
|
error_no_display:
|
|
|
|
{
|
2012-12-27 09:35:45 +00:00
|
|
|
GST_ERROR("failed to retrieve VA display");
|
2010-05-12 09:22:49 +00:00
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
error_no_decode_caps:
|
|
|
|
{
|
2012-12-27 09:35:45 +00:00
|
|
|
GST_ERROR("failed to retrieve VA decode caps");
|
2010-05-12 09:22:49 +00:00
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
error_no_memory:
|
|
|
|
{
|
2012-12-27 09:35:45 +00:00
|
|
|
GST_ERROR("failed to allocate allowed-caps set");
|
2010-05-12 09:22:49 +00:00
|
|
|
gst_caps_unref(decode_caps);
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static GstCaps *
|
|
|
|
gst_vaapidecode_get_caps(GstPad *pad)
|
|
|
|
{
|
|
|
|
GstVaapiDecode * const decode = GST_VAAPIDECODE(GST_OBJECT_PARENT(pad));
|
|
|
|
|
2011-12-08 10:54:59 +00:00
|
|
|
if (!decode->is_ready)
|
|
|
|
return gst_static_pad_template_get_caps(&gst_vaapidecode_sink_factory);
|
|
|
|
|
2010-05-12 09:22:49 +00:00
|
|
|
if (!gst_vaapidecode_ensure_allowed_caps(decode))
|
|
|
|
return gst_caps_new_empty();
|
|
|
|
|
|
|
|
return gst_caps_ref(decode->allowed_caps);
|
|
|
|
}
|
|
|
|
|
2011-11-04 20:50:15 +00:00
|
|
|
static gboolean
|
|
|
|
gst_vaapidecode_query (GstPad *pad, GstQuery *query) {
|
|
|
|
GstVaapiDecode *decode = GST_VAAPIDECODE (gst_pad_get_parent_element (pad));
|
|
|
|
gboolean res;
|
|
|
|
|
|
|
|
GST_DEBUG ("sharing display %p", decode->display);
|
|
|
|
|
|
|
|
if (gst_vaapi_reply_to_query (query, decode->display))
|
|
|
|
res = TRUE;
|
2012-12-04 13:45:29 +00:00
|
|
|
else if (GST_PAD_IS_SINK(pad))
|
|
|
|
res = decode->sinkpad_query(decode->sinkpad, query);
|
2011-11-04 20:50:15 +00:00
|
|
|
else
|
2012-12-04 13:45:29 +00:00
|
|
|
res = decode->srcpad_query(decode->srcpad, query);
|
2011-11-04 20:50:15 +00:00
|
|
|
|
|
|
|
g_object_unref (decode);
|
|
|
|
return res;
|
|
|
|
}
|
|
|
|
|
2010-04-29 17:55:58 +00:00
|
|
|
static void
|
2012-06-30 17:58:36 +00:00
|
|
|
gst_vaapidecode_init(GstVaapiDecode *decode)
|
2010-04-29 17:55:58 +00:00
|
|
|
{
|
2012-12-04 13:45:29 +00:00
|
|
|
GstVideoDecoder * const vdec = GST_VIDEO_DECODER(decode);
|
2010-04-29 17:55:58 +00:00
|
|
|
|
2010-05-16 21:04:32 +00:00
|
|
|
decode->display = NULL;
|
|
|
|
decode->decoder = NULL;
|
|
|
|
decode->decoder_caps = NULL;
|
|
|
|
decode->allowed_caps = NULL;
|
2012-09-11 13:54:20 +00:00
|
|
|
decode->render_time_base = 0;
|
|
|
|
decode->last_buffer_time = 0;
|
2011-12-08 10:54:59 +00:00
|
|
|
decode->is_ready = FALSE;
|
2010-04-29 17:55:58 +00:00
|
|
|
|
2012-10-04 16:39:53 +00:00
|
|
|
g_mutex_init(&decode->decoder_mutex);
|
|
|
|
g_cond_init(&decode->decoder_ready);
|
|
|
|
|
2012-12-04 13:45:29 +00:00
|
|
|
gst_video_decoder_set_packetized(vdec, FALSE);
|
2010-04-29 17:55:58 +00:00
|
|
|
|
2012-12-04 13:45:29 +00:00
|
|
|
/* Pad through which data comes in to the element */
|
|
|
|
decode->sinkpad = GST_VIDEO_DECODER_SINK_PAD(vdec);
|
|
|
|
decode->sinkpad_query = GST_PAD_QUERYFUNC(decode->sinkpad);
|
2011-11-25 17:28:04 +00:00
|
|
|
gst_pad_set_query_function(decode->sinkpad, gst_vaapidecode_query);
|
2012-12-04 13:45:29 +00:00
|
|
|
gst_pad_set_getcaps_function(decode->sinkpad, gst_vaapidecode_get_caps);
|
2010-04-29 17:55:58 +00:00
|
|
|
|
|
|
|
/* Pad through which data goes out of the element */
|
2012-12-04 13:45:29 +00:00
|
|
|
decode->srcpad = GST_VIDEO_DECODER_SRC_PAD(vdec);
|
|
|
|
decode->srcpad_query = GST_PAD_QUERYFUNC(decode->srcpad);
|
2011-11-04 20:50:15 +00:00
|
|
|
gst_pad_set_query_function(decode->srcpad, gst_vaapidecode_query);
|
2010-04-29 17:55:58 +00:00
|
|
|
}
|