libs/common/MovingAverage.h

Go to the documentation of this file.

#ifndef QRK_MOVING_AVERAGE_H
#define QRK_MOVING_AVERAGE_H

#include <deque>


namespace qrk
{

    template <class T>
    class MovingAverage
    {
    public:
        MovingAverage(size_t size, T initial_value = 0)
            : buffer_size_(size), total_(0)
        {
            if (buffer_size_ == 0) {
                buffer_size_ = 1;
            }
            buffer_set(buffer_size_, initial_value);
        }


        void setAverageValue(T value)
        {
            buffer_set(buffer_size_, value);
        }


        T average(void)
        {
            return average_;
        }


        T push_back(T value)
        {
            total_ -= buffer_.front();
            buffer_.pop_front();
            total_ += value;
            buffer_.push_back(value);

            average_ = total_ / buffer_size_;

            return average_;
        }

    private:
        MovingAverage(const MovingAverage& rhs);
        MovingAverage& operator = (const MovingAverage& rhs);

        std::deque<T> buffer_;
        long buffer_size_;
        T average_;
        T total_;


        void buffer_set(size_t size, T value)
        {
            buffer_.assign(size, value);
            average_ = value;
            total_ = average_ * size;
        }
    };
}

#endif /* !QRK_MOVING_AVERAGE_H */