movingaverage.hh

#ifndef __WT_MOVINGAVERAGE_HH__
#define __WT_MOVINGAVERAGE_HH__

#include <Eigen/Eigen>

namespace wt {

template <class Scalar>
class MovingSum
{
public:
  typedef Eigen::Matrix<Scalar, 1, Eigen::Dynamic> RowVector;

public:
  MovingSum(size_t N, size_t M)
    : _hist(N,M), _current(N), _currentIdx(0)
  {
    _hist.setZero();
    _current.setZero();
  }

  void resize(size_t N, size_t M) {
    _hist.resize(N,M);  _hist.setZero();
    _current.resize(N); _current.setZero();
    _currentIdx=0;
  }

  template <class Derived>
  const RowVector &operator() (const Eigen::DenseBase<Derived> &in) {
    _current -= _hist.row(_currentIdx);
    _hist.row(_currentIdx) = in;
    _current += _hist.row(_currentIdx);
    _currentIdx++;
    // modulo N
    if (_hist.rows()==_currentIdx) { _currentIdx=0; }
    return _current;
  }

  const RowVector &addZero() {
    _current -= _hist.row(_currentIdx);
    _hist.row(_currentIdx).setZero();
    _currentIdx++;
    // modulo N
    if (_hist.rows()==_currentIdx) { _currentIdx=0; }
    return _current;
  }

protected:
  Eigen::Matrix<Scalar, Eigen::Dynamic, Eigen::Dynamic> _hist;
  Eigen::Matrix<Scalar, 1, Eigen::Dynamic> _current;
  size_t _currentIdx;
};

}

#endif // __WT_MOVINGAVERAGE_HH__