ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/util/genskyvec.pl
Revision: 2.5
Committed: Thu Jan 27 22:28:12 2011 UTC (11 years, 10 months ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.4: +5 -5 lines
Log Message:
Fixed issue with leading white space in split command

File Contents

# Content
1 #!/usr/bin/perl -w
2 # RCSid $Id: genskyvec.pl,v 2.4 2010/01/23 22:30:29 greg Exp $
3 #
4 # Generate Reinhart vector for a given sky description
5 #
6 # G. Ward
7 #
8 use strict;
9 my @skycolor = (0.960, 1.004, 1.118);
10 my $mf = 4;
11 while ($#ARGV >= 0) {
12 if ("$ARGV[0]" eq "-c") {
13 @skycolor = @ARGV[1..3];
14 shift @ARGV; shift @ARGV; shift @ARGV;
15 } elsif ("$ARGV[0]" eq "-m") {
16 $mf = $ARGV[1];
17 shift @ARGV;
18 }
19 shift @ARGV;
20 }
21 # Load sky description into line array, separating sun if one
22 my @skydesc;
23 my $lightline;
24 my @sunval;
25 my $sunline;
26 my $skyOK = 0;
27 my $srcmod; # putting this inside loop breaks code(?!)
28 while (<>) {
29 push @skydesc, $_;
30 if (/^\w+\s+light\s+/) {
31 s/\s*$//; s/^.*\s//;
32 $srcmod = $_;
33 $lightline = $#skydesc;
34 } elsif (defined($srcmod) && /^($srcmod)\s+source\s/) {
35 @sunval = split(' ', $skydesc[$lightline + 3]);
36 shift @sunval;
37 $sunline = $#skydesc;
38 } elsif (/\sskyfunc\s*$/) {
39 $skyOK = 1;
40 }
41 }
42 die "Bad sky description!\n" if (! $skyOK);
43 # Strip out the solar source if present
44 my @sundir;
45 if (defined $sunline) {
46 @sundir = split(' ', $skydesc[$sunline + 3]);
47 shift @sundir;
48 undef @sundir if ($sundir[2] <= 0);
49 splice(@skydesc, $sunline, 5);
50 }
51 # Reinhart sky sample generator
52 my $rhcal = '
53 DEGREE : PI/180;
54 x1 = .5; x2 = .5;
55 alpha : 90/(MF*7 + .5);
56 tnaz(r) : select(r, 30, 30, 24, 24, 18, 12, 6);
57 rnaz(r) : if(r-(7*MF-.5), 1, MF*tnaz(floor((r+.5)/MF) + 1));
58 raccum(r) : if(r-.5, rnaz(r-1) + raccum(r-1), 0);
59 RowMax : 7*MF + 1;
60 Rmax : raccum(RowMax);
61 Rfindrow(r, rem) : if(rem-rnaz(r)-.5, Rfindrow(r+1, rem-rnaz(r)), r);
62 Rrow = if(Rbin-(Rmax-.5), RowMax-1, Rfindrow(0, Rbin));
63 Rcol = Rbin - raccum(Rrow) - 1;
64 Razi_width = 2*PI / rnaz(Rrow);
65 RAH : alpha*DEGREE;
66 Razi = if(Rbin-.5, (Rcol + x2 - .5)*Razi_width, 2*PI*x2);
67 Ralt = if(Rbin-.5, (Rrow + x1)*RAH, asin(-x1));
68 Romega = if(.5-Rbin, 2*PI, if(Rmax-.5-Rbin,
69 Razi_width*(sin(RAH*(Rrow+1)) - sin(RAH*Rrow)),
70 2*PI*(1 - cos(RAH/2)) ) );
71 cos_ralt = cos(Ralt);
72 Dx = sin(Razi)*cos_ralt;
73 Dy = cos(Razi)*cos_ralt;
74 Dz = sin(Ralt);
75 ';
76 my $nbins = `rcalc -n -e MF:$mf -e \'$rhcal\' -e \'\$1=Rmax+1\'`;
77 chomp $nbins;
78 # Create octree for rtrace
79 my $octree = "/tmp/gtv$$.oct";
80 open OCONV, "| oconv - > $octree";
81 print OCONV @skydesc;
82 print OCONV "skyfunc glow skyglow 0 0 4 @skycolor 0\n";
83 print OCONV "skyglow source sky 0 0 4 0 0 1 360\n";
84 close OCONV;
85 # Run rtrace and average output for every 16 samples
86 my $tregcommand = "cnt $nbins 16 | rcalc -of -e MF:$mf -e '$rhcal' " .
87 q{-e 'Rbin=$1;x1=rand(recno*.37-5.3);x2=rand(recno*-1.47+.86)' } .
88 q{-e '$1=0;$2=0;$3=0;$4=Dx;$5=Dy;$6=Dz' } .
89 "| rtrace -h -ff -ab 0 -w $octree | total -if3 -16 -m";
90 my @tregval = `$tregcommand`;
91 unlink $octree;
92 # Find closest 3 patches to sun and divvy up direct solar contribution
93 my @bestdir;
94 if (@sundir) {
95 my $somega = ($sundir[3]/360)**2 * 3.141592654**3;
96 my $cmd = "cnt " . ($nbins-1) .
97 " | rcalc -e MF:$mf -e '$rhcal' -e Rbin=recno " .
98 "-e 'dot=Dx*$sundir[0] + Dy*$sundir[1] + Dz*$sundir[2]' " .
99 "-e 'cond=dot-.866' " .
100 q{-e '$1=if(1-dot,acos(dot),0);$2=Romega;$3=recno' };
101 @bestdir = `$cmd | sort -n | head -3`;
102 my (@ang, @dom, @ndx);
103 my $wtot = 0;
104 for my $i (0..2) {
105 ($ang[$i], $dom[$i], $ndx[$i]) = split(' ', $bestdir[$i]);
106 $wtot += 1./($ang[$i]+.02);
107 }
108 for my $i (0..2) {
109 my $wt = 1./($ang[$i]+.02)/$wtot * $somega / $dom[$i];
110 my @scolor = split(' ', $tregval[$ndx[$i]]);
111 for my $j (0..2) { $scolor[$j] += $wt * $sunval[$j]; }
112 $tregval[$ndx[$i]] = "$scolor[0]\t$scolor[1]\t$scolor[2]\n";
113 }
114 }
115 # Output our final vector
116 print @tregval;