151 lines
5.5 KiB
Python
151 lines
5.5 KiB
Python
type_to_name = []
|
|
|
|
PIECE_TO_UNICODE = {"wk": "\u2654 ", "wq": "\u2655 ", "wr": "\u2656 ", "wb": "\u2657 ", "wn": "\u2658 ", "wp": "\u2659 ",
|
|
"bk": "\u265A ", "bq": "\u265B ", "br": "\u265C ", "bb": "\u265D ", "bn": "\u265E ", "bp": "\u265F "}
|
|
|
|
INDEX_TO_FIELD = [
|
|
"a1", "b1", "c1", "d1", "e1", "f1", "g1", "h1",
|
|
"a2", "b2", "c2", "d2", "e2", "f2", "g2", "h2",
|
|
"a3", "b3", "c3", "d3", "e3", "f3", "g3", "h3",
|
|
"a4", "b4", "c4", "d4", "e4", "f4", "g4", "h4",
|
|
"a5", "b5", "c5", "d5", "e5", "f5", "g5", "h5",
|
|
"a6", "b6", "c6", "d6", "e6", "f6", "g6", "h6",
|
|
"a7", "b7", "c7", "d7", "e7", "f7", "g7", "h7",
|
|
"a8", "b8", "c8", "d8", "e8", "f8", "g8", "h8",
|
|
]
|
|
|
|
''' Board Index:
|
|
a b c d e f g h
|
|
8 [56] [57] [58] [59] [60] [61] [62] [63]
|
|
7 [48] [49] [50] [51] [52] [53] [54] [55]
|
|
6 [40] [41] [42] [43] [44] [45] [46] [47]
|
|
5 [32] [33] [34] [35] [36] [37] [38] [39]
|
|
4 [24] [25] [26] [27] [28] [29] [30] [31]
|
|
3 [16] [17] [18] [19] [20] [21] [22] [23]
|
|
2 [08] [09] [10] [11] [12] [13] [14] [15]
|
|
1 [00] [01] [02] [03] [03] [05] [06] [07]
|
|
|
|
'''
|
|
|
|
class Piece():
|
|
def __init__(self, pce_type, is_white, index):
|
|
|
|
self.type = pce_type
|
|
self.is_white = is_white
|
|
self.has_castled = False if type == 'K' else True
|
|
self.index = index
|
|
self.bit_mask = 1 << index
|
|
|
|
class Move():
|
|
def __init__(self, start_index, end_index, piece_moved, piece_captured=None):
|
|
self.piece_moved = piece_moved
|
|
self.piece_captured = piece_captured
|
|
self.index_start = start_index
|
|
self.index_end = end_index
|
|
self.bitmask_start: int = 1 << start_index
|
|
self.bitmask_end = 1 << end_index
|
|
self.chess_notation = f"{self.piece_moved[1].upper()}{INDEX_TO_FIELD[self.index_start]}{INDEX_TO_FIELD[self.index_end]}"
|
|
|
|
class Board():
|
|
def __init__(self):
|
|
|
|
self.bit_mask_w = 0x0000000000000001FE7F800
|
|
self.bit_mask_b = 0x1FE7F800000000000000000
|
|
self.bit_mask_all=0x1FE7F80000000001FE7F800
|
|
self.white_to_move = True
|
|
|
|
# self.wr1 = Piece('R', True, 11)
|
|
# self.wn1 = Piece('N', True, 12)
|
|
# self.wb1 = Piece('B', True, 13)
|
|
# self.wk = Piece('K', True, 14)
|
|
# self.wq = Piece('Q', True, 15)
|
|
# self.wb2 = Piece('B', True, 16)
|
|
# self.wn2 = Piece('N', True, 17)
|
|
# self.wr2 = Piece('R', True, 18)
|
|
|
|
# self.wp = [Piece('P', True, 21),
|
|
# Piece('P', True, 22),
|
|
# Piece('P', True, 23),
|
|
# Piece('P', True, 24),
|
|
# Piece('P', True, 25),
|
|
# Piece('P', True, 26),
|
|
# Piece('P', True, 27),
|
|
# Piece('P', True, 28)]
|
|
|
|
self.piece_dict = {
|
|
'br1':81, 'bn1':82, 'bb1':83, 'bk' :84, 'bq' :85, 'bb2':86, 'bn2':87, 'br2':88,
|
|
'bp1':71, 'bp2':72, 'bp3':73, 'bp4':74, 'bp5':75, 'bp6':76, 'bp7':77, 'bp8':78,
|
|
'wp1':21, 'wp2':22, 'wp3':23, 'wp4':24, 'wp5':25, 'wp6':26, 'wp7':27, 'wp8':28,
|
|
'wr1':11, 'wn1':12, 'wb1':13, 'wk' :14, 'wq' :15, 'wb2':16, 'wn2':17, 'wr2':18,
|
|
}
|
|
|
|
self.move_log = []
|
|
|
|
def print_board(self):
|
|
for x in range(100):
|
|
if OUT_OF_BOARD & 1 << x != 0:
|
|
if (x+1) % 10 == 0:
|
|
print("")
|
|
continue
|
|
continue
|
|
if self.bit_mask_all & 1 << x != 0:
|
|
for p in self.piece_dict:
|
|
if self.piece_dict[p] == x:
|
|
print(f"{PIECE_TO_UNICODE[p[:2]]} ", end="")
|
|
else:
|
|
print("\u25FB ", end="")
|
|
|
|
def notation_to_move(self, notation_str): # Kb1a5
|
|
if len(notation_str) < 6:
|
|
start = INDEX_TO_FIELD.index(notation_str[1]+ notation_str[2])
|
|
end = INDEX_TO_FIELD.index(notation_str[3]+ notation_str[4])
|
|
for p in self.piece_dict:
|
|
if self.piece_dict[p] == start:
|
|
piece_moved = p
|
|
if self.piece_dict[p] == end:
|
|
piece_captured = p
|
|
if piece_moved and piece_captured:
|
|
return Move(start, end, piece_moved, piece_captured)
|
|
elif piece_moved:
|
|
return Move(start, end, piece_moved)
|
|
else:
|
|
print("No Piece to Move")
|
|
return
|
|
else:
|
|
print("Move to long")
|
|
return
|
|
|
|
def make_move(self, move):
|
|
self.piece_dict[move.piece_moved] = move.index_start
|
|
|
|
if self.white_to_move:
|
|
self.bit_mask_w = self.bit_mask_w & ~move.bitmask_start
|
|
self.bit_mask_w = self.bit_mask_w | move.bitmask_end
|
|
|
|
if move.piece_captured:
|
|
self.piece_dict[move.piece_captured] = -1
|
|
self.bit_mask_b = self.bit_mask_b & ~move.bitmask_end
|
|
|
|
else:
|
|
self.bit_mask_b = self.bit_mask_b & ~move.bitmask_start
|
|
self.bit_mask_b = self.bit_mask_b | move.bitmask_end
|
|
|
|
if move.piece_captured:
|
|
self.piece_dict[move.piece_captured] = -1
|
|
self.bit_mask_b = self.bit_mask_b & ~move.bitmask_end
|
|
|
|
self.bit_mask_all = self.bit_mask_w | self.bit_mask_b
|
|
|
|
self.move_log.append(move)
|
|
|
|
board = Board()
|
|
board.print_board()
|
|
|
|
while True:
|
|
print("Your Move, Please: ")
|
|
move_str = input()
|
|
if len(move_str) != 4:
|
|
print("Wrong Syntax")
|
|
continue
|
|
move = board.notation_to_move(move_str)
|
|
board.make_move(move) |