ffmpeg.c

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