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*[numberOfScreens()];
 111   for (unsigned int s = 0, n = numberOfScreens(); s < n; ++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 + numberOfScreens(), 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   const ScreenInfo *screen_info = getScreenInfo(screen);
 140
 141   if (rootpmap_id == None) {
 142     rootpmap_id = XInternAtom(getXDisplay(), "_XROOTPMAP_ID", False);
 143     esetroot_id = XInternAtom(getXDisplay(), "ESETROOT_PMAP_ID", False);
 144   }
 145
 146   XGrabServer(getXDisplay());
 147
 148   // Clear out the old pixmap
 149   XGetWindowProperty(getXDisplay(), screen_info->getRootWindow(),
 150                      rootpmap_id, 0L, 1L, False, XA_PIXMAP,
 151                      &type, &format, &length, &after, &data);
 152   if (type == XA_PIXMAP && format == 32) {
 153     XKillClient(getXDisplay(), *((Pixmap *) data));
 154     XSync(getXDisplay(), False);
 155     XFree(data);
 156   }
 157
 158   if (pixmap) {
 159     XChangeProperty(getXDisplay(), screen_info->getRootWindow(),
 160         rootpmap_id, XA_PIXMAP, 32, PropModeReplace,
 161         (unsigned char *) &pixmap, 1);
 162     XChangeProperty(getXDisplay(), screen_info->getRootWindow(),
 163         esetroot_id, XA_PIXMAP, 32, PropModeReplace,
 164         (unsigned char *) &pixmap, 1);
 165   } else {
 166     XDeleteProperty(getXDisplay(), screen_info->getRootWindow(),
 167         rootpmap_id);
 168     XDeleteProperty(getXDisplay(), screen_info->getRootWindow(),
 169         esetroot_id);
 170   }
 171
 172   XUngrabServer(getXDisplay());
 173   XFlush(getXDisplay());
 174 }
 175
 176
 177 // adapted from wmsetbg
 178 Pixmap bsetroot::duplicatePixmap(int screen, Pixmap pixmap,
 179          int width, int height) {
 180   XSync(getXDisplay(), False);
 181
 182   Pixmap copyP = XCreatePixmap(getXDisplay(),
 183              getScreenInfo(screen)->getRootWindow(),
 184              width, height,
 185              DefaultDepth(getXDisplay(), screen));
 186   XCopyArea(getXDisplay(), pixmap, copyP, DefaultGC(getXDisplay(), screen),
 187       0, 0, width, height, 0, 0);
 188   XSync(getXDisplay(), False);
 189
 190   return copyP;
 191 }
 192
 193
 194 void bsetroot::solid(void) {
 195   for (unsigned int screen = 0, n = numberOfScreens(); screen < n; screen++) {
 196     BColor c;
 197
 198     img_ctrl[screen]->parseColor(&c, fore);
 199     if (! c.isAllocated()) c.setPixel(BlackPixel(getXDisplay(), screen));
 200
 201     const ScreenInfo *screen_info = getScreenInfo(screen);
 202
 203     XSetWindowBackground(getXDisplay(), screen_info->getRootWindow(),
 204                          c.getPixel());
 205     XClearWindow(getXDisplay(), screen_info->getRootWindow());
 206
 207     Pixmap pixmap = XCreatePixmap(getXDisplay(),
 208           screen_info->getRootWindow(),
 209           8, 8, DefaultDepth(getXDisplay(), screen));
 210
 211     XSetForeground(getXDisplay(), DefaultGC(getXDisplay(), screen),
 212                    c.getPixel());
 213     XFillRectangle(getXDisplay(), pixmap, DefaultGC(getXDisplay(), screen),
 214                    0, 0, 8, 8);
 215
 216     setPixmapProperty(screen, duplicatePixmap(screen, pixmap, 8, 8));
 217
 218     XFreePixmap(getXDisplay(), pixmap);
 219   }
 220 }
 221
 222
 223 void bsetroot::modula(int x, int y) {
 224   char data[32];
 225   long pattern;
 226
 227   unsigned int screen, i;
 228   unsigned int n = numberOfScreens();
 229
 230   for (pattern = 0, screen = 0; screen < n; screen++) {
 231     for (i = 0; i < 16; i++) {
 232       pattern <<= 1;
 233       if ((i % x) == 0)
 234         pattern |= 0x0001;
 235     }
 236
 237     for (i = 0; i < 16; i++) {
 238       if ((i %  y) == 0) {
 239         data[(i * 2)] = (char) 0xff;
 240         data[(i * 2) + 1] = (char) 0xff;
 241       } else {
 242         data[(i * 2)] = pattern & 0xff;
 243         data[(i * 2) + 1] = (pattern >> 8) & 0xff;
 244       }
 245     }
 246
 247     BColor f, b;
 248     GC gc;
 249     Pixmap bitmap;
 250
 251     const ScreenInfo *screen_info = getScreenInfo(screen);
 252
 253     bitmap =
 254       XCreateBitmapFromData(getXDisplay(),
 255                             screen_info->getRootWindow(), data,
 256                             16, 16);
 257
 258     img_ctrl[screen]->parseColor(&f, fore);
 259     img_ctrl[screen]->parseColor(&b, back);
 260
 261     if (! f.isAllocated()) f.setPixel(WhitePixel(getXDisplay(), screen));
 262     if (! b.isAllocated()) b.setPixel(BlackPixel(getXDisplay(), screen));
 263
 264     XGCValues gcv;
 265     gcv.foreground = f.getPixel();
 266     gcv.background = b.getPixel();
 267
 268     gc = XCreateGC(getXDisplay(), screen_info->getRootWindow(),
 269                    GCForeground | GCBackground, &gcv);
 270
 271     Pixmap pixmap = XCreatePixmap(getXDisplay(),
 272           screen_info->getRootWindow(),
 273           16, 16, screen_info->getDepth());
 274
 275     XCopyPlane(getXDisplay(), bitmap, pixmap, gc,
 276                0, 0, 16, 16, 0, 0, 1l);
 277     XSetWindowBackgroundPixmap(getXDisplay(),
 278                                screen_info->getRootWindow(),
 279                                pixmap);
 280     XClearWindow(getXDisplay(), screen_info->getRootWindow());
 281
 282     setPixmapProperty(screen,
 283           duplicatePixmap(screen, pixmap, 16, 16));
 284
 285     XFreeGC(getXDisplay(), gc);
 286     XFreePixmap(getXDisplay(), bitmap);
 287
 288     if (! (screen_info->getVisual()->c_class & 1))
 289       XFreePixmap(getXDisplay(), pixmap);
 290   }
 291 }
 292
 293
 294 void bsetroot::gradient(void) {
 295   for (unsigned int screen = 0, n = numberOfScreens(); screen < n; screen++) {
 296     BTexture texture;
 297     img_ctrl[screen]->parseTexture(&texture, grad);
 298     img_ctrl[screen]->parseColor(texture.getColor(), fore);
 299     img_ctrl[screen]->parseColor(texture.getColorTo(), back);
 300     const ScreenInfo *screen_info = getScreenInfo(screen);
 301
 302     if (! texture.getColor()->isAllocated())
 303       texture.getColor()->setPixel(WhitePixel(getXDisplay(), screen));
 304     if (! texture.getColorTo()->isAllocated())
 305       texture.getColorTo()->setPixel(BlackPixel(getXDisplay(), screen));
 306
 307     Pixmap pixmap =
 308       img_ctrl[screen]->renderImage(screen_info->size().w(),
 309                                     screen_info->size().h(),
 310                                     &texture);
 311
 312     XSetWindowBackgroundPixmap(getXDisplay(),
 313                                screen_info->getRootWindow(),
 314                                pixmap);
 315     XClearWindow(getXDisplay(), screen_info->getRootWindow());
 316
 317     setPixmapProperty(screen,
 318           duplicatePixmap(screen, pixmap,
 319               screen_info->size().w(),
 320               screen_info->size().h()));
 321
 322     if (! (screen_info->getVisual()->c_class & 1)) {
 323       img_ctrl[screen]->removeImage(pixmap);
 324     }
 325   }
 326 }
 327
 328
 329 void bsetroot::usage(int exit_code) {
 330     fprintf(stderr,
 331       i18n->getMessage(bsetrootSet, bsetrootUsage,
 332      "%s 2.0\n\n"
 333      "Copyright (c) 1997-2000, 2002 Bradley T Hughes\n"
 334      "Copyright (c) 2001-2002 Sean 'Shaleh' Perry\n\n"
 335      "  -display <string>        display connection\n"
 336      "  -mod <x> <y>             modula pattern\n"
 337      "  -foreground, -fg <color> modula foreground color\n"
 338      "  -background, -bg <color> modula background color\n\n"
 339      "  -gradient <texture>      gradient texture\n"
 340      "  -from <color>            gradient start color\n"
 341      "  -to <color>              gradient end color\n\n"
 342      "  -solid <color>           solid color\n\n"
 343      "  -help                    print this help text and exit\n"),
 344       getApplicationName());
 345
 346     exit(exit_code);
 347 }
 348
 349 int main(int argc, char **argv) {
 350   char *display_name = (char *) 0;
 351
 352   NLSInit("openbox.cat");
 353
 354   for (int i = 1; i < argc; i++) {
 355     if (! strcmp(argv[i], "-display")) {
 356       // check for -display option
 357
 358       if ((++i) >= argc) {
 359         fprintf(stderr, i18n->getMessage(mainSet, mainDISPLAYRequiresArg,
 360                  "error: '-display' requires an argument\n"));
 361
 362         ::exit(1);
 363       }
 364
 365       display_name = argv[i];
 366     }
 367   }
 368
 369   bsetroot app(argc, argv, display_name);
 370
 371   return 0;
 372 }
 373