src/gen/mkillum2.c

#ifndef lint
static const char       RCSid[] = "$Id: mkillum2.c,v 2.37 2011/08/16 18:09:53 greg Exp $";
#endif
/*
 * Routines to do the actual calculation for mkillum
 */

#include <string.h>

#include  "mkillum.h"
#include  "face.h"
#include  "cone.h"
#include  "source.h"
#include  "paths.h"

#ifndef NBSDFSAMPS
#define NBSDFSAMPS      256             /* BSDF resampling count */
#endif

COLORV *        distarr = NULL;         /* distribution array */
int             distsiz = 0;


void
newdist(                        /* allocate & clear distribution array */
        int siz
)
{
        if (siz <= 0) {
                if (distsiz > 0)
                        free(distarr);
                distarr = NULL;
                distsiz = 0;
                return;
        }
        if (distsiz < siz) {
                if (distsiz > 0)
                        free(distarr);
                distarr = (COLORV *)malloc(sizeof(COLOR)*siz);
                if (distarr == NULL)
                        error(SYSTEM, "out of memory in newdist");
                distsiz = siz;
        }
        memset(distarr, '\0', sizeof(COLOR)*siz);
}


int
process_ray(                    /* process a ray result or report error */
        RAY *r,
        int rv
)
{
        COLORV  *colp;

        if (rv == 0)                    /* no result ready */
                return(0);
        if (rv < 0)
                error(USER, "ray tracing process died");
        if (r->rno >= distsiz)
                error(INTERNAL, "bad returned index in process_ray");
        multcolor(r->rcol, r->rcoef);   /* in case it's a source ray */
        colp = &distarr[r->rno * 3];
        addcolor(colp, r->rcol);
        return(1);
}


void
raysamp(                        /* queue a ray sample */
        int  ndx,
        FVECT  org,
        FVECT  dir
)
{
        RAY     myRay;
        int     rv;

        if ((ndx < 0) | (ndx >= distsiz))
                error(INTERNAL, "bad index in raysamp");
        VCOPY(myRay.rorg, org);
        VCOPY(myRay.rdir, dir);
        myRay.rmax = .0;
        rayorigin(&myRay, PRIMARY|SPECULAR, NULL, NULL);
        myRay.rno = ndx;
                                        /* queue ray, check result */
        process_ray(&myRay, ray_pqueue(&myRay));
}


void
srcsamps(                       /* sample sources from this surface position */
        struct illum_args *il,
        FVECT org,
        FVECT nrm,
        MAT4 ixfm
)
{
        int  nalt=1, nazi=1;
        SRCINDEX  si;
        RAY  sr;
        FVECT   v;
        double  d;
        int  i, j;
                                                /* get sampling density */
        if (il->sampdens > 0) {
                i = PI * il->sampdens;
                nalt = sqrt(i/PI) + .5;
                nazi = PI*nalt + .5;
        }
        initsrcindex(&si);                      /* loop over (sub)sources */
        for ( ; ; ) {
                VCOPY(sr.rorg, org);            /* pick side to shoot from */
                d = 5.*FTINY;
                VSUM(sr.rorg, sr.rorg, nrm, d);
                samplendx++;                    /* increment sample counter */
                if (!srcray(&sr, NULL, &si))
                        break;                  /* end of sources */
                                                /* index direction */
                if (ixfm != NULL)
                        multv3(v, sr.rdir, ixfm);
                else
                        VCOPY(v, sr.rdir);
                if (v[2] >= -FTINY)
                        continue;       /* only sample transmission */
                v[0] = -v[0]; v[1] = -v[1]; v[2] = -v[2];
                sr.rno = flatindex(v, nalt, nazi);
                d = nalt*nazi*(1./PI) * v[2];
                d *= si.dom;                    /* solid angle correction */
                scalecolor(sr.rcoef, d);
                process_ray(&sr, ray_pqueue(&sr));
        }
}


void
rayclean()                      /* finish all pending rays */
{
        RAY     myRay;

        while (process_ray(&myRay, ray_presult(&myRay, 0)))
                ;
}


static void
mkaxes(                 /* compute u and v to go with n */
        FVECT  u,
        FVECT  v,
        FVECT  n
)
{
        register int  i;

        v[0] = v[1] = v[2] = 0.0;
        for (i = 0; i < 3; i++)
                if (n[i] < 0.6 && n[i] > -0.6)
                        break;
        v[i] = 1.0;
        fcross(u, v, n);
        normalize(u);
        fcross(v, n, u);
}


static void
rounddir(               /* compute uniform spherical direction */
        register FVECT  dv,
        double  alt,
        double  azi
)
{
        double  d1, d2;

        dv[2] = 1. - 2.*alt;
        d1 = sqrt(1. - dv[2]*dv[2]);
        d2 = 2.*PI * azi;
        dv[0] = d1*cos(d2);
        dv[1] = d1*sin(d2);
}


