TerminalOutput.h

 
#ifndef WRENCH_S4U_COLORLOGGING_H
#define WRENCH_S4U_COLORLOGGING_H
 
#include <map>
 
#include <simgrid/s4u/Actor.hpp>
#include <xbt/log.h>
 
#include <iostream>
 
namespace wrench {
 
 
/* Wrappers around XBT_* macros, using a bit of those macro's internal magic as well
 * to avoid generating useless (but space consuming) color ASCII codes
 */
 
#define WRENCH_LOG_CATEGORY(cname, desc) XBT_LOG_NEW_DEFAULT_CATEGORY(cname, desc)
 
#define WRENCH_INFO(...);  \
    if (_XBT_LOG_ISENABLEDV((*_simgrid_log_category__default), xbt_log_priority_info)) { \
      wrench::TerminalOutput::beginThisProcessColor();  \
      XBT_INFO(__VA_ARGS__) ;  \
      wrench::TerminalOutput::endThisProcessColor(); \
    }
 
#define WRENCH_DEBUG(...);  \
    if (_XBT_LOG_ISENABLEDV((*_simgrid_log_category__default), xbt_log_priority_debug)) { \
      wrench::TerminalOutput::beginThisProcessColor();  \
      XBT_DEBUG(__VA_ARGS__) ;  \
      wrench::TerminalOutput::endThisProcessColor(); \
    }
 
#define WRENCH_WARN(...);  \
    if (_XBT_LOG_ISENABLEDV((*_simgrid_log_category__default), xbt_log_priority_warning)) { \
      wrench::TerminalOutput::beginThisProcessColor();  \
      XBT_WARN(__VA_ARGS__) ;  \
      wrench::TerminalOutput::endThisProcessColor(); \
    }
 
    /***********************/
    /***********************/
 
    class TerminalOutput {
 
    public:
 
        enum Color {
                    COLOR_BLACK,
                    COLOR_RED,
                    COLOR_GREEN,
                    COLOR_YELLOW,
                    COLOR_BLUE,
                    COLOR_MAGENTA,
                    COLOR_CYAN,
                    COLOR_WHITE,
        };
 
        static void setThisProcessLoggingColor(Color color);
 
 
 
        /***********************/
        /***********************/
 
        static void beginThisProcessColor();
 
        static void endThisProcessColor();
 
        static void disableColor();
 
//        static void disableLog();
 
        /***********************/
        /***********************/
 
    private:
 
        static const char * color_codes[];
 
        static std::unordered_map<simgrid::s4u::Actor *, std::string> colormap;
 
        static std::string getThisProcessLoggingColor();
 
        static bool color_enabled;
 
    };
 
    /***********************/
    /***********************/
};
 
 
#endif //WRENCH_S4U_COLORLOGGING_H