--- ray/src/gen/mkillum2.c 1991/08/22 12:12:23 1.11 +++ ray/src/gen/mkillum2.c 2004/09/17 21:43:50 2.16 @@ -1,41 +1,58 @@ -/* Copyright (c) 1991 Regents of the University of California */ - #ifndef lint -static char SCCSid[] = "$SunId$ LBL"; +static const char RCSid[] = "$Id: mkillum2.c,v 2.16 2004/09/17 21:43:50 greg Exp $"; #endif - /* * Routines to do the actual calculation for mkillum */ -#include "mkillum.h" +#include +#include "mkillum.h" #include "face.h" - #include "cone.h" - #include "random.h" +#include "selcall.h" -o_default(ob, il, rt, nm) /* default illum action */ -OBJREC *ob; -struct illum_args *il; -struct rtproc *rt; -char *nm; +static void mkaxes(FVECT u, FVECT v, FVECT n); +static void rounddir(FVECT dv, double alt, double azi); +static void flatdir(FVECT dv, double alt, double azi); + + +static void +rayclean( /* finish all pending rays */ + struct rtproc *rt0 +) { + rayflush(rt0, 1); + while (raywait(rt0) != NULL) + ; +} + + +int /* XXX type conflict with otypes.h */ +o_default( /* default illum action */ + OBJREC *ob, + struct illum_args *il, + struct rtproc *rt0, + char *nm +) +{ sprintf(errmsg, "(%s): cannot make illum for %s \"%s\"", nm, ofun[ob->otype].funame, ob->oname); error(WARNING, errmsg); - if (!(il->flags & IL_LIGHT)) - printobj(il->altmat, ob); + printobj(il->altmat, ob); + return(1); } -o_face(ob, il, rt, nm) /* make an illum face */ -OBJREC *ob; -struct illum_args *il; -struct rtproc *rt; -char *nm; +int +o_face( /* make an illum face */ + OBJREC *ob, + struct illum_args *il, + struct rtproc *rt0, + char *nm +) { #define MAXMISS (5*n*il->nsamps) int dim[3]; @@ -52,8 +69,7 @@ char *nm; fa = getface(ob); if (fa->area == 0.0) { freeface(ob); - o_default(ob, il, rt, nm); - return; + return(o_default(ob, il, rt0, nm)); } /* set up sampling */ if (il->sampdens <= 0) @@ -67,7 +83,21 @@ char *nm; distarr = (float *)calloc(n, 3*sizeof(float)); if (distarr == NULL) error(SYSTEM, "out of memory in o_face"); - mkaxes(u, v, fa->norm); + /* take first edge longer than sqrt(area) */ + for (j = fa->nv-1, i = 0; i < fa->nv; j = i++) { + u[0] = VERTEX(fa,i)[0] - VERTEX(fa,j)[0]; + u[1] = VERTEX(fa,i)[1] - VERTEX(fa,j)[1]; + u[2] = VERTEX(fa,i)[2] - VERTEX(fa,j)[2]; + if ((r1 = DOT(u,u)) >= fa->area-FTINY) + break; + } + if (i < fa->nv) { /* got one! -- let's align our axes */ + r2 = 1.0/sqrt(r1); + u[0] *= r2; u[1] *= r2; u[2] *= r2; + fcross(v, fa->norm, u); + } else /* oh well, we'll just have to wing it */ + mkaxes(u, v, fa->norm); + /* now, find limits in (u,v) coordinates */ ur[0] = vr[0] = FHUGE; ur[1] = vr[1] = -FHUGE; for (i = 0; i < fa->nv; i++) { @@ -88,7 +118,7 @@ char *nm; h = ilhash(dim, 3) + i; multisamp(sp, 2, urand(h)); r1 = (dim[1] + sp[0])/nalt; - r2 = (dim[2] + sp[1])/nazi; + r2 = (dim[2] + sp[1] - .5)/nazi; flatdir(dn, r1, r2); for (j = 0; j < 3; j++) dir[j] = -dn[0]*u[j] - dn[1]*v[j] - dn[2]*fa->norm[j]; @@ -103,35 +133,39 @@ char *nm; } while (!inface(org, fa) && nmisses++ < MAXMISS); if (nmisses > MAXMISS) { objerror(ob, WARNING, "bad aspect"); - rt->nrays = 0; + rayclean(rt0); freeface(ob); - free((char *)distarr); - o_default(ob, il, rt, nm); - return; + free((void *)distarr); + return(o_default(ob, il, rt0, nm)); } for (j = 0; j < 3; j++) org[j] += .001*fa->norm[j]; /* send sample */ - raysamp(distarr+3*(dim[1]*nazi+dim[2]), org, dir, rt); + raysamp(distarr+3*(dim[1]*nazi+dim[2]), org, dir, rt0); } - rayflush(rt); + rayclean(rt0); /* write out the face and its distribution */ - average(il, distarr, nalt*nazi); - if (il->sampdens > 0) - flatout(il, distarr, nalt, nazi, u, v, fa->norm); - illumout(il, ob); + if (average(il, distarr, nalt*nazi)) { + if (il->sampdens > 0) + flatout(il, distarr, nalt, nazi, u, v, fa->norm); + illumout(il, ob); + } else + printobj(il->altmat, ob); /* clean up */ freeface(ob); - free((char *)distarr); + free((void *)distarr); + return(0); #undef MAXMISS } -o_sphere(ob, il, rt, nm) /* make an illum sphere */ -register OBJREC *ob; -struct illum_args *il; -struct rtproc *rt; -char *nm; +int +o_sphere( /* make an illum sphere */ + register OBJREC *ob, + struct illum_args *il, + struct rtproc *rt0, + char *nm +) { int dim[3]; int n, nalt, nazi; @@ -148,8 +182,8 @@ char *nm; nalt = nazi = 1; else { n = 4.*PI * il->sampdens; - nalt = sqrt(n/PI) + .5; - nazi = PI*nalt + .5; + nalt = sqrt(2./PI*n) + .5; + nazi = PI/2.*nalt + .5; } n = nalt*nazi; distarr = (float *)calloc(n, 3*sizeof(float)); @@ -164,7 +198,7 @@ char *nm; multisamp(sp, 4, urand(ilhash(dim,3)+i)); /* random direction */ r1 = (dim[1] + sp[0])/nalt; - r2 = (dim[2] + sp[1])/nazi; + r2 = (dim[2] + sp[1] - .5)/nazi; rounddir(dir, r1, r2); /* random location */ mkaxes(u, v, dir); /* yuck! */ @@ -179,26 +213,31 @@ char *nm; dir[j] = -dir[j]; } /* send sample */ - raysamp(distarr+3*(dim[1]*nazi+dim[2]), org, dir, rt); + raysamp(distarr+3*(dim[1]*nazi+dim[2]), org, dir, rt0); } - rayflush(rt); + rayclean(rt0); /* write out the sphere and its distribution */ - average(il, distarr, nalt*nazi); - if (il->sampdens > 0) - roundout(il, distarr, nalt, nazi); - else - objerror(ob, WARNING, "diffuse distribution"); - illumout(il, ob); + if (average(il, distarr, nalt*nazi)) { + if (il->sampdens > 0) + roundout(il, distarr, nalt, nazi); + else + objerror(ob, WARNING, "diffuse distribution"); + illumout(il, ob); + } else + printobj(il->altmat, ob); /* clean up */ - free((char *)distarr); + free((void *)distarr); + return(1); } -o_ring(ob, il, rt, nm) /* make an illum ring */ -OBJREC *ob; -struct illum_args *il; -struct rtproc *rt; -char *nm; +int +o_ring( /* make an illum ring */ + OBJREC *ob, + struct illum_args *il, + struct rtproc *rt0, + char *nm +) { int dim[3]; int n, nalt, nazi; @@ -232,7 +271,7 @@ char *nm; multisamp(sp, 4, urand(ilhash(dim,3)+i)); /* random direction */ r1 = (dim[1] + sp[0])/nalt; - r2 = (dim[2] + sp[1])/nalt; + r2 = (dim[2] + sp[1] - .5)/nazi; flatdir(dn, r1, r2); for (j = 0; j < 3; j++) dir[j] = -dn[0]*u[j] - dn[1]*v[j] - dn[2]*co->ad[j]; @@ -243,65 +282,142 @@ char *nm; r1 = r3*cos(r2); r2 = r3*sin(r2); for (j = 0; j < 3; j++) - org[j] = CO_P0(co)[j] + r1*u[j] + r1*v[j] + + org[j] = CO_P0(co)[j] + r1*u[j] + r2*v[j] + .001*co->ad[j]; /* send sample */ - raysamp(distarr+3*(dim[1]*nazi+dim[2]), org, dir, rt); + raysamp(distarr+3*(dim[1]*nazi+dim[2]), org, dir, rt0); } - rayflush(rt); + rayclean(rt0); /* write out the ring and its distribution */ - average(il, distarr, nalt*nazi); - if (il->sampdens > 0) - flatout(il, distarr, nalt, nazi, u, v, co->ad); - illumout(il, ob); + if (average(il, distarr, nalt*nazi)) { + if (il->sampdens > 0) + flatout(il, distarr, nalt, nazi, u, v, co->ad); + illumout(il, ob); + } else + printobj(il->altmat, ob); /* clean up */ freecone(ob); - free((char *)distarr); + free((void *)distarr); + return(1); } -raysamp(res, org, dir, rt) /* compute a ray sample */ -float res[3]; -FVECT org, dir; -register struct rtproc *rt; +void +raysamp( /* queue a ray sample */ + float res[3], + FVECT org, + FVECT dir, + struct rtproc *rt0 +) { + register struct rtproc *rt; register float *fp; - if (rt->nrays == rt->bsiz) - rayflush(rt); - rt->dest[rt->nrays] = res; - fp = rt->buf + 6*rt->nrays++; + for (rt = rt0; rt != NULL; rt = rt->next) + if (rt->nrays < rt->bsiz && rt->dest[rt->nrays] == NULL) + break; + if (rt == NULL) /* need to free up buffer? */ + rt = raywait(rt0); + if (rt == NULL) + error(SYSTEM, "raywait() returned NULL"); + fp = rt->buf + 6*rt->nrays; *fp++ = org[0]; *fp++ = org[1]; *fp++ = org[2]; *fp++ = dir[0]; *fp++ = dir[1]; *fp = dir[2]; + rt->dest[rt->nrays++] = res; + if (rt->nrays == rt->bsiz) + rayflush(rt, 0); } -rayflush(rt) /* flush buffered rays */ -register struct rtproc *rt; +void +rayflush( /* flush queued rays to rtrace */ + register struct rtproc *rt, + int doall +) { - register int i; + int nw; - if (rt->nrays <= 0) - return; - bzero(rt->buf+6*rt->nrays, 6*sizeof(float)); - if ( process(rt->pd, (char *)rt->buf, (char *)rt->buf, - 3*sizeof(float)*rt->nrays, - 6*sizeof(float)*(rt->nrays+1)) < - 3*sizeof(float)*rt->nrays ) - error(SYSTEM, "error reading from rtrace process"); - i = rt->nrays; - while (i--) { - rt->dest[i][0] += rt->buf[3*i]; - rt->dest[i][1] += rt->buf[3*i+1]; - rt->dest[i][2] += rt->buf[3*i+2]; + do { + if (rt->nrays <= 0) + continue; + memset(rt->buf+6*rt->nrays, 0, 6*sizeof(float)); + nw = 6*sizeof(float)*(rt->nrays+1); + errno = 0; + if (writebuf(rt->pd.w, (char *)rt->buf, nw) < nw) + error(SYSTEM, "error writing to rtrace process"); + rt->nrays = 0; /* flag buffer as flushed */ + } while (doall && (rt = rt->next) != NULL); +} + + +struct rtproc * +raywait( /* retrieve rtrace results */ + struct rtproc *rt0 +) +{ + fd_set readset, errset; + int nr; + struct rtproc *rt, *rtfree; + register int n; + /* prepare select call */ + FD_ZERO(&readset); FD_ZERO(&errset); n = 0; + nr = 0; + for (rt = rt0; rt != NULL; rt = rt->next) { + if (rt->nrays == 0 && rt->dest[0] != NULL) { + FD_SET(rt->pd.r, &readset); + ++nr; + } + FD_SET(rt->pd.r, &errset); + if (rt->pd.r >= n) + n = rt->pd.r + 1; } - rt->nrays = 0; + if (!nr) /* no rays pending */ + return(NULL); + if (nr > 1) /* call select for multiple processes */ + n = select(n, &readset, (fd_set *)NULL, &errset, + (struct timeval *)NULL); + else + FD_ZERO(&errset); + if (n <= 0) + return(NULL); + rtfree = NULL; /* read from ready process(es) */ + for (rt = rt0; rt != NULL; rt = rt->next) { + if (!FD_ISSET(rt->pd.r, &readset) && + !FD_ISSET(rt->pd.r, &errset)) + continue; + for (n = 0; n < rt->bsiz && rt->dest[n] != NULL; n++) + ; + errno = 0; + nr = read(rt->pd.r, (char *)rt->buf, 3*sizeof(float)*(n+1)); + if (nr < 0) + error(SYSTEM, "read error in raywait()"); + if (nr == 0) /* unexpected EOF */ + error(USER, "rtrace process died"); + if (nr < 3*sizeof(float)*(n+1)) { /* read the rest */ + nr = readbuf(rt->pd.r, (char *)rt->buf, + 3*sizeof(float)*(n+1) - nr); + if (nr < 0) + error(USER, "readbuf error in raywait()"); + } + while (n-- > 0) { + rt->dest[n][0] += rt->buf[3*n]; + rt->dest[n][1] += rt->buf[3*n+1]; + rt->dest[n][2] += rt->buf[3*n+2]; + rt->dest[n] = NULL; + } + rtfree = rt; + } + return(rtfree); } -mkaxes(u, v, n) /* compute u and v to go with n */ -FVECT u, v, n; +static void +mkaxes( /* compute u and v to go with n */ + FVECT u, + FVECT v, + FVECT n +) { register int i; @@ -316,9 +432,12 @@ FVECT u, v, n; } -rounddir(dv, alt, azi) /* compute uniform spherical direction */ -register FVECT dv; -double alt, azi; +static void +rounddir( /* compute uniform spherical direction */ + register FVECT dv, + double alt, + double azi +) { double d1, d2; @@ -330,9 +449,12 @@ double alt, azi; } -flatdir(dv, alt, azi) /* compute uniform hemispherical direction */ -register FVECT dv; -double alt, azi; +static void +flatdir( /* compute uniform hemispherical direction */ + register FVECT dv, + double alt, + double azi +) { double d1, d2;