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