sum_values.hpp

Go to the documentation of this file.

#ifndef STAN__COMMON__RECORDER__SUM_VALUES_HPP
#define STAN__COMMON__RECORDER__SUM_VALUES_HPP

#include <vector>
#include <stdexcept>
#include <string>

namespace stan {
  namespace common {
    namespace recorder {
      
      class sum_values {
      public:
        sum_values(const size_t N) 
          : N_(N), m_(0), skip_(0), sum_(N_, 0.0) { }

        sum_values(const size_t N, const size_t skip) 
          : N_(N), m_(0), skip_(skip), sum_(N_, 0.0) { }
        

        void operator()(const std::vector<std::string>& x) { 
        }
        
        template <class T>
        void operator()(const std::vector<T>& x) { 
          if (N_ != x.size()) 
            throw std::length_error("vector provided does not match the parameter length");
          if (m_ >= skip_)
            for (size_t n = 0; n < N_; n++) {
              sum_[n] += x[n];
            }
          m_++;
        }

      
        void operator()(const std::string x) { }
      
        void operator()() { }
      
        bool is_recording() const {
          return true;
        }

        const std::vector<double>& sum() const {
          return sum_;
        }
        
        const size_t called() const {
          return m_;
        }

        const size_t recorded() const {
          if (m_ >= skip_)
            return m_ - skip_;
          else
            return 0;
        }
        

      private:
        size_t N_;
        size_t m_;
        size_t skip_;
        std::vector<double> sum_;
      };
      
    }
  }
}

#endif