...
[iramuteq] / tree.py
diff --git a/tree.py b/tree.py
old mode 100644 (file)
new mode 100755 (executable)
index 2bbe02b..a5d85e6
--- a/tree.py
+++ b/tree.py
@@ -11,14 +11,15 @@ import logging
 from openanalyse import OpenAnalyse
 from corpus import Corpus, copycorpus
 from tableau import Tableau, copymatrix
-from functions import DoConf, GetTxtProfile, TGen, BugReport
+from functions import DoConf, GetTxtProfile, TGen, BugReport, open_folder, translateprofile, ReadProfileAsDico, write_translation_profile, progressbar, doconcorde
 from profile_segment import ProfileSegment, ProfilType
 from search_tools import SearchFrame
-from dialog import PrefSimpleFile, PrefExport
+from dialog import PrefSimpleFile, PrefExport, SearchCorpus, translate_dialog, PrefUCECarac
 from layout import open_antiprofil, TgenLayout
 from guifunct import TGenFrame
 from textaslexico import TgenSpec
 from textreinert import TgenProf
+from mergeclustergraph import MergeClusterGraph
 
 log = logging.getLogger('iramuteq.tree')
 
@@ -44,36 +45,38 @@ def buildmenu(menu, parent_menu):
             menu.AppendItem(nitem)
 
 class InfoDialog ( wx.Dialog ):
-    
+
     def __init__( self, parent, txt, parametres ):
         wx.Dialog.__init__ ( self, parent, id = wx.ID_ANY, title = u"Informations", pos = wx.DefaultPosition, size = wx.DefaultSize, style = wx.DEFAULT_DIALOG_STYLE )
         if len(parametres) > 30 :
             nb = 4
         else :
-            nb = 2       
+            nb = 2
         self.SetSizeHintsSz( wx.Size( 500,200 ), wx.DefaultSize )
-        
+
         bSizer1 = wx.BoxSizer( wx.VERTICAL )
-        
+
         self.m_panel2 = wx.Panel( self, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, wx.TAB_TRAVERSAL )
         bSizer2 = wx.BoxSizer( wx.VERTICAL )
-        
+
         self.m_staticText4 = wx.StaticText( self.m_panel2, wx.ID_ANY, txt, wx.DefaultPosition, wx.DefaultSize, 0 )
         self.m_staticText4.Wrap( -1 )
         bSizer2.Add( self.m_staticText4, 0, wx.ALL, 5 )
-        
-        
+
         self.m_panel2.SetSizer( bSizer2 )
         self.m_panel2.Layout()
         bSizer2.Fit( self.m_panel2 )
         bSizer1.Add( self.m_panel2, 0, wx.EXPAND |wx.ALL, 5 )
-        
-        self.m_panel1 = wx.Panel( self, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, wx.TAB_TRAVERSAL )
+
+        self.m_panel1 = wx.Panel( self, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, wx.TAB_TRAVERSAL|wx.VSCROLL )
 
         fgSizer1 = wx.FlexGridSizer( 0, nb, 0, 0 )
+        fgSizer1.AddGrowableCol( 1 )
+        if nb == 4 :
+            fgSizer1.AddGrowableCol( 3 )
         fgSizer1.SetFlexibleDirection( wx.BOTH )
-        fgSizer1.SetNonFlexibleGrowMode( wx.FLEX_GROWMODE_SPECIFIED )
-        
+        fgSizer1.SetNonFlexibleGrowMode( wx.FLEX_GROWMODE_NONE )
+
         txtctrl = []
         for val in parametres :
             fgSizer1.Add( wx.StaticText( self.m_panel1, wx.ID_ANY, val[0], wx.DefaultPosition, wx.DefaultSize, 0 ), 0, wx.ALL, 0)
@@ -88,22 +91,20 @@ class InfoDialog ( wx.Dialog ):
         self.m_panel1.SetSizer( fgSizer1 )
         self.m_panel1.Layout()
         fgSizer1.Fit( self.m_panel1 )
