[wxWidgets.git] / wxPython / wx / lib / gridmovers.py

#----------------------------------------------------------------------------
# Name:         GridColMover.py
# Purpose:      Grid Column Mover Extension
#
# Author:       Gerrit van Dyk (email: gerritvd@decillion.net)
#
# Version       0.1
# Date:         Nov 19, 2002
# RCS-ID:       $Id$
# Licence:      wxWindows license
#----------------------------------------------------------------------------
# 12/07/2003 - Jeff Grimmett (grimmtooth@softhome.net)
#
# o 2.5 Compatability changes
#
# 12/18/2003 - Jeff Grimmett (grimmtooth@softhome.net)
#
# o wxGridColMoveEvent -> GridColMoveEvent
# o wxGridRowMoveEvent -> GridRowMoveEvent
# o wxGridColMover -> GridColMover
# o wxGridRowMover -> GridRowMover
#


import  wx
import  wx.grid

#----------------------------------------------------------------------------
# event class and macros
# 
# New style 12/7/03
#

wxEVT_COMMAND_GRID_COL_MOVE = wx.NewEventType()
wxEVT_COMMAND_GRID_ROW_MOVE = wx.NewEventType()

EVT_GRID_COL_MOVE = wx.PyEventBinder(wxEVT_COMMAND_GRID_COL_MOVE, 1)
EVT_GRID_ROW_MOVE = wx.PyEventBinder(wxEVT_COMMAND_GRID_ROW_MOVE, 1)

#----------------------------------------------------------------------------

class GridColMoveEvent(wx.PyCommandEvent):
    def __init__(self, id, dCol, bCol):
        wx.PyCommandEvent.__init__(self, id = id)
        self.SetEventType(wxEVT_COMMAND_GRID_COL_MOVE)
        self.moveColumn = dCol
        self.beforeColumn = bCol

    def GetMoveColumn(self):
        return self.moveColumn

    def GetBeforeColumn(self):
        return self.beforeColumn


class GridRowMoveEvent(wx.PyCommandEvent):
    def __init__(self, id, dRow, bRow):
        wx.PyCommandEvent.__init__(self,id = id)
        self.SetEventType(wxEVT_COMMAND_GRID_ROW_MOVE)
        self.moveRow = dRow
        self.beforeRow = bRow

    def GetMoveRow(self):
        return self.moveRow

    def GetBeforeRow(self):
        return self.beforeRow


#----------------------------------------------------------------------------
# graft new methods into the wxGrid class

def _ColToRect(self,col):
    if self.GetNumberRows() > 0:
        rect = self.CellToRect(0,col)
    else:
        rect = wxRect()
        rect.height = self.GetColLabelSize()
        rect.width = self.GetColSize(col)

        for cCol in range(0,col):
            rect.x += self.GetColSize(cCol)

    rect.y = self.GetGridColLabelWindow().GetPosition()[1]
    return rect

wx.grid.Grid.ColToRect = _ColToRect


def _RowToRect(self,row):
    if self.GetNumberCols() > 0:
        rect = self.CellToRect(row,0)
    else:
        rect = wxRect()
        rect.width = self.GetRowLabelSize()
        rect.height = self.GetRowSize(row)

        for cRow in range(0,row):
            rect.y += self.GetRowSize(cRow)

    rect.x = self.GetGridRowLabelWindow().GetPosition()[0]
    return rect

wx.grid.Grid.RowToRect = _RowToRect


#----------------------------------------------------------------------------

