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: Laurent@738: import re lbessard@332: import math laurent@667: import time laurent@471: from types import TupleType laurent@471: from threading import Lock lbessard@332: etisserant@0: import wx etisserant@0: etisserant@0: from plcopen.structures import * Laurent@694: from PLCControler import ITEM_POU, ITEM_PROGRAM, ITEM_FUNCTIONBLOCK lbessard@27: laurent@409: from dialogs import * laurent@409: from graphics import * laurent@586: from controls import EditorPanel lbessard@27: lbessard@27: SCROLLBAR_UNIT = 10 lbessard@27: WINDOW_BORDER = 10 lbessard@27: SCROLL_ZONE = 10 etisserant@0: laurent@381: CURSORS = None laurent@381: laurent@381: def ResetCursors(): laurent@381: global CURSORS laurent@381: if CURSORS == None: laurent@381: CURSORS = [wx.NullCursor, laurent@381: wx.StockCursor(wx.CURSOR_HAND), laurent@381: wx.StockCursor(wx.CURSOR_SIZENWSE), laurent@381: wx.StockCursor(wx.CURSOR_SIZENESW), laurent@381: wx.StockCursor(wx.CURSOR_SIZEWE), laurent@381: wx.StockCursor(wx.CURSOR_SIZENS)] laurent@381: 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', laurent@563: 'size' : 10, lbessard@157: } lbessard@157: else: lbessard@157: faces = { 'times': 'Times', lbessard@157: 'mono' : 'Courier', lbessard@157: 'helv' : 'Helvetica', lbessard@157: 'other': 'new century schoolbook', laurent@563: 'size' : 12, lbessard@157: } lbessard@157: lbessard@332: ZOOM_FACTORS = [math.sqrt(2) ** x for x in xrange(-6, 7)] lbessard@332: laurent@383: def GetVariableCreationFunction(variable_type): laurent@383: def variableCreationFunction(viewer, id, specific_values): laurent@383: return FBD_Variable(viewer, variable_type, laurent@383: specific_values["name"], laurent@383: specific_values["value_type"], laurent@383: id, laurent@383: specific_values["executionOrder"]) laurent@383: return variableCreationFunction laurent@383: laurent@383: def GetConnectorCreationFunction(connector_type): laurent@383: def connectorCreationFunction(viewer, id, specific_values): laurent@383: return FBD_Connector(viewer, connector_type, laurent@383: specific_values["name"], id) laurent@383: return connectorCreationFunction laurent@383: laurent@383: def commentCreationFunction(viewer, id, specific_values): laurent@383: return Comment(viewer, specific_values["content"], id) laurent@383: laurent@383: def GetPowerRailCreationFunction(powerrail_type): laurent@383: def powerRailCreationFunction(viewer, id, specific_values): laurent@383: return LD_PowerRail(viewer, powerrail_type, id, laurent@383: specific_values["connectors"]) laurent@383: return powerRailCreationFunction laurent@383: laurent@383: CONTACT_TYPES = {(True, "none"): CONTACT_REVERSE, laurent@383: (False, "rising"): CONTACT_RISING, laurent@383: (False, "falling"): CONTACT_FALLING} laurent@383: laurent@383: def contactCreationFunction(viewer, id, specific_values): laurent@383: contact_type = CONTACT_TYPES.get((specific_values.get("negated", False), laurent@383: specific_values.get("edge", "none")), laurent@383: CONTACT_NORMAL) laurent@383: return LD_Contact(viewer, contact_type, specific_values["name"], id) laurent@383: laurent@383: COIL_TYPES = {(True, "none", "none"): COIL_REVERSE, laurent@383: (False, "none", "set"): COIL_SET, laurent@383: (False, "none", "reset"): COIL_RESET, laurent@383: (False, "rising", "none"): COIL_RISING, laurent@383: (False, "falling", "none"): COIL_FALLING} laurent@383: laurent@383: def coilCreationFunction(viewer, id, specific_values): laurent@383: coil_type = COIL_TYPES.get((specific_values.get("negated", False), laurent@383: specific_values.get("edge", "none"), laurent@383: specific_values.get("storage", "none")), laurent@383: COIL_NORMAL) laurent@383: return LD_Coil(viewer, coil_type, specific_values["name"], id) laurent@383: laurent@383: def stepCreationFunction(viewer, id, specific_values): laurent@383: step = SFC_Step(viewer, specific_values["name"], laurent@383: specific_values.get("initial", False), id) laurent@383: if specific_values.get("action", None): laurent@383: step.AddAction() laurent@383: connector = step.GetActionConnector() laurent@383: connector.SetPosition(wx.Point(*specific_values["action"]["position"])) laurent@383: return step laurent@383: laurent@383: def transitionCreationFunction(viewer, id, specific_values): laurent@383: transition = SFC_Transition(viewer, specific_values["condition_type"], laurent@383: specific_values.get("condition", None), laurent@383: specific_values["priority"], id) laurent@383: return transition laurent@383: laurent@383: def GetDivergenceCreationFunction(divergence_type): laurent@383: def divergenceCreationFunction(viewer, id, specific_values): laurent@383: return SFC_Divergence(viewer, divergence_type, laurent@383: specific_values["connectors"], id) laurent@383: return divergenceCreationFunction laurent@383: laurent@383: def jumpCreationFunction(viewer, id, specific_values): laurent@383: return SFC_Jump(viewer, specific_values["target"], id) laurent@383: laurent@383: def actionBlockCreationFunction(viewer, id, specific_values): laurent@383: return SFC_ActionBlock(viewer, specific_values["actions"], id) laurent@383: laurent@383: ElementCreationFunctions = { laurent@383: "input": GetVariableCreationFunction(INPUT), laurent@383: "output": GetVariableCreationFunction(OUTPUT), laurent@383: "inout": GetVariableCreationFunction(INOUT), laurent@383: "connector": GetConnectorCreationFunction(CONNECTOR), laurent@383: "continuation": GetConnectorCreationFunction(CONTINUATION), laurent@383: "comment": commentCreationFunction, laurent@383: "leftPowerRail": GetPowerRailCreationFunction(LEFTRAIL), laurent@383: "rightPowerRail": GetPowerRailCreationFunction(RIGHTRAIL), laurent@383: "contact": contactCreationFunction, laurent@383: "coil": coilCreationFunction, laurent@383: "step": stepCreationFunction, laurent@383: "transition": transitionCreationFunction, laurent@383: "selectionDivergence": GetDivergenceCreationFunction(SELECTION_DIVERGENCE), laurent@383: "selectionConvergence": GetDivergenceCreationFunction(SELECTION_CONVERGENCE), laurent@383: "simultaneousDivergence": GetDivergenceCreationFunction(SIMULTANEOUS_DIVERGENCE), laurent@383: "simultaneousConvergence": GetDivergenceCreationFunction(SIMULTANEOUS_CONVERGENCE), laurent@383: "jump": jumpCreationFunction, laurent@383: "actionBlock": actionBlockCreationFunction, laurent@383: } laurent@383: Laurent@738: def sort_blocks(block_infos1, block_infos2): Laurent@738: x1, y1 = block_infos1[0].GetPosition() Laurent@738: x2, y2 = block_infos2[0].GetPosition() Laurent@738: if y1 == y2: Laurent@738: return cmp(x1, x2) Laurent@738: else: Laurent@738: return cmp(y1, y2) Laurent@738: 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): laurent@437: tagname = self.ParentWindow.GetTagName() laurent@437: pou_name, pou_type = self.ParentWindow.Controler.GetEditedElementType(tagname, self.ParentWindow.Debug) lbessard@108: x, y = self.ParentWindow.CalcUnscrolledPosition(x, y) lbessard@323: x = int(x / self.ParentWindow.ViewScale[0]) lbessard@323: y = int(y / self.ParentWindow.ViewScale[1]) lbessard@145: scaling = self.ParentWindow.Scaling lbessard@218: message = None lbessard@218: try: lbessard@218: values = eval(data) lbessard@218: except: laurent@391: message = _("Invalid value \"%s\" for viewer block")%data lbessard@218: values = None lbessard@218: if not isinstance(values, TupleType): laurent@391: message = _("Invalid value \"%s\" for viewer block")%data lbessard@218: values = None lbessard@218: if values is not None: lbessard@249: if values[1] == "debug": lbessard@249: pass lbessard@249: elif values[1] == "program": laurent@391: message = _("Programs can't be used by other POUs!") laurent@437: elif values[1] in ["function", "functionBlock"]: laurent@437: words = tagname.split("::") laurent@437: if pou_name == values[0]: laurent@437: message = _("\"%s\" can't use itself!")%pou_name laurent@437: elif pou_type == "function" and values[1] != "function": laurent@391: message = _("Function Blocks can't be used in Functions!") lbessard@253: elif words[0] == "T" and values[1] != "function": laurent@391: message = _("Function Blocks can't be used in Transitions!") laurent@437: elif self.ParentWindow.Controler.PouIsUsedBy(pou_name, values[0], self.ParentWindow.Debug): laurent@437: message = _("\"%s\" is already used by \"%s\"!")%(pou_name, values[0]) lbessard@195: else: lbessard@218: blockname = values[2] lbessard@239: if len(values) > 3: lbessard@239: blockinputs = values[3] lbessard@239: else: lbessard@239: blockinputs = None lbessard@218: if values[1] != "function" and blockname == "": Laurent@704: blockname = self.ParentWindow.GenerateNewName(blocktype=values[0]) lbessard@249: if blockname.upper() in [name.upper() for name in self.ParentWindow.Controler.GetProjectPouNames(self.ParentWindow.Debug)]: laurent@391: message = _("\"%s\" pou already exists!")%blockname laurent@437: elif blockname.upper() in [name.upper() for name in self.ParentWindow.Controler.GetEditedElementVariables(tagname, self.ParentWindow.Debug)]: laurent@391: message = _("\"%s\" element for this pou already exists!")%blockname lbessard@218: else: lbessard@218: id = self.ParentWindow.GetNewId() lbessard@239: block = FBD_Block(self.ParentWindow, values[0], blockname, id, inputs = blockinputs) lbessard@218: width, height = block.GetMinSize() lbessard@218: if scaling is not None: lbessard@218: x = round(float(x) / float(scaling[0])) * scaling[0] lbessard@218: y = round(float(y) / float(scaling[1])) * scaling[1] lbessard@218: width = round(float(width) / float(scaling[0]) + 0.5) * scaling[0] lbessard@218: height = round(float(height) / float(scaling[1]) + 0.5) * scaling[1] lbessard@218: block.SetPosition(x, y) lbessard@218: block.SetSize(width, height) lbessard@218: self.ParentWindow.AddBlock(block) laurent@437: self.ParentWindow.Controler.AddEditedElementBlock(tagname, id, values[0], blockname) lbessard@218: self.ParentWindow.RefreshBlockModel(block) lbessard@218: self.ParentWindow.RefreshBuffer() lbessard@218: self.ParentWindow.RefreshScrollBars() lbessard@249: self.ParentWindow.RefreshVisibleElements() laurent@586: self.ParentWindow.RefreshVariablePanel() lbessard@218: self.ParentWindow.Refresh(False) laurent@437: elif values[1] == "location": Laurent@713: if pou_type == "program": Laurent@713: location = values[0] Laurent@713: if not location.startswith("%"): Laurent@720: dialog = wx.SingleChoiceDialog(self.ParentWindow, Laurent@720: _("Select a variable class:"), _("Variable class"), Laurent@720: ["Input", "Output", "Memory"], Laurent@720: wx.DEFAULT_DIALOG_STYLE|wx.OK|wx.CANCEL) Laurent@713: if dialog.ShowModal() == wx.ID_OK: Laurent@713: selected = dialog.GetSelection() Laurent@713: else: Laurent@713: selected = None Laurent@713: dialog.Destroy() Laurent@713: if selected is None: Laurent@713: return Laurent@713: if selected == 0: Laurent@713: location = "%I" + location Laurent@713: elif selected == 1: Laurent@713: location = "%Q" + location Laurent@713: else: Laurent@713: location = "%M" + location laurent@437: var_name = values[3] laurent@437: if var_name.upper() in [name.upper() for name in self.ParentWindow.Controler.GetProjectPouNames(self.ParentWindow.Debug)]: laurent@441: message = _("\"%s\" pou already exists!")%var_name lbessard@218: else: Laurent@713: if location[1] == "Q": laurent@437: var_class = OUTPUT lbessard@297: else: laurent@437: var_class = INPUT laurent@437: if values[2] is not None: laurent@437: var_type = values[2] laurent@437: else: Laurent@713: var_type = LOCATIONDATATYPES.get(location[2], ["BOOL"])[0] laurent@441: if not var_name.upper() in [name.upper() for name in self.ParentWindow.Controler.GetEditedElementVariables(tagname, self.ParentWindow.Debug)]: Laurent@713: self.ParentWindow.Controler.AddEditedElementPouVar(tagname, var_type, var_name, location, values[4]) laurent@616: self.ParentWindow.RefreshVariablePanel() laurent@616: self.ParentWindow.AddVariableBlock(x, y, scaling, var_class, var_name, var_type) laurent@616: elif values[1] == "Global": laurent@616: var_name = values[0] laurent@616: if var_name.upper() in [name.upper() for name in self.ParentWindow.Controler.GetProjectPouNames(self.ParentWindow.Debug)]: laurent@616: message = _("\"%s\" pou already exists!")%var_name laurent@616: else: laurent@616: if not var_name.upper() in [name.upper() for name in self.ParentWindow.Controler.GetEditedElementVariables(tagname, self.ParentWindow.Debug)]: laurent@616: self.ParentWindow.Controler.AddEditedElementPouExternalVar(tagname, values[2], var_name) laurent@586: self.ParentWindow.RefreshVariablePanel() laurent@616: self.ParentWindow.AddVariableBlock(x, y, scaling, INPUT, var_name, values[2]) Laurent@716: elif values[1] == "Constant": Laurent@716: self.ParentWindow.AddVariableBlock(x, y, scaling, INPUT, values[0], None) laurent@437: elif values[3] == tagname: laurent@437: if values[1] == "Output": laurent@437: var_class = OUTPUT laurent@437: elif values[1] == "InOut": laurent@437: var_class = INPUT laurent@437: else: laurent@437: var_class = INPUT laurent@437: tree = dict([(var["Name"], var["Tree"]) for var in self.ParentWindow.Controler.GetEditedElementInterfaceVars(tagname, self.ParentWindow.Debug)]).get(values[0], None) laurent@437: if tree is not None: laurent@437: if len(tree[0]) > 0: laurent@437: menu = wx.Menu(title='') laurent@437: self.GenerateTreeMenu(x, y, scaling, menu, "", var_class, [(values[0], values[2], tree)]) laurent@437: self.ParentWindow.PopupMenuXY(menu) lbessard@297: else: laurent@510: self.ParentWindow.AddVariableBlock(x, y, scaling, var_class, values[0], values[2]) lbessard@121: else: laurent@437: message = _("Unknown variable \"%s\" for this POU!") % values[0] laurent@437: else: laurent@437: message = _("Variable don't belong to this POU!") lbessard@249: if message is not None: lbessard@249: wx.CallAfter(self.ShowMessage, message) lbessard@218: lbessard@297: def GenerateTreeMenu(self, x, y, scaling, menu, base_path, var_class, tree): lbessard@299: for child_name, child_type, (child_tree, child_dimensions) in tree: lbessard@297: if base_path: lbessard@297: child_path = "%s.%s" % (base_path, child_name) lbessard@297: else: lbessard@297: child_path = child_name lbessard@299: if len(child_dimensions) > 0: lbessard@299: child_path += "[%s]" % ",".join([str(dimension[0]) for dimension in child_dimensions]) lbessard@297: child_name += "[]" lbessard@297: new_id = wx.NewId() lbessard@297: AppendMenu(menu, help='', id=new_id, kind=wx.ITEM_NORMAL, text=child_name) lbessard@297: self.ParentWindow.Bind(wx.EVT_MENU, self.GetAddVariableBlockFunction(x, y, scaling, var_class, child_path, child_type), id=new_id) lbessard@297: if len(child_tree) > 0: lbessard@297: new_id = wx.NewId() lbessard@297: child_menu = wx.Menu(title='') lbessard@297: self.GenerateTreeMenu(x, y, scaling, child_menu, child_path, var_class, child_tree) lbessard@297: menu.AppendMenu(new_id, "%s." % child_name, child_menu) lbessard@297: lbessard@297: def GetAddVariableBlockFunction(self, x, y, scaling, var_class, var_name, var_type): lbessard@297: def AddVariableFunction(event): laurent@510: self.ParentWindow.AddVariableBlock(x, y, scaling, var_class, var_name, var_type) lbessard@297: return AddVariableFunction lbessard@297: lbessard@218: def ShowMessage(self, message): laurent@391: message = wx.MessageDialog(self.ParentWindow, message, _("Error"), wx.OK|wx.ICON_ERROR) lbessard@249: message.ShowModal() lbessard@249: message.Destroy() lbessard@235: 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: laurent@586: class Viewer(EditorPanel, DebugViewer): 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: laurent@666: # Add list of menu items to the given menu laurent@666: def AddMenuItems(self, menu, items): laurent@666: for item in items: laurent@666: if item is None: laurent@666: menu.AppendSeparator() laurent@666: else: laurent@666: id, kind, text, help, callback = item laurent@666: AppendMenu(menu, help=help, id=id, kind=kind, text=text) laurent@666: # Link menu event to corresponding called functions laurent@666: self.Bind(wx.EVT_MENU, callback, id=id) laurent@666: laurent@666: # Add Block Pin Menu items to the given menu laurent@666: def AddBlockPinMenuItems(self, menu, connector): laurent@666: [ID_NO_MODIFIER, ID_NEGATED, ID_RISING_EDGE, laurent@666: ID_FALLING_EDGE] = [wx.NewId() for i in xrange(4)] laurent@666: lbessard@138: # Create menu items laurent@666: self.AddMenuItems(menu, [ laurent@666: (ID_NO_MODIFIER, wx.ITEM_RADIO, _(u'No Modifier'), '', self.OnNoModifierMenu), laurent@666: (ID_NEGATED, wx.ITEM_RADIO, _(u'Negated'), '', self.OnNegatedMenu), laurent@666: (ID_RISING_EDGE, wx.ITEM_RADIO, _(u'Rising Edge'), '', self.OnRisingEdgeMenu), laurent@666: (ID_FALLING_EDGE, wx.ITEM_RADIO, _(u'Falling Edge'), '', self.OnFallingEdgeMenu)]) laurent@666: laurent@666: type = self.Controler.GetEditedElementType(self.TagName, self.Debug) laurent@666: menu.Enable(ID_RISING_EDGE, type != "function") laurent@666: menu.Enable(ID_FALLING_EDGE, type != "function") laurent@666: laurent@666: if connector.IsNegated(): laurent@666: menu.Check(ID_NEGATED, True) laurent@666: elif connector.GetEdge() == "rising": laurent@666: menu.Check(ID_RISING_EDGE, True) laurent@666: elif connector.GetEdge() == "falling": laurent@666: menu.Check(ID_FALLING_EDGE, True) laurent@666: else: laurent@666: menu.Check(ID_NO_MODIFIER, True) laurent@666: laurent@666: # Add Alignment Menu items to the given menu laurent@666: def AddAlignmentMenuItems(self, menu): laurent@666: [ID_ALIGN_LEFT, ID_ALIGN_CENTER, ID_ALIGN_RIGHT, laurent@666: ID_ALIGN_TOP, ID_ALIGN_MIDDLE, ID_ALIGN_BOTTOM, laurent@666: ] = [wx.NewId() for i in xrange(6)] laurent@666: etisserant@0: # Create menu items laurent@666: self.AddMenuItems(menu, [ laurent@666: (ID_ALIGN_LEFT, wx.ITEM_NORMAL, _(u'Left'), '', self.OnAlignLeftMenu), laurent@666: (ID_ALIGN_CENTER, wx.ITEM_NORMAL, _(u'Center'), '', self.OnAlignCenterMenu), laurent@666: (ID_ALIGN_RIGHT, wx.ITEM_NORMAL, _(u'Right'), '', self.OnAlignRightMenu), laurent@666: None, laurent@666: (ID_ALIGN_TOP, wx.ITEM_NORMAL, _(u'Top'), '', self.OnAlignTopMenu), laurent@666: (ID_ALIGN_MIDDLE, wx.ITEM_NORMAL, _(u'Middle'), '', self.OnAlignMiddleMenu), laurent@666: (ID_ALIGN_BOTTOM, wx.ITEM_NORMAL, _(u'Bottom'), '', self.OnAlignBottomMenu)]) laurent@666: laurent@666: # Add Wire Menu items to the given menu laurent@666: def AddWireMenuItems(self, menu, delete=False): laurent@666: [ID_ADD_SEGMENT, ID_DELETE_SEGMENT] = [wx.NewId() for i in xrange(2)] laurent@666: laurent@666: # Create menu items laurent@666: self.AddMenuItems(menu, [ laurent@666: (ID_ADD_SEGMENT, wx.ITEM_NORMAL, _(u'Add Wire Segment'), '', self.OnAddSegmentMenu), laurent@666: (ID_DELETE_SEGMENT, wx.ITEM_NORMAL, _(u'Delete Wire Segment'), '', self.OnDeleteSegmentMenu)]) laurent@666: laurent@666: menu.Enable(ID_DELETE_SEGMENT, delete) laurent@666: laurent@666: # Add Divergence Menu items to the given menu laurent@666: def AddDivergenceMenuItems(self, menu, delete=False): laurent@666: [ID_ADD_BRANCH, ID_DELETE_BRANCH] = [wx.NewId() for i in xrange(2)] laurent@666: laurent@666: # Create menu items laurent@666: self.AddMenuItems(menu, [ laurent@666: (ID_ADD_BRANCH, wx.ITEM_NORMAL, _(u'Add Divergence Branch'), '', self.OnAddBranchMenu), laurent@666: (ID_DELETE_BRANCH, wx.ITEM_NORMAL, _(u'Delete Divergence Branch'), '', self.OnDeleteBranchMenu)]) laurent@666: laurent@666: menu.Enable(ID_DELETE_BRANCH, delete) laurent@666: laurent@666: # Add Add Menu items to the given menu laurent@666: def AddAddMenuItems(self, menu): laurent@666: [ID_ADD_BLOCK, ID_ADD_VARIABLE, ID_ADD_CONNECTION, laurent@666: ID_ADD_COMMENT] = [wx.NewId() for i in xrange(4)] laurent@666: laurent@666: # Create menu items laurent@666: self.AddMenuItems(menu, [ laurent@666: (ID_ADD_BLOCK, wx.ITEM_NORMAL, _(u'Block'), '', self.GetAddMenuCallBack(self.AddNewBlock)), laurent@666: (ID_ADD_VARIABLE, wx.ITEM_NORMAL, _(u'Variable'), '', self.GetAddMenuCallBack(self.AddNewVariable)), laurent@666: (ID_ADD_CONNECTION, wx.ITEM_NORMAL, _(u'Connection'), '', self.GetAddMenuCallBack(self.AddNewConnection)), laurent@666: None]) laurent@666: laurent@666: if self.CurrentLanguage != "FBD": laurent@666: [ID_ADD_POWER_RAIL, ID_ADD_CONTACT, ID_ADD_COIL, laurent@666: ] = [wx.NewId() for i in xrange(3)] laurent@666: laurent@666: # Create menu items laurent@666: self.AddMenuItems(menu, [ laurent@666: (ID_ADD_POWER_RAIL, wx.ITEM_NORMAL, _(u'Power Rail'), '', self.GetAddMenuCallBack(self.AddNewPowerRail)), laurent@666: (ID_ADD_CONTACT, wx.ITEM_NORMAL, _(u'Contact'), '', self.GetAddMenuCallBack(self.AddNewContact))]) laurent@666: laurent@666: if self.CurrentLanguage != "SFC": laurent@666: self.AddMenuItems(menu, [ laurent@666: (ID_ADD_COIL, wx.ITEM_NORMAL, _(u'Coil'), '', self.GetAddMenuCallBack(self.AddNewCoil))]) laurent@666: laurent@666: menu.AppendSeparator() laurent@666: laurent@666: if self.CurrentLanguage == "SFC": laurent@666: [ID_ADD_INITIAL_STEP, ID_ADD_STEP, ID_ADD_TRANSITION, laurent@666: ID_ADD_ACTION_BLOCK, ID_ADD_DIVERGENCE, ID_ADD_JUMP, laurent@666: ] = [wx.NewId() for i in xrange(6)] laurent@666: laurent@666: # Create menu items laurent@666: self.AddMenuItems(menu, [ laurent@666: (ID_ADD_INITIAL_STEP, wx.ITEM_NORMAL, _(u'Initial Step'), '', self.GetAddMenuCallBack(self.AddNewStep, True)), laurent@666: (ID_ADD_STEP, wx.ITEM_NORMAL, _(u'Step'), '', self.GetAddMenuCallBack(self.AddNewStep)), laurent@666: (ID_ADD_TRANSITION, wx.ITEM_NORMAL, _(u'Transition'), '', self.GetAddMenuCallBack(self.AddNewTransition)), laurent@666: (ID_ADD_ACTION_BLOCK, wx.ITEM_NORMAL, _(u'Action Block'), '', self.GetAddMenuCallBack(self.AddNewActionBlock)), laurent@666: (ID_ADD_DIVERGENCE, wx.ITEM_NORMAL, _(u'Divergence'), '', self.GetAddMenuCallBack(self.AddNewDivergence)), laurent@666: (ID_ADD_JUMP, wx.ITEM_NORMAL, _(u'Jump'), '', self.GetAddMenuCallBack(self.AddNewJump)), laurent@666: None]) laurent@666: laurent@666: self.AddMenuItems(menu, [ laurent@666: (ID_ADD_COMMENT, wx.ITEM_NORMAL, _(u'Comment'), '', self.GetAddMenuCallBack(self.AddNewComment))]) laurent@666: laurent@666: # Add Default Menu items to the given menu laurent@666: def AddDefaultMenuItems(self, menu, edit=False, block=False): laurent@666: if block: laurent@666: [ID_EDIT_BLOCK, ID_DELETE] = [wx.NewId() for i in xrange(2)] laurent@666: laurent@666: # Create menu items laurent@666: self.AddMenuItems(menu, [ laurent@666: (ID_EDIT_BLOCK, wx.ITEM_NORMAL, _(u'Edit Block'), '', self.OnEditBlockMenu), laurent@666: (ID_DELETE, wx.ITEM_NORMAL, _(u'Delete'), '', self.OnDeleteMenu)]) laurent@666: laurent@666: menu.Enable(ID_EDIT_BLOCK, edit) laurent@666: laurent@666: else: laurent@666: [ID_CLEAR_EXEC_ORDER, ID_RESET_EXEC_ORDER] = [wx.NewId() for i in xrange(2)] laurent@666: laurent@666: # Create menu items laurent@666: self.AddMenuItems(menu, [ laurent@666: (ID_CLEAR_EXEC_ORDER, wx.ITEM_NORMAL, _(u'Clear Execution Order'), '', self.OnClearExecutionOrderMenu), laurent@666: (ID_RESET_EXEC_ORDER, wx.ITEM_NORMAL, _(u'Reset Execution Order'), '', self.OnResetExecutionOrderMenu)]) laurent@666: laurent@666: menu.AppendSeparator() laurent@666: laurent@666: add_menu = wx.Menu(title='') laurent@666: self.AddAddMenuItems(add_menu) laurent@666: menu.AppendMenu(-1, _(u'Add'), add_menu) laurent@666: laurent@666: menu.AppendSeparator() laurent@666: laurent@666: [ID_CUT, ID_COPY, ID_PASTE] = [wx.NewId() for i in xrange(3)] laurent@666: laurent@666: # Create menu items laurent@666: self.AddMenuItems(menu, [ laurent@666: (ID_CUT, wx.ITEM_NORMAL, _(u'Cut'), '', self.GetClipboardCallBack(self.Cut)), laurent@666: (ID_COPY, wx.ITEM_NORMAL, _(u'Copy'), '', self.GetClipboardCallBack(self.Copy)), laurent@666: (ID_PASTE, wx.ITEM_NORMAL, _(u'Paste'), '', self.GetAddMenuCallBack(self.Paste))]) laurent@666: laurent@666: menu.Enable(ID_CUT, block) laurent@666: menu.Enable(ID_COPY, block) laurent@666: menu.Enable(ID_PASTE, self.ParentWindow.GetCopyBuffer() is not None) laurent@666: laurent@586: def _init_Editor(self, prnt): laurent@586: self.Editor = wx.ScrolledWindow(prnt, name="Viewer", laurent@586: pos=wx.Point(0, 0), size=wx.Size(0, 0), laurent@586: style=wx.HSCROLL | wx.VSCROLL | wx.ALWAYS_SHOW_SB) Laurent@699: self.Editor.ParentWindow = self laurent@586: etisserant@0: # Create a new Viewer lbessard@249: def __init__(self, parent, tagname, window, controler, debug = False, instancepath = ""): laurent@586: self.VARIABLE_PANEL_TYPE = controler.GetPouType(tagname.split("::")[1]) laurent@586: laurent@600: EditorPanel.__init__(self, parent, tagname, window, controler, debug) greg@361: DebugViewer.__init__(self, controler, debug) laurent@586: etisserant@0: # Adding a rubberband to Viewer laurent@586: self.rubberBand = RubberBand(viewer=self) laurent@586: self.Editor.SetBackgroundColour(wx.Colour(255,255,255)) laurent@586: self.Editor.SetBackgroundStyle(wx.BG_STYLE_CUSTOM) 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 laurent@566: self.Highlights = [] Laurent@738: self.SearchParams = None Laurent@738: self.SearchResults = None Laurent@738: self.CurrentFindHighlight = None lbessard@249: self.InstancePath = instancepath lbessard@319: self.StartMousePos = None lbessard@319: self.StartScreenPos = None laurent@667: self.Buffering = False etisserant@0: laurent@381: # Initialize Cursors laurent@381: ResetCursors() laurent@381: self.CurrentCursor = 0 laurent@381: lbessard@42: # Initialize Block, Wire and Comment numbers laurent@383: self.wire_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@249: if not self.Debug: laurent@586: self.Editor.SetDropTarget(ViewerDropTarget(self)) lbessard@47: laurent@634: self.ElementRefreshList = [] laurent@634: self.ElementRefreshList_lock = Lock() greg@361: laurent@586: dc = wx.ClientDC(self.Editor) 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) greg@361: self.MiniTextDC = wx.MemoryDC() greg@361: self.MiniTextDC.SetFont(wx.Font(faces["size"] * 0.75, wx.SWISS, wx.NORMAL, wx.NORMAL, faceName = faces["helv"])) lbessard@156: laurent@586: self.CurrentScale = None laurent@586: self.SetScale(len(ZOOM_FACTORS) / 2, False) lbessard@332: laurent@566: self.RefreshHighlightsTimer = wx.Timer(self, -1) laurent@566: self.Bind(wx.EVT_TIMER, self.OnRefreshHighlightsTimer, self.RefreshHighlightsTimer) laurent@566: lbessard@249: self.ResetView() lbessard@249: etisserant@0: # Link Viewer event to corresponding methods laurent@586: self.Editor.Bind(wx.EVT_PAINT, self.OnPaint) laurent@586: self.Editor.Bind(wx.EVT_LEFT_DOWN, self.OnViewerLeftDown) laurent@586: self.Editor.Bind(wx.EVT_LEFT_UP, self.OnViewerLeftUp) laurent@586: self.Editor.Bind(wx.EVT_LEFT_DCLICK, self.OnViewerLeftDClick) laurent@586: self.Editor.Bind(wx.EVT_RIGHT_DOWN, self.OnViewerRightDown) laurent@586: self.Editor.Bind(wx.EVT_RIGHT_UP, self.OnViewerRightUp) laurent@586: self.Editor.Bind(wx.EVT_MIDDLE_DOWN, self.OnViewerMiddleDown) laurent@586: self.Editor.Bind(wx.EVT_MIDDLE_UP, self.OnViewerMiddleUp) laurent@586: self.Editor.Bind(wx.EVT_LEAVE_WINDOW, self.OnLeaveViewer) laurent@586: self.Editor.Bind(wx.EVT_MOTION, self.OnViewerMotion) laurent@586: self.Editor.Bind(wx.EVT_CHAR, self.OnChar) laurent@586: self.Editor.Bind(wx.EVT_SCROLLWIN, self.OnScrollWindow) laurent@586: self.Editor.Bind(wx.EVT_SCROLLWIN_THUMBRELEASE, self.OnScrollStop) laurent@586: self.Editor.Bind(wx.EVT_MOUSEWHEEL, self.OnMouseWheelWindow) laurent@586: self.Editor.Bind(wx.EVT_SIZE, self.OnMoveWindow) laurent@633: self.Editor.Bind(wx.EVT_MOUSE_EVENTS, self.OnViewerMouseEvent) laurent@641: laurent@566: def __del__(self): laurent@566: DebugViewer.__del__(self) laurent@566: self.RefreshHighlightsTimer.Stop() laurent@566: laurent@381: def SetCurrentCursor(self, cursor): laurent@575: if self.Mode != MODE_MOTION: laurent@575: global CURSORS laurent@575: if self.CurrentCursor != cursor: laurent@575: self.CurrentCursor = cursor laurent@586: self.Editor.SetCursor(CURSORS[cursor]) laurent@381: lbessard@144: def GetScrolledRect(self, rect): laurent@586: rect.x, rect.y = self.Editor.CalcScrolledPosition(int(rect.x * self.ViewScale[0]), lbessard@323: int(rect.y * self.ViewScale[1])) lbessard@323: rect.width = int(rect.width * self.ViewScale[0]) + 2 lbessard@323: rect.height = int(rect.height * self.ViewScale[1]) + 2 lbessard@144: return rect etisserant@0: laurent@586: def GetTitle(self): laurent@586: if self.Debug: laurent@586: if len(self.InstancePath) > 15: laurent@586: return "..." + self.InstancePath[-12:] laurent@586: return self.InstancePath laurent@586: return EditorPanel.GetTitle(self) laurent@586: lbessard@145: def GetScaling(self): lbessard@145: return self.Scaling lbessard@145: lbessard@249: def GetInstancePath(self): lbessard@249: return self.InstancePath lbessard@249: lbessard@121: def IsViewing(self, tagname): lbessard@249: if self.Debug: lbessard@249: return self.InstancePath == tagname laurent@586: return EditorPanel.IsViewing(self, 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): greg@361: DebugViewer.__del__(self) lbessard@249: self.Flush() etisserant@0: self.ResetView() lbessard@323: laurent@586: def SetScale(self, scale_number, refresh=True, mouse_event=None): laurent@586: new_scale = max(0, min(scale_number, len(ZOOM_FACTORS) - 1)) laurent@586: if self.CurrentScale != new_scale: laurent@586: if refresh: laurent@586: dc = self.GetLogicalDC() laurent@586: self.CurrentScale = new_scale laurent@586: self.ViewScale = (ZOOM_FACTORS[self.CurrentScale], ZOOM_FACTORS[self.CurrentScale]) laurent@586: if refresh: laurent@586: self.Editor.Freeze() laurent@586: if mouse_event is None: laurent@586: client_size = self.Editor.GetClientSize() laurent@586: mouse_pos = wx.Point(client_size[0] / 2, client_size[1] / 2) laurent@586: mouse_event = wx.MouseEvent(wx.EVT_MOUSEWHEEL.typeId) laurent@586: mouse_event.m_x = mouse_pos.x laurent@586: mouse_event.m_y = mouse_pos.y laurent@586: else: laurent@586: mouse_pos = mouse_event.GetPosition() laurent@586: pos = mouse_event.GetLogicalPosition(dc) laurent@586: xmax = self.GetScrollRange(wx.HORIZONTAL) - self.GetScrollThumb(wx.HORIZONTAL) laurent@586: ymax = self.GetScrollRange(wx.VERTICAL) - self.GetScrollThumb(wx.VERTICAL) laurent@586: scrollx = max(0, round(pos.x * self.ViewScale[0] - mouse_pos.x) / SCROLLBAR_UNIT) laurent@586: scrolly = max(0, round(pos.y * self.ViewScale[1] - mouse_pos.y) / SCROLLBAR_UNIT) laurent@586: if scrollx > xmax or scrolly > ymax: laurent@586: self.RefreshScrollBars(max(0, scrollx - xmax), max(0, scrolly - ymax)) laurent@586: self.Scroll(scrollx, scrolly) laurent@586: else: laurent@586: self.Scroll(scrollx, scrolly) laurent@586: self.RefreshScrollBars() laurent@586: self.RefreshScaling(refresh) laurent@586: self.Editor.Thaw() laurent@586: lbessard@332: def GetScale(self): lbessard@332: return self.CurrentScale lbessard@332: laurent@633: def GetViewScale(self): laurent@633: return self.ViewScale laurent@633: laurent@675: def GetState(self): laurent@675: return {"position": self.Editor.GetViewStart(), laurent@675: "zoom": self.CurrentScale} laurent@675: laurent@675: def SetState(self, state): laurent@687: if self: Laurent@704: if state.has_key("zoom"): Laurent@704: self.SetScale(state["zoom"]) Laurent@704: if state.has_key("position"): Laurent@704: self.Scroll(*state["position"]) laurent@687: self.RefreshVisibleElements() laurent@675: lbessard@90: def GetLogicalDC(self, buffered=False): lbessard@90: if buffered: laurent@586: bitmap = wx.EmptyBitmap(*self.Editor.GetClientSize()) greg@352: dc = wx.MemoryDC(bitmap) lbessard@90: else: laurent@586: dc = wx.ClientDC(self.Editor) lbessard@165: dc.SetFont(self.GetFont()) lbessard@114: if wx.VERSION >= (2, 6, 0): laurent@586: self.Editor.DoPrepareDC(dc) lbessard@114: else: laurent@586: self.Editor.PrepareDC(dc) lbessard@323: dc.SetUserScale(self.ViewScale[0], self.ViewScale[1]) lbessard@27: return dc laurent@586: laurent@586: def RefreshRect(self, rect, eraseBackground=True): laurent@586: self.Editor.RefreshRect(rect, eraseBackground) laurent@586: laurent@586: def Scroll(self, x, y): laurent@586: self.Editor.Scroll(x, y) laurent@586: laurent@586: def GetScrollPos(self, orientation): laurent@586: return self.Editor.GetScrollPos(orientation) laurent@586: laurent@586: def GetScrollRange(self, orientation): laurent@586: return self.Editor.GetScrollRange(orientation) laurent@586: laurent@586: def GetScrollThumb(self, orientation): laurent@586: return self.Editor.GetScrollThumb(orientation) laurent@586: laurent@586: def CalcUnscrolledPosition(self, x, y): laurent@586: return self.Editor.CalcUnscrolledPosition(x, y) laurent@586: laurent@586: def GetViewStart(self): laurent@586: return self.Editor.GetViewStart() laurent@586: laurent@586: def GetTextExtent(self, text): laurent@586: return self.Editor.GetTextExtent(text) laurent@586: laurent@586: def GetFont(self): laurent@586: return self.Editor.GetFont() laurent@586: greg@361: def GetMiniTextExtent(self, text): greg@361: return self.MiniTextDC.GetTextExtent(text) greg@361: lbessard@253: def GetMiniFont(self): greg@361: return self.MiniTextDC.GetFont() greg@361: etisserant@0: #------------------------------------------------------------------------------- lbessard@42: # Element management functions lbessard@42: #------------------------------------------------------------------------------- lbessard@42: lbessard@42: def AddBlock(self, block): laurent@383: self.Blocks[block.GetId()] = block 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): laurent@383: self.Comments[comment.GetId()] = comment lbessard@42: lbessard@42: def IsBlock(self, block): laurent@383: return self.Blocks.get(block.GetId(), 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): laurent@383: return self.Comments.get(comment.GetId(), False) lbessard@42: lbessard@42: def RemoveBlock(self, block): laurent@383: self.Blocks.pop(block.GetId()) lbessard@42: lbessard@42: def RemoveWire(self, wire): lbessard@42: self.Wires.pop(wire) lbessard@42: lbessard@42: def RemoveComment(self, comment): laurent@383: self.Comments.pop(comment.GetId()) lbessard@42: lbessard@42: def GetElements(self, sort_blocks=False, sort_wires=False, sort_comments=False): laurent@383: blocks = self.Blocks.values() lbessard@42: wires = self.Wires.keys() laurent@383: comments = self.Comments.values() lbessard@42: if sort_blocks: laurent@383: blocks.sort(lambda x, y: cmp(x.GetId(), y.GetId())) lbessard@42: if sort_wires: lbessard@283: wires.sort(lambda x, y: cmp(self.Wires[x], self.Wires[y])) lbessard@42: if sort_comments: laurent@383: comments.sort(lambda x, y: cmp(x.GetId(), y.GetId())) lbessard@42: return blocks + wires + comments lbessard@42: laurent@388: def GetConnectorByName(self, name): laurent@388: for block in self.Blocks.itervalues(): laurent@388: if isinstance(block, FBD_Connector) and\ laurent@388: block.GetType() == CONNECTOR and\ laurent@388: block.GetName() == name: laurent@388: return block laurent@388: return None laurent@388: lbessard@249: def RefreshVisibleElements(self, xp = None, yp = None): laurent@586: x, y = self.Editor.CalcUnscrolledPosition(0, 0) lbessard@249: if xp is not None: laurent@586: x = xp * self.Editor.GetScrollPixelsPerUnit()[0] lbessard@249: if yp is not None: laurent@586: y = yp * self.Editor.GetScrollPixelsPerUnit()[1] laurent@586: width, height = self.Editor.GetClientSize() lbessard@323: screen = wx.Rect(int(x / self.ViewScale[0]), int(y / self.ViewScale[1]), lbessard@323: int(width / self.ViewScale[0]), int(height / self.ViewScale[1])) laurent@383: for comment in self.Comments.itervalues(): lbessard@249: comment.TestVisible(screen) laurent@383: for wire in self.Wires.iterkeys(): lbessard@249: wire.TestVisible(screen) laurent@383: for block in self.Blocks.itervalues(): lbessard@249: block.TestVisible(screen) lbessard@343: lbessard@343: def GetElementIECPath(self, element): lbessard@343: iec_path = None lbessard@343: if isinstance(element, Wire) and element.EndConnected is not None: lbessard@343: block = element.EndConnected.GetParentBlock() lbessard@343: if isinstance(block, FBD_Block): lbessard@343: blockname = block.GetName() lbessard@343: connectorname = element.EndConnected.GetName() lbessard@343: if blockname != "": lbessard@343: iec_path = "%s.%s.%s"%(self.InstancePath, blockname, connectorname) lbessard@343: else: lbessard@343: if connectorname == "": lbessard@343: iec_path = "%s.%s%d"%(self.InstancePath, block.GetType(), block.GetId()) lbessard@343: else: lbessard@343: iec_path = "%s.%s%d_%s"%(self.InstancePath, block.GetType(), block.GetId(), connectorname) lbessard@343: elif isinstance(block, FBD_Variable): lbessard@343: iec_path = "%s.%s"%(self.InstancePath, block.GetName()) laurent@388: elif isinstance(block, FBD_Connector): laurent@388: connection = self.GetConnectorByName(block.GetName()) laurent@388: if connection is not None: laurent@388: connector = connection.GetConnector() laurent@388: if len(connector.Wires) == 1: laurent@388: iec_path = self.GetElementIECPath(connector.Wires[0][0]) Laurent@697: elif isinstance(element, LD_Contact): lbessard@343: iec_path = "%s.%s"%(self.InstancePath, element.GetName()) lbessard@343: elif isinstance(element, SFC_Step): lbessard@343: iec_path = "%s.%s.X"%(self.InstancePath, element.GetName()) lbessard@343: elif isinstance(element, SFC_Transition): lbessard@343: connectors = element.GetConnectors() laurent@477: previous_steps = self.GetPreviousSteps(connectors["inputs"]) laurent@477: next_steps = self.GetNextSteps(connectors["outputs"]) lbessard@343: iec_path = "%s.%s->%s"%(self.InstancePath, ",".join(previous_steps), ",".join(next_steps)) lbessard@343: return iec_path lbessard@343: 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 = {} lbessard@249: self.Subscribed = {} etisserant@0: self.SelectedElement = None lbessard@140: self.HighlightedElement = None laurent@633: self.ToolTipElement = None etisserant@0: lbessard@249: def Flush(self): greg@361: self.DeleteDataConsumers() laurent@383: for block in self.Blocks.itervalues(): lbessard@249: block.Flush() lbessard@249: lbessard@47: # Remove all elements lbessard@47: def CleanView(self): laurent@383: for block in self.Blocks.itervalues(): lbessard@47: block.Clean() lbessard@47: self.ResetView() lbessard@47: etisserant@0: # Changes Viewer mode etisserant@0: def SetMode(self, mode): laurent@575: if self.Mode != mode or mode == MODE_SELECTION: laurent@575: if self.Mode == MODE_MOTION: laurent@586: wx.CallAfter(self.Editor.SetCursor, wx.NullCursor) 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 laurent@575: if self.Mode == MODE_MOTION: laurent@586: wx.CallAfter(self.Editor.SetCursor, wx.StockCursor(wx.CURSOR_HAND)) laurent@575: self.SavedMode = True laurent@575: 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() laurent@667: if self.ParentWindow: laurent@667: self.ParentWindow.RefreshTitle() laurent@667: self.ParentWindow.RefreshFileMenu() Laurent@745: self.ParentWindow.RefreshEditMenu() laurent@667: laurent@667: def StartBuffering(self): laurent@667: if not self.Buffering: laurent@667: self.Buffering = True laurent@667: self.Controler.StartBuffering() laurent@667: if self.ParentWindow: laurent@667: self.ParentWindow.RefreshTitle() laurent@667: self.ParentWindow.RefreshFileMenu() laurent@667: self.ParentWindow.RefreshEditMenu() laurent@667: laurent@667: def ResetBuffer(self): laurent@667: if self.Buffering: laurent@667: self.Controler.EndBuffering() laurent@667: self.Buffering = False laurent@586: laurent@586: def GetBufferState(self): laurent@586: if not self.Debug: laurent@586: return self.Controler.GetBufferState() laurent@586: return False, False laurent@586: laurent@586: def Undo(self): laurent@586: if not self.Debug: laurent@586: self.Controler.LoadPrevious() laurent@586: self.ParentWindow.CloseTabsWithoutModel() laurent@586: laurent@586: def Redo(self): laurent@586: if not self.Debug: laurent@586: self.Controler.LoadNext() laurent@586: self.ParentWindow.CloseTabsWithoutModel() laurent@654: laurent@586: def HasNoModel(self): laurent@586: if not self.Debug: laurent@586: return self.Controler.GetEditedElement(self.TagName) is None laurent@586: return False laurent@586: lbessard@145: # Refresh the current scaling lbessard@145: def RefreshScaling(self, refresh=True): lbessard@249: properties = self.Controler.GetProjectProperties(self.Debug) lbessard@145: scaling = properties["scaling"][self.CurrentLanguage] laurent@380: if scaling[0] != 0 and scaling[1] != 0: lbessard@145: self.Scaling = scaling lbessard@145: if self.DrawGrid: lbessard@323: width = max(2, int(scaling[0] * self.ViewScale[0])) lbessard@323: height = max(2, int(scaling[1] * self.ViewScale[1])) lbessard@323: bitmap = wx.EmptyBitmap(width, height) lbessard@145: dc = wx.MemoryDC(bitmap) laurent@586: dc.SetBackground(wx.Brush(self.Editor.GetBackgroundColour())) lbessard@145: dc.Clear() laurent@563: dc.SetPen(MiterPen(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) laurent@563: self.PagePen = MiterPen(wx.Colour(180, 180, 180)) lbessard@213: else: lbessard@213: self.PageSize = None lbessard@213: self.PagePen = wx.TRANSPARENT_PEN lbessard@145: if refresh: greg@363: self.RefreshVisibleElements() lbessard@155: self.Refresh(False) lbessard@145: lbessard@145: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Refresh functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: laurent@634: def ElementNeedRefresh(self, element): laurent@634: self.ElementRefreshList_lock.acquire() laurent@634: self.ElementRefreshList.append(element) laurent@634: self.ElementRefreshList_lock.release() laurent@471: greg@361: def RefreshNewData(self): laurent@634: refresh_rect = None laurent@634: self.ElementRefreshList_lock.acquire() laurent@634: for element in self.ElementRefreshList: laurent@634: if refresh_rect is None: laurent@634: refresh_rect = element.GetRedrawRect() laurent@634: else: laurent@634: refresh_rect.Union(element.GetRedrawRect()) laurent@634: self.ElementRefreshList = [] laurent@634: self.ElementRefreshList_lock.release() laurent@634: laurent@634: if refresh_rect is not None: greg@361: self.RefreshRect(self.GetScrolledRect(refresh_rect), False) greg@375: else: greg@375: DebugViewer.RefreshNewData(self) greg@372: etisserant@0: # Refresh Viewer elements laurent@586: def RefreshView(self, variablepanel=True, selection=None): laurent@586: EditorPanel.RefreshView(self, variablepanel) laurent@586: laurent@641: if self.ToolTipElement is not None: laurent@641: self.ToolTipElement.ClearToolTip() laurent@641: self.ToolTipElement = None laurent@641: greg@372: self.Inhibit(True) etisserant@0: self.current_id = 0 etisserant@0: # Start by reseting Viewer lbessard@249: self.Flush() etisserant@0: self.ResetView() laurent@667: self.ResetBuffer() lbessard@249: instance = {} etisserant@0: # List of ids of already loaded blocks etisserant@0: ids = [] etisserant@0: # Load Blocks until they are all loaded lbessard@249: while instance is not None: lbessard@249: instance = self.Controler.GetEditedElementInstanceInfos(self.TagName, exclude = ids, debug = self.Debug) lbessard@249: if instance is not None: laurent@383: self.loadInstance(instance, ids, selection) lbessard@42: self.RefreshScrollBars() lbessard@121: lbessard@121: for wire in self.Wires: lbessard@121: if not wire.IsConnectedCompatible(): lbessard@343: wire.SetValid(False) lbessard@249: if self.Debug: laurent@383: iec_path = self.GetElementIECPath(wire) b@432: if iec_path is None: b@432: block = wire.EndConnected.GetParentBlock() b@432: if isinstance(block, LD_PowerRail): b@432: wire.SetValue(True) b@432: elif self.AddDataConsumer(iec_path.upper(), wire) is None: lbessard@253: wire.SetValue("undefined") lbessard@249: lbessard@249: if self.Debug: laurent@383: for block in self.Blocks.itervalues(): lbessard@253: block.SpreadCurrent() Laurent@697: iec_path = self.GetElementIECPath(block) Laurent@697: if iec_path is not None: Laurent@697: self.AddDataConsumer(iec_path.upper(), block) greg@372: greg@372: self.Inhibit(False) lbessard@249: self.RefreshVisibleElements() laurent@566: self.ShowHighlights() lbessard@155: self.Refresh(False) etisserant@0: laurent@477: def GetPreviousSteps(self, connectors): lbessard@253: steps = [] laurent@477: for connector in connectors: laurent@477: for wire, handle in connector.GetWires(): laurent@477: previous = wire.GetOtherConnected(connector).GetParentBlock() laurent@477: if isinstance(previous, SFC_Step): laurent@477: steps.append(previous.GetName()) laurent@477: elif isinstance(previous, SFC_Divergence) and previous.GetType() in [SIMULTANEOUS_CONVERGENCE, SELECTION_DIVERGENCE]: laurent@477: connectors = previous.GetConnectors() laurent@477: steps.extend(self.GetPreviousSteps(connectors["inputs"])) lbessard@253: return steps lbessard@253: laurent@477: def GetNextSteps(self, connectors): lbessard@253: steps = [] laurent@477: for connector in connectors: laurent@477: for wire, handle in connector.GetWires(): laurent@477: next = wire.GetOtherConnected(connector).GetParentBlock() laurent@477: if isinstance(next, SFC_Step): laurent@477: steps.append(next.GetName()) laurent@477: elif isinstance(next, SFC_Jump): laurent@477: steps.append(next.GetTarget()) laurent@477: elif isinstance(next, SFC_Divergence) and next.GetType() in [SIMULTANEOUS_DIVERGENCE, SELECTION_CONVERGENCE]: laurent@477: connectors = next.GetConnectors() laurent@477: steps.extend(self.GetNextSteps(connectors["outputs"])) lbessard@253: return steps lbessard@253: 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@319: def RefreshScrollBars(self, width_incr=0, height_incr=0): lbessard@213: xstart, ystart = self.GetViewStart() laurent@586: window_size = self.Editor.GetClientSize() lbessard@213: maxx, maxy = self.GetMaxSize() lbessard@323: maxx = max(maxx + WINDOW_BORDER, (xstart * SCROLLBAR_UNIT + window_size[0]) / self.ViewScale[0]) lbessard@323: maxy = max(maxy + WINDOW_BORDER, (ystart * SCROLLBAR_UNIT + window_size[1]) / self.ViewScale[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@323: maxx = int(maxx * self.ViewScale[0]) lbessard@323: maxy = int(maxy * self.ViewScale[1]) laurent@586: self.Editor.SetScrollbars(SCROLLBAR_UNIT, SCROLLBAR_UNIT, lbessard@319: round(maxx / SCROLLBAR_UNIT) + width_incr, round(maxy / SCROLLBAR_UNIT) + height_incr, lbessard@213: xstart, ystart, True) laurent@383: Laurent@738: def EnsureVisible(self, block): Laurent@738: xstart, ystart = self.GetViewStart() Laurent@738: window_size = self.Editor.GetClientSize() Laurent@738: block_bbx = block.GetBoundingBox() Laurent@738: Laurent@738: screen_minx, screen_miny = xstart * SCROLLBAR_UNIT, ystart * SCROLLBAR_UNIT Laurent@738: screen_maxx, screen_maxy = screen_minx + window_size[0], screen_miny + window_size[1] Laurent@738: block_minx = int(block_bbx.x * self.ViewScale[0]) Laurent@738: block_miny = int(block_bbx.y * self.ViewScale[1]) Laurent@738: block_maxx = int(round((block_bbx.x + block_bbx.width) * self.ViewScale[0])) Laurent@738: block_maxy = int(round((block_bbx.y + block_bbx.height) * self.ViewScale[1])) Laurent@738: Laurent@738: xpos, ypos = xstart, ystart Laurent@738: if block_minx < screen_minx and block_maxx < screen_maxx: Laurent@738: xpos -= (screen_minx - block_minx) / SCROLLBAR_UNIT + 1 Laurent@738: elif block_maxx > screen_maxx and block_minx > screen_minx: Laurent@738: xpos += (block_maxx - screen_maxx) / SCROLLBAR_UNIT + 1 Laurent@738: if block_miny < screen_miny and block_maxy < screen_maxy: Laurent@738: ypos -= (screen_miny - block_miny) / SCROLLBAR_UNIT + 1 Laurent@738: elif block_maxy > screen_maxy and block_miny > screen_miny: Laurent@738: ypos += (block_maxy - screen_maxy) / SCROLLBAR_UNIT + 1 Laurent@738: self.Scroll(xpos, ypos) Laurent@738: laurent@383: def SelectInGroup(self, element): laurent@383: element.SetSelected(True) laurent@383: if self.SelectedElement is None: laurent@383: self.SelectedElement = element laurent@383: elif isinstance(self.SelectedElement, Graphic_Group): laurent@383: self.SelectedElement.SelectElement(element) laurent@383: else: laurent@383: group = Graphic_Group(self) laurent@383: group.SelectElement(self.SelectedElement) laurent@383: group.SelectElement(element) laurent@383: self.SelectedElement = group lbessard@95: etisserant@0: # Load instance from given informations laurent@383: def loadInstance(self, instance, ids, selection): etisserant@0: ids.append(instance["id"]) laurent@383: self.current_id = max(self.current_id, instance["id"]) laurent@383: creation_function = ElementCreationFunctions.get(instance["type"], None) laurent@383: connectors = {"inputs" : [], "outputs" : []} laurent@383: specific_values = instance["specific_values"] laurent@383: if creation_function is not None: laurent@383: element = creation_function(self, instance["id"], specific_values) laurent@383: if isinstance(element, SFC_Step): laurent@383: if len(instance["inputs"]) > 0: laurent@383: element.AddInput() laurent@400: else: laurent@400: element.RemoveInput() laurent@383: if len(instance["outputs"]) > 0: laurent@383: element.AddOutput() laurent@383: if isinstance(element, SFC_Transition) and specific_values["condition_type"] == "connection": laurent@383: connector = element.GetConditionConnector() laurent@383: self.CreateWires(connector, id, specific_values["connection"]["links"], ids, selection) etisserant@0: else: lbessard@269: executionControl = False laurent@383: for input in instance["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")) laurent@383: for output in instance["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")) greg@290: if len(connectors["inputs"]) > 0 and connectors["inputs"][0][0] == "EN": laurent@383: connectors["inputs"].pop(0) laurent@383: executionControl = True greg@290: if len(connectors["outputs"]) > 0 and connectors["outputs"][0][0] == "ENO": lbessard@269: connectors["outputs"].pop(0) lbessard@269: executionControl = True laurent@383: if specific_values["name"] is None: laurent@383: specific_values["name"] = "" laurent@383: element = FBD_Block(self, instance["type"], specific_values["name"], laurent@383: instance["id"], len(connectors["inputs"]), laurent@383: connectors=connectors, executionControl=executionControl, laurent@383: executionOrder=specific_values["executionOrder"]) laurent@383: if isinstance(element, Comment): laurent@383: self.AddComment(element) laurent@383: else: laurent@383: self.AddBlock(element) laurent@383: connectors = element.GetConnectors() Laurent@736: if isinstance(element, SFC_Divergence): Laurent@736: element.SetPosition(instance["x"], instance["y"]) Laurent@736: element.SetSize(instance["width"], instance["height"]) laurent@383: for i, input_connector in enumerate(instance["inputs"]): laurent@383: if i < len(connectors["inputs"]): laurent@383: connector = connectors["inputs"][i] laurent@383: connector.SetPosition(wx.Point(*input_connector["position"])) laurent@383: if input_connector.get("negated", False): laurent@383: connector.SetNegated(True) laurent@383: if input_connector.get("edge", "none") != "none": laurent@383: connector.SetEdge(input_connector["edge"]) laurent@383: self.CreateWires(connector, instance["id"], input_connector["links"], ids, selection) laurent@383: for i, output_connector in enumerate(instance["outputs"]): laurent@383: if i < len(connectors["outputs"]): laurent@383: connector = connectors["outputs"][i] laurent@383: if output_connector.get("negated", False): laurent@383: connector.SetNegated(True) laurent@383: if output_connector.get("edge", "none") != "none": laurent@383: connector.SetEdge(output_connector["edge"]) laurent@383: connector.SetPosition(wx.Point(*output_connector["position"])) Laurent@736: if not isinstance(element, SFC_Divergence): Laurent@736: element.SetPosition(instance["x"], instance["y"]) Laurent@736: element.SetSize(instance["width"], instance["height"]) laurent@383: if selection is not None and selection[0].get(instance["id"], False): laurent@383: self.SelectInGroup(element) laurent@383: laurent@383: def CreateWires(self, start_connector, id, links, ids, selection=None): etisserant@0: for link in links: etisserant@0: refLocalId = link["refLocalId"] lbessard@249: if refLocalId is not None: etisserant@0: if refLocalId not in ids: lbessard@249: new_instance = self.Controler.GetEditedElementInstanceInfos(self.TagName, refLocalId, debug = self.Debug) lbessard@249: if new_instance is not None: laurent@383: self.loadInstance(new_instance, ids, selection) etisserant@0: connected = self.FindElementById(refLocalId) lbessard@249: if connected is not None: etisserant@0: points = link["points"] lbessard@64: end_connector = connected.GetConnector(wx.Point(points[-1][0], points[-1][1]), link["formalParameter"]) lbessard@249: if end_connector is not None: 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) laurent@383: if selection is not None and (\ laurent@383: selection[1].get((id, refLocalId), False) or \ laurent@383: selection[1].get((refLocalId, id), False)): laurent@383: self.SelectInGroup(wire) etisserant@0: lbessard@231: def IsOfType(self, type, reference): lbessard@249: return self.Controler.IsOfType(type, reference, self.Debug) lbessard@231: lbessard@231: def IsEndType(self, type): lbessard@231: return self.Controler.IsEndType(type) lbessard@231: lbessard@231: def GetBlockType(self, type, inputs = None): lbessard@249: return self.Controler.GetBlockType(type, inputs, self.Debug) lbessard@231: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Search Element functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: laurent@563: def FindBlock(self, event): laurent@563: dc = self.GetLogicalDC() laurent@563: pos = event.GetLogicalPosition(dc) laurent@383: for block in self.Blocks.itervalues(): laurent@563: if block.HitTest(pos) or block.TestHandle(event) != (0, 0): etisserant@0: return block etisserant@0: return None etisserant@0: laurent@563: def FindWire(self, event): laurent@563: dc = self.GetLogicalDC() laurent@563: pos = event.GetLogicalPosition(dc) etisserant@0: for wire in self.Wires: laurent@563: if wire.HitTest(pos) or wire.TestHandle(event) != (0, 0): etisserant@0: return wire etisserant@0: return None etisserant@0: laurent@633: def FindElement(self, event, exclude_group = False, connectors = True): laurent@563: dc = self.GetLogicalDC() laurent@563: pos = event.GetLogicalPosition(dc) etisserant@0: if self.SelectedElement and not (exclude_group and isinstance(self.SelectedElement, Graphic_Group)): laurent@633: if self.SelectedElement.HitTest(pos, connectors) or self.SelectedElement.TestHandle(event) != (0, 0): etisserant@0: return self.SelectedElement lbessard@42: for element in self.GetElements(): laurent@633: if element.HitTest(pos, connectors) or element.TestHandle(event) != (0, 0): etisserant@0: return element etisserant@0: return None etisserant@0: lbessard@249: def FindBlockConnector(self, pos, direction = None, exclude = None): laurent@383: for block in self.Blocks.itervalues(): lbessard@243: result = block.TestConnector(pos, direction, exclude) etisserant@0: if result: etisserant@0: return result etisserant@0: return None etisserant@0: etisserant@0: def FindElementById(self, id): laurent@383: block = self.Blocks.get(id, None) laurent@383: if block is not None: laurent@383: return block laurent@383: comment = self.Comments.get(id, None) laurent@383: if comment is not None: laurent@383: return comment 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: lbessard@331: def SelectAll(self): lbessard@331: if self.SelectedElement is not None: lbessard@331: self.SelectedElement.SetSelected(False) lbessard@331: self.SelectedElement = Graphic_Group(self) lbessard@331: for element in self.GetElements(): lbessard@331: self.SelectedElement.SelectElement(element) lbessard@331: self.SelectedElement.SetSelected(True) lbessard@331: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Popup menu functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: laurent@479: def GetForceVariableMenuFunction(self, iec_path, element): laurent@469: iec_type = self.GetDataType(iec_path) laurent@469: def ForceVariableFunction(event): laurent@469: if iec_type is not None: laurent@479: dialog = ForceVariableDialog(self.ParentWindow, iec_type, str(element.GetValue())) laurent@469: if dialog.ShowModal() == wx.ID_OK: laurent@479: self.ParentWindow.AddDebugVariable(iec_path) laurent@469: self.ForceDataValue(iec_path, dialog.GetValue()) laurent@469: return ForceVariableFunction laurent@469: laurent@469: def GetReleaseVariableMenuFunction(self, iec_path): laurent@469: def ReleaseVariableFunction(event): laurent@469: self.ReleaseDataValue(iec_path) laurent@469: return ReleaseVariableFunction laurent@469: laurent@467: def PopupForceMenu(self): laurent@467: iec_path = self.GetElementIECPath(self.SelectedElement) laurent@467: if iec_path is not None: laurent@467: menu = wx.Menu(title='') laurent@467: new_id = wx.NewId() laurent@467: AppendMenu(menu, help='', id=new_id, kind=wx.ITEM_NORMAL, text=_("Force value")) laurent@479: self.Bind(wx.EVT_MENU, self.GetForceVariableMenuFunction(iec_path.upper(), self.SelectedElement), id=new_id) laurent@467: new_id = wx.NewId() laurent@467: AppendMenu(menu, help='', id=new_id, kind=wx.ITEM_NORMAL, text=_("Release value")) laurent@469: self.Bind(wx.EVT_MENU, self.GetReleaseVariableMenuFunction(iec_path.upper()), id=new_id) laurent@467: if self.SelectedElement.IsForced(): laurent@467: menu.Enable(new_id, True) laurent@467: else: laurent@467: menu.Enable(new_id, False) laurent@586: self.Editor.PopupMenu(menu) laurent@666: menu.Destroy() laurent@467: etisserant@0: def PopupBlockMenu(self, connector = None): laurent@666: menu = wx.Menu(title='') lbessard@102: if connector is not None and connector.IsCompatible("BOOL"): laurent@666: self.AddBlockPinMenuItems(menu, connector) lbessard@102: else: laurent@666: edit = self.SelectedElement.GetType() in self.Controler.GetProjectPouNames(self.Debug) laurent@666: self.AddDefaultMenuItems(menu, block=True, edit=edit) laurent@666: self.Editor.PopupMenu(menu) laurent@666: menu.Destroy() etisserant@0: lbessard@235: def PopupWireMenu(self, delete=True): laurent@666: menu = wx.Menu(title='') laurent@666: self.AddWireMenuItems(menu, delete) laurent@666: menu.AppendSeparator() laurent@666: self.AddDefaultMenuItems(menu, block=True) laurent@666: self.Editor.PopupMenu(menu) laurent@666: menu.Destroy() laurent@666: etisserant@0: def PopupDivergenceMenu(self, connector): laurent@666: menu = wx.Menu(title='') laurent@666: self.AddDivergenceMenuItems(menu, connector) laurent@666: menu.AppendSeparator() laurent@666: self.AddDefaultMenuItems(menu, block=True) laurent@666: self.Editor.PopupMenu(menu) laurent@666: menu.Destroy() etisserant@0: lbessard@138: def PopupGroupMenu(self): laurent@666: menu = wx.Menu(title='') laurent@666: align_menu = wx.Menu(title='') laurent@666: self.AddAlignmentMenuItems(align_menu) laurent@670: menu.AppendMenu(-1, _(u'Alignment'), align_menu) laurent@666: menu.AppendSeparator() laurent@666: self.AddDefaultMenuItems(menu, block=True) laurent@666: self.Editor.PopupMenu(menu) laurent@666: menu.Destroy() laurent@666: laurent@666: def PopupDefaultMenu(self, block=True): laurent@666: menu = wx.Menu(title='') laurent@666: self.AddDefaultMenuItems(menu, block=block) laurent@666: self.Editor.PopupMenu(menu) laurent@666: menu.Destroy() etisserant@0: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Menu items functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: lbessard@138: def OnAlignLeftMenu(self, event): lbessard@249: if self.SelectedElement is not None 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: lbessard@138: def OnAlignCenterMenu(self, event): lbessard@249: if self.SelectedElement is not None 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: lbessard@138: def OnAlignRightMenu(self, event): lbessard@249: if self.SelectedElement is not None 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: lbessard@138: def OnAlignTopMenu(self, event): lbessard@249: if self.SelectedElement is not None 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: lbessard@138: def OnAlignMiddleMenu(self, event): lbessard@249: if self.SelectedElement is not None 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: lbessard@138: def OnAlignBottomMenu(self, event): lbessard@249: if self.SelectedElement is not None 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: etisserant@0: def OnNoModifierMenu(self, event): lbessard@249: if self.SelectedElement is not None and self.IsBlock(self.SelectedElement): etisserant@0: self.SelectedElement.SetConnectorNegated(False) lbessard@206: self.SelectedElement.Refresh() lbessard@80: self.RefreshBuffer() etisserant@0: etisserant@0: def OnNegatedMenu(self, event): lbessard@249: if self.SelectedElement is not None and self.IsBlock(self.SelectedElement): etisserant@0: self.SelectedElement.SetConnectorNegated(True) lbessard@206: self.SelectedElement.Refresh() lbessard@80: self.RefreshBuffer() etisserant@0: etisserant@0: def OnRisingEdgeMenu(self, event): lbessard@249: if self.SelectedElement is not None and self.IsBlock(self.SelectedElement): etisserant@0: self.SelectedElement.SetConnectorEdge("rising") lbessard@206: self.SelectedElement.Refresh() lbessard@80: self.RefreshBuffer() etisserant@0: etisserant@0: def OnFallingEdgeMenu(self, event): lbessard@249: if self.SelectedElement is not None and self.IsBlock(self.SelectedElement): etisserant@0: self.SelectedElement.SetConnectorEdge("falling") lbessard@206: self.SelectedElement.Refresh() lbessard@80: self.RefreshBuffer() etisserant@0: etisserant@0: def OnAddSegmentMenu(self, event): lbessard@249: if self.SelectedElement is not None and self.IsWire(self.SelectedElement): etisserant@0: self.SelectedElement.AddSegment() lbessard@206: self.SelectedElement.Refresh() etisserant@0: etisserant@0: def OnDeleteSegmentMenu(self, event): lbessard@249: if self.SelectedElement is not None and self.IsWire(self.SelectedElement): etisserant@0: self.SelectedElement.DeleteSegment() lbessard@206: self.SelectedElement.Refresh() etisserant@0: etisserant@0: def OnAddBranchMenu(self, event): lbessard@249: if self.SelectedElement is not None and self.IsBlock(self.SelectedElement): etisserant@0: self.AddDivergenceBranch(self.SelectedElement) etisserant@0: etisserant@0: def OnDeleteBranchMenu(self, event): lbessard@249: if self.SelectedElement is not None and self.IsBlock(self.SelectedElement): etisserant@0: self.RemoveDivergenceBranch(self.SelectedElement) etisserant@0: lbessard@102: def OnEditBlockMenu(self, event): lbessard@249: if self.SelectedElement is not None: lbessard@138: self.ParentWindow.EditProjectElement(ITEM_POU, "P::%s"%self.SelectedElement.GetType()) lbessard@102: etisserant@0: def OnDeleteMenu(self, event): lbessard@249: if self.SelectedElement is not None: etisserant@0: self.SelectedElement.Delete() etisserant@0: self.SelectedElement = None lbessard@80: self.RefreshBuffer() lbessard@206: self.Refresh(False) 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: laurent@666: def GetAddMenuCallBack(self, func, *args): laurent@666: def AddMenuCallBack(event): laurent@666: wx.CallAfter(func, self.rubberBand.GetCurrentExtent(), *args) laurent@666: return AddMenuCallBack laurent@666: laurent@666: def GetClipboardCallBack(self, func): laurent@666: def ClipboardCallback(event): laurent@666: wx.CallAfter(func) laurent@666: return ClipboardCallback laurent@666: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Mouse event functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: laurent@633: def OnViewerMouseEvent(self, event): laurent@633: if not event.Entering(): laurent@667: self.ResetBuffer() laurent@633: element = None laurent@643: if not event.Leaving() and not event.LeftUp() and not event.LeftDClick(): laurent@633: element = self.FindElement(event, True, False) laurent@633: if self.ToolTipElement is not None: laurent@633: self.ToolTipElement.ClearToolTip() laurent@633: self.ToolTipElement = element laurent@633: if self.ToolTipElement is not None: laurent@633: tooltip_pos = self.Editor.ClientToScreen(event.GetPosition()) laurent@633: tooltip_pos.x += 10 laurent@633: tooltip_pos.y += 10 laurent@633: self.ToolTipElement.CreateToolTip(tooltip_pos) laurent@633: event.Skip() laurent@641: etisserant@0: def OnViewerLeftDown(self, event): laurent@666: self.Editor.CaptureMouse() lbessard@27: if self.Mode == MODE_SELECTION: lbessard@27: dc = self.GetLogicalDC() lbessard@27: pos = event.GetLogicalPosition(dc) lbessard@331: if event.ShiftDown() and not event.ControlDown() and self.SelectedElement is not None: laurent@586: element = self.FindElement(event, True) lbessard@249: if element is not None: lbessard@27: if isinstance(self.SelectedElement, Graphic_Group): lbessard@27: self.SelectedElement.SetSelected(False) lbessard@27: self.SelectedElement.SelectElement(element) lbessard@249: elif self.SelectedElement is not None: 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) laurent@620: else: laurent@620: self.rubberBand.Reset() laurent@620: self.rubberBand.OnLeftDown(event, dc, self.Scaling) lbessard@27: else: laurent@563: element = self.FindElement(event) laurent@563: if not self.Debug and (element is None or element.TestHandle(event) == (0, 0)): lbessard@145: connector = self.FindBlockConnector(pos) lbessard@145: else: lbessard@145: connector = None lbessard@249: if not self.Debug and self.DrawingWire: lbessard@138: self.DrawingWire = False lbessard@174: if self.SelectedElement is not None: laurent@563: if element is None or element.TestHandle(event) == (0, 0): lbessard@243: connector = self.FindBlockConnector(pos, self.SelectedElement.GetConnectionDirection()) lbessard@249: if connector is not None: lbessard@174: event.Dragging = lambda : True laurent@563: self.SelectedElement.OnMotion(event, dc, 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) laurent@652: elif not self.Debug and connector is not None and not event.ControlDown(): lbessard@138: self.DrawingWire = True laurent@563: scaled_pos = GetScaledEventPosition(event, dc, 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@327: wire.ProcessDragging(0, 0, event, None) lbessard@138: wire.Handle = (HANDLE_POINT, 1) lbessard@138: self.AddWire(wire) lbessard@249: if self.SelectedElement is not None: lbessard@138: self.SelectedElement.SetSelected(False) lbessard@138: self.SelectedElement = wire laurent@633: if self.HighlightedElement is not None: laurent@633: self.HighlightedElement.SetHighlighted(False) laurent@633: self.HighlightedElement = wire lbessard@259: self.RefreshVisibleElements() laurent@633: self.SelectedElement.SetHighlighted(True) lbessard@27: else: lbessard@249: if self.SelectedElement is not None and self.SelectedElement != element: lbessard@138: self.SelectedElement.SetSelected(False) lbessard@138: self.SelectedElement = None lbessard@249: if element is not None: lbessard@138: self.SelectedElement = element lbessard@249: if self.Debug: lbessard@343: self.StartMousePos = event.GetPosition() lbessard@249: Graphic_Element.OnLeftDown(self.SelectedElement, event, dc, self.Scaling) lbessard@249: else: lbessard@249: 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) laurent@575: elif self.Mode == MODE_MOTION: laurent@575: self.StartMousePos = event.GetPosition() laurent@575: self.StartScreenPos = self.GetScrollPos(wx.HORIZONTAL), self.GetScrollPos(wx.VERTICAL) etisserant@0: event.Skip() etisserant@0: etisserant@0: def OnViewerLeftUp(self, event): lbessard@343: self.StartMousePos = None lbessard@27: if self.rubberBand.IsShown(): lbessard@27: if self.Mode == MODE_SELECTION: laurent@620: new_elements = self.SearchElements(self.rubberBand.GetCurrentExtent()) lbessard@27: self.rubberBand.OnLeftUp(event, self.GetLogicalDC(), self.Scaling) laurent@620: if event.ShiftDown() and self.SelectedElement is not None: laurent@620: if isinstance(self.SelectedElement, Graphic_Group): laurent@620: elements = self.SelectedElement.GetElements() laurent@620: else: laurent@620: elements = [self.SelectedElement] laurent@620: for element in elements: laurent@620: if element not in new_elements: laurent@620: new_elements.append(element) laurent@620: if len(new_elements) == 1: laurent@620: self.SelectedElement = new_elements[0] lbessard@80: self.SelectedElement.SetSelected(True) laurent@620: elif len(new_elements) > 1: lbessard@27: self.SelectedElement = Graphic_Group(self) laurent@620: self.SelectedElement.SetElements(new_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@249: elif self.Mode == MODE_SELECTION and self.SelectedElement is not None: lbessard@138: dc = self.GetLogicalDC() lbessard@249: if not self.Debug and self.DrawingWire: lbessard@138: pos = event.GetLogicalPosition(dc) lbessard@243: connector = self.FindBlockConnector(pos, self.SelectedElement.GetConnectionDirection()) 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@138: self.SelectedElement.SetSelected(True) laurent@633: self.SelectedElement.HighlightPoint(pos) 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() Laurent@735: wire = self.SelectedElement Laurent@735: self.SelectedElement = self.SelectedElement.StartConnected.GetParentBlock() Laurent@735: self.SelectedElement.SetSelected(True) Laurent@735: rect.Union(self.SelectedElement.GetRedrawRect()) Laurent@735: wire.Delete() lbessard@155: self.RefreshRect(self.GetScrolledRect(rect), False) lbessard@138: else: lbessard@249: if self.Debug: lbessard@249: Graphic_Element.OnLeftUp(self.SelectedElement, event, dc, self.Scaling) lbessard@249: else: lbessard@249: self.SelectedElement.OnLeftUp(event, dc, self.Scaling) laurent@381: wx.CallAfter(self.SetCurrentCursor, 0) laurent@575: elif self.Mode == MODE_MOTION: laurent@575: self.StartMousePos = None laurent@575: self.StartScreenPos = None lbessard@106: if self.Mode != MODE_SELECTION and not self.SavedMode: lbessard@90: wx.CallAfter(self.ParentWindow.ResetCurrentMode) laurent@666: if self.Editor.HasCapture(): laurent@666: self.Editor.ReleaseMouse() etisserant@0: event.Skip() etisserant@0: lbessard@319: def OnViewerMiddleDown(self, event): laurent@666: self.Editor.CaptureMouse() lbessard@319: self.StartMousePos = event.GetPosition() lbessard@319: self.StartScreenPos = self.GetScrollPos(wx.HORIZONTAL), self.GetScrollPos(wx.VERTICAL) lbessard@319: event.Skip() lbessard@319: lbessard@319: def OnViewerMiddleUp(self, event): lbessard@319: self.StartMousePos = None lbessard@319: self.StartScreenPos = None laurent@666: if self.Editor.HasCapture(): laurent@666: self.Editor.ReleaseMouse() lbessard@319: event.Skip() lbessard@319: lbessard@145: def OnViewerRightDown(self, event): laurent@666: self.Editor.CaptureMouse() lbessard@145: if self.Mode == MODE_SELECTION: laurent@563: element = self.FindElement(event) lbessard@249: if self.SelectedElement is not None and self.SelectedElement != element: lbessard@145: self.SelectedElement.SetSelected(False) lbessard@145: self.SelectedElement = None Laurent@735: if element is not None: lbessard@145: self.SelectedElement = element lbessard@249: if self.Debug: laurent@563: Graphic_Element.OnRightDown(self.SelectedElement, event, self.GetLogicalDC(), self.Scaling) lbessard@249: else: laurent@563: self.SelectedElement.OnRightDown(event, self.GetLogicalDC(), self.Scaling) lbessard@145: self.SelectedElement.Refresh() lbessard@145: event.Skip() lbessard@145: etisserant@0: def OnViewerRightUp(self, event): laurent@666: dc = self.GetLogicalDC() laurent@666: self.rubberBand.Reset() laurent@666: self.rubberBand.OnLeftDown(event, dc, self.Scaling) laurent@666: self.rubberBand.OnLeftUp(event, dc, self.Scaling) lbessard@249: if self.SelectedElement is not None: lbessard@249: if self.Debug: laurent@563: Graphic_Element.OnRightUp(self.SelectedElement, event, self.GetLogicalDC(), self.Scaling) lbessard@249: else: laurent@563: self.SelectedElement.OnRightUp(event, self.GetLogicalDC(), self.Scaling) laurent@381: wx.CallAfter(self.SetCurrentCursor, 0) lbessard@249: elif not self.Debug: lbessard@118: self.PopupDefaultMenu(False) laurent@666: if self.Editor.HasCapture(): laurent@666: self.Editor.ReleaseMouse() etisserant@0: event.Skip() etisserant@0: etisserant@0: def OnViewerLeftDClick(self, event): Laurent@735: element = self.FindElement(event, connectors=False) Laurent@735: if self.Mode == MODE_SELECTION and element is not None: Laurent@735: if self.SelectedElement is not None and self.SelectedElement != element: Laurent@735: self.SelectedElement.SetSelected(False) Laurent@735: if self.HighlightedElement is not None and self.HighlightedElement != element: Laurent@735: self.HighlightedElement.SetHighlighted(False) Laurent@735: Laurent@735: self.SelectedElement = element Laurent@735: self.HighlightedElement = element Laurent@735: self.SelectedElement.SetHighlighted(True) Laurent@735: lbessard@249: if self.Debug: Laurent@694: if self.IsBlock(self.SelectedElement): Laurent@694: instance_type = self.SelectedElement.GetType() Laurent@694: pou_type = { Laurent@694: "program": ITEM_PROGRAM, Laurent@694: "functionBlock": ITEM_FUNCTIONBLOCK, Laurent@694: }.get(self.Controler.GetPouType(instance_type)) Laurent@694: if pou_type is not None and instance_type in self.Controler.GetProjectPouNames(self.Debug): Laurent@694: self.ParentWindow.OpenDebugViewer(pou_type, Laurent@697: "%s.%s"%(self.InstancePath, self.SelectedElement.GetName()), Laurent@694: self.Controler.ComputePouName(instance_type)) Laurent@694: else: Laurent@694: iec_path = self.GetElementIECPath(self.SelectedElement) Laurent@694: if iec_path is not None: Laurent@694: if isinstance(self.SelectedElement, Wire): Laurent@694: if self.SelectedElement.EndConnected is not None: Laurent@694: var_type = self.SelectedElement.EndConnected.GetType() Laurent@694: if self.Controler.IsOfType(var_type, "ANY_NUM", self.Debug) or\ Laurent@694: self.Controler.IsOfType(var_type, "ANY_BIT", self.Debug): Laurent@694: self.ParentWindow.OpenGraphicViewer(iec_path) Laurent@694: else: Laurent@694: self.ParentWindow.OpenGraphicViewer(iec_path) lbessard@331: elif event.ControlDown() and not event.ShiftDown(): Laurent@694: instance_type = self.SelectedElement.GetType() Laurent@694: if self.IsBlock(self.SelectedElement) and instance_type in self.Controler.GetProjectPouNames(self.Debug): Laurent@694: self.ParentWindow.EditProjectElement(ITEM_POU, Laurent@694: self.Controler.ComputePouName(instance_type)) lbessard@343: else: lbessard@343: self.SelectedElement.OnLeftDClick(event, self.GetLogicalDC(), self.Scaling) lbessard@331: elif event.ControlDown() and event.ShiftDown(): lbessard@331: movex, movey = self.SelectedElement.AdjustToScaling(self.Scaling) lbessard@331: self.SelectedElement.RefreshModel() lbessard@331: self.RefreshBuffer() lbessard@331: if movex != 0 or movey != 0: lbessard@331: self.RefreshRect(self.GetScrolledRect(self.SelectedElement.GetRedrawRect(movex, movey)), False) 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): laurent@666: if self.Editor.HasCapture() and not event.Dragging(): laurent@666: return lbessard@140: refresh = False lbessard@140: dc = self.GetLogicalDC() lbessard@140: pos = GetScaledEventPosition(event, dc, self.Scaling) laurent@575: if event.MiddleIsDown() or self.Mode == MODE_MOTION: lbessard@323: if self.StartMousePos is not None and self.StartScreenPos is not None: lbessard@323: new_pos = event.GetPosition() lbessard@323: xmax = self.GetScrollRange(wx.HORIZONTAL) - self.GetScrollThumb(wx.HORIZONTAL) lbessard@323: ymax = self.GetScrollRange(wx.VERTICAL) - self.GetScrollThumb(wx.VERTICAL) lbessard@331: scrollx = max(0, self.StartScreenPos[0] - (new_pos[0] - self.StartMousePos[0]) / SCROLLBAR_UNIT) lbessard@331: scrolly = max(0, self.StartScreenPos[1] - (new_pos[1] - self.StartMousePos[1]) / SCROLLBAR_UNIT) lbessard@323: if scrollx > xmax or scrolly > ymax: lbessard@323: self.RefreshScrollBars(max(0, scrollx - xmax), max(0, scrolly - ymax)) lbessard@323: self.Scroll(scrollx, scrolly) lbessard@323: else: lbessard@323: self.Scroll(scrollx, scrolly) lbessard@323: self.RefreshScrollBars() lbessard@323: self.RefreshVisibleElements() lbessard@323: else: lbessard@319: if not event.Dragging(): laurent@633: highlighted = self.FindElement(event, connectors=False) laurent@625: if self.HighlightedElement is not None and self.HighlightedElement != highlighted: laurent@625: self.HighlightedElement.SetHighlighted(False) laurent@625: self.HighlightedElement = None laurent@625: if highlighted is not None: laurent@633: if isinstance(highlighted, (Wire, Graphic_Group)): laurent@633: highlighted.HighlightPoint(pos) laurent@625: if self.HighlightedElement != highlighted: laurent@625: highlighted.SetHighlighted(True) lbessard@319: self.HighlightedElement = highlighted lbessard@319: if self.rubberBand.IsShown(): lbessard@319: self.rubberBand.OnMotion(event, dc, self.Scaling) lbessard@319: elif not self.Debug and self.Mode == MODE_SELECTION and self.SelectedElement is not None: lbessard@319: if self.DrawingWire: lbessard@319: connector = self.FindBlockConnector(pos, self.SelectedElement.GetConnectionDirection(), self.SelectedElement.EndConnected) lbessard@319: if not connector or self.SelectedElement.EndConnected == None: lbessard@319: self.SelectedElement.ResetPoints() lbessard@319: movex, movey = self.SelectedElement.OnMotion(event, dc, self.Scaling) lbessard@319: self.SelectedElement.GeneratePoints() lbessard@319: if movex != 0 or movey != 0: lbessard@319: self.RefreshRect(self.GetScrolledRect(self.SelectedElement.GetRedrawRect(movex, movey)), False) laurent@633: else: laurent@633: self.SelectedElement.HighlightPoint(pos) lbessard@319: 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@343: elif self.Debug and self.StartMousePos is not None and event.Dragging(): lbessard@343: pos = event.GetPosition() lbessard@343: if abs(self.StartMousePos.x - pos.x) > 5 or abs(self.StartMousePos.y - pos.y) > 5: lbessard@343: iec_path = self.GetElementIECPath(self.SelectedElement) lbessard@343: if iec_path is not None: lbessard@343: self.StartMousePos = None lbessard@343: if self.HighlightedElement is not None: lbessard@343: self.HighlightedElement.SetHighlighted(False) lbessard@343: self.HighlightedElement = None lbessard@343: data = wx.TextDataObject(str((iec_path, "debug"))) laurent@586: dragSource = wx.DropSource(self.Editor) lbessard@343: dragSource.SetData(data) lbessard@343: dragSource.DoDragDrop() lbessard@319: self.UpdateScrollPos(event) lbessard@140: event.Skip() lbessard@140: lbessard@140: def OnLeaveViewer(self, event): laurent@575: if self.StartScreenPos is None: laurent@575: self.StartMousePos = None lbessard@249: if self.SelectedElement is not None 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@249: if (event.Dragging() and self.SelectedElement is not None) or self.rubberBand.IsShown(): lbessard@27: position = event.GetPosition() lbessard@64: move_window = wx.Point() laurent@586: window_size = self.Editor.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@267: self.RefreshVisibleElements(xp = xstart + move_window.x, yp = ystart + move_window.y) lbessard@27: self.Scroll(xstart + move_window.x, ystart + move_window.y) laurent@586: self.RefreshScrollBars(move_window.x, move_window.y) etisserant@0: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Keyboard event functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: laurent@667: ARROW_KEY_MOVE = { laurent@667: wx.WXK_LEFT: (-1, 0), laurent@667: wx.WXK_RIGHT: (1, 0), laurent@667: wx.WXK_UP: (0, -1), laurent@667: wx.WXK_DOWN: (0, 1), laurent@667: } laurent@667: 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@249: if self.Scaling is not None: lbessard@27: scaling = self.Scaling lbessard@27: else: lbessard@27: scaling = (8, 8) lbessard@249: if not self.Debug and keycode == wx.WXK_DELETE and self.SelectedElement is not None: 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() laurent@381: wx.CallAfter(self.SetCurrentCursor, 0) lbessard@155: self.RefreshRect(self.GetScrolledRect(rect), False) lbessard@249: elif not self.Debug and keycode == wx.WXK_RETURN and self.SelectedElement is not None: lbessard@138: self.SelectedElement.OnLeftDClick(event, self.GetLogicalDC(), self.Scaling) laurent@667: elif self.ARROW_KEY_MOVE.has_key(keycode): laurent@667: move = self.ARROW_KEY_MOVE[keycode] lbessard@42: if event.ControlDown() and event.ShiftDown(): laurent@667: self.Scroll({-1: 0, 0: xpos, 1: xmax}[move[0]], laurent@667: {-1: 0, 0: ypos, 1: ymax}[move[1]]) lbessard@319: self.RefreshVisibleElements() lbessard@42: elif event.ControlDown(): laurent@667: self.Scroll(xpos + move[0], ypos + move[1]) lbessard@319: self.RefreshScrollBars() lbessard@319: self.RefreshVisibleElements() lbessard@249: elif not self.Debug and self.SelectedElement is not None: laurent@667: movex, movey = move laurent@667: if not event.AltDown() or event.ShiftDown(): laurent@667: movex *= scaling[0] laurent@667: movey *= scaling[1] laurent@667: if event.ShiftDown() and not event.AltDown(): laurent@667: movex *= 10 laurent@667: movey *= 10 laurent@667: self.SelectedElement.Move(movex, movey) laurent@667: self.StartBuffering() lbessard@67: self.SelectedElement.RefreshModel() lbessard@56: self.RefreshScrollBars() laurent@667: self.RefreshRect(self.GetScrolledRect(self.SelectedElement.GetRedrawRect(movex, movey)), False) lbessard@249: elif not self.Debug and 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@332: block = self.CopyBlock(self.SelectedElement, wx.Point(*self.SelectedElement.GetPosition())) lbessard@332: event = wx.MouseEvent() laurent@586: event.m_x, event.m_y = self.Editor.ScreenToClient(wx.GetMousePosition()) lbessard@332: dc = self.GetLogicalDC() lbessard@332: self.SelectedElement.OnLeftUp(event, dc, self.Scaling) lbessard@332: self.SelectedElement.SetSelected(False) lbessard@332: block.OnLeftDown(event, dc, self.Scaling) lbessard@332: self.SelectedElement = block lbessard@332: self.SelectedElement.SetSelected(True) laurent@586: self.RefreshVariablePanel() lbessard@267: self.RefreshVisibleElements() lbessard@216: else: lbessard@216: event.Skip() laurent@586: elif keycode == ord("+"): laurent@586: self.SetScale(self.CurrentScale + 1) laurent@586: self.ParentWindow.RefreshDisplayMenu() laurent@586: elif keycode == ord("-"): laurent@586: self.SetScale(self.CurrentScale - 1) laurent@586: self.ParentWindow.RefreshDisplayMenu() lbessard@97: else: lbessard@97: event.Skip() lbessard@27: lbessard@27: #------------------------------------------------------------------------------- laurent@510: # Model adding functions from Drop Target laurent@510: #------------------------------------------------------------------------------- laurent@510: laurent@510: def AddVariableBlock(self, x, y, scaling, var_class, var_name, var_type): laurent@510: id = self.GetNewId() laurent@510: variable = FBD_Variable(self, var_class, var_name, var_type, id) laurent@510: width, height = variable.GetMinSize() laurent@510: if scaling is not None: laurent@510: x = round(float(x) / float(scaling[0])) * scaling[0] laurent@510: y = round(float(y) / float(scaling[1])) * scaling[1] laurent@510: width = round(float(width) / float(scaling[0]) + 0.5) * scaling[0] laurent@510: height = round(float(height) / float(scaling[1]) + 0.5) * scaling[1] laurent@510: variable.SetPosition(x, y) laurent@510: variable.SetSize(width, height) laurent@510: self.AddBlock(variable) laurent@510: self.Controler.AddEditedElementVariable(self.GetTagName(), id, var_class) laurent@510: self.RefreshVariableModel(variable) laurent@510: self.RefreshBuffer() laurent@510: self.RefreshScrollBars() laurent@510: self.RefreshVisibleElements() laurent@510: self.Refresh(False) laurent@510: laurent@510: #------------------------------------------------------------------------------- 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@332: lbessard@27: def AddNewBlock(self, bbox): laurent@409: dialog = FBDBlockDialog(self.ParentWindow, self.Controler) lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@249: dialog.SetBlockList(self.Controler.GetBlockTypes(self.TagName, self.Debug)) lbessard@249: dialog.SetPouNames(self.Controler.GetProjectPouNames(self.Debug)) lbessard@249: dialog.SetPouElementNames(self.Controler.GetEditedElementVariables(self.TagName, self.Debug)) 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@269: values.setdefault("name", "") lbessard@269: block = FBD_Block(self, values["type"], values["name"], id, lbessard@269: values["extension"], values["inputs"], lbessard@269: executionControl = values["executionControl"], lbessard@269: executionOrder = values["executionOrder"]) 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@249: self.RefreshVisibleElements() laurent@586: self.RefreshVariablePanel() laurent@684: self.ParentWindow.RefreshPouInstanceVariablesPanel() lbessard@144: block.Refresh() lbessard@27: dialog.Destroy() lbessard@27: lbessard@27: def AddNewVariable(self, bbox): lbessard@253: words = self.TagName.split("::") lbessard@253: if words[0] == "T": laurent@409: dialog = FBDVariableDialog(self.ParentWindow, self.Controler, words[2]) lbessard@253: else: laurent@409: dialog = FBDVariableDialog(self.ParentWindow, self.Controler) lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@27: dialog.SetMinVariableSize((bbox.width, bbox.height)) lbessard@27: varlist = [] lbessard@249: vars = self.Controler.GetEditedElementInterfaceVars(self.TagName, self.Debug) 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@249: returntype = self.Controler.GetEditedElementInterfaceReturnType(self.TagName, self.Debug) 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@249: self.RefreshVisibleElements() lbessard@144: variable.Refresh() lbessard@27: dialog.Destroy() lbessard@27: lbessard@27: def AddNewConnection(self, bbox): laurent@409: dialog = ConnectionDialog(self.ParentWindow, self.Controler) lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@330: dialog.SetPouNames(self.Controler.GetProjectPouNames(self.Debug)) lbessard@330: dialog.SetPouElementNames(self.Controler.GetEditedElementVariables(self.TagName, self.Debug)) 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@249: self.RefreshVisibleElements() lbessard@144: connection.Refresh() lbessard@27: dialog.Destroy() lbessard@27: lbessard@27: def AddNewComment(self, bbox): lbessard@114: if wx.VERSION >= (2, 5, 0): laurent@391: dialog = wx.TextEntryDialog(self.ParentWindow, _("Edit comment"), _("Please enter comment text"), "", wx.OK|wx.CANCEL|wx.TE_MULTILINE) lbessard@114: else: laurent@391: dialog = wx.TextEntryDialog(self.ParentWindow, _("Edit comment"), _("Please enter comment text"), "", wx.OK|wx.CANCEL) laurent@566: dialog.SetClientSize(wx.Size(400, 200)) 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@249: self.RefreshVisibleElements() lbessard@144: comment.Refresh() lbessard@27: dialog.Destroy() lbessard@27: lbessard@27: def AddNewContact(self, bbox): lbessard@231: dialog = LDElementDialog(self.ParentWindow, self.Controler, "contact") lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@27: varlist = [] lbessard@249: vars = self.Controler.GetEditedElementInterfaceVars(self.TagName, self.Debug) lbessard@27: if vars: lbessard@27: for var in vars: lbessard@277: if 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@249: self.RefreshVisibleElements() lbessard@144: contact.Refresh() lbessard@27: dialog.Destroy() lbessard@27: lbessard@27: def AddNewCoil(self, bbox): lbessard@231: dialog = LDElementDialog(self.ParentWindow, self.Controler, "coil") lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@27: varlist = [] lbessard@249: vars = self.Controler.GetEditedElementInterfaceVars(self.TagName, self.Debug) 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@249: returntype = self.Controler.GetEditedElementInterfaceReturnType(self.TagName, self.Debug) 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@249: self.RefreshVisibleElements() lbessard@144: coil.Refresh() lbessard@27: dialog.Destroy() lbessard@27: lbessard@27: def AddNewPowerRail(self, bbox): lbessard@231: dialog = LDPowerRailDialog(self.ParentWindow, self.Controler) 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() pizza@550: powerrail = LD_PowerRail(self, values["type"], id, 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@249: self.RefreshVisibleElements() lbessard@144: powerrail.Refresh() lbessard@27: dialog.Destroy() lbessard@27: lbessard@71: def AddNewStep(self, bbox, initial = False): laurent@409: dialog = SFCStepDialog(self.ParentWindow, self.Controler, initial) lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@249: dialog.SetPouNames(self.Controler.GetProjectPouNames(self.Debug)) lbessard@249: dialog.SetVariables(self.Controler.GetEditedElementInterfaceVars(self.TagName, self.Debug)) laurent@383: dialog.SetStepNames([block.GetName() for block in self.Blocks.itervalues() 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@249: self.RefreshVisibleElements() lbessard@144: step.Refresh() lbessard@71: dialog.Destroy() lbessard@71: lbessard@27: def AddNewTransition(self, bbox): laurent@409: dialog = SFCTransitionDialog(self.ParentWindow, self.Controler, self.GetDrawingMode() == FREEDRAWING_MODE) lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@249: dialog.SetTransitions(self.Controler.GetEditedElementTransitions(self.TagName, self.Debug)) 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@249: self.RefreshVisibleElements() lbessard@144: transition.Refresh() lbessard@27: dialog.Destroy() lbessard@27: lbessard@27: def AddNewDivergence(self, bbox): laurent@409: dialog = SFCDivergenceDialog(self.ParentWindow, self.Controler) 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@249: self.RefreshVisibleElements() lbessard@144: divergence.Refresh() lbessard@27: dialog.Destroy() etisserant@0: lbessard@108: def AddNewJump(self, bbox): lbessard@108: choices = [] laurent@383: for block in self.Blocks.itervalues(): lbessard@108: if isinstance(block, SFC_Step): lbessard@108: choices.append(block.GetName()) Laurent@720: dialog = wx.SingleChoiceDialog(self.ParentWindow, Laurent@720: _("Add a new jump"), _("Please choose a target"), Laurent@720: choices, wx.DEFAULT_DIALOG_STYLE|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@249: self.RefreshVisibleElements() 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@249: dialog.SetActionList(self.Controler.GetEditedElementActions(self.TagName, self.Debug)) lbessard@249: dialog.SetVariableList(self.Controler.GetEditedElementInterfaceVars(self.TagName, self.Debug)) 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@249: self.RefreshVisibleElements() 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): laurent@409: dialog = FBDBlockDialog(self.ParentWindow, self.Controler) lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@249: dialog.SetBlockList(self.Controler.GetBlockTypes(self.TagName, self.Debug)) lbessard@249: dialog.SetPouNames(self.Controler.GetProjectPouNames(self.Debug)) lbessard@249: variable_names = self.Controler.GetEditedElementVariables(self.TagName, self.Debug) 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@269: old_values = {"name" : block.GetName(), lbessard@269: "type" : block.GetType(), lbessard@269: "extension" : block.GetExtension(), lbessard@269: "inputs" : block.GetInputTypes(), lbessard@269: "executionControl" : block.GetExecutionControl(), lbessard@269: "executionOrder" : block.GetExecutionOrder()} 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@269: block.SetType(new_values["type"], new_values["extension"], executionControl = new_values["executionControl"]) lbessard@118: block.SetExecutionOrder(new_values["executionOrder"]) lbessard@154: rect = rect.Union(block.GetRedrawRect()) lbessard@27: self.RefreshBlockModel(block) lbessard@56: self.RefreshBuffer() Laurent@710: if old_values["executionOrder"] != new_values["executionOrder"]: Laurent@710: self.RefreshView(selection=({block.GetId(): True}, {})) Laurent@710: else: Laurent@710: self.RefreshScrollBars() Laurent@710: self.RefreshVisibleElements() Laurent@710: block.Refresh(rect) laurent@586: self.RefreshVariablePanel() laurent@684: self.ParentWindow.RefreshPouInstanceVariablesPanel() lbessard@27: dialog.Destroy() lbessard@27: lbessard@27: def EditVariableContent(self, variable): lbessard@253: words = self.TagName.split("::") lbessard@253: if words[0] == "T": laurent@409: dialog = FBDVariableDialog(self.ParentWindow, self.Controler, words[2]) lbessard@253: else: laurent@409: dialog = FBDVariableDialog(self.ParentWindow, self.Controler) lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@27: dialog.SetMinVariableSize(variable.GetSize()) lbessard@27: varlist = [] lbessard@249: vars = self.Controler.GetEditedElementInterfaceVars(self.TagName, self.Debug) 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@249: returntype = self.Controler.GetEditedElementInterfaceReturnType(self.TagName, self.Debug) 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) Laurent@710: self.RefreshBuffer() lbessard@118: if old_values["executionOrder"] != new_values["executionOrder"]: Laurent@710: self.RefreshView(selection=({variable.GetId(): True}, {})) Laurent@710: else: Laurent@710: self.RefreshVisibleElements() Laurent@710: self.RefreshScrollBars() Laurent@710: variable.Refresh(rect) lbessard@27: dialog.Destroy() lbessard@27: lbessard@28: def EditConnectionContent(self, connection): laurent@409: dialog = ConnectionDialog(self.ParentWindow, self.Controler) lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@330: dialog.SetPouNames(self.Controler.GetProjectPouNames(self.Debug)) lbessard@330: dialog.SetPouElementNames(self.Controler.GetEditedElementVariables(self.TagName, self.Debug)) 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() Laurent@710: old_name = connection.GetName() 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() Laurent@710: if old_name != values["name"]: Laurent@710: self.Controler.UpdateEditedElementUsedVariable(self.TagName, old_name, values["name"]) Laurent@710: self.RefreshView(selection=({connection.GetId(): True}, {})) Laurent@710: else: Laurent@710: self.RefreshScrollBars() Laurent@710: self.RefreshVisibleElements() Laurent@710: connection.Refresh(rect) lbessard@28: dialog.Destroy() lbessard@28: lbessard@28: def EditContactContent(self, contact): lbessard@231: dialog = LDElementDialog(self.ParentWindow, self.Controler, "contact") lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@28: varlist = [] lbessard@249: vars = self.Controler.GetEditedElementInterfaceVars(self.TagName, self.Debug) lbessard@28: if vars: lbessard@28: for var in vars: lbessard@277: if 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@249: self.RefreshVisibleElements() lbessard@154: contact.Refresh(rect) lbessard@28: dialog.Destroy() lbessard@28: lbessard@28: def EditCoilContent(self, coil): lbessard@231: dialog = LDElementDialog(self.ParentWindow, self.Controler, "coil") lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@28: varlist = [] lbessard@249: vars = self.Controler.GetEditedElementInterfaceVars(self.TagName, self.Debug) 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@249: returntype = self.Controler.GetEditedElementInterfaceReturnType(self.TagName, self.Debug) 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@249: self.RefreshVisibleElements() lbessard@154: coil.Refresh(rect) lbessard@28: dialog.Destroy() lbessard@28: lbessard@28: def EditPowerRailContent(self, powerrail): laurent@585: connectors = powerrail.GetConnectors() laurent@585: type = powerrail.GetType() laurent@585: if type == LEFTRAIL: laurent@585: pin_number = len(connectors["outputs"]) laurent@585: else: laurent@585: pin_number = len(connectors["inputs"]) laurent@585: dialog = LDPowerRailDialog(self.ParentWindow, self.Controler, type, pin_number) 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) laurent@585: powerrail.SetType(values["type"], 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@249: self.RefreshVisibleElements() lbessard@154: powerrail.Refresh(rect) lbessard@28: dialog.Destroy() lbessard@64: lbessard@71: def EditStepContent(self, step): laurent@409: dialog = SFCStepDialog(self.ParentWindow, self.Controler, step.GetInitial()) lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@249: dialog.SetPouNames(self.Controler.GetProjectPouNames(self.Debug)) lbessard@249: dialog.SetVariables(self.Controler.GetEditedElementInterfaceVars(self.TagName, self.Debug)) laurent@383: dialog.SetStepNames([block.GetName() for block in self.Blocks.itervalues() 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() laurent@391: values["input"] = len(connectors["inputs"]) > 0 laurent@391: values["output"] = len(connectors["outputs"]) > 0 laurent@391: values["action"] = step.GetActionConnector() != 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@249: self.RefreshVisibleElements() lbessard@154: step.Refresh(rect) lbessard@71: lbessard@64: def EditTransitionContent(self, transition): laurent@409: dialog = SFCTransitionDialog(self.ParentWindow, self.Controler, self.GetDrawingMode() == FREEDRAWING_MODE) lbessard@165: dialog.SetPreviewFont(self.GetFont()) lbessard@249: dialog.SetTransitions(self.Controler.GetEditedElementTransitions(self.TagName, self.Debug)) 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@249: self.RefreshVisibleElements() lbessard@154: transition.Refresh(rect) lbessard@64: dialog.Destroy() lbessard@64: lbessard@64: def EditJumpContent(self, jump): lbessard@64: choices = [] laurent@383: for block in self.Blocks.itervalues(): lbessard@64: if isinstance(block, SFC_Step): lbessard@64: choices.append(block.GetName()) Laurent@720: dialog = wx.SingleChoiceDialog(self.ParentWindow, Laurent@720: _("Edit jump target"), _("Please choose a target"), Laurent@720: choices, wx.DEFAULT_DIALOG_STYLE|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@249: self.RefreshVisibleElements() 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@249: dialog.SetActionList(self.Controler.GetEditedElementActions(self.TagName, self.Debug)) lbessard@249: dialog.SetVariableList(self.Controler.GetEditedElementInterfaceVars(self.TagName, self.Debug)) 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@249: self.RefreshVisibleElements() 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): laurent@391: dialog = wx.TextEntryDialog(self.ParentWindow, _("Edit comment"), _("Please enter comment text"), comment.GetContent(), wx.OK|wx.CANCEL|wx.TE_MULTILINE) lbessard@114: else: laurent@391: dialog = wx.TextEntryDialog(self.ParentWindow, _("Edit comment"), _("Please enter comment text"), comment.GetContent(), wx.OK|wx.CANCEL) laurent@566: dialog.SetClientSize(wx.Size(400, 200)) 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@249: self.RefreshVisibleElements() 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() laurent@383: infos["action"] = step.GetActionConnector() 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() laurent@383: infos["connection"] = transition.GetConditionConnector() 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() laurent@586: wx.CallAfter(self.RefreshVariablePanel) laurent@684: wx.CallAfter(self.ParentWindow.RefreshPouInstanceVariablesPanel) lbessard@239: lbessard@28: def DeleteVariable(self, variable): lbessard@28: connectors = variable.GetConnectors() laurent@383: if len(connectors["outputs"]) > 0: laurent@383: elements = connectors["outputs"][0].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() laurent@383: elements = connectors["outputs"][0].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() laurent@383: elements = connectors["outputs"][0].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: laurent@383: connectors = powerrail.GetConnectors() laurent@383: for connector in connectors["outputs"]: 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() laurent@383: action_connector = step.GetActionConnector() laurent@383: if len(connectors["outputs"]) > 0: laurent@383: for element in connectors["outputs"][0].GetConnectedBlocks(): lbessard@28: if element not in elements: lbessard@28: elements.append(element) laurent@383: if action_connector is not None: laurent@383: for element in action_connector.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() laurent@383: for element in connectors["outputs"][0].GetConnectedBlocks(): laurent@383: if element not in elements: laurent@383: 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@283: if not self.Debug and (self.IsBlock(self.SelectedElement) or self.IsComment(self.SelectedElement) or isinstance(self.SelectedElement, Graphic_Group)): laurent@384: blocks, wires = self.SelectedElement.GetDefinition() laurent@384: text = self.Controler.GetEditedElementInstancesCopy(self.TagName, blocks, wires, self.Debug) laurent@384: self.ParentWindow.SetCopyBuffer(text) 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() laurent@586: self.RefreshVariablePanel() laurent@684: self.ParentWindow.RefreshPouInstanceVariablesPanel() lbessard@155: self.RefreshRect(self.GetScrolledRect(rect), False) etisserant@0: etisserant@0: def Copy(self): lbessard@283: if not self.Debug and (self.IsBlock(self.SelectedElement) or self.IsComment(self.SelectedElement) or isinstance(self.SelectedElement, Graphic_Group)): laurent@384: blocks, wires = self.SelectedElement.GetDefinition() laurent@384: text = self.Controler.GetEditedElementInstancesCopy(self.TagName, blocks, wires, self.Debug) laurent@384: self.ParentWindow.SetCopyBuffer(text) lbessard@144: laurent@666: def Paste(self, bbx=None): laurent@384: if not self.Debug: laurent@384: element = self.ParentWindow.GetCopyBuffer() laurent@666: if bbx is None: laurent@666: mouse_pos = self.Editor.ScreenToClient(wx.GetMousePosition()) laurent@666: middle = wx.Rect(0, 0, *self.Editor.GetClientSize()).InsideXY(mouse_pos.x, mouse_pos.y) laurent@666: if middle: laurent@666: x, y = self.CalcUnscrolledPosition(mouse_pos.x, mouse_pos.y) laurent@666: else: laurent@666: x, y = self.CalcUnscrolledPosition(0, 0) laurent@666: new_pos = [int(x / self.ViewScale[0]), int(y / self.ViewScale[1])] lbessard@332: else: laurent@666: middle = True laurent@666: new_pos = [bbx.x, bbx.y] laurent@384: result = self.Controler.PasteEditedElementInstances(self.TagName, element, new_pos, middle, self.Debug) laurent@384: if not isinstance(result, (StringType, UnicodeType)): laurent@384: self.RefreshBuffer() laurent@586: self.RefreshView(selection=result) laurent@586: self.RefreshVariablePanel() laurent@684: self.ParentWindow.RefreshPouInstanceVariablesPanel() lbessard@332: else: laurent@586: message = wx.MessageDialog(self.Editor, result, "Error", wx.OK|wx.ICON_ERROR) laurent@384: message.ShowModal() laurent@384: message.Destroy() lbessard@283: lbessard@283: def CanAddElement(self, block): lbessard@283: if isinstance(block, Graphic_Group): lbessard@283: return block.CanAddBlocks(self) lbessard@283: elif 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: Laurent@704: def GenerateNewName(self, element=None, blocktype=None, exclude={}): Laurent@704: if element is not None and isinstance(element, SFC_Step): lbessard@283: format = "Step%d" laurent@384: else: Laurent@704: if element is not None: Laurent@704: blocktype = element.GetType() Laurent@704: if blocktype is None: Laurent@704: blocktype = "Block" Laurent@704: format = "%s%%d" % blocktype Laurent@704: return self.Controler.GenerateNewName(self.TagName, None, format, exclude, self.Debug) lbessard@283: lbessard@283: def IsNamedElement(self, element): lbessard@283: return isinstance(element, FBD_Block) and element.GetName() != "" or isinstance(element, SFC_Step) lbessard@283: lbessard@114: def CopyBlock(self, element, pos): lbessard@114: id = self.GetNewId() lbessard@283: if isinstance(element, Graphic_Group): lbessard@283: block = element.Clone(self, pos=pos) lbessard@144: else: lbessard@283: if self.IsNamedElement(element): lbessard@283: name = self.GenerateNewName(element) lbessard@283: block = element.Clone(self, id, name, pos) lbessard@283: else: lbessard@283: name = None lbessard@283: block = element.Clone(self, id, pos=pos) lbessard@283: self.AddBlockInModel(block) lbessard@283: return block lbessard@283: lbessard@283: def AddBlockInModel(self, block): lbessard@114: if isinstance(block, Comment): lbessard@178: self.AddComment(block) lbessard@283: self.Controler.AddEditedElementComment(self.TagName, block.GetId()) lbessard@144: self.RefreshCommentModel(block) lbessard@178: else: lbessard@178: self.AddBlock(block) lbessard@178: if isinstance(block, FBD_Block): lbessard@283: self.Controler.AddEditedElementBlock(self.TagName, block.GetId(), block.GetType(), block.GetName()) lbessard@178: self.RefreshBlockModel(block) lbessard@178: elif isinstance(block, FBD_Variable): lbessard@283: self.Controler.AddEditedElementVariable(self.TagName, block.GetId(), block.GetType()) lbessard@178: self.RefreshVariableModel(block) lbessard@178: elif isinstance(block, FBD_Connector): lbessard@283: self.Controler.AddEditedElementConnection(self.TagName, block.GetId(), block.GetType()) lbessard@178: self.RefreshConnectionModel(block) lbessard@178: elif isinstance(block, LD_Contact): lbessard@283: self.Controler.AddEditedElementContact(self.TagName, block.GetId()) lbessard@178: self.RefreshContactModel(block) lbessard@178: elif isinstance(block, LD_Coil): lbessard@283: self.Controler.AddEditedElementCoil(self.TagName, block.GetId()) lbessard@178: self.RefreshCoilModel(block) lbessard@178: elif isinstance(block, LD_PowerRail): lbessard@283: self.Controler.AddEditedElementPowerRail(self.TagName, block.GetId(), block.GetType()) lbessard@178: self.RefreshPowerRailModel(block) lbessard@178: elif isinstance(block, SFC_Step): lbessard@283: self.Controler.AddEditedElementStep(self.TagName, block.GetId()) lbessard@178: self.RefreshStepModel(block) lbessard@178: elif isinstance(block, SFC_Transition): lbessard@283: self.Controler.AddEditedElementTransition(self.TagName, block.GetId()) lbessard@178: self.RefreshTransitionModel(block) lbessard@178: elif isinstance(block, SFC_Divergence): lbessard@283: self.Controler.AddEditedElementDivergence(self.TagName, block.GetId(), block.GetType()) lbessard@202: self.RefreshDivergenceModel(block) lbessard@178: elif isinstance(block, SFC_Jump): lbessard@283: self.Controler.AddEditedElementJump(self.TagName, block.GetId()) lbessard@178: self.RefreshJumpModel(block) lbessard@178: elif isinstance(block, SFC_ActionBlock): lbessard@283: self.Controler.AddEditedElementActionBlock(self.TagName, block.GetId()) lbessard@178: self.RefreshActionBlockModel(block) lbessard@231: Laurent@738: #------------------------------------------------------------------------------- Laurent@738: # Find and Replace functions Laurent@738: #------------------------------------------------------------------------------- Laurent@738: Laurent@738: def Find(self, direction, search_params): Laurent@738: if self.SearchParams != search_params: Laurent@738: self.ClearHighlights(SEARCH_RESULT_HIGHLIGHT) Laurent@738: Laurent@738: self.SearchParams = search_params Laurent@738: criteria = { Laurent@738: "raw_pattern": search_params["find_pattern"], Laurent@738: "pattern": re.compile(search_params["find_pattern"]), Laurent@738: "case_sensitive": search_params["case_sensitive"], Laurent@738: "regular_expression": search_params["regular_expression"], Laurent@738: "filter": "all"} Laurent@738: Laurent@738: self.SearchResults = [] Laurent@738: blocks = [] Laurent@738: for infos, start, end, text in self.Controler.SearchInPou(self.TagName, criteria, self.Debug): Laurent@738: if infos[1] in ["var_local", "var_input", "var_output", "var_inout"]: Laurent@738: self.SearchResults.append((infos[1:], start, end, SEARCH_RESULT_HIGHLIGHT)) Laurent@738: else: Laurent@738: block = self.Blocks.get(infos[2]) Laurent@738: if block is not None: Laurent@738: blocks.append((block, (infos[1:], start, end, SEARCH_RESULT_HIGHLIGHT))) Laurent@738: blocks.sort(sort_blocks) Laurent@738: self.SearchResults.extend([infos for block, infos in blocks]) Laurent@738: Laurent@738: if len(self.SearchResults) > 0: Laurent@738: if self.CurrentFindHighlight is not None: Laurent@738: old_idx = self.SearchResults.index(self.CurrentFindHighlight) Laurent@738: if self.SearchParams["wrap"]: Laurent@738: idx = (old_idx + direction) % len(self.SearchResults) Laurent@738: else: Laurent@738: idx = max(0, min(old_idx + direction, len(self.SearchResults) - 1)) Laurent@738: if idx != old_idx: Laurent@738: self.RemoveHighlight(*self.CurrentFindHighlight) Laurent@738: self.CurrentFindHighlight = self.SearchResults[idx] Laurent@738: self.AddHighlight(*self.CurrentFindHighlight) Laurent@738: else: Laurent@738: self.CurrentFindHighlight = self.SearchResults[0] Laurent@738: self.AddHighlight(*self.CurrentFindHighlight) Laurent@738: Laurent@738: else: Laurent@738: if self.CurrentFindHighlight is not None: Laurent@738: self.RemoveHighlight(*self.CurrentFindHighlight) Laurent@738: self.CurrentFindHighlight = None Laurent@738: lbessard@231: #------------------------------------------------------------------------------- laurent@566: # Highlights showing functions laurent@566: #------------------------------------------------------------------------------- laurent@566: laurent@566: def OnRefreshHighlightsTimer(self, event): lbessard@231: self.RefreshView() laurent@566: event.Skip() laurent@566: laurent@566: def ClearHighlights(self, highlight_type=None): laurent@617: EditorPanel.ClearHighlights(self, highlight_type) laurent@617: laurent@566: if highlight_type is None: laurent@566: self.Highlights = [] laurent@566: else: laurent@566: self.Highlights = [(infos, start, end, highlight) for (infos, start, end, highlight) in self.Highlights if highlight != highlight_type] laurent@566: self.RefreshView() laurent@566: laurent@566: def AddHighlight(self, infos, start, end, highlight_type): laurent@617: EditorPanel.AddHighlight(self, infos, start, end, highlight_type) laurent@617: laurent@566: self.Highlights.append((infos, start, end, highlight_type)) Laurent@738: if infos[0] not in ["var_local", "var_input", "var_output", "var_inout"]: Laurent@738: block = self.Blocks.get(infos[1]) Laurent@738: if block is not None: Laurent@738: self.EnsureVisible(block) Laurent@738: self.RefreshHighlightsTimer.Start(int(REFRESH_HIGHLIGHT_PERIOD * 1000), oneShot=True) Laurent@738: Laurent@738: def RemoveHighlight(self, infos, start, end, highlight_type): Laurent@738: EditorPanel.RemoveHighlight(self, infos, start, end, highlight_type) Laurent@738: Laurent@738: if (infos, start, end, highlight_type) in self.Highlights: Laurent@738: self.Highlights.remove((infos, start, end, highlight_type)) Laurent@738: self.RefreshHighlightsTimer.Start(int(REFRESH_HIGHLIGHT_PERIOD * 1000), oneShot=True) Laurent@738: laurent@566: def ShowHighlights(self): laurent@566: for infos, start, end, highlight_type in self.Highlights: laurent@566: if infos[0] in ["comment", "io_variable", "block", "connector", "coil", "contact", "step", "transition", "jump", "action_block"]: lbessard@231: block = self.FindElementById(infos[1]) lbessard@231: if block is not None: laurent@566: block.AddHighlight(infos[2:], start, end, highlight_type) lbessard@231: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Drawing functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: lbessard@249: def OnScrollWindow(self, event): laurent@669: if self.Editor.HasCapture() and self.StartMousePos: laurent@666: return laurent@621: if wx.Platform == '__WXMSW__': laurent@621: wx.CallAfter(self.RefreshVisibleElements) laurent@621: elif event.GetOrientation() == wx.HORIZONTAL: lbessard@249: self.RefreshVisibleElements(xp = event.GetPosition()) lbessard@249: else: lbessard@249: self.RefreshVisibleElements(yp = event.GetPosition()) lbessard@249: event.Skip() lbessard@249: lbessard@319: def OnScrollStop(self, event): lbessard@319: self.RefreshScrollBars() lbessard@319: event.Skip() lbessard@319: lbessard@253: def OnMouseWheelWindow(self, event): lbessard@319: if self.StartMousePos is None or self.StartScreenPos is None: lbessard@323: rotation = event.GetWheelRotation() / event.GetWheelDelta() lbessard@323: if event.ShiftDown(): lbessard@323: x, y = self.GetViewStart() laurent@586: xp = max(0, min(x - rotation * 3, self.Editor.GetVirtualSize()[0] / self.Editor.GetScrollPixelsPerUnit()[0])) lbessard@323: self.RefreshVisibleElements(xp = xp) lbessard@323: self.Scroll(xp, y) lbessard@323: elif event.ControlDown(): lbessard@323: dc = self.GetLogicalDC() laurent@586: self.SetScale(self.CurrentScale + rotation, mouse_event = event) lbessard@332: self.ParentWindow.RefreshDisplayMenu() lbessard@323: else: lbessard@323: x, y = self.GetViewStart() laurent@586: yp = max(0, min(y - rotation * 3, self.Editor.GetVirtualSize()[1] / self.Editor.GetScrollPixelsPerUnit()[1])) lbessard@323: self.RefreshVisibleElements(yp = yp) lbessard@323: self.Scroll(x, yp) lbessard@253: lbessard@27: def OnMoveWindow(self, event): lbessard@42: self.RefreshScrollBars() lbessard@249: self.RefreshVisibleElements() 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: laurent@586: dc.SetBackground(wx.Brush(self.Editor.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() laurent@586: window_size = self.Editor.GetClientSize() laurent@586: width, height = self.Editor.GetVirtualSize() lbessard@323: width = int(max(width, xstart * SCROLLBAR_UNIT + window_size[0]) / self.ViewScale[0]) lbessard@323: height = int(max(height, ystart * SCROLLBAR_UNIT + window_size[1]) / self.ViewScale[1]) lbessard@331: dc.DrawRectangle(1, 1, 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() laurent@586: window_size = self.Editor.GetClientSize() lbessard@331: for x in xrange(self.PageSize[0] - (xstart * SCROLLBAR_UNIT) % self.PageSize[0], int(window_size[0] / self.ViewScale[0]), self.PageSize[0]): lbessard@331: dc.DrawLine(xstart * SCROLLBAR_UNIT + x + 1, int(ystart * SCROLLBAR_UNIT / self.ViewScale[0]), lbessard@331: xstart * SCROLLBAR_UNIT + x + 1, int((ystart * SCROLLBAR_UNIT + window_size[1]) / self.ViewScale[0])) lbessard@331: for y in xrange(self.PageSize[1] - (ystart * SCROLLBAR_UNIT) % self.PageSize[1], int(window_size[1] / self.ViewScale[1]), self.PageSize[1]): lbessard@331: dc.DrawLine(int(xstart * SCROLLBAR_UNIT / self.ViewScale[0]), ystart * SCROLLBAR_UNIT + y + 1, lbessard@331: int((xstart * SCROLLBAR_UNIT + window_size[0]) / self.ViewScale[1]), ystart * SCROLLBAR_UNIT + y + 1) lbessard@140: lbessard@140: # Draw all elements laurent@383: for comment in self.Comments.itervalues(): lbessard@249: if comment != self.SelectedElement and (comment.IsVisible() or printing): lbessard@42: comment.Draw(dc) laurent@383: for wire in self.Wires.iterkeys(): lbessard@249: if wire != self.SelectedElement and (wire.IsVisible() or printing): lbessard@249: if not self.Debug or wire.GetValue() != True: lbessard@249: wire.Draw(dc) lbessard@249: if self.Debug: laurent@383: for wire in self.Wires.iterkeys(): lbessard@249: if wire != self.SelectedElement and (wire.IsVisible() or printing) and wire.GetValue() == True: lbessard@249: wire.Draw(dc) laurent@383: for block in self.Blocks.itervalues(): lbessard@249: if block != self.SelectedElement and (block.IsVisible() or printing): lbessard@42: block.Draw(dc) lbessard@144: lbessard@249: if self.SelectedElement is not None and (self.SelectedElement.IsVisible() or printing): lbessard@249: self.SelectedElement.Draw(dc) lbessard@213: lbessard@213: if not printing: laurent@407: if self.Debug: laurent@407: xstart, ystart = self.GetViewStart() laurent@407: dc.DrawText(_("Debug: %s") % self.InstancePath, 2, 2) lbessard@213: if self.rubberBand.IsShown(): lbessard@213: self.rubberBand.Draw(dc) lbessard@213: dc.EndDrawing() lbessard@213: lbessard@213: def OnPaint(self, event): greg@352: dc = self.GetLogicalDC(True) greg@352: self.DoDrawing(dc) laurent@586: wx.BufferedPaintDC(self.Editor, dc.GetAsBitmap()) greg@372: if self.Debug: greg@372: DebugViewer.RefreshNewData(self) lbessard@213: event.Skip() lbessard@213: lbessard@213: