csvwriter.h

Go to the documentation of this file.

/*
 * Copyright 2021 Ingemar Hedvall
 * SPDX-License-Identifier: MIT
 */
#pragma once
#include <cstdio>
#include <fstream>
#include <sstream>
#include <string>
#include <utility>
#include <vector>
 
namespace util::plot {

 */
class CsvWriter {
 public:
  explicit CsvWriter(const std::string& filename);

  CsvWriter();
  virtual ~CsvWriter();

  void CloseFile();

  [[nodiscard]] std::string FileName() const;

  [[nodiscard]] const std::string& FullName() const { return filename_; }

  void OpenFile(const std::string& filename);

  [[nodiscard]] bool IsOk() const;

  void AddColumnHeader(const std::string& name, const std::string& unit,
                       bool valid = true);

  void SetColumnValid(size_t column, bool valid = true);

  [[nodiscard]] bool IsColumnValid(size_t column) const;

  void AddRow();

  template <typename T>
  void AddColumnValue(const T& value);

  [[nodiscard]] std::string GetText() const { return text_.str(); }

  void Reset();

  [[nodiscard]] size_t Columns() const { return max_columns_; }

  [[nodiscard]] size_t Rows() const { return row_count_; }

  [[nodiscard]] const std::string Label(size_t index) const;

  [[nodiscard]] const std::string& Name(size_t index) const;

  [[nodiscard]] const std::string& Unit(size_t index) const;
 
 private:
  std::ofstream file_;
  std::ostringstream text_;
  std::ostream* output_ = nullptr;
 
  std::string filename_;  
  size_t column_count_ = 0;  
  size_t max_columns_ =
      0;  
  size_t row_count_ = 0;  
 
  struct Header {
    std::string name;    
    std::string unit;    
    bool valid = false;  
  };
  using NameUnitList = std::vector<Header>;
  NameUnitList header_list_;

  void SaveText(const std::string& text);
};
 
template <typename T>
void CsvWriter::AddColumnValue(const T& value) {
  auto temp = std::to_string(value);
  SaveText(temp);
  if (row_count_ == 0) {
    max_columns_ = std::max(column_count_, max_columns_);
  }
}

template <>
void CsvWriter::AddColumnValue<std::string>(const std::string& value);

template <>
void CsvWriter::AddColumnValue<float>(const float& value);

template <>
void CsvWriter::AddColumnValue<double>(const double& value);

template <>
void CsvWriter::AddColumnValue<bool>(const bool& value);
}  // namespace util::plot