gstreamer/gst-libs/gst/vaapi/gstvaapivideobuffer.c

521 lines
14 KiB
C
Raw Normal View History

2010-03-12 23:48:50 +00:00
/*
* gstvaapivideobuffer.c - Gst VA video buffer
*
* gstreamer-vaapi (C) 2010 Splitted-Desktop Systems
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
*/
2010-03-19 16:41:52 +00:00
/**
2010-03-24 08:16:32 +00:00
* SECTION:gstvaapivideobuffer
* @short_description: VA video buffer for GStreamer
2010-03-19 16:41:52 +00:00
*/
2010-03-12 23:48:50 +00:00
#include "config.h"
#include "gstvaapivideobuffer.h"
#include "gstvaapiimagepool.h"
#include "gstvaapisurfacepool.h"
#include "gstvaapiobject_priv.h"
2010-03-12 23:48:50 +00:00
#define DEBUG 1
#include "gstvaapidebug.h"
2010-03-12 23:48:50 +00:00
G_DEFINE_TYPE(GstVaapiVideoBuffer, gst_vaapi_video_buffer, GST_TYPE_BUFFER);
#define GST_VAAPI_VIDEO_BUFFER_GET_PRIVATE(obj) \
(G_TYPE_INSTANCE_GET_PRIVATE((obj), \
GST_VAAPI_TYPE_VIDEO_BUFFER, \
GstVaapiVideoBufferPrivate))
struct _GstVaapiVideoBufferPrivate {
GstVaapiDisplay *display;
GstVaapiVideoPool *image_pool;
GstVaapiImage *image;
GstVaapiVideoPool *surface_pool;
GstVaapiSurface *surface;
GstVaapiSurfaceProxy *proxy;
2010-03-12 23:48:50 +00:00
};
static void
set_display(GstVaapiVideoBuffer *buffer, GstVaapiDisplay *display)
{
GstVaapiVideoBufferPrivate * const priv = buffer->priv;
if (priv->display) {
g_object_unref(priv->display);
priv->display = NULL;
}
if (display)
priv->display = g_object_ref(display);
}
static inline void
set_image(GstVaapiVideoBuffer *buffer, GstVaapiImage *image)
{
buffer->priv->image = g_object_ref(image);
set_display(buffer, GST_VAAPI_OBJECT_DISPLAY(image));
}
static inline void
set_surface(GstVaapiVideoBuffer *buffer, GstVaapiSurface *surface)
{
buffer->priv->surface = g_object_ref(surface);
set_display(buffer, GST_VAAPI_OBJECT_DISPLAY(surface));
}
2010-03-12 23:48:50 +00:00
static void
gst_vaapi_video_buffer_destroy_image(GstVaapiVideoBuffer *buffer)
2010-03-12 23:48:50 +00:00
{
GstVaapiVideoBufferPrivate * const priv = buffer->priv;
2010-03-12 23:48:50 +00:00
if (priv->image) {
if (priv->image_pool)
gst_vaapi_video_pool_put_object(priv->image_pool, priv->image);
2010-03-12 23:48:50 +00:00
else
g_object_unref(priv->image);
priv->image = NULL;
}
if (priv->image_pool) {
g_object_unref(priv->image_pool);
priv->image_pool = NULL;
}
}
static void
gst_vaapi_video_buffer_destroy_surface(GstVaapiVideoBuffer *buffer)
{
GstVaapiVideoBufferPrivate * const priv = buffer->priv;
if (priv->proxy) {
g_object_unref(priv->proxy);
priv->proxy = NULL;
}
2010-03-12 23:48:50 +00:00
if (priv->surface) {
if (priv->surface_pool)
gst_vaapi_video_pool_put_object(priv->surface_pool, priv->surface);
2010-03-12 23:48:50 +00:00
else
g_object_unref(priv->surface);
priv->surface = NULL;
}
if (priv->surface_pool) {
g_object_unref(priv->surface_pool);
priv->surface_pool = NULL;
2010-03-12 23:48:50 +00:00
}
}
static void
gst_vaapi_video_buffer_finalize(GstMiniObject *object)
{
GstVaapiVideoBuffer * const buffer = GST_VAAPI_VIDEO_BUFFER(object);
GstMiniObjectClass *parent_class;
gst_vaapi_video_buffer_destroy_image(buffer);
gst_vaapi_video_buffer_destroy_surface(buffer);
2010-03-12 23:48:50 +00:00
set_display(buffer, NULL);
2010-03-12 23:48:50 +00:00
parent_class = GST_MINI_OBJECT_CLASS(gst_vaapi_video_buffer_parent_class);
if (parent_class->finalize)
parent_class->finalize(object);
}
static void
gst_vaapi_video_buffer_class_init(GstVaapiVideoBufferClass *klass)
{
GstMiniObjectClass * const object_class = GST_MINI_OBJECT_CLASS(klass);
g_type_class_add_private(klass, sizeof(GstVaapiVideoBufferPrivate));
object_class->finalize = gst_vaapi_video_buffer_finalize;
}
static void
gst_vaapi_video_buffer_init(GstVaapiVideoBuffer *buffer)
{
GstVaapiVideoBufferPrivate *priv;
priv = GST_VAAPI_VIDEO_BUFFER_GET_PRIVATE(buffer);
buffer->priv = priv;
priv->display = NULL;
priv->image_pool = NULL;
2010-03-12 23:48:50 +00:00
priv->image = NULL;
priv->surface_pool = NULL;
2010-03-12 23:48:50 +00:00
priv->surface = NULL;
priv->proxy = NULL;
2010-03-12 23:48:50 +00:00
}
2010-04-29 21:56:10 +00:00
/**
* gst_vaapi_video_buffer_new:
* @display: a #GstVaapiDisplay
2010-04-29 21:56:10 +00:00
*
* Creates an empty #GstBuffer. The caller is responsible for completing
* the initialization of the buffer with the gst_vaapi_video_buffer_set_*()
* functions.
*
* Return value: the newly allocated #GstBuffer, or %NULL or error
*/
static inline gpointer
_gst_vaapi_video_buffer_new(void)
2010-03-12 23:48:50 +00:00
{
2010-04-29 21:56:10 +00:00
return gst_mini_object_new(GST_VAAPI_TYPE_VIDEO_BUFFER);
}
2010-03-12 23:48:50 +00:00
2010-04-29 21:56:10 +00:00
GstBuffer *
gst_vaapi_video_buffer_new(GstVaapiDisplay *display)
2010-04-29 21:56:10 +00:00
{
GstBuffer *buffer;
g_return_val_if_fail(GST_VAAPI_IS_DISPLAY(display), NULL);
buffer = _gst_vaapi_video_buffer_new();
if (!buffer)
return NULL;
set_display(GST_VAAPI_VIDEO_BUFFER(buffer), display);
return buffer;
2010-03-12 23:48:50 +00:00
}
2010-03-19 16:41:52 +00:00
/**
* gst_vaapi_video_buffer_new_from_pool:
* @pool: a #GstVaapiVideoPool
*
* Creates a #GstBuffer with a video object allocated from a @pool.
* Only #GstVaapiSurfacePool and #GstVaapiImagePool pools are supported.
*
* The buffer is destroyed through the last call to gst_buffer_unref()
* and the video objects are pushed back to their respective pools.
*
* Return value: the newly allocated #GstBuffer, or %NULL on error
*/
2010-03-12 23:48:50 +00:00
GstBuffer *
gst_vaapi_video_buffer_new_from_pool(GstVaapiVideoPool *pool)
{
GstVaapiVideoBuffer *buffer;
gboolean is_image_pool, is_surface_pool;
2010-03-12 23:48:50 +00:00
g_return_val_if_fail(GST_VAAPI_IS_VIDEO_POOL(pool), NULL);
is_image_pool = GST_VAAPI_IS_IMAGE_POOL(pool);
is_surface_pool = GST_VAAPI_IS_SURFACE_POOL(pool);
if (!is_image_pool && !is_surface_pool)
return NULL;
2010-04-29 21:56:10 +00:00
buffer = _gst_vaapi_video_buffer_new();
if (buffer &&
((is_image_pool &&
gst_vaapi_video_buffer_set_image_from_pool(buffer, pool)) ||
(is_surface_pool &&
gst_vaapi_video_buffer_set_surface_from_pool(buffer, pool)))) {
set_display(buffer, gst_vaapi_video_pool_get_display(pool));
return GST_BUFFER(buffer);
}
gst_mini_object_unref(GST_MINI_OBJECT(buffer));
return NULL;
2010-03-12 23:48:50 +00:00
}
2010-03-19 16:41:52 +00:00
/**
* gst_vaapi_video_buffer_new_with_image:
* @image: a #GstVaapiImage
*
* Creates a #GstBuffer with the specified @image. The resulting
* buffer holds an additional reference to the @image.
*
* Return value: the newly allocated #GstBuffer, or %NULL on error
*/
2010-03-12 23:48:50 +00:00
GstBuffer *
gst_vaapi_video_buffer_new_with_image(GstVaapiImage *image)
{
GstVaapiVideoBuffer *buffer;
2010-03-12 23:48:50 +00:00
g_return_val_if_fail(GST_VAAPI_IS_IMAGE(image), NULL);
2010-04-29 21:56:10 +00:00
buffer = _gst_vaapi_video_buffer_new();
if (buffer)
gst_vaapi_video_buffer_set_image(buffer, image);
return GST_BUFFER(buffer);
2010-03-12 23:48:50 +00:00
}
2010-03-19 16:41:52 +00:00
/**
* gst_vaapi_video_buffer_new_with_surface:
* @surface: a #GstVaapiSurface
*
* Creates a #GstBuffer with the specified @surface. The resulting
* buffer holds an additional reference to the @surface.
*
* Return value: the newly allocated #GstBuffer, or %NULL on error
*/
2010-03-12 23:48:50 +00:00
GstBuffer *
gst_vaapi_video_buffer_new_with_surface(GstVaapiSurface *surface)
{
GstVaapiVideoBuffer *buffer;
2010-03-12 23:48:50 +00:00
g_return_val_if_fail(GST_VAAPI_IS_SURFACE(surface), NULL);
2010-04-29 21:56:10 +00:00
buffer = _gst_vaapi_video_buffer_new();
if (buffer)
gst_vaapi_video_buffer_set_surface(buffer, surface);
return GST_BUFFER(buffer);
2010-03-12 23:48:50 +00:00
}
/**
* gst_vaapi_video_buffer_new_with_surface_proxy:
* @proxy: a #GstVaapiSurfaceProxy
*
* Creates a #GstBuffer with the specified surface @proxy. The
* resulting buffer holds an additional reference to the @proxy.
*
* Return value: the newly allocated #GstBuffer, or %NULL on error
*/
GstBuffer *
gst_vaapi_video_buffer_new_with_surface_proxy(GstVaapiSurfaceProxy *proxy)
{
GstVaapiVideoBuffer *buffer;
g_return_val_if_fail(GST_VAAPI_IS_SURFACE_PROXY(proxy), NULL);
2010-04-29 21:56:10 +00:00
buffer = _gst_vaapi_video_buffer_new();
if (buffer)
gst_vaapi_video_buffer_set_surface_proxy(buffer, proxy);
return GST_BUFFER(buffer);
}
/**
* gst_vaapi_video_buffer_get_display:
* @buffer: a #GstVaapiVideoBuffer
*
* Retrieves the #GstVaapiDisplay the @buffer is bound to. The @buffer
* owns the returned #GstVaapiDisplay object so the caller is
* responsible for calling g_object_ref() when needed.
*
* Return value: the #GstVaapiDisplay the @buffer is bound to
*/
GstVaapiDisplay *
gst_vaapi_video_buffer_get_display(GstVaapiVideoBuffer *buffer)
{
g_return_val_if_fail(GST_VAAPI_IS_VIDEO_BUFFER(buffer), NULL);
return buffer->priv->display;
}
2010-03-19 16:41:52 +00:00
/**
* gst_vaapi_video_buffer_get_image:
* @buffer: a #GstVaapiVideoBuffer
*
* Retrieves the #GstVaapiImage bound to the @buffer. The @buffer owns
* the #GstVaapiImage so the caller is responsible for calling
* g_object_ref() when needed.
*
* Return value: the #GstVaapiImage bound to the @buffer, or %NULL if
* there is none
*/
2010-03-12 23:48:50 +00:00
GstVaapiImage *
gst_vaapi_video_buffer_get_image(GstVaapiVideoBuffer *buffer)
{
g_return_val_if_fail(GST_VAAPI_IS_VIDEO_BUFFER(buffer), NULL);
return buffer->priv->image;
}
2010-03-19 16:41:52 +00:00
/**
* gst_vaapi_video_buffer_set_image:
* @buffer: a #GstVaapiVideoBuffer
* @image: a #GstVaapiImage
*
* Binds @image to the @buffer. If the @buffer contains another image
* previously allocated from a pool, it's pushed back to its parent
* pool and the pool is also released.
*/
void
gst_vaapi_video_buffer_set_image(
GstVaapiVideoBuffer *buffer,
GstVaapiImage *image
)
{
g_return_if_fail(GST_VAAPI_IS_VIDEO_BUFFER(buffer));
g_return_if_fail(GST_VAAPI_IS_IMAGE(image));
gst_vaapi_video_buffer_destroy_image(buffer);
if (image)
set_image(buffer, image);
}
2010-03-19 16:41:52 +00:00
/**
* gst_vaapi_video_buffer_set_image_from_pool
* @buffer: a #GstVaapiVideoBuffer
* @pool: a #GstVaapiVideoPool
*
* Binds a newly allocated video object from the @pool. The @pool
* shall be of type #GstVaapiImagePool. Previously allocated objects
* are released and returned to their parent pools, if any.
*
* Return value: %TRUE on success
*/
gboolean
gst_vaapi_video_buffer_set_image_from_pool(
GstVaapiVideoBuffer *buffer,
GstVaapiVideoPool *pool
)
{
GstVaapiImage *image;
g_return_val_if_fail(GST_VAAPI_IS_VIDEO_BUFFER(buffer), FALSE);
g_return_val_if_fail(GST_VAAPI_IS_IMAGE_POOL(pool), FALSE);
gst_vaapi_video_buffer_destroy_image(buffer);
if (pool) {
image = gst_vaapi_video_pool_get_object(pool);
if (!image)
return FALSE;
set_image(buffer, image);
buffer->priv->image_pool = g_object_ref(pool);
}
return TRUE;
}
2010-03-19 16:41:52 +00:00
/**
* gst_vaapi_video_buffer_get_surface:
* @buffer: a #GstVaapiVideoBuffer
*
* Retrieves the #GstVaapiSurface bound to the @buffer. The @buffer
* owns the #GstVaapiSurface so the caller is responsible for calling
* g_object_ref() when needed.
*
* Return value: the #GstVaapiSurface bound to the @buffer, or %NULL if
* there is none
*/
2010-03-12 23:48:50 +00:00
GstVaapiSurface *
gst_vaapi_video_buffer_get_surface(GstVaapiVideoBuffer *buffer)
{
g_return_val_if_fail(GST_VAAPI_IS_VIDEO_BUFFER(buffer), NULL);
return buffer->priv->surface;
}
2010-03-19 16:41:52 +00:00
/**
* gst_vaapi_video_buffer_set_surface:
* @buffer: a #GstVaapiVideoBuffer
* @surface: a #GstVaapiSurface
*
* Binds @surface to the @buffer. If the @buffer contains another
* surface previously allocated from a pool, it's pushed back to its
* parent pool and the pool is also released.
*/
void
gst_vaapi_video_buffer_set_surface(
GstVaapiVideoBuffer *buffer,
GstVaapiSurface *surface
)
{
g_return_if_fail(GST_VAAPI_IS_VIDEO_BUFFER(buffer));
g_return_if_fail(GST_VAAPI_IS_SURFACE(surface));
gst_vaapi_video_buffer_destroy_surface(buffer);
if (surface)
set_surface(buffer, surface);
}
2010-03-19 16:41:52 +00:00
/**
* gst_vaapi_video_buffer_set_surface_from_pool
* @buffer: a #GstVaapiVideoBuffer
* @pool: a #GstVaapiVideoPool
*
* Binds a newly allocated video object from the @pool. The @pool
* shall be of type #GstVaapiSurfacePool. Previously allocated objects
* are released and returned to their parent pools, if any.
*
* Return value: %TRUE on success
*/
gboolean
gst_vaapi_video_buffer_set_surface_from_pool(
GstVaapiVideoBuffer *buffer,
GstVaapiVideoPool *pool
)
{
GstVaapiSurface *surface;
g_return_val_if_fail(GST_VAAPI_IS_VIDEO_BUFFER(buffer), FALSE);
g_return_val_if_fail(GST_VAAPI_IS_SURFACE_POOL(pool), FALSE);
gst_vaapi_video_buffer_destroy_surface(buffer);
if (pool) {
surface = gst_vaapi_video_pool_get_object(pool);
if (!surface)
return FALSE;
set_surface(buffer, surface);
buffer->priv->surface_pool = g_object_ref(pool);
}
return TRUE;
}
/**
* gst_vaapi_video_buffer_get_surface_proxy:
* @buffer: a #GstVaapiVideoBuffer
*
* Retrieves the #GstVaapiSurfaceProxy bound to the @buffer. The @buffer
* owns the #GstVaapiSurfaceProxy so the caller is responsible for calling
* g_object_ref() when needed.
*
* Return value: the #GstVaapiSurfaceProxy bound to the @buffer, or
* %NULL if there is none
*/
GstVaapiSurfaceProxy *
gst_vaapi_video_buffer_get_surface_proxy(GstVaapiVideoBuffer *buffer)
{
g_return_val_if_fail(GST_VAAPI_IS_VIDEO_BUFFER(buffer), NULL);
return buffer->priv->proxy;
}
/**
* gst_vaapi_video_buffer_set_surface_proxy:
* @buffer: a #GstVaapiVideoBuffer
* @proxy: a #GstVaapiSurfaceProxy
*
* Binds surface @proxy to the @buffer. If the @buffer contains another
* surface previously allocated from a pool, it's pushed back to its
* parent pool and the pool is also released.
*/
void
gst_vaapi_video_buffer_set_surface_proxy(
GstVaapiVideoBuffer *buffer,
GstVaapiSurfaceProxy *proxy
)
{
GstVaapiSurface *surface;
g_return_if_fail(GST_VAAPI_IS_VIDEO_BUFFER(buffer));
g_return_if_fail(GST_VAAPI_IS_SURFACE_PROXY(proxy));
gst_vaapi_video_buffer_destroy_surface(buffer);
if (proxy) {
surface = GST_VAAPI_SURFACE_PROXY_SURFACE(proxy);
if (!surface)
return;
set_surface(buffer, surface);
buffer->priv->proxy = g_object_ref(proxy);
}
}