#! /usr/bin/perl

use strict;

if(@ARGV < 4 || @ARGV > 6) {
	die <<EOHELP;
  useage: create_3d_map.pl foo.txt x_step y_step z_step [z_max] [output_filename]

runs complete_data.pl foo.txt x_step y_step z_step > /tmp/input.data
creates /tmp/heat_plot.gnuplot
gnuplot < /tmp/heat_plot.gnuplot
moves /tmp/gnuplot.png to foo.png
EOHELP
}

# Complete the data
system("complete_data.pl $ARGV[0] $ARGV[1] $ARGV[2] $ARGV[3] > /tmp/input.data") == 0 or die $@;

# Create the gnuplot instructions
open(my $gp, ">", "/tmp/heat_plot.gnuplot");
print $gp <<PLOT;
set pm3d map
set terminal png truecolor size 800,600
set output "/tmp/gnuplot.png"
PLOT
if(defined($ARGV[4])) {
	my $z_max = $ARGV[4];
	print $gp qq{set zrange[0:${z_max}]\n};
}
print $gp qq{splot "/tmp/input.data" with pm3d};
close($gp);

# Run gnuplot
system("gnuplot", "/tmp/heat_plot.gnuplot") == 0 or die $@;

# Move the result
if($ARGV[5] =~ m/\S/) {
	system("cp", "/tmp/gnuplot.png", $ARGV[5]);
} elsif($ARGV[0] =~ m/(.*)\.[^\.]+/) {
	system("cp", "/tmp/gnuplot.png", "$1.png");
} else {
	print "can't determine the output file name!\nPlease try again.\n"
}

# Clean up
unlink("/tmp/input.data");
unlink("/tmp/gnuplot.png");
unlink("/tmp/heat_plot.gnuplot");