Compare commits
12 Commits
06f78487d9
...
e95caa0015
Author | SHA1 | Date | |
---|---|---|---|
|
e95caa0015 | ||
|
bb0b8cdd27 | ||
|
55ba824b13 | ||
|
16d107e5ea | ||
|
baa09135ee | ||
|
eae87f353b | ||
|
362b0e157d | ||
|
c900ebcfa0 | ||
|
c3e46017eb | ||
|
324484aa31 | ||
|
eca7a6ae0c | ||
|
ffe76b161a |
@ -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
25
src/logic/move.py
Normal 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
|
@ -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
|
||||
|
||||
|
@ -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
|
||||
|
@ -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")
|
||||
|
@ -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
|
||||
|
@ -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)
|
||||
|
@ -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,
|
||||
|
Loading…
x
Reference in New Issue
Block a user