BCIgui/UIModellTaktil.py

'''--------------------------------------------------------------------------------------------------
Das taktile Model steuert den Ablauf der aufrufe der Openvibe Funktionen fuer das taktile BCI
noch nicht implementiert da noch kein Programm dafuer
'''

from subprocess import *
from threading import Thread
import time
from UIModell import *
import re

class ModellTaktil(Modell):

    def __init__(self,c, dll):
        Thread.__init__(self)
        Modell.__init__(self, c, dll)
        #self.dll = 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.
        '''
        print("start copySpelling")
        self.controller.resetInfo()
        self.controller.addInfoText('Starten des gefuerten Buchstabierens -- ')

        command = self.dll.getCommandCopyspellingTaktil().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")
            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
            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_taktil() #.split(" ")
        print("Command: " + command)
        command = command.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('Training beendet\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()

        #self.controller.filterClassic()


    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 des Classifier-Trainings -- ")

        command = self.dll.getCommandClassifier_taktil().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('Training beendet (' + 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):
                #print("Output:")
                #print(output)
                ansi_escape = re.compile(r'\x1B(?:[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])')
                test = output.encode("windows-1252").decode("utf-8")
                i = len("Cross-validation test")
                #print(output[accuracy:accuracy+i+12])
                #print(test[accuracy+i+12:accuracy+i+40].strip())
                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.
        '''
        print("start freeSpelling")
        self.controller.resetInfo()
        self.controller.addInfoText('Starten des freien Buchstabierens -- ')
        command = self.dll.getCommandFreespellingTaktil().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("Application terminated")
            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
            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()