mirror of
https://github.com/momo5502/emulator.git
synced 2026-01-19 03:33:56 +00:00
101 lines
2.0 KiB
C++
101 lines
2.0 KiB
C++
#include "std_include.hpp"
|
|
#include "logger.hpp"
|
|
|
|
#include <utils/finally.hpp>
|
|
|
|
namespace
|
|
{
|
|
#ifdef _WIN32
|
|
#define COLOR(win, posix) win
|
|
using color_type = WORD;
|
|
#else
|
|
#define COLOR(win, posix) posix
|
|
using color_type = const char*;
|
|
#endif
|
|
|
|
color_type get_reset_color()
|
|
{
|
|
return COLOR(7, "\033[0m");
|
|
}
|
|
|
|
color_type get_color_type(const color c)
|
|
{
|
|
using enum color;
|
|
|
|
switch (c)
|
|
{
|
|
case black: return COLOR(0x8, "\033[0;90m");
|
|
case red: return COLOR(0xC, "\033[0;91m");
|
|
case green: return COLOR(0xA, "\033[0;92m");
|
|
case yellow: return COLOR(0xE, "\033[0;93m");
|
|
case blue: return COLOR(0x9, "\033[0;94m");
|
|
case cyan: return COLOR(0xB, "\033[0;96m");
|
|
case pink: return COLOR(0xD, "\033[0;95m");
|
|
case white: return COLOR(0xF, "\033[0;97m");
|
|
case dark_gray: return COLOR(0x8, "\033[0;97m");
|
|
case gray:
|
|
default: return get_reset_color();
|
|
}
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
HANDLE get_console_handle()
|
|
{
|
|
return GetStdHandle(STD_OUTPUT_HANDLE);
|
|
}
|
|
#endif
|
|
|
|
void set_color(const color_type color)
|
|
{
|
|
#ifdef _WIN32
|
|
SetConsoleTextAttribute(get_console_handle(), color);
|
|
#else
|
|
printf("%s", color);
|
|
#endif
|
|
}
|
|
|
|
void reset_color()
|
|
{
|
|
(void)fflush(stdout);
|
|
set_color(get_reset_color());
|
|
(void)fflush(stdout);
|
|
}
|
|
|
|
std::string_view format(va_list* ap, const char* message)
|
|
{
|
|
thread_local char buffer[0x1000];
|
|
|
|
#ifdef _WIN32
|
|
const int count = _vsnprintf_s(buffer, sizeof(buffer), sizeof(buffer), message, *ap);
|
|
#else
|
|
const int count = vsnprintf(buffer, sizeof(buffer), message, *ap);
|
|
#endif
|
|
|
|
if (count < 0) return {};
|
|
return {buffer, static_cast<size_t>(count)};
|
|
}
|
|
|
|
void print_colored(const std::string_view& line, const color_type base_color)
|
|
{
|
|
const auto _ = utils::finally(&reset_color);
|
|
set_color(base_color);
|
|
(void)fwrite(line.data(), 1, line.size(), stdout);
|
|
}
|
|
}
|
|
|
|
void logger::print(const color c, const char* message, ...) const
|
|
{
|
|
if (this->disable_output_)
|
|
{
|
|
return;
|
|
}
|
|
|
|
va_list ap;
|
|
va_start(ap, message);
|
|
|
|
const auto data = format(&ap, message);
|
|
print_colored(data, get_color_type(c));
|
|
|
|
va_end(ap);
|
|
}
|