imstkAccumulationBuffer.h

/*
** This file is part of the Interactive Medical Simulation Toolkit (iMSTK)
** iMSTK is distributed under the Apache License, Version 2.0.
** See accompanying NOTICE for details.
*/
#include <vector>
#include <imstkLogger.h>

template<typename T>
class AccumulationBuffer
{
public:
    explicit AccumulationBuffer(size_t capacity = 1024) : m_capacity(capacity), m_data(capacity, {})
    {
    }

    size_t getCapacity() const
    {
        return m_capacity;
    }

    void pushBack(T val)
    {
        m_sum += val;
        m_sum -= m_data[m_index];
        m_data[m_index] = val;
        m_index = (m_index + 1) % m_capacity;
    }

    T getAverage() const
    {
        return m_sum / static_cast<T>(m_capacity);
    }

private:
    size_t m_capacity;
    std::vector<T> m_data;
    T      m_sum   = {}; 
    size_t m_index = 0;  
};