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) : AbstractAdaptationLogic(mpdManager)
  37 {
  38     this->mpdManager    = mpdManager;
  39     this->currentPeriod = this->mpdManager->getFirstPeriod();
  40     this->count         = 0;
  41 }
  42 RateBasedAdaptationLogic::~RateBasedAdaptationLogic ()
  43 {
  44 }
  45
  46 Chunk*  RateBasedAdaptationLogic::getNextChunk () throw(EOFException)
  47 {
  48     if(this->mpdManager == NULL)
  49         throw EOFException();
  50
  51     if(this->currentPeriod == NULL)
  52         throw EOFException();
  53
  54     long bitrate = this->getBpsAvg();
  55
  56     Representation *rep = this->mpdManager->getRepresentation(this->currentPeriod, bitrate);
  57
  58     if(rep == NULL)
  59         throw EOFException();
  60
  61     std::vector<ISegment *> segments = this->mpdManager->getSegments(rep);
  62
  63     if(this->count == segments.size())
  64     {
  65         this->currentPeriod = this->mpdManager->getNextPeriod(this->currentPeriod);
  66         this->count = 0;
  67         return this->getNextChunk();
  68     }
  69
  70     for(size_t i = 0; i < segments.size(); i++)
  71     {
  72         if(i == this->count)
  73         {
  74             this->count++;
  75             Chunk *chunk = new Chunk();
  76             chunk->setUrl(segments.at(i)->getSourceUrl());
  77             return chunk;
  78         }
  79     }
  80     return NULL;
  81 }