]> Pileus Git - wmpus/blobdiff - wm-wmii.c
Add initial Xinerama support
[wmpus] / wm-wmii.c
index 444a5968084fb5fffe0d6208ce817af959d77f14..e5c1a6ca48a85a06e3d674b46f86615c37440621 100644 (file)
--- a/wm-wmii.c
+++ b/wm-wmii.c
 #include <stdio.h>
+#include <stdlib.h>
 
+#include "util.h"
 #include "sys.h"
 #include "wm.h"
 
+#define MODKEY ctrl
+#define MARGIN 0
+#define STACK  25
+
+/* Enums */
 typedef enum {
        none, move, resize
-} Mode_t;
+} drag_t;
+
+typedef enum {
+       split, stack, max, tab
+} mode_t;
+
+
+/* Window structure types */
+struct win_wm { };
+
+typedef struct {
+       win_t  *win;
+       int     height;
+} row_t;
+
+typedef struct {
+       list_t *rows; // of row_t
+       row_t  *row;
+       int     width;
+       mode_t  mode;
+} col_t;
+
+typedef struct {
+       list_t *cols; // of col_t
+       col_t  *col;
+       win_t  *geom;
+} dpy_t;
+
+typedef struct {
+       list_t *dpys; // of dpy_t
+       dpy_t  *dpy;
+       int     name;
+} tag_t;
+
+typedef struct {
+       list_t *tags; // of tag_t
+       tag_t  *tag;
+       win_t  *root;
+       list_t *screens;
+} wm_t;
+
+/* Mouse drag data */
+static drag_t  move_mode;
+static list_t *move_lrow;
+static list_t *move_lcol;
+static ptr_t   move_prev;
+static struct { int v, h; } move_dir;
+
+/* Window management data */
+static wm_t  *wm;
+#define wm_focus wm->tag->dpy->col->row->win
+#define wm_row   wm->tag->dpy->col->row
+#define wm_col   wm->tag->dpy->col
+#define wm_dpy   wm->tag->dpy
+#define wm_tag   wm->tag
+
+#define ROW(l) ((row_t*)(l)->data)
+#define COL(l) ((col_t*)(l)->data)
+#define DPY(l) ((dpy_t*)(l)->data)
+#define TAG(l) ((tag_t*)(l)->data)
+
+/* Helper functions */
+static int searchl(tag_t *tag, win_t *target,
+               list_t **_dpy, list_t **_col, list_t **_row)
+{
+       for (list_t *dpy =     tag ->dpys; dpy; dpy = dpy->next)
+       for (list_t *col = DPY(dpy)->cols; col; col = col->next)
+       for (list_t *row = COL(col)->rows; row; row = row->next) {
+               win_t *win = ROW(row)->win;
+               if (win == target) {
+                       if (_dpy) *_dpy = dpy;
+                       if (_col) *_col = col;
+                       if (_row) *_row = row;
+                       return 1;
+               }
+       }
+       return 0;
+}
+
+static int search(tag_t *tag, win_t *target,
+               dpy_t **_dpy, col_t **_col, row_t **_row)
+{
+       list_t *dpy, *col, *row;
+       if (searchl(tag, target, &dpy, &col, &row)) {
+               if (_dpy) *_dpy = DPY(dpy);
+               if (_col) *_col = COL(col);
+               if (_row) *_row = ROW(row);
+               return 1;
+       }
+       return 0;
+}
+
+static void set_mode(win_t *win, mode_t mode)
+{
+       col_t *col;
+       search(wm_tag, win, NULL, &col, NULL);
+       printf("set_mode: %p, %d -> %d\n",
+                       col, col->mode, mode);
+       col->mode = mode;
+       if (col->mode == split)
+               for (list_t *cur = col->rows; cur; cur = cur->next) {
+                       row_t *row = cur->data;
+                       row->height = wm_dpy->geom->h;
+               }
+       wm_update();
+}
+
+static void set_focus(win_t *win)
+{
+       /* - Only grab mouse button on unfocused window,
+        *   this prevents stealing all mouse clicks from client windows,
+        * - A better way may be to re-send mouse clicks to client windows
+        *   using the return value from wm_handle_key */
+       for (int i = key_mouse1; i < key_mouse7; i++) {
+               if (wm_focus)
+                       sys_watch(wm_focus, i, MOD());
+               sys_unwatch(win, i, MOD());
+       }
+
+       dpy_t *dpy; col_t *col; row_t *row;
+       if (search(wm_tag, win, &dpy, &col, &row)) {
+               wm_dpy = dpy;
+               wm_col = col;
+               wm_row = row;
+       }
+       sys_focus(win);
+}
+
+static void set_move(win_t *win, ptr_t ptr, drag_t drag)
+{
+       printf("set_move: %d - %p@%d,%d\n",
+                       drag, win, ptr.rx, ptr.ry);
+       move_mode = drag;
+       if (drag == move || drag == resize) {
+               searchl(wm_tag, win, NULL, &move_lcol, &move_lrow);
+               move_prev = ptr;
+               int my = win->y + (win->h/2);
+               int mx = win->x + (win->w/2);
+               move_dir.v = ptr.ry < my ? -1 : +1;
+               move_dir.h = ptr.rx < mx ? -1 : +1;
+       }
+}
+
+static void print_txt(list_t *cols)
+{
+       for (list_t *lcol = cols; lcol; lcol = lcol->next) {
+               col_t *col = lcol->data;
+               printf("col:\t<%-9p [%-20p] >%-9p  -  %dpx @ %d !!%p\n",
+                               ( lcol->prev ? lcol->prev->data : NULL ),
+                               col,
+                               ( lcol->next ? lcol->next->data : NULL ),
+                               col->width, col->mode, col->row);
+               for (list_t *lrow = col->rows; lrow; lrow = lrow->next) {
+                       row_t *row = lrow->data;
+                       win_t *win = row->win;
+                       printf("  win:\t<%-9p [%p>>%p] >%-9p  -  %4dpx focus=%d%d\n",
+                                       lrow->prev, lrow, win, lrow->next,
+                                       win->h, col->row == row, wm_focus == win);
+               }
+       }
+}
+
+static void cut_win(win_t *win)
+{
+       list_t *ldpy, *lcol, *lrow;
+       searchl(wm_tag, win, &ldpy, &lcol, &lrow);
+       col_t  *col  = COL(lcol);
+       dpy_t  *dpy  = DPY(ldpy);
+
+       col->row  = lrow->prev ? lrow->prev->data :
+                   lrow->next ? lrow->next->data : NULL;
+       col->rows = list_remove(col->rows, lrow);
+
+       if (col->rows == NULL && (lcol->next || lcol->prev)) {
+               dpy->col  = lcol->prev ? lcol->prev->data :
+                           lcol->next ? lcol->next->data : NULL;
+               dpy->cols = list_remove(wm_dpy->cols, lcol);
+       }
+}
+
+static void put_win(win_t *win, col_t *col)
+{
+       row_t *row = new0(row_t);
+       row->win = win;
 
-win_t *kwin;
-ptr_t  kptr;
-Mode_t mode;
+       if (col == NULL) {
+               col = new0(col_t);
+               wm_dpy->cols = list_insert(wm_dpy->cols, col);
+       }
+
+       int nrows = list_length(col->rows);
+       if (col->row) {
+               list_t *prev = list_find(col->rows, col->row);
+               list_insert_after(prev, row);
+       } else {
+               col->rows = list_insert(col->rows, row);
+       }
+       col->row    = row;
+       wm_dpy->col = col;
+
+       row->height = wm_dpy->geom->h / MAX(nrows,1);
+       if (nrows == 0) {
+               int ncols = list_length(wm_dpy->cols);
+               col->width = wm_dpy->geom->w / MAX(ncols-1,1);
+       }
+}
+
+static void shift_window(win_t *win, int col, int row)
+{
+       printf("shift_window: %p - %+d,%+d\n", win, col, row);
+       print_txt(wm_dpy->cols);
+       printf("shift_window: >>>\n");
+       list_t *ldpy, *lcol, *lrow;
+       searchl(wm_tag, win, &ldpy, &lcol, &lrow);
+       if (row != 0) {
+               list_t *src = lrow, *dst = NULL;
+               if (row < 0) dst = src->prev;
+               if (row > 0) dst = src->next;
+               if (src && dst) {
+                       printf("swap: %p <-> %p\n", src->data, dst->data);
+                       row_t *tmp = src->data;
+                       src->data = dst->data;
+                       dst->data = tmp;
+                       goto update;
+               }
+       } else {
+               int onlyrow = !lrow->prev && !lrow->next;
+               list_t *src = lcol, *dst = NULL;
+               if (col < 0) {
+                       if (!src->prev && !onlyrow)
+                               wm_dpy->cols = list_insert(wm_dpy->cols, new0(col_t));
+                       dst = src->prev;
+               }
+               if (col > 0) {
+                       if (!src->next && !onlyrow)
+                               wm_dpy->cols = list_append(wm_dpy->cols, new0(col_t));
+                       dst = src->next;
+               }
+               if (src && dst) {
+                       cut_win(win);
+                       put_win(win, COL(dst));
+                       goto update;
+               }
+       }
+       return;
+update:
+       print_txt(wm_dpy->cols);
+       wm_update();
+}
+
+static list_t *get_next(list_t *list, int forward)
+{
+       list_t *next = forward ? list->next : list->prev;
+       if (next == NULL) {
+               next = list;
+               while ((list = forward ? next->prev : next->next))
+                       next = list;
+       }
+       return next;
+}
+static void shift_focus(win_t *win, int col, int row)
+{
+       printf("shift_focus: %p - %+d,%+d\n", win, col, row);
+       list_t *ldpy, *lcol, *lrow;
+       searchl(wm_tag, win, &ldpy, &lcol, &lrow);
+       if (row != 0) {
+               row_t *next = get_next(lrow, row > 0)->data;
+               set_focus(next->win);
+               if (COL(lcol)->mode != split)
+                       wm_update();
+       }
+       if (col != 0) {
+               col_t *next = get_next(lcol, col > 0)->data;
+               set_focus(next->row->win);
+       }
+}
+
+/* Window management functions */
+void wm_update(void)
+{
+       int  x=0,  y=0; // Current window top-left position
+       int tx=0, ty=0; // Total x/y size
+       int mx=0, my=0; // Maximum x/y size (screen size)
+       int       sy=0; // Size of focused stack window
+
+       /* Scale horizontally */
+       x  = wm_dpy->geom->x;
+       mx = wm_dpy->geom->w - (list_length(wm_dpy->cols)+1)*MARGIN;
+       for (list_t *lx = wm_dpy->cols; lx; lx = lx->next)
+               tx += COL(lx)->width;
+       for (list_t *lx = wm_dpy->cols; lx; lx = lx->next)
+               COL(lx)->width *= (float)mx / tx;
+
+       /* Scale each column vertically */
+       for (list_t *lx = wm_dpy->cols; lx; lx = lx->next) {
+               col_t *col = lx->data;
+               ty = 0;
+               for (list_t *ly = col->rows; ly; ly = ly->next)
+                       ty += ROW(ly)->height;
+               y  = wm_dpy->geom->y;
+               my = wm_dpy->geom->h - (list_length(col->rows)+1)*MARGIN;
+               sy = my              - (list_length(col->rows)-1)*STACK;
+               for (list_t *ly = col->rows; ly; ly = ly->next) {
+                       win_t *win = ROW(ly)->win;
+                       win->h = ROW(ly)->height;
+                       int height = 0;
+                       switch (col->mode) {
+                       case split:
+                               sys_move(win, x+MARGIN, y+MARGIN,
+                                       col->width, win->h * ((float)my / ty));
+                               height = win->h;
+                               break;
+                       case stack:
+                               height = col->row->win == win ? sy : STACK;
+                               sys_move(win, x+MARGIN, y+MARGIN,
+                                       col->width, height);
+                               break;
+                       case max:
+                       case tab:
+                               sys_move(win, x+MARGIN, 0+MARGIN,
+                                       col->width, wm_dpy->geom->h-2*MARGIN);
+                               if (col->row->win == win)
+                                       sys_raise(win);
+                               break;
+                       }
+                       y += height + MARGIN;
+                       ROW(ly)->height = win->h;
+               }
+               x += col->width + MARGIN;
+       }
+}
 
 int wm_handle_key(win_t *win, Key_t key, mod_t mod, ptr_t ptr)
 {
-       printf("wm_handle_key: %p - %x\n", win, key);
-       kptr = ptr;
-       kwin = win;
+       if (!win || win == wm_dpy->geom) return 0;
+       //printf("wm_handle_key: %p - %x %c%c%c%c%c\n", win, key,
+       //      mod.up    ? '^' : 'v',
+       //      mod.alt   ? 'a' : '-',
+       //      mod.ctrl  ? 'c' : '-',
+       //      mod.shift ? 's' : '-',
+       //      mod.win   ? 'w' : '-');
+
+       /* Mouse movement */
+       if (key_mouse0 <= key && key <= key_mouse7 && mod.up)
+               return set_move(win,ptr,none), 1;
+       else if (key == key_mouse1 && mod.MODKEY)
+               return set_move(win,ptr,move), 1;
+       else if (key == key_mouse3 && mod.MODKEY)
+               return set_move(win,ptr,resize), 1;
+
+       /* Only handle key-down */
+       if (mod.up)
+               return 0;
 
-       /* Raise */
-       if ((key == key_f1 && mod.ctrl) ||
-                       (key_mouse0 <= key && key <= key_mouse7))
+       /* Misc */
+       if (mod.MODKEY) {
+               if (key == key_f1) return sys_raise(win), 1;
+               if (key == key_f2) return set_focus(win), 1;
+               if (key == key_f5) return wm_update(),    1;
+               if (key == key_f6) return print_txt(wm_dpy->cols), 1;
+       }
+       if (key_mouse0 <= key && key <= key_mouse7)
                sys_raise(win);
 
-       /* Movement */
-       if (key_mouse0 <= key && key <= key_mouse7 &&
-                       mod.up && mode != none)
-               return mode = none, 1;
-       else if (key == key_mouse1 && mod.ctrl)
-               return mode = move, 1;
-       else if (key == key_mouse3 && mod.ctrl)
-               return mode = resize, 1;
+       /* Movement commands */
+       if (mod.MODKEY && mod.shift) {
+               switch (key) {
+               case 'h': return shift_window(win,-1, 0), 1;
+               case 'j': return shift_window(win, 0,+1), 1;
+               case 'k': return shift_window(win, 0,-1), 1;
+               case 'l': return shift_window(win,+1, 0), 1;
+               default: break;
+               }
+       }
+       else if (mod.MODKEY) {
+               switch (key) {
+               case 'h': return shift_focus(win,-1, 0), 1;
+               case 'j': return shift_focus(win, 0,+1), 1;
+               case 'k': return shift_focus(win, 0,-1), 1;
+               case 'l': return shift_focus(win,+1, 0), 1;
+               default: break;
+               }
+       }
+
+       /* Column mode commands */
+       if (mod.MODKEY) {
+               switch (key) {
+               case 'd': return set_mode(win, split), 1;
+               case 's': return set_mode(win, stack), 1;
+               case 'm': return set_mode(win, max),   1;
+               case 't': return set_mode(win, tab),   1;
+               default: break;
+               }
+       }
+
+       /* Focus change */
+       if (key == key_enter)
+               return set_focus(win), 1;
+
+       if (key_mouse0 <= key && key <= key_mouse7)
+               return set_focus(win), 0;
+
+       /* Reset focus after after focus change,
+        * not sure what is causing the focus change in the first place
+        * but preventing that would be a better solution */
+       if (key == key_focus)
+               set_focus(wm_focus);
 
        return 0;
 }
 
