37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
import random
|
|
|
|
import chess
|
|
import chess.engine
|
|
|
|
from chesspp.i_strategy import IStrategy
|
|
from chesspp.eval import score_stockfish
|
|
|
|
|
|
class RandomStockfishStrategy(IStrategy):
|
|
def __init__(self, rollout_depth: int, path="../stockfish/stockfish-windows-x86-64-avx2",
|
|
random_seed: random.Random = random.Random()) -> None:
|
|
super().__init__(rollout_depth)
|
|
self._stockfish = None
|
|
self.path = path
|
|
self.random_seed = random_seed
|
|
|
|
def __del__(self):
|
|
if self._stockfish is not None:
|
|
self._stockfish.quit()
|
|
|
|
@property
|
|
def stockfish(self) -> chess.engine.SimpleEngine:
|
|
if self._stockfish is None:
|
|
self._stockfish = self.stockfish = chess.engine.SimpleEngine.popen_uci(self.path)
|
|
return self._stockfish
|
|
|
|
@stockfish.setter
|
|
def stockfish(self, stockfish):
|
|
self._stockfish = stockfish
|
|
|
|
def pick_next_move(self, board: chess.Board) -> chess.Move:
|
|
return self.random_seed.choice(list(board.legal_moves))
|
|
|
|
def analyze_board(self, board: chess.Board) -> int:
|
|
return score_stockfish(board, self.stockfish)
|