-        bSizer1.Add( self.m_panel1, 0, wx.EXPAND|wx.ALL, 3 )
-        
+        bSizer1.Add( self.m_panel1, 1, wx.EXPAND|wx.ALL, 3 )
+
         m_sdbSizer1 = wx.StdDialogButtonSizer()
         self.m_sdbSizer1OK = wx.Button( self, wx.ID_OK )
         m_sdbSizer1.AddButton( self.m_sdbSizer1OK )
         m_sdbSizer1.Realize();
-        
+
         bSizer1.Add( m_sdbSizer1, 0, wx.EXPAND, 5 )
-        
-        
+
         self.SetSizer( bSizer1 )
         self.Layout()
         bSizer1.Fit( self )
-        
         self.Centre( wx.BOTH )
-    
+
     def __del__( self ):
         pass
 
@@ -116,6 +117,11 @@ class LeftTree(CT.CustomTreeCtrl):
                  agwStyle=CT.TR_HIDE_ROOT|CT.TR_HAS_BUTTONS|CT.TR_HAS_VARIABLE_ROW_HEIGHT):
 
         CT.CustomTreeCtrl.__init__(self, parent, id, pos, size, style, agwStyle)
+        #FIXME : test for bigger font on HIDPI screen
+        font = wx.Font(pointSize=16, family=wx.DEFAULT, style=wx.NORMAL, weight=wx.FONTWEIGHT_NORMAL,
+               underline=False, face="", encoding=wx.FONTENCODING_DEFAULT)
+        self.SetFont(font)
+        ##################
         self.log = log
         alldata = dir(CT)
         treestyles = []
@@ -130,30 +136,40 @@ class LeftTree(CT.CustomTreeCtrl):
         self.events = events
         self.styles = treestyles
         self.item = None
-        
+
 
         self.il = wx.ImageList(16, 16)
         self.ild = {}
         for img in self.ira.images_analyses :
             self.ild[img] = self.il.Add(self.ira.images_analyses[img])
         self.SetImageList(self.il)
-        
+
         self.count = 0
         self.log = log
 
         self.history = parent.history
         self.h = self.history.history
+
+        idopenfolder = wx.NewId()
+        #accel_tbl = wx.AcceleratorTable([(wx.ACCEL_CTRL,  ord('E'),  idopenfolder)])
+        self.Bind(wx.EVT_MENU, self.OnOpenFolder, id=idopenfolder)
+
+        idsearchcorpus = wx.NewId()
+        accel_tbl = wx.AcceleratorTable([(wx.ACCEL_CTRL,  ord('E'),  idopenfolder), (wx.ACCEL_CTRL,  ord('F'),  idsearchcorpus)])
+        self.Bind(wx.EVT_MENU, self.OnSearchCorpus, id=idsearchcorpus)
+        self.SetAcceleratorTable(accel_tbl)
+
         self.root = self.AddRoot("Iramuteq")
-        
+
         if not(self.GetAGWWindowStyleFlag() & CT.TR_HIDE_ROOT):
             self.SetPyData(self.root, None)
             self.SetItemImage(self.root, 24, CT.TreeItemIcon_Normal)
             self.SetItemImage(self.root, 13, CT.TreeItemIcon_Expanded)
-        
+
         self.textroot = self.AppendItem(self.root, _(u'Textual corpus'))
         self.SetPyData(self.textroot, {'uuid': 'textroot'})
         self.SetItemImage(self.textroot, self.ild['textroot'], CT.TreeItemIcon_Normal)
-        self.SetItemImage(self.textroot, self.ild['textroot'], CT.TreeItemIcon_Expanded)     
+        self.SetItemImage(self.textroot, self.ild['textroot'], CT.TreeItemIcon_Expanded)
 
         for corpus in reversed(self.h) :
             child = self.AppendItem(self.textroot, corpus['corpus_name'])
