#!/usr/bin/perl

# Convert Radiance animation frames to TIFF output

#

# This is a re-write of Greg's ran2tiff.csh

#

# Axel, Oct 2010

use strict;

use warnings;

use File::Basename;

my $histosiz = 200;

my $histof = '';      # Histogram file (-H)

my $fwt = 0.9;        # Frame weight (-W)

my $outdir = '';      # Output directory for TIFFs (-D)

my @pcopts = ();      # Options passed through to pcond

my @tfopts = ();      # Options passed through to ra_tiff

if ($#ARGV < 0) {

	@ARGV = ('DUMMY');

}

# Process options for pcond and ra_tiff

while (@ARGV) {

	$_ = $ARGV[0];

	if (m/-W/) {   # Weight of each frame

		$pfwt = $ARGV[1];

		shift @ARGV;

	} elsif (m/-H/) {   # Histogram file

		$histof = "$ARGV[1]";

		shift @ARGV;

	} elsif (m/-D/) {   # Output directory for TIFFs

		die("$0: Directory $ARGV[1] does not exist.\n")

				unless (-d "$ARGV[1]");

		$outdir = "$argv[1]";

		shift @ARGV;

	} elsif ((m/-h\w+/) or (m/-a\w+/) or   # human, acuity

			(m/-v\w+/) or (m/-s\w+/) or    # veiling glare, contrast sensitiviy

			(m/-c\w+/) or (m/-l\w+/)) {    # mesopic, linear response

		push(@pcopts, $_);

	} elsif ((m/-u/) or (m/-d/) or (m/-f/)) {

		# Lmax of device, dynamic range of device, macbeth.cal file

		push(@pcopts, $_, $ARGV[1]);

		shift @ARGV;

	} elsif (m/-p/) {   # pcond -p (primaries)

		push(@pcopts, '-p');

		for (my $i=0; $i<6, $i++) {

			push(@pcopts, $ARGV[1]);

			shift @ARGV;

		}

	} elsif ((m/-z/) or (m/-b/) or (m/-w/)) {

		# LZW-compression, 8-bit greyscale, 16-bit

		push(@tfopts, $_);

	} elsif (m/-g/) {

		push(@tfopts, '-g', $_);   # Gamma value for ra_tiff

		shift @ARGV;

	} elsif (m/^-/) {

		die("$0: Bad option: $_\n");

	} else {

		last;

	}

	shift @ARGV;

}

die("$0: Need at least two frames.\n") unless ($#ARGV >= 1);

#echo Usage: "$0 [-W prev_frame_wt][-H histo][-D dir][pcond opts][ra_tiff opts] frame1 frame2 .."

my $td = tempdir( CLEANPU => 1 );

# Get shrunken image luminances

foreach (@ARGV) {

	my ($name, undef, undef) = fileparse($_);

	my $datf = "$td/$name.dat";

	my $cmd = "pfilt -1 -x 128 -y 128 -p 1 \"$_\"";

	$cmd .= " |pvalue -o -h -H -b -df";

	$cmd .= " |rcalc -if1 -e 'L=\$1*179;cond=L-1e-7;\$1=log10(L)' > $datf";

	system("$cmd");

}

# Get Min. and Max. log values

my $Lmin = `cat $td/*.dat | total -l | rcalc -e '\$1=\$1-.01'`;

chomp($Lmin);

my $Lmax = `cat $td/*.dat | total -u | rcalc -e '\$1=\$1+.01'`;

chomp($Lmax);

if ($histof) {

	if (-r $histof) then

		# Fix min/max and translate histogram

		set Lmin=`sed -n '1p' $histof | rcalc -e 'min(a,b):if(a-b,b,a);$1=min($1,'"$Lmin)"`

		set Lmax=`sed -n '$p' $histof | rcalc -e 'max(a,b):if(a-b,a,b);$1=max($1,'"$Lmax)"`

		tabfunc -i hfunc < $histof > $td/oldhist.cal

		cnt $histosiz \

			| rcalc -e "L10=$Lmin+($Lmax-$Lmin)/$histosiz"'*($1+.5)' \

				-f $td/oldhist.cal -e '$1=L10;$2=hfunc(L10)' \

			> $td/oldhisto.dat

	endif

}

foreach ($ARGV) {

	#inp = $_

	set datf="$inp:t"

	set datf="$td/$datf:r.dat"

	set outp="$inp:t"

	set outp="$outdir$outp:r.tif"

	endif

	histo $Lmin $Lmax $histosiz < $datf > $td/newhisto.dat

	if (-f $td/oldhisto.dat) then

		rlam $td/newhisto.dat $td/oldhisto.dat \

			| rcalc -e '$1=$1;$2=$2+$4*'$pfwt \

			> $td/histo.dat

	else

		mv $td/{new,}histo.dat

	endif

	pcond $pcopts -I $inp:q < $td/histo.dat \

		| ra_tiff $tfopts - $outp:q

	mv $td/{,old}histo.dat

}

if ($?histof) then

	cp -f $td/oldhisto.dat $histof

endif

#EOF