LogGuard.hh

#ifndef VTESTBED_GUI_LOGGUARD_HH
#define VTESTBED_GUI_LOGGUARD_HH

#include <iostream>
#include <streambuf>

#include <vtestbed/gui/CircularLineBuffer.hh>
#include <vtestbed/gui/LogStreamBuffer.hh>

namespace vtb {

    namespace gui {

        class LogGuard {

        private:
            CircularLineBuffer _lineBuffer;
            std::streambuf *_oldclog, *_oldcerr, *_oldcout;
            LogStreamBuffer _newclog, _newcerr, _newcout;

        public:

            inline
            LogGuard():
            _oldclog(std::clog.rdbuf()), _oldcerr(std::cerr.rdbuf()),
            _newclog(std::clog.rdbuf(), &_lineBuffer, LogLineMarker::Log),
            _newcerr(std::cerr.rdbuf(), &_lineBuffer, LogLineMarker::Error),
            _newcout(std::cout.rdbuf(), &_lineBuffer, LogLineMarker::Output) {
                std::clog.rdbuf(&_newclog);
                std::cerr.rdbuf(&_newcerr);
                std::cout.rdbuf(&_newcout);
            }

            inline
            ~LogGuard() {
                std::clog.rdbuf(_oldclog);
                std::cerr.rdbuf(_oldcerr);
                std::cout.rdbuf(_oldcout);
            }

            inline CircularLineBuffer& line_buffer() { return this->_lineBuffer; }

        };

    }

}

#endif // vim:filetype=cpp