]> Pileus Git - wmpus/blob - wm-wmii.c
491d1bb7b5d428f35261ad42e41f525edff7e2a9
[wmpus] / wm-wmii.c
1 #include <stdio.h>
2 #include <stdlib.h>
3
4 #include "util.h"
5 #include "sys.h"
6 #include "wm.h"
7
8 #define MODKEY alt
9 #define MARGIN 0
10 #define STACK  25
11
12 /* Enums */
13 typedef enum {
14         none, move, resize
15 } drag_t;
16
17 typedef enum {
18         split, stack, max, tab
19 } mode_t;
20
21
22 /* Window structure types */
23 struct win_wm { };
24
25 typedef struct {
26         win_t  *win;
27         int     height;
28 } row_t;
29
30 typedef struct {
31         list_t *rows; // of row_t
32         row_t  *row;
33         int     width;
34         mode_t  mode;
35 } col_t;
36
37 typedef struct {
38         list_t *cols; // of col_t
39         col_t  *col;
40         win_t  *geom;
41 } dpy_t;
42
43 typedef struct {
44         list_t *dpys; // of dpy_t
45         dpy_t  *dpy;
46         int     name;
47 } tag_t;
48
49 typedef struct {
50         list_t *tags; // of tag_t
51         tag_t  *tag;
52         win_t  *root;
53         list_t *screens;
54 } wm_t;
55
56 /* Mouse drag data */
57 static drag_t  move_mode;
58 static list_t *move_lrow;
59 static list_t *move_lcol;
60 static ptr_t   move_prev;
61 static struct { int v, h; } move_dir;
62
63 /* Window management data */
64 static wm_t  *wm;
65 #define wm_win   wm->tag->dpy->col->row->win
66 #define wm_row   wm->tag->dpy->col->row
67 #define wm_col   wm->tag->dpy->col
68 #define wm_dpy   wm->tag->dpy
69 #define wm_tag   wm->tag
70 #define wm_focus (wm_tag && wm_dpy && wm_col && wm_row ? wm_win : NULL)
71
72 #define WIN(l) ((win_t*)(l)->data)
73 #define ROW(l) ((row_t*)(l)->data)
74 #define COL(l) ((col_t*)(l)->data)
75 #define DPY(l) ((dpy_t*)(l)->data)
76 #define TAG(l) ((tag_t*)(l)->data)
77
78 #define tag_foreach(tag, dpy, col, row, win) \
79         for (list_t *dpy =     tag ->dpys; dpy; dpy = dpy->next) \
80         for (list_t *col = DPY(dpy)->cols; col; col = col->next) \
81         for (list_t *row = COL(col)->rows; row; row = row->next) \
82         for (win_t  *win = ROW(row)->win;  win; win = NULL)      \
83
84 /* Helper functions */
85 static int searchl(tag_t *tag, win_t *target,
86                 list_t **_dpy, list_t **_col, list_t **_row)
87 {
88         tag_foreach(tag, dpy, col, row, win) {
89                 if (win == target) {
90                         if (_dpy) *_dpy = dpy;
91                         if (_col) *_col = col;
92                         if (_row) *_row = row;
93                         return 1;
94                 }
95         }
96         return 0;
97 }
98
99 static int search(tag_t *tag, win_t *target,
100                 dpy_t **_dpy, col_t **_col, row_t **_row)
101 {
102         list_t *dpy, *col, *row;
103         if (searchl(tag, target, &dpy, &col, &row)) {
104                 if (_dpy) *_dpy = DPY(dpy);
105                 if (_col) *_col = COL(col);
106                 if (_row) *_row = ROW(row);
107                 return 1;
108         }
109         return 0;
110 }
111
112 static void set_mode(win_t *win, mode_t mode)
113 {
114         col_t *col;
115         if (!search(wm_tag, win, NULL, &col, NULL))
116                 return;
117         printf("set_mode: %p, %d -> %d\n",
118                         col, col->mode, mode);
119         col->mode = mode;
120         if (col->mode == split)
121                 for (list_t *cur = col->rows; cur; cur = cur->next) {
122                         row_t *row = cur->data;
123                         row->height = wm_dpy->geom->h;
124                 }
125         wm_update();
126 }
127
128 static void set_focus(win_t *win)
129 {
130         if (win == NULL || win == wm->root) {
131                 sys_focus(wm->root);
132                 return;
133         }
134
135         /* - Only grab mouse button on unfocused window,
136          *   this prevents stealing all mouse clicks from client windows,
137          * - A better way may be to re-send mouse clicks to client windows
138          *   using the return value from wm_handle_key */
139         for (int i = key_mouse1; i < key_mouse7; i++) {
140                 if (wm_focus)
141                         sys_watch(wm_focus, i, MOD());
142                 sys_unwatch(win, i, MOD());
143         }
144
145         dpy_t *dpy; col_t *col; row_t *row;
146         if (search(wm_tag, win, &dpy, &col, &row)) {
147                 wm_dpy = dpy;
148                 wm_col = col;
149                 wm_row = row;
150         }
151         sys_focus(win);
152 }
153
154 static void set_move(win_t *win, ptr_t ptr, drag_t drag)
155 {
156         printf("set_move: %d - %p@%d,%d\n",
157                         drag, win, ptr.rx, ptr.ry);
158         move_mode = drag;
159         if (drag == move || drag == resize) {
160                 searchl(wm_tag, win, NULL, &move_lcol, &move_lrow);
161                 move_prev = ptr;
162                 int my = win->y + (win->h/2);
163                 int mx = win->x + (win->w/2);
164                 move_dir.v = ptr.ry < my ? -1 : +1;
165                 move_dir.h = ptr.rx < mx ? -1 : +1;
166         }
167 }
168
169 static void print_txt(void)
170 {
171         for (list_t *ltag = wm->tags; ltag; ltag = ltag->next) {
172                 tag_t *tag = ltag->data;
173                 printf("tag:       <%-9p [%p->%p] >%-9p !%-9p -  %d\n",
174                                 ltag->prev, ltag, ltag->data, ltag->next,
175                                 tag->dpy, tag->name);
176         for (list_t *ldpy = tag->dpys; ldpy; ldpy = ldpy->next) {
177                 dpy_t *dpy  = ldpy->data;
178                 win_t *geom = dpy->geom;
179                 printf("  dpy:     <%-9p [%p->%p] >%-9p !%-9p -  %d,%d %dx%d\n",
180                                 ldpy->prev, ldpy, ldpy->data, ldpy->next,
181                                 dpy->col, geom->x, geom->y, geom->h, geom->w);
182         for (list_t *lcol = dpy->cols; lcol; lcol = lcol->next) {
183                 col_t *col = lcol->data;
184                 printf("    col:   <%-9p [%p->%p] >%-9p !%-9p -  %dpx @ %d\n",
185                                 lcol->prev, lcol, lcol->data, lcol->next,
186                                 col->row, col->width, col->mode);
187         for (list_t *lrow = col->rows; lrow; lrow = lrow->next) {
188                 row_t *row = lrow->data;
189                 win_t *win = row->win;
190                 printf("      win: <%-9p [%p>>%p] >%-9p !%-9p -  %4dpx focus=%d%d\n",
191                                 lrow->prev, lrow, win, lrow->next,
192                                 win, win->h, col->row == row, wm_focus == win);
193         } } } }
194 }
195
196 static void cut_win(tag_t *tag, win_t *win)
197 {
198         list_t *ldpy, *lcol, *lrow;
199         if (!searchl(tag, win, &ldpy, &lcol, &lrow))
200                 return;
201         col_t  *col  = COL(lcol);
202         dpy_t  *dpy  = DPY(ldpy);
203
204         col->row  = lrow->prev ? lrow->prev->data :
205                     lrow->next ? lrow->next->data : NULL;
206         col->rows = list_remove(col->rows, lrow);
207
208         if (col->rows == NULL && (lcol->next || lcol->prev)) {
209                 dpy->col  = lcol->prev ? lcol->prev->data :
210                             lcol->next ? lcol->next->data : NULL;
211                 dpy->cols = list_remove(dpy->cols, lcol);
212         }
213 }
214
215 static void put_win(win_t *win, tag_t *tag, dpy_t *dpy, col_t *col)
216 {
217         row_t *row = new0(row_t);
218         row->win = win;
219
220         if (col == NULL) {
221                 col = new0(col_t);
222                 dpy->cols = list_insert(dpy->cols, col);
223         }
224
225         int nrows = list_length(col->rows);
226         if (col->row) {
227                 list_t *prev = list_find(col->rows, col->row);
228                 list_insert_after(prev, row);
229         } else {
230                 col->rows = list_insert(col->rows, row);
231         }
232         tag->dpy           = dpy;
233         tag->dpy->col      = col;
234         tag->dpy->col->row = row;
235
236         row->height = dpy->geom->h / MAX(nrows,1);
237         if (nrows == 0) {
238                 int ncols = list_length(dpy->cols);
239                 col->width = dpy->geom->w / MAX(ncols-1,1);
240         }
241 }
242
243 static void shift_window(win_t *win, int col, int row)
244 {
245         if (!win) return;
246         printf("shift_window: %p - %+d,%+d\n", win, col, row);
247         print_txt();
248         printf("shift_window: >>>\n");
249         list_t *ldpy, *lcol, *lrow;
250         if (!searchl(wm_tag, win, &ldpy, &lcol, &lrow))
251                 return;
252         dpy_t *dpy = ldpy->data;
253         if (row != 0) {
254                 list_t *src = lrow, *dst = NULL;
255                 if (row < 0) dst = src->prev;
256                 if (row > 0) dst = src->next;
257                 if (src && dst) {
258                         printf("swap: %p <-> %p\n", src->data, dst->data);
259                         row_t *tmp = src->data;
260                         src->data = dst->data;
261                         dst->data = tmp;
262                         goto update;
263                 }
264         } else {
265                 int onlyrow = !lrow->prev && !lrow->next;
266                 list_t *src = lcol, *dst = NULL;
267                 if (col < 0) {
268                         if (src->prev) {
269                                 dst = src->prev;
270                         } else if (!onlyrow) {
271                                 dpy->cols = list_insert(dpy->cols, new0(col_t));
272                                 dst = src->prev;
273                         } else if (ldpy->prev) {
274                                 dpy = ldpy->prev->data;
275                                 dst = list_last(dpy->cols);
276                         } else {
277                                 return;
278                         }
279                 }
280                 if (col > 0) {
281                         if (src->next) {
282                                 dst = src->next;
283                         } else if (!onlyrow) {
284                                 dpy->cols = list_append(dpy->cols, new0(col_t));
285                                 dst = src->next;
286                         } else if (ldpy->next) {
287                                 dpy = ldpy->next->data;
288                                 dst = dpy->cols;
289                         } else {
290                                 return;
291                         }
292                 }
293                 cut_win(wm_tag, win);
294                 put_win(win, wm_tag, dpy, dst ? dst->data : NULL);
295                 goto update;
296         }
297 update:
298         print_txt();
299         wm_update();
300 }
301
302 static list_t *get_next(list_t *list, int forward)
303 {
304         list_t *next = forward ? list->next : list->prev;
305         if (next == NULL) {
306                 next = list;
307                 while ((list = forward ? next->prev : next->next))
308                         next = list;
309         }
310         return next;
311 }
312 static void shift_focus(int cols, int rows)
313 {
314         printf("shift_focus: %+d,%+d\n", cols, rows);
315         if (rows != 0 && wm_focus) {
316                 list_t *dpy, *col, *row;
317                 if (!searchl(wm_tag, wm_focus, &dpy, &col, &row))
318                         return;
319                 row_t *next = get_next(row, rows > 0)->data;
320                 set_focus(next->win);
321                 if (COL(col)->mode != split)
322                         wm_update();
323         }
324         if (cols != 0) {
325                 list_t *dpy, *col, *row, *ndpy, *ncol = NULL;
326                 if (wm_focus) {
327                         if (!searchl(wm_tag, wm_focus, &dpy, &col, &row))
328                                 return;
329                         ncol = cols > 0 ? col->next : col->prev;
330                 } else {
331                         dpy = list_find(wm_tag->dpys, wm_dpy);
332                 }
333                 if (ncol == NULL) {
334                         ndpy = get_next(dpy, cols > 0);
335                         ncol = cols > 0 ? DPY(ndpy)->cols :
336                                 list_last(DPY(ndpy)->cols);
337                         wm_dpy = ndpy->data;
338                 }
339                 if (ncol && COL(ncol) && COL(ncol)->row)
340                         set_focus(COL(ncol)->row->win);
341                 else
342                         sys_focus(wm->root);
343         }
344 }
345
346 static tag_t *tag_new(list_t *screens, int name)
347 {
348         tag_t *tag = new0(tag_t);
349         tag->name  = name;
350         for (list_t *cur = screens; cur; cur = cur->next) {
351                 dpy_t *dpy  = new0(dpy_t);
352                 dpy->geom = cur->data;
353                 tag->dpys = list_append(tag->dpys, dpy);
354         }
355         tag->dpy  = tag->dpys->data;
356         return tag;
357 }
358
359 static tag_t *tag_find(int name)
360 {
361         tag_t *tag = NULL;
362         for (list_t *cur = wm->tags; cur; cur = cur->next)
363                 if (name == TAG(cur)->name) {
364                         tag = cur->data;
365                         break;
366                 }
367         if (!tag) {
368                 tag = tag_new(wm->screens, name);
369                 wm->tags = list_append(wm->tags, tag);
370         }
371         return tag;
372 }
373
374 static void tag_set(win_t *win, int name)
375 {
376         printf("tag_set: %p %d\n", win, name);
377         if (wm_tag->name == name)
378                 return;
379         tag_t *tag = tag_find(name);
380         cut_win(wm_tag, win);
381         put_win(win, tag, tag->dpy, tag->dpy->col);
382         set_focus(wm_focus);
383 }
384
385 static void tag_switch(int name)
386 {
387         printf("tag_switch: %d\n", name);
388         if (wm_col == NULL || wm_row == NULL)
389                 wm->tags = list_remove(wm->tags,
390                                 list_find(wm->tags, wm_tag));
391         wm_tag = tag_find(name);
392 }
393
394 /* Window management functions */
395 void wm_update_dpy(dpy_t *dpy)
396 {
397         int  x=0,  y=0; // Current window top-left position
398         int tx=0, ty=0; // Total x/y size
399         int mx=0, my=0; // Maximum x/y size (screen size)
400         int       sy=0; // Size of focused stack window
401
402         /* Scale horizontally */
403         x  = dpy->geom->x;
404         mx = dpy->geom->w - (list_length(dpy->cols)+1)*MARGIN;
405         for (list_t *lcol = dpy->cols; lcol; lcol = lcol->next)
406                 tx += COL(lcol)->width;
407         for (list_t *lcol = dpy->cols; lcol; lcol = lcol->next)
408                 COL(lcol)->width *= (float)mx / tx;
409
410         /* Scale each column vertically */
411         for (list_t *lcol = dpy->cols; lcol; lcol = lcol->next) {
412                 col_t *col = lcol->data;
413                 ty = 0;
414                 for (list_t *lrow = col->rows; lrow; lrow = lrow->next)
415                         ty += ROW(lrow)->height;
416                 y  = dpy->geom->y;
417                 my = dpy->geom->h - (list_length(col->rows)+1)*MARGIN;
418                 sy = my           - (list_length(col->rows)-1)*STACK;
419                 for (list_t *lrow = col->rows; lrow; lrow = lrow->next) {
420                         win_t *win = ROW(lrow)->win;
421                         win->h = ROW(lrow)->height;
422                         int height = 0;
423                         switch (col->mode) {
424                         case split:
425                                 sys_move(win, x+MARGIN, y+MARGIN,
426                                         col->width, win->h * ((float)my / ty));
427                                 height = win->h;
428                                 break;
429                         case stack:
430                                 if (lrow->next && ROW(lrow->next)->win == col->row->win) {
431                                         /* Hack to prevent flashing */
432                                         win_t *next = ROW(lrow->next)->win;
433                                         sys_move(next, x+MARGIN, y+MARGIN+STACK+MARGIN,
434                                                 col->width, sy);
435                                 }
436                                 height = win == col->row->win ? sy : STACK;
437                                 sys_move(win, x+MARGIN, y+MARGIN,
438                                         col->width, height);
439                                 break;
440                         case max:
441                         case tab:
442                                 sys_move(win, x+MARGIN, 0+MARGIN,
443                                         col->width, dpy->geom->h-2*MARGIN);
444                                 if (col->row->win == win)
445                                         sys_raise(win);
446                                 break;
447                         }
448                         y += height + MARGIN;
449                         ROW(lrow)->height = win->h;
450                 }
451                 x += col->width + MARGIN;
452         }
453 }
454
455 void wm_update(void)
456 {
457         /* Show/hide tags */
458         tag_foreach(wm_tag, dpy, col, row, win)
459                 sys_show(win, st_show);
460         for (list_t *tag = wm ->tags; tag; tag = tag->next)
461                 tag_foreach(TAG(tag), dpy, col, row, win)
462                         if (tag->data != wm_tag)
463                                 sys_show(win, st_hide);
464
465         /* Refrsh the display */
466         for (list_t *ldpy = wm_tag->dpys; ldpy; ldpy = ldpy->next)
467                 wm_update_dpy(ldpy->data);
468         if (wm_focus)
469                 set_focus(wm_focus);
470 }
471
472 int wm_handle_key(win_t *win, Key_t key, mod_t mod, ptr_t ptr)
473 {
474         if (!win || win == wm_dpy->geom) return 0;
475         //printf("wm_handle_key: %p - %x %c%c%c%c%c\n", win, key,
476         //      mod.up    ? '^' : 'v',
477         //      mod.alt   ? 'a' : '-',
478         //      mod.ctrl  ? 'c' : '-',
479         //      mod.shift ? 's' : '-',
480         //      mod.win   ? 'w' : '-');
481
482         /* Mouse movement */
483         if (key_mouse0 <= key && key <= key_mouse7 && mod.up)
484                 return set_move(win,ptr,none),   0;
485         else if (key == key_mouse1 && mod.MODKEY)
486                 return set_move(win,ptr,move),   1;
487         else if (key == key_mouse3 && mod.MODKEY)
488                 return set_move(win,ptr,resize), 1;
489
490         /* Only handle key-down */
491         if (mod.up)
492                 return 0;
493
494         /* Misc */
495         if (mod.MODKEY) {
496 #ifdef DEBUG
497                 if (key == key_f1) return sys_raise(win), 1;
498                 if (key == key_f2) return set_focus(win), 1;
499                 if (key == key_f3) return sys_show(win, st_show), 1;
500                 if (key == key_f4) return sys_show(win, st_hide), 1;
501 #endif
502                 if (key == key_f5) return wm_update(),    1;
503                 if (key == key_f6) return print_txt(),    1;
504         }
505         if (key_mouse0 <= key && key <= key_mouse7)
506                 sys_raise(win);
507
508         /* Movement commands */
509         if (mod.MODKEY && mod.shift) {
510                 switch (key) {
511                 case 'h': return shift_window(wm_focus,-1, 0), 1;
512                 case 'j': return shift_window(wm_focus, 0,+1), 1;
513                 case 'k': return shift_window(wm_focus, 0,-1), 1;
514                 case 'l': return shift_window(wm_focus,+1, 0), 1;
515                 default: break;
516                 }
517         }
518         else if (mod.MODKEY) {
519                 switch (key) {
520                 case 'h': return shift_focus(-1, 0), 1;
521                 case 'j': return shift_focus( 0,+1), 1;
522                 case 'k': return shift_focus( 0,-1), 1;
523                 case 'l': return shift_focus(+1, 0), 1;
524                 default: break;
525                 }
526         }
527
528         /* Column mode commands */
529         if (mod.MODKEY) {
530                 switch (key) {
531                 case 'd': return set_mode(win, split), 1;
532                 case 's': return set_mode(win, stack), 1;
533                 case 'm': return set_mode(win, max),   1;
534                 case 't': return set_mode(win, tab),   1;
535                 default: break;
536                 }
537         }
538
539         /* Tag switching */
540         if (mod.MODKEY && '0' <= key && key <= '9') {
541                 int name = key - '0';
542                 if (mod.shift)
543                         tag_set(win, name);
544                 else
545                         tag_switch(name);
546                 wm_update();
547         }
548
549         /* Focus change */
550         if (key == key_enter)
551                 return set_focus(win), 1;
552
553         if (key_mouse0 <= key && key <= key_mouse7)
554                 return set_focus(win), 0;
555
556         /* Reset focus after after focus change,
557          * not sure what is causing the focus change in the first place
558          * but preventing that would be a better solution */
559         if (key == key_focus)
560                 sys_focus(wm_focus ?: wm->root);
561
562         return 0;
563 }
564
565 int wm_handle_ptr(win_t *cwin, ptr_t ptr)
566 {
567         //printf("wm_handle_ptr: %p - %d,%d %d,%d (%d) -- \n",
568         //              cwin, ptr.x, ptr.y, ptr.rx, ptr.ry, move_mode);
569
570         if (move_mode == none)
571                 return 0;
572
573         /* Tiling */
574         int dx = ptr.rx - move_prev.rx;
575         int dy = ptr.ry - move_prev.ry;
576         move_prev = ptr;
577         if (move_mode == resize) {
578                 list_t *vert = move_dir.v < 0 ? move_lrow->prev : move_lrow->next;
579                 list_t *horz = move_dir.h < 0 ? move_lcol->prev : move_lcol->next;
580                 if (vert) {
581                         ROW(move_lrow)->height += move_dir.v * dy;
582                         ROW(vert)->height      -= move_dir.v * dy;
583                 }
584                 if (horz) {
585                         COL(move_lcol)->width  += move_dir.h * dx;
586                         COL(horz)->width       -= move_dir.h * dx;
587                 }
588                 wm_update();
589         }
590
591         /* Floating */
592         //win_t *mwin = move_win;
593         //int dx = ptr.rx - move_prev.rx;
594         //int dy = ptr.ry - move_prev.ry;
595         //move_prev = ptr;
596         //if (move_mode == move)
597         //      sys_move(mwin, mwin->x+dx, mwin->y+dy, mwin->w, mwin->h);
598         //else if (move_mode == resize)
599         //      sys_move(mwin, mwin->x, mwin->y, mwin->w+dx, mwin->h+dy);
600
601         return 0;
602 }
603
604 void wm_insert(win_t *win)
605 {
606         printf("wm_insert: %p\n", win);
607         print_txt();
608
609         /* Initialize window */
610         win->wm = new0(win_wm_t);
611         sys_watch(win, key_enter, MOD());
612         sys_watch(win, key_focus, MOD());
613
614         /* Add to screen */
615         put_win(win, wm_tag, wm_dpy, wm_col);
616
617         /* Arrange */
618         wm_update();
619         set_focus(wm_focus);
620         print_txt();
621 }
622
623 void wm_remove(win_t *win)
624 {
625         printf("wm_remove: %p\n", win);
626         print_txt();
627         for (list_t *tag = wm->tags; tag; tag = tag->next)
628                 cut_win(tag->data, win);
629         set_focus(wm_focus);
630         wm_update();
631         print_txt();
632 }
633
634 void wm_init(win_t *root)
635 {
636         printf("wm_init: %p\n", root);
637
638         /* Hack, fix screen order */
639         list_t *screens = sys_info(root);
640         list_t *left  = screens;
641         list_t *right = screens->next;
642         if (left && right && WIN(left)->x > WIN(right)->x) {
643                 void *tmp   = left->data;
644                 left->data  = right->data;
645                 right->data = tmp;
646         }
647
648         wm          = new0(wm_t);
649         wm->root    = root;
650         wm->screens = screens;
651         wm->tag     = tag_new(wm->screens, 1);
652         wm->tags    = list_insert(NULL, wm->tag);
653
654         Key_t keys_e[] = {key_enter, key_focus};
655         Key_t keys_s[] = {'h', 'j', 'k', 'l',
656                 '0', '1', '2', '3', '4', '5', '6', '7', '8', '9'};
657         Key_t keys_m[] = {'h', 'j', 'k', 'l', 'd', 's', 'm', 't',
658                 '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
659                 key_f1, key_f2, key_f3, key_f4, key_f5, key_f6,
660                 key_mouse1, key_mouse3};
661         for (int i = 0; i < countof(keys_e); i++)
662                 sys_watch(root, keys_e[i],  MOD());
663         for (int i = 0; i < countof(keys_m); i++)
664                 sys_watch(root, keys_m[i], MOD(.MODKEY=1));
665         for (int i = 0; i < countof(keys_s); i++)
666                 sys_watch(root, keys_s[i], MOD(.MODKEY=1,.shift=1));
667 }