src/util/rtpict.pl

#!/usr/bin/perl -w
# RCSid $Id: rtpict.pl,v 2.10 2019/08/14 22:18:03 greg Exp $
#
# Run rtrace in parallel mode to simulate rpict -n option
# May also be used to render layered images with -o* option
#
#       G. Ward
#
use strict;
# we'll run rpict if no -n or -o* option
my $nprocs = 1;
# rtrace options and the associated number of arguments
my %rtraceC = ('-dt',1, '-dc',1, '-dj',1, '-ds',1, '-dr',1, '-dp',1,
                '-ss',1, '-st',1, '-e',1, '-am',1,
                '-ab',1, '-af',1, '-ai',1, '-aI',1, '-ae',1, '-aE',1,
                '-av',3, '-aw',1, '-aa',1, '-ar',1, '-ad',1, '-as',1,
                '-me',3, '-ma',3, '-mg',1, '-ms',1, '-lr',1, '-lw',1);
# boolean rtrace options
my @boolO = ('-w', '-bv', '-dv', '-i', '-u');
# view options and the associated number of arguments
my %vwraysC = ('-vf',1, '-vtv',0, '-vtl',0, '-vth',0, '-vta',0, '-vts',0, '-vtc',0,
                '-x',1, '-y',1, '-vp',3, '-vd',3, '-vu',3, '-vh',1, '-vv',1,
                '-vo',1, '-va',1, '-vs',1, '-vl',1, '-pa',1, '-pj',1, '-pd',1);
# options we need to silently ignore
my %ignoreC = ('-t',1, '-ps',1, '-pt',1, '-pm',1);
# Starting options for rtrace (rpict values)
my @rtraceA = split(' ', 'rtrace -u- -dt .05 -dc .5 -ds .25 -dr 1 ' .
                                '-aa .2 -ar 64 -ad 512 -as 128 -lr 7 -lw 1e-03');
