falkTX
/
FFmpeg
mirror of https://github.com/falkTX/FFmpeg.git
1
0
Fork 0
Code Issues 0 Releases 338 Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
2888 Commits
32 Branches
580MB
Tree: 0c9bbaec60
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '0c9bbaec60'
${ noResults }
FFmpeg/ffmpeg.c

3150 lines
104KB
Raw Blame History 

  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 me_method = ME_EPZS;

  109. static int video_disable = 0;

  110. static int video_codec_id = CODEC_ID_NONE;

  111. static int same_quality = 0;

  112. static int b_frames = 0;

  113. static int mb_decision = FF_MB_DECISION_SIMPLE;

  114. static int mb_cmp = FF_CMP_SAD;

  115. static int sub_cmp = FF_CMP_SAD;

  116. static int cmp = FF_CMP_SAD;

  117. static int use_4mv = 0;

  118. static int use_obmc = 0;

  119. static int use_aic = 0;

  120. static int use_aiv = 0;

  121. static int use_umv = 0;

  122. static int do_deinterlace = 0;

  123. static int do_interlace = 0;

  124. static int workaround_bugs = FF_BUG_AUTODETECT;

  125. static int error_resilience = 2;

  126. static int error_concealment = 3;

  127. static int dct_algo = 0;

  128. static int idct_algo = 0;

  129. static int use_part = 0;

  130. static int packet_size = 0;

  131. static int error_rate = 0;

  132. static int strict = 0;

  133. static int debug = 0;

  134. static int debug_mv = 0;

  135. extern int loop_input; /* currently a hack */

  136. 

  137. static int gop_size = 12;

  138. static int intra_only = 0;

  139. static int audio_sample_rate = 44100;

  140. static int audio_bit_rate = 64000;

  141. static int audio_disable = 0;

  142. static int audio_channels = 1;

  143. static int audio_codec_id = CODEC_ID_NONE;

  144. 

  145. static int64_t recording_time = 0;

  146. static int64_t start_time = 0;

  147. static int file_overwrite = 0;

  148. static char *str_title = NULL;

  149. static char *str_author = NULL;

  150. static char *str_copyright = NULL;

  151. static char *str_comment = NULL;

  152. static int do_benchmark = 0;

  153. static int do_hex_dump = 0;

  154. static int do_pkt_dump = 0;

  155. static int do_psnr = 0;

  156. static int do_vstats = 0;

  157. static int do_pass = 0;

  158. static int bitexact = 0;

  159. static char *pass_logfilename = NULL;

  160. static int audio_stream_copy = 0;

  161. static int video_stream_copy = 0;

  162. 

  163. static int rate_emu = 0;

  164. 

  165. static char *video_grab_format = "video4linux";

  166. static char *video_device = NULL;

  167. static int  video_channel = 0;

  168. static char *video_standard = "ntsc";

  169. 

  170. static char *audio_grab_format = "audio_device";

  171. static char *audio_device = NULL;

  172. 

  173. static int using_stdin = 0;

  174. static int using_vhook = 0;

  175. static int verbose = 1;

  176. 

  177. #define DEFAULT_PASS_LOGFILENAME "ffmpeg2pass"

  178. 

  179. typedef struct AVOutputStream {

  180.     int file_index;          /* file index */

  181.     int index;               /* stream index in the output file */

  182.     int source_index;        /* AVInputStream index */

  183.     AVStream *st;            /* stream in the output file */

  184.     int encoding_needed;     /* true if encoding needed for this stream */

  185.     int frame_number;

  186.     /* input pts and corresponding output pts

  187.        for A/V sync */

  188.     double sync_ipts;

  189.     double sync_ipts_offset;

  190.     int64_t sync_opts;

  191.     /* video only */

  192.     int video_resample;      /* video_resample and video_crop are mutually exclusive */

  193.     AVPicture pict_tmp;      /* temporary image for resampling */

  194.     ImgReSampleContext *img_resample_ctx; /* for image resampling */

  195. 

  196.     int video_crop;          /* video_resample and video_crop are mutually exclusive */

  197.     int topBand;             /* cropping area sizes */

  198.     int leftBand;

  199.     

  200.     /* audio only */

  201.     int audio_resample;

  202.     ReSampleContext *resample; /* for audio resampling */

  203.     FifoBuffer fifo;     /* for compression: one audio fifo per codec */

  204.     FILE *logfile;

  205. } AVOutputStream;

  206. 

  207. typedef struct AVInputStream {

  208.     int file_index;

  209.     int index;

  210.     AVStream *st;

  211.     int discard;             /* true if stream data should be discarded */

  212.     int decoding_needed;     /* true if the packets must be decoded in 'raw_fifo' */

  213.     int64_t sample_index;      /* current sample */

  214. 

  215.     int64_t       start;     /* time when read started */

  216.     unsigned long frame;     /* current frame */

  217.     int64_t       next_pts;  /* synthetic pts for cases where pkt.pts

  218.                                 is not defined */

  219.     int64_t       pts;       /* current pts */

  220. } AVInputStream;

  221. 

  222. typedef struct AVInputFile {

  223.     int eof_reached;      /* true if eof reached */

  224.     int ist_index;        /* index of first stream in ist_table */

  225.     int buffer_size;      /* current total buffer size */

  226.     int buffer_size_max;  /* buffer size at which we consider we can stop

  227.                              buffering */

  228.     int nb_streams;       /* nb streams we are aware of */

  229. } AVInputFile;

  230. 

  231. #ifndef CONFIG_WIN32

  232. 

  233. /* init terminal so that we can grab keys */

  234. static struct termios oldtty;

  235. 

  236. static void term_exit(void)

  237. {

  238.     tcsetattr (0, TCSANOW, &oldtty);

  239. }

  240. 

  241. static volatile sig_atomic_t received_sigterm = 0;

  242. 

  243. static void

  244. sigterm_handler(int sig)

  245. {

  246.     received_sigterm = sig;

  247.     term_exit();

  248. }

  249. 

  250. static void term_init(void)

  251. {

  252.     struct termios tty;

  253. 

  254.     tcgetattr (0, &tty);

  255.     oldtty = tty;

  256. 

  257.     tty.c_iflag &= ~(IGNBRK|BRKINT|PARMRK|ISTRIP

  258.                           |INLCR|IGNCR|ICRNL|IXON);

  259.     tty.c_oflag |= OPOST;

  260.     tty.c_lflag &= ~(ECHO|ECHONL|ICANON|IEXTEN);

  261.     tty.c_cflag &= ~(CSIZE|PARENB);

  262.     tty.c_cflag |= CS8;

  263.     tty.c_cc[VMIN] = 1;

  264.     tty.c_cc[VTIME] = 0;

  265.     

  266.     tcsetattr (0, TCSANOW, &tty);

  267. 

  268.     signal(SIGINT , sigterm_handler); /* Interrupt (ANSI).  */

  269.     signal(SIGQUIT, sigterm_handler); /* Quit (POSIX).  */

  270.     signal(SIGTERM, sigterm_handler); /* Termination (ANSI).  */

  271.     /*

  272.     register a function to be called at normal program termination

  273.     */

  274.     atexit(term_exit);

  275. #ifdef CONFIG_BEOS_NETSERVER

  276.     fcntl(0, F_SETFL, fcntl(0, F_GETFL) | O_NONBLOCK);

  277. #endif

  278. }

  279. 

  280. /* read a key without blocking */

  281. static int read_key(void)

  282. {

  283.     int n = 1;

  284.     unsigned char ch;

  285. #ifndef CONFIG_BEOS_NETSERVER

  286.     struct timeval tv;

  287.     fd_set rfds;

  288. 

  289.     FD_ZERO(&rfds);

  290.     FD_SET(0, &rfds);

  291.     tv.tv_sec = 0;

  292.     tv.tv_usec = 0;

  293.     n = select(1, &rfds, NULL, NULL, &tv);

  294. #endif

  295.     if (n > 0) {

  296.         n = read(0, &ch, 1);

  297.         if (n == 1)

  298.             return ch;

  299. 

  300.         return n;

  301.     }

  302.     return -1;

  303. }

  304. 

  305. #else

  306. 

  307. static volatile int received_sigterm = 0;

  308. 

  309. /* no interactive support */

  310. static void term_exit(void)

  311. {

  312. }

  313. 

  314. static void term_init(void)

  315. {

  316. }

  317. 

  318. static int read_key(void)

  319. {

  320.     return 0;

  321. }

  322. 

  323. #endif

  324. 

  325. static int read_ffserver_streams(AVFormatContext *s, const char *filename)

  326. {

  327.     int i, err;

  328.     AVFormatContext *ic;

  329. 

  330.     err = av_open_input_file(&ic, filename, NULL, FFM_PACKET_SIZE, NULL);

  331.     if (err < 0)

  332.         return err;

  333.     /* copy stream format */

  334.     s->nb_streams = ic->nb_streams;

  335.     for(i=0;i<ic->nb_streams;i++) {

  336.         AVStream *st;

  337. 

  338.         st = av_mallocz(sizeof(AVStream));

  339.         memcpy(st, ic->streams[i], sizeof(AVStream));

  340.         s->streams[i] = st;

  341.     }

  342. 

  343.     av_close_input_file(ic);

  344.     return 0;

  345. }

  346. 

  347. #define MAX_AUDIO_PACKET_SIZE (128 * 1024)

  348. 

  349. static void do_audio_out(AVFormatContext *s, 

  350.                          AVOutputStream *ost, 

  351.                          AVInputStream *ist,

  352.                          unsigned char *buf, int size)

  353. {

  354.     uint8_t *buftmp;

  355.     static uint8_t *audio_buf = NULL;

  356.     static uint8_t *audio_out = NULL;

  357.     const int audio_out_size= 4*MAX_AUDIO_PACKET_SIZE;

  358. 

  359.     int size_out, frame_bytes, ret;

  360.     AVCodecContext *enc;

  361. 

  362.     /* SC: dynamic allocation of buffers */

  363.     if (!audio_buf)

  364.         audio_buf = av_malloc(2*MAX_AUDIO_PACKET_SIZE);

  365.     if (!audio_out)

  366.         audio_out = av_malloc(audio_out_size);

  367.     if (!audio_buf || !audio_out)

  368.         return;               /* Should signal an error ! */

  369. 

  370.     

  371.     enc = &ost->st->codec;

  372. 

  373.     if (ost->audio_resample) {

  374.         buftmp = audio_buf;

  375.         size_out = audio_resample(ost->resample, 

  376.                                   (short *)buftmp, (short *)buf,

  377.                                   size / (ist->st->codec.channels * 2));

  378.         size_out = size_out * enc->channels * 2;

  379.     } else {

  380.         buftmp = buf;

  381.         size_out = size;

  382.     }

  383. 

  384.     /* now encode as many frames as possible */

  385.     if (enc->frame_size > 1) {

  386.         /* output resampled raw samples */

  387.         fifo_write(&ost->fifo, buftmp, size_out, 

  388.                    &ost->fifo.wptr);

  389. 

  390.         frame_bytes = enc->frame_size * 2 * enc->channels;

  391.         

  392.         while (fifo_read(&ost->fifo, audio_buf, frame_bytes, 

  393.                      &ost->fifo.rptr) == 0) {

  394.             ret = avcodec_encode_audio(enc, audio_out, audio_out_size, 

  395.                                        (short *)audio_buf);

  396.             av_write_frame(s, ost->index, audio_out, ret);

  397.         }

  398.     } else {

  399.         /* output a pcm frame */

  400.         /* XXX: change encoding codec API to avoid this ? */

  401.         switch(enc->codec->id) {

  402.         case CODEC_ID_PCM_S16LE:

  403.         case CODEC_ID_PCM_S16BE:

  404.         case CODEC_ID_PCM_U16LE:

  405.         case CODEC_ID_PCM_U16BE:

  406.             break;

  407.         default:

  408.             size_out = size_out >> 1;

  409.             break;

  410.         }

  411.         ret = avcodec_encode_audio(enc, audio_out, size_out, 

  412. 				   (short *)buftmp);

  413.         av_write_frame(s, ost->index, audio_out, ret);

  414.     }

  415. }

  416. 

  417. static void pre_process_video_frame(AVInputStream *ist, AVPicture *picture, void **bufp)

  418. {

  419.     AVCodecContext *dec;

  420.     AVPicture *picture2;

  421.     AVPicture picture_tmp;

  422.     uint8_t *buf = 0;

  423. 

  424.     dec = &ist->st->codec;

  425. 

  426.     /* deinterlace : must be done before any resize */

  427.     if (do_deinterlace || using_vhook) {

  428.         int size;

  429. 

  430.         /* create temporary picture */

  431.         size = avpicture_get_size(dec->pix_fmt, dec->width, dec->height);

  432.         buf = av_malloc(size);

  433.         if (!buf)

  434.             return;

  435.         

  436.         picture2 = &picture_tmp;

  437.         avpicture_fill(picture2, buf, dec->pix_fmt, dec->width, dec->height);

  438. 

  439.         if (do_deinterlace){

  440.             if(avpicture_deinterlace(picture2, picture, 

  441.                                      dec->pix_fmt, dec->width, dec->height) < 0) {

  442.                 /* if error, do not deinterlace */

  443.                 av_free(buf);

  444.                 buf = NULL;

  445.                 picture2 = picture;

  446.             }

  447.         } else {

  448.             if (img_convert(picture2, dec->pix_fmt, picture, 

  449.                             dec->pix_fmt, dec->width, dec->height) < 0) {

  450.                 /* if error, do not copy */

  451.                 av_free(buf);

  452.                 buf = NULL;

  453.                 picture2 = picture;

  454.             }

  455.         }

  456.     } else {

  457.         picture2 = picture;

  458.     }

  459. 

  460.     frame_hook_process(picture2, dec->pix_fmt, dec->width, dec->height);

  461. 

  462.     if (picture != picture2)

  463.         *picture = *picture2;

  464.     *bufp = buf;

  465. }

  466. 

  467. /* we begin to correct av delay at this threshold */

  468. #define AV_DELAY_MAX 0.100

  469. 

  470. static void do_video_out(AVFormatContext *s, 

  471.                          AVOutputStream *ost, 

  472.                          AVInputStream *ist,

  473.                          AVFrame *in_picture,

  474.                          int *frame_size, AVOutputStream *audio_sync)

  475. {

  476.     int nb_frames, i, ret;

  477.     AVPicture *final_picture, *formatted_picture;

  478.     AVPicture picture_format_temp, picture_crop_temp;

  479.     static uint8_t *video_buffer;

  480.     uint8_t *buf = NULL, *buf1 = NULL;

  481.     AVCodecContext *enc, *dec;

  482.     enum PixelFormat target_pixfmt;

  483. 

  484. #define VIDEO_BUFFER_SIZE (1024*1024)

  485. 

  486.     enc = &ost->st->codec;

  487.     dec = &ist->st->codec;

  488. 

  489.     /* by default, we output a single frame */

  490.     nb_frames = 1;

  491. 

  492.     *frame_size = 0;

  493. 

  494.     /* NOTE: the A/V sync is always done by considering the audio is

  495.        the master clock. It is suffisant for transcoding or playing,

  496.        but not for the general case */

  497.     if (audio_sync) {

  498.         /* compute the A-V delay and duplicate/remove frames if needed */

  499.         double adelta, vdelta, av_delay;

  500. 

  501.         adelta = audio_sync->sync_ipts - ((double)audio_sync->sync_opts * 

  502.             s->pts_num / s->pts_den);

  503. 

  504.         vdelta = ost->sync_ipts - ((double)ost->sync_opts *

  505.             s->pts_num / s->pts_den);

  506. 

  507.         av_delay = adelta - vdelta;

  508.         //            printf("delay=%f\n", av_delay);

  509.         if (av_delay < -AV_DELAY_MAX)

  510.             nb_frames = 2;

  511.         else if (av_delay > AV_DELAY_MAX)

  512.             nb_frames = 0;

  513.     } else {

  514.         double vdelta;

  515. 

  516.         vdelta = (double)(ost->st->pts.val) * s->pts_num / s->pts_den - (ost->sync_ipts - ost->sync_ipts_offset);

  517.         if (vdelta < 100 && vdelta > -100 && ost->sync_ipts_offset) {

  518.             if (vdelta < -AV_DELAY_MAX)

  519.                 nb_frames = 2;

  520.             else if (vdelta > AV_DELAY_MAX)

  521.                 nb_frames = 0;

  522.         } else {

  523.             ost->sync_ipts_offset -= vdelta;

  524.             if (!ost->sync_ipts_offset)

  525.                 ost->sync_ipts_offset = 0.000001; /* one microsecond */

  526.         }

  527.     }

  528.     

  529. #if defined(AVSYNC_DEBUG)

  530.     {

  531.         static char *action[] = { "drop frame", "copy frame", "dup frame" };

  532.         if (audio_sync)

  533.             fprintf(stderr, "Input APTS %12.6f, output APTS %12.6f, ",

  534.                     (double) audio_sync->sync_ipts, 

  535.                     (double) audio_sync->st->pts.val * s->pts_num / s->pts_den);

  536.         fprintf(stderr, "Input VPTS %12.6f, output VPTS %12.6f: %s\n",

  537.                 (double) ost->sync_ipts, 

  538.                 (double) ost->st->pts.val * s->pts_num / s->pts_den,

  539.                 action[nb_frames]);

  540.     }

  541. #endif

  542. 

  543.     if (nb_frames <= 0) 

  544.         return;

  545. 

  546.     if (!video_buffer)

  547.         video_buffer = av_malloc(VIDEO_BUFFER_SIZE);

  548.     if (!video_buffer)

  549.         return;

  550. 

  551.     /* convert pixel format if needed */

  552.     target_pixfmt = ost->video_resample ? PIX_FMT_YUV420P : enc->pix_fmt;

  553.     if (dec->pix_fmt != target_pixfmt) {

  554.         int size;

  555. 

  556.         /* create temporary picture */

  557.         size = avpicture_get_size(target_pixfmt, dec->width, dec->height);

  558.         buf = av_malloc(size);

  559.         if (!buf)

  560.             return;

  561.         formatted_picture = &picture_format_temp;

  562.         avpicture_fill(formatted_picture, buf, target_pixfmt, dec->width, dec->height);

  563.         

  564.         if (img_convert(formatted_picture, target_pixfmt, 

  565.                         (AVPicture *)in_picture, dec->pix_fmt, 

  566.                         dec->width, dec->height) < 0) {

  567.             fprintf(stderr, "pixel format conversion not handled\n");

  568.             goto the_end;

  569.         }

  570.     } else {

  571.         formatted_picture = (AVPicture *)in_picture;

  572.     }

  573. 

  574.     /* XXX: resampling could be done before raw format convertion in

  575.        some cases to go faster */

  576.     /* XXX: only works for YUV420P */

  577.     if (ost->video_resample) {

  578.         final_picture = &ost->pict_tmp;

  579.         img_resample(ost->img_resample_ctx, final_picture, formatted_picture);

  580. 	if (enc->pix_fmt != PIX_FMT_YUV420P) {

  581.             int size;

  582. 	    

  583. 	    av_free(buf);

  584.             /* create temporary picture */

  585.             size = avpicture_get_size(enc->pix_fmt, enc->width, enc->height);

  586.             buf = av_malloc(size);

  587.             if (!buf)

  588.                 return;

  589.             final_picture = &picture_format_temp;

  590.             avpicture_fill(final_picture, buf, enc->pix_fmt, enc->width, enc->height);

  591.         

  592.             if (img_convert(final_picture, enc->pix_fmt, 

  593.                             &ost->pict_tmp, PIX_FMT_YUV420P, 

  594.                             enc->width, enc->height) < 0) {

  595.                 fprintf(stderr, "pixel format conversion not handled\n");

  596.                 goto the_end;

  597.             }

  598. 	}

  599.     } else if (ost->video_crop) {

  600.         picture_crop_temp.data[0] = formatted_picture->data[0] +

  601.                 (ost->topBand * formatted_picture->linesize[0]) + ost->leftBand;

  602. 

  603.         picture_crop_temp.data[1] = formatted_picture->data[1] +

  604.                 ((ost->topBand >> 1) * formatted_picture->linesize[1]) +

  605.                 (ost->leftBand >> 1);

  606. 

  607.         picture_crop_temp.data[2] = formatted_picture->data[2] +

  608.                 ((ost->topBand >> 1) * formatted_picture->linesize[2]) +

  609.                 (ost->leftBand >> 1);

  610. 

  611.         picture_crop_temp.linesize[0] = formatted_picture->linesize[0];

  612.         picture_crop_temp.linesize[1] = formatted_picture->linesize[1];

  613.         picture_crop_temp.linesize[2] = formatted_picture->linesize[2];

  614.         final_picture = &picture_crop_temp;

  615.     } else {

  616.         final_picture = formatted_picture;

  617.     }

  618.     /* duplicates frame if needed */

  619.     /* XXX: pb because no interleaving */

  620.     for(i=0;i<nb_frames;i++) {

  621.         if (s->oformat->flags & AVFMT_RAWPICTURE) {

  622.             /* raw pictures are written as AVPicture structure to

  623.                avoid any copies. We support temorarily the older

  624.                method. */

  625.             AVFrame* old_frame = enc->coded_frame;

  626. 	    enc->coded_frame = dec->coded_frame;

  627.             av_write_frame(s, ost->index, 

  628.                            (uint8_t *)final_picture, sizeof(AVPicture));

  629. 	    enc->coded_frame = old_frame;

  630.         } else {

  631.             AVFrame big_picture;

  632.             

  633.             memset(&big_picture, 0, sizeof(AVFrame));

  634.             *(AVPicture*)&big_picture= *final_picture;

  635.             /* better than nothing: use input picture interlaced

  636.                settings */

  637.             big_picture.interlaced_frame = in_picture->interlaced_frame;

  638.             big_picture.top_field_first = in_picture->top_field_first;

  639. 

  640.             /* handles sameq here. This is not correct because it may

  641.                not be a global option */

  642.             if (same_quality) {

  643.                 big_picture.quality = ist->st->quality;

  644.             }else

  645.                 big_picture.quality = ost->st->quality;

  646.             

  647.             ret = avcodec_encode_video(enc, 

  648.                                        video_buffer, VIDEO_BUFFER_SIZE,

  649.                                        &big_picture);

  650.             //enc->frame_number = enc->real_pict_num;

  651.             av_write_frame(s, ost->index, video_buffer, ret);

  652.             *frame_size = ret;

  653.             //fprintf(stderr,"\nFrame: %3d %3d size: %5d type: %d",

  654.             //        enc->frame_number-1, enc->real_pict_num, ret,

  655.             //        enc->pict_type);

  656.             /* if two pass, output log */

  657.             if (ost->logfile && enc->stats_out) {

  658.                 fprintf(ost->logfile, "%s", enc->stats_out);

  659.             }

  660.         }

  661.         ost->frame_number++;

  662.     }

  663.  the_end:

  664.     av_free(buf);

  665.     av_free(buf1);

  666. }

  667. 

  668. static double psnr(double d){

  669.     if(d==0) return INFINITY;

  670.     return -10.0*log(d)/log(10.0);

  671. }

  672. 

  673. static void do_video_stats(AVFormatContext *os, AVOutputStream *ost, 

  674.                            int frame_size)

  675. {

  676.     static FILE *fvstats=NULL;

  677.     static int64_t total_size = 0;

  678.     char filename[40];

  679.     time_t today2;

  680.     struct tm *today;

  681.     AVCodecContext *enc;

  682.     int frame_number;

  683.     int64_t ti;

  684.     double ti1, bitrate, avg_bitrate;

  685.     

  686.     if (!fvstats) {

  687.         today2 = time(NULL);

  688.         today = localtime(&today2);

  689.         sprintf(filename, "vstats_%02d%02d%02d.log", today->tm_hour,

  690.                                                today->tm_min,

  691.                                                today->tm_sec);

  692.         fvstats = fopen(filename,"w");

  693.         if (!fvstats) {

  694.             perror("fopen");

  695.             exit(1);

  696.         }

  697.     }

  698.     

  699.     ti = MAXINT64;

  700.     enc = &ost->st->codec;

  701.     total_size += frame_size;

  702.     if (enc->codec_type == CODEC_TYPE_VIDEO) {

  703.         frame_number = ost->frame_number;

  704.         fprintf(fvstats, "frame= %5d q= %2.1f ", frame_number, enc->coded_frame->quality/(float)FF_QP2LAMBDA);

  705.         if (enc->flags&CODEC_FLAG_PSNR)

  706.             fprintf(fvstats, "PSNR= %6.2f ", psnr(enc->coded_frame->error[0]/(enc->width*enc->height*255.0*255.0)));

  707.         

  708.         fprintf(fvstats,"f_size= %6d ", frame_size);

  709.         /* compute pts value */

  710.         ti1 = (double)ost->st->pts.val * os->pts_num / os->pts_den;

  711.         if (ti1 < 0.01)

  712.             ti1 = 0.01;

  713.     

  714.         bitrate = (double)(frame_size * 8) * enc->frame_rate / enc->frame_rate_base / 1000.0;

  715.         avg_bitrate = (double)(total_size * 8) / ti1 / 1000.0;

  716.         fprintf(fvstats, "s_size= %8.0fkB time= %0.3f br= %7.1fkbits/s avg_br= %7.1fkbits/s ",

  717.             (double)total_size / 1024, ti1, bitrate, avg_bitrate);

  718.         fprintf(fvstats,"type= %c\n", av_get_pict_type_char(enc->coded_frame->pict_type));        

  719.     }

  720. }

  721. 

  722. static void print_report(AVFormatContext **output_files,

  723. 			 AVOutputStream **ost_table, int nb_ostreams,

  724. 			 int is_last_report)

  725. {

  726.     char buf[1024];

  727.     AVOutputStream *ost;

  728.     AVFormatContext *oc, *os;

  729.     int64_t total_size;

  730.     AVCodecContext *enc;

  731.     int frame_number, vid, i;

  732.     double bitrate, ti1, pts;

  733.     static int64_t last_time = -1;

  734.     

  735.     if (!is_last_report) {

  736.         int64_t cur_time;

  737.         /* display the report every 0.5 seconds */

  738.         cur_time = av_gettime();

  739.         if (last_time == -1) {

  740.             last_time = cur_time;

  741.             return;

  742.         } 

  743.         if ((cur_time - last_time) < 500000)

  744.             return;

  745.         last_time = cur_time;

  746.     }

  747. 

  748. 

  749.     oc = output_files[0];

  750. 

  751.     total_size = url_ftell(&oc->pb);

  752.     

  753.     buf[0] = '\0';

  754.     ti1 = 1e10;

  755.     vid = 0;

  756.     for(i=0;i<nb_ostreams;i++) {

  757.         ost = ost_table[i];

  758.         os = output_files[ost->file_index];

  759.         enc = &ost->st->codec;

  760.         if (vid && enc->codec_type == CODEC_TYPE_VIDEO) {

  761.             sprintf(buf + strlen(buf), "q=%2.1f ",

  762.                     enc->coded_frame->quality/(float)FF_QP2LAMBDA);

  763.         }

  764.         if (!vid && enc->codec_type == CODEC_TYPE_VIDEO) {

  765.             frame_number = ost->frame_number;

  766.             sprintf(buf + strlen(buf), "frame=%5d q=%2.1f ",

  767.                     frame_number, enc->coded_frame ? enc->coded_frame->quality/(float)FF_QP2LAMBDA : 0);

  768.             if(is_last_report)

  769.                 sprintf(buf + strlen(buf), "L");

  770.             if (enc->flags&CODEC_FLAG_PSNR){

  771.                 int j;

  772.                 double error, error_sum=0;

  773.                 double scale, scale_sum=0;

  774.                 char type[3]= {'Y','U','V'};

  775.                 sprintf(buf + strlen(buf), "PSNR=");

  776.                 for(j=0; j<3; j++){

  777.                     if(is_last_report){

  778.                         error= enc->error[j];

  779.                         scale= enc->width*enc->height*255.0*255.0*frame_number;

  780.                     }else{

  781.                         error= enc->coded_frame->error[j];

  782.                         scale= enc->width*enc->height*255.0*255.0;

  783.                     }

  784.                     if(j) scale/=4;

  785.                     error_sum += error;

  786.                     scale_sum += scale;

  787.                     sprintf(buf + strlen(buf), "%c:%2.2f ", type[j], psnr(error/scale));

  788.                 }

  789.                 sprintf(buf + strlen(buf), "*:%2.2f ", psnr(error_sum/scale_sum));

  790.             }

  791.             vid = 1;

  792.         }

  793.         /* compute min output value */

  794.         pts = (double)ost->st->pts.val * os->pts_num / os->pts_den;

  795.         if ((pts < ti1) && (pts > 0))

  796.             ti1 = pts;

  797.     }

  798.     if (ti1 < 0.01)

  799.         ti1 = 0.01;

  800.     

  801.     if (verbose || is_last_report) {

  802.         bitrate = (double)(total_size * 8) / ti1 / 1000.0;

  803.         

  804.         sprintf(buf + strlen(buf), 

  805.             "size=%8.0fkB time=%0.1f bitrate=%6.1fkbits/s",

  806.             (double)total_size / 1024, ti1, bitrate);

  807.         

  808.         fprintf(stderr, "%s    \r", buf);

  809.         fflush(stderr);

  810.     }

  811.         

  812.     if (is_last_report)

  813.         fprintf(stderr, "\n");

  814. }

  815. 

  816. /* pkt = NULL means EOF (needed to flush decoder buffers) */

  817. static int output_packet(AVInputStream *ist, int ist_index,

  818.                          AVOutputStream **ost_table, int nb_ostreams,

  819.                          AVPacket *pkt)

  820. {

  821.     AVFormatContext *os;

  822.     AVOutputStream *ost;

  823.     uint8_t *ptr;

  824.     int len, ret, i;

  825.     uint8_t *data_buf;

  826.     int data_size, got_picture;

  827.     AVFrame picture;

  828.     short samples[AVCODEC_MAX_AUDIO_FRAME_SIZE / 2];

  829.     void *buffer_to_free;

  830.         

  831.     if (pkt && pkt->pts != AV_NOPTS_VALUE) {

  832.         ist->pts = pkt->pts;

  833.     } else {

  834.         ist->pts = ist->next_pts;

  835.     }

  836.     

  837.     if (pkt == NULL) {

  838.         /* EOF handling */

  839.         ptr = NULL;

  840.         len = 0;

  841.         goto handle_eof;

  842.     }

  843. 

  844.     len = pkt->size;

  845.     ptr = pkt->data;

  846.     while (len > 0) {

  847.     handle_eof:

  848.         /* decode the packet if needed */

  849.         data_buf = NULL; /* fail safe */

  850.         data_size = 0;

  851.         if (ist->decoding_needed) {

  852.             switch(ist->st->codec.codec_type) {

  853.             case CODEC_TYPE_AUDIO:

  854.                     /* XXX: could avoid copy if PCM 16 bits with same

  855.                        endianness as CPU */

  856.                 ret = avcodec_decode_audio(&ist->st->codec, samples, &data_size,

  857.                                            ptr, len);

  858.                 if (ret < 0)

  859.                     goto fail_decode;

  860.                 ptr += ret;

  861.                 len -= ret;

  862.                 /* Some bug in mpeg audio decoder gives */

  863.                 /* data_size < 0, it seems they are overflows */

  864.                 if (data_size <= 0) {

  865.                     /* no audio frame */

  866.                     continue;

  867.                 }

  868.                 data_buf = (uint8_t *)samples;

  869.                 ist->next_pts += ((int64_t)AV_TIME_BASE * data_size) / 

  870.                     (2 * ist->st->codec.channels);

  871.                 break;

  872.             case CODEC_TYPE_VIDEO:

  873.                     data_size = (ist->st->codec.width * ist->st->codec.height * 3) / 2;

  874.                     /* XXX: allocate picture correctly */

  875.                     memset(&picture, 0, sizeof(picture));

  876.                     ret = avcodec_decode_video(&ist->st->codec, 

  877.                                                &picture, &got_picture, ptr, len);

  878.                     ist->st->quality= picture.quality;

  879.                     if (ret < 0) 

  880.                         goto fail_decode;

  881.                     if (!got_picture) {

  882.                         /* no picture yet */

  883.                         goto discard_packet;

  884.                     }

  885.                     if (ist->st->codec.frame_rate_base != 0) {

  886.                         ist->next_pts += ((int64_t)AV_TIME_BASE * 

  887.                                           ist->st->codec.frame_rate_base) /

  888.                             ist->st->codec.frame_rate;

  889.                     }

  890.                     len = 0;

  891.                     break;

  892.                 default:

  893.                     goto fail_decode;

  894.                 }

  895.             } else {

  896.                 data_buf = ptr;

  897.                 data_size = len;

  898.                 ret = len;

  899.                 len = 0;

  900.             }

  901. 

  902.             buffer_to_free = NULL;

  903.             if (ist->st->codec.codec_type == CODEC_TYPE_VIDEO) {

  904.                 pre_process_video_frame(ist, (AVPicture *)&picture, 

  905.                                         &buffer_to_free);

  906.             }

  907. 

  908.             /* frame rate emulation */

  909.             if (ist->st->codec.rate_emu) {

  910.                 int64_t pts = av_rescale((int64_t) ist->frame * ist->st->codec.frame_rate_base, 1000000, ist->st->codec.frame_rate);

  911.                 int64_t now = av_gettime() - ist->start;

  912.                 if (pts > now)

  913.                     usleep(pts - now);

  914. 

  915.                 ist->frame++;

  916.             }

  917. 

  918. #if 0

  919.             /* mpeg PTS deordering : if it is a P or I frame, the PTS

  920.                is the one of the next displayed one */

  921.             /* XXX: add mpeg4 too ? */

  922.             if (ist->st->codec.codec_id == CODEC_ID_MPEG1VIDEO) {

  923.                 if (ist->st->codec.pict_type != B_TYPE) {

  924.                     int64_t tmp;

  925.                     tmp = ist->last_ip_pts;

  926.                     ist->last_ip_pts  = ist->frac_pts.val;

  927.                     ist->frac_pts.val = tmp;

  928.                 }

  929.             }

  930. #endif

  931.             /* if output time reached then transcode raw format, 

  932. 	       encode packets and output them */

  933.             if (start_time == 0 || ist->pts >= start_time)

  934.                 for(i=0;i<nb_ostreams;i++) {

  935.                     int frame_size;

  936. 

  937.                     ost = ost_table[i];

  938.                     if (ost->source_index == ist_index) {

  939.                         os = output_files[ost->file_index];

  940. 

  941. #if 0

  942.                         printf("%d: got pts=%0.3f %0.3f\n", i, 

  943.                                (double)pkt->pts / AV_TIME_BASE, 

  944.                                ((double)ist->pts / AV_TIME_BASE) - 

  945.                                ((double)ost->st->pts.val * os->pts_num / os->pts_den));

  946. #endif

  947.                         /* set the input output pts pairs */

  948.                         ost->sync_ipts = (double)ist->pts / AV_TIME_BASE;

  949.                         /* XXX: take into account the various fifos,

  950.                            in particular for audio */

  951.                         ost->sync_opts = ost->st->pts.val;

  952.                         //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); 

  953. 

  954.                         if (ost->encoding_needed) {

  955.                             switch(ost->st->codec.codec_type) {

  956.                             case CODEC_TYPE_AUDIO:

  957.                                 do_audio_out(os, ost, ist, data_buf, data_size);

  958.                                 break;

  959.                             case CODEC_TYPE_VIDEO:

  960.                                 /* find an audio stream for synchro */

  961.                                 {

  962.                                     int i;

  963.                                     AVOutputStream *audio_sync, *ost1;

  964.                                     audio_sync = NULL;

  965.                                     for(i=0;i<nb_ostreams;i++) {

  966.                                         ost1 = ost_table[i];

  967.                                         if (ost1->file_index == ost->file_index &&

  968.                                             ost1->st->codec.codec_type == CODEC_TYPE_AUDIO) {

  969.                                             audio_sync = ost1;

  970.                                             break;

  971.                                         }

  972.                                     }

  973. 

  974.                                     do_video_out(os, ost, ist, &picture, &frame_size, audio_sync);

  975.                                     if (do_vstats && frame_size)

  976.                                         do_video_stats(os, ost, frame_size);

  977.                                 }

  978.                                 break;

  979.                             default:

  980.                                 av_abort();

  981.                             }

  982.                         } else {

  983.                             AVFrame avframe;

  984.                                                 

  985.                             /* no reencoding needed : output the packet directly */

  986.                             /* force the input stream PTS */

  987.                         

  988.                             memset(&avframe, 0, sizeof(AVFrame));

  989.                             ost->st->codec.coded_frame= &avframe;

  990.                             avframe.key_frame = pkt->flags & PKT_FLAG_KEY; 

  991.                         

  992.                             av_write_frame(os, ost->index, data_buf, data_size);

  993.                             ost->st->codec.frame_number++;

  994.                             ost->frame_number++;

  995.                         }

  996.                     }

  997.                 }

  998.             av_free(buffer_to_free);

  999.         }

  1000.  discard_packet:

  1001.     return 0;

  1002.  fail_decode:

  1003.     return -1;

  1004. }

  1005. 

  1006. 

  1007. /*

  1008.  * The following code is the main loop of the file converter

  1009.  */

  1010. static int av_encode(AVFormatContext **output_files,

  1011.                      int nb_output_files,

  1012.                      AVFormatContext **input_files,

  1013.                      int nb_input_files,

  1014.                      AVStreamMap *stream_maps, int nb_stream_maps)

  1015. {

  1016.     int ret, i, j, k, n, nb_istreams = 0, nb_ostreams = 0;

  1017.     AVFormatContext *is, *os;

  1018.     AVCodecContext *codec, *icodec;

  1019.     AVOutputStream *ost, **ost_table = NULL;

  1020.     AVInputStream *ist, **ist_table = NULL;

  1021.     AVInputFile *file_table;

  1022.     AVFormatContext *stream_no_data;

  1023.     int key;

  1024. 

  1025.     file_table= (AVInputFile*) av_mallocz(nb_input_files * sizeof(AVInputFile));

  1026.     if (!file_table)

  1027.         goto fail;

  1028. 

  1029.     /* input stream init */

  1030.     j = 0;

  1031.     for(i=0;i<nb_input_files;i++) {

  1032.         is = input_files[i];

  1033.         file_table[i].ist_index = j;

  1034.         file_table[i].nb_streams = is->nb_streams;

  1035.         j += is->nb_streams;

  1036.     }

  1037.     nb_istreams = j;

  1038. 

  1039.     ist_table = av_mallocz(nb_istreams * sizeof(AVInputStream *));

  1040.     if (!ist_table)

  1041.         goto fail;

  1042.     

  1043.     for(i=0;i<nb_istreams;i++) {

  1044.         ist = av_mallocz(sizeof(AVInputStream));

  1045.         if (!ist)

  1046.             goto fail;

  1047.         ist_table[i] = ist;

  1048.     }

  1049.     j = 0;

  1050.     for(i=0;i<nb_input_files;i++) {

  1051.         is = input_files[i];

  1052.         for(k=0;k<is->nb_streams;k++) {

  1053.             ist = ist_table[j++];

  1054.             ist->st = is->streams[k];

  1055.             ist->file_index = i;

  1056.             ist->index = k;

  1057.             ist->discard = 1; /* the stream is discarded by default

  1058.                                  (changed later) */

  1059. 

  1060.             if (ist->st->codec.rate_emu) {

  1061.                 ist->start = av_gettime();

  1062.                 ist->frame = 0;

  1063.             }

  1064.         }

  1065.     }

  1066. 

  1067.     /* output stream init */

  1068.     nb_ostreams = 0;

  1069.     for(i=0;i<nb_output_files;i++) {

  1070.         os = output_files[i];

  1071.         nb_ostreams += os->nb_streams;

  1072.     }

  1073.     if (nb_stream_maps > 0 && nb_stream_maps != nb_ostreams) {

  1074.         fprintf(stderr, "Number of stream maps must match number of output streams\n");

  1075.         exit(1);

  1076.     }

  1077. 

  1078.     /* Sanity check the mapping args -- do the input files & streams exist? */

  1079.     for(i=0;i<nb_stream_maps;i++) {

  1080.         int fi = stream_maps[i].file_index;

  1081.         int si = stream_maps[i].stream_index;

  1082.         

  1083.         if (fi < 0 || fi > nb_input_files - 1 ||

  1084.             si < 0 || si > file_table[fi].nb_streams - 1) {

  1085.             fprintf(stderr,"Could not find input stream #%d.%d\n", fi, si);

  1086.             exit(1);

  1087.         }

  1088.     }

  1089.     

  1090.     ost_table = av_mallocz(sizeof(AVOutputStream *) * nb_ostreams);

  1091.     if (!ost_table)

  1092.         goto fail;

  1093.     for(i=0;i<nb_ostreams;i++) {

  1094.         ost = av_mallocz(sizeof(AVOutputStream));

  1095.         if (!ost)

  1096.             goto fail;

  1097.         ost_table[i] = ost;

  1098.     }

  1099.     

  1100.     n = 0;

  1101.     for(k=0;k<nb_output_files;k++) {

  1102.         os = output_files[k];

  1103.         for(i=0;i<os->nb_streams;i++) {

  1104.             int found;

  1105.             ost = ost_table[n++];

  1106.             ost->file_index = k;

  1107.             ost->index = i;

  1108.             ost->st = os->streams[i];

  1109.             if (nb_stream_maps > 0) {

  1110.                 ost->source_index = file_table[stream_maps[n-1].file_index].ist_index + 

  1111.                     stream_maps[n-1].stream_index;

  1112.                     

  1113.                 /* Sanity check that the stream types match */

  1114.                 if (ist_table[ost->source_index]->st->codec.codec_type != ost->st->codec.codec_type) {

  1115.                     fprintf(stderr, "Codec type mismatch for mapping #%d.%d -> #%d.%d\n",

  1116.                         stream_maps[n-1].file_index, stream_maps[n-1].stream_index,

  1117.                         ost->file_index, ost->index);

  1118.                     exit(1);

  1119.                 }

  1120.                 

  1121.             } else {

  1122.                 /* get corresponding input stream index : we select the first one with the right type */

  1123.                 found = 0;

  1124.                 for(j=0;j<nb_istreams;j++) {

  1125.                     ist = ist_table[j];

  1126.                     if (ist->discard && 

  1127.                         ist->st->codec.codec_type == ost->st->codec.codec_type) {

  1128.                         ost->source_index = j;

  1129.                         found = 1;

  1130.                     }

  1131.                 }

  1132.                 

  1133.                 if (!found) {

  1134.                     /* try again and reuse existing stream */

  1135.                     for(j=0;j<nb_istreams;j++) {

  1136.                         ist = ist_table[j];

  1137.                         if (ist->st->codec.codec_type == ost->st->codec.codec_type) {

  1138.                             ost->source_index = j;

  1139.                             found = 1;

  1140.                         }

  1141.                     }

  1142.                     if (!found) {

  1143.                         fprintf(stderr, "Could not find input stream matching output stream #%d.%d\n",

  1144.                                 ost->file_index, ost->index);

  1145.                         exit(1);

  1146.                     }

  1147.                 }

  1148.             }

  1149.             ist = ist_table[ost->source_index];

  1150.             ist->discard = 0;

  1151.         }

  1152.     }

  1153. 

  1154.     /* for each output stream, we compute the right encoding parameters */

  1155.     for(i=0;i<nb_ostreams;i++) {

  1156.         ost = ost_table[i];

  1157.         ist = ist_table[ost->source_index];

  1158. 

  1159.         codec = &ost->st->codec;

  1160.         icodec = &ist->st->codec;

  1161. 

  1162.         if (ost->st->stream_copy) {

  1163.             /* if stream_copy is selected, no need to decode or encode */

  1164.             codec->codec_id = icodec->codec_id;

  1165.             codec->codec_type = icodec->codec_type;

  1166.             codec->codec_tag = icodec->codec_tag;

  1167.             codec->bit_rate = icodec->bit_rate;

  1168.             switch(codec->codec_type) {

  1169.             case CODEC_TYPE_AUDIO:

  1170.                 codec->sample_rate = icodec->sample_rate;

  1171.                 codec->channels = icodec->channels;

  1172.                 break;

  1173.             case CODEC_TYPE_VIDEO:

  1174.                 codec->frame_rate = icodec->frame_rate;

  1175.                 codec->frame_rate_base = icodec->frame_rate_base;

  1176.                 codec->width = icodec->width;

  1177.                 codec->height = icodec->height;

  1178.                 break;

  1179.             default:

  1180.                 av_abort();

  1181.             }

  1182.         } else {

  1183.             switch(codec->codec_type) {

  1184.             case CODEC_TYPE_AUDIO:

  1185.                 if (fifo_init(&ost->fifo, 2 * MAX_AUDIO_PACKET_SIZE))

  1186.                     goto fail;

  1187.                 

  1188.                 if (codec->channels == icodec->channels &&

  1189.                     codec->sample_rate == icodec->sample_rate) {

  1190.                     ost->audio_resample = 0;

  1191.                 } else {

  1192.                     if (codec->channels != icodec->channels &&

  1193.                         icodec->codec_id == CODEC_ID_AC3) {

  1194.                         /* Special case for 5:1 AC3 input */

  1195.                         /* and mono or stereo output      */

  1196.                         /* Request specific number of channels */

  1197.                         icodec->channels = codec->channels;

  1198.                         if (codec->sample_rate == icodec->sample_rate)

  1199.                             ost->audio_resample = 0;

  1200.                         else {

  1201.                             ost->audio_resample = 1;

  1202.                             ost->resample = audio_resample_init(codec->channels, icodec->channels,

  1203.                                                         codec->sample_rate, 

  1204.                                                         icodec->sample_rate);

  1205. 			    if(!ost->resample)

  1206. 			      {

  1207. 				printf("Can't resample.  Aborting.\n");

  1208. 				av_abort();

  1209. 			      }

  1210.                         }

  1211.                         /* Request specific number of channels */

  1212.                         icodec->channels = codec->channels;

  1213.                     } else {

  1214.                         ost->audio_resample = 1; 

  1215.                         ost->resample = audio_resample_init(codec->channels, icodec->channels,

  1216.                                                         codec->sample_rate, 

  1217.                                                         icodec->sample_rate);

  1218. 			if(!ost->resample)

  1219. 			  {

  1220. 			    printf("Can't resample.  Aborting.\n");

  1221. 			    av_abort();

  1222. 			  }

  1223.                     }

  1224.                 }

  1225.                 ist->decoding_needed = 1;

  1226.                 ost->encoding_needed = 1;

  1227.                 break;

  1228.             case CODEC_TYPE_VIDEO:

  1229.                 if (codec->width == icodec->width &&

  1230.                     codec->height == icodec->height &&

  1231.                     frame_topBand == 0 &&

  1232.                     frame_bottomBand == 0 &&

  1233.                     frame_leftBand == 0 &&

  1234.                     frame_rightBand == 0)

  1235.                 {

  1236.                     ost->video_resample = 0;

  1237.                     ost->video_crop = 0;

  1238.                 } else if ((codec->width == icodec->width -

  1239.                                 (frame_leftBand + frame_rightBand)) &&

  1240.                         (codec->height == icodec->height -

  1241.                                 (frame_topBand  + frame_bottomBand)))

  1242.                 {

  1243.                     ost->video_resample = 0;

  1244.                     ost->video_crop = 1;

  1245.                     ost->topBand = frame_topBand;

  1246.                     ost->leftBand = frame_leftBand;

  1247.                 } else {

  1248.                     uint8_t *buf;

  1249.                     ost->video_resample = 1;

  1250.                     ost->video_crop = 0; // cropping is handled as part of resample

  1251.                     buf = av_malloc((codec->width * codec->height * 3) / 2);

  1252.                     if (!buf)

  1253.                         goto fail;

  1254.                     ost->pict_tmp.data[0] = buf;

  1255.                     ost->pict_tmp.data[1] = ost->pict_tmp.data[0] + (codec->width * codec->height);

  1256.                     ost->pict_tmp.data[2] = ost->pict_tmp.data[1] + (codec->width * codec->height) / 4;

  1257.                     ost->pict_tmp.linesize[0] = codec->width;

  1258.                     ost->pict_tmp.linesize[1] = codec->width / 2;

  1259.                     ost->pict_tmp.linesize[2] = codec->width / 2;

  1260. 

  1261.                     ost->img_resample_ctx = img_resample_full_init( 

  1262.                                       ost->st->codec.width, ost->st->codec.height,

  1263.                                       ist->st->codec.width, ist->st->codec.height,

  1264.                                       frame_topBand, frame_bottomBand,

  1265.                                       frame_leftBand, frame_rightBand);

  1266.                 }

  1267.                 ost->encoding_needed = 1;

  1268.                 ist->decoding_needed = 1;

  1269.                 break;

  1270.             default:

  1271.                 av_abort();

  1272.             }

  1273.             /* two pass mode */

  1274.             if (ost->encoding_needed && 

  1275.                 (codec->flags & (CODEC_FLAG_PASS1 | CODEC_FLAG_PASS2))) {

  1276.                 char logfilename[1024];

  1277.                 FILE *f;

  1278.                 int size;

  1279.                 char *logbuffer;

  1280.                 

  1281.                 snprintf(logfilename, sizeof(logfilename), "%s-%d.log", 

  1282.                          pass_logfilename ? 

  1283.                          pass_logfilename : DEFAULT_PASS_LOGFILENAME, i);

  1284.                 if (codec->flags & CODEC_FLAG_PASS1) {

  1285.                     f = fopen(logfilename, "w");

  1286.                     if (!f) {

  1287.                         perror(logfilename);

  1288.                         exit(1);

  1289.                     }

  1290.                     ost->logfile = f;

  1291.                 } else {

  1292.                     /* read the log file */

  1293.                     f = fopen(logfilename, "r");

  1294.                     if (!f) {

  1295.                         perror(logfilename);

  1296.                         exit(1);

  1297.                     }

  1298.                     fseek(f, 0, SEEK_END);

  1299.                     size = ftell(f);

  1300.                     fseek(f, 0, SEEK_SET);

  1301.                     logbuffer = av_malloc(size + 1);

  1302.                     if (!logbuffer) {

  1303.                         fprintf(stderr, "Could not allocate log buffer\n");

  1304.                         exit(1);

  1305.                     }

  1306.                     fread(logbuffer, 1, size, f);

  1307.                     fclose(f);

  1308.                     logbuffer[size] = '\0';

  1309.                     codec->stats_in = logbuffer;

  1310.                 }

  1311.             }

  1312.         }

  1313.     }

  1314. 

  1315.     /* dump the file output parameters - cannot be done before in case

  1316.        of stream copy */

  1317.     for(i=0;i<nb_output_files;i++) {

  1318.         dump_format(output_files[i], i, output_files[i]->filename, 1);

  1319.     }

  1320. 

  1321.     /* dump the stream mapping */

  1322.     fprintf(stderr, "Stream mapping:\n");

  1323.     for(i=0;i<nb_ostreams;i++) {

  1324.         ost = ost_table[i];

  1325.         fprintf(stderr, "  Stream #%d.%d -> #%d.%d\n",

  1326.                 ist_table[ost->source_index]->file_index,

  1327.                 ist_table[ost->source_index]->index,

  1328.                 ost->file_index, 

  1329.                 ost->index);

  1330.     }

  1331. 

  1332.     /* open each encoder */

  1333.     for(i=0;i<nb_ostreams;i++) {

  1334.         ost = ost_table[i];

  1335.         if (ost->encoding_needed) {

  1336.             AVCodec *codec;

  1337.             codec = avcodec_find_encoder(ost->st->codec.codec_id);

  1338.             if (!codec) {

  1339.                 fprintf(stderr, "Unsupported codec for output stream #%d.%d\n", 

  1340.                         ost->file_index, ost->index);

  1341.                 exit(1);

  1342.             }

  1343.             if (avcodec_open(&ost->st->codec, codec) < 0) {

  1344.                 fprintf(stderr, "Error while opening codec for stream #%d.%d - maybe incorrect parameters such as bit_rate, rate, width or height\n", 

  1345.                         ost->file_index, ost->index);

  1346.                 exit(1);

  1347.             }

  1348.         }

  1349.     }

  1350. 

  1351.     /* open each decoder */

  1352.     for(i=0;i<nb_istreams;i++) {

  1353.         ist = ist_table[i];

  1354.         if (ist->decoding_needed) {

  1355.             AVCodec *codec;

  1356.             codec = avcodec_find_decoder(ist->st->codec.codec_id);

  1357.             if (!codec) {

  1358.                 fprintf(stderr, "Unsupported codec (id=%d) for input stream #%d.%d\n", 

  1359.                         ist->st->codec.codec_id, ist->file_index, ist->index);

  1360.                 exit(1);

  1361.             }

  1362.             if (avcodec_open(&ist->st->codec, codec) < 0) {

  1363.                 fprintf(stderr, "Error while opening codec for input stream #%d.%d\n", 

  1364.                         ist->file_index, ist->index);

  1365.                 exit(1);

  1366.             }

  1367.             //if (ist->st->codec.codec_type == CODEC_TYPE_VIDEO)

  1368.             //    ist->st->codec.flags |= CODEC_FLAG_REPEAT_FIELD;

  1369.         }

  1370.     }

  1371. 

  1372.     /* init pts */

  1373.     for(i=0;i<nb_istreams;i++) {

  1374.         ist = ist_table[i];

  1375. 	is = input_files[ist->file_index];

  1376.         ist->pts = 0;

  1377.         ist->next_pts = 0;

  1378.     }

  1379.     

  1380.     /* compute buffer size max (should use a complete heuristic) */

  1381.     for(i=0;i<nb_input_files;i++) {

  1382.         file_table[i].buffer_size_max = 2048;

  1383.     }

  1384. 

  1385.     /* open files and write file headers */

  1386.     for(i=0;i<nb_output_files;i++) {

  1387.         os = output_files[i];

  1388.         if (av_write_header(os) < 0) {

  1389.             fprintf(stderr, "Could not write header for output file #%d (incorrect codec paramters ?)\n", i);

  1390.             ret = -EINVAL;

  1391.             goto fail;

  1392.         }

  1393.     }

  1394. 

  1395. #ifndef CONFIG_WIN32

  1396.     if ( !using_stdin )

  1397.         fprintf(stderr, "Press [q] to stop encoding\n");

  1398. #endif

  1399.     term_init();

  1400. 

  1401.     stream_no_data = 0;

  1402.     key = -1;

  1403. 

  1404.     for(; received_sigterm == 0;) {

  1405.         int file_index, ist_index;

  1406.         AVPacket pkt;

  1407.         double pts_min;

  1408.         

  1409.     redo:

  1410.         /* if 'q' pressed, exits */

  1411.         if (!using_stdin) {

  1412.             /* read_key() returns 0 on EOF */

  1413.             key = read_key();

  1414.             if (key == 'q')

  1415.                 break;

  1416.         }

  1417. 

  1418.         /* select the stream that we must read now by looking at the

  1419.            smallest output pts */

  1420.         file_index = -1;

  1421.         pts_min = 1e10;

  1422.         for(i=0;i<nb_ostreams;i++) {

  1423.             double pts;

  1424.             ost = ost_table[i];

  1425.             os = output_files[ost->file_index];

  1426.             ist = ist_table[ost->source_index];

  1427.             pts = (double)ost->st->pts.val * os->pts_num / os->pts_den;

  1428.             if (!file_table[ist->file_index].eof_reached && 

  1429.                 pts < pts_min) {

  1430.                 pts_min = pts;

  1431.                 file_index = ist->file_index;

  1432.             }

  1433.         }

  1434.         /* if none, if is finished */

  1435.         if (file_index < 0) {

  1436.             break;

  1437.         }

  1438. 

  1439.         /* finish if recording time exhausted */

  1440.         if (recording_time > 0 && pts_min >= (recording_time / 1000000.0))

  1441.             break;

  1442. 

  1443.         /* read a frame from it and output it in the fifo */

  1444.         is = input_files[file_index];

  1445.         if (av_read_frame(is, &pkt) < 0) {

  1446.             file_table[file_index].eof_reached = 1;

  1447.             continue;

  1448.         }

  1449.         if (!pkt.size) {

  1450.             stream_no_data = is;

  1451.         } else {

  1452.             stream_no_data = 0;

  1453.         }

  1454.         if (do_pkt_dump) {

  1455.             av_pkt_dump(stdout, &pkt, do_hex_dump);

  1456.         }

  1457.         /* the following test is needed in case new streams appear

  1458.            dynamically in stream : we ignore them */

  1459.         if (pkt.stream_index >= file_table[file_index].nb_streams)

  1460.             goto discard_packet;

  1461.         ist_index = file_table[file_index].ist_index + pkt.stream_index;

  1462.         ist = ist_table[ist_index];

  1463.         if (ist->discard)

  1464.             goto discard_packet;

  1465. 

  1466.         //fprintf(stderr,"read #%d.%d size=%d\n", ist->file_index, ist->index, pkt.size);

  1467.         if (output_packet(ist, ist_index, ost_table, nb_ostreams, &pkt) < 0) {

  1468.             fprintf(stderr, "Error while decoding stream #%d.%d\n",

  1469.                     ist->file_index, ist->index);

  1470.             av_free_packet(&pkt);

  1471.             goto redo;

  1472.         }

  1473.         

  1474.     discard_packet:

  1475.         av_free_packet(&pkt);

  1476.         

  1477.         /* dump report by using the output first video and audio streams */

  1478.         print_report(output_files, ost_table, nb_ostreams, 0);

  1479.     }

  1480. 

  1481.     /* at the end of stream, we must flush the decoder buffers */

  1482.     for(i=0;i<nb_istreams;i++) {

  1483.         ist = ist_table[i];

  1484.         if (ist->decoding_needed) {

  1485.             output_packet(ist, i, ost_table, nb_ostreams, NULL);

  1486.         }

  1487.     }

  1488. 

  1489.     term_exit();

  1490. 

  1491.     /* dump report by using the first video and audio streams */

  1492.     print_report(output_files, ost_table, nb_ostreams, 1);

  1493. 

  1494.     /* close each encoder */

  1495.     for(i=0;i<nb_ostreams;i++) {

  1496.         ost = ost_table[i];

  1497.         if (ost->encoding_needed) {

  1498.             av_freep(&ost->st->codec.stats_in);

  1499.             avcodec_close(&ost->st->codec);

  1500.         }

  1501.     }

  1502.     

  1503.     /* close each decoder */

  1504.     for(i=0;i<nb_istreams;i++) {

  1505.         ist = ist_table[i];

  1506.         if (ist->decoding_needed) {

  1507.             avcodec_close(&ist->st->codec);

  1508.         }

  1509.     }

  1510.     

  1511. 

  1512.     /* write the trailer if needed and close file */

  1513.     for(i=0;i<nb_output_files;i++) {

  1514.         os = output_files[i];

  1515.         av_write_trailer(os);

  1516.     }

  1517.     /* finished ! */

  1518.     

  1519.     ret = 0;

  1520.  fail1:

  1521.     av_free(file_table);

  1522. 

  1523.     if (ist_table) {

  1524.         for(i=0;i<nb_istreams;i++) {

  1525.             ist = ist_table[i];

  1526.             av_free(ist);

  1527.         }

  1528.         av_free(ist_table);

  1529.     }

  1530.     if (ost_table) {

  1531.         for(i=0;i<nb_ostreams;i++) {

  1532.             ost = ost_table[i];

  1533.             if (ost) {

  1534.                 if (ost->logfile) {

  1535.                     fclose(ost->logfile);

  1536.                     ost->logfile = NULL;

  1537.                 }

  1538.                 fifo_free(&ost->fifo); /* works even if fifo is not

  1539.                                           initialized but set to zero */

  1540.                 av_free(ost->pict_tmp.data[0]);

  1541.                 if (ost->video_resample)

  1542.                     img_resample_close(ost->img_resample_ctx);

  1543.                 if (ost->audio_resample)

  1544.                     audio_resample_close(ost->resample);

  1545.                 av_free(ost);

  1546.             }

  1547.         }

  1548.         av_free(ost_table);

  1549.     }

  1550.     return ret;

  1551.  fail:

  1552.     ret = -ENOMEM;

  1553.     goto fail1;

  1554. }

  1555. 

  1556. #if 0

  1557. int file_read(const char *filename)

  1558. {

  1559.     URLContext *h;

  1560.     unsigned char buffer[1024];

  1561.     int len, i;

  1562. 

  1563.     if (url_open(&h, filename, O_RDONLY) < 0) {

  1564.         printf("could not open '%s'\n", filename);

  1565.         return -1;

  1566.     }

  1567.     for(;;) {

  1568.         len = url_read(h, buffer, sizeof(buffer));

  1569.         if (len <= 0)

  1570.             break;

  1571.         for(i=0;i<len;i++) putchar(buffer[i]);

  1572.     }

  1573.     url_close(h);

  1574.     return 0;

  1575. }

  1576. #endif

  1577. 

  1578. static void opt_image_format(const char *arg)

  1579. {

  1580.     AVImageFormat *f;

  1581.     

  1582.     for(f = first_image_format; f != NULL; f = f->next) {

  1583.         if (!strcmp(arg, f->name))

  1584.             break;

  1585.     }

  1586.     if (!f) {

  1587.         fprintf(stderr, "Unknown image format: '%s'\n", arg);

  1588.         exit(1);

  1589.     }

  1590.     image_format = f;

  1591. }

  1592. 

  1593. static void opt_format(const char *arg)

  1594. {

  1595.     /* compatibility stuff for pgmyuv */

  1596.     if (!strcmp(arg, "pgmyuv")) {

  1597.         opt_image_format(arg);

  1598.         arg = "image";

  1599.     }

  1600. 

  1601.     file_iformat = av_find_input_format(arg);

  1602.     file_oformat = guess_format(arg, NULL, NULL);

  1603.     if (!file_iformat && !file_oformat) {

  1604.         fprintf(stderr, "Unknown input or output format: %s\n", arg);

  1605.         exit(1);

  1606.     }

  1607. }

  1608. 

  1609. static void opt_video_bitrate(const char *arg)

  1610. {

  1611.     video_bit_rate = atoi(arg) * 1000;

  1612. }

  1613. 

  1614. static void opt_video_bitrate_tolerance(const char *arg)

  1615. {

  1616.     video_bit_rate_tolerance = atoi(arg) * 1000;

  1617. }

  1618. 

  1619. static void opt_video_bitrate_max(const char *arg)

  1620. {

  1621.     video_rc_max_rate = atoi(arg) * 1000;

  1622. }

  1623. 

  1624. static void opt_video_bitrate_min(const char *arg)

  1625. {

  1626.     video_rc_min_rate = atoi(arg) * 1000;

  1627. }

  1628. 

  1629. static void opt_video_buffer_size(const char *arg)

  1630. {

  1631.     video_rc_buffer_size = atoi(arg) * 1024;

  1632. }

  1633. 

  1634. static void opt_video_rc_eq(char *arg)

  1635. {

  1636.     video_rc_eq = arg;

  1637. }

  1638. 

  1639. static void opt_video_rc_override_string(char *arg)

  1640. {

  1641.     video_rc_override_string = arg;

  1642. }

  1643. 

  1644. 

  1645. static void opt_workaround_bugs(const char *arg)

  1646. {

  1647.     workaround_bugs = atoi(arg);

  1648. }

  1649. 

  1650. static void opt_dct_algo(const char *arg)

  1651. {

  1652.     dct_algo = atoi(arg);

  1653. }

  1654. 

  1655. static void opt_idct_algo(const char *arg)

  1656. {

  1657.     idct_algo = atoi(arg);

  1658. }

  1659. 

  1660. 

  1661. static void opt_error_resilience(const char *arg)

  1662. {

  1663.     error_resilience = atoi(arg);

  1664. }

  1665. 

  1666. static void opt_error_concealment(const char *arg)

  1667. {

  1668.     error_concealment = atoi(arg);

  1669. }

  1670. 

  1671. static void opt_debug(const char *arg)

  1672. {

  1673.     debug = atoi(arg);

  1674. }

  1675. 

  1676. static void opt_vismv(const char *arg)

  1677. {

  1678.     debug_mv = atoi(arg);

  1679. }

  1680.     

  1681. static void opt_verbose(const char *arg)

  1682. {

  1683.     verbose = atoi(arg);

  1684. }

  1685. 

  1686. static void opt_frame_rate(const char *arg)

  1687. {

  1688.     if (parse_frame_rate(&frame_rate, &frame_rate_base, arg) < 0) {

  1689.         fprintf(stderr, "Incorrect frame rate\n");

  1690. 	exit(1);

  1691.     }

  1692. }

  1693. 

  1694. static void opt_frame_crop_top(const char *arg)

  1695. {

  1696.     frame_topBand = atoi(arg); 

  1697.     if (frame_topBand < 0) {

  1698.         fprintf(stderr, "Incorrect top crop size\n");

  1699.         exit(1);

  1700.     }

  1701.     if ((frame_topBand % 2) != 0) {

  1702.         fprintf(stderr, "Top crop size must be a multiple of 2\n");

  1703.         exit(1);

  1704.     }

  1705.     if ((frame_topBand) >= frame_height){

  1706.     	fprintf(stderr, "Vertical crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");

  1707.         exit(1);

  1708.     }

  1709.     frame_height -= frame_topBand;

  1710. }

  1711. 

  1712. static void opt_frame_crop_bottom(const char *arg)

  1713. {

  1714.     frame_bottomBand = atoi(arg);

  1715.     if (frame_bottomBand < 0) {

  1716.         fprintf(stderr, "Incorrect bottom crop size\n");

  1717.         exit(1);

  1718.     }

  1719.     if ((frame_bottomBand % 2) != 0) {

  1720.         fprintf(stderr, "Bottom crop size must be a multiple of 2\n");

  1721.         exit(1);        

  1722.     }

  1723.     if ((frame_bottomBand) >= frame_height){

  1724.     	fprintf(stderr, "Vertical crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");

  1725.         exit(1);

  1726.     }

  1727.     frame_height -= frame_bottomBand;

  1728. }

  1729. 

  1730. static void opt_frame_crop_left(const char *arg)

  1731. {

  1732.     frame_leftBand = atoi(arg);

  1733.     if (frame_leftBand < 0) {

  1734.         fprintf(stderr, "Incorrect left crop size\n");

  1735.         exit(1);

  1736.     }

  1737.     if ((frame_leftBand % 2) != 0) {

  1738.         fprintf(stderr, "Left crop size must be a multiple of 2\n");

  1739.         exit(1);

  1740.     }

  1741.     if ((frame_leftBand) >= frame_width){

  1742.     	fprintf(stderr, "Horizontal crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");

  1743.         exit(1);

  1744.     }

  1745.     frame_width -= frame_leftBand;

  1746. }

  1747. 

  1748. static void opt_frame_crop_right(const char *arg)

  1749. {

  1750.     frame_rightBand = atoi(arg);

  1751.     if (frame_rightBand < 0) {

  1752.         fprintf(stderr, "Incorrect right crop size\n");

  1753.         exit(1);

  1754.     }

  1755.     if ((frame_rightBand % 2) != 0) {

  1756.         fprintf(stderr, "Right crop size must be a multiple of 2\n");

  1757.         exit(1);        

  1758.     }

  1759.     if ((frame_rightBand) >= frame_width){

  1760.     	fprintf(stderr, "Horizontal crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");

  1761.         exit(1);

  1762.     }

  1763.     frame_width -= frame_rightBand;

  1764. }

  1765. 

  1766. static void opt_frame_size(const char *arg)

  1767. {

  1768.     if (parse_image_size(&frame_width, &frame_height, arg) < 0) {

  1769.         fprintf(stderr, "Incorrect frame size\n");

  1770.         exit(1);

  1771.     }

  1772.     if ((frame_width % 2) != 0 || (frame_height % 2) != 0) {

  1773.         fprintf(stderr, "Frame size must be a multiple of 2\n");

  1774.         exit(1);

  1775.     }

  1776. }

  1777. 

  1778. static void opt_frame_pix_fmt(const char *arg)

  1779. {

  1780.     frame_pix_fmt = avcodec_get_pix_fmt(arg);

  1781. }

  1782. 

  1783. static void opt_frame_aspect_ratio(const char *arg)

  1784. {

  1785.     int x = 0, y = 0;

  1786.     double ar = 0;

  1787.     const char *p;

  1788.     

  1789.     p = strchr(arg, ':');

  1790.     if (p) {

  1791.         x = strtol(arg, (char **)&arg, 10);

  1792. 	if (arg == p)

  1793. 	    y = strtol(arg+1, (char **)&arg, 10);

  1794. 	if (x > 0 && y > 0)

  1795. 	    ar = (double)x / (double)y;

  1796.     } else

  1797.         ar = strtod(arg, (char **)&arg);

  1798. 

  1799.     if (!ar) {

  1800.         fprintf(stderr, "Incorrect aspect ratio specification.\n");

  1801. 	exit(1);

  1802.     }

  1803.     frame_aspect_ratio = ar;

  1804. }

  1805. 

  1806. static void opt_gop_size(const char *arg)

  1807. {

  1808.     gop_size = atoi(arg);

  1809. }

  1810. 

  1811. static void opt_b_frames(const char *arg)

  1812. {

  1813.     b_frames = atoi(arg);

  1814.     if (b_frames > FF_MAX_B_FRAMES) {

  1815.         fprintf(stderr, "\nCannot have more than %d B frames, increase FF_MAX_B_FRAMES.\n", FF_MAX_B_FRAMES);

  1816.         exit(1);

  1817.     } else if (b_frames < 1) {

  1818.         fprintf(stderr, "\nNumber of B frames must be higher than 0\n");

  1819.         exit(1);

  1820.     }

  1821. }

  1822. 

  1823. static void opt_mb_decision(const char *arg)

  1824. {

  1825.     mb_decision = atoi(arg);

  1826. }

  1827. 

  1828. static void opt_mb_cmp(const char *arg)

  1829. {

  1830.     mb_cmp = atoi(arg);

  1831. }

  1832. 

  1833. static void opt_sub_cmp(const char *arg)

  1834. {

  1835.     sub_cmp = atoi(arg);

  1836. }

  1837. 

  1838. static void opt_cmp(const char *arg)

  1839. {

  1840.     cmp = atoi(arg);

  1841. }

  1842. 

  1843. static void opt_qscale(const char *arg)

  1844. {

  1845.     video_qscale = atof(arg);

  1846.     if (video_qscale < 0.01 ||

  1847.         video_qscale > 255) {

  1848.         fprintf(stderr, "qscale must be >= 0.01 and <= 255\n");

  1849.         exit(1);

  1850.     }

  1851. }

  1852. 

  1853. static void opt_qmin(const char *arg)

  1854. {

  1855.     video_qmin = atoi(arg);

  1856.     if (video_qmin < 0 ||

  1857.         video_qmin > 31) {

  1858.         fprintf(stderr, "qmin must be >= 1 and <= 31\n");

  1859.         exit(1);

  1860.     }

  1861. }

  1862. 

  1863. static void opt_qmax(const char *arg)

  1864. {

  1865.     video_qmax = atoi(arg);

  1866.     if (video_qmax < 0 ||

  1867.         video_qmax > 31) {

  1868.         fprintf(stderr, "qmax must be >= 1 and <= 31\n");

  1869.         exit(1);

  1870.     }

  1871. }

  1872. 

  1873. static void opt_mb_qmin(const char *arg)

  1874. {

  1875.     video_mb_qmin = atoi(arg);

  1876.     if (video_mb_qmin < 0 ||

  1877.         video_mb_qmin > 31) {

  1878.         fprintf(stderr, "qmin must be >= 1 and <= 31\n");

  1879.         exit(1);

  1880.     }

  1881. }

  1882. 

  1883. static void opt_mb_qmax(const char *arg)

  1884. {

  1885.     video_mb_qmax = atoi(arg);

  1886.     if (video_mb_qmax < 0 ||

  1887.         video_mb_qmax > 31) {

  1888.         fprintf(stderr, "qmax must be >= 1 and <= 31\n");

  1889.         exit(1);

  1890.     }

  1891. }

  1892. 

  1893. static void opt_qdiff(const char *arg)

  1894. {

  1895.     video_qdiff = atoi(arg);

  1896.     if (video_qdiff < 0 ||

  1897.         video_qdiff > 31) {

  1898.         fprintf(stderr, "qdiff must be >= 1 and <= 31\n");

  1899.         exit(1);

  1900.     }

  1901. }

  1902. 

  1903. static void opt_qblur(const char *arg)

  1904. {

  1905.     video_qblur = atof(arg);

  1906. }

  1907. 

  1908. static void opt_qcomp(const char *arg)

  1909. {

  1910.     video_qcomp = atof(arg);

  1911. }

  1912. 

  1913. static void opt_rc_initial_cplx(const char *arg)

  1914. {

  1915.     video_rc_initial_cplx = atof(arg);

  1916. }

  1917. static void opt_b_qfactor(const char *arg)

  1918. {

  1919.     video_b_qfactor = atof(arg);

  1920. }

  1921. static void opt_i_qfactor(const char *arg)

  1922. {

  1923.     video_i_qfactor = atof(arg);

  1924. }

  1925. static void opt_b_qoffset(const char *arg)

  1926. {

  1927.     video_b_qoffset = atof(arg);

  1928. }

  1929. static void opt_i_qoffset(const char *arg)

  1930. {

  1931.     video_i_qoffset = atof(arg);

  1932. }

  1933. 

  1934. static void opt_packet_size(const char *arg)

  1935. {

  1936.     packet_size= atoi(arg);

  1937. }

  1938. 

  1939. static void opt_error_rate(const char *arg)

  1940. {

  1941.     error_rate= atoi(arg);

  1942. }

  1943. 

  1944. static void opt_strict(const char *arg)

  1945. {

  1946.     strict= atoi(arg);

  1947. }

  1948. 

  1949. static void opt_audio_bitrate(const char *arg)

  1950. {

  1951.     audio_bit_rate = atoi(arg) * 1000;

  1952. }

  1953. 

  1954. static void opt_audio_rate(const char *arg)

  1955. {

  1956.     audio_sample_rate = atoi(arg);

  1957. }

  1958. 

  1959. static void opt_audio_channels(const char *arg)

  1960. {

  1961.     audio_channels = atoi(arg);

  1962. }

  1963. 

  1964. static void opt_video_device(const char *arg)

  1965. {

  1966.     video_device = av_strdup(arg);

  1967. }

  1968. 

  1969. static void opt_video_channel(const char *arg)

  1970. {

  1971.     video_channel = strtol(arg, NULL, 0);

  1972. }

  1973. 

  1974. static void opt_video_standard(const char *arg)

  1975. {

  1976.     video_standard = av_strdup(arg);

  1977. }

  1978. 

  1979. static void opt_audio_device(const char *arg)

  1980. {

  1981.     audio_device = av_strdup(arg);

  1982. }

  1983. 

  1984. static void opt_dv1394(const char *arg)

  1985. {

  1986.     video_grab_format = "dv1394";

  1987.     audio_grab_format = NULL;

  1988. }

  1989. 

  1990. static void opt_audio_codec(const char *arg)

  1991. {

  1992.     AVCodec *p;

  1993. 

  1994.     if (!strcmp(arg, "copy")) {

  1995.         audio_stream_copy = 1;

  1996.     } else {

  1997.         p = first_avcodec;

  1998.         while (p) {

  1999.             if (!strcmp(p->name, arg) && p->type == CODEC_TYPE_AUDIO)

  2000.                 break;

  2001.             p = p->next;

  2002.         }

  2003.         if (p == NULL) {

  2004.             fprintf(stderr, "Unknown audio codec '%s'\n", arg);

  2005.             exit(1);

  2006.         } else {

  2007.             audio_codec_id = p->id;

  2008.         }

  2009.     }

  2010. }

  2011. 

  2012. static void add_frame_hooker(const char *arg)

  2013. {

  2014.     int argc = 0;

  2015.     char *argv[64];

  2016.     int i;

  2017.     char *args = av_strdup(arg);

  2018. 

  2019.     using_vhook = 1;

  2020. 

  2021.     argv[0] = strtok(args, " ");

  2022.     while (argc < 62 && (argv[++argc] = strtok(NULL, " "))) {

  2023.     }

  2024. 

  2025.     i = frame_hook_add(argc, argv);

  2026. 

  2027.     if (i != 0) {

  2028.         fprintf(stderr, "Failed to add video hook function: %s\n", arg);

  2029.         exit(1);

  2030.     }

  2031. }

  2032. 

  2033. const char *motion_str[] = {

  2034.     "zero",

  2035.     "full",

  2036.     "log",

  2037.     "phods",

  2038.     "epzs",

  2039.     "x1",

  2040.     NULL,

  2041. };

  2042. 

  2043. static void opt_motion_estimation(const char *arg)

  2044. {

  2045.     const char **p;

  2046.     p = motion_str;

  2047.     for(;;) {

  2048.         if (!*p) {

  2049.             fprintf(stderr, "Unknown motion estimation method '%s'\n", arg);

  2050.             exit(1);

  2051.         }

  2052.         if (!strcmp(*p, arg))

  2053.             break;

  2054.         p++;

  2055.     }

  2056.     me_method = (p - motion_str) + 1;

  2057. }

  2058. 

  2059. static void opt_video_codec(const char *arg)

  2060. {

  2061.     AVCodec *p;

  2062. 

  2063.     if (!strcmp(arg, "copy")) {

  2064.         video_stream_copy = 1;

  2065.     } else {

  2066.         p = first_avcodec;

  2067.         while (p) {

  2068.             if (!strcmp(p->name, arg) && p->type == CODEC_TYPE_VIDEO)

  2069.                 break;

  2070.             p = p->next;

  2071.         }

  2072.         if (p == NULL) {

  2073.             fprintf(stderr, "Unknown video codec '%s'\n", arg);

  2074.             exit(1);

  2075.         } else {

  2076.             video_codec_id = p->id;

  2077.         }

  2078.     }

  2079. }

  2080. 

  2081. static void opt_map(const char *arg)

  2082. {

  2083.     AVStreamMap *m;

  2084.     const char *p;

  2085. 

  2086.     p = arg;

  2087.     m = &stream_maps[nb_stream_maps++];

  2088. 

  2089.     m->file_index = strtol(arg, (char **)&p, 0);

  2090.     if (*p)

  2091.         p++;

  2092. 

  2093.     m->stream_index = strtol(p, (char **)&p, 0);

  2094. }

  2095. 

  2096. static void opt_recording_time(const char *arg)

  2097. {

  2098.     recording_time = parse_date(arg, 1);

  2099. }

  2100. 

  2101. static void opt_start_time(const char *arg)

  2102. {

  2103.     start_time = parse_date(arg, 1);

  2104. }

  2105. 

  2106. static void opt_input_file(const char *filename)

  2107. {

  2108.     AVFormatContext *ic;

  2109.     AVFormatParameters params, *ap = &params;

  2110.     int err, i, ret, rfps, rfps_base;

  2111. 

  2112.     if (!strcmp(filename, "-"))

  2113.         filename = "pipe:";

  2114. 

  2115.     using_stdin |= !strcmp(filename, "pipe:" ) || 

  2116.                    !strcmp( filename, "/dev/stdin" );

  2117. 

  2118.     /* get default parameters from command line */

  2119.     memset(ap, 0, sizeof(*ap));

  2120.     ap->sample_rate = audio_sample_rate;

  2121.     ap->channels = audio_channels;

  2122.     ap->frame_rate = frame_rate;

  2123.     ap->frame_rate_base = frame_rate_base;

  2124.     ap->width = frame_width;

  2125.     ap->height = frame_height;

  2126.     ap->image_format = image_format;

  2127.     ap->pix_fmt = frame_pix_fmt;

  2128. 

  2129.     /* open the input file with generic libav function */

  2130.     err = av_open_input_file(&ic, filename, file_iformat, 0, ap);

  2131.     if (err < 0) {

  2132.         print_error(filename, err);

  2133.         exit(1);

  2134.     }

  2135.     

  2136.     /* If not enough info to get the stream parameters, we decode the

  2137.        first frames to get it. (used in mpeg case for example) */

  2138.     ret = av_find_stream_info(ic);

  2139.     if (ret < 0) {

  2140.         fprintf(stderr, "%s: could not find codec parameters\n", filename);

  2141.         exit(1);

  2142.     }

  2143. 

  2144.     /* if seeking requested, we execute it */

  2145.     if (start_time != 0) {

  2146.         int64_t timestamp;

  2147. 

  2148.         timestamp = start_time;

  2149.         /* add the stream start time */

  2150.         if (ic->start_time != AV_NOPTS_VALUE)

  2151.             timestamp += ic->start_time;

  2152.         ret = av_seek_frame(ic, -1, timestamp);

  2153.         if (ret < 0) {

  2154.             fprintf(stderr, "%s: could not seek to position %0.3f\n", 

  2155.                     filename, (double)timestamp / AV_TIME_BASE);

  2156.         }

  2157.         /* reset seek info */

  2158.         start_time = 0;

  2159.     }

  2160. 

  2161.     /* update the current parameters so that they match the one of the input stream */

  2162.     for(i=0;i<ic->nb_streams;i++) {

  2163.         AVCodecContext *enc = &ic->streams[i]->codec;

  2164.         switch(enc->codec_type) {

  2165.         case CODEC_TYPE_AUDIO:

  2166.             //fprintf(stderr, "\nInput Audio channels: %d", enc->channels);

  2167.             audio_channels = enc->channels;

  2168.             audio_sample_rate = enc->sample_rate;

  2169.             break;

  2170.         case CODEC_TYPE_VIDEO:

  2171.             frame_height = enc->height;

  2172.             frame_width = enc->width;

  2173. 	    frame_aspect_ratio = av_q2d(enc->sample_aspect_ratio) * enc->width / enc->height;

  2174. 	    frame_pix_fmt = enc->pix_fmt;

  2175.             rfps      = ic->streams[i]->r_frame_rate;

  2176.             rfps_base = ic->streams[i]->r_frame_rate_base;

  2177.             enc->workaround_bugs = workaround_bugs;

  2178.             enc->error_resilience = error_resilience; 

  2179.             enc->error_concealment = error_concealment; 

  2180.             enc->idct_algo = idct_algo;

  2181.             enc->debug = debug;

  2182.             enc->debug_mv = debug_mv;            

  2183.             if(bitexact)

  2184.                 enc->flags|= CODEC_FLAG_BITEXACT;

  2185. 

  2186.             assert(enc->frame_rate_base == rfps_base); // should be true for now

  2187.             if (enc->frame_rate != rfps) { 

  2188.                 fprintf(stderr,"\nSeems that stream %d comes from film source: %2.2f->%2.2f\n",

  2189.                     i, (float)enc->frame_rate / enc->frame_rate_base,

  2190.                     (float)rfps / rfps_base);

  2191.             }

  2192.             /* update the current frame rate to match the stream frame rate */

  2193.             frame_rate      = rfps;

  2194.             frame_rate_base = rfps_base;

  2195. 

  2196.             enc->rate_emu = rate_emu;

  2197.             break;

  2198.         case CODEC_TYPE_DATA:

  2199.             break;

  2200.         default:

  2201.             av_abort();

  2202.         }

  2203.     }

  2204.     

  2205.     input_files[nb_input_files] = ic;

  2206.     /* dump the file content */

  2207.     dump_format(ic, nb_input_files, filename, 0);

  2208.     nb_input_files++;

  2209.     file_iformat = NULL;

  2210.     file_oformat = NULL;

  2211.     image_format = NULL;

  2212. 

  2213.     rate_emu = 0;

  2214. }

  2215. 

  2216. static void check_audio_video_inputs(int *has_video_ptr, int *has_audio_ptr)

  2217. {

  2218.     int has_video, has_audio, i, j;

  2219.     AVFormatContext *ic;

  2220. 

  2221.     has_video = 0;

  2222.     has_audio = 0;

  2223.     for(j=0;j<nb_input_files;j++) {

  2224.         ic = input_files[j];

  2225.         for(i=0;i<ic->nb_streams;i++) {

  2226.             AVCodecContext *enc = &ic->streams[i]->codec;

  2227.             switch(enc->codec_type) {

  2228.             case CODEC_TYPE_AUDIO:

  2229.                 has_audio = 1;

  2230.                 break;

  2231.             case CODEC_TYPE_VIDEO:

  2232.                 has_video = 1;

  2233.                 break;

  2234.             default:

  2235.                 av_abort();

  2236.             }

  2237.         }

  2238.     }

  2239.     *has_video_ptr = has_video;

  2240.     *has_audio_ptr = has_audio;

  2241. }

  2242. 

  2243. static void opt_output_file(const char *filename)

  2244. {

  2245.     AVStream *st;

  2246.     AVFormatContext *oc;

  2247.     int use_video, use_audio, nb_streams, input_has_video, input_has_audio;

  2248.     int codec_id;

  2249.     AVFormatParameters params, *ap = &params;

  2250. 

  2251.     if (!strcmp(filename, "-"))

  2252.         filename = "pipe:";

  2253. 

  2254.     oc = av_mallocz(sizeof(AVFormatContext));

  2255. 

  2256.     if (!file_oformat) {

  2257.         file_oformat = guess_format(NULL, filename, NULL);

  2258.         if (!file_oformat) {

  2259.             fprintf(stderr, "Unable for find a suitable output format for '%s'\n",

  2260.                     filename);

  2261.             exit(1);

  2262.         }

  2263.     }

  2264.     

  2265.     oc->oformat = file_oformat;

  2266. 

  2267.     if (!strcmp(file_oformat->name, "ffm") && 

  2268.         strstart(filename, "http:", NULL)) {

  2269.         /* special case for files sent to ffserver: we get the stream

  2270.            parameters from ffserver */

  2271.         if (read_ffserver_streams(oc, filename) < 0) {

  2272.             fprintf(stderr, "Could not read stream parameters from '%s'\n", filename);

  2273.             exit(1);

  2274.         }

  2275.     } else {

  2276.         use_video = file_oformat->video_codec != CODEC_ID_NONE;

  2277.         use_audio = file_oformat->audio_codec != CODEC_ID_NONE;

  2278. 

  2279.         /* disable if no corresponding type found and at least one

  2280.            input file */

  2281.         if (nb_input_files > 0) {

  2282.             check_audio_video_inputs(&input_has_video, &input_has_audio);

  2283.             if (!input_has_video)

  2284.                 use_video = 0;

  2285.             if (!input_has_audio)

  2286.                 use_audio = 0;

  2287.         }

  2288. 

  2289.         /* manual disable */

  2290.         if (audio_disable) {

  2291.             use_audio = 0;

  2292.         }

  2293.         if (video_disable) {

  2294.             use_video = 0;

  2295.         }

  2296.         

  2297.         nb_streams = 0;

  2298.         if (use_video) {

  2299.             AVCodecContext *video_enc;

  2300.             

  2301.             st = av_mallocz(sizeof(AVStream));

  2302.             if (!st) {

  2303.                 fprintf(stderr, "Could not alloc stream\n");

  2304.                 exit(1);

  2305.             }

  2306.             avcodec_get_context_defaults(&st->codec);

  2307. 

  2308.             video_enc = &st->codec;

  2309.             

  2310.             if(!strcmp(file_oformat->name, "mp4") || !strcmp(file_oformat->name, "mov") || !strcmp(file_oformat->name, "3gp"))

  2311.                 video_enc->flags |= CODEC_FLAG_GLOBAL_HEADER;

  2312.             if (video_stream_copy) {

  2313.                 st->stream_copy = 1;

  2314.                 video_enc->codec_type = CODEC_TYPE_VIDEO;

  2315.             } else {

  2316.                 char *p;

  2317.                 int i;

  2318.             

  2319.                 codec_id = file_oformat->video_codec;

  2320.                 if (video_codec_id != CODEC_ID_NONE)

  2321.                     codec_id = video_codec_id;

  2322.                 

  2323.                 video_enc->codec_id = codec_id;

  2324.                 

  2325.                 video_enc->bit_rate = video_bit_rate;

  2326.                 video_enc->bit_rate_tolerance = video_bit_rate_tolerance;

  2327.                 video_enc->frame_rate = frame_rate; 

  2328.                 video_enc->frame_rate_base = frame_rate_base; 

  2329.                 

  2330.                 video_enc->width = frame_width;

  2331.                 video_enc->height = frame_height;

  2332. 		video_enc->sample_aspect_ratio = av_d2q(frame_aspect_ratio*frame_height/frame_width, 255);

  2333. 		video_enc->pix_fmt = frame_pix_fmt;

  2334. 

  2335.                 if (!intra_only)

  2336.                     video_enc->gop_size = gop_size;

  2337.                 else

  2338.                     video_enc->gop_size = 0;

  2339.                 if (video_qscale || same_quality) {

  2340.                     video_enc->flags |= CODEC_FLAG_QSCALE;

  2341.                     st->quality = FF_QP2LAMBDA * video_qscale;

  2342.                 }

  2343. 

  2344.                 if(intra_matrix)

  2345.                     video_enc->intra_matrix = intra_matrix;

  2346.                 if(inter_matrix)

  2347.                     video_enc->inter_matrix = inter_matrix;

  2348. 

  2349.                 if(bitexact)

  2350.                     video_enc->flags |= CODEC_FLAG_BITEXACT;

  2351. 

  2352.                 video_enc->mb_decision = mb_decision;

  2353.                 video_enc->mb_cmp = mb_cmp;

  2354.                 video_enc->me_sub_cmp = sub_cmp;

  2355.                 video_enc->me_cmp = cmp;

  2356.                 

  2357.                 if (use_umv) {

  2358.                     video_enc->flags |= CODEC_FLAG_H263P_UMV;

  2359.                 }

  2360.            	if (use_aic) {

  2361.                     video_enc->flags |= CODEC_FLAG_H263P_AIC;

  2362.                 }

  2363.            	if (use_aiv) {

  2364.                     video_enc->flags |= CODEC_FLAG_H263P_AIV;

  2365.                 }

  2366.                 if (use_4mv) {

  2367.                     video_enc->flags |= CODEC_FLAG_4MV;

  2368.                 }

  2369.                 if (use_obmc) {

  2370.                     video_enc->flags |= CODEC_FLAG_OBMC;

  2371.                 }

  2372.             

  2373.                 if(use_part) {

  2374.                     video_enc->flags |= CODEC_FLAG_PART;

  2375.                 }

  2376.                 if (b_frames) {

  2377.                     video_enc->max_b_frames = b_frames;

  2378.                     video_enc->b_frame_strategy = 0;

  2379.                     video_enc->b_quant_factor = 2.0;

  2380.                 }

  2381.                 if (do_interlace) {

  2382.                     video_enc->flags |= CODEC_FLAG_INTERLACED_DCT;

  2383.                 }

  2384.                 video_enc->qmin = video_qmin;

  2385.                 video_enc->qmax = video_qmax;

  2386.                 video_enc->mb_qmin = video_mb_qmin;

  2387.                 video_enc->mb_qmax = video_mb_qmax;

  2388.                 video_enc->max_qdiff = video_qdiff;

  2389.                 video_enc->qblur = video_qblur;

  2390.                 video_enc->qcompress = video_qcomp;

  2391.                 video_enc->rc_eq = video_rc_eq;

  2392.                 video_enc->debug = debug;

  2393.                 video_enc->debug_mv = debug_mv;                

  2394.                 p= video_rc_override_string;

  2395.                 for(i=0; p; i++){

  2396.                     int start, end, q;

  2397.                     int e=sscanf(p, "%d,%d,%d", &start, &end, &q);

  2398.                     if(e!=3){

  2399.                         fprintf(stderr, "error parsing rc_override\n");

  2400.                         exit(1);

  2401.                     }

  2402.                     video_enc->rc_override= 

  2403.                         av_realloc(video_enc->rc_override, 

  2404.                                    sizeof(RcOverride)*(i+1));

  2405.                     video_enc->rc_override[i].start_frame= start;

  2406.                     video_enc->rc_override[i].end_frame  = end;

  2407.                     if(q>0){

  2408.                         video_enc->rc_override[i].qscale= q;

  2409.                         video_enc->rc_override[i].quality_factor= 1.0;

  2410.                     }

  2411.                     else{

  2412.                         video_enc->rc_override[i].qscale= 0;

  2413.                         video_enc->rc_override[i].quality_factor= -q/100.0;

  2414.                     }

  2415.                     p= strchr(p, '/');

  2416.                     if(p) p++;

  2417.                 }

  2418.                 video_enc->rc_override_count=i;

  2419. 

  2420.                 video_enc->rc_max_rate = video_rc_max_rate;

  2421.                 video_enc->rc_min_rate = video_rc_min_rate;

  2422.                 video_enc->rc_buffer_size = video_rc_buffer_size;

  2423.                 video_enc->rc_buffer_aggressivity= video_rc_buffer_aggressivity;

  2424.                 video_enc->rc_initial_cplx= video_rc_initial_cplx;

  2425.                 video_enc->i_quant_factor = video_i_qfactor;

  2426.                 video_enc->b_quant_factor = video_b_qfactor;

  2427.                 video_enc->i_quant_offset = video_i_qoffset;

  2428.                 video_enc->b_quant_offset = video_b_qoffset;

  2429.                 video_enc->dct_algo = dct_algo;

  2430.                 video_enc->idct_algo = idct_algo;

  2431.                 video_enc->strict_std_compliance = strict;

  2432.                 video_enc->error_rate = error_rate;

  2433.                 if(packet_size){

  2434.                     video_enc->rtp_mode= 1;

  2435.                     video_enc->rtp_payload_size= packet_size;

  2436.                 }

  2437.             

  2438.                 if (do_psnr)

  2439.                     video_enc->flags|= CODEC_FLAG_PSNR;

  2440.             

  2441.                 video_enc->me_method = me_method;

  2442. 

  2443.                 /* two pass mode */

  2444.                 if (do_pass) {

  2445.                     if (do_pass == 1) {

  2446.                         video_enc->flags |= CODEC_FLAG_PASS1;

  2447.                     } else {

  2448.                         video_enc->flags |= CODEC_FLAG_PASS2;

  2449.                     }

  2450.                 }

  2451.             }

  2452.             oc->streams[nb_streams] = st;

  2453.             nb_streams++;

  2454.         }

  2455.     

  2456.         if (use_audio) {

  2457.             AVCodecContext *audio_enc;

  2458. 

  2459.             st = av_mallocz(sizeof(AVStream));

  2460.             if (!st) {

  2461.                 fprintf(stderr, "Could not alloc stream\n");

  2462.                 exit(1);

  2463.             }

  2464.             avcodec_get_context_defaults(&st->codec);

  2465. 

  2466.             audio_enc = &st->codec;

  2467.             audio_enc->codec_type = CODEC_TYPE_AUDIO;

  2468. 

  2469.             if(!strcmp(file_oformat->name, "mp4") || !strcmp(file_oformat->name, "mov") || !strcmp(file_oformat->name, "3gp"))

  2470.                 audio_enc->flags |= CODEC_FLAG_GLOBAL_HEADER;

  2471.             if (audio_stream_copy) {

  2472.                 st->stream_copy = 1;

  2473.             } else {

  2474.                 codec_id = file_oformat->audio_codec;

  2475.                 if (audio_codec_id != CODEC_ID_NONE)

  2476.                     codec_id = audio_codec_id;

  2477.                 audio_enc->codec_id = codec_id;

  2478.                 

  2479.                 audio_enc->bit_rate = audio_bit_rate;

  2480.                 audio_enc->sample_rate = audio_sample_rate;

  2481.                 /* For audio codecs other than AC3 we limit */

  2482.                 /* the number of coded channels to stereo   */

  2483.                 if (audio_channels > 2 && codec_id != CODEC_ID_AC3) {

  2484.                     audio_enc->channels = 2;

  2485.                 } else

  2486.                     audio_enc->channels = audio_channels;

  2487.             }

  2488.             oc->streams[nb_streams] = st;

  2489.             nb_streams++;

  2490.         }

  2491. 

  2492.         oc->nb_streams = nb_streams;

  2493. 

  2494.         if (!nb_streams) {

  2495.             fprintf(stderr, "No audio or video streams available\n");

  2496.             exit(1);

  2497.         }

  2498. 

  2499.         if (str_title)

  2500.             pstrcpy(oc->title, sizeof(oc->title), str_title);

  2501.         if (str_author)

  2502.             pstrcpy(oc->author, sizeof(oc->author), str_author);

  2503.         if (str_copyright)

  2504.             pstrcpy(oc->copyright, sizeof(oc->copyright), str_copyright);

  2505.         if (str_comment)

  2506.             pstrcpy(oc->comment, sizeof(oc->comment), str_comment);

  2507.     }

  2508. 

  2509.     output_files[nb_output_files++] = oc;

  2510. 

  2511.     strcpy(oc->filename, filename);

  2512. 

  2513.     /* check filename in case of an image number is expected */

  2514.     if (oc->oformat->flags & AVFMT_NEEDNUMBER) {

  2515.         if (filename_number_test(oc->filename) < 0) {

  2516.             print_error(oc->filename, AVERROR_NUMEXPECTED);

  2517.             exit(1);

  2518.         }

  2519.     }

  2520. 

  2521.     if (!(oc->oformat->flags & AVFMT_NOFILE)) {

  2522.         /* test if it already exists to avoid loosing precious files */

  2523.         if (!file_overwrite && 

  2524.             (strchr(filename, ':') == NULL ||

  2525.              strstart(filename, "file:", NULL))) {

  2526.             if (url_exist(filename)) {

  2527.                 int c;

  2528.                 

  2529.                 if ( !using_stdin ) {

  2530.                     fprintf(stderr,"File '%s' already exists. Overwrite ? [y/N] ", filename);

  2531.                     fflush(stderr);

  2532.                     c = getchar();

  2533.                     if (toupper(c) != 'Y') {

  2534.                         fprintf(stderr, "Not overwriting - exiting\n");

  2535.                         exit(1);

  2536.                     }

  2537. 				}

  2538. 				else {

  2539.                     fprintf(stderr,"File '%s' already exists. Exiting.\n", filename);

  2540.                     exit(1);

  2541. 				}

  2542.             }

  2543.         }

  2544.         

  2545.         /* open the file */

  2546.         if (url_fopen(&oc->pb, filename, URL_WRONLY) < 0) {

  2547.             fprintf(stderr, "Could not open '%s'\n", filename);

  2548.             exit(1);

  2549.         }

  2550.     }

  2551. 

  2552.     memset(ap, 0, sizeof(*ap));

  2553.     ap->image_format = image_format;

  2554.     if (av_set_parameters(oc, ap) < 0) {

  2555.         fprintf(stderr, "%s: Invalid encoding parameters\n",

  2556.                 oc->filename);

  2557.         exit(1);

  2558.     }

  2559. 

  2560.     /* reset some options */

  2561.     file_oformat = NULL;

  2562.     file_iformat = NULL;

  2563.     image_format = NULL;

  2564.     audio_disable = 0;

  2565.     video_disable = 0;

  2566.     audio_codec_id = CODEC_ID_NONE;

  2567.     video_codec_id = CODEC_ID_NONE;

  2568.     audio_stream_copy = 0;

  2569.     video_stream_copy = 0;

  2570. }

  2571. 

  2572. /* prepare dummy protocols for grab */

  2573. static void prepare_grab(void)

  2574. {

  2575.     int has_video, has_audio, i, j;

  2576.     AVFormatContext *oc;

  2577.     AVFormatContext *ic;

  2578.     AVFormatParameters vp1, *vp = &vp1;

  2579.     AVFormatParameters ap1, *ap = &ap1;

  2580.     

  2581.     /* see if audio/video inputs are needed */

  2582.     has_video = 0;

  2583.     has_audio = 0;

  2584.     memset(ap, 0, sizeof(*ap));

  2585.     memset(vp, 0, sizeof(*vp));

  2586.     for(j=0;j<nb_output_files;j++) {

  2587.         oc = output_files[j];

  2588.         for(i=0;i<oc->nb_streams;i++) {

  2589.             AVCodecContext *enc = &oc->streams[i]->codec;

  2590.             switch(enc->codec_type) {

  2591.             case CODEC_TYPE_AUDIO:

  2592.                 if (enc->sample_rate > ap->sample_rate)

  2593.                     ap->sample_rate = enc->sample_rate;

  2594.                 if (enc->channels > ap->channels)

  2595.                     ap->channels = enc->channels;

  2596.                 has_audio = 1;

  2597.                 break;

  2598.             case CODEC_TYPE_VIDEO:

  2599.                 if (enc->width > vp->width)

  2600.                     vp->width = enc->width;

  2601.                 if (enc->height > vp->height)

  2602.                     vp->height = enc->height;

  2603.                 

  2604.                 assert(enc->frame_rate_base == DEFAULT_FRAME_RATE_BASE);

  2605.                 if (enc->frame_rate > vp->frame_rate){

  2606.                     vp->frame_rate      = enc->frame_rate;

  2607.                     vp->frame_rate_base = enc->frame_rate_base;

  2608.                 }

  2609.                 has_video = 1;

  2610.                 break;

  2611.             default:

  2612.                 av_abort();

  2613.             }

  2614.         }

  2615.     }

  2616.     

  2617.     if (has_video == 0 && has_audio == 0) {

  2618.         fprintf(stderr, "Output file must have at least one audio or video stream\n");

  2619.         exit(1);

  2620.     }

  2621.     

  2622.     if (has_video) {

  2623.         AVInputFormat *fmt1;

  2624.         fmt1 = av_find_input_format(video_grab_format);

  2625.         vp->device  = video_device;

  2626.         vp->channel = video_channel;

  2627. 	vp->standard = video_standard;

  2628.         if (av_open_input_file(&ic, "", fmt1, 0, vp) < 0) {

  2629.             fprintf(stderr, "Could not find video grab device\n");

  2630.             exit(1);

  2631.         }

  2632.         /* If not enough info to get the stream parameters, we decode the

  2633.            first frames to get it. */

  2634. 	if ((ic->ctx_flags & AVFMTCTX_NOHEADER) && av_find_stream_info(ic) < 0) {

  2635.             fprintf(stderr, "Could not find video grab parameters\n");

  2636.             exit(1);

  2637.         }

  2638.         /* by now video grab has one stream */

  2639.         ic->streams[0]->r_frame_rate      = vp->frame_rate;

  2640.         ic->streams[0]->r_frame_rate_base = vp->frame_rate_base;

  2641.         input_files[nb_input_files] = ic;

  2642.         dump_format(ic, nb_input_files, "", 0);

  2643.         nb_input_files++;

  2644.     }

  2645.     if (has_audio && audio_grab_format) {

  2646.         AVInputFormat *fmt1;

  2647.         fmt1 = av_find_input_format(audio_grab_format);

  2648.         ap->device = audio_device;

  2649.         if (av_open_input_file(&ic, "", fmt1, 0, ap) < 0) {

  2650.             fprintf(stderr, "Could not find audio grab device\n");

  2651.             exit(1);

  2652.         }

  2653.         input_files[nb_input_files] = ic;

  2654.         dump_format(ic, nb_input_files, "", 0);

  2655.         nb_input_files++;

  2656.     }

  2657. }

  2658. 

  2659. /* same option as mencoder */

  2660. static void opt_pass(const char *pass_str)

  2661. {

  2662.     int pass;

  2663.     pass = atoi(pass_str);

  2664.     if (pass != 1 && pass != 2) {

  2665.         fprintf(stderr, "pass number can be only 1 or 2\n");

  2666.         exit(1);

  2667.     }

  2668.     do_pass = pass;

  2669. }

  2670. 

  2671. #if defined(CONFIG_WIN32) || defined(CONFIG_OS2)

  2672. static int64_t getutime(void)

  2673. {

  2674.   return av_gettime();

  2675. }

  2676. #else

  2677. static int64_t getutime(void)

  2678. {

  2679.     struct rusage rusage;

  2680. 

  2681.     getrusage(RUSAGE_SELF, &rusage);

  2682.     return (rusage.ru_utime.tv_sec * 1000000LL) + rusage.ru_utime.tv_usec;

  2683. }

  2684. #endif

  2685. 

  2686. extern int ffm_nopts;

  2687. 

  2688. static void opt_bitexact(void)

  2689. {

  2690.     bitexact=1;

  2691.     /* disable generate of real time pts in ffm (need to be supressed anyway) */

  2692.     ffm_nopts = 1;

  2693. }

  2694. 

  2695. static void show_formats(void)

  2696. {

  2697.     AVInputFormat *ifmt;

  2698.     AVOutputFormat *ofmt;

  2699.     AVImageFormat *image_fmt;

  2700.     URLProtocol *up;

  2701.     AVCodec *p;

  2702.     const char **pp;

  2703. 

  2704.     printf("Output audio/video file formats:");

  2705.     for(ofmt = first_oformat; ofmt != NULL; ofmt = ofmt->next) {

  2706.         printf(" %s", ofmt->name);

  2707.     }

  2708.     printf("\n");

  2709. 

  2710.     printf("Input audio/video file formats:");

  2711.     for(ifmt = first_iformat; ifmt != NULL; ifmt = ifmt->next) {

  2712.         printf(" %s", ifmt->name);

  2713.     }

  2714.     printf("\n");

  2715. 

  2716.     printf("Output image formats:");

  2717.     for(image_fmt = first_image_format; image_fmt != NULL; 

  2718.         image_fmt = image_fmt->next) {

  2719.         if (image_fmt->img_write)

  2720.             printf(" %s", image_fmt->name);

  2721.     }

  2722.     printf("\n");

  2723. 

  2724.     printf("Input image formats:");

  2725.     for(image_fmt = first_image_format; image_fmt != NULL; 

  2726.         image_fmt = image_fmt->next) {

  2727.         if (image_fmt->img_read)

  2728.             printf(" %s", image_fmt->name);

  2729.     }

  2730.     printf("\n");

  2731. 

  2732.     printf("Codecs:\n");

  2733.     printf("  Encoders:");

  2734.     for(p = first_avcodec; p != NULL; p = p->next) {

  2735.         if (p->encode)

  2736.             printf(" %s", p->name);

  2737.     }

  2738.     printf("\n");

  2739. 

  2740.     printf("  Decoders:");

  2741.     for(p = first_avcodec; p != NULL; p = p->next) {

  2742.         if (p->decode)

  2743.             printf(" %s", p->name);

  2744.     }

  2745.     printf("\n");

  2746. 

  2747.     printf("Supported file protocols:");

  2748.     for(up = first_protocol; up != NULL; up = up->next)

  2749.         printf(" %s:", up->name);

  2750.     printf("\n");

  2751.     

  2752.     printf("Frame size, frame rate abbreviations: ntsc pal qntsc qpal sntsc spal film ntsc-film sqcif qcif cif 4cif\n");

  2753.     printf("Motion estimation methods:");

  2754.     pp = motion_str;

  2755.     while (*pp) {

  2756.         printf(" %s", *pp);

  2757.         if ((pp - motion_str + 1) == ME_ZERO) 

  2758.             printf("(fastest)");

  2759.         else if ((pp - motion_str + 1) == ME_FULL) 

  2760.             printf("(slowest)");

  2761.         else if ((pp - motion_str + 1) == ME_EPZS) 

  2762.             printf("(default)");

  2763.         pp++;

  2764.     }

  2765.     printf("\n");

  2766.     exit(1);

  2767. }

  2768. 

  2769. void parse_matrix_coeffs(uint16_t *dest, const char *str)

  2770. {

  2771.     int i;

  2772.     const char *p = str;

  2773.     for(i = 0;; i++) {

  2774.         dest[i] = atoi(p);

  2775.         if(i == 63)

  2776.             break;

  2777.         p = strchr(p, ',');

  2778.         if(!p) {

  2779.             fprintf(stderr, "Syntax error in matrix \"%s\" at coeff %d\n", str, i);

  2780.             exit(1);

  2781.         }

  2782.         p++;

  2783.     }

  2784. }

  2785. 

  2786. void opt_inter_matrix(const char *arg)

  2787. {

  2788.     inter_matrix = av_mallocz(sizeof(uint16_t) * 64);

  2789.     parse_matrix_coeffs(inter_matrix, arg);

  2790. }

  2791. 

  2792. void opt_intra_matrix(const char *arg)

  2793. {

  2794.     intra_matrix = av_mallocz(sizeof(uint16_t) * 64);

  2795.     parse_matrix_coeffs(intra_matrix, arg);

  2796. }

  2797. 

  2798. static void opt_target(const char *arg)

  2799. {

  2800.     int norm = -1;

  2801. 

  2802.     if(!strncmp(arg, "pal-", 4)) {

  2803.         norm = 0;

  2804.         arg += 4;

  2805.     } else if(!strncmp(arg, "ntsc-", 5)) {

  2806.         norm = 1;

  2807.         arg += 5;

  2808.     } else {

  2809.         int fr;

  2810.         /* Calculate FR via float to avoid int overflow */

  2811.         fr = (int)(frame_rate * 1000.0 / frame_rate_base);

  2812.         if(fr == 25000) {

  2813.             norm = 0;

  2814.         } else if((fr == 29970) || (fr == 23976)) {

  2815.             norm = 1;

  2816.         } else {

  2817.             /* Try to determine PAL/NTSC by peeking in the input files */

  2818.             if(nb_input_files) {

  2819.                 int i, j;

  2820.                 for(j = 0; j < nb_input_files; j++) {

  2821.                     for(i = 0; i < input_files[j]->nb_streams; i++) {

  2822.                         AVCodecContext *c = &input_files[j]->streams[i]->codec;

  2823.                         if(c->codec_type != CODEC_TYPE_VIDEO)

  2824.                             continue;

  2825.                         fr = c->frame_rate * 1000 / c->frame_rate_base;

  2826.                         if(fr == 25000) {

  2827.                             norm = 0;

  2828.                             break;

  2829.                         } else if((fr == 29970) || (fr == 23976)) {

  2830.                             norm = 1;

  2831.                             break;

  2832.                         }

  2833.                     }

  2834.                     if(norm >= 0)

  2835.                         break;

  2836.                 }

  2837.             }

  2838.         }

  2839.         if(verbose && norm >= 0)

  2840.             printf("Assuming %s for target.\n", norm ? "NTSC" : "PAL");

  2841.     }

  2842. 

  2843.     if(norm < 0) {

  2844.         fprintf(stderr, "Could not determine norm (PAL/NTSC) for target.\n");

  2845.         fprintf(stderr, "Please prefix target with \"pal-\" or \"ntsc-\",\n");

  2846.         fprintf(stderr, "or set a framerate with \"-r xxx\".\n");

  2847.         exit(1);

  2848.     }

  2849. 

  2850.     if(!strcmp(arg, "vcd")) {

  2851. 

  2852.         opt_video_codec("mpeg1video");

  2853.         opt_audio_codec("mp2");

  2854.         opt_format("vcd");

  2855. 

  2856.         opt_frame_size(norm ? "352x240" : "352x288");

  2857. 

  2858.         video_bit_rate = 1150000;

  2859.         video_rc_max_rate = 1150000;

  2860.         video_rc_min_rate = 1150000;

  2861.         video_rc_buffer_size = 40*1024*8;

  2862. 

  2863.         audio_bit_rate = 224000;

  2864.         audio_sample_rate = 44100;

  2865. 

  2866.     } else if(!strcmp(arg, "svcd")) {

  2867. 

  2868.         opt_video_codec("mpeg2video");

  2869.         opt_audio_codec("mp2");

  2870.         opt_format("vcd");

  2871. 

  2872.         opt_frame_size(norm ? "480x480" : "480x576");

  2873.         opt_gop_size(norm ? "18" : "15");

  2874. 

  2875.         video_bit_rate = 2040000;

  2876.         video_rc_max_rate = 2516000;

  2877.         video_rc_min_rate = 1145000;

  2878.         video_rc_buffer_size = 224*1024*8;

  2879. 

  2880.         audio_bit_rate = 224000;

  2881.         audio_sample_rate = 44100;

  2882. 

  2883.     } else if(!strcmp(arg, "dvd")) {

  2884. 

  2885.         opt_video_codec("mpeg2video");

  2886.         opt_audio_codec("ac3");

  2887.         opt_format("vob");

  2888. 

  2889.         opt_frame_size(norm ? "720x480" : "720x576");

  2890.         opt_gop_size(norm ? "18" : "15");

  2891. 

  2892.         video_bit_rate = 6000000;

  2893.         video_rc_max_rate = 9000000;

  2894.         video_rc_min_rate = 1500000;

  2895.         video_rc_buffer_size = 224*1024*8;

  2896. 

  2897.         audio_bit_rate = 448000;

  2898.         audio_sample_rate = 48000;

  2899. 

  2900.     } else {

  2901.         fprintf(stderr, "Unknown target: %s\n", arg);

  2902.         exit(1);

  2903.     }

  2904. }

  2905. 

  2906. const OptionDef options[] = {

  2907.     /* main options */

  2908.     { "L", 0, {(void*)show_license}, "show license" },

  2909.     { "h", 0, {(void*)show_help}, "show help" },

  2910.     { "formats", 0, {(void*)show_formats}, "show available formats, codecs, protocols, ..." },

  2911.     { "f", HAS_ARG, {(void*)opt_format}, "force format", "fmt" },

  2912.     { "img", HAS_ARG, {(void*)opt_image_format}, "force image format", "img_fmt" },

  2913.     { "i", HAS_ARG, {(void*)opt_input_file}, "input file name", "filename" },

  2914.     { "y", OPT_BOOL, {(void*)&file_overwrite}, "overwrite output files" },

  2915.     { "map", HAS_ARG | OPT_EXPERT, {(void*)opt_map}, "set input stream mapping", "file:stream" },

  2916.     { "t", HAS_ARG, {(void*)opt_recording_time}, "set the recording time", "duration" },

  2917.     { "ss", HAS_ARG, {(void*)opt_start_time}, "set the start time offset", "time_off" },

  2918.     { "title", HAS_ARG | OPT_STRING, {(void*)&str_title}, "set the title", "string" },

  2919.     { "author", HAS_ARG | OPT_STRING, {(void*)&str_author}, "set the author", "string" },

  2920.     { "copyright", HAS_ARG | OPT_STRING, {(void*)&str_copyright}, "set the copyright", "string" },

  2921.     { "comment", HAS_ARG | OPT_STRING, {(void*)&str_comment}, "set the comment", "string" },

  2922.     { "debug", HAS_ARG | OPT_EXPERT, {(void*)opt_debug}, "print specific debug info", "" },

  2923.     { "vismv", HAS_ARG | OPT_EXPERT, {(void*)opt_vismv}, "visualize motion vectors", "" },

  2924.     { "benchmark", OPT_BOOL | OPT_EXPERT, {(void*)&do_benchmark}, 

  2925.       "add timings for benchmarking" },

  2926.     { "dump", OPT_BOOL | OPT_EXPERT, {(void*)&do_pkt_dump}, 

  2927.       "dump each input packet" },

  2928.     { "hex", OPT_BOOL | OPT_EXPERT, {(void*)&do_hex_dump}, 

  2929.       "when dumping packets, also dump the payload" },

  2930.     { "bitexact", OPT_EXPERT, {(void*)opt_bitexact}, "only use bit exact algorithms (for codec testing)" }, 

  2931.     { "re", OPT_BOOL | OPT_EXPERT, {(void*)&rate_emu}, "read input at native frame rate", "" },

  2932.     { "loop", OPT_BOOL | OPT_EXPERT, {(void*)&loop_input}, "loop (current only works with images)" },

  2933.     { "v", HAS_ARG, {(void*)opt_verbose}, "control amount of logging", "verbose" },

  2934.     { "target", HAS_ARG, {(void*)opt_target}, "specify target file type (\"vcd\", \"svcd\" or \"dvd\")", "type" },

  2935. 

  2936.     /* video options */

  2937.     { "b", HAS_ARG | OPT_VIDEO, {(void*)opt_video_bitrate}, "set video bitrate (in kbit/s)", "bitrate" },

  2938.     { "r", HAS_ARG | OPT_VIDEO, {(void*)opt_frame_rate}, "set frame rate (Hz value, fraction or abbreviation)", "rate" },

  2939.     { "s", HAS_ARG | OPT_VIDEO, {(void*)opt_frame_size}, "set frame size (WxH or abbreviation)", "size" },

  2940.     { "aspect", HAS_ARG | OPT_VIDEO, {(void*)opt_frame_aspect_ratio}, "set aspect ratio (4:3, 16:9 or 1.3333, 1.7777)", "aspect" },

  2941.     { "pix_fmt", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_frame_pix_fmt}, "set pixel format", "format" },

  2942.     { "croptop", HAS_ARG | OPT_VIDEO, {(void*)opt_frame_crop_top}, "set top crop band size (in pixels)", "size" },

  2943.     { "cropbottom", HAS_ARG | OPT_VIDEO, {(void*)opt_frame_crop_bottom}, "set bottom crop band size (in pixels)", "size" },

  2944.     { "cropleft", HAS_ARG | OPT_VIDEO, {(void*)opt_frame_crop_left}, "set left crop band size (in pixels)", "size" },

  2945.     { "cropright", HAS_ARG | OPT_VIDEO, {(void*)opt_frame_crop_right}, "set right crop band size (in pixels)", "size" },

  2946.     { "g", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_gop_size}, "set the group of picture size", "gop_size" },

  2947.     { "intra", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&intra_only}, "use only intra frames"},

  2948.     { "vn", OPT_BOOL | OPT_VIDEO, {(void*)&video_disable}, "disable video" },

  2949.     { "qscale", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_qscale}, "use fixed video quantiser scale (VBR)", "q" },

  2950.     { "qmin", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_qmin}, "min video quantiser scale (VBR)", "q" },

  2951.     { "qmax", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_qmax}, "max video quantiser scale (VBR)", "q" },

  2952.     { "mbqmin", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_mb_qmin}, "min macroblock quantiser scale (VBR)", "q" },

  2953.     { "mbqmax", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_mb_qmax}, "max macroblock quantiser scale (VBR)", "q" },

  2954.     { "qdiff", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_qdiff}, "max difference between the quantiser scale (VBR)", "q" },

  2955.     { "qblur", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_qblur}, "video quantiser scale blur (VBR)", "blur" },

  2956.     { "qcomp", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_qcomp}, "video quantiser scale compression (VBR)", "compression" },

  2957.     { "rc_init_cplx", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_rc_initial_cplx}, "initial complexity for 1-pass encoding", "complexity" },

  2958.     { "b_qfactor", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_b_qfactor}, "qp factor between p and b frames", "factor" },

  2959.     { "i_qfactor", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_i_qfactor}, "qp factor between p and i frames", "factor" },

  2960.     { "b_qoffset", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_b_qoffset}, "qp offset between p and b frames", "offset" },

  2961.     { "i_qoffset", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_i_qoffset}, "qp offset between p and i frames", "offset" },

  2962. //    { "b_strategy", HAS_ARG | OPT_EXPERT, {(void*)opt_b_strategy}, "dynamic b frame selection strategy", "strategy" },

  2963.     { "rc_eq", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_video_rc_eq}, "set rate control equation", "equation" },

  2964.     { "rc_override", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_video_rc_override_string}, "rate control override for specific intervals", "override" },

  2965.     { "bt", HAS_ARG | OPT_VIDEO, {(void*)opt_video_bitrate_tolerance}, "set video bitrate tolerance (in kbit/s)", "tolerance" },

  2966.     { "maxrate", HAS_ARG | OPT_VIDEO, {(void*)opt_video_bitrate_max}, "set max video bitrate tolerance (in kbit/s)", "bitrate" },

  2967.     { "minrate", HAS_ARG | OPT_VIDEO, {(void*)opt_video_bitrate_min}, "set min video bitrate tolerance (in kbit/s)", "bitrate" },

  2968.     { "bufsize", HAS_ARG | OPT_VIDEO, {(void*)opt_video_buffer_size}, "set ratecontrol buffere size (in kbit)", "size" },

  2969.     { "vcodec", HAS_ARG | OPT_VIDEO, {(void*)opt_video_codec}, "force video codec ('copy' to copy stream)", "codec" },

  2970.     { "me", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_motion_estimation}, "set motion estimation method", 

  2971.       "method" },

  2972.     { "dct_algo", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_dct_algo}, "set dct algo",  "algo" },

  2973.     { "idct_algo", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_idct_algo}, "set idct algo",  "algo" },

  2974.     { "er", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_error_resilience}, "set error resilience",  "n" },

  2975.     { "ec", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_error_concealment}, "set error concealment",  "bit_mask" },

  2976.     { "bf", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_b_frames}, "use 'frames' B frames", "frames" },

  2977.     { "hq", OPT_BOOL, {(void*)&mb_decision}, "activate high quality settings" },

  2978.     { "mbd", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_mb_decision}, "macroblock decision", "mode" },

  2979.     { "mbcmp", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_mb_cmp}, "macroblock compare function", "cmp function" },

  2980.     { "subcmp", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_sub_cmp}, "subpel compare function", "cmp function" },

  2981.     { "cmp", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_cmp}, "fullpel compare function", "cmp function" },

  2982.     { "4mv", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&use_4mv}, "use four motion vector by macroblock (MPEG4)" },

  2983.     { "obmc", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&use_obmc}, "use overlapped block motion compensation (h263+)" },

  2984.     { "part", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&use_part}, "use data partitioning (MPEG4)" },

  2985.     { "bug", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_workaround_bugs}, "workaround not auto detected encoder bugs", "param" },

  2986.     { "ps", HAS_ARG | OPT_EXPERT, {(void*)opt_packet_size}, "set packet size in bits", "size" },

  2987.     { "error", HAS_ARG | OPT_EXPERT, {(void*)opt_error_rate}, "error rate", "rate" },

  2988.     { "strict", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_strict}, "how strictly to follow the standarts", "strictness" },

  2989.     { "sameq", OPT_BOOL | OPT_VIDEO, {(void*)&same_quality}, 

  2990.       "use same video quality as source (implies VBR)" },

  2991.     { "pass", HAS_ARG | OPT_VIDEO, {(void*)&opt_pass}, "select the pass number (1 or 2)", "n" },

  2992.     { "passlogfile", HAS_ARG | OPT_STRING | OPT_VIDEO, {(void*)&pass_logfilename}, "select two pass log file name", "file" },

  2993.     { "deinterlace", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&do_deinterlace}, 

  2994.       "deinterlace pictures" },

  2995.     { "interlace", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&do_interlace}, 

  2996.       "force interlacing support in encoder (MPEG2/MPEG4)" },

  2997.     { "psnr", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&do_psnr}, "calculate PSNR of compressed frames" },

  2998.     { "vstats", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&do_vstats}, "dump video coding statistics to file" }, 

  2999.     { "vhook", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)add_frame_hooker}, "insert video processing module", "module" },

  3000.     { "aic", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&use_aic}, "enable Advanced intra coding (h263+)" },

  3001.     { "aiv", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&use_aiv}, "enable Alternative inter vlc (h263+)" },

  3002.     { "umv", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&use_umv}, "enable Unlimited Motion Vector (h263+)" },

  3003.     { "intra_matrix", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_intra_matrix}, "specify intra matrix coeffs", "matrix" },

  3004.     { "inter_matrix", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_inter_matrix}, "specify inter matrix coeffs", "matrix" },

  3005. 

  3006.     /* audio options */

  3007.     { "ab", HAS_ARG | OPT_AUDIO, {(void*)opt_audio_bitrate}, "set audio bitrate (in kbit/s)", "bitrate", },

  3008.     { "ar", HAS_ARG | OPT_AUDIO, {(void*)opt_audio_rate}, "set audio sampling rate (in Hz)", "rate" },

  3009.     { "ac", HAS_ARG | OPT_AUDIO, {(void*)opt_audio_channels}, "set number of audio channels", "channels" },

  3010.     { "an", OPT_BOOL | OPT_AUDIO, {(void*)&audio_disable}, "disable audio" },

  3011.     { "acodec", HAS_ARG | OPT_AUDIO, {(void*)opt_audio_codec}, "force audio codec ('copy' to copy stream)", "codec" },

  3012. 

  3013.     /* grab options */

  3014.     { "vd", HAS_ARG | OPT_EXPERT | OPT_VIDEO | OPT_GRAB, {(void*)opt_video_device}, "set video grab device", "device" },

  3015.     { "vc", HAS_ARG | OPT_EXPERT | OPT_VIDEO | OPT_GRAB, {(void*)opt_video_channel}, "set video grab channel (DV1394 only)", "channel" },

  3016.     { "tvstd", HAS_ARG | OPT_EXPERT | OPT_VIDEO | OPT_GRAB, {(void*)opt_video_standard}, "set television standard (NTSC, PAL (SECAM))", "standard" },

  3017.     { "dv1394", OPT_EXPERT | OPT_GRAB, {(void*)opt_dv1394}, "set DV1394 grab", "" },

  3018.     { "ad", HAS_ARG | OPT_EXPERT | OPT_AUDIO | OPT_GRAB, {(void*)opt_audio_device}, "set audio device", "device" },

  3019.     { NULL, },

  3020. };

  3021. 

  3022. static void show_banner(void)

  3023. {

  3024.     printf("ffmpeg version " FFMPEG_VERSION ", Copyright (c) 2000-2003 Fabrice Bellard\n");

  3025. }

  3026. 

  3027. static void show_license(void)

  3028. {

  3029.     show_banner();

  3030.     printf(

  3031.     "This library is free software; you can redistribute it and/or\n"

  3032.     "modify it under the terms of the GNU Lesser General Public\n"

  3033.     "License as published by the Free Software Foundation; either\n"

  3034.     "version 2 of the License, or (at your option) any later version.\n"

  3035.     "\n"

  3036.     "This library is distributed in the hope that it will be useful,\n"

  3037.     "but WITHOUT ANY WARRANTY; without even the implied warranty of\n"

  3038.     "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU\n"

  3039.     "Lesser General Public License for more details.\n"

  3040.     "\n"

  3041.     "You should have received a copy of the GNU Lesser General Public\n"

  3042.     "License along with this library; if not, write to the Free Software\n"

  3043.     "Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA\n"

  3044.     );

  3045.     exit(1);

  3046. }

  3047. 

  3048. static void show_help(void)

  3049. {

  3050.     show_banner();

  3051.     printf("usage: ffmpeg [[options] -i input_file]... {[options] outfile}...\n"

  3052.            "Hyper fast Audio and Video encoder\n");

  3053.     printf("\n");

  3054.     show_help_options(options, "Main options:\n",

  3055.                       OPT_EXPERT | OPT_AUDIO | OPT_VIDEO, 0);

  3056.     show_help_options(options, "\nVideo options:\n",

  3057.                       OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_GRAB, 

  3058.                       OPT_VIDEO);

  3059.     show_help_options(options, "\nAdvanced Video options:\n",

  3060.                       OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_GRAB, 

  3061.                       OPT_VIDEO | OPT_EXPERT);

  3062.     show_help_options(options, "\nAudio options:\n",

  3063.                       OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_GRAB, 

  3064.                       OPT_AUDIO);

  3065.     show_help_options(options, "\nAdvanced Audio options:\n",

  3066.                       OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_GRAB, 

  3067.                       OPT_AUDIO | OPT_EXPERT);

  3068.     show_help_options(options, "\nAudio/Video grab options:\n",

  3069.                       OPT_GRAB, 

  3070.                       OPT_GRAB);

  3071.     show_help_options(options, "\nAdvanced options:\n",

  3072.                       OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_GRAB, 

  3073.                       OPT_EXPERT);

  3074.     exit(1);

  3075. }

  3076. 

  3077. void parse_arg_file(const char *filename)

  3078. {

  3079.     opt_output_file(filename);

  3080. }

  3081. 

  3082. int main(int argc, char **argv)

  3083. {

  3084.     int i;

  3085.     int64_t ti;

  3086. 

  3087.     av_register_all();

  3088. 

  3089.     if (argc <= 1)

  3090.         show_help();

  3091.     

  3092.     /* parse options */

  3093.     parse_options(argc, argv, options);

  3094. 

  3095.     /* file converter / grab */

  3096.     if (nb_output_files <= 0) {

  3097.         fprintf(stderr, "Must supply at least one output file\n");

  3098.         exit(1);

  3099.     }

  3100.     

  3101.     if (nb_input_files == 0) {

  3102.         prepare_grab();

  3103.     }

  3104. 

  3105.     ti = getutime();

  3106.     av_encode(output_files, nb_output_files, input_files, nb_input_files, 

  3107.               stream_maps, nb_stream_maps);

  3108.     ti = getutime() - ti;

  3109.     if (do_benchmark) {

  3110.         printf("bench: utime=%0.3fs\n", ti / 1000000.0);

  3111.     }

  3112. 

  3113.     /* close files */

  3114.     for(i=0;i<nb_output_files;i++) {

  3115.         /* maybe av_close_output_file ??? */

  3116.         AVFormatContext *s = output_files[i];

  3117. 	int j;

  3118.         if (!(s->oformat->flags & AVFMT_NOFILE))

  3119. 	    url_fclose(&s->pb);

  3120. 	for(j=0;j<s->nb_streams;j++)

  3121. 	    av_free(s->streams[j]);

  3122.         av_free(s);

  3123.     }

  3124.     for(i=0;i<nb_input_files;i++)

  3125.         av_close_input_file(input_files[i]);

  3126. 

  3127.     av_free_static();

  3128. 

  3129.     if(intra_matrix)

  3130.         av_free(intra_matrix);

  3131.     if(inter_matrix)

  3132.         av_free(inter_matrix);

  3133.     

  3134. #ifdef POWERPC_PERFORMANCE_REPORT

  3135.     extern void powerpc_display_perf_report(void);

  3136.     powerpc_display_perf_report();

  3137. #endif /* POWERPC_PERFORMANCE_REPORT */

  3138. 

  3139. #ifndef CONFIG_WIN32

  3140.     if (received_sigterm) {

  3141.         fprintf(stderr,

  3142.             "Received signal %d: terminating.\n",

  3143.             (int) received_sigterm);

  3144.         exit (255);

  3145.     }

  3146. #endif

  3147.     exit(0); /* not all OS-es handle main() return value */

  3148.     return 0;

  3149. }