ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/util/rtpict.pl
Revision: 2.10
Committed: Wed Aug 14 22:18:03 2019 UTC (4 years, 7 months ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.9: +7 -7 lines
Log Message:
Made it so output names don't collide when rcode2bmp run on all

File Contents

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