Add GPU frequency sensors to sensor classes

Author: mathoudebine

library/sensors/sensors.py

@@ -66,6 +66,11 @@ def fps() -> int:
     def fan_percent() -> float:
         pass
 
+    @staticmethod
+    @abstractmethod
+    def frequency() -> float:
+        pass
+
     @staticmethod
     @abstractmethod
     def is_available() -> bool:

library/sensors/sensors_librehardwaremonitor.py

@@ -330,6 +330,23 @@ def fan_percent(cls) -> float:
         # No Fan Speed sensor for this GPU model
         return math.nan
 
+    @classmethod
+    def frequency(cls) -> float:
+        gpu_to_use = cls.get_gpu_to_use()
+        if gpu_to_use is None:
+            # GPU not supported
+            return math.nan
+
+        try:
+            for sensor in gpu_to_use.Sensors:
+                if sensor.SensorType == Hardware.SensorType.Control:
+                    pass
+        except:
+            pass
+
+        # No Frequency sensor for this GPU model
+        return math.nan
+
     @classmethod
     def is_available(cls) -> bool:
         cls.gpu_name = get_gpu_name()

library/sensors/sensors_python.py

@@ -176,6 +176,16 @@ def fan_percent() -> float:
         else:
             return math.nan
 
+    @staticmethod
+    def frequency() -> float:
+        global DETECTED_GPU
+        if DETECTED_GPU == GpuType.AMD:
+            return GpuAmd.frequency()
+        elif DETECTED_GPU == GpuType.NVIDIA:
+            return GpuNvidia.frequency()
+        else:
+            return math.nan
+
     @staticmethod
     def is_available() -> bool:
         global DETECTED_GPU
@@ -247,6 +257,11 @@ def fan_percent() -> float:
 
         return math.nan
 
+    @staticmethod
+    def frequency() -> float:
+        # Not supported by Python libraries
+        return math.nan
+
     @staticmethod
     def is_available() -> bool:
         try:
@@ -331,6 +346,11 @@ def fan_percent() -> float:
 
         return math.nan
 
+    @staticmethod
+    def frequency() -> float:
+        # Not supported by Python libraries
+        return math.nan
+
     @staticmethod
     def is_available() -> bool:
         try:

library/sensors/sensors_stub_random.py

@@ -60,6 +60,10 @@ def fps() -> int:
     def fan_percent() -> float:
         return random.uniform(0, 100)
 
+    @staticmethod
+    def frequency() -> float:
+        return random.uniform(800, 3400)
+
     @staticmethod
     def is_available() -> bool:
         return True

library/sensors/sensors_stub_static.py

@@ -35,6 +35,7 @@
 GPU_MEM_TOTAL_SIZE_GB = 32
 NETWORK_SPEED_BYTES = 1061000000
 GPU_FPS = 120
+GPU_FREQ_MHZ = 1500.0
 
 
 class Cpu(sensors.Cpu):
@@ -73,6 +74,10 @@ def fps() -> int:
     def fan_percent() -> float:
         return PERCENTAGE_SENSOR_VALUE
 
+    @staticmethod
+    def frequency() -> float:
+        return GPU_FREQ_MHZ
+
     @staticmethod
     def is_available() -> bool:
         return True