Dogcows Code - chaz/openbox/blob - util/bsetroot.cc

   1 // bsetroot.cc for Openbox
   2 // Copyright (c) 2002 - 2002 Ben Janens (ben@orodu.net)
   3 // Copyright (c) 2001 - 2002 Sean 'Shaleh' Perry <shaleh at debian.org>
   4 // Copyright (c) 1997 - 2000, 2002 Brad Hughes <bhughes at trolltech.com>
   5 //
   6 // Permission is hereby granted, free of charge, to any person obtaining a
   7 // copy of this software and associated documentation files (the "Software"),
   8 // to deal in the Software without restriction, including without limitation
   9 // the rights to use, copy, modify, merge, publish, distribute, sublicense,
  10 // and/or sell copies of the Software, and to permit persons to whom the
  11 // Software is furnished to do so, subject to the following conditions:
  12 //
  13 // The above copyright notice and this permission notice shall be included in
  14 // all copies or substantial portions of the Software.
  15 //
  16 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  19 // THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  21 // FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  22 // DEALINGS IN THE SOFTWARE.
  23
  24 #ifdef    HAVE_CONFIG_H
  25 #  include "../config.h"
  26 #endif // HAVE_CONFIG_H
  27
  28 #ifdef    HAVE_STRING_H
  29 #  include <string.h>
  30 #endif // HAVE_STRING_H
  31
  32 #ifdef    HAVE_STDLIB_H
  33 #  include <stdlib.h>
  34 #endif // HAVE_STDLIB_H
  35
  36 #ifdef    HAVE_STDIO_H
  37 #  include <stdio.h>
  38 #endif // HAVE_STDIO_H
  39
  40 #include "../src/i18n.h"
  41 #include "../src/BaseDisplay.h"
  42 #include "../src/Image.h"
  43 #include "bsetroot.h"
  44
  45 #include <algorithm>
  46
  47 bsetroot::bsetroot(int argc, char **argv, char *dpy_name)
  48   : BaseDisplay(argv[0], dpy_name)
  49 {
  50   grad = fore = back = (char *) 0;
  51
  52   Bool mod = False, sol = False, grd = False;
  53   int mod_x = 0, mod_y = 0;
  54
  55   for (int i = 1; i < argc; i++) {
  56     if (! strcmp("-help", argv[i])) {
  57       usage();
  58     } else if ((! strcmp("-fg", argv[i])) ||
  59                (! strcmp("-foreground", argv[i])) ||
  60                (! strcmp("-from", argv[i]))) {
  61       if ((++i) >= argc) usage(1);
  62
  63       fore = argv[i];
  64     } else if ((! strcmp("-bg", argv[i])) ||
  65                (! strcmp("-background", argv[i])) ||
  66                (! strcmp("-to", argv[i]))) {
  67       if ((++i) >= argc) usage(1);
  68
  69       back = argv[i];
  70     } else if (! strcmp("-solid", argv[i])) {
  71       if ((++i) >= argc) usage(1);
  72
  73       fore = argv[i];
  74       sol = True;
  75     } else if (! strcmp("-mod", argv[i])) {
  76       if ((++i) >= argc) usage();
  77
  78       mod_x = atoi(argv[i]);
  79
  80       if ((++i) >= argc) usage();
  81
  82       mod_y = atoi(argv[i]);
  83
  84       if (mod_x < 1) mod_x = 1;
  85       if (mod_y < 1) mod_y = 1;
  86
  87       mod = True;
  88     } else if (! strcmp("-gradient", argv[i])) {
  89       if ((++i) >= argc) usage();
  90
  91       grad = argv[i];
  92       grd = True;
  93     } else if (! strcmp("-display", argv[i])) {
  94       // -display passed through tests ealier... we just skip it now
  95       i++;
  96     } else
  97       usage();
  98   }
  99
 100   if ((mod + sol + grd) != True) {
 101     fprintf(stderr,
 102       i18n->getMessage(bsetrootSet, bsetrootMustSpecify,
 103                        "%s: error: must specify one of: "
 104                        "-solid, -mod, -gradient\n"),
 105       getApplicationName());
 106
 107     usage(2);
 108   }
 109
 110   img_ctrl = new BImageControl*[getNumberOfScreens()];
 111   for (unsigned int s = 0; s < getNumberOfScreens(); ++s)
 112     img_ctrl[s] = new BImageControl(*this, *getScreenInfo(s), true);
 113
 114   if (sol && fore) solid();
 115   else if (mod && mod_x && mod_y && fore && back) modula(mod_x, mod_y);
 116   else if (grd && grad && fore && back) gradient();
 117   else usage();
 118 }
 119
 120
 121 bsetroot::~bsetroot(void) {
 122   XSetCloseDownMode(getXDisplay(), RetainPermanent);
 123
 124   XKillClient(getXDisplay(), AllTemporary);
 125
 126   std::for_each(img_ctrl, img_ctrl + getNumberOfScreens(), PointerAssassin());
 127
 128   delete [] img_ctrl;
 129 }
 130
 131
 132 // adapted from wmsetbg
 133 void bsetroot::setPixmapProperty(int screen, Pixmap pixmap) {
 134   static Atom rootpmap_id = None, esetroot_id = None;
 135   Atom type;
 136   int format;
 137   unsigned long length, after;
 138   unsigned char *data;
 139   int mode = PropModeAppend;
 140   const ScreenInfo *screen_info = getScreenInfo(screen);
 141
 142   if (rootpmap_id == None) {
 143     rootpmap_id = XInternAtom(getXDisplay(), "_XROOTPMAP_ID", True);
 144     esetroot_id = XInternAtom(getXDisplay(), "_ESETROOT_PMAP_ID", True);
 145   }
 146
 147   XGrabServer(getXDisplay());
 148
 149   /* Clear out the old pixmap */
 150   XGetWindowProperty(getXDisplay(), screen_info->getRootWindow(),
 151          rootpmap_id, 0L, 1L, False, AnyPropertyType,
 152          &type, &format, &length, &after, &data);
 153
 154   if ((type == XA_PIXMAP) && (format == 32) && (length == 1)) {
 155     unsigned char* data_esetroot = 0;
 156     XGetWindowProperty(getXDisplay(), screen_info->getRootWindow(),
 157                        esetroot_id, 0L, 1L, False, AnyPropertyType,
 158                        &type, &format, &length, &after, &data);
 159     if (data && data_esetroot && *((Pixmap *) data)) {
 160       XKillClient(getXDisplay(), *((Pixmap *) data));
 161       XSync(getXDisplay(), False);
 162       mode = PropModeReplace;
 163     }
 164   }
 165
 166   if (pixmap) {
 167     XChangeProperty(getXDisplay(), screen_info->getRootWindow(),
 168         rootpmap_id, XA_PIXMAP, 32, mode,
 169         (unsigned char *) &pixmap, 1);
 170     XChangeProperty(getXDisplay(), screen_info->getRootWindow(),
 171         esetroot_id, XA_PIXMAP, 32, mode,
 172         (unsigned char *) &pixmap, 1);
 173   } else {
 174     XDeleteProperty(getXDisplay(), screen_info->getRootWindow(),
 175         rootpmap_id);
 176     XDeleteProperty(getXDisplay(), screen_info->getRootWindow(),
 177         esetroot_id);
 178   }
 179
 180   XUngrabServer(getXDisplay());
 181   XFlush(getXDisplay());
 182 }
 183
 184
 185 // adapted from wmsetbg
 186 Pixmap bsetroot::duplicatePixmap(int screen, Pixmap pixmap,
 187          int width, int height) {
 188   XSync(getXDisplay(), False);
 189
 190   Pixmap copyP = XCreatePixmap(getXDisplay(),
 191              getScreenInfo(screen)->getRootWindow(),
 192              width, height,
 193              DefaultDepth(getXDisplay(), screen));
 194   XCopyArea(getXDisplay(), pixmap, copyP, DefaultGC(getXDisplay(), screen),
 195       0, 0, width, height, 0, 0);
 196   XSync(getXDisplay(), False);
 197
 198   return copyP;
 199 }
 200
 201
 202 void bsetroot::solid(void) {
 203   for (unsigned int screen = 0; screen < getNumberOfScreens(); screen++) {
 204     BColor c;
 205
 206     img_ctrl[screen]->parseColor(&c, fore);
 207     if (! c.isAllocated()) c.setPixel(BlackPixel(getXDisplay(), screen));
 208
 209     const ScreenInfo *screen_info = getScreenInfo(screen);
 210
 211     XSetWindowBackground(getXDisplay(), screen_info->getRootWindow(),
 212                          c.getPixel());
 213     XClearWindow(getXDisplay(), screen_info->getRootWindow());
 214
 215     Pixmap pixmap = XCreatePixmap(getXDisplay(),
 216           screen_info->getRootWindow(),
 217           8, 8, DefaultDepth(getXDisplay(), screen));
 218     XFillRectangle(getXDisplay(), pixmap, DefaultGC(getXDisplay(), screen),
 219                    0, 0, 8, 8);
 220
 221     setPixmapProperty(screen, duplicatePixmap(screen, pixmap, 8, 8));
 222
 223     XFreePixmap(getXDisplay(), pixmap);
 224   }
 225 }
 226
 227
 228 void bsetroot::modula(int x, int y) {
 229   char data[32];
 230   long pattern;
 231
 232   unsigned int screen, i;
 233
 234   for (pattern = 0, screen = 0; screen < getNumberOfScreens(); screen++) {
 235     for (i = 0; i < 16; i++) {
 236       pattern <<= 1;
 237       if ((i % x) == 0)
 238         pattern |= 0x0001;
 239     }
 240
 241     for (i = 0; i < 16; i++) {
 242       if ((i %  y) == 0) {
 243         data[(i * 2)] = (char) 0xff;
 244         data[(i * 2) + 1] = (char) 0xff;
 245       } else {
 246         data[(i * 2)] = pattern & 0xff;
 247         data[(i * 2) + 1] = (pattern >> 8) & 0xff;
 248       }
 249     }
 250
 251     BColor f, b;
 252     GC gc;
 253     Pixmap bitmap;
 254
 255     const ScreenInfo *screen_info = getScreenInfo(screen);
 256
 257     bitmap =
 258       XCreateBitmapFromData(getXDisplay(),
 259                             screen_info->getRootWindow(), data,
 260                             16, 16);
 261
 262     img_ctrl[screen]->parseColor(&f, fore);
 263     img_ctrl[screen]->parseColor(&b, back);
 264
 265     if (! f.isAllocated()) f.setPixel(WhitePixel(getXDisplay(), screen));
 266     if (! b.isAllocated()) b.setPixel(BlackPixel(getXDisplay(), screen));
 267
 268     XGCValues gcv;
 269     gcv.foreground = f.getPixel();
 270     gcv.background = b.getPixel();
 271
 272     gc = XCreateGC(getXDisplay(), screen_info->getRootWindow(),
 273                    GCForeground | GCBackground, &gcv);
 274
 275     Pixmap pixmap = XCreatePixmap(getXDisplay(),
 276           screen_info->getRootWindow(),
 277           16, 16, screen_info->getDepth());
 278
 279     XCopyPlane(getXDisplay(), bitmap, pixmap, gc,
 280                0, 0, 16, 16, 0, 0, 1l);
 281     XSetWindowBackgroundPixmap(getXDisplay(),
 282                                screen_info->getRootWindow(),
 283                                pixmap);
 284     XClearWindow(getXDisplay(), screen_info->getRootWindow());
 285
 286     setPixmapProperty(screen,
 287           duplicatePixmap(screen, pixmap, 16, 16));
 288
 289     XFreeGC(getXDisplay(), gc);
 290     XFreePixmap(getXDisplay(), bitmap);
 291
 292     if (! (screen_info->getVisual()->c_class & 1))
 293       XFreePixmap(getXDisplay(), pixmap);
 294   }
 295 }
 296
 297
 298 void bsetroot::gradient(void) {
 299   for (unsigned int screen = 0; screen < getNumberOfScreens(); screen++) {
 300     BTexture texture;
 301     img_ctrl[screen]->parseTexture(&texture, grad);
 302     img_ctrl[screen]->parseColor(texture.getColor(), fore);
 303     img_ctrl[screen]->parseColor(texture.getColorTo(), back);
 304     const ScreenInfo *screen_info = getScreenInfo(screen);
 305
 306     if (! texture.getColor()->isAllocated())
 307       texture.getColor()->setPixel(WhitePixel(getXDisplay(), screen));
 308     if (! texture.getColorTo()->isAllocated())
 309       texture.getColorTo()->setPixel(BlackPixel(getXDisplay(), screen));
 310
 311     Pixmap pixmap =
 312       img_ctrl[screen]->renderImage(screen_info->size().w(),
 313                                     screen_info->size().h(),
 314                                     &texture);
 315
 316     XSetWindowBackgroundPixmap(getXDisplay(),
 317                                screen_info->getRootWindow(),
 318                                pixmap);
 319     XClearWindow(getXDisplay(), screen_info->getRootWindow());
 320
 321     setPixmapProperty(screen,
 322           duplicatePixmap(screen, pixmap,
 323               screen_info->size().w(),
 324               screen_info->size().h()));
 325
 326     if (! (screen_info->getVisual()->c_class & 1)) {
 327       img_ctrl[screen]->removeImage(pixmap);
 328     }
 329   }
 330 }
 331
 332
 333 void bsetroot::usage(int exit_code) {
 334     fprintf(stderr,
 335       i18n->getMessage(bsetrootSet, bsetrootUsage,
 336      "%s 2.0\n\n"
 337      "Copyright (c) 1997-2000, 2002 Bradley T Hughes\n"
 338      "Copyright (c) 2001-2002 Sean 'Shaleh' Perry\n\n"
 339      "  -display <string>        display connection\n"
 340      "  -mod <x> <y>             modula pattern\n"
 341      "  -foreground, -fg <color> modula foreground color\n"
 342      "  -background, -bg <color> modula background color\n\n"
 343      "  -gradient <texture>      gradient texture\n"
 344      "  -from <color>            gradient start color\n"
 345      "  -to <color>              gradient end color\n\n"
 346      "  -solid <color>           solid color\n\n"
 347      "  -help                    print this help text and exit\n"),
 348       getApplicationName());
 349
 350     exit(exit_code);
 351 }
 352
 353 int main(int argc, char **argv) {
 354   char *display_name = (char *) 0;
 355
 356   i18n->openCatalog("blackbox.cat");
 357
 358   for (int i = 1; i < argc; i++) {
 359     if (! strcmp(argv[i], "-display")) {
 360       // check for -display option
 361
 362       if ((++i) >= argc) {
 363         fprintf(stderr, i18n->getMessage(mainSet, mainDISPLAYRequiresArg,
 364                  "error: '-display' requires an argument\n"));
 365
 366         ::exit(1);
 367       }
 368
 369       display_name = argv[i];
 370     }
 371   }
 372
 373   bsetroot app(argc, argv, display_name);
 374
 375   return 0;
 376 }
 377