class ColDragWindow(wx.Window):
    def __init__(self,parent,image,dragCol):
        wx.Window.__init__(self,parent,-1, style=wx.SIMPLE_BORDER)
        self.image = image
        self.SetSize((self.image.GetWidth(),self.image.GetHeight()))
        self.ux = parent.GetScrollPixelsPerUnit()[0]
        self.moveColumn = dragCol

        self.Bind(wx.EVT_PAINT, self.OnPaint)

    def DisplayAt(self,pos,y):
        x = self.GetPositionTuple()[0]
        if x == pos:
            self.Refresh()              # Need to display insertion point
        else:
            self.MoveXY(pos,y)

    def GetMoveColumn(self):
        return self.moveColumn

    def _GetInsertionInfo(self):
        parent = self.GetParent()
        sx = parent.GetViewStart()[0] * self.ux
        sx -= parent._rlSize
        x = self.GetPosition()[0]
        w = self.GetSize()[0]
        sCol = parent.XToCol(x + sx)
        eCol = parent.XToCol(x + w + sx)
        iPos = xPos = xCol = 99999
        centerPos = x + sx + (w / 2)

        for col in range(sCol,eCol + 1):
            cx = parent.ColToRect(col)[0]

            if abs(cx - centerPos) < iPos:
                iPos = abs(cx - centerPos)
                xCol = col
                xPos = cx

        if xCol < 0 or xCol > parent.GetNumberCols():
            xCol = parent.GetNumberCols()

        return (xPos - sx - x,xCol)

    def GetInsertionColumn(self):
        return self._GetInsertionInfo()[1]

    def GetInsertionPos(self):
        return self._GetInsertionInfo()[0]

    def OnPaint(self,evt):
        dc = wx.PaintDC(self)
        w,h = self.GetSize()
        dc.DrawBitmap(self.image, (0,0))
        dc.SetPen(wx.Pen(wx.BLACK,1,wx.SOLID))
        dc.SetBrush(wx.TRANSPARENT_BRUSH)
        dc.DrawRectangle((0,0), (w,h))
        iPos = self.GetInsertionPos()
        dc.DrawLine((iPos,h - 10), (iPos,h))


class RowDragWindow(wx.Window):
    def __init__(self,parent,image,dragRow):
        wx.Window.__init__(self,parent,-1, style=wx.SIMPLE_BORDER)
        self.image = image
        self.SetSize((self.image.GetWidth(),self.image.GetHeight()))
        self.uy = parent.GetScrollPixelsPerUnit()[1]
        self.moveRow = dragRow

        self.Bind(wx.EVT_PAINT, self.OnPaint)

    def DisplayAt(self,x,pos):
        y = self.GetPosition()[1]
        if y == pos:
            self.Refresh()              # Need to display insertion point
        else:
            self.MoveXY(x,pos)

    def GetMoveRow(self):
        return self.moveRow

    def _GetInsertionInfo(self):
        parent = self.GetParent()
        sy = parent.GetViewStart()[1] * self.uy
        sy -= parent._clSize
        y = self.GetPosition()[1]
        h = self.GetSize()[1]
        sRow = parent.YToRow(y + sy)
        eRow = parent.YToRow(y + h + sy)
        iPos = yPos = yRow = 99999
        centerPos = y + sy + (h / 2)

        for row in range(sRow,eRow + 1):
            cy = parent.RowToRect(row)[1]

            if abs(cy - centerPos) < iPos:
                iPos = abs(cy - centerPos)
                yRow = row
                yPos = cy

        if yRow < 0 or yRow > parent.GetNumberRows():
            yRow = parent.GetNumberRows()

        return (yPos - sy - y,yRow)

    def GetInsertionRow(self):
        return self._GetInsertionInfo()[1]

    def GetInsertionPos(self):
        return self._GetInsertionInfo()[0]

    def OnPaint(self,evt):
        dc = wx.PaintDC(self)
        w,h = self.GetSize()
        dc.DrawBitmap(self.image, (0,0))
        dc.SetPen(wx.Pen(wx.BLACK,1,wx.SOLID))
        dc.SetBrush(wx.TRANSPARENT_BRUSH)
        dc.DrawRectangle((0,0), (w,h))
        iPos = self.GetInsertionPos()
        dc.DrawLine((w - 10,iPos), (w,iPos))

#----------------------------------------------------------------------------

