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

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