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
  35 PangoFontDescription *bat1_font_desc;
  36 PangoFontDescription *bat2_font_desc;
  37 struct batstate battery_state;
  38
  39 static char buf_bat_percentage[10];
  40 static char buf_bat_time[20];
  41
  42 int8_t battery_low_status;
  43 char *battery_low_cmd;
  44 char *path_energy_now, *path_energy_full, *path_current_now, *path_status;
  45
  46
  47 void init_battery()
  48 {
  49         // check battery
  50         GDir *directory = 0;
  51         GError *error = NULL;
  52         const char *entryname;
  53         char *battery_dir = 0;
  54
  55         path_energy_now = path_energy_full = path_current_now = path_status = 0;
  56         directory = g_dir_open("/sys/class/power_supply", 0, &error);
  57         if (error)
  58                 g_error_free(error);
  59         else {
  60                 while ((entryname=g_dir_read_name(directory))) {
  61                         if (strncmp(entryname,"AC", 2) == 0) continue;
  62
  63                         char *path1 = g_build_filename("/sys/class/power_supply", entryname, "present", NULL);
  64                         if (g_file_test (path1, G_FILE_TEST_EXISTS)) {
  65                                 g_free(path1);
  66                                 battery_dir = g_build_filename("/sys/class/power_supply", entryname, NULL);
  67                                 break;
  68                         }
  69                         g_free(path1);
  70                 }
  71         }
  72         if (directory)
  73                 g_dir_close(directory);
  74         if (battery_dir) {
  75                 char *path1 = g_build_filename(battery_dir, "energy_now", NULL);
  76                 if (g_file_test (path1, G_FILE_TEST_EXISTS)) {
  77                         path_energy_now = g_build_filename(battery_dir, "energy_now", NULL);
  78                         path_energy_full = g_build_filename(battery_dir, "energy_full", NULL);
  79                 }
  80                 else {
  81                         char *path2 = g_build_filename(battery_dir, "charge_now", NULL);
  82                         if (g_file_test (path2, G_FILE_TEST_EXISTS)) {
  83                                 path_energy_now = g_build_filename(battery_dir, "charge_now", NULL);
  84                                 path_energy_full = g_build_filename(battery_dir, "charge_full", NULL);
  85                         }
  86                         else {
  87                                 g_free(battery_dir);
  88                                 battery_dir = 0;
  89                                 fprintf(stderr, "ERROR: can't found energy_* or charge_*\n");
  90                         }
  91                         g_free(path2);
  92                 }
  93                 path_current_now = g_build_filename(battery_dir, "current_now", NULL);
  94                 path_status = g_build_filename(battery_dir, "status", NULL);
  95                 g_free(path1);
  96         }
  97
  98         if (battery_dir)
  99                 g_free(battery_dir);
 100 }
 101
 102
 103 void init_battery_panel(void *p)
 104 {
 105         Panel *panel = (Panel*)p;
 106         Battery *battery = &panel->battery;
 107         FILE *fp;
 108         int bat_percentage_height, bat_percentage_height_ink, bat_time_height, bat_time_height_ink;
 109         char *battery_dir = 0;
 110
 111         if (battery_dir == 0) battery->area.on_screen = 0;
 112         if (!battery->area.on_screen) return;
 113
 114         battery->area.parent = p;
 115         battery->area.panel = p;
 116         battery->area._draw_foreground = draw_battery;
 117         battery->area._resize = resize_battery;
 118         battery->area.resize = 1;
 119         battery->area.redraw = 1;
 120
 121         if((fp = fopen(path_energy_now, "r")) == NULL) {
 122                 fprintf(stderr, "ERROR: battery applet can't open energy_now\n");
 123                 panel->battery.area.on_screen = 0;
 124                 return;
 125         }
 126         fclose(fp);
 127         if((fp = fopen(path_energy_full, "r")) == NULL) {
 128                 fprintf(stderr, "ERROR: battery applet can't open energy_full\n");
 129                 panel->battery.area.on_screen = 0;
 130                 return;
 131         }
 132         fclose(fp);
 133         if((fp = fopen(path_current_now, "r")) == NULL) {
 134                 fprintf(stderr, "ERROR: battery applet can't open current_now\n");
 135                 panel->battery.area.on_screen = 0;
 136                 return;
 137         }
 138         fclose(fp);
 139         if((fp = fopen(path_status, "r")) == NULL) {
 140                 fprintf(stderr, "ERROR: battery applet can't open status");
 141                 panel->battery.area.on_screen = 0;
 142                 return;
 143         }
 144         fclose(fp);
 145
 146         update_battery(&battery_state);
 147         snprintf(buf_bat_percentage, sizeof(buf_bat_percentage), "%d%%", battery_state.percentage);
 148         if(battery_state.state == BATTERY_FULL) {
 149                 strcpy(buf_bat_time, "Full");
 150         } else {
 151                 snprintf(buf_bat_time, sizeof(buf_bat_time), "%02d:%02d", battery_state.time.hours, battery_state.time.minutes);
 152         }
 153
 154         get_text_size(bat1_font_desc, &bat_percentage_height_ink, &bat_percentage_height, panel->area.height, buf_bat_percentage, strlen(buf_bat_percentage));
 155         get_text_size(bat2_font_desc, &bat_time_height_ink, &bat_time_height, panel->area.height, buf_bat_time, strlen(buf_bat_time));
 156
 157         if (panel_horizontal) {
 158                 // panel horizonal => fixed height and posy
 159                 battery->area.posy = panel->area.pix.border.width + panel->area.paddingy;
 160                 battery->area.height = panel->area.height - (2 * battery->area.posy);
 161         }
 162         else {
 163                 // panel vertical => fixed width, height, posy and posx
 164                 battery->area.posy = panel->clock.area.posy + panel->clock.area.height + panel->area.paddingx;
 165                 battery->area.height = (2 * battery->area.paddingxlr) + (bat_time_height + bat_percentage_height);
 166                 battery->area.posx = panel->area.pix.border.width + panel->area.paddingy;
 167                 battery->area.width = panel->area.width - (2 * panel->area.pix.border.width) - (2 * panel->area.paddingy);
 168         }
 169
 170         battery->bat1_posy = (battery->area.height - bat_percentage_height) / 2;
 171         battery->bat1_posy -= ((bat_time_height_ink + 2) / 2);
 172         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;
 173 }
 174
 175
 176 void update_battery() {
 177         FILE *fp;
 178         char tmp[25];
 179         int64_t energy_now = 0, energy_full = 0, current_now = 0;
 180         int seconds = 0;
 181         int8_t new_percentage = 0;
 182
 183         fp = fopen(path_energy_now, "r");
 184         if(fp != NULL) {
 185                 fgets(tmp, sizeof tmp, fp);
 186                 energy_now = atoi(tmp);
 187                 fclose(fp);
 188         }
 189
 190         fp = fopen(path_energy_full, "r");
 191         if(fp != NULL) {
 192                 fgets(tmp, sizeof tmp, fp);
 193                 energy_full = atoi(tmp);
 194                 fclose(fp);
 195         }
 196
 197         fp = fopen(path_current_now, "r");
 198         if(fp != NULL) {
 199                 fgets(tmp, sizeof tmp, fp);
 200                 current_now = atoi(tmp);
 201                 fclose(fp);
 202         }
 203
 204         fp = fopen(path_status, "r");
 205         if(fp != NULL) {
 206                 fgets(tmp, sizeof tmp, fp);
 207                 fclose(fp);
 208         }
 209
 210         battery_state.state = BATTERY_UNKNOWN;
 211         if(strcasecmp(tmp, "Charging\n")==0) battery_state.state = BATTERY_CHARGING;
 212         if(strcasecmp(tmp, "Discharging\n")==0) battery_state.state = BATTERY_DISCHARGING;
 213         if(strcasecmp(tmp, "Full\n")==0) battery_state.state = BATTERY_FULL;
 214
 215         if(current_now > 0) {
 216                 switch(battery_state.state) {
 217                         case BATTERY_CHARGING:
 218                                 seconds = 3600 * (energy_full - energy_now) / current_now;
 219                                 break;
 220                         case BATTERY_DISCHARGING:
 221                                 seconds = 3600 * energy_now / current_now;
 222                                 break;
 223                         default:
 224                                 seconds = 0;
 225                                 break;
 226                 }
 227         } else seconds = 0;
 228
 229         battery_state.time.hours = seconds / 3600;
 230         seconds -= 3600 * battery_state.time.hours;
 231         battery_state.time.minutes = seconds / 60;
 232         seconds -= 60 * battery_state.time.minutes;
 233         battery_state.time.seconds = seconds;
 234
 235         if(energy_full > 0)
 236                 new_percentage = (energy_now*100)/energy_full;
 237
 238         if(battery_low_status != 0 && battery_low_status == new_percentage && battery_state.percentage > new_percentage) {
 239                 //printf("battery low, executing: %s\n", battery_low_cmd);
 240                 if (battery_low_cmd) system(battery_low_cmd);
 241         }
 242
 243         battery_state.percentage = new_percentage;
 244
 245         // clamp percentage to 100 in case battery is misreporting that its current charge is more than its max
 246         if(battery_state.percentage > 100) {
 247                 battery_state.percentage = 100;
 248         }
 249 }
 250
 251
 252 void draw_battery (void *obj, cairo_t *c, int active)
 253 {
 254         Battery *battery = obj;
 255         PangoLayout *layout;
 256
 257         layout = pango_cairo_create_layout (c);
 258
 259         // draw layout
 260         pango_layout_set_font_description(layout, bat1_font_desc);
 261         pango_layout_set_width(layout, battery->area.width * PANGO_SCALE);
 262         pango_layout_set_alignment(layout, PANGO_ALIGN_CENTER);
 263         pango_layout_set_text(layout, buf_bat_percentage, strlen(buf_bat_percentage));
 264
 265         cairo_set_source_rgba(c, battery->font.color[0], battery->font.color[1], battery->font.color[2], battery->font.alpha);
 266
 267         pango_cairo_update_layout(c, layout);
 268         cairo_move_to(c, 0, battery->bat1_posy);
 269         pango_cairo_show_layout(c, layout);
 270
 271         pango_layout_set_font_description(layout, bat2_font_desc);
 272         pango_layout_set_indent(layout, 0);
 273         pango_layout_set_text(layout, buf_bat_time, strlen(buf_bat_time));
 274         pango_layout_set_width(layout, battery->area.width * PANGO_SCALE);
 275
 276         pango_cairo_update_layout(c, layout);
 277         cairo_move_to(c, 0, battery->bat2_posy);
 278         pango_cairo_show_layout(c, layout);
 279
 280         g_object_unref(layout);
 281 }
 282
 283
 284 void resize_battery(void *obj)
 285 {
 286         Battery *battery = obj;
 287         PangoLayout *layout;
 288         int percentage_width, time_width, new_width;
 289
 290         percentage_width = time_width = 0;
 291         battery->area.redraw = 1;
 292
 293         snprintf(buf_bat_percentage, sizeof(buf_bat_percentage), "%d%%", battery_state.percentage);
 294         if(battery_state.state == BATTERY_FULL) {
 295                 strcpy(buf_bat_time, "Full");
 296         } else {
 297                 snprintf(buf_bat_time, sizeof(buf_bat_time), "%02d:%02d", battery_state.time.hours, battery_state.time.minutes);
 298         }
 299         // vertical panel doen't adjust width
 300         if (!panel_horizontal) return;
 301
 302         cairo_surface_t *cs;
 303         cairo_t *c;
 304         Pixmap pmap;
 305         pmap = XCreatePixmap(server.dsp, server.root_win, battery->area.width, battery->area.height, server.depth);
 306
 307         cs = cairo_xlib_surface_create(server.dsp, pmap, server.visual, battery->area.width, battery->area.height);
 308         c = cairo_create(cs);
 309         layout = pango_cairo_create_layout(c);
 310
 311         // check width
 312         pango_layout_set_font_description(layout, bat1_font_desc);
 313         pango_layout_set_indent(layout, 0);
 314         pango_layout_set_text(layout, buf_bat_percentage, strlen(buf_bat_percentage));
 315         pango_layout_get_pixel_size(layout, &percentage_width, NULL);
 316
 317         pango_layout_set_font_description(layout, bat2_font_desc);
 318         pango_layout_set_indent(layout, 0);
 319         pango_layout_set_text(layout, buf_bat_time, strlen(buf_bat_time));
 320         pango_layout_get_pixel_size(layout, &time_width, NULL);
 321
 322         if(percentage_width > time_width) new_width = percentage_width;
 323         else new_width = time_width;
 324
 325         new_width += (2*battery->area.paddingxlr) + (2*battery->area.pix.border.width);
 326
 327         int old_width = battery->area.width;
 328
 329         Panel *panel = ((Area*)obj)->panel;
 330         battery->area.width = new_width + 1;
 331         battery->area.posx = panel->area.width - battery->area.width - panel->area.paddingxlr - panel->area.pix.border.width;
 332         if (panel->clock.area.on_screen)
 333                 battery->area.posx -= (panel->clock.area.width + panel->area.paddingx);
 334
 335         if(new_width > old_width || new_width < (old_width-6)) {
 336                 // refresh and resize other objects on panel
 337                 // we try to limit the number of refresh
 338                 // printf("battery_width %d, new_width %d\n", battery->area.width, new_width);
 339                 panel->area.resize = 1;
 340                 systray.area.resize = 1;
 341                 panel_refresh = 1;
 342         }
 343
 344         g_object_unref (layout);
 345         cairo_destroy (c);
 346         cairo_surface_destroy (cs);
 347         XFreePixmap (server.dsp, pmap);
 348 }
 349