GeneticBot.py

#!/usr/bin/env python
import sys
from GameState import GameState
import ants
import utils
import Genes
# define a class with a do_turn method
# the Ants.run method will parse and update bot input
# it will also run the do_turn method for us
class GeneticBot:
	def __init__(self):
		sys.stderr = open('errors.txt', 'w')
	
	# This method is called in ants.py (can be removed)
	def do_setup(self, ants_instance):
		self.gs = GameState(ants_instance) # Keep track of important game state stuff
		# GENOME IS DEFINED HERE
		self.genome = [Genes.ExploreGene(), Genes.FoodGene()]
	
	# Called once a turn in ants.py, orders called from here
	def do_turn(self, ants_instance):
		sys.stderr.write('rem: ' + str(ants_instance.time_remaining()) + '\n')
		self.gs.updateAll(ants_instance)
		sys.stderr.write('rem (building af): ' + str(ants_instance.time_remaining()) + '\n')

		# Get the hill protectors to do their business
		# This is a necessary evil as it would be far messier to screw around with the Genes
		sys.stderr.write('rem (hill protection...): ' + str(ants_instance.time_remaining()) + '\n')
		self.gs.hp.protect(ants_instance, self.gs, self.genome) # does dispersion while staying near the hill

		afs = [] # see Utils.py -> AntForce
		# Build default ant force holders based on current ants
		for ant_loc in ants_instance.my_ants():
			# Do not do dispersions for the normal ants
			if ant_loc in self.gs.hp.getProtectors():
				continue
			row, col = ant_loc
			afs.append(utils.AntForce(row, col))
		sys.stderr.write('rem (expressing genome...): ' + str(ants_instance.time_remaining()) + '\n')

		# Express the genes (not related to hill protectors)
		for gene in self.genome:
			sys.stderr.write('rem (expressing gene {0}): '.format(gene.id) + str(ants_instance.time_remaining()) + '\n')
			df = gene.express(ants_instance, self.gs, afs)
			utils.logdf(gene.id, df, self.gs.turn)
		sys.stderr.write('rem (applying forces): ' + str(ants_instance.time_remaining()) + '\n')
		
		# Resolve the forces to produce actions
		for af in afs:
			af.applyForces(ants_instance, self.gs)
		sys.stderr.write('rem: ' + str(ants_instance.time_remaining()) + '\n')
		sys.stderr.flush()
				
		self.gs.tickTurn()

if __name__ == '__main__':
	# psyco will speed up python a little, but is not needed
	try:
		import psyco
		psyco.full()
	except ImportError:
		pass	
	try:
		# if run is passed a class with a do_turn method, it will do the work
		# this is not needed, in which case you will need to write your own
		# parsing function and your own game state class
		import cProfile
		cProfile.run('ants.Ants.run(GeneticBot())')
	except KeyboardInterrupt:
		print('ctrl-c, leaving ...')