Étirement de la vidéo de la kinect.

[minwii.git] / src / minwii / logapp.py

diff --git a/src/minwii/logapp.py b/src/minwii/logapp.py
index 1bcd703..e03f451 100755 (executable)
--- a/src/minwii/logapp.py
+++ b/src/minwii/logapp.py
@@ -6,24 +6,267 @@ $Id$
 $URL$
 """
 
 $URL$
 """
 

+import os
+os.environ['WINWII_NO_LOG'] = '1'

 from Tkinter import *
 import tkFileDialog
 from Tkinter import *
 import tkFileDialog

+from glob import glob
+from os.path import join as pjoin
+from os.path import basename
+from os.path import getsize
+import os.path
+from minwii.loganalyse import LogFileAnalyser
+from minwii.config import LOGS_DIR
+from pprint import pprint

 
 class Application(Frame) :
     def __init__(self, master=None) :
         Frame.__init__(self, master)
 
 class Application(Frame) :
     def __init__(self, master=None) :
         Frame.__init__(self, master)

-        self.grid()
+        self.configureStretching()

         self.createWidgets()
         self.createWidgets()

+        self.logDir = ''
+        self.logFiles = []
+        self.currentFilePath = ''
+        self.resultsFrame = None
+        
+        if os.path.exists(LOGS_DIR) :
+            self.chooseDirDialog(dir=LOGS_DIR)
+    
+    def configureStretching(self) :
+        top=self.winfo_toplevel()
+        top.rowconfigure(0, weight=1)
+        top.columnconfigure(0, weight=1)
+        
+        self.grid(sticky=N+S+E+W, padx=10, pady=10)
+        self.rowconfigure(0, weight=1)
+        self.columnconfigure(0, weight=1)
+    
+    def createWidgets(self) :
+        # zone d'affichage des données'
+        self.dataFrame = df = Frame(self)
+        
+        self.identFrame = Identification(df)
+        self.identFrame.grid(sticky=NW)
+        
+        # barre de boutons
+        self.btnFrame = bf = Frame(self)
+        bf.grid(row=1, column=0, sticky=W+S+E)
+        bf.rowconfigure(0, weight=1)
+        for i in range(3) :
+            bf.columnconfigure(i, weight=1)        
+
+        self.chooseLogDir = Button(bf, text="Parcourir…", command=self.chooseDirDialog)
+        self.chooseLogDir.grid(row=0, column=0, sticky=W)
+        
+        self.nav = Navbar(bf, incCallback=self.loadLogFile, decCallback=self.loadLogFile)
+
+        self.quitButton = Button(bf, text='Terminer', command=self.quit)
+        self.quitButton.grid(row=0, column=2, sticky=E)
+    
+    def chooseDirDialog(self, dir=None) :
+        if dir is None :
+            self.logDir = tkFileDialog.askdirectory(title='Sélectionnez un dossier de fichiers de logs')
+        else :
+            self.logDir = dir
+        if self.logDir :
+            self.logFiles = glob(pjoin(self.logDir, '*.log'))
+            self._cleanupJunkFiles()
+            self.logFiles.sort()
+            self.logFiles.reverse()
+            self.dataFrame.grid(row=0, column=0, sticky=NW)
+            self.nav.setSize(len(self.logFiles))
+            self.nav.grid(row=0, column=1)
+            self.loadLogFile(self.nav)
+    
+    def _cleanupJunkFiles(self) :
+        files = []
+        while self.logFiles :
+            f = self.logFiles.pop()
+            if not getsize(f) :
+                os.remove(f)
+                continue
+            else :
+                of = open(f)
+                lfa = LogFileAnalyser(of)
+                if lfa.getLastEventTicks() is None :
+                    of.close()
+                    os.remove(f)
+                    continue
+                else :
+                    of.close()
+                
+                files.append(f)
+        
+        self.logFiles = files
+        
+    
+    def loadLogFile(self, nav) :
+        index = nav.index - 1
+        filepath = self.logFiles[index]
+        self.currentFilePath = filepath
+        lfa = LogFileAnalyser(self.currentFilePath)
+        self.identFrame.refresh(lfa)
+        if self.resultsFrame :
+            self.resultsFrame.destroy()
+        self.resultsFrame = ResultsFrame(self.dataFrame)
+        self.resultsFrame.layResults(lfa)
+        lfa.close()
+        self.resultsFrame.grid()
+
+
+class Navbar(Frame) :
+    def __init__(self, master=None, size=1, incCallback=None, decCallback=None) :
+        Frame.__init__(self, master)
+        self.caption = StringVar()
+        self.createWidgets()
+        self.setSize(size)
+        self.incCallback = incCallback if incCallback else lambda x : None
+        self.decCallback = decCallback if decCallback else lambda x : None
+        self.caption.set('%d / %d' % (self.index, self.to))
+    
+    def createWidgets(self) :
+        self.backBtn = Button(self,
+                              text='◀',
+                              command = self.dec
+                              )
+        self.backBtn.grid(row=0, column=0)
+        
+        self.lbl = Label(self, textvariable=self.caption)
+        self.lbl.grid(row=0, column=1)
+
+        self.nextBtn = Button(self,
+                              text='▶',
+                              command = self.inc)
+        self.nextBtn.grid(row=0, column=2)
+    
+    def refreshStates(self) :
+        if self.index == self.from_ :
+            self.backBtn.configure(state=DISABLED)
+        else :
+            self.backBtn.configure(state=NORMAL)
+
+        if self.index < self.to :
+            self.nextBtn.configure(state=NORMAL)
+        else :
+            self.nextBtn.configure(state=DISABLED)
+
+        self.caption.set('%d / %d' % (self.index, self.to))
+        
+    
+    def dec(self) :
+        self.index = self.index - 1
+        self.refreshStates()
+        self.decCallback(self)
+    
+    def inc(self) :
+        self.index = self.index + 1
+        self.refreshStates()
+        self.incCallback(self)
+
+    def setSize(self, size) :
+        self.from_ = 1
+        self.to = size
+        self.index = 1
+        self.refreshStates()
+
+
+class Identification(Frame) :
+    def __init__(self, master=None) :
+        Frame.__init__(self, master)
+        self.fileName = StringVar()
+        self.hid = StringVar()
+        self.patientName = StringVar()
+        self.createWidgets()
+    
+    def refresh(self, lfa) :
+        filename = basename(lfa.logfile.name)
+        self.fileName.set(filename)
+        self.hid.set(lfa.getHID())
+        metadata = lfa.getMetadata()
+        self.patientName.set(metadata.get('PatientName', ''))
+        self.commentsText.delete(1.0, END)
+        self.commentsText.insert(1.0, metadata.get('Comments', ''))

     
     def createWidgets(self) :
     
     def createWidgets(self) :

