wxPython/wx/lib/gridmovers.py

   1 #----------------------------------------------------------------------------
   2 # Name:         GridColMover.py
   3 # Purpose:      Grid Column Mover Extension
   4 #
   5 # Author:       Gerrit van Dyk (email: gerritvd@decillion.net)
   6 #
   7 # Version       0.1
   8 # Date:         Nov 19, 2002
   9 # RCS-ID:       $Id$
  10 # Licence:      wxWindows license
  11 #----------------------------------------------------------------------------
  12 # 12/07/2003 - Jeff Grimmett (grimmtooth@softhome.net)
  13 #
  14 # o 2.5 Compatability changes
  15 #
  16 # 12/18/2003 - Jeff Grimmett (grimmtooth@softhome.net)
  17 #
  18 # o wxGridColMoveEvent -> GridColMoveEvent
  19 # o wxGridRowMoveEvent -> GridRowMoveEvent
  20 # o wxGridColMover -> GridColMover
  21 # o wxGridRowMover -> GridRowMover
  22 #
  23
  24
  25 import  wx
  26 import  wx.grid
  27
  28 #----------------------------------------------------------------------------
  29 # event class and macros
  30 #
  31 # New style 12/7/03
  32 #
  33
  34 wxEVT_COMMAND_GRID_COL_MOVE = wx.NewEventType()
  35 wxEVT_COMMAND_GRID_ROW_MOVE = wx.NewEventType()
  36
  37 EVT_GRID_COL_MOVE = wx.PyEventBinder(wxEVT_COMMAND_GRID_COL_MOVE, 1)
  38 EVT_GRID_ROW_MOVE = wx.PyEventBinder(wxEVT_COMMAND_GRID_ROW_MOVE, 1)
  39
  40 #----------------------------------------------------------------------------
  41
  42 class GridColMoveEvent(wx.PyCommandEvent):
  43     def __init__(self, id, dCol, bCol):
  44         wx.PyCommandEvent.__init__(self, id = id)
  45         self.SetEventType(wxEVT_COMMAND_GRID_COL_MOVE)
  46         self.moveColumn = dCol
  47         self.beforeColumn = bCol
  48
  49     def GetMoveColumn(self):
  50         return self.moveColumn
  51
  52     def GetBeforeColumn(self):
  53         return self.beforeColumn
  54
  55
  56 class GridRowMoveEvent(wx.PyCommandEvent):
  57     def __init__(self, id, dRow, bRow):
  58         wx.PyCommandEvent.__init__(self,id = id)
  59         self.SetEventType(wxEVT_COMMAND_GRID_ROW_MOVE)
  60         self.moveRow = dRow
  61         self.beforeRow = bRow
  62
  63     def GetMoveRow(self):
  64         return self.moveRow
  65
  66     def GetBeforeRow(self):
  67         return self.beforeRow
  68
  69
  70 #----------------------------------------------------------------------------
  71 # graft new methods into the wxGrid class
  72
  73 def _ColToRect(self,col):
  74     if self.GetNumberRows() > 0:
  75         rect = self.CellToRect(0,col)
  76     else:
  77         rect = wxRect()
  78         rect.height = self.GetColLabelSize()
  79         rect.width = self.GetColSize(col)
  80
  81         for cCol in range(0,col):
  82             rect.x += self.GetColSize(cCol)
  83
  84     rect.y = self.GetGridColLabelWindow().GetPosition()[1]
  85     return rect
  86
  87 wx.grid.Grid.ColToRect = _ColToRect
  88
  89
  90 def _RowToRect(self,row):
  91     if self.GetNumberCols() > 0:
  92         rect = self.CellToRect(row,0)
  93     else:
  94         rect = wxRect()
  95         rect.width = self.GetRowLabelSize()
  96         rect.height = self.GetRowSize(row)
  97
  98         for cRow in range(0,row):
  99             rect.y += self.GetRowSize(cRow)
 100
 101     rect.x = self.GetGridRowLabelWindow().GetPosition()[0]
 102     return rect
 103
 104 wx.grid.Grid.RowToRect = _RowToRect
 105
 106
 107 #----------------------------------------------------------------------------
 108
 109 class ColDragWindow(wx.Window):
 110     def __init__(self,parent,image,dragCol):
 111         wx.Window.__init__(self,parent,-1, style=wx.SIMPLE_BORDER)
 112         self.image = image
 113         self.SetSize((self.image.GetWidth(),self.image.GetHeight()))
 114         self.ux = parent.GetScrollPixelsPerUnit()[0]
 115         self.moveColumn = dragCol
 116
 117         self.Bind(wx.EVT_PAINT, self.OnPaint)
 118
 119     def DisplayAt(self,pos,y):
 120         x = self.GetPositionTuple()[0]
 121         if x == pos:
 122             self.Refresh()              # Need to display insertion point
 123         else:
 124             self.MoveXY(pos,y)
 125
 126     def GetMoveColumn(self):
 127         return self.moveColumn
 128
 129     def _GetInsertionInfo(self):
 130         parent = self.GetParent()
 131         sx = parent.GetViewStart()[0] * self.ux
 132         sx -= parent._rlSize
 133         x = self.GetPosition()[0]
 134         w = self.GetSize()[0]
 135         sCol = parent.XToCol(x + sx)
 136         eCol = parent.XToCol(x + w + sx)
 137         iPos = xPos = xCol = 99999
 138         centerPos = x + sx + (w / 2)
 139
 140         for col in range(sCol,eCol + 1):
 141             cx = parent.ColToRect(col)[0]
 142
 143             if abs(cx - centerPos) < iPos:
 144                 iPos = abs(cx - centerPos)
 145                 xCol = col
 146                 xPos = cx
 147
 148         if xCol < 0 or xCol > parent.GetNumberCols():
 149             xCol = parent.GetNumberCols()
 150
 151         return (xPos - sx - x,xCol)
 152
 153     def GetInsertionColumn(self):
 154         return self._GetInsertionInfo()[1]
 155
 156     def GetInsertionPos(self):
 157         return self._GetInsertionInfo()[0]
 158
 159     def OnPaint(self,evt):
 160         dc = wx.PaintDC(self)
 161         w,h = self.GetSize()
 162         dc.DrawBitmap(self.image, (0,0))
 163         dc.SetPen(wx.Pen(wx.BLACK,1,wx.SOLID))
 164         dc.SetBrush(wx.TRANSPARENT_BRUSH)
 165         dc.DrawRectangle((0,0), (w,h))
 166         iPos = self.GetInsertionPos()
 167         dc.DrawLine((iPos,h - 10), (iPos,h))
 168
 169
 170
 171
 172 class RowDragWindow(wx.Window):
 173     def __init__(self,parent,image,dragRow):
 174         wx.Window.__init__(self,parent,-1, style=wx.SIMPLE_BORDER)
 175         self.image = image
 176         self.SetSize((self.image.GetWidth(),self.image.GetHeight()))
 177         self.uy = parent.GetScrollPixelsPerUnit()[1]
 178         self.moveRow = dragRow
 179
 180         self.Bind(wx.EVT_PAINT, self.OnPaint)
 181
 182     def DisplayAt(self,x,pos):
 183         y = self.GetPosition()[1]
 184         if y == pos:
 185             self.Refresh()              # Need to display insertion point
 186         else:
 187             self.MoveXY(x,pos)
 188
 189     def GetMoveRow(self):
 190         return self.moveRow
 191
 192     def _GetInsertionInfo(self):
 193         parent = self.GetParent()
 194         sy = parent.GetViewStart()[1] * self.uy
 195         sy -= parent._clSize
 196         y = self.GetPosition()[1]
 197         h = self.GetSize()[1]
 198         sRow = parent.YToRow(y + sy)
 199         eRow = parent.YToRow(y + h + sy)
 200         iPos = yPos = yRow = 99999
 201         centerPos = y + sy + (h / 2)
 202
 203         for row in range(sRow,eRow + 1):
 204             cy = parent.RowToRect(row)[1]
 205
 206             if abs(cy - centerPos) < iPos:
 207                 iPos = abs(cy - centerPos)
 208                 yRow = row
 209                 yPos = cy
 210
 211         if yRow < 0 or yRow > parent.GetNumberRows():
 212             yRow = parent.GetNumberRows()
 213
 214         return (yPos - sy - y,yRow)
 215
 216     def GetInsertionRow(self):
 217         return self._GetInsertionInfo()[1]
 218
 219     def GetInsertionPos(self):
 220         return self._GetInsertionInfo()[0]
 221
 222     def OnPaint(self,evt):
 223         dc = wx.PaintDC(self)
 224         w,h = self.GetSize()
 225         dc.DrawBitmap(self.image, (0,0))
 226         dc.SetPen(wx.Pen(wx.BLACK,1,wx.SOLID))
 227         dc.SetBrush(wx.TRANSPARENT_BRUSH)
 228         dc.DrawRectangle((0,0), (w,h))
 229         iPos = self.GetInsertionPos()
 230         dc.DrawLine((w - 10,iPos), (w,iPos))
 231
 232 #----------------------------------------------------------------------------
 233
 234 class GridColMover(wx.EvtHandler):
 235     def __init__(self,grid):
 236         wx.EvtHandler.__init__(self)
 237
 238         self.grid = grid
 239         self.grid._rlSize = self.grid.GetRowLabelSize()
 240         self.lwin = grid.GetGridColLabelWindow()
 241         self.lwin.PushEventHandler(self)
 242         self.colWin = None
 243         self.ux = self.grid.GetScrollPixelsPerUnit()[0]
 244         self.startX = -10
 245         self.cellX = 0
 246         self.didMove = False
 247         self.isDragging = False
 248
 249         self.Bind(wx.EVT_MOTION, self.OnMouseMove)
 250         self.Bind(wx.EVT_LEFT_DOWN, self.OnPress)
 251         self.Bind(wx.EVT_LEFT_UP, self.OnRelease)
 252
 253     def OnMouseMove(self,evt):
 254         if self.isDragging:
 255             if abs(self.startX - evt.m_x) >= 3:
 256                 self.didMove = True
 257                 sx,y = self.grid.GetViewStart()
 258                 w,h = self.lwin.GetClientSize()
 259                 x = sx * self.ux
 260
 261                 if (evt.m_x + x) < x:
 262                     x = evt.m_x + x
 263                 elif evt.m_x > w:
 264                     x += evt.m_x - w
 265
 266                 if x < 1: x = 0
 267                 else: x /= self.ux
 268
 269                 if x != sx:
 270                     if wx.Platform == '__WXMSW__':
 271                         self.colWin.Show(False)
 272
 273                     self.grid.Scroll(x,y)
 274
 275                 x,y = self.lwin.ClientToScreenXY(evt.m_x,0)
 276                 x,y = self.grid.ScreenToClientXY(x,y)
 277
 278                 if not self.colWin.IsShown():
 279                     self.colWin.Show(True)
 280
 281                 px = x - self.cellX
 282
 283                 if px < 0 + self.grid._rlSize: px = 0 + self.grid._rlSize
 284
 285                 if px > w - self.colWin.GetSize()[0] + self.grid._rlSize:
 286                     px = w - self.colWin.GetSize()[0] + self.grid._rlSize
 287
 288                 self.colWin.DisplayAt(px,y)
 289                 return
 290
 291         evt.Skip()
 292
 293     def OnPress(self,evt):
 294         self.startX = evt.m_x
 295         sx = self.grid.GetViewStart()[0] * self.ux
 296         sx -= self.grid._rlSize
 297         px,py = self.lwin.ClientToScreenXY(evt.m_x,evt.m_y)
 298         px,py = self.grid.ScreenToClientXY(px,py)
 299
 300         if self.grid.XToEdgeOfCol(px + sx) != wx.NOT_FOUND:
 301             evt.Skip()
 302             return
 303
 304         self.isDragging = True
 305         self.didMove = False
 306         col = self.grid.XToCol(px + sx)
 307         rect = self.grid.ColToRect(col)
 308         self.cellX = px + sx - rect.x
 309         size = self.lwin.GetSize()
 310         rect.y = 0
 311         rect.x -= sx + self.grid._rlSize
 312         rect.height = size[1]
 313         colImg = self._CaptureImage(rect)
 314         self.colWin = ColDragWindow(self.grid,colImg,col)
 315         self.colWin.Show(False)
 316         self.lwin.CaptureMouse()
 317
 318     def OnRelease(self,evt):
 319         if self.isDragging:
 320             self.lwin.ReleaseMouse()
 321             self.colWin.Show(False)
 322             self.isDragging = False
 323
 324             if not self.didMove:
 325                 px = self.lwin.ClientToScreenXY(self.startX,0)[0]
 326                 px = self.grid.ScreenToClientXY(px,0)[0]
 327                 sx = self.grid.GetViewStart()[0] * self.ux
 328                 sx -= self.grid._rlSize
 329                 col = self.grid.XToCol(px+sx)
 330
 331                 if col != wx.NOT_FOUND:
 332                     self.grid.SelectCol(col,evt.m_controlDown)
 333
 334                 return
 335             else:
 336                 bCol = self.colWin.GetInsertionColumn()
 337                 dCol = self.colWin.GetMoveColumn()
 338                 wx.PostEvent(self,
 339                              GridColMoveEvent(self.grid.GetId(), dCol, bCol))
 340
 341             self.colWin.Destroy()
 342         evt.Skip()
 343
 344     def _CaptureImage(self,rect):
 345         bmp = wx.EmptyBitmap(rect.width,rect.height)
 346         memdc = wx.MemoryDC()
 347         memdc.SelectObject(bmp)
 348         dc = wx.WindowDC(self.lwin)
 349         memdc.Blit((0,0), rect.GetSize(), dc, rect.GetPosition())
 350         memdc.SelectObject(wx.NullBitmap)
 351         return bmp
 352
 353
 354 class GridRowMover(wx.EvtHandler):
 355     def __init__(self,grid):
 356         wx.EvtHandler.__init__(self)
 357
 358         self.grid = grid
 359         self.grid._clSize = self.grid.GetColLabelSize()
 360         self.lwin = grid.GetGridRowLabelWindow()
 361         self.lwin.PushEventHandler(self)
 362         self.rowWin = None
 363         self.uy = self.grid.GetScrollPixelsPerUnit()[1]
 364         self.startY = -10
 365         self.cellY = 0
 366         self.didMove = False
 367         self.isDragging = False
 368
 369         self.Bind(wx.EVT_MOTION, self.OnMouseMove)
 370         self.Bind(wx.EVT_LEFT_DOWN, self.OnPress)
 371         self.Bind(wx.EVT_LEFT_UP, self.OnRelease)
 372
 373     def OnMouseMove(self,evt):
 374         if self.isDragging:
 375             if abs(self.startY - evt.m_y) >= 3:
 376                 self.didMove = True
 377                 x,sy = self.grid.GetViewStart()
 378                 w,h = self.lwin.GetClientSizeTuple()
 379                 y = sy * self.uy
 380
 381                 if (evt.m_y + y) < y:
 382                     y = evt.m_y + y
 383                 elif evt.m_y > h:
 384                     y += evt.m_y - h
 385
 386                 if y < 1:
 387                     y = 0
 388                 else:
 389                     y /= self.uy
 390
 391                 if y != sy:
 392                     if wx.Platform == '__WXMSW__':
 393                         self.rowWin.Show(False)
 394
 395                     self.grid.Scroll(x,y)
 396
 397                 x,y = self.lwin.ClientToScreenXY(0,evt.m_y)
 398                 x,y = self.grid.ScreenToClientXY(x,y)
 399
 400                 if not self.rowWin.IsShown():
 401                     self.rowWin.Show(True)
 402
 403                 py = y - self.cellY
 404
 405                 if py < 0 + self.grid._clSize:
 406                     py = 0 + self.grid._clSize
 407
 408                 if py > h - self.rowWin.GetSize()[1] + self.grid._clSize:
 409                     py = h - self.rowWin.GetSize()[1] + self.grid._clSize
 410
 411                 self.rowWin.DisplayAt(x,py)
 412                 return
 413
 414         evt.Skip()
 415
 416     def OnPress(self,evt):
 417         self.startY = evt.m_y
 418         sy = self.grid.GetViewStart()[1] * self.uy
 419         sy -= self.grid._clSize
 420         px,py = self.lwin.ClientToScreenXY(evt.m_x,evt.m_y)
 421         px,py = self.grid.ScreenToClientXY(px,py)
 422
 423         if self.grid.YToEdgeOfRow(py + sy) != wx.NOT_FOUND:
 424             evt.Skip()
 425             return
 426
 427         self.isDragging = True
 428         self.didMove = False
 429         row = self.grid.YToRow(py + sy)
 430         rect = self.grid.RowToRect(row)
 431         self.cellY = py + sy - rect.y
 432         size = self.lwin.GetSize()
 433         rect.x = 0
 434         rect.y -= sy + self.grid._clSize
 435         rect.width = size[0]
 436         rowImg = self._CaptureImage(rect)
 437         self.rowWin = RowDragWindow(self.grid,rowImg,row)
 438         self.rowWin.Show(False)
 439         self.lwin.CaptureMouse()
 440
 441     def OnRelease(self,evt):
 442         if self.isDragging:
 443             self.lwin.ReleaseMouse()
 444             self.rowWin.Show(False)
 445             self.isDragging = False
 446
 447             if not self.didMove:
 448                 py = self.lwin.ClientToScreenXY(0,self.startY)[1]
 449                 py = self.grid.ScreenToClientXY(0,py)[1]
 450                 sy = self.grid.GetViewStart()[1] * self.uy
 451                 sy -= self.grid._clSize
 452                 row = self.grid.YToRow(py + sy)
 453
 454                 if row != wx.NOT_FOUND:
 455                     self.grid.SelectRow(row,evt.m_controlDown)
 456                 return
 457             else:
 458                 bRow = self.rowWin.GetInsertionRow()
 459                 dRow = self.rowWin.GetMoveRow()
 460
 461                 wx.PostEvent(self,
 462                              GridRowMoveEvent(self.grid.GetId(), dRow, bRow))
 463
 464             self.rowWin.Destroy()
 465         evt.Skip()
 466
 467     def _CaptureImage(self,rect):
 468         bmp = wx.EmptyBitmap(rect.width,rect.height)
 469         memdc = wx.MemoryDC()
 470         memdc.SelectObject(bmp)
 471         dc = wx.WindowDC(self.lwin)
 472         memdc.Blit((0,0), rect.GetSize(), dc, rect.GetPosition())
 473         memdc.SelectObject(wx.NullBitmap)
 474         return bmp
 475
 476
 477 #----------------------------------------------------------------------------