behavior.py

import boardUtils, math, copy
from random import choice,randrange, shuffle

los = 5

moves = {

'left': (-1,0),
'right': (1,0),
'up': (0,1),
'down':(0,-1),
'left and up': (-1,1),
'right and up': (1,1),
'left and down': (-1,-1),
'right and down':(1,-1)

}
def dist(p1, p2):
    return math.sqrt((p2[0] - p1[0]) ** 2 + (p2[1] - p1[1]) ** 2)

def mario1(mario,toads,board):
    while True:
        move = choice(moves.values())
        
        if not( board[mario[0] + move[0]] [mario[1] + move[1]]==boardUtils.wall):
            mario[0] += move[0]
            mario[1] += move[1]
            break
    return mario

def mario2(mario,toads,board):
            closestToad = 0
            for toad in toads:
                if dist(mario,[toad['x'],toad['y']]) > closestToad:
                    closestToad = toad

            xdiff = mario[0]-closestToad['x']
            ydiff = mario[1]-closestToad['y']
            if abs(xdiff) < abs(ydiff):
                if xdiff > 0:
                    dx = 1
                    dy = 0
                else:
                    dx = -1
                    dy = 0
            elif abs(ydiff) < abs(xdiff):
                if ydiff > 0:
                    dy = 1
                    dx = 0
                else:
                    dy = -1
                    dx = 0
            else:
                move =  moves[choice(moves.keys())]
                dx = move[0]
                dy = move[1]
            while True:
                if not( board[mario[0] + dx][ mario[1] + dy]==boardUtils.wall):
                    mario[0] += dx
                    mario[1] += dy
                    break
                move =  moves[choice(moves.keys())]
                dx = move[0]
                dy = move[1]

            return mario

def mario3(mario,toads,board):

    randomizedMoves = moves.values()
    #print randomizedMoves
    for move in randomizedMoves:
        #print move[0]
        #print move[1]
        if not( board[mario[0] + move[0]] [mario[1] + move[1]]==boardUtils.wall):
            #print "Not Wall"
            for toad in toads:
                prevDist = dist([mario[0] + move[0], mario[1] + move[1]], [toad['x'],toad['y']])
                newDist = dist(mario,[toad['x'],toad['y']])
                if newDist < prevDist  :

                    mario[0] += move[0]
                    mario[1] += move[1]
                    return mario
    return mario




def toad1(mario,toads,board):
    for toad in toads:
        move = choice(moves.values())
        toad['x'] = move[0]
        toad['y'] = move[1]

def toad2(mario,toads,board):

    for toad in toads:
        toad['previousDistance'] = toad['distanceToMario']
        toad['distanceToMario'] = dist(mario,[toad['x'],toad['y']])

        if toad['previousDistance'] < toad['distanceToMario']:
            toad['direction'] = choice(moves.keys())
        toadMove = moves[toad['direction']]

        board[toad['x']][toad['y']]=boardUtils.toadTrail
        if not( board[toad['x'] + toadMove[0]][ toad['y'] + toadMove[1]]==boardUtils.wall):
            toad['x'] += toadMove[0]
            toad['y'] += toadMove[1]

    return toads

def toad3(mario,toads,board):
#Toad AI
    closestToad = toads[0]
    for toad in toads:
        if dist(mario,[toad['x'],toad['y']]) > closestToad:
            closestToad = toad

    for toad in toads:
        approxclose = closestToad['x']+randrange(-3,3),closestToad['y']+randrange(-3,3)
        possibleMoves = []
        toad['previousDistance'] = toad['distanceToMario']
        toad['distanceToMario'] = dist(mario,[toad['x'],toad['y']])
        #Start new collaboration code
        if toad['distanceToMario'] > los:
            if toad['id'] != closestToad['id']:

                for move in moves.keys():

                    x,y = moves[move][0],moves[move][1]

                    if dist([approxclose[0],approxclose[1]], [toad['x'],toad['y']]) >= dist([approxclose[0], approxclose[1]], [toad['x']+x,toad['y']+y]):
                        possibleMoves.append(move)
                try:
                    toad['direction'] = choice(possibleMoves)
                except:
                    toad['direction'] = choice(moves.keys())
            else:
                if toad['previousDistance'] < toad['distanceToMario']:
                    toad['direction'] = choice(moves.keys())
        elif toad['distanceToMario'] <= los:

            for move in moves.keys():
                x,y = moves[move][0], moves[move][1]
                if dist(mario, [toad['x']+x,toad['y']+y]) <= toad['previousDistance']:
                    possibleMoves.append(move)
            try:
                toad['direction'] = choice(possibleMoves)
            except:
                toad['direction'] = choice(moves.keys())
            #End new Collaboration code
        toadMove = moves[toad['direction']]

        board[toad['x']][toad['y']]=boardUtils.toadTrail
        if not( board[toad['x'] + toadMove[0]][ toad['y'] + toadMove[1]]==boardUtils.wall):
            toad['x'] += toadMove[0]
            toad['y'] += toadMove[1]
    return toads

marioMap = {1:mario1,2:mario2,3:mario3}
toadMap = {1: toad1,2:toad2,3:toad3}

def mario(mario,toads,board,behavior):
    f = marioMap[behavior](mario,toads,board)
    return f

def toad(mario,toads,board,behavior):
    f = toadMap[behavior](mario,toads,board)
    return f