def is_move_ok(match, srcx, srcy, dstx, dsty, piece): DIRS = rules.DIRS direction, stepx, stepy = rk_step(None, srcx, srcy, dstx, dsty) if(direction == DIRS['undefined']): return False color = Match.color_of_piece(piece) pin_dir = rules.pin_dir(match, srcx, srcy) if(direction == DIRS['north'] or direction == DIRS['south']): if(pin_dir != DIRS['north'] and pin_dir != DIRS['south'] and pin_dir != DIRS['undefined']): return False elif(direction == DIRS['east'] or direction == DIRS['west']): if(pin_dir != DIRS['east'] and pin_dir != DIRS['west'] and pin_dir != DIRS['undefined']): return False x = srcx + stepx y = srcy + stepy while(x >= 0 and x <= 7 and y >= 0 and y <= 7): field = match.readfield(x, y) if(x == dstx and y == dsty): if(Match.color_of_piece(field) == color): return False else: return True elif(field != Match.PIECES['blk']): return False x += stepx y += stepy return False
def is_lg_castling_ok(match, srcx, srcy, dstx, dsty, piece): color = Match.color_of_piece(piece) opp_color = Match.REVERSED_COLORS[color] for i in range(1, 3, 1): fieldx = srcx - i field = match.readfield(fieldx, srcy) if(field != Match.PIECES['blk']): return False if(color == Match.COLORS['white']): if(match.wKg_first_movecnt != 0 or match.wRk_a1_first_movecnt != 0): return False else: if(match.bKg_first_movecnt != 0 or match.bRk_a8_first_movecnt != 0): return False king = match.readfield(srcx, srcy) match.writefield(srcx, srcy, Match.PIECES['blk']) for i in range(0, -3, -1): castlingx = srcx + i attacked = rules.attacked(match, castlingx, srcy, opp_color) if(attacked == True): match.writefield(srcx, srcy, king) return False match.writefield(srcx, srcy, king) return True
def is_move_ok(match, srcx, srcy, dstx, dsty, piece): DIRS = rules.DIRS color = Match.color_of_piece(piece) opp_color = Match.REVERSED_COLORS[color] direction = kg_dir(srcx, srcy, dstx, dsty) if(direction == DIRS['sh-castling']): return is_sh_castling_ok(match, srcx, srcy, dstx, dsty, piece) if(direction == DIRS['lg-castling']): return is_lg_castling_ok(match, srcx, srcy, dstx, dsty, piece) if(direction == DIRS['undefined']): return False king = match.readfield(srcx, srcy) captured = match.readfield(dstx, dsty) match.writefield(srcx, srcy, Match.PIECES['blk']) match.writefield(dstx, dsty, king) attacked = rules.attacked(match, dstx, dsty, opp_color) match.writefield(srcx, srcy, king) match.writefield(dstx, dsty, captured) if(attacked == True): return False field = match.readfield(dstx, dsty) if(match.color_of_piece(field) == color): return False return True
def is_move_color_ok_OLD(piece, count): color = Match.color_of_piece(piece) if count % 2 == 0 and color == Match.COLORS["white"]: return True elif count % 2 == 1 and color == Match.COLORS["black"]: return True else: return False
def pin_dir(match, scrx, srcy): piece = match.readfield(scrx, srcy) color = Match.color_of_piece(piece) if color == Match.COLORS["white"]: kgx = match.wKg_x kgy = match.wKg_y else: kgx = match.bKg_x kgy = match.bKg_y direction, stepx, stepy = rook.rk_step(None, scrx, srcy, kgx, kgy) if direction != DIRS["undefined"]: dstx, dsty = search(match, scrx, srcy, stepx, stepy) if dstx != UNDEF_X: piece = match.readfield(dstx, dsty) if (color == Match.COLORS["white"] and piece == Match.PIECES["wKg"]) or ( color == Match.COLORS["black"] and piece == Match.PIECES["bKg"] ): reverse_dir = REVERSE_DIRS[direction] reverse_dir, stepx, stepy = rook.rk_step(reverse_dir, None, None, None, None) dstx, dsty = search(match, scrx, srcy, stepx, stepy) if dstx != UNDEF_X: piece = match.readfield(dstx, dsty) if color == Match.COLORS["white"]: if piece == Match.PIECES["bQu"] or piece == Match.PIECES["bRk"]: return direction else: return DIRS["undefined"] else: if piece == Match.PIECES["wQu"] or piece == Match.PIECES["wRk"]: return direction else: return DIRS["undefined"] direction, stepx, stepy = bishop.bp_step(None, scrx, srcy, kgx, kgy) if direction != DIRS["undefined"]: dstx, dsty = search(match, scrx, srcy, stepx, stepy) if dstx != UNDEF_X: piece = match.readfield(dstx, dsty) if (color == Match.COLORS["white"] and piece == Match.PIECES["wKg"]) or ( color == Match.COLORS["black"] and piece == Match.PIECES["bKg"] ): reverse_dir = REVERSE_DIRS[direction] reverse_dir, stepx, stepy = bishop.bp_step(reverse_dir, None, None, None, None) dstx, dsty = search(match, scrx, srcy, stepx, stepy) if dstx != UNDEF_X: piece = match.readfield(dstx, dsty) if color == Match.COLORS["white"]: if piece == Match.PIECES["bQu"] or piece == Match.PIECES["bBp"]: return direction else: return DIRS["undefined"] else: if piece == Match.PIECES["wQu"] or piece == Match.PIECES["wBp"]: return direction else: return DIRS["undefined"] return DIRS["undefined"]
def evaluate_contacts(match): supported_whites = 0 attacked_whites = 0 supported_blacks = 0 attacked_blacks = 0 color = match.next_color() for y in range(0, 8, 1): for x in range(0, 8, 1): piece = match.readfield(x, y) if(Match.color_of_piece(piece) == Match.COLORS['undefined']): continue elif(Match.color_of_piece(piece) == Match.COLORS['white']): supported_whites += rules.count_attacks(match, x, y, Match.COLORS['white']) attacked_whites += rules.count_attacks(match, x, y, Match.COLORS['black']) else: supported_blacks += rules.count_attacks(match, x, y, Match.COLORS['black']) attacked_blacks += rules.count_attacks(match, x, y, Match.COLORS['white']) return (supported_whites - attacked_whites) - (supported_blacks - attacked_blacks)
def evaluate_developments(match): developed_whites = 0 developed_blacks = 0 for y in range(0, 8, 1): for x in range(0, 8, 1): piece = match.readfield(x, y) if(Match.color_of_piece(piece) == Match.COLORS['undefined']): continue elif(Match.color_of_piece(piece) == Match.COLORS['white']): if(piece == Match.PIECES['wKn']): if(y > 0): developed_whites += 3 elif(piece == Match.PIECES['wBp']): if(y > 0): developed_whites += 2 elif(piece == Match.PIECES['wQu']): if(y > 0): developed_whites += 1 elif(piece == Match.PIECES['wKg']): if(y == 0 and (x == 6 or x == 2)): developed_whites += 2 else: if(piece == Match.PIECES['bKn']): if(y < 7): developed_blacks -= 3 elif(piece == Match.PIECES['bBp']): if(y < 7): developed_blacks -= 2 elif(piece == Match.PIECES['bQu']): if(y < 7): developed_blacks -= 1 elif(piece == Match.PIECES['bKg']): if(y == 7 and (x == 6 or x == 2)): developed_blacks -= 2 return developed_whites + developed_blacks
def is_move_valid(match, srcx, srcy, dstx, dsty, prom_piece): # print(" counts " + str(match.wKg_first_movecnt) + " " + str(match.wRk_h1_first_movecnt) + " " + str(match.wRk_a1_first_movecnt)) # print(" counts " + str(match.bKg_first_movecnt) + " " + str(match.bRk_h8_first_movecnt) + " " + str(match.bRk_a8_first_movecnt)) # print("-------------------------------------------") if not is_move_inbounds(srcx, srcy, dstx, dsty): return False, ERROR_CODES["out-of-bounds"] piece = match.readfield(srcx, srcy) if match.next_color() != Match.color_of_piece(piece): return False, ERROR_CODES["wrong-color"] if piece != Match.PIECES["wKg"] and piece != Match.PIECES["bKg"]: if is_king_after_move_attacked(match, srcx, srcy, dstx, dsty): return False, ERROR_CODES["king-error"] if piece == Match.PIECES["wPw"] or piece == Match.PIECES["bPw"]: if not pawn.is_move_ok(match, srcx, srcy, dstx, dsty, piece, prom_piece): return False, ERROR_CODES["pawn-error"] else: return True, ERROR_CODES["none"] elif piece == Match.PIECES["wRk"] or piece == Match.PIECES["bRk"]: if not rook.is_move_ok(match, srcx, srcy, dstx, dsty, piece): return False, ERROR_CODES["rook-error"] else: return True, ERROR_CODES["none"] elif piece == Match.PIECES["wKn"] or piece == Match.PIECES["bKn"]: if not knight.is_move_ok(match, srcx, srcy, dstx, dsty, piece): return False, ERROR_CODES["knight-error"] else: return True, ERROR_CODES["none"] elif piece == Match.PIECES["wBp"] or piece == Match.PIECES["bBp"]: if not bishop.is_move_ok(match, srcx, srcy, dstx, dsty, piece): return False, ERROR_CODES["bishop-error"] else: return True, ERROR_CODES["none"] elif piece == Match.PIECES["wQu"] or piece == Match.PIECES["bQu"]: if not queen.is_move_ok(match, srcx, srcy, dstx, dsty, piece): return False, ERROR_CODES["queen-error"] else: return True, ERROR_CODES["none"] elif piece == Match.PIECES["wKg"] or piece == Match.PIECES["bKg"]: if not king.is_move_ok(match, srcx, srcy, dstx, dsty, piece): return False, ERROR_CODES["king-error"] else: return True, ERROR_CODES["none"] else: return False, ERROR_CODES["general-error"]
def is_move_ok(match, srcx, srcy, dstx, dsty, piece, prom_piece): DIRS = rules.DIRS direction = pw_dir(srcx, srcy, dstx, dsty, piece) if(direction == DIRS['undefined']): return False pin_dir = rules.pin_dir(match, srcx, srcy) if(direction == DIRS['north'] or direction == DIRS['south'] or direction == DIRS['2north'] or direction == DIRS['2south']): if(pin_dir != DIRS['north'] and pin_dir != DIRS['south'] and pin_dir != DIRS['undefined']): return False elif(direction == DIRS['north-west'] or direction == DIRS['south-east']): if(pin_dir != DIRS['north-west'] and pin_dir != DIRS['south-east'] and pin_dir != DIRS['undefined']): return False elif(direction == DIRS['north-east'] or direction == DIRS['south-west']): if(pin_dir != DIRS['north-east'] and pin_dir != DIRS['south-west'] and pin_dir != DIRS['undefined']): return False dstpiece = match.readfield(dstx, dsty) dstcolor = Match.color_of_piece(dstpiece) if(direction == DIRS['north'] or direction == DIRS['south']): if(dstpiece != Match.PIECES['blk']): return False elif(direction == DIRS['2north']): midpiece = match.readfield(dstx, srcy + 1) if(midpiece != Match.PIECES['blk'] or dstpiece != Match.PIECES['blk']): return False elif(direction == DIRS['2south']): midpiece = match.readfield(dstx, srcy - 1) if(midpiece != Match.PIECES['blk'] or dstpiece != Match.PIECES['blk']): return False if(direction == DIRS['north-west'] or direction == DIRS['north-east']): if(dstcolor != Match.COLORS['black']): return is_white_ep_move_ok(match, srcx, srcy, dstx, dsty) elif(direction == DIRS['south-east'] or direction == DIRS['south-west']): if(dstcolor != Match.COLORS['white']): return is_black_ep_move_ok(match, srcx, srcy, dstx, dsty) if(piece == Match.PIECES['wPw'] and dsty == 7 and not (prom_piece == Match.PIECES['wQu'] or prom_piece == Match.PIECES['wRk'] or prom_piece == Match.PIECES['wBp'] or prom_piece == Match.PIECES['wKn'])): return False elif(piece == Match.PIECES['bPw'] and dsty == 0 and not (prom_piece == Match.PIECES['bQu'] or prom_piece == Match.PIECES['bRk'] or prom_piece == Match.PIECES['bBp'] or prom_piece == Match.PIECES['bKn'])): return False return True
def evaluate_piece_moves(match, srcx, srcy): color = match.next_color() piece = match.readfield(srcx, srcy) movecnt = 0 if(Match.color_of_piece(piece) != color): return movecnt if(piece == Match.PIECES['wQu'] or piece == Match.PIECES['bQu']): dirs = [ [0, 1], [0, -1], [1, 0], [-1, 0], [1, 1], [-1, -1], [-1, 1], [1, -1] ] dircnt = 8 stepcnt = 7 value = 2 elif(piece == Match.PIECES['wRk'] or piece == Match.PIECES['bRk']): dirs = [ [0, 1], [0, -1], [1, 0], [-1, 0] ] dircnt = 4 stepcnt = 7 value = 4 elif(piece == Match.PIECES['wBp'] or piece == Match.PIECES['bBp']): dirs = [ [1, 1], [-1, -1], [-1, 1], [1, -1] ] dircnt = 4 stepcnt = 7 value = 6 elif(piece == Match.PIECES['wKn'] or piece == Match.PIECES['bKn']): dirs = [ [1, 2], [2, 1], [2, -1], [1, -2], [-1, -2], [-2, -1], [-2, 1], [-1, 2] ] dircnt = 8 stepcnt = 1 value = 6 else: return movecnt for j in range(dircnt): stepx = dirs[j][0] stepy = dirs[j][1] dstx = srcx dsty = srcy for i in range(stepcnt): dstx += stepx dsty += stepy flag,errcode = rules.is_move_valid(match, srcx, srcy, dstx, dsty, Match.PIECES['blk']) if(flag): movecnt += value elif(errcode == rules.ERROR_CODES['out-of-bounds']): break return (movecnt // 10)
def is_move_available(match): color = match.next_color() for y1 in range(8): for x1 in range(8): piece = match.readfield(x1, y1) if color == Match.color_of_piece(piece): if color == Match.COLORS["white"]: prom_piece = Match.PIECES["wQu"] else: prom_piece = Match.PIECES["bQu"] for y2 in range(8): for x2 in range(8): flag = is_move_valid(match, x1, y1, x2, y2, prom_piece)[0] if flag: return True return False
def is_king_after_move_attacked(match, srcx, srcy, dstx, dsty): piece = match.readfield(srcx, srcy) match.writefield(srcx, srcy, Match.PIECES["blk"]) dstpiece = match.readfield(dstx, dsty) match.writefield(dstx, dsty, piece) color = Match.color_of_piece(piece) if color == Match.COLORS["white"]: flag = attacked(match, match.wKg_x, match.wKg_y, Match.COLORS["black"]) else: flag = attacked(match, match.bKg_x, match.bKg_y, Match.COLORS["white"]) match.writefield(dstx, dsty, dstpiece) match.writefield(srcx, srcy, piece) return flag
def is_move_ok(match, srcx, srcy, dstx, dsty, piece): DIRS = rules.DIRS direction = kn_dir(srcx, srcy, dstx, dsty) if(direction == DIRS['undefined']): return False color = Match.color_of_piece(piece) pin_dir = rules.pin_dir(match, srcx, srcy) if(pin_dir != DIRS['undefined']): return False field = match.readfield(dstx, dsty) if(match.color_of_piece(field) == color): return False return True
def generate_move(self): gmove = GenMove() color = self.match.next_color() while(self.active): if(self.steps == None): piece = self.match.readfield(self.board_x, self.board_y) if(piece == Match.PIECES['blk'] or color != Match.color_of_piece(piece)): if(self.rotate_field()): continue else: return False, gmove else: if(piece == Match.PIECES['wPw']): if(self.board_y < 6): self.steps = WPW_STEPS self.dir_idx = 0 self.max_dir = 4 self.step_idx = 0 self.max_step = 1 else: self.steps = WPROM_STEPS self.dir_idx = 0 self.max_dir = 3 self.step_idx = 0 self.max_step = 4 elif(piece == Match.PIECES['bPw']): if(self.board_y > 1): self.steps = BPW_STEPS self.dir_idx = 0 self.max_dir = 4 self.step_idx = 0 self.max_step = 1 else: self.steps = BPROM_STEPS self.dir_idx = 0 self.max_dir = 3 self.step_idx = 0 self.max_step = 4 elif(piece == Match.PIECES['wRk'] or piece == Match.PIECES['bRk']): self.steps = RK_STEPS self.dir_idx = 0 self.max_dir = 4 self.step_idx = 0 self.max_step = 7 elif(piece == Match.PIECES['wBp'] or piece == Match.PIECES['bBp']): self.steps = BP_STEPS self.dir_idx = 0 self.max_dir = 4 self.step_idx = 0 self.max_step = 7 elif(piece == Match.PIECES['wKn'] or piece == Match.PIECES['bKn']): self.steps = KN_STEPS self.dir_idx = 0 self.max_dir = 8 self.step_idx = 0 self.max_step = 1 elif(piece == Match.PIECES['wQu'] or piece == Match.PIECES['bQu']): self.steps = QU_STEPS self.dir_idx = 0 self.max_dir = 8 self.step_idx = 0 self.max_step = 7 else: self.steps = KG_STEPS self.dir_idx = 0 self.max_dir = 10 self.step_idx = 0 self.max_step = 1 stepx, stepy, prom_piece = self.read_steps() dstx = self.board_x + stepx dsty = self.board_y + stepy flag, errmsg = rules.is_move_valid(self.match, self.board_x, self.board_y, dstx, dsty, prom_piece) if(flag): gmove = GenMove(self.board_x, self.board_y, dstx, dsty, prom_piece) self.rotate() return True, gmove else: if(errmsg == rules.ERROR_CODES['king-error']): self.rotate() else: if(self.rotate_dir() == False): return False, gmove return False, gmove