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