avconv: saner output video timebase.
[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 = ret;
1479                 video_size += ret;
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             av_free(buffer_to_free);
2019             return AVERROR(ENOMEM);
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 (ost->output_video_filter)
2028                 get_filtered_video_frame(ost->output_video_filter, filtered_frame, &ost->picref, &ist_pts_tb);
2029             if (ost->picref)
2030                 filtered_frame->pts = av_rescale_q(ost->picref->pts, ist_pts_tb, AV_TIME_BASE_Q);
2031             if (ost->picref->video && !ost->frame_aspect_ratio)
2032                 ost->st->codec->sample_aspect_ratio = ost->picref->video->pixel_aspect;
2033 #else
2034             filtered_frame = decoded_frame;
2035 #endif
2036
2037             do_video_out(output_files[ost->file_index].ctx, ost, ist, filtered_frame, &frame_size,
2038                          same_quant ? quality : ost->st->codec->global_quality);
2039             if (vstats_filename && frame_size)
2040                 do_video_stats(output_files[ost->file_index].ctx, ost, frame_size);
2041 #if CONFIG_AVFILTER
2042             frame_available = ost->output_video_filter && avfilter_poll_frame(ost->output_video_filter->inputs[0]);
2043             if (ost->picref)
2044                 avfilter_unref_buffer(ost->picref);
2045         }
2046 #endif
2047     }
2048
2049     av_free(buffer_to_free);
2050     return ret;
2051 }
2052
2053 static int transcode_subtitles(InputStream *ist, AVPacket *pkt, int *got_output)
2054 {
2055     AVSubtitle subtitle;
2056     int i, ret = avcodec_decode_subtitle2(ist->st->codec,
2057                                           &subtitle, got_output, pkt);
2058     if (ret < 0)
2059         return ret;
2060     if (!*got_output)
2061         return ret;
2062
2063     rate_emu_sleep(ist);
2064
2065     for (i = 0; i < nb_output_streams; i++) {
2066         OutputStream *ost = &output_streams[i];
2067
2068         if (!check_output_constraints(ist, ost) || !ost->encoding_needed)
2069             continue;
2070
2071         do_subtitle_out(output_files[ost->file_index].ctx, ost, ist, &subtitle, pkt->pts);
2072     }
2073
2074     avsubtitle_free(&subtitle);
2075     return ret;
2076 }
2077
2078 /* pkt = NULL means EOF (needed to flush decoder buffers) */
2079 static int output_packet(InputStream *ist,
2080                          OutputStream *ost_table, int nb_ostreams,
2081                          const AVPacket *pkt)
2082 {
2083     int i;
2084     int got_output;
2085     int64_t pkt_pts = AV_NOPTS_VALUE;
2086     AVPacket avpkt;
2087
2088     if (ist->next_dts == AV_NOPTS_VALUE)
2089         ist->next_dts = ist->last_dts;
2090
2091     if (pkt == NULL) {
2092         /* EOF handling */
2093         av_init_packet(&avpkt);
2094         avpkt.data = NULL;
2095         avpkt.size = 0;
2096         goto handle_eof;
2097     } else {
2098         avpkt = *pkt;
2099     }
2100
2101     if (pkt->dts != AV_NOPTS_VALUE)
2102         ist->next_dts = ist->last_dts = av_rescale_q(pkt->dts, ist->st->time_base, AV_TIME_BASE_Q);
2103     if (pkt->pts != AV_NOPTS_VALUE)
2104         pkt_pts = av_rescale_q(pkt->pts, ist->st->time_base, AV_TIME_BASE_Q);
2105
2106     // while we have more to decode or while the decoder did output something on EOF
2107     while (ist->decoding_needed && (avpkt.size > 0 || (!pkt && got_output))) {
2108         int ret = 0;
2109     handle_eof:
2110
2111         ist->last_dts = ist->next_dts;
2112
2113         if (avpkt.size && avpkt.size != pkt->size) {
2114             av_log(NULL, ist->showed_multi_packet_warning ? AV_LOG_VERBOSE : AV_LOG_WARNING,
2115                    "Multiple frames in a packet from stream %d\n", pkt->stream_index);
2116             ist->showed_multi_packet_warning = 1;
2117         }
2118
2119         switch (ist->st->codec->codec_type) {
2120         case AVMEDIA_TYPE_AUDIO:
2121             ret = transcode_audio    (ist, &avpkt, &got_output);
2122             break;
2123         case AVMEDIA_TYPE_VIDEO:
2124             ret = transcode_video    (ist, &avpkt, &got_output, &pkt_pts);
2125             if (avpkt.duration)
2126                 ist->next_dts += av_rescale_q(avpkt.duration, ist->st->time_base, AV_TIME_BASE_Q);
2127             else if (ist->st->r_frame_rate.num)
2128                 ist->next_dts += av_rescale_q(1, (AVRational){ist->st->r_frame_rate.den,
2129                                                               ist->st->r_frame_rate.num},
2130                                               AV_TIME_BASE_Q);
2131             else if (ist->st->codec->time_base.num != 0) {
2132                 int ticks      = ist->st->parser ? ist->st->parser->repeat_pict + 1 :
2133                                                    ist->st->codec->ticks_per_frame;
2134                 ist->next_dts += av_rescale_q(ticks, ist->st->codec->time_base, AV_TIME_BASE_Q);
2135             }
2136             break;
2137         case AVMEDIA_TYPE_SUBTITLE:
2138             ret = transcode_subtitles(ist, &avpkt, &got_output);
2139             break;
2140         default:
2141             return -1;
2142         }
2143
2144         if (ret < 0)
2145             return ret;
2146         // touch data and size only if not EOF
2147         if (pkt) {
2148             avpkt.data += ret;
2149             avpkt.size -= ret;
2150         }
2151         if (!got_output) {
2152             continue;
2153         }
2154     }
2155
2156     /* handle stream copy */
2157     if (!ist->decoding_needed) {
2158         rate_emu_sleep(ist);
2159         ist->last_dts = ist->next_dts;
2160         switch (ist->st->codec->codec_type) {
2161         case AVMEDIA_TYPE_AUDIO:
2162             ist->next_dts += ((int64_t)AV_TIME_BASE * ist->st->codec->frame_size) /
2163                              ist->st->codec->sample_rate;
2164             break;
2165         case AVMEDIA_TYPE_VIDEO:
2166             if (ist->st->codec->time_base.num != 0) {
2167                 int ticks = ist->st->parser ? ist->st->parser->repeat_pict + 1 : ist->st->codec->ticks_per_frame;
2168                 ist->next_dts += ((int64_t)AV_TIME_BASE *
2169                                   ist->st->codec->time_base.num * ticks) /
2170                                   ist->st->codec->time_base.den;
2171             }
2172             break;
2173         }
2174     }
2175     for (i = 0; pkt && i < nb_ostreams; i++) {
2176         OutputStream *ost = &ost_table[i];
2177
2178         if (!check_output_constraints(ist, ost) || ost->encoding_needed)
2179             continue;
2180
2181         do_streamcopy(ist, ost, pkt);
2182     }
2183
2184     return 0;
2185 }
2186
2187 static void print_sdp(OutputFile *output_files, int n)
2188 {
2189     char sdp[2048];
2190     int i;
2191     AVFormatContext **avc = av_malloc(sizeof(*avc) * n);
2192
2193     if (!avc)
2194         exit_program(1);
2195     for (i = 0; i < n; i++)
2196         avc[i] = output_files[i].ctx;
2197
2198     av_sdp_create(avc, n, sdp, sizeof(sdp));
2199     printf("SDP:\n%s\n", sdp);
2200     fflush(stdout);
2201     av_freep(&avc);
2202 }
2203
2204 static int init_input_stream(int ist_index, OutputStream *output_streams, int nb_output_streams,
2205                              char *error, int error_len)
2206 {
2207     int i;
2208     InputStream *ist = &input_streams[ist_index];
2209     if (ist->decoding_needed) {
2210         AVCodec *codec = ist->dec;
2211         if (!codec) {
2212             snprintf(error, error_len, "Decoder (codec id %d) not found for input stream #%d:%d",
2213                     ist->st->codec->codec_id, ist->file_index, ist->st->index);
2214             return AVERROR(EINVAL);
2215         }
2216
2217         /* update requested sample format for the decoder based on the
2218            corresponding encoder sample format */
2219         for (i = 0; i < nb_output_streams; i++) {
2220             OutputStream *ost = &output_streams[i];
2221             if (ost->source_index == ist_index) {
2222                 update_sample_fmt(ist->st->codec, codec, ost->st->codec);
2223                 break;
2224             }
2225         }
2226
2227         if (codec->type == AVMEDIA_TYPE_VIDEO && codec->capabilities & CODEC_CAP_DR1) {
2228             ist->st->codec->get_buffer     = codec_get_buffer;
2229             ist->st->codec->release_buffer = codec_release_buffer;
2230             ist->st->codec->opaque         = ist;
2231         }
2232
2233         if (!av_dict_get(ist->opts, "threads", NULL, 0))
2234             av_dict_set(&ist->opts, "threads", "auto", 0);
2235         if (avcodec_open2(ist->st->codec, codec, &ist->opts) < 0) {
2236             snprintf(error, error_len, "Error while opening decoder for input stream #%d:%d",
2237                     ist->file_index, ist->st->index);
2238             return AVERROR(EINVAL);
2239         }
2240         assert_codec_experimental(ist->st->codec, 0);
2241         assert_avoptions(ist->opts);
2242     }
2243
2244     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;
2245     ist->next_dts = AV_NOPTS_VALUE;
2246     init_pts_correction(&ist->pts_ctx);
2247     ist->is_start = 1;
2248
2249     return 0;
2250 }
2251
2252 static int transcode_init(OutputFile *output_files,
2253                           int nb_output_files,
2254                           InputFile *input_files,
2255                           int nb_input_files)
2256 {
2257     int ret = 0, i, j, k;
2258     AVFormatContext *oc;
2259     AVCodecContext *codec, *icodec;
2260     OutputStream *ost;
2261     InputStream *ist;
2262     char error[1024];
2263     int want_sdp = 1;
2264
2265     /* init framerate emulation */
2266     for (i = 0; i < nb_input_files; i++) {
2267         InputFile *ifile = &input_files[i];
2268         if (ifile->rate_emu)
2269             for (j = 0; j < ifile->nb_streams; j++)
2270                 input_streams[j + ifile->ist_index].start = av_gettime();
2271     }
2272
2273     /* output stream init */
2274     for (i = 0; i < nb_output_files; i++) {
2275         oc = output_files[i].ctx;
2276         if (!oc->nb_streams && !(oc->oformat->flags & AVFMT_NOSTREAMS)) {
2277             av_dump_format(oc, i, oc->filename, 1);
2278             av_log(NULL, AV_LOG_ERROR, "Output file #%d does not contain any stream\n", i);
2279             return AVERROR(EINVAL);
2280         }
2281     }
2282
2283     /* for each output stream, we compute the right encoding parameters */
2284     for (i = 0; i < nb_output_streams; i++) {
2285         ost = &output_streams[i];
2286         oc  = output_files[ost->file_index].ctx;
2287         ist = &input_streams[ost->source_index];
2288
2289         if (ost->attachment_filename)
2290             continue;
2291
2292         codec  = ost->st->codec;
2293         icodec = ist->st->codec;
2294
2295         ost->st->disposition          = ist->st->disposition;
2296         codec->bits_per_raw_sample    = icodec->bits_per_raw_sample;
2297         codec->chroma_sample_location = icodec->chroma_sample_location;
2298
2299         if (ost->stream_copy) {
2300             uint64_t extra_size = (uint64_t)icodec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE;
2301
2302             if (extra_size > INT_MAX) {
2303                 return AVERROR(EINVAL);
2304             }
2305
2306             /* if stream_copy is selected, no need to decode or encode */
2307             codec->codec_id   = icodec->codec_id;
2308             codec->codec_type = icodec->codec_type;
2309
2310             if (!codec->codec_tag) {
2311                 if (!oc->oformat->codec_tag ||
2312                      av_codec_get_id (oc->oformat->codec_tag, icodec->codec_tag) == codec->codec_id ||
2313                      av_codec_get_tag(oc->oformat->codec_tag, icodec->codec_id) <= 0)
2314                     codec->codec_tag = icodec->codec_tag;
2315             }
2316
2317             codec->bit_rate       = icodec->bit_rate;
2318             codec->rc_max_rate    = icodec->rc_max_rate;
2319             codec->rc_buffer_size = icodec->rc_buffer_size;
2320             codec->field_order    = icodec->field_order;
2321             codec->extradata      = av_mallocz(extra_size);
2322             if (!codec->extradata) {
2323                 return AVERROR(ENOMEM);
2324             }
2325             memcpy(codec->extradata, icodec->extradata, icodec->extradata_size);
2326             codec->extradata_size = icodec->extradata_size;
2327             if (!copy_tb) {
2328                 codec->time_base      = icodec->time_base;
2329                 codec->time_base.num *= icodec->ticks_per_frame;
2330                 av_reduce(&codec->time_base.num, &codec->time_base.den,
2331                           codec->time_base.num, codec->time_base.den, INT_MAX);
2332             } else
2333                 codec->time_base = ist->st->time_base;
2334
2335             switch (codec->codec_type) {
2336             case AVMEDIA_TYPE_AUDIO:
2337                 if (audio_volume != 256) {
2338                     av_log(NULL, AV_LOG_FATAL, "-acodec copy and -vol are incompatible (frames are not decoded)\n");
2339                     exit_program(1);
2340                 }
2341                 codec->channel_layout     = icodec->channel_layout;
2342                 codec->sample_rate        = icodec->sample_rate;
2343                 codec->channels           = icodec->channels;
2344                 codec->frame_size         = icodec->frame_size;
2345                 codec->audio_service_type = icodec->audio_service_type;
2346                 codec->block_align        = icodec->block_align;
2347                 break;
2348             case AVMEDIA_TYPE_VIDEO:
2349                 codec->pix_fmt            = icodec->pix_fmt;
2350                 codec->width              = icodec->width;
2351                 codec->height             = icodec->height;
2352                 codec->has_b_frames       = icodec->has_b_frames;
2353                 if (!codec->sample_aspect_ratio.num) {
2354                     codec->sample_aspect_ratio   =
2355                     ost->st->sample_aspect_ratio =
2356                         ist->st->sample_aspect_ratio.num ? ist->st->sample_aspect_ratio :
2357                         ist->st->codec->sample_aspect_ratio.num ?
2358                         ist->st->codec->sample_aspect_ratio : (AVRational){0, 1};
2359                 }
2360                 break;
2361             case AVMEDIA_TYPE_SUBTITLE:
2362                 codec->width  = icodec->width;
2363                 codec->height = icodec->height;
2364                 break;
2365             case AVMEDIA_TYPE_DATA:
2366             case AVMEDIA_TYPE_ATTACHMENT:
2367                 break;
2368             default:
2369                 abort();
2370             }
2371         } else {
2372             if (!ost->enc)
2373                 ost->enc = avcodec_find_encoder(ost->st->codec->codec_id);
2374
2375             ist->decoding_needed = 1;
2376             ost->encoding_needed = 1;
2377
2378             switch (codec->codec_type) {
2379             case AVMEDIA_TYPE_AUDIO:
2380                 ost->fifo = av_fifo_alloc(1024);
2381                 if (!ost->fifo) {
2382                     return AVERROR(ENOMEM);
2383                 }
2384                 ost->reformat_pair = MAKE_SFMT_PAIR(AV_SAMPLE_FMT_NONE,AV_SAMPLE_FMT_NONE);
2385
2386                 if (!codec->sample_rate)
2387                     codec->sample_rate = icodec->sample_rate;
2388                 choose_sample_rate(ost->st, ost->enc);
2389                 codec->time_base = (AVRational){ 1, codec->sample_rate };
2390
2391                 if (codec->sample_fmt == AV_SAMPLE_FMT_NONE)
2392                     codec->sample_fmt = icodec->sample_fmt;
2393                 choose_sample_fmt(ost->st, ost->enc);
2394
2395                 if (!codec->channels)
2396                     codec->channels = icodec->channels;
2397                 codec->channel_layout = icodec->channel_layout;
2398                 if (av_get_channel_layout_nb_channels(codec->channel_layout) != codec->channels)
2399                     codec->channel_layout = 0;
2400
2401                 ost->audio_resample       = codec-> sample_rate != icodec->sample_rate || audio_sync_method > 1;
2402                 icodec->request_channels  = codec-> channels;
2403                 ost->resample_sample_fmt  = icodec->sample_fmt;
2404                 ost->resample_sample_rate = icodec->sample_rate;
2405                 ost->resample_channels    = icodec->channels;
2406                 break;
2407             case AVMEDIA_TYPE_VIDEO:
2408                 if (codec->pix_fmt == PIX_FMT_NONE)
2409                     codec->pix_fmt = icodec->pix_fmt;
2410                 choose_pixel_fmt(ost->st, ost->enc);
2411
2412                 if (ost->st->codec->pix_fmt == PIX_FMT_NONE) {
2413                     av_log(NULL, AV_LOG_FATAL, "Video pixel format is unknown, stream cannot be encoded\n");
2414                     exit_program(1);
2415                 }
2416
2417                 if (!codec->width || !codec->height) {
2418                     codec->width  = icodec->width;
2419                     codec->height = icodec->height;
2420                 }
2421
2422                 ost->video_resample = codec->width   != icodec->width  ||
2423                                       codec->height  != icodec->height ||
2424                                       codec->pix_fmt != icodec->pix_fmt;
2425                 if (ost->video_resample) {
2426 #if !CONFIG_AVFILTER
2427                     avcodec_get_frame_defaults(&ost->pict_tmp);
2428                     if (avpicture_alloc((AVPicture*)&ost->pict_tmp, codec->pix_fmt,
2429                                        codec->width, codec->height)) {
2430                         av_log(NULL, AV_LOG_FATAL, "Cannot allocate temp picture, check pix fmt\n");
2431                         exit_program(1);
2432                     }
2433                     ost->img_resample_ctx = sws_getContext(
2434                         icodec->width,
2435                         icodec->height,
2436                         icodec->pix_fmt,
2437                         codec->width,
2438                         codec->height,
2439                         codec->pix_fmt,
2440                         ost->sws_flags, NULL, NULL, NULL);
2441                     if (ost->img_resample_ctx == NULL) {
2442                         av_log(NULL, AV_LOG_FATAL, "Cannot get resampling context\n");
2443                         exit_program(1);
2444                     }
2445 #endif
2446                     codec->bits_per_raw_sample = 0;
2447                 }
2448
2449                 ost->resample_height  = icodec->height;
2450                 ost->resample_width   = icodec->width;
2451                 ost->resample_pix_fmt = icodec->pix_fmt;
2452
2453                 /*
2454                  * We want CFR output if and only if one of those is true:
2455                  * 1) user specified output framerate with -r
2456                  * 2) user specified -vsync cfr
2457                  * 3) output format is CFR and the user didn't force vsync to
2458                  *    something else than CFR
2459                  *
2460                  * in such a case, set ost->frame_rate
2461                  */
2462                 if (!ost->frame_rate.num &&
2463                     (video_sync_method ==  VSYNC_CFR ||
2464                      (video_sync_method ==  VSYNC_AUTO &&
2465                       !(oc->oformat->flags & (AVFMT_NOTIMESTAMPS | AVFMT_VARIABLE_FPS))))) {
2466                     ost->frame_rate = ist->st->r_frame_rate.num ? ist->st->r_frame_rate : (AVRational){25, 1};
2467                     if (ost->enc && ost->enc->supported_framerates && !ost->force_fps) {
2468                         int idx = av_find_nearest_q_idx(ost->frame_rate, ost->enc->supported_framerates);
2469                         ost->frame_rate = ost->enc->supported_framerates[idx];
2470                     }
2471                 }
2472                 if (ost->frame_rate.num) {
2473                     codec->time_base = (AVRational){ost->frame_rate.den, ost->frame_rate.num};
2474                     video_sync_method = VSYNC_CFR;
2475                 } else
2476                     codec->time_base = ist->st->time_base;
2477
2478 #if CONFIG_AVFILTER
2479                 if (configure_video_filters(ist, ost)) {
2480                     av_log(NULL, AV_LOG_FATAL, "Error opening filters!\n");
2481                     exit(1);
2482                 }
2483 #endif
2484                 break;
2485             case AVMEDIA_TYPE_SUBTITLE:
2486                 codec->time_base = (AVRational){1, 1000};
2487                 break;
2488             default:
2489                 abort();
2490                 break;
2491             }
2492             /* two pass mode */
2493             if ((codec->flags & (CODEC_FLAG_PASS1 | CODEC_FLAG_PASS2))) {
2494                 char logfilename[1024];
2495                 FILE *f;
2496
2497                 snprintf(logfilename, sizeof(logfilename), "%s-%d.log",
2498                          pass_logfilename_prefix ? pass_logfilename_prefix : DEFAULT_PASS_LOGFILENAME_PREFIX,
2499                          i);
2500                 if (codec->flags & CODEC_FLAG_PASS1) {
2501                     f = fopen(logfilename, "wb");
2502                     if (!f) {
2503                         av_log(NULL, AV_LOG_FATAL, "Cannot write log file '%s' for pass-1 encoding: %s\n",
2504                                logfilename, strerror(errno));
2505                         exit_program(1);
2506                     }
2507                     ost->logfile = f;
2508                 } else {
2509                     char  *logbuffer;
2510                     size_t logbuffer_size;
2511                     if (cmdutils_read_file(logfilename, &logbuffer, &logbuffer_size) < 0) {
2512                         av_log(NULL, AV_LOG_FATAL, "Error reading log file '%s' for pass-2 encoding\n",
2513                                logfilename);
2514                         exit_program(1);
2515                     }
2516                     codec->stats_in = logbuffer;
2517                 }
2518             }
2519         }
2520     }
2521
2522     /* open each encoder */
2523     for (i = 0; i < nb_output_streams; i++) {
2524         ost = &output_streams[i];
2525         if (ost->encoding_needed) {
2526             AVCodec      *codec = ost->enc;
2527             AVCodecContext *dec = input_streams[ost->source_index].st->codec;
2528             if (!codec) {
2529                 snprintf(error, sizeof(error), "Encoder (codec id %d) not found for output stream #%d:%d",
2530                          ost->st->codec->codec_id, ost->file_index, ost->index);
2531                 ret = AVERROR(EINVAL);
2532                 goto dump_format;
2533             }
2534             if (dec->subtitle_header) {
2535                 ost->st->codec->subtitle_header = av_malloc(dec->subtitle_header_size);
2536                 if (!ost->st->codec->subtitle_header) {
2537                     ret = AVERROR(ENOMEM);
2538                     goto dump_format;
2539                 }
2540                 memcpy(ost->st->codec->subtitle_header, dec->subtitle_header, dec->subtitle_header_size);
2541                 ost->st->codec->subtitle_header_size = dec->subtitle_header_size;
2542             }
2543             if (!av_dict_get(ost->opts, "threads", NULL, 0))
2544                 av_dict_set(&ost->opts, "threads", "auto", 0);
2545             if (avcodec_open2(ost->st->codec, codec, &ost->opts) < 0) {
2546                 snprintf(error, sizeof(error), "Error while opening encoder for output stream #%d:%d - maybe incorrect parameters such as bit_rate, rate, width or height",
2547                         ost->file_index, ost->index);
2548                 ret = AVERROR(EINVAL);
2549                 goto dump_format;
2550             }
2551             assert_codec_experimental(ost->st->codec, 1);
2552             assert_avoptions(ost->opts);
2553             if (ost->st->codec->bit_rate && ost->st->codec->bit_rate < 1000)
2554                 av_log(NULL, AV_LOG_WARNING, "The bitrate parameter is set too low."
2555                                              "It takes bits/s as argument, not kbits/s\n");
2556             extra_size += ost->st->codec->extradata_size;
2557
2558             if (ost->st->codec->me_threshold)
2559                 input_streams[ost->source_index].st->codec->debug |= FF_DEBUG_MV;
2560         }
2561     }
2562
2563     /* init input streams */
2564     for (i = 0; i < nb_input_streams; i++)
2565         if ((ret = init_input_stream(i, output_streams, nb_output_streams, error, sizeof(error))) < 0)
2566             goto dump_format;
2567
2568     /* discard unused programs */
2569     for (i = 0; i < nb_input_files; i++) {
2570         InputFile *ifile = &input_files[i];
2571         for (j = 0; j < ifile->ctx->nb_programs; j++) {
2572             AVProgram *p = ifile->ctx->programs[j];
2573             int discard  = AVDISCARD_ALL;
2574
2575             for (k = 0; k < p->nb_stream_indexes; k++)
2576                 if (!input_streams[ifile->ist_index + p->stream_index[k]].discard) {
2577                     discard = AVDISCARD_DEFAULT;
2578                     break;
2579                 }
2580             p->discard = discard;
2581         }
2582     }
2583
2584     /* open files and write file headers */
2585     for (i = 0; i < nb_output_files; i++) {
2586         oc = output_files[i].ctx;
2587         oc->interrupt_callback = int_cb;
2588         if (avformat_write_header(oc, &output_files[i].opts) < 0) {
2589             snprintf(error, sizeof(error), "Could not write header for output file #%d (incorrect codec parameters ?)", i);
2590             ret = AVERROR(EINVAL);
2591             goto dump_format;
2592         }
2593         assert_avoptions(output_files[i].opts);
2594         if (strcmp(oc->oformat->name, "rtp")) {
2595             want_sdp = 0;
2596         }
2597     }
2598
2599  dump_format:
2600     /* dump the file output parameters - cannot be done before in case
2601        of stream copy */
2602     for (i = 0; i < nb_output_files; i++) {
2603         av_dump_format(output_files[i].ctx, i, output_files[i].ctx->filename, 1);
2604     }
2605
2606     /* dump the stream mapping */
2607     av_log(NULL, AV_LOG_INFO, "Stream mapping:\n");
2608     for (i = 0; i < nb_output_streams; i++) {
2609         ost = &output_streams[i];
2610
2611         if (ost->attachment_filename) {
2612             /* an attached file */
2613             av_log(NULL, AV_LOG_INFO, "  File %s -> Stream #%d:%d\n",
2614                    ost->attachment_filename, ost->file_index, ost->index);
2615             continue;
2616         }
2617         av_log(NULL, AV_LOG_INFO, "  Stream #%d:%d -> #%d:%d",
2618                input_streams[ost->source_index].file_index,
2619                input_streams[ost->source_index].st->index,
2620                ost->file_index,
2621                ost->index);
2622         if (ost->sync_ist != &input_streams[ost->source_index])
2623             av_log(NULL, AV_LOG_INFO, " [sync #%d:%d]",
2624                    ost->sync_ist->file_index,
2625                    ost->sync_ist->st->index);
2626         if (ost->stream_copy)
2627             av_log(NULL, AV_LOG_INFO, " (copy)");
2628         else
2629             av_log(NULL, AV_LOG_INFO, " (%s -> %s)", input_streams[ost->source_index].dec ?
2630                    input_streams[ost->source_index].dec->name : "?",
2631                    ost->enc ? ost->enc->name : "?");
2632         av_log(NULL, AV_LOG_INFO, "\n");
2633     }
2634
2635     if (ret) {
2636         av_log(NULL, AV_LOG_ERROR, "%s\n", error);
2637         return ret;
2638     }
2639
2640     if (want_sdp) {
2641         print_sdp(output_files, nb_output_files);
2642     }
2643
2644     return 0;
2645 }
2646
2647 /*
2648  * The following code is the main loop of the file converter
2649  */
2650 static int transcode(OutputFile *output_files,
2651                      int nb_output_files,
2652                      InputFile *input_files,
2653                      int nb_input_files)
2654 {
2655     int ret, i;
2656     AVFormatContext *is, *os;
2657     OutputStream *ost;
2658     InputStream *ist;
2659     uint8_t *no_packet;
2660     int no_packet_count = 0;
2661     int64_t timer_start;
2662
2663     if (!(no_packet = av_mallocz(nb_input_files)))
2664         exit_program(1);
2665
2666     ret = transcode_init(output_files, nb_output_files, input_files, nb_input_files);
2667     if (ret < 0)
2668         goto fail;
2669
2670     av_log(NULL, AV_LOG_INFO, "Press ctrl-c to stop encoding\n");
2671     term_init();
2672
2673     timer_start = av_gettime();
2674
2675     for (; received_sigterm == 0;) {
2676         int file_index, ist_index;
2677         AVPacket pkt;
2678         int64_t ipts_min;
2679         double opts_min;
2680
2681         ipts_min = INT64_MAX;
2682         opts_min = 1e100;
2683
2684         /* select the stream that we must read now by looking at the
2685            smallest output pts */
2686         file_index = -1;
2687         for (i = 0; i < nb_output_streams; i++) {
2688             OutputFile *of;
2689             int64_t ipts;
2690             double  opts;
2691             ost = &output_streams[i];
2692             of = &output_files[ost->file_index];
2693             os = output_files[ost->file_index].ctx;
2694             ist = &input_streams[ost->source_index];
2695             if (ost->is_past_recording_time || no_packet[ist->file_index] ||
2696                 (os->pb && avio_tell(os->pb) >= of->limit_filesize))
2697                 continue;
2698             opts = ost->st->pts.val * av_q2d(ost->st->time_base);
2699             ipts = ist->last_dts;
2700             if (!input_files[ist->file_index].eof_reached) {
2701                 if (ipts < ipts_min) {
2702                     ipts_min = ipts;
2703                     if (input_sync)
2704                         file_index = ist->file_index;
2705                 }
2706                 if (opts < opts_min) {
2707                     opts_min = opts;
2708                     if (!input_sync) file_index = ist->file_index;
2709                 }
2710             }
2711             if (ost->frame_number >= ost->max_frames) {
2712                 int j;
2713                 for (j = 0; j < of->ctx->nb_streams; j++)
2714                     output_streams[of->ost_index + j].is_past_recording_time = 1;
2715                 continue;
2716             }
2717         }
2718         /* if none, if is finished */
2719         if (file_index < 0) {
2720             if (no_packet_count) {
2721                 no_packet_count = 0;
2722                 memset(no_packet, 0, nb_input_files);
2723                 usleep(10000);
2724                 continue;
2725             }
2726             break;
2727         }
2728
2729         /* read a frame from it and output it in the fifo */
2730         is  = input_files[file_index].ctx;
2731         ret = av_read_frame(is, &pkt);
2732         if (ret == AVERROR(EAGAIN)) {
2733             no_packet[file_index] = 1;
2734             no_packet_count++;
2735             continue;
2736         }
2737         if (ret < 0) {
2738             input_files[file_index].eof_reached = 1;
2739             if (opt_shortest)
2740                 break;
2741             else
2742                 continue;
2743         }
2744
2745         no_packet_count = 0;
2746         memset(no_packet, 0, nb_input_files);
2747
2748         if (do_pkt_dump) {
2749             av_pkt_dump_log2(NULL, AV_LOG_DEBUG, &pkt, do_hex_dump,
2750                              is->streams[pkt.stream_index]);
2751         }
2752         /* the following test is needed in case new streams appear
2753            dynamically in stream : we ignore them */
2754         if (pkt.stream_index >= input_files[file_index].nb_streams)
2755             goto discard_packet;
2756         ist_index = input_files[file_index].ist_index + pkt.stream_index;
2757         ist = &input_streams[ist_index];
2758         if (ist->discard)
2759             goto discard_packet;
2760
2761         if (pkt.dts != AV_NOPTS_VALUE)
2762             pkt.dts += av_rescale_q(input_files[ist->file_index].ts_offset, AV_TIME_BASE_Q, ist->st->time_base);
2763         if (pkt.pts != AV_NOPTS_VALUE)
2764             pkt.pts += av_rescale_q(input_files[ist->file_index].ts_offset, AV_TIME_BASE_Q, ist->st->time_base);
2765
2766         if (pkt.pts != AV_NOPTS_VALUE)
2767             pkt.pts *= ist->ts_scale;
2768         if (pkt.dts != AV_NOPTS_VALUE)
2769             pkt.dts *= ist->ts_scale;
2770
2771         //fprintf(stderr, "next:%"PRId64" dts:%"PRId64" off:%"PRId64" %d\n",
2772         //        ist->next_dts,
2773         //        pkt.dts, input_files[ist->file_index].ts_offset,
2774         //        ist->st->codec->codec_type);
2775         if (pkt.dts != AV_NOPTS_VALUE && ist->next_dts != AV_NOPTS_VALUE
2776             && (is->iformat->flags & AVFMT_TS_DISCONT)) {
2777             int64_t pkt_dts = av_rescale_q(pkt.dts, ist->st->time_base, AV_TIME_BASE_Q);
2778             int64_t delta   = pkt_dts - ist->next_dts;
2779             if ((FFABS(delta) > 1LL * dts_delta_threshold * AV_TIME_BASE || pkt_dts + 1 < ist->last_dts) && !copy_ts) {
2780                 input_files[ist->file_index].ts_offset -= delta;
2781                 av_log(NULL, AV_LOG_DEBUG,
2782                        "timestamp discontinuity %"PRId64", new offset= %"PRId64"\n",
2783                        delta, input_files[ist->file_index].ts_offset);
2784                 pkt.dts-= av_rescale_q(delta, AV_TIME_BASE_Q, ist->st->time_base);
2785                 if (pkt.pts != AV_NOPTS_VALUE)
2786                     pkt.pts-= av_rescale_q(delta, AV_TIME_BASE_Q, ist->st->time_base);
2787             }
2788         }
2789
2790         // fprintf(stderr,"read #%d.%d size=%d\n", ist->file_index, ist->st->index, pkt.size);
2791         if (output_packet(ist, output_streams, nb_output_streams, &pkt) < 0) {
2792
2793             av_log(NULL, AV_LOG_ERROR, "Error while decoding stream #%d:%d\n",
2794                    ist->file_index, ist->st->index);
2795             if (exit_on_error)
2796                 exit_program(1);
2797             av_free_packet(&pkt);
2798             continue;
2799         }
2800
2801     discard_packet:
2802         av_free_packet(&pkt);
2803
2804         /* dump report by using the output first video and audio streams */
2805         print_report(output_files, output_streams, nb_output_streams, 0, timer_start);
2806     }
2807
2808     /* at the end of stream, we must flush the decoder buffers */
2809     for (i = 0; i < nb_input_streams; i++) {
2810         ist = &input_streams[i];
2811         if (ist->decoding_needed) {
2812             output_packet(ist, output_streams, nb_output_streams, NULL);
2813         }
2814     }
2815     flush_encoders(output_streams, nb_output_streams);
2816
2817     term_exit();
2818
2819     /* write the trailer if needed and close file */
2820     for (i = 0; i < nb_output_files; i++) {
2821         os = output_files[i].ctx;
2822         av_write_trailer(os);
2823     }
2824
2825     /* dump report by using the first video and audio streams */
2826     print_report(output_files, output_streams, nb_output_streams, 1, timer_start);
2827
2828     /* close each encoder */
2829     for (i = 0; i < nb_output_streams; i++) {
2830         ost = &output_streams[i];
2831         if (ost->encoding_needed) {
2832             av_freep(&ost->st->codec->stats_in);
2833             avcodec_close(ost->st->codec);
2834         }
2835 #if CONFIG_AVFILTER
2836         avfilter_graph_free(&ost->graph);
2837 #endif
2838     }
2839
2840     /* close each decoder */
2841     for (i = 0; i < nb_input_streams; i++) {
2842         ist = &input_streams[i];
2843         if (ist->decoding_needed) {
2844             avcodec_close(ist->st->codec);
2845         }
2846     }
2847
2848     /* finished ! */
2849     ret = 0;
2850
2851  fail:
2852     av_freep(&no_packet);
2853
2854     if (output_streams) {
2855         for (i = 0; i < nb_output_streams; i++) {
2856             ost = &output_streams[i];
2857             if (ost) {
2858                 if (ost->stream_copy)
2859                     av_freep(&ost->st->codec->extradata);
2860                 if (ost->logfile) {
2861                     fclose(ost->logfile);
2862                     ost->logfile = NULL;
2863                 }
2864                 av_fifo_free(ost->fifo); /* works even if fifo is not
2865                                              initialized but set to zero */
2866                 av_freep(&ost->st->codec->subtitle_header);
2867                 av_free(ost->pict_tmp.data[0]);
2868                 av_free(ost->forced_kf_pts);
2869                 if (ost->video_resample)
2870                     sws_freeContext(ost->img_resample_ctx);
2871                 if (ost->resample)
2872                     audio_resample_close(ost->resample);
2873                 if (ost->reformat_ctx)
2874                     av_audio_convert_free(ost->reformat_ctx);
2875                 av_dict_free(&ost->opts);
2876             }
2877         }
2878     }
2879     return ret;
2880 }
2881
2882 static double parse_frame_aspect_ratio(const char *arg)
2883 {
2884     int x = 0, y = 0;
2885     double ar = 0;
2886     const char *p;
2887     char *end;
2888
2889     p = strchr(arg, ':');
2890     if (p) {
2891         x = strtol(arg, &end, 10);
2892         if (end == p)
2893             y = strtol(end + 1, &end, 10);
2894         if (x > 0 && y > 0)
2895             ar = (double)x / (double)y;
2896     } else
2897         ar = strtod(arg, NULL);
2898
2899     if (!ar) {
2900         av_log(NULL, AV_LOG_FATAL, "Incorrect aspect ratio specification.\n");
2901         exit_program(1);
2902     }
2903     return ar;
2904 }
2905
2906 static int opt_audio_codec(OptionsContext *o, const char *opt, const char *arg)
2907 {
2908     return parse_option(o, "codec:a", arg, options);
2909 }
2910
2911 static int opt_video_codec(OptionsContext *o, const char *opt, const char *arg)
2912 {
2913     return parse_option(o, "codec:v", arg, options);
2914 }
2915
2916 static int opt_subtitle_codec(OptionsContext *o, const char *opt, const char *arg)
2917 {
2918     return parse_option(o, "codec:s", arg, options);
2919 }
2920
2921 static int opt_data_codec(OptionsContext *o, const char *opt, const char *arg)
2922 {
2923     return parse_option(o, "codec:d", arg, options);
2924 }
2925
2926 static int opt_map(OptionsContext *o, const char *opt, const char *arg)
2927 {
2928     StreamMap *m = NULL;
2929     int i, negative = 0, file_idx;
2930     int sync_file_idx = -1, sync_stream_idx;
2931     char *p, *sync;
2932     char *map;
2933
2934     if (*arg == '-') {
2935         negative = 1;
2936         arg++;
2937     }
2938     map = av_strdup(arg);
2939
2940     /* parse sync stream first, just pick first matching stream */
2941     if (sync = strchr(map, ',')) {
2942         *sync = 0;
2943         sync_file_idx = strtol(sync + 1, &sync, 0);
2944         if (sync_file_idx >= nb_input_files || sync_file_idx < 0) {
2945             av_log(NULL, AV_LOG_FATAL, "Invalid sync file index: %d.\n", sync_file_idx);
2946             exit_program(1);
2947         }
2948         if (*sync)
2949             sync++;
2950         for (i = 0; i < input_files[sync_file_idx].nb_streams; i++)
2951             if (check_stream_specifier(input_files[sync_file_idx].ctx,
2952                                        input_files[sync_file_idx].ctx->streams[i], sync) == 1) {
2953                 sync_stream_idx = i;
2954                 break;
2955             }
2956         if (i == input_files[sync_file_idx].nb_streams) {
2957             av_log(NULL, AV_LOG_FATAL, "Sync stream specification in map %s does not "
2958                                        "match any streams.\n", arg);
2959             exit_program(1);
2960         }
2961     }
2962
2963
2964     file_idx = strtol(map, &p, 0);
2965     if (file_idx >= nb_input_files || file_idx < 0) {
2966         av_log(NULL, AV_LOG_FATAL, "Invalid input file index: %d.\n", file_idx);
2967         exit_program(1);
2968     }
2969     if (negative)
2970         /* disable some already defined maps */
2971         for (i = 0; i < o->nb_stream_maps; i++) {
2972             m = &o->stream_maps[i];
2973             if (file_idx == m->file_index &&
2974                 check_stream_specifier(input_files[m->file_index].ctx,
2975                                        input_files[m->file_index].ctx->streams[m->stream_index],
2976                                        *p == ':' ? p + 1 : p) > 0)
2977                 m->disabled = 1;
2978         }
2979     else
2980         for (i = 0; i < input_files[file_idx].nb_streams; i++) {
2981             if (check_stream_specifier(input_files[file_idx].ctx, input_files[file_idx].ctx->streams[i],
2982                         *p == ':' ? p + 1 : p) <= 0)
2983                 continue;
2984             o->stream_maps = grow_array(o->stream_maps, sizeof(*o->stream_maps),
2985                                         &o->nb_stream_maps, o->nb_stream_maps + 1);
2986             m = &o->stream_maps[o->nb_stream_maps - 1];
2987
2988             m->file_index   = file_idx;
2989             m->stream_index = i;
2990
2991             if (sync_file_idx >= 0) {
2992                 m->sync_file_index   = sync_file_idx;
2993                 m->sync_stream_index = sync_stream_idx;
2994             } else {
2995                 m->sync_file_index   = file_idx;
2996                 m->sync_stream_index = i;
2997             }
2998         }
2999
3000     if (!m) {
3001         av_log(NULL, AV_LOG_FATAL, "Stream map '%s' matches no streams.\n", arg);
3002         exit_program(1);
3003     }
3004
3005     av_freep(&map);
3006     return 0;
3007 }
3008
3009 static int opt_attach(OptionsContext *o, const char *opt, const char *arg)
3010 {
3011     o->attachments = grow_array(o->attachments, sizeof(*o->attachments),
3012                                 &o->nb_attachments, o->nb_attachments + 1);
3013     o->attachments[o->nb_attachments - 1] = arg;
3014     return 0;
3015 }
3016
3017 /**
3018  * Parse a metadata specifier in arg.
3019  * @param type metadata type is written here -- g(lobal)/s(tream)/c(hapter)/p(rogram)
3020  * @param index for type c/p, chapter/program index is written here
3021  * @param stream_spec for type s, the stream specifier is written here
3022  */
3023 static void parse_meta_type(char *arg, char *type, int *index, const char **stream_spec)
3024 {
3025     if (*arg) {
3026         *type = *arg;
3027         switch (*arg) {
3028         case 'g':
3029             break;
3030         case 's':
3031             if (*(++arg) && *arg != ':') {
3032                 av_log(NULL, AV_LOG_FATAL, "Invalid metadata specifier %s.\n", arg);
3033                 exit_program(1);
3034             }
3035             *stream_spec = *arg == ':' ? arg + 1 : "";
3036             break;
3037         case 'c':
3038         case 'p':
3039             if (*(++arg) == ':')
3040                 *index = strtol(++arg, NULL, 0);
3041             break;
3042         default:
3043             av_log(NULL, AV_LOG_FATAL, "Invalid metadata type %c.\n", *arg);
3044             exit_program(1);
3045         }
3046     } else
3047         *type = 'g';
3048 }
3049
3050 static int copy_metadata(char *outspec, char *inspec, AVFormatContext *oc, AVFormatContext *ic, OptionsContext *o)
3051 {
3052     AVDictionary **meta_in = NULL;
3053     AVDictionary **meta_out;
3054     int i, ret = 0;
3055     char type_in, type_out;
3056     const char *istream_spec = NULL, *ostream_spec = NULL;
3057     int idx_in = 0, idx_out = 0;
3058
3059     parse_meta_type(inspec,  &type_in,  &idx_in,  &istream_spec);
3060     parse_meta_type(outspec, &type_out, &idx_out, &ostream_spec);
3061
3062     if (type_in == 'g' || type_out == 'g')
3063         o->metadata_global_manual = 1;
3064     if (type_in == 's' || type_out == 's')
3065         o->metadata_streams_manual = 1;
3066     if (type_in == 'c' || type_out == 'c')
3067         o->metadata_chapters_manual = 1;
3068
3069 #define METADATA_CHECK_INDEX(index, nb_elems, desc)\
3070     if ((index) < 0 || (index) >= (nb_elems)) {\
3071         av_log(NULL, AV_LOG_FATAL, "Invalid %s index %d while processing metadata maps.\n",\
3072                 (desc), (index));\
3073         exit_program(1);\
3074     }
3075
3076 #define SET_DICT(type, meta, context, index)\
3077         switch (type) {\
3078         case 'g':\
3079             meta = &context->metadata;\
3080             break;\
3081         case 'c':\
3082             METADATA_CHECK_INDEX(index, context->nb_chapters, "chapter")\
3083             meta = &context->chapters[index]->metadata;\
3084             break;\
3085         case 'p':\
3086             METADATA_CHECK_INDEX(index, context->nb_programs, "program")\
3087             meta = &context->programs[index]->metadata;\
3088             break;\
3089         }\
3090
3091     SET_DICT(type_in, meta_in, ic, idx_in);
3092     SET_DICT(type_out, meta_out, oc, idx_out);
3093
3094     /* for input streams choose first matching stream */
3095     if (type_in == 's') {
3096         for (i = 0; i < ic->nb_streams; i++) {
3097             if ((ret = check_stream_specifier(ic, ic->streams[i], istream_spec)) > 0) {
3098                 meta_in = &ic->streams[i]->metadata;
3099                 break;
3100             } else if (ret < 0)
3101                 exit_program(1);
3102         }
3103         if (!meta_in) {
3104             av_log(NULL, AV_LOG_FATAL, "Stream specifier %s does not match  any streams.\n", istream_spec);
3105             exit_program(1);
3106         }
3107     }
3108
3109     if (type_out == 's') {
3110         for (i = 0; i < oc->nb_streams; i++) {
3111             if ((ret = check_stream_specifier(oc, oc->streams[i], ostream_spec)) > 0) {
3112                 meta_out = &oc->streams[i]->metadata;
3113                 av_dict_copy(meta_out, *meta_in, AV_DICT_DONT_OVERWRITE);
3114             } else if (ret < 0)
3115                 exit_program(1);
3116         }
3117     } else
3118         av_dict_copy(meta_out, *meta_in, AV_DICT_DONT_OVERWRITE);
3119
3120     return 0;
3121 }
3122
3123 static AVCodec *find_codec_or_die(const char *name, enum AVMediaType type, int encoder)
3124 {
3125     const char *codec_string = encoder ? "encoder" : "decoder";
3126     AVCodec *codec;
3127
3128     codec = encoder ?
3129         avcodec_find_encoder_by_name(name) :
3130         avcodec_find_decoder_by_name(name);
3131     if (!codec) {
3132         av_log(NULL, AV_LOG_FATAL, "Unknown %s '%s'\n", codec_string, name);
3133         exit_program(1);
3134     }
3135     if (codec->type != type) {
3136         av_log(NULL, AV_LOG_FATAL, "Invalid %s type '%s'\n", codec_string, name);
3137         exit_program(1);
3138     }
3139     return codec;
3140 }
3141
3142 static AVCodec *choose_decoder(OptionsContext *o, AVFormatContext *s, AVStream *st)
3143 {
3144     char *codec_name = NULL;
3145
3146     MATCH_PER_STREAM_OPT(codec_names, str, codec_name, s, st);
3147     if (codec_name) {
3148         AVCodec *codec = find_codec_or_die(codec_name, st->codec->codec_type, 0);
3149         st->codec->codec_id = codec->id;
3150         return codec;
3151     } else
3152         return avcodec_find_decoder(st->codec->codec_id);
3153 }
3154
3155 /**
3156  * Add all the streams from the given input file to the global
3157  * list of input streams.
3158  */
3159 static void add_input_streams(OptionsContext *o, AVFormatContext *ic)
3160 {
3161     int i;
3162
3163     for (i = 0; i < ic->nb_streams; i++) {
3164         AVStream *st = ic->streams[i];
3165         AVCodecContext *dec = st->codec;
3166         InputStream *ist;
3167
3168         input_streams = grow_array(input_streams, sizeof(*input_streams), &nb_input_streams, nb_input_streams + 1);
3169         ist = &input_streams[nb_input_streams - 1];
3170         ist->st = st;
3171         ist->file_index = nb_input_files;
3172         ist->discard = 1;
3173         ist->opts = filter_codec_opts(codec_opts, ist->st->codec->codec_id, ic, st);
3174
3175         ist->ts_scale = 1.0;
3176         MATCH_PER_STREAM_OPT(ts_scale, dbl, ist->ts_scale, ic, st);
3177
3178         ist->dec = choose_decoder(o, ic, st);
3179
3180         switch (dec->codec_type) {
3181         case AVMEDIA_TYPE_AUDIO:
3182             if (o->audio_disable)
3183                 st->discard = AVDISCARD_ALL;
3184             break;
3185         case AVMEDIA_TYPE_VIDEO:
3186             if (dec->lowres) {
3187                 dec->flags |= CODEC_FLAG_EMU_EDGE;
3188                 dec->height >>= dec->lowres;
3189                 dec->width  >>= dec->lowres;
3190             }
3191
3192             if (o->video_disable)
3193                 st->discard = AVDISCARD_ALL;
3194             else if (video_discard)
3195                 st->discard = video_discard;
3196             break;
3197         case AVMEDIA_TYPE_DATA:
3198             break;
3199         case AVMEDIA_TYPE_SUBTITLE:
3200             if (o->subtitle_disable)
3201                 st->discard = AVDISCARD_ALL;
3202             break;
3203         case AVMEDIA_TYPE_ATTACHMENT:
3204         case AVMEDIA_TYPE_UNKNOWN:
3205             break;
3206         default:
3207             abort();
3208         }
3209     }
3210 }
3211
3212 static void assert_file_overwrite(const char *filename)
3213 {
3214     if (!file_overwrite &&
3215         (strchr(filename, ':') == NULL || filename[1] == ':' ||
3216          av_strstart(filename, "file:", NULL))) {
3217         if (avio_check(filename, 0) == 0) {
3218             if (!using_stdin) {
3219                 fprintf(stderr,"File '%s' already exists. Overwrite ? [y/N] ", filename);
3220                 fflush(stderr);
3221                 if (!read_yesno()) {
3222                     fprintf(stderr, "Not overwriting - exiting\n");
3223                     exit_program(1);
3224                 }
3225             }
3226             else {
3227                 fprintf(stderr,"File '%s' already exists. Exiting.\n", filename);
3228                 exit_program(1);
3229             }
3230         }
3231     }
3232 }
3233
3234 static void dump_attachment(AVStream *st, const char *filename)
3235 {
3236     int ret;
3237     AVIOContext *out = NULL;
3238     AVDictionaryEntry *e;
3239
3240     if (!st->codec->extradata_size) {
3241         av_log(NULL, AV_LOG_WARNING, "No extradata to dump in stream #%d:%d.\n",
3242                nb_input_files - 1, st->index);
3243         return;
3244     }
3245     if (!*filename && (e = av_dict_get(st->metadata, "filename", NULL, 0)))
3246         filename = e->value;
3247     if (!*filename) {
3248         av_log(NULL, AV_LOG_FATAL, "No filename specified and no 'filename' tag"
3249                "in stream #%d:%d.\n", nb_input_files - 1, st->index);
3250         exit_program(1);
3251     }
3252
3253     assert_file_overwrite(filename);
3254
3255     if ((ret = avio_open2(&out, filename, AVIO_FLAG_WRITE, &int_cb, NULL)) < 0) {
3256         av_log(NULL, AV_LOG_FATAL, "Could not open file %s for writing.\n",
3257                filename);
3258         exit_program(1);
3259     }
3260
3261     avio_write(out, st->codec->extradata, st->codec->extradata_size);
3262     avio_flush(out);
3263     avio_close(out);
3264 }
3265
3266 static int opt_input_file(OptionsContext *o, const char *opt, const char *filename)
3267 {
3268     AVFormatContext *ic;
3269     AVInputFormat *file_iformat = NULL;
3270     int err, i, ret;
3271     int64_t timestamp;
3272     uint8_t buf[128];
3273     AVDictionary **opts;
3274     int orig_nb_streams;                     // number of streams before avformat_find_stream_info
3275
3276     if (o->format) {
3277         if (!(file_iformat = av_find_input_format(o->format))) {
3278             av_log(NULL, AV_LOG_FATAL, "Unknown input format: '%s'\n", o->format);
3279             exit_program(1);
3280         }
3281     }
3282
3283     if (!strcmp(filename, "-"))
3284         filename = "pipe:";
3285
3286     using_stdin |= !strncmp(filename, "pipe:", 5) ||
3287                     !strcmp(filename, "/dev/stdin");
3288
3289     /* get default parameters from command line */
3290     ic = avformat_alloc_context();
3291     if (!ic) {
3292         print_error(filename, AVERROR(ENOMEM));
3293         exit_program(1);
3294     }
3295     if (o->nb_audio_sample_rate) {
3296         snprintf(buf, sizeof(buf), "%d", o->audio_sample_rate[o->nb_audio_sample_rate - 1].u.i);
3297         av_dict_set(&format_opts, "sample_rate", buf, 0);
3298     }
3299     if (o->nb_audio_channels) {
3300         snprintf(buf, sizeof(buf), "%d", o->audio_channels[o->nb_audio_channels - 1].u.i);
3301         av_dict_set(&format_opts, "channels", buf, 0);
3302     }
3303     if (o->nb_frame_rates) {
3304         av_dict_set(&format_opts, "framerate", o->frame_rates[o->nb_frame_rates - 1].u.str, 0);
3305     }
3306     if (o->nb_frame_sizes) {
3307         av_dict_set(&format_opts, "video_size", o->frame_sizes[o->nb_frame_sizes - 1].u.str, 0);
3308     }
3309     if (o->nb_frame_pix_fmts)
3310         av_dict_set(&format_opts, "pixel_format", o->frame_pix_fmts[o->nb_frame_pix_fmts - 1].u.str, 0);
3311
3312     ic->flags |= AVFMT_FLAG_NONBLOCK;
3313     ic->interrupt_callback = int_cb;
3314
3315     /* open the input file with generic libav function */
3316     err = avformat_open_input(&ic, filename, file_iformat, &format_opts);
3317     if (err < 0) {
3318         print_error(filename, err);
3319         exit_program(1);
3320     }
3321     assert_avoptions(format_opts);
3322
3323     /* apply forced codec ids */
3324     for (i = 0; i < ic->nb_streams; i++)
3325         choose_decoder(o, ic, ic->streams[i]);
3326
3327     /* Set AVCodecContext options for avformat_find_stream_info */
3328     opts = setup_find_stream_info_opts(ic, codec_opts);
3329     orig_nb_streams = ic->nb_streams;
3330
3331     /* If not enough info to get the stream parameters, we decode the
3332        first frames to get it. (used in mpeg case for example) */
3333     ret = avformat_find_stream_info(ic, opts);
3334     if (ret < 0) {
3335         av_log(NULL, AV_LOG_FATAL, "%s: could not find codec parameters\n", filename);
3336         avformat_close_input(&ic);
3337         exit_program(1);
3338     }
3339
3340     timestamp = o->start_time;
3341     /* add the stream start time */
3342     if (ic->start_time != AV_NOPTS_VALUE)
3343         timestamp += ic->start_time;
3344
3345     /* if seeking requested, we execute it */
3346     if (o->start_time != 0) {
3347         ret = av_seek_frame(ic, -1, timestamp, AVSEEK_FLAG_BACKWARD);
3348         if (ret < 0) {
3349             av_log(NULL, AV_LOG_WARNING, "%s: could not seek to position %0.3f\n",
3350                    filename, (double)timestamp / AV_TIME_BASE);
3351         }
3352     }
3353
3354     /* update the current parameters so that they match the one of the input stream */
3355     add_input_streams(o, ic);
3356
3357     /* dump the file content */
3358     av_dump_format(ic, nb_input_files, filename, 0);
3359
3360     input_files = grow_array(input_files, sizeof(*input_files), &nb_input_files, nb_input_files + 1);
3361     input_files[nb_input_files - 1].ctx        = ic;
3362     input_files[nb_input_files - 1].ist_index  = nb_input_streams - ic->nb_streams;
3363     input_files[nb_input_files - 1].ts_offset  = o->input_ts_offset - (copy_ts ? 0 : timestamp);
3364     input_files[nb_input_files - 1].nb_streams = ic->nb_streams;
3365     input_files[nb_input_files - 1].rate_emu   = o->rate_emu;
3366
3367     for (i = 0; i < o->nb_dump_attachment; i++) {
3368         int j;
3369
3370         for (j = 0; j < ic->nb_streams; j++) {
3371             AVStream *st = ic->streams[j];
3372
3373             if (check_stream_specifier(ic, st, o->dump_attachment[i].specifier) == 1)
3374                 dump_attachment(st, o->dump_attachment[i].u.str);
3375         }
3376     }
3377
3378     for (i = 0; i < orig_nb_streams; i++)
3379         av_dict_free(&opts[i]);
3380     av_freep(&opts);
3381
3382     reset_options(o);
3383     return 0;
3384 }
3385
3386 static void parse_forced_key_frames(char *kf, OutputStream *ost,
3387                                     AVCodecContext *avctx)
3388 {
3389     char *p;
3390     int n = 1, i;
3391     int64_t t;
3392
3393     for (p = kf; *p; p++)
3394         if (*p == ',')
3395             n++;
3396     ost->forced_kf_count = n;
3397     ost->forced_kf_pts   = av_malloc(sizeof(*ost->forced_kf_pts) * n);
3398     if (!ost->forced_kf_pts) {
3399         av_log(NULL, AV_LOG_FATAL, "Could not allocate forced key frames array.\n");
3400         exit_program(1);
3401     }
3402     for (i = 0; i < n; i++) {
3403         p = i ? strchr(p, ',') + 1 : kf;
3404         t = parse_time_or_die("force_key_frames", p, 1);
3405         ost->forced_kf_pts[i] = av_rescale_q(t, AV_TIME_BASE_Q, avctx->time_base);
3406     }
3407 }
3408
3409 static uint8_t *get_line(AVIOContext *s)
3410 {
3411     AVIOContext *line;
3412     uint8_t *buf;
3413     char c;
3414
3415     if (avio_open_dyn_buf(&line) < 0) {
3416         av_log(NULL, AV_LOG_FATAL, "Could not alloc buffer for reading preset.\n");
3417         exit_program(1);
3418     }
3419
3420     while ((c = avio_r8(s)) && c != '\n')
3421         avio_w8(line, c);
3422     avio_w8(line, 0);
3423     avio_close_dyn_buf(line, &buf);
3424
3425     return buf;
3426 }
3427
3428 static int get_preset_file_2(const char *preset_name, const char *codec_name, AVIOContext **s)
3429 {
3430     int i, ret = 1;
3431     char filename[1000];
3432     const char *base[3] = { getenv("AVCONV_DATADIR"),
3433                             getenv("HOME"),
3434                             AVCONV_DATADIR,
3435                             };
3436
3437     for (i = 0; i < FF_ARRAY_ELEMS(base) && ret; i++) {
3438         if (!base[i])
3439             continue;
3440         if (codec_name) {
3441             snprintf(filename, sizeof(filename), "%s%s/%s-%s.avpreset", base[i],
3442                      i != 1 ? "" : "/.avconv", codec_name, preset_name);
3443             ret = avio_open2(s, filename, AVIO_FLAG_READ, &int_cb, NULL);
3444         }
3445         if (ret) {
3446             snprintf(filename, sizeof(filename), "%s%s/%s.avpreset", base[i],
3447                      i != 1 ? "" : "/.avconv", preset_name);
3448             ret = avio_open2(s, filename, AVIO_FLAG_READ, &int_cb, NULL);
3449         }
3450     }
3451     return ret;
3452 }
3453
3454 static void choose_encoder(OptionsContext *o, AVFormatContext *s, OutputStream *ost)
3455 {
3456     char *codec_name = NULL;
3457
3458     MATCH_PER_STREAM_OPT(codec_names, str, codec_name, s, ost->st);
3459     if (!codec_name) {
3460         ost->st->codec->codec_id = av_guess_codec(s->oformat, NULL, s->filename,
3461                                                   NULL, ost->st->codec->codec_type);
3462         ost->enc = avcodec_find_encoder(ost->st->codec->codec_id);
3463     } else if (!strcmp(codec_name, "copy"))
3464         ost->stream_copy = 1;
3465     else {
3466         ost->enc = find_codec_or_die(codec_name, ost->st->codec->codec_type, 1);
3467         ost->st->codec->codec_id = ost->enc->id;
3468     }
3469 }
3470
3471 static OutputStream *new_output_stream(OptionsContext *o, AVFormatContext *oc, enum AVMediaType type)
3472 {
3473     OutputStream *ost;
3474     AVStream *st = avformat_new_stream(oc, NULL);
3475     int idx      = oc->nb_streams - 1, ret = 0;
3476     char *bsf = NULL, *next, *codec_tag = NULL;
3477     AVBitStreamFilterContext *bsfc, *bsfc_prev = NULL;
3478     double qscale = -1;
3479     char *buf = NULL, *arg = NULL, *preset = NULL;
3480     AVIOContext *s = NULL;
3481
3482     if (!st) {
3483         av_log(NULL, AV_LOG_FATAL, "Could not alloc stream.\n");
3484         exit_program(1);
3485     }
3486
3487     if (oc->nb_streams - 1 < o->nb_streamid_map)
3488         st->id = o->streamid_map[oc->nb_streams - 1];
3489
3490     output_streams = grow_array(output_streams, sizeof(*output_streams), &nb_output_streams,
3491                                 nb_output_streams + 1);
3492     ost = &output_streams[nb_output_streams - 1];
3493     ost->file_index = nb_output_files;
3494     ost->index      = idx;
3495     ost->st         = st;
3496     st->codec->codec_type = type;
3497     choose_encoder(o, oc, ost);
3498     if (ost->enc) {
3499         ost->opts  = filter_codec_opts(codec_opts, ost->enc->id, oc, st);
3500     }
3501
3502     avcodec_get_context_defaults3(st->codec, ost->enc);
3503     st->codec->codec_type = type; // XXX hack, avcodec_get_context_defaults2() sets type to unknown for stream copy
3504
3505     MATCH_PER_STREAM_OPT(presets, str, preset, oc, st);
3506     if (preset && (!(ret = get_preset_file_2(preset, ost->enc->name, &s)))) {
3507         do  {
3508             buf = get_line(s);
3509             if (!buf[0] || buf[0] == '#') {
3510                 av_free(buf);
3511                 continue;
3512             }
3513             if (!(arg = strchr(buf, '='))) {
3514                 av_log(NULL, AV_LOG_FATAL, "Invalid line found in the preset file.\n");
3515                 exit_program(1);
3516             }
3517             *arg++ = 0;
3518             av_dict_set(&ost->opts, buf, arg, AV_DICT_DONT_OVERWRITE);
3519             av_free(buf);
3520         } while (!s->eof_reached);
3521         avio_close(s);
3522     }
3523     if (ret) {
3524         av_log(NULL, AV_LOG_FATAL,
3525                "Preset %s specified for stream %d:%d, but could not be opened.\n",
3526                preset, ost->file_index, ost->index);
3527         exit_program(1);
3528     }
3529
3530     ost->max_frames = INT64_MAX;
3531     MATCH_PER_STREAM_OPT(max_frames, i64, ost->max_frames, oc, st);
3532
3533     MATCH_PER_STREAM_OPT(bitstream_filters, str, bsf, oc, st);
3534     while (bsf) {
3535         if (next = strchr(bsf, ','))
3536             *next++ = 0;
3537         if (!(bsfc = av_bitstream_filter_init(bsf))) {
3538             av_log(NULL, AV_LOG_FATAL, "Unknown bitstream filter %s\n", bsf);
3539             exit_program(1);
3540         }
3541         if (bsfc_prev)
3542             bsfc_prev->next = bsfc;
3543         else
3544             ost->bitstream_filters = bsfc;
3545
3546         bsfc_prev = bsfc;
3547         bsf       = next;
3548     }
3549
3550     MATCH_PER_STREAM_OPT(codec_tags, str, codec_tag, oc, st);
3551     if (codec_tag) {
3552         uint32_t tag = strtol(codec_tag, &next, 0);
3553         if (*next)
3554             tag = AV_RL32(codec_tag);
3555         st->codec->codec_tag = tag;
3556     }
3557
3558     MATCH_PER_STREAM_OPT(qscale, dbl, qscale, oc, st);
3559     if (qscale >= 0 || same_quant) {
3560         st->codec->flags |= CODEC_FLAG_QSCALE;
3561         st->codec->global_quality = FF_QP2LAMBDA * qscale;
3562     }
3563
3564     if (oc->oformat->flags & AVFMT_GLOBALHEADER)
3565         st->codec->flags |= CODEC_FLAG_GLOBAL_HEADER;
3566
3567     av_opt_get_int(sws_opts, "sws_flags", 0, &ost->sws_flags);
3568     return ost;
3569 }
3570
3571 static void parse_matrix_coeffs(uint16_t *dest, const char *str)
3572 {
3573     int i;
3574     const char *p = str;
3575     for (i = 0;; i++) {
3576         dest[i] = atoi(p);
3577         if (i == 63)
3578             break;
3579         p = strchr(p, ',');
3580         if (!p) {
3581             av_log(NULL, AV_LOG_FATAL, "Syntax error in matrix \"%s\" at coeff %d\n", str, i);
3582             exit_program(1);
3583         }
3584         p++;
3585     }
3586 }
3587
3588 static OutputStream *new_video_stream(OptionsContext *o, AVFormatContext *oc)
3589 {
3590     AVStream *st;
3591     OutputStream *ost;
3592     AVCodecContext *video_enc;
3593
3594     ost = new_output_stream(o, oc, AVMEDIA_TYPE_VIDEO);
3595     st  = ost->st;
3596     video_enc = st->codec;
3597
3598     if (!ost->stream_copy) {
3599         const char *p = NULL;
3600         char *forced_key_frames = NULL, *frame_rate = NULL, *frame_size = NULL;
3601         char *frame_aspect_ratio = NULL, *frame_pix_fmt = NULL;
3602         char *intra_matrix = NULL, *inter_matrix = NULL, *filters = NULL;
3603         int i;
3604
3605         MATCH_PER_STREAM_OPT(frame_rates, str, frame_rate, oc, st);
3606         if (frame_rate && av_parse_video_rate(&ost->frame_rate, frame_rate) < 0) {
3607             av_log(NULL, AV_LOG_FATAL, "Invalid framerate value: %s\n", frame_rate);
3608             exit_program(1);
3609         }
3610
3611         MATCH_PER_STREAM_OPT(frame_sizes, str, frame_size, oc, st);
3612         if (frame_size && av_parse_video_size(&video_enc->width, &video_enc->height, frame_size) < 0) {
3613             av_log(NULL, AV_LOG_FATAL, "Invalid frame size: %s.\n", frame_size);
3614             exit_program(1);
3615         }
3616
3617         MATCH_PER_STREAM_OPT(frame_aspect_ratios, str, frame_aspect_ratio, oc, st);
3618         if (frame_aspect_ratio)
3619             ost->frame_aspect_ratio = parse_frame_aspect_ratio(frame_aspect_ratio);
3620
3621         MATCH_PER_STREAM_OPT(frame_pix_fmts, str, frame_pix_fmt, oc, st);
3622         if (frame_pix_fmt && (video_enc->pix_fmt = av_get_pix_fmt(frame_pix_fmt)) == PIX_FMT_NONE) {
3623             av_log(NULL, AV_LOG_FATAL, "Unknown pixel format requested: %s.\n", frame_pix_fmt);
3624             exit_program(1);
3625         }
3626         st->sample_aspect_ratio = video_enc->sample_aspect_ratio;
3627
3628         MATCH_PER_STREAM_OPT(intra_matrices, str, intra_matrix, oc, st);
3629         if (intra_matrix) {
3630             if (!(video_enc->intra_matrix = av_mallocz(sizeof(*video_enc->intra_matrix) * 64))) {
3631                 av_log(NULL, AV_LOG_FATAL, "Could not allocate memory for intra matrix.\n");
3632                 exit_program(1);
3633             }
3634             parse_matrix_coeffs(video_enc->intra_matrix, intra_matrix);
3635         }
3636         MATCH_PER_STREAM_OPT(inter_matrices, str, inter_matrix, oc, st);
3637         if (inter_matrix) {
3638             if (!(video_enc->inter_matrix = av_mallocz(sizeof(*video_enc->inter_matrix) * 64))) {
3639                 av_log(NULL, AV_LOG_FATAL, "Could not allocate memory for inter matrix.\n");
3640                 exit_program(1);
3641             }
3642             parse_matrix_coeffs(video_enc->inter_matrix, inter_matrix);
3643         }
3644
3645         MATCH_PER_STREAM_OPT(rc_overrides, str, p, oc, st);
3646         for (i = 0; p; i++) {
3647             int start, end, q;
3648             int e = sscanf(p, "%d,%d,%d", &start, &end, &q);
3649             if (e != 3) {
3650                 av_log(NULL, AV_LOG_FATAL, "error parsing rc_override\n");
3651                 exit_program(1);
3652             }
3653             video_enc->rc_override =
3654                 av_realloc(video_enc->rc_override,
3655                            sizeof(RcOverride) * (i + 1));
3656             video_enc->rc_override[i].start_frame = start;
3657             video_enc->rc_override[i].end_frame   = end;
3658             if (q > 0) {
3659                 video_enc->rc_override[i].qscale         = q;
3660                 video_enc->rc_override[i].quality_factor = 1.0;
3661             }
3662             else {
3663                 video_enc->rc_override[i].qscale         = 0;
3664                 video_enc->rc_override[i].quality_factor = -q/100.0;
3665             }
3666             p = strchr(p, '/');
3667             if (p) p++;
3668         }
3669         video_enc->rc_override_count = i;
3670         if (!video_enc->rc_initial_buffer_occupancy)
3671             video_enc->rc_initial_buffer_occupancy = video_enc->rc_buffer_size * 3 / 4;
3672         video_enc->intra_dc_precision = intra_dc_precision - 8;
3673
3674         /* two pass mode */
3675         if (do_pass) {
3676             if (do_pass == 1) {
3677                 video_enc->flags |= CODEC_FLAG_PASS1;
3678             } else {
3679                 video_enc->flags |= CODEC_FLAG_PASS2;
3680             }
3681         }
3682
3683         MATCH_PER_STREAM_OPT(forced_key_frames, str, forced_key_frames, oc, st);
3684         if (forced_key_frames)
3685             parse_forced_key_frames(forced_key_frames, ost, video_enc);
3686
3687         MATCH_PER_STREAM_OPT(force_fps, i, ost->force_fps, oc, st);
3688
3689         ost->top_field_first = -1;
3690         MATCH_PER_STREAM_OPT(top_field_first, i, ost->top_field_first, oc, st);
3691
3692 #if CONFIG_AVFILTER
3693         MATCH_PER_STREAM_OPT(filters, str, filters, oc, st);
3694         if (filters)
3695             ost->avfilter = av_strdup(filters);
3696 #endif
3697     } else {
3698         MATCH_PER_STREAM_OPT(copy_initial_nonkeyframes, i, ost->copy_initial_nonkeyframes, oc ,st);
3699     }
3700
3701     return ost;
3702 }
3703
3704 static OutputStream *new_audio_stream(OptionsContext *o, AVFormatContext *oc)
3705 {
3706     AVStream *st;
3707     OutputStream *ost;
3708     AVCodecContext *audio_enc;
3709
3710     ost = new_output_stream(o, oc, AVMEDIA_TYPE_AUDIO);
3711     st  = ost->st;
3712
3713     audio_enc = st->codec;
3714     audio_enc->codec_type = AVMEDIA_TYPE_AUDIO;
3715
3716     if (!ost->stream_copy) {
3717         char *sample_fmt = NULL;
3718
3719         MATCH_PER_STREAM_OPT(audio_channels, i, audio_enc->channels, oc, st);
3720
3721         MATCH_PER_STREAM_OPT(sample_fmts, str, sample_fmt, oc, st);
3722         if (sample_fmt &&
3723             (audio_enc->sample_fmt = av_get_sample_fmt(sample_fmt)) == AV_SAMPLE_FMT_NONE) {
3724             av_log(NULL, AV_LOG_FATAL, "Invalid sample format '%s'\n", sample_fmt);
3725             exit_program(1);
3726         }
3727
3728         MATCH_PER_STREAM_OPT(audio_sample_rate, i, audio_enc->sample_rate, oc, st);
3729     }
3730
3731     return ost;
3732 }
3733
3734 static OutputStream *new_data_stream(OptionsContext *o, AVFormatContext *oc)
3735 {
3736     OutputStream *ost;
3737
3738     ost = new_output_stream(o, oc, AVMEDIA_TYPE_DATA);
3739     if (!ost->stream_copy) {
3740         av_log(NULL, AV_LOG_FATAL, "Data stream encoding not supported yet (only streamcopy)\n");
3741         exit_program(1);
3742     }
3743
3744     return ost;
3745 }
3746
3747 static OutputStream *new_attachment_stream(OptionsContext *o, AVFormatContext *oc)
3748 {
3749     OutputStream *ost = new_output_stream(o, oc, AVMEDIA_TYPE_ATTACHMENT);
3750     ost->stream_copy = 1;
3751     return ost;
3752 }
3753
3754 static OutputStream *new_subtitle_stream(OptionsContext *o, AVFormatContext *oc)
3755 {
3756     AVStream *st;
3757     OutputStream *ost;
3758     AVCodecContext *subtitle_enc;
3759
3760     ost = new_output_stream(o, oc, AVMEDIA_TYPE_SUBTITLE);
3761     st  = ost->st;
3762     subtitle_enc = st->codec;
3763
3764     subtitle_enc->codec_type = AVMEDIA_TYPE_SUBTITLE;
3765
3766     return ost;
3767 }
3768
3769 /* arg format is "output-stream-index:streamid-value". */
3770 static int opt_streamid(OptionsContext *o, const char *opt, const char *arg)
3771 {
3772     int idx;
3773     char *p;
3774     char idx_str[16];
3775
3776     av_strlcpy(idx_str, arg, sizeof(idx_str));
3777     p = strchr(idx_str, ':');
3778     if (!p) {
3779         av_log(NULL, AV_LOG_FATAL,
3780                "Invalid value '%s' for option '%s', required syntax is 'index:value'\n",
3781                arg, opt);
3782         exit_program(1);
3783     }
3784     *p++ = '\0';
3785     idx = parse_number_or_die(opt, idx_str, OPT_INT, 0, INT_MAX);
3786     o->streamid_map = grow_array(o->streamid_map, sizeof(*o->streamid_map), &o->nb_streamid_map, idx+1);
3787     o->streamid_map[idx] = parse_number_or_die(opt, p, OPT_INT, 0, INT_MAX);
3788     return 0;
3789 }
3790
3791 static int copy_chapters(InputFile *ifile, OutputFile *ofile, int copy_metadata)
3792 {
3793     AVFormatContext *is = ifile->ctx;
3794     AVFormatContext *os = ofile->ctx;
3795     int i;
3796
3797     for (i = 0; i < is->nb_chapters; i++) {
3798         AVChapter *in_ch = is->chapters[i], *out_ch;
3799         int64_t ts_off   = av_rescale_q(ofile->start_time - ifile->ts_offset,
3800                                        AV_TIME_BASE_Q, in_ch->time_base);
3801         int64_t rt       = (ofile->recording_time == INT64_MAX) ? INT64_MAX :
3802                            av_rescale_q(ofile->recording_time, AV_TIME_BASE_Q, in_ch->time_base);
3803
3804
3805         if (in_ch->end < ts_off)
3806             continue;
3807         if (rt != INT64_MAX && in_ch->start > rt + ts_off)
3808             break;
3809
3810         out_ch = av_mallocz(sizeof(AVChapter));
3811         if (!out_ch)
3812             return AVERROR(ENOMEM);
3813
3814         out_ch->id        = in_ch->id;
3815         out_ch->time_base = in_ch->time_base;
3816         out_ch->start     = FFMAX(0,  in_ch->start - ts_off);
3817         out_ch->end       = FFMIN(rt, in_ch->end   - ts_off);
3818
3819         if (copy_metadata)
3820             av_dict_copy(&out_ch->metadata, in_ch->metadata, 0);
3821
3822         os->nb_chapters++;
3823         os->chapters = av_realloc(os->chapters, sizeof(AVChapter) * os->nb_chapters);
3824         if (!os->chapters)
3825             return AVERROR(ENOMEM);
3826         os->chapters[os->nb_chapters - 1] = out_ch;
3827     }
3828     return 0;
3829 }
3830
3831 static void opt_output_file(void *optctx, const char *filename)
3832 {
3833     OptionsContext *o = optctx;
3834     AVFormatContext *oc;
3835     int i, err;
3836     AVOutputFormat *file_oformat;
3837     OutputStream *ost;
3838     InputStream  *ist;
3839
3840     if (!strcmp(filename, "-"))
3841         filename = "pipe:";
3842
3843     oc = avformat_alloc_context();
3844     if (!oc) {
3845         print_error(filename, AVERROR(ENOMEM));
3846         exit_program(1);
3847     }
3848
3849     if (o->format) {
3850         file_oformat = av_guess_format(o->format, NULL, NULL);
3851         if (!file_oformat) {
3852             av_log(NULL, AV_LOG_FATAL, "Requested output format '%s' is not a suitable output format\n", o->format);
3853             exit_program(1);
3854         }
3855     } else {
3856         file_oformat = av_guess_format(NULL, filename, NULL);
3857         if (!file_oformat) {
3858             av_log(NULL, AV_LOG_FATAL, "Unable to find a suitable output format for '%s'\n",
3859                    filename);
3860             exit_program(1);
3861         }
3862     }
3863
3864     oc->oformat = file_oformat;
3865     oc->interrupt_callback = int_cb;
3866     av_strlcpy(oc->filename, filename, sizeof(oc->filename));
3867
3868     if (!o->nb_stream_maps) {
3869         /* pick the "best" stream of each type */
3870 #define NEW_STREAM(type, index)\
3871         if (index >= 0) {\
3872             ost = new_ ## type ## _stream(o, oc);\
3873             ost->source_index = index;\
3874             ost->sync_ist     = &input_streams[index];\
3875             input_streams[index].discard = 0;\
3876         }
3877
3878         /* video: highest resolution */
3879         if (!o->video_disable && oc->oformat->video_codec != CODEC_ID_NONE) {
3880             int area = 0, idx = -1;
3881             for (i = 0; i < nb_input_streams; i++) {
3882                 ist = &input_streams[i];
3883                 if (ist->st->codec->codec_type == AVMEDIA_TYPE_VIDEO &&
3884                     ist->st->codec->width * ist->st->codec->height > area) {
3885                     area = ist->st->codec->width * ist->st->codec->height;
3886                     idx = i;
3887                 }
3888             }
3889             NEW_STREAM(video, idx);
3890         }
3891
3892         /* audio: most channels */
3893         if (!o->audio_disable && oc->oformat->audio_codec != CODEC_ID_NONE) {
3894             int channels = 0, idx = -1;
3895             for (i = 0; i < nb_input_streams; i++) {
3896                 ist = &input_streams[i];
3897                 if (ist->st->codec->codec_type == AVMEDIA_TYPE_AUDIO &&
3898                     ist->st->codec->channels > channels) {
3899                     channels = ist->st->codec->channels;
3900                     idx = i;
3901                 }
3902             }
3903             NEW_STREAM(audio, idx);
3904         }
3905
3906         /* subtitles: pick first */
3907         if (!o->subtitle_disable && oc->oformat->subtitle_codec != CODEC_ID_NONE) {
3908             for (i = 0; i < nb_input_streams; i++)
3909                 if (input_streams[i].st->codec->codec_type == AVMEDIA_TYPE_SUBTITLE) {
3910                     NEW_STREAM(subtitle, i);
3911                     break;
3912                 }
3913         }
3914         /* do something with data? */
3915     } else {
3916         for (i = 0; i < o->nb_stream_maps; i++) {
3917             StreamMap *map = &o->stream_maps[i];
3918
3919             if (map->disabled)
3920                 continue;
3921
3922             ist = &input_streams[input_files[map->file_index].ist_index + map->stream_index];
3923             switch (ist->st->codec->codec_type) {
3924             case AVMEDIA_TYPE_VIDEO:    ost = new_video_stream(o, oc);    break;
3925             case AVMEDIA_TYPE_AUDIO:    ost = new_audio_stream(o, oc);    break;
3926             case AVMEDIA_TYPE_SUBTITLE: ost = new_subtitle_stream(o, oc); break;
3927             case AVMEDIA_TYPE_DATA:     ost = new_data_stream(o, oc);     break;
3928             case AVMEDIA_TYPE_ATTACHMENT: ost = new_attachment_stream(o, oc); break;
3929             default:
3930                 av_log(NULL, AV_LOG_FATAL, "Cannot map stream #%d:%d - unsupported type.\n",
3931                        map->file_index, map->stream_index);
3932                 exit_program(1);
3933             }
3934
3935             ost->source_index = input_files[map->file_index].ist_index + map->stream_index;
3936             ost->sync_ist     = &input_streams[input_files[map->sync_file_index].ist_index +
3937                                            map->sync_stream_index];
3938             ist->discard = 0;
3939         }
3940     }
3941
3942     /* handle attached files */
3943     for (i = 0; i < o->nb_attachments; i++) {
3944         AVIOContext *pb;
3945         uint8_t *attachment;
3946         const char *p;
3947         int64_t len;
3948
3949         if ((err = avio_open2(&pb, o->attachments[i], AVIO_FLAG_READ, &int_cb, NULL)) < 0) {
3950             av_log(NULL, AV_LOG_FATAL, "Could not open attachment file %s.\n",
3951                    o->attachments[i]);
3952             exit_program(1);
3953         }
3954         if ((len = avio_size(pb)) <= 0) {
3955             av_log(NULL, AV_LOG_FATAL, "Could not get size of the attachment %s.\n",
3956                    o->attachments[i]);
3957             exit_program(1);
3958         }
3959         if (!(attachment = av_malloc(len))) {
3960             av_log(NULL, AV_LOG_FATAL, "Attachment %s too large to fit into memory.\n",
3961                    o->attachments[i]);
3962             exit_program(1);
3963         }
3964         avio_read(pb, attachment, len);
3965
3966         ost = new_attachment_stream(o, oc);
3967         ost->stream_copy               = 0;
3968         ost->source_index              = -1;
3969         ost->attachment_filename       = o->attachments[i];
3970         ost->st->codec->extradata      = attachment;
3971         ost->st->codec->extradata_size = len;
3972
3973         p = strrchr(o->attachments[i], '/');
3974         av_dict_set(&ost->st->metadata, "filename", (p && *p) ? p + 1 : o->attachments[i], AV_DICT_DONT_OVERWRITE);
3975         avio_close(pb);
3976     }
3977
3978     output_files = grow_array(output_files, sizeof(*output_files), &nb_output_files, nb_output_files + 1);
3979     output_files[nb_output_files - 1].ctx       = oc;
3980     output_files[nb_output_files - 1].ost_index = nb_output_streams - oc->nb_streams;
3981     output_files[nb_output_files - 1].recording_time = o->recording_time;
3982     if (o->recording_time != INT64_MAX)
3983         oc->duration = o->recording_time;
3984     output_files[nb_output_files - 1].start_time     = o->start_time;
3985     output_files[nb_output_files - 1].limit_filesize = o->limit_filesize;
3986     av_dict_copy(&output_files[nb_output_files - 1].opts, format_opts, 0);
3987
3988     /* check filename in case of an image number is expected */
3989     if (oc->oformat->flags & AVFMT_NEEDNUMBER) {
3990         if (!av_filename_number_test(oc->filename)) {
3991             print_error(oc->filename, AVERROR(EINVAL));
3992             exit_program(1);
3993         }
3994     }
3995
3996     if (!(oc->oformat->flags & AVFMT_NOFILE)) {
3997         /* test if it already exists to avoid losing precious files */
3998         assert_file_overwrite(filename);
3999
4000         /* open the file */
4001         if ((err = avio_open2(&oc->pb, filename, AVIO_FLAG_WRITE,
4002                               &oc->interrupt_callback,
4003                               &output_files[nb_output_files - 1].opts)) < 0) {
4004             print_error(filename, err);
4005             exit_program(1);
4006         }
4007     }
4008
4009     if (o->mux_preload) {
4010         uint8_t buf[64];
4011         snprintf(buf, sizeof(buf), "%d", (int)(o->mux_preload*AV_TIME_BASE));
4012         av_dict_set(&output_files[nb_output_files - 1].opts, "preload", buf, 0);
4013     }
4014     oc->max_delay = (int)(o->mux_max_delay * AV_TIME_BASE);
4015     oc->flags |= AVFMT_FLAG_NONBLOCK;
4016
4017     /* copy metadata */
4018     for (i = 0; i < o->nb_metadata_map; i++) {
4019         char *p;
4020         int in_file_index = strtol(o->metadata_map[i].u.str, &p, 0);
4021
4022         if (in_file_index < 0)
4023             continue;
4024         if (in_file_index >= nb_input_files) {
4025             av_log(NULL, AV_LOG_FATAL, "Invalid input file index %d while processing metadata maps\n", in_file_index);
4026             exit_program(1);
4027         }
4028         copy_metadata(o->metadata_map[i].specifier, *p ? p + 1 : p, oc, input_files[in_file_index].ctx, o);
4029     }
4030
4031     /* copy chapters */
4032     if (o->chapters_input_file >= nb_input_files) {
4033         if (o->chapters_input_file == INT_MAX) {
4034             /* copy chapters from the first input file that has them*/
4035             o->chapters_input_file = -1;
4036             for (i = 0; i < nb_input_files; i++)
4037                 if (input_files[i].ctx->nb_chapters) {
4038                     o->chapters_input_file = i;
4039                     break;
4040                 }
4041         } else {
4042             av_log(NULL, AV_LOG_FATAL, "Invalid input file index %d in chapter mapping.\n",
4043                    o->chapters_input_file);
4044             exit_program(1);
4045         }
4046     }
4047     if (o->chapters_input_file >= 0)
4048         copy_chapters(&input_files[o->chapters_input_file], &output_files[nb_output_files - 1],
4049                       !o->metadata_chapters_manual);
4050
4051     /* copy global metadata by default */
4052     if (!o->metadata_global_manual && nb_input_files)
4053         av_dict_copy(&oc->metadata, input_files[0].ctx->metadata,
4054                      AV_DICT_DONT_OVERWRITE);
4055     if (!o->metadata_streams_manual)
4056         for (i = output_files[nb_output_files - 1].ost_index; i < nb_output_streams; i++) {
4057             InputStream *ist;
4058             if (output_streams[i].source_index < 0)         /* this is true e.g. for attached files */
4059                 continue;
4060             ist = &input_streams[output_streams[i].source_index];
4061             av_dict_copy(&output_streams[i].st->metadata, ist->st->metadata, AV_DICT_DONT_OVERWRITE);
4062         }
4063
4064     /* process manually set metadata */
4065     for (i = 0; i < o->nb_metadata; i++) {
4066         AVDictionary **m;
4067         char type, *val;
4068         const char *stream_spec;
4069         int index = 0, j, ret;
4070
4071         val = strchr(o->metadata[i].u.str, '=');
4072         if (!val) {
4073             av_log(NULL, AV_LOG_FATAL, "No '=' character in metadata string %s.\n",
4074                    o->metadata[i].u.str);
4075             exit_program(1);
4076         }
4077         *val++ = 0;
4078
4079         parse_meta_type(o->metadata[i].specifier, &type, &index, &stream_spec);
4080         if (type == 's') {
4081             for (j = 0; j < oc->nb_streams; j++) {
4082                 if ((ret = check_stream_specifier(oc, oc->streams[j], stream_spec)) > 0) {
4083                     av_dict_set(&oc->streams[j]->metadata, o->metadata[i].u.str, *val ? val : NULL, 0);
4084                 } else if (ret < 0)
4085                     exit_program(1);
4086             }
4087             printf("ret %d, stream_spec %s\n", ret, stream_spec);
4088         }
4089         else {
4090             switch (type) {
4091             case 'g':
4092                 m = &oc->metadata;
4093                 break;
4094             case 'c':
4095                 if (index < 0 || index >= oc->nb_chapters) {
4096                     av_log(NULL, AV_LOG_FATAL, "Invalid chapter index %d in metadata specifier.\n", index);
4097                     exit_program(1);
4098                 }
4099                 m = &oc->chapters[index]->metadata;
4100                 break;
4101             default:
4102                 av_log(NULL, AV_LOG_FATAL, "Invalid metadata specifier %s.\n", o->metadata[i].specifier);
4103                 exit_program(1);
4104             }
4105             av_dict_set(m, o->metadata[i].u.str, *val ? val : NULL, 0);
4106         }
4107     }
4108
4109     reset_options(o);
4110 }
4111
4112 /* same option as mencoder */
4113 static int opt_pass(const char *opt, const char *arg)
4114 {
4115     do_pass = parse_number_or_die(opt, arg, OPT_INT, 1, 2);
4116     return 0;
4117 }
4118
4119 static int64_t getutime(void)
4120 {
4121 #if HAVE_GETRUSAGE
4122     struct rusage rusage;
4123
4124     getrusage(RUSAGE_SELF, &rusage);
4125     return (rusage.ru_utime.tv_sec * 1000000LL) + rusage.ru_utime.tv_usec;
4126 #elif HAVE_GETPROCESSTIMES
4127     HANDLE proc;
4128     FILETIME c, e, k, u;
4129     proc = GetCurrentProcess();
4130     GetProcessTimes(proc, &c, &e, &k, &u);
4131     return ((int64_t) u.dwHighDateTime << 32 | u.dwLowDateTime) / 10;
4132 #else
4133     return av_gettime();
4134 #endif
4135 }
4136
4137 static int64_t getmaxrss(void)
4138 {
4139 #if HAVE_GETRUSAGE && HAVE_STRUCT_RUSAGE_RU_MAXRSS
4140     struct rusage rusage;
4141     getrusage(RUSAGE_SELF, &rusage);
4142     return (int64_t)rusage.ru_maxrss * 1024;
4143 #elif HAVE_GETPROCESSMEMORYINFO
4144     HANDLE proc;
4145     PROCESS_MEMORY_COUNTERS memcounters;
4146     proc = GetCurrentProcess();
4147     memcounters.cb = sizeof(memcounters);
4148     GetProcessMemoryInfo(proc, &memcounters, sizeof(memcounters));
4149     return memcounters.PeakPagefileUsage;
4150 #else
4151     return 0;
4152 #endif
4153 }
4154
4155 static int opt_audio_qscale(OptionsContext *o, const char *opt, const char *arg)
4156 {
4157     return parse_option(o, "q:a", arg, options);
4158 }
4159
4160 static void show_usage(void)
4161 {
4162     printf("Hyper fast Audio and Video encoder\n");
4163     printf("usage: %s [options] [[infile options] -i infile]... {[outfile options] outfile}...\n", program_name);
4164     printf("\n");
4165 }
4166
4167 static void show_help(void)
4168 {
4169     int flags = AV_OPT_FLAG_DECODING_PARAM | AV_OPT_FLAG_ENCODING_PARAM;
4170     av_log_set_callback(log_callback_help);
4171     show_usage();
4172     show_help_options(options, "Main options:\n",
4173                       OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_SUBTITLE | OPT_GRAB, 0);
4174     show_help_options(options, "\nAdvanced options:\n",
4175                       OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_SUBTITLE | OPT_GRAB,
4176                       OPT_EXPERT);
4177     show_help_options(options, "\nVideo options:\n",
4178                       OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_GRAB,
4179                       OPT_VIDEO);
4180     show_help_options(options, "\nAdvanced Video options:\n",
4181                       OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_GRAB,
4182                       OPT_VIDEO | OPT_EXPERT);
4183     show_help_options(options, "\nAudio options:\n",
4184                       OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_GRAB,
4185                       OPT_AUDIO);
4186     show_help_options(options, "\nAdvanced Audio options:\n",
4187                       OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_GRAB,
4188                       OPT_AUDIO | OPT_EXPERT);
4189     show_help_options(options, "\nSubtitle options:\n",
4190                       OPT_SUBTITLE | OPT_GRAB,
4191                       OPT_SUBTITLE);
4192     show_help_options(options, "\nAudio/Video grab options:\n",
4193                       OPT_GRAB,
4194                       OPT_GRAB);
4195     printf("\n");
4196     show_help_children(avcodec_get_class(), flags);
4197     show_help_children(avformat_get_class(), flags);
4198     show_help_children(sws_get_class(), flags);
4199 }
4200
4201 static int opt_target(OptionsContext *o, const char *opt, const char *arg)
4202 {
4203     enum { PAL, NTSC, FILM, UNKNOWN } norm = UNKNOWN;
4204     static const char *const frame_rates[] = { "25", "30000/1001", "24000/1001" };
4205
4206     if (!strncmp(arg, "pal-", 4)) {
4207         norm = PAL;
4208         arg += 4;
4209     } else if (!strncmp(arg, "ntsc-", 5)) {
4210         norm = NTSC;
4211         arg += 5;
4212     } else if (!strncmp(arg, "film-", 5)) {
4213         norm = FILM;
4214         arg += 5;
4215     } else {
4216         /* Try to determine PAL/NTSC by peeking in the input files */
4217         if (nb_input_files) {
4218             int i, j, fr;
4219             for (j = 0; j < nb_input_files; j++) {
4220                 for (i = 0; i < input_files[j].nb_streams; i++) {
4221                     AVCodecContext *c = input_files[j].ctx->streams[i]->codec;
4222                     if (c->codec_type != AVMEDIA_TYPE_VIDEO)
4223                         continue;
4224                     fr = c->time_base.den * 1000 / c->time_base.num;
4225                     if (fr == 25000) {
4226                         norm = PAL;
4227                         break;
4228                     } else if ((fr == 29970) || (fr == 23976)) {
4229                         norm = NTSC;
4230                         break;
4231                     }
4232                 }
4233                 if (norm != UNKNOWN)
4234                     break;
4235             }
4236         }
4237         if (norm != UNKNOWN)
4238             av_log(NULL, AV_LOG_INFO, "Assuming %s for target.\n", norm == PAL ? "PAL" : "NTSC");
4239     }
4240
4241     if (norm == UNKNOWN) {
4242         av_log(NULL, AV_LOG_FATAL, "Could not determine norm (PAL/NTSC/NTSC-Film) for target.\n");
4243         av_log(NULL, AV_LOG_FATAL, "Please prefix target with \"pal-\", \"ntsc-\" or \"film-\",\n");
4244         av_log(NULL, AV_LOG_FATAL, "or set a framerate with \"-r xxx\".\n");
4245         exit_program(1);
4246     }
4247
4248     if (!strcmp(arg, "vcd")) {
4249         opt_video_codec(o, "c:v", "mpeg1video");
4250         opt_audio_codec(o, "c:a", "mp2");
4251         parse_option(o, "f", "vcd", options);
4252
4253         parse_option(o, "s", norm == PAL ? "352x288" : "352x240", options);
4254         parse_option(o, "r", frame_rates[norm], options);
4255         opt_default("g", norm == PAL ? "15" : "18");
4256
4257         opt_default("b", "1150000");
4258         opt_default("maxrate", "1150000");
4259         opt_default("minrate", "1150000");
4260         opt_default("bufsize", "327680"); // 40*1024*8;
4261
4262         opt_default("b:a", "224000");
4263         parse_option(o, "ar", "44100", options);
4264         parse_option(o, "ac", "2", options);
4265
4266         opt_default("packetsize", "2324");
4267         opt_default("muxrate", "1411200"); // 2352 * 75 * 8;
4268
4269         /* We have to offset the PTS, so that it is consistent with the SCR.
4270            SCR starts at 36000, but the first two packs contain only padding
4271            and the first pack from the other stream, respectively, may also have
4272            been written before.
4273            So the real data starts at SCR 36000+3*1200. */
4274         o->mux_preload = (36000 + 3 * 1200) / 90000.0; // 0.44
4275     } else if (!strcmp(arg, "svcd")) {
4276
4277         opt_video_codec(o, "c:v", "mpeg2video");
4278         opt_audio_codec(o, "c:a", "mp2");
4279         parse_option(o, "f", "svcd", options);
4280
4281         parse_option(o, "s", norm == PAL ? "480x576" : "480x480", options);
4282         parse_option(o, "r", frame_rates[norm], options);
4283         opt_default("g", norm == PAL ? "15" : "18");
4284
4285         opt_default("b", "2040000");
4286         opt_default("maxrate", "2516000");
4287         opt_default("minrate", "0"); // 1145000;
4288         opt_default("bufsize", "1835008"); // 224*1024*8;
4289         opt_default("flags", "+scan_offset");
4290
4291
4292         opt_default("b:a", "224000");
4293         parse_option(o, "ar", "44100", options);
4294
4295         opt_default("packetsize", "2324");
4296
4297     } else if (!strcmp(arg, "dvd")) {
4298
4299         opt_video_codec(o, "c:v", "mpeg2video");
4300         opt_audio_codec(o, "c:a", "ac3");
4301         parse_option(o, "f", "dvd", options);
4302
4303         parse_option(o, "s", norm == PAL ? "720x576" : "720x480", options);
4304         parse_option(o, "r", frame_rates[norm], options);
4305         opt_default("g", norm == PAL ? "15" : "18");
4306
4307         opt_default("b", "6000000");
4308         opt_default("maxrate", "9000000");
4309         opt_default("minrate", "0"); // 1500000;
4310         opt_default("bufsize", "1835008"); // 224*1024*8;
4311
4312         opt_default("packetsize", "2048");  // from www.mpucoder.com: DVD sectors contain 2048 bytes of data, this is also the size of one pack.
4313         opt_default("muxrate", "10080000"); // from mplex project: data_rate = 1260000. mux_rate = data_rate * 8
4314
4315         opt_default("b:a", "448000");
4316         parse_option(o, "ar", "48000", options);
4317
4318     } else if (!strncmp(arg, "dv", 2)) {
4319
4320         parse_option(o, "f", "dv", options);
4321
4322         parse_option(o, "s", norm == PAL ? "720x576" : "720x480", options);
4323         parse_option(o, "pix_fmt", !strncmp(arg, "dv50", 4) ? "yuv422p" :
4324                           norm == PAL ? "yuv420p" : "yuv411p", options);
4325         parse_option(o, "r", frame_rates[norm], options);
4326
4327         parse_option(o, "ar", "48000", options);
4328         parse_option(o, "ac", "2", options);
4329
4330     } else {
4331         av_log(NULL, AV_LOG_ERROR, "Unknown target: %s\n", arg);
4332         return AVERROR(EINVAL);
4333     }
4334     return 0;
4335 }
4336
4337 static int opt_vstats_file(const char *opt, const char *arg)
4338 {
4339     av_free (vstats_filename);
4340     vstats_filename = av_strdup (arg);
4341     return 0;
4342 }
4343
4344 static int opt_vstats(const char *opt, const char *arg)
4345 {
4346     char filename[40];
4347     time_t today2 = time(NULL);
4348     struct tm *today = localtime(&today2);
4349
4350     snprintf(filename, sizeof(filename), "vstats_%02d%02d%02d.log", today->tm_hour, today->tm_min,
4351              today->tm_sec);
4352     return opt_vstats_file(opt, filename);
4353 }
4354
4355 static int opt_video_frames(OptionsContext *o, const char *opt, const char *arg)