src/util/genskyvec.pl

#!/usr/bin/perl -w
# RCSid $Id: genskyvec.pl,v 2.9 2014/09/17 22:40:49 greg Exp $
#
# Generate Reinhart vector for a given sky description
#
#       G. Ward
#
use strict;
my $windoz = ($^O eq "MSWin32" or $^O eq "MSWin64");
my @skycolor = (0.960, 1.004, 1.118);
my $mf = 4;
my $dosky = 1;
my $headout = 1;
my @origARGV = @ARGV;
while ($#ARGV >= 0) {
        if ("$ARGV[0]" eq "-c") {
                @skycolor = @ARGV[1..3];
                shift @ARGV for (1..3);
        } elsif ("$ARGV[0]" eq "-m") {
                $mf = $ARGV[1];
                shift @ARGV;
        } elsif ("$ARGV[0]" eq "-d") {
                $dosky = 0;
        } elsif ("$ARGV[0]" eq "-h") {
                $headout = 0;
        } else {
                die "Unexpected command-line argument: $ARGV[0]\n";
        }
        shift @ARGV;
}
# Load sky description into line array, separating sun if one
my @skydesc;
my $lightline;
my @sunval;
my $sunline;
my $skyOK = 0;
my $srcmod;     # putting this inside loop breaks code(?!)
while (<>) {
        push @skydesc, $_;
        if (/^\w+\s+light\s+/) {
                s/\s*$//; s/^.*\s//;
                $srcmod = $_;
                $lightline = $#skydesc;
        } elsif (defined($srcmod) && /^($srcmod)\s+source\s/) {
                @sunval = split(' ', $skydesc[$lightline + 3]);
                shift @sunval;
                $sunline = $#skydesc;
        } elsif (/\sskyfunc\s*$/) {
                $skyOK = 1;
        }
}
die "Bad sky description!\n" if (! $skyOK);
# Strip out the solar source if present
my @sundir;
if (defined $sunline) {
        @sundir = split(' ', $skydesc[$sunline + 3]);
        shift @sundir;
        undef @sundir if ($sundir[2] <= 0);
        splice(@skydesc, $sunline, 5);
}
# Reinhart sky sample generator
my $rhcal = 'DEGREE : PI/180;' .
        'x1 = .5; x2 = .5;' .
        'alpha : 90/(MF*7 + .5);' .
        'tnaz(r) : select(r, 30, 30, 24, 24, 18, 12, 6);' .
        'rnaz(r) : if(r-(7*MF-.5), 1, MF*tnaz(floor((r+.5)/MF) + 1));' .
        'raccum(r) : if(r-.5, rnaz(r-1) + raccum(r-1), 0);' .
        'RowMax : 7*MF + 1;' .
        'Rmax : raccum(RowMax);' .
        'Rfindrow(r, rem) : if(rem-rnaz(r)-.5, Rfindrow(r+1, rem-rnaz(r)), r);' .
        'Rrow = if(Rbin-(Rmax-.5), RowMax-1, Rfindrow(0, Rbin));' .
        'Rcol = Rbin - raccum(Rrow) - 1;' .
        'Razi_width = 2*PI / rnaz(Rrow);' .
        'RAH : alpha*DEGREE;' .
        'Razi = if(Rbin-.5, (Rcol + x2 - .5)*Razi_width, 2*PI*x2);' .
        'Ralt = if(Rbin-.5, (Rrow + x1)*RAH, asin(-x1));' .
        'Romega = if(.5-Rbin, 2*PI, if(Rmax-.5-Rbin, ' .
        '       Razi_width*(sin(RAH*(Rrow+1)) - sin(RAH*Rrow)),' .
        '       2*PI*(1 - cos(RAH/2)) ) );' .
        'cos_ralt = cos(Ralt);' .
        'Dx = sin(Razi)*cos_ralt;' .
        'Dy = cos(Razi)*cos_ralt;' .
        'Dz = sin(Ralt);' ;
