1 files changed, 158 insertions, 0 deletions
diff --git a/MPD/SegmentTemplateStream.cpp b/MPD/SegmentTemplateStream.cpp
new file mode 100644
index 00000000..f0f475e6
--- /dev/null
+++ b/MPD/SegmentTemplateStream.cpp
@@ -0,0 +1,158 @@
+/*
+ * SegmentTemplateStream.cpp
+ *****************************************************************************
+ * Copyright (C) 2012, bitmovin Softwareentwicklung OG, All Rights Reserved
+ *
+ * Email: libdash-dev@vicky.bitmovin.net
+ *
+ * This source code and its use and distribution, is subject to the terms
+ * and conditions of the applicable license agreement.
+ *****************************************************************************/
+
+#include "SegmentTemplateStream.h"
+
+using namespace dash::mpd;
+using namespace libdash::framework::mpd;
+
+SegmentTemplateStream::SegmentTemplateStream(IMPD *mpd, IPeriod *period, IAdaptationSet *adaptationSet, IRepresentation *representation) :
+    AbstractRepresentationStream     (mpd, period, adaptationSet, representation)
+{
+    this->baseUrls          = BaseUrlResolver::resolveBaseUrl(mpd, period, adaptationSet, 0, 0, 0);
+    this->segmentTemplate   = findSegmentTemplate();
+    calculateSegmentStartTimes();
+}
+SegmentTemplateStream::~SegmentTemplateStream()
+{
+}
+
+ISegment* SegmentTemplateStream::getInitializationSegment()
+{
+    if (this->segmentTemplate->GetInitialization())
+        return this->segmentTemplate->GetInitialization()->ToSegment(baseUrls);
+
+    return this->segmentTemplate->ToInitializationSegment(baseUrls, representation->GetId(), representation->GetBandwidth());
+}
+
+ISegment* SegmentTemplateStream::getIndexSegment(size_t segmentNumber)
+{
+    /* time-based template */
+    if (this->segmentTemplate->GetSegmentTimeline())
+    {
+        if (this->segmentStartTimes.size() > segmentNumber)
+            return this->segmentTemplate->GetIndexSegmentFromTime(baseUrls, representation->GetId(), representation->GetBandwidth(), this->segmentStartTimes.at(segmentNumber));
+
+        return NULL;
+    }
+
+    /* number-based template */
+    return this->segmentTemplate->GetIndexSegmentFromNumber(baseUrls, representation->GetId(), representation->GetBandwidth(),
+                                                            this->segmentTemplate->GetStartNumber() + segmentNumber);
+}
+
+ISegment* SegmentTemplateStream::getMediaSegment(size_t segmentNumber)
+{
+    /* time-based template */
+    if (this->segmentTemplate->GetSegmentTimeline())
+    {
+        if (this->segmentStartTimes.size() > segmentNumber)
+            return this->segmentTemplate->GetMediaSegmentFromTime(baseUrls, representation->GetId(), representation->GetBandwidth(), this->segmentStartTimes.at(segmentNumber));
+
+        return NULL;
+    }
+
+    /* number-based template */
+    return this->segmentTemplate->GetMediaSegmentFromNumber(baseUrls, representation->GetId(), representation->GetBandwidth(),
+                                                            this->segmentTemplate->GetStartNumber() + segmentNumber);
+
+}
+
+ISegment* SegmentTemplateStream::getBitstreamSwitchingSegment ()
+{
+    if (this->segmentTemplate->GetBitstreamSwitching())
+        return this->segmentTemplate->GetBitstreamSwitching()->ToSegment(baseUrls);
+    return this->segmentTemplate->ToBitstreamSwitchingSegment(baseUrls, representation->GetId(), representation->GetBandwidth());
+}
+
+RepresentationStreamType SegmentTemplateStream::getStreamType()
+{
+    return SegmentList;
+}
+
+uint32_t SegmentTemplateStream::getSize()
+{
+    if (!this->segmentStartTimes.empty())
+        return this->segmentStartTimes.size();
+
+    uint32_t numberOfSegments          = 0;
+    double   mediaPresentationDuration = 0;
+
+    if (this->mpd->GetType() == "static")
+    {
+        mediaPresentationDuration = TimeResolver::getDurationInSec(this->mpd->GetMediaPresentationDuration());
+        numberOfSegments = (uint32_t) ceil(mediaPresentationDuration / (this->segmentTemplate->GetDuration() / this->segmentTemplate->GetTimescale()));
+    }
+    else
+    {
+        numberOfSegments = UINT32_MAX - 1;
+    }
+    return numberOfSegments;
+}
+
+uint32_t SegmentTemplateStream::getAverageSegmentDuration()
+{
+    /* TODO calculate average segment durations for SegmentTimeline */
+    return this->segmentTemplate->GetDuration();
+}
+
+ISegmentTemplate* SegmentTemplateStream::findSegmentTemplate()
+{
+    if (this->representation->GetSegmentTemplate())
+        return this->representation->GetSegmentTemplate();
+
+    if (this->adaptationSet->GetSegmentTemplate())
+        return this->adaptationSet->GetSegmentTemplate();
+
+    if (this->period->GetSegmentTemplate())
+        return this->period->GetSegmentTemplate();
+
+    return NULL;
+}
+
+void SegmentTemplateStream::calculateSegmentStartTimes()
+{
+    if (!this->segmentTemplate->GetSegmentTimeline())
+        return;
+
+    size_t   numOfTimelines = 0;
+    uint32_t segStartTime   = 0;
+    uint32_t segDuration    = 0;
+    size_t   repeatCount    = 0;
+
+    numOfTimelines      = this->segmentTemplate->GetSegmentTimeline()->GetTimelines().size();
+
+    for (size_t i = 0; i < numOfTimelines; i++)
+    {
+        repeatCount     = this->segmentTemplate->GetSegmentTimeline()->GetTimelines().at(i)->GetRepeatCount();
+        segStartTime    = this->segmentTemplate->GetSegmentTimeline()->GetTimelines().at(i)->GetStartTime();
+        segDuration     = this->segmentTemplate->GetSegmentTimeline()->GetTimelines().at(i)->GetDuration();
+
+        if (repeatCount > 0)
+        {
+            for (size_t j = 0; j <= repeatCount; j++)
+            {
+                if (segStartTime > 0)
+                {
+                    this->segmentStartTimes.push_back(segStartTime + segDuration * j);
+                }
+                else
+                {
+                    /* TODO: implement if SegmentTemplate.SegmentTimeline.S@t is not specified */
+                }
+            }
+        }
+        else
+        {
+            this->segmentStartTimes.push_back(segStartTime);
+        }
+    }
+}