Subversion Repositories Kolibri OS

Rev

Go to most recent revision | Blame | Last modification | View Log | RSS feed

  1. /*
  2.  * Copyright (c) 2008-2010 Stefano Sabatini
  3.  *
  4.  * This file is part of FFmpeg.
  5.  *
  6.  * FFmpeg is free software; you can redistribute it and/or
  7.  * modify it under the terms of the GNU Lesser General Public
  8.  * License as published by the Free Software Foundation; either
  9.  * version 2.1 of the License, or (at your option) any later version.
  10.  *
  11.  * FFmpeg is distributed in the hope that it will be useful,
  12.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14.  * Lesser General Public License for more details.
  15.  *
  16.  * You should have received a copy of the GNU Lesser General Public
  17.  * License along with FFmpeg; if not, write to the Free Software
  18.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  19.  */
  20.  
  21. #include "config.h"
  22. #if HAVE_UNISTD_H
  23. #include <unistd.h>             /* getopt */
  24. #endif
  25. #include <stdio.h>
  26. #include <string.h>
  27.  
  28. #include "libavutil/channel_layout.h"
  29. #include "libavutil/mem.h"
  30. #include "libavutil/pixdesc.h"
  31. #include "libavfilter/avfilter.h"
  32.  
  33. #if !HAVE_GETOPT
  34. #include "compat/getopt.c"
  35. #endif
  36.  
  37. static void usage(void)
  38. {
  39.     printf("Convert a libavfilter graph to a dot file.\n");
  40.     printf("Usage: graph2dot [OPTIONS]\n");
  41.     printf("\n"
  42.            "Options:\n"
  43.            "-i INFILE         set INFILE as input file, stdin if omitted\n"
  44.            "-o OUTFILE        set OUTFILE as output file, stdout if omitted\n"
  45.            "-h                print this help\n");
  46. }
  47.  
  48. struct line {
  49.     char data[256];
  50.     struct line *next;
  51. };
  52.  
  53. static void print_digraph(FILE *outfile, AVFilterGraph *graph)
  54. {
  55.     int i, j;
  56.  
  57.     fprintf(outfile, "digraph G {\n");
  58.     fprintf(outfile, "node [shape=box]\n");
  59.     fprintf(outfile, "rankdir=LR\n");
  60.  
  61.     for (i = 0; i < graph->nb_filters; i++) {
  62.         char filter_ctx_label[128];
  63.         const AVFilterContext *filter_ctx = graph->filters[i];
  64.  
  65.         snprintf(filter_ctx_label, sizeof(filter_ctx_label), "%s\\n(%s)",
  66.                  filter_ctx->name,
  67.                  filter_ctx->filter->name);
  68.  
  69.         for (j = 0; j < filter_ctx->nb_outputs; j++) {
  70.             AVFilterLink *link = filter_ctx->outputs[j];
  71.             if (link) {
  72.                 char dst_filter_ctx_label[128];
  73.                 const AVFilterContext *dst_filter_ctx = link->dst;
  74.  
  75.                 snprintf(dst_filter_ctx_label, sizeof(dst_filter_ctx_label),
  76.                          "%s\\n(%s)",
  77.                          dst_filter_ctx->name,
  78.                          dst_filter_ctx->filter->name);
  79.  
  80.                 fprintf(outfile, "\"%s\" -> \"%s\" [ label= \"inpad:%s -> outpad:%s\\n",
  81.                         filter_ctx_label, dst_filter_ctx_label,
  82.                         link->srcpad->name, link->dstpad->name);
  83.  
  84.                 if (link->type == AVMEDIA_TYPE_VIDEO) {
  85.                     const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(link->format);
  86.                     fprintf(outfile,
  87.                             "fmt:%s w:%d h:%d tb:%d/%d",
  88.                             desc->name,
  89.                             link->w, link->h,
  90.                             link->time_base.num, link->time_base.den);
  91.                 } else if (link->type == AVMEDIA_TYPE_AUDIO) {
  92.                     char buf[255];
  93.                     av_get_channel_layout_string(buf, sizeof(buf), -1,
  94.                                                  link->channel_layout);
  95.                     fprintf(outfile,
  96.                             "fmt:%s sr:%d cl:%s tb:%d/%d",
  97.                             av_get_sample_fmt_name(link->format),
  98.                             link->sample_rate, buf,
  99.                             link->time_base.num, link->time_base.den);
  100.                 }
  101.                 fprintf(outfile, "\" ];\n");
  102.             }
  103.         }
  104.     }
  105.     fprintf(outfile, "}\n");
  106. }
  107.  
  108. int main(int argc, char **argv)
  109. {
  110.     const char *outfilename = NULL;
  111.     const char *infilename  = NULL;
  112.     FILE *outfile           = NULL;
  113.     FILE *infile            = NULL;
  114.     char *graph_string      = NULL;
  115.     AVFilterGraph *graph = av_mallocz(sizeof(AVFilterGraph));
  116.     char c;
  117.  
  118.     av_log_set_level(AV_LOG_DEBUG);
  119.  
  120.     while ((c = getopt(argc, argv, "hi:o:")) != -1) {
  121.         switch (c) {
  122.         case 'h':
  123.             usage();
  124.             return 0;
  125.         case 'i':
  126.             infilename = optarg;
  127.             break;
  128.         case 'o':
  129.             outfilename = optarg;
  130.             break;
  131.         case '?':
  132.             return 1;
  133.         }
  134.     }
  135.  
  136.     if (!infilename || !strcmp(infilename, "-"))
  137.         infilename = "/dev/stdin";
  138.     infile = fopen(infilename, "r");
  139.     if (!infile) {
  140.         fprintf(stderr, "Failed to open input file '%s': %s\n",
  141.                 infilename, strerror(errno));
  142.         return 1;
  143.     }
  144.  
  145.     if (!outfilename || !strcmp(outfilename, "-"))
  146.         outfilename = "/dev/stdout";
  147.     outfile = fopen(outfilename, "w");
  148.     if (!outfile) {
  149.         fprintf(stderr, "Failed to open output file '%s': %s\n",
  150.                 outfilename, strerror(errno));
  151.         return 1;
  152.     }
  153.  
  154.     /* read from infile and put it in a buffer */
  155.     {
  156.         unsigned int count = 0;
  157.         struct line *line, *last_line, *first_line;
  158.         char *p;
  159.         last_line = first_line = av_malloc(sizeof(struct line));
  160.  
  161.         while (fgets(last_line->data, sizeof(last_line->data), infile)) {
  162.             struct line *new_line = av_malloc(sizeof(struct line));
  163.             count += strlen(last_line->data);
  164.             last_line->next = new_line;
  165.             last_line       = new_line;
  166.         }
  167.         last_line->next = NULL;
  168.  
  169.         graph_string = av_malloc(count + 1);
  170.         p = graph_string;
  171.         for (line = first_line; line->next; line = line->next) {
  172.             unsigned int l = strlen(line->data);
  173.             memcpy(p, line->data, l);
  174.             p += l;
  175.         }
  176.         *p = '\0';
  177.     }
  178.  
  179.     avfilter_register_all();
  180.  
  181.     if (avfilter_graph_parse(graph, graph_string, NULL, NULL, NULL) < 0) {
  182.         fprintf(stderr, "Failed to parse the graph description\n");
  183.         return 1;
  184.     }
  185.  
  186.     if (avfilter_graph_config(graph, NULL) < 0)
  187.         return 1;
  188.  
  189.     print_digraph(outfile, graph);
  190.     fflush(outfile);
  191.  
  192.     return 0;
  193. }
  194.