Handle dynamic desktops by default
[pager.git] / pager.c
1 /* gcc -o pager pager.c -lX11 -lXmu -ldockapp */
2 #include <X11/Xatom.h>
3 #include <dockapp.h>
4 #include <string.h>
5
6 #define DOCKAPP_SIZE 56
7 #define SPEED 100
8 #ifndef FromTool
9 #define FromTool 2
10 #endif
11
12 static GC dockapp_border1GC, dockapp_border2GC, window_border1GC, window_border2GC, active_desktopGC, inactive_desktopGC, active_windowGC, inactive_windowGC;
13 static Atom supported_atom, num_desktops_atom, desktop_geometry_atom, current_desktop_atom, client_list_atom, client_desktop_atom, client_state_atom, active_window_atom, moveresize_window_atom;
14 static Atom above_state, below_state, shaded_state, skip_pager_state, hidden_state;
15 static int desktop = -1;
16 static Window drag_window;
17 static int drag_x, drag_y;
18 static int drag_button;
19 static int dockapp_x, dockapp_y;
20 static unsigned int dockapp_width, dockapp_height, dockapp_border, dockapp_depth;
21 static double scale;
22
23 static int desktop_geometry_supported;
24 static int current_desktop_supported;
25 static int active_window_supported;
26 static int moveresize_supported;
27
28 typedef struct {
29   Window *window;
30   DARect rect;
31 } client_t;
32
33 int error_handler(Display *display, XErrorEvent *event) {
34   char buffer[512];
35
36   switch (event->error_code) {
37     case BadWindow:
38       /* The window may have gone away since we queried the window manager. */
39     break;
40
41     default:
42       if (XGetErrorText(display, event->error_code, buffer, sizeof(buffer))) {
43         fprintf(stderr, "%s\n", buffer);
44         exit(100);
45       }
46     break;
47   }
48
49   return 0;
50 }
51
52 void setup_atom(Atom *atom, const char *prop) {
53   *atom = XInternAtom(DADisplay, prop, True);
54   if (*atom != None) return;
55   fprintf(stderr, "Unknown atom %s!\n", prop);
56   exit(111);
57 }
58
59 void get_atom_longs(Atom atom, Atom type, Window window, long **data, unsigned long *num_items) {
60   Atom actual;
61   int format;
62   unsigned long num_bytes;
63
64   XGetWindowProperty(DADisplay, window, atom, 0, 8192, False, type, &actual, &format, num_items, &num_bytes, (unsigned char **) data);
65 }
66
67 long get_atom_long(Atom atom, Atom type, Window window) {
68   long *data;
69   long ret = 0;
70   unsigned long num_items;
71
72   get_atom_longs(atom, type, window, &data, &num_items);
73   if (num_items) ret = data[0];
74   XFree(data);
75
76   return ret;
77 }
78
79 void check_support() {
80   long *data;
81   unsigned long i, num_items;
82
83   get_atom_longs(supported_atom, XA_ATOM, DefaultRootWindow(DADisplay), &data, &num_items);
84   for (i = 0; i < num_items; i++) {
85     if (data[i] == desktop_geometry_atom) desktop_geometry_supported = 1;
86     if (data[i] == current_desktop_atom) current_desktop_supported = 1;
87     if (data[i] == active_window_atom) active_window_supported = 1;
88     if (data[i] == moveresize_window_atom) moveresize_supported = 1;
89   }
90
91   XFree(data);
92 }
93
94 void enumerate_clients(client_t ***clients, unsigned long *num_clients, unsigned long *active_desktop, unsigned long *active_window) {
95   Window root, dockapp_root, child;
96   int x, y;
97   unsigned int width, height, border, depth;
98   unsigned int root_width, root_height;
99   long *data;
100   long *states;
101   unsigned long i, j, num_states;
102   long client_desktop;
103   int draw;
104
105   if (desktop_geometry_supported) {
106     get_atom_longs(desktop_geometry_atom, XA_CARDINAL, DefaultRootWindow(DADisplay), &states, &num_states);
107     root_width = states[0];
108     root_height = states[0];
109     XFree(states);
110   }
111   else {
112     root_width = DisplayWidth(DADisplay, DefaultScreen(DADisplay));
113     root_height = DisplayWidth(DADisplay, DefaultScreen(DADisplay));
114   }
115
116   XGetGeometry(DADisplay, DAWindow, &dockapp_root, &dockapp_x, &dockapp_y, &dockapp_width, &dockapp_height, &dockapp_border, &dockapp_depth);
117   scale = (double) root_width / (double) dockapp_width;
118
119   *active_window = get_atom_long(active_window_atom, XA_WINDOW, dockapp_root);
120   *active_desktop = get_atom_long(current_desktop_atom, XA_CARDINAL, DefaultRootWindow(DADisplay));
121
122   get_atom_longs(client_list_atom, XA_WINDOW, dockapp_root, &data, num_clients);
123
124   *clients = (client_t **) malloc(sizeof(client_t *) * *num_clients);
125   if (! *clients) {
126     fprintf(stderr, "Out of memory for clients!\n");
127     exit(111);
128   }
129
130   for (i = 0; i < *num_clients; i++) {
131     (*clients)[i] = (client_t *) malloc(sizeof(client_t));
132     if (! (*clients)[i]) {
133       fprintf(stderr, "Out of memory for client!\n");
134       exit(111);
135     }
136     (*clients)[i]->window = 0;
137
138     /* Check the window is on our desktop (or all desktops). */
139     client_desktop = get_atom_long(client_desktop_atom, XA_CARDINAL, data[i]);
140     if (desktop > -1) {
141       if (client_desktop != desktop && client_desktop != -1) continue;
142     }
143     else {
144       if (client_desktop != *active_desktop && client_desktop != -1) continue;
145     }
146
147     /* Try to get its dimensions. */
148     if (! XGetGeometry(DADisplay, data[i], &root, &x, &y, &width, &height, &border, &depth)) continue;
149     if (! XTranslateCoordinates(DADisplay, data[i], root, x, y, &x, &y, &child)) continue;
150
151     /* Make sure it isn't hidden or shaded. */
152     draw = 1;
153     get_atom_longs(client_state_atom, XA_ATOM, data[i], &states, &num_states);
154     for (j = 0; j < num_states; j++) {
155       if (states[j] == shaded_state) draw = 0;
156       if (states[j] == skip_pager_state) draw = 0;
157       if (states[j] == hidden_state) draw = 0;
158     }
159     if (! draw) continue;
160
161     (*clients)[i]->window = data[i];
162
163     /* Scale it. */
164     (*clients)[i]->rect.x = (double) x / scale;
165     (*clients)[i]->rect.y = (double) y / scale;
166     (*clients)[i]->rect.width = (double) width / scale;
167     (*clients)[i]->rect.height = (double) height / scale;
168   }
169
170   XFree(data);
171 }
172
173 void destroy_clients(client_t ***clients, unsigned long num_clients) {
174   unsigned long i;
175
176   if (! *clients) return;
177
178   for (i = 0; i < num_clients; i++) free((*clients)[i]);
179   free(*clients);
180 }
181
182 void setup_GCs() {
183   XGCValues gcv;
184
185   /* GC's */
186   gcv.foreground = DAGetColor("darkGray");
187   XChangeGC(DADisplay, DAClearGC, GCForeground, &gcv);
188
189   gcv.foreground = DAGetColor("lightGray");
190   gcv.graphics_exposures = False;
191
192   dockapp_border1GC     = XCreateGC(DADisplay, DAWindow, GCForeground|GCGraphicsExposures, &gcv);
193
194   gcv.foreground = DAGetColor("#222222");
195   dockapp_border2GC     =  XCreateGC(DADisplay, DAWindow, GCForeground|GCGraphicsExposures, &gcv);
196
197   gcv.foreground = DAGetColor("#0088ff");
198   active_desktopGC = XCreateGC(DADisplay, DAWindow, GCForeground|GCGraphicsExposures, &gcv);
199
200   gcv.foreground = DAGetColor("darkGray");
201   inactive_desktopGC = XCreateGC(DADisplay, DAWindow, GCForeground|GCGraphicsExposures, &gcv);
202
203   gcv.foreground = DAGetColor("white");
204   active_windowGC = XCreateGC(DADisplay, DAWindow, GCForeground|GCGraphicsExposures, &gcv);
205
206   gcv.foreground = DAGetColor("#999999");
207   inactive_windowGC = XCreateGC(DADisplay, DAWindow, GCForeground|GCGraphicsExposures, &gcv);
208
209   gcv.foreground = DAGetColor("black");
210   window_border1GC      =  XCreateGC(DADisplay, DAWindow, GCForeground|GCGraphicsExposures, &gcv);
211
212   gcv.foreground = DAGetColor("#333333");
213   window_border2GC      =  XCreateGC(DADisplay, DAWindow, GCForeground|GCGraphicsExposures, &gcv);
214 }
215
216 void draw_relief(Pixmap pixmap, unsigned int width, unsigned int height, GC desktopGC) {
217   /* Drawing */
218   XFillRectangle(DADisplay, pixmap, desktopGC, 0, 0, width, height);
219
220   XDrawLine(DADisplay, pixmap, dockapp_border2GC, 0, 0, 0, height - 2);
221   XDrawLine(DADisplay, pixmap, dockapp_border2GC, 1, 0, width - 1, 0);
222
223   XDrawLine(DADisplay, pixmap, dockapp_border1GC, 0, height - 1, width - 1, height - 1);
224   XDrawLine(DADisplay, pixmap, dockapp_border1GC, width - 1, 1, width - 1, height - 2);
225 }
226
227 int clicked(DARect *rect, int x, int y) {
228   if (rect->x > x) return 0;
229   if (rect->y > y) return 0;
230   if (rect->x + rect->width < x) return 0;
231   if (rect->y + rect->height < y) return 0;
232   return 1;
233 }
234
235 int client_message(Atom message_type, Window window, long data0, long data1, long data2, long data3, long data4) {
236   XEvent event;
237
238   memset(&event, 0, sizeof(event));
239   event.type = ClientMessage;
240   event.xclient.type = ClientMessage;
241   event.xclient.send_event = True;
242   event.xclient.window = window;
243   event.xclient.message_type = message_type;
244   event.xclient.format = 32;
245   event.xclient.data.l[0] = data0;
246   event.xclient.data.l[1] = data1;
247   event.xclient.data.l[2] = data2;
248   event.xclient.data.l[3] = data3;
249   event.xclient.data.l[4] = data4;
250
251   return XSendEvent(DADisplay, DefaultRootWindow(DADisplay), False, SubstructureRedirectMask | SubstructureNotifyMask, &event);
252 }
253
254 void change(int button, int state, int x, int y) {
255   client_t **clients;
256   unsigned long i, num_clients;
257   unsigned long active_desktop, active_window;
258
259   enumerate_clients(&clients, &num_clients, &active_desktop, &active_window);
260   for (i = num_clients - 1; i > 0; i--) {
261     if (! clients[i]->window) continue;
262     if (! clicked(&clients[i]->rect, x, y)) continue;
263
264     drag_window = (Window) clients[i]->window;
265     drag_button = button;
266     drag_x = x - clients[i]->rect.x;
267     drag_y = y - clients[i]->rect.y;
268     break;
269   }
270   destroy_clients(&clients, num_clients);
271
272   if (drag_window && button == 1) {
273     client_message(active_window_atom, drag_window, FromTool, CurrentTime, 0, 0 ,0);
274   }
275   else {
276     client_message(current_desktop_atom, DefaultRootWindow(DADisplay), desktop, 0, 0, 0, 0);
277   }
278 }
279
280 void release(int button, int state, int x, int y) {
281   if (button == drag_button) drag_window = 0;
282 }
283
284 void page() {
285   Window active_window;
286   Pixmap pixmap;
287   long i;
288   unsigned long num_clients;
289   unsigned long active_desktop;
290   GC gc;
291   client_t **clients;
292
293   enumerate_clients(&clients, &num_clients, &active_desktop, &active_window);
294
295   pixmap = XCreatePixmap(DADisplay, DAWindow, dockapp_width, dockapp_height, dockapp_depth);
296   if (active_desktop == desktop || desktop < 0) gc = active_desktopGC;
297   else gc = inactive_desktopGC;
298   draw_relief(pixmap, dockapp_width, dockapp_height, gc);
299
300   for (i = 0; i < num_clients; i++) {
301     if (! clients[i]->window) continue;
302
303     /* Draw windows which are partially off-screen. */
304     if (clients[i]->rect.x < 0) {
305       clients[i]->rect.width += clients[i]->rect.x;
306       clients[i]->rect.x = 0;
307     }
308     if (clients[i]->rect.y < 0) {
309       clients[i]->rect.height += clients[i]->rect.y;
310       clients[i]->rect.y = 0;
311     }
312
313     /* Draw small windows. */
314     if (clients[i]->rect.width < 2 || clients[i]->rect.height < 2) gc = window_border1GC;
315     else if ((long) clients[i]->window == active_window) gc = active_windowGC;
316     else gc = inactive_windowGC;
317
318     XFillRectangle(DADisplay, pixmap, gc, clients[i]->rect.x, clients[i]->rect.y, clients[i]->rect.width, clients[i]->rect.height);
319     if (clients[i]->rect.width < 2 || clients[i]->rect.height < 2) continue;
320
321     XDrawLine(DADisplay, pixmap, window_border1GC, clients[i]->rect.x, clients[i]->rect.y, clients[i]->rect.x, clients[i]->rect.y + clients[i]->rect.height - 2);
322     XDrawLine(DADisplay, pixmap, window_border1GC, clients[i]->rect.x + 1, clients[i]->rect.y, clients[i]->rect.x + clients[i]->rect.width - 1, clients[i]->rect.y);
323     XDrawLine(DADisplay, pixmap, window_border2GC, clients[i]->rect.x, clients[i]->rect.y + clients[i]->rect.height - 1, clients[i]->rect.x + clients[i]->rect.width - 1, clients[i]->rect.y + clients[i]->rect.height - 1);
324     XDrawLine(DADisplay, pixmap, window_border2GC, clients[i]->rect.x + clients[i]->rect.width - 1, clients[i]->rect.y + 1, clients[i]->rect.x + clients[i]->rect.width - 1, clients[i]->rect.y + clients[i]->rect.height - 2);
325   }
326
327   destroy_clients(&clients, num_clients);
328
329   DASetPixmap(pixmap);
330   XFreePixmap(DADisplay, pixmap);
331 }
332
333 void move(int x, int y) {
334   long dest_x, dest_y;
335
336   if (! drag_window) return;
337   if (x < 0 || y < 0) return;
338   if (x >= dockapp_width || y >= dockapp_height) return;
339
340   dest_x = (long) ((double) (x - drag_x) * scale);
341   dest_y = (long) ((double) (y - drag_y) * scale);
342
343   if (moveresize_supported) {
344     client_message(moveresize_window_atom, drag_window, StaticGravity | (3 << 8) | (FromTool << 12), drag_x, drag_y, 0, 0);
345   }
346   else {
347     XMoveWindow(DADisplay, drag_window, dest_x, dest_y);
348   }
349
350   page();
351 }
352
353 int main(int argc, char **argv) {
354   Display *display;
355   unsigned int root_width, root_height;
356   long num_desktops;
357   double aspect;
358   DACallbacks callbacks = { 0, change, release, move, 0, 0, page };
359   DAProgramOption options = { "-d", "--desktop", "Desktop to page", DOInteger, 0, { (int *) &desktop } };
360
361   DASetExpectedVersion(20020126);
362
363   display = XOpenDisplay(0);
364
365   root_width = DisplayWidth(display, DefaultScreen(display));
366   root_height = DisplayHeight(display, DefaultScreen(display));
367   aspect = (double) root_width / (double) root_height;
368
369   if (root_width > root_height) {
370     dockapp_width = DOCKAPP_SIZE;
371     dockapp_height = (double) DOCKAPP_SIZE / aspect;
372   }
373   else {
374     dockapp_height = DOCKAPP_SIZE;
375     dockapp_width = (double) DOCKAPP_SIZE / aspect;
376   }
377
378   DAParseArguments(argc, argv, &options, 1, "OPTIONS", "pager");
379   DAOpenDisplay(0, argc, argv);
380   DACreateIcon("pager", dockapp_width, dockapp_height, argc, argv);
381
382   DASetCallbacks(&callbacks);
383   DASetTimeout(SPEED);
384
385   setup_atom(&supported_atom, "_NET_SUPPORTED");
386   setup_atom(&num_desktops_atom, "_NET_NUMBER_OF_DESKTOPS");
387   setup_atom(&desktop_geometry_atom, "_NET_DESKTOP_GEOMETRY");
388   setup_atom(&current_desktop_atom, "_NET_CURRENT_DESKTOP");
389   setup_atom(&client_list_atom, "_NET_CLIENT_LIST_STACKING");
390   setup_atom(&client_desktop_atom, "_NET_WM_DESKTOP");
391   setup_atom(&client_state_atom, "_NET_WM_STATE");
392   setup_atom(&active_window_atom, "_NET_ACTIVE_WINDOW");
393   setup_atom(&moveresize_window_atom, "_NET_MOVERESIZE_WINDOW");
394
395   setup_atom(&above_state, "_NET_WM_STATE_ABOVE");
396   setup_atom(&below_state, "_NET_WM_STATE_BELOW");
397   setup_atom(&shaded_state, "_NET_WM_STATE_SHADED");
398   setup_atom(&skip_pager_state, "_NET_WM_STATE_SKIP_PAGER");
399   setup_atom(&hidden_state, "_NET_WM_STATE_HIDDEN");
400
401   check_support();
402
403   num_desktops = get_atom_long(num_desktops_atom, XA_CARDINAL, DefaultRootWindow(DADisplay));
404   if (desktop == 0) {
405     desktop = get_atom_long(current_desktop_atom, XA_CARDINAL, DefaultRootWindow(DADisplay));
406   }
407   else if (desktop < 0) desktop = -1;
408   else if (desktop-- > num_desktops) {
409     fprintf(stderr, "There are only %ld desktops!\n", num_desktops);
410     exit(111);
411   }
412
413   XSetErrorHandler(error_handler);
414   setup_GCs();
415   DAShow();
416   DAEventLoop();
417
418   exit(0);
419 }