diff options
Diffstat (limited to 'chain.c')
| -rw-r--r-- | chain.c | 189 |
1 files changed, 189 insertions, 0 deletions
@@ -0,0 +1,189 @@ +#include <u.h> +#include <libc.h> +#include <thread.h> +#include <draw.h> +#include <mouse.h> +#include <keyboard.h> +#include <geometry.h> + +enum { + MAXMAGNITUDE = 40 +}; + +enum { + Cbg, + Cfg, + NCOLOR +}; + +typedef struct Strut Strut; + +struct Strut +{ + Point2 v; + Strut *next; +}; + +RFrame worldrf; +Image *pal[NCOLOR]; +Strut *struts; +Point2 origin; + +Strut* +newstrut(Point2 v) +{ + Strut *s; + + s = malloc(sizeof(Strut)); + if(s == nil) + sysfatal("malloc: %r"); + setmalloctag(s, getcallerpc(&v)); + s->v = v; + s->next = nil; + return s; +} + +Point +toscreen(Point2 p) +{ + p = invrframexform(p, worldrf); + return Pt(p.x,p.y); +} + +Point2 +fromscreen(Point p) +{ + return rframexform(Pt2(p.x,p.y,1), worldrf); +} + +void +initpalette(void) +{ + pal[Cbg] = allocimage(display, Rect(0,0,1,1), screen->chan, 1, 0x000000ff); + pal[Cfg] = allocimage(display, Rect(0,0,1,1), screen->chan, 1, 0xffffffff); +} + +void +redraw(void) +{ + Strut *s; + Point2 p0, p1; + + p0 = origin; + + lockdisplay(display); + draw(screen, screen->r, pal[Cbg], nil, ZP); + for(s = struts; s != nil; s = s->next){ + p1 = addpt2(p0, s->v); + line(screen, toscreen(p0), toscreen(p1), Endsquare, Endarrow, 0, pal[Cfg], ZP); + p0 = p1; + } + flushimage(display, 1); + unlockdisplay(display); +} + +void +resized(void) +{ + lockdisplay(display); + if(getwindow(display, Refnone) < 0) + sysfatal("resize failed"); + unlockdisplay(display); + worldrf.p = Pt2(screen->r.min.x,screen->r.max.y,1); + origin = Pt2(Dx(screen->r)/2,Dy(screen->r)/2,1); + redraw(); +} + +void +mouse(Mouse m) +{ + Strut *s; + Point2 mpos, v; + double θ; + + if((m.buttons & 1) != 0){ + mpos = subpt2(fromscreen(m.xy), origin); + θ = atan2(mpos.y, mpos.x); + //v = Vec2(cos(frand() * 2*PI),sin(frand() * 2*PI)); + v = Vec2(cos(θ),sin(θ)); + v = mulpt2(v, frand()*MAXMAGNITUDE); + s = newstrut(v); + if(struts == nil) + struts = s; + else{ + s->next = struts; + struts = s; + } + } +} + +void +key(Rune r) +{ + switch(r){ + case Kdel: + case 'q': + threadexitsall(nil); + } +} + +void +usage(void) +{ + fprint(2, "usage: %s\n", argv0); + exits("usage"); +} + +void +threadmain(int argc, char *argv[]) +{ + Mousectl *mc; + Keyboardctl *kc; + Rune r; + + ARGBEGIN{ + default: usage(); + }ARGEND; + if(argc > 0) + usage(); + + if(initdraw(nil, nil, nil) < 0) + sysfatal("initdraw: %r"); + if((mc = initmouse(nil, screen)) == nil) + sysfatal("initmouse: %r"); + if((kc = initkeyboard(nil)) == nil) + sysfatal("initkeyboard: %r"); + initpalette(); + worldrf.p = Pt2(screen->r.min.x,screen->r.max.y,1); + worldrf.bx = Vec2(1, 0); + worldrf.by = Vec2(0,-1); + origin = Pt2(Dx(screen->r)/2,Dy(screen->r)/2,1); + + display->locking = 1; + unlockdisplay(display); + redraw(); + + for(;;){ + enum { MOUSE, RESIZE, KEYBOARD }; + Alt a[] = { + {mc->c, &mc->Mouse, CHANRCV}, + {mc->resizec, nil, CHANRCV}, + {kc->c, &r, CHANRCV}, + {nil, nil, CHANEND} + }; + + switch(alt(a)){ + case MOUSE: + mouse(mc->Mouse); + break; + case RESIZE: + resized(); + break; + case KEYBOARD: + key(r); + break; + } + + redraw(); + } +} |