1 # -*- coding: utf-8 -*-
3 #----------------------------------------------------------------------------
4 # Author: Pierre Ratinaud
7 #comes from ListCtrl.py from the demo tool of wxPython:
8 # Author: Robin Dunn & Gary Dumer
11 # Copyright: (c) 1998 by Total Control Software
12 # Licence: wxWindows license
13 #----------------------------------------------------------------------------
18 import wx.lib.mixins.listctrl as listmix
21 from functions import exec_rcode, MessageImage
22 from dialog import message
23 from chemins import ffr
24 from PrintRScript import barplot, dendroandbarplot
25 #---------------------------------------------------------------------------
27 class SearchList(wx.ListCtrl, listmix.ListCtrlAutoWidthMixin, listmix.ColumnSorterMixin): #wx.Panel, listmix.ColumnSorterMixin):
28 def __init__(self, parent,gparent, dlist,first, nbactives, nbetoiles, add_dendro=True):
29 wx.ListCtrl.__init__( self, parent, -1, style=wx.LC_REPORT|wx.LC_VIRTUAL|wx.LC_HRULES|wx.LC_VRULES)
33 self.add_dendro=add_dendro
34 self.first = ['id','formes'] + first
35 self.lenact = nbactives
36 self.lensup = len(dlist) - (self.lenact + nbetoiles)
39 self.il = wx.ImageList(16, 16)
40 a={"sm_up":"GO_UP","sm_dn":"GO_DOWN","w_idx":"WARNING","e_idx":"ERROR","i_idx":"QUESTION"}
42 s="self.%s= self.il.Add(wx.ArtProvider_GetBitmap(wx.ART_%s,wx.ART_TOOLBAR,(16,16)))" % (k,v)
44 self.SetImageList(self.il, wx.IMAGE_LIST_SMALL)
46 self.attr1 = wx.ListItemAttr()
47 self.attr1.SetBackgroundColour((220, 220, 220))
48 self.attrsg = wx.ListItemAttr()
49 self.attrsg.SetBackgroundColour((230, 230, 230))
50 self.attr2 = wx.ListItemAttr()
51 self.attr2.SetBackgroundColour((190, 249, 236))
52 self.attr2s = wx.ListItemAttr()
53 self.attr2s.SetBackgroundColour((211, 252, 244))
54 self.attr3 = wx.ListItemAttr()
55 self.attr3.SetBackgroundColour((245, 180, 180))
56 self.attr3s = wx.ListItemAttr()
57 self.attr3s.SetBackgroundColour((245, 190, 190))
63 self.Bind(wx.EVT_LIST_ITEM_SELECTED, self.OnItemSelected, self)
64 #self.Bind(wx.EVT_LIST_COL_CLICK, self.OnColClick)
67 self.Bind(wx.EVT_COMMAND_RIGHT_CLICK, self.OnRightClick)
70 self.Bind(wx.EVT_RIGHT_UP, self.OnRightClick)
73 self.itemDataMap = dlist
74 self.itemIndexMap = dlist.keys()
75 self.SetItemCount(len(dlist))
76 listmix.ColumnSorterMixin.__init__(self, len(first)+2)
79 #-----------------------------------------------------------------------------------------
81 first = ['id','formes']+first
82 for i, name in enumerate(first) :
83 self.InsertColumn(i, name, wx.LIST_FORMAT_LEFT)
85 self.SetColumnWidth(0, wx.LIST_AUTOSIZE)
86 for i in range(1,len(first)-1):
87 self.SetColumnWidth(i, 130)
89 self.SortListItems(0, True)
91 # Used by the ColumnSorterMixin, see wx/lib/mixins/listctrl.py
92 def GetListCtrl(self):
95 # Used by the ColumnSorterMixin, see wx/lib/mixins/listctrl.py
96 def GetSortImages(self):
97 return (self.sm_dn, self.sm_up)
100 def OnRightDown(self, event):
103 item, flags = self.HitTest((x, y))
105 if flags & wx.LIST_HITTEST_ONITEM:
110 def getColumnText(self, index, col):
111 item = self.GetItem(index, col)
112 return item.GetText()
114 def OnGetItemText(self, item, col):
115 index=self.itemIndexMap[item]
116 s = self.itemDataMap[index][col]
119 def OnGetItemData(self, item) :
120 index = self.itemIndexMap[item]
121 s = self.itemDataMap[index]
124 def OnItemSelected(self, event):
125 self.currentItem = event.m_itemIndex
128 def OnGetItemAttr(self, item):
129 index=self.itemIndexMap[item]
130 #genre=self.itemDataMap[index][2]
131 if index < self.lenact :
136 elif index >= self.lenact and index < (self.lenact + self.lensup) :
141 elif index >= (self.lenact + self.lensup) :
149 def SortItems(self,sorter=cmp):
150 items = list(self.itemDataMap.keys())
152 self.itemIndexMap = items
157 def OnRightClick(self, event):
159 # only do this part the first time so the events are only bound once
160 if not hasattr(self, "popupID1"):
161 # self.popupID1 = wx.NewId()
162 #self.popupID2 = wx.NewId()
163 self.popupID3 = wx.NewId()
165 self.id_adddendro = wx.NewId()
166 self.Bind(wx.EVT_MENU, self.ongraphdendro, id = self.id_adddendro)
167 # self.Bind(wx.EVT_MENU, self.OnPopupOne, id=self.popupID1)
168 #self.Bind(wx.EVT_MENU, self.OnPopupTwo, id=self.popupID2)
169 self.Bind(wx.EVT_MENU, self.OnPopupThree, id=self.popupID3)
175 # menu.Append(self.popupID1, u"Formes associƩes")
176 #menu.Append(self.popupID2, u"Concordancier")
177 menu.Append(self.popupID3, "Graphique")
179 menu.Append(self.id_adddendro, u"Graphique + dendrogramme")
184 def getselectedwords(self) :
185 words = [self.getColumnText(self.GetFirstSelected(), 6)]
186 last = self.GetFirstSelected()
187 while self.GetNextSelected(last) != -1:
188 last = self.GetNextSelected(last)
189 words.append(self.getColumnText(last, 6))
192 def OnPopupOne(self, event):
193 activenotebook = self.parent.nb.GetSelection()
194 page = self.parent.nb.GetPage(activenotebook)
196 word = self.getColumnText(self.GetFirstSelected(), 0)
199 for forme in lems[word] :
200 rep.append([forme, corpus.formes[forme][0]])
201 win = message(self, u"Formes associƩes", size=(300, 200))
202 win.html = '<html>\n' + '<br>'.join([' : '.join([str(val) for val in forme]) for forme in rep]) + '\n</html>'
203 win.HtmlPage.SetPage(win.html)
206 def OnPopupTwo(self, event):
207 activenotebook = self.parent.nb.GetSelection()
208 page = self.parent.nb.GetPage(activenotebook)
209 item=self.getColumnText(self.GetFirstSelected(), 0)
211 win = message(self, u"Concordancier", size=(600, 200))
213 listmot = corpus.lems[item]
214 uce_ok = [corpus.formes[forme][1] for forme in listmot]
215 uce_ok = list(set([tuple(val) for line in uce_ok for val in line]))
216 txt = '<h1>Concordancier</h1>'
218 content = ' '+' '.join(corpus.ucis_paras_uces[uce[0]][uce[1]][uce[2]])+' '
219 for form in listmot :
222 forme = ' ' + form + ' '
223 while i < len(content):
224 coordword = content[i:].find(forme)
225 if coordword != -1 and i == 0:
226 txt += '<br><b>' + ' '.join(corpus.ucis[uce[0]][0]) + '</b><br>'
228 sp = ' ' * (avap - coordword)
231 deb = i + coordword - avap
232 if len(content) < i + coordword + avap:
233 fin = len(content) - 1
235 fin = i + coordword + avap
236 txt += '<TT>' + sp + content[deb:fin].replace(forme, '<font color=red>' + forme + '</font>') + '</TT><br>'
237 i += coordword + len(forme)
239 elif coordword != -1 and i != 0 :
241 sp = ' ' * (avap - coordword)
244 deb = i + coordword - avap
245 if len(content) < i + coordword + avap:
246 fin = len(content) - 1
248 fin = i + coordword + avap
249 txt += '<TT>' + sp + content[deb:fin].replace(forme, '<font color=red>' + forme + '</font>') + '</TT><br>'
250 i += coordword + len(forme)
255 win.HtmlPage.SetPage(txt)
258 def OnPopupThree(self, event) :
259 datas = [self.OnGetItemData(self.GetFirstSelected())]
260 last = self.GetFirstSelected()
261 while self.GetNextSelected(last) != -1:
262 last = self.GetNextSelected(last)
263 data = self.OnGetItemData(last)
265 colnames = self.first[2:]
266 rownames = [val[1] for val in datas]
267 table = [[str(val) for val in line[2:]] for line in datas]
268 tmpgraph = tempfile.mktemp(dir=self.parent.parent.TEMPDIR)
269 txt = barplot(table, rownames, colnames, self.parent.parent.RscriptsPath['Rgraph'], tmpgraph)
270 tmpscript = tempfile.mktemp(dir=self.parent.parent.TEMPDIR)
271 with open(tmpscript,'w') as f :
273 exec_rcode(self.parent.parent.RPath, tmpscript, wait = True)
274 win = MessageImage(self, u"Graphique", size=(700, 500))
275 win.addsaveimage(tmpgraph)
276 txt = "<img src='%s'>" % tmpgraph
277 win.HtmlPage.SetPage(txt)
280 def ongraphdendro(self, evt) :
281 corpus = self.parent.corpus
282 datas = [self.OnGetItemData(self.GetFirstSelected())]
283 last = self.GetFirstSelected()
284 while self.GetNextSelected(last) != -1:
285 last = self.GetNextSelected(last)
286 data = self.OnGetItemData(last)
288 colnames = self.first[2:]
289 rownames = [val[1] for val in datas]
290 table = [[str(val) for val in line[2:]] for line in datas]
291 tmpgraph = tempfile.mktemp(dir=self.parent.parent.TEMPDIR)
292 txt = dendroandbarplot(table, rownames, colnames, self.parent.parent.RscriptsPath['Rgraph'], tmpgraph, dendro=corpus.dictpathout['Rdendro'])
293 tmpscript = tempfile.mktemp(dir=self.parent.parent.TEMPDIR)
294 with open(tmpscript,'w') as f :
296 exec_rcode(self.parent.parent.RPath, tmpscript, wait = True)
297 win = MessageImage(self, u"Graphique", size=(700, 500))
298 win.addsaveimage(tmpgraph)
299 txt = "<img src='%s'>" % tmpgraph
300 win.HtmlPage.SetPage(txt)