ffmpeg_filter.c

   1 /*
   2  * ffmpeg filter configuration
   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 <stdint.h>
  22
  23 #include "ffmpeg.h"
  24
  25 #include "libavfilter/avfilter.h"
  26 #include "libavfilter/buffersink.h"
  27 #include "libavfilter/buffersrc.h"
  28
  29 #include "libavresample/avresample.h"
  30
  31 #include "libavutil/avassert.h"
  32 #include "libavutil/avstring.h"
  33 #include "libavutil/bprint.h"
  34 #include "libavutil/channel_layout.h"
  35 #include "libavutil/display.h"
  36 #include "libavutil/opt.h"
  37 #include "libavutil/pixdesc.h"
  38 #include "libavutil/pixfmt.h"
  39 #include "libavutil/imgutils.h"
  40 #include "libavutil/samplefmt.h"
  41
  42 static const enum AVPixelFormat *get_compliance_unofficial_pix_fmts(enum AVCodecID codec_id, const enum AVPixelFormat default_formats[])
  43 {
  44     static const enum AVPixelFormat mjpeg_formats[] =
  45         { AV_PIX_FMT_YUVJ420P, AV_PIX_FMT_YUVJ422P, AV_PIX_FMT_YUVJ444P,
  46           AV_PIX_FMT_YUV420P,  AV_PIX_FMT_YUV422P,  AV_PIX_FMT_YUV444P,
  47           AV_PIX_FMT_NONE };
  48     static const enum AVPixelFormat ljpeg_formats[] =
  49         { AV_PIX_FMT_BGR24   , AV_PIX_FMT_BGRA    , AV_PIX_FMT_BGR0,
  50           AV_PIX_FMT_YUVJ420P, AV_PIX_FMT_YUVJ444P, AV_PIX_FMT_YUVJ422P,
  51           AV_PIX_FMT_YUV420P , AV_PIX_FMT_YUV444P , AV_PIX_FMT_YUV422P,
  52           AV_PIX_FMT_NONE};
  53
  54     if (codec_id == AV_CODEC_ID_MJPEG) {
  55         return mjpeg_formats;
  56     } else if (codec_id == AV_CODEC_ID_LJPEG) {
  57         return ljpeg_formats;
  58     } else {
  59         return default_formats;
  60     }
  61 }
  62
  63 enum AVPixelFormat choose_pixel_fmt(AVStream *st, AVCodecContext *enc_ctx, AVCodec *codec, enum AVPixelFormat target)
  64 {
  65     if (codec && codec->pix_fmts) {
  66         const enum AVPixelFormat *p = codec->pix_fmts;
  67         const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(target);
  68         int has_alpha = desc ? desc->nb_components % 2 == 0 : 0;
  69         enum AVPixelFormat best= AV_PIX_FMT_NONE;
  70
  71         if (enc_ctx->strict_std_compliance <= FF_COMPLIANCE_UNOFFICIAL) {
  72             p = get_compliance_unofficial_pix_fmts(enc_ctx->codec_id, p);
  73         }
  74         for (; *p != AV_PIX_FMT_NONE; p++) {
  75             best= avcodec_find_best_pix_fmt_of_2(best, *p, target, has_alpha, NULL);
  76             if (*p == target)
  77                 break;
  78         }
  79         if (*p == AV_PIX_FMT_NONE) {
  80             if (target != AV_PIX_FMT_NONE)
  81                 av_log(NULL, AV_LOG_WARNING,
  82                        "Incompatible pixel format '%s' for codec '%s', auto-selecting format '%s'\n",
  83                        av_get_pix_fmt_name(target),
  84                        codec->name,
  85                        av_get_pix_fmt_name(best));
  86             return best;
  87         }
  88     }
  89     return target;
  90 }
  91
  92 void choose_sample_fmt(AVStream *st, AVCodec *codec)
  93 {
  94     if (codec && codec->sample_fmts) {
  95         const enum AVSampleFormat *p = codec->sample_fmts;
  96         for (; *p != -1; p++) {
  97             if (*p == st->codecpar->format)
  98                 break;
  99         }
 100         if (*p == -1) {
 101             if((codec->capabilities & AV_CODEC_CAP_LOSSLESS) && av_get_sample_fmt_name(st->codecpar->format) > av_get_sample_fmt_name(codec->sample_fmts[0]))
 102                 av_log(NULL, AV_LOG_ERROR, "Conversion will not be lossless.\n");
 103             if(av_get_sample_fmt_name(st->codecpar->format))
 104             av_log(NULL, AV_LOG_WARNING,
 105                    "Incompatible sample format '%s' for codec '%s', auto-selecting format '%s'\n",
 106                    av_get_sample_fmt_name(st->codecpar->format),
 107                    codec->name,
 108                    av_get_sample_fmt_name(codec->sample_fmts[0]));
 109             st->codecpar->format = codec->sample_fmts[0];
 110         }
 111     }
 112 }
 113
 114 static char *choose_pix_fmts(OutputFilter *ofilter)
 115 {
 116     OutputStream *ost = ofilter->ost;
 117     AVDictionaryEntry *strict_dict = av_dict_get(ost->encoder_opts, "strict", NULL, 0);
 118     if (strict_dict)
 119         // used by choose_pixel_fmt() and below
 120         av_opt_set(ost->enc_ctx, "strict", strict_dict->value, 0);
 121
 122      if (ost->keep_pix_fmt) {
 123         avfilter_graph_set_auto_convert(ofilter->graph->graph,
 124                                             AVFILTER_AUTO_CONVERT_NONE);
 125         if (ost->enc_ctx->pix_fmt == AV_PIX_FMT_NONE)
 126             return NULL;
 127         return av_strdup(av_get_pix_fmt_name(ost->enc_ctx->pix_fmt));
 128     }
 129     if (ost->enc_ctx->pix_fmt != AV_PIX_FMT_NONE) {
 130         return av_strdup(av_get_pix_fmt_name(choose_pixel_fmt(ost->st, ost->enc_ctx, ost->enc, ost->enc_ctx->pix_fmt)));
 131     } else if (ost->enc && ost->enc->pix_fmts) {
 132         const enum AVPixelFormat *p;
 133         AVIOContext *s = NULL;
 134         uint8_t *ret;
 135         int len;
 136
 137         if (avio_open_dyn_buf(&s) < 0)
 138             exit_program(1);
 139
 140         p = ost->enc->pix_fmts;
 141         if (ost->enc_ctx->strict_std_compliance <= FF_COMPLIANCE_UNOFFICIAL) {
 142             p = get_compliance_unofficial_pix_fmts(ost->enc_ctx->codec_id, p);
 143         }
 144
 145         for (; *p != AV_PIX_FMT_NONE; p++) {
 146             const char *name = av_get_pix_fmt_name(*p);
 147             avio_printf(s, "%s|", name);
 148         }
 149         len = avio_close_dyn_buf(s, &ret);
 150         ret[len - 1] = 0;
 151         return ret;
 152     } else
 153         return NULL;
 154 }
 155
 156 /* Define a function for building a string containing a list of
 157  * allowed formats. */
 158 #define DEF_CHOOSE_FORMAT(suffix, type, var, supported_list, none, get_name)   \
 159 static char *choose_ ## suffix (OutputFilter *ofilter)                         \
 160 {                                                                              \
 161     if (ofilter->var != none) {                                                \
 162         get_name(ofilter->var);                                                \
 163         return av_strdup(name);                                                \
 164     } else if (ofilter->supported_list) {                                      \
 165         const type *p;                                                         \
 166         AVIOContext *s = NULL;                                                 \
 167         uint8_t *ret;                                                          \
 168         int len;                                                               \
 169                                                                                \
 170         if (avio_open_dyn_buf(&s) < 0)                                         \
 171             exit_program(1);                                                           \
 172                                                                                \
 173         for (p = ofilter->supported_list; *p != none; p++) {                   \
 174             get_name(*p);                                                      \
 175             avio_printf(s, "%s|", name);                                       \
 176         }                                                                      \
 177         len = avio_close_dyn_buf(s, &ret);                                     \
 178         ret[len - 1] = 0;                                                      \
 179         return ret;                                                            \
 180     } else                                                                     \
 181         return NULL;                                                           \
 182 }
 183
 184 //DEF_CHOOSE_FORMAT(pix_fmts, enum AVPixelFormat, format, formats, AV_PIX_FMT_NONE,
 185 //                  GET_PIX_FMT_NAME)
 186
 187 DEF_CHOOSE_FORMAT(sample_fmts, enum AVSampleFormat, format, formats,
 188                   AV_SAMPLE_FMT_NONE, GET_SAMPLE_FMT_NAME)
 189
 190 DEF_CHOOSE_FORMAT(sample_rates, int, sample_rate, sample_rates, 0,
 191                   GET_SAMPLE_RATE_NAME)
 192
 193 DEF_CHOOSE_FORMAT(channel_layouts, uint64_t, channel_layout, channel_layouts, 0,
 194                   GET_CH_LAYOUT_NAME)
 195
 196 int init_simple_filtergraph(InputStream *ist, OutputStream *ost)
 197 {
 198     FilterGraph *fg = av_mallocz(sizeof(*fg));
 199
 200     if (!fg)
 201         exit_program(1);
 202     fg->index = nb_filtergraphs;
 203
 204     GROW_ARRAY(fg->outputs, fg->nb_outputs);
 205     if (!(fg->outputs[0] = av_mallocz(sizeof(*fg->outputs[0]))))
 206         exit_program(1);
 207     fg->outputs[0]->ost   = ost;
 208     fg->outputs[0]->graph = fg;
 209     fg->outputs[0]->format = -1;
 210
 211     ost->filter = fg->outputs[0];
 212
 213     GROW_ARRAY(fg->inputs, fg->nb_inputs);
 214     if (!(fg->inputs[0] = av_mallocz(sizeof(*fg->inputs[0]))))
 215         exit_program(1);
 216     fg->inputs[0]->ist   = ist;
 217     fg->inputs[0]->graph = fg;
 218     fg->inputs[0]->format = -1;
 219
 220     fg->inputs[0]->frame_queue = av_fifo_alloc(8 * sizeof(AVFrame*));
 221     if (!fg->inputs[0]->frame_queue)
 222         exit_program(1);
 223
 224     GROW_ARRAY(ist->filters, ist->nb_filters);
 225     ist->filters[ist->nb_filters - 1] = fg->inputs[0];
 226
 227     GROW_ARRAY(filtergraphs, nb_filtergraphs);
 228     filtergraphs[nb_filtergraphs - 1] = fg;
 229
 230     return 0;
 231 }
 232
 233 static char *describe_filter_link(FilterGraph *fg, AVFilterInOut *inout, int in)
 234 {
 235     AVFilterContext *ctx = inout->filter_ctx;
 236     AVFilterPad *pads = in ? ctx->input_pads  : ctx->output_pads;
 237     int       nb_pads = in ? ctx->nb_inputs   : ctx->nb_outputs;
 238     AVIOContext *pb;
 239     uint8_t *res = NULL;
 240
 241     if (avio_open_dyn_buf(&pb) < 0)
 242         exit_program(1);
 243
 244     avio_printf(pb, "%s", ctx->filter->name);
 245     if (nb_pads > 1)
 246         avio_printf(pb, ":%s", avfilter_pad_get_name(pads, inout->pad_idx));
 247     avio_w8(pb, 0);
 248     avio_close_dyn_buf(pb, &res);
 249     return res;
 250 }
 251
 252 static void init_input_filter(FilterGraph *fg, AVFilterInOut *in)
 253 {
 254     InputStream *ist = NULL;
 255     enum AVMediaType type = avfilter_pad_get_type(in->filter_ctx->input_pads, in->pad_idx);
 256     int i;
 257
 258     // TODO: support other filter types
 259     if (type != AVMEDIA_TYPE_VIDEO && type != AVMEDIA_TYPE_AUDIO) {
 260         av_log(NULL, AV_LOG_FATAL, "Only video and audio filters supported "
 261                "currently.\n");
 262         exit_program(1);
 263     }
 264
 265     if (in->name) {
 266         AVFormatContext *s;
 267         AVStream       *st = NULL;
 268         char *p;
 269         int file_idx = strtol(in->name, &p, 0);
 270
 271         if (file_idx < 0 || file_idx >= nb_input_files) {
 272             av_log(NULL, AV_LOG_FATAL, "Invalid file index %d in filtergraph description %s.\n",
 273                    file_idx, fg->graph_desc);
 274             exit_program(1);
 275         }
 276         s = input_files[file_idx]->ctx;
 277
 278         for (i = 0; i < s->nb_streams; i++) {
 279             enum AVMediaType stream_type = s->streams[i]->codecpar->codec_type;
 280             if (stream_type != type &&
 281                 !(stream_type == AVMEDIA_TYPE_SUBTITLE &&
 282                   type == AVMEDIA_TYPE_VIDEO /* sub2video hack */))
 283                 continue;
 284             if (check_stream_specifier(s, s->streams[i], *p == ':' ? p + 1 : p) == 1) {
 285                 st = s->streams[i];
 286                 break;
 287             }
 288         }
 289         if (!st) {
 290             av_log(NULL, AV_LOG_FATAL, "Stream specifier '%s' in filtergraph description %s "
 291                    "matches no streams.\n", p, fg->graph_desc);
 292             exit_program(1);
 293         }
 294         ist = input_streams[input_files[file_idx]->ist_index + st->index];
 295     } else {
 296         /* find the first unused stream of corresponding type */
 297         for (i = 0; i < nb_input_streams; i++) {
 298             ist = input_streams[i];
 299             if (ist->dec_ctx->codec_type == type && ist->discard)
 300                 break;
 301         }
 302         if (i == nb_input_streams) {
 303             av_log(NULL, AV_LOG_FATAL, "Cannot find a matching stream for "
 304                    "unlabeled input pad %d on filter %s\n", in->pad_idx,
 305                    in->filter_ctx->name);
 306             exit_program(1);
 307         }
 308     }
 309     av_assert0(ist);
 310
 311     ist->discard         = 0;
 312     ist->decoding_needed |= DECODING_FOR_FILTER;
 313     ist->st->discard = AVDISCARD_NONE;
 314
 315     GROW_ARRAY(fg->inputs, fg->nb_inputs);
 316     if (!(fg->inputs[fg->nb_inputs - 1] = av_mallocz(sizeof(*fg->inputs[0]))))
 317         exit_program(1);
 318     fg->inputs[fg->nb_inputs - 1]->ist   = ist;
 319     fg->inputs[fg->nb_inputs - 1]->graph = fg;
 320     fg->inputs[fg->nb_inputs - 1]->format = -1;
 321     fg->inputs[fg->nb_inputs - 1]->type = ist->st->codecpar->codec_type;
 322     fg->inputs[fg->nb_inputs - 1]->name = describe_filter_link(fg, in, 1);
 323
 324     fg->inputs[fg->nb_inputs - 1]->frame_queue = av_fifo_alloc(8 * sizeof(AVFrame*));
 325     if (!fg->inputs[fg->nb_inputs - 1]->frame_queue)
 326         exit_program(1);
 327
 328     GROW_ARRAY(ist->filters, ist->nb_filters);
 329     ist->filters[ist->nb_filters - 1] = fg->inputs[fg->nb_inputs - 1];
 330 }
 331
 332 int init_complex_filtergraph(FilterGraph *fg)
 333 {
 334     AVFilterInOut *inputs, *outputs, *cur;
 335     AVFilterGraph *graph;
 336     int ret = 0;
 337
 338     /* this graph is only used for determining the kinds of inputs
 339      * and outputs we have, and is discarded on exit from this function */
 340     graph = avfilter_graph_alloc();
 341     if (!graph)
 342         return AVERROR(ENOMEM);
 343
 344     ret = avfilter_graph_parse2(graph, fg->graph_desc, &inputs, &outputs);
 345     if (ret < 0)
 346         goto fail;
 347
 348     for (cur = inputs; cur; cur = cur->next)
 349         init_input_filter(fg, cur);
 350
 351     for (cur = outputs; cur;) {
 352         GROW_ARRAY(fg->outputs, fg->nb_outputs);
 353         fg->outputs[fg->nb_outputs - 1] = av_mallocz(sizeof(*fg->outputs[0]));
 354         if (!fg->outputs[fg->nb_outputs - 1])
 355             exit_program(1);
 356
 357         fg->outputs[fg->nb_outputs - 1]->graph   = fg;
 358         fg->outputs[fg->nb_outputs - 1]->out_tmp = cur;
 359         fg->outputs[fg->nb_outputs - 1]->type    = avfilter_pad_get_type(cur->filter_ctx->output_pads,
 360                                                                          cur->pad_idx);
 361         fg->outputs[fg->nb_outputs - 1]->name = describe_filter_link(fg, cur, 0);
 362         cur = cur->next;
 363         fg->outputs[fg->nb_outputs - 1]->out_tmp->next = NULL;
 364     }
 365
 366 fail:
 367     avfilter_inout_free(&inputs);
 368     avfilter_graph_free(&graph);
 369     return ret;
 370 }
 371
 372 static int insert_trim(int64_t start_time, int64_t duration,
 373                        AVFilterContext **last_filter, int *pad_idx,
 374                        const char *filter_name)
 375 {
 376     AVFilterGraph *graph = (*last_filter)->graph;
 377     AVFilterContext *ctx;
 378     const AVFilter *trim;
 379     enum AVMediaType type = avfilter_pad_get_type((*last_filter)->output_pads, *pad_idx);
 380     const char *name = (type == AVMEDIA_TYPE_VIDEO) ? "trim" : "atrim";
 381     int ret = 0;
 382
 383     if (duration == INT64_MAX && start_time == AV_NOPTS_VALUE)
 384         return 0;
 385
 386     trim = avfilter_get_by_name(name);
 387     if (!trim) {
 388         av_log(NULL, AV_LOG_ERROR, "%s filter not present, cannot limit "
 389                "recording time.\n", name);
 390         return AVERROR_FILTER_NOT_FOUND;
 391     }
 392
 393     ctx = avfilter_graph_alloc_filter(graph, trim, filter_name);
 394     if (!ctx)
 395         return AVERROR(ENOMEM);
 396
 397     if (duration != INT64_MAX) {
 398         ret = av_opt_set_int(ctx, "durationi", duration,
 399                                 AV_OPT_SEARCH_CHILDREN);
 400     }
 401     if (ret >= 0 && start_time != AV_NOPTS_VALUE) {
 402         ret = av_opt_set_int(ctx, "starti", start_time,
 403                                 AV_OPT_SEARCH_CHILDREN);
 404     }
 405     if (ret < 0) {
 406         av_log(ctx, AV_LOG_ERROR, "Error configuring the %s filter", name);
 407         return ret;
 408     }
 409
 410     ret = avfilter_init_str(ctx, NULL);
 411     if (ret < 0)
 412         return ret;
 413
 414     ret = avfilter_link(*last_filter, *pad_idx, ctx, 0);
 415     if (ret < 0)
 416         return ret;
 417
 418     *last_filter = ctx;
 419     *pad_idx     = 0;
 420     return 0;
 421 }
 422
 423 static int insert_filter(AVFilterContext **last_filter, int *pad_idx,
 424                          const char *filter_name, const char *args)
 425 {
 426     AVFilterGraph *graph = (*last_filter)->graph;
 427     AVFilterContext *ctx;
 428     int ret;
 429
 430     ret = avfilter_graph_create_filter(&ctx,
 431                                        avfilter_get_by_name(filter_name),
 432                                        filter_name, args, NULL, graph);
 433     if (ret < 0)
 434         return ret;
 435
 436     ret = avfilter_link(*last_filter, *pad_idx, ctx, 0);
 437     if (ret < 0)
 438         return ret;
 439
 440     *last_filter = ctx;
 441     *pad_idx     = 0;
 442     return 0;
 443 }
 444
 445 static int configure_output_video_filter(FilterGraph *fg, OutputFilter *ofilter, AVFilterInOut *out)
 446 {
 447     char *pix_fmts;
 448     OutputStream *ost = ofilter->ost;
 449     OutputFile    *of = output_files[ost->file_index];
 450     AVFilterContext *last_filter = out->filter_ctx;
 451     int pad_idx = out->pad_idx;
 452     int ret;
 453     char name[255];
 454
 455     snprintf(name, sizeof(name), "out_%d_%d", ost->file_index, ost->index);
 456     ret = avfilter_graph_create_filter(&ofilter->filter,
 457                                        avfilter_get_by_name("buffersink"),
 458                                        name, NULL, NULL, fg->graph);
 459
 460     if (ret < 0)
 461         return ret;
 462
 463     if (ofilter->width || ofilter->height) {
 464         char args[255];
 465         AVFilterContext *filter;
 466         AVDictionaryEntry *e = NULL;
 467
 468         snprintf(args, sizeof(args), "%d:%d",
 469                  ofilter->width, ofilter->height);
 470
 471         while ((e = av_dict_get(ost->sws_dict, "", e,
 472                                 AV_DICT_IGNORE_SUFFIX))) {
 473             av_strlcatf(args, sizeof(args), ":%s=%s", e->key, e->value);
 474         }
 475
 476         snprintf(name, sizeof(name), "scaler_out_%d_%d",
 477                  ost->file_index, ost->index);
 478         if ((ret = avfilter_graph_create_filter(&filter, avfilter_get_by_name("scale"),
 479                                                 name, args, NULL, fg->graph)) < 0)
 480             return ret;
 481         if ((ret = avfilter_link(last_filter, pad_idx, filter, 0)) < 0)
 482             return ret;
 483
 484         last_filter = filter;
 485         pad_idx = 0;
 486     }
 487
 488     if ((pix_fmts = choose_pix_fmts(ofilter))) {
 489         AVFilterContext *filter;
 490         snprintf(name, sizeof(name), "format_out_%d_%d",
 491                  ost->file_index, ost->index);
 492         ret = avfilter_graph_create_filter(&filter,
 493                                            avfilter_get_by_name("format"),
 494                                            "format", pix_fmts, NULL, fg->graph);
 495         av_freep(&pix_fmts);
 496         if (ret < 0)
 497             return ret;
 498         if ((ret = avfilter_link(last_filter, pad_idx, filter, 0)) < 0)
 499             return ret;
 500
 501         last_filter = filter;
 502         pad_idx     = 0;
 503     }
 504
 505     if (ost->frame_rate.num && 0) {
 506         AVFilterContext *fps;
 507         char args[255];
 508
 509         snprintf(args, sizeof(args), "fps=%d/%d", ost->frame_rate.num,
 510                  ost->frame_rate.den);
 511         snprintf(name, sizeof(name), "fps_out_%d_%d",
 512                  ost->file_index, ost->index);
 513         ret = avfilter_graph_create_filter(&fps, avfilter_get_by_name("fps"),
 514                                            name, args, NULL, fg->graph);
 515         if (ret < 0)
 516             return ret;
 517
 518         ret = avfilter_link(last_filter, pad_idx, fps, 0);
 519         if (ret < 0)
 520             return ret;
 521         last_filter = fps;
 522         pad_idx = 0;
 523     }
 524
 525     snprintf(name, sizeof(name), "trim_out_%d_%d",
 526              ost->file_index, ost->index);
 527     ret = insert_trim(of->start_time, of->recording_time,
 528                       &last_filter, &pad_idx, name);
 529     if (ret < 0)
 530         return ret;
 531
 532
 533     if ((ret = avfilter_link(last_filter, pad_idx, ofilter->filter, 0)) < 0)
 534         return ret;
 535
 536     return 0;
 537 }
 538
 539 static int configure_output_audio_filter(FilterGraph *fg, OutputFilter *ofilter, AVFilterInOut *out)
 540 {
 541     OutputStream *ost = ofilter->ost;
 542     OutputFile    *of = output_files[ost->file_index];
 543     AVCodecContext *codec  = ost->enc_ctx;
 544     AVFilterContext *last_filter = out->filter_ctx;
 545     int pad_idx = out->pad_idx;
 546     char *sample_fmts, *sample_rates, *channel_layouts;
 547     char name[255];
 548     int ret;
 549
 550     snprintf(name, sizeof(name), "out_%d_%d", ost->file_index, ost->index);
 551     ret = avfilter_graph_create_filter(&ofilter->filter,
 552                                        avfilter_get_by_name("abuffersink"),
 553                                        name, NULL, NULL, fg->graph);
 554     if (ret < 0)
 555         return ret;
 556     if ((ret = av_opt_set_int(ofilter->filter, "all_channel_counts", 1, AV_OPT_SEARCH_CHILDREN)) < 0)
 557         return ret;
 558
 559 #define AUTO_INSERT_FILTER(opt_name, filter_name, arg) do {                 \
 560     AVFilterContext *filt_ctx;                                              \
 561                                                                             \
 562     av_log(NULL, AV_LOG_INFO, opt_name " is forwarded to lavfi "            \
 563            "similarly to -af " filter_name "=%s.\n", arg);                  \
 564                                                                             \
 565     ret = avfilter_graph_create_filter(&filt_ctx,                           \
 566                                        avfilter_get_by_name(filter_name),   \
 567                                        filter_name, arg, NULL, fg->graph);  \
 568     if (ret < 0)                                                            \
 569         return ret;                                                         \
 570                                                                             \
 571     ret = avfilter_link(last_filter, pad_idx, filt_ctx, 0);                 \
 572     if (ret < 0)                                                            \
 573         return ret;                                                         \
 574                                                                             \
 575     last_filter = filt_ctx;                                                 \
 576     pad_idx = 0;                                                            \
 577 } while (0)
 578     if (ost->audio_channels_mapped) {
 579         int i;
 580         AVBPrint pan_buf;
 581         av_bprint_init(&pan_buf, 256, 8192);
 582         av_bprintf(&pan_buf, "0x%"PRIx64,
 583                    av_get_default_channel_layout(ost->audio_channels_mapped));
 584         for (i = 0; i < ost->audio_channels_mapped; i++)
 585             if (ost->audio_channels_map[i] != -1)
 586                 av_bprintf(&pan_buf, "|c%d=c%d", i, ost->audio_channels_map[i]);
 587
 588         AUTO_INSERT_FILTER("-map_channel", "pan", pan_buf.str);
 589         av_bprint_finalize(&pan_buf, NULL);
 590     }
 591
 592     if (codec->channels && !codec->channel_layout)
 593         codec->channel_layout = av_get_default_channel_layout(codec->channels);
 594
 595     sample_fmts     = choose_sample_fmts(ofilter);
 596     sample_rates    = choose_sample_rates(ofilter);
 597     channel_layouts = choose_channel_layouts(ofilter);
 598     if (sample_fmts || sample_rates || channel_layouts) {
 599         AVFilterContext *format;
 600         char args[256];
 601         args[0] = 0;
 602
 603         if (sample_fmts)
 604             av_strlcatf(args, sizeof(args), "sample_fmts=%s:",
 605                             sample_fmts);
 606         if (sample_rates)
 607             av_strlcatf(args, sizeof(args), "sample_rates=%s:",
 608                             sample_rates);
 609         if (channel_layouts)
 610             av_strlcatf(args, sizeof(args), "channel_layouts=%s:",
 611                             channel_layouts);
 612
 613         av_freep(&sample_fmts);
 614         av_freep(&sample_rates);
 615         av_freep(&channel_layouts);
 616
 617         snprintf(name, sizeof(name), "format_out_%d_%d",
 618                  ost->file_index, ost->index);
 619         ret = avfilter_graph_create_filter(&format,
 620                                            avfilter_get_by_name("aformat"),
 621                                            name, args, NULL, fg->graph);
 622         if (ret < 0)
 623             return ret;
 624
 625         ret = avfilter_link(last_filter, pad_idx, format, 0);
 626         if (ret < 0)
 627             return ret;
 628
 629         last_filter = format;
 630         pad_idx = 0;
 631     }
 632
 633     if (audio_volume != 256 && 0) {
 634         char args[256];
 635
 636         snprintf(args, sizeof(args), "%f", audio_volume / 256.);
 637         AUTO_INSERT_FILTER("-vol", "volume", args);
 638     }
 639
 640     if (ost->apad && of->shortest) {
 641         char args[256];
 642         int i;
 643
 644         for (i=0; i<of->ctx->nb_streams; i++)
 645             if (of->ctx->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO)
 646                 break;
 647
 648         if (i<of->ctx->nb_streams) {
 649             snprintf(args, sizeof(args), "%s", ost->apad);
 650             AUTO_INSERT_FILTER("-apad", "apad", args);
 651         }
 652     }
 653
 654     snprintf(name, sizeof(name), "trim for output stream %d:%d",
 655              ost->file_index, ost->index);
 656     ret = insert_trim(of->start_time, of->recording_time,
 657                       &last_filter, &pad_idx, name);
 658     if (ret < 0)
 659         return ret;
 660
 661     if ((ret = avfilter_link(last_filter, pad_idx, ofilter->filter, 0)) < 0)
 662         return ret;
 663
 664     return 0;
 665 }
 666
 667 int configure_output_filter(FilterGraph *fg, OutputFilter *ofilter, AVFilterInOut *out)
 668 {
 669     if (!ofilter->ost) {
 670         av_log(NULL, AV_LOG_FATAL, "Filter %s has an unconnected output\n", ofilter->name);
 671         exit_program(1);
 672     }
 673
 674     switch (avfilter_pad_get_type(out->filter_ctx->output_pads, out->pad_idx)) {
 675     case AVMEDIA_TYPE_VIDEO: return configure_output_video_filter(fg, ofilter, out);
 676     case AVMEDIA_TYPE_AUDIO: return configure_output_audio_filter(fg, ofilter, out);
 677     default: av_assert0(0);
 678     }
 679 }
 680
 681 static int sub2video_prepare(InputStream *ist, InputFilter *ifilter)
 682 {
 683     AVFormatContext *avf = input_files[ist->file_index]->ctx;
 684     int i, w, h;
 685
 686     /* Compute the size of the canvas for the subtitles stream.
 687        If the subtitles codecpar has set a size, use it. Otherwise use the
 688        maximum dimensions of the video streams in the same file. */
 689     w = ifilter->width;
 690     h = ifilter->height;
 691     if (!(w && h)) {
 692         for (i = 0; i < avf->nb_streams; i++) {
 693             if (avf->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) {
 694                 w = FFMAX(w, avf->streams[i]->codecpar->width);
 695                 h = FFMAX(h, avf->streams[i]->codecpar->height);
 696             }
 697         }
 698         if (!(w && h)) {
 699             w = FFMAX(w, 720);
 700             h = FFMAX(h, 576);
 701         }
 702         av_log(avf, AV_LOG_INFO, "sub2video: using %dx%d canvas\n", w, h);
 703     }
 704     ist->sub2video.w = ifilter->width  = w;
 705     ist->sub2video.h = ifilter->height = h;
 706
 707     ifilter->width  = ist->dec_ctx->width  ? ist->dec_ctx->width  : ist->sub2video.w;
 708     ifilter->height = ist->dec_ctx->height ? ist->dec_ctx->height : ist->sub2video.h;
 709
 710     /* rectangles are AV_PIX_FMT_PAL8, but we have no guarantee that the
 711        palettes for all rectangles are identical or compatible */
 712     ifilter->format = AV_PIX_FMT_RGB32;
 713
 714     ist->sub2video.frame = av_frame_alloc();
 715     if (!ist->sub2video.frame)
 716         return AVERROR(ENOMEM);
 717     ist->sub2video.last_pts = INT64_MIN;
 718     return 0;
 719 }
 720
 721 static int configure_input_video_filter(FilterGraph *fg, InputFilter *ifilter,
 722                                         AVFilterInOut *in)
 723 {
 724     AVFilterContext *last_filter;
 725     const AVFilter *buffer_filt = avfilter_get_by_name("buffer");
 726     InputStream *ist = ifilter->ist;
 727     InputFile     *f = input_files[ist->file_index];
 728     AVRational tb = ist->framerate.num ? av_inv_q(ist->framerate) :
 729                                          ist->st->time_base;
 730     AVRational fr = ist->framerate;
 731     AVRational sar;
 732     AVBPrint args;
 733     char name[255];
 734     int ret, pad_idx = 0;
 735     int64_t tsoffset = 0;
 736     AVBufferSrcParameters *par = av_buffersrc_parameters_alloc();
 737
 738     if (!par)
 739         return AVERROR(ENOMEM);
 740     memset(par, 0, sizeof(*par));
 741     par->format = AV_PIX_FMT_NONE;
 742
 743     if (ist->dec_ctx->codec_type == AVMEDIA_TYPE_AUDIO) {
 744         av_log(NULL, AV_LOG_ERROR, "Cannot connect video filter to audio input\n");
 745         ret = AVERROR(EINVAL);
 746         goto fail;
 747     }
 748
 749     if (!fr.num)
 750         fr = av_guess_frame_rate(input_files[ist->file_index]->ctx, ist->st, NULL);
 751
 752     if (ist->dec_ctx->codec_type == AVMEDIA_TYPE_SUBTITLE) {
 753         ret = sub2video_prepare(ist, ifilter);
 754         if (ret < 0)
 755             goto fail;
 756     }
 757
 758     sar = ifilter->sample_aspect_ratio;
 759     if(!sar.den)
 760         sar = (AVRational){0,1};
 761     av_bprint_init(&args, 0, 1);
 762     av_bprintf(&args,
 763              "video_size=%dx%d:pix_fmt=%d:time_base=%d/%d:"
 764              "pixel_aspect=%d/%d:sws_param=flags=%d",
 765              ifilter->width, ifilter->height, ifilter->format,
 766              tb.num, tb.den, sar.num, sar.den,
 767              SWS_BILINEAR + ((ist->dec_ctx->flags&AV_CODEC_FLAG_BITEXACT) ? SWS_BITEXACT:0));
 768     if (fr.num && fr.den)
 769         av_bprintf(&args, ":frame_rate=%d/%d", fr.num, fr.den);
 770     snprintf(name, sizeof(name), "graph %d input from stream %d:%d", fg->index,
 771              ist->file_index, ist->st->index);
 772
 773
 774     if ((ret = avfilter_graph_create_filter(&ifilter->filter, buffer_filt, name,
 775                                             args.str, NULL, fg->graph)) < 0)
 776         goto fail;
 777     par->hw_frames_ctx = ifilter->hw_frames_ctx;
 778     ret = av_buffersrc_parameters_set(ifilter->filter, par);
 779     if (ret < 0)
 780         goto fail;
 781     av_freep(&par);
 782     last_filter = ifilter->filter;
 783
 784     if (ist->autorotate) {
 785         double theta = get_rotation(ist->st);
 786
 787         if (fabs(theta - 90) < 1.0) {
 788             ret = insert_filter(&last_filter, &pad_idx, "transpose", "clock");
 789         } else if (fabs(theta - 180) < 1.0) {
 790             ret = insert_filter(&last_filter, &pad_idx, "hflip", NULL);
 791             if (ret < 0)
 792                 return ret;
 793             ret = insert_filter(&last_filter, &pad_idx, "vflip", NULL);
 794         } else if (fabs(theta - 270) < 1.0) {
 795             ret = insert_filter(&last_filter, &pad_idx, "transpose", "cclock");
 796         } else if (fabs(theta) > 1.0) {
 797             char rotate_buf[64];
 798             snprintf(rotate_buf, sizeof(rotate_buf), "%f*PI/180", theta);
 799             ret = insert_filter(&last_filter, &pad_idx, "rotate", rotate_buf);
 800         }
 801         if (ret < 0)
 802             return ret;
 803     }
 804
 805     if (do_deinterlace) {
 806         AVFilterContext *yadif;
 807
 808         snprintf(name, sizeof(name), "deinterlace_in_%d_%d",
 809                  ist->file_index, ist->st->index);
 810         if ((ret = avfilter_graph_create_filter(&yadif,
 811                                                 avfilter_get_by_name("yadif"),
 812                                                 name, "", NULL,
 813                                                 fg->graph)) < 0)
 814             return ret;
 815
 816         if ((ret = avfilter_link(last_filter, 0, yadif, 0)) < 0)
 817             return ret;
 818
 819         last_filter = yadif;
 820     }
 821
 822     snprintf(name, sizeof(name), "trim_in_%d_%d",
 823              ist->file_index, ist->st->index);
 824     if (copy_ts) {
 825         tsoffset = f->start_time == AV_NOPTS_VALUE ? 0 : f->start_time;
 826         if (!start_at_zero && f->ctx->start_time != AV_NOPTS_VALUE)
 827             tsoffset += f->ctx->start_time;
 828     }
 829     ret = insert_trim(((f->start_time == AV_NOPTS_VALUE) || !f->accurate_seek) ?
 830                       AV_NOPTS_VALUE : tsoffset, f->recording_time,
 831                       &last_filter, &pad_idx, name);
 832     if (ret < 0)
 833         return ret;
 834
 835     if ((ret = avfilter_link(last_filter, 0, in->filter_ctx, in->pad_idx)) < 0)
 836         return ret;
 837     return 0;
 838 fail:
 839     av_freep(&par);
 840
 841     return ret;
 842 }
 843
 844 static int configure_input_audio_filter(FilterGraph *fg, InputFilter *ifilter,
 845                                         AVFilterInOut *in)
 846 {
 847     AVFilterContext *last_filter;
 848     const AVFilter *abuffer_filt = avfilter_get_by_name("abuffer");
 849     InputStream *ist = ifilter->ist;
 850     InputFile     *f = input_files[ist->file_index];
 851     AVBPrint args;
 852     char name[255];
 853     int ret, pad_idx = 0;
 854     int64_t tsoffset = 0;
 855
 856     if (ist->dec_ctx->codec_type != AVMEDIA_TYPE_AUDIO) {
 857         av_log(NULL, AV_LOG_ERROR, "Cannot connect audio filter to non audio input\n");
 858         return AVERROR(EINVAL);
 859     }
 860
 861     av_bprint_init(&args, 0, AV_BPRINT_SIZE_AUTOMATIC);
 862     av_bprintf(&args, "time_base=%d/%d:sample_rate=%d:sample_fmt=%s",
 863              1, ifilter->sample_rate,
 864              ifilter->sample_rate,
 865              av_get_sample_fmt_name(ifilter->format));
 866     if (ifilter->channel_layout)
 867         av_bprintf(&args, ":channel_layout=0x%"PRIx64,
 868                    ifilter->channel_layout);
 869     else
 870         av_bprintf(&args, ":channels=%d", ifilter->channels);
 871     snprintf(name, sizeof(name), "graph_%d_in_%d_%d", fg->index,
 872              ist->file_index, ist->st->index);
 873
 874     if ((ret = avfilter_graph_create_filter(&ifilter->filter, abuffer_filt,
 875                                             name, args.str, NULL,
 876                                             fg->graph)) < 0)
 877         return ret;
 878     last_filter = ifilter->filter;
 879
 880 #define AUTO_INSERT_FILTER_INPUT(opt_name, filter_name, arg) do {                 \
 881     AVFilterContext *filt_ctx;                                              \
 882                                                                             \
 883     av_log(NULL, AV_LOG_INFO, opt_name " is forwarded to lavfi "            \
 884            "similarly to -af " filter_name "=%s.\n", arg);                  \
 885                                                                             \
 886     snprintf(name, sizeof(name), "graph_%d_%s_in_%d_%d",      \
 887                 fg->index, filter_name, ist->file_index, ist->st->index);   \
 888     ret = avfilter_graph_create_filter(&filt_ctx,                           \
 889                                        avfilter_get_by_name(filter_name),   \
 890                                        name, arg, NULL, fg->graph);         \
 891     if (ret < 0)                                                            \
 892         return ret;                                                         \
 893                                                                             \
 894     ret = avfilter_link(last_filter, 0, filt_ctx, 0);                       \
 895     if (ret < 0)                                                            \
 896         return ret;                                                         \
 897                                                                             \
 898     last_filter = filt_ctx;                                                 \
 899 } while (0)
 900
 901     if (audio_sync_method > 0) {
 902         char args[256] = {0};
 903
 904         av_strlcatf(args, sizeof(args), "async=%d", audio_sync_method);
 905         if (audio_drift_threshold != 0.1)
 906             av_strlcatf(args, sizeof(args), ":min_hard_comp=%f", audio_drift_threshold);
 907         if (!fg->reconfiguration)
 908             av_strlcatf(args, sizeof(args), ":first_pts=0");
 909         AUTO_INSERT_FILTER_INPUT("-async", "aresample", args);
 910     }
 911
 912 //     if (ost->audio_channels_mapped) {
 913 //         int i;
 914 //         AVBPrint pan_buf;
 915 //         av_bprint_init(&pan_buf, 256, 8192);
 916 //         av_bprintf(&pan_buf, "0x%"PRIx64,
 917 //                    av_get_default_channel_layout(ost->audio_channels_mapped));
 918 //         for (i = 0; i < ost->audio_channels_mapped; i++)
 919 //             if (ost->audio_channels_map[i] != -1)
 920 //                 av_bprintf(&pan_buf, ":c%d=c%d", i, ost->audio_channels_map[i]);
 921 //         AUTO_INSERT_FILTER_INPUT("-map_channel", "pan", pan_buf.str);
 922 //         av_bprint_finalize(&pan_buf, NULL);
 923 //     }
 924
 925     if (audio_volume != 256) {
 926         char args[256];
 927
 928         av_log(NULL, AV_LOG_WARNING, "-vol has been deprecated. Use the volume "
 929                "audio filter instead.\n");
 930
 931         snprintf(args, sizeof(args), "%f", audio_volume / 256.);
 932         AUTO_INSERT_FILTER_INPUT("-vol", "volume", args);
 933     }
 934
 935     snprintf(name, sizeof(name), "trim for input stream %d:%d",
 936              ist->file_index, ist->st->index);
 937     if (copy_ts) {
 938         tsoffset = f->start_time == AV_NOPTS_VALUE ? 0 : f->start_time;
 939         if (!start_at_zero && f->ctx->start_time != AV_NOPTS_VALUE)
 940             tsoffset += f->ctx->start_time;
 941     }
 942     ret = insert_trim(((f->start_time == AV_NOPTS_VALUE) || !f->accurate_seek) ?
 943                       AV_NOPTS_VALUE : tsoffset, f->recording_time,
 944                       &last_filter, &pad_idx, name);
 945     if (ret < 0)
 946         return ret;
 947
 948     if ((ret = avfilter_link(last_filter, 0, in->filter_ctx, in->pad_idx)) < 0)
 949         return ret;
 950
 951     return 0;
 952 }
 953
 954 static int configure_input_filter(FilterGraph *fg, InputFilter *ifilter,
 955                                   AVFilterInOut *in)
 956 {
 957     if (!ifilter->ist->dec) {
 958         av_log(NULL, AV_LOG_ERROR,
 959                "No decoder for stream #%d:%d, filtering impossible\n",
 960                ifilter->ist->file_index, ifilter->ist->st->index);
 961         return AVERROR_DECODER_NOT_FOUND;
 962     }
 963     switch (avfilter_pad_get_type(in->filter_ctx->input_pads, in->pad_idx)) {
 964     case AVMEDIA_TYPE_VIDEO: return configure_input_video_filter(fg, ifilter, in);
 965     case AVMEDIA_TYPE_AUDIO: return configure_input_audio_filter(fg, ifilter, in);
 966     default: av_assert0(0);
 967     }
 968 }
 969
 970 static void cleanup_filtergraph(FilterGraph *fg)
 971 {
 972     int i;
 973     for (i = 0; i < fg->nb_outputs; i++)
 974         fg->outputs[i]->filter = (AVFilterContext *)NULL;
 975     for (i = 0; i < fg->nb_inputs; i++)
 976         fg->inputs[i]->filter = (AVFilterContext *)NULL;
 977     avfilter_graph_free(&fg->graph);
 978 }
 979
 980 int configure_filtergraph(FilterGraph *fg)
 981 {
 982     AVFilterInOut *inputs, *outputs, *cur;
 983     int ret, i, simple = filtergraph_is_simple(fg);
 984     const char *graph_desc = simple ? fg->outputs[0]->ost->avfilter :
 985                                       fg->graph_desc;
 986
 987     cleanup_filtergraph(fg);
 988     if (!(fg->graph = avfilter_graph_alloc()))
 989         return AVERROR(ENOMEM);
 990
 991     if (simple) {
 992         OutputStream *ost = fg->outputs[0]->ost;
 993         char args[512];
 994         AVDictionaryEntry *e = NULL;
 995
 996         fg->graph->nb_threads = filter_nbthreads;
 997
 998         args[0] = 0;
 999         while ((e = av_dict_get(ost->sws_dict, "", e,
1000                                 AV_DICT_IGNORE_SUFFIX))) {
1001             av_strlcatf(args, sizeof(args), "%s=%s:", e->key, e->value);
1002         }
1003         if (strlen(args))
1004             args[strlen(args)-1] = 0;
1005         fg->graph->scale_sws_opts = av_strdup(args);
1006
1007         args[0] = 0;
1008         while ((e = av_dict_get(ost->swr_opts, "", e,
1009                                 AV_DICT_IGNORE_SUFFIX))) {
1010             av_strlcatf(args, sizeof(args), "%s=%s:", e->key, e->value);
1011         }
1012         if (strlen(args))
1013             args[strlen(args)-1] = 0;
1014         av_opt_set(fg->graph, "aresample_swr_opts", args, 0);
1015
1016         args[0] = '\0';
1017         while ((e = av_dict_get(fg->outputs[0]->ost->resample_opts, "", e,
1018                                 AV_DICT_IGNORE_SUFFIX))) {
1019             av_strlcatf(args, sizeof(args), "%s=%s:", e->key, e->value);
1020         }
1021         if (strlen(args))
1022             args[strlen(args) - 1] = '\0';
1023
1024         e = av_dict_get(ost->encoder_opts, "threads", NULL, 0);
1025         if (e)
1026             av_opt_set(fg->graph, "threads", e->value, 0);
1027     } else {
1028         fg->graph->nb_threads = filter_complex_nbthreads;
1029     }
1030
1031     if ((ret = avfilter_graph_parse2(fg->graph, graph_desc, &inputs, &outputs)) < 0)
1032         goto fail;
1033
1034     if (hw_device_ctx) {
1035         for (i = 0; i < fg->graph->nb_filters; i++) {
1036             fg->graph->filters[i]->hw_device_ctx = av_buffer_ref(hw_device_ctx);
1037         }
1038     }
1039
1040     if (simple && (!inputs || inputs->next || !outputs || outputs->next)) {
1041         const char *num_inputs;
1042         const char *num_outputs;
1043         if (!outputs) {
1044             num_outputs = "0";
1045         } else if (outputs->next) {
1046             num_outputs = ">1";
1047         } else {
1048             num_outputs = "1";
1049         }
1050         if (!inputs) {
1051             num_inputs = "0";
1052         } else if (inputs->next) {
1053             num_inputs = ">1";
1054         } else {
1055             num_inputs = "1";
1056         }
1057         av_log(NULL, AV_LOG_ERROR, "Simple filtergraph '%s' was expected "
1058                "to have exactly 1 input and 1 output."
1059                " However, it had %s input(s) and %s output(s)."
1060                " Please adjust, or use a complex filtergraph (-filter_complex) instead.\n",
1061                graph_desc, num_inputs, num_outputs);
1062         ret = AVERROR(EINVAL);
1063         goto fail;
1064     }
1065
1066     for (cur = inputs, i = 0; cur; cur = cur->next, i++)
1067         if ((ret = configure_input_filter(fg, fg->inputs[i], cur)) < 0) {
1068             avfilter_inout_free(&inputs);
1069             avfilter_inout_free(&outputs);
1070             goto fail;
1071         }
1072     avfilter_inout_free(&inputs);
1073
1074     for (cur = outputs, i = 0; cur; cur = cur->next, i++)
1075         configure_output_filter(fg, fg->outputs[i], cur);
1076     avfilter_inout_free(&outputs);
1077
1078     if ((ret = avfilter_graph_config(fg->graph, NULL)) < 0)
1079         goto fail;
1080
1081     /* limit the lists of allowed formats to the ones selected, to
1082      * make sure they stay the same if the filtergraph is reconfigured later */
1083     for (i = 0; i < fg->nb_outputs; i++) {
1084         OutputFilter *ofilter = fg->outputs[i];
1085         AVFilterContext *sink = ofilter->filter;
1086
1087         ofilter->format = av_buffersink_get_format(sink);
1088
1089         ofilter->width  = av_buffersink_get_w(sink);
1090         ofilter->height = av_buffersink_get_h(sink);
1091
1092         ofilter->sample_rate    = av_buffersink_get_sample_rate(sink);
1093         ofilter->channel_layout = av_buffersink_get_channel_layout(sink);
1094     }
1095
1096     fg->reconfiguration = 1;
1097
1098     for (i = 0; i < fg->nb_outputs; i++) {
1099         OutputStream *ost = fg->outputs[i]->ost;
1100         if (!ost->enc) {
1101             /* identical to the same check in ffmpeg.c, needed because
1102                complex filter graphs are initialized earlier */
1103             av_log(NULL, AV_LOG_ERROR, "Encoder (codec %s) not found for output stream #%d:%d\n",
1104                      avcodec_get_name(ost->st->codecpar->codec_id), ost->file_index, ost->index);
1105             ret = AVERROR(EINVAL);
1106             goto fail;
1107         }
1108         if (ost->enc->type == AVMEDIA_TYPE_AUDIO &&
1109             !(ost->enc->capabilities & AV_CODEC_CAP_VARIABLE_FRAME_SIZE))
1110             av_buffersink_set_frame_size(ost->filter->filter,
1111                                          ost->enc_ctx->frame_size);
1112     }
1113
1114     for (i = 0; i < fg->nb_inputs; i++) {
1115         while (av_fifo_size(fg->inputs[i]->frame_queue)) {
1116             AVFrame *tmp;
1117             av_fifo_generic_read(fg->inputs[i]->frame_queue, &tmp, sizeof(tmp), NULL);
1118             ret = av_buffersrc_add_frame(fg->inputs[i]->filter, tmp);
1119             av_frame_free(&tmp);
1120             if (ret < 0)
1121                 goto fail;
1122         }
1123     }
1124
1125     /* send the EOFs for the finished inputs */
1126     for (i = 0; i < fg->nb_inputs; i++) {
1127         if (fg->inputs[i]->eof) {
1128             ret = av_buffersrc_add_frame(fg->inputs[i]->filter, NULL);
1129             if (ret < 0)
1130                 goto fail;
1131         }
1132     }
1133
1134     /* process queued up subtitle packets */
1135     for (i = 0; i < fg->nb_inputs; i++) {
1136         InputStream *ist = fg->inputs[i]->ist;
1137         if (ist->sub2video.sub_queue && ist->sub2video.frame) {
1138             while (av_fifo_size(ist->sub2video.sub_queue)) {
1139                 AVSubtitle tmp;
1140                 av_fifo_generic_read(ist->sub2video.sub_queue, &tmp, sizeof(tmp), NULL);
1141                 sub2video_update(ist, &tmp);
1142                 avsubtitle_free(&tmp);
1143             }
1144         }
1145     }
1146
1147     return 0;
1148
1149 fail:
1150     cleanup_filtergraph(fg);
1151     return ret;
1152 }
1153
1154 int ifilter_parameters_from_frame(InputFilter *ifilter, const AVFrame *frame)
1155 {
1156     av_buffer_unref(&ifilter->hw_frames_ctx);
1157
1158     ifilter->format = frame->format;
1159
1160     ifilter->width               = frame->width;
1161     ifilter->height              = frame->height;
1162     ifilter->sample_aspect_ratio = frame->sample_aspect_ratio;
1163
1164     ifilter->sample_rate         = frame->sample_rate;
1165     ifilter->channels            = frame->channels;
1166     ifilter->channel_layout      = frame->channel_layout;
1167
1168     if (frame->hw_frames_ctx) {
1169         ifilter->hw_frames_ctx = av_buffer_ref(frame->hw_frames_ctx);
1170         if (!ifilter->hw_frames_ctx)
1171             return AVERROR(ENOMEM);
1172     }
1173
1174     return 0;
1175 }
1176
1177 int ist_in_filtergraph(FilterGraph *fg, InputStream *ist)
1178 {
1179     int i;
1180     for (i = 0; i < fg->nb_inputs; i++)
1181         if (fg->inputs[i]->ist == ist)
1182             return 1;
1183     return 0;
1184 }
1185
1186 int filtergraph_is_simple(FilterGraph *fg)
1187 {
1188     return !fg->graph_desc;
1189 }