FairMQ/fairmq/logger/logger_def.h

/********************************************************************************
 *    Copyright (C) 2014 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH    *
 *                                                                              *
 *              This software is distributed under the terms of the             * 
 *         GNU Lesser General Public Licence version 3 (LGPL) version 3,        *  
 *                  copied verbatim in the file "LICENSE"                       *
 ********************************************************************************/
/* 
 * File:   logger_def.h
 * Author: winckler
 *
 * Created on September 2, 2015, 11:58 AM
 */

#ifndef LOGGER_DEF_H
#define LOGGER_DEF_H

#include <array>
#include <fstream>
#include <sstream>

#include <boost/log/utility/manipulators/to_log.hpp>
#include <boost/log/sources/record_ostream.hpp>

namespace FairMQ
{

enum severity_level
{
    TRACE,
    DEBUG,
    RESULTS,
    INFO,
    STATE,
    WARN,
    ERROR,
    NOLOG
};

static const std::array<std::string, 8> g_LogSeverityLevelString
{
    {
        "TRACE",
        "DEBUG",
        "RESULTS",
        "INFO",
        "STATE",
        "WARN",
        "ERROR",
        "NOLOG"
    }
};

namespace Color
{

enum code
{
    FG_BLACK    = 30,
    FG_RED      = 31,
    FG_GREEN    = 32,
    FG_YELLOW   = 33,
    FG_BLUE     = 34,
    FG_MAGENTA  = 35,
    FG_CYAN     = 36,
    FG_WHITE    = 37,
    FG_DEFAULT  = 39,
    BG_RED      = 41,
    BG_GREEN    = 42,
    BG_BLUE     = 44,
    BG_DEFAULT  = 49
};

}

} // FairMQ namespace

// helper function to format in color console output
template <FairMQ::Color::code color> 
inline std::string write_in(const std::string& text_in_bold)
{
    std::ostringstream os;
    os << "\033[01;" << color << "m" << text_in_bold << "\033[0m";

    return os.str();
}

// typedef
typedef FairMQ::severity_level custom_severity_level;
#define SEVERITY_MINIMUM custom_severity_level::TRACE
#define SEVERITY_ERROR custom_severity_level::ERROR
#define SEVERITY_NOLOG custom_severity_level::NOLOG

// tags used for log console or file formatting
struct tag_console;
struct tag_file;

// overload operator for console output
inline boost::log::formatting_ostream& operator<<
        (
            boost::log::formatting_ostream& strm,
            boost::log::to_log_manip<custom_severity_level, tag_console> const& manip
        )
{
    custom_severity_level level = manip.get();
    std::size_t idx = static_cast<std::size_t>(level);
    if (idx < FairMQ::g_LogSeverityLevelString.size())
    {
        // strm << " idx = " << idx << " ";
        switch (level)
        {
            case custom_severity_level::TRACE :
                strm << write_in<FairMQ::Color::FG_BLUE>(FairMQ::g_LogSeverityLevelString.at(idx));
                break;

            case custom_severity_level::DEBUG :
                strm << write_in<FairMQ::Color::FG_BLUE>(FairMQ::g_LogSeverityLevelString.at(idx));
                break;

            case custom_severity_level::RESULTS :
                strm << write_in<FairMQ::Color::FG_MAGENTA>(FairMQ::g_LogSeverityLevelString.at(idx));
                break;

            case custom_severity_level::INFO :
                strm << write_in<FairMQ::Color::FG_GREEN>(FairMQ::g_LogSeverityLevelString.at(idx));
                break;

            case custom_severity_level::WARN :
                strm << write_in<FairMQ::Color::FG_YELLOW>(FairMQ::g_LogSeverityLevelString.at(idx));
                break;

            case custom_severity_level::STATE :
                strm << write_in<FairMQ::Color::FG_MAGENTA>(FairMQ::g_LogSeverityLevelString.at(idx));
                break;

            case custom_severity_level::ERROR :
                strm << write_in<FairMQ::Color::FG_RED>(FairMQ::g_LogSeverityLevelString.at(idx));
                break;

            case custom_severity_level::NOLOG :
                strm << write_in<FairMQ::Color::FG_DEFAULT>(FairMQ::g_LogSeverityLevelString.at(idx));
                break;

            default:
                break;
        }
    }
    else
    {
        strm << write_in<FairMQ::Color::FG_RED>("Unknown log level ")
             << "(int level = " << static_cast<int>(level) << ")";
    }
    return strm;
}

// overload operator for file output
inline boost::log::formatting_ostream& operator<<
        (
            boost::log::formatting_ostream& strm,
            boost::log::to_log_manip<custom_severity_level, tag_file> const& manip
        )
{
    custom_severity_level level = manip.get();
    std::size_t idx = static_cast<std::size_t>(level);
    if (idx < FairMQ::g_LogSeverityLevelString.size())
    {
        strm << FairMQ::g_LogSeverityLevelString.at(idx);
    }
    else
    {
        strm << write_in<FairMQ::Color::FG_RED>("Unknown log level ")
             << "(int level = " << static_cast<int>(level) << ")";
    }
    return strm;
}

#endif /* LOGGER_DEF_H */