...
[iramuteq] / ProfList.py
1 # -*- coding: utf-8 -*-
2
3 #----------------------------------------------------------------------------
4 # Name:         ListCtrl.py
5 # Author:       Pierre Ratinaud
6
7
8 #comes from ListCtrl.py from the demo tool of wxPython:
9 # Author:       Robin Dunn & Gary Dumer
10 #
11 # Created:
12 # Copyright:    (c) 1998 by Total Control Software
13 # Licence:      wxWindows license
14 #----------------------------------------------------------------------------
15
16 import os
17 import  wx
18 import  wx.lib.mixins.listctrl  as  listmix
19 from listlex import ListForSpec
20 from chemins import ConstructPathOut, ffr
21 from dialog import PrefUCECarac, SearchDial, message, BarFrame, ChronoFrame
22 from tableau import copymatrix
23 from search_tools import SearchFrame
24 import webbrowser
25 #import cStringIO
26 import tempfile
27 import codecs
28 from functions import progressbar, treat_var_mod, doconcorde
29 from operator import itemgetter
30 from copy import copy
31
32 #---------------------------------------------------------------------------
33 class ProfListctrlPanel(wx.ListCtrl, listmix.ListCtrlAutoWidthMixin, listmix.ColumnSorterMixin):
34     def __init__(self, parent, gparent, profclasse, Alceste=False, cl=0, translation = False):
35         wx.ListCtrl.__init__( self, parent, -1, style=wx.LC_REPORT|wx.LC_VIRTUAL|wx.LC_HRULES|wx.LC_VRULES)
36
37         self.parent = parent
38         self.Alceste = Alceste
39         self.Source = gparent
40         self.translation = translation
41         self.cl = cl
42         self.var_mod = {}
43         self.them_mod = {}
44         self.ira = wx.GetApp().GetTopWindow()
45
46         line1 = profclasse.pop(0)
47         classen = [line for line in profclasse if line[0] != '*' and line[0] != '*****']
48         try :
49             self.lenact = profclasse.index([u'*****', u'*', u'*', u'*', u'*', u'*', '', ''])
50             profclasse.pop(self.lenact)
51         except ValueError:
52             try :
53                 self.lenact = profclasse.index([u'*', u'*', u'*', u'*', u'*', u'*', '', ''])
54                 profclasse.pop(self.lenact)
55             except ValueError:
56                 self.lenact = len(profclasse)
57         try :
58             self.lensup = profclasse.index([u'*', u'*', u'*', u'*', u'*', u'*', '', ''])
59             self.lensup = self.lensup - self.lenact
60             profclasse.pop(self.lensup)
61         except ValueError: 
62             self.lensup = len(profclasse) - self.lenact
63         self.lenet = len(profclasse) - (self.lenact + self.lensup)
64 #        print self.lenact, self.lensup, self.lenet
65         for i,  line in enumerate(classen) :
66             line[0] = i
67         dictdata = dict(zip([i for i in range(0,len(classen))], classen))
68
69         if self.lenact != 0 :
70             self.la = [dictdata[i][6] for i in range(0, self.lenact)]
71             self.lchi = [dictdata[i][4] for i in range(0, self.lenact)]
72             self.lfreq = [dictdata[i][1] for i in range(0, self.lenact)]
73         else :
74             self.la = []
75             self.lchi = []
76             self.lfreq = []
77         self.tmpchi = None
78             
79         #adding some art
80         self.il = wx.ImageList(16, 16)
81         a={"sm_up":"GO_UP","sm_dn":"GO_DOWN","w_idx":"WARNING","e_idx":"ERROR","i_idx":"QUESTION"}
82         for k,v in a.items():
83             s="self.%s= self.il.Add(wx.ArtProvider_GetBitmap(wx.ART_%s,wx.ART_TOOLBAR,(16,16)))" % (k,v)
84             exec(s)
85         self.SetImageList(self.il, wx.IMAGE_LIST_SMALL)
86
87         #adding some attributes (colourful background for each item rows)
88         self.attr1 = wx.ListItemAttr()
89         self.attr1.SetBackgroundColour((220, 220, 220))
90         self.attrsg = wx.ListItemAttr()
91         self.attrsg.SetBackgroundColour((230, 230, 230))
92         self.attr2 = wx.ListItemAttr()
93         self.attr2.SetBackgroundColour((190, 249, 236))
94         self.attr2s = wx.ListItemAttr()
95         self.attr2s.SetBackgroundColour((211, 252, 244))
96         self.attr3 = wx.ListItemAttr()
97         self.attr3.SetBackgroundColour((245, 180, 180))
98         self.attr3s = wx.ListItemAttr()
99         self.attr3s.SetBackgroundColour((245, 190, 190))
100
101
102         self.InsertColumn(0, "num", wx.LIST_FORMAT_RIGHT)
103         self.InsertColumn(1, "eff. s.t.", wx.LIST_FORMAT_RIGHT)
104         self.InsertColumn(2, "eff. total", wx.LIST_FORMAT_RIGHT)
105         self.InsertColumn(3, "pourcentage", wx.LIST_FORMAT_RIGHT)
106         self.InsertColumn(4, "chi2", wx.LIST_FORMAT_RIGHT)
107         self.InsertColumn(5, "Type", wx.LIST_FORMAT_RIGHT)
108         self.InsertColumn(6, "forme", wx.LIST_FORMAT_RIGHT)
109         self.InsertColumn(7, "p", wx.LIST_FORMAT_RIGHT)
110
111
112         self.SetColumnWidth(0, 60)
113         self.SetColumnWidth(1, 70)
114         self.SetColumnWidth(2, 80)
115         self.SetColumnWidth(3, 100)
116         self.SetColumnWidth(4, 70)
117         self.SetColumnWidth(5, 60)
118         self.SetColumnWidth(6, 140)
119         self.SetColumnWidth(7, wx.LIST_AUTOSIZE)
120
121         #These two should probably be passed to init more cleanly
122         #setting the numbers of items = number of elements in the dictionary
123         self.itemDataMap = dictdata
124         self.itemIndexMap = dictdata.keys()
125         self.SetItemCount(len(dictdata))
126
127         #mixins
128         listmix.ListCtrlAutoWidthMixin.__init__(self)
129         listmix.ColumnSorterMixin.__init__(self, len(classen[0]))
130
131         #sort by genre (column 2), A->Z ascending order (1)
132         self.SortListItems(0, 1)
133
134         #events
135         #self.Bind(wx.EVT_LIST_ITEM_SELECTED, self.OnItemSelected)
136         self.Bind(wx.EVT_LIST_ITEM_ACTIVATED, self.OnPopupTwo, self)
137         #self.Bind(wx.EVT_LIST_ITEM_DESELECTED, self.OnItemDeselected)
138         self.Bind(wx.EVT_LIST_COL_CLICK, self.OnColClick)
139
140         # for wxMSW
141         self.Bind(wx.EVT_COMMAND_RIGHT_CLICK, self.OnRightClick)
142
143         # for wxGTK
144         self.Bind(wx.EVT_RIGHT_UP, self.OnRightClick)
145
146         #for searching
147         search_id = wx.NewId()
148         searchall_id = wx.NewId()
149         concord_id = wx.NewId()
150         self.parent.Bind(wx.EVT_MENU, self.onsearch, id = search_id)
151         self.parent.Bind(wx.EVT_MENU, self.onsearchall, id = searchall_id)
152         self.accel_tbl = wx.AcceleratorTable([(wx.ACCEL_CTRL, ord('F'), search_id),
153                                               (wx.ACCEL_CTRL|wx.ACCEL_SHIFT, ord('F'), searchall_id)])
154         self.SetAcceleratorTable(self.accel_tbl)
155
156
157
158     def OnColClick(self,event):
159         event.Skip()
160
161     def OnItemSelected(self, event):
162         self.currentItem = event.m_itemIndex
163
164     def OnItemActivated(self, event):
165         self.currentItem = event.m_itemIndex
166
167     def getColumnText(self, index, col):
168         item = self.GetItem(index, col)
169         return item.GetText()
170
171     def OnItemDeselected(self, evt):
172         pass
173     #---------------------------------------------------
174     # These methods are callbacks for implementing the
175     # "virtualness" of the list...
176
177     def OnGetItemText(self, item, col):
178         index=self.itemIndexMap[item]
179         s = self.itemDataMap[index][col]
180         return s
181
182     def OnGetItemImage(self, item):
183         index=self.itemIndexMap[item]
184         genre=self.itemDataMap[index][2]
185
186         if genre=="Rock":
187             return self.w_idx
188         elif genre=="Jazz":
189             return self.e_idx
190         elif genre=="New Age":
191             return self.i_idx
192         else:
193             return -1
194
195     def OnGetItemAttr(self, item):
196         index=self.itemIndexMap[item]
197         if index < self.lenact :
198             if item % 2 :
199                 return self.attr1
200             else :
201                 return self.attrsg
202         elif index >= self.lenact and index < (self.lenact + self.lensup) :
203             if item % 2 :
204                 return self.attr2
205             else :
206                 return self.attr2s
207         elif index >= (self.lenact + self.lensup) :
208             if item % 2 :
209                 return self.attr3
210             else :
211                 return self.attr3s
212         else :
213             return None
214
215     #---------------------------------------------------
216     # Matt C, 2006/02/22
217     # Here's a better SortItems() method --
218     # the ColumnSorterMixin.__ColumnSorter() method already handles the ascending/descending,
219     # and it knows to sort on another column if the chosen columns have the same value.
220
221     def SortItems(self,sorter=cmp):
222         items = list(self.itemDataMap.keys())
223         items.sort(sorter)
224         self.itemIndexMap = items
225
226         # redraw the list
227         self.Refresh()
228
229     # Used by the ColumnSorterMixin, see wx/lib/mixins/listctrl.py
230     def GetListCtrl(self):
231         return self
232
233     # Used by the ColumnSorterMixin, see wx/lib/mixins/listctrl.py
234     def GetSortImages(self):
235         return (self.sm_dn, self.sm_up)
236
237     def onsearch(self, evt) :
238         self.dial = SearchDial(self, self, 6, True)
239         self.dial.CenterOnParent()
240         self.dial.Show()
241         #self.dial.Destroy()
242
243     def onsearchall(self, evt) :
244         if 'FrameSearch' not in dir(self.Source) :
245             self.Source.FrameSearch = SearchFrame(self.parent, -1, _(u"Search...").decode('utf8'), self.Source.corpus)
246         self.dial = SearchDial(self, self.Source.FrameSearch.liste, 1, False)
247         self.dial.CenterOnParent()
248         self.dial.Show()
249         #self.dial.Destroy()
250
251     def OnRightClick(self, event):
252
253         # only do this part the first time so the events are only bound once
254         if self.Alceste:
255             if not hasattr(self, "popupID1"):
256                 self.popupID1 = wx.NewId()
257                 self.popupID2 = wx.NewId()
258                 self.popupID3 = wx.NewId()
259                 self.popupID4 = wx.NewId()
260                 self.popupID5 = wx.NewId()
261                 self.popupID6 = wx.NewId()
262                 self.popupID7 = wx.NewId()
263                 self.popupID8 = wx.NewId()
264                 self.popupID9 = wx.NewId()
265                 #self.popupID10 = wx.NewId()
266                 self.popupIDgraph = wx.NewId()
267                 self.idseg = wx.NewId()
268                 self.iducecarac = wx.NewId()
269                 self.idtablex = wx.NewId()
270                 self.idchimod = wx.NewId()
271                 self.idwordgraph = wx.NewId()
272                 self.popup_proxe = wx.NewId()
273                 self.idlexdendro = wx.NewId()
274                 self.idcloud = wx.NewId()
275                 self.idexport = wx.NewId()
276                 self.idexporttropes = wx.NewId()
277                 self.idexportowledge = wx.NewId()
278                 self.onmaketgen = wx.NewId()
279                 self.onchronochi2 = wx.NewId()
280                 self.onchronoprop = wx.NewId()
281             #    self.export_classes = wx.NewId()
282    
283                 self.Bind(wx.EVT_MENU, self.OnPopupOne, id=self.popupID1)
284                 self.Bind(wx.EVT_MENU, self.OnPopupTwo, id=self.popupID2)
285                 self.Bind(wx.EVT_MENU, self.OnPopupThree, id=self.popupID3)
286                 self.Bind(wx.EVT_MENU, self.OnPopupFour, id=self.popupID4)
287                 self.Bind(wx.EVT_MENU, self.OnPopupFive, id=self.popupID5)
288                 self.Bind(wx.EVT_MENU, self.OnPopupSix, id=self.popupID6)
289                 self.Bind(wx.EVT_MENU, self.OnPopupSeven, id=self.popupID7)
290                 self.Bind(wx.EVT_MENU, self.OnPopupHeight, id=self.popupID8)
291                 self.Bind(wx.EVT_MENU, self.OnPopupNine, id=self.popupID9)
292                 #self.Bind(wx.EVT_MENU, self.OnPopupSpec, id=self.popupID10)
293                 self.Bind(wx.EVT_MENU, self.on_graph, id=self.popupIDgraph)
294                 self.Bind(wx.EVT_MENU, self.on_segments, id=self.idseg)
295                 self.Bind(wx.EVT_MENU, self.on_uce_carac, id = self.iducecarac)
296                 self.Bind(wx.EVT_MENU, self.on_tablex, id = self.idtablex)
297                 self.Bind(wx.EVT_MENU, self.quest_var_mod, id = self.idchimod)
298                 self.Bind(wx.EVT_MENU, self.onwordgraph, id = self.idwordgraph)
299                 self.Bind(wx.EVT_MENU, self.onproxe, id = self.popup_proxe)
300                 self.Bind(wx.EVT_MENU, self.onlexdendro, id = self.idlexdendro)
301                 self.Bind(wx.EVT_MENU, self.oncloud, id = self.idcloud)
302                 self.Bind(wx.EVT_MENU, self.onexport, id = self.idexport)
303                 self.Bind(wx.EVT_MENU, self.onexporttropes, id = self.idexporttropes)
304                 self.Bind(wx.EVT_MENU, self.onexportowledge, id = self.idexportowledge)
305                 self.Bind(wx.EVT_MENU, self.OnMakeTgen, id=self.onmaketgen)
306                 self.Bind(wx.EVT_MENU, self.OnChronoChi2, id=self.onchronochi2)
307                 self.Bind(wx.EVT_MENU, self.OnChronoProp, id=self.onchronoprop)
308              #  self.Bind(wx.EVT_MENU, self.on_export_classes, id = self.export_classes)
309    #            self.Bind(wx.EVT_MENU, self.OnPopupThree, id=self.popupID3)
310     
311             # make a menu
312             menu = wx.Menu()
313             menu.Append(self.popupID1, _(u"Associated forms").decode('utf8'))
314             menu.Append(self.idtablex, _(u"Chi2 by cluster").decode('utf8'))
315             menu.Append(self.idlexdendro, _(u"Chi2 by cluster on dendrogram").decode('utf8'))
316             menu.Append(self.idchimod, _(u"Chi2 modalities of variable").decode('utf8'))
317             menu_chrono = wx.Menu()
318             menu_chrono.Append(self.onchronochi2, _(u'Chi2').decode('utf8'))
319             menu_chrono.Append(self.onchronoprop, _(u'Proportion').decode('utf8'))
320             menu.AppendMenu(-1, _(u"Chronological view").decode('utf8'), menu_chrono)
321             menu.Append(self.idwordgraph, _(u"Word graph").decode('utf8'))
322             #menu.Append(self.export_classes, u"Exporter le corpus...") 
323
324             #menu.Append(self.popupID10, u"Spécificités")
325
326             menu_conc = wx.Menu()
327             menu_conc.Append(self.popupID2, _(u"In segments of this cluster").decode('utf8'))
328             menu_conc.Append(self.popupID3, _(u"In segments of this clustering").decode('utf8'))
329             menu_conc.Append(self.popupID4, _(u"In all segments").decode('utf8'))
330             menu.AppendMenu(-1, _(u"Concordance").decode('utf8'), menu_conc)
331             menu.Append(self.onmaketgen, _(u"Make Tgen").decode('utf8'))
332             menu_cnrtl = wx.Menu()
333             menu_cnrtl.Append(self.popupID5, _(u"Definition").decode('utf8'))
334             menu_cnrtl.Append(self.popupID6, _(u"Etymology").decode('utf8'))
335             menu_cnrtl.Append(self.popupID7, _(u"Synonymous").decode('utf8'))
336             menu_cnrtl.Append(self.popupID8, _(u"Antonym").decode('utf8'))
337             menu_cnrtl.Append(self.popupID9, _(u"Morphology").decode('utf8'))
338             menu_cnrtl.Append(self.popup_proxe, _(u"Proxemy").decode('utf8'))
339             menu.AppendMenu(-1, _(u"Tools from CNRTL (french only)").decode('utf8'), menu_cnrtl)
340             menu.AppendSeparator()
341             menu.Append(self.popupIDgraph, _(u"Graph of cluster").decode('utf8'))
342             menu.Append(self.idseg, _(u"Repeated segments").decode('utf8'))
343             menu.Append(self.iducecarac, _(u"Typical text segments").decode('utf8'))
344             menu.Append(self.idcloud, _(u"Word cloud of cluster").decode('utf8'))
345             menu.Append(self.idexport, _(u'Export...').decode('utf8'))
346             menu.Append(self.idexporttropes, _(u'Export for Tropes').decode('utf8'))
347             menu.Append(self.idexportowledge, _('Exporter for Owledge').decode('utf8'))
348             #menu.Append(self.popupID2, u"Concordancier")
349     #        menu.Append(self.popupID3, "recharger")
350
351             self.PopupMenu(menu)
352             menu.Destroy()
353         elif 'tableau' in dir(self.Source) :
354             if not hasattr(self, "pop1"):
355                 self.pop1 = wx.NewId()
356                 self.pop2 = wx.NewId()
357                 self.pop3 = wx.NewId()
358                 self.Bind(wx.EVT_MENU, self.quest_simi, id=self.pop1)
359                 self.Bind(wx.EVT_MENU, self.on_tablex, id=self.pop2)
360                 self.Bind(wx.EVT_MENU, self.quest_var_mod, id=self.pop3)
361
362             menu = wx.Menu()
363             menu.Append(self.pop2, _(u"Chi2 by cluster").decode('utf8'))
364             menu.Append(self.pop3, _(u"Chi2 modalities of variable").decode('utf8'))
365             menu.AppendSeparator()
366             menu.Append(self.pop1, _(u"Graph of cluster").decode('utf8'))
367             self.PopupMenu(menu)
368             menu.Destroy()
369
370     def oncloud(self, evt) :
371         if 'corpus' in dir(self.Source):
372             corpus = self.Source.corpus
373         prof = [[self.la[i], self.lchi[i], self.lfreq[i]] for i, val in enumerate(self.la)]
374         parametres = copy(self.Source.parametres)
375         parametres['clusterprof'] = prof
376         parametres['type'] = 'clustercloud'
377         parametres['prof'] = self.Source.pathout['actprof_classe_%i.csv' % self.cl]
378         del  parametres['uuid']
379         self.parent.OnClusterCloud(self.Source.corpus, parametres = parametres)
380
381     def onexport(self, evt) :
382         if 'corpus' in dir(self.Source):
383             corpus = self.Source.corpus
384         if self.Source.parametres['classif_mode'] != 2 :
385             uci = False
386         else :
387             uci = True
388         corpus.export_classe(self.Source.pathout['classe_%i_export.txt' % self.cl], self.cl, uci = uci)
389         dial = wx.MessageDialog(self, self.Source.pathout['classe_%i_export.txt' % self.cl], u"Export", wx.OK|wx.ICON_INFORMATION)
390         dial.ShowModal()
391         dial.Destroy()
392
393     def onexporttropes(self, evt) :
394         if 'corpus' in dir(self.Source):
395             corpus = self.Source.corpus
396         if self.Source.parametres['classif_mode'] != 2 :
397             uci = False
398         else :
399             uci = True
400         fileout = self.Source.pathout['export_tropes_classe_%i.txt' % self.cl]
401         corpus.export_tropes(fileout, self.cl, uci = uci)
402
403     def onexportowledge(self, evt):
404         if 'corpus' in dir(self.Source):
405             corpus = self.Source.corpus
406         if self.Source.parametres['classif_mode'] != 2 :
407             uci = False
408         else :
409             uci = True
410         repout = self.Source.pathout['export_owledge_classe_%i' % self.cl]
411         if not os.path.exists(repout) :
412             os.mkdir(repout)
413         corpus.export_owledge(repout, self.cl, uci = uci)
414
415     def getselectedwords(self) :
416         words = [self.getColumnText(self.GetFirstSelected(), 6)]
417         last = self.GetFirstSelected()
418         while self.GetNextSelected(last) != -1:
419             last = self.GetNextSelected(last)
420             words.append(self.getColumnText(last, 6))
421         return words
422
423     def quest_var_mod(self, evt) :
424         word = self.getselectedwords()[0]
425         if len(word.split('_')) <= 1 :
426             dial = wx.MessageDialog(self, _(u"This is not a variable_modality form").decode('utf8'), _(u"Problem").decode('utf8'), wx.OK | wx.ICON_WARNING)
427             dial.CenterOnParent()
428             dial.ShowModal()
429             dial.Destroy()
430             return
431
432         if 'corpus' in dir(self.Source):
433             corpus = self.Source.corpus
434             if word.startswith(u'-*') :
435                 if self.them_mod == {} :
436                     self.them_mod = self.Source.corpus.make_theme_dict()
437                 var_mod = self.them_mod
438             else :
439                 if self.var_mod == {} :
440                     self.var_mod = self.Source.corpus.make_etoiles_dict()
441                 var_mod = self.var_mod
442         else :
443             corpus = self.Source.tableau
444             if self.var_mod == {} :
445                 self.var_mod = treat_var_mod([val for val in corpus.actives] + [val for val in corpus.sups])
446             var_mod = self.var_mod
447         with codecs.open(self.Source.pathout['chisqtable'], 'r', corpus.parametres['syscoding']) as f :
448             chistable = [line.replace('\n','').replace('\r','').replace('"','').replace(',','.').split(';') for line in f]
449         title = chistable[0]
450         title.pop(0)
451         chistable.pop(0)
452         vchistable = [line[1:] for line in chistable]
453         fchistable = [line[0] for line in chistable]
454
455         var = word.split('_')
456         #words = ['_'.join([var[0],word]) for word in self.var_mod[var[0]]]
457         try :
458             words = [word for word in var_mod[var[0]]]
459         except KeyError:
460             dial = wx.MessageDialog(self, _(u"This is not a meta-data").decode('utf8'), _(u"Problem").decode('utf8'), wx.OK | wx.ICON_WARNING)
461             dial.CenterOnParent()
462             dial.ShowModal()
463             dial.Destroy()
464             return
465         words.sort()
466         tableout = []
467         kwords = []
468         for word in words :
469             if word in fchistable :
470                 tableout.append(vchistable[fchistable.index(word)])
471                 kwords.append(word)
472         BarFrame(self.Source.parent, tableout, title, kwords)
473
474     def OnChronoChi2(self, evt) :
475         word = self.getselectedwords()[0]
476         if len(word.split('_')) <= 1 :
477             dial = wx.MessageDialog(self, _(u"This is not a variable_modality form").decode('utf8'), _(u"Problem").decode('utf8'), wx.OK | wx.ICON_WARNING)
478             dial.CenterOnParent()
479             dial.ShowModal()
480             dial.Destroy()
481             return
482
483         if 'corpus' in dir(self.Source):
484             corpus = self.Source.corpus
485             if word.startswith(u'-*') :
486                 if self.them_mod == {} :
487                     self.them_mod = self.Source.corpus.make_theme_dict()
488                 var_mod = self.them_mod
489             else :
490                 if self.var_mod == {} :
491                     self.var_mod = self.Source.corpus.make_etoiles_dict()
492                 var_mod = self.var_mod
493         else :
494             corpus = self.Source.tableau
495             if self.var_mod == {} :
496                 self.var_mod = treat_var_mod([val for val in corpus.actives] + [val for val in corpus.sups])
497             var_mod = self.var_mod
498         var = word.split('_')
499         #words = ['_'.join([var[0],word]) for word in self.var_mod[var[0]]]
500         try :
501             words = [word for word in var_mod[var[0]]]
502         except KeyError:
503             dial = wx.MessageDialog(self, _(u"This is not a meta-data").decode('utf8'), _(u"Problem").decode('utf8'), wx.OK | wx.ICON_WARNING)
504             dial.CenterOnParent()
505             dial.ShowModal()
506             dial.Destroy()
507             return
508         words.sort()
509         vartoplot = var[0] + '_'
510         parametres = {'var' : vartoplot}
511         ChronoFrame(self.Source.parent, parametres, self.Source.pathout, which = 'chi2')
512
513     def OnChronoProp(self, evt) :
514         word = self.getselectedwords()[0]
515         if len(word.split('_')) <= 1 :
516             dial = wx.MessageDialog(self, _(u"This is not a variable_modality form").decode('utf8'), _(u"Problem").decode('utf8'), wx.OK | wx.ICON_WARNING)
517             dial.CenterOnParent()
518             dial.ShowModal()
519             dial.Destroy()
520             return
521
522         if 'corpus' in dir(self.Source):
523             corpus = self.Source.corpus
524             if word.startswith(u'-*') :
525                 if self.them_mod == {} :
526                     self.them_mod = self.Source.corpus.make_theme_dict()
527                 var_mod = self.them_mod
528             else :
529                 if self.var_mod == {} :
530                     self.var_mod = self.Source.corpus.make_etoiles_dict()
531                 var_mod = self.var_mod
532         else :
533             corpus = self.Source.tableau
534             if self.var_mod == {} :
535                 self.var_mod = treat_var_mod([val for val in corpus.actives] + [val for val in corpus.sups])
536             var_mod = self.var_mod
537         var = word.split('_')
538         #words = ['_'.join([var[0],word]) for word in self.var_mod[var[0]]]
539         try :
540             words = [word for word in var_mod[var[0]]]
541         except KeyError:
542             dial = wx.MessageDialog(self, _(u"This is not a meta-data").decode('utf8'), _(u"Problem").decode('utf8'), wx.OK | wx.ICON_WARNING)
543             dial.CenterOnParent()
544             dial.ShowModal()
545             dial.Destroy()
546             return
547         words.sort()
548         vartoplot = var[0] + '_'
549         parametres = {'var' : vartoplot}
550         ChronoFrame(self.Source.parent, parametres, self.Source.pathout, which = 'prop')
551
552
553     def quest_simi(self, evt) :
554         tableau = self.Source.tableau
555         tab = tableau.make_table_from_classe(self.cl, self.la)
556         pathout = ConstructPathOut(self.Source.pathout.dirout, 'simi_classe_%i' %self.cl)
557         if self.tmpchi is None :
558             self.tmpchi = os.path.join(pathout,'chi.csv')
559             with open(self.tmpchi, 'w') as f:
560                 f.write('\n'.join([str(val) for val in self.lchi]))
561         self.filename = os.path.join(pathout,'mat01.csv')
562         tableau.printtable(self.filename, tab)
563         del tab
564         paramsimi = {'coeff' : 0,
565                           'layout' : 2,
566                           'type_graph' : 1,
567                           'arbremax' : 1,
568                           'coeff_tv' : 1,
569                           'coeff_tv_nb' : 0,
570                           'tvprop' : 0,
571                           'tvmin' : 5,
572                           'tvmax' : 30,
573                           'coeff_te' : 1,
574                           'coeff_temin' : 1,
575                           'coeff_temax' : 10,
576                           'label_v': 1,
577                           'label_e': 1,
578                           'vcex' : 0,
579                           'vcexmin' : 10,
580                           'vcexmax' : 25,
581                           'cex' : 10,
582                           'cexfromchi' : True,
583                           'sfromchi': False,
584                           'seuil_ok' : 0,
585                           'seuil' : 1,
586                           'cols' : (255,0,0),
587                           'cola' : (200,200,200),
588                           'width' : 1000,
589                           'height' : 1000,
590                           'first' : True,
591                           'keep_coord' : True,
592                           'alpha' : 20,
593                           'film': False,
594                           'com' : 0,
595                           'communities' : 0,
596                           'halo' : 0,
597                           'tmpchi': self.tmpchi,
598                           'fromprof' : True,
599                           'edgecurved' : True,
600                           }
601         act = {}
602         tableau = copymatrix(tableau)
603         tableau.chi = {}
604         tableau.lchi = self.lchi
605         #tableau.parametres['fromprof'] = True
606         for i, val in enumerate(self.la) :
607             act[val] = [self.lfreq[i]]
608             tableau.chi[val] = [self.lchi[i]]
609         paramsimi['listactives'] = copy(self.la)
610         paramsimi['actives'] = copy(act)
611         paramsimi['pathout'] = pathout
612         self.parent.SimiCluster(parametres = paramsimi, fromprof = ffr(self.filename), tableau = tableau)
613
614     def onwordgraph(self, evt):
615         word = self.getColumnText(self.GetFirstSelected(), 6)
616         if self.tmpchi is None :
617             self.tmpchi = os.path.join(self.Source.parametres['pathout'],'chi_%i.csv' % self.cl)
618             with open(self.tmpchi, 'w') as f:
619                 f.write('\n'.join([str(val) for val in self.lchi]))
620         index = self.la.index(word)
621         parametres = {'type' : 'clustersimitxt',
622                         'pathout' : self.Source.parametres['pathout'],
623                         'word' : index ,
624                         'lem' : self.Source.parametres['lem'],
625                         'tmpchi' : self.tmpchi}
626         #try :
627         self.parent.SimiFromCluster(self.parent, self.Source.corpus, self.la, self.lfreq, self.lchi, self.cl - 1, parametres = parametres, dlg = progressbar(self.ira, 4))
628         #except :
629         #    print 'not acitve'
630
631     def on_graph(self, evt):
632         if self.tmpchi is None :
633             self.tmpchi = os.path.join(self.Source.parametres['pathout'],'chi_%i.csv' % self.cl)
634             with open(self.tmpchi, 'w') as f:
635                 f.write('\n'.join([str(val) for val in self.lchi]))
636         parametres = {'type' : 'clustersimitxt',
637                         'pathout' : self.Source.parametres['pathout'],
638                         'lem' : self.Source.parametres['lem'],
639                         'tmpchi' : self.tmpchi}
640
641         self.parent.SimiFromCluster(self.parent, self.Source.corpus, self.la, self.lfreq, self.lchi, self.cl - 1, parametres = parametres, dlg = progressbar(self.ira, 4))
642
643     def on_segments(self,evt) :
644         dlg = progressbar(self, 2)
645         corpus = self.Source.corpus
646         uces = corpus.lc[self.cl-1]
647         if self.Source.parametres['classif_mode'] != 2 :
648             uci = False
649         else :
650             uci = True
651         l = []
652         dlg.Update(1, u'Segments...')
653         for i in range(2,10) :
654             li = corpus.find_segments_in_classe(uces, i, 1000, uci = uci)
655             if li == [] :
656                 break
657             else :
658                 l += li
659         l.sort(reverse = True)
660         d = {}
661         dlg.Update(2, 'Tri...')
662         for i, line in enumerate(l) :
663             d[i] = [line[1],line[0], line[2]]
664         first = ['','','']
665         para={'dico': d,'fline':first}
666         dlg.Destroy()
667         win = wliste(self, -1, ' - '.join([_(u"Repeated segments").decode('utf8'), "Classe %i" % self.cl]), d, first, size=(600, 500))
668         win.Show(True)
669
670     def on_uce_carac(self,evt) :
671         dial = PrefUCECarac(self, self.parent)
672         dial.CenterOnParent()
673         if dial.ShowModal() == wx.ID_OK :
674             limite = dial.spin_eff.GetValue()
675             atype = dial.radio_type.GetSelection()
676             dlg = progressbar(self.ira,maxi = 4)
677             corpus = self.Source.corpus
678             uces = corpus.lc[self.cl-1]
679             if self.Source.parametres['classif_mode'] != 2 :
680                 uci = False
681             else :
682                 uci = True
683             tab = corpus.make_table_with_classe(uces, self.la, uci = uci)
684             tab.pop(0)
685             dlg.Update(2, u'score...')
686             if atype == 0 :
687                 ntab = [round(sum([self.lchi[i] for i, word in enumerate(line) if word == 1]),2) for line in tab]
688             else :
689                 ntab = [round(sum([self.lchi[i] for i, word in enumerate(line) if word == 1])/float(sum(line)),2) if sum(line)!=0 else 0 for line in tab]
690             ntab2 = [[ntab[i], uces[i]] for i, val in enumerate(ntab)]
691             del ntab
692             ntab2.sort(reverse = True)
693             ntab2 = ntab2[:limite]
694             nuces = [val[1] for val in ntab2]
695             dlg.Update(3, u'concordancier...')
696             ucis_txt, ucestxt = doconcorde(corpus, nuces, self.la, uci = uci)
697             items = dict([[i, '<br>'.join([ucis_txt[i], '<table bgcolor = #1BF0F7 border=0><tr><td><b>score : %.2f</b></td></tr></table><br>' % ntab2[i][0], ucestxt[i]])] for i, uce in enumerate(nuces)])
698             dlg.Destroy()
699             win = message(self, items, ' - '.join([_(u"Typical text segments").decode('utf8'), "Classe %i" % self.cl]), (750, 600), uceids = nuces)
700             #win.SetWindowStyle(wx.STAY_ON_TOP)
701             #win.html = '<html>\n' + '<br>'.join(['<br>'.join([ucis_txt[i], '<table bgcolor = #1BF0F7 border=0><tr><td><b>score : %.2f</b></td></tr></table>' % ntab2[i][0], ucestxt[i]]) for i in range(0,len(ucestxt))]) + '\n</html>'
702             #win.HtmlPage.SetPage(win.html)
703             win.Show(True)
704
705     def on_tablex(self, evt):
706         if 'corpus' in dir(self.Source):
707             corpus = self.Source.corpus
708         else :
709             corpus = self.Source.tableau
710         with codecs.open(self.Source.pathout['chisqtable'], 'r', corpus.parametres['syscoding']) as f :
711             chistable = [line.replace('\n','').replace('\r','').replace('"','').replace(',','.').split(';') for line in f]
712         title = chistable[0]
713         title.pop(0)
714         chistable.pop(0)
715         vchistable = [line[1:] for line in chistable]
716         fchistable = [line[0] for line in chistable]
717         words = self.getselectedwords()
718         tableout = [vchistable[fchistable.index(self.getword(word))] for word in words]
719         tmpgraph = tempfile.mktemp(dir=self.Source.parent.TEMPDIR)
720         nbcl = len(title)
721         nbwords = len(words)
722         BarFrame(self.Source.parent, tableout, title, words)
723
724     def onlexdendro(self, evt):
725         if 'corpus' in dir(self.Source):
726             corpus = self.Source.corpus
727         else :
728             corpus = self.Source.tableau
729         with codecs.open(self.Source.pathout['chisqtable'], 'r', corpus.parametres['syscoding']) as f :
730             chistable = [line.replace('\n','').replace('\r','').replace('"','').replace(',','.').split(';') for line in f]
731         title = chistable[0]
732         title.pop(0)
733         chistable.pop(0)
734         vchistable = [line[1:] for line in chistable]
735         fchistable = [line[0] for line in chistable]
736         words = self.getselectedwords()
737         tableout = [vchistable[fchistable.index(self.getword(word))] for word in words]
738         BarFrame(self.Source.parent, tableout, title, words, tree = self.Source.pathout['Rdendro'])
739
740     def getword(self, word) :
741         if self.translation :
742             return self.lems[word]
743         else :
744             return word
745
746     def make_concord(self, uces, title, color = 'red') :
747         corpus = self.Source.corpus
748         ListWord = [self.getColumnText(self.GetFirstSelected(), 6)]
749         last = self.GetFirstSelected()
750         while self.GetNextSelected(last) != -1:
751             last = self.GetNextSelected(last)
752             ListWord.append(self.getColumnText(last, 6))
753         ucef = []
754         ListWord = [self.getword(word) for word in ListWord]
755         if self.Source.parametres['classif_mode'] != 2 :
756             for word in ListWord :
757                 uci = False
758                 ucef += list(set(corpus.getlemuces(word)).intersection(uces))
759         else :
760             for word in ListWord :
761                 ucef += list(set(corpus.getlemucis(word)).intersection(uces))
762                 uci = True
763         ucis_txt, ucestxt = doconcorde(corpus, ucef, ListWord, uci = uci)
764         items = dict([[i, '<br><br>'.join([ucis_txt[i], ucestxt[i]])] for i in range(0,len(ucestxt))])
765         win = message(self, items, title, (800, 500), uceids = ucef)
766         return win
767
768     def OnPopupTwo(self, event):
769         if 'corpus' in dir(self.Source) :
770             corpus = self.Source.corpus
771             uces = corpus.lc[self.cl-1]
772             win = self.make_concord(uces, ' - '.join([_(u"Concordance").decode('utf8'), "Classe %i" % self.cl]))
773             win.Show(True)
774
775     def OnPopupThree(self, event):
776         corpus = self.Source.corpus
777         uces = [classe[i] for classe in corpus.lc for i in range(0,len(classe))]
778         win = self.make_concord(uces, ' - '.join([_(u"Concordance").decode('utf8'), _(u"Segments of this clustering").decode('utf8')]))
779         win.Show(True)
780
781     def OnPopupFour(self, event):
782         corpus = self.Source.corpus
783         uces = [classe[i] for classe in corpus.lc for i in range(0,len(classe))] + corpus.lc0
784         win = self.make_concord(uces, ' - '.join([_(u"Concordance").decode('utf8'), _(u"All segments").decode('utf8')]))
785         win.Show(True)
786
787     def OnPopupFive(self, event):
788         word = self.getColumnText(self.GetFirstSelected(), 6)
789         lk = "http://www.cnrtl.fr/definition/" + word
790         webbrowser.open(lk)
791
792     def OnPopupSix(self, event):
793         word = self.getColumnText(self.GetFirstSelected(), 6)
794         lk = "http://www.cnrtl.fr/etymologie/" + word
795         webbrowser.open(lk)
796
797     def OnPopupSeven(self, event):
798         word = self.getColumnText(self.GetFirstSelected(), 6)
799         lk = "http://www.cnrtl.fr/synonymie/" + word
800         webbrowser.open(lk)
801
802     def OnPopupHeight(self, event):
803         word = self.getColumnText(self.GetFirstSelected(), 6)
804         lk = "http://www.cnrtl.fr/antonymie/" + word
805         webbrowser.open(lk)
806
807     def OnPopupNine(self, event):
808         word = self.getColumnText(self.GetFirstSelected(), 6)
809         lk = "http://www.cnrtl.fr/morphologie/" + word
810         webbrowser.open(lk)
811
812     def onproxe(self, evt) :
813         word = self.getColumnText(self.GetFirstSelected(), 6)
814         lk = "http://www.cnrtl.fr/proxemie/" + word
815         webbrowser.open(lk)
816
817     def OnPopupOne(self, event):
818         corpus = self.Source.corpus
819         #print 'ATTENTION PRINT ET TABLE'
820         #corpus.make_et_table()
821         word = self.getColumnText(self.GetFirstSelected(), 6)
822         word = self.getword(word)
823         lems = corpus.getlems()
824         uces = corpus.lc[self.cl-1]
825         rep = []
826         #FIXME : donner aussi eff reel a la place de nb uce
827         for forme in lems[word].formes :
828             if self.Source.parametres['classif_mode'] != 2 :
829                 ucef = list(set(corpus.getworduces(forme)).intersection(uces))
830             else :
831                 ucef = list(set(corpus.getworducis(forme)).intersection(uces))
832             #ucef = [uce for uce in corpus.formes[forme][1] if uce in uces]
833             if ucef != [] :
834                 nb = len(ucef)
835                 rep.append([corpus.getforme(forme).forme, nb])
836         rep.sort(key = itemgetter(1), reverse = True)
837         #win = message(self, u"Formes associées", wx.Size(300, 200))
838         items = dict([[i, '\t:\t'.join([str(val) for val in forme])] for i, forme in enumerate(rep)])
839         win = message(self, items, _(u"Associated forms").decode('utf8'), (300, 200))
840         #win.html = '<html>\n' + '<br>'.join([' : '.join([str(val) for val in forme]) for forme in rep]) + '\n</html>'
841         #win.HtmlPage.SetPage(win.html)
842         win.Show(True)
843
844     def OnMakeTgen(self, evt):
845         self.parent.tree.OnTgenEditor(self.getselectedwords())
846
847
848 class wliste(wx.Frame):
849     def __init__(self, parent, id, title, d, fline, size=(600, 500)):
850         wx.Frame.__init__(self, parent, id)
851         self.liste = ListForSpec(self, parent, d, fline[1:], menu = False)
852         self.button_1 = wx.Button(self, -1, _(u"Close").decode('utf8'))
853         self.Bind(wx.EVT_BUTTON, self.OnCloseMe, self.button_1)
854         self.Bind(wx.EVT_CLOSE, self.OnCloseWindow)
855         self.__do_layout()
856
857     def __do_layout(self):
858         sizer_1 = wx.BoxSizer(wx.VERTICAL)
859         sizer_2 = wx.BoxSizer(wx.VERTICAL)
860         sizer_2.Add(self.liste, 1, wx.EXPAND | wx.ADJUST_MINSIZE, 0)
861         sizer_2.Add(self.button_1, 0, wx.ALIGN_CENTER_HORIZONTAL | wx.ADJUST_MINSIZE, 0)
862         sizer_1.Add(sizer_2, 1, wx.EXPAND, 0)
863         self.SetAutoLayout(True)
864         self.SetSizer(sizer_1)
865         self.Layout()
866
867     def OnCloseMe(self, event):
868         self.Close(True)
869
870     def OnCloseWindow(self, event):
871         self.Destroy()