Merge remote-tracking branch 'qatar/master'
[ffmpeg.git] / ffmpeg.c
1 /*
2  * Copyright (c) 2000-2003 Fabrice Bellard
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 /**
22  * @file
23  * multimedia converter based on the FFmpeg libraries
24  */
25
26 #include "config.h"
27 #include <ctype.h>
28 #include <string.h>
29 #include <math.h>
30 #include <stdlib.h>
31 #include <errno.h>
32 #include <signal.h>
33 #include <limits.h>
34 #include <unistd.h>
35 #include "libavformat/avformat.h"
36 #include "libavdevice/avdevice.h"
37 #include "libswscale/swscale.h"
38 #include "libswresample/swresample.h"
39 #include "libavutil/opt.h"
40 #include "libavutil/audioconvert.h"
41 #include "libavutil/parseutils.h"
42 #include "libavutil/samplefmt.h"
43 #include "libavutil/colorspace.h"
44 #include "libavutil/fifo.h"
45 #include "libavutil/intreadwrite.h"
46 #include "libavutil/dict.h"
47 #include "libavutil/mathematics.h"
48 #include "libavutil/pixdesc.h"
49 #include "libavutil/avstring.h"
50 #include "libavutil/libm.h"
51 #include "libavutil/imgutils.h"
52 #include "libavutil/timestamp.h"
53 #include "libavutil/bprint.h"
54 #include "libavformat/os_support.h"
55
56 #include "libavformat/ffm.h" // not public API
57
58 # include "libavfilter/avcodec.h"
59 # include "libavfilter/avfilter.h"
60 # include "libavfilter/avfiltergraph.h"
61 # include "libavfilter/buffersrc.h"
62 # include "libavfilter/buffersink.h"
63
64 #if HAVE_SYS_RESOURCE_H
65 #include <sys/types.h>
66 #include <sys/time.h>
67 #include <sys/resource.h>
68 #elif HAVE_GETPROCESSTIMES
69 #include <windows.h>
70 #endif
71 #if HAVE_GETPROCESSMEMORYINFO
72 #include <windows.h>
73 #include <psapi.h>
74 #endif
75
76 #if HAVE_SYS_SELECT_H
77 #include <sys/select.h>
78 #endif
79
80 #if HAVE_TERMIOS_H
81 #include <fcntl.h>
82 #include <sys/ioctl.h>
83 #include <sys/time.h>
84 #include <termios.h>
85 #elif HAVE_KBHIT
86 #include <conio.h>
87 #endif
88 #include <time.h>
89
90 #include "cmdutils.h"
91
92 #include "libavutil/avassert.h"
93
94 #define VSYNC_AUTO       -1
95 #define VSYNC_PASSTHROUGH 0
96 #define VSYNC_CFR         1
97 #define VSYNC_VFR         2
98 #define VSYNC_DROP        0xff
99
100 #define SINKA
101
102 const char program_name[] = "ffmpeg";
103 const int program_birth_year = 2000;
104
105 /* select an input stream for an output stream */
106 typedef struct StreamMap {
107     int disabled;           /** 1 is this mapping is disabled by a negative map */
108     int file_index;
109     int stream_index;
110     int sync_file_index;
111     int sync_stream_index;
112     char *linklabel;       /** name of an output link, for mapping lavfi outputs */
113 } StreamMap;
114
115 typedef struct {
116     int  file_idx,  stream_idx,  channel_idx; // input
117     int ofile_idx, ostream_idx;               // output
118 } AudioChannelMap;
119
120 static const OptionDef options[];
121
122 #define MAX_STREAMS 1024    /* arbitrary sanity check value */
123
124 static int frame_bits_per_raw_sample = 0;
125 static int video_discard = 0;
126 static int same_quant = 0;
127 static int do_deinterlace = 0;
128 static int intra_dc_precision = 8;
129 static int qp_hist = 0;
130 static int intra_only = 0;
131 static const char *video_codec_name    = NULL;
132 static const char *audio_codec_name    = NULL;
133 static const char *subtitle_codec_name = NULL;
134
135 static int file_overwrite = 0;
136 static int no_file_overwrite = 0;
137 static int do_benchmark = 0;
138 static int do_benchmark_all = 0;
139 static int do_hex_dump = 0;
140 static int do_pkt_dump = 0;
141 static int do_psnr = 0;
142 static int do_pass = 0;
143 static const char *pass_logfilename_prefix;
144 static int video_sync_method = VSYNC_AUTO;
145 static int audio_sync_method = 0;
146 static float audio_drift_threshold = 0.1;
147 static int copy_ts = 0;
148 static int copy_tb = -1;
149 static int opt_shortest = 0;
150 static char *vstats_filename;
151 static FILE *vstats_file;
152
153 static int audio_volume = 256;
154
155 static int exit_on_error = 0;
156 static int using_stdin = 0;
157 static int run_as_daemon  = 0;
158 static volatile int received_nb_signals = 0;
159 static int64_t video_size = 0;
160 static int64_t audio_size = 0;
161 static int64_t extra_size = 0;
162 static int nb_frames_dup = 0;
163 static int nb_frames_drop = 0;
164 static int input_sync;
165
166 static float dts_delta_threshold = 10;
167 static float dts_error_threshold = 3600*30;
168
169 static int print_stats = 1;
170 static int debug_ts = 0;
171 static int current_time;
172
173 #define DEFAULT_PASS_LOGFILENAME_PREFIX "ffmpeg2pass"
174
175 typedef struct InputFilter {
176     AVFilterContext    *filter;
177     struct InputStream *ist;
178     struct FilterGraph *graph;
179 } InputFilter;
180
181 typedef struct OutputFilter {
182     AVFilterContext     *filter;
183     struct OutputStream *ost;
184     struct FilterGraph  *graph;
185
186     /* temporary storage until stream maps are processed */
187     AVFilterInOut       *out_tmp;
188 } OutputFilter;
189
190 typedef struct FilterGraph {
191     int            index;
192     const char    *graph_desc;
193
194     AVFilterGraph *graph;
195
196     InputFilter   **inputs;
197     int          nb_inputs;
198     OutputFilter **outputs;
199     int         nb_outputs;
200 } FilterGraph;
201
202 typedef struct FrameBuffer {
203     uint8_t *base[4];
204     uint8_t *data[4];
205     int  linesize[4];
206
207     int h, w;
208     enum PixelFormat pix_fmt;
209
210     int refcount;
211     struct InputStream *ist;
212     struct FrameBuffer *next;
213 } FrameBuffer;
214
215 typedef struct InputStream {
216     int file_index;
217     AVStream *st;
218     int discard;             /* true if stream data should be discarded */
219     int decoding_needed;     /* true if the packets must be decoded in 'raw_fifo' */
220     AVCodec *dec;
221     AVFrame *decoded_frame;
222
223     int64_t       start;     /* time when read started */
224     /* predicted dts of the next packet read for this stream or (when there are
225      * several frames in a packet) of the next frame in current packet */
226     int64_t       next_dts;
227     /* dts of the last packet read for this stream */
228     int64_t       dts;
229
230     int64_t       next_pts;  /* synthetic pts for the next decode frame */
231     int64_t       pts;       /* current pts of the decoded frame */
232     double ts_scale;
233     int is_start;            /* is 1 at the start and after a discontinuity */
234     int saw_first_ts;
235     int showed_multi_packet_warning;
236     AVDictionary *opts;
237
238     int resample_height;
239     int resample_width;
240     int resample_pix_fmt;
241
242     int      resample_sample_fmt;
243     int      resample_sample_rate;
244     int      resample_channels;
245     uint64_t resample_channel_layout;
246
247     /* a pool of free buffers for decoded data */
248     FrameBuffer *buffer_pool;
249     int dr1;
250
251     /* decoded data from this stream goes into all those filters
252      * currently video and audio only */
253     InputFilter **filters;
254     int        nb_filters;
255 } InputStream;
256
257 typedef struct InputFile {
258     AVFormatContext *ctx;
259     int eof_reached;      /* true if eof reached */
260     int ist_index;        /* index of first stream in input_streams */
261     int buffer_size;      /* current total buffer size */
262     int64_t ts_offset;
263     int nb_streams;       /* number of stream that ffmpeg is aware of; may be different
264                              from ctx.nb_streams if new streams appear during av_read_frame() */
265     int rate_emu;
266 } InputFile;
267
268 typedef struct OutputStream {
269     int file_index;          /* file index */
270     int index;               /* stream index in the output file */
271     int source_index;        /* InputStream index */
272     AVStream *st;            /* stream in the output file */
273     int encoding_needed;     /* true if encoding needed for this stream */
274     int frame_number;
275     /* input pts and corresponding output pts
276        for A/V sync */
277     struct InputStream *sync_ist; /* input stream to sync against */
278     int64_t sync_opts;       /* output frame counter, could be changed to some true timestamp */ // FIXME look at frame_number
279     /* pts of the first frame encoded for this stream, used for limiting
280      * recording time */
281     int64_t first_pts;
282     AVBitStreamFilterContext *bitstream_filters;
283     AVCodec *enc;
284     int64_t max_frames;
285     AVFrame *filtered_frame;
286
287     /* video only */
288     AVRational frame_rate;
289     int force_fps;
290     int top_field_first;
291
292     float frame_aspect_ratio;
293     float last_quality;
294
295     /* forced key frames */
296     int64_t *forced_kf_pts;
297     int forced_kf_count;
298     int forced_kf_index;
299
300     /* audio only */
301     int audio_channels_map[SWR_CH_MAX];  /* list of the channels id to pick from the source stream */
302     int audio_channels_mapped;           /* number of channels in audio_channels_map */
303
304     FILE *logfile;
305
306     OutputFilter *filter;
307     char *avfilter;
308
309     int64_t sws_flags;
310     int64_t swr_dither_method;
311     double swr_dither_scale;
312     AVDictionary *opts;
313     int is_past_recording_time;
314     int stream_copy;
315     const char *attachment_filename;
316     int copy_initial_nonkeyframes;
317
318     int keep_pix_fmt;
319 } OutputStream;
320
321
322 #if HAVE_TERMIOS_H
323
324 /* init terminal so that we can grab keys */
325 static struct termios oldtty;
326 static int restore_tty;
327 #endif
328
329 typedef struct OutputFile {
330     AVFormatContext *ctx;
331     AVDictionary *opts;
332     int ost_index;       /* index of the first stream in output_streams */
333     int64_t recording_time; /* desired length of the resulting file in microseconds */
334     int64_t start_time;     /* start time in microseconds */
335     uint64_t limit_filesize; /* filesize limit expressed in bytes */
336 } OutputFile;
337
338 static InputStream **input_streams = NULL;
339 static int        nb_input_streams = 0;
340 static InputFile   **input_files   = NULL;
341 static int        nb_input_files   = 0;
342
343 static OutputStream **output_streams = NULL;
344 static int         nb_output_streams = 0;
345 static OutputFile   **output_files   = NULL;
346 static int         nb_output_files   = 0;
347
348 static FilterGraph **filtergraphs;
349 int               nb_filtergraphs;
350
351 typedef struct OptionsContext {
352     /* input/output options */
353     int64_t start_time;
354     const char *format;
355
356     SpecifierOpt *codec_names;
357     int        nb_codec_names;
358     SpecifierOpt *audio_channels;
359     int        nb_audio_channels;
360     SpecifierOpt *audio_sample_rate;
361     int        nb_audio_sample_rate;
362     SpecifierOpt *frame_rates;
363     int        nb_frame_rates;
364     SpecifierOpt *frame_sizes;
365     int        nb_frame_sizes;
366     SpecifierOpt *frame_pix_fmts;
367     int        nb_frame_pix_fmts;
368
369     /* input options */
370     int64_t input_ts_offset;
371     int rate_emu;
372
373     SpecifierOpt *ts_scale;
374     int        nb_ts_scale;
375     SpecifierOpt *dump_attachment;
376     int        nb_dump_attachment;
377
378     /* output options */
379     StreamMap *stream_maps;
380     int     nb_stream_maps;
381     AudioChannelMap *audio_channel_maps; /* one info entry per -map_channel */
382     int           nb_audio_channel_maps; /* number of (valid) -map_channel settings */
383     int metadata_global_manual;
384     int metadata_streams_manual;
385     int metadata_chapters_manual;
386     const char **attachments;
387     int       nb_attachments;
388
389     int chapters_input_file;
390
391     int64_t recording_time;
392     uint64_t limit_filesize;
393     float mux_preload;
394     float mux_max_delay;
395
396     int video_disable;
397     int audio_disable;
398     int subtitle_disable;
399     int data_disable;
400
401     /* indexed by output file stream index */
402     int   *streamid_map;
403     int nb_streamid_map;
404
405     SpecifierOpt *metadata;
406     int        nb_metadata;
407     SpecifierOpt *max_frames;
408     int        nb_max_frames;
409     SpecifierOpt *bitstream_filters;
410     int        nb_bitstream_filters;
411     SpecifierOpt *codec_tags;
412     int        nb_codec_tags;
413     SpecifierOpt *sample_fmts;
414     int        nb_sample_fmts;
415     SpecifierOpt *qscale;
416     int        nb_qscale;
417     SpecifierOpt *forced_key_frames;
418     int        nb_forced_key_frames;
419     SpecifierOpt *force_fps;
420     int        nb_force_fps;
421     SpecifierOpt *frame_aspect_ratios;
422     int        nb_frame_aspect_ratios;
423     SpecifierOpt *rc_overrides;
424     int        nb_rc_overrides;
425     SpecifierOpt *intra_matrices;
426     int        nb_intra_matrices;
427     SpecifierOpt *inter_matrices;
428     int        nb_inter_matrices;
429     SpecifierOpt *top_field_first;
430     int        nb_top_field_first;
431     SpecifierOpt *metadata_map;
432     int        nb_metadata_map;
433     SpecifierOpt *presets;
434     int        nb_presets;
435     SpecifierOpt *copy_initial_nonkeyframes;
436     int        nb_copy_initial_nonkeyframes;
437     SpecifierOpt *filters;
438     int        nb_filters;
439 } OptionsContext;
440
441 static void do_video_stats(AVFormatContext *os, OutputStream *ost, int frame_size);
442
443 #define MATCH_PER_STREAM_OPT(name, type, outvar, fmtctx, st)\
444 {\
445     int i, ret;\
446     for (i = 0; i < o->nb_ ## name; i++) {\
447         char *spec = o->name[i].specifier;\
448         if ((ret = check_stream_specifier(fmtctx, st, spec)) > 0)\
449             outvar = o->name[i].u.type;\
450         else if (ret < 0)\
451             exit_program(1);\
452     }\
453 }
454
455 static int64_t getutime(void)
456 {
457 #if HAVE_GETRUSAGE
458     struct rusage rusage;
459
460     getrusage(RUSAGE_SELF, &rusage);
461     return (rusage.ru_utime.tv_sec * 1000000LL) + rusage.ru_utime.tv_usec;
462 #elif HAVE_GETPROCESSTIMES
463     HANDLE proc;
464     FILETIME c, e, k, u;
465     proc = GetCurrentProcess();
466     GetProcessTimes(proc, &c, &e, &k, &u);
467     return ((int64_t) u.dwHighDateTime << 32 | u.dwLowDateTime) / 10;
468 #else
469     return av_gettime();
470 #endif
471 }
472
473 static void update_benchmark(const char *fmt, ...)
474 {
475     if (do_benchmark_all) {
476         int64_t t = getutime();
477         va_list va;
478         char buf[1024];
479
480         if (fmt) {
481             va_start(va, fmt);
482             vsnprintf(buf, sizeof(buf), fmt, va);
483             va_end(va);
484             printf("bench: %8"PRIu64" %s \n", t - current_time, buf);
485         }
486         current_time = t;
487     }
488 }
489
490 static void reset_options(OptionsContext *o, int is_input)
491 {
492     const OptionDef *po = options;
493     OptionsContext bak= *o;
494     int i;
495
496     /* all OPT_SPEC and OPT_STRING can be freed in generic way */
497     while (po->name) {
498         void *dst = (uint8_t*)o + po->u.off;
499
500         if (po->flags & OPT_SPEC) {
501             SpecifierOpt **so = dst;
502             int i, *count = (int*)(so + 1);
503             for (i = 0; i < *count; i++) {
504                 av_freep(&(*so)[i].specifier);
505                 if (po->flags & OPT_STRING)
506                     av_freep(&(*so)[i].u.str);
507             }
508             av_freep(so);
509             *count = 0;
510         } else if (po->flags & OPT_OFFSET && po->flags & OPT_STRING)
511             av_freep(dst);
512         po++;
513     }
514
515     for (i = 0; i < o->nb_stream_maps; i++)
516         av_freep(&o->stream_maps[i].linklabel);
517     av_freep(&o->stream_maps);
518     av_freep(&o->audio_channel_maps);
519     av_freep(&o->streamid_map);
520
521     memset(o, 0, sizeof(*o));
522
523     if(is_input) o->recording_time = bak.recording_time;
524     else         o->recording_time = INT64_MAX;
525     o->mux_max_delay  = 0.7;
526     o->limit_filesize = UINT64_MAX;
527     o->chapters_input_file = INT_MAX;
528
529     uninit_opts();
530     init_opts();
531 }
532
533 static int alloc_buffer(InputStream *ist, AVCodecContext *s, FrameBuffer **pbuf)
534 {
535     FrameBuffer  *buf = av_mallocz(sizeof(*buf));
536     int i, ret;
537     const int pixel_size = av_pix_fmt_descriptors[s->pix_fmt].comp[0].step_minus1+1;
538     int h_chroma_shift, v_chroma_shift;
539     int edge = 32; // XXX should be avcodec_get_edge_width(), but that fails on svq1
540     int w = s->width, h = s->height;
541
542     if (!buf)
543         return AVERROR(ENOMEM);
544
545     avcodec_align_dimensions(s, &w, &h);
546
547     if (!(s->flags & CODEC_FLAG_EMU_EDGE)) {
548         w += 2*edge;
549         h += 2*edge;
550     }
551
552     if ((ret = av_image_alloc(buf->base, buf->linesize, w, h,
553                               s->pix_fmt, 32)) < 0) {
554         av_freep(&buf);
555         return ret;
556     }
557     /* XXX this shouldn't be needed, but some tests break without this line
558      * those decoders are buggy and need to be fixed.
559      * the following tests fail:
560      * cdgraphics, ansi, aasc, fraps-v1, qtrle-1bit
561      */
562     memset(buf->base[0], 128, ret);
563
564     avcodec_get_chroma_sub_sample(s->pix_fmt, &h_chroma_shift, &v_chroma_shift);
565     for (i = 0; i < FF_ARRAY_ELEMS(buf->data); i++) {
566         const int h_shift = i==0 ? 0 : h_chroma_shift;
567         const int v_shift = i==0 ? 0 : v_chroma_shift;
568         if ((s->flags & CODEC_FLAG_EMU_EDGE) || !buf->linesize[1] || !buf->base[i])
569             buf->data[i] = buf->base[i];
570         else
571             buf->data[i] = buf->base[i] +
572                            FFALIGN((buf->linesize[i]*edge >> v_shift) +
573                                    (pixel_size*edge >> h_shift), 32);
574     }
575     buf->w       = s->width;
576     buf->h       = s->height;
577     buf->pix_fmt = s->pix_fmt;
578     buf->ist     = ist;
579
580     *pbuf = buf;
581     return 0;
582 }
583
584 static void free_buffer_pool(InputStream *ist)
585 {
586     FrameBuffer *buf = ist->buffer_pool;
587     while (buf) {
588         ist->buffer_pool = buf->next;
589         av_freep(&buf->base[0]);
590         av_free(buf);
591         buf = ist->buffer_pool;
592     }
593 }
594
595 static void unref_buffer(InputStream *ist, FrameBuffer *buf)
596 {
597     av_assert0(buf->refcount > 0);
598     buf->refcount--;
599     if (!buf->refcount) {
600         FrameBuffer *tmp;
601         for(tmp= ist->buffer_pool; tmp; tmp= tmp->next)
602             av_assert1(tmp != buf);
603         buf->next = ist->buffer_pool;
604         ist->buffer_pool = buf;
605     }
606 }
607
608 static int codec_get_buffer(AVCodecContext *s, AVFrame *frame)
609 {
610     InputStream *ist = s->opaque;
611     FrameBuffer *buf;
612     int ret, i;
613
614     if(av_image_check_size(s->width, s->height, 0, s) || s->pix_fmt<0)
615         return -1;
616
617     if (!ist->buffer_pool && (ret = alloc_buffer(ist, s, &ist->buffer_pool)) < 0)
618         return ret;
619
620     buf              = ist->buffer_pool;
621     ist->buffer_pool = buf->next;
622     buf->next        = NULL;
623     if (buf->w != s->width || buf->h != s->height || buf->pix_fmt != s->pix_fmt) {
624         av_freep(&buf->base[0]);
625         av_free(buf);
626         if ((ret = alloc_buffer(ist, s, &buf)) < 0)
627             return ret;
628     }
629     av_assert0(!buf->refcount);
630     buf->refcount++;
631
632     frame->opaque        = buf;
633     frame->type          = FF_BUFFER_TYPE_USER;
634     frame->extended_data = frame->data;
635     frame->pkt_pts       = s->pkt ? s->pkt->pts : AV_NOPTS_VALUE;
636     frame->width         = buf->w;
637     frame->height        = buf->h;
638     frame->format        = buf->pix_fmt;
639     frame->sample_aspect_ratio = s->sample_aspect_ratio;
640
641     for (i = 0; i < FF_ARRAY_ELEMS(buf->data); i++) {
642         frame->base[i]     = buf->base[i];  // XXX h264.c uses base though it shouldn't
643         frame->data[i]     = buf->data[i];
644         frame->linesize[i] = buf->linesize[i];
645     }
646
647     return 0;
648 }
649
650 static void codec_release_buffer(AVCodecContext *s, AVFrame *frame)
651 {
652     InputStream *ist = s->opaque;
653     FrameBuffer *buf = frame->opaque;
654     int i;
655
656     if(frame->type!=FF_BUFFER_TYPE_USER)
657         return avcodec_default_release_buffer(s, frame);
658
659     for (i = 0; i < FF_ARRAY_ELEMS(frame->data); i++)
660         frame->data[i] = NULL;
661
662     unref_buffer(ist, buf);
663 }
664
665 static void filter_release_buffer(AVFilterBuffer *fb)
666 {
667     FrameBuffer *buf = fb->priv;
668     av_free(fb);
669     unref_buffer(buf->ist, buf);
670 }
671
672 static enum PixelFormat choose_pixel_fmt(AVStream *st, AVCodec *codec, enum PixelFormat target)
673 {
674     if (codec && codec->pix_fmts) {
675         const enum PixelFormat *p = codec->pix_fmts;
676         int has_alpha= av_pix_fmt_descriptors[target].nb_components % 2 == 0;
677         enum PixelFormat best= PIX_FMT_NONE;
678         if (st->codec->strict_std_compliance <= FF_COMPLIANCE_UNOFFICIAL) {
679             if (st->codec->codec_id == CODEC_ID_MJPEG) {
680                 p = (const enum PixelFormat[]) { PIX_FMT_YUVJ420P, PIX_FMT_YUVJ422P, PIX_FMT_YUV420P, PIX_FMT_YUV422P, PIX_FMT_NONE };
681             } else if (st->codec->codec_id == CODEC_ID_LJPEG) {
682                 p = (const enum PixelFormat[]) { PIX_FMT_YUVJ420P, PIX_FMT_YUVJ422P, PIX_FMT_YUVJ444P, PIX_FMT_YUV420P,
683                                                  PIX_FMT_YUV422P, PIX_FMT_YUV444P, PIX_FMT_BGRA, PIX_FMT_NONE };
684             }
685         }
686         for (; *p != PIX_FMT_NONE; p++) {
687             best= avcodec_find_best_pix_fmt2(best, *p, target, has_alpha, NULL);
688             if (*p == target)
689                 break;
690         }
691         if (*p == PIX_FMT_NONE) {
692             if (target != PIX_FMT_NONE)
693                 av_log(NULL, AV_LOG_WARNING,
694                        "Incompatible pixel format '%s' for codec '%s', auto-selecting format '%s'\n",
695                        av_pix_fmt_descriptors[target].name,
696                        codec->name,
697                        av_pix_fmt_descriptors[best].name);
698             return best;
699         }
700     }
701     return target;
702 }
703
704 static char *choose_pix_fmts(OutputStream *ost)
705 {
706      if (ost->keep_pix_fmt) {
707         if (ost->filter)
708             avfilter_graph_set_auto_convert(ost->filter->graph->graph,
709                                             AVFILTER_AUTO_CONVERT_NONE);
710         if (ost->st->codec->pix_fmt == PIX_FMT_NONE)
711             return NULL;
712         return av_strdup(av_get_pix_fmt_name(ost->st->codec->pix_fmt));
713     }
714     if (ost->st->codec->pix_fmt != PIX_FMT_NONE) {
715         return av_strdup(av_get_pix_fmt_name(choose_pixel_fmt(ost->st, ost->enc, ost->st->codec->pix_fmt)));
716     } else if (ost->enc && ost->enc->pix_fmts) {
717         const enum PixelFormat *p;
718         AVIOContext *s = NULL;
719         uint8_t *ret;
720         int len;
721
722         if (avio_open_dyn_buf(&s) < 0)
723             exit_program(1);
724
725         p = ost->enc->pix_fmts;
726         if (ost->st->codec->strict_std_compliance <= FF_COMPLIANCE_UNOFFICIAL) {
727             if (ost->st->codec->codec_id == CODEC_ID_MJPEG) {
728                 p = (const enum PixelFormat[]) { PIX_FMT_YUVJ420P, PIX_FMT_YUVJ422P, PIX_FMT_YUV420P, PIX_FMT_YUV422P, PIX_FMT_NONE };
729             } else if (ost->st->codec->codec_id == CODEC_ID_LJPEG) {
730                 p = (const enum PixelFormat[]) { PIX_FMT_YUVJ420P, PIX_FMT_YUVJ422P, PIX_FMT_YUVJ444P, PIX_FMT_YUV420P,
731                                                     PIX_FMT_YUV422P, PIX_FMT_YUV444P, PIX_FMT_BGRA, PIX_FMT_NONE };
732             }
733         }
734
735         for (; *p != PIX_FMT_NONE; p++) {
736             const char *name = av_get_pix_fmt_name(*p);
737             avio_printf(s, "%s:", name);
738         }
739         len = avio_close_dyn_buf(s, &ret);
740         ret[len - 1] = 0;
741         return ret;
742     } else
743         return NULL;
744 }
745
746 /**
747  * Define a function for building a string containing a list of
748  * allowed formats,
749  */
750 #define DEF_CHOOSE_FORMAT(type, var, supported_list, none, get_name, separator) \
751 static char *choose_ ## var ## s(OutputStream *ost)                             \
752 {                                                                               \
753     if (ost->st->codec->var != none) {                                          \
754         get_name(ost->st->codec->var);                                          \
755         return av_strdup(name);                                                 \
756     } else if (ost->enc->supported_list) {                                      \
757         const type *p;                                                          \
758         AVIOContext *s = NULL;                                                  \
759         uint8_t *ret;                                                           \
760         int len;                                                                \
761                                                                                 \
762         if (avio_open_dyn_buf(&s) < 0)                                          \
763             exit_program(1);                                                    \
764                                                                                 \
765         for (p = ost->enc->supported_list; *p != none; p++) {                   \
766             get_name(*p);                                                       \
767             avio_printf(s, "%s" separator, name);                               \
768         }                                                                       \
769         len = avio_close_dyn_buf(s, &ret);                                      \
770         ret[len - 1] = 0;                                                       \
771         return ret;                                                             \
772     } else                                                                      \
773         return NULL;                                                            \
774 }
775
776 #define GET_PIX_FMT_NAME(pix_fmt)\
777     const char *name = av_get_pix_fmt_name(pix_fmt);
778
779 // DEF_CHOOSE_FORMAT(enum PixelFormat, pix_fmt, pix_fmts, PIX_FMT_NONE,
780 //                   GET_PIX_FMT_NAME, ":")
781
782 #define GET_SAMPLE_FMT_NAME(sample_fmt)\
783     const char *name = av_get_sample_fmt_name(sample_fmt)
784
785 DEF_CHOOSE_FORMAT(enum AVSampleFormat, sample_fmt, sample_fmts,
786                   AV_SAMPLE_FMT_NONE, GET_SAMPLE_FMT_NAME, ",")
787
788 #define GET_SAMPLE_RATE_NAME(rate)\
789     char name[16];\
790     snprintf(name, sizeof(name), "%d", rate);
791
792 DEF_CHOOSE_FORMAT(int, sample_rate, supported_samplerates, 0,
793                   GET_SAMPLE_RATE_NAME, ",")
794
795 #define GET_CH_LAYOUT_NAME(ch_layout)\
796     char name[16];\
797     snprintf(name, sizeof(name), "0x%"PRIx64, ch_layout);
798
799 DEF_CHOOSE_FORMAT(uint64_t, channel_layout, channel_layouts, 0,
800                   GET_CH_LAYOUT_NAME, ",")
801
802 static int configure_audio_filters(FilterGraph *fg, AVFilterContext **in_filter,
803                                    AVFilterContext **out_filter)
804 {
805     InputStream  *ist = fg->inputs[0]->ist;
806     OutputStream *ost = fg->outputs[0]->ost;
807     AVCodecContext *codec  = ost->st->codec;
808     AVCodecContext *icodec = ist->st->codec;
809     char *sample_fmts, *sample_rates, *channel_layouts;
810     char args[256];
811     int ret;
812
813     avfilter_graph_free(&fg->graph);
814     if (!(fg->graph = avfilter_graph_alloc()))
815         return AVERROR(ENOMEM);
816
817     snprintf(args, sizeof(args), "time_base=%d/%d:sample_rate=%d:sample_fmt=%s:"
818              "channel_layout=0x%"PRIx64, ist->st->time_base.num,
819              ist->st->time_base.den, icodec->sample_rate,
820              av_get_sample_fmt_name(icodec->sample_fmt), icodec->channel_layout);
821     ret = avfilter_graph_create_filter(&fg->inputs[0]->filter,
822                                        avfilter_get_by_name("abuffer"),
823                                        "src", args, NULL, fg->graph);
824     if (ret < 0)
825         return ret;
826
827     ret = avfilter_graph_create_filter(&fg->outputs[0]->filter,
828                                        avfilter_get_by_name("abuffersink_old"),
829                                        "out", NULL, NULL, fg->graph);
830     if (ret < 0)
831         return ret;
832
833     *in_filter  = fg->inputs[0]->filter;
834     *out_filter = fg->outputs[0]->filter;
835
836     if (codec->channels && !codec->channel_layout)
837         codec->channel_layout = av_get_default_channel_layout(codec->channels);
838
839     sample_fmts     = choose_sample_fmts(ost);
840     sample_rates    = choose_sample_rates(ost);
841     channel_layouts = choose_channel_layouts(ost);
842     if (sample_fmts || sample_rates || channel_layouts) {
843         AVFilterContext *format;
844         char args[256];
845         int len = 0;
846
847         if (sample_fmts)
848             len += snprintf(args + len, sizeof(args) - len, "sample_fmts=%s:",
849                             sample_fmts);
850         if (sample_rates)
851             len += snprintf(args + len, sizeof(args) - len, "sample_rates=%s:",
852                             sample_rates);
853         if (channel_layouts)
854             len += snprintf(args + len, sizeof(args) - len, "channel_layouts=%s:",
855                             channel_layouts);
856         args[len - 1] = 0;
857
858         av_freep(&sample_fmts);
859         av_freep(&sample_rates);
860         av_freep(&channel_layouts);
861
862         ret = avfilter_graph_create_filter(&format,
863                                            avfilter_get_by_name("aformat"),
864                                            "aformat", args, NULL, fg->graph);
865         if (ret < 0)
866             return ret;
867
868         ret = avfilter_link(format, 0, fg->outputs[0]->filter, 0);
869         if (ret < 0)
870             return ret;
871
872         *out_filter = format;
873     }
874
875 #define AUTO_INSERT_FILTER(opt_name, filter_name, arg) do {                 \
876     AVFilterContext *filt_ctx;                                              \
877                                                                             \
878     av_log(NULL, AV_LOG_INFO, opt_name " is forwarded to lavfi "            \
879            "similarly to -af " filter_name "=%s.\n", arg);                  \
880                                                                             \
881     ret = avfilter_graph_create_filter(&filt_ctx,                           \
882                                        avfilter_get_by_name(filter_name),   \
883                                        filter_name, arg, NULL, fg->graph);  \
884     if (ret < 0)                                                            \
885         return ret;                                                         \
886                                                                             \
887     ret = avfilter_link(*in_filter, 0, filt_ctx, 0);                        \
888     if (ret < 0)                                                            \
889         return ret;                                                         \
890                                                                             \
891     *in_filter = filt_ctx;                                                  \
892 } while (0)
893
894     if (audio_sync_method > 0) {
895         char args[256] = {0};
896
897         av_strlcatf(args, sizeof(args), "min_comp=0.001:min_hard_comp=%f", audio_drift_threshold);
898         if (audio_sync_method > 1)
899             av_strlcatf(args, sizeof(args), ":max_soft_comp=%f", audio_sync_method/(double)icodec->sample_rate);
900         AUTO_INSERT_FILTER("-async", "aresample", args);
901     }
902
903     if (ost->audio_channels_mapped) {
904         int i;
905         AVBPrint pan_buf;
906
907         av_bprint_init(&pan_buf, 256, 8192);
908         av_bprintf(&pan_buf, "0x%"PRIx64,
909                    av_get_default_channel_layout(ost->audio_channels_mapped));
910         for (i = 0; i < ost->audio_channels_mapped; i++)
911             if (ost->audio_channels_map[i] != -1)
912                 av_bprintf(&pan_buf, ":c%d=c%d", i, ost->audio_channels_map[i]);
913
914         AUTO_INSERT_FILTER("-map_channel", "pan", pan_buf.str);
915         av_bprint_finalize(&pan_buf, NULL);
916     }
917
918     if (audio_volume != 256) {
919         char args[256];
920
921         snprintf(args, sizeof(args), "%lf", audio_volume / 256.);
922         AUTO_INSERT_FILTER("-vol", "volume", args);
923     }
924
925     return 0;
926 }
927
928 static int configure_video_filters(FilterGraph *fg, AVFilterContext **in_filter,
929                                    AVFilterContext **out_filter)
930 {
931     InputStream  *ist = fg->inputs[0]->ist;
932     OutputStream *ost = fg->outputs[0]->ost;
933     AVFilterContext *filter;
934     AVCodecContext *codec = ost->st->codec;
935     AVBufferSinkParams *buffersink_params = av_buffersink_params_alloc();
936     char *pix_fmts;
937     AVRational sample_aspect_ratio;
938     char args[255];
939     int ret;
940
941     if (ist->st->sample_aspect_ratio.num) {
942         sample_aspect_ratio = ist->st->sample_aspect_ratio;
943     } else
944         sample_aspect_ratio = ist->st->codec->sample_aspect_ratio;
945
946     snprintf(args, 255, "%d:%d:%d:%d:%d:%d:%d:flags=%d", ist->st->codec->width,
947              ist->st->codec->height, ist->st->codec->pix_fmt,
948              ist->st->time_base.num, ist->st->time_base.den,
949              sample_aspect_ratio.num, sample_aspect_ratio.den,
950              SWS_BILINEAR + ((ist->st->codec->flags&CODEC_FLAG_BITEXACT) ? SWS_BITEXACT:0));
951
952     ret = avfilter_graph_create_filter(&fg->inputs[0]->filter,
953                                        avfilter_get_by_name("buffer"),
954                                        "src", args, NULL, fg->graph);
955     if (ret < 0)
956         return ret;
957
958 #if FF_API_OLD_VSINK_API
959     ret = avfilter_graph_create_filter(&fg->outputs[0]->filter,
960                                        avfilter_get_by_name("buffersink"),
961                                        "out", NULL, NULL, fg->graph);
962 #else
963     ret = avfilter_graph_create_filter(&fg->outputs[0]->filter,
964                                        avfilter_get_by_name("buffersink"),
965                                        "out", NULL, buffersink_params, fg->graph);
966 #endif
967     av_freep(&buffersink_params);
968
969     if (ret < 0)
970         return ret;
971     *in_filter  = fg->inputs[0]->filter;
972     *out_filter = fg->outputs[0]->filter;
973
974     if (codec->width || codec->height) {
975         snprintf(args, 255, "%d:%d:flags=0x%X",
976                  codec->width,
977                  codec->height,
978                  (unsigned)ost->sws_flags);
979         if ((ret = avfilter_graph_create_filter(&filter, avfilter_get_by_name("scale"),
980                                                 NULL, args, NULL, fg->graph)) < 0)
981             return ret;
982         if ((ret = avfilter_link(*in_filter, 0, filter, 0)) < 0)
983             return ret;
984         *in_filter = filter;
985     }
986
987     if ((pix_fmts = choose_pix_fmts(ost))) {
988         if ((ret = avfilter_graph_create_filter(&filter,
989                                                 avfilter_get_by_name("format"),
990                                                 "format", pix_fmts, NULL,
991                                                 fg->graph)) < 0)
992             return ret;
993         if ((ret = avfilter_link(filter, 0, *out_filter, 0)) < 0)
994             return ret;
995
996         *out_filter = filter;
997         av_freep(&pix_fmts);
998     }
999
1000     snprintf(args, sizeof(args), "flags=0x%X", (unsigned)ost->sws_flags);
1001     fg->graph->scale_sws_opts = av_strdup(args);
1002
1003     return 0;
1004 }
1005
1006 static int configure_simple_filtergraph(FilterGraph *fg)
1007 {
1008     OutputStream *ost = fg->outputs[0]->ost;
1009     AVFilterContext *in_filter, *out_filter;
1010     int ret;
1011
1012     avfilter_graph_free(&fg->graph);
1013     fg->graph = avfilter_graph_alloc();
1014     if (!fg->graph)
1015         return AVERROR(ENOMEM);
1016
1017     switch (ost->st->codec->codec_type) {
1018     case AVMEDIA_TYPE_VIDEO:
1019         ret = configure_video_filters(fg, &in_filter, &out_filter);
1020         break;
1021     case AVMEDIA_TYPE_AUDIO:
1022         ret = configure_audio_filters(fg, &in_filter, &out_filter);
1023         break;
1024     default: av_assert0(0);
1025     }
1026     if (ret < 0)
1027         return ret;
1028
1029     if (ost->avfilter) {
1030         AVFilterInOut *outputs = avfilter_inout_alloc();
1031         AVFilterInOut *inputs  = avfilter_inout_alloc();
1032
1033         outputs->name    = av_strdup("in");
1034         outputs->filter_ctx = in_filter;
1035         outputs->pad_idx = 0;
1036         outputs->next    = NULL;
1037
1038         inputs->name    = av_strdup("out");
1039         inputs->filter_ctx = out_filter;
1040         inputs->pad_idx = 0;
1041         inputs->next    = NULL;
1042
1043         if ((ret = avfilter_graph_parse(fg->graph, ost->avfilter, &inputs, &outputs, NULL)) < 0)
1044             return ret;
1045         av_freep(&ost->avfilter);
1046     } else {
1047         if ((ret = avfilter_link(in_filter, 0, out_filter, 0)) < 0)
1048             return ret;
1049     }
1050
1051     if (ost->keep_pix_fmt)
1052         avfilter_graph_set_auto_convert(fg->graph,
1053                                         AVFILTER_AUTO_CONVERT_NONE);
1054
1055     if ((ret = avfilter_graph_config(fg->graph, NULL)) < 0)
1056         return ret;
1057
1058     ost->filter = fg->outputs[0];
1059
1060     return 0;
1061 }
1062
1063 static FilterGraph *init_simple_filtergraph(InputStream *ist, OutputStream *ost)
1064 {
1065     FilterGraph *fg = av_mallocz(sizeof(*fg));
1066
1067     if (!fg)
1068         exit_program(1);
1069     fg->index = nb_filtergraphs;
1070
1071     fg->outputs = grow_array(fg->outputs, sizeof(*fg->outputs), &fg->nb_outputs,
1072                              fg->nb_outputs + 1);
1073     if (!(fg->outputs[0] = av_mallocz(sizeof(*fg->outputs[0]))))
1074         exit_program(1);
1075     fg->outputs[0]->ost   = ost;
1076     fg->outputs[0]->graph = fg;
1077
1078     fg->inputs = grow_array(fg->inputs, sizeof(*fg->inputs), &fg->nb_inputs,
1079                             fg->nb_inputs + 1);
1080     if (!(fg->inputs[0] = av_mallocz(sizeof(*fg->inputs[0]))))
1081         exit_program(1);
1082     fg->inputs[0]->ist   = ist;
1083     fg->inputs[0]->graph = fg;
1084
1085     ist->filters = grow_array(ist->filters, sizeof(*ist->filters),
1086                               &ist->nb_filters, ist->nb_filters + 1);
1087     ist->filters[ist->nb_filters - 1] = fg->inputs[0];
1088
1089     filtergraphs = grow_array(filtergraphs, sizeof(*filtergraphs),
1090                               &nb_filtergraphs, nb_filtergraphs + 1);
1091     filtergraphs[nb_filtergraphs - 1] = fg;
1092
1093     return fg;
1094 }
1095
1096 static void init_input_filter(FilterGraph *fg, AVFilterInOut *in)
1097 {
1098     InputStream *ist = NULL;
1099     enum AVMediaType type = in->filter_ctx->input_pads[in->pad_idx].type;
1100     int i;
1101
1102     // TODO: support other filter types
1103     if (type != AVMEDIA_TYPE_VIDEO && type != AVMEDIA_TYPE_AUDIO) {
1104         av_log(NULL, AV_LOG_FATAL, "Only video and audio filters supported "
1105                "currently.\n");
1106         exit_program(1);
1107     }
1108
1109     if (in->name) {
1110         AVFormatContext *s;
1111         AVStream       *st = NULL;
1112         char *p;
1113         int file_idx = strtol(in->name, &p, 0);
1114
1115         if (file_idx < 0 || file_idx >= nb_input_files) {
1116             av_log(NULL, AV_LOG_FATAL, "Invalid file index %d in filtergraph description %s.\n",
1117                    file_idx, fg->graph_desc);
1118             exit_program(1);
1119         }
1120         s = input_files[file_idx]->ctx;
1121
1122         for (i = 0; i < s->nb_streams; i++) {
1123             if (s->streams[i]->codec->codec_type != type)
1124                 continue;
1125             if (check_stream_specifier(s, s->streams[i], *p == ':' ? p + 1 : p) == 1) {
1126                 st = s->streams[i];
1127                 break;
1128             }
1129         }
1130         if (!st) {
1131             av_log(NULL, AV_LOG_FATAL, "Stream specifier '%s' in filtergraph description %s "
1132                    "matches no streams.\n", p, fg->graph_desc);
1133             exit_program(1);
1134         }
1135         ist = input_streams[input_files[file_idx]->ist_index + st->index];
1136     } else {
1137         /* find the first unused stream of corresponding type */
1138         for (i = 0; i < nb_input_streams; i++) {
1139             ist = input_streams[i];
1140             if (ist->st->codec->codec_type == type && ist->discard)
1141                 break;
1142         }
1143         if (i == nb_input_streams) {
1144             av_log(NULL, AV_LOG_FATAL, "Cannot find a matching stream for "
1145                    "unlabeled input pad %d on filter %s", in->pad_idx,
1146                    in->filter_ctx->name);
1147             exit_program(1);
1148         }
1149     }
1150     ist->discard         = 0;
1151     ist->decoding_needed = 1;
1152     ist->st->discard = AVDISCARD_NONE;
1153
1154     fg->inputs = grow_array(fg->inputs, sizeof(*fg->inputs),
1155                             &fg->nb_inputs, fg->nb_inputs + 1);
1156     if (!(fg->inputs[fg->nb_inputs - 1] = av_mallocz(sizeof(*fg->inputs[0]))))
1157         exit_program(1);
1158     fg->inputs[fg->nb_inputs - 1]->ist   = ist;
1159     fg->inputs[fg->nb_inputs - 1]->graph = fg;
1160
1161     ist->filters = grow_array(ist->filters, sizeof(*ist->filters),
1162                               &ist->nb_filters, ist->nb_filters + 1);
1163     ist->filters[ist->nb_filters - 1] = fg->inputs[fg->nb_inputs - 1];
1164 }
1165
1166 static int configure_output_video_filter(FilterGraph *fg, OutputFilter *ofilter, AVFilterInOut *out)
1167 {
1168     char *pix_fmts;
1169     AVCodecContext *codec = ofilter->ost->st->codec;
1170     AVFilterContext *last_filter = out->filter_ctx;
1171     int pad_idx = out->pad_idx;
1172     int ret;
1173     AVBufferSinkParams *buffersink_params = av_buffersink_params_alloc();
1174
1175 #if FF_API_OLD_VSINK_API
1176     ret = avfilter_graph_create_filter(&ofilter->filter,
1177                                        avfilter_get_by_name("buffersink"),
1178                                        "out", NULL, NULL, fg->graph);
1179 #else
1180     ret = avfilter_graph_create_filter(&ofilter->filter,
1181                                        avfilter_get_by_name("buffersink"),
1182                                        "out", NULL, buffersink_params, fg->graph);
1183 #endif
1184     av_freep(&buffersink_params);
1185
1186     if (ret < 0)
1187         return ret;
1188
1189     if (codec->width || codec->height) {
1190         char args[255];
1191         AVFilterContext *filter;
1192
1193         snprintf(args, sizeof(args), "%d:%d:flags=0x%X",
1194                  codec->width,
1195                  codec->height,
1196                  (unsigned)ofilter->ost->sws_flags);
1197         if ((ret = avfilter_graph_create_filter(&filter, avfilter_get_by_name("scale"),
1198                                                 NULL, args, NULL, fg->graph)) < 0)
1199             return ret;
1200         if ((ret = avfilter_link(last_filter, pad_idx, filter, 0)) < 0)
1201             return ret;
1202
1203         last_filter = filter;
1204         pad_idx = 0;
1205     }
1206
1207     if ((pix_fmts = choose_pix_fmts(ofilter->ost))) {
1208         AVFilterContext *filter;
1209         if ((ret = avfilter_graph_create_filter(&filter,
1210                                                 avfilter_get_by_name("format"),
1211                                                 "format", pix_fmts, NULL,
1212                                                 fg->graph)) < 0)
1213             return ret;
1214         if ((ret = avfilter_link(last_filter, pad_idx, filter, 0)) < 0)
1215             return ret;
1216
1217         last_filter = filter;
1218         pad_idx     = 0;
1219         av_freep(&pix_fmts);
1220     }
1221
1222     if ((ret = avfilter_link(last_filter, pad_idx, ofilter->filter, 0)) < 0)
1223         return ret;
1224
1225     return 0;
1226 }
1227
1228 static int configure_output_audio_filter(FilterGraph *fg, OutputFilter *ofilter, AVFilterInOut *out)
1229 {
1230     OutputStream *ost = ofilter->ost;
1231     AVCodecContext *codec  = ost->st->codec;
1232     AVFilterContext *last_filter = out->filter_ctx;
1233     int pad_idx = out->pad_idx;
1234     char *sample_fmts, *sample_rates, *channel_layouts;
1235     int ret;
1236
1237     ret = avfilter_graph_create_filter(&ofilter->filter,
1238                                        avfilter_get_by_name("abuffersink"),
1239                                        "out", NULL, NULL, fg->graph);
1240     if (ret < 0)
1241         return ret;
1242
1243     if (codec->channels && !codec->channel_layout)
1244         codec->channel_layout = av_get_default_channel_layout(codec->channels);
1245
1246     sample_fmts     = choose_sample_fmts(ost);
1247     sample_rates    = choose_sample_rates(ost);
1248     channel_layouts = choose_channel_layouts(ost);
1249     if (sample_fmts || sample_rates || channel_layouts) {
1250         AVFilterContext *format;
1251         char args[256];
1252         int len = 0;
1253
1254         if (sample_fmts)
1255             len += snprintf(args + len, sizeof(args) - len, "sample_fmts=%s:",
1256                             sample_fmts);
1257         if (sample_rates)
1258             len += snprintf(args + len, sizeof(args) - len, "sample_rates=%s:",
1259                             sample_rates);
1260         if (channel_layouts)
1261             len += snprintf(args + len, sizeof(args) - len, "channel_layouts=%s:",
1262                             channel_layouts);
1263         args[len - 1] = 0;
1264
1265         av_freep(&sample_fmts);
1266         av_freep(&sample_rates);
1267         av_freep(&channel_layouts);
1268
1269         ret = avfilter_graph_create_filter(&format,
1270                                            avfilter_get_by_name("aformat"),
1271                                            "aformat", args, NULL, fg->graph);
1272         if (ret < 0)
1273             return ret;
1274
1275         ret = avfilter_link(last_filter, pad_idx, format, 0);
1276         if (ret < 0)
1277             return ret;
1278
1279         last_filter = format;
1280         pad_idx = 0;
1281     }
1282
1283     if (audio_sync_method > 0) {
1284         AVFilterContext *async;
1285         char args[256];
1286         int  len = 0;
1287
1288         av_log(NULL, AV_LOG_WARNING, "-async has been deprecated. Used the "
1289                "asyncts audio filter instead.\n");
1290
1291         if (audio_sync_method > 1)
1292             len += snprintf(args + len, sizeof(args) - len, "compensate=1:"
1293                             "max_comp=%d:", audio_sync_method);
1294         snprintf(args + len, sizeof(args) - len, "min_delta=%f",
1295                  audio_drift_threshold);
1296
1297         ret = avfilter_graph_create_filter(&async,
1298                                            avfilter_get_by_name("asyncts"),
1299                                            "async", args, NULL, fg->graph);
1300         if (ret < 0)
1301             return ret;
1302
1303         ret = avfilter_link(last_filter, pad_idx, async, 0);
1304         if (ret < 0)
1305             return ret;
1306
1307         last_filter = async;
1308         pad_idx = 0;
1309     }
1310
1311     if ((ret = avfilter_link(last_filter, pad_idx, ofilter->filter, 0)) < 0)
1312         return ret;
1313
1314     return 0;
1315 }
1316
1317 static int configure_output_filter(FilterGraph *fg, OutputFilter *ofilter, AVFilterInOut *out)
1318 {
1319     switch (out->filter_ctx->output_pads[out->pad_idx].type) {
1320     case AVMEDIA_TYPE_VIDEO: return configure_output_video_filter(fg, ofilter, out);
1321     case AVMEDIA_TYPE_AUDIO: return configure_output_audio_filter(fg, ofilter, out);
1322     default: av_assert0(0);
1323     }
1324 }
1325
1326 static int configure_complex_filter(FilterGraph *fg)
1327 {
1328     AVFilterInOut *inputs, *outputs, *cur;
1329     int ret, i, init = !fg->graph;
1330
1331     avfilter_graph_free(&fg->graph);
1332     if (!(fg->graph = avfilter_graph_alloc()))
1333         return AVERROR(ENOMEM);
1334
1335     if ((ret = avfilter_graph_parse2(fg->graph, fg->graph_desc, &inputs, &outputs)) < 0)
1336         return ret;
1337
1338     for (cur = inputs; init && cur; cur = cur->next)
1339         init_input_filter(fg, cur);
1340
1341     for (cur = inputs, i = 0; cur; cur = cur->next, i++) {
1342         InputFilter *ifilter = fg->inputs[i];
1343         InputStream     *ist = ifilter->ist;
1344         AVRational       sar;
1345         AVFilter     *filter;
1346         char            args[255];
1347
1348         switch (cur->filter_ctx->input_pads[cur->pad_idx].type) {
1349         case AVMEDIA_TYPE_VIDEO:
1350             sar = ist->st->sample_aspect_ratio.num ?
1351                   ist->st->sample_aspect_ratio :
1352                   ist->st->codec->sample_aspect_ratio;
1353             snprintf(args, sizeof(args), "%d:%d:%d:%d:%d:%d:%d", ist->st->codec->width,
1354                      ist->st->codec->height, ist->st->codec->pix_fmt,
1355                      ist->st->time_base.num, ist->st->time_base.den,
1356                      sar.num, sar.den);
1357             filter = avfilter_get_by_name("buffer");
1358             break;
1359         case AVMEDIA_TYPE_AUDIO:
1360             snprintf(args, sizeof(args), "time_base=%d/%d:sample_rate=%d:"
1361                      "sample_fmt=%s:channel_layout=0x%"PRIx64,
1362                      ist->st->time_base.num, ist->st->time_base.den,
1363                      ist->st->codec->sample_rate,
1364                      av_get_sample_fmt_name(ist->st->codec->sample_fmt),
1365                      ist->st->codec->channel_layout);
1366             filter = avfilter_get_by_name("abuffer");
1367             break;
1368         default:
1369             av_assert0(0);
1370         }
1371
1372         if ((ret = avfilter_graph_create_filter(&ifilter->filter,
1373                                                 filter, cur->name,
1374                                                 args, NULL, fg->graph)) < 0)
1375             return ret;
1376         if ((ret = avfilter_link(ifilter->filter, 0,
1377                                  cur->filter_ctx, cur->pad_idx)) < 0)
1378             return ret;
1379     }
1380     avfilter_inout_free(&inputs);
1381
1382     if (!init) {
1383         /* we already know the mappings between lavfi outputs and output streams,
1384          * so we can finish the setup */
1385         for (cur = outputs, i = 0; cur; cur = cur->next, i++)
1386             configure_output_filter(fg, fg->outputs[i], cur);
1387         avfilter_inout_free(&outputs);
1388
1389         if ((ret = avfilter_graph_config(fg->graph, NULL)) < 0)
1390             return ret;
1391     } else {
1392         /* wait until output mappings are processed */
1393         for (cur = outputs; cur;) {
1394             fg->outputs = grow_array(fg->outputs, sizeof(*fg->outputs),
1395                                      &fg->nb_outputs, fg->nb_outputs + 1);
1396             if (!(fg->outputs[fg->nb_outputs - 1] = av_mallocz(sizeof(*fg->outputs[0]))))
1397                 exit_program(1);
1398             fg->outputs[fg->nb_outputs - 1]->graph   = fg;
1399             fg->outputs[fg->nb_outputs - 1]->out_tmp = cur;
1400             cur = cur->next;
1401             fg->outputs[fg->nb_outputs - 1]->out_tmp->next = NULL;
1402         }
1403     }
1404
1405     return 0;
1406 }
1407
1408 static int configure_complex_filters(void)
1409 {
1410     int i, ret = 0;
1411
1412     for (i = 0; i < nb_filtergraphs; i++)
1413         if (!filtergraphs[i]->graph &&
1414             (ret = configure_complex_filter(filtergraphs[i])) < 0)
1415             return ret;
1416     return 0;
1417 }
1418
1419 static int configure_filtergraph(FilterGraph *fg)
1420 {
1421     return fg->graph_desc ? configure_complex_filter(fg) :
1422                             configure_simple_filtergraph(fg);
1423 }
1424
1425 static int ist_in_filtergraph(FilterGraph *fg, InputStream *ist)
1426 {
1427     int i;
1428     for (i = 0; i < fg->nb_inputs; i++)
1429         if (fg->inputs[i]->ist == ist)
1430             return 1;
1431     return 0;
1432 }
1433
1434 static void term_exit(void)
1435 {
1436     av_log(NULL, AV_LOG_QUIET, "%s", "");
1437 #if HAVE_TERMIOS_H
1438     if(restore_tty)
1439         tcsetattr (0, TCSANOW, &oldtty);
1440 #endif
1441 }
1442
1443 static volatile int received_sigterm = 0;
1444
1445 static void sigterm_handler(int sig)
1446 {
1447     received_sigterm = sig;
1448     received_nb_signals++;
1449     term_exit();
1450     if(received_nb_signals > 3)
1451         exit(123);
1452 }
1453
1454 static void term_init(void)
1455 {
1456 #if HAVE_TERMIOS_H
1457     if(!run_as_daemon){
1458         struct termios tty;
1459
1460         if (tcgetattr (0, &tty) == 0) {
1461             oldtty = tty;
1462             restore_tty = 1;
1463             atexit(term_exit);
1464
1465             tty.c_iflag &= ~(IGNBRK|BRKINT|PARMRK|ISTRIP
1466                              |INLCR|IGNCR|ICRNL|IXON);
1467             tty.c_oflag |= OPOST;
1468             tty.c_lflag &= ~(ECHO|ECHONL|ICANON|IEXTEN);
1469             tty.c_cflag &= ~(CSIZE|PARENB);
1470             tty.c_cflag |= CS8;
1471             tty.c_cc[VMIN] = 1;
1472             tty.c_cc[VTIME] = 0;
1473
1474             tcsetattr (0, TCSANOW, &tty);
1475         }
1476         signal(SIGQUIT, sigterm_handler); /* Quit (POSIX).  */
1477     }
1478 #endif
1479     avformat_network_deinit();
1480
1481     signal(SIGINT , sigterm_handler); /* Interrupt (ANSI).    */
1482     signal(SIGTERM, sigterm_handler); /* Termination (ANSI).  */
1483 #ifdef SIGXCPU
1484     signal(SIGXCPU, sigterm_handler);
1485 #endif
1486 }
1487
1488 /* read a key without blocking */
1489 static int read_key(void)
1490 {
1491     unsigned char ch;
1492 #if HAVE_TERMIOS_H
1493     int n = 1;
1494     struct timeval tv;
1495     fd_set rfds;
1496
1497     FD_ZERO(&rfds);
1498     FD_SET(0, &rfds);
1499     tv.tv_sec = 0;
1500     tv.tv_usec = 0;
1501     n = select(1, &rfds, NULL, NULL, &tv);
1502     if (n > 0) {
1503         n = read(0, &ch, 1);
1504         if (n == 1)
1505             return ch;
1506
1507         return n;
1508     }
1509 #elif HAVE_KBHIT
1510 #    if HAVE_PEEKNAMEDPIPE
1511     static int is_pipe;
1512     static HANDLE input_handle;
1513     DWORD dw, nchars;
1514     if(!input_handle){
1515         input_handle = GetStdHandle(STD_INPUT_HANDLE);
1516         is_pipe = !GetConsoleMode(input_handle, &dw);
1517     }
1518
1519     if (stdin->_cnt > 0) {
1520         read(0, &ch, 1);
1521         return ch;
1522     }
1523     if (is_pipe) {
1524         /* When running under a GUI, you will end here. */
1525         if (!PeekNamedPipe(input_handle, NULL, 0, NULL, &nchars, NULL))
1526             return -1;
1527         //Read it
1528         if(nchars != 0) {
1529             read(0, &ch, 1);
1530             return ch;
1531         }else{
1532             return -1;
1533         }
1534     }
1535 #    endif
1536     if(kbhit())
1537         return(getch());
1538 #endif
1539     return -1;
1540 }
1541
1542 static int decode_interrupt_cb(void *ctx)
1543 {
1544     return received_nb_signals > 1;
1545 }
1546
1547 static const AVIOInterruptCB int_cb = { decode_interrupt_cb, NULL };
1548
1549 void av_noreturn exit_program(int ret)
1550 {
1551     int i, j;
1552
1553     for (i = 0; i < nb_filtergraphs; i++) {
1554         avfilter_graph_free(&filtergraphs[i]->graph);
1555         for (j = 0; j < filtergraphs[i]->nb_inputs; j++)
1556             av_freep(&filtergraphs[i]->inputs[j]);
1557         av_freep(&filtergraphs[i]->inputs);
1558         for (j = 0; j < filtergraphs[i]->nb_outputs; j++)
1559             av_freep(&filtergraphs[i]->outputs[j]);
1560         av_freep(&filtergraphs[i]->outputs);
1561         av_freep(&filtergraphs[i]);
1562     }
1563     av_freep(&filtergraphs);
1564
1565     /* close files */
1566     for (i = 0; i < nb_output_files; i++) {
1567         AVFormatContext *s = output_files[i]->ctx;
1568         if (!(s->oformat->flags & AVFMT_NOFILE) && s->pb)
1569             avio_close(s->pb);
1570         avformat_free_context(s);
1571         av_dict_free(&output_files[i]->opts);
1572         av_freep(&output_files[i]);
1573     }
1574     for (i = 0; i < nb_output_streams; i++) {
1575         AVBitStreamFilterContext *bsfc = output_streams[i]->bitstream_filters;
1576         while (bsfc) {
1577             AVBitStreamFilterContext *next = bsfc->next;
1578             av_bitstream_filter_close(bsfc);
1579             bsfc = next;
1580         }
1581         output_streams[i]->bitstream_filters = NULL;
1582
1583         av_freep(&output_streams[i]->filtered_frame);
1584         av_freep(&output_streams[i]);
1585     }
1586     for (i = 0; i < nb_input_files; i++) {
1587         avformat_close_input(&input_files[i]->ctx);
1588         av_freep(&input_files[i]);
1589     }
1590     for (i = 0; i < nb_input_streams; i++) {
1591         av_freep(&input_streams[i]->decoded_frame);
1592         av_dict_free(&input_streams[i]->opts);
1593         free_buffer_pool(input_streams[i]);
1594         av_freep(&input_streams[i]->filters);
1595         av_freep(&input_streams[i]);
1596     }
1597
1598     if (vstats_file)
1599         fclose(vstats_file);
1600     av_free(vstats_filename);
1601
1602     av_freep(&input_streams);
1603     av_freep(&input_files);
1604     av_freep(&output_streams);
1605     av_freep(&output_files);
1606
1607     uninit_opts();
1608
1609     avfilter_uninit();
1610     avformat_network_deinit();
1611
1612     if (received_sigterm) {
1613         av_log(NULL, AV_LOG_INFO, "Received signal %d: terminating.\n",
1614                (int) received_sigterm);
1615         exit (255);
1616     }
1617
1618     exit(ret);
1619 }
1620
1621 static void assert_avoptions(AVDictionary *m)
1622 {
1623     AVDictionaryEntry *t;
1624     if ((t = av_dict_get(m, "", NULL, AV_DICT_IGNORE_SUFFIX))) {
1625         av_log(NULL, AV_LOG_FATAL, "Option %s not found.\n", t->key);
1626         exit_program(1);
1627     }
1628 }
1629
1630 static void assert_codec_experimental(AVCodecContext *c, int encoder)
1631 {
1632     const char *codec_string = encoder ? "encoder" : "decoder";
1633     AVCodec *codec;
1634     if (c->codec->capabilities & CODEC_CAP_EXPERIMENTAL &&
1635         c->strict_std_compliance > FF_COMPLIANCE_EXPERIMENTAL) {
1636         av_log(NULL, AV_LOG_FATAL, "%s '%s' is experimental and might produce bad "
1637                 "results.\nAdd '-strict experimental' if you want to use it.\n",
1638                 codec_string, c->codec->name);
1639         codec = encoder ? avcodec_find_encoder(c->codec->id) : avcodec_find_decoder(c->codec->id);
1640         if (!(codec->capabilities & CODEC_CAP_EXPERIMENTAL))
1641             av_log(NULL, AV_LOG_FATAL, "Or use the non experimental %s '%s'.\n",
1642                    codec_string, codec->name);
1643         exit_program(1);
1644     }
1645 }
1646
1647 static void choose_sample_fmt(AVStream *st, AVCodec *codec)
1648 {
1649     if (codec && codec->sample_fmts) {
1650         const enum AVSampleFormat *p = codec->sample_fmts;
1651         for (; *p != -1; p++) {
1652             if (*p == st->codec->sample_fmt)
1653                 break;
1654         }
1655         if (*p == -1) {
1656             if((codec->capabilities & CODEC_CAP_LOSSLESS) && av_get_sample_fmt_name(st->codec->sample_fmt) > av_get_sample_fmt_name(codec->sample_fmts[0]))
1657                 av_log(NULL, AV_LOG_ERROR, "Conversion will not be lossless.\n");
1658             if(av_get_sample_fmt_name(st->codec->sample_fmt))
1659             av_log(NULL, AV_LOG_WARNING,
1660                    "Incompatible sample format '%s' for codec '%s', auto-selecting format '%s'\n",
1661                    av_get_sample_fmt_name(st->codec->sample_fmt),
1662                    codec->name,
1663                    av_get_sample_fmt_name(codec->sample_fmts[0]));
1664             st->codec->sample_fmt = codec->sample_fmts[0];
1665         }
1666     }
1667 }
1668
1669 static void write_frame(AVFormatContext *s, AVPacket *pkt, OutputStream *ost)
1670 {
1671     AVBitStreamFilterContext *bsfc = ost->bitstream_filters;
1672     AVCodecContext          *avctx = ost->st->codec;
1673     int ret;
1674
1675     if ((avctx->codec_type == AVMEDIA_TYPE_VIDEO && video_sync_method == VSYNC_DROP) ||
1676         (avctx->codec_type == AVMEDIA_TYPE_AUDIO && audio_sync_method < 0))
1677         pkt->pts = pkt->dts = AV_NOPTS_VALUE;
1678
1679     if (avctx->codec_type == AVMEDIA_TYPE_AUDIO && pkt->dts != AV_NOPTS_VALUE) {
1680         int64_t max = ost->st->cur_dts + !(s->oformat->flags & AVFMT_TS_NONSTRICT);
1681         if (ost->st->cur_dts && ost->st->cur_dts != AV_NOPTS_VALUE &&  max > pkt->dts) {
1682             av_log(s, max - pkt->dts > 2 ? AV_LOG_WARNING : AV_LOG_DEBUG, "Audio timestamp %"PRId64" < %"PRId64" invalid, cliping\n", pkt->dts, max);
1683             pkt->pts = pkt->dts = max;
1684         }
1685     }
1686
1687     /*
1688      * Audio encoders may split the packets --  #frames in != #packets out.
1689      * But there is no reordering, so we can limit the number of output packets
1690      * by simply dropping them here.
1691      * Counting encoded video frames needs to be done separately because of
1692      * reordering, see do_video_out()
1693      */
1694     if (!(avctx->codec_type == AVMEDIA_TYPE_VIDEO && avctx->codec)) {
1695         if (ost->frame_number >= ost->max_frames) {
1696             av_free_packet(pkt);
1697             return;
1698         }
1699         ost->frame_number++;
1700     }
1701
1702     while (bsfc) {
1703         AVPacket new_pkt = *pkt;
1704         int a = av_bitstream_filter_filter(bsfc, avctx, NULL,
1705                                            &new_pkt.data, &new_pkt.size,
1706                                            pkt->data, pkt->size,
1707                                            pkt->flags & AV_PKT_FLAG_KEY);
1708         if (a > 0) {
1709             av_free_packet(pkt);
1710             new_pkt.destruct = av_destruct_packet;
1711         } else if (a < 0) {
1712             av_log(NULL, AV_LOG_ERROR, "Failed to open bitstream filter %s for stream %d with codec %s",
1713                    bsfc->filter->name, pkt->stream_index,
1714                    avctx->codec ? avctx->codec->name : "copy");
1715             print_error("", a);
1716             if (exit_on_error)
1717                 exit_program(1);
1718         }
1719         *pkt = new_pkt;
1720
1721         bsfc = bsfc->next;
1722     }
1723
1724     pkt->stream_index = ost->index;
1725     ret = av_interleaved_write_frame(s, pkt);
1726     if (ret < 0) {
1727         print_error("av_interleaved_write_frame()", ret);
1728         exit_program(1);
1729     }
1730 }
1731
1732 static int check_recording_time(OutputStream *ost)
1733 {
1734     OutputFile *of = output_files[ost->file_index];
1735
1736     if (of->recording_time != INT64_MAX &&
1737         av_compare_ts(ost->sync_opts - ost->first_pts, ost->st->codec->time_base, of->recording_time,
1738                       AV_TIME_BASE_Q) >= 0) {
1739         ost->is_past_recording_time = 1;
1740         return 0;
1741     }
1742     return 1;
1743 }
1744
1745 static void do_audio_out(AVFormatContext *s, OutputStream *ost,
1746                          AVFrame *frame)
1747 {
1748     AVCodecContext *enc = ost->st->codec;
1749     AVPacket pkt;
1750     int got_packet = 0;
1751
1752     av_init_packet(&pkt);
1753     pkt.data = NULL;
1754     pkt.size = 0;
1755 #if 0
1756     if (!check_recording_time(ost))
1757         return;
1758 #endif
1759     if (frame->pts == AV_NOPTS_VALUE || audio_sync_method < 0)
1760         frame->pts = ost->sync_opts;
1761     ost->sync_opts = frame->pts + frame->nb_samples;
1762
1763     av_assert0(pkt.size || !pkt.data);
1764     update_benchmark(NULL);
1765     if (avcodec_encode_audio2(enc, &pkt, frame, &got_packet) < 0) {
1766         av_log(NULL, AV_LOG_FATAL, "Audio encoding failed (avcodec_encode_audio2)\n");
1767         exit_program(1);
1768     }
1769     update_benchmark("encode_audio %d.%d", ost->file_index, ost->index);
1770
1771     if (got_packet) {
1772         if (pkt.pts != AV_NOPTS_VALUE)
1773             pkt.pts      = av_rescale_q(pkt.pts,      enc->time_base, ost->st->time_base);
1774         if (pkt.dts != AV_NOPTS_VALUE)
1775             pkt.dts      = av_rescale_q(pkt.dts,      enc->time_base, ost->st->time_base);
1776         if (pkt.duration > 0)
1777             pkt.duration = av_rescale_q(pkt.duration, enc->time_base, ost->st->time_base);
1778
1779         if (debug_ts) {
1780             av_log(NULL, AV_LOG_INFO, "encoder -> type:audio "
1781                    "pkt_pts:%s pkt_pts_time:%s pkt_dts:%s pkt_dts_time:%s\n",
1782                    av_ts2str(pkt.pts), av_ts2timestr(pkt.pts, &ost->st->time_base),
1783                    av_ts2str(pkt.dts), av_ts2timestr(pkt.dts, &ost->st->time_base));
1784         }
1785
1786         write_frame(s, &pkt, ost);
1787
1788         audio_size += pkt.size;
1789         av_free_packet(&pkt);
1790     }
1791 }
1792
1793 static void pre_process_video_frame(InputStream *ist, AVPicture *picture, void **bufp)
1794 {
1795     AVCodecContext *dec;
1796     AVPicture *picture2;
1797     AVPicture picture_tmp;
1798     uint8_t *buf = 0;
1799
1800     dec = ist->st->codec;
1801
1802     /* deinterlace : must be done before any resize */
1803     if (do_deinterlace) {
1804         int size;
1805
1806         /* create temporary picture */
1807         size = avpicture_get_size(dec->pix_fmt, dec->width, dec->height);
1808         buf  = av_malloc(size);
1809         if (!buf)
1810             return;
1811
1812         picture2 = &picture_tmp;
1813         avpicture_fill(picture2, buf, dec->pix_fmt, dec->width, dec->height);
1814
1815         if (avpicture_deinterlace(picture2, picture,
1816                                  dec->pix_fmt, dec->width, dec->height) < 0) {
1817             /* if error, do not deinterlace */
1818             av_log(NULL, AV_LOG_WARNING, "Deinterlacing failed\n");
1819             av_free(buf);
1820             buf = NULL;
1821             picture2 = picture;
1822         }
1823     } else {
1824         picture2 = picture;
1825     }
1826
1827     if (picture != picture2)
1828         *picture = *picture2;
1829     *bufp = buf;
1830 }
1831
1832 static void do_subtitle_out(AVFormatContext *s,
1833                             OutputStream *ost,
1834                             InputStream *ist,
1835                             AVSubtitle *sub,
1836                             int64_t pts)
1837 {
1838     static uint8_t *subtitle_out = NULL;
1839     int subtitle_out_max_size = 1024 * 1024;
1840     int subtitle_out_size, nb, i;
1841     AVCodecContext *enc;
1842     AVPacket pkt;
1843
1844     if (pts == AV_NOPTS_VALUE) {
1845         av_log(NULL, AV_LOG_ERROR, "Subtitle packets must have a pts\n");
1846         if (exit_on_error)
1847             exit_program(1);
1848         return;
1849     }
1850
1851     enc = ost->st->codec;
1852
1853     if (!subtitle_out) {
1854         subtitle_out = av_malloc(subtitle_out_max_size);
1855     }
1856
1857     /* Note: DVB subtitle need one packet to draw them and one other
1858        packet to clear them */
1859     /* XXX: signal it in the codec context ? */
1860     if (enc->codec_id == CODEC_ID_DVB_SUBTITLE)
1861         nb = 2;
1862     else
1863         nb = 1;
1864
1865     for (i = 0; i < nb; i++) {
1866         ost->sync_opts = av_rescale_q(pts, ist->st->time_base, enc->time_base);
1867
1868         sub->pts = av_rescale_q(pts, ist->st->time_base, AV_TIME_BASE_Q);
1869         // start_display_time is required to be 0
1870         sub->pts               += av_rescale_q(sub->start_display_time, (AVRational){ 1, 1000 }, AV_TIME_BASE_Q);
1871         sub->end_display_time  -= sub->start_display_time;
1872         sub->start_display_time = 0;
1873         subtitle_out_size = avcodec_encode_subtitle(enc, subtitle_out,
1874                                                     subtitle_out_max_size, sub);
1875         if (subtitle_out_size < 0) {
1876             av_log(NULL, AV_LOG_FATAL, "Subtitle encoding failed\n");
1877             exit_program(1);
1878         }
1879
1880         av_init_packet(&pkt);
1881         pkt.data = subtitle_out;
1882         pkt.size = subtitle_out_size;
1883         pkt.pts  = av_rescale_q(sub->pts, AV_TIME_BASE_Q, ost->st->time_base);
1884         if (enc->codec_id == CODEC_ID_DVB_SUBTITLE) {
1885             /* XXX: the pts correction is handled here. Maybe handling
1886                it in the codec would be better */
1887             if (i == 0)
1888                 pkt.pts += 90 * sub->start_display_time;
1889             else
1890                 pkt.pts += 90 * sub->end_display_time;
1891         }
1892         write_frame(s, &pkt, ost);
1893     }
1894 }
1895
1896 static void do_video_out(AVFormatContext *s,
1897                          OutputStream *ost,
1898                          AVFrame *in_picture,
1899                          float quality)
1900 {
1901     int ret, format_video_sync;
1902     AVPacket pkt;
1903     AVCodecContext *enc = ost->st->codec;
1904     int nb_frames, i;
1905     double sync_ipts, delta;
1906     double duration = 0;
1907     int frame_size = 0;
1908     InputStream *ist = NULL;
1909
1910     if (ost->source_index >= 0)
1911         ist = input_streams[ost->source_index];
1912
1913     if(ist && ist->st->start_time != AV_NOPTS_VALUE && ist->st->first_dts != AV_NOPTS_VALUE && ost->frame_rate.num)
1914         duration = 1/(av_q2d(ost->frame_rate) * av_q2d(enc->time_base));
1915
1916     sync_ipts = in_picture->pts;
1917     delta = sync_ipts - ost->sync_opts + duration;
1918
1919     /* by default, we output a single frame */
1920     nb_frames = 1;
1921
1922     format_video_sync = video_sync_method;
1923     if (format_video_sync == VSYNC_AUTO)
1924         format_video_sync = (s->oformat->flags & AVFMT_VARIABLE_FPS) ? ((s->oformat->flags & AVFMT_NOTIMESTAMPS) ? VSYNC_PASSTHROUGH : VSYNC_VFR) : 1;
1925
1926     switch (format_video_sync) {
1927     case VSYNC_CFR:
1928         // FIXME set to 0.5 after we fix some dts/pts bugs like in avidec.c
1929         if (delta < -1.1)
1930             nb_frames = 0;
1931         else if (delta > 1.1)
1932             nb_frames = lrintf(delta);
1933         break;
1934     case VSYNC_VFR:
1935         if (delta <= -0.6)
1936             nb_frames = 0;
1937         else if (delta > 0.6)
1938             ost->sync_opts = lrint(sync_ipts);
1939         break;
1940     case VSYNC_DROP:
1941     case VSYNC_PASSTHROUGH:
1942         ost->sync_opts = lrint(sync_ipts);
1943         break;
1944     default:
1945         av_assert0(0);
1946     }
1947
1948     nb_frames = FFMIN(nb_frames, ost->max_frames - ost->frame_number);
1949     if (nb_frames == 0) {
1950         nb_frames_drop++;
1951         av_log(NULL, AV_LOG_VERBOSE, "*** drop!\n");
1952         return;
1953     } else if (nb_frames > 1) {
1954         nb_frames_dup += nb_frames - 1;
1955         av_log(NULL, AV_LOG_VERBOSE, "*** %d dup!\n", nb_frames - 1);
1956     }
1957
1958
1959 duplicate_frame:
1960     av_init_packet(&pkt);
1961     pkt.data = NULL;
1962     pkt.size = 0;
1963
1964     in_picture->pts = ost->sync_opts;
1965
1966     if (s->oformat->flags & AVFMT_RAWPICTURE &&
1967         enc->codec->id == CODEC_ID_RAWVIDEO) {
1968         /* raw pictures are written as AVPicture structure to
1969            avoid any copies. We support temporarily the older
1970            method. */
1971         enc->coded_frame->interlaced_frame = in_picture->interlaced_frame;
1972         enc->coded_frame->top_field_first  = in_picture->top_field_first;
1973         pkt.data   = (uint8_t *)in_picture;
1974         pkt.size   =  sizeof(AVPicture);
1975         pkt.pts    = av_rescale_q(in_picture->pts, enc->time_base, ost->st->time_base);
1976         pkt.flags |= AV_PKT_FLAG_KEY;
1977
1978         write_frame(s, &pkt, ost);
1979     } else {
1980         int got_packet;
1981         AVFrame big_picture;
1982
1983         big_picture = *in_picture;
1984         /* better than nothing: use input picture interlaced
1985            settings */
1986         big_picture.interlaced_frame = in_picture->interlaced_frame;
1987         if (ost->st->codec->flags & (CODEC_FLAG_INTERLACED_DCT|CODEC_FLAG_INTERLACED_ME)) {
1988             if (ost->top_field_first == -1)
1989                 big_picture.top_field_first = in_picture->top_field_first;
1990             else
1991                 big_picture.top_field_first = !!ost->top_field_first;
1992         }
1993
1994         /* handles same_quant here. This is not correct because it may
1995            not be a global option */
1996         big_picture.quality = quality;
1997         if (!enc->me_threshold)
1998             big_picture.pict_type = 0;
1999         if (ost->forced_kf_index < ost->forced_kf_count &&
2000             big_picture.pts >= ost->forced_kf_pts[ost->forced_kf_index]) {
2001             big_picture.pict_type = AV_PICTURE_TYPE_I;
2002             ost->forced_kf_index++;
2003         }
2004         update_benchmark(NULL);
2005         ret = avcodec_encode_video2(enc, &pkt, &big_picture, &got_packet);
2006         update_benchmark("encode_video %d.%d", ost->file_index, ost->index);
2007         if (ret < 0) {
2008             av_log(NULL, AV_LOG_FATAL, "Video encoding failed\n");
2009             exit_program(1);
2010         }
2011
2012         if (got_packet) {
2013             if (pkt.pts == AV_NOPTS_VALUE && !(enc->codec->capabilities & CODEC_CAP_DELAY))
2014                 pkt.pts = ost->sync_opts;
2015
2016             if (pkt.pts != AV_NOPTS_VALUE)
2017                 pkt.pts = av_rescale_q(pkt.pts, enc->time_base, ost->st->time_base);
2018             if (pkt.dts != AV_NOPTS_VALUE)
2019                 pkt.dts = av_rescale_q(pkt.dts, enc->time_base, ost->st->time_base);
2020
2021             if (debug_ts) {
2022                 av_log(NULL, AV_LOG_INFO, "encoder -> type:video "
2023                     "pkt_pts:%s pkt_pts_time:%s pkt_dts:%s pkt_dts_time:%s\n",
2024                     av_ts2str(pkt.pts), av_ts2timestr(pkt.pts, &ost->st->time_base),
2025                     av_ts2str(pkt.dts), av_ts2timestr(pkt.dts, &ost->st->time_base));
2026             }
2027
2028             write_frame(s, &pkt, ost);
2029             frame_size = pkt.size;
2030             video_size += pkt.size;
2031             av_free_packet(&pkt);
2032
2033             /* if two pass, output log */
2034             if (ost->logfile && enc->stats_out) {
2035                 fprintf(ost->logfile, "%s", enc->stats_out);
2036             }
2037         }
2038     }
2039     ost->sync_opts++;
2040     /*
2041      * For video, number of frames in == number of packets out.
2042      * But there may be reordering, so we can't throw away frames on encoder
2043      * flush, we need to limit them here, before they go into encoder.
2044      */
2045     ost->frame_number++;
2046
2047     if(--nb_frames)
2048         goto duplicate_frame;
2049
2050     if (vstats_filename && frame_size)
2051         do_video_stats(output_files[ost->file_index]->ctx, ost, frame_size);
2052 }
2053
2054 static double psnr(double d)
2055 {
2056     return -10.0 * log(d) / log(10.0);
2057 }
2058
2059 static void do_video_stats(AVFormatContext *os, OutputStream *ost,
2060                            int frame_size)
2061 {
2062     AVCodecContext *enc;
2063     int frame_number;
2064     double ti1, bitrate, avg_bitrate;
2065
2066     /* this is executed just the first time do_video_stats is called */
2067     if (!vstats_file) {
2068         vstats_file = fopen(vstats_filename, "w");
2069         if (!vstats_file) {
2070             perror("fopen");
2071             exit_program(1);
2072         }
2073     }
2074
2075     enc = ost->st->codec;
2076     if (enc->codec_type == AVMEDIA_TYPE_VIDEO) {
2077         frame_number = ost->frame_number;
2078         fprintf(vstats_file, "frame= %5d q= %2.1f ", frame_number, enc->coded_frame->quality / (float)FF_QP2LAMBDA);
2079         if (enc->flags&CODEC_FLAG_PSNR)
2080             fprintf(vstats_file, "PSNR= %6.2f ", psnr(enc->coded_frame->error[0] / (enc->width * enc->height * 255.0 * 255.0)));
2081
2082         fprintf(vstats_file,"f_size= %6d ", frame_size);
2083         /* compute pts value */
2084         ti1 = ost->sync_opts * av_q2d(enc->time_base);
2085         if (ti1 < 0.01)
2086             ti1 = 0.01;
2087
2088         bitrate     = (frame_size * 8) / av_q2d(enc->time_base) / 1000.0;
2089         avg_bitrate = (double)(video_size * 8) / ti1 / 1000.0;
2090         fprintf(vstats_file, "s_size= %8.0fkB time= %0.3f br= %7.1fkbits/s avg_br= %7.1fkbits/s ",
2091                (double)video_size / 1024, ti1, bitrate, avg_bitrate);
2092         fprintf(vstats_file, "type= %c\n", av_get_picture_type_char(enc->coded_frame->pict_type));
2093     }
2094 }
2095
2096 /* check for new output on any of the filtergraphs */
2097 static int poll_filters(void)
2098 {
2099     AVFilterBufferRef *picref;
2100     AVFrame *filtered_frame = NULL;
2101     int i, ret, ret_all;
2102     unsigned nb_success, nb_eof;
2103     int64_t frame_pts;
2104
2105     while (1) {
2106         /* Reap all buffers present in the buffer sinks */
2107         for (i = 0; i < nb_output_streams; i++) {
2108             OutputStream *ost = output_streams[i];
2109             OutputFile    *of = output_files[ost->file_index];
2110             int ret = 0;
2111
2112             if (!ost->filter || ost->is_past_recording_time)
2113                 continue;
2114
2115             if (!ost->filtered_frame && !(ost->filtered_frame = avcodec_alloc_frame())) {
2116                 return AVERROR(ENOMEM);
2117             } else
2118                 avcodec_get_frame_defaults(ost->filtered_frame);
2119             filtered_frame = ost->filtered_frame;
2120
2121             while (1) {
2122                 AVRational ist_pts_tb = ost->filter->filter->inputs[0]->time_base;
2123                 if (ost->enc->type == AVMEDIA_TYPE_AUDIO &&
2124                     !(ost->enc->capabilities & CODEC_CAP_VARIABLE_FRAME_SIZE))
2125                     ret = av_buffersink_read_samples(ost->filter->filter, &picref,
2126                                                     ost->st->codec->frame_size);
2127                 else
2128 #ifdef SINKA
2129                     ret = av_buffersink_read(ost->filter->filter, &picref);
2130 #else
2131                     ret = av_buffersink_get_buffer_ref(ost->filter->filter, &picref,
2132                                                        AV_BUFFERSINK_FLAG_NO_REQUEST);
2133 #endif
2134                 if (ret < 0) {
2135                     if (ret != AVERROR(EAGAIN) && ret != AVERROR_EOF) {
2136                         char buf[256];
2137                         av_strerror(ret, buf, sizeof(buf));
2138                         av_log(NULL, AV_LOG_WARNING,
2139                                "Error in av_buffersink_get_buffer_ref(): %s\n", buf);
2140                     }
2141                     break;
2142                 }
2143                 frame_pts = AV_NOPTS_VALUE;
2144                 if (picref->pts != AV_NOPTS_VALUE) {
2145                     filtered_frame->pts = frame_pts = av_rescale_q(picref->pts,
2146                                                     ost->filter->filter->inputs[0]->time_base,
2147                                                     ost->st->codec->time_base) -
2148                                         av_rescale_q(of->start_time,
2149                                                     AV_TIME_BASE_Q,
2150                                                     ost->st->codec->time_base);
2151
2152                     if (of->start_time && filtered_frame->pts < 0) {
2153                         avfilter_unref_buffer(picref);
2154                         continue;
2155                     }
2156                 }
2157                 //if (ost->source_index >= 0)
2158                 //    *filtered_frame= *input_streams[ost->source_index]->decoded_frame; //for me_threshold
2159
2160
2161                 switch (ost->filter->filter->inputs[0]->type) {
2162                 case AVMEDIA_TYPE_VIDEO:
2163                     avfilter_fill_frame_from_video_buffer_ref(filtered_frame, picref);
2164                     filtered_frame->pts = frame_pts;
2165                     if (!ost->frame_aspect_ratio)
2166                         ost->st->codec->sample_aspect_ratio = picref->video->sample_aspect_ratio;
2167
2168                     do_video_out(of->ctx, ost, filtered_frame,
2169                                  same_quant ? ost->last_quality :
2170                                               ost->st->codec->global_quality);
2171                     break;
2172                 case AVMEDIA_TYPE_AUDIO:
2173                     avfilter_copy_buf_props(filtered_frame, picref);
2174                     filtered_frame->pts = frame_pts;
2175                     do_audio_out(of->ctx, ost, filtered_frame);
2176                     break;
2177                 default:
2178                     // TODO support subtitle filters
2179                     av_assert0(0);
2180                 }
2181
2182                 avfilter_unref_buffer(picref);
2183             }
2184         }
2185         /* Request frames through all the graphs */
2186         ret_all = nb_success = nb_eof = 0;
2187         for (i = 0; i < nb_filtergraphs; i++) {
2188             ret = avfilter_graph_request_oldest(filtergraphs[i]->graph);
2189             if (!ret) {
2190                 nb_success++;
2191             } else if (ret == AVERROR_EOF) {
2192                 nb_eof++;
2193             } else if (ret != AVERROR(EAGAIN)) {
2194                 char buf[256];
2195                 av_strerror(ret, buf, sizeof(buf));
2196                 av_log(NULL, AV_LOG_WARNING,
2197                        "Error in request_frame(): %s\n", buf);
2198                 ret_all = ret;
2199             }
2200         }
2201         if (!nb_success)
2202             break;
2203         /* Try again if anything succeeded */
2204     }
2205     return nb_eof == nb_filtergraphs ? AVERROR_EOF : ret_all;
2206 }
2207
2208 static void print_report(int is_last_report, int64_t timer_start, int64_t cur_time)
2209 {
2210     char buf[1024];
2211     OutputStream *ost;
2212     AVFormatContext *oc;
2213     int64_t total_size;
2214     AVCodecContext *enc;
2215     int frame_number, vid, i;
2216     double bitrate;
2217     int64_t pts = INT64_MAX;
2218     static int64_t last_time = -1;
2219     static int qp_histogram[52];
2220     int hours, mins, secs, us;
2221
2222     if (!print_stats && !is_last_report)
2223         return;
2224
2225     if (!is_last_report) {
2226         if (last_time == -1) {
2227             last_time = cur_time;
2228             return;
2229         }
2230         if ((cur_time - last_time) < 500000)
2231             return;
2232         last_time = cur_time;
2233     }
2234
2235
2236     oc = output_files[0]->ctx;
2237
2238     total_size = avio_size(oc->pb);
2239     if (total_size < 0) { // FIXME improve avio_size() so it works with non seekable output too
2240         total_size = avio_tell(oc->pb);
2241         if (total_size < 0)
2242             total_size = 0;
2243     }
2244
2245     buf[0] = '\0';
2246     vid = 0;
2247     for (i = 0; i < nb_output_streams; i++) {
2248         float q = -1;
2249         ost = output_streams[i];
2250         enc = ost->st->codec;
2251         if (!ost->stream_copy && enc->coded_frame)
2252             q = enc->coded_frame->quality / (float)FF_QP2LAMBDA;
2253         if (vid && enc->codec_type == AVMEDIA_TYPE_VIDEO) {
2254             snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "q=%2.1f ", q);
2255         }
2256         if (!vid && enc->codec_type == AVMEDIA_TYPE_VIDEO) {
2257             float fps, t = (cur_time-timer_start) / 1000000.0;
2258
2259             frame_number = ost->frame_number;
2260             fps = t > 1 ? frame_number / t : 0;
2261             snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "frame=%5d fps=%3.*f q=%3.1f ",
2262                      frame_number, fps < 9.95, fps, q);
2263             if (is_last_report)
2264                 snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "L");
2265             if (qp_hist) {
2266                 int j;
2267                 int qp = lrintf(q);
2268                 if (qp >= 0 && qp < FF_ARRAY_ELEMS(qp_histogram))
2269                     qp_histogram[qp]++;
2270                 for (j = 0; j < 32; j++)
2271                     snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "%X", (int)lrintf(log(qp_histogram[j] + 1) / log(2)));
2272             }
2273             if (enc->flags&CODEC_FLAG_PSNR) {
2274                 int j;
2275                 double error, error_sum = 0;
2276                 double scale, scale_sum = 0;
2277                 char type[3] = { 'Y','U','V' };
2278                 snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "PSNR=");
2279                 for (j = 0; j < 3; j++) {
2280                     if (is_last_report) {
2281                         error = enc->error[j];
2282                         scale = enc->width * enc->height * 255.0 * 255.0 * frame_number;
2283                     } else {
2284                         error = enc->coded_frame->error[j];
2285                         scale = enc->width * enc->height * 255.0 * 255.0;
2286                     }
2287                     if (j)
2288                         scale /= 4;
2289                     error_sum += error;
2290                     scale_sum += scale;
2291                     snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "%c:%2.2f ", type[j], psnr(error / scale));
2292                 }
2293                 snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "*:%2.2f ", psnr(error_sum / scale_sum));
2294             }
2295             vid = 1;
2296         }
2297         /* compute min output value */
2298         pts = FFMIN(pts, av_rescale_q(ost->st->pts.val,
2299                                       ost->st->time_base, AV_TIME_BASE_Q));
2300     }
2301
2302     secs = pts / AV_TIME_BASE;
2303     us = pts % AV_TIME_BASE;
2304     mins = secs / 60;
2305     secs %= 60;
2306     hours = mins / 60;
2307     mins %= 60;
2308
2309     bitrate = pts ? total_size * 8 / (pts / 1000.0) : 0;
2310
2311     snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf),
2312              "size=%8.0fkB time=", total_size / 1024.0);
2313     snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf),
2314              "%02d:%02d:%02d.%02d ", hours, mins, secs,
2315              (100 * us) / AV_TIME_BASE);
2316     snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf),
2317              "bitrate=%6.1fkbits/s", bitrate);
2318
2319     if (nb_frames_dup || nb_frames_drop)
2320         snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), " dup=%d drop=%d",
2321                 nb_frames_dup, nb_frames_drop);
2322
2323     av_log(NULL, AV_LOG_INFO, "%s    \r", buf);
2324
2325     fflush(stderr);
2326
2327     if (is_last_report) {
2328         int64_t raw= audio_size + video_size + extra_size;
2329         av_log(NULL, AV_LOG_INFO, "\n");
2330         av_log(NULL, AV_LOG_INFO, "video:%1.0fkB audio:%1.0fkB global headers:%1.0fkB muxing overhead %f%%\n",
2331                video_size / 1024.0,
2332                audio_size / 1024.0,
2333                extra_size / 1024.0,
2334                100.0 * (total_size - raw) / raw
2335         );
2336         if(video_size + audio_size + extra_size == 0){
2337             av_log(NULL, AV_LOG_WARNING, "Output file is empty, nothing was encoded (check -ss / -t / -frames parameters if used)\n");
2338         }
2339     }
2340 }
2341
2342 static void flush_encoders(void)
2343 {
2344     int i, ret;
2345
2346     for (i = 0; i < nb_output_streams; i++) {
2347         OutputStream   *ost = output_streams[i];
2348         AVCodecContext *enc = ost->st->codec;
2349         AVFormatContext *os = output_files[ost->file_index]->ctx;
2350         int stop_encoding = 0;
2351
2352         if (!ost->encoding_needed)
2353             continue;
2354
2355         if (ost->st->codec->codec_type == AVMEDIA_TYPE_AUDIO && enc->frame_size <= 1)
2356             continue;
2357         if (ost->st->codec->codec_type == AVMEDIA_TYPE_VIDEO && (os->oformat->flags & AVFMT_RAWPICTURE) && enc->codec->id == CODEC_ID_RAWVIDEO)
2358             continue;
2359
2360         for (;;) {
2361             int (*encode)(AVCodecContext*, AVPacket*, const AVFrame*, int*) = NULL;
2362             const char *desc;
2363             int64_t *size;
2364
2365             switch (ost->st->codec->codec_type) {
2366             case AVMEDIA_TYPE_AUDIO:
2367                 encode = avcodec_encode_audio2;
2368                 desc   = "Audio";
2369                 size   = &audio_size;
2370                 break;
2371             case AVMEDIA_TYPE_VIDEO:
2372                 encode = avcodec_encode_video2;
2373                 desc   = "Video";
2374                 size   = &video_size;
2375                 break;
2376             default:
2377                 stop_encoding = 1;
2378             }
2379
2380             if (encode) {
2381                 AVPacket pkt;
2382                 int got_packet;
2383                 av_init_packet(&pkt);
2384                 pkt.data = NULL;
2385                 pkt.size = 0;
2386
2387                 update_benchmark(NULL);
2388                 ret = encode(enc, &pkt, NULL, &got_packet);
2389                 update_benchmark("flush %s %d.%d", desc, ost->file_index, ost->index);
2390                 if (ret < 0) {
2391                     av_log(NULL, AV_LOG_FATAL, "%s encoding failed\n", desc);
2392                     exit_program(1);
2393                 }
2394                 *size += pkt.size;
2395                 if (ost->logfile && enc->stats_out) {
2396                     fprintf(ost->logfile, "%s", enc->stats_out);
2397                 }
2398                 if (!got_packet) {
2399                     stop_encoding = 1;
2400                     break;
2401                 }
2402                 if (pkt.pts != AV_NOPTS_VALUE)
2403                     pkt.pts = av_rescale_q(pkt.pts, enc->time_base, ost->st->time_base);
2404                 if (pkt.dts != AV_NOPTS_VALUE)
2405                     pkt.dts = av_rescale_q(pkt.dts, enc->time_base, ost->st->time_base);
2406                 write_frame(os, &pkt, ost);
2407             }
2408
2409             if (stop_encoding)
2410                 break;
2411         }
2412     }
2413 }
2414
2415 /*
2416  * Check whether a packet from ist should be written into ost at this time
2417  */
2418 static int check_output_constraints(InputStream *ist, OutputStream *ost)
2419 {
2420     OutputFile *of = output_files[ost->file_index];
2421     int ist_index  = input_files[ist->file_index]->ist_index + ist->st->index;
2422
2423     if (ost->source_index != ist_index)
2424         return 0;
2425
2426     if (of->start_time && ist->pts < of->start_time)
2427         return 0;
2428
2429     if (of->recording_time != INT64_MAX &&
2430         av_compare_ts(ist->pts, AV_TIME_BASE_Q, of->recording_time + of->start_time,
2431                       (AVRational){ 1, 1000000 }) >= 0) {
2432         ost->is_past_recording_time = 1;
2433         return 0;
2434     }
2435
2436     return 1;
2437 }
2438
2439 static void do_streamcopy(InputStream *ist, OutputStream *ost, const AVPacket *pkt)
2440 {
2441     OutputFile *of = output_files[ost->file_index];
2442     int64_t ost_tb_start_time = av_rescale_q(of->start_time, AV_TIME_BASE_Q, ost->st->time_base);
2443     AVPicture pict;
2444     AVPacket opkt;
2445
2446     av_init_packet(&opkt);
2447
2448     if ((!ost->frame_number && !(pkt->flags & AV_PKT_FLAG_KEY)) &&
2449         !ost->copy_initial_nonkeyframes)
2450         return;
2451
2452     /* force the input stream PTS */
2453     if (ost->st->codec->codec_type == AVMEDIA_TYPE_AUDIO)
2454         audio_size += pkt->size;
2455     else if (ost->st->codec->codec_type == AVMEDIA_TYPE_VIDEO) {
2456         video_size += pkt->size;
2457         ost->sync_opts++;
2458     }
2459
2460     if (pkt->pts != AV_NOPTS_VALUE)
2461         opkt.pts = av_rescale_q(pkt->pts, ist->st->time_base, ost->st->time_base) - ost_tb_start_time;
2462     else
2463         opkt.pts = AV_NOPTS_VALUE;
2464
2465     if (pkt->dts == AV_NOPTS_VALUE)
2466         opkt.dts = av_rescale_q(ist->dts, AV_TIME_BASE_Q, ost->st->time_base);
2467     else
2468         opkt.dts = av_rescale_q(pkt->dts, ist->st->time_base, ost->st->time_base);
2469     opkt.dts -= ost_tb_start_time;
2470
2471     opkt.duration = av_rescale_q(pkt->duration, ist->st->time_base, ost->st->time_base);
2472     opkt.flags    = pkt->flags;
2473
2474     // FIXME remove the following 2 lines they shall be replaced by the bitstream filters
2475     if (  ost->st->codec->codec_id != CODEC_ID_H264
2476        && ost->st->codec->codec_id != CODEC_ID_MPEG1VIDEO
2477        && ost->st->codec->codec_id != CODEC_ID_MPEG2VIDEO
2478        && ost->st->codec->codec_id != CODEC_ID_VC1
2479        ) {
2480         if (av_parser_change(ist->st->parser, ost->st->codec, &opkt.data, &opkt.size, pkt->data, pkt->size, pkt->flags & AV_PKT_FLAG_KEY))
2481             opkt.destruct = av_destruct_packet;
2482     } else {
2483         opkt.data = pkt->data;
2484         opkt.size = pkt->size;
2485     }
2486     if (of->ctx->oformat->flags & AVFMT_RAWPICTURE) {
2487         /* store AVPicture in AVPacket, as expected by the output format */
2488         avpicture_fill(&pict, opkt.data, ost->st->codec->pix_fmt, ost->st->codec->width, ost->st->codec->height);
2489         opkt.data = (uint8_t *)&pict;
2490         opkt.size = sizeof(AVPicture);
2491         opkt.flags |= AV_PKT_FLAG_KEY;
2492     }
2493
2494     write_frame(of->ctx, &opkt, ost);
2495     ost->st->codec->frame_number++;
2496     av_free_packet(&opkt);
2497 }
2498
2499 static void rate_emu_sleep(InputStream *ist)
2500 {
2501     if (input_files[ist->file_index]->rate_emu) {
2502         int64_t pts = av_rescale(ist->dts, 1000000, AV_TIME_BASE);
2503         int64_t now = av_gettime() - ist->start;
2504         if (pts > now)
2505             usleep(pts - now);
2506     }
2507 }
2508
2509 static int guess_input_channel_layout(InputStream *ist)
2510 {
2511     AVCodecContext *dec = ist->st->codec;
2512
2513     if (!dec->channel_layout) {
2514         char layout_name[256];
2515
2516         dec->channel_layout = av_get_default_channel_layout(dec->channels);
2517         if (!dec->channel_layout)
2518             return 0;
2519         av_get_channel_layout_string(layout_name, sizeof(layout_name),
2520                                      dec->channels, dec->channel_layout);
2521         av_log(NULL, AV_LOG_WARNING, "Guessed Channel Layout for  Input Stream "
2522                "#%d.%d : %s\n", ist->file_index, ist->st->index, layout_name);
2523     }
2524     return 1;
2525 }
2526
2527 static int decode_audio(InputStream *ist, AVPacket *pkt, int *got_output)
2528 {
2529     AVFrame *decoded_frame;
2530     AVCodecContext *avctx = ist->st->codec;
2531     int i, ret, resample_changed;
2532
2533     if (!ist->decoded_frame && !(ist->decoded_frame = avcodec_alloc_frame()))
2534         return AVERROR(ENOMEM);
2535     else
2536         avcodec_get_frame_defaults(ist->decoded_frame);
2537     decoded_frame = ist->decoded_frame;
2538
2539     update_benchmark(NULL);
2540     ret = avcodec_decode_audio4(avctx, decoded_frame, got_output, pkt);
2541     update_benchmark("decode_audio %d.%d", ist->file_index, ist->st->index);
2542     if (ret < 0) {
2543         return ret;
2544     }
2545     if (avctx->sample_rate <= 0) {
2546         av_log(avctx, AV_LOG_ERROR, "Sample rate %d invalid\n", avctx->sample_rate);
2547         return AVERROR_INVALIDDATA;
2548     }
2549
2550     if (!*got_output) {
2551         /* no audio frame */
2552         if (!pkt->size)
2553             for (i = 0; i < ist->nb_filters; i++)
2554                 av_buffersrc_add_ref(ist->filters[i]->filter, NULL,
2555                                      AV_BUFFERSRC_FLAG_NO_COPY);
2556         return ret;
2557     }
2558
2559     /* if the decoder provides a pts, use it instead of the last packet pts.
2560        the decoder could be delaying output by a packet or more. */
2561     if (decoded_frame->pts != AV_NOPTS_VALUE)
2562         ist->dts = ist->next_dts = ist->pts = ist->next_pts = decoded_frame->pts;
2563     else if (pkt->pts != AV_NOPTS_VALUE) {
2564         decoded_frame->pts = pkt->pts;
2565         pkt->pts           = AV_NOPTS_VALUE;
2566     }else
2567         decoded_frame->pts = av_rescale_q(ist->dts, AV_TIME_BASE_Q, ist->st->time_base);
2568
2569
2570 #if 1
2571     /* increment next_dts to use for the case where the input stream does not
2572        have timestamps or there are multiple frames in the packet */
2573     ist->next_pts += ((int64_t)AV_TIME_BASE * decoded_frame->nb_samples) /
2574                      avctx->sample_rate;
2575     ist->next_dts += ((int64_t)AV_TIME_BASE * decoded_frame->nb_samples) /
2576                      avctx->sample_rate;
2577 #endif
2578
2579     rate_emu_sleep(ist);
2580
2581     resample_changed = ist->resample_sample_fmt     != decoded_frame->format         ||
2582                        ist->resample_channels       != avctx->channels               ||
2583                        ist->resample_channel_layout != decoded_frame->channel_layout ||
2584                        ist->resample_sample_rate    != decoded_frame->sample_rate;
2585     if (resample_changed) {
2586         char layout1[64], layout2[64];
2587
2588         if (!guess_input_channel_layout(ist)) {
2589             av_log(NULL, AV_LOG_FATAL, "Unable to find default channel "
2590                    "layout for Input Stream #%d.%d\n", ist->file_index,
2591                    ist->st->index);
2592             exit_program(1);
2593         }
2594         decoded_frame->channel_layout = avctx->channel_layout;
2595
2596         av_get_channel_layout_string(layout1, sizeof(layout1), ist->resample_channels,
2597                                      ist->resample_channel_layout);
2598         av_get_channel_layout_string(layout2, sizeof(layout2), avctx->channels,
2599                                      decoded_frame->channel_layout);
2600
2601         av_log(NULL, AV_LOG_INFO,
2602                "Input stream #%d:%d frame changed from rate:%d fmt:%s ch:%d chl:%s to rate:%d fmt:%s ch:%d chl:%s\n",
2603                ist->file_index, ist->st->index,
2604                ist->resample_sample_rate,  av_get_sample_fmt_name(ist->resample_sample_fmt),
2605                ist->resample_channels, layout1,
2606                decoded_frame->sample_rate, av_get_sample_fmt_name(decoded_frame->format),
2607                avctx->channels, layout2);
2608
2609         ist->resample_sample_fmt     = decoded_frame->format;
2610         ist->resample_sample_rate    = decoded_frame->sample_rate;
2611         ist->resample_channel_layout = decoded_frame->channel_layout;
2612         ist->resample_channels       = avctx->channels;
2613
2614         for (i = 0; i < nb_filtergraphs; i++)
2615             if (ist_in_filtergraph(filtergraphs[i], ist) &&
2616                 configure_filtergraph(filtergraphs[i]) < 0) {
2617                 av_log(NULL, AV_LOG_FATAL, "Error reinitializing filters!\n");
2618                 exit_program(1);
2619             }
2620     }
2621
2622     for (i = 0; i < ist->nb_filters; i++)
2623         av_buffersrc_add_frame(ist->filters[i]->filter, decoded_frame, 0);
2624
2625     return ret;
2626 }
2627
2628 static int decode_video(InputStream *ist, AVPacket *pkt, int *got_output)
2629 {
2630     AVFrame *decoded_frame;
2631     void *buffer_to_free = NULL;
2632     int i, ret = 0, resample_changed;
2633     int64_t best_effort_timestamp;
2634     AVRational *frame_sample_aspect;
2635     float quality;
2636
2637     if (!ist->decoded_frame && !(ist->decoded_frame = avcodec_alloc_frame()))
2638         return AVERROR(ENOMEM);
2639     else
2640         avcodec_get_frame_defaults(ist->decoded_frame);
2641     decoded_frame = ist->decoded_frame;
2642     pkt->dts  = av_rescale_q(ist->dts, AV_TIME_BASE_Q, ist->st->time_base);
2643
2644     update_benchmark(NULL);
2645     ret = avcodec_decode_video2(ist->st->codec,
2646                                 decoded_frame, got_output, pkt);
2647     update_benchmark("decode_video %d.%d", ist->file_index, ist->st->index);
2648     if (ret < 0)
2649         return ret;
2650
2651     quality = same_quant ? decoded_frame->quality : 0;
2652     if (!*got_output) {
2653         /* no picture yet */
2654         if (!pkt->size)
2655             for (i = 0; i < ist->nb_filters; i++)
2656                 av_buffersrc_add_ref(ist->filters[i]->filter, NULL, AV_BUFFERSRC_FLAG_NO_COPY);
2657         return ret;
2658     }
2659
2660     best_effort_timestamp= av_frame_get_best_effort_timestamp(decoded_frame);
2661     if(best_effort_timestamp != AV_NOPTS_VALUE)
2662         ist->next_pts = ist->pts = av_rescale_q(decoded_frame->pts = best_effort_timestamp, ist->st->time_base, AV_TIME_BASE_Q);
2663
2664     pkt->size = 0;
2665     pre_process_video_frame(ist, (AVPicture *)decoded_frame, &buffer_to_free);
2666
2667     rate_emu_sleep(ist);
2668
2669     if (ist->st->sample_aspect_ratio.num)
2670         decoded_frame->sample_aspect_ratio = ist->st->sample_aspect_ratio;
2671
2672     resample_changed = ist->resample_width   != decoded_frame->width  ||
2673                        ist->resample_height  != decoded_frame->height ||
2674                        ist->resample_pix_fmt != decoded_frame->format;
2675     if (resample_changed) {
2676         av_log(NULL, AV_LOG_INFO,
2677                "Input stream #%d:%d frame changed from size:%dx%d fmt:%s to size:%dx%d fmt:%s\n",
2678                ist->file_index, ist->st->index,
2679                ist->resample_width,  ist->resample_height,  av_get_pix_fmt_name(ist->resample_pix_fmt),
2680                decoded_frame->width, decoded_frame->height, av_get_pix_fmt_name(decoded_frame->format));
2681
2682         ist->resample_width   = decoded_frame->width;
2683         ist->resample_height  = decoded_frame->height;
2684         ist->resample_pix_fmt = decoded_frame->format;
2685
2686         for (i = 0; i < nb_filtergraphs; i++)
2687             if (ist_in_filtergraph(filtergraphs[i], ist) &&
2688                 configure_filtergraph(filtergraphs[i]) < 0) {
2689                 av_log(NULL, AV_LOG_FATAL, "Error reinitializing filters!\n");
2690                 exit_program(1);
2691             }
2692     }
2693
2694     frame_sample_aspect= av_opt_ptr(avcodec_get_frame_class(), decoded_frame, "sample_aspect_ratio");
2695     for (i = 0; i < ist->nb_filters; i++) {
2696         int changed =      ist->st->codec->width   != ist->filters[i]->filter->outputs[0]->w
2697                         || ist->st->codec->height  != ist->filters[i]->filter->outputs[0]->h
2698                         || ist->st->codec->pix_fmt != ist->filters[i]->filter->outputs[0]->format;
2699         // XXX what an ugly hack
2700         if (ist->filters[i]->graph->nb_outputs == 1)
2701             ist->filters[i]->graph->outputs[0]->ost->last_quality = quality;
2702
2703         if (!frame_sample_aspect->num)
2704             *frame_sample_aspect = ist->st->sample_aspect_ratio;
2705         if (ist->dr1 && decoded_frame->type==FF_BUFFER_TYPE_USER && !changed) {
2706             FrameBuffer      *buf = decoded_frame->opaque;
2707             AVFilterBufferRef *fb = avfilter_get_video_buffer_ref_from_arrays(
2708                                         decoded_frame->data, decoded_frame->linesize,
2709                                         AV_PERM_READ | AV_PERM_PRESERVE,
2710                                         ist->st->codec->width, ist->st->codec->height,
2711                                         ist->st->codec->pix_fmt);
2712
2713             avfilter_copy_frame_props(fb, decoded_frame);
2714             fb->buf->priv           = buf;
2715             fb->buf->free           = filter_release_buffer;
2716
2717             av_assert0(buf->refcount>0);
2718             buf->refcount++;
2719             av_buffersrc_add_ref(ist->filters[i]->filter, fb,
2720                                  AV_BUFFERSRC_FLAG_NO_CHECK_FORMAT |
2721                                  AV_BUFFERSRC_FLAG_NO_COPY);
2722         } else
2723         if(av_buffersrc_add_frame(ist->filters[i]->filter, decoded_frame, 0)<0) {
2724             av_log(NULL, AV_LOG_FATAL, "Failed to inject frame into filter network\n");
2725             exit_program(1);
2726         }
2727
2728     }
2729
2730     av_free(buffer_to_free);
2731     return ret;
2732 }
2733
2734 static int transcode_subtitles(InputStream *ist, AVPacket *pkt, int *got_output)
2735 {
2736     AVSubtitle subtitle;
2737     int i, ret = avcodec_decode_subtitle2(ist->st->codec,
2738                                           &subtitle, got_output, pkt);
2739     if (ret < 0)
2740         return ret;
2741     if (!*got_output)
2742         return ret;
2743
2744     rate_emu_sleep(ist);
2745
2746     for (i = 0; i < nb_output_streams; i++) {
2747         OutputStream *ost = output_streams[i];
2748
2749         if (!check_output_constraints(ist, ost) || !ost->encoding_needed)
2750             continue;
2751
2752         do_subtitle_out(output_files[ost->file_index]->ctx, ost, ist, &subtitle, pkt->pts);
2753     }
2754
2755     avsubtitle_free(&subtitle);
2756     return ret;
2757 }
2758
2759 /* pkt = NULL means EOF (needed to flush decoder buffers) */
2760 static int output_packet(InputStream *ist, const AVPacket *pkt)
2761 {
2762     int ret = 0, i;
2763     int got_output;
2764
2765     AVPacket avpkt;
2766     if (!ist->saw_first_ts) {
2767         ist->dts = ist->st->avg_frame_rate.num ? - ist->st->codec->has_b_frames * AV_TIME_BASE / av_q2d(ist->st->avg_frame_rate) : 0;
2768         ist->pts = 0;
2769         if (pkt != NULL && pkt->pts != AV_NOPTS_VALUE && !ist->decoding_needed) {
2770             ist->dts += av_rescale_q(pkt->pts, ist->st->time_base, AV_TIME_BASE_Q);
2771             ist->pts = ist->dts; //unused but better to set it to a value thats not totally wrong
2772         }
2773         ist->saw_first_ts = 1;
2774     }
2775
2776     if (ist->next_dts == AV_NOPTS_VALUE)
2777         ist->next_dts = ist->dts;
2778     if (ist->next_pts == AV_NOPTS_VALUE)
2779         ist->next_pts = ist->pts;
2780
2781     if (pkt == NULL) {
2782         /* EOF handling */
2783         av_init_packet(&avpkt);
2784         avpkt.data = NULL;
2785         avpkt.size = 0;
2786         goto handle_eof;
2787     } else {
2788         avpkt = *pkt;
2789     }
2790
2791     if (pkt->dts != AV_NOPTS_VALUE) {
2792         ist->next_dts = ist->dts = av_rescale_q(pkt->dts, ist->st->time_base, AV_TIME_BASE_Q);
2793         if (ist->st->codec->codec_type != AVMEDIA_TYPE_VIDEO || !ist->decoding_needed)
2794             ist->next_pts = ist->pts = av_rescale_q(pkt->dts, ist->st->time_base, AV_TIME_BASE_Q);
2795     }
2796
2797     // while we have more to decode or while the decoder did output something on EOF
2798     while (ist->decoding_needed && (avpkt.size > 0 || (!pkt && got_output))) {
2799         int duration;
2800     handle_eof:
2801
2802         ist->pts = ist->next_pts;
2803         ist->dts = ist->next_dts;
2804
2805         if (avpkt.size && avpkt.size != pkt->size) {
2806             av_log(NULL, ist->showed_multi_packet_warning ? AV_LOG_VERBOSE : AV_LOG_WARNING,
2807                    "Multiple frames in a packet from stream %d\n", pkt->stream_index);
2808             ist->showed_multi_packet_warning = 1;
2809         }
2810
2811         switch (ist->st->codec->codec_type) {
2812         case AVMEDIA_TYPE_AUDIO:
2813             ret = decode_audio    (ist, &avpkt, &got_output);
2814             break;
2815         case AVMEDIA_TYPE_VIDEO:
2816             ret = decode_video    (ist, &avpkt, &got_output);
2817             if (avpkt.duration) {
2818                 duration = av_rescale_q(avpkt.duration, ist->st->time_base, AV_TIME_BASE_Q);
2819             } else if(ist->st->codec->time_base.num != 0 && ist->st->codec->time_base.den != 0) {
2820                 int ticks= ist->st->parser ? ist->st->parser->repeat_pict+1 : ist->st->codec->ticks_per_frame;
2821                 duration = ((int64_t)AV_TIME_BASE *
2822                                 ist->st->codec->time_base.num * ticks) /
2823                                 ist->st->codec->time_base.den;
2824             } else
2825                 duration = 0;
2826
2827             if(ist->dts != AV_NOPTS_VALUE && duration) {
2828                 ist->next_dts += duration;
2829             }else
2830                 ist->next_dts = AV_NOPTS_VALUE;
2831
2832             if (got_output)
2833                 ist->next_pts += duration; //FIXME the duration is not correct in some cases
2834             break;
2835         case AVMEDIA_TYPE_SUBTITLE:
2836             ret = transcode_subtitles(ist, &avpkt, &got_output);
2837             break;
2838         default:
2839             return -1;
2840         }
2841
2842         if (ret < 0)
2843             return ret;
2844
2845         avpkt.dts=
2846         avpkt.pts= AV_NOPTS_VALUE;
2847
2848         // touch data and size only if not EOF
2849         if (pkt) {
2850             if(ist->st->codec->codec_type != AVMEDIA_TYPE_AUDIO)
2851                 ret = avpkt.size;
2852             avpkt.data += ret;
2853             avpkt.size -= ret;
2854         }
2855         if (!got_output) {
2856             continue;
2857         }
2858     }
2859
2860     /* handle stream copy */
2861     if (!ist->decoding_needed) {
2862         rate_emu_sleep(ist);
2863         ist->dts = ist->next_dts;
2864         switch (ist->st->codec->codec_type) {
2865         case AVMEDIA_TYPE_AUDIO:
2866             ist->next_dts += ((int64_t)AV_TIME_BASE * ist->st->codec->frame_size) /
2867                              ist->st->codec->sample_rate;
2868             break;
2869         case AVMEDIA_TYPE_VIDEO:
2870             if (pkt->duration) {
2871                 ist->next_dts += av_rescale_q(pkt->duration, ist->st->time_base, AV_TIME_BASE_Q);
2872             } else if(ist->st->codec->time_base.num != 0) {
2873                 int ticks= ist->st->parser ? ist->st->parser->repeat_pict + 1 : ist->st->codec->ticks_per_frame;
2874                 ist->next_dts += ((int64_t)AV_TIME_BASE *
2875                                   ist->st->codec->time_base.num * ticks) /
2876                                   ist->st->codec->time_base.den;
2877             }
2878             break;
2879         }
2880         ist->pts = ist->dts;
2881         ist->next_pts = ist->next_dts;
2882     }
2883     for (i = 0; pkt && i < nb_output_streams; i++) {
2884         OutputStream *ost = output_streams[i];
2885
2886         if (!check_output_constraints(ist, ost) || ost->encoding_needed)
2887             continue;
2888
2889         do_streamcopy(ist, ost, pkt);
2890     }
2891
2892     return 0;
2893 }
2894
2895 static void print_sdp(void)
2896 {
2897     char sdp[2048];
2898     int i;
2899     AVFormatContext **avc = av_malloc(sizeof(*avc) * nb_output_files);
2900
2901     if (!avc)
2902         exit_program(1);
2903     for (i = 0; i < nb_output_files; i++)
2904         avc[i] = output_files[i]->ctx;
2905
2906     av_sdp_create(avc, nb_output_files, sdp, sizeof(sdp));
2907     printf("SDP:\n%s\n", sdp);
2908     fflush(stdout);
2909     av_freep(&avc);
2910 }
2911
2912 static int init_input_stream(int ist_index, char *error, int error_len)
2913 {
2914     InputStream *ist = input_streams[ist_index];
2915
2916     if (ist->decoding_needed) {
2917         AVCodec *codec = ist->dec;
2918         if (!codec) {
2919             snprintf(error, error_len, "Decoder (codec %s) not found for input stream #%d:%d",
2920                     avcodec_get_name(ist->st->codec->codec_id), ist->file_index, ist->st->index);
2921             return AVERROR(EINVAL);
2922         }
2923
2924         ist->dr1 = (codec->capabilities & CODEC_CAP_DR1) && !do_deinterlace;
2925         if (codec->type == AVMEDIA_TYPE_VIDEO && ist->dr1) {
2926             ist->st->codec->get_buffer     = codec_get_buffer;
2927             ist->st->codec->release_buffer = codec_release_buffer;
2928             ist->st->codec->opaque         = ist;
2929         }
2930
2931         if (!av_dict_get(ist->opts, "threads", NULL, 0))
2932             av_dict_set(&ist->opts, "threads", "auto", 0);
2933         if (avcodec_open2(ist->st->codec, codec, &ist->opts) < 0) {
2934             snprintf(error, error_len, "Error while opening decoder for input stream #%d:%d",
2935                     ist->file_index, ist->st->index);
2936             return AVERROR(EINVAL);
2937         }
2938         assert_codec_experimental(ist->st->codec, 0);
2939         assert_avoptions(ist->opts);
2940     }
2941
2942     ist->next_pts = AV_NOPTS_VALUE;
2943     ist->next_dts = AV_NOPTS_VALUE;
2944     ist->is_start = 1;
2945
2946     return 0;
2947 }
2948
2949 static InputStream *get_input_stream(OutputStream *ost)
2950 {
2951     if (ost->source_index >= 0)
2952         return input_streams[ost->source_index];
2953
2954     if (ost->filter) {
2955         FilterGraph *fg = ost->filter->graph;
2956         int i;
2957
2958         for (i = 0; i < fg->nb_inputs; i++)
2959             if (fg->inputs[i]->ist->st->codec->codec_type == ost->st->codec->codec_type)
2960                 return fg->inputs[i]->ist;
2961     }
2962
2963     return NULL;
2964 }
2965
2966 static int transcode_init(void)
2967 {
2968     int ret = 0, i, j, k;
2969     AVFormatContext *oc;
2970     AVCodecContext *codec, *icodec;
2971     OutputStream *ost;
2972     InputStream *ist;
2973     char error[1024];
2974     int want_sdp = 1;
2975
2976     /* init framerate emulation */
2977     for (i = 0; i < nb_input_files; i++) {
2978         InputFile *ifile = input_files[i];
2979         if (ifile->rate_emu)
2980             for (j = 0; j < ifile->nb_streams; j++)
2981                 input_streams[j + ifile->ist_index]->start = av_gettime();
2982     }
2983
2984     /* output stream init */
2985     for (i = 0; i < nb_output_files; i++) {
2986         oc = output_files[i]->ctx;
2987         if (!oc->nb_streams && !(oc->oformat->flags & AVFMT_NOSTREAMS)) {
2988             av_dump_format(oc, i, oc->filename, 1);
2989             av_log(NULL, AV_LOG_ERROR, "Output file #%d does not contain any stream\n", i);
2990             return AVERROR(EINVAL);
2991         }
2992     }
2993
2994     /* init complex filtergraphs */
2995     for (i = 0; i < nb_filtergraphs; i++)
2996         if ((ret = avfilter_graph_config(filtergraphs[i]->graph, NULL)) < 0)
2997             return ret;
2998
2999     /* for each output stream, we compute the right encoding parameters */
3000     for (i = 0; i < nb_output_streams; i++) {
3001         ost = output_streams[i];
3002         oc  = output_files[ost->file_index]->ctx;
3003         ist = get_input_stream(ost);
3004
3005         if (ost->attachment_filename)
3006             continue;
3007
3008         codec  = ost->st->codec;
3009
3010         if (ist) {
3011             icodec = ist->st->codec;
3012
3013             ost->st->disposition          = ist->st->disposition;
3014             codec->bits_per_raw_sample    = icodec->bits_per_raw_sample;
3015             codec->chroma_sample_location = icodec->chroma_sample_location;
3016         }
3017
3018         if (ost->stream_copy) {
3019             uint64_t extra_size;
3020
3021             av_assert0(ist && !ost->filter);
3022
3023             extra_size = (uint64_t)icodec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE;
3024
3025             if (extra_size > INT_MAX) {
3026                 return AVERROR(EINVAL);
3027             }
3028
3029             /* if stream_copy is selected, no need to decode or encode */
3030             codec->codec_id   = icodec->codec_id;
3031             codec->codec_type = icodec->codec_type;
3032
3033             if (!codec->codec_tag) {
3034                 if (!oc->oformat->codec_tag ||
3035                      av_codec_get_id (oc->oformat->codec_tag, icodec->codec_tag) == codec->codec_id ||
3036                      av_codec_get_tag(oc->oformat->codec_tag, icodec->codec_id) <= 0)
3037                     codec->codec_tag = icodec->codec_tag;
3038             }
3039
3040             codec->bit_rate       = icodec->bit_rate;
3041             codec->rc_max_rate    = icodec->rc_max_rate;
3042             codec->rc_buffer_size = icodec->rc_buffer_size;
3043             codec->field_order    = icodec->field_order;
3044             codec->extradata      = av_mallocz(extra_size);
3045             if (!codec->extradata) {
3046                 return AVERROR(ENOMEM);
3047             }
3048             memcpy(codec->extradata, icodec->extradata, icodec->extradata_size);
3049             codec->extradata_size= icodec->extradata_size;
3050             codec->bits_per_coded_sample  = icodec->bits_per_coded_sample;
3051
3052             codec->time_base = ist->st->time_base;
3053             /*
3054              * Avi is a special case here because it supports variable fps but
3055              * having the fps and timebase differe significantly adds quite some
3056              * overhead
3057              */
3058             if(!strcmp(oc->oformat->name, "avi")) {
3059                 if (   copy_tb<0 && av_q2d(icodec->time_base)*icodec->ticks_per_frame > 2*av_q2d(ist->st->time_base)
3060                                  && av_q2d(ist->st->time_base) < 1.0/500
3061                     || copy_tb==0){
3062                     codec->time_base = icodec->time_base;
3063                     codec->time_base.num *= icodec->ticks_per_frame;
3064                     codec->time_base.den *= 2;
3065                     codec->ticks_per_frame = 2;
3066                 }
3067             } else if(!(oc->oformat->flags & AVFMT_VARIABLE_FPS)
3068                       && strcmp(oc->oformat->name, "mov") && strcmp(oc->oformat->name, "mp4") && strcmp(oc->oformat->name, "3gp")
3069                       && strcmp(oc->oformat->name, "3g2") && strcmp(oc->oformat->name, "psp") && strcmp(oc->oformat->name, "ipod")
3070             ) {
3071                 if(   copy_tb<0 && av_q2d(icodec->time_base)*icodec->ticks_per_frame > av_q2d(ist->st->time_base)
3072                                 && av_q2d(ist->st->time_base) < 1.0/500
3073                    || copy_tb==0){
3074                     codec->time_base = icodec->time_base;
3075                     codec->time_base.num *= icodec->ticks_per_frame;
3076                 }
3077             }
3078             av_reduce(&codec->time_base.num, &codec->time_base.den,
3079                         codec->time_base.num, codec->time_base.den, INT_MAX);
3080
3081             switch (codec->codec_type) {
3082             case AVMEDIA_TYPE_AUDIO:
3083                 if (audio_volume != 256) {
3084                     av_log(NULL, AV_LOG_FATAL, "-acodec copy and -vol are incompatible (frames are not decoded)\n");
3085                     exit_program(1);
3086                 }
3087                 codec->channel_layout     = icodec->channel_layout;
3088                 codec->sample_rate        = icodec->sample_rate;
3089                 codec->channels           = icodec->channels;
3090                 codec->frame_size         = icodec->frame_size;
3091                 codec->audio_service_type = icodec->audio_service_type;
3092                 codec->block_align        = icodec->block_align;
3093                 if(codec->block_align == 1 && codec->codec_id == CODEC_ID_MP3)
3094                     codec->block_align= 0;
3095                 if(codec->codec_id == CODEC_ID_AC3)
3096                     codec->block_align= 0;
3097                 break;
3098             case AVMEDIA_TYPE_VIDEO:
3099                 codec->pix_fmt            = icodec->pix_fmt;
3100                 codec->width              = icodec->width;
3101                 codec->height             = icodec->height;
3102                 codec->has_b_frames       = icodec->has_b_frames;
3103                 if (!codec->sample_aspect_ratio.num) {
3104                     codec->sample_aspect_ratio   =
3105                     ost->st->sample_aspect_ratio =
3106                         ist->st->sample_aspect_ratio.num ? ist->st->sample_aspect_ratio :
3107                         ist->st->codec->sample_aspect_ratio.num ?
3108                         ist->st->codec->sample_aspect_ratio : (AVRational){0, 1};
3109                 }
3110                 ost->st->avg_frame_rate = ist->st->avg_frame_rate;
3111                 break;
3112             case AVMEDIA_TYPE_SUBTITLE:
3113                 codec->width  = icodec->width;
3114                 codec->height = icodec->height;
3115                 break;
3116             case AVMEDIA_TYPE_DATA:
3117             case AVMEDIA_TYPE_ATTACHMENT:
3118                 break;
3119             default:
3120                 abort();
3121             }
3122         } else {
3123             if (!ost->enc)
3124                 ost->enc = avcodec_find_encoder(codec->codec_id);
3125             if (!ost->enc) {
3126                 /* should only happen when a default codec is not present. */
3127                 snprintf(error, sizeof(error), "Encoder (codec %s) not found for output stream #%d:%d",
3128                          avcodec_get_name(ost->st->codec->codec_id), ost->file_index, ost->index);
3129                 ret = AVERROR(EINVAL);
3130                 goto dump_format;
3131             }
3132
3133             if (ist)
3134                 ist->decoding_needed = 1;
3135             ost->encoding_needed = 1;
3136
3137             if (codec->codec_type == AVMEDIA_TYPE_VIDEO) {
3138                 if (ist && !ost->frame_rate.num)
3139                     ost->frame_rate = ist->st->r_frame_rate.num ? ist->st->r_frame_rate : (AVRational){25, 1};
3140                 if (ost->enc && ost->enc->supported_framerates && !ost->force_fps) {
3141                     int idx = av_find_nearest_q_idx(ost->frame_rate, ost->enc->supported_framerates);
3142                     ost->frame_rate = ost->enc->supported_framerates[idx];
3143                 }
3144             }
3145
3146             if (!ost->filter &&
3147                 (codec->codec_type == AVMEDIA_TYPE_VIDEO ||
3148                  codec->codec_type == AVMEDIA_TYPE_AUDIO)) {
3149                     FilterGraph *fg;
3150                     fg = init_simple_filtergraph(ist, ost);
3151                     if (configure_simple_filtergraph(fg)) {
3152                         av_log(NULL, AV_LOG_FATAL, "Error opening filters!\n");
3153                         exit(1);
3154                     }
3155             }
3156
3157             switch (codec->codec_type) {
3158             case AVMEDIA_TYPE_AUDIO:
3159                 codec->sample_fmt     = ost->filter->filter->inputs[0]->format;
3160                 codec->sample_rate    = ost->filter->filter->inputs[0]->sample_rate;
3161                 codec->channel_layout = ost->filter->filter->inputs[0]->channel_layout;
3162                 codec->channels       = av_get_channel_layout_nb_channels(codec->channel_layout);
3163                 codec->time_base      = (AVRational){ 1, codec->sample_rate };
3164                 break;
3165             case AVMEDIA_TYPE_VIDEO:
3166                 codec->time_base = (AVRational){ost->frame_rate.den, ost->frame_rate.num};
3167                 if (   av_q2d(codec->time_base) < 0.001 && video_sync_method != VSYNC_PASSTHROUGH
3168                    && (video_sync_method == VSYNC_CFR || (video_sync_method == VSYNC_AUTO && !(oc->oformat->flags & AVFMT_VARIABLE_FPS)))){
3169                     av_log(oc, AV_LOG_WARNING, "Frame rate very high for a muxer not efficiently supporting it.\n"
3170                                                "Please consider specifying a lower framerate, a different muxer or -vsync 2\n");
3171                 }
3172                 for (j = 0; j < ost->forced_kf_count; j++)
3173                     ost->forced_kf_pts[j] = av_rescale_q(ost->forced_kf_pts[j],
3174                                                          AV_TIME_BASE_Q,
3175                                                          codec->time_base);
3176
3177                 codec->width  = ost->filter->filter->inputs[0]->w;
3178                 codec->height = ost->filter->filter->inputs[0]->h;
3179                 codec->sample_aspect_ratio = ost->st->sample_aspect_ratio =
3180                     ost->frame_aspect_ratio ? // overridden by the -aspect cli option
3181                     av_d2q(ost->frame_aspect_ratio * codec->height/codec->width, 255) :
3182                     ost->filter->filter->inputs[0]->sample_aspect_ratio;
3183                 codec->pix_fmt = ost->filter->filter->inputs[0]->format;
3184
3185                 if (codec->width   != icodec->width  ||
3186                     codec->height  != icodec->height ||
3187                     codec->pix_fmt != icodec->pix_fmt) {
3188                     codec->bits_per_raw_sample = frame_bits_per_raw_sample;
3189                 }
3190
3191                 break;
3192             case AVMEDIA_TYPE_SUBTITLE:
3193                 codec->time_base = (AVRational){1, 1000};
3194                 break;
3195             default:
3196                 abort();
3197                 break;
3198             }
3199             /* two pass mode */
3200             if (codec->flags & (CODEC_FLAG_PASS1 | CODEC_FLAG_PASS2)) {
3201                 char logfilename[1024];
3202                 FILE *f;
3203
3204                 snprintf(logfilename, sizeof(logfilename), "%s-%d.log",
3205                          pass_logfilename_prefix ? pass_logfilename_prefix : DEFAULT_PASS_LOGFILENAME_PREFIX,
3206                          i);
3207                 if (!strcmp(ost->enc->name, "libx264")) {
3208                     av_dict_set(&ost->opts, "stats", logfilename, AV_DICT_DONT_OVERWRITE);
3209                 } else {
3210                     if (codec->flags & CODEC_FLAG_PASS2) {
3211                         char  *logbuffer;
3212                         size_t logbuffer_size;
3213                         if (cmdutils_read_file(logfilename, &logbuffer, &logbuffer_size) < 0) {
3214                             av_log(NULL, AV_LOG_FATAL, "Error reading log file '%s' for pass-2 encoding\n",
3215                                    logfilename);
3216                             exit_program(1);
3217                         }
3218                         codec->stats_in = logbuffer;
3219                     }
3220                     if (codec->flags & CODEC_FLAG_PASS1) {
3221                         f = fopen(logfilename, "wb");
3222                         if (!f) {
3223                             av_log(NULL, AV_LOG_FATAL, "Cannot write log file '%s' for pass-1 encoding: %s\n",
3224                                 logfilename, strerror(errno));
3225                             exit_program(1);
3226                         }
3227                         ost->logfile = f;
3228                     }
3229                 }
3230             }
3231         }
3232     }
3233
3234     /* open each encoder */
3235     for (i = 0; i < nb_output_streams; i++) {
3236         ost = output_streams[i];
3237         if (ost->encoding_needed) {
3238             AVCodec      *codec = ost->enc;
3239             AVCodecContext *dec = NULL;
3240
3241             if ((ist = get_input_stream(ost)))
3242                 dec = ist->st->codec;
3243             if (dec && dec->subtitle_header) {
3244                 ost->st->codec->subtitle_header = av_malloc(dec->subtitle_header_size);
3245                 if (!ost->st->codec->subtitle_header) {
3246                     ret = AVERROR(ENOMEM);
3247                     goto dump_format;
3248                 }
3249                 memcpy(ost->st->codec->subtitle_header, dec->subtitle_header, dec->subtitle_header_size);
3250                 ost->st->codec->subtitle_header_size = dec->subtitle_header_size;
3251             }
3252             if (!av_dict_get(ost->opts, "threads", NULL, 0))
3253                 av_dict_set(&ost->opts, "threads", "auto", 0);
3254             if (avcodec_open2(ost->st->codec, codec, &ost->opts) < 0) {
3255                 snprintf(error, sizeof(error), "Error while opening encoder for output stream #%d:%d - maybe incorrect parameters such as bit_rate, rate, width or height",
3256                         ost->file_index, ost->index);
3257                 ret = AVERROR(EINVAL);
3258                 goto dump_format;
3259             }
3260             assert_codec_experimental(ost->st->codec, 1);
3261             assert_avoptions(ost->opts);
3262             if (ost->st->codec->bit_rate && ost->st->codec->bit_rate < 1000)
3263                 av_log(NULL, AV_LOG_WARNING, "The bitrate parameter is set too low."
3264                                              " It takes bits/s as argument, not kbits/s\n");
3265             extra_size += ost->st->codec->extradata_size;
3266
3267             if (ost->st->codec->me_threshold)
3268                 input_streams[ost->source_index]->st->codec->debug |= FF_DEBUG_MV;
3269         }
3270     }
3271
3272     /* init input streams */
3273     for (i = 0; i < nb_input_streams; i++)
3274         if ((ret = init_input_stream(i, error, sizeof(error))) < 0)
3275             goto dump_format;
3276
3277     /* discard unused programs */
3278     for (i = 0; i < nb_input_files; i++) {
3279         InputFile *ifile = input_files[i];
3280         for (j = 0; j < ifile->ctx->nb_programs; j++) {
3281             AVProgram *p = ifile->ctx->programs[j];
3282             int discard  = AVDISCARD_ALL;
3283
3284             for (k = 0; k < p->nb_stream_indexes; k++)
3285                 if (!input_streams[ifile->ist_index + p->stream_index[k]]->discard) {
3286                     discard = AVDISCARD_DEFAULT;
3287                     break;
3288                 }
3289             p->discard = discard;
3290         }
3291     }
3292
3293     /* open files and write file headers */
3294     for (i = 0; i < nb_output_files; i++) {
3295         oc = output_files[i]->ctx;
3296         oc->interrupt_callback = int_cb;
3297         if ((ret = avformat_write_header(oc, &output_files[i]->opts)) < 0) {
3298             char errbuf[128];
3299             const char *errbuf_ptr = errbuf;
3300             if (av_strerror(ret, errbuf, sizeof(errbuf)) < 0)
3301                 errbuf_ptr = strerror(AVUNERROR(ret));
3302             snprintf(error, sizeof(error), "Could not write header for output file #%d (incorrect codec parameters ?): %s", i, errbuf_ptr);
3303             ret = AVERROR(EINVAL);
3304             goto dump_format;
3305         }
3306 //         assert_avoptions(output_files[i]->opts);
3307         if (strcmp(oc->oformat->name, "rtp")) {
3308             want_sdp = 0;
3309         }
3310     }
3311
3312  dump_format:
3313     /* dump the file output parameters - cannot be done before in case
3314        of stream copy */
3315     for (i = 0; i < nb_output_files; i++) {
3316         av_dump_format(output_files[i]->ctx, i, output_files[i]->ctx->filename, 1);
3317     }
3318
3319     /* dump the stream mapping */
3320     av_log(NULL, AV_LOG_INFO, "Stream mapping:\n");
3321     for (i = 0; i < nb_input_streams; i++) {
3322         ist = input_streams[i];
3323
3324         for (j = 0; j < ist->nb_filters; j++) {
3325             AVFilterLink *link = ist->filters[j]->filter->outputs[0];
3326             if (ist->filters[j]->graph->graph_desc) {
3327                 av_log(NULL, AV_LOG_INFO, "  Stream #%d:%d (%s) -> %s",
3328                        ist->file_index, ist->st->index, ist->dec ? ist->dec->name : "?",
3329                        link->dst->filter->name);
3330                 if (link->dst->input_count > 1)
3331                     av_log(NULL, AV_LOG_INFO, ":%s", link->dstpad->name);
3332                 if (nb_filtergraphs > 1)
3333                     av_log(NULL, AV_LOG_INFO, " (graph %d)", ist->filters[j]->graph->index);
3334                 av_log(NULL, AV_LOG_INFO, "\n");
3335             }
3336         }
3337     }
3338
3339     for (i = 0; i < nb_output_streams; i++) {
3340         ost = output_streams[i];
3341
3342         if (ost->attachment_filename) {
3343             /* an attached file */
3344             av_log(NULL, AV_LOG_INFO, "  File %s -> Stream #%d:%d\n",
3345                    ost->attachment_filename, ost->file_index, ost->index);
3346             continue;
3347         }
3348
3349         if (ost->filter && ost->filter->graph->graph_desc) {
3350             /* output from a complex graph */
3351             AVFilterLink *link = ost->filter->filter->inputs[0];
3352             av_log(NULL, AV_LOG_INFO, "  %s", link->src->filter->name);
3353             if (link->src->output_count > 1)
3354                 av_log(NULL, AV_LOG_INFO, ":%s", link->srcpad->name);
3355             if (nb_filtergraphs > 1)
3356                 av_log(NULL, AV_LOG_INFO, " (graph %d)", ost->filter->graph->index);
3357
3358             av_log(NULL, AV_LOG_INFO, " -> Stream #%d:%d (%s)\n", ost->file_index,
3359                    ost->index, ost->enc ? ost->enc->name : "?");
3360             continue;
3361         }
3362
3363         av_log(NULL, AV_LOG_INFO, "  Stream #%d:%d -> #%d:%d",
3364                input_streams[ost->source_index]->file_index,
3365                input_streams[ost->source_index]->st->index,
3366                ost->file_index,
3367                ost->index);
3368         if (ost->sync_ist != input_streams[ost->source_index])
3369             av_log(NULL, AV_LOG_INFO, " [sync #%d:%d]",
3370                    ost->sync_ist->file_index,
3371                    ost->sync_ist->st->index);
3372         if (ost->stream_copy)
3373             av_log(NULL, AV_LOG_INFO, " (copy)");
3374         else
3375             av_log(NULL, AV_LOG_INFO, " (%s -> %s)", input_streams[ost->source_index]->dec ?
3376                    input_streams[ost->source_index]->dec->name : "?",
3377                    ost->enc ? ost->enc->name : "?");
3378         av_log(NULL, AV_LOG_INFO, "\n");
3379     }
3380
3381     if (ret) {
3382         av_log(NULL, AV_LOG_ERROR, "%s\n", error);
3383         return ret;
3384     }
3385
3386     if (want_sdp) {
3387         print_sdp();
3388     }
3389
3390     return 0;
3391 }
3392
3393 /*
3394  * The following code is the main loop of the file converter
3395  */
3396 static int transcode(void)
3397 {
3398     int ret, i;
3399     AVFormatContext *is, *os;
3400     OutputStream *ost;
3401     InputStream *ist;
3402     uint8_t *no_packet;
3403     int no_packet_count = 0;
3404     int64_t timer_start;
3405     int key;
3406
3407     if (!(no_packet = av_mallocz(nb_input_files)))
3408         exit_program(1);
3409
3410     ret = transcode_init();
3411     if (ret < 0)
3412         goto fail;
3413
3414     if (!using_stdin) {
3415         av_log(NULL, AV_LOG_INFO, "Press [q] to stop, [?] for help\n");
3416     }
3417
3418     timer_start = av_gettime();
3419
3420     for (; received_sigterm == 0;) {
3421         int file_index, ist_index, past_recording_time = 1;
3422         AVPacket pkt;
3423         int64_t ipts_min;
3424         int64_t cur_time= av_gettime();
3425
3426         ipts_min = INT64_MAX;
3427         /* if 'q' pressed, exits */
3428         if (!using_stdin) {
3429             static int64_t last_time;
3430             if (received_nb_signals)
3431                 break;
3432             /* read_key() returns 0 on EOF */
3433             if(cur_time - last_time >= 100000 && !run_as_daemon){
3434                 key =  read_key();
3435                 last_time = cur_time;
3436             }else
3437                 key = -1;
3438             if (key == 'q')
3439                 break;
3440             if (key == '+') av_log_set_level(av_log_get_level()+10);
3441             if (key == '-') av_log_set_level(av_log_get_level()-10);
3442             if (key == 's') qp_hist     ^= 1;
3443             if (key == 'h'){
3444                 if (do_hex_dump){
3445                     do_hex_dump = do_pkt_dump = 0;
3446                 } else if(do_pkt_dump){
3447                     do_hex_dump = 1;
3448                 } else
3449                     do_pkt_dump = 1;
3450                 av_log_set_level(AV_LOG_DEBUG);
3451             }
3452             if (key == 'c' || key == 'C'){
3453                 char buf[4096], target[64], command[256], arg[256] = {0};
3454                 double time;
3455                 int k, n = 0;
3456                 fprintf(stderr, "\nEnter command: <target> <time> <command>[ <argument>]\n");
3457                 i = 0;
3458                 while ((k = read_key()) != '\n' && k != '\r' && i < sizeof(buf)-1)
3459                     if (k > 0)
3460                         buf[i++] = k;
3461                 buf[i] = 0;
3462                 if (k > 0 &&
3463                     (n = sscanf(buf, "%63[^ ] %lf %255[^ ] %255[^\n]", target, &time, command, arg)) >= 3) {
3464                     av_log(NULL, AV_LOG_DEBUG, "Processing command target:%s time:%f command:%s arg:%s",
3465                            target, time, command, arg);
3466                     for (i = 0; i < nb_filtergraphs; i++) {
3467                         FilterGraph *fg = filtergraphs[i];
3468                         if (fg->graph) {
3469                             if (time < 0) {
3470                                 ret = avfilter_graph_send_command(fg->graph, target, command, arg, buf, sizeof(buf),
3471                                                                   key == 'c' ? AVFILTER_CMD_FLAG_ONE : 0);
3472                                 fprintf(stderr, "Command reply for stream %d: ret:%d res:%s\n", i, ret, buf);
3473                             } else {
3474                                 ret = avfilter_graph_queue_command(fg->graph, target, command, arg, 0, time);
3475                             }
3476                         }
3477                     }
3478                 } else {
3479                     av_log(NULL, AV_LOG_ERROR,
3480                            "Parse error, at least 3 arguments were expected, "
3481                            "only %d given in string '%s'\n", n, buf);
3482                 }
3483             }
3484             if (key == 'd' || key == 'D'){
3485                 int debug=0;
3486                 if(key == 'D') {
3487                     debug = input_streams[0]->st->codec->debug<<1;
3488                     if(!debug) debug = 1;
3489                     while(debug & (FF_DEBUG_DCT_COEFF|FF_DEBUG_VIS_QP|FF_DEBUG_VIS_MB_TYPE)) //unsupported, would just crash
3490                         debug += debug;
3491                 }else
3492                     if(scanf("%d", &debug)!=1)
3493                         fprintf(stderr,"error parsing debug value\n");
3494                 for(i=0;i<nb_input_streams;i++) {
3495                     input_streams[i]->st->codec->debug = debug;
3496                 }
3497                 for(i=0;i<nb_output_streams;i++) {
3498                     ost = output_streams[i];
3499                     ost->st->codec->debug = debug;
3500                 }
3501                 if(debug) av_log_set_level(AV_LOG_DEBUG);
3502                 fprintf(stderr,"debug=%d\n", debug);
3503             }
3504             if (key == '?'){
3505                 fprintf(stderr, "key    function\n"
3506                                 "?      show this help\n"
3507                                 "+      increase verbosity\n"
3508                                 "-      decrease verbosity\n"
3509                                 "c      Send command to filtergraph\n"
3510                                 "D      cycle through available debug modes\n"
3511                                 "h      dump packets/hex press to cycle through the 3 states\n"
3512                                 "q      quit\n"
3513                                 "s      Show QP histogram\n"
3514                 );
3515             }
3516         }
3517
3518         /* check if there's any stream where output is still needed */
3519         for (i = 0; i < nb_output_streams; i++) {
3520             OutputFile *of;
3521             ost = output_streams[i];
3522             of  = output_files[ost->file_index];
3523             os  = output_files[ost->file_index]->ctx;
3524             if (ost->is_past_recording_time ||
3525                 (os->pb && avio_tell(os->pb) >= of->limit_filesize))
3526                 continue;
3527             if (ost->frame_number >= ost->max_frames) {
3528                 int j;
3529                 for (j = 0; j < of->ctx->nb_streams; j++)
3530                     output_streams[of->ost_index + j]->is_past_recording_time = 1;
3531                 continue;
3532             }
3533             past_recording_time = 0;
3534         }
3535         if (past_recording_time)
3536             break;
3537
3538         /* select the stream that we must read now by looking at the
3539            smallest output pts */
3540         file_index = -1;
3541         for (i = 0; i < nb_input_streams; i++) {
3542             int64_t ipts;
3543             ist = input_streams[i];
3544             ipts = ist->pts;
3545             if (ist->discard || no_packet[ist->file_index])
3546                 continue;
3547             if (!input_files[ist->file_index]->eof_reached) {
3548                 if (ipts < ipts_min) {
3549                     ipts_min = ipts;
3550                     file_index = ist->file_index;
3551                 }
3552             }
3553         }
3554         /* if none, if is finished */
3555         if (file_index < 0) {
3556             if (no_packet_count) {
3557                 no_packet_count = 0;
3558                 memset(no_packet, 0, nb_input_files);
3559                 usleep(10000);
3560                 continue;
3561             }
3562             break;
3563         }
3564
3565         /* read a frame from it and output it in the fifo */
3566         is  = input_files[file_index]->ctx;
3567         ret = av_read_frame(is, &pkt);
3568         if (ret == AVERROR(EAGAIN)) {
3569             no_packet[file_index] = 1;
3570             no_packet_count++;
3571             continue;
3572         }
3573         if (ret < 0) {
3574             input_files[file_index]->eof_reached = 1;
3575
3576             for (i = 0; i < input_files[file_index]->nb_streams; i++) {
3577                 ist = input_streams[input_files[file_index]->ist_index + i];
3578                 if (ist->decoding_needed)
3579                     output_packet(ist, NULL);
3580             }
3581
3582             if (opt_shortest)
3583                 break;
3584             else
3585                 continue;
3586         }
3587
3588         no_packet_count = 0;
3589         memset(no_packet, 0, nb_input_files);
3590
3591         if (do_pkt_dump) {
3592             av_pkt_dump_log2(NULL, AV_LOG_DEBUG, &pkt, do_hex_dump,
3593                              is->streams[pkt.stream_index]);
3594         }
3595         /* the following test is needed in case new streams appear
3596            dynamically in stream : we ignore them */
3597         if (pkt.stream_index >= input_files[file_index]->nb_streams)
3598             goto discard_packet;
3599         ist_index = input_files[file_index]->ist_index + pkt.stream_index;
3600         ist = input_streams[ist_index];
3601         if (ist->discard)
3602             goto discard_packet;
3603
3604         if (pkt.dts != AV_NOPTS_VALUE)
3605             pkt.dts += av_rescale_q(input_files[ist->file_index]->ts_offset, AV_TIME_BASE_Q, ist->st->time_base);
3606         if (pkt.pts != AV_NOPTS_VALUE)
3607             pkt.pts += av_rescale_q(input_files[ist->file_index]->ts_offset, AV_TIME_BASE_Q, ist->st->time_base);
3608
3609         if (pkt.pts != AV_NOPTS_VALUE)
3610             pkt.pts *= ist->ts_scale;
3611         if (pkt.dts != AV_NOPTS_VALUE)
3612             pkt.dts *= ist->ts_scale;
3613
3614         if (debug_ts) {
3615             av_log(NULL, AV_LOG_INFO, "demuxer -> ist_index:%d type:%s "
3616                     "next_dts:%s next_dts_time:%s next_pts:%s next_pts_time:%s  pkt_pts:%s pkt_pts_time:%s pkt_dts:%s pkt_dts_time:%s off:%"PRId64"\n",
3617                     ist_index, av_get_media_type_string(ist->st->codec->codec_type),
3618                     av_ts2str(ist->next_dts), av_ts2timestr(ist->next_dts, &ist->st->time_base),
3619                     av_ts2str(ist->next_pts), av_ts2timestr(ist->next_pts, &ist->st->time_base),
3620                     av_ts2str(pkt.pts), av_ts2timestr(pkt.pts, &ist->st->time_base),
3621                     av_ts2str(pkt.dts), av_ts2timestr(pkt.dts, &ist->st->time_base),
3622                     input_files[ist->file_index]->ts_offset);
3623         }
3624
3625         if (pkt.dts != AV_NOPTS_VALUE && ist->next_dts != AV_NOPTS_VALUE && !copy_ts) {
3626             int64_t pkt_dts = av_rescale_q(pkt.dts, ist->st->time_base, AV_TIME_BASE_Q);
3627             int64_t delta   = pkt_dts - ist->next_dts;
3628             if (is->iformat->flags & AVFMT_TS_DISCONT) {
3629             if(delta < -1LL*dts_delta_threshold*AV_TIME_BASE ||
3630                 (delta > 1LL*dts_delta_threshold*AV_TIME_BASE &&
3631                  ist->st->codec->codec_type != AVMEDIA_TYPE_SUBTITLE) ||
3632                 pkt_dts+1<ist->pts){
3633                 input_files[ist->file_index]->ts_offset -= delta;
3634                 av_log(NULL, AV_LOG_DEBUG,
3635                        "timestamp discontinuity %"PRId64", new offset= %"PRId64"\n",
3636                        delta, input_files[ist->file_index]->ts_offset);
3637                 pkt.dts-= av_rescale_q(delta, AV_TIME_BASE_Q, ist->st->time_base);
3638                 if (pkt.pts != AV_NOPTS_VALUE)
3639                     pkt.pts-= av_rescale_q(delta, AV_TIME_BASE_Q, ist->st->time_base);
3640             }
3641             } else {
3642                 if ( delta < -1LL*dts_error_threshold*AV_TIME_BASE ||
3643                     (delta > 1LL*dts_error_threshold*AV_TIME_BASE && ist->st->codec->codec_type != AVMEDIA_TYPE_SUBTITLE) ||
3644                      pkt_dts+1<ist->pts){
3645                     av_log(NULL, AV_LOG_WARNING, "DTS %"PRId64", next:%"PRId64" st:%d invalid dropping\n", pkt.dts, ist->next_dts, pkt.stream_index);
3646                     pkt.dts = AV_NOPTS_VALUE;
3647                 }
3648                 if (pkt.pts != AV_NOPTS_VALUE){
3649                     int64_t pkt_pts = av_rescale_q(pkt.pts, ist->st->time_base, AV_TIME_BASE_Q);
3650                     delta   = pkt_pts - ist->next_dts;
3651                     if ( delta < -1LL*dts_error_threshold*AV_TIME_BASE ||
3652                         (delta > 1LL*dts_error_threshold*AV_TIME_BASE && ist->st->codec->codec_type != AVMEDIA_TYPE_SUBTITLE) ||
3653                         pkt_pts+1<ist->pts) {
3654                         av_log(NULL, AV_LOG_WARNING, "PTS %"PRId64", next:%"PRId64" invalid dropping st:%d\n", pkt.pts, ist->next_dts, pkt.stream_index);
3655                         pkt.pts = AV_NOPTS_VALUE;
3656                     }
3657                 }
3658             }
3659         }
3660
3661         // fprintf(stderr,"read #%d.%d size=%d\n", ist->file_index, ist->st->index, pkt.size);
3662         if (output_packet(ist, &pkt) < 0 ||
3663             ((ret = poll_filters()) < 0 && ret != AVERROR_EOF)) {
3664             av_log(NULL, AV_LOG_ERROR, "Error while decoding stream #%d:%d\n",
3665                    ist->file_index, ist->st->index);
3666             if (exit_on_error)
3667                 exit_program(1);
3668             av_free_packet(&pkt);
3669             continue;
3670         }
3671
3672     discard_packet:
3673         av_free_packet(&pkt);
3674
3675         /* dump report by using the output first video and audio streams */
3676         print_report(0, timer_start, cur_time);
3677     }
3678
3679     /* at the end of stream, we must flush the decoder buffers */
3680     for (i = 0; i < nb_input_streams; i++) {
3681         ist = input_streams[i];
3682         if (!input_files[ist->file_index]->eof_reached && ist->decoding_needed) {
3683             output_packet(ist, NULL);
3684         }
3685     }
3686     poll_filters();
3687     flush_encoders();
3688
3689     term_exit();
3690
3691     /* write the trailer if needed and close file */
3692     for (i = 0; i < nb_output_files; i++) {
3693         os = output_files[i]->ctx;
3694         av_write_trailer(os);
3695     }
3696
3697     /* dump report by using the first video and audio streams */
3698     print_report(1, timer_start, av_gettime());
3699
3700     /* close each encoder */
3701     for (i = 0; i < nb_output_streams; i++) {
3702         ost = output_streams[i];
3703         if (ost->encoding_needed) {
3704             av_freep(&ost->st->codec->stats_in);
3705             avcodec_close(ost->st->codec);
3706         }
3707     }
3708
3709     /* close each decoder */
3710     for (i = 0; i < nb_input_streams; i++) {
3711         ist = input_streams[i];
3712         if (ist->decoding_needed) {
3713             avcodec_close(ist->st->codec);
3714         }
3715     }
3716
3717     /* finished ! */
3718     ret = 0;
3719
3720  fail:
3721     av_freep(&no_packet);
3722
3723     if (output_streams) {
3724         for (i = 0; i < nb_output_streams; i++) {
3725             ost = output_streams[i];
3726             if (ost) {
3727                 if (ost->stream_copy)
3728                     av_freep(&ost->st->codec->extradata);
3729                 if (ost->logfile) {
3730                     fclose(ost->logfile);
3731                     ost->logfile = NULL;
3732                 }
3733                 av_freep(&ost->st->codec->subtitle_header);
3734                 av_free(ost->forced_kf_pts);
3735                 av_dict_free(&ost->opts);
3736             }
3737         }
3738     }
3739     return ret;
3740 }
3741
3742 static int opt_frame_crop(const char *opt, const char *arg)
3743 {
3744     av_log(NULL, AV_LOG_FATAL, "Option '%s' has been removed, use the crop filter instead\n", opt);
3745     return AVERROR(EINVAL);
3746 }
3747
3748 static int opt_pad(const char *opt, const char *arg)
3749 {
3750     av_log(NULL, AV_LOG_FATAL, "Option '%s' has been removed, use the pad filter instead\n", opt);
3751     return -1;
3752 }
3753
3754 static int opt_video_channel(const char *opt, const char *arg)
3755 {
3756     av_log(NULL, AV_LOG_WARNING, "This option is deprecated, use -channel.\n");
3757     return opt_default("channel", arg);
3758 }
3759
3760 static int opt_video_standard(const char *opt, const char *arg)
3761 {
3762     av_log(NULL, AV_LOG_WARNING, "This option is deprecated, use -standard.\n");
3763     return opt_default("standard", arg);
3764 }
3765
3766 static int opt_audio_codec(OptionsContext *o, const char *opt, const char *arg)
3767 {
3768     audio_codec_name = arg;
3769     return parse_option(o, "codec:a", arg, options);
3770 }
3771
3772 static int opt_video_codec(OptionsContext *o, const char *opt, const char *arg)
3773 {
3774     video_codec_name = arg;
3775     return parse_option(o, "codec:v", arg, options);
3776 }
3777
3778 static int opt_subtitle_codec(OptionsContext *o, const char *opt, const char *arg)
3779 {
3780     subtitle_codec_name = arg;
3781     return parse_option(o, "codec:s", arg, options);
3782 }
3783
3784 static int opt_data_codec(OptionsContext *o, const char *opt, const char *arg)
3785 {
3786     return parse_option(o, "codec:d", arg, options);
3787 }
3788
3789 static int opt_map(OptionsContext *o, const char *opt, const char *arg)
3790 {
3791     StreamMap *m = NULL;
3792     int i, negative = 0, file_idx;
3793     int sync_file_idx = -1, sync_stream_idx = 0;
3794     char *p, *sync;
3795     char *map;
3796
3797     if (*arg == '-') {
3798         negative = 1;
3799         arg++;
3800     }
3801     map = av_strdup(arg);
3802
3803     /* parse sync stream first, just pick first matching stream */
3804     if (sync = strchr(map, ',')) {
3805         *sync = 0;
3806         sync_file_idx = strtol(sync + 1, &sync, 0);
3807         if (sync_file_idx >= nb_input_files || sync_file_idx < 0) {
3808             av_log(NULL, AV_LOG_FATAL, "Invalid sync file index: %d.\n", sync_file_idx);
3809             exit_program(1);
3810         }
3811         if (*sync)
3812             sync++;
3813         for (i = 0; i < input_files[sync_file_idx]->nb_streams; i++)
3814             if (check_stream_specifier(input_files[sync_file_idx]->ctx,
3815                                        input_files[sync_file_idx]->ctx->streams[i], sync) == 1) {
3816                 sync_stream_idx = i;
3817                 break;
3818             }
3819         if (i == input_files[sync_file_idx]->nb_streams) {
3820             av_log(NULL, AV_LOG_FATAL, "Sync stream specification in map %s does not "
3821                                        "match any streams.\n", arg);
3822             exit_program(1);
3823         }
3824     }
3825
3826
3827     if (map[0] == '[') {
3828         /* this mapping refers to lavfi output */
3829         const char *c = map + 1;
3830         o->stream_maps = grow_array(o->stream_maps, sizeof(*o->stream_maps),
3831                                     &o->nb_stream_maps, o->nb_stream_maps + 1);
3832         m = &o->stream_maps[o->nb_stream_maps - 1];
3833         m->linklabel = av_get_token(&c, "]");
3834         if (!m->linklabel) {
3835             av_log(NULL, AV_LOG_ERROR, "Invalid output link label: %s.\n", map);
3836             exit_program(1);
3837         }
3838     } else {
3839         file_idx = strtol(map, &p, 0);
3840         if (file_idx >= nb_input_files || file_idx < 0) {
3841             av_log(NULL, AV_LOG_FATAL, "Invalid input file index: %d.\n", file_idx);
3842             exit_program(1);
3843         }
3844         if (negative)
3845             /* disable some already defined maps */
3846             for (i = 0; i < o->nb_stream_maps; i++) {
3847                 m = &o->stream_maps[i];
3848                 if (file_idx == m->file_index &&
3849                     check_stream_specifier(input_files[m->file_index]->ctx,
3850                                            input_files[m->file_index]->ctx->streams[m->stream_index],
3851                                            *p == ':' ? p + 1 : p) > 0)
3852                     m->disabled = 1;
3853             }
3854         else
3855             for (i = 0; i < input_files[file_idx]->nb_streams; i++) {
3856                 if (check_stream_specifier(input_files[file_idx]->ctx, input_files[file_idx]->ctx->streams[i],
3857                             *p == ':' ? p + 1 : p) <= 0)
3858                     continue;
3859                 o->stream_maps = grow_array(o->stream_maps, sizeof(*o->stream_maps),
3860                                             &o->nb_stream_maps, o->nb_stream_maps + 1);
3861                 m = &o->stream_maps[o->nb_stream_maps - 1];
3862
3863                 m->file_index   = file_idx;
3864                 m->stream_index = i;
3865
3866                 if (sync_file_idx >= 0) {
3867                     m->sync_file_index   = sync_file_idx;
3868                     m->sync_stream_index = sync_stream_idx;
3869                 } else {
3870                     m->sync_file_index   = file_idx;
3871                     m->sync_stream_index = i;
3872                 }
3873             }
3874     }
3875
3876     if (!m) {
3877         av_log(NULL, AV_LOG_FATAL, "Stream map '%s' matches no streams.\n", arg);
3878         exit_program(1);
3879     }
3880
3881     av_freep(&map);
3882     return 0;
3883 }
3884
3885 static int opt_attach(OptionsContext *o, const char *opt, const char *arg)
3886 {
3887     o->attachments = grow_array(o->attachments, sizeof(*o->attachments),
3888                                 &o->nb_attachments, o->nb_attachments + 1);
3889     o->attachments[o->nb_attachments - 1] = arg;
3890     return 0;
3891 }
3892
3893 static int opt_map_channel(OptionsContext *o, const char *opt, const char *arg)
3894 {
3895     int n;
3896     AVStream *st;
3897     AudioChannelMap *m;
3898
3899     o->audio_channel_maps =
3900         grow_array(o->audio_channel_maps, sizeof(*o->audio_channel_maps),
3901                    &o->nb_audio_channel_maps, o->nb_audio_channel_maps + 1);
3902     m = &o->audio_channel_maps[o->nb_audio_channel_maps - 1];
3903
3904     /* muted channel syntax */
3905     n = sscanf(arg, "%d:%d.%d", &m->channel_idx, &m->ofile_idx, &m->ostream_idx);
3906     if ((n == 1 || n == 3) && m->channel_idx == -1) {
3907         m->file_idx = m->stream_idx = -1;
3908         if (n == 1)
3909             m->ofile_idx = m->ostream_idx = -1;
3910         return 0;
3911     }
3912
3913     /* normal syntax */
3914     n = sscanf(arg, "%d.%d.%d:%d.%d",
3915                &m->file_idx,  &m->stream_idx, &m->channel_idx,
3916                &m->ofile_idx, &m->ostream_idx);
3917
3918     if (n != 3 && n != 5) {
3919         av_log(NULL, AV_LOG_FATAL, "Syntax error, mapchan usage: "
3920                "[file.stream.channel|-1][:syncfile:syncstream]\n");
3921         exit_program(1);
3922     }
3923
3924     if (n != 5) // only file.stream.channel specified
3925         m->ofile_idx = m->ostream_idx = -1;
3926
3927     /* check input */
3928     if (m->file_idx < 0 || m->file_idx >= nb_input_files) {
3929         av_log(NULL, AV_LOG_FATAL, "mapchan: invalid input file index: %d\n",
3930                m->file_idx);
3931         exit_program(1);
3932     }
3933     if (m->stream_idx < 0 ||
3934         m->stream_idx >= input_files[m->file_idx]->nb_streams) {
3935         av_log(NULL, AV_LOG_FATAL, "mapchan: invalid input file stream index #%d.%d\n",
3936                m->file_idx, m->stream_idx);
3937         exit_program(1);
3938     }
3939     st = input_files[m->file_idx]->ctx->streams[m->stream_idx];
3940     if (st->codec->codec_type != AVMEDIA_TYPE_AUDIO) {
3941         av_log(NULL, AV_LOG_FATAL, "mapchan: stream #%d.%d is not an audio stream.\n",
3942                m->file_idx, m->stream_idx);
3943         exit_program(1);
3944     }
3945     if (m->channel_idx < 0 || m->channel_idx >= st->codec->channels) {
3946         av_log(NULL, AV_LOG_FATAL, "mapchan: invalid audio channel #%d.%d.%d\n",
3947                m->file_idx, m->stream_idx, m->channel_idx);
3948         exit_program(1);
3949     }
3950     return 0;
3951 }
3952
3953 /**
3954  * Parse a metadata specifier in arg.
3955  * @param type metadata type is written here -- g(lobal)/s(tream)/c(hapter)/p(rogram)
3956  * @param index for type c/p, chapter/program index is written here
3957  * @param stream_spec for type s, the stream specifier is written here
3958  */
3959 static void parse_meta_type(char *arg, char *type, int *index, const char **stream_spec)
3960 {
3961     if (*arg) {
3962         *type = *arg;
3963         switch (*arg) {
3964         case 'g':
3965             break;
3966         case 's':
3967             if (*(++arg) && *arg != ':') {
3968                 av_log(NULL, AV_LOG_FATAL, "Invalid metadata specifier %s.\n", arg);
3969                 exit_program(1);
3970             }
3971             *stream_spec = *arg == ':' ? arg + 1 : "";
3972             break;
3973         case 'c':
3974         case 'p':
3975             if (*(++arg) == ':')
3976                 *index = strtol(++arg, NULL, 0);
3977             break;
3978         default:
3979             av_log(NULL, AV_LOG_FATAL, "Invalid metadata type %c.\n", *arg);
3980             exit_program(1);
3981         }
3982     } else
3983         *type = 'g';
3984 }
3985
3986 static int copy_metadata(char *outspec, char *inspec, AVFormatContext *oc, AVFormatContext *ic, OptionsContext *o)
3987 {
3988     AVDictionary **meta_in = NULL;
3989     AVDictionary **meta_out = NULL;
3990     int i, ret = 0;
3991     char type_in, type_out;
3992     const char *istream_spec = NULL, *ostream_spec = NULL;
3993     int idx_in = 0, idx_out = 0;
3994
3995     parse_meta_type(inspec,  &type_in,  &idx_in,  &istream_spec);
3996     parse_meta_type(outspec, &type_out, &idx_out, &ostream_spec);
3997
3998     if (!ic) {
3999         if (type_out == 'g' || !*outspec)
4000             o->metadata_global_manual = 1;
4001         if (type_out == 's' || !*outspec)
4002             o->metadata_streams_manual = 1;
4003         if (type_out == 'c' || !*outspec)
4004             o->metadata_chapters_manual = 1;
4005         return 0;
4006     }
4007
4008     if (type_in == 'g' || type_out == 'g')
4009         o->metadata_global_manual = 1;
4010     if (type_in == 's' || type_out == 's')
4011         o->metadata_streams_manual = 1;
4012     if (type_in == 'c' || type_out == 'c')
4013         o->metadata_chapters_manual = 1;
4014
4015 #define METADATA_CHECK_INDEX(index, nb_elems, desc)\
4016     if ((index) < 0 || (index) >= (nb_elems)) {\
4017         av_log(NULL, AV_LOG_FATAL, "Invalid %s index %d while processing metadata maps.\n",\
4018                 (desc), (index));\
4019         exit_program(1);\
4020     }
4021
4022 #define SET_DICT(type, meta, context, index)\
4023         switch (type) {\
4024         case 'g':\
4025             meta = &context->metadata;\
4026             break;\
4027         case 'c':\
4028             METADATA_CHECK_INDEX(index, context->nb_chapters, "chapter")\
4029             meta = &context->chapters[index]->metadata;\
4030             break;\
4031         case 'p':\
4032             METADATA_CHECK_INDEX(index, context->nb_programs, "program")\
4033             meta = &context->programs[index]->metadata;\
4034             break;\
4035         }\
4036
4037     SET_DICT(type_in, meta_in, ic, idx_in);
4038     SET_DICT(type_out, meta_out, oc, idx_out);
4039
4040     /* for input streams choose first matching stream */
4041     if (type_in == 's') {
4042         for (i = 0; i < ic->nb_streams; i++) {
4043             if ((ret = check_stream_specifier(ic, ic->streams[i], istream_spec)) > 0) {
4044                 meta_in = &ic->streams[i]->metadata;
4045                 break;
4046             } else if (ret < 0)
4047                 exit_program(1);
4048         }
4049         if (!meta_in) {
4050             av_log(NULL, AV_LOG_FATAL, "Stream specifier %s does not match  any streams.\n", istream_spec);
4051             exit_program(1);
4052         }
4053     }
4054
4055     if (type_out == 's') {
4056         for (i = 0; i < oc->nb_streams; i++) {
4057             if ((ret = check_stream_specifier(oc, oc->streams[i], ostream_spec)) > 0) {
4058                 meta_out = &oc->streams[i]->metadata;
4059                 av_dict_copy(meta_out, *meta_in, AV_DICT_DONT_OVERWRITE);
4060             } else if (ret < 0)
4061                 exit_program(1);
4062         }
4063     } else
4064         av_dict_copy(meta_out, *meta_in, AV_DICT_DONT_OVERWRITE);
4065
4066     return 0;
4067 }
4068
4069 static int opt_recording_timestamp(OptionsContext *o, const char *opt, const char *arg)
4070 {
4071     char buf[128];
4072     int64_t recording_timestamp = parse_time_or_die(opt, arg, 0) / 1E6;
4073     struct tm time = *gmtime((time_t*)&recording_timestamp);
4074     strftime(buf, sizeof(buf), "creation_time=%FT%T%z", &time);
4075     parse_option(o, "metadata", buf, options);
4076
4077     av_log(NULL, AV_LOG_WARNING, "%s is deprecated, set the 'creation_time' metadata "
4078                                  "tag instead.\n", opt);
4079     return 0;
4080 }
4081
4082 static AVCodec *find_codec_or_die(const char *name, enum AVMediaType type, int encoder)
4083 {
4084     const char *codec_string = encoder ? "encoder" : "decoder";
4085     AVCodec *codec;
4086
4087     codec = encoder ?
4088         avcodec_find_encoder_by_name(name) :
4089         avcodec_find_decoder_by_name(name);
4090     if (!codec) {
4091         av_log(NULL, AV_LOG_FATAL, "Unknown %s '%s'\n", codec_string, name);
4092         exit_program(1);
4093     }
4094     if (codec->type != type) {
4095         av_log(NULL, AV_LOG_FATAL, "Invalid %s type '%s'\n", codec_string, name);
4096         exit_program(1);
4097     }
4098     return codec;
4099 }
4100
4101 static AVCodec *choose_decoder(OptionsContext *o, AVFormatContext *s, AVStream *st)
4102 {
4103     char *codec_name = NULL;
4104
4105     MATCH_PER_STREAM_OPT(codec_names, str, codec_name, s, st);
4106     if (codec_name) {
4107         AVCodec *codec = find_codec_or_die(codec_name, st->codec->codec_type, 0);
4108         st->codec->codec_id = codec->id;
4109         return codec;
4110     } else
4111         return avcodec_find_decoder(st->codec->codec_id);
4112 }
4113
4114 /**
4115  * Add all the streams from the given input file to the global
4116  * list of input streams.
4117  */
4118 static void add_input_streams(OptionsContext *o, AVFormatContext *ic)
4119 {
4120     int i;
4121     char *next, *codec_tag = NULL;
4122
4123     for (i = 0; i < ic->nb_streams; i++) {
4124         AVStream *st = ic->streams[i];
4125         AVCodecContext *dec = st->codec;
4126         InputStream *ist = av_mallocz(sizeof(*ist));
4127
4128         if (!ist)
4129             exit_program(1);
4130
4131         input_streams = grow_array(input_streams, sizeof(*input_streams), &nb_input_streams, nb_input_streams + 1);
4132         input_streams[nb_input_streams - 1] = ist;
4133
4134         ist->st = st;
4135         ist->file_index = nb_input_files;
4136         ist->discard = 1;
4137         st->discard  = AVDISCARD_ALL;
4138         ist->opts = filter_codec_opts(codec_opts, choose_decoder(o, ic, st), ic, st);
4139
4140         ist->ts_scale = 1.0;
4141         MATCH_PER_STREAM_OPT(ts_scale, dbl, ist->ts_scale, ic, st);
4142
4143         MATCH_PER_STREAM_OPT(codec_tags, str, codec_tag, ic, st);
4144         if (codec_tag) {
4145             uint32_t tag = strtol(codec_tag, &next, 0);
4146             if (*next)
4147                 tag = AV_RL32(codec_tag);
4148             st->codec->codec_tag = tag;
4149         }
4150
4151         ist->dec = choose_decoder(o, ic, st);
4152
4153         switch (dec->codec_type) {
4154         case AVMEDIA_TYPE_VIDEO:
4155             if(!ist->dec)
4156                 ist->dec = avcodec_find_decoder(dec->codec_id);
4157             if (dec->lowres) {
4158                 dec->flags |= CODEC_FLAG_EMU_EDGE;
4159             }
4160
4161             ist->resample_height  = dec->height;
4162             ist->resample_width   = dec->width;
4163             ist->resample_pix_fmt = dec->pix_fmt;
4164
4165             break;
4166         case AVMEDIA_TYPE_AUDIO:
4167             guess_input_channel_layout(ist);
4168
4169             ist->resample_sample_fmt     = dec->sample_fmt;
4170             ist->resample_sample_rate    = dec->sample_rate;
4171             ist->resample_channels       = dec->channels;
4172             ist->resample_channel_layout = dec->channel_layout;
4173
4174             break;
4175         case AVMEDIA_TYPE_DATA:
4176         case AVMEDIA_TYPE_SUBTITLE:
4177             if(!ist->dec)
4178                 ist->dec = avcodec_find_decoder(dec->codec_id);
4179             break;
4180         case AVMEDIA_TYPE_ATTACHMENT:
4181         case AVMEDIA_TYPE_UNKNOWN:
4182             break;
4183         default:
4184             abort();
4185         }
4186     }
4187 }
4188
4189 static void assert_file_overwrite(const char *filename)
4190 {
4191     if ((!file_overwrite || no_file_overwrite) &&
4192         (strchr(filename, ':') == NULL || filename[1] == ':' ||
4193          av_strstart(filename, "file:", NULL))) {
4194         if (avio_check(filename, 0) == 0) {
4195             if (!using_stdin && (!no_file_overwrite || file_overwrite)) {
4196                 fprintf(stderr,"File '%s' already exists. Overwrite ? [y/N] ", filename);
4197                 fflush(stderr);
4198                 term_exit();
4199                 signal(SIGINT, SIG_DFL);
4200                 if (!read_yesno()) {
4201                     av_log(NULL, AV_LOG_FATAL, "Not overwriting - exiting\n");
4202                     exit_program(1);
4203                 }
4204                 term_init();
4205             }
4206             else {
4207                 av_log(NULL, AV_LOG_FATAL, "File '%s' already exists. Exiting.\n", filename);
4208                 exit_program(1);
4209             }
4210         }
4211     }
4212 }
4213
4214 static void dump_attachment(AVStream *st, const char *filename)
4215 {
4216     int ret;
4217     AVIOContext *out = NULL;
4218     AVDictionaryEntry *e;
4219
4220     if (!st->codec->extradata_size) {
4221         av_log(NULL, AV_LOG_WARNING, "No extradata to dump in stream #%d:%d.\n",
4222                nb_input_files - 1, st->index);
4223         return;
4224     }
4225     if (!*filename && (e = av_dict_get(st->metadata, "filename", NULL, 0)))
4226         filename = e->value;
4227     if (!*filename) {
4228         av_log(NULL, AV_LOG_FATAL, "No filename specified and no 'filename' tag"
4229                "in stream #%d:%d.\n", nb_input_files - 1, st->index);
4230         exit_program(1);
4231     }
4232
4233     assert_file_overwrite(filename);
4234
4235     if ((ret = avio_open2(&out, filename, AVIO_FLAG_WRITE, &int_cb, NULL)) < 0) {
4236         av_log(NULL, AV_LOG_FATAL, "Could not open file %s for writing.\n",
4237                filename);
4238         exit_program(1);
4239     }
4240
4241     avio_write(out, st->codec->extradata, st->codec->extradata_size);
4242     avio_flush(out);
4243     avio_close(out);
4244 }
4245
4246 static int opt_input_file(OptionsContext *o, const char *opt, const char *filename)
4247 {
4248     AVFormatContext *ic;
4249     AVInputFormat *file_iformat = NULL;
4250     int err, i, ret;
4251     int64_t timestamp;
4252     uint8_t buf[128];
4253     AVDictionary **opts;
4254     int orig_nb_streams;                     // number of streams before avformat_find_stream_info
4255
4256     if (o->format) {
4257         if (!(file_iformat = av_find_input_format(o->format))) {
4258             av_log(NULL, AV_LOG_FATAL, "Unknown input format: '%s'\n", o->format);
4259             exit_program(1);
4260         }
4261     }
4262
4263     if (!strcmp(filename, "-"))
4264         filename = "pipe:";
4265
4266     using_stdin |= !strncmp(filename, "pipe:", 5) ||
4267                     !strcmp(filename, "/dev/stdin");
4268
4269     /* get default parameters from command line */
4270     ic = avformat_alloc_context();
4271     if (!ic) {
4272         print_error(filename, AVERROR(ENOMEM));
4273         exit_program(1);
4274     }
4275     if (o->nb_audio_sample_rate) {
4276         snprintf(buf, sizeof(buf), "%d", o->audio_sample_rate[o->nb_audio_sample_rate - 1].u.i);
4277         av_dict_set(&format_opts, "sample_rate", buf, 0);
4278     }
4279     if (o->nb_audio_channels) {
4280         /* because we set audio_channels based on both the "ac" and
4281          * "channel_layout" options, we need to check that the specified
4282          * demuxer actually has the "channels" option before setting it */
4283         if (file_iformat && file_iformat->priv_class &&
4284             av_opt_find(&file_iformat->priv_class, "channels", NULL, 0,
4285    &nbs