ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/rt/rcontrib.c
Revision: 2.27
Committed: Mon Mar 21 19:06:08 2016 UTC (8 years, 1 month ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.26: +12 -4 lines
Log Message:
Improved error-checking for too many words in file or string

File Contents

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