Consoles.h

Go to the documentation of this file.

// ----------------------------------------------------------------------------
// -                        CloudViewer: www.cloudViewer.org                  -
// ----------------------------------------------------------------------------
// Copyright (c) 2018-2024 www.cloudViewer.org
// SPDX-License-Identifier: MIT
// ----------------------------------------------------------------------------
 
#pragma once
 
#include <QListWidget>
#include <QString>
 
#include <CVLog.h>
 
#include <functional>
#include <utility>
 
class ConsoleWrapper
{
  public:
    explicit ConsoleWrapper(std::function<void(const QString &)> printFn)
        : m_printFn(std::move(printFn))
    {
    }
 
    void write(const char *messagePart)
    {
 
        const size_t len = strlen(messagePart);
        const char *messageEnd = messagePart + len;
        const char *start = messagePart;
 
        const char *newLinePos;
        while ((newLinePos = strchr(start, '\n')) != nullptr)
        {
            m_currentMessage += QString::fromUtf8(start, static_cast<int>(newLinePos - start));
            m_printFn(m_currentMessage);
            m_currentMessage.clear();
            start = newLinePos + 1;
        }
 
        if (start != messageEnd)
        {
            m_currentMessage += QString::fromUtf8(start, static_cast<int>(messageEnd - start));
        }
    }
 
    void flush() const
    {
        m_printFn(m_currentMessage);
    }
 
  private:
    QString m_currentMessage{};
    std::function<void(const QString &)> m_printFn;
};
 
class ccConsoleOutput
{
  public:
    ccConsoleOutput() = default;
    explicit ccConsoleOutput(const char *prefix) : m_prefix(prefix) {}
 
    void write(const char *messagePart)
    {
        if (!messagePart)
        {
            CVLog::Warning("Found invalid message given!");
        }
        else
        {
            m_output.write(messagePart);
        }
    }
 
    bool isatty() const
    {
        return false;
    }
 
    void flush() const
    {
        m_output.flush();
    }
 
  private:
    const QString m_prefix;
    ConsoleWrapper m_output{[this](const QString &message)
                            {
                                if (m_prefix.isEmpty())
                                {
                                    CVLog::Print(message);
                                }
                                else
                                {
                                    CVLog::Print(m_prefix + message);
                                }
                            }};
};
 
class ListWidgetConsole
{
  public:
    ListWidgetConsole(QListWidget *view, const Qt::GlobalColor color) : m_view(view), m_brush(color)
    {
    }
    ListWidgetConsole(QListWidget *view, const QColor &color) : m_view(view), m_brush() {}
    explicit ListWidgetConsole(QListWidget *view) : m_view(view), m_brush() {}
 
    void write(const char *messagePart)
    {
        if (!messagePart)
        {
            CVLog::Warning("Found invalid message given!");
        }
        else
        {
            m_output.write(messagePart);
        }
    }
    bool isatty() const
    {
        return false;
    }
 
    void flush() const
    {
        m_output.flush();
    }
 
  private:
    QListWidget *m_view;
    QBrush m_brush;
    QString m_prefix;
    ConsoleWrapper m_output{[this](const QString &message)
                            {
                                if (m_view)
                                {
                                    auto *messageItem = new QListWidgetItem(message);
                                    messageItem->setForeground(m_brush);
                                    m_view->addItem(messageItem);
                                }
                                else
                                {
                                    if (m_prefix.isEmpty())
                                    {
                                        CVLog::Print(message);
                                    }
                                    else
                                    {
                                        CVLog::Print(m_prefix + message);
                                    }
                                }
                            }};
};