@@ -176,7 +192,7 @@ class LeftTree(CT.CustomTreeCtrl):
         self.SetPyData(self.matroot, {'uuid': 'matroot'})
         self.SetItemImage(self.matroot, self.ild['matroot'], CT.TreeItemIcon_Normal)
         self.SetItemImage(self.matroot, self.ild['matroot'], CT.TreeItemIcon_Expanded)
-        
+
         orphmat = []
         for matrix in reversed(self.history.matrix) :
             if 'matrix_name' in matrix :
@@ -195,7 +211,7 @@ class LeftTree(CT.CustomTreeCtrl):
                         self.SetItemImage(last, img, CT.TreeItemIcon_Normal)
                         self.SetItemImage(last, img, CT.TreeItemIcon_Expanded)
             else :
-                orphmat.append(matrix)     
+                orphmat.append(matrix)
 
         self.Bind(wx.EVT_LEFT_DCLICK, self.OnLeftDClick)
         #self.Bind(wx.EVT_IDLE, self.OnIdle)
@@ -212,7 +228,7 @@ class LeftTree(CT.CustomTreeCtrl):
                           'EVT_TREE_SEL_CHANGING': self.OnSelChanging, "EVT_TREE_ITEM_HYPERLINK": self.OnHyperLink}
 
         mainframe = wx.GetTopLevelParent(self)
-        
+
         if not hasattr(mainframe, "leftpanel"):
             #self.Bind(CT.EVT_TREE_ITEM_EXPANDED, self.OnItemExpanded)
             #self.Bind(CT.EVT_TREE_ITEM_COLLAPSED, self.OnItemCollapsed)
@@ -236,7 +252,7 @@ class LeftTree(CT.CustomTreeCtrl):
 
         value = choice.GetValue()
         text = choice.GetLabel()
-        
+
         evt = "CT." + text
         binder = self.eventdict[text]
 
@@ -261,13 +277,12 @@ class LeftTree(CT.CustomTreeCtrl):
 
         if self.GetAGWWindowStyleFlag() != style:
             self.SetAGWWindowStyleFlag(style)
-            
+
 
     def OnCompareItems(self, item1, item2):
-        
+
         t1 = self.GetItemText(item1)
         t2 = self.GetItemText(item2)
-        
 
         if t1 < t2:
             return -1
@@ -276,7 +291,6 @@ class LeftTree(CT.CustomTreeCtrl):
 
         return 1
 
-    
     def OnIdle(self, event):
 
     #    if self.gauge:
@@ -325,7 +339,7 @@ class LeftTree(CT.CustomTreeCtrl):
         if itemParent is None :
             itemParent = self.root
         child, cookie = self.GetFirstChild(itemParent)
-        
+
         while child :
             pydata = self.GetPyData(child)
             if pydata['uuid'] == uuid :
@@ -336,7 +350,6 @@ class LeftTree(CT.CustomTreeCtrl):
 
 
     def OnRightDown(self, event):
-        
         pt = event.GetPosition()
         item, flags = self.HitTest(pt)
 
@@ -349,7 +362,7 @@ class LeftTree(CT.CustomTreeCtrl):
     def OnRightUp(self, event):
 
         item = self.item
-        
+
         if not item:
             event.Skip()
             return
@@ -383,24 +396,29 @@ class LeftTree(CT.CustomTreeCtrl):
         text = self.GetItemText(item)
         pydata = self.GetPyData(item)
         self.pydata = pydata
-        
+
         self.current = item
         self.itemdict = {"ishtml": ishtml, "back": back, "fore": fore, "isbold": isbold,
                          "font": font, "normal": normal, "selected": selected, "expanded": expanded,
                          "selexp": selexp, "haswin": haswin, "children": children,
                          "itemtype": itemtype, "text": text, "pydata": pydata, "enabled": enabled}
