csv.hpp

Go to the documentation of this file.

#ifndef STAN__COMMON__RECORDER__CSV_HPP
#define STAN__COMMON__RECORDER__CSV_HPP

#include <ostream>
#include <string>
#include <vector>

#include <stan/math/meta/index_type.hpp>

namespace stan {

  namespace common {

    namespace recorder {
      
      class csv {
      private:
        std::ostream *o_;
        const bool has_stream_;
        const std::string prefix_;
      
      public:
        csv(std::ostream *o, std::string prefix) 
          : o_(o), has_stream_(o != 0), prefix_(prefix) { }
      
        template <class T>
        void operator()(const std::vector<T>& x) {
          typedef typename stan::math::index_type<std::vector<T> >::type idx_t;
          if (!has_stream_)
            return;
        
          if (x.size() != 0) {
            *o_ << x[0];
            for (idx_t n = 1; n < x.size(); n++) {
              *o_ << "," << x[n];
            }
          }
          *o_ << std::endl;
        }
      
        void operator()(const std::string x) {
          if (!has_stream_)
            return;
          *o_ << prefix_ << x << std::endl;
        }
      
        void operator()() {
          if (!has_stream_)
            return;
          *o_ << std::endl;
        }
      
        bool is_recording() const {
          return has_stream_;
        }
      };


    }
  }
}

#endif