'''-------------------------------------------------------------------------------------------------- Das Visuelle Model steuert den Ablauf der aufrufe der Openvibe Funktionen fuer das visuelle BCI ''' from subprocess import * from threading import Thread import time import re from UIModell import Modell class ModellVisuell(Modell): def __init__(self,c, dll): Thread.__init__(self) Modell.__init__(self, c, dll) self.aktiv= True self.openVibeAktiv = False def stop(self): '''-------------------------------------------------------------------------------------------------- Die Funktion stop stopt den ablaufenden Thread ''' print("stop thread") self.aktiv = False def setFunktion(self, func, args=None, kwargs=None): '''-------------------------------------------------------------------------------------------------- Mittels dieser Funktion kann die im Thread ablaufende Funktion und deren Argumente eingestellt werden ''' self.func = func self.args = args or [] self.kwargs = kwargs or {} def run(self): '''-------------------------------------------------------------------------------------------------- Wird bei Thread.start() aus dem Controller aufgerufen und laeuft ab sobald der Thread destartet wurde ''' print("start thread") #self.aktiv = True t = Thread(target=self.func, args=self.args, kwargs=self.kwargs) t.setDaemon(True) t.start() while self.aktiv: time.sleep(0.1) def startCopySpelling(self): '''-------------------------------------------------------------------------------------------------- Holt aus der shared Librarz die Befehle und startet das Copyspelling. Reagiert auf Error oder beenden. Bei Beenden startet es das XDawnTraining, bei Error wird der Vorgang gestoppt. Positioniert die visuelle Matrix. ''' print("start copySpelling") self.controller.resetInfo() self.controller.addInfoText('Starten des gefuerhten Buchstabierens -- ') command = self.dll.getCommandCopyspellingVisuell().split(" ") process = Popen(command, stdout=PIPE, universal_newlines=True) self.openVibeAktiv = True while True: output = process.stdout.readline() print(output.strip()) x = output.find("Application terminated") y = output.find("Error") z = output.find("Initialization") if(x != -1): print("Training finished") process.terminate() self.controller.addInfoText('Buchstabieren beendet\n') self.controller.stop("filterXdawn") break elif(y != -1 ): print("Error occured") self.controller.changeScreen("StartPage") self.controller.addInfoText("Fehler beim Buchstabieren aufgetaucht\n") process.terminate() self.controller.stop() break elif(z != -1): print("Initiation finisched") time.sleep(2) self.positionWindow("p300") if not self.aktiv: print("stop") break #self.controller.stop() def trainXDawn(self): '''-------------------------------------------------------------------------------------------------- Holt aus der shared Library die Befehle und startet das XDawnTraining. Reagiert auf Error oder die Ausgabe das das Training beendet wurde. Bei Beenden startet es das ClassifierTraining, bei Error wird der Vorgang gestoppt. ''' print("start training Xdawn") self.controller.addInfoText('Starten des XDawn/Trainings -- ') command = self.dll.getCommandXDawn_visuell().split(" ") process = Popen(command, stdout=PIPE, universal_newlines=True) self.openVibeAktiv = True while self.aktiv: output = process.stdout.readline() print(output.strip()) x = output.find("Training finished and saved") y = output.find("Error") if(x != -1): print("Training finished") process.terminate() self.controller.addInfoText('Beenden des Trainings\n') self.controller.stop("filterClassic") break elif(y != -1 ): print("Error occured") self.controller.changeScreen("StartPage") self.controller.addInfoText("Fehler beim XDawn Training aufgetaucht\n") self.controller.stop() process.terminate() break if not self.aktiv: print("stop") break #self.controller.stop() def trainClassifier(self): '''-------------------------------------------------------------------------------------------------- Holt aus der shared Library die Befehle und startet das ClassifierTraining. Reagiert auf Error oder die Ausgabe das das Training beendet wurde. Bei Beenden startet es das speichern der Daten, bei Error wird der Vorgang gestoppt. ''' print("start training Classifier") self.controller.addInfoText("Starten Classifier-Trainings -- ") command = self.dll.getCommandClassifier_visuell().split(" ") process = Popen(command, stdout=PIPE, universal_newlines=True) self.openVibeAktiv = True counter = 0 acc = 0 while True: output = process.stdout.readline() print(output.strip()) x = output.find("aka Classifier trainer") accuracy = output.find("Cross-validation test accuracy is ") y = output.find("Error") if(x != -1): counter = counter +1 #counter = 18 if(counter >= 17): print("Training finished") self.controller.addInfoText('Beenden des Training (' + acc + ')\n') process.terminate() self.controller.stop("save") break elif(y != -1 ): print("Error occured") self.controller.changeScreen("StartPage") self.controller.addInfoText("Fehler beim Classifier Training aufgetaucht\n") process.terminate() break if(accuracy != -1): ansi_escape = re.compile(r'\x1B(?:[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])') test = output.encode("windows-1252").decode("utf-8") i = len("Cross-validation test") acc_s = test[accuracy+i:accuracy+i+41].strip() acc = ansi_escape.sub('', acc_s) print(acc) if not self.aktiv: print("stop") break self.controller.changeScreen("StartPage") self.controller.stop() def freeSpelling(self): '''-------------------------------------------------------------------------------------------------- Holt aus der shared Library die Befehle und startet das Freespelling. Reagiert auf Error oder beenden. Daraufhin wird der Vorgang gestoppt. Positioniert die visuelle Matrix. ''' print("start freeSpelling") self.controller.resetInfo() self.controller.addInfoText('Starten des freien Buchstabierens -- ') command = self.dll.getCommandFreespellingVisuell().split(" ") process = Popen(command, stdout=PIPE, universal_newlines=True) self.openVibeAktiv = True while True: output = process.stdout.readline() print(output.strip()) y = output.find("Error") x = output.find("Schlagwort") z = output.find("Initialization") if(x != -1): print("End Spelling") process.terminate() self.controller.addInfoText('Buchstabieren beendet\n') self.controller.changeScreen("StartPage") self.controller.stop() break elif(y != -1 ): print("Error occured") self.controller.changeScreen("StartPage") self.controller.addInfoText("Fehler beim Buchstabieren aufgetaucht\n") process.terminate() self.controller.stop() break elif(z != -1): print("Initiation finisched") time.sleep(2) self.positionWindow("p300") if not self.aktiv: print("stop") break self.controller.changeScreen("StartPage") self.controller.stop() def killProzess(self): '''-------------------------------------------------------------------------------------------------- Ruft den Parent auf, dass Openvibe falls es noch laeuft geschlossen wird ''' if(self.openVibeAktiv): self.openVibeAktiv = False self.killProzessParent()