Pileus Git - wmpus/blob - 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
  10 /* Loca types */
  11 struct win_wm {
  12         list_t *col;
  13         list_t *row;
  14 };
  15
  16 typedef enum {
  17         none, move, resize
  18 } drag_t;
  19
  20 typedef enum {
  21         stack, split, max, tab
  22 } group_t;
  23
  24 typedef struct {
  25         int     width;
  26         group_t group;
  27         list_t *rows;
  28 } col_t;
  29
  30 /* Mouse drag data */
  31 drag_t move_mode;
  32 win_t *move_win;
  33 ptr_t  move_prev;
  34
  35 /* Window management data */
  36 list_t *wm_cols;
  37 win_t  *wm_root;
  38
  39 /* Helper functions */
  40 static void set_mode(drag_t drag, win_t *win, ptr_t ptr)
  41 {
  42         printf("set_mode: %d - %p@%d,%d\n",
  43                         drag, win, ptr.rx, ptr.ry);
  44         move_mode = drag;
  45         if (drag == move || drag == resize) {
  46                 move_win  = win;
  47                 move_prev = ptr;
  48         }
  49 }
  50
  51 static void print_txt(list_t *cols)
  52 {
  53         for (list_t *cnode = cols; cnode; cnode = cnode->next) {
  54                 col_t *col = cnode->data;
  55                 printf("col:\t%p - %dpx @ %d\n", col, col->width, col->group);
  56                 for (list_t *wnode = col->rows; wnode; wnode = wnode->next) {
  57                         win_t *win = wnode->data;
  58                         printf("  win:\t%p - %dpx\n", win, win->h);
  59                 }
  60         }
  61 }
  62
  63 static void arrange(list_t *cols)
  64 {
  65         int  x=0,  y=0; // Current window top-left position
  66         int tx=0, ty=0; // Total x/y size
  67         int mx=0, my=0; // Maximum x/y size (screen size)
  68
  69         mx = wm_root->w;
  70         my = wm_root->h;
  71
  72         /* Scale horizontally */
  73         for (list_t *lx = cols; lx; lx = lx->next)
  74                 tx += ((col_t*)lx->data)->width;
  75         for (list_t *lx = cols; lx; lx = lx->next)
  76                 ((col_t*)lx->data)->width *= (float)mx / tx;
  77
  78         /* Scale each column vertically */
  79         for (list_t *lx = cols; lx; lx = lx->next) {
  80                 col_t *col = lx->data;
  81                 ty = 0;
  82                 for (list_t *ly = col->rows; ly; ly = ly->next)
  83                         ty += ((win_t*)ly->data)->h;
  84                 y = 0;
  85                 for (list_t *ly = col->rows; ly; ly = ly->next) {
  86                         win_t *win = ly->data;
  87                         sys_move(win, x, y, col->width,
  88                                 win->h * ((float)my / ty));
  89                         y += win->h;
  90                 }
  91                 x += col->width;
  92         }
  93 }
  94
  95 static void shift_window(win_t *win, int col, int row)
  96 {
  97         printf("shift_window: %p - %+d,%+d\n", win, col, row);
  98         print_txt(wm_cols);
  99         printf("shift_window: >>>\n");
 100         if (row != 0) {
 101                 list_t *src = win->wm->row, *dst = NULL;
 102                 if (row < 0) dst = src->prev;
 103                 if (row > 0) dst = src->next;
 104                 if (src && dst) {
 105                         printf("swap: %p <-> %p\n", src->data, dst->data);
 106                         src->data = dst->data;
 107                         dst->data = win;
 108                         ((win_t*)src->data)->wm->row = src;
 109                         ((win_t*)dst->data)->wm->row = dst;
 110                         arrange(wm_cols);
 111                 }
 112         } else {
 113                 int onlyrow = !win->wm->row->prev && !win->wm->row->next;
 114                 list_t *src = win->wm->col, *dst = NULL;
 115                 if (col < 0) {
 116                         if (!src->prev && !onlyrow)
 117                                 wm_cols = list_insert(wm_cols, new0(col_t));
 118                         dst = src->prev;
 119                 }
 120                 if (col > 0) {
 121                         if (!src->next && !onlyrow)
 122                                 wm_cols = list_append(wm_cols, new0(col_t));
 123                         dst = src->next;
 124                 }
 125                 if (src && dst) {
 126                         col_t *scol = src->data;
 127                         col_t *dcol = dst->data;
 128                         scol->rows = list_remove(scol->rows, win->wm->row);
 129                         dcol->rows = list_insert(dcol->rows, win);
 130                         win->wm->row = dcol->rows;
 131                         win->wm->col = dst;
 132                         if (onlyrow) // delete column
 133                                 wm_cols = list_remove(wm_cols, src);
 134                         if (dcol->width == 0) // new column
 135                                 dcol->width = wm_root->w / (list_length(wm_cols)-1);
 136                         else
 137                                 win->h      = wm_root->h / (list_length(dcol->rows)-1);
 138                         arrange(wm_cols);
 139                 }
 140         }
 141         print_txt(wm_cols);
 142 }
 143
 144 static void shift_focus(win_t *win, int col, int row)
 145 {
 146         printf("shift_focus: %p - %+d,%+d\n", win, col, row);
 147         list_t *node  = NULL;
 148         if (row != 0) {
 149                 if (row < 0) node = win->wm->row->prev;
 150                 if (row > 0) node = win->wm->row->next;
 151         } else {
 152                 if (col < 0) node = win->wm->col->prev;
 153                 if (col > 0) node = win->wm->col->next;
 154                 if (node) node = ((col_t*)node->data)->rows;
 155         }
 156         if (node)
 157                 sys_focus(node->data);
 158 }
 159
 160 /* Window management functions */
 161 int wm_handle_key(win_t *win, Key_t key, mod_t mod, ptr_t ptr)
 162 {
 163         if (!win) return 0;
 164         //printf("wm_handle_key: %p - %x %x\n", win, key, mod);
 165
 166         /* Raise */
 167         if (key == key_f2)
 168                 return sys_focus(win), 1;
 169         if (key == key_f4)
 170                 return sys_raise(win), 1;
 171         if (key == key_f1 && mod.MODKEY)
 172                 sys_raise(win);
 173         if (key_mouse0 <= key && key <= key_mouse7)
 174                 sys_raise(win);
 175
 176         /* Key commands */
 177         if (mod.MODKEY && mod.shift) {
 178                 switch (key) {
 179                 case 'h': return shift_window(win,-1, 0), 1;
 180                 case 'j': return shift_window(win, 0,+1), 1;
 181                 case 'k': return shift_window(win, 0,-1), 1;
 182                 case 'l': return shift_window(win,+1, 0), 1;
 183                 default: break;
 184                 }
 185         }
 186         else if (mod.MODKEY) {
 187                 switch (key) {
 188                 case 'h': return shift_focus(win,-1, 0), 1;
 189                 case 'j': return shift_focus(win, 0,+1), 1;
 190                 case 'k': return shift_focus(win, 0,-1), 1;
 191                 case 'l': return shift_focus(win,+1, 0), 1;
 192                 default: break;
 193                 }
 194         }
 195
 196         /* Mouse movement */
 197         if (key_mouse0 <= key && key <= key_mouse7 && mod.up)
 198                 return set_mode(none,win,ptr), 1;
 199         else if (key == key_mouse1 && mod.MODKEY)
 200                 return set_mode(move,win,ptr), 1;
 201         else if (key == key_mouse3 && mod.MODKEY)
 202                 return set_mode(resize,win,ptr), 1;
 203
 204         /* Focus change */
 205         if (key == key_enter)
 206                 sys_focus(win);
 207
 208         return 0;
 209 }
 210
 211 int wm_handle_ptr(win_t *cwin, ptr_t ptr)
 212 {
 213         //printf("wm_handle_ptr: %p - %d,%d %d,%d (%d) -- \n",
 214         //              cwin, ptr.x, ptr.y, ptr.rx, ptr.ry, move_mode);
 215
 216         if (move_mode == none)
 217                 return 0;
 218
 219         /* Tiling */
 220         int dx = ptr.rx - move_prev.rx;
 221         int dy = ptr.ry - move_prev.ry;
 222         move_prev = ptr;
 223         if (move_mode == resize) {
 224                 list_t *row   = move_win->wm->row;
 225                 list_t *col   = move_win->wm->col;
 226                 list_t *lower = row->next;
 227                 list_t *right = col->next;
 228                 if (lower) {
 229                         ((win_t*)row->data)->h       += dy;
 230                         ((win_t*)lower->data)->h     -= dy;
 231                 }
 232                 if (right) {
 233                         ((col_t*)col->data)->width   += dx;
 234                         ((col_t*)right->data)->width -= dx;
 235                 }
 236                 arrange(wm_cols);
 237         }
 238
 239         /* Floating */
 240         //win_t *mwin = move_win;
 241         //int dx = ptr.rx - move_prev.rx;
 242         //int dy = ptr.ry - move_prev.ry;
 243         //move_prev = ptr;
 244         //if (move_mode == move)
 245         //      sys_move(mwin, mwin->x+dx, mwin->y+dy, mwin->w, mwin->h);
 246         //else if (move_mode == resize)
 247         //      sys_move(mwin, mwin->x, mwin->y, mwin->w+dx, mwin->h+dy);
 248
 249         return 0;
 250 }
 251
 252 void wm_insert(win_t *win)
 253 {
 254         printf("wm_insert: %p\n", win);
 255         print_txt(wm_cols);
 256
 257         /* Watch enter/leave */
 258         sys_watch(win, key_enter, MOD());
 259
 260         /* Add to screen */
 261         col_t *col = wm_cols->data;
 262         int nrows = list_length(col->rows);
 263         col->rows = list_insert(col->rows, win);
 264
 265         /* Setup window */
 266         win->wm = new0(win_wm_t);
 267         win->wm->col = wm_cols;
 268         win->wm->row = col->rows;
 269         if (nrows) win->h = wm_root->h / nrows;
 270
 271         /* Arrange */
 272         arrange(wm_cols);
 273         print_txt(wm_cols);
 274 }
 275
 276 void wm_remove(win_t *win)
 277 {
 278         printf("wm_remove: %p - (%p,%p)\n", win,
 279                         win->wm->col, win->wm->row);
 280         print_txt(wm_cols);
 281         col_t *col = win->wm->col->data;
 282         col->rows = list_remove(col->rows, win->wm->row);
 283         arrange(wm_cols);
 284         print_txt(wm_cols);
 285 }
 286
 287 void wm_init(win_t *root)
 288 {
 289         printf("wm_init: %p\n", root);
 290         wm_root = root;
 291         sys_watch(root, key_f1,     MOD(.MODKEY=1));
 292         sys_watch(root, key_mouse1, MOD(.MODKEY=1));
 293         sys_watch(root, key_mouse3, MOD(.MODKEY=1));
 294         sys_watch(root, key_enter,  MOD());
 295         Key_t keys[] = {'h', 'j', 'k', 'l'};
 296         for (int i = 0; i < countof(keys); i++) {
 297                 sys_watch(root, keys[i], MOD(.MODKEY=1));
 298                 sys_watch(root, keys[i], MOD(.MODKEY=1,.shift=1));
 299         }
 300         col_t *col = new0(col_t);
 301         col->group = stack;
 302         col->width = root->w;
 303         wm_cols = list_insert(wm_cols, col);
 304 }