-int wm_handle_ptr(win_t *win, ptr_t ptr)
+int wm_handle_ptr(win_t *cwin, ptr_t ptr)
 {
-       printf("wm_handle_ptr: %p - %d,%d %d,%d (%d) -- \n",
-                       win, ptr.x, ptr.y, ptr.rx, ptr.ry, mode);
-       int dx = ptr.rx - kptr.rx;
-       int dy = ptr.ry - kptr.ry;
-       if (mode == move)
-               sys_move(kwin, kwin->x+dx, kwin->y+dy, kwin->w, kwin->h);
-       else if (mode == resize)
-               sys_move(kwin, kwin->x, kwin->y, kwin->w+dx, kwin->h+dy);
+       //printf("wm_handle_ptr: %p - %d,%d %d,%d (%d) -- \n",
+       //              cwin, ptr.x, ptr.y, ptr.rx, ptr.ry, move_mode);
+
+       if (move_mode == none)
+               return 0;
+
+       /* Tiling */
+       int dx = ptr.rx - move_prev.rx;
+       int dy = ptr.ry - move_prev.ry;
+       move_prev = ptr;
+       if (move_mode == resize) {
+               list_t *vert = move_dir.v < 0 ? move_lrow->prev : move_lrow->next;
+               list_t *horz = move_dir.h < 0 ? move_lcol->prev : move_lcol->next;
+               if (vert) {
+                       ROW(move_lrow)->height += move_dir.v * dy;
+                       ROW(vert)->height      -= move_dir.v * dy;
+               }
+               if (horz) {
+                       COL(move_lcol)->width  += move_dir.h * dx;
+                       COL(horz)->width       -= move_dir.h * dx;
+               }
+               wm_update();
+       }
+
+       /* Floating */
+       //win_t *mwin = move_win;
+       //int dx = ptr.rx - move_prev.rx;
+       //int dy = ptr.ry - move_prev.ry;
+       //move_prev = ptr;
+       //if (move_mode == move)
+       //      sys_move(mwin, mwin->x+dx, mwin->y+dy, mwin->w, mwin->h);
+       //else if (move_mode == resize)
+       //      sys_move(mwin, mwin->x, mwin->y, mwin->w+dx, mwin->h+dy);
+
        return 0;
 }
 
