static ChessBoolean move_is_legal(const ChessPosition* position, ChessMove move) { ChessPosition temp_position; chess_position_copy(position, &temp_position); chess_position_make_move(&temp_position, move); chess_position_set_to_move(&temp_position, chess_position_to_move(position)); return !chess_position_is_check(&temp_position); }
ChessResult chess_position_check_result(const ChessPosition* position) { ChessMoveGenerator generator; chess_move_generator_init(&generator, position); if (chess_move_generator_next(&generator) > 0) return CHESS_RESULT_NONE; if (!chess_position_is_check(position)) return CHESS_RESULT_DRAW; /* Stalemate */ return (position->to_move == CHESS_COLOR_WHITE) ? CHESS_RESULT_BLACK_WINS : CHESS_RESULT_WHITE_WINS; }
static ChessMove gen_next(ChessMoveGenerator* gen) { const ChessPosition* position = gen->position; ChessPiece piece; ChessColor color = chess_position_to_move(position); ChessPiece target; int dirs, dir; int piece_dirs; ChessRank start_rank, end_rank; int slide; int capture_dirs; ChessFile ep_file; ChessSquare ep; ChessMove move; if (gen->promote != CHESS_MOVE_PROMOTE_NONE) { gen_pawn_promotes: if (++gen->promote <= CHESS_MOVE_PROMOTE_QUEEN) return chess_move_make_promote(gen->sq, gen->to, gen->promote); gen->promote = CHESS_MOVE_PROMOTE_NONE; } for (; gen->sq <= CHESS_SQUARE_H8; gen->sq++) { piece = chess_position_piece(position, gen->sq); if (piece == CHESS_PIECE_NONE) continue; if (color != chess_piece_color(piece)) continue; switch (piece) { case CHESS_PIECE_WHITE_PAWN: case CHESS_PIECE_BLACK_PAWN: start_rank = (color == CHESS_COLOR_WHITE) ? CHESS_RANK_2 : CHESS_RANK_7; end_rank = (color == CHESS_COLOR_WHITE) ? CHESS_RANK_8 : CHESS_RANK_1; slide = (color == CHESS_COLOR_WHITE) ? SLIDE_N : SLIDE_S; capture_dirs = (color == CHESS_COLOR_WHITE) ? DIR_NE | DIR_NW : DIR_SE | DIR_SW; dirs = capture_dirs & slide_dirs[gen->sq]; ep_file = chess_position_ep(position); ep = (ep_file == CHESS_FILE_INVALID) ? CHESS_SQUARE_INVALID : chess_square_from_fr(ep_file, (color == CHESS_COLOR_WHITE) ? CHESS_RANK_6 : CHESS_RANK_3); if (gen->d == 0) { if (gen->to == CHESS_SQUARE_INVALID) { gen->to = gen->sq + slide; if (chess_position_piece(position, gen->to) == CHESS_PIECE_NONE) { if (chess_square_rank(gen->to) == end_rank) goto gen_pawn_promotes; else return chess_move_make(gen->sq, gen->to); } } else if (chess_square_rank(gen->sq) == start_rank && gen->to == gen->sq + slide) { gen->to += slide; if (chess_position_piece(position, gen->to) == CHESS_PIECE_NONE) return chess_move_make(gen->sq, gen->to); } gen->to = CHESS_SQUARE_INVALID; gen->d = 1; } while (gen->d < 8) { if (dirs_array[gen->d] & dirs) { gen->to = gen->sq + slides_array[gen->d]; piece = chess_position_piece(position, gen->to); gen->d += 2; if (piece != CHESS_PIECE_NONE && chess_piece_color(piece) != color) { if (chess_square_rank(gen->to) == end_rank) goto gen_pawn_promotes; else return chess_move_make(gen->sq, gen->to); } else if (gen->to == ep) { return chess_move_make(gen->sq, gen->to); } } else { gen->d += 2; } } gen->to = CHESS_SQUARE_INVALID; gen->d = 0; break; case CHESS_PIECE_WHITE_KNIGHT: case CHESS_PIECE_BLACK_KNIGHT: dirs = jump_dirs[gen->sq]; for (; gen->d < 8; gen->d++) { dir = dirs_array[gen->d]; if ((dir & dirs) == 0) continue; gen->to = gen->sq + jumps_array[gen->d]; target = chess_position_piece(position, gen->to); if (target == CHESS_PIECE_NONE || chess_piece_color(target) != color) { gen->d++; return chess_move_make(gen->sq, gen->to); } } gen->to = CHESS_SQUARE_INVALID; gen->d = 0; break; case CHESS_PIECE_WHITE_BISHOP: case CHESS_PIECE_BLACK_BISHOP: piece_dirs = bishop_dirs; goto gen_slide_moves; case CHESS_PIECE_WHITE_ROOK: case CHESS_PIECE_BLACK_ROOK: piece_dirs = rook_dirs; goto gen_slide_moves; case CHESS_PIECE_WHITE_QUEEN: case CHESS_PIECE_BLACK_QUEEN: piece_dirs = queen_dirs; gen_slide_moves: while (gen->d < 8) { dir = dirs_array[gen->d] & piece_dirs; if (gen->to == CHESS_SQUARE_INVALID) gen->to = gen->sq; do { dirs = slide_dirs[gen->to]; if ((dir & dirs) == 0) break; gen->to += slides_array[gen->d]; target = chess_position_piece(position, gen->to); if (target == CHESS_PIECE_NONE) return chess_move_make(gen->sq, gen->to); else if (chess_piece_color(target) != color) { move = chess_move_make(gen->sq, gen->to); gen->d++; gen->to = CHESS_SQUARE_INVALID; return move; } } while (target == CHESS_PIECE_NONE); gen->to = CHESS_SQUARE_INVALID; gen->d++; } gen->to = CHESS_SQUARE_INVALID; gen->d = 0; break; case CHESS_PIECE_WHITE_KING: case CHESS_PIECE_BLACK_KING: dirs = slide_dirs[gen->sq]; for (; gen->d < 8; gen->d++) { dir = dirs_array[gen->d]; if ((dir & dirs) == 0) continue; gen->to = gen->sq + slides_array[gen->d]; target = chess_position_piece(position, gen->to); if (target == CHESS_PIECE_NONE || chess_piece_color(target) != color) { gen->d++; return chess_move_make(gen->sq, gen->to); } } gen->to = CHESS_SQUARE_INVALID; gen->d = 0; break; default: assert(0); break; } } if (gen->castle == -1) { if (!chess_position_is_check(position)) gen->castle = chess_position_castle(position); else { gen->castle = CHESS_CASTLE_STATE_NONE; } } if (color == CHESS_COLOR_WHITE) { if ((gen->castle & CHESS_CASTLE_STATE_WK) && chess_position_piece(position, CHESS_SQUARE_F1) == CHESS_PIECE_NONE && chess_position_piece(position, CHESS_SQUARE_G1) == CHESS_PIECE_NONE && !chess_generate_is_square_attacked(position, CHESS_SQUARE_F1, CHESS_COLOR_BLACK) && !chess_generate_is_square_attacked(position, CHESS_SQUARE_G1, CHESS_COLOR_BLACK)) { gen->castle &= ~CHESS_CASTLE_STATE_WK; return chess_move_make(CHESS_SQUARE_E1, CHESS_SQUARE_G1); } if ((gen->castle & CHESS_CASTLE_STATE_WQ) && chess_position_piece(position, CHESS_SQUARE_B1) == CHESS_PIECE_NONE && chess_position_piece(position, CHESS_SQUARE_C1) == CHESS_PIECE_NONE && chess_position_piece(position, CHESS_SQUARE_D1) == CHESS_PIECE_NONE && !chess_generate_is_square_attacked(position, CHESS_SQUARE_D1, CHESS_COLOR_BLACK) && !chess_generate_is_square_attacked(position, CHESS_SQUARE_C1, CHESS_COLOR_BLACK)) { gen->castle &= ~CHESS_CASTLE_STATE_WQ; return chess_move_make(CHESS_SQUARE_E1, CHESS_SQUARE_C1); } } else { if ((gen->castle & CHESS_CASTLE_STATE_BK) && chess_position_piece(position, CHESS_SQUARE_F8) == CHESS_PIECE_NONE && chess_position_piece(position, CHESS_SQUARE_G8) == CHESS_PIECE_NONE && !chess_generate_is_square_attacked(position, CHESS_SQUARE_F8, CHESS_COLOR_WHITE) && !chess_generate_is_square_attacked(position, CHESS_SQUARE_G8, CHESS_COLOR_WHITE)) { gen->castle &= ~CHESS_CASTLE_STATE_BK; return chess_move_make(CHESS_SQUARE_E8, CHESS_SQUARE_G8); } if ((gen->castle & CHESS_CASTLE_STATE_BQ) && chess_position_piece(position, CHESS_SQUARE_B8) == CHESS_PIECE_NONE && chess_position_piece(position, CHESS_SQUARE_C8) == CHESS_PIECE_NONE && chess_position_piece(position, CHESS_SQUARE_D8) == CHESS_PIECE_NONE && !chess_generate_is_square_attacked(position, CHESS_SQUARE_D8, CHESS_COLOR_WHITE) && !chess_generate_is_square_attacked(position, CHESS_SQUARE_C8, CHESS_COLOR_WHITE)) { gen->castle &= ~CHESS_CASTLE_STATE_BQ; return chess_move_make(CHESS_SQUARE_E8, CHESS_SQUARE_C8); } } return 0; }