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