/* * Copyright (c) 2008-2010 Stefano Sabatini * * This file is part of FFmpeg. * * FFmpeg is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * FFmpeg is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with FFmpeg; if not, write to the Free Software * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ #include "config.h" #if HAVE_UNISTD_H #include <unistd.h> /* getopt */ #endif #include <stdio.h> #include <string.h> #include "libavutil/channel_layout.h" #include "libavutil/mem.h" #include "libavutil/pixdesc.h" #include "libavfilter/avfilter.h" #if !HAVE_GETOPT #include "compat/getopt.c" #endif static void usage(void) { printf("Convert a libavfilter graph to a dot file.\n"); printf("Usage: graph2dot [OPTIONS]\n"); printf("\n" "Options:\n" "-i INFILE set INFILE as input file, stdin if omitted\n" "-o OUTFILE set OUTFILE as output file, stdout if omitted\n" "-h print this help\n"); } struct line { char data[256]; struct line *next; }; static void print_digraph(FILE *outfile, AVFilterGraph *graph) { int i, j; fprintf(outfile, "digraph G {\n"); fprintf(outfile, "node [shape=box]\n"); fprintf(outfile, "rankdir=LR\n"); for (i = 0; i < graph->nb_filters; i++) { char filter_ctx_label[128]; const AVFilterContext *filter_ctx = graph->filters[i]; snprintf(filter_ctx_label, sizeof(filter_ctx_label), "%s\\n(%s)", filter_ctx->name, filter_ctx->filter->name); for (j = 0; j < filter_ctx->nb_outputs; j++) { AVFilterLink *link = filter_ctx->outputs[j]; if (link) { char dst_filter_ctx_label[128]; const AVFilterContext *dst_filter_ctx = link->dst; snprintf(dst_filter_ctx_label, sizeof(dst_filter_ctx_label), "%s\\n(%s)", dst_filter_ctx->name, dst_filter_ctx->filter->name); fprintf(outfile, "\"%s\" -> \"%s\" [ label= \"inpad:%s -> outpad:%s\\n", filter_ctx_label, dst_filter_ctx_label, avfilter_pad_get_name(link->srcpad, 0), avfilter_pad_get_name(link->dstpad, 0)); if (link->type == AVMEDIA_TYPE_VIDEO) { const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(link->format); fprintf(outfile, "fmt:%s w:%d h:%d tb:%d/%d", desc->name, link->w, link->h, link->time_base.num, link->time_base.den); } else if (link->type == AVMEDIA_TYPE_AUDIO) { char buf[255]; av_get_channel_layout_string(buf, sizeof(buf), -1, link->channel_layout); fprintf(outfile, "fmt:%s sr:%d cl:%s tb:%d/%d", av_get_sample_fmt_name(link->format), link->sample_rate, buf, link->time_base.num, link->time_base.den); } fprintf(outfile, "\" ];\n"); } } } fprintf(outfile, "}\n"); } int main(int argc, char **argv) { const char *outfilename = NULL; const char *infilename = NULL; FILE *outfile = NULL; FILE *infile = NULL; char *graph_string = NULL; AVFilterGraph *graph = av_mallocz(sizeof(AVFilterGraph)); char c; av_log_set_level(AV_LOG_DEBUG); while ((c = getopt(argc, argv, "hi:o:")) != -1) { switch (c) { case 'h': usage(); return 0; case 'i': infilename = optarg; break; case 'o': outfilename = optarg; break; case '?': return 1; } } if (!infilename || !strcmp(infilename, "-")) infilename = "/dev/stdin"; infile = fopen(infilename, "r"); if (!infile) { fprintf(stderr, "Failed to open input file '%s': %s\n", infilename, strerror(errno)); return 1; } if (!outfilename || !strcmp(outfilename, "-")) outfilename = "/dev/stdout"; outfile = fopen(outfilename, "w"); if (!outfile) { fprintf(stderr, "Failed to open output file '%s': %s\n", outfilename, strerror(errno)); return 1; } /* read from infile and put it in a buffer */ { int64_t count = 0; struct line *line, *last_line, *first_line; char *p; last_line = first_line = av_malloc(sizeof(struct line)); if (!last_line) { fprintf(stderr, "Memory allocation failure\n"); return 1; } while (fgets(last_line->data, sizeof(last_line->data), infile)) { struct line *new_line = av_malloc(sizeof(struct line)); if (!new_line) { fprintf(stderr, "Memory allocation failure\n"); return 1; } count += strlen(last_line->data); last_line->next = new_line; last_line = new_line; } last_line->next = NULL; graph_string = av_malloc(count + 1); if (!graph_string) { fprintf(stderr, "Memory allocation failure\n"); return 1; } p = graph_string; for (line = first_line; line->next; line = line->next) { size_t l = strlen(line->data); memcpy(p, line->data, l); p += l; } *p = '\0'; } if (avfilter_graph_parse(graph, graph_string, NULL, NULL, NULL) < 0) { fprintf(stderr, "Failed to parse the graph description\n"); return 1; } if (avfilter_graph_config(graph, NULL) < 0) return 1; print_digraph(outfile, graph); fflush(outfile); return 0; }