gstreamer/gst-libs/gst/vaapi/gstvaapiencoder.c
Gwenole Beauchesne 8ecc35ecf2 encoder: fix subclassing process.
Fix the GstVaapiEncoderClass parent class type. Make sure to validate
subclass hooks as early as possible, i.e. in gst_vaapi_encoder_init(),
thus avoiding useless run-time checks. Also simplify the subclass
initialization process to be less error prone.
2013-12-04 19:14:38 +01:00

526 lines
15 KiB
C

/*
* gstvaapiencoder.c - VA encoder abstraction
*
* Copyright (C) 2013 Intel Corporation
*
* 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 "gstvaapicompat.h"
#include "gstvaapiencoder.h"
#include "gstvaapiencoder_priv.h"
#include "gstvaapicontext.h"
#include "gstvaapidisplay_priv.h"
#define DEBUG 1
#include "gstvaapidebug.h"
#define GST_VAAPI_ENCODER_LOCK(encoder) \
G_STMT_START { \
g_mutex_lock (&(GST_VAAPI_ENCODER_CAST(encoder))->lock); \
} G_STMT_END
#define GST_VAAPI_ENCODER_UNLOCK(encoder) \
G_STMT_START { \
g_mutex_unlock (&(GST_VAAPI_ENCODER_CAST(encoder))->lock); \
} G_STMT_END
#define GST_VAAPI_ENCODER_BUF_FREE_WAIT(encoder) \
G_STMT_START { \
g_cond_wait (&(GST_VAAPI_ENCODER_CAST(encoder))->codedbuf_free, \
&(GST_VAAPI_ENCODER_CAST(encoder))->lock); \
} G_STMT_END
#define GST_VAAPI_ENCODER_BUF_FREE_SIGNAL(encoder) \
G_STMT_START { \
g_cond_signal (&(GST_VAAPI_ENCODER_CAST(encoder))->codedbuf_free); \
} G_STMT_END
#define GST_VAAPI_ENCODER_FREE_SURFACE_WAIT(encoder) \
G_STMT_START { \
g_cond_wait (&(GST_VAAPI_ENCODER_CAST(encoder))->surface_free, \
&(GST_VAAPI_ENCODER_CAST(encoder))->lock); \
} G_STMT_END
#define GST_VAAPI_ENCODER_FREE_SURFACE_SIGNAL(encoder) \
G_STMT_START { \
g_cond_signal (&(GST_VAAPI_ENCODER_CAST(encoder))->surface_free); \
} G_STMT_END
#define GST_VAAPI_ENCODER_SYNC_SIGNAL(encoder) \
G_STMT_START { \
g_cond_signal (&(GST_VAAPI_ENCODER_CAST(encoder))->sync_ready); \
} G_STMT_END
static inline gboolean
GST_VAAPI_ENCODER_SYNC_WAIT_TIMEOUT (GstVaapiEncoder * encoder, gint64 timeout)
{
gint64 end_time = g_get_monotonic_time () + timeout;
return g_cond_wait_until (&encoder->sync_ready, &encoder->lock, end_time);
}
typedef struct
{
GstVaapiEncPicture *picture;
GstVaapiCodedBufferProxy *buf;
} GstVaapiEncoderSyncPic;
GstVaapiEncoder *
gst_vaapi_encoder_ref (GstVaapiEncoder * encoder)
{
return gst_vaapi_object_ref (encoder);
}
void
gst_vaapi_encoder_unref (GstVaapiEncoder * encoder)
{
gst_vaapi_object_unref (encoder);
}
void
gst_vaapi_encoder_replace (GstVaapiEncoder ** old_encoder_ptr,
GstVaapiEncoder * new_encoder)
{
gst_vaapi_object_replace (old_encoder_ptr, new_encoder);
}
static void
_coded_buffer_proxy_released_notify (GstVaapiEncoder * encoder)
{
GST_VAAPI_ENCODER_LOCK (encoder);
GST_VAAPI_ENCODER_BUF_FREE_SIGNAL (encoder);
GST_VAAPI_ENCODER_UNLOCK (encoder);
}
static GstVaapiCodedBufferProxy *
gst_vaapi_encoder_create_coded_buffer (GstVaapiEncoder * encoder)
{
GstVaapiCodedBufferPool *const pool =
GST_VAAPI_CODED_BUFFER_POOL (encoder->codedbuf_pool);
GstVaapiCodedBufferProxy *codedbuf_proxy;
GST_VAAPI_ENCODER_LOCK (encoder);
do {
codedbuf_proxy = gst_vaapi_coded_buffer_proxy_new_from_pool (pool);
if (codedbuf_proxy)
break;
/* Wait for a free coded buffer to become available */
GST_VAAPI_ENCODER_BUF_FREE_WAIT (encoder);
codedbuf_proxy = gst_vaapi_coded_buffer_proxy_new_from_pool (pool);
} while (0);
GST_VAAPI_ENCODER_UNLOCK (encoder);
if (!codedbuf_proxy)
return NULL;
gst_vaapi_coded_buffer_proxy_set_destroy_notify (codedbuf_proxy,
(GDestroyNotify)_coded_buffer_proxy_released_notify, encoder);
return codedbuf_proxy;
}
static void
_surface_proxy_released_notify (GstVaapiEncoder * encoder)
{
GST_VAAPI_ENCODER_FREE_SURFACE_SIGNAL (encoder);
}
GstVaapiSurfaceProxy *
gst_vaapi_encoder_create_surface (GstVaapiEncoder * encoder)
{
GstVaapiSurfaceProxy *proxy;
g_assert (encoder && encoder->context);
g_return_val_if_fail (encoder->context, NULL);
GST_VAAPI_ENCODER_LOCK (encoder);
while (!gst_vaapi_context_get_surface_count (encoder->context)) {
GST_VAAPI_ENCODER_FREE_SURFACE_WAIT (encoder);
}
proxy = gst_vaapi_context_get_surface_proxy (encoder->context);
GST_VAAPI_ENCODER_UNLOCK (encoder);
gst_vaapi_surface_proxy_set_destroy_notify (proxy,
(GDestroyNotify) _surface_proxy_released_notify, encoder);
return proxy;
}
void
gst_vaapi_encoder_release_surface (GstVaapiEncoder * encoder,
GstVaapiSurfaceProxy * surface)
{
GST_VAAPI_ENCODER_LOCK (encoder);
gst_vaapi_surface_proxy_unref (surface);
GST_VAAPI_ENCODER_UNLOCK (encoder);
}
static GstVaapiEncoderSyncPic *
_create_sync_picture (GstVaapiEncPicture * picture,
GstVaapiCodedBufferProxy * coded_buf)
{
GstVaapiEncoderSyncPic *sync = g_slice_new0 (GstVaapiEncoderSyncPic);
g_assert (picture && coded_buf);
sync->picture = gst_vaapi_enc_picture_ref (picture);
sync->buf = gst_vaapi_coded_buffer_proxy_ref (coded_buf);
return sync;
}
static void
_free_sync_picture (GstVaapiEncoder * encoder,
GstVaapiEncoderSyncPic * sync_pic)
{
g_assert (sync_pic);
if (sync_pic->picture)
gst_vaapi_enc_picture_unref (sync_pic->picture);
if (sync_pic->buf)
gst_vaapi_coded_buffer_proxy_unref (sync_pic->buf);
g_slice_free (GstVaapiEncoderSyncPic, sync_pic);
}
static void
gst_vaapi_encoder_free_sync_pictures (GstVaapiEncoder * encoder)
{
GstVaapiEncoderSyncPic *sync;
GST_VAAPI_ENCODER_LOCK (encoder);
while (!g_queue_is_empty (&encoder->sync_pictures)) {
sync =
(GstVaapiEncoderSyncPic *) g_queue_pop_head (&encoder->sync_pictures);
GST_VAAPI_ENCODER_UNLOCK (encoder);
_free_sync_picture (encoder, sync);
GST_VAAPI_ENCODER_LOCK (encoder);
}
GST_VAAPI_ENCODER_UNLOCK (encoder);
}
static gboolean
gst_vaapi_encoder_push_sync_picture (GstVaapiEncoder * encoder,
GstVaapiEncoderSyncPic * sync_pic)
{
GST_VAAPI_ENCODER_LOCK (encoder);
g_queue_push_tail (&encoder->sync_pictures, sync_pic);
GST_VAAPI_ENCODER_SYNC_SIGNAL (encoder);
GST_VAAPI_ENCODER_UNLOCK (encoder);
return TRUE;
}
static GstVaapiEncoderStatus
gst_vaapi_encoder_pop_sync_picture (GstVaapiEncoder * encoder,
GstVaapiEncoderSyncPic ** sync_pic, guint64 timeout)
{
GstVaapiEncoderStatus ret = GST_VAAPI_ENCODER_STATUS_SUCCESS;
*sync_pic = NULL;
GST_VAAPI_ENCODER_LOCK (encoder);
if (g_queue_is_empty (&encoder->sync_pictures) &&
!GST_VAAPI_ENCODER_SYNC_WAIT_TIMEOUT (encoder, timeout))
goto timeout;
if (g_queue_is_empty (&encoder->sync_pictures)) {
ret = GST_VAAPI_ENCODER_STATUS_UNKNOWN_ERR;
goto end;
}
*sync_pic =
(GstVaapiEncoderSyncPic *) g_queue_pop_head (&encoder->sync_pictures);
g_assert (*sync_pic);
ret = GST_VAAPI_ENCODER_STATUS_SUCCESS;
goto end;
timeout:
ret = GST_VAAPI_ENCODER_STATUS_TIMEOUT;
end:
GST_VAAPI_ENCODER_UNLOCK (encoder);
return ret;
}
GstVaapiEncoderStatus
gst_vaapi_encoder_put_frame (GstVaapiEncoder * encoder,
GstVideoCodecFrame * frame)
{
GstVaapiEncoderStatus ret = GST_VAAPI_ENCODER_STATUS_SUCCESS;
GstVaapiEncoderClass *klass = GST_VAAPI_ENCODER_GET_CLASS (encoder);
GstVaapiEncPicture *picture = NULL;
GstVaapiCodedBufferProxy *coded_buf = NULL;
GstVaapiEncoderSyncPic *sync_pic = NULL;
again:
picture = NULL;
sync_pic = NULL;
ret = klass->reordering (encoder, frame, FALSE, &picture);
if (ret == GST_VAAPI_ENCODER_STATUS_FRAME_NOT_READY)
return GST_VAAPI_ENCODER_STATUS_SUCCESS;
g_assert (picture);
if (ret != GST_VAAPI_ENCODER_STATUS_SUCCESS)
goto error;
if (!picture) {
ret = GST_VAAPI_ENCODER_STATUS_PICTURE_ERR;
goto error;
}
coded_buf = gst_vaapi_encoder_create_coded_buffer (encoder);
if (!coded_buf) {
ret = GST_VAAPI_ENCODER_STATUS_OBJECT_ERR;
goto error;
}
ret = klass->encode (encoder, picture, coded_buf);
if (ret != GST_VAAPI_ENCODER_STATUS_SUCCESS)
goto error;
/* another thread would sync and get coded buffer */
sync_pic = _create_sync_picture (picture, coded_buf);
gst_vaapi_coded_buffer_proxy_replace (&coded_buf, NULL);
gst_vaapi_enc_picture_replace (&picture, NULL);
if (!gst_vaapi_encoder_push_sync_picture (encoder, sync_pic)) {
ret = GST_VAAPI_ENCODER_STATUS_THREAD_ERR;
goto error;
}
frame = NULL;
goto again;
error:
gst_vaapi_enc_picture_replace (&picture, NULL);
gst_vaapi_coded_buffer_proxy_replace (&coded_buf, NULL);
if (sync_pic)
_free_sync_picture (encoder, sync_pic);
GST_ERROR ("encoding failed, error:%d", ret);
return ret;
}
GstVaapiEncoderStatus
gst_vaapi_encoder_get_buffer (GstVaapiEncoder * encoder,
GstVideoCodecFrame ** frame,
GstVaapiCodedBufferProxy ** codedbuf, gint64 us_of_timeout)
{
GstVaapiEncoderStatus ret = GST_VAAPI_ENCODER_STATUS_SUCCESS;
GstVaapiEncoderSyncPic *sync_pic = NULL;
GstVaapiSurfaceStatus surface_status;
GstVaapiEncPicture *picture;
ret = gst_vaapi_encoder_pop_sync_picture (encoder, &sync_pic, us_of_timeout);
if (ret != GST_VAAPI_ENCODER_STATUS_SUCCESS)
goto end;
picture = sync_pic->picture;
if (!picture->surface || !gst_vaapi_surface_sync (picture->surface)) {
ret = GST_VAAPI_ENCODER_STATUS_PARAM_ERR;
goto end;
}
if (!gst_vaapi_surface_query_status (picture->surface, &surface_status)) {
ret = GST_VAAPI_ENCODER_STATUS_PICTURE_ERR;
goto end;
}
if (frame)
*frame = gst_video_codec_frame_ref (picture->frame);
if (codedbuf)
*codedbuf = gst_vaapi_coded_buffer_proxy_ref (sync_pic->buf);
end:
if (sync_pic)
_free_sync_picture (encoder, sync_pic);
return ret;
}
GstVaapiEncoderStatus
gst_vaapi_encoder_flush (GstVaapiEncoder * encoder)
{
GstVaapiEncoderClass *const klass = GST_VAAPI_ENCODER_GET_CLASS (encoder);
return klass->flush (encoder);
}
GstVaapiEncoderStatus
gst_vaapi_encoder_get_codec_data (GstVaapiEncoder * encoder,
GstBuffer ** codec_data)
{
GstVaapiEncoderStatus ret = GST_VAAPI_ENCODER_STATUS_SUCCESS;
GstVaapiEncoderClass *const klass = GST_VAAPI_ENCODER_GET_CLASS (encoder);
*codec_data = NULL;
if (!klass->get_codec_data)
return GST_VAAPI_ENCODER_STATUS_SUCCESS;
ret = klass->get_codec_data (encoder, codec_data);
return ret;
}
static gboolean
gst_vaapi_encoder_ensure_context (GstVaapiEncoder * encoder)
{
GstVaapiEncoderClass *const klass = GST_VAAPI_ENCODER_GET_CLASS (encoder);
GstVaapiContextInfo info;
GstVaapiContext *context;
if (GST_VAAPI_ENCODER_CONTEXT (encoder))
return TRUE;
memset (&info, 0, sizeof (info));
if (!klass->get_context_info (encoder, &info))
return FALSE;
context = gst_vaapi_context_new_full (GST_VAAPI_ENCODER_DISPLAY (encoder),
&info);
if (!context)
return FALSE;
GST_VAAPI_ENCODER_CONTEXT (encoder) = context;
GST_VAAPI_ENCODER_VA_CONTEXT (encoder) = gst_vaapi_context_get_id (context);
return TRUE;
}
GstCaps *
gst_vaapi_encoder_set_format (GstVaapiEncoder * encoder,
GstVideoCodecState * in_state, GstCaps * ref_caps)
{
GstVaapiEncoderClass *const klass = GST_VAAPI_ENCODER_GET_CLASS (encoder);
GstCaps *out_caps = NULL;
if (!GST_VIDEO_INFO_WIDTH (&in_state->info) ||
!GST_VIDEO_INFO_HEIGHT (&in_state->info)) {
GST_WARNING ("encoder set format failed, width or height equal to 0.");
return NULL;
}
GST_VAAPI_ENCODER_VIDEO_INFO (encoder) = in_state->info;
out_caps = klass->set_format (encoder, in_state, ref_caps);
if (!out_caps)
goto error;
if (GST_VAAPI_ENCODER_CAPS (encoder) &&
gst_caps_is_equal (out_caps, GST_VAAPI_ENCODER_CAPS (encoder))) {
gst_caps_unref (out_caps);
return GST_VAAPI_ENCODER_CAPS (encoder);
}
gst_caps_replace (&GST_VAAPI_ENCODER_CAPS (encoder), out_caps);
g_assert (GST_VAAPI_ENCODER_CONTEXT (encoder) == NULL);
gst_vaapi_object_replace (&GST_VAAPI_ENCODER_CONTEXT (encoder), NULL);
if (!gst_vaapi_encoder_ensure_context (encoder))
goto error;
encoder->codedbuf_size = (GST_VAAPI_ENCODER_WIDTH (encoder) *
GST_VAAPI_ENCODER_HEIGHT (encoder) * 400) / (16 * 16);
encoder->codedbuf_pool = gst_vaapi_coded_buffer_pool_new (encoder,
encoder->codedbuf_size);
if (!encoder->codedbuf_pool) {
GST_ERROR ("failed to initialized coded buffer pool");
goto error;
}
gst_vaapi_video_pool_set_capacity (encoder->codedbuf_pool, 5);
return out_caps;
error:
gst_caps_replace (&GST_VAAPI_ENCODER_CAPS (encoder), NULL);
gst_caps_replace (&out_caps, NULL);
GST_ERROR ("encoder set format failed");
return NULL;
}
static gboolean
gst_vaapi_encoder_init (GstVaapiEncoder * encoder, GstVaapiDisplay * display)
{
GstVaapiEncoderClass *const klass = GST_VAAPI_ENCODER_GET_CLASS (encoder);
g_return_val_if_fail (display != NULL, FALSE);
#define CHECK_VTABLE_HOOK(FUNC) do { \
if (!klass->FUNC) \
goto error_invalid_vtable; \
} while (0)
CHECK_VTABLE_HOOK (init);
CHECK_VTABLE_HOOK (finalize);
CHECK_VTABLE_HOOK (encode);
CHECK_VTABLE_HOOK (reordering);
CHECK_VTABLE_HOOK (flush);
CHECK_VTABLE_HOOK (get_context_info);
CHECK_VTABLE_HOOK (set_format);
#undef CHECK_VTABLE_HOOK
encoder->display = gst_vaapi_display_ref (display);
encoder->va_display = gst_vaapi_display_get_display (display);
encoder->va_context = VA_INVALID_ID;
gst_video_info_init (&encoder->video_info);
g_mutex_init (&encoder->lock);
g_cond_init (&encoder->codedbuf_free);
g_cond_init (&encoder->surface_free);
g_queue_init (&encoder->sync_pictures);
g_cond_init (&encoder->sync_ready);
return klass->init (encoder);
/* ERRORS */
error_invalid_vtable:
{
GST_ERROR ("invalid subclass hook (internal error)");
return FALSE;
}
}
void
gst_vaapi_encoder_finalize (GstVaapiEncoder * encoder)
{
GstVaapiEncoderClass *const klass = GST_VAAPI_ENCODER_GET_CLASS (encoder);
klass->finalize (encoder);
gst_vaapi_encoder_free_sync_pictures (encoder);
gst_vaapi_video_pool_replace (&encoder->codedbuf_pool, NULL);
gst_vaapi_object_replace (&encoder->context, NULL);
gst_vaapi_display_replace (&encoder->display, NULL);
encoder->va_display = NULL;
g_mutex_clear (&encoder->lock);
g_cond_clear (&encoder->codedbuf_free);
g_cond_clear (&encoder->surface_free);
g_queue_clear (&encoder->sync_pictures);
g_cond_clear (&encoder->sync_ready);
}
GstVaapiEncoder *
gst_vaapi_encoder_new (const GstVaapiEncoderClass * klass,
GstVaapiDisplay * display)
{
GstVaapiEncoder *encoder;
encoder = (GstVaapiEncoder *)
gst_vaapi_mini_object_new0 (GST_VAAPI_MINI_OBJECT_CLASS (klass));
if (!encoder)
return NULL;
if (!gst_vaapi_encoder_init (encoder, display))
goto error;
return encoder;
error:
gst_vaapi_encoder_unref (encoder);
return NULL;
}