my @vwraysA = ('vwrays', '-ff', '-pj', '.67');
my @vwrightA = ('vwright', '-vtv');
my @rpictA = ('rpict');
my $outpatt = '^-o[vrxlLRXnNsmM]+';
my $refDepth = "";
my $irrad = 0;
my $outlyr;
my $outdir;
my $outpic;
my $outzbf;
OPTION:                         # sort through options
while ($#ARGV >= 0 && "$ARGV[0]" =~ /^[-\@]/) {
        # Check for file inclusion
        if ("$ARGV[0]" =~ /^\@/) {
                open my $handle, '<', substr($ARGV[0], 1);
                shift @ARGV;
                chomp(my @args = <$handle>);
                close $handle;
                unshift @ARGV, split(/\s+/, "@args");
                next OPTION;
        }
        # Check booleans
        for my $boopt (@boolO) {
                if ("$ARGV[0]" =~ ('^' . $boopt . '[-+01tfynTFYN]?$')) {
                        if ("$ARGV[0]" eq '-i') {
                                $irrad = ! $irrad;
                        } elsif ("$ARGV[0]" =~ /^-i[-+01tfynTFYN]/) {
                                $irrad = ("$ARGV[0]" =~ /^-i[+1tyTY]/);
                        }
                        push @rtraceA, $ARGV[0];
                        push @rpictA, shift(@ARGV);
                        next OPTION;
                }
        }
        # Check view options
        if (defined $vwraysC{$ARGV[0]}) {
                push @vwraysA, $ARGV[0];
                my $isvopt = ("$ARGV[0]" =~ /^-v/);
                push(@vwrightA, $ARGV[0]) if ($isvopt);
                push @rpictA, shift(@ARGV);
                for (my $i = $vwraysC{$rpictA[-1]}; $i-- > 0; ) {
                        push @vwraysA, $ARGV[0];
                        push(@vwrightA, $ARGV[0]) if ($isvopt);
                        push @rpictA, shift(@ARGV);
                }
                next OPTION;
        }
        # Check rtrace options
        if (defined $rtraceC{$ARGV[0]}) {
                push @rtraceA, $ARGV[0];
                push @rpictA, shift(@ARGV);
                for (my $i = $rtraceC{$rpictA[-1]}; $i-- > 0; ) {
                        push @rtraceA, $ARGV[0];
                        push @rpictA, shift(@ARGV);
                }
                next OPTION;
        }
        # Check options to ignore
        if (defined $ignoreC{$ARGV[0]}) {
                push @rpictA, shift(@ARGV);
                for (my $i = $ignoreC{$rpictA[-1]}; $i-- > 0; ) {
                        push @rpictA, shift(@ARGV);
                }
                next OPTION;
        }
        # Check for output file or number of processes
        if ("$ARGV[0]" eq '-o') {
                shift @ARGV;
                $outpic = shift(@ARGV);
        } elsif ("$ARGV[0]" eq '-z') {
                push @rpictA, shift(@ARGV);
                $outzbf = $ARGV[0];
                push @rpictA, shift(@ARGV);
        } elsif ("$ARGV[0]" eq '-n') {
                shift @ARGV;
                $nprocs = shift(@ARGV);
        } elsif ("$ARGV[0]" eq '-d') {
                shift @ARGV;
                $refDepth = ' -d ' . shift(@ARGV);
        } elsif ("$ARGV[0]" =~ "$outpatt") {
                push @rtraceA, $ARGV[0];
                $outlyr = substr(shift(@ARGV), 2);
                $outdir = shift(@ARGV);
        } else {
                die "Unsupported option: " . $ARGV[0] . "\n";
        }
}
die "Number of processes must be positive" if ($nprocs <= 0);
if (defined $outdir) {          # check conflicting options
        die "Options -o and -o* are mutually exclusive\n" if (defined $outpic);
        die "Options -z and -o* are mutually exclusive\n" if (defined $outzbf);
}
if (defined $outpic) {          # redirect output?
        die "File '$outpic' already exists\n" if (-e $outpic);
        open STDOUT, '>', "$outpic";
}
#####################################################################
##### May as well run rpict?
if ($nprocs == 1 && !defined($outdir)) {
        push(@rpictA, $ARGV[0]) if ($#ARGV == 0);
        exec @rpictA ;
}
# OK, we're running rtrace in some capacity...
push @rtraceA, ('-n', "$nprocs");
die "Need single octree argument\n" if ($#ARGV != 0);
my $oct = $ARGV[0];
my $view = `@vwrightA 0`;
chomp $view;
my @res = split(/\s/, `@vwraysA -d`);
#####################################################################
##### Generating picture with depth buffer?
if (defined $outzbf) {
        exec "@vwraysA | @rtraceA -fff -olv @res $oct | " .
                "rsplit -ih -iH -f -of $outzbf -oh -oH -of3 - | " .
                "pvalue -r -df | getinfo -a 'VIEW=$view'";
}
#####################################################################
##### Base case with output picture only?
if (! defined $outdir) {
        exec "@vwraysA | @rtraceA -ffc @res $oct | getinfo -a 'VIEW=$view'";
}
#####################################################################
##### Layered image output case
my @rsplitA = ("rsplit", "'-t   '", "-ih", "-iH", "-oh", "-oH");
# Supported rtrace -o* options and output file name.typ
my %rtoutC = (
        v =>    'radiance.hdr',
        r =>    'r_refl.hdr',
        x =>    'r_unrefl.hdr',
        l =>    'd_effective.dpt',
        L =>    'd_firstsurf.dpt',
        R =>    'd_refl.dpt',
        X =>    'd_unrefl.dpt',
        n =>    'perturbed.nrm',
        N =>    'unperturbed.nrm',
        s =>    'surface.idx',
        m =>    'modifier.idx',
        M =>    'material.idx'
);
# Arguments for rsplit based on output file type
my %rcodeC = (
        '.hdr', ['-of3', '!pvalue -r -df -u'],
        '.dpt', ['-of', "!rcode_depth$refDepth -ff"],
        '.nrm', ['-of3', '!rcode_norm -ff'],
        '.idx', ['-oa', '!rcode_ident "-t       "']
);
if ($irrad) {           # adjust actions for -i+ option
        $rtoutC{'v'} = 'irradiance.hdr';
        delete $rtoutC{'r'};
        delete $rtoutC{'x'};
        delete $rtoutC{'R'};
        delete $rtoutC{'X'};
}
if (! -d $outdir) {
        mkdir $outdir or die("Cannot create output directory '$outdir'\n");
}
                        # construct rsplit command
foreach my $oval (split //, $outlyr) {
        die "Duplicate or unsupported type '$oval' in -o$outlyr\n"
                                        if (!defined $rtoutC{$oval});
        my ($otyp) = ($rtoutC{$oval} =~ /(\.[^.]+)$/);
        push @rsplitA, $rcodeC{$otyp}[0];
        push @rsplitA, qq{'$rcodeC{$otyp}[1] > "$outdir/$rtoutC{$oval}"'};
        delete $rtoutC{$oval};
}
                        # call rtrace + rsplit
exec "@vwraysA | @rtraceA -fff @res $oct | getinfo -a 'VIEW=$view' | @rsplitA";
Revision:	2.11
Committed:	Wed Aug 14 22:57:24 2019 UTC (6 years, 2 months ago) by greg
Content type:	text/plain
Branch:	MAIN
Changes since 2.10:	+3 -5 lines
Log Message:	Simplified -o with -z slightly
#	User	Rev	Content
1	greg	2.1	#!/usr/bin/perl -w
2	greg	2.11	# RCSid $Id: rtpict.pl,v 2.10 2019/08/14 22:18:03 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	greg	2.11	"rsplit -ih -iH -f -of $outzbf -oh -oH -of3 - \| " .
143			"pvalue -r -df \| getinfo -a 'VIEW=$view'";
144	greg	2.5	}
145	greg	2.9	#####################################################################
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	greg	2.10	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	greg	2.9	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 ($otyp) = ($rtoutC{$oval} =~ /(\.[^.]+)$/);
190			push @rsplitA, $rcodeC{$otyp}[0];
191			push @rsplitA, qq{'$rcodeC{$otyp}[1] > "$outdir/$rtoutC{$oval}"'};
192			delete $rtoutC{$oval};
193			}
194			# call rtrace + rsplit
195			exec "@vwraysA \| @rtraceA -fff @res $oct \| getinfo -a 'VIEW=$view' \| @rsplitA";