RollingStat.h

/*
 * RollingStat.h
 *
 * This file is part of Mozzi.
 *
 * Copyright 2013-2024 Tim Barrass and the Mozzi Team
 *
 * Mozzi is licensed under the GNU Lesser General Public Licence (LGPL) Version 2.1 or later.
 *
 * WARNING: this code is incomplete, it doesn't work yet
 */
 
#ifndef ROLLINGSTAT_H
#define ROLLINGSTAT_H
 
#include "RollingAverage.h"
#include "mozzi_fixmath.h"
 
// adapted from RunningStat, http://www.johndcook.com/standard_deviation.html
template <class T, int WINDOW_LENGTH>
class RollingStat
{
public:
 
    RollingStat() : _previous_mean(0), _mean(0), _variance(0), WINDOW_LENGTH_AS_RSHIFT(trailingZerosConst(WINDOW_LENGTH))
    {}
 
 
    void update(T x) {
        _mean = rollingMean.next(x);
        _variance = (((long)x - _previous_mean)*((long)x - _mean))>>WINDOW_LENGTH_AS_RSHIFT; // should really be div by (WINDOW_LENGTH-1), but exact values are not important for this application
        _previous_mean = _mean;
    }
 
 
    void update(int8_t x) {
        _mean = rollingMean.next(x);
        _variance = (((int)x - _previous_mean)*((int)x - _mean))>>WINDOW_LENGTH_AS_RSHIFT; // should really be div by (WINDOW_LENGTH-1), but exact values are not important for this application
        _previous_mean = _mean;
    }
 
 
    T getMean() const {
        return _mean;
    }
 
 
    T getVariance() const {
        return _variance;
    }
 
    T getStandardDeviation() const {
        return isqrt16(_variance);
    }
 
 
 
private:
    T _previous_mean, _mean, _variance;
    RollingAverage <T, WINDOW_LENGTH> rollingMean;
    const uint8_t WINDOW_LENGTH_AS_RSHIFT;
};
 
// no need to show the specialisations
template <int WINDOW_LENGTH>
class RollingStat <float, WINDOW_LENGTH>
{
public:
 
    RollingStat() : _previous_mean(0), _mean(0), _variance(0), WINDOW_LENGTH_AS_RSHIFT(trailingZerosConst(WINDOW_LENGTH))
    {}
 
 
    void update(float x) {
        _mean = rollingMean.next(x);
        _variance = ((x - _previous_mean)*(x - _mean))/(WINDOW_LENGTH-1);
        _previous_mean = _mean;
    }
 
 
    float getMean() const {
        return _mean;
    }
 
 
    float getVariance() const {
        return _variance;
    }
 
    float getStandardDeviation() const {
        return sqrt(_variance);
    }
 
 
 
private:
    float _previous_mean, _mean, _variance;
    RollingAverage <float, WINDOW_LENGTH> rollingMean;
    const uint8_t WINDOW_LENGTH_AS_RSHIFT;
};
 
// no need to show the specialisations
#endif        //  #ifndef ROLLINGSTAT_H