Logger.h

Go to the documentation of this file.

 
#pragma once
 
#include <spdlog/fmt/bundled/ranges.h>
#include <spdlog/sinks/basic_file_sink.h>
#include <spdlog/sinks/ostream_sink.h>
#include <spdlog/sinks/stdout_color_sinks.h>
#include <spdlog/spdlog.h>
 
#include <iostream>
 
#define AutoPasLog(lvl, fmt, ...) SPDLOG_LOGGER_##lvl(spdlog::get("AutoPasLog"), fmt, ##__VA_ARGS__)
 
namespace autopas {
class Logger {
 private:
  static inline auto loggerName() { return "AutoPasLog"; };
 
 public:
  using LogLevel = spdlog::level::level_enum;
  static void create(std::string &filename) {
    // drop an already registered Logger if it exists
    if (spdlog::get(loggerName())) {
      unregister();
    }
    spdlog::basic_logger_mt(loggerName(), filename);
  }
 
  static void create(std::ostream &oss = std::cout) {
    // drop an already registered Logger if it exists
    if (spdlog::get(loggerName())) {
      unregister();
    }
    std::shared_ptr<spdlog::sinks::sink> ostreamSink;
#ifdef AUTOPAS_COLORED_CONSOLE_LOGGING
    if (oss.rdbuf() == std::cout.rdbuf()) {
      ostreamSink = std::make_shared<spdlog::sinks::stdout_color_sink_mt>();
    } else if (oss.rdbuf() == std::cerr.rdbuf()) {
      ostreamSink = std::make_shared<spdlog::sinks::stderr_color_sink_mt>();
    } else {  // no color for streams other than cout / cerr
      ostreamSink = std::make_shared<spdlog::sinks::ostream_sink_mt>(oss);
    }
#else
    ostreamSink = std::make_shared<spdlog::sinks::ostream_sink_mt>(oss);
#endif
    auto logger = std::make_shared<spdlog::logger>(loggerName(), ostreamSink);
    spdlog::register_logger(logger);
  }
 
  static void unregister() { spdlog::drop(loggerName()); }
 
  static auto get() { return spdlog::get(loggerName()); }
};  // class Logger
}  // namespace autopas