class Game:
textoX = escrever.render('X', True, (200, 0, 0))
def __init__(self):
global escrever
self.screen = None
self.atraso = 350 # quanto tempo entre os movimentos
self.game = GameEngine()
self.novoJogo()
pass
def novoJogo(self, tamanhoCobra=3):
self.game.iniciar(tamanhoCobra)
self.width, self.height = self.game.tamanhoX * tamanhoPeca + tamanhoPeca * 2, self.game.tamanhoY * tamanhoPeca + tamanhoPeca * 2
self.size = self.width, self.height + painelPontuacaoH
self.proxDir = GameEngine.BAIXO
self.clock = pygame.time.Clock()
self.clock.tick()
self.tempo = 0
self.frutaIdx = random.randint(0, len(frutas) - 1)
if self.screen is None:
self.screen = pygame.display.set_mode(self.size)
pass
def esperarTick(self):
# permitir o movimento da cobra
self.clock.tick()
self.tempo += self.clock.get_time()
if self.tempo >= self.atraso:
self.tempo = 0
self.game.moverCobra(self.proxDir)
return True
return False
def draw(self):
# limpar a tela
self.screen.fill(branco)
# pintar linhas dos quadros
self.screen.fill(
(200, 200, 200),
pygame.Rect(tamanhoPeca,
(1 + self.game.tamanhoY / 2) * tamanhoPeca,
self.game.tamanhoX * tamanhoPeca, 1))
self.screen.fill(
(200, 200, 200),
pygame.Rect((1 + self.game.tamanhoX / 2) * tamanhoPeca,
tamanhoPeca, 1, self.game.tamanhoY * tamanhoPeca))
self.screen.blit(escrever.render('1', True, (200, 200, 200)),
(1 * self.width / 4, 1 * self.height / 4))
self.screen.blit(escrever.render('2', True, (200, 200, 200)),
(3 * self.width / 4, 1 * self.height / 4))
self.screen.blit(escrever.render('3', True, (200, 200, 200)),
(1 * self.width / 4, 3 * self.height / 4))
self.screen.blit(escrever.render('4', True, (200, 200, 200)),
(3 * self.width / 4, 3 * self.height / 4))
# pintar espacos da tela
for i in range(0, self.game.tamanhoY + 2):
self.screen.fill(
marrom,
pygame.Rect(0, i * tamanhoPeca, tamanhoPeca, tamanhoPeca))
self.screen.fill(
marrom,
pygame.Rect(self.width - tamanhoPeca, i * tamanhoPeca,
tamanhoPeca, tamanhoPeca))
pass
for i in range(0, self.game.tamanhoX + 2):
self.screen.fill(
marrom,
pygame.Rect(i * tamanhoPeca, 0, tamanhoPeca, tamanhoPeca))
self.screen.fill(
marrom,
pygame.Rect(i * tamanhoPeca, self.height - tamanhoPeca,
tamanhoPeca, tamanhoPeca))
pass
# pintar fruta
# self.screen.fill( verm, pygame.Rect( (self.game.comida[0]+1) * tamanhoPeca, (self.game.comida[1]+1) * tamanhoPeca, tamanhoPeca, tamanhoPeca ) )
self.screen.blit(frutas[self.frutaIdx],
((self.game.comida[0] + 1) * tamanhoPeca,
(self.game.comida[1] + 1) * tamanhoPeca))
# pintar corpo
for i in range(1, len(self.game.corpo)):
corpo = self.game.corpo[i]
self.screen.fill(
verde2,
pygame.Rect((corpo[0] + 1) * tamanhoPeca,
(corpo[1] + 1) * tamanhoPeca, tamanhoPeca,
tamanhoPeca))
pass
cabeca = self.game.corpo[0]
self.screen.fill(
verde1,
pygame.Rect((cabeca[0] + 1) * tamanhoPeca,
(cabeca[1] + 1) * tamanhoPeca, tamanhoPeca,
tamanhoPeca))
# verificar e pintar 'X' na cabeça da cobra caso o jogo tenha terminado
if self.game.terminado:
self.screen.blit(Game.textoX, ((cabeca[0] + 1) * tamanhoPeca + 4,
(cabeca[1] + 1) * tamanhoPeca - 2))
pass
# escrever pontuação
textoPontuacao = escrever.render('Pontos: %d' % (self.game.pontos),
True, (0, 0, 0))
self.screen.blit(textoPontuacao, (10, self.height + 20))
# escrever entradas do treinamento para o estado atual
entradas = ia_engine.criarEntradas(self.game)
# direção da fruta
txt = '%d%d%d%d' % (entradas[0], entradas[1], entradas[2], entradas[3])
self.screen.blit(escrever.render('^ > . <', True, (160, 160, 160)),
(110, self.height))
self.screen.blit(escrever.render(txt, True, (200, 40, 40)),
(110, self.height + 20))
self.screen.blit(escrever.render('fruta', True, (200, 200, 200)),
(110, self.height + 40))
# direção da cobra
txt = '%d%d%d%d' % (entradas[4], entradas[5], entradas[6], entradas[7])
self.screen.blit(escrever.render('^ > . <', True, (160, 160, 160)),
(180, self.height))
self.screen.blit(escrever.render(txt, True, (20, 140, 20)),
(180, self.height + 20))
self.screen.blit(escrever.render('cobra', True, (200, 200, 200)),
(180, self.height + 40))
# perigo nas direções de decisão
txt = '%d%d%d' % (entradas[8], entradas[9], entradas[10])
self.screen.blit(escrever.render('D F E', True, (160, 160, 160)),
(250, self.height))
self.screen.blit(escrever.render(txt, True, (200, 40, 40)),
(250, self.height + 20))
self.screen.blit(escrever.render('perigo', True, (200, 200, 200)),
(250, self.height + 40))
# espaços vazios
txt = '%3d %3d %3d %3d' % (entradas[11] * 100, entradas[12] * 100,
entradas[13] * 100, entradas[14] * 100)
self.screen.blit(
escrever.render('%3d %3d %3d %3d' % (1, 2, 3, 4), True,
(160, 160, 160)), (320, self.height))
self.screen.blit(escrever.render(txt, True, (200, 40, 40)),
(320, self.height + 20))
self.screen.blit(escrever.render('espaços', True, (200, 200, 200)),
(320, self.height + 40))
pygame.display.flip()
pass
saidas = model.predict(entradas.reshape((1, 15)))[0]
hashEntradas = ia_engine.hashEntradas(entradas)
# if not comModeloTreinado and randint(0, aleatorio) < (partidas - par):
# resposta = to_categorical(randint(0, 2), 3)
# else:
# resposta = saidas
resposta = saidas
maxIdx = np.argmax(saidas)
novaDir = ia_engine.novaDirecao(
resposta, game) # nova direção que a IA encontrou
pontuacao = game.pontos
distancia1 = ia_engine.distancia(game.corpo[0], game.comida)
game.moverCobra(novaDir)
distancia2 = ia_engine.distancia(game.corpo[0], game.comida)
# buscar os valores do estado futuro
entradas_2 = ia_engine.criarEntradas(game)
# saidas_2 = model.predict( entradas.reshape((1, 11)) )[0]
hashEntradas_2 = ia_engine.hashEntradas(entradas_2)
premio = 0
if game.pontos > pontuacao: # entao encontrou a comida
premio = 1
elif game.terminado:
premio = -1
else:
if distancia1 <= distancia2: # moveu para longe
premio = -0.15