Fix bug with two or more wires connected to one input. Now only one wire can be connected to one input, except BOOLean signals in LD and SFC. If user trying to connect wire with already connected input, wire highlight will become red.
Signed-off-by: Andrey Skvortsov <andrej.skvortzov@gmail.com>
#!/usr/bin/env python
# -*- coding: utf-8 -*-
#This file is part of PLCOpenEditor, a library implementing an IEC 61131-3 editor
#based on the plcopen standard.
#
#Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD
#
#See COPYING file for copyrights details.
#
#This library 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.1 of the License, or (at your option) any later version.
#
#This library 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 library; if not, write to the Free Software
#Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
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, 77, wx.NORMAL, wx.NORMAL, False, 'Sans'))
self.SetLabelFont(wx.Font(10, 77, 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()