MaCh3Logger.h

Go to the documentation of this file.

#pragma once
 
// C++ Includes
#include <iostream>
#include <sstream>
#include <functional>
#include <string>
#include <exception>
 
// MaCh3 Includes
#include "Manager/Core.h"
 
_MaCh3_Safe_Include_Start_ //{
// spdlog Includes
#include "spdlog/spdlog.h"
_MaCh3_Safe_Include_End_ //}
 
 
#define MACH3LOG_TRACE SPDLOG_TRACE
#define MACH3LOG_DEBUG SPDLOG_DEBUG
#define MACH3LOG_INFO SPDLOG_INFO
#define MACH3LOG_WARN SPDLOG_WARN
#define MACH3LOG_ERROR SPDLOG_ERROR
#define MACH3LOG_CRITICAL SPDLOG_CRITICAL
#define MACH3LOG_OFF SPDLOG_OFF
 
inline spdlog::level::level_enum get_default_log_level() {
  #ifdef SPDLOG_ACTIVE_LEVEL
  switch (SPDLOG_ACTIVE_LEVEL) {
    case SPDLOG_LEVEL_TRACE:    return spdlog::level::trace;
    case SPDLOG_LEVEL_DEBUG:    return spdlog::level::debug;
    case SPDLOG_LEVEL_INFO:     return spdlog::level::info;
    case SPDLOG_LEVEL_WARN:     return spdlog::level::warn;
    case SPDLOG_LEVEL_ERROR:    return spdlog::level::err;
    case SPDLOG_LEVEL_CRITICAL: return spdlog::level::critical;
    case SPDLOG_LEVEL_OFF:      return spdlog::level::off;
    default: throw std::runtime_error("Unknown SPDLOG_ACTIVE_LEVEL");
  }
  #else
  throw std::runtime_error("SPDLOG_ACTIVE_LEVEL is not defined");
  #endif
}
 
inline void SetMaCh3LoggerFormat()
{
  //KS: %H for hour, %M for minute, %S for second, [%s:%#] for class and line
  //For documentation see https://github.com/gabime/spdlog/wiki/3.-Custom-formatting
  #ifdef DEBUG
  //spdlog::set_pattern("[%H:%M:%S][%s:%#][%^%l%$] %v");
  spdlog::set_pattern("[%s:%#][%^%l%$] %v");
  #else
  //spdlog::set_pattern("[%H:%M:%S][%s][%^%l%$] %v");
  spdlog::set_pattern("[%s][%^%l%$] %v");
  #endif
 
  spdlog::set_level(get_default_log_level());
}
 
template <typename Func, typename LogFunc, typename... Args>
void LoggerPrint(const std::string& LibName, LogFunc logFunction, Func&& func, Args&&... args)
{
  // Create a stringstream to capture the output
  std::stringstream sss_cout;
  std::stringstream sss_cerr;
  
  // Save the original stream buffers
  std::streambuf* coutBuf = nullptr;
  std::streambuf* cerrBuf = nullptr;
 
  // This should be rare but in case buffers no longer exist ignore
  if (std::cout.rdbuf() && std::cerr.rdbuf()) {
    coutBuf = std::cout.rdbuf();  // Save original cout buffer
    cerrBuf = std::cerr.rdbuf();  // Save original cerr buffer
 
    // Redirect std::cout and std::cerr to the stringstream buffer
    std::cout.rdbuf(sss_cout.rdbuf());
    std::cerr.rdbuf(sss_cerr.rdbuf());
  }
 
  try {
    // Call the provided function
    func(std::forward<Args>(args)...);
 
    // Restore the original stream buffers
    if (coutBuf) std::cout.rdbuf(coutBuf);
    if (cerrBuf) std::cerr.rdbuf(cerrBuf);
 
    std::string line;
    while (std::getline(sss_cout, line))
    {
      auto formatted_message = fmt::format("[{}] {}", LibName, line);
      logFunction(formatted_message);
    }
    while (std::getline(sss_cerr, line))
    {
      auto formatted_message = fmt::format("[{}] {}", LibName, line);
      logFunction(formatted_message);
    }
  } catch (std::runtime_error &err) {
    // Restore the original buffers in case of an exception
    std::cout.rdbuf(coutBuf);
    std::cerr.rdbuf(cerrBuf);
 
    std::cout << "\nConsole cout output:" << std::endl;
    std::cout << sss_cout.rdbuf()->str() << std::endl;
 
    std::cout << "\nConsole cerr output:" << std::endl;
    std::cout << sss_cerr.rdbuf()->str() << std::endl;
    throw;
  }
}