mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-24 01:00:37 +00:00
acd33a240f
Previously it was created in the init function and destroyed in ::stop, which lead to segfaults when reusing the element. Now the upload object is created in ::transform_caps if it is NULL, which is the earliest we need it. The other vfuncs already bail out if the upload object is NULL, which means that negotiation wasn't done.
255 lines
8 KiB
C
255 lines
8 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 <stdio.h>
|
|
|
|
#include <gst/gl/gl.h>
|
|
#include "gstgluploadelement.h"
|
|
|
|
GST_DEBUG_CATEGORY_STATIC (gst_gl_upload_element_debug);
|
|
#define GST_CAT_DEFAULT gst_gl_upload_element_debug
|
|
|
|
#define gst_gl_upload_element_parent_class parent_class
|
|
G_DEFINE_TYPE_WITH_CODE (GstGLUploadElement, gst_gl_upload_element,
|
|
GST_TYPE_GL_BASE_FILTER,
|
|
GST_DEBUG_CATEGORY_INIT (gst_gl_upload_element_debug, "gluploadelement", 0,
|
|
"glupload Element"););
|
|
|
|
static gboolean gst_gl_upload_element_get_unit_size (GstBaseTransform * trans,
|
|
GstCaps * caps, gsize * size);
|
|
static GstCaps *_gst_gl_upload_element_transform_caps (GstBaseTransform * bt,
|
|
GstPadDirection direction, GstCaps * caps, GstCaps * filter);
|
|
static gboolean _gst_gl_upload_element_set_caps (GstBaseTransform * bt,
|
|
GstCaps * in_caps, GstCaps * out_caps);
|
|
static gboolean gst_gl_upload_element_filter_meta (GstBaseTransform * trans,
|
|
GstQuery * query, GType api, const GstStructure * params);
|
|
static gboolean _gst_gl_upload_element_propose_allocation (GstBaseTransform *
|
|
bt, GstQuery * decide_query, GstQuery * query);
|
|
static gboolean _gst_gl_upload_element_decide_allocation (GstBaseTransform *
|
|
trans, GstQuery * query);
|
|
static GstFlowReturn
|
|
gst_gl_upload_element_prepare_output_buffer (GstBaseTransform * bt,
|
|
GstBuffer * buffer, GstBuffer ** outbuf);
|
|
static GstFlowReturn gst_gl_upload_element_transform (GstBaseTransform * bt,
|
|
GstBuffer * buffer, GstBuffer * outbuf);
|
|
static gboolean gst_gl_upload_element_stop (GstBaseTransform * bt);
|
|
|
|
static GstStaticPadTemplate gst_gl_upload_element_src_pad_template =
|
|
GST_STATIC_PAD_TEMPLATE ("src",
|
|
GST_PAD_SRC,
|
|
GST_PAD_ALWAYS,
|
|
GST_STATIC_CAPS ("video/x-raw(ANY)"));
|
|
|
|
static void
|
|
gst_gl_upload_element_class_init (GstGLUploadElementClass * klass)
|
|
{
|
|
GstBaseTransformClass *bt_class = GST_BASE_TRANSFORM_CLASS (klass);
|
|
GstElementClass *element_class = GST_ELEMENT_CLASS (klass);
|
|
GstCaps *upload_caps;
|
|
|
|
bt_class->transform_caps = _gst_gl_upload_element_transform_caps;
|
|
bt_class->set_caps = _gst_gl_upload_element_set_caps;
|
|
bt_class->filter_meta = gst_gl_upload_element_filter_meta;
|
|
bt_class->propose_allocation = _gst_gl_upload_element_propose_allocation;
|
|
bt_class->decide_allocation = _gst_gl_upload_element_decide_allocation;
|
|
bt_class->get_unit_size = gst_gl_upload_element_get_unit_size;
|
|
bt_class->prepare_output_buffer = gst_gl_upload_element_prepare_output_buffer;
|
|
bt_class->transform = gst_gl_upload_element_transform;
|
|
bt_class->stop = gst_gl_upload_element_stop;
|
|
|
|
bt_class->passthrough_on_same_caps = TRUE;
|
|
|
|
gst_element_class_add_static_pad_template (element_class,
|
|
&gst_gl_upload_element_src_pad_template);
|
|
|
|
upload_caps = gst_gl_upload_get_input_template_caps ();
|
|
gst_element_class_add_pad_template (element_class,
|
|
gst_pad_template_new ("sink", GST_PAD_SINK, GST_PAD_ALWAYS, upload_caps));
|
|
gst_caps_unref (upload_caps);
|
|
|
|
gst_element_class_set_metadata (element_class,
|
|
"OpenGL uploader", "Filter/Video",
|
|
"Uploads data into OpenGL", "Matthew Waters <matthew@centricular.com>");
|
|
}
|
|
|
|
static void
|
|
gst_gl_upload_element_init (GstGLUploadElement * upload)
|
|
{
|
|
gst_base_transform_set_prefer_passthrough (GST_BASE_TRANSFORM (upload), TRUE);
|
|
}
|
|
|
|
static gboolean
|
|
gst_gl_upload_element_stop (GstBaseTransform * bt)
|
|
{
|
|
GstGLUploadElement *upload = GST_GL_UPLOAD_ELEMENT (bt);
|
|
|
|
if (upload->upload) {
|
|
gst_object_unref (upload->upload);
|
|
upload->upload = NULL;
|
|
}
|
|
|
|
gst_caps_replace (&upload->in_caps, NULL);
|
|
gst_caps_replace (&upload->out_caps, NULL);
|
|
|
|
return GST_BASE_TRANSFORM_CLASS (parent_class)->stop (bt);
|
|
}
|
|
|
|
static gboolean
|
|
gst_gl_upload_element_get_unit_size (GstBaseTransform * trans, GstCaps * caps,
|
|
gsize * size)
|
|
{
|
|
gboolean ret = FALSE;
|
|
GstVideoInfo info;
|
|
|
|
ret = gst_video_info_from_caps (&info, caps);
|
|
if (ret)
|
|
*size = GST_VIDEO_INFO_SIZE (&info);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
static GstCaps *
|
|
_gst_gl_upload_element_transform_caps (GstBaseTransform * bt,
|
|
GstPadDirection direction, GstCaps * caps, GstCaps * filter)
|
|
{
|
|
GstGLUploadElement *upload = GST_GL_UPLOAD_ELEMENT (bt);
|
|
GstGLContext *context = GST_GL_BASE_FILTER (bt)->context;
|
|
|
|
if (upload->upload == NULL)
|
|
upload->upload = gst_gl_upload_new (NULL);
|
|
|
|
return gst_gl_upload_transform_caps (upload->upload, context, direction, caps,
|
|
filter);
|
|
}
|
|
|
|
static gboolean
|
|
gst_gl_upload_element_filter_meta (GstBaseTransform * trans, GstQuery * query,
|
|
GType api, const GstStructure * params)
|
|
{
|
|
/* propose all metadata upstream */
|
|
return TRUE;
|
|
}
|
|
|
|
static gboolean
|
|
_gst_gl_upload_element_propose_allocation (GstBaseTransform * bt,
|
|
GstQuery * decide_query, GstQuery * query)
|
|
{
|
|
GstGLUploadElement *upload = GST_GL_UPLOAD_ELEMENT (bt);
|
|
gboolean ret;
|
|
|
|
if (!upload->upload)
|
|
return FALSE;
|
|
|
|
ret = GST_BASE_TRANSFORM_CLASS (parent_class)->propose_allocation (bt,
|
|
decide_query, query);
|
|
gst_gl_upload_propose_allocation (upload->upload, decide_query, query);
|
|
|
|
return ret;
|
|
}
|
|
|
|
static gboolean
|
|
_gst_gl_upload_element_decide_allocation (GstBaseTransform * trans,
|
|
GstQuery * query)
|
|
{
|
|
GstGLUploadElement *upload = GST_GL_UPLOAD_ELEMENT (trans);
|
|
GstGLContext *context;
|
|
gboolean ret;
|
|
|
|
ret =
|
|
GST_BASE_TRANSFORM_CLASS
|
|
(gst_gl_upload_element_parent_class)->decide_allocation (trans, query);
|
|
if (!ret)
|
|
return FALSE;
|
|
|
|
/* GstGLBaseFilter populates ->context in ::decide_allocation so now it's the
|
|
* time to set the ->upload context */
|
|
context = GST_GL_BASE_FILTER (trans)->context;
|
|
gst_gl_upload_set_context (upload->upload, context);
|
|
|
|
return gst_gl_upload_set_caps (upload->upload, upload->in_caps,
|
|
upload->out_caps);
|
|
}
|
|
|
|
static gboolean
|
|
_gst_gl_upload_element_set_caps (GstBaseTransform * bt, GstCaps * in_caps,
|
|
GstCaps * out_caps)
|
|
{
|
|
GstGLUploadElement *upload = GST_GL_UPLOAD_ELEMENT (bt);
|
|
|
|
gst_caps_replace (&upload->in_caps, in_caps);
|
|
gst_caps_replace (&upload->out_caps, out_caps);
|
|
|
|
if (upload->upload)
|
|
return gst_gl_upload_set_caps (upload->upload, in_caps, out_caps);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
GstFlowReturn
|
|
gst_gl_upload_element_prepare_output_buffer (GstBaseTransform * bt,
|
|
GstBuffer * buffer, GstBuffer ** outbuf)
|
|
{
|
|
GstGLUploadElement *upload = GST_GL_UPLOAD_ELEMENT (bt);
|
|
GstGLUploadReturn ret;
|
|
GstBaseTransformClass *bclass;
|
|
|
|
bclass = GST_BASE_TRANSFORM_GET_CLASS (bt);
|
|
|
|
if (gst_base_transform_is_passthrough (bt)) {
|
|
*outbuf = buffer;
|
|
return GST_FLOW_OK;
|
|
}
|
|
|
|
if (!upload->upload)
|
|
return GST_FLOW_NOT_NEGOTIATED;
|
|
|
|
ret = gst_gl_upload_perform_with_buffer (upload->upload, buffer, outbuf);
|
|
if (ret == GST_GL_UPLOAD_RECONFIGURE) {
|
|
gst_base_transform_reconfigure_src (bt);
|
|
return GST_FLOW_OK;
|
|
}
|
|
|
|
if (ret != GST_GL_UPLOAD_DONE || *outbuf == NULL) {
|
|
GST_ELEMENT_ERROR (bt, RESOURCE, NOT_FOUND, ("%s",
|
|
"Failed to upload buffer"), (NULL));
|
|
if (*outbuf)
|
|
gst_buffer_unref (*outbuf);
|
|
return GST_FLOW_ERROR;
|
|
}
|
|
|
|
/* basetransform doesn't unref if they're the same */
|
|
if (buffer == *outbuf)
|
|
gst_buffer_unref (*outbuf);
|
|
else
|
|
bclass->copy_metadata (bt, buffer, *outbuf);
|
|
|
|
return GST_FLOW_OK;
|
|
}
|
|
|
|
static GstFlowReturn
|
|
gst_gl_upload_element_transform (GstBaseTransform * bt, GstBuffer * buffer,
|
|
GstBuffer * outbuf)
|
|
{
|
|
return GST_FLOW_OK;
|
|
}
|