def getPossibleMoves(self): currentPos = self.position movements = [C(0, 1), C(0, -1), C(1, 0), C(-1, 0), C(1, 1), C(1, -1), C(-1, 1), C(-1, -1)] for movement in movements: newPos = currentPos + movement if self.board.isValidPos(newPos): pieceAtNewPos = self.board.pieceAtPosition(newPos) if self.board.pieceAtPosition(newPos) is None: yield Move(self, newPos) elif pieceAtNewPos.side != self.side: yield Move(self, newPos, pieceToCapture=pieceAtNewPos) # Castling if self.movesMade == 0: inCheck = False kingsideCastleBlocked = False queensideCastleBlocked = False kingsideCastleCheck = False queensideCastleCheck = False kingsideRookMoved = True queensideRookMoved = True kingsideCastlePositions = [self.position + C(1, 0), self.position + C(2, 0)] for pos in kingsideCastlePositions: if self.board.pieceAtPosition(pos): kingsideCastleBlocked = True queensideCastlePositions = [self.position - C(1, 0), self.position - C(2, 0), self.position - C(3, 0)] for pos in queensideCastlePositions: if self.board.pieceAtPosition(pos): queensideCastleBlocked = True if kingsideCastleBlocked and queensideCastleBlocked: return otherSideMoves = self.board.getAllMovesUnfiltered(not self.side, includeKing=False) for move in otherSideMoves: if move.newPos == self.position: inCheck = True break if move.newPos == self.position + C(1, 0) or move.newPos == self.position + C(2, 0): kingsideCastleCheck = True if move.newPos == self.position - C(1, 0) or move.newPos == self.position - C(2, 0): queensideCastleCheck = True kingsideRookPos = self.position + C(3, 0) kingsideRook = self.board.pieceAtPosition(kingsideRookPos) if self.board.isValidPos(kingsideRookPos) else None if kingsideRook and kingsideRook.stringRep == 'R' and kingsideRook.movesMade == 0: kingsideRookMoved = False queensideRookPos = self.position - C(4, 0) queensideRook = self.board.pieceAtPosition(queensideRookPos) if self.board.isValidPos(queensideRookPos) else None if queensideRook and queensideRook.stringRep == 'R' and queensideRook.movesMade == 0: queensideRookMoved = False if not inCheck: if not kingsideCastleBlocked and not kingsideCastleCheck and not kingsideRookMoved: move = Move(self, self.position + C(2, 0)) rookMove = Move(self.position, self.position + C(1, 0)) move.specialMovePiece = self.board.pieceAtPosition(kingsideRookPos) move.kingsideCastle = True move.rookMove = rookMove yield move if not queensideCastleBlocked and not queensideCastleCheck and not queensideRookMoved: move = Move(self, self.position - C(2, 0)) rookMove = Move(self.position, self.position - C(1, 0)) move.specialMovePiece = self.board.pieceAtPosition(queensideRookPos) move.queensideCastle = True move.rookMove = rookMove yield move
def getPossibleMoves(self): # Function run to yield a group of all possible legal and illegal # 'Move' objects to the calling function. Does this by running through # a list of movements and testing for valid moves before yielding them. # Also goes through another special algorithm to test if it can also # perform another special type of move, a 'castle'. pos = self.position movements = [ C(0, 1), C(-1, -1), C(1, 0), C(-1, 0), C(1, 1), C(1, -1), C(-1, 1), C(0, -1) ] for movement in movements: newPos = pos + movement if self.board.isValidPos(newPos): pieceAtNewPos = self.board.pieceAtPosition(newPos) if self.board.pieceAtPosition(newPos) is None: yield Move(self, newPos) elif pieceAtNewPos.side != self.side: yield Move(self, newPos, pieceToCapture=pieceAtNewPos) if self.movesMade == 0: # CASTLING, HIGHLY WIP inCheck = False kingSideCastleBlocked = False queenSideCastleBlocked = False kingSideCastleCheck = False queenSideCastleCheck = False kingSideRookMoved = True queenSideRookMoved = True kingSideCastlePositions = [ self.position - C(1, 0), self.position - C(2, 0) ] for position in kingSideCastlePositions: if self.board.pieceAtPosition(position): kingSideCastleBlocked = True queenSideCastlePositions = [ self.position + C(1, 0), self.position + C(2, 0), self.position + C(3, 0) ] for position in queenSideCastlePositions: if self.board.pieceAtPosition(position): queenSideCastleBlocked = True if kingSideCastleBlocked and queenSideCastleBlocked: return opponentMoves = self.board.getAllMovesUnfiltered(not self.side, includeKing=False) for move in opponentMoves: if move.newPos == self.position: inCheck = True break if move.newPos == self.position - C(1, 0) or \ move.newPos == self.position - C(2, 0): kingSideCastleCheck = True if move.newPos == self.position + C(1, 0) or \ move.newPos == self.position + C(2, 0): queenSideCastleCheck = True kingSideRookPos = self.position - C(3, 0) kingSideRook = self.board.pieceAtPosition(kingSideRookPos) \ if self.board.isValidPos(kingSideRookPos) else None if kingSideRook and \ kingSideRook.stringRep == 'R' and \ kingSideRook.movesMade == 0: kingSideRookMoved = False queenSideRookPos = self.position + C(4, 0) queenSideRook = self.board.pieceAtPosition(queenSideRookPos) \ if self.board.isValidPos(kingSideRookPos) else None if queenSideRook and \ queenSideRook.stringRep == 'R' and \ queenSideRook.movesMade == 0: queenSideRookMoved = False if not inCheck: if not kingSideCastleBlocked and \ not kingSideCastleCheck and \ not kingSideRookMoved: move = Move(self, self.position - C(2, 0)) move.specialMovePiece = self.board.pieceAtPosition( kingSideRookPos) rookMove = Move(move.specialMovePiece, self.position - C(1, 0)) move.kingSideCastle = True move.rookMove = rookMove move.castle = True yield move if not queenSideCastleBlocked and \ not queenSideCastleCheck and \ not queenSideRookMoved: move = Move(self, self.position + C(2, 0)) move.specialMovePiece = self.board.pieceAtPosition( queenSideRookPos) rookMove = Move(move.specialMovePiece, self.position + C(1, 0)) move.queenSideCastle = True move.rookMove = rookMove move.castle = True yield move