def __init__(self, parent=None):
super(Uci_worker, self).__init__(parent)
self.command_queue = queue.Queue()
self.process = QProcess()
self.go_infinite = False
self.current_fen = ""
self.engine_info = EngineInfo()
def on_set_engines(self):
user_settings = self.model.user_settings
dialog = DialogEngines(user_settings=user_settings)
self.on_enter_moves_mode()
if dialog.exec_() == QDialog.Accepted:
user_settings.engines = dialog.engines
user_settings.active_engine = dialog.active_engine
info = EngineInfo()
info.id = user_settings.active_engine.name
self.mainAppWindow.gamestateController.receive_engine_info(info)
def on_set_engines(self):
user_settings = self.model.user_settings
dialog = DialogEngines(user_settings=user_settings)
self.on_enter_moves_mode()
if dialog.exec_() == QDialog.Accepted:
user_settings.engines = dialog.engines
user_settings.active_engine = dialog.active_engine
info = EngineInfo()
info.id = user_settings.active_engine.name
self.mainAppWindow.gamestateController.receive_engine_info(info)
#todo update views, i.e. label above engine window
def __init__(self,parent=None):
super(Uci_worker,self).__init__(parent)
self.command_queue = queue.Queue()
self.process = QProcess()
self.go_infinite = False
self.current_fen = ""
self.engine_info = EngineInfo()
def __init__(self):
super(GameState, self).__init__()
#self.game = chess.pgn.Game()
#self.root = self.game
#self.current = self.game
self.current = chess.pgn.Game()
self.mode = MODE_ENTER_MOVES
self.printer = GUIPrinter()
self.computer_think_time = 3000
self.display_engine_info = True
self.score = 0
self.position_bad = 0
self.position_draw = 0
self.pv = []
self.analysis_threshold = 0.5
self.engine_info = EngineInfo()
self.mate_threat = None
self.next_mate_threat = None
self.best_score = None
self.best_pv = []
self.timed_game = False
self.time_white = 0
self.time_black = 0
self.add_secs_per_move = 0
self.strength_level = 3
self.initialize_headers()
self.pgn_filename = None
self.last_save_dir = None
self.last_open_dir = None
self.unsaved_changes = False
self.game_analysis_white = False
self.game_analysis_black = False
class Uci_worker(QObject):
MOVES = re.compile('\s[a-z]\d[a-z]\d([a-z]{0,1})')
BESTMOVE = re.compile('bestmove\s[a-z]\d[a-z]\d[a-z]{0,1}')
STRENGTH = re.compile('Skill Level value \d+')
def __init__(self, parent=None):
super(Uci_worker, self).__init__(parent)
self.command_queue = queue.Queue()
self.process = QProcess()
self.go_infinite = False
self.current_fen = ""
self.engine_info = EngineInfo()
def process_command(self):
if (self.process.state() == QProcess.NotRunning
and not self.command_queue.empty()):
msg = self.command_queue.get()
if (msg.startswith("start_engine?")):
path = msg.split("?")[1]
self.process.start(path + "\n")
self.engine_info.strength = None
elif (self.process.state() == QProcess.Running):
output = str(self.process.readAllStandardOutput(), "utf-8")
if (not output == ""):
self.engine_info.update_from_string(output, self.current_fen)
self.emit(SIGNAL("info(PyQt_PyObject)"),
deepcopy(self.engine_info))
lines = output.splitlines()
for line in lines:
bm = self.BESTMOVE.search(line)
if (bm):
move = bm.group()[9:]
self.emit(SIGNAL("bestmove(QString)"), move)
if (not self.command_queue.empty()):
# first check if we are in go infinite mode
# then first send a stop command to engine
# before processing further commands
if (self.go_infinite):
self.process.write(bytes(("stop\n"), "utf-8"))
self.process.waitForBytesWritten()
msg = self.command_queue.get()
self.go_infinite = False
# if command is position fen moves, first count the
# numbers of moves so far to generate move numbers in engine info
if (msg.startswith("position")):
match = self.MOVES.findall(msg)
if (len(match) > 0):
self.engine_info.no_game_halfmoves = len(match)
if (msg.startswith("quit")):
self.process.write(bytes(("quit\n"), "utf-8"))
self.process.waitForBytesWritten()
self.process.waitForFinished()
elif (msg.startswith("go infinite")):
self.go_infinite = True
self.process.write(bytes(("go infinite\n"), "utf-8"))
self.process.waitForBytesWritten()
elif (msg.startswith("setoption name Skill Level")):
m = self.STRENGTH.search(msg)
if (m):
self.engine_info.strength = 1200 + int(
m.group()[18:]) * 100
self.process.write(bytes(msg + "\n", "utf-8"))
self.process.waitForBytesWritten()
else:
self.process.write(bytes(msg + "\n", "utf-8"))
self.process.waitForBytesWritten()
def add_command(self, command):
self.command_queue.put(command)
def update_fen(self, fen_string):
self.current_fen = fen_string
class Uci_worker(QObject):
MOVES = re.compile('\s[a-z]\d[a-z]\d([a-z]{0,1})')
BESTMOVE = re.compile('bestmove\s[a-z]\d[a-z]\d[a-z]{0,1}')
STRENGTH = re.compile('Skill Level value \d+')
def __init__(self,parent=None):
super(Uci_worker,self).__init__(parent)
self.command_queue = queue.Queue()
self.process = QProcess()
self.go_infinite = False
self.current_fen = ""
self.engine_info = EngineInfo()
def process_command(self):
if(self.process.state() == QProcess.NotRunning and not self.command_queue.empty()):
msg = self.command_queue.get()
if(msg.startswith("start_engine?")):
path = msg.split("?")[1]
self.process.start(path+"\n")
self.engine_info.strength = None
elif(self.process.state() == QProcess.Running):
output = str(self.process.readAllStandardOutput(),"utf-8")
self.engine_info.update_from_string(output,self.current_fen)
self.emit(SIGNAL("info(PyQt_PyObject)"),deepcopy(self.engine_info))
lines = output.splitlines()
for line in lines:
bm = self.BESTMOVE.search(line)
if(bm):
move = bm.group()[9:]
self.emit(SIGNAL("bestmove(QString)"),move)
if(not self.command_queue.empty()):
# first check if we are in go infinite mode
# then first send a stop command to engine
# before processing further commands
if(self.go_infinite):
self.process.write(bytes(("stop\n"),"utf-8"))
self.process.waitForBytesWritten()
msg = self.command_queue.get()
self.go_infinite = False
# if command is position fen moves, first count the
# numbers of moves so far to generate move numbers in engine info
if(msg.startswith("position")):
match = self.MOVES.findall(msg)
if(len(match) > 0):
self.engine_info.no_game_halfmoves = len(match)
if(msg.startswith("quit")):
self.process.write(bytes(("quit\n"),"utf-8"))
self.process.waitForBytesWritten()
self.process.waitForFinished()
elif(msg.startswith("go infinite")):
self.go_infinite = True
self.process.write(bytes(("go infinite\n"),"utf-8"))
self.process.waitForBytesWritten()
elif(msg.startswith("setoption name Skill Level")):
m = self.STRENGTH.search(msg)
if(m):
#print(str(m.group()[18:]))
self.engine_info.strength = 1200+int(m.group()[18:])*100
self.process.write(bytes(msg+"\n","utf-8"))
self.process.waitForBytesWritten()
else:
self.process.write(bytes(msg+"\n","utf-8"))
self.process.waitForBytesWritten()
def add_command(self,command):
self.command_queue.put(command)
def update_fen(self,fen_string):
self.current_fen = fen_string
class Uci_worker(QObject):
MOVES = re.compile('\s[a-z]\d[a-z]\d([a-z]{0,1})')
BESTMOVE = re.compile('bestmove\s[a-z]\d[a-z]\d[a-z]{0,1}')
STRENGTH = re.compile('Skill Level value \d+')
def __init__(self,parent=None):
super(Uci_worker,self).__init__(parent)
self.command_queue = queue.Queue()
self.process = QProcess()
self.go_infinite = False
self.current_fen = ""
self.engine_info = EngineInfo()
def process_command(self):
if(self.process.state() == QProcess.NotRunning and not self.command_queue.empty()):
msg = self.command_queue.get()
if(msg.startswith("start_engine?")):
print("engine start command received, starting engine")
path = msg.split("?")[1]
self.process.start(path+"\n")
self.engine_info.strength = None
print("process started")
elif(self.process.state() == QProcess.Running):
#time.sleep(0.001)
output = str(self.process.readAllStandardOutput(),"utf-8")
self.engine_info.update_from_string(output,self.current_fen)
self.emit(SIGNAL("info(PyQt_PyObject)"),deepcopy(self.engine_info))
lines = output.splitlines()
for line in lines:
bm = self.BESTMOVE.search(line)
if(bm):
move = bm.group()[9:]
self.emit(SIGNAL("bestmove(QString)"),move)
if(not self.command_queue.empty()):
print("processing queue")
# first check if we are in go infinite mode
# then first send a stop command to engine
# before processing further commands
if(self.go_infinite):
self.process.write(bytes(("stop\n"),"utf-8"))
self.process.waitForBytesWritten()
msg = self.command_queue.get()
self.go_infinite = False
# if command is position fen moves, first count the
# numbers of moves so far to generate move numbers in engine info
if(msg.startswith("position")):
# there could be a lot of position + go infinite commands in the queue
# if the user clicks very quickly through the game. then
# check for the most _recent_ go infinite command in the
# queue and forget about previous ones in order to get
# quicker responses in the GUI for the user
finish = False
while(not finish):
print("msg is: "+msg)
next_msg = None
next_next_msg = None
try:
next_msg = self.command_queue.get(timeout=0.01)
except queue.Empty:
finish = True
if not finish:
try:
next_next_msg = self.command_queue.get(timeout=0.01)
except queue.Empty:
finish = True
self.command_queue.put(next_msg)
if not finish:
if next_msg.startswith("go infinite") and next_next_msg.startswith("position"):
# just skip current position command and associated go infinite
# and go to the last go infinite command
print("prev skipping, now: "+next_next_msg)
msg = next_next_msg
else:
self.command_queue.put(next_next_msg)
self.command_queue.put(next_msg)
else:
finish = True
match = self.MOVES.findall(msg)
if(len(match) > 0):
self.engine_info.no_game_halfmoves = len(match)
if(msg.startswith("quit")):
self.process.write(bytes(("quit\n"),"utf-8"))
self.process.waitForBytesWritten()
self.process.waitForFinished()
elif(msg.startswith("go infinite")):
self.go_infinite = True
self.process.write(bytes(("go infinite\n"),"utf-8"))
self.process.waitForBytesWritten()
elif(msg.startswith("setoption name Skill Level")):
m = self.STRENGTH.search(msg)
if(m):
#print(str(m.group()[18:]))
self.engine_info.strength = 1200+int(m.group()[18:])*100
self.process.write(bytes(msg+"\n","utf-8"))
self.process.waitForBytesWritten()
else:
self.process.write(bytes(msg+"\n","utf-8"))
self.process.waitForBytesWritten()
def add_command(self,command):
self.command_queue.put(command)
def update_fen(self,fen_string):
self.current_fen = fen_string
