tools/gst-discoverer.c

   1 /* GStreamer
   2  * Copyright (C) 2009 Edward Hervey <edward.hervey@collabora.co.uk>
   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 <stdlib.h>
  25 #include <glib.h>
  26 #include <gst/gst.h>
  27 #include <gst/pbutils/pbutils.h>
  28
  29 static gboolean async = FALSE;
  30 static gboolean silent = FALSE;
  31 static gboolean verbose = FALSE;
  32
  33 typedef struct
  34 {
  35   GstDiscoverer *dc;
  36   int argc;
  37   char **argv;
  38 } PrivStruct;
  39
  40 #define my_g_string_append_printf(str, format, ...) \
  41   g_string_append_printf (str, "%*s" format, 2*depth, " ", ##__VA_ARGS__)
  42
  43 static gchar *
  44 gst_stream_audio_information_to_string (GstDiscovererStreamInfo * info,
  45     gint depth)
  46 {
  47   GstDiscovererAudioInfo *audio_info;
  48   GString *s;
  49   gchar *tmp;
  50   const gchar *ctmp;
  51   int len = 400;
  52   const GstTagList *tags;
  53   GstCaps *caps;
  54
  55   g_return_val_if_fail (info != NULL, NULL);
  56
  57   s = g_string_sized_new (len);
  58
  59   my_g_string_append_printf (s, "Codec:\n");
  60   caps = gst_discoverer_stream_info_get_caps (info);
  61   tmp = gst_caps_to_string (caps);
  62   gst_caps_unref (caps);
  63   my_g_string_append_printf (s, "  %s\n", tmp);
  64   g_free (tmp);
  65
  66   my_g_string_append_printf (s, "Additional info:\n");
  67   if (gst_discoverer_stream_info_get_misc (info)) {
  68     tmp = gst_structure_to_string (gst_discoverer_stream_info_get_misc (info));
  69     my_g_string_append_printf (s, "  %s\n", tmp);
  70     g_free (tmp);
  71   } else {
  72     my_g_string_append_printf (s, "  None\n");
  73   }
  74
  75   audio_info = (GstDiscovererAudioInfo *) info;
  76   ctmp = gst_discoverer_audio_info_get_language (audio_info);
  77   my_g_string_append_printf (s, "Language: %s\n", ctmp ? ctmp : "<unknown>");
  78   my_g_string_append_printf (s, "Channels: %u\n",
  79       gst_discoverer_audio_info_get_channels (audio_info));
  80   my_g_string_append_printf (s, "Sample rate: %u\n",
  81       gst_discoverer_audio_info_get_sample_rate (audio_info));
  82   my_g_string_append_printf (s, "Depth: %u\n",
  83       gst_discoverer_audio_info_get_depth (audio_info));
  84
  85   my_g_string_append_printf (s, "Bitrate: %u\n",
  86       gst_discoverer_audio_info_get_bitrate (audio_info));
  87   my_g_string_append_printf (s, "Max bitrate: %u\n",
  88       gst_discoverer_audio_info_get_max_bitrate (audio_info));
  89
  90   my_g_string_append_printf (s, "Tags:\n");
  91   tags = gst_discoverer_stream_info_get_tags (info);
  92   if (tags) {
  93     tmp = gst_structure_to_string ((GstStructure *) tags);
  94     my_g_string_append_printf (s, "  %s\n", tmp);
  95     g_free (tmp);
  96   } else {
  97     my_g_string_append_printf (s, "  None\n");
  98   }
  99   if (verbose)
 100     my_g_string_append_printf (s, "\n");
 101
 102   return g_string_free (s, FALSE);
 103 }
 104
 105 static gchar *
 106 gst_stream_video_information_to_string (GstDiscovererStreamInfo * info,
 107     gint depth)
 108 {
 109   GstDiscovererVideoInfo *video_info;
 110   GString *s;
 111   gchar *tmp;
 112   int len = 500;
 113   const GstStructure *misc;
 114   const GstTagList *tags;
 115   GstCaps *caps;
 116
 117   g_return_val_if_fail (info != NULL, NULL);
 118
 119   s = g_string_sized_new (len);
 120
 121   my_g_string_append_printf (s, "Codec:\n");
 122   caps = gst_discoverer_stream_info_get_caps (info);
 123   tmp = gst_caps_to_string (caps);
 124   gst_caps_unref (caps);
 125   my_g_string_append_printf (s, "  %s\n", tmp);
 126   g_free (tmp);
 127
 128   my_g_string_append_printf (s, "Additional info:\n");
 129   misc = gst_discoverer_stream_info_get_misc (info);
 130   if (misc) {
 131     tmp = gst_structure_to_string (misc);
 132     my_g_string_append_printf (s, "  %s\n", tmp);
 133     g_free (tmp);
 134   } else {
 135     my_g_string_append_printf (s, "  None\n");
 136   }
 137
 138   video_info = (GstDiscovererVideoInfo *) info;
 139   my_g_string_append_printf (s, "Width: %u\n",
 140       gst_discoverer_video_info_get_width (video_info));
 141   my_g_string_append_printf (s, "Height: %u\n",
 142       gst_discoverer_video_info_get_height (video_info));
 143   my_g_string_append_printf (s, "Depth: %u\n",
 144       gst_discoverer_video_info_get_depth (video_info));
 145
 146   my_g_string_append_printf (s, "Frame rate: %u/%u\n",
 147       gst_discoverer_video_info_get_framerate_num (video_info),
 148       gst_discoverer_video_info_get_framerate_denom (video_info));
 149
 150   my_g_string_append_printf (s, "Pixel aspect ratio: %u/%u\n",
 151       gst_discoverer_video_info_get_par_num (video_info),
 152       gst_discoverer_video_info_get_par_denom (video_info));
 153
 154   my_g_string_append_printf (s, "Interlaced: %s\n",
 155       gst_discoverer_video_info_is_interlaced (video_info) ? "true" : "false");
 156
 157   my_g_string_append_printf (s, "Bitrate: %u\n",
 158       gst_discoverer_video_info_get_bitrate (video_info));
 159   my_g_string_append_printf (s, "Max bitrate: %u\n",
 160       gst_discoverer_video_info_get_max_bitrate (video_info));
 161
 162   my_g_string_append_printf (s, "Tags:\n");
 163   tags = gst_discoverer_stream_info_get_tags (info);
 164   if (tags) {
 165     tmp = gst_structure_to_string ((GstStructure *) tags);
 166     my_g_string_append_printf (s, "  %s\n", tmp);
 167     g_free (tmp);
 168   } else {
 169     my_g_string_append_printf (s, "  None\n");
 170   }
 171   if (verbose)
 172     my_g_string_append_printf (s, "\n");
 173
 174   return g_string_free (s, FALSE);
 175 }
 176
 177 static gchar *
 178 gst_stream_subtitle_information_to_string (GstDiscovererStreamInfo * info,
 179     gint depth)
 180 {
 181   GstDiscovererSubtitleInfo *subtitle_info;
 182   GString *s;
 183   gchar *tmp;
 184   const gchar *ctmp;
 185   int len = 400;
 186   const GstTagList *tags;
 187   GstCaps *caps;
 188
 189   g_return_val_if_fail (info != NULL, NULL);
 190
 191   s = g_string_sized_new (len);
 192
 193   my_g_string_append_printf (s, "Codec:\n");
 194   caps = gst_discoverer_stream_info_get_caps (info);
 195   tmp = gst_caps_to_string (caps);
 196   gst_caps_unref (caps);
 197   my_g_string_append_printf (s, "  %s\n", tmp);
 198   g_free (tmp);
 199
 200   my_g_string_append_printf (s, "Additional info:\n");
 201   if (gst_discoverer_stream_info_get_misc (info)) {
 202     tmp = gst_structure_to_string (gst_discoverer_stream_info_get_misc (info));
 203     my_g_string_append_printf (s, "  %s\n", tmp);
 204     g_free (tmp);
 205   } else {
 206     my_g_string_append_printf (s, "  None\n");
 207   }
 208
 209   subtitle_info = (GstDiscovererSubtitleInfo *) info;
 210   ctmp = gst_discoverer_subtitle_info_get_language (subtitle_info);
 211   my_g_string_append_printf (s, "Language: %s\n", ctmp ? ctmp : "<unknown>");
 212
 213   my_g_string_append_printf (s, "Tags:\n");
 214   tags = gst_discoverer_stream_info_get_tags (info);
 215   if (tags) {
 216     tmp = gst_structure_to_string ((GstStructure *) tags);
 217     my_g_string_append_printf (s, "  %s\n", tmp);
 218     g_free (tmp);
 219   } else {
 220     my_g_string_append_printf (s, "  None\n");
 221   }
 222   if (verbose)
 223     my_g_string_append_printf (s, "\n");
 224
 225   return g_string_free (s, FALSE);
 226 }
 227
 228 static void
 229 print_stream_info (GstDiscovererStreamInfo * info, void *depth)
 230 {
 231   gchar *desc = NULL;
 232   GstCaps *caps;
 233
 234   caps = gst_discoverer_stream_info_get_caps (info);
 235
 236   if (caps) {
 237     if (gst_caps_is_fixed (caps) && !verbose)
 238       desc = gst_pb_utils_get_codec_description (caps);
 239     else
 240       desc = gst_caps_to_string (caps);
 241     gst_caps_unref (caps);
 242   }
 243
 244   g_print ("%*s%s: %s\n", 2 * GPOINTER_TO_INT (depth), " ",
 245       gst_discoverer_stream_info_get_stream_type_nick (info), desc);
 246
 247   if (desc) {
 248     g_free (desc);
 249     desc = NULL;
 250   }
 251
 252   if (verbose) {
 253     if (GST_IS_DISCOVERER_AUDIO_INFO (info))
 254       desc =
 255           gst_stream_audio_information_to_string (info,
 256           GPOINTER_TO_INT (depth) + 1);
 257     else if (GST_IS_DISCOVERER_VIDEO_INFO (info))
 258       desc =
 259           gst_stream_video_information_to_string (info,
 260           GPOINTER_TO_INT (depth) + 1);
 261     else if (GST_IS_DISCOVERER_SUBTITLE_INFO (info))
 262       desc =
 263           gst_stream_subtitle_information_to_string (info,
 264           GPOINTER_TO_INT (depth) + 1);
 265     if (desc) {
 266       g_print ("%s", desc);
 267       g_free (desc);
 268     }
 269   }
 270 }
 271
 272 static void
 273 print_topology (GstDiscovererStreamInfo * info, gint depth)
 274 {
 275   GstDiscovererStreamInfo *next;
 276
 277   if (!info)
 278     return;
 279
 280   print_stream_info (info, GINT_TO_POINTER (depth));
 281
 282   next = gst_discoverer_stream_info_get_next (info);
 283   if (next) {
 284     print_topology (next, depth + 1);
 285     gst_discoverer_stream_info_unref (next);
 286   } else if (GST_IS_DISCOVERER_CONTAINER_INFO (info)) {
 287     GList *tmp, *streams;
 288
 289     streams =
 290         gst_discoverer_container_info_get_streams (GST_DISCOVERER_CONTAINER_INFO
 291         (info));
 292     for (tmp = streams; tmp; tmp = tmp->next) {
 293       GstDiscovererStreamInfo *tmpinf = (GstDiscovererStreamInfo *) tmp->data;
 294       print_topology (tmpinf, depth + 1);
 295     }
 296     gst_discoverer_stream_info_list_free (streams);
 297   }
 298 }
 299
 300 static gboolean
 301 print_tag_each (GQuark field_id, const GValue * value, gpointer user_data)
 302 {
 303   gint tab = GPOINTER_TO_INT (user_data);
 304   gchar *ser;
 305
 306   if (G_VALUE_HOLDS_STRING (value))
 307     ser = g_value_dup_string (value);
 308   else if (GST_VALUE_HOLDS_BUFFER (value)) {
 309     GstBuffer *buf = gst_value_get_buffer (value);
 310     ser =
 311         g_strdup_printf ("<GstBuffer [%" G_GSIZE_FORMAT " bytes]>",
 312         gst_buffer_get_size (buf));
 313   } else
 314     ser = gst_value_serialize (value);
 315
 316   g_print ("%*s%s: %s\n", tab, " ",
 317       gst_tag_get_nick (g_quark_to_string (field_id)), ser);
 318   g_free (ser);
 319
 320   return TRUE;
 321 }
 322
 323 static void
 324 print_properties (GstDiscovererInfo * info, gint tab)
 325 {
 326   const GstTagList *tags;
 327
 328   g_print ("%*sDuration: %" GST_TIME_FORMAT "\n", tab + 1, " ",
 329       GST_TIME_ARGS (gst_discoverer_info_get_duration (info)));
 330   g_print ("%*sSeekable: %s\n", tab + 1, " ",
 331       (gst_discoverer_info_get_seekable (info) ? "yes" : "no"));
 332   if ((tags = gst_discoverer_info_get_tags (info))) {
 333     g_print ("%*sTags: \n", tab + 1, " ");
 334     gst_structure_foreach ((const GstStructure *) tags, print_tag_each,
 335         GINT_TO_POINTER (tab + 5));
 336   }
 337 }
 338
 339 static void
 340 print_info (GstDiscovererInfo * info, GError * err)
 341 {
 342   GstDiscovererResult result = gst_discoverer_info_get_result (info);
 343   GstDiscovererStreamInfo *sinfo;
 344
 345   g_print ("Done discovering %s\n", gst_discoverer_info_get_uri (info));
 346   switch (result) {
 347     case GST_DISCOVERER_OK:
 348     {
 349       break;
 350     }
 351     case GST_DISCOVERER_URI_INVALID:
 352     {
 353       g_print ("URI is not valid\n");
 354       break;
 355     }
 356     case GST_DISCOVERER_ERROR:
 357     {
 358       g_print ("An error was encountered while discovering the file\n");
 359       g_print (" %s\n", err->message);
 360       break;
 361     }
 362     case GST_DISCOVERER_TIMEOUT:
 363     {
 364       g_print ("Analyzing URI timed out\n");
 365       break;
 366     }
 367     case GST_DISCOVERER_BUSY:
 368     {
 369       g_print ("Discoverer was busy\n");
 370       break;
 371     }
 372     case GST_DISCOVERER_MISSING_PLUGINS:
 373     {
 374       g_print ("Missing plugins\n");
 375       if (verbose) {
 376         gchar *tmp =
 377             gst_structure_to_string (gst_discoverer_info_get_misc (info));
 378         g_print (" (%s)\n", tmp);
 379         g_free (tmp);
 380       }
 381       break;
 382     }
 383   }
 384
 385   if ((sinfo = gst_discoverer_info_get_stream_info (info))) {
 386     g_print ("\nTopology:\n");
 387     print_topology (sinfo, 1);
 388     g_print ("\nProperties:\n");
 389     print_properties (info, 1);
 390     gst_discoverer_stream_info_unref (sinfo);
 391   }
 392
 393   g_print ("\n");
 394 }
 395
 396 static void
 397 process_file (GstDiscoverer * dc, const gchar * filename)
 398 {
 399   GError *err = NULL;
 400   GDir *dir;
 401   gchar *uri, *path;
 402   GstDiscovererInfo *info;
 403   GstStructure *st = NULL;
 404
 405   if (!gst_uri_is_valid (filename)) {
 406     /* Recurse into directories */
 407     if ((dir = g_dir_open (filename, 0, NULL))) {
 408       const gchar *entry;
 409
 410       while ((entry = g_dir_read_name (dir))) {
 411         gchar *path;
 412         path = g_strconcat (filename, G_DIR_SEPARATOR_S, entry, NULL);
 413         process_file (dc, path);
 414         g_free (path);
 415       }
 416
 417       g_dir_close (dir);
 418       return;
 419     }
 420
 421     if (!g_path_is_absolute (filename)) {
 422       gchar *cur_dir;
 423
 424       cur_dir = g_get_current_dir ();
 425       path = g_build_filename (cur_dir, filename, NULL);
 426       g_free (cur_dir);
 427     } else {
 428       path = g_strdup (filename);
 429     }
 430
 431     uri = g_filename_to_uri (path, NULL, &err);
 432     g_free (path);
 433     path = NULL;
 434
 435     if (err) {
 436       g_warning ("Couldn't convert filename to URI: %s\n", err->message);
 437       g_error_free (err);
 438       return;
 439     }
 440   } else {
 441     uri = g_strdup (filename);
 442   }
 443
 444   if (async == FALSE) {
 445     g_print ("Analyzing %s\n", uri);
 446     info = gst_discoverer_discover_uri (dc, uri, &err);
 447     print_info (info, err);
 448     gst_discoverer_info_unref (info);
 449     if (st)
 450       gst_structure_free (st);
 451   } else {
 452     gst_discoverer_discover_uri_async (dc, uri);
 453   }
 454
 455   g_free (uri);
 456 }
 457
 458 static void
 459 _new_discovered_uri (GstDiscoverer * dc, GstDiscovererInfo * info, GError * err)
 460 {
 461   print_info (info, err);
 462 }
 463
 464 static gboolean
 465 _run_async (PrivStruct * ps)
 466 {
 467   gint i;
 468
 469   for (i = 1; i < ps->argc; i++)
 470     process_file (ps->dc, ps->argv[i]);
 471
 472   return FALSE;
 473 }
 474
 475 static void
 476 _discoverer_finished (GstDiscoverer * dc, GMainLoop * ml)
 477 {
 478   g_main_loop_quit (ml);
 479 }
 480
 481 int
 482 main (int argc, char **argv)
 483 {
 484   GError *err = NULL;
 485   GstDiscoverer *dc;
 486   gint timeout = 10;
 487   GOptionEntry options[] = {
 488     {"async", 'a', 0, G_OPTION_ARG_NONE, &async,
 489         "Run asynchronously", NULL},
 490     {"silent", 's', 0, G_OPTION_ARG_NONE, &silent,
 491         "Don't output the information structure", NULL},
 492     {"timeout", 't', 0, G_OPTION_ARG_INT, &timeout,
 493         "Specify timeout (in seconds, default 10)", "T"},
 494     /* {"elem", 'e', 0, G_OPTION_ARG_NONE, &elem_seek, */
 495     /*     "Seek on elements instead of pads", NULL}, */
 496     {"verbose", 'v', 0, G_OPTION_ARG_NONE, &verbose,
 497         "Verbose properties", NULL},
 498     {NULL}
 499   };
 500   GOptionContext *ctx;
 501
 502 #if !GLIB_CHECK_VERSION (2, 31, 0)
 503   if (!g_thread_supported ())
 504     g_thread_init (NULL);
 505 #endif
 506
 507   ctx =
 508       g_option_context_new
 509       ("- discover files synchronously with GstDiscoverer");
 510   g_option_context_add_main_entries (ctx, options, NULL);
 511   g_option_context_add_group (ctx, gst_init_get_option_group ());
 512
 513   if (!g_option_context_parse (ctx, &argc, &argv, &err)) {
 514     g_print ("Error initializing: %s\n", err->message);
 515     exit (1);
 516   }
 517
 518   g_option_context_free (ctx);
 519
 520   if (argc < 2) {
 521     g_print ("usage: %s <uris>\n", argv[0]);
 522     exit (-1);
 523   }
 524
 525   dc = gst_discoverer_new (timeout * GST_SECOND, &err);
 526   if (G_UNLIKELY (dc == NULL)) {
 527     g_print ("Error initializing: %s\n", err->message);
 528     exit (1);
 529   }
 530
 531   if (async == FALSE) {
 532     gint i;
 533     for (i = 1; i < argc; i++)
 534       process_file (dc, argv[i]);
 535   } else {
 536     PrivStruct *ps = g_new0 (PrivStruct, 1);
 537     GMainLoop *ml = g_main_loop_new (NULL, FALSE);
 538
 539     ps->dc = dc;
 540     ps->argc = argc;
 541     ps->argv = argv;
 542
 543     /* adding uris will be started when the mainloop runs */
 544     g_idle_add ((GSourceFunc) _run_async, ps);
 545
 546     /* connect signals */
 547     g_signal_connect (dc, "discovered", G_CALLBACK (_new_discovered_uri), NULL);
 548     g_signal_connect (dc, "finished", G_CALLBACK (_discoverer_finished), ml);
 549
 550     gst_discoverer_start (dc);
 551     /* run mainloop */
 552     g_main_loop_run (ml);
 553
 554     gst_discoverer_stop (dc);
 555     g_free (ps);
 556     g_main_loop_unref (ml);
 557   }
 558   g_object_unref (dc);
 559
 560   return 0;
 561 }