-        
+
         if not item in [self.textroot, self.matroot] :
             menu = wx.Menu()
             info = wx.MenuItem(menu, wx.ID_ANY, _(u"Informations").decode('utf8'))
             info.SetBitmap(wx.ArtProvider_GetBitmap(wx.ART_INFORMATION, size = (16,16)))
             menu.AppendItem(info)
-            
+
             rename = wx.MenuItem(menu, wx.ID_ANY, _(u"Rename").decode('utf8'))
             rename.SetBitmap(wx.ArtProvider_GetBitmap(wx.ART_TIP, size = (16,16)))
             menu.AppendItem(rename)
+
+            openfolder = wx.MenuItem(menu, wx.ID_ANY, _(u"Open directory").decode('utf8'))
+            openfolder.SetBitmap(wx.ArtProvider_GetBitmap(wx.ART_FOLDER_OPEN, size = (16,16)))
+            menu.AppendItem(openfolder)
+
             menu.AppendSeparator()
-    
+
             if 'corpus_name' in pydata :
                 buildmenu(menu, self.parent.text_menu)
                 menu.AppendSeparator()
@@ -411,11 +429,14 @@ class LeftTree(CT.CustomTreeCtrl):
                 openmenu = wx.Menu()
                 antipro = openmenu.Append(wx.ID_ANY, _(u"Antiprofiles").decode('utf8'))
                 menu.AppendMenu(wx.ID_ANY, _(u"Open ...").decode('utf8'), openmenu)
-    
+
+                translate = menu.Append(wx.ID_ANY, _(u"Translate Profile").decode('utf8'))
                 profsr = menu.Append(wx.ID_ANY, _(u"Repeated segments profiles").decode('utf8'))
                 profgram = menu.Append(wx.ID_ANY, _(u"POS profiles").decode('utf8'))
+                stcaract = menu.Append(wx.ID_ANY, _(u"Characteristic text segments").decode('utf8'))
                 tgen = menu.Append(wx.ID_ANY, _(u"Tgen Editor").decode('utf8'))
                 computetgen = menu.Append(wx.ID_ANY, _(u"Compute Tgen").decode('utf8'))
+                mergeclustergraph = menu.Append(wx.ID_ANY, _(u"Merge Cluster Graph").decode('utf8'))
                 export_corpus = menu.Append(wx.ID_ANY, _(u"Export corpus").decode('utf8'))
                 colored = menu.Append(wx.ID_ANY, _(u"Colored corpus").decode('utf8'))
                 navig = menu.Append(wx.ID_ANY, _(u"Navigator").decode('utf8'))
@@ -425,10 +446,13 @@ class LeftTree(CT.CustomTreeCtrl):
                 subcorpusfromcl = menu.Append(wx.ID_ANY, _(u"Sub corpus from clusters").decode('utf8'))
                 menu.AppendSeparator()
                 self.Bind(wx.EVT_MENU, self.OpenAntipro, antipro)
+                self.Bind(wx.EVT_MENU, self.OnTranslate, translate)
                 self.Bind(wx.EVT_MENU, self.OnProfSR, profsr)
                 self.Bind(wx.EVT_MENU, self.OnProfGram, profgram)
+                self.Bind(wx.EVT_MENU, self.OnStCaract, stcaract)
                 self.Bind(wx.EVT_MENU, self.OnTgenEditor, tgen)
                 self.Bind(wx.EVT_MENU, self.OnTgenCompute, computetgen)
+                self.Bind(wx.EVT_MENU, self.OnMergeClusterGraph, mergeclustergraph)
                 self.Bind(wx.EVT_MENU, self.OnExportCorpus, export_corpus)
                 self.Bind(wx.EVT_MENU, self.OnColored, colored)
                 self.Bind(wx.EVT_MENU, self.OnNavig, navig)
@@ -439,8 +463,10 @@ class LeftTree(CT.CustomTreeCtrl):
             elif pydata.get('type', False) == 'stat'  and pydata['uuid'] in self.parent.history.opened :
                 export_dictionary =  menu.Append(wx.ID_ANY, _(u"Export dictionary").decode('utf8'))
                 export_lems =  menu.Append(wx.ID_ANY, _(u"Export lemma dictionary").decode('utf8'))
