[FFmpeg-devel] [PATCH v7 1/3] avfilter/graphdump: support for the graph2dot function

lance.lmwang at gmail.com lance.lmwang at gmail.com
Thu May 28 17:12:38 EEST 2020


On Mon, May 25, 2020 at 07:50:24AM +0800, lance.lmwang at gmail.com wrote:
> From: Limin Wang <lance.lmwang at gmail.com>
> 
> Signed-off-by: Limin Wang <lance.lmwang at gmail.com>
> ---
>  libavfilter/Makefile    |   1 -
>  libavfilter/graphdump.c |  89 +++++++++++++++++++++
>  tools/graph2dot.c       | 204 ------------------------------------------------
>  3 files changed, 89 insertions(+), 205 deletions(-)
>  delete mode 100644 tools/graph2dot.c
> 
> diff --git a/libavfilter/Makefile b/libavfilter/Makefile
> index 4d07bb6..291126e 100644
> --- a/libavfilter/Makefile
> +++ b/libavfilter/Makefile
> @@ -526,7 +526,6 @@ SKIPHEADERS-$(CONFIG_VULKAN)                 += vulkan.h
>  
>  OBJS-$(CONFIG_LIBGLSLANG)                    += glslang.o
>  
> -TOOLS     = graph2dot
>  TESTPROGS = drawutils filtfmts formats integral
>  
>  TOOLS-$(CONFIG_LIBZMQ) += zmqsend
> diff --git a/libavfilter/graphdump.c b/libavfilter/graphdump.c
> index 79ef1a7..97d4f39 100644
> --- a/libavfilter/graphdump.c
> +++ b/libavfilter/graphdump.c
> @@ -151,15 +151,104 @@ static void avfilter_graph_dump_to_buf(AVBPrint *buf, AVFilterGraph *graph)
>      }
>  }
>  
> +static void avfilter_graph2dot_to_buf(AVBPrint *buf, AVFilterGraph *graph)
> +{
> +    int i, j;
> +
> +    av_bprintf(buf, "digraph G {\n");
> +    av_bprintf(buf, "node [shape=box]\n");
> +    av_bprintf(buf, "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);
> +
> +                av_bprintf(buf, "\"%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);
> +                    av_bprintf(buf,
> +                            "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 audio_buf[255];
> +                    av_get_channel_layout_string(audio_buf, sizeof(audio_buf), -1,
> +                                                 link->channel_layout);
> +                    av_bprintf(buf,
> +                            "fmt:%s sr:%d cl:%s tb:%d/%d",
> +                            av_get_sample_fmt_name(link->format),
> +                            link->sample_rate, audio_buf,
> +                            link->time_base.num, link->time_base.den);
> +                }
> +                av_bprintf(buf, "\" ];\n");
> +            }
> +        }
> +    }
> +    av_bprintf(buf, "}\n");
> +}
> +
>  char *avfilter_graph_dump(AVFilterGraph *graph, const char *options)
>  {
>      AVBPrint buf;
>      char *dump = NULL;
> +    int ret;
> +    AVDictionary *dict = NULL;
> +    AVDictionaryEntry *format = NULL;
> +    AVDictionaryEntry *filename = NULL;
> +
> +    ret = av_dict_parse_string(&dict, options, "=", ":", 0);
> +    if (ret < 0) {
> +        av_dict_free(&dict);
> +        return NULL;
> +    }
> +    format = av_dict_get(dict, "fmt", NULL, 0);
>  
> +    if (format && !av_strcasecmp(format->value, "DOT")) {
> +    av_bprint_init(&buf, 0, AV_BPRINT_SIZE_UNLIMITED);
> +    avfilter_graph2dot_to_buf(&buf, graph);
> +    av_bprint_finalize(&buf, &dump);
> +    } else {
>      av_bprint_init(&buf, 0, AV_BPRINT_SIZE_COUNT_ONLY);
>      avfilter_graph_dump_to_buf(&buf, graph);
>      av_bprint_init(&buf, buf.len + 1, buf.len + 1);
>      avfilter_graph_dump_to_buf(&buf, graph);
>      av_bprint_finalize(&buf, &dump);
> +    }
> +
> +    if (filename = av_dict_get(dict, "filename", NULL, 0)) {
> +        FILE* file = fopen(filename->value, "w");
> +        if (!file) {
> +            av_log(graph, AV_LOG_ERROR, "failed to open: %s \n", filename->value);
> +            av_freep(&dump);
> +            return NULL;
> +        }
> +        if (dump) {
> +            fputs(dump, file);
> +            fflush(file);
> +        }
> +        fclose(file);
> +    }
> +
> +    av_dict_free(&dict);
>      return dump;
>  }
> diff --git a/tools/graph2dot.c b/tools/graph2dot.c
> deleted file mode 100644
> index d5c1e4e..0000000
> --- a/tools/graph2dot.c
> +++ /dev/null
> @@ -1,204 +0,0 @@
> -/*
> - * 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;
> -}
> -- 
> 2.6.4
> 

will apply the patchset tomorrow if no further comments.


-- 
Thanks,
Limin Wang


More information about the ffmpeg-devel mailing list