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 10
  10 #define STACK  20
  11
  12 /* Loca types */
  13 struct win_wm {
  14         list_t *col; // node in wm_cols
  15         list_t *row; // node in col->rows
  16 };
  17
  18 typedef enum {
  19         none, move, resize
  20 } drag_t;
  21
  22 typedef enum {
  23         split, stack, max, tab
  24 } mode_t;
  25
  26 typedef struct {
  27         int     width;
  28         mode_t  mode;
  29         win_t  *focus;
  30         list_t *rows;
  31 } col_t;
  32
  33 /* Mouse drag data */
  34 static drag_t move_mode;
  35 static win_t *move_win;
  36 static ptr_t  move_prev;
  37
  38 /* Window management data */
  39 static win_t  *wm_focus;
  40 static list_t *wm_cols;
  41 static win_t  *wm_root;
  42
  43 /* Helper functions */
  44 static void set_mode(win_t *win, mode_t mode)
  45 {
  46         if (!win->wm || !win->wm->col)
  47                 return;
  48         col_t *col = win->wm->col->data;
  49         printf("set_mode: %p (%p), %d -> %d\n",
  50                         win, col, col->mode, mode);
  51         col->mode = mode;
  52         if (col->mode == split)
  53                 for (list_t *cur = col->rows; cur; cur = cur->next) {
  54                         win_t *row = cur->data;
  55                         row->h = wm_root->h;
  56                 }
  57         wm_update();
  58 }
  59
  60 static void set_focus(win_t *win)
  61 {
  62         if (win->wm && win->wm->col)
  63                 ((col_t*)win->wm->col->data)->focus = win;
  64         wm_focus = win;
  65         sys_focus(win);
  66 }
  67
  68 static void set_move(drag_t drag, win_t *win, ptr_t ptr)
  69 {
  70         printf("set_move: %d - %p@%d,%d\n",
  71                         drag, win, ptr.rx, ptr.ry);
  72         move_mode = drag;
  73         if (drag == move || drag == resize) {
  74                 move_win  = win;
  75                 move_prev = ptr;
  76         }
  77 }
  78
  79 static void print_txt(list_t *cols)
  80 {
  81         for (list_t *lcol = cols; lcol; lcol = lcol->next) {
  82                 col_t *col = lcol->data;
  83                 printf("col:\t           <%-9p [%-19p] >%-9p  -  %dpx @ %d !!%p\n",
  84                                 ( lcol->prev ? lcol->prev->data : NULL ),
  85                                 col,
  86                                 ( lcol->next ? lcol->next->data : NULL ),
  87                                 col->width, col->mode, col->focus);
  88                 for (list_t *lrow = col->rows; lrow; lrow = lrow->next) {
  89                         win_t *win = lrow->data;
  90                         printf("  win:\t^%-9p <%-9p [%p=%p] >%-9p  -  %4dpx focus=%d%d\n",
  91                                         win->wm->col->data,
  92                                         (win->wm->row->prev ? win->wm->row->prev->data : NULL),
  93                                         win->wm->row->data, win,
  94                                         (win->wm->row->next ? win->wm->row->next->data : NULL),
  95                                         win->h, col->focus == win, wm_focus == win);
  96                 }
  97         }
  98 }
  99
 100 static void cut_window(win_t *win)
 101 {
 102         list_t *lrow = win->wm->row;
 103         list_t *lcol = win->wm->col;
 104         col_t  *col  = lcol->data;
 105
 106         col->focus = lrow->prev ? lrow->prev->data  :
 107                      lrow->next ? lrow->next->data  : NULL;
 108
 109         wm_focus   = col->focus ? col->focus        :
 110                      lcol->prev ? ((col_t*)lcol->prev->data)->focus :
 111                      lcol->next ? ((col_t*)lcol->next->data)->focus : NULL;
 112
 113         col->rows  = list_remove(col->rows, lrow);
 114         if (col->rows == NULL && (lcol->next || lcol->prev))
 115                 wm_cols = list_remove(wm_cols, lcol);
 116 }
 117
 118 static void put_window(win_t *win, list_t *lcol)
 119 {
 120         if (lcol == NULL)
 121                 lcol = wm_cols = list_insert(wm_cols, new0(col_t));
 122
 123         col_t *col = lcol->data;
 124         int nrows = list_length(col->rows);
 125         if (col->focus) {
 126                 list_insert_after(col->focus->wm->row, win);
 127                 win->wm->row = col->focus->wm->row->next;
 128         } else {
 129                 col->rows = list_insert(col->rows, win);
 130                 win->wm->row = col->rows;
 131         }
 132         win->wm->col = lcol;
 133         col->focus   = win;
 134         wm_focus     = win;
 135
 136         win->h = wm_root->h / MAX(nrows,1);
 137         if (nrows == 0) {
 138                 int ncols = list_length(wm_cols);
 139                 col->width = wm_root->w / MAX(ncols-1,1);
 140         }
 141 }
 142
 143 static void shift_window(win_t *win, int col, int row)
 144 {
 145         printf("shift_window: %p - %+d,%+d\n", win, col, row);
 146         print_txt(wm_cols);
 147         printf("shift_window: >>>\n");
 148         if (row != 0) {
 149                 list_t *src = win->wm->row, *dst = NULL;
 150                 if (row < 0) dst = src->prev;
 151                 if (row > 0) dst = src->next;
 152                 if (src && dst) {
 153                         printf("swap: %p <-> %p\n", src->data, dst->data);
 154                         src->data = dst->data;
 155                         dst->data = win;
 156                         ((win_t*)src->data)->wm->row = src;
 157                         ((win_t*)dst->data)->wm->row = dst;
 158                         wm_update();
 159                 }
 160         } else {
 161                 int onlyrow = !win->wm->row->prev && !win->wm->row->next;
 162                 list_t *src = win->wm->col, *dst = NULL;
 163                 if (col < 0) {
 164                         if (!src->prev && !onlyrow)
 165                                 wm_cols = list_insert(wm_cols, new0(col_t));
 166                         dst = src->prev;
 167                 }
 168                 if (col > 0) {
 169                         if (!src->next && !onlyrow)
 170                                 wm_cols = list_append(wm_cols, new0(col_t));
 171                         dst = src->next;
 172                 }
 173                 if (src && dst) {
 174                         cut_window(win);
 175                         put_window(win, dst);
 176                         wm_update();
 177                 }
 178         }
 179         print_txt(wm_cols);
 180 }
 181
 182 static void shift_focus(win_t *win, int col, int row)
 183 {
 184         printf("shift_focus: %p - %+d,%+d\n", win, col, row);
 185         list_t *node  = NULL;
 186         int update = 0;
 187         if (row != 0) {
 188                 if (row < 0) node = win->wm->row->prev;
 189                 if (row > 0) node = win->wm->row->next;
 190                 if (((col_t*)win->wm->col->data)->mode != split)
 191                         update = 1;
 192         } else {
 193                 if (col < 0) node = win->wm->col->prev;
 194                 if (col > 0) node = win->wm->col->next;
 195                 if (node) {
 196                         col_t *col = node->data;
 197                         node = col->focus->wm->row;
 198                 }
 199         }
 200         if (node)
 201                 set_focus(node->data);
 202         if (update)
 203                 wm_update();
 204 }
 205
 206 /* Window management functions */
 207 void wm_update(void)
 208 {
 209         int  x=0,  y=0; // Current window top-left position
 210         int tx=0, ty=0; // Total x/y size
 211         int mx=0, my=0; // Maximum x/y size (screen size)
 212         int       sy=0; // Size of focused stack window
 213
 214         /* Scale horizontally */
 215         x  = wm_root->x;
 216         mx = wm_root->w - (list_length(wm_cols)+1)*MARGIN;
 217         for (list_t *lx = wm_cols; lx; lx = lx->next)
 218                 tx += ((col_t*)lx->data)->width;
 219         for (list_t *lx = wm_cols; lx; lx = lx->next)
 220                 ((col_t*)lx->data)->width *= (float)mx / tx;
 221
 222         /* Scale each column vertically */
 223         for (list_t *lx = wm_cols; lx; lx = lx->next) {
 224                 col_t *col = lx->data;
 225                 ty = 0;
 226                 for (list_t *ly = col->rows; ly; ly = ly->next)
 227                         ty += ((win_t*)ly->data)->h;
 228                 y  = wm_root->y;
 229                 my = wm_root->h - (list_length(col->rows)+1)*MARGIN;
 230                 sy = my         - (list_length(col->rows)-1)*STACK;
 231                 for (list_t *ly = col->rows; ly; ly = ly->next) {
 232                         win_t *win = ly->data;
 233                         int height = 0;
 234                         switch (col->mode) {
 235                         case split:
 236                                 sys_move(win, x+MARGIN, y+MARGIN,
 237                                         col->width, win->h * ((float)my / ty));
 238                                 height = win->h;
 239                                 break;
 240                         case stack:
 241                                 height = col->focus == win ? sy : STACK;
 242                                 sys_move(win, x+MARGIN, y+MARGIN,
 243                                         col->width, height);
 244                                 break;
 245                         case max:
 246                         case tab:
 247                                 sys_move(win, x+MARGIN, 0+MARGIN,
 248                                         col->width, wm_root->h-2*MARGIN);
 249                                 if (col->focus == win)
 250                                         sys_raise(win);
 251                                 break;
 252                         }
 253                         y += height + MARGIN;
 254                 }
 255                 x += col->width + MARGIN;
 256         }
 257 }
 258
 259 int wm_handle_key(win_t *win, Key_t key, mod_t mod, ptr_t ptr)
 260 {
 261         if (!win || win == wm_root) return 0;
 262         //printf("wm_handle_key: %p - %x %x\n", win, key, mod);
 263
 264         /* Raise */
 265         if (key == key_f2)
 266                 return set_focus(win), 1;
 267         if (key == key_f4)
 268                 return sys_raise(win), 1;
 269         if (key == key_f1 && mod.MODKEY)
 270                 sys_raise(win);
 271         if (key == key_f12 && mod.MODKEY)
 272                 print_txt(wm_cols);
 273         if (key_mouse0 <= key && key <= key_mouse7)
 274                 sys_raise(win);
 275
 276         /* Movement commands */
 277         if (mod.MODKEY && mod.shift) {
 278                 switch (key) {
 279                 case 'h': return shift_window(win,-1, 0), 1;
 280                 case 'j': return shift_window(win, 0,+1), 1;
 281                 case 'k': return shift_window(win, 0,-1), 1;
 282                 case 'l': return shift_window(win,+1, 0), 1;
 283                 default: break;
 284                 }
 285         }
 286         else if (mod.MODKEY) {
 287                 switch (key) {
 288                 case 'h': return shift_focus(win,-1, 0), 1;
 289                 case 'j': return shift_focus(win, 0,+1), 1;
 290                 case 'k': return shift_focus(win, 0,-1), 1;
 291                 case 'l': return shift_focus(win,+1, 0), 1;
 292                 default: break;
 293                 }
 294         }
 295
 296         /* Column mode commands */
 297         if (mod.MODKEY) {
 298                 switch (key) {
 299                 case 'd': return set_mode(win, split), 1;
 300                 case 's': return set_mode(win, stack), 1;
 301                 case 'm': return set_mode(win, max),   1;
 302                 case 't': return set_mode(win, tab),   1;
 303                 default: break;
 304                 }
 305         }
 306
 307         /* Mouse movement */
 308         if (key_mouse0 <= key && key <= key_mouse7 && mod.up)
 309                 return set_move(none,win,ptr), 1;
 310         else if (key == key_mouse1 && mod.MODKEY)
 311                 return set_move(move,win,ptr), 1;
 312         else if (key == key_mouse3 && mod.MODKEY)
 313                 return set_move(resize,win,ptr), 1;
 314
 315         /* Focus change */
 316         if (key == key_enter || (key_mouse0 <= key && key <= key_mouse7))
 317                 set_focus(win);
 318
 319         /* Reset focus after after focus change,
 320          * not sure what is causing the focus change in the first place
 321          * but preventing that would be a better solution */
 322         if (key == key_focus)
 323                 set_focus(wm_focus);
 324
 325         return 0;
 326 }
 327
 328 int wm_handle_ptr(win_t *cwin, ptr_t ptr)
 329 {
 330         //printf("wm_handle_ptr: %p - %d,%d %d,%d (%d) -- \n",
 331         //              cwin, ptr.x, ptr.y, ptr.rx, ptr.ry, move_mode);
 332
 333         if (move_mode == none)
 334                 return 0;
 335
 336         /* Tiling */
 337         int dx = ptr.rx - move_prev.rx;
 338         int dy = ptr.ry - move_prev.ry;
 339         move_prev = ptr;
 340         if (move_mode == resize) {
 341                 list_t *row   = move_win->wm->row;
 342                 list_t *col   = move_win->wm->col;
 343                 list_t *lower = row->next;
 344                 list_t *right = col->next;
 345                 if (lower) {
 346                         ((win_t*)row->data)->h       += dy;
 347                         ((win_t*)lower->data)->h     -= dy;
 348                 }
 349                 if (right) {
 350                         ((col_t*)col->data)->width   += dx;
 351                         ((col_t*)right->data)->width -= dx;
 352                 }
 353                 wm_update();
 354         }
 355
 356         /* Floating */
 357         //win_t *mwin = move_win;
 358         //int dx = ptr.rx - move_prev.rx;
 359         //int dy = ptr.ry - move_prev.ry;
 360         //move_prev = ptr;
 361         //if (move_mode == move)
 362         //      sys_move(mwin, mwin->x+dx, mwin->y+dy, mwin->w, mwin->h);
 363         //else if (move_mode == resize)
 364         //      sys_move(mwin, mwin->x, mwin->y, mwin->w+dx, mwin->h+dy);
 365
 366         return 0;
 367 }
 368
 369 void wm_insert(win_t *win)
 370 {
 371         printf("wm_insert: %p\n", win);
 372         print_txt(wm_cols);
 373
 374         /* Initialize window */
 375         win->wm = new0(win_wm_t);
 376         sys_watch(win, key_enter, MOD());
 377         sys_watch(win, key_focus, MOD());
 378
 379         /* Add to screen */
 380         list_t *lcol = wm_focus && wm_focus->wm ?
 381                 wm_focus->wm->col : wm_cols;
 382         put_window(win, lcol);
 383
 384         /* Arrange */
 385         wm_update();
 386         sys_focus(wm_focus);
 387         print_txt(wm_cols);
 388 }
 389
 390 void wm_remove(win_t *win)
 391 {
 392         printf("wm_remove: %p - (%p,%p)\n", win,
 393                         win->wm->col, win->wm->row);
 394         print_txt(wm_cols);
 395         cut_window(win);
 396         if (wm_focus)
 397                 sys_focus(wm_focus);
 398         else
 399                 sys_focus(wm_root);
 400         wm_update();
 401         print_txt(wm_cols);
 402 }
 403
 404 void wm_init(win_t *root)
 405 {
 406         printf("wm_init: %p\n", root);
 407         wm_root = root;
 408         sys_watch(root, key_f1,     MOD(.MODKEY=1));
 409         sys_watch(root, key_f12,    MOD(.MODKEY=1));
 410         sys_watch(root, key_mouse1, MOD(.MODKEY=1));
 411         sys_watch(root, key_mouse3, MOD(.MODKEY=1));
 412         sys_watch(root, key_enter,  MOD());
 413         sys_watch(root, key_focus,  MOD());
 414         Key_t keys_m[] = {'h', 'j', 'k', 'l', 'd', 's', 'm', 't'};
 415         Key_t keys_s[] = {'h', 'j', 'k', 'l'};
 416         for (int i = 0; i < countof(keys_m); i++)
 417                 sys_watch(root, keys_m[i], MOD(.MODKEY=1));
 418         for (int i = 0; i < countof(keys_s); i++)
 419                 sys_watch(root, keys_s[i], MOD(.MODKEY=1,.shift=1));
 420         for (int i = key_mouse1; i < key_mouse7; i++)
 421                 sys_watch(root, i, MOD());
 422 }