gecko-dev/netwerk/dash/mpd/nsDASHWebMODManager.cpp

/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=2 et sw=2 tw=80: */
/*
 * nsDASHWebMODManager.cpp
 *****************************************************************************
 * Copyrigh(C) 2010 - 2012 Klagenfurt University
 *
 * Created on: May 1, 2012
 * Based on IsoffMainManager.cpp by:
 *          Christopher Mueller <christopher.mueller@itec.uni-klu.ac.at>
 *          Christian Timmerer  <christian.timmerer@itec.uni-klu.ac.at>
 * Author:
 *          Steve Workman <sworkman@mozilla.com>
 *
 * This Source Code Form Is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this file,
 * You can obtain one at http://mozilla.org/MPL/2.0/.
 *****************************************************************************/

/* DASH - Dynamic Adaptive Streaming over HTTP.
 *
 * DASH is an adaptive bitrate streaming technology where a multimedia file is
 * partitioned into one or more segments and delivered to a client using HTTP.
 *
 * (see DASHDecoder.cpp for info on DASH interaction with the media engine).
 *
 * Media Presentation Description (MPD) Manager for WebM On Demand Profile.
 *
 * Implements MPD Manager interface to use Adaptation Algorithm to determine
 * which stream to download from WebM On Demand-based MPD.
 *
 *   Note: Adaptation algorithm is separate and passed into manager.
 */

#include "nsTArray.h"
#include "nsMimeTypes.h"
#include "nsNetUtil.h"
#include "prlog.h"
#include "SegmentBase.h"
#include "nsDASHWebMODManager.h"

