robocode-svm/robocode-svm

#!/usr/bin/env bash

set -u
#set -x

# File locations
dir=$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )
datadir="$dir/data"
results="$datadir/battle.data"
settings="$datadir/settings.last.battle"
training="$datadir/training.scale.data"
testing="$datadir/testing.scale.data"

# Battle parameters
rounds=6
width=500
height=500
robot1="sample.Corners"
robot2="sample.Fire"


# Iterate over arguments
while :; do
	# If there are any
	if [ $# -gt 0 ] ; then
		case $1 in
			# Regenerate data files
			-c|--clean)
			rm -rf "$datadir"
			;;
			# Call a "show_help" function to display a synopsis, then exit.
			-h|-\?|--help)   
			echo "Generate the data required for SVM classification and show the accuracy"
			echo "of the generated SVM model. If coordinates are supported, one visual battle"
			echo "is shown to support the prediction."
			echo ""
			echo "USAGE: robocode-svm [OPTION] [COORDINATES]"
			echo "  -v, --visualize	Shows visual run of the battle"
			echo "  -x			The x coordinate of the second robot, used with -v"
			echo "  -y			The y coordinate of the second robot, used with -v"
			echo "  -c, --clean		Regenerate the battle data (may take a very long time)"
			echo "  -h, --help		Show this help"
			exit
			;;
			# Unrecognized parameter
			-?*)
			printf 'WARNING: Unknown option (ignored): %s\n' "$1" >&2
			;;
			# Default case: If no more options then break out of the loop.
			*)               
			break
		esac
		# "command" reduces the chance of fatal errors in many shells.
		command shift        
	else
		break
	fi
done

# Prepare the data directory
mkdir -p "$datadir/"
touch "$results"

# If the data file has required amount if data
lines=$(wc -l < "$results")
if [[ "$lines" -eq "$rounds" ]] ; then
	# Calculate the lines needed to split the data to 90% and 10%
	tr=$(bc <<< "scale=0; $rounds * 0.9 / 1");
	te=$(bc <<< "scale=0; ($rounds - $tr) / 1");

	# Scale the results to interval <0, 1> and split them to training set and testing set;
	# misuse tee for 'process substituion' and send its stdout do /dev/null
	svm-scale -l 0 -u 1 "$results" | tee  >(head -n "$tr" > "$training") >(tail -n "$te" > "$testing") > /dev/null
# Generate the battle data (run 1000 battles) for a SVM classification
else
	# Repeat the battle desired number of times (till the battle data file does not contain $rounds entries
	echo "Generating the data for SVM predition. Depending on the machine, this may take very long time."
	delta=$((rounds - lines))
	for ((i = 1; i <= "$delta"; i++)) ; do
		# Generate input parameters
		number=$RANDOM; let "number %= $width"; x=$number
		number=$RANDOM; let "number %= $height"; y=$number
		number=$RANDOM; let "number %= 360"; alpha=$number

		# Write current settings to a file; first robot starts in the middle, with fixed gun angle;
		# the other one start at random known position, with random known gun angle
		{
			echo "#Battle Properties" 
			echo "robocode.battleField.width=$width" 
			echo "robocode.battleField.height=$height"
			echo "robocode.battle.numRounds=1"
			echo "robocode.battle.gunCoolingRate=0.07"
			echo "robocode.battle.rules.inactivityTime=450"
			echo "robocode.battle.selectedRobots=$robot1,$robot2"
			echo "robocode.battle.initialPositions=($((width/2)),$((height/2)),0),($x,$y,$alpha)" 
		} > "$settings"

		# Run battle without GUI, following the settings file, and returning the lines containing winners
		winner=$(robocode -nodisplay -battle "$settings" | tail -2| awk 'NR==1F {print $2}') 
		if [ "$winner" == "$robot1" ] ; then
			winner=1
		else
			winner=2
		fi

		# Append the result of the battle to the battle data file
		echo "$winner 1:$x 2:$y 3:$alpha" >> "$results"
		# Print a dot to infrom user tht something is happening
		printf "."
	done
	printf "\nData generation successful. Run the command again the show the accuracy.\n"
fi