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);
401 mpd_options_(mpd_options),
402 clock_(new base::DefaultClock()) {}
404 MpdBuilder::~MpdBuilder() {}
407 base_urls_.push_back(base_url);
411 std::unique_ptr<AdaptationSet> adaptation_set(
412 new AdaptationSet(adaptation_set_counter_.GetNext(), lang, mpd_options_,
413 type_, &representation_counter_));
414 DCHECK(adaptation_set);
416 if (!lang.empty() && lang == mpd_options_.default_language) {
417 adaptation_set->AddRole(AdaptationSet::kRoleMain);
420 adaptation_sets_.push_back(std::move(adaptation_set));
421 return adaptation_sets_.back().get();
426 return WriteMpdToOutput(output_file);
431 return WriteMpdToOutput(output);
433 template <
typename OutputType>
434 bool MpdBuilder::WriteMpdToOutput(OutputType* output) {
435 static LibXmlInitializer lib_xml_initializer;
437 xml::scoped_xml_ptr<xmlDoc> doc(GenerateMpd());
441 static const int kNiceFormat = 1;
442 int doc_str_size = 0;
443 xmlChar* doc_str = NULL;
444 xmlDocDumpFormatMemoryEnc(doc.get(), &doc_str, &doc_str_size,
"UTF-8",
447 bool result = WriteXmlCharArrayToOutput(doc_str, doc_str_size, output);
455 xmlDocPtr MpdBuilder::GenerateMpd() {
457 static const char kXmlVersion[] =
"1.0";
458 xml::scoped_xml_ptr<xmlDoc> doc(xmlNewDoc(BAD_CAST kXmlVersion));
462 XmlNode period(
"Period");
468 for (
const std::unique_ptr<AdaptationSet>& adaptation_set :
470 xml::scoped_xml_ptr<xmlNode> child(adaptation_set->GetXml());
471 if (!child.get() || !period.AddChild(std::move(child)))
476 std::list<std::string>::const_iterator base_urls_it = base_urls_.begin();
477 for (; base_urls_it != base_urls_.end(); ++base_urls_it) {
478 XmlNode base_url(
"BaseURL");
479 base_url.SetContent(*base_urls_it);
481 if (!mpd.AddChild(base_url.PassScopedPtr()))
485 if (type_ == kDynamic) {
487 period.SetStringAttribute(
"start",
"PT0S");
490 if (!mpd.AddChild(period.PassScopedPtr()))
493 AddMpdNameSpaceInfo(&mpd);
494 AddCommonMpdInfo(&mpd);
497 AddStaticMpdInfo(&mpd);
500 AddDynamicMpdInfo(&mpd);
503 NOTREACHED() <<
"Unknown MPD type: " << type_;
508 const std::string version = GetPackagerVersion();
509 if (!version.empty()) {
510 std::string version_string =
511 base::StringPrintf(
"Generated with %s version %s",
512 GetPackagerProjectUrl().c_str(), version.c_str());
513 xml::scoped_xml_ptr<xmlNode> comment(
514 xmlNewDocComment(doc.get(), BAD_CAST version_string.c_str()));
515 xmlDocSetRootElement(doc.get(), comment.get());
516 xmlAddSibling(comment.release(), mpd.Release());
518 xmlDocSetRootElement(doc.get(), mpd.Release());
520 return doc.release();
523 void MpdBuilder::AddCommonMpdInfo(XmlNode* mpd_node) {
524 if (Positive(mpd_options_.min_buffer_time)) {
525 mpd_node->SetStringAttribute(
526 "minBufferTime", SecondsToXmlDuration(mpd_options_.min_buffer_time));
528 LOG(ERROR) <<
"minBufferTime value not specified.";
533 void MpdBuilder::AddStaticMpdInfo(XmlNode* mpd_node) {
535 DCHECK_EQ(MpdBuilder::kStatic, type_);
537 static const char kStaticMpdType[] =
"static";
538 static const char kStaticMpdProfile[] =
539 "urn:mpeg:dash:profile:isoff-on-demand:2011";
540 mpd_node->SetStringAttribute(
"type", kStaticMpdType);
541 mpd_node->SetStringAttribute(
"profiles", kStaticMpdProfile);
542 mpd_node->SetStringAttribute(
543 "mediaPresentationDuration",
544 SecondsToXmlDuration(GetStaticMpdDuration(mpd_node)));
547 void MpdBuilder::AddDynamicMpdInfo(XmlNode* mpd_node) {
549 DCHECK_EQ(MpdBuilder::kDynamic, type_);
551 static const char kDynamicMpdType[] =
"dynamic";
552 static const char kDynamicMpdProfile[] =
553 "urn:mpeg:dash:profile:isoff-live:2011";
554 mpd_node->SetStringAttribute(
"type", kDynamicMpdType);
555 mpd_node->SetStringAttribute(
"profiles", kDynamicMpdProfile);
558 mpd_node->SetStringAttribute(
"publishTime",
559 XmlDateTimeNowWithOffset(0, clock_.get()));
563 if (availability_start_time_.empty()) {
564 double earliest_presentation_time;
565 if (GetEarliestTimestamp(&earliest_presentation_time)) {
566 availability_start_time_ =
567 XmlDateTimeNowWithOffset(mpd_options_.availability_time_offset -
568 std::ceil(earliest_presentation_time),
571 LOG(ERROR) <<
"Could not determine the earliest segment presentation "
572 "time for availabilityStartTime calculation.";
576 if (!availability_start_time_.empty())
577 mpd_node->SetStringAttribute(
"availabilityStartTime",
578 availability_start_time_);
580 if (Positive(mpd_options_.minimum_update_period)) {
581 mpd_node->SetStringAttribute(
582 "minimumUpdatePeriod",
583 SecondsToXmlDuration(mpd_options_.minimum_update_period));
585 LOG(WARNING) <<
"The profile is dynamic but no minimumUpdatePeriod "
589 SetIfPositive(
"timeShiftBufferDepth", mpd_options_.time_shift_buffer_depth,
591 SetIfPositive(
"suggestedPresentationDelay",
592 mpd_options_.suggested_presentation_delay, mpd_node);
595 float MpdBuilder::GetStaticMpdDuration(XmlNode* mpd_node) {
597 DCHECK_EQ(MpdBuilder::kStatic, type_);
599 xmlNodePtr period_node = FindPeriodNode(mpd_node);
600 DCHECK(period_node) <<
"Period element must be a child of mpd_node.";
601 DCHECK(IsPeriodNode(period_node));
606 float max_duration = 0.0f;
607 for (xmlNodePtr adaptation_set = xmlFirstElementChild(period_node);
608 adaptation_set; adaptation_set = xmlNextElementSibling(adaptation_set)) {
609 for (xmlNodePtr representation = xmlFirstElementChild(adaptation_set);
611 representation = xmlNextElementSibling(representation)) {
612 float duration = 0.0f;
613 if (GetDurationAttribute(representation, &duration)) {
614 max_duration = max_duration > duration ? max_duration : duration;
618 xmlUnsetProp(representation, BAD_CAST
"duration");
626 bool MpdBuilder::GetEarliestTimestamp(
double* timestamp_seconds) {
627 DCHECK(timestamp_seconds);
629 double earliest_timestamp(-1);
630 for (
const std::unique_ptr<AdaptationSet>& adaptation_set :
633 if (adaptation_set->GetEarliestTimestamp(×tamp) &&
634 ((earliest_timestamp < 0) || (timestamp < earliest_timestamp))) {
635 earliest_timestamp = timestamp;
638 if (earliest_timestamp < 0)
641 *timestamp_seconds = earliest_timestamp;
646 MediaInfo* media_info) {
648 const std::string kFileProtocol(
"file://");
649 std::string mpd_file_path = (mpd_path.find(kFileProtocol) == 0)
650 ? mpd_path.substr(kFileProtocol.size())
653 if (!mpd_file_path.empty()) {
654 std::string mpd_dir(FilePath::FromUTF8Unsafe(mpd_file_path)
655 .DirName().AsEndingWithSeparator().AsUTF8Unsafe());
656 if (!mpd_dir.empty()) {
657 if (media_info->has_media_file_name()) {
658 media_info->set_media_file_name(
659 MakePathRelative(media_info->media_file_name(), mpd_dir));
661 if (media_info->has_init_segment_name()) {
662 media_info->set_init_segment_name(
663 MakePathRelative(media_info->init_segment_name(), mpd_dir));
665 if (media_info->has_segment_template()) {
666 media_info->set_segment_template(
667 MakePathRelative(media_info->segment_template(), mpd_dir));
674 const std::string& lang,
676 MpdBuilder::MpdType mpd_type,
677 base::AtomicSequenceNumber* counter)
678 : representation_counter_(counter),
679 id_(adaptation_set_id),
681 mpd_options_(mpd_options),
683 segments_aligned_(kSegmentAlignmentUnknown),
684 force_set_segment_alignment_(false) {
688 AdaptationSet::~AdaptationSet() {}
691 const uint32_t representation_id = representation_counter_->GetNext();
694 std::unique_ptr<RepresentationStateChangeListener> listener(
695 new RepresentationStateChangeListenerImpl(representation_id,
this));
696 std::unique_ptr<Representation> representation(
new Representation(
697 media_info, mpd_options_, representation_id, std::move(listener)));
699 if (!representation->Init())
704 if (media_info.has_video_info()) {
705 const MediaInfo::VideoInfo& video_info = media_info.video_info();
706 DCHECK(video_info.has_width());
707 DCHECK(video_info.has_height());
708 video_widths_.insert(video_info.width());
709 video_heights_.insert(video_info.height());
711 if (video_info.has_time_scale() && video_info.has_frame_duration())
712 RecordFrameRate(video_info.frame_duration(), video_info.time_scale());
714 AddPictureAspectRatio(video_info, &picture_aspect_ratio_);
717 if (media_info.has_video_info()) {
718 content_type_ =
"video";
719 }
else if (media_info.has_audio_info()) {
720 content_type_ =
"audio";
721 }
else if (media_info.has_text_info()) {
722 content_type_ =
"text";
724 if (media_info.text_info().has_type() &&
725 (media_info.text_info().type() != MediaInfo::TextInfo::UNKNOWN)) {
726 roles_.insert(MediaInfoTextTypeToRole(media_info.text_info().type()));
730 representations_.push_back(std::move(representation));
731 return representations_.back().get();
736 content_protection_elements_.push_back(content_protection_element);
737 RemoveDuplicateAttributes(&content_protection_elements_.back());
741 const std::string& pssh) {
742 UpdateContentProtectionPsshHelper(drm_uuid, pssh,
743 &content_protection_elements_);
757 AdaptationSetXmlNode adaptation_set;
759 bool suppress_representation_width =
false;
760 bool suppress_representation_height =
false;
761 bool suppress_representation_frame_rate =
false;
763 adaptation_set.SetId(id_);
764 adaptation_set.SetStringAttribute(
"contentType", content_type_);
765 if (!lang_.empty() && lang_ !=
"und") {
770 if (video_widths_.size() == 1) {
771 suppress_representation_width =
true;
772 adaptation_set.SetIntegerAttribute(
"width", *video_widths_.begin());
773 }
else if (video_widths_.size() > 1) {
774 adaptation_set.SetIntegerAttribute(
"maxWidth", *video_widths_.rbegin());
776 if (video_heights_.size() == 1) {
777 suppress_representation_height =
true;
778 adaptation_set.SetIntegerAttribute(
"height", *video_heights_.begin());
779 }
else if (video_heights_.size() > 1) {
780 adaptation_set.SetIntegerAttribute(
"maxHeight", *video_heights_.rbegin());
783 if (video_frame_rates_.size() == 1) {
784 suppress_representation_frame_rate =
true;
785 adaptation_set.SetStringAttribute(
"frameRate",
786 video_frame_rates_.begin()->second);
787 }
else if (video_frame_rates_.size() > 1) {
788 adaptation_set.SetStringAttribute(
"maxFrameRate",
789 video_frame_rates_.rbegin()->second);
794 if (mpd_type_ == MpdBuilder::kStatic) {
795 CheckVodSegmentAlignment();
798 if (segments_aligned_ == kSegmentAlignmentTrue) {
799 adaptation_set.SetStringAttribute(mpd_type_ == MpdBuilder::kStatic
800 ?
"subsegmentAlignment"
801 :
"segmentAlignment",
805 if (picture_aspect_ratio_.size() == 1)
806 adaptation_set.SetStringAttribute(
"par", *picture_aspect_ratio_.begin());
808 if (!adaptation_set.AddContentProtectionElements(
809 content_protection_elements_)) {
810 return xml::scoped_xml_ptr<xmlNode>();
813 std::string switching_ids;
814 for (uint32_t
id : adaptation_set_switching_ids_) {
815 if (!switching_ids.empty())
816 switching_ids +=
',';
817 switching_ids += base::UintToString(
id);
819 if (!switching_ids.empty()) {
820 adaptation_set.AddSupplementalProperty(
821 "urn:mpeg:dash:adaptation-set-switching:2016", switching_ids);
824 for (AdaptationSet::Role role : roles_)
825 adaptation_set.AddRoleElement(
"urn:mpeg:dash:role:2011", RoleToText(role));
827 for (
const std::unique_ptr<Representation>& representation :
829 if (suppress_representation_width)
830 representation->SuppressOnce(Representation::kSuppressWidth);
831 if (suppress_representation_height)
832 representation->SuppressOnce(Representation::kSuppressHeight);
833 if (suppress_representation_frame_rate)
834 representation->SuppressOnce(Representation::kSuppressFrameRate);
835 xml::scoped_xml_ptr<xmlNode> child(representation->GetXml());
836 if (!child || !adaptation_set.AddChild(std::move(child)))
837 return xml::scoped_xml_ptr<xmlNode>();
840 return adaptation_set.PassScopedPtr();
845 segment_alignment ? kSegmentAlignmentTrue : kSegmentAlignmentFalse;
846 force_set_segment_alignment_ =
true;
850 adaptation_set_switching_ids_.push_back(adaptation_set_id);
863 if (mpd_type_ == MpdBuilder::kDynamic) {
864 CheckLiveSegmentAlignment(representation_id, start_time, duration);
866 representation_segment_start_times_[representation_id].push_back(
872 uint32_t representation_id,
873 uint32_t frame_duration,
874 uint32_t timescale) {
875 RecordFrameRate(frame_duration, timescale);
878 bool AdaptationSet::GetEarliestTimestamp(
double* timestamp_seconds) {
879 DCHECK(timestamp_seconds);
881 double earliest_timestamp(-1);
882 for (
const std::unique_ptr<Representation>& representation :
885 if (representation->GetEarliestTimestamp(×tamp) &&
886 ((earliest_timestamp < 0) || (timestamp < earliest_timestamp))) {
887 earliest_timestamp = timestamp;
890 if (earliest_timestamp < 0)
893 *timestamp_seconds = earliest_timestamp;
921 void AdaptationSet::CheckLiveSegmentAlignment(uint32_t representation_id,
924 if (segments_aligned_ == kSegmentAlignmentFalse ||
925 force_set_segment_alignment_) {
929 std::list<uint64_t>& representation_start_times =
930 representation_segment_start_times_[representation_id];
931 representation_start_times.push_back(start_time);
934 if (representation_segment_start_times_.size() != representations_.size())
937 DCHECK(!representation_start_times.empty());
938 const uint64_t expected_start_time = representation_start_times.front();
939 for (RepresentationTimeline::const_iterator it =
940 representation_segment_start_times_.begin();
941 it != representation_segment_start_times_.end(); ++it) {
945 if (it->second.empty())
948 if (expected_start_time != it->second.front()) {
951 segments_aligned_ = kSegmentAlignmentFalse;
952 representation_segment_start_times_.clear();
956 segments_aligned_ = kSegmentAlignmentTrue;
958 for (RepresentationTimeline::iterator it =
959 representation_segment_start_times_.begin();
960 it != representation_segment_start_times_.end(); ++it) {
961 it->second.pop_front();
967 void AdaptationSet::CheckVodSegmentAlignment() {
968 if (segments_aligned_ == kSegmentAlignmentFalse ||
969 force_set_segment_alignment_) {
972 if (representation_segment_start_times_.empty())
974 if (representation_segment_start_times_.size() == 1) {
975 segments_aligned_ = kSegmentAlignmentTrue;
982 const std::list<uint64_t>& expected_time_line =
983 representation_segment_start_times_.begin()->second;
985 bool all_segment_time_line_same_length =
true;
987 RepresentationTimeline::const_iterator it =
988 representation_segment_start_times_.begin();
989 for (++it; it != representation_segment_start_times_.end(); ++it) {
990 const std::list<uint64_t>& other_time_line = it->second;
991 if (expected_time_line.size() != other_time_line.size()) {
992 all_segment_time_line_same_length =
false;
995 const std::list<uint64_t>* longer_list = &other_time_line;
996 const std::list<uint64_t>* shorter_list = &expected_time_line;
997 if (expected_time_line.size() > other_time_line.size()) {
998 shorter_list = &other_time_line;
999 longer_list = &expected_time_line;
1002 if (!std::equal(shorter_list->begin(), shorter_list->end(),
1003 longer_list->begin())) {
1005 segments_aligned_ = kSegmentAlignmentFalse;
1006 representation_segment_start_times_.clear();
1017 if (!all_segment_time_line_same_length) {
1018 segments_aligned_ = kSegmentAlignmentUnknown;
1022 segments_aligned_ = kSegmentAlignmentTrue;
1027 void AdaptationSet::RecordFrameRate(uint32_t frame_duration,
1028 uint32_t timescale) {
1029 if (frame_duration == 0) {
1030 LOG(ERROR) <<
"Frame duration is 0 and cannot be set.";
1033 video_frame_rates_[
static_cast<double>(timescale) / frame_duration] =
1034 base::IntToString(timescale) +
"/" + base::IntToString(frame_duration);
1038 const MediaInfo& media_info,
1041 std::unique_ptr<RepresentationStateChangeListener> state_change_listener)
1042 : media_info_(media_info),
1045 mpd_options_(mpd_options),
1047 state_change_listener_(std::move(state_change_listener)),
1048 output_suppression_flags_(0) {}
1050 Representation::~Representation() {}
1053 if (!AtLeastOneTrue(media_info_.has_video_info(),
1054 media_info_.has_audio_info(),
1055 media_info_.has_text_info())) {
1059 LOG(ERROR) <<
"Representation needs one of video, audio, or text.";
1063 if (MoreThanOneTrue(media_info_.has_video_info(),
1064 media_info_.has_audio_info(),
1065 media_info_.has_text_info())) {
1066 LOG(ERROR) <<
"Only one of VideoInfo, AudioInfo, or TextInfo can be set.";
1070 if (media_info_.container_type() == MediaInfo::CONTAINER_UNKNOWN) {
1071 LOG(ERROR) <<
"'container_type' in MediaInfo cannot be CONTAINER_UNKNOWN.";
1075 if (media_info_.has_video_info()) {
1076 mime_type_ = GetVideoMimeType();
1077 if (!HasRequiredVideoFields(media_info_.video_info())) {
1078 LOG(ERROR) <<
"Missing required fields to create a video Representation.";
1081 }
else if (media_info_.has_audio_info()) {
1082 mime_type_ = GetAudioMimeType();
1083 }
else if (media_info_.has_text_info()) {
1084 mime_type_ = GetTextMimeType();
1087 if (mime_type_.empty())
1090 codecs_ = GetCodecs(media_info_);
1096 content_protection_elements_.push_back(content_protection_element);
1097 RemoveDuplicateAttributes(&content_protection_elements_.back());
1101 const std::string& pssh) {
1102 UpdateContentProtectionPsshHelper(drm_uuid, pssh,
1103 &content_protection_elements_);
1109 if (start_time == 0 && duration == 0) {
1110 LOG(WARNING) <<
"Got segment with start_time and duration == 0. Ignoring.";
1114 if (state_change_listener_)
1115 state_change_listener_->OnNewSegmentForRepresentation(start_time, duration);
1116 if (IsContiguous(start_time, duration, size)) {
1117 ++segment_infos_.back().repeat;
1120 segment_infos_.push_back(s);
1123 bandwidth_estimator_.AddBlock(
1124 size, static_cast<double>(duration) / media_info_.reference_time_scale());
1127 DCHECK_GE(segment_infos_.size(), 1u);
1131 if (media_info_.has_video_info()) {
1132 media_info_.mutable_video_info()->set_frame_duration(sample_duration);
1133 if (state_change_listener_) {
1134 state_change_listener_->OnSetFrameRateForRepresentation(
1135 sample_duration, media_info_.video_info().time_scale());
1147 if (!HasRequiredMediaInfoFields()) {
1148 LOG(ERROR) <<
"MediaInfo missing required fields.";
1149 return xml::scoped_xml_ptr<xmlNode>();
1152 const uint64_t bandwidth = media_info_.has_bandwidth()
1153 ? media_info_.bandwidth()
1154 : bandwidth_estimator_.Estimate();
1156 DCHECK(!(HasVODOnlyFields(media_info_) && HasLiveOnlyFields(media_info_)));
1158 RepresentationXmlNode representation;
1160 representation.SetId(id_);
1161 representation.SetIntegerAttribute(
"bandwidth", bandwidth);
1162 if (!codecs_.empty())
1163 representation.SetStringAttribute(
"codecs", codecs_);
1164 representation.SetStringAttribute(
"mimeType", mime_type_);
1166 const bool has_video_info = media_info_.has_video_info();
1167 const bool has_audio_info = media_info_.has_audio_info();
1169 if (has_video_info &&
1170 !representation.AddVideoInfo(
1171 media_info_.video_info(),
1172 !(output_suppression_flags_ & kSuppressWidth),
1173 !(output_suppression_flags_ & kSuppressHeight),
1174 !(output_suppression_flags_ & kSuppressFrameRate))) {
1175 LOG(ERROR) <<
"Failed to add video info to Representation XML.";
1176 return xml::scoped_xml_ptr<xmlNode>();
1179 if (has_audio_info &&
1180 !representation.AddAudioInfo(media_info_.audio_info())) {
1181 LOG(ERROR) <<
"Failed to add audio info to Representation XML.";
1182 return xml::scoped_xml_ptr<xmlNode>();
1185 if (!representation.AddContentProtectionElements(
1186 content_protection_elements_)) {
1187 return xml::scoped_xml_ptr<xmlNode>();
1190 if (HasVODOnlyFields(media_info_) &&
1191 !representation.AddVODOnlyInfo(media_info_)) {
1192 LOG(ERROR) <<
"Failed to add VOD segment info.";
1193 return xml::scoped_xml_ptr<xmlNode>();
1196 if (HasLiveOnlyFields(media_info_) &&
1197 !representation.AddLiveOnlyInfo(media_info_, segment_infos_,
1199 LOG(ERROR) <<
"Failed to add Live info.";
1200 return xml::scoped_xml_ptr<xmlNode>();
1205 output_suppression_flags_ = 0;
1206 return representation.PassScopedPtr();
1210 output_suppression_flags_ |= flag;
1213 bool Representation::HasRequiredMediaInfoFields() {
1214 if (HasVODOnlyFields(media_info_) && HasLiveOnlyFields(media_info_)) {
1215 LOG(ERROR) <<
"MediaInfo cannot have both VOD and Live fields.";
1219 if (!media_info_.has_container_type()) {
1220 LOG(ERROR) <<
"MediaInfo missing required field: container_type.";
1224 if (HasVODOnlyFields(media_info_) && !media_info_.has_bandwidth()) {
1225 LOG(ERROR) <<
"Missing 'bandwidth' field. MediaInfo requires bandwidth for "
1226 "static profile for generating a valid MPD.";
1230 VLOG_IF(3, HasLiveOnlyFields(media_info_) && !media_info_.has_bandwidth())
1231 <<
"MediaInfo missing field 'bandwidth'. Using estimated from "
1237 bool Representation::IsContiguous(uint64_t start_time,
1239 uint64_t size)
const {
1240 if (segment_infos_.empty())
1244 const SegmentInfo& previous = segment_infos_.back();
1245 const uint64_t previous_segment_end_time =
1246 previous.start_time + previous.duration * (previous.repeat + 1);
1247 if (previous_segment_end_time == start_time &&
1248 segment_infos_.back().duration == duration) {
1253 const uint64_t previous_segment_start_time =
1254 previous.start_time + previous.duration * previous.repeat;
1255 if (previous_segment_start_time >= start_time) {
1256 LOG(ERROR) <<
"Segments should not be out of order segment. Adding segment "
1257 "with start_time == "
1258 << start_time <<
" but the previous segment starts at "
1259 << previous.start_time <<
".";
1264 const uint64_t kRoundingErrorGrace = 5;
1265 if (previous_segment_end_time + kRoundingErrorGrace < start_time) {
1266 LOG(WARNING) <<
"Found a gap of size "
1267 << (start_time - previous_segment_end_time)
1268 <<
" > kRoundingErrorGrace (" << kRoundingErrorGrace
1269 <<
"). The new segment starts at " << start_time
1270 <<
" but the previous segment ends at "
1271 << previous_segment_end_time <<
".";
1276 if (start_time < previous_segment_end_time - kRoundingErrorGrace) {
1278 <<
"Segments should not be overlapping. The new segment starts at "
1279 << start_time <<
" but the previous segment ends at "
1280 << previous_segment_end_time <<
".";
1288 void Representation::SlideWindow() {
1289 DCHECK(!segment_infos_.empty());
1290 if (mpd_options_.time_shift_buffer_depth <= 0.0)
1293 const uint32_t time_scale = GetTimeScale(media_info_);
1294 DCHECK_GT(time_scale, 0u);
1296 uint64_t time_shift_buffer_depth =
1297 static_cast<uint64_t
>(mpd_options_.time_shift_buffer_depth * time_scale);
1301 const uint64_t current_play_time = LatestSegmentStartTime(segment_infos_);
1302 if (current_play_time <= time_shift_buffer_depth)
1305 const uint64_t timeshift_limit = current_play_time - time_shift_buffer_depth;
1309 std::list<SegmentInfo>::iterator first = segment_infos_.begin();
1310 std::list<SegmentInfo>::iterator last = first;
1311 size_t num_segments_removed = 0;
1312 for (; last != segment_infos_.end(); ++last) {
1313 const uint64_t last_segment_end_time = LastSegmentEndTime(*last);
1314 if (timeshift_limit < last_segment_end_time)
1316 num_segments_removed += last->repeat + 1;
1318 segment_infos_.erase(first, last);
1319 start_number_ += num_segments_removed;
1322 SegmentInfo* first_segment_info = &segment_infos_.front();
1323 DCHECK_LE(timeshift_limit, LastSegmentEndTime(*first_segment_info));
1326 const int repeat_index =
1327 SearchTimedOutRepeatIndex(timeshift_limit, *first_segment_info);
1328 CHECK_GE(repeat_index, 0);
1329 if (repeat_index == 0)
1332 first_segment_info->start_time = first_segment_info->start_time +
1333 first_segment_info->duration * repeat_index;
1335 first_segment_info->repeat = first_segment_info->repeat - repeat_index;
1336 start_number_ += repeat_index;
1339 std::string Representation::GetVideoMimeType()
const {
1340 return GetMimeType(
"video", media_info_.container_type());
1343 std::string Representation::GetAudioMimeType()
const {
1344 return GetMimeType(
"audio", media_info_.container_type());
1347 std::string Representation::GetTextMimeType()
const {
1348 CHECK(media_info_.has_text_info());
1349 if (media_info_.text_info().format() ==
"ttml") {
1350 switch (media_info_.container_type()) {
1351 case MediaInfo::CONTAINER_TEXT:
1352 return "application/ttml+xml";
1353 case MediaInfo::CONTAINER_MP4:
1354 return "application/mp4";
1356 LOG(ERROR) <<
"Failed to determine MIME type for TTML container: "
1357 << media_info_.container_type();
1361 if (media_info_.text_info().format() ==
"vtt") {
1362 if (media_info_.container_type() == MediaInfo::CONTAINER_TEXT) {
1365 LOG(ERROR) <<
"Failed to determine MIME type for VTT container: "
1366 << media_info_.container_type();
1370 LOG(ERROR) <<
"Cannot determine MIME type for format: "
1371 << media_info_.text_info().format()
1372 <<
" container: " << media_info_.container_type();
1376 bool Representation::GetEarliestTimestamp(
double* timestamp_seconds) {
1377 DCHECK(timestamp_seconds);
1379 if (segment_infos_.empty())
1382 *timestamp_seconds =
static_cast<double>(segment_infos_.begin()->start_time) /
1383 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)
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)
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)
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)