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