src/lib/MeanAccumulator/MeanAccumulator.h

   1 #pragma once
   2
   3 template <typename StorageType, typename IncrementType, IncrementType NoValueReturn>
   4 class MeanAccumulator
   5 {
   6 public:
   7     void add(IncrementType val)
   8     {
   9         _accumulator += val;
  10         ++_count;
  11     }
  12
  13     IncrementType mean()
  14     {
  15         if (_count)
  16         {
  17             _previousMean = _accumulator / _count;
  18             reset();
  19
  20             return _previousMean;
  21         }
  22         return NoValueReturn;
  23     }
  24
  25     IncrementType previousMean()
  26     {
  27         return _previousMean;
  28     }
  29
  30     void reset()
  31     {
  32         _accumulator = 0;
  33         _count = 0;
  34     }
  35
  36     size_t getCount() const
  37     {
  38         return _count;
  39     }
  40
  41 private:
  42     StorageType _accumulator;
  43     StorageType _count;
  44     IncrementType _previousMean;
  45 };