ConsoleStream.h

Go to the documentation of this file.

/*****************************************************************************
 * $CAMITK_LICENCE_BEGIN$
 *
 * CamiTK - Computer Assisted Medical Intervention ToolKit
 * (c) 2001-2018 Univ. Grenoble Alpes, CNRS, TIMC-IMAG UMR 5525 (GMCAO)
 *
 * Visit http://camitk.imag.fr for more information
 *
 * This file is part of CamiTK.
 *
 * CamiTK is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License version 3
 * only, as published by the Free Software Foundation.
 *
 * CamiTK is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License version 3 for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * version 3 along with CamiTK.  If not, see <http://www.gnu.org/licenses/>.
 *
 * $CAMITK_LICENCE_END$
 ****************************************************************************/

#ifndef ConsoleStream_H
#define ConsoleStream_H

// -- stl stuff
#include <iostream>
#include <streambuf>
#include <string>

// -- QT stuff
#include <QTextEdit>

namespace camitk {
class ConsoleStream : public std::basic_streambuf<char> {
public:
    ConsoleStream(std::ostream* stream, QTextEdit* textEdit) {
        init(stream, textEdit);
    }

    ConsoleStream() {
        previousBuffer = nullptr;
        logTextEdit = nullptr;
        myStream = nullptr;
    }

    ~ConsoleStream() {
        // output anything that is left
        if (!myString.empty()) {
            logTextEdit->append(myString.c_str());
        }

        free();
    }

    void setStream(std::ostream* stream) {
        free();
        myStream = stream;
        previousBuffer = stream->rdbuf();
        stream->rdbuf(this);
    }

    void setTextEdit(QTextEdit* text_edit) {
        logTextEdit = text_edit;
    }

    void init(std::ostream* stream, QTextEdit* textEdit) {
        setTextEdit(textEdit);
        setStream(stream);
    }

    void free() {
        if (previousBuffer != nullptr && myStream != nullptr) {
            myStream->rdbuf(previousBuffer);
        }
    }
protected:
    int_type overflow(int_type v) {
        if (v == '\n') {
            logTextEdit->append(myString.c_str());
            myString.erase(myString.begin(), myString.end());
        }
        else {
            myString += v;
        }

        return v;
    }

    std::streamsize xsputn(const char* p, std::streamsize n) {
        myString.append(p, p + n);

        std::string::size_type pos = 0;
        while (pos != std::string::npos) {
            pos = myString.find('\n');
            if (pos != std::string::npos) {
                std::string tmp(myString.begin(), myString.begin() + pos);
                logTextEdit->append(tmp.c_str());
                myString.erase(myString.begin(), myString.begin() + pos + 1);
            }
        }

        return n;
    }

private:
    std::ostream* myStream;
    std::streambuf* previousBuffer;
    std::string myString;
    QTextEdit* logTextEdit;
};

}



#endif