class GridColMover(wx.EvtHandler):
    def __init__(self,grid):
        wx.EvtHandler.__init__(self)

        self.grid = grid
        self.grid._rlSize = self.grid.GetRowLabelSize()
        self.lwin = grid.GetGridColLabelWindow()
        self.lwin.PushEventHandler(self)
        self.colWin = None
        self.ux = self.grid.GetScrollPixelsPerUnit()[0]
        self.startX = -10
        self.cellX = 0
        self.didMove = False
        self.isDragging = False

        self.Bind(wx.EVT_MOTION, self.OnMouseMove)
        self.Bind(wx.EVT_LEFT_DOWN, self.OnPress)
        self.Bind(wx.EVT_LEFT_UP, self.OnRelease)

    def OnMouseMove(self,evt):
        if self.isDragging:
            if abs(self.startX - evt.m_x) >= 3:
                self.didMove = True
                sx,y = self.grid.GetViewStart()
                w,h = self.lwin.GetClientSize()
                x = sx * self.ux

                if (evt.m_x + x) < x:
                    x = evt.m_x + x
                elif evt.m_x > w:
                    x += evt.m_x - w

                if x < 1: x = 0
                else: x /= self.ux

                if x != sx:
                    if wx.Platform == '__WXMSW__':
                        self.colWin.Show(False)

                    self.grid.Scroll(x,y)

                x,y = self.lwin.ClientToScreenXY(evt.m_x,0)
                x,y = self.grid.ScreenToClientXY(x,y)

                if not self.colWin.IsShown():
                    self.colWin.Show(True)

                px = x - self.cellX

                if px < 0 + self.grid._rlSize: px = 0 + self.grid._rlSize

                if px > w - self.colWin.GetSize()[0] + self.grid._rlSize:
                    px = w - self.colWin.GetSize()[0] + self.grid._rlSize

                self.colWin.DisplayAt(px,y)
                return

        evt.Skip()

    def OnPress(self,evt):
        self.startX = evt.m_x
        sx = self.grid.GetViewStart()[0] * self.ux
        sx -= self.grid._rlSize
        px,py = self.lwin.ClientToScreenXY(evt.m_x,evt.m_y)
        px,py = self.grid.ScreenToClientXY(px,py)

        if self.grid.XToEdgeOfCol(px + sx) != wx.NOT_FOUND:
            evt.Skip()
            return

        self.isDragging = True
        self.didMove = False
        col = self.grid.XToCol(px + sx)
        rect = self.grid.ColToRect(col)
        self.cellX = px + sx - rect.x
        size = self.lwin.GetSize()
        rect.y = 0
        rect.x -= sx + self.grid._rlSize
        rect.height = size[1]
        colImg = self._CaptureImage(rect)
        self.colWin = ColDragWindow(self.grid,colImg,col)
        self.colWin.Show(False)
        self.lwin.CaptureMouse()

    def OnRelease(self,evt):
        if self.isDragging:
            self.lwin.ReleaseMouse()
            self.colWin.Show(False)
            self.isDragging = False

            if not self.didMove:
                px = self.lwin.ClientToScreenXY(self.startX,0)[0]
                px = self.grid.ScreenToClientXY(px,0)[0]
                sx = self.grid.GetViewStart()[0] * self.ux
                sx -= self.grid._rlSize
                col = self.grid.XToCol(px+sx)

                if col != wx.NOT_FOUND:
                    self.grid.SelectCol(col,evt.m_controlDown)

                return
            else:
                bCol = self.colWin.GetInsertionColumn()
                dCol = self.colWin.GetMoveColumn()
                wx.PostEvent(self,
                             GridColMoveEvent(self.grid.GetId(), dCol, bCol))

            self.colWin.Destroy()
        evt.Skip()

    def _CaptureImage(self,rect):
        bmp = wx.EmptyBitmap(rect.width,rect.height)
        memdc = wx.MemoryDC()
        memdc.SelectObject(bmp)
        dc = wx.WindowDC(self.lwin)
        memdc.Blit((0,0), rect.GetSize(), dc, rect.GetPosition())
        memdc.SelectObject(wx.NullBitmap)
        return bmp


