/*
- Copyright (C) 2010-2011 wpitchoune@gmail.com
-
- This program is free software; you can redistribute it and/or modify
- it under the terms of the GNU General Public License as published by
- the Free Software Foundation; either version 2 of the License, or
- (at your option) any later version.
-
- This program is distributed in the hope that it will be useful,
- but WITHOUT ANY WARRANTY; without even the implied warranty of
- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- GNU General Public License for more details.
-
- You should have received a copy of the GNU General Public License
- along with this program; if not, write to the Free Software
- Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
- 02110-1301 USA
-*/
-
+ * Copyright (C) 2010-2011 jeanfi@gmail.com
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License as
+ * published by the Free Software Foundation; either version 2 of the
+ * License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
+ * 02110-1301 USA
+ */
#include <locale.h>
#include <libintl.h>
#define _(str) gettext(str)
#include <NVCtrl/NVCtrl.h>
#include <NVCtrl/NVCtrlLib.h>
-
#include "psensor.h"
-Display *nvidia_sensors_dpy;
+Display *display;
-int nvidia_get_sensor_temp(struct psensor *sensor)
+/*
+ Returns the temperature (Celcius) of a NVidia GPU.
+*/
+static int get_temp(struct psensor *sensor)
{
int temp;
Bool res;
- res = XNVCTRLQueryTargetAttribute(nvidia_sensors_dpy,
+ res = XNVCTRLQueryTargetAttribute(display,
NV_CTRL_TARGET_TYPE_GPU,
sensor->nvidia_id,
0,
- NV_CTRL_GPU_CORE_TEMPERATURE, &temp);
+ NV_CTRL_GPU_CORE_TEMPERATURE,
+ &temp);
- if (res == True) {
+ if (res == True)
return temp;
- } else {
- fprintf(stderr,
- _("ERROR: failed to retrieve nvidia temperature\n"));
- return 0;
- }
+
+ fprintf(stderr, _("ERROR: failed to retrieve nvidia temperature\n"));
+ return 0;
}
-struct psensor *nvidia_create_sensor(int id, int values_max_length)
+static struct psensor *create_sensor(int id, int values_len)
{
char name[200];
char *sid;
sprintf(sid, "nvidia %s", name);
s = psensor_create(sid, strdup(name),
- SENSOR_TYPE_NVIDIA, values_max_length);
+ SENSOR_TYPE_NVIDIA_TEMP, values_len);
s->nvidia_id = id;
return s;
}
-int nvidia_init()
+/*
+ Opens connection to X server and returns the number
+ of NVidia GPUs.
+
+ Return 0 if no NVidia gpus or cannot get information.
+*/
+static int init()
{
- int event_base, error_base;
- int num_gpus;
+ int evt, err, n;
- nvidia_sensors_dpy = XOpenDisplay(NULL);
+ display = XOpenDisplay(NULL);
- if (!nvidia_sensors_dpy) {
- fprintf(stderr, _("ERROR: nvidia initialization failure\n"));
+ if (!display) {
+ fprintf(stderr,
+ _("ERROR: Cannot open connection to X Server\n"));
return 0;
}
- if (XNVCTRLQueryExtension(nvidia_sensors_dpy, &event_base,
- &error_base)) {
- if (XNVCTRLQueryTargetCount(nvidia_sensors_dpy,
- NV_CTRL_TARGET_TYPE_GPU,
- &num_gpus)) {
- return num_gpus;
- }
-
- }
+ if (XNVCTRLQueryExtension(display, &evt, &err) &&
+ XNVCTRLQueryTargetCount(display, NV_CTRL_TARGET_TYPE_GPU, &n))
+ return n;
- fprintf(stderr, _("ERROR: nvidia initialization failure: %d\n"),
- error_base);
+ fprintf(stderr, _("ERROR: Cannot retrieve NVidia information\n"));
return 0;
}
void nvidia_psensor_list_update(struct psensor **sensors)
{
- struct psensor **s_ptr = sensors;
-
- while (*s_ptr) {
- struct psensor *sensor = *s_ptr;
+ struct psensor **ss, *s;
- if (sensor->type == SENSOR_TYPE_NVIDIA) {
- int val = nvidia_get_sensor_temp(sensor);
+ ss = sensors;
+ while (*ss) {
+ s = *ss;
- psensor_set_current_value(sensor, (double)val);
- }
+ if (s->type == SENSOR_TYPE_NVIDIA_TEMP)
+ psensor_set_current_value(s, get_temp(s));
- s_ptr++;
+ ss++;
}
}
struct psensor **nvidia_psensor_list_add(struct psensor **sensors,
- int values_max_length)
+ int values_len)
{
- int i;
- int nvidia_gpus_count = nvidia_init();
- struct psensor **res = sensors;
-
+ int i, n;
+ struct psensor **tmp, **ss, *s;
- if (!nvidia_gpus_count) {
- fprintf(stderr,
- _("ERROR: "
- "no nvidia chips or initialization failure\n"));
- }
+ n = init();
- for (i = 0; i < nvidia_gpus_count; i++) {
- struct psensor *sensor
- = nvidia_create_sensor(i, values_max_length);
+ ss = sensors;
+ for (i = 0; i < n; i++) {
+ s = create_sensor(i, values_len);
- struct psensor **tmp_psensors = psensor_list_add(res,
- sensor);
+ tmp = psensor_list_add(ss, s);
- if (res != sensors)
- free(res);
+ if (ss != tmp)
+ free(ss);
- res = tmp_psensors;
+ ss = tmp;
}
- return res;
+ return ss;
+}
+
+void nvidia_cleanup()
+{
+ if (display) {
+ XCloseDisplay(display);
+ display = NULL;
+ }
}