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_));
417 DCHECK(adaptation_set);
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()))
483 if (type_ == kDynamic) {
485 period.SetStringAttribute(
"start",
"PT0S");
488 if (!mpd.AddChild(period.PassScopedPtr()))
491 AddMpdNameSpaceInfo(&mpd);
492 AddCommonMpdInfo(&mpd);
495 AddStaticMpdInfo(&mpd);
498 AddDynamicMpdInfo(&mpd);
501 NOTREACHED() <<
"Unknown MPD type: " << type_;
506 const std::string version = GetPackagerVersion();
507 if (!version.empty()) {
508 std::string version_string =
509 base::StringPrintf(
"Generated with %s version %s",
510 GetPackagerProjectUrl().c_str(), version.c_str());
511 xml::scoped_xml_ptr<xmlNode> comment(
512 xmlNewDocComment(doc.get(), BAD_CAST version_string.c_str()));
513 xmlDocSetRootElement(doc.get(), comment.get());
514 xmlAddSibling(comment.release(), mpd.Release());
516 xmlDocSetRootElement(doc.get(), mpd.Release());
518 return doc.release();
521 void MpdBuilder::AddCommonMpdInfo(XmlNode* mpd_node) {
522 if (Positive(mpd_options_.min_buffer_time)) {
523 mpd_node->SetStringAttribute(
524 "minBufferTime", SecondsToXmlDuration(mpd_options_.min_buffer_time));
526 LOG(ERROR) <<
"minBufferTime value not specified.";
531 void MpdBuilder::AddStaticMpdInfo(XmlNode* mpd_node) {
533 DCHECK_EQ(MpdBuilder::kStatic, type_);
535 static const char kStaticMpdType[] =
"static";
536 static const char kStaticMpdProfile[] =
537 "urn:mpeg:dash:profile:isoff-on-demand:2011";
538 mpd_node->SetStringAttribute(
"type", kStaticMpdType);
539 mpd_node->SetStringAttribute(
"profiles", kStaticMpdProfile);
540 mpd_node->SetStringAttribute(
541 "mediaPresentationDuration",
542 SecondsToXmlDuration(GetStaticMpdDuration(mpd_node)));
545 void MpdBuilder::AddDynamicMpdInfo(XmlNode* mpd_node) {
547 DCHECK_EQ(MpdBuilder::kDynamic, type_);
549 static const char kDynamicMpdType[] =
"dynamic";
550 static const char kDynamicMpdProfile[] =
551 "urn:mpeg:dash:profile:isoff-live:2011";
552 mpd_node->SetStringAttribute(
"type", kDynamicMpdType);
553 mpd_node->SetStringAttribute(
"profiles", kDynamicMpdProfile);
556 mpd_node->SetStringAttribute(
"publishTime",
557 XmlDateTimeNowWithOffset(0, clock_.get()));
561 if (availability_start_time_.empty()) {
562 double earliest_presentation_time;
563 if (GetEarliestTimestamp(&earliest_presentation_time)) {
564 availability_start_time_ =
565 XmlDateTimeNowWithOffset(mpd_options_.availability_time_offset -
566 std::ceil(earliest_presentation_time),
569 LOG(ERROR) <<
"Could not determine the earliest segment presentation "
570 "time for availabilityStartTime calculation.";
574 if (!availability_start_time_.empty())
575 mpd_node->SetStringAttribute(
"availabilityStartTime",
576 availability_start_time_);
578 if (Positive(mpd_options_.minimum_update_period)) {
579 mpd_node->SetStringAttribute(
580 "minimumUpdatePeriod",
581 SecondsToXmlDuration(mpd_options_.minimum_update_period));
583 LOG(WARNING) <<
"The profile is dynamic but no minimumUpdatePeriod "
587 SetIfPositive(
"timeShiftBufferDepth", mpd_options_.time_shift_buffer_depth,
589 SetIfPositive(
"suggestedPresentationDelay",
590 mpd_options_.suggested_presentation_delay, mpd_node);
593 float MpdBuilder::GetStaticMpdDuration(XmlNode* mpd_node) {
595 DCHECK_EQ(MpdBuilder::kStatic, type_);
597 xmlNodePtr period_node = FindPeriodNode(mpd_node);
598 DCHECK(period_node) <<
"Period element must be a child of mpd_node.";
599 DCHECK(IsPeriodNode(period_node));
604 float max_duration = 0.0f;
605 for (xmlNodePtr adaptation_set = xmlFirstElementChild(period_node);
606 adaptation_set; adaptation_set = xmlNextElementSibling(adaptation_set)) {
607 for (xmlNodePtr representation = xmlFirstElementChild(adaptation_set);
609 representation = xmlNextElementSibling(representation)) {
610 float duration = 0.0f;
611 if (GetDurationAttribute(representation, &duration)) {
612 max_duration = max_duration > duration ? max_duration : duration;
616 xmlUnsetProp(representation, BAD_CAST
"duration");
624 bool MpdBuilder::GetEarliestTimestamp(
double* timestamp_seconds) {
625 DCHECK(timestamp_seconds);
627 double earliest_timestamp(-1);
628 for (
const std::unique_ptr<AdaptationSet>& adaptation_set :
631 if (adaptation_set->GetEarliestTimestamp(×tamp) &&
632 ((earliest_timestamp < 0) || (timestamp < earliest_timestamp))) {
633 earliest_timestamp = timestamp;
636 if (earliest_timestamp < 0)
639 *timestamp_seconds = earliest_timestamp;
644 MediaInfo* media_info) {
646 const std::string kFileProtocol(
"file://");
647 std::string mpd_file_path = (mpd_path.find(kFileProtocol) == 0)
648 ? mpd_path.substr(kFileProtocol.size())
651 if (!mpd_file_path.empty()) {
652 std::string mpd_dir(FilePath::FromUTF8Unsafe(mpd_file_path)
653 .DirName().AsEndingWithSeparator().AsUTF8Unsafe());
654 if (!mpd_dir.empty()) {
655 if (media_info->has_media_file_name()) {
656 media_info->set_media_file_name(
657 MakePathRelative(media_info->media_file_name(), mpd_dir));
659 if (media_info->has_init_segment_name()) {
660 media_info->set_init_segment_name(
661 MakePathRelative(media_info->init_segment_name(), mpd_dir));
663 if (media_info->has_segment_template()) {
664 media_info->set_segment_template(
665 MakePathRelative(media_info->segment_template(), mpd_dir));
672 const std::string& lang,
674 MpdBuilder::MpdType mpd_type,
675 base::AtomicSequenceNumber* counter)
676 : representation_counter_(counter),
677 id_(adaptation_set_id),
679 mpd_options_(mpd_options),
681 group_(kAdaptationSetGroupNotSet),
682 segments_aligned_(kSegmentAlignmentUnknown),
683 force_set_segment_alignment_(false) {
687 AdaptationSet::~AdaptationSet() {}
690 const uint32_t representation_id = representation_counter_->GetNext();
693 std::unique_ptr<RepresentationStateChangeListener> listener(
694 new RepresentationStateChangeListenerImpl(representation_id,
this));
695 std::unique_ptr<Representation> representation(
new Representation(
696 media_info, mpd_options_, representation_id, std::move(listener)));
698 if (!representation->Init())
703 if (media_info.has_video_info()) {
704 const MediaInfo::VideoInfo& video_info = media_info.video_info();
705 DCHECK(video_info.has_width());
706 DCHECK(video_info.has_height());
707 video_widths_.insert(video_info.width());
708 video_heights_.insert(video_info.height());
710 if (video_info.has_time_scale() && video_info.has_frame_duration())
711 RecordFrameRate(video_info.frame_duration(), video_info.time_scale());
713 AddPictureAspectRatio(video_info, &picture_aspect_ratio_);
716 if (media_info.has_video_info()) {
717 content_type_ =
"video";
718 }
else if (media_info.has_audio_info()) {
719 content_type_ =
"audio";
720 }
else if (media_info.has_text_info()) {
721 content_type_ =
"text";
723 if (media_info.text_info().has_type() &&
724 (media_info.text_info().type() != MediaInfo::TextInfo::UNKNOWN)) {
725 roles_.insert(MediaInfoTextTypeToRole(media_info.text_info().type()));
729 representations_.push_back(std::move(representation));
730 return representations_.back().get();
735 content_protection_elements_.push_back(content_protection_element);
736 RemoveDuplicateAttributes(&content_protection_elements_.back());
740 const std::string& pssh) {
741 UpdateContentProtectionPsshHelper(drm_uuid, pssh,
742 &content_protection_elements_);
756 AdaptationSetXmlNode adaptation_set;
758 bool suppress_representation_width =
false;
759 bool suppress_representation_height =
false;
760 bool suppress_representation_frame_rate =
false;
762 adaptation_set.SetId(id_);
763 adaptation_set.SetStringAttribute(
"contentType", content_type_);
764 if (!lang_.empty() && lang_ !=
"und") {
769 if (video_widths_.size() == 1) {
770 suppress_representation_width =
true;
771 adaptation_set.SetIntegerAttribute(
"width", *video_widths_.begin());
772 }
else if (video_widths_.size() > 1) {
773 adaptation_set.SetIntegerAttribute(
"maxWidth", *video_widths_.rbegin());
775 if (video_heights_.size() == 1) {
776 suppress_representation_height =
true;
777 adaptation_set.SetIntegerAttribute(
"height", *video_heights_.begin());
778 }
else if (video_heights_.size() > 1) {
779 adaptation_set.SetIntegerAttribute(
"maxHeight", *video_heights_.rbegin());
782 if (video_frame_rates_.size() == 1) {
783 suppress_representation_frame_rate =
true;
784 adaptation_set.SetStringAttribute(
"frameRate",
785 video_frame_rates_.begin()->second);
786 }
else if (video_frame_rates_.size() > 1) {
787 adaptation_set.SetStringAttribute(
"maxFrameRate",
788 video_frame_rates_.rbegin()->second);
793 if (mpd_type_ == MpdBuilder::kStatic) {
794 CheckVodSegmentAlignment();
797 if (segments_aligned_ == kSegmentAlignmentTrue) {
798 adaptation_set.SetStringAttribute(mpd_type_ == MpdBuilder::kStatic
799 ?
"subsegmentAlignment"
800 :
"segmentAlignment",
804 if (picture_aspect_ratio_.size() == 1)
805 adaptation_set.SetStringAttribute(
"par", *picture_aspect_ratio_.begin());
808 adaptation_set.SetIntegerAttribute(
"group", group_);
810 if (!adaptation_set.AddContentProtectionElements(
811 content_protection_elements_)) {
812 return xml::scoped_xml_ptr<xmlNode>();
814 for (AdaptationSet::Role role : roles_)
815 adaptation_set.AddRoleElement(
"urn:mpeg:dash:role:2011", RoleToText(role));
817 for (
const std::unique_ptr<Representation>& representation :
819 if (suppress_representation_width)
820 representation->SuppressOnce(Representation::kSuppressWidth);
821 if (suppress_representation_height)
822 representation->SuppressOnce(Representation::kSuppressHeight);
823 if (suppress_representation_frame_rate)
824 representation->SuppressOnce(Representation::kSuppressFrameRate);
825 xml::scoped_xml_ptr<xmlNode> child(representation->GetXml());
826 if (!child || !adaptation_set.AddChild(std::move(child)))
827 return xml::scoped_xml_ptr<xmlNode>();
830 return adaptation_set.PassScopedPtr();
835 segment_alignment ? kSegmentAlignmentTrue : kSegmentAlignmentFalse;
836 force_set_segment_alignment_ =
true;
840 group_ = group_number;
857 if (mpd_type_ == MpdBuilder::kDynamic) {
858 CheckLiveSegmentAlignment(representation_id, start_time, duration);
860 representation_segment_start_times_[representation_id].push_back(
866 uint32_t representation_id,
867 uint32_t frame_duration,
868 uint32_t timescale) {
869 RecordFrameRate(frame_duration, timescale);
872 bool AdaptationSet::GetEarliestTimestamp(
double* timestamp_seconds) {
873 DCHECK(timestamp_seconds);
875 double earliest_timestamp(-1);
876 for (
const std::unique_ptr<Representation>& representation :
879 if (representation->GetEarliestTimestamp(×tamp) &&
880 ((earliest_timestamp < 0) || (timestamp < earliest_timestamp))) {
881 earliest_timestamp = timestamp;
884 if (earliest_timestamp < 0)
887 *timestamp_seconds = earliest_timestamp;
915 void AdaptationSet::CheckLiveSegmentAlignment(uint32_t representation_id,
918 if (segments_aligned_ == kSegmentAlignmentFalse ||
919 force_set_segment_alignment_) {
923 std::list<uint64_t>& representation_start_times =
924 representation_segment_start_times_[representation_id];
925 representation_start_times.push_back(start_time);
928 if (representation_segment_start_times_.size() != representations_.size())
931 DCHECK(!representation_start_times.empty());
932 const uint64_t expected_start_time = representation_start_times.front();
933 for (RepresentationTimeline::const_iterator it =
934 representation_segment_start_times_.begin();
935 it != representation_segment_start_times_.end(); ++it) {
939 if (it->second.empty())
942 if (expected_start_time != it->second.front()) {
945 segments_aligned_ = kSegmentAlignmentFalse;
946 representation_segment_start_times_.clear();
950 segments_aligned_ = kSegmentAlignmentTrue;
952 for (RepresentationTimeline::iterator it =
953 representation_segment_start_times_.begin();
954 it != representation_segment_start_times_.end(); ++it) {
955 it->second.pop_front();
961 void AdaptationSet::CheckVodSegmentAlignment() {
962 if (segments_aligned_ == kSegmentAlignmentFalse ||
963 force_set_segment_alignment_) {
966 if (representation_segment_start_times_.empty())
968 if (representation_segment_start_times_.size() == 1) {
969 segments_aligned_ = kSegmentAlignmentTrue;
976 const std::list<uint64_t>& expected_time_line =
977 representation_segment_start_times_.begin()->second;
979 bool all_segment_time_line_same_length =
true;
981 RepresentationTimeline::const_iterator it =
982 representation_segment_start_times_.begin();
983 for (++it; it != representation_segment_start_times_.end(); ++it) {
984 const std::list<uint64_t>& other_time_line = it->second;
985 if (expected_time_line.size() != other_time_line.size()) {
986 all_segment_time_line_same_length =
false;
989 const std::list<uint64_t>* longer_list = &other_time_line;
990 const std::list<uint64_t>* shorter_list = &expected_time_line;
991 if (expected_time_line.size() > other_time_line.size()) {
992 shorter_list = &other_time_line;
993 longer_list = &expected_time_line;
996 if (!std::equal(shorter_list->begin(), shorter_list->end(),
997 longer_list->begin())) {
999 segments_aligned_ = kSegmentAlignmentFalse;
1000 representation_segment_start_times_.clear();
1011 if (!all_segment_time_line_same_length) {
1012 segments_aligned_ = kSegmentAlignmentUnknown;
1016 segments_aligned_ = kSegmentAlignmentTrue;
1021 void AdaptationSet::RecordFrameRate(uint32_t frame_duration,
1022 uint32_t timescale) {
1023 if (frame_duration == 0) {
1024 LOG(ERROR) <<
"Frame duration is 0 and cannot be set.";
1027 video_frame_rates_[
static_cast<double>(timescale) / frame_duration] =
1028 base::IntToString(timescale) +
"/" + base::IntToString(frame_duration);
1032 const MediaInfo& media_info,
1035 std::unique_ptr<RepresentationStateChangeListener> state_change_listener)
1036 : media_info_(media_info),
1039 mpd_options_(mpd_options),
1041 state_change_listener_(std::move(state_change_listener)),
1042 output_suppression_flags_(0) {}
1044 Representation::~Representation() {}
1047 if (!AtLeastOneTrue(media_info_.has_video_info(),
1048 media_info_.has_audio_info(),
1049 media_info_.has_text_info())) {
1053 LOG(ERROR) <<
"Representation needs one of video, audio, or text.";
1057 if (MoreThanOneTrue(media_info_.has_video_info(),
1058 media_info_.has_audio_info(),
1059 media_info_.has_text_info())) {
1060 LOG(ERROR) <<
"Only one of VideoInfo, AudioInfo, or TextInfo can be set.";
1064 if (media_info_.container_type() == MediaInfo::CONTAINER_UNKNOWN) {
1065 LOG(ERROR) <<
"'container_type' in MediaInfo cannot be CONTAINER_UNKNOWN.";
1069 if (media_info_.has_video_info()) {
1070 mime_type_ = GetVideoMimeType();
1071 if (!HasRequiredVideoFields(media_info_.video_info())) {
1072 LOG(ERROR) <<
"Missing required fields to create a video Representation.";
1075 }
else if (media_info_.has_audio_info()) {
1076 mime_type_ = GetAudioMimeType();
1077 }
else if (media_info_.has_text_info()) {
1078 mime_type_ = GetTextMimeType();
1081 if (mime_type_.empty())
1084 codecs_ = GetCodecs(media_info_);
1090 content_protection_elements_.push_back(content_protection_element);
1091 RemoveDuplicateAttributes(&content_protection_elements_.back());
1095 const std::string& pssh) {
1096 UpdateContentProtectionPsshHelper(drm_uuid, pssh,
1097 &content_protection_elements_);
1103 if (start_time == 0 && duration == 0) {
1104 LOG(WARNING) <<
"Got segment with start_time and duration == 0. Ignoring.";
1108 if (state_change_listener_)
1109 state_change_listener_->OnNewSegmentForRepresentation(start_time, duration);
1110 if (IsContiguous(start_time, duration, size)) {
1111 ++segment_infos_.back().repeat;
1114 segment_infos_.push_back(s);
1117 bandwidth_estimator_.AddBlock(
1118 size, static_cast<double>(duration) / media_info_.reference_time_scale());
1121 DCHECK_GE(segment_infos_.size(), 1u);
1125 if (media_info_.has_video_info()) {
1126 media_info_.mutable_video_info()->set_frame_duration(sample_duration);
1127 if (state_change_listener_) {
1128 state_change_listener_->OnSetFrameRateForRepresentation(
1129 sample_duration, media_info_.video_info().time_scale());
1141 if (!HasRequiredMediaInfoFields()) {
1142 LOG(ERROR) <<
"MediaInfo missing required fields.";
1143 return xml::scoped_xml_ptr<xmlNode>();
1146 const uint64_t bandwidth = media_info_.has_bandwidth()
1147 ? media_info_.bandwidth()
1148 : bandwidth_estimator_.Estimate();
1150 DCHECK(!(HasVODOnlyFields(media_info_) && HasLiveOnlyFields(media_info_)));
1152 RepresentationXmlNode representation;
1154 representation.SetId(id_);
1155 representation.SetIntegerAttribute(
"bandwidth", bandwidth);
1156 if (!codecs_.empty())
1157 representation.SetStringAttribute(
"codecs", codecs_);
1158 representation.SetStringAttribute(
"mimeType", mime_type_);
1160 const bool has_video_info = media_info_.has_video_info();
1161 const bool has_audio_info = media_info_.has_audio_info();
1163 if (has_video_info &&
1164 !representation.AddVideoInfo(
1165 media_info_.video_info(),
1166 !(output_suppression_flags_ & kSuppressWidth),
1167 !(output_suppression_flags_ & kSuppressHeight),
1168 !(output_suppression_flags_ & kSuppressFrameRate))) {
1169 LOG(ERROR) <<
"Failed to add video info to Representation XML.";
1170 return xml::scoped_xml_ptr<xmlNode>();
1173 if (has_audio_info &&
1174 !representation.AddAudioInfo(media_info_.audio_info())) {
1175 LOG(ERROR) <<
"Failed to add audio info to Representation XML.";
1176 return xml::scoped_xml_ptr<xmlNode>();
1179 if (!representation.AddContentProtectionElements(
1180 content_protection_elements_)) {
1181 return xml::scoped_xml_ptr<xmlNode>();
1184 if (HasVODOnlyFields(media_info_) &&
1185 !representation.AddVODOnlyInfo(media_info_)) {
1186 LOG(ERROR) <<
"Failed to add VOD segment info.";
1187 return xml::scoped_xml_ptr<xmlNode>();
1190 if (HasLiveOnlyFields(media_info_) &&
1191 !representation.AddLiveOnlyInfo(media_info_, segment_infos_,
1193 LOG(ERROR) <<
"Failed to add Live info.";
1194 return xml::scoped_xml_ptr<xmlNode>();
1199 output_suppression_flags_ = 0;
1200 return representation.PassScopedPtr();
1204 output_suppression_flags_ |= flag;
1207 bool Representation::HasRequiredMediaInfoFields() {
1208 if (HasVODOnlyFields(media_info_) && HasLiveOnlyFields(media_info_)) {
1209 LOG(ERROR) <<
"MediaInfo cannot have both VOD and Live fields.";
1213 if (!media_info_.has_container_type()) {
1214 LOG(ERROR) <<
"MediaInfo missing required field: container_type.";
1218 if (HasVODOnlyFields(media_info_) && !media_info_.has_bandwidth()) {
1219 LOG(ERROR) <<
"Missing 'bandwidth' field. MediaInfo requires bandwidth for "
1220 "static profile for generating a valid MPD.";
1224 VLOG_IF(3, HasLiveOnlyFields(media_info_) && !media_info_.has_bandwidth())
1225 <<
"MediaInfo missing field 'bandwidth'. Using estimated from "
1231 bool Representation::IsContiguous(uint64_t start_time,
1233 uint64_t size)
const {
1234 if (segment_infos_.empty())
1238 const SegmentInfo& previous = segment_infos_.back();
1239 const uint64_t previous_segment_end_time =
1240 previous.start_time + previous.duration * (previous.repeat + 1);
1241 if (previous_segment_end_time == start_time &&
1242 segment_infos_.back().duration == duration) {
1247 const uint64_t previous_segment_start_time =
1248 previous.start_time + previous.duration * previous.repeat;
1249 if (previous_segment_start_time >= start_time) {
1250 LOG(ERROR) <<
"Segments should not be out of order segment. Adding segment "
1251 "with start_time == "
1252 << start_time <<
" but the previous segment starts at "
1253 << previous.start_time <<
".";
1258 const uint64_t kRoundingErrorGrace = 5;
1259 if (previous_segment_end_time + kRoundingErrorGrace < start_time) {
1260 LOG(WARNING) <<
"Found a gap of size "
1261 << (start_time - previous_segment_end_time)
1262 <<
" > kRoundingErrorGrace (" << kRoundingErrorGrace
1263 <<
"). The new segment starts at " << start_time
1264 <<
" but the previous segment ends at "
1265 << previous_segment_end_time <<
".";
1270 if (start_time < previous_segment_end_time - kRoundingErrorGrace) {
1272 <<
"Segments should not be overlapping. The new segment starts at "
1273 << start_time <<
" but the previous segment ends at "
1274 << previous_segment_end_time <<
".";
1282 void Representation::SlideWindow() {
1283 DCHECK(!segment_infos_.empty());
1284 if (mpd_options_.time_shift_buffer_depth <= 0.0)
1287 const uint32_t time_scale = GetTimeScale(media_info_);
1288 DCHECK_GT(time_scale, 0u);
1290 uint64_t time_shift_buffer_depth =
1291 static_cast<uint64_t
>(mpd_options_.time_shift_buffer_depth * time_scale);
1295 const uint64_t current_play_time = LatestSegmentStartTime(segment_infos_);
1296 if (current_play_time <= time_shift_buffer_depth)
1299 const uint64_t timeshift_limit = current_play_time - time_shift_buffer_depth;
1303 std::list<SegmentInfo>::iterator first = segment_infos_.begin();
1304 std::list<SegmentInfo>::iterator last = first;
1305 size_t num_segments_removed = 0;
1306 for (; last != segment_infos_.end(); ++last) {
1307 const uint64_t last_segment_end_time = LastSegmentEndTime(*last);
1308 if (timeshift_limit < last_segment_end_time)
1310 num_segments_removed += last->repeat + 1;
1312 segment_infos_.erase(first, last);
1313 start_number_ += num_segments_removed;
1316 SegmentInfo* first_segment_info = &segment_infos_.front();
1317 DCHECK_LE(timeshift_limit, LastSegmentEndTime(*first_segment_info));
1320 const int repeat_index =
1321 SearchTimedOutRepeatIndex(timeshift_limit, *first_segment_info);
1322 CHECK_GE(repeat_index, 0);
1323 if (repeat_index == 0)
1326 first_segment_info->start_time = first_segment_info->start_time +
1327 first_segment_info->duration * repeat_index;
1329 first_segment_info->repeat = first_segment_info->repeat - repeat_index;
1330 start_number_ += repeat_index;
1333 std::string Representation::GetVideoMimeType()
const {
1334 return GetMimeType(
"video", media_info_.container_type());
1337 std::string Representation::GetAudioMimeType()
const {
1338 return GetMimeType(
"audio", media_info_.container_type());
1341 std::string Representation::GetTextMimeType()
const {
1342 CHECK(media_info_.has_text_info());
1343 if (media_info_.text_info().format() ==
"ttml") {
1344 switch (media_info_.container_type()) {
1345 case MediaInfo::CONTAINER_TEXT:
1346 return "application/ttml+xml";
1347 case MediaInfo::CONTAINER_MP4:
1348 return "application/mp4";
1350 LOG(ERROR) <<
"Failed to determine MIME type for TTML container: "
1351 << media_info_.container_type();
1355 if (media_info_.text_info().format() ==
"vtt") {
1356 if (media_info_.container_type() == MediaInfo::CONTAINER_TEXT) {
1359 LOG(ERROR) <<
"Failed to determine MIME type for VTT container: "
1360 << media_info_.container_type();
1364 LOG(ERROR) <<
"Cannot determine MIME type for format: "
1365 << media_info_.text_info().format()
1366 <<
" container: " << media_info_.container_type();
1370 bool Representation::GetEarliestTimestamp(
double* timestamp_seconds) {
1371 DCHECK(timestamp_seconds);
1373 if (segment_infos_.empty())
1376 *timestamp_seconds =
static_cast<double>(segment_infos_.begin()->start_time) /
1377 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)