etisserant@0: #!/usr/bin/env python etisserant@0: # -*- coding: utf-8 -*- etisserant@0: etisserant@0: #This file is part of PLCOpenEditor, a library implementing an IEC 61131-3 editor etisserant@0: #based on the plcopen standard. etisserant@0: # lbessard@58: #Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD etisserant@0: # etisserant@0: #See COPYING file for copyrights details. etisserant@0: # etisserant@0: #This library is free software; you can redistribute it and/or etisserant@5: #modify it under the terms of the GNU General Public etisserant@0: #License as published by the Free Software Foundation; either etisserant@0: #version 2.1 of the License, or (at your option) any later version. etisserant@0: # etisserant@0: #This library is distributed in the hope that it will be useful, etisserant@0: #but WITHOUT ANY WARRANTY; without even the implied warranty of etisserant@0: #MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU lbessard@58: #General Public License for more details. etisserant@0: # etisserant@5: #You should have received a copy of the GNU General Public etisserant@0: #License along with this library; if not, write to the Free Software etisserant@0: #Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA etisserant@0: etisserant@0: import wx etisserant@0: etisserant@0: from plcopen.structures import * lbessard@102: from PLCControler import ITEM_POU lbessard@27: lbessard@27: from Dialogs import * lbessard@27: lbessard@27: SCROLLBAR_UNIT = 10 lbessard@27: WINDOW_BORDER = 10 lbessard@27: SCROLL_ZONE = 10 etisserant@0: lbessard@114: def AppendMenu(parent, help, id, kind, text): lbessard@114: if wx.VERSION >= (2, 6, 0): lbessard@114: parent.Append(help=help, id=id, kind=kind, text=text) lbessard@114: else: lbessard@114: parent.Append(helpString=help, id=id, kind=kind, item=text) lbessard@114: lbessard@157: if wx.Platform == '__WXMSW__': lbessard@157: faces = { 'times': 'Times New Roman', lbessard@157: 'mono' : 'Courier New', lbessard@157: 'helv' : 'Arial', lbessard@157: 'other': 'Comic Sans MS', lbessard@158: 'size' : 20, lbessard@157: } lbessard@157: else: lbessard@157: faces = { 'times': 'Times', lbessard@157: 'mono' : 'Courier', lbessard@157: 'helv' : 'Helvetica', lbessard@157: 'other': 'new century schoolbook', lbessard@158: 'size' : 20, lbessard@157: } lbessard@157: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Graphic elements Viewer base class etisserant@0: #------------------------------------------------------------------------------- etisserant@0: lbessard@138: # ID Constants for alignment menu items lbessard@138: [ID_VIEWERALIGNMENTMENUITEMS0, ID_VIEWERALIGNMENTMENUITEMS1, lbessard@138: ID_VIEWERALIGNMENTMENUITEMS2, ID_VIEWERALIGNMENTMENUITEMS4, lbessard@138: ID_VIEWERALIGNMENTMENUITEMS5, ID_VIEWERALIGNMENTMENUITEMS6, lbessard@138: ] = [wx.NewId() for _init_coll_AlignmentMenu_Items in range(6)] lbessard@138: lbessard@138: # ID Constants for contextual menu items lbessard@138: [ID_VIEWERCONTEXTUALMENUITEMS0, ID_VIEWERCONTEXTUALMENUITEMS1, lbessard@138: ID_VIEWERCONTEXTUALMENUITEMS2, ID_VIEWERCONTEXTUALMENUITEMS3, lbessard@138: ID_VIEWERCONTEXTUALMENUITEMS5, ID_VIEWERCONTEXTUALMENUITEMS6, lbessard@138: ID_VIEWERCONTEXTUALMENUITEMS8, ID_VIEWERCONTEXTUALMENUITEMS9, lbessard@138: ID_VIEWERCONTEXTUALMENUITEMS11, ID_VIEWERCONTEXTUALMENUITEMS12, lbessard@138: ID_VIEWERCONTEXTUALMENUITEMS14, ID_VIEWERCONTEXTUALMENUITEMS16, lbessard@138: ID_VIEWERCONTEXTUALMENUITEMS17, lbessard@138: ] = [wx.NewId() for _init_coll_ContextualMenu_Items in range(13)] etisserant@0: etisserant@0: lbessard@47: class ViewerDropTarget(wx.TextDropTarget): lbessard@47: lbessard@47: def __init__(self, parent): lbessard@47: wx.TextDropTarget.__init__(self) lbessard@90: self.ParentWindow = parent lbessard@47: lbessard@47: def OnDropText(self, x, y, data): lbessard@108: x, y = self.ParentWindow.CalcUnscrolledPosition(x, y) lbessard@145: scaling = self.ParentWindow.Scaling lbessard@47: values = eval(data) lbessard@97: if values[1] == "program": lbessard@97: message = wx.MessageDialog(self.ParentWindow, "Programs can't be used by other POUs!", "Error", wx.OK|wx.ICON_ERROR) lbessard@97: message.ShowModal() lbessard@97: message.Destroy() lbessard@97: elif values[1] in ["function", "functionBlock", "program"]: lbessard@122: name, type = self.ParentWindow.Controler.GetEditedElementType(self.ParentWindow.GetTagName()) lbessard@97: if name == values[0]: lbessard@97: message = wx.MessageDialog(self.ParentWindow, "\"%s\" can't use itself!"%name, "Error", wx.OK|wx.ICON_ERROR) lbessard@97: message.ShowModal() lbessard@97: message.Destroy() lbessard@97: elif type == "function" and values[1] != "function": lbessard@97: message = wx.MessageDialog(self.ParentWindow, "Function Blocks can't be used by Functions!", "Error", wx.OK|wx.ICON_ERROR) lbessard@97: message.ShowModal() lbessard@97: message.Destroy() lbessard@97: elif self.ParentWindow.Controler.PouIsUsedBy(name, values[0]): lbessard@97: message = wx.MessageDialog(self.ParentWindow, "\"%s\" is already used by \"%s\"!"%(name, values[0]), "Error", wx.OK|wx.ICON_ERROR) lbessard@97: message.ShowModal() lbessard@97: message.Destroy() lbessard@97: else: lbessard@106: blockname = values[2] lbessard@106: if values[1] != "function" and blockname == "": lbessard@106: dialog = wx.TextEntryDialog(self.ParentWindow, "Block name", "Please enter a block name", "", wx.OK|wx.CANCEL|wx.CENTRE) lbessard@106: if dialog.ShowModal(): lbessard@106: blockname = dialog.GetValue() lbessard@106: dialog.Destroy() lbessard@198: if blockname.upper() in [name.upper() for name in self.ParentWindow.Controler.GetProjectPouNames()]: lbessard@195: message = wx.MessageDialog(self.ParentWindow, "\"%s\" pou already exists!"%blockname, "Error", wx.OK|wx.ICON_ERROR) lbessard@195: message.ShowModal() lbessard@195: message.Destroy() lbessard@198: elif blockname.upper() in [name.upper() for name in self.ParentWindow.Controler.GetEditedElementVariables(self.ParentWindow.GetTagName())]: lbessard@195: message = wx.MessageDialog(self.ParentWindow, "\"%s\" element for this pou already exists!"%blockname, "Error", wx.OK|wx.ICON_ERROR) lbessard@195: message.ShowModal() lbessard@195: message.Destroy() lbessard@195: else: lbessard@195: id = self.ParentWindow.GetNewId() lbessard@195: block = FBD_Block(self.ParentWindow, values[0], blockname, id) lbessard@195: width, height = block.GetMinSize() lbessard@195: if scaling is not None: lbessard@195: x = round(float(x) / float(scaling[0])) * scaling[0] lbessard@195: y = round(float(y) / float(scaling[1])) * scaling[1] lbessard@195: width = round(float(width) / float(scaling[0]) + 0.5) * scaling[0] lbessard@195: height = round(float(height) / float(scaling[1]) + 0.5) * scaling[1] lbessard@195: block.SetPosition(x, y) lbessard@195: block.SetSize(width, height) lbessard@195: self.ParentWindow.AddBlock(block) lbessard@195: self.ParentWindow.Controler.AddEditedElementBlock(self.ParentWindow.GetTagName(), id, values[0], blockname) lbessard@195: self.ParentWindow.RefreshBlockModel(block) lbessard@195: self.ParentWindow.RefreshBuffer() lbessard@195: self.ParentWindow.RefreshScrollBars() lbessard@195: self.ParentWindow.ParentWindow.RefreshVariablePanel(self.ParentWindow.GetTagName()) lbessard@195: self.ParentWindow.Refresh(False) lbessard@53: elif values[1] != "location": lbessard@121: if values[3] == self.ParentWindow.GetTagName(): lbessard@121: id = self.ParentWindow.GetNewId() lbessard@121: if values[1] == "Output": lbessard@121: var_type = OUTPUT lbessard@121: elif values[1] == "InOut": lbessard@121: var_type = INPUT lbessard@121: else: lbessard@121: var_type = INPUT lbessard@121: variable = FBD_Variable(self.ParentWindow, var_type, values[0], values[2], id) lbessard@145: width, height = variable.GetMinSize() lbessard@145: if scaling is not None: lbessard@145: x = round(float(x) / float(scaling[0])) * scaling[0] lbessard@145: y = round(float(y) / float(scaling[1])) * scaling[1] lbessard@145: width = round(float(width) / float(scaling[0]) + 0.5) * scaling[0] lbessard@145: height = round(float(height) / float(scaling[1]) + 0.5) * scaling[1] lbessard@121: variable.SetPosition(x, y) lbessard@121: variable.SetSize(width, height) lbessard@121: self.ParentWindow.AddBlock(variable) lbessard@145: self.ParentWindow.Controler.AddEditedElementVariable(self.ParentWindow.GetTagName(), id, var_type) lbessard@121: self.ParentWindow.RefreshVariableModel(variable) lbessard@121: self.ParentWindow.RefreshBuffer() lbessard@121: self.ParentWindow.RefreshScrollBars() lbessard@155: self.ParentWindow.Refresh(False) lbessard@121: else: lbessard@121: message = wx.MessageDialog(self.ParentWindow, "Variable don't belong to this POU!", "Error", wx.OK|wx.ICON_ERROR) lbessard@121: message.ShowModal() lbessard@121: message.Destroy() lbessard@47: lbessard@121: if wx.VERSION >= (2, 8, 0): lbessard@121: import wx.aui lbessard@121: lbessard@121: class MDIViewer(wx.aui.AuiMDIChildFrame): lbessard@121: def __init__(self, parent, tagname, window, controler): lbessard@121: wx.aui.AuiMDIChildFrame.__init__(self, parent, -1, title = "") lbessard@121: lbessard@121: sizer = wx.BoxSizer(wx.HORIZONTAL) lbessard@121: lbessard@121: self.Viewer = Viewer(self, tagname, window, controler) lbessard@121: lbessard@121: sizer.AddWindow(self.Viewer, 1, border=0, flag=wx.GROW) lbessard@121: lbessard@121: self.SetSizer(sizer) lbessard@121: lbessard@121: def GetViewer(self): lbessard@121: return self.Viewer lbessard@47: etisserant@0: """ lbessard@64: Class that implements a Viewer based on a wx.ScrolledWindow for drawing and etisserant@0: manipulating graphic elements etisserant@0: """ etisserant@0: etisserant@0: class Viewer(wx.ScrolledWindow): etisserant@0: lbessard@114: if wx.VERSION < (2, 6, 0): lbessard@114: def Bind(self, event, function, id = None): lbessard@114: if id is not None: lbessard@114: event(self, id, function) lbessard@114: else: lbessard@114: event(self, function) lbessard@114: lbessard@138: # Create Alignment Menu items lbessard@138: def _init_coll_AlignmentMenu_Items(self, parent): lbessard@138: # Create menu items lbessard@138: AppendMenu(parent, help='', id=ID_VIEWERALIGNMENTMENUITEMS0, lbessard@138: kind=wx.ITEM_NORMAL, text=u'Left') lbessard@138: AppendMenu(parent, help='', id=ID_VIEWERALIGNMENTMENUITEMS1, lbessard@138: kind=wx.ITEM_NORMAL, text=u'Center') lbessard@138: AppendMenu(parent, help='', id=ID_VIEWERALIGNMENTMENUITEMS2, lbessard@138: kind=wx.ITEM_NORMAL, text=u'Right') lbessard@138: parent.AppendSeparator() lbessard@138: AppendMenu(parent, help='', id=ID_VIEWERALIGNMENTMENUITEMS4, lbessard@138: kind=wx.ITEM_NORMAL, text=u'Top') lbessard@138: AppendMenu(parent, help='', id=ID_VIEWERALIGNMENTMENUITEMS5, lbessard@138: kind=wx.ITEM_NORMAL, text=u'Middle') lbessard@138: AppendMenu(parent, help='', id=ID_VIEWERALIGNMENTMENUITEMS6, lbessard@138: kind=wx.ITEM_NORMAL, text=u'Bottom') lbessard@138: # Link menu event to corresponding called functions lbessard@138: self.Bind(wx.EVT_MENU, self.OnAlignLeftMenu, lbessard@138: id=ID_VIEWERALIGNMENTMENUITEMS0) lbessard@138: self.Bind(wx.EVT_MENU, self.OnAlignCenterMenu, lbessard@138: id=ID_VIEWERALIGNMENTMENUITEMS1) lbessard@138: self.Bind(wx.EVT_MENU, self.OnAlignRightMenu, lbessard@138: id=ID_VIEWERALIGNMENTMENUITEMS2) lbessard@138: self.Bind(wx.EVT_MENU, self.OnAlignTopMenu, lbessard@138: id=ID_VIEWERALIGNMENTMENUITEMS4) lbessard@138: self.Bind(wx.EVT_MENU, self.OnAlignMiddleMenu, lbessard@138: id=ID_VIEWERALIGNMENTMENUITEMS5) lbessard@138: self.Bind(wx.EVT_MENU, self.OnAlignBottomMenu, lbessard@138: id=ID_VIEWERALIGNMENTMENUITEMS6) lbessard@138: etisserant@0: # Create Contextual Menu items etisserant@0: def _init_coll_ContextualMenu_Items(self, parent): etisserant@0: # Create menu items lbessard@138: AppendMenu(parent, help='', id=ID_VIEWERCONTEXTUALMENUITEMS0, etisserant@0: kind=wx.ITEM_RADIO, text=u'No Modifier') lbessard@138: AppendMenu(parent, help='', id=ID_VIEWERCONTEXTUALMENUITEMS1, etisserant@0: kind=wx.ITEM_RADIO, text=u'Negated') lbessard@138: AppendMenu(parent, help='', id=ID_VIEWERCONTEXTUALMENUITEMS2, etisserant@0: kind=wx.ITEM_RADIO, text=u'Rising Edge') lbessard@138: AppendMenu(parent, help='', id=ID_VIEWERCONTEXTUALMENUITEMS3, etisserant@0: kind=wx.ITEM_RADIO, text=u'Falling Edge') etisserant@0: parent.AppendSeparator() lbessard@138: AppendMenu(parent, help='', id=ID_VIEWERCONTEXTUALMENUITEMS5, etisserant@0: kind=wx.ITEM_NORMAL, text=u'Add Wire Segment') lbessard@138: AppendMenu(parent, help='', id=ID_VIEWERCONTEXTUALMENUITEMS6, etisserant@0: kind=wx.ITEM_NORMAL, text=u'Delete Wire Segment') etisserant@0: parent.AppendSeparator() lbessard@138: AppendMenu(parent, help='', id=ID_VIEWERCONTEXTUALMENUITEMS8, etisserant@0: kind=wx.ITEM_NORMAL, text=u'Add Divergence Branch') lbessard@138: AppendMenu(parent, help='', id=ID_VIEWERCONTEXTUALMENUITEMS9, etisserant@0: kind=wx.ITEM_NORMAL, text=u'Delete Divergence Branch') etisserant@0: parent.AppendSeparator() lbessard@138: AppendMenu(parent, help='', id=ID_VIEWERCONTEXTUALMENUITEMS11, lbessard@118: kind=wx.ITEM_NORMAL, text=u'Clear Execution Order') lbessard@138: AppendMenu(parent, help='', id=ID_VIEWERCONTEXTUALMENUITEMS12, lbessard@118: kind=wx.ITEM_NORMAL, text=u'Reset Execution Order') lbessard@118: parent.AppendSeparator() lbessard@138: parent.AppendMenu(ID_VIEWERCONTEXTUALMENUITEMS14, "Alignment", self.AlignmentMenu) lbessard@138: parent.AppendSeparator() lbessard@138: AppendMenu(parent, help='', id=ID_VIEWERCONTEXTUALMENUITEMS16, lbessard@102: kind=wx.ITEM_NORMAL, text=u'Edit Block') lbessard@138: AppendMenu(parent, help='', id=ID_VIEWERCONTEXTUALMENUITEMS17, etisserant@0: kind=wx.ITEM_NORMAL, text=u'Delete') etisserant@0: # Link menu event to corresponding called functions etisserant@0: self.Bind(wx.EVT_MENU, self.OnNoModifierMenu, lbessard@138: id=ID_VIEWERCONTEXTUALMENUITEMS0) etisserant@0: self.Bind(wx.EVT_MENU, self.OnNegatedMenu, lbessard@138: id=ID_VIEWERCONTEXTUALMENUITEMS1) etisserant@0: self.Bind(wx.EVT_MENU, self.OnRisingEdgeMenu, lbessard@138: id=ID_VIEWERCONTEXTUALMENUITEMS2) etisserant@0: self.Bind(wx.EVT_MENU, self.OnFallingEdgeMenu, lbessard@138: id=ID_VIEWERCONTEXTUALMENUITEMS3) etisserant@0: self.Bind(wx.EVT_MENU, self.OnAddSegmentMenu, lbessard@138: id=ID_VIEWERCONTEXTUALMENUITEMS5) etisserant@0: self.Bind(wx.EVT_MENU, self.OnDeleteSegmentMenu, lbessard@138: id=ID_VIEWERCONTEXTUALMENUITEMS6) etisserant@0: self.Bind(wx.EVT_MENU, self.OnAddBranchMenu, lbessard@138: id=ID_VIEWERCONTEXTUALMENUITEMS8) etisserant@0: self.Bind(wx.EVT_MENU, self.OnDeleteBranchMenu, lbessard@138: id=ID_VIEWERCONTEXTUALMENUITEMS9) lbessard@118: self.Bind(wx.EVT_MENU, self.OnClearExecutionOrderMenu, lbessard@138: id=ID_VIEWERCONTEXTUALMENUITEMS11) lbessard@118: self.Bind(wx.EVT_MENU, self.OnResetExecutionOrderMenu, lbessard@138: id=ID_VIEWERCONTEXTUALMENUITEMS12) lbessard@102: self.Bind(wx.EVT_MENU, self.OnEditBlockMenu, lbessard@138: id=ID_VIEWERCONTEXTUALMENUITEMS16) etisserant@0: self.Bind(wx.EVT_MENU, self.OnDeleteMenu, lbessard@138: id=ID_VIEWERCONTEXTUALMENUITEMS17) etisserant@0: etisserant@0: # Create and initialize Contextual Menu etisserant@0: def _init_menus(self): lbessard@138: self.AlignmentMenu = wx.Menu(title='') etisserant@0: self.ContextualMenu = wx.Menu(title='') etisserant@0: lbessard@138: self._init_coll_AlignmentMenu_Items(self.AlignmentMenu) etisserant@0: self._init_coll_ContextualMenu_Items(self.ContextualMenu) etisserant@0: etisserant@0: # Create a new Viewer lbessard@121: def __init__(self, parent, tagname, window, controler): lbessard@80: wx.ScrolledWindow.__init__(self, parent, pos=wx.Point(0, 0), size=wx.Size(0, 0), lbessard@195: style=wx.HSCROLL | wx.VSCROLL) etisserant@0: self._init_menus() etisserant@0: # Adding a rubberband to Viewer etisserant@0: self.rubberBand = RubberBand(drawingSurface=self) lbessard@64: self.SetBackgroundColour(wx.Colour(255,255,255)) etisserant@0: self.ResetView() etisserant@0: self.Scaling = None etisserant@0: self.DrawGrid = True lbessard@145: self.GridBrush = wx.TRANSPARENT_BRUSH lbessard@213: self.PageSize = None lbessard@213: self.PagePen = wx.TRANSPARENT_PEN lbessard@138: self.DrawingWire = False lbessard@114: self.current_id = 0 lbessard@121: self.TagName = tagname etisserant@0: lbessard@42: # Initialize Block, Wire and Comment numbers lbessard@42: self.block_id = self.wire_id = self.comment_id = 0 lbessard@42: etisserant@0: # Initialize Viewer mode to Selection mode etisserant@0: self.Mode = MODE_SELECTION lbessard@3: self.SavedMode = False lbessard@114: self.CurrentLanguage = "FBD" etisserant@0: lbessard@90: self.ParentWindow = window etisserant@0: self.Controler = controler etisserant@0: lbessard@47: self.SetDropTarget(ViewerDropTarget(self)) lbessard@47: lbessard@156: dc = wx.ClientDC(self) lbessard@213: font = wx.Font(faces["size"], wx.SWISS, wx.NORMAL, wx.NORMAL, faceName = faces["mono"]) lbessard@165: dc.SetFont(font) lbessard@158: width, height = dc.GetTextExtent("ABCDEFGHIJKLMNOPQRSTUVWXYZ") lbessard@158: while width > 260: lbessard@157: faces["size"] -= 1 lbessard@213: font = wx.Font(faces["size"], wx.SWISS, wx.NORMAL, wx.NORMAL, faceName = faces["mono"]) lbessard@165: dc.SetFont(font) lbessard@158: width, height = dc.GetTextExtent("ABCDEFGHIJKLMNOPQRSTUVWXYZ") lbessard@165: self.SetFont(font) lbessard@156: etisserant@0: # Link Viewer event to corresponding methods lbessard@64: self.Bind(wx.EVT_PAINT, self.OnPaint) lbessard@64: self.Bind(wx.EVT_LEFT_DOWN, self.OnViewerLeftDown) lbessard@64: self.Bind(wx.EVT_LEFT_UP, self.OnViewerLeftUp) lbessard@64: self.Bind(wx.EVT_LEFT_DCLICK, self.OnViewerLeftDClick) lbessard@145: self.Bind(wx.EVT_RIGHT_DOWN, self.OnViewerRightDown) lbessard@64: self.Bind(wx.EVT_RIGHT_UP, self.OnViewerRightUp) lbessard@140: self.Bind(wx.EVT_LEAVE_WINDOW, self.OnLeaveViewer) lbessard@64: self.Bind(wx.EVT_MOTION, self.OnViewerMotion) lbessard@64: self.Bind(wx.EVT_CHAR, self.OnChar) lbessard@213: #self.Bind(wx.EVT_SCROLLWIN, self.OnMoveWindow) lbessard@144: self.Bind(wx.EVT_SIZE, self.OnMoveWindow) lbessard@144: lbessard@144: def GetScrolledRect(self, rect): lbessard@144: rect.x, rect.y = self.CalcScrolledPosition(rect.x, rect.y) lbessard@144: return rect etisserant@0: lbessard@145: def GetScaling(self): lbessard@145: return self.Scaling lbessard@145: lbessard@121: def SetTagName(self, tagname): lbessard@121: self.TagName = tagname lbessard@121: lbessard@121: def GetTagName(self): lbessard@121: return self.TagName lbessard@121: lbessard@121: def IsViewing(self, tagname): lbessard@121: return self.TagName == tagname lbessard@121: etisserant@0: # Returns a new id etisserant@0: def GetNewId(self): etisserant@0: self.current_id += 1 etisserant@0: return self.current_id etisserant@0: etisserant@0: # Destructor etisserant@0: def __del__(self): etisserant@0: self.ResetView() etisserant@0: lbessard@90: def GetLogicalDC(self, buffered=False): lbessard@90: if buffered: lbessard@213: dc = wx.AutoBufferedPaintDC(self) lbessard@90: else: lbessard@90: dc = wx.ClientDC(self) lbessard@165: dc.SetFont(self.GetFont()) lbessard@114: if wx.VERSION >= (2, 6, 0): lbessard@114: self.DoPrepareDC(dc) lbessard@114: else: lbessard@114: self.PrepareDC(dc) lbessard@27: return dc lbessard@27: etisserant@0: #------------------------------------------------------------------------------- lbessard@42: # Element management functions lbessard@42: #------------------------------------------------------------------------------- lbessard@42: lbessard@42: def AddBlock(self, block): lbessard@42: self.block_id += 1 lbessard@42: self.Blocks[block] = self.block_id lbessard@42: lbessard@42: def AddWire(self, wire): lbessard@42: self.wire_id += 1 lbessard@42: self.Wires[wire] = self.wire_id lbessard@42: lbessard@42: def AddComment(self, comment): lbessard@42: self.comment_id += 1 lbessard@42: self.Comments[comment] = self.comment_id lbessard@42: lbessard@42: def IsBlock(self, block): lbessard@42: return self.Blocks.get(block, False) lbessard@42: lbessard@42: def IsWire(self, wire): lbessard@42: return self.Wires.get(wire, False) lbessard@42: lbessard@42: def IsComment(self, comment): lbessard@42: return self.Comments.get(comment, False) lbessard@42: lbessard@42: def RemoveBlock(self, block): lbessard@42: self.Blocks.pop(block) lbessard@42: lbessard@42: def RemoveWire(self, wire): lbessard@42: self.Wires.pop(wire) lbessard@42: lbessard@42: def RemoveComment(self, comment): lbessard@42: self.Comments.pop(comment) lbessard@42: lbessard@42: def GetElements(self, sort_blocks=False, sort_wires=False, sort_comments=False): lbessard@42: blocks = self.Blocks.keys() lbessard@42: wires = self.Wires.keys() lbessard@42: comments = self.Comments.keys() lbessard@42: if sort_blocks: lbessard@42: blocks.sort(lambda x,y:self.Blocks[x].__cmp__(self.Blocks[y])) lbessard@42: if sort_wires: lbessard@42: wires.sort(lambda x,y:self.Wires[x].__cmp__(self.Wires[y])) lbessard@42: if sort_comments: lbessard@42: comments.sort(lambda x,y:self.Comments[x].__cmp__(self.Comments[y])) lbessard@42: return blocks + wires + comments lbessard@42: lbessard@42: #------------------------------------------------------------------------------- etisserant@0: # Reset functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: etisserant@0: # Resets Viewer lists etisserant@0: def ResetView(self): lbessard@42: self.Blocks = {} lbessard@42: self.Wires = {} lbessard@42: self.Comments = {} etisserant@0: self.SelectedElement = None lbessard@140: self.HighlightedElement = None etisserant@0: lbessard@47: # Remove all elements lbessard@47: def CleanView(self): lbessard@47: for block in self.Blocks.keys(): lbessard@47: block.Clean() lbessard@47: self.ResetView() lbessard@47: etisserant@0: # Changes Viewer mode etisserant@0: def SetMode(self, mode): lbessard@3: if self.Mode != mode or mode == MODE_SELECTION: lbessard@3: self.Mode = mode lbessard@3: self.SavedMode = False lbessard@3: else: lbessard@3: self.SavedMode = True etisserant@0: # Reset selection lbessard@3: if self.Mode != MODE_SELECTION and self.SelectedElement: etisserant@0: self.SelectedElement.SetSelected(False) etisserant@0: self.SelectedElement = None lbessard@56: lbessard@27: # Return current drawing mode lbessard@27: def GetDrawingMode(self): lbessard@90: return self.ParentWindow.GetDrawingMode() lbessard@56: lbessard@56: # Buffer the last model state lbessard@56: def RefreshBuffer(self): lbessard@56: self.Controler.BufferProject() lbessard@90: self.ParentWindow.RefreshTitle() lbessard@90: self.ParentWindow.RefreshEditMenu() lbessard@27: lbessard@145: # Refresh the current scaling lbessard@145: def RefreshScaling(self, refresh=True): lbessard@145: properties = self.Controler.GetProjectProperties() lbessard@145: scaling = properties["scaling"][self.CurrentLanguage] lbessard@145: if scaling != (0, 0): lbessard@145: self.Scaling = scaling lbessard@145: if self.DrawGrid: lbessard@145: bitmap = wx.EmptyBitmap(*scaling) lbessard@145: dc = wx.MemoryDC(bitmap) lbessard@145: dc.SetBackground(wx.Brush(self.GetBackgroundColour())) lbessard@145: dc.Clear() lbessard@145: dc.SetPen(wx.Pen(wx.Colour(180, 180, 180))) lbessard@145: dc.DrawPoint(0, 0) lbessard@145: self.GridBrush = wx.BrushFromBitmap(bitmap) lbessard@145: else: lbessard@145: self.GridBrush = wx.TRANSPARENT_BRUSH lbessard@145: else: lbessard@145: self.Scaling = None lbessard@145: self.GridBrush = wx.TRANSPARENT_BRUSH lbessard@213: page_size = properties["pageSize"] lbessard@213: if page_size != (0, 0): lbessard@213: self.PageSize = map(int, page_size) lbessard@213: self.PagePen = wx.Pen(wx.Colour(180, 180, 180)) lbessard@213: else: lbessard@213: self.PageSize = None lbessard@213: self.PagePen = wx.TRANSPARENT_PEN lbessard@145: if refresh: lbessard@155: self.Refresh(False) lbessard@145: lbessard@145: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Refresh functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: lbessard@56: def ResetBuffer(self): lbessard@56: pass lbessard@56: etisserant@0: # Refresh Viewer elements etisserant@0: def RefreshView(self): etisserant@0: self.current_id = 0 etisserant@0: # Start by reseting Viewer etisserant@0: self.ResetView() etisserant@0: instance = True etisserant@0: # List of ids of already loaded blocks etisserant@0: ids = [] etisserant@0: # Load Blocks until they are all loaded etisserant@0: while instance: lbessard@121: instance = self.Controler.GetEditedElementInstanceInfos(self.TagName, exclude=ids) etisserant@0: if instance: etisserant@0: self.loadInstance(instance, ids) lbessard@42: self.RefreshScrollBars() lbessard@121: lbessard@121: to_delete = [] lbessard@121: for wire in self.Wires: lbessard@121: if not wire.IsConnectedCompatible(): lbessard@121: to_delete.append(wire) lbessard@121: for wire in to_delete: lbessard@121: wire.Delete() lbessard@121: lbessard@155: self.Refresh(False) etisserant@0: lbessard@213: def GetMaxSize(self): lbessard@27: maxx = maxy = 0 lbessard@42: for element in self.GetElements(): lbessard@144: bbox = element.GetBoundingBox() lbessard@144: maxx = max(maxx, bbox.x + bbox.width) lbessard@144: maxy = max(maxy, bbox.y + bbox.height) lbessard@213: return maxx, maxy lbessard@213: lbessard@213: def RefreshScrollBars(self): lbessard@213: xstart, ystart = self.GetViewStart() lbessard@213: window_size = self.GetClientSize() lbessard@213: maxx, maxy = self.GetMaxSize() lbessard@27: maxx = max(maxx + WINDOW_BORDER, xstart * SCROLLBAR_UNIT + window_size[0]) lbessard@27: maxy = max(maxy + WINDOW_BORDER, ystart * SCROLLBAR_UNIT + window_size[1]) lbessard@27: if self.rubberBand.IsShown(): lbessard@27: extent = self.rubberBand.GetCurrentExtent() lbessard@27: maxx = max(maxx, extent.x + extent.width) lbessard@27: maxy = max(maxy, extent.y + extent.height) lbessard@27: self.SetScrollbars(SCROLLBAR_UNIT, SCROLLBAR_UNIT, lbessard@213: round(maxx / SCROLLBAR_UNIT), round(maxy / SCROLLBAR_UNIT), lbessard@213: xstart, ystart, True) lbessard@95: etisserant@0: # Load instance from given informations etisserant@0: def loadInstance(self, instance, ids): etisserant@0: ids.append(instance["id"]) etisserant@0: self.current_id = max(self.current_id, instance["id"]) etisserant@0: if instance["type"] == "input": lbessard@118: variable = FBD_Variable(self, INPUT, instance["name"], instance["value_type"], instance["id"], instance["executionOrder"]) etisserant@0: variable.SetPosition(instance["x"], instance["y"]) etisserant@0: variable.SetSize(instance["width"], instance["height"]) lbessard@42: self.AddBlock(variable) etisserant@0: connectors = variable.GetConnectors() lbessard@64: connectors["output"].SetPosition(wx.Point(*instance["connector"]["position"])) etisserant@0: if instance["connector"]["negated"]: etisserant@0: connectors["output"].SetNegated(True) etisserant@0: if instance["connector"]["edge"]: etisserant@0: connectors["output"].SetEdge(instance["connector"]["edge"]) etisserant@0: elif instance["type"] == "output": lbessard@118: variable = FBD_Variable(self, OUTPUT, instance["name"], instance["value_type"], instance["id"], instance["executionOrder"]) etisserant@0: variable.SetPosition(instance["x"], instance["y"]) etisserant@0: variable.SetSize(instance["width"], instance["height"]) lbessard@42: self.AddBlock(variable) etisserant@0: connectors = variable.GetConnectors() lbessard@64: connectors["input"].SetPosition(wx.Point(*instance["connector"]["position"])) etisserant@0: if instance["connector"]["negated"]: etisserant@0: connectors["input"].SetNegated(True) etisserant@0: if instance["connector"]["edge"]: etisserant@0: connectors["input"].SetEdge(instance["connector"]["edge"]) etisserant@0: self.CreateWires(connectors["input"], instance["connector"]["links"], ids) etisserant@0: elif instance["type"] == "inout": lbessard@118: variable = FBD_Variable(self, INOUT, instance["name"], instance["value_type"], instance["id"], instance["executionOrder"]) etisserant@0: variable.SetPosition(instance["x"], instance["y"]) etisserant@0: variable.SetSize(instance["width"], instance["height"]) lbessard@42: self.AddBlock(variable) etisserant@0: connectors = variable.GetConnectors() lbessard@64: connectors["output"].SetPosition(wx.Point(*instance["connectors"]["output"]["position"])) lbessard@64: connectors["input"].SetPosition(wx.Point(*instance["connectors"]["input"]["position"])) etisserant@0: if instance["connectors"]["output"]["negated"]: etisserant@0: connectors["output"].SetNegated(True) etisserant@0: if instance["connectors"]["output"]["edge"]: etisserant@0: connectors["output"].SetEdge(instance["connectors"]["output"]["edge"]) etisserant@0: if instance["connectors"]["input"]["negated"]: etisserant@0: connectors["input"].SetNegated(True) etisserant@0: if instance["connectors"]["input"]["edge"]: etisserant@0: connectors["input"].SetEdge(instance["connectors"]["input"]["edge"]) etisserant@0: self.CreateWires(connectors["input"], instance["connectors"]["input"]["links"], ids) etisserant@0: elif instance["type"] == "continuation": etisserant@0: connection = FBD_Connector(self, CONTINUATION, instance["name"], instance["id"]) etisserant@0: connection.SetPosition(instance["x"], instance["y"]) etisserant@0: connection.SetSize(instance["width"], instance["height"]) lbessard@42: self.AddBlock(connection) etisserant@0: connector = connection.GetConnector() lbessard@64: connector.SetPosition(wx.Point(*instance["connector"]["position"])) etisserant@0: elif instance["type"] == "connection": lbessard@28: connection = FBD_Connector(self, CONNECTOR, instance["name"], instance["id"]) etisserant@0: connection.SetPosition(instance["x"], instance["y"]) etisserant@0: connection.SetSize(instance["width"], instance["height"]) lbessard@42: self.AddBlock(connection) etisserant@0: connector = connection.GetConnector() lbessard@64: connector.SetPosition(wx.Point(*instance["connector"]["position"])) etisserant@0: self.CreateWires(connector, instance["connector"]["links"], ids) etisserant@0: elif instance["type"] == "comment": etisserant@0: comment = Comment(self, instance["content"], instance["id"]) etisserant@0: comment.SetPosition(instance["x"], instance["y"]) etisserant@0: comment.SetSize(instance["width"], instance["height"]) lbessard@42: self.AddComment(comment) etisserant@0: elif instance["type"] == "leftPowerRail": etisserant@0: leftpowerrail = LD_PowerRail(self, LEFTRAIL, instance["id"], [True for i in range(len(instance["connectors"]))]) etisserant@0: leftpowerrail.SetPosition(instance["x"], instance["y"]) lbessard@80: leftpowerrail.SetSize(instance["width"], instance["height"]) lbessard@42: self.AddBlock(leftpowerrail) etisserant@0: connectors = leftpowerrail.GetConnectors() etisserant@0: for i, connector in enumerate(instance["connectors"]): lbessard@64: connectors[i].SetPosition(wx.Point(*connector["position"])) etisserant@0: elif instance["type"] == "rightPowerRail": etisserant@0: rightpowerrail = LD_PowerRail(self, RIGHTRAIL, instance["id"], [True for i in range(len(instance["connectors"]))]) etisserant@0: rightpowerrail.SetPosition(instance["x"], instance["y"]) lbessard@80: rightpowerrail.SetSize(instance["width"], instance["height"]) lbessard@42: self.AddBlock(rightpowerrail) etisserant@0: connectors = rightpowerrail.GetConnectors() etisserant@0: for i, connector in enumerate(instance["connectors"]): lbessard@64: connectors[i].SetPosition(wx.Point(*connector["position"])) etisserant@0: self.CreateWires(connectors[i], connector["links"], ids) etisserant@0: elif instance["type"] == "contact": etisserant@0: if instance["negated"]: etisserant@0: negated = instance["negated"] etisserant@0: else: etisserant@0: negated = False etisserant@0: if instance["edge"]: etisserant@0: edge = instance["edge"] etisserant@0: else: etisserant@0: edge = "none" etisserant@0: if negated and edge == "none": etisserant@0: contact_type = CONTACT_REVERSE etisserant@0: elif not negated and edge == "rising": etisserant@0: contact_type = CONTACT_RISING etisserant@0: elif not negated and edge == "falling": etisserant@0: contact_type = CONTACT_FALLING etisserant@0: else: etisserant@0: contact_type = CONTACT_NORMAL etisserant@0: contact = LD_Contact(self, contact_type, instance["name"], instance["id"]) etisserant@0: contact.SetPosition(instance["x"], instance["y"]) lbessard@145: contact.SetSize(instance["width"], instance["height"]) lbessard@42: self.AddBlock(contact) etisserant@0: connectors = contact.GetConnectors() lbessard@64: connectors["input"].SetPosition(wx.Point(*instance["connectors"]["input"]["position"])) etisserant@0: self.CreateWires(connectors["input"], instance["connectors"]["input"]["links"], ids) lbessard@64: connectors["output"].SetPosition(wx.Point(*instance["connectors"]["output"]["position"])) etisserant@0: elif instance["type"] == "coil": etisserant@0: if instance["negated"]: etisserant@0: negated = instance["negated"] etisserant@0: else: etisserant@0: negated = False etisserant@0: if instance["storage"]: etisserant@0: storage = instance["storage"] etisserant@0: else: etisserant@0: storage = "none" etisserant@0: if negated and storage == "none": etisserant@0: coil_type = COIL_REVERSE etisserant@0: elif not negated and storage == "set": etisserant@0: coil_type = COIL_SET etisserant@0: elif not negated and storage == "reset": etisserant@0: coil_type = COIL_RESET etisserant@0: else: etisserant@0: coil_type = COIL_NORMAL etisserant@0: coil = LD_Coil(self, coil_type, instance["name"], instance["id"]) etisserant@0: coil.SetPosition(instance["x"], instance["y"]) lbessard@145: coil.SetSize(instance["width"], instance["height"]) lbessard@42: self.AddBlock(coil) etisserant@0: connectors = coil.GetConnectors() lbessard@64: connectors["input"].SetPosition(wx.Point(*instance["connectors"]["input"]["position"])) etisserant@0: self.CreateWires(connectors["input"], instance["connectors"]["input"]["links"], ids) lbessard@64: connectors["output"].SetPosition(wx.Point(*instance["connectors"]["output"]["position"])) etisserant@0: elif instance["type"] == "step": etisserant@0: if instance["initial"]: etisserant@0: initial = instance["initial"] etisserant@0: else: etisserant@0: initial = False etisserant@0: step = SFC_Step(self, instance["name"], initial, instance["id"]) etisserant@0: step.SetPosition(instance["x"], instance["y"]) etisserant@0: step.SetSize(instance["width"], instance["height"]) lbessard@42: self.AddBlock(step) etisserant@0: if "output" in instance["connectors"]: etisserant@0: step.AddOutput() etisserant@0: if "action" in instance["connectors"]: etisserant@0: step.AddAction() etisserant@0: connectors = step.GetConnectors() etisserant@0: if connectors["input"]: lbessard@64: connectors["input"].SetPosition(wx.Point(*instance["connectors"]["input"]["position"])) etisserant@0: self.CreateWires(connectors["input"], instance["connectors"]["input"]["links"], ids) etisserant@0: if connectors["output"]: lbessard@64: connectors["output"].SetPosition(wx.Point(*instance["connectors"]["output"]["position"])) etisserant@0: if connectors["action"]: lbessard@64: connectors["action"].SetPosition(wx.Point(*instance["connectors"]["action"]["position"])) etisserant@0: elif instance["type"] == "transition": lbessard@80: transition = SFC_Transition(self, instance["condition_type"], instance["condition"], instance["priority"], instance["id"]) etisserant@0: transition.SetPosition(instance["x"], instance["y"]) lbessard@145: transition.SetSize(instance["width"], instance["height"]) lbessard@42: self.AddBlock(transition) etisserant@0: connectors = transition.GetConnectors() lbessard@64: connectors["input"].SetPosition(wx.Point(*instance["connectors"]["input"]["position"])) etisserant@0: self.CreateWires(connectors["input"], instance["connectors"]["input"]["links"], ids) lbessard@64: connectors["output"].SetPosition(wx.Point(*instance["connectors"]["output"]["position"])) lbessard@64: if instance["condition_type"] == "connection": lbessard@64: self.CreateWires(connectors["connection"], instance["connectors"]["connection"]["links"], ids) etisserant@0: elif instance["type"] in ["selectionDivergence", "selectionConvergence", "simultaneousDivergence", "simultaneousConvergence"]: etisserant@0: if instance["type"] == "selectionDivergence": etisserant@0: divergence = SFC_Divergence(self, SELECTION_DIVERGENCE, etisserant@0: len(instance["connectors"]["outputs"]), instance["id"]) etisserant@0: elif instance["type"] == "selectionConvergence": etisserant@0: divergence = SFC_Divergence(self, SELECTION_CONVERGENCE, etisserant@0: len(instance["connectors"]["inputs"]), instance["id"]) etisserant@0: elif instance["type"] == "simultaneousDivergence": etisserant@0: divergence = SFC_Divergence(self, SIMULTANEOUS_DIVERGENCE, etisserant@0: len(instance["connectors"]["outputs"]), instance["id"]) etisserant@0: else: etisserant@0: divergence = SFC_Divergence(self, SIMULTANEOUS_CONVERGENCE, etisserant@0: len(instance["connectors"]["inputs"]), instance["id"]) etisserant@0: divergence.SetPosition(instance["x"], instance["y"]) etisserant@0: divergence.SetSize(instance["width"], instance["height"]) lbessard@42: self.AddBlock(divergence) etisserant@0: connectors = divergence.GetConnectors() etisserant@0: for i, input_connector in enumerate(instance["connectors"]["inputs"]): etisserant@0: connector = connectors["inputs"][i] lbessard@64: connector.SetPosition(wx.Point(*input_connector["position"])) etisserant@0: self.CreateWires(connector, input_connector["links"], ids) etisserant@0: for i, output_connector in enumerate(instance["connectors"]["outputs"]): etisserant@0: connector = connectors["outputs"][i] lbessard@64: connector.SetPosition(wx.Point(*output_connector["position"])) etisserant@0: elif instance["type"] == "jump": etisserant@0: jump = SFC_Jump(self, instance["target"], instance["id"]) etisserant@0: jump.SetPosition(instance["x"], instance["y"]) lbessard@145: jump.SetSize(instance["width"], instance["height"]) lbessard@42: self.AddBlock(jump) etisserant@0: connector = jump.GetConnector() lbessard@64: connector.SetPosition(wx.Point(*instance["connector"]["position"])) etisserant@0: self.CreateWires(connector, instance["connector"]["links"], ids) etisserant@0: elif instance["type"] == "actionBlock": etisserant@0: actionBlock = SFC_ActionBlock(self, instance["actions"], instance["id"]) etisserant@0: actionBlock.SetPosition(instance["x"], instance["y"]) etisserant@0: actionBlock.SetSize(instance["width"], instance["height"]) lbessard@42: self.AddBlock(actionBlock) etisserant@0: connector = actionBlock.GetConnector() lbessard@64: connector.SetPosition(wx.Point(*instance["connector"]["position"])) etisserant@0: self.CreateWires(connector, instance["connector"]["links"], ids) etisserant@0: else: lbessard@90: connectors = {"inputs" : [], "outputs" : []} lbessard@90: for input in instance["connectors"]["inputs"]: lbessard@90: if input["negated"]: lbessard@90: connectors["inputs"].append((input["name"], None, "negated")) lbessard@90: elif input["edge"]: lbessard@90: connectors["inputs"].append((input["name"], None, input["edge"])) lbessard@90: else: lbessard@90: connectors["inputs"].append((input["name"], None, "none")) lbessard@90: for output in instance["connectors"]["outputs"]: lbessard@90: if output["negated"]: lbessard@90: connectors["outputs"].append((output["name"], None, "negated")) lbessard@90: elif output["edge"]: lbessard@90: connectors["outputs"].append((output["name"], None, output["edge"])) lbessard@90: else: lbessard@90: connectors["outputs"].append((output["name"], None, "none")) etisserant@0: if instance["name"] != None: lbessard@118: block = FBD_Block(self, instance["type"], instance["name"], instance["id"], len(instance["connectors"]["inputs"]), connectors=connectors, executionOrder=instance["executionOrder"]) lbessard@118: else: lbessard@118: block = FBD_Block(self, instance["type"], "", instance["id"], len(instance["connectors"]["inputs"]), connectors=connectors, executionOrder=instance["executionOrder"]) etisserant@0: block.SetPosition(instance["x"], instance["y"]) etisserant@0: block.SetSize(instance["width"], instance["height"]) lbessard@42: self.AddBlock(block) etisserant@0: connectors = block.GetConnectors() etisserant@0: for i, input_connector in enumerate(instance["connectors"]["inputs"]): lbessard@195: if i < len(connectors["inputs"]): lbessard@195: connector = connectors["inputs"][i] lbessard@195: connector.SetPosition(wx.Point(*input_connector["position"])) lbessard@195: if input_connector["negated"]: lbessard@195: connector.SetNegated(True) lbessard@195: if input_connector["edge"] != "none": lbessard@195: connector.SetEdge(input_connector["edge"]) lbessard@195: self.CreateWires(connector, input_connector["links"], ids) etisserant@0: for i, output_connector in enumerate(instance["connectors"]["outputs"]): lbessard@195: if i < len(connectors["outputs"]): lbessard@195: connector = connectors["outputs"][i] lbessard@195: if output_connector["negated"]: lbessard@195: connector.SetNegated(True) lbessard@195: if output_connector["edge"] != "none": lbessard@195: connector.SetEdge(output_connector["edge"]) lbessard@195: connector.SetPosition(wx.Point(*output_connector["position"])) etisserant@0: etisserant@0: def CreateWires(self, start_connector, links, ids): etisserant@0: for link in links: etisserant@0: refLocalId = link["refLocalId"] etisserant@0: if refLocalId != None: etisserant@0: if refLocalId not in ids: lbessard@121: new_instance = self.Controler.GetEditedElementInstanceInfos(self.TagName, refLocalId) etisserant@0: if new_instance: etisserant@0: self.loadInstance(new_instance, ids) etisserant@0: connected = self.FindElementById(refLocalId) etisserant@0: if connected: etisserant@0: points = link["points"] lbessard@64: end_connector = connected.GetConnector(wx.Point(points[-1][0], points[-1][1]), link["formalParameter"]) etisserant@0: if end_connector: etisserant@0: wire = Wire(self) etisserant@0: wire.SetPoints(points) etisserant@0: start_connector.Connect((wire, 0), False) etisserant@0: end_connector.Connect((wire, -1), False) etisserant@0: wire.ConnectStartPoint(None, start_connector) etisserant@0: wire.ConnectEndPoint(None, end_connector) lbessard@42: self.AddWire(wire) etisserant@0: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Search Element functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: etisserant@0: def FindBlock(self, pos): etisserant@0: for block in self.Blocks: etisserant@0: if block.HitTest(pos) or block.TestHandle(pos) != (0, 0): etisserant@0: return block etisserant@0: return None etisserant@0: etisserant@0: def FindWire(self, pos): etisserant@0: for wire in self.Wires: etisserant@0: if wire.HitTest(pos) or wire.TestHandle(pos) != (0, 0): etisserant@0: return wire etisserant@0: return None etisserant@0: etisserant@0: def FindElement(self, pos, exclude_group = False): etisserant@0: if self.SelectedElement and not (exclude_group and isinstance(self.SelectedElement, Graphic_Group)): etisserant@0: if self.SelectedElement.HitTest(pos) or self.SelectedElement.TestHandle(pos) != (0, 0): etisserant@0: return self.SelectedElement lbessard@42: for element in self.GetElements(): etisserant@0: if element.HitTest(pos) or element.TestHandle(pos) != (0, 0): etisserant@0: return element etisserant@0: return None etisserant@0: lbessard@3: def FindBlockConnector(self, pos, exclude = True): etisserant@0: for block in self.Blocks: lbessard@3: result = block.TestConnector(pos, exclude) etisserant@0: if result: etisserant@0: return result etisserant@0: return None etisserant@0: etisserant@0: def FindElementById(self, id): lbessard@42: for element in self.Blocks: lbessard@42: if element.GetId() == id: lbessard@42: return element lbessard@42: for element in self.Comments: etisserant@0: if element.GetId() == id: etisserant@0: return element etisserant@0: return None etisserant@0: etisserant@0: def SearchElements(self, bbox): etisserant@0: elements = [] lbessard@42: for element in self.GetElements(): lbessard@42: if element.IsInSelection(bbox): etisserant@0: elements.append(element) etisserant@0: return elements etisserant@0: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Popup menu functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: etisserant@0: def PopupBlockMenu(self, connector = None): lbessard@102: if connector is not None and connector.IsCompatible("BOOL"): lbessard@121: type = self.Controler.GetEditedElementType(self.TagName) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS0, True) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS1, True) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS2, type != "function") lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS3, type != "function") lbessard@102: else: lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS0, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS1, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS2, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS3, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS5, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS6, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS8, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS9, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS14, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS16, self.SelectedElement.GetType() in self.Controler.GetProjectPouNames()) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS17, True) etisserant@0: if connector: etisserant@0: if connector.IsNegated(): lbessard@138: self.ContextualMenu.Check(ID_VIEWERCONTEXTUALMENUITEMS1, True) etisserant@0: elif connector.GetEdge() == "rising": lbessard@138: self.ContextualMenu.Check(ID_VIEWERCONTEXTUALMENUITEMS2, True) etisserant@0: elif connector.GetEdge() == "falling": lbessard@138: self.ContextualMenu.Check(ID_VIEWERCONTEXTUALMENUITEMS3, True) lbessard@138: else: lbessard@138: self.ContextualMenu.Check(ID_VIEWERCONTEXTUALMENUITEMS0, True) etisserant@0: self.PopupMenu(self.ContextualMenu) etisserant@0: etisserant@0: def PopupWireMenu(self): lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS0, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS1, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS2, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS3, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS5, True) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS6, True) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS8, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS9, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS14, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS16, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS17, True) etisserant@0: self.PopupMenu(self.ContextualMenu) etisserant@0: etisserant@0: def PopupDivergenceMenu(self, connector): lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS0, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS1, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS2, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS3, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS5, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS6, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS8, True) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS9, connector) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS14, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS16, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS17, True) etisserant@0: self.PopupMenu(self.ContextualMenu) etisserant@0: lbessard@138: def PopupGroupMenu(self): lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS0, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS1, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS2, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS3, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS5, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS6, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS8, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS9, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS14, True) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS16, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS17, True) lbessard@138: self.PopupMenu(self.ContextualMenu) lbessard@138: lbessard@118: def PopupDefaultMenu(self, block = True): lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS0, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS1, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS2, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS3, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS5, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS6, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS8, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS9, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS14, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS16, False) lbessard@138: self.ContextualMenu.Enable(ID_VIEWERCONTEXTUALMENUITEMS17, block) etisserant@0: self.PopupMenu(self.ContextualMenu) etisserant@0: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Menu items functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: lbessard@138: def OnAlignLeftMenu(self, event): lbessard@138: if self.SelectedElement and isinstance(self.SelectedElement, Graphic_Group): lbessard@138: self.SelectedElement.AlignElements(ALIGN_LEFT, None) lbessard@138: self.RefreshBuffer() lbessard@155: self.Refresh(False) lbessard@138: event.Skip() lbessard@138: lbessard@138: def OnAlignCenterMenu(self, event): lbessard@138: if self.SelectedElement and isinstance(self.SelectedElement, Graphic_Group): lbessard@138: self.SelectedElement.AlignElements(ALIGN_CENTER, None) lbessard@138: self.RefreshBuffer() lbessard@155: self.Refresh(False) lbessard@138: event.Skip() lbessard@138: lbessard@138: def OnAlignRightMenu(self, event): lbessard@138: if self.SelectedElement and isinstance(self.SelectedElement, Graphic_Group): lbessard@138: self.SelectedElement.AlignElements(ALIGN_RIGHT, None) lbessard@138: self.RefreshBuffer() lbessard@155: self.Refresh(False) lbessard@138: event.Skip() lbessard@138: lbessard@138: def OnAlignTopMenu(self, event): lbessard@138: if self.SelectedElement and isinstance(self.SelectedElement, Graphic_Group): lbessard@138: self.SelectedElement.AlignElements(None, ALIGN_TOP) lbessard@138: self.RefreshBuffer() lbessard@155: self.Refresh(False) lbessard@138: event.Skip() lbessard@138: lbessard@138: def OnAlignMiddleMenu(self, event): lbessard@138: if self.SelectedElement and isinstance(self.SelectedElement, Graphic_Group): lbessard@138: self.SelectedElement.AlignElements(None, ALIGN_MIDDLE) lbessard@138: self.RefreshBuffer() lbessard@155: self.Refresh(False) lbessard@138: event.Skip() lbessard@138: lbessard@138: def OnAlignBottomMenu(self, event): lbessard@138: if self.SelectedElement and isinstance(self.SelectedElement, Graphic_Group): lbessard@138: self.SelectedElement.AlignElements(None, ALIGN_BOTTOM) lbessard@138: self.RefreshBuffer() lbessard@155: self.Refresh(False) lbessard@138: event.Skip() lbessard@138: etisserant@0: def OnNoModifierMenu(self, event): lbessard@42: if self.SelectedElement and self.IsBlock(self.SelectedElement): etisserant@0: self.SelectedElement.SetConnectorNegated(False) lbessard@206: self.SelectedElement.Refresh() lbessard@80: self.RefreshBuffer() etisserant@0: event.Skip() etisserant@0: etisserant@0: def OnNegatedMenu(self, event): lbessard@42: if self.SelectedElement and self.IsBlock(self.SelectedElement): etisserant@0: self.SelectedElement.SetConnectorNegated(True) lbessard@206: self.SelectedElement.Refresh() lbessard@80: self.RefreshBuffer() etisserant@0: event.Skip() etisserant@0: etisserant@0: def OnRisingEdgeMenu(self, event): lbessard@42: if self.SelectedElement and self.IsBlock(self.SelectedElement): etisserant@0: self.SelectedElement.SetConnectorEdge("rising") lbessard@206: self.SelectedElement.Refresh() lbessard@80: self.RefreshBuffer() etisserant@0: event.Skip() etisserant@0: etisserant@0: def OnFallingEdgeMenu(self, event): lbessard@42: if self.SelectedElement and self.IsBlock(self.SelectedElement): etisserant@0: self.SelectedElement.SetConnectorEdge("falling") lbessard@206: self.SelectedElement.Refresh() lbessard@80: self.RefreshBuffer() etisserant@0: event.Skip() etisserant@0: etisserant@0: def OnAddSegmentMenu(self, event): lbessard@80: if self.SelectedElement and self.IsWire(self.SelectedElement): etisserant@0: self.SelectedElement.AddSegment() lbessard@206: self.SelectedElement.Refresh() etisserant@0: event.Skip() etisserant@0: etisserant@0: def OnDeleteSegmentMenu(self, event): lbessard@80: if self.SelectedElement and self.IsWire(self.SelectedElement): etisserant@0: self.SelectedElement.DeleteSegment() lbessard@206: self.SelectedElement.Refresh() etisserant@0: event.Skip() etisserant@0: etisserant@0: def OnAddBranchMenu(self, event): lbessard@42: if self.SelectedElement and self.IsBlock(self.SelectedElement): etisserant@0: self.AddDivergenceBranch(self.SelectedElement) lbessard@80: self.RefreshBuffer() etisserant@0: event.Skip() etisserant@0: etisserant@0: def OnDeleteBranchMenu(self, event): lbessard@42: if self.SelectedElement and self.IsBlock(self.SelectedElement): etisserant@0: self.RemoveDivergenceBranch(self.SelectedElement) lbessard@80: self.RefreshBuffer() etisserant@0: event.Skip() etisserant@0: lbessard@102: def OnEditBlockMenu(self, event): lbessard@102: if self.SelectedElement: lbessard@138: self.ParentWindow.EditProjectElement(ITEM_POU, "P::%s"%self.SelectedElement.GetType()) lbessard@102: event.Skip() lbessard@102: etisserant@0: def OnDeleteMenu(self, event): etisserant@0: if self.SelectedElement: etisserant@0: self.SelectedElement.Delete() etisserant@0: self.SelectedElement = None lbessard@80: self.RefreshBuffer() lbessard@206: self.Refresh(False) etisserant@0: event.Skip() etisserant@0: lbessard@118: def OnClearExecutionOrderMenu(self, event): lbessard@121: self.Controler.ClearEditedElementExecutionOrder(self.TagName) lbessard@118: self.RefreshBuffer() lbessard@118: self.RefreshView() lbessard@118: lbessard@118: def OnResetExecutionOrderMenu(self, event): lbessard@121: self.Controler.ResetEditedElementExecutionOrder(self.TagName) lbessard@118: self.RefreshBuffer() lbessard@118: self.RefreshView() lbessard@118: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Mouse event functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: etisserant@0: def OnViewerLeftDown(self, event): lbessard@27: if self.Mode == MODE_SELECTION: lbessard@27: dc = self.GetLogicalDC() lbessard@27: pos = event.GetLogicalPosition(dc) lbessard@27: if event.ControlDown() and self.SelectedElement: lbessard@27: element = self.FindElement(pos, True) lbessard@27: if element: lbessard@27: if isinstance(self.SelectedElement, Graphic_Group): lbessard@27: self.SelectedElement.SetSelected(False) lbessard@27: self.SelectedElement.SelectElement(element) lbessard@27: elif self.SelectedElement: lbessard@27: group = Graphic_Group(self) lbessard@27: group.SelectElement(self.SelectedElement) lbessard@27: group.SelectElement(element) lbessard@27: self.SelectedElement = group lbessard@27: elements = self.SelectedElement.GetElements() lbessard@27: if len(elements) == 0: lbessard@27: self.SelectedElement = element lbessard@27: elif len(elements) == 1: lbessard@27: self.SelectedElement = elements[0] lbessard@27: self.SelectedElement.SetSelected(True) lbessard@27: else: lbessard@27: element = self.FindElement(pos) lbessard@145: if element is None or element.TestHandle(pos) == (0, 0): lbessard@145: connector = self.FindBlockConnector(pos) lbessard@145: else: lbessard@145: connector = None lbessard@138: if self.DrawingWire: lbessard@138: self.DrawingWire = False lbessard@174: if self.SelectedElement is not None: lbessard@174: if connector: lbessard@174: event.Dragging = lambda : True lbessard@174: self.SelectedElement.OnMotion(event, self.GetLogicalDC(), self.Scaling) lbessard@174: if self.SelectedElement.EndConnected is not None: lbessard@174: self.SelectedElement.ResetPoints() lbessard@174: self.SelectedElement.GeneratePoints() lbessard@174: self.SelectedElement.RefreshModel() lbessard@174: self.SelectedElement.SetSelected(True) lbessard@174: element = self.SelectedElement lbessard@174: self.RefreshBuffer() lbessard@174: else: lbessard@174: rect = self.SelectedElement.GetRedrawRect() lbessard@174: self.SelectedElement.Delete() lbessard@174: self.SelectedElement = None lbessard@174: element = None lbessard@174: self.RefreshRect(self.GetScrolledRect(rect), False) lbessard@138: elif connector: lbessard@138: self.DrawingWire = True lbessard@216: scaled_pos = GetScaledEventPosition(event, self.GetLogicalDC(), self.Scaling) lbessard@138: if (connector.GetDirection() == EAST): lbessard@216: wire = Wire(self, [wx.Point(pos.x, pos.y), EAST], [wx.Point(scaled_pos.x, scaled_pos.y), WEST]) lbessard@138: else: lbessard@216: wire = Wire(self, [wx.Point(pos.x, pos.y), WEST], [wx.Point(scaled_pos.x, scaled_pos.y), EAST]) lbessard@216: wire.oldPos = scaled_pos lbessard@138: wire.Handle = (HANDLE_POINT, 0) lbessard@216: wire.ProcessDragging(0, 0, False, None) lbessard@138: wire.Handle = (HANDLE_POINT, 1) lbessard@138: self.AddWire(wire) lbessard@138: if self.SelectedElement: lbessard@138: self.SelectedElement.SetSelected(False) lbessard@138: self.SelectedElement = wire lbessard@144: self.SelectedElement.Refresh() lbessard@27: else: lbessard@138: if self.SelectedElement and self.SelectedElement != element: lbessard@138: self.SelectedElement.SetSelected(False) lbessard@138: self.SelectedElement = None lbessard@138: if element: lbessard@138: self.SelectedElement = element lbessard@138: self.SelectedElement.OnLeftDown(event, dc, self.Scaling) lbessard@144: self.SelectedElement.Refresh() lbessard@138: else: lbessard@138: self.rubberBand.Reset() lbessard@138: self.rubberBand.OnLeftDown(event, dc, self.Scaling) lbessard@27: elif self.Mode in [MODE_BLOCK, MODE_VARIABLE, MODE_CONNECTION, MODE_COMMENT, lbessard@27: MODE_CONTACT, MODE_COIL, MODE_POWERRAIL, MODE_INITIALSTEP, lbessard@27: MODE_STEP, MODE_TRANSITION, MODE_DIVERGENCE, MODE_JUMP, MODE_ACTION]: lbessard@27: self.rubberBand.Reset() lbessard@27: self.rubberBand.OnLeftDown(event, self.GetLogicalDC(), self.Scaling) etisserant@0: event.Skip() etisserant@0: etisserant@0: def OnViewerLeftUp(self, event): lbessard@27: if self.rubberBand.IsShown(): lbessard@27: if self.Mode == MODE_SELECTION: lbessard@27: elements = self.SearchElements(self.rubberBand.GetCurrentExtent()) lbessard@27: self.rubberBand.OnLeftUp(event, self.GetLogicalDC(), self.Scaling) lbessard@80: if len(elements) == 1: lbessard@80: self.SelectedElement = elements[0] lbessard@80: self.SelectedElement.SetSelected(True) lbessard@80: elif len(elements) > 1: lbessard@27: self.SelectedElement = Graphic_Group(self) lbessard@27: self.SelectedElement.SetElements(elements) lbessard@27: self.SelectedElement.SetSelected(True) lbessard@27: else: lbessard@27: bbox = self.rubberBand.GetCurrentExtent() lbessard@27: self.rubberBand.OnLeftUp(event, self.GetLogicalDC(), self.Scaling) lbessard@27: if self.Mode == MODE_BLOCK: lbessard@64: wx.CallAfter(self.AddNewBlock, bbox) lbessard@27: elif self.Mode == MODE_VARIABLE: lbessard@64: wx.CallAfter(self.AddNewVariable, bbox) lbessard@27: elif self.Mode == MODE_CONNECTION: lbessard@64: wx.CallAfter(self.AddNewConnection, bbox) lbessard@27: elif self.Mode == MODE_COMMENT: lbessard@64: wx.CallAfter(self.AddNewComment, bbox) lbessard@27: elif self.Mode == MODE_CONTACT: lbessard@64: wx.CallAfter(self.AddNewContact, bbox) lbessard@27: elif self.Mode == MODE_COIL: lbessard@67: wx.CallAfter(self.AddNewCoil, bbox) lbessard@27: elif self.Mode == MODE_POWERRAIL: lbessard@64: wx.CallAfter(self.AddNewPowerRail, bbox) lbessard@27: elif self.Mode == MODE_INITIALSTEP: lbessard@71: wx.CallAfter(self.AddNewStep, bbox, True) lbessard@27: elif self.Mode == MODE_STEP: lbessard@71: wx.CallAfter(self.AddNewStep, bbox, False) lbessard@27: elif self.Mode == MODE_TRANSITION: lbessard@64: wx.CallAfter(self.AddNewTransition, bbox) lbessard@27: elif self.Mode == MODE_DIVERGENCE: lbessard@64: wx.CallAfter(self.AddNewDivergence, bbox) lbessard@27: elif self.Mode == MODE_JUMP: lbessard@64: wx.CallAfter(self.AddNewJump, bbox) lbessard@27: elif self.Mode == MODE_ACTION: lbessard@64: wx.CallAfter(self.AddNewActionBlock, bbox) lbessard@27: elif self.Mode == MODE_SELECTION and self.SelectedElement: lbessard@138: dc = self.GetLogicalDC() lbessard@138: if self.DrawingWire: lbessard@138: pos = event.GetLogicalPosition(dc) lbessard@138: connector = self.FindBlockConnector(pos) lbessard@138: if self.SelectedElement.EndConnected is not None: lbessard@138: self.DrawingWire = False lbessard@145: self.SelectedElement.StartConnected.HighlightParentBlock(False) lbessard@145: self.SelectedElement.EndConnected.HighlightParentBlock(False) lbessard@138: self.SelectedElement.ResetPoints() lbessard@138: self.SelectedElement.OnMotion(event, dc, self.Scaling) lbessard@138: self.SelectedElement.GeneratePoints() lbessard@138: self.SelectedElement.RefreshModel() lbessard@145: if self.HighlightedElement is not None: lbessard@145: self.HighlightedElement.SetHighlighted(False) lbessard@145: self.HighlightedElement = None lbessard@145: self.SelectedElement.SetHighlighted(True) lbessard@145: self.HighlightedElement = self.SelectedElement lbessard@138: self.SelectedElement.SetSelected(True) lbessard@138: self.RefreshBuffer() lbessard@144: elif connector is None or self.SelectedElement.GetDragging(): lbessard@138: self.DrawingWire = False lbessard@144: rect = self.SelectedElement.GetRedrawRect() lbessard@138: self.SelectedElement.Delete() lbessard@138: self.SelectedElement = None lbessard@155: self.RefreshRect(self.GetScrolledRect(rect), False) lbessard@138: else: lbessard@138: self.SelectedElement.OnLeftUp(event, dc, self.Scaling) lbessard@138: wx.CallAfter(self.SetCursor, wx.NullCursor) lbessard@106: if self.Mode != MODE_SELECTION and not self.SavedMode: lbessard@90: wx.CallAfter(self.ParentWindow.ResetCurrentMode) etisserant@0: event.Skip() etisserant@0: lbessard@145: def OnViewerRightDown(self, event): lbessard@145: if self.Mode == MODE_SELECTION: lbessard@145: dc = self.GetLogicalDC() lbessard@145: pos = event.GetLogicalPosition(dc) lbessard@145: element = self.FindElement(pos) lbessard@145: if self.SelectedElement and self.SelectedElement != element: lbessard@145: self.SelectedElement.SetSelected(False) lbessard@145: self.SelectedElement = None lbessard@145: if element: lbessard@145: self.SelectedElement = element lbessard@145: self.SelectedElement.OnRightDown(event, dc, self.Scaling) lbessard@145: self.SelectedElement.Refresh() lbessard@145: event.Skip() lbessard@145: etisserant@0: def OnViewerRightUp(self, event): lbessard@80: dc = self.GetLogicalDC() lbessard@145: if self.SelectedElement: lbessard@145: self.SelectedElement.OnRightUp(event, dc, self.Scaling) lbessard@64: wx.CallAfter(self.SetCursor, wx.NullCursor) lbessard@118: else: lbessard@118: self.PopupDefaultMenu(False) etisserant@0: event.Skip() etisserant@0: etisserant@0: def OnViewerLeftDClick(self, event): lbessard@27: if self.Mode == MODE_SELECTION and self.SelectedElement: lbessard@102: if event.ControlDown() and self.IsBlock(self.SelectedElement) and self.SelectedElement.GetType() in self.Controler.GetProjectPouNames(): lbessard@102: self.ParentWindow.EditProjectElement(ITEM_POU, self.SelectedElement.GetType()) lbessard@102: else: lbessard@102: self.SelectedElement.OnLeftDClick(event, self.GetLogicalDC(), self.Scaling) etisserant@0: event.Skip() etisserant@0: etisserant@0: def OnViewerMotion(self, event): lbessard@140: refresh = False lbessard@140: dc = self.GetLogicalDC() lbessard@140: pos = GetScaledEventPosition(event, dc, self.Scaling) lbessard@140: if not event.Dragging(): lbessard@140: highlighted = self.FindElement(pos) lbessard@140: if self.HighlightedElement is not None and self.HighlightedElement != highlighted: lbessard@140: self.HighlightedElement.SetHighlighted(False) lbessard@140: self.HighlightedElement = None lbessard@144: if highlighted is not None and self.HighlightedElement != highlighted: lbessard@140: highlighted.SetHighlighted(True) lbessard@140: self.HighlightedElement = highlighted lbessard@27: if self.rubberBand.IsShown(): lbessard@140: self.rubberBand.OnMotion(event, dc, self.Scaling) lbessard@27: elif self.Mode == MODE_SELECTION and self.SelectedElement: lbessard@138: if self.DrawingWire: lbessard@138: connector = self.FindBlockConnector(pos, False) lbessard@138: if not connector or self.SelectedElement.EndConnected == None: lbessard@138: self.SelectedElement.ResetPoints() lbessard@144: movex, movey = self.SelectedElement.OnMotion(event, dc, self.Scaling) lbessard@138: self.SelectedElement.GeneratePoints() lbessard@144: if movex != 0 or movey != 0: lbessard@155: self.RefreshRect(self.GetScrolledRect(self.SelectedElement.GetRedrawRect(movex, movey)), False) lbessard@144: else: lbessard@144: movex, movey = self.SelectedElement.OnMotion(event, dc, self.Scaling) lbessard@144: if movex != 0 or movey != 0: lbessard@155: self.RefreshRect(self.GetScrolledRect(self.SelectedElement.GetRedrawRect(movex, movey)), False) lbessard@95: self.UpdateScrollPos(event) lbessard@140: event.Skip() lbessard@140: lbessard@140: def OnLeaveViewer(self, event): lbessard@144: if self.SelectedElement and self.SelectedElement.GetDragging(): lbessard@144: event.Skip() lbessard@144: elif self.HighlightedElement is not None: lbessard@140: self.HighlightedElement.SetHighlighted(False) lbessard@140: self.HighlightedElement = None lbessard@71: event.Skip() lbessard@71: lbessard@71: def UpdateScrollPos(self, event): lbessard@27: if (event.Dragging() and self.SelectedElement) or self.rubberBand.IsShown(): lbessard@27: position = event.GetPosition() lbessard@64: move_window = wx.Point() lbessard@27: window_size = self.GetClientSize() lbessard@27: xstart, ystart = self.GetViewStart() lbessard@27: if position.x < SCROLL_ZONE and xstart > 0: lbessard@27: move_window.x = -1 lbessard@27: elif position.x > window_size[0] - SCROLL_ZONE: lbessard@27: move_window.x = 1 lbessard@27: if position.y < SCROLL_ZONE and ystart > 0: lbessard@27: move_window.y = -1 lbessard@27: elif position.y > window_size[1] - SCROLL_ZONE: lbessard@27: move_window.y = 1 lbessard@27: if move_window.x != 0 or move_window.y != 0: lbessard@27: self.Scroll(xstart + move_window.x, ystart + move_window.y) lbessard@42: self.RefreshScrollBars() etisserant@0: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Keyboard event functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: etisserant@0: def OnChar(self, event): lbessard@64: xpos, ypos = self.GetScrollPos(wx.HORIZONTAL), self.GetScrollPos(wx.VERTICAL) lbessard@64: xmax = self.GetScrollRange(wx.HORIZONTAL) - self.GetScrollThumb(wx.HORIZONTAL) lbessard@64: ymax = self.GetScrollRange(wx.VERTICAL) - self.GetScrollThumb(wx.VERTICAL) lbessard@27: keycode = event.GetKeyCode() lbessard@27: if self.Scaling: lbessard@27: scaling = self.Scaling lbessard@27: else: lbessard@27: scaling = (8, 8) lbessard@64: if keycode == wx.WXK_DELETE and self.SelectedElement: lbessard@144: rect = self.SelectedElement.GetRedrawRect(1, 1) lbessard@27: self.SelectedElement.Delete() lbessard@27: self.SelectedElement = None lbessard@56: self.RefreshBuffer() lbessard@56: self.RefreshScrollBars() lbessard@138: self.SetCursor(wx.NullCursor) lbessard@155: self.RefreshRect(self.GetScrolledRect(rect), False) lbessard@138: elif keycode == wx.WXK_RETURN and self.SelectedElement: lbessard@138: self.SelectedElement.OnLeftDClick(event, self.GetLogicalDC(), self.Scaling) lbessard@64: elif keycode == wx.WXK_LEFT: lbessard@42: if event.ControlDown() and event.ShiftDown(): lbessard@42: self.Scroll(0, ypos) lbessard@42: elif event.ControlDown(): lbessard@97: event.Skip() lbessard@42: elif self.SelectedElement: lbessard@42: self.SelectedElement.Move(-scaling[0], 0) lbessard@67: self.SelectedElement.RefreshModel() lbessard@56: self.RefreshBuffer() lbessard@56: self.RefreshScrollBars() lbessard@155: self.RefreshRect(self.GetScrolledRect(self.SelectedElement.GetRedrawRect(-scaling[0], 0)), False) lbessard@64: elif keycode == wx.WXK_RIGHT: lbessard@42: if event.ControlDown() and event.ShiftDown(): lbessard@42: self.Scroll(xmax, ypos) lbessard@42: elif event.ControlDown(): lbessard@97: event.Skip() lbessard@42: elif self.SelectedElement: lbessard@42: self.SelectedElement.Move(scaling[0], 0) lbessard@67: self.SelectedElement.RefreshModel() lbessard@56: self.RefreshBuffer() lbessard@56: self.RefreshScrollBars() lbessard@155: self.RefreshRect(self.GetScrolledRect(self.SelectedElement.GetRedrawRect(scaling[0], 0)), False) lbessard@64: elif keycode == wx.WXK_UP: lbessard@42: if event.ControlDown() and event.ShiftDown(): lbessard@42: self.Scroll(xpos, 0) lbessard@42: elif event.ControlDown(): lbessard@97: event.Skip() lbessard@42: elif self.SelectedElement: lbessard@42: self.SelectedElement.Move(0, -scaling[1]) lbessard@67: self.SelectedElement.RefreshModel() lbessard@56: self.RefreshBuffer() lbessard@56: self.RefreshScrollBars() lbessard@155: self.RefreshRect(self.GetScrolledRect(self.SelectedElement.GetRedrawRect(0, -scaling[1])), False) lbessard@64: elif keycode == wx.WXK_DOWN: lbessard@42: if event.ControlDown() and event.ShiftDown(): lbessard@42: self.Scroll(xpos, ymax) lbessard@42: elif event.ControlDown(): lbessard@97: event.Skip() lbessard@42: elif self.SelectedElement: lbessard@42: self.SelectedElement.Move(0, scaling[1]) lbessard@67: self.SelectedElement.RefreshModel() lbessard@56: self.RefreshBuffer() lbessard@56: self.RefreshScrollBars() lbessard@155: self.RefreshRect(self.GetScrolledRect(self.SelectedElement.GetRedrawRect(0, scaling[1])), False) lbessard@114: elif keycode == wx.WXK_SPACE and self.SelectedElement is not None and self.SelectedElement.Dragging: lbessard@216: if self.IsBlock(self.SelectedElement) or self.IsComment(self.SelectedElement): lbessard@216: self.CopyBlock(self.SelectedElement, wx.Point(*self.SelectedElement.GetPosition())) lbessard@216: self.RefreshBuffer() lbessard@216: self.RefreshScrollBars() lbessard@216: self.ParentWindow.RefreshVariablePanel(self.TagName) lbessard@216: self.SelectedElement.Refresh() lbessard@216: else: lbessard@216: event.Skip() lbessard@97: else: lbessard@97: event.Skip() lbessard@27: lbessard@27: #------------------------------------------------------------------------------- lbessard@27: # Model adding functions lbessard@27: #------------------------------------------------------------------------------- lbessard@27: lbessard@145: def GetScaledSize(self, width, height): lbessard@145: if self.Scaling is not None: lbessard@145: width = round(float(width) / float(self.Scaling[0]) + 0.4) * self.Scaling[0] lbessard@145: height = round(float(height) / float(self.Scaling[1]) + 0.4) * self.Scaling[1] lbessard@145: return width, height lbessard@145: lbessard@27: def AddNewBlock(self, bbox): lbessard@90: dialog = BlockPropertiesDialog(self.ParentWindow) lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@121: dialog.SetBlockList(self.Controler.GetBlockTypes(self.TagName)) lbessard@70: dialog.SetPouNames(self.Controler.GetProjectPouNames()) lbessard@121: dialog.SetPouElementNames(self.Controler.GetEditedElementVariables(self.TagName)) lbessard@27: dialog.SetMinBlockSize((bbox.width, bbox.height)) lbessard@64: if dialog.ShowModal() == wx.ID_OK: lbessard@27: id = self.GetNewId() lbessard@27: values = dialog.GetValues() lbessard@27: if "name" in values: lbessard@28: block = FBD_Block(self, values["type"], values["name"], id, values["extension"], values["inputs"]) lbessard@27: else: lbessard@28: block = FBD_Block(self, values["type"], "", id, values["extension"], values["inputs"]) lbessard@27: block.SetPosition(bbox.x, bbox.y) lbessard@145: block.SetSize(*self.GetScaledSize(values["width"], values["height"])) lbessard@42: self.AddBlock(block) lbessard@121: self.Controler.AddEditedElementBlock(self.TagName, id, values["type"], values.get("name", None)) lbessard@27: self.RefreshBlockModel(block) lbessard@56: self.RefreshBuffer() lbessard@42: self.RefreshScrollBars() lbessard@144: self.ParentWindow.RefreshVariablePanel(self.TagName) lbessard@144: block.Refresh() lbessard@27: dialog.Destroy() lbessard@27: lbessard@27: def AddNewVariable(self, bbox): lbessard@90: dialog = VariablePropertiesDialog(self.ParentWindow) lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@27: dialog.SetMinVariableSize((bbox.width, bbox.height)) lbessard@27: varlist = [] lbessard@121: vars = self.Controler.GetEditedElementInterfaceVars(self.TagName) lbessard@27: if vars: lbessard@27: for var in vars: lbessard@70: if var["Edit"]: lbessard@70: varlist.append((var["Name"], var["Class"], var["Type"])) lbessard@121: returntype = self.Controler.GetEditedElementInterfaceReturnType(self.TagName) lbessard@27: if returntype: lbessard@121: varlist.append((self.Controler.GetEditedElementName(self.TagName), "Output", returntype)) lbessard@27: dialog.SetVariables(varlist) lbessard@64: if dialog.ShowModal() == wx.ID_OK: lbessard@27: id = self.GetNewId() lbessard@27: values = dialog.GetValues() lbessard@27: variable = FBD_Variable(self, values["type"], values["name"], values["value_type"], id) lbessard@27: variable.SetPosition(bbox.x, bbox.y) lbessard@145: variable.SetSize(*self.GetScaledSize(values["width"], values["height"])) lbessard@42: self.AddBlock(variable) lbessard@121: self.Controler.AddEditedElementVariable(self.TagName, id, values["type"]) lbessard@27: self.RefreshVariableModel(variable) lbessard@56: self.RefreshBuffer() lbessard@42: self.RefreshScrollBars() lbessard@144: variable.Refresh() lbessard@27: dialog.Destroy() lbessard@27: lbessard@27: def AddNewConnection(self, bbox): lbessard@90: dialog = ConnectionPropertiesDialog(self.ParentWindow) lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@27: dialog.SetMinConnectionSize((bbox.width, bbox.height)) lbessard@64: if dialog.ShowModal() == wx.ID_OK: lbessard@27: id = self.GetNewId() lbessard@27: values = dialog.GetValues() lbessard@27: connection = FBD_Connector(self, values["type"], values["name"], id) lbessard@27: connection.SetPosition(bbox.x, bbox.y) lbessard@145: connection.SetSize(*self.GetScaledSize(values["width"], values["height"])) lbessard@42: self.AddBlock(connection) lbessard@121: self.Controler.AddEditedElementConnection(self.TagName, id, values["type"]) lbessard@27: self.RefreshConnectionModel(connection) lbessard@56: self.RefreshBuffer() lbessard@42: self.RefreshScrollBars() lbessard@144: connection.Refresh() lbessard@27: dialog.Destroy() lbessard@27: lbessard@27: def AddNewComment(self, bbox): lbessard@114: if wx.VERSION >= (2, 5, 0): lbessard@114: dialog = wx.TextEntryDialog(self.ParentWindow, "Edit comment", "Please enter comment text", "", wx.OK|wx.CANCEL|wx.TE_MULTILINE) lbessard@114: else: lbessard@114: dialog = wx.TextEntryDialog(self.ParentWindow, "Edit comment", "Please enter comment text", "", wx.OK|wx.CANCEL) lbessard@64: if dialog.ShowModal() == wx.ID_OK: lbessard@27: value = dialog.GetValue() lbessard@27: id = self.GetNewId() lbessard@27: comment = Comment(self, value, id) lbessard@27: comment.SetPosition(bbox.x, bbox.y) lbessard@27: min_width, min_height = comment.GetMinSize() lbessard@145: comment.SetSize(*self.GetScaledSize(max(min_width,bbox.width),max(min_height,bbox.height))) lbessard@42: self.AddComment(comment) lbessard@121: self.Controler.AddEditedElementComment(self.TagName, id) lbessard@27: self.RefreshCommentModel(comment) lbessard@56: self.RefreshBuffer() lbessard@42: self.RefreshScrollBars() lbessard@144: comment.Refresh() lbessard@27: dialog.Destroy() lbessard@27: lbessard@27: def AddNewContact(self, bbox): lbessard@90: dialog = LDElementDialog(self.ParentWindow, "contact") lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@27: varlist = [] lbessard@121: vars = self.Controler.GetEditedElementInterfaceVars(self.TagName) lbessard@27: if vars: lbessard@27: for var in vars: lbessard@27: if var["Class"] != "Output" and var["Type"] == "BOOL": lbessard@27: varlist.append(var["Name"]) lbessard@27: dialog.SetVariables(varlist) lbessard@27: dialog.SetValues({"name":"","type":CONTACT_NORMAL}) lbessard@28: dialog.SetElementSize((bbox.width, bbox.height)) lbessard@64: if dialog.ShowModal() == wx.ID_OK: lbessard@27: id = self.GetNewId() lbessard@27: values = dialog.GetValues() lbessard@27: contact = LD_Contact(self, values["type"], values["name"], id) lbessard@27: contact.SetPosition(bbox.x, bbox.y) lbessard@145: contact.SetSize(*self.GetScaledSize(values["width"], values["height"])) lbessard@42: self.AddBlock(contact) lbessard@121: self.Controler.AddEditedElementContact(self.TagName, id) lbessard@27: self.RefreshContactModel(contact) lbessard@56: self.RefreshBuffer() lbessard@42: self.RefreshScrollBars() lbessard@144: contact.Refresh() lbessard@27: dialog.Destroy() lbessard@27: lbessard@27: def AddNewCoil(self, bbox): lbessard@90: dialog = LDElementDialog(self.ParentWindow, "coil") lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@27: varlist = [] lbessard@121: vars = self.Controler.GetEditedElementInterfaceVars(self.TagName) lbessard@27: if vars: lbessard@27: for var in vars: lbessard@27: if var["Class"] != "Input" and var["Type"] == "BOOL": lbessard@27: varlist.append(var["Name"]) lbessard@121: returntype = self.Controler.GetEditedElementInterfaceReturnType(self.TagName) lbessard@27: if returntype == "BOOL": lbessard@121: varlist.append(self.Controler.GetEditedElementName(self.TagName)) lbessard@27: dialog.SetVariables(varlist) lbessard@27: dialog.SetValues({"name":"","type":COIL_NORMAL}) lbessard@28: dialog.SetElementSize((bbox.width, bbox.height)) lbessard@64: if dialog.ShowModal() == wx.ID_OK: lbessard@27: id = self.GetNewId() lbessard@27: values = dialog.GetValues() lbessard@27: coil = LD_Coil(self, values["type"], values["name"], id) lbessard@27: coil.SetPosition(bbox.x, bbox.y) lbessard@145: coil.SetSize(*self.GetScaledSize(values["width"], values["height"])) lbessard@42: self.AddBlock(coil) lbessard@121: self.Controler.AddEditedElementCoil(self.TagName, id) lbessard@67: self.RefreshCoilModel(coil) lbessard@56: self.RefreshBuffer() lbessard@42: self.RefreshScrollBars() lbessard@144: coil.Refresh() lbessard@27: dialog.Destroy() lbessard@27: lbessard@27: def AddNewPowerRail(self, bbox): lbessard@90: dialog = LDPowerRailDialog(self.ParentWindow) lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@27: dialog.SetMinSize((bbox.width, bbox.height)) lbessard@64: if dialog.ShowModal() == wx.ID_OK: lbessard@27: id = self.GetNewId() lbessard@27: values = dialog.GetValues() lbessard@27: powerrail = LD_PowerRail(self, values["type"], id, [True for i in xrange(values["number"])]) lbessard@27: powerrail.SetPosition(bbox.x, bbox.y) lbessard@145: powerrail.SetSize(*self.GetScaledSize(values["width"], values["height"])) lbessard@42: self.AddBlock(powerrail) lbessard@121: self.Controler.AddEditedElementPowerRail(self.TagName, id, values["type"]) lbessard@27: self.RefreshPowerRailModel(powerrail) lbessard@56: self.RefreshBuffer() lbessard@42: self.RefreshScrollBars() lbessard@144: powerrail.Refresh() lbessard@27: dialog.Destroy() lbessard@27: lbessard@71: def AddNewStep(self, bbox, initial = False): lbessard@90: dialog = StepContentDialog(self.ParentWindow, initial) lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@71: dialog.SetPouNames(self.Controler.GetProjectPouNames()) lbessard@121: dialog.SetVariables(self.Controler.GetEditedElementInterfaceVars(self.TagName)) lbessard@71: dialog.SetStepNames([block.GetName() for block in self.Blocks if isinstance(block, SFC_Step)]) lbessard@71: dialog.SetMinStepSize((bbox.width, bbox.height)) lbessard@71: if dialog.ShowModal() == wx.ID_OK: lbessard@71: id = self.GetNewId() lbessard@71: values = dialog.GetValues() lbessard@71: step = SFC_Step(self, values["name"], initial, id) lbessard@71: if values["input"]: lbessard@71: step.AddInput() lbessard@71: else: lbessard@71: step.RemoveInput() lbessard@71: if values["output"]: lbessard@71: step.AddOutput() lbessard@71: else: lbessard@71: step.RemoveOutput() lbessard@71: if values["action"]: lbessard@71: step.AddAction() lbessard@71: else: lbessard@71: step.RemoveAction() lbessard@71: step.SetPosition(bbox.x, bbox.y) lbessard@71: min_width, min_height = step.GetMinSize() lbessard@145: step.SetSize(*self.GetScaledSize(max(bbox.width, min_width), max(bbox.height, min_height))) lbessard@71: self.AddBlock(step) lbessard@121: self.Controler.AddEditedElementStep(self.TagName, id) lbessard@71: self.RefreshStepModel(step) lbessard@71: self.RefreshBuffer() lbessard@71: self.RefreshScrollBars() lbessard@144: step.Refresh() lbessard@71: dialog.Destroy() lbessard@71: lbessard@27: def AddNewTransition(self, bbox): lbessard@90: dialog = TransitionContentDialog(self.ParentWindow, self.GetDrawingMode() == FREEDRAWING_MODE) lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@121: dialog.SetTransitions(self.Controler.GetEditedElementTransitions(self.TagName)) lbessard@64: if dialog.ShowModal() == wx.ID_OK: lbessard@27: id = self.GetNewId() lbessard@27: values = dialog.GetValues() lbessard@80: transition = SFC_Transition(self, values["type"], values["value"], values["priority"], id) lbessard@27: transition.SetPosition(bbox.x, bbox.y) lbessard@27: min_width, min_height = transition.GetMinSize() lbessard@145: transition.SetSize(*self.GetScaledSize(max(bbox.width, min_width), max(bbox.height, min_height))) lbessard@42: self.AddBlock(transition) lbessard@121: self.Controler.AddEditedElementTransition(self.TagName, id) lbessard@27: self.RefreshTransitionModel(transition) lbessard@56: self.RefreshBuffer() lbessard@42: self.RefreshScrollBars() lbessard@144: transition.Refresh() lbessard@27: dialog.Destroy() lbessard@27: lbessard@27: def AddNewDivergence(self, bbox): lbessard@90: dialog = DivergenceCreateDialog(self.ParentWindow) lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@27: dialog.SetMinSize((bbox.width, bbox.height)) lbessard@64: if dialog.ShowModal() == wx.ID_OK: lbessard@27: id = self.GetNewId() lbessard@27: values = dialog.GetValues() lbessard@27: divergence = SFC_Divergence(self, values["type"], values["number"], id) lbessard@27: divergence.SetPosition(bbox.x, bbox.y) lbessard@111: min_width, min_height = divergence.GetMinSize(True) lbessard@145: divergence.SetSize(*self.GetScaledSize(max(bbox.width, min_width), max(bbox.height, min_height))) lbessard@42: self.AddBlock(divergence) lbessard@121: self.Controler.AddEditedElementDivergence(self.TagName, id, values["type"]) lbessard@27: self.RefreshDivergenceModel(divergence) lbessard@56: self.RefreshBuffer() lbessard@42: self.RefreshScrollBars() lbessard@144: divergence.Refresh() lbessard@27: dialog.Destroy() etisserant@0: lbessard@108: def AddNewJump(self, bbox): lbessard@108: choices = [] lbessard@108: for block in self.Blocks: lbessard@108: if isinstance(block, SFC_Step): lbessard@108: choices.append(block.GetName()) lbessard@108: dialog = wx.SingleChoiceDialog(self.ParentWindow, "Add a new jump", "Please choose a target", choices, wx.OK|wx.CANCEL) lbessard@108: if dialog.ShowModal() == wx.ID_OK: lbessard@108: id = self.GetNewId() lbessard@108: value = dialog.GetStringSelection() lbessard@108: jump = SFC_Jump(self, value, id) lbessard@108: jump.SetPosition(bbox.x, bbox.y) lbessard@108: min_width, min_height = jump.GetMinSize() lbessard@145: jump.SetSize(*self.GetScaledSize(max(bbox.width, min_width), max(bbox.height, min_height))) lbessard@108: self.AddBlock(jump) lbessard@121: self.Controler.AddEditedElementJump(self.TagName, id) lbessard@108: self.RefreshJumpModel(jump) lbessard@108: self.RefreshBuffer() lbessard@108: self.RefreshScrollBars() lbessard@144: jump.Refresh() lbessard@108: dialog.Destroy() lbessard@108: lbessard@108: def AddNewActionBlock(self, bbox): lbessard@108: dialog = ActionBlockDialog(self.ParentWindow) lbessard@108: dialog.SetQualifierList(self.Controler.GetQualifierTypes()) lbessard@121: dialog.SetActionList(self.Controler.GetEditedElementActions(self.TagName)) lbessard@121: dialog.SetVariableList(self.Controler.GetEditedElementInterfaceVars(self.TagName)) lbessard@108: if dialog.ShowModal() == wx.ID_OK: lbessard@108: actions = dialog.GetValues() lbessard@108: id = self.GetNewId() lbessard@108: actionblock = SFC_ActionBlock(self, actions, id) lbessard@108: actionblock.SetPosition(bbox.x, bbox.y) lbessard@108: min_width, min_height = actionblock.GetMinSize() lbessard@145: actionblock.SetSize(*self.GetScaledSize(max(bbox.width, min_width), max(bbox.height, min_height))) lbessard@108: self.AddBlock(actionblock) lbessard@121: self.Controler.AddEditedElementActionBlock(self.TagName, id) lbessard@108: self.RefreshActionBlockModel(actionblock) lbessard@108: self.RefreshBuffer() lbessard@108: self.RefreshScrollBars() lbessard@144: actionblock.Refresh() lbessard@108: dialog.Destroy() lbessard@27: lbessard@27: #------------------------------------------------------------------------------- lbessard@27: # Edit element content functions lbessard@27: #------------------------------------------------------------------------------- lbessard@27: lbessard@27: def EditBlockContent(self, block): lbessard@90: dialog = BlockPropertiesDialog(self.ParentWindow) lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@121: dialog.SetBlockList(self.Controler.GetBlockTypes(self.TagName)) lbessard@70: dialog.SetPouNames(self.Controler.GetProjectPouNames()) lbessard@121: variable_names = self.Controler.GetEditedElementVariables(self.TagName) lbessard@118: if block.GetName() != "": lbessard@118: variable_names.remove(block.GetName()) lbessard@118: dialog.SetPouElementNames(variable_names) lbessard@27: dialog.SetMinBlockSize(block.GetSize()) lbessard@118: old_values = {"name" : block.GetName(), "type" : block.GetType(), "inputs" : block.GetInputTypes(), lbessard@118: "executionOrder" : block.GetExecutionOrder(), "extension" : block.GetExtension()} lbessard@118: dialog.SetValues(old_values) lbessard@118: if dialog.ShowModal() == wx.ID_OK: lbessard@118: new_values = dialog.GetValues() lbessard@154: rect = block.GetRedrawRect(1, 1) lbessard@118: if "name" in new_values: lbessard@118: block.SetName(new_values["name"]) lbessard@154: else: lbessard@154: block.SetName("") lbessard@145: block.SetSize(*self.GetScaledSize(new_values["width"], new_values["height"])) lbessard@118: block.SetType(new_values["type"], new_values["extension"]) lbessard@118: block.SetExecutionOrder(new_values["executionOrder"]) lbessard@154: rect = rect.Union(block.GetRedrawRect()) lbessard@27: self.RefreshBlockModel(block) lbessard@118: if old_values["executionOrder"] != new_values["executionOrder"]: lbessard@118: self.RefreshView() lbessard@56: self.RefreshBuffer() lbessard@42: self.RefreshScrollBars() lbessard@144: self.ParentWindow.RefreshVariablePanel(self.TagName) lbessard@154: block.Refresh(rect) lbessard@27: dialog.Destroy() lbessard@27: lbessard@27: def EditVariableContent(self, variable): lbessard@90: dialog = VariablePropertiesDialog(self.ParentWindow) lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@27: dialog.SetMinVariableSize(variable.GetSize()) lbessard@27: varlist = [] lbessard@121: vars = self.Controler.GetEditedElementInterfaceVars(self.TagName) lbessard@27: if vars: lbessard@27: for var in vars: lbessard@70: if var["Edit"]: lbessard@70: varlist.append((var["Name"], var["Class"], var["Type"])) lbessard@121: returntype = self.Controler.GetEditedElementInterfaceReturnType(self.TagName) lbessard@27: if returntype: lbessard@121: varlist.append((self.Controler.GetEditedElementName(self.TagName), "Output", returntype)) lbessard@27: dialog.SetVariables(varlist) lbessard@118: old_values = {"name" : variable.GetName(), "type" : variable.GetType(), lbessard@118: "executionOrder" : variable.GetExecutionOrder()} lbessard@118: dialog.SetValues(old_values) lbessard@118: if dialog.ShowModal() == wx.ID_OK: lbessard@118: new_values = dialog.GetValues() lbessard@154: rect = variable.GetRedrawRect(1, 1) lbessard@118: variable.SetName(new_values["name"]) lbessard@118: variable.SetType(new_values["type"], new_values["value_type"]) lbessard@145: variable.SetSize(*self.GetScaledSize(new_values["width"], new_values["height"])) lbessard@118: variable.SetExecutionOrder(new_values["executionOrder"]) lbessard@154: rect = rect.Union(variable.GetRedrawRect()) lbessard@118: if old_values["type"] != new_values["type"]: lbessard@27: id = variable.GetId() lbessard@121: self.Controler.RemoveEditedElementInstance(self.TagName, id) lbessard@121: self.Controler.AddEditedElementVariable(self.TagName, id, new_values["type"]) lbessard@27: self.RefreshVariableModel(variable) lbessard@118: if old_values["executionOrder"] != new_values["executionOrder"]: lbessard@118: self.RefreshView() lbessard@56: self.RefreshBuffer() lbessard@42: self.RefreshScrollBars() lbessard@154: variable.Refresh(rect) lbessard@27: dialog.Destroy() lbessard@27: lbessard@28: def EditConnectionContent(self, connection): lbessard@90: dialog = ConnectionPropertiesDialog(self.ParentWindow) lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@28: dialog.SetMinConnectionSize(connection.GetSize()) lbessard@28: values = {"name" : connection.GetName(), "type" : connection.GetType()} lbessard@28: dialog.SetValues(values) lbessard@64: if dialog.ShowModal() == wx.ID_OK: lbessard@28: old_type = connection.GetType() lbessard@28: values = dialog.GetValues() lbessard@154: rect = connection.GetRedrawRect(1, 1) lbessard@28: connection.SetName(values["name"]) lbessard@28: connection.SetType(values["type"]) lbessard@145: connection.SetSize(*self.GetScaledSize(values["width"], values["height"])) lbessard@154: rect = rect.Union(connection.GetRedrawRect()) lbessard@28: if old_type != values["type"]: lbessard@28: id = connection.GetId() lbessard@121: self.Controler.RemoveEditedElementInstance(self.TagName, id) lbessard@121: self.Controler.AddEditedElementConnection(self.TagName, id, values["type"]) lbessard@28: self.RefreshConnectionModel(connection) lbessard@56: self.RefreshBuffer() lbessard@42: self.RefreshScrollBars() lbessard@154: connection.Refresh(rect) lbessard@28: dialog.Destroy() lbessard@28: lbessard@28: def EditContactContent(self, contact): lbessard@90: dialog = LDElementDialog(self.ParentWindow, "contact") lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@28: varlist = [] lbessard@121: vars = self.Controler.GetEditedElementInterfaceVars(self.TagName) lbessard@28: if vars: lbessard@28: for var in vars: lbessard@28: if var["Class"] != "Output" and var["Type"] == "BOOL": lbessard@28: varlist.append(var["Name"]) lbessard@28: dialog.SetVariables(varlist) lbessard@28: values = {"name" : contact.GetName(), "type" : contact.GetType()} lbessard@28: dialog.SetValues(values) lbessard@28: dialog.SetElementSize(contact.GetSize()) lbessard@64: if dialog.ShowModal() == wx.ID_OK: lbessard@28: values = dialog.GetValues() lbessard@154: rect = contact.GetRedrawRect(1, 1) lbessard@28: contact.SetName(values["name"]) lbessard@28: contact.SetType(values["type"]) lbessard@145: contact.SetSize(*self.GetScaledSize(values["width"], values["height"])) lbessard@154: rect = rect.Union(contact.GetRedrawRect()) lbessard@28: self.RefreshContactModel(contact) lbessard@56: self.RefreshBuffer() lbessard@42: self.RefreshScrollBars() lbessard@154: contact.Refresh(rect) lbessard@28: dialog.Destroy() lbessard@28: lbessard@28: def EditCoilContent(self, coil): lbessard@90: dialog = LDElementDialog(self.ParentWindow, "coil") lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@28: varlist = [] lbessard@121: vars = self.Controler.GetEditedElementInterfaceVars(self.TagName) lbessard@28: if vars: lbessard@28: for var in vars: lbessard@28: if var["Class"] != "Input" and var["Type"] == "BOOL": lbessard@28: varlist.append(var["Name"]) lbessard@121: returntype = self.Controler.GetEditedElementInterfaceReturnType(self.TagName) lbessard@28: if returntype == "BOOL": lbessard@121: varlist.append(self.Controler.GetEditedElementName(self.TagName)) lbessard@28: dialog.SetVariables(varlist) lbessard@28: values = {"name" : coil.GetName(), "type" : coil.GetType()} lbessard@28: dialog.SetValues(values) lbessard@61: dialog.SetElementSize(coil.GetSize()) lbessard@64: if dialog.ShowModal() == wx.ID_OK: lbessard@28: values = dialog.GetValues() lbessard@154: rect = coil.GetRedrawRect(1, 1) lbessard@28: coil.SetName(values["name"]) lbessard@28: coil.SetType(values["type"]) lbessard@145: coil.SetSize(*self.GetScaledSize(values["width"], values["height"])) lbessard@154: rect = rect.Union(coil.GetRedrawRect()) lbessard@64: self.RefreshCoilModel(coil) lbessard@56: self.RefreshBuffer() lbessard@42: self.RefreshScrollBars() lbessard@154: coil.Refresh(rect) lbessard@28: dialog.Destroy() lbessard@28: lbessard@28: def EditPowerRailContent(self, powerrail): lbessard@90: dialog = LDPowerRailDialog(self.ParentWindow, powerrail.GetType(), len(powerrail.GetConnectors())) lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@28: dialog.SetMinSize(powerrail.GetSize()) lbessard@64: if dialog.ShowModal() == wx.ID_OK: lbessard@28: old_type = powerrail.GetType() lbessard@28: values = dialog.GetValues() lbessard@154: rect = powerrail.GetRedrawRect(1, 1) lbessard@61: powerrail.SetType(values["type"], [True for i in xrange(values["number"])]) lbessard@145: powerrail.SetSize(*self.GetScaledSize(values["width"], values["height"])) lbessard@154: rect = rect.Union(powerrail.GetRedrawRect()) lbessard@28: if old_type != values["type"]: lbessard@28: id = powerrail.GetId() lbessard@121: self.Controler.RemoveEditedElementInstance(self.TagName, id) lbessard@121: self.Controler.AddEditedElementPowerRail(self.TagName, id, values["type"]) lbessard@28: self.RefreshPowerRailModel(powerrail) lbessard@56: self.RefreshBuffer() lbessard@42: self.RefreshScrollBars() lbessard@154: powerrail.Refresh(rect) lbessard@28: dialog.Destroy() lbessard@64: lbessard@71: def EditStepContent(self, step): lbessard@90: dialog = StepContentDialog(self.ParentWindow, step.GetInitial()) lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@71: dialog.SetPouNames(self.Controler.GetProjectPouNames()) lbessard@121: dialog.SetVariables(self.Controler.GetEditedElementInterfaceVars(self.TagName)) lbessard@71: dialog.SetStepNames([block.GetName() for block in self.Blocks if isinstance(block, SFC_Step) and block.GetName() != step.GetName()]) lbessard@71: dialog.SetMinStepSize(step.GetSize()) lbessard@71: values = {"name" : step.GetName()} lbessard@71: connectors = step.GetConnectors() lbessard@71: values["input"] = connectors["input"] != None lbessard@71: values["output"] = connectors["output"] != None lbessard@71: values["action"] = connectors["action"] != None lbessard@71: dialog.SetValues(values) lbessard@71: if dialog.ShowModal() == wx.ID_OK: lbessard@71: values = dialog.GetValues() lbessard@154: rect = step.GetRedrawRect(1, 1) lbessard@71: step.SetName(values["name"]) lbessard@71: if values["input"]: lbessard@71: step.AddInput() lbessard@71: else: lbessard@71: step.RemoveInput() lbessard@71: if values["output"]: lbessard@71: step.AddOutput() lbessard@71: else: lbessard@71: step.RemoveOutput() lbessard@71: if values["action"]: lbessard@71: step.AddAction() lbessard@71: else: lbessard@71: step.RemoveAction() lbessard@145: step.UpdateSize(*self.GetScaledSize(values["width"], values["height"])) lbessard@154: rect = rect.Union(step.GetRedrawRect()) lbessard@154: self.RefreshStepModel(step) lbessard@154: self.RefreshBuffer() lbessard@154: self.RefreshScrollBars() lbessard@154: step.Refresh(rect) lbessard@71: lbessard@64: def EditTransitionContent(self, transition): lbessard@90: dialog = TransitionContentDialog(self.ParentWindow, self.GetDrawingMode() == FREEDRAWING_MODE) lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@121: dialog.SetTransitions(self.Controler.GetEditedElementTransitions(self.TagName)) lbessard@80: dialog.SetValues({"type":transition.GetType(),"value":transition.GetCondition(), "priority":transition.GetPriority()}) lbessard@64: dialog.SetElementSize(transition.GetSize()) lbessard@64: if dialog.ShowModal() == wx.ID_OK: lbessard@64: values = dialog.GetValues() lbessard@154: rect = transition.GetRedrawRect(1, 1) lbessard@64: transition.SetType(values["type"],values["value"]) lbessard@80: transition.SetPriority(values["priority"]) lbessard@154: rect = rect.Union(transition.GetRedrawRect()) lbessard@154: self.RefreshTransitionModel(transition) lbessard@154: self.RefreshBuffer() lbessard@154: self.RefreshScrollBars() lbessard@154: transition.Refresh(rect) lbessard@64: dialog.Destroy() lbessard@64: lbessard@64: def EditJumpContent(self, jump): lbessard@64: choices = [] lbessard@64: for block in self.Blocks: lbessard@64: if isinstance(block, SFC_Step): lbessard@64: choices.append(block.GetName()) lbessard@90: dialog = wx.SingleChoiceDialog(self.ParentWindow, "Edit jump target", "Please choose a target", choices, wx.OK|wx.CANCEL) lbessard@64: dialog.SetSelection(choices.index(jump.GetTarget())) lbessard@64: if dialog.ShowModal() == wx.ID_OK: lbessard@64: value = dialog.GetStringSelection() lbessard@154: rect = jump.GetRedrawRect(1, 1) lbessard@64: jump.SetTarget(value) lbessard@154: rect = rect.Union(jump.GetRedrawRect()) lbessard@154: self.RefreshJumpModel(jump) lbessard@154: self.RefreshBuffer() lbessard@154: self.RefreshScrollBars() lbessard@154: jump.Refresh(rect) lbessard@64: dialog.Destroy() lbessard@64: lbessard@64: def EditActionBlockContent(self, actionblock): lbessard@90: dialog = ActionBlockDialog(self.ParentWindow) lbessard@64: dialog.SetQualifierList(self.Controler.GetQualifierTypes()) lbessard@121: dialog.SetActionList(self.Controler.GetEditedElementActions(self.TagName)) lbessard@121: dialog.SetVariableList(self.Controler.GetEditedElementInterfaceVars(self.TagName)) lbessard@64: dialog.SetValues(actionblock.GetActions()) lbessard@64: if dialog.ShowModal() == wx.ID_OK: lbessard@64: actions = dialog.GetValues() lbessard@154: rect = actionblock.GetRedrawRect(1, 1) lbessard@64: actionblock.SetActions(actions) lbessard@145: actionblock.SetSize(*self.GetScaledSize(*actionblock.GetSize())) lbessard@154: rect = rect.Union(actionblock.GetRedrawRect()) lbessard@154: self.RefreshActionBlockModel(actionblock) lbessard@154: self.RefreshBuffer() lbessard@154: self.RefreshScrollBars() lbessard@154: actionblock.Refresh(rect) lbessard@90: dialog.Destroy() lbessard@90: lbessard@90: def EditCommentContent(self, comment): lbessard@114: if wx.VERSION >= (2, 5, 0): lbessard@114: dialog = wx.TextEntryDialog(self.ParentWindow, "Edit comment", "Please enter comment text", comment.GetContent(), wx.OK|wx.CANCEL|wx.TE_MULTILINE) lbessard@114: else: lbessard@114: dialog = wx.TextEntryDialog(self.ParentWindow, "Edit comment", "Please enter comment text", comment.GetContent(), wx.OK|wx.CANCEL) lbessard@90: if dialog.ShowModal() == wx.ID_OK: lbessard@90: value = dialog.GetValue() lbessard@154: rect = comment.GetRedrawRect(1, 1) lbessard@90: comment.SetContent(value) lbessard@145: comment.SetSize(*self.GetScaledSize(*comment.GetSize())) lbessard@154: rect = rect.Union(comment.GetRedrawRect()) lbessard@154: self.RefreshCommentModel(comment) lbessard@154: self.RefreshBuffer() lbessard@154: self.RefreshScrollBars() lbessard@154: comment.Refresh(rect) lbessard@64: dialog.Destroy() lbessard@28: lbessard@28: #------------------------------------------------------------------------------- lbessard@28: # Model update functions lbessard@28: #------------------------------------------------------------------------------- lbessard@28: lbessard@28: def RefreshBlockModel(self, block): lbessard@28: blockid = block.GetId() lbessard@28: infos = {} lbessard@28: infos["type"] = block.GetType() lbessard@28: infos["name"] = block.GetName() lbessard@145: if self.CurrentLanguage == "FBD": lbessard@145: infos["executionOrder"] = block.GetExecutionOrder() lbessard@28: infos["x"], infos["y"] = block.GetPosition() lbessard@28: infos["width"], infos["height"] = block.GetSize() lbessard@28: infos["connectors"] = block.GetConnectors() lbessard@121: self.Controler.SetEditedElementBlockInfos(self.TagName, blockid, infos) lbessard@28: lbessard@28: def RefreshVariableModel(self, variable): lbessard@28: variableid = variable.GetId() lbessard@28: infos = {} lbessard@28: infos["name"] = variable.GetName() lbessard@145: if self.CurrentLanguage == "FBD": lbessard@145: infos["executionOrder"] = variable.GetExecutionOrder() lbessard@28: infos["x"], infos["y"] = variable.GetPosition() lbessard@28: infos["width"], infos["height"] = variable.GetSize() lbessard@28: infos["connectors"] = variable.GetConnectors() lbessard@121: self.Controler.SetEditedElementVariableInfos(self.TagName, variableid, infos) lbessard@28: lbessard@28: def RefreshConnectionModel(self, connection): lbessard@28: connectionid = connection.GetId() lbessard@28: infos = {} lbessard@28: infos["name"] = connection.GetName() lbessard@28: infos["x"], infos["y"] = connection.GetPosition() lbessard@28: infos["width"], infos["height"] = connection.GetSize() lbessard@28: infos["connector"] = connection.GetConnector() lbessard@121: self.Controler.SetEditedElementConnectionInfos(self.TagName, connectionid, infos) lbessard@28: lbessard@28: def RefreshCommentModel(self, comment): lbessard@28: commentid = comment.GetId() lbessard@28: infos = {} lbessard@28: infos["content"] = comment.GetContent() lbessard@28: infos["x"], infos["y"] = comment.GetPosition() lbessard@28: infos["width"], infos["height"] = comment.GetSize() lbessard@121: self.Controler.SetEditedElementCommentInfos(self.TagName, commentid, infos) lbessard@28: lbessard@28: def RefreshPowerRailModel(self, powerrail): lbessard@28: powerrailid = powerrail.GetId() lbessard@28: infos = {} lbessard@28: infos["x"], infos["y"] = powerrail.GetPosition() lbessard@28: infos["width"], infos["height"] = powerrail.GetSize() lbessard@28: infos["connectors"] = powerrail.GetConnectors() lbessard@121: self.Controler.SetEditedElementPowerRailInfos(self.TagName, powerrailid, infos) lbessard@28: lbessard@28: def RefreshContactModel(self, contact): lbessard@28: contactid = contact.GetId() lbessard@28: infos = {} lbessard@28: infos["name"] = contact.GetName() lbessard@28: infos["type"] = contact.GetType() lbessard@28: infos["x"], infos["y"] = contact.GetPosition() lbessard@28: infos["width"], infos["height"] = contact.GetSize() lbessard@28: infos["connectors"] = contact.GetConnectors() lbessard@121: self.Controler.SetEditedElementContactInfos(self.TagName, contactid, infos) lbessard@28: lbessard@28: def RefreshCoilModel(self, coil): lbessard@28: coilid = coil.GetId() lbessard@28: infos = {} lbessard@28: infos["name"] = coil.GetName() lbessard@28: infos["type"] = coil.GetType() lbessard@28: infos["x"], infos["y"] = coil.GetPosition() lbessard@28: infos["width"], infos["height"] = coil.GetSize() lbessard@28: infos["connectors"] = coil.GetConnectors() lbessard@121: self.Controler.SetEditedElementCoilInfos(self.TagName, coilid, infos) lbessard@28: lbessard@28: def RefreshStepModel(self, step): lbessard@28: stepid = step.GetId() lbessard@28: infos = {} lbessard@28: infos["name"] = step.GetName() lbessard@28: infos["initial"] = step.GetInitial() lbessard@28: infos["x"], infos["y"] = step.GetPosition() lbessard@28: infos["width"], infos["height"] = step.GetSize() lbessard@28: infos["connectors"] = step.GetConnectors() lbessard@121: self.Controler.SetEditedElementStepInfos(self.TagName, stepid, infos) lbessard@28: lbessard@28: def RefreshTransitionModel(self, transition): lbessard@28: transitionid = transition.GetId() lbessard@28: infos = {} lbessard@28: infos["type"] = transition.GetType() lbessard@80: infos["priority"] = transition.GetPriority() lbessard@28: infos["condition"] = transition.GetCondition() lbessard@28: infos["x"], infos["y"] = transition.GetPosition() lbessard@28: infos["width"], infos["height"] = transition.GetSize() lbessard@28: infos["connectors"] = transition.GetConnectors() lbessard@121: self.Controler.SetEditedElementTransitionInfos(self.TagName, transitionid, infos) lbessard@28: lbessard@28: def RefreshDivergenceModel(self, divergence): lbessard@28: divergenceid = divergence.GetId() lbessard@28: infos = {} lbessard@28: infos["x"], infos["y"] = divergence.GetPosition() lbessard@28: infos["width"], infos["height"] = divergence.GetSize() lbessard@28: infos["connectors"] = divergence.GetConnectors() lbessard@121: self.Controler.SetEditedElementDivergenceInfos(self.TagName, divergenceid, infos) lbessard@28: lbessard@28: def RefreshJumpModel(self, jump): lbessard@28: jumpid = jump.GetId() lbessard@28: infos = {} lbessard@28: infos["target"] = jump.GetTarget() lbessard@28: infos["x"], infos["y"] = jump.GetPosition() lbessard@28: infos["width"], infos["height"] = jump.GetSize() lbessard@28: infos["connector"] = jump.GetConnector() lbessard@121: self.Controler.SetEditedElementJumpInfos(self.TagName, jumpid, infos) lbessard@28: lbessard@28: def RefreshActionBlockModel(self, actionblock): lbessard@28: actionblockid = actionblock.GetId() lbessard@28: infos = {} lbessard@28: infos["actions"] = actionblock.GetActions() lbessard@28: infos["x"], infos["y"] = actionblock.GetPosition() lbessard@28: infos["width"], infos["height"] = actionblock.GetSize() lbessard@28: infos["connector"] = actionblock.GetConnector() lbessard@121: self.Controler.SetEditedElementActionBlockInfos(self.TagName, actionblockid, infos) lbessard@28: lbessard@28: lbessard@28: #------------------------------------------------------------------------------- lbessard@28: # Model delete functions lbessard@28: #------------------------------------------------------------------------------- lbessard@28: lbessard@28: lbessard@28: def DeleteBlock(self, block): lbessard@28: elements = [] lbessard@28: for output in block.GetConnectors()["outputs"]: lbessard@28: for element in output.GetConnectedBlocks(): lbessard@28: if element not in elements: lbessard@28: elements.append(element) lbessard@28: block.Clean() lbessard@42: self.RemoveBlock(block) lbessard@121: self.Controler.RemoveEditedElementInstance(self.TagName, block.GetId()) lbessard@28: for element in elements: lbessard@28: element.RefreshModel() lbessard@144: wx.CallAfter(self.ParentWindow.RefreshVariablePanel, self.TagName) lbessard@28: lbessard@28: def DeleteVariable(self, variable): lbessard@28: connectors = variable.GetConnectors() lbessard@28: if connectors["output"]: lbessard@28: elements = connectors["output"].GetConnectedBlocks() lbessard@28: else: lbessard@28: elements = [] lbessard@28: variable.Clean() lbessard@42: self.RemoveBlock(variable) lbessard@121: self.Controler.RemoveEditedElementInstance(self.TagName, variable.GetId()) lbessard@28: for element in elements: lbessard@28: element.RefreshModel() lbessard@28: lbessard@28: def DeleteConnection(self, connection): lbessard@28: if connection.GetType() == CONTINUATION: lbessard@28: elements = connection.GetConnector().GetConnectedBlocks() lbessard@28: else: lbessard@28: elements = [] lbessard@28: connection.Clean() lbessard@42: self.RemoveBlock(connection) lbessard@121: self.Controler.RemoveEditedElementInstance(self.TagName, connection.GetId()) lbessard@28: for element in elements: lbessard@28: element.RefreshModel() lbessard@28: lbessard@28: def DeleteComment(self, comment): lbessard@42: self.RemoveComment(comment) lbessard@121: self.Controler.RemoveEditedElementInstance(self.TagName, comment.GetId()) lbessard@28: lbessard@28: def DeleteWire(self, wire): lbessard@28: if wire in self.Wires: lbessard@28: connected = wire.GetConnected() lbessard@28: wire.Clean() lbessard@42: self.RemoveWire(wire) lbessard@28: for connector in connected: lbessard@28: connector.RefreshParentBlock() lbessard@28: lbessard@28: def DeleteContact(self, contact): lbessard@28: connectors = contact.GetConnectors() lbessard@28: elements = connectors["output"].GetConnectedBlocks() lbessard@28: contact.Clean() lbessard@42: self.RemoveBlock(contact) lbessard@121: self.Controler.RemoveEditedElementInstance(self.TagName, contact.GetId()) lbessard@28: for element in elements: lbessard@28: element.RefreshModel() lbessard@28: lbessard@28: def DeleteCoil(self, coil): lbessard@28: connectors = coil.GetConnectors() lbessard@28: elements = connectors["output"].GetConnectedBlocks() lbessard@28: coil.Clean() lbessard@42: self.RemoveBlock(coil) lbessard@121: self.Controler.RemoveEditedElementInstance(self.TagName, coil.GetId()) lbessard@28: for element in elements: lbessard@28: element.RefreshModel() lbessard@28: lbessard@28: def DeletePowerRail(self, powerrail): lbessard@28: elements = [] lbessard@28: if powerrail.GetType() == LEFTRAIL: lbessard@28: for connector in powerrail.GetConnectors(): lbessard@28: for element in connector.GetConnectedBlocks(): lbessard@28: if element not in elements: lbessard@28: elements.append(element) lbessard@108: powerrail.Clean() lbessard@108: self.RemoveBlock(powerrail) lbessard@121: self.Controler.RemoveEditedElementInstance(self.TagName, powerrail.GetId()) lbessard@28: for element in elements: lbessard@28: element.RefreshModel() lbessard@28: lbessard@28: def DeleteStep(self, step): lbessard@28: elements = [] lbessard@28: connectors = step.GetConnectors() lbessard@28: if connectors["output"]: lbessard@28: for element in connectors["output"].GetConnectedBlocks(): lbessard@28: if element not in elements: lbessard@28: elements.append(element) lbessard@28: if connectors["action"]: lbessard@28: for element in connectors["action"].GetConnectedBlocks(): lbessard@28: if element not in elements: lbessard@28: elements.append(element) lbessard@28: step.Clean() lbessard@42: self.RemoveBlock(step) lbessard@121: self.Controler.RemoveEditedElementInstance(self.TagName, step.GetId()) lbessard@28: for element in elements: lbessard@28: element.RefreshModel() lbessard@28: lbessard@28: def DeleteTransition(self, transition): lbessard@28: elements = [] lbessard@28: connectors = transition.GetConnectors() lbessard@28: if connectors["output"]: lbessard@28: for element in connectors["output"].GetConnectedBlocks(): lbessard@28: if element not in elements: lbessard@28: elements.append(element) lbessard@28: transition.Clean() lbessard@42: self.RemoveBlock(transition) lbessard@121: self.Controler.RemoveEditedElementInstance(self.TagName, transition.GetId()) lbessard@28: for element in elements: lbessard@28: element.RefreshModel() lbessard@28: lbessard@28: def DeleteDivergence(self, divergence): lbessard@28: elements = [] lbessard@28: connectors = divergence.GetConnectors() lbessard@28: for output in connectors["outputs"]: lbessard@28: for element in output.GetConnectedBlocks(): lbessard@28: if element not in elements: lbessard@28: elements.append(element) lbessard@28: divergence.Clean() lbessard@42: self.RemoveBlock(divergence) lbessard@121: self.Controler.RemoveEditedElementInstance(self.TagName, divergence.GetId()) lbessard@28: for element in elements: lbessard@28: element.RefreshModel() lbessard@28: lbessard@28: def DeleteJump(self, jump): lbessard@28: jump.Clean() lbessard@42: self.RemoveBlock(jump) lbessard@121: self.Controler.RemoveEditedElementInstance(self.TagName, jump.GetId()) lbessard@28: lbessard@28: def DeleteActionBlock(self, actionblock): lbessard@28: actionblock.Clean() lbessard@42: self.RemoveBlock(actionblock) lbessard@121: self.Controler.RemoveEditedElementInstance(self.TagName, actionblock.GetId()) lbessard@28: lbessard@27: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Editing functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: etisserant@0: def Cut(self): lbessard@178: if self.IsBlock(self.SelectedElement) or self.IsComment(self.SelectedElement): lbessard@162: self.ParentWindow.SetCopyBuffer(self.SelectedElement.Clone(self)) lbessard@144: rect = self.SelectedElement.GetRedrawRect(1, 1) lbessard@114: self.SelectedElement.Delete() lbessard@144: self.SelectedElement = None lbessard@144: self.RefreshBuffer() lbessard@144: self.RefreshScrollBars() lbessard@155: self.RefreshRect(self.GetScrolledRect(rect), False) etisserant@0: etisserant@0: def Copy(self): lbessard@178: if self.IsBlock(self.SelectedElement) or self.IsComment(self.SelectedElement): lbessard@162: self.ParentWindow.SetCopyBuffer(self.SelectedElement.Clone(self)) lbessard@144: etisserant@0: def Paste(self): lbessard@114: element = self.ParentWindow.GetCopyBuffer() lbessard@114: if element is not None and self.CanAddBlock(element): lbessard@114: block = self.CopyBlock(element, wx.Point(*self.CalcUnscrolledPosition(30, 30))) lbessard@114: if self.SelectedElement is not None: lbessard@114: self.SelectedElement.SetSelected(False) lbessard@114: self.SelectedElement = block lbessard@114: self.SelectedElement.SetSelected(True) lbessard@114: self.RefreshBuffer() lbessard@114: self.RefreshScrollBars() lbessard@144: self.ParentWindow.RefreshVariablePanel(self.TagName) lbessard@114: lbessard@114: def CanAddBlock(self, block): lbessard@114: if self.CurrentLanguage == "SFC": lbessard@114: return True lbessard@114: elif self.CurrentLanguage == "LD" and not isinstance(block, (SFC_Step, SFC_Transition, SFC_Divergence, SFC_Jump, SFC_ActionBlock)): lbessard@114: return True lbessard@114: elif self.CurrentLanguage == "FBD" and isinstance(block, (FBD_Block, FBD_Variable, FBD_Connector, Comment)): lbessard@114: return True lbessard@114: return False lbessard@114: lbessard@114: def CopyBlock(self, element, pos): lbessard@114: id = self.GetNewId() lbessard@144: if isinstance(element, FBD_Block) and element.GetName() != "" or isinstance(element, SFC_Step): lbessard@144: if isinstance(element, FBD_Block): lbessard@144: names = [varname.upper() for varname in self.Controler.GetEditedElementVariables(self.TagName)] lbessard@144: format = "Block%d" lbessard@144: elif isinstance(element, SFC_Step): lbessard@144: names = [block.GetName().upper() for block in self.Blocks if isinstance(block, SFC_Step)] lbessard@144: format = "Step%d" lbessard@144: i = 1 lbessard@144: while (format%i).upper() in names: lbessard@144: i += 1 lbessard@144: name = format%i lbessard@162: block = element.Clone(self, id, name, pos) lbessard@144: else: lbessard@144: name = None lbessard@162: block = element.Clone(self, id, pos=pos) lbessard@114: if isinstance(block, Comment): lbessard@178: self.AddComment(block) lbessard@121: self.Controler.AddEditedElementComment(self.TagName, id) lbessard@144: self.RefreshCommentModel(block) lbessard@178: else: lbessard@178: self.AddBlock(block) lbessard@178: if isinstance(block, FBD_Block): lbessard@178: self.Controler.AddEditedElementBlock(self.TagName, id, block.GetType(), name) lbessard@178: self.RefreshBlockModel(block) lbessard@178: elif isinstance(block, FBD_Variable): lbessard@178: self.Controler.AddEditedElementVariable(self.TagName, id, block.GetType()) lbessard@178: self.RefreshVariableModel(block) lbessard@178: elif isinstance(block, FBD_Connector): lbessard@178: self.Controler.AddEditedElementConnection(self.TagName, id, block.GetType()) lbessard@178: self.RefreshConnectionModel(block) lbessard@178: elif isinstance(block, LD_Contact): lbessard@178: self.Controler.AddEditedElementContact(self.TagName, id) lbessard@178: self.RefreshContactModel(block) lbessard@178: elif isinstance(block, LD_Coil): lbessard@178: self.Controler.AddEditedElementCoil(self.TagName, id) lbessard@178: self.RefreshCoilModel(block) lbessard@178: elif isinstance(block, LD_PowerRail): lbessard@178: self.Controler.AddEditedElementPowerRail(self.TagName, id, block.GetType()) lbessard@178: self.RefreshPowerRailModel(block) lbessard@178: elif isinstance(block, SFC_Step): lbessard@178: self.Controler.AddEditedElementStep(self.TagName, id) lbessard@178: self.RefreshStepModel(block) lbessard@178: elif isinstance(block, SFC_Transition): lbessard@178: self.Controler.AddEditedElementTransition(self.TagName, id) lbessard@178: self.RefreshTransitionModel(block) lbessard@178: elif isinstance(block, SFC_Divergence): lbessard@178: self.Controler.AddEditedElementDivergence(self.TagName, id, block.GetType()) lbessard@202: self.RefreshDivergenceModel(block) lbessard@178: elif isinstance(block, SFC_Jump): lbessard@178: self.Controler.AddEditedElementJump(self.TagName, id) lbessard@178: self.RefreshJumpModel(block) lbessard@178: elif isinstance(block, SFC_ActionBlock): lbessard@178: self.Controler.AddEditedElementActionBlock(self.TagName, id) lbessard@178: self.RefreshActionBlockModel(block) lbessard@114: return block lbessard@114: etisserant@0: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Drawing functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: lbessard@27: def OnMoveWindow(self, event): lbessard@80: self.GetBestSize() lbessard@42: self.RefreshScrollBars() lbessard@27: event.Skip() lbessard@27: lbessard@213: def DoDrawing(self, dc, printing = False): lbessard@213: if printing: lbessard@213: if getattr(dc, "printing", False): lbessard@213: font = wx.Font(self.GetFont().GetPointSize(), wx.MODERN, wx.NORMAL, wx.NORMAL) lbessard@213: dc.SetFont(font) lbessard@213: else: lbessard@213: dc.SetFont(self.GetFont()) lbessard@213: else: lbessard@213: dc.SetBackground(wx.Brush(self.GetBackgroundColour())) lbessard@213: dc.Clear() lbessard@213: dc.BeginDrawing() lbessard@213: if self.Scaling is not None and self.DrawGrid and not printing: lbessard@145: dc.SetPen(wx.TRANSPARENT_PEN) lbessard@145: dc.SetBrush(self.GridBrush) lbessard@213: xstart, ystart = self.GetViewStart() lbessard@213: window_size = self.GetClientSize() lbessard@145: width, height = self.GetVirtualSize() lbessard@213: width = max(width, xstart * SCROLLBAR_UNIT + window_size[0]) lbessard@213: height = max(height, ystart * SCROLLBAR_UNIT + window_size[1]) lbessard@145: dc.DrawRectangle(0, 0, width, height) lbessard@213: if self.PageSize is not None and not printing: lbessard@213: dc.SetPen(self.PagePen) lbessard@213: xstart, ystart = self.GetViewStart() lbessard@213: window_size = self.GetClientSize() lbessard@213: for x in xrange(self.PageSize[0] - (xstart * SCROLLBAR_UNIT) % self.PageSize[0], window_size[0], self.PageSize[0]): lbessard@213: dc.DrawLine(xstart * SCROLLBAR_UNIT + x + 1, ystart * SCROLLBAR_UNIT, lbessard@213: xstart * SCROLLBAR_UNIT + x + 1, ystart * SCROLLBAR_UNIT + window_size[1]) lbessard@213: for y in xrange(self.PageSize[1] - (ystart * SCROLLBAR_UNIT) % self.PageSize[1], window_size[1], self.PageSize[1]): lbessard@213: dc.DrawLine(xstart * SCROLLBAR_UNIT, ystart * SCROLLBAR_UNIT + y + 1, lbessard@213: xstart * SCROLLBAR_UNIT + window_size[0], ystart * SCROLLBAR_UNIT + y + 1) lbessard@140: lbessard@140: # Draw all elements lbessard@42: for comment in self.Comments: lbessard@42: if comment != self.SelectedElement: lbessard@42: comment.Draw(dc) etisserant@0: for wire in self.Wires: etisserant@0: if wire != self.SelectedElement: etisserant@0: wire.Draw(dc) lbessard@42: for block in self.Blocks: lbessard@42: if block != self.SelectedElement: lbessard@42: block.Draw(dc) lbessard@144: etisserant@0: if self.SelectedElement: lbessard@213: self.SelectedElement.Draw(dc) lbessard@213: lbessard@213: if not printing: lbessard@213: if self.rubberBand.IsShown(): lbessard@213: self.rubberBand.Draw(dc) lbessard@213: dc.EndDrawing() lbessard@213: lbessard@213: def OnPaint(self, event): lbessard@213: self.DoDrawing(self.GetLogicalDC(True)) lbessard@213: event.Skip() lbessard@213: lbessard@213: