2 * Copyright (C) 2010-2014 jeanfi@gmail.com
4 * This program is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU General Public License as
6 * published by the Free Software Foundation; either version 2 of the
7 * License, or (at your option) any later version.
9 * This program is distributed in the hope that it will be useful, but
10 * WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
21 #define _(str) gettext(str)
29 #include <NVCtrl/NVCtrl.h>
30 #include <NVCtrl/NVCtrlLib.h>
36 static char *get_product_name(int id)
41 res = XNVCTRLQueryTargetStringAttribute(display,
42 NV_CTRL_TARGET_TYPE_GPU,
45 NV_CTRL_STRING_PRODUCT_NAME,
48 if (strcmp(name, "Unknown")) {
51 log_err(_("Unknown NVIDIA product name for GPU %d"),
56 log_err(_("Failed to retrieve NVIDIA product name for GPU %d"),
60 return strdup("NVIDIA");
63 static double nv_get_temp(int id)
68 res = XNVCTRLQueryTargetAttribute(display,
69 NV_CTRL_TARGET_TYPE_GPU,
72 NV_CTRL_GPU_CORE_TEMPERATURE,
78 return UNKNOWN_DBL_VALUE;
81 static double get_ambient_temp(int id)
86 res = XNVCTRLQueryTargetAttribute(display,
87 NV_CTRL_TARGET_TYPE_GPU,
90 NV_CTRL_AMBIENT_TEMPERATURE,
96 return UNKNOWN_DBL_VALUE;
99 static double get_temp(int id, int type)
101 if (type & SENSOR_TYPE_AMBIENT)
102 return get_ambient_temp(id);
104 return nv_get_temp(id);
107 static double get_usage_att(char *atts, char *att)
109 char *c, *key, *strv, *s;
115 v = UNKNOWN_DBL_VALUE;
140 strv = strndup(s, n);
141 if (!strcmp(key, att))
147 if (v != UNKNOWN_DBL_VALUE)
150 while (*c && (*c == ' ' || *c == ','))
157 static double get_usage(int id, int type)
163 if (type & SENSOR_TYPE_GRAPHICS)
165 else if (type & SENSOR_TYPE_VIDEO)
167 else if (type & SENSOR_TYPE_MEMORY)
169 else if (type & SENSOR_TYPE_PCIE)
172 return UNKNOWN_DBL_VALUE;
174 res = XNVCTRLQueryTargetStringAttribute(display,
175 NV_CTRL_TARGET_TYPE_GPU,
178 NV_CTRL_STRING_GPU_UTILIZATION,
182 return UNKNOWN_DBL_VALUE;
184 v = get_usage_att(atts, stype);
191 static void update(struct psensor *sensor)
195 if (sensor->type & SENSOR_TYPE_TEMP) {
196 v = get_temp(sensor->nvidia_id, sensor->type);
197 } else { /* SENSOR_TYPE_USAGE */
198 v = get_usage(sensor->nvidia_id, sensor->type);
201 if (v == UNKNOWN_DBL_VALUE)
202 log_err(_("Failed to retrieve measure of type %x "
203 "for NVIDIA GPU %d"),
206 psensor_set_current_value(sensor, v);
209 static struct psensor *create_temp_sensor(int id, int subtype, int values_len)
216 pname = get_product_name(id);
218 if (subtype & SENSOR_TYPE_AMBIENT)
219 sprintf(name, "%s %d ambient", pname, id);
221 sprintf(name, "%s %d", pname, id);
224 sid = malloc(strlen("NVIDIA") + 1 + strlen(name) + 1);
225 sprintf(sid, "NVIDIA %s", name);
227 t = SENSOR_TYPE_NVCTRL | SENSOR_TYPE_GPU | SENSOR_TYPE_TEMP | subtype;
229 s = psensor_create(sid,
231 strdup(_("NVIDIA GPU")),
240 static struct psensor *create_usage_sensor(int id,
249 if (subtype & SENSOR_TYPE_GRAPHICS)
250 sprintf(name, "GPU%d graphics", id);
251 else if (subtype & SENSOR_TYPE_MEMORY)
252 sprintf(name, "GPU%d memory", id);
253 else if (subtype & SENSOR_TYPE_VIDEO)
254 sprintf(name, "GPU%d video", id);
255 else /* if (subtype & SENSOR_TYPE_PCIE) */
256 sprintf(name, "GPU%d PCIe", id);
259 sid = malloc(strlen("NVIDIA") + 1 + strlen(name) + 1);
260 sprintf(sid, "NVIDIA %s", name);
262 t = SENSOR_TYPE_NVCTRL | SENSOR_TYPE_GPU | SENSOR_TYPE_USAGE | subtype;
264 s = psensor_create(sid,
266 strdup(_("NVIDIA GPU")),
276 Opens connection to X server and returns the number
279 Return 0 if no NVIDIA gpus or cannot get information.
285 display = XOpenDisplay(NULL);
288 log_err(_("Cannot open connection to X11 server."));
292 if (XNVCTRLQueryExtension(display, &evt, &err) &&
293 XNVCTRLQueryTargetCount(display, NV_CTRL_TARGET_TYPE_GPU, &n))
296 log_err(_("Failed to retrieve NVIDIA information."));
301 void nvidia_psensor_list_update(struct psensor **sensors)
303 struct psensor **ss, *s;
309 if (s->type & SENSOR_TYPE_NVCTRL)
316 struct psensor **nvidia_psensor_list_add(struct psensor **sensors,
320 struct psensor **tmp, **ss, *s;
325 for (i = 0; i < n; i++) {
326 s = create_temp_sensor(i, 0, values_len);
327 tmp = psensor_list_add(ss, s);
332 s = create_temp_sensor(i, SENSOR_TYPE_AMBIENT, values_len);
333 tmp = psensor_list_add(ss, s);
338 s = create_usage_sensor(i, SENSOR_TYPE_GRAPHICS, values_len);
339 tmp = psensor_list_add(ss, s);
344 s = create_usage_sensor(i, SENSOR_TYPE_VIDEO, values_len);
345 tmp = psensor_list_add(ss, s);
350 s = create_usage_sensor(i, SENSOR_TYPE_MEMORY, values_len);
351 tmp = psensor_list_add(ss, s);
356 s = create_usage_sensor(i, SENSOR_TYPE_PCIE, values_len);
357 tmp = psensor_list_add(ss, s);
367 void nvidia_cleanup()
370 XCloseDisplay(display);