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

Comparing ray/src/util/genBSDF.pl (file contents):
Revision 2.64 by greg, Wed Jul 13 00:51:17 2016 UTC vs.
Revision 2.75 by greg, Tue Feb 14 07:06:55 2017 UTC

# Line 13 | Line 13 | sub userror {
13          exit 1;
14   }
15   my ($td,$radscn,$mgfscn,$octree,$fsender,$bsender,$receivers,$facedat,$behinddat,$rmtmp);
16 < my ($tf,$rf,$tb,$rb,$tfx,$rfx,$tbx,$rbx,$tfz,$rfz,$tbz,$rbz);
17 < if ($windoz) {
16 > my ($tf,$rf,$tb,$rb,$tfx,$rfx,$tbx,$rbx,$tfz,$rfz,$tbz,$rbz,$cph);
17 > my ($curphase, $recovery);
18 > if ($#ARGV == 1 && "$ARGV[0]" =~ /^-rec/) {
19 >        $td = $ARGV[1];
20 >        open(MYAVH, "< $td/savedARGV.txt") or die "$td: invalid path\n";
21 >        @ARGV = <MYAVH>;
22 >        close MYAVH;
23 >        chomp @ARGV;
24 >        $recovery = 0;
25 >        if (open(MYPH, "< $td/phase.txt")) {
26 >                while (<MYPH>) {
27 >                        chomp($recovery = $_);
28 >                }
29 >                close MYPH;
30 >        }
31 > } elsif ($windoz) {
32          my $tmploc = `echo \%TMP\%`;
33 <        chomp($tmploc);
33 >        chomp $tmploc;
34          $td = mkdtemp("$tmploc\\genBSDF.XXXXXX");
35 + } else {
36 +        $td = mkdtemp("/tmp/genBSDF.XXXXXX");
37 +        chomp $td;
38 + }
39 + if ($windoz) {
40          $radscn = "$td\\device.rad";
41          $mgfscn = "$td\\device.mgf";
42          $octree = "$td\\device.oct";
# Line 38 | Line 57 | if ($windoz) {
57          $rfz = "$td\\rfz.dat";
58          $tbz = "$td\\tbz.dat";
59          $rbz = "$td\\rbz.dat";
60 <        chomp $td;
60 >        $cph = "$td\\phase.txt";
61          $rmtmp = "rd /S /Q $td";
62   } else {
44 –        $td = mkdtemp("/tmp/genBSDF.XXXXXX");
45 –        chomp $td;
63          $radscn = "$td/device.rad";
64          $mgfscn = "$td/device.mgf";
65          $octree = "$td/device.oct";
# Line 63 | Line 80 | if ($windoz) {
80          $rfz = "$td/rfz.dat";
81          $tbz = "$td/tbz.dat";
82          $rbz = "$td/rbz.dat";
83 +        $cph = "$td/phase.txt";
84          $rmtmp = "rm -rf $td";
85   }
86   my @savedARGV = @ARGV;
87 < my $rfluxmtx = "rfluxmtx -ab 5 -ad 700 -lw 3e-6";
87 > my $rfluxmtx = "rfluxmtx -ab 5 -ad 700 -lw 3e-6 -w-";
88   my $wrapper = "wrapBSDF";
89   my $tensortree = 0;
90   my $ttlog2 = 4;
# Line 136 | Line 154 | while ($#ARGV >= 0) {
154   }
155   # Check that we're actually being asked to do something
156   die "Must have at least one of +forward or +backward\n" if (!$doforw && !$doback);
139 – # Issue warning for unhandled reciprocity case
140 – print STDERR "Warning: recommend both +forward and +backward with -t3\n" if
141 –                ($tensortree==3 && !($doforw && $doback));
157   $wrapper .= $tensortree ? " -a t$tensortree" : " -a kf -c";
158   $wrapper .= " -u $gunit";
159 < # Get scene description and dimensions
160 < if ( $mgfin ) {
161 <        system "mgf2rad @ARGV > $radscn";
162 <        die "Could not load MGF input\n" if ( $? );
163 < } else {
164 <        system "xform -e @ARGV > $radscn";
165 <        die "Could not load Radiance input\n" if ( $? );
159 > if ( !defined $recovery ) {
160 >        # Issue warning for unhandled reciprocity case
161 >        print STDERR "Warning: recommend both +forward and +backward with -t3\n" if
162 >                        ($tensortree==3 && !($doforw && $doback));
163 >        # Get scene description
164 >        if ( $mgfin ) {
165 >                system "mgf2rad @ARGV > $radscn";
166 >                die "Could not load MGF input\n" if ( $? );
167 >        } else {
168 >                system "xform -e @ARGV > $radscn";
169 >                die "Could not load Radiance input\n" if ( $? );
170 >        }
171   }
172 < if ($#dim != 5) {
173 <        @dim = split ' ', `getbbox -h $radscn`;
172 > if ( $#dim != 5 ) {
173 >        @dim = split ' ', `getbbox -h -w $radscn`;
174   }
175 < die "Device entirely inside room!\n" if ($dim[4] >= 0);
176 < if ($dim[5] > 1e-5) {
175 > die "Device entirely inside room!\n" if ( $dim[4] >= 0 );
176 > if ( $dim[5] > 1e-5 ) {
177          print STDERR "Warning: Device extends into room\n";
178 < } elsif ($dim[5]*$dim[5] < .01*($dim[1]-$dim[0])*($dim[3]-$dim[2])) {
178 > } elsif ( $dim[5]*$dim[5] > .01*($dim[1]-$dim[0])*($dim[3]-$dim[2]) ) {
179          print STDERR "Warning: Device far behind Z==0 plane\n";
180   }
181   # Assume Zmax==0 to derive thickness so pkgBSDF will work
182   $wrapper .= ' -f "t=' . (-$dim[4]) . ';w=' . ($dim[1] - $dim[0]) .
183                  ';h=' . ($dim[3] - $dim[2]) . '"';
184 < # Generate octree
185 < system "oconv -w $radscn > $octree";
186 < die "Could not compile scene\n" if ( $? );
187 < # Add MGF description if requested
188 < if ( $geout ) {
189 <        open(MGFSCN, "> $mgfscn");
190 <        printf MGFSCN "xf -t %.6f %.6f 0\n", -($dim[0]+$dim[1])/2, -($dim[2]+$dim[3])/2;
171 <        close MGFSCN;
172 <        if ( $mgfin ) {
173 <                system qq{mgfilt "#,o,xf,c,cxy,cspec,cmix,m,sides,rd,td,rs,ts,ir,v,p,n,f,fh,sph,cyl,cone,prism,ring,torus" @ARGV >> $mgfscn};
174 <        } else {
175 <                system "rad2mgf $radscn >> $mgfscn";
176 <        }
177 <        open(MGFSCN, ">> $mgfscn");
178 <        print MGFSCN "xf\n";
179 <        close MGFSCN;
180 <        $wrapper .= " -g $mgfscn";
181 < }
182 < # Create receiver & sender surfaces (rectangular)
184 > $wrapper .= " -g $mgfscn" if ( $geout );
185 > # Calculate CIE (u',v') from Radiance RGB:
186 > my $CIEuv =     'Xi=.5141*Ri+.3239*Gi+.1620*Bi;' .
187 >                'Yi=.2651*Ri+.6701*Gi+.0648*Bi;' .
188 >                'Zi=.0241*Ri+.1229*Gi+.8530*Bi;' .
189 >                'den=Xi+15*Yi+3*Zi;' .
190 >                'uprime=4*Xi/den;vprime=9*Yi/den;' ;
191   my $FEPS = 1e-5;
192 + my $ns = 2**$ttlog2;
193   my $nx = int(sqrt($nsamp*($dim[1]-$dim[0])/($dim[3]-$dim[2])) + 1);
194   my $ny = int($nsamp/$nx + 1);
195   $nsamp = $nx * $ny;
187 – my $ns = 2**$ttlog2;
188 – open(RADSCN, "> $receivers");
189 – print RADSCN '#@rfluxmtx ' . ($tensortree ? "h=-sc$ns\n" : "h=-kf\n");
190 – print RADSCN '#@rfluxmtx ' . "u=-Y o=$facedat\n\n";
191 – print RADSCN "void glow receiver_face\n0\n0\n4 1 1 1 0\n\n";
192 – print RADSCN "receiver_face source f_receiver\n0\n0\n4 0 0 1 180\n\n";
193 – print RADSCN '#@rfluxmtx ' . ($tensortree ? "h=+sc$ns\n" : "h=+kf\n");
194 – print RADSCN '#@rfluxmtx ' . "u=-Y o=$behinddat\n\n";
195 – print RADSCN "void glow receiver_behind\n0\n0\n4 1 1 1 0\n\n";
196 – print RADSCN "receiver_behind source b_receiver\n0\n0\n4 0 0 -1 180\n";
197 – close RADSCN;
198 – # Finish rfluxmtx command and prepare sender surfaces
196   $rfluxmtx .= " -n $nproc -c $nsamp";
197 < if ( $tensortree != 3 ) {       # Isotropic tensor tree is exception
198 <        open (RADSCN, "> $fsender");
199 <        print RADSCN '#@rfluxmtx u=-Y ' . ($tensortree ? "h=-sc$ns\n\n" : "h=-kf\n\n");
200 <        print RADSCN "void polygon fwd_sender\n0\n0\n12\n";
201 <        printf RADSCN "\t%e\t%e\t%e\n", $dim[0], $dim[2], $dim[4]-$FEPS;
202 <        printf RADSCN "\t%e\t%e\t%e\n", $dim[0], $dim[3], $dim[4]-$FEPS;
203 <        printf RADSCN "\t%e\t%e\t%e\n", $dim[1], $dim[3], $dim[4]-$FEPS;
204 <        printf RADSCN "\t%e\t%e\t%e\n", $dim[1], $dim[2], $dim[4]-$FEPS;
197 > if ( !defined $recovery ) {
198 >        open(MYAVH, "> $td/savedARGV.txt");
199 >        foreach (@savedARGV) {
200 >                print MYAVH "$_\n";
201 >        }
202 >        close MYAVH;
203 >        # Generate octree
204 >        system "oconv -w -f $radscn > $octree";
205 >        die "Could not compile scene\n" if ( $? );
206 >        # Add MGF description if requested
207 >        if ( $geout ) {
208 >                open(MGFSCN, "> $mgfscn");
209 >                printf MGFSCN "xf -t %.6f %.6f 0\n", -($dim[0]+$dim[1])/2, -($dim[2]+$dim[3])/2;
210 >                close MGFSCN;
211 >                if ( $mgfin ) {
212 >                        system qq{mgfilt "#,o,xf,c,cxy,cspec,cmix,m,sides,rd,td,rs,ts,ir,v,p,n,f,fh,sph,cyl,cone,prism,ring,torus" @ARGV >> $mgfscn};
213 >                } else {
214 >                        system "rad2mgf $radscn >> $mgfscn";
215 >                }
216 >                open(MGFSCN, ">> $mgfscn");
217 >                print MGFSCN "xf\n";
218 >                close MGFSCN;
219 >        }
220 >        # Create receiver & sender surfaces (rectangular)
221 >        open(RADSCN, "> $receivers");
222 >        print RADSCN '#@rfluxmtx ' . ($tensortree ? "h=-sc$ns\n" : "h=-kf\n");
223 >        print RADSCN '#@rfluxmtx ' . "u=-Y o=$facedat\n\n";
224 >        print RADSCN "void glow receiver_face\n0\n0\n4 1 1 1 0\n\n";
225 >        print RADSCN "receiver_face source f_receiver\n0\n0\n4 0 0 1 180\n\n";
226 >        print RADSCN '#@rfluxmtx ' . ($tensortree ? "h=+sc$ns\n" : "h=+kf\n");
227 >        print RADSCN '#@rfluxmtx ' . "u=-Y o=$behinddat\n\n";
228 >        print RADSCN "void glow receiver_behind\n0\n0\n4 1 1 1 0\n\n";
229 >        print RADSCN "receiver_behind source b_receiver\n0\n0\n4 0 0 -1 180\n";
230          close RADSCN;
231 <        open (RADSCN, "> $bsender");
232 <        print RADSCN '#@rfluxmtx u=-Y ' . ($tensortree ? "h=+sc$ns\n\n" : "h=+kf\n\n");
233 <        print RADSCN "void polygon bwd_sender\n0\n0\n12\n";
234 <        printf RADSCN "\t%e\t%e\t%e\n", $dim[0], $dim[2], $dim[5]+$FEPS;
235 <        printf RADSCN "\t%e\t%e\t%e\n", $dim[1], $dim[2], $dim[5]+$FEPS;
236 <        printf RADSCN "\t%e\t%e\t%e\n", $dim[1], $dim[3], $dim[5]+$FEPS;
237 <        printf RADSCN "\t%e\t%e\t%e\n", $dim[0], $dim[3], $dim[5]+$FEPS;
238 <        close RADSCN;
231 >        # Prepare sender surfaces
232 >        if ( $tensortree != 3 ) {       # Isotropic tensor tree is exception
233 >                open (RADSCN, "> $fsender");
234 >                print RADSCN '#@rfluxmtx u=-Y ' . ($tensortree ? "h=-sc$ns\n\n" : "h=-kf\n\n");
235 >                print RADSCN "void polygon fwd_sender\n0\n0\n12\n";
236 >                printf RADSCN "\t%e\t%e\t%e\n", $dim[0], $dim[2], $dim[4]-$FEPS;
237 >                printf RADSCN "\t%e\t%e\t%e\n", $dim[0], $dim[3], $dim[4]-$FEPS;
238 >                printf RADSCN "\t%e\t%e\t%e\n", $dim[1], $dim[3], $dim[4]-$FEPS;
239 >                printf RADSCN "\t%e\t%e\t%e\n", $dim[1], $dim[2], $dim[4]-$FEPS;
240 >                close RADSCN;
241 >                open (RADSCN, "> $bsender");
242 >                print RADSCN '#@rfluxmtx u=-Y ' . ($tensortree ? "h=+sc$ns\n\n" : "h=+kf\n\n");
243 >                print RADSCN "void polygon bwd_sender\n0\n0\n12\n";
244 >                printf RADSCN "\t%e\t%e\t%e\n", $dim[0], $dim[2], $dim[5]+$FEPS;
245 >                printf RADSCN "\t%e\t%e\t%e\n", $dim[1], $dim[2], $dim[5]+$FEPS;
246 >                printf RADSCN "\t%e\t%e\t%e\n", $dim[1], $dim[3], $dim[5]+$FEPS;
247 >                printf RADSCN "\t%e\t%e\t%e\n", $dim[0], $dim[3], $dim[5]+$FEPS;
248 >                close RADSCN;
249 >        }
250 >        print STDERR "Recover using: $0 -recover $td\n";
251   }
252 < # Calculate CIE (u',v') from Radiance RGB:
219 < my $CIEuv =     'Xi=.5141*Ri+.3239*Gi+.1620*Bi;' .
220 <                'Yi=.2651*Ri+.6701*Gi+.0648*Bi;' .
221 <                'Zi=.0241*Ri+.1229*Gi+.8530*Bi;' .
222 <                'den=Xi+15*Yi+3*Zi;' .
223 <                'uprime=4*Xi/den;vprime=9*Yi/den;' ;
252 > $curphase = 0;
253   # Create data segments (all the work happens here)
254   if ( $tensortree ) {
255          do_tree_bsdf();
# Line 234 | Line 263 | die "Could not wrap BSDF data\n" if ( $? );
263   # Clean up temporary files and exit
264   exec $rmtmp;
265  
266 < #-------------- End of main program segment --------------#
266 > #============== End of main program segment ==============#
267  
268 + # Function to determine if next phase should be skipped or recovered
269 + sub do_phase {
270 +        $curphase++;
271 +        if ( defined $recovery ) {
272 +                if ( $recovery > $curphase ) { return 0; }
273 +                if ( $recovery == $curphase ) { return -1; }
274 +        }
275 +        open(MYPH, ">> $td/phase.txt");
276 +        print MYPH "$curphase\n";
277 +        close MYPH;
278 +        return 1;
279 + }
280 +
281 + # Check if we are in active phase (not skipping parts)
282 + sub active_phase {
283 +        if ( defined $recovery ) {
284 +                if ( $recovery > $curphase ) { return 0; }
285 +                if ( $recovery == $curphase ) { return -1; }
286 +        }
287 +        return 1;
288 + }
289 +
290 + # Function to run program and check output if in active phase
291 + sub run_check {
292 +        if ( !active_phase() ) { return; }
293 +        my $cmd = shift;
294 +        # print STDERR "Running: $cmd\n";
295 +        system $cmd;
296 +        die "Failure running: $cmd\n" if ( $? );
297 + }
298 +
299   #++++++++++++++ Tensor tree BSDF generation ++++++++++++++#
300   sub do_tree_bsdf {
301  
# Line 248 | Line 308 | sub do_tree_bsdf {
308   # Call rfluxmtx and process tensor tree BSDF for the given direction
309   sub do_ttree_dir {
310          my $forw = shift;
311 +        my $dop = do_phase();
312 +        my $r = ($dop < 0) ? " -r" : "";
313          my $cmd;
314          if ( $tensortree == 3 ) {
315                  # Isotropic BSDF
# Line 263 | Line 325 | sub do_ttree_dir {
325                                  qq{-e "zp=$dim[5-$forw]" -e "myDz=Dz*($forw*2-1)" } .
326                                  qq{-e "\$1=xp-Dx;\$2=yp-Dy;\$3=zp-myDz" } .
327                                  qq{-e "\$4=Dx;\$5=Dy;\$6=myDz" } .
328 <                                "| $rfluxmtx -fa -y $ns2 - $receivers -i $octree";
328 >                                "| $rfluxmtx$r -fa -y $ns2 - $receivers -i $octree";
329                  } else {
330                          $cmd = "cnt $ns2 $ny $nx " .
331 <                                qq{| rcalc -e "r1=rand(.8681*recno-.673892)" } .
332 <                                qq{-e "r2=rand(-5.37138*recno+67.1737811)" } .
333 <                                qq{-e "r3=rand(+3.17603772*recno+83.766771)" } .
334 <                                qq{-e "Dx=1-2*(\$1+r1)/$ns;Dy:0;Dz=sqrt(1-Dx*Dx)" } .
335 <                                qq{-e "xp=(\$3+r2)*(($dim[1]-$dim[0])/$nx)+$dim[0]" } .
336 <                                qq{-e "yp=(\$2+r3)*(($dim[3]-$dim[2])/$ny)+$dim[2]" } .
337 <                                qq{-e "zp=$dim[5-$forw]" -e "myDz=Dz*($forw*2-1)" } .
331 >                                qq{| rcalc -e 'r1=rand(.8681*recno-.673892)' } .
332 >                                qq{-e 'r2=rand(-5.37138*recno+67.1737811)' } .
333 >                                qq{-e 'r3=rand(+3.17603772*recno+83.766771)' } .
334 >                                qq{-e 'Dx=1-2*(\$1+r1)/$ns;Dy:0;Dz=sqrt(1-Dx*Dx)' } .
335 >                                qq{-e 'xp=(\$3+r2)*(($dim[1]-$dim[0])/$nx)+$dim[0]' } .
336 >                                qq{-e 'yp=(\$2+r3)*(($dim[3]-$dim[2])/$ny)+$dim[2]' } .
337 >                                qq{-e 'zp=$dim[5-$forw]' -e 'myDz=Dz*($forw*2-1)' } .
338                                  qq{-e '\$1=xp-Dx;\$2=yp-Dy;\$3=zp-myDz' } .
339                                  qq{-e '\$4=Dx;\$5=Dy;\$6=myDz' -of } .
340 <                                "| $rfluxmtx -h -ff -y $ns2 - $receivers -i $octree";
340 >                                "| $rfluxmtx$r -h -ff -y $ns2 - $receivers -i $octree";
341                  }
342          } else {
343                  # Anisotropic BSDF
344                  my $sender = ($bsender,$fsender)[$forw];
345                  if ($windoz) {
346 <                        $cmd = "$rfluxmtx -fa $sender $receivers -i $octree";
346 >                        $cmd = "$rfluxmtx$r -fa $sender $receivers -i $octree";
347                  } else {
348 <                        $cmd = "$rfluxmtx -h -ff $sender $receivers -i $octree";
348 >                        $cmd = "$rfluxmtx$r -h -ff $sender $receivers -i $octree";
349                  }
350          }
351 <        # print STDERR "Starting: $cmd\n";
352 <        system $cmd;
353 <        die "Failure running rfluxmtx" if ( $? );
351 >        if ( $dop ) {
352 >                # print STDERR "Running: $cmd\n";
353 >                system $cmd;
354 >                die "Failure running rfluxmtx" if ( $? );
355 >        }
356          ttree_out($forw);
357   }       # end of do_ttree_dir()
358  
# Line 371 | Line 435 | sub ttree_comp {
435                          $cmd .= " -of $src " .
436                                          "| rttree_reduce$avg -h -ff -t $pcull -r $tensortree -g $ttlog2";
437                  }
438 <                # print STDERR "Running: $cmd\n";
375 <                system "$cmd > $dest";
376 <                die "Failure running rttree_reduce" if ( $? );
438 >                run_check "$cmd > $dest";
439          } else {
440                  if ($windoz) {
441                          $cmd = "rcollate -ho -oc 1 $src | " . $cmd ;
442                  } else {
443                          $cmd .= " $src";
444                  }
445 <                open(DATOUT, "> $dest");
446 <                print DATOUT "{\n";
447 <                close DATOUT;
448 <                # print STDERR "Running: $cmd\n";
449 <                system "$cmd >> $dest";
450 <                die "Failure running rcalc" if ( $? );
451 <                open(DATOUT, ">> $dest");
452 <                for (my $i = ($tensortree==3)*$ns*$ns*$ns/2; $i-- > 0; ) {
453 <                        print DATOUT "0\n";
445 >                if ( active_phase() ) {
446 >                        open(DATOUT, "> $dest");
447 >                        print DATOUT "{\n";
448 >                        close DATOUT;
449 >                        # print STDERR "Running: $cmd\n";
450 >                        system "$cmd >> $dest";
451 >                        die "Failure running rcalc" if ( $? );
452 >                        open(DATOUT, ">> $dest");
453 >                        for (my $i = ($tensortree==3)*$ns*$ns*$ns/2; $i-- > 0; ) {
454 >                                print DATOUT "0\n";
455 >                        }
456 >                        print DATOUT "}\n";
457 >                        close DATOUT;
458                  }
393 –                print DATOUT "}\n";
394 –                close DATOUT;
459          }
460          if ( "$spec" ne "$curspec" ) {
461                  $wrapper .= " -s $spec";
# Line 414 | Line 478 | sub do_matrix_bsdf {
478   # Call rfluxmtx and process tensor tree BSDF for the given direction
479   sub do_matrix_dir {
480          my $forw = shift;
481 <        my $cmd;
481 >        my $dop = do_phase();
482 >        my $r = ($dop < 0) ? " -r" : "";
483          my $sender = ($bsender,$fsender)[$forw];
484 <        $cmd = "$rfluxmtx -fd $sender $receivers -i $octree";
485 <        # print STDERR "Starting: $cmd\n";
486 <        system $cmd;
487 <        die "Failure running rfluxmtx" if ( $? );
484 >        my $cmd = "$rfluxmtx$r -fd $sender $receivers -i $octree";
485 >        if ( $dop ) {
486 >                # print STDERR "Running: $cmd\n";
487 >                system $cmd;
488 >                die "Failure running rfluxmtx" if ( $? );
489 >        }
490          matrix_out($forw);
491   }       # end of do_matrix_dir()
492  
# Line 463 | Line 530 | sub matrix_comp {
530          } elsif ("$spec" eq "CIE-Z") {
531                  $cmd .= " -c 0.0241 0.1229 0.8530";
532          }
533 <        $cmd .= " $src | rcollate -ho -oc 145";
534 <        # print STDERR "Running: $cmd\n";
468 <        system "$cmd > $dest";
469 <        die "Failure running rmtxop" if ( $? );
533 >        $cmd .= " $src | getinfo -";
534 >        run_check "$cmd > $dest";
535          if ( "$spec" ne "$curspec" ) {
536                  $wrapper .= " -s $spec";
537                  $curspec = $spec;

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines