initial commit

engine.py

@@ -0,0 +1,67 @@
+import chess
+import chess.engine
+import random
+import eval
+
+
+def main():
+    fools_mate = "rnbqkbnr/pppp1ppp/4p3/8/5PP1/8/PPPPP2P/RNBQKBNR b KQkq f3 0 2"
+    board = chess.Board(fools_mate)
+    print(board, '\n')
+    moves = {}
+    for i in range(10):
+        move = pick_move(board)
+        if move is None:
+            break
+
+        simulate_game(board, move, 100)
+        moves[move] = board
+        board = chess.Board(fools_mate)
+
+    analyze_results(moves)
+
+
+def analyze_results(moves: dict):
+    for m, b in moves.items():
+        manual_score = eval.score_game(b)
+        engine_score = eval.analyze_with_stockfish(b)
+        print(f"score for move {m}: manual_score={manual_score}, engine_score={engine_score}")
+
+
+def pick_move(board: chess.Board) -> chess.Move | None:
+    """
+    Pick a random move
+    :param board: chess board
+    :return: a valid move or None if no valid move available
+    """
+    if len(list(board.legal_moves)) == 0:
+        return None
+    return random.choice(list(board.legal_moves))
+
+
+def simulate_game(board: chess.Board, move: chess.Move, depth: int):
+    """
+    Simulate a game starting with the given move
+    :param board: chess board
+    :param move: chosen move
+    :param depth: number of moves that should be simulated after playing the chosen move
+    :return: the score for the simulated game
+    """
+    engine = chess.engine.SimpleEngine.popen_uci("./stockfish/stockfish-ubuntu-x86-64-avx2")
+    board.push(move)
+    print(move)
+    print(board, '\n')
+    for i in range(depth):
+        if board.is_game_over():
+            engine.quit()
+            return
+        r = engine.play(board, chess.engine.Limit(depth=2))
+        print(r)
+        board.push(r.move)
+        print(board, '\n')
+
+    engine.quit()
+
+
+if __name__ == '__main__':
+    main()