weighted_moving_average_impl.h

Go to the documentation of this file.

#pragma once
 
#include "fl/stl/circular_buffer.h"
#include "fl/stl/span.h"
#include "fl/stl/noexcept.h"
 
namespace fl {
namespace detail {
 
template <typename T, fl::size N = 0>
class WeightedMovingAverageImpl {
  public:
    WeightedMovingAverageImpl() FL_NO_EXCEPT : mLastValue(T(0)) {}
    explicit WeightedMovingAverageImpl(fl::size capacity)
        : mBuf(capacity), mLastValue(T(0)) {}
 
    T update(T input) {
        mBuf.push_back(input);
        return recompute();
    }
 
    T update(fl::span<const T> values) {
        if (values.size() == 0) return mLastValue;
        for (fl::size i = 0; i < values.size(); ++i) {
            mBuf.push_back(values[i]);
        }
        return recompute();
    }
 
    T value() const { return mLastValue; }
    void reset() { mBuf.clear(); mLastValue = T(0); }
    bool full() const { return mBuf.full(); }
    fl::size size() const { return mBuf.size(); }
    fl::size capacity() const { return mBuf.capacity(); }
 
    void resize(fl::size new_capacity) {
        mBuf = circular_buffer<T, N>(new_capacity);
        mLastValue = T(0);
    }
 
  private:
    T recompute() {
        fl::size n = mBuf.size();
        T weighted_sum = T(0);
        T weight_total = T(0);
        for (fl::size i = 0; i < n; ++i) {
            T w = T(static_cast<float>(i + 1));
            weighted_sum = weighted_sum + mBuf[i] * w;
            weight_total = weight_total + w;
        }
        mLastValue = weighted_sum / weight_total;
        return mLastValue;
    }
 
    circular_buffer<T, N> mBuf;
    T mLastValue;
};
 
} // namespace detail
} // namespace fl