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