Python Fights Code

# Python Fights online AI creation game
# Copyright (C) 2009  Maxim Sloyko
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

#!/usr/bin/python

from optparse import OptionParser
import cProfile
from core import *
from serialize import serializer

parser = OptionParser()
parser.add_option('-o', '--output',
    dest = 'output',
    default = 'results.prof',
    metavar = 'FILENAME',
    help = 'Where to write profiler\'s logs')

parser.add_option('-r', '--round-count',
    dest = 'rounds',
    type = 'int',
    default = 20,
    metavar = 'NUMBER',
    help = 'Number of rounds to play')

parser.add_option('-n', '--game-count',
    dest = 'games',
    type = 'int',
    default = 1,
    metavar = 'NUMBER',
    help = 'Number of games to play')

opt, args = parser.parse_args()

def main():
    py1 = Python('First')
    mslot1 = MemorySlot(py1)
    mslot1.set_cell(3, 3, MemoryCell(kind = MC_OWN_HEAD))
    mslot1.set_cell(3, 2, MemoryCell(kind = MC_ENEMY_TAIL))

    mslot = MemorySlot(py1)
    mslot.set_cell(3, 3, MemoryCell(kind = MC_OWN_HEAD))
    mslot.set_cell(3, 2, MemoryCell(kind = MC_ENEMY_TAIL, trait = MCT_OR))
    mslot.set_cell(3, 1, MemoryCell(kind = MC_ENEMY_TAIL, trait = MCT_OR))
    mslot.set_cell(3, 0, MemoryCell(kind = MC_ENEMY_TAIL, trait = MCT_OR))
    mslot.set_cell(4, 1, MemoryCell(kind = MC_ENEMY_TAIL, trait = MCT_OR))
    mslot.set_cell(4, 0, MemoryCell(kind = MC_ENEMY_TAIL, trait = MCT_OR))
    mslot.set_cell(5, 0, MemoryCell(kind = MC_ENEMY_TAIL, trait = MCT_OR))
    py1.memslot_list = [mslot1, mslot]

    py2 = Python('Second')
    py3 = Python('Third')

    for i in range(opt.games):
        g = Game([py1, py2, py3], round_num = opt.rounds)
        g.play()
        #moves = b.play()
        print "Score:"
        for py, score in g.get_score().iteritems():
            print '%s: %s' % (py.name, score)

        print serializer.save_game(g)
        #print "Moves:"
        #nmame_map = {MOVE_NORTH: 'N', MOVE_WEST: 'W', MOVE_EAST: 'E', MOVE_SOUTH: 'S'}
        #print ''.join([nmame_map.get(m, '-') for m in moves])

if opt.output:
    cProfile.run('main()', opt.output)
else:
    main()