Add handling for missing nvml lib

PKGBUILD

@@ -1,5 +1,5 @@
 pkgname=fantasize
-pkgver=0.1.9
+pkgver=0.1.7
 pkgrel=1
 pkgdesc='C++ fan control for Linux'
 url='https://github.com/Tabascl/fantasize.git'

app/.clang-format

@@ -0,0 +1 @@
+BasedOnStyle: Google

app/CMakeLists.txt

@@ -1,25 +1,38 @@
 cmake_minimum_required(VERSION 3.0)
 
-project(fantasize VERSION 0.1.8)
+project(fantasize VERSION 0.1.7)
 
 set(CMAKE_EXPORT_COMPILE_COMMANDS ON)
 
-find_package(nlohmann_json 3.11.2 REQUIRED)
-find_package(Boost 1.74 COMPONENTS program_options log log_setup date_time REQUIRED)
+find_package(nlohmann_json 3.7 REQUIRED)
+find_package(Boost 1.80 COMPONENTS program_options log log_setup date_time REQUIRED)
+
+find_library(NVML_LIB nvidia-ml)
+
+set(ADDITIONAL_LIBS)
+
+if (NVML_LIB)
+    set(HAVE_NVML)
+
+    add_library(nvidia-sensor-lib STATIC
+      src/sensor/NvidiaSensor.cxx
+      src/sensor/GPUSensorsFacade.cxx)
+
+    target_link_libraries(nvidia-sensor-lib PUBLIC NVML_LIB)
+
+    list(APPEND ADDITIONAL_LIBS nvidia-sensor-lib)
+endif()
 
 add_executable(${PROJECT_NAME}
   src/main.cxx
   src/sensor/LMSensorsFacade.cxx
-  src/sensor/GPUSensorsFacade.cxx
   src/sensor/Sensor.cxx
-  src/sensor/NvidiaSensor.cxx
   src/sensor/LMSensor.cxx
   src/pwm/PWMControl.cxx
   src/pwm/PWMControlFacade.cxx
   src/fan/HwmonFan.cxx
   src/fan/FanCurve.cxx
   src/fan/FanLabeler.cxx
-  src/fan/Aggregators.cxx
   src/FanGenerator.cxx
   src/Serializer.cxx
   src/sensor/SensorManager.cxx
@@ -31,7 +44,8 @@ add_executable(${PROJECT_NAME}
 set_property(TARGET ${PROJECT_NAME} PROPERTY CXX_STANDARD 20)
 
 target_include_directories(${PROJECT_NAME} PUBLIC include /opt/cuda)
-target_link_libraries(${PROJECT_NAME} PUBLIC sensors nvidia-ml nlohmann_json::nlohmann_json tbb ${Boost_LIBRARIES})
+
+target_link_libraries(${PROJECT_NAME} PUBLIC sensors nlohmann_json::nlohmann_json tbb ${Boost_LIBRARIES} ${ADDITIONAL_LIBS})
 
 install(TARGETS ${PROJECT_NAME} DESTINATION usr/local/bin)
 install(FILES ${CMAKE_CURRENT_SOURCE_DIR}/unit/fantasize.service DESTINATION usr/lib/systemd/system)

app/include/Serializer.h

@@ -7,7 +7,6 @@
 #include <nlohmann/json.hpp>
 
 #include <Settings.h>
-#include <fan/Aggregators.h>
 #include <fan/Fan.h>
 #include <fan/FanCurve.h>
 #include <sensor/Sensor.h>
@@ -17,22 +16,20 @@
 
 using json = nlohmann::json;
 
-class Serializer
-{
+class Serializer {
 public:
   Serializer();
   void SerializeFans(std::vector<std::shared_ptr<Fan>> fans);
-  std::vector<std::shared_ptr<Fan>> DeserializeFans(
-    std::vector<std::shared_ptr<Sensor>> availableSensors);
-  std::vector<std::shared_ptr<FanCurve>> DeserializeFanCurves(
-    std::vector<std::shared_ptr<Sensor>> availableSensors,
-    std::vector<std::shared_ptr<Fan>> availableFans);
+  std::vector<std::shared_ptr<Fan>>
+  DeserializeFans(std::vector<std::shared_ptr<Sensor>> availableSensors);
+  std::vector<std::shared_ptr<FanCurve>>
+  DeserializeFanCurves(std::vector<std::shared_ptr<Sensor>> availableSensors,
+                       std::vector<std::shared_ptr<Fan>> availableFans);
   std::shared_ptr<Settings> DeserializeSettings();
 
 private:
   void WriteJson(json o);
   json ReadJson();
-  std::unique_ptr<Aggregator> aggregatorFromString(std::string str) const;
 };
 
 #endif // SERIALIZER_H_

app/include/fan/Aggregators.h

@@ -1,27 +0,0 @@
-#ifndef AGGREGATORS_H_
-#define AGGREGATORS_H_
-
-#include <sensor/Sensor.h>
-
-class Aggregator
-{
-public:
-  virtual int aggregate(std::vector<std::shared_ptr<Sensor>> sensors) const = 0;
-  virtual const std::string toString() const = 0;
-};
-
-class AverageAggregator : public Aggregator
-{
-public:
-  int aggregate(std::vector<std::shared_ptr<Sensor>> sensors) const override;
-  const std::string toString() const override;
-};
-
-class MaxAggregator : public Aggregator
-{
-public:
-  int aggregate(std::vector<std::shared_ptr<Sensor>> sensors) const override;
-  const std::string toString() const override;
-};
-
-#endif // AGGERGATORS_H_

app/include/fan/FanCurve.h

@@ -4,23 +4,19 @@
 #include <memory>
 #include <vector>
 
-#include <fan/Aggregators.h>
 #include <fan/HwmonFan.h>
 #include <sensor/Sensor.h>
 
-struct FanStep
-{
+struct FanStep {
   int Temp;
   int Percent;
 };
 
-class FanCurve
-{
+class FanCurve {
 public:
   FanCurve(std::vector<FanStep> steps,
            std::vector<std::shared_ptr<Sensor>> sensors,
-           std::vector<std::shared_ptr<Fan>> fans,
-           std::unique_ptr<Aggregator> aggregator);
+           std::vector<std::shared_ptr<Fan>> fans);
 
   void DoFanControl();
 
@@ -31,7 +27,6 @@ private:
   std::vector<FanStep> mSteps;
   std::vector<std::shared_ptr<Sensor>> mTempSensors;
   std::vector<std::shared_ptr<Fan>> mFans;
-  std::unique_ptr<Aggregator> mAggregator;
 };
 
 #endif // FANCURVE_H_

app/include/fan/HwmonFan.h

@@ -4,6 +4,8 @@
 #include <chrono>
 #include <memory>
 
+#include <boost/json/object.hpp>
+
 #include <fan/Fan.h>
 #include <pwm/PWMControl.h>
 #include <sensor/Sensor.h>

app/include/pwm/PWMControl.h

@@ -1,6 +1,7 @@
 #ifndef PWMCONTROL_H_
 #define PWMCONTROL_H_
 
+#include <boost/json/object.hpp>
 #include <string>
 
 #include <Printable.h>

app/src/Serializer.cxx

@@ -1,3 +1,4 @@
+#include "Settings.h"
 #include <filesystem>
 #include <fstream>
 #include <iostream>
@@ -8,21 +9,17 @@
 #include <fan/HwmonFan.h>
 #include <pwm/PWMControl.h>
 #include <sensor/LMSensor.h>
-#include <string>
 
 using namespace std;
 namespace fs = filesystem;
 
-Serializer::Serializer()
-{
+Serializer::Serializer() {
   if (!fs::exists(SERIALIZATION_DIR)) {
     fs::create_directory(SERIALIZATION_DIR);
   }
 }
 
-void
-Serializer::SerializeFans(vector<shared_ptr<Fan>> fans)
-{
+void Serializer::SerializeFans(vector<shared_ptr<Fan>> fans) {
   json obj;
 
   for (auto f : fans) {
@@ -33,8 +30,7 @@ Serializer::SerializeFans(vector<shared_ptr<Fan>> fans)
 }
 
 vector<shared_ptr<Fan>>
-Serializer::DeserializeFans(vector<shared_ptr<Sensor>> availableSensors)
-{
+Serializer::DeserializeFans(vector<shared_ptr<Sensor>> availableSensors) {
   vector<shared_ptr<Fan>> fans;
 
   // Create a for the sensors first, then searching becomes cheaper
@@ -45,7 +41,7 @@ Serializer::DeserializeFans(vector<shared_ptr<Sensor>> availableSensors)
 
   auto data = ReadJson();
   try {
-    for (auto& el : data["fans"].items()) {
+    for (auto &el : data["fans"].items()) {
       auto pwmControl = make_shared<PWMControl>(el.value()["PWMControl"]);
       auto rpmSensor = sensorMap[el.value()["LMSensor"]];
 
@@ -60,22 +56,20 @@ Serializer::DeserializeFans(vector<shared_ptr<Sensor>> availableSensors)
 
       fans.push_back(fan);
     }
-  } catch (const std::exception& e) {
+  } catch (const std::exception &e) {
     std::cout << "Deserialization error! Message: " << e.what() << std::endl;
   }
   return fans;
 }
 
-void
-Serializer::WriteJson(json o)
-{
+void Serializer::WriteJson(json o) {
   json obj;
 
   if (fs::exists(fs::path(SERIALIZATION_DIR) / FANS_JSON_FILENAME)) {
     obj = ReadJson();
   }
 
-  for (auto& [key, value] : o.items()) {
+  for (auto &[key, value] : o.items()) {
     obj[key] = value;
   }
 
@@ -83,18 +77,14 @@ Serializer::WriteJson(json o)
   ostrm << obj.dump(2) << "\n";
 }
 
-json
-Serializer::ReadJson()
-{
+json Serializer::ReadJson() {
   ifstream istrm(fs::path(SERIALIZATION_DIR) / FANS_JSON_FILENAME);
   return json::parse(istrm);
 }
 
-vector<shared_ptr<FanCurve>>
-Serializer::DeserializeFanCurves(
-  std::vector<std::shared_ptr<Sensor>> availableSensors,
-  std::vector<std::shared_ptr<Fan>> availableFans)
-{
+vector<shared_ptr<FanCurve>> Serializer::DeserializeFanCurves(
+    std::vector<std::shared_ptr<Sensor>> availableSensors,
+    std::vector<std::shared_ptr<Fan>> availableFans) {
   auto data = ReadJson();
 
   map<string, shared_ptr<Sensor>> sensorMap;
@@ -109,47 +99,32 @@ Serializer::DeserializeFanCurves(
 
   vector<shared_ptr<FanCurve>> curves;
 
-  for (auto& el : data["fancurves"].items()) {
+  for (auto &el : data["fancurves"].items()) {
     vector<FanStep> steps;
     vector<shared_ptr<Sensor>> sensors;
     vector<shared_ptr<Fan>> fans;
 
-    for (auto& step : el.value()["FanSteps"].items()) {
-      steps.push_back(FanStep{ step.value()[0], step.value()[1] });
+    for (auto &step : el.value()["FanSteps"].items()) {
+      steps.push_back(FanStep{step.value()[0], step.value()[1]});
     }
 
-    for (auto& sensor : el.value()["Sensors"].items()) {
+    for (auto &sensor : el.value()["Sensors"].items()) {
       if (sensorMap.contains(sensor.value()))
         sensors.push_back(sensorMap[sensor.value()]);
     }
 
-    for (auto& fan : el.value()["Fans"].items()) {
+    for (auto &fan : el.value()["Fans"].items()) {
       if (fanMap.contains(fan.value()))
         fans.push_back(fanMap[fan.value()]);
     }
 
-    std::unique_ptr<Aggregator> aggregator =
-      aggregatorFromString(el.value()["AggregateFunction"]);
-
-    curves.push_back(
-      make_shared<FanCurve>(steps, sensors, fans, std::move(aggregator)));
+    curves.push_back(make_shared<FanCurve>(steps, sensors, fans));
   }
 
   return curves;
 }
 
-std::unique_ptr<Aggregator>
-Serializer::aggregatorFromString(std::string str) const
-{
-  if (str == "max")
-    return std::make_unique<MaxAggregator>();
-  else
-    return std::make_unique<AverageAggregator>();
-}
-
-shared_ptr<Settings>
-Serializer::DeserializeSettings()
-{
+shared_ptr<Settings> Serializer::DeserializeSettings() {
   int frequency = FREQUENCY_DEFAULT;
 
   auto data = ReadJson();

app/src/fan/Aggregators.cxx

@@ -1,39 +0,0 @@
-#include <climits>
-#include <fan/Aggregators.h>
-
-int
-AverageAggregator::aggregate(std::vector<std::shared_ptr<Sensor>> sensors) const
-{
-  int sum = 0;
-
-  for (auto sensor : sensors) {
-    sum += sensor->value();
-  }
-
-  return sum / sensors.size();
-}
-
-const std::string
-AverageAggregator::toString() const
-{
-  return "Average";
-}
-
-int
-MaxAggregator::aggregate(std::vector<std::shared_ptr<Sensor>> sensors) const
-{
-  int max = INT_MIN;
-
-  for (auto sensor : sensors) {
-    if (sensor->value() > max)
-      max = sensor->value();
-  }
-
-  return max;
-}
-
-const std::string
-MaxAggregator::toString() const
-{
-  return "Max";
-}

app/src/fan/FanCurve.cxx

@@ -1,28 +1,19 @@
-#include <boost/log/attributes/named_scope.hpp>
-#include <iostream>
-
-#include <boost/log/trivial.hpp>
-
 #include <fan/FanCurve.h>
-#include <memory>
+
+#include <boost/log/attributes/named_scope.hpp>
+#include <boost/log/trivial.hpp>
+#include <sstream>
 
 using namespace std;
 
 FanCurve::FanCurve(std::vector<FanStep> steps,
                    std::vector<std::shared_ptr<Sensor>> sensors,
-                   std::vector<std::shared_ptr<Fan>> fans,
-                   std::unique_ptr<Aggregator> aggregator)
-  : mSteps(steps)
-  , mTempSensors(sensors)
-  , mFans(fans)
-  , mAggregator(std::move(aggregator))
-{
+                   std::vector<std::shared_ptr<Fan>> fans)
+    : mSteps(steps), mTempSensors(sensors), mFans(fans) {
   PrintInfo();
 }
 
-void
-FanCurve::DoFanControl()
-{
+void FanCurve::DoFanControl() {
   BOOST_LOG_FUNCTION();
 
   int temp = AggregateTemperature();
@@ -59,15 +50,17 @@ FanCurve::DoFanControl()
   }
 }
 
-int
-FanCurve::AggregateTemperature()
-{
-  return mAggregator->aggregate(mTempSensors);
+// Dummy Implementation using AVG
+int FanCurve::AggregateTemperature() {
+  int sum = 0;
+  for (auto s : mTempSensors) {
+    sum += s->value();
+  }
+
+  return sum / mTempSensors.size();
 }
 
-void
-FanCurve::PrintInfo()
-{
+void FanCurve::PrintInfo() {
   BOOST_LOG_FUNCTION()
 
   BOOST_LOG_TRIVIAL(info) << "### Fan curve:";
@@ -96,10 +89,4 @@ FanCurve::PrintInfo()
   }
 
   BOOST_LOG_TRIVIAL(info) << sStream.str();
-
-  sStream.str(string());
-
-  sStream << "Aggregate function: " << mAggregator->toString();
-
-  BOOST_LOG_TRIVIAL(info) << sStream.str();
 }

app/src/main.cxx

@@ -20,7 +20,7 @@
 
 #include <App.h>
 
-#define PROJECT_VERSION "v0.1.9"
+#define PROJECT_VERSION "v0.1.6"
 
 namespace po = boost::program_options;
 namespace logging = boost::log;
@@ -29,32 +29,26 @@ App app;
 
 static int doInitialSetup = 0;
 
-void
-signal_handler(int s)
-{
-  app.Shutdown();
-}
+void signal_handler(int s) { app.Shutdown(); }
 
-void
-InitLogging(bool verbose)
-{
+void InitLogging(bool verbose) {
   logging::add_console_log(
-    std::clog,
-    logging::keywords::format =
-      (logging::expressions::stream
-       << "["
-       << logging::expressions::format_date_time<boost::posix_time::ptime>(
-            "TimeStamp", "%Y-%m-%d %H:%M:%S")
-       << "]["
-       << logging::expressions::format_named_scope(
-            "Scope", logging::keywords::format = "%c")
-       << "]"
-       << "[" << logging::trivial::severity << "] "
-       << logging::expressions::smessage));
+      std::clog,
+      logging::keywords::format =
+          (logging::expressions::stream
+           << "["
+           << logging::expressions::format_date_time<boost::posix_time::ptime>(
+                  "TimeStamp", "%Y-%m-%d %H:%M:%S")
+           << "]["
+           << logging::expressions::format_named_scope(
+                  "Scope", logging::keywords::format = "%c")
+           << "]"
+           << "[" << logging::trivial::severity << "] "
+           << logging::expressions::smessage));
 
   logging::add_common_attributes();
   logging::core::get()->add_global_attribute(
-    "Scope", logging::attributes::named_scope());
+      "Scope", logging::attributes::named_scope());
 
   BOOST_LOG_FUNCTION();
 
@@ -66,9 +60,7 @@ InitLogging(bool verbose)
   }
 }
 
-int
-main(int argc, char** argv)
-{
+int main(int argc, char **argv) {
   BOOST_LOG_FUNCTION()
   BOOST_LOG_TRIVIAL(info) << "Version: " << PROJECT_VERSION;
 
@@ -77,8 +69,8 @@ main(int argc, char** argv)
 
   po::options_description desc("Allowed options");
   desc.add_options()("help,h", "produce help message")(
-    "setup,s", po::bool_switch(), "run initial setup")(
-    "verbose,v", po::bool_switch(), "print debug info");
+      "setup,s", po::bool_switch(),
+      "run initial setup")("verbose,v", po::bool_switch(), "print debug info");
 
   po::variables_map vm;
   po::store(po::parse_command_line(argc, argv, desc), vm);
@@ -98,7 +90,7 @@ main(int argc, char** argv)
       app.Init();
       app.NormalOperation();
     }
-  } catch (const std::exception& e) {
+  } catch (const std::exception &e) {
     std::cout << "An exception was caught: " << e.what() << std::endl;
   }
 

app/src/pwm/PWMControl.cxx

@@ -15,9 +15,7 @@
 using namespace std;
 namespace fs = filesystem;
 
-PWMControl::PWMControl(string controlPath)
-  : mControlPath(controlPath)
-{
+PWMControl::PWMControl(string controlPath) : mControlPath(controlPath) {
   fs::path pathEnable(mControlPath + PWM_POSTFIX_ENABLE);
   fs::path pathMode(mControlPath + PWM_POSTFIX_MODE);
 
@@ -37,24 +35,21 @@ PWMControl::PWMControl(string controlPath)
   istrm.close();
 }
 
-PWMControl::~PWMControl()
-{
+PWMControl::~PWMControl() {
   BOOST_LOG_FUNCTION();
 
   BOOST_LOG_TRIVIAL(trace) << "Cleanup";
   Reset();
 }
 
-void
-PWMControl::Power(int percent)
-{
+void PWMControl::Power(int percent) {
   BOOST_LOG_FUNCTION();
 
   int pwmValue = (PWM_MAX_VALUE * percent) / 100;
 
   if (percent != mCurrentValue) {
-    BOOST_LOG_TRIVIAL(trace) << "Updating control value of " << toString()
-                             << " to " << percent << "% (" << pwmValue << ")";
+    BOOST_LOG_TRIVIAL(trace)
+        << "Updating control value to " << percent << "% (" << pwmValue << ")";
     ofstream ostrm(mControlPath, ios::trunc);
     ostrm << pwmValue;
     ostrm.close();
@@ -63,9 +58,7 @@ PWMControl::Power(int percent)
   }
 }
 
-int
-PWMControl::Power()
-{
+int PWMControl::Power() {
   int value;
   ifstream istrm;
 
@@ -75,17 +68,13 @@ PWMControl::Power()
   return (value * 100) / PWM_MAX_VALUE;
 }
 
-void
-PWMControl::EnableManualControl()
-{
+void PWMControl::EnableManualControl() {
   ofstream ostrm(mEnablePath, ios::trunc);
   ostrm << static_cast<int>(PWM_ENABLE::MANUAL_CONTROL);
   ostrm.close();
 }
 
-void
-PWMControl::Reset()
-{
+void PWMControl::Reset() {
   ofstream ostrm(mEnablePath, ios::trunc);
 
   ostrm << mInitialEnable;
@@ -97,15 +86,11 @@ PWMControl::Reset()
   ostrm.close();
 }
 
-const string
-PWMControl::toString() const
-{
+const string PWMControl::toString() const {
   return fs::path(mControlPath).filename();
 }
 
-json
-PWMControl::toJson() const
-{
-  json obj = { "PWMControl", mControlPath };
+json PWMControl::toJson() const {
+  json obj = {"PWMControl", mControlPath};
   return obj;
 }

app/src/sensor/NvidiaSensor.cxx

@@ -1,7 +1,10 @@
-#include <include/nvml.h>
+#ifdef HAVE_NVML
 
+#include <include/nvml.h>
 #include <sensor/NvidiaSensor.h>
 
+#include <boost/json/object.hpp>
+
 using namespace std;
 
 NvidiaSensor::NvidiaSensor() { nvmlInit_v2(); }
@@ -24,3 +27,5 @@ json NvidiaSensor::toJson() const {
   json obj = {"NvidiaSensor", toString()};
   return obj;
 }
+
+#endif

app/src/sensor/SensorManager.cxx

@@ -3,20 +3,26 @@
 using namespace std;
 
 SensorManager::SensorManager()
-    : mLMSensorsFacade(make_unique<LMSensorsFacade>()),
-      mGPUSensorsFacade(make_unique<GPUSensorsFacade>()) {}
-
-vector<shared_ptr<Sensor>> SensorManager::TemperatureSensors() {
-  vector<shared_ptr<Sensor>> tempSensors;
-
-  tempSensors = mLMSensorsFacade->TemperatureSensors();
-
-  auto gpuSensors = mGPUSensorsFacade->TemperatureSensors();
-  tempSensors.insert(tempSensors.end(), gpuSensors.begin(), gpuSensors.end());
-
-  return tempSensors;
+    : mLMSensorsFacade(make_unique<LMSensorsFacade>())
+    , mGPUSensorsFacade(make_unique<GPUSensorsFacade>())
+{
 }
 
-vector<shared_ptr<Sensor>> SensorManager::RPMSensors() {
-  return mLMSensorsFacade->RPMSensors();
+vector<shared_ptr<Sensor>> SensorManager::TemperatureSensors()
+{
+    vector<shared_ptr<Sensor>> tempSensors;
+
+    tempSensors = mLMSensorsFacade->TemperatureSensors();
+
+#ifdef HAVE_NVML
+    auto gpuSensors = mGPUSensorsFacade->TemperatureSensors();
+    tempSensors.insert(tempSensors.end(), gpuSensors.begin(), gpuSensors.end());
+#endif
+
+    return tempSensors;
+}
+
+vector<shared_ptr<Sensor>> SensorManager::RPMSensors()
+{
+    return mLMSensorsFacade->RPMSensors();
 }

app/unit/fantasize.service

@@ -7,4 +7,4 @@ ExecStart=/usr/local/bin/fantasize
 Restart=always
 
 [Install]
-WantedBy=graphical.target
+WantedBy=multi-user.target