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
31 #include "libavformat/avformat.h"
32 #include "libavdevice/avdevice.h"
33 #include "libswscale/swscale.h"
34 #include "libavutil/opt.h"
35 #include "libavcodec/audioconvert.h"
36 #include "libavutil/audioconvert.h"
37 #include "libavutil/parseutils.h"
38 #include "libavutil/samplefmt.h"
39 #include "libavutil/colorspace.h"
40 #include "libavutil/fifo.h"
41 #include "libavutil/intreadwrite.h"
42 #include "libavutil/dict.h"
43 #include "libavutil/mathematics.h"
44 #include "libavutil/pixdesc.h"
45 #include "libavutil/avstring.h"
46 #include "libavutil/libm.h"
47 #include "libavutil/imgutils.h"
48 #include "libavformat/os_support.h"
50 #include "libavformat/ffm.h" // not public API
53 # include "libavfilter/avcodec.h"
54 # include "libavfilter/avfilter.h"
55 # include "libavfilter/avfiltergraph.h"
56 # include "libavfilter/buffersink.h"
57 # include "libavfilter/buffersrc.h"
58 # include "libavfilter/vsrc_buffer.h"
61 #if HAVE_SYS_RESOURCE_H
62 #include <sys/types.h>
64 #include <sys/resource.h>
65 #elif HAVE_GETPROCESSTIMES
68 #if HAVE_GETPROCESSMEMORYINFO
74 #include <sys/select.h>
79 #include <sys/ioctl.h>
89 #include "libavutil/avassert.h"
92 #define VSYNC_PASSTHROUGH 0
96 const char program_name[] = "avconv";
97 const int program_birth_year = 2000;
99 /* select an input stream for an output stream */
100 typedef struct StreamMap {
101 int disabled; /** 1 is this mapping is disabled by a negative map */
105 int sync_stream_index;
109 * select an input file for an output file
111 typedef struct MetadataMap {
112 int file; ///< file index
113 char type; ///< type of metadata to copy -- (g)lobal, (s)tream, (c)hapter or (p)rogram
114 int index; ///< stream/chapter/program number
117 static const OptionDef options[];
119 #define MAX_STREAMS 1024 /* arbitrary sanity check value */
121 static int frame_bits_per_raw_sample = 0;
122 static int video_discard = 0;
123 static int same_quant = 0;
124 static int do_deinterlace = 0;
125 static int intra_dc_precision = 8;
126 static int qp_hist = 0;
128 static int file_overwrite = 0;
129 static int no_file_overwrite = 0;
130 static int do_benchmark = 0;
131 static int do_hex_dump = 0;
132 static int do_pkt_dump = 0;
133 static int do_pass = 0;
134 static const char *pass_logfilename_prefix;
135 static int video_sync_method = VSYNC_AUTO;
136 static int audio_sync_method = 0;
137 static float audio_drift_threshold = 0.1;
138 static int copy_ts = 0;
139 static int copy_tb = 1;
140 static int opt_shortest = 0;
141 static char *vstats_filename;
142 static FILE *vstats_file;
144 static int audio_volume = 256;
146 static int exit_on_error = 0;
147 static int using_stdin = 0;
148 static int run_as_daemon = 0;
149 static int q_pressed = 0;
150 static int64_t video_size = 0;
151 static int64_t audio_size = 0;
152 static int64_t extra_size = 0;
153 static int nb_frames_dup = 0;
154 static int nb_frames_drop = 0;
155 static int input_sync;
157 static float dts_delta_threshold = 10;
159 static int print_stats = 1;
161 static uint8_t *audio_buf;
162 static uint8_t *audio_out;
163 static unsigned int allocated_audio_out_size, allocated_audio_buf_size;
165 #define DEFAULT_PASS_LOGFILENAME_PREFIX "av2pass"
167 typedef struct FrameBuffer {
173 enum PixelFormat pix_fmt;
176 struct InputStream *ist;
177 struct FrameBuffer *next;
180 typedef struct InputStream {
183 int discard; /* true if stream data should be discarded */
184 int decoding_needed; /* true if the packets must be decoded in 'raw_fifo' */
186 AVFrame *decoded_frame;
187 AVFrame *filtered_frame;
189 int64_t start; /* time when read started */
190 int64_t next_pts; /* synthetic pts for cases where pkt.pts
192 int64_t pts; /* current pts */
194 int is_start; /* is 1 at the start and after a discontinuity */
195 int showed_multi_packet_warning;
198 /* a pool of free buffers for decoded data */
199 FrameBuffer *buffer_pool;
202 typedef struct InputFile {
203 AVFormatContext *ctx;
204 int eof_reached; /* true if eof reached */
205 int ist_index; /* index of first stream in ist_table */
206 int buffer_size; /* current total buffer size */
208 int nb_streams; /* number of stream that avconv is aware of; may be different
209 from ctx.nb_streams if new streams appear during av_read_frame() */
213 typedef struct OutputStream {
214 int file_index; /* file index */
215 int index; /* stream index in the output file */
216 int source_index; /* InputStream index */
217 AVStream *st; /* stream in the output file */
218 int encoding_needed; /* true if encoding needed for this stream */
220 /* input pts and corresponding output pts
222 // double sync_ipts; /* dts from the AVPacket of the demuxer in second units */
223 struct InputStream *sync_ist; /* input stream to sync against */
224 int64_t sync_opts; /* output frame counter, could be changed to some true timestamp */ // FIXME look at frame_number
225 AVBitStreamFilterContext *bitstream_filters;
231 AVFrame resample_frame; /* temporary frame for image resampling */
232 struct SwsContext *img_resample_ctx; /* for image resampling */
235 int resample_pix_fmt;
236 AVRational frame_rate;
240 float frame_aspect_ratio;
242 /* forced key frames */
243 int64_t *forced_kf_pts;
249 ReSampleContext *resample; /* for audio resampling */
250 int resample_sample_fmt;
251 int resample_channels;
252 int resample_sample_rate;
254 AVAudioConvert *reformat_ctx;
255 AVFifoBuffer *fifo; /* for compression: one audio fifo per codec */
259 AVFilterContext *output_video_filter;
260 AVFilterContext *input_video_filter;
261 AVFilterBufferRef *picref;
263 AVFilterGraph *graph;
268 int is_past_recording_time;
270 const char *attachment_filename;
271 int copy_initial_nonkeyframes;
276 /* init terminal so that we can grab keys */
277 static struct termios oldtty;
280 typedef struct OutputFile {
281 AVFormatContext *ctx;
283 int ost_index; /* index of the first stream in output_streams */
284 int64_t recording_time; /* desired length of the resulting file in microseconds */
285 int64_t start_time; /* start time in microseconds */
286 uint64_t limit_filesize;
289 static InputStream *input_streams = NULL;
290 static int nb_input_streams = 0;
291 static InputFile *input_files = NULL;
292 static int nb_input_files = 0;
294 static OutputStream *output_streams = NULL;
295 static int nb_output_streams = 0;
296 static OutputFile *output_files = NULL;
297 static int nb_output_files = 0;
299 typedef struct OptionsContext {
300 /* input/output options */
304 SpecifierOpt *codec_names;
306 SpecifierOpt *audio_channels;
307 int nb_audio_channels;
308 SpecifierOpt *audio_sample_rate;
309 int nb_audio_sample_rate;
310 SpecifierOpt *frame_rates;
312 SpecifierOpt *frame_sizes;
314 SpecifierOpt *frame_pix_fmts;
315 int nb_frame_pix_fmts;
318 int64_t input_ts_offset;
321 SpecifierOpt *ts_scale;
323 SpecifierOpt *dump_attachment;
324 int nb_dump_attachment;
327 StreamMap *stream_maps;
329 /* first item specifies output metadata, second is input */
330 MetadataMap (*meta_data_maps)[2];
331 int nb_meta_data_maps;
332 int metadata_global_manual;
333 int metadata_streams_manual;
334 int metadata_chapters_manual;
335 const char **attachments;
338 int chapters_input_file;
340 int64_t recording_time;
341 uint64_t limit_filesize;
347 int subtitle_disable;
350 /* indexed by output file stream index */
354 SpecifierOpt *metadata;
356 SpecifierOpt *max_frames;
358 SpecifierOpt *bitstream_filters;
359 int nb_bitstream_filters;
360 SpecifierOpt *codec_tags;
362 SpecifierOpt *sample_fmts;
364 SpecifierOpt *qscale;
366 SpecifierOpt *forced_key_frames;
367 int nb_forced_key_frames;
368 SpecifierOpt *force_fps;
370 SpecifierOpt *frame_aspect_ratios;
371 int nb_frame_aspect_ratios;
372 SpecifierOpt *rc_overrides;
374 SpecifierOpt *intra_matrices;
375 int nb_intra_matrices;
376 SpecifierOpt *inter_matrices;
377 int nb_inter_matrices;
378 SpecifierOpt *top_field_first;
379 int nb_top_field_first;
380 SpecifierOpt *metadata_map;
382 SpecifierOpt *presets;
384 SpecifierOpt *copy_initial_nonkeyframes;
385 int nb_copy_initial_nonkeyframes;
387 SpecifierOpt *filters;
392 #define MATCH_PER_STREAM_OPT(name, type, outvar, fmtctx, st)\
395 for (i = 0; i < o->nb_ ## name; i++) {\
396 char *spec = o->name[i].specifier;\
397 if ((ret = check_stream_specifier(fmtctx, st, spec)) > 0)\
398 outvar = o->name[i].u.type;\
404 static void reset_options(OptionsContext *o)
406 const OptionDef *po = options;
408 /* all OPT_SPEC and OPT_STRING can be freed in generic way */
410 void *dst = (uint8_t*)o + po->u.off;
412 if (po->flags & OPT_SPEC) {
413 SpecifierOpt **so = dst;
414 int i, *count = (int*)(so + 1);
415 for (i = 0; i < *count; i++) {
416 av_freep(&(*so)[i].specifier);
417 if (po->flags & OPT_STRING)
418 av_freep(&(*so)[i].u.str);
422 } else if (po->flags & OPT_OFFSET && po->flags & OPT_STRING)
427 av_freep(&o->stream_maps);
428 av_freep(&o->meta_data_maps);
429 av_freep(&o->streamid_map);
431 memset(o, 0, sizeof(*o));
433 o->mux_max_delay = 0.7;
434 o->recording_time = INT64_MAX;
435 o->limit_filesize = UINT64_MAX;
436 o->chapters_input_file = INT_MAX;
442 static int alloc_buffer(InputStream *ist, FrameBuffer **pbuf)
444 AVCodecContext *s = ist->st->codec;
445 FrameBuffer *buf = av_mallocz(sizeof(*buf));
447 const int pixel_size = av_pix_fmt_descriptors[s->pix_fmt].comp[0].step_minus1+1;
448 int h_chroma_shift, v_chroma_shift;
449 int edge = 32; // XXX should be avcodec_get_edge_width(), but that fails on svq1
450 int w = s->width, h = s->height;
453 return AVERROR(ENOMEM);
455 if (!(s->flags & CODEC_FLAG_EMU_EDGE)) {
460 avcodec_align_dimensions(s, &w, &h);
461 if ((ret = av_image_alloc(buf->base, buf->linesize, w, h,
462 s->pix_fmt, 32)) < 0) {
466 /* XXX this shouldn't be needed, but some tests break without this line
467 * those decoders are buggy and need to be fixed.
468 * the following tests fail:
469 * bethsoft-vid, cdgraphics, ansi, aasc, fraps-v1, qtrle-1bit
471 memset(buf->base[0], 128, ret);
473 avcodec_get_chroma_sub_sample(s->pix_fmt, &h_chroma_shift, &v_chroma_shift);
474 for (int i = 0; i < FF_ARRAY_ELEMS(buf->data); i++) {
475 const int h_shift = i==0 ? 0 : h_chroma_shift;
476 const int v_shift = i==0 ? 0 : v_chroma_shift;
477 if (s->flags & CODEC_FLAG_EMU_EDGE)
478 buf->data[i] = buf->base[i];
480 buf->data[i] = buf->base[i] +
481 FFALIGN((buf->linesize[i]*edge >> v_shift) +
482 (pixel_size*edge >> h_shift), 32);
486 buf->pix_fmt = s->pix_fmt;
493 static void free_buffer_pool(InputStream *ist)
495 FrameBuffer *buf = ist->buffer_pool;
497 ist->buffer_pool = buf->next;
498 av_freep(&buf->base[0]);
500 buf = ist->buffer_pool;
504 static void unref_buffer(InputStream *ist, FrameBuffer *buf)
506 av_assert0(buf->refcount);
508 if (!buf->refcount) {
509 buf->next = ist->buffer_pool;
510 ist->buffer_pool = buf;
514 static int codec_get_buffer(AVCodecContext *s, AVFrame *frame)
516 InputStream *ist = s->opaque;
520 if (!ist->buffer_pool && (ret = alloc_buffer(ist, &ist->buffer_pool)) < 0)
523 buf = ist->buffer_pool;
524 ist->buffer_pool = buf->next;
526 if (buf->w != s->width || buf->h != s->height || buf->pix_fmt != s->pix_fmt) {
527 av_freep(&buf->base[0]);
529 if ((ret = alloc_buffer(ist, &buf)) < 0)
535 frame->type = FF_BUFFER_TYPE_USER;
536 frame->extended_data = frame->data;
537 frame->pkt_pts = s->pkt ? s->pkt->pts : AV_NOPTS_VALUE;
539 for (i = 0; i < FF_ARRAY_ELEMS(buf->data); i++) {
540 frame->base[i] = buf->base[i]; // XXX h264.c uses base though it shouldn't
541 frame->data[i] = buf->data[i];
542 frame->linesize[i] = buf->linesize[i];
548 static void codec_release_buffer(AVCodecContext *s, AVFrame *frame)
550 InputStream *ist = s->opaque;
551 FrameBuffer *buf = frame->opaque;
554 for (i = 0; i < FF_ARRAY_ELEMS(frame->data); i++)
555 frame->data[i] = NULL;
557 unref_buffer(ist, buf);
560 static void filter_release_buffer(AVFilterBuffer *fb)
562 FrameBuffer *buf = fb->priv;
564 unref_buffer(buf->ist, buf);
569 static int configure_video_filters(InputStream *ist, OutputStream *ost)
571 AVFilterContext *last_filter, *filter;
572 /** filter graph containing all filters including input & output */
573 AVCodecContext *codec = ost->st->codec;
574 AVCodecContext *icodec = ist->st->codec;
575 enum PixelFormat pix_fmts[] = { codec->pix_fmt, PIX_FMT_NONE };
576 AVBufferSinkParams *buffersink_params = av_buffersink_params_alloc();
577 AVRational sample_aspect_ratio;
581 ost->graph = avfilter_graph_alloc();
583 if (ist->st->sample_aspect_ratio.num) {
584 sample_aspect_ratio = ist->st->sample_aspect_ratio;
586 sample_aspect_ratio = ist->st->codec->sample_aspect_ratio;
588 snprintf(args, 255, "%d:%d:%d:%d:%d:%d:%d", ist->st->codec->width,
589 ist->st->codec->height, ist->st->codec->pix_fmt, 1, AV_TIME_BASE,
590 sample_aspect_ratio.num, sample_aspect_ratio.den);
592 ret = avfilter_graph_create_filter(&ost->input_video_filter, avfilter_get_by_name("buffer"),
593 "src", args, NULL, ost->graph);
596 #if FF_API_OLD_VSINK_API
597 ret = avfilter_graph_create_filter(&ost->output_video_filter, avfilter_get_by_name("buffersink"),
598 "out", NULL, pix_fmts, ost->graph);
600 buffersink_params->pixel_fmts = pix_fmts;
601 ret = avfilter_graph_create_filter(&ost->output_video_filter, avfilter_get_by_name("buffersink"),
602 "out", NULL, buffersink_params, ost->graph);
604 av_freep(&buffersink_params);
607 last_filter = ost->input_video_filter;
609 if (codec->width != icodec->width || codec->height != icodec->height) {
610 snprintf(args, 255, "%d:%d:flags=0x%X",
613 (unsigned)ost->sws_flags);
614 if ((ret = avfilter_graph_create_filter(&filter, avfilter_get_by_name("scale"),
615 NULL, args, NULL, ost->graph)) < 0)
617 if ((ret = avfilter_link(last_filter, 0, filter, 0)) < 0)
619 last_filter = filter;
622 snprintf(args, sizeof(args), "flags=0x%X", (unsigned)ost->sws_flags);
623 ost->graph->scale_sws_opts = av_strdup(args);
626 AVFilterInOut *outputs = avfilter_inout_alloc();
627 AVFilterInOut *inputs = avfilter_inout_alloc();
629 outputs->name = av_strdup("in");
630 outputs->filter_ctx = last_filter;
631 outputs->pad_idx = 0;
632 outputs->next = NULL;
634 inputs->name = av_strdup("out");
635 inputs->filter_ctx = ost->output_video_filter;
639 if ((ret = avfilter_graph_parse(ost->graph, ost->avfilter, &inputs, &outputs, NULL)) < 0)
642 if ((ret = avfilter_link(last_filter, 0, ost->output_video_filter, 0)) < 0)
646 if ((ret = avfilter_graph_config(ost->graph, NULL)) < 0)
649 codec->width = ost->output_video_filter->inputs[0]->w;
650 codec->height = ost->output_video_filter->inputs[0]->h;
651 codec->sample_aspect_ratio = ost->st->sample_aspect_ratio =
652 ost->frame_aspect_ratio ? // overridden by the -aspect cli option
653 av_d2q(ost->frame_aspect_ratio * codec->height/codec->width, 255) :
654 ost->output_video_filter->inputs[0]->sample_aspect_ratio;
658 #endif /* CONFIG_AVFILTER */
660 static void term_exit(void)
662 av_log(NULL, AV_LOG_QUIET, "%s", "");
665 tcsetattr (0, TCSANOW, &oldtty);
669 static volatile int received_sigterm = 0;
672 sigterm_handler(int sig)
674 received_sigterm = sig;
679 static void term_init(void)
689 tty.c_iflag &= ~(IGNBRK|BRKINT|PARMRK|ISTRIP
690 |INLCR|IGNCR|ICRNL|IXON);
691 tty.c_oflag |= OPOST;
692 tty.c_lflag &= ~(ECHO|ECHONL|ICANON|IEXTEN);
693 tty.c_cflag &= ~(CSIZE|PARENB);
698 tcsetattr (0, TCSANOW, &tty);
699 signal(SIGQUIT, sigterm_handler); /* Quit (POSIX). */
703 signal(SIGINT , sigterm_handler); /* Interrupt (ANSI). */
704 signal(SIGTERM, sigterm_handler); /* Termination (ANSI). */
706 signal(SIGXCPU, sigterm_handler);
710 /* read a key without blocking */
711 static int read_key(void)
726 n = select(1, &rfds, NULL, NULL, &tv);
741 static int decode_interrupt_cb(void *ctx)
743 q_pressed += read_key() == 'q';
744 return q_pressed > 1;
747 static const AVIOInterruptCB int_cb = { decode_interrupt_cb, NULL };
749 void exit_program(int ret)
754 for (i = 0; i < nb_output_files; i++) {
755 AVFormatContext *s = output_files[i].ctx;
756 if (!(s->oformat->flags & AVFMT_NOFILE) && s->pb)
758 avformat_free_context(s);
759 av_dict_free(&output_files[i].opts);
761 for (i = 0; i < nb_output_streams; i++) {
762 AVBitStreamFilterContext *bsfc = output_streams[i].bitstream_filters;
764 AVBitStreamFilterContext *next = bsfc->next;
765 av_bitstream_filter_close(bsfc);
768 output_streams[i].bitstream_filters = NULL;
771 av_freep(&output_streams[i].avfilter);
774 for (i = 0; i < nb_input_files; i++) {
775 avformat_close_input(&input_files[i].ctx);
777 for (i = 0; i < nb_input_streams; i++) {
778 av_freep(&input_streams[i].decoded_frame);
779 av_freep(&input_streams[i].filtered_frame);
780 av_dict_free(&input_streams[i].opts);
781 free_buffer_pool(&input_streams[i]);
786 av_free(vstats_filename);
788 av_freep(&input_streams);
789 av_freep(&input_files);
790 av_freep(&output_streams);
791 av_freep(&output_files);
796 allocated_audio_buf_size = allocated_audio_out_size = 0;
801 avformat_network_deinit();
803 if (received_sigterm) {
804 av_log(NULL, AV_LOG_INFO, "Received signal %d: terminating.\n",
805 (int) received_sigterm);
812 static void assert_avoptions(AVDictionary *m)
814 AVDictionaryEntry *t;
815 if ((t = av_dict_get(m, "", NULL, AV_DICT_IGNORE_SUFFIX))) {
816 av_log(NULL, AV_LOG_FATAL, "Option %s not found.\n", t->key);
821 static void assert_codec_experimental(AVCodecContext *c, int encoder)
823 const char *codec_string = encoder ? "encoder" : "decoder";
825 if (c->codec->capabilities & CODEC_CAP_EXPERIMENTAL &&
826 c->strict_std_compliance > FF_COMPLIANCE_EXPERIMENTAL) {
827 av_log(NULL, AV_LOG_FATAL, "%s '%s' is experimental and might produce bad "
828 "results.\nAdd '-strict experimental' if you want to use it.\n",
829 codec_string, c->codec->name);
830 codec = encoder ? avcodec_find_encoder(c->codec->id) : avcodec_find_decoder(c->codec->id);
831 if (!(codec->capabilities & CODEC_CAP_EXPERIMENTAL))
832 av_log(NULL, AV_LOG_FATAL, "Or use the non experimental %s '%s'.\n",
833 codec_string, codec->name);
838 static void choose_sample_fmt(AVStream *st, AVCodec *codec)
840 if (codec && codec->sample_fmts) {
841 const enum AVSampleFormat *p = codec->sample_fmts;
842 for (; *p != -1; p++) {
843 if (*p == st->codec->sample_fmt)
847 if((codec->capabilities & CODEC_CAP_LOSSLESS) && av_get_sample_fmt_name(st->codec->sample_fmt) > av_get_sample_fmt_name(codec->sample_fmts[0]))
848 av_log(NULL, AV_LOG_ERROR, "Convertion will not be lossless'\n");
849 if(av_get_sample_fmt_name(st->codec->sample_fmt))
850 av_log(NULL, AV_LOG_WARNING,
851 "Incompatible sample format '%s' for codec '%s', auto-selecting format '%s'\n",
852 av_get_sample_fmt_name(st->codec->sample_fmt),
854 av_get_sample_fmt_name(codec->sample_fmts[0]));
855 st->codec->sample_fmt = codec->sample_fmts[0];
860 static void choose_sample_rate(AVStream *st, AVCodec *codec)
862 if (codec && codec->supported_samplerates) {
863 const int *p = codec->supported_samplerates;
865 int best_dist = INT_MAX;
867 int dist = abs(st->codec->sample_rate - *p);
868 if (dist < best_dist) {
874 av_log(st->codec, AV_LOG_WARNING, "Requested sampling rate unsupported using closest supported (%d)\n", best);
876 st->codec->sample_rate = best;
880 static void choose_pixel_fmt(AVStream *st, AVCodec *codec)
882 if (codec && codec->pix_fmts) {
883 const enum PixelFormat *p = codec->pix_fmts;
884 if (st->codec->strict_std_compliance <= FF_COMPLIANCE_UNOFFICIAL) {
885 if (st->codec->codec_id == CODEC_ID_MJPEG) {
886 p = (const enum PixelFormat[]) { PIX_FMT_YUVJ420P, PIX_FMT_YUVJ422P, PIX_FMT_YUV420P, PIX_FMT_YUV422P, PIX_FMT_NONE };
887 } else if (st->codec->codec_id == CODEC_ID_LJPEG) {
888 p = (const enum PixelFormat[]) { PIX_FMT_YUVJ420P, PIX_FMT_YUVJ422P, PIX_FMT_YUVJ444P, PIX_FMT_YUV420P,
889 PIX_FMT_YUV422P, PIX_FMT_YUV444P, PIX_FMT_BGRA, PIX_FMT_NONE };
892 for (; *p != PIX_FMT_NONE; p++) {
893 if (*p == st->codec->pix_fmt)
896 if (*p == PIX_FMT_NONE) {
897 if (st->codec->pix_fmt != PIX_FMT_NONE)
898 av_log(NULL, AV_LOG_WARNING,
899 "Incompatible pixel format '%s' for codec '%s', auto-selecting format '%s'\n",
900 av_pix_fmt_descriptors[st->codec->pix_fmt].name,
902 av_pix_fmt_descriptors[codec->pix_fmts[0]].name);
903 st->codec->pix_fmt = codec->pix_fmts[0];
909 get_sync_ipts(const OutputStream *ost)
911 const InputStream *ist = ost->sync_ist;
912 OutputFile *of = &output_files[ost->file_index];
913 return (double)(ist->pts - of->start_time) / AV_TIME_BASE;
916 static void write_frame(AVFormatContext *s, AVPacket *pkt, OutputStream *ost)
918 AVBitStreamFilterContext *bsfc = ost->bitstream_filters;
919 AVCodecContext *avctx = ost->st->codec;
923 AVPacket new_pkt = *pkt;
924 int a = av_bitstream_filter_filter(bsfc, avctx, NULL,
925 &new_pkt.data, &new_pkt.size,
926 pkt->data, pkt->size,
927 pkt->flags & AV_PKT_FLAG_KEY);
930 new_pkt.destruct = av_destruct_packet;
932 av_log(NULL, AV_LOG_ERROR, "%s failed for stream %d, codec %s",
933 bsfc->filter->name, pkt->stream_index,
934 avctx->codec ? avctx->codec->name : "copy");
944 ret = av_interleaved_write_frame(s, pkt);
946 print_error("av_interleaved_write_frame()", ret);
952 static void generate_silence(uint8_t* buf, enum AVSampleFormat sample_fmt, size_t size)
954 int fill_char = 0x00;
955 if (sample_fmt == AV_SAMPLE_FMT_U8)
957 memset(buf, fill_char, size);
960 static void do_audio_out(AVFormatContext *s, OutputStream *ost,
961 InputStream *ist, AVFrame *decoded_frame)
964 int64_t audio_out_size, audio_buf_size;
966 int size_out, frame_bytes, ret, resample_changed;
967 AVCodecContext *enc = ost->st->codec;
968 AVCodecContext *dec = ist->st->codec;
969 int osize = av_get_bytes_per_sample(enc->sample_fmt);
970 int isize = av_get_bytes_per_sample(dec->sample_fmt);
971 const int coded_bps = av_get_bits_per_sample(enc->codec->id);
972 uint8_t *buf = decoded_frame->data[0];
973 int size = decoded_frame->nb_samples * dec->channels * isize;
974 int64_t allocated_for_size = size;
977 audio_buf_size = (allocated_for_size + isize * dec->channels - 1) / (isize * dec->channels);
978 audio_buf_size = (audio_buf_size * enc->sample_rate + dec->sample_rate) / dec->sample_rate;
979 audio_buf_size = audio_buf_size * 2 + 10000; // safety factors for the deprecated resampling API
980 audio_buf_size = FFMAX(audio_buf_size, enc->frame_size);
981 audio_buf_size *= osize * enc->channels;
983 audio_out_size = FFMAX(audio_buf_size, enc->frame_size * osize * enc->channels);
984 if (coded_bps > 8 * osize)
985 audio_out_size = audio_out_size * coded_bps / (8*osize);
986 audio_out_size += FF_MIN_BUFFER_SIZE;
988 if (audio_out_size > INT_MAX || audio_buf_size > INT_MAX) {
989 av_log(NULL, AV_LOG_FATAL, "Buffer sizes too large\n");
993 av_fast_malloc(&audio_buf, &allocated_audio_buf_size, audio_buf_size);
994 av_fast_malloc(&audio_out, &allocated_audio_out_size, audio_out_size);
995 if (!audio_buf || !audio_out) {
996 av_log(NULL, AV_LOG_FATAL, "Out of memory in do_audio_out\n");
1000 if (enc->channels != dec->channels)
1001 ost->audio_resample = 1;
1003 resample_changed = ost->resample_sample_fmt != dec->sample_fmt ||
1004 ost->resample_channels != dec->channels ||
1005 ost->resample_sample_rate != dec->sample_rate;
1007 if ((ost->audio_resample && !ost->resample) || resample_changed) {
1008 if (resample_changed) {
1009 av_log(NULL, AV_LOG_INFO, "Input stream #%d:%d frame changed from rate:%d fmt:%s ch:%d to rate:%d fmt:%s ch:%d\n",
1010 ist->file_index, ist->st->index,
1011 ost->resample_sample_rate, av_get_sample_fmt_name(ost->resample_sample_fmt), ost->resample_channels,
1012 dec->sample_rate, av_get_sample_fmt_name(dec->sample_fmt), dec->channels);
1013 ost->resample_sample_fmt = dec->sample_fmt;
1014 ost->resample_channels = dec->channels;
1015 ost->resample_sample_rate = dec->sample_rate;
1017 audio_resample_close(ost->resample);
1019 /* if audio_sync_method is >1 the resampler is needed for audio drift compensation */
1020 if (audio_sync_method <= 1 &&
1021 ost->resample_sample_fmt == enc->sample_fmt &&
1022 ost->resample_channels == enc->channels &&
1023 ost->resample_sample_rate == enc->sample_rate) {
1024 ost->resample = NULL;
1025 ost->audio_resample = 0;
1027 if (dec->sample_fmt != AV_SAMPLE_FMT_S16)
1028 av_log(NULL, AV_LOG_WARNING, "Using s16 intermediate sample format for resampling\n");
1029 ost->resample = av_audio_resample_init(enc->channels, dec->channels,
1030 enc->sample_rate, dec->sample_rate,
1031 enc->sample_fmt, dec->sample_fmt,
1033 if (!ost->resample) {
1034 av_log(NULL, AV_LOG_FATAL, "Can not resample %d channels @ %d Hz to %d channels @ %d Hz\n",
1035 dec->channels, dec->sample_rate,
1036 enc->channels, enc->sample_rate);
1042 #define MAKE_SFMT_PAIR(a,b) ((a)+AV_SAMPLE_FMT_NB*(b))
1043 if (!ost->audio_resample && dec->sample_fmt != enc->sample_fmt &&
1044 MAKE_SFMT_PAIR(enc->sample_fmt,dec->sample_fmt) != ost->reformat_pair) {
1045 if (ost->reformat_ctx)
1046 av_audio_convert_free(ost->reformat_ctx);
1047 ost->reformat_ctx = av_audio_convert_alloc(enc->sample_fmt, 1,
1048 dec->sample_fmt, 1, NULL, 0);
1049 if (!ost->reformat_ctx) {
1050 av_log(NULL, AV_LOG_FATAL, "Cannot convert %s sample format to %s sample format\n",
1051 av_get_sample_fmt_name(dec->sample_fmt),
1052 av_get_sample_fmt_name(enc->sample_fmt));
1055 ost->reformat_pair = MAKE_SFMT_PAIR(enc->sample_fmt,dec->sample_fmt);
1058 if (audio_sync_method) {
1059 double delta = get_sync_ipts(ost) * enc->sample_rate - ost->sync_opts -
1060 av_fifo_size(ost->fifo) / (enc->channels * osize);
1061 int idelta = delta * dec->sample_rate / enc->sample_rate;
1062 int byte_delta = idelta * isize * dec->channels;
1064 // FIXME resample delay
1065 if (fabs(delta) > 50) {
1066 if (ist->is_start || fabs(delta) > audio_drift_threshold*enc->sample_rate) {
1067 if (byte_delta < 0) {
1068 byte_delta = FFMAX(byte_delta, -size);
1071 av_log(NULL, AV_LOG_VERBOSE, "discarding %d audio samples\n",
1072 -byte_delta / (isize * dec->channels));
1077 static uint8_t *input_tmp = NULL;
1078 input_tmp = av_realloc(input_tmp, byte_delta + size);
1080 if (byte_delta > allocated_for_size - size) {
1081 allocated_for_size = byte_delta + (int64_t)size;
1086 generate_silence(input_tmp, dec->sample_fmt, byte_delta);
1087 memcpy(input_tmp + byte_delta, buf, size);
1090 av_log(NULL, AV_LOG_VERBOSE, "adding %d audio samples of silence\n", idelta);
1092 } else if (audio_sync_method > 1) {
1093 int comp = av_clip(delta, -audio_sync_method, audio_sync_method);
1094 av_assert0(ost->audio_resample);
1095 av_log(NULL, AV_LOG_VERBOSE, "compensating audio timestamp drift:%f compensation:%d in:%d\n",
1096 delta, comp, enc->sample_rate);
1097 // 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));
1098 av_resample_compensate(*(struct AVResampleContext**)ost->resample, comp, enc->sample_rate);
1102 ost->sync_opts = lrintf(get_sync_ipts(ost) * enc->sample_rate) -
1103 av_fifo_size(ost->fifo) / (enc->channels * osize); // FIXME wrong
1105 if (ost->audio_resample) {
1107 size_out = audio_resample(ost->resample,
1108 (short *)buftmp, (short *)buf,
1109 size / (dec->channels * isize));
1110 size_out = size_out * enc->channels * osize;
1116 if (!ost->audio_resample && dec->sample_fmt != enc->sample_fmt) {
1117 const void *ibuf[6] = { buftmp };
1118 void *obuf[6] = { audio_buf };
1119 int istride[6] = { isize };
1120 int ostride[6] = { osize };
1121 int len = size_out / istride[0];
1122 if (av_audio_convert(ost->reformat_ctx, obuf, ostride, ibuf, istride, len) < 0) {
1123 printf("av_audio_convert() failed\n");
1129 size_out = len * osize;
1132 /* now encode as many frames as possible */
1133 if (enc->frame_size > 1) {
1134 /* output resampled raw samples */
1135 if (av_fifo_realloc2(ost->fifo, av_fifo_size(ost->fifo) + size_out) < 0) {
1136 av_log(NULL, AV_LOG_FATAL, "av_fifo_realloc2() failed\n");
1139 av_fifo_generic_write(ost->fifo, buftmp, size_out, NULL);
1141 frame_bytes = enc->frame_size * osize * enc->channels;
1143 while (av_fifo_size(ost->fifo) >= frame_bytes) {
1145 av_init_packet(&pkt);
1147 av_fifo_generic_read(ost->fifo, audio_buf, frame_bytes, NULL);
1149 // FIXME pass ost->sync_opts as AVFrame.pts in avcodec_encode_audio()
1151 ret = avcodec_encode_audio(enc, audio_out, audio_out_size,
1152 (short *)audio_buf);
1154 av_log(NULL, AV_LOG_FATAL, "Audio encoding failed\n");
1158 pkt.stream_index = ost->index;
1159 pkt.data = audio_out;
1161 if (enc->coded_frame && enc->coded_frame->pts != AV_NOPTS_VALUE)
1162 pkt.pts = av_rescale_q(enc->coded_frame->pts, enc->time_base, ost->st->time_base);
1163 pkt.flags |= AV_PKT_FLAG_KEY;
1164 write_frame(s, &pkt, ost);
1166 ost->sync_opts += enc->frame_size;
1170 av_init_packet(&pkt);
1172 ost->sync_opts += size_out / (osize * enc->channels);
1174 /* output a pcm frame */
1175 /* determine the size of the coded buffer */
1178 size_out = size_out * coded_bps / 8;
1180 if (size_out > audio_out_size) {
1181 av_log(NULL, AV_LOG_FATAL, "Internal error, buffer size too small\n");
1185 // FIXME pass ost->sync_opts as AVFrame.pts in avcodec_encode_audio()
1186 ret = avcodec_encode_audio(enc, audio_out, size_out,
1189 av_log(NULL, AV_LOG_FATAL, "Audio encoding failed\n");
1193 pkt.stream_index = ost->index;
1194 pkt.data = audio_out;
1196 if (enc->coded_frame && enc->coded_frame->pts != AV_NOPTS_VALUE)
1197 pkt.pts = av_rescale_q(enc->coded_frame->pts, enc->time_base, ost->st->time_base);
1198 pkt.flags |= AV_PKT_FLAG_KEY;
1199 write_frame(s, &pkt, ost);
1203 static void pre_process_video_frame(InputStream *ist, AVPicture *picture, void **bufp)
1205 AVCodecContext *dec;
1206 AVPicture *picture2;
1207 AVPicture picture_tmp;
1210 dec = ist->st->codec;
1212 /* deinterlace : must be done before any resize */
1213 if (do_deinterlace) {
1216 /* create temporary picture */
1217 size = avpicture_get_size(dec->pix_fmt, dec->width, dec->height);
1218 buf = av_malloc(size);
1222 picture2 = &picture_tmp;
1223 avpicture_fill(picture2, buf, dec->pix_fmt, dec->width, dec->height);
1225 if (avpicture_deinterlace(picture2, picture,
1226 dec->pix_fmt, dec->width, dec->height) < 0) {
1227 /* if error, do not deinterlace */
1228 av_log(NULL, AV_LOG_WARNING, "Deinterlacing failed\n");
1237 if (picture != picture2)
1238 *picture = *picture2;
1242 static void do_subtitle_out(AVFormatContext *s,
1248 static uint8_t *subtitle_out = NULL;
1249 int subtitle_out_max_size = 1024 * 1024;
1250 int subtitle_out_size, nb, i;
1251 AVCodecContext *enc;
1254 if (pts == AV_NOPTS_VALUE) {
1255 av_log(NULL, AV_LOG_ERROR, "Subtitle packets must have a pts\n");
1261 enc = ost->st->codec;
1263 if (!subtitle_out) {
1264 subtitle_out = av_malloc(subtitle_out_max_size);
1267 /* Note: DVB subtitle need one packet to draw them and one other
1268 packet to clear them */
1269 /* XXX: signal it in the codec context ? */
1270 if (enc->codec_id == CODEC_ID_DVB_SUBTITLE)
1275 for (i = 0; i < nb; i++) {
1276 sub->pts = av_rescale_q(pts, ist->st->time_base, AV_TIME_BASE_Q);
1277 // start_display_time is required to be 0
1278 sub->pts += av_rescale_q(sub->start_display_time, (AVRational){ 1, 1000 }, AV_TIME_BASE_Q);
1279 sub->end_display_time -= sub->start_display_time;
1280 sub->start_display_time = 0;
1281 subtitle_out_size = avcodec_encode_subtitle(enc, subtitle_out,
1282 subtitle_out_max_size, sub);
1283 if (subtitle_out_size < 0) {
1284 av_log(NULL, AV_LOG_FATAL, "Subtitle encoding failed\n");
1288 av_init_packet(&pkt);
1289 pkt.stream_index = ost->index;
1290 pkt.data = subtitle_out;
1291 pkt.size = subtitle_out_size;
1292 pkt.pts = av_rescale_q(sub->pts, AV_TIME_BASE_Q, ost->st->time_base);
1293 if (enc->codec_id == CODEC_ID_DVB_SUBTITLE) {
1294 /* XXX: the pts correction is handled here. Maybe handling
1295 it in the codec would be better */
1297 pkt.pts += 90 * sub->start_display_time;
1299 pkt.pts += 90 * sub->end_display_time;
1301 write_frame(s, &pkt, ost);
1305 static int bit_buffer_size = 1024 * 256;
1306 static uint8_t *bit_buffer = NULL;
1308 #if !CONFIG_AVFILTER
1309 static void do_video_resample(OutputStream *ost,
1311 AVFrame *in_picture,
1312 AVFrame **out_picture)
1314 int resample_changed = 0;
1315 AVCodecContext *dec = ist->st->codec;
1316 AVCodecContext *enc = ost->st->codec;
1317 *out_picture = in_picture;
1319 resample_changed = ost->resample_width != dec->width ||
1320 ost->resample_height != dec->height ||
1321 ost->resample_pix_fmt != dec->pix_fmt;
1323 if (resample_changed) {
1324 av_log(NULL, AV_LOG_INFO,
1325 "Input stream #%d:%d frame changed from size:%dx%d fmt:%s to size:%dx%d fmt:%s\n",
1326 ist->file_index, ist->st->index,
1327 ost->resample_width, ost->resample_height, av_get_pix_fmt_name(ost->resample_pix_fmt),
1328 dec->width , dec->height , av_get_pix_fmt_name(dec->pix_fmt));
1329 ost->resample_width = dec->width;
1330 ost->resample_height = dec->height;
1331 ost->resample_pix_fmt = dec->pix_fmt;
1334 ost->video_resample = dec->width != enc->width ||
1335 dec->height != enc->height ||
1336 dec->pix_fmt != enc->pix_fmt;
1339 if (ost->video_resample) {
1340 *out_picture = &ost->resample_frame;
1341 if (!ost->img_resample_ctx || resample_changed) {
1342 /* initialize the destination picture */
1343 if (!ost->resample_frame.data[0]) {
1344 avcodec_get_frame_defaults(&ost->resample_frame);
1345 if (avpicture_alloc((AVPicture *)&ost->resample_frame, enc->pix_fmt,
1346 enc->width, enc->height)) {
1347 fprintf(stderr, "Cannot allocate temp picture, check pix fmt\n");
1351 /* initialize a new scaler context */
1352 sws_freeContext(ost->img_resample_ctx);
1353 ost->img_resample_ctx = sws_getContext(dec->width, dec->height, dec->pix_fmt,
1354 enc->width, enc->height, enc->pix_fmt,
1355 ost->sws_flags, NULL, NULL, NULL);
1356 if (ost->img_resample_ctx == NULL) {
1357 av_log(NULL, AV_LOG_FATAL, "Cannot get resampling context\n");
1361 sws_scale(ost->img_resample_ctx, in_picture->data, in_picture->linesize,
1362 0, ost->resample_height, (*out_picture)->data, (*out_picture)->linesize);
1364 if (resample_changed) {
1365 ost->resample_width = dec->width;
1366 ost->resample_height = dec->height;
1367 ost->resample_pix_fmt = dec->pix_fmt;
1373 static void do_video_out(AVFormatContext *s,
1376 AVFrame *in_picture,
1377 int *frame_size, float quality)
1379 int nb_frames, i, ret, format_video_sync;
1380 AVFrame *final_picture;
1381 AVCodecContext *enc;
1384 enc = ost->st->codec;
1386 sync_ipts = get_sync_ipts(ost) / av_q2d(enc->time_base);
1388 /* by default, we output a single frame */
1393 format_video_sync = video_sync_method;
1394 if (format_video_sync == VSYNC_AUTO)
1395 format_video_sync = (s->oformat->flags & AVFMT_NOTIMESTAMPS) ? VSYNC_PASSTHROUGH :
1396 (s->oformat->flags & AVFMT_VARIABLE_FPS) ? VSYNC_VFR : VSYNC_CFR;
1398 if (format_video_sync != VSYNC_PASSTHROUGH) {
1399 double vdelta = sync_ipts - ost->sync_opts;
1400 // FIXME set to 0.5 after we fix some dts/pts bugs like in avidec.c
1403 else if (format_video_sync == VSYNC_VFR) {
1404 if (vdelta <= -0.6) {
1406 } else if (vdelta > 0.6)
1407 ost->sync_opts = lrintf(sync_ipts);
1408 } else if (vdelta > 1.1)
1409 nb_frames = lrintf(vdelta);
1410 //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);
1411 if (nb_frames == 0) {
1413 av_log(NULL, AV_LOG_VERBOSE, "*** drop!\n");
1414 } else if (nb_frames > 1) {
1415 nb_frames_dup += nb_frames - 1;
1416 av_log(NULL, AV_LOG_VERBOSE, "*** %d dup!\n", nb_frames - 1);
1419 ost->sync_opts = lrintf(sync_ipts);
1421 nb_frames = FFMIN(nb_frames, ost->max_frames - ost->frame_number);
1425 #if !CONFIG_AVFILTER
1426 do_video_resample(ost, ist, in_picture, &final_picture);
1428 final_picture = in_picture;
1431 /* duplicates frame if needed */
1432 for (i = 0; i < nb_frames; i++) {
1434 av_init_packet(&pkt);
1435 pkt.stream_index = ost->index;
1437 if (s->oformat->flags & AVFMT_RAWPICTURE &&
1438 enc->codec->id == CODEC_ID_RAWVIDEO) {
1439 /* raw pictures are written as AVPicture structure to
1440 avoid any copies. We support temporarily the older
1442 enc->coded_frame->interlaced_frame = in_picture->interlaced_frame;
1443 enc->coded_frame->top_field_first = in_picture->top_field_first;
1444 pkt.data = (uint8_t *)final_picture;
1445 pkt.size = sizeof(AVPicture);
1446 pkt.pts = av_rescale_q(ost->sync_opts, enc->time_base, ost->st->time_base);
1447 pkt.flags |= AV_PKT_FLAG_KEY;
1449 write_frame(s, &pkt, ost);
1451 AVFrame big_picture;
1453 big_picture = *final_picture;
1454 /* better than nothing: use input picture interlaced
1456 big_picture.interlaced_frame = in_picture->interlaced_frame;
1457 if (ost->st->codec->flags & (CODEC_FLAG_INTERLACED_DCT|CODEC_FLAG_INTERLACED_ME)) {
1458 if (ost->top_field_first == -1)
1459 big_picture.top_field_first = in_picture->top_field_first;
1461 big_picture.top_field_first = !!ost->top_field_first;
1464 /* handles same_quant here. This is not correct because it may
1465 not be a global option */
1466 big_picture.quality = quality;
1467 if (!enc->me_threshold)
1468 big_picture.pict_type = 0;
1469 // big_picture.pts = AV_NOPTS_VALUE;
1470 big_picture.pts = ost->sync_opts;
1471 // big_picture.pts= av_rescale(ost->sync_opts, AV_TIME_BASE*(int64_t)enc->time_base.num, enc->time_base.den);
1472 // av_log(NULL, AV_LOG_DEBUG, "%"PRId64" -> encoder\n", ost->sync_opts);
1473 if (ost->forced_kf_index < ost->forced_kf_count &&
1474 big_picture.pts >= ost->forced_kf_pts[ost->forced_kf_index]) {
1475 big_picture.pict_type = AV_PICTURE_TYPE_I;
1476 ost->forced_kf_index++;
1478 ret = avcodec_encode_video(enc,
1479 bit_buffer, bit_buffer_size,
1482 av_log(NULL, AV_LOG_FATAL, "Video encoding failed\n");
1487 pkt.data = bit_buffer;
1489 if (enc->coded_frame->pts != AV_NOPTS_VALUE)
1490 pkt.pts = av_rescale_q(enc->coded_frame->pts, enc->time_base, ost->st->time_base);
1491 /*av_log(NULL, AV_LOG_DEBUG, "encoder -> %"PRId64"/%"PRId64"\n",
1492 pkt.pts != AV_NOPTS_VALUE ? av_rescale(pkt.pts, enc->time_base.den, AV_TIME_BASE*(int64_t)enc->time_base.num) : -1,
1493 pkt.dts != AV_NOPTS_VALUE ? av_rescale(pkt.dts, enc->time_base.den, AV_TIME_BASE*(int64_t)enc->time_base.num) : -1);*/
1495 if (enc->coded_frame->key_frame)
1496 pkt.flags |= AV_PKT_FLAG_KEY;
1497 write_frame(s, &pkt, ost);
1500 // fprintf(stderr,"\nFrame: %3d size: %5d type: %d",
1501 // enc->frame_number-1, ret, enc->pict_type);
1502 /* if two pass, output log */
1503 if (ost->logfile && enc->stats_out) {
1504 fprintf(ost->logfile, "%s", enc->stats_out);
1512 static double psnr(double d)
1514 return -10.0 * log(d) / log(10.0);
1517 static void do_video_stats(AVFormatContext *os, OutputStream *ost,
1520 AVCodecContext *enc;
1522 double ti1, bitrate, avg_bitrate;
1524 /* this is executed just the first time do_video_stats is called */
1526 vstats_file = fopen(vstats_filename, "w");
1533 enc = ost->st->codec;
1534 if (enc->codec_type == AVMEDIA_TYPE_VIDEO) {
1535 frame_number = ost->frame_number;
1536 fprintf(vstats_file, "frame= %5d q= %2.1f ", frame_number, enc->coded_frame->quality / (float)FF_QP2LAMBDA);
1537 if (enc->flags&CODEC_FLAG_PSNR)
1538 fprintf(vstats_file, "PSNR= %6.2f ", psnr(enc->coded_frame->error[0] / (enc->width * enc->height * 255.0 * 255.0)));
1540 fprintf(vstats_file,"f_size= %6d ", frame_size);
1541 /* compute pts value */
1542 ti1 = ost->sync_opts * av_q2d(enc->time_base);
1546 bitrate = (frame_size * 8) / av_q2d(enc->time_base) / 1000.0;
1547 avg_bitrate = (double)(video_size * 8) / ti1 / 1000.0;
1548 fprintf(vstats_file, "s_size= %8.0fkB time= %0.3f br= %7.1fkbits/s avg_br= %7.1fkbits/s ",
1549 (double)video_size / 1024, ti1, bitrate, avg_bitrate);
1550 fprintf(vstats_file, "type= %c\n", av_get_picture_type_char(enc->coded_frame->pict_type));
1554 static void print_report(OutputFile *output_files,
1555 OutputStream *ost_table, int nb_ostreams,
1556 int is_last_report, int64_t timer_start)
1560 AVFormatContext *oc;
1562 AVCodecContext *enc;
1563 int frame_number, vid, i;
1565 int64_t pts = INT64_MAX;
1566 static int64_t last_time = -1;
1567 static int qp_histogram[52];
1568 int hours, mins, secs, us;
1570 if (!print_stats && !is_last_report)
1573 if (!is_last_report) {
1575 /* display the report every 0.5 seconds */
1576 cur_time = av_gettime();
1577 if (last_time == -1) {
1578 last_time = cur_time;
1581 if ((cur_time - last_time) < 500000)
1583 last_time = cur_time;
1587 oc = output_files[0].ctx;
1589 total_size = avio_size(oc->pb);
1590 if (total_size < 0) // FIXME improve avio_size() so it works with non seekable output too
1591 total_size = avio_tell(oc->pb);
1595 for (i = 0; i < nb_ostreams; i++) {
1597 ost = &ost_table[i];
1598 enc = ost->st->codec;
1599 if (!ost->stream_copy && enc->coded_frame)
1600 q = enc->coded_frame->quality / (float)FF_QP2LAMBDA;
1601 if (vid && enc->codec_type == AVMEDIA_TYPE_VIDEO) {
1602 snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "q=%2.1f ", q);
1604 if (!vid && enc->codec_type == AVMEDIA_TYPE_VIDEO) {
1605 float t = (av_gettime() - timer_start) / 1000000.0;
1607 frame_number = ost->frame_number;
1608 snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "frame=%5d fps=%3d q=%3.1f ",
1609 frame_number, (t > 1) ? (int)(frame_number / t + 0.5) : 0, q);
1611 snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "L");
1615 if (qp >= 0 && qp < FF_ARRAY_ELEMS(qp_histogram))
1617 for (j = 0; j < 32; j++)
1618 snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "%X", (int)lrintf(log(qp_histogram[j] + 1) / log(2)));
1620 if (enc->flags&CODEC_FLAG_PSNR) {
1622 double error, error_sum = 0;
1623 double scale, scale_sum = 0;
1624 char type[3] = { 'Y','U','V' };
1625 snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "PSNR=");
1626 for (j = 0; j < 3; j++) {
1627 if (is_last_report) {
1628 error = enc->error[j];
1629 scale = enc->width * enc->height * 255.0 * 255.0 * frame_number;
1631 error = enc->coded_frame->error[j];
1632 scale = enc->width * enc->height * 255.0 * 255.0;
1638 snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "%c:%2.2f ", type[j], psnr(error / scale));
1640 snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "*:%2.2f ", psnr(error_sum / scale_sum));
1644 /* compute min output value */
1645 pts = FFMIN(pts, av_rescale_q(ost->st->pts.val,
1646 ost->st->time_base, AV_TIME_BASE_Q));
1649 secs = pts / AV_TIME_BASE;
1650 us = pts % AV_TIME_BASE;
1656 bitrate = pts ? total_size * 8 / (pts / 1000.0) : 0;
1658 snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf),
1659 "size=%8.0fkB time=", total_size / 1024.0);
1660 snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf),
1661 "%02d:%02d:%02d.%02d ", hours, mins, secs,
1662 (100 * us) / AV_TIME_BASE);
1663 snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf),
1664 "bitrate=%6.1fkbits/s", bitrate);
1666 if (nb_frames_dup || nb_frames_drop)
1667 snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), " dup=%d drop=%d",
1668 nb_frames_dup, nb_frames_drop);
1670 av_log(NULL, AV_LOG_INFO, "%s \r", buf);
1674 if (is_last_report) {
1675 int64_t raw= audio_size + video_size + extra_size;
1676 av_log(NULL, AV_LOG_INFO, "\n");
1677 av_log(NULL, AV_LOG_INFO, "video:%1.0fkB audio:%1.0fkB global headers:%1.0fkB muxing overhead %f%%\n",
1678 video_size / 1024.0,
1679 audio_size / 1024.0,
1680 extra_size / 1024.0,
1681 100.0 * (total_size - raw) / raw
1686 static void flush_encoders(OutputStream *ost_table, int nb_ostreams)
1690 for (i = 0; i < nb_ostreams; i++) {
1691 OutputStream *ost = &ost_table[i];
1692 AVCodecContext *enc = ost->st->codec;
1693 AVFormatContext *os = output_files[ost->file_index].ctx;
1695 if (!ost->encoding_needed)
1698 if (ost->st->codec->codec_type == AVMEDIA_TYPE_AUDIO && enc->frame_size <= 1)
1700 if (ost->st->codec->codec_type == AVMEDIA_TYPE_VIDEO && (os->oformat->flags & AVFMT_RAWPICTURE) && enc->codec->id == CODEC_ID_RAWVIDEO)
1706 av_init_packet(&pkt);
1707 pkt.stream_index = ost->index;
1709 switch (ost->st->codec->codec_type) {
1710 case AVMEDIA_TYPE_AUDIO:
1711 fifo_bytes = av_fifo_size(ost->fifo);
1713 /* encode any samples remaining in fifo */
1714 if (fifo_bytes > 0) {
1715 int osize = av_get_bytes_per_sample(enc->sample_fmt);
1716 int fs_tmp = enc->frame_size;
1718 av_fifo_generic_read(ost->fifo, audio_buf, fifo_bytes, NULL);
1719 if (enc->codec->capabilities & CODEC_CAP_SMALL_LAST_FRAME) {
1720 enc->frame_size = fifo_bytes / (osize * enc->channels);
1722 int frame_bytes = enc->frame_size*osize*enc->channels;
1723 if (allocated_audio_buf_size < frame_bytes)
1725 generate_silence(audio_buf+fifo_bytes, enc->sample_fmt, frame_bytes - fifo_bytes);
1728 ret = avcodec_encode_audio(enc, bit_buffer, bit_buffer_size, (short *)audio_buf);
1729 pkt.duration = av_rescale((int64_t)enc->frame_size*ost->st->time_base.den,
1730 ost->st->time_base.num, enc->sample_rate);
1731 enc->frame_size = fs_tmp;
1734 ret = avcodec_encode_audio(enc, bit_buffer, bit_buffer_size, NULL);
1737 av_log(NULL, AV_LOG_FATAL, "Audio encoding failed\n");
1741 pkt.flags |= AV_PKT_FLAG_KEY;
1743 case AVMEDIA_TYPE_VIDEO:
1744 ret = avcodec_encode_video(enc, bit_buffer, bit_buffer_size, NULL);
1746 av_log(NULL, AV_LOG_FATAL, "Video encoding failed\n");
1750 if (enc->coded_frame && enc->coded_frame->key_frame)
1751 pkt.flags |= AV_PKT_FLAG_KEY;
1752 if (ost->logfile && enc->stats_out) {
1753 fprintf(ost->logfile, "%s", enc->stats_out);
1762 pkt.data = bit_buffer;
1764 if (enc->coded_frame && enc->coded_frame->pts != AV_NOPTS_VALUE)
1765 pkt.pts = av_rescale_q(enc->coded_frame->pts, enc->time_base, ost->st->time_base);
1766 write_frame(os, &pkt, ost);
1772 * Check whether a packet from ist should be written into ost at this time
1774 static int check_output_constraints(InputStream *ist, OutputStream *ost)
1776 OutputFile *of = &output_files[ost->file_index];
1777 int ist_index = ist - input_streams;
1779 if (ost->source_index != ist_index)
1782 if (of->start_time && ist->pts < of->start_time)
1785 if (of->recording_time != INT64_MAX &&
1786 av_compare_ts(ist->pts, AV_TIME_BASE_Q, of->recording_time + of->start_time,
1787 (AVRational){ 1, 1000000 }) >= 0) {
1788 ost->is_past_recording_time = 1;
1795 static void do_streamcopy(InputStream *ist, OutputStream *ost, const AVPacket *pkt)
1797 OutputFile *of = &output_files[ost->file_index];
1798 int64_t ost_tb_start_time = av_rescale_q(of->start_time, AV_TIME_BASE_Q, ost->st->time_base);
1802 av_init_packet(&opkt);
1804 if ((!ost->frame_number && !(pkt->flags & AV_PKT_FLAG_KEY)) &&
1805 !ost->copy_initial_nonkeyframes)
1808 /* force the input stream PTS */
1809 if (ost->st->codec->codec_type == AVMEDIA_TYPE_AUDIO)
1810 audio_size += pkt->size;
1811 else if (ost->st->codec->codec_type == AVMEDIA_TYPE_VIDEO) {
1812 video_size += pkt->size;
1816 opkt.stream_index = ost->index;
1817 if (pkt->pts != AV_NOPTS_VALUE)
1818 opkt.pts = av_rescale_q(pkt->pts, ist->st->time_base, ost->st->time_base) - ost_tb_start_time;
1820 opkt.pts = AV_NOPTS_VALUE;
1822 if (pkt->dts == AV_NOPTS_VALUE)
1823 opkt.dts = av_rescale_q(ist->pts, AV_TIME_BASE_Q, ost->st->time_base);
1825 opkt.dts = av_rescale_q(pkt->dts, ist->st->time_base, ost->st->time_base);
1826 opkt.dts -= ost_tb_start_time;
1828 opkt.duration = av_rescale_q(pkt->duration, ist->st->time_base, ost->st->time_base);
1829 opkt.flags = pkt->flags;
1831 // FIXME remove the following 2 lines they shall be replaced by the bitstream filters
1832 if ( ost->st->codec->codec_id != CODEC_ID_H264
1833 && ost->st->codec->codec_id != CODEC_ID_MPEG1VIDEO
1834 && ost->st->codec->codec_id != CODEC_ID_MPEG2VIDEO
1836 if (av_parser_change(ist->st->parser, ost->st->codec, &opkt.data, &opkt.size, pkt->data, pkt->size, pkt->flags & AV_PKT_FLAG_KEY))
1837 opkt.destruct = av_destruct_packet;
1839 opkt.data = pkt->data;
1840 opkt.size = pkt->size;
1842 if (of->ctx->oformat->flags & AVFMT_RAWPICTURE) {
1843 /* store AVPicture in AVPacket, as expected by the output format */
1844 avpicture_fill(&pict, opkt.data, ost->st->codec->pix_fmt, ost->st->codec->width, ost->st->codec->height);
1845 opkt.data = (uint8_t *)&pict;
1846 opkt.size = sizeof(AVPicture);
1847 opkt.flags |= AV_PKT_FLAG_KEY;
1850 write_frame(of->ctx, &opkt, ost);
1851 ost->st->codec->frame_number++;
1852 av_free_packet(&opkt);
1855 static void rate_emu_sleep(InputStream *ist)
1857 if (input_files[ist->file_index].rate_emu) {
1858 int64_t pts = av_rescale(ist->pts, 1000000, AV_TIME_BASE);
1859 int64_t now = av_gettime() - ist->start;
1865 static int transcode_audio(InputStream *ist, AVPacket *pkt, int *got_output)
1867 AVFrame *decoded_frame;
1868 AVCodecContext *avctx = ist->st->codec;
1869 int bps = av_get_bytes_per_sample(ist->st->codec->sample_fmt);
1872 if (!ist->decoded_frame && !(ist->decoded_frame = avcodec_alloc_frame()))
1873 return AVERROR(ENOMEM);
1875 avcodec_get_frame_defaults(ist->decoded_frame);
1876 decoded_frame = ist->decoded_frame;
1878 ret = avcodec_decode_audio4(avctx, decoded_frame, got_output, pkt);
1884 /* no audio frame */
1888 /* if the decoder provides a pts, use it instead of the last packet pts.
1889 the decoder could be delaying output by a packet or more. */
1890 if (decoded_frame->pts != AV_NOPTS_VALUE)
1891 ist->next_pts = decoded_frame->pts;
1893 /* increment next_pts to use for the case where the input stream does not
1894 have timestamps or there are multiple frames in the packet */
1895 ist->next_pts += ((int64_t)AV_TIME_BASE * decoded_frame->nb_samples) /
1898 // preprocess audio (volume)
1899 if (audio_volume != 256) {
1900 int decoded_data_size = decoded_frame->nb_samples * avctx->channels * bps;
1901 void *samples = decoded_frame->data[0];
1902 switch (avctx->sample_fmt) {
1903 case AV_SAMPLE_FMT_U8:
1905 uint8_t *volp = samples;
1906 for (i = 0; i < (decoded_data_size / sizeof(*volp)); i++) {
1907 int v = (((*volp - 128) * audio_volume + 128) >> 8) + 128;
1908 *volp++ = av_clip_uint8(v);
1912 case AV_SAMPLE_FMT_S16:
1914 int16_t *volp = samples;
1915 for (i = 0; i < (decoded_data_size / sizeof(*volp)); i++) {
1916 int v = ((*volp) * audio_volume + 128) >> 8;
1917 *volp++ = av_clip_int16(v);
1921 case AV_SAMPLE_FMT_S32:
1923 int32_t *volp = samples;
1924 for (i = 0; i < (decoded_data_size / sizeof(*volp)); i++) {
1925 int64_t v = (((int64_t)*volp * audio_volume + 128) >> 8);
1926 *volp++ = av_clipl_int32(v);
1930 case AV_SAMPLE_FMT_FLT:
1932 float *volp = samples;
1933 float scale = audio_volume / 256.f;
1934 for (i = 0; i < (decoded_data_size / sizeof(*volp)); i++) {
1939 case AV_SAMPLE_FMT_DBL:
1941 double *volp = samples;
1942 double scale = audio_volume / 256.;
1943 for (i = 0; i < (decoded_data_size / sizeof(*volp)); i++) {
1949 av_log(NULL, AV_LOG_FATAL,
1950 "Audio volume adjustment on sample format %s is not supported.\n",
1951 av_get_sample_fmt_name(ist->st->codec->sample_fmt));
1956 rate_emu_sleep(ist);
1958 for (i = 0; i < nb_output_streams; i++) {
1959 OutputStream *ost = &output_streams[i];
1961 if (!check_output_constraints(ist, ost) || !ost->encoding_needed)
1963 do_audio_out(output_files[ost->file_index].ctx, ost, ist, decoded_frame);
1969 static int transcode_video(InputStream *ist, AVPacket *pkt, int *got_output, int64_t *pkt_pts)
1971 AVFrame *decoded_frame, *filtered_frame = NULL;
1972 void *buffer_to_free = NULL;
1976 int frame_available = 1;
1979 if (!ist->decoded_frame && !(ist->decoded_frame = avcodec_alloc_frame()))
1980 return AVERROR(ENOMEM);
1982 avcodec_get_frame_defaults(ist->decoded_frame);
1983 decoded_frame = ist->decoded_frame;
1984 pkt->pts = *pkt_pts;
1985 pkt->dts = ist->pts;
1986 *pkt_pts = AV_NOPTS_VALUE;
1988 ret = avcodec_decode_video2(ist->st->codec,
1989 decoded_frame, got_output, pkt);
1993 quality = same_quant ? decoded_frame->quality : 0;
1995 /* no picture yet */
1998 ist->next_pts = ist->pts = decoded_frame->best_effort_timestamp;
2000 ist->next_pts += av_rescale_q(pkt->duration, ist->st->time_base, AV_TIME_BASE_Q);
2001 else if (ist->st->codec->time_base.num != 0) {
2002 int ticks = ist->st->parser ? ist->st->parser->repeat_pict + 1 :
2003 ist->st->codec->ticks_per_frame;
2004 ist->next_pts += ((int64_t)AV_TIME_BASE *
2005 ist->st->codec->time_base.num * ticks) /
2006 ist->st->codec->time_base.den;
2009 pre_process_video_frame(ist, (AVPicture *)decoded_frame, &buffer_to_free);
2011 rate_emu_sleep(ist);
2013 for (i = 0; i < nb_output_streams; i++) {
2014 OutputStream *ost = &output_streams[i];
2015 int frame_size, resample_changed;
2017 if (!check_output_constraints(ist, ost) || !ost->encoding_needed)
2021 resample_changed = ost->resample_width != decoded_frame->width ||
2022 ost->resample_height != decoded_frame->height ||
2023 ost->resample_pix_fmt != decoded_frame->format;
2024 if (resample_changed) {
2025 av_log(NULL, AV_LOG_INFO,
2026 "Input stream #%d:%d frame changed from size:%dx%d fmt:%s to size:%dx%d fmt:%s\n",
2027 ist->file_index, ist->st->index,
2028 ost->resample_width, ost->resample_height, av_get_pix_fmt_name(ost->resample_pix_fmt),
2029 decoded_frame->width, decoded_frame->height, av_get_pix_fmt_name(decoded_frame->format));
2031 avfilter_graph_free(&ost->graph);
2032 if (configure_video_filters(ist, ost)) {
2033 av_log(NULL, AV_LOG_FATAL, "Error reinitializing filters!\n");
2037 ost->resample_width = decoded_frame->width;
2038 ost->resample_height = decoded_frame->height;
2039 ost->resample_pix_fmt = decoded_frame->format;
2042 if (!decoded_frame->sample_aspect_ratio.num)
2043 decoded_frame->sample_aspect_ratio = ist->st->sample_aspect_ratio;
2044 decoded_frame->pts = ist->pts;
2045 if (ist->st->codec->codec->capabilities & CODEC_CAP_DR1) {
2046 FrameBuffer *buf = decoded_frame->opaque;
2047 AVFilterBufferRef *fb = avfilter_get_video_buffer_ref_from_arrays(
2048 decoded_frame->data, decoded_frame->linesize,
2049 AV_PERM_READ | AV_PERM_PRESERVE,
2050 ist->st->codec->width, ist->st->codec->height,
2051 ist->st->codec->pix_fmt);
2053 avfilter_copy_frame_props(fb, decoded_frame);
2055 fb->buf->priv = buf;
2056 fb->buf->free = filter_release_buffer;
2059 av_buffersrc_buffer(ost->input_video_filter, fb);
2061 av_vsrc_buffer_add_frame(ost->input_video_filter, decoded_frame, AV_VSRC_BUF_FLAG_OVERWRITE);
2063 if (!ist->filtered_frame && !(ist->filtered_frame = avcodec_alloc_frame())) {
2064 av_free(buffer_to_free);
2065 return AVERROR(ENOMEM);
2067 avcodec_get_frame_defaults(ist->filtered_frame);
2068 filtered_frame = ist->filtered_frame;
2070 frame_available = avfilter_poll_frame(ost->output_video_filter->inputs[0]);
2071 while (frame_available) {
2072 if (ost->output_video_filter) {
2073 AVRational ist_pts_tb = ost->output_video_filter->inputs[0]->time_base;
2074 if (av_buffersink_get_buffer_ref(ost->output_video_filter, &ost->picref, 0) < 0)
2077 avfilter_fill_frame_from_video_buffer_ref(filtered_frame, ost->picref);
2078 ist->pts = av_rescale_q(ost->picref->pts, ist_pts_tb, AV_TIME_BASE_Q);
2081 if (ost->picref->video && !ost->frame_aspect_ratio)
2082 ost->st->codec->sample_aspect_ratio = ost->picref->video->sample_aspect_ratio;
2084 filtered_frame = decoded_frame;
2087 do_video_out(output_files[ost->file_index].ctx, ost, ist, filtered_frame, &frame_size,
2088 same_quant ? quality : ost->st->codec->global_quality);
2089 if (vstats_filename && frame_size)
2090 do_video_stats(output_files[ost->file_index].ctx, ost, frame_size);
2093 frame_available = ost->output_video_filter && avfilter_poll_frame(ost->output_video_filter->inputs[0]);
2095 avfilter_unref_buffer(ost->picref);
2100 av_free(buffer_to_free);
2104 static int transcode_subtitles(InputStream *ist, AVPacket *pkt, int *got_output)
2106 AVSubtitle subtitle;
2107 int i, ret = avcodec_decode_subtitle2(ist->st->codec,
2108 &subtitle, got_output, pkt);
2114 rate_emu_sleep(ist);
2116 for (i = 0; i < nb_output_streams; i++) {
2117 OutputStream *ost = &output_streams[i];
2119 if (!check_output_constraints(ist, ost) || !ost->encoding_needed)
2122 do_subtitle_out(output_files[ost->file_index].ctx, ost, ist, &subtitle, pkt->pts);
2125 avsubtitle_free(&subtitle);
2129 /* pkt = NULL means EOF (needed to flush decoder buffers) */
2130 static int output_packet(InputStream *ist,
2131 OutputStream *ost_table, int nb_ostreams,
2132 const AVPacket *pkt)
2136 int64_t pkt_pts = AV_NOPTS_VALUE;
2139 if (ist->next_pts == AV_NOPTS_VALUE)
2140 ist->next_pts = ist->pts;
2144 av_init_packet(&avpkt);
2152 if (pkt->dts != AV_NOPTS_VALUE)
2153 ist->next_pts = ist->pts = av_rescale_q(pkt->dts, ist->st->time_base, AV_TIME_BASE_Q);
2154 if (pkt->pts != AV_NOPTS_VALUE)
2155 pkt_pts = av_rescale_q(pkt->pts, ist->st->time_base, AV_TIME_BASE_Q);
2157 // while we have more to decode or while the decoder did output something on EOF
2158 while (ist->decoding_needed && (avpkt.size > 0 || (!pkt && got_output))) {
2162 ist->pts = ist->next_pts;
2164 if (avpkt.size && avpkt.size != pkt->size) {
2165 av_log(NULL, ist->showed_multi_packet_warning ? AV_LOG_VERBOSE : AV_LOG_WARNING,
2166 "Multiple frames in a packet from stream %d\n", pkt->stream_index);
2167 ist->showed_multi_packet_warning = 1;
2170 switch (ist->st->codec->codec_type) {
2171 case AVMEDIA_TYPE_AUDIO:
2172 ret = transcode_audio (ist, &avpkt, &got_output);
2174 case AVMEDIA_TYPE_VIDEO:
2175 ret = transcode_video (ist, &avpkt, &got_output, &pkt_pts);
2177 case AVMEDIA_TYPE_SUBTITLE:
2178 ret = transcode_subtitles(ist, &avpkt, &got_output);
2186 // touch data and size only if not EOF
2196 /* handle stream copy */
2197 if (!ist->decoding_needed) {
2198 rate_emu_sleep(ist);
2199 ist->pts = ist->next_pts;
2200 switch (ist->st->codec->codec_type) {
2201 case AVMEDIA_TYPE_AUDIO:
2202 ist->next_pts += ((int64_t)AV_TIME_BASE * ist->st->codec->frame_size) /
2203 ist->st->codec->sample_rate;
2205 case AVMEDIA_TYPE_VIDEO:
2206 if (ist->st->codec->time_base.num != 0) {
2207 int ticks = ist->st->parser ? ist->st->parser->repeat_pict + 1 : ist->st->codec->ticks_per_frame;
2208 ist->next_pts += ((int64_t)AV_TIME_BASE *
2209 ist->st->codec->time_base.num * ticks) /
2210 ist->st->codec->time_base.den;
2215 for (i = 0; pkt && i < nb_ostreams; i++) {
2216 OutputStream *ost = &ost_table[i];
2218 if (!check_output_constraints(ist, ost) || ost->encoding_needed)
2221 do_streamcopy(ist, ost, pkt);
2227 static void print_sdp(OutputFile *output_files, int n)
2231 AVFormatContext **avc = av_malloc(sizeof(*avc) * n);
2235 for (i = 0; i < n; i++)
2236 avc[i] = output_files[i].ctx;
2238 av_sdp_create(avc, n, sdp, sizeof(sdp));
2239 printf("SDP:\n%s\n", sdp);
2244 static int init_input_stream(int ist_index, OutputStream *output_streams, int nb_output_streams,
2245 char *error, int error_len)
2248 InputStream *ist = &input_streams[ist_index];
2249 if (ist->decoding_needed) {
2250 AVCodec *codec = ist->dec;
2252 snprintf(error, error_len, "Decoder (codec id %d) not found for input stream #%d:%d",
2253 ist->st->codec->codec_id, ist->file_index, ist->st->index);
2254 return AVERROR(EINVAL);
2257 if (codec->type == AVMEDIA_TYPE_VIDEO && codec->capabilities & CODEC_CAP_DR1) {
2258 ist->st->codec->get_buffer = codec_get_buffer;
2259 ist->st->codec->release_buffer = codec_release_buffer;
2260 ist->st->codec->opaque = ist;
2263 if (avcodec_open2(ist->st->codec, codec, &ist->opts) < 0) {
2264 snprintf(error, error_len, "Error while opening decoder for input stream #%d:%d",
2265 ist->file_index, ist->st->index);
2266 return AVERROR(EINVAL);
2268 assert_codec_experimental(ist->st->codec, 0);
2269 assert_avoptions(ist->opts);
2272 ist->pts = ist->st->avg_frame_rate.num ? - ist->st->codec->has_b_frames * AV_TIME_BASE / av_q2d(ist->st->avg_frame_rate) : 0;
2273 ist->next_pts = AV_NOPTS_VALUE;
2279 static int transcode_init(OutputFile *output_files,
2280 int nb_output_files,
2281 InputFile *input_files,
2284 int ret = 0, i, j, k;
2285 AVFormatContext *oc;
2286 AVCodecContext *codec, *icodec;
2292 /* init framerate emulation */
2293 for (i = 0; i < nb_input_files; i++) {
2294 InputFile *ifile = &input_files[i];
2295 if (ifile->rate_emu)
2296 for (j = 0; j < ifile->nb_streams; j++)
2297 input_streams[j + ifile->ist_index].start = av_gettime();
2300 /* output stream init */
2301 for (i = 0; i < nb_output_files; i++) {
2302 oc = output_files[i].ctx;
2303 if (!oc->nb_streams && !(oc->oformat->flags & AVFMT_NOSTREAMS)) {
2304 av_dump_format(oc, i, oc->filename, 1);
2305 av_log(NULL, AV_LOG_ERROR, "Output file #%d does not contain any stream\n", i);
2306 return AVERROR(EINVAL);
2310 /* for each output stream, we compute the right encoding parameters */
2311 for (i = 0; i < nb_output_streams; i++) {
2312 ost = &output_streams[i];
2313 oc = output_files[ost->file_index].ctx;
2314 ist = &input_streams[ost->source_index];
2316 if (ost->attachment_filename)
2319 codec = ost->st->codec;
2320 icodec = ist->st->codec;
2322 ost->st->disposition = ist->st->disposition;
2323 codec->bits_per_raw_sample = icodec->bits_per_raw_sample;
2324 codec->chroma_sample_location = icodec->chroma_sample_location;
2326 if (ost->stream_copy) {
2327 uint64_t extra_size = (uint64_t)icodec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE;
2329 if (extra_size > INT_MAX) {
2330 return AVERROR(EINVAL);
2333 /* if stream_copy is selected, no need to decode or encode */
2334 codec->codec_id = icodec->codec_id;
2335 codec->codec_type = icodec->codec_type;
2337 if (!codec->codec_tag) {
2338 if (!oc->oformat->codec_tag ||
2339 av_codec_get_id (oc->oformat->codec_tag, icodec->codec_tag) == codec->codec_id ||
2340 av_codec_get_tag(oc->oformat->codec_tag, icodec->codec_id) <= 0)
2341 codec->codec_tag = icodec->codec_tag;
2344 codec->bit_rate = icodec->bit_rate;
2345 codec->rc_max_rate = icodec->rc_max_rate;
2346 codec->rc_buffer_size = icodec->rc_buffer_size;
2347 codec->field_order = icodec->field_order;
2348 codec->extradata = av_mallocz(extra_size);
2349 if (!codec->extradata) {
2350 return AVERROR(ENOMEM);
2352 memcpy(codec->extradata, icodec->extradata, icodec->extradata_size);
2354 codec->extradata_size = icodec->extradata_size;
2356 codec->time_base = icodec->time_base;
2357 codec->time_base.num *= icodec->ticks_per_frame;
2358 av_reduce(&codec->time_base.num, &codec->time_base.den,
2359 codec->time_base.num, codec->time_base.den, INT_MAX);
2361 codec->time_base = ist->st->time_base;
2363 switch (codec->codec_type) {
2364 case AVMEDIA_TYPE_AUDIO:
2365 if (audio_volume != 256) {
2366 av_log(NULL, AV_LOG_FATAL, "-acodec copy and -vol are incompatible (frames are not decoded)\n");
2369 codec->channel_layout = icodec->channel_layout;
2370 codec->sample_rate = icodec->sample_rate;
2371 codec->channels = icodec->channels;
2372 codec->frame_size = icodec->frame_size;
2373 codec->audio_service_type = icodec->audio_service_type;
2374 codec->block_align = icodec->block_align;
2376 case AVMEDIA_TYPE_VIDEO:
2377 codec->pix_fmt = icodec->pix_fmt;
2378 codec->width = icodec->width;
2379 codec->height = icodec->height;
2380 codec->has_b_frames = icodec->has_b_frames;
2381 if (!codec->sample_aspect_ratio.num) {
2382 codec->sample_aspect_ratio =
2383 ost->st->sample_aspect_ratio =
2384 ist->st->sample_aspect_ratio.num ? ist->st->sample_aspect_ratio :
2385 ist->st->codec->sample_aspect_ratio.num ?
2386 ist->st->codec->sample_aspect_ratio : (AVRational){0, 1};
2389 case AVMEDIA_TYPE_SUBTITLE:
2390 codec->width = icodec->width;
2391 codec->height = icodec->height;
2393 case AVMEDIA_TYPE_DATA:
2394 case AVMEDIA_TYPE_ATTACHMENT:
2401 ost->enc = avcodec_find_encoder(ost->st->codec->codec_id);
2403 ist->decoding_needed = 1;
2404 ost->encoding_needed = 1;
2406 switch (codec->codec_type) {
2407 case AVMEDIA_TYPE_AUDIO:
2408 ost->fifo = av_fifo_alloc(1024);
2410 return AVERROR(ENOMEM);
2412 ost->reformat_pair = MAKE_SFMT_PAIR(AV_SAMPLE_FMT_NONE,AV_SAMPLE_FMT_NONE);
2414 if (!codec->sample_rate)
2415 codec->sample_rate = icodec->sample_rate;
2416 choose_sample_rate(ost->st, ost->enc);
2417 codec->time_base = (AVRational){ 1, codec->sample_rate };
2419 if (codec->sample_fmt == AV_SAMPLE_FMT_NONE)
2420 codec->sample_fmt = icodec->sample_fmt;
2421 choose_sample_fmt(ost->st, ost->enc);
2423 if (!codec->channels) {
2424 codec->channels = icodec->channels;
2425 codec->channel_layout = icodec->channel_layout;
2427 if (av_get_channel_layout_nb_channels(codec->channel_layout) != codec->channels)
2428 codec->channel_layout = 0;
2430 ost->audio_resample = codec-> sample_rate != icodec->sample_rate || audio_sync_method > 1;
2431 icodec->request_channels = codec-> channels;
2432 ost->resample_sample_fmt = icodec->sample_fmt;
2433 ost->resample_sample_rate = icodec->sample_rate;
2434 ost->resample_channels = icodec->channels;
2436 case AVMEDIA_TYPE_VIDEO:
2437 if (codec->pix_fmt == PIX_FMT_NONE)
2438 codec->pix_fmt = icodec->pix_fmt;
2439 choose_pixel_fmt(ost->st, ost->enc);
2441 if (ost->st->codec->pix_fmt == PIX_FMT_NONE) {
2442 av_log(NULL, AV_LOG_FATAL, "Video pixel format is unknown, stream cannot be encoded\n");
2446 if (!codec->width || !codec->height) {
2447 codec->width = icodec->width;
2448 codec->height = icodec->height;
2451 ost->video_resample = codec->width != icodec->width ||
2452 codec->height != icodec->height ||
2453 codec->pix_fmt != icodec->pix_fmt;
2454 if (ost->video_resample) {
2455 codec->bits_per_raw_sample= frame_bits_per_raw_sample;
2458 ost->resample_height = icodec->height;
2459 ost->resample_width = icodec->width;
2460 ost->resample_pix_fmt = icodec->pix_fmt;
2462 if (!ost->frame_rate.num)
2463 ost->frame_rate = ist->st->r_frame_rate.num ? ist->st->r_frame_rate : (AVRational) { 25, 1 };
2464 if (ost->enc && ost->enc->supported_framerates && !ost->force_fps) {
2465 int idx = av_find_nearest_q_idx(ost->frame_rate, ost->enc->supported_framerates);
2466 ost->frame_rate = ost->enc->supported_framerates[idx];
2468 codec->time_base = (AVRational){ost->frame_rate.den, ost->frame_rate.num};
2469 if( av_q2d(codec->time_base) < 0.001 && video_sync_method
2470 && (video_sync_method==1 || (video_sync_method<0 && !(oc->oformat->flags & AVFMT_VARIABLE_FPS)))){
2471 av_log(oc, AV_LOG_WARNING, "Frame rate very high for a muxer not effciciently supporting it.\n"
2472 "Please consider specifiying a lower framerate, a different muxer or -vsync 2\n");
2476 if (configure_video_filters(ist, ost)) {
2477 av_log(NULL, AV_LOG_FATAL, "Error opening filters!\n");
2482 case AVMEDIA_TYPE_SUBTITLE:
2489 if (codec->codec_id != CODEC_ID_H264 &&
2490 (codec->flags & (CODEC_FLAG_PASS1 | CODEC_FLAG_PASS2))) {
2491 char logfilename[1024];
2494 snprintf(logfilename, sizeof(logfilename), "%s-%d.log",
2495 pass_logfilename_prefix ? pass_logfilename_prefix : DEFAULT_PASS_LOGFILENAME_PREFIX,
2497 if (codec->flags & CODEC_FLAG_PASS1) {
2498 f = fopen(logfilename, "wb");
2500 av_log(NULL, AV_LOG_FATAL, "Cannot write log file '%s' for pass-1 encoding: %s\n",
2501 logfilename, strerror(errno));
2507 size_t logbuffer_size;
2508 if (cmdutils_read_file(logfilename, &logbuffer, &logbuffer_size) < 0) {
2509 av_log(NULL, AV_LOG_FATAL, "Error reading log file '%s' for pass-2 encoding\n",
2513 codec->stats_in = logbuffer;
2517 if (codec->codec_type == AVMEDIA_TYPE_VIDEO) {
2518 /* maximum video buffer size is 6-bytes per pixel, plus DPX header size */
2519 int size = codec->width * codec->height;
2520 bit_buffer_size = FFMAX(bit_buffer_size, 6 * size + 1664);
2525 bit_buffer = av_malloc(bit_buffer_size);
2527 av_log(NULL, AV_LOG_ERROR, "Cannot allocate %d bytes output buffer\n",
2529 return AVERROR(ENOMEM);
2532 /* open each encoder */
2533 for (i = 0; i < nb_output_streams; i++) {
2534 ost = &output_streams[i];
2535 if (ost->encoding_needed) {
2536 AVCodec *codec = ost->enc;
2537 AVCodecContext *dec = input_streams[ost->source_index].st->codec;
2539 snprintf(error, sizeof(error), "Encoder (codec id %d) not found for output stream #%d:%d",
2540 ost->st->codec->codec_id, ost->file_index, ost->index);
2541 ret = AVERROR(EINVAL);
2544 if (dec->subtitle_header) {
2545 ost->st->codec->subtitle_header = av_malloc(dec->subtitle_header_size);
2546 if (!ost->st->codec->subtitle_header) {
2547 ret = AVERROR(ENOMEM);
2550 memcpy(ost->st->codec->subtitle_header, dec->subtitle_header, dec->subtitle_header_size);
2551 ost->st->codec->subtitle_header_size = dec->subtitle_header_size;
2553 if (avcodec_open2(ost->st->codec, codec, &ost->opts) < 0) {
2554 snprintf(error, sizeof(error), "Error while opening encoder for output stream #%d:%d - maybe incorrect parameters such as bit_rate, rate, width or height",
2555 ost->file_index, ost->index);
2556 ret = AVERROR(EINVAL);
2559 assert_codec_experimental(ost->st->codec, 1);
2560 assert_avoptions(ost->opts);
2561 if (ost->st->codec->bit_rate && ost->st->codec->bit_rate < 1000)
2562 av_log(NULL, AV_LOG_WARNING, "The bitrate parameter is set too low."
2563 "It takes bits/s as argument, not kbits/s\n");
2564 extra_size += ost->st->codec->extradata_size;
2566 if (ost->st->codec->me_threshold)
2567 input_streams[ost->source_index].st->codec->debug |= FF_DEBUG_MV;
2571 /* init input streams */
2572 for (i = 0; i < nb_input_streams; i++)
2573 if ((ret = init_input_stream(i, output_streams, nb_output_streams, error, sizeof(error))) < 0)
2576 /* discard unused programs */
2577 for (i = 0; i < nb_input_files; i++) {
2578 InputFile *ifile = &input_files[i];
2579 for (j = 0; j < ifile->ctx->nb_programs; j++) {
2580 AVProgram *p = ifile->ctx->programs[j];
2581 int discard = AVDISCARD_ALL;
2583 for (k = 0; k < p->nb_stream_indexes; k++)
2584 if (!input_streams[ifile->ist_index + p->stream_index[k]].discard) {
2585 discard = AVDISCARD_DEFAULT;
2588 p->discard = discard;
2592 /* open files and write file headers */
2593 for (i = 0; i < nb_output_files; i++) {
2594 oc = output_files[i].ctx;
2595 oc->interrupt_callback = int_cb;
2596 if (avformat_write_header(oc, &output_files[i].opts) < 0) {
2597 snprintf(error, sizeof(error), "Could not write header for output file #%d (incorrect codec parameters ?)", i);
2598 ret = AVERROR(EINVAL);
2601 // assert_avoptions(output_files[i].opts);
2602 if (strcmp(oc->oformat->name, "rtp")) {
2608 /* dump the file output parameters - cannot be done before in case
2610 for (i = 0; i < nb_output_files; i++) {
2611 av_dump_format(output_files[i].ctx, i, output_files[i].ctx->filename, 1);
2614 /* dump the stream mapping */
2615 av_log(NULL, AV_LOG_INFO, "Stream mapping:\n");
2616 for (i = 0; i < nb_output_streams; i++) {
2617 ost = &output_streams[i];
2619 if (ost->attachment_filename) {
2620 /* an attached file */
2621 av_log(NULL, AV_LOG_INFO, " File %s -> Stream #%d:%d\n",
2622 ost->attachment_filename, ost->file_index, ost->index);
2625 av_log(NULL, AV_LOG_INFO, " Stream #%d:%d -> #%d:%d",
2626 input_streams[ost->source_index].file_index,
2627 input_streams[ost->source_index].st->index,
2630 if (ost->sync_ist != &input_streams[ost->source_index])
2631 av_log(NULL, AV_LOG_INFO, " [sync #%d:%d]",
2632 ost->sync_ist->file_index,
2633 ost->sync_ist->st->index);
2634 if (ost->stream_copy)
2635 av_log(NULL, AV_LOG_INFO, " (copy)");
2637 av_log(NULL, AV_LOG_INFO, " (%s -> %s)", input_streams[ost->source_index].dec ?
2638 input_streams[ost->source_index].dec->name : "?",
2639 ost->enc ? ost->enc->name : "?");
2640 av_log(NULL, AV_LOG_INFO, "\n");
2644 av_log(NULL, AV_LOG_ERROR, "%s\n", error);
2649 print_sdp(output_files, nb_output_files);
2656 * The following code is the main loop of the file converter
2658 static int transcode(OutputFile *output_files,
2659 int nb_output_files,
2660 InputFile *input_files,
2664 AVFormatContext *is, *os;
2668 int no_packet_count = 0;
2669 int64_t timer_start;
2672 if (!(no_packet = av_mallocz(nb_input_files)))
2675 ret = transcode_init(output_files, nb_output_files, input_files, nb_input_files);
2680 av_log(NULL, AV_LOG_INFO, "Press [q] to stop, [?] for help\n");
2681 avio_set_interrupt_cb(decode_interrupt_cb);
2685 timer_start = av_gettime();
2687 for (; received_sigterm == 0;) {
2688 int file_index, ist_index;
2693 ipts_min = INT64_MAX;
2695 /* if 'q' pressed, exits */
2699 /* read_key() returns 0 on EOF */
2703 if (key == '+') av_log_set_level(av_log_get_level()+10);
2704 if (key == '-') av_log_set_level(av_log_get_level()-10);
2705 if (key == 's') qp_hist ^= 1;
2708 do_hex_dump = do_pkt_dump = 0;
2709 } else if(do_pkt_dump){
2713 av_log_set_level(AV_LOG_DEBUG);
2715 if (key == 'd' || key == 'D'){
2718 debug = input_streams[0].st->codec->debug<<1;
2719 if(!debug) debug = 1;
2720 while(debug & (FF_DEBUG_DCT_COEFF|FF_DEBUG_VIS_QP|FF_DEBUG_VIS_MB_TYPE)) //unsupported, would just crash
2723 scanf("%d", &debug);
2724 for(i=0;i<nb_input_streams;i++) {
2725 input_streams[i].st->codec->debug = debug;
2727 for(i=0;i<nb_output_streams;i++) {
2728 ost = &output_streams[i];
2729 ost->st->codec->debug = debug;
2731 if(debug) av_log_set_level(AV_LOG_DEBUG);
2732 fprintf(stderr,"debug=%d\n", debug);
2735 fprintf(stderr, "key function\n"
2736 "? show this help\n"
2737 "+ increase verbosity\n"
2738 "- decrease verbosity\n"
2739 "D cycle through available debug modes\n"
2740 "h dump packets/hex press to cycle through the 3 states\n"
2742 "s Show QP histogram\n"
2747 /* select the stream that we must read now by looking at the
2748 smallest output pts */
2750 for (i = 0; i < nb_output_streams; i++) {
2754 ost = &output_streams[i];
2755 of = &output_files[ost->file_index];
2756 os = output_files[ost->file_index].ctx;
2757 ist = &input_streams[ost->source_index];
2758 if (ost->is_past_recording_time || no_packet[ist->file_index] ||
2759 (os->pb && avio_tell(os->pb) >= of->limit_filesize))
2761 opts = ost->st->pts.val * av_q2d(ost->st->time_base);
2763 if (!input_files[ist->file_index].eof_reached) {
2764 if (ipts < ipts_min) {
2767 file_index = ist->file_index;
2769 if (opts < opts_min) {
2771 if (!input_sync) file_index = ist->file_index;
2774 if (ost->frame_number >= ost->max_frames) {
2776 for (j = 0; j < of->ctx->nb_streams; j++)
2777 output_streams[of->ost_index + j].is_past_recording_time = 1;
2781 /* if none, if is finished */
2782 if (file_index < 0) {
2783 if (no_packet_count) {
2784 no_packet_count = 0;
2785 memset(no_packet, 0, nb_input_files);
2792 /* read a frame from it and output it in the fifo */
2793 is = input_files[file_index].ctx;
2794 ret = av_read_frame(is, &pkt);
2795 if (ret == AVERROR(EAGAIN)) {
2796 no_packet[file_index] = 1;
2801 input_files[file_index].eof_reached = 1;
2808 no_packet_count = 0;
2809 memset(no_packet, 0, nb_input_files);
2812 av_pkt_dump_log2(NULL, AV_LOG_DEBUG, &pkt, do_hex_dump,
2813 is->streams[pkt.stream_index]);
2815 /* the following test is needed in case new streams appear
2816 dynamically in stream : we ignore them */
2817 if (pkt.stream_index >= input_files[file_index].nb_streams)
2818 goto discard_packet;
2819 ist_index = input_files[file_index].ist_index + pkt.stream_index;
2820 ist = &input_streams[ist_index];
2822 goto discard_packet;
2824 if (pkt.dts != AV_NOPTS_VALUE)
2825 pkt.dts += av_rescale_q(input_files[ist->file_index].ts_offset, AV_TIME_BASE_Q, ist->st->time_base);
2826 if (pkt.pts != AV_NOPTS_VALUE)
2827 pkt.pts += av_rescale_q(input_files[ist->file_index].ts_offset, AV_TIME_BASE_Q, ist->st->time_base);
2829 if (pkt.pts != AV_NOPTS_VALUE)
2830 pkt.pts *= ist->ts_scale;
2831 if (pkt.dts != AV_NOPTS_VALUE)
2832 pkt.dts *= ist->ts_scale;
2834 //fprintf(stderr, "next:%"PRId64" dts:%"PRId64" off:%"PRId64" %d\n",
2836 // pkt.dts, input_files[ist->file_index].ts_offset,
2837 // ist->st->codec->codec_type);
2838 if (pkt.dts != AV_NOPTS_VALUE && ist->next_pts != AV_NOPTS_VALUE
2839 && (is->iformat->flags & AVFMT_TS_DISCONT)) {
2840 int64_t pkt_dts = av_rescale_q(pkt.dts, ist->st->time_base, AV_TIME_BASE_Q);
2841 int64_t delta = pkt_dts - ist->next_pts;
2842 if ((FFABS(delta) > 1LL * dts_delta_threshold * AV_TIME_BASE || pkt_dts + 1 < ist->pts) && !copy_ts) {
2843 input_files[ist->file_index].ts_offset -= delta;
2844 av_log(NULL, AV_LOG_DEBUG,
2845 "timestamp discontinuity %"PRId64", new offset= %"PRId64"\n",
2846 delta, input_files[ist->file_index].ts_offset);
2847 pkt.dts-= av_rescale_q(delta, AV_TIME_BASE_Q, ist->st->time_base);
2848 if (pkt.pts != AV_NOPTS_VALUE)
2849 pkt.pts-= av_rescale_q(delta, AV_TIME_BASE_Q, ist->st->time_base);
2853 // fprintf(stderr,"read #%d.%d size=%d\n", ist->file_index, ist->st->index, pkt.size);
2854 if (output_packet(ist, output_streams, nb_output_streams, &pkt) < 0) {
2856 av_log(NULL, AV_LOG_ERROR, "Error while decoding stream #%d:%d\n",
2857 ist->file_index, ist->st->index);
2860 av_free_packet(&pkt);
2865 av_free_packet(&pkt);
2867 /* dump report by using the output first video and audio streams */
2868 print_report(output_files, output_streams, nb_output_streams, 0, timer_start);
2871 /* at the end of stream, we must flush the decoder buffers */
2872 for (i = 0; i < nb_input_streams; i++) {
2873 ist = &input_streams[i];
2874 if (ist->decoding_needed) {
2875 output_packet(ist, output_streams, nb_output_streams, NULL);
2878 flush_encoders(output_streams, nb_output_streams);
2882 /* write the trailer if needed and close file */
2883 for (i = 0; i < nb_output_files; i++) {
2884 os = output_files[i].ctx;
2885 av_write_trailer(os);
2888 /* dump report by using the first video and audio streams */
2889 print_report(output_files, output_streams, nb_output_streams, 1, timer_start);
2891 /* close each encoder */
2892 for (i = 0; i < nb_output_streams; i++) {
2893 ost = &output_streams[i];
2894 if (ost->encoding_needed) {
2895 av_freep(&ost->st->codec->stats_in);
2896 avcodec_close(ost->st->codec);
2899 avfilter_graph_free(&ost->graph);
2903 /* close each decoder */
2904 for (i = 0; i < nb_input_streams; i++) {
2905 ist = &input_streams[i];
2906 if (ist->decoding_needed) {
2907 avcodec_close(ist->st->codec);
2915 av_freep(&bit_buffer);
2916 av_freep(&no_packet);
2918 if (output_streams) {
2919 for (i = 0; i < nb_output_streams; i++) {
2920 ost = &output_streams[i];
2922 if (ost->stream_copy)
2923 av_freep(&ost->st->codec->extradata);
2925 fclose(ost->logfile);
2926 ost->logfile = NULL;
2928 av_fifo_free(ost->fifo); /* works even if fifo is not
2929 initialized but set to zero */
2930 av_freep(&ost->st->codec->subtitle_header);
2931 av_free(ost->resample_frame.data[0]);
2932 av_free(ost->forced_kf_pts);
2933 if (ost->video_resample)
2934 sws_freeContext(ost->img_resample_ctx);
2936 audio_resample_close(ost->resample);
2937 if (ost->reformat_ctx)
2938 av_audio_convert_free(ost->reformat_ctx);
2939 av_dict_free(&ost->opts);
2946 static double parse_frame_aspect_ratio(const char *arg)
2953 p = strchr(arg, ':');
2955 x = strtol(arg, &end, 10);
2957 y = strtol(end + 1, &end, 10);
2959 ar = (double)x / (double)y;
2961 ar = strtod(arg, NULL);
2964 av_log(NULL, AV_LOG_FATAL, "Incorrect aspect ratio specification.\n");
2970 static int opt_audio_codec(OptionsContext *o, const char *opt, const char *arg)
2972 return parse_option(o, "codec:a", arg, options);
2975 static int opt_video_codec(OptionsContext *o, const char *opt, const char *arg)
2977 return parse_option(o, "codec:v", arg, options);
2980 static int opt_subtitle_codec(OptionsContext *o, const char *opt, const char *arg)
2982 return parse_option(o, "codec:s", arg, options);
2985 static int opt_data_codec(OptionsContext *o, const char *opt, const char *arg)
2987 return parse_option(o, "codec:d", arg, options);
2990 static int opt_map(OptionsContext *o, const char *opt, const char *arg)
2992 StreamMap *m = NULL;
2993 int i, negative = 0, file_idx;
2994 int sync_file_idx = -1, sync_stream_idx;
3002 map = av_strdup(arg);
3004 /* parse sync stream first, just pick first matching stream */
3005 if (sync = strchr(map, ',')) {
3007 sync_file_idx = strtol(sync + 1, &sync, 0);
3008 if (sync_file_idx >= nb_input_files || sync_file_idx < 0) {
3009 av_log(NULL, AV_LOG_FATAL, "Invalid sync file index: %d.\n", sync_file_idx);
3014 for (i = 0; i < input_files[sync_file_idx].nb_streams; i++)
3015 if (check_stream_specifier(input_files[sync_file_idx].ctx,
3016 input_files[sync_file_idx].ctx->streams[i], sync) == 1) {
3017 sync_stream_idx = i;
3020 if (i == input_files[sync_file_idx].nb_streams) {
3021 av_log(NULL, AV_LOG_FATAL, "Sync stream specification in map %s does not "
3022 "match any streams.\n", arg);
3028 file_idx = strtol(map, &p, 0);
3029 if (file_idx >= nb_input_files || file_idx < 0) {
3030 av_log(NULL, AV_LOG_FATAL, "Invalid input file index: %d.\n", file_idx);
3034 /* disable some already defined maps */
3035 for (i = 0; i < o->nb_stream_maps; i++) {
3036 m = &o->stream_maps[i];
3037 if (file_idx == m->file_index &&
3038 check_stream_specifier(input_files[m->file_index].ctx,
3039 input_files[m->file_index].ctx->streams[m->stream_index],
3040 *p == ':' ? p + 1 : p) > 0)
3044 for (i = 0; i < input_files[file_idx].nb_streams; i++) {
3045 if (check_stream_specifier(input_files[file_idx].ctx, input_files[file_idx].ctx->streams[i],
3046 *p == ':' ? p + 1 : p) <= 0)
3048 o->stream_maps = grow_array(o->stream_maps, sizeof(*o->stream_maps),
3049 &o->nb_stream_maps, o->nb_stream_maps + 1);
3050 m = &o->stream_maps[o->nb_stream_maps - 1];
3052 m->file_index = file_idx;
3053 m->stream_index = i;
3055 if (sync_file_idx >= 0) {
3056 m->sync_file_index = sync_file_idx;
3057 m->sync_stream_index = sync_stream_idx;
3059 m->sync_file_index = file_idx;
3060 m->sync_stream_index = i;
3065 av_log(NULL, AV_LOG_FATAL, "Stream map '%s' matches no streams.\n", arg);
3073 static int opt_attach(OptionsContext *o, const char *opt, const char *arg)
3075 o->attachments = grow_array(o->attachments, sizeof(*o->attachments),
3076 &o->nb_attachments, o->nb_attachments + 1);
3077 o->attachments[o->nb_attachments - 1] = arg;
3082 * Parse a metadata specifier in arg.
3083 * @param type metadata type is written here -- g(lobal)/s(tream)/c(hapter)/p(rogram)
3084 * @param index for type c/p, chapter/program index is written here
3085 * @param stream_spec for type s, the stream specifier is written here
3087 static void parse_meta_type(char *arg, char *type, int *index, const char **stream_spec)
3095 if (*(++arg) && *arg != ':') {
3096 av_log(NULL, AV_LOG_FATAL, "Invalid metadata specifier %s.\n", arg);
3099 *stream_spec = *arg == ':' ? arg + 1 : "";
3103 if (*(++arg) == ':')
3104 *index = strtol(++arg, NULL, 0);
3107 av_log(NULL, AV_LOG_FATAL, "Invalid metadata type %c.\n", *arg);
3114 static int copy_metadata(char *outspec, char *inspec, AVFormatContext *oc, AVFormatContext *ic, OptionsContext *o)
3116 AVDictionary **meta_in = NULL;
3117 AVDictionary **meta_out;
3119 char type_in, type_out;
3120 const char *istream_spec = NULL, *ostream_spec = NULL;
3121 int idx_in = 0, idx_out = 0;
3123 parse_meta_type(inspec, &type_in, &idx_in, &istream_spec);
3124 parse_meta_type(outspec, &type_out, &idx_out, &ostream_spec);
3126 if (type_in == 'g' || type_out == 'g')
3127 o->metadata_global_manual = 1;
3128 if (type_in == 's' || type_out == 's')
3129 o->metadata_streams_manual = 1;
3130 if (type_in == 'c' || type_out == 'c')
3131 o->metadata_chapters_manual = 1;
3133 #define METADATA_CHECK_INDEX(index, nb_elems, desc)\
3134 if ((index) < 0 || (index) >= (nb_elems)) {\
3135 av_log(NULL, AV_LOG_FATAL, "Invalid %s index %d while processing metadata maps.\n",\
3140 #define SET_DICT(type, meta, context, index)\
3143 meta = &context->metadata;\
3146 METADATA_CHECK_INDEX(index, context->nb_chapters, "chapter")\
3147 meta = &context->chapters[index]->metadata;\
3150 METADATA_CHECK_INDEX(index, context->nb_programs, "program")\
3151 meta = &context->programs[index]->metadata;\
3155 SET_DICT(type_in, meta_in, ic, idx_in);
3156 SET_DICT(type_out, meta_out, oc, idx_out);
3158 /* for input streams choose first matching stream */
3159 if (type_in == 's') {
3160 for (i = 0; i < ic->nb_streams; i++) {
3161 if ((ret = check_stream_specifier(ic, ic->streams[i], istream_spec)) > 0) {
3162 meta_in = &ic->streams[i]->metadata;
3168 av_log(NULL, AV_LOG_FATAL, "Stream specifier %s does not match any streams.\n", istream_spec);
3173 if (type_out == 's') {
3174 for (i = 0; i < oc->nb_streams; i++) {
3175 if ((ret = check_stream_specifier(oc, oc->streams[i], ostream_spec)) > 0) {
3176 meta_out = &oc->streams[i]->metadata;
3177 av_dict_copy(meta_out, *meta_in, AV_DICT_DONT_OVERWRITE);
3182 av_dict_copy(meta_out, *meta_in, AV_DICT_DONT_OVERWRITE);
3187 static AVCodec *find_codec_or_die(const char *name, enum AVMediaType type, int encoder)
3189 const char *codec_string = encoder ? "encoder" : "decoder";
3193 avcodec_find_encoder_by_name(name) :
3194 avcodec_find_decoder_by_name(name);
3196 av_log(NULL, AV_LOG_FATAL, "Unknown %s '%s'\n", codec_string, name);
3199 if (codec->type != type) {
3200 av_log(NULL, AV_LOG_FATAL, "Invalid %s type '%s'\n", codec_string, name);
3206 static AVCodec *choose_decoder(OptionsContext *o, AVFormatContext *s, AVStream *st)
3208 char *codec_name = NULL;
3210 MATCH_PER_STREAM_OPT(codec_names, str, codec_name, s, st);
3212 AVCodec *codec = find_codec_or_die(codec_name, st->codec->codec_type, 0);
3213 st->codec->codec_id = codec->id;
3216 return avcodec_find_decoder(st->codec->codec_id);
3220 * Add all the streams from the given input file to the global
3221 * list of input streams.
3223 static void add_input_streams(OptionsContext *o, AVFormatContext *ic)
3227 for (i = 0; i < ic->nb_streams; i++) {
3228 AVStream *st = ic->streams[i];
3229 AVCodecContext *dec = st->codec;
3232 input_streams = grow_array(input_streams, sizeof(*input_streams), &nb_input_streams, nb_input_streams + 1);
3233 ist = &input_streams[nb_input_streams - 1];
3235 ist->file_index = nb_input_files;
3237 ist->opts = filter_codec_opts(codec_opts, choose_decoder(o, ic, st), ic, st);
3239 ist->ts_scale = 1.0;
3240 MATCH_PER_STREAM_OPT(ts_scale, dbl, ist->ts_scale, ic, st);
3242 ist->dec = choose_decoder(o, ic, st);
3244 switch (dec->codec_type) {
3245 case AVMEDIA_TYPE_AUDIO:
3247 ist->dec = avcodec_find_decoder(dec->codec_id);
3248 if(o->audio_disable)
3249 st->discard = AVDISCARD_ALL;
3251 case AVMEDIA_TYPE_VIDEO:
3253 ist->dec = avcodec_find_decoder(dec->codec_id);
3255 dec->flags |= CODEC_FLAG_EMU_EDGE;
3258 if (o->video_disable)
3259 st->discard = AVDISCARD_ALL;
3260 else if (video_discard)
3261 st->discard = video_discard;
3263 case AVMEDIA_TYPE_DATA:
3265 case AVMEDIA_TYPE_SUBTITLE:
3267 ist->dec = avcodec_find_decoder(dec->codec_id);
3268 if(o->subtitle_disable)
3269 st->discard = AVDISCARD_ALL;
3271 case AVMEDIA_TYPE_ATTACHMENT:
3272 case AVMEDIA_TYPE_UNKNOWN:
3280 static void assert_file_overwrite(const char *filename)
3282 if ((!file_overwrite || no_file_overwrite) &&
3283 (strchr(filename, ':') == NULL || filename[1] == ':' ||
3284 av_strstart(filename, "file:", NULL))) {
3285 if (avio_check(filename, 0) == 0) {
3286 if (!using_stdin && (!no_file_overwrite || file_overwrite)) {
3287 fprintf(stderr,"File '%s' already exists. Overwrite ? [y/N] ", filename);
3289 if (!read_yesno()) {
3290 fprintf(stderr, "Not overwriting - exiting\n");
3295 fprintf(stderr,"File '%s' already exists. Exiting.\n", filename);
3302 static void dump_attachment(AVStream *st, const char *filename)
3305 AVIOContext *out = NULL;
3306 AVDictionaryEntry *e;
3308 if (!st->codec->extradata_size) {
3309 av_log(NULL, AV_LOG_WARNING, "No extradata to dump in stream #%d:%d.\n",
3310 nb_input_files - 1, st->index);
3313 if (!*filename && (e = av_dict_get(st->metadata, "filename", NULL, 0)))
3314 filename = e->value;
3316 av_log(NULL, AV_LOG_FATAL, "No filename specified and no 'filename' tag"
3317 "in stream #%d:%d.\n", nb_input_files - 1, st->index);
3321 assert_file_overwrite(filename);
3323 if ((ret = avio_open2(&out, filename, AVIO_FLAG_WRITE, &int_cb, NULL)) < 0) {
3324 av_log(NULL, AV_LOG_FATAL, "Could not open file %s for writing.\n",
3329 avio_write(out, st->codec->extradata, st->codec->extradata_size);
3334 static int opt_input_file(OptionsContext *o, const char *opt, const char *filename)
3336 AVFormatContext *ic;
3337 AVInputFormat *file_iformat = NULL;
3341 AVDictionary **opts;
3342 int orig_nb_streams; // number of streams before avformat_find_stream_info
3345 if (!(file_iformat = av_find_input_format(o->format))) {
3346 av_log(NULL, AV_LOG_FATAL, "Unknown input format: '%s'\n", o->format);
3351 if (!strcmp(filename, "-"))
3354 using_stdin |= !strncmp(filename, "pipe:", 5) ||
3355 !strcmp(filename, "/dev/stdin");
3357 /* get default parameters from command line */
3358 ic = avformat_alloc_context();
3360 print_error(filename, AVERROR(ENOMEM));
3363 if (o->nb_audio_sample_rate) {
3364 snprintf(buf, sizeof(buf), "%d", o->audio_sample_rate[o->nb_audio_sample_rate - 1].u.i);
3365 av_dict_set(&format_opts, "sample_rate", buf, 0);
3367 if (o->nb_audio_channels) {
3368 snprintf(buf, sizeof(buf), "%d", o->audio_channels[o->nb_audio_channels - 1].u.i);
3369 av_dict_set(&format_opts, "channels", buf, 0);
3371 if (o->nb_frame_rates) {
3372 av_dict_set(&format_opts, "framerate", o->frame_rates[o->nb_frame_rates - 1].u.str, 0);
3374 if (o->nb_frame_sizes) {
3375 av_dict_set(&format_opts, "video_size", o->frame_sizes[o->nb_frame_sizes - 1].u.str, 0);
3377 if (o->nb_frame_pix_fmts)
3378 av_dict_set(&format_opts, "pixel_format", o->frame_pix_fmts[o->nb_frame_pix_fmts - 1].u.str, 0);
3380 ic->flags |= AVFMT_FLAG_NONBLOCK;
3381 ic->interrupt_callback = int_cb;
3383 /* open the input file with generic avformat function */
3384 err = avformat_open_input(&ic, filename, file_iformat, &format_opts);
3386 print_error(filename, err);
3389 assert_avoptions(format_opts);
3391 /* apply forced codec ids */
3392 for (i = 0; i < ic->nb_streams; i++)
3393 choose_decoder(o, ic, ic->streams[i]);
3395 /* Set AVCodecContext options for avformat_find_stream_info */
3396 opts = setup_find_stream_info_opts(ic, codec_opts);
3397 orig_nb_streams = ic->nb_streams;
3399 /* If not enough info to get the stream parameters, we decode the
3400 first frames to get it. (used in mpeg case for example) */
3401 ret = avformat_find_stream_info(ic, opts);
3403 av_log(NULL, AV_LOG_FATAL, "%s: could not find codec parameters\n", filename);
3404 avformat_close_input(&ic);
3408 timestamp = o->start_time;
3409 /* add the stream start time */
3410 if (ic->start_time != AV_NOPTS_VALUE)
3411 timestamp += ic->start_time;
3413 /* if seeking requested, we execute it */
3414 if (o->start_time != 0) {
3415 ret = av_seek_frame(ic, -1, timestamp, AVSEEK_FLAG_BACKWARD);
3417 av_log(NULL, AV_LOG_WARNING, "%s: could not seek to position %0.3f\n",
3418 filename, (double)timestamp / AV_TIME_BASE);
3422 /* update the current parameters so that they match the one of the input stream */
3423 add_input_streams(o, ic);
3425 /* dump the file content */
3426 av_dump_format(ic, nb_input_files, filename, 0);
3428 input_files = grow_array(input_files, sizeof(*input_files), &nb_input_files, nb_input_files + 1);
3429 input_files[nb_input_files - 1].ctx = ic;
3430 input_files[nb_input_files - 1].ist_index = nb_input_streams - ic->nb_streams;
3431 input_files[nb_input_files - 1].ts_offset = o->input_ts_offset - (copy_ts ? 0 : timestamp);
3432 input_files[nb_input_files - 1].nb_streams = ic->nb_streams;
3433 input_files[nb_input_files - 1].rate_emu = o->rate_emu;
3435 for (i = 0; i < o->nb_dump_attachment; i++) {
3438 for (j = 0; j < ic->nb_streams; j++) {
3439 AVStream *st = ic->streams[j];
3441 if (check_stream_specifier(ic, st, o->dump_attachment[i].specifier) == 1)
3442 dump_attachment(st, o->dump_attachment[i].u.str);
3446 for (i = 0; i < orig_nb_streams; i++)
3447 av_dict_free(&opts[i]);
3454 static void parse_forced_key_frames(char *kf, OutputStream *ost,
3455 AVCodecContext *avctx)
3461 for (p = kf; *p; p++)
3464 ost->forced_kf_count = n;
3465 ost->forced_kf_pts = av_malloc(sizeof(*ost->forced_kf_pts) * n);
3466 if (!ost->forced_kf_pts) {
3467 av_log(NULL, AV_LOG_FATAL, "Could not allocate forced key frames array.\n");
3470 for (i = 0; i < n; i++) {
3471 p = i ? strchr(p, ',') + 1 : kf;
3472 t = parse_time_or_die("force_key_frames", p, 1);
3473 ost->forced_kf_pts[i] = av_rescale_q(t, AV_TIME_BASE_Q, avctx->time_base);
3477 static uint8_t *get_line(AVIOContext *s)
3483 if (avio_open_dyn_buf(&line) < 0) {
3484 av_log(NULL, AV_LOG_FATAL, "Could not alloc buffer for reading preset.\n");
3488 while ((c = avio_r8(s)) && c != '\n')
3491 avio_close_dyn_buf(line, &buf);
3496 static int get_preset_file_2(const char *preset_name, const char *codec_name, AVIOContext **s)
3499 char filename[1000];
3500 const char *base[3] = { getenv("AVCONV_DATADIR"),
3505 for (i = 0; i < FF_ARRAY_ELEMS(base) && ret; i++) {
3509 snprintf(filename, sizeof(filename), "%s%s/%s-%s.avpreset", base[i],
3510 i != 1 ? "" : "/.avconv", codec_name, preset_name);
3511 ret = avio_open2(s, filename, AVIO_FLAG_READ, &int_cb, NULL);
3514 snprintf(filename, sizeof(filename), "%s%s/%s.avpreset", base[i],
3515 i != 1 ? "" : "/.avconv", preset_name);
3516 ret = avio_open2(s, filename, AVIO_FLAG_READ, &int_cb, NULL);
3522 static void choose_encoder(OptionsContext *o, AVFormatContext *s, OutputStream *ost)
3524 char *codec_name = NULL;
3526 MATCH_PER_STREAM_OPT(codec_names, str, codec_name, s, ost->st);
3528 ost->st->codec->codec_id = av_guess_codec(s->oformat, NULL, s->filename,
3529 NULL, ost->st->codec->codec_type);
3530 ost->enc = avcodec_find_encoder(ost->st->codec->codec_id);
3531 } else if (!strcmp(codec_name, "copy"))
3532 ost->stream_copy = 1;
3534 ost->enc = find_codec_or_die(codec_name, ost->st->codec->codec_type, 1);
3535 ost->st->codec->codec_id = ost->enc->id;
3539 static OutputStream *new_output_stream(OptionsContext *o, AVFormatContext *oc, enum AVMediaType type)
3542 AVStream *st = avformat_new_stream(oc, NULL);
3543 int idx = oc->nb_streams - 1, ret = 0;
3544 char *bsf = NULL, *next, *codec_tag = NULL;
3545 AVBitStreamFilterContext *bsfc, *bsfc_prev = NULL;
3547 char *buf = NULL, *arg = NULL, *preset = NULL;
3548 AVIOContext *s = NULL;
3551 av_log(NULL, AV_LOG_FATAL, "Could not alloc stream.\n");
3555 if (oc->nb_streams - 1 < o->nb_streamid_map)
3556 st->id = o->streamid_map[oc->nb_streams - 1];
3558 output_streams = grow_array(output_streams, sizeof(*output_streams), &nb_output_streams,
3559 nb_output_streams + 1);
3560 ost = &output_streams[nb_output_streams - 1];
3561 ost->file_index = nb_output_files;
3564 st->codec->codec_type = type;
3565 choose_encoder(o, oc, ost);
3567 ost->opts = filter_codec_opts(codec_opts, ost->enc, oc, st);
3570 avcodec_get_context_defaults3(st->codec, ost->enc);
3571 st->codec->codec_type = type; // XXX hack, avcodec_get_context_defaults2() sets type to unknown for stream copy
3573 MATCH_PER_STREAM_OPT(presets, str, preset, oc, st);
3574 if (preset && (!(ret = get_preset_file_2(preset, ost->enc->name, &s)))) {
3577 if (!buf[0] || buf[0] == '#') {
3581 if (!(arg = strchr(buf, '='))) {
3582 av_log(NULL, AV_LOG_FATAL, "Invalid line found in the preset file.\n");
3586 av_dict_set(&ost->opts, buf, arg, AV_DICT_DONT_OVERWRITE);
3588 } while (!s->eof_reached);
3592 av_log(NULL, AV_LOG_FATAL,
3593 "Preset %s specified for stream %d:%d, but could not be opened.\n",
3594 preset, ost->file_index, ost->index);
3598 ost->max_frames = INT64_MAX;
3599 MATCH_PER_STREAM_OPT(max_frames, i64, ost->max_frames, oc, st);
3601 MATCH_PER_STREAM_OPT(bitstream_filters, str, bsf, oc, st);
3603 if (next = strchr(bsf, ','))
3605 if (!(bsfc = av_bitstream_filter_init(bsf))) {
3606 av_log(NULL, AV_LOG_FATAL, "Unknown bitstream filter %s\n", bsf);
3610 bsfc_prev->next = bsfc;
3612 ost->bitstream_filters = bsfc;
3618 MATCH_PER_STREAM_OPT(codec_tags, str, codec_tag, oc, st);
3620 uint32_t tag = strtol(codec_tag, &next, 0);
3622 tag = AV_RL32(codec_tag);
3623 st->codec->codec_tag = tag;
3626 MATCH_PER_STREAM_OPT(qscale, dbl, qscale, oc, st);
3627 if (qscale >= 0 || same_quant) {
3628 st->codec->flags |= CODEC_FLAG_QSCALE;
3629 st->codec->global_quality = FF_QP2LAMBDA * qscale;
3632 if (oc->oformat->flags & AVFMT_GLOBALHEADER)
3633 st->codec->flags |= CODEC_FLAG_GLOBAL_HEADER;
3635 av_opt_get_int(sws_opts, "sws_flags", 0, &ost->sws_flags);
3639 static void parse_matrix_coeffs(uint16_t *dest, const char *str)
3642 const char *p = str;
3649 av_log(NULL, AV_LOG_FATAL, "Syntax error in matrix \"%s\" at coeff %d\n", str, i);
3656 static OutputStream *new_video_stream(OptionsContext *o, AVFormatContext *oc)
3660 AVCodecContext *video_enc;
3662 ost = new_output_stream(o, oc, AVMEDIA_TYPE_VIDEO);
3664 video_enc = st->codec;
3666 if (!ost->stream_copy) {
3667 const char *p = NULL;
3668 char *forced_key_frames = NULL, *frame_rate = NULL, *frame_size = NULL;
3669 char *frame_aspect_ratio = NULL, *frame_pix_fmt = NULL;
3670 char *intra_matrix = NULL, *inter_matrix = NULL, *filters = NULL;
3673 MATCH_PER_STREAM_OPT(frame_rates, str, frame_rate, oc, st);
3674 if (frame_rate && av_parse_video_rate(&ost->frame_rate, frame_rate) < 0) {
3675 av_log(NULL, AV_LOG_FATAL, "Invalid framerate value: %s\n", frame_rate);
3679 MATCH_PER_STREAM_OPT(frame_sizes, str, frame_size, oc, st);
3680 if (frame_size && av_parse_video_size(&video_enc->width, &video_enc->height, frame_size) < 0) {
3681 av_log(NULL, AV_LOG_FATAL, "Invalid frame size: %s.\n", frame_size);
3685 MATCH_PER_STREAM_OPT(frame_aspect_ratios, str, frame_aspect_ratio, oc, st);
3686 if (frame_aspect_ratio)
3687 ost->frame_aspect_ratio = parse_frame_aspect_ratio(frame_aspect_ratio);
3689 MATCH_PER_STREAM_OPT(frame_pix_fmts, str, frame_pix_fmt, oc, st);
3690 if (frame_pix_fmt && (video_enc->pix_fmt = av_get_pix_fmt(frame_pix_fmt)) == PIX_FMT_NONE) {
3691 av_log(NULL, AV_LOG_FATAL, "Unknown pixel format requested: %s.\n", frame_pix_fmt);
3694 st->sample_aspect_ratio = video_enc->sample_aspect_ratio;
3696 MATCH_PER_STREAM_OPT(intra_matrices, str, intra_matrix, oc, st);
3698 if (!(video_enc->intra_matrix = av_mallocz(sizeof(*video_enc->intra_matrix) * 64))) {
3699 av_log(NULL, AV_LOG_FATAL, "Could not allocate memory for intra matrix.\n");
3702 parse_matrix_coeffs(video_enc->intra_matrix, intra_matrix);
3704 MATCH_PER_STREAM_OPT(inter_matrices, str, inter_matrix, oc, st);
3706 if (!(video_enc->inter_matrix = av_mallocz(sizeof(*video_enc->inter_matrix) * 64))) {
3707 av_log(NULL, AV_LOG_FATAL, "Could not allocate memory for inter matrix.\n");
3710 parse_matrix_coeffs(video_enc->inter_matrix, inter_matrix);
3713 MATCH_PER_STREAM_OPT(rc_overrides, str, p, oc, st);
3714 for (i = 0; p; i++) {
3716 int e = sscanf(p, "%d,%d,%d", &start, &end, &q);
3718 av_log(NULL, AV_LOG_FATAL, "error parsing rc_override\n");
3721 video_enc->rc_override =
3722 av_realloc(video_enc->rc_override,
3723 sizeof(RcOverride) * (i + 1));
3724 video_enc->rc_override[i].start_frame = start;
3725 video_enc->rc_override[i].end_frame = end;
3727 video_enc->rc_override[i].qscale = q;
3728 video_enc->rc_override[i].quality_factor = 1.0;
3731 video_enc->rc_override[i].qscale = 0;
3732 video_enc->rc_override[i].quality_factor = -q/100.0;
3737 video_enc->rc_override_count = i;
3738 if (!video_enc->rc_initial_buffer_occupancy)
3739 video_enc->rc_initial_buffer_occupancy = video_enc->rc_buffer_size * 3 / 4;
3740 video_enc->intra_dc_precision = intra_dc_precision - 8;
3745 video_enc->flags |= CODEC_FLAG_PASS1;
3747 video_enc->flags |= CODEC_FLAG_PASS2;
3751 MATCH_PER_STREAM_OPT(forced_key_frames, str, forced_key_frames, oc, st);
3752 if (forced_key_frames)
3753 parse_forced_key_frames(forced_key_frames, ost, video_enc);
3755 MATCH_PER_STREAM_OPT(force_fps, i, ost->force_fps, oc, st);
3757 ost->top_field_first = -1;
3758 MATCH_PER_STREAM_OPT(top_field_first, i, ost->top_field_first, oc, st);
3761 MATCH_PER_STREAM_OPT(filters, str, filters, oc, st);
3763 ost->avfilter = av_strdup(filters);
3766 MATCH_PER_STREAM_OPT(copy_initial_nonkeyframes, i, ost->copy_initial_nonkeyframes, oc ,st);
3772 static OutputStream *new_audio_stream(OptionsContext *o, AVFormatContext *oc)
3776 AVCodecContext *audio_enc;
3778 ost = new_output_stream(o, oc, AVMEDIA_TYPE_AUDIO);
3781 audio_enc = st->codec;
3782 audio_enc->codec_type = AVMEDIA_TYPE_AUDIO;
3784 if (!ost->stream_copy) {
3785 char *sample_fmt = NULL;
3787 MATCH_PER_STREAM_OPT(audio_channels, i, audio_enc->channels, oc, st);
3789 MATCH_PER_STREAM_OPT(sample_fmts, str, sample_fmt, oc, st);
3791 (audio_enc->sample_fmt = av_get_sample_fmt(sample_fmt)) == AV_SAMPLE_FMT_NONE) {
3792 av_log(NULL, AV_LOG_FATAL, "Invalid sample format '%s'\n", sample_fmt);
3796 MATCH_PER_STREAM_OPT(audio_sample_rate, i, audio_enc->sample_rate, oc, st);
3802 static OutputStream *new_data_stream(OptionsContext *o, AVFormatContext *oc)
3806 ost = new_output_stream(o, oc, AVMEDIA_TYPE_DATA);
3807 if (!ost->stream_copy) {
3808 av_log(NULL, AV_LOG_FATAL, "Data stream encoding not supported yet (only streamcopy)\n");
3815 static OutputStream *new_attachment_stream(OptionsContext *o, AVFormatContext *oc)
3817 OutputStream *ost = new_output_stream(o, oc, AVMEDIA_TYPE_ATTACHMENT);
3818 ost->stream_copy = 1;
3822 static OutputStream *new_subtitle_stream(OptionsContext *o, AVFormatContext *oc)
3826 AVCodecContext *subtitle_enc;
3828 ost = new_output_stream(o, oc, AVMEDIA_TYPE_SUBTITLE);
3830 subtitle_enc = st->codec;
3832 subtitle_enc->codec_type = AVMEDIA_TYPE_SUBTITLE;
3837 /* arg format is "output-stream-index:streamid-value". */
3838 static int opt_streamid(OptionsContext *o, const char *opt, const char *arg)
3844 av_strlcpy(idx_str, arg, sizeof(idx_str));
3845 p = strchr(idx_str, ':');
3847 av_log(NULL, AV_LOG_FATAL,
3848 "Invalid value '%s' for option '%s', required syntax is 'index:value'\n",
3853 idx = parse_number_or_die(opt, idx_str, OPT_INT, 0, MAX_STREAMS-1);
3854 o->streamid_map = grow_array(o->streamid_map, sizeof(*o->streamid_map), &o->nb_streamid_map, idx+1);
3855 o->streamid_map[idx] = parse_number_or_die(opt, p, OPT_INT, 0, INT_MAX);
3859 static int copy_chapters(InputFile *ifile, OutputFile *ofile, int copy_metadata)
3861 AVFormatContext *is = ifile->ctx;
3862 AVFormatContext *os = ofile->ctx;
3865 for (i = 0; i < is->nb_chapters; i++) {
3866 AVChapter *in_ch = is->chapters[i], *out_ch;
3867 int64_t ts_off = av_rescale_q(ofile->start_time - ifile->ts_offset,
3868 AV_TIME_BASE_Q, in_ch->time_base);
3869 int64_t rt = (ofile->recording_time == INT64_MAX) ? INT64_MAX :
3870 av_rescale_q(ofile->recording_time, AV_TIME_BASE_Q, in_ch->time_base);
3873 if (in_ch->end < ts_off)
3875 if (rt != INT64_MAX && in_ch->start > rt + ts_off)
3878 out_ch = av_mallocz(sizeof(AVChapter));
3880 return AVERROR(ENOMEM);
3882 out_ch->id = in_ch->id;
3883 out_ch->time_base = in_ch->time_base;
3884 out_ch->start = FFMAX(0, in_ch->start - ts_off);
3885 out_ch->end = FFMIN(rt, in_ch->end - ts_off);
3888 av_dict_copy(&out_ch->metadata, in_ch->metadata, 0);
3891 os->chapters = av_realloc(os->chapters, sizeof(AVChapter) * os->nb_chapters);
3893 return AVERROR(ENOMEM);
3894 os->chapters[os->nb_chapters - 1] = out_ch;
3899 static void opt_output_file(void *optctx, const char *filename)
3901 OptionsContext *o = optctx;
3902 AVFormatContext *oc;
3904 AVOutputFormat *file_oformat;
3908 if (!strcmp(filename, "-"))
3911 err = avformat_alloc_output_context2(&oc, NULL, o->format, filename);
3913 print_error(filename, err);
3917 file_oformat= oc->oformat;
3918 oc->interrupt_callback = int_cb;
3920 if (!o->nb_stream_maps) {
3921 /* pick the "best" stream of each type */
3922 #define NEW_STREAM(type, index)\
3924 ost = new_ ## type ## _stream(o, oc);\
3925 ost->source_index = index;\
3926 ost->sync_ist = &input_streams[index];\
3927 input_streams[index].discard = 0;\
3930 /* video: highest resolution */
3931 if (!o->video_disable && oc->oformat->video_codec != CODEC_ID_NONE) {
3932 int area = 0, idx = -1;
3933 for (i = 0; i < nb_input_streams; i++) {
3934 ist = &input_streams[i];
3935 if (ist->st->codec->codec_type == AVMEDIA_TYPE_VIDEO &&
3936 ist->st->codec->width * ist->st->codec->height > area) {
3937 area = ist->st->codec->width * ist->st->codec->height;
3941 NEW_STREAM(video, idx);
3944 /* audio: most channels */
3945 if (!o->audio_disable && oc->oformat->audio_codec != CODEC_ID_NONE) {
3946 int channels = 0, idx = -1;
3947 for (i = 0; i < nb_input_streams; i++) {
3948 ist = &input_streams[i];
3949 if (ist->st->codec->codec_type == AVMEDIA_TYPE_AUDIO &&
3950 ist->st->codec->channels > channels) {
3951 channels = ist->st->codec->channels;
3955 NEW_STREAM(audio, idx);
3958 /* subtitles: pick first */
3959 if (!o->subtitle_disable && oc->oformat->subtitle_codec != CODEC_ID_NONE) {
3960 for (i = 0; i < nb_input_streams; i++)
3961 if (input_streams[i].st->codec->codec_type == AVMEDIA_TYPE_SUBTITLE) {
3962 NEW_STREAM(subtitle, i);
3966 /* do something with data? */
3968 for (i = 0; i < o->nb_stream_maps; i++) {
3969 StreamMap *map = &o->stream_maps[i];
3974 ist = &input_streams[input_files[map->file_index].ist_index + map->stream_index];
3975 switch (ist->st->codec->codec_type) {
3976 case AVMEDIA_TYPE_VIDEO: ost = new_video_stream(o, oc); break;
3977 case AVMEDIA_TYPE_AUDIO: ost = new_audio_stream(o, oc); break;
3978 case AVMEDIA_TYPE_SUBTITLE: ost = new_subtitle_stream(o, oc); break;
3979 case AVMEDIA_TYPE_DATA: ost = new_data_stream(o, oc); break;
3980 case AVMEDIA_TYPE_ATTACHMENT: ost = new_attachment_stream(o, oc); break;
3982 av_log(NULL, AV_LOG_FATAL, "Cannot map stream #%d:%d - unsupported type.\n",
3983 map->file_index, map->stream_index);
3987 ost->source_index = input_files[map->file_index].ist_index + map->stream_index;
3988 ost->sync_ist = &input_streams[input_files[map->sync_file_index].ist_index +
3989 map->sync_stream_index];
3994 /* handle attached files */
3995 for (i = 0; i < o->nb_attachments; i++) {
3997 uint8_t *attachment;
4001 if ((err = avio_open2(&pb, o->attachments[i], AVIO_FLAG_READ, &int_cb, NULL)) < 0) {
4002 av_log(NULL, AV_LOG_FATAL, "Could not open attachment file %s.\n",
4006 if ((len = avio_size(pb)) <= 0) {
4007 av_log(NULL, AV_LOG_FATAL, "Could not get size of the attachment %s.\n",
4011 if (!(attachment = av_malloc(len))) {
4012 av_log(NULL, AV_LOG_FATAL, "Attachment %s too large to fit into memory.\n",
4016 avio_read(pb, attachment, len);
4018 ost = new_attachment_stream(o, oc);
4019 ost->stream_copy = 0;
4020 ost->source_index = -1;
4021 ost->attachment_filename = o->attachments[i];
4022 ost->st->codec->extradata = attachment;
4023 ost->st->codec->extradata_size = len;
4025 p = strrchr(o->attachments[i], '/');
4026 av_dict_set(&ost->st->metadata, "filename", (p && *p) ? p + 1 : o->attachments[i], AV_DICT_DONT_OVERWRITE);
4030 output_files = grow_array(output_files, sizeof(*output_files), &nb_output_files, nb_output_files + 1);
4031 output_files[nb_output_files - 1].ctx = oc;
4032 output_files[nb_output_files - 1].ost_index = nb_output_streams - oc->nb_streams;
4033 output_files[nb_output_files - 1].recording_time = o->recording_time;
4034 output_files[nb_output_files - 1].start_time = o->start_time;
4035 output_files[nb_output_files - 1].limit_filesize = o->limit_filesize;
4036 av_dict_copy(&output_files[nb_output_files - 1].opts, format_opts, 0);
4038 /* check filename in case of an image number is expected */
4039 if (oc->oformat->flags & AVFMT_NEEDNUMBER) {
4040 if (!av_filename_number_test(oc->filename)) {
4041 print_error(oc->filename, AVERROR(EINVAL));
4046 if (!(oc->oformat->flags & AVFMT_NOFILE)) {
4047 /* test if it already exists to avoid losing precious files */
4048 assert_file_overwrite(filename);
4051 if ((err = avio_open2(&oc->pb, filename, AVIO_FLAG_WRITE,
4052 &oc->interrupt_callback,
4053 &output_files[nb_output_files - 1].opts)) < 0) {
4054 print_error(filename, err);
4059 if (o->mux_preload) {
4061 snprintf(buf, sizeof(buf), "%d", (int)(o->mux_preload*AV_TIME_BASE));
4062 av_dict_set(&output_files[nb_output_files - 1].opts, "preload", buf, 0);
4064 oc->max_delay = (int)(o->mux_max_delay * AV_TIME_BASE);
4067 for (i = 0; i < o->nb_metadata_map; i++) {
4069 int in_file_index = strtol(o->metadata_map[i].u.str, &p, 0);
4071 if (in_file_index < 0)
4073 if (in_file_index >= nb_input_files) {
4074 av_log(NULL, AV_LOG_FATAL, "Invalid input file index %d while processing metadata maps\n", in_file_index);
4077 copy_metadata(o->metadata_map[i].specifier, *p ? p + 1 : p, oc, input_files[in_file_index].ctx, o);
4081 if (o->chapters_input_file >= nb_input_files) {
4082 if (o->chapters_input_file == INT_MAX) {
4083 /* copy chapters from the first input file that has them*/
4084 o->chapters_input_file = -1;
4085 for (i = 0; i < nb_input_files; i++)
4086 if (input_files[i].ctx->nb_chapters) {
4087 o->chapters_input_file = i;
4091 av_log(NULL, AV_LOG_FATAL, "Invalid input file index %d in chapter mapping.\n",
4092 o->chapters_input_file);
4096 if (o->chapters_input_file >= 0)
4097 copy_chapters(&input_files[o->chapters_input_file], &output_files[nb_output_files - 1],
4098 !o->metadata_chapters_manual);
4100 /* copy global metadata by default */
4101 if (!o->metadata_global_manual && nb_input_files)
4102 av_dict_copy(&oc->metadata, input_files[0].ctx->metadata,
4103 AV_DICT_DONT_OVERWRITE);
4104 if (!o->metadata_streams_manual)
4105 for (i = output_files[nb_output_files - 1].ost_index; i < nb_output_streams; i++) {
4107 if (output_streams[i].source_index < 0) /* this is true e.g. for attached files */
4109 ist = &input_streams[output_streams[i].source_index];
4110 av_dict_copy(&output_streams[i].st->metadata, ist->st->metadata, AV_DICT_DONT_OVERWRITE);
4113 /* process manually set metadata */
4114 for (i = 0; i < o->nb_metadata; i++) {
4117 const char *stream_spec;
4118 int index = 0, j, ret;
4120 val = strchr(o->metadata[i].u.str, '=');
4122 av_log(NULL, AV_LOG_FATAL, "No '=' character in metadata string %s.\n",
4123 o->metadata[i].u.str);
4128 parse_meta_type(o->metadata[i].specifier, &type, &index, &stream_spec);
4130 for (j = 0; j < oc->nb_streams; j++) {
4131 if ((ret = check_stream_specifier(oc, oc->streams[j], stream_spec)) > 0) {
4132 av_dict_set(&oc->streams[j]->metadata, o->metadata[i].u.str, *val ? val : NULL, 0);
4136 printf("ret %d, stream_spec %s\n", ret, stream_spec);
4144 if (index < 0 || index >= oc->nb_chapters) {
4145 av_log(NULL, AV_LOG_FATAL, "Invalid chapter index %d in metadata specifier.\n", index);
4148 m = &oc->chapters[index]->metadata;
4151 av_log(NULL, AV_LOG_FATAL, "Invalid metadata specifier %s.\n", o->metadata[i].specifier);
4154 av_dict_set(m, o->metadata[i].u.str, *val ? val : NULL, 0);
4161 /* same option as mencoder */
4162 static int opt_pass(const char *opt, const char *arg)
4164 do_pass = parse_number_or_die(opt, arg, OPT_INT, 1, 2);
4168 static int64_t getutime(void)
4171 struct rusage rusage;
4173 getrusage(RUSAGE_SELF, &rusage);
4174 return (rusage.ru_utime.tv_sec * 1000000LL) + rusage.ru_utime.tv_usec;
4175 #elif HAVE_GETPROCESSTIMES
4177 FILETIME c, e, k, u;
4178 proc = GetCurrentProcess();
4179 GetProcessTimes(proc, &c, &e, &k, &u);
4180 return ((int64_t) u.dwHighDateTime << 32 | u.dwLowDateTime) / 10;
4182 return av_gettime();
4186 static int64_t getmaxrss(void)
4188 #if HAVE_GETRUSAGE && HAVE_STRUCT_RUSAGE_RU_MAXRSS
4189 struct rusage rusage;
4190 getrusage(RUSAGE_SELF, &rusage);
4191 return (int64_t)rusage.ru_maxrss * 1024;
4192 #elif HAVE_GETPROCESSMEMORYINFO
4194 PROCESS_MEMORY_COUNTERS memcounters;
4195 proc = GetCurrentProcess();
4196 memcounters.cb = sizeof(memcounters);
4197 GetProcessMemoryInfo(proc, &memcounters, sizeof(memcounters));
4198 return memcounters.PeakPagefileUsage;
4204 static int opt_audio_qscale(OptionsContext *o, const char *opt, const char *arg)
4206 return parse_option(o, "q:a", arg, options);
4209 static void show_usage(void)
4211 av_log(NULL, AV_LOG_INFO, "Hyper fast Audio and Video encoder\n");
4212 av_log(NULL, AV_LOG_INFO, "usage: %s [options] [[infile options] -i infile]... {[outfile options] outfile}...\n", program_name);
4213 av_log(NULL, AV_LOG_INFO, "\n");
4216 static int opt_help(const char *opt, const char *arg)
4218 int flags = AV_OPT_FLAG_DECODING_PARAM | AV_OPT_FLAG_ENCODING_PARAM;
4219 av_log_set_callback(log_callback_help);
4221 show_help_options(options, "Main options:\n",
4222 OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_SUBTITLE | OPT_GRAB, 0);
4223 show_help_options(options, "\nAdvanced options:\n",
4224 OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_SUBTITLE | OPT_GRAB,
4226 show_help_options(options, "\nVideo options:\n",
4227 OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_GRAB,
4229 show_help_options(options, "\nAdvanced Video options:\n",
4230 OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_GRAB,
4231 OPT_VIDEO | OPT_EXPERT);
4232 show_help_options(options, "\nAudio options:\n",
4233 OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_GRAB,
4235 show_help_options(options, "\nAdvanced Audio options:\n",
4236 OPT_EXPERT | OPT_AUDIO | OPT_VIDEO | OPT_GRAB,
4237 OPT_AUDIO | OPT_EXPERT);
4238 show_help_options(options, "\nSubtitle options:\n",
4239 OPT_SUBTITLE | OPT_GRAB,
4241 show_help_options(options, "\nAudio/Video grab options:\n",
4245 show_help_children(avcodec_get_class(), flags);
4246 show_help_children(avformat_get_class(), flags);
4247 show_help_children(sws_get_class(), flags);
4252 static int opt_target(OptionsContext *o, const char *opt, const char *arg)
4254 enum { PAL, NTSC, FILM, UNKNOWN } norm = UNKNOWN;
4255 static const char *const frame_rates[] = { "25", "30000/1001", "24000/1001" };
4257 if (!strncmp(arg, "pal-", 4)) {
4260 } else if (!strncmp(arg, "ntsc-", 5)) {
4263 } else if (!strncmp(arg, "film-", 5)) {
4267 /* Try to determine PAL/NTSC by peeking in the input files */
4268 if (nb_input_files) {
4270 for (j = 0; j < nb_input_files; j++) {
4271 for (i = 0; i < input_files[j].nb_streams; i++) {
4272 AVCodecContext *c = input_files[j].ctx->streams[i]->codec;
4273 if (c->codec_type != AVMEDIA_TYPE_VIDEO)
4275 fr = c->time_base.den * 1000 / c->time_base.num;
4279 } else if ((fr == 29970) || (fr == 23976)) {
4284 if (norm != UNKNOWN)
4288 if (norm != UNKNOWN)
4289 av_log(NULL, AV_LOG_INFO, "Assuming %s for target.\n", norm == PAL ? "PAL" : "NTSC");
4292 if (norm == UNKNOWN) {
4293 av_log(NULL, AV_LOG_FATAL, "Could not determine norm (PAL/NTSC/NTSC-Film) for target.\n");
4294 av_log(NULL, AV_LOG_FATAL, "Please prefix target with \"pal-\", \"ntsc-\" or \"film-\",\n");
4295 av_log(NULL, AV_LOG_FATAL, "or set a framerate with \"-r xxx\".\n");
4299 if (!strcmp(arg, "vcd")) {
4300 opt_video_codec(o, "c:v", "mpeg1video");
4301 opt_audio_codec(o, "c:a", "mp2");
4302 parse_option(o, "f", "vcd", options);
4304 parse_option(o, "s", norm == PAL ? "352x288" : "352x240", options);
4305 parse_option(o, "r", frame_rates[norm], options);
4306 opt_default("g", norm == PAL ? "15" : "18");
4308 opt_default("b", "1150000");
4309 opt_default("maxrate", "1150000");
4310 opt_default("minrate", "1150000");
4311 opt_default("bufsize", "327680"); // 40*1024*8;
4313 opt_default("b:a", "224000");
4314 parse_option(o, "ar", "44100", options);
4315 parse_option(o, "ac", "2", options);
4317 opt_default("packetsize", "2324");
4318 opt_default("muxrate", "1411200"); // 2352 * 75 * 8;
4320 /* We have to offset the PTS, so that it is consistent with the SCR.
4321 SCR starts at 36000, but the first two packs contain only padding
4322 and the first pack from the other stream, respectively, may also have
4323 been written before.
4324 So the real data starts at SCR 36000+3*1200. */
4325 o->mux_preload = (36000 + 3 * 1200) / 90000.0; // 0.44
4326 } else if (!strcmp(arg, "svcd")) {
4328 opt_video_codec(o, "c:v", "mpeg2video");
4329 opt_audio_codec(o, "c:a", "mp2");
4330 parse_option(o, "f", "svcd", options);
4332 parse_option(o, "s", norm == PAL ? "480x576" : "480x480", options);
4333 parse_option(o, "r", frame_rates[norm], options);
4334 opt_default("g", norm == PAL ? "15" : "18");
4336 opt_default("b", "2040000");
4337 opt_default("maxrate", "2516000");
4338 opt_default("minrate", "0"); // 1145000;
4339 opt_default("bufsize", "1835008"); // 224*1024*8;
4340 opt_default("flags", "+scan_offset");
4343 opt_default("b:a", "224000");
4344 parse_option(o, "ar", "44100", options);
4346 opt_default("packetsize", "2324");
4348 } else if (!strcmp(arg, "dvd")) {
4350 opt_video_codec(o, "c:v", "mpeg2video");
4351 opt_audio_codec(o, "c:a", "ac3");
4352 parse_option(o, "f", "dvd", options);
4354 parse_option(o, "s", norm == PAL ? "720x576" : "720x480", options);
4355 parse_option(o, "r", frame_rates[norm], options);
4356 opt_default("g", norm == PAL ? "15" : "18");
4358 opt_default("b", "6000000");
4359 opt_default("maxrate", "9000000");
4360 opt_default("minrate", "0"); // 1500000;
4361 opt_default("bufsize", "1835008"); // 224*1024*8;
4363 opt_default("packetsize", "2048"); // from www.mpucoder.com: DVD sectors contain 2048 bytes of data, this is also the size of one pack.
4364 opt_default("muxrate", "10080000"); // from mplex project: data_rate = 1260000. mux_rate = data_rate * 8
4366 opt_default("b:a", "448000");
4367 parse_option(o, "ar", "48000", options);
4369 } else if (!strncmp(arg, "dv", 2)) {
4371 parse_option(o, "f", "dv", options);
4373 parse_option(o, "s", norm == PAL ? "720x576" : "720x480", options);
4374 parse_option(o, "pix_fmt", !strncmp(arg, "dv50", 4) ? "yuv422p" :
4375 norm == PAL ? "yuv420p" : "yuv411p", options);
4376 parse_option(o, "r", frame_rates[norm], options);
4378 parse_option(o, "ar", "48000", options);
4379 parse_option(o, "ac", "2", options);
4382 av_log(NULL, AV_LOG_ERROR, "Unknown target: %s\n", arg);
4383 return AVERROR(EINVAL);
4388 static int opt_vstats_file(const char *opt, const char *arg)
4390 av_free (vstats_filename);
4391 vstats_filename = av_strdup (arg);
4395 static int opt_vstats(const char *opt, const char *arg)
4398 time_t today2 = time(NULL);
4399 struct tm *today = localtime(&today2);
4401 snprintf(filename, sizeof(filename), "vstats_%02d%02d%02d.log", today->tm_hour, today->tm_min,
4403 return opt_vstats_file(opt, filename);
4406 static int opt_video_frames(OptionsContext *o, const char *opt, const char *arg)
4408 return parse_option(o, "frames:v", arg, options);
4411 static int opt_audio_frames(OptionsContext *o, const char *opt, const char *arg)
4413 return parse_option(o, "frames:a", arg, options);
4416 static int opt_data_frames(OptionsContext *o, const char *opt, const char *arg)
4418 return parse_option(o, "frames:d", arg, options);
4421 static void log_callback_null(void* ptr, int level, const char* fmt, va_list vl)
4425 static int opt_passlogfile(const char *opt, const char *arg)
4427 pass_logfilename_prefix = arg;
4428 #if CONFIG_LIBX264_ENCODER
4429 return opt_default("passlogfile", arg);
4435 static int opt_video_tag(OptionsContext *o, const char *opt, const char *arg)
4437 return parse_option(o, "tag:v", arg, options);
4440 static int opt_audio_tag(OptionsContext *o, const char *opt, const char *arg)
4442 return parse_option(o, "tag:a", arg, options);
4445 static int opt_subtitle_tag(OptionsContext *o, const char *opt, const char *arg)
4447 return parse_option(o, "tag:s", arg, options);
4450 static int opt_video_filters(OptionsContext *o, const char *opt, const char *arg)
4452 return parse_option(o, "filter:v", arg, options);
4455 static int opt_vsync(const char *opt, const char *arg)
4457 if (!av_strcasecmp(arg, "cfr")) video_sync_method = VSYNC_CFR;
4458 else if (!av_strcasecmp(arg, "vfr")) video_sync_method = VSYNC_VFR;
4459 else if (!av_strcasecmp(arg, "passthrough")) video_sync_method = VSYNC_PASSTHROUGH;
4461 if (video_sync_method == VSYNC_AUTO)
4462 video_sync_method = parse_number_or_die("vsync", arg, OPT_INT, VSYNC_AUTO, VSYNC_VFR);
4466 #define OFFSET(x) offsetof(OptionsContext, x)
4467 static const OptionDef options[] = {
4469 #include "cmdutils_common_opts.h"
4470 { "f", HAS_ARG | OPT_STRING | OPT_OFFSET, {.off = OFFSET(format)}, "force format", "fmt" },
4471 { "i", HAS_ARG | OPT_FUNC2, {(void*)opt_input_file}, "input file name", "filename" },
4472 { "y", OPT_BOOL, {(void*)&file_overwrite}, "overwrite output files" },
4473 { "n", OPT_BOOL, {(void*)&no_file_overwrite}, "do not overwrite output files" },
4474 { "c", HAS_ARG | OPT_STRING | OPT_SPEC, {.off = OFFSET(codec_names)}, "codec name", "codec" },
4475 { "codec", HAS_ARG | OPT_STRING | OPT_SPEC, {.off = OFFSET(codec_names)}, "codec name", "codec" },
4476 { "pre", HAS_ARG | OPT_STRING | OPT_SPEC, {.off = OFFSET(presets)}, "preset name", "preset" },
4477 { "map", HAS_ARG | OPT_EXPERT | OPT_FUNC2, {(void*)opt_map}, "set input stream mapping", "[-]input_file_id[:stream_specifier][,sync_file_id[:stream_specifier]]" },
4478 { "map_metadata", HAS_ARG | OPT_STRING | OPT_SPEC, {.off = OFFSET(metadata_map)}, "set metadata information of outfile from infile",
4479 "outfile[,metadata]:infile[,metadata]" },
4480 { "map_chapters", OPT_INT | HAS_ARG | OPT_EXPERT | OPT_OFFSET, {.off = OFFSET(chapters_input_file)}, "set chapters mapping", "input_file_index" },
4481 { "t", HAS_ARG | OPT_TIME | OPT_OFFSET, {.off = OFFSET(recording_time)}, "record or transcode \"duration\" seconds of audio/video", "duration" },
4482 { "fs", HAS_ARG | OPT_INT64 | OPT_OFFSET, {.off = OFFSET(limit_filesize)}, "set the limit file size in bytes", "limit_size" }, //
4483 { "ss", HAS_ARG | OPT_TIME | OPT_OFFSET, {.off = OFFSET(start_time)}, "set the start time offset", "time_off" },
4484 { "itsoffset", HAS_ARG | OPT_TIME | OPT_OFFSET, {.off = OFFSET(input_ts_offset)}, "set the input ts offset", "time_off" },
4485 { "itsscale", HAS_ARG | OPT_DOUBLE | OPT_SPEC, {.off = OFFSET(ts_scale)}, "set the input ts scale", "scale" },
4486 { "metadata", HAS_ARG | OPT_STRING | OPT_SPEC, {.off = OFFSET(metadata)}, "add metadata", "string=string" },
4487 { "dframes", HAS_ARG | OPT_FUNC2, {(void*)opt_data_frames}, "set the number of data frames to record", "number" },
4488 { "benchmark", OPT_BOOL | OPT_EXPERT, {(void*)&do_benchmark},
4489 "add timings for benchmarking" },
4490 { "timelimit", HAS_ARG, {(void*)opt_timelimit}, "set max runtime in seconds", "limit" },
4491 { "dump", OPT_BOOL | OPT_EXPERT, {(void*)&do_pkt_dump},
4492 "dump each input packet" },
4493 { "hex", OPT_BOOL | OPT_EXPERT, {(void*)&do_hex_dump},
4494 "when dumping packets, also dump the payload" },
4495 { "re", OPT_BOOL | OPT_EXPERT | OPT_OFFSET, {.off = OFFSET(rate_emu)}, "read input at native frame rate", "" },
4496 { "target", HAS_ARG | OPT_FUNC2, {(void*)opt_target}, "specify target file type (\"vcd\", \"svcd\", \"dvd\", \"dv\", \"dv50\", \"pal-vcd\", \"ntsc-svcd\", ...)", "type" },
4497 { "vsync", HAS_ARG | OPT_EXPERT, {(void*)opt_vsync}, "video sync method", "" },
4498 { "async", HAS_ARG | OPT_INT | OPT_EXPERT, {(void*)&audio_sync_method}, "audio sync method", "" },
4499 { "adrift_threshold", HAS_ARG | OPT_FLOAT | OPT_EXPERT, {(void*)&audio_drift_threshold}, "audio drift threshold", "threshold" },
4500 { "copyts", OPT_BOOL | OPT_EXPERT, {(void*)©_ts}, "copy timestamps" },
4501 { "copytb", OPT_BOOL | OPT_EXPERT, {(void*)©_tb}, "copy input stream time base when stream copying" },
4502 { "shortest", OPT_BOOL | OPT_EXPERT, {(void*)&opt_shortest}, "finish encoding within shortest input" }, //
4503 { "dts_delta_threshold", HAS_ARG | OPT_FLOAT | OPT_EXPERT, {(void*)&dts_delta_threshold}, "timestamp discontinuity delta threshold", "threshold" },
4504 { "xerror", OPT_BOOL, {(void*)&exit_on_error}, "exit on error", "error" },
4505 { "copyinkf", OPT_BOOL | OPT_EXPERT | OPT_SPEC, {.off = OFFSET(copy_initial_nonkeyframes)}, "copy initial non-keyframes" },
4506 { "frames", OPT_INT64 | HAS_ARG | OPT_SPEC, {.off = OFFSET(max_frames)}, "set the number of frames to record", "number" },
4507 { "tag", OPT_STRING | HAS_ARG | OPT_SPEC, {.off = OFFSET(codec_tags)}, "force codec tag/fourcc", "fourcc/tag" },
4508 { "q", HAS_ARG | OPT_EXPERT | OPT_DOUBLE | OPT_SPEC, {.off = OFFSET(qscale)}, "use fixed quality scale (VBR)", "q" },
4509 { "qscale", HAS_ARG | OPT_EXPERT | OPT_DOUBLE | OPT_SPEC, {.off = OFFSET(qscale)}, "use fixed quality scale (VBR)", "q" },
4511 { "filter", HAS_ARG | OPT_STRING | OPT_SPEC, {.off = OFFSET(filters)}, "set stream filterchain", "filter_list" },
4513 { "stats", OPT_BOOL, {&print_stats}, "print progress report during encoding", },
4514 { "attach", HAS_ARG | OPT_FUNC2, {(void*)opt_attach}, "add an attachment to the output file", "filename" },
4515 { "dump_attachment", HAS_ARG | OPT_STRING | OPT_SPEC, {.off = OFFSET(dump_attachment)}, "extract an attachment into a file", "filename" },
4518 { "vframes", HAS_ARG | OPT_VIDEO | OPT_FUNC2, {(void*)opt_video_frames}, "set the number of video frames to record", "number" },
4519 { "r", HAS_ARG | OPT_VIDEO | OPT_STRING | OPT_SPEC, {.off = OFFSET(frame_rates)}, "set frame rate (Hz value, fraction or abbreviation)", "rate" },
4520 { "s", HAS_ARG | OPT_VIDEO | OPT_STRING | OPT_SPEC, {.off = OFFSET(frame_sizes)}, "set frame size (WxH or abbreviation)", "size" },
4521 { "aspect", HAS_ARG | OPT_VIDEO | OPT_STRING | OPT_SPEC, {.off = OFFSET(frame_aspect_ratios)}, "set aspect ratio (4:3, 16:9 or 1.3333, 1.7777)", "aspect" },
4522 { "pix_fmt", HAS_ARG | OPT_EXPERT | OPT_VIDEO | OPT_STRING | OPT_SPEC, {.off = OFFSET(frame_pix_fmts)}, "set pixel format", "format" },
4523 { "bits_per_raw_sample", OPT_INT | HAS_ARG | OPT_VIDEO, {(void*)&frame_bits_per_raw_sample}, "set the number of bits per raw sample", "number" },
4524 { "vn", OPT_BOOL | OPT_VIDEO | OPT_OFFSET, {.off = OFFSET(video_disable)}, "disable video" },
4525 { "vdt", OPT_INT | HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)&video_discard}, "discard threshold", "n" },
4526 { "rc_override", HAS_ARG | OPT_EXPERT | OPT_VIDEO | OPT_STRING | OPT_SPEC, {.off = OFFSET(rc_overrides)}, "rate control override for specific intervals", "override" },
4527 { "vcodec", HAS_ARG | OPT_VIDEO | OPT_FUNC2, {(void*)opt_video_codec}, "force video codec ('copy' to copy stream)", "codec" },
4528 { "same_quant", OPT_BOOL | OPT_VIDEO, {(void*)&same_quant},
4529 "use same quantizer as source (implies VBR)" },
4530 { "pass", HAS_ARG | OPT_VIDEO, {(void*)opt_pass}, "select the pass number (1 or 2)", "n" },
4531 { "passlogfile", HAS_ARG | OPT_VIDEO, {(void*)&opt_passlogfile}, "select two pass log file name prefix", "prefix" },
4532 { "deinterlace", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, {(void*)&do_deinterlace},
4533 "deinterlace pictures" },
4534 { "vstats", OPT_EXPERT | OPT_VIDEO, {(void*)&opt_vstats}, "dump video coding statistics to file" },
4535 { "vstats_file", HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)opt_vstats_file}, "dump video coding statistics to file", "file" },
4537 { "vf", HAS_ARG | OPT_VIDEO | OPT_FUNC2, {(void*)opt_video_filters}, "video filters", "filter list" },
4539 { "intra_matrix", HAS_ARG | OPT_EXPERT | OPT_VIDEO | OPT_STRING | OPT_SPEC, {.off = OFFSET(intra_matrices)}, "specify intra matrix coeffs", "matrix" },
4540 { "inter_matrix", HAS_ARG | OPT_EXPERT | OPT_VIDEO | OPT_STRING | OPT_SPEC, {.off = OFFSET(inter_matrices)}, "specify inter matrix coeffs", "matrix" },
4541 { "top", HAS_ARG | OPT_EXPERT | OPT_VIDEO | OPT_INT| OPT_SPEC, {.off = OFFSET(top_field_first)}, "top=1/bottom=0/auto=-1 field first", "" },
4542 { "dc", OPT_INT | HAS_ARG | OPT_EXPERT | OPT_VIDEO, {(void*)&intra_dc_precision}, "intra_dc_precision", "precision" },
4543 { "vtag", HAS_ARG | OPT_EXPERT | OPT_VIDEO | OPT_FUNC2, {(void*)opt_video_tag}, "force video tag/fourcc", "fourcc/tag" },
4544 { "qphist", OPT_BOOL | OPT_EXPERT | OPT_VIDEO, { (void *)&qp_hist }, "show QP histogram" },
4545 { "force_fps", OPT_BOOL | OPT_EXPERT | OPT_VIDEO | OPT_SPEC, {.off = OFFSET(force_fps)}, "force the selected framerate, disable the best supported framerate selection" },
4546 { "streamid", HAS_ARG | OPT_EXPERT | OPT_FUNC2, {(void*)opt_streamid}, "set the value of an outfile streamid", "streamIndex:value" },
4547 { "force_key_frames", OPT_STRING | HAS_ARG | OPT_EXPERT | OPT_VIDEO | OPT_SPEC, {.off = OFFSET(forced_key_frames)}, "force key frames at specified timestamps", "timestamps" },
4550 { "aframes", HAS_ARG | OPT_AUDIO | OPT_FUNC2, {(void*)opt_audio_frames}, "set the number of audio frames to record", "number" },
4551 { "aq", HAS_ARG | OPT_AUDIO | OPT_FUNC2, {(void*)opt_audio_qscale}, "set audio quality (codec-specific)", "quality", },
4552 { "ar", HAS_ARG | OPT_AUDIO | OPT_INT | OPT_SPEC, {.off = OFFSET(audio_sample_rate)}, "set audio sampling rate (in Hz)", "rate" },
4553 { "ac", HAS_ARG | OPT_AUDIO | OPT_INT | OPT_SPEC, {.off = OFFSET(audio_channels)}, "set number of audio channels", "channels" },
4554 { "an", OPT_BOOL | OPT_AUDIO | OPT_OFFSET, {.off = OFFSET(audio_disable)}, "disable audio" },
4555 { "acodec", HAS_ARG | OPT_AUDIO | OPT_FUNC2, {(void*)opt_audio_codec}, "force audio codec ('copy' to copy stream)", "codec" },
4556 { "atag", HAS_ARG | OPT_EXPERT | OPT_AUDIO | OPT_FUNC2, {(void*)opt_audio_tag}, "force audio tag/fourcc", "fourcc/tag" },
4557 { "vol", OPT_INT | HAS_ARG | OPT_AUDIO, {(void*)&audio_volume}, "change audio volume (256=normal)" , "volume" }, //
4558 { "sample_fmt", HAS_ARG | OPT_EXPERT | OPT_AUDIO | OPT_SPEC | OPT_STRING, {.off = OFFSET(sample_fmts)}, "set sample format", "format" },
4560 /* subtitle options */
4561 { "sn", OPT_BOOL | OPT_SUBTITLE | OPT_OFFSET, {.off = OFFSET(subtitle_disable)}, "disable subtitle" },
4562 { "scodec", HAS_ARG | OPT_SUBTITLE | OPT_FUNC2, {(void*)opt_subtitle_codec}, "force subtitle codec ('copy' to copy stream)", "codec" },
4563 { "stag", HAS_ARG | OPT_EXPERT | OPT_SUBTITLE | OPT_FUNC2, {(void*)opt_subtitle_tag}, "force subtitle tag/fourcc", "fourcc/tag" },
4566 { "isync", OPT_BOOL | OPT_EXPERT | OPT_GRAB, {(void*)&input_sync}, "sync read on input", "" },
4569 { "muxdelay", OPT_FLOAT | HAS_ARG | OPT_EXPERT | OPT_OFFSET, {.off = OFFSET(mux_max_delay)}, "set the maximum demux-decode delay", "seconds" },
4570 { "muxpreload", OPT_FLOAT | HAS_ARG | OPT_EXPERT | OPT_OFFSET, {.off = OFFSET(mux_preload)}, "set the initial demux-decode delay", "seconds" },
4572 { "bsf", HAS_ARG | OPT_STRING | OPT_SPEC, {.off = OFFSET(bitstream_filters)}, "A comma-separated list of bitstream filters", "bitstream_filters" },
4574 /* data codec support */
4575 { "dcodec", HAS_ARG | OPT_DATA | OPT_FUNC2, {(void*)opt_data_codec}, "force data codec ('copy' to copy stream)", "codec" },
4577 { "default", HAS_ARG | OPT_AUDIO | OPT_VIDEO | OPT_EXPERT, {(void*)opt_default}, "generic catch all option", "" },
4581 int main(int argc, char **argv)
4583 OptionsContext o = { 0 };
4588 av_log_set_flags(AV_LOG_SKIP_REPEATED);
4589 parse_loglevel(argc, argv, options);
4591 if(argc>1 && !strcmp(argv[1], "-d")){
4593 av_log_set_callback(log_callback_null);
4598 avcodec_register_all();
4600 avdevice_register_all();
4603 avfilter_register_all();
4606 avformat_network_init();
4608 show_banner(argc, argv, options);
4611 parse_options(&o, argc, argv, options, opt_output_file);
4613 if (nb_output_files <= 0 && nb_input_files == 0) {
4615 av_log(NULL, AV_LOG_WARNING, "Use -h to get full help or, even better, run 'man %s'\n", program_name);
4619 /* file converter / grab */
4620 if (nb_output_files <= 0) {
4621 fprintf(stderr, "At least one output file must be specified\n");
4625 if (nb_input_files == 0) {
4626 av_log(NULL, AV_LOG_FATAL, "At least one input file must be specified\n");
4631 if (transcode(output_files, nb_output_files, input_files, nb_input_files) < 0)
4633 ti = getutime() - ti;
4635 int maxrss = getmaxrss() / 1024;
4636 printf("bench: utime=%0.3fs maxrss=%ikB\n", ti / 1000000.0, maxrss);