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 0
  10 #define STACK  25
  11
  12 /* Enums */
  13 typedef enum {
  14         none, move, resize
  15 } drag_t;
  16
  17 typedef enum {
  18         split, stack, max, tab
  19 } mode_t;
  20
  21
  22 /* Window structure types */
  23 struct win_wm { };
  24
  25 typedef struct {
  26         win_t  *win;
  27         int     height;
  28 } row_t;
  29
  30 typedef struct {
  31         list_t *rows; // of row_t
  32         row_t  *row;
  33         int     width;
  34         mode_t  mode;
  35 } col_t;
  36
  37 typedef struct {
  38         list_t *cols; // of col_t
  39         col_t  *col;
  40         win_t  *geom;
  41 } dpy_t;
  42
  43 typedef struct {
  44         list_t *dpys; // of dpy_t
  45         dpy_t  *dpy;
  46         int     name;
  47 } tag_t;
  48
  49 typedef struct {
  50         list_t *tags; // of tag_t
  51         tag_t  *tag;
  52         win_t  *root;
  53         list_t *screens;
  54 } wm_t;
  55
  56 /* Mouse drag data */
  57 static drag_t  move_mode;
  58 static list_t *move_lrow;
  59 static list_t *move_lcol;
  60 static ptr_t   move_prev;
  61 static struct { int v, h; } move_dir;
  62
  63 /* Window management data */
  64 static wm_t  *wm;
  65 #define wm_win   wm->tag->dpy->col->row->win
  66 #define wm_row   wm->tag->dpy->col->row
  67 #define wm_col   wm->tag->dpy->col
  68 #define wm_dpy   wm->tag->dpy
  69 #define wm_tag   wm->tag
  70 #define wm_focus (wm_tag && wm_dpy && wm_col && wm_row ? wm_win : NULL)
  71
  72 #define ROW(l) ((row_t*)(l)->data)
  73 #define COL(l) ((col_t*)(l)->data)
  74 #define DPY(l) ((dpy_t*)(l)->data)
  75 #define TAG(l) ((tag_t*)(l)->data)
  76
  77 /* Helper functions */
  78 static int searchl(tag_t *tag, win_t *target,
  79                 list_t **_dpy, list_t **_col, list_t **_row)
  80 {
  81         for (list_t *dpy =     tag ->dpys; dpy; dpy = dpy->next)
  82         for (list_t *col = DPY(dpy)->cols; col; col = col->next)
  83         for (list_t *row = COL(col)->rows; row; row = row->next) {
  84                 win_t *win = ROW(row)->win;
  85                 if (win == target) {
  86                         if (_dpy) *_dpy = dpy;
  87                         if (_col) *_col = col;
  88                         if (_row) *_row = row;
  89                         return 1;
  90                 }
  91         }
  92         return 0;
  93 }
  94
  95 static int search(tag_t *tag, win_t *target,
  96                 dpy_t **_dpy, col_t **_col, row_t **_row)
  97 {
  98         list_t *dpy, *col, *row;
  99         if (searchl(tag, target, &dpy, &col, &row)) {
 100                 if (_dpy) *_dpy = DPY(dpy);
 101                 if (_col) *_col = COL(col);
 102                 if (_row) *_row = ROW(row);
 103                 return 1;
 104         }
 105         return 0;
 106 }
 107
 108 static void set_mode(win_t *win, mode_t mode)
 109 {
 110         col_t *col;
 111         search(wm_tag, win, NULL, &col, NULL);
 112         printf("set_mode: %p, %d -> %d\n",
 113                         col, col->mode, mode);
 114         col->mode = mode;
 115         if (col->mode == split)
 116                 for (list_t *cur = col->rows; cur; cur = cur->next) {
 117                         row_t *row = cur->data;
 118                         row->height = wm_dpy->geom->h;
 119                 }
 120         wm_update();
 121 }
 122
 123 static void set_focus(win_t *win)
 124 {
 125         /* - Only grab mouse button on unfocused window,
 126          *   this prevents stealing all mouse clicks from client windows,
 127          * - A better way may be to re-send mouse clicks to client windows
 128          *   using the return value from wm_handle_key */
 129         for (int i = key_mouse1; i < key_mouse7; i++) {
 130                 if (wm_focus)
 131                         sys_watch(wm_focus, i, MOD());
 132                 sys_unwatch(win, i, MOD());
 133         }
 134
 135         dpy_t *dpy; col_t *col; row_t *row;
 136         if (search(wm_tag, win, &dpy, &col, &row)) {
 137                 wm_dpy = dpy;
 138                 wm_col = col;
 139                 wm_row = row;
 140         }
 141         sys_focus(win);
 142 }
 143
 144 static void set_move(win_t *win, ptr_t ptr, drag_t drag)
 145 {
 146         printf("set_move: %d - %p@%d,%d\n",
 147                         drag, win, ptr.rx, ptr.ry);
 148         move_mode = drag;
 149         if (drag == move || drag == resize) {
 150                 searchl(wm_tag, win, NULL, &move_lcol, &move_lrow);
 151                 move_prev = ptr;
 152                 int my = win->y + (win->h/2);
 153                 int mx = win->x + (win->w/2);
 154                 move_dir.v = ptr.ry < my ? -1 : +1;
 155                 move_dir.h = ptr.rx < mx ? -1 : +1;
 156         }
 157 }
 158
 159 static void print_txt(void)
 160 {
 161         for (list_t *ltag = wm->tags; ltag; ltag = ltag->next) {
 162                 tag_t *tag = ltag->data;
 163                 printf("tag:       <%-9p [%p->%p] >%-9p  -  %d\n",
 164                                 ltag->prev, ltag, ltag->data, ltag->next, tag->name);
 165         for (list_t *ldpy = tag->dpys; ldpy; ldpy = ldpy->next) {
 166                 dpy_t *dpy  = ldpy->data;
 167                 win_t *geom = dpy->geom;
 168                 printf("  dpy:     <%-9p [%p->%p] >%-9p  -  %d,%d %dx%d\n",
 169                                 ldpy->prev, ldpy, ldpy->data, ldpy->next,
 170                                 geom->x, geom->y, geom->h, geom->w);
 171         for (list_t *lcol = dpy->cols; lcol; lcol = lcol->next) {
 172                 col_t *col = lcol->data;
 173                 printf("    col:   <%-9p [%p->%p] >%-9p  -  %dpx @ %d !!%p\n",
 174                                 lcol->prev, lcol, lcol->data, lcol->next,
 175                                 col->width, col->mode, col->row);
 176         for (list_t *lrow = col->rows; lrow; lrow = lrow->next) {
 177                 row_t *row = lrow->data;
 178                 win_t *win = row->win;
 179                 printf("      win: <%-9p [%p>>%p] >%-9p  -  %4dpx focus=%d%d\n",
 180                                 lrow->prev, lrow, win, lrow->next,
 181                                 win->h, col->row == row, wm_focus == win);
 182         } } } }
 183 }
 184
 185 static void cut_win(tag_t *tag, win_t *win)
 186 {
 187         list_t *ldpy, *lcol, *lrow;
 188         if (!searchl(tag, win, &ldpy, &lcol, &lrow))
 189                 return;
 190         col_t  *col  = COL(lcol);
 191         dpy_t  *dpy  = DPY(ldpy);
 192
 193         col->row  = lrow->prev ? lrow->prev->data :
 194                     lrow->next ? lrow->next->data : NULL;
 195         col->rows = list_remove(col->rows, lrow);
 196
 197         if (col->rows == NULL && (lcol->next || lcol->prev)) {
 198                 dpy->col  = lcol->prev ? lcol->prev->data :
 199                             lcol->next ? lcol->next->data : NULL;
 200                 dpy->cols = list_remove(dpy->cols, lcol);
 201         }
 202 }
 203
 204 static void put_win(win_t *win, dpy_t *dpy, col_t *col)
 205 {
 206         row_t *row = new0(row_t);
 207         row->win = win;
 208
 209         if (col == NULL) {
 210                 col = new0(col_t);
 211                 dpy->cols = list_insert(dpy->cols, col);
 212         }
 213
 214         int nrows = list_length(col->rows);
 215         if (col->row) {
 216                 list_t *prev = list_find(col->rows, col->row);
 217                 list_insert_after(prev, row);
 218         } else {
 219                 col->rows = list_insert(col->rows, row);
 220         }
 221         wm_tag->dpy           = dpy;
 222         wm_tag->dpy->col      = col;
 223         wm_tag->dpy->col->row = row;
 224
 225         row->height = dpy->geom->h / MAX(nrows,1);
 226         if (nrows == 0) {
 227                 int ncols = list_length(dpy->cols);
 228                 col->width = dpy->geom->w / MAX(ncols-1,1);
 229         }
 230 }
 231
 232 static void shift_window(win_t *win, int col, int row)
 233 {
 234         if (!win) return;
 235         printf("shift_window: %p - %+d,%+d\n", win, col, row);
 236         print_txt();
 237         printf("shift_window: >>>\n");
 238         list_t *ldpy, *lcol, *lrow;
 239         searchl(wm_tag, win, &ldpy, &lcol, &lrow);
 240         dpy_t *dpy = ldpy->data;
 241         if (row != 0) {
 242                 list_t *src = lrow, *dst = NULL;
 243                 if (row < 0) dst = src->prev;
 244                 if (row > 0) dst = src->next;
 245                 if (src && dst) {
 246                         printf("swap: %p <-> %p\n", src->data, dst->data);
 247                         row_t *tmp = src->data;
 248                         src->data = dst->data;
 249                         dst->data = tmp;
 250                         goto update;
 251                 }
 252         } else {
 253                 int onlyrow = !lrow->prev && !lrow->next;
 254                 list_t *src = lcol, *dst = NULL;
 255                 if (col < 0) {
 256                         if (src->prev) {
 257                                 dst = src->prev;
 258                         } else if (!onlyrow) {
 259                                 dpy->cols = list_insert(dpy->cols, new0(col_t));
 260                                 dst = src->prev;
 261                         } else if (ldpy->prev) {
 262                                 dpy = ldpy->prev->data;
 263                                 dst = list_last(dpy->cols);
 264                         } else {
 265                                 return;
 266                         }
 267                 }
 268                 if (col > 0) {
 269                         if (src->next) {
 270                                 dst = src->next;
 271                         } else if (!onlyrow) {
 272                                 dpy->cols = list_append(dpy->cols, new0(col_t));
 273                                 dst = src->next;
 274                         } else if (ldpy->next) {
 275                                 dpy = ldpy->next->data;
 276                                 dst = dpy->cols;
 277                         } else {
 278                                 return;
 279                         }
 280                 }
 281                 cut_win(wm_tag, win);
 282                 put_win(win, dpy, dst ? dst->data : NULL);
 283                 goto update;
 284         }
 285 update:
 286         print_txt();
 287         wm_update();
 288 }
 289
 290 static list_t *get_next(list_t *list, int forward)
 291 {
 292         list_t *next = forward ? list->next : list->prev;
 293         if (next == NULL) {
 294                 next = list;
 295                 while ((list = forward ? next->prev : next->next))
 296                         next = list;
 297         }
 298         return next;
 299 }
 300 static void shift_focus(int cols, int rows)
 301 {
 302         printf("shift_focus: %+d,%+d\n", cols, rows);
 303         if (rows != 0 && wm_focus) {
 304                 list_t *dpy, *col, *row;
 305                 searchl(wm_tag, wm_focus, &dpy, &col, &row);
 306                 row_t *next = get_next(row, rows > 0)->data;
 307                 set_focus(next->win);
 308                 if (COL(col)->mode != split)
 309                         wm_update();
 310         }
 311         if (cols != 0) {
 312                 list_t *dpy, *col, *row, *ndpy, *ncol = NULL;
 313                 if (wm_focus) {
 314                         searchl(wm_tag, wm_focus, &dpy, &col, &row);
 315                         ncol = cols > 0 ? col->next : col->prev;
 316                 } else {
 317                         dpy = list_find(wm_tag->dpys, wm_dpy);
 318                 }
 319                 if (ncol == NULL) {
 320                         ndpy = get_next(dpy, cols > 0);
 321                         ncol = cols > 0 ? DPY(ndpy)->cols :
 322                                 list_last(DPY(ndpy)->cols);
 323                         wm_dpy = ndpy->data;
 324                 }
 325                 if (ncol && COL(ncol) && COL(ncol)->row)
 326                         set_focus(COL(ncol)->row->win);
 327                 else
 328                         sys_focus(wm->root);
 329         }
 330 }
 331
 332 static tag_t *tag_new(list_t *screens, int name)
 333 {
 334         tag_t *tag = new0(tag_t);
 335         tag->name  = name;
 336         for (list_t *cur = screens; cur; cur = cur->next) {
 337                 dpy_t *dpy  = new0(dpy_t);
 338                 dpy->geom = cur->data;
 339                 tag->dpys = list_append(tag->dpys, dpy);
 340         }
 341         tag->dpy  = tag->dpys->data;
 342         return tag;
 343 }
 344
 345 static tag_t *tag_find(int name)
 346 {
 347         tag_t *tag = NULL;
 348         for (list_t *cur = wm->tags; cur; cur = cur->next)
 349                 if (name == TAG(cur)->name) {
 350                         tag = cur->data;
 351                         break;
 352                 }
 353         if (!tag) {
 354                 tag = tag_new(wm->screens, name);
 355                 wm->tags = list_insert(wm->tags, tag);
 356         }
 357         return tag;
 358 }
 359
 360 static void tag_set(win_t *win, int name)
 361 {
 362         printf("tag_set: %p %d\n", win, name);
 363         if (wm_tag->name == name)
 364                 return;
 365         cut_win(wm_tag, win);
 366         win_t *focus = wm_focus;
 367
 368         tag_t *tag = tag_find(name);
 369         put_win(win, tag->dpy, tag->dpy->col);
 370         set_focus(focus);
 371 }
 372
 373 /* Window management functions */
 374 void wm_update_dpy(dpy_t *dpy)
 375 {
 376         int  x=0,  y=0; // Current window top-left position
 377         int tx=0, ty=0; // Total x/y size
 378         int mx=0, my=0; // Maximum x/y size (screen size)
 379         int       sy=0; // Size of focused stack window
 380
 381         /* Scale horizontally */
 382         x  = dpy->geom->x;
 383         mx = dpy->geom->w - (list_length(dpy->cols)+1)*MARGIN;
 384         for (list_t *lx = dpy->cols; lx; lx = lx->next)
 385                 tx += COL(lx)->width;
 386         for (list_t *lx = dpy->cols; lx; lx = lx->next)
 387                 COL(lx)->width *= (float)mx / tx;
 388
 389         /* Scale each column vertically */
 390         for (list_t *lx = dpy->cols; lx; lx = lx->next) {
 391                 col_t *col = lx->data;
 392                 ty = 0;
 393                 for (list_t *ly = col->rows; ly; ly = ly->next)
 394                         ty += ROW(ly)->height;
 395                 y  = dpy->geom->y;
 396                 my = dpy->geom->h - (list_length(col->rows)+1)*MARGIN;
 397                 sy = my              - (list_length(col->rows)-1)*STACK;
 398                 for (list_t *ly = col->rows; ly; ly = ly->next) {
 399                         win_t *win = ROW(ly)->win;
 400                         win->h = ROW(ly)->height;
 401                         int height = 0;
 402                         switch (col->mode) {
 403                         case split:
 404                                 sys_move(win, x+MARGIN, y+MARGIN,
 405                                         col->width, win->h * ((float)my / ty));
 406                                 height = win->h;
 407                                 break;
 408                         case stack:
 409                                 height = col->row->win == win ? sy : STACK;
 410                                 sys_move(win, x+MARGIN, y+MARGIN,
 411                                         col->width, height);
 412                                 break;
 413                         case max:
 414                         case tab:
 415                                 sys_move(win, x+MARGIN, 0+MARGIN,
 416                                         col->width, dpy->geom->h-2*MARGIN);
 417                                 if (col->row->win == win)
 418                                         sys_raise(win);
 419                                 break;
 420                         }
 421                         y += height + MARGIN;
 422                         ROW(ly)->height = win->h;
 423                 }
 424                 x += col->width + MARGIN;
 425         }
 426 }
 427
 428 void wm_update(void)
 429 {
 430         /* Show/hide tags */
 431         for (list_t *ltag =       wm ->tags; ltag; ltag = ltag->next)
 432         for (list_t *ldpy = TAG(ltag)->dpys; ldpy; ldpy = ldpy->next)
 433         for (list_t *lcol = DPY(ldpy)->cols; lcol; lcol = lcol->next)
 434         for (list_t *lrow = COL(lcol)->rows; lrow; lrow = lrow->next)
 435                 sys_show(ROW(lrow)->win,
 436                         ltag->data == wm_tag ? st_show : st_hide);
 437
 438         /* Refrsh the display */
 439         for (list_t *ldpy = wm_tag->dpys; ldpy; ldpy = ldpy->next)
 440                 wm_update_dpy(ldpy->data);
 441         if (wm_focus)
 442                 set_focus(wm_focus);
 443 }
 444
 445 int wm_handle_key(win_t *win, Key_t key, mod_t mod, ptr_t ptr)
 446 {
 447         if (!win || win == wm_dpy->geom) return 0;
 448         //printf("wm_handle_key: %p - %x %c%c%c%c%c\n", win, key,
 449         //      mod.up    ? '^' : 'v',
 450         //      mod.alt   ? 'a' : '-',
 451         //      mod.ctrl  ? 'c' : '-',
 452         //      mod.shift ? 's' : '-',
 453         //      mod.win   ? 'w' : '-');
 454
 455         /* Mouse movement */
 456         if (key_mouse0 <= key && key <= key_mouse7 && mod.up)
 457                 return set_move(win,ptr,none), 1;
 458         else if (key == key_mouse1 && mod.MODKEY)
 459                 return set_move(win,ptr,move), 1;
 460         else if (key == key_mouse3 && mod.MODKEY)
 461                 return set_move(win,ptr,resize), 1;
 462
 463         /* Only handle key-down */
 464         if (mod.up)
 465                 return 0;
 466
 467         /* Misc */
 468         if (mod.MODKEY) {
 469                 if (key == key_f1) return sys_raise(win), 1;
 470                 if (key == key_f2) return set_focus(win), 1;
 471                 if (key == key_f3) return sys_show(win, st_show), 1;
 472                 if (key == key_f4) return sys_show(win, st_hide), 1;
 473                 if (key == key_f5) return wm_update(),    1;
 474                 if (key == key_f6) return print_txt(), 1;
 475         }
 476         if (key_mouse0 <= key && key <= key_mouse7)
 477                 sys_raise(win);
 478
 479         /* Movement commands */
 480         if (mod.MODKEY && mod.shift) {
 481                 switch (key) {
 482                 case 'h': return shift_window(wm_focus,-1, 0), 1;
 483                 case 'j': return shift_window(wm_focus, 0,+1), 1;
 484                 case 'k': return shift_window(wm_focus, 0,-1), 1;
 485                 case 'l': return shift_window(wm_focus,+1, 0), 1;
 486                 default: break;
 487                 }
 488         }
 489         else if (mod.MODKEY) {
 490                 switch (key) {
 491                 case 'h': return shift_focus(-1, 0), 1;
 492                 case 'j': return shift_focus( 0,+1), 1;
 493                 case 'k': return shift_focus( 0,-1), 1;
 494                 case 'l': return shift_focus(+1, 0), 1;
 495                 default: break;
 496                 }
 497         }
 498
 499         /* Column mode commands */
 500         if (mod.MODKEY) {
 501                 switch (key) {
 502                 case 'd': return set_mode(win, split), 1;
 503                 case 's': return set_mode(win, stack), 1;
 504                 case 'm': return set_mode(win, max),   1;
 505                 case 't': return set_mode(win, tab),   1;
 506                 default: break;
 507                 }
 508         }
 509
 510         /* Tag switching */
 511         if (mod.MODKEY && '0' <= key && key <= '9') {
 512                 int name = key - '0';
 513                 if (mod.shift) {
 514                         tag_set(win, name);
 515                 } else {
 516                         printf("tag_switch: %d\n", name);
 517                         wm_tag = tag_find(name);
 518                 }
 519                 wm_update();
 520         }
 521
 522         /* Focus change */
 523         if (key == key_enter)
 524                 return set_focus(win), 1;
 525
 526         if (key_mouse0 <= key && key <= key_mouse7)
 527                 return set_focus(win), 0;
 528
 529         /* Reset focus after after focus change,
 530          * not sure what is causing the focus change in the first place
 531          * but preventing that would be a better solution */
 532         if (key == key_focus)
 533                 sys_focus(wm_focus ?: wm->root);
 534
 535         return 0;
 536 }
 537
 538 int wm_handle_ptr(win_t *cwin, ptr_t ptr)
 539 {
 540         //printf("wm_handle_ptr: %p - %d,%d %d,%d (%d) -- \n",
 541         //              cwin, ptr.x, ptr.y, ptr.rx, ptr.ry, move_mode);
 542
 543         if (move_mode == none)
 544                 return 0;
 545
 546         /* Tiling */
 547         int dx = ptr.rx - move_prev.rx;
 548         int dy = ptr.ry - move_prev.ry;
 549         move_prev = ptr;
 550         if (move_mode == resize) {
 551                 list_t *vert = move_dir.v < 0 ? move_lrow->prev : move_lrow->next;
 552                 list_t *horz = move_dir.h < 0 ? move_lcol->prev : move_lcol->next;
 553                 if (vert) {
 554                         ROW(move_lrow)->height += move_dir.v * dy;
 555                         ROW(vert)->height      -= move_dir.v * dy;
 556                 }
 557                 if (horz) {
 558                         COL(move_lcol)->width  += move_dir.h * dx;
 559                         COL(horz)->width       -= move_dir.h * dx;
 560                 }
 561                 wm_update();
 562         }
 563
 564         /* Floating */
 565         //win_t *mwin = move_win;
 566         //int dx = ptr.rx - move_prev.rx;
 567         //int dy = ptr.ry - move_prev.ry;
 568         //move_prev = ptr;
 569         //if (move_mode == move)
 570         //      sys_move(mwin, mwin->x+dx, mwin->y+dy, mwin->w, mwin->h);
 571         //else if (move_mode == resize)
 572         //      sys_move(mwin, mwin->x, mwin->y, mwin->w+dx, mwin->h+dy);
 573
 574         return 0;
 575 }
 576
 577 void wm_insert(win_t *win)
 578 {
 579         printf("wm_insert: %p\n", win);
 580         print_txt();
 581
 582         /* Initialize window */
 583         win->wm = new0(win_wm_t);
 584         sys_watch(win, key_enter, MOD());
 585         sys_watch(win, key_focus, MOD());
 586
 587         /* Add to screen */
 588         put_win(win, wm_dpy, wm_col);
 589
 590         /* Arrange */
 591         wm_update();
 592         sys_focus(wm_focus);
 593         print_txt();
 594 }
 595
 596 void wm_remove(win_t *win)
 597 {
 598         printf("wm_remove: %p\n", win);
 599         print_txt();
 600         for (list_t *tag = wm->tags; tag; tag = tag->next)
 601                 cut_win(tag->data, win);
 602         if (wm_focus)
 603                 sys_focus(wm_focus);
 604         else
 605                 sys_focus(wm->root);
 606         wm_update();
 607         print_txt();
 608 }
 609
 610 void wm_init(win_t *root)
 611 {
 612         printf("wm_init: %p\n", root);
 613
 614         wm          = new0(wm_t);
 615         wm->root    = root;
 616         wm->screens = sys_info(root);
 617         wm->tag     = tag_new(wm->screens, 1);
 618         wm->tags    = list_insert(NULL, wm->tag);
 619
 620         Key_t keys_e[] = {key_enter, key_focus};
 621         Key_t keys_s[] = {'h', 'j', 'k', 'l',
 622                 '0', '1', '2', '3', '4', '5', '6', '7', '8', '9'};
 623         Key_t keys_m[] = {'h', 'j', 'k', 'l', 'd', 's', 'm', 't',
 624                 '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
 625                 key_f1, key_f2, key_f3, key_f4, key_f5, key_f6,
 626                 key_mouse1, key_mouse3};
 627         for (int i = 0; i < countof(keys_e); i++)
 628                 sys_watch(root, keys_e[i],  MOD());
 629         for (int i = 0; i < countof(keys_m); i++)
 630                 sys_watch(root, keys_m[i], MOD(.MODKEY=1));
 631         for (int i = 0; i < countof(keys_s); i++)
 632                 sys_watch(root, keys_s[i], MOD(.MODKEY=1,.shift=1));
 633 }