tests/icles/videocrop-test.c

   1 /* GStreamer interactive test for the videocrop element
   2  * Copyright (C) 2006 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
  26 #include <stdlib.h>
  27 #include <math.h>
  28
  29 GST_DEBUG_CATEGORY_STATIC (videocrop_test_debug);
  30 #define GST_CAT_DEFAULT videocrop_test_debug
  31
  32 #define OUT_WIDTH      640
  33 #define OUT_HEIGHT     480
  34 #define TIME_PER_TEST   10      /* seconds each format is tested */
  35 #define FRAMERATE       15      /* frames per second             */
  36
  37 #ifndef DEFAULT_VIDEOSINK
  38 #define DEFAULT_VIDEOSINK "xvimagesink"
  39 #endif
  40
  41 static gboolean
  42 check_bus_for_errors (GstBus * bus, GstClockTime max_wait_time)
  43 {
  44   GstMessage *msg;
  45
  46   msg = gst_bus_poll (bus, GST_MESSAGE_ERROR, max_wait_time);
  47
  48   if (msg) {
  49     GError *err = NULL;
  50     gchar *debug = NULL;
  51
  52     g_assert (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_ERROR);
  53     gst_message_parse_error (msg, &err, &debug);
  54     GST_ERROR ("ERROR: %s [%s]", err->message, debug);
  55     g_print ("\n===========> ERROR: %s\n%s\n\n", err->message, debug);
  56     g_error_free (err);
  57     g_free (debug);
  58     gst_message_unref (msg);
  59   }
  60
  61   return (msg != NULL);
  62 }
  63
  64 static void
  65 test_with_caps (GstElement * src, GstElement * videocrop, GstCaps * caps)
  66 {
  67   GstClockTime time_run;
  68   GstElement *pipeline;
  69   GTimer *timer;
  70   GstBus *bus;
  71   GstPad *pad;
  72   guint hcrop;
  73   guint vcrop;
  74
  75   /* caps must be writable, we can't check that here though */
  76   g_assert (GST_CAPS_REFCOUNT_VALUE (caps) == 1);
  77
  78   timer = g_timer_new ();
  79   vcrop = 0;
  80   hcrop = 0;
  81
  82   pipeline = GST_ELEMENT (gst_element_get_parent (videocrop));
  83   g_assert (GST_IS_PIPELINE (pipeline));
  84
  85   /* at this point the pipeline is in PLAYING state; we only want to capture
  86    * errors resulting from our on-the-fly changing of the filtercaps */
  87   bus = gst_pipeline_get_bus (GST_PIPELINE (pipeline));
  88
  89   /* pad to block */
  90   pad = gst_element_get_static_pad (src, "src");
  91
  92   time_run = 0;
  93   do {
  94     GstClockTime wait_time, waited_for_block;
  95
  96     if (check_bus_for_errors (bus, 0))
  97       break;
  98
  99     wait_time = GST_SECOND / FRAMERATE;
 100
 101     GST_LOG ("hcrop = %3d, vcrop = %3d", vcrop, hcrop);
 102
 103     g_timer_reset (timer);
 104
 105     /* need to block the streaming thread while changing these properties,
 106      * otherwise we might get random not-negotiated errors (when caps are
 107      * changed in between upstream calling pad_alloc_buffer() and pushing
 108      * the processed buffer?) */
 109     gst_pad_set_blocked (pad, TRUE);
 110     g_object_set (videocrop, "left", hcrop, "top", vcrop, NULL);
 111     gst_pad_set_blocked (pad, FALSE);
 112
 113     waited_for_block = g_timer_elapsed (timer, NULL) * (double) GST_SECOND;
 114     /* GST_LOG ("waited: %" GST_TIME_FORMAT ", frame len: %" GST_TIME_FORMAT,
 115        GST_TIME_ARGS (waited_for_block), GST_TIME_ARGS (wait_time)); */
 116     ++vcrop;
 117     ++hcrop;
 118
 119     if (wait_time > waited_for_block) {
 120       g_usleep ((wait_time - waited_for_block) / GST_MSECOND);
 121     }
 122
 123     time_run += wait_time;
 124   }
 125   while (time_run < (TIME_PER_TEST * GST_SECOND));
 126
 127   g_timer_destroy (timer);
 128   gst_object_unref (bus);
 129   gst_object_unref (pad);
 130 }
 131
 132 /* return a list of caps where we only need to set
 133  * width and height to get fixed caps */
 134 static GList *
 135 video_crop_get_test_caps (GstElement * videocrop)
 136 {
 137   const GstCaps *allowed_caps;
 138   GstPad *srcpad;
 139   GList *list = NULL;
 140   guint i;
 141
 142   srcpad = gst_element_get_static_pad (videocrop, "src");
 143   g_assert (srcpad != NULL);
 144   allowed_caps = gst_pad_get_pad_template_caps (srcpad);
 145   g_assert (allowed_caps != NULL);
 146
 147   for (i = 0; i < gst_caps_get_size (allowed_caps); ++i) {
 148     GstStructure *new_structure;
 149     GstCaps *single_caps;
 150
 151     single_caps = gst_caps_new_empty ();
 152     new_structure =
 153         gst_structure_copy (gst_caps_get_structure (allowed_caps, i));
 154     gst_structure_set (new_structure, "framerate", GST_TYPE_FRACTION,
 155         FRAMERATE, 1, NULL);
 156     gst_structure_remove_field (new_structure, "width");
 157     gst_structure_remove_field (new_structure, "height");
 158     gst_caps_append_structure (single_caps, new_structure);
 159
 160     /* should be fixed without width/height */
 161     g_assert (gst_caps_is_fixed (single_caps));
 162
 163     list = g_list_prepend (list, single_caps);
 164   }
 165
 166   gst_object_unref (srcpad);
 167
 168   return list;
 169 }
 170
 171 static gchar *opt_videosink_str;        /* NULL */
 172 static gchar *opt_filtercaps_str;       /* NULL */
 173 static gboolean opt_with_ffmpegcolorspace;      /* FALSE */
 174
 175 int
 176 main (int argc, char **argv)
 177 {
 178   static const GOptionEntry test_goptions[] = {
 179     {"videosink", '\0', 0, G_OPTION_ARG_STRING, &opt_videosink_str,
 180         "videosink to use (default: " DEFAULT_VIDEOSINK ")", NULL},
 181     {"caps", '\0', 0, G_OPTION_ARG_STRING, &opt_filtercaps_str,
 182         "filter caps to narrow down formats to test", NULL},
 183     {"with-ffmpegcolorspace", '\0', 0, G_OPTION_ARG_NONE,
 184           &opt_with_ffmpegcolorspace,
 185           "whether to add an ffmpegcolorspace element in front of the sink",
 186         NULL},
 187     {NULL, '\0', 0, 0, NULL, NULL, NULL}
 188   };
 189   GOptionContext *ctx;
 190   GError *opt_err = NULL;
 191
 192   GstElement *pipeline, *src, *filter1, *crop, *scale, *filter2, *csp, *sink;
 193   GMainLoop *loop;
 194   GstCaps *filter_caps = NULL;
 195   GList *caps_list, *l;
 196
 197   if (!g_thread_supported ())
 198     g_thread_init (NULL);
 199
 200   /* command line option parsing */
 201   ctx = g_option_context_new ("");
 202   g_option_context_add_group (ctx, gst_init_get_option_group ());
 203   g_option_context_add_main_entries (ctx, test_goptions, NULL);
 204
 205   if (!g_option_context_parse (ctx, &argc, &argv, &opt_err)) {
 206     g_error ("Error parsing command line options: %s", opt_err->message);
 207     return -1;
 208   }
 209
 210   GST_DEBUG_CATEGORY_INIT (videocrop_test_debug, "videocroptest", 0, "vctest");
 211
 212   loop = g_main_loop_new (NULL, FALSE);
 213
 214   pipeline = gst_pipeline_new ("pipeline");
 215   src = gst_element_factory_make ("videotestsrc", "videotestsrc");
 216   g_assert (src != NULL);
 217   filter1 = gst_element_factory_make ("capsfilter", "capsfilter1");
 218   g_assert (filter1 != NULL);
 219   crop = gst_element_factory_make ("videocrop", "videocrop");
 220   g_assert (crop != NULL);
 221   scale = gst_element_factory_make ("videoscale", "videoscale");
 222   g_assert (scale != NULL);
 223   filter2 = gst_element_factory_make ("capsfilter", "capsfilter2");
 224   g_assert (filter2 != NULL);
 225
 226   if (opt_with_ffmpegcolorspace) {
 227     g_print ("Adding ffmpegcolorspace\n");
 228     csp = gst_element_factory_make ("ffmpegcolorspace", "colorspace");
 229   } else {
 230     csp = gst_element_factory_make ("identity", "colorspace");
 231   }
 232   g_assert (csp != NULL);
 233
 234   if (opt_filtercaps_str) {
 235     filter_caps = gst_caps_from_string (opt_filtercaps_str);
 236     if (filter_caps == NULL) {
 237       g_error ("Invalid filter caps string '%s'", opt_filtercaps_str);
 238     } else {
 239       g_print ("Using filter caps '%s'\n", opt_filtercaps_str);
 240     }
 241   }
 242
 243   if (opt_videosink_str) {
 244     g_print ("Trying videosink '%s' ...", opt_videosink_str);
 245     sink = gst_element_factory_make (opt_videosink_str, "sink");
 246     g_print ("%s\n", (sink) ? "ok" : "element couldn't be created");
 247   } else {
 248     sink = NULL;
 249   }
 250
 251   if (sink == NULL) {
 252     g_print ("Trying videosink '%s' ...", DEFAULT_VIDEOSINK);
 253     sink = gst_element_factory_make (DEFAULT_VIDEOSINK, "sink");
 254     g_print ("%s\n", (sink) ? "ok" : "element couldn't be created");
 255   }
 256   if (sink == NULL) {
 257     g_print ("Trying videosink '%s' ...", "xvimagesink");
 258     sink = gst_element_factory_make ("xvimagesink", "sink");
 259     g_print ("%s\n", (sink) ? "ok" : "element couldn't be created");
 260   }
 261   if (sink == NULL) {
 262     g_print ("Trying videosink '%s' ...", "ximagesink");
 263     sink = gst_element_factory_make ("ximagesink", "sink");
 264     g_print ("%s\n", (sink) ? "ok" : "element couldn't be created");
 265   }
 266
 267   g_assert (sink != NULL);
 268
 269   gst_bin_add_many (GST_BIN (pipeline), src, filter1, crop, scale, filter2,
 270       csp, sink, NULL);
 271
 272   if (!gst_element_link (src, filter1))
 273     g_error ("Failed to link videotestsrc to capsfilter1");
 274
 275   if (!gst_element_link (filter1, crop))
 276     g_error ("Failed to link capsfilter1 to videocrop");
 277
 278   if (!gst_element_link (crop, scale))
 279     g_error ("Failed to link videocrop to videoscale");
 280
 281   if (!gst_element_link (scale, filter2))
 282     g_error ("Failed to link videoscale to capsfilter2");
 283
 284   if (!gst_element_link (filter2, csp))
 285     g_error ("Failed to link capsfilter2 to ffmpegcolorspace");
 286
 287   if (!gst_element_link (csp, sink))
 288     g_error ("Failed to link ffmpegcolorspace to video sink");
 289
 290   caps_list = video_crop_get_test_caps (crop);
 291   for (l = caps_list; l != NULL; l = l->next) {
 292     GstStateChangeReturn ret;
 293     GstCaps *caps, *out_caps;
 294     gboolean skip = FALSE;
 295     gchar *s;
 296
 297     if (filter_caps) {
 298       GstCaps *icaps;
 299
 300       icaps = gst_caps_intersect (filter_caps, GST_CAPS (l->data));
 301       skip = gst_caps_is_empty (icaps);
 302       gst_caps_unref (icaps);
 303     }
 304
 305     /* this is the size of our window (stays fixed) */
 306     out_caps = gst_caps_copy (GST_CAPS (l->data));
 307     gst_structure_set (gst_caps_get_structure (out_caps, 0), "width",
 308         G_TYPE_INT, OUT_WIDTH, "height", G_TYPE_INT, OUT_HEIGHT, NULL);
 309
 310     g_object_set (filter2, "caps", out_caps, NULL);
 311
 312     /* filter1 gets these too to prevent videotestsrc from renegotiating */
 313     g_object_set (filter1, "caps", out_caps, NULL);
 314     gst_caps_unref (out_caps);
 315
 316     caps = gst_caps_copy (GST_CAPS (l->data));
 317     GST_INFO ("testing format: %" GST_PTR_FORMAT, caps);
 318
 319     s = gst_caps_to_string (caps);
 320
 321     if (skip) {
 322       g_print ("Skipping format: %s\n", s);
 323       g_free (s);
 324       continue;
 325     }
 326
 327     g_print ("Format: %s\n", s);
 328
 329     caps = gst_caps_make_writable (caps);
 330
 331     /* FIXME: check return values */
 332     ret = gst_element_set_state (pipeline, GST_STATE_PLAYING);
 333     if (ret != GST_STATE_CHANGE_FAILURE) {
 334       ret = gst_element_get_state (pipeline, NULL, NULL, -1);
 335
 336       if (ret != GST_STATE_CHANGE_FAILURE) {
 337         test_with_caps (src, crop, caps);
 338       } else {
 339         g_print ("Format: %s not supported (failed to go to PLAYING)\n", s);
 340       }
 341     } else {
 342       g_print ("Format: %s not supported\n", s);
 343     }
 344
 345     gst_element_set_state (pipeline, GST_STATE_NULL);
 346
 347     gst_caps_unref (caps);
 348     g_free (s);
 349   }
 350
 351   g_list_foreach (caps_list, (GFunc) gst_caps_unref, NULL);
 352   g_list_free (caps_list);
 353
 354   gst_element_set_state (pipeline, GST_STATE_NULL);
 355   gst_object_unref (pipeline);
 356
 357   return 0;
 358 }