mirror of
https://github.com/optiscaler/OptiScaler.git
synced 2026-05-12 20:51:00 +00:00
cdozdil
119 lines
3.3 KiB
C++
119 lines
3.3 KiB
C++
#include "pch.h"
|
|
#include "spdlog/sinks/basic_file_sink.h"
|
|
#include "spdlog/sinks/stdout_color_sinks.h"
|
|
|
|
bool InitializeConsole()
|
|
{
|
|
// Allocate a console for this app
|
|
if (!AllocConsole()) {
|
|
return false;
|
|
}
|
|
|
|
FILE* pFile;
|
|
|
|
// Redirect STDIN if the console has an input handle
|
|
if (GetStdHandle(STD_INPUT_HANDLE) != INVALID_HANDLE_VALUE) {
|
|
if (freopen_s(&pFile, "CONIN$", "r", stdin) != 0) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
// Redirect STDOUT if the console has an output handle
|
|
if (GetStdHandle(STD_OUTPUT_HANDLE) != INVALID_HANDLE_VALUE) {
|
|
if (freopen_s(&pFile, "CONOUT$", "w", stdout) != 0) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
// Redirect STDERR if the console has an error handle
|
|
if (GetStdHandle(STD_ERROR_HANDLE) != INVALID_HANDLE_VALUE) {
|
|
if (freopen_s(&pFile, "CONOUT$", "w", stderr) != 0) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
// Clear the error state for each of the C++ standard streams
|
|
std::cin.clear();
|
|
std::cout.clear();
|
|
std::cerr.clear();
|
|
std::wcin.clear();
|
|
std::wcout.clear();
|
|
std::wcerr.clear();
|
|
|
|
// Make C++ standard streams point to console as well.
|
|
std::ios::sync_with_stdio();
|
|
|
|
return true;
|
|
}
|
|
|
|
void PrepareLogger()
|
|
{
|
|
try
|
|
{
|
|
auto config = Config(L"nvngx.ini");
|
|
|
|
if (config.LoggingEnabled.value_or(true))
|
|
{
|
|
if (config.OpenConsole.value_or(false))
|
|
InitializeConsole();
|
|
|
|
std::shared_ptr<spdlog::logger> shared_logger = nullptr;
|
|
|
|
if (config.LogToConsole.value_or(true) && config.LogToFile.value_or(false))
|
|
{
|
|
auto console_sink = std::make_shared<spdlog::sinks::stdout_color_sink_mt>();
|
|
console_sink->set_level(spdlog::level::level_enum::info);
|
|
console_sink->set_pattern("[XeSS] [%H:%M:%S.%f] [%L] %v");
|
|
|
|
auto file_sink = std::make_shared<spdlog::sinks::basic_file_sink_mt>(config.LogFileName.value(), true);
|
|
file_sink->set_level((spdlog::level::level_enum)config.LogLevel.value_or(2));
|
|
file_sink->set_pattern("[%H:%M:%S.%f] [%L] %v");
|
|
|
|
spdlog::logger logger("multi_sink", { console_sink, file_sink });
|
|
shared_logger = std::make_shared<spdlog::logger>(logger);
|
|
}
|
|
else if (config.LogToFile.value_or(false))
|
|
{
|
|
auto file_sink = std::make_shared<spdlog::sinks::basic_file_sink_mt>(config.LogFileName.value(), true);
|
|
file_sink->set_level((spdlog::level::level_enum)config.LogLevel.value_or(2));
|
|
file_sink->set_pattern("[%H:%M:%S.%f] [%L] %v");
|
|
|
|
spdlog::logger logger("file_sink", { file_sink });
|
|
shared_logger = std::make_shared<spdlog::logger>(logger);
|
|
}
|
|
else if (config.LogToConsole.value_or(true))
|
|
{
|
|
auto console_sink = std::make_shared<spdlog::sinks::stdout_color_sink_mt>();
|
|
console_sink->set_level(spdlog::level::level_enum::info);
|
|
console_sink->set_pattern("[XeSS] [%H:%M:%S.%f] [%L] %v");
|
|
|
|
spdlog::logger logger("console_sink", { console_sink });
|
|
shared_logger = std::make_shared<spdlog::logger>(logger);
|
|
}
|
|
|
|
if (shared_logger)
|
|
{
|
|
shared_logger->set_level(spdlog::level::level_enum::trace);
|
|
shared_logger->flush_on(spdlog::level::err);
|
|
spdlog::set_default_logger(shared_logger);
|
|
}
|
|
}
|
|
}
|
|
catch (const spdlog::spdlog_ex& ex)
|
|
{
|
|
std::cerr << ex.what() << std::endl;
|
|
|
|
auto logger = spdlog::stdout_color_mt("xess");
|
|
logger->set_pattern("[%H:%M:%S.%f] [%L] %v");
|
|
logger->set_level((spdlog::level::level_enum)2);
|
|
spdlog::set_default_logger(logger);
|
|
}
|
|
}
|
|
|
|
void CloseLogger()
|
|
{
|
|
spdlog::default_logger()->flush();
|
|
spdlog::shutdown();
|
|
}
|
|
|