better setting of atoms. plugged memory leak.
[mikachu/openbox.git] / 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