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