diff -r 4c282e7dd6d3 -r 5505e8908944 gst_plugins_base/gst/playback/gstscreenshot.c --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/gst_plugins_base/gst/playback/gstscreenshot.c Fri Jan 22 09:59:59 2010 +0200 @@ -0,0 +1,206 @@ +/* Small helper element for format conversion + * Copyright (C) 2005 Tim-Philipp Müller + * + * 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., 59 Temple Place - Suite 330, + * Boston, MA 02111-1307, USA. + */ + +#ifdef HAVE_CONFIG_H +#include "config.h" +#endif + +#include +#include + +#include "gstscreenshot.h" + +static void +feed_fakesrc (GstElement * src, GstBuffer * buf, GstPad * pad, gpointer data) +{ + GstBuffer *in_buf = GST_BUFFER (data); + + gst_buffer_set_caps (buf, GST_BUFFER_CAPS (in_buf)); + + memcpy (GST_BUFFER_DATA (buf), GST_BUFFER_DATA (in_buf), + GST_BUFFER_SIZE (in_buf)); + + GST_BUFFER_SIZE (buf) = GST_BUFFER_SIZE (in_buf); + + GST_DEBUG ("feeding buffer %p, size %u, caps %" GST_PTR_FORMAT, + buf, GST_BUFFER_SIZE (buf), GST_BUFFER_CAPS (buf)); +} + +static void +save_result (GstElement * sink, GstBuffer * buf, GstPad * pad, gpointer data) +{ + GstBuffer **p_buf = (GstBuffer **) data; + + *p_buf = gst_buffer_ref (buf); + + GST_DEBUG ("received converted buffer %p with caps %" GST_PTR_FORMAT, + *p_buf, GST_BUFFER_CAPS (*p_buf)); +} + +static gboolean +create_element (const gchar * factory_name, GstElement ** element, + GError ** err) +{ + *element = gst_element_factory_make (factory_name, NULL); + if (*element) + return TRUE; + + if (err && *err == NULL) { + *err = g_error_new (GST_CORE_ERROR, GST_CORE_ERROR_MISSING_PLUGIN, + "cannot create element '%s' - please check your GStreamer installation", + factory_name); + } + + return FALSE; +} + +/* takes ownership of the input buffer */ +#ifdef __SYMBIAN32__ +EXPORT_C +#endif + +GstBuffer * +gst_play_frame_conv_convert (GstBuffer * buf, GstCaps * to_caps) +{ + GstElement *src, *csp, *filter1, *vscale, *filter2, *sink, *pipeline; + GstMessage *msg; + GstBuffer *result = NULL; + GError *error = NULL; + GstBus *bus; + GstCaps *to_caps_no_par; + + g_return_val_if_fail (GST_BUFFER_CAPS (buf) != NULL, NULL); + + /* videoscale is here to correct for the pixel-aspect-ratio for us */ + GST_DEBUG ("creating elements"); + if (!create_element ("fakesrc", &src, &error) || + !create_element ("ffmpegcolorspace", &csp, &error) || + !create_element ("videoscale", &vscale, &error) || + !create_element ("capsfilter", &filter1, &error) || + !create_element ("capsfilter", &filter2, &error) || + !create_element ("fakesink", &sink, &error)) + goto no_elements; + + pipeline = gst_pipeline_new ("screenshot-pipeline"); + if (pipeline == NULL) + goto no_pipeline; + + GST_DEBUG ("adding elements"); + gst_bin_add_many (GST_BIN (pipeline), src, csp, filter1, vscale, filter2, + sink, NULL); + + g_signal_connect (src, "handoff", G_CALLBACK (feed_fakesrc), buf); + + /* set to 'fixed' sizetype */ + g_object_set (src, "sizemax", GST_BUFFER_SIZE (buf), "sizetype", 2, + "num-buffers", 1, "signal-handoffs", TRUE, NULL); + + /* adding this superfluous capsfilter makes linking cheaper */ + to_caps_no_par = gst_caps_copy (to_caps); + gst_structure_remove_field (gst_caps_get_structure (to_caps_no_par, 0), + "pixel-aspect-ratio"); + g_object_set (filter1, "caps", to_caps_no_par, NULL); + gst_caps_unref (to_caps_no_par); + + g_object_set (filter2, "caps", to_caps, NULL); + + g_signal_connect (sink, "handoff", G_CALLBACK (save_result), &result); + + g_object_set (sink, "preroll-queue-len", 1, "signal-handoffs", TRUE, NULL); + + /* FIXME: linking is still way too expensive, profile this properly */ + GST_DEBUG ("linking src->csp"); + if (!gst_element_link_pads (src, "src", csp, "sink")) + return NULL; + + GST_DEBUG ("linking csp->filter1"); + if (!gst_element_link_pads (csp, "src", filter1, "sink")) + return NULL; + + GST_DEBUG ("linking filter1->vscale"); + if (!gst_element_link_pads (filter1, "src", vscale, "sink")) + return NULL; + + GST_DEBUG ("linking vscale->capsfilter"); + if (!gst_element_link_pads (vscale, "src", filter2, "sink")) + return NULL; + + GST_DEBUG ("linking capsfilter->sink"); + if (!gst_element_link_pads (filter2, "src", sink, "sink")) + return NULL; + + GST_DEBUG ("running conversion pipeline"); + gst_element_set_state (pipeline, GST_STATE_PLAYING); + + bus = gst_element_get_bus (pipeline); + msg = + gst_bus_poll (bus, GST_MESSAGE_ERROR | GST_MESSAGE_EOS, 25 * GST_SECOND); + + if (msg) { + switch (GST_MESSAGE_TYPE (msg)) { + case GST_MESSAGE_EOS:{ + if (result) { + GST_DEBUG ("conversion successful: result = %p", result); + } else { + GST_WARNING ("EOS but no result frame?!"); + } + break; + } + case GST_MESSAGE_ERROR:{ + gchar *dbg = NULL; + + gst_message_parse_error (msg, &error, &dbg); + if (error) { + g_warning ("Could not take screenshot: %s", error->message); + GST_DEBUG ("%s [debug: %s]", error->message, GST_STR_NULL (dbg)); + g_error_free (error); + } else { + g_warning ("Could not take screenshot (and NULL error!)"); + } + g_free (dbg); + result = NULL; + break; + } + default:{ + g_return_val_if_reached (NULL); + } + } + } else { + g_warning ("Could not take screenshot: %s", "timeout during conversion"); + result = NULL; + } + + gst_element_set_state (pipeline, GST_STATE_NULL); + gst_object_unref (pipeline); + + return result; + + /* ERRORS */ +no_elements: + { + g_warning ("Could not take screenshot: %s", error->message); + g_error_free (error); + return NULL; + } +no_pipeline: + { + g_warning ("Could not take screenshot: %s", "no pipeline (unknown error)"); + return NULL; + } +}