Source: lib/abr/ewma.js

  1. /*! @license
  2.  * Shaka Player
  3.  * Copyright 2016 Google LLC
  4.  * SPDX-License-Identifier: Apache-2.0
  5.  */
  7. goog.provide('shaka.abr.Ewma');
  9. goog.require('goog.asserts');
  12. /**
  13.  * @summary
  14.  * This class computes an exponentionally-weighted moving average.
  15.  */
  16. shaka.abr.Ewma = class {
  17.   /**
  18.    * @param {number} halfLife The quantity of prior samples (by weight) used
  19.    *   when creating a new estimate.  Those prior samples make up half of the
  20.    *   new estimate.
  21.    */
  22.   constructor(halfLife) {
  23.     goog.asserts.assert(halfLife > 0, 'expected halfLife to be positive');
  25.     /**
  26.      * Larger values of alpha expire historical data more slowly.
  27.      * @private {number}
  28.      */
  29.     this.alpha_ = Math.exp(Math.log(0.5) / halfLife);
  31.     /** @private {number} */
  32.     this.estimate_ = 0;
  34.     /** @private {number} */
  35.     this.totalWeight_ = 0;
  36.   }
  38.   /**
  39.    * Takes a sample.
  40.    *
  41.    * @param {number} weight
  42.    * @param {number} value
  43.    */
  44.   sample(weight, value) {
  45.     const adjAlpha = Math.pow(this.alpha_, weight);
  46.     const newEstimate = value * (1 - adjAlpha) + adjAlpha * this.estimate_;
  48.     if (!isNaN(newEstimate)) {
  49.       this.estimate_ = newEstimate;
  50.       this.totalWeight_ += weight;
  51.     }
  52.   }
  55.   /**
  56.    * @return {number}
  57.    */
  58.   getEstimate() {
  59.     const zeroFactor = 1 - Math.pow(this.alpha_, this.totalWeight_);
  60.     return this.estimate_ / zeroFactor;
  61.   }
  62. };