3 * Copyright (c) 2000-2011 The Libav developers
5 * This file is part of Libav.
7 * Libav 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 * Libav 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 Libav; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
32 #include "libavformat/avformat.h"
33 #include "libavdevice/avdevice.h"
34 #include "libswscale/swscale.h"
35 #include "libavresample/avresample.h"
36 #include "libavutil/opt.h"
37 #include "libavutil/channel_layout.h"
38 #include "libavutil/parseutils.h"
39 #include "libavutil/samplefmt.h"
40 #include "libavutil/fifo.h"
41 #include "libavutil/hwcontext.h"
42 #include "libavutil/internal.h"
43 #include "libavutil/intreadwrite.h"
44 #include "libavutil/dict.h"
45 #include "libavutil/mathematics.h"
46 #include "libavutil/pixdesc.h"
47 #include "libavutil/avstring.h"
48 #include "libavutil/libm.h"
49 #include "libavutil/imgutils.h"
50 #include "libavutil/time.h"
51 #include "libavformat/os_support.h"
53 # include "libavfilter/avfilter.h"
54 # include "libavfilter/buffersrc.h"
55 # include "libavfilter/buffersink.h"
57 #if HAVE_SYS_RESOURCE_H
59 #include <sys/types.h>
60 #include <sys/resource.h>
61 #elif HAVE_GETPROCESSTIMES
64 #if HAVE_GETPROCESSMEMORYINFO
70 #include <sys/select.h>
82 #include "libavutil/avassert.h"
84 const char program_name
[] = "avconv";
85 const int program_birth_year
= 2000;
87 static FILE *vstats_file
;
89 static int nb_frames_drop
= 0;
91 static int want_sdp
= 1;
94 /* signal to input threads that they should exit; set by the main thread */
95 static int transcoding_finished
;
98 InputStream
**input_streams
= NULL
;
99 int nb_input_streams
= 0;
100 InputFile
**input_files
= NULL
;
101 int nb_input_files
= 0;
103 OutputStream
**output_streams
= NULL
;
104 int nb_output_streams
= 0;
105 OutputFile
**output_files
= NULL
;
106 int nb_output_files
= 0;
108 FilterGraph
**filtergraphs
;
111 static void term_exit(void)
113 av_log(NULL
, AV_LOG_QUIET
, "");
116 static volatile int received_sigterm
= 0;
117 static volatile int received_nb_signals
= 0;
120 sigterm_handler(int sig
)
122 received_sigterm
= sig
;
123 received_nb_signals
++;
127 static void term_init(void)
129 signal(SIGINT
, sigterm_handler
); /* Interrupt (ANSI). */
130 signal(SIGTERM
, sigterm_handler
); /* Termination (ANSI). */
132 signal(SIGXCPU
, sigterm_handler
);
136 static int decode_interrupt_cb(void *ctx
)
138 return received_nb_signals
> 1;
141 const AVIOInterruptCB int_cb
= { decode_interrupt_cb
, NULL
};
143 static void avconv_cleanup(int ret
)
147 for (i
= 0; i
< nb_filtergraphs
; i
++) {
148 FilterGraph
*fg
= filtergraphs
[i
];
149 avfilter_graph_free(&fg
->graph
);
150 for (j
= 0; j
< fg
->nb_inputs
; j
++) {
151 while (av_fifo_size(fg
->inputs
[j
]->frame_queue
)) {
153 av_fifo_generic_read(fg
->inputs
[j
]->frame_queue
, &frame
,
154 sizeof(frame
), NULL
);
155 av_frame_free(&frame
);
157 av_fifo_free(fg
->inputs
[j
]->frame_queue
);
158 av_buffer_unref(&fg
->inputs
[j
]->hw_frames_ctx
);
159 av_freep(&fg
->inputs
[j
]->name
);
160 av_freep(&fg
->inputs
[j
]);
162 av_freep(&fg
->inputs
);
163 for (j
= 0; j
< fg
->nb_outputs
; j
++) {
164 av_freep(&fg
->outputs
[j
]->name
);
165 av_freep(&fg
->outputs
[j
]->formats
);
166 av_freep(&fg
->outputs
[j
]->channel_layouts
);
167 av_freep(&fg
->outputs
[j
]->sample_rates
);
168 av_freep(&fg
->outputs
[j
]);
170 av_freep(&fg
->outputs
);
171 av_freep(&fg
->graph_desc
);
173 av_freep(&filtergraphs
[i
]);
175 av_freep(&filtergraphs
);
178 for (i
= 0; i
< nb_output_files
; i
++) {
179 OutputFile
*of
= output_files
[i
];
180 AVFormatContext
*s
= of
->ctx
;
181 if (s
&& s
->oformat
&& !(s
->oformat
->flags
& AVFMT_NOFILE
) && s
->pb
)
183 avformat_free_context(s
);
184 av_dict_free(&of
->opts
);
186 av_freep(&output_files
[i
]);
188 for (i
= 0; i
< nb_output_streams
; i
++) {
189 OutputStream
*ost
= output_streams
[i
];
191 for (j
= 0; j
< ost
->nb_bitstream_filters
; j
++)
192 av_bsf_free(&ost
->bsf_ctx
[j
]);
193 av_freep(&ost
->bsf_ctx
);
195 av_frame_free(&ost
->filtered_frame
);
197 av_parser_close(ost
->parser
);
198 avcodec_free_context(&ost
->parser_avctx
);
200 av_freep(&ost
->forced_keyframes
);
201 av_freep(&ost
->avfilter
);
202 av_freep(&ost
->logfile_prefix
);
204 avcodec_free_context(&ost
->enc_ctx
);
206 if (ost
->muxing_queue
) {
207 while (av_fifo_size(ost
->muxing_queue
)) {
209 av_fifo_generic_read(ost
->muxing_queue
, &pkt
, sizeof(pkt
), NULL
);
210 av_packet_unref(&pkt
);
212 av_fifo_free(ost
->muxing_queue
);
214 av_freep(&output_streams
[i
]);
216 for (i
= 0; i
< nb_input_files
; i
++) {
217 avformat_close_input(&input_files
[i
]->ctx
);
218 av_freep(&input_files
[i
]);
220 for (i
= 0; i
< nb_input_streams
; i
++) {
221 InputStream
*ist
= input_streams
[i
];
223 av_frame_free(&ist
->decoded_frame
);
224 av_frame_free(&ist
->filter_frame
);
225 av_dict_free(&ist
->decoder_opts
);
226 av_freep(&ist
->filters
);
227 av_freep(&ist
->hwaccel_device
);
229 avcodec_free_context(&ist
->dec_ctx
);
231 av_freep(&input_streams
[i
]);
236 av_free(vstats_filename
);
238 av_freep(&input_streams
);
239 av_freep(&input_files
);
240 av_freep(&output_streams
);
241 av_freep(&output_files
);
245 avformat_network_deinit();
247 if (received_sigterm
) {
248 av_log(NULL
, AV_LOG_INFO
, "Received signal %d: terminating.\n",
249 (int) received_sigterm
);
254 void assert_avoptions(AVDictionary
*m
)
256 AVDictionaryEntry
*t
;
257 if ((t
= av_dict_get(m
, "", NULL
, AV_DICT_IGNORE_SUFFIX
))) {
258 av_log(NULL
, AV_LOG_FATAL
, "Option %s not found.\n", t
->key
);
263 static void abort_codec_experimental(AVCodec
*c
, int encoder
)
265 const char *codec_string
= encoder ?
"encoder" : "decoder";
267 av_log(NULL
, AV_LOG_FATAL
, "%s '%s' is experimental and might produce bad "
268 "results.\nAdd '-strict experimental' if you want to use it.\n",
269 codec_string
, c
->name
);
270 codec
= encoder ?
avcodec_find_encoder(c
->id
) : avcodec_find_decoder(c
->id
);
271 if (!(codec
->capabilities
& AV_CODEC_CAP_EXPERIMENTAL
))
272 av_log(NULL
, AV_LOG_FATAL
, "Or use the non experimental %s '%s'.\n",
273 codec_string
, codec
->name
);
277 static void write_packet(OutputFile
*of
, AVPacket
*pkt
, OutputStream
*ost
)
279 AVFormatContext
*s
= of
->ctx
;
280 AVStream
*st
= ost
->st
;
283 if (!of
->header_written
) {
285 /* the muxer is not initialized yet, buffer the packet */
286 if (!av_fifo_space(ost
->muxing_queue
)) {
287 int new_size
= FFMIN(2 * av_fifo_size(ost
->muxing_queue
),
288 ost
->max_muxing_queue_size
);
289 if (new_size
<= av_fifo_size(ost
->muxing_queue
)) {
290 av_log(NULL
, AV_LOG_ERROR
,
291 "Too many packets buffered for output stream %d:%d.\n",
292 ost
->file_index
, ost
->st
->index
);
295 ret
= av_fifo_realloc2(ost
->muxing_queue
, new_size
);
299 av_packet_move_ref(&tmp_pkt
, pkt
);
300 av_fifo_generic_write(ost
->muxing_queue
, &tmp_pkt
, sizeof(tmp_pkt
), NULL
);
305 * Audio encoders may split the packets -- #frames in != #packets out.
306 * But there is no reordering, so we can limit the number of output packets
307 * by simply dropping them here.
308 * Counting encoded video frames needs to be done separately because of
309 * reordering, see do_video_out()
311 if (!(st
->codecpar
->codec_type
== AVMEDIA_TYPE_VIDEO
&& ost
->encoding_needed
)) {
312 if (ost
->frame_number
>= ost
->max_frames
) {
313 av_packet_unref(pkt
);
318 if (st
->codecpar
->codec_type
== AVMEDIA_TYPE_VIDEO
) {
319 uint8_t *sd
= av_packet_get_side_data(pkt
, AV_PKT_DATA_QUALITY_FACTOR
,
321 ost
->quality
= sd ?
*(int *)sd
: -1;
323 if (ost
->frame_rate
.num
) {
324 pkt
->duration
= av_rescale_q(1, av_inv_q(ost
->frame_rate
),
329 if (!(s
->oformat
->flags
& AVFMT_NOTIMESTAMPS
) &&
330 ost
->last_mux_dts
!= AV_NOPTS_VALUE
&&
331 pkt
->dts
< ost
->last_mux_dts
+ !(s
->oformat
->flags
& AVFMT_TS_NONSTRICT
)) {
332 av_log(NULL
, AV_LOG_WARNING
, "Non-monotonous DTS in output stream "
333 "%d:%d; previous: %"PRId64
", current: %"PRId64
"; ",
334 ost
->file_index
, ost
->st
->index
, ost
->last_mux_dts
, pkt
->dts
);
336 av_log(NULL
, AV_LOG_FATAL
, "aborting.\n");
339 av_log(NULL
, AV_LOG_WARNING
, "changing to %"PRId64
". This may result "
340 "in incorrect timestamps in the output file.\n",
341 ost
->last_mux_dts
+ 1);
342 pkt
->dts
= ost
->last_mux_dts
+ 1;
343 if (pkt
->pts
!= AV_NOPTS_VALUE
)
344 pkt
->pts
= FFMAX(pkt
->pts
, pkt
->dts
);
346 ost
->last_mux_dts
= pkt
->dts
;
348 ost
->data_size
+= pkt
->size
;
349 ost
->packets_written
++;
351 pkt
->stream_index
= ost
->index
;
352 av_packet_rescale_ts(pkt
, ost
->mux_timebase
, ost
->st
->time_base
);
354 ret
= av_interleaved_write_frame(s
, pkt
);
356 print_error("av_interleaved_write_frame()", ret
);
361 static void output_packet(OutputFile
*of
, AVPacket
*pkt
, OutputStream
*ost
)
365 /* apply the output bitstream filters, if any */
366 if (ost
->nb_bitstream_filters
) {
369 ret
= av_bsf_send_packet(ost
->bsf_ctx
[0], pkt
);
375 /* get a packet from the previous filter up the chain */
376 ret
= av_bsf_receive_packet(ost
->bsf_ctx
[idx
- 1], pkt
);
377 if (ret
== AVERROR(EAGAIN
)) {
384 /* send it to the next filter down the chain or to the muxer */
385 if (idx
< ost
->nb_bitstream_filters
) {
386 ret
= av_bsf_send_packet(ost
->bsf_ctx
[idx
], pkt
);
391 write_packet(of
, pkt
, ost
);
394 write_packet(of
, pkt
, ost
);
397 if (ret
< 0 && ret
!= AVERROR_EOF
) {
398 av_log(NULL
, AV_LOG_FATAL
, "Error applying bitstream filters to an output "
399 "packet for stream #%d:%d.\n", ost
->file_index
, ost
->index
);
404 static int check_recording_time(OutputStream
*ost
)
406 OutputFile
*of
= output_files
[ost
->file_index
];
408 if (of
->recording_time
!= INT64_MAX
&&
409 av_compare_ts(ost
->sync_opts
- ost
->first_pts
, ost
->enc_ctx
->time_base
, of
->recording_time
,
410 AV_TIME_BASE_Q
) >= 0) {
417 static void do_audio_out(OutputFile
*of
, OutputStream
*ost
,
420 AVCodecContext
*enc
= ost
->enc_ctx
;
424 av_init_packet(&pkt
);
428 if (frame
->pts
== AV_NOPTS_VALUE
|| audio_sync_method
< 0)
429 frame
->pts
= ost
->sync_opts
;
430 ost
->sync_opts
= frame
->pts
+ frame
->nb_samples
;
432 ost
->samples_encoded
+= frame
->nb_samples
;
433 ost
->frames_encoded
++;
435 ret
= avcodec_send_frame(enc
, frame
);
440 ret
= avcodec_receive_packet(enc
, &pkt
);
441 if (ret
== AVERROR(EAGAIN
))
446 output_packet(of
, &pkt
, ost
);
451 av_log(NULL
, AV_LOG_FATAL
, "Audio encoding failed\n");
455 static void do_subtitle_out(OutputFile
*of
,
461 static uint8_t *subtitle_out
= NULL
;
462 int subtitle_out_max_size
= 1024 * 1024;
463 int subtitle_out_size
, nb
, i
;
467 if (pts
== AV_NOPTS_VALUE
) {
468 av_log(NULL
, AV_LOG_ERROR
, "Subtitle packets must have a pts\n");
477 subtitle_out
= av_malloc(subtitle_out_max_size
);
480 /* Note: DVB subtitle need one packet to draw them and one other
481 packet to clear them */
482 /* XXX: signal it in the codec context ? */
483 if (enc
->codec_id
== AV_CODEC_ID_DVB_SUBTITLE
)
488 for (i
= 0; i
< nb
; i
++) {
489 ost
->sync_opts
= av_rescale_q(pts
, ist
->st
->time_base
, enc
->time_base
);
490 if (!check_recording_time(ost
))
493 sub
->pts
= av_rescale_q(pts
, ist
->st
->time_base
, AV_TIME_BASE_Q
);
494 // start_display_time is required to be 0
495 sub
->pts
+= av_rescale_q(sub
->start_display_time
, (AVRational
){ 1, 1000 }, AV_TIME_BASE_Q
);
496 sub
->end_display_time
-= sub
->start_display_time
;
497 sub
->start_display_time
= 0;
499 ost
->frames_encoded
++;
501 subtitle_out_size
= avcodec_encode_subtitle(enc
, subtitle_out
,
502 subtitle_out_max_size
, sub
);
503 if (subtitle_out_size
< 0) {
504 av_log(NULL
, AV_LOG_FATAL
, "Subtitle encoding failed\n");
508 av_init_packet(&pkt
);
509 pkt
.data
= subtitle_out
;
510 pkt
.size
= subtitle_out_size
;
511 pkt
.pts
= av_rescale_q(sub
->pts
, AV_TIME_BASE_Q
, ost
->mux_timebase
);
512 if (enc
->codec_id
== AV_CODEC_ID_DVB_SUBTITLE
) {
513 /* XXX: the pts correction is handled here. Maybe handling
514 it in the codec would be better */
516 pkt
.pts
+= 90 * sub
->start_display_time
;
518 pkt
.pts
+= 90 * sub
->end_display_time
;
520 output_packet(of
, &pkt
, ost
);
524 static void do_video_out(OutputFile
*of
,
529 int ret
, format_video_sync
;
531 AVCodecContext
*enc
= ost
->enc_ctx
;
535 format_video_sync
= video_sync_method
;
536 if (format_video_sync
== VSYNC_AUTO
)
537 format_video_sync
= (of
->ctx
->oformat
->flags
& AVFMT_NOTIMESTAMPS
) ? VSYNC_PASSTHROUGH
:
538 (of
->ctx
->oformat
->flags
& AVFMT_VARIABLE_FPS
) ? VSYNC_VFR
: VSYNC_CFR
;
539 if (format_video_sync
!= VSYNC_PASSTHROUGH
&&
541 in_picture
->pts
!= AV_NOPTS_VALUE
&&
542 in_picture
->pts
< ost
->sync_opts
) {
544 av_log(NULL
, AV_LOG_WARNING
,
545 "*** dropping frame %d from stream %d at ts %"PRId64
"\n",
546 ost
->frame_number
, ost
->st
->index
, in_picture
->pts
);
550 if (in_picture
->pts
== AV_NOPTS_VALUE
)
551 in_picture
->pts
= ost
->sync_opts
;
552 ost
->sync_opts
= in_picture
->pts
;
555 if (!ost
->frame_number
)
556 ost
->first_pts
= in_picture
->pts
;
558 av_init_packet(&pkt
);
562 if (ost
->frame_number
>= ost
->max_frames
)
565 if (enc
->flags
& (AV_CODEC_FLAG_INTERLACED_DCT
| AV_CODEC_FLAG_INTERLACED_ME
) &&
566 ost
->top_field_first
>= 0)
567 in_picture
->top_field_first
= !!ost
->top_field_first
;
569 in_picture
->quality
= enc
->global_quality
;
570 in_picture
->pict_type
= 0;
571 if (ost
->forced_kf_index
< ost
->forced_kf_count
&&
572 in_picture
->pts
>= ost
->forced_kf_pts
[ost
->forced_kf_index
]) {
573 in_picture
->pict_type
= AV_PICTURE_TYPE_I
;
574 ost
->forced_kf_index
++;
577 ost
->frames_encoded
++;
579 ret
= avcodec_send_frame(enc
, in_picture
);
584 * For video, there may be reordering, so we can't throw away frames on
585 * encoder flush, we need to limit them here, before they go into encoder.
590 ret
= avcodec_receive_packet(enc
, &pkt
);
591 if (ret
== AVERROR(EAGAIN
))
596 output_packet(of
, &pkt
, ost
);
597 *frame_size
= pkt
.size
;
599 /* if two pass, output log */
600 if (ost
->logfile
&& enc
->stats_out
) {
601 fprintf(ost
->logfile
, "%s", enc
->stats_out
);
609 av_assert0(ret
!= AVERROR(EAGAIN
) && ret
!= AVERROR_EOF
);
610 av_log(NULL
, AV_LOG_FATAL
, "Video encoding failed\n");
614 #if FF_API_CODED_FRAME && FF_API_ERROR_FRAME
615 static double psnr(double d
)
617 return -10.0 * log(d
) / log(10.0);
621 static void do_video_stats(OutputStream
*ost
, int frame_size
)
625 double ti1
, bitrate
, avg_bitrate
;
627 /* this is executed just the first time do_video_stats is called */
629 vstats_file
= fopen(vstats_filename
, "w");
637 if (enc
->codec_type
== AVMEDIA_TYPE_VIDEO
) {
638 frame_number
= ost
->frame_number
;
639 fprintf(vstats_file
, "frame= %5d q= %2.1f ", frame_number
,
640 ost
->quality
/ (float)FF_QP2LAMBDA
);
642 #if FF_API_CODED_FRAME && FF_API_ERROR_FRAME
643 FF_DISABLE_DEPRECATION_WARNINGS
644 if (enc
->flags
& AV_CODEC_FLAG_PSNR
)
645 fprintf(vstats_file
, "PSNR= %6.2f ", psnr(enc
->coded_frame
->error
[0] / (enc
->width
* enc
->height
* 255.0 * 255.0)));
646 FF_ENABLE_DEPRECATION_WARNINGS
649 fprintf(vstats_file
,"f_size= %6d ", frame_size
);
650 /* compute pts value */
651 ti1
= ost
->sync_opts
* av_q2d(enc
->time_base
);
655 bitrate
= (frame_size
* 8) / av_q2d(enc
->time_base
) / 1000.0;
656 avg_bitrate
= (double)(ost
->data_size
* 8) / ti1
/ 1000.0;
657 fprintf(vstats_file
, "s_size= %8.0fkB time= %0.3f br= %7.1fkbits/s avg_br= %7.1fkbits/s ",
658 (double)ost
->data_size
/ 1024, ti1
, bitrate
, avg_bitrate
);
659 #if FF_API_CODED_FRAME
660 FF_DISABLE_DEPRECATION_WARNINGS
661 fprintf(vstats_file
, "type= %c\n", av_get_picture_type_char(enc
->coded_frame
->pict_type
));
662 FF_ENABLE_DEPRECATION_WARNINGS
667 static int init_output_stream(OutputStream
*ost
, char *error
, int error_len
);
670 * Read one frame for lavfi output for ost and encode it.
672 static int poll_filter(OutputStream
*ost
)
674 OutputFile
*of
= output_files
[ost
->file_index
];
675 AVFrame
*filtered_frame
= NULL
;
678 if (!ost
->filtered_frame
&& !(ost
->filtered_frame
= av_frame_alloc())) {
679 return AVERROR(ENOMEM
);
681 filtered_frame
= ost
->filtered_frame
;
683 if (!ost
->initialized
) {
685 ret
= init_output_stream(ost
, error
, sizeof(error
));
687 av_log(NULL
, AV_LOG_ERROR
, "Error initializing output stream %d:%d -- %s\n",
688 ost
->file_index
, ost
->index
, error
);
693 if (ost
->enc
->type
== AVMEDIA_TYPE_AUDIO
&&
694 !(ost
->enc
->capabilities
& AV_CODEC_CAP_VARIABLE_FRAME_SIZE
))
695 ret
= av_buffersink_get_samples(ost
->filter
->filter
, filtered_frame
,
696 ost
->enc_ctx
->frame_size
);
698 ret
= av_buffersink_get_frame(ost
->filter
->filter
, filtered_frame
);
703 if (filtered_frame
->pts
!= AV_NOPTS_VALUE
) {
704 int64_t start_time
= (of
->start_time
== AV_NOPTS_VALUE
) ?
0 : of
->start_time
;
705 filtered_frame
->pts
= av_rescale_q(filtered_frame
->pts
,
706 ost
->filter
->filter
->inputs
[0]->time_base
,
707 ost
->enc_ctx
->time_base
) -
708 av_rescale_q(start_time
,
710 ost
->enc_ctx
->time_base
);
713 switch (ost
->filter
->filter
->inputs
[0]->type
) {
714 case AVMEDIA_TYPE_VIDEO
:
715 if (!ost
->frame_aspect_ratio
)
716 ost
->enc_ctx
->sample_aspect_ratio
= filtered_frame
->sample_aspect_ratio
;
718 do_video_out(of
, ost
, filtered_frame
, &frame_size
);
719 if (vstats_filename
&& frame_size
)
720 do_video_stats(ost
, frame_size
);
722 case AVMEDIA_TYPE_AUDIO
:
723 do_audio_out(of
, ost
, filtered_frame
);
726 // TODO support subtitle filters
730 av_frame_unref(filtered_frame
);
735 static void finish_output_stream(OutputStream
*ost
)
737 OutputFile
*of
= output_files
[ost
->file_index
];
743 for (i
= 0; i
< of
->ctx
->nb_streams
; i
++)
744 output_streams
[of
->ost_index
+ i
]->finished
= 1;
749 * Read as many frames from possible from lavfi and encode them.
751 * Always read from the active stream with the lowest timestamp. If no frames
752 * are available for it then return EAGAIN and wait for more input. This way we
753 * can use lavfi sources that generate unlimited amount of frames without memory
756 static int poll_filters(void)
760 while (ret
>= 0 && !received_sigterm
) {
761 OutputStream
*ost
= NULL
;
762 int64_t min_pts
= INT64_MAX
;
764 /* choose output stream with the lowest timestamp */
765 for (i
= 0; i
< nb_output_streams
; i
++) {
766 int64_t pts
= output_streams
[i
]->sync_opts
;
768 if (output_streams
[i
]->filter
&& !output_streams
[i
]->filter
->graph
->graph
&&
769 !output_streams
[i
]->filter
->graph
->nb_inputs
) {
770 ret
= configure_filtergraph(output_streams
[i
]->filter
->graph
);
772 av_log(NULL
, AV_LOG_ERROR
, "Error reinitializing filters!\n");
777 if (!output_streams
[i
]->filter
|| output_streams
[i
]->finished
||
778 !output_streams
[i
]->filter
->graph
->graph
)
781 pts
= av_rescale_q(pts
, output_streams
[i
]->enc_ctx
->time_base
,
785 ost
= output_streams
[i
];
792 ret
= poll_filter(ost
);
794 if (ret
== AVERROR_EOF
) {
795 finish_output_stream(ost
);
797 } else if (ret
== AVERROR(EAGAIN
))
804 static void print_final_stats(int64_t total_size
)
806 uint64_t video_size
= 0, audio_size
= 0, extra_size
= 0, other_size
= 0;
807 uint64_t data_size
= 0;
808 float percent
= -1.0;
811 for (i
= 0; i
< nb_output_streams
; i
++) {
812 OutputStream
*ost
= output_streams
[i
];
813 switch (ost
->enc_ctx
->codec_type
) {
814 case AVMEDIA_TYPE_VIDEO
: video_size
+= ost
->data_size
; break;
815 case AVMEDIA_TYPE_AUDIO
: audio_size
+= ost
->data_size
; break;
816 default: other_size
+= ost
->data_size
; break;
818 extra_size
+= ost
->enc_ctx
->extradata_size
;
819 data_size
+= ost
->data_size
;
822 if (data_size
&& total_size
>= data_size
)
823 percent
= 100.0 * (total_size
- data_size
) / data_size
;
825 av_log(NULL
, AV_LOG_INFO
, "\n");
826 av_log(NULL
, AV_LOG_INFO
, "video:%1.0fkB audio:%1.0fkB other streams:%1.0fkB global headers:%1.0fkB muxing overhead: ",
830 extra_size
/ 1024.0);
832 av_log(NULL
, AV_LOG_INFO
, "%f%%", percent
);
834 av_log(NULL
, AV_LOG_INFO
, "unknown");
835 av_log(NULL
, AV_LOG_INFO
, "\n");
837 /* print verbose per-stream stats */
838 for (i
= 0; i
< nb_input_files
; i
++) {
839 InputFile
*f
= input_files
[i
];
840 uint64_t total_packets
= 0, total_size
= 0;
842 av_log(NULL
, AV_LOG_VERBOSE
, "Input file #%d (%s):\n",
843 i
, f
->ctx
->filename
);
845 for (j
= 0; j
< f
->nb_streams
; j
++) {
846 InputStream
*ist
= input_streams
[f
->ist_index
+ j
];
847 enum AVMediaType type
= ist
->dec_ctx
->codec_type
;
849 total_size
+= ist
->data_size
;
850 total_packets
+= ist
->nb_packets
;
852 av_log(NULL
, AV_LOG_VERBOSE
, " Input stream #%d:%d (%s): ",
853 i
, j
, media_type_string(type
));
854 av_log(NULL
, AV_LOG_VERBOSE
, "%"PRIu64
" packets read (%"PRIu64
" bytes); ",
855 ist
->nb_packets
, ist
->data_size
);
857 if (ist
->decoding_needed
) {
858 av_log(NULL
, AV_LOG_VERBOSE
, "%"PRIu64
" frames decoded",
859 ist
->frames_decoded
);
860 if (type
== AVMEDIA_TYPE_AUDIO
)
861 av_log(NULL
, AV_LOG_VERBOSE
, " (%"PRIu64
" samples)", ist
->samples_decoded
);
862 av_log(NULL
, AV_LOG_VERBOSE
, "; ");
865 av_log(NULL
, AV_LOG_VERBOSE
, "\n");
868 av_log(NULL
, AV_LOG_VERBOSE
, " Total: %"PRIu64
" packets (%"PRIu64
" bytes) demuxed\n",
869 total_packets
, total_size
);
872 for (i
= 0; i
< nb_output_files
; i
++) {
873 OutputFile
*of
= output_files
[i
];
874 uint64_t total_packets
= 0, total_size
= 0;
876 av_log(NULL
, AV_LOG_VERBOSE
, "Output file #%d (%s):\n",
877 i
, of
->ctx
->filename
);
879 for (j
= 0; j
< of
->ctx
->nb_streams
; j
++) {
880 OutputStream
*ost
= output_streams
[of
->ost_index
+ j
];
881 enum AVMediaType type
= ost
->enc_ctx
->codec_type
;
883 total_size
+= ost
->data_size
;
884 total_packets
+= ost
->packets_written
;
886 av_log(NULL
, AV_LOG_VERBOSE
, " Output stream #%d:%d (%s): ",
887 i
, j
, media_type_string(type
));
888 if (ost
->encoding_needed
) {
889 av_log(NULL
, AV_LOG_VERBOSE
, "%"PRIu64
" frames encoded",
890 ost
->frames_encoded
);
891 if (type
== AVMEDIA_TYPE_AUDIO
)
892 av_log(NULL
, AV_LOG_VERBOSE
, " (%"PRIu64
" samples)", ost
->samples_encoded
);
893 av_log(NULL
, AV_LOG_VERBOSE
, "; ");
896 av_log(NULL
, AV_LOG_VERBOSE
, "%"PRIu64
" packets muxed (%"PRIu64
" bytes); ",
897 ost
->packets_written
, ost
->data_size
);
899 av_log(NULL
, AV_LOG_VERBOSE
, "\n");
902 av_log(NULL
, AV_LOG_VERBOSE
, " Total: %"PRIu64
" packets (%"PRIu64
" bytes) muxed\n",
903 total_packets
, total_size
);
907 static void print_report(int is_last_report
, int64_t timer_start
)
914 int frame_number
, vid
, i
;
915 double bitrate
, ti1
, pts
;
916 static int64_t last_time
= -1;
917 static int qp_histogram
[52];
919 if (!print_stats
&& !is_last_report
)
922 if (!is_last_report
) {
924 /* display the report every 0.5 seconds */
925 cur_time
= av_gettime_relative();
926 if (last_time
== -1) {
927 last_time
= cur_time
;
930 if ((cur_time
- last_time
) < 500000)
932 last_time
= cur_time
;
936 oc
= output_files
[0]->ctx
;
938 total_size
= avio_size(oc
->pb
);
939 if (total_size
<= 0) // FIXME improve avio_size() so it works with non seekable output too
940 total_size
= avio_tell(oc
->pb
);
941 if (total_size
< 0) {
943 av_strerror(total_size
, errbuf
, sizeof(errbuf
));
944 av_log(NULL
, AV_LOG_VERBOSE
, "Bitrate not available, "
945 "avio_tell() failed: %s\n", errbuf
);
952 for (i
= 0; i
< nb_output_streams
; i
++) {
954 ost
= output_streams
[i
];
956 if (!ost
->stream_copy
)
957 q
= ost
->quality
/ (float) FF_QP2LAMBDA
;
959 if (vid
&& enc
->codec_type
== AVMEDIA_TYPE_VIDEO
) {
960 snprintf(buf
+ strlen(buf
), sizeof(buf
) - strlen(buf
), "q=%2.1f ", q
);
962 if (!vid
&& enc
->codec_type
== AVMEDIA_TYPE_VIDEO
) {
963 float t
= (av_gettime_relative() - timer_start
) / 1000000.0;
965 frame_number
= ost
->frame_number
;
966 snprintf(buf
+ strlen(buf
), sizeof(buf
) - strlen(buf
), "frame=%5d fps=%3d q=%3.1f ",
967 frame_number
, (t
> 1) ?
(int)(frame_number
/ t
+ 0.5) : 0, q
);
969 snprintf(buf
+ strlen(buf
), sizeof(buf
) - strlen(buf
), "L");
973 if (qp
>= 0 && qp
< FF_ARRAY_ELEMS(qp_histogram
))
975 for (j
= 0; j
< 32; j
++)
976 snprintf(buf
+ strlen(buf
), sizeof(buf
) - strlen(buf
), "%X", (int)lrintf(log2(qp_histogram
[j
] + 1)));
979 #if FF_API_CODED_FRAME && FF_API_ERROR_FRAME
980 FF_DISABLE_DEPRECATION_WARNINGS
981 if (enc
->flags
& AV_CODEC_FLAG_PSNR
) {
983 double error
, error_sum
= 0;
984 double scale
, scale_sum
= 0;
985 char type
[3] = { 'Y','U','V' };
986 snprintf(buf
+ strlen(buf
), sizeof(buf
) - strlen(buf
), "PSNR=");
987 for (j
= 0; j
< 3; j
++) {
988 if (is_last_report
) {
989 error
= enc
->error
[j
];
990 scale
= enc
->width
* enc
->height
* 255.0 * 255.0 * frame_number
;
992 error
= enc
->coded_frame
->error
[j
];
993 scale
= enc
->width
* enc
->height
* 255.0 * 255.0;
999 snprintf(buf
+ strlen(buf
), sizeof(buf
) - strlen(buf
), "%c:%2.2f ", type
[j
], psnr(error
/ scale
));
1001 snprintf(buf
+ strlen(buf
), sizeof(buf
) - strlen(buf
), "*:%2.2f ", psnr(error_sum
/ scale_sum
));
1003 FF_ENABLE_DEPRECATION_WARNINGS
1007 /* compute min output value */
1008 pts
= (double)ost
->last_mux_dts
* av_q2d(ost
->mux_timebase
);
1009 if ((pts
< ti1
) && (pts
> 0))
1015 bitrate
= (double)(total_size
* 8) / ti1
/ 1000.0;
1017 snprintf(buf
+ strlen(buf
), sizeof(buf
) - strlen(buf
),
1018 "size=%8.0fkB time=%0.2f bitrate=%6.1fkbits/s",
1019 (double)total_size
/ 1024, ti1
, bitrate
);
1022 snprintf(buf
+ strlen(buf
), sizeof(buf
) - strlen(buf
), " drop=%d",
1025 av_log(NULL
, AV_LOG_INFO
, "%s \r", buf
);
1030 print_final_stats(total_size
);
1034 static void flush_encoders(void)
1038 for (i
= 0; i
< nb_output_streams
; i
++) {
1039 OutputStream
*ost
= output_streams
[i
];
1040 AVCodecContext
*enc
= ost
->enc_ctx
;
1041 OutputFile
*of
= output_files
[ost
->file_index
];
1042 int stop_encoding
= 0;
1044 if (!ost
->encoding_needed
)
1047 if (enc
->codec_type
== AVMEDIA_TYPE_AUDIO
&& enc
->frame_size
<= 1)
1050 if (enc
->codec_type
!= AVMEDIA_TYPE_VIDEO
&& enc
->codec_type
!= AVMEDIA_TYPE_AUDIO
)
1053 avcodec_send_frame(enc
, NULL
);
1056 const char *desc
= NULL
;
1058 switch (enc
->codec_type
) {
1059 case AVMEDIA_TYPE_AUDIO
:
1062 case AVMEDIA_TYPE_VIDEO
:
1071 av_init_packet(&pkt
);
1075 ret
= avcodec_receive_packet(enc
, &pkt
);
1076 if (ret
< 0 && ret
!= AVERROR_EOF
) {
1077 av_log(NULL
, AV_LOG_FATAL
, "%s encoding failed\n", desc
);
1080 if (ost
->logfile
&& enc
->stats_out
) {
1081 fprintf(ost
->logfile
, "%s", enc
->stats_out
);
1083 if (ret
== AVERROR_EOF
) {
1087 output_packet(of
, &pkt
, ost
);
1097 * Check whether a packet from ist should be written into ost at this time
1099 static int check_output_constraints(InputStream
*ist
, OutputStream
*ost
)
1101 OutputFile
*of
= output_files
[ost
->file_index
];
1102 int ist_index
= input_files
[ist
->file_index
]->ist_index
+ ist
->st
->index
;
1104 if (ost
->source_index
!= ist_index
)
1107 if (of
->start_time
!= AV_NOPTS_VALUE
&& ist
->last_dts
< of
->start_time
)
1113 static void do_streamcopy(InputStream
*ist
, OutputStream
*ost
, const AVPacket
*pkt
)
1115 OutputFile
*of
= output_files
[ost
->file_index
];
1116 InputFile
*f
= input_files
[ist
->file_index
];
1117 int64_t start_time
= (of
->start_time
== AV_NOPTS_VALUE
) ?
0 : of
->start_time
;
1118 int64_t ost_tb_start_time
= av_rescale_q(start_time
, AV_TIME_BASE_Q
, ost
->mux_timebase
);
1121 av_init_packet(&opkt
);
1123 if ((!ost
->frame_number
&& !(pkt
->flags
& AV_PKT_FLAG_KEY
)) &&
1124 !ost
->copy_initial_nonkeyframes
)
1127 if (of
->recording_time
!= INT64_MAX
&&
1128 ist
->last_dts
>= of
->recording_time
+ start_time
) {
1133 if (f
->recording_time
!= INT64_MAX
) {
1134 start_time
= f
->ctx
->start_time
;
1135 if (f
->start_time
!= AV_NOPTS_VALUE
)
1136 start_time
+= f
->start_time
;
1137 if (ist
->last_dts
>= f
->recording_time
+ start_time
) {
1143 /* force the input stream PTS */
1144 if (ost
->enc_ctx
->codec_type
== AVMEDIA_TYPE_VIDEO
)
1147 if (pkt
->pts
!= AV_NOPTS_VALUE
)
1148 opkt
.pts
= av_rescale_q(pkt
->pts
, ist
->st
->time_base
, ost
->mux_timebase
) - ost_tb_start_time
;
1150 opkt
.pts
= AV_NOPTS_VALUE
;
1152 if (pkt
->dts
== AV_NOPTS_VALUE
)
1153 opkt
.dts
= av_rescale_q(ist
->last_dts
, AV_TIME_BASE_Q
, ost
->mux_timebase
);
1155 opkt
.dts
= av_rescale_q(pkt
->dts
, ist
->st
->time_base
, ost
->mux_timebase
);
1156 opkt
.dts
-= ost_tb_start_time
;
1158 opkt
.duration
= av_rescale_q(pkt
->duration
, ist
->st
->time_base
, ost
->mux_timebase
);
1159 opkt
.flags
= pkt
->flags
;
1161 // FIXME remove the following 2 lines they shall be replaced by the bitstream filters
1162 if ( ost
->enc_ctx
->codec_id
!= AV_CODEC_ID_H264
1163 && ost
->enc_ctx
->codec_id
!= AV_CODEC_ID_MPEG1VIDEO
1164 && ost
->enc_ctx
->codec_id
!= AV_CODEC_ID_MPEG2VIDEO
1165 && ost
->enc_ctx
->codec_id
!= AV_CODEC_ID_VC1
1167 if (av_parser_change(ost
->parser
, ost
->parser_avctx
,
1168 &opkt
.data
, &opkt
.size
,
1169 pkt
->data
, pkt
->size
,
1170 pkt
->flags
& AV_PKT_FLAG_KEY
)) {
1171 opkt
.buf
= av_buffer_create(opkt
.data
, opkt
.size
, av_buffer_default_free
, NULL
, 0);
1176 opkt
.data
= pkt
->data
;
1177 opkt
.size
= pkt
->size
;
1180 output_packet(of
, &opkt
, ost
);
1183 static int ifilter_send_frame(InputFilter
*ifilter
, AVFrame
*frame
)
1185 FilterGraph
*fg
= ifilter
->graph
;
1186 int need_reinit
, ret
, i
;
1188 /* determine if the parameters for this input changed */
1189 need_reinit
= ifilter
->format
!= frame
->format
;
1190 if (!!ifilter
->hw_frames_ctx
!= !!frame
->hw_frames_ctx
||
1191 (ifilter
->hw_frames_ctx
&& ifilter
->hw_frames_ctx
->data
!= frame
->hw_frames_ctx
->data
))
1194 switch (ifilter
->ist
->st
->codecpar
->codec_type
) {
1195 case AVMEDIA_TYPE_AUDIO
:
1196 need_reinit
|= ifilter
->sample_rate
!= frame
->sample_rate
||
1197 ifilter
->channel_layout
!= frame
->channel_layout
;
1199 case AVMEDIA_TYPE_VIDEO
:
1200 need_reinit
|= ifilter
->width
!= frame
->width
||
1201 ifilter
->height
!= frame
->height
;
1206 ret
= ifilter_parameters_from_frame(ifilter
, frame
);
1211 /* (re)init the graph if possible, otherwise buffer the frame and return */
1212 if (need_reinit
|| !fg
->graph
) {
1213 for (i
= 0; i
< fg
->nb_inputs
; i
++) {
1214 if (fg
->inputs
[i
]->format
< 0) {
1215 AVFrame
*tmp
= av_frame_clone(frame
);
1217 return AVERROR(ENOMEM
);
1218 av_frame_unref(frame
);
1220 if (!av_fifo_space(ifilter
->frame_queue
)) {
1221 ret
= av_fifo_realloc2(ifilter
->frame_queue
, 2 * av_fifo_size(ifilter
->frame_queue
));
1225 av_fifo_generic_write(ifilter
->frame_queue
, &tmp
, sizeof(tmp
), NULL
);
1230 ret
= poll_filters();
1231 if (ret
< 0 && ret
!= AVERROR_EOF
) {
1233 av_strerror(ret
, errbuf
, sizeof(errbuf
));
1235 av_log(NULL
, AV_LOG_ERROR
, "Error while filtering: %s\n", errbuf
);
1239 ret
= configure_filtergraph(fg
);
1241 av_log(NULL
, AV_LOG_ERROR
, "Error reinitializing filters!\n");
1246 ret
= av_buffersrc_add_frame(ifilter
->filter
, frame
);
1248 av_log(NULL
, AV_LOG_ERROR
, "Error while filtering\n");
1255 static int ifilter_send_eof(InputFilter
*ifilter
)
1261 if (ifilter
->filter
) {
1262 ret
= av_buffersrc_add_frame(ifilter
->filter
, NULL
);
1266 // the filtergraph was never configured
1267 FilterGraph
*fg
= ifilter
->graph
;
1268 for (i
= 0; i
< fg
->nb_inputs
; i
++)
1269 if (!fg
->inputs
[i
]->eof
)
1271 if (i
== fg
->nb_inputs
) {
1272 // All the input streams have finished without the filtergraph
1273 // ever being configured.
1274 // Mark the output streams as finished.
1275 for (j
= 0; j
< fg
->nb_outputs
; j
++)
1276 finish_output_stream(fg
->outputs
[j
]->ost
);
1283 // This does not quite work like avcodec_decode_audio4/avcodec_decode_video2.
1284 // There is the following difference: if you got a frame, you must call
1285 // it again with pkt=NULL. pkt==NULL is treated differently from pkt.size==0
1286 // (pkt==NULL means get more output, pkt.size==0 is a flush/drain packet)
1287 static int decode(AVCodecContext
*avctx
, AVFrame
*frame
, int *got_frame
, AVPacket
*pkt
)
1294 ret
= avcodec_send_packet(avctx
, pkt
);
1295 // In particular, we don't expect AVERROR(EAGAIN), because we read all
1296 // decoded frames with avcodec_receive_frame() until done.
1298 return ret
== AVERROR_EOF ?
0 : ret
;
1301 ret
= avcodec_receive_frame(avctx
, frame
);
1302 if (ret
< 0 && ret
!= AVERROR(EAGAIN
) && ret
!= AVERROR_EOF
)
1310 int guess_input_channel_layout(InputStream
*ist
)
1312 AVCodecContext
*dec
= ist
->dec_ctx
;
1314 if (!dec
->channel_layout
) {
1315 char layout_name
[256];
1317 dec
->channel_layout
= av_get_default_channel_layout(dec
->channels
);
1318 if (!dec
->channel_layout
)
1320 av_get_channel_layout_string(layout_name
, sizeof(layout_name
),
1321 dec
->channels
, dec
->channel_layout
);
1322 av_log(NULL
, AV_LOG_WARNING
, "Guessed Channel Layout for Input Stream "
1323 "#%d.%d : %s\n", ist
->file_index
, ist
->st
->index
, layout_name
);
1328 static int decode_audio(InputStream
*ist
, AVPacket
*pkt
, int *got_output
,
1331 AVFrame
*decoded_frame
, *f
;
1332 AVCodecContext
*avctx
= ist
->dec_ctx
;
1333 int i
, ret
, err
= 0;
1335 if (!ist
->decoded_frame
&& !(ist
->decoded_frame
= av_frame_alloc()))
1336 return AVERROR(ENOMEM
);
1337 if (!ist
->filter_frame
&& !(ist
->filter_frame
= av_frame_alloc()))
1338 return AVERROR(ENOMEM
);
1339 decoded_frame
= ist
->decoded_frame
;
1341 ret
= decode(avctx
, decoded_frame
, got_output
, pkt
);
1344 if (!*got_output
|| ret
< 0)
1347 ist
->samples_decoded
+= decoded_frame
->nb_samples
;
1348 ist
->frames_decoded
++;
1350 /* if the decoder provides a pts, use it instead of the last packet pts.
1351 the decoder could be delaying output by a packet or more. */
1352 if (decoded_frame
->pts
!= AV_NOPTS_VALUE
)
1353 ist
->next_dts
= decoded_frame
->pts
;
1354 else if (pkt
&& pkt
->pts
!= AV_NOPTS_VALUE
) {
1355 decoded_frame
->pts
= pkt
->pts
;
1358 if (decoded_frame
->pts
!= AV_NOPTS_VALUE
)
1359 decoded_frame
->pts
= av_rescale_q(decoded_frame
->pts
,
1361 (AVRational
){1, avctx
->sample_rate
});
1362 ist
->nb_samples
= decoded_frame
->nb_samples
;
1363 for (i
= 0; i
< ist
->nb_filters
; i
++) {
1364 if (i
< ist
->nb_filters
- 1) {
1365 f
= ist
->filter_frame
;
1366 err
= av_frame_ref(f
, decoded_frame
);
1372 err
= ifilter_send_frame(ist
->filters
[i
], f
);
1377 av_frame_unref(ist
->filter_frame
);
1378 av_frame_unref(decoded_frame
);
1379 return err
< 0 ? err
: ret
;
1382 static int decode_video(InputStream
*ist
, AVPacket
*pkt
, int *got_output
,
1385 AVFrame
*decoded_frame
, *f
;
1386 int i
, ret
= 0, err
= 0;
1388 if (!ist
->decoded_frame
&& !(ist
->decoded_frame
= av_frame_alloc()))
1389 return AVERROR(ENOMEM
);
1390 if (!ist
->filter_frame
&& !(ist
->filter_frame
= av_frame_alloc()))
1391 return AVERROR(ENOMEM
);
1392 decoded_frame
= ist
->decoded_frame
;
1394 ret
= decode(ist
->dec_ctx
, decoded_frame
, got_output
, pkt
);
1397 if (!*got_output
|| ret
< 0)
1400 ist
->frames_decoded
++;
1402 if (ist
->hwaccel_retrieve_data
&& decoded_frame
->format
== ist
->hwaccel_pix_fmt
) {
1403 err
= ist
->hwaccel_retrieve_data(ist
->dec_ctx
, decoded_frame
);
1407 ist
->hwaccel_retrieved_pix_fmt
= decoded_frame
->format
;
1409 decoded_frame
->pts
= guess_correct_pts(&ist
->pts_ctx
, decoded_frame
->pts
,
1410 decoded_frame
->pkt_dts
);
1411 if (ist
->framerate
.num
)
1412 decoded_frame
->pts
= ist
->cfr_next_pts
++;
1414 if (ist
->st
->sample_aspect_ratio
.num
)
1415 decoded_frame
->sample_aspect_ratio
= ist
->st
->sample_aspect_ratio
;
1417 for (i
= 0; i
< ist
->nb_filters
; i
++) {
1418 if (i
< ist
->nb_filters
- 1) {
1419 f
= ist
->filter_frame
;
1420 err
= av_frame_ref(f
, decoded_frame
);
1426 err
= ifilter_send_frame(ist
->filters
[i
], f
);
1432 av_frame_unref(ist
->filter_frame
);
1433 av_frame_unref(decoded_frame
);
1434 return err
< 0 ? err
: ret
;
1437 static int transcode_subtitles(InputStream
*ist
, AVPacket
*pkt
, int *got_output
,
1440 AVSubtitle subtitle
;
1441 int i
, ret
= avcodec_decode_subtitle2(ist
->dec_ctx
,
1442 &subtitle
, got_output
, pkt
);
1450 ist
->frames_decoded
++;
1452 for (i
= 0; i
< nb_output_streams
; i
++) {
1453 OutputStream
*ost
= output_streams
[i
];
1455 if (!check_output_constraints(ist
, ost
) || !ost
->encoding_needed
)
1458 do_subtitle_out(output_files
[ost
->file_index
], ost
, ist
, &subtitle
, pkt
->pts
);
1461 avsubtitle_free(&subtitle
);
1465 static int send_filter_eof(InputStream
*ist
)
1468 for (i
= 0; i
< ist
->nb_filters
; i
++) {
1469 ret
= ifilter_send_eof(ist
->filters
[i
]);
1476 /* pkt = NULL means EOF (needed to flush decoder buffers) */
1477 static void process_input_packet(InputStream
*ist
, const AVPacket
*pkt
, int no_eof
)
1483 if (ist
->next_dts
== AV_NOPTS_VALUE
)
1484 ist
->next_dts
= ist
->last_dts
;
1488 av_init_packet(&avpkt
);
1495 if (pkt
&& pkt
->dts
!= AV_NOPTS_VALUE
)
1496 ist
->next_dts
= ist
->last_dts
= av_rescale_q(pkt
->dts
, ist
->st
->time_base
, AV_TIME_BASE_Q
);
1498 // while we have more to decode or while the decoder did output something on EOF
1499 while (ist
->decoding_needed
&& (!pkt
|| avpkt
.size
> 0)) {
1502 int decode_failed
= 0;
1505 ist
->last_dts
= ist
->next_dts
;
1507 switch (ist
->dec_ctx
->codec_type
) {
1508 case AVMEDIA_TYPE_AUDIO
:
1509 ret
= decode_audio (ist
, repeating ? NULL
: &avpkt
, &got_output
,
1512 case AVMEDIA_TYPE_VIDEO
:
1513 ret
= decode_video (ist
, repeating ? NULL
: &avpkt
, &got_output
,
1515 if (repeating
&& !got_output
)
1517 else if (pkt
&& pkt
->duration
)
1518 ist
->next_dts
+= av_rescale_q(pkt
->duration
, ist
->st
->time_base
, AV_TIME_BASE_Q
);
1519 else if (ist
->st
->avg_frame_rate
.num
)
1520 ist
->next_dts
+= av_rescale_q(1, av_inv_q(ist
->st
->avg_frame_rate
),
1522 else if (ist
->dec_ctx
->framerate
.num
!= 0) {
1523 int ticks
= ist
->st
->parser ? ist
->st
->parser
->repeat_pict
+ 1 :
1524 ist
->dec_ctx
->ticks_per_frame
;
1525 ist
->next_dts
+= av_rescale_q(ticks
, ist
->dec_ctx
->framerate
, AV_TIME_BASE_Q
);
1528 case AVMEDIA_TYPE_SUBTITLE
:
1531 ret
= transcode_subtitles(ist
, &avpkt
, &got_output
, &decode_failed
);
1538 if (decode_failed
) {
1539 av_log(NULL
, AV_LOG_ERROR
, "Error while decoding stream #%d:%d\n",
1540 ist
->file_index
, ist
->st
->index
);
1542 av_log(NULL
, AV_LOG_FATAL
, "Error while processing the decoded "
1543 "data for stream #%d:%d\n", ist
->file_index
, ist
->st
->index
);
1545 if (!decode_failed
|| exit_on_error
)
1556 /* after flushing, send an EOF on all the filter inputs attached to the stream */
1557 /* except when looping we need to flush but not to send an EOF */
1558 if (!pkt
&& ist
->decoding_needed
&& !no_eof
) {
1559 int ret
= send_filter_eof(ist
);
1561 av_log(NULL
, AV_LOG_FATAL
, "Error marking filters as finished\n");
1566 /* handle stream copy */
1567 if (!ist
->decoding_needed
) {
1568 ist
->last_dts
= ist
->next_dts
;
1569 switch (ist
->dec_ctx
->codec_type
) {
1570 case AVMEDIA_TYPE_AUDIO
:
1571 ist
->next_dts
+= ((int64_t)AV_TIME_BASE
* ist
->dec_ctx
->frame_size
) /
1572 ist
->dec_ctx
->sample_rate
;
1574 case AVMEDIA_TYPE_VIDEO
:
1575 if (ist
->dec_ctx
->framerate
.num
!= 0) {
1576 int ticks
= ist
->st
->parser ? ist
->st
->parser
->repeat_pict
+ 1 : ist
->dec_ctx
->ticks_per_frame
;
1577 ist
->next_dts
+= ((int64_t)AV_TIME_BASE
*
1578 ist
->dec_ctx
->framerate
.den
* ticks
) /
1579 ist
->dec_ctx
->framerate
.num
;
1584 for (i
= 0; pkt
&& i
< nb_output_streams
; i
++) {
1585 OutputStream
*ost
= output_streams
[i
];
1587 if (!check_output_constraints(ist
, ost
) || ost
->encoding_needed
)
1590 do_streamcopy(ist
, ost
, pkt
);
1596 static void print_sdp(void)
1600 AVFormatContext
**avc
;
1602 for (i
= 0; i
< nb_output_files
; i
++) {
1603 if (!output_files
[i
]->header_written
)
1607 avc
= av_malloc(sizeof(*avc
) * nb_output_files
);
1610 for (i
= 0; i
< nb_output_files
; i
++)
1611 avc
[i
] = output_files
[i
]->ctx
;
1613 av_sdp_create(avc
, nb_output_files
, sdp
, sizeof(sdp
));
1614 printf("SDP:\n%s\n", sdp
);
1619 static const HWAccel
*get_hwaccel(enum AVPixelFormat pix_fmt
)
1622 for (i
= 0; hwaccels
[i
].name
; i
++)
1623 if (hwaccels
[i
].pix_fmt
== pix_fmt
)
1624 return &hwaccels
[i
];
1628 static enum AVPixelFormat
get_format(AVCodecContext
*s
, const enum AVPixelFormat
*pix_fmts
)
1630 InputStream
*ist
= s
->opaque
;
1631 const enum AVPixelFormat
*p
;
1634 for (p
= pix_fmts
; *p
!= -1; p
++) {
1635 const AVPixFmtDescriptor
*desc
= av_pix_fmt_desc_get(*p
);
1636 const HWAccel
*hwaccel
;
1638 if (!(desc
->flags
& AV_PIX_FMT_FLAG_HWACCEL
))
1641 hwaccel
= get_hwaccel(*p
);
1643 (ist
->active_hwaccel_id
&& ist
->active_hwaccel_id
!= hwaccel
->id
) ||
1644 (ist
->hwaccel_id
!= HWACCEL_AUTO
&& ist
->hwaccel_id
!= hwaccel
->id
))
1647 ret
= hwaccel
->init(s
);
1649 if (ist
->hwaccel_id
== hwaccel
->id
) {
1650 av_log(NULL
, AV_LOG_FATAL
,
1651 "%s hwaccel requested for input stream #%d:%d, "
1652 "but cannot be initialized.\n", hwaccel
->name
,
1653 ist
->file_index
, ist
->st
->index
);
1654 return AV_PIX_FMT_NONE
;
1659 if (ist
->hw_frames_ctx
) {
1660 s
->hw_frames_ctx
= av_buffer_ref(ist
->hw_frames_ctx
);
1661 if (!s
->hw_frames_ctx
)
1662 return AV_PIX_FMT_NONE
;
1665 ist
->active_hwaccel_id
= hwaccel
->id
;
1666 ist
->hwaccel_pix_fmt
= *p
;
1673 static int get_buffer(AVCodecContext
*s
, AVFrame
*frame
, int flags
)
1675 InputStream
*ist
= s
->opaque
;
1677 if (ist
->hwaccel_get_buffer
&& frame
->format
== ist
->hwaccel_pix_fmt
)
1678 return ist
->hwaccel_get_buffer(s
, frame
, flags
);
1680 return avcodec_default_get_buffer2(s
, frame
, flags
);
1683 static int init_input_stream(int ist_index
, char *error
, int error_len
)
1686 InputStream
*ist
= input_streams
[ist_index
];
1688 if (ist
->decoding_needed
) {
1689 AVCodec
*codec
= ist
->dec
;
1691 snprintf(error
, error_len
, "Decoder (codec id %d) not found for input stream #%d:%d",
1692 ist
->dec_ctx
->codec_id
, ist
->file_index
, ist
->st
->index
);
1693 return AVERROR(EINVAL
);
1696 ist
->dec_ctx
->opaque
= ist
;
1697 ist
->dec_ctx
->get_format
= get_format
;
1698 ist
->dec_ctx
->get_buffer2
= get_buffer
;
1699 ist
->dec_ctx
->thread_safe_callbacks
= 1;
1701 av_opt_set_int(ist
->dec_ctx
, "refcounted_frames", 1, 0);
1703 if (!av_dict_get(ist
->decoder_opts
, "threads", NULL
, 0))
1704 av_dict_set(&ist
->decoder_opts
, "threads", "auto", 0);
1705 if ((ret
= avcodec_open2(ist
->dec_ctx
, codec
, &ist
->decoder_opts
)) < 0) {
1707 if (ret
== AVERROR_EXPERIMENTAL
)
1708 abort_codec_experimental(codec
, 0);
1710 av_strerror(ret
, errbuf
, sizeof(errbuf
));
1712 snprintf(error
, error_len
,
1713 "Error while opening decoder for input stream "
1715 ist
->file_index
, ist
->st
->index
, errbuf
);
1718 assert_avoptions(ist
->decoder_opts
);
1721 ist
->last_dts
= ist
->st
->avg_frame_rate
.num ?
- ist
->dec_ctx
->has_b_frames
* AV_TIME_BASE
/ av_q2d(ist
->st
->avg_frame_rate
) : 0;
1722 ist
->next_dts
= AV_NOPTS_VALUE
;
1723 init_pts_correction(&ist
->pts_ctx
);
1728 static InputStream
*get_input_stream(OutputStream
*ost
)
1730 if (ost
->source_index
>= 0)
1731 return input_streams
[ost
->source_index
];
1734 FilterGraph
*fg
= ost
->filter
->graph
;
1737 for (i
= 0; i
< fg
->nb_inputs
; i
++)
1738 if (fg
->inputs
[i
]->ist
->dec_ctx
->codec_type
== ost
->enc_ctx
->codec_type
)
1739 return fg
->inputs
[i
]->ist
;
1745 /* open the muxer when all the streams are initialized */
1746 static int check_init_output_file(OutputFile
*of
, int file_index
)
1750 for (i
= 0; i
< of
->ctx
->nb_streams
; i
++) {
1751 OutputStream
*ost
= output_streams
[of
->ost_index
+ i
];
1752 if (!ost
->initialized
)
1756 of
->ctx
->interrupt_callback
= int_cb
;
1758 ret
= avformat_write_header(of
->ctx
, &of
->opts
);
1762 av_strerror(ret
, errbuf
, sizeof(errbuf
));
1764 av_log(NULL
, AV_LOG_ERROR
,
1765 "Could not write header for output file #%d "
1766 "(incorrect codec parameters ?): %s",
1767 file_index
, errbuf
);
1770 assert_avoptions(of
->opts
);
1771 of
->header_written
= 1;
1773 av_dump_format(of
->ctx
, file_index
, of
->ctx
->filename
, 1);
1778 /* flush the muxing queues */
1779 for (i
= 0; i
< of
->ctx
->nb_streams
; i
++) {
1780 OutputStream
*ost
= output_streams
[of
->ost_index
+ i
];
1782 while (av_fifo_size(ost
->muxing_queue
)) {
1784 av_fifo_generic_read(ost
->muxing_queue
, &pkt
, sizeof(pkt
), NULL
);
1785 write_packet(of
, &pkt
, ost
);
1792 static int init_output_bsfs(OutputStream
*ost
)
1797 if (!ost
->nb_bitstream_filters
)
1800 for (i
= 0; i
< ost
->nb_bitstream_filters
; i
++) {
1801 ctx
= ost
->bsf_ctx
[i
];
1803 ret
= avcodec_parameters_copy(ctx
->par_in
,
1804 i ? ost
->bsf_ctx
[i
- 1]->par_out
: ost
->st
->codecpar
);
1808 ctx
->time_base_in
= i ? ost
->bsf_ctx
[i
- 1]->time_base_out
: ost
->st
->time_base
;
1810 ret
= av_bsf_init(ctx
);
1812 av_log(NULL
, AV_LOG_ERROR
, "Error initializing bitstream filter: %s\n",
1818 ret
= avcodec_parameters_copy(ost
->st
->codecpar
, ctx
->par_out
);
1822 ost
->st
->time_base
= ctx
->time_base_out
;
1827 static int init_output_stream_streamcopy(OutputStream
*ost
)
1829 OutputFile
*of
= output_files
[ost
->file_index
];
1830 InputStream
*ist
= get_input_stream(ost
);
1831 AVCodecParameters
*par_dst
= ost
->st
->codecpar
;
1832 AVCodecParameters
*par_src
= ist
->st
->codecpar
;
1834 uint32_t codec_tag
= par_dst
->codec_tag
;
1838 if (!of
->ctx
->oformat
->codec_tag
||
1839 av_codec_get_id (of
->ctx
->oformat
->codec_tag
, par_src
->codec_tag
) == par_src
->codec_id
||
1840 av_codec_get_tag(of
->ctx
->oformat
->codec_tag
, par_src
->codec_id
) <= 0)
1841 codec_tag
= par_src
->codec_tag
;
1844 ret
= avcodec_parameters_copy(par_dst
, par_src
);
1848 par_dst
->codec_tag
= codec_tag
;
1850 ost
->st
->disposition
= ist
->st
->disposition
;
1852 ost
->st
->time_base
= ist
->st
->time_base
;
1854 if (ist
->st
->nb_side_data
) {
1855 ost
->st
->side_data
= av_realloc_array(NULL
, ist
->st
->nb_side_data
,
1856 sizeof(*ist
->st
->side_data
));
1857 if (!ost
->st
->side_data
)
1858 return AVERROR(ENOMEM
);
1860 for (i
= 0; i
< ist
->st
->nb_side_data
; i
++) {
1861 const AVPacketSideData
*sd_src
= &ist
->st
->side_data
[i
];
1862 AVPacketSideData
*sd_dst
= &ost
->st
->side_data
[i
];
1864 sd_dst
->data
= av_malloc(sd_src
->size
);
1866 return AVERROR(ENOMEM
);
1867 memcpy(sd_dst
->data
, sd_src
->data
, sd_src
->size
);
1868 sd_dst
->size
= sd_src
->size
;
1869 sd_dst
->type
= sd_src
->type
;
1870 ost
->st
->nb_side_data
++;
1874 ost
->parser
= av_parser_init(par_dst
->codec_id
);
1875 ost
->parser_avctx
= avcodec_alloc_context3(NULL
);
1876 if (!ost
->parser_avctx
)
1877 return AVERROR(ENOMEM
);
1879 if (par_dst
->codec_type
== AVMEDIA_TYPE_VIDEO
) {
1880 if (ost
->frame_aspect_ratio
)
1881 sar
= av_d2q(ost
->frame_aspect_ratio
* par_dst
->height
/ par_dst
->width
, 255);
1882 else if (ist
->st
->sample_aspect_ratio
.num
)
1883 sar
= ist
->st
->sample_aspect_ratio
;
1885 sar
= par_src
->sample_aspect_ratio
;
1886 ost
->st
->sample_aspect_ratio
= par_dst
->sample_aspect_ratio
= sar
;
1892 static void set_encoder_id(OutputFile
*of
, OutputStream
*ost
)
1894 AVDictionaryEntry
*e
;
1896 uint8_t *encoder_string
;
1897 int encoder_string_len
;
1898 int format_flags
= 0;
1900 e
= av_dict_get(of
->opts
, "fflags", NULL
, 0);
1902 const AVOption
*o
= av_opt_find(of
->ctx
, "fflags", NULL
, 0, 0);
1905 av_opt_eval_flags(of
->ctx
, o
, e
->value
, &format_flags
);
1908 encoder_string_len
= sizeof(LIBAVCODEC_IDENT
) + strlen(ost
->enc
->name
) + 2;
1909 encoder_string
= av_mallocz(encoder_string_len
);
1910 if (!encoder_string
)
1913 if (!(format_flags
& AVFMT_FLAG_BITEXACT
))
1914 av_strlcpy(encoder_string
, LIBAVCODEC_IDENT
" ", encoder_string_len
);
1915 av_strlcat(encoder_string
, ost
->enc
->name
, encoder_string_len
);
1916 av_dict_set(&ost
->st
->metadata
, "encoder", encoder_string
,
1917 AV_DICT_DONT_STRDUP_VAL
| AV_DICT_DONT_OVERWRITE
);
1920 static void parse_forced_key_frames(char *kf
, OutputStream
*ost
,
1921 AVCodecContext
*avctx
)
1927 for (p
= kf
; *p
; p
++)
1930 ost
->forced_kf_count
= n
;
1931 ost
->forced_kf_pts
= av_malloc(sizeof(*ost
->forced_kf_pts
) * n
);
1932 if (!ost
->forced_kf_pts
) {
1933 av_log(NULL
, AV_LOG_FATAL
, "Could not allocate forced key frames array.\n");
1938 for (i
= 0; i
< n
; i
++) {
1939 char *next
= strchr(p
, ',');
1944 t
= parse_time_or_die("force_key_frames", p
, 1);
1945 ost
->forced_kf_pts
[i
] = av_rescale_q(t
, AV_TIME_BASE_Q
, avctx
->time_base
);
1951 static int init_output_stream_encode(OutputStream
*ost
)
1953 InputStream
*ist
= get_input_stream(ost
);
1954 AVCodecContext
*enc_ctx
= ost
->enc_ctx
;
1955 AVCodecContext
*dec_ctx
= NULL
;
1957 set_encoder_id(output_files
[ost
->file_index
], ost
);
1960 ost
->st
->disposition
= ist
->st
->disposition
;
1962 dec_ctx
= ist
->dec_ctx
;
1964 enc_ctx
->bits_per_raw_sample
= dec_ctx
->bits_per_raw_sample
;
1965 enc_ctx
->chroma_sample_location
= dec_ctx
->chroma_sample_location
;
1968 switch (enc_ctx
->codec_type
) {
1969 case AVMEDIA_TYPE_AUDIO
:
1970 enc_ctx
->sample_fmt
= ost
->filter
->filter
->inputs
[0]->format
;
1971 enc_ctx
->sample_rate
= ost
->filter
->filter
->inputs
[0]->sample_rate
;
1972 enc_ctx
->channel_layout
= ost
->filter
->filter
->inputs
[0]->channel_layout
;
1973 enc_ctx
->channels
= av_get_channel_layout_nb_channels(enc_ctx
->channel_layout
);
1974 enc_ctx
->time_base
= (AVRational
){ 1, enc_ctx
->sample_rate
};
1976 case AVMEDIA_TYPE_VIDEO
:
1977 enc_ctx
->time_base
= ost
->filter
->filter
->inputs
[0]->time_base
;
1979 enc_ctx
->width
= ost
->filter
->filter
->inputs
[0]->w
;
1980 enc_ctx
->height
= ost
->filter
->filter
->inputs
[0]->h
;
1981 enc_ctx
->sample_aspect_ratio
= ost
->st
->sample_aspect_ratio
=
1982 ost
->frame_aspect_ratio ?
// overridden by the -aspect cli option
1983 av_d2q(ost
->frame_aspect_ratio
* enc_ctx
->height
/enc_ctx
->width
, 255) :
1984 ost
->filter
->filter
->inputs
[0]->sample_aspect_ratio
;
1985 enc_ctx
->pix_fmt
= ost
->filter
->filter
->inputs
[0]->format
;
1987 enc_ctx
->framerate
= ost
->frame_rate
;
1989 ost
->st
->avg_frame_rate
= ost
->frame_rate
;
1992 (enc_ctx
->width
!= dec_ctx
->width
||
1993 enc_ctx
->height
!= dec_ctx
->height
||
1994 enc_ctx
->pix_fmt
!= dec_ctx
->pix_fmt
)) {
1995 enc_ctx
->bits_per_raw_sample
= 0;
1998 if (ost
->forced_keyframes
)
1999 parse_forced_key_frames(ost
->forced_keyframes
, ost
,
2002 case AVMEDIA_TYPE_SUBTITLE
:
2003 enc_ctx
->time_base
= (AVRational
){1, 1000};
2013 static int init_output_stream(OutputStream
*ost
, char *error
, int error_len
)
2017 if (ost
->encoding_needed
) {
2018 AVCodec
*codec
= ost
->enc
;
2019 AVCodecContext
*dec
= NULL
;
2022 ret
= init_output_stream_encode(ost
);
2026 if ((ist
= get_input_stream(ost
)))
2028 if (dec
&& dec
->subtitle_header
) {
2029 ost
->enc_ctx
->subtitle_header
= av_malloc(dec
->subtitle_header_size
);
2030 if (!ost
->enc_ctx
->subtitle_header
)
2031 return AVERROR(ENOMEM
);
2032 memcpy(ost
->enc_ctx
->subtitle_header
, dec
->subtitle_header
, dec
->subtitle_header_size
);
2033 ost
->enc_ctx
->subtitle_header_size
= dec
->subtitle_header_size
;
2035 if (!av_dict_get(ost
->encoder_opts
, "threads", NULL
, 0))
2036 av_dict_set(&ost
->encoder_opts
, "threads", "auto", 0);
2038 if (ost
->filter
&& ost
->filter
->filter
->inputs
[0]->hw_frames_ctx
&&
2039 ((AVHWFramesContext
*)ost
->filter
->filter
->inputs
[0]->hw_frames_ctx
->data
)->format
==
2040 ost
->filter
->filter
->inputs
[0]->format
) {
2041 ost
->enc_ctx
->hw_frames_ctx
= av_buffer_ref(ost
->filter
->filter
->inputs
[0]->hw_frames_ctx
);
2042 if (!ost
->enc_ctx
->hw_frames_ctx
)
2043 return AVERROR(ENOMEM
);
2046 if ((ret
= avcodec_open2(ost
->enc_ctx
, codec
, &ost
->encoder_opts
)) < 0) {
2047 if (ret
== AVERROR_EXPERIMENTAL
)
2048 abort_codec_experimental(codec
, 1);
2049 snprintf(error
, error_len
,
2050 "Error while opening encoder for output stream #%d:%d - "
2051 "maybe incorrect parameters such as bit_rate, rate, width or height",
2052 ost
->file_index
, ost
->index
);
2055 assert_avoptions(ost
->encoder_opts
);
2056 if (ost
->enc_ctx
->bit_rate
&& ost
->enc_ctx
->bit_rate
< 1000)
2057 av_log(NULL
, AV_LOG_WARNING
, "The bitrate parameter is set too low."
2058 "It takes bits/s as argument, not kbits/s\n");
2060 ret
= avcodec_parameters_from_context(ost
->st
->codecpar
, ost
->enc_ctx
);
2062 av_log(NULL
, AV_LOG_FATAL
,
2063 "Error initializing the output stream codec context.\n");
2067 if (ost
->enc_ctx
->nb_coded_side_data
) {
2070 ost
->st
->side_data
= av_realloc_array(NULL
, ost
->enc_ctx
->nb_coded_side_data
,
2071 sizeof(*ost
->st
->side_data
));
2072 if (!ost
->st
->side_data
)
2073 return AVERROR(ENOMEM
);
2075 for (i
= 0; i
< ost
->enc_ctx
->nb_coded_side_data
; i
++) {
2076 const AVPacketSideData
*sd_src
= &ost
->enc_ctx
->coded_side_data
[i
];
2077 AVPacketSideData
*sd_dst
= &ost
->st
->side_data
[i
];
2079 sd_dst
->data
= av_malloc(sd_src
->size
);
2081 return AVERROR(ENOMEM
);
2082 memcpy(sd_dst
->data
, sd_src
->data
, sd_src
->size
);
2083 sd_dst
->size
= sd_src
->size
;
2084 sd_dst
->type
= sd_src
->type
;
2085 ost
->st
->nb_side_data
++;
2089 ost
->st
->time_base
= ost
->enc_ctx
->time_base
;
2090 } else if (ost
->stream_copy
) {
2091 ret
= init_output_stream_streamcopy(ost
);
2096 * FIXME: will the codec context used by the parser during streamcopy
2097 * This should go away with the new parser API.
2099 ret
= avcodec_parameters_to_context(ost
->parser_avctx
, ost
->st
->codecpar
);
2104 /* initialize bitstream filters for the output stream
2105 * needs to be done here, because the codec id for streamcopy is not
2106 * known until now */
2107 ret
= init_output_bsfs(ost
);
2111 ost
->mux_timebase
= ost
->st
->time_base
;
2113 ost
->initialized
= 1;
2115 ret
= check_init_output_file(output_files
[ost
->file_index
], ost
->file_index
);
2122 static int transcode_init(void)
2124 int ret
= 0, i
, j
, k
;
2129 /* init framerate emulation */
2130 for (i
= 0; i
< nb_input_files
; i
++) {
2131 InputFile
*ifile
= input_files
[i
];
2132 if (ifile
->rate_emu
)
2133 for (j
= 0; j
< ifile
->nb_streams
; j
++)
2134 input_streams
[j
+ ifile
->ist_index
]->start
= av_gettime_relative();
2137 /* init input streams */
2138 for (i
= 0; i
< nb_input_streams
; i
++)
2139 if ((ret
= init_input_stream(i
, error
, sizeof(error
))) < 0)
2142 /* open each encoder */
2143 for (i
= 0; i
< nb_output_streams
; i
++) {
2144 // skip streams fed from filtergraphs until we have a frame for them
2145 if (output_streams
[i
]->filter
)
2148 ret
= init_output_stream(output_streams
[i
], error
, sizeof(error
));
2154 /* discard unused programs */
2155 for (i
= 0; i
< nb_input_files
; i
++) {
2156 InputFile
*ifile
= input_files
[i
];
2157 for (j
= 0; j
< ifile
->ctx
->nb_programs
; j
++) {
2158 AVProgram
*p
= ifile
->ctx
->programs
[j
];
2159 int discard
= AVDISCARD_ALL
;
2161 for (k
= 0; k
< p
->nb_stream_indexes
; k
++)
2162 if (!input_streams
[ifile
->ist_index
+ p
->stream_index
[k
]]->discard
) {
2163 discard
= AVDISCARD_DEFAULT
;
2166 p
->discard
= discard
;
2171 /* dump the stream mapping */
2172 av_log(NULL
, AV_LOG_INFO
, "Stream mapping:\n");
2173 for (i
= 0; i
< nb_input_streams
; i
++) {
2174 ist
= input_streams
[i
];
2176 for (j
= 0; j
< ist
->nb_filters
; j
++) {
2177 if (!filtergraph_is_simple(ist
->filters
[j
]->graph
)) {
2178 av_log(NULL
, AV_LOG_INFO
, " Stream #%d:%d (%s) -> %s",
2179 ist
->file_index
, ist
->st
->index
, ist
->dec ? ist
->dec
->name
: "?",
2180 ist
->filters
[j
]->name
);
2181 if (nb_filtergraphs
> 1)
2182 av_log(NULL
, AV_LOG_INFO
, " (graph %d)", ist
->filters
[j
]->graph
->index
);
2183 av_log(NULL
, AV_LOG_INFO
, "\n");
2188 for (i
= 0; i
< nb_output_streams
; i
++) {
2189 ost
= output_streams
[i
];
2191 if (ost
->attachment_filename
) {
2192 /* an attached file */
2193 av_log(NULL
, AV_LOG_INFO
, " File %s -> Stream #%d:%d\n",
2194 ost
->attachment_filename
, ost
->file_index
, ost
->index
);
2198 if (ost
->filter
&& !filtergraph_is_simple(ost
->filter
->graph
)) {
2199 /* output from a complex graph */
2200 av_log(NULL
, AV_LOG_INFO
, " %s", ost
->filter
->name
);
2201 if (nb_filtergraphs
> 1)
2202 av_log(NULL
, AV_LOG_INFO
, " (graph %d)", ost
->filter
->graph
->index
);
2204 av_log(NULL
, AV_LOG_INFO
, " -> Stream #%d:%d (%s)\n", ost
->file_index
,
2205 ost
->index
, ost
->enc ? ost
->enc
->name
: "?");
2209 av_log(NULL
, AV_LOG_INFO
, " Stream #%d:%d -> #%d:%d",
2210 input_streams
[ost
->source_index
]->file_index
,
2211 input_streams
[ost
->source_index
]->st
->index
,
2214 if (ost
->sync_ist
!= input_streams
[ost
->source_index
])
2215 av_log(NULL
, AV_LOG_INFO
, " [sync #%d:%d]",
2216 ost
->sync_ist
->file_index
,
2217 ost
->sync_ist
->st
->index
);
2218 if (ost
->stream_copy
)
2219 av_log(NULL
, AV_LOG_INFO
, " (copy)");
2221 const AVCodec
*in_codec
= input_streams
[ost
->source_index
]->dec
;
2222 const AVCodec
*out_codec
= ost
->enc
;
2223 const char *decoder_name
= "?";
2224 const char *in_codec_name
= "?";
2225 const char *encoder_name
= "?";
2226 const char *out_codec_name
= "?";
2227 const AVCodecDescriptor
*desc
;
2230 decoder_name
= in_codec
->name
;
2231 desc
= avcodec_descriptor_get(in_codec
->id
);
2233 in_codec_name
= desc
->name
;
2234 if (!strcmp(decoder_name
, in_codec_name
))
2235 decoder_name
= "native";
2239 encoder_name
= out_codec
->name
;
2240 desc
= avcodec_descriptor_get(out_codec
->id
);
2242 out_codec_name
= desc
->name
;
2243 if (!strcmp(encoder_name
, out_codec_name
))
2244 encoder_name
= "native";
2247 av_log(NULL
, AV_LOG_INFO
, " (%s (%s) -> %s (%s))",
2248 in_codec_name
, decoder_name
,
2249 out_codec_name
, encoder_name
);
2251 av_log(NULL
, AV_LOG_INFO
, "\n");
2255 av_log(NULL
, AV_LOG_ERROR
, "%s\n", error
);
2262 /* Return 1 if there remain streams where more output is wanted, 0 otherwise. */
2263 static int need_output(void)
2267 for (i
= 0; i
< nb_output_streams
; i
++) {
2268 OutputStream
*ost
= output_streams
[i
];
2269 OutputFile
*of
= output_files
[ost
->file_index
];
2270 AVFormatContext
*os
= output_files
[ost
->file_index
]->ctx
;
2272 if (ost
->finished
||
2273 (os
->pb
&& avio_tell(os
->pb
) >= of
->limit_filesize
))
2275 if (ost
->frame_number
>= ost
->max_frames
) {
2277 for (j
= 0; j
< of
->ctx
->nb_streams
; j
++)
2278 output_streams
[of
->ost_index
+ j
]->finished
= 1;
2288 static InputFile
*select_input_file(void)
2290 InputFile
*ifile
= NULL
;
2291 int64_t ipts_min
= INT64_MAX
;
2294 for (i
= 0; i
< nb_input_streams
; i
++) {
2295 InputStream
*ist
= input_streams
[i
];
2296 int64_t ipts
= ist
->last_dts
;
2298 if (ist
->discard
|| input_files
[ist
->file_index
]->eagain
)
2300 if (!input_files
[ist
->file_index
]->eof_reached
) {
2301 if (ipts
< ipts_min
) {
2303 ifile
= input_files
[ist
->file_index
];
2312 static void *input_thread(void *arg
)
2317 while (!transcoding_finished
&& ret
>= 0) {
2319 ret
= av_read_frame(f
->ctx
, &pkt
);
2321 if (ret
== AVERROR(EAGAIN
)) {
2328 pthread_mutex_lock(&f
->fifo_lock
);
2329 while (!av_fifo_space(f
->fifo
))
2330 pthread_cond_wait(&f
->fifo_cond
, &f
->fifo_lock
);
2332 av_fifo_generic_write(f
->fifo
, &pkt
, sizeof(pkt
), NULL
);
2334 pthread_mutex_unlock(&f
->fifo_lock
);
2341 static void free_input_threads(void)
2345 if (nb_input_files
== 1)
2348 transcoding_finished
= 1;
2350 for (i
= 0; i
< nb_input_files
; i
++) {
2351 InputFile
*f
= input_files
[i
];
2354 if (!f
->fifo
|| f
->joined
)
2357 pthread_mutex_lock(&f
->fifo_lock
);
2358 while (av_fifo_size(f
->fifo
)) {
2359 av_fifo_generic_read(f
->fifo
, &pkt
, sizeof(pkt
), NULL
);
2360 av_packet_unref(&pkt
);
2362 pthread_cond_signal(&f
->fifo_cond
);
2363 pthread_mutex_unlock(&f
->fifo_lock
);
2365 pthread_join(f
->thread
, NULL
);
2368 while (av_fifo_size(f
->fifo
)) {
2369 av_fifo_generic_read(f
->fifo
, &pkt
, sizeof(pkt
), NULL
);
2370 av_packet_unref(&pkt
);
2372 av_fifo_free(f
->fifo
);
2376 static int init_input_threads(void)
2380 if (nb_input_files
== 1)
2383 for (i
= 0; i
< nb_input_files
; i
++) {
2384 InputFile
*f
= input_files
[i
];
2386 if (!(f
->fifo
= av_fifo_alloc(8*sizeof(AVPacket
))))
2387 return AVERROR(ENOMEM
);
2389 pthread_mutex_init(&f
->fifo_lock
, NULL
);
2390 pthread_cond_init (&f
->fifo_cond
, NULL
);
2392 if ((ret
= pthread_create(&f
->thread
, NULL
, input_thread
, f
)))
2393 return AVERROR(ret
);
2398 static int get_input_packet_mt(InputFile
*f
, AVPacket
*pkt
)
2402 pthread_mutex_lock(&f
->fifo_lock
);
2404 if (av_fifo_size(f
->fifo
)) {
2405 av_fifo_generic_read(f
->fifo
, pkt
, sizeof(*pkt
), NULL
);
2406 pthread_cond_signal(&f
->fifo_cond
);
2411 ret
= AVERROR(EAGAIN
);
2414 pthread_mutex_unlock(&f
->fifo_lock
);
2420 static int get_input_packet(InputFile
*f
, AVPacket
*pkt
)
2424 for (i
= 0; i
< f
->nb_streams
; i
++) {
2425 InputStream
*ist
= input_streams
[f
->ist_index
+ i
];
2426 int64_t pts
= av_rescale(ist
->last_dts
, 1000000, AV_TIME_BASE
);
2427 int64_t now
= av_gettime_relative() - ist
->start
;
2429 return AVERROR(EAGAIN
);
2434 if (nb_input_files
> 1)
2435 return get_input_packet_mt(f
, pkt
);
2437 return av_read_frame(f
->ctx
, pkt
);
2440 static int got_eagain(void)
2443 for (i
= 0; i
< nb_input_files
; i
++)
2444 if (input_files
[i
]->eagain
)
2449 static void reset_eagain(void)
2452 for (i
= 0; i
< nb_input_files
; i
++)
2453 input_files
[i
]->eagain
= 0;
2456 // set duration to max(tmp, duration) in a proper time base and return duration's time_base
2457 static AVRational
duration_max(int64_t tmp
, int64_t *duration
, AVRational tmp_time_base
,
2458 AVRational time_base
)
2464 return tmp_time_base
;
2467 ret
= av_compare_ts(*duration
, time_base
, tmp
, tmp_time_base
);
2470 return tmp_time_base
;
2476 static int seek_to_start(InputFile
*ifile
, AVFormatContext
*is
)
2479 AVCodecContext
*avctx
;
2480 int i
, ret
, has_audio
= 0;
2481 int64_t duration
= 0;
2483 ret
= av_seek_frame(is
, -1, is
->start_time
, 0);
2487 for (i
= 0; i
< ifile
->nb_streams
; i
++) {
2488 ist
= input_streams
[ifile
->ist_index
+ i
];
2489 avctx
= ist
->dec_ctx
;
2492 if (ist
->decoding_needed
) {
2493 process_input_packet(ist
, NULL
, 1);
2494 avcodec_flush_buffers(avctx
);
2497 /* duration is the length of the last frame in a stream
2498 * when audio stream is present we don't care about
2499 * last video frame length because it's not defined exactly */
2500 if (avctx
->codec_type
== AVMEDIA_TYPE_AUDIO
&& ist
->nb_samples
)
2504 for (i
= 0; i
< ifile
->nb_streams
; i
++) {
2505 ist
= input_streams
[ifile
->ist_index
+ i
];
2506 avctx
= ist
->dec_ctx
;
2509 if (avctx
->codec_type
== AVMEDIA_TYPE_AUDIO
&& ist
->nb_samples
) {
2510 AVRational sample_rate
= {1, avctx
->sample_rate
};
2512 duration
= av_rescale_q(ist
->nb_samples
, sample_rate
, ist
->st
->time_base
);
2516 if (ist
->framerate
.num
) {
2517 duration
= av_rescale_q(1, ist
->framerate
, ist
->st
->time_base
);
2518 } else if (ist
->st
->avg_frame_rate
.num
) {
2519 duration
= av_rescale_q(1, ist
->st
->avg_frame_rate
, ist
->st
->time_base
);
2520 } else duration
= 1;
2522 if (!ifile
->duration
)
2523 ifile
->time_base
= ist
->st
->time_base
;
2524 /* the total duration of the stream, max_pts - min_pts is
2525 * the duration of the stream without the last frame */
2526 duration
+= ist
->max_pts
- ist
->min_pts
;
2527 ifile
->time_base
= duration_max(duration
, &ifile
->duration
, ist
->st
->time_base
,
2531 if (ifile
->loop
> 0)
2538 * Read one packet from an input file and send it for
2539 * - decoding -> lavfi (audio/video)
2540 * - decoding -> encoding -> muxing (subtitles)
2541 * - muxing (streamcopy)
2544 * - 0 -- one packet was read and processed
2545 * - AVERROR(EAGAIN) -- no packets were available for selected file,
2546 * this function should be called again
2547 * - AVERROR_EOF -- this function should not be called again
2549 static int process_input(void)
2552 AVFormatContext
*is
;
2558 /* select the stream that we must read now */
2559 ifile
= select_input_file();
2560 /* if none, if is finished */
2565 return AVERROR(EAGAIN
);
2567 av_log(NULL
, AV_LOG_VERBOSE
, "No more inputs to read from.\n");
2572 ret
= get_input_packet(ifile
, &pkt
);
2574 if (ret
== AVERROR(EAGAIN
)) {
2578 if (ret
< 0 && ifile
->loop
) {
2579 if ((ret
= seek_to_start(ifile
, is
)) < 0)
2581 ret
= get_input_packet(ifile
, &pkt
);
2584 if (ret
!= AVERROR_EOF
) {
2585 print_error(is
->filename
, ret
);
2589 ifile
->eof_reached
= 1;
2591 for (i
= 0; i
< ifile
->nb_streams
; i
++) {
2592 ist
= input_streams
[ifile
->ist_index
+ i
];
2593 if (ist
->decoding_needed
)
2594 process_input_packet(ist
, NULL
, 0);
2596 /* mark all outputs that don't go through lavfi as finished */
2597 for (j
= 0; j
< nb_output_streams
; j
++) {
2598 OutputStream
*ost
= output_streams
[j
];
2600 if (ost
->source_index
== ifile
->ist_index
+ i
&&
2601 (ost
->stream_copy
|| ost
->enc
->type
== AVMEDIA_TYPE_SUBTITLE
))
2602 finish_output_stream(ost
);
2606 return AVERROR(EAGAIN
);
2612 av_pkt_dump_log2(NULL
, AV_LOG_DEBUG
, &pkt
, do_hex_dump
,
2613 is
->streams
[pkt
.stream_index
]);
2615 /* the following test is needed in case new streams appear
2616 dynamically in stream : we ignore them */
2617 if (pkt
.stream_index
>= ifile
->nb_streams
)
2618 goto discard_packet
;
2620 ist
= input_streams
[ifile
->ist_index
+ pkt
.stream_index
];
2622 ist
->data_size
+= pkt
.size
;
2626 goto discard_packet
;
2628 /* add the stream-global side data to the first packet */
2629 if (ist
->nb_packets
== 1)
2630 for (i
= 0; i
< ist
->st
->nb_side_data
; i
++) {
2631 AVPacketSideData
*src_sd
= &ist
->st
->side_data
[i
];
2634 if (av_packet_get_side_data(&pkt
, src_sd
->type
, NULL
))
2636 if (ist
->autorotate
&& src_sd
->type
== AV_PKT_DATA_DISPLAYMATRIX
)
2639 dst_data
= av_packet_new_side_data(&pkt
, src_sd
->type
, src_sd
->size
);
2643 memcpy(dst_data
, src_sd
->data
, src_sd
->size
);
2646 if (pkt
.dts
!= AV_NOPTS_VALUE
)
2647 pkt
.dts
+= av_rescale_q(ifile
->ts_offset
, AV_TIME_BASE_Q
, ist
->st
->time_base
);
2648 if (pkt
.pts
!= AV_NOPTS_VALUE
)
2649 pkt
.pts
+= av_rescale_q(ifile
->ts_offset
, AV_TIME_BASE_Q
, ist
->st
->time_base
);
2651 if (pkt
.pts
!= AV_NOPTS_VALUE
)
2652 pkt
.pts
*= ist
->ts_scale
;
2653 if (pkt
.dts
!= AV_NOPTS_VALUE
)
2654 pkt
.dts
*= ist
->ts_scale
;
2656 if ((ist
->dec_ctx
->codec_type
== AVMEDIA_TYPE_VIDEO
||
2657 ist
->dec_ctx
->codec_type
== AVMEDIA_TYPE_AUDIO
) &&
2658 pkt
.dts
!= AV_NOPTS_VALUE
&& ist
->next_dts
!= AV_NOPTS_VALUE
&&
2659 (is
->iformat
->flags
& AVFMT_TS_DISCONT
)) {
2660 int64_t pkt_dts
= av_rescale_q(pkt
.dts
, ist
->st
->time_base
, AV_TIME_BASE_Q
);
2661 int64_t delta
= pkt_dts
- ist
->next_dts
;
2663 if ((FFABS(delta
) > 1LL * dts_delta_threshold
* AV_TIME_BASE
|| pkt_dts
+ 1 < ist
->last_dts
) && !copy_ts
) {
2664 ifile
->ts_offset
-= delta
;
2665 av_log(NULL
, AV_LOG_DEBUG
,
2666 "timestamp discontinuity %"PRId64
", new offset= %"PRId64
"\n",
2667 delta
, ifile
->ts_offset
);
2668 pkt
.dts
-= av_rescale_q(delta
, AV_TIME_BASE_Q
, ist
->st
->time_base
);
2669 if (pkt
.pts
!= AV_NOPTS_VALUE
)
2670 pkt
.pts
-= av_rescale_q(delta
, AV_TIME_BASE_Q
, ist
->st
->time_base
);
2673 duration
= av_rescale_q(ifile
->duration
, ifile
->time_base
, ist
->st
->time_base
);
2674 if (pkt
.pts
!= AV_NOPTS_VALUE
) {
2675 pkt
.pts
+= duration
;
2676 ist
->max_pts
= FFMAX(pkt
.pts
, ist
->max_pts
);
2677 ist
->min_pts
= FFMIN(pkt
.pts
, ist
->min_pts
);
2680 if (pkt
.dts
!= AV_NOPTS_VALUE
)
2681 pkt
.dts
+= duration
;
2683 process_input_packet(ist
, &pkt
, 0);
2686 av_packet_unref(&pkt
);
2692 * The following code is the main loop of the file converter
2694 static int transcode(void)
2696 int ret
, i
, need_input
= 1;
2697 AVFormatContext
*os
;
2700 int64_t timer_start
;
2702 ret
= transcode_init();
2706 av_log(NULL
, AV_LOG_INFO
, "Press ctrl-c to stop encoding\n");
2709 timer_start
= av_gettime_relative();
2712 if ((ret
= init_input_threads()) < 0)
2716 while (!received_sigterm
) {
2717 /* check if there's any stream where output is still needed */
2718 if (!need_output()) {
2719 av_log(NULL
, AV_LOG_VERBOSE
, "No more output streams to write to, finishing.\n");
2723 /* read and process one input packet if needed */
2725 ret
= process_input();
2726 if (ret
== AVERROR_EOF
)
2730 ret
= poll_filters();
2731 if (ret
< 0 && ret
!= AVERROR_EOF
) {
2733 av_strerror(ret
, errbuf
, sizeof(errbuf
));
2735 av_log(NULL
, AV_LOG_ERROR
, "Error while filtering: %s\n", errbuf
);
2739 /* dump report by using the output first video and audio streams */
2740 print_report(0, timer_start
);
2743 free_input_threads();
2746 /* at the end of stream, we must flush the decoder buffers */
2747 for (i
= 0; i
< nb_input_streams
; i
++) {
2748 ist
= input_streams
[i
];
2749 if (!input_files
[ist
->file_index
]->eof_reached
&& ist
->decoding_needed
) {
2750 process_input_packet(ist
, NULL
, 0);
2758 /* write the trailer if needed and close file */
2759 for (i
= 0; i
< nb_output_files
; i
++) {
2760 os
= output_files
[i
]->ctx
;
2761 if (!output_files
[i
]->header_written
) {
2762 av_log(NULL
, AV_LOG_ERROR
,
2763 "Nothing was written into output file %d (%s), because "
2764 "at least one of its streams received no packets.\n",
2768 av_write_trailer(os
);
2771 /* dump report by using the first video and audio streams */
2772 print_report(1, timer_start
);
2774 /* close each encoder */
2775 for (i
= 0; i
< nb_output_streams
; i
++) {
2776 ost
= output_streams
[i
];
2777 if (ost
->encoding_needed
) {
2778 av_freep(&ost
->enc_ctx
->stats_in
);
2782 /* close each decoder */
2783 for (i
= 0; i
< nb_input_streams
; i
++) {
2784 ist
= input_streams
[i
];
2785 if (ist
->decoding_needed
) {
2786 avcodec_close(ist
->dec_ctx
);
2787 if (ist
->hwaccel_uninit
)
2788 ist
->hwaccel_uninit(ist
->dec_ctx
);
2792 av_buffer_unref(&hw_device_ctx
);
2799 free_input_threads();
2802 if (output_streams
) {
2803 for (i
= 0; i
< nb_output_streams
; i
++) {
2804 ost
= output_streams
[i
];
2807 fclose(ost
->logfile
);
2808 ost
->logfile
= NULL
;
2810 av_free(ost
->forced_kf_pts
);
2811 av_dict_free(&ost
->encoder_opts
);
2812 av_dict_free(&ost
->resample_opts
);
2819 static int64_t getutime(void)
2822 struct rusage rusage
;
2824 getrusage(RUSAGE_SELF
, &rusage
);
2825 return (rusage
.ru_utime
.tv_sec
* 1000000LL) + rusage
.ru_utime
.tv_usec
;
2826 #elif HAVE_GETPROCESSTIMES
2828 FILETIME c
, e
, k
, u
;
2829 proc
= GetCurrentProcess();
2830 GetProcessTimes(proc
, &c
, &e
, &k
, &u
);
2831 return ((int64_t) u
.dwHighDateTime
<< 32 | u
.dwLowDateTime
) / 10;
2833 return av_gettime_relative();
2837 static int64_t getmaxrss(void)
2839 #if HAVE_GETRUSAGE && HAVE_STRUCT_RUSAGE_RU_MAXRSS
2840 struct rusage rusage
;
2841 getrusage(RUSAGE_SELF
, &rusage
);
2842 return (int64_t)rusage
.ru_maxrss
* 1024;
2843 #elif HAVE_GETPROCESSMEMORYINFO
2845 PROCESS_MEMORY_COUNTERS memcounters
;
2846 proc
= GetCurrentProcess();
2847 memcounters
.cb
= sizeof(memcounters
);
2848 GetProcessMemoryInfo(proc
, &memcounters
, sizeof(memcounters
));
2849 return memcounters
.PeakPagefileUsage
;
2855 int main(int argc
, char **argv
)
2860 register_exit(avconv_cleanup
);
2862 av_log_set_flags(AV_LOG_SKIP_REPEATED
);
2863 parse_loglevel(argc
, argv
, options
);
2865 avcodec_register_all();
2867 avdevice_register_all();
2869 avfilter_register_all();
2871 avformat_network_init();
2875 /* parse options and open all input/output files */
2876 ret
= avconv_parse_options(argc
, argv
);
2880 if (nb_output_files
<= 0 && nb_input_files
== 0) {
2882 av_log(NULL
, AV_LOG_WARNING
, "Use -h to get full help or, even better, run 'man %s'\n", program_name
);
2886 /* file converter / grab */
2887 if (nb_output_files
<= 0) {
2888 fprintf(stderr
, "At least one output file must be specified\n");
2892 for (i
= 0; i
< nb_output_files
; i
++) {
2893 if (strcmp(output_files
[i
]->ctx
->oformat
->name
, "rtp"))
2898 if (transcode() < 0)
2900 ti
= getutime() - ti
;
2902 int maxrss
= getmaxrss() / 1024;
2903 printf("bench: utime=%0.3fs maxrss=%ikB\n", ti
/ 1000000.0, maxrss
);