#include <iomanip>
#include <iostream>
#include <sstream>
-#include <streambuf>
#include "misc.h"
#include "thread.h"
}
-/// Our fancy logging facility. The trick here is to replace cout.rdbuf() with
-/// this one that sends the output both to console and to a file, this allow us
-/// to toggle the logging of std::cout to a file while preserving output to
-/// stdout and without changing a single line of code! Idea and code from:
-/// http://groups.google.com/group/comp.lang.c++/msg/1d941c0f26ea0d81
+/// Our fancy logging facility. The trick here is to replace cin.rdbuf() and
+/// cout.rdbuf() with two Tie objects that tie cin and cout to a file stream. We
+/// can toggle the logging of std::cout and std:cin at runtime while preserving
+/// usual i/o functionality and without changing a single line of code!
+/// Idea from http://groups.google.com/group/comp.lang.c++/msg/1d941c0f26ea0d81
-class Tee: public streambuf {
-public:
- typedef char_traits<char> traits_type;
- typedef traits_type::int_type int_type;
-
- Tee(ios& s, ofstream& f) : stream(s), file(f), stream_buf(s.rdbuf()) {}
- ~Tee() { set(false); }
-
- void set(bool b) { stream.rdbuf(b ? this : stream_buf); }
-
-private:
- int_type overflow(int_type c) {
-
- if (traits_type::eq_int_type(c, traits_type::eof()))
- return traits_type::not_eof(c);
-
- c = stream_buf->sputc(traits_type::to_char_type(c));
+class Logger {
- if (!traits_type::eq_int_type(c, traits_type::eof()))
- c = file.rdbuf()->sputc(traits_type::to_char_type(c));
+ Logger() : in(cin.rdbuf(), file), out(cout.rdbuf(), file) {}
+ ~Logger() { start(false); }
- return c;
- }
+ struct Tie: public streambuf { // MSVC requires splitted streambuf for cin and cout
- int sync() {
+ Tie(streambuf* b, ofstream& f) : buf(b), file(f) {}
- int c = stream_buf->pubsync();
+ int sync() { return file.rdbuf()->pubsync(), buf->pubsync(); }
+ int overflow(int c) { return log(buf->sputc((char)c), "<< "); }
+ int underflow() { return buf->sgetc(); }
+ int uflow() { return log(buf->sbumpc(), ">> "); }
- if (c != -1)
- c = file.rdbuf()->pubsync();
-
- return c;
- }
+ int log(int c, const char* prefix) {
- int underflow() { return traits_type::not_eof(stream_buf->sgetc()); }
+ static int last = '\n';
- int uflow() {
+ if (last == '\n')
+ file.rdbuf()->sputn(prefix, 3);
- int c = stream_buf->sbumpc();
-
- if (!traits_type::eq_int_type(c, traits_type::eof()))
- file.rdbuf()->sputc(traits_type::to_char_type(c));
+ return last = file.rdbuf()->sputc((char)c);
+ }
- return traits_type::not_eof(c);
- }
+ streambuf* buf;
+ ofstream& file;
+ };
- ios& stream;
- ofstream& file;
- streambuf* stream_buf;
-};
+ ofstream file;
+ Tie in, out;
-class Logger {
public:
- Logger() : in(cin, file), out(cout, file) {}
- ~Logger() { set(false); }
+ static void start(bool b) {
- void set(bool b) {
+ static Logger l;
- if (b && !file.is_open())
+ if (b && !l.file.is_open())
{
- file.open("io_log.txt", ifstream::out | ifstream::app);
- in.set(true);
- out.set(true);
+ l.file.open("io_log.txt", ifstream::out | ifstream::app);
+ cin.rdbuf(&l.in);
+ cout.rdbuf(&l.out);
}
- else if (!b && file.is_open())
+ else if (!b && l.file.is_open())
{
- out.set(false);
- in.set(false);
- file.close();
+ cout.rdbuf(l.out.buf);
+ cin.rdbuf(l.in.buf);
+ l.file.close();
}
}
-
-private:
- Tee in, out;
- ofstream file;
};
-void logger_set(bool b) {
- static Logger l;
- l.set(b);
-}
+/// Trampoline helper to avoid moving Logger to misc.h
+void start_logger(bool b) { Logger::start(b); }
/// cpu_count() tries to detect the number of CPU cores