1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249 |
- /*
- * This file is part of the UCB release of Plan 9. It is subject to the license
- * terms in the LICENSE file found in the top-level directory of this
- * distribution and at http://akaros.cs.berkeley.edu/files/Plan9License. No
- * part of the UCB release of Plan 9, including this file, may be copied,
- * modified, propagated, or distributed except according to the terms contained
- * in the LICENSE file.
- */
- #include <u.h>
- #include <libc.h>
- #include <draw.h>
- #include <event.h>
- #include <cursor.h>
- #define initstate muginitstate
- typedef struct State State;
- struct State {
- double black;
- double white;
- double stretch;
- double gamma;
- int depth;
- int gtab[1001];
- Rectangle selr;
- };
- typedef struct Face Face;
- struct Face {
- Rectangle r;
- State state;
- Image *small;
- };
- double GAMMA = 1.0; /* theory tells me this should be 2.2, but 1.0 sure looks better */
- enum {
- Left=0,
- Right,
- Top,
- Bottom,
- RTopLeft=0,
- RTop,
- RTopRight,
- RLeft,
- RMiddle,
- RRight,
- RBotLeft,
- RBot,
- RBotRight,
- };
- void*
- emalloc(uint32_t sz)
- {
- void *v;
- v = malloc(sz);
- if(v == nil)
- sysfatal("malloc %lud fails", sz);
- memset(v, 0, sz);
- return v;
- }
- Face *face[8];
- int nface;
- uint8_t grey2cmap[256];
- Image *bkgd;
- Image *orig;
- Image *ramp, *small, *osmall, *tmp8, *red, *green, *blue;
- State state, ostate;
- uint8_t val2cmap[256];
- uint8_t clamp[3*256];
- Rectangle rbig, rramp, rface[nelem(face)], rsmall;
- double *rdata;
- int sdy, sdx;
- void
- geometry(Rectangle r)
- {
- int i;
- Rectangle fr[9];
- rramp.min = addpt(r.min, Pt(4,4));
- rramp.max = addpt(rramp.min, Pt(256,256));
- rbig.min = Pt(rramp.max.x+6, rramp.min.y);
- rbig.max = addpt(rbig.min, Pt(Dx(orig->r), Dy(orig->r)));
- for(i=0; i<9; i++)
- fr[i] = rectaddpt(Rect(0,0,48,48), Pt(rramp.min.x+48+56*(i%3), rramp.max.y+6+56*(i/3)));
- rsmall = fr[4];
- for(i=0; i<4; i++)
- rface[i] = fr[i];
- for(i=4; i<8; i++)
- rface[i] = fr[i+1];
- }
- double
- y2gamma(int y)
- {
- double g;
- g = (double)y / 128.0;
- return 0.5+g*g; /* gamma from 0.5 to 4.5, with 1.0 near the middle */
- }
- int
- gamma2y(double g)
- {
- g -= 0.5;
- return (int)(128.0*sqrt(g)+0.5);
- }
- void
- drawface(int i)
- {
- if(i==-1){
- border(screen, rsmall, -3, blue, ZP);
- draw(screen, rsmall, small, nil, ZP);
- return;
- }
- border(screen, rface[i], -1, display->black, ZP);
- if(face[i])
- draw(screen, rface[i], face[i]->small, nil, ZP);
- else
- draw(screen, rface[i], display->white, nil, ZP);
- }
- void
- drawrampbar(Image *color, State *s)
- {
- Rectangle liner, r;
- static Rectangle br;
- if(Dx(br))
- draw(screen, br, ramp, nil, subpt(br.min, rramp.min));
- r = rramp;
- r.max.x = r.min.x + (int)(s->white*255.0);
- r.min.x += (int)(s->black*255.0);
- r.min.y += gamma2y(s->gamma);
- r.max.y = r.min.y+1;
- rectclip(&r, rramp);
- draw(screen, r, color, nil, ZP);
- br = r;
- r.min.y -= 2;
- r.max.y += 2;
-
- liner = r;
- r.min.x += Dx(liner)/3;
- r.max.x -= Dx(liner)/3;
- rectclip(&r, rramp);
- draw(screen, r, color, nil, ZP);
- combinerect(&br, r);
- r = liner;
- r.max.x = r.min.x+3;
- rectclip(&r, rramp);
- draw(screen, r, color, nil, ZP);
- combinerect(&br, r);
- r = liner;
- r.min.x = r.max.x-3;
- rectclip(&r, rramp);
- draw(screen, r, color, nil, ZP);
- combinerect(&br, r);
- }
- void
- drawscreen(int clear)
- {
- int i;
- if(clear){
- geometry(screen->r);
- draw(screen, screen->r, bkgd, nil, ZP);
- }
- border(screen, rbig, -1, display->black, ZP);
- draw(screen, rbig, orig, nil, orig->r.min);
- border(screen, rramp, -1, display->black, ZP);
- draw(screen, rramp, ramp, nil, ramp->r.min);
- drawrampbar(red, &state);
- border(screen, rectaddpt(state.selr, subpt(rbig.min, orig->r.min)), -2, red, ZP);
- if(clear){
- drawface(-1);
- for(i=0; i<nelem(face); i++)
- drawface(i);
- }
- }
- void
- moveframe(Rectangle old, Rectangle new)
- {
- border(screen, rectaddpt(old, subpt(rbig.min, orig->r.min)), -2, orig, old.min);
- border(screen, rectaddpt(new, subpt(rbig.min, orig->r.min)), -2, red, ZP);
- }
- /*
- * Initialize gamma ramp; should dither for
- * benefit of non-true-color displays.
- */
- void
- initramp(void)
- {
- int k, x, y;
- uint8_t dat[256*256];
- double g;
- k = 0;
- for(y=0; y<256; y++) {
- g = y2gamma(y);
- for(x=0; x<256; x++)
- dat[k++] = 255.0 * pow(x/255.0, g);
- }
- assert(k == sizeof dat);
- ramp = allocimage(display, Rect(0,0,256,256), GREY8, 0, DNofill);
- if(ramp == nil)
- sysfatal("allocimage: %r");
- if(loadimage(ramp, ramp->r, dat, sizeof dat) != sizeof dat)
- sysfatal("loadimage: %r");
- }
- void
- initclamp(void)
- {
- int i;
- for(i=0; i<256; i++) {
- clamp[i] = 0;
- clamp[256+i] = i;
- clamp[512+i] = 255;
- }
- }
- void
- changestretch(double stretch)
- {
- state.stretch = stretch;
- }
- /*
- * There is greyscale data for the rectangle datar in data;
- * extract square r and write it into the 48x48 pixel image small.
- */
- void
- process(double *data, Rectangle datar, Rectangle r, Image *small)
- {
- double black, center, delta, *k, shrink, sum, *tmp[48], *tt, w, white, x;
- int datadx, dp, dx, dy, error, i, ii, j, jj;
- int ksize, ksizeby2, sdata[48*48], sd, sh, sm, sv, u, uu, uuu, v, vv;
- uint8_t bdata[48*48];
- datadx = Dx(datar);
- dx = Dx(r);
- dy = Dy(r);
- shrink = dx/48.0;
- ksize = 1+2*(int)(shrink/2.0);
- if(ksize <= 2)
- return;
- k = emalloc(ksize*sizeof(k[0]));
- /* center of box */
- for(i=1; i<ksize-1; i++)
- k[i] = 1.0;
- /* edges */
- x = shrink - floor(shrink);
- k[0] = x;
- k[ksize-1] = x;
- sum = 0.0;
- for(i=0; i<ksize; i++)
- sum += k[i];
- for(i=0; i<ksize; i++)
- k[i] /= sum;
- ksizeby2 = ksize/2;
- for(i=0; i<48; i++)
- tmp[i] = emalloc(datadx*sizeof(tmp[i][0]));
- /* squeeze vertically */
- for(i=0; i<48; i++) {
- ii = r.min.y+i*dy/48;
- tt = tmp[i];
- uu = ii - ksizeby2;
- for(j=r.min.x-ksize; j<r.max.x+ksize; j++) {
- if(j<datar.min.x || j>=datar.max.x)
- continue;
- w = 0.0;
- uuu = uu*datadx+j;
- if(uu>=datar.min.y && uu+ksize<datar.max.y)
- for(u=0; u<ksize; u++){
- w += k[u]*data[uuu];
- uuu += datadx;
- }
- else
- for(u=0; u<ksize; u++){
- if(uu+u>=datar.min.y && uu+u<datar.max.y)
- w += k[u]*data[uuu];
- uuu+=datadx;
- }
- tt[j-datar.min.x] = w;
- }
- }
- /* stretch value scale */
- center = (state.black+state.white)/2;
- delta = state.stretch*(state.white-state.black)/2;
- black = center - delta;
- white = center + delta;
- /* squeeze horizontally */
- for(i=0; i<48; i++) {
- tt = tmp[i];
- for(j=0; j<48; j++) {
- jj = r.min.x+j*dx/48;
- w = 0.0;
- for(v=0; v<ksize; v++) {
- vv = jj - ksizeby2 + v;
- if(vv<datar.min.x || vv>=datar.max.x) {
- w += k[v]; /* assume white surround */
- continue;
- }
- w += k[v]*tt[vv-datar.min.x];
- }
- if(w < black || black==white)
- w = 0.0;
- else if(w > white)
- w = 1.0;
- else
- w = (w-black)/(white-black);
- sdata[i*48+j] = state.gtab[(int)(1000.0*w)];
- }
- }
- /* dither to lower depth before copying into GREY8 version */
- if(small->chan != GREY8) {
- u = 0;
- dp = small->depth;
- for(i=0; i<48; i++) {
- sm = 0xFF ^ (0xFF>>dp);
- sh = 0;
- v = 0;
- for(j=0; j<48; j++) {
- ii = 48*i+j;
- sd = clamp[sdata[ii]+256];
- sv = sd&sm;
- v |= sv>>sh;
- sh += dp;
- if(sh == 8) {
- bdata[u++] = v;
- v = 0;
- sh = 0;
- }
- /* propagate error, with decay (sum errors < 1) */
- error = sd - sv;
- if(ii+49 < 48*48) { /* one test is enough, really */
- sdata[ii+1] = sdata[ii+1]+((3*error)>>4);
- sdata[ii+48] = sdata[ii+48]+((3*error)>>4);
- sdata[ii+49] = sdata[ii+49]+((3*error)>>3);
- }
- /* produce correct color map value by copying bits */
- switch(dp){
- case 1:
- sv |= sv>>1;
- case 2:
- sv |= sv>>2;
- case 4:
- sv |= sv>>4;
- }
- sdata[ii] = sv;
- }
- }
- for(i=0; i<nelem(bdata); i++)
- bdata[i] = sdata[i];
- if(loadimage(tmp8, tmp8->r, bdata, sizeof bdata) != sizeof bdata)
- sysfatal("loadimage: %r");
- draw(small, small->r, tmp8, nil, tmp8->r.min);
- } else {
- for(i=0; i<nelem(bdata); i++)
- bdata[i] = sdata[i];
- if(loadimage(small, small->r, bdata, sizeof bdata) != sizeof bdata)
- sysfatal("loadimage: %r");
- }
- free(k);
- for(i=0; i<48; i++)
- free(tmp[i]);
- }
- void
- initval2cmap(void)
- {
- int i;
- for(i=0; i<256; i++)
- val2cmap[i] = rgb2cmap(i, i, i);
- }
- void
- setgtab(State *s)
- {
- int i;
- for(i=0; i<=1000; i++)
- s->gtab[i] = val2cmap[(int)(255.0*pow((i/1000.0), 1.0/s->gamma))];
- }
- int
- section(int x)
- {
- int ib, iw;
- ib = state.black * 255.0;
- iw = state.white * 255.0;
- if(x<ib-5 || iw+5<x)
- return -1;
- iw -= ib;
- x -= ib;
- if(x < iw/3)
- return 0;
- if(x < 2*iw/3)
- return 1;
- return 2;
- }
- Image*
- copyimage(Image *i)
- {
- Image *n;
- if(i == nil)
- return nil;
- n = allocimage(display, i->r, i->chan, 0, DNofill);
- if(n == nil)
- sysfatal("allocimage: %r");
- draw(n, n->r, i, nil, i->r.min);
- return n;
- }
- Image*
- grey8image(Image *i)
- {
- Image *n;
- if(i->chan == GREY8)
- return i;
- n = allocimage(display, i->r, GREY8, 0, DNofill);
- if(n == nil)
- sysfatal("allocimage: %r");
- draw(n, n->r, i, nil, i->r.min);
- freeimage(i);
- return n;
- }
- void
- mark(void)
- {
- if(osmall != small){
- freeimage(osmall);
- osmall = small;
- }
- ostate = state;
- }
- void
- undo(void)
- {
- if(small != osmall){
- freeimage(small);
- small = osmall;
- }
- state = ostate;
- process(rdata, orig->r, state.selr, small);
- drawface(-1);
- drawscreen(0);
- }
- void
- saveface(Face *f, int slot)
- {
- if(slot == -1){
- mark();
- state = f->state;
- small = copyimage(f->small);
- drawface(-1);
- drawscreen(0);
- return;
- }
- if(face[slot]==nil)
- face[slot] = emalloc(sizeof(*face[slot]));
- else{
- freeimage(face[slot]->small);
- face[slot]->small = nil;
- }
- if(f == nil){
- face[slot]->small = copyimage(small);
- face[slot]->state = state;
- }else{
- face[slot]->small = copyimage(f->small);
- face[slot]->state = f->state;
- }
- drawface(slot);
- }
- int
- writeface(char *outfile, Image *image)
- {
- int i, fd, rv, y;
- uint8_t data[48*48/2];
- if(outfile == nil)
- fd = 1;
- else{
- if((fd = create(outfile, OWRITE, 0666)) < 0)
- return -1;
- }
- switch(image->chan) {
- default:
- rv = -1;
- break;
- case GREY1:
- if(unloadimage(image, image->r, data, 48*48/8) != 48*48/8)
- sysfatal("unloadimage: %r");
- for(y=0; y<48; y++) {
- for(i=0; i<3; i++)
- fprint(fd, "0x%.2x%.2x,", data[y*6+i*2+0], data[y*6+i*2+1]);
- fprint(fd, "\n");
- }
- rv = 0;
- break;
-
- case GREY2:
- if(unloadimage(image, image->r, data, 48*48/4) != 48*48/4)
- sysfatal("unloadimage: %r");
- for(y=0; y<48; y++) {
- for(i=0; i<3; i++)
- fprint(fd, "0x%.2x%.2x,%.2x%.2x,",
- data[y*12+i*4+0], data[y*12+i*4+1],
- data[y*12+i*4+2], data[y*12+i*4+3]);
- fprint(fd, "\n");
- }
- rv = 0;
- break;
- case GREY4:
- case GREY8:
- rv = writeimage(fd, image, 0); /* dolock? */
- break;
- }
- if(outfile)
- close(fd);
- return rv;
- }
- void
- room(Rectangle out, Rectangle in, int *a)
- {
- a[Left] = out.min.x - in.min.x;
- a[Right] = out.max.x - in.max.x;
- a[Top] = out.min.y - in.min.y;
- a[Bottom] = out.max.y - in.max.y;
- }
- int
- min(int a, int b)
- {
- if(a < b)
- return a;
- return b;
- }
- int
- max(int a, int b)
- {
- if(a > b)
- return a;
- return b;
- }
- int
- move(Rectangle r, Rectangle picr, Point d, int k, Rectangle *rp)
- {
- int a[4], i;
- Rectangle oldr;
- static int toggle;
- oldr = r;
- room(picr, r, a);
- switch(k){
- case RTopLeft:
- i = (d.x+d.y)/2;
- if(i>=Dx(r) || i>=Dy(r))
- break;
- i = max(i, a[Left]);
- i = max(i, a[Top]);
- r.min.x += i;
- r.min.y += i;
- break;
- case RTop:
- i = d.y;
- if(i < 0){
- /*
- * should really check i/2, but this is safe and feedback
- * makes the control feel right
- */
- i = -min(-i, a[Right]);
- i = max(i, a[Left]);
- }
- i = max(i, a[Top]);
- if(i >= Dy(r))
- break;
- r.min.y += i;
- /* divide the half bit equally */
- toggle = 1-toggle;
- if(toggle){
- r.min.x += i/2;
- r.max.x = r.min.x+Dy(r);
- }else{
- r.max.x -= i/2;
- r.min.x = r.max.x-Dy(r);
- }
- break;
- case RTopRight:
- i = (-d.x+d.y)/2;
- if(i>=Dx(r) || i>=Dy(r))
- break;
- i = -min(-i, a[Right]);
- i = max(i, a[Top]);
- r.max.x -= i;
- r.min.y += i;
- break;
- case RLeft:
- i = d.x;
- if(i < 0){
- i = -min(-i, a[Bottom]);
- i = max(i, a[Top]);
- }
- i = max(i, a[Left]);
- if(i >= Dx(r))
- break;
- r.min.x += i;
- /* divide the half bit equally */
- toggle = 1-toggle;
- if(toggle){
- r.min.y += i/2;
- r.max.y = r.min.y+Dx(r);
- }else{
- r.max.y -= i/2;
- r.min.y = r.max.y-Dx(r);
- }
- break;
- case RMiddle:
- if(d.x >= 0)
- d.x = min(d.x, a[Right]);
- else
- d.x = max(d.x, a[Left]);
- if(d.y >= 0)
- d.y = min(d.y, a[Bottom]);
- else
- d.y = max(d.y, a[Top]);
- r = rectaddpt(r, d);
- break;
- case RRight:
- i = d.x;
- if(i > 0){
- i = min(i, a[Bottom]);
- i = -max(-i, a[Top]);
- }
- i = min(i, a[Right]);
- if(-i >= Dx(r))
- break;
- r.max.x += i;
- /* divide the half bit equally */
- toggle = 1-toggle;
- if(toggle){
- r.min.y -= i/2;
- r.max.y = r.min.y+Dx(r);
- }else{
- r.max.y += i/2;
- r.min.y = r.max.y-Dx(r);
- }
- break;
- case RBotLeft:
- i = (d.x+-d.y)/2;
- if(i>=Dx(r) || i>=Dy(r))
- break;
- i = max(i, a[Left]);
- i = -min(-i, a[Bottom]);
- r.min.x += i;
- r.max.y -= i;
- break;
- case RBot:
- i = d.y;
- if(i > 0){
- i = min(i, a[Right]);
- i = -max(-i, a[Left]);
- }
- i = min(i, a[Bottom]);
- if(i >= Dy(r))
- break;
- r.max.y += i;
- /* divide the half bit equally */
- toggle = 1-toggle;
- if(toggle){
- r.min.x -= i/2;
- r.max.x = r.min.x+Dy(r);
- }else{
- r.max.x += i/2;
- r.min.x = r.max.x-Dy(r);
- }
- break;
- case RBotRight:
- i = (-d.x+-d.y)/2;
- if(i>=Dx(r) || i>=Dy(r))
- break;
- i = -min(-i, a[Right]);
- i = -min(-i, a[Bottom]);
- r.max.x -= i;
- r.max.y -= i;
- break;
- }
- if(Dx(r)<3 || Dy(r)<3){
- *rp = oldr;
- return 0;
- }
- *rp = r;
- return !eqrect(r, oldr);
- }
- void
- rlist(Rectangle r, Rectangle *ra)
- {
- Rectangle tr;
- tr = r;
- tr.max.y = r.min.y+Dy(r)/4;
- ra[0] = tr;
- ra[0].max.x = tr.min.x+Dx(tr)/4;
- ra[1] = tr;
- ra[1].min.x = ra[0].max.x;
- ra[1].max.x = tr.max.x-Dx(tr)/4;
- ra[2] = tr;
- ra[2].min.x = ra[1].max.x;
- tr.min.y = tr.max.y;
- tr.max.y = r.max.y-Dy(r)/4;
- ra[3] = tr;
- ra[3].max.x = tr.min.x+Dx(tr)/4;
- ra[4] = tr;
- ra[4].min.x = ra[3].max.x;
- ra[4].max.x = tr.max.x-Dx(tr)/4;
- ra[5] = tr;
- ra[5].min.x = ra[4].max.x;
- tr.min.y = tr.max.y;
- tr.max.y = r.max.y;
- ra[6] = tr;
- ra[6].max.x = tr.min.x+Dx(tr)/4;
- ra[7] = tr;
- ra[7].min.x = ra[6].max.x;
- ra[7].max.x = tr.max.x-Dx(tr)/4;
- ra[8] = tr;
- ra[8].min.x = ra[7].max.x;
- }
- int
- abs(int a)
- {
- if(a < 0)
- return -a;
- return a;
- }
- void
- usage(void)
- {
- fprint(2, "usage: mug [file.bit]\n");
- exits("usage");
- }
- void
- eresized(int new)
- {
- if(new && getwindow(display, Refmesg) < 0)
- fprint(2,"can't reattach to window");
- drawscreen(1);
- }
- /*
- interface notes
- cursor changes while in rbig to indicate region.
- only button 1 works for resizing region
- only button 1 works for moving thingy in ramp
- button-3 menu: Reset, Depth, Undo, Save, Write
- */
- Cursor tl = {
- {-4, -4},
- {0xfe, 0x00, 0x82, 0x00, 0x8c, 0x00, 0x87, 0xff,
- 0xa0, 0x01, 0xb0, 0x01, 0xd0, 0x01, 0x11, 0xff,
- 0x11, 0x00, 0x11, 0x00, 0x11, 0x00, 0x11, 0x00,
- 0x11, 0x00, 0x11, 0x00, 0x11, 0x00, 0x1f, 0x00, },
- {0x00, 0x00, 0x7c, 0x00, 0x70, 0x00, 0x78, 0x00,
- 0x5f, 0xfe, 0x4f, 0xfe, 0x0f, 0xfe, 0x0e, 0x00,
- 0x0e, 0x00, 0x0e, 0x00, 0x0e, 0x00, 0x0e, 0x00,
- 0x0e, 0x00, 0x0e, 0x00, 0x0e, 0x00, 0x00, 0x00, }
- };
- Cursor t = {
- {-7, -8},
- {0x00, 0x00, 0x00, 0x00, 0x03, 0x80, 0x06, 0xc0,
- 0x1c, 0x70, 0x10, 0x10, 0x0c, 0x60, 0xfc, 0x7f,
- 0x80, 0x01, 0x80, 0x01, 0x80, 0x01, 0xff, 0xff,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, },
- {0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00,
- 0x03, 0x80, 0x0f, 0xe0, 0x03, 0x80, 0x03, 0x80,
- 0x7f, 0xfe, 0x7f, 0xfe, 0x7f, 0xfe, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, }
- };
- Cursor tr = {
- {-11, -4},
- {0x00, 0x7f, 0x00, 0x41, 0x00, 0x31, 0xff, 0xe1,
- 0x80, 0x05, 0x80, 0x0d, 0x80, 0x0b, 0xff, 0x88,
- 0x00, 0x88, 0x0, 0x88, 0x00, 0x88, 0x00, 0x88,
- 0x00, 0x88, 0x00, 0x88, 0x00, 0x88, 0x00, 0xf8, },
- {0x00, 0x00, 0x00, 0x3e, 0x00, 0x0e, 0x00, 0x1e,
- 0x7f, 0xfa, 0x7f, 0xf2, 0x7f, 0xf0, 0x00, 0x70,
- 0x00, 0x70, 0x00, 0x70, 0x00, 0x70, 0x00, 0x70,
- 0x00, 0x70, 0x00, 0x70, 0x00, 0x70, 0x00, 0x00, }
- };
- Cursor r = {
- {-8, -7},
- {0x07, 0xc0, 0x04, 0x40, 0x04, 0x40, 0x04, 0x58,
- 0x04, 0x68, 0x04, 0x6c, 0x04, 0x06, 0x04, 0x02,
- 0x04, 0x06, 0x04, 0x6c, 0x04, 0x68, 0x04, 0x58,
- 0x04, 0x40, 0x04, 0x40, 0x04, 0x40, 0x07, 0xc0, },
- {0x00, 0x00, 0x03, 0x80, 0x03, 0x80, 0x03, 0x80,
- 0x03, 0x90, 0x03, 0x90, 0x03, 0xf8, 0x03, 0xfc,
- 0x03, 0xf8, 0x03, 0x90, 0x03, 0x90, 0x03, 0x80,
- 0x03, 0x80, 0x03, 0x80, 0x03, 0x80, 0x00, 0x00, }
- };
- Cursor br = {
- {-11, -11},
- {0x00, 0xf8, 0x00, 0x88, 0x00, 0x88, 0x00, 0x88,
- 0x00, 0x88, 0x00, 0x88, 0x00, 0x88, 0x00, 0x88,
- 0xff, 0x88, 0x80, 0x0b, 0x80, 0x0d, 0x80, 0x05,
- 0xff, 0xe1, 0x00, 0x31, 0x00, 0x41, 0x00, 0x7f, },
- {0x00, 0x00, 0x00, 0x70, 0x00, 0x70, 0x00, 0x70,
- 0x0, 0x70, 0x00, 0x70, 0x00, 0x70, 0x00, 0x70,
- 0x00, 0x70, 0x7f, 0xf0, 0x7f, 0xf2, 0x7f, 0xfa,
- 0x00, 0x1e, 0x00, 0x0e, 0x00, 0x3e, 0x00, 0x00, }
- };
- Cursor b = {
- {-7, -7},
- {0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0xff, 0xff, 0x80, 0x01, 0x80, 0x01, 0x80, 0x01,
- 0xfc, 0x7f, 0x0c, 0x60, 0x10, 0x10, 0x1c, 0x70,
- 0x06, 0xc0, 0x03, 0x80, 0x00, 0x00, 0x00, 0x00, },
- {0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x7f, 0xfe, 0x7f, 0xfe, 0x7f, 0xfe,
- 0x03, 0x80, 0x03, 0x80, 0x0f, 0xe0, 0x03, 0x80,
- 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, }
- };
- Cursor bl = {
- {-4, -11},
- {0x1f, 0x00, 0x11, 0x00, 0x11, 0x00, 0x11, 0x00,
- 0x11, 0x00, 0x11, 0x00, 0x11, 0x00, 0x11, 0x00,
- 0x11, 0xff, 0xd0, 0x01, 0xb0, 0x01, 0xa0, 0x01,
- 0x87, 0xff, 0x8c, 0x00, 0x82, 0x00, 0xfe, 0x00, },
- {0x00, 0x00, 0x0e, 0x00, 0x0e, 0x00, 0x0e, 0x00,
- 0x0e, 0x00, 0x0e, 0x00, 0x0e, 0x00, 0x0e, 0x00,
- 0x0e, 0x00, 0x0f, 0xfe, 0x4f, 0xfe, 0x5f, 0xfe,
- 0x78, 0x00, 0x70, 0x00, 0x7c, 0x00, 0x00, 0x0, }
- };
- Cursor l = {
- {-7, -7},
- {0x03, 0xe0, 0x02, 0x20, 0x02, 0x20, 0x1a, 0x20,
- 0x16, 0x20, 0x36, 0x20, 0x60, 0x20, 0x40, 0x20,
- 0x60, 0x20, 0x36, 0x20, 0x16, 0x20, 0x1a, 0x20,
- 0x02, 0x20, 0x02, 0x20, 0x02, 0x20, 0x03, 0xe0, },
- {0x00, 0x00, 0x01, 0xc0, 0x01, 0xc0, 0x01, 0xc0,
- 0x09, 0xc0, 0x09, 0xc0, 0x1f, 0xc0, 0x3f, 0xc0,
- 0x1f, 0xc0, 0x09, 0xc0, 0x09, 0xc0, 0x01, 0xc0,
- 0x01, 0xc0, 0x01, 0xc0, 0x01, 0xc0, 0x00, 0x00, }
- };
- Cursor boxcursor = {
- {-7, -7},
- {0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
- 0xFF, 0xFF, 0xF8, 0x1F, 0xF8, 0x1F, 0xF8, 0x1F,
- 0xF8, 0x1F, 0xF8, 0x1F, 0xF8, 0x1F, 0xFF, 0xFF,
- 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, },
- {0x00, 0x00, 0x7F, 0xFE, 0x7F, 0xFE, 0x7F, 0xFE,
- 0x70, 0x0E, 0x70, 0x0E, 0x70, 0x0E, 0x70, 0x0E,
- 0x70, 0x0E, 0x70, 0x0E, 0x70, 0x0E, 0x70, 0x0E,
- 0x7F, 0xFE, 0x7F, 0xFE, 0x7F, 0xFE, 0x00, 0x00, }
- };
- Cursor clearcursor;
- Cursor *corners[10] = {
- &tl, &t, &tr,
- &l, &boxcursor, &r,
- &bl, &b, &br,
- nil, /* default arrow */
- };
- char *item[] = {
- "Reset",
- "Depth",
- "Undo",
- "Write",
- "Exit",
- nil
- };
- Menu menu = {
- item,
- nil,
- 2
- };
- /*BUG make less flashy */
- void
- moveface(Image *back, Point lastp, Image *face, Point p, Point d)
- {
- draw(screen, rectaddpt(back->r, subpt(lastp, d)), back, nil, back->r.min);
- draw(back, back->r, screen, nil, addpt(back->r.min, subpt(p, d)));
- border(screen, rectaddpt(face->r, subpt(p, d)),
- -1, display->black, ZP);
- draw(screen, rectaddpt(face->r, subpt(p, d)),
- face, nil, face->r.min);
- }
- int
- dragface(Mouse *m, Image *im, Point d, int x)
- {
- int i;
- Point lastp;
- static Image *back;
- if(back == nil){
- back = allocimage(display, Rect(-1,-1,49,49), display->image->chan, 0, DNofill);
- if(back == nil)
- sysfatal("dragface backing store: %r");
- }
- lastp = m->xy;
- draw(back, back->r, screen, nil, addpt(back->r.min, subpt(lastp, d)));
- esetcursor(&clearcursor);
- do{
- moveface(back, lastp, im, m->xy, d);
- lastp = m->xy;
- }while(*m=emouse(), m->buttons==1);
- draw(screen, rectaddpt(back->r, subpt(lastp, d)), back, nil, back->r.min);
- esetcursor(nil);
- if(m->buttons==0){
- for(i=0; i<nelem(face); i++)
- if(ptinrect(m->xy, rface[i]))
- return i;
- if(ptinrect(m->xy, rsmall))
- return -1;
- return x;
- }
- while(*m=emouse(), m->buttons)
- ;
- return x;
- }
- void
- initstate(void)
- {
- state.black = 0.0;
- state.white = 1.0;
- state.stretch = 1.0;
- state.depth = 4;
- state.gamma = 1.0;
- setgtab(&state);
- state.selr = insetrect(orig->r, 5);
- sdx = Dx(state.selr);
- sdy = Dy(state.selr);
- if(sdx > sdy)
- state.selr.max.x = state.selr.min.x+sdy;
- else
- state.selr.max.y = state.selr.min.y+sdx;
- }
- void
- main(int argc, char **argv)
- {
- int ccursor, i, fd, k, n, y;
- unsigned char *data;
- double gammatab[256];
- Event e;
- Mouse m;
- Point lastp, p;
- Rectangle nselr, rbig9[9];
- ARGBEGIN{
- default:
- usage();
- }ARGEND
- if(argc > 1)
- usage();
- if(argc == 1){
- if((fd = open(argv[0], OREAD)) < 0)
- sysfatal("open %s: %r", argv[0]);
- }else
- fd = 0;
- if (initdraw(0, 0, "mug") < 0)
- sysfatal("initdraw failed");
- if((orig = readimage(display, fd, 0)) == nil)
- sysfatal("readimage: %r");
- orig = grey8image(orig);
- initramp();
- initclamp();
- initval2cmap();
- bkgd = allocimagemix(display, DPaleyellow, DWhite);
- small = allocimage(display, Rect(0,0,48,48), GREY4, 0, DWhite);
- tmp8 = allocimage(display, Rect(0,0,48,48), GREY8, 0, DWhite);
- red = allocimage(display, Rect(0,0,1,1), display->image->chan, 1, DRed);
- green = allocimage(display, Rect(0,0,1,1), display->image->chan, 1, DGreen);
- blue = allocimage(display, Rect(0,0,1,1), display->image->chan, 1, DBlue);
- if(bkgd==nil || small==nil || tmp8==nil || red==nil || green==nil || blue==nil)
- sysfatal("allocimage: %r");
- n = Dx(orig->r)*Dy(orig->r);
- data = emalloc(n*sizeof data[0]);
- rdata = emalloc(n*sizeof rdata[0]);
- if(unloadimage(orig, orig->r, data, n) != n)
- sysfatal("unloadimage: %r");
-
- for(i=0; i<256; i++)
- gammatab[i] = pow((255-i)/(double)255.0, GAMMA);
- for(i=0; i<n; i++)
- rdata[i] = gammatab[255-data[i]];
- initstate();
- process(rdata, orig->r, state.selr, small);
- drawscreen(1);
- flushimage(display, 1);
- einit(Emouse|Ekeyboard);
- ccursor = 9;
- for(;;){
- if((n=eread(Emouse|Ekeyboard, &e))==Ekeyboard)
- continue;
- if(n != Emouse)
- break;
- m = e.mouse;
- if(m.buttons&4){
- ccursor = 9;
- esetcursor(corners[ccursor]);
- switch(emenuhit(3, &m, &menu)){
- case -1:
- continue;
- case 0: /* Reset */
- mark();
- initstate();
- small = allocimage(display, Rect(0,0,48,48), CHAN1(CGrey, state.depth), 0, DWhite);
- if(small == nil)
- sysfatal("allocimage: %r");
- process(rdata, orig->r, state.selr, small);
- drawface(-1);
- drawscreen(0);
- break;
- case 1: /* Depth */
- mark();
- /* osmall = small, so no freeimage */
- state.depth /= 2;
- if(state.depth == 0)
- state.depth = 8;
- small = allocimage(display, Rect(0,0,48,48), CHAN1(CGrey, state.depth), 0, DWhite);
- if(small == nil)
- sysfatal("allocimage: %r");
- process(rdata, orig->r, state.selr, small);
- drawface(-1);
- break;
- case 2: /* Undo */
- undo();
- break;
- case 3: /* Write */
- writeface(nil, small);
- break;
- case 4: /* Exit */
- exits(nil);
- break;
- }
- }
-
- if(ptinrect(m.xy, rbig)){
- rlist(rectaddpt(state.selr, subpt(rbig.min, orig->r.min)), rbig9);
- for(i=0; i<9; i++)
- if(ptinrect(m.xy, rbig9[i]))
- break;
- if(i != ccursor){
- ccursor = i;
- esetcursor(corners[ccursor]);
- }
- if(i==9)
- continue;
- if(m.buttons & 1){
- mark();
- lastp = m.xy;
- while(m=emouse(), m.buttons&1){
- if(move(state.selr, orig->r, subpt(m.xy, lastp), i, &nselr)){
- moveframe(state.selr, nselr);
- state.selr = nselr;
- lastp = m.xy;
- process(rdata, orig->r, state.selr, small);
- drawface(-1);
- }
- }
- }
- continue;
- }
- if(ccursor != 9){ /* default cursor */
- ccursor = 9;
- esetcursor(corners[ccursor]);
- }
- if(ptinrect(m.xy, rramp)){
- if(m.buttons != 1)
- continue;
- mark();
- y = gamma2y(state.gamma);
- if(abs(y-(m.xy.y-rramp.min.y)) > 5)
- continue;
- k = section(m.xy.x-rramp.min.x);
- drawrampbar(green, &state);
- lastp = m.xy;
- while(m=emouse(), m.buttons&1){
- if(!ptinrect(m.xy, rramp))
- continue;
- switch(k){
- case -1:
- continue;
- case 0:
- if((m.xy.x-rramp.min.x)/255.0 < state.white){
- state.black = (m.xy.x-rramp.min.x)/255.0;
- break;
- }
- continue;
- case 1:
- state.gamma = y2gamma(m.xy.y-rramp.min.y);
- setgtab(&state);
- break;
- case 2:
- if((m.xy.x-rramp.min.x)/255.0 > state.black){
- state.white = (m.xy.x-rramp.min.x)/255.0;
- break;
- }
- continue;
- case 10:
- state.black += (m.xy.x-lastp.x)/255.0;
- state.white += (m.xy.x-lastp.x)/255.0;
- state.gamma = y2gamma(p.y);
- break;
- }
- process(rdata, orig->r, state.selr, small);
- drawface(-1);
- drawrampbar(green, &state);
- }
- if(m.buttons == 0){
- process(rdata, orig->r, state.selr, small);
- drawface(-1);
- drawrampbar(red, &state);
- }else
- undo();
- continue;
- }
-
- if(ptinrect(m.xy, rsmall)){
- if(m.buttons != 1)
- continue;
- n=dragface(&m, small, subpt(m.xy, rsmall.min), -1);
- if(n == -1)
- continue;
- saveface(nil, n);
- }
-
- for(i=0; i<nelem(face); i++)
- if(ptinrect(m.xy, rface[i]))
- break;
- if(i<nelem(face) && face[i] != nil){
- if(m.buttons != 1)
- continue;
- n=dragface(&m, face[i]->small, subpt(m.xy, rface[i].min), i);
- if(n == i)
- continue;
- saveface(face[i], n);
- continue;
- }
- do
- m = emouse();
- while(m.buttons==1);
- }
- exits(nil);
- }
|