mirror of
https://github.com/arcan1s/awesome-widgets.git
synced 2025-04-24 15:37:23 +00:00
296 lines
11 KiB
C++
296 lines
11 KiB
C++
/***************************************************************************
|
|
* This file is part of awesome-widgets *
|
|
* *
|
|
* awesome-widgets is free software: you can redistribute it and/or *
|
|
* modify it under the terms of the GNU General Public License as *
|
|
* published by the Free Software Foundation, either version 3 of the *
|
|
* License, or (at your option) any later version. *
|
|
* *
|
|
* awesome-widgets 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 General Public License for more details. *
|
|
* *
|
|
* You should have received a copy of the GNU General Public License *
|
|
* along with awesome-widgets. If not, see http://www.gnu.org/licenses/ *
|
|
***************************************************************************/
|
|
|
|
#include "awdataaggregator.h"
|
|
|
|
#include <KI18n/KLocalizedString>
|
|
|
|
#include <QBuffer>
|
|
#include <QGraphicsScene>
|
|
#include <QGraphicsView>
|
|
|
|
#include <cmath>
|
|
|
|
#include "awactions.h"
|
|
#include "awdebug.h"
|
|
|
|
|
|
AWDataAggregator::AWDataAggregator(QObject *_parent)
|
|
: QObject(_parent)
|
|
{
|
|
qCDebug(LOG_AW) << __PRETTY_FUNCTION__;
|
|
|
|
m_boundaries["cpuTooltip"] = 100.0;
|
|
m_boundaries["cpuclTooltip"] = 4000.0;
|
|
m_boundaries["memTooltip"] = 100.0;
|
|
m_boundaries["swapTooltip"] = 100.0;
|
|
m_boundaries["downkbTooltip"] = 1.0;
|
|
m_boundaries["upkbTooltip"] = 1.0;
|
|
m_boundaries["batTooltip"] = 100.0;
|
|
|
|
initScene();
|
|
}
|
|
|
|
|
|
AWDataAggregator::~AWDataAggregator()
|
|
{
|
|
qCDebug(LOG_AW) << __PRETTY_FUNCTION__;
|
|
|
|
m_toolTipView->deleteLater();
|
|
m_toolTipScene->deleteLater();
|
|
}
|
|
|
|
|
|
QString AWDataAggregator::htmlImage(const QPixmap &_source)
|
|
{
|
|
QByteArray byteArray;
|
|
QBuffer buffer(&byteArray);
|
|
_source.save(&buffer, "PNG");
|
|
|
|
return byteArray.isEmpty() ? ""
|
|
: QString("<img src=\"data:image/png;base64,%1\"/>").arg(QString(byteArray.toBase64()));
|
|
}
|
|
|
|
|
|
void AWDataAggregator::setParameters(const QVariantMap &_settings)
|
|
{
|
|
qCDebug(LOG_AW) << "Settings" << _settings;
|
|
|
|
// cast from QVariantMap to QVariantHash without data lost
|
|
m_configuration = qvariant_cast<QVariantHash>(_settings);
|
|
|
|
m_enablePopup = m_configuration["notify"].toBool();
|
|
|
|
m_counts = 0;
|
|
m_counts += m_configuration["cpuTooltip"].toInt();
|
|
m_counts += m_configuration["cpuclTooltip"].toInt();
|
|
m_counts += m_configuration["memTooltip"].toInt();
|
|
m_counts += m_configuration["swapTooltip"].toInt();
|
|
m_counts += m_configuration["downkbTooltip"].toInt();
|
|
m_counts += m_configuration["batTooltip"].toInt();
|
|
// resize tooltip image
|
|
m_toolTipView->resize(100 * m_counts, 105);
|
|
|
|
m_requiredKeys.clear();
|
|
if (m_configuration["cpuTooltip"].toBool())
|
|
m_requiredKeys.append("cpuTooltip");
|
|
if (m_configuration["cpuclTooltip"].toBool())
|
|
m_requiredKeys.append("cpuclTooltip");
|
|
if (m_configuration["memTooltip"].toBool())
|
|
m_requiredKeys.append("memTooltip");
|
|
if (m_configuration["swapTooltip"].toBool())
|
|
m_requiredKeys.append("swapTooltip");
|
|
if (m_configuration["downkbTooltip"].toBool())
|
|
m_requiredKeys.append("downkbTooltip");
|
|
if (m_configuration["upkbTooltip"].toBool())
|
|
m_requiredKeys.append("upkbTooltip");
|
|
if (m_configuration["batTooltip"].toBool())
|
|
m_requiredKeys.append("batTooltip");
|
|
|
|
// background
|
|
m_toolTipScene->setBackgroundBrush(m_configuration["useTooltipBackground"].toBool()
|
|
? QColor(m_configuration["tooltipBackground"].toString())
|
|
: Qt::NoBrush);
|
|
}
|
|
|
|
|
|
QPixmap AWDataAggregator::tooltipImage()
|
|
{
|
|
// create image
|
|
m_toolTipScene->clear();
|
|
QPen pen;
|
|
auto shift = 0.0;
|
|
|
|
for (auto i = 0; i < m_requiredKeys.count(); ++i) {
|
|
auto key = m_requiredKeys[i];
|
|
|
|
// create frame
|
|
auto normX = 100.0 / static_cast<float>(m_values[key].count());
|
|
auto normY = 100.0 / (1.5 * m_boundaries[key]);
|
|
|
|
// apply pen color
|
|
if (key != "batTooltip")
|
|
pen.setColor(m_configuration[QString("%1Color").arg(key)].toString());
|
|
|
|
// paint data inside frame
|
|
for (int j = 0; j < m_values[key].count() - 1; j++) {
|
|
// some magic here
|
|
auto x1 = j * normX + shift;
|
|
auto y1 = -std::fabs(m_values[key].at(j)) * normY + 5.0;
|
|
auto x2 = (j + 1) * normX + shift;
|
|
auto y2 = -std::fabs(m_values[key].at(j + 1)) * normY + 5.0;
|
|
// apply color for the battery tooltip based on charge/discharge
|
|
if (key == "batTooltip") {
|
|
if (m_values[key].at(j + 1) > 0)
|
|
pen.setColor(QColor(m_configuration["batTooltipColor"].toString()));
|
|
else
|
|
pen.setColor(QColor(m_configuration["batInTooltipColor"].toString()));
|
|
}
|
|
|
|
m_toolTipScene->addLine(x1, y1, x2, y2, pen);
|
|
}
|
|
|
|
// increase frame shift if not downkbtooltip
|
|
// Additional workaround is required because there is frame (uokb and downkb) which contains two charts
|
|
// with the same shift
|
|
if (key != "downkbTooltip")
|
|
shift += 100.0;
|
|
}
|
|
|
|
return m_toolTipView->grab();
|
|
}
|
|
|
|
|
|
void AWDataAggregator::dataUpdate(const QVariantHash &_values)
|
|
{
|
|
// do not log these arguments
|
|
setData(_values);
|
|
emit(toolTipPainted(htmlImage(tooltipImage())));
|
|
}
|
|
|
|
|
|
void AWDataAggregator::checkValue(const QString &_source, const double _value, const double _extremum) const
|
|
{
|
|
qCDebug(LOG_AW) << "Notification source" << _source << "with value" << _value << "called with extremum"
|
|
<< _extremum;
|
|
|
|
if (_value >= 0.0) {
|
|
if ((m_enablePopup) && (_value > _extremum) && (m_values[_source].last() < _extremum))
|
|
return AWActions::sendNotification("event", notificationText(_source, _value));
|
|
} else {
|
|
if ((m_enablePopup) && (_value < _extremum) && (m_values[_source].last() > _extremum))
|
|
return AWActions::sendNotification("event", notificationText(_source, _value));
|
|
}
|
|
}
|
|
|
|
|
|
void AWDataAggregator::checkValue(const QString &_source, const QString &_current, const QString &_received) const
|
|
{
|
|
qCDebug(LOG_AW) << "Notification source" << _source << "with current value" << _current << "and received one"
|
|
<< _received;
|
|
|
|
if ((m_enablePopup) && (_current != _received) && (!_received.isEmpty()))
|
|
return AWActions::sendNotification("event", notificationText(_source, _received));
|
|
}
|
|
|
|
|
|
void AWDataAggregator::initScene()
|
|
{
|
|
m_toolTipScene = new QGraphicsScene(nullptr);
|
|
m_toolTipView = new QGraphicsView(m_toolTipScene);
|
|
m_toolTipView->setStyleSheet("background: transparent");
|
|
m_toolTipView->setContentsMargins(0, 0, 0, 0);
|
|
m_toolTipView->setFrameShape(QFrame::NoFrame);
|
|
m_toolTipView->setHorizontalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
|
|
m_toolTipView->setVerticalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
|
|
}
|
|
|
|
|
|
QString AWDataAggregator::notificationText(const QString &_source, const float _value)
|
|
{
|
|
qCDebug(LOG_AW) << "Notification source" << _source << "with value" << _value;
|
|
|
|
QString output;
|
|
if (_source == "batTooltip")
|
|
output = _value > 0.0 ? i18n("AC online") : i18n("AC offline");
|
|
else if (_source == "cpuTooltip")
|
|
output = i18n("High CPU load");
|
|
else if (_source == "memTooltip")
|
|
output = i18n("High memory usage");
|
|
else if (_source == "swapTooltip")
|
|
output = i18n("Swap is used");
|
|
else if (_source == "gpu")
|
|
output = i18n("High GPU load");
|
|
|
|
return output;
|
|
}
|
|
|
|
|
|
QString AWDataAggregator::notificationText(const QString &_source, const QString &_value)
|
|
{
|
|
qCDebug(LOG_AW) << "Notification source" << _source << "with value" << _value;
|
|
|
|
QString output;
|
|
if (_source == "netdev")
|
|
output = i18n("Network device has been changed to %1", _value);
|
|
|
|
return output;
|
|
}
|
|
|
|
|
|
void AWDataAggregator::setData(const QVariantHash &_values)
|
|
{
|
|
// do not log these arguments
|
|
// battery update requires info is AC online or not
|
|
setData(_values["ac"].toString() == m_configuration["acOnline"], "batTooltip", _values["bat"].toDouble());
|
|
|
|
// usual case
|
|
setData("cpuTooltip", _values["cpu"].toDouble(), 90.0);
|
|
setData("cpuclTooltip", _values["cpucl"].toDouble());
|
|
setData("memTooltip", _values["mem"].toDouble(), 80.0);
|
|
setData("swapTooltip", _values["swap"].toDouble(), 0.0);
|
|
setData("downkbTooltip", _values["downkb"].toDouble());
|
|
setData("upkbTooltip", _values["upkb"].toDouble());
|
|
|
|
// additional check for network device
|
|
auto currentNetworkDevice = _values["netdev"].toString();
|
|
checkValue("netdev", m_currentNetworkDevice, currentNetworkDevice);
|
|
m_currentNetworkDevice = currentNetworkDevice;
|
|
|
|
// additional check for GPU load
|
|
checkValue("gpu", _values["gpu"].toDouble(), 90.0);
|
|
}
|
|
|
|
|
|
void AWDataAggregator::setData(const QString &_source, double _value, const double _extremum)
|
|
{
|
|
qCDebug(LOG_AW) << "Source" << _source << "to value" << _value << "with extremum" << _extremum;
|
|
|
|
if (m_values[_source].count() == 0)
|
|
m_values[_source].append(0.0);
|
|
else if (m_values[_source].count() > m_configuration["tooltipNumber"].toInt())
|
|
m_values[_source].removeFirst();
|
|
if (std::isnan(_value))
|
|
_value = 0.0;
|
|
|
|
// notifications
|
|
checkValue(_source, _value, _extremum);
|
|
|
|
m_values[_source].append(_value);
|
|
if (_source == "downkbTooltip") {
|
|
// to avoid copying of objects to another list we find max elements in each sequence and compare them
|
|
auto downMax = m_values["downkbTooltip"].empty()
|
|
? 1.0
|
|
: *std::max_element(m_values["downkbTooltip"].cbegin(), m_values["downkbTooltip"].cend());
|
|
auto upMax = m_values["upkbTooltip"].empty()
|
|
? 1.0
|
|
: *std::max_element(m_values["upkbTooltip"].cbegin(), m_values["upkbTooltip"].cend());
|
|
// assign both
|
|
m_boundaries["upkbTooltip"] = m_boundaries["downkbTooltip"] = 1.2 * std::max(downMax, upMax);
|
|
}
|
|
}
|
|
|
|
|
|
void AWDataAggregator::setData(const bool _dontInvert, const QString &_source, double _value)
|
|
{
|
|
qCDebug(LOG_AW) << "Do not invert" << _dontInvert << "value" << _value << "for source" << _source;
|
|
|
|
// invert values for different battery colours
|
|
_value = _dontInvert ? _value : -_value;
|
|
return setData(_source, _value, 0.0);
|
|
}
|