fixed wrong background color, get the background color of
[psensor.git] / src / graph.c
1 /*
2  * Copyright (C) 2010-2012 jeanfi@gmail.com
3  *
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.
8  *
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.
13  *
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
17  * 02110-1301 USA
18  */
19 #include <stdlib.h>
20 #include <sys/time.h>
21 #include <gtk/gtk.h>
22
23 #include "cfg.h"
24 #include "psensor.h"
25
26 /* horizontal padding */
27 #define GRAPH_H_PADDING 4
28 /* vertical padding */
29 #define GRAPH_V_PADDING 4
30
31 static time_t get_graph_end_time_s()
32 {
33         struct timeval tv;
34
35         if (gettimeofday(&tv, NULL) == 0)
36                 return tv.tv_sec;
37         else
38                 return 0;
39 }
40
41 static time_t get_graph_begin_time_s(struct config *cfg)
42 {
43         int ct;
44
45         ct = get_graph_end_time_s();
46
47         if (!ct)
48                 return 0;
49
50         return ct - cfg->graph_monitoring_duration * 60;
51 }
52
53 static int compute_y(double value, double min, double max, int height, int off)
54 {
55         double t = value - min;
56         return height - ((double)height * (t / (max - min))) + off;
57 }
58
59 static char *time_to_str(time_t s)
60 {
61         char *str;
62         /* note: localtime returns a static field, no free required */
63         struct tm *tm = localtime(&s);
64
65         if (!tm)
66                 return NULL;
67
68         str = malloc(6);
69         strftime(str, 6, "%H:%M", tm);
70
71         return str;
72 }
73
74 static void
75 draw_graph_background(cairo_t *cr,
76                       int g_xoff, int g_yoff,
77                       int g_width, int g_height,
78                       int width, int height, struct config *config,
79                       GtkWidget *widget,
80                       GtkWidget *window)
81 {
82         GtkStyleContext *style_ctx;
83         struct color *bgcolor;
84         GdkRGBA rgba;
85
86         bgcolor = config->graph_bgcolor;
87
88         style_ctx = gtk_widget_get_style_context(window);
89         gtk_style_context_get_background_color(style_ctx,
90                                                GTK_STATE_FLAG_NORMAL,
91                                                &rgba);
92
93         if (config->alpha_channel_enabled)
94                 cairo_set_source_rgba(cr,
95                                       rgba.red,
96                                       rgba.green,
97                                       rgba.blue,
98                                       config->graph_bg_alpha);
99         else
100                 cairo_set_source_rgb(cr,
101                                      rgba.red,
102                                      rgba.green,
103                                      rgba.blue);
104
105         cairo_rectangle(cr, 0, 0, width, height);
106         cairo_fill(cr);
107         if (config->alpha_channel_enabled)
108                 cairo_set_source_rgba(cr,
109                                       bgcolor->f_red,
110                                       bgcolor->f_green,
111                                       bgcolor->f_blue,
112                                       config->graph_bg_alpha);
113         else
114                 cairo_set_source_rgb(cr,
115                                      bgcolor->f_red,
116                                      bgcolor->f_green,
117                                      bgcolor->f_blue);
118
119         cairo_rectangle(cr, g_xoff, g_yoff, g_width, g_height);
120         cairo_fill(cr);
121 }
122
123 /* setup dash style */
124 static double dashes[] = {
125         1.0,            /* ink */
126         2.0,            /* skip */
127 };
128 static int ndash = sizeof(dashes) / sizeof(dashes[0]);
129
130 static void draw_background_lines(cairo_t *cr,
131                                   struct color *color,
132                                   int g_width, int g_height,
133                                   int g_xoff, int g_yoff,
134                                   int min, int max)
135 {
136         int i;
137
138         /* draw background lines */
139         cairo_set_line_width(cr, 1);
140         cairo_set_dash(cr, dashes, ndash, 0);
141         cairo_set_source_rgb(cr,
142                              color->f_red, color->f_green, color->f_blue);
143
144         /* vertical lines representing time steps */
145         for (i = 0; i <= 5; i++) {
146                 int x = i * (g_width / 5) + g_xoff;
147                 cairo_move_to(cr, x, g_yoff);
148                 cairo_line_to(cr, x, g_yoff + g_height);
149                 cairo_stroke(cr);
150         }
151
152         /* horizontal lines draws a line for each 10C step */
153         for (i = min; i < max; i++) {
154                 if (i % 10 == 0) {
155                         int y = compute_y(i, min, max, g_height, g_yoff);
156
157                         cairo_move_to(cr, g_xoff, y);
158                         cairo_line_to(cr, g_xoff + g_width, y);
159                         cairo_stroke(cr);
160                 }
161         }
162
163         /* back to normal line style */
164         cairo_set_dash(cr, 0, 0, 0);
165 }
166
167 static void draw_sensor_curve(struct psensor *s,
168                               cairo_t *cr,
169                               double min,
170                               double max,
171                               int bt,
172                               int et,
173                               int g_width,
174                               int g_height,
175                               int g_xoff,
176                               int g_yoff)
177 {
178         int first, i, x, y, t, dt, vdt;
179         double v;
180
181         cairo_set_source_rgb(cr,
182                              s->color->f_red,
183                              s->color->f_green,
184                              s->color->f_blue);
185
186         dt = et - bt;
187         first = 1;
188         for (i = 0; i < s->values_max_length; i++) {
189                 t = s->measures[i].time.tv_sec;
190                 v = s->measures[i].value.d_num;
191
192                 if (v == UNKNOWN_DBL_VALUE || !t)
193                         continue;
194
195                 vdt = t - bt;
196                 if (vdt < 0)
197                         continue;
198
199                 x = vdt * g_width / dt + g_xoff;
200
201                 y = compute_y(v, min, max, g_height, g_yoff);
202
203                 if (first) {
204                         cairo_move_to(cr, x, y);
205                         first = 0;
206                 } else {
207                         cairo_line_to(cr, x, y);
208                 }
209
210         }
211         cairo_stroke(cr);
212 }
213
214
215 void
216 graph_update(struct psensor **sensors,
217              GtkWidget *w_graph,
218              struct config *config,
219              GtkWidget *window)
220 {
221         struct color *fgcolor = config->graph_fgcolor;
222         int et, bt, width, height, g_width, g_height;
223         double min_rpm, max_rpm, mint, maxt;
224         char *strmin, *strmax;
225         /* horizontal and vertical offset of the graph */
226         int g_xoff, g_yoff;
227         cairo_surface_t *cst;
228         cairo_t *cr, *cr_pixmap;
229         char *str_btime, *str_etime;
230         cairo_text_extents_t te_btime, te_etime, te_max, te_min;
231         struct psensor **sensor_cur;
232         GtkAllocation galloc;
233         GtkStyleContext *style_ctx;
234         GdkRGBA rgba;
235
236         if (!gtk_widget_is_drawable(w_graph))
237                 return ;
238
239         min_rpm = get_min_rpm(sensors);
240         max_rpm = get_max_rpm(sensors);
241
242         mint = get_min_temp(sensors);
243         strmin = psensor_value_to_string(SENSOR_TYPE_TEMP, mint);
244
245         maxt = get_max_temp(sensors);
246         strmax = psensor_value_to_string(SENSOR_TYPE_TEMP, maxt);
247
248         str_btime = time_to_str(get_graph_begin_time_s(config));
249         str_etime = time_to_str(get_graph_end_time_s());
250
251         gtk_widget_get_allocation(w_graph, &galloc);
252         width = galloc.width;
253         height = galloc.height;
254
255         cst = cairo_image_surface_create(CAIRO_FORMAT_ARGB32, width, height);
256         cr = cairo_create(cst);
257
258         cairo_select_font_face(cr,
259                                "sans-serif",
260                                CAIRO_FONT_SLANT_NORMAL,
261                                CAIRO_FONT_WEIGHT_NORMAL);
262         cairo_set_font_size(cr, 10.0);
263
264         cairo_text_extents(cr, str_etime, &te_etime);
265         cairo_text_extents(cr, str_btime, &te_btime);
266         cairo_text_extents(cr, strmax, &te_max);
267         cairo_text_extents(cr, strmin, &te_min);
268
269         g_yoff = GRAPH_V_PADDING;
270
271         g_height = height - GRAPH_V_PADDING;
272         if (te_etime.height > te_btime.height)
273                 g_height -= GRAPH_V_PADDING + te_etime.height + GRAPH_V_PADDING;
274         else
275                 g_height -= GRAPH_V_PADDING + te_btime.height + GRAPH_V_PADDING;
276
277         if (te_min.width > te_max.width)
278                 g_xoff = (2 * GRAPH_H_PADDING) + te_max.width;
279         else
280                 g_xoff = (2 * GRAPH_H_PADDING) + te_min.width;
281
282         g_width = width - g_xoff - GRAPH_H_PADDING;
283
284         draw_graph_background(cr,
285                               g_xoff, g_yoff, g_width, g_height,
286                               width, height, config,
287                               w_graph,
288                               window);
289
290         /** Set the color for text drawing */
291         style_ctx = gtk_widget_get_style_context(w_graph);
292         gtk_style_context_get_color(style_ctx, GTK_STATE_FLAG_NORMAL, &rgba);
293         cairo_set_source_rgb(cr, rgba.red, rgba.green, rgba.blue);
294
295         /* draw graph begin time */
296         cairo_move_to(cr, g_xoff, height - GRAPH_V_PADDING);
297         cairo_show_text(cr, str_btime);
298         free(str_btime);
299
300         /* draw graph end time */
301         cairo_move_to(cr,
302                       width - te_etime.width - GRAPH_H_PADDING,
303                       height - GRAPH_V_PADDING);
304         cairo_show_text(cr, str_etime);
305         free(str_etime);
306
307         /* draw min and max temp */
308         cairo_move_to(cr, GRAPH_H_PADDING, te_max.height + GRAPH_V_PADDING);
309         cairo_show_text(cr, strmax);
310         free(strmax);
311
312         cairo_move_to(cr,
313                       GRAPH_H_PADDING, height - (te_min.height / 2) - g_yoff);
314         cairo_show_text(cr, strmin);
315         free(strmin);
316
317         draw_background_lines(cr, fgcolor,
318                               g_width, g_height,
319                               g_xoff, g_yoff,
320                               mint, maxt);
321
322         /* .. and finaly draws the temperature graphs */
323         bt = get_graph_begin_time_s(config);
324         et = get_graph_end_time_s();
325
326         if (bt && et) {
327                 sensor_cur = sensors;
328
329                 cairo_set_line_join(cr, CAIRO_LINE_JOIN_ROUND);
330                 cairo_set_line_width(cr, 1);
331                 while (*sensor_cur) {
332                         struct psensor *s = *sensor_cur;
333
334                         if (s->enabled) {
335                                 double min, max;
336
337                                 if (is_fan_type(s->type)) {
338                                         min = min_rpm;
339                                         max = max_rpm;
340                                 } else if (s->type & SENSOR_TYPE_CPU_USAGE) {
341                                         min = 0;
342                                         max = get_max_value
343                                                 (sensors,
344                                                  SENSOR_TYPE_CPU_USAGE);
345                                 } else {
346                                         min = mint;
347                                         max = maxt;
348                                 }
349
350                                 draw_sensor_curve(s, cr,
351                                                   min, max,
352                                                   bt, et,
353                                                   g_width, g_height,
354                                                   g_xoff, g_yoff);
355                         }
356
357                         sensor_cur++;
358                 }
359         }
360
361         cr_pixmap = gdk_cairo_create(gtk_widget_get_window(w_graph));
362
363         if (cr_pixmap) {
364
365                 if (config->alpha_channel_enabled)
366                         cairo_set_operator(cr_pixmap, CAIRO_OPERATOR_SOURCE);
367
368                 cairo_set_source_surface(cr_pixmap, cst, 0, 0);
369                 cairo_paint(cr_pixmap);
370         }
371
372         cairo_destroy(cr_pixmap);
373         cairo_surface_destroy(cst);
374         cairo_destroy(cr);
375 }