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 const int kAdaptationSetGroupNotSet = -1;
45 AdaptationSet::Role MediaInfoTextTypeToRole(
46 MediaInfo::TextInfo::TextType type) {
48 case MediaInfo::TextInfo::UNKNOWN:
49 LOG(WARNING) <<
"Unknown text type, assuming subtitle.";
50 return AdaptationSet::kRoleSubtitle;
51 case MediaInfo::TextInfo::CAPTION:
52 return AdaptationSet::kRoleCaption;
53 case MediaInfo::TextInfo::SUBTITLE:
54 return AdaptationSet::kRoleSubtitle;
56 NOTREACHED() <<
"Unknown MediaInfo TextType: " << type
57 <<
" assuming subtitle.";
58 return AdaptationSet::kRoleSubtitle;
62 std::string GetMimeType(
const std::string& prefix,
63 MediaInfo::ContainerType container_type) {
64 switch (container_type) {
65 case MediaInfo::CONTAINER_MP4:
66 return prefix +
"/mp4";
67 case MediaInfo::CONTAINER_MPEG2_TS:
69 return prefix +
"/MP2T";
70 case MediaInfo::CONTAINER_WEBM:
71 return prefix +
"/webm";
77 LOG(ERROR) <<
"Unrecognized container type: " << container_type;
81 void AddMpdNameSpaceInfo(XmlNode* mpd) {
84 static const char kXmlNamespace[] =
"urn:mpeg:dash:schema:mpd:2011";
85 static const char kXmlNamespaceXsi[] =
86 "http://www.w3.org/2001/XMLSchema-instance";
87 static const char kXmlNamespaceXlink[] =
"http://www.w3.org/1999/xlink";
88 static const char kDashSchemaMpd2011[] =
89 "urn:mpeg:dash:schema:mpd:2011 DASH-MPD.xsd";
90 static const char kCencNamespace[] =
"urn:mpeg:cenc:2013";
92 mpd->SetStringAttribute(
"xmlns", kXmlNamespace);
93 mpd->SetStringAttribute(
"xmlns:xsi", kXmlNamespaceXsi);
94 mpd->SetStringAttribute(
"xmlns:xlink", kXmlNamespaceXlink);
95 mpd->SetStringAttribute(
"xsi:schemaLocation", kDashSchemaMpd2011);
96 mpd->SetStringAttribute(
"xmlns:cenc", kCencNamespace);
99 bool IsPeriodNode(xmlNodePtr node) {
102 return xmlStrcmp(node->name, reinterpret_cast<const xmlChar*>(
"Period")) ==
111 xmlNodePtr FindPeriodNode(XmlNode* xml_node) {
112 for (xmlNodePtr node = xml_node->GetRawPtr()->xmlChildrenNode; node != NULL;
114 if (IsPeriodNode(node))
121 bool Positive(
double d) {
127 std::string XmlDateTimeNowWithOffset(
128 int32_t offset_seconds,
129 base::Clock* clock) {
130 base::Time time = clock->Now();
131 time += base::TimeDelta::FromSeconds(offset_seconds);
132 base::Time::Exploded time_exploded;
133 time.UTCExplode(&time_exploded);
135 return base::StringPrintf(
"%4d-%02d-%02dT%02d:%02d:%02dZ", time_exploded.year,
136 time_exploded.month, time_exploded.day_of_month,
137 time_exploded.hour, time_exploded.minute,
138 time_exploded.second);
141 void SetIfPositive(
const char* attr_name,
double value, XmlNode* mpd) {
142 if (Positive(value)) {
143 mpd->SetStringAttribute(attr_name, SecondsToXmlDuration(value));
147 uint32_t GetTimeScale(
const MediaInfo& media_info) {
148 if (media_info.has_reference_time_scale()) {
149 return media_info.reference_time_scale();
152 if (media_info.has_video_info()) {
153 return media_info.video_info().time_scale();
156 if (media_info.has_audio_info()) {
157 return media_info.audio_info().time_scale();
160 LOG(WARNING) <<
"No timescale specified, using 1 as timescale.";
164 uint64_t LastSegmentStartTime(
const SegmentInfo& segment_info) {
165 return segment_info.start_time + segment_info.duration * segment_info.repeat;
169 uint64_t LastSegmentEndTime(
const SegmentInfo& segment_info) {
170 return segment_info.start_time +
171 segment_info.duration * (segment_info.repeat + 1);
174 uint64_t LatestSegmentStartTime(
const std::list<SegmentInfo>& segments) {
175 DCHECK(!segments.empty());
176 const SegmentInfo& latest_segment = segments.back();
177 return LastSegmentStartTime(latest_segment);
182 int SearchTimedOutRepeatIndex(uint64_t timeshift_limit,
183 const SegmentInfo& segment_info) {
184 DCHECK_LE(timeshift_limit, LastSegmentEndTime(segment_info));
185 if (timeshift_limit < segment_info.start_time)
188 return (timeshift_limit - segment_info.start_time) / segment_info.duration;
194 bool WriteXmlCharArrayToOutput(xmlChar* doc,
196 std::string* output) {
199 output->assign(doc, doc + doc_size);
203 bool WriteXmlCharArrayToOutput(xmlChar* doc,
205 media::File* output) {
208 if (output->Write(doc, doc_size) < doc_size)
211 return output->Flush();
214 std::string MakePathRelative(
const std::string& path,
215 const std::string& mpd_dir) {
216 return (path.find(mpd_dir) == 0) ? path.substr(mpd_dir.size()) : path;
222 bool HasRequiredVideoFields(
const MediaInfo_VideoInfo& video_info) {
223 if (!video_info.has_height() || !video_info.has_width()) {
225 <<
"Width and height are required fields for generating a valid MPD.";
230 LOG_IF(WARNING, !video_info.has_time_scale())
231 <<
"Video info does not contain timescale required for "
232 "calculating framerate. @frameRate is required for DASH IOP.";
233 LOG_IF(WARNING, !video_info.has_frame_duration())
234 <<
"Video info does not contain frame duration required "
235 "for calculating framerate. @frameRate is required for DASH IOP.";
236 LOG_IF(WARNING, !video_info.has_pixel_width())
237 <<
"Video info does not contain pixel_width to calculate the sample "
238 "aspect ratio required for DASH IOP.";
239 LOG_IF(WARNING, !video_info.has_pixel_height())
240 <<
"Video info does not contain pixel_height to calculate the sample "
241 "aspect ratio required for DASH IOP.";
252 std::string GetPictureAspectRatio(uint32_t width,
254 uint32_t pixel_width,
255 uint32_t pixel_height) {
256 const uint32_t scaled_width = pixel_width * width;
257 const uint32_t scaled_height = pixel_height * height;
258 const double par =
static_cast<double>(scaled_width) / scaled_height;
262 const uint32_t kLargestPossibleParY = 19;
264 uint32_t par_num = 0;
265 uint32_t par_den = 0;
266 double min_error = 1.0;
267 for (uint32_t den = 1; den <= kLargestPossibleParY; ++den) {
268 uint32_t num = par * den + 0.5;
269 double error = fabs(par - static_cast<double>(num) / den);
270 if (error < min_error) {
274 if (error == 0)
break;
277 VLOG(2) <<
"width*pix_width : height*pixel_height (" << scaled_width <<
":"
278 << scaled_height <<
") reduced to " << par_num <<
":" << par_den
279 <<
" with error " << min_error <<
".";
281 return base::IntToString(par_num) +
":" + base::IntToString(par_den);
286 void AddPictureAspectRatio(
287 const MediaInfo::VideoInfo& video_info,
288 std::set<std::string>* picture_aspect_ratio) {
291 if (picture_aspect_ratio->size() > 1)
294 if (video_info.width() == 0 || video_info.height() == 0 ||
295 video_info.pixel_width() == 0 || video_info.pixel_height() == 0) {
300 picture_aspect_ratio->insert(
"bogus");
301 picture_aspect_ratio->insert(
"entries");
305 const std::string par = GetPictureAspectRatio(
306 video_info.width(), video_info.height(),
307 video_info.pixel_width(), video_info.pixel_height());
308 DVLOG(1) <<
"Setting par as: " << par
309 <<
" for video with width: " << video_info.width()
310 <<
" height: " << video_info.height()
311 <<
" pixel_width: " << video_info.pixel_width() <<
" pixel_height; "
312 << video_info.pixel_height();
313 picture_aspect_ratio->insert(par);
316 std::string RoleToText(AdaptationSet::Role role) {
320 case AdaptationSet::kRoleCaption:
322 case AdaptationSet::kRoleSubtitle:
324 case AdaptationSet::kRoleMain:
326 case AdaptationSet::kRoleAlternate:
328 case AdaptationSet::kRoleSupplementary:
329 return "supplementary";
330 case AdaptationSet::kRoleCommentary:
332 case AdaptationSet::kRoleDub:
343 class LibXmlInitializer {
345 LibXmlInitializer() : initialized_(false) {
346 base::AutoLock lock(lock_);
353 ~LibXmlInitializer() {
354 base::AutoLock lock(lock_);
357 initialized_ =
false;
365 DISALLOW_COPY_AND_ASSIGN(LibXmlInitializer);
368 class RepresentationStateChangeListenerImpl
369 :
public RepresentationStateChangeListener {
372 RepresentationStateChangeListenerImpl(uint32_t representation_id,
373 AdaptationSet* adaptation_set)
374 : representation_id_(representation_id), adaptation_set_(adaptation_set) {
375 DCHECK(adaptation_set_);
377 ~RepresentationStateChangeListenerImpl()
override {}
380 void OnNewSegmentForRepresentation(uint64_t start_time,
381 uint64_t duration)
override {
382 adaptation_set_->OnNewSegmentForRepresentation(representation_id_,
383 start_time, duration);
386 void OnSetFrameRateForRepresentation(uint32_t frame_duration,
387 uint32_t timescale)
override {
388 adaptation_set_->OnSetFrameRateForRepresentation(representation_id_,
389 frame_duration, timescale);
393 const uint32_t representation_id_;
394 AdaptationSet*
const adaptation_set_;
396 DISALLOW_COPY_AND_ASSIGN(RepresentationStateChangeListenerImpl);
403 mpd_options_(mpd_options),
404 clock_(new base::DefaultClock()) {}
406 MpdBuilder::~MpdBuilder() {}
409 base_urls_.push_back(base_url);
413 std::unique_ptr<AdaptationSet> adaptation_set(
414 new AdaptationSet(adaptation_set_counter_.GetNext(), lang, mpd_options_,
415 type_, &representation_counter_));
416 DCHECK(adaptation_set);
418 if (!lang.empty() && lang == mpd_options_.default_language) {
419 adaptation_set->AddRole(AdaptationSet::kRoleMain);
422 adaptation_sets_.push_back(std::move(adaptation_set));
423 return adaptation_sets_.back().get();
428 return WriteMpdToOutput(output_file);
433 return WriteMpdToOutput(output);
435 template <
typename OutputType>
436 bool MpdBuilder::WriteMpdToOutput(OutputType* output) {
437 static LibXmlInitializer lib_xml_initializer;
439 xml::scoped_xml_ptr<xmlDoc> doc(GenerateMpd());
443 static const int kNiceFormat = 1;
444 int doc_str_size = 0;
445 xmlChar* doc_str = NULL;
446 xmlDocDumpFormatMemoryEnc(doc.get(), &doc_str, &doc_str_size,
"UTF-8",
449 bool result = WriteXmlCharArrayToOutput(doc_str, doc_str_size, output);
457 xmlDocPtr MpdBuilder::GenerateMpd() {
459 static const char kXmlVersion[] =
"1.0";
460 xml::scoped_xml_ptr<xmlDoc> doc(xmlNewDoc(BAD_CAST kXmlVersion));
464 XmlNode period(
"Period");
470 for (
const std::unique_ptr<AdaptationSet>& adaptation_set :
472 xml::scoped_xml_ptr<xmlNode> child(adaptation_set->GetXml());
473 if (!child.get() || !period.AddChild(std::move(child)))
478 std::list<std::string>::const_iterator base_urls_it = base_urls_.begin();
479 for (; base_urls_it != base_urls_.end(); ++base_urls_it) {
480 XmlNode base_url(
"BaseURL");
481 base_url.SetContent(*base_urls_it);
483 if (!mpd.AddChild(base_url.PassScopedPtr()))
487 if (type_ == kDynamic) {
489 period.SetStringAttribute(
"start",
"PT0S");
492 if (!mpd.AddChild(period.PassScopedPtr()))
495 AddMpdNameSpaceInfo(&mpd);
496 AddCommonMpdInfo(&mpd);
499 AddStaticMpdInfo(&mpd);
502 AddDynamicMpdInfo(&mpd);
505 NOTREACHED() <<
"Unknown MPD type: " << type_;
510 const std::string version = GetPackagerVersion();
511 if (!version.empty()) {
512 std::string version_string =
513 base::StringPrintf(
"Generated with %s version %s",
514 GetPackagerProjectUrl().c_str(), version.c_str());
515 xml::scoped_xml_ptr<xmlNode> comment(
516 xmlNewDocComment(doc.get(), BAD_CAST version_string.c_str()));
517 xmlDocSetRootElement(doc.get(), comment.get());
518 xmlAddSibling(comment.release(), mpd.Release());
520 xmlDocSetRootElement(doc.get(), mpd.Release());
522 return doc.release();
525 void MpdBuilder::AddCommonMpdInfo(XmlNode* mpd_node) {
526 if (Positive(mpd_options_.min_buffer_time)) {
527 mpd_node->SetStringAttribute(
528 "minBufferTime", SecondsToXmlDuration(mpd_options_.min_buffer_time));
530 LOG(ERROR) <<
"minBufferTime value not specified.";
535 void MpdBuilder::AddStaticMpdInfo(XmlNode* mpd_node) {
537 DCHECK_EQ(MpdBuilder::kStatic, type_);
539 static const char kStaticMpdType[] =
"static";
540 static const char kStaticMpdProfile[] =
541 "urn:mpeg:dash:profile:isoff-on-demand:2011";
542 mpd_node->SetStringAttribute(
"type", kStaticMpdType);
543 mpd_node->SetStringAttribute(
"profiles", kStaticMpdProfile);
544 mpd_node->SetStringAttribute(
545 "mediaPresentationDuration",
546 SecondsToXmlDuration(GetStaticMpdDuration(mpd_node)));
549 void MpdBuilder::AddDynamicMpdInfo(XmlNode* mpd_node) {
551 DCHECK_EQ(MpdBuilder::kDynamic, type_);
553 static const char kDynamicMpdType[] =
"dynamic";
554 static const char kDynamicMpdProfile[] =
555 "urn:mpeg:dash:profile:isoff-live:2011";
556 mpd_node->SetStringAttribute(
"type", kDynamicMpdType);
557 mpd_node->SetStringAttribute(
"profiles", kDynamicMpdProfile);
560 mpd_node->SetStringAttribute(
"publishTime",
561 XmlDateTimeNowWithOffset(0, clock_.get()));
565 if (availability_start_time_.empty()) {
566 double earliest_presentation_time;
567 if (GetEarliestTimestamp(&earliest_presentation_time)) {
568 availability_start_time_ =
569 XmlDateTimeNowWithOffset(mpd_options_.availability_time_offset -
570 std::ceil(earliest_presentation_time),
573 LOG(ERROR) <<
"Could not determine the earliest segment presentation "
574 "time for availabilityStartTime calculation.";
578 if (!availability_start_time_.empty())
579 mpd_node->SetStringAttribute(
"availabilityStartTime",
580 availability_start_time_);
582 if (Positive(mpd_options_.minimum_update_period)) {
583 mpd_node->SetStringAttribute(
584 "minimumUpdatePeriod",
585 SecondsToXmlDuration(mpd_options_.minimum_update_period));
587 LOG(WARNING) <<
"The profile is dynamic but no minimumUpdatePeriod "
591 SetIfPositive(
"timeShiftBufferDepth", mpd_options_.time_shift_buffer_depth,
593 SetIfPositive(
"suggestedPresentationDelay",
594 mpd_options_.suggested_presentation_delay, mpd_node);
597 float MpdBuilder::GetStaticMpdDuration(XmlNode* mpd_node) {
599 DCHECK_EQ(MpdBuilder::kStatic, type_);
601 xmlNodePtr period_node = FindPeriodNode(mpd_node);
602 DCHECK(period_node) <<
"Period element must be a child of mpd_node.";
603 DCHECK(IsPeriodNode(period_node));
608 float max_duration = 0.0f;
609 for (xmlNodePtr adaptation_set = xmlFirstElementChild(period_node);
610 adaptation_set; adaptation_set = xmlNextElementSibling(adaptation_set)) {
611 for (xmlNodePtr representation = xmlFirstElementChild(adaptation_set);
613 representation = xmlNextElementSibling(representation)) {
614 float duration = 0.0f;
615 if (GetDurationAttribute(representation, &duration)) {
616 max_duration = max_duration > duration ? max_duration : duration;
620 xmlUnsetProp(representation, BAD_CAST
"duration");
628 bool MpdBuilder::GetEarliestTimestamp(
double* timestamp_seconds) {
629 DCHECK(timestamp_seconds);
631 double earliest_timestamp(-1);
632 for (
const std::unique_ptr<AdaptationSet>& adaptation_set :
635 if (adaptation_set->GetEarliestTimestamp(×tamp) &&
636 ((earliest_timestamp < 0) || (timestamp < earliest_timestamp))) {
637 earliest_timestamp = timestamp;
640 if (earliest_timestamp < 0)
643 *timestamp_seconds = earliest_timestamp;
648 MediaInfo* media_info) {
650 const std::string kFileProtocol(
"file://");
651 std::string mpd_file_path = (mpd_path.find(kFileProtocol) == 0)
652 ? mpd_path.substr(kFileProtocol.size())
655 if (!mpd_file_path.empty()) {
656 std::string mpd_dir(FilePath::FromUTF8Unsafe(mpd_file_path)
657 .DirName().AsEndingWithSeparator().AsUTF8Unsafe());
658 if (!mpd_dir.empty()) {
659 if (media_info->has_media_file_name()) {
660 media_info->set_media_file_name(
661 MakePathRelative(media_info->media_file_name(), mpd_dir));
663 if (media_info->has_init_segment_name()) {
664 media_info->set_init_segment_name(
665 MakePathRelative(media_info->init_segment_name(), mpd_dir));
667 if (media_info->has_segment_template()) {
668 media_info->set_segment_template(
669 MakePathRelative(media_info->segment_template(), mpd_dir));
676 const std::string& lang,
678 MpdBuilder::MpdType mpd_type,
679 base::AtomicSequenceNumber* counter)
680 : representation_counter_(counter),
681 id_(adaptation_set_id),
683 mpd_options_(mpd_options),
685 group_(kAdaptationSetGroupNotSet),
686 segments_aligned_(kSegmentAlignmentUnknown),
687 force_set_segment_alignment_(false) {
691 AdaptationSet::~AdaptationSet() {}
694 const uint32_t representation_id = representation_counter_->GetNext();
697 std::unique_ptr<RepresentationStateChangeListener> listener(
698 new RepresentationStateChangeListenerImpl(representation_id,
this));
699 std::unique_ptr<Representation> representation(
new Representation(
700 media_info, mpd_options_, representation_id, std::move(listener)));
702 if (!representation->Init())
707 if (media_info.has_video_info()) {
708 const MediaInfo::VideoInfo& video_info = media_info.video_info();
709 DCHECK(video_info.has_width());
710 DCHECK(video_info.has_height());
711 video_widths_.insert(video_info.width());
712 video_heights_.insert(video_info.height());
714 if (video_info.has_time_scale() && video_info.has_frame_duration())
715 RecordFrameRate(video_info.frame_duration(), video_info.time_scale());
717 AddPictureAspectRatio(video_info, &picture_aspect_ratio_);
720 if (media_info.has_video_info()) {
721 content_type_ =
"video";
722 }
else if (media_info.has_audio_info()) {
723 content_type_ =
"audio";
724 }
else if (media_info.has_text_info()) {
725 content_type_ =
"text";
727 if (media_info.text_info().has_type() &&
728 (media_info.text_info().type() != MediaInfo::TextInfo::UNKNOWN)) {
729 roles_.insert(MediaInfoTextTypeToRole(media_info.text_info().type()));
733 representations_.push_back(std::move(representation));
734 return representations_.back().get();
739 content_protection_elements_.push_back(content_protection_element);
740 RemoveDuplicateAttributes(&content_protection_elements_.back());
744 const std::string& pssh) {
745 UpdateContentProtectionPsshHelper(drm_uuid, pssh,
746 &content_protection_elements_);
760 AdaptationSetXmlNode adaptation_set;
762 bool suppress_representation_width =
false;
763 bool suppress_representation_height =
false;
764 bool suppress_representation_frame_rate =
false;
766 adaptation_set.SetId(id_);
767 adaptation_set.SetStringAttribute(
"contentType", content_type_);
768 if (!lang_.empty() && lang_ !=
"und") {
773 if (video_widths_.size() == 1) {
774 suppress_representation_width =
true;
775 adaptation_set.SetIntegerAttribute(
"width", *video_widths_.begin());
776 }
else if (video_widths_.size() > 1) {
777 adaptation_set.SetIntegerAttribute(
"maxWidth", *video_widths_.rbegin());
779 if (video_heights_.size() == 1) {
780 suppress_representation_height =
true;
781 adaptation_set.SetIntegerAttribute(
"height", *video_heights_.begin());
782 }
else if (video_heights_.size() > 1) {
783 adaptation_set.SetIntegerAttribute(
"maxHeight", *video_heights_.rbegin());
786 if (video_frame_rates_.size() == 1) {
787 suppress_representation_frame_rate =
true;
788 adaptation_set.SetStringAttribute(
"frameRate",
789 video_frame_rates_.begin()->second);
790 }
else if (video_frame_rates_.size() > 1) {
791 adaptation_set.SetStringAttribute(
"maxFrameRate",
792 video_frame_rates_.rbegin()->second);
797 if (mpd_type_ == MpdBuilder::kStatic) {
798 CheckVodSegmentAlignment();
801 if (segments_aligned_ == kSegmentAlignmentTrue) {
802 adaptation_set.SetStringAttribute(mpd_type_ == MpdBuilder::kStatic
803 ?
"subsegmentAlignment"
804 :
"segmentAlignment",
808 if (picture_aspect_ratio_.size() == 1)
809 adaptation_set.SetStringAttribute(
"par", *picture_aspect_ratio_.begin());
812 adaptation_set.SetIntegerAttribute(
"group", group_);
814 if (!adaptation_set.AddContentProtectionElements(
815 content_protection_elements_)) {
816 return xml::scoped_xml_ptr<xmlNode>();
818 for (AdaptationSet::Role role : roles_)
819 adaptation_set.AddRoleElement(
"urn:mpeg:dash:role:2011", RoleToText(role));
821 for (
const std::unique_ptr<Representation>& representation :
823 if (suppress_representation_width)
824 representation->SuppressOnce(Representation::kSuppressWidth);
825 if (suppress_representation_height)
826 representation->SuppressOnce(Representation::kSuppressHeight);
827 if (suppress_representation_frame_rate)
828 representation->SuppressOnce(Representation::kSuppressFrameRate);
829 xml::scoped_xml_ptr<xmlNode> child(representation->GetXml());
830 if (!child || !adaptation_set.AddChild(std::move(child)))
831 return xml::scoped_xml_ptr<xmlNode>();
834 return adaptation_set.PassScopedPtr();
839 segment_alignment ? kSegmentAlignmentTrue : kSegmentAlignmentFalse;
840 force_set_segment_alignment_ =
true;
844 group_ = group_number;
861 if (mpd_type_ == MpdBuilder::kDynamic) {
862 CheckLiveSegmentAlignment(representation_id, start_time, duration);
864 representation_segment_start_times_[representation_id].push_back(
870 uint32_t representation_id,
871 uint32_t frame_duration,
872 uint32_t timescale) {
873 RecordFrameRate(frame_duration, timescale);
876 bool AdaptationSet::GetEarliestTimestamp(
double* timestamp_seconds) {
877 DCHECK(timestamp_seconds);
879 double earliest_timestamp(-1);
880 for (
const std::unique_ptr<Representation>& representation :
883 if (representation->GetEarliestTimestamp(×tamp) &&
884 ((earliest_timestamp < 0) || (timestamp < earliest_timestamp))) {
885 earliest_timestamp = timestamp;
888 if (earliest_timestamp < 0)
891 *timestamp_seconds = earliest_timestamp;
919 void AdaptationSet::CheckLiveSegmentAlignment(uint32_t representation_id,
922 if (segments_aligned_ == kSegmentAlignmentFalse ||
923 force_set_segment_alignment_) {
927 std::list<uint64_t>& representation_start_times =
928 representation_segment_start_times_[representation_id];
929 representation_start_times.push_back(start_time);
932 if (representation_segment_start_times_.size() != representations_.size())
935 DCHECK(!representation_start_times.empty());
936 const uint64_t expected_start_time = representation_start_times.front();
937 for (RepresentationTimeline::const_iterator it =
938 representation_segment_start_times_.begin();
939 it != representation_segment_start_times_.end(); ++it) {
943 if (it->second.empty())
946 if (expected_start_time != it->second.front()) {
949 segments_aligned_ = kSegmentAlignmentFalse;
950 representation_segment_start_times_.clear();
954 segments_aligned_ = kSegmentAlignmentTrue;
956 for (RepresentationTimeline::iterator it =
957 representation_segment_start_times_.begin();
958 it != representation_segment_start_times_.end(); ++it) {
959 it->second.pop_front();
965 void AdaptationSet::CheckVodSegmentAlignment() {
966 if (segments_aligned_ == kSegmentAlignmentFalse ||
967 force_set_segment_alignment_) {
970 if (representation_segment_start_times_.empty())
972 if (representation_segment_start_times_.size() == 1) {
973 segments_aligned_ = kSegmentAlignmentTrue;
980 const std::list<uint64_t>& expected_time_line =
981 representation_segment_start_times_.begin()->second;
983 bool all_segment_time_line_same_length =
true;
985 RepresentationTimeline::const_iterator it =
986 representation_segment_start_times_.begin();
987 for (++it; it != representation_segment_start_times_.end(); ++it) {
988 const std::list<uint64_t>& other_time_line = it->second;
989 if (expected_time_line.size() != other_time_line.size()) {
990 all_segment_time_line_same_length =
false;
993 const std::list<uint64_t>* longer_list = &other_time_line;
994 const std::list<uint64_t>* shorter_list = &expected_time_line;
995 if (expected_time_line.size() > other_time_line.size()) {
996 shorter_list = &other_time_line;
997 longer_list = &expected_time_line;
1000 if (!std::equal(shorter_list->begin(), shorter_list->end(),
1001 longer_list->begin())) {
1003 segments_aligned_ = kSegmentAlignmentFalse;
1004 representation_segment_start_times_.clear();
1015 if (!all_segment_time_line_same_length) {
1016 segments_aligned_ = kSegmentAlignmentUnknown;
1020 segments_aligned_ = kSegmentAlignmentTrue;
1025 void AdaptationSet::RecordFrameRate(uint32_t frame_duration,
1026 uint32_t timescale) {
1027 if (frame_duration == 0) {
1028 LOG(ERROR) <<
"Frame duration is 0 and cannot be set.";
1031 video_frame_rates_[
static_cast<double>(timescale) / frame_duration] =
1032 base::IntToString(timescale) +
"/" + base::IntToString(frame_duration);
1036 const MediaInfo& media_info,
1039 std::unique_ptr<RepresentationStateChangeListener> state_change_listener)
1040 : media_info_(media_info),
1043 mpd_options_(mpd_options),
1045 state_change_listener_(std::move(state_change_listener)),
1046 output_suppression_flags_(0) {}
1048 Representation::~Representation() {}
1051 if (!AtLeastOneTrue(media_info_.has_video_info(),
1052 media_info_.has_audio_info(),
1053 media_info_.has_text_info())) {
1057 LOG(ERROR) <<
"Representation needs one of video, audio, or text.";
1061 if (MoreThanOneTrue(media_info_.has_video_info(),
1062 media_info_.has_audio_info(),
1063 media_info_.has_text_info())) {
1064 LOG(ERROR) <<
"Only one of VideoInfo, AudioInfo, or TextInfo can be set.";
1068 if (media_info_.container_type() == MediaInfo::CONTAINER_UNKNOWN) {
1069 LOG(ERROR) <<
"'container_type' in MediaInfo cannot be CONTAINER_UNKNOWN.";
1073 if (media_info_.has_video_info()) {
1074 mime_type_ = GetVideoMimeType();
1075 if (!HasRequiredVideoFields(media_info_.video_info())) {
1076 LOG(ERROR) <<
"Missing required fields to create a video Representation.";
1079 }
else if (media_info_.has_audio_info()) {
1080 mime_type_ = GetAudioMimeType();
1081 }
else if (media_info_.has_text_info()) {
1082 mime_type_ = GetTextMimeType();
1085 if (mime_type_.empty())
1088 codecs_ = GetCodecs(media_info_);
1094 content_protection_elements_.push_back(content_protection_element);
1095 RemoveDuplicateAttributes(&content_protection_elements_.back());
1099 const std::string& pssh) {
1100 UpdateContentProtectionPsshHelper(drm_uuid, pssh,
1101 &content_protection_elements_);
1107 if (start_time == 0 && duration == 0) {
1108 LOG(WARNING) <<
"Got segment with start_time and duration == 0. Ignoring.";
1112 if (state_change_listener_)
1113 state_change_listener_->OnNewSegmentForRepresentation(start_time, duration);
1114 if (IsContiguous(start_time, duration, size)) {
1115 ++segment_infos_.back().repeat;
1118 segment_infos_.push_back(s);
1121 bandwidth_estimator_.AddBlock(
1122 size, static_cast<double>(duration) / media_info_.reference_time_scale());
1125 DCHECK_GE(segment_infos_.size(), 1u);
1129 if (media_info_.has_video_info()) {
1130 media_info_.mutable_video_info()->set_frame_duration(sample_duration);
1131 if (state_change_listener_) {
1132 state_change_listener_->OnSetFrameRateForRepresentation(
1133 sample_duration, media_info_.video_info().time_scale());
1145 if (!HasRequiredMediaInfoFields()) {
1146 LOG(ERROR) <<
"MediaInfo missing required fields.";
1147 return xml::scoped_xml_ptr<xmlNode>();
1150 const uint64_t bandwidth = media_info_.has_bandwidth()
1151 ? media_info_.bandwidth()
1152 : bandwidth_estimator_.Estimate();
1154 DCHECK(!(HasVODOnlyFields(media_info_) && HasLiveOnlyFields(media_info_)));
1156 RepresentationXmlNode representation;
1158 representation.SetId(id_);
1159 representation.SetIntegerAttribute(
"bandwidth", bandwidth);
1160 if (!codecs_.empty())
1161 representation.SetStringAttribute(
"codecs", codecs_);
1162 representation.SetStringAttribute(
"mimeType", mime_type_);
1164 const bool has_video_info = media_info_.has_video_info();
1165 const bool has_audio_info = media_info_.has_audio_info();
1167 if (has_video_info &&
1168 !representation.AddVideoInfo(
1169 media_info_.video_info(),
1170 !(output_suppression_flags_ & kSuppressWidth),
1171 !(output_suppression_flags_ & kSuppressHeight),
1172 !(output_suppression_flags_ & kSuppressFrameRate))) {
1173 LOG(ERROR) <<
"Failed to add video info to Representation XML.";
1174 return xml::scoped_xml_ptr<xmlNode>();
1177 if (has_audio_info &&
1178 !representation.AddAudioInfo(media_info_.audio_info())) {
1179 LOG(ERROR) <<
"Failed to add audio info to Representation XML.";
1180 return xml::scoped_xml_ptr<xmlNode>();
1183 if (!representation.AddContentProtectionElements(
1184 content_protection_elements_)) {
1185 return xml::scoped_xml_ptr<xmlNode>();
1188 if (HasVODOnlyFields(media_info_) &&
1189 !representation.AddVODOnlyInfo(media_info_)) {
1190 LOG(ERROR) <<
"Failed to add VOD segment info.";
1191 return xml::scoped_xml_ptr<xmlNode>();
1194 if (HasLiveOnlyFields(media_info_) &&
1195 !representation.AddLiveOnlyInfo(media_info_, segment_infos_,
1197 LOG(ERROR) <<
"Failed to add Live info.";
1198 return xml::scoped_xml_ptr<xmlNode>();
1203 output_suppression_flags_ = 0;
1204 return representation.PassScopedPtr();
1208 output_suppression_flags_ |= flag;
1211 bool Representation::HasRequiredMediaInfoFields() {
1212 if (HasVODOnlyFields(media_info_) && HasLiveOnlyFields(media_info_)) {
1213 LOG(ERROR) <<
"MediaInfo cannot have both VOD and Live fields.";
1217 if (!media_info_.has_container_type()) {
1218 LOG(ERROR) <<
"MediaInfo missing required field: container_type.";
1222 if (HasVODOnlyFields(media_info_) && !media_info_.has_bandwidth()) {
1223 LOG(ERROR) <<
"Missing 'bandwidth' field. MediaInfo requires bandwidth for "
1224 "static profile for generating a valid MPD.";
1228 VLOG_IF(3, HasLiveOnlyFields(media_info_) && !media_info_.has_bandwidth())
1229 <<
"MediaInfo missing field 'bandwidth'. Using estimated from "
1235 bool Representation::IsContiguous(uint64_t start_time,
1237 uint64_t size)
const {
1238 if (segment_infos_.empty())
1242 const SegmentInfo& previous = segment_infos_.back();
1243 const uint64_t previous_segment_end_time =
1244 previous.start_time + previous.duration * (previous.repeat + 1);
1245 if (previous_segment_end_time == start_time &&
1246 segment_infos_.back().duration == duration) {
1251 const uint64_t previous_segment_start_time =
1252 previous.start_time + previous.duration * previous.repeat;
1253 if (previous_segment_start_time >= start_time) {
1254 LOG(ERROR) <<
"Segments should not be out of order segment. Adding segment "
1255 "with start_time == "
1256 << start_time <<
" but the previous segment starts at "
1257 << previous.start_time <<
".";
1262 const uint64_t kRoundingErrorGrace = 5;
1263 if (previous_segment_end_time + kRoundingErrorGrace < start_time) {
1264 LOG(WARNING) <<
"Found a gap of size "
1265 << (start_time - previous_segment_end_time)
1266 <<
" > kRoundingErrorGrace (" << kRoundingErrorGrace
1267 <<
"). The new segment starts at " << start_time
1268 <<
" but the previous segment ends at "
1269 << previous_segment_end_time <<
".";
1274 if (start_time < previous_segment_end_time - kRoundingErrorGrace) {
1276 <<
"Segments should not be overlapping. The new segment starts at "
1277 << start_time <<
" but the previous segment ends at "
1278 << previous_segment_end_time <<
".";
1286 void Representation::SlideWindow() {
1287 DCHECK(!segment_infos_.empty());
1288 if (mpd_options_.time_shift_buffer_depth <= 0.0)
1291 const uint32_t time_scale = GetTimeScale(media_info_);
1292 DCHECK_GT(time_scale, 0u);
1294 uint64_t time_shift_buffer_depth =
1295 static_cast<uint64_t
>(mpd_options_.time_shift_buffer_depth * time_scale);
1299 const uint64_t current_play_time = LatestSegmentStartTime(segment_infos_);
1300 if (current_play_time <= time_shift_buffer_depth)
1303 const uint64_t timeshift_limit = current_play_time - time_shift_buffer_depth;
1307 std::list<SegmentInfo>::iterator first = segment_infos_.begin();
1308 std::list<SegmentInfo>::iterator last = first;
1309 size_t num_segments_removed = 0;
1310 for (; last != segment_infos_.end(); ++last) {
1311 const uint64_t last_segment_end_time = LastSegmentEndTime(*last);
1312 if (timeshift_limit < last_segment_end_time)
1314 num_segments_removed += last->repeat + 1;
1316 segment_infos_.erase(first, last);
1317 start_number_ += num_segments_removed;
1320 SegmentInfo* first_segment_info = &segment_infos_.front();
1321 DCHECK_LE(timeshift_limit, LastSegmentEndTime(*first_segment_info));
1324 const int repeat_index =
1325 SearchTimedOutRepeatIndex(timeshift_limit, *first_segment_info);
1326 CHECK_GE(repeat_index, 0);
1327 if (repeat_index == 0)
1330 first_segment_info->start_time = first_segment_info->start_time +
1331 first_segment_info->duration * repeat_index;
1333 first_segment_info->repeat = first_segment_info->repeat - repeat_index;
1334 start_number_ += repeat_index;
1337 std::string Representation::GetVideoMimeType()
const {
1338 return GetMimeType(
"video", media_info_.container_type());
1341 std::string Representation::GetAudioMimeType()
const {
1342 return GetMimeType(
"audio", media_info_.container_type());
1345 std::string Representation::GetTextMimeType()
const {
1346 CHECK(media_info_.has_text_info());
1347 if (media_info_.text_info().format() ==
"ttml") {
1348 switch (media_info_.container_type()) {
1349 case MediaInfo::CONTAINER_TEXT:
1350 return "application/ttml+xml";
1351 case MediaInfo::CONTAINER_MP4:
1352 return "application/mp4";
1354 LOG(ERROR) <<
"Failed to determine MIME type for TTML container: "
1355 << media_info_.container_type();
1359 if (media_info_.text_info().format() ==
"vtt") {
1360 if (media_info_.container_type() == MediaInfo::CONTAINER_TEXT) {
1363 LOG(ERROR) <<
"Failed to determine MIME type for VTT container: "
1364 << media_info_.container_type();
1368 LOG(ERROR) <<
"Cannot determine MIME type for format: "
1369 << media_info_.text_info().format()
1370 <<
" container: " << media_info_.container_type();
1374 bool Representation::GetEarliestTimestamp(
double* timestamp_seconds) {
1375 DCHECK(timestamp_seconds);
1377 if (segment_infos_.empty())
1380 *timestamp_seconds =
static_cast<double>(segment_infos_.begin()->start_time) /
1381 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)
AdaptationSet(uint32_t adaptation_set_id, const std::string &lang, const MpdOptions &mpd_options, MpdBuilder::MpdType mpd_type, base::AtomicSequenceNumber *representation_counter)
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 AddRole(Role role)
void AddBaseUrl(const std::string &base_url)
virtual void UpdateContentProtectionPssh(const std::string &drm_uuid, const std::string &pssh)
virtual void SetGroup(int group_number)
xml::scoped_xml_ptr< xmlNode > GetXml()
virtual bool ToString(std::string *output)
virtual void ForceSetSegmentAlignment(bool segment_alignment)
static void MakePathsRelativeToMpd(const std::string &mpd_path, MediaInfo *media_info)
MpdBuilder(MpdType type, const MpdOptions &mpd_options)
xml::scoped_xml_ptr< xmlNode > GetXml()
virtual void AddContentProtectionElement(const ContentProtectionElement &element)
virtual AdaptationSet * AddAdaptationSet(const std::string &lang)
virtual int Group() const
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)