cleanup
This commit is contained in:
parent
3b005ad506
commit
58d3f51a61
|
@ -1,140 +0,0 @@
|
|||
import os
|
||||
from pathlib import Path
|
||||
from pygame import mixer
|
||||
import random
|
||||
from stockfish import Stockfish
|
||||
import time
|
||||
|
||||
settings = {
|
||||
"Debug Log File": "",
|
||||
"Contempt": 0,
|
||||
"Min Split Depth": 0,
|
||||
"Threads": 1,
|
||||
# More threads will make the engine stronger, but should be kept at less than the
|
||||
# number of logical processors on your computer.
|
||||
"Ponder": "false",
|
||||
"Hash": 16,
|
||||
# Default size is 16 MB. It's recommended that you increase this value, but keep it
|
||||
# as some power of 2. E.g., if you're fine using 2 GB of RAM, set Hash to 2048
|
||||
# (11th power of 2).
|
||||
"MultiPV": 1,
|
||||
"Skill Level": 20,
|
||||
"Move Overhead": 10,
|
||||
"Minimum Thinking Time": 20,
|
||||
"Slow Mover": 100,
|
||||
"UCI_Chess960": "false",
|
||||
"UCI_LimitStrength": "false",
|
||||
"UCI_Elo": 1350,
|
||||
"NNUE": "true",
|
||||
}
|
||||
|
||||
sound_path = Path("../../sounds")
|
||||
|
||||
|
||||
def play_sound() -> None:
|
||||
"""
|
||||
Plays a random sound from the sound path
|
||||
:return: None
|
||||
"""
|
||||
mixer.init()
|
||||
mixer.music.load("sounds/" + random.choice(os.listdir(sound_path)))
|
||||
mixer.music.play()
|
||||
while mixer.music.get_busy():
|
||||
time.sleep(0.1)
|
||||
|
||||
|
||||
class Game:
|
||||
"""
|
||||
A class representing the game state
|
||||
"""
|
||||
|
||||
def __init__(self, engine_settings: dict):
|
||||
self.engine = Stockfish("/usr/bin/stockfish")
|
||||
self.settings = engine_settings
|
||||
self.engine.update_engine_parameters(self.settings)
|
||||
self.current_evaluation = (
|
||||
self.engine.get_evaluation()
|
||||
) # This is not necessary, now that I think about it.
|
||||
self.evaluations: list[dict] = []
|
||||
self.current_wdl = self.engine.get_wdl_stats()
|
||||
self.wdls: list[tuple[int, int, int]] = []
|
||||
self.engine.set_position()
|
||||
|
||||
def move_was_blunder(self) -> bool:
|
||||
"""
|
||||
Returns true if the last move was a blunder
|
||||
:return: bool
|
||||
"""
|
||||
if len(self.wdls) > 1: # Don't check for blunders on the first move
|
||||
previous_wdl = self.wdls[len(self.evaluations) - 2]
|
||||
if abs(previous_wdl[0] - self.current_wdl[0]) > 300:
|
||||
return True
|
||||
elif abs(previous_wdl[2] - self.current_wdl[2]) > 300:
|
||||
return True
|
||||
else:
|
||||
return False
|
||||
return False
|
||||
|
||||
def make_move(self, move) -> None:
|
||||
"""
|
||||
Makes a move on the board and updates the game state
|
||||
:param move: str
|
||||
:return: None
|
||||
"""
|
||||
if self.engine.is_move_correct(move):
|
||||
self.engine.make_moves_from_current_position([move])
|
||||
self.current_evaluation = self.engine.get_evaluation()
|
||||
self.evaluations.append(self.current_evaluation)
|
||||
self.current_wdl = self.engine.get_wdl_stats()
|
||||
self.wdls.append(self.current_wdl)
|
||||
print(self.current_wdl)
|
||||
print(self.current_evaluation)
|
||||
if self.move_was_blunder():
|
||||
# If the played move was a blunder play a random sound from the sound
|
||||
# path
|
||||
# play_sound()
|
||||
print("Blunder!")
|
||||
else:
|
||||
print("Invalid move")
|
||||
|
||||
def __str__(self):
|
||||
return ""
|
||||
|
||||
|
||||
test_game = Game(settings)
|
||||
|
||||
moves_manual = [
|
||||
"e2e4",
|
||||
"e7e6",
|
||||
"e4e5",
|
||||
"d7d5",
|
||||
"e5d6",
|
||||
"c7d6",
|
||||
"b1c3",
|
||||
"b8c6",
|
||||
"f1b5",
|
||||
"a7a6",
|
||||
"b5a4",
|
||||
"b7b5",
|
||||
"a4b3",
|
||||
"d6d5",
|
||||
"a2a4",
|
||||
"c8b7",
|
||||
"a4b5",
|
||||
"a6b5",
|
||||
"a1a8",
|
||||
"d8a8",
|
||||
"c3b5",
|
||||
"a8a5",
|
||||
"c2c4",
|
||||
"b7a6",
|
||||
"b5c3",
|
||||
"a6c4",
|
||||
"b3c4",
|
||||
"d5c4",
|
||||
"d1g4",
|
||||
"a5a1",
|
||||
]
|
||||
for move in moves_manual:
|
||||
print(move)
|
||||
test_game.make_move(move)
|
Loading…
Reference in a new issue