+                export_cut_corpus = menu.Append(wx.ID_ANY, _(u"Export segmented corpus").decode('utf8'))
                 self.Bind(wx.EVT_MENU, self.OnExportDictionary, export_dictionary)
                 self.Bind(wx.EVT_MENU, self.OnExportLems, export_lems)
+                self.Bind(wx.EVT_MENU, self.OnExportCutCorpus, export_cut_corpus)
                 menu.AppendSeparator()
             elif pydata.get('type', False) == 'spec'  and pydata['uuid'] in self.parent.history.opened :
                 tgen = menu.Append(wx.ID_ANY, _(u"Tgen Editor").decode('utf8'))
@@ -455,13 +481,13 @@ class LeftTree(CT.CustomTreeCtrl):
                 menu.AppendMenu(wx.ID_ANY, _(u"Open ...").decode('utf8'), openmenu)
                 self.Bind(wx.EVT_MENU, self.OpenAntipro, antipro)
                 self.Bind(wx.EVT_MENU, self.OnRapport, rapport)
-    
+
             itemdelete = wx.MenuItem(menu, wx.ID_ANY, _(u"Delete from history").decode('utf8'))
             itemdelete.SetBitmap(wx.ArtProvider_GetBitmap(wx.ART_DELETE, size = (16,16)))
             menu.AppendItem(itemdelete)
             #item11 = menu.Append(wx.ID_ANY, "Prepend An Item")
             #item12 = menu.Append(wx.ID_ANY, "Append An Item")
-    
+
             #self.Bind(wx.EVT_MENU, self.OnItemBackground, item1)
             #self.Bind(wx.EVT_MENU, self.OnItemForeground, item2)
             #self.Bind(wx.EVT_MENU, self.OnItemBold, item3)
@@ -473,9 +499,10 @@ class LeftTree(CT.CustomTreeCtrl):
             self.Bind(wx.EVT_MENU, self.OnItemInfo, info)
             self.Bind(wx.EVT_MENU, self.OnRename, rename)
             self.Bind(wx.EVT_MENU, self.OnItemDelete, itemdelete)
+            self.Bind(wx.EVT_MENU, self.OnOpenFolder, openfolder)
             #self.Bind(wx.EVT_MENU, self.OnItemPrepend, item11)
             #self.Bind(wx.EVT_MENU, self.OnItemAppend, item12)
-            
+
             self.PopupMenu(menu)
             menu.Destroy()
 
@@ -495,7 +522,7 @@ class LeftTree(CT.CustomTreeCtrl):
                 corpus = Corpus(self.parent, parametres = DoConf(irapath).getoptions('corpus'), read = True)
         del busy
         return corpus
-    
+
     def getmatrix(self):
         if 'matrix_name' in self.pydata :
             matrix = Tableau(self.parent, parametres = DoConf(self.pydata['ira']).getoptions('matrix'))
@@ -512,7 +539,7 @@ class LeftTree(CT.CustomTreeCtrl):
 
     def OnStat(self, evt) :
         self.parent.OnTextStat(evt, self.getcorpus())
-        
+
     def OnReinert(self, evt) :
         self.parent.OnTextReinert(evt, self.getcorpus())
 
@@ -524,13 +551,13 @@ class LeftTree(CT.CustomTreeCtrl):
 
     def OnWordCloud(self, evt) :
         self.parent.OnWordCloud(evt, self.getcorpus())
-    
+
 #    def OnFreq(self, evt):
 #        self.parent.OnFreq(evt, self.getmatrix())
-        
+
 #    def OnChiSquare(self, evt):
 #        self.parent.OnChi2(evt, self.getmatrix())
-        
+
 #    def OnSimiTab(self, evt): 
 #        self.parent.OnSimiTab(evt, self.getmatrix())
     
