Commit | Line | Data |
---|---|---|
85f07f22 | 1 | /* |
89b503b5 | 2 | * ffmpeg main |
01310af2 | 3 | * Copyright (c) 2000-2003 Fabrice Bellard |
85f07f22 | 4 | * |
2912e87a | 5 | * This file is part of Libav. |
b78e7197 | 6 | * |
2912e87a | 7 | * Libav is free software; you can redistribute it and/or |
bf5af568 FB |
8 | * modify it under the terms of the GNU Lesser General Public |
9 | * License as published by the Free Software Foundation; either | |
b78e7197 | 10 | * version 2.1 of the License, or (at your option) any later version. |
85f07f22 | 11 | * |
2912e87a | 12 | * Libav is distributed in the hope that it will be useful, |
85f07f22 | 13 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
bf5af568 FB |
14 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
15 | * Lesser General Public License for more details. | |
85f07f22 | 16 | * |
bf5af568 | 17 | * You should have received a copy of the GNU Lesser General Public |
2912e87a | 18 | * License along with Libav; if not, write to the Free Software |
5509bffa | 19 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
85f07f22 | 20 | */ |
364a9607 | 21 | |
0f4e8165 RB |
22 | #include "config.h" |
23 | #include <ctype.h> | |
24 | #include <string.h> | |
25 | #include <math.h> | |
26 | #include <stdlib.h> | |
27 | #include <errno.h> | |
d86b83f8 | 28 | #include <signal.h> |
22f7a060 | 29 | #include <limits.h> |
7246177d | 30 | #include <unistd.h> |
245976da DB |
31 | #include "libavformat/avformat.h" |
32 | #include "libavdevice/avdevice.h" | |
33 | #include "libswscale/swscale.h" | |
41d0eb1c | 34 | #include "libavutil/opt.h" |
ce1ee094 | 35 | #include "libavcodec/audioconvert.h" |
737eb597 RT |
36 | #include "libavutil/audioconvert.h" |
37 | #include "libavutil/parseutils.h" | |
38 | #include "libavutil/samplefmt.h" | |
2b4abbd6 | 39 | #include "libavutil/colorspace.h" |
245976da | 40 | #include "libavutil/fifo.h" |
2839dc97 | 41 | #include "libavutil/intreadwrite.h" |
d2d67e42 | 42 | #include "libavutil/dict.h" |
0ebcdf5c | 43 | #include "libavutil/mathematics.h" |
718c7b18 | 44 | #include "libavutil/pixdesc.h" |
245976da | 45 | #include "libavutil/avstring.h" |
335ee1aa | 46 | #include "libavutil/libm.h" |
245976da | 47 | #include "libavformat/os_support.h" |
daf8e955 | 48 | |
46847a33 MN |
49 | #if CONFIG_AVFILTER |
50 | # include "libavfilter/avfilter.h" | |
51 | # include "libavfilter/avfiltergraph.h" | |
46847a33 MN |
52 | # include "libavfilter/vsrc_buffer.h" |
53 | #endif | |
54 | ||
b250f9c6 | 55 | #if HAVE_SYS_RESOURCE_H |
0a1b29de | 56 | #include <sys/types.h> |
fc5607f8 | 57 | #include <sys/time.h> |
b091aa44 | 58 | #include <sys/resource.h> |
b250f9c6 | 59 | #elif HAVE_GETPROCESSTIMES |
7495c306 RP |
60 | #include <windows.h> |
61 | #endif | |
fc5607f8 RD |
62 | #if HAVE_GETPROCESSMEMORYINFO |
63 | #include <windows.h> | |
64 | #include <psapi.h> | |
65 | #endif | |
7495c306 | 66 | |
b250f9c6 | 67 | #if HAVE_SYS_SELECT_H |
fb1d2d7b BC |
68 | #include <sys/select.h> |
69 | #endif | |
70 | ||
bf5af568 | 71 | #include <time.h> |
85f07f22 | 72 | |
01310af2 FB |
73 | #include "cmdutils.h" |
74 | ||
b64b4134 | 75 | #include "libavutil/avassert.h" |
2b18dcd0 | 76 | |
89b503b5 | 77 | const char program_name[] = "ffmpeg"; |
ea9c581f | 78 | const int program_birth_year = 2000; |
86074ed1 | 79 | |
85f07f22 | 80 | /* select an input stream for an output stream */ |
17c8cc55 | 81 | typedef struct StreamMap { |
85f07f22 FB |
82 | int file_index; |
83 | int stream_index; | |
b4a3389e WG |
84 | int sync_file_index; |
85 | int sync_stream_index; | |
17c8cc55 | 86 | } StreamMap; |
85f07f22 | 87 | |
3f07e8db MN |
88 | /** |
89 | * select an input file for an output file | |
90 | */ | |
17c8cc55 | 91 | typedef struct MetadataMap { |
1829e195 AK |
92 | int file; //< file index |
93 | char type; //< type of metadata to copy -- (g)lobal, (s)tream, (c)hapter or (p)rogram | |
94 | int index; //< stream/chapter/program number | |
17c8cc55 | 95 | } MetadataMap; |
0a38bafd | 96 | |
17c8cc55 | 97 | typedef struct ChapterMap { |
91e96eba AK |
98 | int in_file; |
99 | int out_file; | |
17c8cc55 | 100 | } ChapterMap; |
91e96eba | 101 | |
580a6c57 | 102 | static const OptionDef options[]; |
85f07f22 | 103 | |
60402344 | 104 | #define MAX_FILES 100 |
84fd51e5 | 105 | #define MAX_STREAMS 1024 /* arbitrary sanity check value */ |
85f07f22 | 106 | |
3ee53dab AK |
107 | #define FFM_PACKET_SIZE 4096 //XXX a duplicate of the line in ffm.h |
108 | ||
ef6fc647 | 109 | static const char *last_asked_format = NULL; |
2c6958aa | 110 | static double *input_files_ts_scale[MAX_FILES] = {NULL}; |
2c6958aa | 111 | static int nb_input_files_ts_scale[MAX_FILES] = {0}; |
85f07f22 FB |
112 | |
113 | static AVFormatContext *output_files[MAX_FILES]; | |
8035f429 | 114 | static AVDictionary *output_opts[MAX_FILES]; |
85f07f22 FB |
115 | static int nb_output_files = 0; |
116 | ||
17c8cc55 | 117 | static StreamMap *stream_maps = NULL; |
85f07f22 FB |
118 | static int nb_stream_maps; |
119 | ||
1829e195 | 120 | /* first item specifies output metadata, second is input */ |
17c8cc55 | 121 | static MetadataMap (*meta_data_maps)[2] = NULL; |
0a38bafd | 122 | static int nb_meta_data_maps; |
477b1aea | 123 | static int metadata_global_autocopy = 1; |
d0abe80a AK |
124 | static int metadata_streams_autocopy = 1; |
125 | static int metadata_chapters_autocopy = 1; | |
0a38bafd | 126 | |
17c8cc55 | 127 | static ChapterMap *chapter_maps = NULL; |
91e96eba AK |
128 | static int nb_chapter_maps; |
129 | ||
006e8108 | 130 | /* indexed by output file stream index */ |
e640f261 AJ |
131 | static int *streamid_map = NULL; |
132 | static int nb_streamid_map = 0; | |
006e8108 | 133 | |
55cf1959 MN |
134 | static int frame_width = 0; |
135 | static int frame_height = 0; | |
880e8ba7 | 136 | static float frame_aspect_ratio = 0; |
644a9262 | 137 | static enum PixelFormat frame_pix_fmt = PIX_FMT_NONE; |
5d6e4c16 | 138 | static enum AVSampleFormat audio_sample_fmt = AV_SAMPLE_FMT_NONE; |
cf7fc795 | 139 | static int max_frames[4] = {INT_MAX, INT_MAX, INT_MAX, INT_MAX}; |
89129c6b | 140 | static AVRational frame_rate; |
158c7f05 | 141 | static float video_qscale = 0; |
84f608f4 VM |
142 | static uint16_t *intra_matrix = NULL; |
143 | static uint16_t *inter_matrix = NULL; | |
464a631c | 144 | static const char *video_rc_override_string=NULL; |
85f07f22 | 145 | static int video_disable = 0; |
f3356e9c | 146 | static int video_discard = 0; |
4a897224 | 147 | static char *video_codec_name = NULL; |
83a36b2e | 148 | static unsigned int video_codec_tag = 0; |
0fc2c0f6 | 149 | static char *video_language = NULL; |
85f07f22 | 150 | static int same_quality = 0; |
cfcf0ffd | 151 | static int do_deinterlace = 0; |
bb198e19 | 152 | static int top_field_first = -1; |
f4f3223f | 153 | static int me_threshold = 0; |
1a11cbcc | 154 | static int intra_dc_precision = 8; |
5894e1bb | 155 | static int loop_input = 0; |
8108551a | 156 | static int loop_output = AVFMT_NOOUTPUTLOOP; |
0888fd22 | 157 | static int qp_hist = 0; |
46847a33 MN |
158 | #if CONFIG_AVFILTER |
159 | static char *vfilters = NULL; | |
46847a33 | 160 | #endif |
85f07f22 | 161 | |
85f07f22 | 162 | static int intra_only = 0; |
d7ee4402 | 163 | static int audio_sample_rate = 0; |
13367a46 | 164 | static int64_t channel_layout = 0; |
c57c770d JR |
165 | #define QSCALE_NONE -99999 |
166 | static float audio_qscale = QSCALE_NONE; | |
85f07f22 | 167 | static int audio_disable = 0; |
8f3e9997 | 168 | static int audio_channels = 0; |
4a897224 | 169 | static char *audio_codec_name = NULL; |
83a36b2e | 170 | static unsigned int audio_codec_tag = 0; |
cf7fc795 FB |
171 | static char *audio_language = NULL; |
172 | ||
11bf3847 | 173 | static int subtitle_disable = 0; |
4a897224 | 174 | static char *subtitle_codec_name = NULL; |
cf7fc795 | 175 | static char *subtitle_language = NULL; |
83a36b2e | 176 | static unsigned int subtitle_codec_tag = 0; |
85f07f22 | 177 | |
e3b540b4 LB |
178 | static int data_disable = 0; |
179 | static char *data_codec_name = NULL; | |
180 | static unsigned int data_codec_tag = 0; | |
181 | ||
17c88cb0 MN |
182 | static float mux_preload= 0.5; |
183 | static float mux_max_delay= 0.7; | |
2db3c638 | 184 | |
fc7ad2af | 185 | static int64_t recording_time = INT64_MAX; |
8831db5c | 186 | static int64_t start_time = 0; |
25d34458 | 187 | static int64_t recording_timestamp = 0; |
a6a92a9a | 188 | static int64_t input_ts_offset = 0; |
85f07f22 | 189 | static int file_overwrite = 0; |
d2d67e42 | 190 | static AVDictionary *metadata; |
5727b222 | 191 | static int do_benchmark = 0; |
a0663ba4 | 192 | static int do_hex_dump = 0; |
254abc2e | 193 | static int do_pkt_dump = 0; |
43f1708f | 194 | static int do_psnr = 0; |
5abdb4b1 | 195 | static int do_pass = 0; |
ad16627f | 196 | static char *pass_logfilename_prefix = NULL; |
1629626f FB |
197 | static int audio_stream_copy = 0; |
198 | static int video_stream_copy = 0; | |
cf7fc795 | 199 | static int subtitle_stream_copy = 0; |
e3b540b4 | 200 | static int data_stream_copy = 0; |
8858816d | 201 | static int video_sync_method= -1; |
986ebcdb | 202 | static int audio_sync_method= 0; |
d4d226a8 | 203 | static float audio_drift_threshold= 0.1; |
72bd8100 | 204 | static int copy_ts= 0; |
0f27e6b4 | 205 | static int copy_tb; |
76bdac6d | 206 | static int opt_shortest = 0; |
b60d1379 | 207 | static char *vstats_filename; |
032aa7df | 208 | static FILE *vstats_file; |
50e143c4 | 209 | static int opt_programid = 0; |
50e3477f | 210 | static int copy_initial_nonkeyframes = 0; |
5abdb4b1 | 211 | |
bdfcbbed MK |
212 | static int rate_emu = 0; |
213 | ||
a9aa3467 | 214 | static int audio_volume = 256; |
8aa3ee32 | 215 | |
f2abc559 | 216 | static int exit_on_error = 0; |
d9a916e2 | 217 | static int using_stdin = 0; |
f068206e | 218 | static int verbose = 1; |
9c3d33d6 | 219 | static int thread_count= 1; |
1008ceb3 MN |
220 | static int64_t video_size = 0; |
221 | static int64_t audio_size = 0; | |
222 | static int64_t extra_size = 0; | |
a6a92a9a WG |
223 | static int nb_frames_dup = 0; |
224 | static int nb_frames_drop = 0; | |
6e454c38 | 225 | static int input_sync; |
76bdac6d | 226 | static uint64_t limit_filesize = 0; |
d2845b75 | 227 | static int force_fps = 0; |
4ad08021 | 228 | static char *forced_key_frames = NULL; |
d9a916e2 | 229 | |
a8482aab | 230 | static float dts_delta_threshold = 10; |
5b6d5596 | 231 | |
29cc1c23 | 232 | static int64_t timer_start; |
8bbf6db9 | 233 | |
3321cb3f BC |
234 | static uint8_t *audio_buf; |
235 | static uint8_t *audio_out; | |
a6d1bd05 | 236 | static unsigned int allocated_audio_out_size, allocated_audio_buf_size; |
3321cb3f BC |
237 | |
238 | static short *samples; | |
239 | ||
748c2fca MN |
240 | static AVBitStreamFilterContext *video_bitstream_filters=NULL; |
241 | static AVBitStreamFilterContext *audio_bitstream_filters=NULL; | |
e1cc8339 | 242 | static AVBitStreamFilterContext *subtitle_bitstream_filters=NULL; |
5b6d5596 | 243 | |
ad16627f | 244 | #define DEFAULT_PASS_LOGFILENAME_PREFIX "ffmpeg2pass" |
85f07f22 | 245 | |
17c8cc55 | 246 | struct InputStream; |
b4a3389e | 247 | |
17c8cc55 | 248 | typedef struct OutputStream { |
85f07f22 FB |
249 | int file_index; /* file index */ |
250 | int index; /* stream index in the output file */ | |
17c8cc55 | 251 | int source_index; /* InputStream index */ |
85f07f22 | 252 | AVStream *st; /* stream in the output file */ |
ec5517d5 FB |
253 | int encoding_needed; /* true if encoding needed for this stream */ |
254 | int frame_number; | |
255 | /* input pts and corresponding output pts | |
256 | for A/V sync */ | |
b4a3389e | 257 | //double sync_ipts; /* dts from the AVPacket of the demuxer in second units */ |
17c8cc55 | 258 | struct InputStream *sync_ist; /* input stream to sync against */ |
e928649b | 259 | int64_t sync_opts; /* output frame counter, could be changed to some true timestamp */ //FIXME look at frame_number |
0b6d358a | 260 | AVBitStreamFilterContext *bitstream_filters; |
9446d759 AK |
261 | AVCodec *enc; |
262 | ||
85f07f22 | 263 | /* video only */ |
07d0cdfc | 264 | int video_resample; |
a4d36c11 | 265 | AVFrame pict_tmp; /* temporary image for resampling */ |
18a54b04 LA |
266 | struct SwsContext *img_resample_ctx; /* for image resampling */ |
267 | int resample_height; | |
352666c1 | 268 | int resample_width; |
01a3c821 | 269 | int resample_pix_fmt; |
a6286bda | 270 | AVRational frame_rate; |
34b10a57 | 271 | |
901ff511 BC |
272 | float frame_aspect_ratio; |
273 | ||
4ad08021 NG |
274 | /* forced key frames */ |
275 | int64_t *forced_kf_pts; | |
276 | int forced_kf_count; | |
277 | int forced_kf_index; | |
278 | ||
85f07f22 FB |
279 | /* audio only */ |
280 | int audio_resample; | |
281 | ReSampleContext *resample; /* for audio resampling */ | |
8afab686 SS |
282 | int resample_sample_fmt; |
283 | int resample_channels; | |
284 | int resample_sample_rate; | |
a79db0f7 PR |
285 | int reformat_pair; |
286 | AVAudioConvert *reformat_ctx; | |
41dd680d | 287 | AVFifoBuffer *fifo; /* for compression: one audio fifo per codec */ |
5abdb4b1 | 288 | FILE *logfile; |
1435f2fa | 289 | |
9d5fa618 MN |
290 | #if CONFIG_AVFILTER |
291 | AVFilterContext *output_video_filter; | |
292 | AVFilterContext *input_video_filter; | |
293 | AVFilterBufferRef *picref; | |
294 | char *avfilter; | |
295 | AVFilterGraph *graph; | |
296 | #endif | |
297 | ||
1435f2fa | 298 | int sws_flags; |
17c8cc55 | 299 | } OutputStream; |
85f07f22 | 300 | |
17c8cc55 | 301 | static OutputStream **output_streams_for_file[MAX_FILES] = { NULL }; |
9fdf4b58 NG |
302 | static int nb_output_streams_for_file[MAX_FILES] = { 0 }; |
303 | ||
17c8cc55 | 304 | typedef struct InputStream { |
85f07f22 | 305 | int file_index; |
85f07f22 FB |
306 | AVStream *st; |
307 | int discard; /* true if stream data should be discarded */ | |
308 | int decoding_needed; /* true if the packets must be decoded in 'raw_fifo' */ | |
9e253c13 | 309 | AVCodec *dec; |
bdfcbbed MK |
310 | |
311 | int64_t start; /* time when read started */ | |
254abc2e FB |
312 | int64_t next_pts; /* synthetic pts for cases where pkt.pts |
313 | is not defined */ | |
e7d0374f | 314 | int64_t pts; /* current pts */ |
0ff4f0c0 | 315 | PtsCorrectionContext pts_ctx; |
ff4905a5 | 316 | int is_start; /* is 1 at the start and after a discontinuity */ |
3ff0daf0 | 317 | int showed_multi_packet_warning; |
55a7e946 | 318 | int is_past_recording_time; |
17c8cc55 | 319 | } InputStream; |
85f07f22 | 320 | |
17c8cc55 | 321 | typedef struct InputFile { |
07633154 | 322 | AVFormatContext *ctx; |
85f07f22 FB |
323 | int eof_reached; /* true if eof reached */ |
324 | int ist_index; /* index of first stream in ist_table */ | |
325 | int buffer_size; /* current total buffer size */ | |
27e91f37 | 326 | int64_t ts_offset; |
17c8cc55 | 327 | } InputFile; |
85f07f22 | 328 | |
17c8cc55 | 329 | static InputStream *input_streams = NULL; |
07633154 | 330 | static int nb_input_streams = 0; |
17c8cc55 | 331 | static InputFile *input_files = NULL; |
07633154 AK |
332 | static int nb_input_files = 0; |
333 | ||
46847a33 | 334 | #if CONFIG_AVFILTER |
46847a33 | 335 | |
17c8cc55 | 336 | static int configure_video_filters(InputStream *ist, OutputStream *ost) |
46847a33 | 337 | { |
a9f3cb93 | 338 | AVFilterContext *last_filter, *filter; |
46847a33 MN |
339 | /** filter graph containing all filters including input & output */ |
340 | AVCodecContext *codec = ost->st->codec; | |
341 | AVCodecContext *icodec = ist->st->codec; | |
f7ead94c | 342 | FFSinkContext ffsink_ctx = { .pix_fmt = codec->pix_fmt }; |
7a11c82f | 343 | AVRational sample_aspect_ratio; |
46847a33 | 344 | char args[255]; |
4ddf0d29 | 345 | int ret; |
46847a33 | 346 | |
9d5fa618 | 347 | ost->graph = avfilter_graph_alloc(); |
46847a33 | 348 | |
7a11c82f MN |
349 | if (ist->st->sample_aspect_ratio.num){ |
350 | sample_aspect_ratio = ist->st->sample_aspect_ratio; | |
351 | }else | |
352 | sample_aspect_ratio = ist->st->codec->sample_aspect_ratio; | |
353 | ||
354 | snprintf(args, 255, "%d:%d:%d:%d:%d:%d:%d", ist->st->codec->width, | |
355 | ist->st->codec->height, ist->st->codec->pix_fmt, 1, AV_TIME_BASE, | |
356 | sample_aspect_ratio.num, sample_aspect_ratio.den); | |
357 | ||
9d5fa618 MN |
358 | ret = avfilter_graph_create_filter(&ost->input_video_filter, avfilter_get_by_name("buffer"), |
359 | "src", args, NULL, ost->graph); | |
037be76e | 360 | if (ret < 0) |
4ddf0d29 | 361 | return ret; |
9d5fa618 MN |
362 | ret = avfilter_graph_create_filter(&ost->output_video_filter, &ffsink, |
363 | "out", NULL, &ffsink_ctx, ost->graph); | |
037be76e | 364 | if (ret < 0) |
4ddf0d29 | 365 | return ret; |
9d5fa618 | 366 | last_filter = ost->input_video_filter; |
46847a33 | 367 | |
5879ea6d | 368 | if (codec->width != icodec->width || codec->height != icodec->height) { |
6e82e7fa | 369 | snprintf(args, 255, "%d:%d:flags=0x%X", |
0c22311b MN |
370 | codec->width, |
371 | codec->height, | |
1435f2fa | 372 | ost->sws_flags); |
037be76e | 373 | if ((ret = avfilter_graph_create_filter(&filter, avfilter_get_by_name("scale"), |
9d5fa618 | 374 | NULL, args, NULL, ost->graph)) < 0) |
4ddf0d29 SS |
375 | return ret; |
376 | if ((ret = avfilter_link(last_filter, 0, filter, 0)) < 0) | |
377 | return ret; | |
a9f3cb93 | 378 | last_filter = filter; |
46847a33 MN |
379 | } |
380 | ||
1435f2fa | 381 | snprintf(args, sizeof(args), "flags=0x%X", ost->sws_flags); |
9d5fa618 | 382 | ost->graph->scale_sws_opts = av_strdup(args); |
f96363df | 383 | |
9d5fa618 | 384 | if (ost->avfilter) { |
46847a33 MN |
385 | AVFilterInOut *outputs = av_malloc(sizeof(AVFilterInOut)); |
386 | AVFilterInOut *inputs = av_malloc(sizeof(AVFilterInOut)); | |
387 | ||
388 | outputs->name = av_strdup("in"); | |
7313132b | 389 | outputs->filter_ctx = last_filter; |
46847a33 MN |
390 | outputs->pad_idx = 0; |
391 | outputs->next = NULL; | |
392 | ||
393 | inputs->name = av_strdup("out"); | |
9d5fa618 | 394 | inputs->filter_ctx = ost->output_video_filter; |
46847a33 MN |
395 | inputs->pad_idx = 0; |
396 | inputs->next = NULL; | |
397 | ||
9d5fa618 | 398 | if ((ret = avfilter_graph_parse(ost->graph, ost->avfilter, inputs, outputs, NULL)) < 0) |
4ddf0d29 | 399 | return ret; |
9d5fa618 | 400 | av_freep(&ost->avfilter); |
46847a33 | 401 | } else { |
9d5fa618 | 402 | if ((ret = avfilter_link(last_filter, 0, ost->output_video_filter, 0)) < 0) |
4ddf0d29 | 403 | return ret; |
46847a33 MN |
404 | } |
405 | ||
9d5fa618 | 406 | if ((ret = avfilter_graph_config(ost->graph, NULL)) < 0) |
4ddf0d29 | 407 | return ret; |
46847a33 | 408 | |
9d5fa618 MN |
409 | codec->width = ost->output_video_filter->inputs[0]->w; |
410 | codec->height = ost->output_video_filter->inputs[0]->h; | |
7a11c82f | 411 | codec->sample_aspect_ratio = ost->st->sample_aspect_ratio = |
901ff511 BC |
412 | ost->frame_aspect_ratio ? // overriden by the -aspect cli option |
413 | av_d2q(ost->frame_aspect_ratio*codec->height/codec->width, 255) : | |
9d5fa618 | 414 | ost->output_video_filter->inputs[0]->sample_aspect_ratio; |
46847a33 MN |
415 | |
416 | return 0; | |
417 | } | |
418 | #endif /* CONFIG_AVFILTER */ | |
419 | ||
85f07f22 FB |
420 | static void term_exit(void) |
421 | { | |
6b6bca64 | 422 | av_log(NULL, AV_LOG_QUIET, ""); |
64f6e357 | 423 | } |
85f07f22 | 424 | |
e9a832e5 | 425 | static volatile int received_sigterm = 0; |
a1217548 | 426 | static volatile int received_nb_signals = 0; |
9680a722 RP |
427 | |
428 | static void | |
429 | sigterm_handler(int sig) | |
430 | { | |
431 | received_sigterm = sig; | |
a1217548 | 432 | received_nb_signals++; |
9680a722 RP |
433 | term_exit(); |
434 | } | |
435 | ||
85f07f22 FB |
436 | static void term_init(void) |
437 | { | |
9680a722 | 438 | signal(SIGINT , sigterm_handler); /* Interrupt (ANSI). */ |
9680a722 | 439 | signal(SIGTERM, sigterm_handler); /* Termination (ANSI). */ |
ffcc6e24 MR |
440 | #ifdef SIGXCPU |
441 | signal(SIGXCPU, sigterm_handler); | |
442 | #endif | |
85f07f22 FB |
443 | } |
444 | ||
b51469a0 LS |
445 | static int decode_interrupt_cb(void) |
446 | { | |
a1217548 | 447 | return received_nb_signals > 1; |
b51469a0 LS |
448 | } |
449 | ||
639e4ec8 | 450 | static int ffmpeg_exit(int ret) |
e5295c0d RP |
451 | { |
452 | int i; | |
453 | ||
454 | /* close files */ | |
455 | for(i=0;i<nb_output_files;i++) { | |
e5295c0d | 456 | AVFormatContext *s = output_files[i]; |
8767060c | 457 | if (!(s->oformat->flags & AVFMT_NOFILE) && s->pb) |
22a3212e | 458 | avio_close(s->pb); |
b22dbb29 | 459 | avformat_free_context(s); |
9fdf4b58 | 460 | av_free(output_streams_for_file[i]); |
8035f429 | 461 | av_dict_free(&output_opts[i]); |
e5295c0d | 462 | } |
2c6958aa | 463 | for(i=0;i<nb_input_files;i++) { |
07633154 | 464 | av_close_input_file(input_files[i].ctx); |
2c6958aa AJ |
465 | av_free(input_files_ts_scale[i]); |
466 | } | |
e5295c0d | 467 | |
e5295c0d RP |
468 | av_free(intra_matrix); |
469 | av_free(inter_matrix); | |
470 | ||
471 | if (vstats_file) | |
472 | fclose(vstats_file); | |
473 | av_free(vstats_filename); | |
474 | ||
e640f261 | 475 | av_free(streamid_map); |
3a8e8824 | 476 | av_free(stream_maps); |
63e856df | 477 | av_free(meta_data_maps); |
e5295c0d | 478 | |
07633154 AK |
479 | av_freep(&input_streams); |
480 | av_freep(&input_files); | |
481 | ||
e5295c0d RP |
482 | av_free(video_codec_name); |
483 | av_free(audio_codec_name); | |
484 | av_free(subtitle_codec_name); | |
e3b540b4 | 485 | av_free(data_codec_name); |
e5295c0d | 486 | |
a5c33faa | 487 | uninit_opts(); |
3321cb3f BC |
488 | av_free(audio_buf); |
489 | av_free(audio_out); | |
b8919a30 | 490 | allocated_audio_buf_size= allocated_audio_out_size= 0; |
3321cb3f | 491 | av_free(samples); |
5973490a | 492 | |
46847a33 MN |
493 | #if CONFIG_AVFILTER |
494 | avfilter_uninit(); | |
495 | #endif | |
496 | ||
e5295c0d RP |
497 | if (received_sigterm) { |
498 | fprintf(stderr, | |
499 | "Received signal %d: terminating.\n", | |
500 | (int) received_sigterm); | |
501 | exit (255); | |
502 | } | |
503 | ||
296df4e7 RP |
504 | exit(ret); /* not all OS-es handle main() return value */ |
505 | return ret; | |
dba249ab AJ |
506 | } |
507 | ||
8035f429 AK |
508 | static void assert_avoptions(AVDictionary *m) |
509 | { | |
510 | AVDictionaryEntry *t; | |
511 | if ((t = av_dict_get(m, "", NULL, AV_DICT_IGNORE_SUFFIX))) { | |
512 | av_log(NULL, AV_LOG_ERROR, "Option %s not found.\n", t->key); | |
513 | ffmpeg_exit(1); | |
514 | } | |
515 | } | |
516 | ||
dba249ab AJ |
517 | /* similar to ff_dynarray_add() and av_fast_realloc() */ |
518 | static void *grow_array(void *array, int elem_size, int *size, int new_size) | |
519 | { | |
520 | if (new_size >= INT_MAX / elem_size) { | |
521 | fprintf(stderr, "Array too big.\n"); | |
522 | ffmpeg_exit(1); | |
523 | } | |
524 | if (*size < new_size) { | |
525 | uint8_t *tmp = av_realloc(array, new_size*elem_size); | |
526 | if (!tmp) { | |
527 | fprintf(stderr, "Could not alloc buffer.\n"); | |
528 | ffmpeg_exit(1); | |
529 | } | |
530 | memset(tmp + *size*elem_size, 0, (new_size-*size) * elem_size); | |
531 | *size = new_size; | |
532 | return tmp; | |
533 | } | |
534 | return array; | |
e5295c0d RP |
535 | } |
536 | ||
aa1de0d9 RB |
537 | static void choose_sample_fmt(AVStream *st, AVCodec *codec) |
538 | { | |
539 | if(codec && codec->sample_fmts){ | |
5d6e4c16 | 540 | const enum AVSampleFormat *p= codec->sample_fmts; |
aa1de0d9 RB |
541 | for(; *p!=-1; p++){ |
542 | if(*p == st->codec->sample_fmt) | |
543 | break; | |
544 | } | |
fa34a362 SS |
545 | if (*p == -1) { |
546 | av_log(NULL, AV_LOG_WARNING, | |
547 | "Incompatible sample format '%s' for codec '%s', auto-selecting format '%s'\n", | |
548 | av_get_sample_fmt_name(st->codec->sample_fmt), | |
549 | codec->name, | |
550 | av_get_sample_fmt_name(codec->sample_fmts[0])); | |
aa1de0d9 | 551 | st->codec->sample_fmt = codec->sample_fmts[0]; |
fa34a362 | 552 | } |
aa1de0d9 RB |
553 | } |
554 | } | |
555 | ||
bc778a0c JR |
556 | /** |
557 | * Update the requested input sample format based on the output sample format. | |
558 | * This is currently only used to request float output from decoders which | |
559 | * support multiple sample formats, one of which is AV_SAMPLE_FMT_FLT. | |
560 | * Ideally this will be removed in the future when decoders do not do format | |
561 | * conversion and only output in their native format. | |
562 | */ | |
563 | static void update_sample_fmt(AVCodecContext *dec, AVCodec *dec_codec, | |
564 | AVCodecContext *enc) | |
565 | { | |
566 | /* if sample formats match or a decoder sample format has already been | |
567 | requested, just return */ | |
568 | if (enc->sample_fmt == dec->sample_fmt || | |
569 | dec->request_sample_fmt > AV_SAMPLE_FMT_NONE) | |
570 | return; | |
571 | ||
572 | /* if decoder supports more than one output format */ | |
573 | if (dec_codec && dec_codec->sample_fmts && | |
574 | dec_codec->sample_fmts[0] != AV_SAMPLE_FMT_NONE && | |
575 | dec_codec->sample_fmts[1] != AV_SAMPLE_FMT_NONE) { | |
9a1b7912 | 576 | const enum AVSampleFormat *p; |
bc778a0c JR |
577 | int min_dec = -1, min_inc = -1; |
578 | ||
579 | /* find a matching sample format in the encoder */ | |
580 | for (p = dec_codec->sample_fmts; *p != AV_SAMPLE_FMT_NONE; p++) { | |
581 | if (*p == enc->sample_fmt) { | |
582 | dec->request_sample_fmt = *p; | |
583 | return; | |
584 | } else if (*p > enc->sample_fmt) { | |
585 | min_inc = FFMIN(min_inc, *p - enc->sample_fmt); | |
586 | } else | |
587 | min_dec = FFMIN(min_dec, enc->sample_fmt - *p); | |
588 | } | |
589 | ||
590 | /* if none match, provide the one that matches quality closest */ | |
591 | dec->request_sample_fmt = min_inc > 0 ? enc->sample_fmt + min_inc : | |
592 | enc->sample_fmt - min_dec; | |
593 | } | |
594 | } | |
595 | ||
10d0f5e0 MN |
596 | static void choose_sample_rate(AVStream *st, AVCodec *codec) |
597 | { | |
598 | if(codec && codec->supported_samplerates){ | |
599 | const int *p= codec->supported_samplerates; | |
3c5e1b36 | 600 | int best=0; |
10d0f5e0 MN |
601 | int best_dist=INT_MAX; |
602 | for(; *p; p++){ | |
603 | int dist= abs(st->codec->sample_rate - *p); | |
604 | if(dist < best_dist){ | |
605 | best_dist= dist; | |
606 | best= *p; | |
607 | } | |
608 | } | |
ff866063 MN |
609 | if(best_dist){ |
610 | av_log(st->codec, AV_LOG_WARNING, "Requested sampling rate unsupported using closest supported (%d)\n", best); | |
611 | } | |
10d0f5e0 MN |
612 | st->codec->sample_rate= best; |
613 | } | |
614 | } | |
615 | ||
aa1de0d9 RB |
616 | static void choose_pixel_fmt(AVStream *st, AVCodec *codec) |
617 | { | |
618 | if(codec && codec->pix_fmts){ | |
619 | const enum PixelFormat *p= codec->pix_fmts; | |
b26847b7 MN |
620 | if(st->codec->strict_std_compliance <= FF_COMPLIANCE_UNOFFICIAL){ |
621 | if(st->codec->codec_id==CODEC_ID_MJPEG){ | |
622 | p= (const enum PixelFormat[]){PIX_FMT_YUVJ420P, PIX_FMT_YUVJ422P, PIX_FMT_YUV420P, PIX_FMT_YUV422P, PIX_FMT_NONE}; | |
623 | }else if(st->codec->codec_id==CODEC_ID_LJPEG){ | |
624 | p= (const enum PixelFormat[]){PIX_FMT_YUVJ420P, PIX_FMT_YUVJ422P, PIX_FMT_YUVJ444P, PIX_FMT_YUV420P, PIX_FMT_YUV422P, PIX_FMT_YUV444P, PIX_FMT_BGRA, PIX_FMT_NONE}; | |
625 | } | |
626 | } | |
aa1de0d9 RB |
627 | for(; *p!=-1; p++){ |
628 | if(*p == st->codec->pix_fmt) | |
629 | break; | |
630 | } | |
b568d6d9 SS |
631 | if (*p == -1) { |
632 | if(st->codec->pix_fmt != PIX_FMT_NONE) | |
633 | av_log(NULL, AV_LOG_WARNING, | |
634 | "Incompatible pixel format '%s' for codec '%s', auto-selecting format '%s'\n", | |
635 | av_pix_fmt_descriptors[st->codec->pix_fmt].name, | |
636 | codec->name, | |
637 | av_pix_fmt_descriptors[codec->pix_fmts[0]].name); | |
aa1de0d9 | 638 | st->codec->pix_fmt = codec->pix_fmts[0]; |
b568d6d9 | 639 | } |
aa1de0d9 RB |
640 | } |
641 | } | |
642 | ||
17c8cc55 | 643 | static OutputStream *new_output_stream(AVFormatContext *oc, int file_idx) |
ec1ca41c CEH |
644 | { |
645 | int idx = oc->nb_streams - 1; | |
17c8cc55 | 646 | OutputStream *ost; |
ec1ca41c CEH |
647 | |
648 | output_streams_for_file[file_idx] = | |
649 | grow_array(output_streams_for_file[file_idx], | |
650 | sizeof(*output_streams_for_file[file_idx]), | |
651 | &nb_output_streams_for_file[file_idx], | |
652 | oc->nb_streams); | |
653 | ost = output_streams_for_file[file_idx][idx] = | |
17c8cc55 | 654 | av_mallocz(sizeof(OutputStream)); |
ec1ca41c CEH |
655 | if (!ost) { |
656 | fprintf(stderr, "Could not alloc output stream\n"); | |
657 | ffmpeg_exit(1); | |
658 | } | |
659 | ost->file_index = file_idx; | |
660 | ost->index = idx; | |
1435f2fa AK |
661 | |
662 | ost->sws_flags = av_get_int(sws_opts, "sws_flags", NULL); | |
ec1ca41c CEH |
663 | return ost; |
664 | } | |
665 | ||
b29f97d1 | 666 | static int read_ffserver_streams(AVFormatContext *s, const char *filename) |
85f07f22 | 667 | { |
79fdaa4c | 668 | int i, err; |
8035f429 | 669 | AVFormatContext *ic = NULL; |
3438d82d | 670 | int nopts = 0; |
85f07f22 | 671 | |
8035f429 | 672 | err = avformat_open_input(&ic, filename, NULL, NULL); |
79fdaa4c FB |
673 | if (err < 0) |
674 | return err; | |
85f07f22 | 675 | /* copy stream format */ |
b67f3d65 | 676 | s->nb_streams = 0; |
db3262b7 | 677 | s->streams = av_mallocz(sizeof(AVStream *) * ic->nb_streams); |
85f07f22 FB |
678 | for(i=0;i<ic->nb_streams;i++) { |
679 | AVStream *st; | |
d9521cb1 | 680 | AVCodec *codec; |
1e491e29 | 681 | |
b67f3d65 RC |
682 | s->nb_streams++; |
683 | ||
f37f8d4c | 684 | // FIXME: a more elegant solution is needed |
e1031171 | 685 | st = av_mallocz(sizeof(AVStream)); |
85f07f22 | 686 | memcpy(st, ic->streams[i], sizeof(AVStream)); |
0be37367 | 687 | st->info = NULL; |
f37f8d4c | 688 | st->codec = avcodec_alloc_context(); |
7ef61879 RP |
689 | if (!st->codec) { |
690 | print_error(filename, AVERROR(ENOMEM)); | |
639e4ec8 | 691 | ffmpeg_exit(1); |
7ef61879 | 692 | } |
d9521cb1 | 693 | avcodec_copy_context(st->codec, ic->streams[i]->codec); |
85f07f22 | 694 | s->streams[i] = st; |
837d248d | 695 | |
d9521cb1 RB |
696 | codec = avcodec_find_encoder(st->codec->codec_id); |
697 | if (st->codec->codec_type == AVMEDIA_TYPE_AUDIO) { | |
698 | if (audio_stream_copy) { | |
699 | st->stream_copy = 1; | |
700 | } else | |
701 | choose_sample_fmt(st, codec); | |
702 | } else if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO) { | |
703 | if (video_stream_copy) { | |
704 | st->stream_copy = 1; | |
705 | } else | |
706 | choose_pixel_fmt(st, codec); | |
707 | } | |
837d248d | 708 | |
3438d82d BC |
709 | if(st->codec->flags & CODEC_FLAG_BITEXACT) |
710 | nopts = 1; | |
b67f3d65 RC |
711 | |
712 | new_output_stream(s, nb_output_files); | |
85f07f22 FB |
713 | } |
714 | ||
3438d82d BC |
715 | if (!nopts) |
716 | s->timestamp = av_gettime(); | |
717 | ||
85f07f22 FB |
718 | av_close_input_file(ic); |
719 | return 0; | |
720 | } | |
721 | ||
b4a3389e | 722 | static double |
17c8cc55 | 723 | get_sync_ipts(const OutputStream *ost) |
b4a3389e | 724 | { |
17c8cc55 | 725 | const InputStream *ist = ost->sync_ist; |
fec401f7 | 726 | return (double)(ist->pts - start_time)/AV_TIME_BASE; |
b4a3389e WG |
727 | } |
728 | ||
748c2fca | 729 | static void write_frame(AVFormatContext *s, AVPacket *pkt, AVCodecContext *avctx, AVBitStreamFilterContext *bsfc){ |
0ac07031 MN |
730 | int ret; |
731 | ||
748c2fca MN |
732 | while(bsfc){ |
733 | AVPacket new_pkt= *pkt; | |
734 | int a= av_bitstream_filter_filter(bsfc, avctx, NULL, | |
735 | &new_pkt.data, &new_pkt.size, | |
736 | pkt->data, pkt->size, | |
cc947f04 | 737 | pkt->flags & AV_PKT_FLAG_KEY); |
7055cdac | 738 | if(a>0){ |
748c2fca MN |
739 | av_free_packet(pkt); |
740 | new_pkt.destruct= av_destruct_packet; | |
7055cdac | 741 | } else if(a<0){ |
1f8e32cd DB |
742 | fprintf(stderr, "%s failed for stream %d, codec %s", |
743 | bsfc->filter->name, pkt->stream_index, | |
744 | avctx->codec ? avctx->codec->name : "copy"); | |
7055cdac | 745 | print_error("", a); |
f2abc559 | 746 | if (exit_on_error) |
639e4ec8 | 747 | ffmpeg_exit(1); |
748c2fca MN |
748 | } |
749 | *pkt= new_pkt; | |
750 | ||
751 | bsfc= bsfc->next; | |
752 | } | |
753 | ||
0ac07031 MN |
754 | ret= av_interleaved_write_frame(s, pkt); |
755 | if(ret < 0){ | |
756 | print_error("av_interleaved_write_frame()", ret); | |
639e4ec8 | 757 | ffmpeg_exit(1); |
0ac07031 | 758 | } |
748c2fca MN |
759 | } |
760 | ||
817b23ff | 761 | #define MAX_AUDIO_PACKET_SIZE (128 * 1024) |
85f07f22 | 762 | |
115329f1 | 763 | static void do_audio_out(AVFormatContext *s, |
17c8cc55 AK |
764 | OutputStream *ost, |
765 | InputStream *ist, | |
85f07f22 FB |
766 | unsigned char *buf, int size) |
767 | { | |
0c1a9eda | 768 | uint8_t *buftmp; |
15bfe412 | 769 | int64_t audio_out_size, audio_buf_size; |
7a086a85 | 770 | int64_t allocated_for_size= size; |
d66c7abc | 771 | |
8afab686 | 772 | int size_out, frame_bytes, ret, resample_changed; |
01f4895c | 773 | AVCodecContext *enc= ost->st->codec; |
2886f311 | 774 | AVCodecContext *dec= ist->st->codec; |
e6c52cee JR |
775 | int osize = av_get_bytes_per_sample(enc->sample_fmt); |
776 | int isize = av_get_bytes_per_sample(dec->sample_fmt); | |
15bfe412 MN |
777 | const int coded_bps = av_get_bits_per_sample(enc->codec->id); |
778 | ||
7a086a85 MN |
779 | need_realloc: |
780 | audio_buf_size= (allocated_for_size + isize*dec->channels - 1) / (isize*dec->channels); | |
15bfe412 | 781 | audio_buf_size= (audio_buf_size*enc->sample_rate + dec->sample_rate) / dec->sample_rate; |
8b484d0f | 782 | audio_buf_size= audio_buf_size*2 + 10000; //safety factors for the deprecated resampling API |
79c85beb | 783 | audio_buf_size= FFMAX(audio_buf_size, enc->frame_size); |
15bfe412 MN |
784 | audio_buf_size*= osize*enc->channels; |
785 | ||
786 | audio_out_size= FFMAX(audio_buf_size, enc->frame_size * osize * enc->channels); | |
787 | if(coded_bps > 8*osize) | |
788 | audio_out_size= audio_out_size * coded_bps / (8*osize); | |
789 | audio_out_size += FF_MIN_BUFFER_SIZE; | |
790 | ||
791 | if(audio_out_size > INT_MAX || audio_buf_size > INT_MAX){ | |
792 | fprintf(stderr, "Buffer sizes too large\n"); | |
639e4ec8 | 793 | ffmpeg_exit(1); |
15bfe412 | 794 | } |
85f07f22 | 795 | |
b8919a30 MN |
796 | av_fast_malloc(&audio_buf, &allocated_audio_buf_size, audio_buf_size); |
797 | av_fast_malloc(&audio_out, &allocated_audio_out_size, audio_out_size); | |
e185a2f6 MN |
798 | if (!audio_buf || !audio_out){ |
799 | fprintf(stderr, "Out of memory in do_audio_out\n"); | |
639e4ec8 | 800 | ffmpeg_exit(1); |
e185a2f6 | 801 | } |
d66c7abc | 802 | |
bc778a0c | 803 | if (enc->channels != dec->channels || enc->sample_rate != dec->sample_rate) |
2886f311 AÖ |
804 | ost->audio_resample = 1; |
805 | ||
8afab686 SS |
806 | resample_changed = ost->resample_sample_fmt != dec->sample_fmt || |
807 | ost->resample_channels != dec->channels || | |
808 | ost->resample_sample_rate != dec->sample_rate; | |
809 | ||
810 | if ((ost->audio_resample && !ost->resample) || resample_changed) { | |
811 | if (resample_changed) { | |
812 | 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", | |
d2bc4da1 | 813 | ist->file_index, ist->st->index, |
8afab686 SS |
814 | ost->resample_sample_rate, av_get_sample_fmt_name(ost->resample_sample_fmt), ost->resample_channels, |
815 | dec->sample_rate, av_get_sample_fmt_name(dec->sample_fmt), dec->channels); | |
816 | ost->resample_sample_fmt = dec->sample_fmt; | |
817 | ost->resample_channels = dec->channels; | |
818 | ost->resample_sample_rate = dec->sample_rate; | |
819 | if (ost->resample) | |
820 | audio_resample_close(ost->resample); | |
821 | } | |
07b48f8c SS |
822 | /* if audio_sync_method is >1 the resampler is needed for audio drift compensation */ |
823 | if (audio_sync_method <= 1 && | |
824 | ost->resample_sample_fmt == enc->sample_fmt && | |
8afab686 SS |
825 | ost->resample_channels == enc->channels && |
826 | ost->resample_sample_rate == enc->sample_rate) { | |
827 | ost->resample = NULL; | |
828 | ost->audio_resample = 0; | |
bc778a0c | 829 | } else if (ost->audio_resample) { |
f6715848 SS |
830 | if (dec->sample_fmt != AV_SAMPLE_FMT_S16) |
831 | fprintf(stderr, "Warning, using s16 intermediate sample format for resampling\n"); | |
832 | ost->resample = av_audio_resample_init(enc->channels, dec->channels, | |
833 | enc->sample_rate, dec->sample_rate, | |
834 | enc->sample_fmt, dec->sample_fmt, | |
835 | 16, 10, 0, 0.8); | |
836 | if (!ost->resample) { | |
837 | fprintf(stderr, "Can not resample %d channels @ %d Hz to %d channels @ %d Hz\n", | |
838 | dec->channels, dec->sample_rate, | |
839 | enc->channels, enc->sample_rate); | |
840 | ffmpeg_exit(1); | |
841 | } | |
8afab686 | 842 | } |
2886f311 AÖ |
843 | } |
844 | ||
5d6e4c16 | 845 | #define MAKE_SFMT_PAIR(a,b) ((a)+AV_SAMPLE_FMT_NB*(b)) |
d1e3c6fd | 846 | if (!ost->audio_resample && dec->sample_fmt!=enc->sample_fmt && |
a79db0f7 | 847 | MAKE_SFMT_PAIR(enc->sample_fmt,dec->sample_fmt)!=ost->reformat_pair) { |
a79db0f7 PR |
848 | if (ost->reformat_ctx) |
849 | av_audio_convert_free(ost->reformat_ctx); | |
850 | ost->reformat_ctx = av_audio_convert_alloc(enc->sample_fmt, 1, | |
851 | dec->sample_fmt, 1, NULL, 0); | |
852 | if (!ost->reformat_ctx) { | |
853 | fprintf(stderr, "Cannot convert %s sample format to %s sample format\n", | |
ba7d6e79 SS |
854 | av_get_sample_fmt_name(dec->sample_fmt), |
855 | av_get_sample_fmt_name(enc->sample_fmt)); | |
639e4ec8 | 856 | ffmpeg_exit(1); |
a79db0f7 PR |
857 | } |
858 | ost->reformat_pair=MAKE_SFMT_PAIR(enc->sample_fmt,dec->sample_fmt); | |
859 | } | |
860 | ||
986ebcdb | 861 | if(audio_sync_method){ |
115329f1 | 862 | double delta = get_sync_ipts(ost) * enc->sample_rate - ost->sync_opts |
37f5a713 JR |
863 | - av_fifo_size(ost->fifo)/(enc->channels * 2); |
864 | double idelta= delta*dec->sample_rate / enc->sample_rate; | |
ff19d16b | 865 | int byte_delta= ((int)idelta)*2*dec->channels; |
ff4905a5 | 866 | |
986ebcdb MN |
867 | //FIXME resample delay |
868 | if(fabs(delta) > 50){ | |
d4d226a8 | 869 | if(ist->is_start || fabs(delta) > audio_drift_threshold*enc->sample_rate){ |
ff4905a5 | 870 | if(byte_delta < 0){ |
f41dd5aa | 871 | byte_delta= FFMAX(byte_delta, -size); |
ff4905a5 MN |
872 | size += byte_delta; |
873 | buf -= byte_delta; | |
874 | if(verbose > 2) | |
875 | fprintf(stderr, "discarding %d audio samples\n", (int)-delta); | |
876 | if(!size) | |
877 | return; | |
878 | ist->is_start=0; | |
879 | }else{ | |
880 | static uint8_t *input_tmp= NULL; | |
881 | input_tmp= av_realloc(input_tmp, byte_delta + size); | |
882 | ||
7a086a85 MN |
883 | if(byte_delta > allocated_for_size - size){ |
884 | allocated_for_size= byte_delta + (int64_t)size; | |
885 | goto need_realloc; | |
886 | } | |
887 | ist->is_start=0; | |
ff4905a5 MN |
888 | |
889 | memset(input_tmp, 0, byte_delta); | |
890 | memcpy(input_tmp + byte_delta, buf, size); | |
891 | buf= input_tmp; | |
892 | size += byte_delta; | |
893 | if(verbose > 2) | |
894 | fprintf(stderr, "adding %d audio samples of silence\n", (int)delta); | |
895 | } | |
896 | }else if(audio_sync_method>1){ | |
f66e4f5f | 897 | int comp= av_clip(delta, -audio_sync_method, audio_sync_method); |
b926b628 | 898 | av_assert0(ost->audio_resample); |
ff4905a5 MN |
899 | if(verbose > 2) |
900 | fprintf(stderr, "compensating audio timestamp drift:%f compensation:%d in:%d\n", delta, comp, enc->sample_rate); | |
41dd680d | 901 | // 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)); |
ff4905a5 MN |
902 | av_resample_compensate(*(struct AVResampleContext**)ost->resample, comp, enc->sample_rate); |
903 | } | |
115329f1 | 904 | } |
986ebcdb | 905 | }else |
b4a3389e | 906 | ost->sync_opts= lrintf(get_sync_ipts(ost) * enc->sample_rate) |
37f5a713 | 907 | - av_fifo_size(ost->fifo)/(enc->channels * 2); //FIXME wrong |
85f07f22 FB |
908 | |
909 | if (ost->audio_resample) { | |
910 | buftmp = audio_buf; | |
115329f1 | 911 | size_out = audio_resample(ost->resample, |
85f07f22 | 912 | (short *)buftmp, (short *)buf, |
37f5a713 | 913 | size / (dec->channels * isize)); |
287ba997 | 914 | size_out = size_out * enc->channels * osize; |
85f07f22 FB |
915 | } else { |
916 | buftmp = buf; | |
917 | size_out = size; | |
918 | } | |
919 | ||
d1e3c6fd | 920 | if (!ost->audio_resample && dec->sample_fmt!=enc->sample_fmt) { |
a79db0f7 | 921 | const void *ibuf[6]= {buftmp}; |
80f47250 | 922 | void *obuf[6]= {audio_buf}; |
287ba997 PR |
923 | int istride[6]= {isize}; |
924 | int ostride[6]= {osize}; | |
a79db0f7 PR |
925 | int len= size_out/istride[0]; |
926 | if (av_audio_convert(ost->reformat_ctx, obuf, ostride, ibuf, istride, len)<0) { | |
927 | printf("av_audio_convert() failed\n"); | |
f2abc559 | 928 | if (exit_on_error) |
639e4ec8 | 929 | ffmpeg_exit(1); |
a79db0f7 PR |
930 | return; |
931 | } | |
80f47250 | 932 | buftmp = audio_buf; |
287ba997 | 933 | size_out = len*osize; |
a79db0f7 PR |
934 | } |
935 | ||
85f07f22 | 936 | /* now encode as many frames as possible */ |
a0663ba4 | 937 | if (enc->frame_size > 1) { |
85f07f22 | 938 | /* output resampled raw samples */ |
41dd680d | 939 | if (av_fifo_realloc2(ost->fifo, av_fifo_size(ost->fifo) + size_out) < 0) { |
745b39d5 | 940 | fprintf(stderr, "av_fifo_realloc2() failed\n"); |
639e4ec8 | 941 | ffmpeg_exit(1); |
745b39d5 | 942 | } |
41dd680d | 943 | av_fifo_generic_write(ost->fifo, buftmp, size_out, NULL); |
85f07f22 | 944 | |
287ba997 | 945 | frame_bytes = enc->frame_size * osize * enc->channels; |
115329f1 | 946 | |
41dd680d | 947 | while (av_fifo_size(ost->fifo) >= frame_bytes) { |
e928649b MN |
948 | AVPacket pkt; |
949 | av_init_packet(&pkt); | |
950 | ||
3898eed8 | 951 | av_fifo_generic_read(ost->fifo, audio_buf, frame_bytes, NULL); |
0871ae1a | 952 | |
5bc440e7 MN |
953 | //FIXME pass ost->sync_opts as AVFrame.pts in avcodec_encode_audio() |
954 | ||
115329f1 | 955 | ret = avcodec_encode_audio(enc, audio_out, audio_out_size, |
a0663ba4 | 956 | (short *)audio_buf); |
528271ff MN |
957 | if (ret < 0) { |
958 | fprintf(stderr, "Audio encoding failed\n"); | |
639e4ec8 | 959 | ffmpeg_exit(1); |
528271ff | 960 | } |
1008ceb3 | 961 | audio_size += ret; |
e928649b MN |
962 | pkt.stream_index= ost->index; |
963 | pkt.data= audio_out; | |
964 | pkt.size= ret; | |
e7902f20 | 965 | if(enc->coded_frame && enc->coded_frame->pts != AV_NOPTS_VALUE) |
c0df9d75 | 966 | pkt.pts= av_rescale_q(enc->coded_frame->pts, enc->time_base, ost->st->time_base); |
cc947f04 | 967 | pkt.flags |= AV_PKT_FLAG_KEY; |
0b6d358a | 968 | write_frame(s, &pkt, enc, ost->bitstream_filters); |
115329f1 | 969 | |
986ebcdb | 970 | ost->sync_opts += enc->frame_size; |
85f07f22 FB |
971 | } |
972 | } else { | |
e928649b MN |
973 | AVPacket pkt; |
974 | av_init_packet(&pkt); | |
986ebcdb | 975 | |
287ba997 | 976 | ost->sync_opts += size_out / (osize * enc->channels); |
986ebcdb | 977 | |
a0663ba4 | 978 | /* output a pcm frame */ |
287ba997 PR |
979 | /* determine the size of the coded buffer */ |
980 | size_out /= osize; | |
981 | if (coded_bps) | |
060b8592 | 982 | size_out = size_out*coded_bps/8; |
287ba997 | 983 | |
5ee05a62 MN |
984 | if(size_out > audio_out_size){ |
985 | fprintf(stderr, "Internal error, buffer size too small\n"); | |
639e4ec8 | 986 | ffmpeg_exit(1); |
5ee05a62 MN |
987 | } |
988 | ||
5bc440e7 | 989 | //FIXME pass ost->sync_opts as AVFrame.pts in avcodec_encode_audio() |
115329f1 | 990 | ret = avcodec_encode_audio(enc, audio_out, size_out, |
bb270c08 | 991 | (short *)buftmp); |
528271ff MN |
992 | if (ret < 0) { |
993 | fprintf(stderr, "Audio encoding failed\n"); | |
639e4ec8 | 994 | ffmpeg_exit(1); |
528271ff | 995 | } |
1008ceb3 | 996 | audio_size += ret; |
e928649b MN |
997 | pkt.stream_index= ost->index; |
998 | pkt.data= audio_out; | |
999 | pkt.size= ret; | |
e7902f20 | 1000 | if(enc->coded_frame && enc->coded_frame->pts != AV_NOPTS_VALUE) |
c0df9d75 | 1001 | pkt.pts= av_rescale_q(enc->coded_frame->pts, enc->time_base, ost->st->time_base); |
cc947f04 | 1002 | pkt.flags |= AV_PKT_FLAG_KEY; |
0b6d358a | 1003 | write_frame(s, &pkt, enc, ost->bitstream_filters); |
85f07f22 FB |
1004 | } |
1005 | } | |
1006 | ||
17c8cc55 | 1007 | static void pre_process_video_frame(InputStream *ist, AVPicture *picture, void **bufp) |
10d104e4 PG |
1008 | { |
1009 | AVCodecContext *dec; | |
1010 | AVPicture *picture2; | |
1011 | AVPicture picture_tmp; | |
0c1a9eda | 1012 | uint8_t *buf = 0; |
10d104e4 | 1013 | |
01f4895c | 1014 | dec = ist->st->codec; |
10d104e4 PG |
1015 | |
1016 | /* deinterlace : must be done before any resize */ | |
fdf11906 | 1017 | if (do_deinterlace) { |
10d104e4 PG |
1018 | int size; |
1019 | ||
1020 | /* create temporary picture */ | |
1021 | size = avpicture_get_size(dec->pix_fmt, dec->width, dec->height); | |
1022 | buf = av_malloc(size); | |
1023 | if (!buf) | |
1024 | return; | |
115329f1 | 1025 | |
10d104e4 PG |
1026 | picture2 = &picture_tmp; |
1027 | avpicture_fill(picture2, buf, dec->pix_fmt, dec->width, dec->height); | |
1028 | ||
9e1cc598 RP |
1029 | if(avpicture_deinterlace(picture2, picture, |
1030 | dec->pix_fmt, dec->width, dec->height) < 0) { | |
1031 | /* if error, do not deinterlace */ | |
1032 | fprintf(stderr, "Deinterlacing failed\n"); | |
1033 | av_free(buf); | |
1034 | buf = NULL; | |
1035 | picture2 = picture; | |
1036 | } | |
10d104e4 PG |
1037 | } else { |
1038 | picture2 = picture; | |
1039 | } | |
1040 | ||
10d104e4 PG |
1041 | if (picture != picture2) |
1042 | *picture = *picture2; | |
1043 | *bufp = buf; | |
1044 | } | |
1045 | ||
ec5517d5 FB |
1046 | /* we begin to correct av delay at this threshold */ |
1047 | #define AV_DELAY_MAX 0.100 | |
85f07f22 | 1048 | |
115329f1 | 1049 | static void do_subtitle_out(AVFormatContext *s, |
17c8cc55 AK |
1050 | OutputStream *ost, |
1051 | InputStream *ist, | |
cf7fc795 FB |
1052 | AVSubtitle *sub, |
1053 | int64_t pts) | |
1054 | { | |
1055 | static uint8_t *subtitle_out = NULL; | |
7f4fca03 | 1056 | int subtitle_out_max_size = 1024 * 1024; |
cf7fc795 FB |
1057 | int subtitle_out_size, nb, i; |
1058 | AVCodecContext *enc; | |
1059 | AVPacket pkt; | |
1060 | ||
1061 | if (pts == AV_NOPTS_VALUE) { | |
1062 | fprintf(stderr, "Subtitle packets must have a pts\n"); | |
f2abc559 | 1063 | if (exit_on_error) |
639e4ec8 | 1064 | ffmpeg_exit(1); |
cf7fc795 FB |
1065 | return; |
1066 | } | |
1067 | ||
01f4895c | 1068 | enc = ost->st->codec; |
cf7fc795 FB |
1069 | |
1070 | if (!subtitle_out) { | |
1071 | subtitle_out = av_malloc(subtitle_out_max_size); | |
1072 | } | |
1073 | ||
1074 | /* Note: DVB subtitle need one packet to draw them and one other | |
1075 | packet to clear them */ | |
1076 | /* XXX: signal it in the codec context ? */ | |
1077 | if (enc->codec_id == CODEC_ID_DVB_SUBTITLE) | |
1078 | nb = 2; | |
1079 | else | |
1080 | nb = 1; | |
1081 | ||
1082 | for(i = 0; i < nb; i++) { | |
4bbe788a | 1083 | sub->pts = av_rescale_q(pts, ist->st->time_base, AV_TIME_BASE_Q); |
8b03c014 RD |
1084 | // start_display_time is required to be 0 |
1085 | sub->pts += av_rescale_q(sub->start_display_time, (AVRational){1, 1000}, AV_TIME_BASE_Q); | |
1086 | sub->end_display_time -= sub->start_display_time; | |
1087 | sub->start_display_time = 0; | |
115329f1 | 1088 | subtitle_out_size = avcodec_encode_subtitle(enc, subtitle_out, |
cf7fc795 | 1089 | subtitle_out_max_size, sub); |
266649a5 RD |
1090 | if (subtitle_out_size < 0) { |
1091 | fprintf(stderr, "Subtitle encoding failed\n"); | |
639e4ec8 | 1092 | ffmpeg_exit(1); |
266649a5 | 1093 | } |
115329f1 | 1094 | |
cf7fc795 FB |
1095 | av_init_packet(&pkt); |
1096 | pkt.stream_index = ost->index; | |
1097 | pkt.data = subtitle_out; | |
1098 | pkt.size = subtitle_out_size; | |
8b03c014 | 1099 | pkt.pts = av_rescale_q(sub->pts, AV_TIME_BASE_Q, ost->st->time_base); |
cf7fc795 FB |
1100 | if (enc->codec_id == CODEC_ID_DVB_SUBTITLE) { |
1101 | /* XXX: the pts correction is handled here. Maybe handling | |
1102 | it in the codec would be better */ | |
1103 | if (i == 0) | |
1104 | pkt.pts += 90 * sub->start_display_time; | |
1105 | else | |
1106 | pkt.pts += 90 * sub->end_display_time; | |
1107 | } | |
0b6d358a | 1108 | write_frame(s, &pkt, ost->st->codec, ost->bitstream_filters); |
cf7fc795 FB |
1109 | } |
1110 | } | |
1111 | ||
8a6cb114 | 1112 | static int bit_buffer_size= 1024*256; |
27537106 | 1113 | static uint8_t *bit_buffer= NULL; |
1ff93ffc | 1114 | |
115329f1 | 1115 | static void do_video_out(AVFormatContext *s, |
17c8cc55 AK |
1116 | OutputStream *ost, |
1117 | InputStream *ist, | |
7a0f9d7e | 1118 | AVFrame *in_picture, |
5e8d2e33 | 1119 | int *frame_size, float quality) |
85f07f22 | 1120 | { |
9aa797cd | 1121 | int nb_frames, i, ret, resample_changed; |
e65ab9d9 | 1122 | AVFrame *final_picture, *formatted_picture; |
cfcf0ffd | 1123 | AVCodecContext *enc, *dec; |
47b229db | 1124 | double sync_ipts; |
115329f1 | 1125 | |
01f4895c MN |
1126 | enc = ost->st->codec; |
1127 | dec = ist->st->codec; | |
85f07f22 | 1128 | |
47b229db AS |
1129 | sync_ipts = get_sync_ipts(ost) / av_q2d(enc->time_base); |
1130 | ||
ec5517d5 FB |
1131 | /* by default, we output a single frame */ |
1132 | nb_frames = 1; | |
1133 | ||
204c0f48 PG |
1134 | *frame_size = 0; |
1135 | ||
9ff261a2 | 1136 | if(video_sync_method){ |
d55065a2 | 1137 | double vdelta = sync_ipts - ost->sync_opts; |
e928649b MN |
1138 | //FIXME set to 0.5 after we fix some dts/pts bugs like in avidec.c |
1139 | if (vdelta < -1.1) | |
1140 | nb_frames = 0; | |
e6fdc2b1 MN |
1141 | else if (video_sync_method == 2 || (video_sync_method<0 && (s->oformat->flags & AVFMT_VARIABLE_FPS))){ |
1142 | if(vdelta<=-0.6){ | |
1143 | nb_frames=0; | |
1144 | }else if(vdelta>0.6) | |
7f48bfa1 | 1145 | ost->sync_opts= lrintf(sync_ipts); |
e6fdc2b1 | 1146 | }else if (vdelta > 1.1) |
70122f29 | 1147 | nb_frames = lrintf(vdelta); |
fc6765d7 | 1148 | //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); |
50c3dd32 MN |
1149 | if (nb_frames == 0){ |
1150 | ++nb_frames_drop; | |
1151 | if (verbose>2) | |
1152 | fprintf(stderr, "*** drop!\n"); | |
8300609b | 1153 | }else if (nb_frames > 1) { |
ed30e518 | 1154 | nb_frames_dup += nb_frames - 1; |
50c3dd32 | 1155 | if (verbose>2) |
8300609b | 1156 | fprintf(stderr, "*** %d dup!\n", nb_frames-1); |
50c3dd32 MN |
1157 | } |
1158 | }else | |
47b229db | 1159 | ost->sync_opts= lrintf(sync_ipts); |
445f1b83 | 1160 | |
72415b2a | 1161 | nb_frames= FFMIN(nb_frames, max_frames[AVMEDIA_TYPE_VIDEO] - ost->frame_number); |
115329f1 | 1162 | if (nb_frames <= 0) |
85f07f22 | 1163 | return; |
ce7c56c2 | 1164 | |
46847a33 | 1165 | formatted_picture = in_picture; |
07d0cdfc | 1166 | final_picture = formatted_picture; |
07d0cdfc | 1167 | |
9aa797cd SS |
1168 | resample_changed = ost->resample_width != dec->width || |
1169 | ost->resample_height != dec->height || | |
1170 | ost->resample_pix_fmt != dec->pix_fmt; | |
1171 | ||
1172 | if (resample_changed) { | |
2ecc5b70 SS |
1173 | av_log(NULL, AV_LOG_INFO, |
1174 | "Input stream #%d.%d frame changed from size:%dx%d fmt:%s to size:%dx%d fmt:%s\n", | |
d2bc4da1 | 1175 | ist->file_index, ist->st->index, |
94bed8e5 SS |
1176 | ost->resample_width, ost->resample_height, av_get_pix_fmt_name(ost->resample_pix_fmt), |
1177 | dec->width , dec->height , av_get_pix_fmt_name(dec->pix_fmt)); | |
b83ccbff | 1178 | if(!ost->video_resample) |
639e4ec8 | 1179 | ffmpeg_exit(1); |
b83ccbff MN |
1180 | } |
1181 | ||
46847a33 | 1182 | #if !CONFIG_AVFILTER |
85f07f22 | 1183 | if (ost->video_resample) { |
34b10a57 | 1184 | final_picture = &ost->pict_tmp; |
9aa797cd | 1185 | if (resample_changed) { |
352666c1 EB |
1186 | /* initialize a new scaler context */ |
1187 | sws_freeContext(ost->img_resample_ctx); | |
352666c1 | 1188 | ost->img_resample_ctx = sws_getContext( |
5879ea6d SS |
1189 | ist->st->codec->width, |
1190 | ist->st->codec->height, | |
352666c1 | 1191 | ist->st->codec->pix_fmt, |
0c22311b MN |
1192 | ost->st->codec->width, |
1193 | ost->st->codec->height, | |
352666c1 | 1194 | ost->st->codec->pix_fmt, |
1435f2fa | 1195 | ost->sws_flags, NULL, NULL, NULL); |
352666c1 EB |
1196 | if (ost->img_resample_ctx == NULL) { |
1197 | fprintf(stderr, "Cannot get resampling context\n"); | |
639e4ec8 | 1198 | ffmpeg_exit(1); |
352666c1 EB |
1199 | } |
1200 | } | |
18a54b04 | 1201 | sws_scale(ost->img_resample_ctx, formatted_picture->data, formatted_picture->linesize, |
e65ab9d9 | 1202 | 0, ost->resample_height, final_picture->data, final_picture->linesize); |
f122deb4 | 1203 | } |
46847a33 | 1204 | #endif |
07d0cdfc | 1205 | |
85f07f22 | 1206 | /* duplicates frame if needed */ |
ec5517d5 | 1207 | for(i=0;i<nb_frames;i++) { |
e928649b MN |
1208 | AVPacket pkt; |
1209 | av_init_packet(&pkt); | |
1210 | pkt.stream_index= ost->index; | |
1211 | ||
e8750b00 FR |
1212 | if (s->oformat->flags & AVFMT_RAWPICTURE) { |
1213 | /* raw pictures are written as AVPicture structure to | |
1214 | avoid any copies. We support temorarily the older | |
1215 | method. */ | |
2744ca9a | 1216 | AVFrame* old_frame = enc->coded_frame; |
bb270c08 | 1217 | enc->coded_frame = dec->coded_frame; //FIXME/XXX remove this hack |
e928649b MN |
1218 | pkt.data= (uint8_t *)final_picture; |
1219 | pkt.size= sizeof(AVPicture); | |
b4dba580 | 1220 | pkt.pts= av_rescale_q(ost->sync_opts, enc->time_base, ost->st->time_base); |
cc947f04 | 1221 | pkt.flags |= AV_PKT_FLAG_KEY; |
e928649b | 1222 | |
0b6d358a | 1223 | write_frame(s, &pkt, ost->st->codec, ost->bitstream_filters); |
bb270c08 | 1224 | enc->coded_frame = old_frame; |
e8750b00 | 1225 | } else { |
492cd3a9 | 1226 | AVFrame big_picture; |
a4d36c11 MN |
1227 | |
1228 | big_picture= *final_picture; | |
7a0f9d7e FB |
1229 | /* better than nothing: use input picture interlaced |
1230 | settings */ | |
1231 | big_picture.interlaced_frame = in_picture->interlaced_frame; | |
648e55ff | 1232 | if (ost->st->codec->flags & (CODEC_FLAG_INTERLACED_DCT|CODEC_FLAG_INTERLACED_ME)) { |
bb198e19 MN |
1233 | if(top_field_first == -1) |
1234 | big_picture.top_field_first = in_picture->top_field_first; | |
1235 | else | |
2a8edc5d | 1236 | big_picture.top_field_first = top_field_first; |
bb198e19 | 1237 | } |
7a0f9d7e | 1238 | |
85f07f22 FB |
1239 | /* handles sameq here. This is not correct because it may |
1240 | not be a global option */ | |
5e8d2e33 | 1241 | big_picture.quality = quality; |
f4f3223f MN |
1242 | if(!me_threshold) |
1243 | big_picture.pict_type = 0; | |
50c3dd32 | 1244 | // big_picture.pts = AV_NOPTS_VALUE; |
c0df9d75 MN |
1245 | big_picture.pts= ost->sync_opts; |
1246 | // big_picture.pts= av_rescale(ost->sync_opts, AV_TIME_BASE*(int64_t)enc->time_base.num, enc->time_base.den); | |
949b1a13 | 1247 | //av_log(NULL, AV_LOG_DEBUG, "%"PRId64" -> encoder\n", ost->sync_opts); |
4ad08021 NG |
1248 | if (ost->forced_kf_index < ost->forced_kf_count && |
1249 | big_picture.pts >= ost->forced_kf_pts[ost->forced_kf_index]) { | |
975a1447 | 1250 | big_picture.pict_type = AV_PICTURE_TYPE_I; |
4ad08021 NG |
1251 | ost->forced_kf_index++; |
1252 | } | |
115329f1 | 1253 | ret = avcodec_encode_video(enc, |
8a6cb114 | 1254 | bit_buffer, bit_buffer_size, |
1e491e29 | 1255 | &big_picture); |
95af5e1c | 1256 | if (ret < 0) { |
4156a436 | 1257 | fprintf(stderr, "Video encoding failed\n"); |
639e4ec8 | 1258 | ffmpeg_exit(1); |
4156a436 | 1259 | } |
54e28a85 | 1260 | |
4bbc6260 | 1261 | if(ret>0){ |
27537106 | 1262 | pkt.data= bit_buffer; |
e928649b | 1263 | pkt.size= ret; |
e6b4e4ff | 1264 | if(enc->coded_frame->pts != AV_NOPTS_VALUE) |
c0df9d75 | 1265 | pkt.pts= av_rescale_q(enc->coded_frame->pts, enc->time_base, ost->st->time_base); |
949b1a13 | 1266 | /*av_log(NULL, AV_LOG_DEBUG, "encoder -> %"PRId64"/%"PRId64"\n", |
c0df9d75 MN |
1267 | pkt.pts != AV_NOPTS_VALUE ? av_rescale(pkt.pts, enc->time_base.den, AV_TIME_BASE*(int64_t)enc->time_base.num) : -1, |
1268 | pkt.dts != AV_NOPTS_VALUE ? av_rescale(pkt.dts, enc->time_base.den, AV_TIME_BASE*(int64_t)enc->time_base.num) : -1);*/ | |
50c3dd32 | 1269 | |
e6b4e4ff | 1270 | if(enc->coded_frame->key_frame) |
cc947f04 | 1271 | pkt.flags |= AV_PKT_FLAG_KEY; |
0b6d358a | 1272 | write_frame(s, &pkt, ost->st->codec, ost->bitstream_filters); |
e928649b | 1273 | *frame_size = ret; |
44eb047a | 1274 | video_size += ret; |
54e28a85 BC |
1275 | //fprintf(stderr,"\nFrame: %3d size: %5d type: %d", |
1276 | // enc->frame_number-1, ret, enc->pict_type); | |
e928649b MN |
1277 | /* if two pass, output log */ |
1278 | if (ost->logfile && enc->stats_out) { | |
1279 | fprintf(ost->logfile, "%s", enc->stats_out); | |
1280 | } | |
5abdb4b1 | 1281 | } |
85f07f22 | 1282 | } |
50c3dd32 | 1283 | ost->sync_opts++; |
ec5517d5 | 1284 | ost->frame_number++; |
85f07f22 FB |
1285 | } |
1286 | } | |
1287 | ||
140cb663 | 1288 | static double psnr(double d){ |
b29f97d1 | 1289 | return -10.0*log(d)/log(10.0); |
140cb663 MN |
1290 | } |
1291 | ||
17c8cc55 | 1292 | static void do_video_stats(AVFormatContext *os, OutputStream *ost, |
ec5517d5 | 1293 | int frame_size) |
ce7c56c2 | 1294 | { |
ce7c56c2 J |
1295 | AVCodecContext *enc; |
1296 | int frame_number; | |
ce7c56c2 | 1297 | double ti1, bitrate, avg_bitrate; |
115329f1 | 1298 | |
b60d1379 | 1299 | /* this is executed just the first time do_video_stats is called */ |
032aa7df SS |
1300 | if (!vstats_file) { |
1301 | vstats_file = fopen(vstats_filename, "w"); | |
1302 | if (!vstats_file) { | |
4bd0c2b1 | 1303 | perror("fopen"); |
639e4ec8 | 1304 | ffmpeg_exit(1); |
4bd0c2b1 BF |
1305 | } |
1306 | } | |
1307 | ||
01f4895c | 1308 | enc = ost->st->codec; |
72415b2a | 1309 | if (enc->codec_type == AVMEDIA_TYPE_VIDEO) { |
ec5517d5 | 1310 | frame_number = ost->frame_number; |
032aa7df | 1311 | fprintf(vstats_file, "frame= %5d q= %2.1f ", frame_number, enc->coded_frame->quality/(float)FF_QP2LAMBDA); |
140cb663 | 1312 | if (enc->flags&CODEC_FLAG_PSNR) |
032aa7df | 1313 | fprintf(vstats_file, "PSNR= %6.2f ", psnr(enc->coded_frame->error[0]/(enc->width*enc->height*255.0*255.0))); |
115329f1 | 1314 | |
032aa7df | 1315 | fprintf(vstats_file,"f_size= %6d ", frame_size); |
ec5517d5 | 1316 | /* compute pts value */ |
c0df9d75 | 1317 | ti1 = ost->sync_opts * av_q2d(enc->time_base); |
ce7c56c2 J |
1318 | if (ti1 < 0.01) |
1319 | ti1 = 0.01; | |
115329f1 | 1320 | |
c0df9d75 | 1321 | bitrate = (frame_size * 8) / av_q2d(enc->time_base) / 1000.0; |
1008ceb3 | 1322 | avg_bitrate = (double)(video_size * 8) / ti1 / 1000.0; |
032aa7df | 1323 | fprintf(vstats_file, "s_size= %8.0fkB time= %0.3f br= %7.1fkbits/s avg_br= %7.1fkbits/s ", |
1008ceb3 | 1324 | (double)video_size / 1024, ti1, bitrate, avg_bitrate); |
6209669d | 1325 | fprintf(vstats_file, "type= %c\n", av_get_picture_type_char(enc->coded_frame->pict_type)); |
ce7c56c2 | 1326 | } |
ec5517d5 FB |
1327 | } |
1328 | ||
b29f97d1 | 1329 | static void print_report(AVFormatContext **output_files, |
17c8cc55 | 1330 | OutputStream **ost_table, int nb_ostreams, |
bb270c08 | 1331 | int is_last_report) |
ec5517d5 FB |
1332 | { |
1333 | char buf[1024]; | |
17c8cc55 | 1334 | OutputStream *ost; |
b5ee9c23 | 1335 | AVFormatContext *oc; |
0c1a9eda | 1336 | int64_t total_size; |
ec5517d5 FB |
1337 | AVCodecContext *enc; |
1338 | int frame_number, vid, i; | |
1339 | double bitrate, ti1, pts; | |
0c1a9eda | 1340 | static int64_t last_time = -1; |
0888fd22 | 1341 | static int qp_histogram[52]; |
115329f1 | 1342 | |
ec5517d5 | 1343 | if (!is_last_report) { |
0c1a9eda | 1344 | int64_t cur_time; |
ec5517d5 FB |
1345 | /* display the report every 0.5 seconds */ |
1346 | cur_time = av_gettime(); | |
1347 | if (last_time == -1) { | |
1348 | last_time = cur_time; | |
1349 | return; | |
115329f1 | 1350 | } |
ec5517d5 FB |
1351 | if ((cur_time - last_time) < 500000) |
1352 | return; | |
1353 | last_time = cur_time; | |
1354 | } | |
1355 | ||
ce7c56c2 | 1356 | |
ec5517d5 FB |
1357 | oc = output_files[0]; |
1358 | ||
76aa876e AK |
1359 | total_size = avio_size(oc->pb); |
1360 | if(total_size<0) // FIXME improve avio_size() so it works with non seekable output too | |
a2704c97 | 1361 | total_size= avio_tell(oc->pb); |
115329f1 | 1362 | |
ec5517d5 FB |
1363 | buf[0] = '\0'; |
1364 | ti1 = 1e10; | |
1365 | vid = 0; | |
1366 | for(i=0;i<nb_ostreams;i++) { | |
5da116a3 | 1367 | float q = -1; |
ec5517d5 | 1368 | ost = ost_table[i]; |
01f4895c | 1369 | enc = ost->st->codec; |
5da116a3 MN |
1370 | if (!ost->st->stream_copy && enc->coded_frame) |
1371 | q = enc->coded_frame->quality/(float)FF_QP2LAMBDA; | |
72415b2a | 1372 | if (vid && enc->codec_type == AVMEDIA_TYPE_VIDEO) { |
5da116a3 | 1373 | snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "q=%2.1f ", q); |
10d104e4 | 1374 | } |
72415b2a | 1375 | if (!vid && enc->codec_type == AVMEDIA_TYPE_VIDEO) { |
7fc98937 LW |
1376 | float t = (av_gettime()-timer_start) / 1000000.0; |
1377 | ||
ec5517d5 | 1378 | frame_number = ost->frame_number; |
7fc98937 | 1379 | snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "frame=%5d fps=%3d q=%3.1f ", |
5da116a3 | 1380 | frame_number, (t>1)?(int)(frame_number/t+0.5) : 0, q); |
890972be | 1381 | if(is_last_report) |
0ecca7a4 | 1382 | snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "L"); |
e6b4e4ff | 1383 | if(qp_hist){ |
0888fd22 | 1384 | int j; |
5da116a3 | 1385 | int qp = lrintf(q); |
37d3e066 | 1386 | if(qp>=0 && qp<FF_ARRAY_ELEMS(qp_histogram)) |
0888fd22 MN |
1387 | qp_histogram[qp]++; |
1388 | for(j=0; j<32; j++) | |
1389 | snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "%X", (int)lrintf(log(qp_histogram[j]+1)/log(2))); | |
1390 | } | |
890972be MN |
1391 | if (enc->flags&CODEC_FLAG_PSNR){ |
1392 | int j; | |
1393 | double error, error_sum=0; | |
1394 | double scale, scale_sum=0; | |
1395 | char type[3]= {'Y','U','V'}; | |
0ecca7a4 | 1396 | snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "PSNR="); |
890972be MN |
1397 | for(j=0; j<3; j++){ |
1398 | if(is_last_report){ | |
1399 | error= enc->error[j]; | |
1400 | scale= enc->width*enc->height*255.0*255.0*frame_number; | |
1401 | }else{ | |
1402 | error= enc->coded_frame->error[j]; | |
1403 | scale= enc->width*enc->height*255.0*255.0; | |
1404 | } | |
1405 | if(j) scale/=4; | |
1406 | error_sum += error; | |
1407 | scale_sum += scale; | |
0ecca7a4 | 1408 | snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "%c:%2.2f ", type[j], psnr(error/scale)); |
890972be | 1409 | } |
0ecca7a4 | 1410 | snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "*:%2.2f ", psnr(error_sum/scale_sum)); |
890972be | 1411 | } |
ec5517d5 FB |
1412 | vid = 1; |
1413 | } | |
1414 | /* compute min output value */ | |
43924f01 | 1415 | pts = (double)ost->st->pts.val * av_q2d(ost->st->time_base); |
5d9827bc | 1416 | if ((pts < ti1) && (pts > 0)) |
ec5517d5 FB |
1417 | ti1 = pts; |
1418 | } | |
1419 | if (ti1 < 0.01) | |
1420 | ti1 = 0.01; | |
115329f1 | 1421 | |
d1991f51 | 1422 | if (verbose > 0 || is_last_report) { |
f068206e | 1423 | bitrate = (double)(total_size * 8) / ti1 / 1000.0; |
115329f1 DB |
1424 | |
1425 | snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), | |
475f4d8d | 1426 | "size=%8.0fkB time=%0.2f bitrate=%6.1fkbits/s", |
ec5517d5 | 1427 | (double)total_size / 1024, ti1, bitrate); |
a6a92a9a | 1428 | |
0f649d66 | 1429 | if (nb_frames_dup || nb_frames_drop) |
bb270c08 DB |
1430 | snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), " dup=%d drop=%d", |
1431 | nb_frames_dup, nb_frames_drop); | |
115329f1 | 1432 | |
d8019eb5 AD |
1433 | if (verbose >= 0) |
1434 | fprintf(stderr, "%s \r", buf); | |
1435 | ||
ec5517d5 FB |
1436 | fflush(stderr); |
1437 | } | |
115329f1 | 1438 | |
1008ceb3 MN |
1439 | if (is_last_report && verbose >= 0){ |
1440 | int64_t raw= audio_size + video_size + extra_size; | |
f068206e | 1441 | fprintf(stderr, "\n"); |
1008ceb3 MN |
1442 | fprintf(stderr, "video:%1.0fkB audio:%1.0fkB global headers:%1.0fkB muxing overhead %f%%\n", |
1443 | video_size/1024.0, | |
1444 | audio_size/1024.0, | |
1445 | extra_size/1024.0, | |
1446 | 100.0*(total_size - raw)/raw | |
1447 | ); | |
1448 | } | |
ce7c56c2 J |
1449 | } |
1450 | ||
14cf0fd2 AC |
1451 | static void generate_silence(uint8_t* buf, enum AVSampleFormat sample_fmt, size_t size) |
1452 | { | |
1453 | int fill_char = 0x00; | |
1454 | if (sample_fmt == AV_SAMPLE_FMT_U8) | |
1455 | fill_char = 0x80; | |
1456 | memset(buf, fill_char, size); | |
1457 | } | |
1458 | ||
a700a6ae | 1459 | /* pkt = NULL means EOF (needed to flush decoder buffers) */ |
17c8cc55 AK |
1460 | static int output_packet(InputStream *ist, int ist_index, |
1461 | OutputStream **ost_table, int nb_ostreams, | |
4b85a28f | 1462 | const AVPacket *pkt) |
a700a6ae FB |
1463 | { |
1464 | AVFormatContext *os; | |
17c8cc55 | 1465 | OutputStream *ost; |
ede0e475 | 1466 | int ret, i; |
cb48fdf6 | 1467 | int got_output; |
a700a6ae | 1468 | AVFrame picture; |
3fd62c6e | 1469 | void *buffer_to_free = NULL; |
f038fe8b | 1470 | static unsigned int samples_size= 0; |
cf7fc795 | 1471 | AVSubtitle subtitle, *subtitle_to_free; |
0ff4f0c0 | 1472 | int64_t pkt_pts = AV_NOPTS_VALUE; |
46847a33 | 1473 | #if CONFIG_AVFILTER |
d21f58b5 | 1474 | int frame_available; |
46847a33 | 1475 | #endif |
5e8d2e33 | 1476 | float quality; |
46847a33 | 1477 | |
ede0e475 | 1478 | AVPacket avpkt; |
e6c52cee | 1479 | int bps = av_get_bytes_per_sample(ist->st->codec->sample_fmt); |
ede0e475 | 1480 | |
ed923859 MN |
1481 | if(ist->next_pts == AV_NOPTS_VALUE) |
1482 | ist->next_pts= ist->pts; | |
1483 | ||
a700a6ae FB |
1484 | if (pkt == NULL) { |
1485 | /* EOF handling */ | |
031e14ea | 1486 | av_init_packet(&avpkt); |
ede0e475 TB |
1487 | avpkt.data = NULL; |
1488 | avpkt.size = 0; | |
a700a6ae | 1489 | goto handle_eof; |
031e14ea TB |
1490 | } else { |
1491 | avpkt = *pkt; | |
a700a6ae FB |
1492 | } |
1493 | ||
b1b818fc MN |
1494 | if(pkt->dts != AV_NOPTS_VALUE) |
1495 | ist->next_pts = ist->pts = av_rescale_q(pkt->dts, ist->st->time_base, AV_TIME_BASE_Q); | |
0ff4f0c0 AS |
1496 | if(pkt->pts != AV_NOPTS_VALUE) |
1497 | pkt_pts = av_rescale_q(pkt->pts, ist->st->time_base, AV_TIME_BASE_Q); | |
b1b818fc | 1498 | |
bd6754aa | 1499 | //while we have more to decode or while the decoder did output something on EOF |
cb48fdf6 | 1500 | while (avpkt.size > 0 || (!pkt && got_output)) { |
036c1382 MN |
1501 | uint8_t *data_buf, *decoded_data_buf; |
1502 | int data_size, decoded_data_size; | |
a700a6ae | 1503 | handle_eof: |
b1b818fc | 1504 | ist->pts= ist->next_pts; |
19d5da50 | 1505 | |
d859bb1d | 1506 | if(avpkt.size && avpkt.size != pkt->size && |
6e2fdc3e | 1507 | ((!ist->showed_multi_packet_warning && verbose>0) || verbose>1)){ |
40cb57a2 | 1508 | fprintf(stderr, "Multiple frames in a packet from stream %d\n", pkt->stream_index); |
3ff0daf0 MN |
1509 | ist->showed_multi_packet_warning=1; |
1510 | } | |
40cb57a2 | 1511 | |
a700a6ae | 1512 | /* decode the packet if needed */ |
036c1382 MN |
1513 | decoded_data_buf = NULL; /* fail safe */ |
1514 | decoded_data_size= 0; | |
1515 | data_buf = avpkt.data; | |
1516 | data_size = avpkt.size; | |
cf7fc795 | 1517 | subtitle_to_free = NULL; |
a700a6ae | 1518 | if (ist->decoding_needed) { |
01f4895c | 1519 | switch(ist->st->codec->codec_type) { |
72415b2a | 1520 | case AVMEDIA_TYPE_AUDIO:{ |
81b060fa LM |
1521 | if(pkt && samples_size < FFMAX(pkt->size*sizeof(*samples), AVCODEC_MAX_AUDIO_FRAME_SIZE)) { |
1522 | samples_size = FFMAX(pkt->size*sizeof(*samples), AVCODEC_MAX_AUDIO_FRAME_SIZE); | |
1523 | av_free(samples); | |
1524 | samples= av_malloc(samples_size); | |
1525 | } | |
036c1382 | 1526 | decoded_data_size= samples_size; |
a700a6ae FB |
1527 | /* XXX: could avoid copy if PCM 16 bits with same |
1528 | endianness as CPU */ | |
036c1382 | 1529 | ret = avcodec_decode_audio3(ist->st->codec, samples, &decoded_data_size, |
ede0e475 | 1530 | &avpkt); |
a700a6ae FB |
1531 | if (ret < 0) |
1532 | goto fail_decode; | |
ede0e475 TB |
1533 | avpkt.data += ret; |
1534 | avpkt.size -= ret; | |
036c1382 | 1535 | data_size = ret; |
cb48fdf6 | 1536 | got_output = decoded_data_size > 0; |
a700a6ae | 1537 | /* Some bug in mpeg audio decoder gives */ |
036c1382 | 1538 | /* decoded_data_size < 0, it seems they are overflows */ |
cb48fdf6 | 1539 | if (!got_output) { |
a700a6ae FB |
1540 | /* no audio frame */ |
1541 | continue; | |
1542 | } | |
036c1382 MN |
1543 | decoded_data_buf = (uint8_t *)samples; |
1544 | ist->next_pts += ((int64_t)AV_TIME_BASE/bps * decoded_data_size) / | |
01f4895c | 1545 | (ist->st->codec->sample_rate * ist->st->codec->channels); |
df84ac2e | 1546 | break;} |
72415b2a | 1547 | case AVMEDIA_TYPE_VIDEO: |
036c1382 | 1548 | decoded_data_size = (ist->st->codec->width * ist->st->codec->height * 3) / 2; |
a700a6ae | 1549 | /* XXX: allocate picture correctly */ |
9740beff | 1550 | avcodec_get_frame_defaults(&picture); |
6b474953 AS |
1551 | avpkt.pts = pkt_pts; |
1552 | avpkt.dts = ist->pts; | |
0ff4f0c0 | 1553 | pkt_pts = AV_NOPTS_VALUE; |
9740beff | 1554 | |
ede0e475 | 1555 | ret = avcodec_decode_video2(ist->st->codec, |
cb48fdf6 | 1556 | &picture, &got_output, &avpkt); |
5e8d2e33 | 1557 | quality = same_quality ? picture.quality : 0; |
115329f1 | 1558 | if (ret < 0) |
a700a6ae | 1559 | goto fail_decode; |
cb48fdf6 | 1560 | if (!got_output) { |
a700a6ae FB |
1561 | /* no picture yet */ |
1562 | goto discard_packet; | |
1563 | } | |
6b474953 | 1564 | ist->next_pts = ist->pts = guess_correct_pts(&ist->pts_ctx, picture.pkt_pts, picture.pkt_dts); |
01f4895c | 1565 | if (ist->st->codec->time_base.num != 0) { |
3797c74b | 1566 | int ticks= ist->st->parser ? ist->st->parser->repeat_pict+1 : ist->st->codec->ticks_per_frame; |
115329f1 | 1567 | ist->next_pts += ((int64_t)AV_TIME_BASE * |
34583e1b | 1568 | ist->st->codec->time_base.num * ticks) / |
01f4895c | 1569 | ist->st->codec->time_base.den; |
a700a6ae | 1570 | } |
ede0e475 | 1571 | avpkt.size = 0; |
3fd62c6e SS |
1572 | buffer_to_free = NULL; |
1573 | pre_process_video_frame(ist, (AVPicture *)&picture, &buffer_to_free); | |
a700a6ae | 1574 | break; |
72415b2a | 1575 | case AVMEDIA_TYPE_SUBTITLE: |
ede0e475 | 1576 | ret = avcodec_decode_subtitle2(ist->st->codec, |
cb48fdf6 | 1577 | &subtitle, &got_output, &avpkt); |
cf7fc795 | 1578 | if (ret < 0) |
a700a6ae | 1579 | goto fail_decode; |
cb48fdf6 | 1580 | if (!got_output) { |
cf7fc795 | 1581 | goto discard_packet; |
a700a6ae | 1582 | } |
cf7fc795 | 1583 | subtitle_to_free = &subtitle; |
ede0e475 | 1584 | avpkt.size = 0; |
cf7fc795 FB |
1585 | break; |
1586 | default: | |
1587 | goto fail_decode; | |
1588 | } | |
1589 | } else { | |
bde0705c | 1590 | switch(ist->st->codec->codec_type) { |
72415b2a | 1591 | case AVMEDIA_TYPE_AUDIO: |
bde0705c | 1592 | ist->next_pts += ((int64_t)AV_TIME_BASE * ist->st->codec->frame_size) / |
1c715415 | 1593 | ist->st->codec->sample_rate; |
bde0705c | 1594 | break; |
72415b2a | 1595 | case AVMEDIA_TYPE_VIDEO: |
bde0705c | 1596 | if (ist->st->codec->time_base.num != 0) { |
3797c74b | 1597 | int ticks= ist->st->parser ? ist->st->parser->repeat_pict+1 : ist->st->codec->ticks_per_frame; |
bde0705c | 1598 | ist->next_pts += ((int64_t)AV_TIME_BASE * |
34583e1b | 1599 | ist->st->codec->time_base.num * ticks) / |
bde0705c | 1600 | ist->st->codec->time_base.den; |
2fef0bdf | 1601 | } |
bde0705c | 1602 | break; |
a700a6ae | 1603 | } |
ede0e475 TB |
1604 | ret = avpkt.size; |
1605 | avpkt.size = 0; | |
bde0705c | 1606 | } |
a700a6ae | 1607 | |
46847a33 | 1608 | #if CONFIG_AVFILTER |
9d5fa618 MN |
1609 | if (ist->st->codec->codec_type == AVMEDIA_TYPE_VIDEO) { |
1610 | for (i = 0; i < nb_ostreams; i++) { | |
1611 | ost = ost_table[i]; | |
1612 | if (ost->input_video_filter && ost->source_index == ist_index) { | |
1613 | AVRational sar; | |
1614 | if (ist->st->sample_aspect_ratio.num) | |
1615 | sar = ist->st->sample_aspect_ratio; | |
1616 | else | |
1617 | sar = ist->st->codec->sample_aspect_ratio; | |
1618 | // add it to be filtered | |
1619 | av_vsrc_buffer_add_frame(ost->input_video_filter, &picture, | |
1620 | ist->pts, | |
1621 | sar); | |
1622 | } | |
1623 | } | |
46847a33 MN |
1624 | } |
1625 | #endif | |
1626 | ||
bde0705c | 1627 | // preprocess audio (volume) |
72415b2a | 1628 | if (ist->st->codec->codec_type == AVMEDIA_TYPE_AUDIO) { |
bde0705c LW |
1629 | if (audio_volume != 256) { |
1630 | short *volp; | |
1631 | volp = samples; | |
036c1382 | 1632 | for(i=0;i<(decoded_data_size / sizeof(short));i++) { |
bde0705c LW |
1633 | int v = ((*volp) * audio_volume + 128) >> 8; |
1634 | if (v < -32768) v = -32768; | |
1635 | if (v > 32767) v = 32767; | |
1636 | *volp++ = v; | |
7e987c33 C |
1637 | } |
1638 | } | |
bde0705c | 1639 | } |
7e987c33 | 1640 | |
bde0705c | 1641 | /* frame rate emulation */ |
3a25ca18 | 1642 | if (rate_emu) { |
cb103a19 | 1643 | int64_t pts = av_rescale(ist->pts, 1000000, AV_TIME_BASE); |
bde0705c LW |
1644 | int64_t now = av_gettime() - ist->start; |
1645 | if (pts > now) | |
1646 | usleep(pts - now); | |
bde0705c | 1647 | } |
bde0705c LW |
1648 | /* if output time reached then transcode raw format, |
1649 | encode packets and output them */ | |
1650 | if (start_time == 0 || ist->pts >= start_time) | |
1651 | for(i=0;i<nb_ostreams;i++) { | |
1652 | int frame_size; | |
a700a6ae | 1653 | |
bde0705c LW |
1654 | ost = ost_table[i]; |
1655 | if (ost->source_index == ist_index) { | |
9d5fa618 MN |
1656 | #if CONFIG_AVFILTER |
1657 | frame_available = ist->st->codec->codec_type != AVMEDIA_TYPE_VIDEO || | |
1658 | !ost->output_video_filter || avfilter_poll_frame(ost->output_video_filter->inputs[0]); | |
1659 | while (frame_available) { | |
1660 | AVRational ist_pts_tb; | |
1661 | if (ist->st->codec->codec_type == AVMEDIA_TYPE_VIDEO && ost->output_video_filter) | |
1662 | get_filtered_video_frame(ost->output_video_filter, &picture, &ost->picref, &ist_pts_tb); | |
1663 | if (ost->picref) | |
1664 | ist->pts = av_rescale_q(ost->picref->pts, ist_pts_tb, AV_TIME_BASE_Q); | |
1665 | #endif | |
bde0705c | 1666 | os = output_files[ost->file_index]; |
a700a6ae | 1667 | |
bde0705c | 1668 | /* set the input output pts pairs */ |
27e91f37 | 1669 | //ost->sync_ipts = (double)(ist->pts + input_files[ist->file_index].ts_offset - start_time)/ AV_TIME_BASE; |
bde0705c LW |
1670 | |
1671 | if (ost->encoding_needed) { | |
b926b628 | 1672 | av_assert0(ist->decoding_needed); |
bde0705c | 1673 | switch(ost->st->codec->codec_type) { |
72415b2a | 1674 | case AVMEDIA_TYPE_AUDIO: |
036c1382 | 1675 | do_audio_out(os, ost, ist, decoded_data_buf, decoded_data_size); |
bde0705c | 1676 | break; |
72415b2a | 1677 | case AVMEDIA_TYPE_VIDEO: |
46847a33 | 1678 | #if CONFIG_AVFILTER |
901ff511 | 1679 | if (ost->picref->video && !ost->frame_aspect_ratio) |
9d5fa618 | 1680 | ost->st->codec->sample_aspect_ratio = ost->picref->video->pixel_aspect; |
46847a33 | 1681 | #endif |
5e8d2e33 AK |
1682 | do_video_out(os, ost, ist, &picture, &frame_size, |
1683 | same_quality ? quality : ost->st->codec->global_quality); | |
b60d1379 | 1684 | if (vstats_filename && frame_size) |
bde0705c LW |
1685 | do_video_stats(os, ost, frame_size); |
1686 | break; | |
72415b2a | 1687 | case AVMEDIA_TYPE_SUBTITLE: |
bde0705c LW |
1688 | do_subtitle_out(os, ost, ist, &subtitle, |
1689 | pkt->pts); | |
1690 | break; | |
1691 | default: | |
0f4e8165 | 1692 | abort(); |
bde0705c LW |
1693 | } |
1694 | } else { | |
1695 | AVFrame avframe; //FIXME/XXX remove this | |
1696 | AVPacket opkt; | |
cdf38a17 MN |
1697 | int64_t ost_tb_start_time= av_rescale_q(start_time, AV_TIME_BASE_Q, ost->st->time_base); |
1698 | ||
bde0705c LW |
1699 | av_init_packet(&opkt); |
1700 | ||
cc947f04 | 1701 | if ((!ost->frame_number && !(pkt->flags & AV_PKT_FLAG_KEY)) && !copy_initial_nonkeyframes) |
9d5fa618 | 1702 | #if !CONFIG_AVFILTER |
7cacf1e8 | 1703 | continue; |
9d5fa618 MN |
1704 | #else |
1705 | goto cont; | |
1706 | #endif | |
7cacf1e8 | 1707 | |
bde0705c LW |
1708 | /* no reencoding needed : output the packet directly */ |
1709 | /* force the input stream PTS */ | |
1710 | ||
1711 | avcodec_get_frame_defaults(&avframe); | |
1712 | ost->st->codec->coded_frame= &avframe; | |
cc947f04 | 1713 | avframe.key_frame = pkt->flags & AV_PKT_FLAG_KEY; |
bde0705c | 1714 | |
72415b2a | 1715 | if(ost->st->codec->codec_type == AVMEDIA_TYPE_AUDIO) |
bde0705c | 1716 | audio_size += data_size; |
72415b2a | 1717 | else if (ost->st->codec->codec_type == AVMEDIA_TYPE_VIDEO) { |
bde0705c LW |
1718 | video_size += data_size; |
1719 | ost->sync_opts++; | |
1720 | } | |
1721 | ||
1722 | opkt.stream_index= ost->index; | |
1723 | if(pkt->pts != AV_NOPTS_VALUE) | |
cdf38a17 | 1724 | opkt.pts= av_rescale_q(pkt->pts, ist->st->time_base, ost->st->time_base) - ost_tb_start_time; |
bde0705c LW |
1725 | else |
1726 | opkt.pts= AV_NOPTS_VALUE; | |
1727 | ||
d2ce2f5e | 1728 | if (pkt->dts == AV_NOPTS_VALUE) |
181782ae | 1729 | opkt.dts = av_rescale_q(ist->pts, AV_TIME_BASE_Q, ost->st->time_base); |
d2ce2f5e BC |
1730 | else |
1731 | opkt.dts = av_rescale_q(pkt->dts, ist->st->time_base, ost->st->time_base); | |
cdf38a17 | 1732 | opkt.dts -= ost_tb_start_time; |
c0dd7b7c | 1733 | |
a03d59b7 | 1734 | opkt.duration = av_rescale_q(pkt->duration, ist->st->time_base, ost->st->time_base); |
bde0705c LW |
1735 | opkt.flags= pkt->flags; |
1736 | ||
1737 | //FIXME remove the following 2 lines they shall be replaced by the bitstream filters | |
5bfe91e6 MN |
1738 | if( ost->st->codec->codec_id != CODEC_ID_H264 |
1739 | && ost->st->codec->codec_id != CODEC_ID_MPEG1VIDEO | |
1740 | && ost->st->codec->codec_id != CODEC_ID_MPEG2VIDEO | |
1741 | ) { | |
cc947f04 | 1742 | if(av_parser_change(ist->st->parser, ost->st->codec, &opkt.data, &opkt.size, data_buf, data_size, pkt->flags & AV_PKT_FLAG_KEY)) |
9f907d85 | 1743 | opkt.destruct= av_destruct_packet; |
d310d56a BC |
1744 | } else { |
1745 | opkt.data = data_buf; | |
1746 | opkt.size = data_size; | |
1747 | } | |
bde0705c | 1748 | |
0b6d358a | 1749 | write_frame(os, &opkt, ost->st->codec, ost->bitstream_filters); |
bde0705c LW |
1750 | ost->st->codec->frame_number++; |
1751 | ost->frame_number++; | |
1752 | av_free_packet(&opkt); | |
a700a6ae | 1753 | } |
9d5fa618 MN |
1754 | #if CONFIG_AVFILTER |
1755 | cont: | |
1756 | frame_available = (ist->st->codec->codec_type == AVMEDIA_TYPE_VIDEO) && | |
1757 | ost->output_video_filter && avfilter_poll_frame(ost->output_video_filter->inputs[0]); | |
1758 | if (ost->picref) | |
1759 | avfilter_unref_buffer(ost->picref); | |
1760 | } | |
1761 | #endif | |
a700a6ae | 1762 | } |
bde0705c | 1763 | } |
46847a33 | 1764 | |
bde0705c LW |
1765 | av_free(buffer_to_free); |
1766 | /* XXX: allocate the subtitles in the codec ? */ | |
1767 | if (subtitle_to_free) { | |
1d6233d3 | 1768 | avsubtitle_free(subtitle_to_free); |
bde0705c | 1769 | subtitle_to_free = NULL; |
a700a6ae | 1770 | } |
bde0705c | 1771 | } |
a700a6ae | 1772 | discard_packet: |
6f824977 MN |
1773 | if (pkt == NULL) { |
1774 | /* EOF handling */ | |
115329f1 | 1775 | |
6f824977 MN |
1776 | for(i=0;i<nb_ostreams;i++) { |
1777 | ost = ost_table[i]; | |
1778 | if (ost->source_index == ist_index) { | |
01f4895c | 1779 | AVCodecContext *enc= ost->st->codec; |
6f824977 | 1780 | os = output_files[ost->file_index]; |
115329f1 | 1781 | |
72415b2a | 1782 | if(ost->st->codec->codec_type == AVMEDIA_TYPE_AUDIO && enc->frame_size <=1) |
6f824977 | 1783 | continue; |
72415b2a | 1784 | if(ost->st->codec->codec_type == AVMEDIA_TYPE_VIDEO && (os->oformat->flags & AVFMT_RAWPICTURE)) |
6f824977 MN |
1785 | continue; |
1786 | ||
1787 | if (ost->encoding_needed) { | |
1788 | for(;;) { | |
1789 | AVPacket pkt; | |
cef7cc72 | 1790 | int fifo_bytes; |
6f824977 MN |
1791 | av_init_packet(&pkt); |
1792 | pkt.stream_index= ost->index; | |
115329f1 | 1793 | |
01f4895c | 1794 | switch(ost->st->codec->codec_type) { |
72415b2a | 1795 | case AVMEDIA_TYPE_AUDIO: |
41dd680d | 1796 | fifo_bytes = av_fifo_size(ost->fifo); |
cef7cc72 JR |
1797 | ret = 0; |
1798 | /* encode any samples remaining in fifo */ | |
b10d7e4e | 1799 | if (fifo_bytes > 0) { |
f6d6783a | 1800 | int osize = av_get_bytes_per_sample(enc->sample_fmt); |
cef7cc72 | 1801 | int fs_tmp = enc->frame_size; |
b10d7e4e | 1802 | |
79c85beb | 1803 | av_fifo_generic_read(ost->fifo, audio_buf, fifo_bytes, NULL); |
b10d7e4e BC |
1804 | if (enc->codec->capabilities & CODEC_CAP_SMALL_LAST_FRAME) { |
1805 | enc->frame_size = fifo_bytes / (osize * enc->channels); | |
1806 | } else { /* pad */ | |
1807 | int frame_bytes = enc->frame_size*osize*enc->channels; | |
79c85beb | 1808 | if (allocated_audio_buf_size < frame_bytes) |
639e4ec8 | 1809 | ffmpeg_exit(1); |
14cf0fd2 | 1810 | generate_silence(audio_buf+fifo_bytes, enc->sample_fmt, frame_bytes - fifo_bytes); |
b10d7e4e BC |
1811 | } |
1812 | ||
79c85beb | 1813 | ret = avcodec_encode_audio(enc, bit_buffer, bit_buffer_size, (short *)audio_buf); |
7c8689ef BC |
1814 | pkt.duration = av_rescale((int64_t)enc->frame_size*ost->st->time_base.den, |
1815 | ost->st->time_base.num, enc->sample_rate); | |
cef7cc72 | 1816 | enc->frame_size = fs_tmp; |
9e0db7d5 MN |
1817 | } |
1818 | if(ret <= 0) { | |
cef7cc72 JR |
1819 | ret = avcodec_encode_audio(enc, bit_buffer, bit_buffer_size, NULL); |
1820 | } | |
528271ff MN |
1821 | if (ret < 0) { |
1822 | fprintf(stderr, "Audio encoding failed\n"); | |
639e4ec8 | 1823 | ffmpeg_exit(1); |
528271ff | 1824 | } |
6f824977 | 1825 | audio_size += ret; |
cc947f04 | 1826 | pkt.flags |= AV_PKT_FLAG_KEY; |
6f824977 | 1827 | break; |
72415b2a | 1828 | case AVMEDIA_TYPE_VIDEO: |
8a6cb114 | 1829 | ret = avcodec_encode_video(enc, bit_buffer, bit_buffer_size, NULL); |
528271ff MN |
1830 | if (ret < 0) { |
1831 | fprintf(stderr, "Video encoding failed\n"); | |
639e4ec8 | 1832 | ffmpeg_exit(1); |
528271ff | 1833 | } |
6f824977 MN |
1834 | video_size += ret; |
1835 | if(enc->coded_frame && enc->coded_frame->key_frame) | |
cc947f04 | 1836 | pkt.flags |= AV_PKT_FLAG_KEY; |
6f824977 MN |
1837 | if (ost->logfile && enc->stats_out) { |
1838 | fprintf(ost->logfile, "%s", enc->stats_out); | |
1839 | } | |
1840 | break; | |
1841 | default: | |
1842 | ret=-1; | |
1843 | } | |
115329f1 | 1844 | |
6f824977 MN |
1845 | if(ret<=0) |
1846 | break; | |
27537106 | 1847 | pkt.data= bit_buffer; |
6f824977 | 1848 | pkt.size= ret; |
e7902f20 | 1849 | if(enc->coded_frame && enc->coded_frame->pts != AV_NOPTS_VALUE) |
c0df9d75 | 1850 | pkt.pts= av_rescale_q(enc->coded_frame->pts, enc->time_base, ost->st->time_base); |
0b6d358a | 1851 | write_frame(os, &pkt, ost->st->codec, ost->bitstream_filters); |
6f824977 MN |
1852 | } |
1853 | } | |
1854 | } | |
1855 | } | |
1856 | } | |
115329f1 | 1857 | |
a700a6ae FB |
1858 | return 0; |
1859 | fail_decode: | |
1860 | return -1; | |
1861 | } | |
1862 | ||
6d1ba1ac LA |
1863 | static void print_sdp(AVFormatContext **avc, int n) |
1864 | { | |
1865 | char sdp[2048]; | |
1866 | ||
c3675dfe | 1867 | av_sdp_create(avc, n, sdp, sizeof(sdp)); |
6d1ba1ac | 1868 | printf("SDP:\n%s\n", sdp); |
536cd1db | 1869 | fflush(stdout); |
6d1ba1ac | 1870 | } |
a700a6ae | 1871 | |
7a39f142 AK |
1872 | static int copy_chapters(int infile, int outfile) |
1873 | { | |
07633154 | 1874 | AVFormatContext *is = input_files[infile].ctx; |
7a39f142 AK |
1875 | AVFormatContext *os = output_files[outfile]; |
1876 | int i; | |
1877 | ||
1878 | for (i = 0; i < is->nb_chapters; i++) { | |
1879 | AVChapter *in_ch = is->chapters[i], *out_ch; | |
27e91f37 | 1880 | int64_t ts_off = av_rescale_q(start_time - input_files[infile].ts_offset, |
7a39f142 AK |
1881 | AV_TIME_BASE_Q, in_ch->time_base); |
1882 | int64_t rt = (recording_time == INT64_MAX) ? INT64_MAX : | |
1883 | av_rescale_q(recording_time, AV_TIME_BASE_Q, in_ch->time_base); | |
1884 | ||
1885 | ||
1886 | if (in_ch->end < ts_off) | |
1887 | continue; | |
1888 | if (rt != INT64_MAX && in_ch->start > rt + ts_off) | |
1889 | break; | |
1890 | ||
1891 | out_ch = av_mallocz(sizeof(AVChapter)); | |
1892 | if (!out_ch) | |
1893 | return AVERROR(ENOMEM); | |
1894 | ||
1895 | out_ch->id = in_ch->id; | |
1896 | out_ch->time_base = in_ch->time_base; | |
1897 | out_ch->start = FFMAX(0, in_ch->start - ts_off); | |
1898 | out_ch->end = FFMIN(rt, in_ch->end - ts_off); | |
1899 | ||
d0abe80a | 1900 | if (metadata_chapters_autocopy) |
d2d67e42 | 1901 | av_dict_copy(&out_ch->metadata, in_ch->metadata, 0); |
7a39f142 AK |
1902 | |
1903 | os->nb_chapters++; | |
1904 | os->chapters = av_realloc(os->chapters, sizeof(AVChapter)*os->nb_chapters); | |
1905 | if (!os->chapters) | |
1906 | return AVERROR(ENOMEM); | |
1907 | os->chapters[os->nb_chapters - 1] = out_ch; | |
1908 | } | |
1909 | return 0; | |
1910 | } | |
1911 | ||
17c8cc55 | 1912 | static void parse_forced_key_frames(char *kf, OutputStream *ost, |
4ad08021 NG |
1913 | AVCodecContext *avctx) |
1914 | { | |
1915 | char *p; | |
1916 | int n = 1, i; | |
1917 | int64_t t; | |
1918 | ||
1919 | for (p = kf; *p; p++) | |
1920 | if (*p == ',') | |
1921 | n++; | |
1922 | ost->forced_kf_count = n; | |
1923 | ost->forced_kf_pts = av_malloc(sizeof(*ost->forced_kf_pts) * n); | |
1924 | if (!ost->forced_kf_pts) { | |
1925 | av_log(NULL, AV_LOG_FATAL, "Could not allocate forced key frames array.\n"); | |
1926 | ffmpeg_exit(1); | |
1927 | } | |
1928 | for (i = 0; i < n; i++) { | |
1929 | p = i ? strchr(p, ',') + 1 : kf; | |
1930 | t = parse_time_or_die("force_key_frames", p, 1); | |
1931 | ost->forced_kf_pts[i] = av_rescale_q(t, AV_TIME_BASE_Q, avctx->time_base); | |
1932 | } | |
1933 | } | |
1934 | ||
85f07f22 FB |
1935 | /* |
1936 | * The following code is the main loop of the file converter | |
1937 | */ | |
3aace1bc | 1938 | static int transcode(AVFormatContext **output_files, |
065a20cb | 1939 | int nb_output_files, |
17c8cc55 | 1940 | InputFile *input_files, |
065a20cb | 1941 | int nb_input_files, |
17c8cc55 | 1942 | StreamMap *stream_maps, int nb_stream_maps) |
85f07f22 | 1943 | { |
07633154 | 1944 | int ret = 0, i, j, k, n, nb_ostreams = 0; |
85f07f22 FB |
1945 | AVFormatContext *is, *os; |
1946 | AVCodecContext *codec, *icodec; | |
17c8cc55 AK |
1947 | OutputStream *ost, **ost_table = NULL; |
1948 | InputStream *ist; | |
002c95d7 | 1949 | char error[1024]; |
6d1ba1ac | 1950 | int want_sdp = 1; |
545465ec MN |
1951 | uint8_t no_packet[MAX_FILES]={0}; |
1952 | int no_packet_count=0; | |
bdc4796f | 1953 | |
07633154 AK |
1954 | if (rate_emu) |
1955 | for (i = 0; i < nb_input_streams; i++) | |
1956 | input_streams[i].start = av_gettime(); | |
85f07f22 FB |
1957 | |
1958 | /* output stream init */ | |
1959 | nb_ostreams = 0; | |
1960 | for(i=0;i<nb_output_files;i++) { | |
1961 | os = output_files[i]; | |
bb62d5c1 | 1962 | if (!os->nb_streams && !(os->oformat->flags & AVFMT_NOSTREAMS)) { |
610219a5 | 1963 | av_dump_format(output_files[i], i, output_files[i]->filename, 1); |
fc5d0db5 | 1964 | fprintf(stderr, "Output file #%d does not contain any stream\n", i); |
d62ccec8 JM |
1965 | ret = AVERROR(EINVAL); |
1966 | goto fail; | |
8a7bde1c | 1967 | } |
85f07f22 FB |
1968 | nb_ostreams += os->nb_streams; |
1969 | } | |
1970 | if (nb_stream_maps > 0 && nb_stream_maps != nb_ostreams) { | |
1971 | fprintf(stderr, "Number of stream maps must match number of output streams\n"); | |
d62ccec8 JM |
1972 | ret = AVERROR(EINVAL); |
1973 | goto fail; | |
85f07f22 FB |
1974 | } |
1975 | ||
bd073980 BF |
1976 | /* Sanity check the mapping args -- do the input files & streams exist? */ |
1977 | for(i=0;i<nb_stream_maps;i++) { | |
1978 | int fi = stream_maps[i].file_index; | |
1979 | int si = stream_maps[i].stream_index; | |
115329f1 | 1980 | |
bd073980 | 1981 | if (fi < 0 || fi > nb_input_files - 1 || |
2cf8355f | 1982 | si < 0 || si > input_files[fi].ctx->nb_streams - 1) { |
bd073980 | 1983 | fprintf(stderr,"Could not find input stream #%d.%d\n", fi, si); |
d62ccec8 JM |
1984 | ret = AVERROR(EINVAL); |
1985 | goto fail; | |
bd073980 | 1986 | } |
b4a3389e WG |
1987 | fi = stream_maps[i].sync_file_index; |
1988 | si = stream_maps[i].sync_stream_index; | |
1989 | if (fi < 0 || fi > nb_input_files - 1 || | |
2cf8355f | 1990 | si < 0 || si > input_files[fi].ctx->nb_streams - 1) { |
b4a3389e | 1991 | fprintf(stderr,"Could not find sync stream #%d.%d\n", fi, si); |
d62ccec8 JM |
1992 | ret = AVERROR(EINVAL); |
1993 | goto fail; | |
b4a3389e | 1994 | } |
bd073980 | 1995 | } |
115329f1 | 1996 | |
17c8cc55 | 1997 | ost_table = av_mallocz(sizeof(OutputStream *) * nb_ostreams); |
85f07f22 FB |
1998 | if (!ost_table) |
1999 | goto fail; | |
85f07f22 FB |
2000 | n = 0; |
2001 | for(k=0;k<nb_output_files;k++) { | |
2002 | os = output_files[k]; | |
de427ff4 | 2003 | for(i=0;i<os->nb_streams;i++,n++) { |
85f07f22 | 2004 | int found; |
9fdf4b58 | 2005 | ost = ost_table[n] = output_streams_for_file[k][i]; |
85f07f22 FB |
2006 | ost->st = os->streams[i]; |
2007 | if (nb_stream_maps > 0) { | |
07633154 | 2008 | ost->source_index = input_files[stream_maps[n].file_index].ist_index + |
de427ff4 | 2009 | stream_maps[n].stream_index; |
115329f1 | 2010 | |
bd073980 | 2011 | /* Sanity check that the stream types match */ |
07633154 | 2012 | if (input_streams[ost->source_index].st->codec->codec_type != ost->st->codec->codec_type) { |
150d5a25 | 2013 | int i= ost->file_index; |
610219a5 | 2014 | av_dump_format(output_files[i], i, output_files[i]->filename, 1); |
bd073980 | 2015 | fprintf(stderr, "Codec type mismatch for mapping #%d.%d -> #%d.%d\n", |
de427ff4 | 2016 | stream_maps[n].file_index, stream_maps[n].stream_index, |
bd073980 | 2017 | ost->file_index, ost->index); |
639e4ec8 | 2018 | ffmpeg_exit(1); |
bd073980 | 2019 | } |
115329f1 | 2020 | |
85f07f22 | 2021 | } else { |
247e3954 | 2022 | int best_nb_frames=-1; |
3f1710e7 RP |
2023 | /* get corresponding input stream index : we select the first one with the right type */ |
2024 | found = 0; | |
07633154 | 2025 | for (j = 0; j < nb_input_streams; j++) { |
3f1710e7 | 2026 | int skip=0; |
07633154 | 2027 | ist = &input_streams[j]; |
3f1710e7 RP |
2028 | if(opt_programid){ |
2029 | int pi,si; | |
07633154 | 2030 | AVFormatContext *f = input_files[ist->file_index].ctx; |
3f1710e7 RP |
2031 | skip=1; |
2032 | for(pi=0; pi<f->nb_programs; pi++){ | |
2033 | AVProgram *p= f->programs[pi]; | |
2034 | if(p->id == opt_programid) | |
2035 | for(si=0; si<p->nb_stream_indexes; si++){ | |
2036 | if(f->streams[ p->stream_index[si] ] == ist->st) | |
2037 | skip=0; | |
2038 | } | |
6d3d3b83 | 2039 | } |
3f1710e7 RP |
2040 | } |
2041 | if (ist->discard && ist->st->discard != AVDISCARD_ALL && !skip && | |
2042 | ist->st->codec->codec_type == ost->st->codec->codec_type) { | |
2043 | if(best_nb_frames < ist->st->codec_info_nb_frames){ | |
2044 | best_nb_frames= ist->st->codec_info_nb_frames; | |
2045 | ost->source_index = j; | |
2046 | found = 1; | |
85f07f22 FB |
2047 | } |
2048 | } | |
3f1710e7 | 2049 | } |
a15bc651 NS |
2050 | |
2051 | if (!found) { | |
2052 | if(! opt_programid) { | |
2053 | /* try again and reuse existing stream */ | |
07633154 AK |
2054 | for (j = 0; j < nb_input_streams; j++) { |
2055 | ist = &input_streams[j]; | |
6d3d3b83 MN |
2056 | if ( ist->st->codec->codec_type == ost->st->codec->codec_type |
2057 | && ist->st->discard != AVDISCARD_ALL) { | |
a15bc651 NS |
2058 | ost->source_index = j; |
2059 | found = 1; | |
2060 | } | |
2061 | } | |
50e143c4 | 2062 | } |
85f07f22 | 2063 | if (!found) { |
462cca10 | 2064 | int i= ost->file_index; |
610219a5 | 2065 | av_dump_format(output_files[i], i, output_files[i]->filename, 1); |
85f07f22 FB |
2066 | fprintf(stderr, "Could not find input stream matching output stream #%d.%d\n", |
2067 | ost->file_index, ost->index); | |
639e4ec8 | 2068 | ffmpeg_exit(1); |
85f07f22 FB |
2069 | } |
2070 | } | |
2071 | } | |
07633154 | 2072 | ist = &input_streams[ost->source_index]; |
85f07f22 | 2073 | ist->discard = 0; |
b4a3389e | 2074 | ost->sync_ist = (nb_stream_maps > 0) ? |
07633154 | 2075 | &input_streams[input_files[stream_maps[n].sync_file_index].ist_index + |
de427ff4 | 2076 | stream_maps[n].sync_stream_index] : ist; |
85f07f22 FB |
2077 | } |
2078 | } | |
2079 | ||
85f07f22 FB |
2080 | /* for each output stream, we compute the right encoding parameters */ |
2081 | for(i=0;i<nb_ostreams;i++) { | |
2082 | ost = ost_table[i]; | |
365515ac | 2083 | os = output_files[ost->file_index]; |
07633154 | 2084 | ist = &input_streams[ost->source_index]; |
85f07f22 | 2085 | |
01f4895c MN |
2086 | codec = ost->st->codec; |
2087 | icodec = ist->st->codec; | |
85f07f22 | 2088 | |
d0abe80a | 2089 | if (metadata_streams_autocopy) |
d2d67e42 AK |
2090 | av_dict_copy(&ost->st->metadata, ist->st->metadata, |
2091 | AV_DICT_DONT_OVERWRITE); | |
8e0d882b | 2092 | |
90c2295b | 2093 | ost->st->disposition = ist->st->disposition; |
a39b76ea | 2094 | codec->bits_per_raw_sample= icodec->bits_per_raw_sample; |
de961801 | 2095 | codec->chroma_sample_location = icodec->chroma_sample_location; |
90c2295b | 2096 | |
1629626f | 2097 | if (ost->st->stream_copy) { |
b659c8b4 LA |
2098 | uint64_t extra_size = (uint64_t)icodec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE; |
2099 | ||
2100 | if (extra_size > INT_MAX) | |
2101 | goto fail; | |
2102 | ||
1629626f FB |
2103 | /* if stream_copy is selected, no need to decode or encode */ |
2104 | codec->codec_id = icodec->codec_id; | |
2105 | codec->codec_type = icodec->codec_type; | |
365515ac MN |
2106 | |
2107 | if(!codec->codec_tag){ | |
2108 | if( !os->oformat->codec_tag | |
37ce3d6b | 2109 | || av_codec_get_id (os->oformat->codec_tag, icodec->codec_tag) == codec->codec_id |
365515ac MN |
2110 | || av_codec_get_tag(os->oformat->codec_tag, icodec->codec_id) <= 0) |
2111 | codec->codec_tag = icodec->codec_tag; | |
2112 | } | |
2113 | ||
1629626f | 2114 | codec->bit_rate = icodec->bit_rate; |
2dec2bb8 MN |
2115 | codec->rc_max_rate = icodec->rc_max_rate; |
2116 | codec->rc_buffer_size = icodec->rc_buffer_size; | |
b659c8b4 LA |
2117 | codec->extradata= av_mallocz(extra_size); |
2118 | if (!codec->extradata) | |
2119 | goto fail; | |
2120 | memcpy(codec->extradata, icodec->extradata, icodec->extradata_size); | |
dffbfd06 | 2121 | codec->extradata_size= icodec->extradata_size; |
59e2118e | 2122 | if(!copy_tb && av_q2d(icodec->time_base)*icodec->ticks_per_frame > av_q2d(ist->st->time_base) && av_q2d(ist->st->time_base) < 1.0/500){ |
d29de719 | 2123 | codec->time_base = icodec->time_base; |
3797c74b | 2124 | codec->time_base.num *= icodec->ticks_per_frame; |
8abcbf2d BC |
2125 | av_reduce(&codec->time_base.num, &codec->time_base.den, |
2126 | codec->time_base.num, codec->time_base.den, INT_MAX); | |
3797c74b | 2127 | }else |
d29de719 | 2128 | codec->time_base = ist->st->time_base; |
1629626f | 2129 | switch(codec->codec_type) { |
72415b2a | 2130 | case AVMEDIA_TYPE_AUDIO: |
d08e3e91 RP |
2131 | if(audio_volume != 256) { |
2132 | fprintf(stderr,"-acodec copy and -vol are incompatible (frames are not decoded)\n"); | |
639e4ec8 | 2133 | ffmpeg_exit(1); |
d08e3e91 | 2134 | } |
13367a46 | 2135 | codec->channel_layout = icodec->channel_layout; |
1629626f FB |
2136 | codec->sample_rate = icodec->sample_rate; |
2137 | codec->channels = icodec->channels; | |
0a3b0447 | 2138 | codec->frame_size = icodec->frame_size; |
34b47d7c | 2139 | codec->audio_service_type = icodec->audio_service_type; |
c1344911 | 2140 | codec->block_align= icodec->block_align; |
4efd6f58 MN |
2141 | if(codec->block_align == 1 && codec->codec_id == CODEC_ID_MP3) |
2142 | codec->block_align= 0; | |
bcbd328e MN |
2143 | if(codec->codec_id == CODEC_ID_AC3) |
2144 | codec->block_align= 0; | |
1629626f | 2145 | break; |
72415b2a | 2146 | case AVMEDIA_TYPE_VIDEO: |
9df3437f | 2147 | codec->pix_fmt = icodec->pix_fmt; |
1629626f FB |
2148 | codec->width = icodec->width; |
2149 | codec->height = icodec->height; | |
ff8cc24b | 2150 | codec->has_b_frames = icodec->has_b_frames; |
901ff511 BC |
2151 | if (!codec->sample_aspect_ratio.num) { |
2152 | codec->sample_aspect_ratio = | |
2153 | ost->st->sample_aspect_ratio = | |
2154 | ist->st->sample_aspect_ratio.num ? ist->st->sample_aspect_ratio : | |
2155 | ist->st->codec->sample_aspect_ratio.num ? | |
2156 | ist->st->codec->sample_aspect_ratio : (AVRational){0, 1}; | |
2157 | } | |
1629626f | 2158 | break; |
72415b2a | 2159 | case AVMEDIA_TYPE_SUBTITLE: |
d43b26ea DC |
2160 | codec->width = icodec->width; |
2161 | codec->height = icodec->height; | |
cf7fc795 | 2162 | break; |
e3b540b4 LB |
2163 | case AVMEDIA_TYPE_DATA: |
2164 | break; | |
1629626f | 2165 | default: |
0f4e8165 | 2166 | abort(); |
1629626f FB |
2167 | } |
2168 | } else { | |
62940bb4 AK |
2169 | if (!ost->enc) |
2170 | ost->enc = avcodec_find_encoder(ost->st->codec->codec_id); | |
1629626f | 2171 | switch(codec->codec_type) { |
72415b2a | 2172 | case AVMEDIA_TYPE_AUDIO: |
41dd680d MN |
2173 | ost->fifo= av_fifo_alloc(1024); |
2174 | if(!ost->fifo) | |
85f07f22 | 2175 | goto fail; |
5d6e4c16 | 2176 | ost->reformat_pair = MAKE_SFMT_PAIR(AV_SAMPLE_FMT_NONE,AV_SAMPLE_FMT_NONE); |
d7ee4402 AK |
2177 | if (!codec->sample_rate) { |
2178 | codec->sample_rate = icodec->sample_rate; | |
2179 | if (icodec->lowres) | |
2180 | codec->sample_rate >>= icodec->lowres; | |
2181 | } | |
62940bb4 | 2182 | choose_sample_rate(ost->st, ost->enc); |
d7ee4402 | 2183 | codec->time_base = (AVRational){1, codec->sample_rate}; |
8f3e9997 AK |
2184 | if (!codec->channels) |
2185 | codec->channels = icodec->channels; | |
2186 | if (av_get_channel_layout_nb_channels(codec->channel_layout) != codec->channels) | |
2187 | codec->channel_layout = 0; | |
2886f311 AÖ |
2188 | ost->audio_resample = codec->sample_rate != icodec->sample_rate || audio_sync_method > 1; |
2189 | icodec->request_channels = codec->channels; | |
85f07f22 FB |
2190 | ist->decoding_needed = 1; |
2191 | ost->encoding_needed = 1; | |
8afab686 SS |
2192 | ost->resample_sample_fmt = icodec->sample_fmt; |
2193 | ost->resample_sample_rate = icodec->sample_rate; | |
2194 | ost->resample_channels = icodec->channels; | |
1629626f | 2195 | break; |
72415b2a | 2196 | case AVMEDIA_TYPE_VIDEO: |
10de86b8 AK |
2197 | if (codec->pix_fmt == PIX_FMT_NONE) |
2198 | codec->pix_fmt = icodec->pix_fmt; | |
2199 | choose_pixel_fmt(ost->st, ost->enc); | |
2200 | ||
761c8c92 | 2201 | if (ost->st->codec->pix_fmt == PIX_FMT_NONE) { |
562f22a6 | 2202 | fprintf(stderr, "Video pixel format is unknown, stream cannot be encoded\n"); |
639e4ec8 | 2203 | ffmpeg_exit(1); |
761c8c92 | 2204 | } |
2b95602e SS |
2205 | ost->video_resample = codec->width != icodec->width || |
2206 | codec->height != icodec->height || | |
2207 | codec->pix_fmt != icodec->pix_fmt; | |
c3f11d19 | 2208 | if (ost->video_resample) { |
8a774d3d | 2209 | #if !CONFIG_AVFILTER |
2de28abb | 2210 | avcodec_get_frame_defaults(&ost->pict_tmp); |
9d58e0a9 BC |
2211 | if(avpicture_alloc((AVPicture*)&ost->pict_tmp, codec->pix_fmt, |
2212 | codec->width, codec->height)) { | |
eddc482d | 2213 | fprintf(stderr, "Cannot allocate temp picture, check pix fmt\n"); |
639e4ec8 | 2214 | ffmpeg_exit(1); |
eddc482d | 2215 | } |
18a54b04 | 2216 | ost->img_resample_ctx = sws_getContext( |
5879ea6d SS |
2217 | icodec->width, |
2218 | icodec->height, | |
18a54b04 | 2219 | icodec->pix_fmt, |
0c22311b MN |
2220 | codec->width, |
2221 | codec->height, | |
18a54b04 | 2222 | codec->pix_fmt, |
1435f2fa | 2223 | ost->sws_flags, NULL, NULL, NULL); |
0b50ac8a LA |
2224 | if (ost->img_resample_ctx == NULL) { |
2225 | fprintf(stderr, "Cannot get resampling context\n"); | |
639e4ec8 | 2226 | ffmpeg_exit(1); |
0b50ac8a | 2227 | } |
46847a33 | 2228 | #endif |
a39b76ea | 2229 | codec->bits_per_raw_sample= 0; |
85f07f22 | 2230 | } |
0b7ccad6 AK |
2231 | if (!codec->width || !codec->height) { |
2232 | codec->width = icodec->width; | |
2233 | codec->height = icodec->height; | |
2234 | } | |
5879ea6d SS |
2235 | ost->resample_height = icodec->height; |
2236 | ost->resample_width = icodec->width; | |
b83ccbff | 2237 | ost->resample_pix_fmt= icodec->pix_fmt; |
85f07f22 FB |
2238 | ost->encoding_needed = 1; |
2239 | ist->decoding_needed = 1; | |
46847a33 | 2240 | |
a6286bda AK |
2241 | if (!ost->frame_rate.num) |
2242 | ost->frame_rate = ist->st->r_frame_rate.num ? ist->st->r_frame_rate : (AVRational){25,1}; | |
62940bb4 AK |
2243 | if (ost->enc && ost->enc->supported_framerates && !force_fps) { |
2244 | int idx = av_find_nearest_q_idx(ost->frame_rate, ost->enc->supported_framerates); | |
2245 | ost->frame_rate = ost->enc->supported_framerates[idx]; | |
a6286bda AK |
2246 | } |
2247 | codec->time_base = (AVRational){ost->frame_rate.den, ost->frame_rate.num}; | |
2248 | ||
46847a33 | 2249 | #if CONFIG_AVFILTER |
0420bf09 | 2250 | if (configure_video_filters(ist, ost)) { |
46847a33 MN |
2251 | fprintf(stderr, "Error opening filters!\n"); |
2252 | exit(1); | |
2253 | } | |
2254 | #endif | |
1629626f | 2255 | break; |
72415b2a | 2256 | case AVMEDIA_TYPE_SUBTITLE: |
cf7fc795 FB |
2257 | ost->encoding_needed = 1; |
2258 | ist->decoding_needed = 1; | |
2259 | break; | |
1629626f | 2260 | default: |
0f4e8165 | 2261 | abort(); |
cf7fc795 | 2262 | break; |
85f07f22 | 2263 | } |
1629626f | 2264 | /* two pass mode */ |
115329f1 | 2265 | if (ost->encoding_needed && |
1629626f FB |
2266 | (codec->flags & (CODEC_FLAG_PASS1 | CODEC_FLAG_PASS2))) { |
2267 | char logfilename[1024]; | |
2268 | FILE *f; | |
115329f1 DB |
2269 | |
2270 | snprintf(logfilename, sizeof(logfilename), "%s-%d.log", | |
22730e87 SS |
2271 | pass_logfilename_prefix ? pass_logfilename_prefix : DEFAULT_PASS_LOGFILENAME_PREFIX, |
2272 | i); | |
1629626f | 2273 | if (codec->flags & CODEC_FLAG_PASS1) { |
c56e9e05 | 2274 | f = fopen(logfilename, "wb"); |
1629626f | 2275 | if (!f) { |
42d1d06e | 2276 | fprintf(stderr, "Cannot write log file '%s' for pass-1 encoding: %s\n", logfilename, strerror(errno)); |
639e4ec8 | 2277 | ffmpeg_exit(1); |
1629626f FB |
2278 | } |
2279 | ost->logfile = f; | |
2280 | } else { | |
458b062d SS |
2281 | char *logbuffer; |
2282 | size_t logbuffer_size; | |
2283 | if (read_file(logfilename, &logbuffer, &logbuffer_size) < 0) { | |
2284 | fprintf(stderr, "Error reading log file '%s' for pass-2 encoding\n", logfilename); | |
639e4ec8 | 2285 | ffmpeg_exit(1); |
1629626f | 2286 | } |
1629626f | 2287 | codec->stats_in = logbuffer; |
5abdb4b1 | 2288 | } |
5abdb4b1 FB |
2289 | } |
2290 | } | |
72415b2a | 2291 | if(codec->codec_type == AVMEDIA_TYPE_VIDEO){ |
8a6cb114 | 2292 | int size= codec->width * codec->height; |
c027e91a | 2293 | bit_buffer_size= FFMAX(bit_buffer_size, 6*size + 200); |
8a6cb114 | 2294 | } |
85f07f22 FB |
2295 | } |
2296 | ||
8a6cb114 MN |
2297 | if (!bit_buffer) |
2298 | bit_buffer = av_malloc(bit_buffer_size); | |
002c95d7 | 2299 | if (!bit_buffer) { |
50f3fabc MR |
2300 | fprintf(stderr, "Cannot allocate %d bytes output buffer\n", |
2301 | bit_buffer_size); | |
002c95d7 | 2302 | ret = AVERROR(ENOMEM); |
8a6cb114 | 2303 | goto fail; |
1629626f FB |
2304 | } |
2305 | ||
85f07f22 FB |
2306 | /* open each encoder */ |
2307 | for(i=0;i<nb_ostreams;i++) { | |
2308 | ost = ost_table[i]; | |
2309 | if (ost->encoding_needed) { | |
9446d759 | 2310 | AVCodec *codec = ost->enc; |
07633154 | 2311 | AVCodecContext *dec = input_streams[ost->source_index].st->codec; |
85f07f22 | 2312 | if (!codec) { |
f356fc57 BC |
2313 | snprintf(error, sizeof(error), "Encoder (codec id %d) not found for output stream #%d.%d", |
2314 | ost->st->codec->codec_id, ost->file_index, ost->index); | |
002c95d7 BC |
2315 | ret = AVERROR(EINVAL); |
2316 | goto dump_format; | |
85f07f22 | 2317 | } |
cb2c971d AJ |
2318 | if (dec->subtitle_header) { |
2319 | ost->st->codec->subtitle_header = av_malloc(dec->subtitle_header_size); | |
2320 | if (!ost->st->codec->subtitle_header) { | |
2321 | ret = AVERROR(ENOMEM); | |
2322 | goto dump_format; | |
2323 | } | |
2324 | memcpy(ost->st->codec->subtitle_header, dec->subtitle_header, dec->subtitle_header_size); | |
2325 | ost->st->codec->subtitle_header_size = dec->subtitle_header_size; | |
2326 | } | |
01f4895c | 2327 | if (avcodec_open(ost->st->codec, codec) < 0) { |
f356fc57 | 2328 | snprintf(error, sizeof(error), "Error while opening encoder for output stream #%d.%d - maybe incorrect parameters such as bit_rate, rate, width or height", |
85f07f22 | 2329 | ost->file_index, ost->index); |
002c95d7 BC |
2330 | ret = AVERROR(EINVAL); |
2331 | goto dump_format; | |
85f07f22 | 2332 | } |
01f4895c | 2333 | extra_size += ost->st->codec->extradata_size; |
85f07f22 FB |
2334 | } |
2335 | } | |
2336 | ||
2337 | /* open each decoder */ | |
07633154 AK |
2338 | for (i = 0; i < nb_input_streams; i++) { |
2339 | ist = &input_streams[i]; | |
85f07f22 | 2340 | if (ist->decoding_needed) { |
9e253c13 | 2341 | AVCodec *codec = ist->dec; |
6488cf9b | 2342 | if (!codec) |
fd2b356a | 2343 | codec = avcodec_find_decoder(ist->st->codec->codec_id); |
85f07f22 | 2344 | if (!codec) { |
f356fc57 | 2345 | snprintf(error, sizeof(error), "Decoder (codec id %d) not found for input stream #%d.%d", |
d2bc4da1 | 2346 | ist->st->codec->codec_id, ist->file_index, ist->st->index); |
002c95d7 BC |
2347 | ret = AVERROR(EINVAL); |
2348 | goto dump_format; | |
85f07f22 | 2349 | } |
bc778a0c JR |
2350 | |
2351 | /* update requested sample format for the decoder based on the | |
2352 | corresponding encoder sample format */ | |
2353 | for (j = 0; j < nb_ostreams; j++) { | |
2354 | ost = ost_table[j]; | |
2355 | if (ost->source_index == i) { | |
2356 | update_sample_fmt(ist->st->codec, codec, ost->st->codec); | |
2357 | break; | |
2358 | } | |
2359 | } | |
2360 | ||
01f4895c | 2361 | if (avcodec_open(ist->st->codec, codec) < 0) { |
f356fc57 | 2362 | snprintf(error, sizeof(error), "Error while opening decoder for input stream #%d.%d", |
d2bc4da1 | 2363 | ist->file_index, ist->st->index); |
002c95d7 BC |
2364 | ret = AVERROR(EINVAL); |
2365 | goto dump_format; | |
85f07f22 | 2366 | } |
72415b2a | 2367 | //if (ist->st->codec->codec_type == AVMEDIA_TYPE_VIDEO) |
01f4895c | 2368 | // ist->st->codec->flags |= CODEC_FLAG_REPEAT_FIELD; |
85f07f22 FB |
2369 | } |
2370 | } | |
2371 | ||
2372 | /* init pts */ | |
07633154 | 2373 | for (i = 0; i < nb_input_streams; i++) { |
b8c93c48 | 2374 | AVStream *st; |
07633154 | 2375 | ist = &input_streams[i]; |
b8c93c48 MN |
2376 | st= ist->st; |
2377 | ist->pts = st->avg_frame_rate.num ? - st->codec->has_b_frames*AV_TIME_BASE / av_q2d(st->avg_frame_rate) : 0; | |
48291040 | 2378 | ist->next_pts = AV_NOPTS_VALUE; |
0ff4f0c0 | 2379 | init_pts_correction(&ist->pts_ctx); |
ff4905a5 | 2380 | ist->is_start = 1; |
85f07f22 | 2381 | } |
c5e1e982 | 2382 | |
0a38bafd PB |
2383 | /* set meta data information from input file if required */ |
2384 | for (i=0;i<nb_meta_data_maps;i++) { | |
1829e195 | 2385 | AVFormatContext *files[2]; |
d2d67e42 | 2386 | AVDictionary **meta[2]; |
1829e195 | 2387 | int j; |
0a38bafd | 2388 | |
1829e195 AK |
2389 | #define METADATA_CHECK_INDEX(index, nb_elems, desc)\ |
2390 | if ((index) < 0 || (index) >= (nb_elems)) {\ | |
2391 | snprintf(error, sizeof(error), "Invalid %s index %d while processing metadata maps\n",\ | |
2392 | (desc), (index));\ | |
2393 | ret = AVERROR(EINVAL);\ | |
2394 | goto dump_format;\ | |
115329f1 DB |
2395 | } |
2396 | ||
1829e195 AK |
2397 | int out_file_index = meta_data_maps[i][0].file; |
2398 | int in_file_index = meta_data_maps[i][1].file; | |
7b393736 AK |
2399 | if (in_file_index < 0 || out_file_index < 0) |
2400 | continue; | |
1829e195 AK |
2401 | METADATA_CHECK_INDEX(out_file_index, nb_output_files, "output file") |
2402 | METADATA_CHECK_INDEX(in_file_index, nb_input_files, "input file") | |
0a38bafd | 2403 | |
1829e195 | 2404 | files[0] = output_files[out_file_index]; |
07633154 | 2405 | files[1] = input_files[in_file_index].ctx; |
1829e195 AK |
2406 | |
2407 | for (j = 0; j < 2; j++) { | |
17c8cc55 | 2408 | MetadataMap *map = &meta_data_maps[i][j]; |
1829e195 AK |
2409 | |
2410 | switch (map->type) { | |
2411 | case 'g': | |
2412 | meta[j] = &files[j]->metadata; | |
2413 | break; | |
2414 | case 's': | |
2415 | METADATA_CHECK_INDEX(map->index, files[j]->nb_streams, "stream") | |
2416 | meta[j] = &files[j]->streams[map->index]->metadata; | |
2417 | break; | |
2418 | case 'c': | |
2419 | METADATA_CHECK_INDEX(map->index, files[j]->nb_chapters, "chapter") | |
2420 | meta[j] = &files[j]->chapters[map->index]->metadata; | |
2421 | break; | |
2422 | case 'p': | |
2423 | METADATA_CHECK_INDEX(map->index, files[j]->nb_programs, "program") | |
2424 | meta[j] = &files[j]->programs[map->index]->metadata; | |
2425 | break; | |
2426 | } | |
2427 | } | |
a5926d85 | 2428 | |
d2d67e42 | 2429 | av_dict_copy(meta[0], *meta[1], AV_DICT_DONT_OVERWRITE); |
0a38bafd | 2430 | } |
115329f1 | 2431 | |
477b1aea AK |
2432 | /* copy global metadata by default */ |
2433 | if (metadata_global_autocopy) { | |
477b1aea | 2434 | |
8e8a3cc2 | 2435 | for (i = 0; i < nb_output_files; i++) |
d2d67e42 AK |
2436 | av_dict_copy(&output_files[i]->metadata, input_files[0].ctx->metadata, |
2437 | AV_DICT_DONT_OVERWRITE); | |
477b1aea AK |
2438 | } |
2439 | ||
91e96eba AK |
2440 | /* copy chapters according to chapter maps */ |
2441 | for (i = 0; i < nb_chapter_maps; i++) { | |
2442 | int infile = chapter_maps[i].in_file; | |
2443 | int outfile = chapter_maps[i].out_file; | |
2444 | ||
2445 | if (infile < 0 || outfile < 0) | |
2446 | continue; | |
2447 | if (infile >= nb_input_files) { | |
2448 | snprintf(error, sizeof(error), "Invalid input file index %d in chapter mapping.\n", infile); | |
2449 | ret = AVERROR(EINVAL); | |
2450 | goto dump_format; | |
2451 | } | |
2452 | if (outfile >= nb_output_files) { | |
2453 | snprintf(error, sizeof(error), "Invalid output file index %d in chapter mapping.\n",outfile); | |
2454 | ret = AVERROR(EINVAL); | |
2455 | goto dump_format; | |
2456 | } | |
2457 | copy_chapters(infile, outfile); | |
2458 | } | |
2459 | ||
7a39f142 | 2460 | /* copy chapters from the first input file that has them*/ |
91e96eba | 2461 | if (!nb_chapter_maps) |
a9c2bf9d | 2462 | for (i = 0; i < nb_input_files; i++) { |
07633154 | 2463 | if (!input_files[i].ctx->nb_chapters) |
a9c2bf9d | 2464 | continue; |
7a39f142 | 2465 | |
a9c2bf9d AK |
2466 | for (j = 0; j < nb_output_files; j++) |
2467 | if ((ret = copy_chapters(i, j)) < 0) | |
2468 | goto dump_format; | |
09f47fa4 | 2469 | break; |
a9c2bf9d | 2470 | } |
7a39f142 | 2471 | |
85f07f22 FB |
2472 | /* open files and write file headers */ |
2473 | for(i=0;i<nb_output_files;i++) { | |
2474 | os = output_files[i]; | |
8035f429 | 2475 | if (avformat_write_header(os, &output_opts[i]) < 0) { |
002c95d7 | 2476 | snprintf(error, sizeof(error), "Could not write header for output file #%d (incorrect codec parameters ?)", i); |
8fa36ae0 | 2477 | ret = AVERROR(EINVAL); |
002c95d7 | 2478 | goto dump_format; |
a38469e1 | 2479 | } |
8035f429 | 2480 | assert_avoptions(output_opts[i]); |
6d1ba1ac LA |
2481 | if (strcmp(output_files[i]->oformat->name, "rtp")) { |
2482 | want_sdp = 0; | |
2483 | } | |
2484 | } | |
002c95d7 BC |
2485 | |
2486 | dump_format: | |
2487 | /* dump the file output parameters - cannot be done before in case | |
2488 | of stream copy */ | |
2489 | for(i=0;i<nb_output_files;i++) { | |
610219a5 | 2490 | av_dump_format(output_files[i], i, output_files[i]->filename, 1); |
002c95d7 BC |
2491 | } |
2492 | ||
2493 | /* dump the stream mapping */ | |
2494 | if (verbose >= 0) { | |
2495 | fprintf(stderr, "Stream mapping:\n"); | |
2496 | for(i=0;i<nb_ostreams;i++) { | |
2497 | ost = ost_table[i]; | |
2498 | fprintf(stderr, " Stream #%d.%d -> #%d.%d", | |
07633154 AK |
2499 | input_streams[ost->source_index].file_index, |
2500 | input_streams[ost->source_index].st->index, | |
002c95d7 BC |
2501 | ost->file_index, |
2502 | ost->index); | |
07633154 | 2503 | if (ost->sync_ist != &input_streams[ost->source_index]) |
002c95d7 BC |
2504 | fprintf(stderr, " [sync #%d.%d]", |
2505 | ost->sync_ist->file_index, | |
d2bc4da1 | 2506 | ost->sync_ist->st->index); |
002c95d7 BC |
2507 | fprintf(stderr, "\n"); |
2508 | } | |
2509 | } | |
2510 | ||
2511 | if (ret) { | |
2512 | fprintf(stderr, "%s\n", error); | |
2513 | goto fail; | |
2514 | } | |
2515 | ||
6d1ba1ac LA |
2516 | if (want_sdp) { |
2517 | print_sdp(output_files, nb_output_files); | |
85f07f22 FB |
2518 | } |
2519 | ||
8fb566fd | 2520 | if (verbose >= 0) |
cb48e245 | 2521 | fprintf(stderr, "Press ctrl-c to stop encoding\n"); |
a38469e1 FB |
2522 | term_init(); |
2523 | ||
7fc98937 | 2524 | timer_start = av_gettime(); |
51bd4565 | 2525 | |
d9a916e2 | 2526 | for(; received_sigterm == 0;) { |
85f07f22 FB |
2527 | int file_index, ist_index; |
2528 | AVPacket pkt; | |
a9aeda81 PB |
2529 | double ipts_min; |
2530 | double opts_min; | |
23ffe323 | 2531 | |
85f07f22 | 2532 | redo: |
a9aeda81 PB |
2533 | ipts_min= 1e100; |
2534 | opts_min= 1e100; | |
a38469e1 | 2535 | |
ec5517d5 FB |
2536 | /* select the stream that we must read now by looking at the |
2537 | smallest output pts */ | |
85f07f22 | 2538 | file_index = -1; |
ec5517d5 | 2539 | for(i=0;i<nb_ostreams;i++) { |
23ffe323 | 2540 | double ipts, opts; |
ec5517d5 FB |
2541 | ost = ost_table[i]; |
2542 | os = output_files[ost->file_index]; | |
07633154 | 2543 | ist = &input_streams[ost->source_index]; |
55a7e946 | 2544 | if(ist->is_past_recording_time || no_packet[ist->file_index]) |
545465ec | 2545 | continue; |
c0df9d75 | 2546 | opts = ost->st->pts.val * av_q2d(ost->st->time_base); |
23ffe323 | 2547 | ipts = (double)ist->pts; |
07633154 | 2548 | if (!input_files[ist->file_index].eof_reached){ |
23ffe323 MN |
2549 | if(ipts < ipts_min) { |
2550 | ipts_min = ipts; | |
2551 | if(input_sync ) file_index = ist->file_index; | |
2552 | } | |
2553 | if(opts < opts_min) { | |
2554 | opts_min = opts; | |
2555 | if(!input_sync) file_index = ist->file_index; | |
2556 | } | |
85f07f22 | 2557 | } |
01f4895c | 2558 | if(ost->frame_number >= max_frames[ost->st->codec->codec_type]){ |
7ca60994 MN |
2559 | file_index= -1; |
2560 | break; | |
2561 | } | |
85f07f22 FB |
2562 | } |
2563 | /* if none, if is finished */ | |
51bd4565 | 2564 | if (file_index < 0) { |
545465ec MN |
2565 | if(no_packet_count){ |
2566 | no_packet_count=0; | |
2567 | memset(no_packet, 0, sizeof(no_packet)); | |
d61f30a7 | 2568 | usleep(10000); |
545465ec MN |
2569 | continue; |
2570 | } | |
85f07f22 | 2571 | break; |
ec5517d5 FB |
2572 | } |
2573 | ||
b6e16b86 | 2574 | /* finish if limit size exhausted */ |
a2704c97 | 2575 | if (limit_filesize != 0 && limit_filesize <= avio_tell(output_files[0]->pb)) |
b6e16b86 C |
2576 | break; |
2577 | ||
254abc2e | 2578 | /* read a frame from it and output it in the fifo */ |
07633154 | 2579 | is = input_files[file_index].ctx; |
ad51c68c | 2580 | ret= av_read_frame(is, &pkt); |
b9edbe99 | 2581 | if(ret == AVERROR(EAGAIN)){ |
545465ec MN |
2582 | no_packet[file_index]=1; |
2583 | no_packet_count++; | |
ad51c68c | 2584 | continue; |
545465ec | 2585 | } |
ad51c68c | 2586 | if (ret < 0) { |
07633154 | 2587 | input_files[file_index].eof_reached = 1; |
d24ce947 AB |
2588 | if (opt_shortest) |
2589 | break; | |
2590 | else | |
2591 | continue; | |
85f07f22 | 2592 | } |
303e50e6 | 2593 | |
545465ec MN |
2594 | no_packet_count=0; |
2595 | memset(no_packet, 0, sizeof(no_packet)); | |
2596 | ||
254abc2e | 2597 | if (do_pkt_dump) { |
5e33e7bd MS |
2598 | av_pkt_dump_log2(NULL, AV_LOG_DEBUG, &pkt, do_hex_dump, |
2599 | is->streams[pkt.stream_index]); | |
817b23ff | 2600 | } |
79fdaa4c FB |
2601 | /* the following test is needed in case new streams appear |
2602 | dynamically in stream : we ignore them */ | |
2cf8355f | 2603 | if (pkt.stream_index >= input_files[file_index].ctx->nb_streams) |
bf5af568 | 2604 | goto discard_packet; |
07633154 AK |
2605 | ist_index = input_files[file_index].ist_index + pkt.stream_index; |
2606 | ist = &input_streams[ist_index]; | |
bf5af568 FB |
2607 | if (ist->discard) |
2608 | goto discard_packet; | |
85f07f22 | 2609 | |
fec401f7 | 2610 | if (pkt.dts != AV_NOPTS_VALUE) |
27e91f37 | 2611 | pkt.dts += av_rescale_q(input_files[ist->file_index].ts_offset, AV_TIME_BASE_Q, ist->st->time_base); |
fec401f7 | 2612 | if (pkt.pts != AV_NOPTS_VALUE) |
27e91f37 | 2613 | pkt.pts += av_rescale_q(input_files[ist->file_index].ts_offset, AV_TIME_BASE_Q, ist->st->time_base); |
fec401f7 | 2614 | |
2c6958aa AJ |
2615 | if (pkt.stream_index < nb_input_files_ts_scale[file_index] |
2616 | && input_files_ts_scale[file_index][pkt.stream_index]){ | |
8833f375 MN |
2617 | if(pkt.pts != AV_NOPTS_VALUE) |
2618 | pkt.pts *= input_files_ts_scale[file_index][pkt.stream_index]; | |
2619 | if(pkt.dts != AV_NOPTS_VALUE) | |
2620 | pkt.dts *= input_files_ts_scale[file_index][pkt.stream_index]; | |
2621 | } | |
2622 | ||
27e91f37 | 2623 | // fprintf(stderr, "next:%"PRId64" dts:%"PRId64" off:%"PRId64" %d\n", ist->next_pts, pkt.dts, input_files[ist->file_index].ts_offset, ist->st->codec->codec_type); |
965530e1 MN |
2624 | if (pkt.dts != AV_NOPTS_VALUE && ist->next_pts != AV_NOPTS_VALUE |
2625 | && (is->iformat->flags & AVFMT_TS_DISCONT)) { | |
81d6d520 MN |
2626 | int64_t pkt_dts= av_rescale_q(pkt.dts, ist->st->time_base, AV_TIME_BASE_Q); |
2627 | int64_t delta= pkt_dts - ist->next_pts; | |
2628 | if((FFABS(delta) > 1LL*dts_delta_threshold*AV_TIME_BASE || pkt_dts+1<ist->pts)&& !copy_ts){ | |
27e91f37 | 2629 | input_files[ist->file_index].ts_offset -= delta; |
72bd8100 | 2630 | if (verbose > 2) |
27e91f37 AK |
2631 | fprintf(stderr, "timestamp discontinuity %"PRId64", new offset= %"PRId64"\n", |
2632 | delta, input_files[ist->file_index].ts_offset); | |
fec401f7 MN |
2633 | pkt.dts-= av_rescale_q(delta, AV_TIME_BASE_Q, ist->st->time_base); |
2634 | if(pkt.pts != AV_NOPTS_VALUE) | |
2635 | pkt.pts-= av_rescale_q(delta, AV_TIME_BASE_Q, ist->st->time_base); | |
72bd8100 MN |
2636 | } |
2637 | } | |
2638 | ||
f8ccf720 | 2639 | /* finish if recording time exhausted */ |
6abda15f FC |
2640 | if (recording_time != INT64_MAX && |
2641 | av_compare_ts(pkt.pts, ist->st->time_base, recording_time + start_time, (AVRational){1, 1000000}) >= 0) { | |
55a7e946 | 2642 | ist->is_past_recording_time = 1; |
f8ccf720 | 2643 | goto discard_packet; |
55a7e946 | 2644 | } |
f8ccf720 | 2645 | |
d2bc4da1 | 2646 | //fprintf(stderr,"read #%d.%d size=%d\n", ist->file_index, ist->st->index, pkt.size); |
a700a6ae | 2647 | if (output_packet(ist, ist_index, ost_table, nb_ostreams, &pkt) < 0) { |
d8019eb5 AD |
2648 | |
2649 | if (verbose >= 0) | |
2650 | fprintf(stderr, "Error while decoding stream #%d.%d\n", | |
d2bc4da1 | 2651 | ist->file_index, ist->st->index); |
f2abc559 | 2652 | if (exit_on_error) |
639e4ec8 | 2653 | ffmpeg_exit(1); |
a700a6ae FB |
2654 | av_free_packet(&pkt); |
2655 | goto redo; | |
fe08925f | 2656 | } |
cf7fc795 | 2657 | |
bf5af568 | 2658 | discard_packet: |
85f07f22 | 2659 | av_free_packet(&pkt); |
115329f1 | 2660 | |
ec5517d5 FB |
2661 | /* dump report by using the output first video and audio streams */ |
2662 | print_report(output_files, ost_table, nb_ostreams, 0); | |
85f07f22 | 2663 | } |
a700a6ae FB |
2664 | |
2665 | /* at the end of stream, we must flush the decoder buffers */ | |
07633154 AK |
2666 | for (i = 0; i < nb_input_streams; i++) { |
2667 | ist = &input_streams[i]; | |
a700a6ae FB |
2668 | if (ist->decoding_needed) { |
2669 | output_packet(ist, i, ost_table, nb_ostreams, NULL); | |
2670 | } | |
2671 | } | |
2672 | ||
a38469e1 | 2673 | term_exit(); |
85f07f22 | 2674 | |
c4e37247 MN |
2675 | /* write the trailer if needed and close file */ |
2676 | for(i=0;i<nb_output_files;i++) { | |
2677 | os = output_files[i]; | |
2678 | av_write_trailer(os); | |
2679 | } | |
2680 | ||
37f5cd5a MN |
2681 | /* dump report by using the first video and audio streams */ |
2682 | print_report(output_files, ost_table, nb_ostreams, 1); | |
2683 | ||
85f07f22 FB |
2684 | /* close each encoder */ |
2685 | for(i=0;i<nb_ostreams;i++) { | |
2686 | ost = ost_table[i]; | |
2687 | if (ost->encoding_needed) { | |
01f4895c MN |
2688 | av_freep(&ost->st->codec->stats_in); |
2689 | avcodec_close(ost->st->codec); | |
85f07f22 | 2690 | } |
9d5fa618 MN |
2691 | #if CONFIG_AVFILTER |
2692 | avfilter_graph_free(&ost->graph); | |
2693 | #endif | |
85f07f22 | 2694 | } |
115329f1 | 2695 | |
85f07f22 | 2696 | /* close each decoder */ |
07633154 AK |
2697 | for (i = 0; i < nb_input_streams; i++) { |
2698 | ist = &input_streams[i]; | |
85f07f22 | 2699 | if (ist->decoding_needed) { |
01f4895c | 2700 | avcodec_close(ist->st->codec); |
85f07f22 FB |
2701 | } |
2702 | } | |
85f07f22 | 2703 | |
85f07f22 | 2704 | /* finished ! */ |
00b7fbdc | 2705 | ret = 0; |
115329f1 | 2706 | |
002c95d7 | 2707 | fail: |
83b07470 | 2708 | av_freep(&bit_buffer); |
bdc4796f | 2709 | |
85f07f22 FB |
2710 | if (ost_table) { |
2711 | for(i=0;i<nb_ostreams;i++) { | |
2712 | ost = ost_table[i]; | |
2713 | if (ost) { | |
b659c8b4 LA |
2714 | if (ost->st->stream_copy) |
2715 | av_freep(&ost->st->codec->extradata); | |
5abdb4b1 FB |
2716 | if (ost->logfile) { |
2717 | fclose(ost->logfile); | |
2718 | ost->logfile = NULL; | |
2719 | } | |
41dd680d | 2720 | av_fifo_free(ost->fifo); /* works even if fifo is not |
f5a478f6 | 2721 | initialized but set to zero */ |
cb2c971d | 2722 | av_freep(&ost->st->codec->subtitle_header); |
0f1578af | 2723 | av_free(ost->pict_tmp.data[0]); |
4ad08021 | 2724 | av_free(ost->forced_kf_pts); |
85f07f22 | 2725 | if (ost->video_resample) |
18a54b04 | 2726 | sws_freeContext(ost->img_resample_ctx); |
8e4270c5 | 2727 | if (ost->resample) |
85f07f22 | 2728 | audio_resample_close(ost->resample); |
a79db0f7 PR |
2729 | if (ost->reformat_ctx) |
2730 | av_audio_convert_free(ost->reformat_ctx); | |
0f1578af | 2731 | av_free(ost); |
85f07f22 FB |
2732 | } |
2733 | } | |
0f1578af | 2734 | av_free(ost_table); |
85f07f22 FB |
2735 | } |
2736 | return ret; | |
85f07f22 FB |
2737 | } |
2738 | ||
26513856 | 2739 | static int opt_format(const char *opt, const char *arg) |
85f07f22 | 2740 | { |
a5abfd8f | 2741 | last_asked_format = arg; |
26513856 | 2742 | return 0; |
85f07f22 FB |
2743 | } |
2744 | ||
26513856 | 2745 | static int opt_video_rc_override_string(const char *opt, const char *arg) |
ac2830ec MN |
2746 | { |
2747 | video_rc_override_string = arg; | |
26513856 | 2748 | return 0; |
ac2830ec MN |
2749 | } |
2750 | ||
c48da33c | 2751 | static int opt_me_threshold(const char *opt, const char *arg) |
f4f3223f | 2752 | { |
c48da33c SS |
2753 | me_threshold = parse_number_or_die(opt, arg, OPT_INT64, INT_MIN, INT_MAX); |
2754 | return 0; | |
f4f3223f MN |
2755 | } |
2756 | ||
c48da33c | 2757 | static int opt_verbose(const char *opt, const char *arg) |
f068206e | 2758 | { |
46eab093 | 2759 | verbose = parse_number_or_die(opt, arg, OPT_INT64, -10, 10); |
c48da33c | 2760 | return 0; |
f068206e BE |
2761 | } |
2762 | ||
2fc3866d | 2763 | static int opt_frame_rate(const char *opt, const char *arg) |
85f07f22 | 2764 | { |
126b638e | 2765 | if (av_parse_video_rate(&frame_rate, arg) < 0) { |
2fc3866d | 2766 | fprintf(stderr, "Incorrect value for %s: %s\n", opt, arg); |
639e4ec8 | 2767 | ffmpeg_exit(1); |
445f1b83 | 2768 | } |
2fc3866d | 2769 | return 0; |
85f07f22 FB |
2770 | } |
2771 | ||
ebde2a2c | 2772 | static int opt_bitrate(const char *opt, const char *arg) |
1b1656c6 | 2773 | { |
72415b2a | 2774 | int codec_type = opt[0]=='a' ? AVMEDIA_TYPE_AUDIO : AVMEDIA_TYPE_VIDEO; |
1b1656c6 RP |
2775 | |
2776 | opt_default(opt, arg); | |
2777 | ||
636f1c4c | 2778 | if (av_get_int(avcodec_opts[codec_type], "b", NULL) < 1000) |
1b1656c6 | 2779 | fprintf(stderr, "WARNING: The bitrate parameter is set too low. It takes bits/s as argument, not kbits/s\n"); |
ebde2a2c MH |
2780 | |
2781 | return 0; | |
1b1656c6 RP |
2782 | } |
2783 | ||
5879ea6d | 2784 | static int opt_frame_crop(const char *opt, const char *arg) |
ab6d194a | 2785 | { |
5879ea6d SS |
2786 | fprintf(stderr, "Option '%s' has been removed, use the crop filter instead\n", opt); |
2787 | return AVERROR(EINVAL); | |
ab6d194a MN |
2788 | } |
2789 | ||
26513856 | 2790 | static int opt_frame_size(const char *opt, const char *arg) |
85f07f22 | 2791 | { |
126b638e | 2792 | if (av_parse_video_size(&frame_width, &frame_height, arg) < 0) { |
85f07f22 | 2793 | fprintf(stderr, "Incorrect frame size\n"); |
26513856 | 2794 | return AVERROR(EINVAL); |
85f07f22 | 2795 | } |
26513856 | 2796 | return 0; |
85f07f22 FB |
2797 | } |
2798 | ||
9f434b65 SS |
2799 | static int opt_pad(const char *opt, const char *arg) { |
2800 | fprintf(stderr, "Option '%s' has been removed, use the pad filter instead\n", opt); | |
2801 | return -1; | |
1ff93ffc TK |
2802 | } |
2803 | ||
26513856 | 2804 | static int opt_frame_pix_fmt(const char *opt, const char *arg) |
63167088 | 2805 | { |
90da2b50 | 2806 | if (strcmp(arg, "list")) { |
718c7b18 | 2807 | frame_pix_fmt = av_get_pix_fmt(arg); |
90da2b50 SS |
2808 | if (frame_pix_fmt == PIX_FMT_NONE) { |
2809 | fprintf(stderr, "Unknown pixel format requested: %s\n", arg); | |
26513856 | 2810 | return AVERROR(EINVAL); |
90da2b50 SS |
2811 | } |
2812 | } else { | |
cedac882 | 2813 | show_pix_fmts(); |
639e4ec8 | 2814 | ffmpeg_exit(0); |
c3b95b1d | 2815 | } |
26513856 | 2816 | return 0; |
63167088 RS |
2817 | } |
2818 | ||
26513856 | 2819 | static int opt_frame_aspect_ratio(const char *opt, const char *arg) |
5fe03e38 RS |
2820 | { |
2821 | int x = 0, y = 0; | |
2822 | double ar = 0; | |
2823 | const char *p; | |
815f98cc | 2824 | char *end; |
115329f1 | 2825 | |
5fe03e38 RS |
2826 | p = strchr(arg, ':'); |
2827 | if (p) { | |
815f98cc AJ |
2828 | x = strtol(arg, &end, 10); |
2829 | if (end == p) | |
2830 | y = strtol(end+1, &end, 10); | |
bb270c08 DB |
2831 | if (x > 0 && y > 0) |
2832 | ar = (double)x / (double)y; | |
5fe03e38 | 2833 | } else |
815f98cc | 2834 | ar = strtod(arg, NULL); |
5fe03e38 RS |
2835 | |
2836 | if (!ar) { | |
2837 | fprintf(stderr, "Incorrect aspect ratio specification.\n"); | |
26513856 | 2838 | return AVERROR(EINVAL); |
5fe03e38 RS |
2839 | } |
2840 | frame_aspect_ratio = ar; | |
26513856 | 2841 | return 0; |
5fe03e38 RS |
2842 | } |
2843 | ||
a5926d85 AJ |
2844 | static int opt_metadata(const char *opt, const char *arg) |
2845 | { | |
2846 | char *mid= strchr(arg, '='); | |
2847 | ||
2848 | if(!mid){ | |
2849 | fprintf(stderr, "Missing =\n"); | |
639e4ec8 | 2850 | ffmpeg_exit(1); |
a5926d85 AJ |
2851 | } |
2852 | *mid++= 0; | |
2853 | ||
d2d67e42 | 2854 | av_dict_set(&metadata, arg, mid, 0); |
a5926d85 AJ |
2855 | |
2856 | return 0; | |
2857 | } | |
2858 | ||
dbe94539 | 2859 | static int opt_qscale(const char *opt, const char *arg) |
85f07f22 | 2860 | { |
dbe94539 SS |
2861 | video_qscale = parse_number_or_die(opt, arg, OPT_FLOAT, 0, 255); |
2862 | if (video_qscale == 0) { | |
4e64bead | 2863 | fprintf(stderr, "qscale must be > 0.0 and <= 255\n"); |
dbe94539 | 2864 | return AVERROR(EINVAL); |
85f07f22 | 2865 | } |
dbe94539 | 2866 | return 0; |
85f07f22 FB |
2867 | } |
2868 | ||
dbe94539 | 2869 | static int opt_top_field_first(const char *opt, const char *arg) |
bb198e19 | 2870 | { |
dbe94539 SS |
2871 | top_field_first = parse_number_or_die(opt, arg, OPT_INT, 0, 1); |
2872 | return 0; | |
bb198e19 MN |
2873 | } |
2874 | ||
d18811bb | 2875 | static int opt_thread_count(const char *opt, const char *arg) |
9c3d33d6 | 2876 | { |
d18811bb | 2877 | thread_count= parse_number_or_die(opt, arg, OPT_INT64, 0, INT_MAX); |
b250f9c6 | 2878 | #if !HAVE_THREADS |
d8019eb5 AD |
2879 | if (verbose >= 0) |
2880 | fprintf(stderr, "Warning: not compiled with thread support, using thread emulation\n"); | |
842b556a | 2881 | #endif |
d18811bb | 2882 | return 0; |
9c3d33d6 MN |
2883 | } |
2884 | ||
26513856 | 2885 | static int opt_audio_sample_fmt(const char *opt, const char *arg) |
ce1ee094 | 2886 | { |
6c18f1cd | 2887 | if (strcmp(arg, "list")) { |
ba7d6e79 | 2888 | audio_sample_fmt = av_get_sample_fmt(arg); |
6c18f1cd SS |
2889 | if (audio_sample_fmt == AV_SAMPLE_FMT_NONE) { |
2890 | av_log(NULL, AV_LOG_ERROR, "Invalid sample format '%s'\n", arg); | |
26513856 | 2891 | return AVERROR(EINVAL); |
6c18f1cd SS |
2892 | } |
2893 | } else { | |
5d2c6f42 SS |
2894 | int i; |
2895 | char fmt_str[128]; | |
2896 | for (i = -1; i < AV_SAMPLE_FMT_NB; i++) | |
2897 | printf("%s\n", av_get_sample_fmt_string(fmt_str, sizeof(fmt_str), i)); | |
639e4ec8 | 2898 | ffmpeg_exit(0); |
ce1ee094 | 2899 | } |
26513856 | 2900 | return 0; |
ce1ee094 PR |
2901 | } |
2902 | ||
c48da33c | 2903 | static int opt_audio_rate(const char *opt, const char *arg) |
85f07f22 | 2904 | { |
c48da33c SS |
2905 | audio_sample_rate = parse_number_or_die(opt, arg, OPT_INT64, 0, INT_MAX); |
2906 | return 0; | |
85f07f22 FB |
2907 | } |
2908 | ||
c48da33c | 2909 | static int opt_audio_channels(const char *opt, const char *arg) |
85f07f22 | 2910 | { |
c48da33c SS |
2911 | audio_channels = parse_number_or_die(opt, arg, OPT_INT64, 0, INT_MAX); |
2912 | return 0; | |
85f07f22 FB |
2913 | } |
2914 | ||
dbe94539 | 2915 | static int opt_video_channel(const char *opt, const char *arg) |
a5df11ab | 2916 | { |
55ba12e3 AK |
2917 | av_log(NULL, AV_LOG_WARNING, "This option is deprecated, use -channel.\n"); |
2918 | opt_default("channel", arg); | |
dbe94539 | 2919 | return 0; |
a5df11ab FB |
2920 | } |
2921 | ||
26513856 | 2922 | static int opt_video_standard(const char *opt, const char *arg) |
e3ee3283 | 2923 | { |
55ba12e3 AK |
2924 | av_log(NULL, AV_LOG_WARNING, "This option is deprecated, use -standard.\n"); |
2925 | opt_default("standard", arg); | |
26513856 | 2926 | return 0; |
e3ee3283 AB |
2927 | } |
2928 | ||
26513856 | 2929 | static int opt_codec(int *pstream_copy, char **pcodec_name, |
cf7fc795 | 2930 | int codec_type, const char *arg) |
85f07f22 | 2931 | { |
4a897224 | 2932 | av_freep(pcodec_name); |
1629626f | 2933 | if (!strcmp(arg, "copy")) { |
cf7fc795 | 2934 | *pstream_copy = 1; |
85f07f22 | 2935 | } else { |
4a897224 | 2936 | *pcodec_name = av_strdup(arg); |
85f07f22 | 2937 | } |
26513856 | 2938 | return 0; |
85f07f22 FB |
2939 | } |
2940 | ||
26513856 | 2941 | static int opt_audio_codec(const char *opt, const char *arg) |
cf7fc795 | 2942 | { |
26513856 | 2943 | return opt_codec(&audio_stream_copy, &audio_codec_name, AVMEDIA_TYPE_AUDIO, arg); |
cf7fc795 FB |
2944 | } |
2945 | ||
26513856 | 2946 | static int opt_video_codec(const char *opt, const char *arg) |
85f07f22 | 2947 | { |
26513856 | 2948 | return opt_codec(&video_stream_copy, &video_codec_name, AVMEDIA_TYPE_VIDEO, arg); |
cf7fc795 | 2949 | } |
85f07f22 | 2950 | |
26513856 | 2951 | static int opt_subtitle_codec(const char *opt, const char *arg) |
cf7fc795 | 2952 | { |
26513856 | 2953 | return opt_codec(&subtitle_stream_copy, &subtitle_codec_name, AVMEDIA_TYPE_SUBTITLE, arg); |
85f07f22 FB |
2954 | } |
2955 | ||
26513856 | 2956 | static int opt_data_codec(const char *opt, const char *arg) |
e3b540b4 | 2957 | { |
26513856 | 2958 | return opt_codec(&data_stream_copy, &data_codec_name, AVMEDIA_TYPE_DATA, arg); |
e3b540b4 LB |
2959 | } |
2960 | ||
0d0778b0 | 2961 | static int opt_codec_tag(const char *opt, const char *arg) |
27ad7d3a BC |
2962 | { |
2963 | char *tail; | |
d0242e74 SS |
2964 | uint32_t *codec_tag; |
2965 | ||
2966 | codec_tag = !strcmp(opt, "atag") ? &audio_codec_tag : | |
2967 | !strcmp(opt, "vtag") ? &video_codec_tag : | |
2968 | !strcmp(opt, "stag") ? &subtitle_codec_tag : NULL; | |
0d0778b0 RP |
2969 | if (!codec_tag) |
2970 | return -1; | |
27ad7d3a | 2971 | |
d0242e74 SS |
2972 | *codec_tag = strtol(arg, &tail, 0); |
2973 | if (!tail || *tail) | |
2839dc97 | 2974 | *codec_tag = AV_RL32(arg); |
0d0778b0 RP |
2975 | |
2976 | return 0; | |
27ad7d3a BC |
2977 | } |
2978 | ||
26513856 | 2979 | static int opt_map(const char *opt, const char *arg) |
85f07f22 | 2980 | { |
17c8cc55 | 2981 | StreamMap *m; |
815f98cc | 2982 | char *p; |
85f07f22 | 2983 | |
3a8e8824 AJ |
2984 | stream_maps = grow_array(stream_maps, sizeof(*stream_maps), &nb_stream_maps, nb_stream_maps + 1); |
2985 | m = &stream_maps[nb_stream_maps-1]; | |
85f07f22 | 2986 | |
815f98cc | 2987 | m->file_index = strtol(arg, &p, 0); |
85f07f22 FB |
2988 | if (*p) |
2989 | p++; | |
a5dc85ef | 2990 | |
815f98cc | 2991 | m->stream_index = strtol(p, &p, 0); |
b4a3389e WG |
2992 | if (*p) { |
2993 | p++; | |
815f98cc | 2994 | m->sync_file_index = strtol(p, &p, 0); |
b4a3389e WG |
2995 | if (*p) |
2996 | p++; | |
815f98cc | 2997 | m->sync_stream_index = strtol(p, &p, 0); |
b4a3389e WG |
2998 | } else { |
2999 | m->sync_file_index = m->file_index; | |
3000 | m->sync_stream_index = m->stream_index; | |
3001 | } | |
26513856 | 3002 | return 0; |
85f07f22 FB |
3003 | } |
3004 | ||
c961fb3c | 3005 | static void parse_meta_type(char *arg, char *type, int *index, char **endptr) |
1829e195 AK |
3006 | { |
3007 | *endptr = arg; | |
3008 | if (*arg == ',') { | |
3009 | *type = *(++arg); | |
3010 | switch (*arg) { | |
3011 | case 'g': | |
3012 | break; | |
3013 | case 's': | |
3014 | case 'c': | |
3015 | case 'p': | |
3016 | *index = strtol(++arg, endptr, 0); | |
3017 | break; | |
3018 | default: | |
3019 | fprintf(stderr, "Invalid metadata type %c.\n", *arg); | |
3020 | ffmpeg_exit(1); | |
3021 | } | |
3022 | } else | |
3023 | *type = 'g'; | |
3024 | } | |
3025 | ||
26513856 | 3026 | static int opt_map_metadata(const char *opt, const char *arg) |
0a38bafd | 3027 | { |
17c8cc55 | 3028 | MetadataMap *m, *m1; |
815f98cc | 3029 | char *p; |
115329f1 | 3030 | |
63e856df AK |
3031 | meta_data_maps = grow_array(meta_data_maps, sizeof(*meta_data_maps), |
3032 | &nb_meta_data_maps, nb_meta_data_maps + 1); | |
0a38bafd | 3033 | |
1829e195 AK |
3034 | m = &meta_data_maps[nb_meta_data_maps - 1][0]; |
3035 | m->file = strtol(arg, &p, 0); | |
3036 | parse_meta_type(p, &m->type, &m->index, &p); | |
0a38bafd PB |
3037 | if (*p) |
3038 | p++; | |
3039 | ||
1829e195 AK |
3040 | m1 = &meta_data_maps[nb_meta_data_maps - 1][1]; |
3041 | m1->file = strtol(p, &p, 0); | |
3042 | parse_meta_type(p, &m1->type, &m1->index, &p); | |
d0abe80a | 3043 | |
477b1aea AK |
3044 | if (m->type == 'g' || m1->type == 'g') |
3045 | metadata_global_autocopy = 0; | |
d0abe80a AK |
3046 | if (m->type == 's' || m1->type == 's') |
3047 | metadata_streams_autocopy = 0; | |
3048 | if (m->type == 'c' || m1->type == 'c') | |
3049 | metadata_chapters_autocopy = 0; | |
26513856 SS |
3050 | |
3051 | return 0; | |
0a38bafd PB |
3052 | } |
3053 | ||
26513856 | 3054 | static int opt_map_meta_data(const char *opt, const char *arg) |
87e4d9b2 AK |
3055 | { |
3056 | fprintf(stderr, "-map_meta_data is deprecated and will be removed soon. " | |
3057 | "Use -map_metadata instead.\n"); | |
26513856 | 3058 | return opt_map_metadata(opt, arg); |
87e4d9b2 AK |
3059 | } |
3060 | ||
26513856 | 3061 | static int opt_map_chapters(const char *opt, const char *arg) |
91e96eba | 3062 | { |
17c8cc55 | 3063 | ChapterMap *c; |
91e96eba AK |
3064 | char *p; |
3065 | ||
3066 | chapter_maps = grow_array(chapter_maps, sizeof(*chapter_maps), &nb_chapter_maps, | |
3067 | nb_chapter_maps + 1); | |
3068 | c = &chapter_maps[nb_chapter_maps - 1]; | |
3069 | c->out_file = strtol(arg, &p, 0); | |
3070 | if (*p) | |
3071 | p++; | |
3072 | ||
3073 | c->in_file = strtol(p, &p, 0); | |