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