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
|
#include <string.h>
#include <stdlib.h>
#include <stdio.h>
#include "../slstatus.h"
#include "../util.h"
// Function to run a command and capture its output
char* runCommand(const char* command) {
char buffer[128];
char* result = NULL;
FILE* pipe = popen(command, "r");
if (!pipe) {
perror("popen");
exit(1);
}
while (fgets(buffer, sizeof(buffer), pipe) != NULL) {
if (result == NULL) {
result = strdup(buffer);
} else {
char* old_result = result;
result = (char*)malloc(strlen(old_result) + strlen(buffer) + 1);
strcpy(result, old_result);
strcat(result, buffer);
free(old_result);
}
}
pclose(pipe);
return result;
}
// Funtion to gather used VRAM information (MiB)
float GetUsedVRAM() {
char* command_output;
float vram_used = -1.0;
// Check if an NVIDIA card is present
command_output = runCommand("nvidia-smi --query-gpu=memory.used --format=csv,noheader,nounits");
if (command_output != NULL) {
sscanf(command_output, "%f", &vram_used);
free(command_output);
} else {
// If no NVIDIA card is found, check for an AMD Radeon card
// todo intel: "intel-gpu-top -b -s 1 -o - | grep 'used:'"
command_output = runCommand("cat /sys/class/drm/card0/device/mem_info_vram_used");
if (command_output != NULL) {
sscanf(command_output, "%f", &vram_used);
free(command_output);
}
}
return vram_used;
}
// Funtion to gather total VRAM information (MiB)
float GetTotalVRAM() {
char* command_output;
float vram_total = -1.0;
// Check if an NVIDIA card is present
command_output = runCommand("nvidia-smi --query-gpu=memory.total --format=csv,noheader,nounits");
if (command_output != NULL) {
sscanf(command_output, "%f", &vram_total);
free(command_output);
} else {
// If no NVIDIA card is found, check for an AMD Radeon card
// todo intel: "intel-gpu-top -b -s 1 -o - | grep 'total:'"
command_output = runCommand("cat /sys/class/drm/card0/device/mem_info_vram_total");
if (command_output != NULL) {
sscanf(command_output, "%f", &vram_total);
free(command_output);
}
}
return vram_total;
}
const char *
vram_perc(const char *unused) {
float vram_used = GetUsedVRAM();
float vram_total = GetTotalVRAM();
if (vram_used >= 0.0 && vram_total >= 0.0) {
int vram_percentage = (int) vram_used / vram_total * 100;
return bprintf("%d", vram_percentage);
} else {
return NULL;
}
}
|
