HGKeeper

#!/usr/bin/env python

# -*- coding: utf-8 -*-

# This file is part of Beremiz, a Integrated Development Environment for

# programming IEC 61131-3 automates supporting plcopen standard and CanFestival.

# See COPYING file for copyrights details.

# This program is free software; you can redistribute it and/or

# modify it under the terms of the GNU General Public License

# as published by the Free Software Foundation; either version 2

# of the License, or (at your option) any later version.

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

# GNU General Public License for more details.

# You should have received a copy of the GNU General Public License

# along with this program; if not, write to the Free Software

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.

from __future__ import absolute_import

import wx

import wx.grid

class CustomGrid(wx.grid.Grid):

def __init__(self, *args, **kwargs):

wx.grid.Grid.__init__(self, *args, **kwargs)

self.Editable = True

self.AddButton = None

self.DeleteButton = None

self.UpButton = None

self.DownButton = None

self.SetFont(wx.Font(12, wx.DEFAULT, wx.NORMAL, wx.NORMAL, False, 'Sans'))

self.SetLabelFont(wx.Font(10, wx.DEFAULT, wx.NORMAL, wx.NORMAL, False, 'Sans'))

self.SetSelectionBackground(wx.WHITE)

self.SetSelectionForeground(wx.BLACK)

self.DisableDragRowSize()

self.Bind(wx.grid.EVT_GRID_SELECT_CELL, self.OnSelectCell)

self.Bind(wx.grid.EVT_GRID_EDITOR_HIDDEN, self.OnEditorHidden)

self.Bind(wx.EVT_KEY_DOWN, self.OnKeyDown)

def SetFocus(self):

if self:

wx.grid.Grid.SetFocus(self)

def SetDefaultValue(self, default_value):

self.DefaultValue = default_value

def SetEditable(self, editable=True):

self.Editable = editable

self.RefreshButtons()

def SetButtons(self, buttons):

for name in ["Add", "Delete", "Up", "Down"]:

button = buttons.get(name, None)

setattr(self, "%sButton" % name, button)

if button is not None:

button.Bind(wx.EVT_BUTTON, getattr(self, "On%sButton" % name))

def RefreshButtons(self):

if self:

rows = self.Table.GetNumberRows()

row = self.GetGridCursorRow()

if self.AddButton is not None:

self.AddButton.Enable(self.Editable)

if self.DeleteButton is not None:

self.DeleteButton.Enable(self.Editable and rows > 0)

if self.UpButton is not None:

self.UpButton.Enable(self.Editable and row > 0)

if self.DownButton is not None:

self.DownButton.Enable(self.Editable and 0 <= row < rows - 1)

def CloseEditControl(self):

row, col = self.GetGridCursorRow(), self.GetGridCursorCol()

if row != -1 and col != -1:

self.SetGridCursor(row, col)

def AddRow(self):

self.CloseEditControl()

new_row = self.GetGridCursorRow() + 1

col = max(self.GetGridCursorCol(), 0)

if getattr(self, "_AddRow", None) is not None:

new_row = self._AddRow(new_row)

else:

self.Table.InsertRow(new_row, self.DefaultValue.copy())

self.Table.ResetView(self)

if new_row is not None:

self.SetSelectedCell(new_row, col)

def DeleteRow(self):

self.CloseEditControl()

row = self.GetGridCursorRow()

if row >= 0:

col = self.GetGridCursorCol()

if getattr(self, "_DeleteRow", None) is not None:

self._DeleteRow(row)

else:

self.Table.RemoveRow(row)

self.Table.ResetView(self)

if self.Table.GetNumberRows() > 0:

self.SetSelectedCell(min(row, self.Table.GetNumberRows() - 1), col)

def MoveRow(self, row, move):

self.CloseEditControl()

col = self.GetGridCursorCol()

if getattr(self, "_MoveRow", None) is not None:

new_row = self._MoveRow(row, move)

else:

new_row = self.Table.MoveRow(row, move)

if new_row != row:

self.Table.ResetView(self)

if new_row != row:

self.SetSelectedCell(new_row, col)

def SetSelectedCell(self, row, col):

self.SetGridCursor(row, col)

self.MakeCellVisible(row, col)

self.RefreshButtons()

def OnAddButton(self, event):

self.AddRow()

self.SetFocus()

event.Skip()

def OnDeleteButton(self, event):

self.DeleteRow()

self.SetFocus()

event.Skip()

def OnUpButton(self, event):

self.MoveRow(self.GetGridCursorRow(), -1)

self.SetFocus()

event.Skip()

def OnDownButton(self, event):

self.MoveRow(self.GetGridCursorRow(), 1)

self.SetFocus()

event.Skip()

def OnSelectCell(self, event):

wx.CallAfter(self.RefreshButtons)

event.Skip()

def OnEditorHidden(self, event):

wx.CallAfter(self.SetFocus)

event.Skip()

def OnKeyDown(self, event):

key_handled = False

keycode = event.GetKeyCode()

if keycode == wx.WXK_TAB:

row = self.GetGridCursorRow()

col = self.GetGridCursorCol()

if event.ShiftDown():

if row < 0 or col == 0:

self.Navigate(wx.NavigationKeyEvent.IsBackward)

key_handled = True

elif row < 0 or col == self.Table.GetNumberCols() - 1:

self.Navigate(wx.NavigationKeyEvent.IsForward)

key_handled = True

elif keycode in (wx.WXK_ADD, wx.WXK_NUMPAD_ADD) and self.Editable:

self.AddRow()

key_handled = True

elif keycode in (wx.WXK_DELETE, wx.WXK_NUMPAD_DELETE) and self.Editable:

self.DeleteRow()

key_handled = True

elif keycode == wx.WXK_UP and event.ShiftDown() and self.Editable:

self.MoveRow(self.GetGridCursorRow(), -1)

key_handled = True

elif keycode == wx.WXK_DOWN and event.ShiftDown() and self.Editable:

self.MoveRow(self.GetGridCursorRow(), 1)

key_handled = True

if not key_handled:

event.Skip()

beremiz