git.sesse.net Git - vlc/blob - modules/stream_filter/dash/adaptationlogic/RateBasedAdaptationLogic.cpp

   1 /*
   2  * RateBasedAdaptationLogic.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
  11  * it 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 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
  22  * Foundation, 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 "RateBasedAdaptationLogic.h"
  29
  30 using namespace dash::logic;
  31 using namespace dash::xml;
  32 using namespace dash::http;
  33 using namespace dash::mpd;
  34 using namespace dash::exception;
  35
  36 RateBasedAdaptationLogic::RateBasedAdaptationLogic  (IMPDManager *mpdManager, stream_t *stream) :
  37                           AbstractAdaptationLogic   (mpdManager, stream),
  38                           mpdManager                (mpdManager),
  39                           count                     (0),
  40                           currentPeriod             (mpdManager->getFirstPeriod()),
  41                           width                     (0),
  42                           height                    (0)
  43 {
  44     this->width  = var_InheritInteger(stream, "dash-prefwidth");
  45     this->height = var_InheritInteger(stream, "dash-prefheight");
  46 }
  47
  48 Chunk*  RateBasedAdaptationLogic::getNextChunk() throw(EOFException)
  49 {
  50     if(this->mpdManager == NULL)
  51         throw EOFException();
  52
  53     if(this->currentPeriod == NULL)
  54         throw EOFException();
  55
  56     long bitrate = this->getBpsAvg();
  57
  58     Representation *rep = this->mpdManager->getRepresentation(this->currentPeriod, bitrate, this->width, this->height);
  59
  60     if ( rep == NULL )
  61         throw EOFException();
  62
  63     std::vector<Segment *> segments = this->mpdManager->getSegments(rep);
  64
  65     if ( this->count == segments.size() )
  66     {
  67         this->currentPeriod = this->mpdManager->getNextPeriod(this->currentPeriod);
  68         this->count = 0;
  69         return this->getNextChunk();
  70     }
  71
  72     if ( segments.size() > this->count )
  73     {
  74         Segment *seg = segments.at( this->count );
  75         Chunk *chunk = seg->toChunk();
  76         //In case of UrlTemplate, we must stay on the same segment.
  77         if ( seg->isSingleShot() == true )
  78             this->count++;
  79         seg->done();
  80         return chunk;
  81     }
  82     return NULL;
  83 }
  84
  85 const Representation *RateBasedAdaptationLogic::getCurrentRepresentation() const
  86 {
  87     return this->mpdManager->getRepresentation( this->currentPeriod, this->getBpsAvg() );
  88 }