Add documentation, cleanup

app/src/fan/PwmControl.cxx

@@ -55,6 +55,12 @@ void PwmControl::enableManualControl() {
 
 void PwmControl::reset() {
   ofstream ostrm(mEnablePath, ios::trunc);
+
   ostrm << mInitialEnable;
   ostrm.close();
+
+  ostrm.open(mModePath, ios::trunc);
+  ostrm << mInitialMode;
+
+  ostrm.close();
 }

app/src/main.cxx

@@ -1,23 +1,24 @@
 #include <exception>
+#include <filesystem>
 #include <iostream>
 #include <memory>
+#include <regex>
 #include <vector>
 
 #include <sensors/sensors.h>
 
-#include <fan/Pwm.h>
-#include <sensor/HwmonTemperatureSensor.h>
-#include <sensor/NvidiaTemperatureSensor.h>
+#include <fan/PwmControl.h>
+#include <sensor/HwmonSensor.h>
+#include <sensor/NvidiaSensor.h>
 
 #define CONFIG_FILE "/etc/conf.d/sensors"
+#define HWMON_BASE_PATH "/sys/class/hwmon"
 
-std::vector<std::shared_ptr<TemperatureSensor>> sensors() {
-  std::vector<std::shared_ptr<TemperatureSensor>> sensors;
+namespace fs = std::filesystem;
 
-  auto config = fopen(CONFIG_FILE, "r");
-  if (sensors_init(config) != 0) {
-    throw std::runtime_error("Config file doesn't exist");
-  }
+template <sensors_subfeature_type T>
+std::vector<std::shared_ptr<Sensor>> sensors() {
+  std::vector<std::shared_ptr<Sensor>> sensors;
 
   int c = 0;
   for (const sensors_chip_name *chipName;
@@ -26,32 +27,64 @@ std::vector<std::shared_ptr<TemperatureSensor>> sensors() {
     int d = 0;
     for (const sensors_feature *feature;
          (feature = sensors_get_features(chipName, &d)) != NULL;) {
-      auto tempFeature = sensors_get_subfeature(chipName, feature,
-                                                SENSORS_SUBFEATURE_TEMP_INPUT);
-      if (tempFeature)
-        sensors.push_back(std::make_shared<HwmonTemperatureSensor>(
-            chipName, feature, tempFeature));
+      auto subFeature = sensors_get_subfeature(chipName, feature, T);
+      if (subFeature) {
+        sensors.push_back(
+            std::make_shared<HwmonSensor>(chipName, feature, subFeature));
+      }
     }
   }
 
-  sensors.push_back(std::make_shared<NvidiaTemperatureSensor>());
-
   return sensors;
 }
 
+std::vector<std::shared_ptr<PwmControl>> pwmControls() {
+  std::vector<std::shared_ptr<PwmControl>> controls;
+
+  const std::regex re_ctrl_enable("pwm[0-9]_enable");
+  const std::regex re_ctrl_mode("pwm[0-9]_mode");
+  const std::regex re_ctrl("pwm[0-9]");
+
+  if (!fs::exists(HWMON_BASE_PATH)) {
+    std::cout << HWMON_BASE_PATH << " doesn't exist" << std::endl;
+  } else {
+    for (const fs::directory_entry &hwmon_device :
+         fs::directory_iterator{HWMON_BASE_PATH}) {
+
+      for (const fs::directory_entry &control :
+           fs::directory_iterator{hwmon_device}) {
+        auto filename = control.path().filename().string();
+
+        if (regex_match(filename, re_ctrl)) {
+          auto controlPath = control.path().string();
+
+          controls.push_back(std::make_shared<PwmControl>(controlPath));
+        }
+      }
+    }
+  }
+
+  return controls;
+}
+
 int main() {
-  class PWM pwm;
-  std::cout << '\n';
-  pwm.dumpValues();
+  auto config = fopen(CONFIG_FILE, "r");
+  if (sensors_init(config) != 0) {
+    throw std::runtime_error("Config file doesn't exist");
+  }
 
-  auto controls = pwm.getControls();
-  std::cout << pwm.readValue(controls[0], PWM_CONTROL_PROPERTY::ENABLE)
-            << std::endl;
-
-  auto tempSensors = sensors();
+  auto tempSensors = sensors<SENSORS_SUBFEATURE_TEMP_INPUT>();
+  tempSensors.push_back(std::make_shared<NvidiaSensor>());
 
   for (auto s : tempSensors) {
-    std::cout << s->name() << ": " << s->temperature() << std::endl;
+    std::cout << s->name() << ": " << s->value() << std::endl;
+  }
+
+  std::cout << "pwm" << std::endl;
+
+  auto pwmSensors = sensors<SENSORS_SUBFEATURE_FAN_INPUT>();
+  for (auto s : pwmSensors) {
+    std::cout << s->name() << ": " << s->value() << std::endl;
   }
 
   return 0;

app/src/sensor/HwmonSensor.cxx

@@ -0,0 +1,19 @@
+#include <sensors/sensors.h>
+
+#include <sensor/HwmonSensor.h>
+
+using namespace std;
+
+HwmonSensor::HwmonSensor(const sensors_chip_name *chipName,
+                         const sensors_feature *feature,
+                         const sensors_subfeature *subfeature)
+    : mChipName(chipName), mFeature(feature), mSubFeature(subfeature) {}
+
+int HwmonSensor::value() {
+  double *value;
+  sensors_get_value(mChipName, mSubFeature->number, value);
+
+  return static_cast<int>(*value);
+}
+
+string HwmonSensor::name() { return sensors_get_label(mChipName, mFeature); }

app/src/sensor/HwmonTemperatureSensor.cxx

@@ -1,21 +0,0 @@
-#include <sensors/sensors.h>
-
-#include <sensor/HwmonTemperatureSensor.h>
-
-using namespace std;
-
-HwmonTemperatureSensor::HwmonTemperatureSensor(
-    const sensors_chip_name *chipName, const sensors_feature *feature,
-    const sensors_subfeature *subfeature)
-    : mChipName(chipName), mFeature(feature), mSubFeature(subfeature) {}
-
-int HwmonTemperatureSensor::temperature() {
-  double *value;
-  sensors_get_value(mChipName, mSubFeature->number, value);
-
-  return static_cast<int>(*value);
-}
-
-string HwmonTemperatureSensor::name() {
-  return sensors_get_label(mChipName, mFeature);
-}

app/src/sensor/NvidiaSensor.cxx

@@ -0,0 +1,21 @@
+#include <include/nvml.h>
+
+#include <sensor/NvidiaSensor.h>
+
+using namespace std;
+
+NvidiaSensor::NvidiaSensor() { nvmlInit_v2(); }
+NvidiaSensor::~NvidiaSensor() { nvmlShutdown(); }
+
+int NvidiaSensor::value() {
+  nvmlDevice_t device;
+
+  nvmlDeviceGetHandleByIndex_v2(0, &device);
+
+  unsigned int temp;
+  nvmlDeviceGetTemperature(device, NVML_TEMPERATURE_GPU, &temp);
+
+  return static_cast<int>(temp);
+}
+
+string NvidiaSensor::name() { return "GPU"; }

app/src/sensor/NvidiaTemperatureSensor.cxx

@@ -1,21 +0,0 @@
-#include <include/nvml.h>
-
-#include <sensor/NvidiaTemperatureSensor.h>
-
-using namespace std;
-
-NvidiaTemperatureSensor::NvidiaTemperatureSensor() { nvmlInit_v2(); }
-NvidiaTemperatureSensor::~NvidiaTemperatureSensor() { nvmlShutdown(); }
-
-int NvidiaTemperatureSensor::temperature() {
-  nvmlDevice_t device;
-
-  nvmlDeviceGetHandleByIndex_v2(0, &device);
-
-  unsigned int temp;
-  nvmlDeviceGetTemperature(device, NVML_TEMPERATURE_GPU, &temp);
-
-  return static_cast<int>(temp);
-}
-
-string NvidiaTemperatureSensor::name() { return "GPU"; }