FFmpeg coverage


Directory: ../../../ffmpeg/
File: src/libavformat/dashenc.c
Date: 2022-12-05 03:11:11
Exec Total Coverage
Lines: 0 1402 0.0%
Functions: 0 40 0.0%
Branches: 0 1116 0.0%

Line Branch Exec Source
1 /*
2 * MPEG-DASH ISO BMFF segmenter
3 * Copyright (c) 2014 Martin Storsjo
4 * Copyright (c) 2018 Akamai Technologies, Inc.
5 *
6 * This file is part of FFmpeg.
7 *
8 * FFmpeg is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
12 *
13 * FFmpeg is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
17 *
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with FFmpeg; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21 */
22
23 #include "config.h"
24 #include "config_components.h"
25 #if HAVE_UNISTD_H
26 #include <unistd.h>
27 #endif
28
29 #include "libavutil/avassert.h"
30 #include "libavutil/avutil.h"
31 #include "libavutil/avstring.h"
32 #include "libavutil/bprint.h"
33 #include "libavutil/intreadwrite.h"
34 #include "libavutil/mathematics.h"
35 #include "libavutil/opt.h"
36 #include "libavutil/parseutils.h"
37 #include "libavutil/rational.h"
38 #include "libavutil/time.h"
39 #include "libavutil/time_internal.h"
40
41 #include "libavcodec/avcodec.h"
42
43 #include "av1.h"
44 #include "avc.h"
45 #include "avformat.h"
46 #include "avio_internal.h"
47 #include "hlsplaylist.h"
48 #if CONFIG_HTTP_PROTOCOL
49 #include "http.h"
50 #endif
51 #include "internal.h"
52 #include "isom.h"
53 #include "mux.h"
54 #include "os_support.h"
55 #include "url.h"
56 #include "vpcc.h"
57 #include "dash.h"
58
59 typedef enum {
60 SEGMENT_TYPE_AUTO = 0,
61 SEGMENT_TYPE_MP4,
62 SEGMENT_TYPE_WEBM,
63 SEGMENT_TYPE_NB
64 } SegmentType;
65
66 enum {
67 FRAG_TYPE_NONE = 0,
68 FRAG_TYPE_EVERY_FRAME,
69 FRAG_TYPE_DURATION,
70 FRAG_TYPE_PFRAMES,
71 FRAG_TYPE_NB
72 };
73
74 #define MPD_PROFILE_DASH 1
75 #define MPD_PROFILE_DVB 2
76
77 typedef struct Segment {
78 char file[1024];
79 int64_t start_pos;
80 int range_length, index_length;
81 int64_t time;
82 double prog_date_time;
83 int64_t duration;
84 int n;
85 } Segment;
86
87 typedef struct AdaptationSet {
88 int id;
89 char *descriptor;
90 int64_t seg_duration;
91 int64_t frag_duration;
92 int frag_type;
93 enum AVMediaType media_type;
94 AVDictionary *metadata;
95 AVRational min_frame_rate, max_frame_rate;
96 int ambiguous_frame_rate;
97 int64_t max_frag_duration;
98 int max_width, max_height;
99 int nb_streams;
100 AVRational par;
101 int trick_idx;
102 } AdaptationSet;
103
104 typedef struct OutputStream {
105 AVFormatContext *ctx;
106 int ctx_inited, as_idx;
107 AVIOContext *out;
108 AVCodecParserContext *parser;
109 AVCodecContext *parser_avctx;
110 int packets_written;
111 char initfile[1024];
112 int64_t init_start_pos, pos;
113 int init_range_length;
114 int nb_segments, segments_size, segment_index;
115 int64_t seg_duration;
116 int64_t frag_duration;
117 int64_t last_duration;
118 Segment **segments;
119 int64_t first_pts, start_pts, max_pts;
120 int64_t last_dts, last_pts;
121 int last_flags;
122 int bit_rate;
123 int first_segment_bit_rate;
124 SegmentType segment_type; /* segment type selected for this particular stream */
125 const char *format_name;
126 const char *extension_name;
127 const char *single_file_name; /* file names selected for this particular stream */
128 const char *init_seg_name;
129 const char *media_seg_name;
130
131 char codec_str[100];
132 int written_len;
133 char filename[1024];
134 char full_path[1024];
135 char temp_path[1024];
136 double availability_time_offset;
137 AVProducerReferenceTime producer_reference_time;
138 char producer_reference_time_str[100];
139 int total_pkt_size;
140 int64_t total_pkt_duration;
141 int muxer_overhead;
142 int frag_type;
143 int64_t gop_size;
144 AVRational sar;
145 int coding_dependency;
146 } OutputStream;
147
148 typedef struct DASHContext {
149 const AVClass *class; /* Class for private options. */
150 char *adaptation_sets;
151 AdaptationSet *as;
152 int nb_as;
153 int window_size;
154 int extra_window_size;
155 int64_t seg_duration;
156 int64_t frag_duration;
157 int remove_at_exit;
158 int use_template;
159 int use_timeline;
160 int single_file;
161 OutputStream *streams;
162 int has_video;
163 int64_t last_duration;
164 int64_t total_duration;
165 char availability_start_time[100];
166 time_t start_time_s;
167 int64_t presentation_time_offset;
168 char dirname[1024];
169 const char *single_file_name; /* file names as specified in options */
170 const char *init_seg_name;
171 const char *media_seg_name;
172 const char *utc_timing_url;
173 const char *method;
174 const char *user_agent;
175 AVDictionary *http_opts;
176 int hls_playlist;
177 const char *hls_master_name;
178 int http_persistent;
179 int master_playlist_created;
180 AVIOContext *mpd_out;
181 AVIOContext *m3u8_out;
182 int streaming;
183 int64_t timeout;
184 int index_correction;
185 AVDictionary *format_options;
186 int global_sidx;
187 SegmentType segment_type_option; /* segment type as specified in options */
188 int ignore_io_errors;
189 int lhls;
190 int ldash;
191 int master_publish_rate;
192 int nr_of_streams_to_flush;
193 int nr_of_streams_flushed;
194 int frag_type;
195 int write_prft;
196 int64_t max_gop_size;
197 int64_t max_segment_duration;
198 int profile;
199 int64_t target_latency;
200 int target_latency_refid;
201 AVRational min_playback_rate;
202 AVRational max_playback_rate;
203 int64_t update_period;
204 } DASHContext;
205
206 static struct codec_string {
207 int id;
208 const char *str;
209 } codecs[] = {
210 { AV_CODEC_ID_VP8, "vp8" },
211 { AV_CODEC_ID_VP9, "vp9" },
212 { AV_CODEC_ID_VORBIS, "vorbis" },
213 { AV_CODEC_ID_OPUS, "opus" },
214 { AV_CODEC_ID_FLAC, "flac" },
215 { 0, NULL }
216 };
217
218 static struct format_string {
219 SegmentType segment_type;
220 const char *str;
221 } formats[] = {
222 { SEGMENT_TYPE_AUTO, "auto" },
223 { SEGMENT_TYPE_MP4, "mp4" },
224 { SEGMENT_TYPE_WEBM, "webm" },
225 { 0, NULL }
226 };
227
228 static int dashenc_io_open(AVFormatContext *s, AVIOContext **pb, char *filename,
229 AVDictionary **options) {
230 DASHContext *c = s->priv_data;
231 int http_base_proto = filename ? ff_is_http_proto(filename) : 0;
232 int err = AVERROR_MUXER_NOT_FOUND;
233 if (!*pb || !http_base_proto || !c->http_persistent) {
234 err = s->io_open(s, pb, filename, AVIO_FLAG_WRITE, options);
235 #if CONFIG_HTTP_PROTOCOL
236 } else {
237 URLContext *http_url_context = ffio_geturlcontext(*pb);
238 av_assert0(http_url_context);
239 err = ff_http_do_new_request(http_url_context, filename);
240 if (err < 0)
241 ff_format_io_close(s, pb);
242 #endif
243 }
244 return err;
245 }
246
247 static void dashenc_io_close(AVFormatContext *s, AVIOContext **pb, char *filename) {
248 DASHContext *c = s->priv_data;
249 int http_base_proto = filename ? ff_is_http_proto(filename) : 0;
250
251 if (!*pb)
252 return;
253
254 if (!http_base_proto || !c->http_persistent) {
255 ff_format_io_close(s, pb);
256 #if CONFIG_HTTP_PROTOCOL
257 } else {
258 URLContext *http_url_context = ffio_geturlcontext(*pb);
259 av_assert0(http_url_context);
260 avio_flush(*pb);
261 ffurl_shutdown(http_url_context, AVIO_FLAG_WRITE);
262 #endif
263 }
264 }
265
266 static const char *get_format_str(SegmentType segment_type) {
267 int i;
268 for (i = 0; i < SEGMENT_TYPE_NB; i++)
269 if (formats[i].segment_type == segment_type)
270 return formats[i].str;
271 return NULL;
272 }
273
274 static const char *get_extension_str(SegmentType type, int single_file)
275 {
276 switch (type) {
277
278 case SEGMENT_TYPE_MP4: return single_file ? "mp4" : "m4s";
279 case SEGMENT_TYPE_WEBM: return "webm";
280 default: return NULL;
281 }
282 }
283
284 static int handle_io_open_error(AVFormatContext *s, int err, char *url) {
285 DASHContext *c = s->priv_data;
286 char errbuf[AV_ERROR_MAX_STRING_SIZE];
287 av_strerror(err, errbuf, sizeof(errbuf));
288 av_log(s, c->ignore_io_errors ? AV_LOG_WARNING : AV_LOG_ERROR,
289 "Unable to open %s for writing: %s\n", url, errbuf);
290 return c->ignore_io_errors ? 0 : err;
291 }
292
293 static inline SegmentType select_segment_type(SegmentType segment_type, enum AVCodecID codec_id)
294 {
295 if (segment_type == SEGMENT_TYPE_AUTO) {
296 if (codec_id == AV_CODEC_ID_OPUS || codec_id == AV_CODEC_ID_VORBIS ||
297 codec_id == AV_CODEC_ID_VP8 || codec_id == AV_CODEC_ID_VP9) {
298 segment_type = SEGMENT_TYPE_WEBM;
299 } else {
300 segment_type = SEGMENT_TYPE_MP4;
301 }
302 }
303
304 return segment_type;
305 }
306
307 static int init_segment_types(AVFormatContext *s)
308 {
309 DASHContext *c = s->priv_data;
310 int has_mp4_streams = 0;
311 for (int i = 0; i < s->nb_streams; ++i) {
312 OutputStream *os = &c->streams[i];
313 SegmentType segment_type = select_segment_type(
314 c->segment_type_option, s->streams[i]->codecpar->codec_id);
315 os->segment_type = segment_type;
316 os->format_name = get_format_str(segment_type);
317 if (!os->format_name) {
318 av_log(s, AV_LOG_ERROR, "Could not select DASH segment type for stream %d\n", i);
319 return AVERROR_MUXER_NOT_FOUND;
320 }
321 os->extension_name = get_extension_str(segment_type, c->single_file);
322 if (!os->extension_name) {
323 av_log(s, AV_LOG_ERROR, "Could not get extension type for stream %d\n", i);
324 return AVERROR_MUXER_NOT_FOUND;
325 }
326
327 has_mp4_streams |= segment_type == SEGMENT_TYPE_MP4;
328 }
329
330 if (c->hls_playlist && !has_mp4_streams) {
331 av_log(s, AV_LOG_WARNING, "No mp4 streams, disabling HLS manifest generation\n");
332 c->hls_playlist = 0;
333 }
334
335 return 0;
336 }
337
338 static void set_vp9_codec_str(AVFormatContext *s, AVCodecParameters *par,
339 AVRational *frame_rate, char *str, int size) {
340 VPCC vpcc;
341 int ret = ff_isom_get_vpcc_features(s, par, NULL, 0, frame_rate, &vpcc);
342 if (ret == 0) {
343 av_strlcatf(str, size, "vp09.%02d.%02d.%02d",
344 vpcc.profile, vpcc.level, vpcc.bitdepth);
345 } else {
346 // Default to just vp9 in case of error while finding out profile or level
347 av_log(s, AV_LOG_WARNING, "Could not find VP9 profile and/or level\n");
348 av_strlcpy(str, "vp9", size);
349 }
350 return;
351 }
352
353 static void set_codec_str(AVFormatContext *s, AVCodecParameters *par,
354 AVRational *frame_rate, char *str, int size)
355 {
356 const AVCodecTag *tags[2] = { NULL, NULL };
357 uint32_t tag;
358 int i;
359
360 // common Webm codecs are not part of RFC 6381
361 for (i = 0; codecs[i].id; i++)
362 if (codecs[i].id == par->codec_id) {
363 if (codecs[i].id == AV_CODEC_ID_VP9) {
364 set_vp9_codec_str(s, par, frame_rate, str, size);
365 } else {
366 av_strlcpy(str, codecs[i].str, size);
367 }
368 return;
369 }
370
371 // for codecs part of RFC 6381
372 if (par->codec_type == AVMEDIA_TYPE_VIDEO)
373 tags[0] = ff_codec_movvideo_tags;
374 else if (par->codec_type == AVMEDIA_TYPE_AUDIO)
375 tags[0] = ff_codec_movaudio_tags;
376 else
377 return;
378
379 tag = par->codec_tag;
380 if (!tag)
381 tag = av_codec_get_tag(tags, par->codec_id);
382 if (!tag)
383 return;
384 if (size < 5)
385 return;
386
387 AV_WL32(str, tag);
388 str[4] = '\0';
389 if (!strcmp(str, "mp4a") || !strcmp(str, "mp4v")) {
390 uint32_t oti;
391 tags[0] = ff_mp4_obj_type;
392 oti = av_codec_get_tag(tags, par->codec_id);
393 if (oti)
394 av_strlcatf(str, size, ".%02"PRIx32, oti);
395 else
396 return;
397
398 if (tag == MKTAG('m', 'p', '4', 'a')) {
399 if (par->extradata_size >= 2) {
400 int aot = par->extradata[0] >> 3;
401 if (aot == 31)
402 aot = ((AV_RB16(par->extradata) >> 5) & 0x3f) + 32;
403 av_strlcatf(str, size, ".%d", aot);
404 }
405 } else if (tag == MKTAG('m', 'p', '4', 'v')) {
406 // Unimplemented, should output ProfileLevelIndication as a decimal number
407 av_log(s, AV_LOG_WARNING, "Incomplete RFC 6381 codec string for mp4v\n");
408 }
409 } else if (!strcmp(str, "avc1")) {
410 uint8_t *tmpbuf = NULL;
411 uint8_t *extradata = par->extradata;
412 int extradata_size = par->extradata_size;
413 if (!extradata_size)
414 return;
415 if (extradata[0] != 1) {
416 AVIOContext *pb;
417 if (avio_open_dyn_buf(&pb) < 0)
418 return;
419 if (ff_isom_write_avcc(pb, extradata, extradata_size) < 0) {
420 ffio_free_dyn_buf(&pb);
421 return;
422 }
423 extradata_size = avio_close_dyn_buf(pb, &extradata);
424 tmpbuf = extradata;
425 }
426
427 if (extradata_size >= 4)
428 av_strlcatf(str, size, ".%02x%02x%02x",
429 extradata[1], extradata[2], extradata[3]);
430 av_free(tmpbuf);
431 } else if (!strcmp(str, "av01")) {
432 AV1SequenceParameters seq;
433 if (!par->extradata_size)
434 return;
435 if (ff_av1_parse_seq_header(&seq, par->extradata, par->extradata_size) < 0)
436 return;
437
438 av_strlcatf(str, size, ".%01u.%02u%s.%02u",
439 seq.profile, seq.level, seq.tier ? "H" : "M", seq.bitdepth);
440 if (seq.color_description_present_flag)
441 av_strlcatf(str, size, ".%01u.%01u%01u%01u.%02u.%02u.%02u.%01u",
442 seq.monochrome,
443 seq.chroma_subsampling_x, seq.chroma_subsampling_y, seq.chroma_sample_position,
444 seq.color_primaries, seq.transfer_characteristics, seq.matrix_coefficients,
445 seq.color_range);
446 }
447 }
448
449 static int flush_dynbuf(DASHContext *c, OutputStream *os, int *range_length)
450 {
451 uint8_t *buffer;
452
453 if (!os->ctx->pb) {
454 return AVERROR(EINVAL);
455 }
456
457 // flush
458 av_write_frame(os->ctx, NULL);
459 avio_flush(os->ctx->pb);
460
461 if (!c->single_file) {
462 // write out to file
463 *range_length = avio_close_dyn_buf(os->ctx->pb, &buffer);
464 os->ctx->pb = NULL;
465 if (os->out)
466 avio_write(os->out, buffer + os->written_len, *range_length - os->written_len);
467 os->written_len = 0;
468 av_free(buffer);
469
470 // re-open buffer
471 return avio_open_dyn_buf(&os->ctx->pb);
472 } else {
473 *range_length = avio_tell(os->ctx->pb) - os->pos;
474 return 0;
475 }
476 }
477
478 static void set_http_options(AVDictionary **options, DASHContext *c)
479 {
480 if (c->method)
481 av_dict_set(options, "method", c->method, 0);
482 av_dict_copy(options, c->http_opts, 0);
483 if (c->user_agent)
484 av_dict_set(options, "user_agent", c->user_agent, 0);
485 if (c->http_persistent)
486 av_dict_set_int(options, "multiple_requests", 1, 0);
487 if (c->timeout >= 0)
488 av_dict_set_int(options, "timeout", c->timeout, 0);
489 }
490
491 static void get_hls_playlist_name(char *playlist_name, int string_size,
492 const char *base_url, int id) {
493 if (base_url)
494 snprintf(playlist_name, string_size, "%smedia_%d.m3u8", base_url, id);
495 else
496 snprintf(playlist_name, string_size, "media_%d.m3u8", id);
497 }
498
499 static void get_start_index_number(OutputStream *os, DASHContext *c,
500 int *start_index, int *start_number) {
501 *start_index = 0;
502 *start_number = 1;
503 if (c->window_size) {
504 *start_index = FFMAX(os->nb_segments - c->window_size, 0);
505 *start_number = FFMAX(os->segment_index - c->window_size, 1);
506 }
507 }
508
509 static void write_hls_media_playlist(OutputStream *os, AVFormatContext *s,
510 int representation_id, int final,
511 char *prefetch_url) {
512 DASHContext *c = s->priv_data;
513 int timescale = os->ctx->streams[0]->time_base.den;
514 char temp_filename_hls[1024];
515 char filename_hls[1024];
516 AVDictionary *http_opts = NULL;
517 int target_duration = 0;
518 int ret = 0;
519 const char *proto = avio_find_protocol_name(c->dirname);
520 int use_rename = proto && !strcmp(proto, "file");
521 int i, start_index, start_number;
522 double prog_date_time = 0;
523
524 get_start_index_number(os, c, &start_index, &start_number);
525
526 if (!c->hls_playlist || start_index >= os->nb_segments ||
527 os->segment_type != SEGMENT_TYPE_MP4)
528 return;
529
530 get_hls_playlist_name(filename_hls, sizeof(filename_hls),
531 c->dirname, representation_id);
532
533 snprintf(temp_filename_hls, sizeof(temp_filename_hls), use_rename ? "%s.tmp" : "%s", filename_hls);
534
535 set_http_options(&http_opts, c);
536 ret = dashenc_io_open(s, &c->m3u8_out, temp_filename_hls, &http_opts);
537 av_dict_free(&http_opts);
538 if (ret < 0) {
539 handle_io_open_error(s, ret, temp_filename_hls);
540 return;
541 }
542 for (i = start_index; i < os->nb_segments; i++) {
543 Segment *seg = os->segments[i];
544 double duration = (double) seg->duration / timescale;
545 if (target_duration <= duration)
546 target_duration = lrint(duration);
547 }
548
549 ff_hls_write_playlist_header(c->m3u8_out, 6, -1, target_duration,
550 start_number, PLAYLIST_TYPE_NONE, 0);
551
552 ff_hls_write_init_file(c->m3u8_out, os->initfile, c->single_file,
553 os->init_range_length, os->init_start_pos);
554
555 for (i = start_index; i < os->nb_segments; i++) {
556 Segment *seg = os->segments[i];
557
558 if (fabs(prog_date_time) < 1e-7) {
559 if (os->nb_segments == 1)
560 prog_date_time = c->start_time_s;
561 else
562 prog_date_time = seg->prog_date_time;
563 }
564 seg->prog_date_time = prog_date_time;
565
566 ret = ff_hls_write_file_entry(c->m3u8_out, 0, c->single_file,
567 (double) seg->duration / timescale, 0,
568 seg->range_length, seg->start_pos, NULL,
569 c->single_file ? os->initfile : seg->file,
570 &prog_date_time, 0, 0, 0);
571 if (ret < 0) {
572 av_log(os->ctx, AV_LOG_WARNING, "ff_hls_write_file_entry get error\n");
573 }
574 }
575
576 if (prefetch_url)
577 avio_printf(c->m3u8_out, "#EXT-X-PREFETCH:%s\n", prefetch_url);
578
579 if (final)
580 ff_hls_write_end_list(c->m3u8_out);
581
582 dashenc_io_close(s, &c->m3u8_out, temp_filename_hls);
583
584 if (use_rename)
585 ff_rename(temp_filename_hls, filename_hls, os->ctx);
586 }
587
588 static int flush_init_segment(AVFormatContext *s, OutputStream *os)
589 {
590 DASHContext *c = s->priv_data;
591 int ret, range_length;
592
593 ret = flush_dynbuf(c, os, &range_length);
594 if (ret < 0)
595 return ret;
596
597 os->pos = os->init_range_length = range_length;
598 if (!c->single_file) {
599 char filename[1024];
600 snprintf(filename, sizeof(filename), "%s%s", c->dirname, os->initfile);
601 dashenc_io_close(s, &os->out, filename);
602 }
603 return 0;
604 }
605
606 static void dash_free(AVFormatContext *s)
607 {
608 DASHContext *c = s->priv_data;
609 int i, j;
610
611 if (c->as) {
612 for (i = 0; i < c->nb_as; i++) {
613 av_dict_free(&c->as[i].metadata);
614 av_freep(&c->as[i].descriptor);
615 }
616 av_freep(&c->as);
617 c->nb_as = 0;
618 }
619
620 if (!c->streams)
621 return;
622 for (i = 0; i < s->nb_streams; i++) {
623 OutputStream *os = &c->streams[i];
624 if (os->ctx && os->ctx->pb) {
625 if (!c->single_file)
626 ffio_free_dyn_buf(&os->ctx->pb);
627 else
628 avio_close(os->ctx->pb);
629 }
630 ff_format_io_close(s, &os->out);
631 avformat_free_context(os->ctx);
632 avcodec_free_context(&os->parser_avctx);
633 av_parser_close(os->parser);
634 for (j = 0; j < os->nb_segments; j++)
635 av_free(os->segments[j]);
636 av_free(os->segments);
637 av_freep(&os->single_file_name);
638 av_freep(&os->init_seg_name);
639 av_freep(&os->media_seg_name);
640 }
641 av_freep(&c->streams);
642
643 ff_format_io_close(s, &c->mpd_out);
644 ff_format_io_close(s, &c->m3u8_out);
645 }
646
647 static void output_segment_list(OutputStream *os, AVIOContext *out, AVFormatContext *s,
648 int representation_id, int final)
649 {
650 DASHContext *c = s->priv_data;
651 int i, start_index, start_number;
652 get_start_index_number(os, c, &start_index, &start_number);
653
654 if (c->use_template) {
655 int timescale = c->use_timeline ? os->ctx->streams[0]->time_base.den : AV_TIME_BASE;
656 avio_printf(out, "\t\t\t\t<SegmentTemplate timescale=\"%d\" ", timescale);
657 if (!c->use_timeline) {
658 avio_printf(out, "duration=\"%"PRId64"\" ", os->seg_duration);
659 if (c->streaming && os->availability_time_offset)
660 avio_printf(out, "availabilityTimeOffset=\"%.3f\" ",
661 os->availability_time_offset);
662 }
663 if (c->streaming && os->availability_time_offset && !final)
664 avio_printf(out, "availabilityTimeComplete=\"false\" ");
665
666 avio_printf(out, "initialization=\"%s\" media=\"%s\" startNumber=\"%d\"", os->init_seg_name, os->media_seg_name, c->use_timeline ? start_number : 1);
667 if (c->presentation_time_offset)
668 avio_printf(out, " presentationTimeOffset=\"%"PRId64"\"", c->presentation_time_offset);
669 avio_printf(out, ">\n");
670 if (c->use_timeline) {
671 int64_t cur_time = 0;
672 avio_printf(out, "\t\t\t\t\t<SegmentTimeline>\n");
673 for (i = start_index; i < os->nb_segments; ) {
674 Segment *seg = os->segments[i];
675 int repeat = 0;
676 avio_printf(out, "\t\t\t\t\t\t<S ");
677 if (i == start_index || seg->time != cur_time) {
678 cur_time = seg->time;
679 avio_printf(out, "t=\"%"PRId64"\" ", seg->time);
680 }
681 avio_printf(out, "d=\"%"PRId64"\" ", seg->duration);
682 while (i + repeat + 1 < os->nb_segments &&
683 os->segments[i + repeat + 1]->duration == seg->duration &&
684 os->segments[i + repeat + 1]->time == os->segments[i + repeat]->time + os->segments[i + repeat]->duration)
685 repeat++;
686 if (repeat > 0)
687 avio_printf(out, "r=\"%d\" ", repeat);
688 avio_printf(out, "/>\n");
689 i += 1 + repeat;
690 cur_time += (1 + repeat) * seg->duration;
691 }
692 avio_printf(out, "\t\t\t\t\t</SegmentTimeline>\n");
693 }
694 avio_printf(out, "\t\t\t\t</SegmentTemplate>\n");
695 } else if (c->single_file) {
696 avio_printf(out, "\t\t\t\t<BaseURL>%s</BaseURL>\n", os->initfile);
697 avio_printf(out, "\t\t\t\t<SegmentList timescale=\"%d\" duration=\"%"PRId64"\" startNumber=\"%d\">\n", AV_TIME_BASE, FFMIN(os->seg_duration, os->last_duration), start_number);
698 avio_printf(out, "\t\t\t\t\t<Initialization range=\"%"PRId64"-%"PRId64"\" />\n", os->init_start_pos, os->init_start_pos + os->init_range_length - 1);
699 for (i = start_index; i < os->nb_segments; i++) {
700 Segment *seg = os->segments[i];
701 avio_printf(out, "\t\t\t\t\t<SegmentURL mediaRange=\"%"PRId64"-%"PRId64"\" ", seg->start_pos, seg->start_pos + seg->range_length - 1);
702 if (seg->index_length)
703 avio_printf(out, "indexRange=\"%"PRId64"-%"PRId64"\" ", seg->start_pos, seg->start_pos + seg->index_length - 1);
704 avio_printf(out, "/>\n");
705 }
706 avio_printf(out, "\t\t\t\t</SegmentList>\n");
707 } else {
708 avio_printf(out, "\t\t\t\t<SegmentList timescale=\"%d\" duration=\"%"PRId64"\" startNumber=\"%d\">\n", AV_TIME_BASE, FFMIN(os->seg_duration, os->last_duration), start_number);
709 avio_printf(out, "\t\t\t\t\t<Initialization sourceURL=\"%s\" />\n", os->initfile);
710 for (i = start_index; i < os->nb_segments; i++) {
711 Segment *seg = os->segments[i];
712 avio_printf(out, "\t\t\t\t\t<SegmentURL media=\"%s\" />\n", seg->file);
713 }
714 avio_printf(out, "\t\t\t\t</SegmentList>\n");
715 }
716 if (!c->lhls || final) {
717 write_hls_media_playlist(os, s, representation_id, final, NULL);
718 }
719
720 }
721
722 static char *xmlescape(const char *str) {
723 int outlen = strlen(str)*3/2 + 6;
724 char *out = av_realloc(NULL, outlen + 1);
725 int pos = 0;
726 if (!out)
727 return NULL;
728 for (; *str; str++) {
729 if (pos + 6 > outlen) {
730 char *tmp;
731 outlen = 2 * outlen + 6;
732 tmp = av_realloc(out, outlen + 1);
733 if (!tmp) {
734 av_free(out);
735 return NULL;
736 }
737 out = tmp;
738 }
739 if (*str == '&') {
740 memcpy(&out[pos], "&amp;", 5);
741 pos += 5;
742 } else if (*str == '<') {
743 memcpy(&out[pos], "&lt;", 4);
744 pos += 4;
745 } else if (*str == '>') {
746 memcpy(&out[pos], "&gt;", 4);
747 pos += 4;
748 } else if (*str == '\'') {
749 memcpy(&out[pos], "&apos;", 6);
750 pos += 6;
751 } else if (*str == '\"') {
752 memcpy(&out[pos], "&quot;", 6);
753 pos += 6;
754 } else {
755 out[pos++] = *str;
756 }
757 }
758 out[pos] = '\0';
759 return out;
760 }
761
762 static void write_time(AVIOContext *out, int64_t time)
763 {
764 int seconds = time / AV_TIME_BASE;
765 int fractions = time % AV_TIME_BASE;
766 int minutes = seconds / 60;
767 int hours = minutes / 60;
768 seconds %= 60;
769 minutes %= 60;
770 avio_printf(out, "PT");
771 if (hours)
772 avio_printf(out, "%dH", hours);
773 if (hours || minutes)
774 avio_printf(out, "%dM", minutes);
775 avio_printf(out, "%d.%dS", seconds, fractions / (AV_TIME_BASE / 10));
776 }
777
778 static void format_date(char *buf, int size, int64_t time_us)
779 {
780 struct tm *ptm, tmbuf;
781 int64_t time_ms = time_us / 1000;
782 const time_t time_s = time_ms / 1000;
783 int millisec = time_ms - (time_s * 1000);
784 ptm = gmtime_r(&time_s, &tmbuf);
785 if (ptm) {
786 int len;
787 if (!strftime(buf, size, "%Y-%m-%dT%H:%M:%S", ptm)) {
788 buf[0] = '\0';
789 return;
790 }
791 len = strlen(buf);
792 snprintf(buf + len, size - len, ".%03dZ", millisec);
793 }
794 }
795
796 static int write_adaptation_set(AVFormatContext *s, AVIOContext *out, int as_index,
797 int final)
798 {
799 DASHContext *c = s->priv_data;
800 AdaptationSet *as = &c->as[as_index];
801 AVDictionaryEntry *lang, *role;
802 int i;
803
804 avio_printf(out, "\t\t<AdaptationSet id=\"%d\" contentType=\"%s\" startWithSAP=\"1\" segmentAlignment=\"true\" bitstreamSwitching=\"true\"",
805 as->id, as->media_type == AVMEDIA_TYPE_VIDEO ? "video" : "audio");
806 if (as->media_type == AVMEDIA_TYPE_VIDEO && as->max_frame_rate.num && !as->ambiguous_frame_rate && av_cmp_q(as->min_frame_rate, as->max_frame_rate) < 0)
807 avio_printf(out, " maxFrameRate=\"%d/%d\"", as->max_frame_rate.num, as->max_frame_rate.den);
808 else if (as->media_type == AVMEDIA_TYPE_VIDEO && as->max_frame_rate.num && !as->ambiguous_frame_rate && !av_cmp_q(as->min_frame_rate, as->max_frame_rate))
809 avio_printf(out, " frameRate=\"%d/%d\"", as->max_frame_rate.num, as->max_frame_rate.den);
810 if (as->media_type == AVMEDIA_TYPE_VIDEO) {
811 avio_printf(out, " maxWidth=\"%d\" maxHeight=\"%d\"", as->max_width, as->max_height);
812 avio_printf(out, " par=\"%d:%d\"", as->par.num, as->par.den);
813 }
814 lang = av_dict_get(as->metadata, "language", NULL, 0);
815 if (lang)
816 avio_printf(out, " lang=\"%s\"", lang->value);
817 avio_printf(out, ">\n");
818
819 if (!final && c->ldash && as->max_frag_duration && !(c->profile & MPD_PROFILE_DVB))
820 avio_printf(out, "\t\t\t<Resync dT=\"%"PRId64"\" type=\"0\"/>\n", as->max_frag_duration);
821 if (as->trick_idx >= 0)
822 avio_printf(out, "\t\t\t<EssentialProperty id=\"%d\" schemeIdUri=\"http://dashif.org/guidelines/trickmode\" value=\"%d\"/>\n", as->id, as->trick_idx);
823 role = av_dict_get(as->metadata, "role", NULL, 0);
824 if (role)
825 avio_printf(out, "\t\t\t<Role schemeIdUri=\"urn:mpeg:dash:role:2011\" value=\"%s\"/>\n", role->value);
826 if (as->descriptor)
827 avio_printf(out, "\t\t\t%s\n", as->descriptor);
828 for (i = 0; i < s->nb_streams; i++) {
829 AVStream *st = s->streams[i];
830 OutputStream *os = &c->streams[i];
831 char bandwidth_str[64] = {'\0'};
832
833 if (os->as_idx - 1 != as_index)
834 continue;
835
836 if (os->bit_rate > 0)
837 snprintf(bandwidth_str, sizeof(bandwidth_str), " bandwidth=\"%d\"", os->bit_rate);
838 else if (final) {
839 int average_bit_rate = os->pos * 8 * AV_TIME_BASE / c->total_duration;
840 snprintf(bandwidth_str, sizeof(bandwidth_str), " bandwidth=\"%d\"", average_bit_rate);
841 } else if (os->first_segment_bit_rate > 0)
842 snprintf(bandwidth_str, sizeof(bandwidth_str), " bandwidth=\"%d\"", os->first_segment_bit_rate);
843
844 if (as->media_type == AVMEDIA_TYPE_VIDEO) {
845 avio_printf(out, "\t\t\t<Representation id=\"%d\" mimeType=\"video/%s\" codecs=\"%s\"%s width=\"%d\" height=\"%d\"",
846 i, os->format_name, os->codec_str, bandwidth_str, s->streams[i]->codecpar->width, s->streams[i]->codecpar->height);
847 if (st->codecpar->field_order == AV_FIELD_UNKNOWN)
848 avio_printf(out, " scanType=\"unknown\"");
849 else if (st->codecpar->field_order != AV_FIELD_PROGRESSIVE)
850 avio_printf(out, " scanType=\"interlaced\"");
851 avio_printf(out, " sar=\"%d:%d\"", os->sar.num, os->sar.den);
852 if (st->avg_frame_rate.num && av_cmp_q(as->min_frame_rate, as->max_frame_rate) < 0)
853 avio_printf(out, " frameRate=\"%d/%d\"", st->avg_frame_rate.num, st->avg_frame_rate.den);
854 if (as->trick_idx >= 0) {
855 AdaptationSet *tas = &c->as[as->trick_idx];
856 if (!as->ambiguous_frame_rate && !tas->ambiguous_frame_rate)
857 avio_printf(out, " maxPlayoutRate=\"%d\"", FFMAX((int)av_q2d(av_div_q(tas->min_frame_rate, as->min_frame_rate)), 1));
858 }
859 if (!os->coding_dependency)
860 avio_printf(out, " codingDependency=\"false\"");
861 avio_printf(out, ">\n");
862 } else {
863 avio_printf(out, "\t\t\t<Representation id=\"%d\" mimeType=\"audio/%s\" codecs=\"%s\"%s audioSamplingRate=\"%d\">\n",
864 i, os->format_name, os->codec_str, bandwidth_str, s->streams[i]->codecpar->sample_rate);
865 avio_printf(out, "\t\t\t\t<AudioChannelConfiguration schemeIdUri=\"urn:mpeg:dash:23003:3:audio_channel_configuration:2011\" value=\"%d\" />\n",
866 s->streams[i]->codecpar->ch_layout.nb_channels);
867 }
868 if (!final && c->write_prft && os->producer_reference_time_str[0]) {
869 avio_printf(out, "\t\t\t\t<ProducerReferenceTime id=\"%d\" inband=\"true\" type=\"%s\" wallClockTime=\"%s\" presentationTime=\"%"PRId64"\">\n",
870 i, os->producer_reference_time.flags ? "captured" : "encoder", os->producer_reference_time_str, c->presentation_time_offset);
871 avio_printf(out, "\t\t\t\t\t<UTCTiming schemeIdUri=\"urn:mpeg:dash:utc:http-xsdate:2014\" value=\"%s\"/>\n", c->utc_timing_url);
872 avio_printf(out, "\t\t\t\t</ProducerReferenceTime>\n");
873 }
874 if (!final && c->ldash && os->gop_size && os->frag_type != FRAG_TYPE_NONE && !(c->profile & MPD_PROFILE_DVB) &&
875 (os->frag_type != FRAG_TYPE_DURATION || os->frag_duration != os->seg_duration))
876 avio_printf(out, "\t\t\t\t<Resync dT=\"%"PRId64"\" type=\"1\"/>\n", os->gop_size);
877 output_segment_list(os, out, s, i, final);
878 avio_printf(out, "\t\t\t</Representation>\n");
879 }
880 avio_printf(out, "\t\t</AdaptationSet>\n");
881
882 return 0;
883 }
884
885 static int add_adaptation_set(AVFormatContext *s, AdaptationSet **as, enum AVMediaType type)
886 {
887 DASHContext *c = s->priv_data;
888 void *mem;
889
890 if (c->profile & MPD_PROFILE_DVB && (c->nb_as + 1) > 16) {
891 av_log(s, AV_LOG_ERROR, "DVB-DASH profile allows a max of 16 Adaptation Sets\n");
892 return AVERROR(EINVAL);
893 }
894 mem = av_realloc(c->as, sizeof(*c->as) * (c->nb_as + 1));
895 if (!mem)
896 return AVERROR(ENOMEM);
897 c->as = mem;
898 ++c->nb_as;
899
900 *as = &c->as[c->nb_as - 1];
901 memset(*as, 0, sizeof(**as));
902 (*as)->media_type = type;
903 (*as)->frag_type = -1;
904 (*as)->trick_idx = -1;
905
906 return 0;
907 }
908
909 static int adaptation_set_add_stream(AVFormatContext *s, int as_idx, int i)
910 {
911 DASHContext *c = s->priv_data;
912 AdaptationSet *as = &c->as[as_idx - 1];
913 OutputStream *os = &c->streams[i];
914
915 if (as->media_type != s->streams[i]->codecpar->codec_type) {
916 av_log(s, AV_LOG_ERROR, "Codec type of stream %d doesn't match AdaptationSet's media type\n", i);
917 return AVERROR(EINVAL);
918 } else if (os->as_idx) {
919 av_log(s, AV_LOG_ERROR, "Stream %d is already assigned to an AdaptationSet\n", i);
920 return AVERROR(EINVAL);
921 }
922 if (c->profile & MPD_PROFILE_DVB && (as->nb_streams + 1) > 16) {
923 av_log(s, AV_LOG_ERROR, "DVB-DASH profile allows a max of 16 Representations per Adaptation Set\n");
924 return AVERROR(EINVAL);
925 }
926 os->as_idx = as_idx;
927 ++as->nb_streams;
928
929 return 0;
930 }
931
932 static int parse_adaptation_sets(AVFormatContext *s)
933 {
934 DASHContext *c = s->priv_data;
935 const char *p = c->adaptation_sets;
936 enum { new_set, parse_default, parsing_streams, parse_seg_duration, parse_frag_duration } state;
937 AdaptationSet *as;
938 int i, n, ret;
939
940 // default: one AdaptationSet for each stream
941 if (!p) {
942 for (i = 0; i < s->nb_streams; i++) {
943 if ((ret = add_adaptation_set(s, &as, s->streams[i]->codecpar->codec_type)) < 0)
944 return ret;
945 as->id = i;
946
947 c->streams[i].as_idx = c->nb_as;
948 ++as->nb_streams;
949 }
950 goto end;
951 }
952
953 // syntax id=0,streams=0,1,2 id=1,streams=3,4 and so on
954 // option id=0,descriptor=descriptor_str,streams=0,1,2 and so on
955 // option id=0,seg_duration=2.5,frag_duration=0.5,streams=0,1,2
956 // id=1,trick_id=0,seg_duration=10,frag_type=none,streams=3 and so on
957 // descriptor is useful to the scheme defined by ISO/IEC 23009-1:2014/Amd.2:2015
958 // descriptor_str should be a self-closing xml tag.
959 // seg_duration and frag_duration have the same syntax as the global options of
960 // the same name, and the former have precedence over them if set.
961 state = new_set;
962 while (*p) {
963 if (*p == ' ') {
964 p++;
965 continue;
966 } else if (state == new_set && av_strstart(p, "id=", &p)) {
967 char id_str[10], *end_str;
968
969 n = strcspn(p, ",");
970 snprintf(id_str, sizeof(id_str), "%.*s", n, p);
971
972 i = strtol(id_str, &end_str, 10);
973 if (id_str == end_str || i < 0 || i > c->nb_as) {
974 av_log(s, AV_LOG_ERROR, "\"%s\" is not a valid value for an AdaptationSet id\n", id_str);
975 return AVERROR(EINVAL);
976 }
977
978 if ((ret = add_adaptation_set(s, &as, AVMEDIA_TYPE_UNKNOWN)) < 0)
979 return ret;
980 as->id = i;
981
982 p += n;
983 if (*p)
984 p++;
985 state = parse_default;
986 } else if (state != new_set && av_strstart(p, "seg_duration=", &p)) {
987 state = parse_seg_duration;
988 } else if (state != new_set && av_strstart(p, "frag_duration=", &p)) {
989 state = parse_frag_duration;
990 } else if (state == parse_seg_duration || state == parse_frag_duration) {
991 char str[32];
992 int64_t usecs = 0;
993
994 n = strcspn(p, ",");
995 snprintf(str, sizeof(str), "%.*s", n, p);
996 p += n;
997 if (*p)
998 p++;
999
1000 ret = av_parse_time(&usecs, str, 1);
1001 if (ret < 0) {
1002 av_log(s, AV_LOG_ERROR, "Unable to parse option value \"%s\" as duration\n", str);
1003 return ret;
1004 }
1005
1006 if (state == parse_seg_duration)
1007 as->seg_duration = usecs;
1008 else
1009 as->frag_duration = usecs;
1010 state = parse_default;
1011 } else if (state != new_set && av_strstart(p, "frag_type=", &p)) {
1012 char type_str[16];
1013
1014 n = strcspn(p, ",");
1015 snprintf(type_str, sizeof(type_str), "%.*s", n, p);
1016 p += n;
1017 if (*p)
1018 p++;
1019
1020 if (!strcmp(type_str, "duration"))
1021 as->frag_type = FRAG_TYPE_DURATION;
1022 else if (!strcmp(type_str, "pframes"))
1023 as->frag_type = FRAG_TYPE_PFRAMES;
1024 else if (!strcmp(type_str, "every_frame"))
1025 as->frag_type = FRAG_TYPE_EVERY_FRAME;
1026 else if (!strcmp(type_str, "none"))
1027 as->frag_type = FRAG_TYPE_NONE;
1028 else {
1029 av_log(s, AV_LOG_ERROR, "Unable to parse option value \"%s\" as fragment type\n", type_str);
1030 return ret;
1031 }
1032 state = parse_default;
1033 } else if (state != new_set && av_strstart(p, "descriptor=", &p)) {
1034 n = strcspn(p, ">") + 1; //followed by one comma, so plus 1
1035 if (n < strlen(p)) {
1036 as->descriptor = av_strndup(p, n);
1037 } else {
1038 av_log(s, AV_LOG_ERROR, "Parse error, descriptor string should be a self-closing xml tag\n");
1039 return AVERROR(EINVAL);
1040 }
1041 p += n;
1042 if (*p)
1043 p++;
1044 state = parse_default;
1045 } else if ((state != new_set) && av_strstart(p, "trick_id=", &p)) {
1046 char trick_id_str[10], *end_str;
1047
1048 n = strcspn(p, ",");
1049 snprintf(trick_id_str, sizeof(trick_id_str), "%.*s", n, p);
1050 p += n;
1051
1052 as->trick_idx = strtol(trick_id_str, &end_str, 10);
1053 if (trick_id_str == end_str || as->trick_idx < 0)
1054 return AVERROR(EINVAL);
1055
1056 if (*p)
1057 p++;
1058 state = parse_default;
1059 } else if ((state != new_set) && av_strstart(p, "streams=", &p)) { //descriptor and durations are optional
1060 state = parsing_streams;
1061 } else if (state == parsing_streams) {
1062 AdaptationSet *as = &c->as[c->nb_as - 1];
1063 char idx_str[8], *end_str;
1064
1065 n = strcspn(p, " ,");
1066 snprintf(idx_str, sizeof(idx_str), "%.*s", n, p);
1067 p += n;
1068
1069 // if value is "a" or "v", map all streams of that type
1070 if (as->media_type == AVMEDIA_TYPE_UNKNOWN && (idx_str[0] == 'v' || idx_str[0] == 'a')) {
1071 enum AVMediaType type = (idx_str[0] == 'v') ? AVMEDIA_TYPE_VIDEO : AVMEDIA_TYPE_AUDIO;
1072 av_log(s, AV_LOG_DEBUG, "Map all streams of type %s\n", idx_str);
1073
1074 for (i = 0; i < s->nb_streams; i++) {
1075 if (s->streams[i]->codecpar->codec_type != type)
1076 continue;
1077
1078 as->media_type = s->streams[i]->codecpar->codec_type;
1079
1080 if ((ret = adaptation_set_add_stream(s, c->nb_as, i)) < 0)
1081 return ret;
1082 }
1083 } else { // select single stream
1084 i = strtol(idx_str, &end_str, 10);
1085 if (idx_str == end_str || i < 0 || i >= s->nb_streams) {
1086 av_log(s, AV_LOG_ERROR, "Selected stream \"%s\" not found!\n", idx_str);
1087 return AVERROR(EINVAL);
1088 }
1089 av_log(s, AV_LOG_DEBUG, "Map stream %d\n", i);
1090
1091 if (as->media_type == AVMEDIA_TYPE_UNKNOWN) {
1092 as->media_type = s->streams[i]->codecpar->codec_type;
1093 }
1094
1095 if ((ret = adaptation_set_add_stream(s, c->nb_as, i)) < 0)
1096 return ret;
1097 }
1098
1099 if (*p == ' ')
1100 state = new_set;
1101 if (*p)
1102 p++;
1103 } else {
1104 return AVERROR(EINVAL);
1105 }
1106 }
1107
1108 end:
1109 // check for unassigned streams
1110 for (i = 0; i < s->nb_streams; i++) {
1111 OutputStream *os = &c->streams[i];
1112 if (!os->as_idx) {
1113 av_log(s, AV_LOG_ERROR, "Stream %d is not mapped to an AdaptationSet\n", i);
1114 return AVERROR(EINVAL);
1115 }
1116 }
1117
1118 // check references for trick mode AdaptationSet
1119 for (i = 0; i < c->nb_as; i++) {
1120 as = &c->as[i];
1121 if (as->trick_idx < 0)
1122 continue;
1123 for (n = 0; n < c->nb_as; n++) {
1124 if (c->as[n].id == as->trick_idx)
1125 break;
1126 }
1127 if (n >= c->nb_as) {
1128 av_log(s, AV_LOG_ERROR, "reference AdaptationSet id \"%d\" not found for trick mode AdaptationSet id \"%d\"\n", as->trick_idx, as->id);
1129 return AVERROR(EINVAL);
1130 }
1131 }
1132
1133 return 0;
1134 }
1135
1136 static int write_manifest(AVFormatContext *s, int final)
1137 {
1138 DASHContext *c = s->priv_data;
1139 AVIOContext *out;
1140 char temp_filename[1024];
1141 int ret, i;
1142 const char *proto = avio_find_protocol_name(s->url);
1143 int use_rename = proto && !strcmp(proto, "file");
1144 static unsigned int warned_non_file = 0;
1145 AVDictionaryEntry *title = av_dict_get(s->metadata, "title", NULL, 0);
1146 AVDictionary *opts = NULL;
1147
1148 if (!use_rename && !warned_non_file++)
1149 av_log(s, AV_LOG_ERROR, "Cannot use rename on non file protocol, this may lead to races and temporary partial files\n");
1150
1151 snprintf(temp_filename, sizeof(temp_filename), use_rename ? "%s.tmp" : "%s", s->url);
1152 set_http_options(&opts, c);
1153 ret = dashenc_io_open(s, &c->mpd_out, temp_filename, &opts);
1154 av_dict_free(&opts);
1155 if (ret < 0) {
1156 return handle_io_open_error(s, ret, temp_filename);
1157 }
1158 out = c->mpd_out;
1159 avio_printf(out, "<?xml version=\"1.0\" encoding=\"utf-8\"?>\n");
1160 avio_printf(out, "<MPD xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\"\n"
1161 "\txmlns=\"urn:mpeg:dash:schema:mpd:2011\"\n"
1162 "\txmlns:xlink=\"http://www.w3.org/1999/xlink\"\n"
1163 "\txsi:schemaLocation=\"urn:mpeg:DASH:schema:MPD:2011 http://standards.iso.org/ittf/PubliclyAvailableStandards/MPEG-DASH_schema_files/DASH-MPD.xsd\"\n"
1164 "\tprofiles=\"");
1165 if (c->profile & MPD_PROFILE_DASH)
1166 avio_printf(out, "%s%s", "urn:mpeg:dash:profile:isoff-live:2011", c->profile & MPD_PROFILE_DVB ? "," : "\"\n");
1167 if (c->profile & MPD_PROFILE_DVB)
1168 avio_printf(out, "%s", "urn:dvb:dash:profile:dvb-dash:2014\"\n");
1169 avio_printf(out, "\ttype=\"%s\"\n",
1170 final ? "static" : "dynamic");
1171 if (final) {
1172 avio_printf(out, "\tmediaPresentationDuration=\"");
1173 write_time(out, c->total_duration);
1174 avio_printf(out, "\"\n");
1175 } else {
1176 int64_t update_period = c->last_duration / AV_TIME_BASE;
1177 char now_str[100];
1178 if (c->use_template && !c->use_timeline)
1179 update_period = 500;
1180 if (c->update_period)
1181 update_period = c->update_period;
1182 avio_printf(out, "\tminimumUpdatePeriod=\"PT%"PRId64"S\"\n", update_period);
1183 if (!c->ldash)
1184 avio_printf(out, "\tsuggestedPresentationDelay=\"PT%"PRId64"S\"\n", c->last_duration / AV_TIME_BASE);
1185 if (c->availability_start_time[0])
1186 avio_printf(out, "\tavailabilityStartTime=\"%s\"\n", c->availability_start_time);
1187 format_date(now_str, sizeof(now_str), av_gettime());
1188 if (now_str[0])
1189 avio_printf(out, "\tpublishTime=\"%s\"\n", now_str);
1190 if (c->window_size && c->use_template) {
1191 avio_printf(out, "\ttimeShiftBufferDepth=\"");
1192 write_time(out, c->last_duration * c->window_size);
1193 avio_printf(out, "\"\n");
1194 }
1195 }
1196 avio_printf(out, "\tmaxSegmentDuration=\"");
1197 write_time(out, c->max_segment_duration);
1198 avio_printf(out, "\"\n");
1199 avio_printf(out, "\tminBufferTime=\"");
1200 write_time(out, c->ldash && c->max_gop_size ? c->max_gop_size : c->last_duration * 2);
1201 avio_printf(out, "\">\n");
1202 avio_printf(out, "\t<ProgramInformation>\n");
1203 if (title) {
1204 char *escaped = xmlescape(title->value);
1205 avio_printf(out, "\t\t<Title>%s</Title>\n", escaped);
1206 av_free(escaped);
1207 }
1208 avio_printf(out, "\t</ProgramInformation>\n");
1209
1210 avio_printf(out, "\t<ServiceDescription id=\"0\">\n");
1211 if (!final && c->target_latency && c->target_latency_refid >= 0) {
1212 avio_printf(out, "\t\t<Latency target=\"%"PRId64"\"", c->target_latency / 1000);
1213 if (s->nb_streams > 1)
1214 avio_printf(out, " referenceId=\"%d\"", c->target_latency_refid);
1215 avio_printf(out, "/>\n");
1216 }
1217 if (av_cmp_q(c->min_playback_rate, (AVRational) {1, 1}) ||
1218 av_cmp_q(c->max_playback_rate, (AVRational) {1, 1}))
1219 avio_printf(out, "\t\t<PlaybackRate min=\"%.2f\" max=\"%.2f\"/>\n",
1220 av_q2d(c->min_playback_rate), av_q2d(c->max_playback_rate));
1221 avio_printf(out, "\t</ServiceDescription>\n");
1222
1223 if (c->window_size && s->nb_streams > 0 && c->streams[0].nb_segments > 0 && !c->use_template) {
1224 OutputStream *os = &c->streams[0];
1225 int start_index = FFMAX(os->nb_segments - c->window_size, 0);
1226 int64_t start_time = av_rescale_q(os->segments[start_index]->time, s->streams[0]->time_base, AV_TIME_BASE_Q);
1227 avio_printf(out, "\t<Period id=\"0\" start=\"");
1228 write_time(out, start_time);
1229 avio_printf(out, "\">\n");
1230 } else {
1231 avio_printf(out, "\t<Period id=\"0\" start=\"PT0.0S\">\n");
1232 }
1233
1234 for (i = 0; i < c->nb_as; i++) {
1235 if ((ret = write_adaptation_set(s, out, i, final)) < 0)
1236 return ret;
1237 }
1238 avio_printf(out, "\t</Period>\n");
1239
1240 if (c->utc_timing_url)
1241 avio_printf(out, "\t<UTCTiming schemeIdUri=\"urn:mpeg:dash:utc:http-xsdate:2014\" value=\"%s\"/>\n", c->utc_timing_url);
1242
1243 avio_printf(out, "</MPD>\n");
1244 avio_flush(out);
1245 dashenc_io_close(s, &c->mpd_out, temp_filename);
1246
1247 if (use_rename) {
1248 if ((ret = ff_rename(temp_filename, s->url, s)) < 0)
1249 return ret;
1250 }
1251
1252 if (c->hls_playlist) {
1253 char filename_hls[1024];
1254
1255 // Publish master playlist only the configured rate
1256 if (c->master_playlist_created && (!c->master_publish_rate ||
1257 c->streams[0].segment_index % c->master_publish_rate))
1258 return 0;
1259
1260 if (*c->dirname)
1261 snprintf(filename_hls, sizeof(filename_hls), "%s%s", c->dirname, c->hls_master_name);
1262 else
1263 snprintf(filename_hls, sizeof(filename_hls), "%s", c->hls_master_name);
1264
1265 snprintf(temp_filename, sizeof(temp_filename), use_rename ? "%s.tmp" : "%s", filename_hls);
1266
1267 set_http_options(&opts, c);
1268 ret = dashenc_io_open(s, &c->m3u8_out, temp_filename, &opts);
1269 av_dict_free(&opts);
1270 if (ret < 0) {
1271 return handle_io_open_error(s, ret, temp_filename);
1272 }
1273
1274 ff_hls_write_playlist_version(c->m3u8_out, 7);
1275
1276 if (c->has_video) {
1277 // treat audio streams as alternative renditions for video streams
1278 const char *audio_group = "A1";
1279 char audio_codec_str[128] = "\0";
1280 int is_default = 1;
1281 int max_audio_bitrate = 0;
1282
1283 for (i = 0; i < s->nb_streams; i++) {
1284 char playlist_file[64];
1285 AVStream *st = s->streams[i];
1286 OutputStream *os = &c->streams[i];
1287 if (st->codecpar->codec_type != AVMEDIA_TYPE_AUDIO)
1288 continue;
1289 if (os->segment_type != SEGMENT_TYPE_MP4)
1290 continue;
1291 get_hls_playlist_name(playlist_file, sizeof(playlist_file), NULL, i);
1292 ff_hls_write_audio_rendition(c->m3u8_out, (char *)audio_group,
1293 playlist_file, NULL, i, is_default);
1294 max_audio_bitrate = FFMAX(st->codecpar->bit_rate +
1295 os->muxer_overhead, max_audio_bitrate);
1296 if (!av_strnstr(audio_codec_str, os->codec_str, sizeof(audio_codec_str))) {
1297 if (strlen(audio_codec_str))
1298 av_strlcat(audio_codec_str, ",", sizeof(audio_codec_str));
1299 av_strlcat(audio_codec_str, os->codec_str, sizeof(audio_codec_str));
1300 }
1301 is_default = 0;
1302 }
1303
1304 for (i = 0; i < s->nb_streams; i++) {
1305 char playlist_file[64];
1306 char codec_str[128];
1307 AVStream *st = s->streams[i];
1308 OutputStream *os = &c->streams[i];
1309 char *agroup = NULL;
1310 int stream_bitrate = os->muxer_overhead;
1311 if (os->bit_rate > 0)
1312 stream_bitrate += os->bit_rate;
1313 else if (final)
1314 stream_bitrate += os->pos * 8 * AV_TIME_BASE / c->total_duration;
1315 else if (os->first_segment_bit_rate > 0)
1316 stream_bitrate += os->first_segment_bit_rate;
1317 if (st->codecpar->codec_type != AVMEDIA_TYPE_VIDEO)
1318 continue;
1319 if (os->segment_type != SEGMENT_TYPE_MP4)
1320 continue;
1321 av_strlcpy(codec_str, os->codec_str, sizeof(codec_str));
1322 if (max_audio_bitrate) {
1323 agroup = (char *)audio_group;
1324 stream_bitrate += max_audio_bitrate;
1325 av_strlcat(codec_str, ",", sizeof(codec_str));
1326 av_strlcat(codec_str, audio_codec_str, sizeof(codec_str));
1327 }
1328 get_hls_playlist_name(playlist_file, sizeof(playlist_file), NULL, i);
1329 ff_hls_write_stream_info(st, c->m3u8_out, stream_bitrate,
1330 playlist_file, agroup,
1331 codec_str, NULL, NULL);
1332 }
1333
1334 } else {
1335 // treat audio streams as separate renditions
1336
1337 for (i = 0; i < s->nb_streams; i++) {
1338 char playlist_file[64];
1339 char codec_str[128];
1340 AVStream *st = s->streams[i];
1341 OutputStream *os = &c->streams[i];
1342 int stream_bitrate = os->muxer_overhead;
1343 if (os->bit_rate > 0)
1344 stream_bitrate += os->bit_rate;
1345 else if (final)
1346 stream_bitrate += os->pos * 8 * AV_TIME_BASE / c->total_duration;
1347 else if (os->first_segment_bit_rate > 0)
1348 stream_bitrate += os->first_segment_bit_rate;
1349 if (st->codecpar->codec_type != AVMEDIA_TYPE_AUDIO)
1350 continue;
1351 if (os->segment_type != SEGMENT_TYPE_MP4)
1352 continue;
1353 av_strlcpy(codec_str, os->codec_str, sizeof(codec_str));
1354 get_hls_playlist_name(playlist_file, sizeof(playlist_file), NULL, i);
1355 ff_hls_write_stream_info(st, c->m3u8_out, stream_bitrate,
1356 playlist_file, NULL,
1357 codec_str, NULL, NULL);
1358 }
1359 }
1360
1361 dashenc_io_close(s, &c->m3u8_out, temp_filename);
1362 if (use_rename)
1363 if ((ret = ff_rename(temp_filename, filename_hls, s)) < 0)
1364 return ret;
1365 c->master_playlist_created = 1;
1366 }
1367
1368 return 0;
1369 }
1370
1371 static int dict_copy_entry(AVDictionary **dst, const AVDictionary *src, const char *key)
1372 {
1373 AVDictionaryEntry *entry = av_dict_get(src, key, NULL, 0);
1374 if (entry)
1375 av_dict_set(dst, key, entry->value, AV_DICT_DONT_OVERWRITE);
1376 return 0;
1377 }
1378
1379 static int dash_init(AVFormatContext *s)
1380 {
1381 DASHContext *c = s->priv_data;
1382 int ret = 0, i;
1383 char *ptr;
1384 char basename[1024];
1385
1386 c->nr_of_streams_to_flush = 0;
1387 if (c->single_file_name)
1388 c->single_file = 1;
1389 if (c->single_file)
1390 c->use_template = 0;
1391
1392 if (!c->profile) {
1393 av_log(s, AV_LOG_ERROR, "At least one profile must be enabled.\n");
1394 return AVERROR(EINVAL);
1395 }
1396 if (c->lhls && s->strict_std_compliance > FF_COMPLIANCE_EXPERIMENTAL) {
1397 av_log(s, AV_LOG_ERROR,
1398 "LHLS is experimental, Please set -strict experimental in order to enable it.\n");
1399 return AVERROR_EXPERIMENTAL;
1400 }
1401
1402 if (c->lhls && !c->streaming) {
1403 av_log(s, AV_LOG_WARNING, "Enabling streaming as LHLS is enabled\n");
1404 c->streaming = 1;
1405 }
1406
1407 if (c->lhls && !c->hls_playlist) {
1408 av_log(s, AV_LOG_INFO, "Enabling hls_playlist as LHLS is enabled\n");
1409 c->hls_playlist = 1;
1410 }
1411
1412 if (c->ldash && !c->streaming) {
1413 av_log(s, AV_LOG_WARNING, "Enabling streaming as LDash is enabled\n");
1414 c->streaming = 1;
1415 }
1416
1417 if (c->target_latency && !c->streaming) {
1418 av_log(s, AV_LOG_WARNING, "Target latency option will be ignored as streaming is not enabled\n");
1419 c->target_latency = 0;
1420 }
1421
1422 if (c->global_sidx && !c->single_file) {
1423 av_log(s, AV_LOG_WARNING, "Global SIDX option will be ignored as single_file is not enabled\n");
1424 c->global_sidx = 0;
1425 }
1426
1427 if (c->global_sidx && c->streaming) {
1428 av_log(s, AV_LOG_WARNING, "Global SIDX option will be ignored as streaming is enabled\n");
1429 c->global_sidx = 0;
1430 }
1431 if (c->frag_type == FRAG_TYPE_NONE && c->streaming) {
1432 av_log(s, AV_LOG_VERBOSE, "Changing frag_type from none to every_frame as streaming is enabled\n");
1433 c->frag_type = FRAG_TYPE_EVERY_FRAME;
1434 }
1435
1436 if (c->write_prft < 0) {
1437 c->write_prft = c->ldash;
1438 if (c->ldash)
1439 av_log(s, AV_LOG_VERBOSE, "Enabling Producer Reference Time element for Low Latency mode\n");
1440 }
1441
1442 if (c->write_prft && !c->utc_timing_url) {
1443 av_log(s, AV_LOG_WARNING, "Producer Reference Time element option will be ignored as utc_timing_url is not set\n");
1444 c->write_prft = 0;
1445 }
1446
1447 if (c->write_prft && !c->streaming) {
1448 av_log(s, AV_LOG_WARNING, "Producer Reference Time element option will be ignored as streaming is not enabled\n");
1449 c->write_prft = 0;
1450 }
1451
1452 if (c->ldash && !c->write_prft) {
1453 av_log(s, AV_LOG_WARNING, "Low Latency mode enabled without Producer Reference Time element option! Resulting manifest may not be complaint\n");
1454 }
1455
1456 if (c->target_latency && !c->write_prft) {
1457 av_log(s, AV_LOG_WARNING, "Target latency option will be ignored as Producer Reference Time element will not be written\n");
1458 c->target_latency = 0;
1459 }
1460
1461 if (av_cmp_q(c->max_playback_rate, c->min_playback_rate) < 0) {
1462 av_log(s, AV_LOG_WARNING, "Minimum playback rate value is higer than the Maximum. Both will be ignored\n");
1463 c->min_playback_rate = c->max_playback_rate = (AVRational) {1, 1};
1464 }
1465
1466 av_strlcpy(c->dirname, s->url, sizeof(c->dirname));
1467 ptr = strrchr(c->dirname, '/');
1468 if (ptr) {
1469 av_strlcpy(basename, &ptr[1], sizeof(basename));
1470 ptr[1] = '\0';
1471 } else {
1472 c->dirname[0] = '\0';
1473 av_strlcpy(basename, s->url, sizeof(basename));
1474 }
1475
1476 ptr = strrchr(basename, '.');
1477 if (ptr)
1478 *ptr = '\0';
1479
1480 c->streams = av_mallocz(sizeof(*c->streams) * s->nb_streams);
1481 if (!c->streams)
1482 return AVERROR(ENOMEM);
1483
1484 if ((ret = parse_adaptation_sets(s)) < 0)
1485 return ret;
1486
1487 if ((ret = init_segment_types(s)) < 0)
1488 return ret;
1489
1490 for (i = 0; i < s->nb_streams; i++) {
1491 OutputStream *os = &c->streams[i];
1492 AdaptationSet *as = &c->as[os->as_idx - 1];
1493 AVFormatContext *ctx;
1494 AVStream *st;
1495 AVDictionary *opts = NULL;
1496 char filename[1024];
1497
1498 os->bit_rate = s->streams[i]->codecpar->bit_rate;
1499 if (!os->bit_rate) {
1500 int level = s->strict_std_compliance >= FF_COMPLIANCE_STRICT ?
1501 AV_LOG_ERROR : AV_LOG_WARNING;
1502 av_log(s, level, "No bit rate set for stream %d\n", i);
1503 if (s->strict_std_compliance >= FF_COMPLIANCE_STRICT)
1504 return AVERROR(EINVAL);
1505 }
1506
1507 // copy AdaptationSet language and role from stream metadata
1508 dict_copy_entry(&as->metadata, s->streams[i]->metadata, "language");
1509 dict_copy_entry(&as->metadata, s->streams[i]->metadata, "role");
1510
1511 if (c->init_seg_name) {
1512 os->init_seg_name = av_strireplace(c->init_seg_name, "$ext$", os->extension_name);
1513 if (!os->init_seg_name)
1514 return AVERROR(ENOMEM);
1515 }
1516 if (c->media_seg_name) {
1517 os->media_seg_name = av_strireplace(c->media_seg_name, "$ext$", os->extension_name);
1518 if (!os->media_seg_name)
1519 return AVERROR(ENOMEM);
1520 }
1521 if (c->single_file_name) {
1522 os->single_file_name = av_strireplace(c->single_file_name, "$ext$", os->extension_name);
1523 if (!os->single_file_name)
1524 return AVERROR(ENOMEM);
1525 }
1526
1527 if (os->segment_type == SEGMENT_TYPE_WEBM) {
1528 if ((!c->single_file && !av_match_ext(os->init_seg_name, os->format_name)) ||
1529 (!c->single_file && !av_match_ext(os->media_seg_name, os->format_name)) ||
1530 ( c->single_file && !av_match_ext(os->single_file_name, os->format_name))) {
1531 av_log(s, AV_LOG_WARNING,
1532 "One or many segment file names doesn't end with .webm. "
1533 "Override -init_seg_name and/or -media_seg_name and/or "
1534 "-single_file_name to end with the extension .webm\n");
1535 }
1536 if (c->streaming) {
1537 // Streaming not supported as matroskaenc buffers internally before writing the output
1538 av_log(s, AV_LOG_WARNING, "One or more streams in WebM output format. Streaming option will be ignored\n");
1539 c->streaming = 0;
1540 }
1541 }
1542
1543 os->ctx = ctx = avformat_alloc_context();
1544 if (!ctx)
1545 return AVERROR(ENOMEM);
1546
1547 ctx->oformat = av_guess_format(os->format_name, NULL, NULL);
1548 if (!ctx->oformat)
1549 return AVERROR_MUXER_NOT_FOUND;
1550 ctx->interrupt_callback = s->interrupt_callback;
1551 ctx->opaque = s->opaque;
1552 ctx->io_close = s->io_close;
1553 ctx->io_close2 = s->io_close2;
1554 ctx->io_open = s->io_open;
1555 ctx->strict_std_compliance = s->strict_std_compliance;
1556
1557 if (!(st = avformat_new_stream(ctx, NULL)))
1558 return AVERROR(ENOMEM);
1559 avcodec_parameters_copy(st->codecpar, s->streams[i]->codecpar);
1560 st->sample_aspect_ratio = s->streams[i]->sample_aspect_ratio;
1561 st->time_base = s->streams[i]->time_base;
1562 st->avg_frame_rate = s->streams[i]->avg_frame_rate;
1563 ctx->avoid_negative_ts = s->avoid_negative_ts;
1564 ctx->flags = s->flags;
1565
1566 os->parser = av_parser_init(st->codecpar->codec_id);
1567 if (os->parser) {
1568 os->parser_avctx = avcodec_alloc_context3(NULL);
1569 if (!os->parser_avctx)
1570 return AVERROR(ENOMEM);
1571 ret = avcodec_parameters_to_context(os->parser_avctx, st->codecpar);
1572 if (ret < 0)
1573 return ret;
1574 // We only want to parse frame headers
1575 os->parser->flags |= PARSER_FLAG_COMPLETE_FRAMES;
1576 }
1577
1578 if (c->single_file) {
1579 if (os->single_file_name)
1580 ff_dash_fill_tmpl_params(os->initfile, sizeof(os->initfile), os->single_file_name, i, 0, os->bit_rate, 0);
1581 else
1582 snprintf(os->initfile, sizeof(os->initfile), "%s-stream%d.%s", basename, i, os->format_name);
1583 } else {
1584 ff_dash_fill_tmpl_params(os->initfile, sizeof(os->initfile), os->init_seg_name, i, 0, os->bit_rate, 0);
1585 }
1586 snprintf(filename, sizeof(filename), "%s%s", c->dirname, os->initfile);
1587 set_http_options(&opts, c);
1588 if (!c->single_file) {
1589 if ((ret = avio_open_dyn_buf(&ctx->pb)) < 0)
1590 return ret;
1591 ret = s->io_open(s, &os->out, filename, AVIO_FLAG_WRITE, &opts);
1592 } else {
1593 ctx->url = av_strdup(filename);
1594 ret = avio_open2(&ctx->pb, filename, AVIO_FLAG_WRITE, NULL, &opts);
1595 }
1596 av_dict_free(&opts);
1597 if (ret < 0)
1598 return ret;
1599 os->init_start_pos = 0;
1600
1601 av_dict_copy(&opts, c->format_options, 0);
1602 if (!as->seg_duration)
1603 as->seg_duration = c->seg_duration;
1604 if (!as->frag_duration)
1605 as->frag_duration = c->frag_duration;
1606 if (as->frag_type < 0)
1607 as->frag_type = c->frag_type;
1608 os->seg_duration = as->seg_duration;
1609 os->frag_duration = as->frag_duration;
1610 os->frag_type = as->frag_type;
1611
1612 c->max_segment_duration = FFMAX(c->max_segment_duration, as->seg_duration);
1613
1614 if (c->profile & MPD_PROFILE_DVB && (os->seg_duration > 15000000 || os->seg_duration < 960000)) {
1615 av_log(s, AV_LOG_ERROR, "Segment duration %"PRId64" is outside the allowed range for DVB-DASH profile\n", os->seg_duration);
1616 return AVERROR(EINVAL);
1617 }
1618
1619 if (os->frag_type == FRAG_TYPE_DURATION && !os->frag_duration) {
1620 av_log(s, AV_LOG_WARNING, "frag_type set to duration for stream %d but no frag_duration set\n", i);
1621 os->frag_type = c->streaming ? FRAG_TYPE_EVERY_FRAME : FRAG_TYPE_NONE;
1622 }
1623 if (os->frag_type == FRAG_TYPE_DURATION && os->frag_duration > os->seg_duration) {
1624 av_log(s, AV_LOG_ERROR, "Fragment duration %"PRId64" is longer than Segment duration %"PRId64"\n", os->frag_duration, os->seg_duration);
1625 return AVERROR(EINVAL);
1626 }
1627 if (os->frag_type == FRAG_TYPE_PFRAMES && (st->codecpar->codec_type != AVMEDIA_TYPE_VIDEO || !os->parser)) {
1628 if (st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO && !os->parser)
1629 av_log(s, AV_LOG_WARNING, "frag_type set to P-Frame reordering, but no parser found for stream %d\n", i);
1630 os->frag_type = c->streaming ? FRAG_TYPE_EVERY_FRAME : FRAG_TYPE_NONE;
1631 }
1632 if (os->frag_type != FRAG_TYPE_PFRAMES && as->trick_idx < 0)
1633 // Set this now if a parser isn't used
1634 os->coding_dependency = 1;
1635
1636 if (os->segment_type == SEGMENT_TYPE_MP4) {
1637 if (c->streaming)
1638 // skip_sidx : Reduce bitrate overhead
1639 // skip_trailer : Avoids growing memory usage with time
1640 av_dict_set(&opts, "movflags", "+dash+delay_moov+skip_sidx+skip_trailer", AV_DICT_APPEND);
1641 else {
1642 if (c->global_sidx)
1643 av_dict_set(&opts, "movflags", "+dash+delay_moov+global_sidx+skip_trailer", AV_DICT_APPEND);
1644 else
1645 av_dict_set(&opts, "movflags", "+dash+delay_moov+skip_trailer", AV_DICT_APPEND);
1646 }
1647 if (os->frag_type == FRAG_TYPE_EVERY_FRAME)
1648 av_dict_set(&opts, "movflags", "+frag_every_frame", AV_DICT_APPEND);
1649 else
1650 av_dict_set(&opts, "movflags", "+frag_custom", AV_DICT_APPEND);
1651 if (os->frag_type == FRAG_TYPE_DURATION)
1652 av_dict_set_int(&opts, "frag_duration", os->frag_duration, 0);
1653 if (c->write_prft)
1654 av_dict_set(&opts, "write_prft", "wallclock", 0);
1655 } else {
1656 av_dict_set_int(&opts, "cluster_time_limit", c->seg_duration / 1000, 0);
1657 av_dict_set_int(&opts, "cluster_size_limit", 5 * 1024 * 1024, 0); // set a large cluster size limit
1658 av_dict_set_int(&opts, "dash", 1, 0);
1659 av_dict_set_int(&opts, "dash_track_number", i + 1, 0);
1660 av_dict_set_int(&opts, "live", 1, 0);
1661 }
1662 ret = avformat_init_output(ctx, &opts);
1663 av_dict_free(&opts);
1664 if (ret < 0)
1665 return ret;
1666 os->ctx_inited = 1;
1667 avio_flush(ctx->pb);
1668
1669 av_log(s, AV_LOG_VERBOSE, "Representation %d init segment will be written to: %s\n", i, filename);
1670
1671 s->streams[i]->time_base = st->time_base;
1672 // If the muxer wants to shift timestamps, request to have them shifted
1673 // already before being handed to this muxer, so we don't have mismatches
1674 // between the MPD and the actual segments.
1675 s->avoid_negative_ts = ctx->avoid_negative_ts;
1676 if (st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) {
1677 AVRational avg_frame_rate = s->streams[i]->avg_frame_rate;
1678 AVRational par;
1679 if (avg_frame_rate.num > 0) {
1680 if (av_cmp_q(avg_frame_rate, as->min_frame_rate) < 0)
1681 as->min_frame_rate = avg_frame_rate;
1682 if (av_cmp_q(as->max_frame_rate, avg_frame_rate) < 0)
1683 as->max_frame_rate = avg_frame_rate;
1684 } else {
1685 as->ambiguous_frame_rate = 1;
1686 }
1687
1688 if (st->codecpar->width > as->max_width)
1689 as->max_width = st->codecpar->width;
1690 if (st->codecpar->height > as->max_height)
1691 as->max_height = st->codecpar->height;
1692
1693 if (st->sample_aspect_ratio.num)
1694 os->sar = st->sample_aspect_ratio;
1695 else
1696 os->sar = (AVRational){1,1};
1697 av_reduce(&par.num, &par.den,
1698 st->codecpar->width * (int64_t)os->sar.num,
1699 st->codecpar->height * (int64_t)os->sar.den,
1700 1024 * 1024);
1701
1702 if (as->par.num && av_cmp_q(par, as->par)) {
1703 av_log(s, AV_LOG_ERROR, "Conflicting stream aspect ratios values in Adaptation Set %d. Please ensure all adaptation sets have the same aspect ratio\n", os->as_idx);
1704 return AVERROR(EINVAL);
1705 }
1706 as->par = par;
1707
1708 c->has_video = 1;
1709 }
1710
1711 set_codec_str(s, st->codecpar, &st->avg_frame_rate, os->codec_str,
1712 sizeof(os->codec_str));
1713 os->first_pts = AV_NOPTS_VALUE;
1714 os->max_pts = AV_NOPTS_VALUE;
1715 os->last_dts = AV_NOPTS_VALUE;
1716 os->segment_index = 1;
1717
1718 if (s->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO)
1719 c->nr_of_streams_to_flush++;
1720 }
1721
1722 if (!c->has_video && c->seg_duration <= 0) {
1723 av_log(s, AV_LOG_WARNING, "no video stream and no seg duration set\n");
1724 return AVERROR(EINVAL);
1725 }
1726 if (!c->has_video && c->frag_type == FRAG_TYPE_PFRAMES)
1727 av_log(s, AV_LOG_WARNING, "no video stream and P-frame fragmentation set\n");
1728
1729 c->nr_of_streams_flushed = 0;
1730 c->target_latency_refid = -1;
1731
1732 return 0;
1733 }
1734
1735 static int dash_write_header(AVFormatContext *s)
1736 {
1737 DASHContext *c = s->priv_data;
1738 int i, ret;
1739 for (i = 0; i < s->nb_streams; i++) {
1740 OutputStream *os = &c->streams[i];
1741 if ((ret = avformat_write_header(os->ctx, NULL)) < 0)
1742 return ret;
1743
1744 // Flush init segment
1745 // Only for WebM segment, since for mp4 delay_moov is set and
1746 // the init segment is thus flushed after the first packets.
1747 if (os->segment_type == SEGMENT_TYPE_WEBM &&
1748 (ret = flush_init_segment(s, os)) < 0)
1749 return ret;
1750 }
1751 return ret;
1752 }
1753
1754 static int add_segment(OutputStream *os, const char *file,
1755 int64_t time, int64_t duration,
1756 int64_t start_pos, int64_t range_length,
1757 int64_t index_length, int next_exp_index)
1758 {
1759 int err;
1760 Segment *seg;
1761 if (os->nb_segments >= os->segments_size) {
1762 os->segments_size = (os->segments_size + 1) * 2;
1763 if ((err = av_reallocp_array(&os->segments, sizeof(*os->segments),
1764 os->segments_size)) < 0) {
1765 os->segments_size = 0;
1766 os->nb_segments = 0;
1767 return err;
1768 }
1769 }
1770 seg = av_mallocz(sizeof(*seg));
1771 if (!seg)
1772 return AVERROR(ENOMEM);
1773 av_strlcpy(seg->file, file, sizeof(seg->file));
1774 seg->time = time;
1775 seg->duration = duration;
1776 if (seg->time < 0) { // If pts<0, it is expected to be cut away with an edit list
1777 seg->duration += seg->time;
1778 seg->time = 0;
1779 }
1780 seg->start_pos = start_pos;
1781 seg->range_length = range_length;
1782 seg->index_length = index_length;
1783 os->segments[os->nb_segments++] = seg;
1784 os->segment_index++;
1785 //correcting the segment index if it has fallen behind the expected value
1786 if (os->segment_index < next_exp_index) {
1787 av_log(NULL, AV_LOG_WARNING, "Correcting the segment index after file %s: current=%d corrected=%d\n",
1788 file, os->segment_index, next_exp_index);
1789 os->segment_index = next_exp_index;
1790 }
1791 return 0;
1792 }
1793
1794 static void write_styp(AVIOContext *pb)
1795 {
1796 avio_wb32(pb, 24);
1797 ffio_wfourcc(pb, "styp");
1798 ffio_wfourcc(pb, "msdh");
1799 avio_wb32(pb, 0); /* minor */
1800 ffio_wfourcc(pb, "msdh");
1801 ffio_wfourcc(pb, "msix");
1802 }
1803
1804 static void find_index_range(AVFormatContext *s, const char *full_path,
1805 int64_t pos, int *index_length)
1806 {
1807 uint8_t buf[8];
1808 AVIOContext *pb;
1809 int ret;
1810
1811 ret = s->io_open(s, &pb, full_path, AVIO_FLAG_READ, NULL);
1812 if (ret < 0)
1813 return;
1814 if (avio_seek(pb, pos, SEEK_SET) != pos) {
1815 ff_format_io_close(s, &pb);
1816 return;
1817 }
1818 ret = avio_read(pb, buf, 8);
1819 ff_format_io_close(s, &pb);
1820 if (ret < 8)
1821 return;
1822 if (AV_RL32(&buf[4]) != MKTAG('s', 'i', 'd', 'x'))
1823 return;
1824 *index_length = AV_RB32(&buf[0]);
1825 }
1826
1827 static int update_stream_extradata(AVFormatContext *s, OutputStream *os,
1828 AVPacket *pkt, AVRational *frame_rate)
1829 {
1830 AVCodecParameters *par = os->ctx->streams[0]->codecpar;
1831 uint8_t *extradata;
1832 size_t extradata_size;
1833 int ret;
1834
1835 if (par->extradata_size)
1836 return 0;
1837
1838 extradata = av_packet_get_side_data(pkt, AV_PKT_DATA_NEW_EXTRADATA, &extradata_size);
1839 if (!extradata_size)
1840 return 0;
1841
1842 ret = ff_alloc_extradata(par, extradata_size);
1843 if (ret < 0)
1844 return ret;
1845
1846 memcpy(par->extradata, extradata, extradata_size);
1847
1848 set_codec_str(s, par, frame_rate, os->codec_str, sizeof(os->codec_str));
1849
1850 return 0;
1851 }
1852
1853 static void dashenc_delete_file(AVFormatContext *s, char *filename) {
1854 DASHContext *c = s->priv_data;
1855 int http_base_proto = ff_is_http_proto(filename);
1856
1857 if (http_base_proto) {
1858 AVIOContext *out = NULL;
1859 AVDictionary *http_opts = NULL;
1860
1861 set_http_options(&http_opts, c);
1862 av_dict_set(&http_opts, "method", "DELETE", 0);
1863
1864 if (dashenc_io_open(s, &out, filename, &http_opts) < 0) {
1865 av_log(s, AV_LOG_ERROR, "failed to delete %s\n", filename);
1866 }
1867
1868 av_dict_free(&http_opts);
1869 ff_format_io_close(s, &out);
1870 } else {
1871 int res = ffurl_delete(filename);
1872 if (res < 0) {
1873 char errbuf[AV_ERROR_MAX_STRING_SIZE];
1874 av_strerror(res, errbuf, sizeof(errbuf));
1875 av_log(s, (res == AVERROR(ENOENT) ? AV_LOG_WARNING : AV_LOG_ERROR), "failed to delete %s: %s\n", filename, errbuf);
1876 }
1877 }
1878 }
1879
1880 static int dashenc_delete_segment_file(AVFormatContext *s, const char* file)
1881 {
1882 DASHContext *c = s->priv_data;
1883 AVBPrint buf;
1884
1885 av_bprint_init(&buf, 0, AV_BPRINT_SIZE_UNLIMITED);
1886
1887 av_bprintf(&buf, "%s%s", c->dirname, file);
1888 if (!av_bprint_is_complete(&buf)) {
1889 av_bprint_finalize(&buf, NULL);
1890 av_log(s, AV_LOG_WARNING, "Out of memory for filename\n");
1891 return AVERROR(ENOMEM);
1892 }
1893
1894 dashenc_delete_file(s, buf.str);
1895
1896 av_bprint_finalize(&buf, NULL);
1897 return 0;
1898 }
1899
1900 static inline void dashenc_delete_media_segments(AVFormatContext *s, OutputStream *os, int remove_count)
1901 {
1902 for (int i = 0; i < remove_count; ++i) {
1903 dashenc_delete_segment_file(s, os->segments[i]->file);
1904
1905 // Delete the segment regardless of whether the file was successfully deleted
1906 av_free(os->segments[i]);
1907 }
1908
1909 os->nb_segments -= remove_count;
1910 memmove(os->segments, os->segments + remove_count, os->nb_segments * sizeof(*os->segments));
1911 }
1912
1913 static int dash_flush(AVFormatContext *s, int final, int stream)
1914 {
1915 DASHContext *c = s->priv_data;
1916 int i, ret = 0;
1917
1918 const char *proto = avio_find_protocol_name(s->url);
1919 int use_rename = proto && !strcmp(proto, "file");
1920
1921 int cur_flush_segment_index = 0, next_exp_index = -1;
1922 if (stream >= 0) {
1923 cur_flush_segment_index = c->streams[stream].segment_index;
1924
1925 //finding the next segment's expected index, based on the current pts value
1926 if (c->use_template && !c->use_timeline && c->index_correction &&
1927 c->streams[stream].last_pts != AV_NOPTS_VALUE &&
1928 c->streams[stream].first_pts != AV_NOPTS_VALUE) {
1929 int64_t pts_diff = av_rescale_q(c->streams[stream].last_pts -
1930 c->streams[stream].first_pts,
1931 s->streams[stream]->time_base,
1932 AV_TIME_BASE_Q);
1933 next_exp_index = (pts_diff / c->streams[stream].seg_duration) + 1;
1934 }
1935 }
1936
1937 for (i = 0; i < s->nb_streams; i++) {
1938 OutputStream *os = &c->streams[i];
1939 AVStream *st = s->streams[i];
1940 int range_length, index_length = 0;
1941 int64_t duration;
1942
1943 if (!os->packets_written)
1944 continue;
1945
1946 // Flush the single stream that got a keyframe right now.
1947 // Flush all audio streams as well, in sync with video keyframes,
1948 // but not the other video streams.
1949 if (stream >= 0 && i != stream) {
1950 if (s->streams[stream]->codecpar->codec_type != AVMEDIA_TYPE_VIDEO &&
1951 s->streams[i]->codecpar->codec_type != AVMEDIA_TYPE_VIDEO)
1952 continue;
1953 if (s->streams[i]->codecpar->codec_type != AVMEDIA_TYPE_AUDIO)
1954 continue;
1955 // Make sure we don't flush audio streams multiple times, when
1956 // all video streams are flushed one at a time.
1957 if (c->has_video && os->segment_index > cur_flush_segment_index)
1958 continue;
1959 }
1960
1961 if (c->single_file)
1962 snprintf(os->full_path, sizeof(os->full_path), "%s%s", c->dirname, os->initfile);
1963
1964 ret = flush_dynbuf(c, os, &range_length);
1965 if (ret < 0)
1966 break;
1967 os->packets_written = 0;
1968
1969 if (c->single_file) {
1970 find_index_range(s, os->full_path, os->pos, &index_length);
1971 } else {
1972 dashenc_io_close(s, &os->out, os->temp_path);
1973
1974 if (use_rename) {
1975 ret = ff_rename(os->temp_path, os->full_path, os->ctx);
1976 if (ret < 0)
1977 break;
1978 }
1979 }
1980
1981 duration = av_rescale_q(os->max_pts - os->start_pts, st->time_base, AV_TIME_BASE_Q);
1982 os->last_duration = FFMAX(os->last_duration, duration);
1983
1984 if (!os->muxer_overhead && os->max_pts > os->start_pts)
1985 os->muxer_overhead = ((int64_t) (range_length - os->total_pkt_size) *
1986 8 * AV_TIME_BASE) / duration;
1987 os->total_pkt_size = 0;
1988 os->total_pkt_duration = 0;
1989
1990 if (!os->bit_rate && !os->first_segment_bit_rate) {
1991 os->first_segment_bit_rate = (int64_t) range_length * 8 * AV_TIME_BASE / duration;
1992 }
1993 add_segment(os, os->filename, os->start_pts, os->max_pts - os->start_pts, os->pos, range_length, index_length, next_exp_index);
1994 av_log(s, AV_LOG_VERBOSE, "Representation %d media segment %d written to: %s\n", i, os->segment_index, os->full_path);
1995
1996 os->pos += range_length;
1997 }
1998
1999 if (c->window_size) {
2000 for (i = 0; i < s->nb_streams; i++) {
2001 OutputStream *os = &c->streams[i];
2002 int remove_count = os->nb_segments - c->window_size - c->extra_window_size;
2003 if (remove_count > 0)
2004 dashenc_delete_media_segments(s, os, remove_count);
2005 }
2006 }
2007
2008 if (final) {
2009 for (i = 0; i < s->nb_streams; i++) {
2010 OutputStream *os = &c->streams[i];
2011 if (os->ctx && os->ctx_inited) {
2012 int64_t file_size = avio_tell(os->ctx->pb);
2013 av_write_trailer(os->ctx);
2014 if (c->global_sidx) {
2015 int j, start_index, start_number;
2016 int64_t sidx_size = avio_tell(os->ctx->pb) - file_size;
2017 get_start_index_number(os, c, &start_index, &start_number);
2018 if (start_index >= os->nb_segments ||
2019 os->segment_type != SEGMENT_TYPE_MP4)
2020 continue;
2021 os->init_range_length += sidx_size;
2022 for (j = start_index; j < os->nb_segments; j++) {
2023 Segment *seg = os->segments[j];
2024 seg->start_pos += sidx_size;
2025 }
2026 }
2027
2028 }
2029 }
2030 }
2031 if (ret >= 0) {
2032 if (c->has_video && !final) {
2033 c->nr_of_streams_flushed++;
2034 if (c->nr_of_streams_flushed != c->nr_of_streams_to_flush)
2035 return ret;
2036
2037 c->nr_of_streams_flushed = 0;
2038 }
2039 // In streaming mode the manifest is written at the beginning
2040 // of the segment instead
2041 if (!c->streaming || final)
2042 ret = write_manifest(s, final);
2043 }
2044 return ret;
2045