ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/util/genBSDF.pl
(Generate patch)

Comparing ray/src/util/genBSDF.pl (file contents):
Revision 2.13 by greg, Sat Apr 16 00:39:07 2011 UTC vs.
Revision 2.20 by greg, Sat Jun 4 00:33:53 2011 UTC

# Line 8 | Line 8
8   use strict;
9   use File::Temp qw/ :mktemp  /;
10   sub userror {
11 <        print STDERR "Usage: genBSDF [-n Nproc][-c Nsamp][-r \"ropts\"][-dim xmin xmax ymin ymax zmin zmax][{+|-}f][{+|-}b][{+|-}mgf][{+|-}geom] [input ..]\n";
11 >        print STDERR "Usage: genBSDF [-n Nproc][-c Nsamp][-t{3|4} Nlog2][-r \"ropts\"][-dim xmin xmax ymin ymax zmin zmax][{+|-}f][{+|-}b][{+|-}mgf][{+|-}geom] [input ..]\n";
12          exit 1;
13   }
14   my $td = mkdtemp("/tmp/genBSDF.XXXXXX");
15   chomp $td;
16 + my @savedARGV = @ARGV;
17 + my $tensortree = 0;
18 + my $ttlog2 = 4;
19   my $nsamp = 1000;
20   my $rtargs = "-w -ab 5 -ad 700 -lw 3e-6";
21   my $mgfin = 0;
# Line 34 | Line 37 | while ($#ARGV >= 0) {
37                  $doforw = ("$ARGV[0]" =~ /^\+/);
38          } elsif ("$ARGV[0]" =~ /^[-+]b/) {
39                  $doback = ("$ARGV[0]" =~ /^\+/);
40 +        } elsif ("$ARGV[0]" =~ /^-t[34]$/) {
41 +                $tensortree = substr($ARGV[0], 2, 1);
42 +                $ttlog2 = $ARGV[1];
43 +                shift @ARGV;
44          } elsif ("$ARGV[0]" eq "-c") {
45                  $nsamp = $ARGV[1];
46                  shift @ARGV;
# Line 52 | Line 59 | while ($#ARGV >= 0) {
59          shift @ARGV;
60   }
61   # Check that we're actually being asked to do something
62 < die "Must have at least one of +forward or +backward" if (!$doforw && !$doback);
62 > die "Must have at least one of +forward or +backward\n" if (!$doforw && !$doback);
63 > # Name our own persist file?
64 > my $persistfile;
65 > if ( $tensortree && $nproc > 1 && "$rtargs" !~ /-PP /) {
66 >        $persistfile = "$td/pfile.txt";
67 >        $rtargs = "-PP $persistfile $rtargs";
68 > }
69   # Get scene description and dimensions
70   my $radscn = "$td/device.rad";
71   my $mgfscn = "$td/device.mgf";
# Line 82 | Line 95 | close RADSCN;
95   # Generate octree
96   system "oconv -w $radscn > $octree";
97   die "Could not compile scene\n" if ( $? );
98 < # Set up sampling of interior portal
98 > # Output XML prologue
99 > print
100 > '<?xml version="1.0" encoding="UTF-8"?>
101 > <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">
102 > ';
103 > print "<!-- File produced by: genBSDF @savedARGV -->\n";
104 > print
105 > '<WindowElementType>System</WindowElementType>
106 > <Optical>
107 > <Layer>
108 >        <Material>
109 >                <Name>Name</Name>
110 >                <Manufacturer>Manufacturer</Manufacturer>
111 > ';
112 > printf "\t\t<Thickness unit=\"Meter\">%.3f</Thickness>\n", $dim[5] - $dim[4];
113 > printf "\t\t<Width unit=\"Meter\">%.3f</Width>\n", $dim[1] - $dim[0];
114 > printf "\t\t<Height unit=\"Meter\">%.3f</Height>\n", $dim[3] - $dim[2];
115 > print "\t\t<DeviceType>Integral</DeviceType>\n";
116 > # Output MGF description if requested
117 > if ( $geout ) {
118 >        print "\t\t<Geometry format=\"MGF\" unit=\"Meter\">\n";
119 >        printf "xf -t %.6f %.6f 0\n", -($dim[0]+$dim[1])/2, -($dim[2]+$dim[3])/2;
120 >        open(MGFSCN, "< $mgfscn");
121 >        while (<MGFSCN>) { print $_; }
122 >        close MGFSCN;
123 >        print "xf\n";
124 >        print "\t\t</Geometry>\n";
125 > }
126 > print " </Material>\n";
127 > # Set up surface sampling
128 > my $nx = int(sqrt($nsamp*($dim[1]-$dim[0])/($dim[3]-$dim[2])) + .5);
129 > my $ny = int($nsamp/$nx + .5);
130 > $nsamp = $nx * $ny;
131 > my $ns = 2**$ttlog2;
132 > my (@pdiv, $disk2sq, $sq2disk, $tcal, $kcal);
133 > # Create data segments (all the work happens here)
134 > if ( $tensortree ) {
135 >        do_tree_bsdf();
136 > } else {
137 >        do_matrix_bsdf();
138 > }
139 > # Output XML epilogue
140 > print
141 > '</Layer>
142 > </Optical>
143 > </WindowElement>
144 > ';
145 > # Clean up temporary files and exit
146 > if ( $persistfile && open(PFI, "< $persistfile") ) {
147 >        while (<PFI>) {
148 >                s/^[^ ]* //;
149 >                kill('ALRM', $_);
150 >                last;
151 >        }
152 >        close PFI;
153 > }
154 > exec("rm -rf $td");
155 >
156 > #-------------- End of main program segment --------------#
157 >
158 > #++++++++++++++ Tensor tree BSDF generation ++++++++++++++#
159 > sub do_tree_bsdf {
160 > # Get sampling rate and subdivide task
161 > my $ns2 = $ns;
162 > $ns2 /= 2 if ( $tensortree == 3 );
163 > @pdiv = (0, int($ns2/$nproc));
164 > my $nrem = $ns2 % $nproc;
165 > for (my $i = 1; $i < $nproc; $i++) {
166 >        my $nv = $pdiv[$i] + $pdiv[1];
167 >        ++$nv if ( $nrem-- > 0 );
168 >        push @pdiv, $nv;
169 > }
170 > die "Script error 1" if ($pdiv[-1] != $ns2);
171 > # Shirley-Chiu mapping from unit square to disk
172 > $sq2disk = '
173 > in_square_a = 2*in_square_x - 1;
174 > in_square_b = 2*in_square_y - 1;
175 > in_square_rgn = if(in_square_a + in_square_b,
176 >                        if(in_square_a - in_square_b, 1, 2),
177 >                        if(in_square_b - in_square_a, 3, 4));
178 > out_disk_r = .999995*select(in_square_rgn, in_square_a, in_square_b,
179 >                        -in_square_a, -in_square_b);
180 > out_disk_phi = PI/4 * select(in_square_rgn,
181 >                                in_square_b/in_square_a,
182 >                                2 - in_square_a/in_square_b,
183 >                                4 + in_square_b/in_square_a,
184 >                                if(in_square_b*in_square_b,
185 >                                        6 - in_square_a/in_square_b, 0));
186 > Dx = out_disk_r*cos(out_disk_phi);
187 > Dy = out_disk_r*sin(out_disk_phi);
188 > Dz = sqrt(1 - out_disk_r*out_disk_r);
189 > ';
190 > # Shirley-Chiu mapping from unit disk to square
191 > $disk2sq = '
192 > norm_radians(p) : if(-p - PI/4, p + 2*PI, p);
193 > in_disk_r = .999995*sqrt(Dx*Dx + Dy*Dy);
194 > in_disk_phi = norm_radians(atan2(Dy, Dx));
195 > in_disk_rgn = floor((in_disk_phi + PI/4)/(PI/2)) + 1;
196 > out_square_a = select(in_disk_rgn,
197 >                        in_disk_r,
198 >                        (PI/2 - in_disk_phi)*in_disk_r/(PI/4),
199 >                        -in_disk_r,
200 >                        (in_disk_phi - 3*PI/2)*in_disk_r/(PI/4));
201 > out_square_b = select(in_disk_rgn,
202 >                        in_disk_phi*in_disk_r/(PI/4),
203 >                        in_disk_r,
204 >                        (PI - in_disk_phi)*in_disk_r/(PI/4),
205 >                        -in_disk_r);
206 > out_square_x = (out_square_a + 1)/2;
207 > out_square_y = (out_square_b + 1)/2;
208 > ';
209 > # Announce ourselves in XML output
210 > print "\t<DataDefinition>\n";
211 > print "\t\t<IncidentDataStructure>TensorTree$tensortree</IncidentDataStructure>\n";
212 > print "\t</DataDefinition>\n";
213 > # Fork parallel rtcontrib processes to compute each side
214 > if ( $doback ) {
215 >        for (my $proc = 0; $proc < $nproc; $proc++) {
216 >                bg_tree_rtcontrib(0, $proc);
217 >        }
218 >        while (wait() >= 0) {
219 >                die "rtcontrib process reported error" if ( $? );
220 >        }
221 >        ttree_out(0);
222 > }
223 > if ( $doforw ) {
224 >        for (my $proc = 0; $proc < $nproc; $proc++) {
225 >                bg_tree_rtcontrib(1, $proc);
226 >        }
227 >        while (wait() >= 0) {
228 >                die "rtcontrib process reported error" if ( $? );
229 >        }
230 >        ttree_out(1);
231 > }
232 > }       # end of sub do_tree_bsdf()
233 >
234 > # Run i'th rtcontrib process for generating tensor tree samples
235 > sub bg_tree_rtcontrib {
236 >        my $pid = fork();
237 >        die "Cannot fork new process" unless defined $pid;
238 >        if ($pid > 0) { return $pid; }
239 >        my $forw = shift;
240 >        my $pn = shift;
241 >        my $pbeg = $pdiv[$pn];
242 >        my $plen = $pdiv[$pn+1] - $pbeg;
243 >        my $matargs = "-m $bmodnm";
244 >        if ( !$forw || !$doback ) { $matargs .= " -m $fmodnm"; }
245 >        my $cmd = "rtcontrib $rtargs -h -ff -fo -c $nsamp " .
246 >                "-e '$disk2sq' -bn '$ns*$ns' " .
247 >                "-b '$ns*floor(out_square_x*$ns)+floor(out_square_y*$ns)' " .
248 >                "-o $td/%s_" . sprintf("%03d", $pn) . ".flt $matargs $octree";
249 >        if ( $tensortree == 3 ) {
250 >                # Isotropic BSDF
251 >                $cmd = "cnt $plen $ny $nx " .
252 >                        "| rcalc -e 'r1=rand(($pn+.8681)*recno-.673892)' " .
253 >                        "-e 'r2=rand(($pn-5.37138)*recno+67.1737811)' " .
254 >                        "-e 'r3=rand(($pn+3.17603772)*recno+83.766771)' " .
255 >                        "-e 'Dx=1-2*($pbeg+\$1+r1)/$ns;Dy:0;Dz=sqrt(1-Dx*Dx)' " .
256 >                        "-e 'xp=(\$3+r2)*(($dim[1]-$dim[0])/$nx)+$dim[0]' " .
257 >                        "-e 'yp=(\$2+r3)*(($dim[3]-$dim[2])/$ny)+$dim[2]' " .
258 >                        "-e 'zp=$dim[5-$forw]' -e 'myDz=Dz*($forw*2-1)' " .
259 >                        "-e '\$1=xp-Dx;\$2=yp-Dy;\$3=zp-myDz' " .
260 >                        "-e '\$4=Dx;\$5=Dy;\$6=myDz' -of " .
261 >                        "| $cmd";
262 >        } else {
263 >                # Anisotropic BSDF
264 >                # Sample area vertically to improve load balance, since
265 >                # shading systems usually have bilateral symmetry (L-R)
266 >                $cmd = "cnt $plen $ns $ny $nx " .
267 >                        "| rcalc -e 'r1=rand(($pn+.8681)*recno-.673892)' " .
268 >                        "-e 'r2=rand(($pn-5.37138)*recno+67.1737811)' " .
269 >                        "-e 'r3=rand(($pn+3.17603772)*recno+83.766771)' " .
270 >                        "-e 'r4=rand(($pn-2.3857833)*recno-964.72738)' " .
271 >                        "-e 'in_square_x=($pbeg+\$1+r1)/$ns' " .
272 >                        "-e 'in_square_y=(\$2+r2)/$ns' -e '$sq2disk' " .
273 >                        "-e 'xp=(\$4+r3)*(($dim[1]-$dim[0])/$nx)+$dim[0]' " .
274 >                        "-e 'yp=(\$3+r4)*(($dim[3]-$dim[2])/$ny)+$dim[2]' " .
275 >                        "-e 'zp=$dim[5-$forw]' -e 'myDz=Dz*($forw*2-1)' " .
276 >                        "-e '\$1=xp-Dx;\$2=yp-Dy;\$3=zp-myDz' " .
277 >                        "-e '\$4=Dx;\$5=Dy;\$6=myDz' -of " .
278 >                        "| $cmd";
279 >        }
280 > # print STDERR "Starting: $cmd\n";
281 >        exec($cmd);             # no return; status report to parent via wait
282 >        die "Cannot exec: $cmd\n";
283 > }       # end of bg_tree_rtcontrib()
284 >
285 > # Simplify and output tensor tree results
286 > sub ttree_out {
287 >        my $forw = shift;
288 >        my $side = ("Back","Front")[$forw];
289 > # Only output one transmitted distribution, preferring backwards
290 > if ( !$forw || !$doback ) {
291 > print
292 > '       <WavelengthData>
293 >                <LayerNumber>System</LayerNumber>
294 >                <Wavelength unit="Integral">Visible</Wavelength>
295 >                <SourceSpectrum>CIE Illuminant D65 1nm.ssp</SourceSpectrum>
296 >                <DetectorSpectrum>ASTM E308 1931 Y.dsp</DetectorSpectrum>
297 >                <WavelengthDataBlock>
298 >                        <WavelengthDataDirection>Transmission</WavelengthDataDirection>
299 >                        <AngleBasis>LBNL/Shirley-Chiu</AngleBasis>
300 >                        <ScatteringDataType>BTDF</ScatteringDataType>
301 >                        <ScatteringData>
302 > ';
303 > system "rcalc -if3 -e 'Omega:PI/($ns*$ns)' " .
304 >        q{-e '$1=(0.265*$1+0.670*$2+0.065*$3)/Omega' -of } .
305 >        "$td/" . ($bmodnm,$fmodnm)[$forw] . "_???.flt " .
306 >        "| rttree_reduce -h -ff -r $tensortree -g $ttlog2";
307 > die "Failure running rttree_reduce" if ( $? );
308 > print
309 > '                       </ScatteringData>
310 >                </WavelengthDataBlock>
311 >        </WavelengthData>
312 > ';
313 > }
314 > # Output reflection
315 > print
316 > '       <WavelengthData>
317 >                <LayerNumber>System</LayerNumber>
318 >                <Wavelength unit="Integral">Visible</Wavelength>
319 >                <SourceSpectrum>CIE Illuminant D65 1nm.ssp</SourceSpectrum>
320 >                <DetectorSpectrum>ASTM E308 1931 Y.dsp</DetectorSpectrum>
321 >                <WavelengthDataBlock>
322 > ';
323 > print "\t\t\t<WavelengthDataDirection>Reflection $side</WavelengthDataDirection>\n";
324 > print
325 > '                       <AngleBasis>LBNL/Shirley-Chiu</AngleBasis>
326 >                        <ScatteringDataType>BRDF</ScatteringDataType>
327 >                        <ScatteringData>
328 > ';
329 > system "rcalc -if3 -e 'Omega:PI/($ns*$ns)' " .
330 >        q{-e '$1=(0.265*$1+0.670*$2+0.065*$3)/Omega' -of } .
331 >        "$td/" . ($fmodnm,$bmodnm)[$forw] . "_???.flt " .
332 >        "| rttree_reduce -h -ff -r $tensortree -g $ttlog2";
333 > die "Failure running rttree_reduce" if ( $? );
334 > print
335 > '                       </ScatteringData>
336 >                </WavelengthDataBlock>
337 >        </WavelengthData>
338 > ';
339 > }       # end of ttree_out()
340 >
341 > #------------- End of do_tree_bsdf() & subroutines -------------#
342 >
343 > #+++++++++++++++ Klems matrix BSDF generation +++++++++++++++#
344 > sub do_matrix_bsdf {
345 > # Set up sampling of portal
346   # Kbin to produce incident direction in full Klems basis with (x1,x2) randoms
347 < my $tcal = '
347 > $tcal = '
348   DEGREE : PI/180;
349   sq(x) : x*x;
350   Kpola(r) : select(r+1, -5, 5, 15, 25, 35, 45, 55, 65, 75, 90);
# Line 105 | Line 365 | KprojOmega = PI * if(Kbin-.5,
365          1 - sq(cos(Kpola(1)*DEGREE)));
366   ';
367   # Compute Klems bin from exiting ray direction (forward or backward)
368 < my $kcal = '
368 > $kcal = '
369   DEGREE : PI/180;
370   abs(x) : if(x, x, -x);
371   Acos(x) : 1/DEGREE * if(x-1, 0, if(-1-x, 0, acos(x)));
# Line 131 | Line 391 | kbin2(pol,azi) = select(kfindrow(1, pol),
391   kbin = kbin2(Acos(abs(Dz)),Atan2(Dy,Dx));
392   ';
393   my $ndiv = 145;
134 my $nx = int(sqrt($nsamp*($dim[1]-$dim[0])/($dim[3]-$dim[2])) + .5);
135 my $ny = int($nsamp/$nx + .5);
136 $nsamp = $nx * $ny;
394   # Compute scattering data using rtcontrib
395   my @tfarr;
396   my @rfarr;
397   my @tbarr;
398   my @rbarr;
399   my $cmd;
400 < my $rtcmd = "rtcontrib -h -ff -fo -n $nproc -c $nsamp " .
400 > my $rtcmd = "rtcontrib $rtargs -h -ff -fo -n $nproc -c $nsamp " .
401          "-e '$kcal' -b kbin -bn $ndiv " .
402 <        "-o '$td/%s.flt' -m $fmodnm -m $bmodnm $rtargs $octree";
402 >        "-o '$td/%s.flt' -m $fmodnm -m $bmodnm $octree";
403   my $rccmd = "rcalc -e '$tcal' " .
404          "-e 'mod(n,d):n-floor(n/d)*d' -e 'Kbin=mod(recno-.999,$ndiv)' " .
405          q{-if3 -e '$1=(0.265*$1+0.670*$2+0.065*$3)/KprojOmega'};
# Line 174 | Line 431 | die "Failure running: $rccmd $td/$bmodnm.flt\n" if ( $
431   @rbarr = `$rccmd $td/$fmodnm.flt`;
432   die "Failure running: $rccmd $td/$fmodnm.flt\n" if ( $? );
433   }
434 < # Output XML prologue
434 > # Output angle basis
435   print
436 < '<?xml version="1.0" encoding="UTF-8"?>
437 < <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">
438 <        <WindowElementType>System</WindowElementType>
182 <        <Optical>
183 <                <Layer>
184 <                <Material>
185 <                        <Name>Name</Name>
186 <                        <Manufacturer>Manufacturer</Manufacturer>
187 < ';
188 < printf "\t\t\t<Thickness unit=\"Meter\">%.3f</Thickness>\n", $dim[5] - $dim[4];
189 < printf "\t\t\t<Width unit=\"Meter\">%.3f</Width>\n", $dim[1] - $dim[0];
190 < printf "\t\t\t<Height unit=\"Meter\">%.3f</Height>\n", $dim[3] - $dim[2];
191 < print "\t\t\t<DeviceType>Integral</DeviceType>\n";
192 < # Output MGF description if requested
193 < if ( $geout ) {
194 <        print "\t\t\t<Geometry format=\"MGF\" unit=\"Meter\">\n";
195 <        printf "xf -t %.6f %.6f 0\n", -($dim[0]+$dim[1])/2, -($dim[2]+$dim[3])/2;
196 <        system "cat $mgfscn";
197 <        print "xf\n";
198 <        print "\t\t\t</Geometry>\n";
199 < }
200 < print '                 </Material>
201 <                <DataDefinition>
202 <                        <IncidentDataStructure>Columns</IncidentDataStructure>
203 <                        <AngleBasis>
436 > '       <DataDefinition>
437 >                <IncidentDataStructure>Columns</IncidentDataStructure>
438 >                <AngleBasis>
439                          <AngleBasisName>LBNL/Klems Full</AngleBasisName>
440 <                                <AngleBasisBlock>
440 >                        <AngleBasisBlock>
441                                  <Theta>0</Theta>
442                                  <nPhis>1</nPhis>
443                                  <ThetaBounds>
444 <                                <LowerTheta>0</LowerTheta>
445 <                                <UpperTheta>5</UpperTheta>
444 >                                        <LowerTheta>0</LowerTheta>
445 >                                        <UpperTheta>5</UpperTheta>
446                                  </ThetaBounds>
447                                  </AngleBasisBlock>
448                                  <AngleBasisBlock>
# Line 278 | Line 513 | print '                        </Material>
513          </DataDefinition>
514   ';
515   if ( $doforw ) {
516 < print '         <WavelengthData>
516 > print
517 > '       <WavelengthData>
518                  <LayerNumber>System</LayerNumber>
519                  <Wavelength unit="Integral">Visible</Wavelength>
520                  <SourceSpectrum>CIE Illuminant D65 1nm.ssp</SourceSpectrum>
# Line 298 | Line 534 | for (my $od = 0; $od < $ndiv; $od++) {
534          print "\n";
535   }
536   print
537 < '               </ScatteringData>
538 <        </WavelengthDataBlock>
537 > '                       </ScatteringData>
538 >                </WavelengthDataBlock>
539          </WavelengthData>
540          <WavelengthData>
541                  <LayerNumber>System</LayerNumber>
# Line 321 | Line 557 | for (my $od = 0; $od < $ndiv; $od++) {
557          print "\n";
558   }
559   print
560 < '               </ScatteringData>
561 <        </WavelengthDataBlock>
560 > '                       </ScatteringData>
561 >                </WavelengthDataBlock>
562          </WavelengthData>
563   ';
564   }
565   if ( $doback ) {
566 < print '         <WavelengthData>
566 > print
567 > '       <WavelengthData>
568                  <LayerNumber>System</LayerNumber>
569                  <Wavelength unit="Integral">Visible</Wavelength>
570                  <SourceSpectrum>CIE Illuminant D65 1nm.ssp</SourceSpectrum>
# Line 347 | Line 584 | for (my $od = 0; $od < $ndiv; $od++) {
584          print "\n";
585   }
586   print
587 < '               </ScatteringData>
588 <        </WavelengthDataBlock>
587 > '                       </ScatteringData>
588 >                </WavelengthDataBlock>
589          </WavelengthData>
590          <WavelengthData>
591                  <LayerNumber>System</LayerNumber>
# Line 370 | Line 607 | for (my $od = 0; $od < $ndiv; $od++) {
607          print "\n";
608   }
609   print
610 < '               </ScatteringData>
611 <        </WavelengthDataBlock>
610 > '                       </ScatteringData>
611 >                </WavelengthDataBlock>
612          </WavelengthData>
613   ';
614   }
615 < # Output XML epilogue
616 < print '</Layer>
380 < </Optical>
381 < </WindowElement>
382 < ';
383 < # Clean up temporary files
384 < system "rm -rf $td";
615 > }
616 > #------------- End of do_matrix_bsdf() --------------#

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines