#!/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
from time import localtime
from datetime import datetime
import wx, wx.grid
if wx.VERSION >= (2, 8, 0):
import wx.aui
from SFCViewer import *
from LDViewer import *
from Viewer import *
from TextViewer import *
from RessourceEditor import *
from DataTypeEditor import *
from PLCControler import *
from plcopen import OpenPDFDoc
from plcopen.structures import LOCATIONDATATYPES
import os, re, platform, sys, time, traceback, getopt
__version__ = "$Revision$"
CWD = os.path.split(os.path.realpath(__file__))[0]
[ID_PLCOPENEDITOR, ID_PLCOPENEDITORPROJECTTREE,
ID_PLCOPENEDITORMAINSPLITTER, ID_PLCOPENEDITORSECONDSPLITTER,
ID_PLCOPENEDITOREDITORPANEL, ID_PLCOPENEDITORTABSOPENED,
ID_PLCOPENEDITORTOOLBAR, ID_PLCOPENEDITORDEFAULTTOOLBAR,
ID_PLCOPENEDITORSFCTOOLBAR, ID_PLCOPENEDITORFBDTOOLBAR,
ID_PLCOPENEDITORLDTOOLBAR,
] = [wx.NewId() for _init_ctrls in range(11)]
[ID_PLCOPENEDITORFILEMENUITEMS0, ID_PLCOPENEDITORFILEMENUITEMS1,
ID_PLCOPENEDITORFILEMENUITEMS2, ID_PLCOPENEDITORFILEMENUITEMS3,
ID_PLCOPENEDITORFILEMENUITEMS5, ID_PLCOPENEDITORFILEMENUITEMS6,
ID_PLCOPENEDITORFILEMENUITEMS7, ID_PLCOPENEDITORFILEMENUITEMS9,
ID_PLCOPENEDITORFILEMENUITEMS11,
] = [wx.NewId() for _init_coll_FileMenu_Items in range(9)]
[ID_PLCOPENEDITORHELPMENUITEMS0, ID_PLCOPENEDITORHELPMENUITEMS1,
ID_PLCOPENEDITORHELPMENUITEMS2, ID_PLCOPENEDITORHELPMENUITEMS3,
] = [wx.NewId() for _init_coll_HelpMenu_Items in range(4)]
[ID_PLCOPENEDITORSFCMENUITEMS0, ID_PLCOPENEDITORSFCMENUITEMS1,
ID_PLCOPENEDITORSFCMENUITEMS2, ID_PLCOPENEDITORSFCMENUITEMS3,
] = [wx.NewId() for _init_coll_HelpMenu_Items in range(4)]
[ID_PLCOPENEDITORCONFIGMENUITEMS0, ID_PLCOPENEDITORCONFIGMENUITEMS1,
] = [wx.NewId() for _init_coll_HelpMenu_Items in range(2)]
[ID_PLCOPENEDITOREDITMENUITEMS0, ID_PLCOPENEDITOREDITMENUITEMS1,
ID_PLCOPENEDITOREDITMENUITEMS2, ID_PLCOPENEDITOREDITMENUITEMS4,
ID_PLCOPENEDITOREDITMENUITEMS5, ID_PLCOPENEDITOREDITMENUITEMS6,
ID_PLCOPENEDITOREDITMENUITEMS8, ID_PLCOPENEDITOREDITMENUITEMS9,
ID_PLCOPENEDITOREDITMENUITEMS11, ID_PLCOPENEDITOREDITMENUITEMS12,
ID_PLCOPENEDITOREDITMENUITEMS14, ID_PLCOPENEDITOREDITMENUITEMS15,
] = [wx.NewId() for _init_coll_EditMenu_Items in range(12)]
[ID_PLCOPENEDITORSFCMENUITEMS0, ID_PLCOPENEDITORSFCMENUITEMS1,
ID_PLCOPENEDITORSFCMENUITEMS2, ID_PLCOPENEDITORSFCMENUITEMS3,
] = [wx.NewId() for _init_coll_SFCMenu_Items in range(4)]
[ID_PLCOPENEDITORCONFIGMENUITEMS0, ID_PLCOPENEDITORCONFIGMENUITEMS1,
] = [wx.NewId() for _init_coll_ConfigMenu_Items in range(2)]
#-------------------------------------------------------------------------------
# ToolBars definitions
#-------------------------------------------------------------------------------
[ID_PLCOPENEDITORTOOLBARSELECTION, ID_PLCOPENEDITORTOOLBARCOMMENT,
ID_PLCOPENEDITORTOOLBARVARIABLE, ID_PLCOPENEDITORTOOLBARBLOCK,
ID_PLCOPENEDITORTOOLBARCONNECTION, ID_PLCOPENEDITORTOOLBARWIRE,
ID_PLCOPENEDITORTOOLBARPOWERRAIL, ID_PLCOPENEDITORTOOLBARRUNG,
ID_PLCOPENEDITORTOOLBARCOIL, ID_PLCOPENEDITORTOOLBARCONTACT,
ID_PLCOPENEDITORTOOLBARBRANCH, ID_PLCOPENEDITORTOOLBARINITIALSTEP,
ID_PLCOPENEDITORTOOLBARSTEP, ID_PLCOPENEDITORTOOLBARTRANSITION,
ID_PLCOPENEDITORTOOLBARACTIONBLOCK, ID_PLCOPENEDITORTOOLBARDIVERGENCE,
ID_PLCOPENEDITORTOOLBARJUMP,
] = [wx.NewId() for _init_coll_DefaultToolBar_Items in range(17)]
ToolBarItems = {
"FBD" : [(True, FREEDRAWING_MODE|DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARCOMMENT, "OnCommentTool", "comment.png", "Create a new comment"),
(True, FREEDRAWING_MODE|DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARVARIABLE, "OnVariableTool", "variable.png", "Create a new variable"),
(True, FREEDRAWING_MODE|DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARBLOCK, "OnBlockTool", "block.png", "Create a new block"),
(True, FREEDRAWING_MODE|DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARCONNECTION, "OnConnectionTool", "connection.png", "Create a new connection")],
"LD" : [(True, FREEDRAWING_MODE, ID_PLCOPENEDITORTOOLBARCOMMENT, "OnCommentTool", "comment.png", "Create a new comment"),
(True, FREEDRAWING_MODE, ID_PLCOPENEDITORTOOLBARPOWERRAIL, "OnPowerRailTool", "powerrail.png", "Create a new power rail"),
(False, DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARRUNG, "OnRungTool", "rung.png", "Create a new rung"),
(True, FREEDRAWING_MODE, ID_PLCOPENEDITORTOOLBARCOIL, "OnCoilTool", "coil.png", "Create a new coil"),
(False, FREEDRAWING_MODE|DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARCONTACT, "OnContactTool", "contact.png", "Create a new contact"),
(False, DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARBRANCH, "OnBranchTool", "branch.png", "Create a new branch"),
(True, FREEDRAWING_MODE, ID_PLCOPENEDITORTOOLBARVARIABLE, "OnVariableTool", "variable.png", "Create a new variable"),
(False, FREEDRAWING_MODE|DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARBLOCK, "OnBlockTool", "block.png", "Create a new block"),
(True, FREEDRAWING_MODE, ID_PLCOPENEDITORTOOLBARCONNECTION, "OnConnectionTool", "connection.png", "Create a new connection")],
"SFC" : [(True, FREEDRAWING_MODE|DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARCOMMENT, "OnCommentTool", "comment.png", "Create a new comment"),
(True, FREEDRAWING_MODE|DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARINITIALSTEP, "OnInitialStepTool", "initial_step.png", "Create a new initial step"),
(False, FREEDRAWING_MODE|DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARSTEP, "OnStepTool", "step.png", "Create a new step"),
(True, FREEDRAWING_MODE, ID_PLCOPENEDITORTOOLBARTRANSITION, "OnTransitionTool", "transition.png", "Create a new transition"),
(False, FREEDRAWING_MODE|DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARACTIONBLOCK, "OnActionBlockTool", "action.png", "Create a new action block"),
(False, FREEDRAWING_MODE|DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARDIVERGENCE, "OnDivergenceTool", "divergence.png", "Create a new divergence"),
(False, FREEDRAWING_MODE|DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARJUMP, "OnJumpTool", "jump.png", "Create a new jump"),
(True, FREEDRAWING_MODE, ID_PLCOPENEDITORTOOLBARVARIABLE, "OnVariableTool", "variable.png", "Create a new variable"),
(True, FREEDRAWING_MODE, ID_PLCOPENEDITORTOOLBARBLOCK, "OnBlockTool", "block.png", "Create a new block"),
(True, FREEDRAWING_MODE, ID_PLCOPENEDITORTOOLBARCONNECTION, "OnConnectionTool", "connection.png", "Create a new connection"),
(True, FREEDRAWING_MODE, ID_PLCOPENEDITORTOOLBARPOWERRAIL, "OnPowerRailTool", "powerrail.png", "Create a new power rail"),
(True, FREEDRAWING_MODE, ID_PLCOPENEDITORTOOLBARCONTACT, "OnContactTool", "contact.png", "Create a new contact")],
"ST" : [],
"IL" : []
}
def AppendMenu(parent, help, id, kind, text):
if wx.VERSION >= (2, 6, 0):
parent.Append(help=help, id=id, kind=kind, text=text)
else:
parent.Append(helpString=help, id=id, kind=kind, item=text)
if wx.VERSION >= (2, 8, 0):
base_class = wx.aui.AuiMDIParentFrame
else:
base_class = wx.Frame
class PLCOpenEditor(base_class):
_custom_classes = {'wx.SashWindow' : ['Viewer']}
if wx.VERSION < (2, 6, 0):
def Bind(self, event, function, id = None):
if id is not None:
event(self, id, function)
else:
event(self, function)
def _init_coll_EditMenu_Items(self, parent):
AppendMenu(parent, help='', id=ID_PLCOPENEDITOREDITMENUITEMS0,
kind=wx.ITEM_NORMAL, text=u'Refresh\tCTRL+R')
AppendMenu(parent, help='', id=ID_PLCOPENEDITOREDITMENUITEMS1,
kind=wx.ITEM_NORMAL, text=u'Undo\tCTRL+Z')
AppendMenu(parent, help='', id=ID_PLCOPENEDITOREDITMENUITEMS2,
kind=wx.ITEM_NORMAL, text=u'Redo\tCTRL+Y')
parent.AppendSeparator()
AppendMenu(parent, help='', id=ID_PLCOPENEDITOREDITMENUITEMS4,
kind=wx.ITEM_NORMAL, text=u'Cut\tCTRL+X')
AppendMenu(parent, help='', id=ID_PLCOPENEDITOREDITMENUITEMS5,
kind=wx.ITEM_NORMAL, text=u'Copy\tCTRL+C')
AppendMenu(parent, help='', id=ID_PLCOPENEDITOREDITMENUITEMS6,
kind=wx.ITEM_NORMAL, text=u'Paste\tCTRL+V')
parent.AppendSeparator()
AppendMenu(parent, help='', id=ID_PLCOPENEDITOREDITMENUITEMS8,
kind=wx.ITEM_NORMAL, text=u'Add Data Type')
AppendMenu(parent, help='', id=ID_PLCOPENEDITOREDITMENUITEMS9,
kind=wx.ITEM_NORMAL, text=u'Remove Data Type')
parent.AppendSeparator()
AppendMenu(parent, help='', id=ID_PLCOPENEDITOREDITMENUITEMS11,
kind=wx.ITEM_NORMAL, text=u'Add POU')
AppendMenu(parent, help='', id=ID_PLCOPENEDITOREDITMENUITEMS12,
kind=wx.ITEM_NORMAL, text=u'Remove POU')
parent.AppendSeparator()
AppendMenu(parent, help='', id=ID_PLCOPENEDITOREDITMENUITEMS14,
kind=wx.ITEM_NORMAL, text=u'Add Configuration')
AppendMenu(parent, help='', id=ID_PLCOPENEDITOREDITMENUITEMS15,
kind=wx.ITEM_NORMAL, text=u'Remove Configuration')
self.Bind(wx.EVT_MENU, self.OnRefreshMenu,
id=ID_PLCOPENEDITOREDITMENUITEMS0)
self.Bind(wx.EVT_MENU, self.OnUndoMenu,
id=ID_PLCOPENEDITOREDITMENUITEMS1)
self.Bind(wx.EVT_MENU, self.OnRedoMenu,
id=ID_PLCOPENEDITOREDITMENUITEMS2)
self.Bind(wx.EVT_MENU, self.OnCutMenu,
id=ID_PLCOPENEDITOREDITMENUITEMS4)
self.Bind(wx.EVT_MENU, self.OnCopyMenu,
id=ID_PLCOPENEDITOREDITMENUITEMS5)
self.Bind(wx.EVT_MENU, self.OnPasteMenu,
id=ID_PLCOPENEDITOREDITMENUITEMS6)
self.Bind(wx.EVT_MENU, self.OnAddDataTypeMenu,
id=ID_PLCOPENEDITOREDITMENUITEMS8)
self.Bind(wx.EVT_MENU, self.OnRemoveDataTypeMenu,
id=ID_PLCOPENEDITOREDITMENUITEMS9)
self.Bind(wx.EVT_MENU, self.OnAddPouMenu,
id=ID_PLCOPENEDITOREDITMENUITEMS11)
self.Bind(wx.EVT_MENU, self.OnRemovePouMenu,
id=ID_PLCOPENEDITOREDITMENUITEMS12)
self.Bind(wx.EVT_MENU, self.OnAddConfigurationMenu,
id=ID_PLCOPENEDITOREDITMENUITEMS14)
self.Bind(wx.EVT_MENU, self.OnRemoveConfigurationMenu,
id=ID_PLCOPENEDITOREDITMENUITEMS15)
def _init_coll_menuBar1_Menus(self, parent):
if self.ModeSolo:
parent.Append(menu=self.FileMenu, title=u'File')
parent.Append(menu=self.EditMenu, title=u'Edit')
parent.Append(menu=self.HelpMenu, title=u'Help')
def _init_coll_ConfigMenu_Items(self, parent):
AppendMenu(parent, help='', id=ID_PLCOPENEDITORCONFIGMENUITEMS0,
kind=wx.ITEM_NORMAL, text=u'Add Resource')
AppendMenu(parent, help='', id=ID_PLCOPENEDITORCONFIGMENUITEMS1,
kind=wx.ITEM_NORMAL, text=u'Remove Resource')
self.Bind(wx.EVT_MENU, self.OnAddResourceMenu,
id=ID_PLCOPENEDITORCONFIGMENUITEMS0)
self.Bind(wx.EVT_MENU, self.OnRemoveResourceMenu,
id=ID_PLCOPENEDITORCONFIGMENUITEMS1)
def _init_coll_HelpMenu_Items(self, parent):
AppendMenu(parent, help='', id=ID_PLCOPENEDITORHELPMENUITEMS0,
kind=wx.ITEM_NORMAL, text=u'PLCOpenEditor\tF1')
AppendMenu(parent, help='', id=ID_PLCOPENEDITORHELPMENUITEMS1,
kind=wx.ITEM_NORMAL, text=u'PLCOpen\tF2')
AppendMenu(parent, help='', id=ID_PLCOPENEDITORHELPMENUITEMS2,
kind=wx.ITEM_NORMAL, text=u'IEC 61131-3\tF3')
if self.ModeSolo:
AppendMenu(parent, help='', id=ID_PLCOPENEDITORHELPMENUITEMS3,
kind=wx.ITEM_NORMAL, text=u'About')
self.Bind(wx.EVT_MENU, self.OnPLCOpenMenu,
id=ID_PLCOPENEDITORHELPMENUITEMS1)
def _init_coll_FileMenu_Items(self, parent):
AppendMenu(parent, help='', id=ID_PLCOPENEDITORFILEMENUITEMS0,
kind=wx.ITEM_NORMAL, text=u'New\tCTRL+N')
AppendMenu(parent, help='', id=ID_PLCOPENEDITORFILEMENUITEMS1,
kind=wx.ITEM_NORMAL, text=u'Open\tCTRL+O')
AppendMenu(parent, help='', id=ID_PLCOPENEDITORFILEMENUITEMS2,
kind=wx.ITEM_NORMAL, text=u'Close Tab\tCTRL+W')
AppendMenu(parent, help='', id=ID_PLCOPENEDITORFILEMENUITEMS3,
kind=wx.ITEM_NORMAL, text=u'Close Project')
parent.AppendSeparator()
AppendMenu(parent, help='', id=ID_PLCOPENEDITORFILEMENUITEMS5,
kind=wx.ITEM_NORMAL, text=u'Save\tCTRL+S')
AppendMenu(parent, help='', id=ID_PLCOPENEDITORFILEMENUITEMS6,
kind=wx.ITEM_NORMAL, text=u'Save As...\tCTRL+SHIFT+S')
AppendMenu(parent, help='', id=ID_PLCOPENEDITORFILEMENUITEMS7,
kind=wx.ITEM_NORMAL, text=u'Generate Program\tCTRL+G')
parent.AppendSeparator()
AppendMenu(parent, help='', id=ID_PLCOPENEDITORFILEMENUITEMS9,
kind=wx.ITEM_NORMAL, text=u'Properties')
parent.AppendSeparator()
AppendMenu(parent, help='', id=ID_PLCOPENEDITORFILEMENUITEMS11,
kind=wx.ITEM_NORMAL, text=u'Quit\tCTRL+Q')
self.Bind(wx.EVT_MENU, self.OnNewProjectMenu,
id=ID_PLCOPENEDITORFILEMENUITEMS0)
self.Bind(wx.EVT_MENU, self.OnOpenProjectMenu,
id=ID_PLCOPENEDITORFILEMENUITEMS1)
self.Bind(wx.EVT_MENU, self.OnCloseTabMenu,
id=ID_PLCOPENEDITORFILEMENUITEMS2)
self.Bind(wx.EVT_MENU, self.OnCloseProjectMenu,
id=ID_PLCOPENEDITORFILEMENUITEMS3)
self.Bind(wx.EVT_MENU, self.OnSaveProjectMenu,
id=ID_PLCOPENEDITORFILEMENUITEMS5)
self.Bind(wx.EVT_MENU, self.OnSaveProjectAsMenu,
id=ID_PLCOPENEDITORFILEMENUITEMS6)
self.Bind(wx.EVT_MENU, self.OnGenerateProgramMenu,
id=ID_PLCOPENEDITORFILEMENUITEMS7)
self.Bind(wx.EVT_MENU, self.OnPropertiesMenu,
id=ID_PLCOPENEDITORFILEMENUITEMS9)
self.Bind(wx.EVT_MENU, self.OnQuitMenu,
id=ID_PLCOPENEDITORFILEMENUITEMS11)
def _init_coll_SFCMenu_Items(self, parent):
AppendMenu(parent, help='', id=ID_PLCOPENEDITORSFCMENUITEMS0,
kind=wx.ITEM_NORMAL, text=u'Add Transition')
AppendMenu(parent, help='', id=ID_PLCOPENEDITORSFCMENUITEMS1,
kind=wx.ITEM_NORMAL, text=u'Add Action')
AppendMenu(parent, help='', id=ID_PLCOPENEDITORSFCMENUITEMS2,
kind=wx.ITEM_NORMAL, text=u'Remove Transition')
AppendMenu(parent, help='', id=ID_PLCOPENEDITORSFCMENUITEMS3,
kind=wx.ITEM_NORMAL, text=u'Remove Action')
self.Bind(wx.EVT_MENU, self.OnAddPouTransitionMenu,
id=ID_PLCOPENEDITORSFCMENUITEMS0)
self.Bind(wx.EVT_MENU, self.OnAddPouActionMenu,
id=ID_PLCOPENEDITORSFCMENUITEMS1)
self.Bind(wx.EVT_MENU, self.OnRemovePouTransitionMenu,
id=ID_PLCOPENEDITORSFCMENUITEMS2)
self.Bind(wx.EVT_MENU, self.OnRemovePouActionMenu,
id=ID_PLCOPENEDITORSFCMENUITEMS3)
def _init_utils(self):
self.menuBar1 = wx.MenuBar()
if self.ModeSolo:
self.FileMenu = wx.Menu(title='')
else:
self.FileMenu = None
self.EditMenu = wx.Menu(title='')
self.HelpMenu = wx.Menu(title='')
self.SFCMenu = wx.Menu(title='')
self.ConfigMenu = wx.Menu(title='')
self._init_coll_menuBar1_Menus(self.menuBar1)
if self.ModeSolo:
self._init_coll_FileMenu_Items(self.FileMenu)
self._init_coll_EditMenu_Items(self.EditMenu)
self._init_coll_HelpMenu_Items(self.HelpMenu)
self._init_coll_SFCMenu_Items(self.SFCMenu)
self._init_coll_ConfigMenu_Items(self.ConfigMenu)
def _init_ctrls(self, prnt):
if wx.VERSION >= (2, 8, 0):
wx.aui.AuiMDIParentFrame.__init__(self, winid=ID_PLCOPENEDITOR, name=u'PLCOpenEditor',
parent=prnt, pos=wx.DefaultPosition, size=wx.Size(1000, 600),
style=wx.DEFAULT_FRAME_STYLE|wx.SUNKEN_BORDER|wx.CLIP_CHILDREN, title=u'PLCOpenEditor')
else:
wx.Frame.__init__(self, id=ID_PLCOPENEDITOR, name=u'PLCOpenEditor',
parent=prnt, pos=wx.DefaultPosition, size=wx.Size(1000, 600),
style=wx.DEFAULT_FRAME_STYLE, title=u'PLCOpenEditor')
self._init_utils()
self.SetClientSize(wx.Size(1000, 600))
self.SetMenuBar(self.menuBar1)
self.Bind(wx.EVT_CLOSE, self.OnCloseFrame)
if not self.ModeSolo:
self.Bind(wx.EVT_MENU, self.OnCloseTabMenu, id=ID_PLCOPENEDITORFILEMENUITEMS2)
self.Bind(wx.EVT_MENU, self.OnSaveProjectMenu, id=ID_PLCOPENEDITORFILEMENUITEMS5)
accel = wx.AcceleratorTable([wx.AcceleratorEntry(wx.ACCEL_CTRL, 119, ID_PLCOPENEDITORFILEMENUITEMS2),
wx.AcceleratorEntry(wx.ACCEL_CTRL, 83, ID_PLCOPENEDITORFILEMENUITEMS5)])
self.SetAcceleratorTable(accel)
if wx.VERSION >= (2, 8, 0):
self.AUIManager = wx.aui.AuiManager(self)
self.AUIManager.SetDockSizeConstraint(0.5, 0.5)
self.Panes = {}
if wx.VERSION < (2, 8, 0):
self.MainSplitter = wx.SplitterWindow(id=ID_PLCOPENEDITORMAINSPLITTER,
name='MainSplitter', parent=self, point=wx.Point(0, 0),
size=wx.Size(0, 0), style=wx.SP_3D)
self.MainSplitter.SetNeedUpdating(True)
self.MainSplitter.SetMinimumPaneSize(1)
self.ProjectTree = wx.TreeCtrl(id=ID_PLCOPENEDITORPROJECTTREE,
name='treeCtrl1', parent=self.MainSplitter, pos=wx.Point(0, 0),
size=wx.Size(-1, -1),
style=wx.TR_HAS_BUTTONS|wx.TR_EDIT_LABELS|wx.TR_SINGLE|wx.SUNKEN_BORDER)
else:
self.ProjectTree = wx.TreeCtrl(id=ID_PLCOPENEDITORPROJECTTREE,
name='treeCtrl1', parent=self, pos=wx.Point(0, 0),
size=wx.Size(200, -1),
style=wx.TR_HAS_BUTTONS|wx.TR_EDIT_LABELS|wx.TR_SINGLE|wx.SUNKEN_BORDER)
self.AUIManager.AddPane(self.ProjectTree, wx.aui.AuiPaneInfo().Caption("Project Tree").Left().Layer(1).BestSize(wx.Size(200, 500)).CloseButton(False))
if wx.Platform == '__WXMSW__':
self.Bind(wx.EVT_TREE_ITEM_RIGHT_CLICK, self.OnProjectTreeRightUp,
id=ID_PLCOPENEDITORPROJECTTREE)
self.Bind(wx.EVT_TREE_SEL_CHANGED, self.OnProjectTreeItemSelected,
id=ID_PLCOPENEDITORPROJECTTREE)
else:
if wx.VERSION >= (2, 6, 0):
self.ProjectTree.Bind(wx.EVT_RIGHT_UP, self.OnProjectTreeRightUp)
self.ProjectTree.Bind(wx.EVT_LEFT_UP, self.OnProjectTreeLeftUp)
else:
wx.EVT_RIGHT_UP(self.ProjectTree, self.OnProjectTreeRightUp)
wx.EVT_LEFT_UP(self.ProjectTree, self.OnProjectTreeLeftUp)
self.Bind(wx.EVT_TREE_SEL_CHANGING, self.OnProjectTreeItemChanging,
id=ID_PLCOPENEDITORPROJECTTREE)
self.Bind(wx.EVT_TREE_BEGIN_DRAG, self.OnProjectTreeBeginDrag,
id=ID_PLCOPENEDITORPROJECTTREE)
self.Bind(wx.EVT_TREE_BEGIN_LABEL_EDIT, self.OnProjectTreeItemBeginEdit,
id=ID_PLCOPENEDITORPROJECTTREE)
self.Bind(wx.EVT_TREE_END_LABEL_EDIT, self.OnProjectTreeItemEndEdit,
id=ID_PLCOPENEDITORPROJECTTREE)
self.Bind(wx.EVT_TREE_ITEM_ACTIVATED, self.OnProjectTreeItemActivated,
id=ID_PLCOPENEDITORPROJECTTREE)
if wx.VERSION < (2, 8, 0):
self.ToolBar = self.CreateToolBar(wx.TB_HORIZONTAL|wx.TB_FLAT|wx.NO_BORDER,
ID_PLCOPENEDITORTOOLBAR, 'ToolBar')
self.ToolBar.SetToolBitmapSize(wx.Size(25, 25))
self.ToolBar.AddRadioTool(ID_PLCOPENEDITORTOOLBARSELECTION,
wx.Bitmap(os.path.join(CWD, 'Images', 'select.png')), wx.NullBitmap, "Select an object")
self.ToolBar.Realize()
else:
ToolBar = wx.ToolBar(self, ID_PLCOPENEDITORTOOLBAR, wx.DefaultPosition, wx.DefaultSize,
wx.TB_FLAT | wx.TB_NODIVIDER | wx.NO_BORDER)
ToolBar.SetToolBitmapSize(wx.Size(25, 25))
ToolBar.AddRadioTool(ID_PLCOPENEDITORTOOLBARSELECTION,
wx.Bitmap(os.path.join(CWD, 'Images', 'select.png')), wx.NullBitmap, "Select an object")
ToolBar.Realize()
self.Panes["ToolBar"] = ToolBar
self.AUIManager.AddPane(ToolBar, wx.aui.AuiPaneInfo().
Name("ToolBar").Caption("Toolbar").
ToolbarPane().Top().
LeftDockable(False).RightDockable(False))
self.Bind(wx.EVT_TOOL, self.OnSelectionTool,
id=ID_PLCOPENEDITORTOOLBARSELECTION)
if wx.VERSION < (2, 8, 0):
self.SecondSplitter = wx.SplitterWindow(id=ID_PLCOPENEDITORSECONDSPLITTER,
name='SecondSplitter', parent=self.MainSplitter, point=wx.Point(0, 0),
size=wx.Size(0, 0), style=wx.SP_3D)
self.SecondSplitter.SetMinimumPaneSize(1)
self.MainSplitter.SplitVertically(self.ProjectTree, self.SecondSplitter, 200)
self.TabsOpened = wx.Notebook(id=ID_PLCOPENEDITORTABSOPENED,
name='TabsOpened', parent=self.SecondSplitter, pos=wx.Point(0,
0), size=wx.Size(0, 0), style=0)
if wx.VERSION >= (2, 6, 0):
self.TabsOpened.Bind(wx.EVT_NOTEBOOK_PAGE_CHANGED,
self.OnPouSelectedChanged, id=ID_PLCOPENEDITORTABSOPENED)
else:
wx.EVT_NOTEBOOK_PAGE_CHANGED(self.TabsOpened, ID_PLCOPENEDITORTABSOPENED,
self.OnPouSelectedChanged)
if wx.VERSION < (2, 8, 0):
self.VariablePanelIndexer = VariablePanelIndexer(self.SecondSplitter, self.Controler)
self.SecondSplitter.SplitHorizontally(self.TabsOpened, self.VariablePanelIndexer, -200)
else:
self.VariablePanelIndexer = VariablePanelIndexer(self, self.Controler)
self.AUIManager.AddPane(self.VariablePanelIndexer, wx.aui.AuiPaneInfo().Caption("Variable Panel").Bottom().Layer(0).BestSize(wx.Size(800, 200)).CloseButton(False))
self.AUIManager.Update()
def __init__(self, parent, controler = None, fileOpen = None):
self.ModeSolo = controler == None
if self.ModeSolo:
self.Controler = PLCControler()
if fileOpen is not None:
self.Controler.OpenXMLFile(fileOpen)
else:
self.Controler = controler
self._init_ctrls(parent)
if wx.Platform != '__WXMSW__':
self.TreeImageList = wx.ImageList(16, 16)
for language in LANGUAGES:
self.TreeImageList.Add(wx.Bitmap(os.path.join(CWD, 'Images', '%s.png'%language)))
self.ProjectTree.AssignImageList(self.TreeImageList)
self.CurrentToolBar = []
self.CurrentLanguage = ""
self.SelectedItem = None
self.CopyBuffer = None
self.DrawingMode = FREEDRAWING_MODE
#self.DrawingMode = DRIVENDRAWING_MODE
if not self.ModeSolo or fileOpen is not None:
self.RefreshProjectTree()
self.RefreshFileMenu()
self.RefreshEditMenu()
self.RefreshTitle()
self.RefreshToolBar()
def ResetSelectedItem(self):
self.SelectedItem = None
def GetPageCount(self):
if wx.VERSION >= (2, 8, 0):
notebook = self.GetNotebook()
if notebook is not None:
return notebook.GetPageCount()
else:
return 0
else:
return self.TabsOpened.GetPageCount()
def GetPage(self, idx):
if wx.VERSION >= (2, 8, 0):
notebook = self.GetNotebook()
if notebook is not None:
return notebook.GetPage(idx).GetViewer()
else:
return None
else:
return self.TabsOpened.GetPage(idx)
def GetPageSelection(self):
if wx.VERSION >= (2, 8, 0):
notebook = self.GetNotebook()
if notebook is not None:
return notebook.GetSelection()
else:
return -1
else:
return self.TabsOpened.GetSelection()
def SetPageSelection(self, idx):
if wx.VERSION >= (2, 8, 0):
notebook = self.GetNotebook()
if notebook is not None:
notebook.SetSelection(idx)
else:
self.TabsOpened.SetSelection(idx)
def DeletePage(self, idx):
if wx.VERSION >= (2, 8, 0):
notebook = self.GetNotebook()
if notebook is not None:
notebook.DeletePage(idx)
else:
self.TabsOpened.DeletePage(idx)
def DeleteAllPages(self):
if wx.VERSION >= (2, 8, 0):
notebook = self.GetNotebook()
if notebook is not None:
for idx in xrange(notebook.GetPageCount()):
notebook.DeletePage(idx)
else:
self.TabsOpened.DeleteAllPages()
def SetPageText(self, idx, text):
if wx.VERSION >= (2, 8, 0):
notebook = self.GetNotebook()
if notebook is not None:
return notebook.SetPageText(idx, text)
else:
return self.TabsOpened.SetPageText(idx, text)
def GetPageText(self, idx):
if wx.VERSION >= (2, 8, 0):
notebook = self.GetNotebook()
if notebook is not None:
return notebook.GetPageText(idx)
else:
return ""
else:
return self.TabsOpened.GetPageText(idx)
def OnPageClose(self, event):
selected = self.GetPageSelection()
if selected >= 0:
tagname = self.GetPage(selected).GetTagName()
self.VariablePanelIndexer.RemoveVariablePanel(tagname)
if self.GetPageCount() > 0:
new_index = min(selected, self.GetPageCount() - 1)
tagname = self.GetPage(new_index).GetTagName()
self.SetPageSelection(new_index)
self.VariablePanelIndexer.ChangeVariablePanel(tagname)
self.RefreshTitle()
self.RefreshFileMenu()
self.RefreshEditMenu()
self.RefreshToolBar()
event.Skip()
def GetCopyBuffer(self):
return self.CopyBuffer
def SetCopyBuffer(self, element):
self.CopyBuffer = element
self.RefreshEditMenu()
def GetDrawingMode(self):
return self.DrawingMode
def RefreshTitle(self):
if self.Controler.HasOpenedProject() > 0:
self.SetTitle("PLCOpenEditor - %s"%self.Controler.GetFilename())
else:
self.SetTitle("PLCOpenEditor")
def RefreshFileMenu(self):
if self.FileMenu:
if self.Controler.HasOpenedProject():
if self.GetPageCount() > 0:
self.FileMenu.Enable(ID_PLCOPENEDITORFILEMENUITEMS2, True)
else:
self.FileMenu.Enable(ID_PLCOPENEDITORFILEMENUITEMS2, False)
self.FileMenu.Enable(ID_PLCOPENEDITORFILEMENUITEMS3, True)
self.FileMenu.Enable(ID_PLCOPENEDITORFILEMENUITEMS5, True)
self.FileMenu.Enable(ID_PLCOPENEDITORFILEMENUITEMS6, True)
self.FileMenu.Enable(ID_PLCOPENEDITORFILEMENUITEMS7, True)
self.FileMenu.Enable(ID_PLCOPENEDITORFILEMENUITEMS9, True)
else:
self.FileMenu.Enable(ID_PLCOPENEDITORFILEMENUITEMS2, False)
self.FileMenu.Enable(ID_PLCOPENEDITORFILEMENUITEMS3, False)
self.FileMenu.Enable(ID_PLCOPENEDITORFILEMENUITEMS5, False)
self.FileMenu.Enable(ID_PLCOPENEDITORFILEMENUITEMS6, False)
self.FileMenu.Enable(ID_PLCOPENEDITORFILEMENUITEMS7, False)
self.FileMenu.Enable(ID_PLCOPENEDITORFILEMENUITEMS9, False)
def RefreshEditMenu(self):
if self.EditMenu:
if self.Controler.HasOpenedProject():
undo, redo = self.Controler.GetBufferState()
self.EditMenu.Enable(ID_PLCOPENEDITOREDITMENUITEMS1, undo)
self.EditMenu.Enable(ID_PLCOPENEDITOREDITMENUITEMS2, redo)
else:
self.EditMenu.Enable(ID_PLCOPENEDITOREDITMENUITEMS1, False)
self.EditMenu.Enable(ID_PLCOPENEDITOREDITMENUITEMS2, False)
if self.Controler.HasOpenedProject():
if self.GetPageCount() > 0:
self.EditMenu.Enable(ID_PLCOPENEDITOREDITMENUITEMS0, True)
else:
self.EditMenu.Enable(ID_PLCOPENEDITOREDITMENUITEMS0, False)
self.EditMenu.Enable(ID_PLCOPENEDITOREDITMENUITEMS8, True)
self.EditMenu.Enable(ID_PLCOPENEDITOREDITMENUITEMS9, True)
self.EditMenu.Enable(ID_PLCOPENEDITOREDITMENUITEMS11, True)
self.EditMenu.Enable(ID_PLCOPENEDITOREDITMENUITEMS12, True)
self.EditMenu.Enable(ID_PLCOPENEDITOREDITMENUITEMS14, True)
self.EditMenu.Enable(ID_PLCOPENEDITOREDITMENUITEMS15, True)
else:
self.EditMenu.Enable(ID_PLCOPENEDITOREDITMENUITEMS0, False)
self.EditMenu.Enable(ID_PLCOPENEDITOREDITMENUITEMS8, False)
self.EditMenu.Enable(ID_PLCOPENEDITOREDITMENUITEMS9, False)
self.EditMenu.Enable(ID_PLCOPENEDITOREDITMENUITEMS11, False)
self.EditMenu.Enable(ID_PLCOPENEDITOREDITMENUITEMS12, False)
self.EditMenu.Enable(ID_PLCOPENEDITOREDITMENUITEMS14, False)
self.EditMenu.Enable(ID_PLCOPENEDITOREDITMENUITEMS15, False)
self.EditMenu.Enable(ID_PLCOPENEDITOREDITMENUITEMS4, True)
self.EditMenu.Enable(ID_PLCOPENEDITOREDITMENUITEMS5, True)
if self.CopyBuffer is not None:
self.EditMenu.Enable(ID_PLCOPENEDITOREDITMENUITEMS6, True)
else:
self.EditMenu.Enable(ID_PLCOPENEDITOREDITMENUITEMS6, False)
def RefreshEditor(self, variablepanel = True):
selected = self.GetPageSelection()
if selected != -1:
window = self.GetPage(selected)
window.RefreshView()
if variablepanel:
self.VariablePanelIndexer.RefreshVariablePanel(window.GetTagName())
def ShowProperties(self):
old_values = self.Controler.GetProjectProperties()
dialog = ProjectDialog(self)
dialog.SetValues(old_values)
if dialog.ShowModal() == wx.ID_OK:
new_values = dialog.GetValues()
projectname = new_values.pop("projectName")
new_values["creationDateTime"] = old_values["creationDateTime"]
self.Controler.SetProjectProperties(projectname, new_values)
self.RefreshProjectTree()
dialog.Destroy()
def OnCloseFrame(self, event):
if not self.ModeSolo and getattr(self, "_onclose", None) is not None:
self.AUIManager.UnInit()
self._onclose()
event.Skip()
elif not self.Controler.ProjectIsSaved():
dialog = wx.MessageDialog(self, "There are changes, do you want to save?", "Close Application", wx.YES_NO|wx.CANCEL|wx.ICON_QUESTION)
answer = dialog.ShowModal()
dialog.Destroy()
if answer == wx.ID_YES:
self.SaveProject()
event.Skip()
elif answer == wx.ID_NO:
self.Controler.Reset()
self.AUIManager.UnInit()
wx.CallAfter(self.Close)
event.Skip()
else:
event.Veto()
else:
event.Skip()
def OnNewProjectMenu(self, event):
dialog = ProjectDialog(self)
if dialog.ShowModal() == wx.ID_OK:
values = dialog.GetValues()
projectname = values.pop("projectName")
values["creationDateTime"] = datetime(*localtime()[:6])
self.Controler.CreateNewProject(projectname)
self.Controler.SetProjectProperties(projectname, values)
self.RefreshTitle()
self.RefreshFileMenu()
self.RefreshEditMenu()
self.RefreshProjectTree()
event.Skip()
def OnOpenProjectMenu(self, event):
filepath = self.Controler.GetFilePath()
if filepath != "":
directory = os.path.dirname(filepath)
else:
directory = os.getcwd()
dialog = wx.FileDialog(self, "Choose a file", directory, "", "PLCOpen files (*.xml)|*.xml|All files|*.*", wx.OPEN)
if dialog.ShowModal() == wx.ID_OK:
filepath = dialog.GetPath()
if os.path.isfile(filepath):
self.Controler.OpenXMLFile(filepath)
self.DeleteAllPages()
self.RefreshProjectTree()
self.RefreshTitle()
self.RefreshFileMenu()
self.RefreshEditMenu()
self.RefreshToolBar()
dialog.Destroy()
event.Skip()
def OnCloseTabMenu(self, event):
selected = self.GetPageSelection()
if selected >= 0:
tagname = self.GetPage(selected).GetTagName()
self.DeletePage(selected)
self.VariablePanelIndexer.RemoveVariablePanel(tagname)
if self.GetPageCount() > 0:
new_index = min(selected, self.GetPageCount() - 1)
tagname = self.GetPage(new_index).GetTagName()
self.SetPageSelection(new_index)
self.VariablePanelIndexer.ChangeVariablePanel(tagname)
self.RefreshTitle()
self.RefreshFileMenu()
self.RefreshEditMenu()
self.RefreshToolBar()
event.Skip()
def OnCloseProjectMenu(self, event):
self.DeleteAllPages()
self.VariablePanelIndexer.RemoveAllPanels()
self.ProjectTree.DeleteAllItems()
self.Controler.Reset()
self.RefreshTitle()
self.RefreshFileMenu()
self.RefreshEditMenu()
self.RefreshToolBar()
event.Skip()
def OnSaveProjectMenu(self, event):
if not self.ModeSolo and getattr(self, "_onsave", None) != None:
self._onsave()
else:
self.SaveProject()
event.Skip()
def OnSaveProjectAsMenu(self, event):
self.SaveProjectAs()
event.Skip()
def OnGenerateProgramMenu(self, event):
dialog = wx.FileDialog(self, "Choose a file", os.getcwd(), self.Controler.GetProgramFilePath(), "ST files (*.st)|*.st|All files|*.*", wx.SAVE|wx.CHANGE_DIR)
if dialog.ShowModal() == wx.ID_OK:
filepath = dialog.GetPath()
if os.path.isdir(os.path.dirname(filepath)):
result = self.Controler.GenerateProgram(filepath)
if not result:
message = wx.MessageDialog(self, "Can't generate program to file %s!"%filepath, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
else:
message = wx.MessageDialog(self, "Program was successfully generated!", "Done", wx.OK|wx.ICON_INFORMATION)
message.ShowModal()
message.Destroy()
else:
message = wx.MessageDialog(self, "%s is not a valid folder!"%os.path.dirname(filepath), "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
dialog.Destroy()
event.Skip()
def SaveProject(self):
result = self.Controler.SaveXMLFile()
if not result:
self.SaveProjectAs()
else:
self.RefreshTitle()
def SaveProjectAs(self):
filepath = self.Controler.GetFilePath()
if filepath != "":
directory, filename = os.path.split(filepath)
else:
directory, filename = os.getcwd(), "%(projectName)s.xml"%self.Controler.GetProjectProperties()
dialog = wx.FileDialog(self, "Choose a file", directory, filename, "PLCOpen files (*.xml)|*.xml|All files|*.*", wx.SAVE|wx.OVERWRITE_PROMPT)
if dialog.ShowModal() == wx.ID_OK:
filepath = dialog.GetPath()
if os.path.isdir(os.path.dirname(filepath)):
result = self.Controler.SaveXMLFile(filepath)
if not result:
message = wx.MessageDialog(self, "Can't save project to file %s!"%filepath, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
else:
message = wx.MessageDialog(self, "%s is not a valid folder!"%os.path.dirname(filepath), "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
self.RefreshTitle()
dialog.Destroy()
def OnPropertiesMenu(self, event):
self.ShowProperties()
event.Skip()
def OnQuitMenu(self, event):
self.Close()
event.Skip()
def ResetCurrentMode(self):
selected = self.GetPageSelection()
if selected != -1:
window = self.GetPage(selected)
window.SetMode(MODE_SELECTION)
if wx.VERSION < (2, 8, 0):
ToolBar = self.ToolBar
else:
ToolBar = self.Panes["ToolBar"]
if ToolBar:
ToolBar.ToggleTool(ID_PLCOPENEDITORTOOLBARSELECTION, True)
def ResetToolToggle(self, id):
if wx.VERSION < (2, 8, 0):
tool = self.ToolBar.FindById(id)
else:
tool = self.Panes["ToolBar"].FindById(id)
tool.SetToggle(False)
def OnSelectionTool(self, event):
selected = self.GetPageSelection()
if selected != -1:
self.GetPage(selected).SetMode(MODE_SELECTION)
event.Skip()
def OnCommentTool(self, event):
self.ResetToolToggle(ID_PLCOPENEDITORTOOLBARCOMMENT)
selected = self.GetPageSelection()
if selected != -1:
self.GetPage(selected).SetMode(MODE_COMMENT)
event.Skip()
def OnVariableTool(self, event):
self.ResetToolToggle(ID_PLCOPENEDITORTOOLBARVARIABLE)
selected = self.GetPageSelection()
if selected != -1:
self.GetPage(selected).SetMode(MODE_VARIABLE)
event.Skip()
def OnBlockTool(self, event):
self.ResetToolToggle(ID_PLCOPENEDITORTOOLBARBLOCK)
selected = self.GetPageSelection()
if selected != -1:
self.GetPage(selected).SetMode(MODE_BLOCK)
event.Skip()
def OnConnectionTool(self, event):
self.ResetToolToggle(ID_PLCOPENEDITORTOOLBARCONNECTION)
selected = self.GetPageSelection()
if selected != -1:
self.GetPage(selected).SetMode(MODE_CONNECTION)
event.Skip()
def OnPowerRailTool(self, event):
self.ResetToolToggle(ID_PLCOPENEDITORTOOLBARPOWERRAIL)
selected = self.GetPageSelection()
if selected != -1:
self.GetPage(selected).SetMode(MODE_POWERRAIL)
event.Skip()
def OnRungTool(self, event):
selected = self.GetPageSelection()
if selected != -1:
self.GetPage(selected).AddLadderRung()
event.Skip()
def OnCoilTool(self, event):
self.ResetToolToggle(ID_PLCOPENEDITORTOOLBARCOIL)
selected = self.GetPageSelection()
if selected != -1:
self.GetPage(selected).SetMode(MODE_COIL)
event.Skip()
def OnContactTool(self, event):
if self.DrawingMode == FREEDRAWING_MODE:
self.ResetToolToggle(ID_PLCOPENEDITORTOOLBARCONTACT)
selected = self.GetPageSelection()
if selected != -1:
if self.DrawingMode == FREEDRAWING_MODE:
self.GetPage(selected).SetMode(MODE_CONTACT)
else:
self.GetPage(selected).AddLadderContact()
event.Skip()
def OnBranchTool(self, event):
selected = self.GetPageSelection()
if selected != -1:
self.GetPage(selected).AddLadderBranch()
event.Skip()
def OnInitialStepTool(self, event):
self.ResetToolToggle(ID_PLCOPENEDITORTOOLBARINITIALSTEP)
selected = self.GetPageSelection()
if selected != -1:
self.GetPage(selected).SetMode(MODE_INITIALSTEP)
event.Skip()
def OnStepTool(self, event):
if self.GetDrawingMode() == FREEDRAWING_MODE:
self.ResetToolToggle(ID_PLCOPENEDITORTOOLBARSTEP)
selected = self.GetPageSelection()
if selected != -1:
if self.GetDrawingMode() == FREEDRAWING_MODE:
self.GetPage(selected).SetMode(MODE_STEP)
else:
self.GetPage(selected).AddStep()
event.Skip()
def OnActionBlockTool(self, event):
if self.GetDrawingMode() == FREEDRAWING_MODE:
self.ResetToolToggle(ID_PLCOPENEDITORTOOLBARACTIONBLOCK)
selected = self.GetPageSelection()
if selected != -1:
if self.GetDrawingMode() == FREEDRAWING_MODE:
self.GetPage(selected).SetMode(MODE_ACTION)
else:
self.GetPage(selected).AddStepAction()
event.Skip()
def OnTransitionTool(self, event):
self.ResetToolToggle(ID_PLCOPENEDITORTOOLBARTRANSITION)
selected = self.GetPageSelection()
if selected != -1:
self.GetPage(selected).SetMode(MODE_TRANSITION)
event.Skip()
def OnDivergenceTool(self, event):
if self.GetDrawingMode() == FREEDRAWING_MODE:
self.ResetToolToggle(ID_PLCOPENEDITORTOOLBARDIVERGENCE)
selected = self.GetPageSelection()
if selected != -1:
if self.GetDrawingMode() == FREEDRAWING_MODE:
self.GetPage(selected).SetMode(MODE_DIVERGENCE)
else:
self.GetPage(selected).AddDivergence()
event.Skip()
def OnJumpTool(self, event):
if self.GetDrawingMode() == FREEDRAWING_MODE:
self.ResetToolToggle(ID_PLCOPENEDITORTOOLBARJUMP)
selected = self.GetPageSelection()
if selected != -1:
if self.GetDrawingMode() == FREEDRAWING_MODE:
self.GetPage(selected).SetMode(MODE_JUMP)
else:
self.GetPage(selected).AddJump()
event.Skip()
def OnPouSelectedChanged(self, event):
if wx.VERSION < (2, 8, 0) or event.GetActive():
old_selected = self.GetPageSelection()
if old_selected >= 0:
self.GetPage(old_selected).ResetBuffer()
if wx.VERSION >= (2, 8, 0):
window = event.GetEventObject().GetViewer()
else:
selected = event.GetSelection()
if selected >= 0:
window = self.GetPage(selected)
else:
window = None
if window:
if self.ProjectTree:
root = self.ProjectTree.GetRootItem()
words = window.GetTagName().split("::")
if words[0] == "P":
self.SelectProjectTreeItem(root, [(words[1], ITEM_POU)])
elif words[0] == "T":
self.SelectProjectTreeItem(root, [(words[1], ITEM_POU), (words[2], ITEM_TRANSITION)])
elif words[0] == "A":
self.SelectProjectTreeItem(root, [(words[1], ITEM_POU), (words[2], ITEM_ACTION)])
elif words[0] == "C":
self.SelectProjectTreeItem(root, [(words[1], ITEM_CONFIGURATION)])
elif words[0] == "R":
self.SelectProjectTreeItem(root, [(words[1], ITEM_CONFIGURATION), (words[2], ITEM_RESOURCE)])
window.RefreshView()
self.VariablePanelIndexer.ChangeVariablePanel(window.GetTagName())
self.RefreshFileMenu()
self.RefreshEditMenu()
self.RefreshToolBar()
event.Skip()
def SelectProjectTreeItem(self, root, items):
found = False
if self.ProjectTree:
if wx.VERSION >= (2, 6, 0):
item, root_cookie = self.ProjectTree.GetFirstChild(root)
else:
item, root_cookie = self.ProjectTree.GetFirstChild(root, 0)
while item.IsOk() and not found:
if (self.ProjectTree.GetItemText(item), self.ProjectTree.GetPyData(item)) == items[0]:
if len(items) == 1:
self.SelectedItem = item
self.ProjectTree.SelectItem(item)
wx.CallAfter(self.ResetSelectedItem)
return True
else:
found = self.SelectProjectTreeItem(item, items[1:])
else:
found = self.SelectProjectTreeItem(item, items)
item, root_cookie = self.ProjectTree.GetNextChild(root, root_cookie)
return found
def OnProjectTreeBeginDrag(self, event):
if wx.Platform == '__WXMSW__':
self.SelectedItem = event.GetItem()
if self.SelectedItem is not None and self.ProjectTree.GetPyData(self.SelectedItem) == ITEM_POU:
block_name = self.ProjectTree.GetItemText(self.SelectedItem)
block_type = self.Controler.GetPouType(block_name)
if block_type != "program":
data = wx.TextDataObject(str((block_name, block_type, "")))
dragSource = wx.DropSource(self.ProjectTree)
dragSource.SetData(data)
dragSource.DoDragDrop()
self.ResetSelectedItem()
def RefreshEditorNames(self, item_type, old_name, new_name):
for i in xrange(self.GetPageCount()):
editor = self.GetPage(i)
editor.RefreshName(item_type, old_name, new_name)
def OnProjectTreeItemEndEdit(self, event):
message = None
abort = False
new_name = event.GetLabel()
if new_name != "":
if not TestIdentifier(new_name):
message = "\"%s\" is not a valid identifier!"%new_name
elif new_name.upper() in IEC_KEYWORDS:
message = "\"%s\" is a keyword. It can't be used!"%new_name
else:
item = event.GetItem()
old_name = self.ProjectTree.GetItemText(item)
itemtype = self.ProjectTree.GetPyData(item)
if itemtype == ITEM_PROJECT:
self.Controler.SetProjectProperties(name = new_name)
elif itemtype == ITEM_DATATYPE:
if new_name.upper() in [name.upper() for name in self.Controler.GetProjectDataTypeNames() if name != old_name]:
message = "\"%s\" data type already exists!"%new_name
abort = True
if not abort:
self.Controler.ChangeDataTypeName(old_name, new_name)
self.RefreshEditorNames(itemtype, old_name, new_name)
self.RefreshPageTitles()
elif itemtype == ITEM_POU:
if new_name.upper() in [name.upper() for name in self.Controler.GetProjectPouNames() if name != old_name]:
message = "\"%s\" pou already exists!"%new_name
abort = True
elif new_name.upper() in [name.upper() for name in self.Controler.GetProjectPouVariables()]:
messageDialog = wx.MessageDialog(self, "A variable is defined with \"%s\" as name. It can generate a conflict. Do you wish to continue?"%new_name, "Error", wx.YES_NO|wx.ICON_QUESTION)
if messageDialog.ShowModal() == wx.ID_NO:
abort = True
messageDialog.Destroy()
if not abort:
self.Controler.ChangePouName(old_name, new_name)
self.RefreshEditorNames(itemtype, old_name, new_name)
self.RefreshPageTitles()
elif itemtype == ITEM_TRANSITION:
parent = self.ProjectTree.GetItemParent(item)
parent_type = self.ProjectTree.GetPyData(parent)
while parent_type != ITEM_POU:
parent = self.ProjectTree.GetItemParent(parent)
parent_type = self.ProjectTree.GetPyData(parent)
pou_name = self.ProjectTree.GetItemText(parent)
if new_name.upper() in [name.upper() for name in self.Controler.GetProjectPouNames()]:
message = "A pou with \"%s\" as name exists!"%new_name
elif new_name.upper() in [name.upper() for name in self.Controler.GetProjectPouVariables(pou_name) if name != old_name]:
message = "A variable with \"%s\" as name already exists in this pou!"%new_name
else:
self.Controler.ChangePouTransitionName(pou_name, old_name, new_name)
self.RefreshEditorNames(itemtype, old_name, new_name)
self.RefreshPageTitles()
elif itemtype == ITEM_ACTION:
parent = self.ProjectTree.GetItemParent(item)
parent_type = self.ProjectTree.GetPyData(parent)
while parent_type != ITEM_POU:
parent = self.ProjectTree.GetItemParent(parent)
parent_type = self.ProjectTree.GetPyData(parent)
pou_name = self.ProjectTree.GetItemText(parent)
if new_name.upper() in [name.upper() for name in self.Controler.GetProjectPouNames()]:
message = "A pou with \"%s\" as name exists!"%new_name
elif new_name.upper() in [name.upper() for name in self.Controler.GetProjectPouVariables(pou_name) if name != old_name]:
message = "A variable with \"%s\" as name already exists in this pou!"%new_name
else:
self.Controler.ChangePouActionName(pou_name, old_name, new_name)
self.RefreshEditorNames(itemtype, old_name, new_name)
self.RefreshPageTitles()
elif itemtype == ITEM_CONFIGURATION:
if new_name.upper() in [name.upper() for name in self.Controler.GetProjectConfigNames() if name != old_name]:
message = "\"%s\" config already exists!"%new_name
abort = True
elif new_name.upper() in [name.upper() for name in self.Controler.GetProjectPouNames()]:
messageDialog = wx.MessageDialog(self, "A pou is defined with \"%s\" as name. It can generate a conflict. Do you wish to continue?"%new_name, "Error", wx.YES_NO|wx.ICON_QUESTION)
if messageDialog.ShowModal() == wx.ID_NO:
abort = True
messageDialog.Destroy()
elif new_name.upper() in [name.upper() for name in self.Controler.GetProjectPouVariables()]:
messageDialog = wx.MessageDialog(self, "A variable is defined with \"%s\" as name. It can generate a conflict. Do you wish to continue?"%new_name, "Error", wx.YES_NO|wx.ICON_QUESTION)
if messageDialog.ShowModal() == wx.ID_NO:
abort = True
messageDialog.Destroy()
if not abort:
self.Controler.ChangeConfigurationName(old_name, new_name)
self.RefreshEditorNames(itemtype, old_name, new_name)
self.RefreshPageTitles()
elif itemtype == ITEM_RESOURCE:
parent = self.ProjectTree.GetItemParent(item)
parent_type = self.ProjectTree.GetPyData(parent)
while parent_type != ITEM_CONFIGURATION:
parent = self.ProjectTree.GetItemParent(parent)
parent_type = self.ProjectTree.GetPyData(parent)
config_name = self.ProjectTree.GetItemText(parent)
if new_name.upper() in [name.upper() for name in self.Controler.GetProjectConfigNames()]:
message = "\"%s\" config already exists!"%new_name
abort = True
elif new_name.upper() in [name.upper() for name in self.Controler.GetProjectPouNames()]:
messageDialog = wx.MessageDialog(self, "A pou is defined with \"%s\" as name. It can generate a conflict. Do you wish to continue?"%new_name, "Error", wx.YES_NO|wx.ICON_QUESTION)
if messageDialog.ShowModal() == wx.ID_NO:
abort = True
messageDialog.Destroy()
elif new_name.upper() in [name.upper() for name in self.Controler.GetProjectPouVariables()]:
messageDialog = wx.MessageDialog(self, "A variable is defined with \"%s\" as name. It can generate a conflict. Do you wish to continue?"%new_name, "Error", wx.YES_NO|wx.ICON_QUESTION)
if messageDialog.ShowModal() == wx.ID_NO:
abort = True
messageDialog.Destroy()
if not abort:
self.Controler.ChangeConfigurationResourceName(config_name, old_name, new_name)
self.RefreshEditorNames(itemtype, old_name, new_name)
self.RefreshPageTitles()
if message or abort:
if message:
messageDialog = wx.MessageDialog(self, message, "Error", wx.OK|wx.ICON_ERROR)
messageDialog.ShowModal()
messageDialog.Destroy()
item = event.GetItem()
wx.CallAfter(self.ProjectTree.EditLabel, item)
event.Veto()
else:
wx.CallAfter(self.RefreshProjectTree)
selected = self.GetPageSelection()
if selected != -1:
self.GetPage(selected).RefreshView()
event.Skip()
def OnProjectTreeItemBeginEdit(self, event):
selected = event.GetItem()
if self.ProjectTree.GetPyData(selected) == ITEM_UNEDITABLE:
event.Veto()
else:
event.Skip()
def OnProjectTreeItemActivated(self, event):
selected = event.GetItem()
if self.ProjectTree.IsExpanded(selected):
self.ProjectTree.Collapse(selected)
else:
self.ProjectTree.Expand(selected)
name = self.ProjectTree.GetItemText(selected)
data = self.ProjectTree.GetPyData(selected)
if name == "Properties":
self.ShowProperties()
if data == ITEM_DATATYPE:
self.EditProjectElement(data, self.Controler.ComputeDataTypeName(name))
elif data == ITEM_POU:
self.EditProjectElement(data, self.Controler.ComputePouName(name))
elif data == ITEM_CONFIGURATION:
self.EditProjectElement(data, self.Controler.ComputeConfigurationName(name))
elif data == ITEM_RESOURCE:
item = self.ProjectTree.GetItemParent(selected)
item_type = self.ProjectTree.GetPyData(item)
while item_type != ITEM_CONFIGURATION:
item = self.ProjectTree.GetItemParent(item)
item_type = self.ProjectTree.GetPyData(item)
config_name = self.ProjectTree.GetItemText(item)
self.EditProjectElement(data, self.Controler.ComputeConfigurationResourceName(config_name, name))
elif data in [ITEM_TRANSITION, ITEM_ACTION]:
item = self.ProjectTree.GetItemParent(selected)
item_type = self.ProjectTree.GetPyData(item)
while item_type != ITEM_POU:
item = self.ProjectTree.GetItemParent(item)
item_type = self.ProjectTree.GetPyData(item)
pou_name = self.ProjectTree.GetItemText(item)
if data == ITEM_TRANSITION:
tagname = self.Controler.ComputePouTransitionName(pou_name, name)
elif data == ITEM_ACTION:
tagname = self.Controler.ComputePouActionName(pou_name, name)
self.EditProjectElement(data, tagname)
event.Skip()
def ProjectTreeItemSelect(self, select_item):
name = self.ProjectTree.GetItemText(select_item)
data = self.ProjectTree.GetPyData(select_item)
if data == ITEM_DATATYPE:
self.EditProjectElement(data, self.Controler.ComputeDataTypeName(name), True)
elif data == ITEM_POU:
self.EditProjectElement(data, self.Controler.ComputePouName(name), True)
elif data == ITEM_CONFIGURATION:
self.EditProjectElement(data, self.Controler.ComputeConfigurationName(name), True)
elif data == ITEM_RESOURCE:
item = self.ProjectTree.GetItemParent(select_item)
item_type = self.ProjectTree.GetPyData(item)
while item_type != ITEM_CONFIGURATION:
item = self.ProjectTree.GetItemParent(item)
item_type = self.ProjectTree.GetPyData(item)
config_name = self.ProjectTree.GetItemText(item)
self.EditProjectElement(data, self.Controler.ComputeConfigurationResourceName(config_name, name), True)
elif data in [ITEM_TRANSITION, ITEM_ACTION]:
item = self.ProjectTree.GetItemParent(select_item)
item_type = self.ProjectTree.GetPyData(item)
while item_type != ITEM_POU:
item = self.ProjectTree.GetItemParent(item)
item_type = self.ProjectTree.GetPyData(item)
pou_name = self.ProjectTree.GetItemText(item)
if data == ITEM_TRANSITION:
tagname = self.Controler.ComputePouTransitionName(pou_name, name)
elif data == ITEM_ACTION:
tagname = self.Controler.ComputePouActionName(pou_name, name)
self.EditProjectElement(data, tagname, True)
def OnProjectTreeLeftUp(self, event):
if self.SelectedItem is not None:
self.ProjectTree.SelectItem(self.SelectedItem)
self.ProjectTreeItemSelect(self.SelectedItem)
wx.CallAfter(self.ResetSelectedItem)
event.Skip()
def OnProjectTreeItemSelected(self, event):
self.ProjectTreeItemSelect(event.GetItem())
event.Skip()
def OnProjectTreeItemChanging(self, event):
if self.ProjectTree.GetPyData(event.GetItem()) != ITEM_UNEDITABLE and self.SelectedItem is None:
self.SelectedItem = event.GetItem()
event.Veto()
else:
event.Skip()
def IsOpened(self, tagname):
for idx in xrange(self.GetPageCount()):
if self.GetPage(idx).IsViewing(tagname):
return idx
return None
def EditProjectElement(self, elementtype, tagname, onlyopened = False):
openedidx = self.IsOpened(tagname)
if openedidx is not None:
old_selected = self.GetPageSelection()
if old_selected != openedidx:
if old_selected >= 0:
self.GetPage(old_selected).ResetBuffer()
self.SetPageSelection(openedidx)
self.GetPage(openedidx).RefreshView()
self.VariablePanelIndexer.ChangeVariablePanel(tagname)
self.RefreshPageTitles()
self.RefreshFileMenu()
self.RefreshEditMenu()
self.RefreshToolBar()
elif not onlyopened:
if wx.VERSION >= (2, 8, 0):
if elementtype == ITEM_CONFIGURATION:
new_window = MDIConfigurationEditor(self, tagname, self, self.Controler)
elif elementtype == ITEM_RESOURCE:
new_window = MDIResourceEditor(self, tagname, self, self.Controler)
elif elementtype in [ITEM_POU, ITEM_TRANSITION, ITEM_ACTION]:
bodytype = self.Controler.GetEditedElementBodyType(tagname)
if bodytype == "FBD":
new_window = MDIViewer(self, tagname, self, self.Controler)
elif bodytype == "LD":
new_window = LD_MDIViewer(self, tagname, self, self.Controler)
elif bodytype == "SFC":
new_window = SFC_MDIViewer(self, tagname, self, self.Controler)
else:
new_window = TextMDIViewer(self, tagname, self, self.Controler)
viewer = new_window.GetViewer()
viewer.SetTextSyntax(elementtype)
if bodytype == "IL":
viewer.SetKeywords(IL_KEYWORDS)
else:
viewer.SetKeywords(ST_KEYWORDS)
elif elementtype == ITEM_DATATYPE:
new_window = MDIDataTypeEditor(self, tagname, self, self.Controler)
new_window.Bind(wx.EVT_ACTIVATE, self.OnPouSelectedChanged)
new_window.Bind(wx.EVT_CLOSE, self.OnPageClose)
new_window.Layout()
else:
if elementtype == ITEM_CONFIGURATION:
new_window = ConfigurationEditor(self.TabsOpened, tagname, self, self.Controler)
self.TabsOpened.AddPage(new_window, "")
elif elementtype == ITEM_RESOURCE:
new_window = ResourceEditor(self.TabsOpened, tagname, self, self.Controler)
self.TabsOpened.AddPage(new_window, "")
elif elementtype in [ITEM_POU, ITEM_TRANSITION, ITEM_ACTION]:
bodytype = self.Controler.GetEditedElementBodyType(tagname)
if bodytype == "FBD":
new_window = Viewer(self.TabsOpened, tagname, self, self.Controler)
elif bodytype == "LD":
new_window = LD_Viewer(self.TabsOpened, tagname, self, self.Controler)
elif bodytype == "SFC":
new_window = SFC_Viewer(self.TabsOpened, tagname, self, self.Controler)
else:
new_window = TextViewer(self.TabsOpened, tagname, self, self.Controler)
new_window.SetTextSyntax(elementtype)
if bodytype == "IL":
new_window.SetKeywords(IL_KEYWORDS)
else:
new_window.SetKeywords(ST_KEYWORDS)
self.TabsOpened.AddPage(new_window, "")
elif elementtype == ITEM_DATATYPE:
new_window = DataTypeEditor(self.TabsOpened, tagname, self, self.Controler)
self.TabsOpened.AddPage(new_window, "")
if elementtype == ITEM_CONFIGURATION:
self.VariablePanelIndexer.AddVariablePanel(tagname, "config")
elif elementtype == ITEM_RESOURCE:
self.VariablePanelIndexer.AddVariablePanel(tagname, "resource")
elif elementtype in [ITEM_POU, ITEM_TRANSITION, ITEM_ACTION]:
words = tagname.split("::")
self.VariablePanelIndexer.AddVariablePanel(tagname, self.Controler.GetPouType(words[1]))
self.VariablePanelIndexer.ChangeVariablePanel(tagname)
openedidx = self.IsOpened(tagname)
old_selected = self.GetPageSelection()
if old_selected != openedidx:
if old_selected >= 0:
self.GetPage(old_selected).ResetBuffer()
for i in xrange(self.GetPageCount()):
window = self.GetPage(i)
if window.GetTagName() == tagname:
self.SetPageSelection(i)
window.RefreshView()
window.SetFocus()
self.RefreshPageTitles()
self.RefreshFileMenu()
self.RefreshEditMenu()
self.RefreshToolBar()
def OnProjectTreeRightUp(self, event):
cursor_pos = event.GetPosition()
selected = self.ProjectTree.GetSelection()
if self.ProjectTree.GetPyData(selected) == ITEM_POU:
name = self.ProjectTree.GetItemText(selected)
if self.Controler.GetPouBodyType(name) == "SFC":
self.PopupMenuXY(self.SFCMenu, cursor_pos.x, cursor_pos.y)
elif self.ProjectTree.GetPyData(selected) == ITEM_CONFIGURATION:
self.PopupMenu(self.ConfigMenu)
event.Skip()
def RefreshProjectTree(self):
infos = self.Controler.GetProjectInfos()
root = self.ProjectTree.GetRootItem()
if not root.IsOk():
root = self.ProjectTree.AddRoot(infos["name"])
self.GenerateTreeBranch(root, infos)
self.ProjectTree.Expand(self.ProjectTree.GetRootItem())
def GenerateTreeBranch(self, root, infos):
to_delete = []
self.ProjectTree.SetItemText(root, infos["name"])
self.ProjectTree.SetPyData(root, infos["type"])
if infos["type"] == ITEM_POU and wx.Platform != '__WXMSW__':
self.ProjectTree.SetItemImage(root, LANGUAGES.index(self.Controler.GetPouBodyType(infos["name"])))
if wx.VERSION >= (2, 5, 0):
item, root_cookie = self.ProjectTree.GetFirstChild(root)
else:
item, root_cookie = self.ProjectTree.GetFirstChild(root, 0)
for values in infos["values"]:
if not item.IsOk():
item = self.ProjectTree.AppendItem(root, "")
if wx.Platform != '__WXMSW__':
item, root_cookie = self.ProjectTree.GetNextChild(root, root_cookie)
self.GenerateTreeBranch(item, values)
item, root_cookie = self.ProjectTree.GetNextChild(root, root_cookie)
while item.IsOk():
to_delete.append(item)
item, root_cookie = self.ProjectTree.GetNextChild(root, root_cookie)
for item in to_delete:
self.ProjectTree.Delete(item)
def ResetToolBar(self):
for item in self.CurrentToolBar:
if wx.VERSION >= (2, 6, 0):
self.Unbind(wx.EVT_MENU, id=item)
else:
self.Disconnect(id=item, eventType=wx.wxEVT_COMMAND_MENU_SELECTED)
if wx.VERSION < (2, 8, 0):
ToolBar = self.ToolBar
else:
ToolBar = self.Panes["ToolBar"]
if ToolBar:
ToolBar.DeleteTool(item)
ToolBar.Realize()
if wx.VERSION >= (2, 8, 0):
self.AUIManager.GetPane("ToolBar").BestSize(ToolBar.GetBestSize())
self.AUIManager.Update()
def RefreshToolBar(self):
selected = self.GetPageSelection()
if selected != -1:
language = self.Controler.GetEditedElementBodyType(self.GetPage(selected).GetTagName())
else:
language = None
if language is not None and language != self.CurrentLanguage:
self.ResetToolBar()
self.CurrentLanguage = language
self.CurrentToolBar = []
if wx.VERSION < (2, 8, 0):
ToolBar = self.ToolBar
else:
ToolBar = self.Panes["ToolBar"]
if ToolBar:
for radio, modes, id, method, picture, help in ToolBarItems[language]:
if modes & self.DrawingMode:
if radio or self.DrawingMode == FREEDRAWING_MODE:
ToolBar.AddRadioTool(id, wx.Bitmap(os.path.join(CWD, "Images", picture)), wx.NullBitmap, help)
else:
ToolBar.AddSimpleTool(id, wx.Bitmap(os.path.join(CWD, "Images", picture)), help)
self.Bind(wx.EVT_TOOL, getattr(self, method), id=id)
self.CurrentToolBar.append(id)
ToolBar.Realize()
if wx.VERSION >= (2, 8, 0):
self.AUIManager.GetPane("ToolBar").BestSize(ToolBar.GetBestSize())
self.AUIManager.Update()
elif not language:
self.ResetToolBar()
self.CurrentLanguage = language
self.ResetCurrentMode()
def RefreshPageTitles(self):
for idx in xrange(self.GetPageCount()):
window = self.GetPage(idx)
words = window.GetTagName().split("::")
self.SetPageText(idx, "-".join(words[1:]))
def OnRefreshMenu(self, event):
selected = self.GetPageSelection()
if selected != -1:
window = self.GetPage(selected)
window.RefreshView()
self.VariablePanelIndexer.RefreshVariablePanel(window.GetTagName())
event.Skip()
def OnUndoMenu(self, event):
self.Controler.LoadPrevious()
selected = self.GetPageSelection()
if selected != -1:
window = self.GetPage(selected)
window.RefreshView()
self.VariablePanelIndexer.RefreshVariablePanel(window.GetTagName())
self.RefreshTitle()
self.RefreshEditMenu()
self.RefreshProjectTree()
event.Skip()
def OnRedoMenu(self, event):
self.Controler.LoadNext()
selected = self.GetPageSelection()
if selected != -1:
window = self.GetPage(selected)
window.RefreshView()
self.VariablePanelIndexer.RefreshVariablePanel(window.GetTagName())
self.RefreshTitle()
self.RefreshEditMenu()
self.RefreshProjectTree()
event.Skip()
def OnCutMenu(self, event):
selected = self.GetPageSelection()
if selected != -1:
try:
self.GetPage(selected).Cut()
except:
pass
event.Skip()
def OnCopyMenu(self, event):
selected = self.GetPageSelection()
if selected != -1:
try:
self.GetPage(selected).Copy()
except:
pass
event.Skip()
def OnPasteMenu(self, event):
selected = self.GetPageSelection()
if selected != -1:
try:
self.GetPage(selected).Paste()
except:
pass
event.Skip()
def OnAddDataTypeMenu(self, event):
dialog = DataTypeDialog(self, "Add a new data type", "Please enter data type name", "", wx.OK|wx.CANCEL)
dialog.SetDataTypeNames(self.Controler.GetProjectDataTypeNames())
if dialog.ShowModal() == wx.ID_OK:
self.Controler.ProjectAddDataType(dialog.GetValue())
self.RefreshTitle()
self.RefreshEditMenu()
self.RefreshProjectTree()
dialog.Destroy()
event.Skip()
def OnRemoveDataTypeMenu(self, event):
datatypes = self.Controler.GetProjectDataTypeNames()
dialog = wx.SingleChoiceDialog(self, "Select Data Type to remove:", "Data Type Remove", datatypes, wx.OK|wx.CANCEL)
if dialog.ShowModal() == wx.ID_OK:
selected = dialog.GetStringSelection()
if not self.Controler.DataTypeIsUsed(selected):
self.Controler.ProjectRemoveDataType(selected)
tagname = self.Controler.ComputeDataTypeName(selected)
idx = self.IsOpened(tagname)
if idx is not None:
self.DeletePage(idx)
self.RefreshTitle()
self.RefreshEditMenu()
self.RefreshProjectTree()
self.RefreshToolBar()
else:
message = wx.MessageDialog(self, "%s is used by one or more POUs. It can't be removed!"%selected, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
event.Skip()
def OnAddPouMenu(self, event):
dialog = PouDialog(self)
dialog.SetPouNames(self.Controler.GetProjectPouNames())
dialog.SetPouElementNames(self.Controler.GetProjectPouVariables())
if dialog.ShowModal() == wx.ID_OK:
values = dialog.GetValues()
self.Controler.ProjectAddPou(values["pouName"], values["pouType"], values["language"])
self.RefreshTitle()
self.RefreshEditMenu()
self.RefreshProjectTree()
dialog.Destroy()
event.Skip()
def OnRemovePouMenu(self, event):
pous = self.Controler.GetProjectPouNames()
dialog = wx.SingleChoiceDialog(self, "Select POU to remove:", "POU Remove", pous, wx.OK|wx.CANCEL)
if dialog.ShowModal() == wx.ID_OK:
selected = dialog.GetStringSelection()
if not self.Controler.PouIsUsed(selected):
self.Controler.ProjectRemovePou(selected)
tagname = self.Controler.ComputePouName(selected)
idx = self.IsOpened(tagname)
if idx is not None:
self.VariablePanelIndexer.RemoveVariablePanel(self.GetPage(idx).GetTagName())
self.DeletePage(idx)
self.RefreshTitle()
self.RefreshEditMenu()
self.RefreshProjectTree()
self.RefreshToolBar()
else:
message = wx.MessageDialog(self, "%s is used by one or more POUs. It can't be removed!"%selected, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
event.Skip()
def OnAddConfigurationMenu(self, event):
dialog = ConfigurationNameDialog(self, "Please enter configuration name", "Add new configuration")
dialog.SetPouNames(self.Controler.GetProjectPouNames())
dialog.SetPouElementNames(self.Controler.GetProjectPouVariables())
if dialog.ShowModal() == wx.ID_OK:
value = dialog.GetValue()
self.Controler.ProjectAddConfiguration(value)
self.RefreshTitle()
self.RefreshEditMenu()
self.RefreshProjectTree()
dialog.Destroy()
event.Skip()
def OnRemoveConfigurationMenu(self, event):
configs = self.Controler.GetProjectConfigNames()
dialog = wx.SingleChoiceDialog(self, "Please select a configuration", "Remove configuration", configs, wx.OK|wx.CANCEL)
if dialog.ShowModal() == wx.ID_OK:
selected = dialog.GetStringSelection()
self.Controler.ProjectRemoveConfiguration(selected)
tagname = self.Controler.ComputeConfigurationName(selected)
idx = self.IsOpened(tagname)
if idx is not None:
self.VariablePanelIndexer.RemoveVariablePanel(self.GetPage(idx).GetTagName())
self.DeletePage(idx)
self.RefreshTitle()
self.RefreshEditMenu()
self.RefreshProjectTree()
event.Skip()
def OnAddPouTransitionMenu(self, event):
selected = self.ProjectTree.GetSelection()
if self.ProjectTree.GetPyData(selected) == ITEM_POU:
pouname = self.ProjectTree.GetItemText(selected)
if self.Controler.GetPouBodyType(pouname) == "SFC":
dialog = PouTransitionDialog(self)
dialog.SetPouNames(self.Controler.GetProjectPouNames())
dialog.SetPouElementNames(self.Controler.GetProjectPouVariables(pouname))
if dialog.ShowModal() == wx.ID_OK:
values = dialog.GetValues()
self.Controler.ProjectAddPouTransition(pouname, values["transitionName"], values["language"])
self.RefreshTitle()
self.RefreshEditMenu()
self.RefreshProjectTree()
dialog.Destroy()
event.Skip()
def OnRemovePouTransitionMenu(self, event):
selected = self.ProjectTree.GetSelection()
if self.ProjectTree.GetPyData(selected) == ITEM_POU:
pouname = self.ProjectTree.GetItemText(selected)
if self.Controler.GetPouBodyType(pouname) == "SFC":
transitions = self.Controler.GetPouTransitions(pouname)
dialog = wx.SingleChoiceDialog(self, "Select Transition to remove:", "Remove transition", transitions, wx.OK|wx.CANCEL)
if dialog.ShowModal() == wx.ID_OK:
selected = dialog.GetStringSelection()
self.Controler.ProjectRemovePouTransition(pouname, selected)
tagname = self.Controler.ComputePouTransitionName(pouname, selected)
idx = self.IsOpened(tagname)
if idx is not None:
self.VariablePanelIndexer.RemoveVariablePanel(self.GetPage(idx).GetTagName())
self.DeletePage(idx)
self.RefreshTitle()
self.RefreshEditMenu()
self.RefreshProjectTree()
dialog.Destroy()
event.Skip()
def OnAddPouActionMenu(self, event):
selected = self.ProjectTree.GetSelection()
if self.ProjectTree.GetPyData(selected) == ITEM_POU:
pouname = self.ProjectTree.GetItemText(selected)
if self.Controler.GetPouBodyType(pouname) == "SFC":
dialog = PouActionDialog(self)
dialog.SetPouNames(self.Controler.GetProjectPouNames())
dialog.SetPouElementNames(self.Controler.GetProjectPouVariables(pouname))
if dialog.ShowModal() == wx.ID_OK:
values = dialog.GetValues()
self.Controler.ProjectAddPouAction(pouname, values["actionName"], values["language"])
self.RefreshTitle()
self.RefreshEditMenu()
self.RefreshProjectTree()
dialog.Destroy()
event.Skip()
def OnRemovePouActionMenu(self, event):
selected = self.ProjectTree.GetSelection()
if self.ProjectTree.GetPyData(selected) == ITEM_POU:
pouname = self.ProjectTree.GetItemText(selected)
if self.Controler.GetPouBodyType(pouname) == "SFC":
actions = self.Controler.GetPouActions(pouname)
dialog = wx.SingleChoiceDialog(self, "Select Action to remove:", "Remove action", actions, wx.OK|wx.CANCEL)
if dialog.ShowModal() == wx.ID_OK:
selected = dialog.GetStringSelection()
self.Controler.ProjectRemovePouAction(pouname, selected)
tagname = self.Controler.ComputePouActionName(pouname, selected)
idx = self.IsOpened(tagname)
if idx is not None:
self.VariablePanelIndexer.RemoveVariablePanel(self.GetPage(idx).GetTagName())
self.DeletePage(idx)
self.RefreshTitle()
self.RefreshEditMenu()
self.RefreshProjectTree()
dialog.Destroy()
event.Skip()
def OnAddResourceMenu(self, event):
selected = self.ProjectTree.GetSelection()
if self.ProjectTree.GetPyData(selected) == ITEM_CONFIGURATION:
config_name = self.ProjectTree.GetItemText(selected)
dialog = ResourceNameDialog(self, "Please enter resource name", "Add new resource")
dialog.SetPouNames(self.Controler.GetProjectPouNames())
dialog.SetPouElementNames(self.Controler.GetProjectPouVariables())
if dialog.ShowModal() == wx.ID_OK:
value = dialog.GetValue()
self.Controler.ProjectAddConfigurationResource(config_name, value)
tagname = self.Controler.ComputeConfigurationResourceName(config_name, selected)
idx = self.IsOpened(tagname)
if idx is not None:
self.VariablePanelIndexer.RemoveVariablePanel(self.GetPage(idx).GetTagName())
self.DeletePage(idx)
self.RefreshTitle()
self.RefreshEditMenu()
self.RefreshProjectTree()
dialog.Destroy()
event.Skip()
def OnRemoveResourceMenu(self, event):
selected = self.ProjectTree.GetSelection()
if self.ProjectTree.GetPyData(selected) == ITEM_CONFIGURATION:
config_name = self.ProjectTree.GetItemText(selected)
infos = self.Controler.GetProjectInfos()
resources = []
for config in infos["configs"]:
if config["name"] == config_name:
resources = config["resources"]
dialog = wx.SingleChoiceDialog(self, "Select Resource to remove:", "Remove resource", resources, wx.OK|wx.CANCEL)
if dialog.ShowModal() == wx.ID_OK:
resource = dialog.GetStringSelection()
self.Controler.ProjectRemoveConfigurationResource(config_name, resource)
self.RefreshTitle()
self.RefreshEditMenu()
self.RefreshProjectTree()
dialog.Destroy()
event.Skip()
def OnPLCOpenMenu(self, event):
result = OpenPDFDoc()
if type(result) == StringType:
message = wx.MessageDialog(self, result, "ERROR", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
event.Skip()
current_num = 0
def GetNewNum():
global current_num
current_num += 1
return current_num
#-------------------------------------------------------------------------------
# Create Project Dialog
#-------------------------------------------------------------------------------
[ID_PROJECTDIALOG, ID_PROJECTDIALOGPROJECTNAME,
ID_PROJECTDIALOGCOMPANYNAME, ID_PROJECTDIALOGCOMPANYURL,
ID_PROJECTDIALOGPRODUCTNAME, ID_PROJECTDIALOGPRODUCTVERSION,
ID_PROJECTDIALOGPRODUCTRELEASE, ID_PROJECTDIALOGCONTENTDESCRIPTION,
ID_PROJECTDIALOGSTATICTEXT1, ID_PROJECTDIALOGSTATICTEXT2,
ID_PROJECTDIALOGSTATICTEXT3, ID_PROJECTDIALOGSTATICTEXT4,
ID_PROJECTDIALOGSTATICTEXT5, ID_PROJECTDIALOGSTATICTEXT6,
ID_PROJECTDIALOGSTATICTEXT7,
] = [wx.NewId() for _init_ctrls in range(15)]
class ProjectDialog(wx.Dialog):
if wx.VERSION < (2, 6, 0):
def Bind(self, event, function, id = None):
if id is not None:
event(self, id, function)
else:
event(self, function)
def _init_coll_flexGridSizer1_Items(self, parent):
parent.AddSizer(self.MainSizer, 0, border=20, flag=wx.GROW|wx.TOP|wx.LEFT|wx.RIGHT)
parent.AddSizer(self.ButtonSizer, 0, border=20, flag=wx.ALIGN_RIGHT|wx.BOTTOM|wx.LEFT|wx.RIGHT)
def _init_coll_flexGridSizer1_Growables(self, parent):
parent.AddGrowableCol(0)
parent.AddGrowableRow(0)
def _init_coll_MainSizer_Items(self, parent):
parent.AddWindow(self.staticText1, 0, border=4, flag=wx.GROW|wx.TOP)
parent.AddWindow(self.ProjectName, 0, border=0, flag=wx.GROW)
parent.AddWindow(self.staticText2, 0, border=4, flag=wx.GROW|wx.TOP)
parent.AddWindow(self.CompanyName, 0, border=0, flag=wx.GROW)
parent.AddWindow(self.staticText3, 0, border=4, flag=wx.GROW|wx.TOP)
parent.AddWindow(self.CompanyURL, 0, border=0, flag=wx.GROW)
parent.AddWindow(self.staticText4, 0, border=4, flag=wx.GROW|wx.TOP)
parent.AddWindow(self.ProductName, 0, border=0, flag=wx.GROW)
parent.AddWindow(self.staticText5, 0, border=4, flag=wx.GROW|wx.TOP)
parent.AddWindow(self.ProductVersion, 0, border=0, flag=wx.GROW)
parent.AddWindow(self.staticText6, 0, border=4, flag=wx.GROW|wx.TOP)
parent.AddWindow(self.ProductRelease, 0, border=0, flag=wx.GROW)
parent.AddWindow(self.staticText7, 0, border=4, flag=wx.GROW|wx.TOP)
parent.AddWindow(self.ContentDescription, 0, border=0, flag=wx.GROW)
def _init_coll_MainSizer_Growables(self, parent):
parent.AddGrowableCol(1)
parent.AddGrowableRow(6)
def _init_sizers(self):
self.flexGridSizer1 = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=10)
self.MainSizer = wx.FlexGridSizer(cols=2, hgap=0, rows=7, vgap=15)
self._init_coll_flexGridSizer1_Items(self.flexGridSizer1)
self._init_coll_flexGridSizer1_Growables(self.flexGridSizer1)
self._init_coll_MainSizer_Items(self.MainSizer)
self._init_coll_MainSizer_Growables(self.MainSizer)
self.SetSizer(self.flexGridSizer1)
def _init_ctrls(self, prnt):
wx.Dialog.__init__(self, id=ID_PROJECTDIALOG,
name='ProjectDialog', parent=prnt, pos=wx.Point(376, 223),
size=wx.Size(550, 450), style=wx.DEFAULT_DIALOG_STYLE,
title='Create a new project')
self.SetClientSize(wx.Size(550, 450))
self.staticText1 = wx.StaticText(id=ID_PROJECTDIALOGSTATICTEXT1,
label='Project Name (required):', name='staticText1', parent=self,
pos=wx.Point(0, 0), size=wx.Size(200, 17), style=0)
self.ProjectName = wx.TextCtrl(id=ID_PROJECTDIALOGPROJECTNAME,
name='ProjectName', parent=self, pos=wx.Point(0, 0),
size=wx.Size(0, 24), style=0)
self.staticText2 = wx.StaticText(id=ID_PROJECTDIALOGSTATICTEXT2,
label='Company Name (required):', name='staticText2', parent=self,
pos=wx.Point(0, 0), size=wx.Size(200, 17), style=0)
self.CompanyName = wx.TextCtrl(id=ID_PROJECTDIALOGCOMPANYNAME,
name='CompanyName', parent=self, pos=wx.Point(0, 0),
size=wx.Size(0, 24), style=0)
self.staticText3 = wx.StaticText(id=ID_PROJECTDIALOGSTATICTEXT3,
label='Company URL (optional):', name='staticText3', parent=self,
pos=wx.Point(0, 0), size=wx.Size(200, 17), style=0)
self.CompanyURL = wx.TextCtrl(id=ID_PROJECTDIALOGCOMPANYURL,
name='CompanyURL', parent=self, pos=wx.Point(0, 0),
size=wx.Size(0, 24), style=0)
self.staticText4 = wx.StaticText(id=ID_PROJECTDIALOGSTATICTEXT4,
label='Product Name (required):', name='staticText4', parent=self,
pos=wx.Point(0, 0), size=wx.Size(200, 17), style=0)
self.ProductName = wx.TextCtrl(id=ID_PROJECTDIALOGPRODUCTNAME,
name='ProductName', parent=self, pos=wx.Point(0, 0),
size=wx.Size(0, 24), style=0)
self.staticText5 = wx.StaticText(id=ID_PROJECTDIALOGSTATICTEXT5,
label='Product Version (required):', name='staticText5', parent=self,
pos=wx.Point(0, 0), size=wx.Size(200, 17), style=0)
self.ProductVersion = wx.TextCtrl(id=ID_PROJECTDIALOGPRODUCTVERSION,
name='ProductVersion', parent=self, pos=wx.Point(0, 0),
size=wx.Size(0, 24), style=0)
self.staticText6 = wx.StaticText(id=ID_PROJECTDIALOGSTATICTEXT6,
label='Product Release (optional):', name='staticText6', parent=self,
pos=wx.Point(0, 0), size=wx.Size(200, 17), style=0)
self.ProductRelease = wx.TextCtrl(id=ID_PROJECTDIALOGPRODUCTRELEASE,
name='ProductRelease', parent=self, pos=wx.Point(0, 0),
size=wx.Size(0, 24), style=0)
self.staticText7 = wx.StaticText(id=ID_PROJECTDIALOGSTATICTEXT7,
label='Content Description (optional):', name='staticText7', parent=self,
pos=wx.Point(0, 0), size=wx.Size(200, 17), style=0)
self.ContentDescription = wx.TextCtrl(id=ID_PROJECTDIALOGCONTENTDESCRIPTION,
name='ProductRelease', parent=self, pos=wx.Point(0, 0),
size=wx.Size(0, 120), style=wx.TE_MULTILINE)
self.ButtonSizer = self.CreateButtonSizer(wx.OK|wx.CANCEL|wx.CENTRE)
self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.ButtonSizer.GetAffirmativeButton().GetId())
self._init_sizers()
def __init__(self, parent):
self._init_ctrls(parent)
def OnOK(self, event):
error = []
if self.ProjectName.GetValue() == "":
error.append("Project Name")
if self.CompanyName.GetValue() == "":
error.append("Company Name")
if self.ProductName.GetValue() == "":
error.append("Product Name")
if self.ProductVersion.GetValue() == "":
error.append("Product Version")
if len(error) > 0:
text = ""
for i, item in enumerate(error):
if i == 0:
text += item
elif i == len(error) - 1:
text += " and %s"%item
else:
text += ", %s"%item
message = wx.MessageDialog(self, "Form isn't complete. %s must be filled!"%text, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
else:
self.EndModal(wx.ID_OK)
def SetValues(self, values):
for item, value in values.items():
if item == "projectName":
self.ProjectName.SetValue(value)
elif item == "companyName":
self.CompanyName.SetValue(value)
elif item == "companyURL":
self.CompanyURL.SetValue(value)
elif item == "productName":
self.ProductName.SetValue(value)
elif item == "productVersion":
self.ProductVersion.SetValue(value)
elif item == "productRelease":
self.ProductRelease.SetValue(value)
elif item == "contentDescription":
self.ContentDescription.SetValue(value)
def GetValues(self):
values = {}
values["projectName"] = self.ProjectName.GetValue()
values["companyName"] = self.CompanyName.GetValue()
if self.CompanyURL.GetValue() != None:
values["companyURL"] = self.CompanyURL.GetValue()
values["productName"] = self.ProductName.GetValue()
values["productVersion"] = self.ProductVersion.GetValue()
if self.ProductRelease.GetValue() != None:
values["productRelease"] = self.ProductRelease.GetValue()
if self.ProductRelease.GetValue() != None:
values["contentDescription"] = self.ContentDescription.GetValue()
return values
#-------------------------------------------------------------------------------
# Edit Step Name Dialog
#-------------------------------------------------------------------------------
class DataTypeDialog(wx.TextEntryDialog):
if wx.VERSION < (2, 6, 0):
def Bind(self, event, function, id = None):
if id is not None:
event(self, id, function)
else:
event(self, function)
def __init__(self, parent, message, caption = "Please enter text", defaultValue = "",
style = wx.OK|wx.CANCEL|wx.CENTRE, pos = wx.DefaultPosition):
wx.TextEntryDialog.__init__(self, parent, message, caption, defaultValue, style, pos)
self.DataTypeNames = []
if wx.VERSION >= (2, 8, 0):
self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.GetSizer().GetItem(2).GetSizer().GetItem(1).GetSizer().GetAffirmativeButton().GetId())
elif wx.VERSION >= (2, 6, 0):
self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.GetSizer().GetItem(3).GetSizer().GetAffirmativeButton().GetId())
else:
self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.GetSizer().GetItem(3).GetSizer().GetChildren()[0].GetSizer().GetChildren()[0].GetWindow().GetId())
def OnOK(self, event):
datatype_name = self.GetSizer().GetItem(1).GetWindow().GetValue()
if datatype_name == "":
message = wx.MessageDialog(self, "You must type a name!", "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
elif not TestIdentifier(datatype_name):
message = wx.MessageDialog(self, "\"%s\" is not a valid identifier!"%datatype_name, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
elif datatype_name.upper() in IEC_KEYWORDS:
message = wx.MessageDialog(self, "\"%s\" is a keyword. It can't be used!"%datatype_name, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
elif datatype_name.upper() in self.DataTypeNames:
message = wx.MessageDialog(self, "\"%s\" data type already exists!"%datatype_name, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
else:
self.EndModal(wx.ID_OK)
def SetDataTypeNames(self, datatype_names):
self.DataTypeNames = [datatype_name.upper() for datatype_name in datatype_names]
def GetValue(self):
return self.GetSizer().GetItem(1).GetWindow().GetValue()
#-------------------------------------------------------------------------------
# Create Pou Dialog
#-------------------------------------------------------------------------------
[ID_POUDIALOG, ID_POUDIALOGPOUNAME,
ID_POUDIALOGPOUTYPE, ID_POUDIALOGLANGUAGE, ID_POUDIALOGSTATICTEXT1,
ID_POUDIALOGSTATICTEXT2, ID_POUDIALOGSTATICTEXT3,
] = [wx.NewId() for _init_ctrls in range(7)]
class PouDialog(wx.Dialog):
if wx.VERSION < (2, 6, 0):
def Bind(self, event, function, id = None):
if id is not None:
event(self, id, function)
else:
event(self, function)
def _init_coll_flexGridSizer1_Items(self, parent):
parent.AddSizer(self.MainSizer, 0, border=20, flag=wx.GROW|wx.TOP|wx.LEFT|wx.RIGHT)
parent.AddSizer(self.ButtonSizer, 0, border=20, flag=wx.ALIGN_RIGHT|wx.BOTTOM|wx.LEFT|wx.RIGHT)
def _init_coll_flexGridSizer1_Growables(self, parent):
parent.AddGrowableCol(0)
parent.AddGrowableRow(0)
def _init_coll_MainSizer_Items(self, parent):
parent.AddWindow(self.staticText1, 0, border=4, flag=wx.GROW|wx.TOP)
parent.AddWindow(self.PouName, 0, border=0, flag=wx.GROW)
parent.AddWindow(self.staticText2, 0, border=4, flag=wx.GROW|wx.TOP)
parent.AddWindow(self.PouType, 0, border=0, flag=wx.GROW)
parent.AddWindow(self.staticText3, 0, border=4, flag=wx.GROW|wx.TOP)
parent.AddWindow(self.Language, 0, border=0, flag=wx.GROW)
def _init_coll_MainSizer_Growables(self, parent):
parent.AddGrowableCol(1)
def _init_sizers(self):
self.flexGridSizer1 = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=10)
self.MainSizer = wx.FlexGridSizer(cols=2, hgap=0, rows=3, vgap=15)
self._init_coll_flexGridSizer1_Items(self.flexGridSizer1)
self._init_coll_flexGridSizer1_Growables(self.flexGridSizer1)
self._init_coll_MainSizer_Items(self.MainSizer)
self._init_coll_MainSizer_Growables(self.MainSizer)
self.SetSizer(self.flexGridSizer1)
def _init_ctrls(self, prnt):
wx.Dialog.__init__(self, id=ID_POUDIALOG,
name='PouDialog', parent=prnt, pos=wx.Point(376, 223),
size=wx.Size(300, 200), style=wx.DEFAULT_DIALOG_STYLE,
title='Create a new POU')
self.SetClientSize(wx.Size(300, 200))
self.staticText1 = wx.StaticText(id=ID_POUDIALOGSTATICTEXT1,
label='POU Name:', name='staticText1', parent=self,
pos=wx.Point(0, 0), size=wx.Size(100, 17), style=0)
self.PouName = wx.TextCtrl(id=ID_POUDIALOGPOUNAME,
name='POUName', parent=self, pos=wx.Point(0, 0),
size=wx.Size(0, 24), style=0)
self.staticText2 = wx.StaticText(id=ID_POUDIALOGSTATICTEXT2,
label='POU Type:', name='staticText2', parent=self,
pos=wx.Point(0, 0), size=wx.Size(100, 17), style=0)
self.PouType = wx.Choice(id=ID_POUDIALOGPOUTYPE,
name='POUType', parent=self, pos=wx.Point(0, 0),
size=wx.Size(0, 24), style=0)
self.Bind(wx.EVT_CHOICE, self.OnTypeChanged, id=ID_POUDIALOGPOUTYPE)
self.staticText3 = wx.StaticText(id=ID_POUDIALOGSTATICTEXT3,
label='Language:', name='staticText3', parent=self,
pos=wx.Point(0, 0), size=wx.Size(100, 17), style=0)
self.Language = wx.Choice(id=ID_POUDIALOGLANGUAGE,
name='Language', parent=self, pos=wx.Point(0, 0),
size=wx.Size(0, 24), style=0)
self.ButtonSizer = self.CreateButtonSizer(wx.OK|wx.CANCEL|wx.CENTRE)
self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.ButtonSizer.GetAffirmativeButton().GetId())
self._init_sizers()
def __init__(self, parent):
self._init_ctrls(parent)
for option in ["function","functionBlock","program"]:
self.PouType.Append(option)
self.RefreshLanguage()
self.PouNames = []
self.PouElementNames = []
def OnOK(self, event):
error = []
pou_name = self.PouName.GetValue()
if pou_name == "":
error.append("POU Name")
if self.PouType.GetStringSelection() == "":
error.append("POU Type")
if self.Language.GetStringSelection() == "":
error.append("Language")
if len(error) > 0:
text = ""
for i, item in enumerate(error):
if i == 0:
text += item
elif i == len(error) - 1:
text += " and %s"%item
else:
text += ", %s"%item
message = wx.MessageDialog(self, "Form isn't complete. %s must be filled!"%text, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
elif not TestIdentifier(pou_name):
message = wx.MessageDialog(self, "\"%s\" is not a valid identifier!"%pou_name, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
elif pou_name.upper() in IEC_KEYWORDS:
message = wx.MessageDialog(self, "\"%s\" is a keyword. It can't be used!"%pou_name, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
elif pou_name.upper() in self.PouNames:
message = wx.MessageDialog(self, "\"%s\" pou already exists!"%pou_name, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
elif pou_name.upper() in self.PouElementNames:
message = wx.MessageDialog(self, "A pou has an element with \"%s\" for name. It can generate a conflict. Do you wish to continue?"%pou_name, "Warning", wx.YES_NO|wx.ICON_EXCLAMATION)
result = message.ShowModal()
message.Destroy()
if result == wx.ID_YES:
self.EndModal(wx.ID_OK)
else:
self.EndModal(wx.ID_OK)
def RefreshLanguage(self):
selection = self.Language.GetStringSelection()
self.Language.Clear()
for option in ["IL","ST","LD","FBD","SFC"]:
if option != "SFC" or self.PouType.GetStringSelection() != "function":
self.Language.Append(option)
if self.Language.FindString(selection) != wx.NOT_FOUND:
self.Language.SetStringSelection(selection)
def OnTypeChanged(self, event):
self.RefreshLanguage()
event.Skip()
def SetPouNames(self, pou_names):
self.PouNames = [pou_name.upper() for pou_name in pou_names]
def SetPouElementNames(self, element_names):
self.PouElementNames = [element_name.upper() for element_name in element_names]
def SetValues(self, values):
for item, value in values.items():
if item == "pouName":
self.PouName.SetValue(value)
elif item == "pouType":
self.PouType.SetStringSelection(value)
elif item == "language":
self.Language.SetStringSelection(value)
def GetValues(self):
values = {}
values["pouName"] = self.PouName.GetValue()
values["pouType"] = self.PouType.GetStringSelection()
values["language"] = self.Language.GetStringSelection()
return values
#-------------------------------------------------------------------------------
# Create Pou Transition Dialog
#-------------------------------------------------------------------------------
[ID_POUTRANSITIONDIALOG, ID_POUTRANSITIONDIALOGTRANSITIONNAME,
ID_POUTRANSITIONDIALOGLANGUAGE, ID_POUTRANSITIONDIALOGSTATICTEXT1,
ID_POUTRANSITIONDIALOGSTATICTEXT2,
] = [wx.NewId() for _init_ctrls in range(5)]
class PouTransitionDialog(wx.Dialog):
if wx.VERSION < (2, 6, 0):
def Bind(self, event, function, id = None):
if id is not None:
event(self, id, function)
else:
event(self, function)
def _init_coll_flexGridSizer1_Items(self, parent):
parent.AddSizer(self.MainSizer, 0, border=20, flag=wx.GROW|wx.TOP|wx.LEFT|wx.RIGHT)
parent.AddSizer(self.ButtonSizer, 0, border=20, flag=wx.ALIGN_RIGHT|wx.BOTTOM|wx.LEFT|wx.RIGHT)
def _init_coll_flexGridSizer1_Growables(self, parent):
parent.AddGrowableCol(0)
parent.AddGrowableRow(0)
def _init_coll_MainSizer_Items(self, parent):
parent.AddWindow(self.staticText1, 0, border=4, flag=wx.GROW|wx.TOP)
parent.AddWindow(self.TransitionName, 0, border=0, flag=wx.GROW)
parent.AddWindow(self.staticText2, 0, border=4, flag=wx.GROW|wx.TOP)
parent.AddWindow(self.Language, 0, border=0, flag=wx.GROW)
def _init_coll_MainSizer_Growables(self, parent):
parent.AddGrowableCol(1)
def _init_sizers(self):
self.flexGridSizer1 = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=10)
self.MainSizer = wx.FlexGridSizer(cols=2, hgap=0, rows=2, vgap=15)
self._init_coll_flexGridSizer1_Items(self.flexGridSizer1)
self._init_coll_flexGridSizer1_Growables(self.flexGridSizer1)
self._init_coll_MainSizer_Items(self.MainSizer)
self._init_coll_MainSizer_Growables(self.MainSizer)
self.SetSizer(self.flexGridSizer1)
def _init_ctrls(self, prnt):
wx.Dialog.__init__(self, id=ID_POUTRANSITIONDIALOG,
name='PouTransitionDialog', parent=prnt, pos=wx.Point(376, 223),
size=wx.Size(350, 200), style=wx.DEFAULT_DIALOG_STYLE,
title='Create a new transition')
self.SetClientSize(wx.Size(350, 160))
self.staticText1 = wx.StaticText(id=ID_POUTRANSITIONDIALOGSTATICTEXT1,
label='Transition Name:', name='staticText1', parent=self,
pos=wx.Point(0, 0), size=wx.Size(120, 17), style=0)
self.TransitionName = wx.TextCtrl(id=ID_POUTRANSITIONDIALOGTRANSITIONNAME,
name='TransitionName', parent=self, pos=wx.Point(0, 0),
size=wx.Size(0, 24), style=0)
self.staticText2 = wx.StaticText(id=ID_POUTRANSITIONDIALOGSTATICTEXT2,
label='Language:', name='staticText2', parent=self,
pos=wx.Point(0, 0), size=wx.Size(120, 17), style=0)
self.Language = wx.Choice(id=ID_POUTRANSITIONDIALOGLANGUAGE,
name='Language', parent=self, pos=wx.Point(0, 0),
size=wx.Size(0, 24), style=0)
self.ButtonSizer = self.CreateButtonSizer(wx.OK|wx.CANCEL|wx.CENTRE)
self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.ButtonSizer.GetAffirmativeButton().GetId())
self._init_sizers()
def __init__(self, parent):
self._init_ctrls(parent)
for option in ["IL","ST","LD","FBD"]:
self.Language.Append(option)
self.PouNames = []
self.PouElementNames = []
def OnOK(self, event):
error = []
transition_name = self.TransitionName.GetValue()
if self.TransitionName.GetValue() == "":
error.append("Transition Name")
if self.Language.GetStringSelection() == "":
error.append("Language")
if len(error) > 0:
text = ""
for i, item in enumerate(error):
if i == 0:
text += item
elif i == len(error) - 1:
text += " and %s"%item
else:
text += ", %s"%item
message = wx.MessageDialog(self, "Form isn't complete. %s must be filled!"%text, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
elif not TestIdentifier(transition_name):
message = wx.MessageDialog(self, "\"%s\" is not a valid identifier!"%transition_name, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
elif transition_name.upper() in IEC_KEYWORDS:
message = wx.MessageDialog(self, "\"%s\" is a keyword. It can't be used!"%transition_name, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
elif transition_name.upper() in self.PouNames:
message = wx.MessageDialog(self, "A pou with \"%s\" for name exists!"%transition_name, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
elif transition_name.upper() in self.PouElementNames:
message = wx.MessageDialog(self, "\"%s\" element for this pou already exists!"%transition_name, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
else:
self.EndModal(wx.ID_OK)
def SetPouNames(self, pou_names):
self.PouNames = [pou_name.upper() for pou_name in pou_names]
def SetPouElementNames(self, pou_names):
self.PouElementNames = [pou_name.upper() for pou_name in pou_names]
def SetValues(self, values):
for item, value in values.items():
if item == "transitionName":
self.TransitionName.SetValue(value)
elif item == "language":
self.Language.SetStringSelection(value)
def GetValues(self):
values = {}
values["transitionName"] = self.TransitionName.GetValue()
values["language"] = self.Language.GetStringSelection()
return values
#-------------------------------------------------------------------------------
# Create Pou Action Dialog
#-------------------------------------------------------------------------------
[ID_POUACTIONDIALOG, ID_POUACTIONDIALOGACTIONNAME,
ID_POUACTIONDIALOGLANGUAGE, ID_POUACTIONDIALOGSTATICTEXT1,
ID_POUACTIONDIALOGSTATICTEXT2,
] = [wx.NewId() for _init_ctrls in range(5)]
class PouActionDialog(wx.Dialog):
if wx.VERSION < (2, 6, 0):
def Bind(self, event, function, id = None):
if id is not None:
event(self, id, function)
else:
event(self, function)
def _init_coll_flexGridSizer1_Items(self, parent):
parent.AddSizer(self.MainSizer, 0, border=20, flag=wx.GROW|wx.TOP|wx.LEFT|wx.RIGHT)
parent.AddSizer(self.ButtonSizer, 0, border=20, flag=wx.ALIGN_RIGHT|wx.BOTTOM|wx.LEFT|wx.RIGHT)
def _init_coll_flexGridSizer1_Growables(self, parent):
parent.AddGrowableCol(0)
parent.AddGrowableRow(0)
def _init_coll_MainSizer_Items(self, parent):
parent.AddWindow(self.staticText1, 0, border=4, flag=wx.GROW|wx.TOP)
parent.AddWindow(self.ActionName, 0, border=0, flag=wx.GROW)
parent.AddWindow(self.staticText2, 0, border=4, flag=wx.GROW|wx.TOP)
parent.AddWindow(self.Language, 0, border=0, flag=wx.GROW)
def _init_coll_MainSizer_Growables(self, parent):
parent.AddGrowableCol(1)
def _init_sizers(self):
self.flexGridSizer1 = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=10)
self.MainSizer = wx.FlexGridSizer(cols=2, hgap=0, rows=2, vgap=15)
self._init_coll_flexGridSizer1_Items(self.flexGridSizer1)
self._init_coll_flexGridSizer1_Growables(self.flexGridSizer1)
self._init_coll_MainSizer_Items(self.MainSizer)
self._init_coll_MainSizer_Growables(self.MainSizer)
self.SetSizer(self.flexGridSizer1)
def _init_ctrls(self, prnt):
wx.Dialog.__init__(self, id=ID_POUACTIONDIALOG,
name='PouActionDialog', parent=prnt, pos=wx.Point(376, 223),
size=wx.Size(320, 200), style=wx.DEFAULT_DIALOG_STYLE|wx.RESIZE_BORDER,
title='Create a new action')
self.SetClientSize(wx.Size(320, 160))
self.staticText1 = wx.StaticText(id=ID_POUACTIONDIALOGSTATICTEXT1,
label='Action Name:', name='staticText1', parent=self,
pos=wx.Point(0, 0), size=wx.Size(100, 17), style=0)
self.ActionName = wx.TextCtrl(id=ID_POUACTIONDIALOGACTIONNAME,
name='ActionName', parent=self, pos=wx.Point(0, 0),
size=wx.Size(0, 24), style=0)
self.staticText2 = wx.StaticText(id=ID_POUACTIONDIALOGSTATICTEXT2,
label='Language:', name='staticText2', parent=self,
pos=wx.Point(0, 0), size=wx.Size(100, 17), style=0)
self.Language = wx.Choice(id=ID_POUACTIONDIALOGLANGUAGE,
name='Language', parent=self, pos=wx.Point(0, 0),
size=wx.Size(0, 24), style=0)
self.ButtonSizer = self.CreateButtonSizer(wx.OK|wx.CANCEL|wx.CENTRE)
self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.ButtonSizer.GetAffirmativeButton().GetId())
self._init_sizers()
def __init__(self, parent):
self._init_ctrls(parent)
for option in ["IL","ST","LD","FBD"]:
self.Language.Append(option)
self.PouNames = []
self.PouElementNames = []
def OnOK(self, event):
error = []
action_name = self.ActionName.GetValue()
if action_name == "":
error.append("Action Name")
if self.Language.GetStringSelection() == "":
error.append("Language")
if len(error) > 0:
text = ""
for i, item in enumerate(error):
if i == 0:
text += item
elif i == len(error) - 1:
text += " and %s"%item
else:
text += ", %s"%item
message = wx.MessageDialog(self, "Form isn't complete. %s must be filled!"%text, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
elif not TestIdentifier(action_name):
message = wx.MessageDialog(self, "\"%s\" is not a valid identifier!"%action_name, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
elif action_name.upper() in IEC_KEYWORDS:
message = wx.MessageDialog(self, "\"%s\" is a keyword. It can't be used!"%action_name, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
elif action_name.upper() in self.PouNames:
message = wx.MessageDialog(self, "A pou with \"%s\" for name exists!"%action_name, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
elif action_name.upper() in self.PouElementNames:
message = wx.MessageDialog(self, "\"%s\" element for this pou already exists!"%action_name, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
else:
self.EndModal(wx.ID_OK)
def SetPouNames(self, pou_names):
self.PouNames = [pou_name.upper() for pou_name in pou_names]
def SetPouElementNames(self, element_names):
self.PouElementNames = [element_name.upper() for element_name in element_names]
def SetValues(self, values):
for item, value in values.items():
if item == "actionName":
self.ActionName.SetValue(value)
elif item == "language":
self.Language.SetStringSelection(value)
def GetValues(self):
values = {}
values["actionName"] = self.ActionName.GetValue()
values["language"] = self.Language.GetStringSelection()
return values
#-------------------------------------------------------------------------------
# Configuration Name Dialog
#-------------------------------------------------------------------------------
class ConfigurationNameDialog(wx.TextEntryDialog):
if wx.VERSION < (2, 6, 0):
def Bind(self, event, function, id = None):
if id is not None:
event(self, id, function)
else:
event(self, function)
def __init__(self, parent, message, caption = "Please enter configuration name", defaultValue = "",
style = wx.OK|wx.CANCEL|wx.CENTRE, pos = wx.DefaultPosition):
wx.TextEntryDialog.__init__(self, parent, message, caption, defaultValue, style, pos)
self.PouNames = []
self.PouElementNames = []
self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.GetSizer().GetItem(3).GetSizer().GetAffirmativeButton().GetId())
def OnOK(self, event):
config_name = self.GetSizer().GetItem(1).GetWindow().GetValue()
if config_name == "":
message = wx.MessageDialog(self, "You must type a name!", "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
elif not TestIdentifier(config_name):
message = wx.MessageDialog(self, "\"%s\" is not a valid identifier!"%config_name, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
elif config_name.upper() in IEC_KEYWORDS:
message = wx.MessageDialog(self, "\"%s\" is a keyword. It can't be used!"%config_name, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
elif config_name.upper() in self.PouNames:
message = wx.MessageDialog(self, "A pou with \"%s\" as name exists!"%config_name, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
elif config_name.upper() in self.PouElementNames:
message = wx.MessageDialog(self, "A pou has an element with \"%s\" for name. It can generate a conflict. Do you wish to continue?"%config_name, "Warning", wx.YES_NO|wx.ICON_EXCLAMATION)
result = message.ShowModal()
message.Destroy()
if result == wx.ID_YES:
self.EndModal(wx.ID_OK)
else:
self.EndModal(wx.ID_OK)
def SetPouNames(self, pou_names):
self.PouNames = [pou_name.upper() for pou_name in pou_names]
def SetPouElementNames(self, pou_names):
self.PouElementNames = [pou_name.upper() for pou_name in pou_names]
def GetValue(self):
return self.GetSizer().GetItem(1).GetWindow().GetValue()
#-------------------------------------------------------------------------------
# Resource Name Dialog
#-------------------------------------------------------------------------------
class ResourceNameDialog(wx.TextEntryDialog):
if wx.VERSION < (2, 6, 0):
def Bind(self, event, function, id = None):
if id is not None:
event(self, id, function)
else:
event(self, function)
def __init__(self, parent, message, caption = "Please enter resource name", defaultValue = "",
style = wx.OK|wx.CANCEL|wx.CENTRE, pos = wx.DefaultPosition):
wx.TextEntryDialog.__init__(self, parent, message, caption, defaultValue, style, pos)
self.PouNames = []
self.PouElementNames = []
self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.GetSizer().GetItem(3).GetSizer().GetAffirmativeButton().GetId())
def OnOK(self, event):
resource_name = self.GetSizer().GetItem(1).GetWindow().GetValue()
if resource_name == "":
message = wx.MessageDialog(self, "You must type a name!", "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
elif not TestIdentifier(resource_name):
message = wx.MessageDialog(self, "\"%s\" is not a valid identifier!"%resource_name, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
elif resource_name.upper() in IEC_KEYWORDS:
message = wx.MessageDialog(self, "\"%s\" is a keyword. It can't be used!"%resource_name, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
elif resource_name.upper() in self.PouNames:
message = wx.MessageDialog(self, "A pou with \"%s\" as name exists!"%resource_name, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
elif resource_name.upper() in self.PouElementNames:
message = wx.MessageDialog(self, "A pou has an element with \"%s\" for name. It can generate a conflict. Do you wish to continue?"%resource_name, "Warning", wx.YES_NO|wx.ICON_EXCLAMATION)
result = message.ShowModal()
message.Destroy()
if result == wx.ID_YES:
self.EndModal(wx.ID_OK)
else:
self.EndModal(wx.ID_OK)
def SetPouNames(self, pou_names):
self.PouNames = [pou_name.upper() for pou_name in pou_names]
def SetPouElementNames(self, pou_names):
self.PouElementNames = [pou_name.upper() for pou_name in pou_names]
def GetValue(self):
return self.GetSizer().GetItem(1).GetWindow().GetValue()
#-------------------------------------------------------------------------------
# Variables Editor Panel
#-------------------------------------------------------------------------------
class VariablePanelIndexer(wx.Panel):
def _init_sizers(self):
self.MainSizer = wx.BoxSizer(wx.HORIZONTAL)
self.SetSizer(self.MainSizer)
def _init_ctrls(self, prnt):
wx.Panel.__init__(self, id=wx.NewId(),
name='VariablePanelIndexer', parent=prnt, pos=wx.Point(0, 0),
size=wx.Size(0, 300), style=wx.TAB_TRAVERSAL)
self._init_sizers()
def __init__(self, parent, controler):
self._init_ctrls(parent)
self.ParentWindow = parent
self.Controler = controler
self.VariablePanelList = {}
self.CurrentPanel = None
def AddVariablePanel(self, tagname, element_type):
new_panel = VariablePanel(self, self.ParentWindow, self.Controler, element_type)
new_panel.SetTagName(tagname)
new_panel.Hide()
new_panel.RefreshView()
self.MainSizer.AddWindow(new_panel, 1, border=0, flag=wx.GROW)
self.VariablePanelList[tagname] = new_panel
def RemoveVariablePanel(self, tagname):
if tagname in self.VariablePanelList:
panel = self.VariablePanelList.pop(tagname)
panel.Hide()
self.MainSizer.Remove(panel)
if self.CurrentPanel == tagname:
self.CurrentPanel = None
def RemoveAllPanels(self):
if self.CurrentPanel is not None and self.CurrentPanel in self.VariablePanelList:
self.VariablePanelList[self.CurrentPanel].Hide()
self.VariablePanelList = {}
self.MainSizer.Clear()
self.CurrentPanel = None
def UpdateVariablePanelTagName(self, old_tagname, new_tagname):
if old_name in self.VariablePanelList:
self.VariablePanelList[new_tagname] = self.VariablePanelList.pop(old_tagname)
if self.CurrentPanel == old_tagname:
self.CurrentPanel = new_tagname
def ChangeVariablePanel(self, tagname):
if tagname in self.VariablePanelList:
if tagname != self.CurrentPanel:
if self.CurrentPanel is not None:
self.VariablePanelList[self.CurrentPanel].Hide()
self.CurrentPanel = tagname
self.VariablePanelList[self.CurrentPanel].RefreshView()
self.VariablePanelList[self.CurrentPanel].Show()
self.MainSizer.Layout()
else:
if self.CurrentPanel is not None:
self.VariablePanelList[self.CurrentPanel].Hide()
self.CurrentPanel = None
self.MainSizer.Layout()
def RefreshVariablePanel(self, tagname):
if tagname in self.VariablePanelList:
self.VariablePanelList[self.CurrentPanel].RefreshView()
#-------------------------------------------------------------------------------
# Variables Editor Panel
#-------------------------------------------------------------------------------
class VariableTable(wx.grid.PyGridTableBase):
"""
A custom wx.grid.Grid Table using user supplied data
"""
def __init__(self, parent, data, colnames):
# The base class must be initialized *first*
wx.grid.PyGridTableBase.__init__(self)
self.data = data
self.old_value = None
self.colnames = colnames
self.Parent = parent
# XXX
# we need to store the row length and collength to
# see if the table has changed size
self._rows = self.GetNumberRows()
self._cols = self.GetNumberCols()
def GetNumberCols(self):
return len(self.colnames)
def GetNumberRows(self):
return len(self.data)
def GetColLabelValue(self, col):
if col < len(self.colnames):
return self.colnames[col]
def GetRowLabelValues(self, row):
return row
def GetValue(self, row, col):
if row < self.GetNumberRows():
if col == 0:
return self.data[row]["Number"]
name = str(self.data[row].get(self.GetColLabelValue(col), ""))
return name
def SetValue(self, row, col, value):
if col < len(self.colnames):
colname = self.GetColLabelValue(col)
if colname == "Name":
self.old_value = self.data[row][colname]
self.data[row][colname] = value
def GetValueByName(self, row, colname):
if row < self.GetNumberRows():
return self.data[row].get(colname)
def SetValueByName(self, row, colname, value):
if row < self.GetNumberRows():
self.data[row][colname] = value
def GetOldValue(self):
return self.old_value
def ResetView(self, grid):
"""
(wx.grid.Grid) -> Reset the grid view. Call this to
update the grid if rows and columns have been added or deleted
"""
grid.BeginBatch()
for current, new, delmsg, addmsg in [
(self._rows, self.GetNumberRows(), wx.grid.GRIDTABLE_NOTIFY_ROWS_DELETED, wx.grid.GRIDTABLE_NOTIFY_ROWS_APPENDED),
(self._cols, self.GetNumberCols(), wx.grid.GRIDTABLE_NOTIFY_COLS_DELETED, wx.grid.GRIDTABLE_NOTIFY_COLS_APPENDED),
]:
if new < current:
msg = wx.grid.GridTableMessage(self,delmsg,new,current-new)
grid.ProcessTableMessage(msg)
elif new > current:
msg = wx.grid.GridTableMessage(self,addmsg,new-current)
grid.ProcessTableMessage(msg)
self.UpdateValues(grid)
grid.EndBatch()
self._rows = self.GetNumberRows()
self._cols = self.GetNumberCols()
# update the column rendering scheme
self._updateColAttrs(grid)
# update the scrollbars and the displayed part of the grid
grid.AdjustScrollbars()
grid.ForceRefresh()
def UpdateValues(self, grid):
"""Update all displayed values"""
# This sends an event to the grid table to update all of the values
msg = wx.grid.GridTableMessage(self, wx.grid.GRIDTABLE_REQUEST_VIEW_GET_VALUES)
grid.ProcessTableMessage(msg)
def _updateColAttrs(self, grid):
"""
wx.grid.Grid -> update the column attributes to add the
appropriate renderer given the column name.
Otherwise default to the default renderer.
"""
typelist = None
accesslist = None
for row in range(self.GetNumberRows()):
for col in range(self.GetNumberCols()):
editor = None
renderer = None
colname = self.GetColLabelValue(col)
if col != 0 and self.GetValueByName(row, "Edit"):
grid.SetReadOnly(row, col, False)
if colname == "Name":
if self.Parent.PouIsUsed and self.GetValueByName(row, "Class") in ["Input", "Output", "InOut"]:
grid.SetReadOnly(row, col, True)
else:
editor = wx.grid.GridCellTextEditor()
renderer = wx.grid.GridCellStringRenderer()
elif colname == "Initial Value":
editor = wx.grid.GridCellTextEditor()
renderer = wx.grid.GridCellStringRenderer()
elif colname == "Location":
if self.GetValueByName(row, "Class") in ["Local", "Global"]:
editor = wx.grid.GridCellTextEditor()
renderer = wx.grid.GridCellStringRenderer()
else:
grid.SetReadOnly(row, col, True)
elif colname == "Class":
if len(self.Parent.ClassList) == 1 or self.Parent.PouIsUsed and self.GetValueByName(row, "Class") in ["Input", "Output", "InOut"]:
grid.SetReadOnly(row, col, True)
else:
editor = wx.grid.GridCellChoiceEditor()
excluded = []
if self.Parent.PouIsUsed:
excluded.extend(["Input","Output","InOut"])
editor.SetParameters(",".join([choice for choice in self.Parent.ClassList if choice not in excluded]))
elif colname in ["Retain", "Constant"]:
editor = wx.grid.GridCellChoiceEditor()
editor.SetParameters(self.Parent.OptionList)
elif colname == "Type":
editor = wx.grid.GridCellTextEditor()
else:
grid.SetReadOnly(row, col, True)
grid.SetCellEditor(row, col, editor)
grid.SetCellRenderer(row, col, renderer)
grid.SetCellBackgroundColour(row, col, wx.WHITE)
def SetData(self, data):
self.data = data
def GetData(self):
return self.data
def GetCurrentIndex(self):
return self.CurrentIndex
def SetCurrentIndex(self, index):
self.CurrentIndex = index
def AppendRow(self, row_content):
self.data.append(row_content)
def RemoveRow(self, row_index):
self.data.pop(row_index)
def GetRow(self, row_index):
return self.data[row_index]
def Empty(self):
self.data = []
self.editors = []
class VariableDropTarget(wx.TextDropTarget):
def __init__(self, parent):
wx.TextDropTarget.__init__(self)
self.ParentWindow = parent
def OnDropText(self, x, y, data):
col = self.ParentWindow.VariablesGrid.XToCol(x)
row = self.ParentWindow.VariablesGrid.YToRow(y - self.ParentWindow.VariablesGrid.GetColLabelSize())
if col != wx.NOT_FOUND and row != wx.NOT_FOUND:
if self.ParentWindow.Table.GetColLabelValue(col) != "Location":
return
message = None
if not self.ParentWindow.Table.GetValueByName(row, "Edit"):
message = "Can't affect a location to a function block instance"
elif self.ParentWindow.Table.GetValueByName(row, "Class") not in ["Local", "Global"]:
message = "Can affect a location only to local or global variables"
else:
try:
values = eval(data)
except:
values = None
if values and values[1] == "location":
location = values[0]
variable_type = self.ParentWindow.Table.GetValueByName(row, "Type")
message = None
if location[0].isdigit() and variable_type != "BOOL":
message = "Incompatible size of data between \"%s\" and \"BOOL\""%location
elif location[0] not in LOCATIONDATATYPES:
message = "Unrecognized data size \"%s\""%location[0]
elif variable_type not in LOCATIONDATATYPES[location[0]]:
message = "Incompatible size of data between \"%s\" and \"%s\""%(location, variable_type)
else:
dialog = wx.SingleChoiceDialog(self.ParentWindow, "Select a variable class:", "Variable class", ["Input", "Output", "Memory"], wx.OK|wx.CANCEL)
if dialog.ShowModal() == wx.ID_OK:
selected = dialog.GetSelection()
if selected == 0:
location = "%I" + location
elif selected == 1:
location = "%Q" + location
else:
location = "%M" + location
self.ParentWindow.Table.SetValue(row, col, location)
self.ParentWindow.Table.ResetView(self.ParentWindow.VariablesGrid)
self.ParentWindow.SaveValues()
dialog.Destroy()
if message is not None:
message = wx.MessageDialog(self.ParentWindow, message, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
[ID_VARIABLEEDITORPANEL, ID_VARIABLEEDITORPANELVARIABLESGRID,
ID_VARIABLEEDITORPANELRETURNTYPE, ID_VARIABLEEDITORPANELCLASSFILTER,
ID_VARIABLEEDITORPANELADDBUTTON, ID_VARIABLEEDITORPANELDELETEBUTTON,
ID_VARIABLEEDITORPANELUPBUTTON, ID_VARIABLEEDITORPANELDOWNBUTTON,
ID_VARIABLEEDITORPANELSTATICTEXT1, ID_VARIABLEEDITORPANELSTATICTEXT2,
ID_VARIABLEEDITORPANELSTATICTEXT3,
] = [wx.NewId() for _init_ctrls in range(11)]
class VariablePanel(wx.Panel):
if wx.VERSION < (2, 6, 0):
def Bind(self, event, function, id = None):
if id is not None:
event(self, id, function)
else:
event(self, function)
def _init_coll_MainSizer_Items(self, parent):
parent.AddWindow(self.VariablesGrid, 0, border=0, flag=wx.GROW)
parent.AddSizer(self.ControlPanelSizer, 0, border=0, flag=wx.GROW)
def _init_coll_MainSizer_Growables(self, parent):
parent.AddGrowableCol(0)
parent.AddGrowableRow(0)
def _init_coll_ControlPanelSizer_Items(self, parent):
parent.AddSizer(self.ChoicePanelSizer, 0, border=0, flag=wx.GROW)
parent.AddSizer(self.ButtonPanelSizer, 0, border=0, flag=wx.ALIGN_CENTER)
def _init_coll_ControlPanelSizer_Growables(self, parent):
parent.AddGrowableCol(0)
parent.AddGrowableRow(0)
parent.AddGrowableRow(1)
def _init_coll_ChoicePanelSizer_Items(self, parent):
parent.AddWindow(self.staticText1, 0, border=0, flag=wx.ALIGN_BOTTOM)
parent.AddWindow(self.ReturnType, 0, border=0, flag=0)
parent.AddWindow(self.staticText2, 0, border=0, flag=wx.ALIGN_BOTTOM)
parent.AddWindow(self.ClassFilter, 0, border=0, flag=0)
def _init_coll_ButtonPanelSizer_Items(self, parent):
parent.AddWindow(self.UpButton, 0, border=0, flag=0)
parent.AddWindow(self.AddButton, 0, border=0, flag=0)
parent.AddWindow(self.DownButton, 0, border=0, flag=0)
parent.AddWindow(self.DeleteButton, 0, border=0, flag=0)
def _init_coll_ButtonPanelSizer_Growables(self, parent):
parent.AddGrowableCol(0)
parent.AddGrowableCol(1)
parent.AddGrowableCol(2)
parent.AddGrowableCol(3)
parent.AddGrowableRow(0)
def _init_sizers(self):
self.MainSizer = wx.FlexGridSizer(cols=2, hgap=10, rows=1, vgap=0)
self.ControlPanelSizer = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=10)
self.ChoicePanelSizer = wx.GridSizer(cols=1, hgap=5, rows=4, vgap=5)
self.ButtonPanelSizer = wx.FlexGridSizer(cols=2, hgap=5, rows=2, vgap=0)
self._init_coll_MainSizer_Items(self.MainSizer)
self._init_coll_MainSizer_Growables(self.MainSizer)
self._init_coll_ControlPanelSizer_Items(self.ControlPanelSizer)
self._init_coll_ControlPanelSizer_Growables(self.ControlPanelSizer)
self._init_coll_ChoicePanelSizer_Items(self.ChoicePanelSizer)
self._init_coll_ButtonPanelSizer_Items(self.ButtonPanelSizer)
self._init_coll_ButtonPanelSizer_Growables(self.ButtonPanelSizer)
self.SetSizer(self.MainSizer)
def _init_ctrls(self, prnt):
wx.Panel.__init__(self, id=ID_VARIABLEEDITORPANEL,
name='VariableEditorPanel', parent=prnt, pos=wx.Point(0, 0),
size=wx.Size(0, 0), style=wx.TAB_TRAVERSAL)
self.staticText1 = wx.StaticText(id=ID_VARIABLEEDITORPANELSTATICTEXT1,
label='Return Type:', name='staticText1', parent=self,
pos=wx.Point(0, 0), size=wx.Size(95, 17), style=0)
self.ReturnType = wx.Choice(id=ID_VARIABLEEDITORPANELRETURNTYPE,
name='ReturnType', parent=self, pos=wx.Point(0, 0),
size=wx.Size(145, 24), style=0)
self.Bind(wx.EVT_CHOICE, self.OnReturnTypeChanged, id=ID_VARIABLEEDITORPANELRETURNTYPE)
self.staticText2 = wx.StaticText(id=ID_VARIABLEEDITORPANELSTATICTEXT2,
label='Class Filter:', name='staticText2', parent=self,
pos=wx.Point(0, 0), size=wx.Size(95, 17), style=0)
self.ClassFilter = wx.Choice(id=ID_VARIABLEEDITORPANELCLASSFILTER,
name='ClassFilter', parent=self, pos=wx.Point(0, 0),
size=wx.Size(145, 24), style=0)
self.Bind(wx.EVT_CHOICE, self.OnClassFilter, id=ID_VARIABLEEDITORPANELCLASSFILTER)
self.VariablesGrid = wx.grid.Grid(id=ID_VARIABLEEDITORPANELVARIABLESGRID,
name='VariablesGrid', parent=self, pos=wx.Point(0, 0),
size=wx.Size(0, 150), style=wx.VSCROLL)
self.VariablesGrid.SetFont(wx.Font(12, 77, wx.NORMAL, wx.NORMAL, False,
'Sans'))
self.VariablesGrid.SetLabelFont(wx.Font(10, 77, wx.NORMAL, wx.NORMAL,
False, 'Sans'))
self.VariablesGrid.SetSelectionBackground(wx.WHITE)
self.VariablesGrid.SetSelectionForeground(wx.BLACK)
if wx.VERSION >= (2, 6, 0):
self.VariablesGrid.Bind(wx.grid.EVT_GRID_CELL_CHANGE, self.OnVariablesGridCellChange)
self.VariablesGrid.Bind(wx.grid.EVT_GRID_SELECT_CELL, self.OnVariablesGridSelectCell)
self.VariablesGrid.Bind(wx.grid.EVT_GRID_CELL_LEFT_CLICK, self.OnVariablesGridCellLeftClick)
self.VariablesGrid.Bind(wx.grid.EVT_GRID_EDITOR_SHOWN, self.OnVariablesGridEditorShown)
else:
wx.grid.EVT_GRID_CELL_CHANGE(self.VariablesGrid, self.OnVariablesGridCellChange)
wx.grid.EVT_GRID_SELECT_CELL(self.VariablesGrid, self.OnVariablesGridSelectCell)
wx.grid.EVT_GRID_CELL_LEFT_CLICK(self.VariablesGrid, self.OnVariablesGridCellLeftClick)
wx.grid.EVT_GRID_EDITOR_SHOWN(self.VariablesGrid, self.OnVariablesGridEditorShown)
self.VariablesGrid.SetDropTarget(VariableDropTarget(self))
self.AddButton = wx.Button(id=ID_VARIABLEEDITORPANELADDBUTTON, label='Add',
name='AddButton', parent=self, pos=wx.Point(345, 340),
size=wx.Size(72, 32), style=0)
self.Bind(wx.EVT_BUTTON, self.OnAddButton, id=ID_VARIABLEEDITORPANELADDBUTTON)
self.DeleteButton = wx.Button(id=ID_VARIABLEEDITORPANELDELETEBUTTON, label='Delete',
name='DeleteButton', parent=self, pos=wx.Point(425, 340),
size=wx.Size(72, 32), style=0)
self.Bind(wx.EVT_BUTTON, self.OnDeleteButton, id=ID_VARIABLEEDITORPANELDELETEBUTTON)
self.UpButton = wx.Button(id=ID_VARIABLEEDITORPANELUPBUTTON, label='^',
name='UpButton', parent=self, pos=wx.Point(505, 340),
size=wx.Size(32, 32), style=0)
self.Bind(wx.EVT_BUTTON, self.OnUpButton, id=ID_VARIABLEEDITORPANELUPBUTTON)
self.DownButton = wx.Button(id=ID_VARIABLEEDITORPANELDOWNBUTTON, label='v',
name='DownButton', parent=self, pos=wx.Point(545, 340),
size=wx.Size(32, 32), style=0)
self.Bind(wx.EVT_BUTTON, self.OnDownButton, id=ID_VARIABLEEDITORPANELDOWNBUTTON)
self._init_sizers()
def __init__(self, parent, window, controler, element_type):
self._init_ctrls(parent)
self.ParentWindow = window
self.Controler = controler
self.ElementType = element_type
self.Filter = "All"
self.FilterChoices = []
self.FilterChoiceTransfer = {"All" : "All", "Interface" : "Interface",
" Input" : "Input", " Output" : "Output", " InOut" : "InOut",
" External" : "External", "Variables" : "Variables", " Local" : "Local",
" Temp" : "Temp", "Global" : "Global", "Access" : "Access"}
if element_type in ["config", "resource"]:
self.DefaultTypes = {"All" : "Global"}
self.DefaultValue = {"Name" : "", "Class" : "", "Type" : "INT", "Location" : "", "Initial Value" : "", "Retain" : "No", "Constant" : "No", "Edit" : True}
else:
self.DefaultTypes = {"All" : "Local", "Interface" : "Input", "Variables" : "Local"}
self.DefaultValue = {"Name" : "", "Class" : "", "Type" : "INT", "Location" : "", "Initial Value" : "", "Retain" : "No", "Constant" : "No", "Edit" : True}
if element_type in ["config", "resource"] or element_type in ["program", "transition", "action"]:
self.Table = VariableTable(self, [], ["#", "Name", "Class", "Type", "Location", "Initial Value", "Retain", "Constant"])
if element_type not in ["config", "resource"]:
self.FilterChoices = ["All","Interface"," Input"," Output"," InOut"," External","Variables"," Local"," Temp","Global","Access"]
else:
self.FilterChoices = ["All","Global","Access"]
self.ColSizes = [40, 80, 70, 80, 80, 80, 60, 70]
self.ColAlignements = [wx.ALIGN_CENTER, wx.ALIGN_LEFT, wx.ALIGN_LEFT, wx.ALIGN_LEFT, wx.ALIGN_LEFT, wx.ALIGN_LEFT, wx.ALIGN_CENTER, wx.ALIGN_CENTER]
else:
self.Table = VariableTable(self, [], ["#", "Name", "Class", "Type", "Initial Value", "Retain", "Constant"])
self.FilterChoices = ["All","Interface"," Input"," Output"," InOut"," External","Variables"," Local"," Temp"]
self.ColSizes = [40, 120, 70, 80, 120, 60, 70]
self.ColAlignements = [wx.ALIGN_CENTER, wx.ALIGN_LEFT, wx.ALIGN_LEFT, wx.ALIGN_LEFT, wx.ALIGN_LEFT, wx.ALIGN_CENTER, wx.ALIGN_CENTER]
for choice in self.FilterChoices:
self.ClassFilter.Append(choice)
reverse_transfer = {}
for filter, choice in self.FilterChoiceTransfer.items():
reverse_transfer[choice] = filter
self.ClassFilter.SetStringSelection(reverse_transfer[self.Filter])
self.RefreshTypeList()
self.OptionList = "Yes,No"
if element_type == "function":
for base_type in self.Controler.GetBaseTypes():
self.ReturnType.Append(base_type)
self.ReturnType.Enable(True)
else:
self.ReturnType.Enable(False)
self.staticText1.Hide()
self.ReturnType.Hide()
self.VariablesGrid.SetTable(self.Table)
self.VariablesGrid.SetRowLabelSize(0)
for col in range(self.Table.GetNumberCols()):
attr = wx.grid.GridCellAttr()
attr.SetAlignment(self.ColAlignements[col], wx.ALIGN_CENTRE)
self.VariablesGrid.SetColAttr(col, attr)
self.VariablesGrid.SetColSize(col, self.ColSizes[col])
def SetTagName(self, tagname):
self.TagName = tagname
def RefreshView(self):
self.PouNames = self.Controler.GetProjectPouNames()
words = self.TagName.split("::")
if self.ElementType == "config":
self.PouIsUsed = False
returnType = None
self.Values = self.Controler.GetConfigurationGlobalVars(words[1])
elif self.ElementType == "resource":
self.PouIsUsed = False
returnType = None
self.Values = self.Controler.GetConfigurationResourceGlobalVars(words[1], words[2])
else:
self.PouIsUsed = self.Controler.PouIsUsed(words[1])
returnType = self.Controler.GetEditedElementInterfaceReturnType(self.TagName)
self.Values = self.Controler.GetEditedElementInterfaceVars(self.TagName)
if returnType and self.ReturnType.IsEnabled():
self.ReturnType.SetStringSelection(returnType)
self.RefreshValues()
self.RefreshButtons()
def OnReturnTypeChanged(self, event):
words = self.TagName.split("::")
self.Controler.SetPouInterfaceReturnType(words[1], self.ReturnType.GetStringSelection())
self.Controler.BufferProject()
self.ParentWindow.RefreshTitle()
self.ParentWindow.RefreshEditMenu()
event.Skip()
def OnClassFilter(self, event):
self.Filter = self.FilterChoiceTransfer[self.ClassFilter.GetStringSelection()]
self.RefreshTypeList()
self.RefreshValues()
self.RefreshButtons()
event.Skip()
def RefreshTypeList(self):
if self.Filter == "All":
self.ClassList = [self.FilterChoiceTransfer[choice] for choice in self.FilterChoices if self.FilterChoiceTransfer[choice] not in ["All","Interface","Variables"]]
elif self.Filter == "Interface":
self.ClassList = ["Input","Output","InOut","External"]
elif self.Filter == "Variables":
self.ClassList = ["Local","Temp"]
else:
self.ClassList = [self.Filter]
def RefreshButtons(self):
if getattr(self, "Table", None):
table_length = len(self.Table.data)
row_class = None
if table_length and self.PouIsUsed:
row = self.VariablesGrid.GetGridCursorRow()
row_class = self.Table.GetValueByName(row, "Class")
self.AddButton.Enable(not self.PouIsUsed or self.Filter not in ["Interface", "Input", "Output", "InOut"])
self.DeleteButton.Enable(table_length > 0 and row_class not in ["Input", "Output", "InOut"])
self.UpButton.Enable(table_length > 0 and self.Filter == "All" and row_class not in ["Input", "Output", "InOut"])
self.DownButton.Enable(table_length > 0 and self.Filter == "All" and row_class not in ["Input", "Output", "InOut"])
def OnAddButton(self, event):
new_row = self.DefaultValue.copy()
if self.Filter in self.DefaultTypes:
new_row["Class"] = self.DefaultTypes[self.Filter]
else:
new_row["Class"] = self.Filter
self.Values.append(new_row)
self.SaveValues()
self.RefreshValues()
self.RefreshButtons()
event.Skip()
def OnDeleteButton(self, event):
row = self.Table.GetRow(self.VariablesGrid.GetGridCursorRow())
self.Values.remove(row)
self.SaveValues()
self.RefreshValues()
self.RefreshButtons()
event.Skip()
def OnUpButton(self, event):
self.MoveValue(self.VariablesGrid.GetGridCursorRow(), -1)
self.RefreshButtons()
event.Skip()
def OnDownButton(self, event):
self.MoveValue(self.VariablesGrid.GetGridCursorRow(), 1)
self.RefreshButtons()
event.Skip()
def OnVariablesGridCellChange(self, event):
row, col = event.GetRow(), event.GetCol()
colname = self.Table.GetColLabelValue(col)
value = self.Table.GetValue(row, col)
if colname == "Name":
if not TestIdentifier(value):
message = wx.MessageDialog(self, "\"%s\" is not a valid identifier!"%value, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
event.Veto()
elif value.upper() in IEC_KEYWORDS:
message = wx.MessageDialog(self, "\"%s\" is a keyword. It can't be used!"%value, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
event.Veto()
elif value.upper() in self.PouNames:
message = wx.MessageDialog(self, "A pou with \"%s\" as name exists!"%value, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
event.Veto()
elif value.upper() in [var["Name"].upper() for var in self.Values if var != self.Table.data[row]]:
message = wx.MessageDialog(self, "A variable with \"%s\" as name exists in this pou!"%value, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
event.Veto()
else:
self.SaveValues(False)
old_value = self.Table.GetOldValue()
if old_value != "":
self.Controler.UpdateEditedElementUsedVariable(self.TagName, old_value, value)
self.Controler.BufferProject()
self.ParentWindow.RefreshEditor(variablepanel = False)
self.ParentWindow.RefreshTitle()
self.ParentWindow.RefreshEditMenu()
event.Skip()
else:
self.SaveValues()
if colname == "Class":
self.Table.ResetView(self.VariablesGrid)
event.Skip()
def OnVariablesGridEditorShown(self, event):
row, col = event.GetRow(), event.GetCol()
if self.Table.GetColLabelValue(col) == "Type":
type_menu = wx.Menu(title='')
base_menu = wx.Menu(title='')
for base_type in self.Controler.GetBaseTypes():
new_id = wx.NewId()
AppendMenu(base_menu, help='', id=new_id, kind=wx.ITEM_NORMAL, text=base_type)
self.Bind(wx.EVT_MENU, self.GetVariableTypeFunction(base_type), id=new_id)
type_menu.AppendMenu(wx.NewId(), "Base Types", base_menu)
datatype_menu = wx.Menu(title='')
for datatype in self.Controler.GetDataTypes(basetypes = False):
new_id = wx.NewId()
AppendMenu(datatype_menu, help='', id=new_id, kind=wx.ITEM_NORMAL, text=datatype)
self.Bind(wx.EVT_MENU, self.GetVariableTypeFunction(datatype), id=new_id)
type_menu.AppendMenu(wx.NewId(), "User Data Types", datatype_menu)
functionblock_menu = wx.Menu(title='')
for functionblock_type in self.Controler.GetFunctionBlockTypes():
new_id = wx.NewId()
AppendMenu(functionblock_menu, help='', id=new_id, kind=wx.ITEM_NORMAL, text=functionblock_type)
self.Bind(wx.EVT_MENU, self.GetVariableTypeFunction(functionblock_type), id=new_id)
type_menu.AppendMenu(wx.NewId(), "Function Block Types", functionblock_menu)
rect = self.VariablesGrid.BlockToDeviceRect((row, col), (row, col))
self.VariablesGrid.PopupMenuXY(type_menu, rect.x + rect.width, rect.y + self.VariablesGrid.GetColLabelSize())
event.Veto()
else:
event.Skip()
def GetVariableTypeFunction(self, base_type):
def VariableTypeFunction(event):
row = self.VariablesGrid.GetGridCursorRow()
self.Table.SetValueByName(row, "Type", base_type)
self.Table.ResetView(self.VariablesGrid)
self.SaveValues(False)
self.ParentWindow.RefreshEditor(variablepanel = False)
self.Controler.BufferProject()
self.ParentWindow.RefreshTitle()
self.ParentWindow.RefreshEditMenu()
event.Skip()
return VariableTypeFunction
def OnVariablesGridCellLeftClick(self, event):
if event.GetCol() == 0:
row = event.GetRow()
var_name = self.Table.GetValueByName(row, "Name")
var_class = self.Table.GetValueByName(row, "Class")
var_type = self.Table.GetValueByName(row, "Type")
data = wx.TextDataObject(str((var_name, var_class, var_type, self.TagName)))
dragSource = wx.DropSource(self.VariablesGrid)
dragSource.SetData(data)
dragSource.DoDragDrop()
event.Skip()
def OnVariablesGridSelectCell(self, event):
wx.CallAfter(self.RefreshButtons)
event.Skip()
def MoveValue(self, value_index, move):
new_index = max(0, min(value_index + move, len(self.Values) - 1))
if new_index != value_index:
self.Values.insert(new_index, self.Values.pop(value_index))
self.RefreshValues()
self.VariablesGrid.SetGridCursor(new_index, self.VariablesGrid.GetGridCursorCol())
else:
self.RefreshValues()
self.SaveValues()
def RefreshValues(self):
if len(self.Table.data) > 0:
self.VariablesGrid.SetGridCursor(0, 1)
data = []
for num, variable in enumerate(self.Values):
if variable["Class"] in self.ClassList:
variable["Number"] = num + 1
data.append(variable)
self.Table.SetData(data)
self.Table.ResetView(self.VariablesGrid)
def SaveValues(self, buffer = True):
words = self.TagName.split("::")
if self.ElementType == "config":
self.Controler.SetConfigurationGlobalVars(words[1], self.Values)
elif self.ElementType == "resource":
self.Controler.SetConfigurationResourceGlobalVars(words[1], words[2], self.Values)
else:
if self.ReturnType.IsEnabled():
self.Controler.SetPouInterfaceReturnType(words[1], self.ReturnType.GetStringSelection())
self.Controler.SetPouInterfaceVars(words[1], self.Values)
if buffer:
self.Controler.BufferProject()
self.ParentWindow.RefreshTitle()
self.ParentWindow.RefreshEditMenu()
#-------------------------------------------------------------------------------
# Exception Handler
#-------------------------------------------------------------------------------
Max_Traceback_List_Size = 20
def Display_Exception_Dialog(e_type,e_value,e_tb):
trcbck_lst = []
for i,line in enumerate(traceback.extract_tb(e_tb)):
trcbck = " " + str(i+1) + ". "
if line[0].find(os.getcwd()) == -1:
trcbck += "file : " + str(line[0]) + ", "
else:
trcbck += "file : " + str(line[0][len(os.getcwd()):]) + ", "
trcbck += "line : " + str(line[1]) + ", " + "function : " + str(line[2])
trcbck_lst.append(trcbck)
# Allow clicking....
cap = wx.Window_GetCapture()
if cap:
cap.ReleaseMouse()
dlg = wx.SingleChoiceDialog(None,
"""
An error happens.
Click on OK for saving an error report.
Please contact LOLITech at:
+33 (0)3 29 52 95 67
bugs_PLCOpenEditor@lolitech.fr
Error:
""" +
str(e_type) + " : " + str(e_value),
"Error",
trcbck_lst)
try:
res = (dlg.ShowModal() == wx.ID_OK)
finally:
dlg.Destroy()
return res
def Display_Error_Dialog(e_value):
message = wx.MessageDialog(None, str(e_value), "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
def get_last_traceback(tb):
while tb.tb_next:
tb = tb.tb_next
return tb
def format_namespace(d, indent=' '):
return '\n'.join(['%s%s: %s' % (indent, k, repr(v)[:10000]) for k, v in d.iteritems()])
ignored_exceptions = [] # a problem with a line in a module is only reported once per session
def AddExceptHook(path, app_version='[No version]'):#, ignored_exceptions=[]):
def handle_exception(e_type, e_value, e_traceback):
traceback.print_exception(e_type, e_value, e_traceback) # this is very helpful when there's an exception in the rest of this func
last_tb = get_last_traceback(e_traceback)
ex = (last_tb.tb_frame.f_code.co_filename, last_tb.tb_frame.f_lineno)
if str(e_value).startswith("!!!"):
Display_Error_Dialog(e_value)
elif ex not in ignored_exceptions:
result = Display_Exception_Dialog(e_type,e_value,e_traceback)
if result:
ignored_exceptions.append(ex)
info = {
'app-title' : wx.GetApp().GetAppName(), # app_title
'app-version' : app_version,
'wx-version' : wx.VERSION_STRING,
'wx-platform' : wx.Platform,
'python-version' : platform.python_version(), #sys.version.split()[0],
'platform' : platform.platform(),
'e-type' : e_type,
'e-value' : e_value,
'date' : time.ctime(),
'cwd' : os.getcwd(),
}
if e_traceback:
info['traceback'] = ''.join(traceback.format_tb(e_traceback)) + '%s: %s' % (e_type, e_value)
last_tb = get_last_traceback(e_traceback)
exception_locals = last_tb.tb_frame.f_locals # the locals at the level of the stack trace where the exception actually occurred
info['locals'] = format_namespace(exception_locals)
if 'self' in exception_locals:
info['self'] = format_namespace(exception_locals['self'].__dict__)
output = open(path+os.sep+"bug_report_"+info['date'].replace(':','-').replace(' ','_')+".txt",'w')
lst = info.keys()
lst.sort()
for a in lst:
output.write(a+":\n"+str(info[a])+"\n\n")
#sys.excepthook = lambda *args: wx.CallAfter(handle_exception, *args)
sys.excepthook = handle_exception
if __name__ == '__main__':
def usage():
print "\nUsage of PLCOpenEditor.py :"
print "\n %s [Filepath]\n"%sys.argv[0]
try:
opts, args = getopt.getopt(sys.argv[1:], "h", ["help"])
except getopt.GetoptError:
# print help information and exit:
usage()
sys.exit(2)
for o, a in opts:
if o in ("-h", "--help"):
usage()
sys.exit()
fileOpen = None
if len(args) > 1:
usage()
sys.exit()
elif len(args) == 1:
fileOpen = args[0]
app = wx.PySimpleApp()
wx.InitAllImageHandlers()
# Install a exception handle for bug reports
AddExceptHook(os.getcwd(),__version__)
frame = PLCOpenEditor(None, fileOpen=fileOpen)
frame.Show()
app.MainLoop()