avfilter/f_bench: Fix comparission condition
[ffmpeg.git] / libavfilter / graphdump.c
1 /*
2  * Filter graphs to bad ASCII-art
3  * Copyright (c) 2012 Nicolas George
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21
22 #include <string.h>
23
24 #include "libavutil/channel_layout.h"
25 #include "libavutil/bprint.h"
26 #include "libavutil/pixdesc.h"
27 #include "avfilter.h"
28 #include "avfiltergraph.h"
29 #include "internal.h"
30
31 static int print_link_prop(AVBPrint *buf, AVFilterLink *link)
32 {
33     char *format;
34     char layout[64];
35     AVBPrint dummy_buffer = { 0 };
36
37     if (!buf)
38         buf = &dummy_buffer;
39     switch (link->type) {
40         case AVMEDIA_TYPE_VIDEO:
41             format = av_x_if_null(av_get_pix_fmt_name(link->format), "?");
42             av_bprintf(buf, "[%dx%d %d:%d %s]", link->w, link->h,
43                     link->sample_aspect_ratio.num,
44                     link->sample_aspect_ratio.den,
45                     format);
46             break;
47
48         case AVMEDIA_TYPE_AUDIO:
49             av_get_channel_layout_string(layout, sizeof(layout),
50                                          link->channels, link->channel_layout);
51             format = av_x_if_null(av_get_sample_fmt_name(link->format), "?");
52             av_bprintf(buf, "[%dHz %s:%s]",
53                        (int)link->sample_rate, format, layout);
54             break;
55
56         default:
57             av_bprintf(buf, "?");
58             break;
59     }
60     return buf->len;
61 }
62
63 static void avfilter_graph_dump_to_buf(AVBPrint *buf, AVFilterGraph *graph)
64 {
65     unsigned i, j, x, e;
66
67     for (i = 0; i < graph->nb_filters; i++) {
68         AVFilterContext *filter = graph->filters[i];
69         unsigned max_src_name = 0, max_dst_name = 0;
70         unsigned max_in_name  = 0, max_out_name = 0;
71         unsigned max_in_fmt   = 0, max_out_fmt  = 0;
72         unsigned width, height, in_indent;
73         unsigned lname = strlen(filter->name);
74         unsigned ltype = strlen(filter->filter->name);
75
76         for (j = 0; j < filter->nb_inputs; j++) {
77             AVFilterLink *l = filter->inputs[j];
78             unsigned ln = strlen(l->src->name) + 1 + strlen(l->srcpad->name);
79             max_src_name = FFMAX(max_src_name, ln);
80             max_in_name = FFMAX(max_in_name, strlen(l->dstpad->name));
81             max_in_fmt = FFMAX(max_in_fmt, print_link_prop(NULL, l));
82         }
83         for (j = 0; j < filter->nb_outputs; j++) {
84             AVFilterLink *l = filter->outputs[j];
85             unsigned ln = strlen(l->dst->name) + 1 + strlen(l->dstpad->name);
86             max_dst_name = FFMAX(max_dst_name, ln);
87             max_out_name = FFMAX(max_out_name, strlen(l->srcpad->name));
88             max_out_fmt = FFMAX(max_out_fmt, print_link_prop(NULL, l));
89         }
90         in_indent = max_src_name + max_in_name + max_in_fmt;
91         in_indent += in_indent ? 4 : 0;
92         width = FFMAX(lname + 2, ltype + 4);
93         height = FFMAX3(2, filter->nb_inputs, filter->nb_outputs);
94         av_bprint_chars(buf, ' ', in_indent);
95         av_bprintf(buf, "+");
96         av_bprint_chars(buf, '-', width);
97         av_bprintf(buf, "+\n");
98         for (j = 0; j < height; j++) {
99             unsigned in_no  = j - (height - filter->nb_inputs ) / 2;
100             unsigned out_no = j - (height - filter->nb_outputs) / 2;
101
102             /* Input link */
103             if (in_no < filter->nb_inputs) {
104                 AVFilterLink *l = filter->inputs[in_no];
105                 e = buf->len + max_src_name + 2;
106                 av_bprintf(buf, "%s:%s", l->src->name, l->srcpad->name);
107                 av_bprint_chars(buf, '-', e - buf->len);
108                 e = buf->len + max_in_fmt + 2 +
109                     max_in_name - strlen(l->dstpad->name);
110                 print_link_prop(buf, l);
111                 av_bprint_chars(buf, '-', e - buf->len);
112                 av_bprintf(buf, "%s", l->dstpad->name);
113             } else {
114                 av_bprint_chars(buf, ' ', in_indent);
115             }
116
117             /* Filter */
118             av_bprintf(buf, "|");
119             if (j == (height - 2) / 2) {
120                 x = (width - lname) / 2;
121                 av_bprintf(buf, "%*s%-*s", x, "", width - x, filter->name);
122             } else if (j == (height - 2) / 2 + 1) {
123                 x = (width - ltype - 2) / 2;
124                 av_bprintf(buf, "%*s(%s)%*s", x, "", filter->filter->name,
125                         width - ltype - 2 - x, "");
126             } else {
127                 av_bprint_chars(buf, ' ', width);
128             }
129             av_bprintf(buf, "|");
130
131             /* Output link */
132             if (out_no < filter->nb_outputs) {
133                 AVFilterLink *l = filter->outputs[out_no];
134                 unsigned ln = strlen(l->dst->name) + 1 +
135                               strlen(l->dstpad->name);
136                 e = buf->len + max_out_name + 2;
137                 av_bprintf(buf, "%s", l->srcpad->name);
138                 av_bprint_chars(buf, '-', e - buf->len);
139                 e = buf->len + max_out_fmt + 2 +
140                     max_dst_name - ln;
141                 print_link_prop(buf, l);
142                 av_bprint_chars(buf, '-', e - buf->len);
143                 av_bprintf(buf, "%s:%s", l->dst->name, l->dstpad->name);
144             }
145             av_bprintf(buf, "\n");
146         }
147         av_bprint_chars(buf, ' ', in_indent);
148         av_bprintf(buf, "+");
149         av_bprint_chars(buf, '-', width);
150         av_bprintf(buf, "+\n");
151         av_bprintf(buf, "\n");
152     }
153 }
154
155 char *avfilter_graph_dump(AVFilterGraph *graph, const char *options)
156 {
157     AVBPrint buf;
158     char *dump;
159
160     av_bprint_init(&buf, 0, 0);
161     avfilter_graph_dump_to_buf(&buf, graph);
162     av_bprint_init(&buf, buf.len + 1, buf.len + 1);
163     avfilter_graph_dump_to_buf(&buf, graph);
164     av_bprint_finalize(&buf, &dump);
165     return dump;
166 }