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/file/file.h"
28 #include "packager/mpd/base/content_protection_element.h"
29 #include "packager/mpd/base/language_utils.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;
192 bool WriteXmlCharArrayToOutput(xmlChar* doc,
194 std::string* output) {
197 output->assign(doc, doc + doc_size);
201 bool WriteXmlCharArrayToOutput(xmlChar* doc,
203 media::File* output) {
206 if (output->Write(doc, doc_size) < doc_size)
209 return output->Flush();
212 std::string MakePathRelative(
const std::string& path,
213 const std::string& mpd_dir) {
214 return (path.find(mpd_dir) == 0) ? path.substr(mpd_dir.size()) : path;
220 bool HasRequiredVideoFields(
const MediaInfo_VideoInfo& video_info) {
221 if (!video_info.has_height() || !video_info.has_width()) {
223 <<
"Width and height are required fields for generating a valid MPD.";
228 LOG_IF(WARNING, !video_info.has_time_scale())
229 <<
"Video info does not contain timescale required for "
230 "calculating framerate. @frameRate is required for DASH IOP.";
231 LOG_IF(WARNING, !video_info.has_frame_duration())
232 <<
"Video info does not contain frame duration required "
233 "for calculating framerate. @frameRate is required for DASH IOP.";
234 LOG_IF(WARNING, !video_info.has_pixel_width())
235 <<
"Video info does not contain pixel_width to calculate the sample "
236 "aspect ratio required for DASH IOP.";
237 LOG_IF(WARNING, !video_info.has_pixel_height())
238 <<
"Video info does not contain pixel_height to calculate the sample "
239 "aspect ratio required for DASH IOP.";
250 std::string GetPictureAspectRatio(uint32_t width,
252 uint32_t pixel_width,
253 uint32_t pixel_height) {
254 const uint32_t scaled_width = pixel_width * width;
255 const uint32_t scaled_height = pixel_height * height;
256 const double par =
static_cast<double>(scaled_width) / scaled_height;
260 const uint32_t kLargestPossibleParY = 19;
262 uint32_t par_num = 0;
263 uint32_t par_den = 0;
264 double min_error = 1.0;
265 for (uint32_t den = 1; den <= kLargestPossibleParY; ++den) {
266 uint32_t num = par * den + 0.5;
267 double error = fabs(par - static_cast<double>(num) / den);
268 if (error < min_error) {
272 if (error == 0)
break;
275 VLOG(2) <<
"width*pix_width : height*pixel_height (" << scaled_width <<
":"
276 << scaled_height <<
") reduced to " << par_num <<
":" << par_den
277 <<
" with error " << min_error <<
".";
279 return base::IntToString(par_num) +
":" + base::IntToString(par_den);
284 void AddPictureAspectRatio(
285 const MediaInfo::VideoInfo& video_info,
286 std::set<std::string>* picture_aspect_ratio) {
289 if (picture_aspect_ratio->size() > 1)
292 if (video_info.width() == 0 || video_info.height() == 0 ||
293 video_info.pixel_width() == 0 || video_info.pixel_height() == 0) {
298 picture_aspect_ratio->insert(
"bogus");
299 picture_aspect_ratio->insert(
"entries");
303 const std::string par = GetPictureAspectRatio(
304 video_info.width(), video_info.height(),
305 video_info.pixel_width(), video_info.pixel_height());
306 DVLOG(1) <<
"Setting par as: " << par
307 <<
" for video with width: " << video_info.width()
308 <<
" height: " << video_info.height()
309 <<
" pixel_width: " << video_info.pixel_width() <<
" pixel_height; "
310 << video_info.pixel_height();
311 picture_aspect_ratio->insert(par);
314 std::string RoleToText(AdaptationSet::Role role) {
318 case AdaptationSet::kRoleCaption:
320 case AdaptationSet::kRoleSubtitle:
322 case AdaptationSet::kRoleMain:
324 case AdaptationSet::kRoleAlternate:
326 case AdaptationSet::kRoleSupplementary:
327 return "supplementary";
328 case AdaptationSet::kRoleCommentary:
330 case AdaptationSet::kRoleDub:
341 class LibXmlInitializer {
343 LibXmlInitializer() : initialized_(false) {
344 base::AutoLock lock(lock_);
351 ~LibXmlInitializer() {
352 base::AutoLock lock(lock_);
355 initialized_ =
false;
363 DISALLOW_COPY_AND_ASSIGN(LibXmlInitializer);
366 class RepresentationStateChangeListenerImpl
367 :
public RepresentationStateChangeListener {
370 RepresentationStateChangeListenerImpl(uint32_t representation_id,
371 AdaptationSet* adaptation_set)
372 : representation_id_(representation_id), adaptation_set_(adaptation_set) {
373 DCHECK(adaptation_set_);
375 ~RepresentationStateChangeListenerImpl()
override {}
378 void OnNewSegmentForRepresentation(uint64_t start_time,
379 uint64_t duration)
override {
380 adaptation_set_->OnNewSegmentForRepresentation(representation_id_,
381 start_time, duration);
384 void OnSetFrameRateForRepresentation(uint32_t frame_duration,
385 uint32_t timescale)
override {
386 adaptation_set_->OnSetFrameRateForRepresentation(representation_id_,
387 frame_duration, timescale);
391 const uint32_t representation_id_;
392 AdaptationSet*
const adaptation_set_;
394 DISALLOW_COPY_AND_ASSIGN(RepresentationStateChangeListenerImpl);
400 : mpd_options_(mpd_options), clock_(new base::DefaultClock()) {}
402 MpdBuilder::~MpdBuilder() {}
405 base_urls_.push_back(base_url);
409 std::unique_ptr<AdaptationSet> adaptation_set(
410 new AdaptationSet(adaptation_set_counter_.GetNext(), lang, mpd_options_,
411 &representation_counter_));
412 DCHECK(adaptation_set);
414 if (!lang.empty() && lang == mpd_options_.default_language) {
415 adaptation_set->AddRole(AdaptationSet::kRoleMain);
418 adaptation_sets_.push_back(std::move(adaptation_set));
419 return adaptation_sets_.back().get();
424 return WriteMpdToOutput(output_file);
429 return WriteMpdToOutput(output);
431 template <
typename OutputType>
432 bool MpdBuilder::WriteMpdToOutput(OutputType* output) {
433 static LibXmlInitializer lib_xml_initializer;
435 xml::scoped_xml_ptr<xmlDoc> doc(GenerateMpd());
439 static const int kNiceFormat = 1;
440 int doc_str_size = 0;
441 xmlChar* doc_str = NULL;
442 xmlDocDumpFormatMemoryEnc(doc.get(), &doc_str, &doc_str_size,
"UTF-8",
445 bool result = WriteXmlCharArrayToOutput(doc_str, doc_str_size, output);
453 xmlDocPtr MpdBuilder::GenerateMpd() {
455 static const char kXmlVersion[] =
"1.0";
456 xml::scoped_xml_ptr<xmlDoc> doc(xmlNewDoc(BAD_CAST kXmlVersion));
460 XmlNode period(
"Period");
466 for (
const std::unique_ptr<AdaptationSet>& adaptation_set :
468 xml::scoped_xml_ptr<xmlNode> child(adaptation_set->GetXml());
469 if (!child.get() || !period.AddChild(std::move(child)))
474 std::list<std::string>::const_iterator base_urls_it = base_urls_.begin();
475 for (; base_urls_it != base_urls_.end(); ++base_urls_it) {
476 XmlNode base_url(
"BaseURL");
477 base_url.SetContent(*base_urls_it);
479 if (!mpd.AddChild(base_url.PassScopedPtr()))
484 if (mpd_options_.mpd_type == MpdType::kDynamic) {
486 period.SetStringAttribute(
"start",
"PT0S");
489 if (!mpd.AddChild(period.PassScopedPtr()))
492 AddMpdNameSpaceInfo(&mpd);
494 static const char kOnDemandProfile[] =
495 "urn:mpeg:dash:profile:isoff-on-demand:2011";
496 static const char kLiveProfile[] =
497 "urn:mpeg:dash:profile:isoff-live:2011";
498 switch (mpd_options_.dash_profile) {
499 case DashProfile::kOnDemand:
500 mpd.SetStringAttribute(
"profiles", kOnDemandProfile);
502 case DashProfile::kLive:
503 mpd.SetStringAttribute(
"profiles", kLiveProfile);
506 NOTREACHED() <<
"Unknown DASH profile: "
507 <<
static_cast<int>(mpd_options_.dash_profile);
511 AddCommonMpdInfo(&mpd);
512 switch (mpd_options_.mpd_type) {
513 case MpdType::kStatic:
514 AddStaticMpdInfo(&mpd);
516 case MpdType::kDynamic:
517 AddDynamicMpdInfo(&mpd);
520 NOTREACHED() <<
"Unknown MPD type: "
521 <<
static_cast<int>(mpd_options_.mpd_type);
526 const std::string version = GetPackagerVersion();
527 if (!version.empty()) {
528 std::string version_string =
529 base::StringPrintf(
"Generated with %s version %s",
530 GetPackagerProjectUrl().c_str(), version.c_str());
531 xml::scoped_xml_ptr<xmlNode> comment(
532 xmlNewDocComment(doc.get(), BAD_CAST version_string.c_str()));
533 xmlDocSetRootElement(doc.get(), comment.get());
534 xmlAddSibling(comment.release(), mpd.Release());
536 xmlDocSetRootElement(doc.get(), mpd.Release());
538 return doc.release();
541 void MpdBuilder::AddCommonMpdInfo(XmlNode* mpd_node) {
542 if (Positive(mpd_options_.min_buffer_time)) {
543 mpd_node->SetStringAttribute(
544 "minBufferTime", SecondsToXmlDuration(mpd_options_.min_buffer_time));
546 LOG(ERROR) <<
"minBufferTime value not specified.";
551 void MpdBuilder::AddStaticMpdInfo(XmlNode* mpd_node) {
553 DCHECK_EQ(MpdType::kStatic, mpd_options_.mpd_type);
555 static const char kStaticMpdType[] =
"static";
556 mpd_node->SetStringAttribute(
"type", kStaticMpdType);
557 mpd_node->SetStringAttribute(
558 "mediaPresentationDuration",
559 SecondsToXmlDuration(GetStaticMpdDuration(mpd_node)));
562 void MpdBuilder::AddDynamicMpdInfo(XmlNode* mpd_node) {
564 DCHECK_EQ(MpdType::kDynamic, mpd_options_.mpd_type);
566 static const char kDynamicMpdType[] =
"dynamic";
567 mpd_node->SetStringAttribute(
"type", kDynamicMpdType);
570 mpd_node->SetStringAttribute(
"publishTime",
571 XmlDateTimeNowWithOffset(0, clock_.get()));
575 if (availability_start_time_.empty()) {
576 double earliest_presentation_time;
577 if (GetEarliestTimestamp(&earliest_presentation_time)) {
578 availability_start_time_ =
579 XmlDateTimeNowWithOffset(mpd_options_.availability_time_offset -
580 std::ceil(earliest_presentation_time),
583 LOG(ERROR) <<
"Could not determine the earliest segment presentation "
584 "time for availabilityStartTime calculation.";
588 if (!availability_start_time_.empty())
589 mpd_node->SetStringAttribute(
"availabilityStartTime",
590 availability_start_time_);
592 if (Positive(mpd_options_.minimum_update_period)) {
593 mpd_node->SetStringAttribute(
594 "minimumUpdatePeriod",
595 SecondsToXmlDuration(mpd_options_.minimum_update_period));
597 LOG(WARNING) <<
"The profile is dynamic but no minimumUpdatePeriod "
601 SetIfPositive(
"timeShiftBufferDepth", mpd_options_.time_shift_buffer_depth,
603 SetIfPositive(
"suggestedPresentationDelay",
604 mpd_options_.suggested_presentation_delay, mpd_node);
607 float MpdBuilder::GetStaticMpdDuration(XmlNode* mpd_node) {
609 DCHECK_EQ(MpdType::kStatic, mpd_options_.mpd_type);
611 xmlNodePtr period_node = FindPeriodNode(mpd_node);
612 DCHECK(period_node) <<
"Period element must be a child of mpd_node.";
613 DCHECK(IsPeriodNode(period_node));
619 float max_duration = 0.0f;
620 for (xmlNodePtr adaptation_set = xmlFirstElementChild(period_node);
621 adaptation_set; adaptation_set = xmlNextElementSibling(adaptation_set)) {
622 for (xmlNodePtr representation = xmlFirstElementChild(adaptation_set);
624 representation = xmlNextElementSibling(representation)) {
625 float duration = 0.0f;
626 if (GetDurationAttribute(representation, &duration)) {
627 max_duration = max_duration > duration ? max_duration : duration;
631 xmlUnsetProp(representation, BAD_CAST
"duration");
639 bool MpdBuilder::GetEarliestTimestamp(
double* timestamp_seconds) {
640 DCHECK(timestamp_seconds);
642 double earliest_timestamp(-1);
643 for (
const std::unique_ptr<AdaptationSet>& adaptation_set :
646 if (adaptation_set->GetEarliestTimestamp(×tamp) &&
647 ((earliest_timestamp < 0) || (timestamp < earliest_timestamp))) {
648 earliest_timestamp = timestamp;
651 if (earliest_timestamp < 0)
654 *timestamp_seconds = earliest_timestamp;
659 MediaInfo* media_info) {
661 const std::string kFileProtocol(
"file://");
662 std::string mpd_file_path = (mpd_path.find(kFileProtocol) == 0)
663 ? mpd_path.substr(kFileProtocol.size())
666 if (!mpd_file_path.empty()) {
667 std::string mpd_dir(FilePath::FromUTF8Unsafe(mpd_file_path)
668 .DirName().AsEndingWithSeparator().AsUTF8Unsafe());
669 if (!mpd_dir.empty()) {
670 if (media_info->has_media_file_name()) {
671 media_info->set_media_file_name(
672 MakePathRelative(media_info->media_file_name(), mpd_dir));
674 if (media_info->has_init_segment_name()) {
675 media_info->set_init_segment_name(
676 MakePathRelative(media_info->init_segment_name(), mpd_dir));
678 if (media_info->has_segment_template()) {
679 media_info->set_segment_template(
680 MakePathRelative(media_info->segment_template(), mpd_dir));
687 const std::string& lang,
689 base::AtomicSequenceNumber* counter)
690 : representation_counter_(counter),
691 id_(adaptation_set_id),
693 mpd_options_(mpd_options),
694 segments_aligned_(kSegmentAlignmentUnknown),
695 force_set_segment_alignment_(false) {
699 AdaptationSet::~AdaptationSet() {}
702 const uint32_t representation_id = representation_counter_->GetNext();
705 std::unique_ptr<RepresentationStateChangeListener> listener(
706 new RepresentationStateChangeListenerImpl(representation_id,
this));
707 std::unique_ptr<Representation> representation(
new Representation(
708 media_info, mpd_options_, representation_id, std::move(listener)));
710 if (!representation->Init())
715 if (media_info.has_video_info()) {
716 const MediaInfo::VideoInfo& video_info = media_info.video_info();
717 DCHECK(video_info.has_width());
718 DCHECK(video_info.has_height());
719 video_widths_.insert(video_info.width());
720 video_heights_.insert(video_info.height());
722 if (video_info.has_time_scale() && video_info.has_frame_duration())
723 RecordFrameRate(video_info.frame_duration(), video_info.time_scale());
725 AddPictureAspectRatio(video_info, &picture_aspect_ratio_);
728 if (media_info.has_video_info()) {
729 content_type_ =
"video";
730 }
else if (media_info.has_audio_info()) {
731 content_type_ =
"audio";
732 }
else if (media_info.has_text_info()) {
733 content_type_ =
"text";
735 if (media_info.text_info().has_type() &&
736 (media_info.text_info().type() != MediaInfo::TextInfo::UNKNOWN)) {
737 roles_.insert(MediaInfoTextTypeToRole(media_info.text_info().type()));
741 representations_.push_back(std::move(representation));
742 return representations_.back().get();
747 content_protection_elements_.push_back(content_protection_element);
748 RemoveDuplicateAttributes(&content_protection_elements_.back());
752 const std::string& pssh) {
753 UpdateContentProtectionPsshHelper(drm_uuid, pssh,
754 &content_protection_elements_);
768 AdaptationSetXmlNode adaptation_set;
770 bool suppress_representation_width =
false;
771 bool suppress_representation_height =
false;
772 bool suppress_representation_frame_rate =
false;
774 adaptation_set.SetId(id_);
775 adaptation_set.SetStringAttribute(
"contentType", content_type_);
776 if (!lang_.empty() && lang_ !=
"und") {
781 if (video_widths_.size() == 1) {
782 suppress_representation_width =
true;
783 adaptation_set.SetIntegerAttribute(
"width", *video_widths_.begin());
784 }
else if (video_widths_.size() > 1) {
785 adaptation_set.SetIntegerAttribute(
"maxWidth", *video_widths_.rbegin());
787 if (video_heights_.size() == 1) {
788 suppress_representation_height =
true;
789 adaptation_set.SetIntegerAttribute(
"height", *video_heights_.begin());
790 }
else if (video_heights_.size() > 1) {
791 adaptation_set.SetIntegerAttribute(
"maxHeight", *video_heights_.rbegin());
794 if (video_frame_rates_.size() == 1) {
795 suppress_representation_frame_rate =
true;
796 adaptation_set.SetStringAttribute(
"frameRate",
797 video_frame_rates_.begin()->second);
798 }
else if (video_frame_rates_.size() > 1) {
799 adaptation_set.SetStringAttribute(
"maxFrameRate",
800 video_frame_rates_.rbegin()->second);
805 if (mpd_options_.dash_profile == DashProfile::kOnDemand) {
806 CheckVodSegmentAlignment();
809 if (segments_aligned_ == kSegmentAlignmentTrue) {
810 adaptation_set.SetStringAttribute(
811 mpd_options_.dash_profile == DashProfile::kOnDemand
812 ?
"subsegmentAlignment"
813 :
"segmentAlignment",
817 if (picture_aspect_ratio_.size() == 1)
818 adaptation_set.SetStringAttribute(
"par", *picture_aspect_ratio_.begin());
820 if (!adaptation_set.AddContentProtectionElements(
821 content_protection_elements_)) {
822 return xml::scoped_xml_ptr<xmlNode>();
825 std::string switching_ids;
826 for (uint32_t
id : adaptation_set_switching_ids_) {
827 if (!switching_ids.empty())
828 switching_ids +=
',';
829 switching_ids += base::UintToString(
id);
831 if (!switching_ids.empty()) {
832 adaptation_set.AddSupplementalProperty(
833 "urn:mpeg:dash:adaptation-set-switching:2016", switching_ids);
836 for (AdaptationSet::Role role : roles_)
837 adaptation_set.AddRoleElement(
"urn:mpeg:dash:role:2011", RoleToText(role));
839 for (
const std::unique_ptr<Representation>& representation :
841 if (suppress_representation_width)
842 representation->SuppressOnce(Representation::kSuppressWidth);
843 if (suppress_representation_height)
844 representation->SuppressOnce(Representation::kSuppressHeight);
845 if (suppress_representation_frame_rate)
846 representation->SuppressOnce(Representation::kSuppressFrameRate);
847 xml::scoped_xml_ptr<xmlNode> child(representation->GetXml());
848 if (!child || !adaptation_set.AddChild(std::move(child)))
849 return xml::scoped_xml_ptr<xmlNode>();
852 return adaptation_set.PassScopedPtr();
857 segment_alignment ? kSegmentAlignmentTrue : kSegmentAlignmentFalse;
858 force_set_segment_alignment_ =
true;
862 adaptation_set_switching_ids_.push_back(adaptation_set_id);
875 if (mpd_options_.dash_profile == DashProfile::kLive) {
876 CheckLiveSegmentAlignment(representation_id, start_time, duration);
878 representation_segment_start_times_[representation_id].push_back(
884 uint32_t representation_id,
885 uint32_t frame_duration,
886 uint32_t timescale) {
887 RecordFrameRate(frame_duration, timescale);
890 bool AdaptationSet::GetEarliestTimestamp(
double* timestamp_seconds) {
891 DCHECK(timestamp_seconds);
893 double earliest_timestamp(-1);
894 for (
const std::unique_ptr<Representation>& representation :
897 if (representation->GetEarliestTimestamp(×tamp) &&
898 ((earliest_timestamp < 0) || (timestamp < earliest_timestamp))) {
899 earliest_timestamp = timestamp;
902 if (earliest_timestamp < 0)
905 *timestamp_seconds = earliest_timestamp;
933 void AdaptationSet::CheckLiveSegmentAlignment(uint32_t representation_id,
936 if (segments_aligned_ == kSegmentAlignmentFalse ||
937 force_set_segment_alignment_) {
941 std::list<uint64_t>& representation_start_times =
942 representation_segment_start_times_[representation_id];
943 representation_start_times.push_back(start_time);
946 if (representation_segment_start_times_.size() != representations_.size())
949 DCHECK(!representation_start_times.empty());
950 const uint64_t expected_start_time = representation_start_times.front();
951 for (RepresentationTimeline::const_iterator it =
952 representation_segment_start_times_.begin();
953 it != representation_segment_start_times_.end(); ++it) {
957 if (it->second.empty())
960 if (expected_start_time != it->second.front()) {
963 segments_aligned_ = kSegmentAlignmentFalse;
964 representation_segment_start_times_.clear();
968 segments_aligned_ = kSegmentAlignmentTrue;
970 for (RepresentationTimeline::iterator it =
971 representation_segment_start_times_.begin();
972 it != representation_segment_start_times_.end(); ++it) {
973 it->second.pop_front();
979 void AdaptationSet::CheckVodSegmentAlignment() {
980 if (segments_aligned_ == kSegmentAlignmentFalse ||
981 force_set_segment_alignment_) {
984 if (representation_segment_start_times_.empty())
986 if (representation_segment_start_times_.size() == 1) {
987 segments_aligned_ = kSegmentAlignmentTrue;
994 const std::list<uint64_t>& expected_time_line =
995 representation_segment_start_times_.begin()->second;
997 bool all_segment_time_line_same_length =
true;
999 RepresentationTimeline::const_iterator it =
1000 representation_segment_start_times_.begin();
1001 for (++it; it != representation_segment_start_times_.end(); ++it) {
1002 const std::list<uint64_t>& other_time_line = it->second;
1003 if (expected_time_line.size() != other_time_line.size()) {
1004 all_segment_time_line_same_length =
false;
1007 const std::list<uint64_t>* longer_list = &other_time_line;
1008 const std::list<uint64_t>* shorter_list = &expected_time_line;
1009 if (expected_time_line.size() > other_time_line.size()) {
1010 shorter_list = &other_time_line;
1011 longer_list = &expected_time_line;
1014 if (!std::equal(shorter_list->begin(), shorter_list->end(),
1015 longer_list->begin())) {
1017 segments_aligned_ = kSegmentAlignmentFalse;
1018 representation_segment_start_times_.clear();
1029 if (!all_segment_time_line_same_length) {
1030 segments_aligned_ = kSegmentAlignmentUnknown;
1034 segments_aligned_ = kSegmentAlignmentTrue;
1039 void AdaptationSet::RecordFrameRate(uint32_t frame_duration,
1040 uint32_t timescale) {
1041 if (frame_duration == 0) {
1042 LOG(ERROR) <<
"Frame duration is 0 and cannot be set.";
1045 video_frame_rates_[
static_cast<double>(timescale) / frame_duration] =
1046 base::IntToString(timescale) +
"/" + base::IntToString(frame_duration);
1050 const MediaInfo& media_info,
1053 std::unique_ptr<RepresentationStateChangeListener> state_change_listener)
1054 : media_info_(media_info),
1057 mpd_options_(mpd_options),
1059 state_change_listener_(std::move(state_change_listener)),
1060 output_suppression_flags_(0) {}
1062 Representation::~Representation() {}
1065 if (!AtLeastOneTrue(media_info_.has_video_info(),
1066 media_info_.has_audio_info(),
1067 media_info_.has_text_info())) {
1071 LOG(ERROR) <<
"Representation needs one of video, audio, or text.";
1075 if (MoreThanOneTrue(media_info_.has_video_info(),
1076 media_info_.has_audio_info(),
1077 media_info_.has_text_info())) {
1078 LOG(ERROR) <<
"Only one of VideoInfo, AudioInfo, or TextInfo can be set.";
1082 if (media_info_.container_type() == MediaInfo::CONTAINER_UNKNOWN) {
1083 LOG(ERROR) <<
"'container_type' in MediaInfo cannot be CONTAINER_UNKNOWN.";
1087 if (media_info_.has_video_info()) {
1088 mime_type_ = GetVideoMimeType();
1089 if (!HasRequiredVideoFields(media_info_.video_info())) {
1090 LOG(ERROR) <<
"Missing required fields to create a video Representation.";
1093 }
else if (media_info_.has_audio_info()) {
1094 mime_type_ = GetAudioMimeType();
1095 }
else if (media_info_.has_text_info()) {
1096 mime_type_ = GetTextMimeType();
1099 if (mime_type_.empty())
1102 codecs_ = GetCodecs(media_info_);
1108 content_protection_elements_.push_back(content_protection_element);
1109 RemoveDuplicateAttributes(&content_protection_elements_.back());
1113 const std::string& pssh) {
1114 UpdateContentProtectionPsshHelper(drm_uuid, pssh,
1115 &content_protection_elements_);
1121 if (start_time == 0 && duration == 0) {
1122 LOG(WARNING) <<
"Got segment with start_time and duration == 0. Ignoring.";
1126 if (state_change_listener_)
1127 state_change_listener_->OnNewSegmentForRepresentation(start_time, duration);
1128 if (IsContiguous(start_time, duration, size)) {
1129 ++segment_infos_.back().repeat;
1132 segment_infos_.push_back(s);
1135 bandwidth_estimator_.AddBlock(
1136 size, static_cast<double>(duration) / media_info_.reference_time_scale());
1139 DCHECK_GE(segment_infos_.size(), 1u);
1143 if (media_info_.has_video_info()) {
1144 media_info_.mutable_video_info()->set_frame_duration(sample_duration);
1145 if (state_change_listener_) {
1146 state_change_listener_->OnSetFrameRateForRepresentation(
1147 sample_duration, media_info_.video_info().time_scale());
1159 if (!HasRequiredMediaInfoFields()) {
1160 LOG(ERROR) <<
"MediaInfo missing required fields.";
1161 return xml::scoped_xml_ptr<xmlNode>();
1164 const uint64_t bandwidth = media_info_.has_bandwidth()
1165 ? media_info_.bandwidth()
1166 : bandwidth_estimator_.Estimate();
1168 DCHECK(!(HasVODOnlyFields(media_info_) && HasLiveOnlyFields(media_info_)));
1170 RepresentationXmlNode representation;
1172 representation.SetId(id_);
1173 representation.SetIntegerAttribute(
"bandwidth", bandwidth);
1174 if (!codecs_.empty())
1175 representation.SetStringAttribute(
"codecs", codecs_);
1176 representation.SetStringAttribute(
"mimeType", mime_type_);
1178 const bool has_video_info = media_info_.has_video_info();
1179 const bool has_audio_info = media_info_.has_audio_info();
1181 if (has_video_info &&
1182 !representation.AddVideoInfo(
1183 media_info_.video_info(),
1184 !(output_suppression_flags_ & kSuppressWidth),
1185 !(output_suppression_flags_ & kSuppressHeight),
1186 !(output_suppression_flags_ & kSuppressFrameRate))) {
1187 LOG(ERROR) <<
"Failed to add video info to Representation XML.";
1188 return xml::scoped_xml_ptr<xmlNode>();
1191 if (has_audio_info &&
1192 !representation.AddAudioInfo(media_info_.audio_info())) {
1193 LOG(ERROR) <<
"Failed to add audio info to Representation XML.";
1194 return xml::scoped_xml_ptr<xmlNode>();
1197 if (!representation.AddContentProtectionElements(
1198 content_protection_elements_)) {
1199 return xml::scoped_xml_ptr<xmlNode>();
1202 if (HasVODOnlyFields(media_info_) &&
1203 !representation.AddVODOnlyInfo(media_info_)) {
1204 LOG(ERROR) <<
"Failed to add VOD segment info.";
1205 return xml::scoped_xml_ptr<xmlNode>();
1208 if (HasLiveOnlyFields(media_info_) &&
1209 !representation.AddLiveOnlyInfo(media_info_, segment_infos_,
1211 LOG(ERROR) <<
"Failed to add Live info.";
1212 return xml::scoped_xml_ptr<xmlNode>();
1217 output_suppression_flags_ = 0;
1218 return representation.PassScopedPtr();
1222 output_suppression_flags_ |= flag;
1225 bool Representation::HasRequiredMediaInfoFields() {
1226 if (HasVODOnlyFields(media_info_) && HasLiveOnlyFields(media_info_)) {
1227 LOG(ERROR) <<
"MediaInfo cannot have both VOD and Live fields.";
1231 if (!media_info_.has_container_type()) {
1232 LOG(ERROR) <<
"MediaInfo missing required field: container_type.";
1236 if (HasVODOnlyFields(media_info_) && !media_info_.has_bandwidth()) {
1237 LOG(ERROR) <<
"Missing 'bandwidth' field. MediaInfo requires bandwidth for "
1238 "static profile for generating a valid MPD.";
1242 VLOG_IF(3, HasLiveOnlyFields(media_info_) && !media_info_.has_bandwidth())
1243 <<
"MediaInfo missing field 'bandwidth'. Using estimated from "
1249 bool Representation::IsContiguous(uint64_t start_time,
1251 uint64_t size)
const {
1252 if (segment_infos_.empty())
1256 const SegmentInfo& previous = segment_infos_.back();
1257 const uint64_t previous_segment_end_time =
1258 previous.start_time + previous.duration * (previous.repeat + 1);
1259 if (previous_segment_end_time == start_time &&
1260 segment_infos_.back().duration == duration) {
1265 const uint64_t previous_segment_start_time =
1266 previous.start_time + previous.duration * previous.repeat;
1267 if (previous_segment_start_time >= start_time) {
1268 LOG(ERROR) <<
"Segments should not be out of order segment. Adding segment "
1269 "with start_time == "
1270 << start_time <<
" but the previous segment starts at "
1271 << previous.start_time <<
".";
1276 const uint64_t kRoundingErrorGrace = 5;
1277 if (previous_segment_end_time + kRoundingErrorGrace < start_time) {
1278 LOG(WARNING) <<
"Found a gap of size "
1279 << (start_time - previous_segment_end_time)
1280 <<
" > kRoundingErrorGrace (" << kRoundingErrorGrace
1281 <<
"). The new segment starts at " << start_time
1282 <<
" but the previous segment ends at "
1283 << previous_segment_end_time <<
".";
1288 if (start_time < previous_segment_end_time - kRoundingErrorGrace) {
1290 <<
"Segments should not be overlapping. The new segment starts at "
1291 << start_time <<
" but the previous segment ends at "
1292 << previous_segment_end_time <<
".";
1300 void Representation::SlideWindow() {
1301 DCHECK(!segment_infos_.empty());
1302 if (mpd_options_.time_shift_buffer_depth <= 0.0)
1305 const uint32_t time_scale = GetTimeScale(media_info_);
1306 DCHECK_GT(time_scale, 0u);
1308 uint64_t time_shift_buffer_depth =
1309 static_cast<uint64_t
>(mpd_options_.time_shift_buffer_depth * time_scale);
1313 const uint64_t current_play_time = LatestSegmentStartTime(segment_infos_);
1314 if (current_play_time <= time_shift_buffer_depth)
1317 const uint64_t timeshift_limit = current_play_time - time_shift_buffer_depth;
1321 std::list<SegmentInfo>::iterator first = segment_infos_.begin();
1322 std::list<SegmentInfo>::iterator last = first;
1323 size_t num_segments_removed = 0;
1324 for (; last != segment_infos_.end(); ++last) {
1325 const uint64_t last_segment_end_time = LastSegmentEndTime(*last);
1326 if (timeshift_limit < last_segment_end_time)
1328 num_segments_removed += last->repeat + 1;
1330 segment_infos_.erase(first, last);
1331 start_number_ += num_segments_removed;
1334 SegmentInfo* first_segment_info = &segment_infos_.front();
1335 DCHECK_LE(timeshift_limit, LastSegmentEndTime(*first_segment_info));
1338 const int repeat_index =
1339 SearchTimedOutRepeatIndex(timeshift_limit, *first_segment_info);
1340 CHECK_GE(repeat_index, 0);
1341 if (repeat_index == 0)
1344 first_segment_info->start_time = first_segment_info->start_time +
1345 first_segment_info->duration * repeat_index;
1347 first_segment_info->repeat = first_segment_info->repeat - repeat_index;
1348 start_number_ += repeat_index;
1351 std::string Representation::GetVideoMimeType()
const {
1352 return GetMimeType(
"video", media_info_.container_type());
1355 std::string Representation::GetAudioMimeType()
const {
1356 return GetMimeType(
"audio", media_info_.container_type());
1359 std::string Representation::GetTextMimeType()
const {
1360 CHECK(media_info_.has_text_info());
1361 if (media_info_.text_info().format() ==
"ttml") {
1362 switch (media_info_.container_type()) {
1363 case MediaInfo::CONTAINER_TEXT:
1364 return "application/ttml+xml";
1365 case MediaInfo::CONTAINER_MP4:
1366 return "application/mp4";
1368 LOG(ERROR) <<
"Failed to determine MIME type for TTML container: "
1369 << media_info_.container_type();
1373 if (media_info_.text_info().format() ==
"vtt") {
1374 if (media_info_.container_type() == MediaInfo::CONTAINER_TEXT) {
1377 LOG(ERROR) <<
"Failed to determine MIME type for VTT container: "
1378 << media_info_.container_type();
1382 LOG(ERROR) <<
"Cannot determine MIME type for format: "
1383 << media_info_.text_info().format()
1384 <<
" container: " << media_info_.container_type();
1388 bool Representation::GetEarliestTimestamp(
double* timestamp_seconds) {
1389 DCHECK(timestamp_seconds);
1391 if (segment_infos_.empty())
1394 *timestamp_seconds =
static_cast<double>(segment_infos_.begin()->start_time) /
1395 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)
bool WriteMpdToFile(media::File *output_file)
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)
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)