cprogs/xbatmon-simple.c

   1 /*
   2  * display outputs, per line:
   3  *
   4  *   Remaining:  | Empty:       | Degraded:
   5  *     blue      |  black       |  dimgrey      discharging
   6  *     green     |  black       |  dimgrey      charging
   7  *     cyan      |  black       |  dimgrey      charged
   8  *     grey      |  black       |  dimgrey      charging&discharching!
   9  *     lightgrey |  black       |  dimgrey      none of the above
  10  *     blue      |  red         |  dimgrey      discharging - low!
  11  *     green     |  red         |  dimgrey      charging - low
  12  *     cyan      |  red         |  dimgrey      charged - low [1]
  13  *     grey      |  red         |  dimgrey      charging&discharching, low [1]
  14  *       ...  darkgreen  ...                    no batteries present
  15  *       ...  yellow  ...                       error
  16  *
  17  * [1] battery must be quite badly degraded
  18  */
  19 /*
  20  * Copyright (C) 2004 Ian Jackson <ian@davenant.greenend.org.uk>
  21  *
  22  * This is free software; you can redistribute it and/or modify
  23  * it under the terms of the GNU General Public License as
  24  * published by the Free Software Foundation; either version 3,
  25  * or (at your option) any later version.
  26  *
  27  * This is distributed in the hope that it will be useful, but
  28  * WITHOUT ANY WARRANTY; without even the implied warranty of
  29  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  30  * GNU General Public License for more details.
  31  *
  32  * You should have received a copy of the GNU General Public
  33  * License along with this file; if not, consult the Free Software
  34  * Foundation's website at www.fsf.org, or the GNU Project website at
  35  * www.gnu.org.
  36  */
  37
  38 #include <stdio.h>
  39 #include <assert.h>
  40 #include <math.h>
  41 #include <stdlib.h>
  42 #include <string.h>
  43 #include <errno.h>
  44 #include <unistd.h>
  45 #include <ctype.h>
  46 #include <stdint.h>
  47 #include <limits.h>
  48 #include <inttypes.h>
  49
  50 #include <sys/poll.h>
  51 #include <sys/types.h>
  52 #include <dirent.h>
  53
  54 #include <X11/Xlib.h>
  55 #include <X11/Xutil.h>
  56 #include <X11/Xresource.h>
  57
  58 #define TOP      60
  59 #define BOTTOM 3600
  60
  61 #define TIMEOUT         5000 /* milliseconds */
  62 #define TIMEOUT_ONERROR 3333 /* milliseconds */
  63
  64 static const char program_name[]= "xbatmon-simple";
  65 static int debug=-1, alarmlevel;
  66
  67 /*---------- general utility stuff and declarations ----------*/
  68
  69 static void fail(const char *m) {
  70   fprintf(stderr,"error: %s\n", m);
  71   exit(-1);
  72 }
  73 static void badusage(void) { fail("bad usage"); }
  74
  75 typedef uint64_t value;
  76 #define VAL_NOTFOUND (~(value)0)
  77
  78 typedef struct fileinfo fileinfo;
  79 typedef int parser(const fileinfo*);
  80
  81 static parser parse_uevent;
  82
  83 struct fileinfo {
  84   const char *filename;
  85   parser *parse;
  86   const void *extra;
  87 };
  88
  89 /*---------- structure of and results from /sys/class/power/... ----------*/
  90 /* variables this_... are the results from readbattery();
  91  * if readbattery() succeeds the appropriate ones are all valid
  92  * and not VAL_NOTFOUND
  93  */
  94
  95 typedef struct batinfo_field {
  96   const char *label;
  97   value *valuep;
  98   const char *enumarray[10];
  99 } batinfo_field;
 100
 101 #define BAT_QTYS(_, _ec, EC_, PC_)                              \
 102   _(design_capacity##_ec,    BATTERY,  EC_##FULL_DESIGN )       \
 103   _(last_full_capacity##_ec, BATTERY,  EC_##FULL        )       \
 104   _(remaining_capacity##_ec, BATTERY,  EC_##NOW         )       \
 105   _(present_rate##_ec,       BATTERY,  PC_##NOW         )
 106  /* ENERGY [mWh]; POWER [mW]; CHARGE [uAh]; CURRENT [uA] */
 107
 108 #define UEVENT_ESSENTIAL_QUANTITY_FIELDS(_)                     \
 109   _(present,                 BATTERY,  PRESENT /* bool */ )     \
 110   _(online,                  MAINS,    ONLINE  /* bool */ )
 111
 112 #define UEVENT_FUNKY_QUANTITY_FIELDS(_)         \
 113   BAT_QTYS(_,_energy,ENERGY_,POWER_)            \
 114   BAT_QTYS(_,_charge,CHARGE_,CURRENT_)
 115
 116 #define UEVENT_OPTIONAL_QUANTITY_FIELDS(_)                      \
 117   _(voltage,                 BATTERY,  VOLTAGE_NOW /* uV */ )
 118
 119 #define UEVENT_ENUM_FIELDS(_)                                           \
 120   _(state,   BATTERY,  STATUS,  "Discharging","Charging","Full","Unknown" ) \
 121   _(type,    BOTH,     TYPE,    "Mains",       "Battery"              )
 122
 123 #define CHGST_DISCHARGING 0 /* Reflects order in _(state,...) above     */
 124 #define CHGST_CHARGING    1 /* Also, much code assumes exactly          */
 125 #define CHGST_CHARGED     2 /* these three possible states.             */
 126 #define CHGST_UNKNOWN     3 /* these three possible states.             */
 127 #define CHGST_ERROR       8 /* Except that this one is an extra bit.    */
 128
 129 #define TYPE_MAINS        0 /* Reflects order in _(type,...) above        */
 130 #define TYPE_BATTERY      1 /* Also, much code assumes exactly these two  */
 131 #define TYPE_BOTH       100 /* Except this is a magic invalid value.      */
 132
 133 #define SEPARATE_QUANTITY_FIELDS(_)             \
 134   /* See commit ec6f5f0be800bc5f2a27046833dba04e0c67ffac for
 135      the code needed to use this */
 136
 137
 138 #define ALL_DIRECT_VARS(_)                      \
 139   UEVENT_ESSENTIAL_QUANTITY_FIELDS(_)           \
 140   UEVENT_FUNKY_QUANTITY_FIELDS(_)               \
 141   UEVENT_OPTIONAL_QUANTITY_FIELDS(_)            \
 142   UEVENT_ENUM_FIELDS(_)                         \
 143   SEPARATE_QUANTITY_FIELDS(_)
 144
 145 #define ALL_VARS(_)                             \
 146   ALL_DIRECT_VARS(_)                            \
 147   BAT_QTYS(_,,,)
 148
 149 #define ALL_NEEDED_FIELDS(_)                    \
 150   UEVENT_ESSENTIAL_QUANTITY_FIELDS(_)           \
 151   UEVENT_ENUM_FIELDS(_)                         \
 152   SEPARATE_QUANTITY_FIELDS(_)
 153
 154 #define ALL_PLAIN_ACCUMULATE_FIELDS(_)          \
 155   UEVENT_ESSENTIAL_QUANTITY_FIELDS(_)           \
 156   SEPARATE_QUANTITY_FIELDS(_)
 157
 158 #define ALL_ACCUMULATE_FIELDS(_)                \
 159   ALL_PLAIN_ACCUMULATE_FIELDS(_)                \
 160   BAT_QTYS(_,,,)
 161
 162
 163 #define F_VAR(f,...) \
 164 static value this_##f;
 165 ALL_VARS(F_VAR)
 166
 167 #define Q_FLD(f,t,l)        { "POWER_SUPPLY_" #l, &this_##f },
 168 #define E_FLD(f,t,l,vl...)  { "POWER_SUPPLY_" #l, &this_##f, { vl } },
 169
 170 static const batinfo_field uevent_fields[]= {
 171   UEVENT_ESSENTIAL_QUANTITY_FIELDS(Q_FLD)
 172   UEVENT_FUNKY_QUANTITY_FIELDS(Q_FLD)
 173   UEVENT_OPTIONAL_QUANTITY_FIELDS(Q_FLD)
 174   UEVENT_ENUM_FIELDS(E_FLD)
 175   { 0 }
 176 };
 177
 178 #define S_FLD(f,t,fn,vl...)                                             \
 179 static const batinfo_field bif_##f = { 0, &this_##f, { vl } };
 180   SEPARATE_QUANTITY_FIELDS(S_FLD)
 181
 182 #define S_FILE(f,t,fn,vl...) { fn, parse_separate, &bif_##f },
 183
 184 static const fileinfo files[]= {
 185   { "uevent",  parse_uevent,  uevent_fields },
 186   SEPARATE_QUANTITY_FIELDS(S_FILE)
 187   { 0 }
 188 };
 189
 190 /*---------- parsing of one thingx in /sys/class/power/... ----------*/
 191
 192 /* variables private to the parser and its error handlers */
 193 static char batlinebuf[1000];
 194 static FILE *batfile;
 195 static const char *batdirname;
 196 static const char *batfilename;
 197 static const char *batlinevalue;
 198
 199 static int batfailf(const char *why) {
 200   if (batlinevalue) {
 201     fprintf(stderr,"%s/%s: %s value `%s': %s\n",
 202             batdirname,batfilename, batlinebuf,batlinevalue,why);
 203   } else {
 204     fprintf(stderr,"%s/%s: %s: `%s'\n",
 205             batdirname,batfilename, why, batlinebuf);
 206   }
 207   return -1;
 208 }
 209
 210 static int batfailc(const char *why) {
 211   fprintf(stderr,"%s/%s: %s\n",
 212           batdirname,batfilename, why);
 213   return -1;
 214 }
 215
 216 static int batfaile(const char *syscall, const char *target) {
 217   fprintf(stderr,"%s: failed to %s %s: %s\n",
 218           batdirname ? batdirname : "*", syscall, target, strerror(errno));
 219   return -1;
 220 }
 221
 222 static int chdir_base(void) {
 223   int r;
 224
 225   r= chdir("/sys/class/power_supply");
 226   if (r) return batfaile("chdir","/sys/class/power_supply");
 227
 228   return 0;
 229 }
 230
 231 static void tidybattery(void) {
 232   if (batfile) { fclose(batfile); batfile=0; }
 233 }
 234
 235 static int parse_value(const fileinfo *cfile, const batinfo_field *field) {
 236   if (*field->valuep != VAL_NOTFOUND)
 237     return batfailf("value specified multiple times");
 238
 239   if (!field->enumarray[0]) {
 240
 241     char *ep;
 242     *field->valuep= strtoull(batlinevalue,&ep,10);
 243     if (*ep)
 244       batfailf("value number syntax incorrect");
 245
 246   } else {
 247
 248     const char *const *enumsearch;
 249     for (*field->valuep=0, enumsearch=field->enumarray;
 250          *enumsearch && strcmp(*enumsearch,batlinevalue);
 251          (*field->valuep)++, enumsearch++);
 252     if (!*enumsearch)
 253       batfailf("unknown enum value");
 254
 255   }
 256   return 0;
 257 }
 258
 259 static int parse_uevent(const fileinfo *cfile) {
 260   char *equals= strchr(batlinebuf,'=');
 261   if (!equals)
 262     return batfailf("line without a equals");
 263   *equals= 0;
 264   batlinevalue = equals+1;
 265
 266   const batinfo_field *field;
 267   for (field=cfile->extra; field->label; field++) {
 268     if (!strcmp(field->label,batlinebuf))
 269       goto found;
 270   }
 271   return 0;
 272
 273  found:
 274   return parse_value(cfile, field);
 275 }
 276
 277 static int readbattery(void) { /* 0=>ok, -1=>couldn't */
 278
 279   const fileinfo *cfile;
 280   char *sr;
 281   int r, l;
 282
 283   r= chdir_base();
 284   if (r) return r;
 285
 286   r= chdir(batdirname);
 287   if (r) return batfaile("chdir",batdirname);
 288
 289 #define V_NOTFOUND(f,...) \
 290   this_##f = VAL_NOTFOUND;
 291 ALL_VARS(V_NOTFOUND)
 292
 293   for (cfile=files;
 294        (batfilename= cfile->filename);
 295        cfile++) {
 296     batfile= fopen(batfilename,"r");
 297     if (!batfile) {
 298       if (errno == ENOENT) continue;
 299       return batfaile("open",batfilename);
 300     }
 301
 302     for (;;) {
 303       batlinevalue= 0;
 304
 305       sr= fgets(batlinebuf,sizeof(batlinebuf),batfile);
 306       if (ferror(batfile)) return batfaile("read",batfilename);
 307       if (!sr && feof(batfile)) break;
 308       l= strlen(batlinebuf);
 309       assert(l>0);
 310       if (batlinebuf[l-1] != '\n')
 311         return batfailf("line too long");
 312       batlinebuf[l-1]= 0;
 313
 314       if (cfile->parse(cfile))
 315         return -1;
 316     }
 317
 318     fclose(batfile);
 319     batfile= 0;
 320   }
 321
 322   if (debug) {
 323     printf("%s:\n",batdirname);
 324 #define V_PRINT(f,...)                                  \
 325     printf(" %-30s = %20"PRId64"\n", #f, (int64_t)this_##f);
 326 ALL_DIRECT_VARS(V_PRINT)
 327   }
 328
 329   int needsfields_MAINS   = this_type == TYPE_MAINS;
 330   int needsfields_BATTERY = this_type == TYPE_BATTERY;
 331   int needsfields_BOTH    = 1;
 332
 333   int missing = 0;
 334
 335 #define V_NEEDED(f,t,...)                               \
 336   if (needsfields_##t && this_##f == VAL_NOTFOUND) {    \
 337     fprintf(stderr,"%s: %s: not found\n",               \
 338             batdirname, #f);                            \
 339     missing++;                                          \
 340   }
 341 ALL_NEEDED_FIELDS(V_NEEDED)
 342
 343   if (missing) return -1;
 344
 345   return 0;
 346 }
 347
 348 /*---------- data collection and analysis ----------*/
 349
 350 /* These next three variables are the results of the charging state */
 351 static unsigned charging_mask; /* 1u<<CHGST_* | ... */
 352 static double nondegraded_norm, fill_norm, ratepersec_norm;
 353 static int alarmed;
 354
 355 #define Q_VAR(f,t,...) \
 356 static double total_##f;
 357   ALL_ACCUMULATE_FIELDS(Q_VAR)
 358
 359 static void acquiredata(void) {
 360   DIR *di = 0;
 361   struct dirent *de;
 362   int r;
 363
 364   charging_mask= 0;
 365   alarmed = 0;
 366
 367   if (debug) printf("\n");
 368
 369 #define Q_ZERO(f,t,...) \
 370   total_##f= 0;
 371 ALL_ACCUMULATE_FIELDS(Q_ZERO)
 372
 373   r = chdir_base();
 374   if (r) goto bad;
 375
 376   di= opendir(".");  if (!di) { batfaile("opendir","battery"); goto bad; }
 377   while ((de= readdir(di))) {
 378     if (de->d_name[0]==0 || de->d_name[0]=='.') continue;
 379
 380     batdirname= de->d_name;
 381     r= readbattery();
 382     tidybattery();
 383
 384     if (r) {
 385     bad:
 386       charging_mask |= (1u << CHGST_ERROR);
 387       break;
 388     }
 389
 390     if (this_type == TYPE_BATTERY) {
 391       if (!this_present)
 392         continue;
 393
 394       charging_mask |= 1u << this_state;
 395
 396 #define QTY_SUPPLIED(f,...)   this_##f != VAL_NOTFOUND &&
 397 #define QTY_USE_ENERGY(f,...) this_##f = this_##f##_energy;
 398 #define QTY_USE_CHARGE(f,...) this_##f = this_##f##_charge;
 399
 400       double funky_multiplier;
 401       if (BAT_QTYS(QTY_SUPPLIED,_energy,,) 1) {
 402         if (debug) printf(" using energy\n");
 403         BAT_QTYS(QTY_USE_ENERGY,,,);
 404         funky_multiplier = 1.0;
 405       } else if (BAT_QTYS(QTY_SUPPLIED,_charge,,)
 406                  this_voltage != VAL_NOTFOUND) {
 407         if (debug) printf(" using charge\n");
 408         BAT_QTYS(QTY_USE_CHARGE,,,);
 409         funky_multiplier = this_voltage * 1e-6;
 410       } else {
 411         batfailc("neither complete set of energy nor charge");
 412         continue;
 413       }
 414       if (this_state == CHGST_DISCHARGING)
 415         /* negate it */
 416         total_present_rate -= 2.0 * this_present_rate * funky_multiplier;
 417
 418 #define Q_ACCUMULATE_FUNKY(f,...)                       \
 419       total_##f += this_##f * funky_multiplier;
 420 BAT_QTYS(Q_ACCUMULATE_FUNKY,,,)
 421     }
 422
 423 #define Q_ACCUMULATE_PLAIN(f,t,...)                     \
 424     if (this_type == TYPE_##t)                  \
 425       total_##f += this_##f;
 426 ALL_PLAIN_ACCUMULATE_FIELDS(Q_ACCUMULATE_PLAIN)
 427
 428
 429   }
 430   if (di) closedir(di);
 431
 432   if (debug) {
 433     printf("TOTAL:\n");
 434     printf(" %-30s = %#20x\n", "mask", charging_mask);
 435 #define T_PRINT(f,...)                                  \
 436     printf(" %-30s = %20.6f\n", #f, total_##f);
 437 BAT_QTYS(T_PRINT,,,)
 438 ALL_PLAIN_ACCUMULATE_FIELDS(T_PRINT)
 439   }
 440
 441   if ((charging_mask & (1u<<CHGST_DISCHARGING)) &&
 442       !total_online/*mains*/) {
 443     double time_remaining =
 444       -total_remaining_capacity * 3600.0 / total_present_rate;
 445     if (debug) printf(" %-30s = %20.6f\n", "time remaining", time_remaining);
 446     if (time_remaining < alarmlevel)
 447       alarmed = 1;
 448   }
 449
 450   if (total_design_capacity < 0.5)
 451     total_design_capacity= 1.0;
 452
 453   if (total_last_full_capacity < total_remaining_capacity)
 454     total_last_full_capacity= total_remaining_capacity;
 455   if (total_design_capacity < total_last_full_capacity)
 456     total_design_capacity= total_last_full_capacity;
 457
 458   nondegraded_norm= total_last_full_capacity / total_design_capacity;
 459   fill_norm= total_remaining_capacity / total_design_capacity;
 460   ratepersec_norm=  total_present_rate
 461     / (3600.0 * total_design_capacity);
 462 }
 463
 464 static void initacquire(void) {
 465 }
 466
 467 /*---------- argument parsing ----------*/
 468
 469 #define COLOURS                                 \
 470   C(blue,           discharging)                \
 471   C(green,         charging)                    \
 472   C(cyan,           charged)                    \
 473   C(lightgrey,      notcharging)                \
 474   C(grey,           confusing)                  \
 475   C(black,          normal)                     \
 476   C(red,            low)                        \
 477   C(dimgrey,        degraded)                   \
 478   C(darkgreen,      absent)                     \
 479   C(yellow,         error)                      \
 480   C(white,          equilibrium)                \
 481   GC(remain)                                    \
 482   GC(white)                                     \
 483   GC(empty)
 484
 485 static XrmDatabase xrm;
 486 static Display *disp;
 487 static int screen;
 488 static const char *parentwindow;
 489
 490 static const char defaultresources[]=
 491 #define GC(g)
 492 #define C(c,u)                                  \
 493   "*" #u "Color: " #c "\n"
 494   COLOURS
 495 #undef GC
 496 #undef C
 497   ;
 498
 499 #define S(s) ((char*)(s))
 500 static const XrmOptionDescRec optiontable[]= {
 501   { S("-debug"),        S("*debug"),        XrmoptionIsArg },
 502   { S("-warningTime"),  S("*warningTime"),  XrmoptionSepArg },
 503   { S("-display"),      S("*display"),      XrmoptionSepArg },
 504   { S("-geometry"),     S("*geometry"),     XrmoptionSepArg },
 505   { S("-into"),         S("*parentWindow"), XrmoptionSepArg },
 506   { S("-iconic"),       S("*iconic"),       XrmoptionIsArg },
 507   { S("-withdrawn"),    S("*withdrawn"),    XrmoptionIsArg },
 508 #define GC(g)
 509 #define C(c,u)                                                  \
 510   { S("-" #u "Color"),  S("*" #u "Color"),  XrmoptionSepArg },  \
 511   { S("-" #u "Colour"), S("*" #u "Color"),  XrmoptionSepArg },
 512   COLOURS
 513 #undef GC
 514 #undef C
 515 };
 516
 517 static const char *getresource(const char *want) {
 518   char name_buf[256], class_buf[256];
 519   XrmValue val;
 520   char *rep_type_dummy;
 521   int r;
 522
 523   assert(strlen(want) < 128);
 524
 525   sprintf(name_buf,"xbatmon-simple.%s",want);
 526   sprintf(class_buf,"Xbatmon-Simple.%s",want);
 527
 528   r= XrmGetResource(xrm, name_buf,class_buf, &rep_type_dummy, &val);
 529   if (r) return val.addr;
 530
 531   sprintf(name_buf,"xacpi-simple.%s",want);
 532   sprintf(class_buf,"Xacpi-Simple.%s",want);
 533
 534   r= XrmGetResource(xrm, name_buf,class_buf, &rep_type_dummy, &val);
 535   if (r) return val.addr;
 536
 537   return 0;
 538 }
 539
 540 static int getresource_bool(const char *want, int def, int *cache) {
 541   /* *cache should be initialised to -1 and will be set to !!value
 542    * alternatively cache==0 is allowed */
 543
 544   if (cache && *cache >= 0) return *cache;
 545
 546   const char *str= getresource(want);
 547   int result = def;
 548   if (str && str[0]) {
 549     char *ep;
 550     long l= strtol(str,&ep,0);
 551     if (!*ep) {
 552       result = l > 0;
 553     } else {
 554       switch (str[0]) {
 555       case 't': case 'T': case 'y': case 'Y':         result= 1;  break;
 556       case 'f': case 'F': case 'n': case 'N':         result= 0;  break;
 557       case '-': /* option name from XrmoptionIsArg */ result= 1;  break;
 558       }
 559     }
 560   }
 561
 562   if (cache) *cache= result;
 563   return result;
 564 }
 565
 566 static void more_resources(const char *str, const char *why) {
 567   XrmDatabase more;
 568
 569   if (!str) return;
 570
 571   more= XrmGetStringDatabase((char*)str);
 572   if (!more) fail(why);
 573   XrmCombineDatabase(more,&xrm,0);
 574 }
 575
 576 static void parseargs(int argc, char **argv) {
 577   Screen *screenscreen;
 578
 579   XrmInitialize();
 580
 581   XrmParseCommand(&xrm, (XrmOptionDescRec*)optiontable,
 582                   sizeof(optiontable)/sizeof(*optiontable),
 583                   program_name, &argc, argv);
 584
 585   if (argc>1) badusage();
 586
 587   getresource_bool("debug",0,&debug);
 588
 589   const char *alarmlevel_string= getresource("alarmLevel");
 590   alarmlevel = alarmlevel_string ? atoi(alarmlevel_string) : 300;
 591
 592   parentwindow = getresource("parentWindow");
 593
 594   disp= XOpenDisplay(getresource("display"));
 595   if (!disp) fail("could not open display");
 596
 597   screen= DefaultScreen(disp);
 598
 599   screenscreen= ScreenOfDisplay(disp,screen);
 600   if (!screenscreen) fail("screenofdisplay");
 601   more_resources(XScreenResourceString(screenscreen), "screen resources");
 602   more_resources(XResourceManagerString(disp), "display resources");
 603   more_resources(defaultresources, "default resources");
 604 }
 605
 606 /*---------- display ----------*/
 607
 608 static Window win;
 609 static int width, height;
 610 static Colormap cmap;
 611 static unsigned long lastbackground;
 612
 613 typedef struct {
 614   GC gc;
 615   unsigned long lastfg;
 616 } Gcstate;
 617
 618 #define C(c,u) static unsigned long pix_##u;
 619 #define GC(g) static Gcstate gc_##g;
 620   COLOURS
 621 #undef C
 622 #undef GC
 623
 624 static void refresh(void);
 625
 626 #define CHGMASK_CHG_DIS ((1u<<CHGST_CHARGING) | (1u<<CHGST_DISCHARGING))
 627
 628 static void failr(const char *m, int r) {
 629   fprintf(stderr,"error: %s (code %d)\n", m, r);
 630   exit(-1);
 631 }
 632
 633 static void setbackground(unsigned long newbg) {
 634   int r;
 635
 636   if (newbg == lastbackground) return;
 637   r= XSetWindowBackground(disp,win,newbg);
 638   if (!r) fail("XSetWindowBackground");
 639   lastbackground= newbg;
 640 }
 641
 642 static void setforeground(Gcstate *g, unsigned long px) {
 643   XGCValues gcv;
 644   int r;
 645
 646   if (g->lastfg == px) return;
 647
 648   memset(&gcv,0,sizeof(gcv));
 649   g->lastfg= gcv.foreground= px;
 650   r= XChangeGC(disp,g->gc,GCForeground,&gcv);
 651   if (!r) fail("XChangeGC");
 652 }
 653
 654 static void show_solid(unsigned long px) {
 655   setbackground(px);
 656   XClearWindow(disp,win);
 657 }
 658
 659 static void show(void) {
 660   double elap, then;
 661   int i, leftmost_lit, leftmost_nondeg, beyond, first_beyond;
 662
 663   if (!charging_mask)
 664     return show_solid(pix_absent);
 665
 666   if (charging_mask & (1u << CHGST_ERROR))
 667     return show_solid(pix_error);
 668
 669   setbackground(pix_degraded);
 670   XClearWindow(disp,win);
 671
 672   setforeground(&gc_remain,
 673                 !(charging_mask & CHGMASK_CHG_DIS) ?
 674                 (~charging_mask & (1u << CHGST_CHARGED) ?
 675                  pix_notcharging : pix_charged) :
 676                 !(~charging_mask & CHGMASK_CHG_DIS) ? pix_confusing :
 677                 charging_mask & (1u<<CHGST_CHARGING)
 678                 ? pix_charging : pix_discharging);
 679
 680   setforeground(&gc_empty, alarmed ? pix_low : pix_normal);
 681
 682   for (i=0, first_beyond=1; i<height; i++) {
 683     elap= !i ? 0 :
 684       height==2 ? BOTTOM :
 685       TOP * exp( (double)i / (height-2) * log( (double)BOTTOM/TOP ) );
 686
 687     then= fill_norm + ratepersec_norm * elap;
 688
 689     beyond=
 690       ((charging_mask & (1u<<CHGST_DISCHARGING) && then <= 0.0) ||
 691        (charging_mask & (1u<<CHGST_CHARGING) && then>=nondegraded_norm));
 692
 693     if (then <= 0.0) then= 0.0;
 694     else if (then >= nondegraded_norm) then= nondegraded_norm;
 695
 696     leftmost_lit= width * then;
 697     leftmost_nondeg= width * nondegraded_norm;
 698
 699     if (beyond && first_beyond) {
 700       XDrawLine(disp, win, gc_white.gc, 0,i, leftmost_nondeg,i);
 701       first_beyond= 0;
 702     } else {
 703       if (leftmost_lit < leftmost_nondeg)
 704         XDrawLine(disp, win, gc_empty.gc,
 705                   leftmost_lit,i, leftmost_nondeg,i);
 706       if (leftmost_lit >= 0)
 707         XDrawLine(disp, win, gc_remain.gc, 0,i, leftmost_lit,i);
 708     }
 709   }
 710 }
 711
 712 static void initgc(Gcstate *gc_r) {
 713   XGCValues gcv;
 714
 715   memset(&gcv,0,sizeof(gcv));
 716   gcv.function= GXcopy;
 717   gcv.line_width= 1;
 718   gc_r->lastfg= gcv.foreground= pix_equilibrium;
 719   gc_r->gc= XCreateGC(disp,win, GCFunction|GCLineWidth|GCForeground, &gcv);
 720 }
 721
 722 static void colour(unsigned long *pix_r, const char *whichcolour) {
 723   XColor xc;
 724   const char *name;
 725   Status st;
 726
 727   name= getresource(whichcolour);
 728   if (!name) fail("get colour resource");
 729
 730   st= XAllocNamedColor(disp,cmap,name,&xc,&xc);
 731   if (!st) fail(name);
 732
 733   *pix_r= xc.pixel;
 734 }
 735
 736 static void initgraphics(int argc, char **argv) {
 737   int xwmgr, r;
 738   const char *geom_string;
 739   XSizeHints *normal_hints;
 740   XWMHints *wm_hints;
 741   XClassHint *class_hint;
 742   int pos_x, pos_y, gravity;
 743   char *program_name_silly;
 744
 745   program_name_silly= (char*)program_name;
 746
 747   normal_hints= XAllocSizeHints();
 748   wm_hints= XAllocWMHints();
 749   class_hint= XAllocClassHint();
 750
 751   if (!normal_hints || !wm_hints || !class_hint)
 752     fail("could not alloc hint(s)");
 753
 754   geom_string= getresource("geometry");
 755
 756   xwmgr= XWMGeometry(disp,screen, geom_string,"128x32", 0,
 757                  normal_hints,
 758                  &pos_x, &pos_y,
 759                  &width, &height,
 760                  &gravity);
 761
 762   unsigned long parentwindowid;
 763   if (parentwindow)
 764     parentwindowid = strtoul(parentwindow,0,0);
 765   else
 766     parentwindowid = DefaultRootWindow(disp);
 767
 768   win= XCreateSimpleWindow(disp,parentwindowid,
 769                            pos_x,pos_y,width,height,0,0,0);
 770   cmap= DefaultColormap(disp,screen);
 771
 772 #define C(c,u) colour(&pix_##u, #u "Color");
 773 #define GC(g) initgc(&gc_##g);
 774   COLOURS
 775 #undef C
 776 #undef GC
 777
 778   r= XSetWindowBackground(disp,win,pix_degraded);
 779   if (!r) fail("init set background");
 780   lastbackground= pix_degraded;
 781
 782   normal_hints->flags= PWinGravity;
 783   normal_hints->win_gravity= gravity;
 784   normal_hints->x= pos_x;
 785   normal_hints->y= pos_y;
 786   normal_hints->width= width;
 787   normal_hints->height= height;
 788   if ((xwmgr & XValue) || (xwmgr & YValue))
 789     normal_hints->flags |= USPosition;
 790
 791   wm_hints->flags= InputHint;
 792   wm_hints->input= False;
 793   wm_hints->initial_state=
 794     (getresource_bool("withdrawn",0,0) ? WithdrawnState :
 795      getresource_bool("iconic",0,0) ? IconicState
 796      : NormalState);
 797
 798   class_hint->res_name= program_name_silly;
 799   class_hint->res_class= program_name_silly;
 800
 801   XmbSetWMProperties(disp,win, program_name,program_name,
 802                      argv,argc, normal_hints, wm_hints, class_hint);
 803
 804   XSelectInput(disp,win, ExposureMask|StructureNotifyMask);
 805   XMapWindow(disp,win);
 806 }
 807
 808 static void refresh(void) {
 809   acquiredata();
 810   show();
 811 }
 812
 813 static void newgeometry(void) {
 814   int dummy;
 815   Window dummyw;
 816
 817   XGetGeometry(disp,win, &dummyw,&dummy,&dummy, &width,&height, &dummy,&dummy);
 818 }
 819
 820 static void eventloop(void) {
 821   XEvent ev;
 822   struct pollfd pfd;
 823   int r, timeout;
 824
 825   newgeometry();
 826   refresh();
 827
 828   for (;;) {
 829     XFlush(disp);
 830
 831     pfd.fd= ConnectionNumber(disp);
 832     pfd.events= POLLIN|POLLERR;
 833
 834     timeout= !(charging_mask & (1u << CHGST_ERROR)) ? TIMEOUT : TIMEOUT_ONERROR;
 835     r= poll(&pfd,1,timeout);
 836     if (r==-1 && errno!=EINTR) failr("poll",errno);
 837
 838     while (XPending(disp)) {
 839       XNextEvent(disp,&ev);
 840       if (ev.type == ConfigureNotify) {
 841         XConfigureEvent *ce= (void*)&ev;
 842         width= ce->width;
 843         height= ce->height;
 844       }
 845     }
 846     refresh();
 847   }
 848 }
 849
 850 int main(int argc, char **argv) {
 851   parseargs(argc,argv);
 852   initacquire();
 853   initgraphics(argc,argv);
 854   eventloop();
 855   return 0;
 856 }