@@ -555,6 +582,49 @@ class LeftTree(CT.CustomTreeCtrl):
     def OnProfGram(self, evt) :
         ProfilType(self.parent, self.page.corpus, self.page.parametres)
 
+    def OnMergeClusterGraph(self, evt) :
+        MergeClusterGraph(self.parent, self.page.corpus, self.page.parametres)
+        print 'merge done !'
+
+    def OnStCaract(self, evt) :
+        dial = PrefUCECarac(self, self.parent)
+        dial.CenterOnParent()
+        if self.page.parametres['classif_mode'] != 2 :
+            uci = False
+        else :
+            uci = True
+
+        if dial.ShowModal() == wx.ID_OK :
+            limite = dial.spin_eff.GetValue()
+            atype = dial.radio_type.GetSelection()
+            dial.Destroy()
+            corpus = self.page.corpus
+            dlg = progressbar(self.ira, maxi = len(corpus.lc))
+            dlg.Update(1, 'wait...')
+            for i in range(0, len(corpus.lc)) :
+                page = self.page.ProfNB.GetPage(i)
+                rcl = page.cl - 1
+                dlg.Update(i, u'Cluster %i' % (i+1))
+                uces = corpus.lc[rcl]
+                tab = corpus.make_table_with_classe(uces, page.la, uci = uci)
+                tab.pop(0)
+                if atype == 0 :
+                    ntab = [round(sum([page.lchi[j] for j, word in enumerate(line) if word == 1]),2) for line in tab]
+                else :
+                    ntab = [round(sum([page.lchi[j] for j, word in enumerate(line) if word == 1])/float(sum(line)),2) if sum(line)!=0 else 0 for line in tab]
+                ntab2 = [[ntab[j], uces[j]] for j, val in enumerate(ntab)]
+                del ntab
+                ntab2.sort(reverse = True)
+                ntab2 = ntab2[:limite]
+                nuces = [val[1] for val in ntab2]
+                ucis_txt, ucestxt = doconcorde(corpus, nuces, page.la, uci = uci)
+                items = ['<br>'.join([ucis_txt[j], '<table bgcolor = #1BF0F7 border=0><tr><td><b>score : %.2f</b></td></tr></table><br>' % ntab2[j][0], ucestxt[j]]) for j, uce in enumerate(nuces)]
+                filename = self.page.pathout['st_caract_cl_%i.html' % (rcl+1)]
+                with open(filename, 'w') as f :
+                    f.write('\n'.join(items))
+            dlg.Destroy()
+
+
     def OnExportCorpus(self, evt) :
         dial = PrefExport(self, self.parent)
         dial.fbb.SetValue(os.path.join(os.path.dirname(self.page.dictpathout['ira']), 'export_corpus.txt'))
@@ -572,11 +642,24 @@ class LeftTree(CT.CustomTreeCtrl):
             self.page.corpus.export_corpus_classes(dial.fbb.GetValue(), alc = alc, lem = lem, uci = uci)
             msg = u"Fini !"
             dial.Destroy()
-            dlg = wx.MessageDialog(self.parent, msg, u"Export", wx.OK | wx.NO_DEFAULT | wx.ICON_INFORMATION)
+            dlg = wx.MessageDialog(self.parent, msg, u"Export", wx.OK | wx.ICON_INFORMATION)
             dlg.CenterOnParent()
             dlg.ShowModal()
             dlg.Destroy()
 
+    def OnExportCutCorpus(self, evt) :
+        uci = False
+        fileout = os.path.join(os.path.dirname(self.page.pathout['ira']), 'segmented_corpus.txt')
+        txt = self.page.corpus.make_cut_corpus(uci = uci)
+        with open(fileout, 'w') as f :
+            f.write(txt)
+        msg = '\n'.join([_(u"Done !").decode('utf8'), fileout])
+        dlg = wx.MessageDialog(self.parent, msg, _(u"Segmented corpus"), wx.OK | wx.ICON_INFORMATION)
+        dlg.CenterOnParent()
+        dlg.ShowModal()
+        dlg.Destroy()
+
+
     def OnColored(self, evt) :
         dial = PrefSimpleFile(self, self.parent, **{'mask' : '*.html', 'title': _(u"Colored corpus").decode('utf8')})
         dial.fbb.SetValue(os.path.join(os.path.dirname(self.page.dictpathout['ira']), 'corpus_couleur.html'))
