Pileus Git - wmpus/blob - sys-x11.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3 #include <search.h>
   4
   5 #include <X11/Xlib.h>
   6 #include <X11/Xproto.h>
   7 #include <X11/Xatom.h>
   8 #include <X11/keysym.h>
   9
  10 #include "util.h"
  11 #include "sys.h"
  12 #include "wm.h"
  13
  14 /* Internal structures */
  15 struct win_sys {
  16         Window   xid;
  17         Display *dpy;
  18         struct {
  19                 int left, right, top, bottom;
  20         } strut;
  21 };
  22
  23 typedef struct {
  24         Key_t key;
  25         int   sym;
  26 } keymap_t;
  27
  28 typedef enum {
  29         wm_proto, wm_focus, net_strut, natoms
  30 } atom_t;
  31
  32 /* Global data */
  33 static void *win_cache = NULL;
  34 static Atom atoms[natoms];
  35 static int (*xerrorxlib)(Display *, XErrorEvent *);
  36
  37 /* Conversion functions */
  38 static keymap_t key2sym[] = {
  39         {key_left    , XK_Left },
  40         {key_right   , XK_Right},
  41         {key_up      , XK_Up   },
  42         {key_down    , XK_Down },
  43         {key_home    , XK_Home },
  44         {key_end     , XK_End  },
  45         {key_pageup  , XK_Prior},
  46         {key_pagedown, XK_Next },
  47         {key_f1      , XK_F1   },
  48         {key_f2      , XK_F2   },
  49         {key_f3      , XK_F3   },
  50         {key_f4      , XK_F4   },
  51         {key_f5      , XK_F5   },
  52         {key_f6      , XK_F6   },
  53         {key_f7      , XK_F7   },
  54         {key_f8      , XK_F8   },
  55         {key_f9      , XK_F9   },
  56         {key_f10     , XK_F10  },
  57         {key_f11     , XK_F11  },
  58         {key_f12     , XK_F12  },
  59 };
  60
  61 /* - Modifiers */
  62 static mod_t x2mod(unsigned int state, int up)
  63 {
  64         return (mod_t){
  65                .alt   = !!(state & Mod1Mask   ),
  66                .ctrl  = !!(state & ControlMask),
  67                .shift = !!(state & ShiftMask  ),
  68                .win   = !!(state & Mod4Mask   ),
  69                .up    = up,
  70         };
  71 }
  72
  73 static unsigned int mod2x(mod_t mod)
  74 {
  75         return (mod.alt   ? Mod1Mask    : 0)
  76              | (mod.ctrl  ? ControlMask : 0)
  77              | (mod.shift ? ShiftMask   : 0)
  78              | (mod.win   ? Mod4Mask    : 0);
  79 }
  80
  81 /* - Keycodes */
  82 static Key_t x2key(KeySym sym)
  83 {
  84         keymap_t *km = map_getr(key2sym,sym);
  85         return km ? km->key : sym;
  86 }
  87
  88 static KeySym key2x(Key_t key)
  89 {
  90         keymap_t *km = map_get(key2sym,key);
  91         return km ? km->sym : key;
  92 }
  93
  94 static Key_t x2btn(int btn)
  95 {
  96         return btn + key_mouse0;
  97 }
  98
  99 static int btn2x(Key_t key)
 100 {
 101         return key - key_mouse0;
 102 }
 103
 104 /* - Pointers */
 105 static ptr_t x2ptr(XEvent *_ev)
 106 {
 107         XKeyEvent *ev = &_ev->xkey;
 108         return (ptr_t){ev->x, ev->y, ev->x_root, ev->y_root};
 109 }
 110
 111 static Window getfocus(win_t *root, XEvent *event)
 112 {
 113         int revert;
 114         Window focus = PointerRoot;
 115         if (event->type == KeyPress || event->type == KeyRelease)
 116                 XGetInputFocus(root->sys->dpy, &focus, &revert);
 117         if (focus == PointerRoot)
 118                 focus = event->xkey.subwindow;
 119         if (focus == None)
 120                 focus = event->xkey.window;
 121         return focus;
 122 }
 123
 124 /* Helpers */
 125 static int add_strut(win_t *root, win_t *win)
 126 {
 127         /* Get X11 strut data */
 128         Atom ret_type;
 129         int ret_size;
 130         unsigned long ret_items, bytes_left;
 131         unsigned char *xdata;
 132         int status = XGetWindowProperty(win->sys->dpy, win->sys->xid,
 133                         atoms[net_strut], 0L, 4L, False, XA_CARDINAL,
 134                         &ret_type, &ret_size, &ret_items, &bytes_left, &xdata);
 135         if (status != Success || ret_size != 32 || ret_items != 4)
 136                 return 0;
 137
 138         int left   = ((int*)xdata)[0];
 139         int right  = ((int*)xdata)[1];
 140         int top    = ((int*)xdata)[2];
 141         int bottom = ((int*)xdata)[3];
 142         if (left == 0 && right == 0 && top == 0 && bottom == 0)
 143                 return 0;
 144
 145         win->sys->strut.left   = left;
 146         win->sys->strut.right  = right;
 147         win->sys->strut.top    = top;
 148         win->sys->strut.bottom = bottom;
 149         root->x += left;
 150         root->y += top;
 151         root->w -= left+right;
 152         root->h -= top+bottom;
 153         return 1;
 154 }
 155
 156 static int del_strut(win_t *root, win_t *win)
 157 {
 158         int left   = win->sys->strut.left;
 159         int right  = win->sys->strut.right;
 160         int top    = win->sys->strut.top;
 161         int bottom = win->sys->strut.bottom;
 162         if (left == 0 && right == 0 && top == 0 && bottom == 0)
 163                 return 0;
 164
 165         root->x -= left;
 166         root->y -= top;
 167         root->w += left+right;
 168         root->h += top+bottom;
 169         return 1;
 170 }
 171
 172 /* Window functions */
 173 static win_t *win_new(Display *dpy, Window xid)
 174 {
 175         XWindowAttributes attr;
 176         if (XGetWindowAttributes(dpy, xid, &attr))
 177                 if (attr.override_redirect)
 178                         return NULL;
 179         win_t *win    = new0(win_t);
 180         win->x        = attr.x;
 181         win->y        = attr.y;
 182         win->w        = attr.width;
 183         win->h        = attr.height;
 184         win->sys      = new0(win_sys_t);
 185         win->sys->dpy = dpy;
 186         win->sys->xid = xid;
 187         printf("%p = %p, %d\n", win, dpy, (int)xid);
 188         return win;
 189 }
 190
 191 static int win_cmp(const void *_a, const void *_b)
 192 {
 193         const win_t *a = _a, *b = _b;
 194         if (a->sys->dpy < b->sys->dpy) return -1;
 195         if (a->sys->dpy > b->sys->dpy) return  1;
 196         if (a->sys->xid < b->sys->xid) return -1;
 197         if (a->sys->xid > b->sys->xid) return  1;
 198         return 0;
 199 }
 200
 201 static win_t *win_find(Display *dpy, Window xid, int create)
 202 {
 203         if (!dpy || !xid)
 204                 return NULL;
 205         //printf("win_find: %p, %d\n", dpy, (int)xid);
 206         win_sys_t sys = {.dpy=dpy, .xid=xid};
 207         win_t     tmp = {.sys=&sys};
 208         win_t **old = NULL, *new = NULL;
 209         if ((old = tfind(&tmp, &win_cache, win_cmp)))
 210                 return *old;
 211         if (create && (new = win_new(dpy,xid)))
 212                 tsearch(new, &win_cache, win_cmp);
 213         return new;
 214 }
 215
 216 static void win_remove(win_t *win)
 217 {
 218         tdelete(win, &win_cache, win_cmp);
 219         free(win->sys);
 220         free(win);
 221 }
 222
 223 static int win_viewable(win_t *win)
 224 {
 225         XWindowAttributes attr;
 226         if (XGetWindowAttributes(win->sys->dpy, win->sys->xid, &attr))
 227                 return attr.map_state == IsViewable;
 228         else
 229                 return True;
 230 }
 231
 232 /* Callbacks */
 233 static void process_event(int type, XEvent *ev, win_t *root)
 234 {
 235         Display  *dpy = root->sys->dpy;
 236         win_t *win = NULL;
 237         printf("event: %d\n", type);
 238
 239         /* Common data for all these events ... */
 240         ptr_t ptr; mod_t mod;
 241         if (type == KeyPress    || type == KeyRelease    ||
 242             type == ButtonPress || type == ButtonRelease ||
 243             type == MotionNotify) {
 244                 Window xid = getfocus(root, ev);
 245                 if (!(win = win_find(dpy,xid,0)))
 246                         return;
 247                 ptr = x2ptr(ev);
 248                 mod = x2mod(ev->xkey.state, type==KeyRelease||type==ButtonRelease);
 249         }
 250
 251         /* Split based on event */
 252         if (type == KeyPress) {
 253                 while (XCheckTypedEvent(dpy, KeyPress, ev));
 254                 KeySym sym = XKeycodeToKeysym(dpy, ev->xkey.keycode, 0);
 255                 wm_handle_key(win, x2key(sym), mod, ptr);
 256         }
 257         else if (type == KeyRelease) {
 258                 //printf("release: %d\n", type);
 259         }
 260         else if (type == ButtonPress) {
 261                 wm_handle_key(win, x2btn(ev->xbutton.button), mod, ptr);
 262                 XGrabPointer(dpy, ev->xbutton.root, True, PointerMotionMask|ButtonReleaseMask,
 263                                 GrabModeAsync, GrabModeAsync, None, None, CurrentTime);
 264         }
 265         else if (type == ButtonRelease) {
 266                 XUngrabPointer(dpy, CurrentTime);
 267                 wm_handle_key(win, x2btn(ev->xbutton.button), mod, ptr);
 268         }
 269         else if (type == MotionNotify) {
 270                 while (XCheckTypedEvent(dpy, MotionNotify, ev));
 271                 wm_handle_ptr(win, ptr);
 272         }
 273         else if (type == EnterNotify || type == LeaveNotify) {
 274                 printf("enter: %d\n", type);
 275                 key_t key = EnterNotify ? key_enter : key_leave;
 276                 if ((win = win_find(dpy,ev->xcrossing.window,0)))
 277                         wm_handle_key(win, key, MOD(), PTR());
 278         }
 279         else if (type == FocusIn || type == FocusOut) {
 280                 printf("focus: %d\n", type);
 281                 key_t key = FocusIn ? key_focus : key_unfocus;
 282                 if ((win = win_find(dpy,ev->xfocus.window,0)))
 283                         wm_handle_key(win, key, MOD(), PTR());
 284         }
 285         else if (type == ConfigureNotify) {
 286                 printf("configure: %d\n", type);
 287         }
 288         else if (type == MapNotify) {
 289                 printf("map: %d\n", type);
 290         }
 291         else if (type == UnmapNotify) {
 292                 //printf("unmap: %d\n", type);
 293                 if ((win = win_find(dpy,ev->xunmap.window,0))) {
 294                         if (!del_strut(root, win))
 295                                 wm_remove(win);
 296                         else
 297                                 wm_update();
 298                         win_remove(win);
 299                 }
 300         }
 301         else if (type == ConfigureRequest) {
 302                 XConfigureRequestEvent *cre = &ev->xconfigurerequest;
 303                 XWindowChanges wc = {
 304                         .x     = cre->x,     .y      = cre->y,
 305                         .width = cre->width, .height = cre->height,
 306                 };
 307                 if ((win = win_find(dpy,ev->xmaprequest.window,1))) {
 308                         wc.x      = win->x;
 309                         wc.y      = win->y;
 310                         wc.width  = win->w;
 311                         wc.height = win->h;
 312                 }
 313                 printf("configure_req: %d - %x, (0x%lx) %dx%d @ %d,%d\n",
 314                                 type, (int)cre->window, cre->value_mask,
 315                                 cre->height, cre->width, cre->x, cre->y);
 316                 XConfigureWindow(dpy, cre->window, cre->value_mask, &wc);
 317         }
 318         else if (type == MapRequest) {
 319                 printf("map_req: %d\n", type);
 320                 if ((win = win_find(dpy,ev->xmaprequest.window,1))) {
 321                         if (!add_strut(root, win))
 322                                 wm_insert(win);
 323                         else
 324                                 wm_update();
 325                 }
 326                 XMapWindow(dpy,ev->xmaprequest.window);
 327         }
 328         else {
 329                 printf("unknown event: %d\n", type);
 330         }
 331 }
 332
 333 static int xerror(Display *dpy, XErrorEvent *err)
 334 {
 335         if (err->error_code == BadWindow ||
 336             (err->request_code == X_SetInputFocus     && err->error_code == BadMatch   ) ||
 337             (err->request_code == X_PolyText8         && err->error_code == BadDrawable) ||
 338             (err->request_code == X_PolyFillRectangle && err->error_code == BadDrawable) ||
 339             (err->request_code == X_PolySegment       && err->error_code == BadDrawable) ||
 340             (err->request_code == X_ConfigureWindow   && err->error_code == BadMatch   ) ||
 341             (err->request_code == X_GrabButton        && err->error_code == BadAccess  ) ||
 342             (err->request_code == X_GrabKey           && err->error_code == BadAccess  ) ||
 343             (err->request_code == X_CopyArea          && err->error_code == BadDrawable))
 344                 return 0;
 345         return xerrorxlib(dpy, err);
 346 }
 347
 348 /*****************
 349  * Sys functions *
 350  *****************/
 351 void sys_move(win_t *win, int x, int y, int w, int h)
 352 {
 353         //printf("sys_move: %p - %d,%d  %dx%d\n", win, x, y, w, h);
 354         win->x = MAX(x,0); win->y = MAX(y,0);
 355         win->w = MAX(w,1); win->h = MAX(h,1);
 356         XMoveResizeWindow(win->sys->dpy, win->sys->xid,
 357                         win->x, win->y, win->w, win->h);
 358
 359         /* Flush events, so moving window doesn't cuase re-focus
 360          * There's probably a better way to do this */
 361         XEvent ev;
 362         XSync(win->sys->dpy, False);
 363         while (XCheckMaskEvent(win->sys->dpy, EnterWindowMask|LeaveWindowMask, &ev))
 364                 printf("Skipping enter/leave event\n");
 365 }
 366
 367 void sys_raise(win_t *win)
 368 {
 369         //printf("sys_raise: %p\n", win);
 370         XRaiseWindow(win->sys->dpy, win->sys->xid);
 371 }
 372
 373 void sys_focus(win_t *win)
 374 {
 375         printf("sys_focus: %p\n", win);
 376         XEvent ev = {
 377                 .type                 = ClientMessage,
 378                 .xclient.window       = win->sys->xid,
 379                 .xclient.message_type = atoms[wm_proto],
 380                 .xclient.format       = 32,
 381                 .xclient.data.l[0]    = atoms[wm_focus],
 382                 .xclient.data.l[1]    = CurrentTime,
 383         };
 384         XSetInputFocus(win->sys->dpy, win->sys->xid,
 385                         RevertToPointerRoot, CurrentTime);
 386         //XSetInputFocus(win->sys->dpy, PointerRoot,
 387         //              RevertToPointerRoot, CurrentTime);
 388         XSendEvent(win->sys->dpy, win->sys->xid, False, NoEventMask, &ev);
 389 }
 390
 391 void sys_watch(win_t *win, Key_t key, mod_t mod)
 392 {
 393         //printf("sys_watch: %p - %x %hhx\n", win, key, mod);
 394         XWindowAttributes attr;
 395         XGetWindowAttributes(win->sys->dpy, win->sys->xid, &attr);
 396         long mask = attr.your_event_mask;
 397         if (key_mouse0 <= key && key <= key_mouse7)
 398                 XGrabButton(win->sys->dpy, btn2x(key), mod2x(mod), win->sys->xid, True,
 399                                 mod.up ? ButtonReleaseMask : ButtonPressMask,
 400                                 GrabModeAsync, GrabModeAsync, None, None);
 401         else if (key == key_enter)
 402                 XSelectInput(win->sys->dpy, win->sys->xid, EnterWindowMask|mask);
 403         else if (key == key_leave)
 404                 XSelectInput(win->sys->dpy, win->sys->xid, LeaveWindowMask|mask);
 405         else if (key == key_focus || key == key_unfocus)
 406                 XSelectInput(win->sys->dpy, win->sys->xid, FocusChangeMask|mask);
 407         else
 408                 XGrabKey(win->sys->dpy, XKeysymToKeycode(win->sys->dpy, key2x(key)),
 409                                 mod2x(mod), win->sys->xid, True, GrabModeAsync, GrabModeAsync);
 410 }
 411
 412 win_t *sys_init(void)
 413 {
 414         Display *dpy;
 415         Window   xid;
 416         if (!(dpy = XOpenDisplay(NULL)))
 417                 error("Unable to get display");
 418         if (!(xid = DefaultRootWindow(dpy)))
 419                 error("Unable to get root window");
 420         atoms[wm_proto]  = XInternAtom(dpy, "WM_PROTOCOLS",  False);
 421         atoms[wm_focus]  = XInternAtom(dpy, "WM_TAKE_FOCUS", False);
 422         atoms[net_strut] = XInternAtom(dpy, "_NET_WM_STRUT", False);
 423         XSelectInput(dpy, xid, SubstructureRedirectMask|SubstructureNotifyMask);
 424         XSetInputFocus(dpy, None, RevertToNone, CurrentTime);
 425         xerrorxlib = XSetErrorHandler(xerror);
 426         return win_find(dpy, xid, 1);
 427 }
 428
 429 void sys_run(win_t *root)
 430 {
 431         /* Add each initial window */
 432         unsigned int nkids;
 433         Window par, xid, *kids = NULL;
 434         if (XQueryTree(root->sys->dpy, root->sys->xid,
 435                                 &par, &xid, &kids, &nkids))
 436                 for(int i = 0; i < nkids; i++) {
 437                         win_t *win = win_find(root->sys->dpy, kids[i], 1);
 438                         if (win && win_viewable(win) && !add_strut(root,win))
 439                                 wm_insert(win);
 440                 }
 441         wm_update(); // For struts
 442
 443         /* Main loop */
 444         for(;;)
 445         {
 446                 XEvent ev;
 447                 XNextEvent(root->sys->dpy, &ev);
 448                 process_event(ev.type, &ev, root);
 449         }
 450 }