iramuteq.org Git - iramuteq/blob - 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
 383         itemdelete = menu.Append(wx.ID_ANY, "Supprimer de l'historique")
 384         if item == self.GetRootItem():
 385             itemdelete.Enable(False)
 386         #item11 = menu.Append(wx.ID_ANY, "Prepend An Item")
 387         #item12 = menu.Append(wx.ID_ANY, "Append An Item")
 388
 389         #self.Bind(wx.EVT_MENU, self.OnItemBackground, item1)
 390         #self.Bind(wx.EVT_MENU, self.OnItemForeground, item2)
 391         #self.Bind(wx.EVT_MENU, self.OnItemBold, item3)
 392         #self.Bind(wx.EVT_MENU, self.OnItemFont, item4)
 393         #self.Bind(wx.EVT_MENU, self.OnItemHyperText, item5)
 394         #self.Bind(wx.EVT_MENU, self.OnEnableWindow, item6)
 395         #self.Bind(wx.EVT_MENU, self.OnDisableItem, item7)
 396         #self.Bind(wx.EVT_MENU, self.OnItemIcons, item8)
 397         self.Bind(wx.EVT_MENU, self.OnItemInfo, info)
 398         self.Bind(wx.EVT_MENU, self.OnItemDelete, itemdelete)
 399         #self.Bind(wx.EVT_MENU, self.OnItemPrepend, item11)
 400         #self.Bind(wx.EVT_MENU, self.OnItemAppend, item12)
 401
 402         self.PopupMenu(menu)
 403         menu.Destroy()
 404
 405     def getcorpus(self):
 406         if self.pydata['uuid'] in self.parent.history.openedcorpus :
 407             return copycorpus(self.parent.history.openedcorpus[self.pydata['uuid']])
 408         elif 'corpus_name' in self.pydata :
 409             return Corpus(self.parent, parametres = DoConf(self.pydata['ira']).getoptions('corpus'), read = True)
 410         else :
 411             cuuid = self.pydata['corpus']
 412             if cuuid in self.parent.history.openedcorpus :
 413                 return copycorpus(self.parent.history.openedcorpus[cuuid])
 414             else :
 415                 irapath = self.parent.history.corpus[cuuid]['ira']
 416                 return Corpus(self.parent, parametres = DoConf(irapath).getoptions('corpus'), read = True)
 417
 418     def OnSpec(self, evt) :
 419         self.parent.OnTextSpec(evt, self.getcorpus())
 420
 421     def OnStat(self, evt) :
 422         self.parent.OnTextStat(evt, self.getcorpus())
 423
 424     def OnAlceste(self, evt) :
 425         self.parent.OnTextAlceste(evt, self.getcorpus())
 426
 427     def OnPam(self, evt) :
 428         self.parent.OnPamSimple(evt, self.getcorpus())
 429
 430     def OnSimiTxt(self, evt) :
 431         self.parent.OnSimiTxt(evt, self.getcorpus())
 432
 433     def OnWordCloud(self, evt) :
 434         self.parent.OnWordCloud(evt, self.getcorpus())
 435
 436     def OnProfSR(self, evt) :
 437         ProfileSegment(self.parent, self.page.dictpathout, self.page.parametres, self.page.corpus)
 438
 439     def OnProfGram(self, evt) :
 440         ProfilType(self.parent, self.page.corpus, self.page.parametres)
 441
 442     def OnExportCorpus(self, evt) :
 443         dial = PrefExport(self, self.parent)
 444         dial.fbb.SetValue(os.path.join(os.path.dirname(self.page.dictpathout['ira']), 'export_corpus.txt'))
 445         dial.CenterOnParent()
 446         res = dial.ShowModal()
 447         if res == wx.ID_OK :
 448             if dial.radio_type.GetSelection() == 0 : alc = True
 449             else : alc = False
 450             if dial.radio_lem.GetSelection() == 0 : lem = True
 451             else : lem = False
 452             self.page.corpus.export_corpus_classes(dial.fbb.GetValue(), alc = alc, lem = lem)
 453             msg = u"Fini !"
 454             dial.Destroy()
 455             dlg = wx.MessageDialog(self.parent, msg, u"Export", wx.OK | wx.NO_DEFAULT | wx.ICON_INFORMATION)
 456             dlg.CenterOnParent()
 457             dlg.ShowModal()
 458             dlg.Destroy()
 459
 460     def OnColored(self, evt) :
 461         dial = PrefSimpleFile(self, self.parent, **{'mask' : '*.html', 'title': 'Corpus en couleur'})
 462         dial.fbb.SetValue(os.path.join(os.path.dirname(self.page.dictpathout['ira']), 'corpus_couleur.html'))
 463         dial.CenterOnParent()
 464         res = dial.ShowModal()
 465         if res == wx.ID_OK :
 466             fileout = dial.fbb.GetValue()
 467             dial.Destroy()
 468             txt = self.page.corpus.make_colored_corpus()
 469             with open(fileout, 'w') as f :
 470                 f.write(txt)
 471             msg = u"Fini !\nVoulez-vous ouvrir le corpus dans votre navigateur ?"
 472             dlg = wx.MessageDialog(self.parent, msg, u"Corpus en couleur", wx.NO | wx.YES | wx.NO_DEFAULT | wx.ICON_QUESTION)
 473             dlg.CenterOnParent()
 474             if dlg.ShowModal() == wx.ID_YES :
 475                 webbrowser.open(fileout)
 476             dlg.Destroy()
 477
 478     def OnNavig(self, evt):
 479         if 'FrameSearch' not in dir(self.page) :
 480             self.page.FrameSearch = SearchFrame(self.parent, -1, u"Rechercher...", self.page.corpus)
 481         self.page.FrameSearch.Show()
 482
 483     def StatClasse(self, evt):
 484         dial = PrefSimpleFile(self, self.parent, **{'mask' : '*.csv', 'title': 'Stat par classe'})
 485         dial.fbb.SetValue( os.path.join(os.path.dirname(self.page.dictpathout['ira']), 'stat_par_classe.csv'))
 486         dial.CenterOnParent()
 487         res = dial.ShowModal()
 488         if res == wx.ID_OK :
 489             fileout = dial.fbb.GetValue()
 490             dial.Destroy()
 491             self.page.corpus.get_stat_by_cluster(fileout)
 492             msg = u"Fini !"
 493             dlg = wx.MessageDialog(self.parent, msg, u"Stat par classe", wx.OK | wx.NO_DEFAULT | wx.ICON_INFORMATION)
 494             dlg.CenterOnParent()
 495             if dlg.ShowModal() == wx.ID_OK :
 496                 dlg.Destroy()
 497
 498     def OpenAntipro(self, evt) :
 499         find = False
 500         for i in range(0, self.page.TabChdSim.GetPageCount()) :
 501             page = self.page.TabChdSim.GetPage(i)
 502             if self.page.TabChdSim.GetPageText(i) == 'Antiprofils' :
 503                 self.page.TabChdSim.SetSelection(i)
 504                 find = True
 505                 break
 506         if not find :
 507             open_antiprofil(self.page, self.page.dictpathout['ANTIPRO_OUT'], self.parent.syscoding)
 508             self.page.TabChdSim.SetSelection(self.page.TabChdSim.GetPageCount() - 1)
 509
 510     def OnRapport(self, evt) :
 511         dial = PrefSimpleFile(self, self.parent, **{'mask' : '*.txt', 'title': 'Rapport'})
 512         dial.fbb.SetValue(self.page.dictpathout['rapport'])
 513         dial.CenterOnParent()
 514         res = dial.ShowModal()
 515         if res == wx.ID_OK :
 516             fileout = dial.fbb.GetValue()
 517             dial.Destroy()
 518             with open(fileout, 'w') as f :
 519                 f.write(self.page.debtext + '\n' + GetTxtProfile(self.page.DictProfile, self.page.cluster_size))
 520             msg = u"Fini !"
 521             dlg = wx.MessageDialog(self.parent, msg, u"Rapport", wx.OK | wx.NO_DEFAULT | wx.ICON_INFORMATION)
 522             dlg.CenterOnParent()
 523             dlg.ShowModal()
 524             dlg.Destroy()
 525         else :
 526             dial.Destroy()
 527
 528     def OnExportDictionary(self, evt) :
 529         corpus = self.page.corpus
 530         corpus.export_dictionary(self.page.pathout['dictionary.csv'], self.parent.syscoding)
 531         log.info('export dictionary %s' % self.page.pathout['dictionary.csv'])
 532         dial = wx.MessageDialog(self.parent, self.page.pathout['dictionary.csv'], 'Export', wx.OK)
 533         dial.ShowModal()
 534         dial.Destroy()
 535
 536     def OnExportLems(self, evt) :
 537         corpus = self.page.corpus
 538         corpus.export_lems(self.page.pathout['lemmes.csv'], self.parent.syscoding)
 539         log.info('export lemmes %s' % self.page.pathout['lemmes.csv'])
 540         dial = wx.MessageDialog(self.parent, self.page.pathout['lemmes.csv'], 'Export', wx.OK)
 541         dial.ShowModal()
 542         dial.Destroy()
 543
 544     def OnItemBackground(self, event):
 545
 546         colourdata = wx.ColourData()
 547         colourdata.SetColour(self.itemdict["back"])
 548         dlg = wx.ColourDialog(self, colourdata)
 549
 550         dlg.GetColourData().SetChooseFull(True)
 551
 552         if dlg.ShowModal() == wx.ID_OK:
 553             data = dlg.GetColourData()
 554             col1 = data.GetColour().Get()
 555             self.SetItemBackgroundColour(self.current, col1)
 556         dlg.Destroy()
 557
 558
 559     def OnItemForeground(self, event):
 560
 561         colourdata = wx.ColourData()
 562         colourdata.SetColour(self.itemdict["fore"])
 563         dlg = wx.ColourDialog(self, colourdata)
 564
 565         dlg.GetColourData().SetChooseFull(True)
 566
 567         if dlg.ShowModal() == wx.ID_OK:
 568             data = dlg.GetColourData()
 569             col1 = data.GetColour().Get()
 570             self.SetItemTextColour(self.current, col1)
 571         dlg.Destroy()
 572
 573
 574     def OnItemBold(self, event):
 575
 576         self.SetItemBold(self.current, not self.itemdict["isbold"])
 577
 578
 579     def OnItemFont(self, event):
 580
 581         data = wx.FontData()
 582         font = self.itemdict["font"]
 583
 584         if font is None:
 585             font = wx.SystemSettings_GetFont(wx.SYS_DEFAULT_GUI_FONT)
 586
 587         data.SetInitialFont(font)
 588
 589         dlg = wx.FontDialog(self, data)
 590
 591         if dlg.ShowModal() == wx.ID_OK:
 592             data = dlg.GetFontData()
 593             font = data.GetChosenFont()
 594             self.SetItemFont(self.current, font)
 595
 596         dlg.Destroy()
 597
 598
 599     def OnItemHyperText(self, event):
 600
 601         self.SetItemHyperText(self.current, not self.itemdict["ishtml"])
 602
 603
 604     def OnEnableWindow(self, event):
 605
 606         enable = self.GetItemWindowEnabled(self.current)
 607         self.SetItemWindowEnabled(self.current, not enable)
 608
 609
 610     def OnDisableItem(self, event):
 611
 612         self.EnableItem(self.current, False)
 613
 614
 615     def OnItemIcons(self, event):
 616
 617         bitmaps = [self.itemdict["normal"], self.itemdict["selected"],
 618                    self.itemdict["expanded"], self.itemdict["selexp"]]
 619
 620         wx.BeginBusyCursor()
 621         dlg = TreeIcons(self, -1, bitmaps=bitmaps)
 622         wx.EndBusyCursor()
 623         dlg.ShowModal()
 624
 625
 626     def SetNewIcons(self, bitmaps):
 627
 628         self.SetItemImage(self.current, bitmaps[0], CT.TreeItemIcon_Normal)
 629         self.SetItemImage(self.current, bitmaps[1], CT.TreeItemIcon_Selected)
 630         self.SetItemImage(self.current, bitmaps[2], CT.TreeItemIcon_Expanded)
 631         self.SetItemImage(self.current, bitmaps[3], CT.TreeItemIcon_SelectedExpanded)
 632
 633
 634     def OnItemInfo(self, event):
 635
 636         itemtext = self.itemdict["text"]
 637         numchildren = str(self.itemdict["children"])
 638         itemtype = self.itemdict["itemtype"]
 639         pydata = self.itemdict['pydata']
 640         #if 'analyses' in pydata :
 641         #    toshow = dict([[val, pydata[val]] for val in pydata if val not in['analyses', 'isload']])
 642         #else :
 643         toshow = pydata['ira']
 644         toshow = DoConf(toshow).getoptions()
 645         txt = DoConf().totext(toshow)
 646         parametres = [val.split('\t\t:') for val in txt.splitlines()]
 647         parametres.sort()
 648
 649         if itemtype == 0:
 650             itemtype = "Normal"
 651         elif itemtype == 1:
 652             itemtype = "CheckBox"
 653         else:
 654             itemtype = "RadioButton"
 655
 656         dlg = InfoDialog(self, itemtext, parametres)
 657         dlg.CenterOnParent()
 658         dlg.ShowModal()
 659         dlg.Destroy()
 660
 661
 662
 663     def OnItemDelete(self, event):
 664
 665         strs = "Are You Sure You Want To Delete Item " + self.GetItemText(self.current) + "?"
 666         dlg = wx.MessageDialog(None, strs, 'Deleting Item', wx.YES | wx.NO_DEFAULT | wx.CANCEL | wx.ICON_QUESTION)
 667
 668         if dlg.ShowModal() in [wx.ID_NO, wx.ID_CANCEL]:
 669             dlg.Destroy()
 670             return
 671
 672         dlg.Destroy()
 673
 674         pydata = self.itemdict['pydata']
 675         if 'corpus_name' in pydata :
 676             self.history.delete(pydata, True)
 677         else :
 678             self.history.delete(pydata)
 679         self.DeleteChildren(self.current)
 680         self.Delete(self.current)
 681         self.current = None
 682
 683
 684
 685     def OnItemPrepend(self, event):
 686
 687         dlg = wx.TextEntryDialog(self, "Please Enter The New Item Name", 'Item Naming', 'Python')
 688
 689         if dlg.ShowModal() == wx.ID_OK:
 690             newname = dlg.GetValue()
 691             newitem = self.PrependItem(self.current, newname)
 692             self.EnsureVisible(newitem)
 693
 694         dlg.Destroy()
 695
 696     def AddAnalyse(self, parametres, itemParent = None, bold = True) :
 697         uuid = parametres.get('corpus', None)
 698         if uuid is not None :
 699             if itemParent is None :
 700                 itemParent = self.root
 701             child, cookie = self.GetFirstChild(itemParent)
 702             corpus = None
 703             while child :
 704                 pydata = self.GetPyData(child)
 705                 if pydata['uuid'] == uuid :
 706                     corpus = child
 707                     break
 708                 self.GiveFocus(child, uuid)
 709                 child, cookie = self.GetNextChild(itemParent, cookie)
 710             #item = self.AppendItem(child, parametres['name'])
 711             if corpus is not None :
 712                 item = self.AppendItem(corpus, parametres['name'])
 713             else :
 714                 item = self.AppendItem(self.root, parametres['name'])
 715         else :
 716             item = self.AppendItem(self.root, parametres['name'])
 717         self.SetPyData(item, parametres)
 718         self.SetItemImage(item, 24, CT.TreeItemIcon_Normal)
 719         self.SetItemImage(item, 13, CT.TreeItemIcon_Expanded)
 720         self.SetItemBold(item, bold)
 721
 722     def OnItemAppend(self, item):
 723         child = self.AppendItem(self.root, item['corpus_name'])
 724         self.SetPyData(child, item)
 725         self.history.addtab(item)
 726         self.SetItemImage(child, 24, CT.TreeItemIcon_Normal)
 727         self.SetItemImage(child, 13, CT.TreeItemIcon_Expanded)
 728         self.SetItemBold(child, True)
 729
 730         #dlg = wx.TextEntryDialog(self, "Please Enter The New Item Name", 'Item Naming', 'Python')
 731
 732         #if dlg.ShowModal() == wx.ID_OK:
 733         #    newname = dlg.GetValue()
 734         #    newitem = self.AppendItem(self.current, newname)
 735         #    self.EnsureVisible(newitem)
 736
 737
 738         #dlg.Destroy()
 739
 740
 741     def OnBeginEdit(self, event):
 742
 743         #self.log.info("OnBeginEdit" + "\n")
 744         # show how to prevent edit...
 745         item = event.GetItem()
 746         if item and self.GetItemText(item) == "The Root Item":
 747             wx.Bell()
 748             #self.log.info("You can't edit this one..." + "\n")
 749
 750             # Lets just see what's visible of its children
 751             cookie = 0
 752             root = event.GetItem()
 753             (child, cookie) = self.GetFirstChild(root)
 754
 755             while child:
 756                 #self.log.info("Child [%s] visible = %d" % (self.GetItemText(child), self.IsVisible(child)) + "\n")
 757                 (child, cookie) = self.GetNextChild(root, cookie)
 758
 759             event.Veto()
 760
 761
 762     def OnEndEdit(self, event):
 763
 764         #self.log.info("OnEndEdit: %s %s" %(event.IsEditCancelled(), event.GetLabel()))
 765         # show how to reject edit, we'll not allow any digits
 766         for x in event.GetLabel():
 767             if x in string.digits:
 768                 #self.log.info(", You can't enter digits..." + "\n")
 769                 event.Veto()
 770                 return
 771
 772         self.log.info("\n")
 773
 774
 775     def OnLeftDClick(self, event):
 776
 777         pt = event.GetPosition()
 778         item, flags = self.HitTest(pt)
 779         if item is not None :
 780             pydata = self.GetPyData(item)
 781             if pydata['uuid'] in self.parent.history.opened :
 782                 for i in range(self.parent.nb.GetPageCount()) :
 783                     page = self.parent.nb.GetPage(i)
 784                     if 'parametres' in dir(page) :
 785                         if page.parametres['uuid'] == pydata['uuid'] :
 786                             self.parent.nb.SetSelection(i)
 787                             break
 788             else :
 789                 OpenAnalyse(self.parent, pydata)
 790                 self.SetItemBold(item, True)
 791                 self.OnSelChanged(pydata = pydata)
 792         #if item and (flags & CT.TREE_HITTEST_ONITEMLABEL):
 793         #    if self.GetAGWWindowStyleFlag() & CT.TR_EDIT_LABELS:
 794         #        self.log.info("OnLeftDClick: %s (manually starting label edit)"% self.GetItemText(item) + "\n")
 795
 796                 #self.EditLabel(item)
 797         #    else:
 798         #        pydata = self.GetPyData(item)
 799         #        print pydata
 800         #        self.log.info("OnLeftDClick: Cannot Start Manual Editing, Missing Style TR_EDIT_LABELS\n")
 801
 802         event.Skip()
 803
 804
 805     def OnItemExpanded(self, event):
 806
 807         item = event.GetItem()
 808         if item:
 809             self.log.info("OnItemExpanded: %s" % self.GetItemText(item) + "\n")
 810
 811
 812     def OnItemExpanding(self, event):
 813
 814         item = event.GetItem()
 815         if item:
 816             self.log.info("OnItemExpanding: %s" % self.GetItemText(item) + "\n")
 817
 818         event.Skip()
 819
 820
 821     def OnItemCollapsed(self, event):
 822
 823         item = event.GetItem()
 824         if item:
 825             self.log.info("OnItemCollapsed: %s" % self.GetItemText(item) + "\n")
 826
 827
 828     def OnItemCollapsing(self, event):
 829
 830         item = event.GetItem()
 831         if item:
 832             self.log.info("OnItemCollapsing: %s" % self.GetItemText(item) + "\n")
 833
 834         event.Skip()
 835
 836
 837     def OnSelChanged(self, event = None, pydata = None):
 838         if event is not None :
 839             item = event.GetItem()
 840             pydata = self.GetPyData(item)
 841         if pydata is not None :
 842             self.pydata = pydata
 843             if pydata['uuid'] in self.parent.history.opened :
 844                 for i in range(self.parent.nb.GetPageCount()) :
 845                     self.page = self.parent.nb.GetPage(i)
 846                     if 'parametres' in dir(self.page) :
 847                         if self.page.parametres['uuid'] == pydata['uuid'] :
 848                             self.parent.nb.SetSelection(i)
 849                             break
 850         if event is not None :
 851             event.Skip()
 852
 853
 854     def OnSelChanging(self, event):
 855
 856         item = event.GetItem()
 857         olditem = event.GetOldItem()
 858
 859         if item:
 860             if not olditem:
 861                 olditemtext = "None"
 862             else:
 863                 olditemtext = self.GetItemText(olditem)
 864             #self.log.info("OnSelChanging: From %s" % olditemtext + " To %s" % self.GetItemText(item) + "\n")
 865
 866         event.Skip()
 867
 868
 869     def OnBeginDrag(self, event):
 870
 871         self.item = event.GetItem()
 872         if self.item:
 873             self.log.info("Beginning Drag..." + "\n")
 874
 875             event.Allow()
 876
 877
 878     def OnBeginRDrag(self, event):
 879
 880         self.item = event.GetItem()
 881         if self.item:
 882             self.log.info("Beginning Right Drag..." + "\n")
 883
 884             event.Allow()
 885
 886
 887     def OnEndDrag(self, event):
 888
 889         self.item = event.GetItem()
 890         if self.item:
 891             self.log.info("Ending Drag!" + "\n")
 892
 893         event.Skip()
 894
 895
 896     def OnDeleteItem(self, event):
 897
 898         item = event.GetItem()
 899
 900         if not item:
 901             return
 902
 903         self.log.info("Deleting Item: %s" % self.GetItemText(item) + "\n")
 904         event.Skip()
 905
 906
 907     def OnItemCheck(self, event):
 908
 909         item = event.GetItem()
 910         self.log.info("Item " + self.GetItemText(item) + " Has Been Checked!\n")
 911         event.Skip()
 912
 913
 914     def OnItemChecking(self, event):
 915
 916         item = event.GetItem()
 917         self.log.info("Item " + self.GetItemText(item) + " Is Being Checked...\n")
 918         event.Skip()
 919
 920
 921     def OnToolTip(self, event):
 922
 923         item = event.GetItem()
 924         if item:
 925             event.SetToolTip(wx.ToolTip(self.GetItemText(item)))
 926
 927
 928     def OnItemMenu(self, event):
 929
 930         item = event.GetItem()
 931         if item:
 932             self.log.info("OnItemMenu: %s" % self.GetItemText(item) + "\n")
 933
 934         event.Skip()
 935
 936
 937     def OnKey(self, event):
 938
 939         keycode = event.GetKeyCode()
 940         keyname = keyMap.get(keycode, None)
 941
 942         if keycode == wx.WXK_BACK:
 943             self.log.info("OnKeyDown: HAHAHAHA! I Vetoed Your Backspace! HAHAHAHA\n")
 944             return
 945
 946         if keyname is None:
 947             if "unicode" in wx.PlatformInfo:
 948                 keycode = event.GetUnicodeKey()
 949                 if keycode <= 127:
 950                     keycode = event.GetKeyCode()
 951                 keyname = "\"" + unichr(event.GetUnicodeKey()) + "\""
 952                 if keycode < 27:
 953                     keyname = "Ctrl-%s" % chr(ord('A') + keycode-1)
 954
 955             elif keycode < 256:
 956                 if keycode == 0:
 957                     keyname = "NUL"
 958                 elif keycode < 27:
 959                     keyname = "Ctrl-%s" % chr(ord('A') + keycode-1)
 960                 else:
 961                     keyname = "\"%s\"" % chr(keycode)
 962             else:
 963                 keyname = "unknown (%s)" % keycode
 964
 965         self.log.info("OnKeyDown: You Pressed '" + keyname + "'\n")
 966
 967         event.Skip()
 968
 969
 970     def OnActivate(self, event):
 971
 972         if self.item:
 973             self.log.info("OnActivate: %s" % self.GetItemText(self.item) + "\n")
 974
 975         event.Skip()
 976
 977
 978     def OnHyperLink(self, event):
 979
 980         item = event.GetItem()
 981         if item:
 982             self.log.info("OnHyperLink: %s" % self.GetItemText(self.item) + "\n")
 983
 984
 985     def OnTextCtrl(self, event):
 986
 987         char = chr(event.GetKeyCode())
 988         self.log.info("EDITING THE TEXTCTRL: You Wrote '" + char + \
 989                        "' (KeyCode = " + str(event.GetKeyCode()) + ")\n")
 990         event.Skip()
 991
 992
 993     def OnComboBox(self, event):
 994
 995         selection = event.GetEventObject().GetValue()
 996         self.log.info("CHOICE FROM COMBOBOX: You Chose '" + selection + "'\n")
 997         event.Skip()