def displayRL(player, food, enemy):
pygame.init()
clock = pygame.time.Clock()
screen = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT), 0, 32)
surface = pygame.Surface(screen.get_size())
surface = surface.convert()
drawGrid(surface)
myfont = pygame.font.SysFont("bahnschrift", 20)
max_iter = 0
while (player.lives > 4):
max_iter += 1
if max_iter == 50:
break
clock.tick(7)
drawGrid(surface)
obs = (player - food, player.get_head_position())
# obs = (player - food)
# no random action in display phase
action_space = np.array(q_table[obs]).copy()
player.action(action_space, "QL")
handleSnakeCollisions(player, enemy)
handleFoodEating([player, enemy], food)
player.draw(surface)
enemy.draw(surface)
food.draw(surface)
screen.blit(surface, (0, 0))
text1 = myfont.render("Score Player {0}".format(player.score), 1,
(250, 250, 250))
# text2 = myfont.render("Score AI {0}".format(enemy.score), 1, (250, 250, 250))
screen.blit(text1, (5, 10))
# screen.blit(text2, (SCREEN_WIDTH - 120, 10))
pygame.display.update()
print("Snake Player Final Score:", player.score)
def step_render(self, key):
'''
:return: [lives, score, player, enemy, food]
'''
self.clock.tick(3)
drawGrid(self.surface)
if not self.food:
self.generate_food()
_lives, _score, _player, _enemy, _food = self.step(key)
self.draw_snake(self.player, self.surface, SNAKE_COLOR[0], SNAKE_HEAD_COLOR[0])
self.draw_snake(self.enemy, self.surface, SNAKE_COLOR[1], SNAKE_HEAD_COLOR[1])
if not self.food:
self.generate_food()
self.draw_food(self.surface, FOOD_COLOR)
self.screen.blit(self.surface, (0, 0))
text1 = self.myfont.render("Score: {0} Lives: {1}".format(round(self.score, 2), self.lives), True, (250, 250, 250))
# text2 = myfont.render("Score AI {0}".format(enemy.score), 1, (250, 250, 250))
self.screen.blit(text1, (5, 10))
# screen.blit(text2, (SCREEN_WIDTH - 120, 10))
pygame.display.update()
return _lives, _score, _player, _enemy, _food
def render_init(self):
pygame.init()
self.clock = pygame.time.Clock()
self.screen = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT), 0, 32)
self.surface = pygame.Surface(self.screen.get_size())
self.surface = self.surface.convert()
drawGrid(self.surface)
self.myfont = pygame.font.SysFont("bahnschrift", 20)
def display():
pygame.init()
clock = pygame.time.Clock()
screen = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT), 0, 32)
surface = pygame.Surface(screen.get_size())
surface = surface.convert()
drawGrid(surface)
player = Snake(0)
player.lives = 10
enemy = Snake(1)
enemy.positions = [(random.randint(0,
SIZE - 1), random.randint(0, SIZE - 1))]
food = Food([player])
myfont = pygame.font.SysFont("bahnschrift", 20)
iterations = 0
while (player.lives > 0):
for event in pygame.event.get():
if event.type == pygame.KEYDOWN:
if event.key == pygame.K_1:
print("SPace")
clock.tick(0)
#print(get_food_distance(player, food))
#getObsSmall([player, enemy], food)
clock.tick(1)
drawGrid(surface)
enemy.handle_keys()
obs = (player - food)
# no random action in display phase
action_space = np.array(q_table[obs]).copy()
player.action(action_space, "QL")
enemy.move()
handleSnakeCollisions(player, enemy)
handleFoodEating([player, enemy], food)
getObsGrid([player, enemy], food, size=7, fullGrid=True)
player.draw(surface)
enemy.draw(surface)
food.draw(surface)
screen.blit(surface, (0, 0))
text1 = myfont.render("Score AI {0}".format(player.score), 1,
(250, 250, 250))
text2 = myfont.render("Score Player {0}".format(enemy.score), 1,
(250, 250, 250))
screen.blit(text1, (5, 10))
screen.blit(text2, (SCREEN_WIDTH - 175, 10))
pygame.display.update()
print("Snake Player Final Score:", player.score)
def render(agent, obs_type: str, board_size: int):
pygame.init()
clock = pygame.time.Clock()
screen = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT), 0, 32)
surface = pygame.Surface(screen.get_size())
surface = surface.convert()
drawGrid(surface)
player = Snake(0)
player.lives = 5
enemy = Snake(1)
enemy.lives = 5
enemy.positions = [[5,5]]
food = Food([player])
myfont = pygame.font.SysFont("bahnschrift", 20)
iterations = 0
while (player.lives > 4 and enemy.lives > 4):
clock.tick(4)
drawGrid(surface)
enemyMovement(food=food,enemy=enemy)
# no random action in display phase
if obs_type == "Grid":
obs = getObsGrid([player], food, size=board_size, fullGrid=False)
if obs_type == 'Small':
obs = getObsSmall([player, enemy], food)
action = agent.choose_action(obs)
action_space = [action]
player.action(action_space, "PG")
handleSnakeCollisions(player,enemy)
handleFoodEating([player, enemy], food)
player.draw(surface)
enemy.draw(surface)
food.draw(surface)
screen.blit(surface, (0, 0))
text1 = myfont.render("Score AI {0}".format(player.score), 1, (250, 250, 250))
#text2 = myfont.render("Score Player {0}".format(enemy.score), 1, (250, 250, 250))
screen.blit(text1, (5, 10))
#screen.blit(text2, (SCREEN_WIDTH - 175, 10))
pygame.display.update()
print("Snake Player Final Score:", player.score)