Moving_average.hpp

Go to the documentation of this file.

#ifndef HRK_MOVING_AVERAGE_H
#define HRK_MOVING_AVERAGE_H

#include <deque>
#include <cstddef>


namespace hrk
{
    template <class T>
    class Moving_average
    {
    public:
        Moving_average(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 set_average_value(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:
        Moving_average(const Moving_average& rhs);
        Moving_average& operator = (const Moving_average& 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