def find_moves(self, chessboard, my_king, d_tool):
count = 0
for i in range(-1, 2):
for j in range(-1, 2):
x = self.col + i
y = self.row + j
while self.can_move(chessboard, x, y, my_king) == 1:
count += 1
pygame.draw.ellipse(d_tool.screen, rgb_blue,
(GameFunctions.calc_x(
x, d_tool.sqr_l, d_tool.w_buffer),
GameFunctions.calc_y(
y, d_tool.sqr_l, d_tool.h_buffer),
d_tool.sqr_l, d_tool.sqr_l), 4)
x += i
y += j
if self.can_move(chessboard, x, y, my_king) == 2:
count += 1
pygame.draw.ellipse(d_tool.screen, rgb_red,
(GameFunctions.calc_x(
x, d_tool.sqr_l, d_tool.w_buffer),
GameFunctions.calc_y(
y, d_tool.sqr_l, d_tool.h_buffer),
d_tool.sqr_l, d_tool.sqr_l), 4)
return count
def find_moves(self, chessboard, my_king, d_tool):
count = 0
for i in range(-2, 3):
check = self.can_move(chessboard, self.col + i,
self.row - (3 - abs(i)), my_king)
if check > 0:
count += 1
pygame.draw.ellipse(
d_tool.screen, colour[check + 1],
(GameFunctions.calc_x(self.col + i, d_tool.sqr_l,
d_tool.w_buffer),
GameFunctions.calc_y(
self.row - (3 - abs(i)), d_tool.sqr_l,
d_tool.h_buffer), d_tool.sqr_l, d_tool.sqr_l), 4)
check = self.can_move(chessboard, self.col + i,
self.row + (3 - abs(i)), my_king)
if check > 0:
count += 1
pygame.draw.ellipse(
d_tool.screen, colour[check + 1],
(GameFunctions.calc_x(self.col + i, d_tool.sqr_l,
d_tool.w_buffer),
GameFunctions.calc_y(
self.row + (3 - abs(i)), d_tool.sqr_l,
d_tool.h_buffer), d_tool.sqr_l, d_tool.sqr_l), 4)
return count
def find_moves(self, chessboard, my_king, d_tool):
count = 0
if self.can_move(chessboard, self.col, self.row + self.owner,
my_king) == 1:
count += 1
pygame.draw.ellipse(
d_tool.screen, rgb_blue,
(GameFunctions.calc_x(self.col, d_tool.sqr_l, d_tool.w_buffer),
GameFunctions.calc_y(
self.row + self.owner, d_tool.sqr_l,
d_tool.h_buffer), d_tool.sqr_l, d_tool.sqr_l), 4)
if self.can_move(chessboard, self.col, self.row + (2 * self.owner),
my_king) == 1:
count += 1
pygame.draw.ellipse(
d_tool.screen, rgb_blue,
(GameFunctions.calc_x(self.col, d_tool.sqr_l,
d_tool.w_buffer),
GameFunctions.calc_y(
self.row + (2 * self.owner), d_tool.sqr_l,
d_tool.h_buffer), d_tool.sqr_l, d_tool.sqr_l), 4)
if self.can_move(chessboard, self.col - 1, self.row + self.owner,
my_king) == 2:
count += 1
pygame.draw.ellipse(
d_tool.screen, rgb_red,
(GameFunctions.calc_x(self.col - 1, d_tool.sqr_l,
d_tool.w_buffer),
GameFunctions.calc_y(
self.row + self.owner, d_tool.sqr_l,
d_tool.h_buffer), d_tool.sqr_l, d_tool.sqr_l), 4)
if self.can_move(chessboard, self.col + 1, self.row + self.owner,
my_king) == 2:
count += 1
pygame.draw.ellipse(
d_tool.screen, rgb_red,
(GameFunctions.calc_x(self.col + 1, d_tool.sqr_l,
d_tool.w_buffer),
GameFunctions.calc_y(
self.row + self.owner, d_tool.sqr_l,
d_tool.h_buffer), d_tool.sqr_l, d_tool.sqr_l), 4)
return count
def display(self, game_display, sqr_l, sqr_w_buffer, sqr_h_buffer):
game_display.blit(
self.img, (GameFunctions.calc_x(self.col, sqr_l, sqr_w_buffer) +
((sqr_l - 64) / 2),
GameFunctions.calc_y(self.row, sqr_l, sqr_h_buffer) +
((sqr_l - 64) / 2)))
def main():
# start pygame modules that need to be started
pygame.init()
# display setting information
display_w = 1200
display_h = 900
length_num = 8
if display_h < display_w:
sqr_l = display_h / length_num
sqr_h_buffer = 0
sqr_w_buffer = (display_w - display_h) / 2
else:
sqr_l = display_w / length_num
sqr_h_buffer = (display_h - display_w) / 2
sqr_w_buffer = 0
# store RGB colour codes for pygame in tuples
rgb_black = (0, 0, 0)
rgb_white = (255, 255, 255)
rgb_red = (255, 0, 0)
rgb_green = (0, 255, 0)
rgb_blue = (0, 0, 255)
# create the window with caption and a game clock
chessboard_window = pygame.display.set_mode((display_w, display_h))
pygame.display.set_caption('Python 2-Player Chess')
clock = pygame.time.Clock()
# create the chessboard including pieces and ready first turn
chessboard = ChessCreateBoard.create_board()
white_turn = True
active_piece = None
king = [chessboard[5][8], chessboard[5][1]]
player_in_check = [False, False]
drawing = GameFunctions.Dimensions(screen=chessboard_window,
sqr_l=sqr_l,
w_buffer=sqr_w_buffer,
h_buffer=sqr_h_buffer)
# start the game loop
continue_game = True
while continue_game:
# pygame event loop stacks events in a frame
for event in pygame.event.get():
# activates if user closes game window
if event.type == pygame.QUIT:
continue_game = False
# activates on click
if event.type == pygame.MOUSEBUTTONDOWN:
click_pos = pygame.mouse.get_pos()
if (sqr_w_buffer < click_pos[0] < display_w - sqr_w_buffer) \
and (sqr_h_buffer < click_pos[1] < display_h - sqr_h_buffer):
click_x = int((click_pos[0] - sqr_w_buffer) / sqr_l) + 1
click_y = int((click_pos[1] - sqr_h_buffer) / sqr_l) + 1
if active_piece and active_piece.can_move(chessboard, click_x, click_y, king[1 - white_turn]) \
and ((active_piece.owner == 1) != white_turn):
# special cases
if active_piece.__class__.__name__ == "King" \
and active_piece.horiz_distance(click_x) > 1:
if click_x < active_piece.col:
chessboard[active_piece.col - 1][
active_piece.row] = chessboard[1][
active_piece.row]
chessboard[1][active_piece.row] = None
chessboard[active_piece.col -
1][active_piece.
row].col = active_piece.col - 1
chessboard[active_piece.col - 1][
active_piece.row].has_moved = True
else:
chessboard[active_piece.col + 1][
active_piece.row] = chessboard[8][
active_piece.row]
chessboard[8][active_piece.row] = None
chessboard[active_piece.col +
1][active_piece.
row].col = active_piece.col + 1
chessboard[active_piece.col + 1][
active_piece.row].has_moved = True
elif active_piece.__class__.__name__ == "Pawn" \
and active_piece.vert_distance(click_y) == 2:
active_piece.en_pass = True
elif active_piece.__class__.__name__ == "Pawn" \
and not chessboard[click_x][click_y] and active_piece.col != click_x:
chessboard[click_x][active_piece.row] = None
elif active_piece.__class__.__name__ == "Pawn" \
and (click_y == 1 or click_y == 8):
chessboard[click_x][click_y] = ChessPieces.Queen(
active_piece.owner, click_x, click_y)
chessboard[click_x][click_y] = active_piece
chessboard[active_piece.col][active_piece.row] = None
for i in range(1, 9):
if chessboard[i][5 - white_turn]:
chessboard[i][5 - white_turn].en_pass = False
active_piece.col = click_x
active_piece.row = click_y
active_piece.has_moved = True
active_piece = None
white_turn = not white_turn
if king[1 - white_turn].in_check(chessboard):
player_in_check[1 - white_turn] = True
print('CHECK!')
else:
active_piece = chessboard[click_x][click_y]
else:
active_piece = None
i = 1
j = 1
legal_moves = 0
while legal_moves == 0 and i < 9:
if j > 8:
j = 1
i += 1
if i < 9 and chessboard[i][j] \
and ((chessboard[i][j].owner == 1) != white_turn):
legal_moves = chessboard[i][j].find_moves(
chessboard, king[1 - white_turn], drawing)
j += 1
if i == 9:
if player_in_check[1 - white_turn]:
print('CHECKMATE')
continue_game = False
else:
print('STALEMATE')
continue_game = False
# starting with the background the chessboard is drawn along with the active piece paths
chessboard_window.fill(rgb_black)
for i in range(1, 9):
for j in range(1, 9):
if (i + j) % 2 == 0:
pygame.draw.rect(
chessboard_window, rgb_white,
(GameFunctions.calc_x(i, sqr_l, sqr_w_buffer),
GameFunctions.calc_y(j, sqr_l,
sqr_h_buffer), sqr_l, sqr_l), 0)
if chessboard[i][j]:
chessboard[i][j].display(chessboard_window, sqr_l,
sqr_w_buffer, sqr_h_buffer)
if active_piece:
active_piece.find_moves(chessboard, king[1 - white_turn], drawing)
# after drawing the board it is updated to the screen and frames are updated
pygame.display.update()
clock.tick(100)
pygame.quit()
quit()