demux: dash: add MPD merging for updates
[vlc.git] / modules / demux / dash / mpd / MPD.cpp
1 /*
2  * MPD.cpp
3  *****************************************************************************
4  * Copyright (C) 2010 - 2011 Klagenfurt University
5  *
6  * Created on: Aug 10, 2010
7  * Authors: Christopher Mueller <christopher.mueller@itec.uni-klu.ac.at>
8  *          Christian Timmerer  <christian.timmerer@itec.uni-klu.ac.at>
9  *
10  * This program is free software; you can redistribute it and/or modify it
11  * under the terms of the GNU Lesser General Public License as published
12  * by the Free Software Foundation; either version 2.1 of the License, or
13  * (at your option) any later version.
14  *
15  * This program is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18  * GNU Lesser General Public License for more details.
19  *
20  * You should have received a copy of the GNU Lesser General Public License
21  * along with this program; if not, write to the Free Software Foundation,
22  * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
23  *****************************************************************************/
24 #ifdef HAVE_CONFIG_H
25 # include "config.h"
26 #endif
27
28 #include "MPD.h"
29 #include "Helper.h"
30 #include "dash.hpp"
31 #include "SegmentTimeline.h"
32 #include <vlc_common.h>
33 #include <vlc_stream.h>
34
35 using namespace dash::mpd;
36
37 MPD::MPD (stream_t *stream_, Profile profile_) :
38     ICanonicalUrl(),
39     stream(stream_),
40     profile( profile_ )
41 {
42     playbackStart.Set(0);
43     availabilityStartTime.Set( 0 );
44     availabilityEndTime.Set( 0 );
45     duration.Set( 0 );
46     minUpdatePeriod.Set( 0 );
47     minBufferTime.Set( 0 );
48     timeShiftBufferDepth.Set( 0 );
49     programInfo.Set( NULL );
50 }
51
52 MPD::~MPD   ()
53 {
54     for(size_t i = 0; i < this->periods.size(); i++)
55         delete(this->periods.at(i));
56
57     for(size_t i = 0; i < this->baseUrls.size(); i++)
58         delete(this->baseUrls.at(i));
59
60     delete(programInfo.Get());
61 }
62
63 const std::vector<Period*>&    MPD::getPeriods             ()
64 {
65     return this->periods;
66 }
67
68 void                    MPD::addBaseUrl             (BaseUrl *url)
69 {
70     this->baseUrls.push_back(url);
71 }
72 void                    MPD::addPeriod              (Period *period)
73 {
74     this->periods.push_back(period);
75 }
76
77 bool                    MPD::isLive() const
78 {
79     if(type.empty())
80     {
81         Profile live(Profile::ISOLive);
82         return profile == live;
83     }
84     else
85         return (type != "static");
86 }
87
88 void MPD::setType(const std::string &type_)
89 {
90     type = type_;
91 }
92
93 Profile MPD::getProfile() const
94 {
95     return profile;
96 }
97 Url MPD::getUrlSegment() const
98 {
99     if (!baseUrls.empty())
100         return Url(baseUrls.front()->getUrl());
101     else
102     {
103         std::stringstream ss;
104         ss << stream->psz_access << "://" << Helper::getDirectoryPath(stream->psz_path) << "/";
105         return Url(ss.str());
106     }
107 }
108
109 vlc_object_t * MPD::getVLCObject() const
110 {
111     return VLC_OBJECT(stream);
112 }
113
114 Period* MPD::getFirstPeriod()
115 {
116     std::vector<Period *> periods = getPeriods();
117
118     if( !periods.empty() )
119         return periods.front();
120     else
121         return NULL;
122 }
123
124 Period* MPD::getNextPeriod(Period *period)
125 {
126     std::vector<Period *> periods = getPeriods();
127
128     for(size_t i = 0; i < periods.size(); i++)
129     {
130         if(periods.at(i) == period && (i + 1) < periods.size())
131             return periods.at(i + 1);
132     }
133
134     return NULL;
135 }
136
137 void MPD::mergeWith(MPD *updatedMPD)
138 {
139     availabilityEndTime.Set(updatedMPD->availabilityEndTime.Get());
140     /* Only merge timelines for now */
141     for(size_t i = 0; i < periods.size() && i < updatedMPD->periods.size(); i++)
142     {
143         std::vector<SegmentTimeline *> timelines;
144         std::vector<SegmentTimeline *> timelinesUpdate;
145         periods.at(i)->collectTimelines(&timelines);
146         updatedMPD->periods.at(i)->collectTimelines(&timelinesUpdate);
147
148         for(size_t j = 0; j < timelines.size() && j < timelinesUpdate.size(); j++)
149             timelines.at(j)->mergeWith(*timelinesUpdate.at(j));
150     }
151 }