X-Git-Url: https://scm.cri.ensmp.fr/git/minwii.git/blobdiff_plain/4722346740569925bc0f88d4b5bba44df6b01a5d..46570d28d9b8eca74768bf3e6f43f18788e4d9a4:/src/minwii/logapp.py diff --git a/src/minwii/logapp.py b/src/minwii/logapp.py index b48140c..e03f451 100755 --- a/src/minwii/logapp.py +++ b/src/minwii/logapp.py @@ -6,11 +6,18 @@ $Id$ $URL$ """ +import os +os.environ['WINWII_NO_LOG'] = '1' 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) : @@ -19,6 +26,11 @@ class Application(Frame) : 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() @@ -32,42 +44,74 @@ class Application(Frame) : def createWidgets(self) : # zone d'affichage des données' self.dataFrame = df = Frame(self) - #df.grid(sticky=NW) self.identFrame = Identification(df) self.identFrame.grid(sticky=NW) - self.nav = Navbar(df, incCallback=self.loadLogFile, decCallback=self.loadLogFile) - self.nav.grid() - # barre de boutons self.btnFrame = bf = Frame(self) bf.grid(row=1, column=0, sticky=W+S+E) bf.rowconfigure(0, weight=1) - bf.columnconfigure(0, weight=1) - bf.columnconfigure(1, weight=1) - + for i in range(3) : + bf.columnconfigure(i, weight=1) - self.chooseLogDir = Button(bf, text="Parcourir…", command=self.openFileDialog) + 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=1, sticky=E) + self.quitButton.grid(row=0, column=2, sticky=E) - def openFileDialog(self) : - self.logDir = tkFileDialog.askdirectory() + 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.logFiles.sort() - self.dataFrame.grid(row=0, column=0, sticky=NW) - self.nav.setSize(len(self.logFiles)) - self.loadLogFile(self.nav) + 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] - filename = basename(filepath) - self.identFrame.setFileName(filename) + 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) : @@ -78,7 +122,6 @@ class Navbar(Frame) : self.setSize(size) self.incCallback = incCallback if incCallback else lambda x : None self.decCallback = decCallback if decCallback else lambda x : None - self.grid() self.caption.set('%d / %d' % (self.index, self.to)) def createWidgets(self) : @@ -131,10 +174,18 @@ class Identification(Frame) : def __init__(self, master=None) : Frame.__init__(self, master) self.fileName = StringVar() + self.hid = StringVar() + self.patientName = StringVar() self.createWidgets() - def setFileName(self, name) : - self.fileName.set(name) + 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) : fileLbl = Label(self, text='Fichier :') @@ -143,19 +194,79 @@ class Identification(Frame) : 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=1, column=0, sticky=E) + nameLbl.grid(row=2, column=0, sticky=E) - self.nameEntry = Entry(self, width=40) - self.nameEntry.grid(row=1, column=1, sticky=W) + 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=2, column=0, sticky=E) + commentsLbl.grid(row=3, column=0, sticky=E) self.commentsText = Text(self, width=40, height=4, undo=True, wrap=WORD) - self.commentsText.grid(row=2, column=1, sticky=W) + 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 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