diff --git a/src/movegen.cpp b/src/movegen.cpp index 289b51ad..cec98f00 100644 --- a/src/movegen.cpp +++ b/src/movegen.cpp @@ -35,7 +35,7 @@ namespace { enum CastlingSide { KING_SIDE, QUEEN_SIDE }; template - MoveStack* generate_castle_moves(const Position& pos, MoveStack* mlist, Color us) { + MoveStack* generate_castle(const Position& pos, MoveStack* mlist, Color us) { const CastleRight CR[] = { Side ? WHITE_OOO : WHITE_OO, Side ? BLACK_OOO : BLACK_OO }; @@ -96,7 +96,7 @@ namespace { : Delta == DELTA_NE ? (p & ~FileHBB) << 9 : Delta == DELTA_SE ? (p & ~FileHBB) >> 7 : Delta == DELTA_NW ? (p & ~FileABB) << 7 - : Delta == DELTA_SW ? (p & ~FileABB) >> 9 : p; + : Delta == DELTA_SW ? (p & ~FileABB) >> 9 : 0; } @@ -119,8 +119,8 @@ namespace { (*mlist++).move = make_promotion(to - Delta, to, KNIGHT); } - // Knight-promotion is the only one that can give a check (direct or - // discovered) not already included in the queen-promotion. + // Knight-promotion is the only one that can give a direct check not + // already included in the queen-promotion. if (Type == MV_NON_CAPTURE_CHECK && bit_is_set(StepAttacksBB[W_KNIGHT][to], ksq)) (*mlist++).move = make_promotion(to - Delta, to, KNIGHT); else @@ -134,7 +134,7 @@ namespace { template MoveStack* generate_pawn_moves(const Position& pos, MoveStack* mlist, Bitboard target, Square ksq = SQ_NONE) { - // Calculate our parametrized parameters at compile time, named according to + // Compute our parametrized parameters at compile time, named according to // the point of view of white side. const Color Them = (Us == WHITE ? BLACK : WHITE); const Bitboard TRank8BB = (Us == WHITE ? Rank8BB : Rank1BB); @@ -240,43 +240,24 @@ namespace { Color us, const CheckInfo& ci) { assert(Pt != KING && Pt != PAWN); - Square from; - const Square* pl = pos.piece_list(us, Pt); - - if ((from = *pl++) == SQ_NONE) - return mlist; - - Bitboard checkSqs = ci.checkSq[Pt] & pos.empty_squares(); - - do - { - if ( (Pt == BISHOP || Pt == ROOK || Pt == QUEEN) - && !(PseudoAttacks[Pt][from] & checkSqs)) - continue; - - if (ci.dcCandidates && bit_is_set(ci.dcCandidates, from)) - continue; - - Bitboard b = pos.attacks_from(from) & checkSqs; - SERIALIZE(b); - - } while ((from = *pl++) != SQ_NONE); - - return mlist; - } - - - template - FORCE_INLINE MoveStack* generate_piece_moves(const Position& pos, MoveStack* mlist, Color us, Bitboard target) { - - Bitboard b; + Bitboard b, target; Square from; const Square* pl = pos.piece_list(us, Pt); if (*pl != SQ_NONE) { + target = ci.checkSq[Pt] & pos.empty_squares(); // Non capture checks only + do { from = *pl; + + if ( (Pt == BISHOP || Pt == ROOK || Pt == QUEEN) + && !(PseudoAttacks[Pt][from] & target)) + continue; + + if (ci.dcCandidates && bit_is_set(ci.dcCandidates, from)) + continue; + b = pos.attacks_from(from) & target; SERIALIZE(b); } while (*++pl != SQ_NONE); @@ -286,9 +267,29 @@ namespace { } - template<> - FORCE_INLINE MoveStack* generate_piece_moves(const Position& pos, MoveStack* mlist, Color us, Bitboard target) { + template + FORCE_INLINE MoveStack* generate_moves(const Position& pos, MoveStack* mlist, + Color us, Bitboard target) { + assert(Pt != KING && Pt != PAWN); + Bitboard b; + Square from; + const Square* pl = pos.piece_list(us, Pt); + + if (*pl != SQ_NONE) + do { + from = *pl; + b = pos.attacks_from(from) & target; + SERIALIZE(b); + } while (*++pl != SQ_NONE); + + return mlist; + } + + + template<> + FORCE_INLINE MoveStack* generate_moves(const Position& pos, MoveStack* mlist, + Color us, Bitboard target) { Square from = pos.king_square(us); Bitboard b = pos.attacks_from(from) & target; SERIALIZE(b); @@ -328,16 +329,16 @@ MoveStack* generate(const Position& pos, MoveStack* mlist) { mlist = (us == WHITE ? generate_pawn_moves(pos, mlist, target) : generate_pawn_moves(pos, mlist, target)); - mlist = generate_piece_moves(pos, mlist, us, target); - mlist = generate_piece_moves(pos, mlist, us, target); - mlist = generate_piece_moves(pos, mlist, us, target); - mlist = generate_piece_moves(pos, mlist, us, target); - mlist = generate_piece_moves(pos, mlist, us, target); + mlist = generate_moves(pos, mlist, us, target); + mlist = generate_moves(pos, mlist, us, target); + mlist = generate_moves(pos, mlist, us, target); + mlist = generate_moves(pos, mlist, us, target); + mlist = generate_moves(pos, mlist, us, target); if (Type != MV_CAPTURE && pos.can_castle(us)) { - mlist = generate_castle_moves(pos, mlist, us); - mlist = generate_castle_moves(pos, mlist, us); + mlist = generate_castle(pos, mlist, us); + mlist = generate_castle(pos, mlist, us); } return mlist; @@ -386,8 +387,8 @@ MoveStack* generate(const Position& pos, MoveStack* mlist) if (pos.can_castle(us)) { - mlist = generate_castle_moves(pos, mlist, us); - mlist = generate_castle_moves(pos, mlist, us); + mlist = generate_castle(pos, mlist, us); + mlist = generate_castle(pos, mlist, us); } return mlist; @@ -458,10 +459,10 @@ MoveStack* generate(const Position& pos, MoveStack* mlist) { mlist = (us == WHITE ? generate_pawn_moves(pos, mlist, target) : generate_pawn_moves(pos, mlist, target)); - mlist = generate_piece_moves(pos, mlist, us, target); - mlist = generate_piece_moves(pos, mlist, us, target); - mlist = generate_piece_moves(pos, mlist, us, target); - return generate_piece_moves(pos, mlist, us, target); + mlist = generate_moves(pos, mlist, us, target); + mlist = generate_moves(pos, mlist, us, target); + mlist = generate_moves(pos, mlist, us, target); + return generate_moves(pos, mlist, us, target); }