Dogcows Code - chaz/tint2/blob - src/battery/battery.c

   1 /**************************************************************************
   2 *
   3 * Tint2 : battery
   4 *
   5 * Copyright (C) 2009 Sebastian Reichel <elektranox@gmail.com>
   6 *
   7 * This program is free software; you can redistribute it and/or
   8 * modify it under the terms of the GNU General Public License version 2
   9 * or any later version as published by the Free Software Foundation.
  10 *
  11 * This program is distributed in the hope that it will be useful,
  12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 * GNU General Public License for more details.
  15 * You should have received a copy of the GNU General Public License
  16 * along with this program; if not, write to the Free Software
  17 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  18 **************************************************************************/
  19
  20 #include <string.h>
  21 #include <stdio.h>
  22 #include <stdlib.h>
  23 #include <cairo.h>
  24 #include <cairo-xlib.h>
  25 #include <pango/pangocairo.h>
  26
  27 #include "window.h"
  28 #include "server.h"
  29 #include "area.h"
  30 #include "panel.h"
  31 #include "taskbar.h"
  32 #include "battery.h"
  33 #include "clock.h"
  34 #include "timer.h"
  35
  36 PangoFontDescription *bat1_font_desc;
  37 PangoFontDescription *bat2_font_desc;
  38 struct batstate battery_state;
  39 int battery_enabled;
  40 int percentage_hide;
  41 static timeout* battery_timeout;
  42
  43 static char buf_bat_percentage[10];
  44 static char buf_bat_time[20];
  45
  46 int8_t battery_low_status;
  47 unsigned char battery_low_cmd_send;
  48 char *battery_low_cmd;
  49 char *path_energy_now;
  50 char *path_energy_full;
  51 char *path_current_now;
  52 char *path_status;
  53
  54 void update_batterys(void* arg)
  55 {
  56         int i;
  57         update_battery();
  58         for (i=0 ; i < nb_panel ; i++) {
  59                 if (battery_state.percentage >= percentage_hide) {
  60                         if (panel1[i].battery.area.on_screen == 1) {
  61                                 panel1[i].battery.area.on_screen = 0;
  62                                 // force resize on panel
  63                                 panel1[i].area.resize = 1;
  64                                 panel_refresh = 1;
  65                         }
  66                         continue;
  67                 }
  68                 else {
  69                         if (panel1[i].battery.area.on_screen == 0) {
  70                                 panel1[i].battery.area.on_screen = 1;
  71                                 // force resize on panel
  72                                 panel1[i].area.resize = 1;
  73                                 panel_refresh = 1;
  74                         }
  75                 }
  76                 panel1[i].battery.area.resize = 1;
  77         }
  78 }
  79
  80 void default_battery()
  81 {
  82         battery_enabled = 0;
  83         percentage_hide = 101;
  84         battery_low_cmd_send = 0;
  85         battery_timeout = 0;
  86         bat1_font_desc = 0;
  87         bat2_font_desc = 0;
  88         battery_low_cmd = 0;
  89         path_energy_now = 0;
  90         path_energy_full = 0;
  91         path_current_now = 0;
  92         path_status = 0;
  93 }
  94
  95 void cleanup_battery()
  96 {
  97         if (bat1_font_desc) pango_font_description_free(bat1_font_desc);
  98         if (bat2_font_desc) pango_font_description_free(bat2_font_desc);
  99         if (path_energy_now) g_free(path_energy_now);
 100         if (path_energy_full) g_free(path_energy_full);
 101         if (path_current_now) g_free(path_current_now);
 102         if (path_status) g_free(path_status);
 103         if (battery_low_cmd) g_free(battery_low_cmd);
 104 }
 105
 106
 107 void init_battery()
 108 {
 109         // check battery
 110         GDir *directory = 0;
 111         GError *error = NULL;
 112         const char *entryname;
 113         char *battery_dir = 0;
 114
 115         if (!battery_enabled) return;
 116
 117         directory = g_dir_open("/sys/class/power_supply", 0, &error);
 118         if (error)
 119                 g_error_free(error);
 120         else {
 121                 while ((entryname=g_dir_read_name(directory))) {
 122                         if (strncmp(entryname,"AC", 2) == 0) continue;
 123
 124                         char *path1 = g_build_filename("/sys/class/power_supply", entryname, "present", NULL);
 125                         if (g_file_test (path1, G_FILE_TEST_EXISTS)) {
 126                                 g_free(path1);
 127                                 battery_dir = g_build_filename("/sys/class/power_supply", entryname, NULL);
 128                                 break;
 129                         }
 130                         g_free(path1);
 131                 }
 132         }
 133         if (directory)
 134                 g_dir_close(directory);
 135         if (!battery_dir) {
 136                 fprintf(stderr, "ERROR: battery applet can't found power_supply\n");
 137                 default_battery();
 138                 return;
 139         }
 140
 141         char *path1 = g_build_filename(battery_dir, "energy_now", NULL);
 142         if (g_file_test (path1, G_FILE_TEST_EXISTS)) {
 143                 path_energy_now = g_build_filename(battery_dir, "energy_now", NULL);
 144                 path_energy_full = g_build_filename(battery_dir, "energy_full", NULL);
 145         }
 146         else {
 147                 char *path2 = g_build_filename(battery_dir, "charge_now", NULL);
 148                 if (g_file_test (path2, G_FILE_TEST_EXISTS)) {
 149                         path_energy_now = g_build_filename(battery_dir, "charge_now", NULL);
 150                         path_energy_full = g_build_filename(battery_dir, "charge_full", NULL);
 151                 }
 152                 else {
 153                         fprintf(stderr, "ERROR: can't found energy_* or charge_*\n");
 154                 }
 155                 g_free(path2);
 156         }
 157         if (path_energy_now && path_energy_full) {
 158                 path_current_now = g_build_filename(battery_dir, "current_now", NULL);
 159                 path_status = g_build_filename(battery_dir, "status", NULL);
 160
 161                 // check file
 162                 FILE *fp1, *fp2, *fp3, *fp4;
 163                 fp1 = fopen(path_energy_now, "r");
 164                 fp2 = fopen(path_energy_full, "r");
 165                 fp3 = fopen(path_current_now, "r");
 166                 fp4 = fopen(path_status, "r");
 167                 if (fp1 == NULL || fp2 == NULL || fp3 == NULL || fp4 == NULL) {
 168                         cleanup_battery();
 169                         default_battery();
 170                         fprintf(stderr, "ERROR: battery applet can't open energy_now\n");
 171                 }
 172                 fclose(fp1);
 173                 fclose(fp2);
 174                 fclose(fp3);
 175                 fclose(fp4);
 176         }
 177
 178         g_free(path1);
 179         g_free(battery_dir);
 180
 181         if (battery_enabled && battery_timeout==0)
 182                 battery_timeout = add_timeout(10, 10000, update_batterys, 0);
 183 }
 184
 185
 186 void init_battery_panel(void *p)
 187 {
 188         Panel *panel = (Panel*)p;
 189         Battery *battery = &panel->battery;
 190         int bat_percentage_height, bat_percentage_height_ink, bat_time_height, bat_time_height_ink;
 191
 192         if (!battery_enabled)
 193                 return;
 194
 195         battery->area.parent = p;
 196         battery->area.panel = p;
 197         battery->area._draw_foreground = draw_battery;
 198         battery->area._resize = resize_battery;
 199         battery->area.resize = 1;
 200         battery->area.redraw = 1;
 201         battery->area.on_screen = 1;
 202
 203         update_battery(&battery_state);
 204         snprintf(buf_bat_percentage, sizeof(buf_bat_percentage), "%d%%", battery_state.percentage);
 205         snprintf(buf_bat_time, sizeof(buf_bat_time), "%02d:%02d", battery_state.time.hours, battery_state.time.minutes);
 206
 207         get_text_size(bat1_font_desc, &bat_percentage_height_ink, &bat_percentage_height, panel->area.height, buf_bat_percentage, strlen(buf_bat_percentage));
 208         get_text_size(bat2_font_desc, &bat_time_height_ink, &bat_time_height, panel->area.height, buf_bat_time, strlen(buf_bat_time));
 209
 210         if (panel_horizontal) {
 211                 // panel horizonal => fixed height and posy
 212                 battery->area.posy = panel->area.bg->border.width + panel->area.paddingy;
 213                 battery->area.height = panel->area.height - (2 * battery->area.posy);
 214         }
 215         else {
 216                 // panel vertical => fixed width, height, posy and posx
 217                 battery->area.posy = panel->clock.area.posy + panel->clock.area.height + panel->area.paddingx;
 218                 battery->area.height = (2 * battery->area.paddingxlr) + (bat_time_height + bat_percentage_height);
 219                 battery->area.posx = panel->area.bg->border.width + panel->area.paddingy;
 220                 battery->area.width = panel->area.width - (2 * panel->area.bg->border.width) - (2 * panel->area.paddingy);
 221         }
 222
 223         battery->bat1_posy = (battery->area.height - bat_percentage_height) / 2;
 224         battery->bat1_posy -= ((bat_time_height_ink + 2) / 2);
 225         battery->bat2_posy = battery->bat1_posy + bat_percentage_height + 2 - (bat_percentage_height - bat_percentage_height_ink)/2 - (bat_time_height - bat_time_height_ink)/2;
 226 }
 227
 228
 229 void update_battery() {
 230         FILE *fp;
 231         char tmp[25];
 232         int64_t energy_now = 0, energy_full = 0, current_now = 0;
 233         int seconds = 0;
 234         int8_t new_percentage = 0;
 235
 236         fp = fopen(path_status, "r");
 237         if(fp != NULL) {
 238                 if (fgets(tmp, sizeof tmp, fp)) {
 239                         battery_state.state = BATTERY_UNKNOWN;
 240                         if(strcasecmp(tmp, "Charging\n")==0) battery_state.state = BATTERY_CHARGING;
 241                         if(strcasecmp(tmp, "Discharging\n")==0) battery_state.state = BATTERY_DISCHARGING;
 242                         if(strcasecmp(tmp, "Full\n")==0) battery_state.state = BATTERY_FULL;
 243                 }
 244                 fclose(fp);
 245         }
 246
 247         fp = fopen(path_energy_now, "r");
 248         if(fp != NULL) {
 249                 if (fgets(tmp, sizeof tmp, fp)) energy_now = atoi(tmp);
 250                 fclose(fp);
 251         }
 252
 253         fp = fopen(path_energy_full, "r");
 254         if(fp != NULL) {
 255                 if (fgets(tmp, sizeof tmp, fp)) energy_full = atoi(tmp);
 256                 fclose(fp);
 257         }
 258
 259         fp = fopen(path_current_now, "r");
 260         if(fp != NULL) {
 261                 if (fgets(tmp, sizeof tmp, fp)) current_now = atoi(tmp);
 262                 fclose(fp);
 263         }
 264
 265         if(current_now > 0) {
 266                 switch(battery_state.state) {
 267                         case BATTERY_CHARGING:
 268                                 seconds = 3600 * (energy_full - energy_now) / current_now;
 269                                 break;
 270                         case BATTERY_DISCHARGING:
 271                                 seconds = 3600 * energy_now / current_now;
 272                                 break;
 273                         default:
 274                                 seconds = 0;
 275                                 break;
 276                 }
 277         } else seconds = 0;
 278
 279         battery_state.time.hours = seconds / 3600;
 280         seconds -= 3600 * battery_state.time.hours;
 281         battery_state.time.minutes = seconds / 60;
 282         seconds -= 60 * battery_state.time.minutes;
 283         battery_state.time.seconds = seconds;
 284
 285         if(energy_full > 0)
 286                 new_percentage = (energy_now*100)/energy_full;
 287
 288         if(battery_low_status > new_percentage && battery_state.state == BATTERY_DISCHARGING && !battery_low_cmd_send) {
 289                         system(battery_low_cmd);  // return value == -1, since we've set SIGCHLD to SIGIGN
 290                         battery_low_cmd_send = 1;
 291         }
 292         if(battery_low_status < new_percentage && battery_state.state == BATTERY_CHARGING && battery_low_cmd_send) {
 293                 battery_low_cmd_send = 0;
 294         }
 295
 296         battery_state.percentage = new_percentage;
 297
 298         // clamp percentage to 100 in case battery is misreporting that its current charge is more than its max
 299         if(battery_state.percentage > 100) {
 300                 battery_state.percentage = 100;
 301         }
 302 }
 303
 304
 305 void draw_battery (void *obj, cairo_t *c)
 306 {
 307         Battery *battery = obj;
 308         PangoLayout *layout;
 309
 310         layout = pango_cairo_create_layout (c);
 311
 312         // draw layout
 313         pango_layout_set_font_description(layout, bat1_font_desc);
 314         pango_layout_set_width(layout, battery->area.width * PANGO_SCALE);
 315         pango_layout_set_alignment(layout, PANGO_ALIGN_CENTER);
 316         pango_layout_set_text(layout, buf_bat_percentage, strlen(buf_bat_percentage));
 317
 318         cairo_set_source_rgba(c, battery->font.color[0], battery->font.color[1], battery->font.color[2], battery->font.alpha);
 319
 320         pango_cairo_update_layout(c, layout);
 321         cairo_move_to(c, 0, battery->bat1_posy);
 322         pango_cairo_show_layout(c, layout);
 323
 324         pango_layout_set_font_description(layout, bat2_font_desc);
 325         pango_layout_set_indent(layout, 0);
 326         pango_layout_set_text(layout, buf_bat_time, strlen(buf_bat_time));
 327         pango_layout_set_width(layout, battery->area.width * PANGO_SCALE);
 328
 329         pango_cairo_update_layout(c, layout);
 330         cairo_move_to(c, 0, battery->bat2_posy);
 331         pango_cairo_show_layout(c, layout);
 332
 333         g_object_unref(layout);
 334 }
 335
 336
 337 void resize_battery(void *obj)
 338 {
 339         Battery *battery = obj;
 340         PangoLayout *layout;
 341         int percentage_width, time_width, new_width;
 342
 343         percentage_width = time_width = 0;
 344         battery->area.redraw = 1;
 345
 346         snprintf(buf_bat_percentage, sizeof(buf_bat_percentage), "%d%%", battery_state.percentage);
 347         if(battery_state.state == BATTERY_FULL) {
 348                 strcpy(buf_bat_time, "Full");
 349         } else {
 350                 snprintf(buf_bat_time, sizeof(buf_bat_time), "%02d:%02d", battery_state.time.hours, battery_state.time.minutes);
 351         }
 352         // vertical panel doen't adjust width
 353         if (!panel_horizontal) return;
 354
 355         cairo_surface_t *cs;
 356         cairo_t *c;
 357         Pixmap pmap;
 358         pmap = XCreatePixmap(server.dsp, server.root_win, battery->area.width, battery->area.height, server.depth);
 359
 360         cs = cairo_xlib_surface_create(server.dsp, pmap, server.visual, battery->area.width, battery->area.height);
 361         c = cairo_create(cs);
 362         layout = pango_cairo_create_layout(c);
 363
 364         // check width
 365         pango_layout_set_font_description(layout, bat1_font_desc);
 366         pango_layout_set_indent(layout, 0);
 367         pango_layout_set_text(layout, buf_bat_percentage, strlen(buf_bat_percentage));
 368         pango_layout_get_pixel_size(layout, &percentage_width, NULL);
 369
 370         pango_layout_set_font_description(layout, bat2_font_desc);
 371         pango_layout_set_indent(layout, 0);
 372         pango_layout_set_text(layout, buf_bat_time, strlen(buf_bat_time));
 373         pango_layout_get_pixel_size(layout, &time_width, NULL);
 374
 375         if(percentage_width > time_width) new_width = percentage_width;
 376         else new_width = time_width;
 377
 378         new_width += (2*battery->area.paddingxlr) + (2*battery->area.bg->border.width);
 379
 380         int old_width = battery->area.width;
 381
 382         Panel *panel = ((Area*)obj)->panel;
 383         battery->area.width = new_width + 1;
 384         battery->area.posx = panel->area.width - battery->area.width - panel->area.paddingxlr - panel->area.bg->border.width;
 385         if (panel->clock.area.on_screen)
 386                 battery->area.posx -= (panel->clock.area.width + panel->area.paddingx);
 387
 388         if(new_width > old_width || new_width < (old_width-6)) {
 389                 // refresh and resize other objects on panel
 390                 // we try to limit the number of refresh
 391                 // printf("battery_width %d, new_width %d\n", battery->area.width, new_width);
 392                 panel->area.resize = 1;
 393                 systray.area.resize = 1;
 394                 panel_refresh = 1;
 395         }
 396
 397         g_object_unref (layout);
 398         cairo_destroy (c);
 399         cairo_surface_destroy (cs);
 400         XFreePixmap (server.dsp, pmap);
 401 }
 402