#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));
}