ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/rt/rcontrib.c
Revision: 2.22
Committed: Sat Jul 19 20:03:03 2014 UTC (9 years, 9 months ago) by greg
Content type: text/plain
Branch: MAIN
CVS Tags: rad4R2
Changes since 2.21: +29 -24 lines
Log Message:
Fixed bug in parameter parser and added check during main arguments processing

File Contents

# User Rev Content
1 greg 2.1 #ifndef lint
2 greg 2.22 static const char RCSid[] = "$Id: rcontrib.c,v 2.21 2014/07/19 18:39:35 greg Exp $";
3 greg 2.1 #endif
4     /*
5     * Accumulate ray contributions for a set of materials
6     * Initialization and calculation routines
7     */
8    
9 greg 2.20 #include "copyright.h"
10    
11     #include <ctype.h>
12 greg 2.1 #include "rcontrib.h"
13     #include "otypes.h"
14 greg 2.14 #include "source.h"
15 greg 2.1
16 greg 2.2 char *shm_boundary = NULL; /* boundary of shared memory */
17    
18 greg 2.1 CUBE thescene; /* our scene */
19     OBJECT nsceneobjs; /* number of objects in our scene */
20    
21     int dimlist[MAXDIM]; /* sampling dimensions */
22     int ndims = 0; /* number of sampling dimensions */
23     int samplendx = 0; /* index for this sample */
24    
25     static void trace_contrib(RAY *r); /* our trace callback */
26     void (*trace)() = trace_contrib;
27    
28     int do_irrad = 0; /* compute irradiance? */
29    
30 greg 2.3 int rand_samp = 1; /* pure Monte Carlo sampling? */
31 greg 2.1
32 greg 2.8 double dstrsrc = 0.9; /* square source distribution */
33 greg 2.1 double shadthresh = .03; /* shadow threshold */
34     double shadcert = .75; /* shadow certainty */
35     int directrelay = 3; /* number of source relays */
36     int vspretest = 512; /* virtual source pretest density */
37     int directvis = 1; /* sources visible? */
38     double srcsizerat = .2; /* maximum ratio source size/dist. */
39    
40     COLOR cextinction = BLKCOLOR; /* global extinction coefficient */
41     COLOR salbedo = BLKCOLOR; /* global scattering albedo */
42     double seccg = 0.; /* global scattering eccentricity */
43     double ssampdist = 0.; /* scatter sampling distance */
44    
45     double specthresh = .15; /* specular sampling threshold */
46     double specjitter = 1.; /* specular sampling jitter */
47    
48     int backvis = 1; /* back face visibility */
49    
50 greg 2.3 int maxdepth = -10; /* maximum recursion depth */
51     double minweight = 2e-3; /* minimum ray weight */
52 greg 2.1
53     char *ambfile = NULL; /* ambient file name */
54     COLOR ambval = BLKCOLOR; /* ambient value */
55     int ambvwt = 0; /* initial weight for ambient value */
56     double ambacc = 0; /* ambient accuracy */
57     int ambres = 256; /* ambient resolution */
58     int ambdiv = 350; /* ambient divisions */
59     int ambssamp = 0; /* ambient super-samples */
60     int ambounce = 1; /* ambient bounces */
61     char *amblist[AMBLLEN+1]; /* ambient include/exclude list */
62     int ambincl = -1; /* include == 1, exclude == 0 */
63    
64     int account; /* current accumulation count */
65     RNUMBER raysleft; /* number of rays left to trace */
66     long waitflush; /* how long until next flush */
67    
68 greg 2.4 RNUMBER lastray = 0; /* last ray number sent */
69     RNUMBER lastdone = 0; /* last ray output */
70 greg 2.1
71     static void mcfree(void *p) { epfree((*(MODCONT *)p).binv); free(p); }
72    
73     LUTAB modconttab = LU_SINIT(NULL,mcfree); /* modifier lookup table */
74    
75     /************************** INITIALIZATION ROUTINES ***********************/
76    
77     char *
78     formstr( /* return format identifier */
79     int f
80     )
81     {
82     switch (f) {
83     case 'a': return("ascii");
84     case 'f': return("float");
85     case 'd': return("double");
86     case 'c': return(COLRFMT);
87     }
88     return("unknown");
89     }
90    
91    
92     /* Add modifier to our list to track */
93     MODCONT *
94 greg 2.20 addmodifier(char *modn, char *outf, char *prms, char *binv, int bincnt)
95 greg 2.1 {
96     LUENT *lep = lu_find(&modconttab,modn);
97     MODCONT *mp;
98     EPNODE *ebinv;
99     int i;
100    
101     if (lep->data != NULL) {
102     sprintf(errmsg, "duplicate modifier '%s'", modn);
103     error(USER, errmsg);
104     }
105     if (nmods >= MAXMODLIST)
106     error(INTERNAL, "too many modifiers");
107     modname[nmods++] = modn; /* XXX assumes static string */
108     lep->key = modn; /* XXX assumes static string */
109     if (binv == NULL)
110     binv = "0"; /* use single bin if unspecified */
111     ebinv = eparse(binv);
112     if (ebinv->type == NUM) { /* check value if constant */
113     bincnt = (int)(evalue(ebinv) + 1.5);
114     if (bincnt != 1) {
115     sprintf(errmsg, "illegal non-zero constant for bin (%s)",
116     binv);
117     error(USER, errmsg);
118     }
119     } else if (bincnt <= 0) {
120     sprintf(errmsg,
121     "unspecified or illegal bin count for modifier '%s'",
122     modn);
123     error(USER, errmsg);
124     }
125     /* initialize results holder */
126     mp = (MODCONT *)malloc(sizeof(MODCONT)+sizeof(DCOLOR)*(bincnt-1));
127     if (mp == NULL)
128     error(SYSTEM, "out of memory in addmodifier");
129     mp->outspec = outf; /* XXX assumes static string */
130     mp->modname = modn; /* XXX assumes static string */
131 greg 2.20 mp->params = prms;
132 greg 2.1 mp->binv = ebinv;
133     mp->nbins = bincnt;
134     memset(mp->cbin, 0, sizeof(DCOLOR)*bincnt);
135     /* allocate output streams */
136     for (i = bincnt; i-- > 0; )
137     getostream(mp->outspec, mp->modname, i, 1);
138     lep->data = (char *)mp;
139     return(mp);
140     }
141    
142    
143 greg 2.10 /* Add modifiers from a file list */
144 greg 2.1 void
145 greg 2.20 addmodfile(char *fname, char *outf, char *prms, char *binv, int bincnt)
146 greg 2.1 {
147     char *mname[MAXMODLIST];
148     int i;
149     /* find the file & store strings */
150     if (wordfile(mname, getpath(fname, getrlibpath(), R_OK)) < 0) {
151     sprintf(errmsg, "cannot find modifier file '%s'", fname);
152     error(SYSTEM, errmsg);
153     }
154     for (i = 0; mname[i]; i++) /* add each one */
155 greg 2.20 addmodifier(mname[i], outf, prms, binv, bincnt);
156 greg 2.1 }
157    
158    
159     void
160     quit( /* quit program */
161     int code
162     )
163     {
164     if (nchild > 0) /* close children if any */
165 greg 2.10 end_children(code != 0);
166 greg 2.1 exit(code);
167     }
168    
169    
170     /* Initialize our process(es) */
171     static void
172     rcinit()
173     {
174     int i;
175    
176     if (nproc > MAXPROCESS)
177     sprintf(errmsg, "too many processes requested -- reducing to %d",
178     nproc = MAXPROCESS);
179 greg 2.2 if (nproc > 1) {
180     preload_objs(); /* preload auxiliary data */
181     /* set shared memory boundary */
182     shm_boundary = strcpy((char *)malloc(16), "SHM_BOUNDARY");
183     }
184 greg 2.14 for (i = 0; i < nsources; i++) /* tracing to sources as well */
185     source[i].sflags |= SFOLLOW;
186 greg 2.1 if (yres > 0) { /* set up flushing & ray counts */
187     if (xres > 0)
188     raysleft = (RNUMBER)xres*yres;
189     else
190     raysleft = yres;
191     } else
192     raysleft = 0;
193     if ((account = accumulate) > 1)
194     raysleft *= accumulate;
195     waitflush = (yres > 0) & (xres > 1) ? 0 : xres;
196    
197 greg 2.5 if (nproc > 1 && in_rchild()) /* forked child? */
198 greg 2.1 return; /* return to main processing loop */
199    
200 greg 2.5 if (recover) { /* recover previous output? */
201 greg 2.8 if (accumulate <= 0)
202 greg 2.5 reload_output();
203 greg 2.8 else
204 greg 2.5 recover_output();
205     }
206     if (nproc == 1) /* single process? */
207     return;
208 greg 2.8 /* else run appropriate controller */
209     if (accumulate <= 0)
210     feeder_loop();
211     else
212     parental_loop();
213 greg 2.1 quit(0); /* parent musn't return! */
214     }
215    
216     /************************** MAIN CALCULATION PROCESS ***********************/
217    
218 greg 2.20 /* Set parameters for current bin evaluation */
219 greg 2.22 void
220 greg 2.20 set_eparams(char *prms)
221     {
222 greg 2.22 static char *last_params = NULL;
223     char vname[RMAXWORD];
224     double value;
225     char *cpd;
226     /* check if already set */
227     if ((prms == NULL) | (prms == last_params))
228     return;
229     if (last_params != NULL && !strcmp(prms, last_params))
230     return;
231     last_params = prms; /* assign each variable */
232     while (*prms) {
233     if (isspace(*prms)) {
234     ++prms; continue;
235     }
236     if (!isalpha(*prms))
237 greg 2.20 goto bad_params;
238     cpd = vname;
239 greg 2.22 while (*prms && (*prms != '=') & !isspace(*prms)) {
240     if (!isid(*prms))
241 greg 2.20 goto bad_params;
242 greg 2.22 *cpd++ = *prms++;
243 greg 2.20 }
244     if (cpd == vname)
245     goto bad_params;
246     *cpd = '\0';
247 greg 2.22 while (isspace(*prms)) prms++;
248     if (*prms++ != '=')
249 greg 2.20 goto bad_params;
250 greg 2.22 value = atof(prms);
251     if ((prms = fskip(prms)) == NULL)
252 greg 2.20 goto bad_params;
253 greg 2.22 while (isspace(*prms)) prms++;
254     prms += (*prms == ',') | (*prms == ';');
255 greg 2.20 varset(vname, '=', value);
256     }
257     return;
258     bad_params:
259 greg 2.22 sprintf(errmsg, "bad parameter list '%s'", last_params);
260 greg 2.20 error(USER, errmsg);
261     }
262    
263 greg 2.22
264 greg 2.1 /* Our trace call to sum contributions */
265     static void
266     trace_contrib(RAY *r)
267     {
268     MODCONT *mp;
269 greg 2.19 double bval;
270 greg 2.1 int bn;
271     RREAL contr[3];
272    
273 greg 2.6 if (r->ro == NULL || r->ro->omod == OVOID)
274 greg 2.1 return;
275 greg 2.20 /* shadow ray not on source? */
276     if (r->rsrc >= 0 && source[r->rsrc].so != r->ro)
277     return;
278 greg 2.1
279     mp = (MODCONT *)lu_find(&modconttab,objptr(r->ro->omod)->oname)->data;
280    
281 greg 2.6 if (mp == NULL) /* not in our list? */
282     return;
283 greg 2.1
284 greg 2.21 worldfunc(RCCONTEXT, r); /* else set context */
285 greg 2.22 set_eparams((char *)mp->params);
286 greg 2.21 if ((bval = evalue(mp->binv)) <= -.5) /* and get bin number */
287     return; /* silently ignore negatives */
288 greg 2.19 if ((bn = (int)(bval + .5)) >= mp->nbins) {
289 greg 2.1 error(WARNING, "bad bin number (ignored)");
290     return;
291     }
292 greg 2.14 raycontrib(contr, r, PRIMARY); /* compute coefficient */
293 greg 2.1 if (contrib)
294 greg 2.14 multcolor(contr, r->rcol); /* -> contribution */
295 greg 2.1 addcolor(mp->cbin[bn], contr);
296     }
297    
298    
299 greg 2.7 /* Evaluate irradiance contributions */
300 greg 2.1 static void
301 greg 2.7 eval_irrad(FVECT org, FVECT dir)
302 greg 2.1 {
303 greg 2.7 RAY thisray;
304    
305     VSUM(thisray.rorg, org, dir, 1.1e-4);
306     thisray.rdir[0] = -dir[0];
307     thisray.rdir[1] = -dir[1];
308     thisray.rdir[2] = -dir[2];
309     thisray.rmax = 0.0;
310     rayorigin(&thisray, PRIMARY, NULL, NULL);
311 greg 2.16 /* pretend we hit surface */
312     thisray.rt = thisray.rot = 1e-5;
313 greg 2.7 thisray.rod = 1.0;
314 greg 2.15 VCOPY(thisray.ron, dir);
315 greg 2.7 VSUM(thisray.rop, org, dir, 1e-4);
316     samplendx++; /* compute result */
317     (*ofun[Lamb.otype].funp)(&Lamb, &thisray);
318 greg 2.1 }
319    
320    
321 greg 2.7 /* Evaluate radiance contributions */
322 greg 2.1 static void
323 greg 2.7 eval_rad(FVECT org, FVECT dir, double dmax)
324 greg 2.1 {
325     RAY thisray;
326     /* set up ray */
327 greg 2.7 VCOPY(thisray.rorg, org);
328     VCOPY(thisray.rdir, dir);
329     thisray.rmax = dmax;
330 greg 2.1 rayorigin(&thisray, PRIMARY, NULL, NULL);
331     samplendx++; /* call ray evaluation */
332     rayvalue(&thisray);
333     }
334    
335    
336     /* Accumulate and/or output ray contributions (child or only process) */
337     static void
338     done_contrib()
339     {
340     MODCONT *mp;
341     int i;
342    
343     if (account <= 0 || --account)
344     return; /* not time yet */
345    
346     for (i = 0; i < nmods; i++) { /* output records & clear */
347     mp = (MODCONT *)lu_find(&modconttab,modname[i])->data;
348     mod_output(mp);
349     memset(mp->cbin, 0, sizeof(DCOLOR)*mp->nbins);
350     }
351     end_record(); /* end lines & flush if time */
352    
353     account = accumulate; /* reset accumulation counter */
354     }
355    
356    
357     /* Principal calculation loop (called by main) */
358     void
359     rcontrib()
360     {
361     static int ignore_warning_given = 0;
362     FVECT orig, direc;
363     double d;
364 greg 2.2 /* initialize (& fork more of us) */
365 greg 2.1 rcinit();
366     /* load rays from stdin & process */
367     #ifdef getc_unlocked
368 greg 2.2 flockfile(stdin); /* avoid mutex overhead */
369 greg 2.1 #endif
370     while (getvec(orig) == 0 && getvec(direc) == 0) {
371     d = normalize(direc);
372 greg 2.17 if (nchild != -1 && (d == 0.0) & (accumulate == 0)) {
373 greg 2.1 if (!ignore_warning_given++)
374     error(WARNING,
375     "dummy ray(s) ignored during accumulation\n");
376     continue;
377     }
378     if (lastray+1 < lastray)
379     lastray = lastdone = 0;
380     ++lastray;
381     if (d == 0.0) { /* zero ==> flush */
382 greg 2.18 if ((yres <= 0) | (xres <= 1))
383 greg 2.17 waitflush = 1; /* flush after */
384     if (nchild == -1)
385     account = 1;
386 greg 2.7 } else if (imm_irrad) { /* else compute */
387     eval_irrad(orig, direc);
388     } else {
389     eval_rad(orig, direc, lim_dist ? d : 0.0);
390 greg 2.1 }
391     done_contrib(); /* accumulate/output */
392     ++lastdone;
393     if (raysleft && !--raysleft)
394     break; /* preemptive EOI */
395     }
396 greg 2.11 if (nchild != -1 && (accumulate <= 0) | (account < accumulate)) {
397 greg 2.1 if (account < accumulate) {
398     error(WARNING, "partial accumulation in final record");
399     accumulate -= account;
400     }
401     account = 1; /* output accumulated totals */
402     done_contrib();
403     }
404 greg 2.9 lu_done(&ofiletab); /* close output files */
405 greg 2.1 if (raysleft)
406     error(USER, "unexpected EOF on input");
407     }