class GridRowMover(wx.EvtHandler):
    def __init__(self,grid):
        wx.EvtHandler.__init__(self)

        self.grid = grid
        self.grid._clSize = self.grid.GetColLabelSize()
        self.lwin = grid.GetGridRowLabelWindow()
        self.lwin.PushEventHandler(self)
        self.rowWin = None
        self.uy = self.grid.GetScrollPixelsPerUnit()[1]
        self.startY = -10
        self.cellY = 0
        self.didMove = False
        self.isDragging = False

        self.Bind(wx.EVT_MOTION, self.OnMouseMove)
        self.Bind(wx.EVT_LEFT_DOWN, self.OnPress)
        self.Bind(wx.EVT_LEFT_UP, self.OnRelease)

    def OnMouseMove(self,evt):
        if self.isDragging:
            if abs(self.startY - evt.m_y) >= 3:
                self.didMove = True
                x,sy = self.grid.GetViewStart()
                w,h = self.lwin.GetClientSizeTuple()
                y = sy * self.uy

                if (evt.m_y + y) < y:
                    y = evt.m_y + y
                elif evt.m_y > h:
                    y += evt.m_y - h

                if y < 1: 
                    y = 0
                else: 
                    y /= self.uy

                if y != sy:
                    if wx.Platform == '__WXMSW__':
                        self.rowWin.Show(False)

                    self.grid.Scroll(x,y)

                x,y = self.lwin.ClientToScreenXY(0,evt.m_y)
                x,y = self.grid.ScreenToClientXY(x,y)

                if not self.rowWin.IsShown():
                    self.rowWin.Show(True)

                py = y - self.cellY

                if py < 0 + self.grid._clSize: 
                    py = 0 + self.grid._clSize

                if py > h - self.rowWin.GetSize()[1] + self.grid._clSize:
                    py = h - self.rowWin.GetSize()[1] + self.grid._clSize

                self.rowWin.DisplayAt(x,py)
                return

        evt.Skip()

    def OnPress(self,evt):
        self.startY = evt.m_y
        sy = self.grid.GetViewStart()[1] * self.uy
        sy -= self.grid._clSize
        px,py = self.lwin.ClientToScreenXY(evt.m_x,evt.m_y)
        px,py = self.grid.ScreenToClientXY(px,py)

        if self.grid.YToEdgeOfRow(py + sy) != wx.NOT_FOUND:
            evt.Skip()
            return

        self.isDragging = True
        self.didMove = False
        row = self.grid.YToRow(py + sy)
        rect = self.grid.RowToRect(row)
        self.cellY = py + sy - rect.y
        size = self.lwin.GetSize()
        rect.x = 0
        rect.y -= sy + self.grid._clSize
        rect.width = size[0]
        rowImg = self._CaptureImage(rect)
        self.rowWin = RowDragWindow(self.grid,rowImg,row)
        self.rowWin.Show(False)
        self.lwin.CaptureMouse()

    def OnRelease(self,evt):
        if self.isDragging:
            self.lwin.ReleaseMouse()
            self.rowWin.Show(False)
            self.isDragging = False

            if not self.didMove:
                py = self.lwin.ClientToScreenXY(0,self.startY)[1]
                py = self.grid.ScreenToClientXY(0,py)[1]
                sy = self.grid.GetViewStart()[1] * self.uy
                sy -= self.grid._clSize
                row = self.grid.YToRow(py + sy)

                if row != wx.NOT_FOUND:
                    self.grid.SelectRow(row,evt.m_controlDown)
                return
            else:
                bRow = self.rowWin.GetInsertionRow()
                dRow = self.rowWin.GetMoveRow()

                wx.PostEvent(self,
                             GridRowMoveEvent(self.grid.GetId(), dRow, bRow))

            self.rowWin.Destroy()
        evt.Skip()

    def _CaptureImage(self,rect):
        bmp = wx.EmptyBitmap(rect.width,rect.height)
        memdc = wx.MemoryDC()
        memdc.SelectObject(bmp)
        dc = wx.WindowDC(self.lwin)
        memdc.Blit((0,0), rect.GetSize(), dc, rect.GetPosition())
        memdc.SelectObject(wx.NullBitmap)
        return bmp


#----------------------------------------------------------------------------
Commit	Line	Data
d14a1e28 RD	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	#----------------------------------------------------------------------------
b881fc78 RD	12	# 12/07/2003 - Jeff Grimmett (grimmtooth@softhome.net)
	13	#
	14	# o 2.5 Compatability changes
	15	#
33785d9f RD	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
1fded56b	24
b881fc78 RD	25	import wx
b881fc78 RD	26	import wx.grid
1fded56b	27
d14a1e28	28	#----------------------------------------------------------------------------
b881fc78 RD	29	# event class and macros
	30	#
	31	# New style 12/7/03
	32	#
