Dogcows Code - chaz/openbox/blob - tests/icons.c

   1 /* -*- indent-tabs-mode: nil; tab-width: 4; c-basic-offset: 4; -*-
   2
   3 icons.c for the Openbox window manager
   4 Copyright (c) 2003        Ben Jansens
   5
   6 This program is free software; you can redistribute it and/or modify
   7 it under the terms of the GNU General Public License as published by
   8 the Free Software Foundation; either version 2 of the License, or
   9 (at your option) any later version.
  10
  11 This program is distributed in the hope that it will be useful,
  12 but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 GNU General Public License for more details.
  15
  16 See the COPYING file for a copy of the GNU General Public License.
  17 */
  18
  19 #include <X11/Xlib.h>
  20 #include <X11/Xutil.h>
  21 #include <X11/Xatom.h>
  22 #include <X11/cursorfont.h>
  23 #include <stdlib.h>
  24 #include <stdio.h>
  25 #include <assert.h>
  26
  27 Window findClient(Display *d, Window win)
  28 {
  29     Window r, *children;
  30     unsigned int n, i;
  31     Atom state = XInternAtom(d, "WM_STATE", True);
  32     Atom ret_type;
  33     int ret_format;
  34     unsigned long ret_items, ret_bytesleft;
  35     unsigned long *prop_return;
  36
  37     XQueryTree(d, win, &r, &r, &children, &n);
  38     for (i = 0; i < n; ++i) {
  39         Window w = findClient(d, children[i]);
  40         if (w) return w;
  41     }
  42
  43     // try me
  44     XGetWindowProperty(d, win, state, 0, 1,
  45                        False, state, &ret_type, &ret_format,
  46                        &ret_items, &ret_bytesleft,
  47                        (unsigned char**) &prop_return);
  48     if (ret_type == None || ret_items < 1)
  49         return None;
  50     return win; // found it!
  51 }
  52
  53 int main(int argc, char **argv)
  54 {
  55     Display *d = XOpenDisplay(NULL);
  56     int s = DefaultScreen(d);
  57     Atom net_wm_icon = XInternAtom(d, "_NET_WM_ICON", True);
  58     Atom ret_type;
  59     unsigned int winw = 0, winh = 0;
  60     int ret_format;
  61     unsigned long ret_items, ret_bytesleft;
  62     const int MAX_IMAGES = 10;
  63     unsigned long *prop_return[MAX_IMAGES];
  64     XImage *i[MAX_IMAGES];
  65     long offset = 0;
  66     unsigned int image = 0;
  67     unsigned int j; // loop counter
  68     Window id, win;
  69     Pixmap p;
  70     Cursor cur;
  71     XEvent ev;
  72
  73     printf("Click on a window with an icon...\n");
  74
  75     //int id = strtol(argv[1], NULL, 16);
  76     XUngrabPointer(d, CurrentTime);
  77     cur = XCreateFontCursor(d, XC_crosshair);
  78     XGrabPointer(d, RootWindow(d, s), False, ButtonPressMask, GrabModeAsync,
  79                  GrabModeAsync, None, cur, CurrentTime);
  80     XEvent ev;
  81     while (1) {
  82         XNextEvent(d, &ev);
  83         if (ev.type == ButtonPress) {
  84             XUngrabPointer(d, CurrentTime);
  85             id = findClient(d, ev.xbutton.subwindow);
  86             break;
  87         }
  88     }
  89
  90     printf("Using window 0x%lx\n", id);
  91
  92     do {
  93         unsigned int w, h;
  94
  95         XGetWindowProperty(d, id, net_wm_icon, offset++, 1,
  96                            False, XA_CARDINAL, &ret_type, &ret_format,
  97                            &ret_items, &ret_bytesleft,
  98                            (unsigned char**) &prop_return[image]);
  99         if (ret_type == None || ret_items < 1) {
 100             printf("No icon found\n");
 101             return 1;
 102         }
 103         w = prop_return[image][0];
 104         XFree(prop_return[image]);
 105
 106         XGetWindowProperty(d, id, net_wm_icon, offset++, 1,
 107                            False, XA_CARDINAL, &ret_type, &ret_format,
 108                            &ret_items, &ret_bytesleft,
 109                            (unsigned char**) &prop_return[image]);
 110         if (ret_type == None || ret_items < 1) {
 111             printf("Failed to get height\n");
 112             return 1;
 113         }
 114         h = prop_return[image][0];
 115         XFree(prop_return[image]);
 116
 117         XGetWindowProperty(d, id, net_wm_icon, offset, w*h,
 118                            False, XA_CARDINAL, &ret_type, &ret_format,
 119                            &ret_items, &ret_bytesleft,
 120                            (unsigned char**) &prop_return[image]);
 121         if (ret_type == None || ret_items < w*h) {
 122             printf("Failed to get image data\n");
 123             return 1;
 124         }
 125         offset += w*h;
 126
 127         printf("Found icon with size %dx%d\n", w, h);
 128
 129         i[image] = XCreateImage(d, DefaultVisual(d, s), DefaultDepth(d, s),
 130                                 ZPixmap, 0, NULL, w, h, 32, 0);
 131         assert(i[image]);
 132         i[image]->byte_order = LSBFirst;
 133         i[image]->data = (char*)prop_return[image];
 134         for (j = 0; j < w*h; j++) {
 135             unsigned char alpha = (unsigned char)i[image]->data[j*4+3];
 136             unsigned char r = (unsigned char) i[image]->data[j*4+0];
 137             unsigned char g = (unsigned char) i[image]->data[j*4+1];
 138             unsigned char b = (unsigned char) i[image]->data[j*4+2];
 139
 140             // background color
 141             unsigned char bgr = 0;
 142             unsigned char bgg = 0;
 143             unsigned char bgb = 0;
 144
 145             r = bgr + (r - bgr) * alpha / 256;
 146             g = bgg + (g - bgg) * alpha / 256;
 147             b = bgb + (b - bgb) * alpha / 256;
 148
 149             i[image]->data[j*4+0] = (char) r;
 150             i[image]->data[j*4+1] = (char) g;
 151             i[image]->data[j*4+2] = (char) b;
 152         }
 153
 154         winw += w;
 155         if (h > winh) winh = h;
 156
 157         ++image;
 158     } while (ret_bytesleft > 0 && image < MAX_IMAGES);
 159
 160     win = XCreateSimpleWindow(d, RootWindow(d, s), 0, 0, winw, winh,
 161                               0, 0, 0);
 162     assert(win);
 163     XMapWindow(d, win);
 164
 165     p = XCreatePixmap(d, win, winw, winh, DefaultDepth(d, s));
 166     XFillRectangle(d, p, DefaultGC(d, s), 0, 0, winw, winh);
 167
 168     for (j = 0; j < image; ++j) {
 169         static unsigned int x = 0;
 170
 171         XPutImage(d, p, DefaultGC(d, s), i[j], 0, 0, x, 0,
 172                   i[j]->width, i[j]->height);
 173         x += i[j]->width;
 174         XDestroyImage(i[j]);
 175     }
 176
 177     XSetWindowBackgroundPixmap(d, win, p);
 178     XClearWindow(d, win);
 179
 180     XFlush(d);
 181
 182     getchar();
 183
 184     XFreePixmap(d, p);
 185     XCloseDisplay(d);
 186 }