65907054c1d5e236eb49025537481265a4b7cbca
[iramuteq] / tree.py
1 #!/bin/env python
2 # -*- coding: utf-8 -*-
3 #Author: Pierre Ratinaud
4 #Copyright (c) 2012, Pierre Ratinaud
5 #Lisense: GNU GPL
6
7 import wx
8 import os
9 import webbrowser
10 import wx.lib.agw.customtreectrl as CT
11 import logging
12 from openanalyse import OpenAnalyse
13 from corpus import Corpus, copycorpus
14 from functions import DoConf, GetTxtProfile
15 from profile_segment import ProfileSegment, ProfilType
16 from search_tools import SearchFrame
17 from dialog import PrefSimpleFile, PrefExport
18 from layout import open_antiprofil
19
20 log = logging.getLogger('iramuteq.tree')
21
22 class InfoDialog ( wx.Dialog ):
23     
24     def __init__( self, parent, txt, parametres ):
25         wx.Dialog.__init__ ( self, parent, id = wx.ID_ANY, title = u"Informations", pos = wx.DefaultPosition, size = wx.DefaultSize, style = wx.DEFAULT_DIALOG_STYLE )
26         if len(parametres) > 30 :
27             nb = 4
28         else :
29             nb = 2       
30         self.SetSizeHintsSz( wx.Size( 500,200 ), wx.DefaultSize )
31         
32         bSizer1 = wx.BoxSizer( wx.VERTICAL )
33         
34         self.m_panel2 = wx.Panel( self, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, wx.TAB_TRAVERSAL )
35         bSizer2 = wx.BoxSizer( wx.VERTICAL )
36         
37         self.m_staticText4 = wx.StaticText( self.m_panel2, wx.ID_ANY, txt, wx.DefaultPosition, wx.DefaultSize, 0 )
38         self.m_staticText4.Wrap( -1 )
39         bSizer2.Add( self.m_staticText4, 0, wx.ALL, 5 )
40         
41         
42         self.m_panel2.SetSizer( bSizer2 )
43         self.m_panel2.Layout()
44         bSizer2.Fit( self.m_panel2 )
45         bSizer1.Add( self.m_panel2, 0, wx.EXPAND |wx.ALL, 5 )
46         
47         self.m_panel1 = wx.Panel( self, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, wx.TAB_TRAVERSAL )
48
49         fgSizer1 = wx.FlexGridSizer( 0, nb, 0, 0 )
50         fgSizer1.SetFlexibleDirection( wx.BOTH )
51         fgSizer1.SetNonFlexibleGrowMode( wx.FLEX_GROWMODE_SPECIFIED )
52         
53         txtctrl = []
54         for val in parametres :
55             fgSizer1.Add( wx.StaticText( self.m_panel1, wx.ID_ANY, val[0], wx.DefaultPosition, wx.DefaultSize, 0 ), 0, wx.ALL, 0)
56             #fgSizer1.Add( wx.StaticText( self.m_panel1, wx.ID_ANY, val[1], wx.DefaultPosition, wx.DefaultSize, 0 ), 0, wx.ALL, 0)
57             txtctrl.append( wx.TextCtrl( self.m_panel1, wx.ID_ANY, val[1], wx.DefaultPosition, (450, 20), wx.TE_READONLY ) )
58             txtctrl[-1].SetBackgroundColour('#DDE8EB')
59             #wx.SystemSettings.GetColour(wx.SYS_COLOUR_GRAYTEXT))
60             fgSizer1.Add( txtctrl[-1], 0, wx.ALL|wx.EXPAND, 0)
61             #fgSizer1.Add( wx.StaticLine( self.m_panel1, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, wx.LI_HORIZONTAL ), 0, wx.EXPAND |wx.ALL, 0)
62             #fgSizer1.Add( wx.StaticLine( self.m_panel1, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, wx.LI_HORIZONTAL ), 0, wx.EXPAND|wx.ALL, 0)
63
64         self.m_panel1.SetSizer( fgSizer1 )
65         self.m_panel1.Layout()
66         fgSizer1.Fit( self.m_panel1 )
67         bSizer1.Add( self.m_panel1, 0, wx.EXPAND|wx.ALL, 3 )
68         
69         m_sdbSizer1 = wx.StdDialogButtonSizer()
70         self.m_sdbSizer1OK = wx.Button( self, wx.ID_OK )
71         m_sdbSizer1.AddButton( self.m_sdbSizer1OK )
72         m_sdbSizer1.Realize();
73         
74         bSizer1.Add( m_sdbSizer1, 0, wx.EXPAND, 5 )
75         
76         
77         self.SetSizer( bSizer1 )
78         self.Layout()
79         bSizer1.Fit( self )
80         
81         self.Centre( wx.BOTH )
82     
83     def __del__( self ):
84         pass
85
86
87 class LeftTree(CT.CustomTreeCtrl):
88
89     def __init__(self, parent, id=wx.ID_ANY, pos=wx.DefaultPosition,
90                  size=wx.DefaultSize,
91                  style=wx.SUNKEN_BORDER|wx.WANTS_CHARS,
92                  agwStyle=CT.TR_HAS_BUTTONS|CT.TR_HAS_VARIABLE_ROW_HEIGHT):
93
94         CT.CustomTreeCtrl.__init__(self, parent, id, pos, size, style, agwStyle)
95         self.log = log
96         alldata = dir(CT)
97         treestyles = []
98         events = []
99         for data in alldata:
100             if data.startswith("TR_"):
101                 treestyles.append(data)
102             elif data.startswith("EVT_"):
103                 events.append(data)
104         self.parent = parent
105         self.events = events
106         self.styles = treestyles
107         self.item = None
108         
109         il = wx.ImageList(16, 16)
110
111         self.count = 0
112         self.log = log
113
114         self.history = parent.history
115         self.h = self.history.history
116         self.root = self.AddRoot("Iramuteq")
117         
118         if not(self.GetAGWWindowStyleFlag() & CT.TR_HIDE_ROOT):
119             self.SetPyData(self.root, None)
120             self.SetItemImage(self.root, 24, CT.TreeItemIcon_Normal)
121             self.SetItemImage(self.root, 13, CT.TreeItemIcon_Expanded)
122
123         for corpus in self.h :
124             child = self.AppendItem(self.root, corpus['corpus_name'])
125             self.SetPyData(child, corpus)
126             self.SetItemImage(child, 24, CT.TreeItemIcon_Normal)
127             self.SetItemImage(child, 13, CT.TreeItemIcon_Expanded)
128
129             if 'analyses' in corpus :
130                 for y in corpus['analyses'] :
131                     last = self.AppendItem(child, y['name'], ct_type=0)
132                     self.SetPyData(last, y)
133                     self.SetItemImage(last, 24, CT.TreeItemIcon_Normal)
134                     self.SetItemImage(last, 13, CT.TreeItemIcon_Expanded)
135
136         for matrix in self.history.matrix :
137             last = self.AppendItem(self.root, matrix['name'])
138             self.SetPyData(last, matrix)
139             self.SetItemImage(last, 24, CT.TreeItemIcon_Normal)
140             self.SetItemImage(last, 13, CT.TreeItemIcon_Expanded)
141
142
143         self.Bind(wx.EVT_LEFT_DCLICK, self.OnLeftDClick)
144         #self.Bind(wx.EVT_IDLE, self.OnIdle)
145
146         self.eventdict = {'EVT_TREE_BEGIN_DRAG': self.OnBeginDrag, 'EVT_TREE_BEGIN_LABEL_EDIT': self.OnBeginEdit,
147                           'EVT_TREE_BEGIN_RDRAG': self.OnBeginRDrag, 'EVT_TREE_DELETE_ITEM': self.OnDeleteItem,
148                           'EVT_TREE_END_DRAG': self.OnEndDrag, 'EVT_TREE_END_LABEL_EDIT': self.OnEndEdit,
149                           'EVT_TREE_ITEM_ACTIVATED': self.OnActivate, 'EVT_TREE_ITEM_CHECKED': self.OnItemCheck,
150                           'EVT_TREE_ITEM_CHECKING': self.OnItemChecking, 'EVT_TREE_ITEM_COLLAPSED': self.OnItemCollapsed,
151                           'EVT_TREE_ITEM_COLLAPSING': self.OnItemCollapsing, 'EVT_TREE_ITEM_EXPANDED': self.OnItemExpanded,
152                           'EVT_TREE_ITEM_EXPANDING': self.OnItemExpanding, 'EVT_TREE_ITEM_GETTOOLTIP': self.OnToolTip,
153                           'EVT_TREE_ITEM_MENU': self.OnItemMenu, 'EVT_TREE_ITEM_RIGHT_CLICK': self.OnRightDown,
154                           'EVT_TREE_KEY_DOWN': self.OnKey, 'EVT_TREE_SEL_CHANGED': self.OnSelChanged,
155                           'EVT_TREE_SEL_CHANGING': self.OnSelChanging, "EVT_TREE_ITEM_HYPERLINK": self.OnHyperLink}
156
157         mainframe = wx.GetTopLevelParent(self)
158         
159         if not hasattr(mainframe, "leftpanel"):
160             #self.Bind(CT.EVT_TREE_ITEM_EXPANDED, self.OnItemExpanded)
161             #self.Bind(CT.EVT_TREE_ITEM_COLLAPSED, self.OnItemCollapsed)
162             self.Bind(CT.EVT_TREE_SEL_CHANGED, self.OnSelChanged)
163             self.Bind(CT.EVT_TREE_SEL_CHANGING, self.OnSelChanging)
164             self.Bind(wx.EVT_RIGHT_DOWN, self.OnRightDown)
165             self.Bind(wx.EVT_RIGHT_UP, self.OnRightUp)
166         else:
167             for combos in mainframe.treeevents:
168                 self.BindEvents(combos)
169
170         if hasattr(mainframe, "leftpanel"):
171             self.ChangeStyle(mainframe.treestyles)
172
173         if not(self.GetAGWWindowStyleFlag() & CT.TR_HIDE_ROOT):
174             self.SelectItem(self.root)
175             self.Expand(self.root)
176
177
178     def BindEvents(self, choice, recreate=False):
179
180         value = choice.GetValue()
181         text = choice.GetLabel()
182         
183         evt = "CT." + text
184         binder = self.eventdict[text]
185
186         if value == 1:
187             if evt == "CT.EVT_TREE_BEGIN_RDRAG":
188                 self.Bind(wx.EVT_RIGHT_DOWN, None)
189                 self.Bind(wx.EVT_RIGHT_UP, None)
190             self.Bind(eval(evt), binder)
191         else:
192             self.Bind(eval(evt), None)
193             if evt == "CT.EVT_TREE_BEGIN_RDRAG":
194                 self.Bind(wx.EVT_RIGHT_DOWN, self.OnRightDown)
195                 self.Bind(wx.EVT_RIGHT_UP, self.OnRightUp)
196
197
198     def ChangeStyle(self, combos):
199
200         style = 0
201         for combo in combos:
202             if combo.GetValue() == 1:
203                 style = style | eval("CT." + combo.GetLabel())
204
205         if self.GetAGWWindowStyleFlag() != style:
206             self.SetAGWWindowStyleFlag(style)
207             
208
209     def OnCompareItems(self, item1, item2):
210         
211         t1 = self.GetItemText(item1)
212         t2 = self.GetItemText(item2)
213         
214
215         if t1 < t2:
216             return -1
217         if t1 == t2:
218             return 0
219
220         return 1
221
222     
223     def OnIdle(self, event):
224
225     #    if self.gauge:
226     #        try:
227     #            if self.gauge.IsEnabled() and self.gauge.IsShown():
228     #                self.count = self.count + 1
229
230     #                if self.count >= 50:
231     #                    self.count = 0
232
233     #                self.gauge.SetValue(self.count)
234
235     #        except:
236     #            self.gauge = None
237
238         event.Skip()
239
240
241     def CloseItem(self, itemParent = None, uuid = None) :
242         if itemParent is None :
243             itemParent = self.root
244         child, cookie = self.GetFirstChild(itemParent)
245         while child :
246             pydata = self.GetPyData(child)
247             if pydata['uuid'] == uuid :
248                 self.SetItemBold(child, False)
249                 break
250             self.CloseItem(child, uuid)
251             child, cookie = self.GetNextChild(itemParent, cookie)
252
253     def GiveFocus(self, itemParent = None, uuid = None, bold = False) :
254         if itemParent is None :
255             itemParent = self.root
256         child, cookie = self.GetFirstChild(itemParent)
257         while child :
258             pydata = self.GetPyData(child)
259             if pydata['uuid'] == uuid :
260                 self.SelectItem(child)
261                 if bold :
262                     self.SetItemBold(child, True)
263                 return
264             self.GiveFocus(child, uuid, bold)
265             child, cookie = self.GetNextChild(itemParent, cookie)
266
267     def IsInTree(self, itemParent = None, uuid = None) :
268         if itemParent is None :
269             itemParent = self.root
270         child, cookie = self.GetFirstChild(itemParent)
271         while child :
272             pydata = self.GetPyData(child)
273             if pydata['uuid'] == uuid :
274                 return True
275             self.GiveFocus(child, uuid)
276             child, cookie = self.GetNextChild(itemParent, cookie)
277         return False
278
279
280     def OnRightDown(self, event):
281         
282         pt = event.GetPosition()
283         item, flags = self.HitTest(pt)
284
285         if item:
286             self.item = item
287             #self.log.info("OnRightClick: %s, %s, %s" % (self.GetItemText(item), type(item), item.__class__) + "\n")
288             self.SelectItem(item)
289
290
291     def OnRightUp(self, event):
292
293         item = self.item
294         
295         if not item:
296             event.Skip()
297             return
298
299         if not self.IsItemEnabled(item):
300             event.Skip()
301             return
302
303         # Item Text Appearance
304         ishtml = self.IsItemHyperText(item)
305         back = self.GetItemBackgroundColour(item)
306         fore = self.GetItemTextColour(item)
307         isbold = self.IsBold(item)
308         font = self.GetItemFont(item)
309
310         # Icons On Item
311         normal = self.GetItemImage(item, CT.TreeItemIcon_Normal)
312         selected = self.GetItemImage(item, CT.TreeItemIcon_Selected)
313         expanded = self.GetItemImage(item, CT.TreeItemIcon_Expanded)
314         selexp = self.GetItemImage(item, CT.TreeItemIcon_SelectedExpanded)
315
316         # Enabling/Disabling Windows Associated To An Item
317         haswin = self.GetItemWindow(item)
318
319         # Enabling/Disabling Items
320         enabled = self.IsItemEnabled(item)
321
322         # Generic Item's Info
323         children = self.GetChildrenCount(item)
324         itemtype = self.GetItemType(item)
325         text = self.GetItemText(item)
326         pydata = self.GetPyData(item)
327         self.pydata = pydata
328         
329         self.current = item
330         self.itemdict = {"ishtml": ishtml, "back": back, "fore": fore, "isbold": isbold,
331                          "font": font, "normal": normal, "selected": selected, "expanded": expanded,
332                          "selexp": selexp, "haswin": haswin, "children": children,
333                          "itemtype": itemtype, "text": text, "pydata": pydata, "enabled": enabled}
334         
335         menu = wx.Menu()
336         info = menu.Append(wx.ID_ANY, "Informations")
337         menu.AppendSeparator()
338
339         if 'corpus_name' in pydata :
340             stat = menu.Append(wx.ID_ANY, u"Statistiques")
341             spec = menu.Append(wx.ID_ANY, u"Spécificté et AFC")
342             classification = wx.Menu()
343             alceste = classification.Append(wx.ID_ANY, u"Méthode GNEPA")
344             #pam = classification.Append(wx.ID_ANY, u"Par matrice des distances")
345             menu.AppendMenu(-1, u"Classification", classification)
346             simi = menu.Append(wx.ID_ANY, u"Analyse de similitude")
347             wdc = menu.Append(wx.ID_ANY, u"Nuage de mots")
348             menu.AppendSeparator()
349             self.Bind(wx.EVT_MENU, self.OnAlceste, alceste)
350             #self.Bind(wx.EVT_MENU, self.OnPam, pam)
351             self.Bind(wx.EVT_MENU, self.OnStat, stat)
352             self.Bind(wx.EVT_MENU, self.OnSpec, spec)
353             self.Bind(wx.EVT_MENU, self.OnSimiTxt, simi)
354             self.Bind(wx.EVT_MENU, self.OnWordCloud, wdc)
355         elif pydata.get('type', False) == 'alceste' and pydata['uuid'] in self.parent.history.opened :
356             openmenu = wx.Menu()
357             antipro = openmenu.Append(wx.ID_ANY, u"antiprofils")
358             menu.AppendMenu(wx.ID_ANY, u"Ouvrir...", openmenu)
359
360             profsr = menu.Append(wx.ID_ANY, u"Profils des segments répétés")
361             profgram = menu.Append(wx.ID_ANY, u"Profils des types")
362             export_corpus = menu.Append(wx.ID_ANY, u"Exporter le corpus")
363             colored = menu.Append(wx.ID_ANY, u"Corpus en couleur")
364             navig = menu.Append(wx.ID_ANY, u"Outil de navigation")
365             statclasse = menu.Append(wx.ID_ANY, u"Statistiques par classe")
366             rapport = menu.Append(wx.ID_ANY, u"Rapport")
367             menu.AppendSeparator()
368             self.Bind(wx.EVT_MENU, self.OpenAntipro, antipro)
369             self.Bind(wx.EVT_MENU, self.OnProfSR, profsr)
370             self.Bind(wx.EVT_MENU, self.OnProfGram, profgram)
371             self.Bind(wx.EVT_MENU, self.OnExportCorpus, export_corpus)
372             self.Bind(wx.EVT_MENU, self.OnColored, colored)
373             self.Bind(wx.EVT_MENU, self.OnNavig, navig)
374             self.Bind(wx.EVT_MENU, self.StatClasse, statclasse)
375             self.Bind(wx.EVT_MENU, self.OnRapport, rapport)
376         elif pydata.get('type', False) == 'stat'  and pydata['uuid'] in self.parent.history.opened :
377             export_dictionary =  menu.Append(wx.ID_ANY, u"Exporter le dictionnaire")
378             export_lems =  menu.Append(wx.ID_ANY, u"Exporter le dictionnaire des lemmes")
379             self.Bind(wx.EVT_MENU, self.OnExportDictionary, export_dictionary)
380             self.Bind(wx.EVT_MENU, self.OnExportLems, export_lems)
381             menu.AppendSeparator()
382         elif pydata.get('type', False) == 'gnepamatrix' and pydata['uuid'] in self.parent.history.opened :
383             openmenu = wx.Menu()
384             antipro = openmenu.Append(wx.ID_ANY, u"antiprofils")
385             menu.AppendMenu(wx.ID_ANY, u"Ouvrir...", openmenu)
386             self.Bind(wx.EVT_MENU, self.OpenAntipro, antipro)
387
388
389         itemdelete = menu.Append(wx.ID_ANY, "Supprimer de l'historique")
390         if item == self.GetRootItem():
391             itemdelete.Enable(False)
392         #item11 = menu.Append(wx.ID_ANY, "Prepend An Item")
393         #item12 = menu.Append(wx.ID_ANY, "Append An Item")
394
395         #self.Bind(wx.EVT_MENU, self.OnItemBackground, item1)
396         #self.Bind(wx.EVT_MENU, self.OnItemForeground, item2)
397         #self.Bind(wx.EVT_MENU, self.OnItemBold, item3)
398         #self.Bind(wx.EVT_MENU, self.OnItemFont, item4)
399         #self.Bind(wx.EVT_MENU, self.OnItemHyperText, item5)
400         #self.Bind(wx.EVT_MENU, self.OnEnableWindow, item6)
401         #self.Bind(wx.EVT_MENU, self.OnDisableItem, item7)
402         #self.Bind(wx.EVT_MENU, self.OnItemIcons, item8)
403         self.Bind(wx.EVT_MENU, self.OnItemInfo, info)
404         self.Bind(wx.EVT_MENU, self.OnItemDelete, itemdelete)
405         #self.Bind(wx.EVT_MENU, self.OnItemPrepend, item11)
406         #self.Bind(wx.EVT_MENU, self.OnItemAppend, item12)
407         
408         self.PopupMenu(menu)
409         menu.Destroy()
410
411     def getcorpus(self):
412         if self.pydata['uuid'] in self.parent.history.openedcorpus :
413             return copycorpus(self.parent.history.openedcorpus[self.pydata['uuid']])
414         elif 'corpus_name' in self.pydata :
415             return Corpus(self.parent, parametres = DoConf(self.pydata['ira']).getoptions('corpus'), read = True)
416         else :
417             cuuid = self.pydata['corpus']
418             if cuuid in self.parent.history.openedcorpus :
419                 return copycorpus(self.parent.history.openedcorpus[cuuid])
420             else :
421                 irapath = self.parent.history.corpus[cuuid]['ira']
422                 return Corpus(self.parent, parametres = DoConf(irapath).getoptions('corpus'), read = True)
423
424     def OnSpec(self, evt) :
425         self.parent.OnTextSpec(evt, self.getcorpus())
426
427     def OnStat(self, evt) :
428         self.parent.OnTextStat(evt, self.getcorpus())
429         
430     def OnAlceste(self, evt) :
431         self.parent.OnTextAlceste(evt, self.getcorpus())
432
433     def OnPam(self, evt) :
434         self.parent.OnPamSimple(evt, self.getcorpus())
435
436     def OnSimiTxt(self, evt) :
437         self.parent.OnSimiTxt(evt, self.getcorpus())
438
439     def OnWordCloud(self, evt) :
440         self.parent.OnWordCloud(evt, self.getcorpus())
441
442     def OnProfSR(self, evt) :
443         ProfileSegment(self.parent, self.page.dictpathout, self.page.parametres, self.page.corpus)
444
445     def OnProfGram(self, evt) :
446         ProfilType(self.parent, self.page.corpus, self.page.parametres)
447
448     def OnExportCorpus(self, evt) :
449         dial = PrefExport(self, self.parent)
450         dial.fbb.SetValue(os.path.join(os.path.dirname(self.page.dictpathout['ira']), 'export_corpus.txt'))
451         dial.CenterOnParent()
452         res = dial.ShowModal()
453         if res == wx.ID_OK :
454             if dial.radio_type.GetSelection() == 0 : alc = True
455             else : alc = False
456             if dial.radio_lem.GetSelection() == 0 : lem = True
457             else : lem = False
458             self.page.corpus.export_corpus_classes(dial.fbb.GetValue(), alc = alc, lem = lem)
459             msg = u"Fini !"
460             dial.Destroy()
461             dlg = wx.MessageDialog(self.parent, msg, u"Export", wx.OK | wx.NO_DEFAULT | wx.ICON_INFORMATION)
462             dlg.CenterOnParent()
463             dlg.ShowModal()
464             dlg.Destroy()
465
466     def OnColored(self, evt) :
467         dial = PrefSimpleFile(self, self.parent, **{'mask' : '*.html', 'title': 'Corpus en couleur'})
468         dial.fbb.SetValue(os.path.join(os.path.dirname(self.page.dictpathout['ira']), 'corpus_couleur.html'))
469         dial.CenterOnParent()
470         res = dial.ShowModal()
471         if res == wx.ID_OK :
472             fileout = dial.fbb.GetValue()
473             dial.Destroy()
474             txt = self.page.corpus.make_colored_corpus()
475             with open(fileout, 'w') as f :
476                 f.write(txt)
477             msg = u"Fini !\nVoulez-vous ouvrir le corpus dans votre navigateur ?"
478             dlg = wx.MessageDialog(self.parent, msg, u"Corpus en couleur", wx.NO | wx.YES | wx.NO_DEFAULT | wx.ICON_QUESTION)
479             dlg.CenterOnParent()
480             if dlg.ShowModal() == wx.ID_YES :
481                 webbrowser.open(fileout)
482             dlg.Destroy()
483
484     def OnNavig(self, evt):
485         if 'FrameSearch' not in dir(self.page) :
486             self.page.FrameSearch = SearchFrame(self.parent, -1, u"Rechercher...", self.page.corpus)
487         self.page.FrameSearch.Show()
488
489     def StatClasse(self, evt):
490         dial = PrefSimpleFile(self, self.parent, **{'mask' : '*.csv', 'title': 'Stat par classe'})
491         dial.fbb.SetValue( os.path.join(os.path.dirname(self.page.dictpathout['ira']), 'stat_par_classe.csv'))
492         dial.CenterOnParent()
493         res = dial.ShowModal()
494         if res == wx.ID_OK :
495             fileout = dial.fbb.GetValue()
496             dial.Destroy()
497             self.page.corpus.get_stat_by_cluster(fileout)
498             msg = u"Fini !"
499             dlg = wx.MessageDialog(self.parent, msg, u"Stat par classe", wx.OK | wx.NO_DEFAULT | wx.ICON_INFORMATION)
500             dlg.CenterOnParent()
501             if dlg.ShowModal() == wx.ID_OK :
502                 dlg.Destroy()        
503
504     def OpenAntipro(self, evt) :
505         find = False
506         for i in range(0, self.page.TabChdSim.GetPageCount()) :
507             page = self.page.TabChdSim.GetPage(i)
508             if self.page.TabChdSim.GetPageText(i) == 'Antiprofils' :
509                 self.page.TabChdSim.SetSelection(i)
510                 find = True
511                 break
512         if not find :
513             open_antiprofil(self.page, self.page.dictpathout['ANTIPRO_OUT'], self.parent.syscoding)
514             self.page.TabChdSim.SetSelection(self.page.TabChdSim.GetPageCount() - 1)
515
516     def OnRapport(self, evt) :
517         dial = PrefSimpleFile(self, self.parent, **{'mask' : '*.txt', 'title': 'Rapport'})
518         dial.fbb.SetValue(self.page.dictpathout['rapport'])
519         dial.CenterOnParent()
520         res = dial.ShowModal()
521         if res == wx.ID_OK :
522             fileout = dial.fbb.GetValue()
523             dial.Destroy()
524             with open(fileout, 'w') as f :
525                 f.write(self.page.debtext + '\n' + GetTxtProfile(self.page.DictProfile, self.page.cluster_size))
526             msg = u"Fini !"
527             dlg = wx.MessageDialog(self.parent, msg, u"Rapport", wx.OK | wx.NO_DEFAULT | wx.ICON_INFORMATION)
528             dlg.CenterOnParent()
529             dlg.ShowModal()
530             dlg.Destroy()
531         else :
532             dial.Destroy()
533
534     def OnExportDictionary(self, evt) :
535         corpus = self.page.corpus
536         corpus.export_dictionary(self.page.pathout['dictionary.csv'], self.parent.syscoding)
537         log.info('export dictionary %s' % self.page.pathout['dictionary.csv'])
538         dial = wx.MessageDialog(self.parent, self.page.pathout['dictionary.csv'], 'Export', wx.OK)
539         dial.ShowModal()
540         dial.Destroy()
541         
542     def OnExportLems(self, evt) :
543         corpus = self.page.corpus
544         corpus.export_lems(self.page.pathout['lemmes.csv'], self.parent.syscoding)
545         log.info('export lemmes %s' % self.page.pathout['lemmes.csv'])
546         dial = wx.MessageDialog(self.parent, self.page.pathout['lemmes.csv'], 'Export', wx.OK)
547         dial.ShowModal()
548         dial.Destroy()
549
550     def OnItemBackground(self, event):
551
552         colourdata = wx.ColourData()
553         colourdata.SetColour(self.itemdict["back"])
554         dlg = wx.ColourDialog(self, colourdata)
555         
556         dlg.GetColourData().SetChooseFull(True)
557
558         if dlg.ShowModal() == wx.ID_OK:
559             data = dlg.GetColourData()
560             col1 = data.GetColour().Get()
561             self.SetItemBackgroundColour(self.current, col1)
562         dlg.Destroy()
563
564
565     def OnItemForeground(self, event):
566
567         colourdata = wx.ColourData()
568         colourdata.SetColour(self.itemdict["fore"])
569         dlg = wx.ColourDialog(self, colourdata)
570         
571         dlg.GetColourData().SetChooseFull(True)
572
573         if dlg.ShowModal() == wx.ID_OK:
574             data = dlg.GetColourData()
575             col1 = data.GetColour().Get()
576             self.SetItemTextColour(self.current, col1)
577         dlg.Destroy()
578
579
580     def OnItemBold(self, event):
581
582         self.SetItemBold(self.current, not self.itemdict["isbold"])
583
584
585     def OnItemFont(self, event):
586
587         data = wx.FontData()
588         font = self.itemdict["font"]
589         
590         if font is None:
591             font = wx.SystemSettings_GetFont(wx.SYS_DEFAULT_GUI_FONT)
592             
593         data.SetInitialFont(font)
594
595         dlg = wx.FontDialog(self, data)
596         
597         if dlg.ShowModal() == wx.ID_OK:
598             data = dlg.GetFontData()
599             font = data.GetChosenFont()
600             self.SetItemFont(self.current, font)
601
602         dlg.Destroy()
603         
604
605     def OnItemHyperText(self, event):
606
607         self.SetItemHyperText(self.current, not self.itemdict["ishtml"])
608
609
610     def OnEnableWindow(self, event):
611
612         enable = self.GetItemWindowEnabled(self.current)
613         self.SetItemWindowEnabled(self.current, not enable)
614
615
616     def OnDisableItem(self, event):
617
618         self.EnableItem(self.current, False)
619         
620
621     def OnItemIcons(self, event):
622
623         bitmaps = [self.itemdict["normal"], self.itemdict["selected"],
624                    self.itemdict["expanded"], self.itemdict["selexp"]]
625
626         wx.BeginBusyCursor()        
627         dlg = TreeIcons(self, -1, bitmaps=bitmaps)
628         wx.EndBusyCursor()
629         dlg.ShowModal()
630
631
632     def SetNewIcons(self, bitmaps):
633
634         self.SetItemImage(self.current, bitmaps[0], CT.TreeItemIcon_Normal)
635         self.SetItemImage(self.current, bitmaps[1], CT.TreeItemIcon_Selected)
636         self.SetItemImage(self.current, bitmaps[2], CT.TreeItemIcon_Expanded)
637         self.SetItemImage(self.current, bitmaps[3], CT.TreeItemIcon_SelectedExpanded)
638
639
640     def OnItemInfo(self, event):
641
642         itemtext = self.itemdict["text"]
643         numchildren = str(self.itemdict["children"])
644         itemtype = self.itemdict["itemtype"]
645         pydata = self.itemdict['pydata']
646         #if 'analyses' in pydata :
647         #    toshow = dict([[val, pydata[val]] for val in pydata if val not in['analyses', 'isload']])
648         #else :
649         toshow = pydata['ira']
650         toshow = DoConf(toshow).getoptions()
651         txt = DoConf().totext(toshow)
652         parametres = [val.split('\t\t:') for val in txt.splitlines()]
653         parametres.sort()
654
655         if itemtype == 0:
656             itemtype = "Normal"
657         elif itemtype == 1:
658             itemtype = "CheckBox"
659         else:
660             itemtype = "RadioButton"
661
662         dlg = InfoDialog(self, itemtext, parametres)
663         dlg.CenterOnParent()
664         dlg.ShowModal()
665         dlg.Destroy()
666                 
667         
668
669     def OnItemDelete(self, event):
670
671         strs = "Are You Sure You Want To Delete Item " + self.GetItemText(self.current) + "?"
672         dlg = wx.MessageDialog(None, strs, 'Deleting Item', wx.OK | wx.CANCEL | wx.ICON_QUESTION)
673
674         if dlg.ShowModal() in [wx.ID_NO, wx.ID_CANCEL]:
675             dlg.Destroy()
676             return
677
678         dlg.Destroy()
679         
680         pydata = self.itemdict['pydata']
681         if 'corpus_name' in pydata :
682             self.history.delete(pydata, True)
683         else :
684             self.history.delete(pydata)
685         self.DeleteChildren(self.current)
686         self.Delete(self.current)
687         self.current = None
688         
689
690
691     def OnItemPrepend(self, event):
692
693         dlg = wx.TextEntryDialog(self, "Please Enter The New Item Name", 'Item Naming', 'Python')
694
695         if dlg.ShowModal() == wx.ID_OK:
696             newname = dlg.GetValue()
697             newitem = self.PrependItem(self.current, newname)
698             self.EnsureVisible(newitem)
699
700         dlg.Destroy()
701
702     def AddAnalyse(self, parametres, itemParent = None, bold = True) :
703         uuid = parametres.get('corpus', None)
704         if uuid is not None :
705             if itemParent is None :
706                 itemParent = self.root
707             child, cookie = self.GetFirstChild(itemParent)
708             corpus = None
709             while child :
710                 pydata = self.GetPyData(child)
711                 if pydata['uuid'] == uuid :
712                     corpus = child
713                     break
714                 self.GiveFocus(child, uuid)
715                 child, cookie = self.GetNextChild(itemParent, cookie)
716             #item = self.AppendItem(child, parametres['name'])
717             if corpus is not None : 
718                 item = self.AppendItem(corpus, parametres['name'])
719             else :
720                 item = self.AppendItem(self.root, parametres['name'])
721         else :
722             item = self.AppendItem(self.root, parametres['name'])
723         self.SetPyData(item, parametres)
724         self.SetItemImage(item, 24, CT.TreeItemIcon_Normal)
725         self.SetItemImage(item, 13, CT.TreeItemIcon_Expanded)
726         self.SetItemBold(item, bold)
727         
728     def OnItemAppend(self, item):
729         child = self.AppendItem(self.root, item['corpus_name'])
730         self.SetPyData(child, item)
731         self.history.addtab(item)
732         self.SetItemImage(child, 24, CT.TreeItemIcon_Normal)
733         self.SetItemImage(child, 13, CT.TreeItemIcon_Expanded)
734         self.SetItemBold(child, True)
735         
736         #dlg = wx.TextEntryDialog(self, "Please Enter The New Item Name", 'Item Naming', 'Python')
737
738         #if dlg.ShowModal() == wx.ID_OK:
739         #    newname = dlg.GetValue()
740         #    newitem = self.AppendItem(self.current, newname)
741         #    self.EnsureVisible(newitem)
742         
743
744         #dlg.Destroy()
745         
746
747     def OnBeginEdit(self, event):
748         
749         #self.log.info("OnBeginEdit" + "\n")
750         # show how to prevent edit...
751         item = event.GetItem()
752         if item and self.GetItemText(item) == "The Root Item":
753             wx.Bell()
754             #self.log.info("You can't edit this one..." + "\n")
755
756             # Lets just see what's visible of its children
757             cookie = 0
758             root = event.GetItem()
759             (child, cookie) = self.GetFirstChild(root)
760
761             while child:
762                 #self.log.info("Child [%s] visible = %d" % (self.GetItemText(child), self.IsVisible(child)) + "\n")
763                 (child, cookie) = self.GetNextChild(root, cookie)
764
765             event.Veto()
766
767
768     def OnEndEdit(self, event):
769         
770         #self.log.info("OnEndEdit: %s %s" %(event.IsEditCancelled(), event.GetLabel()))
771         # show how to reject edit, we'll not allow any digits
772         for x in event.GetLabel():
773             if x in string.digits:
774                 #self.log.info(", You can't enter digits..." + "\n")
775                 event.Veto()
776                 return
777             
778         self.log.info("\n")
779
780
781     def OnLeftDClick(self, event):
782         
783         pt = event.GetPosition()
784         item, flags = self.HitTest(pt)
785         if item is not None :
786             pydata = self.GetPyData(item)
787             if pydata['uuid'] in self.parent.history.opened :
788                 for i in range(self.parent.nb.GetPageCount()) :
789                     page = self.parent.nb.GetPage(i)
790                     if 'parametres' in dir(page) :
791                         if page.parametres['uuid'] == pydata['uuid'] :
792                             self.parent.nb.SetSelection(i)
793                             break
794             else :
795                 OpenAnalyse(self.parent, pydata)
796                 self.SetItemBold(item, True)
797                 self.OnSelChanged(pydata = pydata)
798         #if item and (flags & CT.TREE_HITTEST_ONITEMLABEL):
799         #    if self.GetAGWWindowStyleFlag() & CT.TR_EDIT_LABELS:
800         #        self.log.info("OnLeftDClick: %s (manually starting label edit)"% self.GetItemText(item) + "\n")
801                 
802                 #self.EditLabel(item)
803         #    else:
804         #        pydata = self.GetPyData(item)
805         #        print pydata
806         #        self.log.info("OnLeftDClick: Cannot Start Manual Editing, Missing Style TR_EDIT_LABELS\n")
807
808         event.Skip()                
809         
810
811     def OnItemExpanded(self, event):
812         
813         item = event.GetItem()
814         if item:
815             self.log.info("OnItemExpanded: %s" % self.GetItemText(item) + "\n")
816
817
818     def OnItemExpanding(self, event):
819         
820         item = event.GetItem()
821         if item:
822             self.log.info("OnItemExpanding: %s" % self.GetItemText(item) + "\n")
823             
824         event.Skip()
825
826         
827     def OnItemCollapsed(self, event):
828
829         item = event.GetItem()
830         if item:
831             self.log.info("OnItemCollapsed: %s" % self.GetItemText(item) + "\n")
832             
833
834     def OnItemCollapsing(self, event):
835
836         item = event.GetItem()
837         if item:
838             self.log.info("OnItemCollapsing: %s" % self.GetItemText(item) + "\n")
839     
840         event.Skip()
841
842         
843     def OnSelChanged(self, event = None, pydata = None):
844         if event is not None :
845             item = event.GetItem()
846             pydata = self.GetPyData(item)
847         if pydata is not None :
848             self.pydata = pydata
849             if pydata['uuid'] in self.parent.history.opened :
850                 for i in range(self.parent.nb.GetPageCount()) :
851                     self.page = self.parent.nb.GetPage(i)
852                     if 'parametres' in dir(self.page) :
853                         if self.page.parametres['uuid'] == pydata['uuid'] :
854                             self.parent.nb.SetSelection(i)
855                             break
856         if event is not None :
857             event.Skip()
858
859
860     def OnSelChanging(self, event):
861
862         item = event.GetItem()
863         olditem = event.GetOldItem()
864         
865         if item:
866             if not olditem:
867                 olditemtext = "None"
868             else:
869                 olditemtext = self.GetItemText(olditem)
870             #self.log.info("OnSelChanging: From %s" % olditemtext + " To %s" % self.GetItemText(item) + "\n")
871                 
872         event.Skip()
873
874
875     def OnBeginDrag(self, event):
876
877         self.item = event.GetItem()
878         if self.item:
879             self.log.info("Beginning Drag..." + "\n")
880
881             event.Allow()
882
883
884     def OnBeginRDrag(self, event):
885
886         self.item = event.GetItem()
887         if self.item:
888             self.log.info("Beginning Right Drag..." + "\n")
889
890             event.Allow()
891         
892
893     def OnEndDrag(self, event):
894
895         self.item = event.GetItem()
896         if self.item:
897             self.log.info("Ending Drag!" + "\n")
898
899         event.Skip()            
900
901
902     def OnDeleteItem(self, event):
903
904         item = event.GetItem()
905
906         if not item:
907             return
908
909         self.log.info("Deleting Item: %s" % self.GetItemText(item) + "\n")
910         event.Skip()
911         
912
913     def OnItemCheck(self, event):
914
915         item = event.GetItem()
916         self.log.info("Item " + self.GetItemText(item) + " Has Been Checked!\n")
917         event.Skip()
918
919
920     def OnItemChecking(self, event):
921
922         item = event.GetItem()
923         self.log.info("Item " + self.GetItemText(item) + " Is Being Checked...\n")
924         event.Skip()
925         
926
927     def OnToolTip(self, event):
928
929         item = event.GetItem()
930         if item:
931             event.SetToolTip(wx.ToolTip(self.GetItemText(item)))
932
933
934     def OnItemMenu(self, event):
935
936         item = event.GetItem()
937         if item:
938             self.log.info("OnItemMenu: %s" % self.GetItemText(item) + "\n")
939     
940         event.Skip()
941
942
943     def OnKey(self, event):
944
945         keycode = event.GetKeyCode()
946         keyname = keyMap.get(keycode, None)
947                 
948         if keycode == wx.WXK_BACK:
949             self.log.info("OnKeyDown: HAHAHAHA! I Vetoed Your Backspace! HAHAHAHA\n")
950             return
951
952         if keyname is None:
953             if "unicode" in wx.PlatformInfo:
954                 keycode = event.GetUnicodeKey()
955                 if keycode <= 127:
956                     keycode = event.GetKeyCode()
957                 keyname = "\"" + unichr(event.GetUnicodeKey()) + "\""
958                 if keycode < 27:
959                     keyname = "Ctrl-%s" % chr(ord('A') + keycode-1)
960                 
961             elif keycode < 256:
962                 if keycode == 0:
963                     keyname = "NUL"
964                 elif keycode < 27:
965                     keyname = "Ctrl-%s" % chr(ord('A') + keycode-1)
966                 else:
967                     keyname = "\"%s\"" % chr(keycode)
968             else:
969                 keyname = "unknown (%s)" % keycode
970                 
971         self.log.info("OnKeyDown: You Pressed '" + keyname + "'\n")
972
973         event.Skip()
974         
975         
976     def OnActivate(self, event):
977         
978         if self.item:
979             self.log.info("OnActivate: %s" % self.GetItemText(self.item) + "\n")
980
981         event.Skip()
982
983         
984     def OnHyperLink(self, event):
985
986         item = event.GetItem()
987         if item:
988             self.log.info("OnHyperLink: %s" % self.GetItemText(self.item) + "\n")
989             
990
991     def OnTextCtrl(self, event):
992
993         char = chr(event.GetKeyCode())
994         self.log.info("EDITING THE TEXTCTRL: You Wrote '" + char + \
995                        "' (KeyCode = " + str(event.GetKeyCode()) + ")\n")
996         event.Skip()
997
998
999     def OnComboBox(self, event):
1000
1001         selection = event.GetEventObject().GetValue()
1002         self.log.info("CHOICE FROM COMBOBOX: You Chose '" + selection + "'\n")
1003         event.Skip()