csv_writer.h

Go to the documentation of this file.

#pragma once
 
#include <fstream>
#include <sstream>
#include <string>
 
class CSVWriter {
 public:
  explicit CSVWriter(std::string delimiter = ",") : delimiter_(delimiter) {}
 
  template <typename T>
  void add_data_in_row(T first, T last) {
    // Iterate over the range and add each element to the content
    for (; first != last;) {
      buffer_ << *first;
      if (++first != last) buffer_ << delimiter_;
    }
    buffer_ << std::endl;
  }
 
  bool save_to_file(std::string path) {
    std::fstream file;
    // Open the file in truncate mode
    file.open(path, std::ios::out | std::ios::trunc);
 
    if (!file.good()) {
      return false;
    }
 
    file << buffer_.str();
    file.close();
 
    return true;
  }
 
  std::string get_content() { return buffer_.str(); }
 
 private:
  std::ostringstream buffer_;
  std::string delimiter_;
};