mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-13 03:46:34 +00:00
449 lines
13 KiB
C
449 lines
13 KiB
C
/*
|
|
* GStreamer
|
|
* Copyright (C) 2015 Matthew Waters <matthew@centricular.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 <string.h>
|
|
|
|
#include "gstglbuffer.h"
|
|
#include "gstglutils.h"
|
|
|
|
/**
|
|
* SECTION:gstglbuffer
|
|
* @short_description: memory subclass for GL buffers
|
|
* @see_also: #GstMemory, #GstAllocator
|
|
*
|
|
* GstGLBuffer is a #GstMemory subclass providing support for the mapping of
|
|
* GL buffers.
|
|
*
|
|
* Data is uploaded or downloaded from the GPU as is necessary.
|
|
*/
|
|
|
|
/* Implementation notes:
|
|
*
|
|
* Currently does not take into account GLES2 differences (no mapbuffer)
|
|
*/
|
|
|
|
#define USING_OPENGL(context) (gst_gl_context_check_gl_version (context, GST_GL_API_OPENGL, 1, 0))
|
|
#define USING_OPENGL3(context) (gst_gl_context_check_gl_version (context, GST_GL_API_OPENGL3, 3, 1))
|
|
#define USING_GLES(context) (gst_gl_context_check_gl_version (context, GST_GL_API_GLES, 1, 0))
|
|
#define USING_GLES2(context) (gst_gl_context_check_gl_version (context, GST_GL_API_GLES2, 2, 0))
|
|
#define USING_GLES3(context) (gst_gl_context_check_gl_version (context, GST_GL_API_GLES2, 3, 0))
|
|
|
|
/* compatibility definitions... */
|
|
#ifndef GL_MAP_READ_BIT
|
|
#define GL_MAP_READ_BIT 0x0001
|
|
#endif
|
|
#ifndef GL_MAP_WRITE_BIT
|
|
#define GL_MAP_WRITE_BIT 0x0002
|
|
#endif
|
|
#ifndef GL_COPY_READ_BUFFER
|
|
#define GL_COPY_READ_BUFFER 0x8F36
|
|
#endif
|
|
#ifndef GL_COPY_WRITE_BUFFER
|
|
#define GL_COPY_WRITE_BUFFER 0x8F37
|
|
#endif
|
|
|
|
GST_DEBUG_CATEGORY_STATIC (GST_CAT_GL_BUFFER);
|
|
#define GST_CAT_DEFUALT GST_CAT_GL_BUFFER
|
|
|
|
static GstAllocator *_gl_buffer_allocator;
|
|
|
|
static gboolean
|
|
_gl_buffer_create (GstGLBuffer * gl_mem, GError ** error)
|
|
{
|
|
const GstGLFuncs *gl = gl_mem->mem.context->gl_vtable;
|
|
|
|
gl->GenBuffers (1, &gl_mem->id);
|
|
gl->BindBuffer (gl_mem->target, gl_mem->id);
|
|
gl->BufferData (gl_mem->target, gl_mem->mem.mem.maxsize, NULL,
|
|
gl_mem->usage_hints);
|
|
gl->BindBuffer (gl_mem->target, 0);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
struct create_data
|
|
{
|
|
GstGLBuffer *mem;
|
|
gboolean result;
|
|
};
|
|
|
|
static void
|
|
_gl_buffer_init (GstGLBuffer * mem, GstAllocator * allocator,
|
|
GstMemory * parent, GstGLContext * context, guint gl_target, guint gl_usage,
|
|
GstAllocationParams * params, gsize size)
|
|
{
|
|
mem->target = gl_target;
|
|
mem->usage_hints = gl_usage;
|
|
|
|
gst_gl_base_memory_init ((GstGLBaseMemory *) mem, allocator, parent, context,
|
|
params, size, NULL, NULL);
|
|
|
|
GST_CAT_DEBUG (GST_CAT_GL_BUFFER, "new GL buffer memory:%p size:%"
|
|
G_GSIZE_FORMAT, mem, mem->mem.mem.maxsize);
|
|
}
|
|
|
|
static GstGLBuffer *
|
|
_gl_buffer_new (GstAllocator * allocator, GstMemory * parent,
|
|
GstGLContext * context, guint gl_target, guint gl_usage,
|
|
GstAllocationParams * params, gsize size)
|
|
{
|
|
GstGLBuffer *ret = g_new0 (GstGLBuffer, 1);
|
|
_gl_buffer_init (ret, allocator, parent, context, gl_target, gl_usage,
|
|
params, size);
|
|
|
|
return ret;
|
|
}
|
|
|
|
static gpointer
|
|
gst_gl_buffer_cpu_access (GstGLBuffer * mem, GstMapInfo * info, gsize size)
|
|
{
|
|
const GstGLFuncs *gl = mem->mem.context->gl_vtable;
|
|
gpointer data, ret;
|
|
|
|
if (!gst_gl_base_memory_alloc_data (GST_GL_BASE_MEMORY_CAST (mem)))
|
|
return NULL;
|
|
|
|
ret = mem->mem.data;
|
|
|
|
GST_CAT_LOG (GST_CAT_GL_BUFFER, "mapping id %d size %" G_GSIZE_FORMAT,
|
|
mem->id, size);
|
|
|
|
/* The extra data pointer indirection/memcpy is needed for coherent across
|
|
* concurrent map()'s in both GL and CPU */
|
|
if (GST_MEMORY_FLAG_IS_SET (mem, GST_GL_BASE_MEMORY_TRANSFER_NEED_DOWNLOAD)
|
|
&& (info->flags & GST_MAP_GL) == 0 && (info->flags & GST_MAP_READ) != 0) {
|
|
gl->BindBuffer (mem->target, mem->id);
|
|
|
|
if (gl->MapBufferRange) {
|
|
/* FIXME: optionally remove this with a flag and return the
|
|
* glMapBufferRange pointer (requires
|
|
* GL_ARB_buffer_storage/GL4/GL_COHERENT_BIT) */
|
|
guint gl_map_flags = GL_MAP_READ_BIT;
|
|
|
|
data = gl->MapBufferRange (mem->target, 0, size, gl_map_flags);
|
|
|
|
if (data)
|
|
memcpy (mem->mem.data, data, size);
|
|
|
|
gl->UnmapBuffer (mem->target);
|
|
ret = mem->mem.data;
|
|
} else if (gl->GetBufferSubData) {
|
|
gl->GetBufferSubData (mem->target, 0, size, mem->mem.data);
|
|
ret = mem->mem.data;
|
|
} else {
|
|
ret = NULL;
|
|
}
|
|
gl->BindBuffer (mem->target, 0);
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
static void
|
|
gst_gl_buffer_upload_cpu_write (GstGLBuffer * mem, GstMapInfo * info,
|
|
gsize size)
|
|
{
|
|
const GstGLFuncs *gl = mem->mem.context->gl_vtable;
|
|
gpointer data;
|
|
|
|
if (!mem->mem.data)
|
|
/* no data pointer has been written */
|
|
return;
|
|
|
|
/* The extra data pointer indirection/memcpy is needed for coherent across
|
|
* concurrent map()'s in both GL and CPU */
|
|
/* FIXME: uploading potentially half-written data for libav pushing READWRITE
|
|
* mapped buffers */
|
|
if (GST_MEMORY_FLAG_IS_SET (mem, GST_GL_BASE_MEMORY_TRANSFER_NEED_UPLOAD)
|
|
|| (mem->mem.map_flags & GST_MAP_WRITE) != 0) {
|
|
gl->BindBuffer (mem->target, mem->id);
|
|
|
|
if (gl->MapBufferRange) {
|
|
/* FIXME: optionally remove this with a flag and return the
|
|
* glMapBufferRange pointer (requires
|
|
* GL_ARB_buffer_storage/GL4/GL_COHERENT_BIT) */
|
|
guint gl_map_flags = GL_MAP_WRITE_BIT;
|
|
|
|
data = gl->MapBufferRange (mem->target, 0, size, gl_map_flags);
|
|
|
|
if (data)
|
|
memcpy (data, mem->mem.data, size);
|
|
|
|
gl->UnmapBuffer (mem->target);
|
|
} else if (gl->BufferSubData) {
|
|
gl->BufferSubData (mem->target, 0, size, mem->mem.data);
|
|
}
|
|
gl->BindBuffer (mem->target, 0);
|
|
}
|
|
}
|
|
|
|
static gpointer
|
|
_gl_buffer_map (GstGLBuffer * mem, GstMapInfo * info, gsize size)
|
|
{
|
|
const GstGLFuncs *gl = mem->mem.context->gl_vtable;
|
|
|
|
if ((info->flags & GST_MAP_GL) != 0) {
|
|
if (info->flags & GST_MAP_READ) {
|
|
gst_gl_buffer_upload_cpu_write (mem, info, size);
|
|
}
|
|
gl->BindBuffer (mem->target, mem->id);
|
|
return &mem->id;
|
|
} else {
|
|
return gst_gl_buffer_cpu_access (mem, info, size);
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
static void
|
|
_gl_buffer_unmap (GstGLBuffer * mem, GstMapInfo * info)
|
|
{
|
|
const GstGLFuncs *gl = mem->mem.context->gl_vtable;
|
|
|
|
if ((info->flags & GST_MAP_GL) != 0) {
|
|
gl->BindBuffer (mem->target, 0);
|
|
}
|
|
/* XXX: optimistically transfer data */
|
|
}
|
|
|
|
static gboolean
|
|
gst_gl_buffer_copy_buffer_sub_data (GstGLBuffer * src,
|
|
GstGLBuffer * dest, gssize offset, gssize size)
|
|
{
|
|
const GstGLFuncs *gl = src->mem.context->gl_vtable;
|
|
GstMapInfo sinfo, dinfo;
|
|
|
|
if (!gl->CopyBufferSubData)
|
|
/* This is GL(ES) 3.0+ only */
|
|
return FALSE;
|
|
|
|
if (!gst_memory_map ((GstMemory *) src, &sinfo, GST_MAP_READ | GST_MAP_GL)) {
|
|
GST_CAT_WARNING (GST_CAT_GL_BUFFER,
|
|
"failed to read map source memory %p", src);
|
|
return FALSE;
|
|
}
|
|
|
|
if (!gst_memory_map ((GstMemory *) dest, &dinfo, GST_MAP_WRITE | GST_MAP_GL)) {
|
|
GST_CAT_WARNING (GST_CAT_GL_BUFFER,
|
|
"failed to write map destination memory %p", dest);
|
|
gst_memory_unmap ((GstMemory *) src, &sinfo);
|
|
return FALSE;
|
|
}
|
|
|
|
gl->BindBuffer (GL_COPY_READ_BUFFER, src->id);
|
|
gl->BindBuffer (GL_COPY_WRITE_BUFFER, dest->id);
|
|
gl->CopyBufferSubData (GL_COPY_READ_BUFFER, GL_COPY_WRITE_BUFFER,
|
|
offset, 0, size);
|
|
|
|
gst_memory_unmap ((GstMemory *) src, &sinfo);
|
|
gst_memory_unmap ((GstMemory *) dest, &dinfo);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
static GstGLBuffer *
|
|
_gl_buffer_copy (GstGLBuffer * src, gssize offset, gssize size)
|
|
{
|
|
GstAllocator *allocator = src->mem.mem.allocator;
|
|
GstAllocationParams params = { 0, src->mem.mem.align, 0, 0 };
|
|
GstGLBuffer *dest = NULL;
|
|
|
|
dest = _gl_buffer_new (allocator, NULL, src->mem.context,
|
|
src->target, src->usage_hints, ¶ms, src->mem.mem.maxsize);
|
|
|
|
/* If not doing a full copy, then copy to sysmem, the 2D represention of the
|
|
* texture would become wrong */
|
|
if (GST_MEMORY_FLAG_IS_SET (src, GST_GL_BASE_MEMORY_TRANSFER_NEED_UPLOAD)) {
|
|
if (!gst_gl_base_memory_memcpy (GST_GL_BASE_MEMORY_CAST (src),
|
|
GST_GL_BASE_MEMORY_CAST (dest), offset, size)) {
|
|
GST_CAT_WARNING (GST_CAT_GL_BUFFER, "Could not copy GL Buffer");
|
|
gst_memory_unref (GST_MEMORY_CAST (dest));
|
|
dest = NULL;
|
|
}
|
|
} else {
|
|
if (!gst_gl_buffer_copy_buffer_sub_data (src, dest, offset, size)) {
|
|
if (!gst_gl_base_memory_memcpy (GST_GL_BASE_MEMORY_CAST (src),
|
|
GST_GL_BASE_MEMORY_CAST (dest), offset, size)) {
|
|
GST_CAT_WARNING (GST_CAT_GL_BUFFER, "Could not copy GL Buffer");
|
|
gst_memory_unref (GST_MEMORY_CAST (dest));
|
|
dest = NULL;
|
|
}
|
|
}
|
|
}
|
|
|
|
return dest;
|
|
}
|
|
|
|
static GstMemory *
|
|
_gl_buffer_alloc (GstAllocator * allocator, gsize size,
|
|
GstAllocationParams * params)
|
|
{
|
|
g_critical ("Need to use gst_gl_base_memory_alloc() to allocate from "
|
|
"this allocator");
|
|
|
|
return NULL;
|
|
}
|
|
|
|
static void
|
|
_gl_buffer_destroy (GstGLBuffer * mem)
|
|
{
|
|
const GstGLFuncs *gl = mem->mem.context->gl_vtable;
|
|
|
|
gl->DeleteBuffers (1, &mem->id);
|
|
}
|
|
|
|
static void
|
|
_gst_gl_buffer_allocation_params_copy_data (GstGLBufferAllocationParams * src,
|
|
GstGLBufferAllocationParams * dest)
|
|
{
|
|
memset (dest, 0, sizeof (*dest));
|
|
|
|
gst_gl_allocation_params_copy_data (&src->parent, &dest->parent);
|
|
|
|
dest->gl_target = src->gl_target;
|
|
dest->gl_usage = src->gl_usage;
|
|
}
|
|
|
|
static void
|
|
_gst_gl_buffer_allocation_params_free_data (GstGLBufferAllocationParams *
|
|
params)
|
|
{
|
|
gst_gl_allocation_params_free_data (¶ms->parent);
|
|
}
|
|
|
|
GstGLBufferAllocationParams *
|
|
gst_gl_buffer_allocation_params_new (GstGLContext * context, gsize alloc_size,
|
|
GstAllocationParams * alloc_params, guint gl_target, guint gl_usage)
|
|
{
|
|
GstGLBufferAllocationParams *params;
|
|
|
|
g_return_val_if_fail (GST_IS_GL_CONTEXT (context), NULL);
|
|
g_return_val_if_fail (alloc_size > 0, NULL);
|
|
|
|
params = g_new0 (GstGLBufferAllocationParams, 1);
|
|
|
|
if (!gst_gl_allocation_params_init (¶ms->parent, sizeof (*params),
|
|
GST_GL_ALLOCATION_PARAMS_ALLOC_FLAG_BUFFER |
|
|
GST_GL_ALLOCATION_PARAMS_ALLOC_FLAG_ALLOC,
|
|
(GstGLAllocationParamsCopyFunc)
|
|
_gst_gl_buffer_allocation_params_copy_data,
|
|
(GstGLAllocationParamsFreeFunc)
|
|
_gst_gl_buffer_allocation_params_free_data, context, alloc_size,
|
|
alloc_params, NULL, 0, NULL, NULL)) {
|
|
g_free (params);
|
|
return NULL;
|
|
}
|
|
|
|
params->gl_target = gl_target;
|
|
params->gl_usage = gl_usage;
|
|
|
|
return params;
|
|
}
|
|
|
|
static GstGLBuffer *
|
|
_gl_buffer_alloc_mem (GstGLBufferAllocator * allocator,
|
|
GstGLBufferAllocationParams * params)
|
|
{
|
|
guint alloc_flags = params->parent.alloc_flags;
|
|
|
|
g_return_val_if_fail (alloc_flags &
|
|
GST_GL_ALLOCATION_PARAMS_ALLOC_FLAG_BUFFER, NULL);
|
|
g_return_val_if_fail (alloc_flags & GST_GL_ALLOCATION_PARAMS_ALLOC_FLAG_ALLOC,
|
|
NULL);
|
|
|
|
return _gl_buffer_new (GST_ALLOCATOR (allocator), NULL,
|
|
params->parent.context, params->gl_target, params->gl_usage,
|
|
params->parent.alloc_params, params->parent.alloc_size);
|
|
}
|
|
|
|
G_DEFINE_TYPE (GstGLBufferAllocator, gst_gl_buffer_allocator,
|
|
GST_TYPE_GL_BASE_MEMORY_ALLOCATOR);
|
|
|
|
static void
|
|
gst_gl_buffer_allocator_class_init (GstGLBufferAllocatorClass * klass)
|
|
{
|
|
GstAllocatorClass *allocator_class = (GstAllocatorClass *) klass;
|
|
GstGLBaseMemoryAllocatorClass *gl_base;
|
|
|
|
gl_base = (GstGLBaseMemoryAllocatorClass *) klass;
|
|
|
|
gl_base->alloc = (GstGLBaseMemoryAllocatorAllocFunction) _gl_buffer_alloc_mem;
|
|
gl_base->create = (GstGLBaseMemoryAllocatorCreateFunction) _gl_buffer_create;
|
|
gl_base->map = (GstGLBaseMemoryAllocatorMapFunction) _gl_buffer_map;
|
|
gl_base->unmap = (GstGLBaseMemoryAllocatorUnmapFunction) _gl_buffer_unmap;
|
|
gl_base->copy = (GstGLBaseMemoryAllocatorCopyFunction) _gl_buffer_copy;
|
|
gl_base->destroy =
|
|
(GstGLBaseMemoryAllocatorDestroyFunction) _gl_buffer_destroy;
|
|
|
|
allocator_class->alloc = _gl_buffer_alloc;
|
|
}
|
|
|
|
static void
|
|
gst_gl_buffer_allocator_init (GstGLBufferAllocator * allocator)
|
|
{
|
|
GstAllocator *alloc = GST_ALLOCATOR_CAST (allocator);
|
|
|
|
alloc->mem_type = GST_GL_BUFFER_ALLOCATOR_NAME;
|
|
|
|
GST_OBJECT_FLAG_SET (allocator, GST_ALLOCATOR_FLAG_CUSTOM_ALLOC);
|
|
}
|
|
|
|
/**
|
|
* gst_gl_buffer_init_once:
|
|
*
|
|
* Initializes the GL Buffer allocator. It is safe to call this function
|
|
* multiple times. This must be called before any other GstGLBuffer operation.
|
|
*/
|
|
void
|
|
gst_gl_buffer_init_once (void)
|
|
{
|
|
static volatile gsize _init = 0;
|
|
|
|
if (g_once_init_enter (&_init)) {
|
|
gst_gl_base_memory_init_once ();
|
|
|
|
GST_DEBUG_CATEGORY_INIT (GST_CAT_GL_BUFFER, "glbuffer", 0, "OpenGL Buffer");
|
|
|
|
_gl_buffer_allocator =
|
|
g_object_new (gst_gl_buffer_allocator_get_type (), NULL);
|
|
|
|
gst_allocator_register (GST_GL_BUFFER_ALLOCATOR_NAME,
|
|
gst_object_ref (_gl_buffer_allocator));
|
|
g_once_init_leave (&_init, 1);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* gst_is_gl_buffer:
|
|
* @mem:a #GstMemory
|
|
*
|
|
* Returns: whether the memory at @mem is a #GstGLBuffer
|
|
*/
|
|
gboolean
|
|
gst_is_gl_buffer (GstMemory * mem)
|
|
{
|
|
return mem != NULL && mem->allocator != NULL &&
|
|
g_type_is_a (G_OBJECT_TYPE (mem->allocator),
|
|
GST_TYPE_GL_BUFFER_ALLOCATOR);
|
|
}
|