-
Notifications
You must be signed in to change notification settings - Fork 0
/
q_light.py
347 lines (312 loc) · 17.2 KB
/
q_light.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
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
"""
This file contains the code for the 3D-like Tic-Tac-Toe game using Pygame.
"""
import random
import math
import pygame
import pygame.gfxdraw
QUEEN_COLOR = (100, 200, 250) # Bright red color
SHADOW_COLOR = (0, 0, 100, 10) # Black color
OUTLINE_COLOR = (0, 0, 255, 255) # Black color
SHDOW_OUTLINE_COLOR = (0, 0, 0, 250)
BLACK_CELL_COLOR = [30, 30, 30] # Black color
WHITE_CELL_COLOR = [159, 150, 150] # Bright white color
QUEENS = 8
class GameGUI:
"""
This class is responsible for the graphical user interface of the Tic-Tac-Toe game.
"""
def __init__(self):
pygame.init()
# create QUEENSxQUEENS board
self.board = [[0 for _ in range(QUEENS)] for _ in range(QUEENS)]
self.board_size = 1200 # Size of the board in pixels
self.cell_size = self.board_size // QUEENS # Size of each cell in pixels
self.screen = pygame.display.set_mode(
(self.board_size, self.board_size))
self.clock = pygame.time.Clock()
# Create a grid of indices
self.indices = None
self.diag_indices = None
self.init_cell_sizes()
pygame.display.set_caption(f'{QUEENS} Queens Problem')
self.done_x_pos = 0
self.directions = [(1, 0), (-1, 0), (0, 1), (0, -1)
] # right, left, down, up
self.direction = random.choice(self.directions)
self.pos = [0, 0] # Initial position [x, y]
self.speed = 1 # Speed of movement in pixels per frame
self.last_time = pygame.time.get_ticks()
self.selected_queen = None
self.missing_queens = QUEENS
def init_cell_sizes(self):
self.half_cell_size = self.cell_size // 2
self.sixth_cell_size = self.cell_size // 6
self.eight_cell_size = self.cell_size // 8
def draw_board(self, board=None):
if board is None:
board = self.board
cell_size = self.cell_size
eight_cell_size = self.eight_cell_size
screen = self.screen
game_size = QUEENS
board_size = self.board_size
for i in range(game_size):
for j in range(game_size):
color = BLACK_CELL_COLOR if (i + j) % 2 == 0 else WHITE_CELL_COLOR
pygame.draw.rect(screen, color, pygame.Rect(
j * cell_size, i * cell_size, cell_size, cell_size))
if self.missing_queens > 0:
self.draw_missing_queens(board, cell_size)
for i in range(game_size):
for j in range(game_size):
if board[i][j] == 1:
self.draw_queen(i, j, QUEEN_COLOR, cell_size)
def draw_missing_queens(self, board, cell_size):
# mark red cell for every missing queen
missing_queens = self.missing_queens
for x in range(QUEENS):
for y in range(QUEENS):
if missing_queens == 0:
return
pygame.draw.rect(self.screen, (255, 0, 0), pygame.Rect(
y * cell_size, x * cell_size, cell_size, cell_size))
missing_queens -= 1
def draw_queen(self, x, y, color, cell_size):
center = (y * cell_size + self.half_cell_size,
x * cell_size + self.half_cell_size)
q_size = cell_size + 2 * self.sixth_cell_size
offsets = range(0, q_size, self.sixth_cell_size//3)
gradient_colors = [(max(0, color[0] - 1 * offset), max(0, color[1] -
1 * offset), max(0, color[2] - 1 * offset)) for offset in offsets]
for offset in range(0, q_size, self.sixth_cell_size//3):
highlight_radius = max(
self.half_cell_size - 2, self.half_cell_size - offset)
gradient_color = gradient_colors[offset //
(self.sixth_cell_size // 3)]
alpha = 255
pygame.gfxdraw.aacircle(self.screen, center[0], center[1] - offset, highlight_radius, (
gradient_color[0], gradient_color[1], gradient_color[2], alpha))
if offset < q_size - self.sixth_cell_size // 3:
pygame.gfxdraw.filled_circle(
self.screen, center[0], center[1] - offset, highlight_radius, (gradient_color[0], gradient_color[1], gradient_color[2], alpha))
# add shadow
pygame.gfxdraw.aaellipse(self.screen, center[0], center[1] - offset + 2, highlight_radius - 2, highlight_radius - 2, (
SHDOW_OUTLINE_COLOR))
# add outline
pygame.gfxdraw.aaellipse(
self.screen, center[0], center[1] - offset, highlight_radius, highlight_radius, (OUTLINE_COLOR))
else:
crown_height = self.eight_cell_size # Adjust this value to position the crown correctly
for angle in range(0, 360, 30):
# Calculate the coordinates of the triangle
angle_rad = math.radians(angle)
x1 = center[0]
y1 = center[1] - offset - crown_height # Move the triangle up
x2 = center[0] + highlight_radius * math.cos(angle_rad)
y2 = center[1] + highlight_radius * math.sin(angle_rad) - offset - crown_height # Move the triangle up
x3 = center[0] + highlight_radius * math.cos(angle_rad + math.pi / 3)
y3 = center[1] + highlight_radius * math.sin(angle_rad + math.pi / 3) - offset - crown_height # Move the triangle up
# Draw the triangle
pygame.gfxdraw.aatrigon(self.screen, int(x1), int(y1), int(x2), int(y2), int(x3), int(y3), (
gradient_color[0], gradient_color[1], gradient_color[2], alpha))
pygame.gfxdraw.filled_trigon(self.screen, int(x1), int(y1), int(x2), int(y2), int(x3), int(y3), (
gradient_color[0], gradient_color[1], gradient_color[2], alpha))
def draw_energy(self, energy, board_size, cell_size):
# Convert the energy value to a string
energy_str = str(energy)
# extract the number from "tensor([number])"
energy_str = energy_str[energy_str.index(
'(') + 1:energy_str.index(')')]
# remove the decimal point
energy_str = energy_str.replace('.', '')
# add "Energy: " to the string
energy_str = "Energy: " + energy_str
# Create a mapping from characters to a grid of cells
char_to_pixels = {
'0': [[1, 1, 1], [1, 0, 1], [1, 0, 1], [1, 0, 1], [1, 1, 1]],
'1': [[0, 1, 0], [0, 1, 0], [0, 1, 0], [0, 1, 0], [0, 1, 0]],
'2': [[1, 1, 1], [0, 0, 1], [1, 1, 1], [1, 0, 0], [1, 1, 1]],
'3': [[1, 1, 1], [0, 0, 1], [1, 1, 1], [0, 0, 1], [1, 1, 1]],
'4': [[1, 0, 1], [1, 0, 1], [1, 1, 1], [0, 0, 1], [0, 0, 1]],
'5': [[1, 1, 1], [1, 0, 0], [1, 1, 1], [0, 0, 1], [1, 1, 1]],
'6': [[1, 1, 1], [1, 0, 0], [1, 1, 1], [1, 0, 1], [1, 1, 1]],
'7': [[1, 1, 1], [0, 0, 1], [0, 1, 0], [1, 0, 0], [1, 0, 0]],
'8': [[1, 1, 1], [1, 0, 1], [1, 1, 1], [1, 0, 1], [1, 1, 1]],
'9': [[1, 1, 1], [1, 0, 1], [1, 1, 1], [0, 0, 1], [1, 1, 1]],
' ': [[0, 0, 0], [0, 0, 0], [0, 0, 0], [0, 0, 0], [0, 0, 0]],
'-': [[0, 0, 0], [0, 0, 0], [1, 1, 1], [0, 0, 0], [0, 0, 0]],
'E': [[1, 1, 1], [1, 0, 0], [1, 1, 1], [1, 0, 0], [1, 1, 1]],
'n': [[1, 0, 1], [1, 1, 1], [1, 0, 1], [1, 0, 1], [1, 0, 1]],
'r': [[1, 1, 0], [1, 0, 1], [1, 1, 0], [1, 0, 1], [1, 0, 1]],
'g': [[1, 1, 1], [1, 0, 0], [1, 0, 0], [1, 0, 1], [1, 1, 1]],
'y': [[1, 0, 1], [1, 0, 1], [1, 1, 1], [0, 0, 1], [0, 0, 1]],
'b': [[1, 1, 0], [1, 0, 1], [1, 1, 0], [1, 0, 1], [1, 1, 0]],
'w': [[1, 0, 1], [1, 0, 1], [1, 1, 1], [1, 1, 1], [1, 0, 1]],
'k': [[1, 0, 1], [1, 1, 0], [1, 0, 1], [1, 1, 0], [1, 0, 1]],
'o': [[0, 0, 0], [0, 0, 0], [1, 1, 1], [0, 0, 0], [0, 0, 0]],
'p': [[1, 1, 1], [1, 0, 1], [1, 1, 1], [1, 0, 0], [1, 0, 0]],
'c': [[0, 0, 0], [0, 0, 0], [1, 1, 1], [0, 0, 1], [0, 0, 1]],
'm': [[1, 0, 1], [1, 1, 1], [1, 0, 1], [1, 0, 1], [1, 0, 1]],
'a': [[0, 0, 0], [0, 0, 0], [1, 1, 1], [0, 0, 1], [1, 1, 1]],
't': [[1, 1, 1], [0, 1, 0], [0, 1, 0], [0, 1, 0], [0, 1, 0]],
's': [[1, 1, 1], [1, 0, 0], [1, 1, 1], [0, 0, 1], [1, 1, 1]],
'u': [[1, 0, 1], [1, 0, 1], [1, 0, 1], [1, 0, 1], [1, 1, 1]],
'v': [[1, 0, 1], [1, 0, 1], [1, 0, 1], [1, 0, 1], [0, 1, 0]],
'l': [[1, 0, 0], [1, 0, 0], [1, 0, 0], [1, 0, 0], [1, 1, 1]],
'e': [[1, 1, 1], [1, 0, 0], [1, 1, 1], [1, 0, 0], [1, 1, 1]],
'x': [[1, 0, 1], [1, 0, 1], [0, 1, 0], [1, 0, 1], [1, 0, 1]],
'f': [[1, 1, 1], [1, 0, 0], [1, 1, 1], [1, 0, 0], [1, 0, 0]],
'h': [[1, 0, 1], [1, 0, 1], [1, 1, 1], [1, 0, 1], [1, 0, 1]],
'(': [[0, 1, 0], [1, 0, 0], [1, 0, 0], [1, 0, 0], [0, 1, 0]],
')': [[0, 1, 0], [0, 0, 1], [0, 0, 1], [0, 0, 1], [0, 1, 0]],
'.': [[0, 0, 0], [0, 0, 0], [0, 0, 0], [0, 0, 0], [0, 0, 0]],
',': [[0, 0, 0], [0, 0, 0], [0, 0, 0], [0, 1, 0], [1, 0, 0]],
':': [[0, 0, 0], [0, 1, 0], [0, 0, 0], [0, 1, 0], [0, 0, 0]],
';': [[0, 0, 0], [0, 1, 0], [0, 0, 0], [0, 1, 0], [1, 0, 0]]
}
def draw_energy(self, energy, board_size, cell_size):
# Convert the energy value to a string
energy_str = str(energy)
# extract the number from "tensor([number])"
energy_str = energy_str[energy_str.index(
'(') + 1:energy_str.index(')')]
# remove the decimal point
energy_str = energy_str.replace('.', '')
# add "Energy: " to the string
energy_str = "Energy:" + energy_str
# Create a mapping from characters to a grid of cells
char_to_pixels = {
'0': [[1, 1, 1], [1, 0, 1], [1, 0, 1], [1, 0, 1], [1, 1, 1]],
'1': [[0, 1, 0], [0, 1, 0], [0, 1, 0], [0, 1, 0], [0, 1, 0]],
'2': [[1, 1, 1], [0, 0, 1], [1, 1, 1], [1, 0, 0], [1, 1, 1]],
'3': [[1, 1, 1], [0, 0, 1], [1, 1, 1], [0, 0, 1], [1, 1, 1]],
'4': [[1, 0, 1], [1, 0, 1], [1, 1, 1], [0, 0, 1], [0, 0, 1]],
'5': [[1, 1, 1], [1, 0, 0], [1, 1, 1], [0, 0, 1], [1, 1, 1]],
'6': [[1, 1, 1], [1, 0, 0], [1, 1, 1], [1, 0, 1], [1, 1, 1]],
'7': [[1, 1, 1], [0, 0, 1], [0, 1, 0], [1, 0, 0], [1, 0, 0]],
'8': [[1, 1, 1], [1, 0, 1], [1, 1, 1], [1, 0, 1], [1, 1, 1]],
'9': [[1, 1, 1], [1, 0, 1], [1, 1, 1], [0, 0, 1], [1, 1, 1]],
' ': [[0, 0, 0], [0, 0, 0], [0, 0, 0], [0, 0, 0], [0, 0, 0]],
'-': [[0, 0, 0], [0, 0, 0], [1, 1, 1], [0, 0, 0], [0, 0, 0]],
'E': [[1, 1, 1], [1, 0, 0], [1, 1, 1], [1, 0, 0], [1, 1, 1]],
'n': [[1, 0, 1], [1, 1, 1], [1, 0, 1], [1, 0, 1], [1, 0, 1]],
'r': [[1, 1, 0], [1, 0, 1], [1, 1, 0], [1, 0, 1], [1, 0, 1]],
'g': [[1, 1, 1], [1, 0, 0], [1, 0, 0], [1, 0, 1], [1, 1, 1]],
'y': [[1, 0, 1], [1, 0, 1], [1, 1, 1], [0, 0, 1], [0, 0, 1]],
'b': [[1, 1, 0], [1, 0, 1], [1, 1, 0], [1, 0, 1], [1, 1, 0]],
'w': [[1, 0, 1], [1, 0, 1], [1, 1, 1], [1, 1, 1], [1, 0, 1]],
'k': [[1, 0, 1], [1, 1, 0], [1, 0, 1], [1, 1, 0], [1, 0, 1]],
'o': [[0, 0, 0], [0, 0, 0], [1, 1, 1], [0, 0, 0], [0, 0, 0]],
'p': [[1, 1, 1], [1, 0, 1], [1, 1, 1], [1, 0, 0], [1, 0, 0]],
'c': [[0, 0, 0], [0, 0, 0], [1, 1, 1], [0, 0, 1], [0, 0, 1]],
'm': [[1, 0, 1], [1, 1, 1], [1, 0, 1], [1, 0, 1], [1, 0, 1]],
'a': [[0, 0, 0], [0, 0, 0], [1, 1, 1], [0, 0, 1], [1, 1, 1]],
't': [[1, 1, 1], [0, 1, 0], [0, 1, 0], [0, 1, 0], [0, 1, 0]],
's': [[1, 1, 1], [1, 0, 0], [1, 1, 1], [0, 0, 1], [1, 1, 1]],
'u': [[1, 0, 1], [1, 0, 1], [1, 0, 1], [1, 0, 1], [1, 1, 1]],
'v': [[1, 0, 1], [1, 0, 1], [1, 0, 1], [1, 0, 1], [0, 1, 0]],
'l': [[1, 0, 0], [1, 0, 0], [1, 0, 0], [1, 0, 0], [1, 1, 1]],
'e': [[1, 1, 1], [1, 0, 0], [1, 1, 1], [1, 0, 0], [1, 1, 1]],
'x': [[1, 0, 1], [1, 0, 1], [0, 1, 0], [1, 0, 1], [1, 0, 1]],
'f': [[1, 1, 1], [1, 0, 0], [1, 1, 1], [1, 0, 0], [1, 0, 0]],
'h': [[1, 0, 1], [1, 0, 1], [1, 1, 1], [1, 0, 1], [1, 0, 1]],
'(': [[0, 1, 0], [1, 0, 0], [1, 0, 0], [1, 0, 0], [0, 1, 0]],
')': [[0, 1, 0], [0, 0, 1], [0, 0, 1], [0, 0, 1], [0, 1, 0]],
'.': [[0, 0, 0], [0, 0, 0], [0, 0, 0], [0, 0, 0], [0, 0, 0]],
',': [[0, 0, 0], [0, 0, 0], [0, 0, 0], [0, 1, 0], [1, 0, 0]],
':': [[0, 0, 0], [0, 1, 0], [0, 0, 0], [0, 1, 0], [0, 0, 0]],
';': [[0, 0, 0], [0, 1, 0], [0, 0, 0], [0, 1, 0], [1, 0, 0]]
}
rightest = 0
# Draw the energy value on the board
for i, char in enumerate(energy_str):
pixels = char_to_pixels[char]
for j, row in enumerate(pixels):
for k, pixel in enumerate(row):
if not pixel:
continue
x = self.pos[0] + i * 4 + k
y = self.pos[1] + j
color = (0, 255, 255) if pixel else (0, 0, 0)
pygame.draw.rect(self.screen, color, pygame.Rect(
x * cell_size, y * cell_size, cell_size, cell_size))
rightest = max(rightest, x * cell_size)
# update only after some time
time = pygame.time.get_ticks()
if time - self.last_time > 1000:
self.last_time = time
# Update position for next draw
self.pos[0] += self.direction[0] * self.speed
self.pos[1] += self.direction[1] * self.speed
# if hit right edge change direction to left + up or down
if rightest == board_size - self.cell_size * 2:
self.direction = (-1, self.direction[1])
# if hit left edge change direction to right + up or down
elif self.pos[0] <= 0:
self.direction = (1, self.direction[1])
# if hit top edge change direction to down + left or right
elif self.pos[1] <= 0:
self.direction = (random.choice([-1, 1]), 1)
# if hit bottom edge change direction to up + left or right
elif self.pos[1] * self.cell_size + 50 >= board_size:
self.direction = (random.choice([-1, 1]), -1)
def draw_done(self):
chars_to_pixels = {
'D': [[1, 1, 1, 1, 0], [1, 0, 0, 0, 1], [1, 0, 0, 0, 1], [1, 0, 0, 0, 1], [1, 1, 1, 1, 0]],
'O': [[0, 1, 1, 1, 0], [1, 0, 0, 0, 1], [1, 0, 0, 0, 1], [1, 0, 0, 0, 1], [0, 1, 1, 1, 0]],
'N': [[1, 0, 0, 0, 1], [1, 1, 0, 0, 1], [1, 0, 1, 0, 1], [1, 0, 0, 1, 1], [1, 0, 0, 0, 1]],
'E': [[1, 1, 1, 1, 1], [1, 0, 0, 0, 0], [1, 1, 1, 1, 0], [1, 0, 0, 0, 0], [1, 1, 1, 1, 1]]
}
self.done_x_pos += 1
self.done_x_pos %= QUEENS
for i, char in enumerate("DONE"):
pixels = chars_to_pixels[char]
for j, row in enumerate(pixels):
for k, pixel in enumerate(row):
if not pixel:
continue
x = i * 6 + k + self.done_x_pos # Adjusted for the new character size
y = j + 10
color = (0, 255, 0) if pixel else (0, 0, 0)
pygame.draw.rect(self.screen, color, pygame.Rect(
x * self.cell_size, y * self.cell_size, self.cell_size, self.cell_size))
def handle_events(self):
board_size = QUEENS
cell_size = self.screen.get_width() // board_size
for event in pygame.event.get():
if event.type == pygame.QUIT:
return False
elif event.type == pygame.MOUSEBUTTONDOWN:
x, y = pygame.mouse.get_pos()
# Check if the click is on the board
if 0 <= x <= board_size * cell_size and 0 <= y <= board_size * cell_size:
row, col = y // cell_size, x // cell_size
# If a queen is already selected, move it to the clicked cell
if self.missing_queens > 0 and self.board[row][col] == 0:
self.board[row][col] = 1
self.missing_queens -= 1
break
# If the clicked cell contains a queen, select it
elif self.board[row][col] == 1 and self.missing_queens < board_size:
self.missing_queens += 1
self.board[row][col] = 0
break
return True
def display_message(self, message):
font = pygame.font.Font(None, 36)
text = font.render(message, 1, (255, 255, 255))
textpos = text.get_rect(centerx=self.screen.get_width() / 2)
self.screen.blit(text, textpos)
def run(self):
self.player_turn = True
running = True
while running:
self.clock.tick(60)
self.draw_board(self.board)
pygame.display.flip()
running = self.handle_events()
pygame.quit()
if __name__ == "__main__":
gui = GameGUI()
gui.run()