3fb29b5ecf
However, this drastically increases CPU use of gotop, so leaving this unmerged at the moment.
125 lines
2.6 KiB
Go
125 lines
2.6 KiB
Go
// +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)
|
|
}
|
|
}
|
|
}
|