#!/usr/bin/perl
use Chart::Gnuplot; 

#one must specify FILENAME to get data from (tab-delimited), a name for the output file, a type for output file (e.g. pdf, png, eps) and a plot title
($infile, $title, $type, $plot_title) = @ARGV;
$outfile = $title.'.'.$type;

#add or remove colours accordingly	
@colour_array = ("lblue", "brown", "blue", "lred", "lgreen", "yellow", "green", "red", "purple", "orange", "pink", "dyellow");

@data;
$counter;

open IN, "$infile" or die $!; 
$label_line = <IN>; chomp $label_line;	#find the labels data (for naming the colours in the stacked bars)
@split_labels = split(/\t/, $label_line);

while(<IN>)
{
	chomp; 
  	$counter = 0;
  	for (split/\t/,$_)
	{ push @{$data[$counter++]}, $_; }
}
close IN;

#x-axis labels is the row #0 in the AoA @data
@x = @{$data[0]};

$total_y = @data-1;	#find how many y rows we have => how many colours we need (-1 because #0 is x-axis)
@pick_colours = @colour_array[ 0 .. $total_y - 1 ];

# Initiate the chart object 
$chart = Chart::Gnuplot->new( 
	output => $outfile, 
	title => $title, 
	xlabel => { 
			text => 'Method', 
			color => 'black',
			offset => "-1",
			font => 'Times-Roman, 20'
		  },
	ylabel => { 
			text => $plot_title, 
			color => 'black',
			offset => -1,
			font => 'Times-Roman, 20'
		  },
	yrange => [0, '*'], 
	xtics => { rotate => '-270', font => 'Times-Roman, 15' },	
	ytics => { font => 'Times-Roman, 15' },	
	bg => {
    		color   => 'white',
    		density => 0.2,
	      },
	plotbg => 'white',
	legend => {position => 'outside bottom'}, 
	border => {
    			sides    => 'bottom, left',
    			linetype => 3,
    			width    => 2,
    			color    => 'black',
		  },

	"style histogram" => 'rowstacked'
); 

#all the rest rows in @data are used for creating the histogram bars
#create object foreach series of data (column in excel)
$counter_objects=0;
for $y ( 1 .. $#data ) 
{
	$legend_name = $split_labels[$y];	#get the name for the legend, e.g 1TM, 2TM, >2TM
	@tmp_y = @{$data[$y]};			#create temp array

	$counter_objects++;
	
	${y."$counter_objects"} = Chart::Gnuplot::DataSet->new
	( 
		xdata => \@x, 
		ydata => \@tmp_y, 
		title => "$legend_name", 
		fill => {density => 0.2}, 
		style => "histograms", 
	); 
}
#$chart->plot2d($h1, $h2, $h3, $h4);