autocovariance.hpp

Go to the documentation of this file.

#ifndef __STAN__PROB__AUTOCOVARIANCE_HPP__
#define __STAN__PROB__AUTOCOVARIANCE_HPP__

#include <stan/prob/autocorrelation.hpp>
#include <stan/math/matrix/variance.hpp>

namespace stan {
  
  namespace prob {

    
    template <typename T>
    void autocovariance(const std::vector<T>& y,
                        std::vector<T>& acov,
                        Eigen::FFT<T>& fft) {
      
      stan::prob::autocorrelation(y, acov, fft);

      T var = stan::math::variance(y) * (y.size()-1) / y.size();
      for (size_t i = 0; i < y.size(); i++) {
        acov[i] *= var;
      }
    }

    template <typename T>
    void autocovariance(const std::vector<T>& y,
                        std::vector<T>& acov) {
      Eigen::FFT<T> fft;
      autocovariance(y,acov,fft);
    }


  }
}

#endif