def main():
pygame.init() #Initialize pygame
stan = State() #Initialize state of the game
game_screen = Screen(stan.game_width, stan.game_length) #Initialize game screen
snake_dead = False
clock = pygame.time.Clock()
while not snake_dead:
clock.tick(5)
key = handle_keys()
if key is None:
key = stan.prev_key
stan.move_snake(key)
snake_dead = not stan.check_if_snake_lives()
text = game_screen.myfont.render("Score {0}".format(stan.points), 1, (0, 0, 0))
game_screen.game_display.blit(text, (5, 10))
game_screen.display_move(stan) #Update game screen
pygame.display.update()
def display_evaluation_game(input_file):
pygame.init()
stan = State()
game_screen = Screen(stan.game_width, stan.game_length)
clock = pygame.time.Clock()
with open(input_file, "r") as fp:
for line in fp:
single_move = eval(line)
screen, reward_print, total_reward = single_move
clock.tick(5)
convert_env_to_state(stan, screen)
game_screen.display_move(stan)
text2 = game_screen.myfont.render(
"Score: {:0.0f} | Step reward: {:0.2f} | Total reward: {:0.1f}"
.format(stan.points, reward_print, total_reward), 1, (0, 0, 0))
game_screen.game_display.blit(text2, (10, 10))
pygame.display.update()
