ffmpeg.c 90.1 KB
Newer Older
F
merge  
Fabrice Bellard 已提交
1 2
/*
 * FFmpeg main 
F
Fabrice Bellard 已提交
3
 * Copyright (c) 2000-2003 Fabrice Bellard
F
merge  
Fabrice Bellard 已提交
4
 *
5 6 7 8
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
F
merge  
Fabrice Bellard 已提交
9
 *
10
 * This library is distributed in the hope that it will be useful,
F
merge  
Fabrice Bellard 已提交
11
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 13
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
F
merge  
Fabrice Bellard 已提交
14
 *
15 16 17
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
F
merge  
Fabrice Bellard 已提交
18
 */
19
#define HAVE_AV_CONFIG_H
A
Alex Beregszaszi 已提交
20
#include "common.h"
21
#include "avformat.h"
22
#include "framehook.h"
F
François Revol 已提交
23 24
/* usleep() */
#include "os_support.h"
25

F
Fabrice Bellard 已提交
26
#ifndef CONFIG_WIN32
F
merge  
Fabrice Bellard 已提交
27 28 29 30 31
#include <unistd.h>
#include <fcntl.h>
#include <sys/ioctl.h>
#include <sys/time.h>
#include <termios.h>
F
Fabrice Bellard 已提交
32
#include <sys/resource.h>
33
#include <signal.h>
F
Fabrice Bellard 已提交
34
#endif
35 36 37 38 39
#ifdef CONFIG_OS2
#include <sys/types.h>
#include <sys/select.h>
#include <stdlib.h>
#endif
40 41
#include <time.h>
#include <ctype.h>
F
merge  
Fabrice Bellard 已提交
42

F
Fabrice Bellard 已提交
43 44
#include "cmdutils.h"

45 46 47
#if !defined(INFINITY) && defined(HUGE_VAL)
#define INFINITY HUGE_VAL
#endif
F
merge  
Fabrice Bellard 已提交
48

49
#define MAXINT64 int64_t_C(0x7fffffffffffffff)
F
Fabrice Bellard 已提交
50

F
merge  
Fabrice Bellard 已提交
51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71
/* select an input stream for an output stream */
typedef struct AVStreamMap {
    int file_index;
    int stream_index;
} AVStreamMap;

extern const OptionDef options[];

void show_help(void);

#define MAX_FILES 20

static AVFormatContext *input_files[MAX_FILES];
static int nb_input_files = 0;

static AVFormatContext *output_files[MAX_FILES];
static int nb_output_files = 0;

static AVStreamMap stream_maps[MAX_FILES];
static int nb_stream_maps;

72 73
static AVInputFormat *file_iformat;
static AVOutputFormat *file_oformat;
74
static AVImageFormat *image_format;
F
merge  
Fabrice Bellard 已提交
75 76
static int frame_width  = 160;
static int frame_height = 128;
77
static float frame_aspect_ratio = 0;
78
static enum PixelFormat frame_pix_fmt = PIX_FMT_YUV420P;
79 80 81 82
static int frame_topBand  = 0;
static int frame_bottomBand = 0;
static int frame_leftBand  = 0;
static int frame_rightBand = 0;
83 84
static int frame_rate = 25;
static int frame_rate_base = 1;
M
Michael Niedermayer 已提交
85 86
static int video_bit_rate = 200*1000;
static int video_bit_rate_tolerance = 4000*1000;
F
merge  
Fabrice Bellard 已提交
87
static int video_qscale = 0;
M
Michael Niedermayer 已提交
88 89
static int video_qmin = 2;
static int video_qmax = 31;
M
Michael Niedermayer 已提交
90 91
static int video_mb_qmin = 2;
static int video_mb_qmax = 31;
92 93 94
static int video_qdiff = 3;
static float video_qblur = 0.5;
static float video_qcomp = 0.5;
95
#if 0 //experimental, (can be removed)
M
Michael Niedermayer 已提交
96 97 98
static float video_rc_qsquish=1.0;
static float video_rc_qmod_amp=0;
static int video_rc_qmod_freq=0;
99
#endif
M
Michael Niedermayer 已提交
100 101 102 103 104 105 106 107 108
static char *video_rc_override_string=NULL;
static char *video_rc_eq="tex^qComp";
static int video_rc_buffer_size=0;
static float video_rc_buffer_aggressivity=1.0;
static int video_rc_max_rate=0;
static int video_rc_min_rate=0;
static float video_rc_initial_cplx=0;
static float video_b_qfactor = 1.25;
static float video_b_qoffset = 1.25;
109
static float video_i_qfactor = -0.8;
M
Michael Niedermayer 已提交
110
static float video_i_qoffset = 0.0;
111
static int me_method = ME_EPZS;
F
merge  
Fabrice Bellard 已提交
112 113 114
static int video_disable = 0;
static int video_codec_id = CODEC_ID_NONE;
static int same_quality = 0;
115
static int b_frames = 0;
116
static int use_hq = 0;
J
Juanjo 已提交
117
static int use_4mv = 0;
118 119 120 121
/* Fx */
static int use_aic = 0;
static int use_umv = 0;
/* /Fx */
122
static int do_deinterlace = 0;
123 124 125
static int workaround_bugs = FF_BUG_AUTODETECT;
static int error_resilience = 2;
static int error_concealment = 3;
126
static int dct_algo = 0;
127
static int idct_algo = 0;
128 129
static int use_part = 0;
static int packet_size = 0;
M
Michael Niedermayer 已提交
130
static int strict = 0;
M
Michael Niedermayer 已提交
131
static int debug = 0;
F
merge  
Fabrice Bellard 已提交
132 133 134 135 136 137 138 139 140

static int gop_size = 12;
static int intra_only = 0;
static int audio_sample_rate = 44100;
static int audio_bit_rate = 64000;
static int audio_disable = 0;
static int audio_channels = 1;
static int audio_codec_id = CODEC_ID_NONE;

141
static int64_t recording_time = 0;
F
merge  
Fabrice Bellard 已提交
142 143 144 145 146
static int file_overwrite = 0;
static char *str_title = NULL;
static char *str_author = NULL;
static char *str_copyright = NULL;
static char *str_comment = NULL;
F
Fabrice Bellard 已提交
147
static int do_benchmark = 0;
148
static int do_hex_dump = 0;
149
static int do_psnr = 0;
150
static int do_vstats = 0;
151
static int do_pass = 0;
152
static int bitexact = 0;
153
static char *pass_logfilename = NULL;
154 155
static int audio_stream_copy = 0;
static int video_stream_copy = 0;
156

157 158
static int rate_emu = 0;

159
static char *video_grab_format = "video4linux";
160
static char *video_device = NULL;
161
static int  video_channel = 0;
162
static char *video_standard = "ntsc";
163

164
static char *audio_grab_format = "audio_device";
165
static char *audio_device = NULL;
166

167
#define DEFAULT_PASS_LOGFILENAME "ffmpeg2pass"
F
merge  
Fabrice Bellard 已提交
168 169 170 171 172 173

typedef struct AVOutputStream {
    int file_index;          /* file index */
    int index;               /* stream index in the output file */
    int source_index;        /* AVInputStream index */
    AVStream *st;            /* stream in the output file */
174 175 176 177 178
    int encoding_needed;     /* true if encoding needed for this stream */
    int frame_number;
    /* input pts and corresponding output pts
       for A/V sync */
    double sync_ipts;
179
    double sync_ipts_offset;
180
    int64_t sync_opts;
F
merge  
Fabrice Bellard 已提交
181
    /* video only */
182 183
    int video_resample;      /* video_resample and video_crop are mutually exclusive */
    AVPicture pict_tmp;      /* temporary image for resampling */
F
merge  
Fabrice Bellard 已提交
184
    ImgReSampleContext *img_resample_ctx; /* for image resampling */
185 186 187 188

    int video_crop;          /* video_resample and video_crop are mutually exclusive */
    int topBand;             /* cropping area sizes */
    int leftBand;
F
merge  
Fabrice Bellard 已提交
189 190 191 192 193
    
    /* audio only */
    int audio_resample;
    ReSampleContext *resample; /* for audio resampling */
    FifoBuffer fifo;     /* for compression: one audio fifo per codec */
194
    FILE *logfile;
F
merge  
Fabrice Bellard 已提交
195 196 197 198 199 200 201 202
} AVOutputStream;

typedef struct AVInputStream {
    int file_index;
    int index;
    AVStream *st;
    int discard;             /* true if stream data should be discarded */
    int decoding_needed;     /* true if the packets must be decoded in 'raw_fifo' */
203
    int64_t sample_index;      /* current sample */
204
    int frame_decoded;       /* true if a video or audio frame has been decoded */
205 206 207

    int64_t       start;     /* time when read started */
    unsigned long frame;     /* current frame */
F
merge  
Fabrice Bellard 已提交
208 209 210 211 212 213 214 215
} AVInputStream;

typedef struct AVInputFile {
    int eof_reached;      /* true if eof reached */
    int ist_index;        /* index of first stream in ist_table */
    int buffer_size;      /* current total buffer size */
    int buffer_size_max;  /* buffer size at which we consider we can stop
                             buffering */
216
    int nb_streams;       /* nb streams we are aware of */
F
merge  
Fabrice Bellard 已提交
217 218
} AVInputFile;

F
Fabrice Bellard 已提交
219 220
#ifndef CONFIG_WIN32

F
merge  
Fabrice Bellard 已提交
221 222 223 224 225 226 227 228
/* init terminal so that we can grab keys */
static struct termios oldtty;

static void term_exit(void)
{
    tcsetattr (0, TCSANOW, &oldtty);
}

229 230 231 232 233 234 235 236 237
static volatile sig_atomic_t received_sigterm = 0;

static void
sigterm_handler(int sig)
{
    received_sigterm = sig;
    term_exit();
}

F
merge  
Fabrice Bellard 已提交
238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255
static void term_init(void)
{
    struct termios tty;

    tcgetattr (0, &tty);
    oldtty = tty;

    tty.c_iflag &= ~(IGNBRK|BRKINT|PARMRK|ISTRIP
                          |INLCR|IGNCR|ICRNL|IXON);
    tty.c_oflag |= OPOST;
    tty.c_lflag &= ~(ECHO|ECHONL|ICANON|IEXTEN);
    tty.c_cflag &= ~(CSIZE|PARENB);
    tty.c_cflag |= CS8;
    tty.c_cc[VMIN] = 1;
    tty.c_cc[VTIME] = 0;
    
    tcsetattr (0, TCSANOW, &tty);

256 257 258 259 260 261
    signal(SIGINT , sigterm_handler); /* Interrupt (ANSI).  */
    signal(SIGQUIT, sigterm_handler); /* Quit (POSIX).  */
    signal(SIGTERM, sigterm_handler); /* Termination (ANSI).  */
    /*
    register a function to be called at normal program termination
    */
F
merge  
Fabrice Bellard 已提交
262
    atexit(term_exit);
263 264 265
#ifdef CONFIG_BEOS_NETSERVER
    fcntl(0, F_SETFL, fcntl(0, F_GETFL) | O_NONBLOCK);
#endif
F
merge  
Fabrice Bellard 已提交
266 267 268 269 270
}

/* read a key without blocking */
static int read_key(void)
{
271
    int n = 1;
F
merge  
Fabrice Bellard 已提交
272
    unsigned char ch;
273 274
#ifndef CONFIG_BEOS_NETSERVER
    struct timeval tv;
F
merge  
Fabrice Bellard 已提交
275 276 277 278 279 280 281
    fd_set rfds;

    FD_ZERO(&rfds);
    FD_SET(0, &rfds);
    tv.tv_sec = 0;
    tv.tv_usec = 0;
    n = select(1, &rfds, NULL, NULL, &tv);
282
#endif
F
merge  
Fabrice Bellard 已提交
283
    if (n > 0) {
284 285
        n = read(0, &ch, 1);
        if (n == 1)
F
merge  
Fabrice Bellard 已提交
286
            return ch;
287 288

        return n;
F
merge  
Fabrice Bellard 已提交
289 290 291 292
    }
    return -1;
}

293
#else
F
merge  
Fabrice Bellard 已提交
294

295 296
/* no interactive support */
static void term_exit(void)
F
merge  
Fabrice Bellard 已提交
297
{
298
}
F
merge  
Fabrice Bellard 已提交
299

300 301 302
static void term_init(void)
{
}
F
merge  
Fabrice Bellard 已提交
303

304 305
static int read_key(void)
{
306
    return 0;
F
merge  
Fabrice Bellard 已提交
307 308
}

309
#endif
F
Fabrice Bellard 已提交
310

311
static int read_ffserver_streams(AVFormatContext *s, const char *filename)
F
merge  
Fabrice Bellard 已提交
312
{
313
    int i, err;
F
merge  
Fabrice Bellard 已提交
314 315
    AVFormatContext *ic;

316 317 318
    err = av_open_input_file(&ic, filename, NULL, FFM_PACKET_SIZE, NULL);
    if (err < 0)
        return err;
F
merge  
Fabrice Bellard 已提交
319 320 321 322
    /* copy stream format */
    s->nb_streams = ic->nb_streams;
    for(i=0;i<ic->nb_streams;i++) {
        AVStream *st;
M
cleanup  
Michael Niedermayer 已提交
323

324
        st = av_mallocz(sizeof(AVStream));
F
merge  
Fabrice Bellard 已提交
325 326 327 328 329 330 331 332
        memcpy(st, ic->streams[i], sizeof(AVStream));
        s->streams[i] = st;
    }

    av_close_input_file(ic);
    return 0;
}

333
#define MAX_AUDIO_PACKET_SIZE (128 * 1024)
F
merge  
Fabrice Bellard 已提交
334 335 336 337 338 339

static void do_audio_out(AVFormatContext *s, 
                         AVOutputStream *ost, 
                         AVInputStream *ist,
                         unsigned char *buf, int size)
{
340
    uint8_t *buftmp;
341 342
    static uint8_t *audio_buf = NULL;
    static uint8_t *audio_out = NULL;
M
10l  
Michael Niedermayer 已提交
343
    const int audio_out_size= 4*MAX_AUDIO_PACKET_SIZE;
344

F
merge  
Fabrice Bellard 已提交
345 346 347
    int size_out, frame_bytes, ret;
    AVCodecContext *enc;

348 349 350 351
    /* SC: dynamic allocation of buffers */
    if (!audio_buf)
        audio_buf = av_malloc(2*MAX_AUDIO_PACKET_SIZE);
    if (!audio_out)
M
10l  
Michael Niedermayer 已提交
352
        audio_out = av_malloc(audio_out_size);
353 354 355 356
    if (!audio_buf || !audio_out)
        return;               /* Should signal an error ! */

    
F
merge  
Fabrice Bellard 已提交
357 358 359 360 361 362 363 364 365 366 367 368 369 370
    enc = &ost->st->codec;

    if (ost->audio_resample) {
        buftmp = audio_buf;
        size_out = audio_resample(ost->resample, 
                                  (short *)buftmp, (short *)buf,
                                  size / (ist->st->codec.channels * 2));
        size_out = size_out * enc->channels * 2;
    } else {
        buftmp = buf;
        size_out = size;
    }

    /* now encode as many frames as possible */
371
    if (enc->frame_size > 1) {
F
merge  
Fabrice Bellard 已提交
372 373 374 375 376 377 378 379
        /* output resampled raw samples */
        fifo_write(&ost->fifo, buftmp, size_out, 
                   &ost->fifo.wptr);

        frame_bytes = enc->frame_size * 2 * enc->channels;
        
        while (fifo_read(&ost->fifo, audio_buf, frame_bytes, 
                     &ost->fifo.rptr) == 0) {
M
10l  
Michael Niedermayer 已提交
380
            ret = avcodec_encode_audio(enc, audio_out, audio_out_size, 
381
                                       (short *)audio_buf);
382
            av_write_frame(s, ost->index, audio_out, ret);
F
merge  
Fabrice Bellard 已提交
383 384
        }
    } else {
385 386 387 388 389 390 391 392 393 394 395 396 397
        /* output a pcm frame */
        /* XXX: change encoding codec API to avoid this ? */
        switch(enc->codec->id) {
        case CODEC_ID_PCM_S16LE:
        case CODEC_ID_PCM_S16BE:
        case CODEC_ID_PCM_U16LE:
        case CODEC_ID_PCM_U16BE:
            break;
        default:
            size_out = size_out >> 1;
            break;
        }
        ret = avcodec_encode_audio(enc, audio_out, size_out, 
398
				   (short *)buftmp);
399
        av_write_frame(s, ost->index, audio_out, ret);
F
merge  
Fabrice Bellard 已提交
400 401 402
    }
}

