iramuteq.org Git - iramuteq/blob - Liste.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 sys
  18 import  wx
  19 from dialog import SearchDial
  20 import wx.lib.mixins.listctrl as listmix
  21 import cStringIO
  22
  23 #---------------------------------------------------------------------------
  24
  25 class List(wx.ListCtrl, listmix.ListCtrlAutoWidthMixin):
  26     def __init__(self, parent, ID, pos=wx.DefaultPosition,
  27                  size=wx.DefaultSize, style=0):
  28         wx.ListCtrl.__init__(self, parent, ID, pos, size, style)
  29         listmix.ListCtrlAutoWidthMixin.__init__(self)
  30
  31
  32 class ListPanel(wx.Panel, listmix.ColumnSorterMixin):
  33     def __init__(self, parent, gparent, dlist):
  34         self.parent = parent
  35         self.gparent = gparent
  36         self.source = gparent
  37         self.dlist = dlist
  38         wx.Panel.__init__(self, parent, -1, style=wx.WANTS_CHARS)
  39
  40         search_id = wx.NewId()
  41         self.parent.Bind(wx.EVT_MENU, self.onsearch, id = search_id)
  42         self.accel_tbl = wx.AcceleratorTable([(wx.ACCEL_CTRL, ord('F'), search_id)])
  43         self.SetAcceleratorTable(self.accel_tbl)
  44
  45         self.il = wx.ImageList(16, 16)
  46         self.sm_up = self.il.Add(getSmallUpArrowBitmap())
  47         self.sm_dn = self.il.Add(getSmallDnArrowBitmap())
  48
  49
  50         tID = wx.NewId()
  51
  52         self.list = List(self, tID,
  53                                  style=wx.LC_REPORT
  54                                  | wx.BORDER_NONE
  55                                  | wx.LC_EDIT_LABELS
  56                                  | wx.LC_SORT_ASCENDING
  57                                  )
  58
  59
  60         self.list.SetImageList(self.il, wx.IMAGE_LIST_SMALL)
  61         self.PopulateList(dlist)
  62
  63         self.Bind(wx.EVT_SIZE, self.OnSize)
  64
  65         self.Bind(wx.EVT_LIST_ITEM_SELECTED, self.OnItemSelected, self.list)
  66         self.Bind(wx.EVT_LIST_COL_CLICK, self.OnColClick, self.list)
  67         # for wxMSW
  68         self.list.Bind(wx.EVT_COMMAND_RIGHT_CLICK, self.OnRightClick)
  69
  70         # for wxGTK
  71         self.list.Bind(wx.EVT_RIGHT_UP, self.OnRightClick)
  72         self.itemDataMap = dlist
  73
  74         listmix.ColumnSorterMixin.__init__(self, 3)
  75         self.SortListItems(1, False)
  76         self.do_greyline()
  77 #-----------------------------------------------------------------------------------------
  78
  79     def PopulateList(self, dlist):
  80
  81         #self.list.InsertColumn(0,'id', wx.LIST_FORMAT_LEFT)
  82 #        i=1
  83         self.list.InsertColumn(0, 'forme', wx.LIST_FORMAT_RIGHT)
  84         self.list.InsertColumn(1, 'nb', wx.LIST_FORMAT_RIGHT)
  85         self.list.InsertColumn(2, 'type', wx.LIST_FORMAT_RIGHT)
  86         self.list.InsertColumn(3, '', wx.LIST_FORMAT_RIGHT)
  87
  88         ct = 0
  89         for key, data in dlist.iteritems():
  90             ct += 1
  91             index = self.list.InsertStringItem(sys.maxint, data[0])
  92             self.list.SetStringItem(index, 1, `data[1]`)
  93             self.list.SetStringItem(index, 2, data[2])
  94             self.list.SetStringItem(index, 3, '')
  95             self.list.SetItemData(index, key)
  96
  97         self.list.SetColumnWidth(0, 150)
  98         self.list.SetColumnWidth(1, 100)
  99         self.list.SetColumnWidth(2, 100)
 100         self.list.SetColumnWidth(3, wx.LIST_AUTOSIZE)
 101
 102
 103         self.currentItem = 0
 104
 105     def do_greyline(self):
 106         for row in xrange(self.list.GetItemCount()):
 107             if row % 2 :
 108                 self.list.SetItemBackgroundColour(row, (230, 230, 230))
 109             else :
 110                 self.list.SetItemBackgroundColour(row, wx.WHITE)
 111
 112     def OnColClick(self, event):
 113         self.do_greyline()
 114
 115     # Used by the ColumnSorterMixin, see wx/lib/mixins/listctrl.py
 116     def GetListCtrl(self):
 117         return self.list
 118
 119     # Used by the ColumnSorterMixin, see wx/lib/mixins/listctrl.py
 120     def GetSortImages(self):
 121         return (self.sm_dn, self.sm_up)
 122
 123
 124     def OnRightDown(self, event):
 125         x = event.GetX()
 126         y = event.GetY()
 127         item, flags = self.list.HitTest((x, y))
 128
 129         if flags & wx.LIST_HITTEST_ONITEM:
 130             self.list.Select(item)
 131
 132         event.Skip()
 133
 134
 135     def getColumnText(self, index, col):
 136         item = self.list.GetItem(index, col)
 137         return item.GetText()
 138
 139
 140     def OnItemSelected(self, event):
 141         self.currentItem = event.m_itemIndex
 142         event.Skip()
 143
 144     def onsearch(self, evt) :
 145         self.dial = SearchDial(self, self, 0, True)
 146         self.dial.CenterOnParent()
 147         self.dial.ShowModal()
 148         self.dial.Destroy()
 149
 150     def OnRightClick(self, event):
 151
 152         # only do this part the first time so the events are only bound once
 153         if not hasattr(self, "popupID1"):
 154             self.popupID1 = wx.NewId()
 155             self.popupID2 = wx.NewId()
 156  #           self.popupID3 = wx.NewId()
 157
 158             self.Bind(wx.EVT_MENU, self.OnPopupOne, id=self.popupID1)
 159             self.Bind(wx.EVT_MENU, self.OnPopupTwo, id=self.popupID2)
 160 #            self.Bind(wx.EVT_MENU, self.OnPopupThree, id=self.popupID3)
 161
 162         # make a menu
 163         menu = wx.Menu()
 164         # add some items
 165         menu.Append(self.popupID1, u"Formes associées")
 166         menu.Append(self.popupID2, u"Concordancier")
 167 #        menu.Append(self.popupID3, "recharger")
 168
 169         self.PopupMenu(menu)
 170         menu.Destroy()
 171
 172
 173     def OnPopupOne(self, event):
 174         corpus = self.gparent.corpus
 175         word = self.getColumnText(self.list.GetFirstSelected(), 0)
 176         lems = corpus.getlems()
 177         rep = []
 178         for forme in lems[word].formes :
 179              rep.append([corpus.getforme(forme).forme, corpus.getforme(forme).freq])
 180         win = message(self, -1, u"Formes associées", size=(300, 200), style=wx.DEFAULT_FRAME_STYLE)
 181         win.html = '<html>\n' + '<br>'.join([' : '.join([str(val) for val in forme]) for forme in rep]) + '\n</html>'
 182         win.HtmlPage.SetPage(win.html)
 183         win.Show(True)
 184
 185     def OnPopupTwo(self, event):
 186         corpus = self.gparent.corpus
 187         win = message(self, -1, u"Concordancier", size=(600, 200), style=wx.DEFAULT_FRAME_STYLE)
 188         avap = 60
 189         item = self.getColumnText(self.list.GetFirstSelected(), 0)
 190         listmot = corpus.getlems()[item].formes
 191         #uce_ok = [corpus.formes[corpus.idformes[forme].forme][1] for forme in listmot]
 192         uce_ok = corpus.getlemuces(item)#list(set([tuple(val) for line in uce_ok for val in line]))
 193         txt = '<h1>Concordancier</h1>'
 194         res = corpus.getconcorde(uce_ok)
 195         for uce in res :
 196             ucetxt = ' '+uce[1]+' '
 197             txt += ' '.join(corpus.ucis[corpus.getucefromid(uce[0]).uci].etoiles) + '<br>'
 198             for forme in listmot :
 199                 forme = corpus.getforme(forme).forme
 200                 ucetxt = ucetxt.replace(' '+forme+' ', '<font color=red> ' + forme + ' </font>')
 201             txt += ucetxt + '<br><br>'
 202 #        for uce in uce_ok:
 203 #            content = ' '+' '.join(corpus.ucis_paras_uces[uce[0]][uce[1]][uce[2]])+' '
 204 #            for form in listmot :
 205 #                sp = ''
 206 #                i = 0
 207 #                forme = ' ' + form + ' '
 208 #                while i < len(content):
 209 #                    coordword = content[i:].find(forme)
 210 #                    if coordword != -1 and i == 0:
 211 #                        txt += '<br><b>' + ' '.join(corpus.ucis[uce[0]][0]) + '</b><br>'
 212 #                        if coordword < avap:
 213 #                            sp = '&nbsp;' * (avap - coordword)
 214 #                            deb = i
 215 #                        else:
 216 #                            deb = i + coordword - avap
 217 #                        if len(content) < i + coordword + avap:
 218 #                            fin = len(content) - 1
 219 #                        else:
 220 #                            fin = i + coordword + avap
 221 #                        txt += '<TT>' + sp + content[deb:fin].replace(forme, '<font color=red>' + forme + '</font>') + '</TT><br>'
 222 #                        i += coordword + len(forme)
 223 #                        sp = ''
 224 #                    elif coordword != -1 and i != 0 :
 225 #                        if coordword < avap:
 226 #                            sp = '&nbsp;' * (avap - coordword)
 227 #                            deb = i
 228 #                        else:
 229 #                            deb = i + coordword - avap
 230 #                        if len(content) < i + coordword + avap:
 231 #                            fin = len(content) - 1
 232 #                        else:
 233 #                            fin = i + coordword + avap
 234 #                        txt += '<TT>' + sp + content[deb:fin].replace(forme, '<font color=red>' + forme + '</font>') + '</TT><br>'
 235 #                        i += coordword + len(forme)
 236 #                        sp = ''
 237 #                    else:
 238 #                        i = len(content)
 239 #                        sp = ''
 240         win.HtmlPage.SetPage(txt)
 241         win.Show(True)
 242
 243     def OnSize(self, event):
 244         w, h = self.GetClientSizeTuple()
 245         self.list.SetDimensions(0, 0, w, h)
 246
 247 class message(wx.Frame):
 248     def __init__(self, *args, **kwds):
 249         # begin wxGlade: MyFrame.__init__
 250         kwds["style"] = wx.DEFAULT_FRAME_STYLE
 251         wx.Frame.__init__(self, *args, **kwds)
 252         #self.text_ctrl_1 = wx.TextCtrl(self, -1, "", style=wx.TE_MULTILINE)
 253         self.HtmlPage = wx.html.HtmlWindow(self, -1)
 254         if "gtk2" in wx.PlatformInfo:
 255             self.HtmlPage.SetStandardFonts()
 256         self.HtmlPage.SetFonts('Courier', 'Courier')
 257
 258
 259         self.button_1 = wx.Button(self, -1, "Fermer")
 260         self.Bind(wx.EVT_BUTTON, self.OnCloseMe, self.button_1)
 261         self.Bind(wx.EVT_CLOSE, self.OnCloseWindow)
 262         self.__do_layout()
 263         # end wxGlade
 264
 265     def __do_layout(self):
 266         # begin wxGlade: MyFrame.__do_layout
 267         sizer_1 = wx.BoxSizer(wx.VERTICAL)
 268         sizer_2 = wx.BoxSizer(wx.VERTICAL)
 269         sizer_2.Add(self.HtmlPage, 1, wx.EXPAND | wx.ADJUST_MINSIZE, 0)
 270         sizer_2.Add(self.button_1, 0, wx.ALIGN_CENTER_HORIZONTAL | wx.ADJUST_MINSIZE, 0)
 271         sizer_1.Add(sizer_2, 1, wx.EXPAND, 0)
 272         self.SetAutoLayout(True)
 273         self.SetSizer(sizer_1)
 274         self.Layout()
 275         # end wxGlade
 276
 277     def OnCloseMe(self, event):
 278         self.Close(True)
 279
 280     def OnCloseWindow(self, event):
 281         self.Destroy()
 282
 283
 284 def getSmallUpArrowData():
 285     return \
 286 '\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR\x00\x00\x00\x10\x00\x00\x00\x10\x08\x06\
 287 \x00\x00\x00\x1f\xf3\xffa\x00\x00\x00\x04sBIT\x08\x08\x08\x08|\x08d\x88\x00\
 288 \x00\x00<IDAT8\x8dcddbf\xa0\x040Q\xa4{h\x18\xf0\xff\xdf\xdf\xffd\x1b\x00\xd3\
 289 \x8c\xcf\x10\x9c\x06\xa0k\xc2e\x08m\xc2\x00\x97m\xd8\xc41\x0c \x14h\xe8\xf2\
 290 \x8c\xa3)q\x10\x18\x00\x00R\xd8#\xec\xb2\xcd\xc1Y\x00\x00\x00\x00IEND\xaeB`\
 291 \x82'
 292
 293 def getSmallUpArrowBitmap():
 294     return wx.BitmapFromImage(getSmallUpArrowImage())
 295
 296 def getSmallUpArrowImage():
 297     stream = cStringIO.StringIO(getSmallUpArrowData())
 298     return wx.ImageFromStream(stream)
 299
 300 #----------------------------------------------------------------------
 301 def getSmallDnArrowData():
 302     return \
 303 "\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR\x00\x00\x00\x10\x00\x00\x00\x10\x08\x06\
 304 \x00\x00\x00\x1f\xf3\xffa\x00\x00\x00\x04sBIT\x08\x08\x08\x08|\x08d\x88\x00\
 305 \x00\x00HIDAT8\x8dcddbf\xa0\x040Q\xa4{\xd4\x00\x06\x06\x06\x06\x06\x16t\x81\
 306 \xff\xff\xfe\xfe'\xa4\x89\x91\x89\x99\x11\xa7\x0b\x90%\ti\xc6j\x00>C\xb0\x89\
 307 \xd3.\x10\xd1m\xc3\xe5*\xbc.\x80i\xc2\x17.\x8c\xa3y\x81\x01\x00\xa1\x0e\x04e\
 308 ?\x84B\xef\x00\x00\x00\x00IEND\xaeB`\x82"
 309
 310 def getSmallDnArrowBitmap():
 311     return wx.BitmapFromImage(getSmallDnArrowImage())
 312
 313 def getSmallDnArrowImage():
 314     stream = cStringIO.StringIO(getSmallDnArrowData())
 315     return wx.ImageFromStream(stream)