-        self.chooseLogDir = Button(self, text="Parcourir…", command=self.openFileDialog)
-        self.chooseLogDir.grid()
-        self.quitButton = Button(self, text='Terminer', command=self.quit)
-        self.quitButton.grid()
+        fileLbl = Label(self, text='Fichier :')
+        fileLbl.grid(row=0, column=0, sticky=E)
+
+        fileNameLbl = Label(self, textvariable=self.fileName)
+        fileNameLbl.grid(row=0, column=1, sticky=W)
+        
+        hidLbl = Label(self, text='HID :')
+        hidLbl.grid(row=1, column=0, sticky=E)
+        
+        hidNameLbl = Label(self, textvariable=self.hid)
+        hidNameLbl.grid(row=1, column=1, sticky=W)
+        
+        nameLbl = Label(self, text='Patient :')
+        nameLbl.grid(row=2, column=0, sticky=E)
+        
+        self.nameEntry = Entry(self, width=40, textvariable=self.patientName)
+        self.nameEntry.grid(row=2, column=1, sticky=W)
+        
+        commentsLbl = Label(self, text='Commentaires :')
+        commentsLbl.grid(row=3, column=0, sticky=E)
+        
+        self.commentsText = Text(self, width=40, height=4, undo=True, wrap=WORD)
+        self.commentsText.grid(row=3, column=1, sticky=W)
+        
+        self.saveBtn = Button(self, text='Enregistrer', command=self.saveMetadata)
+        self.saveBtn.grid(row=4, column=1, sticky=E)

     
     

-    def openFileDialog(self) :
-        print tkFileDialog.askopenfilename()
+    def saveMetadata(self):
+        app = self.master.master
+        filepath = app.currentFilePath
+        lfa = LogFileAnalyser(filepath, mode='r+')
+        patientName = '%s\n' % self.nameEntry.get().replace('\n', ' ').strip()
+        comments = '%s\n' % self.commentsText.get(1.0, END).replace('\n', ' ').strip()
+        metadata = (('PatientName', self.nameEntry.get()),
+                    ('Comments', comments))
+        lfa.setMetadata(metadata)
+
+
+class ResultsFrame(Frame) :
+    
+    def layResults(self, lfa) :
+        results = lfa.analyse()
+        if results :
+            for i, kvt in enumerate(results) :
+                k, v, timeBased = kvt
+                kl = Label(self, text='%s :' % k)
+                kl.grid(row=i, column=0, sticky=E)
+
+                if not timeBased :
+                    vl = Label(self, text=v)
+                    vl.grid(row=i, column=1, sticky=W)
+                else :
+                    maxv = max(v)
+                    if maxv :
+                        cw, ch = 200, 100
+                        c = Canvas(self, background='#fff', width=cw, height=ch)
+                        rectW = int(float(cw) / len(v))
+                        unitRectH = float(ch) / maxv
+                        for j, fv in enumerate(v) :
+                            if not fv : continue
+                            x0 = j * rectW
+                            y0 = ch - int(unitRectH * fv)
+                            x1 = (j + 1) * rectW
+                            y1 = ch
+                            c.create_rectangle(x0, y0, x1, y1, fill="#9085ba")
+                        c.grid(row=i, column=1, sticky=W)
+                            
+                    else :
+                        vl = Label(self, text='—')
+                        vl.grid(row=i, column=1, sticky=W)
+        else :
+            msg = Label(self, text="Pas de données exploitables.")
+            msg.grid()
+            
+        
+def main() :
+    app = Application()
+    app.master.title("Analyseur des sessions MINDs")
+    app.mainloop()

 
 

-app = Application()
-app.master.title("Analyseur des sessions MINWii")
-app.mainloop()
+if __name__ == '__main__' :
+    main()
\ No newline at end of file