shaka-player/lib/dash/segment_base.js

/**
 * @license
 * Copyright 2015 Google Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

goog.provide('shaka.dash.SegmentBase');

goog.require('shaka.asserts');
goog.require('shaka.dash.MpdUtils');
goog.require('shaka.log');
goog.require('shaka.media.InitSegmentReference');
goog.require('shaka.media.Mp4SegmentIndexParser');
goog.require('shaka.media.SegmentIndex');
goog.require('shaka.media.WebmSegmentIndexParser');
goog.require('shaka.util.Error');
goog.require('shaka.util.XmlUtils');


/**
 * Creates an init segment reference from a Context object.
 *
 * @param {shaka.dash.DashParser.Context} context
 * @param {string} type One of segmentBase, segmentList, or segmentTemplate
 *     which tells the source of the init segment.
 * @return {shaka.media.InitSegmentReference}
 */
shaka.dash.SegmentBase.createInitSegment = function(context, type) {
  var MpdUtils = shaka.dash.MpdUtils;
  var XmlUtils = shaka.util.XmlUtils;
  var SegmentBase = shaka.dash.SegmentBase;

  var initialization =
      MpdUtils.inheritChild(context, type, 'Initialization');
  if (!initialization)
    return null;

  var uri = initialization.getAttribute('sourceURL');
  if (!uri)
    return null;
  var startByte = 0;
  var endByte = null;
  var range = XmlUtils.parseAttr(initialization, 'range', XmlUtils.parseRange);
  if (range) {
    startByte = range.start;
    endByte = range.end;
  }

  var resolvedUris =
      MpdUtils.resolveUris(context.representation.baseUris, [uri]);
  return new shaka.media.InitSegmentReference(resolvedUris, startByte, endByte);
};


/**
 * Creates a new Stream object.
 *
 * @param {shaka.dash.DashParser.Context} context
 * @param {shaka.dash.DashParser.RequestInitSegmentCallback} requestInitSegment
 * @param {!Object.<string, !shaka.media.SegmentIndex>} segmentIndexMap
 * @throws shaka.util.Error When there is a parsing error.
 * @return {shaka.dash.DashParser.StreamInfo}
 */
shaka.dash.SegmentBase.createStream = function(
    context, requestInitSegment, segmentIndexMap) {
  shaka.asserts.assert(context.representation.segmentBase,
                       'Should only be called with SegmentBase');
  // Since SegmentBase does not need updates, simply treat any call as
  // the initial parse.
  var MpdUtils = shaka.dash.MpdUtils;
  var SegmentBase = shaka.dash.SegmentBase;

  var presentationTimeOffset = MpdUtils.inheritAttribute(
      context, 'segmentBase', 'presentationTimeOffset');

  var init = SegmentBase.createInitSegment(context, 'segmentBase');
  var index =
      SegmentBase.createSegmentIndex_(context, requestInitSegment, init);

  return {
    createSegmentIndex: index.createSegmentIndex,
    findSegmentPosition: index.findSegmentPosition,
    getSegmentReference: index.getSegmentReference,
    initSegmentReference: init,
    presentationTimeOffset: Number(presentationTimeOffset) || 0
  };
};


/**
 * Creates segment index info for the given info.
 *
 * @param {shaka.dash.DashParser.RequestInitSegmentCallback} requestInitSegment
 * @param {shaka.media.InitSegmentReference} init
 * @param {!Array.<string>} uris
 * @param {number} startByte
 * @param {?number} endByte
 * @param {string} containerType
 * @return {shaka.dash.DashParser.SegmentIndexFunctions}
 */
