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