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