ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/cv/bsdf2ttree.c
Revision: 2.23
Committed: Tue Nov 26 17:35:59 2013 UTC (10 years, 4 months ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.22: +4 -3 lines
Log Message:
Include random.h for Windows instead of calling drand48

File Contents

# User Rev Content
1 greg 2.1 #ifndef lint
2 greg 2.23 static const char RCSid[] = "$Id: bsdf2ttree.c,v 2.22 2013/11/26 17:33:55 greg Exp $";
3 greg 2.1 #endif
4     /*
5     * Load measured BSDF interpolant and write out as XML file with tensor tree.
6     *
7     * G. Ward
8     */
9    
10     #define _USE_MATH_DEFINES
11     #include <stdio.h>
12     #include <stdlib.h>
13     #include <math.h>
14 greg 2.23 #include "random.h"
15 greg 2.1 #include "platform.h"
16 greg 2.22 #include "rtprocess.h"
17 greg 2.11 #include "calcomp.h"
18 greg 2.1 #include "bsdfrep.h"
19     /* global argv[0] */
20     char *progname;
21     /* percentage to cull (<0 to turn off) */
22 greg 2.7 double pctcull = 90.;
23 greg 2.1 /* sampling order */
24     int samp_order = 6;
25 greg 2.15 /* super-sampling threshold */
26     const double ssamp_thresh = 0.35;
27     /* number of super-samples */
28     const int nssamp = 100;
29 greg 2.18 /* limit on number of RBF lobes */
30     static int lobe_lim = 15000;
31 greg 2.1
32 greg 2.11 /* Output XML prologue to stdout */
33     static void
34     xml_prologue(int ac, char *av[])
35     {
36     puts("<?xml version=\"1.0\" encoding=\"UTF-8\"?>");
37     puts("<WindowElement xmlns=\"http://windows.lbl.gov\" xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\" xsi:schemaLocation=\"http://windows.lbl.gov/BSDF-v1.4.xsd\">");
38     fputs("<!-- File produced by:", stdout);
39     while (ac-- > 0) {
40     fputc(' ', stdout);
41     fputs(*av++, stdout);
42     }
43     puts(" -->");
44     puts("<WindowElementType>System</WindowElementType>");
45     puts("<FileType>BSDF</FileType>");
46     puts("<Optical>");
47     puts("<Layer>");
48     puts("\t<Material>");
49 greg 2.21 printf("\t\t<Name>%s</Name>\n", bsdf_name[0] ? bsdf_name : "Unknown");
50     printf("\t\t<Manufacturer>%s</Manufacturer>\n",
51     bsdf_manuf[0] ? bsdf_manuf : "Unknown");
52 greg 2.11 puts("\t\t<DeviceType>Other</DeviceType>");
53     puts("\t</Material>");
54     puts("\t<DataDefinition>");
55     printf("\t\t<IncidentDataStructure>TensorTree%c</IncidentDataStructure>\n",
56     single_plane_incident ? '3' : '4');
57     puts("\t</DataDefinition>");
58     }
59    
60     /* Output XML data prologue to stdout */
61     static void
62     data_prologue()
63     {
64     static const char *bsdf_type[4] = {
65     "Reflection Front",
66     "Transmission Front",
67     "Transmission Back",
68     "Reflection Back"
69     };
70    
71     puts("\t<WavelengthData>");
72     puts("\t\t<LayerNumber>System</LayerNumber>");
73     puts("\t\t<Wavelength unit=\"Integral\">Visible</Wavelength>");
74     puts("\t\t<SourceSpectrum>CIE Illuminant D65 1nm.ssp</SourceSpectrum>");
75     puts("\t\t<DetectorSpectrum>ASTM E308 1931 Y.dsp</DetectorSpectrum>");
76     puts("\t\t<WavelengthDataBlock>");
77     printf("\t\t\t<WavelengthDataDirection>%s</WavelengthDataDirection>\n",
78     bsdf_type[(input_orient>0)<<1 | (output_orient>0)]);
79     puts("\t\t\t<AngleBasis>LBNL/Shirley-Chiu</AngleBasis>");
80     puts("\t\t\t<ScatteringDataType>BTDF</ScatteringDataType>");
81     puts("\t\t\t<ScatteringData>");
82     }
83    
84     /* Output XML data epilogue to stdout */
85     static void
86     data_epilogue(void)
87     {
88     puts("\t\t\t</ScatteringData>");
89     puts("\t\t</WavelengthDataBlock>");
90     puts("\t</WavelengthData>");
91     }
92    
93     /* Output XML epilogue to stdout */
94     static void
95     xml_epilogue(void)
96     {
97     puts("</Layer>");
98     puts("</Optical>");
99     puts("</WindowElement>");
100     }
101    
102 greg 2.15 /* Compute absolute relative difference */
103     static double
104     abs_diff(double v1, double v0)
105     {
106     if ((v0 < 0) | (v1 < 0))
107     return(.0);
108     v1 = (v1-v0)*2./(v0+v1+.0001);
109     if (v1 < 0)
110     return(-v1);
111     return(v1);
112     }
113    
114 greg 2.1 /* Interpolate and output isotropic BSDF data */
115     static void
116 greg 2.11 eval_isotropic(char *funame)
117 greg 2.1 {
118     const int sqres = 1<<samp_order;
119     FILE *ofp = NULL;
120 greg 2.17 int assignD = 0;
121 greg 2.1 char cmd[128];
122     int ix, ox, oy;
123 greg 2.11 double iovec[6];
124 greg 2.4 float bsdf;
125 greg 2.15
126 greg 2.11 data_prologue(); /* begin output */
127     if (pctcull >= 0) {
128 greg 2.20 sprintf(cmd, "rttree_reduce -a -h -ff -r 3 -t %f -g %d",
129 greg 2.1 pctcull, samp_order);
130     fflush(stdout);
131     ofp = popen(cmd, "w");
132     if (ofp == NULL) {
133     fprintf(stderr, "%s: cannot create pipe to rttree_reduce\n",
134     progname);
135     exit(1);
136     }
137     SET_FILE_BINARY(ofp);
138     } else
139     fputs("{\n", stdout);
140 greg 2.17 /* need to assign Dx, Dy, Dz? */
141     if (funame != NULL)
142     assignD = (fundefined(funame) < 6);
143 greg 2.1 /* run through directions */
144     for (ix = 0; ix < sqres/2; ix++) {
145 greg 2.11 RBFNODE *rbf = NULL;
146 greg 2.14 iovec[0] = 2.*(ix+.5)/sqres - 1.;
147 greg 2.11 iovec[1] = .0;
148     iovec[2] = input_orient * sqrt(1. - iovec[0]*iovec[0]);
149     if (funame == NULL)
150 greg 2.18 rbf = advect_rbf(iovec, lobe_lim);
151 greg 2.15 for (ox = 0; ox < sqres; ox++) {
152     float last_bsdf = -1;
153 greg 2.1 for (oy = 0; oy < sqres; oy++) {
154 greg 2.11 SDsquare2disk(iovec+3, (ox+.5)/sqres, (oy+.5)/sqres);
155     iovec[5] = output_orient *
156     sqrt(1. - iovec[3]*iovec[3] - iovec[4]*iovec[4]);
157     if (funame == NULL)
158 greg 2.15 bsdf = eval_rbfrep(rbf, iovec+3) *
159 greg 2.11 output_orient/iovec[5];
160 greg 2.15 else {
161 greg 2.16 double ssa[3], ssvec[6], sum;
162 greg 2.15 int ssi;
163 greg 2.17 if (assignD) {
164     varset("Dx", '=', -iovec[3]);
165     varset("Dy", '=', -iovec[4]);
166     varset("Dz", '=', -iovec[5]);
167     ++eclock;
168     }
169 greg 2.15 bsdf = funvalue(funame, 6, iovec);
170     if (abs_diff(bsdf, last_bsdf) > ssamp_thresh) {
171 greg 2.16 sum = 0; /* super-sample voxel */
172 greg 2.15 for (ssi = nssamp; ssi--; ) {
173 greg 2.23 SDmultiSamp(ssa, 3, (ssi+frandom())/nssamp);
174 greg 2.15 ssvec[0] = 2.*(ix+ssa[0])/sqres - 1.;
175     ssvec[1] = .0;
176     ssvec[2] = input_orient *
177     sqrt(1. - ssvec[0]*ssvec[0]);
178     SDsquare2disk(ssvec+3, (ox+ssa[1])/sqres,
179     (oy+ssa[2])/sqres);
180     ssvec[5] = output_orient *
181     sqrt(1. - ssvec[3]*ssvec[3] -
182     ssvec[4]*ssvec[4]);
183 greg 2.17 if (assignD) {
184     varset("Dx", '=', -iovec[3]);
185     varset("Dy", '=', -iovec[4]);
186     varset("Dz", '=', -iovec[5]);
187     ++eclock;
188     }
189 greg 2.16 sum += funvalue(funame, 6, ssvec);
190 greg 2.15 }
191 greg 2.16 bsdf = sum/nssamp;
192 greg 2.15 }
193     }
194 greg 2.1 if (pctcull >= 0)
195     fwrite(&bsdf, sizeof(bsdf), 1, ofp);
196     else
197     printf("\t%.3e\n", bsdf);
198 greg 2.15 last_bsdf = bsdf;
199 greg 2.1 }
200 greg 2.15 }
201 greg 2.3 if (rbf != NULL)
202     free(rbf);
203 greg 2.1 }
204     if (pctcull >= 0) { /* finish output */
205     if (pclose(ofp)) {
206     fprintf(stderr, "%s: error running '%s'\n",
207     progname, cmd);
208     exit(1);
209     }
210     } else {
211     for (ix = sqres*sqres*sqres/2; ix--; )
212     fputs("\t0\n", stdout);
213     fputs("}\n", stdout);
214     }
215 greg 2.11 data_epilogue();
216 greg 2.1 }
217    
218     /* Interpolate and output anisotropic BSDF data */
219     static void
220 greg 2.11 eval_anisotropic(char *funame)
221 greg 2.1 {
222     const int sqres = 1<<samp_order;
223     FILE *ofp = NULL;
224 greg 2.17 int assignD = 0;
225 greg 2.1 char cmd[128];
226     int ix, iy, ox, oy;
227 greg 2.11 double iovec[6];
228 greg 2.4 float bsdf;
229 greg 2.15
230 greg 2.11 data_prologue(); /* begin output */
231     if (pctcull >= 0) {
232 greg 2.19 sprintf(cmd, "rttree_reduce%s -h -ff -r 4 -t %f -g %d",
233     (input_orient>0 ^ output_orient>0) ? "" : " -a",
234 greg 2.1 pctcull, samp_order);
235     fflush(stdout);
236     ofp = popen(cmd, "w");
237     if (ofp == NULL) {
238     fprintf(stderr, "%s: cannot create pipe to rttree_reduce\n",
239     progname);
240     exit(1);
241     }
242     } else
243     fputs("{\n", stdout);
244 greg 2.17 /* need to assign Dx, Dy, Dz? */
245     if (funame != NULL)
246     assignD = (fundefined(funame) < 6);
247 greg 2.1 /* run through directions */
248     for (ix = 0; ix < sqres; ix++)
249     for (iy = 0; iy < sqres; iy++) {
250 greg 2.11 RBFNODE *rbf = NULL; /* Klems reversal */
251 greg 2.15 SDsquare2disk(iovec, 1.-(ix+.5)/sqres, 1.-(iy+.5)/sqres);
252 greg 2.11 iovec[2] = input_orient *
253     sqrt(1. - iovec[0]*iovec[0] - iovec[1]*iovec[1]);
254     if (funame == NULL)
255 greg 2.18 rbf = advect_rbf(iovec, lobe_lim);
256 greg 2.15 for (ox = 0; ox < sqres; ox++) {
257     float last_bsdf = -1;
258 greg 2.1 for (oy = 0; oy < sqres; oy++) {
259 greg 2.11 SDsquare2disk(iovec+3, (ox+.5)/sqres, (oy+.5)/sqres);
260     iovec[5] = output_orient *
261     sqrt(1. - iovec[3]*iovec[3] - iovec[4]*iovec[4]);
262     if (funame == NULL)
263 greg 2.15 bsdf = eval_rbfrep(rbf, iovec+3) *
264 greg 2.11 output_orient/iovec[5];
265 greg 2.15 else {
266 greg 2.16 double ssa[4], ssvec[6], sum;
267 greg 2.15 int ssi;
268 greg 2.17 if (assignD) {
269     varset("Dx", '=', -iovec[3]);
270     varset("Dy", '=', -iovec[4]);
271     varset("Dz", '=', -iovec[5]);
272     ++eclock;
273     }
274 greg 2.15 bsdf = funvalue(funame, 6, iovec);
275     if (abs_diff(bsdf, last_bsdf) > ssamp_thresh) {
276 greg 2.16 sum = 0; /* super-sample voxel */
277 greg 2.15 for (ssi = nssamp; ssi--; ) {
278 greg 2.23 SDmultiSamp(ssa, 4, (ssi+frandom())/nssamp);
279 greg 2.15 SDsquare2disk(ssvec, 1.-(ix+ssa[0])/sqres,
280     1.-(iy+ssa[1])/sqres);
281     ssvec[2] = output_orient *
282     sqrt(1. - ssvec[0]*ssvec[0] -
283     ssvec[1]*ssvec[1]);
284     SDsquare2disk(ssvec+3, (ox+ssa[2])/sqres,
285     (oy+ssa[3])/sqres);
286     ssvec[5] = output_orient *
287     sqrt(1. - ssvec[3]*ssvec[3] -
288     ssvec[4]*ssvec[4]);
289 greg 2.17 if (assignD) {
290     varset("Dx", '=', -iovec[3]);
291     varset("Dy", '=', -iovec[4]);
292     varset("Dz", '=', -iovec[5]);
293     ++eclock;
294     }
295 greg 2.16 sum += funvalue(funame, 6, ssvec);
296 greg 2.15 }
297 greg 2.16 bsdf = sum/nssamp;
298 greg 2.15 }
299     }
300 greg 2.1 if (pctcull >= 0)
301     fwrite(&bsdf, sizeof(bsdf), 1, ofp);
302     else
303     printf("\t%.3e\n", bsdf);
304 greg 2.15 last_bsdf = bsdf;
305 greg 2.1 }
306 greg 2.15 }
307 greg 2.3 if (rbf != NULL)
308     free(rbf);
309 greg 2.1 }
310     if (pctcull >= 0) { /* finish output */
311     if (pclose(ofp)) {
312     fprintf(stderr, "%s: error running '%s'\n",
313     progname, cmd);
314     exit(1);
315     }
316     } else
317     fputs("}\n", stdout);
318 greg 2.11 data_epilogue();
319 greg 2.5 }
320    
321 greg 2.1 /* Read in BSDF and interpolate as tensor tree representation */
322     int
323     main(int argc, char *argv[])
324     {
325 greg 2.11 int dofwd = 0, dobwd = 1;
326     int i, na;
327 greg 2.1
328 greg 2.5 progname = argv[0];
329 greg 2.11 esupport |= E_VARIABLE|E_FUNCTION|E_RCONST;
330     esupport &= ~(E_INCHAN|E_OUTCHAN);
331     scompile("PI:3.14159265358979323846", NULL, 0);
332     biggerlib();
333     for (i = 1; i < argc-1 && (argv[i][0] == '-') | (argv[i][0] == '+'); i++)
334     switch (argv[i][1]) { /* get options */
335     case 'e':
336     scompile(argv[++i], NULL, 0);
337     break;
338     case 'f':
339     if (!argv[i][2])
340     fcompile(argv[++i]);
341     else
342     dofwd = (argv[i][0] == '+');
343     break;
344     case 'b':
345     dobwd = (argv[i][0] == '+');
346     break;
347 greg 2.1 case 't':
348 greg 2.11 switch (argv[i][2]) {
349     case '3':
350 greg 2.13 single_plane_incident = 1;
351 greg 2.11 break;
352     case '4':
353 greg 2.13 single_plane_incident = 0;
354 greg 2.11 break;
355     case '\0':
356     pctcull = atof(argv[++i]);
357     break;
358     default:
359     goto userr;
360     }
361 greg 2.1 break;
362     case 'g':
363 greg 2.5 samp_order = atoi(argv[++i]);
364 greg 2.1 break;
365 greg 2.18 case 'l':
366     lobe_lim = atoi(argv[++i]);
367     break;
368 greg 2.1 default:
369     goto userr;
370     }
371 greg 2.11 if (single_plane_incident >= 0) { /* function-based BSDF? */
372     void (*evf)(char *s) = single_plane_incident ?
373     &eval_isotropic : &eval_anisotropic;
374 greg 2.17 if (i != argc-1 || fundefined(argv[i]) < 3) {
375 greg 2.12 fprintf(stderr,
376     "%s: need single function with 6 arguments: bsdf(ix,iy,iz,ox,oy,oz)\n",
377     progname);
378 greg 2.17 fprintf(stderr, "\tor 3 arguments using Dx,Dy,Dz: bsdf(ix,iy,iz)\n",
379     progname);
380 greg 2.11 goto userr;
381 greg 2.12 }
382 greg 2.17 ++eclock;
383 greg 2.11 xml_prologue(argc, argv); /* start XML output */
384     if (dofwd) {
385     input_orient = -1;
386     output_orient = -1;
387     (*evf)(argv[i]); /* outside reflectance */
388     output_orient = 1;
389     (*evf)(argv[i]); /* outside -> inside */
390     }
391     if (dobwd) {
392     input_orient = 1;
393     output_orient = 1;
394     (*evf)(argv[i]); /* inside reflectance */
395     output_orient = -1;
396     (*evf)(argv[i]); /* inside -> outside */
397     }
398     xml_epilogue(); /* finish XML output & exit */
399     return(0);
400     }
401     if (i < argc) { /* open input files if given */
402 greg 2.12 int nbsdf = 0;
403 greg 2.11 for ( ; i < argc; i++) { /* interpolate each component */
404     FILE *fpin = fopen(argv[i], "rb");
405     if (fpin == NULL) {
406     fprintf(stderr, "%s: cannot open BSDF interpolant '%s'\n",
407     progname, argv[i]);
408     return(1);
409     }
410     if (!load_bsdf_rep(fpin))
411     return(1);
412     fclose(fpin);
413 greg 2.12 if (!nbsdf++) /* start XML on first dist. */
414     xml_prologue(argc, argv);
415 greg 2.11 if (single_plane_incident)
416     eval_isotropic(NULL);
417     else
418     eval_anisotropic(NULL);
419     }
420     xml_epilogue(); /* finish XML output & exit */
421     return(0);
422     }
423     SET_FILE_BINARY(stdin); /* load from stdin */
424     if (!load_bsdf_rep(stdin))
425 greg 2.1 return(1);
426 greg 2.5 xml_prologue(argc, argv); /* start XML output */
427 greg 2.1 if (single_plane_incident) /* resample dist. */
428 greg 2.11 eval_isotropic(NULL);
429 greg 2.1 else
430 greg 2.11 eval_anisotropic(NULL);
431     xml_epilogue(); /* finish XML output & exit */
432 greg 2.1 return(0);
433     userr:
434     fprintf(stderr,
435 greg 2.18 "Usage: %s [-g Nlog2][-t pctcull][-l maxlobes] [bsdf.sir ..] > bsdf.xml\n",
436 greg 2.11 progname);
437     fprintf(stderr,
438     " or: %s -t{3|4} [-g Nlog2][-t pctcull][{+|-}for[ward]][{+|-}b[ackward]][-e expr][-f file] bsdf_func > bsdf.xml\n",
439 greg 2.1 progname);
440     return(1);
441     }