--- /dev/null
+/* Small helper element for format conversion
+ * (c) 2004 Ronald Bultje <rbultje@ronald.bitfreak.net>
+ * Portion Copyright © 2009 Nokia Corporation and/or its
+ * subsidiary(-ies).* All rights reserved. *
+ *
+ * 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 <gst/gst.h>
+#include <string.h>
+
+#include "gstscreenshot.h"
+
+typedef struct {
+ GstBuffer *result;
+ GstElement *src;
+ GstElement *sink;
+ GstElement *pipeline;
+ BvwFrameConvCb cb;
+ gpointer cb_data;
+} GstScreenshotData;
+
+/* GST_DEBUG_CATEGORY_EXTERN (_totem_gst_debug_cat); */
+/* #define GST_CAT_DEFAULT _totem_gst_debug_cat */
+
+static void feed_fakesrc(GstElement *src, GstBuffer *buf, GstPad *pad,
+ gpointer data)
+{
+ GstBuffer *in_buf = GST_BUFFER(data);
+
+ g_assert(GST_BUFFER_SIZE(buf) >= GST_BUFFER_SIZE(in_buf));
+ g_assert(!GST_BUFFER_FLAG_IS_SET(buf, GST_BUFFER_FLAG_READONLY));
+
+ 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));
+
+ gst_buffer_unref(in_buf);
+}
+
+static void save_result(GstElement *sink, GstBuffer *buf, GstPad *pad,
+ gpointer data)
+{
+ GstScreenshotData *gsd = data;
+
+ gsd->result = gst_buffer_ref(buf);
+
+ GST_DEBUG("received converted buffer %p with caps %" GST_PTR_FORMAT,
+ gsd->result, GST_BUFFER_CAPS(gsd->result));
+}
+
+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;
+}
+
+static gboolean finalize_process(GstScreenshotData *gsd)
+{
+ g_signal_handlers_disconnect_matched(gsd->sink, (GSignalMatchType)
+ G_SIGNAL_MATCH_FUNC, 0, 0, NULL,
+ save_result, NULL);
+ g_signal_handlers_disconnect_matched(gsd->src, (GSignalMatchType)
+ G_SIGNAL_MATCH_FUNC, 0, 0, NULL,
+ feed_fakesrc, NULL);
+ gst_element_set_state(gsd->pipeline, GST_STATE_NULL);
+
+ g_free(gsd);
+
+ return FALSE;
+}
+
+static gboolean async_bus_handler(GstBus *bus, GstMessage *msg,
+ gpointer data)
+{
+ GstScreenshotData *gsd = data;
+ gboolean keep_watch = TRUE;
+
+ switch (GST_MESSAGE_TYPE(msg)) {
+ case GST_MESSAGE_EOS: {
+ if (gsd->result != NULL) {
+ GST_DEBUG("conversion successful: result = %p",
+ gsd->result);
+ } else {
+ GST_WARNING("EOS but no result frame?!");
+ }
+ gsd->cb(gsd->result, gsd->cb_data);
+ keep_watch = finalize_process(gsd);
+ break;
+ }
+ case GST_MESSAGE_ERROR: {
+ gchar *dbg = NULL;
+ GError *error = NULL;
+
+ gst_message_parse_error(msg, &error, &dbg);
+ if (error != NULL) {
+ 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);
+ gsd->result = NULL;
+ gsd->cb(gsd->result, gsd->cb_data);
+ keep_watch = finalize_process(gsd);
+ break;
+ }
+ default:
+ break;
+ }
+
+ return keep_watch;
+}
+
+/* takes ownership of the input buffer */
+gboolean bvw_frame_conv_convert(GstBuffer *buf, GstCaps *to_caps,
+ BvwFrameConvCb cb, gpointer cb_data)
+{
+ static GstElement *src = NULL, *sink = NULL, *pipeline = NULL,
+ *filter1 = NULL, *filter2 = NULL;
+ static GstBus *bus;
+ GError *error = NULL;
+ GstCaps *to_caps_no_par;
+ GstScreenshotData *gsd;
+
+ g_return_val_if_fail(GST_BUFFER_CAPS(buf) != NULL, FALSE);
+ g_return_val_if_fail(cb != NULL, FALSE);
+
+ if (pipeline == NULL) {
+ GstElement *csp, *vscale;
+
+ pipeline = gst_pipeline_new("screenshot-pipeline");
+ if(pipeline == NULL) {
+ g_warning("Could not take screenshot: "
+ "no pipeline (unknown error)");
+ return FALSE;
+ }
+
+ /* 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)) {
+ g_warning("Could not take screenshot: %s",
+ error->message);
+ g_error_free(error);
+ return FALSE;
+ }
+
+ GST_DEBUG("adding elements");
+ gst_bin_add_many(GST_BIN(pipeline), src, csp, filter1, vscale,
+ filter2, sink, NULL);
+
+ g_object_set(sink, "preroll-queue-len", 1,
+ "signal-handoffs", TRUE, NULL);
+
+ /* set to 'fixed' sizetype */
+ g_object_set(src, "sizetype", 2, "num-buffers", 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 FALSE;
+
+ GST_DEBUG("linking csp->filter1");
+ if(!gst_element_link_pads(csp, "src", filter1, "sink"))
+ return FALSE;
+
+ GST_DEBUG("linking filter1->vscale");
+ if(!gst_element_link_pads(filter1, "src", vscale, "sink"))
+ return FALSE;
+
+ GST_DEBUG("linking vscale->capsfilter");
+ if(!gst_element_link_pads(vscale, "src", filter2, "sink"))
+ return FALSE;
+
+ GST_DEBUG("linking capsfilter->sink");
+ if(!gst_element_link_pads(filter2, "src", sink, "sink"))
+ return FALSE;
+
+ bus = gst_element_get_bus(pipeline);
+ }
+
+ /* 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);
+ gst_caps_unref(to_caps);
+
+ gsd = g_new0(GstScreenshotData, 1);
+
+ gsd->src = src;
+ gsd->sink = sink;
+ gsd->pipeline = pipeline;
+ gsd->cb = cb;
+ gsd->cb_data = cb_data;
+
+ g_signal_connect(sink, "handoff", G_CALLBACK(save_result), gsd);
+
+ g_signal_connect(src, "handoff", G_CALLBACK(feed_fakesrc), buf);
+
+ gst_bus_add_watch(bus, async_bus_handler, gsd);
+
+ /* set to 'fixed' sizetype */
+ g_object_set(src, "sizemax", GST_BUFFER_SIZE(buf), NULL);
+
+ GST_DEBUG("running conversion pipeline");
+ gst_element_set_state(pipeline, GST_STATE_PLAYING);
+
+ return TRUE;
+}