csv_writer.hpp

Go to the documentation of this file.

#ifndef STAN__IO__CSV_WRITER_HPP
#define STAN__IO__CSV_WRITER_HPP

#include <ostream>
#include <limits>
#include <iomanip>
#include <stan/math/matrix.hpp>
#include <stan/math/meta/index_type.hpp>
#include <stan/math/matrix/meta/index_type.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_ << std::setprecision (std::numeric_limits<double>::digits10 + 1)
           << x;
      }

      template<int R, int C>
      void write_row_major(const Eigen::Matrix<double,R,C>& m) {
        using Eigen::Matrix;
        using stan::math::index_type;
        typedef typename index_type<Matrix<double,R,C> >::type idx_t;
        for (idx_t i = 0; i < m.rows(); ++i)
          for (idx_t j = 0; j < m.cols(); ++j)
            write(m(i,j));
      }

      template<int R, int C>
      void write_col_major(const Eigen::Matrix<double,R,C>& m) {
        using Eigen::Matrix;
        using stan::math::index_type;
        typedef typename index_type<Matrix<double,R,C> >::type idx_t;

        for (idx_t j = 0; j < m.cols(); ++j)
          for (idx_t i = 0; i < m.rows(); ++i)
            write(m(i,j));
      }

      template<int R, int C>
      void write(const Eigen::Matrix<double,R,C>& m) {
        write_col_major(m);
      }

      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