ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/util/rsensor.c
Revision: 2.16
Committed: Thu Apr 23 23:54:08 2015 UTC (8 years, 11 months ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.15: +11 -5 lines
Log Message:
Avoid generating rays behind sensor

File Contents

# Content
1 #ifndef lint
2 static const char RCSid[] = "$Id: rsensor.c,v 2.15 2015/04/23 23:34:18 greg Exp $";
3 #endif
4
5 /*
6 * Compute sensor signal based on spatial sensitivity.
7 *
8 * Created Feb 2008 for Architectural Energy Corp.
9 */
10
11 #include "ray.h"
12 #include "source.h"
13 #include "view.h"
14 #include "random.h"
15
16 #define DEGREE (PI/180.)
17
18 #define MAXNT 181 /* maximum number of theta divisions */
19 #define MAXNP 360 /* maximum number of phi divisions */
20
21 extern char *progname; /* global argv[0] */
22 extern int nowarn; /* don't report warnings? */
23
24 /* current sensor's perspective */
25 VIEW ourview = {VT_ANG,{0.,0.,0.},{0.,0.,1.},{1.,0.,0.},
26 1.,180.,180.,0.,0.,0.,0.,
27 {0.,0.,0.},{0.,0.,0.},0.,0.};
28
29 unsigned long nsamps = 10000; /* desired number of initial samples */
30 unsigned long nssamps = 9000; /* number of super-samples */
31 int ndsamps = 32; /* number of direct samples */
32 int nprocs = 1; /* number of rendering processes */
33
34 float *sensor = NULL; /* current sensor data */
35 int sntp[2]; /* number of sensor theta and phi angles */
36 float maxtheta; /* maximum theta value for this sensor */
37 float tvals[MAXNT+1]; /* theta prob. values (1-D table of 1-cos(t)) */
38 float *pvals = NULL; /* phi prob. values (2-D table in radians) */
39 int ntheta = 0; /* polar angle divisions */
40 int nphi = 0; /* azimuthal angle divisions */
41 double gscale = 1.; /* global scaling value */
42
43 #define s_theta(t) sensor[(t+1)*(sntp[1]+1)]
44 #define s_phi(p) sensor[(p)+1]
45 #define s_val(t,p) sensor[(p)+1+(t+1)*(sntp[1]+1)]
46
47 static void comp_sensor(char *sfile);
48
49 static void
50 over_options() /* overriding options */
51 {
52 directvis = (ndsamps <= 0);
53 do_irrad = 0;
54 }
55
56 static void
57 print_defaults() /* print out default parameters */
58 {
59 over_options();
60 printf("-n %-9d\t\t\t# number of processes\n", nprocs);
61 printf("-rd %-9ld\t\t\t# ray directions\n", nsamps);
62 /* printf("-rs %-9ld\t\t\t# ray super-samples\n", nssamps); */
63 printf("-dn %-9d\t\t\t# direct number of samples\n", ndsamps);
64 printf("-vp %f %f %f\t# view point\n",
65 ourview.vp[0], ourview.vp[1], ourview.vp[2]);
66 printf("-vd %f %f %f\t# view direction\n",
67 ourview.vdir[0], ourview.vdir[1], ourview.vdir[2]);
68 printf("-vu %f %f %f\t# view up\n",
69 ourview.vup[0], ourview.vup[1], ourview.vup[2]);
70 printf("-vo %f\t\t\t# view fore clipping distance\n", ourview.vfore);
71 print_rdefaults();
72 }
73
74
75 void
76 quit(ec) /* make sure exit is called */
77 int ec;
78 {
79 if (ray_pnprocs > 0) /* close children if any */
80 ray_pclose(0);
81 exit(ec);
82 }
83
84
85 int
86 main(
87 int argc,
88 char *argv[]
89 )
90 {
91 int doheader = 1;
92 int optwarn = 0;
93 int i, rval;
94
95 progname = argv[0];
96 /* set up rendering defaults */
97 rand_samp = 1;
98 dstrsrc = 0.65;
99 srcsizerat = 0.1;
100 directrelay = 3;
101 ambounce = 1;
102 maxdepth = -10;
103 /* get options from command line */
104 for (i = 1; i < argc; i++) {
105 while ((rval = expandarg(&argc, &argv, i)) > 0)
106 ;
107 if (rval < 0) {
108 sprintf(errmsg, "cannot expand '%s'", argv[i]);
109 error(SYSTEM, errmsg);
110 }
111 if (argv[i][0] != '-') {
112 if (i >= argc-1)
113 break; /* final octree argument */
114 if (!ray_pnprocs) {
115 over_options();
116 if (doheader) { /* print header */
117 newheader("RADIANCE", stdout);
118 printargs(argc, argv, stdout);
119 fputformat("ascii", stdout);
120 putchar('\n');
121 }
122 /* start process(es) */
123 if (strcmp(argv[argc-1], "."))
124 ray_pinit(argv[argc-1], nprocs);
125 }
126 comp_sensor(argv[i]); /* process a sensor file */
127 continue;
128 }
129 if (argv[i][1] == 'r') { /* sampling options */
130 if (argv[i][2] == 'd')
131 nsamps = atol(argv[++i]);
132 else if (argv[i][2] == 's')
133 nssamps = atol(argv[++i]);
134 else {
135 sprintf(errmsg, "bad option at '%s'", argv[i]);
136 error(USER, errmsg);
137 }
138 continue;
139 }
140 /* direct component samples */
141 if (argv[i][1] == 'd' && argv[i][2] == 'n') {
142 ndsamps = atoi(argv[++i]);
143 continue;
144 }
145 if (argv[i][1] == 'v') { /* next sensor view */
146 if (argv[i][2] == 'f') {
147 rval = viewfile(argv[++i], &ourview, NULL);
148 if (rval < 0) {
149 sprintf(errmsg,
150 "cannot open view file \"%s\"",
151 argv[i]);
152 error(SYSTEM, errmsg);
153 } else if (rval == 0) {
154 sprintf(errmsg,
155 "bad view file \"%s\"",
156 argv[i]);
157 error(USER, errmsg);
158 }
159 continue;
160 }
161 rval = getviewopt(&ourview, argc-i, argv+i);
162 if (rval >= 0) {
163 i += rval;
164 continue;
165 }
166 sprintf(errmsg, "bad view option at '%s'", argv[i]);
167 error(USER, errmsg);
168 }
169 if (!strcmp(argv[i], "-w")) { /* toggle warnings */
170 nowarn = !nowarn;
171 continue;
172 }
173 if (ray_pnprocs) {
174 if (!optwarn++)
175 error(WARNING,
176 "rendering options should appear before first sensor");
177 } else if (!strcmp(argv[i], "-defaults")) {
178 print_defaults();
179 return(0);
180 }
181 if (argv[i][1] == 'h') { /* header toggle */
182 doheader = !doheader;
183 continue;
184 }
185 if (!strcmp(argv[i], "-n")) { /* number of processes */
186 nprocs = atoi(argv[++i]);
187 if (nprocs <= 0)
188 error(USER, "illegal number of processes");
189 continue;
190 }
191 rval = getrenderopt(argc-i, argv+i);
192 if (rval < 0) {
193 sprintf(errmsg, "bad render option at '%s'", argv[i]);
194 error(USER, errmsg);
195 }
196 i += rval;
197 }
198 if (sensor == NULL)
199 error(USER, i<argc ? "missing sensor file" : "missing octree");
200 quit(0);
201 }
202
203 /* Load sensor sensitivities (first row and column are angles) */
204 static float *
205 load_sensor(
206 int ntp[2],
207 char *sfile
208 )
209 {
210 int warnedneg;
211 char linebuf[8192];
212 int last_pos_val = 0;
213 int nelem = 1000;
214 float *sarr = (float *)malloc(sizeof(float)*nelem);
215 FILE *fp;
216 char *cp;
217 int i;
218
219 fp = frlibopen(sfile);
220 if (fp == NULL) {
221 sprintf(errmsg, "cannot open sensor file '%s'", sfile);
222 error(SYSTEM, errmsg);
223 }
224 fgets(linebuf, sizeof(linebuf), fp);
225 if (!strncmp(linebuf, "Elevation ", 10))
226 fgets(linebuf, sizeof(linebuf), fp);
227 /* get phi values */
228 sarr[0] = .0f;
229 if (strncmp(linebuf, "degrees", 7)) {
230 sprintf(errmsg, "Missing 'degrees' in sensor file '%s'", sfile);
231 error(USER, errmsg);
232 }
233 cp = sskip(linebuf);
234 ntp[1] = 0;
235 for ( ; ; ) {
236 sarr[ntp[1]+1] = atof(cp);
237 cp = fskip(cp);
238 if (cp == NULL)
239 break;
240 if (ntp[1] > 1 && sarr[ntp[1]+1] <= sarr[ntp[1]]) {
241 sprintf(errmsg,
242 "Phi values not monotinically increasing in sensor file '%s'",
243 sfile);
244 error(USER, errmsg);
245 }
246 ++ntp[1];
247 }
248 warnedneg = 0;
249 ntp[0] = 0; /* get thetas + data */
250 while (fgets(linebuf, sizeof(linebuf), fp) != NULL) {
251 ++ntp[0];
252 if ((ntp[0]+1)*(ntp[1]+1) > nelem) {
253 nelem += (nelem>>2) + ntp[1];
254 sarr = (float *)realloc((void *)sarr,
255 sizeof(float)*nelem);
256 if (sarr == NULL)
257 error(SYSTEM, "out of memory in load_sensor()");
258 }
259 cp = linebuf;
260 i = ntp[0]*(ntp[1]+1);
261 for ( ; ; ) {
262 sarr[i] = atof(cp);
263 cp = fskip(cp);
264 if (cp == NULL)
265 break;
266 if (i && sarr[i] < .0) {
267 if (!warnedneg++) {
268 sprintf(errmsg,
269 "Negative value(s) in sensor file '%s' (ignored)\n", sfile);
270 error(WARNING, errmsg);
271 }
272 sarr[i] = .0;
273 } else if (i > ntp[0]*(ntp[1]+1) && sarr[i] > FTINY)
274 last_pos_val = i;
275 ++i;
276 }
277 if (i == ntp[0]*(ntp[1]+1))
278 break;
279 if (ntp[0] > 1 && sarr[ntp[0]*(ntp[1]+1)] <=
280 sarr[(ntp[0]-1)*(ntp[1]+1)]) {
281 sprintf(errmsg,
282 "Theta values not monotinically increasing in sensor file '%s'",
283 sfile);
284 error(USER, errmsg);
285 }
286 if (i != (ntp[0]+1)*(ntp[1]+1)) {
287 sprintf(errmsg,
288 "bad column count near line %d in sensor file '%s'",
289 ntp[0]+1, sfile);
290 error(USER, errmsg);
291 }
292 }
293 /* truncate zero region */
294 ntp[0] = (last_pos_val + ntp[1])/(ntp[1]+1) - 1;
295 nelem = (ntp[0]+1)*(ntp[1]+1);
296 fclose(fp);
297 errmsg[0] = '\0'; /* sanity checks */
298 if (!last_pos_val)
299 sprintf(errmsg, "no positive sensor values in file '%s'", sfile);
300 if (ntp[0] <= 0)
301 sprintf(errmsg, "no data in sensor file '%s'", sfile);
302 else if (fabs(sarr[ntp[1]+1]) > FTINY)
303 sprintf(errmsg, "minimum theta must be 0 in sensor file '%s'",
304 sfile);
305 else if (fabs(sarr[1]) > FTINY)
306 sprintf(errmsg, "minimum phi must be 0 in sensor file '%s'",
307 sfile);
308 else if (sarr[ntp[1]] <= FTINY)
309 sprintf(errmsg,
310 "maximum phi must be positive in sensor file '%s'",
311 sfile);
312 else if (sarr[ntp[0]*(ntp[1]+1)] <= FTINY)
313 sprintf(errmsg,
314 "maximum theta (%f) must be positive in sensor file '%s'",
315 sarr[ntp[0]*(ntp[1]+1)], sfile);
316 if (errmsg[0])
317 error(USER, errmsg);
318 return((float *)realloc((void *)sarr, sizeof(float)*nelem));
319 }
320
321 /* Initialize probability table */
322 static void
323 init_ptable(
324 char *sfile
325 )
326 {
327 int samptot = nsamps;
328 float *rowp, *rowp1;
329 double rowsum[MAXNT], rowomega[MAXNT];
330 double thdiv[MAXNT+1], phdiv[MAXNP+1];
331 double tsize, psize;
332 double prob, frac, frac1;
333 int i, j, t, p;
334 /* free old table */
335 if (sensor != NULL)
336 free((void *)sensor);
337 if (pvals != NULL)
338 free((void *)pvals);
339 if (sfile == NULL || !*sfile) {
340 sensor = NULL;
341 sntp[0] = sntp[1] = 0;
342 pvals = NULL;
343 ntheta = nphi = 0;
344 return;
345 }
346 /* load sensor table */
347 sensor = load_sensor(sntp, sfile);
348 if (sntp[0] > MAXNT) {
349 sprintf(errmsg, "Too many theta rows in sensor file '%s'",
350 sfile);
351 error(INTERNAL, errmsg);
352 }
353 if (sntp[1] > MAXNP) {
354 sprintf(errmsg, "Too many phi columns in sensor file '%s'",
355 sfile);
356 error(INTERNAL, errmsg);
357 }
358 /* compute boundary angles */
359 maxtheta = DEGREE*(1.5f*s_theta(sntp[0]-1) - 0.5f*s_theta(sntp[0]-2));
360 if (maxtheta > PI)
361 maxtheta = PI;
362 thdiv[0] = .0;
363 for (t = 1; t < sntp[0]; t++)
364 thdiv[t] = DEGREE/2.*(s_theta(t-1) + s_theta(t));
365 thdiv[sntp[0]] = maxtheta;
366 phdiv[0] = DEGREE*(1.5f*s_phi(0) - 0.5f*s_phi(1));
367 for (p = 1; p < sntp[1]; p++)
368 phdiv[p] = DEGREE/2.*(s_phi(p-1) + s_phi(p));
369 phdiv[sntp[1]] = DEGREE*(1.5f*s_phi(sntp[1]-1) - 0.5f*s_phi(sntp[1]-2));
370 /* size our table */
371 tsize = 1. - cos(maxtheta);
372 psize = PI*tsize/maxtheta;
373 if (sntp[0]*sntp[1] < samptot) /* don't overdo resolution */
374 samptot = sntp[0]*sntp[1];
375 ntheta = (int)(sqrt((double)samptot*tsize/psize) + 0.5);
376 if (ntheta > MAXNT)
377 ntheta = MAXNT;
378 nphi = samptot/ntheta;
379 pvals = (float *)malloc(sizeof(float)*(ntheta+1)*(nphi+1));
380 if (pvals == NULL)
381 error(SYSTEM, "out of memory in init_ptable()");
382 gscale = .0; /* compute our inverse table */
383 for (i = 0; i < sntp[0]; i++) {
384 rowp = &s_val(i,0);
385 rowsum[i] = 1e-20;
386 for (j = 0; j < sntp[1]; j++)
387 rowsum[i] += *rowp++;
388 rowomega[i] = cos(thdiv[i]) - cos(thdiv[i+1]);
389 rowomega[i] *= 2.*PI / (double)sntp[1];
390 gscale += rowsum[i] * rowomega[i];
391 }
392 if (gscale <= FTINY) {
393 sprintf(errmsg, "Sensor values sum to zero in file '%s'", sfile);
394 error(USER, errmsg);
395 }
396 for (i = 0; i < ntheta; i++) {
397 prob = (double)i / (double)ntheta;
398 for (t = 0; t < sntp[0]; t++)
399 if ((prob -= rowsum[t]*rowomega[t]/gscale) <= .0)
400 break;
401 if (t >= sntp[0])
402 error(INTERNAL, "code error 1 in init_ptable()");
403 frac = 1. + prob/(rowsum[t]*rowomega[t]/gscale);
404 tvals[i] = 1. - ( (1.-frac)*cos(thdiv[t]) +
405 frac*cos(thdiv[t+1]) );
406 /* offset b/c sensor values are centered */
407 if ((t < sntp[0]-1) & (!t | (frac >= 0.5))) {
408 frac -= 0.5;
409 } else {
410 frac += 0.5;
411 --t;
412 }
413 pvals[i*(nphi+1)] = phdiv[0];
414 for (j = 1; j < nphi; j++) {
415 prob = (double)j / (double)nphi;
416 rowp = &s_val(t,0);
417 rowp1 = &s_val(t+1,0);
418 for (p = 0; p < sntp[1]; p++)
419 if ((prob -= (1.-frac)*rowp[p]/rowsum[t] +
420 frac*rowp1[p]/rowsum[t+1]) <= .0)
421 break;
422 if (p >= sntp[1]) { /* should never happen? */
423 p = sntp[1] - 1;
424 prob = .5;
425 }
426 frac1 = 1. + prob/((1.-frac)*rowp[p]/rowsum[t]
427 + frac*rowp1[p]/rowsum[t+1]);
428 pvals[i*(nphi+1) + j] = (1.-frac1)*phdiv[p] +
429 frac1*phdiv[p+1];
430 }
431 pvals[i*(nphi+1) + nphi] = phdiv[sntp[1]];
432 }
433 /* duplicate final row */
434 memcpy(pvals+ntheta*(nphi+1), pvals+(ntheta-1)*(nphi+1),
435 sizeof(*pvals)*(nphi+1));
436 tvals[0] = .0f;
437 tvals[ntheta] = (float)tsize;
438 }
439
440 /* Get normalized direction from random variables in [0,1) range */
441 static void
442 get_direc(
443 FVECT dvec,
444 double x,
445 double y
446 )
447 {
448 double xfrac = x*ntheta;
449 int tndx = (int)xfrac;
450 double yfrac = y*nphi;
451 int pndx = (int)yfrac;
452 double rad, phi;
453 FVECT dv;
454 int i;
455
456 xfrac -= (double)tndx;
457 yfrac -= (double)pndx;
458 pndx += tndx*(nphi+1);
459
460 dv[2] = 1. - ((1.-xfrac)*tvals[tndx] + xfrac*tvals[tndx+1]);
461 rad = sqrt(1. - dv[2]*dv[2]);
462 phi = (1.-yfrac)*pvals[pndx] + yfrac*pvals[pndx+1];
463 dv[0] = -rad*sin(phi);
464 dv[1] = rad*cos(phi);
465 for (i = 3; i--; )
466 dvec[i] = dv[0]*ourview.hvec[i] +
467 dv[1]*ourview.vvec[i] +
468 dv[2]*ourview.vdir[i] ;
469 }
470
471 /* Get sensor value in the specified direction (normalized) */
472 static float
473 sens_val(
474 FVECT dvec
475 )
476 {
477 FVECT dv;
478 float theta, phi;
479 int t, p;
480
481 dv[2] = DOT(dvec, ourview.vdir);
482 theta = acos(dv[2]);
483 if (theta >= maxtheta)
484 return(.0f);
485 dv[0] = DOT(dvec, ourview.hvec);
486 dv[1] = DOT(dvec, ourview.vvec);
487 phi = atan2(-dv[0], dv[1]);
488 while (phi < .0f) phi += (float)(2.*PI);
489 t = (int)(theta/maxtheta * sntp[0]);
490 p = (int)(phi*(1./(2.*PI)) * sntp[1]);
491 /* hack for non-uniform sensor grid */
492 theta *= (float)(1./DEGREE);
493 phi *= (float)(1./DEGREE);
494 while (t+1 < sntp[0] && theta >= s_theta(t+1))
495 ++t;
496 while (t-1 >= 0 && theta <= s_theta(t-1))
497 --t;
498 while (p+1 < sntp[1] && phi >= s_phi(p+1))
499 ++p;
500 while (p-1 >= 0 && phi <= s_phi(p-1))
501 --p;
502 return(s_val(t,p));
503 }
504
505 /* Print origin and direction */
506 static void
507 print_ray(
508 FVECT rorg,
509 FVECT rdir
510 )
511 {
512 printf("%.6g %.6g %.6g %.8f %.8f %.8f\n",
513 rorg[0], rorg[1], rorg[2],
514 rdir[0], rdir[1], rdir[2]);
515 }
516
517 /* Compute sensor output */
518 static void
519 comp_sensor(
520 char *sfile
521 )
522 {
523 int ndirs = dstrsrc > FTINY ? ndsamps :
524 ndsamps > 0 ? 1 : 0;
525 char *err;
526 int nt, np;
527 COLOR vsum;
528 RAY rr;
529 double sf;
530 int i, j;
531 /* set view */
532 ourview.type = VT_ANG;
533 ourview.horiz = ourview.vert = 180.;
534 ourview.hoff = ourview.voff = .0;
535 err = setview(&ourview);
536 if (err != NULL)
537 error(USER, err);
538 /* assign probability table */
539 init_ptable(sfile);
540 /* stratified MC sampling */
541 setcolor(vsum, .0f, .0f, .0f);
542 nt = (int)(sqrt((double)nsamps*ntheta/nphi) + .5);
543 np = nsamps/nt;
544 sf = gscale/nsamps;
545 for (i = 0; i < nt; i++)
546 for (j = 0; j < np; j++) {
547 VCOPY(rr.rorg, ourview.vp);
548 get_direc(rr.rdir, (i+frandom())/nt, (j+frandom())/np);
549 if (ourview.vfore > FTINY)
550 VSUM(rr.rorg, rr.rorg, rr.rdir, ourview.vfore);
551 if (!ray_pnprocs) {
552 print_ray(rr.rorg, rr.rdir);
553 continue;
554 }
555 rr.rmax = .0;
556 rayorigin(&rr, PRIMARY|SPECULAR, NULL, NULL);
557 scalecolor(rr.rcoef, sf);
558 if (ray_pqueue(&rr) == 1)
559 addcolor(vsum, rr.rcol);
560 }
561 /* remaining rays pure MC */
562 for (i = nsamps - nt*np; i-- > 0; ) {
563 VCOPY(rr.rorg, ourview.vp);
564 get_direc(rr.rdir, frandom(), frandom());
565 if (ourview.vfore > FTINY)
566 VSUM(rr.rorg, rr.rorg, rr.rdir, ourview.vfore);
567 if (!ray_pnprocs) {
568 print_ray(rr.rorg, rr.rdir);
569 continue;
570 }
571 rr.rmax = .0;
572 rayorigin(&rr, PRIMARY|SPECULAR, NULL, NULL);
573 scalecolor(rr.rcoef, sf);
574 if (ray_pqueue(&rr) == 1)
575 addcolor(vsum, rr.rcol);
576 }
577 if (!ray_pnprocs) /* just printing rays */
578 return;
579 /* scale partial result */
580 scalecolor(vsum, sf);
581 /* add direct component */
582 for (i = ndirs; i-- > 0; ) {
583 SRCINDEX si;
584 initsrcindex(&si);
585 while (srcray(&rr, NULL, &si)) {
586 sf = sens_val(rr.rdir);
587 if (sf <= FTINY)
588 continue;
589 sf *= si.dom/ndirs;
590 scalecolor(rr.rcoef, sf);
591 if (ray_pqueue(&rr) == 1) {
592 multcolor(rr.rcol, rr.rcoef);
593 addcolor(vsum, rr.rcol);
594 }
595 }
596 }
597 /* finish our calculation */
598 while (ray_presult(&rr, 0) > 0) {
599 multcolor(rr.rcol, rr.rcoef);
600 addcolor(vsum, rr.rcol);
601 }
602 /* print our result */
603 printf("%.4e %.4e %.4e\n", colval(vsum,RED),
604 colval(vsum,GRN), colval(vsum,BLU));
605 }