iramuteq.org Git - iramuteq/blob - tree.py

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