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: # etisserant@0: #Copyright (C): 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 etisserant@0: #Lesser General Public License for more details. etisserant@0: # etisserant@5: #You should have received a copy of the GNU General Public etisserant@0: #License along with this library; if not, write to the Free Software etisserant@0: #Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA etisserant@0: etisserant@0: from wxPython.wx import * etisserant@0: import wx etisserant@0: etisserant@0: from plcopen.structures import * etisserant@0: from graphics.GraphicCommons import * etisserant@0: from graphics.FBD_Objects import * etisserant@0: from graphics.LD_Objects import * lbessard@27: from graphics.SFC_Objects import * lbessard@27: lbessard@27: from Dialogs import * lbessard@27: lbessard@27: SCROLLBAR_UNIT = 10 lbessard@27: WINDOW_BORDER = 10 lbessard@27: SCROLL_ZONE = 10 etisserant@0: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Graphic elements Viewer base class etisserant@0: #------------------------------------------------------------------------------- etisserant@0: etisserant@0: # ID Constants for menu items etisserant@0: [wxID_FBDVIEWERCONTEXTUALMENUITEMS0, wxID_FBDVIEWERCONTEXTUALMENUITEMS1, etisserant@0: wxID_FBDVIEWERCONTEXTUALMENUITEMS2, wxID_FBDVIEWERCONTEXTUALMENUITEMS3, etisserant@0: wxID_FBDVIEWERCONTEXTUALMENUITEMS5, wxID_FBDVIEWERCONTEXTUALMENUITEMS6, etisserant@0: wxID_FBDVIEWERCONTEXTUALMENUITEMS8, wxID_FBDVIEWERCONTEXTUALMENUITEMS9, etisserant@0: wxID_FBDVIEWERCONTEXTUALMENUITEMS11, etisserant@0: ] = [wx.NewId() for _init_coll_ContextualMenu_Items in range(9)] etisserant@0: etisserant@0: etisserant@0: """ etisserant@0: Class that implements a Viewer based on a wxScrolledWindow for drawing and etisserant@0: manipulating graphic elements etisserant@0: """ etisserant@0: etisserant@0: class Viewer(wx.ScrolledWindow): etisserant@0: etisserant@0: # Create Contextual Menu items etisserant@0: def _init_coll_ContextualMenu_Items(self, parent): etisserant@0: # Create menu items etisserant@0: parent.Append(help='', id=wxID_FBDVIEWERCONTEXTUALMENUITEMS0, etisserant@0: kind=wx.ITEM_RADIO, text=u'No Modifier') etisserant@0: parent.Append(help='', id=wxID_FBDVIEWERCONTEXTUALMENUITEMS1, etisserant@0: kind=wx.ITEM_RADIO, text=u'Negated') etisserant@0: parent.Append(help='', id=wxID_FBDVIEWERCONTEXTUALMENUITEMS2, etisserant@0: kind=wx.ITEM_RADIO, text=u'Rising Edge') etisserant@0: parent.Append(help='', id=wxID_FBDVIEWERCONTEXTUALMENUITEMS3, etisserant@0: kind=wx.ITEM_RADIO, text=u'Falling Edge') etisserant@0: parent.AppendSeparator() etisserant@0: parent.Append(help='', id=wxID_FBDVIEWERCONTEXTUALMENUITEMS5, etisserant@0: kind=wx.ITEM_NORMAL, text=u'Add Wire Segment') etisserant@0: parent.Append(help='', id=wxID_FBDVIEWERCONTEXTUALMENUITEMS6, etisserant@0: kind=wx.ITEM_NORMAL, text=u'Delete Wire Segment') etisserant@0: parent.AppendSeparator() etisserant@0: parent.Append(help='', id=wxID_FBDVIEWERCONTEXTUALMENUITEMS8, etisserant@0: kind=wx.ITEM_NORMAL, text=u'Add Divergence Branch') etisserant@0: parent.Append(help='', id=wxID_FBDVIEWERCONTEXTUALMENUITEMS9, etisserant@0: kind=wx.ITEM_NORMAL, text=u'Delete Divergence Branch') etisserant@0: parent.AppendSeparator() etisserant@0: parent.Append(help='', id=wxID_FBDVIEWERCONTEXTUALMENUITEMS11, etisserant@0: kind=wx.ITEM_NORMAL, text=u'Delete') etisserant@0: # Link menu event to corresponding called functions etisserant@0: self.Bind(wx.EVT_MENU, self.OnNoModifierMenu, etisserant@0: id=wxID_FBDVIEWERCONTEXTUALMENUITEMS0) etisserant@0: self.Bind(wx.EVT_MENU, self.OnNegatedMenu, etisserant@0: id=wxID_FBDVIEWERCONTEXTUALMENUITEMS1) etisserant@0: self.Bind(wx.EVT_MENU, self.OnRisingEdgeMenu, etisserant@0: id=wxID_FBDVIEWERCONTEXTUALMENUITEMS2) etisserant@0: self.Bind(wx.EVT_MENU, self.OnFallingEdgeMenu, etisserant@0: id=wxID_FBDVIEWERCONTEXTUALMENUITEMS3) etisserant@0: self.Bind(wx.EVT_MENU, self.OnAddSegmentMenu, etisserant@0: id=wxID_FBDVIEWERCONTEXTUALMENUITEMS5) etisserant@0: self.Bind(wx.EVT_MENU, self.OnDeleteSegmentMenu, etisserant@0: id=wxID_FBDVIEWERCONTEXTUALMENUITEMS6) etisserant@0: self.Bind(wx.EVT_MENU, self.OnAddBranchMenu, etisserant@0: id=wxID_FBDVIEWERCONTEXTUALMENUITEMS8) etisserant@0: self.Bind(wx.EVT_MENU, self.OnDeleteBranchMenu, etisserant@0: id=wxID_FBDVIEWERCONTEXTUALMENUITEMS9) etisserant@0: self.Bind(wx.EVT_MENU, self.OnDeleteMenu, etisserant@0: id=wxID_FBDVIEWERCONTEXTUALMENUITEMS11) etisserant@0: etisserant@0: # Create and initialize Contextual Menu etisserant@0: def _init_menus(self): etisserant@0: self.ContextualMenu = wx.Menu(title='') etisserant@0: etisserant@0: self._init_coll_ContextualMenu_Items(self.ContextualMenu) etisserant@0: etisserant@0: # Create a new Viewer etisserant@0: def __init__(self, parent, window, controler): lbessard@27: wx.ScrolledWindow.__init__(self, parent, style=wx.SUNKEN_BORDER | wx.HSCROLL | wx.VSCROLL) etisserant@0: self._init_menus() etisserant@0: # Adding a rubberband to Viewer etisserant@0: self.rubberBand = RubberBand(drawingSurface=self) etisserant@0: self.SetBackgroundColour(wxColour(255,255,255)) etisserant@0: self.ResetView() etisserant@0: self.Scaling = None etisserant@0: #self.Scaling = (8, 8) etisserant@0: self.DrawGrid = True etisserant@0: self.current_id = 0 etisserant@0: etisserant@0: # Initialize Viewer mode to Selection mode etisserant@0: self.Mode = MODE_SELECTION lbessard@3: self.SavedMode = False etisserant@0: etisserant@0: self.Parent = window etisserant@0: self.Controler = controler etisserant@0: etisserant@0: # Link Viewer event to corresponding methods etisserant@0: EVT_PAINT(self, self.OnPaint) etisserant@0: EVT_LEFT_DOWN(self, self.OnViewerLeftDown) etisserant@0: EVT_LEFT_UP(self, self.OnViewerLeftUp) etisserant@0: EVT_LEFT_DCLICK(self, self.OnViewerLeftDClick) etisserant@0: EVT_RIGHT_UP(self, self.OnViewerRightUp) etisserant@0: EVT_MOTION(self, self.OnViewerMotion) etisserant@0: EVT_CHAR(self, self.OnChar) lbessard@27: EVT_SCROLLWIN(self, self.OnMoveWindow) lbessard@27: EVT_SIZE(self, self.OnMoveWindow) etisserant@0: etisserant@0: # Returns a new id etisserant@0: def GetNewId(self): etisserant@0: self.current_id += 1 etisserant@0: return self.current_id etisserant@0: etisserant@0: # Destructor etisserant@0: def __del__(self): etisserant@0: self.ResetView() etisserant@0: lbessard@27: def GetLogicalDC(self): lbessard@27: dc = wxClientDC(self) lbessard@27: self.DoPrepareDC(dc) lbessard@27: return dc lbessard@27: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Reset functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: etisserant@0: # Resets Viewer lists etisserant@0: def ResetView(self): etisserant@0: self.Blocks = [] etisserant@0: self.Wires = [] etisserant@0: self.Elements = [] etisserant@0: self.SelectedElement = None etisserant@0: etisserant@0: # Changes Viewer mode etisserant@0: def SetMode(self, mode): lbessard@3: if self.Mode != mode or mode == MODE_SELECTION: lbessard@3: self.Mode = mode lbessard@3: self.SavedMode = False lbessard@3: else: lbessard@3: self.SavedMode = True etisserant@0: # Reset selection lbessard@3: if self.Mode != MODE_SELECTION and self.SelectedElement: etisserant@0: self.SelectedElement.SetSelected(False) etisserant@0: self.SelectedElement = None etisserant@0: self.Refresh() etisserant@0: lbessard@27: # Return current drawing mode lbessard@27: def GetDrawingMode(self): lbessard@27: return self.Parent.GetDrawingMode() lbessard@27: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Refresh functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: etisserant@0: # Refresh Viewer elements etisserant@0: def RefreshView(self): etisserant@0: self.current_id = 0 etisserant@0: # Start by reseting Viewer etisserant@0: self.ResetView() etisserant@0: instance = True etisserant@0: # List of ids of already loaded blocks etisserant@0: ids = [] etisserant@0: # Load Blocks until they are all loaded etisserant@0: while instance: etisserant@0: instance = self.Controler.GetCurrentElementEditingInstanceInfos(exclude=ids) etisserant@0: if instance: etisserant@0: self.loadInstance(instance, ids) lbessard@27: self.RefreshScrollBar() etisserant@0: self.Refresh() etisserant@0: lbessard@27: def RefreshScrollBar(self): lbessard@27: xstart, ystart = self.GetViewStart() lbessard@27: window_size = self.GetClientSize() lbessard@27: maxx = maxy = 0 lbessard@27: for element in self.Elements: lbessard@27: posx, posy = element.GetPosition() lbessard@27: width, height = element.GetSize() lbessard@27: maxx = max(maxx, posx + width) lbessard@27: maxy = max(maxy, posy + height) lbessard@27: maxx = max(maxx + WINDOW_BORDER, xstart * SCROLLBAR_UNIT + window_size[0]) lbessard@27: maxy = max(maxy + WINDOW_BORDER, ystart * SCROLLBAR_UNIT + window_size[1]) lbessard@27: if self.rubberBand.IsShown(): lbessard@27: extent = self.rubberBand.GetCurrentExtent() lbessard@27: maxx = max(maxx, extent.x + extent.width) lbessard@27: maxy = max(maxy, extent.y + extent.height) lbessard@27: self.SetScrollbars(SCROLLBAR_UNIT, SCROLLBAR_UNIT, lbessard@27: maxx / SCROLLBAR_UNIT, maxy / SCROLLBAR_UNIT, xstart, ystart) lbessard@27: etisserant@0: # Load instance from given informations etisserant@0: def loadInstance(self, instance, ids): etisserant@0: ids.append(instance["id"]) etisserant@0: self.current_id = max(self.current_id, instance["id"]) etisserant@0: if instance["type"] == "input": etisserant@0: variable = FBD_Variable(self, INPUT, instance["name"], instance["value_type"], instance["id"]) etisserant@0: variable.SetPosition(instance["x"], instance["y"]) etisserant@0: variable.SetSize(instance["width"], instance["height"]) etisserant@0: self.Blocks.append(variable) etisserant@0: self.Elements.append(variable) etisserant@0: connectors = variable.GetConnectors() etisserant@0: connectors["output"].SetPosition(wxPoint(*instance["connector"]["position"])) etisserant@0: if instance["connector"]["negated"]: etisserant@0: connectors["output"].SetNegated(True) etisserant@0: if instance["connector"]["edge"]: etisserant@0: connectors["output"].SetEdge(instance["connector"]["edge"]) etisserant@0: elif instance["type"] == "output": etisserant@0: variable = FBD_Variable(self, OUTPUT, instance["name"], instance["value_type"], instance["id"]) etisserant@0: variable.SetPosition(instance["x"], instance["y"]) etisserant@0: variable.SetSize(instance["width"], instance["height"]) etisserant@0: self.Blocks.append(variable) etisserant@0: self.Elements.append(variable) etisserant@0: connectors = variable.GetConnectors() etisserant@0: connectors["input"].SetPosition(wxPoint(*instance["connector"]["position"])) etisserant@0: if instance["connector"]["negated"]: etisserant@0: connectors["input"].SetNegated(True) etisserant@0: if instance["connector"]["edge"]: etisserant@0: connectors["input"].SetEdge(instance["connector"]["edge"]) etisserant@0: self.CreateWires(connectors["input"], instance["connector"]["links"], ids) etisserant@0: elif instance["type"] == "inout": etisserant@0: variable = FBD_Variable(self, INOUT, instance["name"], instance["value_type"], instance["id"]) etisserant@0: variable.SetPosition(instance["x"], instance["y"]) etisserant@0: variable.SetSize(instance["width"], instance["height"]) etisserant@0: self.Blocks.append(variable) etisserant@0: self.Elements.append(variable) etisserant@0: connectors = variable.GetConnectors() etisserant@0: connectors["output"].SetPosition(wxPoint(*instance["connectors"]["output"]["position"])) etisserant@0: connectors["input"].SetPosition(wxPoint(*instance["connectors"]["input"]["position"])) etisserant@0: if instance["connectors"]["output"]["negated"]: etisserant@0: connectors["output"].SetNegated(True) etisserant@0: if instance["connectors"]["output"]["edge"]: etisserant@0: connectors["output"].SetEdge(instance["connectors"]["output"]["edge"]) etisserant@0: if instance["connectors"]["input"]["negated"]: etisserant@0: connectors["input"].SetNegated(True) etisserant@0: if instance["connectors"]["input"]["edge"]: etisserant@0: connectors["input"].SetEdge(instance["connectors"]["input"]["edge"]) etisserant@0: self.CreateWires(connectors["input"], instance["connectors"]["input"]["links"], ids) etisserant@0: elif instance["type"] == "continuation": etisserant@0: connection = FBD_Connector(self, CONTINUATION, instance["name"], instance["id"]) etisserant@0: connection.SetPosition(instance["x"], instance["y"]) etisserant@0: connection.SetSize(instance["width"], instance["height"]) etisserant@0: self.Blocks.append(connection) etisserant@0: self.Elements.append(connection) etisserant@0: connector = connection.GetConnector() etisserant@0: connector.SetPosition(wxPoint(*instance["connector"]["position"])) etisserant@0: elif instance["type"] == "connection": etisserant@0: connection = FBD_Connection(self, CONNECTOR, instance["name"], instance["id"]) etisserant@0: connection.SetPosition(instance["x"], instance["y"]) etisserant@0: connection.SetSize(instance["width"], instance["height"]) etisserant@0: self.Blocks.append(connection) etisserant@0: self.Elements.append(connection) etisserant@0: connector = connection.GetConnector() etisserant@0: connector.SetPosition(wxPoint(*instance["connector"]["position"])) etisserant@0: self.CreateWires(connector, instance["connector"]["links"], ids) etisserant@0: elif instance["type"] == "comment": etisserant@0: comment = Comment(self, instance["content"], instance["id"]) etisserant@0: comment.SetPosition(instance["x"], instance["y"]) etisserant@0: comment.SetSize(instance["width"], instance["height"]) etisserant@0: self.Elements.append(comment) etisserant@0: elif instance["type"] == "leftPowerRail": etisserant@0: leftpowerrail = LD_PowerRail(self, LEFTRAIL, instance["id"], [True for i in range(len(instance["connectors"]))]) etisserant@0: leftpowerrail.SetPosition(instance["x"], instance["y"]) etisserant@0: self.Blocks.append(leftpowerrail) etisserant@0: self.Elements.append(leftpowerrail) etisserant@0: connectors = leftpowerrail.GetConnectors() etisserant@0: for i, connector in enumerate(instance["connectors"]): etisserant@0: connectors[i].SetPosition(wxPoint(*connector["position"])) etisserant@0: elif instance["type"] == "rightPowerRail": etisserant@0: rightpowerrail = LD_PowerRail(self, RIGHTRAIL, instance["id"], [True for i in range(len(instance["connectors"]))]) etisserant@0: rightpowerrail.SetPosition(instance["x"], instance["y"]) etisserant@0: self.Blocks.append(rightpowerrail) etisserant@0: self.Elements.append(rightpowerrail) etisserant@0: connectors = rightpowerrail.GetConnectors() etisserant@0: for i, connector in enumerate(instance["connectors"]): etisserant@0: connectors[i].SetPosition(wxPoint(*connector["position"])) etisserant@0: self.CreateWires(connectors[i], connector["links"], ids) etisserant@0: elif instance["type"] == "contact": etisserant@0: if instance["negated"]: etisserant@0: negated = instance["negated"] etisserant@0: else: etisserant@0: negated = False etisserant@0: if instance["edge"]: etisserant@0: edge = instance["edge"] etisserant@0: else: etisserant@0: edge = "none" etisserant@0: if negated and edge == "none": etisserant@0: contact_type = CONTACT_REVERSE etisserant@0: elif not negated and edge == "rising": etisserant@0: contact_type = CONTACT_RISING etisserant@0: elif not negated and edge == "falling": etisserant@0: contact_type = CONTACT_FALLING etisserant@0: else: etisserant@0: contact_type = CONTACT_NORMAL etisserant@0: contact = LD_Contact(self, contact_type, instance["name"], instance["id"]) etisserant@0: contact.SetPosition(instance["x"], instance["y"]) etisserant@0: self.Blocks.append(contact) etisserant@0: self.Elements.append(contact) etisserant@0: connectors = contact.GetConnectors() etisserant@0: connectors["input"].SetPosition(wxPoint(*instance["connectors"]["input"]["position"])) etisserant@0: self.CreateWires(connectors["input"], instance["connectors"]["input"]["links"], ids) etisserant@0: connectors["output"].SetPosition(wxPoint(*instance["connectors"]["output"]["position"])) etisserant@0: elif instance["type"] == "coil": etisserant@0: if instance["negated"]: etisserant@0: negated = instance["negated"] etisserant@0: else: etisserant@0: negated = False etisserant@0: if instance["storage"]: etisserant@0: storage = instance["storage"] etisserant@0: else: etisserant@0: storage = "none" etisserant@0: if negated and storage == "none": etisserant@0: coil_type = COIL_REVERSE etisserant@0: elif not negated and storage == "set": etisserant@0: coil_type = COIL_SET etisserant@0: elif not negated and storage == "reset": etisserant@0: coil_type = COIL_RESET etisserant@0: else: etisserant@0: coil_type = COIL_NORMAL etisserant@0: coil = LD_Coil(self, coil_type, instance["name"], instance["id"]) etisserant@0: coil.SetPosition(instance["x"], instance["y"]) etisserant@0: self.Blocks.append(coil) etisserant@0: self.Elements.append(coil) etisserant@0: connectors = coil.GetConnectors() etisserant@0: connectors["input"].SetPosition(wxPoint(*instance["connectors"]["input"]["position"])) etisserant@0: self.CreateWires(connectors["input"], instance["connectors"]["input"]["links"], ids) etisserant@0: connectors["output"].SetPosition(wxPoint(*instance["connectors"]["output"]["position"])) etisserant@0: elif instance["type"] == "step": etisserant@0: if instance["initial"]: etisserant@0: initial = instance["initial"] etisserant@0: else: etisserant@0: initial = False etisserant@0: step = SFC_Step(self, instance["name"], initial, instance["id"]) etisserant@0: step.SetPosition(instance["x"], instance["y"]) etisserant@0: step.SetSize(instance["width"], instance["height"]) etisserant@0: self.Blocks.append(step) etisserant@0: self.Elements.append(step) etisserant@0: if "output" in instance["connectors"]: etisserant@0: step.AddOutput() etisserant@0: if "action" in instance["connectors"]: etisserant@0: step.AddAction() etisserant@0: connectors = step.GetConnectors() etisserant@0: if connectors["input"]: etisserant@0: connectors["input"].SetPosition(wxPoint(*instance["connectors"]["input"]["position"])) etisserant@0: self.CreateWires(connectors["input"], instance["connectors"]["input"]["links"], ids) etisserant@0: if connectors["output"]: etisserant@0: connectors["output"].SetPosition(wxPoint(*instance["connectors"]["output"]["position"])) etisserant@0: if connectors["action"]: etisserant@0: connectors["action"].SetPosition(wxPoint(*instance["connectors"]["action"]["position"])) etisserant@0: elif instance["type"] == "transition": etisserant@0: transition = SFC_Transition(self, instance["condition_type"], instance["condition"], instance["id"]) etisserant@0: transition.SetPosition(instance["x"], instance["y"]) etisserant@0: self.Blocks.append(transition) etisserant@0: self.Elements.append(transition) etisserant@0: connectors = transition.GetConnectors() etisserant@0: connectors["input"].SetPosition(wxPoint(*instance["connectors"]["input"]["position"])) etisserant@0: self.CreateWires(connectors["input"], instance["connectors"]["input"]["links"], ids) etisserant@0: connectors["output"].SetPosition(wxPoint(*instance["connectors"]["output"]["position"])) etisserant@0: elif instance["type"] in ["selectionDivergence", "selectionConvergence", "simultaneousDivergence", "simultaneousConvergence"]: etisserant@0: if instance["type"] == "selectionDivergence": etisserant@0: divergence = SFC_Divergence(self, SELECTION_DIVERGENCE, etisserant@0: len(instance["connectors"]["outputs"]), instance["id"]) etisserant@0: elif instance["type"] == "selectionConvergence": etisserant@0: divergence = SFC_Divergence(self, SELECTION_CONVERGENCE, etisserant@0: len(instance["connectors"]["inputs"]), instance["id"]) etisserant@0: elif instance["type"] == "simultaneousDivergence": etisserant@0: divergence = SFC_Divergence(self, SIMULTANEOUS_DIVERGENCE, etisserant@0: len(instance["connectors"]["outputs"]), instance["id"]) etisserant@0: else: etisserant@0: divergence = SFC_Divergence(self, SIMULTANEOUS_CONVERGENCE, etisserant@0: len(instance["connectors"]["inputs"]), instance["id"]) etisserant@0: divergence.SetPosition(instance["x"], instance["y"]) etisserant@0: divergence.SetSize(instance["width"], instance["height"]) etisserant@0: self.Blocks.append(divergence) etisserant@0: self.Elements.append(divergence) etisserant@0: connectors = divergence.GetConnectors() etisserant@0: for i, input_connector in enumerate(instance["connectors"]["inputs"]): etisserant@0: connector = connectors["inputs"][i] etisserant@0: connector.SetPosition(wxPoint(*input_connector["position"])) etisserant@0: self.CreateWires(connector, input_connector["links"], ids) etisserant@0: for i, output_connector in enumerate(instance["connectors"]["outputs"]): etisserant@0: connector = connectors["outputs"][i] etisserant@0: connector.SetPosition(wxPoint(*output_connector["position"])) etisserant@0: elif instance["type"] == "jump": etisserant@0: jump = SFC_Jump(self, instance["target"], instance["id"]) etisserant@0: jump.SetPosition(instance["x"], instance["y"]) etisserant@0: self.Blocks.append(jump) etisserant@0: self.Elements.append(jump) etisserant@0: connector = jump.GetConnector() etisserant@0: connector.SetPosition(wxPoint(*instance["connector"]["position"])) etisserant@0: self.CreateWires(connector, instance["connector"]["links"], ids) etisserant@0: elif instance["type"] == "actionBlock": etisserant@0: actionBlock = SFC_ActionBlock(self, instance["actions"], instance["id"]) etisserant@0: actionBlock.SetPosition(instance["x"], instance["y"]) etisserant@0: actionBlock.SetSize(instance["width"], instance["height"]) etisserant@0: self.Blocks.append(actionBlock) etisserant@0: self.Elements.append(actionBlock) etisserant@0: connector = actionBlock.GetConnector() etisserant@0: connector.SetPosition(wxPoint(*instance["connector"]["position"])) etisserant@0: self.CreateWires(connector, instance["connector"]["links"], ids) etisserant@0: else: etisserant@0: if instance["name"] != None: etisserant@0: block = FBD_Block(self, instance["type"], instance["name"], instance["id"], len(instance["connectors"]["inputs"])) etisserant@0: else: etisserant@0: block = FBD_Block(self, instance["type"], "", instance["id"], len(instance["connectors"]["inputs"])) etisserant@0: block.SetPosition(instance["x"], instance["y"]) etisserant@0: block.SetSize(instance["width"], instance["height"]) etisserant@0: self.Blocks.append(block) etisserant@0: self.Elements.append(block) etisserant@0: connectors = block.GetConnectors() etisserant@0: for i, input_connector in enumerate(instance["connectors"]["inputs"]): etisserant@0: connector = connectors["inputs"][i] etisserant@0: connector.SetPosition(wxPoint(*input_connector["position"])) etisserant@0: if input_connector["negated"]: etisserant@0: connector.SetNegated(True) etisserant@0: if input_connector["edge"]: etisserant@0: connector.SetEdge(input_connector["edge"]) etisserant@0: self.CreateWires(connector, input_connector["links"], ids) etisserant@0: for i, output_connector in enumerate(instance["connectors"]["outputs"]): etisserant@0: connector = connectors["outputs"][i] etisserant@0: if output_connector["negated"]: etisserant@0: connector.SetNegated(True) etisserant@0: if output_connector["edge"]: etisserant@0: connector.SetEdge(output_connector["edge"]) etisserant@0: connector.SetPosition(wxPoint(*output_connector["position"])) etisserant@0: etisserant@0: def CreateWires(self, start_connector, links, ids): etisserant@0: for link in links: etisserant@0: refLocalId = link["refLocalId"] etisserant@0: if refLocalId != None: etisserant@0: if refLocalId not in ids: etisserant@0: new_instance = self.Controler.GetCurrentElementEditingInstanceInfos(refLocalId) etisserant@0: if new_instance: etisserant@0: self.loadInstance(new_instance, ids) etisserant@0: connected = self.FindElementById(refLocalId) etisserant@0: if connected: etisserant@0: points = link["points"] lbessard@27: end_connector = connected.GetConnector(wxPoint(points[-1][0], points[-1][1]), link["formalParameter"]) etisserant@0: if end_connector: etisserant@0: wire = Wire(self) etisserant@0: wire.SetPoints(points) etisserant@0: start_connector.Connect((wire, 0), False) etisserant@0: end_connector.Connect((wire, -1), False) etisserant@0: wire.ConnectStartPoint(None, start_connector) etisserant@0: wire.ConnectEndPoint(None, end_connector) etisserant@0: self.Wires.append(wire) etisserant@0: self.Elements.append(wire) etisserant@0: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Search Element functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: etisserant@0: def FindBlock(self, pos): etisserant@0: for block in self.Blocks: etisserant@0: if block.HitTest(pos) or block.TestHandle(pos) != (0, 0): etisserant@0: return block etisserant@0: return None etisserant@0: etisserant@0: def FindWire(self, pos): etisserant@0: for wire in self.Wires: etisserant@0: if wire.HitTest(pos) or wire.TestHandle(pos) != (0, 0): etisserant@0: return wire etisserant@0: return None etisserant@0: etisserant@0: def FindElement(self, pos, exclude_group = False): etisserant@0: if self.SelectedElement and not (exclude_group and isinstance(self.SelectedElement, Graphic_Group)): etisserant@0: if self.SelectedElement.HitTest(pos) or self.SelectedElement.TestHandle(pos) != (0, 0): etisserant@0: return self.SelectedElement etisserant@0: for element in self.Elements: etisserant@0: if element.HitTest(pos) or element.TestHandle(pos) != (0, 0): etisserant@0: return element etisserant@0: return None etisserant@0: lbessard@3: def FindBlockConnector(self, pos, exclude = True): etisserant@0: for block in self.Blocks: lbessard@3: result = block.TestConnector(pos, exclude) etisserant@0: if result: etisserant@0: return result etisserant@0: return None etisserant@0: etisserant@0: def FindElementById(self, id): etisserant@0: for element in self.Elements: etisserant@0: if element.GetId() == id: etisserant@0: return element etisserant@0: return None etisserant@0: etisserant@0: def SearchElements(self, bbox): etisserant@0: elements = [] etisserant@0: for element in self.Elements: etisserant@0: element_bbox = element.GetBoundingBox() etisserant@0: if element_bbox.x >= bbox.x and element_bbox.y >= bbox.y and element_bbox.x + element_bbox.width <= bbox.x + bbox.width and element_bbox.y + element_bbox.height <= bbox.y + bbox.height: etisserant@0: elements.append(element) etisserant@0: return elements etisserant@0: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Popup menu functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: etisserant@0: def PopupBlockMenu(self, connector = None): lbessard@13: type = self.Controler.GetCurrentElementEditingType() etisserant@0: self.ContextualMenu.FindItemByPosition(0).Enable(connector != None) etisserant@0: self.ContextualMenu.FindItemByPosition(1).Enable(connector != None) lbessard@13: self.ContextualMenu.FindItemByPosition(2).Enable(connector != None and type != "function") lbessard@13: self.ContextualMenu.FindItemByPosition(3).Enable(connector != None and type != "function") etisserant@0: self.ContextualMenu.FindItemByPosition(5).Enable(False) etisserant@0: self.ContextualMenu.FindItemByPosition(6).Enable(False) etisserant@0: self.ContextualMenu.FindItemByPosition(8).Enable(False) etisserant@0: self.ContextualMenu.FindItemByPosition(9).Enable(False) etisserant@0: if connector: etisserant@0: if connector.IsNegated(): etisserant@0: self.ContextualMenu.FindItemByPosition(1).Check(True) etisserant@0: elif connector.GetEdge() == "rising": etisserant@0: self.ContextualMenu.FindItemByPosition(2).Check(True) etisserant@0: elif connector.GetEdge() == "falling": etisserant@0: self.ContextualMenu.FindItemByPosition(3).Check(True) etisserant@0: else: etisserant@0: self.ContextualMenu.FindItemByPosition(0).Check(True) etisserant@0: self.PopupMenu(self.ContextualMenu) etisserant@0: etisserant@0: def PopupVariableMenu(self, connector = None): etisserant@0: self.ContextualMenu.FindItemByPosition(0).Enable(connector != None) etisserant@0: self.ContextualMenu.FindItemByPosition(1).Enable(connector != None) etisserant@0: self.ContextualMenu.FindItemByPosition(2).Enable(False) etisserant@0: self.ContextualMenu.FindItemByPosition(3).Enable(False) etisserant@0: self.ContextualMenu.FindItemByPosition(5).Enable(False) etisserant@0: self.ContextualMenu.FindItemByPosition(6).Enable(False) etisserant@0: self.ContextualMenu.FindItemByPosition(8).Enable(False) etisserant@0: self.ContextualMenu.FindItemByPosition(9).Enable(False) etisserant@0: if connector: etisserant@0: if connector.IsNegated(): etisserant@0: self.ContextualMenu.FindItemByPosition(1).Check(True) etisserant@0: else: etisserant@0: self.ContextualMenu.FindItemByPosition(0).Check(True) etisserant@0: self.PopupMenu(self.ContextualMenu) etisserant@0: etisserant@0: def PopupWireMenu(self): etisserant@0: self.ContextualMenu.FindItemByPosition(0).Enable(False) etisserant@0: self.ContextualMenu.FindItemByPosition(1).Enable(False) etisserant@0: self.ContextualMenu.FindItemByPosition(2).Enable(False) etisserant@0: self.ContextualMenu.FindItemByPosition(3).Enable(False) etisserant@0: self.ContextualMenu.FindItemByPosition(5).Enable(True) etisserant@0: self.ContextualMenu.FindItemByPosition(6).Enable(True) etisserant@0: self.ContextualMenu.FindItemByPosition(8).Enable(False) etisserant@0: self.ContextualMenu.FindItemByPosition(9).Enable(False) etisserant@0: self.PopupMenu(self.ContextualMenu) etisserant@0: etisserant@0: def PopupDivergenceMenu(self, connector): etisserant@0: self.ContextualMenu.FindItemByPosition(0).Enable(False) etisserant@0: self.ContextualMenu.FindItemByPosition(1).Enable(False) etisserant@0: self.ContextualMenu.FindItemByPosition(2).Enable(False) etisserant@0: self.ContextualMenu.FindItemByPosition(3).Enable(False) etisserant@0: self.ContextualMenu.FindItemByPosition(5).Enable(False) etisserant@0: self.ContextualMenu.FindItemByPosition(6).Enable(False) etisserant@0: self.ContextualMenu.FindItemByPosition(8).Enable(True) etisserant@0: self.ContextualMenu.FindItemByPosition(9).Enable(connector) etisserant@0: self.PopupMenu(self.ContextualMenu) etisserant@0: etisserant@0: def PopupDefaultMenu(self): etisserant@0: self.ContextualMenu.FindItemByPosition(0).Enable(False) etisserant@0: self.ContextualMenu.FindItemByPosition(1).Enable(False) etisserant@0: self.ContextualMenu.FindItemByPosition(2).Enable(False) etisserant@0: self.ContextualMenu.FindItemByPosition(3).Enable(False) etisserant@0: self.ContextualMenu.FindItemByPosition(5).Enable(False) etisserant@0: self.ContextualMenu.FindItemByPosition(6).Enable(False) etisserant@0: self.ContextualMenu.FindItemByPosition(8).Enable(False) etisserant@0: self.ContextualMenu.FindItemByPosition(9).Enable(False) etisserant@0: self.PopupMenu(self.ContextualMenu) etisserant@0: etisserant@0: def EditCommentContent(self, comment): etisserant@0: dialog = wxTextEntryDialog(self.Parent, "Edit comment", "Please enter comment text", comment.GetContent(), wxOK|wxCANCEL|wxTE_MULTILINE) etisserant@0: if dialog.ShowModal() == wxID_OK: etisserant@0: value = dialog.GetValue() etisserant@0: comment.SetContent(value) etisserant@0: infos = {"content" : value} etisserant@0: infos["width"], infos["height"] = comment.GetSize() etisserant@0: self.Controler.SetCurrentElementEditingCommentInfos(comment.GetId(), infos) etisserant@0: self.Refresh() etisserant@0: dialog.Destroy() etisserant@0: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Menu items functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: etisserant@0: def OnNoModifierMenu(self, event): etisserant@0: if self.SelectedElement and self.SelectedElement in self.Blocks: etisserant@0: self.SelectedElement.SetConnectorNegated(False) etisserant@0: event.Skip() etisserant@0: etisserant@0: def OnNegatedMenu(self, event): etisserant@0: if self.SelectedElement and self.SelectedElement in self.Blocks: etisserant@0: self.SelectedElement.SetConnectorNegated(True) etisserant@0: event.Skip() etisserant@0: etisserant@0: def OnRisingEdgeMenu(self, event): etisserant@0: if self.SelectedElement and self.SelectedElement in self.Blocks: etisserant@0: self.SelectedElement.SetConnectorEdge("rising") etisserant@0: event.Skip() etisserant@0: etisserant@0: def OnFallingEdgeMenu(self, event): etisserant@0: if self.SelectedElement and self.SelectedElement in self.Blocks: etisserant@0: self.SelectedElement.SetConnectorEdge("falling") etisserant@0: event.Skip() etisserant@0: etisserant@0: def OnAddSegmentMenu(self, event): etisserant@0: if self.SelectedElement and self.SelectedElement in self.Wires: etisserant@0: self.SelectedElement.AddSegment() etisserant@0: event.Skip() etisserant@0: etisserant@0: def OnDeleteSegmentMenu(self, event): etisserant@0: if self.SelectedElement and self.SelectedElement in self.Wires: etisserant@0: self.SelectedElement.DeleteSegment() etisserant@0: event.Skip() etisserant@0: etisserant@0: def OnAddBranchMenu(self, event): etisserant@0: if self.SelectedElement and self.SelectedElement in self.Blocks: etisserant@0: self.AddDivergenceBranch(self.SelectedElement) etisserant@0: event.Skip() etisserant@0: etisserant@0: def OnDeleteBranchMenu(self, event): etisserant@0: if self.SelectedElement and self.SelectedElement in self.Blocks: etisserant@0: self.RemoveDivergenceBranch(self.SelectedElement) etisserant@0: event.Skip() etisserant@0: etisserant@0: def OnDeleteMenu(self, event): etisserant@0: if self.SelectedElement: etisserant@0: self.SelectedElement.Delete() etisserant@0: self.SelectedElement = None etisserant@0: event.Skip() etisserant@0: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Mouse event functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: etisserant@0: def OnViewerLeftDown(self, event): lbessard@27: if self.Mode == MODE_SELECTION: lbessard@27: dc = self.GetLogicalDC() lbessard@27: pos = event.GetLogicalPosition(dc) lbessard@27: if event.ControlDown() and self.SelectedElement: lbessard@27: element = self.FindElement(pos, True) lbessard@27: if element: lbessard@27: if isinstance(self.SelectedElement, Graphic_Group): lbessard@27: self.SelectedElement.SetSelected(False) lbessard@27: self.SelectedElement.SelectElement(element) lbessard@27: elif self.SelectedElement: lbessard@27: group = Graphic_Group(self) lbessard@27: group.SelectElement(self.SelectedElement) lbessard@27: group.SelectElement(element) lbessard@27: self.SelectedElement = group lbessard@27: elements = self.SelectedElement.GetElements() lbessard@27: if len(elements) == 0: lbessard@27: self.SelectedElement = element lbessard@27: elif len(elements) == 1: lbessard@27: self.SelectedElement = elements[0] lbessard@27: self.SelectedElement.SetSelected(True) lbessard@27: else: lbessard@27: element = self.FindElement(pos) lbessard@27: if self.SelectedElement and self.SelectedElement != element: lbessard@27: self.SelectedElement.SetSelected(False) lbessard@27: self.SelectedElement = None lbessard@27: self.Refresh() lbessard@27: if element: lbessard@27: self.SelectedElement = element lbessard@27: self.SelectedElement.OnLeftDown(event, dc, self.Scaling) lbessard@27: self.Refresh() lbessard@27: else: lbessard@27: self.rubberBand.Reset() lbessard@27: 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) lbessard@27: elif self.Mode == MODE_WIRE: lbessard@27: pos = GetScaledEventPosition(event, self.GetLogicalDC(), self.Scaling) lbessard@27: connector = self.FindBlockConnector(pos) lbessard@27: if connector: lbessard@27: if (connector.GetDirection() == EAST): lbessard@27: wire = Wire(self, [wxPoint(pos.x, pos.y), EAST], [wxPoint(pos.x, pos.y), WEST]) lbessard@27: else: lbessard@27: wire = Wire(self, [wxPoint(pos.x, pos.y), WEST], [wxPoint(pos.x, pos.y), EAST]) lbessard@27: wire.oldPos = pos lbessard@27: wire.Handle = (HANDLE_POINT, 0) lbessard@27: wire.ProcessDragging(0, 0) lbessard@27: wire.Handle = (HANDLE_POINT, 1) lbessard@27: self.Wires.append(wire) lbessard@27: self.Elements.append(wire) lbessard@27: if self.SelectedElement: lbessard@27: self.SelectedElement.SetSelected(False) lbessard@27: self.SelectedElement = wire lbessard@27: elif self.SelectedElement: lbessard@27: self.SelectedElement.SetSelected(False) lbessard@27: self.SelectedElement = None lbessard@27: self.Refresh() etisserant@0: event.Skip() etisserant@0: etisserant@0: def OnViewerLeftUp(self, event): lbessard@27: if self.rubberBand.IsShown(): lbessard@27: if self.Mode == MODE_SELECTION: lbessard@27: elements = self.SearchElements(self.rubberBand.GetCurrentExtent()) lbessard@27: self.rubberBand.OnLeftUp(event, self.GetLogicalDC(), self.Scaling) lbessard@27: if len(elements) > 0: lbessard@27: self.SelectedElement = Graphic_Group(self) lbessard@27: self.SelectedElement.SetElements(elements) lbessard@27: self.SelectedElement.SetSelected(True) lbessard@27: self.Refresh() 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@27: wxCallAfter(self.AddNewBlock, bbox) lbessard@27: elif self.Mode == MODE_VARIABLE: lbessard@27: wxCallAfter(self.AddNewVariable, bbox) lbessard@27: elif self.Mode == MODE_CONNECTION: lbessard@27: wxCallAfter(self.AddNewConnection, bbox) lbessard@27: elif self.Mode == MODE_COMMENT: lbessard@27: wxCallAfter(self.AddNewComment, bbox) lbessard@27: elif self.Mode == MODE_CONTACT: lbessard@27: wxCallAfter(self.AddNewContact, bbox) lbessard@27: elif self.Mode == MODE_COIL: lbessard@27: wxCallAfter(self.AddNewContact, bbox) lbessard@27: elif self.Mode == MODE_POWERRAIL: lbessard@27: wxCallAfter(self.AddNewPowerRail, bbox) lbessard@27: elif self.Mode == MODE_INITIALSTEP: lbessard@27: wxCallAfter(self.AddNewInitialStep, bbox) lbessard@27: elif self.Mode == MODE_STEP: lbessard@27: wxCallAfter(self.AddNewStep, bbox) lbessard@27: elif self.Mode == MODE_TRANSITION: lbessard@27: wxCallAfter(self.AddNewTransition, bbox) lbessard@27: elif self.Mode == MODE_DIVERGENCE: lbessard@27: wxCallAfter(self.AddNewDivergence, bbox) lbessard@27: elif self.Mode == MODE_JUMP: lbessard@27: wxCallAfter(self.AddNewJump, bbox) lbessard@27: elif self.Mode == MODE_ACTION: lbessard@27: wxCallAfter(self.AddNewActionBlock, bbox) lbessard@27: elif self.Mode == MODE_SELECTION and self.SelectedElement: lbessard@27: self.SelectedElement.OnLeftUp(event, self.GetLogicalDC(), self.Scaling) lbessard@27: wxCallAfter(self.SetCursor, wxNullCursor) lbessard@27: self.ReleaseMouse() lbessard@27: self.Refresh() lbessard@27: elif self.Mode == MODE_WIRE and self.SelectedElement: lbessard@27: dc = self.GetLogicalDC() lbessard@27: pos = GetScaledEventPosition(event, dc, self.Scaling) lbessard@27: connector = self.FindBlockConnector(pos, False) lbessard@27: if connector and connector != self.SelectedElement.StartConnected: lbessard@27: self.SelectedElement.ResetPoints() lbessard@27: self.SelectedElement.OnMotion(event, dc, self.Scaling) lbessard@27: self.SelectedElement.GeneratePoints() lbessard@27: self.SelectedElement.RefreshModel() lbessard@27: self.SelectedElement.SetSelected(True) lbessard@27: else: lbessard@27: self.SelectedElement.Delete() lbessard@27: self.SelectedElement = None lbessard@27: self.Refresh() lbessard@27: if not self.SavedMode: lbessard@27: wxCallAfter(self.Parent.ResetCurrentMode) etisserant@0: event.Skip() etisserant@0: etisserant@0: def OnViewerRightUp(self, event): lbessard@27: pos = event.GetPosition() lbessard@27: element = self.FindElement(pos) lbessard@27: if element: lbessard@27: if self.SelectedElement and self.SelectedElement != element: lbessard@27: self.SelectedElement.SetSelected(False) lbessard@27: self.SelectedElement = element lbessard@27: self.SelectedElement.SetSelected(True) lbessard@27: self.SelectedElement.OnRightUp(event, self.GetLogicalDC(), self.Scaling) lbessard@27: wxCallAfter(self.SetCursor, wxNullCursor) lbessard@27: self.ReleaseMouse() lbessard@27: self.Refresh() etisserant@0: event.Skip() etisserant@0: etisserant@0: def OnViewerLeftDClick(self, event): lbessard@27: if self.Mode == MODE_SELECTION and self.SelectedElement: lbessard@27: self.SelectedElement.OnLeftDClick(event, self.GetLogicalDC(), self.Scaling) lbessard@27: self.Refresh() etisserant@0: event.Skip() etisserant@0: etisserant@0: def OnViewerMotion(self, event): lbessard@27: if self.rubberBand.IsShown(): lbessard@27: self.rubberBand.OnMotion(event, self.GetLogicalDC(), self.Scaling) lbessard@27: elif self.Mode == MODE_SELECTION and self.SelectedElement: lbessard@27: self.SelectedElement.OnMotion(event, self.GetLogicalDC(), self.Scaling) lbessard@27: self.Refresh() lbessard@27: elif self.Mode == MODE_WIRE and self.SelectedElement: lbessard@27: dc = self.GetLogicalDC() lbessard@27: pos = GetScaledEventPosition(event, dc, self.Scaling) lbessard@27: connector = self.FindBlockConnector(pos, False) lbessard@27: if not connector or self.SelectedElement.EndConnected == None: lbessard@27: self.SelectedElement.ResetPoints() lbessard@27: self.SelectedElement.OnMotion(event, dc, self.Scaling) lbessard@27: self.SelectedElement.GeneratePoints() lbessard@27: self.Refresh() lbessard@27: if (event.Dragging() and self.SelectedElement) or self.rubberBand.IsShown(): lbessard@27: position = event.GetPosition() lbessard@27: move_window = wxPoint() lbessard@27: window_size = self.GetClientSize() lbessard@27: xstart, ystart = self.GetViewStart() lbessard@27: if position.x < SCROLL_ZONE and xstart > 0: lbessard@27: move_window.x = -1 lbessard@27: elif position.x > window_size[0] - SCROLL_ZONE: lbessard@27: move_window.x = 1 lbessard@27: if position.y < SCROLL_ZONE and ystart > 0: lbessard@27: move_window.y = -1 lbessard@27: elif position.y > window_size[1] - SCROLL_ZONE: lbessard@27: move_window.y = 1 lbessard@27: if move_window.x != 0 or move_window.y != 0: lbessard@27: self.Scroll(xstart + move_window.x, ystart + move_window.y) lbessard@27: self.RefreshScrollBar() etisserant@0: event.Skip() etisserant@0: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Keyboard event functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: etisserant@0: def OnChar(self, event): lbessard@27: keycode = event.GetKeyCode() lbessard@27: if self.Scaling: lbessard@27: scaling = self.Scaling lbessard@27: else: lbessard@27: scaling = (8, 8) lbessard@27: if keycode == WXK_DELETE and self.SelectedElement: lbessard@27: self.SelectedElement.Clean() lbessard@27: self.SelectedElement.Delete() lbessard@27: self.SelectedElement = None lbessard@27: elif keycode == WXK_LEFT and self.SelectedElement: lbessard@27: self.SelectedElement.Move(-scaling[0], 0) lbessard@27: elif keycode == WXK_RIGHT and self.SelectedElement: lbessard@27: self.SelectedElement.Move(scaling[0], 0) lbessard@27: elif keycode == WXK_UP and self.SelectedElement: lbessard@27: self.SelectedElement.Move(0, -scaling[1]) lbessard@27: elif keycode == WXK_DOWN and self.SelectedElement: lbessard@27: self.SelectedElement.Move(0, scaling[1]) lbessard@27: self.Refresh() lbessard@27: event.Skip() lbessard@27: lbessard@27: #------------------------------------------------------------------------------- lbessard@27: # Model adding functions lbessard@27: #------------------------------------------------------------------------------- lbessard@27: lbessard@27: def AddNewBlock(self, bbox): lbessard@27: dialog = BlockPropertiesDialog(self.Parent) lbessard@27: dialog.SetBlockList(self.Controler.GetBlockTypes()) lbessard@27: dialog.SetMinBlockSize((bbox.width, bbox.height)) lbessard@27: if dialog.ShowModal() == wxID_OK: lbessard@27: id = self.GetNewId() lbessard@27: values = dialog.GetValues() lbessard@27: if "name" in values: lbessard@27: block = FBD_Block(self, values["type"], values["name"], id, values["extension"]) lbessard@27: else: lbessard@27: block = FBD_Block(self, values["type"], "", id, values["extension"]) lbessard@27: block.SetPosition(bbox.x, bbox.y) lbessard@27: block.SetSize(values["width"], values["height"]) lbessard@27: self.Blocks.append(block) lbessard@27: self.Elements.append(block) lbessard@27: self.Controler.AddCurrentElementEditingBlock(id) lbessard@27: self.RefreshBlockModel(block) lbessard@27: self.Refresh() lbessard@27: dialog.Destroy() lbessard@27: lbessard@27: def AddNewVariable(self, bbox): lbessard@27: dialog = VariablePropertiesDialog(self.Parent) lbessard@27: dialog.SetMinVariableSize((bbox.width, bbox.height)) lbessard@27: varlist = [] lbessard@27: vars = self.Controler.GetCurrentElementEditingInterfaceVars() lbessard@27: if vars: lbessard@27: for var in vars: lbessard@27: varlist.append((var["Name"], var["Class"], var["Type"])) lbessard@27: returntype = self.Controler.GetCurrentElementEditingInterfaceReturnType() lbessard@27: if returntype: lbessard@27: varlist.append((self.Controler.GetCurrentElementEditingName(), "Output", returntype)) lbessard@27: dialog.SetVariables(varlist) lbessard@27: if dialog.ShowModal() == wxID_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@27: variable.SetSize(values["width"], values["height"]) lbessard@27: self.Blocks.append(variable) lbessard@27: self.Elements.append(variable) lbessard@27: self.Controler.AddCurrentElementEditingVariable(id, values["type"]) lbessard@27: self.RefreshVariableModel(variable) lbessard@27: self.Refresh() lbessard@27: dialog.Destroy() lbessard@27: lbessard@27: def AddNewConnection(self, bbox): lbessard@27: dialog = ConnectionPropertiesDialog(self.Parent) lbessard@27: dialog.SetMinConnectionSize((bbox.width, bbox.height)) lbessard@27: if dialog.ShowModal() == wxID_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@27: connection.SetSize(values["width"], values["height"]) lbessard@27: self.Blocks.append(connection) lbessard@27: self.Elements.append(connection) lbessard@27: self.Controler.AddCurrentElementEditingConnection(id, values["type"]) lbessard@27: self.RefreshConnectionModel(connection) lbessard@27: self.Refresh() lbessard@27: dialog.Destroy() lbessard@27: lbessard@27: def AddNewComment(self, bbox): lbessard@27: dialog = wxTextEntryDialog(self.Parent, "Add a new comment", "Please enter comment text", "", wxOK|wxCANCEL|wxTE_MULTILINE) lbessard@27: if dialog.ShowModal() == wxID_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@27: comment.SetSize(max(min_width,bbox.width),max(min_height,bbox.height)) lbessard@27: self.Elements.append(comment) lbessard@27: self.Controler.AddCurrentElementEditingComment(id) lbessard@27: self.RefreshCommentModel(comment) lbessard@27: self.Refresh() lbessard@27: dialog.Destroy() lbessard@27: lbessard@27: def AddNewContact(self, bbox): lbessard@27: dialog = LDElementDialog(self.Parent, "contact") lbessard@27: varlist = [] lbessard@27: vars = self.Controler.GetCurrentElementEditingInterfaceVars() lbessard@27: if vars: lbessard@27: for var in vars: lbessard@27: if var["Class"] != "Output" and var["Type"] == "BOOL": lbessard@27: varlist.append(var["Name"]) lbessard@27: dialog.SetVariables(varlist) lbessard@27: dialog.SetValues({"name":"","type":CONTACT_NORMAL}) lbessard@27: dialog.SetElementSize(bbox.width, bbox.height) lbessard@27: if dialog.ShowModal() == wxID_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@27: contact.SetSize(values["width"], values["height"]) lbessard@27: self.Blocks.append(contact) lbessard@27: self.Elements.append(contact) lbessard@27: self.Controler.AddCurrentElementEditingContact(id) lbessard@27: self.RefreshContactModel(contact) lbessard@27: self.Refresh() lbessard@27: dialog.Destroy() lbessard@27: lbessard@27: def AddNewCoil(self, bbox): lbessard@27: dialog = LDElementDialog(self.Parent, "coil") lbessard@27: varlist = [] lbessard@27: vars = self.Controler.GetCurrentElementEditingInterfaceVars() 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@27: returntype = self.Controler.GetCurrentElementEditingInterfaceReturnType() lbessard@27: if returntype == "BOOL": lbessard@27: varlist.append(self.Controler.GetCurrentElementEditingName()) lbessard@27: dialog.SetVariables(varlist) lbessard@27: dialog.SetValues({"name":"","type":COIL_NORMAL}) lbessard@27: dialog.SetElementSize(bbox.width, bbox.height) lbessard@27: if dialog.ShowModal() == wxID_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@27: coil.SetSize(values["width"], values["height"]) lbessard@27: self.Blocks.append(coil) lbessard@27: self.Elements.append(coil) lbessard@27: self.Controler.AddCurrentElementEditingCoil(id) lbessard@27: self.RefreshCoilModel(contact) lbessard@27: self.Refresh() lbessard@27: dialog.Destroy() lbessard@27: lbessard@27: def AddNewPowerRail(self, bbox): lbessard@27: dialog = LDPowerRailDialog(self.Parent) lbessard@27: dialog.SetMinSize((bbox.width, bbox.height)) lbessard@27: if dialog.ShowModal() == wxID_OK: lbessard@27: id = self.GetNewId() lbessard@27: values = dialog.GetValues() lbessard@27: powerrail = LD_PowerRail(self, values["type"], id, [True for i in xrange(values["number"])]) lbessard@27: powerrail.SetPosition(bbox.x, bbox.y) lbessard@27: powerrail.SetSize(values["width"], values["height"]) lbessard@27: self.Blocks.append(powerrail) lbessard@27: self.Elements.append(powerrail) lbessard@27: self.Controler.AddCurrentElementEditingPowerRail(id, values["type"]) lbessard@27: self.RefreshPowerRailModel(powerrail) lbessard@27: self.Refresh() lbessard@27: dialog.Destroy() lbessard@27: lbessard@27: def AddNewTransition(self, bbox): lbessard@27: dialog = TransitionContentDialog(self.Parent) lbessard@27: dialog.SetTransitions(self.Controler.GetCurrentElementEditingTransitions()) lbessard@27: if dialog.ShowModal() == wxID_OK: lbessard@27: id = self.GetNewId() lbessard@27: values = dialog.GetValues() lbessard@27: transition = SFC_Transition(self, values["type"], values["value"], id) lbessard@27: transition.SetPosition(bbox.x, bbox.y) lbessard@27: min_width, min_height = transition.GetMinSize() lbessard@27: transition.SetSize(max(bbox.width, min_width), max(bbox.height, min_height)) lbessard@27: self.Blocks.append(transition) lbessard@27: self.Elements.append(transition) lbessard@27: self.Controler.AddCurrentElementEditingTransition(id) lbessard@27: self.RefreshTransitionModel(transition) lbessard@27: self.Refresh() lbessard@27: dialog.Destroy() lbessard@27: lbessard@27: def AddNewDivergence(self, bbox): lbessard@27: dialog = DivergenceCreateDialog(self.Parent) lbessard@27: dialog.SetMinSize((bbox.width, bbox.height)) lbessard@27: if dialog.ShowModal() == wxID_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@27: min_width, min_height = divergence.GetMinSize() lbessard@27: divergence.SetSize(max(bbox.width, min_width), max(bbox.height, min_height)) lbessard@27: self.Blocks.append(divergence) lbessard@27: self.Elements.append(divergence) lbessard@27: self.Controler.AddCurrentElementEditingDivergence(id, values["type"]) lbessard@27: self.RefreshDivergenceModel(divergence) lbessard@27: self.Refresh() lbessard@27: dialog.Destroy() etisserant@0: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Model update functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: etisserant@0: def RefreshBlockModel(self, block): etisserant@0: blockid = block.GetId() etisserant@0: infos = {} etisserant@0: infos["type"] = block.GetType() etisserant@0: infos["name"] = block.GetName() etisserant@0: infos["x"], infos["y"] = block.GetPosition() etisserant@0: infos["width"], infos["height"] = block.GetSize() etisserant@0: infos["connectors"] = block.GetConnectors() etisserant@0: self.Controler.SetCurrentElementEditingBlockInfos(blockid, infos) etisserant@0: etisserant@0: def RefreshVariableModel(self, variable): etisserant@0: variableid = variable.GetId() etisserant@0: infos = {} etisserant@0: infos["name"] = variable.GetName() etisserant@0: infos["x"], infos["y"] = variable.GetPosition() etisserant@0: infos["width"], infos["height"] = variable.GetSize() etisserant@0: infos["connectors"] = variable.GetConnectors() etisserant@0: self.Controler.SetCurrentElementEditingVariableInfos(variableid, infos) etisserant@0: etisserant@0: def RefreshConnectionModel(self, connection): etisserant@0: connectionid = connection.GetId() etisserant@0: infos = {} etisserant@0: infos["name"] = connection.GetName() etisserant@0: infos["x"], infos["y"] = connection.GetPosition() etisserant@0: infos["width"], infos["height"] = connection.GetSize() etisserant@0: infos["connector"] = connection.GetConnector() etisserant@0: self.Controler.SetCurrentElementEditingConnectionInfos(connectionid, infos) etisserant@0: etisserant@0: def RefreshCommentModel(self, comment): etisserant@0: commentid = comment.GetId() etisserant@0: infos = {} etisserant@0: infos["content"] = comment.GetContent() etisserant@0: infos["x"], infos["y"] = comment.GetPosition() etisserant@0: infos["width"], infos["height"] = comment.GetSize() etisserant@0: self.Controler.SetCurrentElementEditingCommentInfos(commentid, infos) etisserant@0: etisserant@0: def RefreshPowerRailModel(self, powerrail): etisserant@0: powerrailid = powerrail.GetId() etisserant@0: infos = {} etisserant@0: infos["x"], infos["y"] = powerrail.GetPosition() etisserant@0: infos["width"], infos["height"] = powerrail.GetSize() etisserant@0: infos["connectors"] = powerrail.GetConnectors() etisserant@0: self.Controler.SetCurrentElementEditingPowerRailInfos(powerrailid, infos) etisserant@0: etisserant@0: def RefreshContactModel(self, contact): etisserant@0: contactid = contact.GetId() etisserant@0: infos = {} etisserant@0: infos["name"] = contact.GetName() etisserant@0: infos["type"] = contact.GetType() etisserant@0: infos["x"], infos["y"] = contact.GetPosition() etisserant@0: infos["width"], infos["height"] = contact.GetSize() etisserant@0: infos["connectors"] = contact.GetConnectors() etisserant@0: self.Controler.SetCurrentElementEditingContactInfos(contactid, infos) etisserant@0: etisserant@0: def RefreshCoilModel(self, coil): etisserant@0: coilid = coil.GetId() etisserant@0: infos = {} etisserant@0: infos["name"] = coil.GetName() etisserant@0: infos["type"] = coil.GetType() etisserant@0: infos["x"], infos["y"] = coil.GetPosition() etisserant@0: infos["width"], infos["height"] = coil.GetSize() etisserant@0: infos["connectors"] = coil.GetConnectors() etisserant@0: self.Controler.SetCurrentElementEditingCoilInfos(coilid, infos) etisserant@0: etisserant@0: def RefreshStepModel(self, step): etisserant@0: stepid = step.GetId() etisserant@0: infos = {} etisserant@0: infos["name"] = step.GetName() etisserant@0: infos["initial"] = step.GetInitial() etisserant@0: infos["x"], infos["y"] = step.GetPosition() etisserant@0: infos["width"], infos["height"] = step.GetSize() etisserant@0: infos["connectors"] = step.GetConnectors() etisserant@0: self.Controler.SetCurrentElementEditingStepInfos(stepid, infos) etisserant@0: etisserant@0: def RefreshTransitionModel(self, transition): etisserant@0: transitionid = transition.GetId() etisserant@0: infos = {} etisserant@0: infos["type"] = transition.GetType() etisserant@0: infos["condition"] = transition.GetCondition() etisserant@0: infos["x"], infos["y"] = transition.GetPosition() etisserant@0: infos["width"], infos["height"] = transition.GetSize() etisserant@0: infos["connectors"] = transition.GetConnectors() etisserant@0: self.Controler.SetCurrentElementEditingTransitionInfos(transitionid, infos) etisserant@0: etisserant@0: def RefreshDivergenceModel(self, divergence): etisserant@0: divergenceid = divergence.GetId() etisserant@0: infos = {} etisserant@0: infos["x"], infos["y"] = divergence.GetPosition() etisserant@0: infos["width"], infos["height"] = divergence.GetSize() etisserant@0: infos["connectors"] = divergence.GetConnectors() etisserant@0: self.Controler.SetCurrentElementEditingDivergenceInfos(divergenceid, infos) etisserant@0: etisserant@0: def RefreshJumpModel(self, jump): etisserant@0: jumpid = jump.GetId() etisserant@0: infos = {} etisserant@0: infos["target"] = jump.GetTarget() etisserant@0: infos["x"], infos["y"] = jump.GetPosition() etisserant@0: infos["width"], infos["height"] = jump.GetSize() etisserant@0: infos["connector"] = jump.GetConnector() etisserant@0: self.Controler.SetCurrentElementEditingJumpInfos(jumpid, infos) etisserant@0: etisserant@0: def RefreshActionBlockModel(self, actionblock): etisserant@0: actionblockid = actionblock.GetId() etisserant@0: infos = {} etisserant@0: infos["actions"] = actionblock.GetActions() etisserant@0: infos["x"], infos["y"] = actionblock.GetPosition() etisserant@0: infos["width"], infos["height"] = actionblock.GetSize() etisserant@0: infos["connector"] = actionblock.GetConnector() etisserant@0: self.Controler.SetCurrentElementEditingActionBlockInfos(actionblockid, infos) etisserant@0: lbessard@27: lbessard@27: #------------------------------------------------------------------------------- lbessard@27: # Model delete functions lbessard@27: #------------------------------------------------------------------------------- lbessard@27: lbessard@27: lbessard@27: def DeleteBlock(self, block): lbessard@27: elements = [] lbessard@27: for output in block.GetConnectors()["outputs"]: lbessard@27: for element in output.GetConnectedBlocks(): lbessard@27: if element not in elements: lbessard@27: elements.append(element) lbessard@27: block.Clean() lbessard@27: self.Blocks.remove(block) lbessard@27: self.Elements.remove(block) lbessard@27: self.Controler.RemoveCurrentElementEditingInstance(block.GetId()) lbessard@27: for element in elements: lbessard@27: element.RefreshModel() lbessard@27: lbessard@27: def DeleteVariable(self, variable): lbessard@27: connectors = variable.GetConnectors() lbessard@27: if connectors["output"]: lbessard@27: elements = connectors["output"].GetConnectedBlocks() lbessard@27: else: lbessard@27: elements = [] lbessard@27: variable.Clean() lbessard@27: self.Blocks.remove(variable) lbessard@27: self.Elements.remove(variable) lbessard@27: self.Controler.RemoveCurrentElementEditingInstance(variable.GetId()) lbessard@27: for element in elements: lbessard@27: element.RefreshModel() lbessard@27: lbessard@27: def DeleteConnection(self, connection): lbessard@27: if connection.GetType() == CONTINUATION: lbessard@27: elements = connection.GetConnector().GetConnectedBlocks() lbessard@27: else: lbessard@27: elements = [] lbessard@27: connection.Clean() lbessard@27: self.Blocks.remove(connection) lbessard@27: self.Elements.remove(connection) lbessard@27: self.Controler.RemoveCurrentElementEditingInstance(connection.GetId()) lbessard@27: for element in elements: lbessard@27: element.RefreshModel() lbessard@27: lbessard@27: def DeleteComment(self, comment): lbessard@27: self.Elements.remove(comment) lbessard@27: self.Controler.RemoveCurrentElementEditingInstance(comment.GetId()) lbessard@27: lbessard@27: def DeleteWire(self, wire): lbessard@27: connected = wire.GetConnected() lbessard@27: wire.Clean() lbessard@27: self.Wires.remove(wire) lbessard@27: self.Elements.remove(wire) lbessard@27: for connector in connected: lbessard@27: connector.RefreshParentBlock() lbessard@27: lbessard@27: def DeleteContact(self, contact): lbessard@27: connectors = contact.GetConnectors() lbessard@27: elements = connectors["output"].GetConnectedBlocks() lbessard@27: contact.Clean() lbessard@27: self.Blocks.remove(contact) lbessard@27: self.Elements.remove(contact) lbessard@27: self.Controler.RemoveCurrentElementEditingInstance(contact.GetId()) lbessard@27: for element in elements: lbessard@27: element.RefreshModel() lbessard@27: lbessard@27: def DeleteCoil(self, coil): lbessard@27: connectors = coil.GetConnectors() lbessard@27: elements = connectors["output"].GetConnectedBlocks() lbessard@27: coil.Clean() lbessard@27: self.Blocks.remove(coil) lbessard@27: self.Elements.remove(coil) lbessard@27: self.Controler.RemoveCurrentElementEditingInstance(coil.GetId()) lbessard@27: for element in elements: lbessard@27: element.RefreshModel() lbessard@27: lbessard@27: def DeletePowerRail(self, powerrail): lbessard@27: elements = [] lbessard@27: if powerrail.GetType() == LEFTRAIL: lbessard@27: for connector in powerrail.GetConnectors(): lbessard@27: for element in connector.GetConnectedBlocks(): lbessard@27: if element not in elements: lbessard@27: elements.append(element) lbessard@27: powerrrail.Clean() lbessard@27: self.Blocks.remove(powerrrail) lbessard@27: self.Elements.remove(powerrrail) lbessard@27: self.Controler.RemoveCurrentElementEditingInstance(powerrrail.GetId()) lbessard@27: for element in elements: lbessard@27: element.RefreshModel() lbessard@27: lbessard@27: #------------------------------------------------------------------------------- lbessard@27: # Edit element content functions lbessard@27: #------------------------------------------------------------------------------- lbessard@27: lbessard@27: def EditBlockContent(self, block): lbessard@27: dialog = BlockPropertiesDialog(self.Parent) lbessard@27: dialog.SetBlockList(self.Controler.GetBlockTypes()) lbessard@27: dialog.SetMinBlockSize(block.GetSize()) lbessard@27: values = {"name" : block.GetName(), "type" : block.GetType()} lbessard@27: values["extension"] = block.GetExtension() lbessard@27: dialog.SetValues(values) lbessard@27: if dialog.ShowModal() == wxID_OK: lbessard@27: values = dialog.GetValues() lbessard@27: if "name" in values: lbessard@27: block.SetName(values["name"]) lbessard@27: block.SetSize(values["width"], values["height"]) lbessard@27: block.SetType(values["type"], values["extension"]) lbessard@27: self.RefreshBlockModel(block) lbessard@27: self.Refresh() lbessard@27: dialog.Destroy() lbessard@27: lbessard@27: def EditVariableContent(self, variable): lbessard@27: dialog = VariablePropertiesDialog(self.Parent) lbessard@27: dialog.SetMinVariableSize(variable.GetSize()) lbessard@27: varlist = [] lbessard@27: vars = self.Controler.GetCurrentElementEditingInterfaceVars() lbessard@27: if vars: lbessard@27: for var in vars: lbessard@27: varlist.append((var["Name"], var["Class"], var["Type"])) lbessard@27: returntype = self.Controler.GetCurrentElementEditingInterfaceReturnType() lbessard@27: if returntype: lbessard@27: varlist.append((self.Controler.GetCurrentElementEditingName(), "Output", returntype)) lbessard@27: dialog.SetVariables(varlist) lbessard@27: values = {"name" : variable.GetName(), "type" : variable.GetType()} lbessard@27: dialog.SetValues(values) lbessard@27: if dialog.ShowModal() == wxID_OK: lbessard@27: old_type = variable.GetType() lbessard@27: values = dialog.GetValues() lbessard@27: variable.SetName(values["name"]) lbessard@27: variable.SetType(values["type"], values["value_type"]) lbessard@27: variable.SetSize(values["width"], values["height"]) lbessard@27: if old_type != values["type"]: lbessard@27: id = variable.GetId() lbessard@27: self.Controler.RemoveCurrentElementEditingInstance(id) lbessard@27: self.Controler.AddCurrentElementEditingVariable(id, values["type"]) lbessard@27: self.RefreshVariableModel(variable) lbessard@27: self.Refresh() lbessard@27: dialog.Destroy() lbessard@27: lbessard@27: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Editing functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: etisserant@0: def Cut(self): etisserant@0: pass etisserant@0: etisserant@0: def Copy(self): etisserant@0: pass etisserant@0: etisserant@0: def Paste(self): etisserant@0: pass etisserant@0: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Drawing functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: lbessard@27: def OnMoveWindow(self, event): lbessard@27: self.RefreshScrollBar() lbessard@27: event.Skip() lbessard@27: etisserant@0: def OnPaint(self, event): lbessard@27: dc = self.GetLogicalDC() etisserant@0: dc.Clear() etisserant@0: dc.SetPen(wxPen(wxColour(230, 230, 230))) etisserant@0: if self.Scaling and self.DrawGrid: etisserant@0: width, height = dc.GetSize() etisserant@0: for i in xrange(1, width / self.Scaling[0] + 1): etisserant@0: dc.DrawLine(i * self.Scaling[0], 0, i * self.Scaling[0], height) etisserant@0: for i in xrange(1, height / self.Scaling[1] + 1): etisserant@0: dc.DrawLine(0, i * self.Scaling[1], width, i * self.Scaling[1]) etisserant@0: for wire in self.Wires: etisserant@0: if wire != self.SelectedElement: etisserant@0: wire.Draw(dc) etisserant@0: for element in self.Elements: etisserant@0: if element not in self.Wires and element != self.SelectedElement: etisserant@0: element.Draw(dc) etisserant@0: if self.SelectedElement: etisserant@0: self.SelectedElement.Draw(dc) lbessard@27: if self.rubberBand.IsShown(): lbessard@27: self.rubberBand.Draw() lbessard@27: event.Skip() lbessard@27: lbessard@27: