class Game:
def __init__(self):
self.screen = pygame.display.set_mode((GAME_HEIGHT, GAME_WIDTH))
self.clock = pygame.time.Clock()
self.food = Food(FOOD_COLOR)
self.snake = Snake(SNAKE_COLOR)
self.game_over = False
def run(self):
# update core game components
self.clock.tick(100 * (1 + len(self.snake.body) / 50))
self.screen.fill(pygame.Color(0,0,0))
# render the snake and food for the game
self.food.render(self.screen)
self.snake.render(self.screen)
# update the snake and check for lose conditions
self.snake.snake_controls(self.food)
self.game_over = self.snake.check_lose()
return self.game_over
class Game:
MODE = "AGENT"
HEADER_HEIGHT = 48
WINDOW_WIDTH = 480
WINDOW_HEIGHT = 480
SNAKE_STEP = 16
EATING_REWARD = 1
NUM_GAMES_AGENT = 200
def __init__(self, ind_game=0):
pygame.init()
pygame.display.set_caption("My snake game!")
self._display_surf = pygame.display.set_mode(
(self.WINDOW_WIDTH, self.WINDOW_HEIGHT), pygame.HWSURFACE)
self.score = 0
self.ind_game = ind_game
self._running = True
self.flag_lost = False
self.snake = Snake(
length=1,
step=self.SNAKE_STEP,
x_max=self.WINDOW_WIDTH,
y_max=self.WINDOW_HEIGHT,
)
self.walls = Wall(
self.HEADER_HEIGHT,
self.WINDOW_HEIGHT,
self.WINDOW_WIDTH,
thickness=self.SNAKE_STEP,
)
self.raspi = RasPi(
x=self.SNAKE_STEP *
(-1 + floor(self.WINDOW_WIDTH / 2 / self.SNAKE_STEP)),
y=self.SNAKE_STEP *
(floor(self.WINDOW_HEIGHT / 2 / self.SNAKE_STEP)),
)
self.clock = pygame.time.Clock()
if self.ind_game == 0:
self.high_score = 0
if self.MODE == "AGENT":
self.frame_rate = 60
if self.ind_game == 0:
self.agent = Agent()
self.num_games = self.NUM_GAMES_AGENT
else:
self.frame_rate = 15
self.num_games = 1
def on_event(self, event):
if event.type == QUIT:
self._running = False
Game.on_cleanup()
exit()
def on_loop(self):
"""
Method executed for each clock tick
"""
self.snake.update()
if self.MODE == "AGENT":
self.agent.store(self)
self.flag_lost = self.snake.has_lost(self.walls)
flag_snake_over_raspi = True
if self.snake.has_eaten(self.raspi.x, self.raspi.y):
self.score += self.EATING_REWARD
self.snake.lengthen()
while flag_snake_over_raspi:
self.raspi.spawn_at_random(
x_min=self.walls.x_min_inside + self.walls.thickness,
x_max=self.walls.x_max_inside,
y_min=self.walls.y_min_inside,
y_max=self.walls.y_max_inside,
step=self.SNAKE_STEP,
)
flag_snake_over_raspi = self.snake.is_over(
self.raspi.x, self.raspi.y)
self._running = not self.flag_lost
if self.MODE == "AGENT":
self.agent.train_network_short_term()
def on_render(self):
self._display_surf.fill((0, 0, 0))
self.snake.render(self._display_surf)
self.raspi.render(self._display_surf)
self.walls.render(self._display_surf)
self.display_message(f"Score: {self.score}", (50, 20), fontsize=16)
self.display_message(f"High score: {self.high_score}",
(self.WINDOW_WIDTH - 80, 20),
fontsize=16)
if self.MODE == "AGENT":
self.display_message(
f"Game n°: {self.ind_game} / {self.num_games}",
(int(self.WINDOW_WIDTH / 2), 20),
fontsize=16,
)
pygame.display.flip()
@staticmethod
def on_cleanup():
pygame.quit()
def on_execute(self):
for i_games in range(self.num_games):
while self._running:
if self.MODE == "MANUAL":
self.manual_move()
elif self.MODE == "AGENT":
self.agent_move()
self.on_loop()
self.on_render()
self.clock.tick(self.frame_rate)
if self.flag_lost:
self.set_high_score()
if self.MODE == "MANUAL":
self.display_message(
f"You lost, you big big loser! - final score: {self.score}",
(self.WINDOW_WIDTH // 2, self.WINDOW_HEIGHT // 2),
fontsize=16,
)
self.display_message(
f"Replay: press space, Quit: press escape",
(self.WINDOW_WIDTH // 2,
self.WINDOW_HEIGHT // 2 + 50),
fontsize=16,
)
accepted_replay = "undef"
while accepted_replay == "undef":
accepted_replay = self.propose_replay()
elif self.MODE == "AGENT":
self.agent.train_network()
self.agent.summary(self)
self.__init__(self.ind_game + 1) # Reinit game
Game.on_cleanup()
def manual_move(self):
for event in pygame.event.get():
self.on_event(event)
pygame.event.pump()
keys = pygame.key.get_pressed()
if keys[K_RIGHT] and self.snake.direction != "left":
self.snake.move_right()
if keys[K_LEFT] and self.snake.direction != "right":
self.snake.move_left()
if keys[K_UP] and self.snake.direction != "down":
self.snake.move_up()
if keys[K_DOWN] and self.snake.direction != "up":
self.snake.move_down()
if keys[K_ESCAPE]:
self._running = False
def agent_move(self):
next_move = self.agent.next_move(self)
if next_move == "right" and self.snake.direction != "left":
self.snake.move_right()
if next_move == "left" and self.snake.direction != "right":
self.snake.move_left()
if next_move == "up" and self.snake.direction != "down":
self.snake.move_up()
if next_move == "down" and self.snake.direction != "up":
self.snake.move_down()
def display_message(self, message_string, position, fontsize=32):
font = pygame.font.Font("freesansbold.ttf", fontsize)
text = font.render(message_string, True, (255, 255, 255), (0, 0, 0))
text_rect = text.get_rect()
text_rect.center = position
self._display_surf.blit(text, text_rect)
pygame.display.flip()
def propose_replay(self):
pygame.event.pump()
keys = pygame.key.get_pressed()
if keys[K_SPACE]:
self.__init__()
return "yes"
elif keys[K_ESCAPE]:
self._running = False
return "no"
else:
return "undef"
def set_high_score(self):
if self.score > self.high_score:
self.high_score = self.score