64 lines
2.3 KiB
Python
64 lines
2.3 KiB
Python
from logic.board import Board
|
|
from logic.move import Move
|
|
from logic.pieces.piece import Piece
|
|
from logic.position import Position
|
|
from view.view import View
|
|
|
|
|
|
class Controller:
|
|
def __init__(self, board: Board, view: View) -> None:
|
|
self._board = board
|
|
self._view = view
|
|
|
|
self._view.set_controller(self)
|
|
self._reset_selection()
|
|
|
|
self._selected_piece: Piece = None
|
|
self._legal_moves: list[Move] = []
|
|
|
|
def _reset_selection(self):
|
|
self._selected_piece = None
|
|
self._legal_moves = []
|
|
self._view.update_board(self._board, self._selected_piece, self._legal_moves)
|
|
|
|
|
|
def _show_legal_moves(self, pos: Position):
|
|
piece = self._board.piece_at(pos.x, pos.y)
|
|
|
|
if piece:
|
|
if piece.colour != self._board._turn:
|
|
return
|
|
self._selected_piece = piece
|
|
self._legal_moves = piece.legal_moves(self._board)
|
|
self._view.update_board(self._board, self._selected_piece, self._legal_moves)
|
|
else:
|
|
self._reset_selection()
|
|
|
|
def _make_move(self, move: Move) -> None:
|
|
self._board = self._board.make_move(move)
|
|
self._reset_selection()
|
|
|
|
def on_tile_selected(self, x: int, y: int) -> None:
|
|
pos = Position(x, y)
|
|
|
|
piece = self._board.piece_at(x, y)
|
|
|
|
if self._selected_piece is None \
|
|
or (piece is not None and piece != self._selected_piece and piece.colour == self._selected_piece.colour):
|
|
self._show_legal_moves(pos)
|
|
else:
|
|
legal_moves_positions = [move for move in self._legal_moves if move.pos == pos]
|
|
assert len(legal_moves_positions) <= 1, f"Apparently we can make multiple moves towards {pos.to_algebraic()} with {type(self._selected_piece)}, which doesn't make sense..."
|
|
|
|
if len(legal_moves_positions) == 0: # click on a square outside of the possible moves
|
|
self._reset_selection()
|
|
else:
|
|
move = legal_moves_positions[0]
|
|
self._make_move(move)
|
|
|
|
if self._board.is_checkmate_for(self._board._turn):
|
|
self._view.notify_checkmate(self._board._turn)
|
|
|
|
if self._board.is_stalemate_for(self._board._turn):
|
|
self._view.notify_stalemate(self._board._turn)
|