shaka.dash.SegmentBase.createSegmentIndexFromUris = function(
    requestInitSegment, init, uris, startByte, endByte, containerType) {
  // Create a local variable to bind to so we can set to null to help the GC.
  var localRequest = requestInitSegment;
  var segmentIndex = null;
  var create = function() {
    var async = [
      localRequest(uris, startByte, endByte),
      containerType == 'webm' ?
          localRequest(init.uris, init.startByte, init.endByte) :
          null
    ];

    localRequest = null;
    return Promise.all(async).then(function(results) {
      var indexData = results[0];
      var initData = results[1] || null;
      var references = null;

      if (containerType == 'mp4') {
        var parser = new shaka.media.Mp4SegmentIndexParser();
        references = parser.parse(indexData, startByte, uris);
      } else {
        shaka.asserts.assert(initData, 'WebM requires init data');
        var parser = new shaka.media.WebmSegmentIndexParser();
        references = parser.parse(indexData, initData, uris);
      }

      // Since containers are never updated, we don't need to store the
      // segmentIndex in the map.
      shaka.asserts.assert(!segmentIndex,
                           'Should not call createSegmentIndex twice');
      segmentIndex = new shaka.media.SegmentIndex(references);
    });
  };
  var get = function(i) {
    shaka.asserts.assert(segmentIndex, 'Must call createSegmentIndex first');
    return segmentIndex.get(i);
  };
  var find = function(t) {
    shaka.asserts.assert(segmentIndex, 'Must call createSegmentIndex first');
    return segmentIndex.find(t);
  };

  return {
    createSegmentIndex: create,
    findSegmentPosition: find,
    getSegmentReference: get
  };
};


/**
 * Creates segment index info from a Context object.
 *
 * @param {shaka.dash.DashParser.Context} context
 * @param {shaka.dash.DashParser.RequestInitSegmentCallback} requestInitSegment
 * @param {shaka.media.InitSegmentReference} init
 * @return {shaka.dash.DashParser.SegmentIndexFunctions}
 * @throws shaka.util.Error When there is a parsing error.
 * @private
 */
shaka.dash.SegmentBase.createSegmentIndex_ = function(
    context, requestInitSegment, init) {
  var MpdUtils = shaka.dash.MpdUtils;
  var XmlUtils = shaka.util.XmlUtils;

  var contentType = context.representation.contentType;
  var containerType = context.representation.mimeType.split('/')[1];
  if (contentType != 'text' && containerType != 'mp4' &&
      containerType != 'webm') {
    shaka.log.error(
        'SegmentBase specifies an unsupported container type.',
        context.representation);
    throw new shaka.util.Error(
        shaka.util.Error.Category.MANIFEST,
        shaka.util.Error.Code.DASH_UNSUPPORTED_CONTAINER);
  }

  if ((containerType == 'webm') && !init) {
    shaka.log.error(
        'SegmentBase does not contain sufficient segment information:',
        'the SegmentBase uses a WebM container,',
        'but does not contain an Initialization element.',
        context.representation);
    throw new shaka.util.Error(
        shaka.util.Error.Category.MANIFEST,
        shaka.util.Error.Code.DASH_WEBM_MISSING_INIT);
  }

  var representationIndex =
      MpdUtils.inheritChild(context, 'segmentBase', 'RepresentationIndex');
  var indexRangeElem =
      MpdUtils.inheritAttribute(context, 'segmentBase', 'indexRange');

  var indexUris = context.representation.baseUris;
  var indexRange = XmlUtils.parseRange(indexRangeElem || '');
  if (representationIndex) {
    var representationUri = representationIndex.getAttribute('sourceURL');
    if (representationUri) {
      indexUris = MpdUtils.resolveUris(
          context.representation.baseUris, [representationUri]);
    }

    indexRange = XmlUtils.parseAttr(
        representationIndex, 'range', XmlUtils.parseRange, indexRange);
  }

  if (!indexRange) {
    shaka.log.error(
        'SegmentBase does not contain sufficient segment information:',
        'the SegmentBase does not contain @indexRange',
        'or a RepresentationIndex element.',
        context.representation);
    throw new shaka.util.Error(
        shaka.util.Error.Category.MANIFEST,
        shaka.util.Error.Code.DASH_NO_SEGMENT_INFO);
  }

  return shaka.dash.SegmentBase.createSegmentIndexFromUris(
      requestInitSegment, init, indexUris, indexRange.start, indexRange.end,
      containerType);
};