403 404 405 406 407
static void pre_process_video_frame(AVInputStream *ist, AVPicture *picture, void **bufp)
{
    AVCodecContext *dec;
    AVPicture *picture2;
    AVPicture picture_tmp;
408
    uint8_t *buf = 0;
409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442

    dec = &ist->st->codec;

    /* deinterlace : must be done before any resize */
    if (do_deinterlace) {
        int size;

        /* create temporary picture */
        size = avpicture_get_size(dec->pix_fmt, dec->width, dec->height);
        buf = av_malloc(size);
        if (!buf)
            return;
        
        picture2 = &picture_tmp;
        avpicture_fill(picture2, buf, dec->pix_fmt, dec->width, dec->height);

        if (avpicture_deinterlace(picture2, picture, 
                                  dec->pix_fmt, dec->width, dec->height) < 0) {
            /* if error, do not deinterlace */
            av_free(buf);
            buf = NULL;
            picture2 = picture;
        }
    } else {
        picture2 = picture;
    }

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

    if (picture != picture2)
        *picture = *picture2;
    *bufp = buf;
}

443 444
/* we begin to correct av delay at this threshold */
#define AV_DELAY_MAX 0.100
F
merge  
Fabrice Bellard 已提交
445 446 447 448

static void do_video_out(AVFormatContext *s, 
                         AVOutputStream *ost, 
                         AVInputStream *ist,
449
                         AVPicture *in_picture,
450
                         int *frame_size, AVOutputStream *audio_sync)
F
merge  
Fabrice Bellard 已提交
451
{
452
    int nb_frames, i, ret;
453 454
    AVPicture *final_picture, *formatted_picture;
    AVPicture picture_format_temp, picture_crop_temp;
455 456
    static uint8_t *video_buffer;
    uint8_t *buf = NULL, *buf1 = NULL;
457
    AVCodecContext *enc, *dec;
F
merge  
Fabrice Bellard 已提交
458

M
Michael Niedermayer 已提交
459 460
#define VIDEO_BUFFER_SIZE (1024*1024)

F
merge  
Fabrice Bellard 已提交
461
    enc = &ost->st->codec;
462
    dec = &ist->st->codec;
F
merge  
Fabrice Bellard 已提交
463

464 465 466
    /* by default, we output a single frame */
    nb_frames = 1;

467 468
    *frame_size = 0;

469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493
    /* NOTE: the A/V sync is always done by considering the audio is
       the master clock. It is suffisant for transcoding or playing,
       but not for the general case */
    if (audio_sync) {
        /* compute the A-V delay and duplicate/remove frames if needed */
        double adelta, vdelta, apts, vpts, av_delay;
        
        if (audio_sync->sync_ipts != AV_NOPTS_VALUE &&
            ost->sync_ipts != AV_NOPTS_VALUE) {
            
            adelta = (double)(ost->st->pts.val - audio_sync->sync_opts) * 
                s->pts_num / s->pts_den;
            apts = audio_sync->sync_ipts + adelta; 
            
            vdelta = (double)(ost->st->pts.val - ost->sync_opts) *
                s->pts_num / s->pts_den;
            vpts = ost->sync_ipts + vdelta;
            
            av_delay = apts - vpts;
            //            printf("delay=%f\n", av_delay);
            if (av_delay < -AV_DELAY_MAX)
                nb_frames = 2;
            else if (av_delay > AV_DELAY_MAX)
                nb_frames = 0;
        }
494 495 496 497
    } else {
        double vdelta;

        if (ost->sync_ipts != AV_NOPTS_VALUE) {
498
            vdelta = (double)(ost->st->pts.val) * s->pts_num / s->pts_den - (ost->sync_ipts - ost->sync_ipts_offset);
499
            if (vdelta < 100 && vdelta > -100 && ost->sync_ipts_offset) {
500 501 502 503 504 505
                if (vdelta < -AV_DELAY_MAX)
                    nb_frames = 2;
                else if (vdelta > AV_DELAY_MAX)
                    nb_frames = 0;
            } else {
                ost->sync_ipts_offset -= vdelta;
506 507
                if (!ost->sync_ipts_offset)
                    ost->sync_ipts_offset = 0.000001; /* one microsecond */
508 509
            }

510
#if defined(PJSG)
511 512 513 514 515 516 517
            {
                static char *action[] = { "drop frame", "copy frame", "dup frame" };
                printf("Input PTS %12.6f, output PTS %12.6f: %s\n",
                    (double) ost->sync_ipts, (double) ost->st->pts.val * s->pts_num / s->pts_den,
                    action[nb_frames]);
            }
#endif
518
        }
519 520
    }
    if (nb_frames <= 0) 
F
merge  
Fabrice Bellard 已提交
521
        return;
522

523
    if (!video_buffer)
524 525 526
        video_buffer = av_malloc(VIDEO_BUFFER_SIZE);
    if (!video_buffer)
        return;
527

528 529 530 531 532 533
    /* convert pixel format if needed */
    if (enc->pix_fmt != dec->pix_fmt) {
        int size;

        /* create temporary picture */
        size = avpicture_get_size(enc->pix_fmt, dec->width, dec->height);
F
Fabrice Bellard 已提交
534
        buf = av_malloc(size);
535 536
        if (!buf)
            return;
537 538
        formatted_picture = &picture_format_temp;
        avpicture_fill(formatted_picture, buf, enc->pix_fmt, dec->width, dec->height);
539
        
540 541
        if (img_convert(formatted_picture, enc->pix_fmt, 
                        in_picture, dec->pix_fmt, 
542 543 544 545 546
                        dec->width, dec->height) < 0) {
            fprintf(stderr, "pixel format conversion not handled\n");
            goto the_end;
        }
    } else {
547
        formatted_picture = in_picture;
548 549 550 551 552
    }

    /* XXX: resampling could be done before raw format convertion in
       some cases to go faster */
    /* XXX: only works for YUV420P */
F
merge  
Fabrice Bellard 已提交
553
    if (ost->video_resample) {
554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571
        final_picture = &ost->pict_tmp;
        img_resample(ost->img_resample_ctx, final_picture, formatted_picture);
    } else if (ost->video_crop) {
        picture_crop_temp.data[0] = formatted_picture->data[0] +
                (ost->topBand * formatted_picture->linesize[0]) + ost->leftBand;

        picture_crop_temp.data[1] = formatted_picture->data[1] +
                ((ost->topBand >> 1) * formatted_picture->linesize[1]) +
                (ost->leftBand >> 1);

        picture_crop_temp.data[2] = formatted_picture->data[2] +
                ((ost->topBand >> 1) * formatted_picture->linesize[2]) +
                (ost->leftBand >> 1);

        picture_crop_temp.linesize[0] = formatted_picture->linesize[0];
        picture_crop_temp.linesize[1] = formatted_picture->linesize[1];
        picture_crop_temp.linesize[2] = formatted_picture->linesize[2];
        final_picture = &picture_crop_temp;
F
merge  
Fabrice Bellard 已提交
572
    } else {
573
        final_picture = formatted_picture;
F
merge  
Fabrice Bellard 已提交
574 575 576
    }
    /* duplicates frame if needed */
    /* XXX: pb because no interleaving */
577
    for(i=0;i<nb_frames;i++) {
578 579 580 581 582 583 584
        if (s->oformat->flags & AVFMT_RAWPICTURE) {
            /* raw pictures are written as AVPicture structure to
               avoid any copies. We support temorarily the older
               method. */
            av_write_frame(s, ost->index, 
                           (uint8_t *)final_picture, sizeof(AVPicture));
        } else {
M
Michael Niedermayer 已提交
585
            AVFrame big_picture;
M
cleanup  
Michael Niedermayer 已提交
586
            
M
Michael Niedermayer 已提交
587
            memset(&big_picture, 0, sizeof(AVFrame));
588
            *(AVPicture*)&big_picture= *final_picture;
M
cleanup  
Michael Niedermayer 已提交
589
                        
F
merge  
Fabrice Bellard 已提交
590 591 592
            /* handles sameq here. This is not correct because it may
               not be a global option */
            if (same_quality) {
M
cleanup  
Michael Niedermayer 已提交
593 594 595
                big_picture.quality = ist->st->quality;
            }else
                big_picture.quality = ost->st->quality;
596
            
597
            ret = avcodec_encode_video(enc, 
M
Michael Niedermayer 已提交
598
                                       video_buffer, VIDEO_BUFFER_SIZE,
M
cleanup  
Michael Niedermayer 已提交
599
                                       &big_picture);
600
            //enc->frame_number = enc->real_pict_num;
601
            av_write_frame(s, ost->index, video_buffer, ret);
602
            *frame_size = ret;
J
Juanjo 已提交
603
            //fprintf(stderr,"\nFrame: %3d %3d size: %5d type: %d",
604
            //        enc->frame_number-1, enc->real_pict_num, ret,
J
Juanjo 已提交
605
            //        enc->pict_type);
606 607 608 609
            /* if two pass, output log */
            if (ost->logfile && enc->stats_out) {
                fprintf(ost->logfile, "%s", enc->stats_out);
            }
F
merge  
Fabrice Bellard 已提交
610
        }
611
        ost->frame_number++;
F
merge  
Fabrice Bellard 已提交
612
    }
613
 the_end:
F
Fabrice Bellard 已提交
614 615
    av_free(buf);
    av_free(buf1);
F
merge  
Fabrice Bellard 已提交
616 617
}

618 619
static double psnr(double d){
    if(d==0) return INFINITY;
620
    return -10.0*log(d)/log(10.0);
621 622
}

623 624
static void do_video_stats(AVFormatContext *os, AVOutputStream *ost, 
                           int frame_size)
625 626
{
    static FILE *fvstats=NULL;
627
    static int64_t total_size = 0;
628
    char filename[40];
629 630
    time_t today2;
    struct tm *today;
631 632
    AVCodecContext *enc;
    int frame_number;
633
    int64_t ti;
634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652
    double ti1, bitrate, avg_bitrate;
    
    if (!fvstats) {
        today2 = time(NULL);
        today = localtime(&today2);
        sprintf(filename, "vstats_%02d%02d%02d.log", today->tm_hour,
                                               today->tm_min,
                                               today->tm_sec);
        fvstats = fopen(filename,"w");
        if (!fvstats) {
            perror("fopen");
            exit(1);
        }
    }
    
    ti = MAXINT64;
    enc = &ost->st->codec;
    total_size += frame_size;
    if (enc->codec_type == CODEC_TYPE_VIDEO) {
653
        frame_number = ost->frame_number;
M
Michael Niedermayer 已提交
654
        fprintf(fvstats, "frame= %5d q= %2.1f ", frame_number, enc->coded_frame->quality);
655
        if (enc->flags&CODEC_FLAG_PSNR)
M
Michael Niedermayer 已提交
656
            fprintf(fvstats, "PSNR= %6.2f ", psnr(enc->coded_frame->error[0]/(enc->width*enc->height*255.0*255.0)));
657 658
        
        fprintf(fvstats,"f_size= %6d ", frame_size);
659 660
        /* compute pts value */
        ti1 = (double)ost->st->pts.val * os->pts_num / os->pts_den;
661 662 663
        if (ti1 < 0.01)
            ti1 = 0.01;
    
664
        bitrate = (double)(frame_size * 8) * enc->frame_rate / enc->frame_rate_base / 1000.0;
665 666 667
        avg_bitrate = (double)(total_size * 8) / ti1 / 1000.0;
        fprintf(fvstats, "s_size= %8.0fkB time= %0.3f br= %7.1fkbits/s avg_br= %7.1fkbits/s ",
            (double)total_size / 1024, ti1, bitrate, avg_bitrate);
F
Falk Hüffner 已提交
668
        fprintf(fvstats,"type= %c\n", av_get_pict_type_char(enc->coded_frame->pict_type));        
669
    }
670 671
}

672 673 674
static void print_report(AVFormatContext **output_files,
			 AVOutputStream **ost_table, int nb_ostreams,
			 int is_last_report)
675 676 677 678
{
    char buf[1024];
    AVOutputStream *ost;
    AVFormatContext *oc, *os;
679
    int64_t total_size;
680 681 682
    AVCodecContext *enc;
    int frame_number, vid, i;
    double bitrate, ti1, pts;
683
    static int64_t last_time = -1;
684 685
    
    if (!is_last_report) {
686
        int64_t cur_time;
687 688 689 690 691 692 693 694 695 696 697
        /* display the report every 0.5 seconds */
        cur_time = av_gettime();
        if (last_time == -1) {
            last_time = cur_time;
            return;
        } 
        if ((cur_time - last_time) < 500000)
            return;
        last_time = cur_time;
    }

698

699 700 701 702 703 704 705 706 707 708 709
    oc = output_files[0];

    total_size = url_ftell(&oc->pb);
    
    buf[0] = '\0';
    ti1 = 1e10;
    vid = 0;
    for(i=0;i<nb_ostreams;i++) {
        ost = ost_table[i];
        os = output_files[ost->file_index];
        enc = &ost->st->codec;
710
        if (vid && enc->codec_type == CODEC_TYPE_VIDEO) {
M
cleanup  
Michael Niedermayer 已提交
711
            sprintf(buf + strlen(buf), "q=%2.1f ",
M
Michael Niedermayer 已提交
712
                    enc->coded_frame->quality);
713
        }
714 715
        if (!vid && enc->codec_type == CODEC_TYPE_VIDEO) {
            frame_number = ost->frame_number;
M
cleanup  
Michael Niedermayer 已提交
716
            sprintf(buf + strlen(buf), "frame=%5d q=%2.1f ",
M
Michael Niedermayer 已提交
717
                    frame_number, enc->coded_frame ? enc->coded_frame->quality : 0);
718
            if (enc->flags&CODEC_FLAG_PSNR)
M
Michael Niedermayer 已提交
719
                sprintf(buf + strlen(buf), "PSNR= %6.2f ", psnr(enc->coded_frame->error[0]/(enc->width*enc->height*255.0*255.0)));
720 721 722 723
            vid = 1;
        }
        /* compute min output value */
        pts = (double)ost->st->pts.val * os->pts_num / os->pts_den;
724
        if ((pts < ti1) && (pts > 0))
725 726 727 728 729 730 731 732 733
            ti1 = pts;
    }
    if (ti1 < 0.01)
        ti1 = 0.01;
    bitrate = (double)(total_size * 8) / ti1 / 1000.0;
    
    sprintf(buf + strlen(buf), 
            "size=%8.0fkB time=%0.1f bitrate=%6.1fkbits/s",
            (double)total_size / 1024, ti1, bitrate);
734
    
735
    fprintf(stderr, "%s   ", buf);
736
    
737 738 739 740 741 742
    if (is_last_report) {
        fprintf(stderr, "\n");
    } else {
        fprintf(stderr, "\r");
        fflush(stderr);
    }
743 744
}

