projects / libav.git / blob
? search:


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