ref: f5cc6fbe3a7bcf8bdb002c646ddd519014afafd2
dir: /appl/wm/wm.b/
implement Wm; include "sys.m"; sys: Sys; include "draw.m"; draw: Draw; Screen, Display, Image, Rect, Point, Wmcontext, Pointer: import draw; include "wmsrv.m"; wmsrv: Wmsrv; Window, Client: import wmsrv; include "tk.m"; include "wmclient.m"; wmclient: Wmclient; include "string.m"; str: String; include "sh.m"; include "winplace.m"; winplace: Winplace; Wm: module { init: fn(ctxt: ref Draw->Context, argv: list of string); }; Ptrstarted, Kbdstarted, Controlstarted, Controller, Fixedorigin: con 1<<iota; Bdwidth: con 3; Sminx, Sminy, Smaxx, Smaxy: con iota; Minx, Miny, Maxx, Maxy: con 1<<iota; Background: con int 16r777777FF; screen: ref Screen; display: ref Display; ptrfocus: ref Client; kbdfocus: ref Client; controller: ref Client; allowcontrol := 1; fakekbd: chan of string; fakekbdin: chan of string; buttons := 0; badmodule(p: string) { sys->fprint(sys->fildes(2), "wm: cannot load %s: %r\n", p); raise "fail:bad module"; } init(ctxt: ref Draw->Context, argv: list of string) { sys = load Sys Sys->PATH; draw = load Draw Draw->PATH; if(draw == nil) badmodule(Draw->PATH); str = load String String->PATH; if(str == nil) badmodule(String->PATH); wmsrv = load Wmsrv Wmsrv->PATH; if(wmsrv == nil) badmodule(Wmsrv->PATH); wmclient = load Wmclient Wmclient->PATH; if(wmclient == nil) badmodule(Wmclient->PATH); wmclient->init(); winplace = load Winplace Winplace->PATH; if(winplace == nil) badmodule(Winplace->PATH); winplace->init(); sys->pctl(Sys->NEWPGRP|Sys->FORKNS, nil); if (ctxt == nil) ctxt = wmclient->makedrawcontext(); display = ctxt.display; buts := Wmclient->Appl; if(ctxt.wm == nil) buts = Wmclient->Plain; win := wmclient->window(ctxt, "Wm", buts); wmclient->win.reshape(((0, 0), (100, 100))); wmclient->win.onscreen("place"); if(win.image == nil){ sys->fprint(sys->fildes(2), "wm: cannot get image to draw on\n"); raise "fail:no image"; } wmclient->win.startinput("kbd" :: "ptr" :: nil); wmctxt := win.ctxt; screen = makescreen(win.image); (clientwm, join, req) := wmsrv->init(); clientctxt := ref Draw->Context(ctxt.display, nil, clientwm); wmrectIO := sys->file2chan("/chan", "wmrect"); if(wmrectIO == nil) fatal(sys->sprint("cannot make /chan/wmrect: %r")); sync := chan of string; argv = tl argv; if(argv == nil) argv = "wm/toolbar" :: nil; spawn command(clientctxt, argv, sync); if((e := <-sync) != nil) fatal("cannot run command: " + e); fakekbd = chan of string; for(;;) alt { c := <-win.ctl or c = <-wmctxt.ctl => # XXX could implement "pleaseexit" in order that # applications can raise a warning message before # they're unceremoniously dumped. if(c == "exit") for(z := wmsrv->top(); z != nil; z = z.znext) z.ctl <-= "exit"; wmclient->win.wmctl(c); if(win.image != screen.image) reshaped(win); c := <-wmctxt.kbd or c = int <-fakekbd => if(kbdfocus != nil) kbdfocus.kbd <-= c; p := <-wmctxt.ptr => if(wmclient->win.pointer(*p)) break; if(p.buttons && (ptrfocus == nil || buttons == 0)){ c := wmsrv->find(p.xy); if(c != nil){ ptrfocus = c; c.ctl <-= "raise"; setfocus(win, c); } } if(ptrfocus != nil && (ptrfocus.flags & Ptrstarted) != 0){ # inside currently selected client or it had button down last time (might have come up) buttons = p.buttons; ptrfocus.ptr <-= p; break; } buttons = 0; (c, rc) := <-join => rc <-= nil; # new client; inform it of the available screen rectangle. # XXX do we need to do this now we've got wmrect? c.ctl <-= "rect " + r2s(screen.image.r); if(allowcontrol){ controller = c; c.flags |= Controller; allowcontrol = 0; }else controlevent("newclient " + string c.id); c.cursor = "cursor"; (c, data, rc) := <-req => # if client leaving if(rc == nil){ c.remove(); if(c.stop == nil) break; if(c == ptrfocus) ptrfocus = nil; if(c == kbdfocus) kbdfocus = nil; if(c == controller) controller = nil; controlevent("delclient " + string c.id); for(z := wmsrv->top(); z != nil; z = z.znext) if(z.flags & Kbdstarted) break; setfocus(win, z); c.stop <-= 1; break; } err := handlerequest(win, wmctxt, c, string data); n := len data; if(err != nil) n = -1; alt{ rc <-= (n, err) =>; * =>; } (nil, nil, nil, wc) := <-wmrectIO.write => if(wc == nil) break; alt{ wc <-= (0, "cannot write") =>; * =>; } (off, nil, nil, rc) := <-wmrectIO.read => if(rc == nil) break; d := array of byte r2s(screen.image.r); if(off > len d) off = len d; alt{ rc <-= (d[off:], nil) =>; * =>; } } } handlerequest(win: ref Wmclient->Window, wmctxt: ref Wmcontext, c: ref Client, req: string): string { #sys->print("%d: %s\n", c.id, req); args := str->unquoted(req); if(args == nil) return "no request"; n := len args; if(req[0] == '!' && n < 3) return "bad arg count"; case hd args { "key" => # XXX should we restrict this capability to certain clients only? if(n != 2) return "bad arg count"; if(fakekbdin == nil){ fakekbdin = chan of string; spawn bufferproc(fakekbdin, fakekbd); } fakekbdin <-= hd tl args; "ptr" => # ptr x y if(n != 3) return "bad arg count"; if(ptrfocus != c) return "cannot move pointer"; e := wmclient->win.wmctl(req); if(e == nil){ c.ptr <-= nil; # flush queue c.ptr <-= ref Pointer(buttons, (int hd tl args, int hd tl tl args), sys->millisec()); } "cursor" => # cursor hotx hoty dx dy data if(n != 6 && n != 1) return "bad arg count"; c.cursor = req; if(ptrfocus == c || kbdfocus == c) return wmclient->win.wmctl(c.cursor); "start" => if(n != 2) return "bad arg count"; case hd tl args { "mouse" or "ptr" => c.flags |= Ptrstarted; "kbd" => c.flags |= Kbdstarted; # XXX this means that any new window grabs the focus from the current # application, but usually you want this to happen... how can we distinguish # the two cases? setfocus(win, c); "control" => if((c.flags & Controller) == 0) return "control not available"; c.flags |= Controlstarted; * => return "unknown input source"; } "!reshape" => # reshape tag reqid rect [how] # XXX allow "how" to specify that the origin of the window is never # changed - a new window will be created instead. if(n < 7) return "bad arg count"; args = tl args; tag := hd args; args = tl args; args = tl args; # skip reqid r: Rect; r.min.x = int hd args; args = tl args; r.min.y = int hd args; args = tl args; r.max.x = int hd args; args = tl args; r.max.y = int hd args; args = tl args; if(args != nil){ case hd args{ "onscreen" => r = fitrect(r, screen.image.r); "place" => r = fitrect(r, screen.image.r); r = newrect(r, screen.image.r); "exact" => ; "max" => r = screen.image.r; # XXX don't obscure toolbar? * => return "unkown placement method"; } } return reshape(c, tag, r); "delete" => # delete tag if(tl args == nil) return "tag required"; c.setimage(hd tl args, nil); if(c.wins == nil && c == kbdfocus) setfocus(win, nil); "raise" => c.top(); "lower" => c.bottom(); "!move" or "!size" => # !move tag reqid startx starty # !size tag reqid mindx mindy ismove := hd args == "!move"; if(n < 3) return "bad arg count"; args = tl args; tag := hd args; args = tl args; args = tl args; # skip reqid w := c.window(tag); if(w == nil) return "no such tag"; if(ismove){ if(n != 5) return "bad arg count"; return dragwin(wmctxt.ptr, c, w, Point(int hd args, int hd tl args).sub(w.r.min)); }else{ if(n != 5) return "bad arg count"; sizewin(wmctxt.ptr, c, w, Point(int hd args, int hd tl args)); } "fixedorigin" => c.flags |= Fixedorigin; "rect" => ; "kbdfocus" => if(n != 2) return "bad arg count"; if(int hd tl args) setfocus(win, c); else if(c == kbdfocus) setfocus(win, nil); # controller specific messages: "request" => # can be used to test for control. if((c.flags & Controller) == 0) return "you are not in control"; "ctl" => # ctl id msg if((c.flags & Controlstarted) == 0) return "invalid request"; if(n < 3) return "bad arg count"; id := int hd tl args; for(z := wmsrv->top(); z != nil; z = z.znext) if(z.id == id) break; if(z == nil) return "no such client"; z.ctl <-= str->quoted(tl tl args); "endcontrol" => if(c != controller) return "invalid request"; controller = nil; allowcontrol = 1; c.flags &= ~(Controlstarted | Controller); * => if(c == controller || controller == nil || (controller.flags & Controlstarted) == 0) return "unknown control request"; controller.ctl <-= "request " + string c.id + " " + req; } return nil; } Fix: con 1000; # the window manager window has been reshaped; # allocate a new screen, and move all the reshaped(win: ref Wmclient->Window) { oldr := screen.image.r; newr := win.image.r; mx := Fix; if(oldr.dx() > 0) mx = newr.dx() * Fix / oldr.dx(); my := Fix; if(oldr.dy() > 0) my = newr.dy() * Fix / oldr.dy(); screen = makescreen(win.image); for(z := wmsrv->top(); z != nil; z = z.znext){ for(wl := z.wins; wl != nil; wl = tl wl){ w := hd wl; w.img = nil; nr := w.r.subpt(oldr.min); nr.min.x = nr.min.x * mx / Fix; nr.min.y = nr.min.y * my / Fix; nr.max.x = nr.max.x * mx / Fix; nr.max.y = nr.max.y * my / Fix; nr = nr.addpt(newr.min); w.img = screen.newwindow(nr, Draw->Refbackup, Draw->Nofill); # XXX check for creation failure w.r = nr; z.ctl <-= sys->sprint("!reshape %q -1 %s", w.tag, r2s(nr)); z.ctl <-= "rect " + r2s(newr); } } } controlevent(e: string) { if(controller != nil && (controller.flags & Controlstarted)) controller.ctl <-= e; } dragwin(ptr: chan of ref Pointer, c: ref Client, w: ref Window, off: Point): string { if(buttons == 0) return "too late"; p: ref Pointer; scr := screen.image.r; Margin: con 10; do{ p = <-ptr; org := p.xy.sub(off); if(org.y < scr.min.y) org.y = scr.min.y; else if(org.y > scr.max.y - Margin) org.y = scr.max.y - Margin; if(org.x < scr.min.x && org.x + w.r.dx() < scr.min.x + Margin) org.x = scr.min.x + Margin - w.r.dx(); else if(org.x > scr.max.x - Margin) org.x = scr.max.x - Margin; w.img.origin(w.img.r.min, org); } while (p.buttons != 0); c.ptr <-= p; buttons = 0; r: Rect; r.min = p.xy.sub(off); r.max = r.min.add(w.r.size()); if(r.eq(w.r)) return "not moved"; reshape(c, w.tag, r); return nil; } sizewin(ptrc: chan of ref Pointer, c: ref Client, w: ref Window, minsize: Point): string { borders := array[4] of ref Image; showborders(borders, w.r, Minx|Maxx|Miny|Maxy); screen.image.flush(Draw->Flushnow); while((ptr := <-ptrc).buttons == 0) ; xy := ptr.xy; move, show: int; offset := Point(0, 0); r := w.r; show = Minx|Miny|Maxx|Maxy; if(xy.in(w.r) == 0){ r = (xy, xy); move = Maxx|Maxy; }else { if(xy.x < (r.min.x+r.max.x)/2){ move=Minx; offset.x = xy.x - r.min.x; }else{ move=Maxx; offset.x = xy.x - r.max.x; } if(xy.y < (r.min.y+r.max.y)/2){ move |= Miny; offset.y = xy.y - r.min.y; }else{ move |= Maxy; offset.y = xy.y - r.max.y; } } return reshape(c, w.tag, sweep(ptrc, r, offset, borders, move, show, minsize)); } reshape(c: ref Client, tag: string, r: Rect): string { w := c.window(tag); # if window hasn't changed size, then just change its origin and use the same image. if((c.flags & Fixedorigin) == 0 && w != nil && w.r.size().eq(r.size())){ c.setorigin(tag, r.min); } else { img := screen.newwindow(r, Draw->Refbackup, Draw->Nofill); if(img == nil) return sys->sprint("window creation failed: %r"); if(c.setimage(tag, img) == -1) return "can't do two at once"; } c.top(); return nil; } sweep(ptr: chan of ref Pointer, r: Rect, offset: Point, borders: array of ref Image, move, show: int, min: Point): Rect { while((p := <-ptr).buttons != 0){ xy := p.xy.sub(offset); if(move&Minx) r.min.x = xy.x; if(move&Miny) r.min.y = xy.y; if(move&Maxx) r.max.x = xy.x; if(move&Maxy) r.max.y = xy.y; showborders(borders, r, show); } r = r.canon(); if(r.min.y < screen.image.r.min.y){ r.min.y = screen.image.r.min.y; r = r.canon(); } if(r.dx() < min.x){ if(move & Maxx) r.max.x = r.min.x + min.x; else r.min.x = r.max.x - min.x; } if(r.dy() < min.y){ if(move & Maxy) r.max.y = r.min.y + min.y; else { r.min.y = r.max.y - min.y; if(r.min.y < screen.image.r.min.y){ r.min.y = screen.image.r.min.y; r.max.y = r.min.y + min.y; } } } return r; } showborders(b: array of ref Image, r: Rect, show: int) { r = r.canon(); b[Sminx] = showborder(b[Sminx], show&Minx, (r.min, (r.min.x+Bdwidth, r.max.y))); b[Sminy] = showborder(b[Sminy], show&Miny, ((r.min.x+Bdwidth, r.min.y), (r.max.x-Bdwidth, r.min.y+Bdwidth))); b[Smaxx] = showborder(b[Smaxx], show&Maxx, ((r.max.x-Bdwidth, r.min.y), (r.max.x, r.max.y))); b[Smaxy] = showborder(b[Smaxy], show&Maxy, ((r.min.x+Bdwidth, r.max.y-Bdwidth), (r.max.x-Bdwidth, r.max.y))); } showborder(b: ref Image, show: int, r: Rect): ref Image { if(!show) return nil; if(b != nil && b.r.size().eq(r.size())) b.origin(r.min, r.min); else b = screen.newwindow(r, Draw->Refbackup, Draw->Red); return b; } r2s(r: Rect): string { return string r.min.x + " " + string r.min.y + " " + string r.max.x + " " + string r.max.y; } # XXX for consideration: # do not allow applications to grab the keyboard focus # unless there is currently no keyboard focus... # but what about launching a new app from the taskbar: # surely we should allow that to grab the focus? setfocus(win: ref Wmclient->Window, new: ref Client) { old := kbdfocus; if(old == new) return; if(new == nil) wmclient->win.wmctl("cursor"); else if(old == nil || old.cursor != new.cursor) wmclient->win.wmctl(new.cursor); if(new != nil && (new.flags & Kbdstarted) == 0) return; if(old != nil) old.ctl <-= "haskbdfocus 0"; if(new != nil){ new.ctl <-= "raise"; new.ctl <-= "haskbdfocus 1"; kbdfocus = new; } else kbdfocus = nil; } makescreen(img: ref Image): ref Screen { screen = Screen.allocate(img, img.display.color(Background), 0); img.draw(img.r, screen.fill, nil, screen.fill.r.min); return screen; } kill(pid: int, note: string): int { fd := sys->open("/prog/"+string pid+"/ctl", Sys->OWRITE); if(fd == nil || sys->fprint(fd, "%s", note) < 0) return -1; return 0; } fatal(s: string) { sys->fprint(sys->fildes(2), "wm: %s\n", s); kill(sys->pctl(0, nil), "killgrp"); raise "fail:error"; } # fit a window rectangle to the available space. # try to preserve requested location if possible. # make sure that the window is no bigger than # the screen, and that its top and left-hand edges # will be visible at least. fitrect(w, r: Rect): Rect { if(w.dx() > r.dx()) w.max.x = w.min.x + r.dx(); if(w.dy() > r.dy()) w.max.y = w.min.y + r.dy(); size := w.size(); if (w.max.x > r.max.x) (w.min.x, w.max.x) = (r.min.x - size.x, r.max.x - size.x); if (w.max.y > r.max.y) (w.min.y, w.max.y) = (r.min.y - size.y, r.max.y - size.y); if (w.min.x < r.min.x) (w.min.x, w.max.x) = (r.min.x, r.min.x + size.x); if (w.min.y < r.min.y) (w.min.y, w.max.y) = (r.min.y, r.min.y + size.y); return w; } lastrect: Rect; # find an suitable area for a window newrect(w, r: Rect): Rect { rl: list of Rect; for(z := wmsrv->top(); z != nil; z = z.znext) for(wl := z.wins; wl != nil; wl = tl wl) rl = (hd wl).r :: rl; lastrect = winplace->place(rl, r, lastrect, w.size()); return lastrect; } bufferproc(in, out: chan of string) { h, t: list of string; dummyout := chan of string; for(;;){ outc := dummyout; s: string; if(h != nil || t != nil){ outc = out; if(h == nil) for(; t != nil; t = tl t) h = hd t :: h; s = hd h; } alt{ x := <-in => t = x :: t; outc <-= s => h = tl h; } } } command(ctxt: ref Draw->Context, args: list of string, sync: chan of string) { if((sh := load Sh Sh->PATH) != nil){ sh->run(ctxt, "{$*&}" :: args); sync <-= nil; return; } fds := list of {0, 1, 2}; sys->pctl(sys->NEWFD, fds); cmd := hd args; file := cmd; if(len file<4 || file[len file-4:]!=".dis") file += ".dis"; c := load Wm file; if(c == nil) { err := sys->sprint("%r"); if(err != "permission denied" && err != "access permission denied" && file[0]!='/' && file[0:2]!="./"){ c = load Wm "/dis/"+file; if(c == nil) err = sys->sprint("%r"); } if(c == nil){ sync <-= sys->sprint("%s: %s\n", cmd, err); exit; } } sync <-= nil; c->init(ctxt, args); }