sdrangel/sdrbase/dsp/movingaverage.h

82 lines
2.7 KiB
C++

///////////////////////////////////////////////////////////////////////////////////////
// Copyright (C) 2012 maintech GmbH, Otto-Hahn-Str. 15, 97204 Hoechberg, Germany //
// written by Christian Daniel //
// Copyright (C) 2015-2018 Edouard Griffiths, F4EXB <f4exb06@gmail.com> //
// //
// This program is free software; you can redistribute it and/or modify //
// it under the terms of the GNU General Public License as published by //
// the Free Software Foundation as version 3 of the License, or //
// (at your option) any later version. //
// //
// This program is distributed in the hope that it will be useful, //
// but WITHOUT ANY WARRANTY; without even the implied warranty of //
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the //
// GNU General Public License V3 for more details. //
// //
// You should have received a copy of the GNU General Public License //
// along with this program. If not, see <http://www.gnu.org/licenses/>. //
///////////////////////////////////////////////////////////////////////////////////////
#ifndef INCLUDE_MOVINGAVERAGE_H
#define INCLUDE_MOVINGAVERAGE_H
#include <stdint.h>
#include <vector>
#include <algorithm>
template<typename Type> class MovingAverage {
public:
MovingAverage(int historySize, Type initial) : m_index(0)
{
resize(historySize, initial);
}
void resize(int historySize, Type initial)
{
m_history.resize(historySize);
std::fill(m_history.begin(), m_history.end(), initial);
m_sum = (Type) m_history.size() * initial;
m_index = 0;
}
void feed(Type value)
{
Type& oldest = m_history[m_index];
m_sum += value - oldest;
oldest = value;
if (m_index < m_history.size() - 1) {
m_index++;
} else {
m_index = 0;
}
}
void fill(Type value)
{
std::fill(m_history.begin(), m_history.end(), value);
m_sum = (Type) m_history.size() * value;
}
Type average() const
{
return m_sum / (Type) m_history.size();
}
Type sum() const
{
return m_sum;
}
int historySize() const
{
return m_history.size();
}
protected:
std::vector<Type> m_history;
Type m_sum;
uint32_t m_index;
};
#endif // INCLUDE_MOVINGAVERAGE_H