F
merge  
Fabrice Bellard 已提交
745 746 747 748 749 750 751 752 753
/*
 * The following code is the main loop of the file converter
 */
static int av_encode(AVFormatContext **output_files,
                     int nb_output_files,
                     AVFormatContext **input_files,
                     int nb_input_files,
                     AVStreamMap *stream_maps, int nb_stream_maps)
{
754
    int ret, i, j, k, n, nb_istreams = 0, nb_ostreams = 0, pts_set;
F
merge  
Fabrice Bellard 已提交
755 756 757 758
    AVFormatContext *is, *os;
    AVCodecContext *codec, *icodec;
    AVOutputStream *ost, **ost_table = NULL;
    AVInputStream *ist, **ist_table = NULL;
F
Fabrice Bellard 已提交
759
    AVInputFile *file_table;
760
    AVFormatContext *stream_no_data;
761
    int key;
F
Fabrice Bellard 已提交
762

763
    file_table= (AVInputFile*) av_mallocz(nb_input_files * sizeof(AVInputFile));
F
Fabrice Bellard 已提交
764 765
    if (!file_table)
        goto fail;
F
merge  
Fabrice Bellard 已提交
766 767 768 769 770 771

    /* input stream init */
    j = 0;
    for(i=0;i<nb_input_files;i++) {
        is = input_files[i];
        file_table[i].ist_index = j;
772
        file_table[i].nb_streams = is->nb_streams;
F
merge  
Fabrice Bellard 已提交
773 774 775 776 777 778
        j += is->nb_streams;
    }
    nb_istreams = j;

    ist_table = av_mallocz(nb_istreams * sizeof(AVInputStream *));
    if (!ist_table)
F
Fabrice Bellard 已提交
779
        goto fail;
F
merge  
Fabrice Bellard 已提交
780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796
    
    for(i=0;i<nb_istreams;i++) {
        ist = av_mallocz(sizeof(AVInputStream));
        if (!ist)
            goto fail;
        ist_table[i] = ist;
    }
    j = 0;
    for(i=0;i<nb_input_files;i++) {
        is = input_files[i];
        for(k=0;k<is->nb_streams;k++) {
            ist = ist_table[j++];
            ist->st = is->streams[k];
            ist->file_index = i;
            ist->index = k;
            ist->discard = 1; /* the stream is discarded by default
                                 (changed later) */
797 798 799 800 801

            if (ist->st->codec.rate_emu) {
                ist->start = av_gettime();
                ist->frame = 0;
            }
F
merge  
Fabrice Bellard 已提交
802 803 804 805 806 807 808 809 810 811 812 813 814 815
        }
    }

    /* output stream init */
    nb_ostreams = 0;
    for(i=0;i<nb_output_files;i++) {
        os = output_files[i];
        nb_ostreams += os->nb_streams;
    }
    if (nb_stream_maps > 0 && nb_stream_maps != nb_ostreams) {
        fprintf(stderr, "Number of stream maps must match number of output streams\n");
        exit(1);
    }

816 817 818 819 820 821 822 823 824 825 826 827
    /* Sanity check the mapping args -- do the input files & streams exist? */
    for(i=0;i<nb_stream_maps;i++) {
        int fi = stream_maps[i].file_index;
        int si = stream_maps[i].stream_index;
        
        if (fi < 0 || fi > nb_input_files - 1 ||
            si < 0 || si > file_table[fi].nb_streams - 1) {
            fprintf(stderr,"Could not find input stream #%d.%d\n", fi, si);
            exit(1);
        }
    }
    
F
merge  
Fabrice Bellard 已提交
828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849
    ost_table = av_mallocz(sizeof(AVOutputStream *) * nb_ostreams);
    if (!ost_table)
        goto fail;
    for(i=0;i<nb_ostreams;i++) {
        ost = av_mallocz(sizeof(AVOutputStream));
        if (!ost)
            goto fail;
        ost_table[i] = ost;
    }
    
    n = 0;
    for(k=0;k<nb_output_files;k++) {
        os = output_files[k];
        for(i=0;i<os->nb_streams;i++) {
            int found;
            ost = ost_table[n++];
            ost->file_index = k;
            ost->index = i;
            ost->st = os->streams[i];
            if (nb_stream_maps > 0) {
                ost->source_index = file_table[stream_maps[n-1].file_index].ist_index + 
                    stream_maps[n-1].stream_index;
850 851 852 853 854 855 856 857 858
                    
                /* Sanity check that the stream types match */
                if (ist_table[ost->source_index]->st->codec.codec_type != ost->st->codec.codec_type) {
                    fprintf(stderr, "Codec type mismatch for mapping #%d.%d -> #%d.%d\n",
                        stream_maps[n-1].file_index, stream_maps[n-1].stream_index,
                        ost->file_index, ost->index);
                    exit(1);
                }
                
F
merge  
Fabrice Bellard 已提交
859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899
            } else {
                /* get corresponding input stream index : we select the first one with the right type */
                found = 0;
                for(j=0;j<nb_istreams;j++) {
                    ist = ist_table[j];
                    if (ist->discard && 
                        ist->st->codec.codec_type == ost->st->codec.codec_type) {
                        ost->source_index = j;
                        found = 1;
                    }
                }
                
                if (!found) {
                    /* try again and reuse existing stream */
                    for(j=0;j<nb_istreams;j++) {
                        ist = ist_table[j];
                        if (ist->st->codec.codec_type == ost->st->codec.codec_type) {
                            ost->source_index = j;
                            found = 1;
                        }
                    }
                    if (!found) {
                        fprintf(stderr, "Could not find input stream matching output stream #%d.%d\n",
                                ost->file_index, ost->index);
                        exit(1);
                    }
                }
            }
            ist = ist_table[ost->source_index];
            ist->discard = 0;
        }
    }

    /* for each output stream, we compute the right encoding parameters */
    for(i=0;i<nb_ostreams;i++) {
        ost = ost_table[i];
        ist = ist_table[ost->source_index];

        codec = &ost->st->codec;
        icodec = &ist->st->codec;

900 901 902 903 904 905 906 907 908 909 910 911 912
        if (ost->st->stream_copy) {
            /* if stream_copy is selected, no need to decode or encode */
            codec->codec_id = icodec->codec_id;
            codec->codec_type = icodec->codec_type;
            codec->codec_tag = icodec->codec_tag;
            codec->bit_rate = icodec->bit_rate;
            switch(codec->codec_type) {
            case CODEC_TYPE_AUDIO:
                codec->sample_rate = icodec->sample_rate;
                codec->channels = icodec->channels;
                break;
            case CODEC_TYPE_VIDEO:
                codec->frame_rate = icodec->frame_rate;
913
                codec->frame_rate_base = icodec->frame_rate_base;
914 915 916 917 918 919 920 921 922
                codec->width = icodec->width;
                codec->height = icodec->height;
                break;
            default:
                av_abort();
            }
        } else {
            switch(codec->codec_type) {
            case CODEC_TYPE_AUDIO:
F
merge  
Fabrice Bellard 已提交
923 924
                if (fifo_init(&ost->fifo, 2 * MAX_AUDIO_PACKET_SIZE))
                    goto fail;
925
                
F
merge  
Fabrice Bellard 已提交
926 927 928 929
                if (codec->channels == icodec->channels &&
                    codec->sample_rate == icodec->sample_rate) {
                    ost->audio_resample = 0;
                } else {
930 931 932 933
                    if (codec->channels != icodec->channels &&
                        icodec->codec_id == CODEC_ID_AC3) {
                        /* Special case for 5:1 AC3 input */
                        /* and mono or stereo output      */
934 935 936 937 938 939 940 941 942 943
                        /* Request specific number of channels */
                        icodec->channels = codec->channels;
                        if (codec->sample_rate == icodec->sample_rate)
                            ost->audio_resample = 0;
                        else {
                            ost->audio_resample = 1;
                            ost->resample = audio_resample_init(codec->channels, icodec->channels,
                                                        codec->sample_rate, 
                                                        icodec->sample_rate);
                        }
944 945 946 947 948
                        /* Request specific number of channels */
                        icodec->channels = codec->channels;
                    } else {
                        ost->audio_resample = 1; 
                        ost->resample = audio_resample_init(codec->channels, icodec->channels,
F
merge  
Fabrice Bellard 已提交
949 950
                                                        codec->sample_rate, 
                                                        icodec->sample_rate);
951
                    }
F
merge  
Fabrice Bellard 已提交
952 953 954
                }
                ist->decoding_needed = 1;
                ost->encoding_needed = 1;
955 956
                break;
            case CODEC_TYPE_VIDEO:
F
merge  
Fabrice Bellard 已提交
957
                if (codec->width == icodec->width &&
958 959 960 961 962 963 964 965 966 967 968 969 970
                    codec->height == icodec->height &&
                    frame_topBand == 0 &&
                    frame_bottomBand == 0 &&
                    frame_leftBand == 0 &&
                    frame_rightBand == 0)
                {
                    ost->video_resample = 0;
                    ost->video_crop = 0;
                } else if ((codec->width == icodec->width -
                                (frame_leftBand + frame_rightBand)) &&
                        (codec->height == icodec->height -
                                (frame_topBand  + frame_bottomBand)))
                {
F
merge  
Fabrice Bellard 已提交
971
                    ost->video_resample = 0;
972 973 974
                    ost->video_crop = 1;
                    ost->topBand = frame_topBand;
                    ost->leftBand = frame_leftBand;
F
merge  
Fabrice Bellard 已提交
975
                } else {
976
                    uint8_t *buf;
F
merge  
Fabrice Bellard 已提交
977
                    ost->video_resample = 1;
978
                    ost->video_crop = 0; // cropping is handled as part of resample
F
Fabrice Bellard 已提交
979
                    buf = av_malloc((codec->width * codec->height * 3) / 2);
F
merge  
Fabrice Bellard 已提交
980 981 982 983 984 985 986 987 988
                    if (!buf)
                        goto fail;
                    ost->pict_tmp.data[0] = buf;
                    ost->pict_tmp.data[1] = ost->pict_tmp.data[0] + (codec->width * codec->height);
                    ost->pict_tmp.data[2] = ost->pict_tmp.data[1] + (codec->width * codec->height) / 4;
                    ost->pict_tmp.linesize[0] = codec->width;
                    ost->pict_tmp.linesize[1] = codec->width / 2;
                    ost->pict_tmp.linesize[2] = codec->width / 2;

989
                    ost->img_resample_ctx = img_resample_full_init( 
F
merge  
Fabrice Bellard 已提交
990
                                      ost->st->codec.width, ost->st->codec.height,
991 992 993
                                      ist->st->codec.width, ist->st->codec.height,
                                      frame_topBand, frame_bottomBand,
                                      frame_leftBand, frame_rightBand);
F
merge  
Fabrice Bellard 已提交
994 995 996
                }
                ost->encoding_needed = 1;
                ist->decoding_needed = 1;
997 998 999
                break;
            default:
                av_abort();
F
merge  
Fabrice Bellard 已提交
1000
            }
1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037
            /* two pass mode */
            if (ost->encoding_needed && 
                (codec->flags & (CODEC_FLAG_PASS1 | CODEC_FLAG_PASS2))) {
                char logfilename[1024];
                FILE *f;
                int size;
                char *logbuffer;
                
                snprintf(logfilename, sizeof(logfilename), "%s-%d.log", 
                         pass_logfilename ? 
                         pass_logfilename : DEFAULT_PASS_LOGFILENAME, i);
                if (codec->flags & CODEC_FLAG_PASS1) {
                    f = fopen(logfilename, "w");
                    if (!f) {
                        perror(logfilename);
                        exit(1);
                    }
                    ost->logfile = f;
                } else {
                    /* read the log file */
                    f = fopen(logfilename, "r");
                    if (!f) {
                        perror(logfilename);
                        exit(1);
                    }
                    fseek(f, 0, SEEK_END);
                    size = ftell(f);
                    fseek(f, 0, SEEK_SET);
                    logbuffer = av_malloc(size + 1);
                    if (!logbuffer) {
                        fprintf(stderr, "Could not allocate log buffer\n");
                        exit(1);
                    }
                    fread(logbuffer, 1, size, f);
                    fclose(f);
                    logbuffer[size] = '\0';
                    codec->stats_in = logbuffer;
1038 1039 1040
                }
            }
        }
F
merge  
Fabrice Bellard 已提交
1041 1042
    }

1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059
    /* dump the file output parameters - cannot be done before in case
       of stream copy */
    for(i=0;i<nb_output_files;i++) {
        dump_format(output_files[i], i, output_files[i]->filename, 1);
    }

    /* dump the stream mapping */
    fprintf(stderr, "Stream mapping:\n");
    for(i=0;i<nb_ostreams;i++) {
        ost = ost_table[i];
        fprintf(stderr, "  Stream #%d.%d -> #%d.%d\n",
                ist_table[ost->source_index]->file_index,
                ist_table[ost->source_index]->index,
                ost->file_index, 
                ost->index);
    }

F
merge  
Fabrice Bellard 已提交
1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085
    /* open each encoder */
    for(i=0;i<nb_ostreams;i++) {
        ost = ost_table[i];
        if (ost->encoding_needed) {
            AVCodec *codec;
            codec = avcodec_find_encoder(ost->st->codec.codec_id);
            if (!codec) {
                fprintf(stderr, "Unsupported codec for output stream #%d.%d\n", 
                        ost->file_index, ost->index);
                exit(1);
            }
            if (avcodec_open(&ost->st->codec, codec) < 0) {
                fprintf(stderr, "Error while opening codec for stream #%d.%d - maybe incorrect parameters such as bit_rate, rate, width or height\n", 
                        ost->file_index, ost->index);
                exit(1);
            }
        }
    }

    /* open each decoder */
    for(i=0;i<nb_istreams;i++) {
        ist = ist_table[i];
        if (ist->decoding_needed) {
            AVCodec *codec;
            codec = avcodec_find_decoder(ist->st->codec.codec_id);
            if (!codec) {
1086 1087
                fprintf(stderr, "Unsupported codec (id=%d) for input stream #%d.%d\n", 
                        ist->st->codec.codec_id, ist->file_index, ist->index);
F
merge  
Fabrice Bellard 已提交
1088 1089 1090 1091 1092 1093 1094
                exit(1);
            }
            if (avcodec_open(&ist->st->codec, codec) < 0) {
                fprintf(stderr, "Error while opening codec for input stream #%d.%d\n", 
                        ist->file_index, ist->index);
                exit(1);
            }
1095 1096
            //if (ist->st->codec.codec_type == CODEC_TYPE_VIDEO)
            //    ist->st->codec.flags |= CODEC_FLAG_REPEAT_FIELD;
1097
            ist->frame_decoded = 1;
F
merge  
Fabrice Bellard 已提交
1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113
        }
    }

    /* init pts */
    for(i=0;i<nb_istreams;i++) {
        ist = ist_table[i];
    }
    
    /* compute buffer size max (should use a complete heuristic) */
    for(i=0;i<nb_input_files;i++) {
        file_table[i].buffer_size_max = 2048;
    }

    /* open files and write file headers */
    for(i=0;i<nb_output_files;i++) {
        os = output_files[i];
1114
        if (av_write_header(os) < 0) {
1115 1116 1117 1118
            fprintf(stderr, "Could not write header for output file #%d (incorrect codec paramters ?)\n", i);
            ret = -EINVAL;
            goto fail;
        }
F
merge  
Fabrice Bellard 已提交
1119 1120
    }

1121
#ifndef CONFIG_WIN32
F
Fabrice Bellard 已提交
1122
    fprintf(stderr, "Press [q] to stop encoding\n");
1123 1124 1125
#endif
    term_init();

1126
    stream_no_data = 0;
1127
    key = -1;
1128

F
merge  
Fabrice Bellard 已提交
1129 1130 1131
    for(;;) {
        int file_index, ist_index;
        AVPacket pkt;
1132
        uint8_t *ptr;
F
merge  
Fabrice Bellard 已提交
1133
        int len;
1134
        uint8_t *data_buf;
F
merge  
Fabrice Bellard 已提交
1135 1136 1137
        int data_size, got_picture;
        AVPicture picture;
        short samples[AVCODEC_MAX_AUDIO_FRAME_SIZE / 2];
1138
        void *buffer_to_free;
1139 1140
        double pts_min;
        
F
merge  
Fabrice Bellard 已提交
1141
    redo:
1142
        /* if 'q' pressed, exits */
1143 1144 1145 1146 1147 1148
        if (key) {
            /* read_key() returns 0 on EOF */
            key = read_key();
            if (key == 'q')
                break;
        }
1149

1150 1151
        /* select the stream that we must read now by looking at the
           smallest output pts */
F
merge  
Fabrice Bellard 已提交
1152
        file_index = -1;
1153 1154 1155 1156 1157 1158 1159 1160 1161 1162
        pts_min = 1e10;
        for(i=0;i<nb_ostreams;i++) {
            double pts;
            ost = ost_table[i];
            os = output_files[ost->file_index];
            ist = ist_table[ost->source_index];
            pts = (double)ost->st->pts.val * os->pts_num / os->pts_den;
            if (!file_table[ist->file_index].eof_reached && 
                pts < pts_min) {
                pts_min = pts;
F
merge  
Fabrice Bellard 已提交
1163 1164 1165 1166
                file_index = ist->file_index;
            }
        }
        /* if none, if is finished */
1167
        if (file_index < 0) {
F
merge  
Fabrice Bellard 已提交
1168
            break;
1169 1170
        }

F
merge  
Fabrice Bellard 已提交
1171
        /* finish if recording time exhausted */
1172
        if (recording_time > 0 && pts_min >= (recording_time / 1000000.0))
F
merge  
Fabrice Bellard 已提交
1173
            break;
1174

F
merge  
Fabrice Bellard 已提交
1175 1176 1177 1178 1179 1180
        /* read a packet from it and output it in the fifo */
        is = input_files[file_index];
        if (av_read_packet(is, &pkt) < 0) {
            file_table[file_index].eof_reached = 1;
            continue;
        }
1181 1182 1183 1184 1185
        if (!pkt.size) {
            stream_no_data = is;
        } else {
            stream_no_data = 0;
        }
1186 1187 1188 1189
        if (do_hex_dump) {
            printf("stream #%d, size=%d:\n", pkt.stream_index, pkt.size);
            av_hex_dump(pkt.data, pkt.size);
        }
1190 1191 1192
        /* the following test is needed in case new streams appear
           dynamically in stream : we ignore them */
        if (pkt.stream_index >= file_table[file_index].nb_streams)
1193
            goto discard_packet;
F
merge  
Fabrice Bellard 已提交
1194 1195
        ist_index = file_table[file_index].ist_index + pkt.stream_index;
        ist = ist_table[ist_index];
1196 1197
        if (ist->discard)
            goto discard_packet;
F
merge  
Fabrice Bellard 已提交
1198

1199
        // printf("read #%d.%d size=%d\n", ist->file_index, ist->index, pkt.size);
F
merge  
Fabrice Bellard 已提交
1200 1201 1202

        len = pkt.size;
        ptr = pkt.data;
1203
        pts_set = 0;
F
merge  
Fabrice Bellard 已提交
1204
        while (len > 0) {
1205
            int64_t ipts;
1206 1207

            ipts = AV_NOPTS_VALUE;
F
merge  
Fabrice Bellard 已提交
1208 1209 1210 1211 1212

            /* decode the packet if needed */
            data_buf = NULL; /* fail safe */
            data_size = 0;
            if (ist->decoding_needed) {
1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224
                /* NOTE1: we only take into account the PTS if a new
                   frame has begun (MPEG semantics) */
                /* NOTE2: even if the fraction is not initialized,
                   av_frac_set can be used to set the integer part */
                if (ist->frame_decoded && 
                    pkt.pts != AV_NOPTS_VALUE && 
                    !pts_set) {
                    ipts = pkt.pts;
                    ist->frame_decoded = 0;
                    pts_set = 1;
                }

F
merge  
Fabrice Bellard 已提交
1225 1226
                switch(ist->st->codec.codec_type) {
                case CODEC_TYPE_AUDIO:
1227 1228 1229 1230 1231 1232
                    /* XXX: could avoid copy if PCM 16 bits with same
                       endianness as CPU */
                    ret = avcodec_decode_audio(&ist->st->codec, samples, &data_size,
                                               ptr, len);
                    if (ret < 0)
                        goto fail_decode;
1233 1234 1235
                    /* Some bug in mpeg audio decoder gives */
                    /* data_size < 0, it seems they are overflows */
                    if (data_size <= 0) {
1236 1237 1238 1239
                        /* no audio frame */
                        ptr += ret;
                        len -= ret;
                        continue;
F
merge  
Fabrice Bellard 已提交
1240
                    }
1241
                    data_buf = (uint8_t *)samples;
F
merge  
Fabrice Bellard 已提交
1242 1243
                    break;
                case CODEC_TYPE_VIDEO:
1244
                    {
M
Michael Niedermayer 已提交
1245
                        AVFrame big_picture;
M
cleanup  
Michael Niedermayer 已提交
1246

F
merge  
Fabrice Bellard 已提交
1247 1248
                        data_size = (ist->st->codec.width * ist->st->codec.height * 3) / 2;
                        ret = avcodec_decode_video(&ist->st->codec, 
M
cleanup  
Michael Niedermayer 已提交
1249 1250 1251
                                                   &big_picture, &got_picture, ptr, len);
                        picture= *(AVPicture*)&big_picture;
                        ist->st->quality= big_picture.quality;
F
merge  
Fabrice Bellard 已提交
1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264
                        if (ret < 0) {
                        fail_decode:
                            fprintf(stderr, "Error while decoding stream #%d.%d\n",
                                    ist->file_index, ist->index);
                            av_free_packet(&pkt);
                            goto redo;
                        }
                        if (!got_picture) {
                            /* no picture yet */
                            ptr += ret;
                            len -= ret;
                            continue;
                        }
1265
                                  
F
merge  
Fabrice Bellard 已提交
1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278
                    }
                    break;
                default:
                    goto fail_decode;
                }
            } else {
                data_buf = ptr;
                data_size = len;
                ret = len;
            }
            ptr += ret;
            len -= ret;

1279 1280 1281 1282 1283
            buffer_to_free = 0;
            if (ist->st->codec.codec_type == CODEC_TYPE_VIDEO) {
                pre_process_video_frame(ist, &picture, &buffer_to_free);
            }

1284 1285
            ist->frame_decoded = 1;

1286 1287
            /* frame rate emulation */
            if (ist->st->codec.rate_emu) {
1288
                int64_t pts = av_rescale((int64_t) ist->frame * ist->st->codec.frame_rate_base, 1000000, ist->st->codec.frame_rate);
1289 1290 1291 1292 1293 1294 1295
                int64_t now = av_gettime() - ist->start;
                if (pts > now)
                    usleep(pts - now);

                ist->frame++;
            }

1296 1297 1298 1299 1300 1301
#if 0
            /* mpeg PTS deordering : if it is a P or I frame, the PTS
               is the one of the next displayed one */
            /* XXX: add mpeg4 too ? */
            if (ist->st->codec.codec_id == CODEC_ID_MPEG1VIDEO) {
                if (ist->st->codec.pict_type != B_TYPE) {
1302
                    int64_t tmp;
1303 1304 1305 1306 1307 1308
                    tmp = ist->last_ip_pts;
                    ist->last_ip_pts  = ist->frac_pts.val;
                    ist->frac_pts.val = tmp;
                }
            }
#endif
F
merge  
Fabrice Bellard 已提交
1309
            /* transcode raw format, encode packets and output them */
1310

F
merge  
Fabrice Bellard 已提交
1311
            for(i=0;i<nb_ostreams;i++) {
1312
                int frame_size;
1313

F
merge  
Fabrice Bellard 已提交
1314 1315 1316 1317
                ost = ost_table[i];
                if (ost->source_index == ist_index) {
                    os = output_files[ost->file_index];

1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329
                    if (ipts != AV_NOPTS_VALUE) {
#if 0
                        printf("%d: got pts=%f %f\n", 
                               i, pkt.pts / 90000.0, 
                               (ipts - ost->st->pts.val) / 90000.0);
#endif
                        /* set the input output pts pairs */
                        ost->sync_ipts = (double)ipts * is->pts_num / 
                            is->pts_den;
                        /* XXX: take into account the various fifos,
                           in particular for audio */
                        ost->sync_opts = ost->st->pts.val;
1330 1331 1332
                        //printf("ipts=%lld sync_ipts=%f sync_opts=%lld pts.val=%lld pkt.pts=%lld\n", ipts, ost->sync_ipts, ost->sync_opts, ost->st->pts.val, pkt.pts); 
                    } else {
                        //printf("pts.val=%lld\n", ost->st->pts.val); 
1333
                        ost->sync_ipts = AV_NOPTS_VALUE;
1334 1335
                    }

F
merge  
Fabrice Bellard 已提交
1336 1337 1338 1339 1340 1341
                    if (ost->encoding_needed) {
                        switch(ost->st->codec.codec_type) {
                        case CODEC_TYPE_AUDIO:
                            do_audio_out(os, ost, ist, data_buf, data_size);
                            break;
                        case CODEC_TYPE_VIDEO:
1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356
                            /* find an audio stream for synchro */
                            {
                                int i;
                                AVOutputStream *audio_sync, *ost1;
                                audio_sync = NULL;
                                for(i=0;i<nb_ostreams;i++) {
                                    ost1 = ost_table[i];
                                    if (ost1->file_index == ost->file_index &&
                                        ost1->st->codec.codec_type == CODEC_TYPE_AUDIO) {
                                        audio_sync = ost1;
                                        break;
                                    }
                                }

                                do_video_out(os, ost, ist, &picture, &frame_size, audio_sync);
1357
                                if (do_vstats && frame_size)
1358 1359
                                    do_video_stats(os, ost, frame_size);
                            }
F
merge  
Fabrice Bellard 已提交
1360
                            break;
1361
                        default:
1362
                            av_abort();
F
merge  
Fabrice Bellard 已提交
1363 1364
                        }
                    } else {
1365 1366
                        AVFrame avframe;
                                                
F
merge  
Fabrice Bellard 已提交
1367
                        /* no reencoding needed : output the packet directly */
1368
                        /* force the input stream PTS */
1369 1370 1371
                        
                        memset(&avframe, 0, sizeof(AVFrame));
                        ost->st->codec.coded_frame= &avframe;
1372
			avframe.key_frame = pkt.flags & PKT_FLAG_KEY; 
1373
                        
1374
                        av_write_frame(os, ost->index, data_buf, data_size);
1375
			ost->st->codec.frame_number++;
1376
			ost->frame_number++;
F
merge  
Fabrice Bellard 已提交
1377 1378 1379
                    }
                }
            }
1380
            av_free(buffer_to_free);
1381
            ipts = AV_NOPTS_VALUE;
F
merge  
Fabrice Bellard 已提交
1382
        }
1383
    discard_packet:
F
merge  
Fabrice Bellard 已提交
1384 1385
        av_free_packet(&pkt);
        
1386 1387
        /* dump report by using the output first video and audio streams */
        print_report(output_files, ost_table, nb_ostreams, 0);
F
merge  
Fabrice Bellard 已提交
1388
    }
1389
    term_exit();
F
merge  
Fabrice Bellard 已提交
1390 1391

    /* dump report by using the first video and audio streams */
1392 1393
    print_report(output_files, ost_table, nb_ostreams, 1);

F
merge  
Fabrice Bellard 已提交
1394 1395 1396 1397
    /* close each encoder */
    for(i=0;i<nb_ostreams;i++) {
        ost = ost_table[i];
        if (ost->encoding_needed) {
1398
            av_freep(&ost->st->codec.stats_in);
F
merge  
Fabrice Bellard 已提交
1399 1400 1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414
            avcodec_close(&ost->st->codec);
        }
    }
    
    /* close each decoder */
    for(i=0;i<nb_istreams;i++) {
        ist = ist_table[i];
        if (ist->decoding_needed) {
            avcodec_close(&ist->st->codec);
        }
    }
    

    /* write the trailer if needed and close file */
    for(i=0;i<nb_output_files;i++) {
        os = output_files[i];
1415
        av_write_trailer(os);
F
merge  
Fabrice Bellard 已提交
1416 1417 1418 1419 1420
    }
    /* finished ! */
    
    ret = 0;
 fail1:
F
Fabrice Bellard 已提交
1421
    av_free(file_table);
F
Fabrice Bellard 已提交
1422

F
merge  
Fabrice Bellard 已提交
1423 1424 1425
    if (ist_table) {
        for(i=0;i<nb_istreams;i++) {
            ist = ist_table[i];
F
Fabrice Bellard 已提交
1426
            av_free(ist);
F
merge  
Fabrice Bellard 已提交
1427
        }
F
Fabrice Bellard 已提交
1428
        av_free(ist_table);
F
merge  
Fabrice Bellard 已提交
1429 1430 1431 1432 1433
    }
    if (ost_table) {
        for(i=0;i<nb_ostreams;i++) {
            ost = ost_table[i];
            if (ost) {
1434 1435 1436 1437
                if (ost->logfile) {
                    fclose(ost->logfile);
                    ost->logfile = NULL;
                }
1438 1439
                fifo_free(&ost->fifo); /* works even if fifo is not
                                          initialized but set to zero */
F
Fabrice Bellard 已提交
1440
                av_free(ost->pict_tmp.data[0]);
F
merge  
Fabrice Bellard 已提交
1441 1442 1443 1444
                if (ost->video_resample)
                    img_resample_close(ost->img_resample_ctx);
                if (ost->audio_resample)
                    audio_resample_close(ost->resample);
F
Fabrice Bellard 已提交
1445
                av_free(ost);
F
merge  
Fabrice Bellard 已提交
1446 1447
            }
        }
F
Fabrice Bellard 已提交
1448
        av_free(ost_table);
F
merge  
Fabrice Bellard 已提交
1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477
    }
    return ret;
 fail:
    ret = -ENOMEM;
    goto fail1;
}

