widgets/graph.c

   1 /*
   2  * graph.c - a graph widget
   3  *
   4  * Copyright © 2007-2008 Julien Danjou <julien@danjou.info>
   5  * Copyright © 2007-2008 Marco Candrian <mac@calmar.ws>
   6  *
   7  * This program is free software; you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published by
   9  * the Free Software Foundation; either version 2 of the License, or
  10  * (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License along
  18  * with this program; if not, write to the Free Software Foundation, Inc.,
  19  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  20  *
  21  */
  22
  23 #include <cairo.h>
  24 #include "common/draw.h"
  25 #include "widget.h"
  26 #include "screen.h"
  27 #include "common/util.h"
  28
  29 extern AwesomeConf globalconf;
  30
  31 typedef struct
  32 {
  33     /* general layout */
  34     float *max;                 /* Represents a full graph */
  35     int width;                  /* Width of the widget */
  36     float height;               /* Height of graph (0-1, where 1 is height of statusbar) */
  37     int box_height;             /* Height of the innerbox in pixels */
  38     int padding_left;           /* Left padding */
  39     int size;                   /* Size of lines-array (also innerbox-lenght) */
  40     XColor bg;                  /* Background color */
  41     XColor bordercolor;         /* Border color */
  42
  43     /* markers... */
  44     int index;                  /* Index of current (new) value */
  45     int *max_index;             /* Index of the actual maximum value */
  46     float *current_max;         /* Pointer to current maximum value itself */
  47
  48     /* all data is stored here */
  49     int data_items;             /* Number of data-input items */
  50     int **lines;                /* Keeps the calculated values (line-length); */
  51     float **values;             /* Actual values */
  52
  53     /* additional data + a pointer to **lines accordingly */
  54     int **fillbottom;           /* datatypes holder (same as some **lines pointer) */
  55     int fillbottom_total;       /* total of them */
  56     XColor *fillbottom_color;   /* color of them */
  57     int **filltop;              /* datatypes holder */
  58     int filltop_total;          /* total of them */
  59     XColor *filltop_color;      /* color of them */
  60     int **drawline;             /* datatypes holder */
  61     int drawline_total;         /* total of them */
  62     XColor *drawline_color;     /* color of them */
  63
  64     int *draw_from;             /* Preparation/tmp array for draw_graph(); */
  65     int *draw_to;               /* Preparation/tmp array for draw_graph(); */
  66
  67 } Data;
  68
  69 static int
  70 graph_draw(Widget *widget, DrawCtx *ctx, int offset,
  71                  int used __attribute__ ((unused)))
  72 {
  73     int margin_top, left_offset;
  74     int z, y, x, tmp;
  75     Data *d = widget->data;
  76     Area rectangle;
  77     cairo_surface_t *surface;
  78     cairo_t *cr;
  79
  80     if(d->width < 1 || !d->data_items)
  81         return 0;
  82
  83     if(!widget->user_supplied_x)
  84         widget->area.x = widget_calculate_offset(widget->statusbar->width,
  85                                                  d->width,
  86                                                  offset,
  87                                                  widget->alignment);
  88     if(!widget->user_supplied_y)
  89         widget->area.y = 0;
  90
  91     margin_top = (int) (widget->statusbar->height * (1 - d->height)) / 2 + 0.5 + widget->area.y;
  92     left_offset = widget->area.x + d->padding_left;
  93
  94     if(!(d->box_height))
  95         d->box_height = (int) (widget->statusbar->height * d->height + 0.5) - 2;
  96
  97     rectangle.x = left_offset;
  98     rectangle.y = margin_top;
  99     rectangle.width = d->size + 2;
 100     rectangle.height = d->box_height + 2;
 101     draw_rectangle(ctx, rectangle, False, d->bordercolor);
 102
 103     rectangle.x++;
 104     rectangle.y++;
 105     rectangle.width -= 2;
 106     rectangle.height -= 2;
 107     draw_rectangle(ctx, rectangle, True, d->bg);
 108
 109     draw_graph_init(ctx, &surface, &cr); /* setup drawing surface etc */
 110
 111     /* draw style = top */
 112     for(z = 0; z < d->filltop_total; z++)
 113     {
 114         for(y = 0; y < d->size; y++)
 115         {
 116             for(tmp = 0, x = 0; x < d->filltop_total; x++) /* find largest smaller value */
 117             {
 118                 if (x == z)
 119                     continue;
 120
 121                 if(d->filltop[x][y] > tmp && d->filltop[x][y] < d->filltop[z][y])
 122                     tmp = d->filltop[x][y];
 123             }
 124             d->draw_from[y] = d->box_height - tmp;
 125             d->draw_to[y] = d->box_height - d->filltop[z][y];
 126         }
 127         draw_graph(cr,
 128                 left_offset + 2, margin_top + d->box_height + 1,
 129                 d->size, d->draw_from, d->draw_to, d->index,
 130                 d->filltop_color[z]);
 131     }
 132
 133     /* draw style = bottom */
 134     for(z = 0; z < d->fillbottom_total; z++)
 135     {
 136         for(y = 0; y < d->size; y++)
 137         {
 138             for(tmp = 0, x = 0; x < d->fillbottom_total; x++) /* find largest smaller value */
 139             {
 140                 if (x == z)
 141                     continue;
 142
 143                 if(d->fillbottom[x][y] > tmp && d->fillbottom[x][y] < d->fillbottom[z][y])
 144                     tmp = d->fillbottom[x][y];
 145             }
 146             d->draw_from[y] = tmp;
 147         }
 148
 149         draw_graph(cr,
 150                 left_offset + 2, margin_top + d->box_height + 1,
 151                 d->size, d->draw_from, d->fillbottom[z], d->index,
 152                 d->fillbottom_color[z]);
 153     }
 154
 155     /* draw style = line */
 156     for(z = 0; z < d->drawline_total; z++)
 157     {
 158         draw_graph_line(cr,
 159                 left_offset + 2, margin_top + d->box_height + 1,
 160                 d->size, d->drawline[z], d->index,
 161                 d->drawline_color[z]);
 162     }
 163
 164     draw_graph_end(surface, cr);
 165
 166     widget->area.width = d->width;
 167     widget->area.height = widget->statusbar->height;
 168     return widget->area.width;
 169 }
 170
 171 static void
 172 graph_tell(Widget *widget, char *command)
 173 {
 174     Data *d = widget->data;
 175     int i, z;
 176     float *value;
 177     char *tok;
 178
 179     if(!command || d->width < 1 || !(d->data_items > 0))
 180         return;
 181
 182     value = p_new(float, d->data_items);
 183
 184     for (i = 0, tok = strtok(command, ","); tok && i < d->data_items; tok = strtok(NULL, ","), i++)
 185         value[i] = MAX(atof(tok), 0);
 186
 187     if(++d->index >= d->size) /* cycle inside the arrays (all-in-one) */
 188         d->index = 0;
 189
 190     /* add according values and to-draw-line-lenghts to the according data_items */
 191     for(z = 0; z < d->data_items; z++)
 192     {
 193         if(d->values[z]) /* scale option is true */
 194         {
 195             d->values[z][d->index] = value[z];
 196
 197             if(value[z] > d->current_max[z]) /* a new maximum value found */
 198             {
 199                 d->max_index[z] = d->index;
 200                 d->current_max[z] = value[z];
 201
 202                 /* recalculate */
 203                 for (i = 0; i < d->size; i++)
 204                     d->lines[z][i] = (int) (d->values[z][i] * (d->box_height) / d->current_max[z] + 0.5);
 205             }
 206             else if(d->max_index[z] == d->index) /* old max_index reached, re-check/generate */
 207             {
 208                 /* find the new max */
 209                 for (i = 0; i < d->size; i++)
 210                     if (d->values[z][i] > d->values[z][d->max_index[z]])
 211                         d->max_index[z] = i;
 212
 213                 d->current_max[z] = MAX(d->values[z][d->max_index[z]], d->max[z]);
 214
 215                 /* recalculate */
 216                 for (i = 0; i < d->size; i++)
 217                     d->lines[z][i] = (int) (d->values[z][i] * d->box_height / d->current_max[z] + 0.5);
 218             }
 219             else
 220                 d->lines[z][d->index] = (int) (value[z] * d->box_height / d->current_max[z] + 0.5);
 221
 222         }
 223         else /* scale option is false - limit to d->box_height */
 224         {
 225             if (value[z] < d->current_max[z])
 226                 d->lines[z][d->index] = (int) (value[z] * d->box_height / d->current_max[z] + 0.5);
 227             else
 228                 d->lines[z][d->index] = d->box_height;
 229         }
 230     }
 231 }
 232
 233 Widget *
 234 graph_new(Statusbar *statusbar, cfg_t *config)
 235 {
 236     Widget *w;
 237     Data *d;
 238     cfg_t *cfg;
 239     char *color;
 240     int phys_screen = get_phys_screen(statusbar->screen);
 241     int i;
 242     char *type;
 243     XColor tmp_color;
 244
 245     w = p_new(Widget, 1);
 246     widget_common_new(w, statusbar, config);
 247
 248     w->draw = graph_draw;
 249     w->tell = graph_tell;
 250     d = w->data = p_new(Data, 1);
 251
 252     d->width = cfg_getint(config, "width");
 253     d->height = cfg_getfloat(config, "height");
 254     d->padding_left = cfg_getint(config, "padding_left");
 255     d->size = d->width - d->padding_left - 2;
 256
 257     if(d->size < 1)
 258     {
 259         warn("graph widget needs: (width - padding_left) >= 3\n");
 260         return w;
 261     }
 262
 263     if(!(d->data_items = cfg_size(config, "data")))
 264     {
 265         warn("graph widget needs at least one data section\n");
 266         return w;
 267     }
 268
 269     d->draw_from = p_new(int, d->size);
 270     d->draw_to = p_new(int, d->size);
 271
 272     d->fillbottom = p_new(int *, d->size);
 273     d->filltop = p_new(int *, d->size);
 274     d->drawline = p_new(int *, d->size);
 275
 276     d->values = p_new(float *, d->data_items);
 277     d->lines = p_new(int *, d->data_items);
 278
 279     d->filltop_color = p_new(XColor, d->data_items);
 280     d->fillbottom_color = p_new(XColor, d->data_items);
 281     d->drawline_color = p_new(XColor, d->data_items);
 282
 283     d->max_index = p_new(int, d->data_items);
 284
 285     d->current_max = p_new(float, d->data_items);
 286     d->max = p_new(float, d->data_items);
 287
 288     for(i = 0; i < d->data_items; i++)
 289     {
 290         cfg = cfg_getnsec(config, "data", i);
 291
 292         if((color = cfg_getstr(cfg, "fg")))
 293             tmp_color = draw_color_new(globalconf.display, phys_screen, color);
 294         else
 295             tmp_color = globalconf.screens[statusbar->screen].colors_normal[ColFG];
 296
 297         if (cfg_getbool(cfg, "scale"))
 298             d->values[i] = p_new(float, d->size); /* not null -> scale = true */
 299
 300         /* prevent: division by zero */
 301         d->current_max[i] = d->max[i] = cfg_getfloat(cfg, "max");
 302         if(!(d->max[i] > 0))
 303         {
 304             warn("all graph widget needs a 'max' value greater than zero\n");
 305             d->data_items = 0;
 306             return w;
 307         }
 308
 309         d->lines[i] = p_new(int, d->size);
 310
 311         /* filter each style-typ into it's own array (for easy looping later)*/
 312
 313         if ((type = cfg_getstr(cfg, "style")))
 314         {
 315             if(!strncmp(type, "bottom", sizeof("bottom")))
 316             {
 317                 d->fillbottom[d->fillbottom_total] = d->lines[i];
 318                 d->fillbottom_color[d->fillbottom_total] = tmp_color;
 319                 d->fillbottom_total++;
 320             }
 321             else if (!strncmp(type, "top", sizeof("top")))
 322             {
 323                 d->filltop[d->filltop_total] = d->lines[i];
 324                 d->filltop_color[d->filltop_total] = tmp_color;
 325                 d->filltop_total++;
 326             }
 327             else if (!strncmp(type, "line", sizeof("line")))
 328             {
 329                 d->drawline[d->drawline_total] = d->lines[i];
 330                 d->drawline_color[d->drawline_total] = tmp_color;
 331                 d->drawline_total++;
 332             }
 333         }
 334     }
 335
 336     if((color = cfg_getstr(config, "bg")))
 337         d->bg = draw_color_new(globalconf.display, phys_screen, color);
 338     else
 339         d->bg = globalconf.screens[statusbar->screen].colors_normal[ColBG];
 340
 341     if((color = cfg_getstr(config, "bordercolor")))
 342         d->bordercolor = draw_color_new(globalconf.display, phys_screen, color);
 343     else
 344         d->bordercolor = tmp_color;
 345
 346     return w;
 347 }
 348 // vim: filetype=c:expandtab:shiftwidth=4:tabstop=8:softtabstop=4:encoding=utf-8:textwidth=80