@@ -593,7 +676,7 @@ class LeftTree(CT.CustomTreeCtrl):
             with open(fileout, 'w') as f :
                 f.write(txt)
             msg = ' !\n'.join([_(u"Done").decode('utf8'), _(u"Open in a web browser ?").decode('utf8')])
-            dlg = wx.MessageDialog(self.parent, msg, u"Corpus en couleur", wx.NO | wx.YES | wx.NO_DEFAULT | wx.ICON_QUESTION)
+            dlg = wx.MessageDialog(self.parent, msg, u"Corpus en couleur", wx.NO | wx.YES | wx.ICON_QUESTION)
             dlg.CenterOnParent()
             if dlg.ShowModal() == wx.ID_YES :
                 webbrowser.open(fileout)
@@ -614,10 +697,10 @@ class LeftTree(CT.CustomTreeCtrl):
             dial.Destroy()
             self.page.corpus.get_stat_by_cluster(fileout)
             msg = u"Fini !"
-            dlg = wx.MessageDialog(self.parent, msg, _(u"Clusters statistics").decode('utf8'), wx.OK | wx.NO_DEFAULT | wx.ICON_INFORMATION)
+            dlg = wx.MessageDialog(self.parent, msg, _(u"Clusters statistics").decode('utf8'), wx.OK | wx.ICON_INFORMATION)
             dlg.CenterOnParent()
             if dlg.ShowModal() == wx.ID_OK :
-                dlg.Destroy()        
+                dlg.Destroy()
 
     def OpenAntipro(self, evt) :
         find = False
@@ -631,6 +714,25 @@ class LeftTree(CT.CustomTreeCtrl):
             open_antiprofil(self.page, self.page.dictpathout['ANTIPRO_OUT'], self.parent.syscoding)
             self.page.TabChdSim.SetSelection(self.page.TabChdSim.GetPageCount() - 1)
 
+    def OnTranslate(self, evt) :
+        dial = translate_dialog(self.parent)
+        dial.CenterOnParent()
+        res = dial.ShowModal()
+        if res == wx.ID_OK :
+            to_l = dial.getto_l()
+            from_l = dial.getfrom_l()
+            dial.Destroy()
+            busy = wx.BusyInfo(_("Please wait...").decode('utf8'), self.parent)
+            wx.SafeYield()
+            prof, lems = translateprofile(self.page.corpus, ReadProfileAsDico(self.page.dictpathout['PROFILE_OUT'], True, self.parent.syscoding), lf=from_l, lt=to_l)
+            write_translation_profile(prof, lems, to_l, self.page.dictpathout)
+            open_antiprofil(self.page, prof, 'utf8', title = _(u'Translation').decode('utf8') + ' %s' % to_l, translation = True, lems=lems)
+            del busy
+            self.page.lems = lems
+            self.page.TabChdSim.SetSelection(self.page.TabChdSim.GetPageCount() - 1)
+        else :
+            dial.Destroy()
+
     def OnRapport(self, evt) :
         dial = PrefSimpleFile(self, self.parent, **{'mask' : '*.txt', 'title': _(u"Report").decode('utf8')})
         dial.fbb.SetValue(self.page.dictpathout['rapport'])
@@ -642,7 +744,7 @@ class LeftTree(CT.CustomTreeCtrl):
             with open(fileout, 'w') as f :
                 f.write(self.page.debtext + '\n' + GetTxtProfile(self.page.DictProfile, self.page.cluster_size))
             msg = u"Fini !"
