1 # -*- coding: utf-8 -*-
2 #Author: Pierre Ratinaud
3 #Copyright (c) 2012, Pierre Ratinaud
9 import wx.lib.agw.customtreectrl as CT
11 from openanalyse import OpenAnalyse
12 from corpus import Corpus, copycorpus
13 from tableau import Tableau, copymatrix
14 from functions import DoConf, GetTxtProfile, TGen, BugReport, open_folder
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, TgenLayout
19 from guifunct import TGenFrame
20 from textaslexico import TgenSpec
21 from textreinert import TgenProf
23 log = logging.getLogger('iramuteq.tree')
25 def buildmenu(menu, parent_menu):
26 for i in range(parent_menu.GetMenuItemCount()) :
27 item = parent_menu.FindItemByPosition(i)
29 itemtext = item.GetText()
30 itemicon = item.GetBitmap()
31 nitem = wx.MenuItem(menu, itemid, itemtext)
32 nitem.SetBitmap(itemicon)
35 for val in item.GetSubMenu().GetMenuItems() :
37 itemtext = val.GetText()
38 itemicon = val.GetBitmap()
39 nitem = wx.MenuItem(menu, itemid, itemtext)
40 nitem.SetBitmap(itemicon)
41 nmenu.AppendItem(nitem)
42 menu.AppendMenu(-1, item.GetText(), nmenu)
44 menu.AppendItem(nitem)
46 class InfoDialog ( wx.Dialog ):
48 def __init__( self, parent, txt, parametres ):
49 wx.Dialog.__init__ ( self, parent, id = wx.ID_ANY, title = u"Informations", pos = wx.DefaultPosition, size = wx.DefaultSize, style = wx.DEFAULT_DIALOG_STYLE )
50 if len(parametres) > 30 :
54 self.SetSizeHintsSz( wx.Size( 500,200 ), wx.DefaultSize )
56 bSizer1 = wx.BoxSizer( wx.VERTICAL )
58 self.m_panel2 = wx.Panel( self, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, wx.TAB_TRAVERSAL )
59 bSizer2 = wx.BoxSizer( wx.VERTICAL )
61 self.m_staticText4 = wx.StaticText( self.m_panel2, wx.ID_ANY, txt, wx.DefaultPosition, wx.DefaultSize, 0 )
62 self.m_staticText4.Wrap( -1 )
63 bSizer2.Add( self.m_staticText4, 0, wx.ALL, 5 )
66 self.m_panel2.SetSizer( bSizer2 )
67 self.m_panel2.Layout()
68 bSizer2.Fit( self.m_panel2 )
69 bSizer1.Add( self.m_panel2, 0, wx.EXPAND |wx.ALL, 5 )
71 self.m_panel1 = wx.Panel( self, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, wx.TAB_TRAVERSAL )
73 fgSizer1 = wx.FlexGridSizer( 0, nb, 0, 0 )
74 fgSizer1.SetFlexibleDirection( wx.BOTH )
75 fgSizer1.SetNonFlexibleGrowMode( wx.FLEX_GROWMODE_SPECIFIED )
78 for val in parametres :
79 fgSizer1.Add( wx.StaticText( self.m_panel1, wx.ID_ANY, val[0], wx.DefaultPosition, wx.DefaultSize, 0 ), 0, wx.ALL, 0)
80 #fgSizer1.Add( wx.StaticText( self.m_panel1, wx.ID_ANY, val[1], wx.DefaultPosition, wx.DefaultSize, 0 ), 0, wx.ALL, 0)
81 txtctrl.append( wx.TextCtrl( self.m_panel1, wx.ID_ANY, val[1], wx.DefaultPosition, (450, 20), wx.TE_READONLY ) )
82 txtctrl[-1].SetBackgroundColour('#DDE8EB')
83 #wx.SystemSettings.GetColour(wx.SYS_COLOUR_GRAYTEXT))
84 fgSizer1.Add( txtctrl[-1], 0, wx.ALL|wx.EXPAND, 0)
85 #fgSizer1.Add( wx.StaticLine( self.m_panel1, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, wx.LI_HORIZONTAL ), 0, wx.EXPAND |wx.ALL, 0)
86 #fgSizer1.Add( wx.StaticLine( self.m_panel1, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, wx.LI_HORIZONTAL ), 0, wx.EXPAND|wx.ALL, 0)
88 self.m_panel1.SetSizer( fgSizer1 )
89 self.m_panel1.Layout()
90 fgSizer1.Fit( self.m_panel1 )
91 bSizer1.Add( self.m_panel1, 0, wx.EXPAND|wx.ALL, 3 )
93 m_sdbSizer1 = wx.StdDialogButtonSizer()
94 self.m_sdbSizer1OK = wx.Button( self, wx.ID_OK )
95 m_sdbSizer1.AddButton( self.m_sdbSizer1OK )
96 m_sdbSizer1.Realize();
98 bSizer1.Add( m_sdbSizer1, 0, wx.EXPAND, 5 )
101 self.SetSizer( bSizer1 )
105 self.Centre( wx.BOTH )
111 class LeftTree(CT.CustomTreeCtrl):
113 def __init__(self, parent, id=wx.ID_ANY, pos=wx.DefaultPosition,
115 style=wx.SUNKEN_BORDER|wx.WANTS_CHARS,
116 agwStyle=CT.TR_HIDE_ROOT|CT.TR_HAS_BUTTONS|CT.TR_HAS_VARIABLE_ROW_HEIGHT):
118 CT.CustomTreeCtrl.__init__(self, parent, id, pos, size, style, agwStyle)
124 if data.startswith("TR_"):
125 treestyles.append(data)
126 elif data.startswith("EVT_"):
131 self.styles = treestyles
135 self.il = wx.ImageList(16, 16)
137 for img in self.ira.images_analyses :
138 self.ild[img] = self.il.Add(self.ira.images_analyses[img])
139 self.SetImageList(self.il)
144 self.history = parent.history
145 self.h = self.history.history
147 idopenfolder = wx.NewId()
148 accel_tbl = wx.AcceleratorTable([(wx.ACCEL_CTRL, ord('E'), idopenfolder)])
149 self.SetAcceleratorTable(accel_tbl)
151 self.Bind(wx.EVT_MENU, self.OnOpenFolder, id=idopenfolder)
153 self.root = self.AddRoot("Iramuteq")
155 if not(self.GetAGWWindowStyleFlag() & CT.TR_HIDE_ROOT):
156 self.SetPyData(self.root, None)
157 self.SetItemImage(self.root, 24, CT.TreeItemIcon_Normal)
158 self.SetItemImage(self.root, 13, CT.TreeItemIcon_Expanded)
160 self.textroot = self.AppendItem(self.root, _(u'Textual corpus'))
161 self.SetPyData(self.textroot, {'uuid': 'textroot'})
162 self.SetItemImage(self.textroot, self.ild['textroot'], CT.TreeItemIcon_Normal)
163 self.SetItemImage(self.textroot, self.ild['textroot'], CT.TreeItemIcon_Expanded)
165 for corpus in reversed(self.h) :
166 child = self.AppendItem(self.textroot, corpus['corpus_name'])
167 self.SetPyData(child, corpus)
168 self.SetItemImage(child, self.ild['corpus'], CT.TreeItemIcon_Normal)
169 self.SetItemImage(child, self.ild['corpus'], CT.TreeItemIcon_Expanded)
171 if 'analyses' in corpus :
172 for y in corpus['analyses'] :
173 last = self.AppendItem(child, y['name'], ct_type=0)
174 self.SetPyData(last, y)
175 if y['type'] in self.ild :
176 img = self.ild[y['type']]
179 self.SetItemImage(last, img, CT.TreeItemIcon_Normal)
180 self.SetItemImage(last, img, CT.TreeItemIcon_Expanded)
182 self.matroot = self.AppendItem(self.root, _(u'Matrix'))
183 self.SetPyData(self.matroot, {'uuid': 'matroot'})
184 self.SetItemImage(self.matroot, self.ild['matroot'], CT.TreeItemIcon_Normal)
185 self.SetItemImage(self.matroot, self.ild['matroot'], CT.TreeItemIcon_Expanded)
188 for matrix in reversed(self.history.matrix) :
189 if 'matrix_name' in matrix :
190 child = self.AppendItem(self.matroot, matrix['matrix_name'])
191 self.SetPyData(child, matrix)
192 self.SetItemImage(child, self.ild['matrix'], CT.TreeItemIcon_Normal)
193 self.SetItemImage(child, self.ild['matrix'], CT.TreeItemIcon_Expanded)
194 if 'analyses' in matrix :
195 for y in matrix['analyses'] :
196 last = self.AppendItem(child, y['name'], ct_type=0)
197 self.SetPyData(last, y)
198 if y['type'] in self.ild :
199 img = self.ild[y['type']]
202 self.SetItemImage(last, img, CT.TreeItemIcon_Normal)
203 self.SetItemImage(last, img, CT.TreeItemIcon_Expanded)
205 orphmat.append(matrix)
207 self.Bind(wx.EVT_LEFT_DCLICK, self.OnLeftDClick)
208 #self.Bind(wx.EVT_IDLE, self.OnIdle)
210 self.eventdict = {'EVT_TREE_BEGIN_DRAG': self.OnBeginDrag, 'EVT_TREE_BEGIN_LABEL_EDIT': self.OnBeginEdit,
211 'EVT_TREE_BEGIN_RDRAG': self.OnBeginRDrag, 'EVT_TREE_DELETE_ITEM': self.OnDeleteItem,
212 'EVT_TREE_END_DRAG': self.OnEndDrag, 'EVT_TREE_END_LABEL_EDIT': self.OnEndEdit,
213 'EVT_TREE_ITEM_ACTIVATED': self.OnActivate, 'EVT_TREE_ITEM_CHECKED': self.OnItemCheck,
214 'EVT_TREE_ITEM_CHECKING': self.OnItemChecking, 'EVT_TREE_ITEM_COLLAPSED': self.OnItemCollapsed,
215 'EVT_TREE_ITEM_COLLAPSING': self.OnItemCollapsing, 'EVT_TREE_ITEM_EXPANDED': self.OnItemExpanded,
216 'EVT_TREE_ITEM_EXPANDING': self.OnItemExpanding, 'EVT_TREE_ITEM_GETTOOLTIP': self.OnToolTip,
217 'EVT_TREE_ITEM_MENU': self.OnItemMenu, 'EVT_TREE_ITEM_RIGHT_CLICK': self.OnRightDown,
218 'EVT_TREE_KEY_DOWN': self.OnKey, 'EVT_TREE_SEL_CHANGED': self.OnSelChanged,
219 'EVT_TREE_SEL_CHANGING': self.OnSelChanging, "EVT_TREE_ITEM_HYPERLINK": self.OnHyperLink}
221 mainframe = wx.GetTopLevelParent(self)
223 if not hasattr(mainframe, "leftpanel"):
224 #self.Bind(CT.EVT_TREE_ITEM_EXPANDED, self.OnItemExpanded)
225 #self.Bind(CT.EVT_TREE_ITEM_COLLAPSED, self.OnItemCollapsed)
226 self.Bind(CT.EVT_TREE_SEL_CHANGED, self.OnSelChanged)
227 self.Bind(CT.EVT_TREE_SEL_CHANGING, self.OnSelChanging)
228 self.Bind(wx.EVT_RIGHT_DOWN, self.OnRightDown)
229 self.Bind(wx.EVT_RIGHT_UP, self.OnRightUp)
231 for combos in mainframe.treeevents:
232 self.BindEvents(combos)
234 if hasattr(mainframe, "leftpanel"):
235 self.ChangeStyle(mainframe.treestyles)
237 if not(self.GetAGWWindowStyleFlag() & CT.TR_HIDE_ROOT):
238 self.SelectItem(self.root)
239 self.Expand(self.root)
242 def BindEvents(self, choice, recreate=False):
244 value = choice.GetValue()
245 text = choice.GetLabel()
248 binder = self.eventdict[text]
251 if evt == "CT.EVT_TREE_BEGIN_RDRAG":
252 self.Bind(wx.EVT_RIGHT_DOWN, None)
253 self.Bind(wx.EVT_RIGHT_UP, None)
254 self.Bind(eval(evt), binder)
256 self.Bind(eval(evt), None)
257 if evt == "CT.EVT_TREE_BEGIN_RDRAG":
258 self.Bind(wx.EVT_RIGHT_DOWN, self.OnRightDown)
259 self.Bind(wx.EVT_RIGHT_UP, self.OnRightUp)
262 def ChangeStyle(self, combos):
266 if combo.GetValue() == 1:
267 style = style | eval("CT." + combo.GetLabel())
269 if self.GetAGWWindowStyleFlag() != style:
270 self.SetAGWWindowStyleFlag(style)
273 def OnCompareItems(self, item1, item2):
275 t1 = self.GetItemText(item1)
276 t2 = self.GetItemText(item2)
287 def OnIdle(self, event):
291 # if self.gauge.IsEnabled() and self.gauge.IsShown():
292 # self.count = self.count + 1
294 # if self.count >= 50:
297 # self.gauge.SetValue(self.count)
305 def CloseItem(self, itemParent = None, uuid = None) :
306 if itemParent is None :
307 itemParent = self.root
308 child, cookie = self.GetFirstChild(itemParent)
310 pydata = self.GetPyData(child)
311 if pydata['uuid'] == uuid :
312 self.SetItemBold(child, False)
314 self.CloseItem(child, uuid)
315 child, cookie = self.GetNextChild(itemParent, cookie)
317 def GiveFocus(self, itemParent = None, uuid = None, bold = False) :
318 if itemParent is None :
319 itemParent = self.root
320 child, cookie = self.GetFirstChild(itemParent)
322 pydata = self.GetPyData(child)
323 if pydata['uuid'] == uuid :
324 self.SelectItem(child)
326 self.SetItemBold(child, True)
328 self.GiveFocus(child, uuid, bold)
329 child, cookie = self.GetNextChild(itemParent, cookie)
331 def IsInTree(self, itemParent = None, uuid = None) :
332 if itemParent is None :
333 itemParent = self.root
334 child, cookie = self.GetFirstChild(itemParent)
337 pydata = self.GetPyData(child)
338 if pydata['uuid'] == uuid :
340 self.GiveFocus(child, uuid)
341 child, cookie = self.GetNextChild(itemParent, cookie)
345 def OnRightDown(self, event):
347 pt = event.GetPosition()
348 item, flags = self.HitTest(pt)
352 #self.log.info("OnRightClick: %s, %s, %s" % (self.GetItemText(item), type(item), item.__class__) + "\n")
353 self.SelectItem(item)
356 def OnRightUp(self, event):
364 if not self.IsItemEnabled(item):
368 # Item Text Appearance
369 ishtml = self.IsItemHyperText(item)
370 back = self.GetItemBackgroundColour(item)
371 fore = self.GetItemTextColour(item)
372 isbold = self.IsBold(item)
373 font = self.GetItemFont(item)
376 normal = self.GetItemImage(item, CT.TreeItemIcon_Normal)
377 selected = self.GetItemImage(item, CT.TreeItemIcon_Selected)
378 expanded = self.GetItemImage(item, CT.TreeItemIcon_Expanded)
379 selexp = self.GetItemImage(item, CT.TreeItemIcon_SelectedExpanded)
381 # Enabling/Disabling Windows Associated To An Item
382 haswin = self.GetItemWindow(item)
384 # Enabling/Disabling Items
385 enabled = self.IsItemEnabled(item)
387 # Generic Item's Info
388 children = self.GetChildrenCount(item)
389 itemtype = self.GetItemType(item)
390 text = self.GetItemText(item)
391 pydata = self.GetPyData(item)
395 self.itemdict = {"ishtml": ishtml, "back": back, "fore": fore, "isbold": isbold,
396 "font": font, "normal": normal, "selected": selected, "expanded": expanded,
397 "selexp": selexp, "haswin": haswin, "children": children,
398 "itemtype": itemtype, "text": text, "pydata": pydata, "enabled": enabled}
400 if not item in [self.textroot, self.matroot] :
402 info = wx.MenuItem(menu, wx.ID_ANY, _(u"Informations").decode('utf8'))
403 info.SetBitmap(wx.ArtProvider_GetBitmap(wx.ART_INFORMATION, size = (16,16)))
404 menu.AppendItem(info)
406 rename = wx.MenuItem(menu, wx.ID_ANY, _(u"Rename").decode('utf8'))
407 rename.SetBitmap(wx.ArtProvider_GetBitmap(wx.ART_TIP, size = (16,16)))
408 menu.AppendItem(rename)
410 openfolder = wx.MenuItem(menu, wx.ID_ANY, _(u"Open directory").decode('utf8'))
411 openfolder.SetBitmap(wx.ArtProvider_GetBitmap(wx.ART_FOLDER_OPEN, size = (16,16)))
412 menu.AppendItem(openfolder)
414 menu.AppendSeparator()
416 if 'corpus_name' in pydata :
417 buildmenu(menu, self.parent.text_menu)
418 menu.AppendSeparator()
419 elif 'matrix_name' in pydata :
420 buildmenu(menu, self.parent.matrix_menu)
421 menu.AppendSeparator()
422 elif pydata.get('type', False) == 'alceste' and pydata['uuid'] in self.parent.history.opened :
424 antipro = openmenu.Append(wx.ID_ANY, _(u"Antiprofiles").decode('utf8'))
425 menu.AppendMenu(wx.ID_ANY, _(u"Open ...").decode('utf8'), openmenu)
427 profsr = menu.Append(wx.ID_ANY, _(u"Repeated segments profiles").decode('utf8'))
428 profgram = menu.Append(wx.ID_ANY, _(u"POS profiles").decode('utf8'))
429 tgen = menu.Append(wx.ID_ANY, _(u"Tgen Editor").decode('utf8'))
430 computetgen = menu.Append(wx.ID_ANY, _(u"Compute Tgen").decode('utf8'))
431 export_corpus = menu.Append(wx.ID_ANY, _(u"Export corpus").decode('utf8'))
432 colored = menu.Append(wx.ID_ANY, _(u"Colored corpus").decode('utf8'))
433 navig = menu.Append(wx.ID_ANY, _(u"Navigator").decode('utf8'))
434 statclasse = menu.Append(wx.ID_ANY, _(u"Clusters statistics").decode('utf8'))
435 rapport = menu.Append(wx.ID_ANY, _(u"Report").decode('utf8'))
436 export_classes = menu.Append(wx.ID_ANY, _(u"Export clusters").decode('utf8'))
437 subcorpusfromcl = menu.Append(wx.ID_ANY, _(u"Sub corpus from clusters").decode('utf8'))
438 menu.AppendSeparator()
439 self.Bind(wx.EVT_MENU, self.OpenAntipro, antipro)
440 self.Bind(wx.EVT_MENU, self.OnProfSR, profsr)
441 self.Bind(wx.EVT_MENU, self.OnProfGram, profgram)
442 self.Bind(wx.EVT_MENU, self.OnTgenEditor, tgen)
443 self.Bind(wx.EVT_MENU, self.OnTgenCompute, computetgen)
444 self.Bind(wx.EVT_MENU, self.OnExportCorpus, export_corpus)
445 self.Bind(wx.EVT_MENU, self.OnColored, colored)
446 self.Bind(wx.EVT_MENU, self.OnNavig, navig)
447 self.Bind(wx.EVT_MENU, self.StatClasse, statclasse)
448 self.Bind(wx.EVT_MENU, self.OnRapport, rapport)
449 self.Bind(wx.EVT_MENU, self.OnExportClasses, export_classes)
450 self.Bind(wx.EVT_MENU, self.OnSubCorpusFromClusters, subcorpusfromcl)
451 elif pydata.get('type', False) == 'stat' and pydata['uuid'] in self.parent.history.opened :
452 export_dictionary = menu.Append(wx.ID_ANY, _(u"Export dictionary").decode('utf8'))
453 export_lems = menu.Append(wx.ID_ANY, _(u"Export lemma dictionary").decode('utf8'))
454 self.Bind(wx.EVT_MENU, self.OnExportDictionary, export_dictionary)
455 self.Bind(wx.EVT_MENU, self.OnExportLems, export_lems)
456 menu.AppendSeparator()
457 elif pydata.get('type', False) == 'spec' and pydata['uuid'] in self.parent.history.opened :
458 tgen = menu.Append(wx.ID_ANY, _(u"Tgen Editor").decode('utf8'))
459 computetgen = menu.Append(wx.ID_ANY, _(u"Compute Tgen").decode('utf8'))
460 self.Bind(wx.EVT_MENU, self.OnTgenEditor, tgen)
461 self.Bind(wx.EVT_MENU, self.OnTgenCompute, computetgen)
462 menu.AppendSeparator()
463 elif pydata.get('type', False) == 'reinertmatrix' and pydata['uuid'] in self.parent.history.opened :
465 antipro = openmenu.Append(wx.ID_ANY, _(u"antiprofiles").decode('utf8'))
466 rapport = menu.Append(wx.ID_ANY, _(u"Report").decode('utf8'))
467 menu.AppendMenu(wx.ID_ANY, _(u"Open ...").decode('utf8'), openmenu)
468 self.Bind(wx.EVT_MENU, self.OpenAntipro, antipro)
469 self.Bind(wx.EVT_MENU, self.OnRapport, rapport)
471 itemdelete = wx.MenuItem(menu, wx.ID_ANY, _(u"Delete from history").decode('utf8'))
472 itemdelete.SetBitmap(wx.ArtProvider_GetBitmap(wx.ART_DELETE, size = (16,16)))
473 menu.AppendItem(itemdelete)
474 #item11 = menu.Append(wx.ID_ANY, "Prepend An Item")
475 #item12 = menu.Append(wx.ID_ANY, "Append An Item")
477 #self.Bind(wx.EVT_MENU, self.OnItemBackground, item1)
478 #self.Bind(wx.EVT_MENU, self.OnItemForeground, item2)
479 #self.Bind(wx.EVT_MENU, self.OnItemBold, item3)
480 #self.Bind(wx.EVT_MENU, self.OnItemFont, item4)
481 #self.Bind(wx.EVT_MENU, self.OnItemHyperText, item5)
482 #self.Bind(wx.EVT_MENU, self.OnEnableWindow, item6)
483 #self.Bind(wx.EVT_MENU, self.OnDisableItem, item7)
484 #self.Bind(wx.EVT_MENU, self.OnItemIcons, item8)
485 self.Bind(wx.EVT_MENU, self.OnItemInfo, info)
486 self.Bind(wx.EVT_MENU, self.OnRename, rename)
487 self.Bind(wx.EVT_MENU, self.OnItemDelete, itemdelete)
488 self.Bind(wx.EVT_MENU, self.OnOpenFolder, openfolder)
489 #self.Bind(wx.EVT_MENU, self.OnItemPrepend, item11)
490 #self.Bind(wx.EVT_MENU, self.OnItemAppend, item12)
496 busy = wx.BusyInfo(_("Please wait...Reading corpus").decode('utf8'), self.parent)
498 if self.pydata['uuid'] in self.parent.history.openedcorpus :
499 corpus = copycorpus(self.parent.history.openedcorpus[self.pydata['uuid']])
500 elif 'corpus_name' in self.pydata :
501 corpus = Corpus(self.parent, parametres = DoConf(self.pydata['ira']).getoptions('corpus'), read = True)
503 cuuid = self.pydata['corpus']
504 if cuuid in self.parent.history.openedcorpus :
505 corpus = copycorpus(self.parent.history.openedcorpus[cuuid])
507 irapath = self.parent.history.corpus[cuuid]['ira']
508 corpus = Corpus(self.parent, parametres = DoConf(irapath).getoptions('corpus'), read = True)
513 if 'matrix_name' in self.pydata :
514 matrix = Tableau(self.parent, parametres = DoConf(self.pydata['ira']).getoptions('matrix'))
516 return copymatrix(matrix)
518 cuuid = self.pydata['matrix']
519 matrix = Tableau(self.parent, parametres = DoConf(self.history.matrixanalyse[cuuid]['ira']).getoptions('matrix'))
521 return copymatrix(matrix)
523 def OnSpec(self, evt) :
524 self.parent.OnTextSpec(evt, self.getcorpus())
526 def OnStat(self, evt) :
527 self.parent.OnTextStat(evt, self.getcorpus())
529 def OnReinert(self, evt) :
530 self.parent.OnTextReinert(evt, self.getcorpus())
532 def OnPam(self, evt) :
533 self.parent.OnPamSimple(evt, self.getcorpus())
535 def OnSimiTxt(self, evt) :
536 self.parent.OnSimiTxt(evt, self.getcorpus())
538 def OnWordCloud(self, evt) :
539 self.parent.OnWordCloud(evt, self.getcorpus())
541 # def OnFreq(self, evt):
542 # self.parent.OnFreq(evt, self.getmatrix())
544 # def OnChiSquare(self, evt):
545 # self.parent.OnChi2(evt, self.getmatrix())
547 # def OnSimiTab(self, evt):
548 # self.parent.OnSimiTab(evt, self.getmatrix())
550 # def OnProto(self, evt):
551 # self.parent.OnProto(evt, self.getmatrix())
553 # def OnSplitFromVar(self, evt):
554 # self.parent.OnSplitVar(evt, self.getmatrix())
556 # def OnCHDReinert(self, evt):
557 # self.parent.OnCHDReinert(evt, self.getmatrix())
559 #def OnSubTextFromMeta(self, evt):
560 # self.parent.OnSubText(self.getcorpus(), parametres = {'frommeta' : True})
562 #def OnSubTextFromTheme(self, evt):
563 # self.parent.OnSubText(self.getcorpus(), parametres = {'fromtheme' : True})
565 def OnProfSR(self, evt) :
566 ProfileSegment(self.parent, self.page.dictpathout, self.page.parametres, self.page.corpus)
568 def OnProfGram(self, evt) :
569 ProfilType(self.parent, self.page.corpus, self.page.parametres)
571 def OnExportCorpus(self, evt) :
572 dial = PrefExport(self, self.parent)
573 dial.fbb.SetValue(os.path.join(os.path.dirname(self.page.dictpathout['ira']), 'export_corpus.txt'))
574 dial.CenterOnParent()
575 res = dial.ShowModal()
577 if dial.radio_type.GetSelection() == 0 : alc = True
579 if dial.radio_lem.GetSelection() == 0 : lem = True
581 if self.page.parametres['classif_mode'] != 2 :
585 self.page.corpus.export_corpus_classes(dial.fbb.GetValue(), alc = alc, lem = lem, uci = uci)
588 dlg = wx.MessageDialog(self.parent, msg, u"Export", wx.OK | wx.ICON_INFORMATION)
593 def OnColored(self, evt) :
594 dial = PrefSimpleFile(self, self.parent, **{'mask' : '*.html', 'title': _(u"Colored corpus").decode('utf8')})
595 dial.fbb.SetValue(os.path.join(os.path.dirname(self.page.dictpathout['ira']), 'corpus_couleur.html'))
596 dial.CenterOnParent()
597 res = dial.ShowModal()
599 fileout = dial.fbb.GetValue()
601 if self.page.parametres['classif_mode'] != 2 :
605 txt = self.page.corpus.make_colored_corpus(uci = uci)
606 with open(fileout, 'w') as f :
608 msg = ' !\n'.join([_(u"Done").decode('utf8'), _(u"Open in a web browser ?").decode('utf8')])
609 dlg = wx.MessageDialog(self.parent, msg, u"Corpus en couleur", wx.NO | wx.YES | wx.ICON_QUESTION)
611 if dlg.ShowModal() == wx.ID_YES :
612 webbrowser.open(fileout)
615 def OnNavig(self, evt):
616 if 'FrameSearch' not in dir(self.page) :
617 self.page.FrameSearch = SearchFrame(self.parent, -1, _(u"Search ...").decode('utf8'), self.page.corpus)
618 self.page.FrameSearch.Show()
620 def StatClasse(self, evt):
621 dial = PrefSimpleFile(self, self.parent, **{'mask' : '*.csv', 'title': _(u"Clusters statistics").decode('utf8')})
622 dial.fbb.SetValue( os.path.join(os.path.dirname(self.page.dictpathout['ira']), 'stat_par_classe.csv'))
623 dial.CenterOnParent()
624 res = dial.ShowModal()
626 fileout = dial.fbb.GetValue()
628 self.page.corpus.get_stat_by_cluster(fileout)
630 dlg = wx.MessageDialog(self.parent, msg, _(u"Clusters statistics").decode('utf8'), wx.OK | wx.ICON_INFORMATION)
632 if dlg.ShowModal() == wx.ID_OK :
635 def OpenAntipro(self, evt) :
637 for i in range(0, self.page.TabChdSim.GetPageCount()) :
638 page = self.page.TabChdSim.GetPage(i)
639 if self.page.TabChdSim.GetPageText(i) == _(u"Antiprofiles").decode('utf8') :
640 self.page.TabChdSim.SetSelection(i)
644 open_antiprofil(self.page, self.page.dictpathout['ANTIPRO_OUT'], self.parent.syscoding)
645 self.page.TabChdSim.SetSelection(self.page.TabChdSim.GetPageCount() - 1)
647 def OnRapport(self, evt) :
648 dial = PrefSimpleFile(self, self.parent, **{'mask' : '*.txt', 'title': _(u"Report").decode('utf8')})
649 dial.fbb.SetValue(self.page.dictpathout['rapport'])
650 dial.CenterOnParent()
651 res = dial.ShowModal()
653 fileout = dial.fbb.GetValue()
655 with open(fileout, 'w') as f :
656 f.write(self.page.debtext + '\n' + GetTxtProfile(self.page.DictProfile, self.page.cluster_size))
658 dlg = wx.MessageDialog(self.parent, msg, _(u"Report").decode('utf8'), wx.OK | wx.ICON_INFORMATION)
665 def OnExportDictionary(self, evt) :
666 corpus = self.page.corpus
667 corpus.export_dictionary(self.page.pathout['dictionary.csv'], self.parent.syscoding)
668 log.info('export dictionary %s' % self.page.pathout['dictionary.csv'])
669 dial = wx.MessageDialog(self.parent, self.page.pathout['dictionary.csv'], 'Export', wx.OK)
673 def OnExportLems(self, evt) :
674 corpus = self.page.corpus
675 corpus.export_lems(self.page.pathout['lemmes.csv'], self.parent.syscoding)
676 log.info('export lemmes %s' % self.page.pathout['lemmes.csv'])
677 dial = wx.MessageDialog(self.parent, self.page.pathout['lemmes.csv'], 'Export', wx.OK)
681 def OnTgenEditor(self, evt):
682 corpus = self.page.corpus
683 tgenpath = os.path.join(self.page.parametres['pathout'], 'tgen.csv')
684 tgen = TGen(path = tgenpath, encoding = self.parent.syscoding)
685 if os.path.exists(tgenpath) :
687 if isinstance(evt, list) :
689 while 'tgen%i' %i in tgen.tgen :
691 tgenname = 'tgen%i' %i
692 tgen.tgen[tgenname] = evt
693 tgenframe = TGenFrame(self.parent, corpus, tgen)
695 if isinstance(evt, list) :
696 tgenframe.OnNewTgen(None, tgen = tgenname)
698 def OnTgenCompute(self, evt):
699 corpus = self.page.corpus
700 tgenpath = os.path.join(self.page.parametres['pathout'], 'tgen.csv')
701 if not os.path.exists(tgenpath) :
702 message = wx.MessageDialog(self.parent, _(u"No TGen yet !").decode('utf8'), style = wx.ICON_EXCLAMATION | wx.OK)
706 self.page.parametres['tgenpath'] = tgenpath
707 tgen = TGen(path = tgenpath, encoding = self.parent.syscoding)
708 if self.page.parametres['type'] == 'spec' :
709 self.page.parametres['etoiles'] = self.page.etoiles
710 TgenSpec(self.parent, corpus, self.page.parametres)
711 elif self.page.parametres['type'] == 'alceste' :
712 TgenProf(self.parent, corpus, self.page.parametres, self.page.cluster_size)
713 TgenLayout(self.page)
715 def OnExportClasses(self, event):
716 corpus = self.page.corpus
717 if self.page.parametres['classif_mode'] != 2 :
721 busy = wx.BusyInfo(_("Please wait...").decode('utf8'), self.parent)
723 for i in range(1, self.page.parametres['clnb'] + 1) :
724 corpus.export_classe(self.page.pathout['classe_%i_export.txt' % i], i, uci = uci)
726 dial = wx.MessageDialog(self, self.page.pathout['classe_x_export.txt'], u"Export", wx.OK|wx.ICON_INFORMATION)
730 def OnSubCorpusFromClusters(self, evt):
731 self.parent.OnSubText(evt, corpus = self.getcorpus(), parametres = {'fromclusters' : True, 'clnb': self.page.parametres['clnb'], 'lc' : self.page.corpus.lc})
733 def OnRename(self, event):
734 pydata = self.itemdict['pydata']
735 if 'matrix_name' in pydata :
737 elif 'corpus_name' in pydata :
741 oldname = pydata[name]
742 dlg = wx.TextEntryDialog(self, _("New Name").decode('utf8'), _(u'Rename').decode('utf8'), oldname)
743 if dlg.ShowModal() == wx.ID_OK:
744 newname = dlg.GetValue()
746 pydata[name] = newname
747 Totconf = DoConf(configfile=pydata['ira'])
748 conf = Totconf.getoptions()
750 Totconf.makeoptions(Totconf.getsections(), [conf])
751 self.history.update(pydata)
752 self.SetItemText(self.current, newname)
753 self.EnsureVisible(self.current)
755 def OnOpenFolder(self, evt):
757 open_folder(os.path.dirname(self.pydata['ira']))
759 print 'cannot open folder %s' % self.pydata.get('ira', 'noirapath')
761 def OnItemBackground(self, event):
763 colourdata = wx.ColourData()
764 colourdata.SetColour(self.itemdict["back"])
765 dlg = wx.ColourDialog(self, colourdata)
767 dlg.GetColourData().SetChooseFull(True)
769 if dlg.ShowModal() == wx.ID_OK:
770 data = dlg.GetColourData()
771 col1 = data.GetColour().Get()
772 self.SetItemBackgroundColour(self.current, col1)
776 def OnItemForeground(self, event):
778 colourdata = wx.ColourData()
779 colourdata.SetColour(self.itemdict["fore"])
780 dlg = wx.ColourDialog(self, colourdata)
782 dlg.GetColourData().SetChooseFull(True)
784 if dlg.ShowModal() == wx.ID_OK:
785 data = dlg.GetColourData()
786 col1 = data.GetColour().Get()
787 self.SetItemTextColour(self.current, col1)
791 def OnItemBold(self, event):
793 self.SetItemBold(self.current, not self.itemdict["isbold"])
796 def OnItemFont(self, event):
799 font = self.itemdict["font"]
802 font = wx.SystemSettings_GetFont(wx.SYS_DEFAULT_GUI_FONT)
804 data.SetInitialFont(font)
806 dlg = wx.FontDialog(self, data)
808 if dlg.ShowModal() == wx.ID_OK:
809 data = dlg.GetFontData()
810 font = data.GetChosenFont()
811 self.SetItemFont(self.current, font)
816 def OnItemHyperText(self, event):
818 self.SetItemHyperText(self.current, not self.itemdict["ishtml"])
821 def OnEnableWindow(self, event):
823 enable = self.GetItemWindowEnabled(self.current)
824 self.SetItemWindowEnabled(self.current, not enable)
827 def OnDisableItem(self, event):
829 self.EnableItem(self.current, False)
832 def OnItemIcons(self, event):
834 bitmaps = [self.itemdict["normal"], self.itemdict["selected"],
835 self.itemdict["expanded"], self.itemdict["selexp"]]
838 dlg = TreeIcons(self, -1, bitmaps=bitmaps)
843 def SetNewIcons(self, bitmaps):
845 self.SetItemImage(self.current, bitmaps[0], CT.TreeItemIcon_Normal)
846 self.SetItemImage(self.current, bitmaps[1], CT.TreeItemIcon_Selected)
847 self.SetItemImage(self.current, bitmaps[2], CT.TreeItemIcon_Expanded)
848 self.SetItemImage(self.current, bitmaps[3], CT.TreeItemIcon_SelectedExpanded)
851 def OnItemInfo(self, event):
853 itemtext = self.itemdict["text"]
854 numchildren = str(self.itemdict["children"])
855 itemtype = self.itemdict["itemtype"]
856 pydata = self.itemdict['pydata']
857 #if 'analyses' in pydata :
858 # toshow = dict([[val, pydata[val]] for val in pydata if val not in['analyses', 'isload']])
860 toshow = pydata['ira']
861 toshow = DoConf(toshow).getoptions()
862 txt = DoConf().totext(toshow)
863 parametres = [val.split('\t\t:') for val in txt.splitlines()]
869 itemtype = "CheckBox"
871 itemtype = "RadioButton"
873 dlg = InfoDialog(self, itemtext, parametres)
880 def OnItemDelete(self, event):
882 strs = "Are You Sure You Want To Delete Item " + self.GetItemText(self.current) + "?"
883 dlg = wx.MessageDialog(None, strs, 'Deleting Item', wx.OK | wx.CANCEL | wx.ICON_QUESTION)
885 if dlg.ShowModal() in [wx.ID_NO, wx.ID_CANCEL]:
891 pydata = self.itemdict['pydata']
892 if 'corpus_name' in pydata :
893 self.history.delete(pydata, True)
895 self.history.delete(pydata)
896 self.DeleteChildren(self.current)
897 self.Delete(self.current)
902 def OnItemPrepend(self, event):
904 dlg = wx.TextEntryDialog(self, "Please Enter The New Item Name", 'Item Naming', 'Python')
906 if dlg.ShowModal() == wx.ID_OK:
907 newname = dlg.GetValue()
908 newitem = self.PrependItem(self.current, newname)
909 self.EnsureVisible(newitem)
913 def AddAnalyse(self, parametres, itemParent = None, bold = True) :
914 uuid = parametres.get('corpus', None)
915 if uuid is not None :
916 if itemParent is None :
917 itemParent = self.textroot
918 child, cookie = self.GetFirstChild(itemParent)
921 pydata = self.GetPyData(child)
922 if pydata['uuid'] == uuid :
925 self.GiveFocus(child, uuid)
926 child, cookie = self.GetNextChild(itemParent, cookie)
927 #item = self.AppendItem(child, parametres['name'])
928 if corpus is not None :
929 item = self.AppendItem(corpus, parametres['name'])
931 item = self.AppendItem(self.textroot, parametres['name'])
933 item = self.AppendItem(self.matroot, parametres['name'])
934 self.SetPyData(item, parametres)
935 if parametres['type'] in self.ild :
936 img = self.ild[parametres['type']]
939 self.SetItemImage(item, img, CT.TreeItemIcon_Normal)
940 self.SetItemImage(item, 13, CT.TreeItemIcon_Expanded)
941 self.SetItemBold(item, bold)
942 self.SelectItem(item)
944 def AddMatAnalyse(self, parametres, itemParent = None, bold = True) :
945 uuid = parametres.get('matrix', None)
946 if uuid is not None :
947 if itemParent is None :
948 itemParent = self.matroot
949 child, cookie = self.GetFirstChild(itemParent)
952 pydata = self.GetPyData(child)
953 if pydata['uuid'] == uuid :
956 self.GiveFocus(child, uuid)
957 child, cookie = self.GetNextChild(itemParent, cookie)
958 #item = self.AppendItem(child, parametres['name'])
959 if matrix is not None :
960 item = self.AppendItem(matrix, parametres['name'])
962 item = self.AppendItem(self.matroot, parametres['name'])
963 self.SetPyData(item, parametres)
964 if parametres['type'] in self.ild :
965 img = self.ild[parametres['type']]
968 self.SetItemImage(item, img, CT.TreeItemIcon_Normal)
969 self.SetItemImage(item, 13, CT.TreeItemIcon_Expanded)
970 self.SetItemBold(item, bold)
971 self.SelectItem(item)
973 def OnItemAppend(self, item, select = True):
974 if 'corpus_name' in item :
975 child = self.InsertItem(self.textroot, 0, item['corpus_name'])
977 child = self.InsertItem(self.matroot, 0, item['matrix_name'])
978 self.SetPyData(child, item)
979 if item['type'] in self.ild :
980 img = self.ild[item['type']]
983 self.SetItemImage(child, img, CT.TreeItemIcon_Normal)
984 self.SetItemImage(child, img, CT.TreeItemIcon_Expanded)
986 self.history.addtab(item)
987 self.SetItemBold(child, True)
988 self.SelectItem(child)
990 #dlg = wx.TextEntryDialog(self, "Please Enter The New Item Name", 'Item Naming', 'Python')
992 #if dlg.ShowModal() == wx.ID_OK:
993 # newname = dlg.GetValue()
994 # newitem = self.AppendItem(self.current, newname)
995 # self.EnsureVisible(newitem)
1001 def OnBeginEdit(self, event):
1003 #self.log.info("OnBeginEdit" + "\n")
1004 # show how to prevent edit...
1005 item = event.GetItem()
1006 if item and self.GetItemText(item) == "The Root Item":
1008 #self.log.info("You can't edit this one..." + "\n")
1010 # Lets just see what's visible of its children
1012 root = event.GetItem()
1013 (child, cookie) = self.GetFirstChild(root)
1016 #self.log.info("Child [%s] visible = %d" % (self.GetItemText(child), self.IsVisible(child)) + "\n")
1017 (child, cookie) = self.GetNextChild(root, cookie)
1022 def OnEndEdit(self, event):
1024 #self.log.info("OnEndEdit: %s %s" %(event.IsEditCancelled(), event.GetLabel()))
1025 # show how to reject edit, we'll not allow any digits
1026 for x in event.GetLabel():
1027 if x in string.digits:
1028 #self.log.info(", You can't enter digits..." + "\n")
1035 def OnLeftDClick(self, event):
1036 pt = event.GetPosition()
1037 item, flags = self.HitTest(pt)
1038 if item is not None :
1039 pydata = self.GetPyData(item)
1040 if pydata['uuid'] in self.parent.history.opened :
1041 for i in range(self.parent.nb.GetPageCount()) :
1042 page = self.parent.nb.GetPage(i)
1043 if 'parametres' in dir(page) :
1044 if page.parametres['uuid'] == pydata['uuid'] :
1045 self.parent.nb.SetSelection(i)
1047 elif pydata['uuid'] in ['textroot', 'matroot'] :
1050 if os.path.exists(pydata['ira']) :
1051 busy = wx.BusyInfo(_("Please wait..."), self.parent)
1054 OpenAnalyse(self.parent, pydata)
1056 self.SetItemBold(item, True)
1057 self.OnSelChanged(pydata = pydata)
1062 wx.MessageBox(_(u"This file does not exist : %s" % pydata['ira']).decode('utf8'), 'Information', wx.ICON_EXCLAMATION | wx.STAY_ON_TOP )
1063 #if item and (flags & CT.TREE_HITTEST_ONITEMLABEL):
1064 # if self.GetAGWWindowStyleFlag() & CT.TR_EDIT_LABELS:
1065 # self.log.info("OnLeftDClick: %s (manually starting label edit)"% self.GetItemText(item) + "\n")
1067 #self.EditLabel(item)
1069 # pydata = self.GetPyData(item)
1071 # self.log.info("OnLeftDClick: Cannot Start Manual Editing, Missing Style TR_EDIT_LABELS\n")
1076 def OnItemExpanded(self, event):
1078 item = event.GetItem()
1080 self.log.info("OnItemExpanded: %s" % self.GetItemText(item) + "\n")
1083 def OnItemExpanding(self, event):
1085 item = event.GetItem()
1087 self.log.info("OnItemExpanding: %s" % self.GetItemText(item) + "\n")
1092 def OnItemCollapsed(self, event):
1094 item = event.GetItem()
1096 self.log.info("OnItemCollapsed: %s" % self.GetItemText(item) + "\n")
1099 def OnItemCollapsing(self, event):
1101 item = event.GetItem()
1103 self.log.info("OnItemCollapsing: %s" % self.GetItemText(item) + "\n")
1108 def OnSelChanged(self, event = None, pydata = None):
1109 if event is not None :
1110 item = event.GetItem()
1111 pydata = self.GetPyData(item)
1113 if pydata is not None :
1114 if 'corpus_name' in pydata or 'corpus' in pydata :
1115 self.ira.ShowMenu('matrix', False)
1116 self.ira.ShowMenu('text', True)
1117 if 'matrix_name' in pydata or 'matrix' in pydata:
1118 self.ira.ShowMenu('text', False)
1119 self.ira.ShowMenu('matrix', True)
1120 if 'uuid' in pydata :
1121 if pydata['uuid'] in ['textroot', 'matroot'] :
1122 self.ira.ShowMenu('text', False)
1123 self.ira.ShowMenu('matrix', False)
1124 self.pydata = pydata
1125 if pydata['uuid'] in self.parent.history.opened :
1126 for i in range(self.parent.nb.GetPageCount()) :
1127 self.page = self.parent.nb.GetPage(i)
1128 if 'parametres' in dir(self.page) :
1129 if self.page.parametres['uuid'] == pydata['uuid'] :
1130 self.parent.nb.SetSelection(i)
1132 if event is not None :
1136 def OnSelChanging(self, event):
1138 item = event.GetItem()
1139 olditem = event.GetOldItem()
1143 olditemtext = "None"
1145 olditemtext = self.GetItemText(olditem)
1146 #self.log.info("OnSelChanging: From %s" % olditemtext + " To %s" % self.GetItemText(item) + "\n")
1151 def OnBeginDrag(self, event):
1153 self.item = event.GetItem()
1155 self.log.info("Beginning Drag..." + "\n")
1160 def OnBeginRDrag(self, event):
1162 self.item = event.GetItem()
1164 self.log.info("Beginning Right Drag..." + "\n")
1169 def OnEndDrag(self, event):
1171 self.item = event.GetItem()
1173 self.log.info("Ending Drag!" + "\n")
1178 def OnDeleteItem(self, event):
1180 item = event.GetItem()
1185 self.log.info("Deleting Item: %s" % self.GetItemText(item) + "\n")
1189 def OnItemCheck(self, event):
1191 item = event.GetItem()
1192 self.log.info("Item " + self.GetItemText(item) + " Has Been Checked!\n")
1196 def OnItemChecking(self, event):
1198 item = event.GetItem()
1199 self.log.info("Item " + self.GetItemText(item) + " Is Being Checked...\n")
1203 def OnToolTip(self, event):
1205 item = event.GetItem()
1207 event.SetToolTip(wx.ToolTip(self.GetItemText(item)))
1210 def OnItemMenu(self, event):
1212 item = event.GetItem()
1214 self.log.info("OnItemMenu: %s" % self.GetItemText(item) + "\n")
1219 def OnKey(self, event):
1221 keycode = event.GetKeyCode()
1222 keyname = keyMap.get(keycode, None)
1224 if keycode == wx.WXK_BACK:
1225 self.log.info("OnKeyDown: HAHAHAHA! I Vetoed Your Backspace! HAHAHAHA\n")
1229 if "unicode" in wx.PlatformInfo:
1230 keycode = event.GetUnicodeKey()
1232 keycode = event.GetKeyCode()
1233 keyname = "\"" + unichr(event.GetUnicodeKey()) + "\""
1235 keyname = "Ctrl-%s" % chr(ord('A') + keycode-1)
1241 keyname = "Ctrl-%s" % chr(ord('A') + keycode-1)
1243 keyname = "\"%s\"" % chr(keycode)
1245 keyname = "unknown (%s)" % keycode
1247 self.log.info("OnKeyDown: You Pressed '" + keyname + "'\n")
1252 def OnActivate(self, event):
1255 self.log.info("OnActivate: %s" % self.GetItemText(self.item) + "\n")
1260 def OnHyperLink(self, event):
1262 item = event.GetItem()
1264 self.log.info("OnHyperLink: %s" % self.GetItemText(self.item) + "\n")
1267 def OnTextCtrl(self, event):
1269 char = chr(event.GetKeyCode())
1270 self.log.info("EDITING THE TEXTCTRL: You Wrote '" + char + \
1271 "' (KeyCode = " + str(event.GetKeyCode()) + ")\n")
1275 def OnComboBox(self, event):
1277 selection = event.GetEventObject().GetValue()
1278 self.log.info("CHOICE FROM COMBOBOX: You Chose '" + selection + "'\n")