git.mutantstargoat.com Git - laserbrain_demo/blob - src/logger.cc

   1 #include <stdio.h>
   2 #include <stdarg.h>
   3 #include "logger.h"
   4 #include "ui.h"
   5
   6 #if defined(unix) || defined(__unix__) || defined(__APPLE__)
   7 #include <unistd.h>
   8 #elif defined(WIN32)
   9 #include <windows.h>
  10 #endif
  11
  12 #define UI_MSG_TIMEOUT 4000
  13
  14 enum { LOG_INFO, LOG_WARNING, LOG_ERROR, LOG_FATAL, LOG_DEBUG };
  15
  16 static int typecolor(int type);
  17 static const char *typeprefix(int type);
  18
  19 static FILE *fp = stdout;
  20 static FILE *logfile;
  21
  22 static void logmsg(int type, const char *fmt, va_list ap)
  23 {
  24         va_list ap_orig;
  25
  26         va_copy(ap_orig, ap);
  27
  28 #if defined(unix) || defined(__unix__) || (defined(__APPLE__) && !defined(TARGET_IPHONE))
  29         if(isatty(fileno(fp)) && type != LOG_INFO) {
  30                 int c = typecolor(type);
  31                 fprintf(fp, "\033[%dm", c);
  32                 vfprintf(fp, fmt, ap);
  33                 fprintf(fp, "\033[0m");
  34         } else
  35 #endif
  36         {
  37                 fprintf(fp, "%s", typeprefix(type));
  38                 vfprintf(fp, fmt, ap);
  39         }
  40         if(type == LOG_ERROR || type == LOG_FATAL || type == LOG_DEBUG) {
  41                 fflush(fp);
  42         }
  43
  44         if(logfile) {
  45                 va_end(ap);
  46                 va_copy(ap, ap_orig);
  47                 fprintf(logfile, "%s", typeprefix(type));
  48                 vfprintf(logfile, fmt, ap);
  49         }
  50
  51 #ifdef WIN32
  52         if(type == LOG_FATAL) {
  53                 static char msgbuf[1024];
  54                 vsnprintf(msgbuf, sizeof msgbuf - 1, fmt, ap_orig);
  55                 msgbuf[sizeof msgbuf - 1] = 0;
  56                 MessageBox(0, msgbuf, "Fatal error", MB_OK | MB_ICONSTOP);
  57         }
  58 #endif
  59
  60         va_end(ap_orig);
  61 }
  62
  63 static void close_logfile(void)
  64 {
  65         if(logfile) fclose(logfile);
  66 }
  67
  68 extern "C" void set_log_file(const char *fname)
  69 {
  70         static int init_once;
  71
  72         close_logfile();
  73         logfile = fopen(fname, "w");
  74
  75         if(!init_once) {
  76                 atexit(close_logfile);
  77                 init_once = 1;
  78         }
  79 }
  80
  81 extern "C" void info_log(const char *fmt, ...)
  82 {
  83         va_list ap;
  84
  85         va_start(ap, fmt);
  86         logmsg(LOG_INFO, fmt, ap);
  87         va_end(ap);
  88 }
  89
  90 extern "C" void warning_log(const char *fmt, ...)
  91 {
  92         va_list ap;
  93
  94         va_start(ap, fmt);
  95         logmsg(LOG_WARNING, fmt, ap);
  96         va_end(ap);
  97
  98         va_start(ap, fmt);
  99         show_messagev(UI_MSG_TIMEOUT, Vec3(1.0, 0.8, 0.1), fmt, ap);
 100         va_end(ap);
 101 }
 102
 103 extern "C" void error_log(const char *fmt, ...)
 104 {
 105         va_list ap;
 106
 107         va_start(ap, fmt);
 108         logmsg(LOG_ERROR, fmt, ap);
 109         va_end(ap);
 110
 111         va_start(ap, fmt);
 112         show_messagev(UI_MSG_TIMEOUT, Vec3(1.0, 0.1, 0.1), fmt, ap);
 113         va_end(ap);
 114 }
 115
 116 extern "C" void fatal_log(const char *fmt, ...)
 117 {
 118         va_list ap;
 119
 120         va_start(ap, fmt);
 121         logmsg(LOG_FATAL, fmt, ap);
 122         va_end(ap);
 123 }
 124
 125 extern "C" void debug_log(const char *fmt, ...)
 126 {
 127         va_list ap;
 128
 129         va_start(ap, fmt);
 130         logmsg(LOG_DEBUG, fmt, ap);
 131         va_end(ap);
 132 }
 133
 134 enum {
 135         BLACK = 0,
 136         RED,
 137         GREEN,
 138         YELLOW,
 139         BLUE,
 140         MAGENTA,
 141         CYAN,
 142         WHITE
 143 };
 144
 145 #define ANSI_FGCOLOR(x) (30 + (x))
 146 #define ANSI_BGCOLOR(x) (40 + (x))
 147
 148 static int typecolor(int type)
 149 {
 150         switch(type) {
 151         case LOG_ERROR:
 152                 return ANSI_FGCOLOR(RED);
 153         case LOG_FATAL:
 154                 return ANSI_FGCOLOR(RED);       // TODO differentiate from LOG_ERROR
 155         case LOG_WARNING:
 156                 return ANSI_FGCOLOR(YELLOW);
 157         case LOG_DEBUG:
 158                 return ANSI_FGCOLOR(MAGENTA);
 159         default:
 160                 break;
 161         }
 162         return 37;
 163 }
 164
 165 static const char *typeprefix(int type)
 166 {
 167         switch(type) {
 168         case LOG_ERROR:
 169                 return "E: ";
 170         case LOG_FATAL:
 171                 return "F: ";
 172         case LOG_WARNING:
 173                 return "W: ";
 174         case LOG_DEBUG:
 175                 return "D: ";
 176         default:
 177                 break;
 178         }
 179         return "";
 180 }