3 * Copyright (c) 2000-2003 Fabrice Bellard
5 * This file is part of FFmpeg.
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.
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.
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
22 /* needed for usleep() */
23 #define _XOPEN_SOURCE 600
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"
44 #if HAVE_SYS_RESOURCE_H
45 #include <sys/types.h>
46 #include <sys/resource.h>
47 #elif HAVE_GETPROCESSTIMES
52 #include <sys/select.h>
57 #include <sys/ioctl.h>
63 #undef time //needed because HAVE_AV_CONFIG_H is defined on top
73 const char program_name[] = "FFmpeg";
74 const int program_birth_year = 2000;
76 /* select an input stream for an output stream */
77 typedef struct AVStreamMap {
81 int sync_stream_index;
84 /** select an input file for an output file */
85 typedef struct AVMetaDataMap {
90 static const OptionDef options[];
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;
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;
106 static AVStreamMap stream_maps[MAX_FILES*MAX_STREAMS];
107 static int nb_stream_maps;
109 static AVMetaDataMap meta_data_maps[MAX_FILES];
110 static int nb_meta_data_maps;
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;
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;
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;
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;
168 static float mux_preload= 0.5;
169 static float mux_max_delay= 0.7;
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;
198 static int rate_emu = 0;
200 static int video_channel = 0;
201 static char *video_standard;
203 static int audio_volume = 256;
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;
219 static int pgmyuv_compatibility_hack=0;
220 static float dts_delta_threshold = 10;
222 static unsigned int sws_flags = SWS_BICUBIC;
224 static int64_t timer_start;
226 static uint8_t *audio_buf;
227 static uint8_t *audio_out;
228 static uint8_t *audio_out2;
230 static short *samples;
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];
237 #define DEFAULT_PASS_LOGFILENAME_PREFIX "ffmpeg2pass"
239 struct AVInputStream;
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 */
248 /* input pts and corresponding output pts
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
255 AVFrame pict_tmp; /* temporary image for resampling */
256 struct SwsContext *img_resample_ctx; /* for image resampling */
260 int topBand; /* cropping area sizes */
264 int padtop; /* padding area sizes */
271 ReSampleContext *resample; /* for audio resampling */
273 AVAudioConvert *reformat_ctx;
274 AVFifoBuffer *fifo; /* for compression: one audio fifo per codec */
278 typedef struct AVInputStream {
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 */
286 int64_t start; /* time when read started */
287 int64_t next_pts; /* synthetic pts for cases where pkt.pts
289 int64_t pts; /* current pts */
290 int is_start; /* is 1 at the start and after a discontinuity */
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 */
302 /* init terminal so that we can grab keys */
303 static struct termios oldtty;
306 static void term_exit(void)
309 tcsetattr (0, TCSANOW, &oldtty);
313 static volatile sig_atomic_t received_sigterm = 0;
316 sigterm_handler(int sig)
318 received_sigterm = sig;
322 static void term_init(void)
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);
339 tcsetattr (0, TCSANOW, &tty);
340 signal(SIGQUIT, sigterm_handler); /* Quit (POSIX). */
343 signal(SIGINT , sigterm_handler); /* Interrupt (ANSI). */
344 signal(SIGTERM, sigterm_handler); /* Termination (ANSI). */
346 register a function to be called at normal program termination
349 #if CONFIG_BEOS_NETSERVER
350 fcntl(0, F_SETFL, fcntl(0, F_GETFL) | O_NONBLOCK);
354 /* read a key without blocking */
355 static int read_key(void)
360 #if !CONFIG_BEOS_NETSERVER
368 n = select(1, &rfds, NULL, NULL, &tv);
384 static int decode_interrupt_cb(void)
386 return q_pressed || (q_pressed = read_key() == 'q');
389 static int av_exit(int ret)
394 for(i=0;i<nb_output_files;i++) {
395 /* maybe av_close_output_file ??? */
396 AVFormatContext *s = output_files[i];
398 if (!(s->oformat->flags & AVFMT_NOFILE) && 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]);
405 for(j=0;j<s->nb_programs;j++) {
406 av_metadata_free(&s->programs[j]->metadata);
408 for(j=0;j<s->nb_chapters;j++) {
409 av_metadata_free(&s->chapters[j]->metadata);
411 av_metadata_free(&s->metadata);
414 for(i=0;i<nb_input_files;i++)
415 av_close_input_file(input_files[i]);
417 av_free(intra_matrix);
418 av_free(inter_matrix);
422 av_free(vstats_filename);
426 av_free(video_codec_name);
427 av_free(audio_codec_name);
428 av_free(subtitle_codec_name);
430 av_free(video_standard);
432 #if CONFIG_POWERPC_PERF
433 void powerpc_display_perf_report(void);
434 powerpc_display_perf_report();
435 #endif /* CONFIG_POWERPC_PERF */
437 for (i=0;i<CODEC_TYPE_NB;i++)
438 av_free(avcodec_opts[i]);
439 av_free(avformat_opts);
446 if (received_sigterm) {
448 "Received signal %d: terminating.\n",
449 (int) received_sigterm);
453 exit(ret); /* not all OS-es handle main() return value */
457 static int read_ffserver_streams(AVFormatContext *s, const char *filename)
463 err = av_open_input_file(&ic, filename, NULL, FFM_PACKET_SIZE, NULL);
466 /* copy stream format */
467 s->nb_streams = ic->nb_streams;
468 for(i=0;i<ic->nb_streams;i++) {
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));
478 if (st->codec->codec_type == CODEC_TYPE_AUDIO && audio_stream_copy)
480 else if (st->codec->codec_type == CODEC_TYPE_VIDEO && video_stream_copy)
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);
488 if(st->codec->flags & CODEC_FLAG_BITEXACT)
493 s->timestamp = av_gettime();
495 av_close_input_file(ic);
500 get_sync_ipts(const AVOutputStream *ost)
502 const AVInputStream *ist = ost->sync_ist;
503 return (double)(ist->pts - start_time)/AV_TIME_BASE;
506 static void write_frame(AVFormatContext *s, AVPacket *pkt, AVCodecContext *avctx, AVBitStreamFilterContext *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);
517 new_pkt.destruct= av_destruct_packet;
519 fprintf(stderr, "%s failed for stream %d, codec %s",
520 bsfc->filter->name, pkt->stream_index,
521 avctx->codec ? avctx->codec->name : "copy");
531 ret= av_interleaved_write_frame(s, pkt);
533 print_error("av_interleaved_write_frame()", ret);
538 #define MAX_AUDIO_PACKET_SIZE (128 * 1024)
540 static void do_audio_out(AVFormatContext *s,
543 unsigned char *buf, int size)
546 const int audio_out_size= 4*MAX_AUDIO_PACKET_SIZE;
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;
554 /* SC: dynamic allocation of buffers */
556 audio_buf = av_malloc(2*MAX_AUDIO_PACKET_SIZE);
558 audio_out = av_malloc(audio_out_size);
559 if (!audio_buf || !audio_out)
560 return; /* Should signal an error ! */
562 if (enc->channels != dec->channels)
563 ost->audio_resample = 1;
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,
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);
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) {
584 audio_out2 = av_malloc(audio_out_size);
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));
597 ost->reformat_pair=MAKE_SFMT_PAIR(enc->sample_fmt,dec->sample_fmt);
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;
606 //FIXME resample delay
607 if(fabs(delta) > 50){
608 if(ist->is_start || fabs(delta) > audio_drift_threshold*enc->sample_rate){
610 byte_delta= FFMAX(byte_delta, -size);
614 fprintf(stderr, "discarding %d audio samples\n", (int)-delta);
619 static uint8_t *input_tmp= NULL;
620 input_tmp= av_realloc(input_tmp, byte_delta + size);
622 if(byte_delta + size <= MAX_AUDIO_PACKET_SIZE)
625 byte_delta= MAX_AUDIO_PACKET_SIZE - size;
627 memset(input_tmp, 0, byte_delta);
628 memcpy(input_tmp + byte_delta, buf, size);
632 fprintf(stderr, "adding %d audio samples of silence\n", (int)delta);
634 }else if(audio_sync_method>1){
635 int comp= av_clip(delta, -audio_sync_method, audio_sync_method);
636 assert(ost->audio_resample);
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);
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
647 if (ost->audio_resample) {
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;
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");
671 size_out = len*osize;
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");
681 av_fifo_generic_write(ost->fifo, buftmp, size_out, NULL);
683 frame_bytes = enc->frame_size * osize * enc->channels;
685 while (av_fifo_size(ost->fifo) >= frame_bytes) {
687 av_init_packet(&pkt);
689 av_fifo_generic_read(ost->fifo, audio_buf, frame_bytes, NULL);
691 //FIXME pass ost->sync_opts as AVFrame.pts in avcodec_encode_audio()
693 ret = avcodec_encode_audio(enc, audio_out, audio_out_size,
696 fprintf(stderr, "Audio encoding failed\n");
700 pkt.stream_index= ost->index;
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]);
708 ost->sync_opts += enc->frame_size;
712 int coded_bps = av_get_bits_per_sample(enc->codec->id)/8;
713 av_init_packet(&pkt);
715 ost->sync_opts += size_out / (osize * enc->channels);
717 /* output a pcm frame */
718 /* determine the size of the coded buffer */
721 size_out *= coded_bps;
723 //FIXME pass ost->sync_opts as AVFrame.pts in avcodec_encode_audio()
724 ret = avcodec_encode_audio(enc, audio_out, size_out,
727 fprintf(stderr, "Audio encoding failed\n");
731 pkt.stream_index= ost->index;
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]);
741 static void pre_process_video_frame(AVInputStream *ist, AVPicture *picture, void **bufp)
745 AVPicture picture_tmp;
748 dec = ist->st->codec;
750 /* deinterlace : must be done before any resize */
751 if (do_deinterlace) {
754 /* create temporary picture */
755 size = avpicture_get_size(dec->pix_fmt, dec->width, dec->height);
756 buf = av_malloc(size);
760 picture2 = &picture_tmp;
761 avpicture_fill(picture2, buf, dec->pix_fmt, dec->width, dec->height);
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");
775 if (picture != picture2)
776 *picture = *picture2;
780 /* we begin to correct av delay at this threshold */
781 #define AV_DELAY_MAX 0.100
783 static void do_subtitle_out(AVFormatContext *s,
789 static uint8_t *subtitle_out = NULL;
790 int subtitle_out_max_size = 65536;
791 int subtitle_out_size, nb, i;
795 if (pts == AV_NOPTS_VALUE) {
796 fprintf(stderr, "Subtitle packets must have a pts\n");
802 enc = ost->st->codec;
805 subtitle_out = av_malloc(subtitle_out_max_size);
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)
816 for(i = 0; i < nb; i++) {
817 subtitle_out_size = avcodec_encode_subtitle(enc, subtitle_out,
818 subtitle_out_max_size, sub);
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 */
829 pkt.pts += 90 * sub->start_display_time;
831 pkt.pts += 90 * sub->end_display_time;
833 write_frame(s, &pkt, ost->st->codec, bitstream_filters[ost->file_index][pkt.stream_index]);
837 static int bit_buffer_size= 1024*256;
838 static uint8_t *bit_buffer= NULL;
840 static void do_video_out(AVFormatContext *s,
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;
851 avcodec_get_frame_defaults(&picture_crop_temp);
852 avcodec_get_frame_defaults(&picture_pad_temp);
854 enc = ost->st->codec;
855 dec = ist->st->codec;
857 /* by default, we output a single frame */
862 if(video_sync_method>0 || (video_sync_method && av_q2d(enc->time_base) > 0.001)){
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
868 else if (video_sync_method == 2 || (video_sync_method<0 && (s->oformat->flags & AVFMT_VARIABLE_FPS))){
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);
879 fprintf(stderr, "*** drop!\n");
880 }else if (nb_frames > 1) {
881 nb_frames_dup += nb_frames;
883 fprintf(stderr, "*** %d dup!\n", nb_frames-1);
886 ost->sync_opts= lrintf(get_sync_ipts(ost) / av_q2d(enc->time_base));
888 nb_frames= FFMIN(nb_frames, max_frames[CODEC_TYPE_VIDEO] - ost->frame_number);
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");
899 formatted_picture = &picture_crop_temp;
901 formatted_picture = in_picture;
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");
916 resampling_dst = &picture_pad_temp;
920 if (ost->video_resample) {
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);
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);
933 /* duplicates frame if needed */
934 for(i=0;i<nb_frames;i++) {
936 av_init_packet(&pkt);
937 pkt.stream_index= ost->index;
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
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;
950 write_frame(s, &pkt, ost->st->codec, bitstream_filters[ost->file_index][pkt.stream_index]);
951 enc->coded_frame = old_frame;
955 big_picture= *final_picture;
956 /* better than nothing: use input picture interlaced
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;
963 big_picture.top_field_first = top_field_first;
966 /* handles sameq here. This is not correct because it may
967 not be a global option */
969 big_picture.quality = ist->st->quality;
971 big_picture.quality = ost->st->quality;
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,
982 fprintf(stderr, "Video encoding failed\n");
987 pkt.data= bit_buffer;
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);*/
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]);
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);
1009 ost->frame_number++;
1013 static double psnr(double d){
1014 return -10.0*log(d)/log(10.0);
1017 static void do_video_stats(AVFormatContext *os, AVOutputStream *ost,
1020 AVCodecContext *enc;
1022 double ti1, bitrate, avg_bitrate;
1024 /* this is executed just the first time do_video_stats is called */
1026 vstats_file = fopen(vstats_filename, "w");
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)));
1040 fprintf(vstats_file,"f_size= %6d ", frame_size);
1041 /* compute pts value */
1042 ti1 = ost->sync_opts * av_q2d(enc->time_base);
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));
1054 static void print_report(AVFormatContext **output_files,
1055 AVOutputStream **ost_table, int nb_ostreams,
1059 AVOutputStream *ost;
1060 AVFormatContext *oc;
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];
1068 if (!is_last_report) {
1070 /* display the report every 0.5 seconds */
1071 cur_time = av_gettime();
1072 if (last_time == -1) {
1073 last_time = cur_time;
1076 if ((cur_time - last_time) < 500000)
1078 last_time = cur_time;
1082 oc = output_files[0];
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);
1091 for(i=0;i<nb_ostreams;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);
1099 if (!vid && enc->codec_type == CODEC_TYPE_VIDEO) {
1100 float t = (av_gettime()-timer_start) / 1000000.0;
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);
1108 snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "L");
1111 int qp= lrintf(enc->coded_frame->quality/(float)FF_QP2LAMBDA);
1112 if(qp>=0 && qp<FF_ARRAY_ELEMS(qp_histogram))
1115 snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "%X", (int)lrintf(log(qp_histogram[j]+1)/log(2)));
1117 if (enc->flags&CODEC_FLAG_PSNR){
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=");
1125 error= enc->error[j];
1126 scale= enc->width*enc->height*255.0*255.0*frame_number;
1128 error= enc->coded_frame->error[j];
1129 scale= enc->width*enc->height*255.0*255.0;
1134 snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "%c:%2.2f ", type[j], psnr(error/scale));
1136 snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "*:%2.2f ", psnr(error_sum/scale_sum));
1140 /* compute min output value */
1141 pts = (double)ost->st->pts.val * av_q2d(ost->st->time_base);
1142 if ((pts < ti1) && (pts > 0))
1148 if (verbose || is_last_report) {
1149 bitrate = (double)(total_size * 8) / ti1 / 1000.0;
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);
1156 snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), " dup=%d drop=%d",
1157 nb_frames_dup, nb_frames_drop);
1160 fprintf(stderr, "%s \r", buf);
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",
1172 100.0*(total_size - raw)/raw
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)
1182 AVFormatContext *os;
1183 AVOutputStream *ost;
1186 int data_size, got_picture;
1188 void *buffer_to_free;
1189 static unsigned int samples_size= 0;
1190 AVSubtitle subtitle, *subtitle_to_free;
1194 if(ist->next_pts == AV_NOPTS_VALUE)
1195 ist->next_pts= ist->pts;
1199 av_init_packet(&avpkt);
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);
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)) {
1213 ist->pts= ist->next_pts;
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);
1218 /* decode the packet if needed */
1219 data_buf = NULL; /* fail safe */
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);
1228 samples= av_malloc(samples_size);
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,
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 */
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);
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);
1254 ret = avcodec_decode_video2(ist->st->codec,
1255 &picture, &got_picture, &avpkt);
1256 ist->st->quality= picture.quality;
1260 /* no picture yet */
1261 goto discard_packet;
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;
1271 case CODEC_TYPE_SUBTITLE:
1272 ret = avcodec_decode_subtitle2(ist->st->codec,
1273 &subtitle, &got_subtitle, &avpkt);
1276 if (!got_subtitle) {
1277 goto discard_packet;
1279 subtitle_to_free = &subtitle;
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;
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;
1300 data_buf = avpkt.data;
1301 data_size = avpkt.size;
1306 buffer_to_free = NULL;
1307 if (ist->st->codec->codec_type == CODEC_TYPE_VIDEO) {
1308 pre_process_video_frame(ist, (AVPicture *)&picture,
1312 // preprocess audio (volume)
1313 if (ist->st->codec->codec_type == CODEC_TYPE_AUDIO) {
1314 if (audio_volume != 256) {
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;
1326 /* frame rate emulation */
1328 int64_t pts = av_rescale(ist->pts, 1000000, AV_TIME_BASE);
1329 int64_t now = av_gettime() - ist->start;
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++) {
1341 if (ost->source_index == ist_index) {
1342 os = output_files[ost->file_index];
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));
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;
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);
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);
1363 case CODEC_TYPE_SUBTITLE:
1364 do_subtitle_out(os, ost, ist, &subtitle,
1371 AVFrame avframe; //FIXME/XXX remove this
1373 int64_t ost_tb_start_time= av_rescale_q(start_time, AV_TIME_BASE_Q, ost->st->time_base);
1375 av_init_packet(&opkt);
1377 if ((!ost->frame_number && !(pkt->flags & PKT_FLAG_KEY)) && !copy_initial_nonkeyframes)
1380 /* no reencoding needed : output the packet directly */
1381 /* force the input stream PTS */
1383 avcodec_get_frame_defaults(&avframe);
1384 ost->st->codec->coded_frame= &avframe;
1385 avframe.key_frame = pkt->flags & PKT_FLAG_KEY;
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;
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;
1398 opkt.pts= AV_NOPTS_VALUE;
1400 if (pkt->dts == AV_NOPTS_VALUE)
1401 opkt.dts = av_rescale_q(ist->pts, AV_TIME_BASE_Q, ost->st->time_base);
1403 opkt.dts = av_rescale_q(pkt->dts, ist->st->time_base, ost->st->time_base);
1404 opkt.dts -= ost_tb_start_time;
1406 opkt.duration = av_rescale_q(pkt->duration, ist->st->time_base, ost->st->time_base);
1407 opkt.flags= pkt->flags;
1409 //FIXME remove the following 2 lines they shall be replaced by the bitstream filters
1410 if(av_parser_change(ist->st->parser, ost->st->codec, &opkt.data, &opkt.size, data_buf, data_size, pkt->flags & PKT_FLAG_KEY))
1411 opkt.destruct= av_destruct_packet;
1413 write_frame(os, &opkt, ost->st->codec, bitstream_filters[ost->file_index][opkt.stream_index]);
1414 ost->st->codec->frame_number++;
1415 ost->frame_number++;
1416 av_free_packet(&opkt);
1420 av_free(buffer_to_free);
1421 /* XXX: allocate the subtitles in the codec ? */
1422 if (subtitle_to_free) {
1423 if (subtitle_to_free->rects != NULL) {
1424 for (i = 0; i < subtitle_to_free->num_rects; i++) {
1425 av_freep(&subtitle_to_free->rects[i]->pict.data[0]);
1426 av_freep(&subtitle_to_free->rects[i]->pict.data[1]);
1427 av_freep(&subtitle_to_free->rects[i]);
1429 av_freep(&subtitle_to_free->rects);
1431 subtitle_to_free->num_rects = 0;
1432 subtitle_to_free = NULL;
1439 for(i=0;i<nb_ostreams;i++) {
1441 if (ost->source_index == ist_index) {
1442 AVCodecContext *enc= ost->st->codec;
1443 os = output_files[ost->file_index];
1445 if(ost->st->codec->codec_type == CODEC_TYPE_AUDIO && enc->frame_size <=1)
1447 if(ost->st->codec->codec_type == CODEC_TYPE_VIDEO && (os->oformat->flags & AVFMT_RAWPICTURE))
1450 if (ost->encoding_needed) {
1454 av_init_packet(&pkt);
1455 pkt.stream_index= ost->index;
1457 switch(ost->st->codec->codec_type) {
1458 case CODEC_TYPE_AUDIO:
1459 fifo_bytes = av_fifo_size(ost->fifo);
1461 /* encode any samples remaining in fifo */
1462 if (fifo_bytes > 0) {
1463 int osize = av_get_bits_per_sample_format(enc->sample_fmt) >> 3;
1464 int fs_tmp = enc->frame_size;
1466 av_fifo_generic_read(ost->fifo, samples, fifo_bytes, NULL);
1467 if (enc->codec->capabilities & CODEC_CAP_SMALL_LAST_FRAME) {
1468 enc->frame_size = fifo_bytes / (osize * enc->channels);
1470 int frame_bytes = enc->frame_size*osize*enc->channels;
1471 if (samples_size < frame_bytes)
1473 memset((uint8_t*)samples+fifo_bytes, 0, frame_bytes - fifo_bytes);
1476 ret = avcodec_encode_audio(enc, bit_buffer, bit_buffer_size, samples);
1477 pkt.duration = av_rescale((int64_t)enc->frame_size*ost->st->time_base.den,
1478 ost->st->time_base.num, enc->sample_rate);
1479 enc->frame_size = fs_tmp;
1482 ret = avcodec_encode_audio(enc, bit_buffer, bit_buffer_size, NULL);
1485 fprintf(stderr, "Audio encoding failed\n");
1489 pkt.flags |= PKT_FLAG_KEY;
1491 case CODEC_TYPE_VIDEO:
1492 ret = avcodec_encode_video(enc, bit_buffer, bit_buffer_size, NULL);
1494 fprintf(stderr, "Video encoding failed\n");
1498 if(enc->coded_frame && enc->coded_frame->key_frame)
1499 pkt.flags |= PKT_FLAG_KEY;
1500 if (ost->logfile && enc->stats_out) {
1501 fprintf(ost->logfile, "%s", enc->stats_out);
1510 pkt.data= bit_buffer;
1512 if(enc->coded_frame && enc->coded_frame->pts != AV_NOPTS_VALUE)
1513 pkt.pts= av_rescale_q(enc->coded_frame->pts, enc->time_base, ost->st->time_base);
1514 write_frame(os, &pkt, ost->st->codec, bitstream_filters[ost->file_index][pkt.stream_index]);
1526 static void print_sdp(AVFormatContext **avc, int n)
1530 avf_sdp_create(avc, n, sdp, sizeof(sdp));
1531 printf("SDP:\n%s\n", sdp);
1535 static int stream_index_from_inputs(AVFormatContext **input_files,
1537 AVInputFile *file_table,
1538 AVInputStream **ist_table,
1539 enum CodecType type,
1543 for(z=0; z<nb_input_files; z++) {
1544 AVFormatContext *ic = input_files[z];
1545 for(p=0; p<ic->nb_programs; p++) {
1546 AVProgram *program = ic->programs[p];
1547 if(program->id != programid)
1549 for(q=0; q<program->nb_stream_indexes; q++) {
1550 int sidx = program->stream_index[q];
1551 int ris = file_table[z].ist_index + sidx;
1552 if(ist_table[ris]->discard && ic->streams[sidx]->codec->codec_type == type)
1562 * The following code is the main loop of the file converter
1564 static int av_encode(AVFormatContext **output_files,
1565 int nb_output_files,
1566 AVFormatContext **input_files,
1568 AVStreamMap *stream_maps, int nb_stream_maps)
1570 int ret = 0, i, j, k, n, nb_istreams = 0, nb_ostreams = 0;
1571 AVFormatContext *is, *os;
1572 AVCodecContext *codec, *icodec;
1573 AVOutputStream *ost, **ost_table = NULL;
1574 AVInputStream *ist, **ist_table = NULL;
1575 AVInputFile *file_table;
1579 uint8_t no_packet[MAX_FILES]={0};
1580 int no_packet_count=0;
1582 file_table= av_mallocz(nb_input_files * sizeof(AVInputFile));
1586 /* input stream init */
1588 for(i=0;i<nb_input_files;i++) {
1589 is = input_files[i];
1590 file_table[i].ist_index = j;
1591 file_table[i].nb_streams = is->nb_streams;
1592 j += is->nb_streams;
1596 ist_table = av_mallocz(nb_istreams * sizeof(AVInputStream *));
1600 for(i=0;i<nb_istreams;i++) {
1601 ist = av_mallocz(sizeof(AVInputStream));
1607 for(i=0;i<nb_input_files;i++) {
1608 is = input_files[i];
1609 for(k=0;k<is->nb_streams;k++) {
1610 ist = ist_table[j++];
1611 ist->st = is->streams[k];
1612 ist->file_index = i;
1614 ist->discard = 1; /* the stream is discarded by default
1618 ist->start = av_gettime();
1623 /* output stream init */
1625 for(i=0;i<nb_output_files;i++) {
1626 os = output_files[i];
1627 if (!os->nb_streams) {
1628 dump_format(output_files[i], i, output_files[i]->filename, 1);
1629 fprintf(stderr, "Output file #%d does not contain any stream\n", i);
1632 nb_ostreams += os->nb_streams;
1634 if (nb_stream_maps > 0 && nb_stream_maps != nb_ostreams) {
1635 fprintf(stderr, "Number of stream maps must match number of output streams\n");
1639 /* Sanity check the mapping args -- do the input files & streams exist? */
1640 for(i=0;i<nb_stream_maps;i++) {
1641 int fi = stream_maps[i].file_index;
1642 int si = stream_maps[i].stream_index;
1644 if (fi < 0 || fi > nb_input_files - 1 ||
1645 si < 0 || si > file_table[fi].nb_streams - 1) {
1646 fprintf(stderr,"Could not find input stream #%d.%d\n", fi, si);
1649 fi = stream_maps[i].sync_file_index;
1650 si = stream_maps[i].sync_stream_index;
1651 if (fi < 0 || fi > nb_input_files - 1 ||
1652 si < 0 || si > file_table[fi].nb_streams - 1) {
1653 fprintf(stderr,"Could not find sync stream #%d.%d\n", fi, si);
1658 ost_table = av_mallocz(sizeof(AVOutputStream *) * nb_ostreams);
1661 for(i=0;i<nb_ostreams;i++) {
1662 ost = av_mallocz(sizeof(AVOutputStream));
1669 for(k=0;k<nb_output_files;k++) {
1670 os = output_files[k];
1671 for(i=0;i<os->nb_streams;i++,n++) {
1674 ost->file_index = k;
1676 ost->st = os->streams[i];
1677 if (nb_stream_maps > 0) {
1678 ost->source_index = file_table[stream_maps[n].file_index].ist_index +
1679 stream_maps[n].stream_index;
1681 /* Sanity check that the stream types match */
1682 if (ist_table[ost->source_index]->st->codec->codec_type != ost->st->codec->codec_type) {
1683 int i= ost->file_index;
1684 dump_format(output_files[i], i, output_files[i]->filename, 1);
1685 fprintf(stderr, "Codec type mismatch for mapping #%d.%d -> #%d.%d\n",
1686 stream_maps[n].file_index, stream_maps[n].stream_index,
1687 ost->file_index, ost->index);
1694 j = stream_index_from_inputs(input_files, nb_input_files, file_table, ist_table, ost->st->codec->codec_type, opt_programid);
1696 ost->source_index = j;
1700 /* get corresponding input stream index : we select the first one with the right type */
1702 for(j=0;j<nb_istreams;j++) {
1705 ist->st->codec->codec_type == ost->st->codec->codec_type) {
1706 ost->source_index = j;
1714 if(! opt_programid) {
1715 /* try again and reuse existing stream */
1716 for(j=0;j<nb_istreams;j++) {
1718 if (ist->st->codec->codec_type == ost->st->codec->codec_type) {
1719 ost->source_index = j;
1725 int i= ost->file_index;
1726 dump_format(output_files[i], i, output_files[i]->filename, 1);
1727 fprintf(stderr, "Could not find input stream matching output stream #%d.%d\n",
1728 ost->file_index, ost->index);
1733 ist = ist_table[ost->source_index];
1735 ost->sync_ist = (nb_stream_maps > 0) ?
1736 ist_table[file_table[stream_maps[n].sync_file_index].ist_index +
1737 stream_maps[n].sync_stream_index] : ist;
1741 /* for each output stream, we compute the right encoding parameters */
1742 for(i=0;i<nb_ostreams;i++) {
1743 AVMetadataTag *lang;
1745 os = output_files[ost->file_index];
1746 ist = ist_table[ost->source_index];
1748 codec = ost->st->codec;
1749 icodec = ist->st->codec;
1751 if ((lang=av_metadata_get(ist->st->metadata, "language", NULL, 0))
1752 && !av_metadata_get(ost->st->metadata, "language", NULL, 0))
1753 av_metadata_set(&ost->st->metadata, "language", lang->value);
1755 ost->st->disposition = ist->st->disposition;
1756 codec->bits_per_raw_sample= icodec->bits_per_raw_sample;
1757 codec->chroma_sample_location = icodec->chroma_sample_location;
1759 if (ost->st->stream_copy) {
1760 /* if stream_copy is selected, no need to decode or encode */
1761 codec->codec_id = icodec->codec_id;
1762 codec->codec_type = icodec->codec_type;
1764 if(!codec->codec_tag){
1765 if( !os->oformat->codec_tag
1766 || av_codec_get_id (os->oformat->codec_tag, icodec->codec_tag) > 0
1767 || av_codec_get_tag(os->oformat->codec_tag, icodec->codec_id) <= 0)
1768 codec->codec_tag = icodec->codec_tag;
1771 codec->bit_rate = icodec->bit_rate;
1772 codec->extradata= icodec->extradata;
1773 codec->extradata_size= icodec->extradata_size;
1774 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){
1775 codec->time_base = icodec->time_base;
1776 codec->time_base.num *= icodec->ticks_per_frame;
1778 codec->time_base = ist->st->time_base;
1779 switch(codec->codec_type) {
1780 case CODEC_TYPE_AUDIO:
1781 if(audio_volume != 256) {
1782 fprintf(stderr,"-acodec copy and -vol are incompatible (frames are not decoded)\n");
1785 codec->channel_layout = icodec->channel_layout;
1786 codec->sample_rate = icodec->sample_rate;
1787 codec->channels = icodec->channels;
1788 codec->frame_size = icodec->frame_size;
1789 codec->block_align= icodec->block_align;
1790 if(codec->block_align == 1 && codec->codec_id == CODEC_ID_MP3)
1791 codec->block_align= 0;
1792 if(codec->codec_id == CODEC_ID_AC3)
1793 codec->block_align= 0;
1795 case CODEC_TYPE_VIDEO:
1796 codec->pix_fmt = icodec->pix_fmt;
1797 codec->width = icodec->width;
1798 codec->height = icodec->height;
1799 codec->has_b_frames = icodec->has_b_frames;
1801 case CODEC_TYPE_SUBTITLE:
1802 codec->width = icodec->width;
1803 codec->height = icodec->height;
1809 switch(codec->codec_type) {
1810 case CODEC_TYPE_AUDIO:
1811 ost->fifo= av_fifo_alloc(1024);
1814 ost->reformat_pair = MAKE_SFMT_PAIR(SAMPLE_FMT_NONE,SAMPLE_FMT_NONE);
1815 ost->audio_resample = codec->sample_rate != icodec->sample_rate || audio_sync_method > 1;
1816 icodec->request_channels = codec->channels;
1817 ist->decoding_needed = 1;
1818 ost->encoding_needed = 1;
1820 case CODEC_TYPE_VIDEO:
1821 if (ost->st->codec->pix_fmt == PIX_FMT_NONE) {
1822 fprintf(stderr, "Video pixel format is unknown, stream cannot be decoded\n");
1825 ost->video_crop = ((frame_leftBand + frame_rightBand + frame_topBand + frame_bottomBand) != 0);
1826 ost->video_pad = ((frame_padleft + frame_padright + frame_padtop + frame_padbottom) != 0);
1827 ost->video_resample = ((codec->width != icodec->width -
1828 (frame_leftBand + frame_rightBand) +
1829 (frame_padleft + frame_padright)) ||
1830 (codec->height != icodec->height -
1831 (frame_topBand + frame_bottomBand) +
1832 (frame_padtop + frame_padbottom)) ||
1833 (codec->pix_fmt != icodec->pix_fmt));
1834 if (ost->video_crop) {
1835 ost->topBand = frame_topBand;
1836 ost->leftBand = frame_leftBand;
1838 if (ost->video_pad) {
1839 ost->padtop = frame_padtop;
1840 ost->padleft = frame_padleft;
1841 ost->padbottom = frame_padbottom;
1842 ost->padright = frame_padright;
1843 if (!ost->video_resample) {
1844 avcodec_get_frame_defaults(&ost->pict_tmp);
1845 if(avpicture_alloc((AVPicture*)&ost->pict_tmp, codec->pix_fmt,
1846 codec->width, codec->height))
1850 if (ost->video_resample) {
1851 avcodec_get_frame_defaults(&ost->pict_tmp);
1852 if(avpicture_alloc((AVPicture*)&ost->pict_tmp, codec->pix_fmt,
1853 codec->width, codec->height)) {
1854 fprintf(stderr, "Cannot allocate temp picture, check pix fmt\n");
1857 sws_flags = av_get_int(sws_opts, "sws_flags", NULL);
1858 ost->img_resample_ctx = sws_getContext(
1859 icodec->width - (frame_leftBand + frame_rightBand),
1860 icodec->height - (frame_topBand + frame_bottomBand),
1862 codec->width - (frame_padleft + frame_padright),
1863 codec->height - (frame_padtop + frame_padbottom),
1865 sws_flags, NULL, NULL, NULL);
1866 if (ost->img_resample_ctx == NULL) {
1867 fprintf(stderr, "Cannot get resampling context\n");
1870 ost->resample_height = icodec->height - (frame_topBand + frame_bottomBand);
1871 codec->bits_per_raw_sample= 0;
1873 ost->encoding_needed = 1;
1874 ist->decoding_needed = 1;
1876 case CODEC_TYPE_SUBTITLE:
1877 ost->encoding_needed = 1;
1878 ist->decoding_needed = 1;
1885 if (ost->encoding_needed &&
1886 (codec->flags & (CODEC_FLAG_PASS1 | CODEC_FLAG_PASS2))) {
1887 char logfilename[1024];
1892 snprintf(logfilename, sizeof(logfilename), "%s-%d.log",
1893 pass_logfilename_prefix ? pass_logfilename_prefix : DEFAULT_PASS_LOGFILENAME_PREFIX,
1895 if (codec->flags & CODEC_FLAG_PASS1) {
1896 f = fopen(logfilename, "w");
1898 fprintf(stderr, "Cannot write log file '%s' for pass-1 encoding: %s\n", logfilename, strerror(errno));
1903 /* read the log file */
1904 f = fopen(logfilename, "r");
1906 fprintf(stderr, "Cannot read log file '%s' for pass-2 encoding: %s\n", logfilename, strerror(errno));
1909 fseek(f, 0, SEEK_END);
1911 fseek(f, 0, SEEK_SET);
1912 logbuffer = av_malloc(size + 1);
1914 fprintf(stderr, "Could not allocate log buffer\n");
1917 size = fread(logbuffer, 1, size, f);
1919 logbuffer[size] = '\0';
1920 codec->stats_in = logbuffer;
1924 if(codec->codec_type == CODEC_TYPE_VIDEO){
1925 int size= codec->width * codec->height;
1926 bit_buffer_size= FFMAX(bit_buffer_size, 6*size + 200);
1931 bit_buffer = av_malloc(bit_buffer_size);
1933 fprintf(stderr, "Cannot allocate %d bytes output buffer\n",
1935 ret = AVERROR(ENOMEM);
1939 /* open each encoder */
1940 for(i=0;i<nb_ostreams;i++) {
1942 if (ost->encoding_needed) {
1943 AVCodec *codec = output_codecs[i];
1945 codec = avcodec_find_encoder(ost->st->codec->codec_id);
1947 snprintf(error, sizeof(error), "Unsupported codec for output stream #%d.%d",
1948 ost->file_index, ost->index);
1949 ret = AVERROR(EINVAL);
1952 if (avcodec_open(ost->st->codec, codec) < 0) {
1953 snprintf(error, sizeof(error), "Error while opening codec for output stream #%d.%d - maybe incorrect parameters such as bit_rate, rate, width or height",
1954 ost->file_index, ost->index);
1955 ret = AVERROR(EINVAL);
1958 extra_size += ost->st->codec->extradata_size;
1962 /* open each decoder */
1963 for(i=0;i<nb_istreams;i++) {
1965 if (ist->decoding_needed) {
1966 AVCodec *codec = input_codecs[i];
1968 codec = avcodec_find_decoder(ist->st->codec->codec_id);
1970 snprintf(error, sizeof(error), "Unsupported codec (id=%d) for input stream #%d.%d",
1971 ist->st->codec->codec_id, ist->file_index, ist->index);
1972 ret = AVERROR(EINVAL);
1975 if (avcodec_open(ist->st->codec, codec) < 0) {
1976 snprintf(error, sizeof(error), "Error while opening codec for input stream #%d.%d",
1977 ist->file_index, ist->index);
1978 ret = AVERROR(EINVAL);
1981 //if (ist->st->codec->codec_type == CODEC_TYPE_VIDEO)
1982 // ist->st->codec->flags |= CODEC_FLAG_REPEAT_FIELD;
1987 for(i=0;i<nb_istreams;i++) {
1990 ist->next_pts = AV_NOPTS_VALUE;
1994 /* set meta data information from input file if required */
1995 for (i=0;i<nb_meta_data_maps;i++) {
1996 AVFormatContext *out_file;
1997 AVFormatContext *in_file;
1998 AVMetadataTag *mtag;
2000 int out_file_index = meta_data_maps[i].out_file;
2001 int in_file_index = meta_data_maps[i].in_file;
2002 if (out_file_index < 0 || out_file_index >= nb_output_files) {
2003 snprintf(error, sizeof(error), "Invalid output file index %d map_meta_data(%d,%d)",
2004 out_file_index, out_file_index, in_file_index);
2005 ret = AVERROR(EINVAL);
2008 if (in_file_index < 0 || in_file_index >= nb_input_files) {
2009 snprintf(error, sizeof(error), "Invalid input file index %d map_meta_data(%d,%d)",
2010 in_file_index, out_file_index, in_file_index);
2011 ret = AVERROR(EINVAL);
2015 out_file = output_files[out_file_index];
2016 in_file = input_files[in_file_index];
2020 while((mtag=av_metadata_get(in_file->metadata, "", mtag, AV_METADATA_IGNORE_SUFFIX)))
2021 av_metadata_set(&out_file->metadata, mtag->key, mtag->value);
2022 av_metadata_conv(out_file, out_file->oformat->metadata_conv,
2023 in_file->iformat->metadata_conv);
2026 /* open files and write file headers */
2027 for(i=0;i<nb_output_files;i++) {
2028 os = output_files[i];
2029 if (av_write_header(os) < 0) {
2030 snprintf(error, sizeof(error), "Could not write header for output file #%d (incorrect codec parameters ?)", i);
2031 ret = AVERROR(EINVAL);
2034 if (strcmp(output_files[i]->oformat->name, "rtp")) {
2040 /* dump the file output parameters - cannot be done before in case
2042 for(i=0;i<nb_output_files;i++) {
2043 dump_format(output_files[i], i, output_files[i]->filename, 1);
2046 /* dump the stream mapping */
2048 fprintf(stderr, "Stream mapping:\n");
2049 for(i=0;i<nb_ostreams;i++) {
2051 fprintf(stderr, " Stream #%d.%d -> #%d.%d",
2052 ist_table[ost->source_index]->file_index,
2053 ist_table[ost->source_index]->index,
2056 if (ost->sync_ist != ist_table[ost->source_index])
2057 fprintf(stderr, " [sync #%d.%d]",
2058 ost->sync_ist->file_index,
2059 ost->sync_ist->index);
2060 fprintf(stderr, "\n");
2065 fprintf(stderr, "%s\n", error);
2070 print_sdp(output_files, nb_output_files);
2073 if (!using_stdin && verbose >= 0) {
2074 fprintf(stderr, "Press [q] to stop encoding\n");
2075 url_set_interrupt_cb(decode_interrupt_cb);
2079 timer_start = av_gettime();
2081 for(; received_sigterm == 0;) {
2082 int file_index, ist_index;
2090 /* if 'q' pressed, exits */
2094 /* read_key() returns 0 on EOF */
2100 /* select the stream that we must read now by looking at the
2101 smallest output pts */
2103 for(i=0;i<nb_ostreams;i++) {
2106 os = output_files[ost->file_index];
2107 ist = ist_table[ost->source_index];
2108 if(no_packet[ist->file_index])
2110 if(ost->st->codec->codec_type == CODEC_TYPE_VIDEO)
2111 opts = ost->sync_opts * av_q2d(ost->st->codec->time_base);
2113 opts = ost->st->pts.val * av_q2d(ost->st->time_base);
2114 ipts = (double)ist->pts;
2115 if (!file_table[ist->file_index].eof_reached){
2116 if(ipts < ipts_min) {
2118 if(input_sync ) file_index = ist->file_index;
2120 if(opts < opts_min) {
2122 if(!input_sync) file_index = ist->file_index;
2125 if(ost->frame_number >= max_frames[ost->st->codec->codec_type]){
2130 /* if none, if is finished */
2131 if (file_index < 0) {
2132 if(no_packet_count){
2134 memset(no_packet, 0, sizeof(no_packet));
2141 /* finish if recording time exhausted */
2142 if (opts_min >= (recording_time / 1000000.0))
2145 /* finish if limit size exhausted */
2146 if (limit_filesize != 0 && limit_filesize < url_ftell(output_files[0]->pb))
2149 /* read a frame from it and output it in the fifo */
2150 is = input_files[file_index];
2151 ret= av_read_frame(is, &pkt);
2152 if(ret == AVERROR(EAGAIN)){
2153 no_packet[file_index]=1;
2158 file_table[file_index].eof_reached = 1;
2166 memset(no_packet, 0, sizeof(no_packet));
2169 av_pkt_dump_log(NULL, AV_LOG_DEBUG, &pkt, do_hex_dump);
2171 /* the following test is needed in case new streams appear
2172 dynamically in stream : we ignore them */
2173 if (pkt.stream_index >= file_table[file_index].nb_streams)
2174 goto discard_packet;
2175 ist_index = file_table[file_index].ist_index + pkt.stream_index;
2176 ist = ist_table[ist_index];
2178 goto discard_packet;
2180 if (pkt.dts != AV_NOPTS_VALUE)
2181 pkt.dts += av_rescale_q(input_files_ts_offset[ist->file_index], AV_TIME_BASE_Q, ist->st->time_base);
2182 if (pkt.pts != AV_NOPTS_VALUE)
2183 pkt.pts += av_rescale_q(input_files_ts_offset[ist->file_index], AV_TIME_BASE_Q, ist->st->time_base);
2185 if(input_files_ts_scale[file_index][pkt.stream_index]){
2186 if(pkt.pts != AV_NOPTS_VALUE)
2187 pkt.pts *= input_files_ts_scale[file_index][pkt.stream_index];
2188 if(pkt.dts != AV_NOPTS_VALUE)
2189 pkt.dts *= input_files_ts_scale[file_index][pkt.stream_index];
2192 // 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);
2193 if (pkt.dts != AV_NOPTS_VALUE && ist->next_pts != AV_NOPTS_VALUE
2194 && (is->iformat->flags & AVFMT_TS_DISCONT)) {
2195 int64_t pkt_dts= av_rescale_q(pkt.dts, ist->st->time_base, AV_TIME_BASE_Q);
2196 int64_t delta= pkt_dts - ist->next_pts;
2197 if((FFABS(delta) > 1LL*dts_delta_threshold*AV_TIME_BASE || pkt_dts+1<ist->pts)&& !copy_ts){
2198 input_files_ts_offset[ist->file_index]-= delta;
2200 fprintf(stderr, "timestamp discontinuity %"PRId64", new offset= %"PRId64"\n", delta, input_files_ts_offset[ist->file_index]);
2201 pkt.dts-= av_rescale_q(delta, AV_TIME_BASE_Q, ist->st->time_base);
2202 if(pkt.pts != AV_NOPTS_VALUE)
2203 pkt.pts-= av_rescale_q(delta, AV_TIME_BASE_Q, ist->st->time_base);
2207 //fprintf(stderr,"read #%d.%d size=%d\n", ist->file_index, ist->index, pkt.size);
2208 if (output_packet(ist, ist_index, ost_table, nb_ostreams, &pkt) < 0) {
2211 fprintf(stderr, "Error while decoding stream #%d.%d\n",
2212 ist->file_index, ist->index);
2215 av_free_packet(&pkt);
2220 av_free_packet(&pkt);
2222 /* dump report by using the output first video and audio streams */
2223 print_report(output_files, ost_table, nb_ostreams, 0);
2226 /* at the end of stream, we must flush the decoder buffers */
2227 for(i=0;i<nb_istreams;i++) {
2229 if (ist->decoding_needed) {
2230 output_packet(ist, i, ost_table, nb_ostreams, NULL);
2236 /* write the trailer if needed and close file */
2237 for(i=0;i<nb_output_files;i++) {
2238 os = output_files[i];
2239 av_write_trailer(os);
2242 /* dump report by using the first video and audio streams */
2243 print_report(output_files, ost_table, nb_ostreams, 1);
2245 /* close each encoder */
2246 for(i=0;i<nb_ostreams;i++) {
2248 if (ost->encoding_needed) {
2249 av_freep(&ost->st->codec->stats_in);
2250 avcodec_close(ost->st->codec);
2254 /* close each decoder */
2255 for(i=0;i<nb_istreams;i++) {
2257 if (ist->decoding_needed) {
2258 avcodec_close(ist->st->codec);
2266 av_freep(&bit_buffer);
2267 av_free(file_table);
2270 for(i=0;i<nb_istreams;i++) {
2277 for(i=0;i<nb_ostreams;i++) {
2281 fclose(ost->logfile);
2282 ost->logfile = NULL;
2284 av_fifo_free(ost->fifo); /* works even if fifo is not
2285 initialized but set to zero */
2286 av_free(ost->pict_tmp.data[0]);
2287 if (ost->video_resample)
2288 sws_freeContext(ost->img_resample_ctx);
2290 audio_resample_close(ost->resample);
2291 if (ost->reformat_ctx)
2292 av_audio_convert_free(ost->reformat_ctx);
2302 int file_read(const char *filename)
2305 unsigned char buffer[1024];
2308 if (url_open(&h, filename, O_RDONLY) < 0) {
2309 printf("could not open '%s'\n", filename);
2313 len = url_read(h, buffer, sizeof(buffer));
2316 for(i=0;i<len;i++) putchar(buffer[i]);
2323 static void opt_format(const char *arg)
2325 /* compatibility stuff for pgmyuv */
2326 if (!strcmp(arg, "pgmyuv")) {
2327 pgmyuv_compatibility_hack=1;
2328 // opt_image_format(arg);
2330 fprintf(stderr, "pgmyuv format is deprecated, use image2\n");
2333 file_iformat = av_find_input_format(arg);
2334 file_oformat = guess_format(arg, NULL, NULL);
2335 if (!file_iformat && !file_oformat) {
2336 fprintf(stderr, "Unknown input or output format: %s\n", arg);
2341 static void opt_video_rc_override_string(const char *arg)
2343 video_rc_override_string = arg;
2346 static int opt_me_threshold(const char *opt, const char *arg)
2348 me_threshold = parse_number_or_die(opt, arg, OPT_INT64, INT_MIN, INT_MAX);
2352 static int opt_loglevel(const char *opt, const char *arg)
2354 int level = parse_number_or_die(opt, arg, OPT_INT, INT_MIN, INT_MAX);
2355 av_log_set_level(level);
2359 static int opt_verbose(const char *opt, const char *arg)
2361 verbose = parse_number_or_die(opt, arg, OPT_INT64, -10, 10);
2365 static int opt_frame_rate(const char *opt, const char *arg)
2367 if (av_parse_video_frame_rate(&frame_rate, arg) < 0) {
2368 fprintf(stderr, "Incorrect value for %s: %s\n", opt, arg);
2374 static int opt_bitrate(const char *opt, const char *arg)
2376 int codec_type = opt[0]=='a' ? CODEC_TYPE_AUDIO : CODEC_TYPE_VIDEO;
2378 opt_default(opt, arg);
2380 if (av_get_int(avcodec_opts[codec_type], "b", NULL) < 1000)
2381 fprintf(stderr, "WARNING: The bitrate parameter is set too low. It takes bits/s as argument, not kbits/s\n");
2386 static void opt_frame_crop_top(const char *arg)
2388 frame_topBand = atoi(arg);
2389 if (frame_topBand < 0) {
2390 fprintf(stderr, "Incorrect top crop size\n");
2393 if ((frame_topBand) >= frame_height){
2394 fprintf(stderr, "Vertical crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");
2397 frame_height -= frame_topBand;
2400 static void opt_frame_crop_bottom(const char *arg)
2402 frame_bottomBand = atoi(arg);
2403 if (frame_bottomBand < 0) {
2404 fprintf(stderr, "Incorrect bottom crop size\n");
2407 if ((frame_bottomBand) >= frame_height){
2408 fprintf(stderr, "Vertical crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");
2411 frame_height -= frame_bottomBand;
2414 static void opt_frame_crop_left(const char *arg)
2416 frame_leftBand = atoi(arg);
2417 if (frame_leftBand < 0) {
2418 fprintf(stderr, "Incorrect left crop size\n");
2421 if ((frame_leftBand) >= frame_width){
2422 fprintf(stderr, "Horizontal crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");
2425 frame_width -= frame_leftBand;
2428 static void opt_frame_crop_right(const char *arg)
2430 frame_rightBand = atoi(arg);
2431 if (frame_rightBand < 0) {
2432 fprintf(stderr, "Incorrect right crop size\n");
2435 if ((frame_rightBand) >= frame_width){
2436 fprintf(stderr, "Horizontal crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");
2439 frame_width -= frame_rightBand;
2442 static void opt_frame_size(const char *arg)
2444 if (av_parse_video_frame_size(&frame_width, &frame_height, arg) < 0) {
2445 fprintf(stderr, "Incorrect frame size\n");
2448 if ((frame_width % 2) != 0 || (frame_height % 2) != 0) {
2449 fprintf(stderr, "Frame size must be a multiple of 2\n");
2454 static void opt_pad_color(const char *arg) {
2455 /* Input is expected to be six hex digits similar to
2456 how colors are expressed in html tags (but without the #) */
2457 int rgb = strtol(arg, NULL, 16);
2461 g = ((rgb >> 8) & 255);
2464 padcolor[0] = RGB_TO_Y(r,g,b);
2465 padcolor[1] = RGB_TO_U(r,g,b,0);
2466 padcolor[2] = RGB_TO_V(r,g,b,0);
2469 static void opt_frame_pad_top(const char *arg)
2471 frame_padtop = atoi(arg);
2472 if (frame_padtop < 0) {
2473 fprintf(stderr, "Incorrect top pad size\n");
2478 static void opt_frame_pad_bottom(const char *arg)
2480 frame_padbottom = atoi(arg);
2481 if (frame_padbottom < 0) {
2482 fprintf(stderr, "Incorrect bottom pad size\n");
2488 static void opt_frame_pad_left(const char *arg)
2490 frame_padleft = atoi(arg);
2491 if (frame_padleft < 0) {
2492 fprintf(stderr, "Incorrect left pad size\n");
2498 static void opt_frame_pad_right(const char *arg)
2500 frame_padright = atoi(arg);
2501 if (frame_padright < 0) {
2502 fprintf(stderr, "Incorrect right pad size\n");
2507 static void list_fmts(void (*get_fmt_string)(char *buf, int buf_size, int fmt), int nb_fmts)
2511 for (i=-1; i < nb_fmts; i++) {
2512 get_fmt_string (fmt_str, sizeof(fmt_str), i);
2513 fprintf(stdout, "%s\n", fmt_str);
2517 static void opt_frame_pix_fmt(const char *arg)
2519 if (strcmp(arg, "list")) {
2520 frame_pix_fmt = avcodec_get_pix_fmt(arg);
2521 if (frame_pix_fmt == PIX_FMT_NONE) {
2522 fprintf(stderr, "Unknown pixel format requested: %s\n", arg);
2526 list_fmts(avcodec_pix_fmt_string, PIX_FMT_NB);
2531 static void opt_frame_aspect_ratio(const char *arg)
2538 p = strchr(arg, ':');
2540 x = strtol(arg, &end, 10);
2542 y = strtol(end+1, &end, 10);
2544 ar = (double)x / (double)y;
2546 ar = strtod(arg, NULL);
2549 fprintf(stderr, "Incorrect aspect ratio specification.\n");
2552 frame_aspect_ratio = ar;
2555 static int opt_metadata(const char *opt, const char *arg)
2557 char *mid= strchr(arg, '=');
2560 fprintf(stderr, "Missing =\n");
2566 metadata= av_realloc(metadata, sizeof(*metadata)*metadata_count);
2567 metadata[metadata_count-1].key = av_strdup(arg);
2568 metadata[metadata_count-1].value= av_strdup(mid);
2573 static void opt_qscale(const char *arg)
2575 video_qscale = atof(arg);
2576 if (video_qscale <= 0 ||
2577 video_qscale > 255) {
2578 fprintf(stderr, "qscale must be > 0.0 and <= 255\n");
2583 static void opt_top_field_first(const char *arg)
2585 top_field_first= atoi(arg);
2588 static int opt_thread_count(const char *opt, const char *arg)
2590 thread_count= parse_number_or_die(opt, arg, OPT_INT64, 0, INT_MAX);
2593 fprintf(stderr, "Warning: not compiled with thread support, using thread emulation\n");
2598 static void opt_audio_sample_fmt(const char *arg)
2600 if (strcmp(arg, "list"))
2601 audio_sample_fmt = avcodec_get_sample_fmt(arg);
2603 list_fmts(avcodec_sample_fmt_string, SAMPLE_FMT_NB);
2608 static int opt_audio_rate(const char *opt, const char *arg)
2610 audio_sample_rate = parse_number_or_die(opt, arg, OPT_INT64, 0, INT_MAX);
2614 static int opt_audio_channels(const char *opt, const char *arg)
2616 audio_channels = parse_number_or_die(opt, arg, OPT_INT64, 0, INT_MAX);
2620 static void opt_video_channel(const char *arg)
2622 video_channel = strtol(arg, NULL, 0);
2625 static void opt_video_standard(const char *arg)
2627 video_standard = av_strdup(arg);
2630 static void opt_codec(int *pstream_copy, char **pcodec_name,
2631 int codec_type, const char *arg)
2633 av_freep(pcodec_name);
2634 if (!strcmp(arg, "copy")) {
2637 *pcodec_name = av_strdup(arg);
2641 static void opt_audio_codec(const char *arg)
2643 opt_codec(&audio_stream_copy, &audio_codec_name, CODEC_TYPE_AUDIO, arg);
2646 static void opt_audio_tag(const char *arg)
2649 audio_codec_tag= strtol(arg, &tail, 0);
2652 audio_codec_tag= arg[0] + (arg[1]<<8) + (arg[2]<<16) + (arg[3]<<24);
2655 static void opt_video_tag(const char *arg)
2658 video_codec_tag= strtol(arg, &tail, 0);
2661 video_codec_tag= arg[0] + (arg[1]<<8) + (arg[2]<<16) + (arg[3]<<24);
2664 static void opt_video_codec(const char *arg)
2666 opt_codec(&video_stream_copy, &video_codec_name, CODEC_TYPE_VIDEO, arg);
2669 static void opt_subtitle_codec(const char *arg)
2671 opt_codec(&subtitle_stream_copy, &subtitle_codec_name, CODEC_TYPE_SUBTITLE, arg);
2674 static void opt_subtitle_tag(const char *arg)
2677 subtitle_codec_tag= strtol(arg, &tail, 0);
2680 subtitle_codec_tag= arg[0] + (arg[1]<<8) + (arg[2]<<16) + (arg[3]<<24);
2683 static void opt_map(const char *arg)
2688 m = &stream_maps[nb_stream_maps++];
2690 m->file_index = strtol(arg, &p, 0);
2694 m->stream_index = strtol(p, &p, 0);
2697 m->sync_file_index = strtol(p, &p, 0);
2700 m->sync_stream_index = strtol(p, &p, 0);
2702 m->sync_file_index = m->file_index;
2703 m->sync_stream_index = m->stream_index;
2707 static void opt_map_meta_data(const char *arg)
2712 m = &meta_data_maps[nb_meta_data_maps++];
2714 m->out_file = strtol(arg, &p, 0);
2718 m->in_file = strtol(p, &p, 0);
2721 static void opt_input_ts_scale(const char *arg)
2723 unsigned int stream;
2727 stream = strtol(arg, &p, 0);
2730 scale= strtod(p, &p);
2732 if(stream >= MAX_STREAMS)
2735 input_files_ts_scale[nb_input_files][stream]= scale;
2738 static int opt_recording_time(const char *opt, const char *arg)
2740 recording_time = parse_time_or_die(opt, arg, 1);
2744 static int opt_start_time(const char *opt, const char *arg)
2746 start_time = parse_time_or_die(opt, arg, 1);
2750 static int opt_rec_timestamp(const char *opt, const char *arg)
2752 rec_timestamp = parse_time_or_die(opt, arg, 0) / 1000000;
2756 static int opt_input_ts_offset(const char *opt, const char *arg)
2758 input_ts_offset = parse_time_or_die(opt, arg, 1);
2762 static enum CodecID find_codec_or_die(const char *name, int type, int encoder)
2764 const char *codec_string = encoder ? "encoder" : "decoder";
2768 return CODEC_ID_NONE;
2770 avcodec_find_encoder_by_name(name) :
2771 avcodec_find_decoder_by_name(name);
2773 fprintf(stderr, "Unknown %s '%s'\n", codec_string, name);
2776 if(codec->type != type) {
2777 fprintf(stderr, "Invalid %s type '%s'\n", codec_string, name);
2783 static void opt_input_file(const char *filename)
2785 AVFormatContext *ic;
2786 AVFormatParameters params, *ap = ¶ms;
2787 int err, i, ret, rfps, rfps_base;
2790 if (!strcmp(filename, "-"))
2793 using_stdin |= !strncmp(filename, "pipe:", 5) ||
2794 !strcmp(filename, "/dev/stdin");
2796 /* get default parameters from command line */
2797 ic = avformat_alloc_context();
2799 memset(ap, 0, sizeof(*ap));
2800 ap->prealloced_context = 1;
2801 ap->sample_rate = audio_sample_rate;
2802 ap->channels = audio_channels;
2803 ap->time_base.den = frame_rate.num;
2804 ap->time_base.num = frame_rate.den;
2805 ap->width = frame_width + frame_padleft + frame_padright;
2806 ap->height = frame_height + frame_padtop + frame_padbottom;
2807 ap->pix_fmt = frame_pix_fmt;
2808 // ap->sample_fmt = audio_sample_fmt; //FIXME:not implemented in libavformat
2809 ap->channel = video_channel;
2810 ap->standard = video_standard;
2811 ap->video_codec_id = find_codec_or_die(video_codec_name, CODEC_TYPE_VIDEO, 0);
2812 ap->audio_codec_id = find_codec_or_die(audio_codec_name, CODEC_TYPE_AUDIO, 0);
2813 if(pgmyuv_compatibility_hack)
2814 ap->video_codec_id= CODEC_ID_PGMYUV;
2816 set_context_opts(ic, avformat_opts, AV_OPT_FLAG_DECODING_PARAM);
2818 ic->video_codec_id = find_codec_or_die(video_codec_name , CODEC_TYPE_VIDEO , 0);
2819 ic->audio_codec_id = find_codec_or_die(audio_codec_name , CODEC_TYPE_AUDIO , 0);
2820 ic->subtitle_codec_id= find_codec_or_die(subtitle_codec_name, CODEC_TYPE_SUBTITLE, 0);
2821 ic->flags |= AVFMT_FLAG_NONBLOCK;
2823 /* open the input file with generic libav function */
2824 err = av_open_input_file(&ic, filename, file_iformat, 0, ap);
2826 print_error(filename, err);
2831 for(i=0; i<ic->nb_programs; i++)
2832 if(ic->programs[i]->id != opt_programid)
2833 ic->programs[i]->discard = AVDISCARD_ALL;
2836 ic->loop_input = loop_input;
2838 /* If not enough info to get the stream parameters, we decode the
2839 first frames to get it. (used in mpeg case for example) */
2840 ret = av_find_stream_info(ic);
2841 if (ret < 0 && verbose >= 0) {
2842 fprintf(stderr, "%s: could not find codec parameters\n", filename);
2846 timestamp = start_time;
2847 /* add the stream start time */
2848 if (ic->start_time != AV_NOPTS_VALUE)
2849 timestamp += ic->start_time;
2851 /* if seeking requested, we execute it */
2852 if (start_time != 0) {
2853 ret = av_seek_frame(ic, -1, timestamp, AVSEEK_FLAG_BACKWARD);
2855 fprintf(stderr, "%s: could not seek to position %0.3f\n",
2856 filename, (double)timestamp / AV_TIME_BASE);
2858 /* reset seek info */
2862 /* update the current parameters so that they match the one of the input stream */
2863 for(i=0;i<ic->nb_streams;i++) {
2864 AVCodecContext *enc = ic->streams[i]->codec;
2866 avcodec_thread_init(enc, thread_count);
2867 enc->thread_count= thread_count;
2868 switch(enc->codec_type) {
2869 case CODEC_TYPE_AUDIO:
2870 set_context_opts(enc, avcodec_opts[CODEC_TYPE_AUDIO], AV_OPT_FLAG_AUDIO_PARAM | AV_OPT_FLAG_DECODING_PARAM);
2871 //fprintf(stderr, "\nInput Audio channels: %d", enc->channels);
2872 channel_layout = enc->channel_layout;
2873 audio_channels = enc->channels;
2874 audio_sample_rate = enc->sample_rate;
2875 audio_sample_fmt = enc->sample_fmt;
2876 input_codecs[nb_icodecs++] = avcodec_find_decoder_by_name(audio_codec_name);
2878 ic->streams[i]->discard= AVDISCARD_ALL;
2880 case CODEC_TYPE_VIDEO:
2881 set_context_opts(enc, avcodec_opts[CODEC_TYPE_VIDEO], AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_DECODING_PARAM);
2882 frame_height = enc->height;
2883 frame_width = enc->width;
2884 if(ic->streams[i]->sample_aspect_ratio.num)
2885 frame_aspect_ratio=av_q2d(ic->streams[i]->sample_aspect_ratio);
2887 frame_aspect_ratio=av_q2d(enc->sample_aspect_ratio);
2888 frame_aspect_ratio *= (float) enc->width / enc->height;
2889 frame_pix_fmt = enc->pix_fmt;
2890 rfps = ic->streams[i]->r_frame_rate.num;
2891 rfps_base = ic->streams[i]->r_frame_rate.den;
2892 if(enc->lowres) enc->flags |= CODEC_FLAG_EMU_EDGE;
2894 enc->debug |= FF_DEBUG_MV;
2896 if (enc->time_base.den != rfps || enc->time_base.num != rfps_base) {
2899 fprintf(stderr,"\nSeems stream %d codec frame rate differs from container frame rate: %2.2f (%d/%d) -> %2.2f (%d/%d)\n",
2900 i, (float)enc->time_base.den / enc->time_base.num, enc->time_base.den, enc->time_base.num,
2902 (float)rfps / rfps_base, rfps, rfps_base);
2904 /* update the current frame rate to match the stream frame rate */
2905 frame_rate.num = rfps;
2906 frame_rate.den = rfps_base;
2908 input_codecs[nb_icodecs++] = avcodec_find_decoder_by_name(video_codec_name);
2910 ic->streams[i]->discard= AVDISCARD_ALL;
2911 else if(video_discard)
2912 ic->streams[i]->discard= video_discard;
2914 case CODEC_TYPE_DATA:
2916 case CODEC_TYPE_SUBTITLE:
2917 input_codecs[nb_icodecs++] = avcodec_find_decoder_by_name(subtitle_codec_name);
2918 if(subtitle_disable)
2919 ic->streams[i]->discard = AVDISCARD_ALL;
2921 case CODEC_TYPE_ATTACHMENT:
2922 case CODEC_TYPE_UNKNOWN:
2930 input_files[nb_input_files] = ic;
2931 input_files_ts_offset[nb_input_files] = input_ts_offset - (copy_ts ? 0 : timestamp);
2932 /* dump the file content */
2934 dump_format(ic, nb_input_files, filename, 0);
2937 file_iformat = NULL;
2938 file_oformat = NULL;
2942 av_freep(&video_codec_name);
2943 av_freep(&audio_codec_name);
2944 av_freep(&subtitle_codec_name);
2947 static void check_audio_video_sub_inputs(int *has_video_ptr, int *has_audio_ptr,
2948 int *has_subtitle_ptr)
2950 int has_video, has_audio, has_subtitle, i, j;
2951 AVFormatContext *ic;
2956 for(j=0;j<nb_input_files;j++) {
2957 ic = input_files[j];
2958 for(i=0;i<ic->nb_streams;i++) {
2959 AVCodecContext *enc = ic->streams[i]->codec;
2960 switch(enc->codec_type) {
2961 case CODEC_TYPE_AUDIO:
2964 case CODEC_TYPE_VIDEO:
2967 case CODEC_TYPE_SUBTITLE:
2970 case CODEC_TYPE_DATA:
2971 case CODEC_TYPE_ATTACHMENT:
2972 case CODEC_TYPE_UNKNOWN:
2979 *has_video_ptr = has_video;
2980 *has_audio_ptr = has_audio;
2981 *has_subtitle_ptr = has_subtitle;
2984 static void new_video_stream(AVFormatContext *oc)
2987 AVCodecContext *video_enc;
2990 st = av_new_stream(oc, oc->nb_streams);
2992 fprintf(stderr, "Could not alloc stream\n");
2995 avcodec_get_context_defaults2(st->codec, CODEC_TYPE_VIDEO);
2996 bitstream_filters[nb_output_files][oc->nb_streams - 1]= video_bitstream_filters;
2997 video_bitstream_filters= NULL;
3000 avcodec_thread_init(st->codec, thread_count);
3002 video_enc = st->codec;
3005 video_enc->codec_tag= video_codec_tag;
3007 if( (video_global_header&1)
3008 || (video_global_header==0 && (oc->oformat->flags & AVFMT_GLOBALHEADER))){
3009 video_enc->flags |= CODEC_FLAG_GLOBAL_HEADER;
3010 avcodec_opts[CODEC_TYPE_VIDEO]->flags|= CODEC_FLAG_GLOBAL_HEADER;
3012 if(video_global_header&2){
3013 video_enc->flags2 |= CODEC_FLAG2_LOCAL_HEADER;
3014 avcodec_opts[CODEC_TYPE_VIDEO]->flags2|= CODEC_FLAG2_LOCAL_HEADER;
3017 if (video_stream_copy) {
3018 st->stream_copy = 1;
3019 video_enc->codec_type = CODEC_TYPE_VIDEO;
3020 video_enc->sample_aspect_ratio =
3021 st->sample_aspect_ratio = av_d2q(frame_aspect_ratio*frame_height/frame_width, 255);
3026 AVRational fps= frame_rate.num ? frame_rate : (AVRational){25,1};
3028 if (video_codec_name) {
3029 codec_id = find_codec_or_die(video_codec_name, CODEC_TYPE_VIDEO, 1);
3030 codec = avcodec_find_encoder_by_name(video_codec_name);
3031 output_codecs[nb_ocodecs] = codec;
3033 codec_id = av_guess_codec(oc->oformat, NULL, oc->filename, NULL, CODEC_TYPE_VIDEO);
3034 codec = avcodec_find_encoder(codec_id);
3037 video_enc->codec_id = codec_id;
3039 set_context_opts(video_enc, avcodec_opts[CODEC_TYPE_VIDEO], AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM);
3041 if (codec && codec->supported_framerates && !force_fps)
3042 fps = codec->supported_framerates[av_find_nearest_q_idx(fps, codec->supported_framerates)];
3043 video_enc->time_base.den = fps.num;
3044 video_enc->time_base.num = fps.den;
3046 video_enc->width = frame_width + frame_padright + frame_padleft;
3047 video_enc->height = frame_height + frame_padtop + frame_padbottom;
3048 video_enc->sample_aspect_ratio = av_d2q(frame_aspect_ratio*video_enc->height/video_enc->width, 255);
3049 video_enc->pix_fmt = frame_pix_fmt;
3050 st->sample_aspect_ratio = video_enc->sample_aspect_ratio;
3052 if(codec && codec->pix_fmts){
3053 const enum PixelFormat *p= codec->pix_fmts;
3055 if(*p == video_enc->pix_fmt)
3059 video_enc->pix_fmt = codec->pix_fmts[0];
3063 video_enc->gop_size = 0;
3064 if (video_qscale || same_quality) {
3065 video_enc->flags |= CODEC_FLAG_QSCALE;
3066 video_enc->global_quality=
3067 st->quality = FF_QP2LAMBDA * video_qscale;
3071 video_enc->intra_matrix = intra_matrix;
3073 video_enc->inter_matrix = inter_matrix;
3075 video_enc->thread_count = thread_count;
3076 p= video_rc_override_string;
3079 int e=sscanf(p, "%d,%d,%d", &start, &end, &q);
3081 fprintf(stderr, "error parsing rc_override\n");
3084 video_enc->rc_override=
3085 av_realloc(video_enc->rc_override,
3086 sizeof(RcOverride)*(i+1));
3087 video_enc->rc_override[i].start_frame= start;
3088 video_enc->rc_override[i].end_frame = end;
3090 video_enc->rc_override[i].qscale= q;
3091 video_enc->rc_override[i].quality_factor= 1.0;
3094 video_enc->rc_override[i].qscale= 0;
3095 video_enc->rc_override[i].quality_factor= -q/100.0;
3100 video_enc->rc_override_count=i;
3101 if (!video_enc->rc_initial_buffer_occupancy)
3102 video_enc->rc_initial_buffer_occupancy = video_enc->rc_buffer_size*3/4;
3103 video_enc->me_threshold= me_threshold;
3104 video_enc->intra_dc_precision= intra_dc_precision - 8;
3107 video_enc->flags|= CODEC_FLAG_PSNR;
3112 video_enc->flags |= CODEC_FLAG_PASS1;
3114 video_enc->flags |= CODEC_FLAG_PASS2;
3120 /* reset some key parameters */
3122 av_freep(&video_codec_name);
3123 video_stream_copy = 0;
3126 static void new_audio_stream(AVFormatContext *oc)
3129 AVCodecContext *audio_enc;
3132 st = av_new_stream(oc, oc->nb_streams);
3134 fprintf(stderr, "Could not alloc stream\n");
3137 avcodec_get_context_defaults2(st->codec, CODEC_TYPE_AUDIO);
3139 bitstream_filters[nb_output_files][oc->nb_streams - 1]= audio_bitstream_filters;
3140 audio_bitstream_filters= NULL;
3143 avcodec_thread_init(st->codec, thread_count);
3145 audio_enc = st->codec;
3146 audio_enc->codec_type = CODEC_TYPE_AUDIO;
3149 audio_enc->codec_tag= audio_codec_tag;
3151 if (oc->oformat->flags & AVFMT_GLOBALHEADER) {
3152 audio_enc->flags |= CODEC_FLAG_GLOBAL_HEADER;
3153 avcodec_opts[CODEC_TYPE_AUDIO]->flags|= CODEC_FLAG_GLOBAL_HEADER;
3155 if (audio_stream_copy) {
3156 st->stream_copy = 1;
3157 audio_enc->channels = audio_channels;
3161 set_context_opts(audio_enc, avcodec_opts[CODEC_TYPE_AUDIO], AV_OPT_FLAG_AUDIO_PARAM | AV_OPT_FLAG_ENCODING_PARAM);
3163 if (audio_codec_name) {
3164 codec_id = find_codec_or_die(audio_codec_name, CODEC_TYPE_AUDIO, 1);
3165 codec = avcodec_find_encoder_by_name(audio_codec_name);
3166 output_codecs[nb_ocodecs] = codec;
3168 codec_id = av_guess_codec(oc->oformat, NULL, oc->filename, NULL, CODEC_TYPE_AUDIO);
3169 codec = avcodec_find_encoder(codec_id);
3171 audio_enc->codec_id = codec_id;
3173 if (audio_qscale > QSCALE_NONE) {
3174 audio_enc->flags |= CODEC_FLAG_QSCALE;
3175 audio_enc->global_quality = st->quality = FF_QP2LAMBDA * audio_qscale;
3177 audio_enc->thread_count = thread_count;
3178 audio_enc->channels = audio_channels;
3179 audio_enc->sample_fmt = audio_sample_fmt;
3180 audio_enc->channel_layout = channel_layout;
3181 if (avcodec_channel_layout_num_channels(channel_layout) != audio_channels)
3182 audio_enc->channel_layout = 0;
3184 if(codec && codec->sample_fmts){
3185 const enum SampleFormat *p= codec->sample_fmts;
3187 if(*p == audio_enc->sample_fmt)
3191 audio_enc->sample_fmt = codec->sample_fmts[0];
3195 audio_enc->sample_rate = audio_sample_rate;
3196 audio_enc->time_base= (AVRational){1, audio_sample_rate};
3197 if (audio_language) {
3198 av_metadata_set(&st->metadata, "language", audio_language);
3199 av_free(audio_language);
3200 audio_language = NULL;
3203 /* reset some key parameters */
3205 av_freep(&audio_codec_name);
3206 audio_stream_copy = 0;
3209 static void new_subtitle_stream(AVFormatContext *oc)
3212 AVCodecContext *subtitle_enc;
3214 st = av_new_stream(oc, oc->nb_streams);
3216 fprintf(stderr, "Could not alloc stream\n");
3219 avcodec_get_context_defaults2(st->codec, CODEC_TYPE_SUBTITLE);
3221 bitstream_filters[nb_output_files][oc->nb_streams - 1]= subtitle_bitstream_filters;
3222 subtitle_bitstream_filters= NULL;
3224 subtitle_enc = st->codec;
3225 subtitle_enc->codec_type = CODEC_TYPE_SUBTITLE;
3227 if(subtitle_codec_tag)
3228 subtitle_enc->codec_tag= subtitle_codec_tag;
3230 if (subtitle_stream_copy) {
3231 st->stream_copy = 1;
3233 set_context_opts(avcodec_opts[CODEC_TYPE_SUBTITLE], subtitle_enc, AV_OPT_FLAG_SUBTITLE_PARAM | AV_OPT_FLAG_ENCODING_PARAM);
3234 subtitle_enc->codec_id = find_codec_or_die(subtitle_codec_name, CODEC_TYPE_SUBTITLE, 1);
3235 output_codecs[nb_ocodecs] = avcodec_find_encoder_by_name(subtitle_codec_name);
3239 if (subtitle_language) {
3240 av_metadata_set(&st->metadata, "language", subtitle_language);
3241 av_free(subtitle_language);
3242 subtitle_language = NULL;
3245 subtitle_disable = 0;
3246 av_freep(&subtitle_codec_name);
3247 subtitle_stream_copy = 0;
3250 static void opt_new_audio_stream(void)
3252 AVFormatContext *oc;
3253 if (nb_output_files <= 0) {
3254 fprintf(stderr, "At least one output file must be specified\n");
3257 oc = output_files[nb_output_files - 1];
3258 new_audio_stream(oc);
3261 static void opt_new_video_stream(void)
3263 AVFormatContext *oc;
3264 if (nb_output_files <= 0) {
3265 fprintf(stderr, "At least one output file must be specified\n");
3268 oc = output_files[nb_output_files - 1];
3269 new_video_stream(oc);
3272 static void opt_new_subtitle_stream(void)
3274 AVFormatContext *oc;
3275 if (nb_output_files <= 0) {
3276 fprintf(stderr, "At least one output file must be specified\n");
3279 oc = output_files[nb_output_files - 1];
3280 new_subtitle_stream(oc);
3283 static void opt_output_file(const char *filename)
3285 AVFormatContext *oc;
3286 int use_video, use_audio, use_subtitle;
3287 int input_has_video, input_has_audio, input_has_subtitle;
3288 AVFormatParameters params, *ap = ¶ms;
3290 if (!strcmp(filename, "-"))
3293 oc = avformat_alloc_context();
3295 if (!file_oformat) {
3296 file_oformat = guess_format(NULL, filename, NULL);
3297 if (!file_oformat) {
3298 fprintf(stderr, "Unable to find a suitable output format for '%s'\n",
3304 oc->oformat = file_oformat;
3305 av_strlcpy(oc->filename, filename, sizeof(oc->filename));
3307 if (!strcmp(file_oformat->name, "ffm") &&
3308 av_strstart(filename, "http:", NULL)) {
3309 /* special case for files sent to ffserver: we get the stream
3310 parameters from ffserver */
3311 int err = read_ffserver_streams(oc, filename);
3313 print_error(filename, err);
3317 use_video = file_oformat->video_codec != CODEC_ID_NONE || video_stream_copy || video_codec_name;
3318 use_audio = file_oformat->audio_codec != CODEC_ID_NONE || audio_stream_copy || audio_codec_name;
3319 use_subtitle = file_oformat->subtitle_codec != CODEC_ID_NONE || subtitle_stream_copy || subtitle_codec_name;
3321 /* disable if no corresponding type found and at least one
3323 if (nb_input_files > 0) {
3324 check_audio_video_sub_inputs(&input_has_video, &input_has_audio,
3325 &input_has_subtitle);
3326 if (!input_has_video)
3328 if (!input_has_audio)
3330 if (!input_has_subtitle)
3334 /* manual disable */
3335 if (audio_disable) {
3338 if (video_disable) {
3341 if (subtitle_disable) {
3346 new_video_stream(oc);
3350 new_audio_stream(oc);
3354 new_subtitle_stream(oc);
3357 oc->timestamp = rec_timestamp;
3359 for(; metadata_count>0; metadata_count--){
3360 av_metadata_set(&oc->metadata, metadata[metadata_count-1].key,
3361 metadata[metadata_count-1].value);
3363 av_metadata_conv(oc, oc->oformat->metadata_conv, NULL);
3366 output_files[nb_output_files++] = oc;
3368 /* check filename in case of an image number is expected */
3369 if (oc->oformat->flags & AVFMT_NEEDNUMBER) {
3370 if (!av_filename_number_test(oc->filename)) {
3371 print_error(oc->filename, AVERROR_NUMEXPECTED);
3376 if (!(oc->oformat->flags & AVFMT_NOFILE)) {
3377 /* test if it already exists to avoid loosing precious files */
3378 if (!file_overwrite &&
3379 (strchr(filename, ':') == NULL ||
3380 filename[1] == ':' ||
3381 av_strstart(filename, "file:", NULL))) {
3382 if (url_exist(filename)) {
3384 fprintf(stderr,"File '%s' already exists. Overwrite ? [y/N] ", filename);
3386 if (!read_yesno()) {
3387 fprintf(stderr, "Not overwriting - exiting\n");
3392 fprintf(stderr,"File '%s' already exists. Exiting.\n", filename);
3399 if (url_fopen(&oc->pb, filename, URL_WRONLY) < 0) {
3400 fprintf(stderr, "Could not open '%s'\n", filename);
3405 memset(ap, 0, sizeof(*ap));
3406 if (av_set_parameters(oc, ap) < 0) {
3407 fprintf(stderr, "%s: Invalid encoding parameters\n",
3412 oc->preload= (int)(mux_preload*AV_TIME_BASE);
3413 oc->max_delay= (int)(mux_max_delay*AV_TIME_BASE);
3414 oc->loop_output = loop_output;
3415 oc->flags |= AVFMT_FLAG_NONBLOCK;
3417 set_context_opts(oc, avformat_opts, AV_OPT_FLAG_ENCODING_PARAM);
3419 /* reset some options */
3420 file_oformat = NULL;
3421 file_iformat = NULL;
3424 /* same option as mencoder */
3425 static void opt_pass(const char *pass_str)
3428 pass = atoi(pass_str);
3429 if (pass != 1 && pass != 2) {
3430 fprintf(stderr, "pass number can be only 1 or 2\n");
3436 static int64_t getutime(void)
3439 struct rusage rusage;
3441 getrusage(RUSAGE_SELF, &rusage);
3442 return (rusage.ru_utime.tv_sec * 1000000LL) + rusage.ru_utime.tv_usec;
3443 #elif HAVE_GETPROCESSTIMES
3445 FILETIME c, e, k, u;
3446 proc = GetCurrentProcess();
3447 GetProcessTimes(proc, &c, &e, &k, &u);
3448 return ((int64_t) u.dwHighDateTime << 32 | u.dwLowDateTime) / 10;
3450 return av_gettime();
3454 static void parse_matrix_coeffs(uint16_t *dest, const char *str)
3457 const char *p = str;
3464 fprintf(stderr, "Syntax error in matrix \"%s\" at coeff %d\n", str, i);
3471 static void opt_inter_matrix(const char *arg)
3473 inter_matrix = av_mallocz(sizeof(uint16_t) * 64);
3474 parse_matrix_coeffs(inter_matrix, arg);
3477 static void opt_intra_matrix(const char *arg)
3479 intra_matrix = av_mallocz(sizeof(uint16_t) * 64);
3480 parse_matrix_coeffs(intra_matrix, arg);
3484 * Trivial log callback.
3485 * Only suitable for show_help and similar since it lacks prefix handling.
3487 static void log_callback_help(void* ptr, int level, const char* fmt, va_list vl)
3489 vfprintf(stdout, fmt, vl);
3492 static void show_help(void)
3494 av_log_set_callback(log_callback_help);
3495 printf("usage: ffmpeg [[infile options] -i infile]... {[outfile options] outfile}...\n"
3496 "Hyper fast Audio and Video encoder\n");
3498 show_help_options(options, "Main options:\n",
3499 OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_SUBTITLE | OPT_GRAB, 0);
3500 show_help_options(options, "\nAdvanced options:\n",
3501 OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_SUBTITLE | OPT_GRAB,
3503 show_help_options(options, "\nVideo options:\n",
3504 OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_GRAB,
3506 show_help_options(options, "\nAdvanced Video options:\n",
3507 OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_GRAB,
3508 OPT_VIDEO | OPT_EXPERT);
3509 show_help_options(options, "\nAudio options:\n",
3510 OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_GRAB,
3512 show_help_options(options, "\nAdvanced Audio options:\n",
3513 OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_GRAB,
3514 OPT_AUDIO | OPT_EXPERT);
3515 show_help_options(options, "\nSubtitle options:\n",
3516 OPT_SUBTITLE | OPT_GRAB,
3518 show_help_options(options, "\nAudio/Video grab options:\n",
3522 av_opt_show(avcodec_opts[0], NULL);
3524 av_opt_show(avformat_opts, NULL);
3526 av_opt_show(sws_opts, NULL);
3529 static void opt_target(const char *arg)
3532 static const char *const frame_rates[] = {"25", "30000/1001", "24000/1001"};
3534 if(!strncmp(arg, "pal-", 4)) {
3537 } else if(!strncmp(arg, "ntsc-", 5)) {
3540 } else if(!strncmp(arg, "film-", 5)) {
3545 /* Calculate FR via float to avoid int overflow */
3546 fr = (int)(frame_rate.num * 1000.0 / frame_rate.den);
3549 } else if((fr == 29970) || (fr == 23976)) {
3552 /* Try to determine PAL/NTSC by peeking in the input files */
3553 if(nb_input_files) {
3555 for(j = 0; j < nb_input_files; j++) {
3556 for(i = 0; i < input_files[j]->nb_streams; i++) {
3557 AVCodecContext *c = input_files[j]->streams[i]->codec;
3558 if(c->codec_type != CODEC_TYPE_VIDEO)
3560 fr = c->time_base.den * 1000 / c->time_base.num;
3564 } else if((fr == 29970) || (fr == 23976)) {
3574 if(verbose && norm >= 0)
3575 fprintf(stderr, "Assuming %s for target.\n", norm ? "NTSC" : "PAL");
3579 fprintf(stderr, "Could not determine norm (PAL/NTSC/NTSC-Film) for target.\n");
3580 fprintf(stderr, "Please prefix target with \"pal-\", \"ntsc-\" or \"film-\",\n");
3581 fprintf(stderr, "or set a framerate with \"-r xxx\".\n");
3585 if(!strcmp(arg, "vcd")) {
3587 opt_video_codec("mpeg1video");
3588 opt_audio_codec("mp2");
3591 opt_frame_size(norm ? "352x240" : "352x288");
3592 opt_frame_rate(NULL, frame_rates[norm]);
3593 opt_default("gop", norm ? "18" : "15");
3595 opt_default("b", "1150000");
3596 opt_default("maxrate", "1150000");
3597 opt_default("minrate", "1150000");
3598 opt_default("bufsize", "327680"); // 40*1024*8;
3600 opt_default("ab", "224000");
3601 audio_sample_rate = 44100;
3604 opt_default("packetsize", "2324");
3605 opt_default("muxrate", "1411200"); // 2352 * 75 * 8;
3607 /* We have to offset the PTS, so that it is consistent with the SCR.
3608 SCR starts at 36000, but the first two packs contain only padding
3609 and the first pack from the other stream, respectively, may also have
3610 been written before.
3611 So the real data starts at SCR 36000+3*1200. */
3612 mux_preload= (36000+3*1200) / 90000.0; //0.44
3613 } else if(!strcmp(arg, "svcd")) {
3615 opt_video_codec("mpeg2video");
3616 opt_audio_codec("mp2");
3619 opt_frame_size(norm ? "480x480" : "480x576");
3620 opt_frame_rate(NULL, frame_rates[norm]);
3621 opt_default("gop", norm ? "18" : "15");
3623 opt_default("b", "2040000");
3624 opt_default("maxrate", "2516000");
3625 opt_default("minrate", "0"); //1145000;
3626 opt_default("bufsize", "1835008"); //224*1024*8;
3627 opt_default("flags", "+scan_offset");
3630 opt_default("ab", "224000");
3631 audio_sample_rate = 44100;
3633 opt_default("packetsize", "2324");
3635 } else if(!strcmp(arg, "dvd")) {
3637 opt_video_codec("mpeg2video");
3638 opt_audio_codec("ac3");
3641 opt_frame_size(norm ? "720x480" : "720x576");
3642 opt_frame_rate(NULL, frame_rates[norm]);
3643 opt_default("gop", norm ? "18" : "15");
3645 opt_default("b", "6000000");
3646 opt_default("maxrate", "9000000");
3647 opt_default("minrate", "0"); //1500000;
3648 opt_default("bufsize", "1835008"); //224*1024*8;
3650 opt_default("packetsize", "2048"); // from www.mpucoder.com: DVD sectors contain 2048 bytes of data, this is also the size of one pack.
3651 opt_default("muxrate", "10080000"); // from mplex project: data_rate = 1260000. mux_rate = data_rate * 8
3653 opt_default("ab", "448000");
3654 audio_sample_rate = 48000;
3656 } else if(!strncmp(arg, "dv", 2)) {
3660 opt_frame_size(norm ? "720x480" : "720x576");
3661 opt_frame_pix_fmt(!strncmp(arg, "dv50", 4) ? "yuv422p" :
3662 (norm ? "yuv411p" : "yuv420p"));
3663 opt_frame_rate(NULL, frame_rates[norm]);
3665 audio_sample_rate = 48000;
3669 fprintf(stderr, "Unknown target: %s\n", arg);
3674 static void opt_vstats_file (const char *arg)
3676 av_free (vstats_filename);
3677 vstats_filename=av_strdup (arg);
3680 static void opt_vstats (void)
3683 time_t today2 = time(NULL);
3684 struct tm *today = localtime(&today2);
3686 snprintf(filename, sizeof(filename), "vstats_%02d%02d%02d.log", today->tm_hour, today->tm_min,
3688 opt_vstats_file(filename);
3691 static int opt_bsf(const char *opt, const char *arg)
3693 AVBitStreamFilterContext *bsfc= av_bitstream_filter_init(arg); //FIXME split name and args for filter at '='
3694 AVBitStreamFilterContext **bsfp;
3697 fprintf(stderr, "Unknown bitstream filter %s\n", arg);
3701 bsfp= *opt == 'v' ? &video_bitstream_filters :
3702 *opt == 'a' ? &audio_bitstream_filters :
3703 &subtitle_bitstream_filters;
3705 bsfp= &(*bsfp)->next;