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