1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
// +build freebsd
package widgets
import (
"os/exec"
"strconv"
"strings"
"github.com/rai-project/nvidia-smi"
"github.com/xxxserxxx/gotop/utils"
)
var sensorOIDS = map[string]string{
"dev.cpu.0.temperature": "CPU 0 ",
"hw.acpi.thermal.tz0.temperature": "Thermal zone 0",
}
type sensorMeasurement struct {
name string
temperature float64
}
func removeUnusedChars(s string) string {
s1 := strings.Replace(s, "C", "", 1)
s2 := strings.TrimSuffix(s1, "\n")
return s2
}
func refineOutput(output []byte) (float64, error) {
convertedOutput := utils.ConvertLocalizedString(removeUnusedChars(string(output)))
value, err := strconv.ParseFloat(convertedOutput, 64)
if err != nil {
return 0, err
}
return value, nil
}
func collectSysctlSensors() []sensorMeasurement {
var measurements []sensorMeasurement
for k, v := range sensorOIDS {
output, err := exec.Command("sysctl", "-n", k).Output()
if err != nil {
continue
}
value, err := refineOutput(output)
if err != nil {
continue
}
measurements = append(measurements, sensorMeasurement{v, value})
}
return measurements
}
func collectNvidiaSensors() []sensorMeasurement {
var measurements []sensorMeasurement
info, _ := nvidiasmi.New()
if info.HasGPU() {
for i := range info.GPUS {
gpu := info.GPUS[i]
var s sensorMeasurement
s.name = gpu.ProductName + " " + strconv.Itoa(i)
s.temperature, _ = strconv.ParseFloat(strings.ReplaceAll(gpu.GpuTemp, " C", ""), 10)
measurements = append(measurements, s)
}
}
return measurements
}
func collectAMDGPUSensors() []sensorMeasurement {
var measurments []sensorMeasurement
return measurments
}
func collectGPUSensors() []sensorMeasurement {
var measurements []sensorMeasurement
measurements = append(measurements, collectSysctlSensors()...)
measurements = append(measurements, collectNvidiaSensors()...)
measurements = append(measurements, collectAMDGPUSensors()...)
return measurements
}
func collectSensors() []sensorMeasurement {
var measurements []sensorMeasurement
for k, v := range sensorOIDS {
output, err := exec.Command("sysctl", "-n", k).Output()
if err != nil {
continue
}
value, err := refineOutput(output)
if err != nil {
continue
}
measurements = append(measurements, sensorMeasurement{v, value})
}
measurements = append(measurements, collectGPUSensors()...)
return measurements
}
func (self *TempWidget) update() {
sensors := collectSensors()
for _, sensor := range sensors {
switch self.TempScale {
case Fahrenheit:
self.Data[sensor.name] = utils.CelsiusToFahrenheit(int(sensor.temperature))
case Celsius:
self.Data[sensor.name] = int(sensor.temperature)
}
}
}
|
