wmmemload/src/main.c

   1 /*
   2  *    WMMemLoad - A dockapp to monitor memory usage
   3  *    Copyright (C) 2002  Mark Staggs <me@markstaggs.net>
   4  *
   5  *    Based on work by Seiichi SATO <ssato@sh.rim.or.jp>
   6  *    Copyright (C) 2001,2002  Seiichi SATO <ssato@sh.rim.or.jp>
   7
   8  *    This program is free software; you can redistribute it and/or modify
   9  *    it under the terms of the GNU General Public License as published by
  10  *    the Free Software Foundation; either version 2 of the License, or
  11  *    (at your option) any later version.
  12
  13  *    This program is distributed in the hope that it will be useful,
  14  *    but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  *    GNU General Public License for more details.
  17
  18  *    You should have received a copy of the GNU General Public License
  19  *    along with this program; if not, write to the Free Software
  20  *    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  21  *
  22  */
  23
  24 #ifdef HAVE_CONFIG_H
  25 #include "config.h"
  26 #endif
  27
  28 #include "dockapp.h"
  29 #include "mem.h"
  30 #include "backlight_on.xpm"
  31 #include "backlight_off.xpm"
  32 #include "parts.xpm"
  33
  34 #define SIZE        58
  35 #define WINDOWED_BG "  \tc #AEAAAE"
  36 #define MAX_HISTORY 16
  37 #define CPUNUM_NONE -1
  38
  39 typedef enum { LIGHTON, LIGHTOFF } light;
  40
  41 Pixmap pixmap;
  42 Pixmap backdrop_on;
  43 Pixmap backdrop_off;
  44 Pixmap parts;
  45 Pixmap mask;
  46 static char     *display_name = "";
  47 static char     *light_color = NULL;    /* back-light color */
  48 static unsigned update_interval = 1;
  49 static light    backlight = LIGHTOFF;
  50
  51 static struct   mem_options mem_opts;
  52 static unsigned mem_usage = 0;
  53 static unsigned swap_usage = 0;
  54 static unsigned alarm_mem = 101;
  55 static unsigned alarm_swap = 101;
  56
  57 /* prototypes */
  58 static void update(void);
  59 static void switch_light(void);
  60 static void draw_memdigit(int per);
  61 static void draw_swapdigit(int per);
  62 static void parse_arguments(int argc, char **argv);
  63 static void print_help(char *prog);
  64
  65 int main(int argc, char **argv)
  66 {
  67    XEvent event;
  68    XpmColorSymbol colors[2] = { {"Back0", NULL, 0}, {"Back1", NULL, 0} };
  69    int ncolor = 0;
  70
  71    /* Parse CommandLine */
  72    mem_opts.ignore_buffers = mem_opts.ignore_cached
  73                                      = mem_opts.ignore_wired = False;
  74    parse_arguments(argc, argv);
  75
  76    /* Initialize Application */
  77    mem_init();
  78    dockapp_open_window(display_name, PACKAGE, SIZE, SIZE, argc, argv);
  79    dockapp_set_eventmask(ButtonPressMask);
  80
  81    if(light_color)
  82    {
  83       colors[0].pixel = dockapp_getcolor(light_color);
  84       colors[1].pixel = dockapp_blendedcolor(light_color, -24, -24, -24, 1.0);
  85       ncolor = 2;
  86    }
  87
  88    /* change raw xpm data to pixmap */
  89    if(dockapp_iswindowed)
  90       backlight_on_xpm[1] = backlight_off_xpm[1] = WINDOWED_BG;
  91
  92    if(!dockapp_xpm2pixmap(backlight_on_xpm, &backdrop_on, &mask, colors, ncolor))
  93    {
  94       fprintf(stderr, "Error initializing backlit background image.\n");
  95       exit(1);
  96    }
  97    if(!dockapp_xpm2pixmap(backlight_off_xpm, &backdrop_off, NULL, NULL, 0))
  98    {
  99       fprintf(stderr, "Error initializing background image.\n");
 100       exit(1);
 101    }
 102    if(!dockapp_xpm2pixmap(parts_xpm, &parts, NULL, colors, ncolor))
 103    {
 104       fprintf(stderr, "Error initializing parts image.\n");
 105       exit(1);
 106    }
 107
 108    /* shape window */
 109    if(!dockapp_iswindowed)
 110       dockapp_setshape(mask, 0, 0);
 111    if(mask) XFreePixmap(display, mask);
 112
 113    /* pixmap : draw area */
 114    pixmap = dockapp_XCreatePixmap(SIZE, SIZE);
 115
 116    /* Initialize pixmap */
 117    if(backlight == LIGHTON)
 118       dockapp_copyarea(backdrop_on, pixmap, 0, 0, SIZE, SIZE, 0, 0);
 119    else
 120       dockapp_copyarea(backdrop_off, pixmap, 0, 0, SIZE, SIZE, 0, 0);
 121
 122    dockapp_set_background(pixmap);
 123    dockapp_show();
 124
 125    /* Main loop */
 126    while(1)
 127    {
 128       if (dockapp_nextevent_or_timeout(&event, update_interval * 1000))
 129       {
 130          /* Next Event */
 131          switch(event.type)
 132          {
 133             case ButtonPress:
 134                switch_light();
 135                break;
 136             default: /* make gcc happy */
 137                break;
 138          }
 139       }
 140       else
 141       {
 142          /* Time Out */
 143          update();
 144       }
 145    }
 146
 147    return 0;
 148 }
 149
 150 /* called by timer */
 151 static void update(void)
 152 {
 153    static light pre_backlight;
 154    static Bool in_alarm_mode = False;
 155
 156    /* get current cpu usage in percent */
 157    mem_getusage(&mem_usage, &swap_usage, &mem_opts);
 158
 159    /* alarm mode */
 160    if(mem_usage >= alarm_mem || swap_usage >= alarm_swap)
 161    {
 162       if(!in_alarm_mode)
 163       {
 164          in_alarm_mode = True;
 165          pre_backlight = backlight;
 166       }
 167       if(backlight == LIGHTOFF)
 168       {
 169          switch_light();
 170          return;
 171       }
 172    }
 173    else
 174    {
 175       if(in_alarm_mode)
 176       {
 177          in_alarm_mode = False;
 178          if (backlight != pre_backlight)
 179          {
 180             switch_light();
 181             return;
 182          }
 183       }
 184    }
 185
 186    /* all clear */
 187    if (backlight == LIGHTON)
 188       dockapp_copyarea(backdrop_on, pixmap, 0, 0, 58, 58, 0, 0);
 189    else
 190       dockapp_copyarea(backdrop_off, pixmap, 0, 0, 58, 58, 0, 0);
 191
 192    /* draw digit */
 193    draw_memdigit(mem_usage);
 194    draw_swapdigit(swap_usage);
 195
 196    /* show */
 197    dockapp_copy2window(pixmap);
 198 }
 199
 200 /* called when mouse button pressed */
 201 static void switch_light(void)
 202 {
 203    switch (backlight)
 204    {
 205       case LIGHTOFF:
 206          backlight = LIGHTON;
 207          dockapp_copyarea(backdrop_on, pixmap, 0, 0, 58, 58, 0, 0);
 208          break;
 209       case LIGHTON:
 210          backlight = LIGHTOFF;
 211          dockapp_copyarea(backdrop_off, pixmap, 0, 0, 58, 58, 0, 0);
 212          break;
 213    }
 214
 215    /* redraw digit */
 216    mem_getusage(&mem_usage, &swap_usage, &mem_opts);
 217    draw_memdigit(mem_usage);
 218    draw_swapdigit(swap_usage);
 219
 220    /* show */
 221    dockapp_copy2window(pixmap);
 222 }
 223
 224 static void draw_memdigit(int per)
 225 {
 226    int v100, v10, v1;
 227    int y = 0;
 228
 229    if (per < 0) per = 0;
 230    if (per > 100) per = 100;
 231
 232    v100 = per / 100;
 233    v10  = (per - v100 * 100) / 10;
 234    v1   = (per - v100 * 100 - v10 * 10);
 235
 236    if (backlight == LIGHTON) y = 20;
 237
 238    /* draw digit */
 239    dockapp_copyarea(parts, pixmap, v1 * 10, y, 10, 20, 29, 7);
 240    if (v10 != 0)
 241       dockapp_copyarea(parts, pixmap, v10 * 10, y, 10, 20, 17, 7);
 242    if (v100 == 1)
 243    {
 244       dockapp_copyarea(parts, pixmap, 10, y, 10, 20,  5, 7);
 245       dockapp_copyarea(parts, pixmap, 0, y, 10, 20, 17, 7);
 246    }
 247 }
 248
 249
 250 static void draw_swapdigit(int per)
 251 {
 252    int v100, v10, v1;
 253    int y = 0;
 254
 255    if (per < 0) per = 0;
 256    if (per > 100) per = 100;
 257
 258    v100 = per / 100;
 259    v10  = (per - v100 * 100) / 10;
 260    v1   = (per - v100 * 100 - v10 * 10);
 261
 262    if (backlight == LIGHTON) y = 20;
 263
 264    /* draw digit */
 265    dockapp_copyarea(parts, pixmap, v1 * 10, y, 10, 20, 29, 34);
 266    if (v10 != 0)
 267       dockapp_copyarea(parts, pixmap, v10 * 10, y, 10, 20, 17, 34);
 268    if (v100 == 1)
 269    {
 270       dockapp_copyarea(parts, pixmap, 10, y, 10, 20, 5, 34);
 271       dockapp_copyarea(parts, pixmap, 0, y, 10, 20, 17, 34);
 272    }
 273 }
 274
 275 static void parse_arguments(int argc, char **argv)
 276 {
 277    int i;
 278    int integer;
 279    for (i = 1; i < argc; i++)
 280    {
 281       if (!strcmp(argv[i], "--help") || !strcmp(argv[i], "-h"))
 282          print_help(argv[0]), exit(0);
 283       else if (!strcmp(argv[i], "--version") || !strcmp(argv[i], "-v"))
 284          printf("%s version %s\n", PACKAGE, VERSION), exit(0);
 285       else if (!strcmp(argv[i], "--display") || !strcmp(argv[i], "-d"))
 286       {
 287          display_name = argv[i + 1];
 288          i++;
 289       }
 290       else if (!strcmp(argv[i], "--alarm-mem") || !strcmp(argv[i], "-am"))
 291       {
 292          if (argc == i + 1)
 293             alarm_mem = 90;
 294          else if (sscanf(argv[i + 1], "%i", &integer) != 1)
 295             alarm_mem = 90;
 296          else if (integer < 0 || integer > 100)
 297             fprintf(stderr, "%s: argument %s must be from 0 to 100\n",
 298                     argv[0], argv[i]), exit(1);
 299          else
 300             alarm_mem = integer, i++;
 301       }
 302       else if (!strcmp(argv[i], "--alarm-swap") || !strcmp(argv[i], "-as"))
 303       {
 304          if (argc == i + 1)
 305             alarm_swap = 50;
 306          else if (sscanf(argv[i + 1], "%i", &integer) != 1)
 307             alarm_swap = 50;
 308          else if (integer < 0 || integer > 100)
 309             fprintf(stderr, "%s: argument %s must be from 0 to 100\n",
 310                     argv[0], argv[i]), exit(1);
 311          else
 312             alarm_swap = integer, i++;
 313       }
 314       else if (!strcmp(argv[i], "--backlight") || !strcmp(argv[i], "-bl"))
 315          backlight = LIGHTON;
 316       else if (!strcmp(argv[i], "--light-color") || !strcmp(argv[i], "-lc"))
 317       {
 318          light_color = argv[i + 1];
 319          i++;
 320       }
 321       else if (!strcmp(argv[i], "--interval") || !strcmp(argv[i], "-i"))
 322       {
 323          if (argc == i + 1)
 324             fprintf(stderr, "%s: error parsing argument for option %s\n",
 325                     argv[0], argv[i]), exit(1);
 326          if (sscanf(argv[i + 1], "%i", &integer) != 1)
 327              fprintf(stderr, "%s: error parsing argument for option %s\n",
 328                      argv[0], argv[i]), exit(1);
 329          if (integer < 1)
 330              fprintf(stderr, "%s: argument %s must be >=1\n",
 331                      argv[0], argv[i]), exit(1);
 332          update_interval = integer;
 333          i++;
 334       }
 335       else if (!strcmp(argv[i], "--windowed") || !strcmp(argv[i], "-w"))
 336          dockapp_iswindowed = True;
 337       else if (!strcmp(argv[i], "--broken-wm") || !strcmp(argv[i], "-bw"))
 338          dockapp_isbrokenwm = True;
 339 #ifdef IGNORE_BUFFERS
 340       else if (!strcmp(argv[i], "--ignore-buffers") || !strcmp(argv[i], "-b"))
 341          mem_opts.ignore_buffers = True;
 342 #endif
 343 #ifdef IGNORE_CACHED
 344       else if (!strcmp(argv[i], "--ignore-cached") || !strcmp(argv[i], "-c"))
 345          mem_opts.ignore_cached = True;
 346 #endif
 347 #ifdef IGNORE_WIRED
 348       else if (!strcmp(argv[i], "--ignore-wired") || !strcmp(argv[i], "-wr"))
 349          mem_opts.ignore_wired = True;
 350 #endif
 351       else
 352       {
 353          fprintf(stderr, "%s: unrecognized option '%s'\n", argv[0], argv[i]);
 354          print_help(argv[0]), exit(1);
 355       }
 356    }
 357    if (alarm_mem != 101 && alarm_swap != 101)
 358    {
 359       fprintf(stderr,
 360               "%s: select either '-am, --alarm-mem' or '-as, --alarm-swap'\n",
 361               argv[0]);
 362       exit(1);
 363    }
 364 }
 365
 366 static void print_help(char *prog)
 367 {
 368    printf("Usage : %s [OPTIONS]\n", prog);
 369    printf("WMMemMon - Window Maker memory/swap monitor dockapp\n");
 370    printf("  -d,  --display <string>        display to use\n");
 371    printf("  -bl, --backlight               turn on back-light\n");
 372    printf("  -lc, --light-color <string>    back-light color(rgb:6E/C6/3B is default)\n");
 373    printf("  -i,  --interval <number>       number of secs between updates (1 is default)\n");
 374 #ifdef IGNORE_BUFFERS
 375    printf("  -b,  --ignore-buffers          ignore buffers\n");
 376 #endif
 377 #ifdef IGNORE_CACHED
 378    printf("  -c,  --ignore-cached           ignore cached pages\n");
 379 #endif
 380 #ifdef IGNORE_WIRED
 381    printf("  -wr, --ignore-wired            ignore wired pages\n");
 382 #endif
 383    printf("  -h,  --help                    show this help text and exit\n");
 384    printf("  -v,  --version                 show program version and exit\n");
 385    printf("  -w,  --windowed                run the application in windowed mode\n");
 386    printf("  -bw, --broken-wm               activate broken window manager fix\n");
 387    printf("  -am, --alarm-mem <percentage>  activate alarm mode of memory. <percentage>\n");
 388    printf("                                 is threshold of percentage from 0 to 100.\n");
 389    printf("                                 (90 is default)\n");
 390    printf("  -as, --alarm-swap <percentage> activate alarm mode of swap. <percentage> is\n");
 391    printf("                                 threshold of percentage from 0 to 100.\n");
 392    printf("                                 (50 is default)\n");
 393 }