46a2d9447d3d74a985fc64690ec8e711de92f4ce
[ffmpeg.git] / libavfilter / tests / filtfmts.c
1 /*
2  * Copyright (c) 2009 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 <stdio.h>
22
23 #include "libavutil/channel_layout.h"
24 #include "libavutil/mem.h"
25 #include "libavutil/pixdesc.h"
26 #include "libavutil/samplefmt.h"
27
28 #include "libavfilter/avfilter.h"
29 #include "libavfilter/formats.h"
30
31 static void print_formats(AVFilterContext *filter_ctx)
32 {
33     int i, j;
34
35 #define PRINT_FMTS(inout, outin, INOUT)                                 \
36     for (i = 0; i < filter_ctx->nb_##inout##puts; i++) {                     \
37         if (filter_ctx->inout##puts[i]->type == AVMEDIA_TYPE_VIDEO) {   \
38             AVFilterFormats *fmts =                                     \
39                 filter_ctx->inout##puts[i]->outin##_formats;            \
40             for (j = 0; j < fmts->nb_formats; j++)                    \
41                 if(av_get_pix_fmt_name(fmts->formats[j]))               \
42                 printf(#INOUT "PUT[%d] %s: fmt:%s\n",                   \
43                        i, avfilter_pad_get_name(filter_ctx->inout##put_pads, i),      \
44                        av_get_pix_fmt_name(fmts->formats[j]));          \
45         } else if (filter_ctx->inout##puts[i]->type == AVMEDIA_TYPE_AUDIO) { \
46             AVFilterFormats *fmts;                                      \
47             AVFilterChannelLayouts *layouts;                            \
48                                                                         \
49             fmts = filter_ctx->inout##puts[i]->outin##_formats;         \
50             for (j = 0; j < fmts->nb_formats; j++)                    \
51                 printf(#INOUT "PUT[%d] %s: fmt:%s\n",                   \
52                        i, avfilter_pad_get_name(filter_ctx->inout##put_pads, i),      \
53                        av_get_sample_fmt_name(fmts->formats[j]));       \
54                                                                         \
55             layouts = filter_ctx->inout##puts[i]->outin##_channel_layouts; \
56             for (j = 0; j < layouts->nb_channel_layouts; j++) {                  \
57                 char buf[256];                                          \
58                 av_get_channel_layout_string(buf, sizeof(buf), -1,      \
59                                              layouts->channel_layouts[j]);         \
60                 printf(#INOUT "PUT[%d] %s: chlayout:%s\n",              \
61                        i, avfilter_pad_get_name(filter_ctx->inout##put_pads, i), buf); \
62             }                                                           \
63         }                                                               \
64     }                                                                   \
65
66     PRINT_FMTS(in,  out, IN);
67     PRINT_FMTS(out, in,  OUT);
68 }
69
70 int main(int argc, char **argv)
71 {
72     AVFilter *filter;
73     AVFilterContext *filter_ctx;
74     AVFilterGraph *graph_ctx;
75     const char *filter_name;
76     const char *filter_args = NULL;
77     int i;
78     int ret = 0;
79
80     av_log_set_level(AV_LOG_DEBUG);
81
82     if (argc < 2) {
83         fprintf(stderr, "Missing filter name as argument\n");
84         return 1;
85     }
86
87     filter_name = argv[1];
88     if (argc > 2)
89         filter_args = argv[2];
90
91     /* allocate graph */
92     graph_ctx = avfilter_graph_alloc();
93     if (!graph_ctx)
94         return 1;
95
96     avfilter_register_all();
97
98     /* get a corresponding filter and open it */
99     if (!(filter = avfilter_get_by_name(filter_name))) {
100         fprintf(stderr, "Unrecognized filter with name '%s'\n", filter_name);
101         return 1;
102     }
103
104     /* open filter and add it to the graph */
105     if (!(filter_ctx = avfilter_graph_alloc_filter(graph_ctx, filter, filter_name))) {
106         fprintf(stderr, "Impossible to open filter with name '%s'\n",
107                 filter_name);
108         return 1;
109     }
110     if (avfilter_init_str(filter_ctx, filter_args) < 0) {
111         fprintf(stderr, "Impossible to init filter '%s' with arguments '%s'\n",
112                 filter_name, filter_args);
113         return 1;
114     }
115
116     /* create a link for each of the input pads */
117     for (i = 0; i < filter_ctx->nb_inputs; i++) {
118         AVFilterLink *link = av_mallocz(sizeof(AVFilterLink));
119         if (!link) {
120             fprintf(stderr, "Unable to allocate memory for filter input link\n");
121             ret = 1;
122             goto fail;
123         }
124         link->type = avfilter_pad_get_type(filter_ctx->input_pads, i);
125         filter_ctx->inputs[i] = link;
126     }
127     for (i = 0; i < filter_ctx->nb_outputs; i++) {
128         AVFilterLink *link = av_mallocz(sizeof(AVFilterLink));
129         if (!link) {
130             fprintf(stderr, "Unable to allocate memory for filter output link\n");
131             ret = 1;
132             goto fail;
133         }
134         link->type = avfilter_pad_get_type(filter_ctx->output_pads, i);
135         filter_ctx->outputs[i] = link;
136     }
137
138     if (filter->query_formats)
139         filter->query_formats(filter_ctx);
140     else
141         ff_default_query_formats(filter_ctx);
142
143     print_formats(filter_ctx);
144
145 fail:
146     avfilter_free(filter_ctx);
147     avfilter_graph_free(&graph_ctx);
148     fflush(stdout);
149     return ret;
150 }