git.sesse.net Git - ffmpeg/blob - ffmpeg.c

   1 /*
   2  * FFmpeg main
   3  * Copyright (c) 2000-2003 Fabrice Bellard
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General Public
  16  * License along with this library; if not, write to the Free Software
  17  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  18  */
  19 #define HAVE_AV_CONFIG_H
  20 #include "avformat.h"
  21 #include "framehook.h"
  22
  23 #ifndef CONFIG_WIN32
  24 #include <unistd.h>
  25 #include <fcntl.h>
  26 #include <sys/ioctl.h>
  27 #include <sys/time.h>
  28 #include <termios.h>
  29 #include <sys/resource.h>
  30 #include <signal.h>
  31 #endif
  32 #ifdef CONFIG_OS2
  33 #include <sys/types.h>
  34 #include <sys/select.h>
  35 #include <stdlib.h>
  36 #endif
  37 #include <time.h>
  38
  39 #include "cmdutils.h"
  40
  41 #if !defined(INFINITY) && defined(HUGE_VAL)
  42 #define INFINITY HUGE_VAL
  43 #endif
  44
  45 /* select an input stream for an output stream */
  46 typedef struct AVStreamMap {
  47     int file_index;
  48     int stream_index;
  49 } AVStreamMap;
  50
  51 extern const OptionDef options[];
  52
  53 static void show_help(void);
  54 static void show_license(void);
  55
  56 #define MAX_FILES 20
  57
  58 static AVFormatContext *input_files[MAX_FILES];
  59 static int nb_input_files = 0;
  60
  61 static AVFormatContext *output_files[MAX_FILES];
  62 static int nb_output_files = 0;
  63
  64 static AVStreamMap stream_maps[MAX_FILES];
  65 static int nb_stream_maps;
  66
  67 static AVInputFormat *file_iformat;
  68 static AVOutputFormat *file_oformat;
  69 static AVImageFormat *image_format;
  70 static int frame_width  = 160;
  71 static int frame_height = 128;
  72 static float frame_aspect_ratio = 0;
  73 static enum PixelFormat frame_pix_fmt = PIX_FMT_YUV420P;
  74 static int frame_topBand  = 0;
  75 static int frame_bottomBand = 0;
  76 static int frame_leftBand  = 0;
  77 static int frame_rightBand = 0;
  78 static int frame_rate = 25;
  79 static int frame_rate_base = 1;
  80 static int video_bit_rate = 200*1000;
  81 static int video_bit_rate_tolerance = 4000*1000;
  82 static float video_qscale = 0;
  83 static int video_qmin = 2;
  84 static int video_qmax = 31;
  85 static int video_mb_qmin = 2;
  86 static int video_mb_qmax = 31;
  87 static int video_qdiff = 3;
  88 static float video_qblur = 0.5;
  89 static float video_qcomp = 0.5;
  90 static uint16_t *intra_matrix = NULL;
  91 static uint16_t *inter_matrix = NULL;
  92 #if 0 //experimental, (can be removed)
  93 static float video_rc_qsquish=1.0;
  94 static float video_rc_qmod_amp=0;
  95 static int video_rc_qmod_freq=0;
  96 #endif
  97 static char *video_rc_override_string=NULL;
  98 static char *video_rc_eq="tex^qComp";
  99 static int video_rc_buffer_size=0;
 100 static float video_rc_buffer_aggressivity=1.0;
 101 static int video_rc_max_rate=0;
 102 static int video_rc_min_rate=0;
 103 static float video_rc_initial_cplx=0;
 104 static float video_b_qfactor = 1.25;
 105 static float video_b_qoffset = 1.25;
 106 static float video_i_qfactor = -0.8;
 107 static float video_i_qoffset = 0.0;
 108 static int video_intra_quant_bias= FF_DEFAULT_QUANT_BIAS;
 109 static int video_inter_quant_bias= FF_DEFAULT_QUANT_BIAS;
 110 static int me_method = ME_EPZS;
 111 static int video_disable = 0;
 112 static int video_codec_id = CODEC_ID_NONE;
 113 static int same_quality = 0;
 114 static int b_frames = 0;
 115 static int mb_decision = FF_MB_DECISION_SIMPLE;
 116 static int ildct_cmp = FF_CMP_VSAD;
 117 static int mb_cmp = FF_CMP_SAD;
 118 static int sub_cmp = FF_CMP_SAD;
 119 static int cmp = FF_CMP_SAD;
 120 static int use_4mv = 0;
 121 static int use_obmc = 0;
 122 static int use_aic = 0;
 123 static int use_aiv = 0;
 124 static int use_umv = 0;
 125 static int use_alt_scan = 0;
 126 static int use_trell = 0;
 127 static int use_scan_offset = 0;
 128 static int closed_gop = 0;
 129 static int do_deinterlace = 0;
 130 static int do_interlace_dct = 0;
 131 static int do_interlace_me = 0;
 132 static int workaround_bugs = FF_BUG_AUTODETECT;
 133 static int error_resilience = 2;
 134 static int error_concealment = 3;
 135 static int dct_algo = 0;
 136 static int idct_algo = 0;
 137 static int use_part = 0;
 138 static int packet_size = 0;
 139 static int error_rate = 0;
 140 static int strict = 0;
 141 static int top_field_first = -1;
 142 static int noise_reduction = 0;
 143 static int sc_threshold = 0;
 144 static int debug = 0;
 145 static int debug_mv = 0;
 146 extern int loop_input; /* currently a hack */
 147
 148 static int gop_size = 12;
 149 static int intra_only = 0;
 150 static int audio_sample_rate = 44100;
 151 static int audio_bit_rate = 64000;
 152 static int audio_disable = 0;
 153 static int audio_channels = 1;
 154 static int audio_codec_id = CODEC_ID_NONE;
 155
 156 static int64_t recording_time = 0;
 157 static int64_t start_time = 0;
 158 static int file_overwrite = 0;
 159 static char *str_title = NULL;
 160 static char *str_author = NULL;
 161 static char *str_copyright = NULL;
 162 static char *str_comment = NULL;
 163 static int do_benchmark = 0;
 164 static int do_hex_dump = 0;
 165 static int do_pkt_dump = 0;
 166 static int do_psnr = 0;
 167 static int do_vstats = 0;
 168 static int do_pass = 0;
 169 static int bitexact = 0;
 170 static char *pass_logfilename = NULL;
 171 static int audio_stream_copy = 0;
 172 static int video_stream_copy = 0;
 173
 174 static int rate_emu = 0;
 175
 176 static char *video_grab_format = "video4linux";
 177 static char *video_device = NULL;
 178 static int  video_channel = 0;
 179 static char *video_standard = "ntsc";
 180
 181 static char *audio_grab_format = "audio_device";
 182 static char *audio_device = NULL;
 183
 184 static int using_stdin = 0;
 185 static int using_vhook = 0;
 186 static int verbose = 1;
 187
 188 #define DEFAULT_PASS_LOGFILENAME "ffmpeg2pass"
 189
 190 typedef struct AVOutputStream {
 191     int file_index;          /* file index */
 192     int index;               /* stream index in the output file */
 193     int source_index;        /* AVInputStream index */
 194     AVStream *st;            /* stream in the output file */
 195     int encoding_needed;     /* true if encoding needed for this stream */
 196     int frame_number;
 197     /* input pts and corresponding output pts
 198        for A/V sync */
 199     double sync_ipts;
 200     double sync_ipts_offset;
 201     int64_t sync_opts;
 202     /* video only */
 203     int video_resample;      /* video_resample and video_crop are mutually exclusive */
 204     AVPicture pict_tmp;      /* temporary image for resampling */
 205     ImgReSampleContext *img_resample_ctx; /* for image resampling */
 206
 207     int video_crop;          /* video_resample and video_crop are mutually exclusive */
 208     int topBand;             /* cropping area sizes */
 209     int leftBand;
 210
 211     /* audio only */
 212     int audio_resample;
 213     ReSampleContext *resample; /* for audio resampling */
 214     FifoBuffer fifo;     /* for compression: one audio fifo per codec */
 215     FILE *logfile;
 216 } AVOutputStream;
 217
 218 typedef struct AVInputStream {
 219     int file_index;
 220     int index;
 221     AVStream *st;
 222     int discard;             /* true if stream data should be discarded */
 223     int decoding_needed;     /* true if the packets must be decoded in 'raw_fifo' */
 224     int64_t sample_index;      /* current sample */
 225
 226     int64_t       start;     /* time when read started */
 227     unsigned long frame;     /* current frame */
 228     int64_t       next_pts;  /* synthetic pts for cases where pkt.pts
 229                                 is not defined */
 230     int64_t       pts;       /* current pts */
 231 } AVInputStream;
 232
 233 typedef struct AVInputFile {
 234     int eof_reached;      /* true if eof reached */
 235     int ist_index;        /* index of first stream in ist_table */
 236     int buffer_size;      /* current total buffer size */
 237     int buffer_size_max;  /* buffer size at which we consider we can stop
 238                              buffering */
 239     int nb_streams;       /* nb streams we are aware of */
 240 } AVInputFile;
 241
 242 #ifndef CONFIG_WIN32
 243
 244 /* init terminal so that we can grab keys */
 245 static struct termios oldtty;
 246
 247 static void term_exit(void)
 248 {
 249     tcsetattr (0, TCSANOW, &oldtty);
 250 }
 251
 252 static volatile sig_atomic_t received_sigterm = 0;
 253
 254 static void
 255 sigterm_handler(int sig)
 256 {
 257     received_sigterm = sig;
 258     term_exit();
 259 }
 260
 261 static void term_init(void)
 262 {
 263     struct termios tty;
 264
 265     tcgetattr (0, &tty);
 266     oldtty = tty;
 267
 268     tty.c_iflag &= ~(IGNBRK|BRKINT|PARMRK|ISTRIP
 269                           |INLCR|IGNCR|ICRNL|IXON);
 270     tty.c_oflag |= OPOST;
 271     tty.c_lflag &= ~(ECHO|ECHONL|ICANON|IEXTEN);
 272     tty.c_cflag &= ~(CSIZE|PARENB);
 273     tty.c_cflag |= CS8;
 274     tty.c_cc[VMIN] = 1;
 275     tty.c_cc[VTIME] = 0;
 276
 277     tcsetattr (0, TCSANOW, &tty);
 278
 279     signal(SIGINT , sigterm_handler); /* Interrupt (ANSI).  */
 280     signal(SIGQUIT, sigterm_handler); /* Quit (POSIX).  */
 281     signal(SIGTERM, sigterm_handler); /* Termination (ANSI).  */
 282     /*
 283     register a function to be called at normal program termination
 284     */
 285     atexit(term_exit);
 286 #ifdef CONFIG_BEOS_NETSERVER
 287     fcntl(0, F_SETFL, fcntl(0, F_GETFL) | O_NONBLOCK);
 288 #endif
 289 }
 290
 291 /* read a key without blocking */
 292 static int read_key(void)
 293 {
 294     int n = 1;
 295     unsigned char ch;
 296 #ifndef CONFIG_BEOS_NETSERVER
 297     struct timeval tv;
 298     fd_set rfds;
 299
 300     FD_ZERO(&rfds);
 301     FD_SET(0, &rfds);
 302     tv.tv_sec = 0;
 303     tv.tv_usec = 0;
 304     n = select(1, &rfds, NULL, NULL, &tv);
 305 #endif
 306     if (n > 0) {
 307         n = read(0, &ch, 1);
 308         if (n == 1)
 309             return ch;
 310
 311         return n;
 312     }
 313     return -1;
 314 }
 315
 316 #else
 317
 318 static volatile int received_sigterm = 0;
 319
 320 /* no interactive support */
 321 static void term_exit(void)
 322 {
 323 }
 324
 325 static void term_init(void)
 326 {
 327 }
 328
 329 static int read_key(void)
 330 {
 331     return 0;
 332 }
 333
 334 #endif
 335
 336 static int read_ffserver_streams(AVFormatContext *s, const char *filename)
 337 {
 338     int i, err;
 339     AVFormatContext *ic;
 340
 341     err = av_open_input_file(&ic, filename, NULL, FFM_PACKET_SIZE, NULL);
 342     if (err < 0)
 343         return err;
 344     /* copy stream format */
 345     s->nb_streams = ic->nb_streams;
 346     for(i=0;i<ic->nb_streams;i++) {
 347         AVStream *st;
 348
 349         st = av_mallocz(sizeof(AVStream));
 350         memcpy(st, ic->streams[i], sizeof(AVStream));
 351         s->streams[i] = st;
 352     }
 353
 354     av_close_input_file(ic);
 355     return 0;
 356 }
 357
 358 #define MAX_AUDIO_PACKET_SIZE (128 * 1024)
 359
 360 static void do_audio_out(AVFormatContext *s,
 361                          AVOutputStream *ost,
 362                          AVInputStream *ist,
 363                          unsigned char *buf, int size)
 364 {
 365     uint8_t *buftmp;
 366     static uint8_t *audio_buf = NULL;
 367     static uint8_t *audio_out = NULL;
 368     const int audio_out_size= 4*MAX_AUDIO_PACKET_SIZE;
 369
 370     int size_out, frame_bytes, ret;
 371     AVCodecContext *enc;
 372
 373     /* SC: dynamic allocation of buffers */
 374     if (!audio_buf)
 375         audio_buf = av_malloc(2*MAX_AUDIO_PACKET_SIZE);
 376     if (!audio_out)
 377         audio_out = av_malloc(audio_out_size);
 378     if (!audio_buf || !audio_out)
 379         return;               /* Should signal an error ! */
 380
 381
 382     enc = &ost->st->codec;
 383
 384     if (ost->audio_resample) {
 385         buftmp = audio_buf;
 386         size_out = audio_resample(ost->resample,
 387                                   (short *)buftmp, (short *)buf,
 388                                   size / (ist->st->codec.channels * 2));
 389         size_out = size_out * enc->channels * 2;
 390     } else {
 391         buftmp = buf;
 392         size_out = size;
 393     }
 394
 395     /* now encode as many frames as possible */
 396     if (enc->frame_size > 1) {
 397         /* output resampled raw samples */
 398         fifo_write(&ost->fifo, buftmp, size_out,
 399                    &ost->fifo.wptr);
 400
 401         frame_bytes = enc->frame_size * 2 * enc->channels;
 402
 403         while (fifo_read(&ost->fifo, audio_buf, frame_bytes,
 404                      &ost->fifo.rptr) == 0) {
 405             ret = avcodec_encode_audio(enc, audio_out, audio_out_size,
 406                                        (short *)audio_buf);
 407             av_write_frame(s, ost->index, audio_out, ret);
 408         }
 409     } else {
 410         /* output a pcm frame */
 411         /* XXX: change encoding codec API to avoid this ? */
 412         switch(enc->codec->id) {
 413         case CODEC_ID_PCM_S16LE:
 414         case CODEC_ID_PCM_S16BE:
 415         case CODEC_ID_PCM_U16LE:
 416         case CODEC_ID_PCM_U16BE:
 417             break;
 418         default:
 419             size_out = size_out >> 1;
 420             break;
 421         }
 422         ret = avcodec_encode_audio(enc, audio_out, size_out,
 423                                    (short *)buftmp);
 424         av_write_frame(s, ost->index, audio_out, ret);
 425     }
 426 }
 427
 428 static void pre_process_video_frame(AVInputStream *ist, AVPicture *picture, void **bufp)
 429 {
 430     AVCodecContext *dec;
 431     AVPicture *picture2;
 432     AVPicture picture_tmp;
 433     uint8_t *buf = 0;
 434
 435     dec = &ist->st->codec;
 436
 437     /* deinterlace : must be done before any resize */
 438     if (do_deinterlace || using_vhook) {
 439         int size;
 440
 441         /* create temporary picture */
 442         size = avpicture_get_size(dec->pix_fmt, dec->width, dec->height);
 443         buf = av_malloc(size);
 444         if (!buf)
 445             return;
 446
 447         picture2 = &picture_tmp;
 448         avpicture_fill(picture2, buf, dec->pix_fmt, dec->width, dec->height);
 449
 450         if (do_deinterlace){
 451             if(avpicture_deinterlace(picture2, picture,
 452                                      dec->pix_fmt, dec->width, dec->height) < 0) {
 453                 /* if error, do not deinterlace */
 454                 av_free(buf);
 455                 buf = NULL;
 456                 picture2 = picture;
 457             }
 458         } else {
 459             if (img_convert(picture2, dec->pix_fmt, picture,
 460                             dec->pix_fmt, dec->width, dec->height) < 0) {
 461                 /* if error, do not copy */
 462                 av_free(buf);
 463                 buf = NULL;
 464                 picture2 = picture;
 465             }
 466         }
 467     } else {
 468         picture2 = picture;
 469     }
 470
 471     frame_hook_process(picture2, dec->pix_fmt, dec->width, dec->height);
 472
 473     if (picture != picture2)
 474         *picture = *picture2;
 475     *bufp = buf;
 476 }
 477
 478 /* we begin to correct av delay at this threshold */
 479 #define AV_DELAY_MAX 0.100
 480
 481 static void do_video_out(AVFormatContext *s,
 482                          AVOutputStream *ost,
 483                          AVInputStream *ist,
 484                          AVFrame *in_picture,
 485                          int *frame_size, AVOutputStream *audio_sync)
 486 {
 487     int nb_frames, i, ret;
 488     AVPicture *final_picture, *formatted_picture;
 489     AVPicture picture_format_temp, picture_crop_temp;
 490     static uint8_t *video_buffer= NULL;
 491     uint8_t *buf = NULL, *buf1 = NULL;
 492     AVCodecContext *enc, *dec;
 493     enum PixelFormat target_pixfmt;
 494
 495 #define VIDEO_BUFFER_SIZE (1024*1024)
 496
 497     enc = &ost->st->codec;
 498     dec = &ist->st->codec;
 499
 500     /* by default, we output a single frame */
 501     nb_frames = 1;
 502
 503     *frame_size = 0;
 504
 505     /* NOTE: the A/V sync is always done by considering the audio is
 506        the master clock. It is suffisant for transcoding or playing,
 507        but not for the general case */
 508     if (audio_sync) {
 509         /* compute the A-V delay and duplicate/remove frames if needed */
 510         double adelta, vdelta, av_delay;
 511
 512         adelta = audio_sync->sync_ipts - ((double)audio_sync->sync_opts *
 513             s->pts_num / s->pts_den);
 514
 515         vdelta = ost->sync_ipts - ((double)ost->sync_opts *
 516             s->pts_num / s->pts_den);
 517
 518         av_delay = adelta - vdelta;
 519         //            printf("delay=%f\n", av_delay);
 520         if (av_delay < -AV_DELAY_MAX)
 521             nb_frames = 2;
 522         else if (av_delay > AV_DELAY_MAX)
 523             nb_frames = 0;
 524     } else {
 525         double vdelta;
 526
 527         vdelta = (double)(ost->st->pts.val) * s->pts_num / s->pts_den - (ost->sync_ipts - ost->sync_ipts_offset);
 528         if (vdelta < 100 && vdelta > -100 && ost->sync_ipts_offset) {
 529             if (vdelta < -AV_DELAY_MAX)
 530                 nb_frames = 2;
 531             else if (vdelta > AV_DELAY_MAX)
 532                 nb_frames = 0;
 533         } else {
 534             ost->sync_ipts_offset -= vdelta;
 535             if (!ost->sync_ipts_offset)
 536                 ost->sync_ipts_offset = 0.000001; /* one microsecond */
 537         }
 538     }
 539
 540 #if defined(AVSYNC_DEBUG)
 541     {
 542         static char *action[] = { "drop frame", "copy frame", "dup frame" };
 543         if (audio_sync)
 544             fprintf(stderr, "Input APTS %12.6f, output APTS %12.6f, ",
 545                     (double) audio_sync->sync_ipts,
 546                     (double) audio_sync->st->pts.val * s->pts_num / s->pts_den);
 547         fprintf(stderr, "Input VPTS %12.6f, output VPTS %12.6f: %s\n",
 548                 (double) ost->sync_ipts,
 549                 (double) ost->st->pts.val * s->pts_num / s->pts_den,
 550                 action[nb_frames]);
 551     }
 552 #endif
 553
 554     if (nb_frames <= 0)
 555         return;
 556
 557     if (!video_buffer)
 558         video_buffer = av_malloc(VIDEO_BUFFER_SIZE);
 559     if (!video_buffer)
 560         return;
 561
 562     /* convert pixel format if needed */
 563     target_pixfmt = ost->video_resample ? PIX_FMT_YUV420P : enc->pix_fmt;
 564     if (dec->pix_fmt != target_pixfmt) {
 565         int size;
 566
 567         /* create temporary picture */
 568         size = avpicture_get_size(target_pixfmt, dec->width, dec->height);
 569         buf = av_malloc(size);
 570         if (!buf)
 571             return;
 572         formatted_picture = &picture_format_temp;
 573         avpicture_fill(formatted_picture, buf, target_pixfmt, dec->width, dec->height);
 574
 575         if (img_convert(formatted_picture, target_pixfmt,
 576                         (AVPicture *)in_picture, dec->pix_fmt,
 577                         dec->width, dec->height) < 0) {
 578             fprintf(stderr, "pixel format conversion not handled\n");
 579             goto the_end;
 580         }
 581     } else {
 582         formatted_picture = (AVPicture *)in_picture;
 583     }
 584
 585     /* XXX: resampling could be done before raw format convertion in
 586        some cases to go faster */
 587     /* XXX: only works for YUV420P */
 588     if (ost->video_resample) {
 589         final_picture = &ost->pict_tmp;
 590         img_resample(ost->img_resample_ctx, final_picture, formatted_picture);
 591         if (enc->pix_fmt != PIX_FMT_YUV420P) {
 592             int size;
 593
 594             av_free(buf);
 595             /* create temporary picture */
 596             size = avpicture_get_size(enc->pix_fmt, enc->width, enc->height);
 597             buf = av_malloc(size);
 598             if (!buf)
 599                 return;
 600             final_picture = &picture_format_temp;
 601             avpicture_fill(final_picture, buf, enc->pix_fmt, enc->width, enc->height);
 602
 603             if (img_convert(final_picture, enc->pix_fmt,
 604                             &ost->pict_tmp, PIX_FMT_YUV420P,
 605                             enc->width, enc->height) < 0) {
 606                 fprintf(stderr, "pixel format conversion not handled\n");
 607                 goto the_end;
 608             }
 609         }
 610     } else if (ost->video_crop) {
 611         picture_crop_temp.data[0] = formatted_picture->data[0] +
 612                 (ost->topBand * formatted_picture->linesize[0]) + ost->leftBand;
 613
 614         picture_crop_temp.data[1] = formatted_picture->data[1] +
 615                 ((ost->topBand >> 1) * formatted_picture->linesize[1]) +
 616                 (ost->leftBand >> 1);
 617
 618         picture_crop_temp.data[2] = formatted_picture->data[2] +
 619                 ((ost->topBand >> 1) * formatted_picture->linesize[2]) +
 620                 (ost->leftBand >> 1);
 621
 622         picture_crop_temp.linesize[0] = formatted_picture->linesize[0];
 623         picture_crop_temp.linesize[1] = formatted_picture->linesize[1];
 624         picture_crop_temp.linesize[2] = formatted_picture->linesize[2];
 625         final_picture = &picture_crop_temp;
 626     } else {
 627         final_picture = formatted_picture;
 628     }
 629     /* duplicates frame if needed */
 630     /* XXX: pb because no interleaving */
 631     for(i=0;i<nb_frames;i++) {
 632         if (s->oformat->flags & AVFMT_RAWPICTURE) {
 633             /* raw pictures are written as AVPicture structure to
 634                avoid any copies. We support temorarily the older
 635                method. */
 636             AVFrame* old_frame = enc->coded_frame;
 637             enc->coded_frame = dec->coded_frame;
 638             av_write_frame(s, ost->index,
 639                            (uint8_t *)final_picture, sizeof(AVPicture));
 640             enc->coded_frame = old_frame;
 641         } else {
 642             AVFrame big_picture;
 643
 644             memset(&big_picture, 0, sizeof(AVFrame));
 645             *(AVPicture*)&big_picture= *final_picture;
 646             /* better than nothing: use input picture interlaced
 647                settings */
 648             big_picture.interlaced_frame = in_picture->interlaced_frame;
 649             if(do_interlace_me || do_interlace_dct){
 650                 if(top_field_first == -1)
 651                     big_picture.top_field_first = in_picture->top_field_first;
 652                 else
 653                     big_picture.top_field_first = 1;
 654             }
 655
 656             /* handles sameq here. This is not correct because it may
 657                not be a global option */
 658             if (same_quality) {
 659                 big_picture.quality = ist->st->quality;
 660             }else
 661                 big_picture.quality = ost->st->quality;
 662
 663             ret = avcodec_encode_video(enc,
 664                                        video_buffer, VIDEO_BUFFER_SIZE,
 665                                        &big_picture);
 666             //enc->frame_number = enc->real_pict_num;
 667             av_write_frame(s, ost->index, video_buffer, ret);
 668             *frame_size = ret;
 669             //fprintf(stderr,"\nFrame: %3d %3d size: %5d type: %d",
 670             //        enc->frame_number-1, enc->real_pict_num, ret,
 671             //        enc->pict_type);
 672             /* if two pass, output log */
 673             if (ost->logfile && enc->stats_out) {
 674                 fprintf(ost->logfile, "%s", enc->stats_out);
 675             }
 676         }
 677         ost->frame_number++;
 678     }
 679  the_end:
 680     av_free(buf);
 681     av_free(buf1);
 682 }
 683
 684 static double psnr(double d){
 685     if(d==0) return INFINITY;
 686     return -10.0*log(d)/log(10.0);
 687 }
 688
 689 static void do_video_stats(AVFormatContext *os, AVOutputStream *ost,
 690                            int frame_size)
 691 {
 692     static FILE *fvstats=NULL;
 693     static int64_t total_size = 0;
 694     char filename[40];
 695     time_t today2;
 696     struct tm *today;
 697     AVCodecContext *enc;
 698     int frame_number;
 699     int64_t ti;
 700     double ti1, bitrate, avg_bitrate;
 701
 702     if (!fvstats) {
 703         today2 = time(NULL);
 704         today = localtime(&today2);
 705         sprintf(filename, "vstats_%02d%02d%02d.log", today->tm_hour,
 706                                                today->tm_min,
 707                                                today->tm_sec);
 708         fvstats = fopen(filename,"w");
 709         if (!fvstats) {
 710             perror("fopen");
 711             exit(1);
 712         }
 713     }
 714
 715     ti = MAXINT64;
 716     enc = &ost->st->codec;
 717     total_size += frame_size;
 718     if (enc->codec_type == CODEC_TYPE_VIDEO) {
 719         frame_number = ost->frame_number;
 720         fprintf(fvstats, "frame= %5d q= %2.1f ", frame_number, enc->coded_frame->quality/(float)FF_QP2LAMBDA);
 721         if (enc->flags&CODEC_FLAG_PSNR)
 722             fprintf(fvstats, "PSNR= %6.2f ", psnr(enc->coded_frame->error[0]/(enc->width*enc->height*255.0*255.0)));
 723
 724         fprintf(fvstats,"f_size= %6d ", frame_size);
 725         /* compute pts value */
 726         ti1 = (double)ost->st->pts.val * os->pts_num / os->pts_den;
 727         if (ti1 < 0.01)
 728             ti1 = 0.01;
 729
 730         bitrate = (double)(frame_size * 8) * enc->frame_rate / enc->frame_rate_base / 1000.0;
 731         avg_bitrate = (double)(total_size * 8) / ti1 / 1000.0;
 732         fprintf(fvstats, "s_size= %8.0fkB time= %0.3f br= %7.1fkbits/s avg_br= %7.1fkbits/s ",
 733             (double)total_size / 1024, ti1, bitrate, avg_bitrate);
 734         fprintf(fvstats,"type= %c\n", av_get_pict_type_char(enc->coded_frame->pict_type));
 735     }
 736 }
 737
 738 static void print_report(AVFormatContext **output_files,
 739                          AVOutputStream **ost_table, int nb_ostreams,
 740                          int is_last_report)
 741 {
 742     char buf[1024];
 743     AVOutputStream *ost;
 744     AVFormatContext *oc, *os;
 745     int64_t total_size;
 746     AVCodecContext *enc;
 747     int frame_number, vid, i;
 748     double bitrate, ti1, pts;
 749     static int64_t last_time = -1;
 750
 751     if (!is_last_report) {
 752         int64_t cur_time;
 753         /* display the report every 0.5 seconds */
 754         cur_time = av_gettime();
 755         if (last_time == -1) {
 756             last_time = cur_time;
 757             return;
 758         }
 759         if ((cur_time - last_time) < 500000)
 760             return;
 761         last_time = cur_time;
 762     }
 763
 764
 765     oc = output_files[0];
 766
 767     total_size = url_ftell(&oc->pb);
 768
 769     buf[0] = '\0';
 770     ti1 = 1e10;
 771     vid = 0;
 772     for(i=0;i<nb_ostreams;i++) {
 773         ost = ost_table[i];
 774         os = output_files[ost->file_index];
 775         enc = &ost->st->codec;
 776         if (vid && enc->codec_type == CODEC_TYPE_VIDEO) {
 777             sprintf(buf + strlen(buf), "q=%2.1f ",
 778                     enc->coded_frame->quality/(float)FF_QP2LAMBDA);
 779         }
 780         if (!vid && enc->codec_type == CODEC_TYPE_VIDEO) {
 781             frame_number = ost->frame_number;
 782             sprintf(buf + strlen(buf), "frame=%5d q=%2.1f ",
 783                     frame_number, enc->coded_frame ? enc->coded_frame->quality/(float)FF_QP2LAMBDA : 0);
 784             if(is_last_report)
 785                 sprintf(buf + strlen(buf), "L");
 786             if (enc->flags&CODEC_FLAG_PSNR){
 787                 int j;
 788                 double error, error_sum=0;
 789                 double scale, scale_sum=0;
 790                 char type[3]= {'Y','U','V'};
 791                 sprintf(buf + strlen(buf), "PSNR=");
 792                 for(j=0; j<3; j++){
 793                     if(is_last_report){
 794                         error= enc->error[j];
 795                         scale= enc->width*enc->height*255.0*255.0*frame_number;
 796                     }else{
 797                         error= enc->coded_frame->error[j];
 798                         scale= enc->width*enc->height*255.0*255.0;
 799                     }
 800                     if(j) scale/=4;
 801                     error_sum += error;
 802                     scale_sum += scale;
 803                     sprintf(buf + strlen(buf), "%c:%2.2f ", type[j], psnr(error/scale));
 804                 }
 805                 sprintf(buf + strlen(buf), "*:%2.2f ", psnr(error_sum/scale_sum));
 806             }
 807             vid = 1;
 808         }
 809         /* compute min output value */
 810         pts = (double)ost->st->pts.val * os->pts_num / os->pts_den;
 811         if ((pts < ti1) && (pts > 0))
 812             ti1 = pts;
 813     }
 814     if (ti1 < 0.01)
 815         ti1 = 0.01;
 816
 817     if (verbose || is_last_report) {
 818         bitrate = (double)(total_size * 8) / ti1 / 1000.0;
 819
 820         sprintf(buf + strlen(buf),
 821             "size=%8.0fkB time=%0.1f bitrate=%6.1fkbits/s",
 822             (double)total_size / 1024, ti1, bitrate);
 823
 824         fprintf(stderr, "%s    \r", buf);
 825         fflush(stderr);
 826     }
 827
 828     if (is_last_report)
 829         fprintf(stderr, "\n");
 830 }
 831
 832 /* pkt = NULL means EOF (needed to flush decoder buffers) */
 833 static int output_packet(AVInputStream *ist, int ist_index,
 834                          AVOutputStream **ost_table, int nb_ostreams,
 835                          AVPacket *pkt)
 836 {
 837     AVFormatContext *os;
 838     AVOutputStream *ost;
 839     uint8_t *ptr;
 840     int len, ret, i;
 841     uint8_t *data_buf;
 842     int data_size, got_picture;
 843     AVFrame picture;
 844     short samples[AVCODEC_MAX_AUDIO_FRAME_SIZE / 2];
 845     void *buffer_to_free;
 846
 847     if (pkt && pkt->pts != AV_NOPTS_VALUE) {
 848         ist->pts = pkt->pts;
 849     } else {
 850         ist->pts = ist->next_pts;
 851     }
 852
 853     if (pkt == NULL) {
 854         /* EOF handling */
 855         ptr = NULL;
 856         len = 0;
 857         goto handle_eof;
 858     }
 859
 860     len = pkt->size;
 861     ptr = pkt->data;
 862     while (len > 0) {
 863     handle_eof:
 864         /* decode the packet if needed */
 865         data_buf = NULL; /* fail safe */
 866         data_size = 0;
 867         if (ist->decoding_needed) {
 868             switch(ist->st->codec.codec_type) {
 869             case CODEC_TYPE_AUDIO:
 870                     /* XXX: could avoid copy if PCM 16 bits with same
 871                        endianness as CPU */
 872                 ret = avcodec_decode_audio(&ist->st->codec, samples, &data_size,
 873                                            ptr, len);
 874                 if (ret < 0)
 875                     goto fail_decode;
 876                 ptr += ret;
 877                 len -= ret;
 878                 /* Some bug in mpeg audio decoder gives */
 879                 /* data_size < 0, it seems they are overflows */
 880                 if (data_size <= 0) {
 881                     /* no audio frame */
 882                     continue;
 883                 }
 884                 data_buf = (uint8_t *)samples;
 885                 ist->next_pts += ((int64_t)AV_TIME_BASE * data_size) /
 886                     (2 * ist->st->codec.channels);
 887                 break;
 888             case CODEC_TYPE_VIDEO:
 889                     data_size = (ist->st->codec.width * ist->st->codec.height * 3) / 2;
 890                     /* XXX: allocate picture correctly */
 891                     memset(&picture, 0, sizeof(picture));
 892                     ret = avcodec_decode_video(&ist->st->codec,
 893                                                &picture, &got_picture, ptr, len);
 894                     ist->st->quality= picture.quality;
 895                     if (ret < 0)
 896                         goto fail_decode;
 897                     if (!got_picture) {
 898                         /* no picture yet */
 899                         goto discard_packet;
 900                     }
 901                     if (ist->st->codec.frame_rate_base != 0) {
 902                         ist->next_pts += ((int64_t)AV_TIME_BASE *
 903                                           ist->st->codec.frame_rate_base) /
 904                             ist->st->codec.frame_rate;
 905                     }
 906                     len = 0;
 907                     break;
 908                 default:
 909                     goto fail_decode;
 910                 }
 911             } else {
 912                 data_buf = ptr;
 913                 data_size = len;
 914                 ret = len;
 915                 len = 0;
 916             }
 917
 918             buffer_to_free = NULL;
 919             if (ist->st->codec.codec_type == CODEC_TYPE_VIDEO) {
 920                 pre_process_video_frame(ist, (AVPicture *)&picture,
 921                                         &buffer_to_free);
 922             }
 923
 924             /* frame rate emulation */
 925             if (ist->st->codec.rate_emu) {
 926                 int64_t pts = av_rescale((int64_t) ist->frame * ist->st->codec.frame_rate_base, 1000000, ist->st->codec.frame_rate);
 927                 int64_t now = av_gettime() - ist->start;
 928                 if (pts > now)
 929                     usleep(pts - now);
 930
 931                 ist->frame++;
 932             }
 933
 934 #if 0
 935             /* mpeg PTS deordering : if it is a P or I frame, the PTS
 936                is the one of the next displayed one */
 937             /* XXX: add mpeg4 too ? */
 938             if (ist->st->codec.codec_id == CODEC_ID_MPEG1VIDEO) {
 939                 if (ist->st->codec.pict_type != B_TYPE) {
 940                     int64_t tmp;
 941                     tmp = ist->last_ip_pts;
 942                     ist->last_ip_pts  = ist->frac_pts.val;
 943                     ist->frac_pts.val = tmp;
 944                 }
 945             }
 946 #endif
 947             /* if output time reached then transcode raw format,
 948                encode packets and output them */
 949             if (start_time == 0 || ist->pts >= start_time)
 950                 for(i=0;i<nb_ostreams;i++) {
 951                     int frame_size;
 952
 953                     ost = ost_table[i];
 954                     if (ost->source_index == ist_index) {
 955                         os = output_files[ost->file_index];
 956
 957 #if 0
 958                         printf("%d: got pts=%0.3f %0.3f\n", i,
 959                                (double)pkt->pts / AV_TIME_BASE,
 960                                ((double)ist->pts / AV_TIME_BASE) -
 961                                ((double)ost->st->pts.val * os->pts_num / os->pts_den));
 962 #endif
 963                         /* set the input output pts pairs */
 964                         ost->sync_ipts = (double)ist->pts / AV_TIME_BASE;
 965                         /* XXX: take into account the various fifos,
 966                            in particular for audio */
 967                         ost->sync_opts = ost->st->pts.val;
 968                         //printf("ipts=%lld sync_ipts=%f sync_opts=%lld pts.val=%lld pkt->pts=%lld\n", ist->pts, ost->sync_ipts, ost->sync_opts, ost->st->pts.val, pkt->pts);
 969
 970                         if (ost->encoding_needed) {
 971                             switch(ost->st->codec.codec_type) {
 972                             case CODEC_TYPE_AUDIO:
 973                                 do_audio_out(os, ost, ist, data_buf, data_size);
 974                                 break;
 975                             case CODEC_TYPE_VIDEO:
 976                                 /* find an audio stream for synchro */
 977                                 {
 978                                     int i;
 979                                     AVOutputStream *audio_sync, *ost1;
 980                                     audio_sync = NULL;
 981                                     for(i=0;i<nb_ostreams;i++) {
 982                                         ost1 = ost_table[i];
 983                                         if (ost1->file_index == ost->file_index &&
 984                                             ost1->st->codec.codec_type == CODEC_TYPE_AUDIO) {
 985                                             audio_sync = ost1;
 986                                             break;
 987                                         }
 988                                     }
 989
 990                                     do_video_out(os, ost, ist, &picture, &frame_size, audio_sync);
 991                                     if (do_vstats && frame_size)
 992                                         do_video_stats(os, ost, frame_size);
 993                                 }
 994                                 break;
 995                             default:
 996                                 av_abort();
 997                             }
 998                         } else {
 999                             AVFrame avframe;
1000
1001                             /* no reencoding needed : output the packet directly */
1002                             /* force the input stream PTS */
1003
1004                             memset(&avframe, 0, sizeof(AVFrame));
1005                             ost->st->codec.coded_frame= &avframe;
1006                             avframe.key_frame = pkt->flags & PKT_FLAG_KEY;
1007
1008                             av_write_frame(os, ost->index, data_buf, data_size);
1009                             ost->st->codec.frame_number++;
1010                             ost->frame_number++;
1011                         }
1012                     }
1013                 }
1014             av_free(buffer_to_free);
1015         }
1016  discard_packet:
1017     return 0;
1018  fail_decode:
1019     return -1;
1020 }
1021
1022
1023 /*
1024  * The following code is the main loop of the file converter
1025  */
1026 static int av_encode(AVFormatContext **output_files,
1027                      int nb_output_files,
1028                      AVFormatContext **input_files,
1029                      int nb_input_files,
1030                      AVStreamMap *stream_maps, int nb_stream_maps)
1031 {
1032     int ret, i, j, k, n, nb_istreams = 0, nb_ostreams = 0;
1033     AVFormatContext *is, *os;
1034     AVCodecContext *codec, *icodec;
1035     AVOutputStream *ost, **ost_table = NULL;
1036     AVInputStream *ist, **ist_table = NULL;
1037     AVInputFile *file_table;
1038     AVFormatContext *stream_no_data;
1039     int key;
1040
1041     file_table= (AVInputFile*) av_mallocz(nb_input_files * sizeof(AVInputFile));
1042     if (!file_table)
1043         goto fail;
1044
1045     /* input stream init */
1046     j = 0;
1047     for(i=0;i<nb_input_files;i++) {
1048         is = input_files[i];
1049         file_table[i].ist_index = j;
1050         file_table[i].nb_streams = is->nb_streams;
1051         j += is->nb_streams;
1052     }
1053     nb_istreams = j;
1054
1055     ist_table = av_mallocz(nb_istreams * sizeof(AVInputStream *));
1056     if (!ist_table)
1057         goto fail;
1058
1059     for(i=0;i<nb_istreams;i++) {
1060         ist = av_mallocz(sizeof(AVInputStream));
1061         if (!ist)
1062             goto fail;
1063         ist_table[i] = ist;
1064     }
1065     j = 0;
1066     for(i=0;i<nb_input_files;i++) {
1067         is = input_files[i];
1068         for(k=0;k<is->nb_streams;k++) {
1069             ist = ist_table[j++];
1070             ist->st = is->streams[k];
1071             ist->file_index = i;
1072             ist->index = k;
1073             ist->discard = 1; /* the stream is discarded by default
1074                                  (changed later) */
1075
1076             if (ist->st->codec.rate_emu) {
1077                 ist->start = av_gettime();
1078                 ist->frame = 0;
1079             }
1080         }
1081     }
1082
1083     /* output stream init */
1084     nb_ostreams = 0;
1085     for(i=0;i<nb_output_files;i++) {
1086         os = output_files[i];
1087         nb_ostreams += os->nb_streams;
1088     }
1089     if (nb_stream_maps > 0 && nb_stream_maps != nb_ostreams) {
1090         fprintf(stderr, "Number of stream maps must match number of output streams\n");
1091         exit(1);
1092     }
1093
1094     /* Sanity check the mapping args -- do the input files & streams exist? */
1095     for(i=0;i<nb_stream_maps;i++) {
1096         int fi = stream_maps[i].file_index;
1097         int si = stream_maps[i].stream_index;
1098
1099         if (fi < 0 || fi > nb_input_files - 1 ||
1100             si < 0 || si > file_table[fi].nb_streams - 1) {
1101             fprintf(stderr,"Could not find input stream #%d.%d\n", fi, si);
1102             exit(1);
1103         }
1104     }
1105
1106     ost_table = av_mallocz(sizeof(AVOutputStream *) * nb_ostreams);
1107     if (!ost_table)
1108         goto fail;
1109     for(i=0;i<nb_ostreams;i++) {
1110         ost = av_mallocz(sizeof(AVOutputStream));
1111         if (!ost)
1112             goto fail;
1113         ost_table[i] = ost;
1114     }
1115
1116     n = 0;
1117     for(k=0;k<nb_output_files;k++) {
1118         os = output_files[k];
1119         for(i=0;i<os->nb_streams;i++) {
1120             int found;
1121             ost = ost_table[n++];
1122             ost->file_index = k;
1123             ost->index = i;
1124             ost->st = os->streams[i];
1125             if (nb_stream_maps > 0) {
1126                 ost->source_index = file_table[stream_maps[n-1].file_index].ist_index +
1127                     stream_maps[n-1].stream_index;
1128
1129                 /* Sanity check that the stream types match */
1130                 if (ist_table[ost->source_index]->st->codec.codec_type != ost->st->codec.codec_type) {
1131                     fprintf(stderr, "Codec type mismatch for mapping #%d.%d -> #%d.%d\n",
1132                         stream_maps[n-1].file_index, stream_maps[n-1].stream_index,
1133                         ost->file_index, ost->index);
1134                     exit(1);
1135                 }
1136
1137             } else {
1138                 /* get corresponding input stream index : we select the first one with the right type */
1139                 found = 0;
1140                 for(j=0;j<nb_istreams;j++) {
1141                     ist = ist_table[j];
1142                     if (ist->discard &&
1143                         ist->st->codec.codec_type == ost->st->codec.codec_type) {
1144                         ost->source_index = j;
1145                         found = 1;
1146                     }
1147                 }
1148
1149                 if (!found) {
1150                     /* try again and reuse existing stream */
1151                     for(j=0;j<nb_istreams;j++) {
1152                         ist = ist_table[j];
1153                         if (ist->st->codec.codec_type == ost->st->codec.codec_type) {
1154                             ost->source_index = j;
1155                             found = 1;
1156                         }
1157                     }
1158                     if (!found) {
1159                         fprintf(stderr, "Could not find input stream matching output stream #%d.%d\n",
1160                                 ost->file_index, ost->index);
1161                         exit(1);
1162                     }
1163                 }
1164             }
1165             ist = ist_table[ost->source_index];
1166             ist->discard = 0;
1167         }
1168     }
1169
1170     /* for each output stream, we compute the right encoding parameters */
1171     for(i=0;i<nb_ostreams;i++) {
1172         ost = ost_table[i];
1173         ist = ist_table[ost->source_index];
1174
1175         codec = &ost->st->codec;
1176         icodec = &ist->st->codec;
1177
1178         if (ost->st->stream_copy) {
1179             /* if stream_copy is selected, no need to decode or encode */
1180             codec->codec_id = icodec->codec_id;
1181             codec->codec_type = icodec->codec_type;
1182             codec->codec_tag = icodec->codec_tag;
1183             codec->bit_rate = icodec->bit_rate;
1184             switch(codec->codec_type) {
1185             case CODEC_TYPE_AUDIO:
1186                 codec->sample_rate = icodec->sample_rate;
1187                 codec->channels = icodec->channels;
1188                 break;
1189             case CODEC_TYPE_VIDEO:
1190                 codec->frame_rate = icodec->frame_rate;
1191                 codec->frame_rate_base = icodec->frame_rate_base;
1192                 codec->width = icodec->width;
1193                 codec->height = icodec->height;
1194                 break;
1195             default:
1196                 av_abort();
1197             }
1198         } else {
1199             switch(codec->codec_type) {
1200             case CODEC_TYPE_AUDIO:
1201                 if (fifo_init(&ost->fifo, 2 * MAX_AUDIO_PACKET_SIZE))
1202                     goto fail;
1203
1204                 if (codec->channels == icodec->channels &&
1205                     codec->sample_rate == icodec->sample_rate) {
1206                     ost->audio_resample = 0;
1207                 } else {
1208                     if (codec->channels != icodec->channels &&
1209                         icodec->codec_id == CODEC_ID_AC3) {
1210                         /* Special case for 5:1 AC3 input */
1211                         /* and mono or stereo output      */
1212                         /* Request specific number of channels */
1213                         icodec->channels = codec->channels;
1214                         if (codec->sample_rate == icodec->sample_rate)
1215                             ost->audio_resample = 0;
1216                         else {
1217                             ost->audio_resample = 1;
1218                             ost->resample = audio_resample_init(codec->channels, icodec->channels,
1219                                                         codec->sample_rate,
1220                                                         icodec->sample_rate);
1221                             if(!ost->resample)
1222                               {
1223                                 printf("Can't resample.  Aborting.\n");
1224                                 av_abort();
1225                               }
1226                         }
1227                         /* Request specific number of channels */
1228                         icodec->channels = codec->channels;
1229                     } else {
1230                         ost->audio_resample = 1;
1231                         ost->resample = audio_resample_init(codec->channels, icodec->channels,
1232                                                         codec->sample_rate,
1233                                                         icodec->sample_rate);
1234                         if(!ost->resample)
1235                           {
1236                             printf("Can't resample.  Aborting.\n");
1237                             av_abort();
1238                           }
1239                     }
1240                 }
1241                 ist->decoding_needed = 1;
1242                 ost->encoding_needed = 1;
1243                 break;
1244             case CODEC_TYPE_VIDEO:
1245                 if (codec->width == icodec->width &&
1246                     codec->height == icodec->height &&
1247                     frame_topBand == 0 &&
1248                     frame_bottomBand == 0 &&
1249                     frame_leftBand == 0 &&
1250                     frame_rightBand == 0)
1251                 {
1252                     ost->video_resample = 0;
1253                     ost->video_crop = 0;
1254                 } else if ((codec->width == icodec->width -
1255                                 (frame_leftBand + frame_rightBand)) &&
1256                         (codec->height == icodec->height -
1257                                 (frame_topBand  + frame_bottomBand)))
1258                 {
1259                     ost->video_resample = 0;
1260                     ost->video_crop = 1;
1261                     ost->topBand = frame_topBand;
1262                     ost->leftBand = frame_leftBand;
1263                 } else {
1264                     uint8_t *buf;
1265                     ost->video_resample = 1;
1266                     ost->video_crop = 0; // cropping is handled as part of resample
1267                     buf = av_malloc((codec->width * codec->height * 3) / 2);
1268                     if (!buf)
1269                         goto fail;
1270                     ost->pict_tmp.data[0] = buf;
1271                     ost->pict_tmp.data[1] = ost->pict_tmp.data[0] + (codec->width * codec->height);
1272                     ost->pict_tmp.data[2] = ost->pict_tmp.data[1] + (codec->width * codec->height) / 4;
1273                     ost->pict_tmp.linesize[0] = codec->width;
1274                     ost->pict_tmp.linesize[1] = codec->width / 2;
1275                     ost->pict_tmp.linesize[2] = codec->width / 2;
1276
1277                     ost->img_resample_ctx = img_resample_full_init(
1278                                       ost->st->codec.width, ost->st->codec.height,
1279                                       ist->st->codec.width, ist->st->codec.height,
1280                                       frame_topBand, frame_bottomBand,
1281                                       frame_leftBand, frame_rightBand);
1282                 }
1283                 ost->encoding_needed = 1;
1284                 ist->decoding_needed = 1;
1285                 break;
1286             default:
1287                 av_abort();
1288             }
1289             /* two pass mode */
1290             if (ost->encoding_needed &&
1291                 (codec->flags & (CODEC_FLAG_PASS1 | CODEC_FLAG_PASS2))) {
1292                 char logfilename[1024];
1293                 FILE *f;
1294                 int size;
1295                 char *logbuffer;
1296
1297                 snprintf(logfilename, sizeof(logfilename), "%s-%d.log",
1298                          pass_logfilename ?
1299                          pass_logfilename : DEFAULT_PASS_LOGFILENAME, i);
1300                 if (codec->flags & CODEC_FLAG_PASS1) {
1301                     f = fopen(logfilename, "w");
1302                     if (!f) {
1303                         perror(logfilename);
1304                         exit(1);
1305                     }
1306                     ost->logfile = f;
1307                 } else {
1308                     /* read the log file */
1309                     f = fopen(logfilename, "r");
1310                     if (!f) {
1311                         perror(logfilename);
1312                         exit(1);
1313                     }
1314                     fseek(f, 0, SEEK_END);
1315                     size = ftell(f);
1316                     fseek(f, 0, SEEK_SET);
1317                     logbuffer = av_malloc(size + 1);
1318                     if (!logbuffer) {
1319                         fprintf(stderr, "Could not allocate log buffer\n");
1320                         exit(1);
1321                     }
1322                     fread(logbuffer, 1, size, f);
1323                     fclose(f);
1324                     logbuffer[size] = '\0';
1325                     codec->stats_in = logbuffer;
1326                 }
1327             }
1328         }
1329     }
1330
1331     /* dump the file output parameters - cannot be done before in case
1332        of stream copy */
1333     for(i=0;i<nb_output_files;i++) {
1334         dump_format(output_files[i], i, output_files[i]->filename, 1);
1335     }
1336
1337     /* dump the stream mapping */
1338     fprintf(stderr, "Stream mapping:\n");
1339     for(i=0;i<nb_ostreams;i++) {
1340         ost = ost_table[i];
1341         fprintf(stderr, "  Stream #%d.%d -> #%d.%d\n",
1342                 ist_table[ost->source_index]->file_index,
1343                 ist_table[ost->source_index]->index,
1344                 ost->file_index,
1345                 ost->index);
1346     }
1347
1348     /* open each encoder */
1349     for(i=0;i<nb_ostreams;i++) {
1350         ost = ost_table[i];
1351         if (ost->encoding_needed) {
1352             AVCodec *codec;
1353             codec = avcodec_find_encoder(ost->st->codec.codec_id);
1354             if (!codec) {
1355                 fprintf(stderr, "Unsupported codec for output stream #%d.%d\n",
1356                         ost->file_index, ost->index);
1357                 exit(1);
1358             }
1359             if (avcodec_open(&ost->st->codec, codec) < 0) {
1360                 fprintf(stderr, "Error while opening codec for stream #%d.%d - maybe incorrect parameters such as bit_rate, rate, width or height\n",
1361                         ost->file_index, ost->index);
1362                 exit(1);
1363             }
1364         }
1365     }
1366
1367     /* open each decoder */
1368     for(i=0;i<nb_istreams;i++) {
1369         ist = ist_table[i];
1370         if (ist->decoding_needed) {
1371             AVCodec *codec;
1372             codec = avcodec_find_decoder(ist->st->codec.codec_id);
1373             if (!codec) {
1374                 fprintf(stderr, "Unsupported codec (id=%d) for input stream #%d.%d\n",
1375                         ist->st->codec.codec_id, ist->file_index, ist->index);
1376                 exit(1);
1377             }
1378             if (avcodec_open(&ist->st->codec, codec) < 0) {
1379                 fprintf(stderr, "Error while opening codec for input stream #%d.%d\n",
1380                         ist->file_index, ist->index);
1381                 exit(1);
1382             }
1383             //if (ist->st->codec.codec_type == CODEC_TYPE_VIDEO)
1384             //    ist->st->codec.flags |= CODEC_FLAG_REPEAT_FIELD;
1385         }
1386     }
1387
1388     /* init pts */
1389     for(i=0;i<nb_istreams;i++) {
1390         ist = ist_table[i];
1391         is = input_files[ist->file_index];
1392         ist->pts = 0;
1393         ist->next_pts = 0;
1394     }
1395
1396     /* compute buffer size max (should use a complete heuristic) */
1397     for(i=0;i<nb_input_files;i++) {
1398         file_table[i].buffer_size_max = 2048;
1399     }
1400
1401     /* open files and write file headers */
1402     for(i=0;i<nb_output_files;i++) {
1403         os = output_files[i];
1404         if (av_write_header(os) < 0) {
1405             fprintf(stderr, "Could not write header for output file #%d (incorrect codec paramters ?)\n", i);
1406             ret = -EINVAL;
1407             goto fail;
1408         }
1409     }
1410
1411 #ifndef CONFIG_WIN32
1412     if ( !using_stdin )
1413         fprintf(stderr, "Press [q] to stop encoding\n");
1414 #endif
1415     term_init();
1416
1417     stream_no_data = 0;
1418     key = -1;
1419
1420     for(; received_sigterm == 0;) {
1421         int file_index, ist_index;
1422         AVPacket pkt;
1423         double pts_min;
1424
1425     redo:
1426         /* if 'q' pressed, exits */
1427         if (!using_stdin) {
1428             /* read_key() returns 0 on EOF */
1429             key = read_key();
1430             if (key == 'q')
1431                 break;
1432         }
1433
1434         /* select the stream that we must read now by looking at the
1435            smallest output pts */
1436         file_index = -1;
1437         pts_min = 1e10;
1438         for(i=0;i<nb_ostreams;i++) {
1439             double pts;
1440             ost = ost_table[i];
1441             os = output_files[ost->file_index];
1442             ist = ist_table[ost->source_index];
1443             pts = (double)ost->st->pts.val * os->pts_num / os->pts_den;
1444             if (!file_table[ist->file_index].eof_reached &&
1445                 pts < pts_min) {
1446                 pts_min = pts;
1447                 file_index = ist->file_index;
1448             }
1449         }
1450         /* if none, if is finished */
1451         if (file_index < 0) {
1452             break;
1453         }
1454
1455         /* finish if recording time exhausted */
1456         if (recording_time > 0 && pts_min >= (recording_time / 1000000.0))
1457             break;
1458
1459         /* read a frame from it and output it in the fifo */
1460         is = input_files[file_index];
1461         if (av_read_frame(is, &pkt) < 0) {
1462             file_table[file_index].eof_reached = 1;
1463             continue;
1464         }
1465
1466         if (!pkt.size) {
1467             stream_no_data = is;
1468         } else {
1469             stream_no_data = 0;
1470         }
1471         if (do_pkt_dump) {
1472             av_pkt_dump(stdout, &pkt, do_hex_dump);
1473         }
1474         /* the following test is needed in case new streams appear
1475            dynamically in stream : we ignore them */
1476         if (pkt.stream_index >= file_table[file_index].nb_streams)
1477             goto discard_packet;
1478         ist_index = file_table[file_index].ist_index + pkt.stream_index;
1479         ist = ist_table[ist_index];
1480         if (ist->discard)
1481             goto discard_packet;
1482
1483         //fprintf(stderr,"read #%d.%d size=%d\n", ist->file_index, ist->index, pkt.size);
1484         if (output_packet(ist, ist_index, ost_table, nb_ostreams, &pkt) < 0) {
1485             fprintf(stderr, "Error while decoding stream #%d.%d\n",
1486                     ist->file_index, ist->index);
1487             av_free_packet(&pkt);
1488             goto redo;
1489         }
1490
1491     discard_packet:
1492         av_free_packet(&pkt);
1493
1494         /* dump report by using the output first video and audio streams */
1495         print_report(output_files, ost_table, nb_ostreams, 0);
1496     }
1497
1498     /* at the end of stream, we must flush the decoder buffers */
1499     for(i=0;i<nb_istreams;i++) {
1500         ist = ist_table[i];
1501         if (ist->decoding_needed) {
1502             output_packet(ist, i, ost_table, nb_ostreams, NULL);
1503         }
1504     }
1505
1506     term_exit();
1507
1508     /* dump report by using the first video and audio streams */
1509     print_report(output_files, ost_table, nb_ostreams, 1);
1510
1511     /* close each encoder */
1512     for(i=0;i<nb_ostreams;i++) {
1513         ost = ost_table[i];
1514         if (ost->encoding_needed) {
1515             av_freep(&ost->st->codec.stats_in);
1516             avcodec_close(&ost->st->codec);
1517         }
1518     }
1519
1520     /* close each decoder */
1521     for(i=0;i<nb_istreams;i++) {
1522         ist = ist_table[i];
1523         if (ist->decoding_needed) {
1524             avcodec_close(&ist->st->codec);
1525         }
1526     }
1527
1528
1529     /* write the trailer if needed and close file */
1530     for(i=0;i<nb_output_files;i++) {
1531         os = output_files[i];
1532         av_write_trailer(os);
1533     }
1534     /* finished ! */
1535
1536     ret = 0;
1537  fail1:
1538     av_free(file_table);
1539
1540     if (ist_table) {
1541         for(i=0;i<nb_istreams;i++) {
1542             ist = ist_table[i];
1543             av_free(ist);
1544         }
1545         av_free(ist_table);
1546     }
1547     if (ost_table) {
1548         for(i=0;i<nb_ostreams;i++) {
1549             ost = ost_table[i];
1550             if (ost) {
1551                 if (ost->logfile) {
1552                     fclose(ost->logfile);
1553                     ost->logfile = NULL;
1554                 }
1555                 fifo_free(&ost->fifo); /* works even if fifo is not
1556                                           initialized but set to zero */
1557                 av_free(ost->pict_tmp.data[0]);
1558                 if (ost->video_resample)
1559                     img_resample_close(ost->img_resample_ctx);
1560                 if (ost->audio_resample)
1561                     audio_resample_close(ost->resample);
1562                 av_free(ost);
1563             }
1564         }
1565         av_free(ost_table);
1566     }
1567     return ret;
1568  fail:
1569     ret = -ENOMEM;
1570     goto fail1;
1571 }
1572
1573 #if 0
1574 int file_read(const char *filename)
1575 {
1576     URLContext *h;
1577     unsigned char buffer[1024];
1578     int len, i;
1579
1580     if (url_open(&h, filename, O_RDONLY) < 0) {
1581         printf("could not open '%s'\n", filename);
1582         return -1;
1583     }
1584     for(;;) {
1585         len = url_read(h, buffer, sizeof(buffer));
1586         if (len <= 0)
1587             break;
1588         for(i=0;i<len;i++) putchar(buffer[i]);
1589     }
1590     url_close(h);
1591     return 0;
1592 }
1593 #endif
1594
1595 static void opt_image_format(const char *arg)
1596 {
1597     AVImageFormat *f;
1598
1599     for(f = first_image_format; f != NULL; f = f->next) {
1600         if (!strcmp(arg, f->name))
1601             break;
1602     }
1603     if (!f) {
1604         fprintf(stderr, "Unknown image format: '%s'\n", arg);
1605         exit(1);
1606     }
1607     image_format = f;
1608 }
1609
1610 static void opt_format(const char *arg)
1611 {
1612     /* compatibility stuff for pgmyuv */
1613     if (!strcmp(arg, "pgmyuv")) {
1614         opt_image_format(arg);
1615         arg = "image";
1616     }
1617
1618     file_iformat = av_find_input_format(arg);
1619     file_oformat = guess_format(arg, NULL, NULL);
1620     if (!file_iformat && !file_oformat) {
1621         fprintf(stderr, "Unknown input or output format: %s\n", arg);
1622         exit(1);
1623     }
1624 }
1625
1626 static void opt_video_bitrate(const char *arg)
1627 {
1628     video_bit_rate = atoi(arg) * 1000;
1629 }
1630
1631 static void opt_video_bitrate_tolerance(const char *arg)
1632 {
1633     video_bit_rate_tolerance = atoi(arg) * 1000;
1634 }
1635
1636 static void opt_video_bitrate_max(const char *arg)
1637 {
1638     video_rc_max_rate = atoi(arg) * 1000;
1639 }
1640
1641 static void opt_video_bitrate_min(const char *arg)
1642 {
1643     video_rc_min_rate = atoi(arg) * 1000;
1644 }
1645
1646 static void opt_video_buffer_size(const char *arg)
1647 {
1648     video_rc_buffer_size = atoi(arg) * 8*1024;
1649 }
1650
1651 static void opt_video_rc_eq(char *arg)
1652 {
1653     video_rc_eq = arg;
1654 }
1655
1656 static void opt_video_rc_override_string(char *arg)
1657 {
1658     video_rc_override_string = arg;
1659 }
1660
1661
1662 static void opt_workaround_bugs(const char *arg)
1663 {
1664     workaround_bugs = atoi(arg);
1665 }
1666
1667 static void opt_dct_algo(const char *arg)
1668 {
1669     dct_algo = atoi(arg);
1670 }
1671
1672 static void opt_idct_algo(const char *arg)
1673 {
1674     idct_algo = atoi(arg);
1675 }
1676
1677
1678 static void opt_error_resilience(const char *arg)
1679 {
1680     error_resilience = atoi(arg);
1681 }
1682
1683 static void opt_error_concealment(const char *arg)
1684 {
1685     error_concealment = atoi(arg);
1686 }
1687
1688 static void opt_debug(const char *arg)
1689 {
1690     debug = atoi(arg);
1691 }
1692
1693 static void opt_vismv(const char *arg)
1694 {
1695     debug_mv = atoi(arg);
1696 }
1697
1698 static void opt_verbose(const char *arg)
1699 {
1700     verbose = atoi(arg);
1701 }
1702
1703 static void opt_frame_rate(const char *arg)
1704 {
1705     if (parse_frame_rate(&frame_rate, &frame_rate_base, arg) < 0) {
1706         fprintf(stderr, "Incorrect frame rate\n");
1707         exit(1);
1708     }
1709 }
1710
1711 static void opt_frame_crop_top(const char *arg)
1712 {
1713     frame_topBand = atoi(arg);
1714     if (frame_topBand < 0) {
1715         fprintf(stderr, "Incorrect top crop size\n");
1716         exit(1);
1717     }
1718     if ((frame_topBand % 2) != 0) {
1719         fprintf(stderr, "Top crop size must be a multiple of 2\n");
1720         exit(1);
1721     }
1722     if ((frame_topBand) >= frame_height){
1723         fprintf(stderr, "Vertical crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");
1724         exit(1);
1725     }
1726     frame_height -= frame_topBand;
1727 }
1728
1729 static void opt_frame_crop_bottom(const char *arg)
1730 {
1731     frame_bottomBand = atoi(arg);
1732     if (frame_bottomBand < 0) {
1733         fprintf(stderr, "Incorrect bottom crop size\n");
1734         exit(1);
1735     }
1736     if ((frame_bottomBand % 2) != 0) {
1737         fprintf(stderr, "Bottom crop size must be a multiple of 2\n");
1738         exit(1);
1739     }
1740     if ((frame_bottomBand) >= frame_height){
1741         fprintf(stderr, "Vertical crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");
1742         exit(1);
1743     }
1744     frame_height -= frame_bottomBand;
1745 }
1746
1747 static void opt_frame_crop_left(const char *arg)
1748 {
1749     frame_leftBand = atoi(arg);
1750     if (frame_leftBand < 0) {
1751         fprintf(stderr, "Incorrect left crop size\n");
1752         exit(1);
1753     }
1754     if ((frame_leftBand % 2) != 0) {
1755         fprintf(stderr, "Left crop size must be a multiple of 2\n");
1756         exit(1);
1757     }
1758     if ((frame_leftBand) >= frame_width){
1759         fprintf(stderr, "Horizontal crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");
1760         exit(1);
1761     }
1762     frame_width -= frame_leftBand;
1763 }
1764
1765 static void opt_frame_crop_right(const char *arg)
1766 {
1767     frame_rightBand = atoi(arg);
1768     if (frame_rightBand < 0) {
1769         fprintf(stderr, "Incorrect right crop size\n");
1770         exit(1);
1771     }
1772     if ((frame_rightBand % 2) != 0) {
1773         fprintf(stderr, "Right crop size must be a multiple of 2\n");
1774         exit(1);
1775     }
1776     if ((frame_rightBand) >= frame_width){
1777         fprintf(stderr, "Horizontal crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");
1778         exit(1);
1779     }
1780     frame_width -= frame_rightBand;
1781 }
1782
1783 static void opt_frame_size(const char *arg)
1784 {
1785     if (parse_image_size(&frame_width, &frame_height, arg) < 0) {
1786         fprintf(stderr, "Incorrect frame size\n");
1787         exit(1);
1788     }
1789     if ((frame_width % 2) != 0 || (frame_height % 2) != 0) {
1790         fprintf(stderr, "Frame size must be a multiple of 2\n");
1791         exit(1);
1792     }
1793 }
1794
1795 static void opt_frame_pix_fmt(const char *arg)
1796 {
1797     frame_pix_fmt = avcodec_get_pix_fmt(arg);
1798 }
1799
1800 static void opt_frame_aspect_ratio(const char *arg)
1801 {
1802     int x = 0, y = 0;
1803     double ar = 0;
1804     const char *p;
1805
1806     p = strchr(arg, ':');
1807     if (p) {
1808         x = strtol(arg, (char **)&arg, 10);
1809         if (arg == p)
1810             y = strtol(arg+1, (char **)&arg, 10);
1811         if (x > 0 && y > 0)
1812             ar = (double)x / (double)y;
1813     } else
1814         ar = strtod(arg, (char **)&arg);
1815
1816     if (!ar) {
1817         fprintf(stderr, "Incorrect aspect ratio specification.\n");
1818         exit(1);
1819     }
1820     frame_aspect_ratio = ar;
1821 }
1822
1823 static void opt_gop_size(const char *arg)
1824 {
1825     gop_size = atoi(arg);
1826 }
1827
1828 static void opt_b_frames(const char *arg)
1829 {
1830     b_frames = atoi(arg);
1831     if (b_frames > FF_MAX_B_FRAMES) {
1832         fprintf(stderr, "\nCannot have more than %d B frames, increase FF_MAX_B_FRAMES.\n", FF_MAX_B_FRAMES);
1833         exit(1);
1834     } else if (b_frames < 1) {
1835         fprintf(stderr, "\nNumber of B frames must be higher than 0\n");
1836         exit(1);
1837     }
1838 }
1839
1840 static void opt_mb_decision(const char *arg)
1841 {
1842     mb_decision = atoi(arg);
1843 }
1844
1845 static void opt_mb_cmp(const char *arg)
1846 {
1847     mb_cmp = atoi(arg);
1848 }
1849
1850 static void opt_ildct_cmp(const char *arg)
1851 {
1852     ildct_cmp = atoi(arg);
1853 }
1854
1855 static void opt_sub_cmp(const char *arg)
1856 {
1857     sub_cmp = atoi(arg);
1858 }
1859
1860 static void opt_cmp(const char *arg)
1861 {
1862     cmp = atoi(arg);
1863 }
1864
1865 static void opt_qscale(const char *arg)
1866 {
1867     video_qscale = atof(arg);
1868     if (video_qscale < 0.01 ||
1869         video_qscale > 255) {
1870         fprintf(stderr, "qscale must be >= 0.01 and <= 255\n");
1871         exit(1);
1872     }
1873 }
1874
1875 static void opt_qmin(const char *arg)
1876 {
1877     video_qmin = atoi(arg);
1878     if (video_qmin < 0 ||
1879         video_qmin > 31) {
1880         fprintf(stderr, "qmin must be >= 1 and <= 31\n");
1881         exit(1);
1882     }
1883 }
1884
1885 static void opt_qmax(const char *arg)
1886 {
1887     video_qmax = atoi(arg);
1888     if (video_qmax < 0 ||
1889         video_qmax > 31) {
1890         fprintf(stderr, "qmax must be >= 1 and <= 31\n");
1891         exit(1);
1892     }
1893 }
1894
1895 static void opt_mb_qmin(const char *arg)
1896 {
1897     video_mb_qmin = atoi(arg);
1898     if (video_mb_qmin < 0 ||
1899         video_mb_qmin > 31) {
1900         fprintf(stderr, "qmin must be >= 1 and <= 31\n");
1901         exit(1);
1902     }
1903 }
1904
1905 static void opt_mb_qmax(const char *arg)
1906 {
1907     video_mb_qmax = atoi(arg);
1908     if (video_mb_qmax < 0 ||
1909         video_mb_qmax > 31) {
1910         fprintf(stderr, "qmax must be >= 1 and <= 31\n");
1911         exit(1);
1912     }
1913 }
1914
1915 static void opt_qdiff(const char *arg)
1916 {
1917     video_qdiff = atoi(arg);
1918     if (video_qdiff < 0 ||
1919         video_qdiff > 31) {
1920         fprintf(stderr, "qdiff must be >= 1 and <= 31\n");
1921         exit(1);
1922     }
1923 }
1924
1925 static void opt_qblur(const char *arg)
1926 {
1927     video_qblur = atof(arg);
1928 }
1929
1930 static void opt_qcomp(const char *arg)
1931 {
1932     video_qcomp = atof(arg);
1933 }
1934
1935 static void opt_rc_initial_cplx(const char *arg)
1936 {
1937     video_rc_initial_cplx = atof(arg);
1938 }
1939 static void opt_b_qfactor(const char *arg)
1940 {
1941     video_b_qfactor = atof(arg);
1942 }
1943 static void opt_i_qfactor(const char *arg)
1944 {
1945     video_i_qfactor = atof(arg);
1946 }
1947 static void opt_b_qoffset(const char *arg)
1948 {
1949     video_b_qoffset = atof(arg);
1950 }
1951 static void opt_i_qoffset(const char *arg)
1952 {
1953     video_i_qoffset = atof(arg);
1954 }
1955
1956 static void opt_ibias(const char *arg)
1957 {
1958     video_intra_quant_bias = atoi(arg);
1959 }
1960 static void opt_pbias(const char *arg)
1961 {
1962     video_inter_quant_bias = atoi(arg);
1963 }
1964
1965 static void opt_packet_size(const char *arg)
1966 {
1967     packet_size= atoi(arg);
1968 }
1969
1970 static void opt_error_rate(const char *arg)
1971 {
1972     error_rate= atoi(arg);
1973 }
1974
1975 static void opt_strict(const char *arg)
1976 {
1977     strict= atoi(arg);
1978 }
1979
1980 static void opt_top_field_first(const char *arg)
1981 {
1982     top_field_first= atoi(arg);
1983 }
1984
1985 static void opt_noise_reduction(const char *arg)
1986 {
1987     noise_reduction= atoi(arg);
1988 }
1989
1990 static void opt_sc_threshold(const char *arg)
1991 {
1992     sc_threshold= atoi(arg);
1993 }
1994
1995 static void opt_audio_bitrate(const char *arg)
1996 {
1997     audio_bit_rate = atoi(arg) * 1000;
1998 }
1999
2000 static void opt_audio_rate(const char *arg)
2001 {
2002     audio_sample_rate = atoi(arg);
2003 }
2004
2005 static void opt_audio_channels(const char *arg)
2006 {
2007     audio_channels = atoi(arg);
2008 }
2009
2010 static void opt_video_device(const char *arg)
2011 {
2012     video_device = av_strdup(arg);
2013 }
2014
2015 static void opt_video_channel(const char *arg)
2016 {
2017     video_channel = strtol(arg, NULL, 0);
2018 }
2019
2020 static void opt_video_standard(const char *arg)
2021 {
2022     video_standard = av_strdup(arg);
2023 }
2024
2025 static void opt_audio_device(const char *arg)
2026 {
2027     audio_device = av_strdup(arg);
2028 }
2029
2030 static void opt_dv1394(const char *arg)
2031 {
2032     video_grab_format = "dv1394";
2033     audio_grab_format = NULL;
2034 }
2035
2036 static void opt_audio_codec(const char *arg)
2037 {
2038     AVCodec *p;
2039
2040     if (!strcmp(arg, "copy")) {
2041         audio_stream_copy = 1;
2042     } else {
2043         p = first_avcodec;
2044         while (p) {
2045             if (!strcmp(p->name, arg) && p->type == CODEC_TYPE_AUDIO)
2046                 break;
2047             p = p->next;
2048         }
2049         if (p == NULL) {
2050             fprintf(stderr, "Unknown audio codec '%s'\n", arg);
2051             exit(1);
2052         } else {
2053             audio_codec_id = p->id;
2054         }
2055     }
2056 }
2057
2058 static void add_frame_hooker(const char *arg)
2059 {
2060     int argc = 0;
2061     char *argv[64];
2062     int i;
2063     char *args = av_strdup(arg);
2064
2065     using_vhook = 1;
2066
2067     argv[0] = strtok(args, " ");
2068     while (argc < 62 && (argv[++argc] = strtok(NULL, " "))) {
2069     }
2070
2071     i = frame_hook_add(argc, argv);
2072
2073     if (i != 0) {
2074         fprintf(stderr, "Failed to add video hook function: %s\n", arg);
2075         exit(1);
2076     }
2077 }
2078
2079 const char *motion_str[] = {
2080     "zero",
2081     "full",
2082     "log",
2083     "phods",
2084     "epzs",
2085     "x1",
2086     NULL,
2087 };
2088
2089 static void opt_motion_estimation(const char *arg)
2090 {
2091     const char **p;
2092     p = motion_str;
2093     for(;;) {
2094         if (!*p) {
2095             fprintf(stderr, "Unknown motion estimation method '%s'\n", arg);
2096             exit(1);
2097         }
2098         if (!strcmp(*p, arg))
2099             break;
2100         p++;
2101     }
2102     me_method = (p - motion_str) + 1;
2103 }
2104
2105 static void opt_video_codec(const char *arg)
2106 {
2107     AVCodec *p;
2108
2109     if (!strcmp(arg, "copy")) {
2110         video_stream_copy = 1;
2111     } else {
2112         p = first_avcodec;
2113         while (p) {
2114             if (!strcmp(p->name, arg) && p->type == CODEC_TYPE_VIDEO)
2115                 break;
2116             p = p->next;
2117         }
2118         if (p == NULL) {
2119             fprintf(stderr, "Unknown video codec '%s'\n", arg);
2120             exit(1);
2121         } else {
2122             video_codec_id = p->id;
2123         }
2124     }
2125 }
2126
2127 static void opt_map(const char *arg)
2128 {
2129     AVStreamMap *m;
2130     const char *p;
2131
2132     p = arg;
2133     m = &stream_maps[nb_stream_maps++];
2134
2135     m->file_index = strtol(arg, (char **)&p, 0);
2136     if (*p)
2137         p++;
2138
2139     m->stream_index = strtol(p, (char **)&p, 0);
2140 }
2141
2142 static void opt_recording_time(const char *arg)
2143 {
2144     recording_time = parse_date(arg, 1);
2145 }
2146
2147 static void opt_start_time(const char *arg)
2148 {
2149     start_time = parse_date(arg, 1);
2150 }
2151
2152 static void opt_input_file(const char *filename)
2153 {
2154     AVFormatContext *ic;
2155     AVFormatParameters params, *ap = &params;
2156     int err, i, ret, rfps, rfps_base;
2157
2158     if (!strcmp(filename, "-"))
2159         filename = "pipe:";
2160
2161     using_stdin |= !strcmp(filename, "pipe:" ) ||
2162                    !strcmp( filename, "/dev/stdin" );
2163
2164     /* get default parameters from command line */
2165     memset(ap, 0, sizeof(*ap));
2166     ap->sample_rate = audio_sample_rate;
2167     ap->channels = audio_channels;
2168     ap->frame_rate = frame_rate;
2169     ap->frame_rate_base = frame_rate_base;
2170     ap->width = frame_width;
2171     ap->height = frame_height;
2172     ap->image_format = image_format;
2173     ap->pix_fmt = frame_pix_fmt;
2174
2175     /* open the input file with generic libav function */
2176     err = av_open_input_file(&ic, filename, file_iformat, 0, ap);
2177     if (err < 0) {
2178         print_error(filename, err);
2179         exit(1);
2180     }
2181
2182     /* If not enough info to get the stream parameters, we decode the
2183        first frames to get it. (used in mpeg case for example) */
2184     ret = av_find_stream_info(ic);
2185     if (ret < 0) {
2186         fprintf(stderr, "%s: could not find codec parameters\n", filename);
2187         exit(1);
2188     }
2189
2190     /* if seeking requested, we execute it */
2191     if (start_time != 0) {
2192         int64_t timestamp;
2193
2194         timestamp = start_time;
2195         /* add the stream start time */
2196         if (ic->start_time != AV_NOPTS_VALUE)
2197             timestamp += ic->start_time;
2198         ret = av_seek_frame(ic, -1, timestamp);
2199         if (ret < 0) {
2200             fprintf(stderr, "%s: could not seek to position %0.3f\n",
2201                     filename, (double)timestamp / AV_TIME_BASE);
2202         }
2203         /* reset seek info */
2204         start_time = 0;
2205     }
2206
2207     /* update the current parameters so that they match the one of the input stream */
2208     for(i=0;i<ic->nb_streams;i++) {
2209         AVCodecContext *enc = &ic->streams[i]->codec;
2210         switch(enc->codec_type) {
2211         case CODEC_TYPE_AUDIO:
2212             //fprintf(stderr, "\nInput Audio channels: %d", enc->channels);
2213             audio_channels = enc->channels;
2214             audio_sample_rate = enc->sample_rate;
2215             break;
2216         case CODEC_TYPE_VIDEO:
2217             frame_height = enc->height;
2218             frame_width = enc->width;
2219             frame_aspect_ratio = av_q2d(enc->sample_aspect_ratio) * enc->width / enc->height;
2220             frame_pix_fmt = enc->pix_fmt;
2221             rfps      = ic->streams[i]->r_frame_rate;
2222             rfps_base = ic->streams[i]->r_frame_rate_base;
2223             enc->workaround_bugs = workaround_bugs;
2224             enc->error_resilience = error_resilience;
2225             enc->error_concealment = error_concealment;
2226             enc->idct_algo = idct_algo;
2227             enc->debug = debug;
2228             enc->debug_mv = debug_mv;
2229             if(bitexact)
2230                 enc->flags|= CODEC_FLAG_BITEXACT;
2231
2232             assert(enc->frame_rate_base == rfps_base); // should be true for now
2233             if (enc->frame_rate != rfps) {
2234                 fprintf(stderr,"\nSeems that stream %d comes from film source: %2.2f->%2.2f\n",
2235                     i, (float)enc->frame_rate / enc->frame_rate_base,
2236                     (float)rfps / rfps_base);
2237             }
2238             /* update the current frame rate to match the stream frame rate */
2239             frame_rate      = rfps;
2240             frame_rate_base = rfps_base;
2241
2242             enc->rate_emu = rate_emu;
2243             break;
2244         case CODEC_TYPE_DATA:
2245             break;
2246         default:
2247             av_abort();
2248         }
2249     }
2250
2251     input_files[nb_input_files] = ic;
2252     /* dump the file content */
2253     dump_format(ic, nb_input_files, filename, 0);
2254     nb_input_files++;
2255     file_iformat = NULL;
2256     file_oformat = NULL;
2257     image_format = NULL;
2258
2259     rate_emu = 0;
2260 }
2261
2262 static void check_audio_video_inputs(int *has_video_ptr, int *has_audio_ptr)
2263 {
2264     int has_video, has_audio, i, j;
2265     AVFormatContext *ic;
2266
2267     has_video = 0;
2268     has_audio = 0;
2269     for(j=0;j<nb_input_files;j++) {
2270         ic = input_files[j];
2271         for(i=0;i<ic->nb_streams;i++) {
2272             AVCodecContext *enc = &ic->streams[i]->codec;
2273             switch(enc->codec_type) {
2274             case CODEC_TYPE_AUDIO:
2275                 has_audio = 1;
2276                 break;
2277             case CODEC_TYPE_VIDEO:
2278                 has_video = 1;
2279                 break;
2280             default:
2281                 av_abort();
2282             }
2283         }
2284     }
2285     *has_video_ptr = has_video;
2286     *has_audio_ptr = has_audio;
2287 }
2288
2289 static void opt_output_file(const char *filename)
2290 {
2291     AVStream *st;
2292     AVFormatContext *oc;
2293     int use_video, use_audio, nb_streams, input_has_video, input_has_audio;
2294     int codec_id;
2295     AVFormatParameters params, *ap = &params;
2296
2297     if (!strcmp(filename, "-"))
2298         filename = "pipe:";
2299
2300     oc = av_mallocz(sizeof(AVFormatContext));
2301
2302     if (!file_oformat) {
2303         file_oformat = guess_format(NULL, filename, NULL);
2304         if (!file_oformat) {
2305             fprintf(stderr, "Unable for find a suitable output format for '%s'\n",
2306                     filename);
2307             exit(1);
2308         }
2309     }
2310
2311     oc->oformat = file_oformat;
2312
2313     if (!strcmp(file_oformat->name, "ffm") &&
2314         strstart(filename, "http:", NULL)) {
2315         /* special case for files sent to ffserver: we get the stream
2316            parameters from ffserver */
2317         if (read_ffserver_streams(oc, filename) < 0) {
2318             fprintf(stderr, "Could not read stream parameters from '%s'\n", filename);
2319             exit(1);
2320         }
2321     } else {
2322         use_video = file_oformat->video_codec != CODEC_ID_NONE;
2323         use_audio = file_oformat->audio_codec != CODEC_ID_NONE;
2324
2325         /* disable if no corresponding type found and at least one
2326            input file */
2327         if (nb_input_files > 0) {
2328             check_audio_video_inputs(&input_has_video, &input_has_audio);
2329             if (!input_has_video)
2330                 use_video = 0;
2331             if (!input_has_audio)
2332                 use_audio = 0;
2333         }
2334
2335         /* manual disable */
2336         if (audio_disable) {
2337             use_audio = 0;
2338         }
2339         if (video_disable) {
2340             use_video = 0;
2341         }
2342
2343         nb_streams = 0;
2344         if (use_video) {
2345             AVCodecContext *video_enc;
2346
2347             st = av_mallocz(sizeof(AVStream));
2348             if (!st) {
2349                 fprintf(stderr, "Could not alloc stream\n");
2350                 exit(1);
2351             }
2352             avcodec_get_context_defaults(&st->codec);
2353
2354             video_enc = &st->codec;
2355
2356             if(!strcmp(file_oformat->name, "mp4") || !strcmp(file_oformat->name, "mov") || !strcmp(file_oformat->name, "3gp"))
2357                 video_enc->flags |= CODEC_FLAG_GLOBAL_HEADER;
2358             if (video_stream_copy) {
2359                 st->stream_copy = 1;
2360                 video_enc->codec_type = CODEC_TYPE_VIDEO;
2361             } else {
2362                 char *p;
2363                 int i;
2364
2365                 codec_id = file_oformat->video_codec;
2366                 if (video_codec_id != CODEC_ID_NONE)
2367                     codec_id = video_codec_id;
2368
2369                 video_enc->codec_id = codec_id;
2370
2371                 video_enc->bit_rate = video_bit_rate;
2372                 video_enc->bit_rate_tolerance = video_bit_rate_tolerance;
2373                 video_enc->frame_rate = frame_rate;
2374                 video_enc->frame_rate_base = frame_rate_base;
2375
2376                 video_enc->width = frame_width;
2377                 video_enc->height = frame_height;
2378                 video_enc->sample_aspect_ratio = av_d2q(frame_aspect_ratio*frame_height/frame_width, 255);
2379                 video_enc->pix_fmt = frame_pix_fmt;
2380
2381                 if (!intra_only)
2382                     video_enc->gop_size = gop_size;
2383                 else
2384                     video_enc->gop_size = 0;
2385                 if (video_qscale || same_quality) {
2386                     video_enc->flags |= CODEC_FLAG_QSCALE;
2387                     st->quality = FF_QP2LAMBDA * video_qscale;
2388                 }
2389
2390                 if(intra_matrix)
2391                     video_enc->intra_matrix = intra_matrix;
2392                 if(inter_matrix)
2393                     video_enc->inter_matrix = inter_matrix;
2394
2395                 if(bitexact)
2396                     video_enc->flags |= CODEC_FLAG_BITEXACT;
2397
2398                 video_enc->mb_decision = mb_decision;
2399                 video_enc->mb_cmp = mb_cmp;
2400                 video_enc->ildct_cmp = ildct_cmp;
2401                 video_enc->me_sub_cmp = sub_cmp;
2402                 video_enc->me_cmp = cmp;
2403
2404                 if (use_umv) {
2405                     video_enc->flags |= CODEC_FLAG_H263P_UMV;
2406                 }
2407                 if (use_aic) {
2408                     video_enc->flags |= CODEC_FLAG_H263P_AIC;
2409                 }
2410                 if (use_aiv) {
2411                     video_enc->flags |= CODEC_FLAG_H263P_AIV;
2412                 }
2413                 if (use_4mv) {
2414                     video_enc->flags |= CODEC_FLAG_4MV;
2415                 }
2416                 if (use_obmc) {
2417                     video_enc->flags |= CODEC_FLAG_OBMC;
2418                 }
2419
2420                 if(use_part) {
2421                     video_enc->flags |= CODEC_FLAG_PART;
2422                 }
2423                 if (use_alt_scan) {
2424                     video_enc->flags |= CODEC_FLAG_ALT_SCAN;
2425                 }
2426                 if (use_trell) {
2427                     video_enc->flags |= CODEC_FLAG_TRELLIS_QUANT;
2428                 }
2429                 if (use_scan_offset) {
2430                     video_enc->flags |= CODEC_FLAG_SVCD_SCAN_OFFSET;
2431                 }
2432                 if (closed_gop) {
2433                     video_enc->flags |= CODEC_FLAG_CLOSED_GOP;
2434                 }
2435                 if (b_frames) {
2436                     video_enc->max_b_frames = b_frames;
2437                     video_enc->b_frame_strategy = 0;
2438                     video_enc->b_quant_factor = 2.0;
2439                 }
2440                 if (do_interlace_dct) {
2441                     video_enc->flags |= CODEC_FLAG_INTERLACED_DCT;
2442                 }
2443                 if (do_interlace_me) {
2444                     video_enc->flags |= CODEC_FLAG_INTERLACED_ME;
2445                 }
2446                 video_enc->qmin = video_qmin;
2447                 video_enc->qmax = video_qmax;
2448                 video_enc->mb_qmin = video_mb_qmin;
2449                 video_enc->mb_qmax = video_mb_qmax;
2450                 video_enc->max_qdiff = video_qdiff;
2451                 video_enc->qblur = video_qblur;
2452                 video_enc->qcompress = video_qcomp;
2453                 video_enc->rc_eq = video_rc_eq;
2454                 video_enc->debug = debug;
2455                 video_enc->debug_mv = debug_mv;
2456                 p= video_rc_override_string;
2457                 for(i=0; p; i++){
2458                     int start, end, q;
2459                     int e=sscanf(p, "%d,%d,%d", &start, &end, &q);
2460                     if(e!=3){
2461                         fprintf(stderr, "error parsing rc_override\n");
2462                         exit(1);
2463                     }
2464                     video_enc->rc_override=
2465                         av_realloc(video_enc->rc_override,
2466                                    sizeof(RcOverride)*(i+1));
2467                     video_enc->rc_override[i].start_frame= start;
2468                     video_enc->rc_override[i].end_frame  = end;
2469                     if(q>0){
2470                         video_enc->rc_override[i].qscale= q;
2471                         video_enc->rc_override[i].quality_factor= 1.0;
2472                     }
2473                     else{
2474                         video_enc->rc_override[i].qscale= 0;
2475                         video_enc->rc_override[i].quality_factor= -q/100.0;
2476                     }
2477                     p= strchr(p, '/');
2478                     if(p) p++;
2479                 }
2480                 video_enc->rc_override_count=i;
2481
2482                 video_enc->rc_max_rate = video_rc_max_rate;
2483                 video_enc->rc_min_rate = video_rc_min_rate;
2484                 video_enc->rc_buffer_size = video_rc_buffer_size;
2485                 video_enc->rc_buffer_aggressivity= video_rc_buffer_aggressivity;
2486                 video_enc->rc_initial_cplx= video_rc_initial_cplx;
2487                 video_enc->i_quant_factor = video_i_qfactor;
2488                 video_enc->b_quant_factor = video_b_qfactor;
2489                 video_enc->i_quant_offset = video_i_qoffset;
2490                 video_enc->b_quant_offset = video_b_qoffset;
2491                 video_enc->intra_quant_bias = video_intra_quant_bias;
2492                 video_enc->inter_quant_bias = video_inter_quant_bias;
2493                 video_enc->dct_algo = dct_algo;
2494                 video_enc->idct_algo = idct_algo;
2495                 video_enc->strict_std_compliance = strict;
2496                 video_enc->error_rate = error_rate;
2497                 video_enc->noise_reduction= noise_reduction;
2498                 video_enc->scenechange_threshold= sc_threshold;
2499                 if(packet_size){
2500                     video_enc->rtp_mode= 1;
2501                     video_enc->rtp_payload_size= packet_size;
2502                 }
2503
2504                 if (do_psnr)
2505                     video_enc->flags|= CODEC_FLAG_PSNR;
2506
2507                 video_enc->me_method = me_method;
2508
2509                 /* two pass mode */
2510                 if (do_pass) {
2511                     if (do_pass == 1) {
2512                         video_enc->flags |= CODEC_FLAG_PASS1;
2513                     } else {
2514                         video_enc->flags |= CODEC_FLAG_PASS2;
2515                     }
2516                 }
2517             }
2518             oc->streams[nb_streams] = st;
2519             nb_streams++;
2520         }
2521
2522         if (use_audio) {
2523             AVCodecContext *audio_enc;
2524
2525             st = av_mallocz(sizeof(AVStream));
2526             if (!st) {
2527                 fprintf(stderr, "Could not alloc stream\n");
2528                 exit(1);
2529             }
2530             avcodec_get_context_defaults(&st->codec);
2531
2532             audio_enc = &st->codec;
2533             audio_enc->codec_type = CODEC_TYPE_AUDIO;
2534
2535             if(!strcmp(file_oformat->name, "mp4") || !strcmp(file_oformat->name, "mov") || !strcmp(file_oformat->name, "3gp"))
2536                 audio_enc->flags |= CODEC_FLAG_GLOBAL_HEADER;
2537             if (audio_stream_copy) {
2538                 st->stream_copy = 1;
2539             } else {
2540                 codec_id = file_oformat->audio_codec;
2541                 if (audio_codec_id != CODEC_ID_NONE)
2542                     codec_id = audio_codec_id;
2543                 audio_enc->codec_id = codec_id;
2544
2545                 audio_enc->bit_rate = audio_bit_rate;
2546                 audio_enc->sample_rate = audio_sample_rate;
2547                 /* For audio codecs other than AC3 we limit */
2548                 /* the number of coded channels to stereo   */
2549                 if (audio_channels > 2 && codec_id != CODEC_ID_AC3) {
2550                     audio_enc->channels = 2;
2551                 } else
2552                     audio_enc->channels = audio_channels;
2553             }
2554             oc->streams[nb_streams] = st;
2555             nb_streams++;
2556         }
2557
2558         oc->nb_streams = nb_streams;
2559
2560         if (!nb_streams) {
2561             fprintf(stderr, "No audio or video streams available\n");
2562             exit(1);
2563         }
2564
2565         if (str_title)
2566             pstrcpy(oc->title, sizeof(oc->title), str_title);
2567         if (str_author)
2568             pstrcpy(oc->author, sizeof(oc->author), str_author);
2569         if (str_copyright)
2570             pstrcpy(oc->copyright, sizeof(oc->copyright), str_copyright);
2571         if (str_comment)
2572             pstrcpy(oc->comment, sizeof(oc->comment), str_comment);
2573     }
2574
2575     output_files[nb_output_files++] = oc;
2576
2577     strcpy(oc->filename, filename);
2578
2579     /* check filename in case of an image number is expected */
2580     if (oc->oformat->flags & AVFMT_NEEDNUMBER) {
2581         if (filename_number_test(oc->filename) < 0) {
2582             print_error(oc->filename, AVERROR_NUMEXPECTED);
2583             exit(1);
2584         }
2585     }
2586
2587     if (!(oc->oformat->flags & AVFMT_NOFILE)) {
2588         /* test if it already exists to avoid loosing precious files */
2589         if (!file_overwrite &&
2590             (strchr(filename, ':') == NULL ||
2591              strstart(filename, "file:", NULL))) {
2592             if (url_exist(filename)) {
2593                 int c;
2594
2595                 if ( !using_stdin ) {
2596                     fprintf(stderr,"File '%s' already exists. Overwrite ? [y/N] ", filename);
2597                     fflush(stderr);
2598                     c = getchar();
2599                     if (toupper(c) != 'Y') {
2600                         fprintf(stderr, "Not overwriting - exiting\n");
2601                         exit(1);
2602                     }
2603                                 }
2604                                 else {
2605                     fprintf(stderr,"File '%s' already exists. Exiting.\n", filename);
2606                     exit(1);
2607                                 }
2608             }
2609         }
2610
2611         /* open the file */
2612         if (url_fopen(&oc->pb, filename, URL_WRONLY) < 0) {
2613             fprintf(stderr, "Could not open '%s'\n", filename);
2614             exit(1);
2615         }
2616     }
2617
2618     memset(ap, 0, sizeof(*ap));
2619     ap->image_format = image_format;
2620     if (av_set_parameters(oc, ap) < 0) {
2621         fprintf(stderr, "%s: Invalid encoding parameters\n",
2622                 oc->filename);
2623         exit(1);
2624     }
2625
2626     /* reset some options */
2627     file_oformat = NULL;
2628     file_iformat = NULL;
2629     image_format = NULL;
2630     audio_disable = 0;
2631     video_disable = 0;
2632     audio_codec_id = CODEC_ID_NONE;
2633     video_codec_id = CODEC_ID_NONE;
2634     audio_stream_copy = 0;
2635     video_stream_copy = 0;
2636 }
2637
2638 /* prepare dummy protocols for grab */
2639 static void prepare_grab(void)
2640 {
2641     int has_video, has_audio, i, j;
2642     AVFormatContext *oc;
2643     AVFormatContext *ic;
2644     AVFormatParameters vp1, *vp = &vp1;
2645     AVFormatParameters ap1, *ap = &ap1;
2646
2647     /* see if audio/video inputs are needed */
2648     has_video = 0;
2649     has_audio = 0;
2650     memset(ap, 0, sizeof(*ap));
2651     memset(vp, 0, sizeof(*vp));
2652     for(j=0;j<nb_output_files;j++) {
2653         oc = output_files[j];
2654         for(i=0;i<oc->nb_streams;i++) {
2655             AVCodecContext *enc = &oc->streams[i]->codec;
2656             switch(enc->codec_type) {
2657             case CODEC_TYPE_AUDIO:
2658                 if (enc->sample_rate > ap->sample_rate)
2659                     ap->sample_rate = enc->sample_rate;
2660                 if (enc->channels > ap->channels)
2661                     ap->channels = enc->channels;
2662                 has_audio = 1;
2663                 break;
2664             case CODEC_TYPE_VIDEO:
2665                 if (enc->width > vp->width)
2666                     vp->width = enc->width;
2667                 if (enc->height > vp->height)
2668                     vp->height = enc->height;
2669
2670                 assert(enc->frame_rate_base == DEFAULT_FRAME_RATE_BASE);
2671                 if (enc->frame_rate > vp->frame_rate){
2672                     vp->frame_rate      = enc->frame_rate;
2673                     vp->frame_rate_base = enc->frame_rate_base;
2674                 }
2675                 has_video = 1;
2676                 break;
2677             default:
2678                 av_abort();
2679             }
2680         }
2681     }
2682
2683     if (has_video == 0 && has_audio == 0) {
2684         fprintf(stderr, "Output file must have at least one audio or video stream\n");
2685         exit(1);
2686     }
2687
2688     if (has_video) {
2689         AVInputFormat *fmt1;
2690         fmt1 = av_find_input_format(video_grab_format);
2691         vp->device  = video_device;
2692         vp->channel = video_channel;
2693         vp->standard = video_standard;
2694         if (av_open_input_file(&ic, "", fmt1, 0, vp) < 0) {
2695             fprintf(stderr, "Could not find video grab device\n");
2696             exit(1);
2697         }
2698         /* If not enough info to get the stream parameters, we decode the
2699            first frames to get it. */
2700         if ((ic->ctx_flags & AVFMTCTX_NOHEADER) && av_find_stream_info(ic) < 0) {
2701             fprintf(stderr, "Could not find video grab parameters\n");
2702             exit(1);
2703         }
2704         /* by now video grab has one stream */
2705         ic->streams[0]->r_frame_rate      = vp->frame_rate;
2706         ic->streams[0]->r_frame_rate_base = vp->frame_rate_base;
2707         input_files[nb_input_files] = ic;
2708         dump_format(ic, nb_input_files, "", 0);
2709         nb_input_files++;
2710     }
2711     if (has_audio && audio_grab_format) {
2712         AVInputFormat *fmt1;
2713         fmt1 = av_find_input_format(audio_grab_format);
2714         ap->device = audio_device;
2715         if (av_open_input_file(&ic, "", fmt1, 0, ap) < 0) {
2716             fprintf(stderr, "Could not find audio grab device\n");
2717             exit(1);
2718         }
2719         input_files[nb_input_files] = ic;
2720         dump_format(ic, nb_input_files, "", 0);
2721         nb_input_files++;
2722     }
2723 }
2724
2725 /* same option as mencoder */
2726 static void opt_pass(const char *pass_str)
2727 {
2728     int pass;
2729     pass = atoi(pass_str);
2730     if (pass != 1 && pass != 2) {
2731         fprintf(stderr, "pass number can be only 1 or 2\n");
2732         exit(1);
2733     }
2734     do_pass = pass;
2735 }
2736
2737 #if defined(CONFIG_WIN32) || defined(CONFIG_OS2)
2738 static int64_t getutime(void)
2739 {
2740   return av_gettime();
2741 }
2742 #else
2743 static int64_t getutime(void)
2744 {
2745     struct rusage rusage;
2746
2747     getrusage(RUSAGE_SELF, &rusage);
2748     return (rusage.ru_utime.tv_sec * 1000000LL) + rusage.ru_utime.tv_usec;
2749 }
2750 #endif
2751
2752 extern int ffm_nopts;
2753
2754 static void opt_bitexact(void)
2755 {
2756     bitexact=1;
2757     /* disable generate of real time pts in ffm (need to be supressed anyway) */
2758     ffm_nopts = 1;
2759 }
2760
2761 static void show_formats(void)
2762 {
2763     AVInputFormat *ifmt;
2764     AVOutputFormat *ofmt;
2765     AVImageFormat *image_fmt;
2766     URLProtocol *up;
2767     AVCodec *p;
2768     const char **pp;
2769
2770     printf("Output audio/video file formats:");
2771     for(ofmt = first_oformat; ofmt != NULL; ofmt = ofmt->next) {
2772         printf(" %s", ofmt->name);
2773     }
2774     printf("\n");
2775
2776     printf("Input audio/video file formats:");
2777     for(ifmt = first_iformat; ifmt != NULL; ifmt = ifmt->next) {
2778         printf(" %s", ifmt->name);
2779     }
2780     printf("\n");
2781
2782     printf("Output image formats:");
2783     for(image_fmt = first_image_format; image_fmt != NULL;
2784         image_fmt = image_fmt->next) {
2785         if (image_fmt->img_write)
2786             printf(" %s", image_fmt->name);
2787     }
2788     printf("\n");
2789
2790     printf("Input image formats:");
2791     for(image_fmt = first_image_format; image_fmt != NULL;
2792         image_fmt = image_fmt->next) {
2793         if (image_fmt->img_read)
2794             printf(" %s", image_fmt->name);
2795     }
2796     printf("\n");
2797
2798     printf("Codecs:\n");
2799     printf("  Encoders:");
2800     for(p = first_avcodec; p != NULL; p = p->next) {
2801         if (p->encode)
2802             printf(" %s", p->name);
2803     }
2804     printf("\n");
2805
2806     printf("  Decoders:");
2807     for(p = first_avcodec; p != NULL; p = p->next) {
2808         if (p->decode)
2809             printf(" %s", p->name);
2810     }
2811     printf("\n");
2812
2813     printf("Supported file protocols:");
2814     for(up = first_protocol; up != NULL; up = up->next)
2815         printf(" %s:", up->name);
2816     printf("\n");
2817
2818     printf("Frame size, frame rate abbreviations: ntsc pal qntsc qpal sntsc spal film ntsc-film sqcif qcif cif 4cif\n");
2819     printf("Motion estimation methods:");
2820     pp = motion_str;
2821     while (*pp) {
2822         printf(" %s", *pp);
2823         if ((pp - motion_str + 1) == ME_ZERO)
2824             printf("(fastest)");
2825         else if ((pp - motion_str + 1) == ME_FULL)
2826             printf("(slowest)");
2827         else if ((pp - motion_str + 1) == ME_EPZS)
2828             printf("(default)");
2829         pp++;
2830     }
2831     printf("\n");
2832     exit(1);
2833 }
2834
2835 void parse_matrix_coeffs(uint16_t *dest, const char *str)
2836 {
2837     int i;
2838     const char *p = str;
2839     for(i = 0;; i++) {
2840         dest[i] = atoi(p);
2841         if(i == 63)
2842             break;
2843         p = strchr(p, ',');
2844         if(!p) {
2845             fprintf(stderr, "Syntax error in matrix \"%s\" at coeff %d\n", str, i);
2846             exit(1);
2847         }
2848         p++;
2849     }
2850 }
2851
2852 void opt_inter_matrix(const char *arg)
2853 {
2854     inter_matrix = av_mallocz(sizeof(uint16_t) * 64);
2855     parse_matrix_coeffs(inter_matrix, arg);
2856 }
2857
2858 void opt_intra_matrix(const char *arg)
2859 {
2860     intra_matrix = av_mallocz(sizeof(uint16_t) * 64);
2861     parse_matrix_coeffs(intra_matrix, arg);
2862 }
2863
2864 static void opt_target(const char *arg)
2865 {
2866     int norm = -1;
2867
2868     if(!strncmp(arg, "pal-", 4)) {
2869         norm = 0;
2870         arg += 4;
2871     } else if(!strncmp(arg, "ntsc-", 5)) {
2872         norm = 1;
2873         arg += 5;
2874     } else {
2875         int fr;
2876         /* Calculate FR via float to avoid int overflow */
2877         fr = (int)(frame_rate * 1000.0 / frame_rate_base);
2878         if(fr == 25000) {
2879             norm = 0;
2880         } else if((fr == 29970) || (fr == 23976)) {
2881             norm = 1;
2882         } else {
2883             /* Try to determine PAL/NTSC by peeking in the input files */
2884             if(nb_input_files) {
2885                 int i, j;
2886                 for(j = 0; j < nb_input_files; j++) {
2887                     for(i = 0; i < input_files[j]->nb_streams; i++) {
2888                         AVCodecContext *c = &input_files[j]->streams[i]->codec;
2889                         if(c->codec_type != CODEC_TYPE_VIDEO)
2890                             continue;
2891                         fr = c->frame_rate * 1000 / c->frame_rate_base;
2892                         if(fr == 25000) {
2893                             norm = 0;
2894                             break;
2895                         } else if((fr == 29970) || (fr == 23976)) {
2896                             norm = 1;
2897                             break;
2898                         }
2899                     }
2900                     if(norm >= 0)
2901                         break;
2902                 }
2903             }
2904         }
2905         if(verbose && norm >= 0)
2906             printf("Assuming %s for target.\n", norm ? "NTSC" : "PAL");
2907     }
2908
2909     if(norm < 0) {
2910         fprintf(stderr, "Could not determine norm (PAL/NTSC) for target.\n");
2911         fprintf(stderr, "Please prefix target with \"pal-\" or \"ntsc-\",\n");
2912         fprintf(stderr, "or set a framerate with \"-r xxx\".\n");
2913         exit(1);
2914     }
2915
2916     if(!strcmp(arg, "vcd")) {
2917
2918         opt_video_codec("mpeg1video");
2919         opt_audio_codec("mp2");
2920         opt_format("vcd");
2921
2922         opt_frame_size(norm ? "352x240" : "352x288");
2923
2924         video_bit_rate = 1150000;
2925         video_rc_max_rate = 1150000;
2926         video_rc_min_rate = 1150000;
2927         video_rc_buffer_size = 40*1024*8;
2928
2929         audio_bit_rate = 224000;
2930         audio_sample_rate = 44100;
2931
2932     } else if(!strcmp(arg, "svcd")) {
2933
2934         opt_video_codec("mpeg2video");
2935         opt_audio_codec("mp2");
2936         opt_format("vcd");
2937
2938         opt_frame_size(norm ? "480x480" : "480x576");
2939         opt_gop_size(norm ? "18" : "15");
2940
2941         video_bit_rate = 2040000;
2942         video_rc_max_rate = 2516000;
2943         video_rc_min_rate = 0; //1145000;
2944         video_rc_buffer_size = 224*1024*8;
2945         use_scan_offset = 1;
2946
2947         audio_bit_rate = 224000;
2948         audio_sample_rate = 44100;
2949
2950     } else if(!strcmp(arg, "dvd")) {
2951
2952         opt_video_codec("mpeg2video");
2953         opt_audio_codec("ac3");
2954         opt_format("vob");
2955
2956         opt_frame_size(norm ? "720x480" : "720x576");
2957         opt_gop_size(norm ? "18" : "15");
2958
2959         video_bit_rate = 6000000;
2960         video_rc_max_rate = 9000000;
2961         video_rc_min_rate = 0; //1500000;
2962         video_rc_buffer_size = 224*1024*8;
2963
2964         audio_bit_rate = 448000;
2965         audio_sample_rate = 48000;
2966
2967     } else {
2968         fprintf(stderr, "Unknown target: %s\n", arg);
2969         exit(1);
2970     }
2971 }
2972
2973 const OptionDef options[] = {
2974     /* main options */
2975     { "L", 0, {(void*)show_license}, "show license" },
2976     { "h", 0, {(void*)show_help}, "show help" },
2977     { "formats", 0, {(void*)show_formats}, "show available formats, codecs, protocols, ..." },
2978     { "f", HAS_ARG, {(void*)opt_format}, "force format", "fmt" },
2979     { "img", HAS_ARG, {(void*)opt_image_format}, "force image format", "img_fmt" },
2980     { "i", HAS_ARG, {(void*)opt_input_file}, "input file name", "filename" },
2981     { "y", OPT_BOOL, {(void*)&file_overwrite}, "overwrite output files" },
2982     { "map", HAS_ARG | OPT_EXPERT, {(void*)opt_map}, "set input stream mapping", "file:stream" },
2983     { "t", HAS_ARG, {(void*)opt_recording_time}, "set the recording time", "duration" },
2984     { "ss", HAS_ARG, {(void*)opt_start_time}, "set the start time offset", "time_off" },
2985     { "title", HAS_ARG | OPT_STRING, {(void*)&str_title}, "set the title", "string" },
2986     { "author", HAS_ARG | OPT_STRING, {(void*)&str_author}, "set the author", "string" },
2987     { "copyright", HAS_ARG | OPT_STRING, {(void*)&str_copyright}, "set the copyright", "string" },
2988     { "comment", HAS_ARG | OPT_STRING, {(void*)&str_comment}, "set the comment", "string" },
2989     { "debug", HAS_ARG | OPT_EXPERT, {(void*)opt_debug}, "print specific debug info", "" },
2990     { "vismv", HAS_ARG | OPT_EXPERT, {(void*)opt_vismv}, "visualize motion vectors", "" },
2991     { "benchmark", OPT_BOOL | OPT_EXPERT, {(void*)&do_benchmark},
2992       "add timings for benchmarking" },
2993     { "dump", OPT_BOOL | OPT_EXPERT, {(void*)&do_pkt_dump},
2994       "dump each input packet" },
2995     { "hex", OPT_BOOL | OPT_EXPERT, {(void*)&do_hex_dump},
2996       "when dumping packets, also dump the payload" },
2997     { "bitexact", OPT_EXPERT, {(void*)opt_bitexact}, "only use bit exact algorithms (for codec testing)" },
2998     { "re", OPT_BOOL | OPT_EXPERT, {(void*)&rate_emu}, "read input at native frame rate", "" },
2999     { "loop", OPT_BOOL | OPT_EXPERT, {(void*)&loop_input}, "loop (current only works with images)" },
3000     { "v", HAS_ARG, {(void*)opt_verbose}, "control amount of logging", "verbose" },
3001     { "target", HAS_ARG, {(void*)opt_target}, "specify target file type (\"vcd\", \"svcd\" or \"dvd\")", "type" },
3002
3003     /* video options */
3004     { "b", HAS_ARG | OPT_VIDEO, {(void*)opt_video_bitrate}, "set video bitrate (in kbit/s)", "bitrate" },
3005     { "r", HAS_ARG | OPT_VIDEO, {(void*)opt_frame_rate}, "set frame rate (Hz value, fraction or abbreviation)", "rate" },
3006     { "s", HAS_ARG | OPT_VIDEO, {(void*)opt_frame_size}, "set frame size (WxH or abbreviation)", "size" },
3007     { "aspect", HAS_ARG | OPT_VIDEO, {(void*)opt_frame_aspect_ratio}, "set aspect ratio (4:3, 16:9 or 1.3333, 1.7777)", "aspect" },
3008     { "pix_fmt", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_frame_pix_fmt}, "set pixel format", "format" },
3009     { "croptop", HAS_ARG | OPT_VIDEO, {(void*)opt_frame_crop_top}, "set top crop band size (in pixels)", "size" },
3010     { "cropbottom", HAS_ARG | OPT_VIDEO, {(void*)opt_frame_crop_bottom}, "set bottom crop band size (in pixels)", "size" },
3011     { "cropleft", HAS_ARG | OPT_VIDEO, {(void*)opt_frame_crop_left}, "set left crop band size (in pixels)", "size" },
3012     { "cropright", HAS_ARG | OPT_VIDEO, {(void*)opt_frame_crop_right}, "set right crop band size (in pixels)", "size" },
3013     { "g", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_gop_size}, "set the group of picture size", "gop_size" },
3014     { "intra", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&intra_only}, "use only intra frames"},
3015     { "vn", OPT_BOOL | OPT_VIDEO, {(void*)&video_disable}, "disable video" },
3016     { "qscale", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_qscale}, "use fixed video quantiser scale (VBR)", "q" },
3017     { "qmin", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_qmin}, "min video quantiser scale (VBR)", "q" },
3018     { "qmax", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_qmax}, "max video quantiser scale (VBR)", "q" },
3019     { "mbqmin", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_mb_qmin}, "min macroblock quantiser scale (VBR)", "q" },
3020     { "mbqmax", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_mb_qmax}, "max macroblock quantiser scale (VBR)", "q" },
3021     { "qdiff", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_qdiff}, "max difference between the quantiser scale (VBR)", "q" },
3022     { "qblur", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_qblur}, "video quantiser scale blur (VBR)", "blur" },
3023     { "qcomp", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_qcomp}, "video quantiser scale compression (VBR)", "compression" },
3024     { "rc_init_cplx", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_rc_initial_cplx}, "initial complexity for 1-pass encoding", "complexity" },
3025     { "b_qfactor", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_b_qfactor}, "qp factor between p and b frames", "factor" },
3026     { "i_qfactor", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_i_qfactor}, "qp factor between p and i frames", "factor" },
3027     { "b_qoffset", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_b_qoffset}, "qp offset between p and b frames", "offset" },
3028     { "i_qoffset", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_i_qoffset}, "qp offset between p and i frames", "offset" },
3029     { "ibias", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_ibias}, "intra quant bias", "bias" },
3030     { "pbias", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_pbias}, "inter quant bias", "bias" },
3031 //    { "b_strategy", HAS_ARG | OPT_EXPERT, {(void*)opt_b_strategy}, "dynamic b frame selection strategy", "strategy" },
3032     { "rc_eq", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_video_rc_eq}, "set rate control equation", "equation" },
3033     { "rc_override", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_video_rc_override_string}, "rate control override for specific intervals", "override" },
3034     { "bt", HAS_ARG | OPT_VIDEO, {(void*)opt_video_bitrate_tolerance}, "set video bitrate tolerance (in kbit/s)", "tolerance" },
3035     { "maxrate", HAS_ARG | OPT_VIDEO, {(void*)opt_video_bitrate_max}, "set max video bitrate tolerance (in kbit/s)", "bitrate" },
3036     { "minrate", HAS_ARG | OPT_VIDEO, {(void*)opt_video_bitrate_min}, "set min video bitrate tolerance (in kbit/s)", "bitrate" },
3037     { "bufsize", HAS_ARG | OPT_VIDEO, {(void*)opt_video_buffer_size}, "set ratecontrol buffere size (in kByte)", "size" },
3038     { "vcodec", HAS_ARG | OPT_VIDEO, {(void*)opt_video_codec}, "force video codec ('copy' to copy stream)", "codec" },
3039     { "me", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_motion_estimation}, "set motion estimation method",
3040       "method" },
3041     { "dct_algo", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_dct_algo}, "set dct algo",  "algo" },
3042     { "idct_algo", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_idct_algo}, "set idct algo",  "algo" },
3043     { "er", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_error_resilience}, "set error resilience",  "n" },
3044     { "ec", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_error_concealment}, "set error concealment",  "bit_mask" },
3045     { "bf", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_b_frames}, "use 'frames' B frames", "frames" },
3046     { "hq", OPT_BOOL, {(void*)&mb_decision}, "activate high quality settings" },
3047     { "mbd", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_mb_decision}, "macroblock decision", "mode" },
3048     { "mbcmp", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_mb_cmp}, "macroblock compare function", "cmp function" },
3049     { "ildctcmp", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_ildct_cmp}, "ildct compare function", "cmp function" },
3050     { "subcmp", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_sub_cmp}, "subpel compare function", "cmp function" },
3051     { "cmp", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_cmp}, "fullpel compare function", "cmp function" },
3052     { "4mv", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&use_4mv}, "use four motion vector by macroblock (MPEG4)" },
3053     { "obmc", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&use_obmc}, "use overlapped block motion compensation (h263+)" },
3054     { "part", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&use_part}, "use data partitioning (MPEG4)" },
3055     { "bug", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_workaround_bugs}, "workaround not auto detected encoder bugs", "param" },
3056     { "ps", HAS_ARG | OPT_EXPERT, {(void*)opt_packet_size}, "set packet size in bits", "size" },
3057     { "error", HAS_ARG | OPT_EXPERT, {(void*)opt_error_rate}, "error rate", "rate" },
3058     { "strict", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_strict}, "how strictly to follow the standarts", "strictness" },
3059     { "sameq", OPT_BOOL | OPT_VIDEO, {(void*)&same_quality},
3060       "use same video quality as source (implies VBR)" },
3061     { "pass", HAS_ARG | OPT_VIDEO, {(void*)&opt_pass}, "select the pass number (1 or 2)", "n" },
3062     { "passlogfile", HAS_ARG | OPT_STRING | OPT_VIDEO, {(void*)&pass_logfilename}, "select two pass log file name", "file" },
3063     { "deinterlace", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&do_deinterlace},
3064       "deinterlace pictures" },
3065     { "ildct", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&do_interlace_dct},
3066       "force interlaced dct support in encoder (MPEG2/MPEG4)" },
3067     { "ilme", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&do_interlace_me},
3068       "force interlacied me support in encoder MPEG2" },
3069     { "psnr", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&do_psnr}, "calculate PSNR of compressed frames" },
3070     { "vstats", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&do_vstats}, "dump video coding statistics to file" },
3071     { "vhook", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)add_frame_hooker}, "insert video processing module", "module" },
3072     { "aic", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&use_aic}, "enable Advanced intra coding (h263+)" },
3073     { "aiv", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&use_aiv}, "enable Alternative inter vlc (h263+)" },
3074     { "umv", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&use_umv}, "enable Unlimited Motion Vector (h263+)" },
3075     { "alt", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&use_alt_scan}, "enable alternate scantable (mpeg2)" },
3076     { "trell", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&use_trell}, "enable trellis quantization" },
3077     { "cgop", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&closed_gop}, "closed gop" },
3078     { "scan_offset", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&use_scan_offset}, "enable SVCD Scan Offset placeholder" },
3079     { "intra_matrix", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_intra_matrix}, "specify intra matrix coeffs", "matrix" },
3080     { "inter_matrix", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_inter_matrix}, "specify inter matrix coeffs", "matrix" },
3081     { "top", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_top_field_first}, "top=1/bottom=0/auto=-1 field first", "" },
3082     { "nr", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_noise_reduction}, "noise reduction", "" },
3083     { "sc_threshold", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_sc_threshold}, "scene change threshold", "threshold" },
3084
3085     /* audio options */
3086     { "ab", HAS_ARG | OPT_AUDIO, {(void*)opt_audio_bitrate}, "set audio bitrate (in kbit/s)", "bitrate", },
3087     { "ar", HAS_ARG | OPT_AUDIO, {(void*)opt_audio_rate}, "set audio sampling rate (in Hz)", "rate" },
3088     { "ac", HAS_ARG | OPT_AUDIO, {(void*)opt_audio_channels}, "set number of audio channels", "channels" },
3089     { "an", OPT_BOOL | OPT_AUDIO, {(void*)&audio_disable}, "disable audio" },
3090     { "acodec", HAS_ARG | OPT_AUDIO, {(void*)opt_audio_codec}, "force audio codec ('copy' to copy stream)", "codec" },
3091
3092     /* grab options */
3093     { "vd", HAS_ARG | OPT_EXPERT | OPT_VIDEO | OPT_GRAB, {(void*)opt_video_device}, "set video grab device", "device" },
3094     { "vc", HAS_ARG | OPT_EXPERT | OPT_VIDEO | OPT_GRAB, {(void*)opt_video_channel}, "set video grab channel (DV1394 only)", "channel" },
3095     { "tvstd", HAS_ARG | OPT_EXPERT | OPT_VIDEO | OPT_GRAB, {(void*)opt_video_standard}, "set television standard (NTSC, PAL (SECAM))", "standard" },
3096     { "dv1394", OPT_EXPERT | OPT_GRAB, {(void*)opt_dv1394}, "set DV1394 grab", "" },
3097     { "ad", HAS_ARG | OPT_EXPERT | OPT_AUDIO | OPT_GRAB, {(void*)opt_audio_device}, "set audio device", "device" },
3098     { NULL, },
3099 };
3100
3101 static void show_banner(void)
3102 {
3103     printf("ffmpeg version " FFMPEG_VERSION ", Copyright (c) 2000-2003 Fabrice Bellard\n");
3104 }
3105
3106 static void show_license(void)
3107 {
3108     show_banner();
3109     printf(
3110     "This library is free software; you can redistribute it and/or\n"
3111     "modify it under the terms of the GNU Lesser General Public\n"
3112     "License as published by the Free Software Foundation; either\n"
3113     "version 2 of the License, or (at your option) any later version.\n"
3114     "\n"
3115     "This library is distributed in the hope that it will be useful,\n"
3116     "but WITHOUT ANY WARRANTY; without even the implied warranty of\n"
3117     "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU\n"
3118     "Lesser General Public License for more details.\n"
3119     "\n"
3120     "You should have received a copy of the GNU Lesser General Public\n"
3121     "License along with this library; if not, write to the Free Software\n"
3122     "Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA\n"
3123     );
3124     exit(1);
3125 }
3126
3127 static void show_help(void)
3128 {
3129     show_banner();
3130     printf("usage: ffmpeg [[options] -i input_file]... {[options] outfile}...\n"
3131            "Hyper fast Audio and Video encoder\n");
3132     printf("\n");
3133     show_help_options(options, "Main options:\n",
3134                       OPT_EXPERT | OPT_AUDIO | OPT_VIDEO, 0);
3135     show_help_options(options, "\nVideo options:\n",
3136                       OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_GRAB,
3137                       OPT_VIDEO);
3138     show_help_options(options, "\nAdvanced Video options:\n",
3139                       OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_GRAB,
3140                       OPT_VIDEO | OPT_EXPERT);
3141     show_help_options(options, "\nAudio options:\n",
3142                       OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_GRAB,
3143                       OPT_AUDIO);
3144     show_help_options(options, "\nAdvanced Audio options:\n",
3145                       OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_GRAB,
3146                       OPT_AUDIO | OPT_EXPERT);
3147     show_help_options(options, "\nAudio/Video grab options:\n",
3148                       OPT_GRAB,
3149                       OPT_GRAB);
3150     show_help_options(options, "\nAdvanced options:\n",
3151                       OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_GRAB,
3152                       OPT_EXPERT);
3153     exit(1);
3154 }
3155
3156 void parse_arg_file(const char *filename)
3157 {
3158     opt_output_file(filename);
3159 }
3160
3161 int main(int argc, char **argv)
3162 {
3163     int i;
3164     int64_t ti;
3165
3166     av_register_all();
3167
3168     if (argc <= 1)
3169         show_help();
3170
3171     /* parse options */
3172     parse_options(argc, argv, options);
3173
3174     /* file converter / grab */
3175     if (nb_output_files <= 0) {
3176         fprintf(stderr, "Must supply at least one output file\n");
3177         exit(1);
3178     }
3179
3180     if (nb_input_files == 0) {
3181         prepare_grab();
3182     }
3183
3184     ti = getutime();
3185     av_encode(output_files, nb_output_files, input_files, nb_input_files,
3186               stream_maps, nb_stream_maps);
3187     ti = getutime() - ti;
3188     if (do_benchmark) {
3189         printf("bench: utime=%0.3fs\n", ti / 1000000.0);
3190     }
3191
3192     /* close files */
3193     for(i=0;i<nb_output_files;i++) {
3194         /* maybe av_close_output_file ??? */
3195         AVFormatContext *s = output_files[i];
3196         int j;
3197         if (!(s->oformat->flags & AVFMT_NOFILE))
3198             url_fclose(&s->pb);
3199         for(j=0;j<s->nb_streams;j++)
3200             av_free(s->streams[j]);
3201         av_free(s);
3202     }
3203     for(i=0;i<nb_input_files;i++)
3204         av_close_input_file(input_files[i]);
3205
3206     av_free_static();
3207
3208     if(intra_matrix)
3209         av_free(intra_matrix);
3210     if(inter_matrix)
3211         av_free(inter_matrix);
3212
3213 #ifdef POWERPC_PERFORMANCE_REPORT
3214     extern void powerpc_display_perf_report(void);
3215     powerpc_display_perf_report();
3216 #endif /* POWERPC_PERFORMANCE_REPORT */
3217
3218 #ifndef CONFIG_WIN32
3219     if (received_sigterm) {
3220         fprintf(stderr,
3221             "Received signal %d: terminating.\n",
3222             (int) received_sigterm);
3223         exit (255);
3224     }
3225 #endif
3226     exit(0); /* not all OS-es handle main() return value */
3227     return 0;
3228 }