d14a1e28	33
b881fc78 RD	34	wxEVT_COMMAND_GRID_COL_MOVE = wx.NewEventType()
b881fc78 RD	35	wxEVT_COMMAND_GRID_ROW_MOVE = wx.NewEventType()
d14a1e28	36
b881fc78 RD	37	EVT_GRID_COL_MOVE = wx.PyEventBinder(wxEVT_COMMAND_GRID_COL_MOVE, 1)
b881fc78 RD	38	EVT_GRID_ROW_MOVE = wx.PyEventBinder(wxEVT_COMMAND_GRID_ROW_MOVE, 1)
d14a1e28	39
b881fc78	40	#----------------------------------------------------------------------------
d14a1e28	41
33785d9f	42	class GridColMoveEvent(wx.PyCommandEvent):
d14a1e28	43	def __init__(self, id, dCol, bCol):
b881fc78	44	wx.PyCommandEvent.__init__(self, id = id)
d14a1e28 RD	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
33785d9f	56	class GridRowMoveEvent(wx.PyCommandEvent):
d14a1e28	57	def __init__(self, id, dRow, bRow):
b881fc78	58	wx.PyCommandEvent.__init__(self,id = id)
d14a1e28 RD	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)
b881fc78	80
d14a1e28 RD	81	for cCol in range(0,col):
d14a1e28 RD	82	rect.x += self.GetColSize(cCol)
b881fc78	83
d14a1e28 RD	84	rect.y = self.GetGridColLabelWindow().GetPosition()[1]
	85	return rect
	86
b881fc78	87	wx.grid.Grid.ColToRect = _ColToRect
d14a1e28 RD	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)
b881fc78	97
d14a1e28 RD	98	for cRow in range(0,row):
d14a1e28 RD	99	rect.y += self.GetRowSize(cRow)
b881fc78	100
d14a1e28 RD	101	rect.x = self.GetGridRowLabelWindow().GetPosition()[0]
	102	return rect
	103
b881fc78	104	wx.grid.Grid.RowToRect = _RowToRect
d14a1e28 RD	105
	106
	107	#----------------------------------------------------------------------------
	108
b881fc78	109	class ColDragWindow(wx.Window):
d14a1e28	110	def __init__(self,parent,image,dragCol):
b881fc78	111	wx.Window.__init__(self,parent,-1, style=wx.SIMPLE_BORDER)
d14a1e28 RD	112	self.image = image
	113	self.SetSize((self.image.GetWidth(),self.image.GetHeight()))
	114	self.ux = parent.GetScrollPixelsPerUnit()[0]
	115	self.moveColumn = dragCol
	116
b881fc78	117	self.Bind(wx.EVT_PAINT, self.OnPaint)
d14a1e28 RD	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
b881fc78 RD	133	x = self.GetPosition()[0]
b881fc78 RD	134	w = self.GetSize()[0]
d14a1e28 RD	135	sCol = parent.XToCol(x + sx)
	136	eCol = parent.XToCol(x + w + sx)
	137	iPos = xPos = xCol = 99999
	138	centerPos = x + sx + (w / 2)
b881fc78	139
d14a1e28 RD	140	for col in range(sCol,eCol + 1):
d14a1e28 RD	141	cx = parent.ColToRect(col)[0]
b881fc78	142
d14a1e28 RD	143	if abs(cx - centerPos) < iPos:
	144	iPos = abs(cx - centerPos)
	145	xCol = col
	146	xPos = cx
b881fc78	147
d14a1e28 RD	148	if xCol < 0 or xCol > parent.GetNumberCols():
d14a1e28 RD	149	xCol = parent.GetNumberCols()
b881fc78	150
d14a1e28 RD	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):
b881fc78	160	dc = wx.PaintDC(self)
d14a1e28	161	w,h = self.GetSize()
7722248d	162	dc.DrawBitmap(self.image, (0,0))
b881fc78 RD	163	dc.SetPen(wx.Pen(wx.BLACK,1,wx.SOLID))
b881fc78 RD	164	dc.SetBrush(wx.TRANSPARENT_BRUSH)
7722248d	165	dc.DrawRectangle((0,0), (w,h))
d14a1e28	166	iPos = self.GetInsertionPos()
7722248d	167	dc.DrawLine((iPos,h - 10), (iPos,h))
d14a1e28 RD	168
	169
	170
	171
