tests/icles/gdkpixbufsink-test.c

   1 /* GStreamer interactive test for the gdkpixbufsink element
   2  * Copyright (C) 2008 Tim-Philipp Müller <tim centricular net>
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Library General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Library General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Library General Public
  15  * License along with this library; if not, write to the
  16  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  17  * Boston, MA 02111-1307, USA.
  18  */
  19
  20 #ifdef HAVE_CONFIG_H
  21 # include "config.h"
  22 #endif
  23
  24 #include <gst/gst.h>
  25 #include <gtk/gtk.h>
  26
  27 typedef struct
  28 {
  29   GstElement *pipe;
  30   GstElement *sink;
  31   gboolean got_video;
  32
  33   GtkWidget *win;
  34   GtkWidget *img;
  35   GtkWidget *slider;
  36   GtkWidget *accurate_cb;
  37
  38   gboolean accurate;            /* whether to try accurate seeks */
  39
  40   gint64 cur_pos;
  41   gboolean prerolled;
  42 } AppInfo;
  43
  44 static void seek_to (AppInfo * info, gdouble percent);
  45
  46 static GstElement *
  47 create_element (const gchar * factory_name)
  48 {
  49   GstElement *element;
  50
  51   element = gst_element_factory_make (factory_name, NULL);
  52
  53   if (element == NULL)
  54     g_error ("Failed to create '%s' element", factory_name);
  55
  56   return element;
  57 }
  58
  59 static void
  60 new_decoded_pad (GstElement * dec, GstPad * new_pad, gboolean last,
  61     AppInfo * info)
  62 {
  63   const gchar *sname;
  64   GstElement *csp, *scale, *filter;
  65   GstStructure *s;
  66   GstCaps *caps;
  67   GstPad *sinkpad;
  68
  69   /* already found a video stream? */
  70   if (info->got_video)
  71     return;
  72
  73   /* FIXME: is this racy or does decodebin2 make sure caps are always
  74    * negotiated at this point? */
  75   caps = gst_pad_query_caps (new_pad, NULL);
  76   g_return_if_fail (caps != NULL);
  77
  78   s = gst_caps_get_structure (caps, 0);
  79   sname = gst_structure_get_name (s);
  80   if (!g_str_has_prefix (sname, "video/x-raw-"))
  81     goto not_video;
  82
  83   csp = create_element ("ffmpegcolorspace");
  84   scale = create_element ("videoscale");
  85   filter = create_element ("capsfilter");
  86   info->sink = create_element ("gdkpixbufsink");
  87   g_object_set (info->sink, "qos", FALSE, "max-lateness", (gint64) - 1, NULL);
  88
  89   gst_bin_add_many (GST_BIN (info->pipe), csp, scale, filter, info->sink, NULL);
  90
  91   sinkpad = gst_element_get_static_pad (csp, "sink");
  92   if (GST_PAD_LINK_FAILED (gst_pad_link (new_pad, sinkpad)))
  93     g_error ("Can't link new decoded pad to ffmpegcolorspace's sink pad");
  94   gst_object_unref (sinkpad);
  95
  96   if (!gst_element_link (csp, scale))
  97     g_error ("Can't link ffmpegcolorspace to videoscale");
  98   if (!gst_element_link (scale, filter))
  99     g_error ("Can't link videoscale to capsfilter");
 100   if (!gst_element_link (filter, info->sink))
 101     g_error ("Can't link capsfilter to gdkpixbufsink");
 102
 103   gst_element_set_state (info->sink, GST_STATE_PAUSED);
 104   gst_element_set_state (filter, GST_STATE_PAUSED);
 105   gst_element_set_state (scale, GST_STATE_PAUSED);
 106   gst_element_set_state (csp, GST_STATE_PAUSED);
 107
 108   info->got_video = TRUE;
 109   return;
 110
 111 not_video:
 112   {
 113     if (last) {
 114       g_error ("This file does not contain a video track, or you do not have "
 115           "the necessary decoder(s) installed");
 116     }
 117   }
 118 }
 119
 120 static void
 121 bus_message_cb (GstBus * bus, GstMessage * msg, AppInfo * info)
 122 {
 123   switch (GST_MESSAGE_TYPE (msg)) {
 124     case GST_MESSAGE_ASYNC_DONE:{
 125       /* only interested in async-done messages from the top-level pipeline */
 126       if (msg->src != GST_OBJECT_CAST (info->pipe))
 127         break;
 128
 129       if (!info->prerolled) {
 130         /* make slider visible if it's not visible already */
 131         gtk_widget_show (info->slider);
 132
 133         /* initial frame is often black, so seek to beginning plus a bit */
 134         seek_to (info, 0.001);
 135         info->prerolled = TRUE;
 136       }
 137
 138       /* update position */
 139       if (!gst_element_query_position (info->pipe, GST_FORMAT_TIME,
 140               &info->cur_pos))
 141         info->cur_pos = -1;
 142       break;
 143     }
 144     case GST_MESSAGE_ELEMENT:{
 145       const GValue *val;
 146       GdkPixbuf *pixbuf = NULL;
 147       const GstStructure *structure;
 148
 149       /* only interested in element messages from our gdkpixbufsink */
 150       if (msg->src != GST_OBJECT_CAST (info->sink))
 151         break;
 152
 153       /* only interested in these two messages */
 154       if (!gst_message_has_name (msg, "preroll-pixbuf") &&
 155           !gst_message_has_name (msg, "pixbuf")) {
 156         break;
 157       }
 158
 159       g_print ("pixbuf\n");
 160       structure = gst_message_get_structure (msg);
 161       val = gst_structure_get_value (structure, "pixbuf");
 162       g_return_if_fail (val != NULL);
 163
 164       pixbuf = GDK_PIXBUF (g_value_dup_object (val));
 165       gtk_image_set_from_pixbuf (GTK_IMAGE (info->img), pixbuf);
 166       g_object_unref (pixbuf);
 167       break;
 168     }
 169     case GST_MESSAGE_ERROR:{
 170       GError *err = NULL;
 171       gchar *dbg = NULL;
 172
 173       gst_message_parse_error (msg, &err, &dbg);
 174       g_error ("Error: %s\n%s\n", err->message, (dbg) ? dbg : "");
 175       g_error_free (err);
 176       g_free (dbg);
 177       break;
 178     }
 179     default:
 180       break;
 181   }
 182 }
 183
 184 static gboolean
 185 create_pipeline (AppInfo * info, const gchar * filename)
 186 {
 187   GstElement *src, *dec;
 188   GstBus *bus;
 189
 190   info->pipe = gst_pipeline_new ("pipeline");
 191   src = create_element ("filesrc");
 192   g_object_set (src, "location", filename, NULL);
 193
 194   dec = create_element ("decodebin2");
 195
 196   gst_bin_add_many (GST_BIN (info->pipe), src, dec, NULL);
 197   if (!gst_element_link (src, dec))
 198     g_error ("Can't link filesrc to decodebin2");
 199
 200   g_signal_connect (dec, "new-decoded-pad", G_CALLBACK (new_decoded_pad), info);
 201
 202   /* set up bus */
 203   bus = gst_element_get_bus (info->pipe);
 204   gst_bus_add_signal_watch (bus);
 205   g_signal_connect (bus, "message", G_CALLBACK (bus_message_cb), info);
 206   gst_object_unref (bus);
 207
 208   return TRUE;
 209 }
 210
 211 static void
 212 seek_to (AppInfo * info, gdouble percent)
 213 {
 214   GstSeekFlags seek_flags;
 215   gint64 seek_pos, dur = -1;
 216
 217   if (!gst_element_query_duration (info->pipe, GST_FORMAT_TIME, &dur)
 218       || dur <= 0) {
 219     g_printerr ("Could not query duration\n");
 220     return;
 221   }
 222
 223   seek_pos = gst_gdouble_to_guint64 (gst_guint64_to_gdouble (dur) * percent);
 224   g_print ("Seeking to %" GST_TIME_FORMAT ", accurate: %d\n",
 225       GST_TIME_ARGS (seek_pos), info->accurate);
 226
 227   seek_flags = GST_SEEK_FLAG_FLUSH;
 228
 229   if (info->accurate)
 230     seek_flags |= GST_SEEK_FLAG_ACCURATE;
 231   else
 232     seek_flags |= GST_SEEK_FLAG_KEY_UNIT;
 233
 234   if (!gst_element_seek_simple (info->pipe, GST_FORMAT_TIME, seek_flags,
 235           seek_pos)) {
 236     g_printerr ("Seek failed.\n");
 237     return;
 238   }
 239 }
 240
 241 static void
 242 slider_cb (GtkRange * range, AppInfo * info)
 243 {
 244   gdouble val;
 245
 246   val = gtk_range_get_value (range);
 247   seek_to (info, val);
 248 }
 249
 250 static gchar *
 251 slider_format_value_cb (GtkScale * scale, gdouble value, AppInfo * info)
 252 {
 253   gchar s[64];
 254
 255   if (info->cur_pos < 0)
 256     return g_strdup_printf ("%0.1g%%", value * 100.0);
 257
 258   g_snprintf (s, 64, "%" GST_TIME_FORMAT, GST_TIME_ARGS (info->cur_pos));
 259   s[10] = '\0';
 260   return g_strdup (s);
 261 }
 262
 263 static void
 264 accurate_toggled_cb (GtkToggleButton * toggle, AppInfo * info)
 265 {
 266   info->accurate = gtk_toggle_button_get_active (toggle);
 267 }
 268
 269 static void
 270 run_gui (const gchar * filename)
 271 {
 272   GtkWidget *vbox, *hbox;
 273   AppInfo *info;
 274
 275   info = g_new0 (AppInfo, 1);
 276
 277   /* create pipeline */
 278   if (!create_pipeline (info, filename))
 279     goto done;
 280
 281   /* create window */
 282   info->win = gtk_window_new (GTK_WINDOW_TOPLEVEL);
 283   g_signal_connect (info->win, "delete-event", G_CALLBACK (gtk_main_quit),
 284       NULL);
 285
 286   vbox = gtk_box_new (GTK_ORIENTATION_VERTICAL, 6);
 287   gtk_container_set_border_width (GTK_CONTAINER (vbox), 12);
 288   gtk_container_add (GTK_CONTAINER (info->win), vbox);
 289
 290   info->img = gtk_image_new ();
 291   gtk_box_pack_start (GTK_BOX (vbox), info->img, FALSE, FALSE, 6);
 292
 293   hbox = gtk_box_new (GTK_ORIENTATION_HORIZONTAL, 6);
 294   gtk_box_pack_start (GTK_BOX (vbox), hbox, FALSE, FALSE, 6);
 295
 296   info->accurate_cb = gtk_check_button_new_with_label ("accurate seek "
 297       "(might not work reliably with all demuxers)");
 298   gtk_box_pack_start (GTK_BOX (hbox), info->accurate_cb, FALSE, FALSE, 6);
 299   g_signal_connect (info->accurate_cb, "toggled",
 300       G_CALLBACK (accurate_toggled_cb), info);
 301
 302   info->slider = gtk_scale_new_with_range (GTK_ORIENTATION_HORIZONTAL,
 303       0.0, 1.0, 0.001);
 304   gtk_box_pack_start (GTK_BOX (vbox), info->slider, FALSE, FALSE, 6);
 305   g_signal_connect (info->slider, "value-changed",
 306       G_CALLBACK (slider_cb), info);
 307   g_signal_connect (info->slider, "format-value",
 308       G_CALLBACK (slider_format_value_cb), info);
 309
 310   /* and go! */
 311   gst_element_set_state (info->pipe, GST_STATE_PAUSED);
 312
 313   gtk_widget_show_all (info->win);
 314   gtk_widget_hide (info->slider);       /* hide until we're prerolled */
 315   gtk_main ();
 316
 317 done:
 318
 319   g_free (info);
 320 }
 321
 322 static gchar **filenames = NULL;
 323
 324 int
 325 main (int argc, char **argv)
 326 {
 327   static const GOptionEntry test_goptions[] = {
 328     {G_OPTION_REMAINING, 0, 0, G_OPTION_ARG_FILENAME_ARRAY, &filenames, NULL},
 329     {NULL, '\0', 0, 0, NULL, NULL, NULL}
 330   };
 331   GOptionContext *ctx;
 332   GError *opt_err = NULL;
 333
 334   gtk_init (&argc, &argv);
 335
 336   /* command line option parsing */
 337   ctx = g_option_context_new (" VIDEOFILE");
 338   g_option_context_add_group (ctx, gst_init_get_option_group ());
 339   g_option_context_add_main_entries (ctx, test_goptions, NULL);
 340
 341   if (!g_option_context_parse (ctx, &argc, &argv, &opt_err)) {
 342     g_error ("Error parsing command line options: %s", opt_err->message);
 343     return -1;
 344   }
 345
 346   if (filenames == NULL || filenames[0] == NULL || filenames[0][0] == '\0') {
 347     g_printerr ("Please specify a path to a video file\n\n");
 348     return -1;
 349   }
 350
 351   run_gui (filenames[0]);
 352
 353   g_free (filenames);
 354   g_option_context_free (ctx);
 355
 356   return 0;
 357 }