--- ray/src/gen/mkillum2.c 1992/03/11 12:25:47 2.2 +++ ray/src/gen/mkillum2.c 2004/09/17 21:43:50 2.16 @@ -1,40 +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); 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]; @@ -51,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) @@ -66,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++) { @@ -102,18 +133,17 @@ 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 */ if (average(il, distarr, nalt*nazi)) { if (il->sampdens > 0) @@ -123,16 +153,19 @@ char *nm; 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; @@ -149,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)); @@ -180,9 +213,9 @@ 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 */ if (average(il, distarr, nalt*nazi)) { if (il->sampdens > 0) @@ -193,15 +226,18 @@ char *nm; } 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; @@ -246,13 +282,13 @@ 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 */ if (average(il, distarr, nalt*nazi)) { if (il->sampdens > 0) @@ -262,52 +298,126 @@ char *nm; 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)); - errno = 0; - 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; @@ -322,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; @@ -336,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;