JasonWoof Got questions, comments, patches, etc.? Contact Jason Woofenden
tags should be persistent now during X server run
[dwm.git] / main.c
1 /* See LICENSE file for copyright and license details. */
2 #include "dwm.h"
3 #include <errno.h>
4 #include <locale.h>
5 #include <stdio.h>
6 #include <stdlib.h>
7 #include <string.h>
8 #include <unistd.h>
9 #include <sys/select.h>
10 #include <X11/cursorfont.h>
11 #include <X11/keysym.h>
12 #include <X11/Xatom.h>
13 #include <X11/Xproto.h>
14
15 /* extern */
16
17 char stext[256];
18 int screen, sx, sy, sw, sh, wax, way, waw, wah;
19 unsigned int bh, ntags;
20 unsigned int bpos = BARPOS;
21 unsigned int numlockmask = 0;
22 Atom dwmtags, wmatom[WMLast], netatom[NetLast];
23 Bool *seltag;
24 Bool selscreen = True;
25 Client *clients = NULL;
26 Client *sel = NULL;
27 Client *stack = NULL;
28 Cursor cursor[CurLast];
29 Display *dpy;
30 DC dc = {0};
31 Window root, barwin;
32
33 /* static */
34
35 static int (*xerrorxlib)(Display *, XErrorEvent *);
36 static Bool otherwm, readin;
37 static Bool running = True;
38
39 static void
40 cleanup(void) {
41         close(STDIN_FILENO);
42         while(stack) {
43                 unban(stack);
44                 unmanage(stack);
45         }
46         if(dc.font.set)
47                 XFreeFontSet(dpy, dc.font.set);
48         else
49                 XFreeFont(dpy, dc.font.xfont);
50         XUngrabKey(dpy, AnyKey, AnyModifier, root);
51         XFreePixmap(dpy, dc.drawable);
52         XFreeGC(dpy, dc.gc);
53         XDestroyWindow(dpy, barwin);
54         XFreeCursor(dpy, cursor[CurNormal]);
55         XFreeCursor(dpy, cursor[CurResize]);
56         XFreeCursor(dpy, cursor[CurMove]);
57         XSetInputFocus(dpy, PointerRoot, RevertToPointerRoot, CurrentTime);
58         XSync(dpy, False);
59         free(seltag);
60 }
61
62 static unsigned long
63 initcolor(const char *colstr) {
64         Colormap cmap = DefaultColormap(dpy, screen);
65         XColor color;
66
67         if(!XAllocNamedColor(dpy, cmap, colstr, &color, &color))
68                 eprint("error, cannot allocate color '%s'\n", colstr);
69         return color.pixel;
70 }
71
72 static void
73 initfont(const char *fontstr) {
74         char *def, **missing;
75         int i, n;
76
77         missing = NULL;
78         if(dc.font.set)
79                 XFreeFontSet(dpy, dc.font.set);
80         dc.font.set = XCreateFontSet(dpy, fontstr, &missing, &n, &def);
81         if(missing) {
82                 while(n--)
83                         fprintf(stderr, "dwm: missing fontset: %s\n", missing[n]);
84                 XFreeStringList(missing);
85         }
86         if(dc.font.set) {
87                 XFontSetExtents *font_extents;
88                 XFontStruct **xfonts;
89                 char **font_names;
90                 dc.font.ascent = dc.font.descent = 0;
91                 font_extents = XExtentsOfFontSet(dc.font.set);
92                 n = XFontsOfFontSet(dc.font.set, &xfonts, &font_names);
93                 for(i = 0, dc.font.ascent = 0, dc.font.descent = 0; i < n; i++) {
94                         if(dc.font.ascent < (*xfonts)->ascent)
95                                 dc.font.ascent = (*xfonts)->ascent;
96                         if(dc.font.descent < (*xfonts)->descent)
97                                 dc.font.descent = (*xfonts)->descent;
98                         xfonts++;
99                 }
100         }
101         else {
102                 if(dc.font.xfont)
103                         XFreeFont(dpy, dc.font.xfont);
104                 dc.font.xfont = NULL;
105                 if(!(dc.font.xfont = XLoadQueryFont(dpy, fontstr)))
106                         eprint("error, cannot load font: '%s'\n", fontstr);
107                 dc.font.ascent = dc.font.xfont->ascent;
108                 dc.font.descent = dc.font.xfont->descent;
109         }
110         dc.font.height = dc.font.ascent + dc.font.descent;
111 }
112
113 static void
114 scan(void) {
115         unsigned int i, num;
116         Window *wins, d1, d2;
117         XWindowAttributes wa;
118
119         wins = NULL;
120         if(XQueryTree(dpy, root, &d1, &d2, &wins, &num)) {
121                 for(i = 0; i < num; i++) {
122                         if(!XGetWindowAttributes(dpy, wins[i], &wa)
123                         || wa.override_redirect || XGetTransientForHint(dpy, wins[i], &d1))
124                                 continue;
125                         if(wa.map_state == IsViewable)
126                                 manage(wins[i], &wa);
127                 }
128         }
129         if(wins)
130                 XFree(wins);
131 }
132
133 static void
134 setup(void) {
135         int i, j;
136         unsigned int mask;
137         Window w;
138         XModifierKeymap *modmap;
139         XSetWindowAttributes wa;
140
141         /* init atoms */
142         dwmtags = XInternAtom(dpy, "__DWM_TAGS", False);
143         wmatom[WMProtocols] = XInternAtom(dpy, "WM_PROTOCOLS", False);
144         wmatom[WMDelete] = XInternAtom(dpy, "WM_DELETE_WINDOW", False);
145         wmatom[WMState] = XInternAtom(dpy, "WM_STATE", False);
146         netatom[NetSupported] = XInternAtom(dpy, "_NET_SUPPORTED", False);
147         netatom[NetWMName] = XInternAtom(dpy, "_NET_WM_NAME", False);
148         XChangeProperty(dpy, root, netatom[NetSupported], XA_ATOM, 32,
149                         PropModeReplace, (unsigned char *) netatom, NetLast);
150         /* init cursors */
151         cursor[CurNormal] = XCreateFontCursor(dpy, XC_left_ptr);
152         cursor[CurResize] = XCreateFontCursor(dpy, XC_sizing);
153         cursor[CurMove] = XCreateFontCursor(dpy, XC_fleur);
154         /* init modifier map */
155         modmap = XGetModifierMapping(dpy);
156         for (i = 0; i < 8; i++)
157                 for (j = 0; j < modmap->max_keypermod; j++) {
158                         if(modmap->modifiermap[i * modmap->max_keypermod + j]
159                                         == XKeysymToKeycode(dpy, XK_Num_Lock))
160                                 numlockmask = (1 << i);
161                 }
162         XFreeModifiermap(modmap);
163         /* select for events */
164         wa.event_mask = SubstructureRedirectMask | SubstructureNotifyMask
165                 | EnterWindowMask | LeaveWindowMask | StructureNotifyMask;
166         wa.cursor = cursor[CurNormal];
167         XChangeWindowAttributes(dpy, root, CWEventMask | CWCursor, &wa);
168         XSelectInput(dpy, root, wa.event_mask);
169         grabkeys();
170         compileregs();
171         for(ntags = 0; tags[ntags]; ntags++);
172         seltag = emallocz(sizeof(Bool) * ntags);
173         seltag[0] = True;
174         /* style */
175         dc.norm[ColBorder] = initcolor(NORMBORDERCOLOR);
176         dc.norm[ColBG] = initcolor(NORMBGCOLOR);
177         dc.norm[ColFG] = initcolor(NORMFGCOLOR);
178         dc.sel[ColBorder] = initcolor(SELBORDERCOLOR);
179         dc.sel[ColBG] = initcolor(SELBGCOLOR);
180         dc.sel[ColFG] = initcolor(SELFGCOLOR);
181         initfont(FONT);
182         /* geometry */
183         sx = sy = 0;
184         sw = DisplayWidth(dpy, screen);
185         sh = DisplayHeight(dpy, screen);
186         initlayouts();
187         /* bar */
188         dc.h = bh = dc.font.height + 2;
189         wa.override_redirect = 1;
190         wa.background_pixmap = ParentRelative;
191         wa.event_mask = ButtonPressMask | ExposureMask;
192         barwin = XCreateWindow(dpy, root, sx, sy, sw, bh, 0,
193                         DefaultDepth(dpy, screen), CopyFromParent, DefaultVisual(dpy, screen),
194                         CWOverrideRedirect | CWBackPixmap | CWEventMask, &wa);
195         XDefineCursor(dpy, barwin, cursor[CurNormal]);
196         updatebarpos();
197         XMapRaised(dpy, barwin);
198         strcpy(stext, "dwm-"VERSION);
199         /* pixmap for everything */
200         dc.drawable = XCreatePixmap(dpy, root, sw, bh, DefaultDepth(dpy, screen));
201         dc.gc = XCreateGC(dpy, root, 0, 0);
202         XSetLineAttributes(dpy, dc.gc, 1, LineSolid, CapButt, JoinMiter);
203         if(!dc.font.set)
204                 XSetFont(dpy, dc.gc, dc.font.xfont->fid);
205         /* multihead support */
206         selscreen = XQueryPointer(dpy, root, &w, &w, &i, &i, &i, &i, &mask);
207 }
208
209 /*
210  * Startup Error handler to check if another window manager
211  * is already running.
212  */
213 static int
214 xerrorstart(Display *dsply, XErrorEvent *ee) {
215         otherwm = True;
216         return -1;
217 }
218
219 /* extern */
220
221 void
222 quit(const char *arg) {
223         readin = running = False;
224 }
225
226 void
227 updatebarpos(void) {
228         XEvent ev;
229
230         wax = sx;
231         way = sy;
232         wah = sh;
233         waw = sw;
234         switch(bpos) {
235         default:
236                 wah -= bh;
237                 way += bh;
238                 XMoveWindow(dpy, barwin, sx, sy);
239                 break;
240         case BarBot:
241                 wah -= bh;
242                 XMoveWindow(dpy, barwin, sx, sy + wah);
243                 break;
244         case BarOff:
245                 XMoveWindow(dpy, barwin, sx, sy - bh);
246                 break;
247         }
248         XSync(dpy, False);
249         while(XCheckMaskEvent(dpy, EnterWindowMask, &ev));
250 }
251
252 /* There's no way to check accesses to destroyed windows, thus those cases are
253  * ignored (especially on UnmapNotify's).  Other types of errors call Xlibs
254  * default error handler, which may call exit.
255  */
256 int
257 xerror(Display *dpy, XErrorEvent *ee) {
258         if(ee->error_code == BadWindow
259         || (ee->request_code == X_SetInputFocus && ee->error_code == BadMatch)
260         || (ee->request_code == X_PolyText8 && ee->error_code == BadDrawable)
261         || (ee->request_code == X_PolyFillRectangle && ee->error_code == BadDrawable)
262         || (ee->request_code == X_PolySegment && ee->error_code == BadDrawable)
263         || (ee->request_code == X_ConfigureWindow && ee->error_code == BadMatch)
264         || (ee->request_code == X_GrabKey && ee->error_code == BadAccess)
265         || (ee->request_code == X_CopyArea && ee->error_code == BadDrawable))
266                 return 0;
267         fprintf(stderr, "dwm: fatal error: request code=%d, error code=%d\n",
268                 ee->request_code, ee->error_code);
269         return xerrorxlib(dpy, ee); /* may call exit */
270 }
271
272 int
273 main(int argc, char *argv[]) {
274         char *p;
275         int r, xfd;
276         fd_set rd;
277         XEvent ev;
278
279         if(argc == 2 && !strcmp("-v", argv[1]))
280                 eprint("dwm-"VERSION", © 2006-2007 A. R. Garbe, S. van Dijk, J. Salmi, P. Hruby, S. Nagy\n");
281         else if(argc != 1)
282                 eprint("usage: dwm [-v]\n");
283         setlocale(LC_CTYPE, "");
284         if(!(dpy = XOpenDisplay(0)))
285                 eprint("dwm: cannot open display\n");
286         xfd = ConnectionNumber(dpy);
287         screen = DefaultScreen(dpy);
288         root = RootWindow(dpy, screen);
289         otherwm = False;
290         XSetErrorHandler(xerrorstart);
291         /* this causes an error if some other window manager is running */
292         XSelectInput(dpy, root, SubstructureRedirectMask);
293         XSync(dpy, False);
294         if(otherwm)
295                 eprint("dwm: another window manager is already running\n");
296
297         XSync(dpy, False);
298         XSetErrorHandler(NULL);
299         xerrorxlib = XSetErrorHandler(xerror);
300         XSync(dpy, False);
301         setup();
302         drawstatus();
303         scan();
304
305         /* main event loop, also reads status text from stdin */
306         XSync(dpy, False);
307         readin = True;
308         while(running) {
309                 FD_ZERO(&rd);
310                 if(readin)
311                         FD_SET(STDIN_FILENO, &rd);
312                 FD_SET(xfd, &rd);
313                 if(select(xfd + 1, &rd, NULL, NULL, NULL) == -1) {
314                         if(errno == EINTR)
315                                 continue;
316                         eprint("select failed\n");
317                 }
318                 if(FD_ISSET(STDIN_FILENO, &rd)) {
319                         switch(r = read(STDIN_FILENO, stext, sizeof stext - 1)) {
320                         case -1:
321                                 strncpy(stext, strerror(errno), sizeof stext - 1);
322                                 stext[sizeof stext - 1] = '\0';
323                                 readin = False;
324                                 break;
325                         case 0:
326                                 strncpy(stext, "EOF", 4);
327                                 readin = False;
328                                 break;
329                         default:
330                                 for(stext[r] = '\0', p = stext + strlen(stext) - 1; p >= stext && *p == '\n'; *p-- = '\0');
331                                 for(; p >= stext && *p != '\n'; --p);
332                                 if(p > stext)
333                                         strncpy(stext, p + 1, sizeof stext);
334                         }
335                         drawstatus();
336                 }
337                 while(XPending(dpy)) {
338                         XNextEvent(dpy, &ev);
339                         if(handler[ev.type])
340                                 (handler[ev.type])(&ev); /* call handler */
341                 }
342         }
343         cleanup();
344         XCloseDisplay(dpy);
345         return 0;
346 }