]> git.sesse.net Git - stockfish/blobdiff - src/misc.cpp
Remove cruft from Logger class
[stockfish] / src / misc.cpp
index 8cfba7341fa7fe7256fdaafa440dc3140ae9dbcd..bdb2cc27c4d31372165ebfff8e073f535c913064 100644 (file)
@@ -105,6 +105,62 @@ void dbg_print() {
 }
 
 
+/// Our fancy logging facility. The trick here is to replace cin.rdbuf() and
+/// cout.rdbuf() with this one that tees cin and cout to a file stream. We can
+/// toggle the logging of std::cout and std:cin at runtime while preserving i/o
+/// functionality and without changing a single line of code!
+/// Idea from http://groups.google.com/group/comp.lang.c++/msg/1d941c0f26ea0d81
+
+class Logger: public streambuf {
+
+  Logger() : cinbuf(cin.rdbuf()), coutbuf(cout.rdbuf()) {}
+  ~Logger() { start(false); }
+
+public:
+  static void start(bool b) {
+
+    static Logger l;
+
+    if (b && !l.file.is_open())
+    {
+        l.file.open("io_log.txt", ifstream::out | ifstream::app);
+        cin.rdbuf(&l);
+        cout.rdbuf(&l);
+    }
+    else if (!b && l.file.is_open())
+    {
+        cout.rdbuf(l.coutbuf);
+        cin.rdbuf(l.cinbuf);
+        l.file.close();
+    }
+  }
+
+private:
+  int sync() { return file.rdbuf()->pubsync(), coutbuf->pubsync(); }
+  int overflow(int c) { return log(coutbuf->sputc((char)c), "<< ") ; }
+  int underflow() { return cinbuf->sgetc(); }
+  int uflow() { return log(cinbuf->sbumpc(), ">> "); }
+
+  int log(int c, const char* prefix) {
+
+    static int last = '\n';
+
+    if (last == '\n')
+        file.rdbuf()->sputn(prefix, 3);
+
+    return last = file.rdbuf()->sputc((char)c);
+  }
+
+private:
+  ofstream file;
+  streambuf *cinbuf, *coutbuf;
+};
+
+
+/// Trampoline helper to avoid moving Logger to misc.h header
+void start_logger(bool b) { Logger::start(b); }
+
+
 /// cpu_count() tries to detect the number of CPU cores
 
 int cpu_count() {