void
flatdir(                /* compute uniform hemispherical direction */
        FVECT  dv,
        double  alt,
        double  azi
)
{
        double  d1, d2;

        d1 = sqrt(alt);
        d2 = 2.*PI * azi;
        dv[0] = d1*cos(d2);
        dv[1] = d1*sin(d2);
        dv[2] = sqrt(1. - alt);
}


int
flatindex(              /* compute index for hemispherical direction */
        FVECT   dv,
        int     nalt,
        int     nazi
)
{
        double  d;
        int     i, j;
        
        d = 1.0 - dv[2]*dv[2];
        i = d*nalt;
        d = atan2(dv[1], dv[0]) * (0.5/PI);
        if (d < 0.0) d += 1.0;
        j = d*nazi + 0.5;
        if (j >= nazi) j = 0;
        return(i*nazi + j);
}


int
my_default(     /* default illum action */
        OBJREC  *ob,
        struct illum_args  *il,
        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);
}


int
my_face(                /* make an illum face */
        OBJREC  *ob,
        struct illum_args  *il,
        char  *nm
)
{
        int  dim[2];
        int  n, nalt, nazi, alti;
        double  sp[2], r1, r2;
        int  h;
        FVECT  dn, org, dir;
        FVECT  u, v;
        double  ur[2], vr[2];
        MAT4  xfm;
        char  xfrot[64];
        int  nallow;
        FACE  *fa;
        int  i, j;
                                /* get/check arguments */
        fa = getface(ob);
        if (fa->area == 0.0) {
                freeface(ob);
                return(my_default(ob, il, nm));
        }
                                /* set up sampling */
        if (il->sampdens <= 0) {
                nalt = nazi = 1;        /* diffuse assumption */
        } else {
                n = PI * il->sampdens;
                nalt = sqrt(n/PI) + .5;
                nazi = PI*nalt + .5;
        }
        n = nazi*nalt;
        newdist(n);
                                /* take first edge >= 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++) {
                r1 = DOT(VERTEX(fa,i),u);
                if (r1 < ur[0]) ur[0] = r1;
                if (r1 > ur[1]) ur[1] = r1;
                r2 = DOT(VERTEX(fa,i),v);
                if (r2 < vr[0]) vr[0] = r2;
                if (r2 > vr[1]) vr[1] = r2;
        }
        dim[0] = random();
                                /* sample polygon */
        nallow = 5*n*il->nsamps;
        for (dim[1] = 0; dim[1] < n; dim[1]++)
                for (i = 0; i < il->nsamps; i++) {
                                        /* randomize direction */
                    h = ilhash(dim, 2) + i;
                    multisamp(sp, 2, urand(h));
                    alti = dim[1]/nazi;
                    r1 = (alti + sp[0])/nalt;
                    r2 = (dim[1] - alti*nazi + 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];
                                        /* randomize location */
                    do {
                        multisamp(sp, 2, urand(h+4862+nallow));
                        r1 = ur[0] + (ur[1]-ur[0]) * sp[0];
                        r2 = vr[0] + (vr[1]-vr[0]) * sp[1];
                        for (j = 0; j < 3; j++)
                            org[j] = r1*u[j] + r2*v[j]
                                        + fa->offset*fa->norm[j];
                    } while (!inface(org, fa) && nallow-- > 0);
                    if (nallow < 0) {
                        objerror(ob, WARNING, "bad aspect");
                        rayclean();
                        freeface(ob);
                        return(my_default(ob, il, nm));
                    }
                    r1 = 5.*FTINY;
                    for (j = 0; j < 3; j++)
                        org[j] += r1*fa->norm[j];
                                        /* send sample */
                    raysamp(dim[1], org, dir);
                }
                                /* add in direct component? */
        if (il->flags & IL_LIGHT) {
                MAT4    ixfm;
                for (i = 3; i--; ) {
                        ixfm[i][0] = u[i];
                        ixfm[i][1] = v[i];
                        ixfm[i][2] = fa->norm[i];
                        ixfm[i][3] = 0.;
                }
                ixfm[3][0] = ixfm[3][1] = ixfm[3][2] = 0.;
                ixfm[3][3] = 1.;
                dim[0] = random();
                nallow = 10*il->nsamps;
                for (i = 0; i < il->nsamps; i++) {
                                        /* randomize location */
                    h = dim[0] + samplendx++;
                    do {
                        multisamp(sp, 2, urand(h+nallow));
                        r1 = ur[0] + (ur[1]-ur[0]) * sp[0];
                        r2 = vr[0] + (vr[1]-vr[0]) * sp[1];
                        for (j = 0; j < 3; j++)
                            org[j] = r1*u[j] + r2*v[j]
                                        + fa->offset*fa->norm[j];
                    } while (!inface(org, fa) && nallow-- > 0);
                    if (nallow < 0) {
                        objerror(ob, WARNING, "bad aspect");
                        rayclean();
                        freeface(ob);
                        return(my_default(ob, il, nm));
                    }
                                        /* sample source rays */
                    srcsamps(il, org, fa->norm, ixfm);
                }
        }
                                /* wait for all rays to finish */
        rayclean();
                                /* write out the face and its distribution */
        if (average(il, distarr, n)) {
                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);
        return(0);
}


