This repository has been archived by the owner on May 2, 2019. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Trilha.py
213 lines (200 loc) · 9.02 KB
/
Trilha.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
from aima.games import Game, GameState
class Trilha(Game):
def __init__(self):
"""
Caso utilizar matriz:
vazio = [(1, 1), (1, 4), (1, 7), (2, 2), (2, 4), (2, 6), (3, 3), (3, 4), (3, 5), (4, 1), (4, 2), (4, 3), (4, 5),
(4, 6), (4, 7), (5, 3), (5, 4), (5, 5), (6, 2), (6, 4), (6, 6), (7, 1), (7, 4), (7, 7)]
"""
# Posições do tabuleiro sem peças
vazio = [x for x in range(1, 25)]
# Dicionário com todas as posições e seus adjacentes. Ex: {1: [2,8]}
self.adjacentes = {}
# Cria os adjacentes
for x in range(1, 25):
# Caso x seja ímpar
if (x % 2) == 1:
# Verifica se é 1,9 ou 17
if (x % 8) == 1:
self.adjacentes.update({x: [x + 1, x + 7]})
else:
self.adjacentes.update({x: [x - 1, x + 1]})
# Caso par
else:
# 1 até 8
if x <= 8:
if (x % 8) == 0:
self.adjacentes.update({x: [x - 7, x - 1, x + 8]})
else:
self.adjacentes.update({x: [x - 1, x + 1, x + 8]})
# De 9 até 16
elif x > 8 and x <= 16:
if (x % 8) == 0:
self.adjacentes.update({x: [x - 8, x - 7, x - 1, x + 8]})
else:
self.adjacentes.update({x: [x - 8, x - 1, x + 1, x + 8]})
# De 17 até 24
else:
if (x % 8) == 0:
self.adjacentes.update({x: [x - 8, x - 7, x - 1]})
else:
self.adjacentes.update({x: [x - 8, x - 1, x + 1]})
"""Estado inicial"""
self.initial = GameState(to_move="BRANCO", utility=0, board={"BRANCO": 0, "PRETO": 0, "REMOVE": 0}, moves=vazio)
def actions(self, state):
"""Retorna todos os movimentos possíveis no momento"""
return state.moves
def result(self, state, move):
"""Deve retornar o estado que ficará o jogo quando uma é aplicado um movimento a um estado anterior"""
# Movimentos ilegais não tem efeitos
if move not in state.moves:
return state
# Copia o tabuleiro
board = state.board.copy()
utility = 0
# Jogadas
moves = []
# Lista para não procurar no tabuleiro
lista = ["BRANCO", "PRETO", "REMOVE"]
# Procura uma substring e verifica caso seja a vez de alguém remover a peça
if state.board['REMOVE'] == 1:
# Remove a peça do tabuleiro
del (board[move])
# Define quem é o próximo a jogar
board['REMOVE'] = 0
# Verifica se alguém ganhou com essa jogada
utility = self.compute_utility(board, state.to_move)
# Verifica ainda tem peça para jogar no tabuleiro
elif board[state.to_move] < 9:
board[move] = state.to_move
board[state.to_move] = board[state.to_move] + 1
# Caso já tenham sido jogadas as peças, então é para mover uma peça no tabuleiro
else:
# Define o destino
board[move[1]] = state.to_move
# Remove da origem
del (board[move[0]])
move = move[1]
# Define quem é o próximo a jogar
if state.to_move == "BRANCO":
if state.board['REMOVE'] == 0 and self.verifica_trinca(board=board, to_move=state.to_move, move=move):
to_move = "BRANCO"
board['REMOVE'] = 1
else:
to_move = "PRETO"
else:
if state.board['REMOVE'] == 0 and self.verifica_trinca(board=board, to_move=state.to_move, move=move):
to_move = "PRETO"
board['REMOVE'] = 1
else:
to_move = 'BRANCO'
# Caso o próximo movimento seja de remover
if board['REMOVE'] == 1:
# POS é o número no tabuleiro e to é quem jogou
for pos, to in board.items():
# Caso o estado seja "BRANCO_REMOVE" procura todas as peças preta
if not to == to_move and pos not in lista:
moves.append(pos)
# Caso o próximo movimento seja só colocar uma peça no tabuleiro
elif board[to_move] < 9:
moves = self.vazios(board)
# Caso o próximo movimento seja movimentar uma peça no tabuleiro
else:
jogadas = []
vazios = self.vazios(board)
for pos, to in board.items():
if to == to_move and pos not in lista:
# Salva todas as jogadas do próximo a mover a peça
jogadas.append(pos)
# Verifica cada jogada
for jogada in jogadas:
# Pega os adjacentes de cada jogada
adjacente = self.adjacentes[jogada]
# Verifica em cada adjacente se ele está vazio
for a in adjacente:
if a in vazios:
# Salva um tuple contendo (ORIGEM,DESTINO)
moves.append((jogada, a))
return GameState(to_move=to_move, utility=utility, board=board, moves=moves)
def utility(self, state, player):
"""Retorna o estado final para o jogador. 0 se o jogo não acabou, 1 se o player passado ganhou ou -1 se ele
perdeu"""
return state.utility if player == 'BRANCO' else -state.utility
def terminal_test(self, state):
"""Retorna verdadeiro se é o estado final do jogo"""
return state.utility != 0
def compute_utility(self, board, player):
"""Verifica se ouve algum ganhador"""
if board[player] < 9:
return 0
cont = 0
lista = ["BRANCO", "PRETO", "REMOVE"]
for pos, to in board.items():
if not to == player and pos not in lista:
cont = cont + 1
if cont < 3:
return 1 if player == "BRANCO" else -1
return 0
def verifica_trinca(self, board, to_move, move):
"""Verifica se houve alguma trinca com o movimento"""
# Caso move seja ímpar
if (move % 2) == 1:
# Verifica se é 1,9 ou 17
if (move % 8) == 1:
if ((move + 1) in board.keys() and board[move + 1] == to_move and (
move + 2) in board.keys() and board[move + 2] == to_move) or (
(move + 7) in board.keys() and board[move + 7] == to_move and (
move + 6) in board.keys() and board[move + 6] == to_move):
return True
else:
return False
elif (move % 8) == 7:
if ((move + 1) in board.keys() and board[move + 1] == to_move and (
move - 6) in board.keys() and board[move - 6] == to_move) or (
(move - 1) in board.keys() and board[move - 1] == to_move and (
move - 2) in board.keys() and board[move - 2] == to_move):
return True
else:
return False
elif ((move - 1) in board.keys() and board[move - 1] == to_move and (
move - 2) in board.keys() and board[move - 2] == to_move) or (
(move + 1) in board.keys() and board[move + 1] == to_move and (
move + 2) in board.keys() and board[move + 2] == to_move):
return True
else:
return False
# Caso par
else:
if (move % 8) == 0:
if (move - 7) in board.keys() and board[move - 7] == to_move and (
move - 1) in board.keys() and board[move - 1] == to_move:
return True
elif (move + 1) in board.keys() and board[move + 1] == to_move and (
move - 1) in board.keys() and board[move - 1] == to_move:
return True
if move <= 8:
if (move + 8) in board.keys() and board[move + 8] == to_move and (
move + 16) in board.keys() and board[move + 16] == to_move:
return True
else:
return False
# De 9 até 16
elif move > 8 and move <= 16:
if (move - 8) in board.keys() and board[move - 8] == to_move and (
move + 8) in board.keys() and board[move + 8] == to_move:
return True
else:
return False
# De 17 até 24
else:
if (move - 8) in board.keys() and board[move - 8] == to_move and (
move - 16) in board.keys() and board[move - 16] == to_move:
return True
else:
return False
def vazios(self, board):
vazios = []
for x in range(1, 25):
if x not in board.keys():
vazios.append(x)
return vazios