Compare commits

...

12 Commits

Author SHA1 Message Date
Karma Riuk
e95caa0015 implemented bishop's possible moves
Some checks are pending
pre-release / Pre Release (push) Waiting to run
2025-01-30 11:48:45 +01:00
Karma Riuk
bb0b8cdd27 imported colour also in gui 2025-01-30 11:48:27 +01:00
Karma Riuk
55ba824b13 fixed circular dependency 2025-01-30 11:48:03 +01:00
Karma Riuk
16d107e5ea fixed the colour that was extracted 2025-01-30 11:45:47 +01:00
Karma Riuk
baa09135ee made is_capturing a kwarg 2025-01-30 10:46:06 +01:00
Karma Riuk
eae87f353b now the legal moves of the pawn actually returns
moves
2025-01-30 10:45:43 +01:00
Karma Riuk
362b0e157d annotated the legal moves function better 2025-01-30 10:45:24 +01:00
Karma Riuk
c900ebcfa0 extracted the colour to an enum 2025-01-30 10:45:12 +01:00
Karma Riuk
c3e46017eb now the legal moves function actually returns a
list of moves
2025-01-30 10:28:41 +01:00
Karma Riuk
324484aa31 created the move class 2025-01-30 09:41:46 +01:00
Karma Riuk
eca7a6ae0c now black pawns can also go forward 2025-01-29 16:59:13 +01:00
Karma Riuk
ffe76b161a minor fixes 2025-01-29 16:55:54 +01:00
8 changed files with 125 additions and 31 deletions

View File

@ -4,14 +4,12 @@ from logic.pieces.knight import Knight
from logic.pieces.queen import Queen
from logic.pieces.rook import Rook
from logic.pieces.pawn import Pawn
from logic.pieces.piece import Piece
from logic.pieces.piece import Colour, Piece
from logic.position import Position
from typing import Type
class Board:
KING_SIDE_CASTLE = "king side castle"
QUEEN_SIDE_CASTLE = "queen side castle"
def __init__(self) -> None:
self._white: dict[Position, Piece] = {}
self._black: dict[Position, Piece] = {}
@ -41,6 +39,7 @@ class Board:
@staticmethod
def setup_FEN_position(position: str) -> "Board":
ret = Board()
index = 0
# -- Pieces
pieces = "prnbqk" # possible pieces
@ -49,15 +48,16 @@ class Board:
x = 0
y = 7 # FEN starts from the top left, so 8th rank
for c in position:
index += 1
if c == " ":
break
if c in pieces or c in pieces.upper():
pos = Position(x, y)
piece = Board._piece_class_from_char(c)
if c.isupper():
ret._white[pos] = piece(pos, Piece.WHITE)
ret._white[pos] = piece(pos, Colour.WHITE)
else:
ret._black[pos] = piece(pos, Piece.BLACK)
ret._black[pos] = piece(pos, Colour.BLACK)
x += 1
continue
@ -69,18 +69,18 @@ class Board:
# -- Active colour
index = position.find(" ") # find the first space
index += 1
if position[index] == "w":
ret._turn = Piece.WHITE
ret._turn = Colour.WHITE
elif position[index] == "b":
ret._turn = Piece.BLACK
ret._turn = Colour.BLACK
else:
raise ValueError(f"The FEN position is malformed, the active colour should be either 'w' or 'b', but is '{position[index]}'")
index += 1
# -- Castling Rights
for c in position:
for c in position[index:]:
index += 1
if c == "-" or c == " ":
break
@ -96,7 +96,6 @@ class Board:
ret._black_castling_write.add(Board.QUEEN_SIDE_CASTLE)
# -- En passant target
index = position.find(" ", index + 1)
if position[index] != "-":
ret._en_passant_target = position[index:index+2]

25
src/logic/move.py Normal file
View File

@ -0,0 +1,25 @@
from logic.pieces.piece import Piece
from logic.position import Position
from enum import Enum
class Move:
def __init__(self, is_capturing: bool) -> None:
self.is_capturing = is_capturing
def to_algebraic(self) -> str:
raise NotImplementedError("The move can't be translated to algbraic notation, as it was not implemented")
@staticmethod
def from_algebraic(move: str) -> "Move":
raise NotImplementedError("The move can't be translated from algbraic notation, as it was not implemented")
class PieceMove(Move):
def __init__(self, piece: Piece, pos: Position,/, is_capturing: bool = False) -> None:
super().__init__(is_capturing)
self.piece = piece
self.pos = pos
class Castle(Move, Enum):
KING_SIDE_CASTLE = False
QUEEN_SIDE_CASTLE = False

View File

@ -1,4 +1,52 @@
from logic.move import Move, PieceMove
from logic.position import Position
from .piece import Piece
class Bishop(Piece):
pass
def _move_for_position(self, board: "Board", x: int, y: int) -> Move | None:
if not Position.is_within_bounds(x, y):
return None
piece = board.piece_at(x, y)
if piece is None:
return PieceMove(self, Position(x, y))
if piece.colour != self.colour:
return PieceMove(self, Position(x, y), is_capturing=True)
return None
def _look_direction(self, board: "Board", mult_dx: int, mult_dy: int):
ret = []
for d in range(1, 8):
dx = mult_dx * d
dy = mult_dy * d
move = self._move_for_position(board, self.pos.x + dx, self.pos.y + dy)
if move is None:
break
ret.append(move)
if move.is_capturing:
break
return ret
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))
return ret

