Add test for hddsource, move sources to own library

2025-07-10 04:15:51 +00:00 · 2016-06-07 14:16:48 +03:00
parent 18789f78b3
commit 5a0541d06d
41 changed files with 344 additions and 143 deletions
--- a/sources/extsysmonsources/batterysource.cpp
+++ b/sources/extsysmonsources/batterysource.cpp
@ -0,0 +1,141 @@
+/***************************************************************************
+ *   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 "batterysource.h"
+
+#include <QDir>
+
+#include "awdebug.h"
+
+
+BatterySource::BatterySource(QObject *parent, const QStringList args)
+    : AbstractExtSysMonSource(parent, args)
+{
+    Q_ASSERT(args.count() == 1);
+    qCDebug(LOG_ESS) << __PRETTY_FUNCTION__;
+
+    m_acpiPath = args.at(0);
+    m_sources = getSources();
+}
+
+
+BatterySource::~BatterySource()
+{
+    qCDebug(LOG_ESS) << __PRETTY_FUNCTION__;
+}
+
+
+QVariant BatterySource::data(QString source)
+{
+    qCDebug(LOG_ESS) << "Source" << source;
+
+    if (!m_values.contains(source))
+        run();
+    QVariant value = m_values.take(source);
+    return value;
+}
+
+
+QVariantMap BatterySource::initialData(QString source) const
+{
+    qCDebug(LOG_ESS) << "Source" << source;
+
+    QVariantMap data;
+    if (source == QString("battery/ac")) {
+        data[QString("min")] = false;
+        data[QString("max")] = true;
+        data[QString("name")] = QString("Is AC online or not");
+        data[QString("type")] = QString("bool");
+        data[QString("units")] = QString("");
+    } else if (source == QString("battery/bat")) {
+        data[QString("min")] = 0;
+        data[QString("max")] = 100;
+        data[QString("name")] = QString("Average battery usage");
+        data[QString("type")] = QString("integer");
+        data[QString("units")] = QString("%");
+    } else {
+        data[QString("min")] = 0;
+        data[QString("max")] = 100;
+        data[QString("name")] = QString("Battery %1 usage").arg(index(source));
+        data[QString("type")] = QString("integer");
+        data[QString("units")] = QString("%");
+    }
+
+    return data;
+}
+
+
+void BatterySource::run()
+{
+    // adaptor
+    QFile acFile(QString("%1/AC/online").arg(m_acpiPath));
+    if (acFile.open(QIODevice::ReadOnly))
+        m_values[QString("battery/ac")]
+            = (QString(acFile.readLine()).trimmed().toInt() == 1);
+    acFile.close();
+
+    // batteries
+    float currentLevel = 0.0;
+    float fullLevel = 0.0;
+    for (int i = 0; i < m_batteriesCount; i++) {
+        QFile currentLevelFile(
+            QString("%1/BAT%2/energy_now").arg(m_acpiPath).arg(i));
+        QFile fullLevelFile(
+            QString("%1/BAT%2/energy_full").arg(m_acpiPath).arg(i));
+        if ((currentLevelFile.open(QIODevice::ReadOnly))
+            && (fullLevelFile.open(QIODevice::ReadOnly))) {
+            float batCurrent
+                = QString(currentLevelFile.readLine()).trimmed().toFloat();
+            float batFull
+                = QString(fullLevelFile.readLine()).trimmed().toFloat();
+            m_values[QString("battery/bat%1").arg(i)]
+                = static_cast<int>(100 * batCurrent / batFull);
+            currentLevel += batCurrent;
+            fullLevel += batFull;
+        }
+        currentLevelFile.close();
+        fullLevelFile.close();
+    }
+    m_values[QString("battery/bat")]
+        = static_cast<int>(100 * currentLevel / fullLevel);
+}
+
+
+QStringList BatterySource::sources() const
+{
+    return m_sources;
+}
+
+
+QStringList BatterySource::getSources()
+{
+    QStringList sources;
+    sources.append(QString("battery/ac"));
+    sources.append(QString("battery/bat"));
+    m_batteriesCount
+        = QDir(m_acpiPath)
+              .entryList(QStringList() << QString("BAT*"),
+                         QDir::Dirs | QDir::NoDotAndDotDot, QDir::Name)
+              .count();
+    qCInfo(LOG_ESS) << "Init batteries count as" << m_batteriesCount;
+    for (int i = 0; i < m_batteriesCount; i++)
+        sources.append(QString("battery/bat%1").arg(i));
+
+    qCInfo(LOG_ESS) << "Sources list" << sources;
+    return sources;
+}