writer_def.hpp

Go to the documentation of this file.

 
#ifndef ROOT_WRITER_DEF_HPP
#define ROOT_WRITER_DEF_HPP
 
#include <Eigen/Dense>
#include <fstream>
#include <iostream>
#include <string>
 
enum WritingMethod { CONSOLE, CSV, DAT, GNUPLOT };
 
template <typename V>
class PrinterBase;
 
template <typename T>
class Writer {
  private:
    friend class WriterBaseTester;  //<! Friend test fixture class for unit testing.
    template <typename V>
    void build_printer(std::unique_ptr<PrinterBase<V>>& printer);
 
  protected:
    T values;              
    WritingMethod method;  
    char separator;        
    bool overwrite;        
    std::string filename;  
 
  public:
    Writer(const T& vals_to_write, WritingMethod write_method, std::string filename = "output", char separator = ',',
           bool overwrite = true);
    void write();
};
 
template <typename V>
class PrinterBase {
  public:
    PrinterBase() = default;
    virtual ~PrinterBase() = default;
    virtual void write_values(const V& value) = 0;
};
 
template <typename V>
class PrinterCLI : public PrinterBase<V> {
  public:
    PrinterCLI();
    void write_values(const V& value) override;
};
 
template <typename V>
class PrinterFile : public PrinterBase<V> {
  protected:
    std::string filename;  
    bool append;  
    std::ofstream file;  
 
  public:
    PrinterFile(const std::string& fname, bool ow_mode);
};
 
template <typename V>
class PrinterDAT : public PrinterFile<V> {
  public:
    PrinterDAT(const std::string& fname, bool ow_mode);
    void write_values(const V& value) override;
};
 
template <typename V>
class PrinterCSV : public PrinterFile<V> {
  private:
    char separator;  
 
  public:
    PrinterCSV(const std::string& fname, char sep, bool ow_mode);
    void write_values(const V& value) override;
};
 
template <typename V>
// LLM
class PrinterGNUPlot : public PrinterDAT<V> {
  public:
    PrinterGNUPlot(const std::string& fname, bool ow_mode);
    void generate_gnuplot_script() const;
};
 
#include "writer.hpp"
 
#endif  // ROOT_WRITER_HPP