csv_writer.hpp

Go to the documentation of this file.

#ifndef __STAN__IO__CSV_WRITER_HPP__
#define __STAN__IO__CSV_WRITER_HPP__

#include <cstddef>
#include <iostream>
#include <stan/math/matrix.hpp>

namespace stan {

  namespace io {

    class csv_writer {
    private:
      std::ostream& o_;
      bool at_bol_;

    public: 

      void comma() {
        if (at_bol_) {
          at_bol_ = false;
          return;
        }
        o_ << ",";
      }


      csv_writer(std::ostream& o)
        : o_(o), at_bol_(true) {
      }

      void newline() {
        o_ << "\n";
        at_bol_ = true;
      }

      void write(int n) {
        comma();
        o_ << n;
      }

      void write(double x) {
        comma();
        o_ << x;
      }

      void write(const Eigen::Matrix<double,Eigen::Dynamic,1>& v) {
        typedef Eigen::Matrix<double,Eigen::Dynamic,1>::size_type size_type;
        for (size_type i = 0; i < v.size(); ++i)
          write(v[i]);
      }
      
      void write(const Eigen::Matrix<double,1,Eigen::Dynamic>& rv) {
        typedef Eigen::Matrix<double,1,Eigen::Dynamic>::size_type size_type;
        for (size_type i = 0; i < rv.size(); ++i)
          write(rv[i]);
      }

      void write(const Eigen::Matrix<double,Eigen::Dynamic,Eigen::Dynamic>& m) {
        typedef Eigen::Matrix<double,Eigen::Dynamic,Eigen::Dynamic>::size_type size_type;
        for (size_type i = 0; i < m.rows(); ++i)
          for (size_type j = 0; j < m.cols(); ++j)
            write(m(i,j));
      }

      void write_col_major(const Eigen::Matrix<double,Eigen::Dynamic,Eigen::Dynamic>& m) {
        typedef Eigen::Matrix<double,Eigen::Dynamic,Eigen::Dynamic>::size_type size_type;
        for (size_type j = 0; j < m.cols(); ++j)
          for (size_type i = 0; i < m.rows(); ++i)
            write(m(i,j));
      }

      void write(const std::string& s) {
        comma();

        o_ << '"';
        for (size_t i = 0; i < s.size(); ++i) {
          if (s.at(i) == '"') {
            o_ << '"' << '"'; // double quotes
          } else {
            o_ << s.at(i);
          }
        }
        o_ << '"';
      }


    };

  }

}

#endif