b881fc78	172	class RowDragWindow(wx.Window):
d14a1e28	173	def __init__(self,parent,image,dragRow):
b881fc78	174	wx.Window.__init__(self,parent,-1, style=wx.SIMPLE_BORDER)
d14a1e28 RD	175	self.image = image
	176	self.SetSize((self.image.GetWidth(),self.image.GetHeight()))
	177	self.uy = parent.GetScrollPixelsPerUnit()[1]
	178	self.moveRow = dragRow
	179
b881fc78	180	self.Bind(wx.EVT_PAINT, self.OnPaint)
d14a1e28 RD	181
d14a1e28 RD	182	def DisplayAt(self,x,pos):
b881fc78	183	y = self.GetPosition()[1]
d14a1e28 RD	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
b881fc78 RD	196	y = self.GetPosition()[1]
b881fc78 RD	197	h = self.GetSize()[1]
d14a1e28 RD	198	sRow = parent.YToRow(y + sy)
	199	eRow = parent.YToRow(y + h + sy)
	200	iPos = yPos = yRow = 99999
	201	centerPos = y + sy + (h / 2)
b881fc78	202
d14a1e28 RD	203	for row in range(sRow,eRow + 1):
d14a1e28 RD	204	cy = parent.RowToRect(row)[1]
b881fc78	205
d14a1e28 RD	206	if abs(cy - centerPos) < iPos:
	207	iPos = abs(cy - centerPos)
	208	yRow = row
	209	yPos = cy
b881fc78	210
d14a1e28 RD	211	if yRow < 0 or yRow > parent.GetNumberRows():
d14a1e28 RD	212	yRow = parent.GetNumberRows()
b881fc78	213
d14a1e28 RD	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):
b881fc78	223	dc = wx.PaintDC(self)
d14a1e28	224	w,h = self.GetSize()
7722248d	225	dc.DrawBitmap(self.image, (0,0))
b881fc78 RD	226	dc.SetPen(wx.Pen(wx.BLACK,1,wx.SOLID))
b881fc78 RD	227	dc.SetBrush(wx.TRANSPARENT_BRUSH)
7722248d	228	dc.DrawRectangle((0,0), (w,h))
d14a1e28	229	iPos = self.GetInsertionPos()
7722248d	230	dc.DrawLine((w - 10,iPos), (w,iPos))
d14a1e28 RD	231
	232	#----------------------------------------------------------------------------
	233
33785d9f	234	class GridColMover(wx.EvtHandler):
d14a1e28	235	def __init__(self,grid):
b881fc78	236	wx.EvtHandler.__init__(self)
d14a1e28 RD	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
b881fc78 RD	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)
d14a1e28 RD	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()
b881fc78	258	w,h = self.lwin.GetClientSize()
d14a1e28	259	x = sx * self.ux
b881fc78	260
d14a1e28 RD	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
b881fc78	265
d14a1e28 RD	266	if x < 1: x = 0
d14a1e28 RD	267	else: x /= self.ux
b881fc78	268
d14a1e28	269	if x != sx:
b881fc78	270	if wx.Platform == '__WXMSW__':
d14a1e28	271	self.colWin.Show(False)
b881fc78	272
d14a1e28	273	self.grid.Scroll(x,y)
b881fc78	274
d14a1e28 RD	275	x,y = self.lwin.ClientToScreenXY(evt.m_x,0)
d14a1e28 RD	276	x,y = self.grid.ScreenToClientXY(x,y)
b881fc78	277
d14a1e28 RD	278	if not self.colWin.IsShown():
d14a1e28 RD	279	self.colWin.Show(True)
b881fc78	280
d14a1e28	281	px = x - self.cellX
b881fc78	282
d14a1e28	283	if px < 0 + self.grid._rlSize: px = 0 + self.grid._rlSize
b881fc78 RD	284
	285	if px > w - self.colWin.GetSize()[0] + self.grid._rlSize:
	286	px = w - self.colWin.GetSize()[0] + self.grid._rlSize
	287
