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.64 by greg, Wed Jul 13 00:51:17 2016 UTC vs.
Revision 2.72 by greg, Fri Oct 14 00:54:22 2016 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;
# 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) {
172 > if ( $#dim != 5 ) {
173          @dim = split ' ', `getbbox -h $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 $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:
253 < my $CIEuv =     'Xi=.5141*Ri+.3239*Gi+.1620*Bi;' .
254 <                'Yi=.2651*Ri+.6701*Gi+.0648*Bi;' .
255 <                'Zi=.0241*Ri+.1229*Gi+.8530*Bi;' .
256 <                'den=Xi+15*Yi+3*Zi;' .
257 <                'uprime=4*Xi/den;vprime=9*Yi/den;' ;
252 > # Open unbuffered progress file
253 > open(MYPH, ">> $td/phase.txt");
254 > {
255 >        my $ofh = select MYPH;
256 >        $| = 1;
257 >        select $ofh;
258 > }
259 > $curphase = 0;
260   # Create data segments (all the work happens here)
261   if ( $tensortree ) {
262          do_tree_bsdf();
# Line 234 | Line 270 | die "Could not wrap BSDF data\n" if ( $? );
270   # Clean up temporary files and exit
271   exec $rmtmp;
272  
273 < #-------------- End of main program segment --------------#
273 > #============== End of main program segment ==============#
274  
275 + # Function to determine if next phase should be skipped or recovered
276 + sub do_phase {
277 +        $curphase++;
278 +        if ( defined $recovery ) {
279 +                if ( $recovery > $curphase ) { return 0; }
280 +                if ( $recovery == $curphase ) { return -1; }
281 +        }
282 +        print MYPH "$curphase\n";
283 +        return 1;
284 + }
285 +
286 + # Check if we are in active phase (not skipping parts)
287 + sub active_phase {
288 +        if ( defined $recovery ) {
289 +                if ( $recovery > $curphase ) { return 0; }
290 +                if ( $recovery == $curphase ) { return -1; }
291 +        }
292 +        return 1;
293 + }
294 +
295 + # Function to run program and check output if in active phase
296 + sub run_check {
297 +        if ( !active_phase() ) { return; }
298 +        my $cmd = shift;
299 +        # print STDERR "Running: $cmd\n";
300 +        system $cmd;
301 +        die "Failure running: $cmd\n" if ( $? );
302 + }
303 +
304   #++++++++++++++ Tensor tree BSDF generation ++++++++++++++#
305   sub do_tree_bsdf {
306  
# Line 248 | Line 313 | sub do_tree_bsdf {
313   # Call rfluxmtx and process tensor tree BSDF for the given direction
314   sub do_ttree_dir {
315          my $forw = shift;
316 +        my $dop = do_phase();
317 +        my $r = ($dop < 0) ? " -r" : "";
318          my $cmd;
319          if ( $tensortree == 3 ) {
320                  # Isotropic BSDF
# Line 263 | Line 330 | sub do_ttree_dir {
330                                  qq{-e "zp=$dim[5-$forw]" -e "myDz=Dz*($forw*2-1)" } .
331                                  qq{-e "\$1=xp-Dx;\$2=yp-Dy;\$3=zp-myDz" } .
332                                  qq{-e "\$4=Dx;\$5=Dy;\$6=myDz" } .
333 <                                "| $rfluxmtx -fa -y $ns2 - $receivers -i $octree";
333 >                                "| $rfluxmtx$r -fa -y $ns2 - $receivers -i $octree";
334                  } else {
335                          $cmd = "cnt $ns2 $ny $nx " .
336                                  qq{| rcalc -e "r1=rand(.8681*recno-.673892)" } .
# Line 275 | Line 342 | sub do_ttree_dir {
342                                  qq{-e "zp=$dim[5-$forw]" -e "myDz=Dz*($forw*2-1)" } .
343                                  qq{-e '\$1=xp-Dx;\$2=yp-Dy;\$3=zp-myDz' } .
344                                  qq{-e '\$4=Dx;\$5=Dy;\$6=myDz' -of } .
345 <                                "| $rfluxmtx -h -ff -y $ns2 - $receivers -i $octree";
345 >                                "| $rfluxmtx$r -h -ff -y $ns2 - $receivers -i $octree";
346                  }
347          } else {
348                  # Anisotropic BSDF
349                  my $sender = ($bsender,$fsender)[$forw];
350                  if ($windoz) {
351 <                        $cmd = "$rfluxmtx -fa $sender $receivers -i $octree";
351 >                        $cmd = "$rfluxmtx$r -fa $sender $receivers -i $octree";
352                  } else {
353 <                        $cmd = "$rfluxmtx -h -ff $sender $receivers -i $octree";
353 >                        $cmd = "$rfluxmtx$r -h -ff $sender $receivers -i $octree";
354                  }
355          }
356 <        # print STDERR "Starting: $cmd\n";
357 <        system $cmd;
358 <        die "Failure running rfluxmtx" if ( $? );
356 >        if ( $dop ) {
357 >                # print STDERR "Running: $cmd\n";
358 >                system $cmd;
359 >                die "Failure running rfluxmtx" if ( $? );
360 >        }
361          ttree_out($forw);
362   }       # end of do_ttree_dir()
363  
# Line 371 | Line 440 | sub ttree_comp {
440                          $cmd .= " -of $src " .
441                                          "| rttree_reduce$avg -h -ff -t $pcull -r $tensortree -g $ttlog2";
442                  }
443 <                # print STDERR "Running: $cmd\n";
375 <                system "$cmd > $dest";
376 <                die "Failure running rttree_reduce" if ( $? );
443 >                run_check "$cmd > $dest";
444          } else {
445                  if ($windoz) {
446                          $cmd = "rcollate -ho -oc 1 $src | " . $cmd ;
447                  } else {
448                          $cmd .= " $src";
449                  }
450 <                open(DATOUT, "> $dest");
451 <                print DATOUT "{\n";
452 <                close DATOUT;
453 <                # print STDERR "Running: $cmd\n";
454 <                system "$cmd >> $dest";
455 <                die "Failure running rcalc" if ( $? );
456 <                open(DATOUT, ">> $dest");
457 <                for (my $i = ($tensortree==3)*$ns*$ns*$ns/2; $i-- > 0; ) {
458 <                        print DATOUT "0\n";
450 >                if ( active_phase() ) {
451 >                        open(DATOUT, "> $dest");
452 >                        print DATOUT "{\n";
453 >                        close DATOUT;
454 >                        # print STDERR "Running: $cmd\n";
455 >                        system "$cmd >> $dest";
456 >                        die "Failure running rcalc" if ( $? );
457 >                        open(DATOUT, ">> $dest");
458 >                        for (my $i = ($tensortree==3)*$ns*$ns*$ns/2; $i-- > 0; ) {
459 >                                print DATOUT "0\n";
460 >                        }
461 >                        print DATOUT "}\n";
462 >                        close DATOUT;
463                  }
393                print DATOUT "}\n";
394                close DATOUT;
464          }
465          if ( "$spec" ne "$curspec" ) {
466                  $wrapper .= " -s $spec";
# Line 414 | Line 483 | sub do_matrix_bsdf {
483   # Call rfluxmtx and process tensor tree BSDF for the given direction
484   sub do_matrix_dir {
485          my $forw = shift;
486 <        my $cmd;
486 >        my $dop = do_phase();
487 >        my $r = ($dop < 0) ? " -r" : "";
488          my $sender = ($bsender,$fsender)[$forw];
489 <        $cmd = "$rfluxmtx -fd $sender $receivers -i $octree";
490 <        # print STDERR "Starting: $cmd\n";
491 <        system $cmd;
492 <        die "Failure running rfluxmtx" if ( $? );
489 >        my $cmd = "$rfluxmtx$r -fd $sender $receivers -i $octree";
490 >        if ( $dop ) {
491 >                # print STDERR "Running: $cmd\n";
492 >                system $cmd;
493 >                die "Failure running rfluxmtx" if ( $? );
494 >        }
495          matrix_out($forw);
496   }       # end of do_matrix_dir()
497  
# Line 463 | Line 535 | sub matrix_comp {
535          } elsif ("$spec" eq "CIE-Z") {
536                  $cmd .= " -c 0.0241 0.1229 0.8530";
537          }
538 <        $cmd .= " $src | rcollate -ho -oc 145";
539 <        # print STDERR "Running: $cmd\n";
468 <        system "$cmd > $dest";
469 <        die "Failure running rmtxop" if ( $? );
538 >        $cmd .= " $src | getinfo -";
539 >        run_check "$cmd > $dest";
540          if ( "$spec" ne "$curspec" ) {
541                  $wrapper .= " -s $spec";
542                  $curspec = $spec;

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines