1
0
Fork 0
mirror of https://github.com/sockspls/badfish synced 2025-04-30 08:43:09 +00:00

Cleanup handling of Delta enums

Ispired by Rein's code.

No functional change.

Signed-off-by: Marco Costalba <mcostalba@gmail.com>
This commit is contained in:
Marco Costalba 2011-07-15 11:19:49 +02:00
parent fbdabe2975
commit 9c8c0de538
2 changed files with 28 additions and 26 deletions

View file

@ -46,13 +46,13 @@ namespace {
inline MoveStack* generate_discovered_checks(const Position& pos, MoveStack* mlist, Square from) { inline MoveStack* generate_discovered_checks(const Position& pos, MoveStack* mlist, Square from) {
assert(Pt != QUEEN); assert(Pt != QUEEN);
assert(Pt != PAWN);
Bitboard b = pos.attacks_from<Pt>(from) & pos.empty_squares(); Bitboard b = pos.attacks_from<Pt>(from) & pos.empty_squares();
if (Pt == KING) if (Pt == KING)
{ b &= ~QueenPseudoAttacks[pos.king_square(opposite_color(pos.side_to_move()))];
Square ksq = pos.king_square(opposite_color(pos.side_to_move()));
b &= ~QueenPseudoAttacks[ksq];
}
SERIALIZE_MOVES(b); SERIALIZE_MOVES(b);
return mlist; return mlist;
} }
@ -61,6 +61,7 @@ namespace {
inline MoveStack* generate_direct_checks(const Position& pos, MoveStack* mlist, Color us, inline MoveStack* generate_direct_checks(const Position& pos, MoveStack* mlist, Color us,
Bitboard dc, Square ksq) { Bitboard dc, Square ksq) {
assert(Pt != KING); assert(Pt != KING);
assert(Pt != PAWN);
Bitboard checkSqs, b; Bitboard checkSqs, b;
Square from; Square from;
@ -357,10 +358,11 @@ namespace {
return mlist; return mlist;
} }
template<Color Us, MoveType Type, Square Delta> template<MoveType Type, Square Delta>
inline MoveStack* generate_promotions(const Position& pos, MoveStack* mlist, Bitboard pawnsOn7, Bitboard target) { inline MoveStack* generate_promotions(const Position& pos, MoveStack* mlist, Bitboard pawnsOn7, Bitboard target) {
const Bitboard TFileABB = (Delta == DELTA_NE || Delta == DELTA_SE ? FileABB : FileHBB); const Bitboard TFileABB = (Delta == DELTA_NE || Delta == DELTA_SE ? FileABB : FileHBB);
const Color Them = (Delta > 0 ? BLACK : WHITE);
Bitboard b; Bitboard b;
Square to; Square to;
@ -388,7 +390,7 @@ namespace {
// This is the only possible under promotion that can give a check // This is the only possible under promotion that can give a check
// not already included in the queen-promotion. // not already included in the queen-promotion.
if ( Type == MV_CHECK if ( Type == MV_CHECK
&& bit_is_set(pos.attacks_from<KNIGHT>(to), pos.king_square(opposite_color(Us)))) && bit_is_set(pos.attacks_from<KNIGHT>(to), pos.king_square(Them)))
(*mlist++).move = make_promotion_move(to - Delta, to, KNIGHT); (*mlist++).move = make_promotion_move(to - Delta, to, KNIGHT);
else (void)pos; // Silence a warning under MSVC else (void)pos; // Silence a warning under MSVC
} }
@ -403,9 +405,9 @@ namespace {
const Color Them = (Us == WHITE ? BLACK : WHITE); const Color Them = (Us == WHITE ? BLACK : WHITE);
const Bitboard TRank7BB = (Us == WHITE ? Rank7BB : Rank2BB); const Bitboard TRank7BB = (Us == WHITE ? Rank7BB : Rank2BB);
const Bitboard TRank3BB = (Us == WHITE ? Rank3BB : Rank6BB); const Bitboard TRank3BB = (Us == WHITE ? Rank3BB : Rank6BB);
const Square TDELTA_N = (Us == WHITE ? DELTA_N : DELTA_S); const Square UP = (Us == WHITE ? DELTA_N : DELTA_S);
const Square TDELTA_NE = (Us == WHITE ? DELTA_NE : DELTA_SE); const Square RIGHT_UP = (Us == WHITE ? DELTA_NE : DELTA_SE);
const Square TDELTA_NW = (Us == WHITE ? DELTA_NW : DELTA_SW); const Square LEFT_UP = (Us == WHITE ? DELTA_NW : DELTA_SW);
Square to; Square to;
Bitboard b1, b2, dc1, dc2, pawnPushes, emptySquares; Bitboard b1, b2, dc1, dc2, pawnPushes, emptySquares;
@ -417,7 +419,7 @@ namespace {
if (Type != MV_CAPTURE) if (Type != MV_CAPTURE)
{ {
emptySquares = (Type == MV_NON_CAPTURE ? target : pos.empty_squares()); emptySquares = (Type == MV_NON_CAPTURE ? target : pos.empty_squares());
pawnPushes = move_pawns<TDELTA_N>(pawns & ~TRank7BB) & emptySquares; pawnPushes = move_pawns<UP>(pawns & ~TRank7BB) & emptySquares;
} }
if (Type == MV_EVASION) if (Type == MV_EVASION)
@ -433,23 +435,23 @@ namespace {
emptySquares = pos.empty_squares(); emptySquares = pos.empty_squares();
pawns &= ~TRank7BB; pawns &= ~TRank7BB;
mlist = generate_promotions<Us, Type, TDELTA_NE>(pos, mlist, pawnsOn7, enemyPieces); mlist = generate_promotions<Type, RIGHT_UP>(pos, mlist, pawnsOn7, enemyPieces);
mlist = generate_promotions<Us, Type, TDELTA_NW>(pos, mlist, pawnsOn7, enemyPieces); mlist = generate_promotions<Type, LEFT_UP>(pos, mlist, pawnsOn7, enemyPieces);
mlist = generate_promotions<Us, Type, TDELTA_N >(pos, mlist, pawnsOn7, emptySquares); mlist = generate_promotions<Type, UP>(pos, mlist, pawnsOn7, emptySquares);
} }
// Standard captures // Standard captures
if (Type == MV_CAPTURE || Type == MV_EVASION) if (Type == MV_CAPTURE || Type == MV_EVASION)
{ {
mlist = generate_pawn_captures<Type, TDELTA_NE>(mlist, pawns, enemyPieces); mlist = generate_pawn_captures<Type, RIGHT_UP>(mlist, pawns, enemyPieces);
mlist = generate_pawn_captures<Type, TDELTA_NW>(mlist, pawns, enemyPieces); mlist = generate_pawn_captures<Type, LEFT_UP>(mlist, pawns, enemyPieces);
} }
// Single and double pawn pushes // Single and double pawn pushes
if (Type != MV_CAPTURE) if (Type != MV_CAPTURE)
{ {
b1 = (Type != MV_EVASION ? pawnPushes : pawnPushes & emptySquares); b1 = (Type != MV_EVASION ? pawnPushes : pawnPushes & emptySquares);
b2 = move_pawns<TDELTA_N>(pawnPushes & TRank3BB) & emptySquares; b2 = move_pawns<UP>(pawnPushes & TRank3BB) & emptySquares;
if (Type == MV_CHECK) if (Type == MV_CHECK)
{ {
@ -462,15 +464,15 @@ namespace {
// don't generate captures. // don't generate captures.
if (pawns & target) // For CHECK type target is dc bitboard if (pawns & target) // For CHECK type target is dc bitboard
{ {
dc1 = move_pawns<TDELTA_N>(pawns & target & ~file_bb(ksq)) & emptySquares; dc1 = move_pawns<UP>(pawns & target & ~file_bb(ksq)) & emptySquares;
dc2 = move_pawns<TDELTA_N>(dc1 & TRank3BB) & emptySquares; dc2 = move_pawns<UP>(dc1 & TRank3BB) & emptySquares;
b1 |= dc1; b1 |= dc1;
b2 |= dc2; b2 |= dc2;
} }
} }
SERIALIZE_MOVES_D(b1, -TDELTA_N); SERIALIZE_MOVES_D(b1, -UP);
SERIALIZE_MOVES_D(b2, -TDELTA_N -TDELTA_N); SERIALIZE_MOVES_D(b2, -UP -UP);
} }
// En passant captures // En passant captures
@ -482,7 +484,7 @@ namespace {
// An en passant capture can be an evasion only if the checking piece // An en passant capture can be an evasion only if the checking piece
// is the double pushed pawn and so is in the target. Otherwise this // is the double pushed pawn and so is in the target. Otherwise this
// is a discovery check and we are forced to do otherwise. // is a discovery check and we are forced to do otherwise.
if (Type == MV_EVASION && !bit_is_set(target, pos.ep_square() - TDELTA_N)) if (Type == MV_EVASION && !bit_is_set(target, pos.ep_square() - UP))
return mlist; return mlist;
b1 = pawns & pos.attacks_from<PAWN>(pos.ep_square(), Them); b1 = pawns & pos.attacks_from<PAWN>(pos.ep_square(), Them);

View file

@ -520,7 +520,7 @@ bool Position::pl_move_is_legal(Move m, Bitboard pinned) const {
{ {
Color them = opposite_color(us); Color them = opposite_color(us);
Square to = move_to(m); Square to = move_to(m);
Square capsq = to + Square(us == WHITE ? -8 : 8); Square capsq = to + pawn_push(them);
Square ksq = king_square(us); Square ksq = king_square(us);
Bitboard b = occupied_squares(); Bitboard b = occupied_squares();
@ -930,7 +930,7 @@ void Position::do_move(Move m, StateInfo& newSt, const CheckInfo& ci, bool moveI
// Set en passant square, only if moved pawn can be captured // Set en passant square, only if moved pawn can be captured
if ((to ^ from) == 16) if ((to ^ from) == 16)
{ {
if (attacks_from<PAWN>(from + (us == WHITE ? DELTA_N : DELTA_S), us) & pieces(PAWN, them)) if (attacks_from<PAWN>(from + pawn_push(us), us) & pieces(PAWN, them))
{ {
st->epSquare = Square((int(from) + int(to)) / 2); st->epSquare = Square((int(from) + int(to)) / 2);
key ^= zobEp[st->epSquare]; key ^= zobEp[st->epSquare];
@ -1039,7 +1039,7 @@ void Position::do_capture_move(Key& key, PieceType capture, Color them, Square t
{ {
if (ep) // en passant ? if (ep) // en passant ?
{ {
capsq = (them == BLACK)? (to - DELTA_N) : (to - DELTA_S); capsq = to + pawn_push(them);
assert(to == st->epSquare); assert(to == st->epSquare);
assert(relative_rank(opposite_color(them), to) == RANK_6); assert(relative_rank(opposite_color(them), to) == RANK_6);
@ -1262,7 +1262,7 @@ void Position::undo_move(Move m) {
Square capsq = to; Square capsq = to;
if (ep) if (ep)
capsq = (us == WHITE)? (to - DELTA_N) : (to - DELTA_S); capsq = to - pawn_push(us);
assert(st->capturedType != KING); assert(st->capturedType != KING);
assert(!ep || square_is_empty(capsq)); assert(!ep || square_is_empty(capsq));
@ -1467,7 +1467,7 @@ int Position::see(Move m) const {
// Handle en passant moves // Handle en passant moves
if (st->epSquare == to && piece_type(piece_on(from)) == PAWN) if (st->epSquare == to && piece_type(piece_on(from)) == PAWN)
{ {
Square capQq = (side_to_move() == WHITE ? to - DELTA_N : to - DELTA_S); Square capQq = to - pawn_push(side_to_move());
assert(capturedType == PIECE_TYPE_NONE); assert(capturedType == PIECE_TYPE_NONE);
assert(piece_type(piece_on(capQq)) == PAWN); assert(piece_type(piece_on(capQq)) == PAWN);