#if 0
int file_read(const char *filename)
{
    URLContext *h;
    unsigned char buffer[1024];
    int len, i;

    if (url_open(&h, filename, O_RDONLY) < 0) {
        printf("could not open '%s'\n", filename);
        return -1;
    }
    for(;;) {
        len = url_read(h, buffer, sizeof(buffer));
        if (len <= 0)
            break;
        for(i=0;i<len;i++) putchar(buffer[i]);
    }
    url_close(h);
    return 0;
}
#endif

1478
static void show_licence(void)
F
merge  
Fabrice Bellard 已提交
1479 1480 1481
{
    printf(
    "ffmpeg version " FFMPEG_VERSION "\n"
1482 1483 1484 1485 1486
    "Copyright (c) 2000, 2001, 2002 Fabrice Bellard\n"
    "This library is free software; you can redistribute it and/or\n"
    "modify it under the terms of the GNU Lesser General Public\n"
    "License as published by the Free Software Foundation; either\n"
    "version 2 of the License, or (at your option) any later version.\n"
F
merge  
Fabrice Bellard 已提交
1487
    "\n"
1488
    "This library is distributed in the hope that it will be useful,\n"
F
merge  
Fabrice Bellard 已提交
1489
    "but WITHOUT ANY WARRANTY; without even the implied warranty of\n"
1490 1491
    "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU\n"
    "Lesser General Public License for more details.\n"
F
merge  
Fabrice Bellard 已提交
1492
    "\n"
1493 1494 1495
    "You should have received a copy of the GNU Lesser General Public\n"
    "License along with this library; if not, write to the Free Software\n"
    "Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA\n"
F
merge  
Fabrice Bellard 已提交
1496 1497 1498 1499
    );
    exit(1);
}

