def main():
with open(sys.argv[1]) as file:
data = json.load(file)
# TODO: find and print winning action sequence
print_utils.print_board(make_board_dict(data))
# make the initial state and the initial node
init_state = ai.State(make_state_dict(data, WHITE_PIECE), make_state_dict(data, BLACK_PIECE))
init_node = ai.Node(init_state)
# print the winning move sequence
print(ai.get_winning_sequence(init_node))
clock = pygame.time.Clock()
i_lastClicked = -1
j_lastClicked = -1
sym_lastClicked = ""
madeMove = True
blue_moves = game.blue_moves
initial_frame = False
human_before = None
ai_before = None
human_time = None
current_state = ai.State(game, game.aiPlayer, max_depth)
while True:
clock.tick(60)
if (game.checkWin()):
screen.fill((190, 175, 175))
renderBoardFromState()
if (game.checkWin() == 'red'):
red_text = font.render('RED wins!', True, (234, 43, 31))
red_rect = red_text.get_rect()
red_rect = (300, 20)
screen.blit(red_text, red_rect)
elif (game.checkWin() == 'blue'):
blue_text = font.render('BLUE wins!', True, (41, 88, 170))
blue_rect = red_text.get_rect()
def console(board, algorithm, maximum_depth):
current_state = ai.State(board, board.aiPlayer, maximum_depth)
timerStarted = False
while(True):
board.print()
if (board.checkWin()):
if (board.checkWin() == 'draw'):
print("Draw!")
print("Blue score:", board.scoreHeuristic("blue"))
print("Red score:", board.scoreHeuristic("red"))
break
if (board.checkWin() == 'red'):
print("Red wins!")
print("Blue score:", board.scoreHeuristic("blue"))
print("Red score:", board.scoreHeuristic("red"))
break
if (board.checkWin() == 'blue'):
print("Blue wins!")
print("Blue score:", board.scoreHeuristic("blue"))
print("Red score:", board.scoreHeuristic("red"))
break
if ((board.humanPlayer == 'blue' and board.blue_moves) or (board.humanPlayer == 'red' and not board.blue_moves)):
choice = input("Type quit if you want to quit, else just enter: ")
if choice == "quit":
print("Blue score:", board.scoreHeuristic("blue"))
print("Red score:", board.scoreHeuristic("red"))
break
print("BLUE" if board.blue_moves else "RED", "moves")
print("new move (i, j) -> (x, y)")
if (not timerStarted):
timerStarted = True
t_before = int(round(time.time() * 1000))
try:
i = int(input("i: "))
j = int(input("j: "))
x = int(input("x: "))
y = int(input("y: "))
except Exception as E:
print("Invalid format, try again\n")
continue
try:
board.move(i, j, x, y, board.board[i][j])
except Exception as E:
print("Error:", E)
t_after = int(round(time.time() * 1000))
timerStarted = False
print("Human thought for:", str(t_after - t_before), "miliseconds")
else:
print("BLUE" if board.blue_moves else "RED", "moves")
t_before = int(round(time.time() * 1000))
if (algorithm == 1):
current_state = ai.min_max(current_state)
else:
current_state = ai.ab_pruning(-1000, +1000, current_state)
t_after = int(round(time.time() * 1000))
print("AI thought for:", str(t_after - t_before), "miliseconds")
board.board = current_state.choice.board.board
board.blue_moves = current_state.choice.board.blue_moves
def creaturesTest(rootNode1_acc,
rootNode2_acc,
showing=False,
time_length=None,
playing=False,
online=False):
number_of_balls = 1
number_of_players = 2
number_of_skulls = 0
balls = list()
players = list()
skulls = list()
if online: socket, isClient = tcp_connect(ip_address)
if not playing:
rootNode1_accx = rootNode1_acc[0]
rootNode1_accy = rootNode1_acc[1]
rootNode2_accx = rootNode2_acc[0]
rootNode2_accy = rootNode2_acc[1]
if showing: pygame.init()
if showing: os.environ['SDL_VIDEO_CENTERED'] = '1'
if showing:
vis = general.getVisibleSize()
window = pygame.display.set_mode(
(int(vis[0]), int(vis[1])),
pygame.DOUBLEBUF) # , pygame.FULLSCREEN)
else:
window = pygame.display.set_mode((100, 100))
fitness1 = 0
fitness2 = 0
if showing:
pygame.display.set_caption('Genetic Soccer')
general.surface = pygame.display.get_surface()
field = Field()
goal1 = field.getGoal1()
goal2 = field.getGoal2()
for i in xrange(0, number_of_balls):
balls.append(Ball(surface=general.surface, radius=6, showing=showing))
for i in xrange(0, number_of_players):
players.append(
Player(general.surface, (20, 20), number_of_players - i, showing))
for i in xrange(0, number_of_skulls):
skulls.append(CreatureBodyComputer(surface=general.surface))
ai_state = ai.State(players[0], players[1], balls[0])
my_ai = ai.Ai(players[1], goal1, goal2)
if showing: clock = pygame.time.Clock()
counter = 0
while counter < time_length or time_length is None:
""" Main loop of the simulation (e.g., game). """
moveCamera(balls[0])
stoppingx_event = False
stoppingy_event = False
new_key = False
mouse_pos = pygame.mouse.get_pos()
goal_size = goal1.image.get_size()
if online:
ball = balls[0]
player1 = players[0]
player2 = players[1]
terminals0 = Terminals(counter, ball.pos, ball.vel, player1.pos,
player1.vel, 1, [0, 0], [0, 0], 1)
terminals1 = Terminals(
counter, ball.pos, ball.vel,
[width - player2.pos[0], height - player2.pos[1]],
[-player2.vel[0], -player2.vel[1]], 1, [0, 0], [0, 0], 1
) # width-players[1].pos[0],height-players[1].pos[1], players[1].vel[0], players[1].vel[1], 1)
if not playing:
players[0].acc[0] = running_tree(terminals0).run_tree(
rootNode1_accx)
players[0].acc[1] = running_tree(terminals0).run_tree(
rootNode1_accy)
players[1].acc[0] = running_tree(terminals1).run_tree(
rootNode2_accx)
players[1].acc[1] = -running_tree(terminals1).run_tree(
rootNode2_accy)
if showing: clock.tick(120)
counter += 1
if showing:
""" Background """
if showing: field.blitBackground(general.surface)
""" Keyboard events """
input = pygame.event.get()
for event in input:
whatkey = Keyboard(event)
if whatkey.isKEYDOWN():
if whatkey.isEscape():
sys.exit(0)
elif whatkey.isEquals():
general.global_zoom *= 1.1
elif whatkey.isMinus():
general.global_zoom /= 1.1
elif whatkey.isLeft():
players[0].fireLeft()
elif whatkey.isRight():
players[0].fireRight()
elif whatkey.isUp():
players[0].fireUp()
elif whatkey.isDown():
players[0].fireDown()
elif whatkey.isPeriod():
balls[0].getShooted(players[0])
elif whatkey.isSlash():
balls[0].getShooted(players[0], 0.46)
elif whatkey.isA():
players[1].fireLeft()
elif whatkey.isD():
players[1].fireRight()
elif whatkey.isW():
players[1].fireUp()
elif whatkey.isS():
players[1].fireDown()
elif whatkey.isBackquote():
balls[0].getShooted(players[1])
elif whatkey.isOne():
balls[0].getShooted(players[1], 0.46)
elif whatkey.isKEYUP():
if whatkey.isLeft() or whatkey.isRight():
players[0].stopLeftAndRight()
elif whatkey.isUp() or whatkey.isDown():
players[0].stopUpAndDown()
elif whatkey.isA() or whatkey.isD():
players[1].stopLeftAndRight()
elif whatkey.isW() or whatkey.isS():
players[1].stopUpAndDown()
if event.type == pygame.MOUSEBUTTONDOWN:
left, top = (x - 50 for x in event.pos)
if whatkey.isKEYUP() or whatkey.isKEYDOWN():
new_key = True
if online: net.updatePlayers(new_key, isClient, socket, players)
ai_state.update()
my_ai.do(ai_state)
for player in players:
player.collidingGoal(goal1)
player.collidingGoal(goal2)
player.move(general.surface,
field,
counter,
cameraPos=cameraPos,
goal_size=goal_size)
for ball in balls:
""" All balls """
ball_centre = ball.getCentre()
# fitness1 += 1000/distance(ball_centre, players[0].get_pos())
# fitness2 += 1000/distance(ball_centre, players[1].get_pos())
ball.is_under_player = False
# players colliding with the ball
for player in players:
ball.getKicked(player)
# computer player colliding with the ball and the goal
for cc in skulls:
cc.collidingGoal(goal1)
cc.collidingGoal(goal2)
ball.getKicked(cc)
if debug:
print(mouse_pos)
cc.acc = [(random.random() - 0.5) / 5.0,
(random.random() - 0.5) / 5.0]
cc.move(general.surface,
field,
counter,
cameraPos=cameraPos,
goal_size=goal_size)
# to the pipe (tirake darvaze):
is_colliding_goal_pipe_top_right = (
ball.vel[1] < 0
and (diff(ball_centre[1], goal_size[1]) < ball.radius)
and (diff(ball_centre[0], (width / 2 + goal_size[0] / 2)) < 3))
is_colliding_goal_pipe_top_left = (
ball.vel[1] < 0
and (diff(ball_centre[1], goal_size[1]) < ball.radius)
and (diff(ball_centre[0], (width / 2 - goal_size[0] / 2)) < 3))
is_colliding_goal_pipe_bottom_right = (
ball.vel[1] > 0
and (diff(ball_centre[1],
(height - goal_size[1])) < ball.radius)
and (diff(ball_centre[0], (width / 2 + goal_size[0] / 2)) < 3))
is_colliding_goal_pipe_bottom_left = (
ball.vel[1] > 0
and (diff(ball_centre[1],
(height - goal_size[1])) < ball.radius)
and (diff(ball_centre[0], (width / 2 - goal_size[0] / 2)) < 3))
if is_colliding_goal_pipe_top_right or is_colliding_goal_pipe_top_left or is_colliding_goal_pipe_bottom_right or is_colliding_goal_pipe_bottom_left:
ball.vel[1] = -ball.vel[1]
else:
ball.collidingGoal(goal1)
ball.collidingGoal(goal2)
if goal1.rect.contains(ball.rect):
fitness2 += 100
field.score2 += 1
ball.reset()
players[0].reset()
players[1].reset()
players[0].fire_acc += 0.01
players[1].fire_acc += 0.01
if showing: field.setMessage2()
elif goal2.rect.contains(ball.rect):
fitness1 += 100
field.score1 += 1
ball.reset()
players[0].reset()
players[1].reset()
players[0].fire_acc += 0.01
players[1].fire_acc += 0.01
if showing: field.setMessage1()
ball.move(counter, cameraPos=cameraPos)
if showing:
field.blitField(width=width,
height=height,
surface=general.surface,
cameraPos=cameraPos)
vis = general.getVisibleSize()
if debug:
pygame.draw.rect(general.surface, pygame.Color(0, 0, 0),
(0, 0, vis[0], vis[1]), 1)
if showing: pygame.display.flip()
return (1.0 * fitness1 / counter, 1.0 * fitness2 / counter)