namespace mozilla {
namespace net {

#if defined(PR_LOGGING)
static PRLogModuleInfo* gnsDASHWebMODManagerLog = nullptr;
#define LOG(msg, ...) \
        PR_LOG(gnsDASHWebMODManagerLog, PR_LOG_DEBUG, \
               ("%p [nsDASHWebMODManager] " msg, this, __VA_ARGS__))
#define LOG1(msg) PR_LOG(gDASHMPDParserLog, PR_LOG_DEBUG, \
                         ("%p [nsDASHWebMODManager] " msg, this))
#else
#define LOG(msg, ...)
#define LOG1(msg)
#endif

nsDASHWebMODManager::nsDASHWebMODManager(MPD* aMpd)
{
  MOZ_COUNT_CTOR(nsDASHWebMODManager);
  NS_ENSURE_TRUE_VOID(aMpd);
  mMpd = aMpd;
#if defined(PR_LOGGING)
  if(!gnsDASHWebMODManagerLog)
    gnsDASHWebMODManagerLog = PR_NewLogModule("nsDASHWebMODManager");
#endif
  LOG("Created nsDASHWebMODManager with mMpd(%p)", mMpd.get());
}

nsDASHWebMODManager::~nsDASHWebMODManager()
{
  MOZ_COUNT_DTOR(nsDASHWebMODManager);
}


Period const *
nsDASHWebMODManager::GetFirstPeriod() const
{
  NS_ENSURE_TRUE(mMpd, nullptr);
  NS_ENSURE_TRUE(0 < mMpd->GetNumPeriods(), nullptr);
  return mMpd->GetPeriod(0);
}

nsresult
nsDASHWebMODManager::GetFirstSegmentUrl(uint32_t const aAdaptSetIdx,
                                        uint32_t const aRepIdx,
                                        nsAString &aUrl) const
{
  NS_ENSURE_TRUE(mMpd, NS_ERROR_NULL_POINTER);
  // Append base URL for MPD.
  if (mMpd->HasBaseUrls()) {
    aUrl.Append(mMpd->GetBaseUrl(0));
    LOG("BaseUrl \"%s\"", NS_ConvertUTF16toUTF8(aUrl).get());
  }

  // Append base URL for Representation.
  AdaptationSet const * adaptSet = GetAdaptationSet(aAdaptSetIdx);
  NS_ENSURE_TRUE(adaptSet, NS_ERROR_NULL_POINTER);

  NS_ENSURE_TRUE(aRepIdx < adaptSet->GetNumRepresentations(),
                 NS_ERROR_ILLEGAL_VALUE);
  Representation const * rep = adaptSet->GetRepresentation(aRepIdx);
  NS_ENSURE_TRUE(rep, NS_ERROR_NULL_POINTER);

  if (rep->HasBaseUrls()) {
    aUrl.Append(rep->GetBaseUrl(0));
    LOG("Appending \"%s\"",
        NS_ConvertUTF16toUTF8(rep->GetBaseUrl(0)).get());
  }

  return NS_OK;
}

uint32_t
nsDASHWebMODManager::GetNumAdaptationSets() const
{
  Period const * current = GetFirstPeriod();
  return current ? current->GetNumAdaptationSets() : 0;
}

IMPDManager::AdaptationSetType
nsDASHWebMODManager::GetAdaptationSetType(uint32_t const aAdaptSetIdx) const
{
  AdaptationSet const * adaptSet = GetAdaptationSet(aAdaptSetIdx);
  NS_ENSURE_TRUE(adaptSet, DASH_ASTYPE_INVALID);

  nsAutoString mimeType;
  adaptSet->GetMIMEType(mimeType);
  NS_ENSURE_TRUE(!mimeType.IsEmpty(), DASH_ASTYPE_INVALID);

  return GetAdaptationSetType(mimeType);
}

IMPDManager::AdaptationSetType
nsDASHWebMODManager::GetAdaptationSetType(nsAString const & aMimeType) const
{
  NS_ENSURE_TRUE(!aMimeType.IsEmpty(), DASH_ASTYPE_INVALID);

  if (aMimeType == NS_LITERAL_STRING(VIDEO_WEBM)) {
    return DASH_VIDEO_STREAM;
  } else if (aMimeType == NS_LITERAL_STRING(AUDIO_WEBM)) {
    return DASH_AUDIO_STREAM;
  } else {
    return DASH_ASTYPE_INVALID;
  }
}

uint32_t
nsDASHWebMODManager::GetNumRepresentations(uint32_t const aAdaptSetIdx) const
{
  AdaptationSet const * adaptSet = GetAdaptationSet(aAdaptSetIdx);
  NS_ENSURE_TRUE(adaptSet, DASH_ASTYPE_INVALID);

  return adaptSet->GetNumRepresentations();
}

AdaptationSet const *
nsDASHWebMODManager::GetAdaptationSet(uint32_t const aAdaptSetIdx) const
{
  Period const * current = GetFirstPeriod();
  NS_ENSURE_TRUE(current, nullptr);

  NS_ENSURE_TRUE(0 < current->GetNumAdaptationSets(), nullptr);
  NS_ENSURE_TRUE(aAdaptSetIdx < current->GetNumAdaptationSets(), nullptr);
  AdaptationSet const * adaptSet = current->GetAdaptationSet(aAdaptSetIdx);
  NS_ENSURE_TRUE(adaptSet, nullptr);
  return adaptSet;
}

Representation const *
nsDASHWebMODManager::GetRepresentation(uint32_t const aAdaptSetIdx,
                                       uint32_t const aRepIdx) const
{
  AdaptationSet const * adaptSet = GetAdaptationSet(aAdaptSetIdx);
  NS_ENSURE_TRUE(adaptSet, nullptr);

  NS_ENSURE_TRUE(aRepIdx < adaptSet->GetNumRepresentations(), nullptr);
  Representation const * rep = adaptSet->GetRepresentation(aRepIdx);
  NS_ENSURE_TRUE(rep, nullptr);

  return rep;
}

double
nsDASHWebMODManager::GetStartTime() const
{
  Period const * current = GetFirstPeriod();
  NS_ENSURE_TRUE(current, -1);

  return current->GetStart();
}

double
nsDASHWebMODManager::GetDuration() const
{
  Period const * current = GetFirstPeriod();
  NS_ENSURE_TRUE(current, -1);

  return current->GetDuration();
}

bool
nsDASHWebMODManager::GetBestRepForBandwidth(uint32_t aAdaptSetIdx,
                                            uint64_t aBandwidth,
                                            uint32_t &aRepIdx) const
{
  NS_ENSURE_TRUE(aAdaptSetIdx < GetNumAdaptationSets(), false);
  NS_ENSURE_TRUE(0 < GetNumRepresentations(aAdaptSetIdx), false);
  // Return false if there isn't enough bandwidth for even the lowest bitrate.
  // Let calling function decide what to do. Use 0.95 multiplier to deal with
  // 5% variance in bandwidth.
  // XXX Multiplier is a guess at present.
  if (aBandwidth*0.95 < GetRepresentation(aAdaptSetIdx, 0)->GetBitrate()) {
    aRepIdx = UINT32_MAX;
    return false;
  }
  // Iterate until the current |Representation|'s bitrate is higher than the
  // estimated available bandwidth. Use 0.95 multiplier to deal with 5%
  // variance in bandwidth.
  // XXX Multiplier is a guess at present.
  for (uint32_t i = 1; i < GetNumRepresentations(aAdaptSetIdx); i++) {
    NS_ENSURE_TRUE(GetRepresentation(aAdaptSetIdx, i), false);
    if (aBandwidth*0.95 < GetRepresentation(aAdaptSetIdx, i)->GetBitrate()) {
      // Pick the previous one, since this one's bitrate is too high.
      aRepIdx = i-1;
      return true;
    }
  }
  // If we reach here, all of the |Representation|'s bitrates are lower than the
  // available bandwidth. Just pick the highest, i.e. last in the array.
  aRepIdx = GetNumRepresentations(aAdaptSetIdx)-1;
  return true;
}

}//namespace net
}//namespace mozilla