Affichage du titre de la chanson.

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

diff --git a/src/minwii/loganalyse.py b/src/minwii/loganalyse.py
index 1a1f676..ab8c819 100755 (executable)
--- a/src/minwii/loganalyse.py
+++ b/src/minwii/loganalyse.py
@@ -12,12 +12,14 @@ from minwii.musicxml import musicXml2Song
 from minwii.globals import PLAYING_MODES
 from statlib import stats
 from datetime import timedelta
+from xml.etree import ElementTree
+import os.path
 
 PLAYING_MODES = dict(PLAYING_MODES)
 
 DEFAULT_STATS = (#'geometricmean',
-                 #'harmonicmean',
-                 #'mean',
+                 ('harmonicmean', 'Moyenne harmonique'),
+                 ('mean', 'Moyenne  '),
                  ('median', 'Médiane'),
                  #'medianscore',
                  #'mode',
@@ -73,6 +75,10 @@ class LogFileAnalyser(LogFileReader) :
         try :
             self.mode = mode = self.getMode()
             results.append(('Mode de jeu', PLAYING_MODES.get(mode, mode)))
+
+            self.songTitle = LogFileAnalyser.getSongTitle(self.getSongFile())
+            results.append(('Chanson', self.songTitle))
+
             for name in self.POSSIBLE_ANALYSES[mode] :
                 meth = getattr(self, name)
                 results.append((meth.__doc__, meth()))
@@ -81,6 +87,23 @@ class LogFileAnalyser(LogFileReader) :
         
         return results
     
+    @staticmethod
+    def getSongTitle(file) :
+        if os.path.exists(file) :
+            it = ElementTree.iterparse(file, ['start', 'end'])
+            creditFound = False
+
+            for evt, el in it :
+                if el.tag == 'credit' :
+                    creditFound = True
+                if el.tag == 'credit-words' and creditFound:
+                    return el.text
+                if el.tag == 'part-list' :
+                    # plus de chance de trouver un titre
+                    return os.path.basename(file)
+        else :
+            return os.path.basename(file)
+
     def _toTimeDelta(self, milliseconds) :
         duration = milliseconds / 1000.
         duration = int(round(duration, 0))