main.c (view raw)
1/* (C)opyright MMVI-MMVII Anselm R. Garbe <garbeam at gmail dot com>
2 * See LICENSE file for license details.
3 */
4
5#include "dwm.h"
6#include <errno.h>
7#include <locale.h>
8#include <stdio.h>
9#include <stdlib.h>
10#include <string.h>
11#include <unistd.h>
12#include <sys/select.h>
13#include <X11/cursorfont.h>
14#include <X11/keysym.h>
15#include <X11/Xatom.h>
16#include <X11/Xproto.h>
17
18/* extern */
19
20char stext[256];
21int screen, sx, sy, sw, sh, wax, way, waw, wah;
22unsigned int bh, ntags, numlockmask;
23Atom wmatom[WMLast], netatom[NetLast];
24Bool running = True;
25Bool *seltag;
26Bool selscreen = True;
27Client *clients = NULL;
28Client *sel = NULL;
29Client *stack = NULL;
30Cursor cursor[CurLast];
31Display *dpy;
32DC dc = {0};
33Window root, barwin;
34
35/* static */
36
37static int (*xerrorxlib)(Display *, XErrorEvent *);
38static Bool otherwm, readin;
39
40static void
41cleanup(void) {
42 close(STDIN_FILENO);
43 while(stack) {
44 if(stack->isbanned)
45 XMoveWindow(dpy, stack->win, stack->x, stack->y);
46 unmanage(stack);
47 }
48 if(dc.font.set)
49 XFreeFontSet(dpy, dc.font.set);
50 else
51 XFreeFont(dpy, dc.font.xfont);
52 XUngrabKey(dpy, AnyKey, AnyModifier, root);
53 XFreePixmap(dpy, dc.drawable);
54 XFreeGC(dpy, dc.gc);
55 XDestroyWindow(dpy, barwin);
56 XFreeCursor(dpy, cursor[CurNormal]);
57 XFreeCursor(dpy, cursor[CurResize]);
58 XFreeCursor(dpy, cursor[CurMove]);
59 XSetInputFocus(dpy, PointerRoot, RevertToPointerRoot, CurrentTime);
60 XSync(dpy, False);
61 free(seltag);
62}
63
64static unsigned int
65textnw(const char *text, unsigned int len) {
66 XRectangle r;
67
68 if(dc.font.set) {
69 XmbTextExtents(dc.font.set, text, len, NULL, &r);
70 return r.width;
71 }
72 return XTextWidth(dc.font.xfont, text, len);
73}
74
75static void
76drawtext(const char *text, unsigned long col[ColLast], Bool filledsquare, Bool emptysquare) {
77 int x, y, w, h;
78 static char buf[256];
79 unsigned int len, olen;
80 XGCValues gcv;
81 XRectangle r = { dc.x, dc.y, dc.w, dc.h };
82
83 XSetForeground(dpy, dc.gc, col[ColBG]);
84 XFillRectangles(dpy, dc.drawable, dc.gc, &r, 1);
85 if(!text)
86 return;
87 w = 0;
88 olen = len = strlen(text);
89 if(len >= sizeof buf)
90 len = sizeof buf - 1;
91 memcpy(buf, text, len);
92 buf[len] = 0;
93 h = dc.font.ascent + dc.font.descent;
94 y = dc.y + (dc.h / 2) - (h / 2) + dc.font.ascent;
95 x = dc.x + (h / 2);
96 /* shorten text if necessary */
97 while(len && (w = textnw(buf, len)) > dc.w - h)
98 buf[--len] = 0;
99 if(len < olen) {
100 if(len > 1)
101 buf[len - 1] = '.';
102 if(len > 2)
103 buf[len - 2] = '.';
104 if(len > 3)
105 buf[len - 3] = '.';
106 }
107 if(w > dc.w)
108 return; /* too long */
109 gcv.foreground = col[ColFG];
110 if(dc.font.set) {
111 XChangeGC(dpy, dc.gc, GCForeground, &gcv);
112 XmbDrawString(dpy, dc.drawable, dc.font.set, dc.gc, x, y, buf, len);
113 }
114 else {
115 gcv.font = dc.font.xfont->fid;
116 XChangeGC(dpy, dc.gc, GCForeground | GCFont, &gcv);
117 XDrawString(dpy, dc.drawable, dc.gc, x, y, buf, len);
118 }
119 x = (h + 2) / 4;
120 r.x = dc.x + 1;
121 r.y = dc.y + 1;
122 if(filledsquare) {
123 r.width = r.height = x + 1;
124 XFillRectangles(dpy, dc.drawable, dc.gc, &r, 1);
125 }
126 else if(emptysquare) {
127 r.width = r.height = x;
128 XDrawRectangles(dpy, dc.drawable, dc.gc, &r, 1);
129 }
130}
131
132static unsigned long
133getcolor(const char *colstr) {
134 Colormap cmap = DefaultColormap(dpy, screen);
135 XColor color;
136
137 if(!XAllocNamedColor(dpy, cmap, colstr, &color, &color))
138 eprint("error, cannot allocate color '%s'\n", colstr);
139 return color.pixel;
140}
141
142static Bool
143isoccupied(unsigned int t) {
144 Client *c;
145
146 for(c = clients; c; c = c->next)
147 if(c->tags[t])
148 return True;
149 return False;
150}
151
152static void
153scan(void) {
154 unsigned int i, num;
155 Window *wins, d1, d2;
156 XWindowAttributes wa;
157
158 wins = NULL;
159 if(XQueryTree(dpy, root, &d1, &d2, &wins, &num)) {
160 for(i = 0; i < num; i++) {
161 if(!XGetWindowAttributes(dpy, wins[i], &wa)
162 || wa.override_redirect || XGetTransientForHint(dpy, wins[i], &d1))
163 continue;
164 if(wa.map_state == IsViewable)
165 manage(wins[i], &wa);
166 }
167 }
168 if(wins)
169 XFree(wins);
170}
171
172static void
173setfont(const char *fontstr) {
174 char *def, **missing;
175 int i, n;
176
177 missing = NULL;
178 if(dc.font.set)
179 XFreeFontSet(dpy, dc.font.set);
180 dc.font.set = XCreateFontSet(dpy, fontstr, &missing, &n, &def);
181 if(missing) {
182 while(n--)
183 fprintf(stderr, "missing fontset: %s\n", missing[n]);
184 XFreeStringList(missing);
185 }
186 if(dc.font.set) {
187 XFontSetExtents *font_extents;
188 XFontStruct **xfonts;
189 char **font_names;
190 dc.font.ascent = dc.font.descent = 0;
191 font_extents = XExtentsOfFontSet(dc.font.set);
192 n = XFontsOfFontSet(dc.font.set, &xfonts, &font_names);
193 for(i = 0, dc.font.ascent = 0, dc.font.descent = 0; i < n; i++) {
194 if(dc.font.ascent < (*xfonts)->ascent)
195 dc.font.ascent = (*xfonts)->ascent;
196 if(dc.font.descent < (*xfonts)->descent)
197 dc.font.descent = (*xfonts)->descent;
198 xfonts++;
199 }
200 }
201 else {
202 if(dc.font.xfont)
203 XFreeFont(dpy, dc.font.xfont);
204 dc.font.xfont = NULL;
205 if(!(dc.font.xfont = XLoadQueryFont(dpy, fontstr)))
206 eprint("error, cannot load font: '%s'\n", fontstr);
207 dc.font.ascent = dc.font.xfont->ascent;
208 dc.font.descent = dc.font.xfont->descent;
209 }
210 dc.font.height = dc.font.ascent + dc.font.descent;
211}
212
213static void
214setup(void) {
215 int i, j;
216 unsigned int mask;
217 Window w;
218 XModifierKeymap *modmap;
219 XSetWindowAttributes wa;
220
221 /* init atoms */
222 wmatom[WMProtocols] = XInternAtom(dpy, "WM_PROTOCOLS", False);
223 wmatom[WMDelete] = XInternAtom(dpy, "WM_DELETE_WINDOW", False);
224 wmatom[WMState] = XInternAtom(dpy, "WM_STATE", False);
225 netatom[NetSupported] = XInternAtom(dpy, "_NET_SUPPORTED", False);
226 netatom[NetWMName] = XInternAtom(dpy, "_NET_WM_NAME", False);
227 XChangeProperty(dpy, root, netatom[NetSupported], XA_ATOM, 32,
228 PropModeReplace, (unsigned char *) netatom, NetLast);
229 /* init cursors */
230 cursor[CurNormal] = XCreateFontCursor(dpy, XC_left_ptr);
231 cursor[CurResize] = XCreateFontCursor(dpy, XC_sizing);
232 cursor[CurMove] = XCreateFontCursor(dpy, XC_fleur);
233 /* init modifier map */
234 numlockmask = 0;
235 modmap = XGetModifierMapping(dpy);
236 for (i = 0; i < 8; i++)
237 for (j = 0; j < modmap->max_keypermod; j++) {
238 if(modmap->modifiermap[i * modmap->max_keypermod + j]
239 == XKeysymToKeycode(dpy, XK_Num_Lock))
240 numlockmask = (1 << i);
241 }
242 XFreeModifiermap(modmap);
243 /* select for events */
244 wa.event_mask = SubstructureRedirectMask | SubstructureNotifyMask
245 | EnterWindowMask | LeaveWindowMask;
246 wa.cursor = cursor[CurNormal];
247 XChangeWindowAttributes(dpy, root, CWEventMask | CWCursor, &wa);
248 grabkeys();
249 compileregs();
250 for(ntags = 0; tags[ntags]; ntags++);
251 seltag = emallocz(sizeof(Bool) * ntags);
252 seltag[0] = True;
253 /* style */
254 dc.norm[ColBorder] = getcolor(NORMBORDERCOLOR);
255 dc.norm[ColBG] = getcolor(NORMBGCOLOR);
256 dc.norm[ColFG] = getcolor(NORMFGCOLOR);
257 dc.sel[ColBorder] = getcolor(SELBORDERCOLOR);
258 dc.sel[ColBG] = getcolor(SELBGCOLOR);
259 dc.sel[ColFG] = getcolor(SELFGCOLOR);
260 setfont(FONT);
261 /* geometry */
262 sx = sy = 0;
263 sw = DisplayWidth(dpy, screen);
264 sh = DisplayHeight(dpy, screen);
265 initlayouts();
266 /* bar */
267 dc.h = bh = dc.font.height + 2;
268 wa.override_redirect = 1;
269 wa.background_pixmap = ParentRelative;
270 wa.event_mask = ButtonPressMask | ExposureMask;
271 barwin = XCreateWindow(dpy, root, sx, sy + (TOPBAR ? 0 : sh - bh), sw, bh, 0,
272 DefaultDepth(dpy, screen), CopyFromParent, DefaultVisual(dpy, screen),
273 CWOverrideRedirect | CWBackPixmap | CWEventMask, &wa);
274 XDefineCursor(dpy, barwin, cursor[CurNormal]);
275 XMapRaised(dpy, barwin);
276 strcpy(stext, "dwm-"VERSION);
277 /* windowarea */
278 wax = sx;
279 way = sy + (TOPBAR ? bh : 0);
280 wah = sh - bh;
281 waw = sw;
282 /* pixmap for everything */
283 dc.drawable = XCreatePixmap(dpy, root, sw, bh, DefaultDepth(dpy, screen));
284 dc.gc = XCreateGC(dpy, root, 0, 0);
285 XSetLineAttributes(dpy, dc.gc, 1, LineSolid, CapButt, JoinMiter);
286 /* multihead support */
287 selscreen = XQueryPointer(dpy, root, &w, &w, &i, &i, &i, &i, &mask);
288}
289
290/*
291 * Startup Error handler to check if another window manager
292 * is already running.
293 */
294static int
295xerrorstart(Display *dsply, XErrorEvent *ee) {
296 otherwm = True;
297 return -1;
298}
299
300/* extern */
301
302void
303drawstatus(void) {
304 int i, x;
305
306 dc.x = dc.y = 0;
307 for(i = 0; i < ntags; i++) {
308 dc.w = textw(tags[i]);
309 if(seltag[i])
310 drawtext(tags[i], dc.sel, sel && sel->tags[i], isoccupied(i));
311 else
312 drawtext(tags[i], dc.norm, sel && sel->tags[i], isoccupied(i));
313 dc.x += dc.w;
314 }
315 dc.w = blw;
316 drawtext(lt->symbol, dc.norm, False, False);
317 x = dc.x + dc.w;
318 dc.w = textw(stext);
319 dc.x = sw - dc.w;
320 if(dc.x < x) {
321 dc.x = x;
322 dc.w = sw - x;
323 }
324 drawtext(stext, dc.norm, False, False);
325 if((dc.w = dc.x - x) > bh) {
326 dc.x = x;
327 drawtext(sel ? sel->name : NULL, sel ? dc.sel : dc.norm, False, False);
328 }
329 XCopyArea(dpy, dc.drawable, barwin, dc.gc, 0, 0, sw, bh, 0, 0);
330 XSync(dpy, False);
331}
332
333void
334sendevent(Window w, Atom a, long value) {
335 XEvent e;
336
337 e.type = ClientMessage;
338 e.xclient.window = w;
339 e.xclient.message_type = a;
340 e.xclient.format = 32;
341 e.xclient.data.l[0] = value;
342 e.xclient.data.l[1] = CurrentTime;
343 XSendEvent(dpy, w, False, NoEventMask, &e);
344 XSync(dpy, False);
345}
346
347unsigned int
348textw(const char *text) {
349 return textnw(text, strlen(text)) + dc.font.height;
350}
351
352void
353quit(Arg *arg) {
354 readin = running = False;
355}
356
357/* There's no way to check accesses to destroyed windows, thus those cases are
358 * ignored (especially on UnmapNotify's). Other types of errors call Xlibs
359 * default error handler, which may call exit.
360 */
361int
362xerror(Display *dpy, XErrorEvent *ee) {
363 if(ee->error_code == BadWindow
364 || (ee->request_code == X_SetInputFocus && ee->error_code == BadMatch)
365 || (ee->request_code == X_PolyText8 && ee->error_code == BadDrawable)
366 || (ee->request_code == X_PolyFillRectangle && ee->error_code == BadDrawable)
367 || (ee->request_code == X_PolySegment && ee->error_code == BadDrawable)
368 || (ee->request_code == X_ConfigureWindow && ee->error_code == BadMatch)
369 || (ee->request_code == X_GrabKey && ee->error_code == BadAccess)
370 || (ee->request_code == X_CopyArea && ee->error_code == BadDrawable))
371 return 0;
372 fprintf(stderr, "dwm: fatal error: request code=%d, error code=%d\n",
373 ee->request_code, ee->error_code);
374 return xerrorxlib(dpy, ee); /* may call exit */
375}
376
377int
378main(int argc, char *argv[]) {
379 char *p;
380 int r, xfd;
381 fd_set rd;
382 XEvent ev;
383
384 if(argc == 2 && !strncmp("-v", argv[1], 3))
385 eprint("dwm-"VERSION", (C)opyright MMVI-MMVII Anselm R. Garbe\n");
386 else if(argc != 1)
387 eprint("usage: dwm [-v]\n");
388 setlocale(LC_CTYPE, "");
389 if(!(dpy = XOpenDisplay(0)))
390 eprint("dwm: cannot open display\n");
391 xfd = ConnectionNumber(dpy);
392 screen = DefaultScreen(dpy);
393 root = RootWindow(dpy, screen);
394 otherwm = False;
395 XSetErrorHandler(xerrorstart);
396 /* this causes an error if some other window manager is running */
397 XSelectInput(dpy, root, SubstructureRedirectMask);
398 XSync(dpy, False);
399 if(otherwm)
400 eprint("dwm: another window manager is already running\n");
401
402 XSync(dpy, False);
403 XSetErrorHandler(NULL);
404 xerrorxlib = XSetErrorHandler(xerror);
405 XSync(dpy, False);
406 setup();
407 drawstatus();
408 scan();
409
410 /* main event loop, also reads status text from stdin */
411 XSync(dpy, False);
412 readin = True;
413 while(running) {
414 FD_ZERO(&rd);
415 if(readin)
416 FD_SET(STDIN_FILENO, &rd);
417 FD_SET(xfd, &rd);
418 if(select(xfd + 1, &rd, NULL, NULL, NULL) == -1) {
419 if(errno == EINTR)
420 continue;
421 eprint("select failed\n");
422 }
423 if(FD_ISSET(STDIN_FILENO, &rd)) {
424 switch(r = read(STDIN_FILENO, stext, sizeof stext - 1)) {
425 case -1:
426 strncpy(stext, strerror(errno), sizeof stext - 1);
427 stext[sizeof stext - 1] = '\0';
428 readin = False;
429 break;
430 case 0:
431 strncpy(stext, "EOF", 4);
432 readin = False;
433 break;
434 default:
435 for(stext[r] = '\0', p = stext + strlen(stext) - 1; p >= stext && *p == '\n'; *p-- = '\0');
436 for(; p >= stext && *p != '\n'; --p);
437 if(p > stext)
438 strncpy(stext, p + 1, sizeof stext);
439 }
440 drawstatus();
441 }
442 if(FD_ISSET(xfd, &rd))
443 while(XPending(dpy)) {
444 XNextEvent(dpy, &ev);
445 if(handler[ev.type])
446 (handler[ev.type])(&ev); /* call handler */
447 }
448 }
449 cleanup();
450 XCloseDisplay(dpy);
451 return 0;
452}