1500
static void opt_image_format(const char *arg)
1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514
{
    AVImageFormat *f;
    
    for(f = first_image_format; f != NULL; f = f->next) {
        if (!strcmp(arg, f->name))
            break;
    }
    if (!f) {
        fprintf(stderr, "Unknown image format: '%s'\n", arg);
        exit(1);
    }
    image_format = f;
}

1515
static void opt_format(const char *arg)
F
merge  
Fabrice Bellard 已提交
1516
{
1517 1518 1519 1520 1521 1522
    /* compatibility stuff for pgmyuv */
    if (!strcmp(arg, "pgmyuv")) {
        opt_image_format(arg);
        arg = "image";
    }

1523 1524 1525 1526
    file_iformat = av_find_input_format(arg);
    file_oformat = guess_format(arg, NULL, NULL);
    if (!file_iformat && !file_oformat) {
        fprintf(stderr, "Unknown input or output format: %s\n", arg);
F
merge  
Fabrice Bellard 已提交
1527 1528 1529 1530
        exit(1);
    }
}

1531
static void opt_video_bitrate(const char *arg)
F
merge  
Fabrice Bellard 已提交
1532 1533 1534 1535
{
    video_bit_rate = atoi(arg) * 1000;
}

1536
static void opt_video_bitrate_tolerance(const char *arg)
1537 1538 1539 1540
{
    video_bit_rate_tolerance = atoi(arg) * 1000;
}

1541
static void opt_video_bitrate_max(const char *arg)
M
Michael Niedermayer 已提交
1542 1543 1544 1545
{
    video_rc_max_rate = atoi(arg) * 1000;
}

1546
static void opt_video_bitrate_min(const char *arg)
M
Michael Niedermayer 已提交
1547 1548 1549 1550
{
    video_rc_min_rate = atoi(arg) * 1000;
}

1551
static void opt_video_buffer_size(const char *arg)
M
fixing  
Michael Niedermayer 已提交
1552 1553 1554 1555
{
    video_rc_buffer_size = atoi(arg) * 1000;
}

1556
static void opt_video_rc_eq(char *arg)
M
Michael Niedermayer 已提交
1557 1558 1559 1560
{
    video_rc_eq = arg;
}

1561
static void opt_video_rc_override_string(char *arg)
1562 1563 1564 1565
{
    video_rc_override_string = arg;
}

M
Michael Niedermayer 已提交
1566

1567
static void opt_workaround_bugs(const char *arg)
1568 1569 1570 1571
{
    workaround_bugs = atoi(arg);
}

1572
static void opt_dct_algo(const char *arg)
1573 1574 1575 1576
{
    dct_algo = atoi(arg);
}

1577
static void opt_idct_algo(const char *arg)
1578 1579 1580 1581 1582
{
    idct_algo = atoi(arg);
}


1583
static void opt_error_resilience(const char *arg)
1584 1585 1586 1587
{
    error_resilience = atoi(arg);
}

1588
static void opt_error_concealment(const char *arg)
1589 1590 1591 1592
{
    error_concealment = atoi(arg);
}

1593
static void opt_debug(const char *arg)
M
Michael Niedermayer 已提交
1594 1595 1596
{
    debug = atoi(arg);
}
1597

1598
static void opt_frame_rate(const char *arg)
F
merge  
Fabrice Bellard 已提交
1599
{
1600 1601 1602
    frame_rate_base = DEFAULT_FRAME_RATE_BASE; //FIXME not optimal
    frame_rate = (int)(strtod(arg, 0) * frame_rate_base + 0.5);
    //FIXME parse fractions 
F
merge  
Fabrice Bellard 已提交
1603 1604
}

1605

1606
static void opt_frame_crop_top(const char *arg)
1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619 1620 1621 1622 1623
{
    frame_topBand = atoi(arg); 
    if (frame_topBand < 0) {
        fprintf(stderr, "Incorrect top crop size\n");
        exit(1);
    }
    if ((frame_topBand % 2) != 0) {
        fprintf(stderr, "Top crop size must be a multiple of 2\n");
        exit(1);
    }
    if ((frame_topBand) >= frame_height){
    	fprintf(stderr, "Vertical crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");
        exit(1);
    }
    frame_height -= frame_topBand;
}

1624
static void opt_frame_crop_bottom(const char *arg)
1625 1626 1627 1628 1629 1630 1631 1632 1633 1634 1635 1636 1637 1638 1639 1640 1641
{
    frame_bottomBand = atoi(arg);
    if (frame_bottomBand < 0) {
        fprintf(stderr, "Incorrect bottom crop size\n");
        exit(1);
    }
    if ((frame_bottomBand % 2) != 0) {
        fprintf(stderr, "Bottom crop size must be a multiple of 2\n");
        exit(1);        
    }
    if ((frame_bottomBand) >= frame_height){
    	fprintf(stderr, "Vertical crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");
        exit(1);
    }
    frame_height -= frame_bottomBand;
}

1642
static void opt_frame_crop_left(const char *arg)
1643 1644 1645 1646 1647 1648 1649 1650 1651 1652 1653 1654 1655 1656 1657 1658 1659
{
    frame_leftBand = atoi(arg);
    if (frame_leftBand < 0) {
        fprintf(stderr, "Incorrect left crop size\n");
        exit(1);
    }
    if ((frame_leftBand % 2) != 0) {
        fprintf(stderr, "Left crop size must be a multiple of 2\n");
        exit(1);
    }
    if ((frame_leftBand) >= frame_width){
    	fprintf(stderr, "Horizontal crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");
        exit(1);
    }
    frame_width -= frame_leftBand;
}

1660
static void opt_frame_crop_right(const char *arg)
1661 1662 1663 1664 1665 1666 1667 1668 1669 1670 1671 1672 1673 1674 1675 1676 1677
{
    frame_rightBand = atoi(arg);
    if (frame_rightBand < 0) {
        fprintf(stderr, "Incorrect right crop size\n");
        exit(1);
    }
    if ((frame_rightBand % 2) != 0) {
        fprintf(stderr, "Right crop size must be a multiple of 2\n");
        exit(1);        
    }
    if ((frame_rightBand) >= frame_width){
    	fprintf(stderr, "Horizontal crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");
        exit(1);
    }
    frame_width -= frame_rightBand;
}

1678
static void opt_frame_size(const char *arg)
F
merge  
Fabrice Bellard 已提交
1679 1680 1681 1682 1683 1684 1685 1686 1687 1688 1689 1690
{
    parse_image_size(&frame_width, &frame_height, arg);
    if (frame_width <= 0 || frame_height <= 0) {
        fprintf(stderr, "Incorrect frame size\n");
        exit(1);
    }
    if ((frame_width % 2) != 0 || (frame_height % 2) != 0) {
        fprintf(stderr, "Frame size must be a multiple of 2\n");
        exit(1);
    }
}

1691 1692 1693 1694 1695
static void opt_frame_pix_fmt(const char *arg)
{
    frame_pix_fmt = avcodec_get_pix_fmt(arg);
}

1696 1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707 1708 1709 1710 1711 1712 1713 1714 1715 1716 1717 1718
static void opt_frame_aspect_ratio(const char *arg)
{
    int x = 0, y = 0;
    double ar = 0;
    const char *p;
    
    p = strchr(arg, ':');
    if (p) {
        x = strtol(arg, (char **)&arg, 10);
	if (arg == p)
	    y = strtol(arg+1, (char **)&arg, 10);
	if (x > 0 && y > 0)
	    ar = (double)x / (double)y;
    } else
        ar = strtod(arg, (char **)&arg);

    if (!ar) {
        fprintf(stderr, "Incorrect aspect ratio specification.\n");
	exit(1);
    }
    frame_aspect_ratio = ar;
}

1719
static void opt_gop_size(const char *arg)
F
merge  
Fabrice Bellard 已提交
1720 1721 1722 1723
{
    gop_size = atoi(arg);
}

1724
static void opt_b_frames(const char *arg)
1725 1726 1727 1728 1729 1730 1731 1732 1733 1734 1735
{
    b_frames = atoi(arg);
    if (b_frames > FF_MAX_B_FRAMES) {
        fprintf(stderr, "\nCannot have more than %d B frames, increase FF_MAX_B_FRAMES.\n", FF_MAX_B_FRAMES);
        exit(1);
    } else if (b_frames < 1) {
        fprintf(stderr, "\nNumber of B frames must be higher than 0\n");
        exit(1);
    }
}

1736
static void opt_qscale(const char *arg)
F
merge  
Fabrice Bellard 已提交
1737 1738 1739 1740 1741 1742 1743 1744 1745
{
    video_qscale = atoi(arg);
    if (video_qscale < 0 ||
        video_qscale > 31) {
        fprintf(stderr, "qscale must be >= 1 and <= 31\n");
        exit(1);
    }
}

1746
static void opt_qmin(const char *arg)
1747 1748 1749 1750 1751 1752 1753 1754 1755
{
    video_qmin = atoi(arg);
    if (video_qmin < 0 ||
        video_qmin > 31) {
        fprintf(stderr, "qmin must be >= 1 and <= 31\n");
        exit(1);
    }
}

1756
static void opt_qmax(const char *arg)
1757 1758 1759 1760 1761 1762 1763 1764 1765
{
    video_qmax = atoi(arg);
    if (video_qmax < 0 ||
        video_qmax > 31) {
        fprintf(stderr, "qmax must be >= 1 and <= 31\n");
        exit(1);
    }
}

1766
static void opt_mb_qmin(const char *arg)
M
Michael Niedermayer 已提交
1767 1768 1769 1770 1771 1772 1773 1774 1775
{
    video_mb_qmin = atoi(arg);
    if (video_mb_qmin < 0 ||
        video_mb_qmin > 31) {
        fprintf(stderr, "qmin must be >= 1 and <= 31\n");
        exit(1);
    }
}

1776
static void opt_mb_qmax(const char *arg)
M
Michael Niedermayer 已提交
1777 1778 1779 1780 1781 1782 1783 1784 1785
{
    video_mb_qmax = atoi(arg);
    if (video_mb_qmax < 0 ||
        video_mb_qmax > 31) {
        fprintf(stderr, "qmax must be >= 1 and <= 31\n");
        exit(1);
    }
}

1786
static void opt_qdiff(const char *arg)
1787 1788 1789 1790 1791 1792 1793 1794 1795
{
    video_qdiff = atoi(arg);
    if (video_qdiff < 0 ||
        video_qdiff > 31) {
        fprintf(stderr, "qdiff must be >= 1 and <= 31\n");
        exit(1);
    }
}

1796
static void opt_qblur(const char *arg)
1797 1798 1799 1800
{
    video_qblur = atof(arg);
}

1801
static void opt_qcomp(const char *arg)
1802 1803 1804
{
    video_qcomp = atof(arg);
}
F
merge  
Fabrice Bellard 已提交
1805

1806
static void opt_rc_initial_cplx(const char *arg)
1807 1808 1809
{
    video_rc_initial_cplx = atof(arg);
}
1810
static void opt_b_qfactor(const char *arg)
1811 1812 1813
{
    video_b_qfactor = atof(arg);
}
1814
static void opt_i_qfactor(const char *arg)
1815 1816 1817
{
    video_i_qfactor = atof(arg);
}
1818
static void opt_b_qoffset(const char *arg)
1819 1820 1821
{
    video_b_qoffset = atof(arg);
}
1822
static void opt_i_qoffset(const char *arg)
1823 1824 1825 1826
{
    video_i_qoffset = atof(arg);
}

1827
static void opt_packet_size(const char *arg)
1828 1829 1830 1831
{
    packet_size= atoi(arg);
}

1832
static void opt_strict(const char *arg)
M
Michael Niedermayer 已提交
1833 1834 1835 1836
{
    strict= atoi(arg);
}

1837
static void opt_audio_bitrate(const char *arg)
F
merge  
Fabrice Bellard 已提交
1838 1839 1840 1841
{
    audio_bit_rate = atoi(arg) * 1000;
}

1842
static void opt_audio_rate(const char *arg)
F
merge  
Fabrice Bellard 已提交
1843 1844 1845 1846
{
    audio_sample_rate = atoi(arg);
}

1847
static void opt_audio_channels(const char *arg)
F
merge  
Fabrice Bellard 已提交
1848 1849 1850 1851
{
    audio_channels = atoi(arg);
}

1852
static void opt_video_device(const char *arg)
F
merge  
Fabrice Bellard 已提交
1853
{
F
Fabrice Bellard 已提交
1854
    video_device = av_strdup(arg);
F
merge  
Fabrice Bellard 已提交
1855 1856
}

1857
static void opt_video_channel(const char *arg)
1858 1859 1860 1861
{
    video_channel = strtol(arg, NULL, 0);
}

1862 1863 1864 1865 1866
static void opt_video_standard(const char *arg)
{
    video_standard = av_strdup(arg);
}

1867
static void opt_audio_device(const char *arg)
F
merge  
Fabrice Bellard 已提交
1868
{
F
Fabrice Bellard 已提交
1869
    audio_device = av_strdup(arg);
F
merge  
Fabrice Bellard 已提交
1870 1871
}

1872
static void opt_dv1394(const char *arg)
1873 1874
{
    video_grab_format = "dv1394";
1875
    audio_grab_format = NULL;
1876 1877
}

1878
static void opt_audio_codec(const char *arg)
F
merge  
Fabrice Bellard 已提交
1879 1880 1881
{
    AVCodec *p;

1882 1883
    if (!strcmp(arg, "copy")) {
        audio_stream_copy = 1;
F
merge  
Fabrice Bellard 已提交
1884
    } else {
1885 1886 1887 1888 1889 1890 1891 1892 1893 1894 1895 1896
        p = first_avcodec;
        while (p) {
            if (!strcmp(p->name, arg) && p->type == CODEC_TYPE_AUDIO)
                break;
            p = p->next;
        }
        if (p == NULL) {
            fprintf(stderr, "Unknown audio codec '%s'\n", arg);
            exit(1);
        } else {
            audio_codec_id = p->id;
        }
F
merge  
Fabrice Bellard 已提交
1897 1898 1899
    }
}

