ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/rt/rcontrib.c
(Generate patch)

Comparing ray/src/rt/rcontrib.c (file contents):
Revision 2.1 by greg, Sat Jun 9 07:16:47 2012 UTC vs.
Revision 2.40 by greg, Tue Feb 16 20:06:06 2021 UTC

# Line 6 | Line 6 | static const char RCSid[] = "$Id$";
6   * Initialization and calculation routines
7   */
8  
9 + #include "copyright.h"
10 +
11   #include "rcontrib.h"
10 #include "source.h"
12   #include "otypes.h"
13 < #include "platform.h"
13 > #include "source.h"
14  
15 + char    *shm_boundary = NULL;           /* boundary of shared memory */
16 +
17   CUBE    thescene;                       /* our scene */
18   OBJECT  nsceneobjs;                     /* number of objects in our scene */
19  
# Line 18 | Line 21 | 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 */
22 < void    (*trace)() = trace_contrib;
24 > void    (*trace)() = NULL;              /* trace call (NULL before rcinit) */
25  
26   int     do_irrad = 0;                   /* compute irradiance? */
27  
28 < int     rand_samp = 0;                  /* pure Monte Carlo sampling? */
28 > int     rand_samp = 1;                  /* pure Monte Carlo sampling? */
29  
30 < double  dstrsrc = 0.0;                  /* square source distribution */
31 < double  shadthresh = .03;               /* shadow threshold */
30 > double  dstrsrc = 0.9;                  /* square source distribution */
31 > double  shadthresh = 0.;                /* shadow threshold */
32   double  shadcert = .75;                 /* shadow certainty */
33   int     directrelay = 3;                /* number of source relays */
34   int     vspretest = 512;                /* virtual source pretest density */
# Line 38 | Line 40 | COLOR  salbedo = BLKCOLOR;             /* global scattering albedo
40   double  seccg = 0.;                     /* global scattering eccentricity */
41   double  ssampdist = 0.;                 /* scatter sampling distance */
42  
43 < double  specthresh = .15;               /* specular sampling threshold */
43 > double  specthresh = .02;               /* specular sampling threshold */
44   double  specjitter = 1.;                /* specular sampling jitter */
45  
46   int     backvis = 1;                    /* back face visibility */
47  
48 < int     maxdepth = 8;                   /* maximum recursion depth */
49 < double  minweight = 5e-4;               /* minimum ray weight */
48 > int     maxdepth = -10;                 /* maximum recursion depth */
49 > double  minweight = 2e-3;               /* minimum ray weight */
50  
51   char    *ambfile = NULL;                /* ambient file name */
52   COLOR   ambval = BLKCOLOR;              /* ambient value */
53   int     ambvwt = 0;                     /* initial weight for ambient value */
54 < double  ambacc = 0;                     /* ambient accuracy */
54 > double  ambacc = 0.;                    /* ambient accuracy */
55   int     ambres = 256;                   /* ambient resolution */
56   int     ambdiv = 350;                   /* ambient divisions */
57   int     ambssamp = 0;                   /* ambient super-samples */
# Line 61 | Line 63 | int    account;                        /* current accumulation count */
63   RNUMBER raysleft;                       /* number of rays left to trace */
64   long    waitflush;                      /* how long until next flush */
65  
66 < int     lastray = 0;                    /* last ray number sent */
67 < int     lastdone = 0;                   /* last ray output */
66 > RNUMBER lastray = 0;                    /* last ray number sent */
67 > RNUMBER lastdone = 0;                   /* last ray output */
68  
69 < /* Close output stream and free record */
68 < static void
69 < closestream(void *p)
70 < {
71 <        STREAMOUT       *sop = (STREAMOUT *)p;
72 <        int             status = 0;
73 <        if (sop->outpipe)
74 <                status = pclose(sop->ofp);
75 <        else if (sop->ofp != stdout)
76 <                status = fclose(sop->ofp);
77 <        if (status)
78 <                error(SYSTEM, "error closing output stream");
79 <        free(p);
80 < }
69 > static void     trace_contrib(RAY *r);  /* our trace callback */
70  
82 LUTAB   ofiletab = LU_SINIT(free,closestream);  /* output file table */
83
84 #define OF_MODIFIER     01
85 #define OF_BIN          02
86
71   static void mcfree(void *p) { epfree((*(MODCONT *)p).binv); free(p); }
72  
73   LUTAB   modconttab = LU_SINIT(NULL,mcfree);     /* modifier lookup table */
74  
91 static OBJECT           traset[MAXTSET+1]={0};  /* trace include set */
92
75   /************************** INITIALIZATION ROUTINES ***********************/
76  
95 void
96 tranotify(                      /* record new modifier */
97        OBJECT  obj
98 )
99 {
100        static int      hitlimit = 0;
101        OBJREC          *o = objptr(obj);
102        int             i;
103
104        if (obj == OVOID) {             /* starting over */
105                traset[0] = 0;
106                hitlimit = 0;
107                return;
108        }
109        if (hitlimit || !ismodifier(o->otype))
110                return;
111        for (i = nmods; i-- > 0; )
112                if (!strcmp(o->oname, modname[i])) {
113                        if (traset[0] >= MAXTSET) {
114                                error(WARNING, "too many same-named modifiers");
115                                hitlimit++;
116                                return;         /* should this be fatal? */
117                        }
118                        insertelem(traset, obj);
119                        break;
120                }
121 }
122
123
77   char *
78   formstr(                                /* return format identifier */
79          int  f
# Line 138 | Line 91 | formstr(                               /* return format identifier */
91  
92   /* Add modifier to our list to track */
93   MODCONT *
94 < addmodifier(char *modn, char *outf, char *binv, int bincnt)
94 > addmodifier(char *modn, char *outf, char *prms, char *binv, int bincnt)
95   {
96          LUENT   *lep = lu_find(&modconttab,modn);
97          MODCONT *mp;
# Line 149 | Line 102 | addmodifier(char *modn, char *outf, char *binv, int bi
102                  sprintf(errmsg, "duplicate modifier '%s'", modn);
103                  error(USER, errmsg);
104          }
105 <        if (nmods >= MAXMODLIST)
106 <                error(INTERNAL, "too many modifiers");
105 >        if (!strcmp(modn, VOIDID)) {
106 >                sprintf(errmsg, "cannot track '%s' modifier", VOIDID);
107 >                error(USER, errmsg);
108 >        }
109 >        if (nmods >= modasiz) {         /* need bigger modifier array */
110 >                modasiz += modasiz/3 + 64;
111 >                if (modname == NULL)
112 >                        modname = (char **)malloc(modasiz*sizeof(char *));
113 >                else
114 >                        modname = (char **)realloc(modname, modasiz*sizeof(char *));
115 >                if (modname == NULL)
116 >                        error(SYSTEM, "Out of memory in addmodifier()");
117 >        }
118          modname[nmods++] = modn;        /* XXX assumes static string */
119          lep->key = modn;                /* XXX assumes static string */
120          if (binv == NULL)
# Line 175 | Line 139 | addmodifier(char *modn, char *outf, char *binv, int bi
139                  error(SYSTEM, "out of memory in addmodifier");
140          mp->outspec = outf;             /* XXX assumes static string */
141          mp->modname = modn;             /* XXX assumes static string */
142 +        mp->params = prms;              /* XXX assumes static string */
143          mp->binv = ebinv;
144 +        mp->bin0 = 0;
145          mp->nbins = bincnt;
146          memset(mp->cbin, 0, sizeof(DCOLOR)*bincnt);
147 <                                        /* allocate output streams */
148 <        for (i = bincnt; i-- > 0; )
149 <                getostream(mp->outspec, mp->modname, i, 1);
147 >                                        /* figure out starting bin */
148 >        while (!getostream(mp->outspec, mp->modname, mp->bin0, 1))
149 >                mp->bin0++;
150 >                                        /* allocate other output streams */
151 >        for (i = 0; ++i < mp->nbins; )
152 >                getostream(mp->outspec, mp->modname, mp->bin0+i, 1);
153          lep->data = (char *)mp;
154          return(mp);
155   }
156  
157  
158 < /* add modifiers from a file list */
158 > /* Add modifiers from a file list */
159   void
160 < addmodfile(char *fname, char *outf, char *binv, int bincnt)
160 > addmodfile(char *fname, char *outf, char *prms, char *binv, int bincnt)
161   {
162 <        char    *mname[MAXMODLIST];
163 <        int     i;
164 <                                        /* find the file & store strings */
165 <        if (wordfile(mname, getpath(fname, getrlibpath(), R_OK)) < 0) {
166 <                sprintf(errmsg, "cannot find modifier file '%s'", fname);
162 >        char    *path = getpath(fname, getrlibpath(), R_OK);
163 >        char    mod[MAXSTR];
164 >        FILE    *fp;
165 >
166 >        if (path == NULL || (fp = fopen(path, "r")) == NULL) {
167 >                sprintf(errmsg, "cannot load modifier file '%s'",
168 >                                path != NULL ? path : fname);
169                  error(SYSTEM, errmsg);
170          }
171 <        for (i = 0; mname[i]; i++)      /* add each one */
172 <                addmodifier(mname[i], outf, binv, bincnt);
171 >        while (fgetword(mod, sizeof(mod), fp) != NULL)
172 >                addmodifier(mod, outf, prms, binv, bincnt);
173 >        fclose(fp);
174   }
175  
176  
177 + /* Check if we have any more rays left (and report progress) */
178 + int
179 + morays(void)
180 + {
181 +        static RNUMBER  total_rays;
182 +        static time_t   tstart, last_report;
183 +        time_t          tnow;
184 +
185 +        if (!raysleft)
186 +                return(1);      /* unknown total, so nothing to do or say */
187 +
188 +        if (report_intvl > 0 && (tnow = time(0)) >= last_report+report_intvl) {
189 +                if (!total_rays) {
190 +                        total_rays = raysleft;
191 +                        tstart = tnow;
192 +                } else {
193 +                        sprintf(errmsg, "%.2f%% done after %.3f hours\n",
194 +                                        100.-100.*raysleft/total_rays,
195 +                                        (1./3600.)*(tnow - tstart));
196 +                        eputs(errmsg);
197 +                }
198 +                last_report = tnow;
199 +        }
200 +        return(--raysleft);
201 + }
202 +
203 +
204 + /* Quit program */
205   void
206 < quit(                   /* quit program */
206 > quit(
207          int  code
208   )
209   {
210          if (nchild > 0)         /* close children if any */
211 <                end_children();
211 >                end_children(code != 0);
212 >        else if (nchild < 0)
213 >                _exit(code);    /* avoid flush() in child */
214          exit(code);
215   }
216  
217  
218   /* Initialize our process(es) */
219   static void
220 < rcinit()
220 > rcinit(void)
221   {
222          int     i;
223  
224          if (nproc > MAXPROCESS)
225                  sprintf(errmsg, "too many processes requested -- reducing to %d",
226                                  nproc = MAXPROCESS);
227 <                
228 <        if ((nproc > 1) & (accumulate <= 0))
229 <                zero_record(0);         /* prime our queue to accumulate */
230 <
229 <        if (recover) {                  /* recover previous output? */
230 <                if (accumulate <= 0) {
231 <                        reload_output();
232 <                        if ((nproc > 1) & (accumulate <= 0))
233 <                                queue_modifiers();
234 <                } else
235 <                        recover_output();
227 >        if (nproc > 1) {
228 >                preload_objs();         /* preload auxiliary data */
229 >                                        /* set shared memory boundary */
230 >                shm_boundary = strcpy((char *)malloc(16), "SHM_BOUNDARY");
231          }
232 +        trace = trace_contrib;          /* set up trace call-back */
233 +        for (i = 0; i < nsources; i++)  /* tracing to sources as well */
234 +                source[i].sflags |= SFOLLOW;
235          if (yres > 0) {                 /* set up flushing & ray counts */
236                  if (xres > 0)
237                          raysleft = (RNUMBER)xres*yres;
# Line 244 | Line 242 | rcinit()
242          if ((account = accumulate) > 1)
243                  raysleft *= accumulate;
244          waitflush = (yres > 0) & (xres > 1) ? 0 : xres;
247                                        /* tracing to sources as well */
248        for (i = 0; i < nsources; i++)
249                source[i].sflags |= SFOLLOW;
245  
246 <        if (nproc == 1 || in_rchild())  /* single process or child */
246 >        if (nproc > 1 && in_rchild())   /* forked child? */
247                  return;                 /* return to main processing loop */
248  
249 <        parental_loop();                /* else run controller */
249 >        if (recover) {                  /* recover previous output? */
250 >                if (accumulate <= 0)
251 >                        reload_output();
252 >                else
253 >                        recover_output();
254 >        }
255 >        if (nproc == 1)                 /* single process? */
256 >                return;
257 >                                        /* else run appropriate controller */
258 >        if (accumulate <= 0)
259 >                feeder_loop();
260 >        else
261 >                parental_loop();
262          quit(0);                        /* parent musn't return! */
263   }
264  
# Line 262 | Line 269 | static void
269   trace_contrib(RAY *r)
270   {
271          MODCONT *mp;
272 +        double  bval;
273          int     bn;
274          RREAL   contr[3];
275  
276 <        if (r->ro == NULL || !inset(traset, r->ro->omod))
276 >        if (r->ro == NULL || r->ro->omod == OVOID)
277                  return;
278 +                                                /* shadow ray not on source? */
279 +        if (r->rsrc >= 0 && source[r->rsrc].so != r->ro)
280 +                return;
281  
282          mp = (MODCONT *)lu_find(&modconttab,objptr(r->ro->omod)->oname)->data;
283  
284 <        if (mp == NULL)
285 <                error(CONSISTENCY, "unexpected modifier in trace_contrib()");
284 >        if (mp == NULL)                         /* not in our list? */
285 >                return;
286  
287 <        worldfunc(RCCONTEXT, r);                /* get bin number */
288 <        bn = (int)(evalue(mp->binv) + .5);
289 <        if ((bn < 0) | (bn >= mp->nbins)) {
290 <                error(WARNING, "bad bin number (ignored)");
287 >        worldfunc(RCCONTEXT, r);                /* else set context */
288 >        set_eparams((char *)mp->params);
289 >        if ((bval = evalue(mp->binv)) <= -.5)   /* and get bin number */
290 >                return;                         /* silently ignore negatives */
291 >        if ((bn = (int)(bval + .5)) >= mp->nbins) {
292 >                sprintf(errmsg, "bad bin number (%d ignored)", bn);
293 >                error(WARNING, errmsg);
294                  return;
295          }
296 <        raycontrib(contr, r, PRIMARY);
296 >        raycontrib(contr, r, PRIMARY);          /* compute coefficient */
297          if (contrib)
298 <                multcolor(contr, r->rcol);
298 >                multcolor(contr, r->rcol);      /* -> contribution */
299          addcolor(mp->cbin[bn], contr);
300   }
301  
302  
303 + /* Evaluate irradiance contributions */
304   static void
305 < rayirrad(                       /* compute irradiance rather than radiance */
291 <        RAY *r
292 < )
305 > eval_irrad(FVECT org, FVECT dir)
306   {
307 <        r->rot = 1e-5;                  /* pretend we hit surface */
308 <        VSUM(r->rop, r->rorg, r->rdir, r->rot);
309 <        r->ron[0] = -r->rdir[0];
310 <        r->ron[1] = -r->rdir[1];
311 <        r->ron[2] = -r->rdir[2];
312 <        r->rod = 1.0;
313 <                                        /* compute result */
314 <        r->revf = raytrace;
315 <        (*ofun[Lamb.otype].funp)(&Lamb, r);
316 <        r->revf = rayirrad;
307 >        RAY     thisray;
308 >
309 >        VSUM(thisray.rorg, org, dir, 1.1e-4);
310 >        thisray.rdir[0] = -dir[0];
311 >        thisray.rdir[1] = -dir[1];
312 >        thisray.rdir[2] = -dir[2];
313 >        thisray.rmax = 0.0;
314 >        rayorigin(&thisray, PRIMARY, NULL, NULL);
315 >                                        /* pretend we hit surface */
316 >        thisray.rxt = thisray.rot = 1e-5;
317 >        thisray.rod = 1.0;
318 >        VCOPY(thisray.ron, dir);
319 >        VSUM(thisray.rop, org, dir, 1e-4);
320 >        samplendx++;                    /* compute result */
321 >        (*ofun[Lamb.otype].funp)(&Lamb, &thisray);
322   }
323  
324  
325 < /* Evaluate ray contributions */
325 > /* Evaluate radiance contributions */
326   static void
327 < eval_ray(FVECT  org, FVECT  dir, double dmax)
327 > eval_rad(FVECT org, FVECT dir, double dmax)
328   {
329          RAY     thisray;
330                                          /* set up ray */
331 +        VCOPY(thisray.rorg, org);
332 +        VCOPY(thisray.rdir, dir);
333 +        thisray.rmax = dmax;
334          rayorigin(&thisray, PRIMARY, NULL, NULL);
314        if (imm_irrad) {
315                VSUM(thisray.rorg, org, dir, 1.1e-4);
316                thisray.rdir[0] = -dir[0];
317                thisray.rdir[1] = -dir[1];
318                thisray.rdir[2] = -dir[2];
319                thisray.rmax = 0.0;
320                thisray.revf = rayirrad;
321        } else {
322                VCOPY(thisray.rorg, org);
323                VCOPY(thisray.rdir, dir);
324                thisray.rmax = dmax;
325        }
335          samplendx++;                    /* call ray evaluation */
336          rayvalue(&thisray);
337   }
# Line 330 | Line 339 | eval_ray(FVECT  org, FVECT  dir, double        dmax)
339  
340   /* Accumulate and/or output ray contributions (child or only process) */
341   static void
342 < done_contrib()
342 > done_contrib(void)
343   {
344          MODCONT *mp;
345          int     i;
# Line 351 | Line 360 | done_contrib()
360  
361   /* Principal calculation loop (called by main) */
362   void
363 < rcontrib()
363 > rcontrib(void)
364   {
365          static int      ignore_warning_given = 0;
366          FVECT           orig, direc;
367          double          d;
368 <                                        /* initialize & fork */
368 >                                        /* initialize (& fork more of us) */
369          rcinit();
370                                          /* load rays from stdin & process */
371   #ifdef getc_unlocked
372 <        flockfile(stdin);               /* avoid lock/unlock overhead */
372 >        flockfile(stdin);               /* avoid mutex overhead */
373   #endif
374          while (getvec(orig) == 0 && getvec(direc) == 0) {
375                  d = normalize(direc);
376 <                if ((d == 0.0) & (accumulate != 1)) {
376 >                if (nchild != -1 && (d == 0.0) & (accumulate == 0)) {
377                          if (!ignore_warning_given++)
378                                  error(WARNING,
379                                  "dummy ray(s) ignored during accumulation\n");
# Line 374 | Line 383 | rcontrib()
383                          lastray = lastdone = 0;
384                  ++lastray;
385                  if (d == 0.0) {                         /* zero ==> flush */
386 <                        if ((yres <= 0) | (xres <= 0))
387 <                                waitflush = 1;          /* flush right after */
388 <                } else {                                /* else compute */
389 <                        eval_ray(orig, direc, lim_dist ? d : 0.0);
386 >                        if ((yres <= 0) | (xres <= 1))
387 >                                waitflush = 1;          /* flush after */
388 >                        if (nchild == -1)
389 >                                account = 1;
390 >                } else if (imm_irrad) {                 /* else compute */
391 >                        eval_irrad(orig, direc);
392 >                } else {
393 >                        eval_rad(orig, direc, lim_dist ? d : 0.0);
394                  }
395                  done_contrib();         /* accumulate/output */
396                  ++lastdone;
397 <                if (raysleft && !--raysleft)
397 >                if (!morays())
398                          break;          /* preemptive EOI */
399          }
400 <        if (accumulate <= 0 || account < accumulate) {
400 >        if (nchild != -1 && (accumulate <= 0) | (account < accumulate)) {
401                  if (account < accumulate) {
402                          error(WARNING, "partial accumulation in final record");
403                          accumulate -= account;
# Line 392 | Line 405 | rcontrib()
405                  account = 1;            /* output accumulated totals */
406                  done_contrib();
407          }
408 +        lu_done(&ofiletab);             /* close output files */
409          if (raysleft)
410                  error(USER, "unexpected EOF on input");
397        lu_done(&ofiletab);             /* close output files */
411   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines