1 |
#!/usr/bin/perl -w |
2 |
# RCSid $Id: falsecolor.pl,v 2.12 2017/09/25 18:48:11 greg Exp $ |
3 |
|
4 |
use warnings; |
5 |
use strict; |
6 |
use File::Temp qw/ tempdir /; |
7 |
use POSIX qw/ floor /; |
8 |
|
9 |
my @palettes = ('def', 'spec', 'pm3d', 'hot', 'eco'); |
10 |
|
11 |
my $mult = 179.0; # Multiplier. Default W/sr/m2 -> cd/m2 |
12 |
my $label = 'cd/m2'; # Units shown in legend |
13 |
my $scale = 1000; # Top of the scale |
14 |
my $scaledigits = 3; # Number of maximum digits for numbers in legend |
15 |
my $decades = 0; # Default is linear mapping |
16 |
my $pal = 'def'; # Palette |
17 |
my $redv = "${pal}_red(v)"; # Mapping functions for R,G,B |
18 |
my $grnv = "${pal}_grn(v)"; |
19 |
my $bluv = "${pal}_blu(v)"; |
20 |
my $ndivs = 8; # Number of lines in legend |
21 |
my $picture = '-'; |
22 |
my $cpict = ''; |
23 |
my $legwidth = 100; # Legend width and height |
24 |
my $legheight = 200; |
25 |
my $haszero = 1; # print 0 in scale for falsecolor images only |
26 |
my $docont = ''; # Contours: -cl and -cb |
27 |
my $doposter = 0; # Posterization: -cp |
28 |
my $doextrem = 0; # Don't mark extrema |
29 |
my $needfile = 0; |
30 |
my $showpal = 0; # Show availabel colour palettes |
31 |
|
32 |
while ($#ARGV >= 0) { |
33 |
$_ = shift; |
34 |
# Options with qualifiers |
35 |
if (m/-lw/) { # Legend width |
36 |
$legwidth = shift; |
37 |
} elsif (m/-lh/) { # Legend height |
38 |
$legheight = shift; |
39 |
} elsif (m/-m/) { # Multiplier |
40 |
$mult = shift; |
41 |
} elsif (m/-spec/) { |
42 |
die("depricated option '-spec'. Please use '-pal spec' instead."); |
43 |
} elsif (m/-s/) { # Scale |
44 |
$scale = shift; |
45 |
if ($scale =~ m/[aA].*/) { |
46 |
$needfile = 1; |
47 |
} |
48 |
} elsif (m/-d/) { # Number of maximum digits for numbers in legend |
49 |
$scaledigits = shift; |
50 |
} elsif (m/-l$/) { # Label |
51 |
$label = shift; |
52 |
} elsif (m/-log/) { # Logarithmic mapping |
53 |
$decades = shift; |
54 |
} elsif (m/-r/) { # Custom palette functions for R,G,B |
55 |
$redv = shift; |
56 |
} elsif (m/-g/) { |
57 |
$grnv = shift; |
58 |
} elsif (m/-b/) { |
59 |
$bluv = shift; |
60 |
} elsif (m/-pal$/) { # Color palette |
61 |
$pal = shift; |
62 |
if (! grep $_ eq $pal, @palettes) { |
63 |
die("invalid palette '$pal'.\n"); |
64 |
} |
65 |
$redv = "${pal}_red(v)"; |
66 |
$grnv = "${pal}_grn(v)"; |
67 |
$bluv = "${pal}_blu(v)"; |
68 |
} elsif (m/-i$/) { # Image for intensity mapping |
69 |
$picture = shift; |
70 |
} elsif (m/-p$/) { # Image for background |
71 |
$cpict = shift; |
72 |
} elsif (m/-ip/ || m/-pi/) { |
73 |
$picture = shift; |
74 |
$cpict = $picture; |
75 |
} elsif (m/-n/) { # Number of contour lines |
76 |
$ndivs = shift; |
77 |
|
78 |
# Switches |
79 |
} elsif (m/-cl/) { # Contour lines |
80 |
$docont = 'a'; |
81 |
$haszero = 0; |
82 |
} elsif (m/-cb/) { # Contour bands |
83 |
$docont = 'b'; |
84 |
$haszero = 0; |
85 |
} elsif (m/-cp/) { # Posterize |
86 |
$doposter = 1; |
87 |
$haszero = 0; |
88 |
} elsif (m/-palettes/) { # Show all available palettes |
89 |
$scale = 45824; # 256 * 179 |
90 |
$showpal = 1; |
91 |
} elsif (m/-e/) { |
92 |
$doextrem = 1; |
93 |
$needfile = 1; |
94 |
# Oops! Illegal option |
95 |
} else { |
96 |
die("bad option \"$_\"\n"); |
97 |
} |
98 |
} |
99 |
|
100 |
if (($legwidth <= 20) || ($legheight <= 40)) { |
101 |
# Legend is too small to be legible. Don't bother doing one. |
102 |
$legwidth = 0; |
103 |
$legheight = 0; |
104 |
} |
105 |
|
106 |
# Temporary directory. Will be removed upon successful program exit. |
107 |
my $td = tempdir( CLEANUP => 1 ); |
108 |
|
109 |
if ($needfile == 1 && $picture eq '-') { |
110 |
# Pretend that $td/stdin.rad is the actual filename. |
111 |
$picture = "$td/stdin.hdr"; |
112 |
open(FHpic, ">$picture") or |
113 |
die("Unable to write to file $picture\n"); |
114 |
# Input is from STDIN: Capture to file. |
115 |
while (<>) { |
116 |
print FHpic; |
117 |
} |
118 |
close(FHpic); |
119 |
|
120 |
if ($cpict eq '-') { |
121 |
$cpict = "$td/stdin.hdr"; |
122 |
} |
123 |
} |
124 |
|
125 |
# Find a good scale for auto mode. |
126 |
if ($scale =~ m/[aA].*/) { |
127 |
my @histo = split(/\s/, `phisto $picture| tail -2`); |
128 |
# e.g. $ phisto tests/richmond.hdr| tail -2 |
129 |
# 3.91267 14 |
130 |
# 3.94282 6 |
131 |
my $LogLmax = $histo[0]; |
132 |
$scale = $mult / 179 * 10**$LogLmax; |
133 |
} |
134 |
|
135 |
if ($docont ne '') { |
136 |
# -cl -> $docont = a |
137 |
# -cb -> $docont = b |
138 |
my $newv = join( "(v-1/ndivs)*ndivs/(ndivs-1)", split("v", $redv) ); |
139 |
$redv = $newv; |
140 |
$newv = join( "(v-1/ndivs)*ndivs/(ndivs-1)", split("v", $bluv) ); |
141 |
$bluv = $newv; |
142 |
$newv = join( "(v-1/ndivs)*ndivs/(ndivs-1)", split("v", $grnv) ); |
143 |
$grnv = $newv; |
144 |
} elsif ($doposter == 1) { |
145 |
# -cp -> $doposter = 1 |
146 |
my $newv = join( "seg2(v)", split("v", $redv) ); |
147 |
$redv = $newv; |
148 |
$newv = join( "seg2(v)", split("v", $bluv) ); |
149 |
$bluv = $newv; |
150 |
$newv = join( "seg2(v)", split("v", $grnv) ); |
151 |
$grnv = $newv; |
152 |
} |
153 |
|
154 |
my $pc0 = "$td/pc0.cal"; |
155 |
open(FHpc0, ">$pc0"); |
156 |
print FHpc0 <<EndOfPC0; |
157 |
PI : 3.14159265358979323846 ; |
158 |
scale : $scale ; |
159 |
mult : $mult ; |
160 |
ndivs : $ndivs ; |
161 |
gamma : 2.2; |
162 |
|
163 |
or(a,b) : if(a,a,b); |
164 |
EPS : 1e-7; |
165 |
neq(a,b) : if(a-b-EPS,1,b-a-EPS); |
166 |
btwn(a,x,b) : if(a-x,-1,b-x); |
167 |
clip(x) : if(x-1,1,if(x,x,0)); |
168 |
frac(x) : x - floor(x); |
169 |
boundary(a,b) : neq(floor(ndivs*a),floor(ndivs*b)); |
170 |
|
171 |
spec_red(x) = 1.6*x - .6; |
172 |
spec_grn(x) = if(x-.375, 1.6-1.6*x, 8/3*x); |
173 |
spec_blu(x) = 1 - 8/3*x; |
174 |
|
175 |
pm3d_red(x) = sqrt(clip(x)) ^ gamma; |
176 |
pm3d_grn(x) = clip(x*x*x) ^ gamma; |
177 |
pm3d_blu(x) = clip(sin(2*PI*clip(x))) ^ gamma; |
178 |
|
179 |
hot_red(x) = clip(3*x) ^ gamma; |
180 |
hot_grn(x) = clip(3*x - 1) ^ gamma; |
181 |
hot_blu(x) = clip(3*x - 2) ^ gamma; |
182 |
|
183 |
eco_red(x) = clip(2*x) ^ gamma; |
184 |
eco_grn(x) = clip(2*(x-0.5)) ^ gamma; |
185 |
eco_blu(x) = clip(2*(0.5-x)) ^ gamma; |
186 |
|
187 |
interp_arr2(i,x,f):(i+1-x)*f(i)+(x-i)*f(i+1); |
188 |
interp_arr(x,f):if(x-1,if(f(0)-x,interp_arr2(floor(x),x,f),f(f(0))),f(1)); |
189 |
|
190 |
def_redp(i):select(i,0.18848,0.05468174, |
191 |
0.00103547,8.311144e-08,7.449763e-06,0.0004390987,0.001367254, |
192 |
0.003076,0.01376382,0.06170773,0.1739422,0.2881156,0.3299725, |
193 |
0.3552663,0.372552,0.3921184,0.4363976,0.6102754,0.7757267, |
194 |
0.9087369,1,1,0.9863); |
195 |
def_red(x):interp_arr(x/0.0454545+1,def_redp); |
196 |
|
197 |
def_grnp(i):select(i,0.0009766,2.35501e-05, |
198 |
0.0008966244,0.0264977,0.1256843,0.2865799,0.4247083,0.4739468, |
199 |
0.4402732,0.3671876,0.2629843,0.1725325,0.1206819,0.07316644, |
200 |
0.03761026,0.01612362,0.004773749,6.830967e-06,0.00803605, |
201 |
0.1008085,0.3106831,0.6447838,0.9707); |
202 |
def_grn(x):interp_arr(x/0.0454545+1,def_grnp); |
203 |
|
204 |
def_blup(i):select(i,0.2666,0.3638662,0.4770437, |
205 |
0.5131397,0.5363797,0.5193677,0.4085123,0.1702815,0.05314236, |
206 |
0.05194055,0.08564082,0.09881395,0.08324373,0.06072902, |
207 |
0.0391076,0.02315354,0.01284458,0.005184709,0.001691774, |
208 |
2.432735e-05,1.212949e-05,0.006659406,0.02539); |
209 |
def_blu(x):interp_arr(x/0.0454545+1,def_blup); |
210 |
|
211 |
isconta = if(btwn(1/ndivs/2,v,1+1/ndivs/2),or(boundary(vleft,vright),boundary(vabove,vbelow)),-1); |
212 |
iscontb = if(btwn(1/ndivs/2,v,1+1/ndivs/2),-btwn(.1,frac(ndivs*v),.9),-1); |
213 |
|
214 |
seg(x)=(floor(v*ndivs)+.5)/ndivs; |
215 |
seg2(x)=(seg(x)-1/ndivs)*ndivs/(ndivs-1); |
216 |
|
217 |
ra = 0; |
218 |
ga = 0; |
219 |
ba = 0; |
220 |
|
221 |
in = 1; |
222 |
|
223 |
ro = if(in,clip($redv),ra); |
224 |
go = if(in,clip($grnv),ga); |
225 |
bo = if(in,clip($bluv),ba); |
226 |
EndOfPC0 |
227 |
close FHpc0; |
228 |
|
229 |
my $pc1 = "$td/pc1.cal"; |
230 |
open(FHpc1, ">$pc1"); |
231 |
print FHpc1 <<EndOfPC1; |
232 |
norm : mult/scale/le(1); |
233 |
|
234 |
v = map(li(1)*norm); |
235 |
|
236 |
vleft = map(li(1,-1,0)*norm); |
237 |
vright = map(li(1,1,0)*norm); |
238 |
vabove = map(li(1,0,1)*norm); |
239 |
vbelow = map(li(1,0,-1)*norm); |
240 |
|
241 |
map(x) = x; |
242 |
|
243 |
|
244 |
ra = ri(nfiles); |
245 |
ga = gi(nfiles); |
246 |
ba = bi(nfiles); |
247 |
EndOfPC1 |
248 |
close FHpc1; |
249 |
|
250 |
my $pc0args = "-f $pc0"; |
251 |
my $pc1args = "-f $pc1"; |
252 |
|
253 |
if ($showpal == 1) { |
254 |
my $pc = "pcompos -a 1"; |
255 |
foreach my $pal (@palettes) { |
256 |
my $fcimg = "$td/$pal.hdr"; |
257 |
my $lbimg = "$td/${pal}_label.hdr"; |
258 |
system "psign -cb 0 0 0 -cf 1 1 1 -h 20 $pal > $lbimg"; |
259 |
|
260 |
my $cmd = qq[pcomb $pc0args -e "v=x/256"]; |
261 |
$cmd .= qq[ -e "ro=clip(${pal}_red(v));go=clip(${pal}_grn(v));bo=clip(${pal}_blu(v))"]; |
262 |
$cmd .= qq[ -x 256 -y 30 > $fcimg]; |
263 |
system "$cmd"; |
264 |
$pc .= " $fcimg $lbimg"; |
265 |
} |
266 |
system "$pc"; |
267 |
exit 0; |
268 |
} |
269 |
|
270 |
# Contours |
271 |
if ($docont ne '') { |
272 |
# -cl -> $docont = a |
273 |
# -cb -> $docont = b |
274 |
$pc0args .= qq[ -e "in=iscont$docont"]; |
275 |
} |
276 |
|
277 |
if ($cpict eq '') { |
278 |
$pc1args .= qq[ -e "ra=0;ga=0;ba=0"]; |
279 |
} elsif ($cpict eq $picture) { |
280 |
$cpict = ''; |
281 |
} |
282 |
|
283 |
# Logarithmic mapping |
284 |
if ($decades > 0) { |
285 |
$pc1args .= qq[ -e "map(x)=if(x-10^-$decades,log10(x)/$decades+1,0)"]; |
286 |
} |
287 |
|
288 |
# Colours in the legend |
289 |
my $scolpic = "$td/scol.hdr"; |
290 |
|
291 |
# Labels in the legend |
292 |
my $slabpic = "$td/slab.hdr"; |
293 |
my $cmd; |
294 |
if ($legwidth > 0) { |
295 |
# Legend: Create the text labels |
296 |
my $sheight = floor($legheight / $ndivs ); |
297 |
my $theight = floor($legwidth/(8/1.67)); |
298 |
my $stheight = $sheight <= $theight ? $sheight : $theight; |
299 |
my $vlegheight = $sheight * $ndivs * (1+1.5/$ndivs); |
300 |
my $text = "$label\n"; |
301 |
my $tslabpic = "$td/slabT.hdr"; |
302 |
open PSIGN, "| psign -s -.15 -cf 1 1 1 -cb 0 0 0 -h $stheight > $tslabpic"; |
303 |
print PSIGN "$text"; |
304 |
close PSIGN; |
305 |
my $loop = $ndivs+$haszero; |
306 |
my $hlegheight = $sheight * ($loop) + $sheight * .5; |
307 |
my $pcompost = qq[pcompos -b 0 0 0 =-0 $tslabpic 0 $hlegheight ]; |
308 |
for (my $i=0; $i<$loop; $i++) { |
309 |
my $imap = ($ndivs - $i) / $ndivs; |
310 |
my $value = $scale; |
311 |
if ($decades > 0) { |
312 |
$value *= 10**(($imap - 1) * $decades); |
313 |
} else { |
314 |
$value *= $imap; |
315 |
} |
316 |
# Have no more than 3 decimal places |
317 |
$value =~ s/(\.[0-9]{$scaledigits})[0-9]*/$1/; |
318 |
$text = "$value\n"; |
319 |
$tslabpic = "$td/slab$i.hdr"; |
320 |
open PSIGN, "| psign -s -.15 -cf 1 1 1 -cb 0 0 0 -h $stheight > $tslabpic"; |
321 |
print PSIGN "$text"; |
322 |
close PSIGN; |
323 |
$hlegheight = $sheight * ($loop - $i - 1) + $sheight * .5; |
324 |
$pcompost .= qq[=-0 $tslabpic 0 $hlegheight ]; |
325 |
} |
326 |
$pcompost .= qq[ > $slabpic]; |
327 |
system $pcompost; |
328 |
|
329 |
# Legend: Create the background colours |
330 |
$cmd = qq[pcomb $pc0args]; |
331 |
$cmd .= qq[ -x $legwidth -y $vlegheight]; |
332 |
$cmd .= qq[ -e "v=(y+.5-$sheight)/(yres/(1+1.5/$ndivs));vleft=v;vright=v"]; |
333 |
$cmd .= qq[ -e "vbelow=(y-.5-$sheight)/(yres/(1+1.5/$ndivs));vabove=(y+1.5-$sheight)/(yres/(1+1.5/$ndivs))"]; |
334 |
$cmd .= qq[ -e "ra=0;ga=0;ba=0;"]; |
335 |
$cmd .= qq[ > $scolpic]; |
336 |
system $cmd; |
337 |
} else { |
338 |
# Create dummy colour scale and legend labels so we don't |
339 |
# need to change the final command line. |
340 |
open(FHscolpic, ">$scolpic"); |
341 |
print FHscolpic "\n-Y 1 +X 1\naaa\n"; |
342 |
close(FHscolpic); |
343 |
open(FHslabpic, ">$slabpic"); |
344 |
print FHslabpic "\n-Y 1 +X 1\naaa\n"; |
345 |
close(FHslabpic); |
346 |
} |
347 |
|
348 |
# Legend: Invert the text labels (for dropshadow) |
349 |
my $slabinvpic = "$td/slabinv.hdr"; |
350 |
$cmd = qq[pcomb -e "lo=1-gi(1)" $slabpic > $slabinvpic]; |
351 |
system $cmd; |
352 |
|
353 |
|
354 |
my $sh0 = -floor($legheight / $ndivs / 2); |
355 |
if ($haszero < 1) { |
356 |
$sh0 = -floor($legheight / ($ndivs)*1.5); |
357 |
} |
358 |
|
359 |
# Command line without extrema |
360 |
|
361 |
$cmd = qq[pcomb $pc0args $pc1args "$picture"]; |
362 |
$cmd .= qq[ "$cpict"] if ($cpict); |
363 |
$cmd .= qq[ | pcompos -b 0 0 0 $scolpic 0 $sh0 +t .1 $slabinvpic 2 -1 ]; |
364 |
$cmd .= qq[ -t .5 $slabpic 0 0 - $legwidth 0]; |
365 |
|
366 |
if ($doextrem == 1) { |
367 |
# Get min/max image luminance |
368 |
my $cmd1 = 'pextrem -o ' . $picture; |
369 |
my $retval = `$cmd1`; |
370 |
# e.g. |
371 |
# x y r g b |
372 |
# 193 207 3.070068e-02 3.118896e-02 1.995850e-02 |
373 |
# 211 202 1.292969e+00 1.308594e+00 1.300781e+00 |
374 |
|
375 |
my @extrema = split(/\s/, $retval); |
376 |
my ($lxmin, $ymin, $rmin, $gmin, $bmin, $lxmax, $ymax, $rmax, $gmax, $bmax) = @extrema; |
377 |
$lxmin += $legwidth; |
378 |
$lxmax += $legwidth; |
379 |
|
380 |
# Weighted average of R,G,B |
381 |
my $minpos = "$lxmin $ymin"; |
382 |
my $minval = ($rmin * .27 + $gmin * .67 + $bmin * .06) * $mult; |
383 |
$minval =~ s/(\.[0-9]{$scaledigits})[0-9]*/$1/; |
384 |
my $maxval = ($rmax * .27 + $gmax * .67 + $bmax * .06) * $mult; |
385 |
$maxval =~ s/(\.[0-9]{$scaledigits})[0-9]*/$1/; |
386 |
|
387 |
# Create the labels for min/max intensity |
388 |
my $minvpic = "$td/minv.hdr"; |
389 |
system "psign -s -.15 -a 2 -h 16 $minval > $minvpic"; |
390 |
my $maxvpic = "$td/maxv.hdr"; |
391 |
system "psign -s -.15 -a 2 -h 16 $maxval > $maxvpic"; |
392 |
|
393 |
# Add extrema labels to command line |
394 |
$cmd .= qq[ $minvpic $minpos $maxvpic $lxmax $ymax]; |
395 |
} |
396 |
|
397 |
|
398 |
|
399 |
|
400 |
# Process image and combine with legend |
401 |
system "$cmd"; |
402 |
|
403 |
#EOF |