View File

@ -1,8 +1,9 @@
from logic.position import Position
from logic.move import Move, PieceMove
from logic.pieces.piece import Piece
from logic.position import Position
class Pawn(Piece):
def legal_moves(self, board) -> list[Position]:
def legal_moves(self, board) -> list[Move]:
ret = []
# can we capture to the left?
@ -12,7 +13,7 @@ class Pawn(Piece):
(self.colour == self.BLACK and (capturable_piece := board.piece_at(self.pos.x - 1, self.pos.y - 1)))
):
if capturable_piece.colour != self.colour:
ret.append(capturable_piece.pos)
ret.append(PieceMove(self, capturable_piece.pos, is_capturing = True))
# can we capture to the right?
if self.pos.x < 7 and (
@ -21,11 +22,21 @@ class Pawn(Piece):
(self.colour == self.BLACK and (capturable_piece := board.piece_at(self.pos.x + 1, self.pos.y - 1)))
):
if capturable_piece.colour != self.colour:
ret.append(capturable_piece.pos)
ret.append(PieceMove(self, capturable_piece.pos, is_capturing = True))
for dy in range(1, 3 if self.pos.y == 1 else 2):
if self.pos.y + dy > 7 or board.piece_at(self.pos.x, self.pos.y + dy):
break
ret.append(Position(self.pos.x, self.pos.y + dy))
if self.colour == Piece.WHITE:
for dy in range(1, 3 if self.pos.y == 1 else 2):
if self.pos.y + dy > 7 or board.piece_at(self.pos.x, self.pos.y + dy):
break
pos = Position(self.pos.x, self.pos.y + dy)
ret.append(PieceMove(self, pos))
else:
for dy in range(1, 3 if self.pos.y == 6 else 2):
if self.pos.y - dy < 0 or board.piece_at(self.pos.x, self.pos.y - dy):
break
pos = Position(self.pos.x, self.pos.y - dy)
ret.append(PieceMove(self, pos))
print(ret)
return ret

View File

@ -1,17 +1,19 @@
from logic.position import Position
from enum import Enum
class Piece:
class Colour(Enum):
WHITE = "white"
BLACK = "black"
def __init__(self, pos, colour) -> None:
class Piece:
def __init__(self, pos: Position, colour: Colour) -> None:
self.pos = pos
assert colour == self.WHITE or colour == self.BLACK, "The colour of the piece must be either Piece.WHITE or Piece.BLACK"
assert colour == Colour.WHITE or colour == Colour.BLACK, "The colour of the piece must be either Piece.WHITE or Piece.BLACK"
self.colour = colour
def position(self) -> Position:
return self.pos
def legal_moves(self, board) -> list[Position]:
def legal_moves(self, board: "Board") -> list["Move"]:
raise NotImplementedError(f"Can't say what the legal moves are for {type(self).__name__}, the method hasn't been implemented yet")

View File

@ -9,6 +9,12 @@ class Position:
self.x = x
self.y = y
@staticmethod
def is_within_bounds(x: int, y: int) -> bool:
return x >= Position._MIN_POS and x <= Position._MAX_POS \
and y >= Position._MIN_POS and y <= Position._MAX_POS
def __eq__(self, value: object, /) -> bool:
if type(value) != type(self):
return False

View File

@ -1,9 +1,9 @@
from logic.board import Board, create_board
from logic.board import Board
from view.gui import GUI
from view.tui import TUI
if __name__ == "__main__":
initial_board_position = "rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w"
initial_board_position = "rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1"
board = Board.setup_FEN_position(initial_board_position)
view = GUI(board)

View File

@ -1,7 +1,7 @@
import tkinter as tk
from logic.board import Board
from logic.pieces.piece import Piece
from logic.pieces.piece import Colour, Piece
from logic.position import Position
from view.view import View
@ -30,21 +30,24 @@ class GUI(View):
for y in range(8):
for x in range(8):
colour = colours[(x + y) % 2]
if self.state["selected_piece"] and Position(x, 7-y) in self.state["legal_moves"]:
colour = "#ADD8E6" # Highlight legal moves
if self.state["selected_piece"]:
possible_positions = [move.pos for move in self.state["legal_moves"]]
if Position(x, 7-y) in possible_positions:
colour = "#ADD8E6" # Highlight legal moves
self.canvas.create_rectangle(
x * self.tile_size,
y * self.tile_size,
(x + 1) * self.tile_size,
(y + 1) * self.tile_size,
fill=colour
fill=colour,
outline=colour,
)
piece = self.board.piece_at(x, 7-y)
if piece:
text_colour = "white" if piece.colour == Piece.WHITE else "black"
text_colour = "white" if piece.colour == Colour.WHITE else "black"
self.canvas.create_text(
(x + 0.5) * self.tile_size,
(y + 0.5) * self.tile_size,