ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/util/rtpict.pl
Revision: 2.16
Committed: Wed Apr 8 00:58:32 2020 UTC (3 years, 11 months ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.15: +2 -2 lines
Log Message:
Added support for persist mode, which rtrace will complain about if -n > 1

File Contents

# Content
1 #!/usr/bin/perl -w
2 # RCSid $Id: rtpict.pl,v 2.15 2020/03/15 16:54:19 greg Exp $
3 #
4 # Run rtrace in parallel mode to simulate rpict -n option
5 # May also be used to render layered images with -o* option
6 #
7 # G. Ward
8 #
9 use strict;
10 # we'll run rpict if no -n or -o* option
11 my $nprocs = 1;
12 # rtrace options and the associated number of arguments
13 my %rtraceC = ('-dt',1, '-dc',1, '-dj',1, '-ds',1, '-dr',1, '-dp',1,
14 '-ss',1, '-st',1, '-e',1, '-am',1, '-P',1, '-PP',1,
15 '-ab',1, '-af',1, '-ai',1, '-aI',1, '-ae',1, '-aE',1,
16 '-av',3, '-aw',1, '-aa',1, '-ar',1, '-ad',1, '-as',1,
17 '-me',3, '-ma',3, '-mg',1, '-ms',1, '-lr',1, '-lw',1);
18 # boolean rtrace options
19 my @boolO = ('-w', '-bv', '-dv', '-i', '-u');
20 # view options and the associated number of arguments
21 my %vwraysC = ('-vf',1, '-vtv',0, '-vtl',0, '-vth',0, '-vta',0, '-vts',0, '-vtc',0,
22 '-x',1, '-y',1, '-vp',3, '-vd',3, '-vu',3, '-vh',1, '-vv',1,
23 '-vo',1, '-va',1, '-vs',1, '-vl',1, '-pa',1, '-pj',1);
24 # options we need to silently ignore
25 my %ignoreC = ('-t',1, '-ps',1, '-pt',1, '-pm',1, '-pd',1);
26 # Starting options for rtrace (rpict values)
27 my @rtraceA = split(' ', 'rtrace -u- -dt .05 -dc .5 -ds .25 -dr 1 ' .
28 '-aa .2 -ar 64 -ad 512 -as 128 -lr 7 -lw 1e-03');
29 my @vwraysA = ('vwrays', '-ff', '-pj', '.67');
30 my @vwrightA = ('vwright', '-vtv');
31 my @rpictA = ('rpict');
32 my $outpatt = '^-o[vrxlLRXnNsmM]+';
33 my $refDepth = "";
34 my $irrad = 0;
35 my $outlyr;
36 my $outdir;
37 my $outpic;
38 my $outzbf;
39 OPTION: # sort through options
40 while ($#ARGV >= 0 && "$ARGV[0]" =~ /^[-\@]/) {
41 # Check for file inclusion
42 if ("$ARGV[0]" =~ /^\@/) {
43 open my $handle, '<', substr($ARGV[0], 1) or die "No file for $ARGV[0]\n";
44 shift @ARGV;
45 chomp(my @args = <$handle>);
46 close $handle;
47 unshift @ARGV, split(/\s+/, "@args");
48 next OPTION;
49 }
50 # Check booleans
51 for my $boopt (@boolO) {
52 if ("$ARGV[0]" =~ ('^' . $boopt . '[-+01tfynTFYN]?$')) {
53 if ("$ARGV[0]" eq '-i') {
54 $irrad = ! $irrad;
55 } elsif ("$ARGV[0]" =~ /^-i[-+01tfynTFYN]/) {
56 $irrad = ("$ARGV[0]" =~ /^-i[+1tyTY]/);
57 }
58 push @rtraceA, $ARGV[0];
59 push @rpictA, shift(@ARGV);
60 next OPTION;
61 }
62 }
63 # Check view options
64 if (defined $vwraysC{$ARGV[0]}) {
65 push @vwraysA, $ARGV[0];
66 my $isvopt = ("$ARGV[0]" =~ /^-v/);
67 push(@vwrightA, $ARGV[0]) if ($isvopt);
68 push @rpictA, shift(@ARGV);
69 for (my $i = $vwraysC{$rpictA[-1]}; $i-- > 0; ) {
70 push @vwraysA, $ARGV[0];
71 push(@vwrightA, $ARGV[0]) if ($isvopt);
72 push @rpictA, shift(@ARGV);
73 }
74 next OPTION;
75 }
76 # Check rtrace options
77 if (defined $rtraceC{$ARGV[0]}) {
78 push @rtraceA, $ARGV[0];
79 push @rpictA, shift(@ARGV);
80 for (my $i = $rtraceC{$rpictA[-1]}; $i-- > 0; ) {
81 push @rtraceA, $ARGV[0];
82 push @rpictA, shift(@ARGV);
83 }
84 next OPTION;
85 }
86 # Check options to ignore
87 if (defined $ignoreC{$ARGV[0]}) {
88 push @rpictA, shift(@ARGV);
89 for (my $i = $ignoreC{$rpictA[-1]}; $i-- > 0; ) {
90 push @rpictA, shift(@ARGV);
91 }
92 next OPTION;
93 }
94 # Check for output file or number of processes
95 if ("$ARGV[0]" eq '-o') {
96 shift @ARGV;
97 $outpic = shift(@ARGV);
98 } elsif ("$ARGV[0]" eq '-z') {
99 push @rpictA, shift(@ARGV);
100 $outzbf = $ARGV[0];
101 push @rpictA, shift(@ARGV);
102 } elsif ("$ARGV[0]" eq '-n') {
103 shift @ARGV;
104 $nprocs = shift(@ARGV);
105 } elsif ("$ARGV[0]" eq '-d') {
106 shift @ARGV;
107 $refDepth = ' -d ' . shift(@ARGV);
108 } elsif ("$ARGV[0]" =~ "$outpatt") {
109 push @rtraceA, $ARGV[0];
110 $outlyr = substr(shift(@ARGV), 2);
111 $outdir = shift(@ARGV);
112 } else {
113 die "Unsupported option: " . $ARGV[0] . "\n";
114 }
115 }
116 die "Number of processes must be positive" if ($nprocs <= 0);
117 if (defined $outdir) { # check conflicting options
118 die "Options -o and -o* are mutually exclusive\n" if (defined $outpic);
119 die "Options -z and -o* are mutually exclusive\n" if (defined $outzbf);
120 }
121 if (defined $outpic) { # redirect output?
122 die "File '$outpic' already exists\n" if (-e $outpic);
123 open STDOUT, '>', "$outpic";
124 }
125 #####################################################################
126 ##### May as well run rpict?
127 if ($nprocs == 1 && !defined($outdir)) {
128 push(@rpictA, $ARGV[0]) if ($#ARGV == 0);
129 exec @rpictA ;
130 }
131 # OK, we're running rtrace in some capacity...
132 push @rtraceA, ('-n', "$nprocs");
133 die "Need single octree argument\n" if ($#ARGV != 0);
134 my $oct = $ARGV[0];
135 my $view = `@vwrightA 0`;
136 chomp $view;
137 my @res = split(/\s/, `@vwraysA -d`);
138 #####################################################################
139 ##### Generating picture with depth buffer?
140 if (defined $outzbf) {
141 exec "@vwraysA | @rtraceA -fff -olv @res '$oct' | " .
142 "rsplit -ih -iH -f -of '$outzbf' -oh -oH -of3 - | " .
143 "pvalue -r -df | getinfo -a 'VIEW=$view'";
144 }
145 #####################################################################
146 ##### Base case with output picture only?
147 if (! defined $outdir) {
148 exec "@vwraysA | @rtraceA -ffc @res '$oct' | getinfo -a 'VIEW=$view'";
149 }
150 #####################################################################
151 ##### Layered image output case
152 my @rsplitA = ("rsplit", "'-t '", "-ih", "-iH", "-oh", "-oH");
153 # Supported rtrace -o* options and output file name.typ
154 my %rtoutC = (
155 v => 'radiance.hdr',
156 r => 'r_refl.hdr',
157 x => 'r_unrefl.hdr',
158 l => 'd_effective.dpt',
159 L => 'd_firstsurf.dpt',
160 R => 'd_refl.dpt',
161 X => 'd_unrefl.dpt',
162 n => 'perturbed.nrm',
163 N => 'unperturbed.nrm',
164 s => 'surface.idx',
165 m => 'modifier.idx',
166 M => 'material.idx'
167 );
168 # Arguments for rsplit based on output file type
169 my %rcodeC = (
170 '.hdr', ['-of3', '!pvalue -r -df -u'],
171 '.dpt', ['-of', "!rcode_depth$refDepth -ff"],
172 '.nrm', ['-of3', '!rcode_norm -ff'],
173 '.idx', ['-oa', '!rcode_ident "-t "']
174 );
175 if ($irrad) { # adjust actions for -i+ option
176 $rtoutC{'v'} = 'irradiance.hdr';
177 delete $rtoutC{'r'};
178 delete $rtoutC{'x'};
179 delete $rtoutC{'R'};
180 delete $rtoutC{'X'};
181 }
182 if (! -d $outdir) {
183 mkdir $outdir or die("Cannot create output directory '$outdir'\n");
184 }
185 # construct rsplit command
186 foreach my $oval (split //, $outlyr) {
187 die "Duplicate or unsupported type '$oval' in -o$outlyr\n"
188 if (!defined $rtoutC{$oval});
189 my $outfile = "$outdir/$rtoutC{$oval}";
190 die "File '$outfile' already exists\n" if (-e $outfile);
191 my ($otyp) = ($rtoutC{$oval} =~ /(\.[^.]+)$/);
192 push @rsplitA, $rcodeC{$otyp}[0];
193 push @rsplitA, qq{'$rcodeC{$otyp}[1] > "$outfile"'};
194 delete $rtoutC{$oval};
195 }
196 # call rtrace + rsplit
197 exec "@vwraysA | @rtraceA -fff @res '$oct' | getinfo -a 'VIEW=$view' | @rsplitA";