LogStreamBuffer.hh

#ifndef VTESTBED_GUI_LOGSTREAMBUFFER_HH
#define VTESTBED_GUI_LOGSTREAMBUFFER_HH

#include <streambuf>

#include <vtestbed/gui/CircularLineBuffer.hh>

namespace vtb {

    namespace gui {

        class LogStreamBuffer: public std::streambuf {

        private:
            std::streambuf* sb;
            CircularLineBuffer* ilb;
            LogLineMarker _marker;

        public:

            inline
            LogStreamBuffer(
                std::streambuf* sb,
                CircularLineBuffer* ilb,
                LogLineMarker marker
            ): sb(sb), ilb(ilb), _marker(marker) {}

        protected:

            virtual int
            overflow(int c) {
                if (c == traits_type::eof()) {
                    return !traits_type::eof();
                } else {
                    int const r = sb->sputc(c);
                    const char cc = traits_type::to_char_type(c);
                    ilb->put(&cc, _marker);
                    return r == traits_type::eof() ? traits_type::eof() : c;
                }
            }

            virtual std::streamsize
            xsputn(const char* s, std::streamsize n){
                sb->sputn(s, n);
                ilb->put(s, _marker, n);
                return n;
            }

        };

    }

}

#endif // vim:filetype=cpp