vcs.maemo.org Git - monky/blob - src/nvidia.c

   1 /* -*- mode: c; c-basic-offset: 4; tab-width: 4; indent-tabs-mode: t -*-
   2  * vim: ts=4 sw=4 noet ai cindent syntax=c
   3  *
   4  * Conky, a system monitor, based on torsmo
   5  *
   6  * Any original torsmo code is licensed under the BSD license
   7  *
   8  * All code written since the fork of torsmo is licensed under the GPL
   9  *
  10  * Please see COPYING for details
  11  *
  12  * Copyright (c) 2008 Markus Meissner
  13  * Copyright (c) 2005-2010 Brenden Matthews, Philip Kovacs, et. al.
  14  *      (see AUTHORS)
  15  * All rights reserved.
  16  *
  17  * This program is free software: you can redistribute it and/or modify
  18  * it under the terms of the GNU General Public License as published by
  19  * the Free Software Foundation, either version 3 of the License, or
  20  * (at your option) any later version.
  21  *
  22  * This program is distributed in the hope that it will be useful,
  23  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  24  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  25  * GNU General Public License for more details.
  26  * You should have received a copy of the GNU General Public License
  27  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  28  *
  29  */
  30
  31 #include "conky.h"
  32 #include "logging.h"
  33 #include "nvidia.h"
  34 #include "temphelper.h"
  35 #include <NVCtrl/NVCtrlLib.h>
  36
  37 const int nvidia_query_to_attr[] = {NV_CTRL_GPU_CORE_TEMPERATURE,
  38                                     NV_CTRL_GPU_CORE_THRESHOLD,
  39                                     NV_CTRL_AMBIENT_TEMPERATURE,
  40                                     NV_CTRL_GPU_CURRENT_CLOCK_FREQS,
  41                                     NV_CTRL_GPU_CURRENT_CLOCK_FREQS,
  42                                     NV_CTRL_IMAGE_SETTINGS};
  43
  44 typedef enum _QUERY_ID {
  45         NV_TEMP,
  46         NV_TEMP_THRESHOLD,
  47         NV_TEMP_AMBIENT,
  48         NV_GPU_FREQ,
  49         NV_MEM_FREQ,
  50         NV_IMAGE_QUALITY
  51 } QUERY_ID;
  52
  53 struct nvidia_s {
  54         int interval;
  55         int print_as_float;
  56         QUERY_ID type;
  57 };
  58
  59 static Display *nvdisplay;
  60
  61 static int get_nvidia_value(QUERY_ID qid){
  62         int tmp;
  63         Display *dpy = nvdisplay ? nvdisplay : display;
  64         if(!dpy || !XNVCTRLQueryAttribute(dpy, 0, 0, nvidia_query_to_attr[qid], &tmp)){
  65                 return -1;
  66         }
  67         /* FIXME: when are the low 2 bytes of NV_GPU_FREQ needed? */
  68         if (qid == NV_GPU_FREQ)
  69                 return tmp >> 16;
  70         if (qid == NV_MEM_FREQ)
  71                 return tmp & 0xFFFF;
  72         return tmp;
  73 }
  74
  75 int set_nvidia_type(struct text_object *obj, const char *arg)
  76 {
  77         struct nvidia_s *nvs;
  78
  79         nvs = obj->data.opaque = malloc(sizeof(struct nvidia_s));
  80         memset(nvs, 0, sizeof(struct nvidia_s));
  81
  82         switch(arg[0]) {
  83                 case 't':                              // temp or threshold
  84                         nvs->print_as_float = 1;
  85                         if (arg[1] == 'e')
  86                                 nvs->type = NV_TEMP;
  87                         else if (arg[1] == 'h')
  88                                 nvs->type = NV_TEMP_THRESHOLD;
  89                         else
  90                                 return 1;
  91                         break;
  92                 case 'a':                              // ambient temp
  93                         nvs->print_as_float = 1;
  94                         nvs->type = NV_TEMP_AMBIENT;
  95                         break;
  96                 case 'g':                              // gpufreq
  97                         nvs->type = NV_GPU_FREQ;
  98                         break;
  99                 case 'm':                              // memfreq
 100                         nvs->type = NV_MEM_FREQ;
 101                         break;
 102                 case 'i':                              // imagequality
 103                         nvs->type = NV_IMAGE_QUALITY;
 104                         break;
 105                 default:
 106                         return 1;
 107         }
 108         return 0;
 109 }
 110
 111 void print_nvidia_value(struct text_object *obj, char *p, int p_max_size)
 112 {
 113         int value;
 114         struct nvidia_s *nvs = obj->data.opaque;
 115
 116         if (!nvs ||
 117             (value = get_nvidia_value(nvs->type)) == -1) {
 118                 snprintf(p, p_max_size, "N/A");
 119                 return;
 120         }
 121         if (nvs->type == NV_TEMP)
 122                 temp_print(p, p_max_size, (double)value, TEMP_CELSIUS);
 123         else if (nvs->print_as_float &&
 124                         value > 0 && value < 100)
 125                 snprintf(p, p_max_size, "%.1f", (float)value);
 126         else
 127                 snprintf(p, p_max_size, "%d", value);
 128 }
 129
 130 void free_nvidia(struct text_object *obj)
 131 {
 132         if (obj->data.opaque) {
 133                 free(obj->data.opaque);
 134                 obj->data.opaque = NULL;
 135         }
 136 }
 137
 138 void set_nvidia_display(const char *disp)
 139 {
 140         if(nvdisplay) {
 141                 XCloseDisplay(nvdisplay);
 142                 nvdisplay = NULL;
 143         }
 144         if(disp) {
 145                 if ((nvdisplay = XOpenDisplay(disp)) == NULL) {
 146                         CRIT_ERR(NULL, NULL, "can't open nvidia display: %s", XDisplayName(disp));
 147                 }
 148         }
 149 }