bugfix
[minwii.git] / src / minwii / widgets / songfilebrowser.py
1 # -*- coding: utf-8 -*-
2 """
3 Boîte de dialogue pour sélection des chansons.
4
5 $Id$
6 $URL$
7 """
8
9 import pygame
10 from pygame.locals import K_RETURN
11 from pgu.gui import FileDialog
12 import pgu.gui.basic as basic
13 import pgu.gui.input as input
14 import pgu.gui.button as button
15 import pgu.gui.pguglobals as pguglobals
16 import pgu.gui.table as table
17 import pgu.gui.area as area
18 from pgu.gui.const import *
19 from pgu.gui.dialog import Dialog
20
21 import os
22 import tempfile
23 from xml.etree import ElementTree
24 from minwii.musicxml import musicXml2Song
25
26 INDEX_TXT = 'index.txt'
27
28 class FileOpenDialog(FileDialog):
29
30
31
32 def __init__(self, path):
33 cls1 = 'filedialog'
34 if not path: self.curdir = os.getcwd()
35 else: self.curdir = path
36 self.dir_img = basic.Image(
37 pguglobals.app.theme.get(cls1+'.folder', '', 'image'))
38 td_style = {'padding_left': 4,
39 'padding_right': 4,
40 'padding_top': 2,
41 'padding_bottom': 2}
42 self.title = basic.Label("Ouvrir un chanson", cls="dialog.title.label")
43 self.body = table.Table()
44 self.list = area.List(width=700, height=250)
45 self.input_dir = input.Input()
46 self.input_file = input.Input()
47 self._current_sort = 'alpha'
48 self._list_dir_()
49 self.button_ok = button.Button("Ouvrir")
50 self.button_sort_alpha = button.Button("A-Z")
51 self.button_sort_alpha.connect(CLICK, self._set_current_sort_, 'alpha')
52 self.button_sort_num = button.Button("0-9")
53 self.button_sort_num.connect(CLICK, self._set_current_sort_, 'num')
54 self.body.tr()
55 self.body.td(basic.Label("Dossier"), style=td_style, align=-1)
56 self.body.td(self.input_dir, style=td_style)
57 self.body.td(self.button_sort_alpha)
58 self.body.td(self.button_sort_num)
59 self.body.tr()
60 self.body.td(self.list, colspan=4, style=td_style)
61 self.list.connect(CHANGE, self._item_select_changed_, None)
62 #self.list.connect(CLICK, self._check_dbl_click_, None)
63 self._last_time_click = pygame.time.get_ticks()
64 self.button_ok.connect(CLICK, self._button_okay_clicked_, None)
65 self.body.tr()
66 self.body.td(basic.Label("Fichier"), style=td_style, align=-1)
67 self.body.td(self.input_file, style=td_style)
68 self.body.td(self.button_ok, style=td_style, colspan=2)
69 self.value = None
70 Dialog.__init__(self, self.title, self.body)
71
72 # FileDialog.__init__(self,
73 # title_txt="Ouvrir une chanson",
74 # button_txt="Ouvrir",
75 # path=path,
76 # )
77 # self.list.style.width = 700
78 # self.list.style.height = 250
79
80 def _list_dir_(self):
81 self.input_dir.value = self.curdir
82 self.input_dir.pos = len(self.curdir)
83 self.input_dir.vpos = 0
84 dirs = []
85 files = []
86 try:
87 for i in os.listdir(self.curdir):
88 if i.startswith('.') : continue
89 if os.path.isdir(os.path.join(self.curdir, i)): dirs.append(i)
90 else: files.append(i)
91 except:
92 self.input_file.value = "Dossier innacessible !"
93
94 dirs.sort()
95 dirs.insert(0, '..')
96
97 files.sort()
98 for i in dirs:
99 self.list.add(i, image=self.dir_img, value=i)
100
101 xmlFiles = []
102 for i in files:
103 if not i.endswith('.xml') :
104 continue
105 filepath = os.path.join(self.curdir, i)
106 xmlFiles.append(filepath)
107 # self.list.add(FileOpenDialog.getSongTitle(filepath), value=i)
108
109 if xmlFiles :
110 printableLines = self.getPrintableLines(xmlFiles)
111 for l in printableLines :
112 self.list.add(l[0], value = l[1])
113
114 self.list.set_vertical_scroll(0)
115
116 def getPrintableLines(self, xmlFiles) :
117 index = self.getUpdatedIndex(xmlFiles)
118
119 printableLines = []
120 for l in index :
121 l = l.strip()
122 l = l.split('\t')
123 printableLines.append(('%s - %s / %s' % (l[2], l[3], l[4]), l[0]))
124
125 return printableLines
126
127
128 @staticmethod
129 def getSongTitle(file) :
130 it = ElementTree.iterparse(file, ['start', 'end'])
131 creditFound = False
132 title = os.path.basename(file)
133
134 for evt, el in it :
135 if el.tag == 'credit' :
136 creditFound = True
137 if el.tag == 'credit-words' and creditFound:
138 title = el.text
139 break
140 if el.tag == 'part-list' :
141 # au delà de ce tag : aucune chance de trouver un titre
142 break
143 return title
144
145 @staticmethod
146 def getSongMetadata(file) :
147 metadata = {}
148 metadata['title'] = FileOpenDialog.getSongTitle(file).encode('iso-8859-1')
149 metadata['mtime'] = str(os.stat(file).st_mtime)
150 metadata['file'] = os.path.basename(file)
151 song = musicXml2Song(file)
152 metadata['distinctNotes'] = len(song.distinctNotes)
153
154 histo = song.intervalsHistogram
155 coeffInter = reduce(lambda a, b : a + b,
156 [abs(k) * v for k, v in histo.items()])
157
158 totInter = reduce(lambda a, b: a+b, histo.values())
159 totInter = totInter - histo.get(0, 0)
160 difficulty = int(round(float(coeffInter) / totInter, 0))
161 metadata['difficulty'] = difficulty
162
163 return metadata
164
165 def getUpdatedIndex(self, xmlFiles) :
166 indexTxtPath = os.path.join(self.curdir, INDEX_TXT)
167 index = []
168
169 if not os.path.exists(indexTxtPath) :
170 musicXmlFound = False
171 tmp = tempfile.TemporaryFile(mode='r+')
172 for file in xmlFiles :
173 try :
174 metadata = FileOpenDialog.getSongMetadata(file)
175 musicXmlFound = True
176 except ValueError, e :
177 print e
178 if e.args and e.args[0] == 'not a musicxml file' :
179 continue
180
181 line = '%(file)s\t%(mtime)s\t%(title)s\t%(distinctNotes)d\t%(difficulty)d\n' % metadata
182 index.append(line)
183 tmp.write(line)
184
185 if musicXmlFound :
186 tmp.seek(0)
187 indexFile = open(indexTxtPath, 'w')
188 indexFile.write(tmp.read())
189 indexFile.close()
190 tmp.close()
191 else :
192 indexedFiles = {}
193 indexTxt = open(indexTxtPath, 'r')
194
195 # check if index is up to date, and update entries if so.
196 for l in filter(None, indexTxt.readlines()) :
197 parts = l.split('\t')
198 fileBaseName, modificationTime = parts[0], parts[1]
199 filePath = os.path.join(self.curdir, fileBaseName)
200
201 if not os.path.exists(filePath) :
202 continue
203
204 indexedFiles[fileBaseName] = l
205 currentMtime = str(os.stat(filePath).st_mtime)
206
207 # check modification time missmatch
208 if currentMtime != modificationTime :
209 try :
210 metadata = FileOpenDialog.getSongMetadata(filePath)
211 musicXmlFound = True
212 except ValueError, e :
213 print e
214 if e.args and e.args[0] == 'not a musicxml file' :
215 continue
216
217 metadata = FileOpenDialog.getSongMetadata(filePath)
218 line = '%(file)s\t%(mtime)s\t%(title)s\t%(distinctNotes)d\t%(difficulty)d\n' % metadata
219 indexedFiles[fileBaseName] = line
220
221 # check for new files.
222 for file in xmlFiles :
223 fileBaseName = os.path.basename(file)
224 if not indexedFiles.has_key(fileBaseName) :
225 try :
226 metadata = FileOpenDialog.getSongMetadata(filePath)
227 musicXmlFound = True
228 except ValueError, e :
229 print e
230 if e.args and e.args[0] == 'not a musicxml file' :
231 continue
232
233 metadata = FileOpenDialog.getSongMetadata(file)
234 line = '%(file)s\t%(mtime)s\t%(title)s\t%(distinctNotes)d\t%(difficulty)d\n' % metadata
235 indexedFiles[fileBaseName] = line
236
237 # ok, the index is up to date !
238
239 index = indexedFiles.values()
240
241
242 if self._current_sort == 'alpha' :
243 def s(a, b) :
244 da = desacc(a.split('\t')[2]).lower()
245 db = desacc(b.split('\t')[2]).lower()
246 return cmp(da, db)
247
248 elif self._current_sort == 'num' :
249 def s(a, b) :
250 da = int(a.split('\t')[3])
251 db = int(b.split('\t')[3])
252 return cmp(da, db)
253 else :
254 s = cmp
255
256 index.sort(s)
257 return index
258
259 def _set_current_sort_(self, arg) :
260 self._current_sort = arg
261 self.list.clear()
262 self._list_dir_()
263
264 def _check_dbl_click_(self, arg) :
265 if pygame.time.get_ticks() - self._last_time_click < 300 :
266 self._button_okay_clicked_(None)
267 else :
268 self._last_time_click = pygame.time.get_ticks()
269
270 def event(self, e) :
271 FileDialog.event(self, e)
272
273 if e.type == CLICK and \
274 e.button == 1 and \
275 self.list.rect.collidepoint(e.pos) :
276 self._check_dbl_click_(e)
277
278 if e.type == KEYDOWN and e.key == K_RETURN :
279 self._button_okay_clicked_(None)
280
281
282 # utils
283 from unicodedata import decomposition
284 from string import printable
285 _printable = dict([(c, True) for c in printable])
286 isPrintable = _printable.has_key
287
288 def _recurseDecomposition(uc):
289 deco = decomposition(uc).split()
290 fullDeco = []
291 if deco :
292 while (deco) :
293 code = deco.pop()
294 if code.startswith('<') :
295 continue
296 c = unichr(int(code, 16))
297 subDeco = decomposition(c).split()
298 if subDeco :
299 deco.extend(subDeco)
300 else :
301 fullDeco.append(c)
302 fullDeco.reverse()
303 else :
304 fullDeco.append(uc)
305
306 fullDeco = u''.join(filter(lambda c : isPrintable(c), fullDeco))
307 return fullDeco
308
309 def desacc(s, encoding='iso-8859-1') :
310 us = s.decode(encoding, 'ignore')
311 ret = []
312 for uc in us :
313 ret.append(_recurseDecomposition(uc))
314 return u''.join(ret)