Logger.hpp

#pragma once
 
#include <string>
#include <fstream>
#include <sstream>
#include <vector>
 
// Fix for Windows headers.
#ifdef ERROR
#   undef ERROR
#endif
 
class Log {
 
public:
    enum Domain {
        GPU = 0,
        Resources,
        Input,
        Utilities,
        Config
    };
 
private:
    enum class Level : int {
        INFO = 0,
        WARNING,
        ERROR,
        VERBOSE 
    };
 
    void set(Level l);
 
    const std::vector<std::string> _domainStrings = {"GPU", "Resources", "Input", "Utilities", "Config"}; 
 
    const std::vector<std::string> _levelStrings = {"", "(!) ", "(X) ", ""}; 
 
    const std::vector<std::string> _colorStrings = {"\x1B[0m\x1B[39m", "\x1B[0m\x1B[33m", "\x1B[0m\x1B[31m", "\x1B[2m\x1B[37m"}; 
 
public:
    Log();
 
    Log(const std::string & filePath, bool logToStdin, bool verbose = false);
 
    void setVerbose(bool verbose);
 
    template<class T>
    Log & operator<<(const T & input) {
        appendIfNeeded();
        _stream << input;
        return *this;
    }
 
    Log & operator<<(const Domain & domain);
 
    Log & operator<<(std::ostream & (*modif)(std::ostream &));
 
    Log & operator<<(std::ios_base & (*modif)(std::ios_base &));
 
    static void setDefaultFile(const std::string & filePath);
 
    static void setDefaultVerbose(bool verbose);
 
    static Log & Info();
 
    static Log & Warning();
 
    static Log & Error();
 
    static Log & Verbose();
 
private:
    void setFile(const std::string & filePath, bool flushExisting = true);
 
    void flush();
 
    void appendIfNeeded();
 
    Level _level      = Level::INFO; 
    bool _logToStdOut = true;        
    std::ofstream _file;             
    std::stringstream _stream;       
    bool _verbose          = false;  
    bool _ignoreUntilFlush = false;  
    bool _appendPrefix   = false;  
    bool _useColors        = false;  
 
    static Log * _defaultLogger; 
};