From bb0a3266c7b44507ee8798120b1e287105337858 Mon Sep 17 00:00:00 2001 From: Karma Riuk Date: Thu, 30 Jan 2025 17:12:53 +0100 Subject: [PATCH] implemented queen legal moves --- src/logic/pieces/queen.py | 28 +++++++++++++++++++++++++++- 1 file changed, 27 insertions(+), 1 deletion(-) diff --git a/src/logic/pieces/queen.py b/src/logic/pieces/queen.py index e3427fa..40ccd63 100644 --- a/src/logic/pieces/queen.py +++ b/src/logic/pieces/queen.py @@ -1,5 +1,31 @@ from .piece import Piece class Queen(Piece): - pass + def legal_moves(self, board: "Board") -> list[Move]: + ret = [] + # looking north east + ret.extend(self._look_direction(board, 1, 1)) + + # looking south east + ret.extend(self._look_direction(board, 1, -1)) + + # looking south west + ret.extend(self._look_direction(board, -1, -1)) + + # looking north west + ret.extend(self._look_direction(board, -1, 1)) + + # looking east + ret.extend(self._look_direction(board, 1, 0)) + + # looking south + ret.extend(self._look_direction(board, 0, -1)) + + # looking west + ret.extend(self._look_direction(board, -1, 0)) + + # looking north + ret.extend(self._look_direction(board, 0, 1)) + + return ret