a005d7fc667b6bab74febed532b1e542ad57a444
[ffmpeg.git] / ffmpeg.c
1 /*
2  * FFmpeg main
3  * Copyright (c) 2000-2003 Fabrice Bellard
4  *
5  * This library is free software; you can redistribute it and/or
6  * modify it under the terms of the GNU Lesser General Public
7  * License as published by the Free Software Foundation; either
8  * version 2 of the License, or (at your option) any later version.
9  *
10  * This library is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  * Lesser General Public License for more details.
14  *
15  * You should have received a copy of the GNU Lesser General Public
16  * License along with this library; if not, write to the Free Software
17  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
18  */
19 #define HAVE_AV_CONFIG_H
20 #include <limits.h>
21 #include "avformat.h"
22 #include "framehook.h"
23 #include "dsputil.h"
24 #include "opt.h"
25
26 #ifndef CONFIG_WIN32
27 #include <unistd.h>
28 #include <fcntl.h>
29 #include <sys/ioctl.h>
30 #include <sys/time.h>
31 #include <termios.h>
32 #include <sys/resource.h>
33 #include <signal.h>
34 #endif
35 #ifdef CONFIG_OS2
36 #include <sys/types.h>
37 #include <sys/select.h>
38 #include <stdlib.h>
39 #endif
40 #undef time //needed because HAVE_AV_CONFIG_H is defined on top
41 #include <time.h>
42
43 #include "version.h"
44 #include "cmdutils.h"
45
46 #undef NDEBUG
47 #include <assert.h>
48
49 #if !defined(INFINITY) && defined(HUGE_VAL)
50 #define INFINITY HUGE_VAL
51 #endif
52
53 /* select an input stream for an output stream */
54 typedef struct AVStreamMap {
55     int file_index;
56     int stream_index;
57     int sync_file_index;
58     int sync_stream_index;
59 } AVStreamMap;
60
61 /** select an input file for an output file */
62 typedef struct AVMetaDataMap {
63     int out_file;
64     int in_file;
65 } AVMetaDataMap;
66
67 extern const OptionDef options[];
68
69 static void show_help(void);
70 static void show_license(void);
71 static int opt_default(const char *opt, const char *arg);
72
73 #define MAX_FILES 20
74
75 static AVFormatContext *input_files[MAX_FILES];
76 static int64_t input_files_ts_offset[MAX_FILES];
77 static int nb_input_files = 0;
78
79 static AVFormatContext *output_files[MAX_FILES];
80 static int nb_output_files = 0;
81
82 static AVStreamMap stream_maps[MAX_FILES];
83 static int nb_stream_maps;
84
85 static AVMetaDataMap meta_data_maps[MAX_FILES];
86 static int nb_meta_data_maps;
87
88 static AVInputFormat *file_iformat;
89 static AVOutputFormat *file_oformat;
90 static AVImageFormat *image_format;
91 static int frame_width  = 0;
92 static int frame_height = 0;
93 static float frame_aspect_ratio = 0;
94 static enum PixelFormat frame_pix_fmt = PIX_FMT_NONE;
95 static int frame_padtop  = 0;
96 static int frame_padbottom = 0;
97 static int frame_padleft  = 0;
98 static int frame_padright = 0;
99 static int padcolor[3] = {16,128,128}; /* default to black */
100 static int frame_topBand  = 0;
101 static int frame_bottomBand = 0;
102 static int frame_leftBand  = 0;
103 static int frame_rightBand = 0;
104 static int max_frames[4] = {INT_MAX, INT_MAX, INT_MAX, INT_MAX};
105 static int frame_rate = 25;
106 static int frame_rate_base = 1;
107 static int video_bit_rate = 200*1000;
108 static int video_bit_rate_tolerance = 4000*1000;
109 static float video_qscale = 0;
110 static int video_qmin = 2;
111 static int video_qmax = 31;
112 static int video_lmin = 2*FF_QP2LAMBDA;
113 static int video_lmax = 31*FF_QP2LAMBDA;
114 static int video_mb_lmin = 2*FF_QP2LAMBDA;
115 static int video_mb_lmax = 31*FF_QP2LAMBDA;
116 static int video_qdiff = 3;
117 static float video_qblur = 0.5;
118 static float video_qsquish = 0.0;
119 static float video_qcomp = 0.5;
120 static uint16_t *intra_matrix = NULL;
121 static uint16_t *inter_matrix = NULL;
122 #if 0 //experimental, (can be removed)
123 static float video_rc_qsquish=1.0;
124 static float video_rc_qmod_amp=0;
125 static int video_rc_qmod_freq=0;
126 #endif
127 static char *video_rc_override_string=NULL;
128 static char *video_rc_eq="tex^qComp";
129 static int video_rc_buffer_size=0;
130 static float video_rc_buffer_aggressivity=1.0;
131 static int video_rc_max_rate=0;
132 static int video_rc_min_rate=0;
133 static float video_rc_initial_cplx=0;
134 static float video_b_qfactor = 1.25;
135 static float video_b_qoffset = 1.25;
136 static float video_i_qfactor = -0.8;
137 static float video_i_qoffset = 0.0;
138 static int video_intra_quant_bias= FF_DEFAULT_QUANT_BIAS;
139 static int video_inter_quant_bias= FF_DEFAULT_QUANT_BIAS;
140 static int me_method = ME_EPZS;
141 static int video_disable = 0;
142 static int video_discard = 0;
143 static int video_codec_id = CODEC_ID_NONE;
144 static int video_codec_tag = 0;
145 static int same_quality = 0;
146 static int b_frames = 0;
147 static int pre_me = 0;
148 static int do_deinterlace = 0;
149 static int workaround_bugs = FF_BUG_AUTODETECT;
150 static int packet_size = 0;
151 static int error_rate = 0;
152 static int strict = 0;
153 static int top_field_first = -1;
154 static int sc_threshold = 0;
155 static int me_threshold = 0;
156 static int mb_threshold = 0;
157 static int intra_dc_precision = 8;
158 static int me_penalty_compensation= 256;
159 static int frame_skip_threshold= 0;
160 static int frame_skip_factor= 0;
161 static int frame_skip_exp= 0;
162 extern int loop_input; /* currently a hack */
163 static int loop_output = AVFMT_NOOUTPUTLOOP;
164 static int genpts = 0;
165 static int qp_hist = 0;
166
167 static int gop_size = 12;
168 static int intra_only = 0;
169 static int audio_sample_rate = 44100;
170 static int audio_bit_rate = 64000;
171 #define QSCALE_NONE -99999
172 static float audio_qscale = QSCALE_NONE;
173 static int audio_disable = 0;
174 static int audio_channels = 1;
175 static int audio_codec_id = CODEC_ID_NONE;
176 static int audio_codec_tag = 0;
177 static char *audio_language = NULL;
178
179 static int subtitle_codec_id = CODEC_ID_NONE;
180 static char *subtitle_language = NULL;
181
182 static int mux_rate= 0;
183 static int mux_packet_size= 0;
184 static float mux_preload= 0.5;
185 static float mux_max_delay= 0.7;
186
187 static int64_t recording_time = 0;
188 static int64_t start_time = 0;
189 static int64_t rec_timestamp = 0;
190 static int64_t input_ts_offset = 0;
191 static int file_overwrite = 0;
192 static char *str_title = NULL;
193 static char *str_author = NULL;
194 static char *str_copyright = NULL;
195 static char *str_comment = NULL;
196 static int do_benchmark = 0;
197 static int do_hex_dump = 0;
198 static int do_pkt_dump = 0;
199 static int do_psnr = 0;
200 static int do_vstats = 0;
201 static int do_pass = 0;
202 static char *pass_logfilename = NULL;
203 static int audio_stream_copy = 0;
204 static int video_stream_copy = 0;
205 static int subtitle_stream_copy = 0;
206 static int video_sync_method= 1;
207 static int audio_sync_method= 0;
208 static int copy_ts= 0;
209 static int opt_shortest = 0; //
210 static int video_global_header = 0;
211
212 static int rate_emu = 0;
213
214 #ifdef CONFIG_BKTR
215 static const char *video_grab_format = "bktr";
216 #else
217 #ifdef CONFIG_VIDEO4LINUX2
218 static const char *video_grab_format = "video4linux2";
219 #else
220 static const char *video_grab_format = "video4linux";
221 #endif
222 #endif
223 static char *video_device = NULL;
224 static char *grab_device = NULL;
225 static int  video_channel = 0;
226 static char *video_standard = "ntsc";
227
228 static const char *audio_grab_format = "audio_device";
229 static char *audio_device = NULL;
230 static int audio_volume = 256;
231
232 static int using_stdin = 0;
233 static int using_vhook = 0;
234 static int verbose = 1;
235 static int thread_count= 1;
236 static int q_pressed = 0;
237 static int me_range = 0;
238 static int64_t video_size = 0;
239 static int64_t audio_size = 0;
240 static int64_t extra_size = 0;
241 static int nb_frames_dup = 0;
242 static int nb_frames_drop = 0;
243 static int input_sync;
244 static int limit_filesize = 0; //
245
246 static int pgmyuv_compatibility_hack=0;
247 static int dts_delta_threshold = 10;
248
249 const char **opt_names=NULL;
250 int opt_name_count=0;
251 AVCodecContext *avctx_opts;
252
253 static AVBitStreamFilterContext *video_bitstream_filters=NULL;
254 static AVBitStreamFilterContext *audio_bitstream_filters=NULL;
255 static AVBitStreamFilterContext *bitstream_filters[MAX_FILES][MAX_STREAMS];
256
257 #define DEFAULT_PASS_LOGFILENAME "ffmpeg2pass"
258
259 struct AVInputStream;
260
261 typedef struct AVOutputStream {
262     int file_index;          /* file index */
263     int index;               /* stream index in the output file */
264     int source_index;        /* AVInputStream index */
265     AVStream *st;            /* stream in the output file */
266     int encoding_needed;     /* true if encoding needed for this stream */
267     int frame_number;
268     /* input pts and corresponding output pts
269        for A/V sync */
270     //double sync_ipts;        /* dts from the AVPacket of the demuxer in second units */
271     struct AVInputStream *sync_ist; /* input stream to sync against */
272     int64_t sync_opts;       /* output frame counter, could be changed to some true timestamp */ //FIXME look at frame_number
273     /* video only */
274     int video_resample;
275     AVFrame pict_tmp;      /* temporary image for resampling */
276     ImgReSampleContext *img_resample_ctx; /* for image resampling */
277
278     int video_crop;
279     int topBand;             /* cropping area sizes */
280     int leftBand;
281
282     int video_pad;
283     int padtop;              /* padding area sizes */
284     int padbottom;
285     int padleft;
286     int padright;
287
288     /* audio only */
289     int audio_resample;
290     ReSampleContext *resample; /* for audio resampling */
291     FifoBuffer fifo;     /* for compression: one audio fifo per codec */
292     FILE *logfile;
293 } AVOutputStream;
294
295 typedef struct AVInputStream {
296     int file_index;
297     int index;
298     AVStream *st;
299     int discard;             /* true if stream data should be discarded */
300     int decoding_needed;     /* true if the packets must be decoded in 'raw_fifo' */
301     int64_t sample_index;      /* current sample */
302
303     int64_t       start;     /* time when read started */
304     unsigned long frame;     /* current frame */
305     int64_t       next_pts;  /* synthetic pts for cases where pkt.pts
306                                 is not defined */
307     int64_t       pts;       /* current pts */
308     int is_start;            /* is 1 at the start and after a discontinuity */
309 } AVInputStream;
310
311 typedef struct AVInputFile {
312     int eof_reached;      /* true if eof reached */
313     int ist_index;        /* index of first stream in ist_table */
314     int buffer_size;      /* current total buffer size */
315     int buffer_size_max;  /* buffer size at which we consider we can stop
316                              buffering */
317     int nb_streams;       /* nb streams we are aware of */
318 } AVInputFile;
319
320 #ifndef CONFIG_WIN32
321
322 /* init terminal so that we can grab keys */
323 static struct termios oldtty;
324
325 static void term_exit(void)
326 {
327     tcsetattr (0, TCSANOW, &oldtty);
328 }
329
330 static volatile sig_atomic_t received_sigterm = 0;
331
332 static void
333 sigterm_handler(int sig)
334 {
335     received_sigterm = sig;
336     term_exit();
337 }
338
339 static void term_init(void)
340 {
341     struct termios tty;
342
343     tcgetattr (0, &tty);
344     oldtty = tty;
345
346     tty.c_iflag &= ~(IGNBRK|BRKINT|PARMRK|ISTRIP
347                           |INLCR|IGNCR|ICRNL|IXON);
348     tty.c_oflag |= OPOST;
349     tty.c_lflag &= ~(ECHO|ECHONL|ICANON|IEXTEN);
350     tty.c_cflag &= ~(CSIZE|PARENB);
351     tty.c_cflag |= CS8;
352     tty.c_cc[VMIN] = 1;
353     tty.c_cc[VTIME] = 0;
354
355     tcsetattr (0, TCSANOW, &tty);
356
357     signal(SIGINT , sigterm_handler); /* Interrupt (ANSI).  */
358     signal(SIGQUIT, sigterm_handler); /* Quit (POSIX).  */
359     signal(SIGTERM, sigterm_handler); /* Termination (ANSI).  */
360     /*
361     register a function to be called at normal program termination
362     */
363     atexit(term_exit);
364 #ifdef CONFIG_BEOS_NETSERVER
365     fcntl(0, F_SETFL, fcntl(0, F_GETFL) | O_NONBLOCK);
366 #endif
367 }
368
369 /* read a key without blocking */
370 static int read_key(void)
371 {
372     int n = 1;
373     unsigned char ch;
374 #ifndef CONFIG_BEOS_NETSERVER
375     struct timeval tv;
376     fd_set rfds;
377
378     FD_ZERO(&rfds);
379     FD_SET(0, &rfds);
380     tv.tv_sec = 0;
381     tv.tv_usec = 0;
382     n = select(1, &rfds, NULL, NULL, &tv);
383 #endif
384     if (n > 0) {
385         n = read(0, &ch, 1);
386         if (n == 1)
387             return ch;
388
389         return n;
390     }
391     return -1;
392 }
393
394 static int decode_interrupt_cb(void)
395 {
396     return q_pressed || (q_pressed = read_key() == 'q');
397 }
398
399 #else
400
401 static volatile int received_sigterm = 0;
402
403 /* no interactive support */
404 static void term_exit(void)
405 {
406 }
407
408 static void term_init(void)
409 {
410 }
411
412 static int read_key(void)
413 {
414     return 0;
415 }
416
417 #endif
418
419 static int read_ffserver_streams(AVFormatContext *s, const char *filename)
420 {
421     int i, err;
422     AVFormatContext *ic;
423
424     err = av_open_input_file(&ic, filename, NULL, FFM_PACKET_SIZE, NULL);
425     if (err < 0)
426         return err;
427     /* copy stream format */
428     s->nb_streams = ic->nb_streams;
429     for(i=0;i<ic->nb_streams;i++) {
430         AVStream *st;
431
432         // FIXME: a more elegant solution is needed
433         st = av_mallocz(sizeof(AVStream));
434         memcpy(st, ic->streams[i], sizeof(AVStream));
435         st->codec = avcodec_alloc_context();
436         memcpy(st->codec, ic->streams[i]->codec, sizeof(AVCodecContext));
437         s->streams[i] = st;
438     }
439
440     av_close_input_file(ic);
441     return 0;
442 }
443
444 static double
445 get_sync_ipts(const AVOutputStream *ost)
446 {
447     const AVInputStream *ist = ost->sync_ist;
448     return (double)(ist->pts + input_files_ts_offset[ist->file_index] - start_time)/AV_TIME_BASE;
449 }
450
451 static void write_frame(AVFormatContext *s, AVPacket *pkt, AVCodecContext *avctx, AVBitStreamFilterContext *bsfc){
452     while(bsfc){
453         AVPacket new_pkt= *pkt;
454         int a= av_bitstream_filter_filter(bsfc, avctx, NULL,
455                                           &new_pkt.data, &new_pkt.size,
456                                           pkt->data, pkt->size,
457                                           pkt->flags & PKT_FLAG_KEY);
458         if(a){
459             av_free_packet(pkt);
460             new_pkt.destruct= av_destruct_packet;
461         }
462         *pkt= new_pkt;
463
464         bsfc= bsfc->next;
465     }
466
467     av_interleaved_write_frame(s, pkt);
468 }
469
470 #define MAX_AUDIO_PACKET_SIZE (128 * 1024)
471
472 static void do_audio_out(AVFormatContext *s,
473                          AVOutputStream *ost,
474                          AVInputStream *ist,
475                          unsigned char *buf, int size)
476 {
477     uint8_t *buftmp;
478     static uint8_t *audio_buf = NULL;
479     static uint8_t *audio_out = NULL;
480     const int audio_out_size= 4*MAX_AUDIO_PACKET_SIZE;
481
482     int size_out, frame_bytes, ret;
483     AVCodecContext *enc= ost->st->codec;
484
485     /* SC: dynamic allocation of buffers */
486     if (!audio_buf)
487         audio_buf = av_malloc(2*MAX_AUDIO_PACKET_SIZE);
488     if (!audio_out)
489         audio_out = av_malloc(audio_out_size);
490     if (!audio_buf || !audio_out)
491         return;               /* Should signal an error ! */
492
493     if(audio_sync_method){
494         double delta = get_sync_ipts(ost) * enc->sample_rate - ost->sync_opts
495                 - fifo_size(&ost->fifo, ost->fifo.rptr)/(ost->st->codec->channels * 2);
496         double idelta= delta*ist->st->codec->sample_rate / enc->sample_rate;
497         int byte_delta= ((int)idelta)*2*ist->st->codec->channels;
498
499         //FIXME resample delay
500         if(fabs(delta) > 50){
501             if(ist->is_start){
502                 if(byte_delta < 0){
503                     byte_delta= FFMAX(byte_delta, -size);
504                     size += byte_delta;
505                     buf  -= byte_delta;
506                     if(verbose > 2)
507                         fprintf(stderr, "discarding %d audio samples\n", (int)-delta);
508                     if(!size)
509                         return;
510                     ist->is_start=0;
511                 }else{
512                     static uint8_t *input_tmp= NULL;
513                     input_tmp= av_realloc(input_tmp, byte_delta + size);
514
515                     if(byte_delta + size <= MAX_AUDIO_PACKET_SIZE)
516                         ist->is_start=0;
517                     else
518                         byte_delta= MAX_AUDIO_PACKET_SIZE - size;
519
520                     memset(input_tmp, 0, byte_delta);
521                     memcpy(input_tmp + byte_delta, buf, size);
522                     buf= input_tmp;
523                     size += byte_delta;
524                     if(verbose > 2)
525                         fprintf(stderr, "adding %d audio samples of silence\n", (int)delta);
526                 }
527             }else if(audio_sync_method>1){
528                 int comp= clip(delta, -audio_sync_method, audio_sync_method);
529                 assert(ost->audio_resample);
530                 if(verbose > 2)
531                     fprintf(stderr, "compensating audio timestamp drift:%f compensation:%d in:%d\n", delta, comp, enc->sample_rate);
532 //                fprintf(stderr, "drift:%f len:%d opts:%lld ipts:%lld fifo:%d\n", delta, -1, ost->sync_opts, (int64_t)(get_sync_ipts(ost) * enc->sample_rate), fifo_size(&ost->fifo, ost->fifo.rptr)/(ost->st->codec->channels * 2));
533                 av_resample_compensate(*(struct AVResampleContext**)ost->resample, comp, enc->sample_rate);
534             }
535         }
536     }else
537         ost->sync_opts= lrintf(get_sync_ipts(ost) * enc->sample_rate)
538                         - fifo_size(&ost->fifo, ost->fifo.rptr)/(ost->st->codec->channels * 2); //FIXME wrong
539
540     if (ost->audio_resample) {
541         buftmp = audio_buf;
542         size_out = audio_resample(ost->resample,
543                                   (short *)buftmp, (short *)buf,
544                                   size / (ist->st->codec->channels * 2));
545         size_out = size_out * enc->channels * 2;
546     } else {
547         buftmp = buf;
548         size_out = size;
549     }
550
551     /* now encode as many frames as possible */
552     if (enc->frame_size > 1) {
553         /* output resampled raw samples */
554         fifo_write(&ost->fifo, buftmp, size_out,
555                    &ost->fifo.wptr);
556
557         frame_bytes = enc->frame_size * 2 * enc->channels;
558
559         while (fifo_read(&ost->fifo, audio_buf, frame_bytes,
560                      &ost->fifo.rptr) == 0) {
561             AVPacket pkt;
562             av_init_packet(&pkt);
563
564             ret = avcodec_encode_audio(enc, audio_out, audio_out_size,
565                                        (short *)audio_buf);
566             audio_size += ret;
567             pkt.stream_index= ost->index;
568             pkt.data= audio_out;
569             pkt.size= ret;
570             if(enc->coded_frame && enc->coded_frame->pts != AV_NOPTS_VALUE)
571                 pkt.pts= av_rescale_q(enc->coded_frame->pts, enc->time_base, ost->st->time_base);
572             pkt.flags |= PKT_FLAG_KEY;
573             write_frame(s, &pkt, ost->st->codec, bitstream_filters[ost->file_index][pkt.stream_index]);
574
575             ost->sync_opts += enc->frame_size;
576         }
577     } else {
578         AVPacket pkt;
579         av_init_packet(&pkt);
580
581         ost->sync_opts += size_out / (2 * enc->channels);
582
583         /* output a pcm frame */
584         /* XXX: change encoding codec API to avoid this ? */
585         switch(enc->codec->id) {
586         case CODEC_ID_PCM_S32LE:
587         case CODEC_ID_PCM_S32BE:
588         case CODEC_ID_PCM_U32LE:
589         case CODEC_ID_PCM_U32BE:
590             size_out = size_out << 1;
591             break;
592         case CODEC_ID_PCM_S24LE:
593         case CODEC_ID_PCM_S24BE:
594         case CODEC_ID_PCM_U24LE:
595         case CODEC_ID_PCM_U24BE:
596         case CODEC_ID_PCM_S24DAUD:
597             size_out = size_out / 2 * 3;
598             break;
599         case CODEC_ID_PCM_S16LE:
600         case CODEC_ID_PCM_S16BE:
601         case CODEC_ID_PCM_U16LE:
602         case CODEC_ID_PCM_U16BE:
603             break;
604         default:
605             size_out = size_out >> 1;
606             break;
607         }
608         ret = avcodec_encode_audio(enc, audio_out, size_out,
609                                    (short *)buftmp);
610         audio_size += ret;
611         pkt.stream_index= ost->index;
612         pkt.data= audio_out;
613         pkt.size= ret;
614         if(enc->coded_frame && enc->coded_frame->pts != AV_NOPTS_VALUE)
615             pkt.pts= av_rescale_q(enc->coded_frame->pts, enc->time_base, ost->st->time_base);
616         pkt.flags |= PKT_FLAG_KEY;
617         write_frame(s, &pkt, ost->st->codec, bitstream_filters[ost->file_index][pkt.stream_index]);
618     }
619 }
620
621 static void pre_process_video_frame(AVInputStream *ist, AVPicture *picture, void **bufp)
622 {
623     AVCodecContext *dec;
624     AVPicture *picture2;
625     AVPicture picture_tmp;
626     uint8_t *buf = 0;
627
628     dec = ist->st->codec;
629
630     /* deinterlace : must be done before any resize */
631     if (do_deinterlace || using_vhook) {
632         int size;
633
634         /* create temporary picture */
635         size = avpicture_get_size(dec->pix_fmt, dec->width, dec->height);
636         buf = av_malloc(size);
637         if (!buf)
638             return;
639
640         picture2 = &picture_tmp;
641         avpicture_fill(picture2, buf, dec->pix_fmt, dec->width, dec->height);
642
643         if (do_deinterlace){
644             if(avpicture_deinterlace(picture2, picture,
645                                      dec->pix_fmt, dec->width, dec->height) < 0) {
646                 /* if error, do not deinterlace */
647                 av_free(buf);
648                 buf = NULL;
649                 picture2 = picture;
650             }
651         } else {
652             img_copy(picture2, picture, dec->pix_fmt, dec->width, dec->height);
653         }
654     } else {
655         picture2 = picture;
656     }
657
658     frame_hook_process(picture2, dec->pix_fmt, dec->width, dec->height);
659
660     if (picture != picture2)
661         *picture = *picture2;
662     *bufp = buf;
663 }
664
665 /* we begin to correct av delay at this threshold */
666 #define AV_DELAY_MAX 0.100
667
668 static void do_subtitle_out(AVFormatContext *s,
669                             AVOutputStream *ost,
670                             AVInputStream *ist,
671                             AVSubtitle *sub,
672                             int64_t pts)
673 {
674     static uint8_t *subtitle_out = NULL;
675     int subtitle_out_max_size = 65536;
676     int subtitle_out_size, nb, i;
677     AVCodecContext *enc;
678     AVPacket pkt;
679
680     if (pts == AV_NOPTS_VALUE) {
681         fprintf(stderr, "Subtitle packets must have a pts\n");
682         return;
683     }
684
685     enc = ost->st->codec;
686
687     if (!subtitle_out) {
688         subtitle_out = av_malloc(subtitle_out_max_size);
689     }
690
691     /* Note: DVB subtitle need one packet to draw them and one other
692        packet to clear them */
693     /* XXX: signal it in the codec context ? */
694     if (enc->codec_id == CODEC_ID_DVB_SUBTITLE)
695         nb = 2;
696     else
697         nb = 1;
698
699     for(i = 0; i < nb; i++) {
700         subtitle_out_size = avcodec_encode_subtitle(enc, subtitle_out,
701                                                     subtitle_out_max_size, sub);
702
703         av_init_packet(&pkt);
704         pkt.stream_index = ost->index;
705         pkt.data = subtitle_out;
706         pkt.size = subtitle_out_size;
707         pkt.pts = av_rescale_q(av_rescale_q(pts, ist->st->time_base, AV_TIME_BASE_Q) + input_files_ts_offset[ist->file_index], AV_TIME_BASE_Q,  ost->st->time_base);
708         if (enc->codec_id == CODEC_ID_DVB_SUBTITLE) {
709             /* XXX: the pts correction is handled here. Maybe handling
710                it in the codec would be better */
711             if (i == 0)
712                 pkt.pts += 90 * sub->start_display_time;
713             else
714                 pkt.pts += 90 * sub->end_display_time;
715         }
716         write_frame(s, &pkt, ost->st->codec, bitstream_filters[ost->file_index][pkt.stream_index]);
717     }
718 }
719
720 static int bit_buffer_size= 1024*256;
721 static uint8_t *bit_buffer= NULL;
722
723 static void do_video_out(AVFormatContext *s,
724                          AVOutputStream *ost,
725                          AVInputStream *ist,
726                          AVFrame *in_picture,
727                          int *frame_size)
728 {
729     int nb_frames, i, ret;
730     AVFrame *final_picture, *formatted_picture, *resampling_dst, *padding_src;
731     AVFrame picture_format_temp, picture_crop_temp, picture_pad_temp;
732     uint8_t *buf = NULL, *buf1 = NULL;
733     AVCodecContext *enc, *dec;
734     enum PixelFormat target_pixfmt;
735
736     avcodec_get_frame_defaults(&picture_format_temp);
737     avcodec_get_frame_defaults(&picture_crop_temp);
738     avcodec_get_frame_defaults(&picture_pad_temp);
739
740     enc = ost->st->codec;
741     dec = ist->st->codec;
742
743     /* by default, we output a single frame */
744     nb_frames = 1;
745
746     *frame_size = 0;
747
748     if(video_sync_method){
749         double vdelta;
750         vdelta = get_sync_ipts(ost) / av_q2d(enc->time_base) - ost->sync_opts;
751         //FIXME set to 0.5 after we fix some dts/pts bugs like in avidec.c
752         if (vdelta < -1.1)
753             nb_frames = 0;
754         else if (vdelta > 1.1)
755             nb_frames = lrintf(vdelta);
756 //fprintf(stderr, "vdelta:%f, ost->sync_opts:%lld, ost->sync_ipts:%f nb_frames:%d\n", vdelta, ost->sync_opts, ost->sync_ipts, nb_frames);
757         if (nb_frames == 0){
758             ++nb_frames_drop;
759             if (verbose>2)
760                 fprintf(stderr, "*** drop!\n");
761         }else if (nb_frames > 1) {
762             nb_frames_dup += nb_frames;
763             if (verbose>2)
764                 fprintf(stderr, "*** %d dup!\n", nb_frames-1);
765         }
766     }else
767         ost->sync_opts= lrintf(get_sync_ipts(ost) / av_q2d(enc->time_base));
768
769     nb_frames= FFMIN(nb_frames, max_frames[CODEC_TYPE_VIDEO] - ost->frame_number);
770     if (nb_frames <= 0)
771         return;
772
773     /* convert pixel format if needed */
774     target_pixfmt = ost->video_resample ? PIX_FMT_YUV420P : enc->pix_fmt;
775     if (dec->pix_fmt != target_pixfmt) {
776         int size;
777
778         /* create temporary picture */
779         size = avpicture_get_size(target_pixfmt, dec->width, dec->height);
780         buf = av_malloc(size);
781         if (!buf)
782             return;
783         formatted_picture = &picture_format_temp;
784         avpicture_fill((AVPicture*)formatted_picture, buf, target_pixfmt, dec->width, dec->height);
785
786         if (img_convert((AVPicture*)formatted_picture, target_pixfmt,
787                         (AVPicture *)in_picture, dec->pix_fmt,
788                         dec->width, dec->height) < 0) {
789
790             if (verbose >= 0)
791                 fprintf(stderr, "pixel format conversion not handled\n");
792
793             goto the_end;
794         }
795     } else {
796         formatted_picture = in_picture;
797     }
798
799     if (ost->video_crop) {
800         if (img_crop((AVPicture *)&picture_crop_temp, (AVPicture *)formatted_picture, target_pixfmt, ost->topBand, ost->leftBand) < 0) {
801             av_log(NULL, AV_LOG_ERROR, "error cropping picture\n");
802             goto the_end;
803         }
804         formatted_picture = &picture_crop_temp;
805     }
806
807     final_picture = formatted_picture;
808     padding_src = formatted_picture;
809     resampling_dst = &ost->pict_tmp;
810     if (ost->video_pad) {
811         final_picture = &ost->pict_tmp;
812         if (ost->video_resample) {
813             if (img_crop((AVPicture *)&picture_pad_temp, (AVPicture *)final_picture, target_pixfmt, ost->padtop, ost->padleft) < 0) {
814                 av_log(NULL, AV_LOG_ERROR, "error padding picture\n");
815                 goto the_end;
816             }
817             resampling_dst = &picture_pad_temp;
818         }
819     }
820
821     /* XXX: resampling could be done before raw format conversion in
822        some cases to go faster */
823     /* XXX: only works for YUV420P */
824     if (ost->video_resample) {
825         padding_src = NULL;
826         final_picture = &ost->pict_tmp;
827         img_resample(ost->img_resample_ctx, (AVPicture *)resampling_dst, (AVPicture*)formatted_picture);
828     }
829
830     if (enc->pix_fmt != target_pixfmt) {
831         int size;
832
833         av_free(buf);
834         /* create temporary picture */
835         size = avpicture_get_size(enc->pix_fmt, enc->width, enc->height);
836         buf = av_malloc(size);
837         if (!buf)
838             return;
839         final_picture = &picture_format_temp;
840         avpicture_fill((AVPicture*)final_picture, buf, enc->pix_fmt, enc->width, enc->height);
841
842         if (img_convert((AVPicture*)final_picture, enc->pix_fmt,
843                         (AVPicture*)&ost->pict_tmp, target_pixfmt,
844                         enc->width, enc->height) < 0) {
845
846             if (verbose >= 0)
847                 fprintf(stderr, "pixel format conversion not handled\n");
848
849             goto the_end;
850         }
851     }
852
853     if (ost->video_pad) {
854         img_pad((AVPicture*)final_picture, (AVPicture *)padding_src,
855                 enc->height, enc->width, enc->pix_fmt,
856                 ost->padtop, ost->padbottom, ost->padleft, ost->padright, padcolor);
857     }
858
859     /* duplicates frame if needed */
860     for(i=0;i<nb_frames;i++) {
861         AVPacket pkt;
862         av_init_packet(&pkt);
863         pkt.stream_index= ost->index;
864
865         if (s->oformat->flags & AVFMT_RAWPICTURE) {
866             /* raw pictures are written as AVPicture structure to
867                avoid any copies. We support temorarily the older
868                method. */
869             AVFrame* old_frame = enc->coded_frame;
870             enc->coded_frame = dec->coded_frame; //FIXME/XXX remove this hack
871             pkt.data= (uint8_t *)final_picture;
872             pkt.size=  sizeof(AVPicture);
873             if(dec->coded_frame && enc->coded_frame->pts != AV_NOPTS_VALUE)
874                 pkt.pts= av_rescale_q(enc->coded_frame->pts, enc->time_base, ost->st->time_base);
875             if(dec->coded_frame && dec->coded_frame->key_frame)
876                 pkt.flags |= PKT_FLAG_KEY;
877
878             write_frame(s, &pkt, ost->st->codec, bitstream_filters[ost->file_index][pkt.stream_index]);
879             enc->coded_frame = old_frame;
880         } else {
881             AVFrame big_picture;
882
883             big_picture= *final_picture;
884             /* better than nothing: use input picture interlaced
885                settings */
886             big_picture.interlaced_frame = in_picture->interlaced_frame;
887             if(avctx_opts->flags & (CODEC_FLAG_INTERLACED_DCT|CODEC_FLAG_INTERLACED_ME)){
888                 if(top_field_first == -1)
889                     big_picture.top_field_first = in_picture->top_field_first;
890                 else
891                     big_picture.top_field_first = top_field_first;
892             }
893
894             /* handles sameq here. This is not correct because it may
895                not be a global option */
896             if (same_quality) {
897                 big_picture.quality = ist->st->quality;
898             }else
899                 big_picture.quality = ost->st->quality;
900             if(!me_threshold)
901                 big_picture.pict_type = 0;
902 //            big_picture.pts = AV_NOPTS_VALUE;
903             big_picture.pts= ost->sync_opts;
904 //            big_picture.pts= av_rescale(ost->sync_opts, AV_TIME_BASE*(int64_t)enc->time_base.num, enc->time_base.den);
905 //av_log(NULL, AV_LOG_DEBUG, "%lld -> encoder\n", ost->sync_opts);
906             ret = avcodec_encode_video(enc,
907                                        bit_buffer, bit_buffer_size,
908                                        &big_picture);
909             //enc->frame_number = enc->real_pict_num;
910             if(ret>0){
911                 pkt.data= bit_buffer;
912                 pkt.size= ret;
913                 if(enc->coded_frame && enc->coded_frame->pts != AV_NOPTS_VALUE)
914                     pkt.pts= av_rescale_q(enc->coded_frame->pts, enc->time_base, ost->st->time_base);
915 /*av_log(NULL, AV_LOG_DEBUG, "encoder -> %lld/%lld\n",
916    pkt.pts != AV_NOPTS_VALUE ? av_rescale(pkt.pts, enc->time_base.den, AV_TIME_BASE*(int64_t)enc->time_base.num) : -1,
917    pkt.dts != AV_NOPTS_VALUE ? av_rescale(pkt.dts, enc->time_base.den, AV_TIME_BASE*(int64_t)enc->time_base.num) : -1);*/
918
919                 if(enc->coded_frame && enc->coded_frame->key_frame)
920                     pkt.flags |= PKT_FLAG_KEY;
921                 write_frame(s, &pkt, ost->st->codec, bitstream_filters[ost->file_index][pkt.stream_index]);
922                 *frame_size = ret;
923                 //fprintf(stderr,"\nFrame: %3d %3d size: %5d type: %d",
924                 //        enc->frame_number-1, enc->real_pict_num, ret,
925                 //        enc->pict_type);
926                 /* if two pass, output log */
927                 if (ost->logfile && enc->stats_out) {
928                     fprintf(ost->logfile, "%s", enc->stats_out);
929                 }
930             }
931         }
932         ost->sync_opts++;
933         ost->frame_number++;
934     }
935  the_end:
936     av_free(buf);
937     av_free(buf1);
938 }
939
940 static double psnr(double d){
941     if(d==0) return INFINITY;
942     return -10.0*log(d)/log(10.0);
943 }
944
945 static void do_video_stats(AVFormatContext *os, AVOutputStream *ost,
946                            int frame_size)
947 {
948     static FILE *fvstats=NULL;
949     char filename[40];
950     time_t today2;
951     struct tm *today;
952     AVCodecContext *enc;
953     int frame_number;
954     int64_t ti;
955     double ti1, bitrate, avg_bitrate;
956
957     if (!fvstats) {
958         today2 = time(NULL);
959         today = localtime(&today2);
960         snprintf(filename, sizeof(filename), "vstats_%02d%02d%02d.log", today->tm_hour,
961                                                today->tm_min,
962                                                today->tm_sec);
963         fvstats = fopen(filename,"w");
964         if (!fvstats) {
965             perror("fopen");
966             exit(1);
967         }
968     }
969
970     ti = MAXINT64;
971     enc = ost->st->codec;
972     if (enc->codec_type == CODEC_TYPE_VIDEO) {
973         frame_number = ost->frame_number;
974         fprintf(fvstats, "frame= %5d q= %2.1f ", frame_number, enc->coded_frame->quality/(float)FF_QP2LAMBDA);
975         if (enc->flags&CODEC_FLAG_PSNR)
976             fprintf(fvstats, "PSNR= %6.2f ", psnr(enc->coded_frame->error[0]/(enc->width*enc->height*255.0*255.0)));
977
978         fprintf(fvstats,"f_size= %6d ", frame_size);
979         /* compute pts value */
980         ti1 = ost->sync_opts * av_q2d(enc->time_base);
981         if (ti1 < 0.01)
982             ti1 = 0.01;
983
984         bitrate = (frame_size * 8) / av_q2d(enc->time_base) / 1000.0;
985         avg_bitrate = (double)(video_size * 8) / ti1 / 1000.0;
986         fprintf(fvstats, "s_size= %8.0fkB time= %0.3f br= %7.1fkbits/s avg_br= %7.1fkbits/s ",
987             (double)video_size / 1024, ti1, bitrate, avg_bitrate);
988         fprintf(fvstats,"type= %c\n", av_get_pict_type_char(enc->coded_frame->pict_type));
989     }
990 }
991
992 static void print_report(AVFormatContext **output_files,
993                          AVOutputStream **ost_table, int nb_ostreams,
994                          int is_last_report)
995 {
996     char buf[1024];
997     AVOutputStream *ost;
998     AVFormatContext *oc, *os;
999     int64_t total_size;
1000     AVCodecContext *enc;
1001     int frame_number, vid, i;
1002     double bitrate, ti1, pts;
1003     static int64_t last_time = -1;
1004     static int qp_histogram[52];
1005
1006     if (!is_last_report) {
1007         int64_t cur_time;
1008         /* display the report every 0.5 seconds */
1009         cur_time = av_gettime();
1010         if (last_time == -1) {
1011             last_time = cur_time;
1012             return;
1013         }
1014         if ((cur_time - last_time) < 500000)
1015             return;
1016         last_time = cur_time;
1017     }
1018
1019
1020     oc = output_files[0];
1021
1022     total_size = url_ftell(&oc->pb);
1023
1024     buf[0] = '\0';
1025     ti1 = 1e10;
1026     vid = 0;
1027     for(i=0;i<nb_ostreams;i++) {
1028         ost = ost_table[i];
1029         os = output_files[ost->file_index];
1030         enc = ost->st->codec;
1031         if (vid && enc->codec_type == CODEC_TYPE_VIDEO) {
1032             snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "q=%2.1f ",
1033                     enc->coded_frame->quality/(float)FF_QP2LAMBDA);
1034         }
1035         if (!vid && enc->codec_type == CODEC_TYPE_VIDEO) {
1036             frame_number = ost->frame_number;
1037             snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "frame=%5d q=%3.1f ",
1038                     frame_number, enc->coded_frame ? enc->coded_frame->quality/(float)FF_QP2LAMBDA : -1);
1039             if(is_last_report)
1040                 snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "L");
1041             if(qp_hist && enc->coded_frame){
1042                 int j;
1043                 int qp= lrintf(enc->coded_frame->quality/(float)FF_QP2LAMBDA);
1044                 if(qp>=0 && qp<sizeof(qp_histogram)/sizeof(int))
1045                     qp_histogram[qp]++;
1046                 for(j=0; j<32; j++)
1047                     snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "%X", (int)lrintf(log(qp_histogram[j]+1)/log(2)));
1048             }
1049             if (enc->flags&CODEC_FLAG_PSNR){
1050                 int j;
1051                 double error, error_sum=0;
1052                 double scale, scale_sum=0;
1053                 char type[3]= {'Y','U','V'};
1054                 snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "PSNR=");
1055                 for(j=0; j<3; j++){
1056                     if(is_last_report){
1057                         error= enc->error[j];
1058                         scale= enc->width*enc->height*255.0*255.0*frame_number;
1059                     }else{
1060                         error= enc->coded_frame->error[j];
1061                         scale= enc->width*enc->height*255.0*255.0;
1062                     }
1063                     if(j) scale/=4;
1064                     error_sum += error;
1065                     scale_sum += scale;
1066                     snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "%c:%2.2f ", type[j], psnr(error/scale));
1067                 }
1068                 snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "*:%2.2f ", psnr(error_sum/scale_sum));
1069             }
1070             vid = 1;
1071         }
1072         /* compute min output value */
1073         pts = (double)ost->st->pts.val * ost->st->time_base.num / ost->st->time_base.den;
1074         if ((pts < ti1) && (pts > 0))
1075             ti1 = pts;
1076     }
1077     if (ti1 < 0.01)
1078         ti1 = 0.01;
1079
1080     if (verbose || is_last_report) {
1081         bitrate = (double)(total_size * 8) / ti1 / 1000.0;
1082
1083         snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf),
1084             "size=%8.0fkB time=%0.1f bitrate=%6.1fkbits/s",
1085             (double)total_size / 1024, ti1, bitrate);
1086
1087         if (verbose > 1)
1088           snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), " dup=%d drop=%d",
1089                   nb_frames_dup, nb_frames_drop);
1090
1091         if (verbose >= 0)
1092             fprintf(stderr, "%s    \r", buf);
1093
1094         fflush(stderr);
1095     }
1096
1097     if (is_last_report && verbose >= 0){
1098         int64_t raw= audio_size + video_size + extra_size;
1099         fprintf(stderr, "\n");
1100         fprintf(stderr, "video:%1.0fkB audio:%1.0fkB global headers:%1.0fkB muxing overhead %f%%\n",
1101                 video_size/1024.0,
1102                 audio_size/1024.0,
1103                 extra_size/1024.0,
1104                 100.0*(total_size - raw)/raw
1105         );
1106     }
1107 }
1108
1109 /* pkt = NULL means EOF (needed to flush decoder buffers) */
1110 static int output_packet(AVInputStream *ist, int ist_index,
1111                          AVOutputStream **ost_table, int nb_ostreams,
1112                          const AVPacket *pkt)
1113 {
1114     AVFormatContext *os;
1115     AVOutputStream *ost;
1116     uint8_t *ptr;
1117     int len, ret, i;
1118     uint8_t *data_buf;
1119     int data_size, got_picture;
1120     AVFrame picture;
1121     void *buffer_to_free;
1122     static unsigned int samples_size= 0;
1123     static short *samples= NULL;
1124     AVSubtitle subtitle, *subtitle_to_free;
1125     int got_subtitle;
1126
1127     if(!pkt){
1128         ist->pts= ist->next_pts; // needed for last packet if vsync=0
1129     } else if (pkt->dts != AV_NOPTS_VALUE) { //FIXME seems redundant, as libavformat does this too
1130         ist->next_pts = ist->pts = av_rescale_q(pkt->dts, ist->st->time_base, AV_TIME_BASE_Q);
1131     } else {
1132 //        assert(ist->pts == ist->next_pts);
1133     }
1134
1135     if (pkt == NULL) {
1136         /* EOF handling */
1137         ptr = NULL;
1138         len = 0;
1139         goto handle_eof;
1140     }
1141
1142     len = pkt->size;
1143     ptr = pkt->data;
1144     while (len > 0) {
1145     handle_eof:
1146         /* decode the packet if needed */
1147         data_buf = NULL; /* fail safe */
1148         data_size = 0;
1149         subtitle_to_free = NULL;
1150         if (ist->decoding_needed) {
1151             switch(ist->st->codec->codec_type) {
1152             case CODEC_TYPE_AUDIO:{
1153                 if(pkt)
1154                     samples= av_fast_realloc(samples, &samples_size, FFMAX(pkt->size, AVCODEC_MAX_AUDIO_FRAME_SIZE));
1155                     /* XXX: could avoid copy if PCM 16 bits with same
1156                        endianness as CPU */
1157                 ret = avcodec_decode_audio(ist->st->codec, samples, &data_size,
1158                                            ptr, len);
1159                 if (ret < 0)
1160                     goto fail_decode;
1161                 ptr += ret;
1162                 len -= ret;
1163                 /* Some bug in mpeg audio decoder gives */
1164                 /* data_size < 0, it seems they are overflows */
1165                 if (data_size <= 0) {
1166                     /* no audio frame */
1167                     continue;
1168                 }
1169                 data_buf = (uint8_t *)samples;
1170                 ist->next_pts += ((int64_t)AV_TIME_BASE/2 * data_size) /
1171                     (ist->st->codec->sample_rate * ist->st->codec->channels);
1172                 break;}
1173             case CODEC_TYPE_VIDEO:
1174                     data_size = (ist->st->codec->width * ist->st->codec->height * 3) / 2;
1175                     /* XXX: allocate picture correctly */
1176                     avcodec_get_frame_defaults(&picture);
1177
1178                     ret = avcodec_decode_video(ist->st->codec,
1179                                                &picture, &got_picture, ptr, len);
1180                     ist->st->quality= picture.quality;
1181                     if (ret < 0)
1182                         goto fail_decode;
1183                     if (!got_picture) {
1184                         /* no picture yet */
1185                         goto discard_packet;
1186                     }
1187                     if (ist->st->codec->time_base.num != 0) {
1188                         ist->next_pts += ((int64_t)AV_TIME_BASE *
1189                                           ist->st->codec->time_base.num) /
1190                             ist->st->codec->time_base.den;
1191                     }
1192                     len = 0;
1193                     break;
1194             case CODEC_TYPE_SUBTITLE:
1195                 ret = avcodec_decode_subtitle(ist->st->codec,
1196                                               &subtitle, &got_subtitle, ptr, len);
1197                 if (ret < 0)
1198                     goto fail_decode;
1199                 if (!got_subtitle) {
1200                     goto discard_packet;
1201                 }
1202                 subtitle_to_free = &subtitle;
1203                 len = 0;
1204                 break;
1205             default:
1206                 goto fail_decode;
1207             }
1208         } else {
1209                 switch(ist->st->codec->codec_type) {
1210                 case CODEC_TYPE_AUDIO:
1211                     ist->next_pts += ((int64_t)AV_TIME_BASE * ist->st->codec->frame_size) /
1212                         (ist->st->codec->sample_rate * ist->st->codec->channels);
1213                     break;
1214                 case CODEC_TYPE_VIDEO:
1215                     if (ist->st->codec->time_base.num != 0) {
1216                         ist->next_pts += ((int64_t)AV_TIME_BASE *
1217                                           ist->st->codec->time_base.num) /
1218                             ist->st->codec->time_base.den;
1219                     }
1220                     break;
1221                 }
1222                 data_buf = ptr;
1223                 data_size = len;
1224                 ret = len;
1225                 len = 0;
1226             }
1227
1228             buffer_to_free = NULL;
1229             if (ist->st->codec->codec_type == CODEC_TYPE_VIDEO) {
1230                 pre_process_video_frame(ist, (AVPicture *)&picture,
1231                                         &buffer_to_free);
1232             }
1233
1234             // preprocess audio (volume)
1235             if (ist->st->codec->codec_type == CODEC_TYPE_AUDIO) {
1236                 if (audio_volume != 256) {
1237                     short *volp;
1238                     volp = samples;
1239                     for(i=0;i<(data_size / sizeof(short));i++) {
1240                         int v = ((*volp) * audio_volume + 128) >> 8;
1241                         if (v < -32768) v = -32768;
1242                         if (v >  32767) v = 32767;
1243                         *volp++ = v;
1244                     }
1245                 }
1246             }
1247
1248             /* frame rate emulation */
1249             if (ist->st->codec->rate_emu) {
1250                 int64_t pts = av_rescale((int64_t) ist->frame * ist->st->codec->time_base.num, 1000000, ist->st->codec->time_base.den);
1251                 int64_t now = av_gettime() - ist->start;
1252                 if (pts > now)
1253                     usleep(pts - now);
1254
1255                 ist->frame++;
1256             }
1257
1258 #if 0
1259             /* mpeg PTS deordering : if it is a P or I frame, the PTS
1260                is the one of the next displayed one */
1261             /* XXX: add mpeg4 too ? */
1262             if (ist->st->codec->codec_id == CODEC_ID_MPEG1VIDEO) {
1263                 if (ist->st->codec->pict_type != B_TYPE) {
1264                     int64_t tmp;
1265                     tmp = ist->last_ip_pts;
1266                     ist->last_ip_pts  = ist->frac_pts.val;
1267                     ist->frac_pts.val = tmp;
1268                 }
1269             }
1270 #endif
1271             /* if output time reached then transcode raw format,
1272                encode packets and output them */
1273             if (start_time == 0 || ist->pts >= start_time)
1274                 for(i=0;i<nb_ostreams;i++) {
1275                     int frame_size;
1276
1277                     ost = ost_table[i];
1278                     if (ost->source_index == ist_index) {
1279                         os = output_files[ost->file_index];
1280
1281 #if 0
1282                         printf("%d: got pts=%0.3f %0.3f\n", i,
1283                                (double)pkt->pts / AV_TIME_BASE,
1284                                ((double)ist->pts / AV_TIME_BASE) -
1285                                ((double)ost->st->pts.val * ost->st->time_base.num / ost->st->time_base.den));
1286 #endif
1287                         /* set the input output pts pairs */
1288                         //ost->sync_ipts = (double)(ist->pts + input_files_ts_offset[ist->file_index] - start_time)/ AV_TIME_BASE;
1289
1290                         if (ost->encoding_needed) {
1291                             switch(ost->st->codec->codec_type) {
1292                             case CODEC_TYPE_AUDIO:
1293                                 do_audio_out(os, ost, ist, data_buf, data_size);
1294                                 break;
1295                             case CODEC_TYPE_VIDEO:
1296                                     do_video_out(os, ost, ist, &picture, &frame_size);
1297                                     video_size += frame_size;
1298                                     if (do_vstats && frame_size)
1299                                         do_video_stats(os, ost, frame_size);
1300                                 break;
1301                             case CODEC_TYPE_SUBTITLE:
1302                                 do_subtitle_out(os, ost, ist, &subtitle,
1303                                                 pkt->pts);
1304                                 break;
1305                             default:
1306                                 av_abort();
1307                             }
1308                         } else {
1309                             AVFrame avframe; //FIXME/XXX remove this
1310                             AVPacket opkt;
1311                             av_init_packet(&opkt);
1312
1313                             /* no reencoding needed : output the packet directly */
1314                             /* force the input stream PTS */
1315
1316                             avcodec_get_frame_defaults(&avframe);
1317                             ost->st->codec->coded_frame= &avframe;
1318                             avframe.key_frame = pkt->flags & PKT_FLAG_KEY;
1319
1320                             if(ost->st->codec->codec_type == CODEC_TYPE_AUDIO)
1321                                 audio_size += data_size;
1322                             else if (ost->st->codec->codec_type == CODEC_TYPE_VIDEO) {
1323                                 video_size += data_size;
1324                                 ost->sync_opts++;
1325                             }
1326
1327                             opkt.stream_index= ost->index;
1328                             if(pkt->pts != AV_NOPTS_VALUE)
1329                                 opkt.pts= av_rescale_q(av_rescale_q(pkt->pts, ist->st->time_base, AV_TIME_BASE_Q) + input_files_ts_offset[ist->file_index], AV_TIME_BASE_Q,  ost->st->time_base);
1330                             else
1331                                 opkt.pts= AV_NOPTS_VALUE;
1332
1333                             {
1334                                 int64_t dts;
1335                                 if (pkt->dts == AV_NOPTS_VALUE)
1336                                     dts = ist->next_pts;
1337                                 else
1338                                     dts= av_rescale_q(pkt->dts, ist->st->time_base, AV_TIME_BASE_Q);
1339                                 opkt.dts= av_rescale_q(dts + input_files_ts_offset[ist->file_index], AV_TIME_BASE_Q,  ost->st->time_base);
1340                             }
1341                             opkt.flags= pkt->flags;
1342
1343                             //FIXME remove the following 2 lines they shall be replaced by the bitstream filters
1344                             if(av_parser_change(ist->st->parser, ost->st->codec, &opkt.data, &opkt.size, data_buf, data_size, pkt->flags & PKT_FLAG_KEY))
1345                                 opkt.destruct= av_destruct_packet;
1346
1347                             write_frame(os, &opkt, ost->st->codec, bitstream_filters[ost->file_index][pkt->stream_index]);
1348                             ost->st->codec->frame_number++;
1349                             ost->frame_number++;
1350                             av_free_packet(&opkt);
1351                         }
1352                     }
1353                 }
1354             av_free(buffer_to_free);
1355             /* XXX: allocate the subtitles in the codec ? */
1356             if (subtitle_to_free) {
1357                 if (subtitle_to_free->rects != NULL) {
1358                     for (i = 0; i < subtitle_to_free->num_rects; i++) {
1359                         av_free(subtitle_to_free->rects[i].bitmap);
1360                         av_free(subtitle_to_free->rects[i].rgba_palette);
1361                     }
1362                     av_freep(&subtitle_to_free->rects);
1363                 }
1364                 subtitle_to_free->num_rects = 0;
1365                 subtitle_to_free = NULL;
1366             }
1367         }
1368  discard_packet:
1369     if (pkt == NULL) {
1370         /* EOF handling */
1371
1372         for(i=0;i<nb_ostreams;i++) {
1373             ost = ost_table[i];
1374             if (ost->source_index == ist_index) {
1375                 AVCodecContext *enc= ost->st->codec;
1376                 os = output_files[ost->file_index];
1377
1378                 if(ost->st->codec->codec_type == CODEC_TYPE_AUDIO && enc->frame_size <=1)
1379                     continue;
1380                 if(ost->st->codec->codec_type == CODEC_TYPE_VIDEO && (os->oformat->flags & AVFMT_RAWPICTURE))
1381                     continue;
1382
1383                 if (ost->encoding_needed) {
1384                     for(;;) {
1385                         AVPacket pkt;
1386                         int fifo_bytes;
1387                         av_init_packet(&pkt);
1388                         pkt.stream_index= ost->index;
1389
1390                         switch(ost->st->codec->codec_type) {
1391                         case CODEC_TYPE_AUDIO:
1392                             fifo_bytes = fifo_size(&ost->fifo, NULL);
1393                             ret = 0;
1394                             /* encode any samples remaining in fifo */
1395                             if(fifo_bytes > 0 && enc->codec->capabilities & CODEC_CAP_SMALL_LAST_FRAME) {
1396                                 int fs_tmp = enc->frame_size;
1397                                 enc->frame_size = fifo_bytes / (2 * enc->channels);
1398                                 if(fifo_read(&ost->fifo, (uint8_t *)samples, fifo_bytes,
1399                                         &ost->fifo.rptr) == 0) {
1400                                     ret = avcodec_encode_audio(enc, bit_buffer, bit_buffer_size, samples);
1401                                 }
1402                                 enc->frame_size = fs_tmp;
1403                             }
1404                             if(ret <= 0) {
1405                                 ret = avcodec_encode_audio(enc, bit_buffer, bit_buffer_size, NULL);
1406                             }
1407                             audio_size += ret;
1408                             pkt.flags |= PKT_FLAG_KEY;
1409                             break;
1410                         case CODEC_TYPE_VIDEO:
1411                             ret = avcodec_encode_video(enc, bit_buffer, bit_buffer_size, NULL);
1412                             video_size += ret;
1413                             if(enc->coded_frame && enc->coded_frame->key_frame)
1414                                 pkt.flags |= PKT_FLAG_KEY;
1415                             if (ost->logfile && enc->stats_out) {
1416                                 fprintf(ost->logfile, "%s", enc->stats_out);
1417                             }
1418                             break;
1419                         default:
1420                             ret=-1;
1421                         }
1422
1423                         if(ret<=0)
1424                             break;
1425                         pkt.data= bit_buffer;
1426                         pkt.size= ret;
1427                         if(enc->coded_frame && enc->coded_frame->pts != AV_NOPTS_VALUE)
1428                             pkt.pts= av_rescale_q(enc->coded_frame->pts, enc->time_base, ost->st->time_base);
1429                         write_frame(os, &pkt, ost->st->codec, bitstream_filters[ost->file_index][pkt.stream_index]);
1430                     }
1431                 }
1432             }
1433         }
1434     }
1435
1436     return 0;
1437  fail_decode:
1438     return -1;
1439 }
1440
1441
1442 /*
1443  * The following code is the main loop of the file converter
1444  */
1445 static int av_encode(AVFormatContext **output_files,
1446                      int nb_output_files,
1447                      AVFormatContext **input_files,
1448                      int nb_input_files,
1449                      AVStreamMap *stream_maps, int nb_stream_maps)
1450 {
1451     int ret, i, j, k, n, nb_istreams = 0, nb_ostreams = 0;
1452     AVFormatContext *is, *os;
1453     AVCodecContext *codec, *icodec;
1454     AVOutputStream *ost, **ost_table = NULL;
1455     AVInputStream *ist, **ist_table = NULL;
1456     AVInputFile *file_table;
1457     AVFormatContext *stream_no_data;
1458     int key;
1459
1460     file_table= (AVInputFile*) av_mallocz(nb_input_files * sizeof(AVInputFile));
1461     if (!file_table)
1462         goto fail;
1463
1464     /* input stream init */
1465     j = 0;
1466     for(i=0;i<nb_input_files;i++) {
1467         is = input_files[i];
1468         file_table[i].ist_index = j;
1469         file_table[i].nb_streams = is->nb_streams;
1470         j += is->nb_streams;
1471     }
1472     nb_istreams = j;
1473
1474     ist_table = av_mallocz(nb_istreams * sizeof(AVInputStream *));
1475     if (!ist_table)
1476         goto fail;
1477
1478     for(i=0;i<nb_istreams;i++) {
1479         ist = av_mallocz(sizeof(AVInputStream));
1480         if (!ist)
1481             goto fail;
1482         ist_table[i] = ist;
1483     }
1484     j = 0;
1485     for(i=0;i<nb_input_files;i++) {
1486         is = input_files[i];
1487         for(k=0;k<is->nb_streams;k++) {
1488             ist = ist_table[j++];
1489             ist->st = is->streams[k];
1490             ist->file_index = i;
1491             ist->index = k;
1492             ist->discard = 1; /* the stream is discarded by default
1493                                  (changed later) */
1494
1495             if (ist->st->codec->rate_emu) {
1496                 ist->start = av_gettime();
1497                 ist->frame = 0;
1498             }
1499         }
1500     }
1501
1502     /* output stream init */
1503     nb_ostreams = 0;
1504     for(i=0;i<nb_output_files;i++) {
1505         os = output_files[i];
1506         nb_ostreams += os->nb_streams;
1507     }
1508     if (nb_stream_maps > 0 && nb_stream_maps != nb_ostreams) {
1509         fprintf(stderr, "Number of stream maps must match number of output streams\n");
1510         exit(1);
1511     }
1512
1513     /* Sanity check the mapping args -- do the input files & streams exist? */
1514     for(i=0;i<nb_stream_maps;i++) {
1515         int fi = stream_maps[i].file_index;
1516         int si = stream_maps[i].stream_index;
1517
1518         if (fi < 0 || fi > nb_input_files - 1 ||
1519             si < 0 || si > file_table[fi].nb_streams - 1) {
1520             fprintf(stderr,"Could not find input stream #%d.%d\n", fi, si);
1521             exit(1);
1522         }
1523         fi = stream_maps[i].sync_file_index;
1524         si = stream_maps[i].sync_stream_index;
1525         if (fi < 0 || fi > nb_input_files - 1 ||
1526             si < 0 || si > file_table[fi].nb_streams - 1) {
1527             fprintf(stderr,"Could not find sync stream #%d.%d\n", fi, si);
1528             exit(1);
1529         }
1530     }
1531
1532     ost_table = av_mallocz(sizeof(AVOutputStream *) * nb_ostreams);
1533     if (!ost_table)
1534         goto fail;
1535     for(i=0;i<nb_ostreams;i++) {
1536         ost = av_mallocz(sizeof(AVOutputStream));
1537         if (!ost)
1538             goto fail;
1539         ost_table[i] = ost;
1540     }
1541
1542     n = 0;
1543     for(k=0;k<nb_output_files;k++) {
1544         os = output_files[k];
1545         for(i=0;i<os->nb_streams;i++) {
1546             int found;
1547             ost = ost_table[n++];
1548             ost->file_index = k;
1549             ost->index = i;
1550             ost->st = os->streams[i];
1551             if (nb_stream_maps > 0) {
1552                 ost->source_index = file_table[stream_maps[n-1].file_index].ist_index +
1553                     stream_maps[n-1].stream_index;
1554
1555                 /* Sanity check that the stream types match */
1556                 if (ist_table[ost->source_index]->st->codec->codec_type != ost->st->codec->codec_type) {
1557                     fprintf(stderr, "Codec type mismatch for mapping #%d.%d -> #%d.%d\n",
1558                         stream_maps[n-1].file_index, stream_maps[n-1].stream_index,
1559                         ost->file_index, ost->index);
1560                     exit(1);
1561                 }
1562
1563             } else {
1564                 /* get corresponding input stream index : we select the first one with the right type */
1565                 found = 0;
1566                 for(j=0;j<nb_istreams;j++) {
1567                     ist = ist_table[j];
1568                     if (ist->discard &&
1569                         ist->st->codec->codec_type == ost->st->codec->codec_type) {
1570                         ost->source_index = j;
1571                         found = 1;
1572                         break;
1573                     }
1574                 }
1575
1576                 if (!found) {
1577                     /* try again and reuse existing stream */
1578                     for(j=0;j<nb_istreams;j++) {
1579                         ist = ist_table[j];
1580                         if (ist->st->codec->codec_type == ost->st->codec->codec_type) {
1581                             ost->source_index = j;
1582                             found = 1;
1583                         }
1584                     }
1585                     if (!found) {
1586                         fprintf(stderr, "Could not find input stream matching output stream #%d.%d\n",
1587                                 ost->file_index, ost->index);
1588                         exit(1);
1589                     }
1590                 }
1591             }
1592             ist = ist_table[ost->source_index];
1593             ist->discard = 0;
1594             ost->sync_ist = (nb_stream_maps > 0) ?
1595                 ist_table[file_table[stream_maps[n-1].sync_file_index].ist_index +
1596                          stream_maps[n-1].sync_stream_index] : ist;
1597         }
1598     }
1599
1600     /* for each output stream, we compute the right encoding parameters */
1601     for(i=0;i<nb_ostreams;i++) {
1602         ost = ost_table[i];
1603         ist = ist_table[ost->source_index];
1604
1605         codec = ost->st->codec;
1606         icodec = ist->st->codec;
1607
1608         if (ost->st->stream_copy) {
1609             /* if stream_copy is selected, no need to decode or encode */
1610             codec->codec_id = icodec->codec_id;
1611             codec->codec_type = icodec->codec_type;
1612             if(!codec->codec_tag) codec->codec_tag = icodec->codec_tag;
1613             codec->bit_rate = icodec->bit_rate;
1614             codec->extradata= icodec->extradata;
1615             codec->extradata_size= icodec->extradata_size;
1616             codec->time_base = icodec->time_base;
1617             switch(codec->codec_type) {
1618             case CODEC_TYPE_AUDIO:
1619                 codec->sample_rate = icodec->sample_rate;
1620                 codec->channels = icodec->channels;
1621                 codec->frame_size = icodec->frame_size;
1622                 codec->block_align= icodec->block_align;
1623                 break;
1624             case CODEC_TYPE_VIDEO:
1625                 codec->pix_fmt = icodec->pix_fmt;
1626                 codec->width = icodec->width;
1627                 codec->height = icodec->height;
1628                 codec->has_b_frames = icodec->has_b_frames;
1629                 break;
1630             case CODEC_TYPE_SUBTITLE:
1631                 break;
1632             default:
1633                 av_abort();
1634             }
1635         } else {
1636             switch(codec->codec_type) {
1637             case CODEC_TYPE_AUDIO:
1638                 if (fifo_init(&ost->fifo, 2 * MAX_AUDIO_PACKET_SIZE))
1639                     goto fail;
1640
1641                 if (codec->channels == icodec->channels &&
1642                     codec->sample_rate == icodec->sample_rate) {
1643                     ost->audio_resample = 0;
1644                 } else {
1645                     if (codec->channels != icodec->channels &&
1646                         (icodec->codec_id == CODEC_ID_AC3 ||
1647                          icodec->codec_id == CODEC_ID_DTS)) {
1648                         /* Special case for 5:1 AC3 and DTS input */
1649                         /* and mono or stereo output      */
1650                         /* Request specific number of channels */
1651                         icodec->channels = codec->channels;
1652                         if (codec->sample_rate == icodec->sample_rate)
1653                             ost->audio_resample = 0;
1654                         else {
1655                             ost->audio_resample = 1;
1656                         }
1657                     } else {
1658                         ost->audio_resample = 1;
1659                     }
1660                 }
1661                 if(audio_sync_method>1)
1662                     ost->audio_resample = 1;
1663
1664                 if(ost->audio_resample){
1665                     ost->resample = audio_resample_init(codec->channels, icodec->channels,
1666                                                     codec->sample_rate, icodec->sample_rate);
1667                     if(!ost->resample){
1668                         printf("Can't resample.  Aborting.\n");
1669                         av_abort();
1670                     }
1671                 }
1672                 ist->decoding_needed = 1;
1673                 ost->encoding_needed = 1;
1674                 break;
1675             case CODEC_TYPE_VIDEO:
1676                 ost->video_crop = ((frame_leftBand + frame_rightBand + frame_topBand + frame_bottomBand) != 0);
1677                 ost->video_pad = ((frame_padleft + frame_padright + frame_padtop + frame_padbottom) != 0);
1678                 ost->video_resample = ((codec->width != icodec->width -
1679                                 (frame_leftBand + frame_rightBand) +
1680                                 (frame_padleft + frame_padright)) ||
1681                         (codec->height != icodec->height -
1682                                 (frame_topBand  + frame_bottomBand) +
1683                                 (frame_padtop + frame_padbottom)));
1684                 if (ost->video_crop) {
1685                     ost->topBand = frame_topBand;
1686                     ost->leftBand = frame_leftBand;
1687                 }
1688                 if (ost->video_pad) {
1689                     ost->padtop = frame_padtop;
1690                     ost->padleft = frame_padleft;
1691                     ost->padbottom = frame_padbottom;
1692                     ost->padright = frame_padright;
1693                     if (!ost->video_resample) {
1694                         avcodec_get_frame_defaults(&ost->pict_tmp);
1695                         if( avpicture_alloc( (AVPicture*)&ost->pict_tmp, codec->pix_fmt,
1696                                          codec->width, codec->height ) )
1697                             goto fail;
1698                     }
1699                 }
1700                 if (ost->video_resample) {
1701                     avcodec_get_frame_defaults(&ost->pict_tmp);
1702                     if( avpicture_alloc( (AVPicture*)&ost->pict_tmp, PIX_FMT_YUV420P,
1703                                          codec->width, codec->height ) )
1704                         goto fail;
1705
1706                     ost->img_resample_ctx = img_resample_init(
1707                             codec->width - (frame_padleft + frame_padright),
1708                             codec->height - (frame_padtop + frame_padbottom),
1709                             icodec->width - (frame_leftBand + frame_rightBand),
1710                             icodec->height - (frame_topBand + frame_bottomBand));
1711
1712                 }
1713                 ost->encoding_needed = 1;
1714                 ist->decoding_needed = 1;
1715                 break;
1716             case CODEC_TYPE_SUBTITLE:
1717                 ost->encoding_needed = 1;
1718                 ist->decoding_needed = 1;
1719                 break;
1720             default:
1721                 av_abort();
1722                 break;
1723             }
1724             /* two pass mode */
1725             if (ost->encoding_needed &&
1726                 (codec->flags & (CODEC_FLAG_PASS1 | CODEC_FLAG_PASS2))) {
1727                 char logfilename[1024];
1728                 FILE *f;
1729                 int size;
1730                 char *logbuffer;
1731
1732                 snprintf(logfilename, sizeof(logfilename), "%s-%d.log",
1733                          pass_logfilename ?
1734                          pass_logfilename : DEFAULT_PASS_LOGFILENAME, i);
1735                 if (codec->flags & CODEC_FLAG_PASS1) {
1736                     f = fopen(logfilename, "w");
1737                     if (!f) {
1738                         perror(logfilename);
1739                         exit(1);
1740                     }
1741                     ost->logfile = f;
1742                 } else {
1743                     /* read the log file */
1744                     f = fopen(logfilename, "r");
1745                     if (!f) {
1746                         perror(logfilename);
1747                         exit(1);
1748                     }
1749                     fseek(f, 0, SEEK_END);
1750                     size = ftell(f);
1751                     fseek(f, 0, SEEK_SET);
1752                     logbuffer = av_malloc(size + 1);
1753                     if (!logbuffer) {
1754                         fprintf(stderr, "Could not allocate log buffer\n");
1755                         exit(1);
1756                     }
1757                     size = fread(logbuffer, 1, size, f);
1758                     fclose(f);
1759                     logbuffer[size] = '\0';
1760                     codec->stats_in = logbuffer;
1761                 }
1762             }
1763         }
1764         if(codec->codec_type == CODEC_TYPE_VIDEO){
1765             int size= codec->width * codec->height;
1766             bit_buffer_size= FFMAX(bit_buffer_size, 4*size);
1767         }
1768     }
1769
1770     if (!bit_buffer)
1771         bit_buffer = av_malloc(bit_buffer_size);
1772     if (!bit_buffer)
1773         goto fail;
1774
1775     /* dump the file output parameters - cannot be done before in case
1776        of stream copy */
1777     for(i=0;i<nb_output_files;i++) {
1778         dump_format(output_files[i], i, output_files[i]->filename, 1);
1779     }
1780
1781     /* dump the stream mapping */
1782     if (verbose >= 0) {
1783         fprintf(stderr, "Stream mapping:\n");
1784         for(i=0;i<nb_ostreams;i++) {
1785             ost = ost_table[i];
1786             fprintf(stderr, "  Stream #%d.%d -> #%d.%d",
1787                     ist_table[ost->source_index]->file_index,
1788                     ist_table[ost->source_index]->index,
1789                     ost->file_index,
1790                     ost->index);
1791             if (ost->sync_ist != ist_table[ost->source_index])
1792                 fprintf(stderr, " [sync #%d.%d]",
1793                         ost->sync_ist->file_index,
1794                         ost->sync_ist->index);
1795             fprintf(stderr, "\n");
1796         }
1797     }
1798
1799     /* open each encoder */
1800     for(i=0;i<nb_ostreams;i++) {
1801         ost = ost_table[i];
1802         if (ost->encoding_needed) {
1803             AVCodec *codec;
1804             codec = avcodec_find_encoder(ost->st->codec->codec_id);
1805             if (!codec) {
1806                 fprintf(stderr, "Unsupported codec for output stream #%d.%d\n",
1807                         ost->file_index, ost->index);
1808                 exit(1);
1809             }
1810             if (avcodec_open(ost->st->codec, codec) < 0) {
1811                 fprintf(stderr, "Error while opening codec for output stream #%d.%d - maybe incorrect parameters such as bit_rate, rate, width or height\n",
1812                         ost->file_index, ost->index);
1813                 exit(1);
1814             }
1815             extra_size += ost->st->codec->extradata_size;
1816         }
1817     }
1818
1819     /* open each decoder */
1820     for(i=0;i<nb_istreams;i++) {
1821         ist = ist_table[i];
1822         if (ist->decoding_needed) {
1823             AVCodec *codec;
1824             codec = avcodec_find_decoder(ist->st->codec->codec_id);
1825             if (!codec) {
1826                 fprintf(stderr, "Unsupported codec (id=%d) for input stream #%d.%d\n",
1827                         ist->st->codec->codec_id, ist->file_index, ist->index);
1828                 exit(1);
1829             }
1830             if (avcodec_open(ist->st->codec, codec) < 0) {
1831                 fprintf(stderr, "Error while opening codec for input stream #%d.%d\n",
1832                         ist->file_index, ist->index);
1833                 exit(1);
1834             }
1835             //if (ist->st->codec->codec_type == CODEC_TYPE_VIDEO)
1836             //    ist->st->codec->flags |= CODEC_FLAG_REPEAT_FIELD;
1837         }
1838     }
1839
1840     /* init pts */
1841     for(i=0;i<nb_istreams;i++) {
1842         ist = ist_table[i];
1843         is = input_files[ist->file_index];
1844         ist->pts = 0;
1845         ist->next_pts = av_rescale_q(ist->st->start_time, ist->st->time_base, AV_TIME_BASE_Q);
1846         if(ist->st->start_time == AV_NOPTS_VALUE)
1847             ist->next_pts=0;
1848         if(input_files_ts_offset[ist->file_index])
1849             ist->next_pts= AV_NOPTS_VALUE;
1850         ist->is_start = 1;
1851     }
1852
1853     /* compute buffer size max (should use a complete heuristic) */
1854     for(i=0;i<nb_input_files;i++) {
1855         file_table[i].buffer_size_max = 2048;
1856     }
1857
1858     /* set meta data information from input file if required */
1859     for (i=0;i<nb_meta_data_maps;i++) {
1860         AVFormatContext *out_file;
1861         AVFormatContext *in_file;
1862
1863         int out_file_index = meta_data_maps[i].out_file;
1864         int in_file_index = meta_data_maps[i].in_file;
1865         if ( out_file_index < 0 || out_file_index >= nb_output_files ) {
1866             fprintf(stderr, "Invalid output file index %d map_meta_data(%d,%d)\n", out_file_index, out_file_index, in_file_index);
1867             ret = -EINVAL;
1868             goto fail;
1869         }
1870         if ( in_file_index < 0 || in_file_index >= nb_input_files ) {
1871             fprintf(stderr, "Invalid input file index %d map_meta_data(%d,%d)\n", in_file_index, out_file_index, in_file_index);
1872             ret = -EINVAL;
1873             goto fail;
1874         }
1875
1876         out_file = output_files[out_file_index];
1877         in_file = input_files[in_file_index];
1878
1879         strcpy(out_file->title, in_file->title);
1880         strcpy(out_file->author, in_file->author);
1881         strcpy(out_file->copyright, in_file->copyright);
1882         strcpy(out_file->comment, in_file->comment);
1883         strcpy(out_file->album, in_file->album);
1884         out_file->year = in_file->year;
1885         out_file->track = in_file->track;
1886         strcpy(out_file->genre, in_file->genre);
1887     }
1888
1889     /* open files and write file headers */
1890     for(i=0;i<nb_output_files;i++) {
1891         os = output_files[i];
1892         if (av_write_header(os) < 0) {
1893             fprintf(stderr, "Could not write header for output file #%d (incorrect codec parameters ?)\n", i);
1894             ret = -EINVAL;
1895             goto fail;
1896         }
1897     }
1898
1899 #ifndef CONFIG_WIN32
1900     if ( !using_stdin && verbose >= 0) {
1901         fprintf(stderr, "Press [q] to stop encoding\n");
1902         url_set_interrupt_cb(decode_interrupt_cb);
1903     }
1904 #endif
1905     term_init();
1906
1907     stream_no_data = 0;
1908     key = -1;
1909
1910     for(; received_sigterm == 0;) {
1911         int file_index, ist_index;
1912         AVPacket pkt;
1913         double ipts_min;
1914         double opts_min;
1915
1916     redo:
1917         ipts_min= 1e100;
1918         opts_min= 1e100;
1919         /* if 'q' pressed, exits */
1920         if (!using_stdin) {
1921             if (q_pressed)
1922                 break;
1923             /* read_key() returns 0 on EOF */
1924             key = read_key();
1925             if (key == 'q')
1926                 break;
1927         }
1928
1929         /* select the stream that we must read now by looking at the
1930            smallest output pts */
1931         file_index = -1;
1932         for(i=0;i<nb_ostreams;i++) {
1933             double ipts, opts;
1934             ost = ost_table[i];
1935             os = output_files[ost->file_index];
1936             ist = ist_table[ost->source_index];
1937             if(ost->st->codec->codec_type == CODEC_TYPE_VIDEO)
1938                 opts = ost->sync_opts * av_q2d(ost->st->codec->time_base);
1939             else
1940                 opts = ost->st->pts.val * av_q2d(ost->st->time_base);
1941             ipts = (double)ist->pts;
1942             if (!file_table[ist->file_index].eof_reached){
1943                 if(ipts < ipts_min) {
1944                     ipts_min = ipts;
1945                     if(input_sync ) file_index = ist->file_index;
1946                 }
1947                 if(opts < opts_min) {
1948                     opts_min = opts;
1949                     if(!input_sync) file_index = ist->file_index;
1950                 }
1951             }
1952             if(ost->frame_number >= max_frames[ost->st->codec->codec_type]){
1953                 file_index= -1;
1954                 break;
1955             }
1956         }
1957         /* if none, if is finished */
1958         if (file_index < 0) {
1959             break;
1960         }
1961
1962         /* finish if recording time exhausted */
1963         if (recording_time > 0 && opts_min >= (recording_time / 1000000.0))
1964             break;
1965
1966         /* finish if limit size exhausted */
1967         if (limit_filesize != 0 && (limit_filesize * 1024) < url_ftell(&output_files[0]->pb))
1968             break;
1969
1970         /* read a frame from it and output it in the fifo */
1971         is = input_files[file_index];
1972         if (av_read_frame(is, &pkt) < 0) {
1973             file_table[file_index].eof_reached = 1;
1974             if (opt_shortest) break; else continue; //
1975         }
1976
1977         if (!pkt.size) {
1978             stream_no_data = is;
1979         } else {
1980             stream_no_data = 0;
1981         }
1982         if (do_pkt_dump) {
1983             av_pkt_dump(stdout, &pkt, do_hex_dump);
1984         }
1985         /* the following test is needed in case new streams appear
1986            dynamically in stream : we ignore them */
1987         if (pkt.stream_index >= file_table[file_index].nb_streams)
1988             goto discard_packet;
1989         ist_index = file_table[file_index].ist_index + pkt.stream_index;
1990         ist = ist_table[ist_index];
1991         if (ist->discard)
1992             goto discard_packet;
1993
1994 //        fprintf(stderr, "next:%lld dts:%lld off:%lld %d\n", ist->next_pts, pkt.dts, input_files_ts_offset[ist->file_index], ist->st->codec->codec_type);
1995         if (pkt.dts != AV_NOPTS_VALUE && ist->next_pts != AV_NOPTS_VALUE) {
1996             int64_t delta= av_rescale_q(pkt.dts, ist->st->time_base, AV_TIME_BASE_Q) - ist->next_pts;
1997             if(ABS(delta) > 1LL*dts_delta_threshold*AV_TIME_BASE && !copy_ts){
1998                 input_files_ts_offset[ist->file_index]-= delta;
1999                 if (verbose > 2)
2000                     fprintf(stderr, "timestamp discontinuity %"PRId64", new offset= %"PRId64"\n", delta, input_files_ts_offset[ist->file_index]);
2001                 for(i=0; i<file_table[file_index].nb_streams; i++){
2002                     int index= file_table[file_index].ist_index + i;
2003                     ist_table[index]->next_pts += delta;
2004                     ist_table[index]->is_start=1;
2005                 }
2006             }
2007         }
2008
2009         //fprintf(stderr,"read #%d.%d size=%d\n", ist->file_index, ist->index, pkt.size);
2010         if (output_packet(ist, ist_index, ost_table, nb_ostreams, &pkt) < 0) {
2011
2012             if (verbose >= 0)
2013                 fprintf(stderr, "Error while decoding stream #%d.%d\n",
2014                         ist->file_index, ist->index);
2015
2016             av_free_packet(&pkt);
2017             goto redo;
2018         }
2019
2020     discard_packet:
2021         av_free_packet(&pkt);
2022
2023         /* dump report by using the output first video and audio streams */
2024         print_report(output_files, ost_table, nb_ostreams, 0);
2025     }
2026
2027     /* at the end of stream, we must flush the decoder buffers */
2028     for(i=0;i<nb_istreams;i++) {
2029         ist = ist_table[i];
2030         if (ist->decoding_needed) {
2031             output_packet(ist, i, ost_table, nb_ostreams, NULL);
2032         }
2033     }
2034
2035     term_exit();
2036
2037     /* write the trailer if needed and close file */
2038     for(i=0;i<nb_output_files;i++) {
2039         os = output_files[i];
2040         av_write_trailer(os);
2041     }
2042
2043     /* dump report by using the first video and audio streams */
2044     print_report(output_files, ost_table, nb_ostreams, 1);
2045
2046     /* close each encoder */
2047     for(i=0;i<nb_ostreams;i++) {
2048         ost = ost_table[i];
2049         if (ost->encoding_needed) {
2050             av_freep(&ost->st->codec->stats_in);
2051             avcodec_close(ost->st->codec);
2052         }
2053     }
2054
2055     /* close each decoder */
2056     for(i=0;i<nb_istreams;i++) {
2057         ist = ist_table[i];
2058         if (ist->decoding_needed) {
2059             avcodec_close(ist->st->codec);
2060         }
2061     }
2062
2063     /* finished ! */
2064
2065     ret = 0;
2066  fail1:
2067     av_freep(&bit_buffer);
2068     av_free(file_table);
2069
2070     if (ist_table) {
2071         for(i=0;i<nb_istreams;i++) {
2072             ist = ist_table[i];
2073             av_free(ist);
2074         }
2075         av_free(ist_table);
2076     }
2077     if (ost_table) {
2078         for(i=0;i<nb_ostreams;i++) {
2079             ost = ost_table[i];
2080             if (ost) {
2081                 if (ost->logfile) {
2082                     fclose(ost->logfile);
2083                     ost->logfile = NULL;
2084                 }
2085                 fifo_free(&ost->fifo); /* works even if fifo is not
2086                                           initialized but set to zero */
2087                 av_free(ost->pict_tmp.data[0]);
2088                 if (ost->video_resample)
2089                     img_resample_close(ost->img_resample_ctx);
2090                 if (ost->audio_resample)
2091                     audio_resample_close(ost->resample);
2092                 av_free(ost);
2093             }
2094         }
2095         av_free(ost_table);
2096     }
2097     return ret;
2098  fail:
2099     ret = -ENOMEM;
2100     goto fail1;
2101 }
2102
2103 #if 0
2104 int file_read(const char *filename)
2105 {
2106     URLContext *h;
2107     unsigned char buffer[1024];
2108     int len, i;
2109
2110     if (url_open(&h, filename, O_RDONLY) < 0) {
2111         printf("could not open '%s'\n", filename);
2112         return -1;
2113     }
2114     for(;;) {
2115         len = url_read(h, buffer, sizeof(buffer));
2116         if (len <= 0)
2117             break;
2118         for(i=0;i<len;i++) putchar(buffer[i]);
2119     }
2120     url_close(h);
2121     return 0;
2122 }
2123 #endif
2124
2125 static void opt_image_format(const char *arg)
2126 {
2127     AVImageFormat *f;
2128
2129     for(f = first_image_format; f != NULL; f = f->next) {
2130         if (!strcmp(arg, f->name))
2131             break;
2132     }
2133     if (!f) {
2134         fprintf(stderr, "Unknown image format: '%s'\n", arg);
2135         exit(1);
2136     }
2137     image_format = f;
2138 }
2139
2140 static void opt_format(const char *arg)
2141 {
2142     /* compatibility stuff for pgmyuv */
2143     if (!strcmp(arg, "pgmyuv")) {
2144         pgmyuv_compatibility_hack=1;
2145 //        opt_image_format(arg);
2146         arg = "image2";
2147     }
2148
2149     file_iformat = av_find_input_format(arg);
2150     file_oformat = guess_format(arg, NULL, NULL);
2151     if (!file_iformat && !file_oformat) {
2152         fprintf(stderr, "Unknown input or output format: %s\n", arg);
2153         exit(1);
2154     }
2155 }
2156
2157 static void opt_video_bitrate(const char *arg)
2158 {
2159     video_bit_rate = atoi(arg) * 1000;
2160 }
2161
2162 static void opt_video_bitrate_tolerance(const char *arg)
2163 {
2164     video_bit_rate_tolerance = atoi(arg) * 1000;
2165 }
2166
2167 static void opt_video_bitrate_max(const char *arg)
2168 {
2169     video_rc_max_rate = atoi(arg) * 1000;
2170 }
2171
2172 static void opt_video_bitrate_min(const char *arg)
2173 {
2174     video_rc_min_rate = atoi(arg) * 1000;
2175 }
2176
2177 static void opt_video_buffer_size(const char *arg)
2178 {
2179     video_rc_buffer_size = atoi(arg) * 8*1024;
2180 }
2181
2182 static void opt_video_rc_eq(char *arg)
2183 {
2184     video_rc_eq = arg;
2185 }
2186
2187 static void opt_video_rc_override_string(char *arg)
2188 {
2189     video_rc_override_string = arg;
2190 }
2191
2192
2193 static void opt_workaround_bugs(const char *arg)
2194 {
2195     workaround_bugs = atoi(arg);
2196 }
2197
2198 static void opt_me_threshold(const char *arg)
2199 {
2200     me_threshold = atoi(arg);
2201 }
2202
2203 static void opt_mb_threshold(const char *arg)
2204 {
2205     mb_threshold = atoi(arg);
2206 }
2207
2208 static void opt_verbose(const char *arg)
2209 {
2210     verbose = atoi(arg);
2211     av_log_set_level(atoi(arg));
2212 }
2213
2214 static void opt_frame_rate(const char *arg)
2215 {
2216     if (parse_frame_rate(&frame_rate, &frame_rate_base, arg) < 0) {
2217         fprintf(stderr, "Incorrect frame rate\n");
2218         exit(1);
2219     }
2220 }
2221
2222 static void opt_frame_crop_top(const char *arg)
2223 {
2224     frame_topBand = atoi(arg);
2225     if (frame_topBand < 0) {
2226         fprintf(stderr, "Incorrect top crop size\n");
2227         exit(1);
2228     }
2229     if ((frame_topBand % 2) != 0) {
2230         fprintf(stderr, "Top crop size must be a multiple of 2\n");
2231         exit(1);
2232     }
2233     if ((frame_topBand) >= frame_height){
2234         fprintf(stderr, "Vertical crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");
2235         exit(1);
2236     }
2237     frame_height -= frame_topBand;
2238 }
2239
2240 static void opt_frame_crop_bottom(const char *arg)
2241 {
2242     frame_bottomBand = atoi(arg);
2243     if (frame_bottomBand < 0) {
2244         fprintf(stderr, "Incorrect bottom crop size\n");
2245         exit(1);
2246     }
2247     if ((frame_bottomBand % 2) != 0) {
2248         fprintf(stderr, "Bottom crop size must be a multiple of 2\n");
2249         exit(1);
2250     }
2251     if ((frame_bottomBand) >= frame_height){
2252         fprintf(stderr, "Vertical crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");
2253         exit(1);
2254     }
2255     frame_height -= frame_bottomBand;
2256 }
2257
2258 static void opt_frame_crop_left(const char *arg)
2259 {
2260     frame_leftBand = atoi(arg);
2261     if (frame_leftBand < 0) {
2262         fprintf(stderr, "Incorrect left crop size\n");
2263         exit(1);
2264     }
2265     if ((frame_leftBand % 2) != 0) {
2266         fprintf(stderr, "Left crop size must be a multiple of 2\n");
2267         exit(1);
2268     }
2269     if ((frame_leftBand) >= frame_width){
2270         fprintf(stderr, "Horizontal crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");
2271         exit(1);
2272     }
2273     frame_width -= frame_leftBand;
2274 }
2275
2276 static void opt_frame_crop_right(const char *arg)
2277 {
2278     frame_rightBand = atoi(arg);
2279     if (frame_rightBand < 0) {
2280         fprintf(stderr, "Incorrect right crop size\n");
2281         exit(1);
2282     }
2283     if ((frame_rightBand % 2) != 0) {
2284         fprintf(stderr, "Right crop size must be a multiple of 2\n");
2285         exit(1);
2286     }
2287     if ((frame_rightBand) >= frame_width){
2288         fprintf(stderr, "Horizontal crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");
2289         exit(1);
2290     }
2291     frame_width -= frame_rightBand;
2292 }
2293
2294 static void opt_frame_size(const char *arg)
2295 {
2296     if (parse_image_size(&frame_width, &frame_height, arg) < 0) {
2297         fprintf(stderr, "Incorrect frame size\n");
2298         exit(1);
2299     }
2300     if ((frame_width % 2) != 0 || (frame_height % 2) != 0) {
2301         fprintf(stderr, "Frame size must be a multiple of 2\n");
2302         exit(1);
2303     }
2304 }
2305
2306
2307 #define SCALEBITS 10
2308 #define ONE_HALF  (1 << (SCALEBITS - 1))
2309 #define FIX(x)    ((int) ((x) * (1<<SCALEBITS) + 0.5))
2310
2311 #define RGB_TO_Y(r, g, b) \
2312 ((FIX(0.29900) * (r) + FIX(0.58700) * (g) + \
2313   FIX(0.11400) * (b) + ONE_HALF) >> SCALEBITS)
2314
2315 #define RGB_TO_U(r1, g1, b1, shift)\
2316 (((- FIX(0.16874) * r1 - FIX(0.33126) * g1 +         \
2317      FIX(0.50000) * b1 + (ONE_HALF << shift) - 1) >> (SCALEBITS + shift)) + 128)
2318
2319 #define RGB_TO_V(r1, g1, b1, shift)\
2320 (((FIX(0.50000) * r1 - FIX(0.41869) * g1 -           \
2321    FIX(0.08131) * b1 + (ONE_HALF << shift) - 1) >> (SCALEBITS + shift)) + 128)
2322
2323 static void opt_pad_color(const char *arg) {
2324     /* Input is expected to be six hex digits similar to
2325        how colors are expressed in html tags (but without the #) */
2326     int rgb = strtol(arg, NULL, 16);
2327     int r,g,b;
2328
2329     r = (rgb >> 16);
2330     g = ((rgb >> 8) & 255);
2331     b = (rgb & 255);
2332
2333     padcolor[0] = RGB_TO_Y(r,g,b);
2334     padcolor[1] = RGB_TO_U(r,g,b,0);
2335     padcolor[2] = RGB_TO_V(r,g,b,0);
2336 }
2337
2338 static void opt_frame_pad_top(const char *arg)
2339 {
2340     frame_padtop = atoi(arg);
2341     if (frame_padtop < 0) {
2342         fprintf(stderr, "Incorrect top pad size\n");
2343         exit(1);
2344     }
2345     if ((frame_padtop % 2) != 0) {
2346         fprintf(stderr, "Top pad size must be a multiple of 2\n");
2347         exit(1);
2348     }
2349 }
2350
2351 static void opt_frame_pad_bottom(const char *arg)
2352 {
2353     frame_padbottom = atoi(arg);
2354     if (frame_padbottom < 0) {
2355         fprintf(stderr, "Incorrect bottom pad size\n");
2356         exit(1);
2357     }
2358     if ((frame_padbottom % 2) != 0) {
2359         fprintf(stderr, "Bottom pad size must be a multiple of 2\n");
2360         exit(1);
2361     }
2362 }
2363
2364
2365 static void opt_frame_pad_left(const char *arg)
2366 {
2367     frame_padleft = atoi(arg);
2368     if (frame_padleft < 0) {
2369         fprintf(stderr, "Incorrect left pad size\n");
2370         exit(1);
2371     }
2372     if ((frame_padleft % 2) != 0) {
2373         fprintf(stderr, "Left pad size must be a multiple of 2\n");
2374         exit(1);
2375     }
2376 }
2377
2378
2379 static void opt_frame_pad_right(const char *arg)
2380 {
2381     frame_padright = atoi(arg);
2382     if (frame_padright < 0) {
2383         fprintf(stderr, "Incorrect right pad size\n");
2384         exit(1);
2385     }
2386     if ((frame_padright % 2) != 0) {
2387         fprintf(stderr, "Right pad size must be a multiple of 2\n");
2388         exit(1);
2389     }
2390 }
2391
2392
2393 static void opt_frame_pix_fmt(const char *arg)
2394 {
2395     frame_pix_fmt = avcodec_get_pix_fmt(arg);
2396 }
2397
2398 static void opt_frame_aspect_ratio(const char *arg)
2399 {
2400     int x = 0, y = 0;
2401     double ar = 0;
2402     const char *p;
2403
2404     p = strchr(arg, ':');
2405     if (p) {
2406         x = strtol(arg, (char **)&arg, 10);
2407         if (arg == p)
2408             y = strtol(arg+1, (char **)&arg, 10);
2409         if (x > 0 && y > 0)
2410             ar = (double)x / (double)y;
2411     } else
2412         ar = strtod(arg, (char **)&arg);
2413
2414     if (!ar) {
2415         fprintf(stderr, "Incorrect aspect ratio specification.\n");
2416         exit(1);
2417     }
2418     frame_aspect_ratio = ar;
2419 }
2420
2421 static void opt_gop_size(const char *arg)
2422 {
2423     gop_size = atoi(arg);
2424 }
2425
2426 static void opt_b_frames(const char *arg)
2427 {
2428     b_frames = atoi(arg);
2429     if (b_frames > FF_MAX_B_FRAMES) {
2430         fprintf(stderr, "\nCannot have more than %d B frames, increase FF_MAX_B_FRAMES.\n", FF_MAX_B_FRAMES);
2431         exit(1);
2432     } else if (b_frames < 1) {
2433         fprintf(stderr, "\nNumber of B frames must be higher than 0\n");
2434         exit(1);
2435     }
2436 }
2437
2438 static void opt_pre_me(const char *arg)
2439 {
2440     pre_me = atoi(arg);
2441 }
2442
2443 static void opt_qscale(const char *arg)
2444 {
2445     video_qscale = atof(arg);
2446     if (video_qscale <= 0 ||
2447         video_qscale > 255) {
2448         fprintf(stderr, "qscale must be > 0.0 and <= 255\n");
2449         exit(1);
2450     }
2451 }
2452
2453 static void opt_qsquish(const char *arg)
2454 {
2455     video_qsquish = atof(arg);
2456     if (video_qsquish < 0.0 ||
2457         video_qsquish > 99.0) {
2458         fprintf(stderr, "qsquish must be >= 0.0 and <= 99.0\n");
2459         exit(1);
2460     }
2461 }
2462
2463 static void opt_lmax(const char *arg)
2464 {
2465     video_lmax = atof(arg)*FF_QP2LAMBDA;
2466 }
2467
2468 static void opt_lmin(const char *arg)
2469 {
2470     video_lmin = atof(arg)*FF_QP2LAMBDA;
2471 }
2472
2473 static void opt_qmin(const char *arg)
2474 {
2475     video_qmin = atoi(arg);
2476     if (video_qmin < 1 ||
2477         video_qmin > 51) {
2478         fprintf(stderr, "qmin must be >= 1 and <= 51\n");
2479         exit(1);
2480     }
2481 }
2482
2483 static void opt_qmax(const char *arg)
2484 {
2485     video_qmax = atoi(arg);
2486     if (video_qmax < 1 ||
2487         video_qmax > 51) {
2488         fprintf(stderr, "qmax must be >= 1 and <= 51\n");
2489         exit(1);
2490     }
2491 }
2492
2493 static void opt_mb_lmin(const char *arg)
2494 {
2495     video_mb_lmin = atof(arg)*FF_QP2LAMBDA;
2496     if (video_mb_lmin < 1 ||
2497         video_mb_lmin > FF_LAMBDA_MAX) {
2498         fprintf(stderr, "mblmin must be >= 1 and <= %d\n", FF_LAMBDA_MAX / FF_QP2LAMBDA);
2499         exit(1);
2500     }
2501 }
2502
2503 static void opt_mb_lmax(const char *arg)
2504 {
2505     video_mb_lmax = atof(arg)*FF_QP2LAMBDA;
2506     if (video_mb_lmax < 1 ||
2507         video_mb_lmax > FF_LAMBDA_MAX) {
2508         fprintf(stderr, "mblmax must be >= 1 and <= %d\n", FF_LAMBDA_MAX / FF_QP2LAMBDA);
2509         exit(1);
2510     }
2511 }
2512
2513 static void opt_qdiff(const char *arg)
2514 {
2515     video_qdiff = atoi(arg);
2516     if (video_qdiff < 0 ||
2517         video_qdiff > 31) {
2518         fprintf(stderr, "qdiff must be >= 1 and <= 31\n");
2519         exit(1);
2520     }
2521 }
2522
2523 static void opt_qblur(const char *arg)
2524 {
2525     video_qblur = atof(arg);
2526 }
2527
2528 static void opt_qcomp(const char *arg)
2529 {
2530     video_qcomp = atof(arg);
2531 }
2532
2533 static void opt_rc_initial_cplx(const char *arg)
2534 {
2535     video_rc_initial_cplx = atof(arg);
2536 }
2537 static void opt_b_qfactor(const char *arg)
2538 {
2539     video_b_qfactor = atof(arg);
2540 }
2541 static void opt_i_qfactor(const char *arg)
2542 {
2543     video_i_qfactor = atof(arg);
2544 }
2545 static void opt_b_qoffset(const char *arg)
2546 {
2547     video_b_qoffset = atof(arg);
2548 }
2549 static void opt_i_qoffset(const char *arg)
2550 {
2551     video_i_qoffset = atof(arg);
2552 }
2553
2554 static void opt_ibias(const char *arg)
2555 {
2556     video_intra_quant_bias = atoi(arg);
2557 }
2558 static void opt_pbias(const char *arg)
2559 {
2560     video_inter_quant_bias = atoi(arg);
2561 }
2562
2563 static void opt_packet_size(const char *arg)
2564 {
2565     packet_size= atoi(arg);
2566 }
2567
2568 static void opt_error_rate(const char *arg)
2569 {
2570     error_rate= atoi(arg);
2571 }
2572
2573 static void opt_strict(const char *arg)
2574 {
2575     strict= atoi(arg);
2576 }
2577
2578 static void opt_top_field_first(const char *arg)
2579 {
2580     top_field_first= atoi(arg);
2581 }
2582
2583 static void opt_sc_threshold(const char *arg)
2584 {
2585     sc_threshold= atoi(arg);
2586 }
2587
2588 static void opt_me_range(const char *arg)
2589 {
2590     me_range = atoi(arg);
2591 }
2592
2593 static void opt_thread_count(const char *arg)
2594 {
2595     thread_count= atoi(arg);
2596 #if !defined(HAVE_THREADS)
2597     if (verbose >= 0)
2598         fprintf(stderr, "Warning: not compiled with thread support, using thread emulation\n");
2599 #endif
2600 }
2601
2602 static void opt_audio_bitrate(const char *arg)
2603 {
2604     audio_bit_rate = atoi(arg) * 1000;
2605 }
2606
2607 static void opt_audio_rate(const char *arg)
2608 {
2609     audio_sample_rate = atoi(arg);
2610 }
2611
2612 static void opt_audio_channels(const char *arg)
2613 {
2614     audio_channels = atoi(arg);
2615 }
2616
2617 static void opt_video_device(const char *arg)
2618 {
2619     video_device = av_strdup(arg);
2620 }
2621
2622 static void opt_grab_device(const char *arg)
2623 {
2624     grab_device = av_strdup(arg);
2625 }
2626
2627 static void opt_video_channel(const char *arg)
2628 {
2629     video_channel = strtol(arg, NULL, 0);
2630 }
2631
2632 static void opt_video_standard(const char *arg)
2633 {
2634     video_standard = av_strdup(arg);
2635 }
2636
2637 static void opt_audio_device(const char *arg)
2638 {
2639     audio_device = av_strdup(arg);
2640 }
2641
2642 static void opt_codec(int *pstream_copy, int *pcodec_id,
2643                       int codec_type, const char *arg)
2644 {
2645     AVCodec *p;
2646
2647     if (!strcmp(arg, "copy")) {
2648         *pstream_copy = 1;
2649     } else {
2650         p = first_avcodec;
2651         while (p) {
2652             if (!strcmp(p->name, arg) && p->type == codec_type)
2653                 break;
2654             p = p->next;
2655         }
2656         if (p == NULL) {
2657             fprintf(stderr, "Unknown codec '%s'\n", arg);
2658             exit(1);
2659         } else {
2660             *pcodec_id = p->id;
2661         }
2662     }
2663 }
2664
2665 static void opt_audio_codec(const char *arg)
2666 {
2667     opt_codec(&audio_stream_copy, &audio_codec_id, CODEC_TYPE_AUDIO, arg);
2668 }
2669
2670 static void opt_audio_tag(const char *arg)
2671 {
2672     char *tail;
2673     audio_codec_tag= strtol(arg, &tail, 0);
2674
2675     if(!tail || *tail)
2676         audio_codec_tag= arg[0] + (arg[1]<<8) + (arg[2]<<16) + (arg[3]<<24);
2677 }
2678
2679 static void opt_video_tag(const char *arg)
2680 {
2681     char *tail;
2682     video_codec_tag= strtol(arg, &tail, 0);
2683
2684     if(!tail || *tail)
2685         video_codec_tag= arg[0] + (arg[1]<<8) + (arg[2]<<16) + (arg[3]<<24);
2686 }
2687
2688 static void add_frame_hooker(const char *arg)
2689 {
2690     int argc = 0;
2691     char *argv[64];
2692     int i;
2693     char *args = av_strdup(arg);
2694
2695     using_vhook = 1;
2696
2697     argv[0] = strtok(args, " ");
2698     while (argc < 62 && (argv[++argc] = strtok(NULL, " "))) {
2699     }
2700
2701     i = frame_hook_add(argc, argv);
2702
2703     if (i != 0) {
2704         fprintf(stderr, "Failed to add video hook function: %s\n", arg);
2705         exit(1);
2706     }
2707 }
2708
2709 const char *motion_str[] = {
2710     "zero",
2711     "full",
2712     "log",
2713     "phods",
2714     "epzs",
2715     "x1",
2716     "hex",
2717     "umh",
2718     "iter",
2719     NULL,
2720 };
2721
2722 static void opt_motion_estimation(const char *arg)
2723 {
2724     const char **p;
2725     p = motion_str;
2726     for(;;) {
2727         if (!*p) {
2728             fprintf(stderr, "Unknown motion estimation method '%s'\n", arg);
2729             exit(1);
2730         }
2731         if (!strcmp(*p, arg))
2732             break;
2733         p++;
2734     }
2735     me_method = (p - motion_str) + 1;
2736 }
2737
2738 static void opt_video_codec(const char *arg)
2739 {
2740     opt_codec(&video_stream_copy, &video_codec_id, CODEC_TYPE_VIDEO, arg);
2741 }
2742
2743 static void opt_subtitle_codec(const char *arg)
2744 {
2745     opt_codec(&subtitle_stream_copy, &subtitle_codec_id, CODEC_TYPE_SUBTITLE, arg);
2746 }
2747
2748 static void opt_map(const char *arg)
2749 {
2750     AVStreamMap *m;
2751     const char *p;
2752
2753     p = arg;
2754     m = &stream_maps[nb_stream_maps++];
2755
2756     m->file_index = strtol(arg, (char **)&p, 0);
2757     if (*p)
2758         p++;
2759
2760     m->stream_index = strtol(p, (char **)&p, 0);
2761     if (*p) {
2762         p++;
2763         m->sync_file_index = strtol(p, (char **)&p, 0);
2764         if (*p)
2765             p++;
2766         m->sync_stream_index = strtol(p, (char **)&p, 0);
2767     } else {
2768         m->sync_file_index = m->file_index;
2769         m->sync_stream_index = m->stream_index;
2770     }
2771 }
2772
2773 static void opt_map_meta_data(const char *arg)
2774 {
2775     AVMetaDataMap *m;
2776     const char *p;
2777
2778     p = arg;
2779     m = &meta_data_maps[nb_meta_data_maps++];
2780
2781     m->out_file = strtol(arg, (char **)&p, 0);
2782     if (*p)
2783         p++;
2784
2785     m->in_file = strtol(p, (char **)&p, 0);
2786 }
2787
2788 static void opt_recording_time(const char *arg)
2789 {
2790     recording_time = parse_date(arg, 1);
2791 }
2792
2793 static void opt_start_time(const char *arg)
2794 {
2795     start_time = parse_date(arg, 1);
2796 }
2797
2798 static void opt_rec_timestamp(const char *arg)
2799 {
2800     rec_timestamp = parse_date(arg, 0) / 1000000;
2801 }
2802
2803 static void opt_input_ts_offset(const char *arg)
2804 {
2805     input_ts_offset = parse_date(arg, 1);
2806 }
2807
2808 static void opt_input_file(const char *filename)
2809 {
2810     AVFormatContext *ic;
2811     AVFormatParameters params, *ap = &params;
2812     int err, i, ret, rfps, rfps_base;
2813     int64_t timestamp;
2814
2815     if (!strcmp(filename, "-"))
2816         filename = "pipe:";
2817
2818     using_stdin |= !strncmp(filename, "pipe:", 5) ||
2819                    !strcmp( filename, "/dev/stdin" );
2820
2821     /* get default parameters from command line */
2822     memset(ap, 0, sizeof(*ap));
2823     ap->sample_rate = audio_sample_rate;
2824     ap->channels = audio_channels;
2825     ap->time_base.den = frame_rate;
2826     ap->time_base.num = frame_rate_base;
2827     ap->width = frame_width + frame_padleft + frame_padright;
2828     ap->height = frame_height + frame_padtop + frame_padbottom;
2829     ap->image_format = image_format;
2830     ap->pix_fmt = frame_pix_fmt;
2831     ap->device  = grab_device;
2832     ap->channel = video_channel;
2833     ap->standard = video_standard;
2834     ap->video_codec_id = video_codec_id;
2835     ap->audio_codec_id = audio_codec_id;
2836     if(pgmyuv_compatibility_hack)
2837         ap->video_codec_id= CODEC_ID_PGMYUV;
2838
2839     /* open the input file with generic libav function */
2840     err = av_open_input_file(&ic, filename, file_iformat, 0, ap);
2841     if (err < 0) {
2842         print_error(filename, err);
2843         exit(1);
2844     }
2845
2846     if(genpts)
2847         ic->flags|= AVFMT_FLAG_GENPTS;
2848
2849     /* If not enough info to get the stream parameters, we decode the
2850        first frames to get it. (used in mpeg case for example) */
2851     ret = av_find_stream_info(ic);
2852     if (ret < 0 && verbose >= 0) {
2853         fprintf(stderr, "%s: could not find codec parameters\n", filename);
2854         exit(1);
2855     }
2856
2857     timestamp = start_time;
2858     /* add the stream start time */
2859     if (ic->start_time != AV_NOPTS_VALUE)
2860         timestamp += ic->start_time;
2861
2862     /* if seeking requested, we execute it */
2863     if (start_time != 0) {
2864         ret = av_seek_frame(ic, -1, timestamp, AVSEEK_FLAG_BACKWARD);
2865         if (ret < 0) {
2866             fprintf(stderr, "%s: could not seek to position %0.3f\n",
2867                     filename, (double)timestamp / AV_TIME_BASE);
2868         }
2869         /* reset seek info */
2870         start_time = 0;
2871     }
2872
2873     /* update the current parameters so that they match the one of the input stream */
2874     for(i=0;i<ic->nb_streams;i++) {
2875         int j;
2876         AVCodecContext *enc = ic->streams[i]->codec;
2877 #if defined(HAVE_THREADS)
2878         if(thread_count>1)
2879             avcodec_thread_init(enc, thread_count);
2880 #endif
2881         enc->thread_count= thread_count;
2882         switch(enc->codec_type) {
2883         case CODEC_TYPE_AUDIO:
2884             for(j=0; j<opt_name_count; j++){
2885                 AVOption *opt;
2886                 double d= av_get_double(avctx_opts, opt_names[j], &opt);
2887                 if(d==d && (opt->flags&AV_OPT_FLAG_AUDIO_PARAM) && (opt->flags&AV_OPT_FLAG_DECODING_PARAM))
2888                     av_set_double(enc, opt_names[j], d);
2889             }
2890             //fprintf(stderr, "\nInput Audio channels: %d", enc->channels);
2891             audio_channels = enc->channels;
2892             audio_sample_rate = enc->sample_rate;
2893             if(audio_disable)
2894                 ic->streams[i]->discard= AVDISCARD_ALL;
2895             break;
2896         case CODEC_TYPE_VIDEO:
2897             for(j=0; j<opt_name_count; j++){
2898                 AVOption *opt;
2899                 double d= av_get_double(avctx_opts, opt_names[j], &opt);
2900                 if(d==d && (opt->flags&AV_OPT_FLAG_VIDEO_PARAM) && (opt->flags&AV_OPT_FLAG_DECODING_PARAM))
2901                     av_set_double(enc, opt_names[j], d);
2902             }
2903             frame_height = enc->height;
2904             frame_width = enc->width;
2905             frame_aspect_ratio = av_q2d(enc->sample_aspect_ratio) * enc->width / enc->height;
2906             frame_pix_fmt = enc->pix_fmt;
2907             rfps      = ic->streams[i]->r_frame_rate.num;
2908             rfps_base = ic->streams[i]->r_frame_rate.den;
2909             enc->workaround_bugs = workaround_bugs;
2910             if(enc->lowres) enc->flags |= CODEC_FLAG_EMU_EDGE;
2911             if(me_threshold)
2912                 enc->debug |= FF_DEBUG_MV;
2913
2914             if (enc->time_base.den != rfps || enc->time_base.num != rfps_base) {
2915
2916                 if (verbose >= 0)
2917                     fprintf(stderr,"\nSeems that stream %d comes from film source: %2.2f (%d/%d) -> %2.2f (%d/%d)\n",
2918                             i, (float)enc->time_base.den / enc->time_base.num, enc->time_base.den, enc->time_base.num,
2919
2920                     (float)rfps / rfps_base, rfps, rfps_base);
2921             }
2922             /* update the current frame rate to match the stream frame rate */
2923             frame_rate      = rfps;
2924             frame_rate_base = rfps_base;
2925
2926             enc->rate_emu = rate_emu;
2927             if(video_disable)
2928                 ic->streams[i]->discard= AVDISCARD_ALL;
2929             else if(video_discard)
2930                 ic->streams[i]->discard= video_discard;
2931             break;
2932         case CODEC_TYPE_DATA:
2933             break;
2934         case CODEC_TYPE_SUBTITLE:
2935             break;
2936         case CODEC_TYPE_UNKNOWN:
2937             break;
2938         default:
2939             av_abort();
2940         }
2941     }
2942
2943     input_files[nb_input_files] = ic;
2944     input_files_ts_offset[nb_input_files] = input_ts_offset - (copy_ts ? 0 : timestamp);
2945     /* dump the file content */
2946     if (verbose >= 0)
2947         dump_format(ic, nb_input_files, filename, 0);
2948
2949     nb_input_files++;
2950     file_iformat = NULL;
2951     file_oformat = NULL;
2952     image_format = NULL;
2953
2954     grab_device = NULL;
2955     video_channel = 0;
2956
2957     rate_emu = 0;
2958 }
2959
2960 static void opt_grab(const char *arg)
2961 {
2962     file_iformat = av_find_input_format(arg);
2963     opt_input_file("");
2964 }
2965
2966 static void check_audio_video_inputs(int *has_video_ptr, int *has_audio_ptr)
2967 {
2968     int has_video, has_audio, i, j;
2969     AVFormatContext *ic;
2970
2971     has_video = 0;
2972     has_audio = 0;
2973     for(j=0;j<nb_input_files;j++) {
2974         ic = input_files[j];
2975         for(i=0;i<ic->nb_streams;i++) {
2976             AVCodecContext *enc = ic->streams[i]->codec;
2977             switch(enc->codec_type) {
2978             case CODEC_TYPE_AUDIO:
2979                 has_audio = 1;
2980                 break;
2981             case CODEC_TYPE_VIDEO:
2982                 has_video = 1;
2983                 break;
2984             case CODEC_TYPE_DATA:
2985             case CODEC_TYPE_UNKNOWN:
2986             case CODEC_TYPE_SUBTITLE:
2987                 break;
2988             default:
2989                 av_abort();
2990             }
2991         }
2992     }
2993     *has_video_ptr = has_video;
2994     *has_audio_ptr = has_audio;
2995 }
2996
2997 static void new_video_stream(AVFormatContext *oc)
2998 {
2999     AVStream *st;
3000     AVCodecContext *video_enc;
3001     int codec_id;
3002
3003     st = av_new_stream(oc, oc->nb_streams);
3004     if (!st) {
3005         fprintf(stderr, "Could not alloc stream\n");
3006         exit(1);
3007     }
3008     bitstream_filters[nb_output_files][oc->nb_streams - 1]= video_bitstream_filters;
3009     video_bitstream_filters= NULL;
3010
3011 #if defined(HAVE_THREADS)
3012     if(thread_count>1)
3013         avcodec_thread_init(st->codec, thread_count);
3014 #endif
3015
3016     video_enc = st->codec;
3017
3018     if(video_codec_tag)
3019         video_enc->codec_tag= video_codec_tag;
3020
3021     if(   (video_global_header&1)
3022        || (video_global_header==0 && (oc->oformat->flags & AVFMT_GLOBALHEADER))){
3023         video_enc->flags |= CODEC_FLAG_GLOBAL_HEADER;
3024         avctx_opts->flags|= CODEC_FLAG_GLOBAL_HEADER;
3025     }
3026     if(video_global_header&2){
3027         video_enc->flags2 |= CODEC_FLAG2_LOCAL_HEADER;
3028         avctx_opts->flags2|= CODEC_FLAG2_LOCAL_HEADER;
3029     }
3030
3031     if (video_stream_copy) {
3032         st->stream_copy = 1;
3033         video_enc->codec_type = CODEC_TYPE_VIDEO;
3034     } else {
3035         char *p;
3036         int i;
3037         AVCodec *codec;
3038
3039         codec_id = av_guess_codec(oc->oformat, NULL, oc->filename, NULL, CODEC_TYPE_VIDEO);
3040         if (video_codec_id != CODEC_ID_NONE)
3041             codec_id = video_codec_id;
3042
3043         video_enc->codec_id = codec_id;
3044         codec = avcodec_find_encoder(codec_id);
3045
3046         for(i=0; i<opt_name_count; i++){
3047              AVOption *opt;
3048              double d= av_get_double(avctx_opts, opt_names[i], &opt);
3049              if(d==d && (opt->flags&AV_OPT_FLAG_VIDEO_PARAM) && (opt->flags&AV_OPT_FLAG_ENCODING_PARAM))
3050                  av_set_double(video_enc, opt_names[i], d);
3051         }
3052
3053         video_enc->bit_rate = video_bit_rate;
3054         video_enc->bit_rate_tolerance = video_bit_rate_tolerance;
3055         video_enc->time_base.den = frame_rate;
3056         video_enc->time_base.num = frame_rate_base;
3057         if(codec && codec->supported_framerates){
3058             const AVRational *p= codec->supported_framerates;
3059             AVRational req= (AVRational){frame_rate, frame_rate_base};
3060             const AVRational *best=NULL;
3061             AVRational best_error= (AVRational){INT_MAX, 1};
3062             for(; p->den!=0; p++){
3063                 AVRational error= av_sub_q(req, *p);
3064                 if(error.num <0) error.num *= -1;
3065                 if(av_cmp_q(error, best_error) < 0){
3066                     best_error= error;
3067                     best= p;
3068                 }
3069             }
3070             video_enc->time_base.den= best->num;
3071             video_enc->time_base.num= best->den;
3072         }
3073
3074         video_enc->width = frame_width + frame_padright + frame_padleft;
3075         video_enc->height = frame_height + frame_padtop + frame_padbottom;
3076         video_enc->sample_aspect_ratio = av_d2q(frame_aspect_ratio*frame_height/frame_width, 255);
3077         video_enc->pix_fmt = frame_pix_fmt;
3078
3079         if(codec && codec->pix_fmts){
3080             const enum PixelFormat *p= codec->pix_fmts;
3081             for(; *p!=-1; p++){
3082                 if(*p == video_enc->pix_fmt)
3083                     break;
3084             }
3085             if(*p == -1)
3086                 video_enc->pix_fmt = codec->pix_fmts[0];
3087         }
3088
3089         if (!intra_only)
3090             video_enc->gop_size = gop_size;
3091         else
3092             video_enc->gop_size = 0;
3093         if (video_qscale || same_quality) {
3094             video_enc->flags |= CODEC_FLAG_QSCALE;
3095             video_enc->global_quality=
3096                 st->quality = FF_QP2LAMBDA * video_qscale;
3097         }
3098
3099         if(intra_matrix)
3100             video_enc->intra_matrix = intra_matrix;
3101         if(inter_matrix)
3102             video_enc->inter_matrix = inter_matrix;
3103
3104         video_enc->pre_me = pre_me;
3105
3106         if (b_frames) {
3107             video_enc->max_b_frames = b_frames;
3108             video_enc->b_quant_factor = 2.0;
3109         }
3110         video_enc->qmin = video_qmin;
3111         video_enc->qmax = video_qmax;
3112         video_enc->lmin = video_lmin;
3113         video_enc->lmax = video_lmax;
3114         video_enc->rc_qsquish = video_qsquish;
3115         video_enc->mb_lmin = video_mb_lmin;
3116         video_enc->mb_lmax = video_mb_lmax;
3117         video_enc->max_qdiff = video_qdiff;
3118         video_enc->qblur = video_qblur;
3119         video_enc->qcompress = video_qcomp;
3120         video_enc->rc_eq = video_rc_eq;
3121         video_enc->workaround_bugs = workaround_bugs;
3122         video_enc->thread_count = thread_count;
3123         p= video_rc_override_string;
3124         for(i=0; p; i++){
3125             int start, end, q;
3126             int e=sscanf(p, "%d,%d,%d", &start, &end, &q);
3127             if(e!=3){
3128                 fprintf(stderr, "error parsing rc_override\n");
3129                 exit(1);
3130             }
3131             video_enc->rc_override=
3132                 av_realloc(video_enc->rc_override,
3133                            sizeof(RcOverride)*(i+1));
3134             video_enc->rc_override[i].start_frame= start;
3135             video_enc->rc_override[i].end_frame  = end;
3136             if(q>0){
3137                 video_enc->rc_override[i].qscale= q;
3138                 video_enc->rc_override[i].quality_factor= 1.0;
3139             }
3140             else{
3141                 video_enc->rc_override[i].qscale= 0;
3142                 video_enc->rc_override[i].quality_factor= -q/100.0;
3143             }
3144             p= strchr(p, '/');
3145             if(p) p++;
3146         }
3147         video_enc->rc_override_count=i;
3148
3149         video_enc->rc_max_rate = video_rc_max_rate;
3150         video_enc->rc_min_rate = video_rc_min_rate;
3151         video_enc->rc_buffer_size = video_rc_buffer_size;
3152         video_enc->rc_initial_buffer_occupancy = video_rc_buffer_size*3/4;
3153         video_enc->rc_buffer_aggressivity= video_rc_buffer_aggressivity;
3154         video_enc->rc_initial_cplx= video_rc_initial_cplx;
3155         video_enc->i_quant_factor = video_i_qfactor;
3156         video_enc->b_quant_factor = video_b_qfactor;
3157         video_enc->i_quant_offset = video_i_qoffset;
3158         video_enc->b_quant_offset = video_b_qoffset;
3159         video_enc->intra_quant_bias = video_intra_quant_bias;
3160         video_enc->inter_quant_bias = video_inter_quant_bias;
3161         video_enc->me_threshold= me_threshold;
3162         video_enc->mb_threshold= mb_threshold;
3163         video_enc->intra_dc_precision= intra_dc_precision - 8;
3164         video_enc->strict_std_compliance = strict;
3165         video_enc->error_rate = error_rate;
3166         video_enc->scenechange_threshold= sc_threshold;
3167         video_enc->me_range = me_range;
3168         video_enc->me_penalty_compensation= me_penalty_compensation;
3169         video_enc->frame_skip_threshold= frame_skip_threshold;
3170         video_enc->frame_skip_factor= frame_skip_factor;
3171         video_enc->frame_skip_exp= frame_skip_exp;
3172
3173         if(packet_size){
3174             video_enc->rtp_mode= 1;
3175             video_enc->rtp_payload_size= packet_size;
3176         }
3177
3178         if (do_psnr)
3179             video_enc->flags|= CODEC_FLAG_PSNR;
3180
3181         video_enc->me_method = me_method;
3182
3183         /* two pass mode */
3184         if (do_pass) {
3185             if (do_pass == 1) {
3186                 video_enc->flags |= CODEC_FLAG_PASS1;
3187             } else {
3188                 video_enc->flags |= CODEC_FLAG_PASS2;
3189             }
3190         }
3191     }
3192
3193     /* reset some key parameters */
3194     video_disable = 0;
3195     video_codec_id = CODEC_ID_NONE;
3196     video_stream_copy = 0;
3197 }
3198
3199 static void new_audio_stream(AVFormatContext *oc)
3200 {
3201     AVStream *st;
3202     AVCodecContext *audio_enc;
3203     int codec_id, i;
3204
3205     st = av_new_stream(oc, oc->nb_streams);
3206     if (!st) {
3207         fprintf(stderr, "Could not alloc stream\n");
3208         exit(1);
3209     }
3210
3211     bitstream_filters[nb_output_files][oc->nb_streams - 1]= audio_bitstream_filters;
3212     audio_bitstream_filters= NULL;
3213
3214 #if defined(HAVE_THREADS)
3215     if(thread_count>1)
3216         avcodec_thread_init(st->codec, thread_count);
3217 #endif
3218
3219     audio_enc = st->codec;
3220     audio_enc->codec_type = CODEC_TYPE_AUDIO;
3221
3222     if(audio_codec_tag)
3223         audio_enc->codec_tag= audio_codec_tag;
3224
3225     if (oc->oformat->flags & AVFMT_GLOBALHEADER) {
3226         audio_enc->flags |= CODEC_FLAG_GLOBAL_HEADER;
3227         avctx_opts->flags|= CODEC_FLAG_GLOBAL_HEADER;
3228     }
3229     if (audio_stream_copy) {
3230         st->stream_copy = 1;
3231         audio_enc->channels = audio_channels;
3232     } else {
3233         codec_id = av_guess_codec(oc->oformat, NULL, oc->filename, NULL, CODEC_TYPE_AUDIO);
3234
3235         for(i=0; i<opt_name_count; i++){
3236             AVOption *opt;
3237             double d= av_get_double(avctx_opts, opt_names[i], &opt);
3238             if(d==d && (opt->flags&AV_OPT_FLAG_AUDIO_PARAM) && (opt->flags&AV_OPT_FLAG_ENCODING_PARAM))
3239                 av_set_double(audio_enc, opt_names[i], d);
3240         }
3241
3242         if (audio_codec_id != CODEC_ID_NONE)
3243             codec_id = audio_codec_id;
3244         audio_enc->codec_id = codec_id;
3245
3246         audio_enc->bit_rate = audio_bit_rate;
3247         if (audio_qscale > QSCALE_NONE) {
3248             audio_enc->flags |= CODEC_FLAG_QSCALE;
3249             audio_enc->global_quality = st->quality = FF_QP2LAMBDA * audio_qscale;
3250         }
3251         audio_enc->strict_std_compliance = strict;
3252         audio_enc->thread_count = thread_count;
3253         /* For audio codecs other than AC3 or DTS we limit */
3254         /* the number of coded channels to stereo   */
3255         if (audio_channels > 2 && codec_id != CODEC_ID_AC3
3256             && codec_id != CODEC_ID_DTS) {
3257             audio_enc->channels = 2;
3258         } else
3259             audio_enc->channels = audio_channels;
3260     }
3261     audio_enc->sample_rate = audio_sample_rate;
3262     audio_enc->time_base= (AVRational){1, audio_sample_rate};
3263     if (audio_language) {
3264         pstrcpy(st->language, sizeof(st->language), audio_language);
3265         av_free(audio_language);
3266         audio_language = NULL;
3267     }
3268
3269     /* reset some key parameters */
3270     audio_disable = 0;
3271     audio_codec_id = CODEC_ID_NONE;
3272     audio_stream_copy = 0;
3273 }
3274
3275 static void opt_new_subtitle_stream(void)
3276 {
3277     AVFormatContext *oc;
3278     AVStream *st;
3279     AVCodecContext *subtitle_enc;
3280     int i;
3281
3282     if (nb_output_files <= 0) {
3283         fprintf(stderr, "At least one output file must be specified\n");
3284         exit(1);
3285     }
3286     oc = output_files[nb_output_files - 1];
3287
3288     st = av_new_stream(oc, oc->nb_streams);
3289     if (!st) {
3290         fprintf(stderr, "Could not alloc stream\n");
3291         exit(1);
3292     }
3293
3294     subtitle_enc = st->codec;
3295     subtitle_enc->codec_type = CODEC_TYPE_SUBTITLE;
3296     if (subtitle_stream_copy) {
3297         st->stream_copy = 1;
3298     } else {
3299         for(i=0; i<opt_name_count; i++){
3300              AVOption *opt;
3301              double d= av_get_double(avctx_opts, opt_names[i], &opt);
3302              if(d==d && (opt->flags&AV_OPT_FLAG_SUBTITLE_PARAM) && (opt->flags&AV_OPT_FLAG_ENCODING_PARAM))
3303                  av_set_double(subtitle_enc, opt_names[i], d);
3304         }
3305         subtitle_enc->codec_id = subtitle_codec_id;
3306     }
3307
3308     if (subtitle_language) {
3309         pstrcpy(st->language, sizeof(st->language), subtitle_language);
3310         av_free(subtitle_language);
3311         subtitle_language = NULL;
3312     }
3313
3314     subtitle_codec_id = CODEC_ID_NONE;
3315     subtitle_stream_copy = 0;
3316 }
3317
3318 static void opt_new_audio_stream(void)
3319 {
3320     AVFormatContext *oc;
3321     if (nb_output_files <= 0) {
3322         fprintf(stderr, "At least one output file must be specified\n");
3323         exit(1);
3324     }
3325     oc = output_files[nb_output_files - 1];
3326     new_audio_stream(oc);
3327 }
3328
3329 static void opt_new_video_stream(void)
3330 {
3331     AVFormatContext *oc;
3332     if (nb_output_files <= 0) {
3333         fprintf(stderr, "At least one output file must be specified\n");
3334         exit(1);
3335     }
3336     oc = output_files[nb_output_files - 1];
3337     new_video_stream(oc);
3338 }
3339
3340 static void opt_output_file(const char *filename)
3341 {
3342     AVFormatContext *oc;
3343     int use_video, use_audio, input_has_video, input_has_audio;
3344     AVFormatParameters params, *ap = &params;
3345
3346     if (!strcmp(filename, "-"))
3347         filename = "pipe:";
3348
3349     oc = av_alloc_format_context();
3350
3351     if (!file_oformat) {
3352         file_oformat = guess_format(NULL, filename, NULL);
3353         if (!file_oformat) {
3354             fprintf(stderr, "Unable for find a suitable output format for '%s'\n",
3355                     filename);
3356             exit(1);
3357         }
3358     }
3359
3360     oc->oformat = file_oformat;
3361     pstrcpy(oc->filename, sizeof(oc->filename), filename);
3362
3363     if (!strcmp(file_oformat->name, "ffm") &&
3364         strstart(filename, "http:", NULL)) {
3365         /* special case for files sent to ffserver: we get the stream
3366            parameters from ffserver */
3367         if (read_ffserver_streams(oc, filename) < 0) {
3368             fprintf(stderr, "Could not read stream parameters from '%s'\n", filename);
3369             exit(1);
3370         }
3371     } else {
3372         use_video = file_oformat->video_codec != CODEC_ID_NONE || video_stream_copy || video_codec_id != CODEC_ID_NONE;
3373         use_audio = file_oformat->audio_codec != CODEC_ID_NONE || audio_stream_copy || audio_codec_id != CODEC_ID_NONE;
3374
3375         /* disable if no corresponding type found and at least one
3376            input file */
3377         if (nb_input_files > 0) {
3378             check_audio_video_inputs(&input_has_video, &input_has_audio);
3379             if (!input_has_video)
3380                 use_video = 0;
3381             if (!input_has_audio)
3382                 use_audio = 0;
3383         }
3384
3385         /* manual disable */
3386         if (audio_disable) {
3387             use_audio = 0;
3388         }
3389         if (video_disable) {
3390             use_video = 0;
3391         }
3392
3393         if (use_video) {
3394             new_video_stream(oc);
3395         }
3396
3397         if (use_audio) {
3398             new_audio_stream(oc);
3399         }
3400
3401         if (!oc->nb_streams) {
3402             fprintf(stderr, "No audio or video streams available\n");
3403             exit(1);
3404         }
3405
3406         oc->timestamp = rec_timestamp;
3407
3408         if (str_title)
3409             pstrcpy(oc->title, sizeof(oc->title), str_title);
3410         if (str_author)
3411             pstrcpy(oc->author, sizeof(oc->author), str_author);
3412         if (str_copyright)
3413             pstrcpy(oc->copyright, sizeof(oc->copyright), str_copyright);
3414         if (str_comment)
3415             pstrcpy(oc->comment, sizeof(oc->comment), str_comment);
3416     }
3417
3418     output_files[nb_output_files++] = oc;
3419
3420     /* check filename in case of an image number is expected */
3421     if (oc->oformat->flags & AVFMT_NEEDNUMBER) {
3422         if (filename_number_test(oc->filename) < 0) {
3423             print_error(oc->filename, AVERROR_NUMEXPECTED);
3424             exit(1);
3425         }
3426     }
3427
3428     if (!(oc->oformat->flags & AVFMT_NOFILE)) {
3429         /* test if it already exists to avoid loosing precious files */
3430         if (!file_overwrite &&
3431             (strchr(filename, ':') == NULL ||
3432              strstart(filename, "file:", NULL))) {
3433             if (url_exist(filename)) {
3434                 int c;
3435
3436                 if ( !using_stdin ) {
3437                     fprintf(stderr,"File '%s' already exists. Overwrite ? [y/N] ", filename);
3438                     fflush(stderr);
3439                     c = getchar();
3440                     if (toupper(c) != 'Y') {
3441                         fprintf(stderr, "Not overwriting - exiting\n");
3442                         exit(1);
3443                     }
3444                                 }
3445                                 else {
3446                     fprintf(stderr,"File '%s' already exists. Exiting.\n", filename);
3447                     exit(1);
3448                                 }
3449             }
3450         }
3451
3452         /* open the file */
3453         if (url_fopen(&oc->pb, filename, URL_WRONLY) < 0) {
3454             fprintf(stderr, "Could not open '%s'\n", filename);
3455             exit(1);
3456         }
3457     }
3458
3459     memset(ap, 0, sizeof(*ap));
3460     ap->image_format = image_format;
3461     if (av_set_parameters(oc, ap) < 0) {
3462         fprintf(stderr, "%s: Invalid encoding parameters\n",
3463                 oc->filename);
3464         exit(1);
3465     }
3466
3467     oc->packet_size= mux_packet_size;
3468     oc->mux_rate= mux_rate;
3469     oc->preload= (int)(mux_preload*AV_TIME_BASE);
3470     oc->max_delay= (int)(mux_max_delay*AV_TIME_BASE);
3471     oc->loop_output = loop_output;
3472
3473     /* reset some options */
3474     file_oformat = NULL;
3475     file_iformat = NULL;
3476     image_format = NULL;
3477 }
3478
3479 /* prepare dummy protocols for grab */
3480 static void prepare_grab(void)
3481 {
3482     int has_video, has_audio, i, j;
3483     AVFormatContext *oc;
3484     AVFormatContext *ic;
3485     AVFormatParameters vp1, *vp = &vp1;
3486     AVFormatParameters ap1, *ap = &ap1;
3487
3488     /* see if audio/video inputs are needed */
3489     has_video = 0;
3490     has_audio = 0;
3491     memset(ap, 0, sizeof(*ap));
3492     memset(vp, 0, sizeof(*vp));
3493     vp->time_base.num= 1;
3494     for(j=0;j<nb_output_files;j++) {
3495         oc = output_files[j];
3496         for(i=0;i<oc->nb_streams;i++) {
3497             AVCodecContext *enc = oc->streams[i]->codec;
3498             switch(enc->codec_type) {
3499             case CODEC_TYPE_AUDIO:
3500                 if (enc->sample_rate > ap->sample_rate)
3501                     ap->sample_rate = enc->sample_rate;
3502                 if (enc->channels > ap->channels)
3503                     ap->channels = enc->channels;
3504                 has_audio = 1;
3505                 break;
3506             case CODEC_TYPE_VIDEO:
3507                 if (enc->width > vp->width)
3508                     vp->width = enc->width;
3509                 if (enc->height > vp->height)
3510                     vp->height = enc->height;
3511
3512                 if (vp->time_base.num*(int64_t)enc->time_base.den > enc->time_base.num*(int64_t)vp->time_base.den){
3513                     vp->time_base = enc->time_base;
3514                     vp->width += frame_leftBand + frame_rightBand;
3515                     vp->width -= (frame_padleft + frame_padright);
3516                     vp->height += frame_topBand + frame_bottomBand;
3517                     vp->height -= (frame_padtop + frame_padbottom);
3518                 }
3519                 has_video = 1;
3520                 break;
3521             default:
3522                 av_abort();
3523             }
3524         }
3525     }
3526
3527     if (has_video == 0 && has_audio == 0) {
3528         fprintf(stderr, "Output file must have at least one audio or video stream\n");
3529         exit(1);
3530     }
3531
3532     if (has_video) {
3533         AVInputFormat *fmt1;
3534         fmt1 = av_find_input_format(video_grab_format);
3535         vp->device  = video_device;
3536         vp->channel = video_channel;
3537         vp->standard = video_standard;
3538         vp->pix_fmt = frame_pix_fmt;
3539         if (av_open_input_file(&ic, "", fmt1, 0, vp) < 0) {
3540             fprintf(stderr, "Could not find video grab device\n");
3541             exit(1);
3542         }
3543         /* If not enough info to get the stream parameters, we decode the
3544            first frames to get it. */
3545         if ((ic->ctx_flags & AVFMTCTX_NOHEADER) && av_find_stream_info(ic) < 0) {
3546             fprintf(stderr, "Could not find video grab parameters\n");
3547             exit(1);
3548         }
3549         /* by now video grab has one stream */
3550         ic->streams[0]->r_frame_rate.num = vp->time_base.den;
3551         ic->streams[0]->r_frame_rate.den = vp->time_base.num;
3552         input_files[nb_input_files] = ic;
3553
3554         if (verbose >= 0)
3555             dump_format(ic, nb_input_files, "", 0);
3556
3557         nb_input_files++;
3558     }
3559     if (has_audio && audio_grab_format) {
3560         AVInputFormat *fmt1;
3561         fmt1 = av_find_input_format(audio_grab_format);
3562         ap->device = audio_device;
3563         if (av_open_input_file(&ic, "", fmt1, 0, ap) < 0) {
3564             fprintf(stderr, "Could not find audio grab device\n");
3565             exit(1);
3566         }
3567         input_files[nb_input_files] = ic;
3568
3569         if (verbose >= 0)
3570             dump_format(ic, nb_input_files, "", 0);
3571
3572         nb_input_files++;
3573     }
3574 }
3575
3576 /* same option as mencoder */
3577 static void opt_pass(const char *pass_str)
3578 {
3579     int pass;
3580     pass = atoi(pass_str);
3581     if (pass != 1 && pass != 2) {
3582         fprintf(stderr, "pass number can be only 1 or 2\n");
3583         exit(1);
3584     }
3585     do_pass = pass;
3586 }
3587
3588 #if defined(CONFIG_WIN32) || defined(CONFIG_OS2)
3589 static int64_t getutime(void)
3590 {
3591   return av_gettime();
3592 }
3593 #else
3594 static int64_t getutime(void)
3595 {
3596     struct rusage rusage;
3597
3598     getrusage(RUSAGE_SELF, &rusage);
3599     return (rusage.ru_utime.tv_sec * 1000000LL) + rusage.ru_utime.tv_usec;
3600 }
3601 #endif
3602
3603 extern int ffm_nopts;
3604
3605 static void show_formats(void)
3606 {
3607     AVInputFormat *ifmt;
3608     AVOutputFormat *ofmt;
3609     AVImageFormat *image_fmt;
3610     URLProtocol *up;
3611     AVCodec *p, *p2;
3612     const char **pp, *last_name;
3613
3614     printf("File formats:\n");
3615     last_name= "000";
3616     for(;;){
3617         int decode=0;
3618         int encode=0;
3619         const char *name=NULL;
3620         const char *long_name=NULL;
3621
3622         for(ofmt = first_oformat; ofmt != NULL; ofmt = ofmt->next) {
3623             if((name == NULL || strcmp(ofmt->name, name)<0) &&
3624                 strcmp(ofmt->name, last_name)>0){
3625                 name= ofmt->name;
3626                 long_name= ofmt->long_name;
3627                 encode=1;
3628             }
3629         }
3630         for(ifmt = first_iformat; ifmt != NULL; ifmt = ifmt->next) {
3631             if((name == NULL || strcmp(ifmt->name, name)<0) &&
3632                 strcmp(ifmt->name, last_name)>0){
3633                 name= ifmt->name;
3634                 long_name= ifmt->long_name;
3635                 encode=0;
3636             }
3637             if(name && strcmp(ifmt->name, name)==0)
3638                 decode=1;
3639         }
3640         if(name==NULL)
3641             break;
3642         last_name= name;
3643
3644         printf(
3645             " %s%s %-15s %s\n",
3646             decode ? "D":" ",
3647             encode ? "E":" ",
3648             name,
3649             long_name ? long_name:" ");
3650     }
3651     printf("\n");
3652
3653     printf("Image formats (filename extensions, if any, follow):\n");
3654     for(image_fmt = first_image_format; image_fmt != NULL;
3655         image_fmt = image_fmt->next) {
3656         printf(
3657             " %s%s %-6s %s\n",
3658             image_fmt->img_read  ? "D":" ",
3659             image_fmt->img_write ? "E":" ",
3660             image_fmt->name,
3661             image_fmt->extensions ? image_fmt->extensions:" ");
3662     }
3663     printf("\n");
3664
3665     printf("Codecs:\n");
3666     last_name= "000";
3667     for(;;){
3668         int decode=0;
3669         int encode=0;
3670         int cap=0;
3671         const char *type_str;
3672
3673         p2=NULL;
3674         for(p = first_avcodec; p != NULL; p = p->next) {
3675             if((p2==NULL || strcmp(p->name, p2->name)<0) &&
3676                 strcmp(p->name, last_name)>0){
3677                 p2= p;
3678                 decode= encode= cap=0;
3679             }
3680             if(p2 && strcmp(p->name, p2->name)==0){
3681                 if(p->decode) decode=1;
3682                 if(p->encode) encode=1;
3683                 cap |= p->capabilities;
3684             }
3685         }
3686         if(p2==NULL)
3687             break;
3688         last_name= p2->name;
3689
3690         switch(p2->type) {
3691         case CODEC_TYPE_VIDEO:
3692             type_str = "V";
3693             break;
3694         case CODEC_TYPE_AUDIO:
3695             type_str = "A";
3696             break;
3697         case CODEC_TYPE_SUBTITLE:
3698             type_str = "S";
3699             break;
3700         default:
3701             type_str = "?";
3702             break;
3703         }
3704         printf(
3705             " %s%s%s%s%s%s %s",
3706             decode ? "D": (/*p2->decoder ? "d":*/" "),
3707             encode ? "E":" ",
3708             type_str,
3709             cap & CODEC_CAP_DRAW_HORIZ_BAND ? "S":" ",
3710             cap & CODEC_CAP_DR1 ? "D":" ",
3711             cap & CODEC_CAP_TRUNCATED ? "T":" ",
3712             p2->name);
3713        /* if(p2->decoder && decode==0)
3714             printf(" use %s for decoding", p2->decoder->name);*/
3715         printf("\n");
3716     }
3717     printf("\n");
3718
3719     printf("Supported file protocols:\n");
3720     for(up = first_protocol; up != NULL; up = up->next)
3721         printf(" %s:", up->name);
3722     printf("\n");
3723
3724     printf("Frame size, frame rate abbreviations:\n ntsc pal qntsc qpal sntsc spal film ntsc-film sqcif qcif cif 4cif\n");
3725     printf("Motion estimation methods:\n");
3726     pp = motion_str;
3727     while (*pp) {
3728         printf(" %s", *pp);
3729         if ((pp - motion_str + 1) == ME_ZERO)
3730             printf("(fastest)");
3731         else if ((pp - motion_str + 1) == ME_FULL)
3732             printf("(slowest)");
3733         else if ((pp - motion_str + 1) == ME_EPZS)
3734             printf("(default)");
3735         pp++;
3736     }
3737     printf("\n\n");
3738     printf(
3739 "Note, the names of encoders and decoders dont always match, so there are\n"
3740 "several cases where the above table shows encoder only or decoder only entries\n"
3741 "even though both encoding and decoding are supported for example, the h263\n"
3742 "decoder corresponds to the h263 and h263p encoders, for file formats its even\n"
3743 "worse\n");
3744     exit(1);
3745 }
3746
3747 static void parse_matrix_coeffs(uint16_t *dest, const char *str)
3748 {
3749     int i;
3750     const char *p = str;
3751     for(i = 0;; i++) {
3752         dest[i] = atoi(p);
3753         if(i == 63)
3754             break;
3755         p = strchr(p, ',');
3756         if(!p) {
3757             fprintf(stderr, "Syntax error in matrix \"%s\" at coeff %d\n", str, i);
3758             exit(1);
3759         }