-            dlg = wx.MessageDialog(self.parent, msg, _(u"Report").decode('utf8'), wx.OK | wx.NO_DEFAULT | wx.ICON_INFORMATION)
+            dlg = wx.MessageDialog(self.parent, msg, _(u"Report").decode('utf8'), wx.OK | wx.ICON_INFORMATION)
             dlg.CenterOnParent()
             dlg.ShowModal()
             dlg.Destroy()
@@ -686,7 +788,7 @@ class LeftTree(CT.CustomTreeCtrl):
         corpus = self.page.corpus
         tgenpath = os.path.join(self.page.parametres['pathout'], 'tgen.csv')        
         if not os.path.exists(tgenpath) :
-            message = wx.MessageDialog(self.parent, _(u"No TGen yet !"), style = wx.ICON_EXCLAMATION | wx.OK) 
+            message = wx.MessageDialog(self.parent, _(u"No TGen yet !").decode('utf8'), style = wx.ICON_EXCLAMATION | wx.OK) 
             message.ShowModal()
             message.Destroy()
         else :
@@ -726,7 +828,7 @@ class LeftTree(CT.CustomTreeCtrl):
         else :
             name = 'name'
         oldname = pydata[name]
-        dlg = wx.TextEntryDialog(self, "Please Enter The New Item Name", 'Item Naming', oldname)
+        dlg = wx.TextEntryDialog(self, _("New Name").decode('utf8'), _(u'Rename').decode('utf8'), oldname)
         if dlg.ShowModal() == wx.ID_OK:
             newname = dlg.GetValue()
             dlg.Destroy()
@@ -738,6 +840,36 @@ class LeftTree(CT.CustomTreeCtrl):
             self.history.update(pydata)
             self.SetItemText(self.current, newname)
             self.EnsureVisible(self.current)
+    
+    def OnOpenFolder(self, evt):
+        try :
+            open_folder(os.path.dirname(self.pydata['ira']))
+        except :
+            print 'cannot open folder %s' % self.pydata.get('ira', 'noirapath')
+    
+    def GetCorpusByName(self, corpus_name) :
+        return [corpus for corpus in self.h if corpus_name in corpus['corpus_name']]
+    
+    def OnSearchCorpus(self, evt):
+        searchframe = SearchCorpus(self.ira, self, None, None)
+        searchframe.Show()
+        #res = self.GetCorpusByName('Ministres')
+        #print res
+        
+    def SetContentBackground(self, itemParent = None, uuid = None, color = True) :
+        if itemParent is None :
+            itemParent = self.root
+        child, cookie = self.GetFirstChild(itemParent)
+        while child :
+            pydata = self.GetPyData(child)
+            if pydata['uuid'] == uuid :
+                self.SelectItem(child)
+                if color :
+                    self.SetItemBackgroundColour(child, wx.Colour(15,234,56))
+                return
+            self.SetContentBackground(child, uuid, color)
+            child, cookie = self.GetNextChild(itemParent, cookie)
+         
 
     def OnItemBackground(self, event):
 
@@ -1093,11 +1225,15 @@ class LeftTree(CT.CustomTreeCtrl):
 
         if pydata is not None :
             if 'corpus_name' in pydata or 'corpus' in pydata :
-                self.ira.ShowMenu('text', True)
                 self.ira.ShowMenu('matrix', False)
+                self.ira.ShowMenu('text', True)
             if 'matrix_name' in pydata or 'matrix' in pydata:
                 self.ira.ShowMenu('text', False)
-                self.ira.ShowMenu('matrix', True)                
+                self.ira.ShowMenu('matrix', True)
+            if 'uuid' in pydata :
+                if pydata['uuid'] in ['textroot', 'matroot'] :
+                    self.ira.ShowMenu('text', False)
+                    self.ira.ShowMenu('matrix', False)             
             self.pydata = pydata
             if pydata['uuid'] in self.parent.history.opened :
                 for i in range(self.parent.nb.GetPageCount()) :