my ($nbins, $octree, $tregcommand, $suncmd);
if ($windoz) {
        $nbins = `rcalc -n -e MF:$mf -e \"$rhcal\" -e \"\$1=Rmax+1\"`;
        chomp $nbins;
        $octree = "gtv$$.oct";
        $tregcommand = "cnt $nbins 16 | rcalc -e MF:$mf -e \"$rhcal\" " .
                q{-e "Rbin=$1;x1=rand(recno*.37-5.3);x2=rand(recno*-1.47+.86)" } .
                q{-e "$1=0;$2=0;$3=0;$4=Dx;$5=Dy;$6=Dz" } .
                "| rtrace -h -ab 0 -w $octree | total -16 -m";
        if (@sundir) {
                $suncmd = "cnt " . ($nbins-1) .
                        " | rcalc -e MF:$mf -e \"$rhcal\" -e Rbin=recno " .
                        "-e \"dot=Dx*$sundir[0] + Dy*$sundir[1] + Dz*$sundir[2]\" " .
                        "-e \"cond=dot-.866\" " .
                        q{-e "$1=if(1-dot,acos(dot),0);$2=Romega;$3=recno" };
        }
} else {
        $nbins = `rcalc -n -e MF:$mf -e \'$rhcal\' -e \'\$1=Rmax+1\'`;
        chomp $nbins;
        $octree = "/tmp/gtv$$.oct";
        $tregcommand = "cnt $nbins 16 | rcalc -of -e MF:$mf -e '$rhcal' " .
                q{-e 'Rbin=$1;x1=rand(recno*.37-5.3);x2=rand(recno*-1.47+.86)' } .
                q{-e '$1=0;$2=0;$3=0;$4=Dx;$5=Dy;$6=Dz' } .
                "| rtrace -h -ff -ab 0 -w $octree | total -if3 -16 -m";
        if (@sundir) {
                $suncmd = "cnt " . ($nbins-1) .
                        " | rcalc -e MF:$mf -e '$rhcal' -e Rbin=recno " .
                        "-e 'dot=Dx*$sundir[0] + Dy*$sundir[1] + Dz*$sundir[2]' " .
                        "-e 'cond=dot-.866' " .
                        q{-e '$1=if(1-dot,acos(dot),0);$2=Romega;$3=recno' };
        }
}
my @tregval;
if ($dosky) {
        # Create octree for rtrace
        open OCONV, "| oconv - > $octree";
        print OCONV @skydesc;
        print OCONV "skyfunc glow skyglow 0 0 4 @skycolor 0\n";
        print OCONV "skyglow source sky 0 0 4 0 0 1 360\n";
        close OCONV;
        # Run rtrace and average output for every 16 samples
        @tregval = `$tregcommand`;
        unlink $octree;
} else {
        push @tregval, "0\t0\t0\n" for (1..$nbins);
}
# Find closest 3 patches to sun and divvy up direct solar contribution
sub numSort1 {
        my @a1 = split("\t", $a);
        my @b1 = split("\t", $b);
        return ($a1[0] <=> $b1[0]);
}
if (@sundir) {
        my $somega = ($sundir[3]/360)**2 * 3.141592654**3;
        my @bestdir = `$suncmd`;
        @bestdir = sort numSort1 @bestdir;
        my (@ang, @dom, @ndx);
        my $wtot = 0;
        for my $i (0..2) {
                ($ang[$i], $dom[$i], $ndx[$i]) = split(' ', $bestdir[$i]);
                $wtot += 1./($ang[$i]+.02);
        }
        for my $i (0..2) {
                my $wt = 1./($ang[$i]+.02)/$wtot * $somega / $dom[$i];
                my @scolor = split(' ', $tregval[$ndx[$i]]);
                for my $j (0..2) { $scolor[$j] += $wt * $sunval[$j]; }
                $tregval[$ndx[$i]] = "$scolor[0]\t$scolor[1]\t$scolor[2]\n";
        }
}
# Output header if requested
if ($headout) {
        print "#?RADIANCE\n";
        print "genskyvec @origARGV\n";
        print "NROWS=", $#tregval+1, "\n";
        print "NCOLS=1\nNCOMP=3\n";
        print "FORMAT=ascii\n";
        print "\n";
}
# Output our final vector
print @tregval;
Revision:	2.10
Committed:	Mon Dec 12 17:40:07 2016 UTC (7 years, 4 months ago) by greg
Content type:	text/plain
Branch:	MAIN
CVS Tags:	rad5R1
Changes since 2.9:	+3 -1 lines
Log Message:	Added check for bad argument
#	User	Rev	Content
1	greg	2.1	#!/usr/bin/perl -w
2	greg	2.10	# RCSid $Id: genskyvec.pl,v 2.9 2014/09/17 22:40:49 greg Exp $
3	greg	2.1	#
4			# Generate Reinhart vector for a given sky description
5			#
6			# G. Ward
7			#
8			use strict;
9	greg	2.6	my $windoz = ($^O eq "MSWin32" or $^O eq "MSWin64");
10	greg	2.1	my @skycolor = (0.960, 1.004, 1.118);
11	greg	2.3	my $mf = 4;
12	greg	2.8	my $dosky = 1;
13	greg	2.9	my $headout = 1;
14			my @origARGV = @ARGV;
15	greg	2.1	while ($#ARGV >= 0) {
16			if ("$ARGV[0]" eq "-c") {
17			@skycolor = @ARGV[1..3];
18	greg	2.8	shift @ARGV for (1..3);
19	greg	2.3	} elsif ("$ARGV[0]" eq "-m") {
20			$mf = $ARGV[1];
21			shift @ARGV;
22	greg	2.8	} elsif ("$ARGV[0]" eq "-d") {
23			$dosky = 0;
24	greg	2.9	} elsif ("$ARGV[0]" eq "-h") {
25			$headout = 0;
26	greg	2.10	} else {
27			die "Unexpected command-line argument: $ARGV[0]\n";
28	greg	2.1	}
29			shift @ARGV;
30			}
31			# Load sky description into line array, separating sun if one
32			my @skydesc;
33			my $lightline;
34			my @sunval;
35			my $sunline;
36			my $skyOK = 0;
37			my $srcmod; # putting this inside loop breaks code(?!)
38			while (<>) {
39			push @skydesc, $_;
40			if (/^\w+\s+light\s+/) {
41			s/\s$//; s/^.\s//;
42			$srcmod = $_;
43			$lightline = $#skydesc;
44			} elsif (defined($srcmod) && /^($srcmod)\s+source\s/) {
45	greg	2.5	@sunval = split(' ', $skydesc[$lightline + 3]);
46	greg	2.1	shift @sunval;
47			$sunline = $#skydesc;
48			} elsif (/\sskyfunc\s*$/) {
49			$skyOK = 1;
50			}
51			}
52			die "Bad sky description!\n" if (! $skyOK);
53			# Strip out the solar source if present
54			my @sundir;
55			if (defined $sunline) {
56	greg	2.5	@sundir = split(' ', $skydesc[$sunline + 3]);
57	greg	2.1	shift @sundir;
58			undef @sundir if ($sundir[2] <= 0);
59			splice(@skydesc, $sunline, 5);
60			}
61			# Reinhart sky sample generator
62	greg	2.6	my $rhcal = 'DEGREE : PI/180;' .
63			'x1 = .5; x2 = .5;' .
64			'alpha : 90/(MF*7 + .5);' .
65			'tnaz(r) : select(r, 30, 30, 24, 24, 18, 12, 6);' .
66			'rnaz(r) : if(r-(7MF-.5), 1, MFtnaz(floor((r+.5)/MF) + 1));' .
67			'raccum(r) : if(r-.5, rnaz(r-1) + raccum(r-1), 0);' .
68			'RowMax : 7*MF + 1;' .
69			'Rmax : raccum(RowMax);' .
70			'Rfindrow(r, rem) : if(rem-rnaz(r)-.5, Rfindrow(r+1, rem-rnaz(r)), r);' .
71			'Rrow = if(Rbin-(Rmax-.5), RowMax-1, Rfindrow(0, Rbin));' .
72			'Rcol = Rbin - raccum(Rrow) - 1;' .
73			'Razi_width = 2*PI / rnaz(Rrow);' .
74			'RAH : alpha*DEGREE;' .
75			'Razi = if(Rbin-.5, (Rcol + x2 - .5)Razi_width, 2PI*x2);' .
76			'Ralt = if(Rbin-.5, (Rrow + x1)*RAH, asin(-x1));' .
77			'Romega = if(.5-Rbin, 2*PI, if(Rmax-.5-Rbin, ' .
78			' Razi_width(sin(RAH(Rrow+1)) - sin(RAH*Rrow)),' .
79			' 2PI(1 - cos(RAH/2)) ) );' .
80			'cos_ralt = cos(Ralt);' .
81			'Dx = sin(Razi)*cos_ralt;' .
82			'Dy = cos(Razi)*cos_ralt;' .
83			'Dz = sin(Ralt);' ;
84			my ($nbins, $octree, $tregcommand, $suncmd);
85			if ($windoz) {
86			$nbins = `rcalc -n -e MF:$mf -e \"$rhcal\" -e \"\$1=Rmax+1\"`;
87			chomp $nbins;
88			$octree = "gtv$$.oct";
89			$tregcommand = "cnt $nbins 16 \| rcalc -e MF:$mf -e \"$rhcal\" " .
90			q{-e "Rbin=$1;x1=rand(recno.37-5.3);x2=rand(recno-1.47+.86)" } .
91			q{-e "$1=0;$2=0;$3=0;$4=Dx;$5=Dy;$6=Dz" } .
92			"\| rtrace -h -ab 0 -w $octree \| total -16 -m";
93	greg	2.7	if (@sundir) {
94			$suncmd = "cnt " . ($nbins-1) .
95			" \| rcalc -e MF:$mf -e \"$rhcal\" -e Rbin=recno " .
96			"-e \"dot=Dx$sundir[0] + Dy$sundir[1] + Dz*$sundir[2]\" " .
97			"-e \"cond=dot-.866\" " .
98			q{-e "$1=if(1-dot,acos(dot),0);$2=Romega;$3=recno" };
99			}
100	greg	2.6	} else {
101			$nbins = `rcalc -n -e MF:$mf -e \'$rhcal\' -e \'\$1=Rmax+1\'`;
102			chomp $nbins;
103			$octree = "/tmp/gtv$$.oct";
104			$tregcommand = "cnt $nbins 16 \| rcalc -of -e MF:$mf -e '$rhcal' " .
105			q{-e 'Rbin=$1;x1=rand(recno.37-5.3);x2=rand(recno-1.47+.86)' } .
106			q{-e '$1=0;$2=0;$3=0;$4=Dx;$5=Dy;$6=Dz' } .
107			"\| rtrace -h -ff -ab 0 -w $octree \| total -if3 -16 -m";
108	greg	2.7	if (@sundir) {
109			$suncmd = "cnt " . ($nbins-1) .
110			" \| rcalc -e MF:$mf -e '$rhcal' -e Rbin=recno " .
111			"-e 'dot=Dx$sundir[0] + Dy$sundir[1] + Dz*$sundir[2]' " .
112			"-e 'cond=dot-.866' " .
113			q{-e '$1=if(1-dot,acos(dot),0);$2=Romega;$3=recno' };
114			}
115	greg	2.6	}
116	greg	2.8	my @tregval;
117			if ($dosky) {
118			# Create octree for rtrace
119			open OCONV, "\| oconv - > $octree";
120			print OCONV @skydesc;
121			print OCONV "skyfunc glow skyglow 0 0 4 @skycolor 0\n";
122			print OCONV "skyglow source sky 0 0 4 0 0 1 360\n";
123			close OCONV;
124			# Run rtrace and average output for every 16 samples
125			@tregval = `$tregcommand`;
126			unlink $octree;
127			} else {
128			push @tregval, "0\t0\t0\n" for (1..$nbins);
129			}
130	greg	2.1	# Find closest 3 patches to sun and divvy up direct solar contribution
131	greg	2.6	sub numSort1 {
132			my @a1 = split("\t", $a);
133			my @b1 = split("\t", $b);
134			return ($a1[0] <=> $b1[0]);
135			}
136	greg	2.1	if (@sundir) {
137			my $somega = ($sundir[3]/360)*2 3.141592654**3;
138	greg	2.6	my @bestdir = `$suncmd`;
139			@bestdir = sort numSort1 @bestdir;
140	greg	2.1	my (@ang, @dom, @ndx);
141			my $wtot = 0;
142			for my $i (0..2) {
143	greg	2.5	($ang[$i], $dom[$i], $ndx[$i]) = split(' ', $bestdir[$i]);
144	greg	2.1	$wtot += 1./($ang[$i]+.02);
145			}
146			for my $i (0..2) {
147			my $wt = 1./($ang[$i]+.02)/$wtot * $somega / $dom[$i];
148	greg	2.5	my @scolor = split(' ', $tregval[$ndx[$i]]);
149	greg	2.1	for my $j (0..2) { $scolor[$j] += $wt * $sunval[$j]; }
150	greg	2.4	$tregval[$ndx[$i]] = "$scolor[0]\t$scolor[1]\t$scolor[2]\n";
151	greg	2.1	}
152			}
153	greg	2.9	# Output header if requested
154			if ($headout) {
155			print "#?RADIANCE\n";
156			print "genskyvec @origARGV\n";
157			print "NROWS=", $#tregval+1, "\n";
158			print "NCOLS=1\nNCOMP=3\n";
159			print "FORMAT=ascii\n";
160			print "\n";
161			}
162	greg	2.1	# Output our final vector
163			print @tregval;