/// 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 Logger {
+struct Tie: public streambuf { // MSVC requires splitted streambuf for cin and cout
- Logger() : in(cin.rdbuf(), file), out(cout.rdbuf(), file) {}
- ~Logger() { start(false); }
+ Tie(streambuf* b, ofstream* f) : buf(b), file(f) {}
- struct Tie: public streambuf { // MSVC requires splitted streambuf for cin and cout
+ 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(), ">> "); }
- Tie(streambuf* b, ofstream& f) : buf(b), file(f) {}
+ streambuf* buf;
+ ofstream* file;
- 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(), ">> "); }
+ int log(int c, const char* prefix) {
- int log(int c, const char* prefix) {
+ static int last = '\n';
- static int last = '\n';
+ if (last == '\n')
+ file->rdbuf()->sputn(prefix, 3);
- if (last == '\n')
- file.rdbuf()->sputn(prefix, 3);
+ return last = file->rdbuf()->sputc((char)c);
+ }
+};
- return last = file.rdbuf()->sputc((char)c);
- }
+class Logger {
- streambuf* buf;
- ofstream& file;
- };
+ Logger() : in(cin.rdbuf(), &file), out(cout.rdbuf(), &file) {}
+ ~Logger() { start(false); }
ofstream file;
Tie in, out;
};
+/// Used to serialize access to std::cout to avoid multiple threads to write at
+/// the same time.
+
+std::ostream& operator<<(std::ostream& os, SyncCout sc) {
+
+ static Mutex m;
+
+ if (sc == io_lock)
+ m.lock();
+
+ if (sc == io_unlock)
+ m.unlock();
+
+ return os;
+}
+
+
/// Trampoline helper to avoid moving Logger to misc.h
void start_logger(bool b) { Logger::start(b); }