int
my_sphere(      /* make an illum sphere */
        register OBJREC  *ob,
        struct illum_args  *il,
        char  *nm
)
{
        int  dim[3];
        int  n, nalt, nazi;
        double  sp[4], r1, r2, r3;
        FVECT  org, dir;
        FVECT  u, v;
        register int  i, j;
                                /* check arguments */
        if (ob->oargs.nfargs != 4)
                objerror(ob, USER, "bad # of arguments");
                                /* set up sampling */
        if (il->sampdens <= 0)
                nalt = nazi = 1;
        else {
                n = 4.*PI * il->sampdens;
                nalt = sqrt(2./PI*n) + .5;
                nazi = PI/2.*nalt + .5;
        }
        n = nalt*nazi;
        newdist(n);
        dim[0] = random();
                                /* sample sphere */
        for (dim[1] = 0; dim[1] < nalt; dim[1]++)
            for (dim[2] = 0; dim[2] < nazi; dim[2]++)
                for (i = 0; i < il->nsamps; i++) {
                                        /* next sample point */
                    multisamp(sp, 4, urand(ilhash(dim,3)+i));
                                        /* random direction */
                    r1 = (dim[1] + sp[0])/nalt;
                    r2 = (dim[2] + sp[1] - .5)/nazi;
                    rounddir(dir, r1, r2);
                                        /* random location */
                    mkaxes(u, v, dir);          /* yuck! */
                    r3 = sqrt(sp[2]);
                    r2 = 2.*PI*sp[3];
                    r1 = r3*ob->oargs.farg[3]*cos(r2);
                    r2 = r3*ob->oargs.farg[3]*sin(r2);
                    r3 = ob->oargs.farg[3]*sqrt(1.01-r3*r3);
                    for (j = 0; j < 3; j++) {
                        org[j] = ob->oargs.farg[j] + r1*u[j] + r2*v[j] +
                                        r3*dir[j];
                        dir[j] = -dir[j];
                    }
                                        /* send sample */
                    raysamp(dim[1]*nazi+dim[2], org, dir);
                }
                                /* wait for all rays to finish */
        rayclean();
                                /* write out the sphere and its distribution */
        if (average(il, distarr, n)) {
                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 */
        return(1);
}


int
my_ring(                /* make an illum ring */
        OBJREC  *ob,
        struct illum_args  *il,
        char  *nm
)
{
        int  dim[2];
        int  n, nalt, nazi, alti;
        double  sp[2], r1, r2, r3;
        int  h;
        FVECT  dn, org, dir;
        FVECT  u, v;
        MAT4  xfm;
        CONE  *co;
        int  i, j;
                                /* get/check arguments */
        co = getcone(ob, 0);
                                /* set up sampling */
        if (il->sampdens <= 0) {
                nalt = nazi = 1;        /* diffuse assumption */
        } else {
                n = PI * il->sampdens;
                nalt = sqrt(n/PI) + .5;
                nazi = PI*nalt + .5;
        }
        n = nazi*nalt;
        newdist(n);
        mkaxes(u, v, co->ad);
        dim[0] = random();
                                /* sample disk */
        for (dim[1] = 0; dim[1] < n; dim[1]++)
                for (i = 0; i < il->nsamps; i++) {
                                        /* next sample point */
                    h = ilhash(dim,2) + i;
                                        /* randomize direction */
                    multisamp(sp, 2, urand(h));
                    alti = dim[1]/nazi;
                    r1 = (alti + sp[0])/nalt;
                    r2 = (dim[1] - alti*nazi + 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];
                                        /* randomize location */
                    multisamp(sp, 2, urand(h+8371));
                    r3 = sqrt(CO_R0(co)*CO_R0(co) +
                            sp[0]*(CO_R1(co)*CO_R1(co) - CO_R0(co)*CO_R0(co)));
                    r2 = 2.*PI*sp[1];
                    r1 = r3*cos(r2);
                    r2 = r3*sin(r2);
                    r3 = 5.*FTINY;
                    for (j = 0; j < 3; j++)
                        org[j] = CO_P0(co)[j] + r1*u[j] + r2*v[j] +
                                                r3*co->ad[j];
                                        /* send sample */
                    raysamp(dim[1], org, dir);
                }
                                /* add in direct component? */
        if (il->flags & IL_LIGHT) {
                MAT4    ixfm;
                for (i = 3; i--; ) {
                        ixfm[i][0] = u[i];
                        ixfm[i][1] = v[i];
                        ixfm[i][2] = co->ad[i];
                        ixfm[i][3] = 0.;
                }
                ixfm[3][0] = ixfm[3][1] = ixfm[3][2] = 0.;
                ixfm[3][3] = 1.;
                dim[0] = random();
                for (i = 0; i < il->nsamps; i++) {
                                        /* randomize location */
                    h = dim[0] + samplendx++;
                    multisamp(sp, 2, urand(h));
                    r3 = sqrt(CO_R0(co)*CO_R0(co) +
                            sp[0]*(CO_R1(co)*CO_R1(co) - CO_R0(co)*CO_R0(co)));
                    r2 = 2.*PI*sp[1];
                    r1 = r3*cos(r2);
                    r2 = r3*sin(r2);
                    for (j = 0; j < 3; j++)
                        org[j] = CO_P0(co)[j] + r1*u[j] + r2*v[j];
                                        /* sample source rays */
                    srcsamps(il, org, co->ad, ixfm);
                }
        }
                                /* wait for all rays to finish */
        rayclean();
                                /* write out the ring and its distribution */
        if (average(il, distarr, n)) {
                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);
        return(1);
}
Revision:	2.38
Committed:	Sat Oct 13 20:15:43 2012 UTC (11 years, 6 months ago) by greg
Content type:	text/plain
Branch:	MAIN
CVS Tags:	rad4R2P2, rad4R2, rad4R2P1
Changes since 2.37:	+51 -327 lines
Log Message:	Corrected errors in XML interpreter and genBSDF and removed mkillum BSDF code
#	User	Rev	Content
1	greg	1.1	#ifndef lint
2	greg	2.38	static const char RCSid[] = "$Id: mkillum2.c,v 2.37 2011/08/16 18:09:53 greg Exp $";
3	greg	1.1	#endif
4			/*
5	greg	1.4	* Routines to do the actual calculation for mkillum
6	greg	1.1	*/
7
8	schorsch	2.11	#include <string.h>
9
10	greg	1.1	#include "mkillum.h"
11			#include "face.h"
12			#include "cone.h"
13	greg	2.25	#include "source.h"
14	greg	2.35	#include "paths.h"
15	greg	1.1
16	greg	2.31	#ifndef NBSDFSAMPS
17	greg	2.32	#define NBSDFSAMPS 256 /* BSDF resampling count */
18	greg	2.31	#endif
19	greg	2.13
20	greg	2.21	COLORV * distarr = NULL; /* distribution array */
21			int distsiz = 0;
22	greg	2.16
23	greg	2.31
24	greg	2.21	void
25	greg	2.18	newdist( /* allocate & clear distribution array */
26			int siz
27			)
28			{
29	greg	2.24	if (siz <= 0) {
30	greg	2.18	if (distsiz > 0)
31	greg	2.36	free(distarr);
32	greg	2.18	distarr = NULL;
33			distsiz = 0;
34			return;
35			}
36			if (distsiz < siz) {
37	greg	2.21	if (distsiz > 0)
38	greg	2.36	free(distarr);
39	greg	2.21	distarr = (COLORV )malloc(sizeof(COLOR)siz);
40	greg	2.18	if (distarr == NULL)
41	greg	2.20	error(SYSTEM, "out of memory in newdist");
42	greg	2.18	distsiz = siz;
43			}
44	greg	2.21	memset(distarr, '\0', sizeof(COLOR)*siz);
45	greg	2.18	}
46
47
48	greg	2.21	int
49	greg	2.25	process_ray( /* process a ray result or report error */
50			RAY *r,
51			int rv
52			)
53	greg	2.18	{
54			COLORV *colp;
55
56	greg	2.24	if (rv == 0) /* no result ready */
57	greg	2.18	return(0);
58			if (rv < 0)
59	greg	2.20	error(USER, "ray tracing process died");
60	greg	2.18	if (r->rno >= distsiz)
61	greg	2.20	error(INTERNAL, "bad returned index in process_ray");
62	greg	2.25	multcolor(r->rcol, r->rcoef); /* in case it's a source ray */
63	greg	2.18	colp = &distarr[r->rno * 3];
64			addcolor(colp, r->rcol);
65			return(1);
66			}
67
68
69	greg	2.21	void
70	greg	2.25	raysamp( /* queue a ray sample */
71	greg	2.18	int ndx,
72			FVECT org,
73			FVECT dir
74	greg	2.16	)
75			{
76	greg	2.18	RAY myRay;
77			int rv;
78
79			if ((ndx < 0) \| (ndx >= distsiz))
80	greg	2.20	error(INTERNAL, "bad index in raysamp");
81	greg	2.18	VCOPY(myRay.rorg, org);
82			VCOPY(myRay.rdir, dir);
83			myRay.rmax = .0;
84	greg	2.36	rayorigin(&myRay, PRIMARY\|SPECULAR, NULL, NULL);
85	greg	2.18	myRay.rno = ndx;
86			/* queue ray, check result */
87			process_ray(&myRay, ray_pqueue(&myRay));
88			}
89
90
91	greg	2.21	void
92	greg	2.25	srcsamps( /* sample sources from this surface position */
93			struct illum_args *il,
94			FVECT org,
95			FVECT nrm,
96			MAT4 ixfm
97			)
98			{
99	greg	2.37	int nalt=1, nazi=1;
100	greg	2.25	SRCINDEX si;
101			RAY sr;
102			FVECT v;
103			double d;
104			int i, j;
105			/* get sampling density */
106	greg	2.38	if (il->sampdens > 0) {
107	greg	2.25	i = PI * il->sampdens;
108			nalt = sqrt(i/PI) + .5;
109			nazi = PI*nalt + .5;
110			}
111			initsrcindex(&si); /* loop over (sub)sources */
112			for ( ; ; ) {
113			VCOPY(sr.rorg, org); /* pick side to shoot from */
114	greg	2.38	d = 5.*FTINY;
115	greg	2.37	VSUM(sr.rorg, sr.rorg, nrm, d);
116	greg	2.30	samplendx++; /* increment sample counter */
117	greg	2.25	if (!srcray(&sr, NULL, &si))
118			break; /* end of sources */
119			/* index direction */
120			if (ixfm != NULL)
121			multv3(v, sr.rdir, ixfm);
122			else
123			VCOPY(v, sr.rdir);
124	greg	2.38	if (v[2] >= -FTINY)
125			continue; /* only sample transmission */
126			v[0] = -v[0]; v[1] = -v[1]; v[2] = -v[2];
127			sr.rno = flatindex(v, nalt, nazi);
128			d = naltnazi(1./PI) * v[2];
129	greg	2.25	d = si.dom; / solid angle correction */
130			scalecolor(sr.rcoef, d);
131			process_ray(&sr, ray_pqueue(&sr));
132			}
133			}
134
135
136			void
137	greg	2.18	rayclean() /* finish all pending rays */
138			{
139			RAY myRay;
140
141			while (process_ray(&myRay, ray_presult(&myRay, 0)))
142	greg	2.16	;
143			}
144	schorsch	2.12
145
146	greg	2.21	static void
147			mkaxes( /* compute u and v to go with n */
148			FVECT u,
149			FVECT v,
150			FVECT n
151			)
152			{
153			register int i;
154
155			v[0] = v[1] = v[2] = 0.0;
156			for (i = 0; i < 3; i++)
157			if (n[i] < 0.6 && n[i] > -0.6)
158			break;
159			v[i] = 1.0;
160			fcross(u, v, n);
161			normalize(u);
162			fcross(v, n, u);
163			}
164
165
166			static void
167			rounddir( /* compute uniform spherical direction */
168			register FVECT dv,
169			double alt,
170			double azi
171			)
172			{
173			double d1, d2;
174
175			dv[2] = 1. - 2.*alt;
176			d1 = sqrt(1. - dv[2]*dv[2]);
177			d2 = 2.PI azi;
178			dv[0] = d1*cos(d2);
179			dv[1] = d1*sin(d2);
180			}
181
182
183	greg	2.24	void
184	greg	2.21	flatdir( /* compute uniform hemispherical direction */
185	greg	2.24	FVECT dv,
186	greg	2.21	double alt,
187			double azi
188			)
189			{
190			double d1, d2;
191
192			d1 = sqrt(alt);
193			d2 = 2.PI azi;
194			dv[0] = d1*cos(d2);
195			dv[1] = d1*sin(d2);
196			dv[2] = sqrt(1. - alt);
197			}
198
199	greg	2.35
200	greg	2.29	int
201			flatindex( /* compute index for hemispherical direction */
202			FVECT dv,
203			int nalt,
204			int nazi
205			)
206			{
207			double d;
208			int i, j;
209
210			d = 1.0 - dv[2]*dv[2];
211			i = d*nalt;
212			d = atan2(dv[1], dv[0]) * (0.5/PI);
213			if (d < 0.0) d += 1.0;
214			j = d*nazi + 0.5;
215			if (j >= nazi) j = 0;
216			return(i*nazi + j);
217			}
218
219	greg	2.21
220	greg	2.19	int
221	greg	2.18	my_default( /* default illum action */
222	schorsch	2.12	OBJREC *ob,
223			struct illum_args *il,
224			char *nm
225			)
226	greg	1.1	{
227	greg	1.2	sprintf(errmsg, "(%s): cannot make illum for %s \"%s\"",
228			nm, ofun[ob->otype].funame, ob->oname);
229			error(WARNING, errmsg);
230	greg	2.2	printobj(il->altmat, ob);
231	greg	2.13	return(1);
232	greg	1.2	}
233
234
235	greg	2.13	int
236	greg	2.18	my_face( /* make an illum face */
237	schorsch	2.12	OBJREC *ob,
238			struct illum_args *il,
239			char *nm
240			)
241	greg	1.2	{
242	greg	2.21	int dim[2];
243	greg	2.25	int n, nalt, nazi, alti;
244	greg	1.10	double sp[2], r1, r2;
245	greg	2.25	int h;
246	greg	1.4	FVECT dn, org, dir;
247	greg	1.3	FVECT u, v;
248			double ur[2], vr[2];
249	greg	2.21	MAT4 xfm;
250	greg	2.35	char xfrot[64];
251	greg	2.25	int nallow;
252	greg	2.21	FACE *fa;
253	greg	2.25	int i, j;
254	greg	1.3	/* get/check arguments */
255			fa = getface(ob);
256			if (fa->area == 0.0) {
257			freeface(ob);
258	greg	2.19	return(my_default(ob, il, nm));
259	greg	1.3	}
260			/* set up sampling */
261	greg	2.38	if (il->sampdens <= 0) {
262			nalt = nazi = 1; /* diffuse assumption */
263	greg	2.22	} else {
264	greg	2.38	n = PI * il->sampdens;
265			nalt = sqrt(n/PI) + .5;
266			nazi = PI*nalt + .5;
267	greg	2.22	}
268	greg	2.38	n = nazi*nalt;
269	greg	2.18	newdist(n);
270	greg	2.20	/* take first edge >= sqrt(area) */
271	greg	2.4	for (j = fa->nv-1, i = 0; i < fa->nv; j = i++) {
272			u[0] = VERTEX(fa,i)[0] - VERTEX(fa,j)[0];
273			u[1] = VERTEX(fa,i)[1] - VERTEX(fa,j)[1];
274			u[2] = VERTEX(fa,i)[2] - VERTEX(fa,j)[2];
275	greg	2.5	if ((r1 = DOT(u,u)) >= fa->area-FTINY)
276	greg	2.3	break;
277			}
278			if (i < fa->nv) { /* got one! -- let's align our axes */
279	greg	2.5	r2 = 1.0/sqrt(r1);
280			u[0] = r2; u[1] = r2; u[2] *= r2;
281	greg	2.3	fcross(v, fa->norm, u);
282			} else /* oh well, we'll just have to wing it */
283			mkaxes(u, v, fa->norm);
284			/* now, find limits in (u,v) coordinates */
285	greg	1.3	ur[0] = vr[0] = FHUGE;
286			ur[1] = vr[1] = -FHUGE;
287			for (i = 0; i < fa->nv; i++) {
288			r1 = DOT(VERTEX(fa,i),u);
289			if (r1 < ur[0]) ur[0] = r1;
290			if (r1 > ur[1]) ur[1] = r1;
291			r2 = DOT(VERTEX(fa,i),v);
292			if (r2 < vr[0]) vr[0] = r2;
293			if (r2 > vr[1]) vr[1] = r2;
294			}
295			dim[0] = random();
296			/* sample polygon */
297	greg	2.25	nallow = 5nil->nsamps;
298	greg	2.21	for (dim[1] = 0; dim[1] < n; dim[1]++)
299	greg	1.3	for (i = 0; i < il->nsamps; i++) {
300	greg	2.23	/* randomize direction */
301	greg	2.21	h = ilhash(dim, 2) + i;
302	greg	2.38	multisamp(sp, 2, urand(h));
303			alti = dim[1]/nazi;
304			r1 = (alti + sp[0])/nalt;
305			r2 = (dim[1] - alti*nazi + sp[1] - .5)/nazi;
306			flatdir(dn, r1, r2);
307			for (j = 0; j < 3; j++)
308	greg	2.21	dir[j] = -dn[0]u[j] - dn[1]v[j] -
309			dn[2]*fa->norm[j];
310	greg	2.23	/* randomize location */
311	greg	1.3	do {
312	greg	2.25	multisamp(sp, 2, urand(h+4862+nallow));
313	greg	1.10	r1 = ur[0] + (ur[1]-ur[0]) * sp[0];
314			r2 = vr[0] + (vr[1]-vr[0]) * sp[1];
315	greg	1.3	for (j = 0; j < 3; j++)
316			org[j] = r1u[j] + r2v[j]
317			+ fa->offset*fa->norm[j];
318	greg	2.25	} while (!inface(org, fa) && nallow-- > 0);
319			if (nallow < 0) {
320	greg	1.3	objerror(ob, WARNING, "bad aspect");
321	greg	2.18	rayclean();
322	greg	1.3	freeface(ob);
323	greg	2.19	return(my_default(ob, il, nm));
324	greg	1.3	}
325	greg	2.38	r1 = 5.*FTINY;
326	greg	1.3	for (j = 0; j < 3; j++)
327	greg	2.21	org[j] += r1*fa->norm[j];
328	greg	1.3	/* send sample */
329	greg	2.21	raysamp(dim[1], org, dir);
330	greg	1.3	}
331	greg	2.25	/* add in direct component? */
332	greg	2.38	if (il->flags & IL_LIGHT) {
333	greg	2.25	MAT4 ixfm;
334	greg	2.38	for (i = 3; i--; ) {
335			ixfm[i][0] = u[i];
336			ixfm[i][1] = v[i];
337			ixfm[i][2] = fa->norm[i];
338			ixfm[i][3] = 0.;
339	greg	2.29	}
340	greg	2.38	ixfm[3][0] = ixfm[3][1] = ixfm[3][2] = 0.;
341			ixfm[3][3] = 1.;
342	greg	2.25	dim[0] = random();
343			nallow = 10*il->nsamps;
344			for (i = 0; i < il->nsamps; i++) {
345			/* randomize location */
346			h = dim[0] + samplendx++;
347			do {
348			multisamp(sp, 2, urand(h+nallow));
349			r1 = ur[0] + (ur[1]-ur[0]) * sp[0];
350			r2 = vr[0] + (vr[1]-vr[0]) * sp[1];
351			for (j = 0; j < 3; j++)
352			org[j] = r1u[j] + r2v[j]
353			+ fa->offset*fa->norm[j];
354			} while (!inface(org, fa) && nallow-- > 0);
355			if (nallow < 0) {
356			objerror(ob, WARNING, "bad aspect");
357			rayclean();
358			freeface(ob);
359			return(my_default(ob, il, nm));
360			}
361			/* sample source rays */
362			srcsamps(il, org, fa->norm, ixfm);
363			}
364			}
365			/* wait for all rays to finish */
366	greg	2.18	rayclean();
367	greg	1.11	/* write out the face and its distribution */
368	greg	2.21	if (average(il, distarr, n)) {
369	greg	1.12	if (il->sampdens > 0)
370			flatout(il, distarr, nalt, nazi, u, v, fa->norm);
371			illumout(il, ob);
372	greg	2.2	} else
373	greg	1.12	printobj(il->altmat, ob);
374	greg	1.3	/* clean up */
375			freeface(ob);
376	greg	2.15	return(0);
377	greg	1.2	}
378
379
380	greg	2.13	int
381	greg	2.18	my_sphere( /* make an illum sphere */
382	schorsch	2.12	register OBJREC *ob,
383			struct illum_args *il,
384			char *nm
385			)
386	greg	1.2	{
387	greg	1.10	int dim[3];
388	greg	1.2	int n, nalt, nazi;
389	greg	1.10	double sp[4], r1, r2, r3;
390	greg	1.4	FVECT org, dir;
391	greg	1.2	FVECT u, v;
392			register int i, j;
393			/* check arguments */
394			if (ob->oargs.nfargs != 4)
395			objerror(ob, USER, "bad # of arguments");
396			/* set up sampling */
397	greg	1.11	if (il->sampdens <= 0)
398			nalt = nazi = 1;
399			else {
400			n = 4.PI il->sampdens;
401	greg	2.7	nalt = sqrt(2./PI*n) + .5;
402			nazi = PI/2.*nalt + .5;
403	greg	1.11	}
404	greg	1.2	n = nalt*nazi;
405	greg	2.18	newdist(n);
406	greg	1.2	dim[0] = random();
407			/* sample sphere */
408			for (dim[1] = 0; dim[1] < nalt; dim[1]++)
409	greg	1.8	for (dim[2] = 0; dim[2] < nazi; dim[2]++)
410	greg	1.2	for (i = 0; i < il->nsamps; i++) {
411	greg	1.10	/* next sample point */
412	greg	1.11	multisamp(sp, 4, urand(ilhash(dim,3)+i));
413	greg	1.2	/* random direction */
414	greg	1.10	r1 = (dim[1] + sp[0])/nalt;
415	greg	1.13	r2 = (dim[2] + sp[1] - .5)/nazi;
416	greg	1.2	rounddir(dir, r1, r2);
417			/* random location */
418	greg	1.8	mkaxes(u, v, dir); /* yuck! */
419	greg	1.10	r3 = sqrt(sp[2]);
420			r2 = 2.PIsp[3];
421	greg	1.5	r1 = r3ob->oargs.farg[3]cos(r2);
422			r2 = r3ob->oargs.farg[3]sin(r2);
423			r3 = ob->oargs.farg[3]sqrt(1.01-r3r3);
424			for (j = 0; j < 3; j++) {
425			org[j] = ob->oargs.farg[j] + r1u[j] + r2v[j] +
426			r3*dir[j];
427			dir[j] = -dir[j];
428			}
429	greg	1.2	/* send sample */
430	greg	2.18	raysamp(dim[1]*nazi+dim[2], org, dir);
431	greg	1.2	}
432	greg	2.25	/* wait for all rays to finish */
433	greg	2.18	rayclean();
434	greg	1.11	/* write out the sphere and its distribution */
435	greg	2.21	if (average(il, distarr, n)) {
436	greg	1.12	if (il->sampdens > 0)
437			roundout(il, distarr, nalt, nazi);
438			else
439			objerror(ob, WARNING, "diffuse distribution");
440			illumout(il, ob);
441	greg	2.2	} else
442	greg	1.12	printobj(il->altmat, ob);
443	greg	1.2	/* clean up */
444	greg	2.13	return(1);
445	greg	1.2	}
446
447
448	greg	2.13	int
449	greg	2.18	my_ring( /* make an illum ring */
450	schorsch	2.12	OBJREC *ob,
451			struct illum_args *il,
452			char *nm
453			)
454	greg	1.2	{
455	greg	2.21	int dim[2];
456			int n, nalt, nazi, alti;
457			double sp[2], r1, r2, r3;
458			int h;
459	greg	1.4	FVECT dn, org, dir;
460	greg	1.3	FVECT u, v;
461	greg	2.21	MAT4 xfm;
462			CONE *co;
463	greg	2.25	int i, j;
464	greg	1.3	/* get/check arguments */
465			co = getcone(ob, 0);
466			/* set up sampling */
467	greg	2.38	if (il->sampdens <= 0) {
468			nalt = nazi = 1; /* diffuse assumption */
469	greg	2.22	} else {
470	greg	2.38	n = PI * il->sampdens;
471			nalt = sqrt(n/PI) + .5;
472			nazi = PI*nalt + .5;
473	greg	2.22	}
474	greg	2.38	n = nazi*nalt;
475	greg	2.18	newdist(n);
476	greg	1.3	mkaxes(u, v, co->ad);
477			dim[0] = random();
478			/* sample disk */
479	greg	2.21	for (dim[1] = 0; dim[1] < n; dim[1]++)
480	greg	1.3	for (i = 0; i < il->nsamps; i++) {
481	greg	1.10	/* next sample point */
482	greg	2.21	h = ilhash(dim,2) + i;
483	greg	2.23	/* randomize direction */
484	greg	2.38	multisamp(sp, 2, urand(h));
485			alti = dim[1]/nazi;
486			r1 = (alti + sp[0])/nalt;
487			r2 = (dim[1] - alti*nazi + sp[1] - .5)/nazi;
488			flatdir(dn, r1, r2);
489			for (j = 0; j < 3; j++)
490			dir[j] = -dn[0]u[j] - dn[1]v[j] - dn[2]*co->ad[j];
491	greg	2.23	/* randomize location */
492	greg	2.21	multisamp(sp, 2, urand(h+8371));
493	greg	1.5	r3 = sqrt(CO_R0(co)*CO_R0(co) +
494	greg	2.21	sp[0](CO_R1(co)CO_R1(co) - CO_R0(co)*CO_R0(co)));
495			r2 = 2.PIsp[1];
496	greg	1.5	r1 = r3*cos(r2);
497			r2 = r3*sin(r2);
498	greg	2.38	r3 = 5.*FTINY;
499	greg	1.3	for (j = 0; j < 3; j++)
500	greg	2.6	org[j] = CO_P0(co)[j] + r1u[j] + r2v[j] +
501	greg	2.21	r3*co->ad[j];
502	greg	1.3	/* send sample */
503	greg	2.21	raysamp(dim[1], org, dir);
504	greg	1.3	}
505	greg	2.25	/* add in direct component? */
506	greg	2.38	if (il->flags & IL_LIGHT) {
507	greg	2.25	MAT4 ixfm;
508	greg	2.38	for (i = 3; i--; ) {
509			ixfm[i][0] = u[i];
510			ixfm[i][1] = v[i];
511			ixfm[i][2] = co->ad[i];
512			ixfm[i][3] = 0.;
513	greg	2.29	}
514	greg	2.38	ixfm[3][0] = ixfm[3][1] = ixfm[3][2] = 0.;
515			ixfm[3][3] = 1.;
516	greg	2.25	dim[0] = random();
517			for (i = 0; i < il->nsamps; i++) {
518			/* randomize location */
519			h = dim[0] + samplendx++;
520			multisamp(sp, 2, urand(h));
521			r3 = sqrt(CO_R0(co)*CO_R0(co) +
522			sp[0](CO_R1(co)CO_R1(co) - CO_R0(co)*CO_R0(co)));
523			r2 = 2.PIsp[1];
524			r1 = r3*cos(r2);
525			r2 = r3*sin(r2);
526			for (j = 0; j < 3; j++)
527			org[j] = CO_P0(co)[j] + r1u[j] + r2v[j];
528			/* sample source rays */
529			srcsamps(il, org, co->ad, ixfm);
530			}
531			}
532			/* wait for all rays to finish */
533	greg	2.18	rayclean();
534	greg	1.11	/* write out the ring and its distribution */
535	greg	2.21	if (average(il, distarr, n)) {
536	greg	1.12	if (il->sampdens > 0)
537			flatout(il, distarr, nalt, nazi, u, v, co->ad);
538			illumout(il, ob);
539	greg	2.2	} else
540	greg	1.12	printobj(il->altmat, ob);
541	greg	1.3	/* clean up */
542			freecone(ob);
543	greg	2.13	return(1);
544	greg	1.2	}