pos = pygame.mouse.get_pos()
grid.mark_mine(pos[0]//30, pos[1]//30)
if grid.check_if_win():
state = States.win
if event.type == pygame.KEYDOWN:
if event.key == pygame.K_SPACE and (state == States.game_over or state == States.win):
grid.reload()
state = States.running
if event.key == pygame.K_b:
grid.show_mines()
surface.fill((0,0,0))
if player.get_health() == 0:
state = States.game_over
if state == States.game_over:
Stats.draw(surface, 'Game over!', (970, 350))
Stats.draw(surface, 'Press Space to restart', (920, 400))
elif state == States.win:
Stats.draw(surface, 'You win!', (1000, 350))
Stats.draw(surface, 'Press Space to restart', (920, 400))
grid.draw(surface)
Stats.draw(surface, 'Lives remaining', (950, 100))
Stats.draw(surface, str(player.get_health()), (1020, 200))
Stats.draw(surface, 'RMB to mark mine', (950, 550))
Stats.draw(surface, 'press b to show mines', (920, 650))
pygame.display.flip()
class Game():
def __init__(self):
pygame.init()
self.game_display = pygame.display.set_mode(
(stg.display_x, stg.display_y))
pygame.display.set_caption(stg.display_title)
self.tf_model = tf.keras.models.load_model('num_reader.model')
self.running = True
self.guessing = False
self.drawing = True
self.clock = pygame.time.Clock()
self.click = False
self.prediction_text = None
def new_board(self):
if self.running:
self.guessing = False
self.drawing = True
self.board = Grid()
def new_guess(self):
self.info_text1 = gui.TextWindow(
stg.text3_x, stg.text3_y, stg.text3_w, stg.text3_h, stg.text3_text_color, stg.text3_text, stg.text3_font)
self.info_text2 = gui.TextWindow(
stg.text4_x, stg.text4_y, stg.text4_w, stg.text4_h, stg.text4_text_color, stg.text4_text, stg.text4_font)
while self.running:
self.new_board()
self.run()
def run(self):
self.playing = True
while self.playing:
self.clock.tick(60)
self.events()
self.update()
self.draw()
def events(self):
for event in pygame.event.get():
if event.type == pygame.QUIT:
self.playing = False
self.running = False
pygame.quit()
sys.exit()
if event.type == pygame.MOUSEBUTTONDOWN:
self.click = True
if event.type == pygame.MOUSEBUTTONUP:
self.click = False
self.mouse_pos = pygame.mouse.get_pos()
self.keys = pygame.key.get_pressed()
if self.keys[pygame.K_ESCAPE]:
self.playing = False
self.running = False
pygame.quit()
sys.exit()
if self.keys[pygame.K_SPACE]:
self.drawing = False
self.guessing = True
if self.keys[pygame.K_c]:
self.new_board()
def update(self):
if self.drawing:
if self.click:
self.board.update()
elif self.guessing:
self.guess()
def draw(self):
self.game_display.fill(stg.BG_COLOR)
self.board.draw(self.game_display)
if self.drawing:
self.info_text1.draw(self.game_display)
elif self.prediction_text:
self.prediction_text.draw(self.game_display)
self.info_text2.draw(self.game_display)
pygame.display.update()
def guess(self):
# self.data = self.overwriting_data()
self.data = np.reshape(self.board.grid, (-1, 28, 28))
self.predictions = self.tf_model.predict(self.data)
self.prediction = (np.argmax(self.predictions[0]))
self.prediction_text = gui.TextWindow(stg.text2_x, stg.text2_y, stg.text2_w, stg.text2_h,
stg.text2_text_color, (f'{stg.text2_text}{self.prediction}'), stg.text2_font)
self.guessing = False