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