161 lines
3.5 KiB
C++
161 lines
3.5 KiB
C++
|
|
|
|
#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
|