1900
static void add_frame_hooker(const char *arg)
1901 1902 1903 1904
{
    int argc = 0;
    char *argv[64];
    int i;
F
Fabrice Bellard 已提交
1905
    char *args = av_strdup(arg);
1906 1907 1908 1909 1910 1911 1912 1913 1914 1915 1916 1917 1918

    argv[0] = strtok(args, " ");
    while (argc < 62 && (argv[++argc] = strtok(NULL, " "))) {
    }

    i = frame_hook_add(argc, argv);

    if (i != 0) {
        fprintf(stderr, "Failed to add video hook function: %s\n", arg);
        exit(1);
    }
}

F
merge  
Fabrice Bellard 已提交
1919 1920 1921 1922 1923
const char *motion_str[] = {
    "zero",
    "full",
    "log",
    "phods",
M
Michael Niedermayer 已提交
1924 1925
    "epzs",
    "x1",
F
merge  
Fabrice Bellard 已提交
1926 1927 1928
    NULL,
};

1929
static void opt_motion_estimation(const char *arg)
F
merge  
Fabrice Bellard 已提交
1930 1931 1932 1933 1934 1935 1936 1937 1938 1939 1940 1941
{
    const char **p;
    p = motion_str;
    for(;;) {
        if (!*p) {
            fprintf(stderr, "Unknown motion estimation method '%s'\n", arg);
            exit(1);
        }
        if (!strcmp(*p, arg))
            break;
        p++;
    }
1942
    me_method = (p - motion_str) + 1;
F
merge  
Fabrice Bellard 已提交
1943 1944
}

1945
static void opt_video_codec(const char *arg)
F
merge  
Fabrice Bellard 已提交
1946 1947 1948
{
    AVCodec *p;

1949 1950
    if (!strcmp(arg, "copy")) {
        video_stream_copy = 1;
F
merge  
Fabrice Bellard 已提交
1951
    } else {
1952 1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963
        p = first_avcodec;
        while (p) {
            if (!strcmp(p->name, arg) && p->type == CODEC_TYPE_VIDEO)
                break;
            p = p->next;
        }
        if (p == NULL) {
            fprintf(stderr, "Unknown video codec '%s'\n", arg);
            exit(1);
        } else {
            video_codec_id = p->id;
        }
F
merge  
Fabrice Bellard 已提交
1964 1965 1966
    }
}

1967
static void opt_map(const char *arg)
F
merge  
Fabrice Bellard 已提交
1968 1969 1970 1971 1972 1973 1974 1975 1976 1977
{
    AVStreamMap *m;
    const char *p;

    p = arg;
    m = &stream_maps[nb_stream_maps++];

    m->file_index = strtol(arg, (char **)&p, 0);
    if (*p)
        p++;
J
Juanjo 已提交
1978 1979

    m->stream_index = strtol(p, (char **)&p, 0);
F
merge  
Fabrice Bellard 已提交
1980 1981
}

1982
static void opt_recording_time(const char *arg)
F
merge  
Fabrice Bellard 已提交
1983 1984 1985 1986
{
    recording_time = parse_date(arg, 1);
}

1987
static void opt_input_file(const char *filename)
F
merge  
Fabrice Bellard 已提交
1988 1989 1990
{
    AVFormatContext *ic;
    AVFormatParameters params, *ap = &params;
1991
    int err, i, ret, rfps, rfps_base;
F
merge  
Fabrice Bellard 已提交
1992

1993 1994 1995
    if (!strcmp(filename, "-"))
        filename = "pipe:";

F
merge  
Fabrice Bellard 已提交
1996
    /* get default parameters from command line */
1997 1998 1999 2000
    memset(ap, 0, sizeof(*ap));
    ap->sample_rate = audio_sample_rate;
    ap->channels = audio_channels;
    ap->frame_rate = frame_rate;
2001
    ap->frame_rate_base = frame_rate_base;
2002 2003
    ap->width = frame_width;
    ap->height = frame_height;
2004
    ap->image_format = image_format;
2005
    ap->pix_fmt = frame_pix_fmt;
2006 2007 2008

    /* open the input file with generic libav function */
    err = av_open_input_file(&ic, filename, file_iformat, 0, ap);
F
merge  
Fabrice Bellard 已提交
2009
    if (err < 0) {
2010
        print_error(filename, err);
F
merge  
Fabrice Bellard 已提交
2011 2012 2013
        exit(1);
    }
    
2014 2015 2016
    /* If not enough info to get the stream parameters, we decode the
       first frames to get it. (used in mpeg case for example) */
    ret = av_find_stream_info(ic);
F
merge  
Fabrice Bellard 已提交
2017 2018 2019 2020 2021 2022 2023 2024 2025 2026
    if (ret < 0) {
        fprintf(stderr, "%s: could not find codec parameters\n", filename);
        exit(1);
    }

    /* update the current parameters so that they match the one of the input stream */
    for(i=0;i<ic->nb_streams;i++) {
        AVCodecContext *enc = &ic->streams[i]->codec;
        switch(enc->codec_type) {
        case CODEC_TYPE_AUDIO:
2027
            //fprintf(stderr, "\nInput Audio channels: %d", enc->channels);
F
merge  
Fabrice Bellard 已提交
2028 2029 2030 2031 2032 2033
            audio_channels = enc->channels;
            audio_sample_rate = enc->sample_rate;
            break;
        case CODEC_TYPE_VIDEO:
            frame_height = enc->height;
            frame_width = enc->width;
2034
	    frame_aspect_ratio = enc->aspect_ratio;
2035
	    frame_pix_fmt = enc->pix_fmt;
2036 2037
            rfps      = ic->streams[i]->r_frame_rate;
            rfps_base = ic->streams[i]->r_frame_rate_base;
2038
            enc->workaround_bugs = workaround_bugs;
2039
            enc->error_resilience = error_resilience; 
2040
            enc->error_concealment = error_concealment; 
2041
            enc->idct_algo= idct_algo;
M
Michael Niedermayer 已提交
2042
            enc->debug= debug;
2043 2044
/*            if(enc->codec->capabilities & CODEC_CAP_TRUNCATED)
                enc->flags|= CODEC_FLAG_TRUNCATED; */
M
Michael Niedermayer 已提交
2045
            if(/*enc->codec_id==CODEC_ID_MPEG4 || */enc->codec_id==CODEC_ID_MPEG1VIDEO || enc->codec_id==CODEC_ID_H264)
2046
                enc->flags|= CODEC_FLAG_TRUNCATED;
2047 2048 2049
            
            if(bitexact)
                enc->flags|= CODEC_FLAG_BITEXACT;
2050

2051 2052
            assert(enc->frame_rate_base == rfps_base); // should be true for now
            if (enc->frame_rate != rfps) { 
2053
                fprintf(stderr,"\nSeems that stream %d comes from film source: %2.2f->%2.2f\n",
2054 2055
                    i, (float)enc->frame_rate / enc->frame_rate_base,
                    (float)rfps / rfps_base);
2056
            }
2057
            /* update the current frame rate to match the stream frame rate */
2058 2059
            frame_rate      = rfps;
            frame_rate_base = rfps_base;
2060 2061

            enc->rate_emu = rate_emu;
F
merge  
Fabrice Bellard 已提交
2062
            break;
2063
        default:
2064
            av_abort();
F
merge  
Fabrice Bellard 已提交
2065 2066 2067 2068 2069 2070 2071
        }
    }
    
    input_files[nb_input_files] = ic;
    /* dump the file content */
    dump_format(ic, nb_input_files, filename, 0);
    nb_input_files++;
2072 2073
    file_iformat = NULL;
    file_oformat = NULL;
2074
    image_format = NULL;
2075 2076

    rate_emu = 0;
F
merge  
Fabrice Bellard 已提交
2077 2078
}

2079
static void check_audio_video_inputs(int *has_video_ptr, int *has_audio_ptr)
2080 2081 2082 2083 2084 2085 2086 2087 2088 2089 2090 2091 2092 2093 2094 2095 2096
{
    int has_video, has_audio, i, j;
    AVFormatContext *ic;

    has_video = 0;
    has_audio = 0;
    for(j=0;j<nb_input_files;j++) {
        ic = input_files[j];
        for(i=0;i<ic->nb_streams;i++) {
            AVCodecContext *enc = &ic->streams[i]->codec;
            switch(enc->codec_type) {
            case CODEC_TYPE_AUDIO:
                has_audio = 1;
                break;
            case CODEC_TYPE_VIDEO:
                has_video = 1;
                break;
2097
            default:
2098
                av_abort();
2099 2100 2101 2102 2103 2104 2105
            }
        }
    }
    *has_video_ptr = has_video;
    *has_audio_ptr = has_audio;
}

