chiark / gitweb /
actually reads /proc - compiles but untested
[chiark-utils.git] / cprogs / xacpi-simple.c
1 /*
2  * display outputs, per line:
3  *
4  *    Remaining: | Empty:   | Degraded:
5  *     green     |  red     |  black      discharging
6  *     blue      |  red     |  black      charging
7  *     cyan      |  red     |  black      charged
8  *     grey      |  red     |  black      charging&discharching!
9  *     green     |  yellow  |  black      discharging - low!
10  *     blue      |  yellow  |  black      charging - low
11  *     cyan      |  yellow  |  black      charged - low [1]
12  *     grey      |  yellow  |  black      charging&discharching! - low [1]
13  *       ...  dark grey  ...              no batteries present
14  *
15  * [1] battery must be quite badly degraded
16  */
17
18 #include <stdio.h>
19 #include <assert.h>
20 #include <math.h>
21 #include <stdlib.h>
22 #include <string.h>
23 #include <errno.h>
24 #include <unistd.h>
25 #include <ctype.h>
26
27 #include <sys/types.h>
28 #include <dirent.h>
29
30 #include <X11/Xlib.h>
31
32 /*---------- general utility stuff and declarations ----------*/
33
34 static void fail(const char *m) {
35   fprintf(stderr,"error: %s\n", m);
36   sleep(5);
37   exit(-1);
38 }
39 static void badusage(void) { fail("bad usage"); }
40
41 #define CHGST_DISCHARGING 0 /* Reflects order in E(state,charging_state)  */
42 #define CHGST_CHARGING    1 /*  in fields table.  Also, much code assumes */
43 #define CHGST_CHARGED     2 /*  exactly these three possible states.      */
44
45 /*---------- structure of and results from /proc/acpi/battery/... ----------*/
46 /* variables thisbat_... are the results from readbattery();
47  * if readbattery() succeeds they are all valid and not VAL_NOTFOUND
48  */
49
50 typedef struct batinfo_field {
51   const char *file;
52   const char *label;
53   unsigned long *valuep;
54   const char *unit;
55   const char *enumarray[10];
56 } batinfo_field;
57
58 #define QUANTITY_FIELDS                         \
59   QF(info,  design_capacity, "mWh")             \
60   QF(info,  last_full_capacity, "mWh")          \
61   QF(state, present_rate,       "mW")           \
62   QF(state, remaining_capacity, "mWh")          \
63   QF(alarm, alarm,              "mWh")
64
65 #define QF(f,l,u) static unsigned long thisbat_##f##_##l;
66   QUANTITY_FIELDS
67 #undef QF
68
69 static unsigned long thisbat_info_present, thisbat_state_present;
70 static unsigned long thisbat_state_charging_state;
71
72 #define VAL_NOTFOUND (~0UL)
73
74 static const batinfo_field fields[]= {
75 #define E(f,l)       #f, #l, &thisbat_##f##_##l, 0
76 #define QF(f,l,u)  { #f, #l, &thisbat_##f##_##l, u },
77   { E(info, present),            { "no", "yes" }                    },
78   { E(state,present),            { "no", "yes" }                    },
79   { E(state,charging_state),     { "discharging charging charged" } },
80   QUANTITY_FIELDS           /* take care re charging_state values order -  */
81   { 0 }                     /* if you must change it, search for CHGST_... */
82 #undef E
83 #undef QF
84 };
85
86 static const char *files[]= { "info", "state", "alarm" };
87
88 /*---------- parsing of one battery in /proc/acpi/battery/... ----------*/
89
90 /* variables private to the parser and its error handlers */
91 static char batlinebuf[1000];
92 static FILE *batfile;
93 static const char *batdirname;
94 static const char *batfilename;
95 static const char *batlinevalue;
96
97 static int batfailf(const char *why) {
98   if (batlinevalue) {
99     fprintf(stderr,"battery/%s/%s: %s value `%s': %s\n",
100             batdirname,batfilename, batlinebuf,batlinevalue,why);
101   } else {
102     fprintf(stderr,"battery/%s/%s: %s: `%s'\n",
103             batdirname,batfilename, why, batlinebuf);
104   }
105   return -1;
106 }
107
108 static int batfaile(const char *syscall, const char *target) {
109   fprintf(stderr,"battery/%s: failed to %s %s: %s\n",
110           batdirname ? batdirname : "*", syscall, target, strerror(errno));
111   return -1;
112 }
113
114 static void chdir_base(void) {
115   int r;
116   
117   r= chdir("/proc/acpi/battery");
118   if (r) batfaile("chdir","/proc/acpi/battery");
119 }
120
121 static void tidybattery(void) {
122   if (batfile) { fclose(batfile); batfile=0; }
123   if (batdirname) { chdir_base(); batdirname=0; }
124 }
125
126 static int readbattery(void) { /* 0=>ok, -1=>couldn't */
127   const batinfo_field *field;
128   const char *const *cfilename, *const *enumsearch;
129   char *colon, *ep;
130   int r, l, missing;
131   
132   r= chdir(batdirname);
133   if (r) return batfaile("chdir",batdirname);
134
135   for (field=fields; field->file; field++)
136     *field->valuep= VAL_NOTFOUND;
137
138   for (cfilename=files;
139        (batfilename= *cfilename);
140        cfilename++) {
141     batfile= fopen(batfilename,"r");
142     if (!batfile) return batfaile("open",batfilename);
143
144     for (;;) {
145       batlinevalue= 0;
146       
147       fgets(batlinebuf,sizeof(batlinebuf),batfile);
148       if (ferror(batfile)) return batfaile("read",batfilename);
149       l= strlen(batlinebuf);
150       if (l==0 && feof(batfile)) break;
151       assert(l>0);
152       if (batlinebuf[l-1] != '\n')
153         return batfailf("line too long");
154       batlinebuf[l-1]= 0;
155       colon= strchr(batlinebuf,':');
156       if (!colon)
157         return batfailf("line without a colon");
158       *colon= 0;
159
160       for (field=fields; ; field++) {
161         if (!strcmp(field->file,batfilename) &&
162             !strcmp(field->label,batlinebuf))
163           goto label_interesting;
164       }
165       continue;
166
167     label_interesting:
168       for (batlinevalue= colon+1;
169            *batlinevalue && isspace((unsigned char)*batlinevalue);
170            batlinevalue++);
171
172       if (field->unit) {
173
174         *field->valuep= strtoul(batlinevalue,&ep,10);
175         if (ep==batlinevalue || *ep!=' ')
176           batfailf("value number syntax incorrect");
177         if (strcmp(ep+1,field->unit)) batfailf("incorrect unit");
178
179       } else {
180         
181         for (*field->valuep=0, enumsearch=field->enumarray;
182              *enumsearch && strcmp(*enumsearch,batlinevalue);
183              (*field->valuep)++, enumsearch++);
184         if (!*enumsearch)
185           batfailf("unknown enum value");
186
187       }
188     }
189
190     fclose(batfile);
191     batfile= 0;
192   }
193
194   r= chdir(batdirname);
195   if (r) return batfaile("chdir","..");
196   batdirname= 0;
197
198   for (field=fields, missing=0;
199        field->file;
200        field++) {
201     if (*field->valuep == VAL_NOTFOUND) {
202       fprintf(stderr,"battery/%s/%s: %s: not found\n",
203               batdirname, field->file,field->label);
204       missing++;
205     }
206   }
207   if (missing) return -1;
208
209   return 0;
210 }   
211
212 /*---------- data collection and analysis ----------*/
213
214 /* These next three variables are the results of the charging state */
215 static unsigned charging_state_mask; /* 1u<<CHGST_* | ... */
216 static double nondegraded_norm, fill_norm, ratepersec_norm;
217 static int alarm_level; /* 0=ok, 1=low */
218
219 #define QF(f,l,u) \
220   static double total_##f##_##l;
221   QUANTITY_FIELDS
222 #undef QF
223
224 static void acquiredata(void) {
225   DIR *di;
226   struct dirent *de;
227   int r;
228   
229   charging_state_mask= 0;
230
231 #define QF(f,l,u) \
232   total_##f##_##l= 0;
233   QUANTITY_FIELDS
234 #undef QF
235
236   di= opendir(".");  if (!di) batfaile("opendir","battery");
237   while ((de= readdir(di))) {
238     if (de->d_name[0]==0 || de->d_name[0]=='.') continue;
239
240     batdirname= de->d_name;
241     r= readbattery();
242     tidybattery();
243
244     if (r) continue;
245
246     if (!thisbat_info_present || !thisbat_state_present) {
247       printf("battery/%s not present\n",de->d_name);
248       continue;
249     }
250
251     charging_state_mask |= 1u << thisbat_state_charging_state;
252
253 #define QF(f,l,u) \
254     total_##f##_##l += thisbat_##f##_##l;
255     QUANTITY_FIELDS
256 #undef QF
257
258     if (thisbat_state_charging_state == CHGST_DISCHARGING)
259       /* negate it */
260       total_state_present_rate -= 2.0 * thisbat_state_present_rate;
261       
262   }
263   closedir(di);
264
265   if (total_info_design_capacity < 0.5)
266     total_info_design_capacity= 1.0;
267
268   if (total_info_last_full_capacity < total_state_remaining_capacity)
269     total_info_last_full_capacity= total_state_remaining_capacity;
270   if (total_info_design_capacity < total_info_last_full_capacity)
271     total_info_design_capacity= total_info_last_full_capacity;
272
273   alarm_level=
274     (total_state_remaining_capacity < total_alarm_alarm &&
275      !(charging_state_mask & 1u << CHGST_CHARGING));
276
277   nondegraded_norm= total_info_last_full_capacity / total_info_design_capacity;
278   fill_norm= total_state_remaining_capacity / total_info_design_capacity;
279   ratepersec_norm=  total_state_present_rate
280     / (3600.0 * total_info_design_capacity);
281 }
282
283 static void initacquire(void) {
284   chdir_base();
285 }  
286
287 /*---------- display ----------*/
288
289 #define TOP      60
290 #define BOTTOM 3600
291
292 #define COLOURS                                 \
293   C(black)                                      \
294   C(green)                                      \
295   C(red)                                        \
296   C(yellow)                                     \
297   C(blue)                                       \
298   C(cyan)                                       \
299   C(grey)                                       \
300   C(darkgrey)                                   \
301   GC(remain)                                    \
302   GC(empty)
303
304 static Display *disp;
305 static Window win;
306 static int width, height;
307 static Colormap cmap;
308 static int screen;
309 static unsigned long lastbackground;
310
311 typedef struct {
312   GC gc;
313   unsigned long lastfg;
314 } Gcstate;
315
316 #define C(c) static unsigned long pix_##c;
317 #define GC(g) static Gcstate gc_##g;
318   COLOURS
319 #undef C
320 #undef GC
321
322 static void refresh(void);
323
324 #define CHGMASK_CHG_DIS (1u<<CHGST_CHARGING | 1u<<CHGST_DISCHARGING)
325
326 static void setbackground(unsigned long newbg) {
327   int r;
328   
329   if (newbg == lastbackground) return;
330   r= XSetWindowBackground(disp,win,newbg);
331   if (r) fail("XSetWindowBackground");
332   lastbackground= newbg;
333 }
334
335 static void setforeground(Gcstate *g, unsigned long px) {
336   XGCValues gcv;
337   int r;
338   
339   if (g->lastfg == px) return;
340   
341   memset(&gcv,0,sizeof(gcv));
342   g->lastfg= gcv.foreground= px;
343   r= XChangeGC(disp,g->gc,GCForeground,&gcv);
344   if (r) fail("XChangeGC");
345 }
346
347 static void show(void) {
348   double elap, then;
349   int i, leftmost_lit, leftmost_nondeg;
350
351   if (!charging_state_mask) {
352     setbackground(pix_darkgrey);
353     XClearWindow(disp,win);
354     return;
355   }
356
357   setbackground(pix_black);
358   XClearWindow(disp,win);
359   
360   setforeground(&gc_remain,
361                 !(charging_state_mask & CHGMASK_CHG_DIS) ? pix_cyan :
362                 !(~charging_state_mask & CHGMASK_CHG_DIS) ? pix_grey :
363                 charging_state_mask & (1u<<CHGST_CHARGING)
364                 ? pix_blue : pix_green);
365                 
366   setforeground(&gc_empty, alarm_level ? pix_yellow : pix_red);
367
368   for (i=0; i<height; i++) {
369     elap= !i ? 0 :
370       height==2 ? BOTTOM :
371       TOP * exp( (double)i / (height-2) * log( (double)BOTTOM/TOP ) );
372     
373     then= fill_norm + ratepersec_norm * elap;
374     if (then <= 0.0) then= 0.0;
375     else if (then >= nondegraded_norm) then= nondegraded_norm;
376
377     leftmost_lit= width * then;
378     leftmost_nondeg= width * nondegraded_norm;
379
380     if (leftmost_lit >= 0)
381       XDrawLine(disp, win, gc_remain.gc, 0,i, leftmost_lit,i);
382     if (leftmost_lit < leftmost_nondeg)
383       XDrawLine(disp, win, gc_empty.gc,
384                 leftmost_lit+1,i, leftmost_nondeg,i);
385   }
386 }
387
388 static void initgc(Gcstate *gc_r) {
389   XGCValues gcv;
390
391   memset(&gcv,0,sizeof(gcv));
392   gcv.function= GXcopy;
393   gcv.line_width= 1;
394   gc_r->lastfg= gcv.foreground= pix_black;
395   gc_r->gc= XCreateGC(disp,win, GCFunction|GCLineWidth|GCForeground, &gcv);
396 }
397
398 static void colour(unsigned long *pix_r, const char *name) {
399   XColor xc;
400   Status st;
401   
402   st= XAllocNamedColor(disp,cmap,name,&xc,&xc);
403   if (!st) fail("couldn't allocate colour");
404   
405   *pix_r= xc.pixel;
406 }
407
408 static void initgraphics(void) {
409   int r;
410   
411   disp= XOpenDisplay(0); if (!disp) fail("could not open display");
412
413   screen= DefaultScreen(disp);
414   win= XCreateSimpleWindow(disp,DefaultRootWindow(disp),0,0,100,20,0,0,0);
415   cmap= DefaultColormap(disp,screen);
416   
417 #define C(c) colour(&pix_##c, #c);
418 #define GC(g) initgc(&gc_##g);
419   COLOURS
420 #undef C
421 #undef GC
422
423   r= XSetWindowBackground(disp,win,pix_black);
424   if (r) fail("init set background");
425   lastbackground= pix_black;
426
427   XSelectInput(disp,win, ExposureMask|VisibilityChangeMask);
428   XMapWindow(disp,win);
429 }
430  
431 static void refresh(void) {
432   acquiredata();
433   show();
434 }
435
436 static void newgeometry(void) {
437   int dummy;
438   Window dummyw;
439   
440   XGetGeometry(disp,win, &dummyw,&dummy,&dummy, &width,&height, &dummy,&dummy);
441   refresh();
442 }
443
444 static void eventloop(void) {
445   XEvent ev;
446   
447   newgeometry();
448   
449   for (;;) {
450     XNextEvent(disp,&ev);
451     newgeometry();
452   }
453 }
454
455 int main(int argc, const char *const *argv) {
456   if (!argv[0] || argv[1])
457     badusage();
458
459   initacquire();
460   initgraphics();
461   eventloop();
462   return 0;
463 }