qt-migration-legatus/qt6-after/qt6-migrated-legatus/Lts/Common/LtsNotification/lts_notification_ctl.cpp

#include "lts_notification_ctl.h"

#include <QtGui/QColor>
#include <QtCore/QDebug>
#include <QtGui/QIcon>
#include <QtGui/QImage>
#include <QtGui/QPainter>
#include <QtWidgets/QToolButton>
#include <utility>

namespace LtsNotification
{

LtsNotificationController::LtsNotificationController(const QString& identifier)
    : QObject(), m_count(0), m_identifier(QStringLiteral("LtsNotification%1").arg(identifier))
{
}

LtsNotificationController::~LtsNotificationController()
{
    qDeleteAll(m_notiList);
}

void LtsNotificationController::setIcon(const QIcon& icon)
{
    m_icon = icon;

    setAllIcons();
}

void LtsNotificationController::setCount(const int count)
{
    m_count = count;

    setAllIcons();
}

void LtsNotificationController::setBackground(const int r, const int g, const int b)
{

    for (QToolButton* tb : std::as_const(m_notiList))
    {

        QString stylesheet;

        if (r == -1 && g == -1 && b == -1)
            stylesheet = QStringLiteral("");
        else
            stylesheet = QStringLiteral("QToolButton {background-color: rgb(") + QString::number(r) +
                         QStringLiteral(",") + QString::number(g) + QStringLiteral(",") + QString::number(b) +
                         QStringLiteral(");}");

        tb->setStyleSheet(stylesheet);
    }
}

void LtsNotificationController::setEnabled(bool enable)
{
    m_enable = enable;

    setAllIcons();
}

void LtsNotificationController::setAction(const std::function<void(void)>& action)
{
    m_action = action;
}

void LtsNotificationController::setToolTip(const QString& toolTip)
{
    m_toolTip = toolTip;

    for (QToolButton* tb : std::as_const(m_notiList))
    {
        tb->setToolTip(toolTip);
        tb->setIconSize(QSize(30, 30));
    }
}

void LtsNotificationController::setAllIcons()
{
    if (m_icon.isNull())
        return;

    QIcon icon = modifyIcon(m_icon);

    for (QToolButton* tb : std::as_const(m_notiList))
    {
        tb->setIcon(icon);
        tb->setIconSize(QSize(30, 30));
        tb->setEnabled(m_enable);
    }
}

#define SIZE 40

QIcon LtsNotificationController::modifyIcon(const QIcon& icon)
{

    if (m_count < 1)
        return icon;

    QPixmap iconImage(icon.pixmap(SIZE * 8 / 10, SIZE * 8 / 10));
    QPainter painter(&iconImage);

    painter.save();
    painter.setRenderHints(QPainter::TextAntialiasing | QPainter::Antialiasing | QPainter::SmoothPixmapTransform);

    QFont f;
    f.setFamily(QStringLiteral("Monospace"));
    f.setPointSize(16);
    f.setBold(true);
    painter.setFont(f);

    QColor color(Qt::red);
    painter.setBrush(color);
    painter.setPen(QPen(color, 3.1));
    painter.drawText(QRect(0, 0, 30, 30), QString::number(m_count));
    painter.restore();

    return iconImage;
}

QToolButton* LtsNotificationController::createWidget()
{
    QToolButton* b = new QToolButton();
    b->setToolButtonStyle(Qt::ToolButtonIconOnly);
    b->setObjectName(m_identifier);

    b->setMinimumSize(SIZE, SIZE);
    b->setMaximumSize(SIZE, SIZE);

    connect(b, &QToolButton::destroyed, this, &LtsNotificationController::slotNotiWidgetDestroyed);
    connect(b, &QToolButton::clicked, this, [this]() {
        if (m_action)
            m_action();
    });

    if (!m_icon.isNull())
        b->setIcon(modifyIcon(m_icon));

    b->setToolTip(m_toolTip);

    m_notiList.append(b);
    return b;
}

void LtsNotificationController::slotNotiWidgetDestroyed(QObject* obj)
{
    m_notiList.removeAll(static_cast<QToolButton*>(obj));
}

} // namespace LtsNotification