Skip to content

Commit 67288a0

Browse files
committed
Allow verbose logging
1 parent 2ea00b4 commit 67288a0

3 files changed

Lines changed: 10 additions & 3 deletions

File tree

include/logging.h

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,13 +8,14 @@
88

99
#include "stable-diffusion.h"
1010

11-
enum class LogLevel { Debug, Info, Warning, Error };
11+
enum class LogLevel { Verbose, Debug, Info, Warning, Error };
1212

1313
void log_message(LogLevel level, const char* format, ...);
1414
void set_log_level(LogLevel level);
1515
void set_log_level(const std::string& level_str);
1616

1717
// Convenience macros for logging
18+
#define LOG_VERBOSE(...) log_message(LogLevel::Verbose, __VA_ARGS__)
1819
#define LOG_DEBUG(...) log_message(LogLevel::Debug, __VA_ARGS__)
1920
#define LOG_INFO(...) log_message(LogLevel::Info, __VA_ARGS__)
2021
#define LOG_WARNING(...) log_message(LogLevel::Warning, __VA_ARGS__)

src/logging.cpp

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,8 @@ static std::mutex log_mutex;
1111

1212
const char* log_level_to_string(LogLevel level) {
1313
switch (level) {
14+
case LogLevel::Verbose:
15+
return "VERBOSE";
1416
case LogLevel::Debug:
1517
return "DEBUG";
1618
case LogLevel::Info:
@@ -26,6 +28,8 @@ const char* log_level_to_string(LogLevel level) {
2628

2729
const char* log_level_to_color(LogLevel level) {
2830
switch (level) {
31+
case LogLevel::Verbose:
32+
return "\033[35m"; // Magenta
2933
case LogLevel::Debug:
3034
return "\033[36m"; // Cyan
3135
case LogLevel::Info:
@@ -89,7 +93,9 @@ void set_log_level(LogLevel level) {
8993
void set_log_level(const std::string& level_str) {
9094
LogLevel level = LogLevel::Info; // default
9195

92-
if (level_str == "debug") {
96+
if (level_str == "verbose") {
97+
level = LogLevel::Verbose;
98+
} else if (level_str == "debug") {
9399
level = LogLevel::Debug;
94100
} else if (level_str == "info") {
95101
level = LogLevel::Info;

src/main.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ void print_usage(const char* program_name) {
3737
std::cerr << std::endl;
3838
std::cerr << "Options:" << std::endl;
3939
std::cerr << " --port <port> Server port (default: 8188)" << std::endl;
40-
std::cerr << " --log-level <level> Log level: debug, info, warning, error (default: info)"
40+
std::cerr << " --log-level <level> Log level: verbose, debug, info, warning, error (default: info)"
4141
<< std::endl;
4242
std::cerr << " --parent-pid <pid> Parent process PID" << std::endl;
4343
std::cerr << " --ckpt-dir <path> Checkpoint models directory" << std::endl;

0 commit comments

Comments
 (0)