realise seamless integration of new_bar_in_shell
[monky] / src / specials.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) 2004, Hannu Saransaari and Lauri Hakkarainen
13  * Copyright (c) 2005-2009 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 #include "conky.h"
31 #include "colours.h"
32 #ifdef X11
33 #include "fonts.h"
34 #endif /* X11 */
35 #include "logging.h"
36 #include "specials.h"
37 #include <math.h>
38
39 /* maximum number of special things, e.g. fonts, offsets, aligns, etc. */
40 int max_specials = MAX_SPECIALS_DEFAULT;
41
42 /* create specials array on heap instead of stack with introduction of
43  * max_specials */
44 struct special_t *specials = NULL;
45
46 int special_count;
47
48 int default_bar_width = 0, default_bar_height = 6;
49 #ifdef X11
50 int default_graph_width = 0, default_graph_height = 25;
51 int default_gauge_width = 40, default_gauge_height = 25;
52 #endif /* X11 */
53
54 /*
55  * Special data typedefs
56  */
57
58 struct bar {
59         int width, height;
60 };
61
62 struct gauge {
63         int width, height;
64 };
65
66 struct graph {
67         int width, height;
68         unsigned int first_colour, last_colour;
69         unsigned int scale, showaslog;
70         char tempgrad;
71 };
72
73 struct stippled_hr {
74         int height, arg;
75 };
76
77 struct tab {
78         int width, arg;
79 };
80
81 /*
82  * Scanning arguments to various special text objects
83  */
84
85 #ifdef X11
86 const char *scan_gauge(struct text_object *obj, const char *args)
87 {
88         struct gauge *g;
89
90         g = malloc(sizeof(struct gauge));
91         memset(g, 0, sizeof(struct gauge));
92
93         /*width and height*/
94         g->width = default_gauge_width;
95         g->height = default_gauge_height;
96
97         /* gauge's argument is either height or height,width */
98         if (args) {
99                 int n = 0;
100
101                 if (sscanf(args, "%d,%d %n", &g->height, &g->width, &n) <= 1) {
102                         if (sscanf(args, "%d %n", &g->height, &n) == 2) {
103                                 g->width = g->height; /*square gauge*/
104                         }
105                 }
106                 args += n;
107         }
108
109         obj->special_data = g;
110         return args;
111 }
112 #endif /* X11 */
113
114 const char *scan_bar(struct text_object *obj, const char *args)
115 {
116         struct bar *b;
117
118         b = malloc(sizeof(struct bar));
119         memset(b, 0, sizeof(struct bar));
120
121         /* zero width means all space that is available */
122         b->width = default_bar_width;
123         b->height = default_bar_height;
124         /* bar's argument is either height or height,width */
125         if (args) {
126                 int n = 0;
127
128                 if (sscanf(args, "%d,%d %n", &b->height, &b->width, &n) <= 1) {
129                         sscanf(args, "%d %n", &b->height, &n);
130                 }
131                 args += n;
132         }
133
134         obj->special_data = b;
135         return args;
136 }
137
138 #ifdef X11
139 char *scan_font(const char *args)
140 {
141         if (args && *args) {
142                 return strndup(args, DEFAULT_TEXT_BUFFER_SIZE);
143         }
144
145         return NULL;
146 }
147
148 char *scan_graph(struct text_object *obj, const char *args, int defscale)
149 {
150         struct graph *g;
151         char buf[1024];
152         memset(buf, 0, 1024);
153
154         g = malloc(sizeof(struct graph));
155         memset(g, 0, sizeof(struct graph));
156         obj->special_data = g;
157
158         /* zero width means all space that is available */
159         g->width = default_graph_width;
160         g->height = default_graph_height;
161         g->first_colour = 0;
162         g->last_colour = 0;
163         g->scale = defscale;
164         g->tempgrad = FALSE;
165         g->showaslog = FALSE;
166         if (args) {
167                 if (strstr(args, " "TEMPGRAD) || strncmp(args, TEMPGRAD, strlen(TEMPGRAD)) == 0) {
168                         g->tempgrad = TRUE;
169                 }
170                 if (strstr(args, " "LOGGRAPH) || strncmp(args, LOGGRAPH, strlen(LOGGRAPH)) == 0) {
171                         g->showaslog = TRUE;
172                 }
173                 if (sscanf(args, "%d,%d %x %x %u", &g->height, &g->width, &g->first_colour, &g->last_colour, &g->scale) == 5) {
174                         return NULL;
175                 }
176                 g->scale = defscale;
177                 if (sscanf(args, "%d,%d %x %x", &g->height, &g->width, &g->first_colour, &g->last_colour) == 4) {
178                         return NULL;
179                 }
180                 if (sscanf(args, "%1023s %d,%d %x %x %u", buf, &g->height, &g->width, &g->first_colour, &g->last_colour, &g->scale) == 6) {
181                         return strndup(buf, text_buffer_size);
182                 }
183                 g->scale = defscale;
184                 if (sscanf(args, "%1023s %d,%d %x %x", buf, &g->height, &g->width, &g->first_colour, &g->last_colour) == 5) {
185                         return strndup(buf, text_buffer_size);
186                 }
187                 buf[0] = '\0';
188                 g->height = 25;
189                 g->width = 0;
190                 if (sscanf(args, "%x %x %u", &g->first_colour, &g->last_colour, &g->scale) == 3) {
191                         return NULL;
192                 }
193                 g->scale = defscale;
194                 if (sscanf(args, "%x %x", &g->first_colour, &g->last_colour) == 2) {
195                         return NULL;
196                 }
197                 if (sscanf(args, "%1023s %x %x %u", buf, &g->first_colour, &g->last_colour, &g->scale) == 4) {
198                         return strndup(buf, text_buffer_size);
199                 }
200                 g->scale = defscale;
201                 if (sscanf(args, "%1023s %x %x", buf, &g->first_colour, &g->last_colour) == 3) {
202                         return strndup(buf, text_buffer_size);
203                 }
204                 buf[0] = '\0';
205                 g->first_colour = 0;
206                 g->last_colour = 0;
207                 if (sscanf(args, "%d,%d %u", &g->height, &g->width, &g->scale) == 3) {
208                         return NULL;
209                 }
210                 g->scale = defscale;
211                 if (sscanf(args, "%d,%d", &g->height, &g->width) == 2) {
212                         return NULL;
213                 }
214                 if (sscanf(args, "%1023s %d,%d %u", buf, &g->height, &g->width, &g->scale) < 4) {
215                         g->scale = defscale;
216                         //TODO: check the return value and throw an error?
217                         sscanf(args, "%1023s %d,%d", buf, &g->height, &g->width);
218                 }
219 #undef g
220
221                 return strndup(buf, text_buffer_size);
222         }
223
224         if (buf[0] == '\0') {
225                 return NULL;
226         } else {
227                 return strndup(buf, text_buffer_size);
228         }
229 }
230 #endif /* X11 */
231
232 /*
233  * Printing various special text objects
234  */
235
236 static struct special_t *new_special(char *buf, enum special_types t)
237 {
238         if (special_count >= max_specials) {
239                 CRIT_ERR(NULL, NULL, "too many special things in text");
240         }
241
242         buf[0] = SPECIAL_CHAR;
243         buf[1] = '\0';
244         specials[special_count].type = t;
245         return &specials[special_count++];
246 }
247
248 #ifdef X11
249 void new_gauge(struct text_object *obj, char *buf, int usage)
250 {
251         struct special_t *s = 0;
252         struct gauge *g = obj->special_data;
253
254         if ((output_methods & TO_X) == 0)
255                 return;
256
257         if (!g)
258                 return;
259
260         s = new_special(buf, GAUGE);
261
262         s->arg = (usage > 255) ? 255 : ((usage < 0) ? 0 : usage);
263         s->width = g->width;
264         s->height = g->height;
265 }
266
267 void new_font(char *buf, char *args)
268 {
269         if ((output_methods & TO_X) == 0)
270                 return;
271
272         if (args) {
273                 struct special_t *s = new_special(buf, FONT);
274
275                 if (s->font_added > font_count || !s->font_added || (strncmp(args, fonts[s->font_added].name, DEFAULT_TEXT_BUFFER_SIZE) != EQUAL) ) {
276                         int tmp = selected_font;
277
278                         selected_font = s->font_added = add_font(args);
279                         selected_font = tmp;
280                 }
281         } else {
282                 struct special_t *s = new_special(buf, FONT);
283                 int tmp = selected_font;
284
285                 selected_font = s->font_added = 0;
286                 selected_font = tmp;
287         }
288 }
289
290 static void graph_append(struct special_t *graph, double f, char showaslog)
291 {
292         int i;
293
294         if (showaslog) {
295 #ifdef MATH
296                 f = log10(f + 1);
297 #endif
298         }
299
300         if (!graph->scaled && f > graph->graph_scale) {
301                 f = graph->graph_scale;
302         }
303
304         graph->graph[0] = f;    /* add new data */
305         /* shift all the data by 1 */
306         for (i = graph->graph_width - 1; i > 0; i--) {
307                 graph->graph[i] = graph->graph[i - 1];
308                 if (graph->scaled && graph->graph[i - 1] > graph->graph_scale) {
309                         /* check if we need to update the scale */
310                         graph->graph_scale = graph->graph[i - 1];
311                 }
312         }
313         if (graph->scaled && graph->graph[graph->graph_width] > graph->graph_scale) {
314                 /* check if we need to update the scale */
315                 graph->graph_scale = graph->graph[graph->graph_width];
316         }
317 }
318
319 void new_graph(struct text_object *obj, char *buf, double val)
320 {
321         struct special_t *s = 0;
322         struct graph *g = obj->special_data;
323
324         if ((output_methods & TO_X) == 0)
325                 return;
326
327         if (!g)
328                 return;
329
330         s = new_special(buf, GRAPH);
331
332         s->width = g->width;
333         if (s->graph == NULL) {
334                 if (s->width > 0 && s->width < MAX_GRAPH_DEPTH) {
335                         // subtract 2 for the box
336                         s->graph_width = s->width /* - 2 */;
337                 } else {
338                         s->graph_width = MAX_GRAPH_DEPTH - 2;
339                 }
340                 s->graph = malloc(s->graph_width * sizeof(double));
341                 memset(s->graph, 0, s->graph_width * sizeof(double));
342                 s->graph_scale = 100;
343         }
344         s->height = g->height;
345         s->first_colour = adjust_colours(g->first_colour);
346         s->last_colour = adjust_colours(g->last_colour);
347         if (g->scale != 0) {
348                 s->scaled = 0;
349                 s->graph_scale = g->scale;
350                 s->show_scale = 0;
351         } else {
352                 s->scaled = 1;
353                 s->graph_scale = 1;
354                 s->show_scale = 1;
355         }
356         s->tempgrad = g->tempgrad;
357         /* if (s->width) {
358                 s->graph_width = s->width - 2;  // subtract 2 for rectangle around
359         } */
360 #ifdef MATH
361         if (g->showaslog) {
362                 s->graph_scale = log10(s->graph_scale + 1);
363         }
364 #endif
365         graph_append(s, val, g->showaslog);
366 }
367
368 void new_hr(char *buf, int a)
369 {
370         if ((output_methods & TO_X) == 0)
371                 return;
372
373         new_special(buf, HORIZONTAL_LINE)->height = a;
374 }
375
376 void scan_stippled_hr(struct text_object *obj, const char *arg)
377 {
378         struct stippled_hr *sh;
379
380         sh = malloc(sizeof(struct stippled_hr));
381         memset(sh, 0, sizeof(struct stippled_hr));
382
383         sh->arg = get_stippled_borders();
384         sh->height = 1;
385
386         if (arg) {
387                 if (sscanf(arg, "%d %d", &sh->arg, &sh->height) != 2) {
388                         sscanf(arg, "%d", &sh->height);
389                 }
390         }
391         if (sh->arg <= 0) {
392                 sh->arg = 1;
393         }
394         obj->special_data = sh;
395 }
396
397 void new_stippled_hr(struct text_object *obj, char *buf)
398 {
399         struct special_t *s = 0;
400         struct stippled_hr *sh = obj->special_data;
401
402         if ((output_methods & TO_X) == 0)
403                 return;
404
405         if (!sh)
406                 return;
407
408         s = new_special(buf, STIPPLED_HR);
409
410         s->height = sh->height;
411         s->arg = sh->arg;
412 }
413 #endif /* X11 */
414
415 void new_fg(char *buf, long c)
416 {
417 #ifdef X11
418         if (output_methods & TO_X)
419                 new_special(buf, FG)->arg = c;
420 #endif /* X11 */
421 #ifdef NCURSES
422         if (output_methods & TO_NCURSES)
423                 new_special(buf, FG)->arg = c;
424 #endif /* NCURSES */
425         UNUSED(buf);
426         UNUSED(c);
427 }
428
429 #ifdef X11
430 void new_bg(char *buf, long c)
431 {
432         if ((output_methods & TO_X) == 0)
433                 return;
434
435         new_special(buf, BG)->arg = c;
436 }
437 #endif /* X11 */
438
439 static void new_bar_in_shell(struct text_object *obj, char* buffer, int buf_max_size, double usage)
440 {
441         struct bar *b = obj->special_data;
442         int width, i, scaledusage;
443
444         if (!b)
445                 return;
446
447         width = b->width;
448         if (!width)
449                 width = DEFAULT_BAR_WIDTH_NO_X;
450
451         if (width > buf_max_size)
452                 width = buf_max_size;
453
454         scaledusage = round_to_int( usage * width / 255);
455
456         for (i = 0; i < scaledusage; i++)
457                 buffer[i] = '#';
458
459         for (; i < width; i++)
460                 buffer[i] = '_';
461
462         buffer[i] = 0;
463 }
464
465 #ifdef X11
466 static void new_bar_in_x11(struct text_object *obj, char *buf, int usage)
467 {
468         struct special_t *s = 0;
469         struct bar *b = obj->special_data;
470
471         if ((output_methods & TO_X) == 0)
472                 return;
473
474         if (!b)
475                 return;
476
477         s = new_special(buf, BAR);
478
479         s->arg = usage;
480         s->width = b->width;
481         s->height = b->height;
482 }
483 #endif /* X11 */
484
485 /* usage is in range [0,255] */
486 void new_bar(struct text_object *obj, char *p, int p_max_size, int usage)
487 {
488         if (!p_max_size)
489                 return;
490
491         usage = (usage > 255) ? 255 : ((usage < 0) ? 0 : usage);
492
493 #ifdef X11
494         if ((output_methods & TO_X))
495                 new_bar_in_x11(obj, p, usage);
496         else
497 #endif /* X11 */
498                 new_bar_in_shell(obj, p, p_max_size, usage);
499 }
500
501 void new_outline(char *buf, long c)
502 {
503         new_special(buf, OUTLINE)->arg = c;
504 }
505
506 void new_offset(char *buf, long c)
507 {
508         new_special(buf, OFFSET)->arg = c;
509 }
510
511 void new_voffset(char *buf, long c)
512 {
513         new_special(buf, VOFFSET)->arg = c;
514 }
515
516 void new_alignr(char *buf, long c)
517 {
518         new_special(buf, ALIGNR)->arg = c;
519 }
520
521 // A postive offset pushes the text further left
522 void new_alignc(char *buf, long c)
523 {
524         new_special(buf, ALIGNC)->arg = c;
525 }
526
527 void new_goto(char *buf, long c)
528 {
529         new_special(buf, GOTO)->arg = c;
530 }
531
532 void scan_tab(struct text_object *obj, const char *arg)
533 {
534         struct tab *t;
535
536         t = malloc(sizeof(struct tab));
537         memset(t, 0, sizeof(struct tab));
538
539         t->width = 10;
540         t->arg = 0;
541
542         if (arg) {
543                 if (sscanf(arg, "%d %d", &t->width, &t->arg) != 2) {
544                         sscanf(arg, "%d", &t->arg);
545                 }
546         }
547         if (t->width <= 0) {
548                 t->width = 1;
549         }
550         obj->special_data = t;
551 }
552
553 void new_tab(struct text_object *obj, char *buf)
554 {
555         struct special_t *s = 0;
556         struct tab *t = obj->special_data;
557
558         if (!t)
559                 return;
560
561         s = new_special(buf, TAB);
562         s->width = t->width;
563         s->arg = t->arg;
564 }