rudism
/
zeropod
Archived
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

handle game over state

This commit is contained in:
Rudis Muiznieks 2022-04-30 11:00:14 -05:00
parent 08bd3e835c
commit 0d5a55088f
Signed by: rudism
GPG Key ID: CABF2F86EF7884F9
2 changed files with 56 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
plugin/chess/draw.py
View File

@ -1,6 +1,7 @@
from typing import Optional
from icecream import ic
from graphics import Graphics
from chess import Board, Color, WHITE
from chess import Board, Outcome, Color, WHITE
class Draw:
BOARD_SIZE = 64
@ -79,11 +80,15 @@ class Draw:
self._graphics.fill_rect(
self.BOARD_SIZE, 0, self.BOARD_SIZE, self.BOARD_SIZE, 0)
def _format_move(self, move: str):
if len(move) == 4:
return move.upper()[:2] + "-" + move.upper()[2:]
return move
def draw_board(self, board: Board, player_color: Color):
self._graphics.fill_rect(0, 0, self.BOARD_SIZE, self.BOARD_SIZE, 0)
nb = "".join(str(board).split()).replace(" ", "")
nb = nb if player_color == WHITE else nb[::-1]
ic(player_color, nb)
c = True
for row in range(8):
for col in range(8):
@ -109,7 +114,7 @@ class Draw:
def draw_thinking(self, last_move: str):
self._clear_info()
self._graphics.text(last_move, 12, 0, 1)
self._graphics.text(self._format_move(last_move), 12, 0, 1)
self._graphics.text("Shh, I'm", 12, 2, 1)
self._graphics.text("thinking!", 12, 3, 1)
self._graphics.show()
@ -124,10 +129,22 @@ class Draw:
def draw_select(self, last_move: str, src: str, dst = None):
self._clear_info()
self._graphics.text(last_move, 12, 0, 1)
self._graphics.text(self._format_move(last_move), 12, 0, 1)
self._graphics.text("You move:", 12, 2, 1)
if dst == None:
self._graphics.text("<" + src + "> - __", 12, 3, 1)
self._graphics.text("<" + src.upper() + "> - __", 12, 3, 1)
else:
self._graphics.text(src + " - <" + dst + ">", 12, 3, 1)
self._graphics.text(src.upper() + " - <" + dst.upper() + ">", 12, 3, 1)
self._graphics.show()
def draw_game_over(self, outcome: Optional[Outcome]):
self._clear_info()
self._graphics.text("Game over", 12, 0, 1)
if outcome != None:
if outcome.winner == None:
self._graphics.text("Draw!", 12, 2, 1)
else:
self._graphics.text("Winner:", 12, 2, 1)
self._graphics.text("White!" if outcome.winner == WHITE else "Black!",
12, 3, 1)
self._graphics.show()

48
plugin/chess/game.py
View File

@ -54,6 +54,26 @@ class ChessGame:
self._sf.make_moves_from_current_position([move])
self._draw.draw_board(self._board, self._player_color)
def _get_sources(self):
if len(list(self._all_moves)) == 0:
moves = self._board.generate_legal_moves()
mvarray = map(lambda m: [Move.uci(m)[0:2], Move.uci(m)[2:4]], moves)
for src, dst in list(mvarray):
self._all_moves.setdefault(src, []).append(dst)
if len(list(self._all_moves)) > 0:
return sorted(
list(self._all_moves),
key=lambda x: x[::-1])
return list[str]()
def _get_dests(self):
if self._src_idx < len(list(self._all_moves)):
src = self._get_sources()[self._src_idx]
return sorted(
self._all_moves[src],
key=lambda x: x[::-1])
return list[str]()
def run(self):
# either load the save game or start a new one
self._load_saved_or_init()
@ -66,6 +86,9 @@ class ChessGame:
# HANDLE STATE #
################
if self._board.is_game_over() and self._state != GameState.MAIN_MENU:
self._state = GameState.GAME_OVER
# draw menu
if self._state == GameState.MAIN_MENU:
self._draw.draw_menu(self._menu_index)
@ -74,7 +97,6 @@ class ChessGame:
# computer makes a move
elif self._state == GameState.THINKING:
self._draw.draw_thinking(self._move)
# TODO: check game over
move = self._sf.get_best_move_time(self.STOCKFISH_MOVE_TIME * 1000)
if move != None:
self._make_move(move)
@ -88,13 +110,9 @@ class ChessGame:
# user picks source piece
elif self._state == GameState.CHOOSE_SRC:
if len(list(self._all_moves)) == 0:
moves = self._board.generate_legal_moves()
mvarray = map(lambda m: [Move.uci(m)[0:2], Move.uci(m)[2:4]], moves)
for src, dst in list(mvarray):
self._all_moves.setdefault(src, []).append(dst)
if len(list(self._all_moves)) > 0:
src = list(self._all_moves)[self._src_idx]
srces = self._get_sources()
if len(srces) > 0:
src = srces[self._src_idx]
self._draw.draw_select(self._move, src)
key = self._cinput.get_one_shot()
else:
@ -102,13 +120,14 @@ class ChessGame:
# user picks dest square
elif self._state == GameState.CHOOSE_DST:
src = list(self._all_moves)[self._src_idx]
dst = self._all_moves[src][self._dst_idx]
src = self._get_sources()[self._src_idx]
dst = self._get_dests()[self._dst_idx]
self._draw.draw_select(self._move, src, dst)
key = self._cinput.get_one_shot()
# game has ended
elif self._state == GameState.GAME_OVER:
else:
self._draw.draw_game_over(self._board.outcome())
key = self._cinput.get_one_shot()
################
@ -162,6 +181,7 @@ class ChessGame:
# move to destination picker
elif key == Button.BTN_B:
self._dst_idx = 0
self._state = GameState.CHOOSE_DST
# handle user input when selecting dest piece
@ -183,14 +203,12 @@ class ChessGame:
# make the move
elif key == Button.BTN_B:
dst = self._all_moves[src][self._dst_idx]
src = self._get_sources()[self._src_idx]
dst = self._get_dests()[self._dst_idx]
self._make_move(src + dst)
# TODO: check game over
self._state = GameState.THINKING
# handle user input on game over
else:
if key == Button.BTN_A or key == Button.BTN_B:
self._state = GameState.MAIN_MENU