7 #include "packager/mpd/base/mpd_builder.h"
9 #include <libxml/tree.h>
10 #include <libxml/xmlstring.h>
18 #include "packager/base/base64.h"
19 #include "packager/base/bind.h"
20 #include "packager/base/files/file_path.h"
21 #include "packager/base/logging.h"
22 #include "packager/base/strings/string_number_conversions.h"
23 #include "packager/base/strings/stringprintf.h"
24 #include "packager/base/synchronization/lock.h"
25 #include "packager/base/time/default_clock.h"
26 #include "packager/base/time/time.h"
27 #include "packager/media/base/language_utils.h"
28 #include "packager/media/file/file.h"
29 #include "packager/mpd/base/content_protection_element.h"
30 #include "packager/mpd/base/mpd_utils.h"
31 #include "packager/mpd/base/xml/xml_node.h"
32 #include "packager/version/version.h"
38 using xml::RepresentationXmlNode;
39 using xml::AdaptationSetXmlNode;
43 AdaptationSet::Role MediaInfoTextTypeToRole(
44 MediaInfo::TextInfo::TextType type) {
46 case MediaInfo::TextInfo::UNKNOWN:
47 LOG(WARNING) <<
"Unknown text type, assuming subtitle.";
48 return AdaptationSet::kRoleSubtitle;
49 case MediaInfo::TextInfo::CAPTION:
50 return AdaptationSet::kRoleCaption;
51 case MediaInfo::TextInfo::SUBTITLE:
52 return AdaptationSet::kRoleSubtitle;
54 NOTREACHED() <<
"Unknown MediaInfo TextType: " << type
55 <<
" assuming subtitle.";
56 return AdaptationSet::kRoleSubtitle;
60 std::string GetMimeType(
const std::string& prefix,
61 MediaInfo::ContainerType container_type) {
62 switch (container_type) {
63 case MediaInfo::CONTAINER_MP4:
64 return prefix +
"/mp4";
65 case MediaInfo::CONTAINER_MPEG2_TS:
67 return prefix +
"/MP2T";
68 case MediaInfo::CONTAINER_WEBM:
69 return prefix +
"/webm";
75 LOG(ERROR) <<
"Unrecognized container type: " << container_type;
79 void AddMpdNameSpaceInfo(XmlNode* mpd) {
82 static const char kXmlNamespace[] =
"urn:mpeg:dash:schema:mpd:2011";
83 static const char kXmlNamespaceXsi[] =
84 "http://www.w3.org/2001/XMLSchema-instance";
85 static const char kXmlNamespaceXlink[] =
"http://www.w3.org/1999/xlink";
86 static const char kDashSchemaMpd2011[] =
87 "urn:mpeg:dash:schema:mpd:2011 DASH-MPD.xsd";
88 static const char kCencNamespace[] =
"urn:mpeg:cenc:2013";
90 mpd->SetStringAttribute(
"xmlns", kXmlNamespace);
91 mpd->SetStringAttribute(
"xmlns:xsi", kXmlNamespaceXsi);
92 mpd->SetStringAttribute(
"xmlns:xlink", kXmlNamespaceXlink);
93 mpd->SetStringAttribute(
"xsi:schemaLocation", kDashSchemaMpd2011);
94 mpd->SetStringAttribute(
"xmlns:cenc", kCencNamespace);
97 bool IsPeriodNode(xmlNodePtr node) {
100 return xmlStrcmp(node->name, reinterpret_cast<const xmlChar*>(
"Period")) ==
109 xmlNodePtr FindPeriodNode(XmlNode* xml_node) {
110 for (xmlNodePtr node = xml_node->GetRawPtr()->xmlChildrenNode; node != NULL;
112 if (IsPeriodNode(node))
119 bool Positive(
double d) {
125 std::string XmlDateTimeNowWithOffset(
126 int32_t offset_seconds,
127 base::Clock* clock) {
128 base::Time time = clock->Now();
129 time += base::TimeDelta::FromSeconds(offset_seconds);
130 base::Time::Exploded time_exploded;
131 time.UTCExplode(&time_exploded);
133 return base::StringPrintf(
"%4d-%02d-%02dT%02d:%02d:%02dZ", time_exploded.year,
134 time_exploded.month, time_exploded.day_of_month,
135 time_exploded.hour, time_exploded.minute,
136 time_exploded.second);
139 void SetIfPositive(
const char* attr_name,
double value, XmlNode* mpd) {
140 if (Positive(value)) {
141 mpd->SetStringAttribute(attr_name, SecondsToXmlDuration(value));
145 uint32_t GetTimeScale(
const MediaInfo& media_info) {
146 if (media_info.has_reference_time_scale()) {
147 return media_info.reference_time_scale();
150 if (media_info.has_video_info()) {
151 return media_info.video_info().time_scale();
154 if (media_info.has_audio_info()) {
155 return media_info.audio_info().time_scale();
158 LOG(WARNING) <<
"No timescale specified, using 1 as timescale.";
162 uint64_t LastSegmentStartTime(
const SegmentInfo& segment_info) {
163 return segment_info.start_time + segment_info.duration * segment_info.repeat;
167 uint64_t LastSegmentEndTime(
const SegmentInfo& segment_info) {
168 return segment_info.start_time +
169 segment_info.duration * (segment_info.repeat + 1);
172 uint64_t LatestSegmentStartTime(
const std::list<SegmentInfo>& segments) {
173 DCHECK(!segments.empty());
174 const SegmentInfo& latest_segment = segments.back();
175 return LastSegmentStartTime(latest_segment);
180 int SearchTimedOutRepeatIndex(uint64_t timeshift_limit,
181 const SegmentInfo& segment_info) {
182 DCHECK_LE(timeshift_limit, LastSegmentEndTime(segment_info));
183 if (timeshift_limit < segment_info.start_time)
186 return (timeshift_limit - segment_info.start_time) / segment_info.duration;
189 std::string MakePathRelative(
const std::string& path,
190 const std::string& mpd_dir) {
191 return (path.find(mpd_dir) == 0) ? path.substr(mpd_dir.size()) : path;
197 bool HasRequiredVideoFields(
const MediaInfo_VideoInfo& video_info) {
198 if (!video_info.has_height() || !video_info.has_width()) {
200 <<
"Width and height are required fields for generating a valid MPD.";
205 LOG_IF(WARNING, !video_info.has_time_scale())
206 <<
"Video info does not contain timescale required for "
207 "calculating framerate. @frameRate is required for DASH IOP.";
208 LOG_IF(WARNING, !video_info.has_pixel_width())
209 <<
"Video info does not contain pixel_width to calculate the sample "
210 "aspect ratio required for DASH IOP.";
211 LOG_IF(WARNING, !video_info.has_pixel_height())
212 <<
"Video info does not contain pixel_height to calculate the sample "
213 "aspect ratio required for DASH IOP.";
224 std::string GetPictureAspectRatio(uint32_t width,
226 uint32_t pixel_width,
227 uint32_t pixel_height) {
228 const uint32_t scaled_width = pixel_width * width;
229 const uint32_t scaled_height = pixel_height * height;
230 const double par =
static_cast<double>(scaled_width) / scaled_height;
234 const uint32_t kLargestPossibleParY = 19;
236 uint32_t par_num = 0;
237 uint32_t par_den = 0;
238 double min_error = 1.0;
239 for (uint32_t den = 1; den <= kLargestPossibleParY; ++den) {
240 uint32_t num = par * den + 0.5;
241 double error = fabs(par - static_cast<double>(num) / den);
242 if (error < min_error) {
246 if (error == 0)
break;
249 VLOG(2) <<
"width*pix_width : height*pixel_height (" << scaled_width <<
":"
250 << scaled_height <<
") reduced to " << par_num <<
":" << par_den
251 <<
" with error " << min_error <<
".";
253 return base::IntToString(par_num) +
":" + base::IntToString(par_den);
258 void AddPictureAspectRatio(
259 const MediaInfo::VideoInfo& video_info,
260 std::set<std::string>* picture_aspect_ratio) {
263 if (picture_aspect_ratio->size() > 1)
266 if (video_info.width() == 0 || video_info.height() == 0 ||
267 video_info.pixel_width() == 0 || video_info.pixel_height() == 0) {
272 picture_aspect_ratio->insert(
"bogus");
273 picture_aspect_ratio->insert(
"entries");
277 const std::string par = GetPictureAspectRatio(
278 video_info.width(), video_info.height(),
279 video_info.pixel_width(), video_info.pixel_height());
280 DVLOG(1) <<
"Setting par as: " << par
281 <<
" for video with width: " << video_info.width()
282 <<
" height: " << video_info.height()
283 <<
" pixel_width: " << video_info.pixel_width() <<
" pixel_height; "
284 << video_info.pixel_height();
285 picture_aspect_ratio->insert(par);
288 std::string RoleToText(AdaptationSet::Role role) {
292 case AdaptationSet::kRoleCaption:
294 case AdaptationSet::kRoleSubtitle:
296 case AdaptationSet::kRoleMain:
298 case AdaptationSet::kRoleAlternate:
300 case AdaptationSet::kRoleSupplementary:
301 return "supplementary";
302 case AdaptationSet::kRoleCommentary:
304 case AdaptationSet::kRoleDub:
315 class LibXmlInitializer {
317 LibXmlInitializer() : initialized_(false) {
318 base::AutoLock lock(lock_);
325 ~LibXmlInitializer() {
326 base::AutoLock lock(lock_);
329 initialized_ =
false;
337 DISALLOW_COPY_AND_ASSIGN(LibXmlInitializer);
340 class RepresentationStateChangeListenerImpl
341 :
public RepresentationStateChangeListener {
344 RepresentationStateChangeListenerImpl(uint32_t representation_id,
345 AdaptationSet* adaptation_set)
346 : representation_id_(representation_id), adaptation_set_(adaptation_set) {
347 DCHECK(adaptation_set_);
349 ~RepresentationStateChangeListenerImpl()
override {}
352 void OnNewSegmentForRepresentation(uint64_t start_time,
353 uint64_t duration)
override {
354 adaptation_set_->OnNewSegmentForRepresentation(representation_id_,
355 start_time, duration);
358 void OnSetFrameRateForRepresentation(uint32_t frame_duration,
359 uint32_t timescale)
override {
360 adaptation_set_->OnSetFrameRateForRepresentation(representation_id_,
361 frame_duration, timescale);
365 const uint32_t representation_id_;
366 AdaptationSet*
const adaptation_set_;
368 DISALLOW_COPY_AND_ASSIGN(RepresentationStateChangeListenerImpl);
374 : mpd_options_(mpd_options), clock_(new base::DefaultClock()) {}
376 MpdBuilder::~MpdBuilder() {}
379 base_urls_.push_back(base_url);
383 std::unique_ptr<AdaptationSet> adaptation_set(
384 new AdaptationSet(adaptation_set_counter_.GetNext(), lang, mpd_options_,
385 &representation_counter_));
386 DCHECK(adaptation_set);
388 if (!lang.empty() && lang == mpd_options_.default_language) {
389 adaptation_set->AddRole(AdaptationSet::kRoleMain);
392 adaptation_sets_.push_back(std::move(adaptation_set));
393 return adaptation_sets_.back().get();
398 static LibXmlInitializer lib_xml_initializer;
400 xml::scoped_xml_ptr<xmlDoc> doc(GenerateMpd());
404 static const int kNiceFormat = 1;
405 int doc_str_size = 0;
406 xmlChar* doc_str =
nullptr;
407 xmlDocDumpFormatMemoryEnc(doc.get(), &doc_str, &doc_str_size,
"UTF-8",
409 output->assign(doc_str, doc_str + doc_str_size);
417 xmlDocPtr MpdBuilder::GenerateMpd() {
419 static const char kXmlVersion[] =
"1.0";
420 xml::scoped_xml_ptr<xmlDoc> doc(xmlNewDoc(BAD_CAST kXmlVersion));
424 XmlNode period(
"Period");
430 for (
const std::unique_ptr<AdaptationSet>& adaptation_set :
432 xml::scoped_xml_ptr<xmlNode> child(adaptation_set->GetXml());
433 if (!child.get() || !period.AddChild(std::move(child)))
438 std::list<std::string>::const_iterator base_urls_it = base_urls_.begin();
439 for (; base_urls_it != base_urls_.end(); ++base_urls_it) {
440 XmlNode base_url(
"BaseURL");
441 base_url.SetContent(*base_urls_it);
443 if (!mpd.AddChild(base_url.PassScopedPtr()))
448 if (mpd_options_.mpd_type == MpdType::kDynamic) {
450 period.SetStringAttribute(
"start",
"PT0S");
453 if (!mpd.AddChild(period.PassScopedPtr()))
456 AddMpdNameSpaceInfo(&mpd);
458 static const char kOnDemandProfile[] =
459 "urn:mpeg:dash:profile:isoff-on-demand:2011";
460 static const char kLiveProfile[] =
461 "urn:mpeg:dash:profile:isoff-live:2011";
462 switch (mpd_options_.dash_profile) {
463 case DashProfile::kOnDemand:
464 mpd.SetStringAttribute(
"profiles", kOnDemandProfile);
466 case DashProfile::kLive:
467 mpd.SetStringAttribute(
"profiles", kLiveProfile);
470 NOTREACHED() <<
"Unknown DASH profile: "
471 <<
static_cast<int>(mpd_options_.dash_profile);
475 AddCommonMpdInfo(&mpd);
476 switch (mpd_options_.mpd_type) {
477 case MpdType::kStatic:
478 AddStaticMpdInfo(&mpd);
480 case MpdType::kDynamic:
481 AddDynamicMpdInfo(&mpd);
484 NOTREACHED() <<
"Unknown MPD type: "
485 <<
static_cast<int>(mpd_options_.mpd_type);
490 const std::string version = GetPackagerVersion();
491 if (!version.empty()) {
492 std::string version_string =
493 base::StringPrintf(
"Generated with %s version %s",
494 GetPackagerProjectUrl().c_str(), version.c_str());
495 xml::scoped_xml_ptr<xmlNode> comment(
496 xmlNewDocComment(doc.get(), BAD_CAST version_string.c_str()));
497 xmlDocSetRootElement(doc.get(), comment.get());
498 xmlAddSibling(comment.release(), mpd.Release());
500 xmlDocSetRootElement(doc.get(), mpd.Release());
502 return doc.release();
505 void MpdBuilder::AddCommonMpdInfo(XmlNode* mpd_node) {
506 if (Positive(mpd_options_.min_buffer_time)) {
507 mpd_node->SetStringAttribute(
508 "minBufferTime", SecondsToXmlDuration(mpd_options_.min_buffer_time));
510 LOG(ERROR) <<
"minBufferTime value not specified.";
515 void MpdBuilder::AddStaticMpdInfo(XmlNode* mpd_node) {
517 DCHECK_EQ(MpdType::kStatic, mpd_options_.mpd_type);
519 static const char kStaticMpdType[] =
"static";
520 mpd_node->SetStringAttribute(
"type", kStaticMpdType);
521 mpd_node->SetStringAttribute(
522 "mediaPresentationDuration",
523 SecondsToXmlDuration(GetStaticMpdDuration(mpd_node)));
526 void MpdBuilder::AddDynamicMpdInfo(XmlNode* mpd_node) {
528 DCHECK_EQ(MpdType::kDynamic, mpd_options_.mpd_type);
530 static const char kDynamicMpdType[] =
"dynamic";
531 mpd_node->SetStringAttribute(
"type", kDynamicMpdType);
534 mpd_node->SetStringAttribute(
"publishTime",
535 XmlDateTimeNowWithOffset(0, clock_.get()));
539 if (availability_start_time_.empty()) {
540 double earliest_presentation_time;
541 if (GetEarliestTimestamp(&earliest_presentation_time)) {
542 availability_start_time_ = XmlDateTimeNowWithOffset(
543 -std::ceil(earliest_presentation_time), clock_.get());
545 LOG(ERROR) <<
"Could not determine the earliest segment presentation "
546 "time for availabilityStartTime calculation.";
550 if (!availability_start_time_.empty())
551 mpd_node->SetStringAttribute(
"availabilityStartTime",
552 availability_start_time_);
554 if (Positive(mpd_options_.minimum_update_period)) {
555 mpd_node->SetStringAttribute(
556 "minimumUpdatePeriod",
557 SecondsToXmlDuration(mpd_options_.minimum_update_period));
559 LOG(WARNING) <<
"The profile is dynamic but no minimumUpdatePeriod "
563 SetIfPositive(
"timeShiftBufferDepth", mpd_options_.time_shift_buffer_depth,
565 SetIfPositive(
"suggestedPresentationDelay",
566 mpd_options_.suggested_presentation_delay, mpd_node);
569 float MpdBuilder::GetStaticMpdDuration(XmlNode* mpd_node) {
571 DCHECK_EQ(MpdType::kStatic, mpd_options_.mpd_type);
573 xmlNodePtr period_node = FindPeriodNode(mpd_node);
574 DCHECK(period_node) <<
"Period element must be a child of mpd_node.";
575 DCHECK(IsPeriodNode(period_node));
581 float max_duration = 0.0f;
582 for (xmlNodePtr adaptation_set = xmlFirstElementChild(period_node);
583 adaptation_set; adaptation_set = xmlNextElementSibling(adaptation_set)) {
584 for (xmlNodePtr representation = xmlFirstElementChild(adaptation_set);
586 representation = xmlNextElementSibling(representation)) {
587 float duration = 0.0f;
588 if (GetDurationAttribute(representation, &duration)) {
589 max_duration = max_duration > duration ? max_duration : duration;
593 xmlUnsetProp(representation, BAD_CAST
"duration");
601 bool MpdBuilder::GetEarliestTimestamp(
double* timestamp_seconds) {
602 DCHECK(timestamp_seconds);
604 double earliest_timestamp(-1);
605 for (
const std::unique_ptr<AdaptationSet>& adaptation_set :
608 if (adaptation_set->GetEarliestTimestamp(×tamp) &&
609 ((earliest_timestamp < 0) || (timestamp < earliest_timestamp))) {
610 earliest_timestamp = timestamp;
613 if (earliest_timestamp < 0)
616 *timestamp_seconds = earliest_timestamp;
621 MediaInfo* media_info) {
623 const std::string kFileProtocol(
"file://");
624 std::string mpd_file_path = (mpd_path.find(kFileProtocol) == 0)
625 ? mpd_path.substr(kFileProtocol.size())
628 if (!mpd_file_path.empty()) {
629 std::string mpd_dir(FilePath::FromUTF8Unsafe(mpd_file_path)
630 .DirName().AsEndingWithSeparator().AsUTF8Unsafe());
631 if (!mpd_dir.empty()) {
632 if (media_info->has_media_file_name()) {
633 media_info->set_media_file_name(
634 MakePathRelative(media_info->media_file_name(), mpd_dir));
636 if (media_info->has_init_segment_name()) {
637 media_info->set_init_segment_name(
638 MakePathRelative(media_info->init_segment_name(), mpd_dir));
640 if (media_info->has_segment_template()) {
641 media_info->set_segment_template(
642 MakePathRelative(media_info->segment_template(), mpd_dir));
649 const std::string& lang,
651 base::AtomicSequenceNumber* counter)
652 : representation_counter_(counter),
653 id_(adaptation_set_id),
655 mpd_options_(mpd_options),
656 segments_aligned_(kSegmentAlignmentUnknown),
657 force_set_segment_alignment_(false) {
661 AdaptationSet::~AdaptationSet() {}
664 const uint32_t representation_id = representation_counter_->GetNext();
667 std::unique_ptr<RepresentationStateChangeListener> listener(
668 new RepresentationStateChangeListenerImpl(representation_id,
this));
669 std::unique_ptr<Representation> representation(
new Representation(
670 media_info, mpd_options_, representation_id, std::move(listener)));
672 if (!representation->Init()) {
673 LOG(ERROR) <<
"Failed to initialize Representation.";
679 if (media_info.has_video_info()) {
680 const MediaInfo::VideoInfo& video_info = media_info.video_info();
681 DCHECK(video_info.has_width());
682 DCHECK(video_info.has_height());
683 video_widths_.insert(video_info.width());
684 video_heights_.insert(video_info.height());
686 if (video_info.has_time_scale() && video_info.has_frame_duration())
687 RecordFrameRate(video_info.frame_duration(), video_info.time_scale());
689 AddPictureAspectRatio(video_info, &picture_aspect_ratio_);
692 if (media_info.has_video_info()) {
693 content_type_ =
"video";
694 }
else if (media_info.has_audio_info()) {
695 content_type_ =
"audio";
696 }
else if (media_info.has_text_info()) {
697 content_type_ =
"text";
699 if (media_info.text_info().has_type() &&
700 (media_info.text_info().type() != MediaInfo::TextInfo::UNKNOWN)) {
701 roles_.insert(MediaInfoTextTypeToRole(media_info.text_info().type()));
705 representations_.push_back(std::move(representation));
706 return representations_.back().get();
711 content_protection_elements_.push_back(content_protection_element);
712 RemoveDuplicateAttributes(&content_protection_elements_.back());
716 const std::string& pssh) {
717 UpdateContentProtectionPsshHelper(drm_uuid, pssh,
718 &content_protection_elements_);
732 AdaptationSetXmlNode adaptation_set;
734 bool suppress_representation_width =
false;
735 bool suppress_representation_height =
false;
736 bool suppress_representation_frame_rate =
false;
738 adaptation_set.SetId(id_);
739 adaptation_set.SetStringAttribute(
"contentType", content_type_);
740 if (!lang_.empty() && lang_ !=
"und") {
745 if (video_widths_.size() == 1) {
746 suppress_representation_width =
true;
747 adaptation_set.SetIntegerAttribute(
"width", *video_widths_.begin());
748 }
else if (video_widths_.size() > 1) {
749 adaptation_set.SetIntegerAttribute(
"maxWidth", *video_widths_.rbegin());
751 if (video_heights_.size() == 1) {
752 suppress_representation_height =
true;
753 adaptation_set.SetIntegerAttribute(
"height", *video_heights_.begin());
754 }
else if (video_heights_.size() > 1) {
755 adaptation_set.SetIntegerAttribute(
"maxHeight", *video_heights_.rbegin());
758 if (video_frame_rates_.size() == 1) {
759 suppress_representation_frame_rate =
true;
760 adaptation_set.SetStringAttribute(
"frameRate",
761 video_frame_rates_.begin()->second);
762 }
else if (video_frame_rates_.size() > 1) {
763 adaptation_set.SetStringAttribute(
"maxFrameRate",
764 video_frame_rates_.rbegin()->second);
769 if (mpd_options_.dash_profile == DashProfile::kOnDemand) {
770 CheckVodSegmentAlignment();
773 if (segments_aligned_ == kSegmentAlignmentTrue) {
774 adaptation_set.SetStringAttribute(
775 mpd_options_.dash_profile == DashProfile::kOnDemand
776 ?
"subsegmentAlignment"
777 :
"segmentAlignment",
781 if (picture_aspect_ratio_.size() == 1)
782 adaptation_set.SetStringAttribute(
"par", *picture_aspect_ratio_.begin());
784 if (!adaptation_set.AddContentProtectionElements(
785 content_protection_elements_)) {
786 return xml::scoped_xml_ptr<xmlNode>();
789 if (!trick_play_reference_ids_.empty()) {
790 std::string id_string;
791 for (uint32_t
id : trick_play_reference_ids_) {
792 id_string += std::to_string(
id) +
",";
794 DCHECK(!id_string.empty());
795 id_string.resize(id_string.size() - 1);
796 adaptation_set.AddEssentialProperty(
797 "http://dashif.org/guidelines/trickmode", id_string);
800 std::string switching_ids;
801 for (uint32_t
id : adaptation_set_switching_ids_) {
802 if (!switching_ids.empty())
803 switching_ids +=
',';
804 switching_ids += base::UintToString(
id);
806 if (!switching_ids.empty()) {
807 adaptation_set.AddSupplementalProperty(
808 "urn:mpeg:dash:adaptation-set-switching:2016", switching_ids);
811 for (AdaptationSet::Role role : roles_)
812 adaptation_set.AddRoleElement(
"urn:mpeg:dash:role:2011", RoleToText(role));
814 for (
const std::unique_ptr<Representation>& representation :
816 if (suppress_representation_width)
817 representation->SuppressOnce(Representation::kSuppressWidth);
818 if (suppress_representation_height)
819 representation->SuppressOnce(Representation::kSuppressHeight);
820 if (suppress_representation_frame_rate)
821 representation->SuppressOnce(Representation::kSuppressFrameRate);
822 xml::scoped_xml_ptr<xmlNode> child(representation->GetXml());
823 if (!child || !adaptation_set.AddChild(std::move(child)))
824 return xml::scoped_xml_ptr<xmlNode>();
827 return adaptation_set.PassScopedPtr();
832 segment_alignment ? kSegmentAlignmentTrue : kSegmentAlignmentFalse;
833 force_set_segment_alignment_ =
true;
837 adaptation_set_switching_ids_.push_back(adaptation_set_id);
850 if (mpd_options_.dash_profile == DashProfile::kLive) {
851 CheckLiveSegmentAlignment(representation_id, start_time, duration);
853 representation_segment_start_times_[representation_id].push_back(
859 uint32_t representation_id,
860 uint32_t frame_duration,
861 uint32_t timescale) {
862 RecordFrameRate(frame_duration, timescale);
866 trick_play_reference_ids_.insert(
id);
869 bool AdaptationSet::GetEarliestTimestamp(
double* timestamp_seconds) {
870 DCHECK(timestamp_seconds);
872 double earliest_timestamp(-1);
873 for (
const std::unique_ptr<Representation>& representation :
876 if (representation->GetEarliestTimestamp(×tamp) &&
877 ((earliest_timestamp < 0) || (timestamp < earliest_timestamp))) {
878 earliest_timestamp = timestamp;
881 if (earliest_timestamp < 0)
884 *timestamp_seconds = earliest_timestamp;
912 void AdaptationSet::CheckLiveSegmentAlignment(uint32_t representation_id,
915 if (segments_aligned_ == kSegmentAlignmentFalse ||
916 force_set_segment_alignment_) {
920 std::list<uint64_t>& representation_start_times =
921 representation_segment_start_times_[representation_id];
922 representation_start_times.push_back(start_time);
925 if (representation_segment_start_times_.size() != representations_.size())
928 DCHECK(!representation_start_times.empty());
929 const uint64_t expected_start_time = representation_start_times.front();
930 for (RepresentationTimeline::const_iterator it =
931 representation_segment_start_times_.begin();
932 it != representation_segment_start_times_.end(); ++it) {
936 if (it->second.empty())
939 if (expected_start_time != it->second.front()) {
942 segments_aligned_ = kSegmentAlignmentFalse;
943 representation_segment_start_times_.clear();
947 segments_aligned_ = kSegmentAlignmentTrue;
949 for (RepresentationTimeline::iterator it =
950 representation_segment_start_times_.begin();
951 it != representation_segment_start_times_.end(); ++it) {
952 it->second.pop_front();
958 void AdaptationSet::CheckVodSegmentAlignment() {
959 if (segments_aligned_ == kSegmentAlignmentFalse ||
960 force_set_segment_alignment_) {
963 if (representation_segment_start_times_.empty())
965 if (representation_segment_start_times_.size() == 1) {
966 segments_aligned_ = kSegmentAlignmentTrue;
973 const std::list<uint64_t>& expected_time_line =
974 representation_segment_start_times_.begin()->second;
976 bool all_segment_time_line_same_length =
true;
978 RepresentationTimeline::const_iterator it =
979 representation_segment_start_times_.begin();
980 for (++it; it != representation_segment_start_times_.end(); ++it) {
981 const std::list<uint64_t>& other_time_line = it->second;
982 if (expected_time_line.size() != other_time_line.size()) {
983 all_segment_time_line_same_length =
false;
986 const std::list<uint64_t>* longer_list = &other_time_line;
987 const std::list<uint64_t>* shorter_list = &expected_time_line;
988 if (expected_time_line.size() > other_time_line.size()) {
989 shorter_list = &other_time_line;
990 longer_list = &expected_time_line;
993 if (!std::equal(shorter_list->begin(), shorter_list->end(),
994 longer_list->begin())) {
996 segments_aligned_ = kSegmentAlignmentFalse;
997 representation_segment_start_times_.clear();
1008 if (!all_segment_time_line_same_length) {
1009 segments_aligned_ = kSegmentAlignmentUnknown;
1013 segments_aligned_ = kSegmentAlignmentTrue;
1018 void AdaptationSet::RecordFrameRate(uint32_t frame_duration,
1019 uint32_t timescale) {
1020 if (frame_duration == 0) {
1021 LOG(ERROR) <<
"Frame duration is 0 and cannot be set.";
1024 video_frame_rates_[
static_cast<double>(timescale) / frame_duration] =
1025 base::IntToString(timescale) +
"/" + base::IntToString(frame_duration);
1029 const MediaInfo& media_info,
1032 std::unique_ptr<RepresentationStateChangeListener> state_change_listener)
1033 : media_info_(media_info),
1036 mpd_options_(mpd_options),
1038 state_change_listener_(std::move(state_change_listener)),
1039 output_suppression_flags_(0) {}
1041 Representation::~Representation() {}
1044 if (!AtLeastOneTrue(media_info_.has_video_info(),
1045 media_info_.has_audio_info(),
1046 media_info_.has_text_info())) {
1050 LOG(ERROR) <<
"Representation needs one of video, audio, or text.";
1054 if (MoreThanOneTrue(media_info_.has_video_info(),
1055 media_info_.has_audio_info(),
1056 media_info_.has_text_info())) {
1057 LOG(ERROR) <<
"Only one of VideoInfo, AudioInfo, or TextInfo can be set.";
1061 if (media_info_.container_type() == MediaInfo::CONTAINER_UNKNOWN) {
1062 LOG(ERROR) <<
"'container_type' in MediaInfo cannot be CONTAINER_UNKNOWN.";
1066 if (media_info_.has_video_info()) {
1067 mime_type_ = GetVideoMimeType();
1068 if (!HasRequiredVideoFields(media_info_.video_info())) {
1069 LOG(ERROR) <<
"Missing required fields to create a video Representation.";
1072 }
else if (media_info_.has_audio_info()) {
1073 mime_type_ = GetAudioMimeType();
1074 }
else if (media_info_.has_text_info()) {
1075 mime_type_ = GetTextMimeType();
1078 if (mime_type_.empty())
1081 codecs_ = GetCodecs(media_info_);
1087 content_protection_elements_.push_back(content_protection_element);
1088 RemoveDuplicateAttributes(&content_protection_elements_.back());
1092 const std::string& pssh) {
1093 UpdateContentProtectionPsshHelper(drm_uuid, pssh,
1094 &content_protection_elements_);
1100 if (start_time == 0 && duration == 0) {
1101 LOG(WARNING) <<
"Got segment with start_time and duration == 0. Ignoring.";
1105 if (state_change_listener_)
1106 state_change_listener_->OnNewSegmentForRepresentation(start_time, duration);
1107 if (IsContiguous(start_time, duration, size)) {
1108 ++segment_infos_.back().repeat;
1111 segment_infos_.push_back(s);
1114 bandwidth_estimator_.AddBlock(
1115 size, static_cast<double>(duration) / media_info_.reference_time_scale());
1118 DCHECK_GE(segment_infos_.size(), 1u);
1122 if (media_info_.has_video_info()) {
1123 media_info_.mutable_video_info()->set_frame_duration(sample_duration);
1124 if (state_change_listener_) {
1125 state_change_listener_->OnSetFrameRateForRepresentation(
1126 sample_duration, media_info_.video_info().time_scale());
1138 if (!HasRequiredMediaInfoFields()) {
1139 LOG(ERROR) <<
"MediaInfo missing required fields.";
1140 return xml::scoped_xml_ptr<xmlNode>();
1143 const uint64_t bandwidth = media_info_.has_bandwidth()
1144 ? media_info_.bandwidth()
1145 : bandwidth_estimator_.Estimate();
1147 DCHECK(!(HasVODOnlyFields(media_info_) && HasLiveOnlyFields(media_info_)));
1149 RepresentationXmlNode representation;
1151 representation.SetId(id_);
1152 representation.SetIntegerAttribute(
"bandwidth", bandwidth);
1153 if (!codecs_.empty())
1154 representation.SetStringAttribute(
"codecs", codecs_);
1155 representation.SetStringAttribute(
"mimeType", mime_type_);
1157 const bool has_video_info = media_info_.has_video_info();
1158 const bool has_audio_info = media_info_.has_audio_info();
1160 if (has_video_info &&
1161 !representation.AddVideoInfo(
1162 media_info_.video_info(),
1163 !(output_suppression_flags_ & kSuppressWidth),
1164 !(output_suppression_flags_ & kSuppressHeight),
1165 !(output_suppression_flags_ & kSuppressFrameRate))) {
1166 LOG(ERROR) <<
"Failed to add video info to Representation XML.";
1167 return xml::scoped_xml_ptr<xmlNode>();
1170 if (has_audio_info &&
1171 !representation.AddAudioInfo(media_info_.audio_info())) {
1172 LOG(ERROR) <<
"Failed to add audio info to Representation XML.";
1173 return xml::scoped_xml_ptr<xmlNode>();
1176 if (!representation.AddContentProtectionElements(
1177 content_protection_elements_)) {
1178 return xml::scoped_xml_ptr<xmlNode>();
1182 if (mpd_options_.mpd_type == MpdType::kStatic &&
1183 media_info_.has_media_duration_seconds()) {
1186 representation.SetFloatingPointAttribute(
1187 "duration", media_info_.media_duration_seconds());
1190 if (HasVODOnlyFields(media_info_) &&
1191 !representation.AddVODOnlyInfo(media_info_)) {
1192 LOG(ERROR) <<
"Failed to add VOD segment info.";
1193 return xml::scoped_xml_ptr<xmlNode>();
1196 if (HasLiveOnlyFields(media_info_) &&
1197 !representation.AddLiveOnlyInfo(media_info_, segment_infos_,
1199 LOG(ERROR) <<
"Failed to add Live info.";
1200 return xml::scoped_xml_ptr<xmlNode>();
1205 output_suppression_flags_ = 0;
1206 return representation.PassScopedPtr();
1210 output_suppression_flags_ |= flag;
1213 bool Representation::HasRequiredMediaInfoFields() {
1214 if (HasVODOnlyFields(media_info_) && HasLiveOnlyFields(media_info_)) {
1215 LOG(ERROR) <<
"MediaInfo cannot have both VOD and Live fields.";
1219 if (!media_info_.has_container_type()) {
1220 LOG(ERROR) <<
"MediaInfo missing required field: container_type.";
1224 if (HasVODOnlyFields(media_info_) && !media_info_.has_bandwidth()) {
1225 LOG(ERROR) <<
"Missing 'bandwidth' field. MediaInfo requires bandwidth for "
1226 "static profile for generating a valid MPD.";
1230 VLOG_IF(3, HasLiveOnlyFields(media_info_) && !media_info_.has_bandwidth())
1231 <<
"MediaInfo missing field 'bandwidth'. Using estimated from "
1237 bool Representation::IsContiguous(uint64_t start_time,
1239 uint64_t size)
const {
1240 if (segment_infos_.empty())
1244 const SegmentInfo& previous = segment_infos_.back();
1245 const uint64_t previous_segment_end_time =
1246 previous.start_time + previous.duration * (previous.repeat + 1);
1247 if (previous_segment_end_time == start_time &&
1248 segment_infos_.back().duration == duration) {
1253 const uint64_t previous_segment_start_time =
1254 previous.start_time + previous.duration * previous.repeat;
1255 if (previous_segment_start_time >= start_time) {
1256 LOG(ERROR) <<
"Segments should not be out of order segment. Adding segment "
1257 "with start_time == "
1258 << start_time <<
" but the previous segment starts at "
1259 << previous_segment_start_time <<
".";
1264 const uint64_t kRoundingErrorGrace = 5;
1265 if (previous_segment_end_time + kRoundingErrorGrace < start_time) {
1266 LOG(WARNING) <<
"Found a gap of size "
1267 << (start_time - previous_segment_end_time)
1268 <<
" > kRoundingErrorGrace (" << kRoundingErrorGrace
1269 <<
"). The new segment starts at " << start_time
1270 <<
" but the previous segment ends at "
1271 << previous_segment_end_time <<
".";
1276 if (start_time < previous_segment_end_time - kRoundingErrorGrace) {
1278 <<
"Segments should not be overlapping. The new segment starts at "
1279 << start_time <<
" but the previous segment ends at "
1280 << previous_segment_end_time <<
".";
1288 void Representation::SlideWindow() {
1289 DCHECK(!segment_infos_.empty());
1290 if (mpd_options_.time_shift_buffer_depth <= 0.0 ||
1291 mpd_options_.mpd_type == MpdType::kStatic)
1294 const uint32_t time_scale = GetTimeScale(media_info_);
1295 DCHECK_GT(time_scale, 0u);
1297 uint64_t time_shift_buffer_depth =
1298 static_cast<uint64_t
>(mpd_options_.time_shift_buffer_depth * time_scale);
1302 const uint64_t current_play_time = LatestSegmentStartTime(segment_infos_);
1303 if (current_play_time <= time_shift_buffer_depth)
1306 const uint64_t timeshift_limit = current_play_time - time_shift_buffer_depth;
1310 std::list<SegmentInfo>::iterator first = segment_infos_.begin();
1311 std::list<SegmentInfo>::iterator last = first;
1312 size_t num_segments_removed = 0;
1313 for (; last != segment_infos_.end(); ++last) {
1314 const uint64_t last_segment_end_time = LastSegmentEndTime(*last);
1315 if (timeshift_limit < last_segment_end_time)
1317 num_segments_removed += last->repeat + 1;
1319 segment_infos_.erase(first, last);
1320 start_number_ += num_segments_removed;
1323 SegmentInfo* first_segment_info = &segment_infos_.front();
1324 DCHECK_LE(timeshift_limit, LastSegmentEndTime(*first_segment_info));
1327 const int repeat_index =
1328 SearchTimedOutRepeatIndex(timeshift_limit, *first_segment_info);
1329 CHECK_GE(repeat_index, 0);
1330 if (repeat_index == 0)
1333 first_segment_info->start_time = first_segment_info->start_time +
1334 first_segment_info->duration * repeat_index;
1336 first_segment_info->repeat = first_segment_info->repeat - repeat_index;
1337 start_number_ += repeat_index;
1340 std::string Representation::GetVideoMimeType()
const {
1341 return GetMimeType(
"video", media_info_.container_type());
1344 std::string Representation::GetAudioMimeType()
const {
1345 return GetMimeType(
"audio", media_info_.container_type());
1348 std::string Representation::GetTextMimeType()
const {
1349 CHECK(media_info_.has_text_info());
1350 if (media_info_.text_info().format() ==
"ttml") {
1351 switch (media_info_.container_type()) {
1352 case MediaInfo::CONTAINER_TEXT:
1353 return "application/ttml+xml";
1354 case MediaInfo::CONTAINER_MP4:
1355 return "application/mp4";
1357 LOG(ERROR) <<
"Failed to determine MIME type for TTML container: "
1358 << media_info_.container_type();
1362 if (media_info_.text_info().format() ==
"vtt") {
1363 if (media_info_.container_type() == MediaInfo::CONTAINER_TEXT) {
1365 }
else if (media_info_.container_type() == MediaInfo::CONTAINER_MP4) {
1366 return "application/mp4";
1368 LOG(ERROR) <<
"Failed to determine MIME type for VTT container: "
1369 << media_info_.container_type();
1373 LOG(ERROR) <<
"Cannot determine MIME type for format: "
1374 << media_info_.text_info().format()
1375 <<
" container: " << media_info_.container_type();
1379 bool Representation::GetEarliestTimestamp(
double* timestamp_seconds) {
1380 DCHECK(timestamp_seconds);
1382 if (segment_infos_.empty())
1385 *timestamp_seconds =
static_cast<double>(segment_infos_.begin()->start_time) /
1386 GetTimeScale(media_info_);
void OnSetFrameRateForRepresentation(uint32_t representation_id, uint32_t frame_duration, uint32_t timescale)
virtual void AddNewSegment(uint64_t start_time, uint64_t duration, uint64_t size)
Representation(const MediaInfo &media_info, const MpdOptions &mpd_options, uint32_t representation_id, std::unique_ptr< RepresentationStateChangeListener > state_change_listener)
virtual void SetSampleDuration(uint32_t sample_duration)
virtual Representation * AddRepresentation(const MediaInfo &media_info)
std::string LanguageToShortestForm(const std::string &language)
virtual void AddContentProtectionElement(const ContentProtectionElement &element)
virtual void AddTrickPlayReferenceId(uint32_t id)
MpdBuilder(const MpdOptions &mpd_options)
virtual void AddRole(Role role)
void AddBaseUrl(const std::string &base_url)
virtual void UpdateContentProtectionPssh(const std::string &drm_uuid, const std::string &pssh)
AdaptationSet(uint32_t adaptation_set_id, const std::string &lang, const MpdOptions &mpd_options, base::AtomicSequenceNumber *representation_counter)
xml::scoped_xml_ptr< xmlNode > GetXml()
virtual bool ToString(std::string *output)
void AddAdaptationSetSwitching(uint32_t adaptation_set_id)
virtual void ForceSetSegmentAlignment(bool segment_alignment)
static void MakePathsRelativeToMpd(const std::string &mpd_path, MediaInfo *media_info)
xml::scoped_xml_ptr< xmlNode > GetXml()
virtual void AddContentProtectionElement(const ContentProtectionElement &element)
virtual AdaptationSet * AddAdaptationSet(const std::string &lang)
void OnNewSegmentForRepresentation(uint32_t representation_id, uint64_t start_time, uint64_t duration)
virtual void UpdateContentProtectionPssh(const std::string &drm_uuid, const std::string &pssh)
void SuppressOnce(SuppressFlag flag)