2106
static void opt_output_file(const char *filename)
F
merge  
Fabrice Bellard 已提交
2107 2108 2109
{
    AVStream *st;
    AVFormatContext *oc;
2110
    int use_video, use_audio, nb_streams, input_has_video, input_has_audio;
F
merge  
Fabrice Bellard 已提交
2111
    int codec_id;
2112
    AVFormatParameters params, *ap = &params;
F
merge  
Fabrice Bellard 已提交
2113 2114 2115 2116 2117 2118

    if (!strcmp(filename, "-"))
        filename = "pipe:";

    oc = av_mallocz(sizeof(AVFormatContext));

2119 2120 2121 2122 2123 2124 2125
    if (!file_oformat) {
        file_oformat = guess_format(NULL, filename, NULL);
        if (!file_oformat) {
            fprintf(stderr, "Unable for find a suitable output format for '%s'\n",
                    filename);
            exit(1);
        }
F
merge  
Fabrice Bellard 已提交
2126 2127
    }
    
2128
    oc->oformat = file_oformat;
F
merge  
Fabrice Bellard 已提交
2129

2130
    if (!strcmp(file_oformat->name, "ffm") && 
F
merge  
Fabrice Bellard 已提交
2131 2132 2133 2134 2135 2136 2137 2138
        strstart(filename, "http:", NULL)) {
        /* special case for files sent to ffserver: we get the stream
           parameters from ffserver */
        if (read_ffserver_streams(oc, filename) < 0) {
            fprintf(stderr, "Could not read stream parameters from '%s'\n", filename);
            exit(1);
        }
    } else {
2139 2140
        use_video = file_oformat->video_codec != CODEC_ID_NONE;
        use_audio = file_oformat->audio_codec != CODEC_ID_NONE;
2141

2142 2143 2144 2145 2146 2147 2148 2149 2150
        /* disable if no corresponding type found and at least one
           input file */
        if (nb_input_files > 0) {
            check_audio_video_inputs(&input_has_video, &input_has_audio);
            if (!input_has_video)
                use_video = 0;
            if (!input_has_audio)
                use_audio = 0;
        }
2151 2152

        /* manual disable */
F
merge  
Fabrice Bellard 已提交
2153 2154 2155 2156 2157 2158 2159 2160 2161 2162 2163 2164 2165 2166 2167 2168
        if (audio_disable) {
            use_audio = 0;
        }
        if (video_disable) {
            use_video = 0;
        }
        
        nb_streams = 0;
        if (use_video) {
            AVCodecContext *video_enc;
            
            st = av_mallocz(sizeof(AVStream));
            if (!st) {
                fprintf(stderr, "Could not alloc stream\n");
                exit(1);
            }
M
cleanup  
Michael Niedermayer 已提交
2169
            avcodec_get_context_defaults(&st->codec);
F
merge  
Fabrice Bellard 已提交
2170

2171 2172 2173 2174 2175
            video_enc = &st->codec;
            if (video_stream_copy) {
                st->stream_copy = 1;
                video_enc->codec_type = CODEC_TYPE_VIDEO;
            } else {
2176 2177 2178
                char *p;
                int i;
            
2179 2180 2181 2182 2183 2184 2185 2186 2187
                codec_id = file_oformat->video_codec;
                if (video_codec_id != CODEC_ID_NONE)
                    codec_id = video_codec_id;
                
                video_enc->codec_id = codec_id;
                
                video_enc->bit_rate = video_bit_rate;
                video_enc->bit_rate_tolerance = video_bit_rate_tolerance;
                video_enc->frame_rate = frame_rate; 
2188
                video_enc->frame_rate_base = frame_rate_base; 
2189 2190 2191
                
                video_enc->width = frame_width;
                video_enc->height = frame_height;
2192
		video_enc->aspect_ratio = frame_aspect_ratio;
2193
		video_enc->pix_fmt = frame_pix_fmt;
2194 2195 2196 2197 2198 2199 2200

                if (!intra_only)
                    video_enc->gop_size = gop_size;
                else
                    video_enc->gop_size = 0;
                if (video_qscale || same_quality) {
                    video_enc->flags |= CODEC_FLAG_QSCALE;
M
cleanup  
Michael Niedermayer 已提交
2201
                    st->quality = video_qscale;
2202
                }
2203 2204 2205 2206
                
                if(bitexact)
                    video_enc->flags |= CODEC_FLAG_BITEXACT;

2207 2208 2209
                if (use_hq) {
                    video_enc->flags |= CODEC_FLAG_HQ;
                }
2210 2211 2212 2213 2214 2215 2216 2217
           	/* Fx */ 
                if (use_umv) {
                    video_enc->flags |= CODEC_FLAG_H263P_UMV;
                }
           	if (use_aic) {
                    video_enc->flags |= CODEC_FLAG_H263P_AIC;
                }
           	/* /Fx */ 
2218 2219 2220 2221
                if (use_4mv) {
                    video_enc->flags |= CODEC_FLAG_HQ;
                    video_enc->flags |= CODEC_FLAG_4MV;
                }
2222
            
2223 2224
                if(use_part)
                    video_enc->flags |= CODEC_FLAG_PART;
2225 2226
               
            
2227 2228 2229 2230
                if (b_frames) {
                    video_enc->max_b_frames = b_frames;
                    video_enc->b_frame_strategy = 0;
                    video_enc->b_quant_factor = 2.0;
2231 2232
                }
            
2233 2234
                video_enc->qmin = video_qmin;
                video_enc->qmax = video_qmax;
M
Michael Niedermayer 已提交
2235 2236
                video_enc->mb_qmin = video_mb_qmin;
                video_enc->mb_qmax = video_mb_qmax;
2237 2238 2239 2240
                video_enc->max_qdiff = video_qdiff;
                video_enc->qblur = video_qblur;
                video_enc->qcompress = video_qcomp;
                video_enc->rc_eq = video_rc_eq;
M
Michael Niedermayer 已提交
2241
                video_enc->debug= debug;
2242 2243 2244 2245 2246 2247 2248 2249 2250 2251
                
                p= video_rc_override_string;
                for(i=0; p; i++){
                    int start, end, q;
                    int e=sscanf(p, "%d,%d,%d", &start, &end, &q);
                    if(e!=3){
                        fprintf(stderr, "error parsing rc_override\n");
                        exit(1);
                    }
                    video_enc->rc_override= 
F
Fabrice Bellard 已提交
2252 2253
                        av_realloc(video_enc->rc_override, 
                                   sizeof(RcOverride)*(i+1));
2254 2255 2256 2257 2258 2259 2260 2261 2262 2263 2264 2265 2266 2267 2268
                    video_enc->rc_override[i].start_frame= start;
                    video_enc->rc_override[i].end_frame  = end;
                    if(q>0){
                        video_enc->rc_override[i].qscale= q;
                        video_enc->rc_override[i].quality_factor= 1.0;
                    }
                    else{
                        video_enc->rc_override[i].qscale= 0;
                        video_enc->rc_override[i].quality_factor= -q/100.0;
                    }
                    p= strchr(p, '/');
                    if(p) p++;
                }
                video_enc->rc_override_count=i;

2269 2270 2271 2272
                video_enc->rc_max_rate = video_rc_max_rate;
                video_enc->rc_min_rate = video_rc_min_rate;
                video_enc->rc_buffer_size = video_rc_buffer_size;
                video_enc->rc_buffer_aggressivity= video_rc_buffer_aggressivity;
2273
                video_enc->rc_initial_cplx= video_rc_initial_cplx;
2274 2275 2276 2277 2278 2279
                video_enc->i_quant_factor = video_i_qfactor;
                video_enc->b_quant_factor = video_b_qfactor;
                video_enc->i_quant_offset = video_i_qoffset;
                video_enc->b_quant_offset = video_b_qoffset;
                video_enc->dct_algo = dct_algo;
                video_enc->idct_algo = idct_algo;
M
Michael Niedermayer 已提交
2280
                video_enc->strict_std_compliance = strict;
2281 2282 2283 2284
                if(packet_size){
                    video_enc->rtp_mode= 1;
                    video_enc->rtp_payload_size= packet_size;
                }
2285
            
2286
                if (do_psnr)
2287
                    video_enc->flags|= CODEC_FLAG_PSNR;
2288
            
2289
                video_enc->me_method = me_method;
2290

2291 2292 2293 2294 2295 2296 2297
                /* two pass mode */
                if (do_pass) {
                    if (do_pass == 1) {
                        video_enc->flags |= CODEC_FLAG_PASS1;
                    } else {
                        video_enc->flags |= CODEC_FLAG_PASS2;
                    }
2298
                }
2299
            }
F
merge  
Fabrice Bellard 已提交
2300 2301 2302 2303 2304 2305 2306 2307 2308 2309 2310 2311
            oc->streams[nb_streams] = st;
            nb_streams++;
        }
    
        if (use_audio) {
            AVCodecContext *audio_enc;

            st = av_mallocz(sizeof(AVStream));
            if (!st) {
                fprintf(stderr, "Could not alloc stream\n");
                exit(1);
            }
M
cleanup  
Michael Niedermayer 已提交
2312
            avcodec_get_context_defaults(&st->codec);
2313

F
merge  
Fabrice Bellard 已提交
2314 2315
            audio_enc = &st->codec;
            audio_enc->codec_type = CODEC_TYPE_AUDIO;
2316 2317 2318 2319 2320 2321 2322 2323 2324 2325 2326 2327 2328 2329 2330 2331 2332
            if (audio_stream_copy) {
                st->stream_copy = 1;
            } else {
                codec_id = file_oformat->audio_codec;
                if (audio_codec_id != CODEC_ID_NONE)
                    codec_id = audio_codec_id;
                audio_enc->codec_id = codec_id;
                
                audio_enc->bit_rate = audio_bit_rate;
                audio_enc->sample_rate = audio_sample_rate;
                /* For audio codecs other than AC3 we limit */
                /* the number of coded channels to stereo   */
                if (audio_channels > 2 && codec_id != CODEC_ID_AC3) {
                    audio_enc->channels = 2;
                } else
                    audio_enc->channels = audio_channels;
            }
F
merge  
Fabrice Bellard 已提交
2333 2334 2335 2336 2337 2338 2339
            oc->streams[nb_streams] = st;
            nb_streams++;
        }

        oc->nb_streams = nb_streams;

        if (!nb_streams) {
2340
            fprintf(stderr, "No audio or video streams available\n");
F
merge  
Fabrice Bellard 已提交
2341 2342 2343 2344
            exit(1);
        }

        if (str_title)
2345
            pstrcpy(oc->title, sizeof(oc->title), str_title);
F
merge  
Fabrice Bellard 已提交
2346
        if (str_author)
2347
            pstrcpy(oc->author, sizeof(oc->author), str_author);
F
merge  
Fabrice Bellard 已提交
2348
        if (str_copyright)
2349
            pstrcpy(oc->copyright, sizeof(oc->copyright), str_copyright);
F
merge  
Fabrice Bellard 已提交
2350
        if (str_comment)
2351
            pstrcpy(oc->comment, sizeof(oc->comment), str_comment);
F
merge  
Fabrice Bellard 已提交
2352 2353
    }

2354
    output_files[nb_output_files++] = oc;
F
merge  
Fabrice Bellard 已提交
2355 2356

    strcpy(oc->filename, filename);
2357 2358

    /* check filename in case of an image number is expected */
2359 2360 2361
    if (oc->oformat->flags & AVFMT_NEEDNUMBER) {
        if (filename_number_test(oc->filename) < 0) {
            print_error(oc->filename, AVERROR_NUMEXPECTED);
2362
            exit(1);
2363
        }
2364 2365
    }

2366
    if (!(oc->oformat->flags & AVFMT_NOFILE)) {
F
merge  
Fabrice Bellard 已提交
2367 2368 2369 2370 2371 2372 2373 2374 2375 2376 2377 2378 2379 2380 2381 2382 2383 2384 2385 2386 2387 2388 2389 2390
        /* test if it already exists to avoid loosing precious files */
        if (!file_overwrite && 
            (strchr(filename, ':') == NULL ||
             strstart(filename, "file:", NULL))) {
            if (url_exist(filename)) {
                int c;
                
                printf("File '%s' already exists. Overwrite ? [y/N] ", filename);
                fflush(stdout);
                c = getchar();
                if (toupper(c) != 'Y') {
                    fprintf(stderr, "Not overwriting - exiting\n");
                    exit(1);
                }
            }
        }
        
        /* open the file */
        if (url_fopen(&oc->pb, filename, URL_WRONLY) < 0) {
            fprintf(stderr, "Could not open '%s'\n", filename);
            exit(1);
        }
    }

2391 2392 2393 2394 2395 2396 2397 2398
    memset(ap, 0, sizeof(*ap));
    ap->image_format = image_format;
    if (av_set_parameters(oc, ap) < 0) {
        fprintf(stderr, "%s: Invalid encoding parameters\n",
                oc->filename);
        exit(1);
    }

F
merge  
Fabrice Bellard 已提交
2399
    /* reset some options */
2400 2401
    file_oformat = NULL;
    file_iformat = NULL;
2402
    image_format = NULL;
F
merge  
Fabrice Bellard 已提交
2403 2404 2405 2406
    audio_disable = 0;
    video_disable = 0;
    audio_codec_id = CODEC_ID_NONE;
    video_codec_id = CODEC_ID_NONE;
2407 2408
    audio_stream_copy = 0;
    video_stream_copy = 0;
F
merge  
Fabrice Bellard 已提交
2409 2410
}

2411
/* prepare dummy protocols for grab */
2412
static void prepare_grab(void)
2413 2414 2415 2416
{
    int has_video, has_audio, i, j;
    AVFormatContext *oc;
    AVFormatContext *ic;
2417
    AVFormatParameters vp1, *vp = &vp1;
2418
    AVFormatParameters ap1, *ap = &ap1;
2419
    
2420 2421 2422 2423
    /* see if audio/video inputs are needed */
    has_video = 0;
    has_audio = 0;
    memset(ap, 0, sizeof(*ap));
2424
    memset(vp, 0, sizeof(*vp));
2425 2426 2427 2428 2429 2430 2431 2432 2433 2434 2435 2436 2437
    for(j=0;j<nb_output_files;j++) {
        oc = output_files[j];
        for(i=0;i<oc->nb_streams;i++) {
            AVCodecContext *enc = &oc->streams[i]->codec;
            switch(enc->codec_type) {
            case CODEC_TYPE_AUDIO:
                if (enc->sample_rate > ap->sample_rate)
                    ap->sample_rate = enc->sample_rate;
                if (enc->channels > ap->channels)
                    ap->channels = enc->channels;
                has_audio = 1;
                break;
            case CODEC_TYPE_VIDEO:
2438 2439 2440 2441
                if (enc->width > vp->width)
                    vp->width = enc->width;
                if (enc->height > vp->height)
                    vp->height = enc->height;
2442 2443 2444 2445 2446 2447
                
                assert(enc->frame_rate_base == DEFAULT_FRAME_RATE_BASE);
                if (enc->frame_rate > vp->frame_rate){
                    vp->frame_rate      = enc->frame_rate;
                    vp->frame_rate_base = enc->frame_rate_base;
                }
2448 2449
                has_video = 1;
                break;
2450
            default:
2451
                av_abort();
2452 2453 2454 2455 2456 2457 2458 2459 2460 2461
            }
        }
    }
    
    if (has_video == 0 && has_audio == 0) {
        fprintf(stderr, "Output file must have at least one audio or video stream\n");
        exit(1);
    }
    
    if (has_video) {
2462
        AVInputFormat *fmt1;
2463
        fmt1 = av_find_input_format(video_grab_format);
2464 2465
        vp->device  = video_device;
        vp->channel = video_channel;
2466
	vp->standard = video_standard;
2467
        if (av_open_input_file(&ic, "", fmt1, 0, vp) < 0) {
2468
            fprintf(stderr, "Could not find video grab device\n");
2469 2470
            exit(1);
        }
2471
        /* by now video grab has one stream */
2472 2473
        ic->streams[0]->r_frame_rate      = vp->frame_rate;
        ic->streams[0]->r_frame_rate_base = vp->frame_rate_base;
2474
        input_files[nb_input_files] = ic;
2475
        dump_format(ic, nb_input_files, "", 0);
2476 2477
        nb_input_files++;
    }
2478
    if (has_audio && audio_grab_format) {
2479
        AVInputFormat *fmt1;
2480
        fmt1 = av_find_input_format(audio_grab_format);
2481
        ap->device = audio_device;
2482
        if (av_open_input_file(&ic, "", fmt1, 0, ap) < 0) {
2483
            fprintf(stderr, "Could not find audio grab device\n");
2484 2485 2486
            exit(1);
        }
        input_files[nb_input_files] = ic;
2487
        dump_format(ic, nb_input_files, "", 0);
2488 2489 2490 2491
        nb_input_files++;
    }
}

2492
/* same option as mencoder */
2493
static void opt_pass(const char *pass_str)
2494 2495 2496 2497 2498 2499 2500 2501 2502
{
    int pass;
    pass = atoi(pass_str);
    if (pass != 1 && pass != 2) {
        fprintf(stderr, "pass number can be only 1 or 2\n");
        exit(1);
    }
    do_pass = pass;
}
2503

2504
#if defined(CONFIG_WIN32) || defined(CONFIG_OS2)
2505
static int64_t getutime(void)
F
Fabrice Bellard 已提交
2506
{
2507
  return av_gettime();
F
Fabrice Bellard 已提交
2508
}
F
Fabrice Bellard 已提交
2509
#else
2510
static int64_t getutime(void)
F
Fabrice Bellard 已提交
2511
{
2512 2513 2514 2515
    struct rusage rusage;

    getrusage(RUSAGE_SELF, &rusage);
    return (rusage.ru_utime.tv_sec * 1000000LL) + rusage.ru_utime.tv_usec;
F
Fabrice Bellard 已提交
2516 2517
}
#endif
F
Fabrice Bellard 已提交
2518

2519 2520
extern int ffm_nopts;

2521
static void opt_bitexact(void)
2522
{
2523
    bitexact=1;
2524 2525 2526 2527
    /* disable generate of real time pts in ffm (need to be supressed anyway) */
    ffm_nopts = 1;
}

2528
static void show_formats(void)
F
merge  
Fabrice Bellard 已提交
2529
{
2530 2531
    AVInputFormat *ifmt;
    AVOutputFormat *ofmt;
2532
    AVImageFormat *image_fmt;
F
merge  
Fabrice Bellard 已提交
2533 2534 2535 2536
    URLProtocol *up;
    AVCodec *p;
    const char **pp;

2537
    printf("Output audio/video file formats:");
2538 2539
    for(ofmt = first_oformat; ofmt != NULL; ofmt = ofmt->next) {
        printf(" %s", ofmt->name);
F
merge  
Fabrice Bellard 已提交
2540 2541
    }
    printf("\n");
2542 2543

    printf("Input audio/video file formats:");
2544 2545
    for(ifmt = first_iformat; ifmt != NULL; ifmt = ifmt->next) {
        printf(" %s", ifmt->name);
F
merge  
Fabrice Bellard 已提交
2546 2547 2548
    }
    printf("\n");

2549 2550 2551 2552 2553 2554 2555 2556 2557 2558 2559 2560 2561 2562 2563 2564
    printf("Output image formats:");
    for(image_fmt = first_image_format; image_fmt != NULL; 
        image_fmt = image_fmt->next) {
        if (image_fmt->img_write)
            printf(" %s", image_fmt->name);
    }
    printf("\n");

    printf("Input image formats:");
    for(image_fmt = first_image_format; image_fmt != NULL; 
        image_fmt = image_fmt->next) {
        if (image_fmt->img_read)
            printf(" %s", image_fmt->name);
    }
    printf("\n");

F
merge  
Fabrice Bellard 已提交
2565 2566 2567 2568 2569 2570 2571 2572 2573 2574 2575 2576 2577 2578 2579 2580 2581 2582 2583 2584 2585 2586 2587 2588 2589
    printf("Codecs:\n");
    printf("  Encoders:");
    for(p = first_avcodec; p != NULL; p = p->next) {
        if (p->encode)
            printf(" %s", p->name);
    }
    printf("\n");

    printf("  Decoders:");
    for(p = first_avcodec; p != NULL; p = p->next) {
        if (p->decode)
            printf(" %s", p->name);
    }
    printf("\n");

    printf("Supported file protocols:");
    for(up = first_protocol; up != NULL; up = up->next)
        printf(" %s:", up->name);
    printf("\n");
    
    printf("Frame size abbreviations: sqcif qcif cif 4cif\n");
    printf("Motion estimation methods:");
    pp = motion_str;
    while (*pp) {
        printf(" %s", *pp);
2590
        if ((pp - motion_str + 1) == ME_ZERO) 
F
merge  
Fabrice Bellard 已提交
2591
            printf("(fastest)");
2592
        else if ((pp - motion_str + 1) == ME_FULL) 
F
merge  
Fabrice Bellard 已提交
2593
            printf("(slowest)");
2594
        else if ((pp - motion_str + 1) == ME_EPZS) 
F
merge  
Fabrice Bellard 已提交
2595 2596 2597 2598 2599 2600 2601 2602
            printf("(default)");
        pp++;
    }
    printf("\n");
    exit(1);
}

