CEPS/src/Logger.cpp at main · MarkGillespie/CEPS · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
#include "Logger.h"

std::string to_string(LogType lot) {
    switch (lot) {
    case LogType::STRING:
        return "string";
    case LogType::DOUBLE:
        return "double";
    }
    return "Unrecognized log type";
}

Logger::Logger() {}

template <>
void Logger::log(std::string name, const char* val) {
    log(name, std::string(val));
}

template <>
void Logger::log(std::string name, std::string val) {
    logs.push_back(std::make_tuple(name, LogType::STRING, stringLogs.size()));
    stringLogs.push_back(val);
}

template <>
void Logger::log(std::string name, bool val) {
    logs.push_back(std::make_tuple(name, LogType::STRING, stringLogs.size()));
    stringLogs.push_back(val ? "True" : "False");
}

template <>
void Logger::log(std::string name, double val) {
    logs.push_back(std::make_tuple(name, LogType::DOUBLE, doubleLogs.size()));
    doubleLogs.push_back(val);
}

template <>
void Logger::log(std::string name, size_t val) {
    logs.push_back(std::make_tuple(name, LogType::DOUBLE, doubleLogs.size()));
    doubleLogs.push_back(val);
}

template <>
void Logger::log(std::string name, int val) {
    logs.push_back(std::make_tuple(name, LogType::DOUBLE, doubleLogs.size()));
    doubleLogs.push_back(val);
}

bool Logger::writeLog(std::string filename) {
    std::ofstream out;

    // std::ios::trunc ensures that we overwrite anything previously in the file
    out.open(filename, std::ios::trunc);
    if (out.is_open()) {
        writeLog(out);
        out.close();
        return true;
    } else {
        return false;
    }
}

void Logger::writeLog(std::ostream& out) {
    out << std::setw(12);
    size_t N = logs.size();
    for (size_t iL = 0; iL + 1 < N; ++iL) {
        out << std::get<0>(logs[iL]) << "\t";
    }
    out << std::get<0>(logs[logs.size() - 1]) << std::endl;

    auto printLog = [&](size_t iL) {
        switch (std::get<1>(logs[iL])) {
        case LogType::STRING:
            out << stringLogs[std::get<2>(logs[iL])];
            break;
        case LogType::DOUBLE:
            out << doubleLogs[std::get<2>(logs[iL])];
            break;
        }
    };

    for (size_t iL = 0; iL + 1 < N; ++iL) {
        printLog(iL);
        out << "\t";
    }
    printLog(logs.size() - 1);
}