pe

#!/bin/bash

# Program Name: pe
# Title: Program Evaluator

# Purpose: Evalute the program for given input and compare the results with the required output. This script is written
# keeping Compettive Programming Competitions in mind.

#-------------------------------------------------------------------------------------------------------------------------

# Colors
red='\033[37;41m'
brown='\033[37;43m'
green='\033[37;42m'


# Variables
program=
input=
output=

# Creating tmp directory in $HOME if it does not exists
[[ -d $HOME/tmp ]] || mkdir $HOME/tmp

# Creating tmp file in $HOME/tmp directory
programOutput=$(mktemp $HOME/tmp/pe.$$.XXXXXXXXXXXXXXXXXX)


#Functions
usage () {
	cat <<- eof
	
	Program Name: pe
	Title: Program Evaluator
	Usage: pe [ -i ] PROGRAM INPUT_FILE OUTPUT_FILE

	Purpose: Evalute the program for given input and compare the results with the required output. This script is written
	keeping Competitive Programming Competitions in mind.

	Options:

	-i		Show incorrect results only
	-h		Show help

	eof
}

# debug function
log () {
	echo "Debug: $1"
}

# validating correct number of positional parameters and assigning values

if [[ "$1" == "-h" ]]; then
	usage
	exit 0;
fi

if [[ "$#" == "4" ]]; then

	if [[ "$1" == "-i" ]]; then
		incorrectOnly=1
	else
		usage 1>&2
		exit 1
	fi

	program=$2
	input=$3
	output=$4
elif [[ "$#" == "3" ]]; then
	
	program=$1
	input=$2
	output=$3

else
	usage 1>&2
	exit 1
fi

# validating variables and their respective permissions

if [[ ! -x $program ]]; then
	echo "pe: Cannot execute '$program' or file does not exists" 1>&2
	exit 2 
fi

if [[ ! -r $input ]]; then
	echo "pe: Cannot read '$input' or file does not exists" 1>&2
	exit 3
fi

if [[ ! -r $output ]]; then
	echo "pe: Cannot read '$output' or file does not exists" 1>&2
	exit 4
fi

if [[ ! -w $programOutput ]]; then
	##log "cant create tmp output file"
	exit -1
fi

# Executing the PROGRAM while redirecting input and output to respective files

$program < $input > $programOutput

# comparing output files and displaying output

lines=$(wc -l $input)
lines=${lines%%" $input"}

echo -e "\n${brown}Input\tOutput\tProgram Output\033[0m"

for (( n = 0; n < $lines; ++n));
do
	IFS='' read -r i <&3
	IFS='' read -r o <&4
	IFS='' read -r po <&5 

	color=

	if [[ "${o}" == "${po}" ]]; then
		color=$green

		if [[ $incorrectOnly == 1 ]]; then
			continue
		fi
	else
		color=$red
	fi

	printf "${color}%5s\t%6s\t%14s\033[0m\n" $i $o $po

done 3<$input 4<$output 5<$programOutput

echo -e "\033[0m"

# Removing temp file
rm $progamOutput