+
+#----------------------------------------------------------------------------
+#----------------------------------------------------------------------------
+from bisect import bisect
+
+
+class TextEditMixin:
+ """
+ A mixin class that enables any text in any column of a
+ multi-column listctrl to be edited by clicking on the given row
+ and column. You close the text editor by hitting the ENTER key or
+ clicking somewhere else on the listctrl. You switch to the next
+ column by hiting TAB.
+
+ To use the mixin you have to include it in the class definition
+ and call the __init__ function::
+
+ class TestListCtrl(wx.ListCtrl, TextEditMixin):
+ def __init__(self, parent, ID, pos=wx.DefaultPosition,
+ size=wx.DefaultSize, style=0):
+ wx.ListCtrl.__init__(self, parent, ID, pos, size, style)
+ TextEditMixin.__init__(self)
+
+
+ Authors: Steve Zatz, Pim Van Heuven (pim@think-wize.com)
+ """
+
+ editorBgColour = wx.Colour(255,255,175) # Yellow
+ editorFgColour = wx.Colour(0,0,0) # black
+
+ def __init__(self):
+ #editor = wx.TextCtrl(self, -1, pos=(-1,-1), size=(-1,-1),
+ # style=wx.TE_PROCESS_ENTER|wx.TE_PROCESS_TAB \
+ # |wx.TE_RICH2)
+
+ self.make_editor()
+ self.Bind(wx.EVT_TEXT_ENTER, self.CloseEditor)
+ self.Bind(wx.EVT_LEFT_DOWN, self.OnLeftDown)
+ self.Bind(wx.EVT_LEFT_DCLICK, self.OnLeftDown)
+ self.Bind(wx.EVT_LIST_ITEM_SELECTED, self.OnItemSelected)
+
+
+ def make_editor(self, col_style=wx.LIST_FORMAT_LEFT):
+
+ style =wx.TE_PROCESS_ENTER|wx.TE_PROCESS_TAB|wx.TE_RICH2
+ style |= {wx.LIST_FORMAT_LEFT: wx.TE_LEFT,
+ wx.LIST_FORMAT_RIGHT: wx.TE_RIGHT,
+ wx.LIST_FORMAT_CENTRE : wx.TE_CENTRE
+ }[col_style]
+
+ editor = wx.TextCtrl(self, -1, style=style)
+ editor.SetBackgroundColour(self.editorBgColour)
+ editor.SetForegroundColour(self.editorFgColour)
+ font = self.GetFont()
+ editor.SetFont(font)
+
+ self.curRow = 0
+ self.curCol = 0
+
+ editor.Hide()
+ if hasattr(self, 'editor'):
+ self.editor.Destroy()
+ self.editor = editor
+
+ self.col_style = col_style
+ self.editor.Bind(wx.EVT_CHAR, self.OnChar)
+ self.editor.Bind(wx.EVT_KILL_FOCUS, self.CloseEditor)
+
+
+ def OnItemSelected(self, evt):
+ self.curRow = evt.GetIndex()
+ evt.Skip()
+
+
+ def OnChar(self, event):
+ ''' Catch the TAB, Shift-TAB, cursor DOWN/UP key code
+ so we can open the editor at the next column (if any).'''
+
+ keycode = event.GetKeyCode()
+ if keycode == wx.WXK_TAB and event.ShiftDown():
+ self.CloseEditor()
+ if self.curCol-1 >= 0:
+ self.OpenEditor(self.curCol-1, self.curRow)
+
+ elif keycode == wx.WXK_TAB:
+ self.CloseEditor()
+ if self.curCol+1 < self.GetColumnCount():
+ self.OpenEditor(self.curCol+1, self.curRow)
+
+ elif keycode == wx.WXK_ESCAPE:
+ self.CloseEditor()
+
+ elif keycode == wx.WXK_DOWN:
+ self.CloseEditor()
+ if self.curRow+1 < self.GetItemCount():
+ self._SelectIndex(self.curRow+1)
+ self.OpenEditor(self.curCol, self.curRow)
+
+ elif keycode == wx.WXK_UP:
+ self.CloseEditor()
+ if self.curRow > 0:
+ self._SelectIndex(self.curRow-1)
+ self.OpenEditor(self.curCol, self.curRow)
+
+ else:
+ event.Skip()
+
+
+ def OnLeftDown(self, evt=None):
+ ''' Examine the click and double
+ click events to see if a row has been click on twice. If so,
+ determine the current row and columnn and open the editor.'''
+
+ if self.editor.IsShown():
+ self.CloseEditor()
+
+ x,y = evt.GetPosition()
+ row,flags = self.HitTest((x,y))
+
+ if row != self.curRow: # self.curRow keeps track of the current row
+ evt.Skip()
+ return
+
+ # the following should really be done in the mixin's init but
+ # the wx.ListCtrl demo creates the columns after creating the
+ # ListCtrl (generally not a good idea) on the other hand,
+ # doing this here handles adjustable column widths
+
+ self.col_locs = [0]
+ loc = 0
+ for n in range(self.GetColumnCount()):
+ loc = loc + self.GetColumnWidth(n)
+ self.col_locs.append(loc)
+
+
+ col = bisect(self.col_locs, x+self.GetScrollPos(wx.HORIZONTAL)) - 1
+ self.OpenEditor(col, row)
+
+
+ def OpenEditor(self, col, row):
+ ''' Opens an editor at the current position. '''
+
+ # give the derived class a chance to Allow/Veto this edit.
+ evt = wx.ListEvent(wx.wxEVT_COMMAND_LIST_BEGIN_LABEL_EDIT, self.GetId())
+ evt.m_itemIndex = row
+ evt.m_col = col
+ item = self.GetItem(row, col)
+ evt.m_item.SetId(item.GetId())
+ evt.m_item.SetColumn(item.GetColumn())
+ evt.m_item.SetData(item.GetData())
+ evt.m_item.SetText(item.GetText())
+ ret = self.GetEventHandler().ProcessEvent(evt)
+ if ret and not evt.IsAllowed():
+ return # user code doesn't allow the edit.
+
+ if self.GetColumn(col).m_format != self.col_style:
+ self.make_editor(self.GetColumn(col).m_format)
+
+ x0 = self.col_locs[col]
+ x1 = self.col_locs[col+1] - x0
+
+ scrolloffset = self.GetScrollPos(wx.HORIZONTAL)
+
+ # scroll forward
+ if x0+x1-scrolloffset > self.GetSize()[0]:
+ if wx.Platform == "__WXMSW__":
+ # don't start scrolling unless we really need to
+ offset = x0+x1-self.GetSize()[0]-scrolloffset
+ # scroll a bit more than what is minimum required
+ # so we don't have to scroll everytime the user presses TAB
+ # which is very tireing to the eye
+ addoffset = self.GetSize()[0]/4
+ # but be careful at the end of the list
+ if addoffset + scrolloffset < self.GetSize()[0]:
+ offset += addoffset
+
+ self.ScrollList(offset, 0)
+ scrolloffset = self.GetScrollPos(wx.HORIZONTAL)
+ else:
+ # Since we can not programmatically scroll the ListCtrl
+ # close the editor so the user can scroll and open the editor
+ # again
+ self.editor.SetValue(self.GetItem(row, col).GetText())
+ self.curRow = row
+ self.curCol = col
+ self.CloseEditor()
+ return
+
+ y0 = self.GetItemRect(row)[1]
+
+ editor = self.editor
+ editor.SetDimensions(x0-scrolloffset,y0, x1,-1)
+
+ editor.SetValue(self.GetItem(row, col).GetText())
+ editor.Show()
+ editor.Raise()
+ editor.SetSelection(-1,-1)
+ editor.SetFocus()
+
+ self.curRow = row
+ self.curCol = col
+
+
+ # FIXME: this function is usually called twice - second time because
+ # it is binded to wx.EVT_KILL_FOCUS. Can it be avoided? (MW)
+ def CloseEditor(self, evt=None):
+ ''' Close the editor and save the new value to the ListCtrl. '''
+ if not self.editor.IsShown():
+ return
+ text = self.editor.GetValue()
+ self.editor.Hide()
+ self.SetFocus()
+
+ # post wxEVT_COMMAND_LIST_END_LABEL_EDIT
+ # Event can be vetoed. It doesn't has SetEditCanceled(), what would
+ # require passing extra argument to CloseEditor()
+ evt = wx.ListEvent(wx.wxEVT_COMMAND_LIST_END_LABEL_EDIT, self.GetId())
+ evt.m_itemIndex = self.curRow
+ evt.m_col = self.curCol
+ item = self.GetItem(self.curRow, self.curCol)
+ evt.m_item.SetId(item.GetId())
+ evt.m_item.SetColumn(item.GetColumn())
+ evt.m_item.SetData(item.GetData())
+ evt.m_item.SetText(text) #should be empty string if editor was canceled
+ ret = self.GetEventHandler().ProcessEvent(evt)
+ if not ret or evt.IsAllowed():
+ if self.IsVirtual():
+ # replace by whather you use to populate the virtual ListCtrl
+ # data source
+ self.SetVirtualData(self.curRow, self.curCol, text)
+ else:
+ self.SetStringItem(self.curRow, self.curCol, text)
+ self.RefreshItem(self.curRow)
+
+ def _SelectIndex(self, row):
+ listlen = self.GetItemCount()
+ if row < 0 and not listlen:
+ return
+ if row > (listlen-1):
+ row = listlen -1
+
+ self.SetItemState(self.curRow, ~wx.LIST_STATE_SELECTED,
+ wx.LIST_STATE_SELECTED)
+ self.EnsureVisible(row)
+ self.SetItemState(row, wx.LIST_STATE_SELECTED,
+ wx.LIST_STATE_SELECTED)
+
+
+
+#----------------------------------------------------------------------------
+#----------------------------------------------------------------------------
+
+"""
+FILENAME: CheckListCtrlMixin.py
+AUTHOR: Bruce Who (bruce.who.hk at gmail.com)
+DATE: 2006-02-09
+$Revision$
+DESCRIPTION:
+ This script provide a mixin for ListCtrl which add a checkbox in the first
+ column of each row. It is inspired by limodou's CheckList.py(which can be
+ got from his NewEdit) and improved:
+ - You can just use InsertStringItem() to insert new items;
+ - Once a checkbox is checked/unchecked, the corresponding item is not
+ selected;
+ - You can use SetItemData() and GetItemData();
+ - Interfaces are changed to OnCheckItem(), IsChecked(), CheckItem().
+
+ You should not set a imagelist for the ListCtrl once this mixin is used.
+
+HISTORY:
+1.3 - You can check/uncheck a group of sequential items by <Shift-click>:
+ First click(or <Shift-Click>) item1 to check/uncheck it, then
+ Shift-click item2 to check/uncheck it, and you'll find that all
+ items between item1 and item2 are check/unchecked!
+1.2 - Add ToggleItem()
+1.1 - Initial version
+"""
+
+from wx import ImageFromStream, BitmapFromImage
+import cStringIO, zlib
+
+def getUncheckData():
+ return zlib.decompress(
+"x\xda\xeb\x0c\xf0s\xe7\xe5\x92\xe2b``\xe0\xf5\xf4p\t\x02\xd2\x02 \xcc\xc1\
+\x06$\xe5?\xffO\x04R,\xc5N\x9e!\x1c@P\xc3\x91\xd2\x01\xe4\xbb{\xba8\x86X\xf4\
+&\xa7\xa4$\xa5-`1\x08\\2\xbb\xb1\xb1\x91\xf5\xd8\x84o\xeb\xff\xfaw\x1d[.=[2\
+\x90'\x01\x08v\xec]\xd3\xa3qvU`l\x81\xd9\xd18\t\xd3\x84+\x0cll[\xa6t\xcc9\
+\xd4\xc1\xda\xc3<O\x9a1\xc3\x88\xc3j\xfa\x86_\xee@#\x19<]\xfd\\\xd69%4\x01\
+\x00\xdc\x80-\x05" )
+
+def getUncheckBitmap():
+ return BitmapFromImage(getUncheckImage())
+
+def getUncheckImage():
+ stream = cStringIO.StringIO(getUncheckData())
+ return ImageFromStream(stream)
+
+def getCheckData():
+ return zlib.decompress(
+'x\xda\xeb\x0c\xf0s\xe7\xe5\x92\xe2b``\xe0\xf5\xf4p\t\x02\xd2\x02 \xcc\xc1\
+\x06$\xe5?\xffO\x04R,\xc5N\x9e!\x1c@P\xc3\x91\xd2\x01\xe47{\xba8\x86X\xf4&\
+\xa7\xa4$\xa5-`1\x08\\2\xbb\xb1\xb1\x91\xf5\xd8\x84o\xeb\xff\xfaw\x1d[.=[2\
+\x90\'\x01\x08v\xec\\2C\xe3\xec+\xc3\xbd\x05fG\xe3\x14n1\xcc5\xad\x8a8\x1a\
+\xb9\xa1\xeb\xd1\x853-\xaa\xc76\xecb\xb8i\x16c&\\\xc2\xb8\xe9Xvbx\xa1T\xc3U\
+\xd6p\'\xbd\x85\x19\xff\xbe\xbf\xd7\xe7R\xcb`\xd8\xa5\xf8\x83\xe1^\xc4\x0e\
+\xa1"\xce\xc3n\x93x\x14\xd8\x16\xb0(\x15q)\x8b\x19\xf0U\xe4\xb10\x08V\xa8\
+\x99\xf3\xdd\xde\xad\x06t\x0e\x83\xa7\xab\x9f\xcb:\xa7\x84&\x00\xe0HE\xab' )
+
+def getCheckBitmap():
+ return BitmapFromImage(getCheckImage())
+
+def getCheckImage():
+ stream = cStringIO.StringIO(getCheckData())
+ return ImageFromStream(stream)
+
+
+
+class CheckListCtrlMixin:
+ """
+ This is a mixin for ListCtrl which add a checkbox in the first
+ column of each row. It is inspired by limodou's CheckList.py(which
+ can be got from his NewEdit) and improved:
+
+ - You can just use InsertStringItem() to insert new items;
+
+ - Once a checkbox is checked/unchecked, the corresponding item
+ is not selected;
+
+ - You can use SetItemData() and GetItemData();
+
+ - Interfaces are changed to OnCheckItem(), IsChecked(),
+ CheckItem().
+
+ You should not set a imagelist for the ListCtrl once this mixin is used.
+ """
+ def __init__(self, check_image=None, uncheck_image=None):
+ self.__imagelist_ = wx.ImageList(16, 16)
+ if not check_image:
+ check_image = getCheckBitmap()
+ if not uncheck_image:
+ uncheck_image = getUncheckBitmap()
+ self.uncheck_image = self.__imagelist_.Add(uncheck_image)
+ self.check_image = self.__imagelist_.Add(check_image)
+ self.SetImageList(self.__imagelist_, wx.IMAGE_LIST_SMALL)
+ self.__last_check_ = None
+
+ self.Bind(wx.EVT_LEFT_DOWN, self.__OnLeftDown_)
+
+ # override the default methods of ListCtrl/ListView
+ self.InsertStringItem = self.__InsertStringItem_
+
+ # NOTE: if you use InsertItem, InsertImageItem or InsertImageStringItem,
+ # you must set the image yourself.
+ def __InsertStringItem_(self, index, label):
+ index = self.InsertImageStringItem(index, label, 0)
+ return index
+
+ def __OnLeftDown_(self, evt):
+ (index, flags) = self.HitTest(evt.GetPosition())
+ if flags == wx.LIST_HITTEST_ONITEMICON:
+ img_idx = self.GetItem(index).GetImage()
+ flag_check = img_idx == 0
+ begin_index = index
+ end_index = index
+ if self.__last_check_ is not None \
+ and wx.GetKeyState(wx.WXK_SHIFT):
+ last_index, last_flag_check = self.__last_check_
+ if last_flag_check == flag_check:
+ # XXX what if the previous item is deleted or new items
+ # are inserted?
+ item_count = self.GetItemCount()
+ if last_index < item_count:
+ if last_index < index:
+ begin_index = last_index
+ end_index = index
+ elif last_index > index:
+ begin_index = index
+ end_index = last_index
+ else:
+ assert False
+ while begin_index <= end_index:
+ self.CheckItem(begin_index, flag_check)
+ begin_index += 1
+ self.__last_check_ = (index, flag_check)
+ else:
+ evt.Skip()
+
+ def OnCheckItem(self, index, flag):
+ pass
+
+ def IsChecked(self, index):
+ return self.GetItem(index).GetImage() == 1
+
+ def CheckItem(self, index, check = True):
+ img_idx = self.GetItem(index).GetImage()
+ if img_idx == 0 and check is True:
+ self.SetItemImage(index, 1)
+ self.OnCheckItem(index, True)
+ elif img_idx == 1 and check is False:
+ self.SetItemImage(index, 0)
+ self.OnCheckItem(index, False)
+
+ def ToggleItem(self, index):
+ self.CheckItem(index, not self.IsChecked(index))
+
+
+#----------------------------------------------------------------------------
projects / wxWidgets.git / blobdiff