DASH Media Packaging SDK
 All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator
mpd_builder.cc
1 // Copyright 2014 Google Inc. All rights reserved.
2 //
3 // Use of this source code is governed by a BSD-style
4 // license that can be found in the LICENSE file or at
5 // https://developers.google.com/open-source/licenses/bsd
6 
7 #include "packager/mpd/base/mpd_builder.h"
8 
9 #include <libxml/tree.h>
10 #include <libxml/xmlstring.h>
11 
12 #include <cmath>
13 #include <iterator>
14 #include <list>
15 #include <memory>
16 #include <string>
17 
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"
33 
34 namespace shaka {
35 
36 using base::FilePath;
37 using xml::XmlNode;
38 using xml::RepresentationXmlNode;
39 using xml::AdaptationSetXmlNode;
40 
41 namespace {
42 
43 AdaptationSet::Role MediaInfoTextTypeToRole(
44  MediaInfo::TextInfo::TextType type) {
45  switch (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;
53  default:
54  NOTREACHED() << "Unknown MediaInfo TextType: " << type
55  << " assuming subtitle.";
56  return AdaptationSet::kRoleSubtitle;
57  }
58 }
59 
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:
66  // NOTE: DASH MPD spec uses lowercase but RFC3555 says uppercase.
67  return prefix + "/MP2T";
68  case MediaInfo::CONTAINER_WEBM:
69  return prefix + "/webm";
70  default:
71  break;
72  }
73 
74  // Unsupported container types should be rejected/handled by the caller.
75  LOG(ERROR) << "Unrecognized container type: " << container_type;
76  return std::string();
77 }
78 
79 void AddMpdNameSpaceInfo(XmlNode* mpd) {
80  DCHECK(mpd);
81 
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";
89 
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);
95 }
96 
97 bool IsPeriodNode(xmlNodePtr node) {
98  DCHECK(node);
99  int kEqual = 0;
100  return xmlStrcmp(node->name, reinterpret_cast<const xmlChar*>("Period")) ==
101  kEqual;
102 }
103 
104 // Find the first <Period> element. This does not recurse down the tree,
105 // only checks direct children. Returns the pointer to Period element on
106 // success, otherwise returns false.
107 // As noted here, we must traverse.
108 // http://www.xmlsoft.org/tutorial/ar01s04.html
109 xmlNodePtr FindPeriodNode(XmlNode* xml_node) {
110  for (xmlNodePtr node = xml_node->GetRawPtr()->xmlChildrenNode; node != NULL;
111  node = node->next) {
112  if (IsPeriodNode(node))
113  return node;
114  }
115 
116  return NULL;
117 }
118 
119 bool Positive(double d) {
120  return d > 0.0;
121 }
122 
123 // Return current time in XML DateTime format. The value is in UTC, so the
124 // string ends with a 'Z'.
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);
132 
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);
137 }
138 
139 void SetIfPositive(const char* attr_name, double value, XmlNode* mpd) {
140  if (Positive(value)) {
141  mpd->SetStringAttribute(attr_name, SecondsToXmlDuration(value));
142  }
143 }
144 
145 uint32_t GetTimeScale(const MediaInfo& media_info) {
146  if (media_info.has_reference_time_scale()) {
147  return media_info.reference_time_scale();
148  }
149 
150  if (media_info.has_video_info()) {
151  return media_info.video_info().time_scale();
152  }
153 
154  if (media_info.has_audio_info()) {
155  return media_info.audio_info().time_scale();
156  }
157 
158  LOG(WARNING) << "No timescale specified, using 1 as timescale.";
159  return 1;
160 }
161 
162 uint64_t LastSegmentStartTime(const SegmentInfo& segment_info) {
163  return segment_info.start_time + segment_info.duration * segment_info.repeat;
164 }
165 
166 // This is equal to |segment_info| end time
167 uint64_t LastSegmentEndTime(const SegmentInfo& segment_info) {
168  return segment_info.start_time +
169  segment_info.duration * (segment_info.repeat + 1);
170 }
171 
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);
176 }
177 
178 // Given |timeshift_limit|, finds out the number of segments that are no longer
179 // valid and should be removed from |segment_info|.
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)
184  return 0;
185 
186  return (timeshift_limit - segment_info.start_time) / segment_info.duration;
187 }
188 
189 // Overload this function to support different types of |output|.
190 // Note that this could be done by call MpdBuilder::ToString() and use the
191 // result to write to a file, it requires an extra copy.
192 bool WriteXmlCharArrayToOutput(xmlChar* doc,
193  int doc_size,
194  std::string* output) {
195  DCHECK(doc);
196  DCHECK(output);
197  output->assign(doc, doc + doc_size);
198  return true;
199 }
200 
201 bool WriteXmlCharArrayToOutput(xmlChar* doc,
202  int doc_size,
203  media::File* output) {
204  DCHECK(doc);
205  DCHECK(output);
206  if (output->Write(doc, doc_size) < doc_size)
207  return false;
208 
209  return output->Flush();
210 }
211 
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;
215 }
216 
217 // Check whether the video info has width and height.
218 // DASH IOP also requires several other fields for video representations, namely
219 // width, height, framerate, and sar.
220 bool HasRequiredVideoFields(const MediaInfo_VideoInfo& video_info) {
221  if (!video_info.has_height() || !video_info.has_width()) {
222  LOG(ERROR)
223  << "Width and height are required fields for generating a valid MPD.";
224  return false;
225  }
226  // These fields are not required for a valid MPD, but required for DASH IOP
227  // compliant MPD. MpdBuilder can keep generating MPDs without these fields.
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.";
240  return true;
241 }
242 
243 // Returns the picture aspect ratio string e.g. "16:9", "4:3".
244 // "Reducing the quotient to minimal form" does not work well in practice as
245 // there may be some rounding performed in the input, e.g. the resolution of
246 // 480p is 854:480 for 16:9 aspect ratio, can only be reduced to 427:240.
247 // The algorithm finds out the pair of integers, num and den, where num / den is
248 // the closest ratio to scaled_width / scaled_height, by looping den through
249 // common values.
250 std::string GetPictureAspectRatio(uint32_t width,
251  uint32_t height,
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;
257 
258  // Typical aspect ratios have par_y less than or equal to 19:
259  // https://en.wikipedia.org/wiki/List_of_common_resolutions
260  const uint32_t kLargestPossibleParY = 19;
261 
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) {
269  min_error = error;
270  par_num = num;
271  par_den = den;
272  if (error == 0) break;
273  }
274  }
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 << ".";
278 
279  return base::IntToString(par_num) + ":" + base::IntToString(par_den);
280 }
281 
282 // Adds an entry to picture_aspect_ratio if the size of picture_aspect_ratio is
283 // less than 2 and video_info has both pixel width and pixel height.
284 void AddPictureAspectRatio(
285  const MediaInfo::VideoInfo& video_info,
286  std::set<std::string>* picture_aspect_ratio) {
287  // If there are more than one entries in picture_aspect_ratio, the @par
288  // attribute cannot be set, so skip.
289  if (picture_aspect_ratio->size() > 1)
290  return;
291 
292  if (video_info.width() == 0 || video_info.height() == 0 ||
293  video_info.pixel_width() == 0 || video_info.pixel_height() == 0) {
294  // If there is even one Representation without a @sar attribute, @par cannot
295  // be calculated.
296  // Just populate the set with at least 2 bogus strings so that further call
297  // to this function will bail out immediately.
298  picture_aspect_ratio->insert("bogus");
299  picture_aspect_ratio->insert("entries");
300  return;
301  }
302 
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);
312 }
313 
314 std::string RoleToText(AdaptationSet::Role role) {
315  // Using switch so that the compiler can detect whether there is a case that's
316  // not being handled.
317  switch (role) {
318  case AdaptationSet::kRoleCaption:
319  return "caption";
320  case AdaptationSet::kRoleSubtitle:
321  return "subtitle";
322  case AdaptationSet::kRoleMain:
323  return "main";
324  case AdaptationSet::kRoleAlternate:
325  return "alternate";
326  case AdaptationSet::kRoleSupplementary:
327  return "supplementary";
328  case AdaptationSet::kRoleCommentary:
329  return "commentary";
330  case AdaptationSet::kRoleDub:
331  return "dub";
332  default:
333  break;
334  }
335 
336  NOTREACHED();
337  return "";
338 }
339 
340 // Spooky static initialization/cleanup of libxml.
341 class LibXmlInitializer {
342  public:
343  LibXmlInitializer() : initialized_(false) {
344  base::AutoLock lock(lock_);
345  if (!initialized_) {
346  xmlInitParser();
347  initialized_ = true;
348  }
349  }
350 
351  ~LibXmlInitializer() {
352  base::AutoLock lock(lock_);
353  if (initialized_) {
354  xmlCleanupParser();
355  initialized_ = false;
356  }
357  }
358 
359  private:
360  base::Lock lock_;
361  bool initialized_;
362 
363  DISALLOW_COPY_AND_ASSIGN(LibXmlInitializer);
364 };
365 
366 class RepresentationStateChangeListenerImpl
367  : public RepresentationStateChangeListener {
368  public:
369  // |adaptation_set| is not owned by this class.
370  RepresentationStateChangeListenerImpl(uint32_t representation_id,
371  AdaptationSet* adaptation_set)
372  : representation_id_(representation_id), adaptation_set_(adaptation_set) {
373  DCHECK(adaptation_set_);
374  }
375  ~RepresentationStateChangeListenerImpl() override {}
376 
377  // RepresentationStateChangeListener implementation.
378  void OnNewSegmentForRepresentation(uint64_t start_time,
379  uint64_t duration) override {
380  adaptation_set_->OnNewSegmentForRepresentation(representation_id_,
381  start_time, duration);
382  }
383 
384  void OnSetFrameRateForRepresentation(uint32_t frame_duration,
385  uint32_t timescale) override {
386  adaptation_set_->OnSetFrameRateForRepresentation(representation_id_,
387  frame_duration, timescale);
388  }
389 
390  private:
391  const uint32_t representation_id_;
392  AdaptationSet* const adaptation_set_;
393 
394  DISALLOW_COPY_AND_ASSIGN(RepresentationStateChangeListenerImpl);
395 };
396 
397 } // namespace
398 
399 MpdBuilder::MpdBuilder(MpdType type, const MpdOptions& mpd_options)
400  : type_(type),
401  mpd_options_(mpd_options),
402  clock_(new base::DefaultClock()) {}
403 
404 MpdBuilder::~MpdBuilder() {}
405 
406 void MpdBuilder::AddBaseUrl(const std::string& base_url) {
407  base_urls_.push_back(base_url);
408 }
409 
410 AdaptationSet* MpdBuilder::AddAdaptationSet(const std::string& lang) {
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);
415 
416  if (!lang.empty() && lang == mpd_options_.default_language) {
417  adaptation_set->AddRole(AdaptationSet::kRoleMain);
418  }
419 
420  adaptation_sets_.push_back(std::move(adaptation_set));
421  return adaptation_sets_.back().get();
422 }
423 
425  DCHECK(output_file);
426  return WriteMpdToOutput(output_file);
427 }
428 
429 bool MpdBuilder::ToString(std::string* output) {
430  DCHECK(output);
431  return WriteMpdToOutput(output);
432 }
433 template <typename OutputType>
434 bool MpdBuilder::WriteMpdToOutput(OutputType* output) {
435  static LibXmlInitializer lib_xml_initializer;
436 
437  xml::scoped_xml_ptr<xmlDoc> doc(GenerateMpd());
438  if (!doc.get())
439  return false;
440 
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",
445  kNiceFormat);
446 
447  bool result = WriteXmlCharArrayToOutput(doc_str, doc_str_size, output);
448  xmlFree(doc_str);
449 
450  // Cleanup, free the doc.
451  doc.reset();
452  return result;
453 }
454 
455 xmlDocPtr MpdBuilder::GenerateMpd() {
456  // Setup nodes.
457  static const char kXmlVersion[] = "1.0";
458  xml::scoped_xml_ptr<xmlDoc> doc(xmlNewDoc(BAD_CAST kXmlVersion));
459  XmlNode mpd("MPD");
460 
461  // Iterate thru AdaptationSets and add them to one big Period element.
462  XmlNode period("Period");
463 
464  // Always set id=0 for now. Since this class can only generate one Period
465  // at the moment, just use a constant.
466  // Required for 'dynamic' MPDs.
467  period.SetId(0);
468  for (const std::unique_ptr<AdaptationSet>& adaptation_set :
469  adaptation_sets_) {
470  xml::scoped_xml_ptr<xmlNode> child(adaptation_set->GetXml());
471  if (!child.get() || !period.AddChild(std::move(child)))
472  return NULL;
473  }
474 
475  // Add baseurls to MPD.
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);
480 
481  if (!mpd.AddChild(base_url.PassScopedPtr()))
482  return NULL;
483  }
484 
485  if (type_ == kDynamic) {
486  // This is the only Period and it is a regular period.
487  period.SetStringAttribute("start", "PT0S");
488  }
489 
490  if (!mpd.AddChild(period.PassScopedPtr()))
491  return NULL;
492 
493  AddMpdNameSpaceInfo(&mpd);
494  AddCommonMpdInfo(&mpd);
495  switch (type_) {
496  case kStatic:
497  AddStaticMpdInfo(&mpd);
498  break;
499  case kDynamic:
500  AddDynamicMpdInfo(&mpd);
501  break;
502  default:
503  NOTREACHED() << "Unknown MPD type: " << type_;
504  break;
505  }
506 
507  DCHECK(doc);
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());
517  } else {
518  xmlDocSetRootElement(doc.get(), mpd.Release());
519  }
520  return doc.release();
521 }
522 
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));
527  } else {
528  LOG(ERROR) << "minBufferTime value not specified.";
529  // TODO(tinskip): Propagate error.
530  }
531 }
532 
533 void MpdBuilder::AddStaticMpdInfo(XmlNode* mpd_node) {
534  DCHECK(mpd_node);
535  DCHECK_EQ(MpdBuilder::kStatic, type_);
536 
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)));
545 }
546 
547 void MpdBuilder::AddDynamicMpdInfo(XmlNode* mpd_node) {
548  DCHECK(mpd_node);
549  DCHECK_EQ(MpdBuilder::kDynamic, type_);
550 
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);
556 
557  // No offset from NOW.
558  mpd_node->SetStringAttribute("publishTime",
559  XmlDateTimeNowWithOffset(0, clock_.get()));
560 
561  // 'availabilityStartTime' is required for dynamic profile. Calculate if
562  // not already calculated.
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),
569  clock_.get());
570  } else {
571  LOG(ERROR) << "Could not determine the earliest segment presentation "
572  "time for availabilityStartTime calculation.";
573  // TODO(tinskip). Propagate an error.
574  }
575  }
576  if (!availability_start_time_.empty())
577  mpd_node->SetStringAttribute("availabilityStartTime",
578  availability_start_time_);
579 
580  if (Positive(mpd_options_.minimum_update_period)) {
581  mpd_node->SetStringAttribute(
582  "minimumUpdatePeriod",
583  SecondsToXmlDuration(mpd_options_.minimum_update_period));
584  } else {
585  LOG(WARNING) << "The profile is dynamic but no minimumUpdatePeriod "
586  "specified.";
587  }
588 
589  SetIfPositive("timeShiftBufferDepth", mpd_options_.time_shift_buffer_depth,
590  mpd_node);
591  SetIfPositive("suggestedPresentationDelay",
592  mpd_options_.suggested_presentation_delay, mpd_node);
593 }
594 
595 float MpdBuilder::GetStaticMpdDuration(XmlNode* mpd_node) {
596  DCHECK(mpd_node);
597  DCHECK_EQ(MpdBuilder::kStatic, type_);
598 
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));
602 
603  // Attribute mediaPresentationDuration must be present for 'static' MPD. So
604  // setting "PT0S" is required even if none of the representaions have duration
605  // attribute.
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);
610  representation;
611  representation = xmlNextElementSibling(representation)) {
612  float duration = 0.0f;
613  if (GetDurationAttribute(representation, &duration)) {
614  max_duration = max_duration > duration ? max_duration : duration;
615 
616  // 'duration' attribute is there only to help generate MPD, not
617  // necessary for MPD, remove the attribute.
618  xmlUnsetProp(representation, BAD_CAST "duration");
619  }
620  }
621  }
622 
623  return max_duration;
624 }
625 
626 bool MpdBuilder::GetEarliestTimestamp(double* timestamp_seconds) {
627  DCHECK(timestamp_seconds);
628 
629  double earliest_timestamp(-1);
630  for (const std::unique_ptr<AdaptationSet>& adaptation_set :
631  adaptation_sets_) {
632  double timestamp;
633  if (adaptation_set->GetEarliestTimestamp(&timestamp) &&
634  ((earliest_timestamp < 0) || (timestamp < earliest_timestamp))) {
635  earliest_timestamp = timestamp;
636  }
637  }
638  if (earliest_timestamp < 0)
639  return false;
640 
641  *timestamp_seconds = earliest_timestamp;
642  return true;
643 }
644 
645 void MpdBuilder::MakePathsRelativeToMpd(const std::string& mpd_path,
646  MediaInfo* media_info) {
647  DCHECK(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())
651  : mpd_path;
652 
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));
660  }
661  if (media_info->has_init_segment_name()) {
662  media_info->set_init_segment_name(
663  MakePathRelative(media_info->init_segment_name(), mpd_dir));
664  }
665  if (media_info->has_segment_template()) {
666  media_info->set_segment_template(
667  MakePathRelative(media_info->segment_template(), mpd_dir));
668  }
669  }
670  }
671 }
672 
673 AdaptationSet::AdaptationSet(uint32_t adaptation_set_id,
674  const std::string& lang,
675  const MpdOptions& mpd_options,
676  MpdBuilder::MpdType mpd_type,
677  base::AtomicSequenceNumber* counter)
678  : representation_counter_(counter),
679  id_(adaptation_set_id),
680  lang_(lang),
681  mpd_options_(mpd_options),
682  mpd_type_(mpd_type),
683  segments_aligned_(kSegmentAlignmentUnknown),
684  force_set_segment_alignment_(false) {
685  DCHECK(counter);
686 }
687 
688 AdaptationSet::~AdaptationSet() {}
689 
690 Representation* AdaptationSet::AddRepresentation(const MediaInfo& media_info) {
691  const uint32_t representation_id = representation_counter_->GetNext();
692  // Note that AdaptationSet outlive Representation, so this object
693  // will die before AdaptationSet.
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)));
698 
699  if (!representation->Init())
700  return NULL;
701 
702  // For videos, record the width, height, and the frame rate to calculate the
703  // max {width,height,framerate} required for DASH IOP.
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());
710 
711  if (video_info.has_time_scale() && video_info.has_frame_duration())
712  RecordFrameRate(video_info.frame_duration(), video_info.time_scale());
713 
714  AddPictureAspectRatio(video_info, &picture_aspect_ratio_);
715  }
716 
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";
723 
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()));
727  }
728  }
729 
730  representations_.push_back(std::move(representation));
731  return representations_.back().get();
732 }
733 
735  const ContentProtectionElement& content_protection_element) {
736  content_protection_elements_.push_back(content_protection_element);
737  RemoveDuplicateAttributes(&content_protection_elements_.back());
738 }
739 
740 void AdaptationSet::UpdateContentProtectionPssh(const std::string& drm_uuid,
741  const std::string& pssh) {
742  UpdateContentProtectionPsshHelper(drm_uuid, pssh,
743  &content_protection_elements_);
744 }
745 
746 void AdaptationSet::AddRole(Role role) {
747  roles_.insert(role);
748 }
749 
750 // Creates a copy of <AdaptationSet> xml element, iterate thru all the
751 // <Representation> (child) elements and add them to the copy.
752 // Set all the attributes first and then add the children elements so that flags
753 // can be passed to Representation to avoid setting redundant attributes. For
754 // example, if AdaptationSet@width is set, then Representation@width is
755 // redundant and should not be set.
756 xml::scoped_xml_ptr<xmlNode> AdaptationSet::GetXml() {
757  AdaptationSetXmlNode adaptation_set;
758 
759  bool suppress_representation_width = false;
760  bool suppress_representation_height = false;
761  bool suppress_representation_frame_rate = false;
762 
763  adaptation_set.SetId(id_);
764  adaptation_set.SetStringAttribute("contentType", content_type_);
765  if (!lang_.empty() && lang_ != "und") {
766  adaptation_set.SetStringAttribute("lang", LanguageToShortestForm(lang_));
767  }
768 
769  // Note that std::{set,map} are ordered, so the last element is the max value.
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());
775  }
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());
781  }
782 
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);
790  }
791 
792  // Note: must be checked before checking segments_aligned_ (below). So that
793  // segments_aligned_ is set before checking below.
794  if (mpd_type_ == MpdBuilder::kStatic) {
795  CheckVodSegmentAlignment();
796  }
797 
798  if (segments_aligned_ == kSegmentAlignmentTrue) {
799  adaptation_set.SetStringAttribute(mpd_type_ == MpdBuilder::kStatic
800  ? "subsegmentAlignment"
801  : "segmentAlignment",
802  "true");
803  }
804 
805  if (picture_aspect_ratio_.size() == 1)
806  adaptation_set.SetStringAttribute("par", *picture_aspect_ratio_.begin());
807 
808  if (!adaptation_set.AddContentProtectionElements(
809  content_protection_elements_)) {
810  return xml::scoped_xml_ptr<xmlNode>();
811  }
812 
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);
818  }
819  if (!switching_ids.empty()) {
820  adaptation_set.AddSupplementalProperty(
821  "urn:mpeg:dash:adaptation-set-switching:2016", switching_ids);
822  }
823 
824  for (AdaptationSet::Role role : roles_)
825  adaptation_set.AddRoleElement("urn:mpeg:dash:role:2011", RoleToText(role));
826 
827  for (const std::unique_ptr<Representation>& representation :
828  representations_) {
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>();
838  }
839 
840  return adaptation_set.PassScopedPtr();
841 }
842 
843 void AdaptationSet::ForceSetSegmentAlignment(bool segment_alignment) {
844  segments_aligned_ =
845  segment_alignment ? kSegmentAlignmentTrue : kSegmentAlignmentFalse;
846  force_set_segment_alignment_ = true;
847 }
848 
849 void AdaptationSet::AddAdaptationSetSwitching(uint32_t adaptation_set_id) {
850  adaptation_set_switching_ids_.push_back(adaptation_set_id);
851 }
852 
853 // Check segmentAlignment for Live here. Storing all start_time and duration
854 // will out-of-memory because there's no way of knowing when it will end.
855 // VOD subsegmentAlignment check is *not* done here because it is possible
856 // that some Representations might not have been added yet (e.g. a thread is
857 // assigned per muxer so one might run faster than others).
858 // To be clear, for Live, all Representations should be added before a
859 // segment is added.
860 void AdaptationSet::OnNewSegmentForRepresentation(uint32_t representation_id,
861  uint64_t start_time,
862  uint64_t duration) {
863  if (mpd_type_ == MpdBuilder::kDynamic) {
864  CheckLiveSegmentAlignment(representation_id, start_time, duration);
865  } else {
866  representation_segment_start_times_[representation_id].push_back(
867  start_time);
868  }
869 }
870 
872  uint32_t representation_id,
873  uint32_t frame_duration,
874  uint32_t timescale) {
875  RecordFrameRate(frame_duration, timescale);
876 }
877 
878 bool AdaptationSet::GetEarliestTimestamp(double* timestamp_seconds) {
879  DCHECK(timestamp_seconds);
880 
881  double earliest_timestamp(-1);
882  for (const std::unique_ptr<Representation>& representation :
883  representations_) {
884  double timestamp;
885  if (representation->GetEarliestTimestamp(&timestamp) &&
886  ((earliest_timestamp < 0) || (timestamp < earliest_timestamp))) {
887  earliest_timestamp = timestamp;
888  }
889  }
890  if (earliest_timestamp < 0)
891  return false;
892 
893  *timestamp_seconds = earliest_timestamp;
894  return true;
895 }
896 
897 // This implementation assumes that each representations' segments' are
898 // contiguous.
899 // Also assumes that all Representations are added before this is called.
900 // This checks whether the first elements of the lists in
901 // representation_segment_start_times_ are aligned.
902 // For example, suppose this method was just called with args rep_id=2
903 // start_time=1.
904 // 1 -> [1, 100, 200]
905 // 2 -> [1]
906 // The timestamps of the first elements match, so this flags
907 // segments_aligned_=true.
908 // Also since the first segment start times match, the first element of all the
909 // lists are removed, so the map of lists becomes:
910 // 1 -> [100, 200]
911 // 2 -> []
912 // Note that there could be false positives.
913 // e.g. just got rep_id=3 start_time=1 duration=300, and the duration of the
914 // whole AdaptationSet is 300.
915 // 1 -> [1, 100, 200]
916 // 2 -> [1, 90, 100]
917 // 3 -> [1]
918 // They are not aligned but this will be marked as aligned.
919 // But since this is unlikely to happen in the packager (and to save
920 // computation), this isn't handled at the moment.
921 void AdaptationSet::CheckLiveSegmentAlignment(uint32_t representation_id,
922  uint64_t start_time,
923  uint64_t /* duration */) {
924  if (segments_aligned_ == kSegmentAlignmentFalse ||
925  force_set_segment_alignment_) {
926  return;
927  }
928 
929  std::list<uint64_t>& representation_start_times =
930  representation_segment_start_times_[representation_id];
931  representation_start_times.push_back(start_time);
932  // There's no way to detemine whether the segments are aligned if some
933  // representations do not have any segments.
934  if (representation_segment_start_times_.size() != representations_.size())
935  return;
936 
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) {
942  // If there are no entries in a list, then there is no way for the
943  // segment alignment status to change.
944  // Note that it can be empty because entries get deleted below.
945  if (it->second.empty())
946  return;
947 
948  if (expected_start_time != it->second.front()) {
949  // Flag as false and clear the start times data, no need to keep it
950  // around.
951  segments_aligned_ = kSegmentAlignmentFalse;
952  representation_segment_start_times_.clear();
953  return;
954  }
955  }
956  segments_aligned_ = kSegmentAlignmentTrue;
957 
958  for (RepresentationTimeline::iterator it =
959  representation_segment_start_times_.begin();
960  it != representation_segment_start_times_.end(); ++it) {
961  it->second.pop_front();
962  }
963 }
964 
965 // Make sure all segements start times match for all Representations.
966 // This assumes that the segments are contiguous.
967 void AdaptationSet::CheckVodSegmentAlignment() {
968  if (segments_aligned_ == kSegmentAlignmentFalse ||
969  force_set_segment_alignment_) {
970  return;
971  }
972  if (representation_segment_start_times_.empty())
973  return;
974  if (representation_segment_start_times_.size() == 1) {
975  segments_aligned_ = kSegmentAlignmentTrue;
976  return;
977  }
978 
979  // This is not the most efficient implementation to compare the values
980  // because expected_time_line is compared against all other time lines, but
981  // probably the most readable.
982  const std::list<uint64_t>& expected_time_line =
983  representation_segment_start_times_.begin()->second;
984 
985  bool all_segment_time_line_same_length = true;
986  // Note that the first entry is skipped because it is expected_time_line.
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;
993  }
994 
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;
1000  }
1001 
1002  if (!std::equal(shorter_list->begin(), shorter_list->end(),
1003  longer_list->begin())) {
1004  // Some segments are definitely unaligned.
1005  segments_aligned_ = kSegmentAlignmentFalse;
1006  representation_segment_start_times_.clear();
1007  return;
1008  }
1009  }
1010 
1011  // TODO(rkuroiwa): The right way to do this is to also check the durations.
1012  // For example:
1013  // (a) 3 4 5
1014  // (b) 3 4 5 6
1015  // could be true or false depending on the length of the third segment of (a).
1016  // i.e. if length of the third segment is 2, then this is not aligned.
1017  if (!all_segment_time_line_same_length) {
1018  segments_aligned_ = kSegmentAlignmentUnknown;
1019  return;
1020  }
1021 
1022  segments_aligned_ = kSegmentAlignmentTrue;
1023 }
1024 
1025 // Since all AdaptationSet cares about is the maxFrameRate, representation_id
1026 // is not passed to this method.
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.";
1031  return;
1032  }
1033  video_frame_rates_[static_cast<double>(timescale) / frame_duration] =
1034  base::IntToString(timescale) + "/" + base::IntToString(frame_duration);
1035 }
1036 
1038  const MediaInfo& media_info,
1039  const MpdOptions& mpd_options,
1040  uint32_t id,
1041  std::unique_ptr<RepresentationStateChangeListener> state_change_listener)
1042  : media_info_(media_info),
1043  id_(id),
1044  bandwidth_estimator_(BandwidthEstimator::kUseAllBlocks),
1045  mpd_options_(mpd_options),
1046  start_number_(1),
1047  state_change_listener_(std::move(state_change_listener)),
1048  output_suppression_flags_(0) {}
1049 
1050 Representation::~Representation() {}
1051 
1053  if (!AtLeastOneTrue(media_info_.has_video_info(),
1054  media_info_.has_audio_info(),
1055  media_info_.has_text_info())) {
1056  // This is an error. Segment information can be in AdaptationSet, Period, or
1057  // MPD but the interface does not provide a way to set them.
1058  // See 5.3.9.1 ISO 23009-1:2012 for segment info.
1059  LOG(ERROR) << "Representation needs one of video, audio, or text.";
1060  return false;
1061  }
1062 
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.";
1067  return false;
1068  }
1069 
1070  if (media_info_.container_type() == MediaInfo::CONTAINER_UNKNOWN) {
1071  LOG(ERROR) << "'container_type' in MediaInfo cannot be CONTAINER_UNKNOWN.";
1072  return false;
1073  }
1074 
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.";
1079  return false;
1080  }
1081  } else if (media_info_.has_audio_info()) {
1082  mime_type_ = GetAudioMimeType();
1083  } else if (media_info_.has_text_info()) {
1084  mime_type_ = GetTextMimeType();
1085  }
1086 
1087  if (mime_type_.empty())
1088  return false;
1089 
1090  codecs_ = GetCodecs(media_info_);
1091  return true;
1092 }
1093 
1095  const ContentProtectionElement& content_protection_element) {
1096  content_protection_elements_.push_back(content_protection_element);
1097  RemoveDuplicateAttributes(&content_protection_elements_.back());
1098 }
1099 
1100 void Representation::UpdateContentProtectionPssh(const std::string& drm_uuid,
1101  const std::string& pssh) {
1102  UpdateContentProtectionPsshHelper(drm_uuid, pssh,
1103  &content_protection_elements_);
1104 }
1105 
1106 void Representation::AddNewSegment(uint64_t start_time,
1107  uint64_t duration,
1108  uint64_t size) {
1109  if (start_time == 0 && duration == 0) {
1110  LOG(WARNING) << "Got segment with start_time and duration == 0. Ignoring.";
1111  return;
1112  }
1113 
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;
1118  } else {
1119  SegmentInfo s = {start_time, duration, /* Not repeat. */ 0};
1120  segment_infos_.push_back(s);
1121  }
1122 
1123  bandwidth_estimator_.AddBlock(
1124  size, static_cast<double>(duration) / media_info_.reference_time_scale());
1125 
1126  SlideWindow();
1127  DCHECK_GE(segment_infos_.size(), 1u);
1128 }
1129 
1130 void Representation::SetSampleDuration(uint32_t sample_duration) {
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());
1136  }
1137  }
1138 }
1139 
1140 // Uses info in |media_info_| and |content_protection_elements_| to create a
1141 // "Representation" node.
1142 // MPD schema has strict ordering. The following must be done in order.
1143 // AddVideoInfo() (possibly adds FramePacking elements), AddAudioInfo() (Adds
1144 // AudioChannelConfig elements), AddContentProtectionElements*(), and
1145 // AddVODOnlyInfo() (Adds segment info).
1146 xml::scoped_xml_ptr<xmlNode> Representation::GetXml() {
1147  if (!HasRequiredMediaInfoFields()) {
1148  LOG(ERROR) << "MediaInfo missing required fields.";
1149  return xml::scoped_xml_ptr<xmlNode>();
1150  }
1151 
1152  const uint64_t bandwidth = media_info_.has_bandwidth()
1153  ? media_info_.bandwidth()
1154  : bandwidth_estimator_.Estimate();
1155 
1156  DCHECK(!(HasVODOnlyFields(media_info_) && HasLiveOnlyFields(media_info_)));
1157 
1158  RepresentationXmlNode representation;
1159  // Mandatory fields for 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_);
1165 
1166  const bool has_video_info = media_info_.has_video_info();
1167  const bool has_audio_info = media_info_.has_audio_info();
1168 
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>();
1177  }
1178 
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>();
1183  }
1184 
1185  if (!representation.AddContentProtectionElements(
1186  content_protection_elements_)) {
1187  return xml::scoped_xml_ptr<xmlNode>();
1188  }
1189 
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>();
1194  }
1195 
1196  if (HasLiveOnlyFields(media_info_) &&
1197  !representation.AddLiveOnlyInfo(media_info_, segment_infos_,
1198  start_number_)) {
1199  LOG(ERROR) << "Failed to add Live info.";
1200  return xml::scoped_xml_ptr<xmlNode>();
1201  }
1202  // TODO(rkuroiwa): It is likely that all representations have the exact same
1203  // SegmentTemplate. Optimize and propagate the tag up to AdaptationSet level.
1204 
1205  output_suppression_flags_ = 0;
1206  return representation.PassScopedPtr();
1207 }
1208 
1209 void Representation::SuppressOnce(SuppressFlag flag) {
1210  output_suppression_flags_ |= flag;
1211 }
1212 
1213 bool Representation::HasRequiredMediaInfoFields() {
1214  if (HasVODOnlyFields(media_info_) && HasLiveOnlyFields(media_info_)) {
1215  LOG(ERROR) << "MediaInfo cannot have both VOD and Live fields.";
1216  return false;
1217  }
1218 
1219  if (!media_info_.has_container_type()) {
1220  LOG(ERROR) << "MediaInfo missing required field: container_type.";
1221  return false;
1222  }
1223 
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.";
1227  return false;
1228  }
1229 
1230  VLOG_IF(3, HasLiveOnlyFields(media_info_) && !media_info_.has_bandwidth())
1231  << "MediaInfo missing field 'bandwidth'. Using estimated from "
1232  "segment size.";
1233 
1234  return true;
1235 }
1236 
1237 bool Representation::IsContiguous(uint64_t start_time,
1238  uint64_t duration,
1239  uint64_t size) const {
1240  if (segment_infos_.empty())
1241  return false;
1242 
1243  // Contiguous segment.
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) {
1249  return true;
1250  }
1251 
1252  // No out of order segments.
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 << ".";
1260  return false;
1261  }
1262 
1263  // A gap since previous.
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 << ".";
1272  return false;
1273  }
1274 
1275  // No overlapping segments.
1276  if (start_time < previous_segment_end_time - kRoundingErrorGrace) {
1277  LOG(WARNING)
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 << ".";
1281  return false;
1282  }
1283 
1284  // Within rounding error grace but technically not contiguous in terms of MPD.
1285  return false;
1286 }
1287 
1288 void Representation::SlideWindow() {
1289  DCHECK(!segment_infos_.empty());
1290  if (mpd_options_.time_shift_buffer_depth <= 0.0)
1291  return;
1292 
1293  const uint32_t time_scale = GetTimeScale(media_info_);
1294  DCHECK_GT(time_scale, 0u);
1295 
1296  uint64_t time_shift_buffer_depth =
1297  static_cast<uint64_t>(mpd_options_.time_shift_buffer_depth * time_scale);
1298 
1299  // The start time of the latest segment is considered the current_play_time,
1300  // and this should guarantee that the latest segment will stay in the list.
1301  const uint64_t current_play_time = LatestSegmentStartTime(segment_infos_);
1302  if (current_play_time <= time_shift_buffer_depth)
1303  return;
1304 
1305  const uint64_t timeshift_limit = current_play_time - time_shift_buffer_depth;
1306 
1307  // First remove all the SegmentInfos that are completely out of range, by
1308  // looking at the very last segment's end time.
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)
1315  break;
1316  num_segments_removed += last->repeat + 1;
1317  }
1318  segment_infos_.erase(first, last);
1319  start_number_ += num_segments_removed;
1320 
1321  // Now some segment in the first SegmentInfo should be left in the list.
1322  SegmentInfo* first_segment_info = &segment_infos_.front();
1323  DCHECK_LE(timeshift_limit, LastSegmentEndTime(*first_segment_info));
1324 
1325  // Identify which segments should still be in the SegmentInfo.
1326  const int repeat_index =
1327  SearchTimedOutRepeatIndex(timeshift_limit, *first_segment_info);
1328  CHECK_GE(repeat_index, 0);
1329  if (repeat_index == 0)
1330  return;
1331 
1332  first_segment_info->start_time = first_segment_info->start_time +
1333  first_segment_info->duration * repeat_index;
1334 
1335  first_segment_info->repeat = first_segment_info->repeat - repeat_index;
1336  start_number_ += repeat_index;
1337 }
1338 
1339 std::string Representation::GetVideoMimeType() const {
1340  return GetMimeType("video", media_info_.container_type());
1341 }
1342 
1343 std::string Representation::GetAudioMimeType() const {
1344  return GetMimeType("audio", media_info_.container_type());
1345 }
1346 
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";
1355  default:
1356  LOG(ERROR) << "Failed to determine MIME type for TTML container: "
1357  << media_info_.container_type();
1358  return "";
1359  }
1360  }
1361  if (media_info_.text_info().format() == "vtt") {
1362  if (media_info_.container_type() == MediaInfo::CONTAINER_TEXT) {
1363  return "text/vtt";
1364  }
1365  LOG(ERROR) << "Failed to determine MIME type for VTT container: "
1366  << media_info_.container_type();
1367  return "";
1368  }
1369 
1370  LOG(ERROR) << "Cannot determine MIME type for format: "
1371  << media_info_.text_info().format()
1372  << " container: " << media_info_.container_type();
1373  return "";
1374 }
1375 
1376 bool Representation::GetEarliestTimestamp(double* timestamp_seconds) {
1377  DCHECK(timestamp_seconds);
1378 
1379  if (segment_infos_.empty())
1380  return false;
1381 
1382  *timestamp_seconds = static_cast<double>(segment_infos_.begin()->start_time) /
1383  GetTimeScale(media_info_);
1384  return true;
1385 }
1386 
1387 } // namespace shaka
void OnSetFrameRateForRepresentation(uint32_t representation_id, uint32_t frame_duration, uint32_t timescale)
Definition: mpd_builder.cc:871
virtual void AddNewSegment(uint64_t start_time, uint64_t duration, uint64_t size)
bool WriteMpdToFile(media::File *output_file)
Definition: mpd_builder.cc:424
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)
Definition: mpd_builder.cc:673
virtual void SetSampleDuration(uint32_t sample_duration)
virtual Representation * AddRepresentation(const MediaInfo &media_info)
Definition: mpd_builder.cc:690
std::string LanguageToShortestForm(const std::string &language)
Define an abstract file interface.
Definition: file.h:24
virtual void AddContentProtectionElement(const ContentProtectionElement &element)
Definition: mpd_builder.cc:734
virtual void AddRole(Role role)
Definition: mpd_builder.cc:746
void AddBaseUrl(const std::string &base_url)
Definition: mpd_builder.cc:406
virtual void UpdateContentProtectionPssh(const std::string &drm_uuid, const std::string &pssh)
Definition: mpd_builder.cc:740
xml::scoped_xml_ptr< xmlNode > GetXml()
virtual bool ToString(std::string *output)
Definition: mpd_builder.cc:429
void AddAdaptationSetSwitching(uint32_t adaptation_set_id)
Definition: mpd_builder.cc:849
virtual void ForceSetSegmentAlignment(bool segment_alignment)
Definition: mpd_builder.cc:843
static void MakePathsRelativeToMpd(const std::string &mpd_path, MediaInfo *media_info)
Definition: mpd_builder.cc:645
MpdBuilder(MpdType type, const MpdOptions &mpd_options)
Definition: mpd_builder.cc:399
xml::scoped_xml_ptr< xmlNode > GetXml()
Definition: mpd_builder.cc:756
virtual void AddContentProtectionElement(const ContentProtectionElement &element)
virtual AdaptationSet * AddAdaptationSet(const std::string &lang)
Definition: mpd_builder.cc:410
Defines Mpd Options.
Definition: mpd_options.h:15
void OnNewSegmentForRepresentation(uint32_t representation_id, uint64_t start_time, uint64_t duration)
Definition: mpd_builder.cc:860
virtual void UpdateContentProtectionPssh(const std::string &drm_uuid, const std::string &pssh)
void SuppressOnce(SuppressFlag flag)