Source: lib/abr/ewma.js

  1. /**
  2.  * @license
  3.  * Copyright 2016 Google Inc.
  4.  *
  5.  * Licensed under the Apache License, Version 2.0 (the "License");
  6.  * you may not use this file except in compliance with the License.
  7.  * You may obtain a copy of the License at
  8.  *
  9.  *     http://www.apache.org/licenses/LICENSE-2.0
  10.  *
  11.  * Unless required by applicable law or agreed to in writing, software
  12.  * distributed under the License is distributed on an "AS IS" BASIS,
  13.  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14.  * See the License for the specific language governing permissions and
  15.  * limitations under the License.
  16.  */
  18. goog.provide('shaka.abr.Ewma');
  20. goog.require('goog.asserts');
  24. /**
  25.  * Computes an exponentionally-weighted moving average.
  26.  *
  27.  * @param {number} halfLife The quantity of prior samples (by weight) used
  28.  *   when creating a new estimate.  Those prior samples make up half of the new
  29.  *   estimate.
  30.  * @struct
  31.  * @constructor
  32.  */
  33. shaka.abr.Ewma = function(halfLife) {
  34.   goog.asserts.assert(halfLife > 0, 'expected halfLife to be positive');
  36.   /**
  37.    * Larger values of alpha expire historical data more slowly.
  38.    * @private {number}
  39.    */
  40.   this.alpha_ = Math.exp(Math.log(0.5) / halfLife);
  42.   /** @private {number} */
  43.   this.estimate_ = 0;
  45.   /** @private {number} */
  46.   this.totalWeight_ = 0;
  47. };
  50. /**
  51.  * Takes a sample.
  52.  *
  53.  * @param {number} weight
  54.  * @param {number} value
  55.  */
  56. shaka.abr.Ewma.prototype.sample = function(weight, value) {
  57.   let adjAlpha = Math.pow(this.alpha_, weight);
  58.   let newEstimate = value * (1 - adjAlpha) + adjAlpha * this.estimate_;
  60.   if (!isNaN(newEstimate)) {
  61.     this.estimate_ = newEstimate;
  62.     this.totalWeight_ += weight;
  63.   }
  64. };
  67. /**
  68.  * @return {number}
  69.  */
  70. shaka.abr.Ewma.prototype.getEstimate = function() {
  71.   let zeroFactor = 1 - Math.pow(this.alpha_, this.totalWeight_);
  72.   return this.estimate_ / zeroFactor;
  73. };