+void wm_insert(win_t *win)
+{
+       printf("wm_insert: %p\n", win);
+       print_txt(wm_dpy->cols);
+
+       /* Initialize window */
+       win->wm = new0(win_wm_t);
+       sys_watch(win, key_enter, MOD());
+       sys_watch(win, key_focus, MOD());
+
+       /* Add to screen */
+       put_win(win, wm_col);
+
+       /* Arrange */
+       wm_update();
+       sys_focus(wm_focus);
+       print_txt(wm_dpy->cols);
+}
+
+void wm_remove(win_t *win)
+{
+       printf("wm_remove: %p\n", win);
+       print_txt(wm_dpy->cols);
+       cut_win(win);
+       if (wm_focus)
+               sys_focus(wm_focus);
+       else
+               sys_focus(wm->root);
+       wm_update();
+       print_txt(wm_dpy->cols);
+}
+
+tag_t *tag_new(list_t *screens, int name)
+{
+       tag_t *tag = new0(tag_t);
+       tag->name  = name;
+       for (list_t *cur = screens; cur; cur = cur->next) {
+               dpy_t *dpy  = new0(dpy_t);
+               dpy->geom = cur->data;
+               tag->dpys = list_append(tag->dpys, dpy);
+       }
+       tag->dpy  = tag->dpys->data;
+       return tag;
+}
+
 void wm_init(win_t *root)
 {
-       sys_watch(root, key_f1,     MOD(.ctrl=1));
-       sys_watch(root, key_mouse1, MOD(.ctrl=1));
-       sys_watch(root, key_mouse3, MOD(.ctrl=1));
+       printf("wm_init: %p\n", root);
+
+       wm          = new0(wm_t);
+       wm->root    = root;
+       wm->screens = sys_info(root);
+       wm->tag     = tag_new(wm->screens, 1);
+       wm->tags    = list_insert(NULL, wm->tag);
+
+       Key_t keys_e[] = {key_enter, key_focus};
+       Key_t keys_s[] = {'h', 'j', 'k', 'l'};
+       Key_t keys_m[] = {'h', 'j', 'k', 'l', 'd', 's', 'm', 't',
+               key_f1, key_f2, key_f5, key_f6, key_mouse1, key_mouse3};
+       for (int i = 0; i < countof(keys_e); i++)
+               sys_watch(root, keys_e[i],  MOD());
+       for (int i = 0; i < countof(keys_m); i++)
+               sys_watch(root, keys_m[i], MOD(.MODKEY=1));
+       for (int i = 0; i < countof(keys_s); i++)
+               sys_watch(root, keys_s[i], MOD(.MODKEY=1,.shift=1));
 }