#!/usr/bin/perl
#-------------------------------------------------------------------------------
#
# Plot (using R) 
# Data is feed as a 1 column of numbers 
#
# Note: Any line that does not match a numeric regular expression, is filtered out).
#
#														Pablo Cingolani
#-------------------------------------------------------------------------------

#-------------------------------------------------------------------------------
# Main
#-------------------------------------------------------------------------------

# Parse command line option (file base name)
$base = 'plot';
if( $ARGV[0] ne '' )	{ $base = $ARGV[0]; }
if( $ARGV[1] ne '' )	{ $label = $ARGV[1]; }
print "label:$label\n";

$pngFile = "$base.png";
$txtFile = "$base.txt";

# Read STDIN and create an R vector
open TXT, "> $txtFile" or die "Cannot open output file '$txtFile'\n";
print TXT "label\tx\n";
for( $ln = 0 ; $l = <STDIN> ; ) {
	chomp $l;

	# Does the string have a label and a number? (can be float)
	if( $l =~ /^(\S+)\s+([-+]?[0-9]*\.?[0-9]+([eE][-+]?[0-9]+)?)$/ ) { 
		print "$1\t$2\n"; 
		print TXT "$1\t$2\n"; 
	}
}
close TXT;

#---
# Create an R program, save plot as PNG image
#---

open R, "| R --vanilla --slave " or die "Cannot open R program\n";
print R <<EOF;

plotChart <- function( x, labels, title, label ) {
	keep <- (labels == label)
	cat('label:', label, '\t', sum(keep), '\n')
	plot(x, xaxt = "n", col=ifelse(keep, "red", "black"), pch=ifelse(keep, 15, 1))
	axis(1, at=1:length(x), labels=labels)

    # Mean & median calculated over the whola data
    abline( h=mean(x), col='blue', lty=2, lwd=2);
    abline( h=median(x), col='green', lty=2, lwd=2);

	legend("topright",c("Mean","Median"),lty=c(1,1),col=c("blue","green"))

}

png('$pngFile', width = 1024, height = 1024);

data <- read.csv("$txtFile", sep='\\t', header = TRUE);
x <- data\$x
labels <- data\$label
plotChart( x, labels, "All data", '$label' );
print( summary( x ) )

dev.off();
quit( save='no' )
EOF

close R;

#---
# Show figure
#---

$os = `uname`;
$show = "eog"; 
if( $os =~ "Darwin" )	{ $show = "open"; }
`$show $pngFile &`;