94 lines
2.3 KiB
C++
94 lines
2.3 KiB
C++
#ifndef CSVFILE_HPP
|
|
#define CSVFILE_HPP
|
|
#include <fstream>
|
|
#include <iostream>
|
|
#include <sstream>
|
|
#include <string>
|
|
|
|
class csvfile;
|
|
|
|
inline static csvfile &endrow(csvfile &file);
|
|
inline static csvfile &flush(csvfile &file);
|
|
|
|
class csvfile {
|
|
std::ofstream fs_;
|
|
bool is_first_;
|
|
const std::string separator_;
|
|
const std::string escape_seq_;
|
|
const std::string special_chars_;
|
|
|
|
public:
|
|
csvfile(const std::string filename, const bool &overwrite,
|
|
const std::string separator = ";")
|
|
: fs_(), is_first_(true), separator_(separator), escape_seq_("\""),
|
|
special_chars_("\"") {
|
|
fs_.exceptions(std::ios::failbit | std::ios::badbit);
|
|
if (!std::filesystem::exists(std::filesystem::path("../OptimizationResults")
|
|
.append("statistics.csv")
|
|
)) {
|
|
std::ofstream outfile(std::filesystem::path("../OptimizationResults")
|
|
.append("statistics.csv")
|
|
.string());
|
|
outfile.close();
|
|
}
|
|
overwrite ? fs_.open(filename, std::ios::trunc)
|
|
: fs_.open(filename, std::ios::app);
|
|
}
|
|
|
|
~csvfile() {
|
|
flush();
|
|
fs_.close();
|
|
}
|
|
|
|
void flush() { fs_.flush(); }
|
|
|
|
void endrow() {
|
|
fs_ << std::endl;
|
|
is_first_ = true;
|
|
}
|
|
|
|
csvfile &operator<<(csvfile &(*val)(csvfile &)) { return val(*this); }
|
|
|
|
csvfile &operator<<(const char *val) { return write(escape(val)); }
|
|
|
|
csvfile &operator<<(const std::string &val) { return write(escape(val)); }
|
|
|
|
template <typename T> csvfile &operator<<(const T &val) { return write(val); }
|
|
|
|
private:
|
|
template <typename T> csvfile &write(const T &val) {
|
|
if (!is_first_) {
|
|
fs_ << separator_;
|
|
} else {
|
|
is_first_ = false;
|
|
}
|
|
fs_ << val;
|
|
return *this;
|
|
}
|
|
|
|
std::string escape(const std::string &val) {
|
|
std::ostringstream result;
|
|
result << '"';
|
|
std::string::size_type to, from = 0u, len = val.length();
|
|
while (from < len && std::string::npos !=
|
|
(to = val.find_first_of(special_chars_, from))) {
|
|
result << val.substr(from, to - from) << escape_seq_ << val[to];
|
|
from = to + 1;
|
|
}
|
|
result << val.substr(from) << '"';
|
|
return result.str();
|
|
}
|
|
};
|
|
|
|
inline static csvfile &endrow(csvfile &file) {
|
|
file.endrow();
|
|
return file;
|
|
}
|
|
|
|
inline static csvfile &flush(csvfile &file) {
|
|
file.flush();
|
|
return file;
|
|
}
|
|
|
|
#endif // CSVFILE_HPP
|