Logger.cpp

Go to the documentation of this file.

 
#include "Logger.h"
 
#include <cstdio>
#include <iomanip>
#include <iostream>
#include <sstream>
#include <unistd.h>
 
namespace {
const char* const kReset = "\033[0m";
}
 
Logger::Logger(bool verbose, bool useColor)
    : verbose_(verbose), useColor_(useColor && isatty(fileno(stdout)) != 0) {}
 
bool Logger::openEventFile(const std::string& path) {
    eventFile_.open(path);
    return eventFile_.is_open();
}
 
bool Logger::verbose() const {
    return verbose_;
}
 
const char* Logger::label(LogCategory category) {
    switch (category) {
        case LogCategory::Arrival: return "ARRIVE ";
        case LogCategory::Assign: return "ASSIGN ";
        case LogCategory::Complete: return "DONE   ";
        case LogCategory::Reject: return "REJECT ";
        case LogCategory::Discard: return "DISCARD";
        case LogCategory::ScaleUp: return "SCALE+ ";
        case LogCategory::ScaleDown: return "SCALE- ";
        case LogCategory::Status: return "STATUS ";
        case LogCategory::Info: return "INFO   ";
    }
    return "";
}
 
const char* Logger::color(LogCategory category) {
    switch (category) {
        case LogCategory::Arrival: return "\033[34m";    // blue
        case LogCategory::Assign: return "\033[36m";     // cyan
        case LogCategory::Complete: return "\033[32m";   // green
        case LogCategory::Reject: return "\033[1;31m";   // bold red
        case LogCategory::Discard: return "\033[31m";    // red
        case LogCategory::ScaleUp: return "\033[1;33m";  // bold yellow
        case LogCategory::ScaleDown: return "\033[1;35m";// bold magenta
        case LogCategory::Status: return "\033[1;37m";   // bold white
        case LogCategory::Info: return "\033[37m";       // white
    }
    return "";
}
 
std::string Logger::colorize(LogCategory category, const std::string& text) const {
    if (!useColor_) {
        return text;
    }
    return std::string(color(category)) + text + kReset;
}
 
void Logger::log(LogCategory category, unsigned long cycle, const std::string& source,
                 const std::string& message) {
    const bool alwaysShown = category == LogCategory::Status || category == LogCategory::Info;
    if (!verbose_ && !alwaysShown) {
        return;
    }
    std::ostringstream line;
    line << "[t=" << std::setw(6) << std::setfill('0') << cycle << "] [" << source << "] "
         << label(category) << " " << message;
    std::cout << colorize(category, line.str()) << '\n';
    if (eventFile_.is_open()) {
        eventFile_ << line.str() << '\n';
    }
}