def Move(self, row, col):
"Move king to coordinates (row, col)."
from pieces.black_funcs import Black_Funcs
Black_ = Black_Funcs(None)
if (self.row, self.col) != (-1, -1):
Black.blocks[self.row][self.col] = 0
Black.blocks[row][col] = 1
self.col = col
self.row = row
self.x = Piece.paddingx + col * UNIT
self.y = Piece.paddingy + row * UNIT
if Black_.get_freeze() == False:
self.moved = 1
def Move(self, i, row, col):
"Move rook i to coordinates (row, col)."
from pieces.black_funcs import Black_Funcs
Black_ = Black_Funcs(None)
if (self.row[i], self.col[i]) != (-1, -1):
Black.blocks[self.row[i]][self.col[i]] = 0
Black.blocks[row][col] = 1
self.col[i] = col
self.row[i] = row
self.x[i] = Piece.paddingx + col * UNIT
self.y[i] = Piece.paddingy + row * UNIT
if Black_.get_freeze() == False:
self.moved[i] = 1
def build_check_movelist(self):
"""Black king has been checked by a bishop. Build and return a movelist going
in the direction of the black king.
"""
from pieces.black_funcs import Black_Funcs
Black_ = Black_Funcs(None)
black_king_row = Black_.get_pos("K")[0]
black_king_col = Black_.get_pos("K")[1]
movelist = []
for i in range(2):
if White.checker != "B" + str(i):
continue
row = self.row[i]
col = self.col[i]
# king in lower right (relative to bishop)
if black_king_row > self.row[i] and black_king_col > self.col[i]:
while row <= 7 and col <= 7:
if (row, col) == (black_king_row, black_king_col):
return movelist
movelist.append((row, col))
row += 1
col += 1
# lower left
elif black_king_row > self.row[i] and black_king_col < self.col[i]:
while row <= 7 and col >= 0:
if (row, col) == (black_king_row, black_king_col):
return movelist
movelist.append((row, col))
row += 1
col -= 1
# upper left
elif black_king_row < self.row[i] and black_king_col < self.col[i]:
while row >= 0 and col >= 0:
if (row, col) == (black_king_row, black_king_col):
return movelist
movelist.append((row, col))
row -= 1
col -= 1
# upper right
elif black_king_row < self.row[i] and black_king_col > self.col[i]:
while row >= 0 and col <= 7:
if (row, col) == (black_king_row, black_king_col):
return movelist
movelist.append((row, col))
row -= 1
col += 1
def get_pinned_piece(self, dir):
"A piece has been pinned in direction dir. Get that piece."
from pieces.black_funcs import Black_Funcs
Black_ = Black_Funcs(None)
for i in range(2):
if self.in_path[i] == 0 or self.alive[i] == 0:
continue
if dir == "UL":
row = self.row[i]
col = self.col[i]
while row >= 0 and col >= 0:
if Black.blocks[row][col] == 1 and (
row, col) != Black_.get_pos("K"):
return Black_.get_piece(row, col)
row -= 1
col -= 1
elif dir == "UR":
row = self.row[i]
col = self.col[i]
while row >= 0 and col <= 7:
if Black.blocks[row][col] == 1 and (
row, col) != Black_.get_pos("K"):
return Black_.get_piece(row, col)
row -= 1
col += 1
elif dir == "LR":
row = self.row[i]
col = self.col[i]
while row <= 7 and col <= 7:
if Black.blocks[row][col] == 1 and (
row, col) != Black_.get_pos("K"):
return Black_.get_piece(row, col)
row += 1
col += 1
elif dir == "LL":
row = self.row[i]
col = self.col[i]
while row <= 7 and col >= 0:
if Black.blocks[row][col] == 1 and (
row, col) != Black_.get_pos("K"):
return Black_.get_piece(row, col)
row += 1
col -= 1
def update_movelist(self):
from pieces.black_funcs import Black_Funcs
Black_ = Black_Funcs(None)
for i in range(8):
self.movelist[i].clear()
self.hit_movelist[i].clear()
if self.alive[i] == 0:
continue
# normal move (1 forward)
if self.row[i] - 1 >= 0:
if Black.blocks[self.row[i]-1][self.col[i]] == 0 and \
White.blocks[self.row[i]-1][self.col[i]] == 0:
self.movelist[i].append((self.row[i] - 1, self.col[i]))
# opponent on left
if self.row[i] - 1 >= 0 and self.col[i] - 1 >= 0:
self.hit_movelist[i].append((self.row[i] - 1, self.col[i] - 1))
if Black.blocks[self.row[i] - 1][self.col[i] - 1] == 1:
self.movelist[i].append((self.row[i] - 1, self.col[i] - 1))
# opponent on right
if self.row[i] - 1 >= 0 and self.col[i] + 1 <= 7:
self.hit_movelist[i].append((self.row[i] - 1, self.col[i] + 1))
if Black.blocks[self.row[i] - 1][self.col[i] + 1] == 1:
self.movelist[i].append((self.row[i] - 1, self.col[i] + 1))
# initial double-step
if self.row[i] == 6 and Black.blocks[self.row[i]-1][self.col[i]] == 0 and \
White.blocks[self.row[i]-1][self.col[i]] == 0 and \
Black.blocks[self.row[i]-2][self.col[i]] == 0 and \
White.blocks[self.row[i]-2][self.col[i]] == 0:
self.movelist[i].append((self.row[i] - 2, self.col[i]))
# en passant
if self.row[i] == 3:
for k in range(len(Black.en_passant)):
if Black.en_passant[k] == 1:
pos = Black_.get_pos("P" + str(k))
if abs(pos[1] - self.col[i]) == 1:
self.movelist[i].append((pos[0] - 1, pos[1]))
def check_insuffience_mat(self):
"""A draw by insufficient material occurs if both sides have any of:
- A lone king
- A king and bishop
- A king and knight
- A king and 2 knights
and no pawns.
"""
from pieces.black_funcs import Black_Funcs
Black_ = Black_Funcs(None)
from pieces.white_funcs import White_Funcs
White_ = White_Funcs()
if (Black_.lone_king() or Black_.one_bishop() or Black_.two_knights() or \
Black_.one_knight()) and (White_.lone_king() or White_.one_bishop() or \
White_.two_knights() or White_.one_knight()):
return True
return False
def king_in_path(self):
"""Determine if king is in path of bishop. Used for pinning.
Returns "UL", "UR", "LL", "LR" for where the king's position is relative
to the bishop (upper left, upper right, lower left, lower right).
Also sets the in_path flag for the bishop to 1.
"""
from pieces.black_funcs import Black_Funcs
Black_ = Black_Funcs(None)
for i in range(2):
if self.alive[i] == 0:
continue
self.in_path[i] = 0
row = self.row[i]
col = self.col[i]
while row >= 0 and col >= 0:
if (row, col) == Black_.get_pos("K"):
self.in_path[i] = 1
return "UL"
row -= 1
col -= 1
row = self.row[i]
col = self.col[i]
while row >= 0 and col <= 7:
if (row, col) == Black_.get_pos("K"):
self.in_path[i] = 1
return "UR"
row -= 1
col += 1
row = self.row[i]
col = self.col[i]
while row <= 7 and col <= 7:
if (row, col) == Black_.get_pos("K"):
self.in_path[i] = 1
return "LR"
row += 1
col += 1
row = self.row[i]
col = self.col[i]
while row <= 7 and col >= 0:
if (row, col) == Black_.get_pos("K"):
self.in_path[i] = 1
return "LL"
row += 1
col -= 1
def king_in_path(self):
"""Determine if king is in path of queen. Used for pinning.
Assigns a list of directions to self.in_path for where the king's
position is relative to the queen (see self.dir).
"""
from pieces.black_funcs import Black_Funcs
Black_ = Black_Funcs(None)
for i in range(White.num_queens):
if self.alive[i] == 0:
continue
self.in_path[i] = 0
col = self.col[i]
row = self.row[i]
# Rook part
while col >= 0:
if (row, col) == Black_.get_pos("K"):
self.in_path[i] = "L"
col -= 1
col = self.col[i]
row = self.row[i]
while col <= 7:
if (row, col) == Black_.get_pos("K"):
self.in_path[i] = "R"
col += 1
col = self.col[i]
row = self.row[i]
while row >= 0:
if (row, col) == Black_.get_pos("K"):
self.in_path[i] = "U"
row -= 1
col = self.col[i]
row = self.row[i]
while row <= 7:
if (row, col) == Black_.get_pos("K"):
self.in_path[i] = "D"
row += 1
# Bishop part
row = self.row[i]
col = self.col[i]
while row >= 0 and col >= 0:
if (row, col) == Black_.get_pos("K"):
self.in_path[i] = "UL"
row -= 1
col -= 1
row = self.row[i]
col = self.col[i]
while row >= 0 and col <= 7:
if (row, col) == Black_.get_pos("K"):
self.in_path[i] = "UR"
row -= 1
col += 1
row = self.row[i]
col = self.col[i]
while row <= 7 and col <= 7:
if (row, col) == Black_.get_pos("K"):
self.in_path[i] = "LR"
row += 1
col += 1
row = self.row[i]
col = self.col[i]
while row <= 7 and col >= 0:
if (row, col) == Black_.get_pos("K"):
self.in_path[i] = "LL"
row += 1
col -= 1
WhiteRook = White_Rook()
WhiteQueen = White_Queen()
WhiteKnight = White_Knight()
from pieces.black.blackking import Black_King
from pieces.white.whiteking import White_King
BlackKing = Black_King()
WhiteKing = White_King()
from board import Game_Board
Board = Game_Board()
from pieces.white_funcs import White_Funcs
from pieces.black_funcs import Black_Funcs
White_ = White_Funcs()
Black_ = Black_Funcs(White_)
from ai_helper import Helper
helper = Helper()
from black_ai import Black_AI
BlackAI = Black_AI()
from white_ai import White_AI
WhiteAI = White_AI()
def init():
if env == "" or env == None:
return
elif env == "only_pawns0":
Black_.destroy("Q0")
for i in range(2):
def build_check_movelist(self):
"""Black king has been checked by queen. Build and return a movelist going
in the direction of the black king.
"""
from pieces.black_funcs import Black_Funcs
Black_ = Black_Funcs(None)
black_king_row = Black_.get_pos("K")[0]
black_king_col = Black_.get_pos("K")[1]
movelist = []
for i in range(White.num_queens):
if White.checker != "Q" + str(i):
continue
row = self.row[i]
col = self.col[i]
# * rook part *
# king is above queen
if black_king_row < self.row[i] and black_king_col == self.col[i]:
while row >= 0:
if (row, col) == (black_king_row, black_king_col):
return movelist
movelist.append((row, col))
row -= 1
# below
elif black_king_row > self.row[i] and black_king_col == self.col[i]:
while row <= 7:
if (row, col) == (black_king_row, black_king_col):
return movelist
movelist.append((row, col))
row += 1
# left
elif black_king_col < self.col[i] and black_king_row == self.row[i]:
while col >= 0:
if (row, col) == (black_king_row, black_king_col):
return movelist
movelist.append((row, col))
col -= 1
# right
elif black_king_col > self.col[i] and black_king_row == self.row[i]:
while col <= 7:
if (row, col) == (black_king_row, black_king_col):
return movelist
movelist.append((row, col))
col += 1
# * bishop part *
# lower right
elif black_king_row > self.row[i] and black_king_col > self.col[i]:
while row <= 7 and col <= 7:
if (row, col) == (black_king_row, black_king_col):
return movelist
movelist.append((row, col))
row += 1
col += 1
# lower left
elif black_king_row > self.row[i] and black_king_col < self.col[i]:
while row <= 7 and col >= 0:
if (row, col) == (black_king_row, black_king_col):
return movelist
movelist.append((row, col))
row += 1
col -= 1
# upper left
elif black_king_row < self.row[i] and black_king_col < self.col[i]:
while row >= 0 and col >= 0:
if (row, col) == (black_king_row, black_king_col):
return movelist
movelist.append((row, col))
row -= 1
col -= 1
# upper right
elif black_king_row < self.row[i] and black_king_col > self.col[i]:
while row >= 0 and col <= 7:
if (row, col) == (black_king_row, black_king_col):
return movelist
movelist.append((row, col))
row -= 1
col += 1
def undo(self):
"Undo the last move played. Used for AI move generation in tree traversal."
from pieces.black_funcs import Black_Funcs
Black_ = Black_Funcs(None)
from pieces.white_funcs import White_Funcs
White_ = White_Funcs()
if len(self.moves["piece"]) == 0:
print("Nothing to undo!")
return
if (Black.ai or White.ai) and self.freeze and len(
self.moves["piece"]) <= self.total_moves:
print("AI undo'ing too far") if verbose else None
return
if (self.checkmate or self.stalemate
or self.insufficient_mat) and self.freeze == False:
self.checkmate = False
self.stalemate = False
self.insufficient_mat = False
(row, col) = (self.moves["pos"][-1][0], self.moves["pos"][-1][1])
piece = self.moves["piece"][-1]
# * Black *
# move back
if self.moves["color"][-1] == "B":
if piece == "K":
BlackKing.Move(row, col)
elif piece == "CK":
BlackKing.Move(0, 4)
BlackRook.Move(1, 0, 7)
Black.castled = 0
BlackRook.moved[1] = 0
BlackKing.moved = 0
elif piece == "CQ":
BlackKing.Move(0, 4)
BlackRook.Move(0, 0, 0)
Black.castled = 0
BlackRook.moved[0] = 0
BlackKing.moved = 0
for i in range(Black.num_queens):
if piece == "Q" + str(i):
BlackQueen.Move(i, row, col)
for i in range(8):
if piece == "P" + str(i):
if row == 6:
# use black queen pos since it was promoted to q
Black.blocks[BlackQueen.row[-1]][
BlackQueen.col[-1]] = 0
Black_.unpromote(i, row, col)
else:
BlackPawn.Move(i, row, col)
for i in range(2):
if piece == "B" + str(i):
BlackBishop.Move(i, row, col)
elif piece == "R" + str(i):
BlackRook.Move(i, row, col)
elif piece == "N" + str(i):
BlackKnight.Move(i, row, col)
# revive if piece killed
if self.moves["killed"][-1] == True:
if self.killed["piece"][-1] == "K":
WhiteKing.alive = 1
WhiteKing.Move(self.killed["pos"][-1][0],
self.killed["pos"][-1][1])
for i in range(White.num_queens):
if self.killed["piece"][-1] == "Q" + str(i):
WhiteQueen.alive[i] = 1
WhiteQueen.Move(i, self.killed["pos"][-1][0],
self.killed["pos"][-1][1])
for i in range(8):
if self.killed["piece"][-1] == "P" + str(i):
WhitePawn.alive[i] = 1
WhitePawn.Move(i, self.killed["pos"][-1][0],
self.killed["pos"][-1][1])
for i in range(2):
if self.killed["piece"][-1] == "B" + str(i):
WhiteBishop.alive[i] = 1
WhiteBishop.Move(i, self.killed["pos"][-1][0],
self.killed["pos"][-1][1])
elif self.killed["piece"][-1] == "N" + str(i):
WhiteKnight.alive[i] = 1
WhiteKnight.Move(i, self.killed["pos"][-1][0],
self.killed["pos"][-1][1])
elif self.killed["piece"][-1] == "R" + str(i):
WhiteRook.alive[i] = 1
WhiteRook.Move(i, self.killed["pos"][-1][0],
self.killed["pos"][-1][1])
self.killed["piece"].pop(-1)
self.killed["pos"].pop(-1)
self.update_all_moves()
Black_.check_white_pin()
if Black_.check_white_check():
White_.update_check_movelists()
Black_.reset_enpassant()
self.moves["color"].pop(-1)
self.moves["piece"].pop(-1)
self.moves["pos"].pop(-1)
self.moves["killed"].pop(-1)
# * White *
# move back
elif self.moves["color"][-1] == "W":
if piece == "K":
WhiteKing.Move(row, col)
elif piece == "CK":
WhiteKing.Move(7, 4)
WhiteRook.Move(1, 7, 7)
White.castled = 0
WhiteRook.moved[1] = 0
WhiteKing.moved = 0
elif piece == "CQ":
WhiteKing.Move(7, 4)
WhiteRook.Move(0, 7, 0)
White.castled = 0
WhiteRook.moved[0] = 0
WhiteKing.moved = 0
for i in range(White.num_queens):
if piece == "Q" + str(i):
WhiteQueen.Move(i, row, col)
for i in range(8):
if piece == "P" + str(i):
if row == 1:
# use white queen pos since it was promoted to q
White.blocks[WhiteQueen.row[-1]][
WhiteQueen.col[-1]] = 0
White_.unpromote(i, row, col)
else:
WhitePawn.Move(i, row, col)
for i in range(2):
if piece == "B" + str(i):
WhiteBishop.Move(i, row, col)
elif piece == "R" + str(i):
WhiteRook.Move(i, row, col)
elif piece == "N" + str(i):
WhiteKnight.Move(i, row, col)
# revive if piece killed
if self.moves["killed"][-1] == True:
if self.killed["piece"][-1] == "K":
BlackKing.alive = 1
BlackKing.Move(self.killed["pos"][-1][0],
self.killed["pos"][-1][1])
for i in range(Black.num_queens):
if self.killed["piece"][-1] == "Q" + str(i):
BlackQueen.alive[i] = 1
BlackQueen.Move(i, self.killed["pos"][-1][0],
self.killed["pos"][-1][1])
for i in range(8):
if self.killed["piece"][-1] == "P" + str(i):
BlackPawn.alive[i] = 1
BlackPawn.Move(i, self.killed["pos"][-1][0],
self.killed["pos"][-1][1])
for i in range(2):
if self.killed["piece"][-1] == "B" + str(i):
BlackBishop.alive[i] = 1
BlackBishop.Move(i, self.killed["pos"][-1][0],
self.killed["pos"][-1][1])
elif self.killed["piece"][-1] == "N" + str(i):
BlackKnight.alive[i] = 1
BlackKnight.Move(i, self.killed["pos"][-1][0],
self.killed["pos"][-1][1])
elif self.killed["piece"][-1] == "R" + str(i):
BlackRook.alive[i] = 1
BlackRook.Move(i, self.killed["pos"][-1][0],
self.killed["pos"][-1][1])
self.killed["piece"].pop(-1)
self.killed["pos"].pop(-1)
self.update_all_moves()
White_.check_black_pin()
if White_.check_black_check():
Black_.update_check_movelists()
White_.reset_enpassant()
self.moves["color"].pop(-1)
self.moves["piece"].pop(-1)
self.moves["pos"].pop(-1)
self.moves["killed"].pop(-1)
if not self.freeze:
self.total_moves -= 1
def update_movelist(self):
# We are importing inside this function because we cannot import at the top (since
# Black_Funcs class does not know about the black pieces at that point yet)
from pieces.black_funcs import Black_Funcs
Black_ = Black_Funcs(None)
for i in range(2):
self.movelist[i].clear()
self.protecting_movelist[i].clear()
if self.alive[i] == 0:
continue
row = self.row[i]
col = self.col[i]
# diagonal pointing to top-left corner of screen
while row - 1 >= 0 and col - 1 >= 0:
if White.blocks[row - 1][col - 1] == 1:
self.protecting_movelist[i].append((row - 1, col - 1))
break
if Black.blocks[row - 1][col - 1] == 1:
blackking_pos = Black_.get_pos("K")
if blackking_pos[0] == row - 1 and blackking_pos[
1] == col - 1:
self.movelist[i].append((row - 1, col - 1))
self.movelist[i].append((row - 2, col - 2))
break
self.movelist[i].append((row - 1, col - 1))
break
row -= 1
col -= 1
self.movelist[i].append((row, col))
row = self.row[i]
col = self.col[i]
# diagonal to top-right corner of screen
while row - 1 >= 0 and col + 1 <= 7:
if White.blocks[row - 1][col + 1] == 1:
self.protecting_movelist[i].append((row - 1, col + 1))
break
if Black.blocks[row - 1][col + 1] == 1:
blackking_pos = Black_.get_pos("K")
if blackking_pos[0] == row - 1 and blackking_pos[
1] == col + 1:
self.movelist[i].append((row - 1, col + 1))
self.movelist[i].append((row - 2, col + 2))
break
self.movelist[i].append((row - 1, col + 1))
break
row -= 1
col += 1
self.movelist[i].append((row, col))
row = self.row[i]
col = self.col[i]
# diagonal to bottom-right corner of screen
while row + 1 <= 7 and col + 1 <= 7:
if White.blocks[row + 1][col + 1] == 1:
self.protecting_movelist[i].append((row + 1, col + 1))
break
if Black.blocks[row + 1][col + 1] == 1:
blackking_pos = Black_.get_pos("K")
if blackking_pos[0] == row + 1 and blackking_pos[
1] == col + 1:
self.movelist[i].append((row + 1, col + 1))
self.movelist[i].append((row + 2, col + 2))
break
self.movelist[i].append((row + 1, col + 1))
break
row += 1
col += 1
self.movelist[i].append((row, col))
row = self.row[i]
col = self.col[i]
# diagonal to bottom-left corner of screen
while row + 1 <= 7 and col - 1 >= 0:
if White.blocks[row + 1][col - 1] == 1:
self.protecting_movelist[i].append((row + 1, col - 1))
break
if Black.blocks[row + 1][col - 1] == 1:
blackking_pos = Black_.get_pos("K")
if blackking_pos[0] == row + 1 and blackking_pos[
1] == col - 1:
self.movelist[i].append((row + 1, col - 1))
self.movelist[i].append((row + 2, col - 2))
break
self.movelist[i].append((row + 1, col - 1))
break
row += 1
col -= 1
self.movelist[i].append((row, col))
def pin_piece(self, pinned_piece, dir):
"""A black piece is pinned between bishop and king in direction dir.
Pin that piece by filtering any moves that are not in the bishop's pinned_movelist.
"""
from pieces.black_funcs import Black_Funcs
Black_ = Black_Funcs(None)
for i in range(2):
if self.in_path[i] == 0 or self.alive[i] == 0:
continue
self.pinned_movelist[i].clear()
if dir == "UL":
row = self.row[i]
col = self.col[i]
while row >= 0 and col >= 0:
if (row, col) == Black_.get_pos("K"):
Black_.filter(pinned_piece, self.pinned_movelist[i])
return
self.pinned_movelist[i].append((row, col))
row -= 1
col -= 1
elif dir == "UR":
row = self.row[i]
col = self.col[i]
while row >= 0 and col <= 7:
if (row, col) == Black_.get_pos("K"):
Black_.filter(pinned_piece, self.pinned_movelist[i])
return
self.pinned_movelist[i].append((row, col))
row -= 1
col += 1
elif dir == "LR":
row = self.row[i]
col = self.col[i]
while row <= 7 and col <= 7:
if (row, col) == Black_.get_pos("K"):
Black_.filter(pinned_piece, self.pinned_movelist[i])
return
self.pinned_movelist[i].append((row, col))
row += 1
col += 1
elif dir == "LL":
row = self.row[i]
col = self.col[i]
while row <= 7 and col >= 0:
if (row, col) == Black_.get_pos("K"):
Black_.filter(pinned_piece, self.pinned_movelist[i])
return
self.pinned_movelist[i].append((row, col))
row += 1
col -= 1
class White_Funcs(object):
"Methods relating to white pieces."
def __init__(self):
self.Black_ = Black_Funcs(self)
self.piece = ""
def play(self):
if Board.stalemate or Board.checkmate or Board.insufficient_mat or Board.tfr:
return
if White.ai:
from white_ai import White_AI
AI = White_AI()
AI.generate_move()
Board.save_img = True
White.turn = False
Black.turn = True
else:
if White.blocks[Queue.row[-1]][Queue.col[-1]] == 1:
self.piece = self.get_piece(Queue.row[-1], Queue.col[-1])
# regular move
if len(Queue.col) >= 2 and White.blocks[Queue.row[0]][Queue.col[0]] == 1 \
and White.blocks[Queue.row[1]][Queue.col[1]] == 0:
self.move_piece(self.piece, Queue.row[-1], Queue.col[-1], True)
Board.ai_move = True
# queen side castle
elif len(Queue.col) >= 2 and (Queue.row[0], Queue.col[0]) == (WhiteKing.row, WhiteKing.col) \
and (Queue.row[1], Queue.col[1]) == (WhiteRook.row[0], WhiteRook.col[0]) and \
self.castle_criteria_queen():
self.castle_queen(True)
# king side castle
elif len(Queue.col) >= 2 and (Queue.row[0], Queue.col[0]) == (WhiteKing.row, WhiteKing.col) \
and (Queue.row[1], Queue.col[1]) == (WhiteRook.row[1], WhiteRook.col[1]) and \
self.castle_criteria_king():
self.castle_king(True)
def filter(self, piece, pinned_movelist):
"""Helper function which removes any coordinates in a white piece's movelist
which is not contained in pinned_movelist. Used to assign pinned movelists to pieces.
"""
new_movelist = []
for i in range(White.num_queens):
if piece == "Q" + str(i):
for k in WhiteQueen.movelist[i]:
if not (k not in pinned_movelist):
new_movelist.append(k)
WhiteQueen.movelist[i] = new_movelist.copy()
for i in range(8):
new_movelist.clear()
if piece == "P" + str(i):
for k in WhitePawn.movelist[i]:
if not (k not in pinned_movelist):
new_movelist.append(k)
WhitePawn.movelist[i] = new_movelist.copy()
for i in range(2):
new_movelist.clear()
if piece == "R" + str(i):
for k in WhiteRook.movelist[i]:
if not (k not in pinned_movelist):
new_movelist.append(k)
WhiteRook.movelist[i] = new_movelist.copy()
elif piece == "B" + str(i):
for k in WhiteBishop.movelist[i]:
if not (k not in pinned_movelist):
new_movelist.append(k)
WhiteBishop.movelist[i] = new_movelist.copy()
elif piece == "N" + str(i):
for k in WhiteKnight.movelist[i]:
if not (k not in pinned_movelist):
new_movelist.append(k)
WhiteKnight.movelist[i] = new_movelist.copy()
def get_pos(self, piece):
"""Getter for accessing white piece positions.
Return coordinates of white piece.
"""
if piece == "K":
return (WhiteKing.row, WhiteKing.col)
for i in range(8):
if piece == "P" + str(i):
return (WhitePawn.row[i], WhitePawn.col[i])
def get_movelist(self, piece):
"""Getter for accessing white piece movelist.
Return movelist of white piece.
"""
if piece == "K":
return WhiteKing.movelist
for i in range(White.num_queens):
if piece == "Q" + str(i):
return WhiteQueen.movelist[i]
for i in range(2):
if piece == "B" + str(i):
return WhiteBishop.movelist[i]
elif piece == "N" + str(i):
return WhiteKnight.movelist[i]
elif piece == "R" + str(i):
return WhiteRook.movelist[i]
for i in range(8):
if piece == "P" + str(i):
return WhitePawn.movelist[i]
def get_piece(self, row, col):
"Get piece (as a string) at (row, col)."
piece = ""
# King
if WhiteKing.row == row and WhiteKing.col == col:
piece = "K"
# Queen
for i in range(White.num_queens):
if WhiteQueen.row[i] == row and WhiteQueen.col[i] == col:
piece = "Q" + str(i)
# Bishop, knight, rook
for i in range(2):
if WhiteBishop.row[i] == row and WhiteBishop.col[i] == col:
piece = "B" + str(i)
if WhiteKnight.row[i] == row and WhiteKnight.col[i] == col:
piece = "N" + str(i)
if WhiteRook.row[i] == row and WhiteRook.col[i] == col:
piece = "R" + str(i)
# Pawn
for i in range(8):
if WhitePawn.row[i] == row and WhitePawn.col[i] == col:
piece = "P" + str(i)
return piece if not None else None
def move_piece(self, piece, row, col, sound):
"Move piece to position (row, col)."
moved = False
killed = False
self.reset_enpassant()
if row > 7 or col > 7 or row < 0 or col < 0:
print("(row, col) bad numbers!")
return
# King
if piece == "K" and (row, col) in WhiteKing.movelist:
killed = self.check_kill(row, col, sound)
Board.moves["pos"].append((WhiteKing.row, WhiteKing.col))
WhiteKing.Move(row, col)
Board.moves["piece"].append("K")
moved = True
elif piece == "K" and (row, col) not in WhiteKing.movelist and sound:
Sound.error.play()
# Queen
for i in range(White.num_queens):
if piece == "Q" + str(i) and (row, col) in WhiteQueen.movelist[i]:
killed = self.check_kill(row, col, sound)
Board.moves["pos"].append(
(WhiteQueen.row[i], WhiteQueen.col[i]))
WhiteQueen.Move(i, row, col)
Board.moves["piece"].append("Q" + str(i))
moved = True
elif piece == "Q" + str(i) and (
row, col) not in WhiteQueen.movelist[i] and sound:
Sound.error.play()
# Pawn
for i in range(8):
if piece == "P" + str(i) and (row, col) in WhitePawn.movelist[i]:
if Black.en_passant[col] == 1 and WhitePawn.row[i] == 3:
killed = self.check_kill(row + 1, col, sound)
else:
killed = self.check_kill(row, col, sound)
Board.moves["pos"].append((WhitePawn.row[i], WhitePawn.col[i]))
WhitePawn.Move(i, row, col)
Board.moves["piece"].append("P" + str(i))
moved = True
if row == 0:
self.promote(i)
elif piece == "P" + str(i) and (
row, col) not in WhitePawn.movelist[i] and sound:
Sound.error.play()
# Knight, bishop, rook
for i in range(2):
if piece == "N" + str(i) and (row, col) in WhiteKnight.movelist[i]:
killed = self.check_kill(row, col, sound)
Board.moves["pos"].append(
(WhiteKnight.row[i], WhiteKnight.col[i]))
WhiteKnight.Move(i, row, col)
Board.moves["piece"].append("N" + str(i))
moved = True
elif piece == "N" + str(i) and (
row, col) not in WhiteKnight.movelist[i] and sound:
Sound.error.play()
if piece == "B" + str(i) and (row, col) in WhiteBishop.movelist[i]:
killed = self.check_kill(row, col, sound)
Board.moves["pos"].append(
(WhiteBishop.row[i], WhiteBishop.col[i]))
WhiteBishop.Move(i, row, col)
Board.moves["piece"].append("B" + str(i))
moved = True
elif piece == "B" + str(i) and (
row, col) not in WhiteBishop.movelist[i] and sound:
Sound.error.play()
if piece == "R" + str(i) and (row, col) in WhiteRook.movelist[i]:
killed = self.check_kill(row, col, sound)
Board.moves["pos"].append((WhiteRook.row[i], WhiteRook.col[i]))
WhiteRook.Move(i, row, col)
Board.moves["piece"].append("R" + str(i))
moved = True
elif piece == "R" + str(i) and (
row, col) not in WhiteRook.movelist[i] and sound:
Sound.error.play()
if moved:
self.valid_move(killed, sound)
def valid_move(self, killed, sound):
"""A valid move has been made. Update the board, check for pins,
checkmate, stalemate,...etc.
"""
Board.update_all_moves()
self.check_black_pin()
# check if black has been checked
if self.check_black_check():
if sound:
Sound.check.play()
self.Black_.update_check_movelists()
# check if it is also a checkmate
if self.Black_.check_no_moves_left():
Board.checkmate = True
Black.lose = True
elif self.check_black_check() == False and not killed:
if sound:
Sound.move.play()
# check if moving caused a stalemate
if self.Black_.check_no_moves_left():
Board.stalemate = True
Board.check_draw()
self.reset_enpassant()
White.turn = False
Black.turn = True
Board.moves["color"].append("W")
if Board.freeze != True:
Board.total_moves += 1
if testing:
White.turn = True
Black.turn = True
def destroy(self, piece):
"Destroy a piece."
if piece == "K":
WhiteKing.alive = 0
WhiteKing.col = -1
WhiteKing.row = -1
WhiteKing.x = -1
WhiteKing.y = -1
WhiteKing.movelist.clear()
# queen
for i in range(White.num_queens):
if piece == "Q" + str(i):
WhiteQueen.alive[i] = 0
WhiteQueen.col[i] = -1
WhiteQueen.row[i] = -1
WhiteQueen.x[i] = -1
WhiteQueen.y[i] = -1
WhiteQueen.movelist[i].clear()
WhiteQueen.protecting_movelist[i].clear()
WhiteQueen.pinned_movelist[i].clear()
WhiteQueen.in_path[i] = 0
# pawn
for i in range(8):
if piece == "P" + str(i):
WhitePawn.alive[i] = 0
WhitePawn.col[i] = -1
WhitePawn.row[i] = -1
WhitePawn.x[i] = -1
WhitePawn.y[i] = -1
WhitePawn.movelist[i].clear()
WhitePawn.hit_movelist[i].clear()
# bishop, rook, knight
for i in range(2):
if piece == "B" + str(i):
WhiteBishop.alive[i] = 0
WhiteBishop.col[i] = -1
WhiteBishop.row[i] = -1
WhiteBishop.x[i] = -1
WhiteBishop.y[i] = -1
WhiteBishop.movelist[i].clear()
WhiteBishop.protecting_movelist[i].clear()
WhiteBishop.pinned_movelist[i].clear()
WhiteBishop.in_path[i] = 0
if piece == "R" + str(i):
WhiteRook.alive[i] = 0
WhiteRook.col[i] = -1
WhiteRook.row[i] = -1
WhiteRook.x[i] = -1
WhiteRook.y[i] = -1
WhiteRook.movelist[i].clear()
WhiteRook.protecting_movelist[i].clear()
WhiteRook.pinned_movelist[i].clear()
WhiteRook.in_path[i] = 0
if piece == "N" + str(i):
WhiteKnight.alive[i] = 0
WhiteKnight.col[i] = -1
WhiteKnight.row[i] = -1
WhiteKnight.x[i] = -1
WhiteKnight.y[i] = -1
WhiteKnight.movelist[i].clear()
WhiteKnight.protecting_movelist[i].clear()
def check_kill(self, row, col, sound):
"""Once a white piece has moved to a new position, check
if a black piece is already there. If so, kill it.
"""
if Black.blocks[row][col] == 1:
Black.blocks[row][col] = 0
killed_piece = self.Black_.get_piece(row, col)
# King
if killed_piece == "K":
Board.moves["killed"].append(True)
Board.killed["piece"].append("K")
Board.killed["pos"].append((BlackKing.row, BlackKing.col))
BlackKing.alive = 0
BlackKing.col = -1
BlackKing.row = -1
BlackKing.x = -1
BlackKing.y = -1
BlackKing.movelist.clear()
if sound:
Sound.kill.play()
return True
# queen
for i in range(Black.num_queens):
if killed_piece == "Q" + str(i):
Board.moves["killed"].append(True)
Board.killed["piece"].append("Q" + str(i))
Board.killed["pos"].append(
(BlackQueen.row[i], BlackQueen.col[i]))
BlackQueen.alive[i] = 0
BlackQueen.col[i] = -1
BlackQueen.row[i] = -1
BlackQueen.x[i] = -1
BlackQueen.y[i] = -1
BlackQueen.movelist[i].clear()
BlackQueen.protecting_movelist[i].clear()
BlackQueen.pinned_movelist[i].clear()
BlackQueen.in_path[i] = 0
if sound:
Sound.kill.play()
return True
# pawn
for i in range(8):
if killed_piece == "P" + str(i):
Board.moves["killed"].append(True)
Board.killed["piece"].append("P" + str(i))
Board.killed["pos"].append(
(BlackPawn.row[i], BlackPawn.col[i]))
BlackPawn.alive[i] = 0
BlackPawn.col[i] = -1
BlackPawn.row[i] = -1
BlackPawn.x[i] = -1
BlackPawn.y[i] = -1
BlackPawn.movelist[i].clear()
BlackPawn.hit_movelist[i].clear()
if sound:
Sound.kill.play()
return True
# bishop, rook, knight
for i in range(2):
if killed_piece == "B" + str(i):
Board.moves["killed"].append(True)
Board.killed["piece"].append("B" + str(i))
Board.killed["pos"].append(
(BlackBishop.row[i], BlackBishop.col[i]))
BlackBishop.alive[i] = 0
BlackBishop.col[i] = -1
BlackBishop.row[i] = -1
BlackBishop.x[i] = -1
BlackBishop.y[i] = -1
BlackBishop.movelist[i].clear()
BlackBishop.protecting_movelist[i].clear()
BlackBishop.pinned_movelist[i].clear()
BlackBishop.in_path[i] = 0
if sound:
Sound.kill.play()
return True
if killed_piece == "R" + str(i):
Board.moves["killed"].append(True)
Board.killed["piece"].append("R" + str(i))
Board.killed["pos"].append(
(BlackRook.row[i], BlackRook.col[i]))
BlackRook.alive[i] = 0
BlackRook.col[i] = -1
BlackRook.row[i] = -1
BlackRook.x[i] = -1
BlackRook.y[i] = -1
BlackRook.movelist[i].clear()
BlackRook.protecting_movelist[i].clear()
BlackRook.pinned_movelist[i].clear()
BlackRook.in_path[i] = 0
if sound:
Sound.kill.play()
return True
if killed_piece == "N" + str(i):
Board.moves["killed"].append(True)
Board.killed["piece"].append("N" + str(i))
Board.killed["pos"].append(
(BlackKnight.row[i], BlackKnight.col[i]))
BlackKnight.alive[i] = 0
BlackKnight.col[i] = -1
BlackKnight.row[i] = -1
BlackKnight.x[i] = -1
BlackKnight.y[i] = -1
BlackKnight.movelist[i].clear()
BlackKnight.protecting_movelist[i].clear()
if sound:
Sound.kill.play()
return True
Board.moves["killed"].append(False)
return False
def check_black_check(self):
"""Check detection. If the black king's position is in the movelist
of any white piece, then it is checked.
"""
pos = (BlackKing.row, BlackKing.col)
for i in range(White.num_queens):
if pos in WhiteQueen.movelist[i]:
White.checker = "Q" + str(i)
return 1
for i in range(8):
if pos in WhitePawn.hit_movelist[i]:
White.checker = "P" + str(i)
return 1
for i in range(2):
if pos in WhiteBishop.movelist[i]:
White.checker = "B" + str(i)
return 1
if pos in WhiteRook.movelist[i]:
White.checker = "R" + str(i)
return 1
if pos in WhiteKnight.movelist[i]:
White.checker = "N" + str(i)
return 1
White.checker = ""
return 0
def update_check_movelists(self):
"""White has been checked. Replace all white piece moves (except king)
with moves such that white is not checked anymore.
"""
# build check movelist from the white piece that is checking the black king
movelist = []
for i in range(Black.num_queens):
if Black.checker == "Q" + str(i):
movelist = BlackQueen.build_check_movelist().copy()
for i in range(2):
if Black.checker == "B" + str(i):
movelist = BlackBishop.build_check_movelist().copy()
elif Black.checker == "R" + str(i):
movelist = BlackRook.build_check_movelist().copy()
elif Black.checker == "N" + str(i):
movelist = [(BlackKnight.row[i], BlackKnight.col[i])]
for i in range(8):
if Black.checker == "P" + str(i):
movelist = [(BlackPawn.row[i], BlackPawn.col[i])]
# now filter all white piece movelists with the new check movelist
for i in range(White.num_queens):
self.filter("Q" + str(i), movelist)
for i in range(2):
self.filter("B" + str(i), movelist)
self.filter("R" + str(i), movelist)
self.filter("N" + str(i), movelist)
for i in range(8):
self.filter("P" + str(i), movelist)
def check_no_moves_left(self):
"Returns true if White has no moves left to play."
if WhiteKing.movelist != []:
return 0
for i in range(White.num_queens):
if WhiteQueen.movelist[i] != []:
return 0
for i in range(2):
if WhiteBishop.movelist[i] != [] or WhiteRook.movelist[i] != [] or \
WhiteKnight.movelist[i] != []:
return 0
for i in range(8):
if WhitePawn.movelist[i] != []:
return 0
return 1
def check_black_pin(self):
"Check if any black piece is pinned by a white bishop, rook, or queen."
WhiteBishop.check_pin()
WhiteQueen.check_pin()
WhiteRook.check_pin()
def reset_enpassant(self):
"Reset opposite side's pawn en passant flags."
for i in range(8):
White.en_passant[i] = 0
def promote(self, i):
"Promote white pawn i to queen."
White.num_queens += 1
WhiteQueen.row.append(WhitePawn.row[i])
WhiteQueen.col.append(WhitePawn.col[i])
WhiteQueen.x.append(Piece.paddingx + WhitePawn.col[i] * UNIT)
WhiteQueen.y.append(Piece.paddingy + WhitePawn.row[i] * UNIT)
WhiteQueen.img.append(
pg.image.load('../assets/sprites/whiteQueen.png'))
WhiteQueen.img[-1] = pg.transform.scale(WhiteQueen.img[-1],
(Piece.scale, Piece.scale))
WhiteQueen.movelist.append([])
WhiteQueen.protecting_movelist.append([])
WhiteQueen.pinned_movelist.append([])
WhiteQueen.alive.append(1)
WhiteQueen.in_path.append(0)
WhitePawn.alive[i] = 0
WhitePawn.col[i] = -1
WhitePawn.row[i] = -1
WhitePawn.x[i] = -1
WhitePawn.y[i] = -1
WhitePawn.movelist[i].clear()
def unpromote(self, i, row, col):
# kill last queen
White.num_queens -= 1
WhiteQueen.row.pop(-1)
WhiteQueen.col.pop(-1)
WhiteQueen.x.pop(-1)
WhiteQueen.y.pop(-1)
WhiteQueen.img.pop(-1)
WhiteQueen.movelist.pop(-1)
WhiteQueen.protecting_movelist.pop(-1)
WhiteQueen.pinned_movelist.pop(-1)
WhiteQueen.alive.pop(-1)
WhiteQueen.in_path.pop(-1)
# revive last pawn
WhitePawn.alive[i] = 1
WhitePawn.col[i] = col
WhitePawn.row[i] = row
WhitePawn.x[i] = Piece.paddingx + col * UNIT
WhitePawn.y[i] = Piece.paddingy + row * UNIT
WhitePawn.movelist[i].clear()
White.blocks[row][col] = 1
def in_movelist(self, row, col):
"""Helper function which returns true if position (row, col) is in the movelist
of any Black piece.
"""
for i in BlackKing.movelist:
if i == (row, col):
return True
for i in range(Black.num_queens):
for k in BlackQueen.movelist[i]:
if k == (row, col):
return True
for i in range(8):
for k in BlackPawn.movelist[i]:
if k == (row, col):
return True
for i in range(2):
for k in BlackBishop.movelist[i]:
if k == (row, col):
return True
for k in BlackRook.movelist[i]:
if k == (row, col):
return True
for k in BlackKnight.movelist[i]:
if k == (row, col):
return True
return False
def castle_criteria_king(self):
"""Return true if king-side castling is allowed. i.e:
- King & rook 1 have not moved
- No pieces (white or black) between king and rook 1
- King is not checked
- Squares between rook 1 & king are not attacked
- King does not end up in check
"""
if WhiteKing.moved == True or WhiteRook.moved[1] == True:
return False
if Black.blocks[7][6] == 1 or Black.blocks[7][5] == 1 or White.blocks[7][6] == 1 \
or White.blocks[7][5] == 1:
return False
if Black.checker != "":
return False
if self.in_movelist(7, 6) == True or self.in_movelist(7, 5) == True:
return False
return True
def castle_criteria_queen(self):
"Same as above but for queen-side castle (& rook 0)."
if WhiteKing.moved == True or WhiteRook.moved[0] == True:
return False
if Black.blocks[7][1] == 1 or Black.blocks[7][2] == 1 or Black.blocks[7][3] == 1 \
or White.blocks[7][1] == 1 or White.blocks[7][2] == 1 or White.blocks[7][3] == 1:
return False
if Black.checker != "":
return False
if self.in_movelist(7, 1) == True or self.in_movelist(
7, 2) == True or self.in_movelist(7, 3):
return False
return True
def castle_queen(self, sound):
"Queen-side castle."
Board.moves["pos"].append("CQ")
WhiteKing.Move(7, 2)
WhiteRook.Move(0, 7, 3)
self.valid_move(False, sound)
White.castled = 1
Board.moves["piece"].append("CQ")
Board.moves["killed"].append(False)
def castle_king(self, sound):
"King-side castle."
Board.moves["pos"].append("CK")
WhiteKing.Move(7, 6)
WhiteRook.Move(1, 7, 5)
self.valid_move(False, sound)
White.castled = 1
Board.moves["piece"].append("CK")
Board.moves["killed"].append(False)
def two_knights(self):
"Below functions are used for draw by insufficient material."
for i in range(White.num_queens):
if WhiteQueen.alive[i] == 1:
return False
for i in range(8):
if WhitePawn.alive[i] == 1:
return False
for i in range(2):
if WhiteBishop.alive[i] == 1:
return False
if WhiteRook.alive[i] == 1:
return False
if WhiteKnight.alive[0] == 1 and WhiteKnight.alive[1] == 1:
return True
return False
def one_knight(self):
for i in range(White.num_queens):
if WhiteQueen.alive[i] == 1:
return False
for i in range(8):
if WhitePawn.alive[i] == 1:
return False
for i in range(2):
if WhiteBishop.alive[i] == 1:
return False
if WhiteRook.alive[i] == 1:
return False
if WhiteKnight.alive[0] != WhiteKnight.alive[1]:
return True
return False
def lone_king(self):
for i in range(White.num_queens):
if WhiteQueen.alive[i] == 1:
return False
for i in range(8):
if WhitePawn.alive[i] == 1:
return False
for i in range(2):
if WhiteBishop.alive[i] == 1:
return False
if WhiteRook.alive[i] == 1:
return False
if WhiteKnight.alive[i] == 1:
return False
return True
def one_bishop(self):
for i in range(White.num_queens):
if WhiteQueen.alive[i] == 1:
return False
for i in range(8):
if WhitePawn.alive[i] == 1:
return False
for i in range(2):
if WhiteRook.alive[i] == 1:
return False
if WhiteKnight.alive[i] == 1:
return False
if WhiteBishop.alive[0] != WhiteBishop.alive[1]:
return True
return False
def num_pieces(self, dir):
"""Determine the number of black pieces between bishop and black
king in direction dir.
"""
from pieces.black_funcs import Black_Funcs
Black_ = Black_Funcs(None)
for i in range(2):
if self.in_path[i] == 0 or self.alive[i] == 0:
continue
if dir == "UL":
row = self.row[i]
col = self.col[i]
num = 0
num_blocks = 0
while row >= 0 and col >= 0:
if White.blocks[row][col] == 1 and num_blocks > 0:
return -1
elif Black.blocks[row][col] == 1 and (
row, col) != Black_.get_pos("K"):
num += 1
elif (row, col) == Black_.get_pos("K"):
return num
row -= 1
col -= 1
num_blocks += 1
elif dir == "UR":
row = self.row[i]
col = self.col[i]
num = 0
num_blocks = 0
while row >= 0 and col <= 7:
if White.blocks[row][col] == 1 and num_blocks > 0:
return -1
elif Black.blocks[row][col] == 1 and (
row, col) != Black_.get_pos("K"):
num += 1
elif (row, col) == Black_.get_pos("K"):
return num
row -= 1
col += 1
num_blocks += 1
elif dir == "LR":
row = self.row[i]
col = self.col[i]
num = 0
num_blocks = 0
while row <= 7 and col <= 7:
if White.blocks[row][col] == 1 and num_blocks > 0:
return -1
elif Black.blocks[row][col] == 1 and (
row, col) != Black_.get_pos("K"):
num += 1
elif (row, col) == Black_.get_pos("K"):
return num
row += 1
col += 1
num_blocks += 1
elif dir == "LL":
row = self.row[i]
col = self.col[i]
num = 0
num_blocks = 0
while row <= 7 and col >= 0:
if White.blocks[row][col] == 1 and num_blocks > 0:
return -1
elif Black.blocks[row][col] == 1 and (
row, col) != Black_.get_pos("K"):
num += 1
elif (row, col) == Black_.get_pos("K"):
return num
row += 1
col -= 1
num_blocks += 1
def update_movelist(self):
"Queen movelist is the sum of Bishop and Rook moves."
from pieces.black_funcs import Black_Funcs
Black_ = Black_Funcs(None)
for i in range(White.num_queens):
if self.alive[i] == 0:
continue
# Rook part
self.movelist[i].clear()
self.protecting_movelist[i].clear()
row = self.row[i]
col = self.col[i]
while row - 1 >= 0:
if White.blocks[row - 1][col] == 1:
self.protecting_movelist[i].append((row - 1, col))
break
if Black.blocks[row - 1][col] == 1:
blackking_pos = Black_.get_pos("K")
if blackking_pos[0] == row - 1 and blackking_pos[1] == col:
self.movelist[i].append((row - 1, col))
self.movelist[i].append((row - 2, col))
break
self.movelist[i].append((row - 1, col))
break
row -= 1
self.movelist[i].append((row, col))
row = self.row[i]
while row + 1 <= 7:
if White.blocks[row + 1][col] == 1:
self.protecting_movelist[i].append((row + 1, col))
break
if Black.blocks[row + 1][col] == 1:
blackking_pos = Black_.get_pos("K")
if blackking_pos[0] == row + 1 and blackking_pos[1] == col:
self.movelist[i].append((row + 1, col))
self.movelist[i].append((row + 2, col))
break
self.movelist[i].append((row + 1, col))
break
row += 1
self.movelist[i].append((row, col))
row = self.row[i]
while col - 1 >= 0:
if White.blocks[row][col - 1] == 1:
self.protecting_movelist[i].append((row, col - 1))
break
if Black.blocks[row][col - 1] == 1:
blackking_pos = Black_.get_pos("K")
if blackking_pos[0] == row and blackking_pos[1] == col - 1:
self.movelist[i].append((row, col - 1))
self.movelist[i].append((row, col - 2))
break
self.movelist[i].append((row, col - 1))
break
col -= 1
self.movelist[i].append((row, col))
col = self.col[i]
while col + 1 <= 7:
if White.blocks[row][col + 1] == 1:
self.protecting_movelist[i].append((row, col + 1))
break
if Black.blocks[row][col + 1] == 1:
blackking_pos = Black_.get_pos("K")
if blackking_pos[0] == row and blackking_pos[1] == col + 1:
self.movelist[i].append((row, col + 1))
self.movelist[i].append((row, col + 2))
break
self.movelist[i].append((row, col + 1))
break
col += 1
self.movelist[i].append((row, col))
# Bishop part
row = self.row[i]
col = self.col[i]
while row - 1 >= 0 and col - 1 >= 0:
if White.blocks[row - 1][col - 1] == 1:
self.protecting_movelist[i].append((row - 1, col - 1))
break
if Black.blocks[row - 1][col - 1] == 1:
blackking_pos = Black_.get_pos("K")
if blackking_pos[0] == row - 1 and blackking_pos[
1] == col - 1:
self.movelist[i].append((row - 1, col - 1))
self.movelist[i].append((row - 2, col - 2))
break
self.movelist[i].append((row - 1, col - 1))
break
row -= 1
col -= 1
self.movelist[i].append((row, col))
row = self.row[i]
col = self.col[i]
while row - 1 >= 0 and col + 1 <= 7:
if White.blocks[row - 1][col + 1] == 1:
self.protecting_movelist[i].append((row - 1, col + 1))
break
if Black.blocks[row - 1][col + 1] == 1:
blackking_pos = Black_.get_pos("K")
if blackking_pos[0] == row - 1 and blackking_pos[
1] == col + 1:
self.movelist[i].append((row - 1, col + 1))
self.movelist[i].append((row - 2, col + 2))
break
self.movelist[i].append((row - 1, col + 1))
break
row -= 1
col += 1
self.movelist[i].append((row, col))
row = self.row[i]
col = self.col[i]
while row + 1 <= 7 and col + 1 <= 7:
if White.blocks[row + 1][col + 1] == 1:
self.protecting_movelist[i].append((row + 1, col + 1))
break
if Black.blocks[row + 1][col + 1] == 1:
blackking_pos = Black_.get_pos("K")
if blackking_pos[0] == row + 1 and blackking_pos[
1] == col + 1:
self.movelist[i].append((row + 1, col + 1))
self.movelist[i].append((row + 2, col + 2))
break
self.movelist[i].append((row + 1, col + 1))
break
row += 1
col += 1
self.movelist[i].append((row, col))
row = self.row[i]
col = self.col[i]
while row + 1 <= 7 and col - 1 >= 0:
if White.blocks[row + 1][col - 1] == 1:
self.protecting_movelist[i].append((row + 1, col - 1))
break
if Black.blocks[row + 1][col - 1] == 1:
blackking_pos = Black_.get_pos("K")
if blackking_pos[0] == row + 1 and blackking_pos[
1] == col - 1:
self.movelist[i].append((row + 1, col - 1))
self.movelist[i].append((row + 2, col - 2))
break
self.movelist[i].append((row + 1, col - 1))
break
row += 1
col -= 1
self.movelist[i].append((row, col))
def __init__(self):
self.Black_ = Black_Funcs(self)
self.piece = ""