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 <string>
16 
17 #include "packager/base/base64.h"
18 #include "packager/base/bind.h"
19 #include "packager/base/files/file_path.h"
20 #include "packager/base/logging.h"
21 #include "packager/base/memory/scoped_ptr.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 const int kAdaptationSetGroupNotSet = -1;
44 
45 AdaptationSet::Role MediaInfoTextTypeToRole(
46  MediaInfo::TextInfo::TextType type) {
47  switch (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;
55  default:
56  NOTREACHED() << "Unknown MediaInfo TextType: " << type
57  << " assuming subtitle.";
58  return AdaptationSet::kRoleSubtitle;
59  }
60 }
61 
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:
68  // NOTE: DASH MPD spec uses lowercase but RFC3555 says uppercase.
69  return prefix + "/MP2T";
70  case MediaInfo::CONTAINER_WEBM:
71  return prefix + "/webm";
72  default:
73  break;
74  }
75 
76  // Unsupported container types should be rejected/handled by the caller.
77  LOG(ERROR) << "Unrecognized container type: " << container_type;
78  return std::string();
79 }
80 
81 void AddMpdNameSpaceInfo(XmlNode* mpd) {
82  DCHECK(mpd);
83 
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";
91 
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);
97 }
98 
99 bool IsPeriodNode(xmlNodePtr node) {
100  DCHECK(node);
101  int kEqual = 0;
102  return xmlStrcmp(node->name, reinterpret_cast<const xmlChar*>("Period")) ==
103  kEqual;
104 }
105 
106 // Find the first <Period> element. This does not recurse down the tree,
107 // only checks direct children. Returns the pointer to Period element on
108 // success, otherwise returns false.
109 // As noted here, we must traverse.
110 // http://www.xmlsoft.org/tutorial/ar01s04.html
111 xmlNodePtr FindPeriodNode(XmlNode* xml_node) {
112  for (xmlNodePtr node = xml_node->GetRawPtr()->xmlChildrenNode; node != NULL;
113  node = node->next) {
114  if (IsPeriodNode(node))
115  return node;
116  }
117 
118  return NULL;
119 }
120 
121 bool Positive(double d) {
122  return d > 0.0;
123 }
124 
125 // Return current time in XML DateTime format. The value is in UTC, so the
126 // string ends with a 'Z'.
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);
134 
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);
139 }
140 
141 void SetIfPositive(const char* attr_name, double value, XmlNode* mpd) {
142  if (Positive(value)) {
143  mpd->SetStringAttribute(attr_name, SecondsToXmlDuration(value));
144  }
145 }
146 
147 uint32_t GetTimeScale(const MediaInfo& media_info) {
148  if (media_info.has_reference_time_scale()) {
149  return media_info.reference_time_scale();
150  }
151 
152  if (media_info.has_video_info()) {
153  return media_info.video_info().time_scale();
154  }
155 
156  if (media_info.has_audio_info()) {
157  return media_info.audio_info().time_scale();
158  }
159 
160  LOG(WARNING) << "No timescale specified, using 1 as timescale.";
161  return 1;
162 }
163 
164 uint64_t LastSegmentStartTime(const SegmentInfo& segment_info) {
165  return segment_info.start_time + segment_info.duration * segment_info.repeat;
166 }
167 
168 // This is equal to |segment_info| end time
169 uint64_t LastSegmentEndTime(const SegmentInfo& segment_info) {
170  return segment_info.start_time +
171  segment_info.duration * (segment_info.repeat + 1);
172 }
173 
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);
178 }
179 
180 // Given |timeshift_limit|, finds out the number of segments that are no longer
181 // valid and should be removed from |segment_info|.
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)
186  return 0;
187 
188  return (timeshift_limit - segment_info.start_time) / segment_info.duration;
189 }
190 
191 // Overload this function to support different types of |output|.
192 // Note that this could be done by call MpdBuilder::ToString() and use the
193 // result to write to a file, it requires an extra copy.
194 bool WriteXmlCharArrayToOutput(xmlChar* doc,
195  int doc_size,
196  std::string* output) {
197  DCHECK(doc);
198  DCHECK(output);
199  output->assign(doc, doc + doc_size);
200  return true;
201 }
202 
203 bool WriteXmlCharArrayToOutput(xmlChar* doc,
204  int doc_size,
205  media::File* output) {
206  DCHECK(doc);
207  DCHECK(output);
208  if (output->Write(doc, doc_size) < doc_size)
209  return false;
210 
211  return output->Flush();
212 }
213 
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;
217 }
218 
219 // Check whether the video info has width and height.
220 // DASH IOP also requires several other fields for video representations, namely
221 // width, height, framerate, and sar.
222 bool HasRequiredVideoFields(const MediaInfo_VideoInfo& video_info) {
223  if (!video_info.has_height() || !video_info.has_width()) {
224  LOG(ERROR)
225  << "Width and height are required fields for generating a valid MPD.";
226  return false;
227  }
228  // These fields are not required for a valid MPD, but required for DASH IOP
229  // compliant MPD. MpdBuilder can keep generating MPDs without these fields.
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.";
242  return true;
243 }
244 
245 // Returns the picture aspect ratio string e.g. "16:9", "4:3".
246 // "Reducing the quotient to minimal form" does not work well in practice as
247 // there may be some rounding performed in the input, e.g. the resolution of
248 // 480p is 854:480 for 16:9 aspect ratio, can only be reduced to 427:240.
249 // The algorithm finds out the pair of integers, num and den, where num / den is
250 // the closest ratio to scaled_width / scaled_height, by looping den through
251 // common values.
252 std::string GetPictureAspectRatio(uint32_t width,
253  uint32_t height,
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;
259 
260  // Typical aspect ratios have par_y less than or equal to 19:
261  // https://en.wikipedia.org/wiki/List_of_common_resolutions
262  const uint32_t kLargestPossibleParY = 19;
263 
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) {
271  min_error = error;
272  par_num = num;
273  par_den = den;
274  if (error == 0) break;
275  }
276  }
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 << ".";
280 
281  return base::IntToString(par_num) + ":" + base::IntToString(par_den);
282 }
283 
284 // Adds an entry to picture_aspect_ratio if the size of picture_aspect_ratio is
285 // less than 2 and video_info has both pixel width and pixel height.
286 void AddPictureAspectRatio(
287  const MediaInfo::VideoInfo& video_info,
288  std::set<std::string>* picture_aspect_ratio) {
289  // If there are more than one entries in picture_aspect_ratio, the @par
290  // attribute cannot be set, so skip.
291  if (picture_aspect_ratio->size() > 1)
292  return;
293 
294  if (video_info.width() == 0 || video_info.height() == 0 ||
295  video_info.pixel_width() == 0 || video_info.pixel_height() == 0) {
296  // If there is even one Representation without a @sar attribute, @par cannot
297  // be calculated.
298  // Just populate the set with at least 2 bogus strings so that further call
299  // to this function will bail out immediately.
300  picture_aspect_ratio->insert("bogus");
301  picture_aspect_ratio->insert("entries");
302  return;
303  }
304 
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);
314 }
315 
316 std::string RoleToText(AdaptationSet::Role role) {
317  // Using switch so that the compiler can detect whether there is a case that's
318  // not being handled.
319  switch (role) {
320  case AdaptationSet::kRoleCaption:
321  return "caption";
322  case AdaptationSet::kRoleSubtitle:
323  return "subtitle";
324  case AdaptationSet::kRoleMain:
325  return "main";
326  case AdaptationSet::kRoleAlternate:
327  return "alternate";
328  case AdaptationSet::kRoleSupplementary:
329  return "supplementary";
330  case AdaptationSet::kRoleCommentary:
331  return "commentary";
332  case AdaptationSet::kRoleDub:
333  return "dub";
334  default:
335  NOTREACHED();
336  return "";
337  }
338 
339  NOTREACHED();
340  return "";
341 }
342 
343 // Spooky static initialization/cleanup of libxml.
344 class LibXmlInitializer {
345  public:
346  LibXmlInitializer() : initialized_(false) {
347  base::AutoLock lock(lock_);
348  if (!initialized_) {
349  xmlInitParser();
350  initialized_ = true;
351  }
352  }
353 
354  ~LibXmlInitializer() {
355  base::AutoLock lock(lock_);
356  if (initialized_) {
357  xmlCleanupParser();
358  initialized_ = false;
359  }
360  }
361 
362  private:
363  base::Lock lock_;
364  bool initialized_;
365 
366  DISALLOW_COPY_AND_ASSIGN(LibXmlInitializer);
367 };
368 
369 class RepresentationStateChangeListenerImpl
370  : public RepresentationStateChangeListener {
371  public:
372  // |adaptation_set| is not owned by this class.
373  RepresentationStateChangeListenerImpl(uint32_t representation_id,
374  AdaptationSet* adaptation_set)
375  : representation_id_(representation_id), adaptation_set_(adaptation_set) {
376  DCHECK(adaptation_set_);
377  }
378  ~RepresentationStateChangeListenerImpl() override {}
379 
380  // RepresentationStateChangeListener implementation.
381  void OnNewSegmentForRepresentation(uint64_t start_time,
382  uint64_t duration) override {
383  adaptation_set_->OnNewSegmentForRepresentation(representation_id_,
384  start_time, duration);
385  }
386 
387  void OnSetFrameRateForRepresentation(uint32_t frame_duration,
388  uint32_t timescale) override {
389  adaptation_set_->OnSetFrameRateForRepresentation(representation_id_,
390  frame_duration, timescale);
391  }
392 
393  private:
394  const uint32_t representation_id_;
395  AdaptationSet* const adaptation_set_;
396 
397  DISALLOW_COPY_AND_ASSIGN(RepresentationStateChangeListenerImpl);
398 };
399 
400 } // namespace
401 
402 MpdBuilder::MpdBuilder(MpdType type, const MpdOptions& mpd_options)
403  : type_(type),
404  mpd_options_(mpd_options),
405  adaptation_sets_deleter_(&adaptation_sets_),
406  clock_(new base::DefaultClock()) {}
407 
408 MpdBuilder::~MpdBuilder() {}
409 
410 void MpdBuilder::AddBaseUrl(const std::string& base_url) {
411  base_urls_.push_back(base_url);
412 }
413 
414 AdaptationSet* MpdBuilder::AddAdaptationSet(const std::string& lang) {
415  scoped_ptr<AdaptationSet> adaptation_set(
416  new AdaptationSet(adaptation_set_counter_.GetNext(), lang, mpd_options_,
417  type_, &representation_counter_));
418 
419  DCHECK(adaptation_set);
420  adaptation_sets_.push_back(adaptation_set.get());
421  return adaptation_set.release();
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  std::list<AdaptationSet*>::iterator adaptation_sets_it =
469  adaptation_sets_.begin();
470  for (; adaptation_sets_it != adaptation_sets_.end(); ++adaptation_sets_it) {
471  xml::scoped_xml_ptr<xmlNode> child((*adaptation_sets_it)->GetXml());
472  if (!child.get() || !period.AddChild(child.Pass()))
473  return NULL;
474  }
475 
476  // Add baseurls to MPD.
477  std::list<std::string>::const_iterator base_urls_it = base_urls_.begin();
478  for (; base_urls_it != base_urls_.end(); ++base_urls_it) {
479  XmlNode base_url("BaseURL");
480  base_url.SetContent(*base_urls_it);
481 
482  if (!mpd.AddChild(base_url.PassScopedPtr()))
483  return NULL;
484  }
485 
486  if (type_ == kDynamic) {
487  // This is the only Period and it is a regular period.
488  period.SetStringAttribute("start", "PT0S");
489  }
490 
491  if (!mpd.AddChild(period.PassScopedPtr()))
492  return NULL;
493 
494  AddMpdNameSpaceInfo(&mpd);
495  AddCommonMpdInfo(&mpd);
496  switch (type_) {
497  case kStatic:
498  AddStaticMpdInfo(&mpd);
499  break;
500  case kDynamic:
501  AddDynamicMpdInfo(&mpd);
502  break;
503  default:
504  NOTREACHED() << "Unknown MPD type: " << type_;
505  break;
506  }
507 
508  DCHECK(doc);
509  const std::string version = GetPackagerVersion();
510  if (!version.empty()) {
511  std::string version_string =
512  base::StringPrintf("Generated with %s version %s",
513  GetPackagerProjectUrl().c_str(), version.c_str());
514  xml::scoped_xml_ptr<xmlNode> comment(
515  xmlNewDocComment(doc.get(), BAD_CAST version_string.c_str()));
516  xmlDocSetRootElement(doc.get(), comment.get());
517  xmlAddSibling(comment.release(), mpd.Release());
518  } else {
519  xmlDocSetRootElement(doc.get(), mpd.Release());
520  }
521  return doc.release();
522 }
523 
524 void MpdBuilder::AddCommonMpdInfo(XmlNode* mpd_node) {
525  if (Positive(mpd_options_.min_buffer_time)) {
526  mpd_node->SetStringAttribute(
527  "minBufferTime", SecondsToXmlDuration(mpd_options_.min_buffer_time));
528  } else {
529  LOG(ERROR) << "minBufferTime value not specified.";
530  // TODO(tinskip): Propagate error.
531  }
532 }
533 
534 void MpdBuilder::AddStaticMpdInfo(XmlNode* mpd_node) {
535  DCHECK(mpd_node);
536  DCHECK_EQ(MpdBuilder::kStatic, type_);
537 
538  static const char kStaticMpdType[] = "static";
539  static const char kStaticMpdProfile[] =
540  "urn:mpeg:dash:profile:isoff-on-demand:2011";
541  mpd_node->SetStringAttribute("type", kStaticMpdType);
542  mpd_node->SetStringAttribute("profiles", kStaticMpdProfile);
543  mpd_node->SetStringAttribute(
544  "mediaPresentationDuration",
545  SecondsToXmlDuration(GetStaticMpdDuration(mpd_node)));
546 }
547 
548 void MpdBuilder::AddDynamicMpdInfo(XmlNode* mpd_node) {
549  DCHECK(mpd_node);
550  DCHECK_EQ(MpdBuilder::kDynamic, type_);
551 
552  static const char kDynamicMpdType[] = "dynamic";
553  static const char kDynamicMpdProfile[] =
554  "urn:mpeg:dash:profile:isoff-live:2011";
555  mpd_node->SetStringAttribute("type", kDynamicMpdType);
556  mpd_node->SetStringAttribute("profiles", kDynamicMpdProfile);
557 
558  // No offset from NOW.
559  mpd_node->SetStringAttribute("publishTime",
560  XmlDateTimeNowWithOffset(0, clock_.get()));
561 
562  // 'availabilityStartTime' is required for dynamic profile. Calculate if
563  // not already calculated.
564  if (availability_start_time_.empty()) {
565  double earliest_presentation_time;
566  if (GetEarliestTimestamp(&earliest_presentation_time)) {
567  availability_start_time_ =
568  XmlDateTimeNowWithOffset(mpd_options_.availability_time_offset -
569  std::ceil(earliest_presentation_time),
570  clock_.get());
571  } else {
572  LOG(ERROR) << "Could not determine the earliest segment presentation "
573  "time for availabilityStartTime calculation.";
574  // TODO(tinskip). Propagate an error.
575  }
576  }
577  if (!availability_start_time_.empty())
578  mpd_node->SetStringAttribute("availabilityStartTime",
579  availability_start_time_);
580 
581  if (Positive(mpd_options_.minimum_update_period)) {
582  mpd_node->SetStringAttribute(
583  "minimumUpdatePeriod",
584  SecondsToXmlDuration(mpd_options_.minimum_update_period));
585  } else {
586  LOG(WARNING) << "The profile is dynamic but no minimumUpdatePeriod "
587  "specified.";
588  }
589 
590  SetIfPositive("timeShiftBufferDepth", mpd_options_.time_shift_buffer_depth,
591  mpd_node);
592  SetIfPositive("suggestedPresentationDelay",
593  mpd_options_.suggested_presentation_delay, mpd_node);
594 }
595 
596 float MpdBuilder::GetStaticMpdDuration(XmlNode* mpd_node) {
597  DCHECK(mpd_node);
598  DCHECK_EQ(MpdBuilder::kStatic, type_);
599 
600  xmlNodePtr period_node = FindPeriodNode(mpd_node);
601  DCHECK(period_node) << "Period element must be a child of mpd_node.";
602  DCHECK(IsPeriodNode(period_node));
603 
604  // Attribute mediaPresentationDuration must be present for 'static' MPD. So
605  // setting "PT0S" is required even if none of the representaions have duration
606  // attribute.
607  float max_duration = 0.0f;
608  for (xmlNodePtr adaptation_set = xmlFirstElementChild(period_node);
609  adaptation_set; adaptation_set = xmlNextElementSibling(adaptation_set)) {
610  for (xmlNodePtr representation = xmlFirstElementChild(adaptation_set);
611  representation;
612  representation = xmlNextElementSibling(representation)) {
613  float duration = 0.0f;
614  if (GetDurationAttribute(representation, &duration)) {
615  max_duration = max_duration > duration ? max_duration : duration;
616 
617  // 'duration' attribute is there only to help generate MPD, not
618  // necessary for MPD, remove the attribute.
619  xmlUnsetProp(representation, BAD_CAST "duration");
620  }
621  }
622  }
623 
624  return max_duration;
625 }
626 
627 bool MpdBuilder::GetEarliestTimestamp(double* timestamp_seconds) {
628  DCHECK(timestamp_seconds);
629 
630  double earliest_timestamp(-1);
631  for (std::list<AdaptationSet*>::const_iterator iter =
632  adaptation_sets_.begin();
633  iter != adaptation_sets_.end(); ++iter) {
634  double timestamp;
635  if ((*iter)->GetEarliestTimestamp(&timestamp) &&
636  ((earliest_timestamp < 0) || (timestamp < earliest_timestamp))) {
637  earliest_timestamp = timestamp;
638  }
639  }
640  if (earliest_timestamp < 0)
641  return false;
642 
643  *timestamp_seconds = earliest_timestamp;
644  return true;
645 }
646 
647 void MpdBuilder::MakePathsRelativeToMpd(const std::string& mpd_path,
648  MediaInfo* media_info) {
649  DCHECK(media_info);
650  const std::string kFileProtocol("file://");
651  std::string mpd_file_path = (mpd_path.find(kFileProtocol) == 0)
652  ? mpd_path.substr(kFileProtocol.size())
653  : mpd_path;
654 
655  if (!mpd_file_path.empty()) {
656  std::string mpd_dir(
657  FilePath(mpd_file_path).DirName().AsEndingWithSeparator().value());
658  if (!mpd_dir.empty()) {
659  if (media_info->has_media_file_name()) {
660  media_info->set_media_file_name(
661  MakePathRelative(media_info->media_file_name(), mpd_dir));
662  }
663  if (media_info->has_init_segment_name()) {
664  media_info->set_init_segment_name(
665  MakePathRelative(media_info->init_segment_name(), mpd_dir));
666  }
667  if (media_info->has_segment_template()) {
668  media_info->set_segment_template(
669  MakePathRelative(media_info->segment_template(), mpd_dir));
670  }
671  }
672  }
673 }
674 
675 AdaptationSet::AdaptationSet(uint32_t adaptation_set_id,
676  const std::string& lang,
677  const MpdOptions& mpd_options,
678  MpdBuilder::MpdType mpd_type,
679  base::AtomicSequenceNumber* counter)
680  : representations_deleter_(&representations_),
681  representation_counter_(counter),
682  id_(adaptation_set_id),
683  lang_(lang),
684  mpd_options_(mpd_options),
685  mpd_type_(mpd_type),
686  group_(kAdaptationSetGroupNotSet),
687  segments_aligned_(kSegmentAlignmentUnknown),
688  force_set_segment_alignment_(false) {
689  DCHECK(counter);
690 }
691 
692 AdaptationSet::~AdaptationSet() {}
693 
694 Representation* AdaptationSet::AddRepresentation(const MediaInfo& media_info) {
695  const uint32_t representation_id = representation_counter_->GetNext();
696  // Note that AdaptationSet outlive Representation, so this object
697  // will die before AdaptationSet.
698  scoped_ptr<RepresentationStateChangeListener> listener(
699  new RepresentationStateChangeListenerImpl(representation_id, this));
700  scoped_ptr<Representation> representation(new Representation(
701  media_info, mpd_options_, representation_id, listener.Pass()));
702 
703  if (!representation->Init())
704  return NULL;
705 
706  // For videos, record the width, height, and the frame rate to calculate the
707  // max {width,height,framerate} required for DASH IOP.
708  if (media_info.has_video_info()) {
709  const MediaInfo::VideoInfo& video_info = media_info.video_info();
710  DCHECK(video_info.has_width());
711  DCHECK(video_info.has_height());
712  video_widths_.insert(video_info.width());
713  video_heights_.insert(video_info.height());
714 
715  if (video_info.has_time_scale() && video_info.has_frame_duration())
716  RecordFrameRate(video_info.frame_duration(), video_info.time_scale());
717 
718  AddPictureAspectRatio(video_info, &picture_aspect_ratio_);
719  }
720 
721  if (media_info.has_video_info()) {
722  content_type_ = "video";
723  } else if (media_info.has_audio_info()) {
724  content_type_ = "audio";
725  } else if (media_info.has_text_info()) {
726  content_type_ = "text";
727 
728  if (media_info.text_info().has_type() &&
729  (media_info.text_info().type() != MediaInfo::TextInfo::UNKNOWN)) {
730  roles_.insert(MediaInfoTextTypeToRole(media_info.text_info().type()));
731  }
732  }
733 
734  representations_.push_back(representation.get());
735  return representation.release();
736 }
737 
739  const ContentProtectionElement& content_protection_element) {
740  content_protection_elements_.push_back(content_protection_element);
741  RemoveDuplicateAttributes(&content_protection_elements_.back());
742 }
743 
744 void AdaptationSet::UpdateContentProtectionPssh(const std::string& drm_uuid,
745  const std::string& pssh) {
746  UpdateContentProtectionPsshHelper(drm_uuid, pssh,
747  &content_protection_elements_);
748 }
749 
750 void AdaptationSet::AddRole(Role role) {
751  roles_.insert(role);
752 }
753 
754 // Creates a copy of <AdaptationSet> xml element, iterate thru all the
755 // <Representation> (child) elements and add them to the copy.
756 // Set all the attributes first and then add the children elements so that flags
757 // can be passed to Representation to avoid setting redundant attributes. For
758 // example, if AdaptationSet@width is set, then Representation@width is
759 // redundant and should not be set.
760 xml::scoped_xml_ptr<xmlNode> AdaptationSet::GetXml() {
761  AdaptationSetXmlNode adaptation_set;
762 
763  bool suppress_representation_width = false;
764  bool suppress_representation_height = false;
765  bool suppress_representation_frame_rate = false;
766 
767  adaptation_set.SetId(id_);
768  adaptation_set.SetStringAttribute("contentType", content_type_);
769  if (!lang_.empty() && lang_ != "und") {
770  adaptation_set.SetStringAttribute("lang", LanguageToShortestForm(lang_));
771  }
772 
773  // Note that std::{set,map} are ordered, so the last element is the max value.
774  if (video_widths_.size() == 1) {
775  suppress_representation_width = true;
776  adaptation_set.SetIntegerAttribute("width", *video_widths_.begin());
777  } else if (video_widths_.size() > 1) {
778  adaptation_set.SetIntegerAttribute("maxWidth", *video_widths_.rbegin());
779  }
780  if (video_heights_.size() == 1) {
781  suppress_representation_height = true;
782  adaptation_set.SetIntegerAttribute("height", *video_heights_.begin());
783  } else if (video_heights_.size() > 1) {
784  adaptation_set.SetIntegerAttribute("maxHeight", *video_heights_.rbegin());
785  }
786 
787  if (video_frame_rates_.size() == 1) {
788  suppress_representation_frame_rate = true;
789  adaptation_set.SetStringAttribute("frameRate",
790  video_frame_rates_.begin()->second);
791  } else if (video_frame_rates_.size() > 1) {
792  adaptation_set.SetStringAttribute("maxFrameRate",
793  video_frame_rates_.rbegin()->second);
794  }
795 
796  // Note: must be checked before checking segments_aligned_ (below). So that
797  // segments_aligned_ is set before checking below.
798  if (mpd_type_ == MpdBuilder::kStatic) {
799  CheckVodSegmentAlignment();
800  }
801 
802  if (segments_aligned_ == kSegmentAlignmentTrue) {
803  adaptation_set.SetStringAttribute(mpd_type_ == MpdBuilder::kStatic
804  ? "subsegmentAlignment"
805  : "segmentAlignment",
806  "true");
807  }
808 
809  if (picture_aspect_ratio_.size() == 1)
810  adaptation_set.SetStringAttribute("par", *picture_aspect_ratio_.begin());
811 
812  if (group_ >= 0)
813  adaptation_set.SetIntegerAttribute("group", group_);
814 
815  if (!adaptation_set.AddContentProtectionElements(
816  content_protection_elements_)) {
817  return xml::scoped_xml_ptr<xmlNode>();
818  }
819  for (AdaptationSet::Role role : roles_)
820  adaptation_set.AddRoleElement("urn:mpeg:dash:role:2011", RoleToText(role));
821 
822  for (Representation* representation : representations_) {
823  if (suppress_representation_width)
824  representation->SuppressOnce(Representation::kSuppressWidth);
825  if (suppress_representation_height)
826  representation->SuppressOnce(Representation::kSuppressHeight);
827  if (suppress_representation_frame_rate)
828  representation->SuppressOnce(Representation::kSuppressFrameRate);
829  xml::scoped_xml_ptr<xmlNode> child(representation->GetXml());
830  if (!child || !adaptation_set.AddChild(child.Pass()))
831  return xml::scoped_xml_ptr<xmlNode>();
832  }
833 
834  return adaptation_set.PassScopedPtr();
835 }
836 
837 void AdaptationSet::ForceSetSegmentAlignment(bool segment_alignment) {
838  segments_aligned_ =
839  segment_alignment ? kSegmentAlignmentTrue : kSegmentAlignmentFalse;
840  force_set_segment_alignment_ = true;
841 }
842 
843 void AdaptationSet::SetGroup(int group_number) {
844  group_ = group_number;
845 }
846 
847 int AdaptationSet::Group() const {
848  return group_;
849 }
850 
851 // Check segmentAlignment for Live here. Storing all start_time and duration
852 // will out-of-memory because there's no way of knowing when it will end.
853 // VOD subsegmentAlignment check is *not* done here because it is possible
854 // that some Representations might not have been added yet (e.g. a thread is
855 // assigned per muxer so one might run faster than others).
856 // To be clear, for Live, all Representations should be added before a
857 // segment is added.
858 void AdaptationSet::OnNewSegmentForRepresentation(uint32_t representation_id,
859  uint64_t start_time,
860  uint64_t duration) {
861  if (mpd_type_ == MpdBuilder::kDynamic) {
862  CheckLiveSegmentAlignment(representation_id, start_time, duration);
863  } else {
864  representation_segment_start_times_[representation_id].push_back(
865  start_time);
866  }
867 }
868 
870  uint32_t representation_id,
871  uint32_t frame_duration,
872  uint32_t timescale) {
873  RecordFrameRate(frame_duration, timescale);
874 }
875 
876 bool AdaptationSet::GetEarliestTimestamp(double* timestamp_seconds) {
877  DCHECK(timestamp_seconds);
878 
879  double earliest_timestamp(-1);
880  for (std::list<Representation*>::const_iterator iter =
881  representations_.begin();
882  iter != representations_.end(); ++iter) {
883  double timestamp;
884  if ((*iter)->GetEarliestTimestamp(&timestamp) &&
885  ((earliest_timestamp < 0) || (timestamp < earliest_timestamp))) {
886  earliest_timestamp = timestamp;
887  }
888  }
889  if (earliest_timestamp < 0)
890  return false;
891 
892  *timestamp_seconds = earliest_timestamp;
893  return true;
894 }
895 
896 // This implementation assumes that each representations' segments' are
897 // contiguous.
898 // Also assumes that all Representations are added before this is called.
899 // This checks whether the first elements of the lists in
900 // representation_segment_start_times_ are aligned.
901 // For example, suppose this method was just called with args rep_id=2
902 // start_time=1.
903 // 1 -> [1, 100, 200]
904 // 2 -> [1]
905 // The timestamps of the first elements match, so this flags
906 // segments_aligned_=true.
907 // Also since the first segment start times match, the first element of all the
908 // lists are removed, so the map of lists becomes:
909 // 1 -> [100, 200]
910 // 2 -> []
911 // Note that there could be false positives.
912 // e.g. just got rep_id=3 start_time=1 duration=300, and the duration of the
913 // whole AdaptationSet is 300.
914 // 1 -> [1, 100, 200]
915 // 2 -> [1, 90, 100]
916 // 3 -> [1]
917 // They are not aligned but this will be marked as aligned.
918 // But since this is unlikely to happen in the packager (and to save
919 // computation), this isn't handled at the moment.
920 void AdaptationSet::CheckLiveSegmentAlignment(uint32_t representation_id,
921  uint64_t start_time,
922  uint64_t /* duration */) {
923  if (segments_aligned_ == kSegmentAlignmentFalse ||
924  force_set_segment_alignment_) {
925  return;
926  }
927 
928  std::list<uint64_t>& representation_start_times =
929  representation_segment_start_times_[representation_id];
930  representation_start_times.push_back(start_time);
931  // There's no way to detemine whether the segments are aligned if some
932  // representations do not have any segments.
933  if (representation_segment_start_times_.size() != representations_.size())
934  return;
935 
936  DCHECK(!representation_start_times.empty());
937  const uint64_t expected_start_time = representation_start_times.front();
938  for (RepresentationTimeline::const_iterator it =
939  representation_segment_start_times_.begin();
940  it != representation_segment_start_times_.end(); ++it) {
941  // If there are no entries in a list, then there is no way for the
942  // segment alignment status to change.
943  // Note that it can be empty because entries get deleted below.
944  if (it->second.empty())
945  return;
946 
947  if (expected_start_time != it->second.front()) {
948  // Flag as false and clear the start times data, no need to keep it
949  // around.
950  segments_aligned_ = kSegmentAlignmentFalse;
951  representation_segment_start_times_.clear();
952  return;
953  }
954  }
955  segments_aligned_ = kSegmentAlignmentTrue;
956 
957  for (RepresentationTimeline::iterator it =
958  representation_segment_start_times_.begin();
959  it != representation_segment_start_times_.end(); ++it) {
960  it->second.pop_front();
961  }
962 }
963 
964 // Make sure all segements start times match for all Representations.
965 // This assumes that the segments are contiguous.
966 void AdaptationSet::CheckVodSegmentAlignment() {
967  if (segments_aligned_ == kSegmentAlignmentFalse ||
968  force_set_segment_alignment_) {
969  return;
970  }
971  if (representation_segment_start_times_.empty())
972  return;
973  if (representation_segment_start_times_.size() == 1) {
974  segments_aligned_ = kSegmentAlignmentTrue;
975  return;
976  }
977 
978  // This is not the most efficient implementation to compare the values
979  // because expected_time_line is compared against all other time lines, but
980  // probably the most readable.
981  const std::list<uint64_t>& expected_time_line =
982  representation_segment_start_times_.begin()->second;
983 
984  bool all_segment_time_line_same_length = true;
985  // Note that the first entry is skipped because it is expected_time_line.
986  RepresentationTimeline::const_iterator it =
987  representation_segment_start_times_.begin();
988  for (++it; it != representation_segment_start_times_.end(); ++it) {
989  const std::list<uint64_t>& other_time_line = it->second;
990  if (expected_time_line.size() != other_time_line.size()) {
991  all_segment_time_line_same_length = false;
992  }
993 
994  const std::list<uint64_t>* longer_list = &other_time_line;
995  const std::list<uint64_t>* shorter_list = &expected_time_line;
996  if (expected_time_line.size() > other_time_line.size()) {
997  shorter_list = &other_time_line;
998  longer_list = &expected_time_line;
999  }
1000 
1001  if (!std::equal(shorter_list->begin(), shorter_list->end(),
1002  longer_list->begin())) {
1003  // Some segments are definitely unaligned.
1004  segments_aligned_ = kSegmentAlignmentFalse;
1005  representation_segment_start_times_.clear();
1006  return;
1007  }
1008  }
1009 
1010  // TODO(rkuroiwa): The right way to do this is to also check the durations.
1011  // For example:
1012  // (a) 3 4 5
1013  // (b) 3 4 5 6
1014  // could be true or false depending on the length of the third segment of (a).
1015  // i.e. if length of the third segment is 2, then this is not aligned.
1016  if (!all_segment_time_line_same_length) {
1017  segments_aligned_ = kSegmentAlignmentUnknown;
1018  return;
1019  }
1020 
1021  segments_aligned_ = kSegmentAlignmentTrue;
1022 }
1023 
1024 // Since all AdaptationSet cares about is the maxFrameRate, representation_id
1025 // is not passed to this method.
1026 void AdaptationSet::RecordFrameRate(uint32_t frame_duration,
1027  uint32_t timescale) {
1028  if (frame_duration == 0) {
1029  LOG(ERROR) << "Frame duration is 0 and cannot be set.";
1030  return;
1031  }
1032  video_frame_rates_[static_cast<double>(timescale) / frame_duration] =
1033  base::IntToString(timescale) + "/" + base::IntToString(frame_duration);
1034 }
1035 
1037  const MediaInfo& media_info,
1038  const MpdOptions& mpd_options,
1039  uint32_t id,
1040  scoped_ptr<RepresentationStateChangeListener> state_change_listener)
1041  : media_info_(media_info),
1042  id_(id),
1043  bandwidth_estimator_(BandwidthEstimator::kUseAllBlocks),
1044  mpd_options_(mpd_options),
1045  start_number_(1),
1046  state_change_listener_(state_change_listener.Pass()),
1047  output_suppression_flags_(0) {}
1048 
1049 Representation::~Representation() {}
1050 
1052  if (!AtLeastOneTrue(media_info_.has_video_info(),
1053  media_info_.has_audio_info(),
1054  media_info_.has_text_info())) {
1055  // This is an error. Segment information can be in AdaptationSet, Period, or
1056  // MPD but the interface does not provide a way to set them.
1057  // See 5.3.9.1 ISO 23009-1:2012 for segment info.
1058  LOG(ERROR) << "Representation needs one of video, audio, or text.";
1059  return false;
1060  }
1061 
1062  if (MoreThanOneTrue(media_info_.has_video_info(),
1063  media_info_.has_audio_info(),
1064  media_info_.has_text_info())) {
1065  LOG(ERROR) << "Only one of VideoInfo, AudioInfo, or TextInfo can be set.";
1066  return false;
1067  }
1068 
1069  if (media_info_.container_type() == MediaInfo::CONTAINER_UNKNOWN) {
1070  LOG(ERROR) << "'container_type' in MediaInfo cannot be CONTAINER_UNKNOWN.";
1071  return false;
1072  }
1073 
1074  if (media_info_.has_video_info()) {
1075  mime_type_ = GetVideoMimeType();
1076  if (!HasRequiredVideoFields(media_info_.video_info())) {
1077  LOG(ERROR) << "Missing required fields to create a video Representation.";
1078  return false;
1079  }
1080  } else if (media_info_.has_audio_info()) {
1081  mime_type_ = GetAudioMimeType();
1082  } else if (media_info_.has_text_info()) {
1083  mime_type_ = GetTextMimeType();
1084  }
1085 
1086  if (mime_type_.empty())
1087  return false;
1088 
1089  codecs_ = GetCodecs(media_info_);
1090  return true;
1091 }
1092 
1094  const ContentProtectionElement& content_protection_element) {
1095  content_protection_elements_.push_back(content_protection_element);
1096  RemoveDuplicateAttributes(&content_protection_elements_.back());
1097 }
1098 
1099 void Representation::UpdateContentProtectionPssh(const std::string& drm_uuid,
1100  const std::string& pssh) {
1101  UpdateContentProtectionPsshHelper(drm_uuid, pssh,
1102  &content_protection_elements_);
1103 }
1104 
1105 void Representation::AddNewSegment(uint64_t start_time,
1106  uint64_t duration,
1107  uint64_t size) {
1108  if (start_time == 0 && duration == 0) {
1109  LOG(WARNING) << "Got segment with start_time and duration == 0. Ignoring.";
1110  return;
1111  }
1112 
1113  if (state_change_listener_)
1114  state_change_listener_->OnNewSegmentForRepresentation(start_time, duration);
1115  if (IsContiguous(start_time, duration, size)) {
1116  ++segment_infos_.back().repeat;
1117  } else {
1118  SegmentInfo s = {start_time, duration, /* Not repeat. */ 0};
1119  segment_infos_.push_back(s);
1120  }
1121 
1122  bandwidth_estimator_.AddBlock(
1123  size, static_cast<double>(duration) / media_info_.reference_time_scale());
1124 
1125  SlideWindow();
1126  DCHECK_GE(segment_infos_.size(), 1u);
1127 }
1128 
1129 void Representation::SetSampleDuration(uint32_t sample_duration) {
1130  if (media_info_.has_video_info()) {
1131  media_info_.mutable_video_info()->set_frame_duration(sample_duration);
1132  if (state_change_listener_) {
1133  state_change_listener_->OnSetFrameRateForRepresentation(
1134  sample_duration, media_info_.video_info().time_scale());
1135  }
1136  }
1137 }
1138 
1139 // Uses info in |media_info_| and |content_protection_elements_| to create a
1140 // "Representation" node.
1141 // MPD schema has strict ordering. The following must be done in order.
1142 // AddVideoInfo() (possibly adds FramePacking elements), AddAudioInfo() (Adds
1143 // AudioChannelConfig elements), AddContentProtectionElements*(), and
1144 // AddVODOnlyInfo() (Adds segment info).
1145 xml::scoped_xml_ptr<xmlNode> Representation::GetXml() {
1146  if (!HasRequiredMediaInfoFields()) {
1147  LOG(ERROR) << "MediaInfo missing required fields.";
1148  return xml::scoped_xml_ptr<xmlNode>();
1149  }
1150 
1151  const uint64_t bandwidth = media_info_.has_bandwidth()
1152  ? media_info_.bandwidth()
1153  : bandwidth_estimator_.Estimate();
1154 
1155  DCHECK(!(HasVODOnlyFields(media_info_) && HasLiveOnlyFields(media_info_)));
1156 
1157  RepresentationXmlNode representation;
1158  // Mandatory fields for Representation.
1159  representation.SetId(id_);
1160  representation.SetIntegerAttribute("bandwidth", bandwidth);
1161  if (!codecs_.empty())
1162  representation.SetStringAttribute("codecs", codecs_);
1163  representation.SetStringAttribute("mimeType", mime_type_);
1164 
1165  const bool has_video_info = media_info_.has_video_info();
1166  const bool has_audio_info = media_info_.has_audio_info();
1167 
1168  if (has_video_info &&
1169  !representation.AddVideoInfo(
1170  media_info_.video_info(),
1171  !(output_suppression_flags_ & kSuppressWidth),
1172  !(output_suppression_flags_ & kSuppressHeight),
1173  !(output_suppression_flags_ & kSuppressFrameRate))) {
1174  LOG(ERROR) << "Failed to add video info to Representation XML.";
1175  return xml::scoped_xml_ptr<xmlNode>();
1176  }
1177 
1178  if (has_audio_info &&
1179  !representation.AddAudioInfo(media_info_.audio_info())) {
1180  LOG(ERROR) << "Failed to add audio info to Representation XML.";
1181  return xml::scoped_xml_ptr<xmlNode>();
1182  }
1183 
1184  if (!representation.AddContentProtectionElements(
1185  content_protection_elements_)) {
1186  return xml::scoped_xml_ptr<xmlNode>();
1187  }
1188 
1189  if (HasVODOnlyFields(media_info_) &&
1190  !representation.AddVODOnlyInfo(media_info_)) {
1191  LOG(ERROR) << "Failed to add VOD segment info.";
1192  return xml::scoped_xml_ptr<xmlNode>();
1193  }
1194 
1195  if (HasLiveOnlyFields(media_info_) &&
1196  !representation.AddLiveOnlyInfo(media_info_, segment_infos_,
1197  start_number_)) {
1198  LOG(ERROR) << "Failed to add Live info.";
1199  return xml::scoped_xml_ptr<xmlNode>();
1200  }
1201  // TODO(rkuroiwa): It is likely that all representations have the exact same
1202  // SegmentTemplate. Optimize and propagate the tag up to AdaptationSet level.
1203 
1204  output_suppression_flags_ = 0;
1205  return representation.PassScopedPtr();
1206 }
1207 
1208 void Representation::SuppressOnce(SuppressFlag flag) {
1209  output_suppression_flags_ |= flag;
1210 }
1211 
1212 bool Representation::HasRequiredMediaInfoFields() {
1213  if (HasVODOnlyFields(media_info_) && HasLiveOnlyFields(media_info_)) {
1214  LOG(ERROR) << "MediaInfo cannot have both VOD and Live fields.";
1215  return false;
1216  }
1217 
1218  if (!media_info_.has_container_type()) {
1219  LOG(ERROR) << "MediaInfo missing required field: container_type.";
1220  return false;
1221  }
1222 
1223  if (HasVODOnlyFields(media_info_) && !media_info_.has_bandwidth()) {
1224  LOG(ERROR) << "Missing 'bandwidth' field. MediaInfo requires bandwidth for "
1225  "static profile for generating a valid MPD.";
1226  return false;
1227  }
1228 
1229  VLOG_IF(3, HasLiveOnlyFields(media_info_) && !media_info_.has_bandwidth())
1230  << "MediaInfo missing field 'bandwidth'. Using estimated from "
1231  "segment size.";
1232 
1233  return true;
1234 }
1235 
1236 bool Representation::IsContiguous(uint64_t start_time,
1237  uint64_t duration,
1238  uint64_t size) const {
1239  if (segment_infos_.empty())
1240  return false;
1241 
1242  // Contiguous segment.
1243  const SegmentInfo& previous = segment_infos_.back();
1244  const uint64_t previous_segment_end_time =
1245  previous.start_time + previous.duration * (previous.repeat + 1);
1246  if (previous_segment_end_time == start_time &&
1247  segment_infos_.back().duration == duration) {
1248  return true;
1249  }
1250 
1251  // No out of order segments.
1252  const uint64_t previous_segment_start_time =
1253  previous.start_time + previous.duration * previous.repeat;
1254  if (previous_segment_start_time >= start_time) {
1255  LOG(ERROR) << "Segments should not be out of order segment. Adding segment "
1256  "with start_time == "
1257  << start_time << " but the previous segment starts at "
1258  << previous.start_time << ".";
1259  return false;
1260  }
1261 
1262  // A gap since previous.
1263  const uint64_t kRoundingErrorGrace = 5;
1264  if (previous_segment_end_time + kRoundingErrorGrace < start_time) {
1265  LOG(WARNING) << "Found a gap of size "
1266  << (start_time - previous_segment_end_time)
1267  << " > kRoundingErrorGrace (" << kRoundingErrorGrace
1268  << "). The new segment starts at " << start_time
1269  << " but the previous segment ends at "
1270  << previous_segment_end_time << ".";
1271  return false;
1272  }
1273 
1274  // No overlapping segments.
1275  if (start_time < previous_segment_end_time - kRoundingErrorGrace) {
1276  LOG(WARNING)
1277  << "Segments should not be overlapping. The new segment starts at "
1278  << start_time << " but the previous segment ends at "
1279  << previous_segment_end_time << ".";
1280  return false;
1281  }
1282 
1283  // Within rounding error grace but technically not contiguous in terms of MPD.
1284  return false;
1285 }
1286 
1287 void Representation::SlideWindow() {
1288  DCHECK(!segment_infos_.empty());
1289  if (mpd_options_.time_shift_buffer_depth <= 0.0)
1290  return;
1291 
1292  const uint32_t time_scale = GetTimeScale(media_info_);
1293  DCHECK_GT(time_scale, 0u);
1294 
1295  uint64_t time_shift_buffer_depth =
1296  static_cast<uint64_t>(mpd_options_.time_shift_buffer_depth * time_scale);
1297 
1298  // The start time of the latest segment is considered the current_play_time,
1299  // and this should guarantee that the latest segment will stay in the list.
1300  const uint64_t current_play_time = LatestSegmentStartTime(segment_infos_);
1301  if (current_play_time <= time_shift_buffer_depth)
1302  return;
1303 
1304  const uint64_t timeshift_limit = current_play_time - time_shift_buffer_depth;
1305 
1306  // First remove all the SegmentInfos that are completely out of range, by
1307  // looking at the very last segment's end time.
1308  std::list<SegmentInfo>::iterator first = segment_infos_.begin();
1309  std::list<SegmentInfo>::iterator last = first;
1310  size_t num_segments_removed = 0;
1311  for (; last != segment_infos_.end(); ++last) {
1312  const uint64_t last_segment_end_time = LastSegmentEndTime(*last);
1313  if (timeshift_limit < last_segment_end_time)
1314  break;
1315  num_segments_removed += last->repeat + 1;
1316  }
1317  segment_infos_.erase(first, last);
1318  start_number_ += num_segments_removed;
1319 
1320  // Now some segment in the first SegmentInfo should be left in the list.
1321  SegmentInfo* first_segment_info = &segment_infos_.front();
1322  DCHECK_LE(timeshift_limit, LastSegmentEndTime(*first_segment_info));
1323 
1324  // Identify which segments should still be in the SegmentInfo.
1325  const int repeat_index =
1326  SearchTimedOutRepeatIndex(timeshift_limit, *first_segment_info);
1327  CHECK_GE(repeat_index, 0);
1328  if (repeat_index == 0)
1329  return;
1330 
1331  first_segment_info->start_time = first_segment_info->start_time +
1332  first_segment_info->duration * repeat_index;
1333 
1334  first_segment_info->repeat = first_segment_info->repeat - repeat_index;
1335  start_number_ += repeat_index;
1336 }
1337 
1338 std::string Representation::GetVideoMimeType() const {
1339  return GetMimeType("video", media_info_.container_type());
1340 }
1341 
1342 std::string Representation::GetAudioMimeType() const {
1343  return GetMimeType("audio", media_info_.container_type());
1344 }
1345 
1346 std::string Representation::GetTextMimeType() const {
1347  CHECK(media_info_.has_text_info());
1348  if (media_info_.text_info().format() == "ttml") {
1349  switch (media_info_.container_type()) {
1350  case MediaInfo::CONTAINER_TEXT:
1351  return "application/ttml+xml";
1352  case MediaInfo::CONTAINER_MP4:
1353  return "application/mp4";
1354  default:
1355  LOG(ERROR) << "Failed to determine MIME type for TTML container: "
1356  << media_info_.container_type();
1357  return "";
1358  }
1359  }
1360  if (media_info_.text_info().format() == "vtt") {
1361  if (media_info_.container_type() == MediaInfo::CONTAINER_TEXT) {
1362  return "text/vtt";
1363  }
1364  LOG(ERROR) << "Failed to determine MIME type for VTT container: "
1365  << media_info_.container_type();
1366  return "";
1367  }
1368 
1369  LOG(ERROR) << "Cannot determine MIME type for format: "
1370  << media_info_.text_info().format()
1371  << " container: " << media_info_.container_type();
1372  return "";
1373 }
1374 
1375 bool Representation::GetEarliestTimestamp(double* timestamp_seconds) {
1376  DCHECK(timestamp_seconds);
1377 
1378  if (segment_infos_.empty())
1379  return false;
1380 
1381  *timestamp_seconds = static_cast<double>(segment_infos_.begin()->start_time) /
1382  GetTimeScale(media_info_);
1383  return true;
1384 }
1385 
1386 } // namespace shaka
void OnSetFrameRateForRepresentation(uint32_t representation_id, uint32_t frame_duration, uint32_t timescale)
Definition: mpd_builder.cc:869
virtual void AddNewSegment(uint64_t start_time, uint64_t duration, uint64_t size)
bool WriteMpdToFile(media::File *output_file)
Definition: mpd_builder.cc:424
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:675
virtual void SetSampleDuration(uint32_t sample_duration)
virtual Representation * AddRepresentation(const MediaInfo &media_info)
Definition: mpd_builder.cc:694
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:738
Representation(const MediaInfo &media_info, const MpdOptions &mpd_options, uint32_t representation_id, scoped_ptr< RepresentationStateChangeListener > state_change_listener)
virtual void AddRole(Role role)
Definition: mpd_builder.cc:750
void AddBaseUrl(const std::string &base_url)
Definition: mpd_builder.cc:410
virtual void UpdateContentProtectionPssh(const std::string &drm_uuid, const std::string &pssh)
Definition: mpd_builder.cc:744
virtual void SetGroup(int group_number)
Definition: mpd_builder.cc:843
xml::scoped_xml_ptr< xmlNode > GetXml()
virtual bool ToString(std::string *output)
Definition: mpd_builder.cc:429
virtual void ForceSetSegmentAlignment(bool segment_alignment)
Definition: mpd_builder.cc:837
static void MakePathsRelativeToMpd(const std::string &mpd_path, MediaInfo *media_info)
Definition: mpd_builder.cc:647
MpdBuilder(MpdType type, const MpdOptions &mpd_options)
Definition: mpd_builder.cc:402
xml::scoped_xml_ptr< xmlNode > GetXml()
Definition: mpd_builder.cc:760
virtual void AddContentProtectionElement(const ContentProtectionElement &element)
virtual AdaptationSet * AddAdaptationSet(const std::string &lang)
Definition: mpd_builder.cc:414
virtual int Group() const
Definition: mpd_builder.cc:847
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:858
virtual void UpdateContentProtectionPssh(const std::string &drm_uuid, const std::string &pssh)
void SuppressOnce(SuppressFlag flag)