[libav-devel] [PATCH] avprobe: output proper INI or JSON

Luca Barbato lu_zero at gentoo.org
Sat May 19 10:15:41 CEST 2012


Make the output valid ini and provide an alternative JSON serialization.
---
 avprobe.c |  627 +++++++++++++++++++++++++++++++++++++++++++++++++------------
 1 files changed, 507 insertions(+), 120 deletions(-)

diff --git a/avprobe.c b/avprobe.c
index 6182875..9cdbeec 100644
--- a/avprobe.c
+++ b/avprobe.c
@@ -35,6 +35,7 @@ const int program_birth_year = 2007;
 static int do_show_format  = 0;
 static AVDictionary *fmt_entries_to_show = NULL;
 static int nb_fmt_entries_to_show;
+static int do_output_json  = 0;
 static int do_show_packets = 0;
 static int do_show_streams = 0;
 
@@ -64,6 +65,358 @@ void exit_program(int ret)
     exit(ret);
 }
 
+#define MAX_INDENT 4
+
+static int print_object_separator;
+static int print_array_separator;
+static int print_item_separator;
+static int indent_level;
+static int omit_name[MAX_INDENT];
+
+
+AVIOContext *avp_out = NULL;
+
+#define AVP_INDENT() avio_printf(avp_out, "%*c", indent_level * 2, ' ')
+
+/*
+ * Default format, INI
+ *
+ * - all key and values are utf8
+ * - '.' is the subgroup separator
+ * - newlines and the following characters are escaped
+ * - '\' is the escape character
+ * - '#' is the comment
+ * - '=' is the key/value separators
+ * - ':' is not used but usually parsed as key/value separator
+ */
+
+static void ini_print_header(void)
+{
+    avio_printf(avp_out, "# avprobe output\n\n");
+}
+static void ini_print_footer(void)
+{
+    avio_w8(avp_out, '\n');
+}
+
+static void ini_escape_print(const char *s)
+{
+    int i = 0;
+    char c = 0;
+
+    while (c = s[i++]) {
+        switch (c) {
+        case '\r': avio_printf(avp_out, "%s", "\\r"); break;
+        case '\n': avio_printf(avp_out, "%s", "\\n"); break;
+        case '\\': avio_printf(avp_out, "%s", "\\\\"); break;
+        case '\f': avio_printf(avp_out, "%s", "\\f"); break;
+        case '\b': avio_printf(avp_out, "%s", "\\b"); break;
+        case '\t': avio_printf(avp_out, "%s", "\\t"); break;
+        case '#': avio_printf(avp_out, "%s", "\\#"); break;
+        case '=': avio_printf(avp_out, "%s", "\\="); break;
+        case ':': avio_printf(avp_out, "%s", "\\:"); break;
+        default:
+            if ((unsigned char)c < 32)
+                avio_printf(avp_out, "\\x00%02x", c & 0xff);
+            else
+                avio_w8(avp_out, c);
+        break;
+        }
+    }
+}
+
+static char const **prefix;
+static int group_index = -1;
+static int in_object;
+
+static void ini_group_header(const char *name)
+{
+    prefix = av_realloc(prefix, sizeof(char *) * (indent_level+1));
+    if (!prefix || !name) {
+        fprintf(stderr, "Out of memory\n");
+        exit(1);
+    }
+    prefix[indent_level++] = name;
+}
+
+static void ini_group_footer(void)
+{
+    av_freep(&prefix[--indent_level]);
+}
+
+static void ini_print_array_header(const char *name)
+{
+    if (print_array_separator)
+        avio_printf(avp_out, "\n");
+    group_index = 0;
+    ini_group_header(av_strdup(name));
+}
+static void ini_print_array_footer(const char *name)
+{
+    ini_group_footer();
+    group_index = -1;
+}
+
+static void ini_print_object_header(const char *name)
+{
+    int i;
+    char *n = av_strdup(name);
+
+    if (print_object_separator)
+        avio_printf(avp_out, "\n");
+
+    avio_printf(avp_out, "[");
+    for (i = 0; i < indent_level; i++)
+        avio_printf(avp_out, "%s.", prefix[i]);
+
+    if (group_index >= 0 && !in_object) {
+        int len = strlen(name) + 22;
+
+        n = av_malloc(len);
+
+        if (!n) {
+            fprintf(stderr, "Out of memory\n");
+            exit(1);
+        }
+
+        snprintf(n, len, "%s.%d", name, group_index++);
+    }
+
+    avio_printf(avp_out, "%s", n);
+    avio_printf(avp_out, "]\n");
+
+    ini_group_header(n);
+    in_object = 1;
+}
+
+static void ini_print_object_footer(const char *name)
+{
+    in_object = 0;
+
+    ini_group_footer();
+}
+
+
+static void ini_print_integer(const char *key, int64_t value)
+{
+    ini_escape_print(key);
+    avio_printf(avp_out, "=%"PRId64"\n", value);
+}
+
+
+static void ini_print_string(const char *key, const char *value)
+{
+    ini_escape_print(key);
+    avio_printf(avp_out, "=");
+    ini_escape_print(value);
+    avio_w8(avp_out, '\n');
+}
+
+/*
+ * Alternate format, JSON
+ */
+
+static void json_print_header(void)
+{
+    avio_printf(avp_out, "{");
+    indent_level++;
+}
+static void json_print_footer(void)
+{
+    avio_printf(avp_out, "}\n");
+    indent_level--;
+}
+
+static void json_print_array_header(const char *name)
+{
+    if (print_array_separator)
+        avio_printf(avp_out, ",\n");
+    AVP_INDENT();
+    if (!omit_name[indent_level])
+        avio_printf(avp_out, "\"%s\" : ", name);
+    avio_printf(avp_out, "[\n");
+    indent_level++;
+    omit_name[indent_level] = 1;
+}
+
+static void json_print_array_footer(const char *name)
+{
+    omit_name[indent_level] = 0;
+    indent_level--;
+    avio_printf(avp_out, "\n");
+    AVP_INDENT();
+    avio_printf(avp_out, "]");
+}
+
+static void json_print_object_header(const char *name)
+{
+    if (print_object_separator)
+        avio_printf(avp_out, ",\n");
+    AVP_INDENT();
+    if (!omit_name[indent_level])
+        avio_printf(avp_out, "\"%s\" : ", name);
+    avio_printf(avp_out, "{\n");
+    indent_level++;
+}
+
+static void json_print_object_footer(const char *name)
+{
+    indent_level--;
+    avio_printf(avp_out, "\n");
+    AVP_INDENT();
+    avio_printf(avp_out, "}");
+}
+
+static void json_print_integer(const char *key, int64_t value)
+{
+    if (print_item_separator)
+        avio_printf(avp_out, ",\n");
+    AVP_INDENT();
+    avio_printf(avp_out, "\"%s\" : %"PRId64"", key, value);
+}
+
+static void json_escape_print(const char *s)
+{
+    int i = 0;
+    char c = 0;
+
+    while (c = s[i++]) {
+        switch (c) {
+        case '\r': avio_printf(avp_out, "%s", "\\r"); break;
+        case '\n': avio_printf(avp_out, "%s", "\\n"); break;
+        case '\\': avio_printf(avp_out, "%s", "\\\\"); break;
+        case '"' : avio_printf(avp_out, "%s", "\\\""); break;
+        case '\f': avio_printf(avp_out, "%s", "\\f"); break;
+        case '\b': avio_printf(avp_out, "%s", "\\b"); break;
+        case '\t': avio_printf(avp_out, "%s", "\\t"); break;
+        default:
+            if ((unsigned char)c < 32)
+                avio_printf(avp_out, "\\u00%02x", c & 0xff);
+            else
+                avio_w8(avp_out, c);
+        break;
+        }
+    }
+}
+
+static void json_print_string(const char *key, const char *value)
+{
+    if (print_item_separator)
+        avio_printf(avp_out, ",\n");
+    AVP_INDENT();
+    avio_w8(avp_out, '\"');
+    json_escape_print(key);
+    avio_printf(avp_out, "\" : \"");
+    json_escape_print(value);
+    avio_w8(avp_out, '\"');
+}
+
+/*
+ * Simple Formatter for single entries.
+ */
+
+static void show_format_entry_integer(const char *key, int64_t value)
+{
+    if (key && av_dict_get(fmt_entries_to_show, key, NULL, 0)) {
+        if (nb_fmt_entries_to_show > 1)
+            avio_printf(avp_out, "%s=", key);
+        avio_printf(avp_out, "%"PRId64"\n", value);
+    }
+}
+
+static void show_format_entry_string(const char *key, const char *value)
+{
+    if (key && av_dict_get(fmt_entries_to_show, key, NULL, 0)) {
+        if (nb_fmt_entries_to_show > 1)
+            avio_printf(avp_out, "%s=", key);
+        avio_printf(avp_out, "%s\n", value);
+    }
+}
+
+
+void (*print_header)(void) = ini_print_header;
+void (*print_footer)(void) = ini_print_footer;
+
+void (*print_array_header) (const char *name) = ini_print_array_header;
+void (*print_array_footer) (const char *name) = ini_print_array_footer;
+void (*print_object_header)(const char *name) = ini_print_object_header;
+void (*print_object_footer)(const char *name) = ini_print_object_footer;
+
+void (*print_integer) (const char *key, int64_t value)     = ini_print_integer;
+void (*print_string)  (const char *key, const char *value) = ini_print_string;
+
+static void avp_header(void)
+{
+    if (print_header)
+        print_header();
+    print_array_separator = 0;
+}
+
+static void avp_footer(void)
+{
+    if (print_footer)
+        print_footer();
+}
+
+static void avp_chapter_header(const char *chapter)
+{
+    if (print_array_header)
+        print_array_header(chapter);
+    print_object_separator = 0;
+}
+
+static void avp_chapter_footer(const char *chapter)
+{
+    if (print_array_footer)
+        print_array_footer(chapter);
+    print_array_separator = 1;
+}
+
+static void avp_section_header(const char *section)
+{
+    if (!fmt_entries_to_show ||
+        (section && av_dict_get(fmt_entries_to_show, section, NULL, 0))) {
+        if (print_object_header)
+            print_object_header(section);
+        print_item_separator = 0;
+    }
+}
+
+static void avp_section_footer(const char *section)
+{
+    if (!fmt_entries_to_show ||
+        (section && av_dict_get(fmt_entries_to_show, section, NULL, 0))) {
+        if (print_object_footer)
+            print_object_footer(section);
+        print_object_separator = 1;
+    }
+}
+
+static void avp_int(const char *key, int64_t value)
+{
+    print_integer(key, value);
+    print_item_separator = 1;
+}
+
+static void avp_str(const char *key, const char *value)
+{
+    print_string(key, value);
+    print_item_separator = 1;
+}
+
+static void avp_dict(AVDictionary *dict, const char *name)
+{
+    AVDictionaryEntry *entry = NULL;
+    if (!dict)
+        return;
+    print_object_separator = print_item_separator;
+    avp_section_header(name);
+    while ((entry = av_dict_get(dict, "", entry, AV_DICT_IGNORE_SUFFIX))) {
+        avp_str(entry->key, entry->value);
+    }
+    avp_section_footer(name);
+}
+
 static char *value_string(char *buf, int buf_size, double val, const char *unit)
 {
     if (unit == unit_second_str && use_value_sexagesimal_format) {
@@ -90,11 +443,12 @@ static char *value_string(char *buf, int buf_size, double val, const char *unit)
             val  /= pow(10, index * 3);
             prefix_string = decimal_unit_prefixes[index];
         }
-
-        snprintf(buf, buf_size, "%.3f %s%s", val, prefix_string,
+        snprintf(buf, buf_size, "%.*f%s%s",
+                 index ? 3 : 0, val,
+                 prefix_string,
                  show_value_unit ? unit : "");
     } else {
-        snprintf(buf, buf_size, "%f %s", val, show_value_unit ? unit : "");
+        snprintf(buf, buf_size, "%f%s", val, show_value_unit ? unit : "");
     }
 
     return buf;
@@ -112,7 +466,7 @@ static char *time_value_string(char *buf, int buf_size, int64_t val,
     return buf;
 }
 
-static char *ts_value_string (char *buf, int buf_size, int64_t ts)
+static char *ts_value_string(char *buf, int buf_size, int64_t ts)
 {
     if (ts == AV_NOPTS_VALUE) {
         snprintf(buf, buf_size, "N/A");
@@ -123,6 +477,21 @@ static char *ts_value_string (char *buf, int buf_size, int64_t ts)
     return buf;
 }
 
+static char *rational_string(char *buf, int buf_size, const char *sep,
+                             const AVRational *rat)
+{
+    snprintf(buf, buf_size, "%d%s%d", rat->num, sep, rat->den);
+    return buf;
+}
+
+static char *tag_string(char *buf, int buf_size, int tag)
+{
+    snprintf(buf, buf_size, "0x%04x", tag);
+    return buf;
+}
+
+
+
 static const char *media_type_string(enum AVMediaType media_type)
 {
     switch (media_type) {
@@ -140,25 +509,25 @@ static void show_packet(AVFormatContext *fmt_ctx, AVPacket *pkt)
     char val_str[128];
     AVStream *st = fmt_ctx->streams[pkt->stream_index];
 
-    printf("[PACKET]\n");
-    printf("codec_type=%s\n", media_type_string(st->codec->codec_type));
-    printf("stream_index=%d\n", pkt->stream_index);
-    printf("pts=%s\n", ts_value_string(val_str, sizeof(val_str), pkt->pts));
-    printf("pts_time=%s\n", time_value_string(val_str, sizeof(val_str),
-                                              pkt->pts, &st->time_base));
-    printf("dts=%s\n", ts_value_string(val_str, sizeof(val_str), pkt->dts));
-    printf("dts_time=%s\n", time_value_string(val_str, sizeof(val_str),
-                                              pkt->dts, &st->time_base));
-    printf("duration=%s\n", ts_value_string(val_str, sizeof(val_str),
-                                            pkt->duration));
-    printf("duration_time=%s\n", time_value_string(val_str, sizeof(val_str),
-                                                   pkt->duration,
-                                                   &st->time_base));
-    printf("size=%s\n", value_string(val_str, sizeof(val_str),
-                                     pkt->size, unit_byte_str));
-    printf("pos=%"PRId64"\n", pkt->pos);
-    printf("flags=%c\n", pkt->flags & AV_PKT_FLAG_KEY ? 'K' : '_');
-    printf("[/PACKET]\n");
+    avp_section_header("packet");
+    avp_str("codec_type", media_type_string(st->codec->codec_type));
+    avp_int("stream_index", pkt->stream_index);
+    avp_str("pts", ts_value_string(val_str, sizeof(val_str), pkt->pts));
+    avp_str("pts_time", time_value_string(val_str, sizeof(val_str),
+                                               pkt->pts, &st->time_base));
+    avp_str("dts", ts_value_string(val_str, sizeof(val_str), pkt->dts));
+    avp_str("dts_time", time_value_string(val_str, sizeof(val_str),
+                                               pkt->dts, &st->time_base));
+    avp_str("duration", ts_value_string(val_str, sizeof(val_str),
+                                             pkt->duration));
+    avp_str("duration_time", time_value_string(val_str, sizeof(val_str),
+                                                    pkt->duration,
+                                                    &st->time_base));
+    avp_str("size", value_string(val_str, sizeof(val_str),
+                                      pkt->size, unit_byte_str));
+    avp_int("pos", pkt->pos);
+    avp_str("flags", pkt->flags & AV_PKT_FLAG_KEY ? "K" : "_");
+    avp_section_footer("packet");
 }
 
 static void show_packets(AVFormatContext *fmt_ctx)
@@ -166,9 +535,10 @@ static void show_packets(AVFormatContext *fmt_ctx)
     AVPacket pkt;
 
     av_init_packet(&pkt);
-
+    avp_chapter_header("packets");
     while (!av_read_frame(fmt_ctx, &pkt))
         show_packet(fmt_ctx, &pkt);
+    avp_chapter_footer("packets");
 }
 
 static void show_stream(AVFormatContext *fmt_ctx, int stream_idx)
@@ -177,138 +547,121 @@ static void show_stream(AVFormatContext *fmt_ctx, int stream_idx)
     AVCodecContext *dec_ctx;
     AVCodec *dec;
     char val_str[128];
-    AVDictionaryEntry *tag = NULL;
     AVRational display_aspect_ratio;
 
-    printf("[STREAM]\n");
+    avp_section_header("stream");
 
-    printf("index=%d\n", stream->index);
+    avp_int("index", stream->index);
 
     if ((dec_ctx = stream->codec)) {
         if ((dec = dec_ctx->codec)) {
-            printf("codec_name=%s\n", dec->name);
-            printf("codec_long_name=%s\n", dec->long_name);
+            avp_str("codec_name", dec->name);
+            avp_str("codec_long_name", dec->long_name);
         } else {
-            printf("codec_name=unknown\n");
+            avp_str("codec_name", "unknown");
         }
 
-        printf("codec_type=%s\n", media_type_string(dec_ctx->codec_type));
-        printf("codec_time_base=%d/%d\n",
-               dec_ctx->time_base.num, dec_ctx->time_base.den);
+        avp_str("codec_type", media_type_string(dec_ctx->codec_type));
+        avp_str("codec_time_base",
+                     rational_string(val_str, sizeof(val_str),
+                                     "/", &dec_ctx->time_base));
 
         /* print AVI/FourCC tag */
         av_get_codec_tag_string(val_str, sizeof(val_str), dec_ctx->codec_tag);
-        printf("codec_tag_string=%s\n", val_str);
-        printf("codec_tag=0x%04x\n", dec_ctx->codec_tag);
+        avp_str("codec_tag_string", val_str);
+        avp_str("codec_tag", tag_string(val_str, sizeof(val_str),
+                                             dec_ctx->codec_tag));
 
         switch (dec_ctx->codec_type) {
         case AVMEDIA_TYPE_VIDEO:
-            printf("width=%d\n", dec_ctx->width);
-            printf("height=%d\n", dec_ctx->height);
-            printf("has_b_frames=%d\n", dec_ctx->has_b_frames);
+            avp_int("width", dec_ctx->width);
+            avp_int("height", dec_ctx->height);
+            avp_int("has_b_frames", dec_ctx->has_b_frames);
             if (dec_ctx->sample_aspect_ratio.num) {
-                printf("sample_aspect_ratio=%d:%d\n",
-                       dec_ctx->sample_aspect_ratio.num,
-                       dec_ctx->sample_aspect_ratio.den);
+                avp_str("sample_aspect_ratio",
+                             rational_string(val_str, sizeof(val_str), ":",
+                             &dec_ctx->sample_aspect_ratio));
                 av_reduce(&display_aspect_ratio.num, &display_aspect_ratio.den,
                           dec_ctx->width  * dec_ctx->sample_aspect_ratio.num,
                           dec_ctx->height * dec_ctx->sample_aspect_ratio.den,
                           1024*1024);
-                printf("display_aspect_ratio=%d:%d\n",
-                       display_aspect_ratio.num, display_aspect_ratio.den);
+                avp_str("display_aspect_ratio",
+                             rational_string(val_str, sizeof(val_str), ":",
+                             &display_aspect_ratio));
             }
-            printf("pix_fmt=%s\n",
+            avp_str("pix_fmt",
                    dec_ctx->pix_fmt != PIX_FMT_NONE ? av_pix_fmt_descriptors[dec_ctx->pix_fmt].name
                                                     : "unknown");
-            printf("level=%d\n", dec_ctx->level);
+            avp_int("level", dec_ctx->level);
             break;
 
         case AVMEDIA_TYPE_AUDIO:
-            printf("sample_rate=%s\n", value_string(val_str, sizeof(val_str),
-                                                    dec_ctx->sample_rate,
-                                                    unit_hertz_str));
-            printf("channels=%d\n", dec_ctx->channels);
-            printf("bits_per_sample=%d\n",
-                   av_get_bits_per_sample(dec_ctx->codec_id));
+            avp_str("sample_rate",
+                         value_string(val_str, sizeof(val_str),
+                                      dec_ctx->sample_rate,
+                                      unit_hertz_str));
+            avp_int("channels", dec_ctx->channels);
+            avp_int("bits_per_sample",
+                          av_get_bits_per_sample(dec_ctx->codec_id));
             break;
         }
     } else {
-        printf("codec_type=unknown\n");
+        avp_str("codec_type", "unknown");
     }
 
     if (fmt_ctx->iformat->flags & AVFMT_SHOW_IDS)
-        printf("id=0x%x\n", stream->id);
-    printf("r_frame_rate=%d/%d\n",
-           stream->r_frame_rate.num, stream->r_frame_rate.den);
-    printf("avg_frame_rate=%d/%d\n",
-           stream->avg_frame_rate.num, stream->avg_frame_rate.den);
-    printf("time_base=%d/%d\n",
-           stream->time_base.num, stream->time_base.den);
-    printf("start_time=%s\n",
-           time_value_string(val_str, sizeof(val_str),
+        avp_int("id", stream->id);
+    avp_str("r_frame_rate",
+                 rational_string(val_str, sizeof(val_str), "/",
+                 &stream->r_frame_rate));
+    avp_str("avg_frame_rate",
+                 rational_string(val_str, sizeof(val_str), "/",
+                 &stream->avg_frame_rate));
+    avp_str("time_base",
+                 rational_string(val_str, sizeof(val_str), "/",
+                 &stream->time_base));
+    avp_str("start_time",
+                 time_value_string(val_str, sizeof(val_str),
                              stream->start_time, &stream->time_base));
-    printf("duration=%s\n",
-           time_value_string(val_str, sizeof(val_str),
-                             stream->duration, &stream->time_base));
+    avp_str("duration",
+                 time_value_string(val_str, sizeof(val_str),
+                                   stream->duration, &stream->time_base));
     if (stream->nb_frames)
-        printf("nb_frames=%"PRId64"\n", stream->nb_frames);
+        avp_int("nb_frames", stream->nb_frames);
 
-    while ((tag = av_dict_get(stream->metadata, "", tag,
-                              AV_DICT_IGNORE_SUFFIX)))
-        printf("TAG:%s=%s\n", tag->key, tag->value);
+    avp_dict(stream->metadata, "tags");
 
-    printf("[/STREAM]\n");
-}
-
-static void print_format_entry(const char *tag,
-                               const char *val)
-{
-    if (!fmt_entries_to_show) {
-        if (tag) {
-            printf("%s=%s\n", tag, val);
-        } else {
-            printf("%s\n", val);
-        }
-    } else if (tag && av_dict_get(fmt_entries_to_show, tag, NULL, 0)) {
-        if (nb_fmt_entries_to_show > 1)
-            printf("%s=", tag);
-        printf("%s\n", val);
-    }
+    avp_section_footer("stream");
 }
 
 static void show_format(AVFormatContext *fmt_ctx)
 {
-    AVDictionaryEntry *tag = NULL;
     char val_str[128];
     int64_t size = fmt_ctx->pb ? avio_size(fmt_ctx->pb) : -1;
 
-    print_format_entry(NULL, "[FORMAT]");
-    print_format_entry("filename",         fmt_ctx->filename);
+    avp_section_header("format");
+    avp_str("filename",         fmt_ctx->filename);
     snprintf(val_str, sizeof(val_str) - 1, "%d", fmt_ctx->nb_streams);
-    print_format_entry("nb_streams",       val_str);
-    print_format_entry("format_name",      fmt_ctx->iformat->name);
-    print_format_entry("format_long_name", fmt_ctx->iformat->long_name);
-    print_format_entry("start_time",
+    avp_str("nb_streams",       val_str);
+    avp_str("format_name",      fmt_ctx->iformat->name);
+    avp_str("format_long_name", fmt_ctx->iformat->long_name);
+    avp_str("start_time",
                        time_value_string(val_str, sizeof(val_str),
                                          fmt_ctx->start_time, &AV_TIME_BASE_Q));
-    print_format_entry("duration",
+    avp_str("duration",
                        time_value_string(val_str, sizeof(val_str),
                                          fmt_ctx->duration, &AV_TIME_BASE_Q));
-    print_format_entry("size",
+    avp_str("size",
                        size >= 0 ? value_string(val_str, sizeof(val_str),
                                                 size, unit_byte_str)
                                   : "unknown");
-    print_format_entry("bit_rate",
+    avp_str("bit_rate",
                        value_string(val_str, sizeof(val_str),
                                     fmt_ctx->bit_rate, unit_bit_per_second_str));
 
-    while ((tag = av_dict_get(fmt_ctx->metadata, "", tag,
-                              AV_DICT_IGNORE_SUFFIX))) {
-        snprintf(val_str, sizeof(val_str) - 1, "TAG:%s", tag->key);
-        print_format_entry(val_str, tag->value);
-    }
+    avp_dict(fmt_ctx->metadata, "tags");
 
-    print_format_entry(NULL, "[/FORMAT]");
+    avp_section_footer("format");
 }
 
 static int open_input_file(AVFormatContext **fmt_ctx_ptr, const char *filename)
@@ -355,20 +708,6 @@ static int open_input_file(AVFormatContext **fmt_ctx_ptr, const char *filename)
     return 0;
 }
 
-static void close_input_file(AVFormatContext **ctx_ptr)
-{
-    int i;
-    AVFormatContext *fmt_ctx = *ctx_ptr;
-
-    /* close decoder for each stream */
-    for (i = 0; i < fmt_ctx->nb_streams; i++) {
-        AVStream *stream = fmt_ctx->streams[i];
-
-        avcodec_close(stream->codec);
-    }
-    avformat_close_input(ctx_ptr);
-}
-
 static int probe_file(const char *filename)
 {
     AVFormatContext *fmt_ctx;
@@ -380,14 +719,17 @@ static int probe_file(const char *filename)
     if (do_show_packets)
         show_packets(fmt_ctx);
 
-    if (do_show_streams)
+    if (do_show_streams) {
+        avp_chapter_header("streams");
         for (i = 0; i < fmt_ctx->nb_streams; i++)
             show_stream(fmt_ctx, i);
+        avp_chapter_footer("streams");
+    }
 
     if (do_show_format)
         show_format(fmt_ctx);
 
-    close_input_file(&fmt_ctx);
+    avformat_close_input(&fmt_ctx);
     return 0;
 }
 
@@ -408,10 +750,33 @@ static int opt_format(const char *opt, const char *arg)
     return 0;
 }
 
+static void opt_output_json(void)
+{
+    print_header        = json_print_header;
+    print_footer        = json_print_footer;
+    print_array_header  = json_print_array_header;
+    print_array_footer  = json_print_array_footer;
+    print_object_header = json_print_object_header;
+    print_object_footer = json_print_object_footer;
+
+    print_integer = json_print_integer;
+    print_string  = json_print_string;
+
+}
+
 static int opt_show_format_entry(const char *opt, const char *arg)
 {
     do_show_format = 1;
     nb_fmt_entries_to_show++;
+    print_header        = NULL;
+    print_footer        = NULL;
+    print_array_header  = NULL;
+    print_array_footer  = NULL;
+    print_object_header = NULL;
+    print_object_footer = NULL;
+
+    print_integer = show_format_entry_integer;
+    print_string  = show_format_entry_string;
     av_dict_set(&fmt_entries_to_show, arg, "", 0);
     return 0;
 }
@@ -459,6 +824,7 @@ static const OptionDef options[] = {
       "use sexagesimal format HOURS:MM:SS.MICROSECONDS for time units" },
     { "pretty", 0, {(void*)&opt_pretty},
       "prettify the format of displayed values, make it more human readable" },
+    { "json", OPT_BOOL, {(void*)&do_output_json}, "output a json document instead of an ini one" },
     { "show_format",  OPT_BOOL, {(void*)&do_show_format} , "show format/container info" },
     { "show_format_entry", HAS_ARG, {(void*)opt_show_format_entry},
       "show a particular entry from the format/container info", "entry" },
@@ -469,9 +835,21 @@ static const OptionDef options[] = {
     { NULL, },
 };
 
+static int avp_buf_write(void *opaque, uint8_t *buf, int buf_size)
+{
+    printf("%.*s", buf_size, buf);
+    return 0;
+}
+
+#define AVP_BUFFSIZE 4096
+
 int main(int argc, char **argv)
 {
     int ret;
+    uint8_t *buffer = av_malloc(AVP_BUFFSIZE);
+
+    if (!buffer)
+        exit(1);
 
     parse_loglevel(argc, argv, options);
     av_register_all();
@@ -493,10 +871,19 @@ int main(int argc, char **argv)
         exit(1);
     }
 
-    ret = probe_file(input_filename);
+    avp_out = avio_alloc_context(buffer, AVP_BUFFSIZE, 1, NULL, NULL,
+                                 avp_buf_write, NULL);
+    if (!avp_out)
+        exit(1);
 
-    uninit_opts();
-    av_dict_free(&fmt_entries_to_show);
+    if (do_output_json)
+        opt_output_json();
+
+    avp_header();
+    ret = probe_file(input_filename);
+    avp_footer();
+    avio_flush(avp_out);
+    avio_close(avp_out);
 
     avformat_network_deinit();
 
-- 
1.7.8.rc1



More information about the libav-devel mailing list