const OptionDef options[] = {
F
Fabrice Bellard 已提交
2603 2604 2605 2606
    { "L", 0, {(void*)show_licence}, "show license" },
    { "h", 0, {(void*)show_help}, "show help" },
    { "formats", 0, {(void*)show_formats}, "show available formats, codecs, protocols, ..." },
    { "f", HAS_ARG, {(void*)opt_format}, "force format", "fmt" },
2607
    { "img", HAS_ARG, {(void*)opt_image_format}, "force image format", "img_fmt" },
F
Fabrice Bellard 已提交
2608 2609 2610 2611 2612 2613 2614 2615
    { "i", HAS_ARG, {(void*)opt_input_file}, "input file name", "filename" },
    { "y", OPT_BOOL, {(void*)&file_overwrite}, "overwrite output files" },
    { "map", HAS_ARG | OPT_EXPERT, {(void*)opt_map}, "set input stream mapping", "file:stream" },
    { "t", HAS_ARG, {(void*)opt_recording_time}, "set the recording time", "duration" },
    { "title", HAS_ARG | OPT_STRING, {(void*)&str_title}, "set the title", "string" },
    { "author", HAS_ARG | OPT_STRING, {(void*)&str_author}, "set the author", "string" },
    { "copyright", HAS_ARG | OPT_STRING, {(void*)&str_copyright}, "set the copyright", "string" },
    { "comment", HAS_ARG | OPT_STRING, {(void*)&str_comment}, "set the comment", "string" },
2616 2617
    { "pass", HAS_ARG, {(void*)&opt_pass}, "select the pass number (1 or 2)", "n" },
    { "passlogfile", HAS_ARG | OPT_STRING, {(void*)&pass_logfilename}, "select two pass log file name", "file" },
F
merge  
Fabrice Bellard 已提交
2618
    /* video options */
F
Fabrice Bellard 已提交
2619 2620
    { "b", HAS_ARG, {(void*)opt_video_bitrate}, "set video bitrate (in kbit/s)", "bitrate" },
    { "r", HAS_ARG, {(void*)opt_frame_rate}, "set frame rate (in Hz)", "rate" },
2621
    { "re", OPT_BOOL|OPT_EXPERT, {(void*)&rate_emu}, "read input at native frame rate" },
F
Fabrice Bellard 已提交
2622
    { "s", HAS_ARG, {(void*)opt_frame_size}, "set frame size (WxH or abbreviation)", "size" },
2623
    { "aspect", HAS_ARG, {(void*)opt_frame_aspect_ratio}, "set aspect ratio (4:3, 16:9 or 1.3333, 1.7777)", "aspect" },
2624
    { "pix_fmt", HAS_ARG | OPT_EXPERT, {(void*)opt_frame_pix_fmt}, "set pixel format", "format" },
2625 2626 2627 2628
    { "croptop", HAS_ARG, {(void*)opt_frame_crop_top}, "set top crop band size (in pixels)", "size" },
    { "cropbottom", HAS_ARG, {(void*)opt_frame_crop_bottom}, "set bottom crop band size (in pixels)", "size" },
    { "cropleft", HAS_ARG, {(void*)opt_frame_crop_left}, "set left crop band size (in pixels)", "size" },
    { "cropright", HAS_ARG, {(void*)opt_frame_crop_right}, "set right crop band size (in pixels)", "size" },
F
Fabrice Bellard 已提交
2629 2630 2631 2632
    { "g", HAS_ARG | OPT_EXPERT, {(void*)opt_gop_size}, "set the group of picture size", "gop_size" },
    { "intra", OPT_BOOL | OPT_EXPERT, {(void*)&intra_only}, "use only intra frames"},
    { "vn", OPT_BOOL, {(void*)&video_disable}, "disable video" },
    { "qscale", HAS_ARG | OPT_EXPERT, {(void*)opt_qscale}, "use fixed video quantiser scale (VBR)", "q" },
2633 2634
    { "qmin", HAS_ARG | OPT_EXPERT, {(void*)opt_qmin}, "min video quantiser scale (VBR)", "q" },
    { "qmax", HAS_ARG | OPT_EXPERT, {(void*)opt_qmax}, "max video quantiser scale (VBR)", "q" },
M
Michael Niedermayer 已提交
2635 2636
    { "mbqmin", HAS_ARG | OPT_EXPERT, {(void*)opt_mb_qmin}, "min macroblock quantiser scale (VBR)", "q" },
    { "mbqmax", HAS_ARG | OPT_EXPERT, {(void*)opt_mb_qmax}, "max macroblock quantiser scale (VBR)", "q" },
2637 2638 2639
    { "qdiff", HAS_ARG | OPT_EXPERT, {(void*)opt_qdiff}, "max difference between the quantiser scale (VBR)", "q" },
    { "qblur", HAS_ARG | OPT_EXPERT, {(void*)opt_qblur}, "video quantiser scale blur (VBR)", "blur" },
    { "qcomp", HAS_ARG | OPT_EXPERT, {(void*)opt_qcomp}, "video quantiser scale compression (VBR)", "compression" },
2640
    { "rc_init_cplx", HAS_ARG | OPT_EXPERT, {(void*)opt_rc_initial_cplx}, "initial complexity for 1-pass encoding", "complexity" },
2641 2642 2643 2644
    { "b_qfactor", HAS_ARG | OPT_EXPERT, {(void*)opt_b_qfactor}, "qp factor between p and b frames", "factor" },
    { "i_qfactor", HAS_ARG | OPT_EXPERT, {(void*)opt_i_qfactor}, "qp factor between p and i frames", "factor" },
    { "b_qoffset", HAS_ARG | OPT_EXPERT, {(void*)opt_b_qoffset}, "qp offset between p and b frames", "offset" },
    { "i_qoffset", HAS_ARG | OPT_EXPERT, {(void*)opt_i_qoffset}, "qp offset between p and i frames", "offset" },
M
Michael Niedermayer 已提交
2645
//    { "b_strategy", HAS_ARG | OPT_EXPERT, {(void*)opt_b_strategy}, "dynamic b frame selection strategy", "strategy" },
M
Michael Niedermayer 已提交
2646
    { "rc_eq", HAS_ARG | OPT_EXPERT, {(void*)opt_video_rc_eq}, "", "equation" },
2647
    { "rc_override", HAS_ARG | OPT_EXPERT, {(void*)opt_video_rc_override_string}, "Rate control override", "qualities for specific intervals" },
2648
    { "bt", HAS_ARG, {(void*)opt_video_bitrate_tolerance}, "set video bitrate tolerance (in kbit/s)", "tolerance" },
M
Michael Niedermayer 已提交
2649 2650
    { "maxrate", HAS_ARG, {(void*)opt_video_bitrate_max}, "set max video bitrate tolerance (in kbit/s)", "bitrate" },
    { "minrate", HAS_ARG, {(void*)opt_video_bitrate_min}, "set min video bitrate tolerance (in kbit/s)", "bitrate" },
M
fixing  
Michael Niedermayer 已提交
2651
    { "bufsize", HAS_ARG, {(void*)opt_video_buffer_size}, "set ratecontrol buffere size (in kbit)", "size" },
2652
    { "vd", HAS_ARG | OPT_EXPERT, {(void*)opt_video_device}, "set video grab device", "device" },
2653
    { "vc", HAS_ARG | OPT_EXPERT, {(void*)opt_video_channel}, "set video grab channel (DV1394 only)", "channel" },
2654
    { "tvstd", HAS_ARG | OPT_EXPERT, {(void*)opt_video_standard}, "set television standard (NTSC, PAL (SECAM))", "standard" },
2655
    { "dv1394", OPT_EXPERT, {(void*)opt_dv1394}, "set DV1394 grab", "" },
2656
    { "vcodec", HAS_ARG | OPT_EXPERT, {(void*)opt_video_codec}, "force video codec ('copy' to copy stream)", "codec" },
F
Fabrice Bellard 已提交
2657
    { "me", HAS_ARG | OPT_EXPERT, {(void*)opt_motion_estimation}, "set motion estimation method", 
F
merge  
Fabrice Bellard 已提交
2658
      "method" },
2659
    { "dct_algo", HAS_ARG | OPT_EXPERT, {(void*)opt_dct_algo}, "set dct algo",  "algo" },
2660
    { "idct_algo", HAS_ARG | OPT_EXPERT, {(void*)opt_idct_algo}, "set idct algo",  "algo" },
2661
    { "er", HAS_ARG | OPT_EXPERT, {(void*)opt_error_resilience}, "set error resilience",  "" },
M
Michael Niedermayer 已提交
2662
    { "ec", HAS_ARG | OPT_EXPERT, {(void*)opt_error_concealment}, "set error concealment",  "" },
2663
    { "bf", HAS_ARG | OPT_EXPERT, {(void*)opt_b_frames}, "use 'frames' B frames (only MPEG-4)", "frames" },
2664
    { "hq", OPT_BOOL | OPT_EXPERT, {(void*)&use_hq}, "activate high quality settings" },
J
Juanjo 已提交
2665
    { "4mv", OPT_BOOL | OPT_EXPERT, {(void*)&use_4mv}, "use four motion vector by macroblock (only MPEG-4)" },
2666
    { "part", OPT_BOOL | OPT_EXPERT, {(void*)&use_part}, "use data partitioning (only MPEG-4)" },
2667
    { "bug", HAS_ARG | OPT_EXPERT, {(void*)opt_workaround_bugs}, "workaround not auto detected encoder bugs", "param" },
2668
    { "ps", HAS_ARG | OPT_EXPERT, {(void*)opt_packet_size}, "packet size", "size in bits" },
M
Michael Niedermayer 已提交
2669
    { "strict", HAS_ARG | OPT_EXPERT, {(void*)opt_strict}, "strictness", "how strictly to follow the standarts" },
F
Fabrice Bellard 已提交
2670
    { "sameq", OPT_BOOL, {(void*)&same_quality}, 
F
merge  
Fabrice Bellard 已提交
2671
      "use same video quality as source (implies VBR)" },
M
10l  
Michael Niedermayer 已提交
2672
    { "debug", HAS_ARG | OPT_EXPERT, {(void*)opt_debug}, "print specific debug info", "" },
F
merge  
Fabrice Bellard 已提交
2673
    /* audio options */
F
Fabrice Bellard 已提交
2674 2675 2676 2677 2678
    { "ab", HAS_ARG, {(void*)opt_audio_bitrate}, "set audio bitrate (in kbit/s)", "bitrate", },
    { "ar", HAS_ARG, {(void*)opt_audio_rate}, "set audio sampling rate (in Hz)", "rate" },
    { "ac", HAS_ARG, {(void*)opt_audio_channels}, "set number of audio channels", "channels" },
    { "an", OPT_BOOL, {(void*)&audio_disable}, "disable audio" },
    { "ad", HAS_ARG | OPT_EXPERT, {(void*)opt_audio_device}, "set audio device", "device" },
2679
    { "acodec", HAS_ARG | OPT_EXPERT, {(void*)opt_audio_codec}, "force audio codec ('copy' to copy stream)", "codec" },
F
Fabrice Bellard 已提交
2680
    { "deinterlace", OPT_BOOL | OPT_EXPERT, {(void*)&do_deinterlace}, 
2681
      "deinterlace pictures" },
F
Fabrice Bellard 已提交
2682
    { "benchmark", OPT_BOOL | OPT_EXPERT, {(void*)&do_benchmark}, 
F
Fabrice Bellard 已提交
2683
      "add timings for benchmarking" },
2684 2685
    { "hex", OPT_BOOL | OPT_EXPERT, {(void*)&do_hex_dump}, 
      "dump each input packet" },
2686 2687
    { "psnr", OPT_BOOL | OPT_EXPERT, {(void*)&do_psnr}, "calculate PSNR of compressed frames" },
    { "vstats", OPT_BOOL | OPT_EXPERT, {(void*)&do_vstats}, "dump video coding statistics to file" }, 
2688
    { "bitexact", OPT_EXPERT, {(void*)opt_bitexact}, "only use bit exact algorithms (for codec testing)" }, 
2689
    { "vhook", HAS_ARG | OPT_EXPERT, {(void*)add_frame_hooker}, "insert video processing module", "module name and parameters" },
2690 2691 2692 2693
    /* Fx */
    { "aic", OPT_BOOL | OPT_EXPERT, {(void*)&use_aic}, "enable Advanced intra coding (h263+)" },
    { "umv", OPT_BOOL | OPT_EXPERT, {(void*)&use_umv}, "enable Unlimited Motion Vector (h263+)" },
    /* /Fx */
F
merge  
Fabrice Bellard 已提交
2694 2695 2696
    { NULL, },
};

F
Fabrice Bellard 已提交
2697 2698 2699 2700 2701 2702 2703 2704 2705 2706 2707 2708 2709 2710 2711
void show_help(void)
{
    printf("ffmpeg version " FFMPEG_VERSION ", Copyright (c) 2000, 2001, 2002 Fabrice Bellard\n");
    printf("usage: ffmpeg [[options] -i input_file]... {[options] outfile}...\n"
           "Hyper fast Audio and Video encoder\n");
    printf("\n");
    show_help_options(options);
    exit(1);
}

void parse_arg_file(const char *filename)
{
    opt_output_file(filename);
}

F
merge  
Fabrice Bellard 已提交
2712 2713
int main(int argc, char **argv)
{
F
Fabrice Bellard 已提交
2714
    int i;
2715
    int64_t ti;
2716

2717
    av_register_all();
F
merge  
Fabrice Bellard 已提交
2718 2719 2720

    if (argc <= 1)
        show_help();
2721 2722
    
    /* parse options */
F
Fabrice Bellard 已提交
2723
    parse_options(argc, argv, options);
F
merge  
Fabrice Bellard 已提交
2724

F
Fabrice Bellard 已提交
2725 2726 2727 2728 2729 2730 2731 2732
    /* file converter / grab */
    if (nb_output_files <= 0) {
        fprintf(stderr, "Must supply at least one output file\n");
        exit(1);
    }
    
    if (nb_input_files == 0) {
        prepare_grab();
2733 2734 2735 2736 2737 2738 2739 2740
    }

    ti = getutime();
    av_encode(output_files, nb_output_files, input_files, nb_input_files, 
              stream_maps, nb_stream_maps);
    ti = getutime() - ti;
    if (do_benchmark) {
        printf("bench: utime=%0.3fs\n", ti / 1000000.0);
F
merge  
Fabrice Bellard 已提交
2741 2742 2743 2744
    }

    /* close files */
    for(i=0;i<nb_output_files;i++) {
2745 2746 2747 2748 2749 2750 2751 2752
        /* maybe av_close_output_file ??? */
        AVFormatContext *s = output_files[i];
	int j;
        if (!(s->oformat->flags & AVFMT_NOFILE))
	    url_fclose(&s->pb);
	for(j=0;j<s->nb_streams;j++)
	    av_free(s->streams[j]);
        av_free(s);
F
merge  
Fabrice Bellard 已提交
2753
    }
2754 2755
    for(i=0;i<nb_input_files;i++)
        av_close_input_file(input_files[i]);
F
merge  
Fabrice Bellard 已提交
2756

2757
    av_free_static();
2758 2759

    
2760 2761 2762 2763
#ifdef POWERPC_TBL_PERFORMANCE_REPORT
    extern void powerpc_display_perf_report(void);
    powerpc_display_perf_report();
#endif /* POWERPC_TBL_PERFORMANCE_REPORT */
2764

2765 2766 2767 2768 2769 2770 2771 2772 2773
#ifndef CONFIG_WIN32
    if (received_sigterm) {
        fprintf(stderr,
            "Received signal %d: terminating.\n",
            (int) received_sigterm);
        exit (255);
    }
#endif
    exit(0); /* not all OS-es handle main() return value */
F
merge  
Fabrice Bellard 已提交
2774 2775
    return 0;
}