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