d14a1e28 RD	288	self.colWin.DisplayAt(px,y)
d14a1e28 RD	289	return
b881fc78	290
d14a1e28 RD	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)
b881fc78 RD	299
b881fc78 RD	300	if self.grid.XToEdgeOfCol(px + sx) != wx.NOT_FOUND:
d14a1e28 RD	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
b881fc78	309	size = self.lwin.GetSize()
d14a1e28 RD	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
b881fc78	323
d14a1e28 RD	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)
b881fc78 RD	330
b881fc78 RD	331	if col != wx.NOT_FOUND:
d14a1e28	332	self.grid.SelectCol(col,evt.m_controlDown)
b881fc78	333
d14a1e28 RD	334	return
	335	else:
	336	bCol = self.colWin.GetInsertionColumn()
	337	dCol = self.colWin.GetMoveColumn()
b881fc78	338	wx.PostEvent(self,
33785d9f	339	GridColMoveEvent(self.grid.GetId(), dCol, bCol))
b881fc78	340
d14a1e28 RD	341	self.colWin.Destroy()
	342	evt.Skip()
	343
	344	def _CaptureImage(self,rect):
b881fc78 RD	345	bmp = wx.EmptyBitmap(rect.width,rect.height)
b881fc78 RD	346	memdc = wx.MemoryDC()
d14a1e28	347	memdc.SelectObject(bmp)
b881fc78	348	dc = wx.WindowDC(self.lwin)
7722248d	349	memdc.Blit((0,0), rect.GetSize(), dc, rect.GetPosition())
b881fc78	350	memdc.SelectObject(wx.NullBitmap)
d14a1e28 RD	351	return bmp
	352
	353
33785d9f	354	class GridRowMover(wx.EvtHandler):
d14a1e28	355	def __init__(self,grid):
b881fc78	356	wx.EvtHandler.__init__(self)
d14a1e28 RD	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
b881fc78 RD	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)
d14a1e28 RD	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
b881fc78	380
d14a1e28 RD	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
b881fc78 RD	385
	386	if y < 1:
	387	y = 0
	388	else:
	389	y /= self.uy
	390
d14a1e28	391	if y != sy:
b881fc78	392	if wx.Platform == '__WXMSW__':
d14a1e28	393	self.rowWin.Show(False)
b881fc78	394
d14a1e28	395	self.grid.Scroll(x,y)
b881fc78	396
d14a1e28 RD	397	x,y = self.lwin.ClientToScreenXY(0,evt.m_y)
d14a1e28 RD	398	x,y = self.grid.ScreenToClientXY(x,y)
b881fc78	399
d14a1e28 RD	400	if not self.rowWin.IsShown():
d14a1e28 RD	401	self.rowWin.Show(True)
b881fc78	402
d14a1e28	403	py = y - self.cellY
b881fc78 RD	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
d14a1e28 RD	411	self.rowWin.DisplayAt(x,py)
d14a1e28 RD	412	return
b881fc78	413
d14a1e28 RD	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)
b881fc78 RD	422
b881fc78 RD	423	if self.grid.YToEdgeOfRow(py + sy) != wx.NOT_FOUND:
d14a1e28 RD	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
b881fc78	432	size = self.lwin.GetSize()
d14a1e28 RD	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
b881fc78	446
d14a1e28 RD	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)
b881fc78 RD	453
b881fc78 RD	454	if row != wx.NOT_FOUND:
d14a1e28 RD	455	self.grid.SelectRow(row,evt.m_controlDown)
	456	return
	457	else:
	458	bRow = self.rowWin.GetInsertionRow()
	459	dRow = self.rowWin.GetMoveRow()
b881fc78 RD	460
b881fc78 RD	461	wx.PostEvent(self,
33785d9f	462	GridRowMoveEvent(self.grid.GetId(), dRow, bRow))
b881fc78	463
d14a1e28 RD	464	self.rowWin.Destroy()
	465	evt.Skip()
	466
	467	def _CaptureImage(self,rect):
b881fc78 RD	468	bmp = wx.EmptyBitmap(rect.width,rect.height)
b881fc78 RD	469	memdc = wx.MemoryDC()
d14a1e28	470	memdc.SelectObject(bmp)
b881fc78	471	dc = wx.WindowDC(self.lwin)
7722248d	472	memdc.Blit((0,0), rect.GetSize(), dc, rect.GetPosition())
b881fc78	473	memdc.SelectObject(wx.NullBitmap)
d14a1e28 RD	474	return bmp
	475
	476
	477	#----------------------------------------------------------------------------