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: from types import *
etisserant@0: 
etisserant@0: from plcopen.structures import *
etisserant@0: from graphics.GraphicCommons import *
etisserant@0: from graphics.FBD_Objects import *
etisserant@0: from Viewer import *
etisserant@0: 
etisserant@0: def ExtractNextBlocks(block, block_list):
etisserant@0:     current_list = [block]
etisserant@0:     while len(current_list) > 0:
etisserant@0:         next_list = []
etisserant@0:         for current in current_list:
etisserant@0:             connectors = current.GetConnectors()
etisserant@0:             input_connectors = []
etisserant@0:             if isinstance(current, LD_PowerRail) and current.GetType() == RIGHTRAIL:
etisserant@0:                 input_connectors = connectors
etisserant@0:             else:
etisserant@0:                 if "inputs" in connectors:
etisserant@0:                     input_connectors = connectors["inputs"]
etisserant@0:                 if "input" in connectors:
etisserant@0:                     input_connectors = [connectors["input"]]
etisserant@0:             for connector in input_connectors:
etisserant@0:                 for wire, handle in connector.GetWires():
etisserant@0:                     next = wire.EndConnected.GetParentBlock()
etisserant@0:                     if not isinstance(next, LD_PowerRail) and next not in block_list:
etisserant@0:                         block_list.append(next)
etisserant@0:                         next_list.append(next)
etisserant@0:         current_list = next_list
etisserant@0:     
etisserant@0: def CalcBranchSize(elements, stop):
etisserant@0:     branch_size = 0
etisserant@0:     stop_list = [stop]
etisserant@0:     ExtractNextBlocks(stop, stop_list)
etisserant@0:     element_tree = {}
etisserant@0:     for element in elements:
etisserant@0:         if element not in element_tree:
etisserant@0:             element_tree[element] = {"parents":["start"], "children":[], "weight":None}
etisserant@0:             GenerateTree(element, element_tree, stop_list)
etisserant@0:         elif element_tree[element]:
etisserant@0:             element_tree[element]["parents"].append("start")
etisserant@0:     for element, values in element_tree.items():
etisserant@0:         if values and values["children"] == ["stop"]:
etisserant@0:             CalcWeight(element, element_tree)
etisserant@0:             if values["weight"]:
etisserant@0:                 branch_size += values["weight"]
etisserant@0:             else:
etisserant@0:                 return 1
etisserant@0:     return branch_size
etisserant@0: 
etisserant@0: def RemoveElement(remove, element_tree):
etisserant@0:     if remove in element_tree and element_tree[remove]:
etisserant@0:         for child in element_tree[remove]["children"]:
etisserant@0:             if child != "stop":
etisserant@0:                 RemoveElement(child, element_tree)
etisserant@0:         element_tree[remove] = None
etisserant@0: 
etisserant@0: def GenerateTree(element, element_tree, stop_list):
etisserant@0:     if element in element_tree:
etisserant@0:         connectors = element.GetConnectors()
etisserant@0:         input_connectors = []
etisserant@0:         if isinstance(element, LD_PowerRail) and element.GetType() == RIGHTRAIL:
etisserant@0:             input_connectors = connectors
etisserant@0:         else:
etisserant@0:             if "inputs" in connectors:
etisserant@0:                 input_connectors = connectors["inputs"]
etisserant@0:             if "input" in connectors:
etisserant@0:                 input_connectors = [connectors["input"]]
etisserant@0:         for connector in input_connectors:
etisserant@0:             for wire, handle in connector.GetWires():
etisserant@0:                 next = wire.EndConnected.GetParentBlock()
etisserant@0:                 if isinstance(next, LD_PowerRail) and next.GetType() == LEFTRAIL or next in stop_list:
etisserant@0:                     for remove in element_tree[element]["children"]:
etisserant@0:                         RemoveElement(remove, element_tree)
etisserant@0:                     element_tree[element]["children"] = ["stop"]
etisserant@0:                 elif element_tree[element]["children"] == ["stop"]:
etisserant@0:                     element_tree[next] = None
etisserant@0:                 elif next not in element_tree or element_tree[next]:
etisserant@0:                     element_tree[element]["children"].append(next)
etisserant@0:                     if next in element_tree:
etisserant@0:                         element_tree[next]["parents"].append(element)
etisserant@0:                     else:
etisserant@0:                         element_tree[next] = {"parents":[element], "children":[], "weight":None}
etisserant@0:                         GenerateTree(next, element_tree, stop_list)
etisserant@0: 
etisserant@0: def CalcWeight(element, element_tree):
etisserant@0:     weight = 0
etisserant@0:     parts = None
etisserant@0:     if element in element_tree:
etisserant@0:         for parent in element_tree[element]["parents"]:
etisserant@0:             if parent == "start":
etisserant@0:                 weight += 1
etisserant@0:             elif parent in element_tree:
etisserant@0:                 if not parts:
etisserant@0:                     parts = len(element_tree[parent]["children"])
etisserant@0:                 else:
etisserant@0:                     parts = min(parts, len(element_tree[parent]["children"]))
etisserant@0:                 if not element_tree[parent]["weight"]:
etisserant@0:                     CalcWeight(parent, element_tree)
etisserant@0:                 if element_tree[parent]["weight"]:
etisserant@0:                     weight += element_tree[parent]["weight"]
etisserant@0:                 else:
etisserant@0:                     element_tree[element]["weight"] = None
etisserant@0:                     return
etisserant@0:             else:
etisserant@0:                 element_tree[element]["weight"] = None
etisserant@0:                 return
etisserant@0:         if not parts:
etisserant@0:             parts = 1
etisserant@0:         element_tree[element]["weight"] = max(1, weight / parts)
etisserant@0: 
etisserant@0: 
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: #                     Ladder Diagram Graphic elements Viewer class
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: 
etisserant@0: """
etisserant@0: Class derived from Viewer class that implements a Viewer of Ladder Diagram
etisserant@0: """
etisserant@0: 
etisserant@0: class LD_Viewer(Viewer):
etisserant@0: 
etisserant@0:     def __init__(self, parent, window, controler):
etisserant@0:         Viewer.__init__(self, parent, window, controler)
etisserant@0:         self.Rungs = []
etisserant@0:         self.Comments = []
etisserant@0: 
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: #                          Refresh functions
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: 
etisserant@0:     def RefreshView(self):
etisserant@0:         Viewer.RefreshView(self)
etisserant@0:         for i, rung in enumerate(self.Rungs):
etisserant@0:             bbox = rung.GetBoundingBox()
etisserant@0:             if i < len(self.Comments):
etisserant@0:                 pos = self.Comments[i].GetPosition()
etisserant@0:                 if pos[1] > bbox.y:
etisserant@0:                     self.Comment.insert(i, None)
etisserant@0:             else:
etisserant@0:                 self.Comment.insert(i, None)
etisserant@0:     
etisserant@0:     def loadInstance(self, instance, ids):
etisserant@0:         Viewer.loadInstance(self, instance, ids)
etisserant@0:         if instance["type"] == "leftPowerRail":
etisserant@0:             element = self.FindElementById(instance["id"])
etisserant@0:             rung = Graphic_Group(self)
etisserant@0:             rung.SelectElement(element)
etisserant@0:             self.Rungs.append(rung)
etisserant@0:         elif instance["type"] == "rightPowerRail":
etisserant@0:             rungs = []
etisserant@0:             for connector in instance["connectors"]:
etisserant@0:                 for link in connector["links"]:
etisserant@0:                     connected = self.FindElementById(link["refLocalId"])
etisserant@0:                     rung = self.FindRung(connected)
etisserant@0:                     if rung not in rungs:
etisserant@0:                         rungs.append(rung)
etisserant@0:             if len(rungs) > 1:
etisserant@0:                 raise "ValueError", "Ladder element with id %d is on more than one rung."%instance["id"]
etisserant@0:             element = self.FindElementById(instance["id"])
etisserant@0:             self.Rungs[rungs[0]].SelectElement(element)
etisserant@0:             for connector in element.GetConnectors():
etisserant@0:                 for wire, num in connector.GetWires():
etisserant@0:                     self.Rungs[rungs[0]].SelectElement(wire)
etisserant@0:             self.RefreshPosition(element)
etisserant@0:         elif instance["type"] in ["contact", "coil"]:
etisserant@0:             rungs = []
etisserant@0:             for link in instance["connectors"]["input"]["links"]:
etisserant@0:                 connected = self.FindElementById(link["refLocalId"])
etisserant@0:                 rung = self.FindRung(connected)
etisserant@0:                 if rung not in rungs:
etisserant@0:                     rungs.append(rung)
etisserant@0:             if len(rungs) > 1:
etisserant@0:                 raise "ValueError", "Ladder element with id %d is on more than one rung."%instance["id"]
etisserant@0:             element = self.FindElementById(instance["id"])
etisserant@0:             self.Rungs[rungs[0]].SelectElement(element)
etisserant@0:             for wire, num in element.GetConnectors()["input"].GetWires():
etisserant@0:                 self.Rungs[rungs[0]].SelectElement(wire)
etisserant@0:             self.RefreshPosition(element)
etisserant@0:         elif instance["type"] == "comment":
etisserant@0:             element = self.FindElementById(instance["id"])
etisserant@0:             pos = element.GetPosition()
etisserant@0:             i = 0
etisserant@0:             inserted = False
etisserant@0:             while i < len(self.Comments) and not inserted: 
etisserant@0:                 ipos = self.Comments[i].GetPosition()
etisserant@0:                 if pos[1] < ipos[1]:
etisserant@0:                     self.Comments.insert(i, element)
etisserant@0:                     inserted = True
etisserant@0:                 i += 1
etisserant@0:             if not inserted:
etisserant@0:                 self.Comments.append(element)
etisserant@0: 
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: #                          Search Element functions
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: 
etisserant@0:     def FindRung(self, element):
etisserant@0:         for i, rung in enumerate(self.Rungs):
etisserant@0:             if rung.IsElementIn(element):
etisserant@0:                 return i
etisserant@0:         return None
etisserant@0: 
etisserant@0:     def FindElement(self, pos):
etisserant@0:         elements = []
etisserant@0:         for element in self.Elements:
etisserant@0:             if element.HitTest(pos) or element.TestHandle(pos) != (0, 0):
etisserant@0:                 elements.append(element)
etisserant@0:         if len(elements) == 1:
etisserant@0:             return elements[0]
etisserant@0:         elif len(elements) > 1:
etisserant@0:             group = Graphic_Group(self)
etisserant@0:             for element in elements:
etisserant@0:                 if element in self.Blocks:
etisserant@0:                     return element
etisserant@0:                 group.SelectElement(element)
etisserant@0:             return group
etisserant@0:         return None
etisserant@0: 
etisserant@0:     def SearchElements(self, bbox):
etisserant@0:         elements = []
etisserant@0:         for element in self.Blocks:
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: #                          Mouse event functions
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: 
etisserant@0:     def OnViewerLeftDown(self, event):
etisserant@0:         if self.Mode == MODE_SELECTION:
etisserant@0:             pos = event.GetPosition()
etisserant@0:             element = self.FindElement(pos)
etisserant@0:             if self.SelectedElement:
etisserant@0:                 if self.SelectedElement in self.Elements:
etisserant@0:                     if self.SelectedElement != element:
etisserant@0:                         if self.SelectedElement in self.Wires:
etisserant@0:                             self.SelectedElement.SetSelectedSegment(None)
etisserant@0:                         else:
etisserant@0:                             self.SelectedElement.SetSelected(False)
etisserant@0:                     else:
etisserant@0:                         self.SelectedElement = None
etisserant@0:                 elif element and element not in self.Elements:
etisserant@0:                     if self.SelectedElement.GetElements() != element.GetElements():
etisserant@0:                         for elt in self.SelectedElement.GetElements():
etisserant@0:                             if elt in self.Wires:
etisserant@0:                                 elt.SetSelectedSegment(None)
etisserant@0:                         self.SelectedElement.SetSelected(False)
etisserant@0:                         self.SelectedElement = None
etisserant@0:                 else:
etisserant@0:                     for elt in self.SelectedElement.GetElements():
etisserant@0:                         if elt in self.Wires:
etisserant@0:                             elt.SetSelectedSegment(None)
etisserant@0:                     self.SelectedElement.SetSelected(False)
etisserant@0:                     self.SelectedElement = None
etisserant@0:                 self.Refresh()
etisserant@0:             if element:
etisserant@0:                 self.SelectedElement = element
etisserant@0:                 self.SelectedElement.OnLeftDown(event, self.Scaling)
etisserant@0:                 self.Refresh()
etisserant@0:             else:
etisserant@0:                 self.rubberBand.Reset()
etisserant@0:                 self.rubberBand.OnLeftDown(event, self.Scaling)
etisserant@0:         event.Skip()
etisserant@0: 
etisserant@0:     def OnViewerLeftUp(self, event):
etisserant@0:         if self.rubberBand.IsShown():
etisserant@0:             if self.Mode == MODE_SELECTION:
etisserant@0:                 elements = self.SearchElements(self.rubberBand.GetCurrentExtent())
etisserant@0:                 self.rubberBand.OnLeftUp(event, self.Scaling)
etisserant@0:                 if len(elements) > 0:
etisserant@0:                     self.SelectedElement = Graphic_Group(self)
etisserant@0:                     self.SelectedElement.SetElements(elements)
etisserant@0:                     self.SelectedElement.SetSelected(True)
etisserant@0:                     self.Refresh()
etisserant@0:         elif self.Mode == MODE_SELECTION and self.SelectedElement:
etisserant@0:             if self.SelectedElement in self.Elements:
etisserant@0:                 if self.SelectedElement in self.Wires:
etisserant@0:                     result = self.SelectedElement.TestSegment(event.GetPosition(), True)
etisserant@0:                     if result and result[1] in [EAST, WEST]:
etisserant@0:                         self.SelectedElement.SetSelectedSegment(result[0])
etisserant@0:                 else:
etisserant@0:                     self.SelectedElement.OnLeftUp(event, self.Scaling)
etisserant@0:             else:
etisserant@0:                 for element in self.SelectedElement.GetElements():
etisserant@0:                     if element in self.Wires:
etisserant@0:                         result = element.TestSegment(event.GetPosition(), True)
etisserant@0:                         if result and result[1] in [EAST, WEST]:
etisserant@0:                             element.SetSelectedSegment(result[0])
etisserant@0:                     else:
etisserant@0:                         element.OnLeftUp(event, self.Scaling)
etisserant@0:             wxCallAfter(self.SetCursor, wxNullCursor)
etisserant@0:             self.ReleaseMouse()
etisserant@0:             self.Refresh()
etisserant@0:         event.Skip()
etisserant@0: 
etisserant@0:     def OnViewerRightUp(self, event):
etisserant@0:         pos = event.GetPosition()
etisserant@0:         element = self.FindElement(pos)
etisserant@0:         if element:
etisserant@0:             if self.SelectedElement and self.SelectedElement != element:
etisserant@0:                 self.SelectedElement.SetSelected(False)
etisserant@0:             self.SelectedElement = element
etisserant@0:             if self.SelectedElement in self.Wires:
etisserant@0:                 self.SelectedElement.SetSelectedSegment(0)
etisserant@0:             else:
etisserant@0:                 self.SelectedElement.SetSelected(True)
etisserant@0:                 self.SelectedElement.OnRightUp(event, self.Scaling)
etisserant@0:             wxCallAfter(self.SetCursor, wxNullCursor)
etisserant@0:             self.ReleaseMouse()
etisserant@0:             self.Refresh()
etisserant@0:         event.Skip()
etisserant@0: 
etisserant@0:     def OnViewerLeftDClick(self, event):
etisserant@0:         if self.Mode == MODE_SELECTION and self.SelectedElement:
etisserant@0:             self.SelectedElement.OnLeftDClick(event, self.Scaling)
etisserant@0:             self.Refresh()
etisserant@0:         event.Skip()
etisserant@0: 
etisserant@0:     def OnViewerMotion(self, event):
etisserant@0:         if self.rubberBand.IsShown():
etisserant@0:             self.rubberBand.OnMotion(event, self.Scaling)
etisserant@0:         event.Skip()
etisserant@0: 
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: #                          Keyboard event functions
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: 
etisserant@0:     def OnChar(self, event):
etisserant@0:         keycode = event.GetKeyCode()
etisserant@0:         if keycode == WXK_DELETE and self.SelectedElement:
etisserant@0:             if self.SelectedElement in self.Blocks:
etisserant@0:                 self.SelectedElement.Delete()
etisserant@0:             elif self.SelectedElement in self.Wires:
etisserant@0:                 self.DeleteWire(self.SelectedElement)
etisserant@0:             elif self.SelectedElement not in self.Elements:
etisserant@0:                 all_wires = True
etisserant@0:                 for element in self.SelectedElement.GetElements():
etisserant@0:                     all_wires &= element in self.Wires
etisserant@0:                 if all_wires:
etisserant@0:                     self.DeleteWire(self.SelectedElement)
etisserant@0:                 else:
etisserant@0:                     self.SelectedElement.Delete()
etisserant@0:         self.Refresh()
etisserant@0:         event.Skip()
etisserant@0: 
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: #                          Adding element functions
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: 
etisserant@0:     def AddRung(self):
etisserant@0:         dialog = LDElementDialog(self.Parent, "coil")
etisserant@0:         varlist = []
etisserant@0:         vars = self.Controler.GetCurrentElementEditingInterfaceVars()
etisserant@0:         if vars:
etisserant@0:             for var in vars:
etisserant@0:                 if var["Type"] != "Input" and var["Value"] == "BOOL":
etisserant@0:                     varlist.append(var["Name"])
etisserant@0:         returntype = self.Controler.GetCurrentElementEditingInterfaceReturnType()
etisserant@0:         if returntype == "BOOL":
etisserant@0:             varlist.append(self.Controler.GetCurrentElementEditingName())
etisserant@0:         dialog.SetVariables(varlist)
etisserant@0:         dialog.SetValues({"name":"","type":COIL_NORMAL})
etisserant@0:         if dialog.ShowModal() == wxID_OK:
etisserant@0:             values = dialog.GetValues()
etisserant@0:             startx, starty = LD_OFFSET[0], 0
etisserant@0:             if len(self.Rungs) > 0:
etisserant@0:                 bbox = self.Rungs[-1].GetBoundingBox()
etisserant@0:                 starty = bbox.y + bbox.height
etisserant@0:             starty += LD_OFFSET[1]
etisserant@0:             rung = Graphic_Group(self)
etisserant@0:             # Create comment
etisserant@0:             id = self.GetNewId()
etisserant@0:             comment = Comment(self, "Commentaire", id)
etisserant@0:             comment.SetPosition(startx, starty)
etisserant@0:             comment.SetSize(LD_COMMENT_DEFAULTSIZE[0], LD_COMMENT_DEFAULTSIZE[1])
etisserant@0:             self.Elements.append(comment)
etisserant@0:             self.Comments.append(comment)
etisserant@0:             self.Controler.AddCurrentElementEditingComment(id)
etisserant@0:             self.RefreshCommentModel(comment)
etisserant@0:             starty += LD_COMMENT_DEFAULTSIZE[1] + LD_OFFSET[1]
etisserant@0:             # Create LeftPowerRail
etisserant@0:             id = self.GetNewId()
etisserant@0:             leftpowerrail = LD_PowerRail(self, LEFTRAIL, id)
etisserant@0:             leftpowerrail.SetPosition(startx, starty)
etisserant@0:             self.Elements.append(leftpowerrail)
etisserant@0:             self.Blocks.append(leftpowerrail)
etisserant@0:             rung.SelectElement(leftpowerrail)
etisserant@0:             self.Controler.AddCurrentElementEditingPowerRail(id, LEFTRAIL)
etisserant@0:             self.RefreshPowerRailModel(leftpowerrail)
etisserant@0:             # Create Coil
etisserant@0:             id = self.GetNewId()
etisserant@0:             coil = LD_Coil(self, values["type"], values["name"], id)
etisserant@0:             coil.SetPosition(startx, starty + (LD_LINE_SIZE - LD_ELEMENT_SIZE[1]) / 2)
etisserant@0:             self.Elements.append(coil)
etisserant@0:             self.Blocks.append(coil)
etisserant@0:             rung.SelectElement(coil)
etisserant@0:             self.Controler.AddCurrentElementEditingCoil(id)
etisserant@0:             # Create Wire between LeftPowerRail and Coil
etisserant@0:             wire = Wire(self)
etisserant@0:             start_connector = coil.GetConnectors()["input"]
etisserant@0:             end_connector = leftpowerrail.GetConnectors()[0]
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:             rung.SelectElement(wire)
etisserant@0:             # Create RightPowerRail
etisserant@0:             id = self.GetNewId()
etisserant@0:             rightpowerrail = LD_PowerRail(self, RIGHTRAIL, id)
etisserant@0:             rightpowerrail.SetPosition(startx, starty)
etisserant@0:             self.Elements.append(rightpowerrail)
etisserant@0:             self.Blocks.append(rightpowerrail)
etisserant@0:             rung.SelectElement(rightpowerrail)
etisserant@0:             self.Controler.AddCurrentElementEditingPowerRail(id, RIGHTRAIL)
etisserant@0:             # Create Wire between LeftPowerRail and Coil
etisserant@0:             wire = Wire(self)
etisserant@0:             start_connector = rightpowerrail.GetConnectors()[0]
etisserant@0:             end_connector = coil.GetConnectors()["output"]
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:             rung.SelectElement(wire)
etisserant@0:             self.RefreshPosition(coil)
etisserant@0:             self.Rungs.append(rung)
etisserant@0:             self.Refresh()
etisserant@0: 
etisserant@0:     def AddContact(self):
etisserant@0:         wires = []
etisserant@0:         if self.SelectedElement in self.Wires:
etisserant@0:             left_element = self.SelectedElement.EndConnected
etisserant@0:             if not isinstance(left_element.GetParentBlock(), LD_Coil):
etisserant@0:                 wires.append(self.SelectedElement)
etisserant@0:         elif self.SelectedElement and self.SelectedElement not in self.Elements:
etisserant@0:             if False not in [element in self.Wires for element in self.SelectedElement.GetElements()]:
etisserant@0:                 for element in self.SelectedElement.GetElements():
etisserant@0:                     wires.append(element)
etisserant@0:         if len(wires) > 0:
etisserant@0:             dialog = LDElementDialog(self.Parent, "contact")
etisserant@0:             varlist = []
etisserant@0:             vars = self.Controler.GetCurrentElementEditingInterfaceVars()
etisserant@0:             if vars:
etisserant@0:                 for var in vars:
etisserant@0:                     if var["Type"] != "Output" and var["Value"] == "BOOL":
etisserant@0:                         varlist.append(var["Name"])
etisserant@0:             dialog.SetVariables(varlist)
etisserant@0:             dialog.SetValues({"name":"","type":CONTACT_NORMAL})
etisserant@0:             if dialog.ShowModal() == wxID_OK:
etisserant@0:                 values = dialog.GetValues()
etisserant@0:                 points = wires[0].GetSelectedSegmentPoints()
etisserant@0:                 id = self.GetNewId()
etisserant@0:                 contact = LD_Contact(self, values["type"], values["name"], id)
etisserant@0:                 contact.SetPosition(0, points[0].y - (LD_ELEMENT_SIZE[1] + 1) / 2)
etisserant@0:                 self.Elements.append(contact)
etisserant@0:                 self.Blocks.append(contact)
etisserant@0:                 self.Controler.AddCurrentElementEditingContact(id)
etisserant@0:                 rungindex = self.FindRung(wires[0])
etisserant@0:                 rung = self.Rungs[rungindex]
etisserant@0:                 old_bbox = rung.GetBoundingBox()
etisserant@0:                 rung.SelectElement(contact)
etisserant@0:                 connectors = contact.GetConnectors()
etisserant@0:                 left_elements = []
etisserant@0:                 right_elements = []
etisserant@0:                 left_index = []
etisserant@0:                 right_index = []
etisserant@0:                 for wire in wires:
etisserant@0:                     if wire.EndConnected not in left_elements:
etisserant@0:                         left_elements.append(wire.EndConnected)
etisserant@0:                         left_index.append(wire.EndConnected.GetWireIndex(wire))
etisserant@0:                     else:
etisserant@0:                         idx = left_elements.index(wire.EndConnected)
etisserant@0:                         left_index[idx] = min(left_index[idx], wire.EndConnected.GetWireIndex(wire))
etisserant@0:                     if wire.StartConnected not in right_elements:
etisserant@0:                         right_elements.append(wire.StartConnected)
etisserant@0:                         right_index.append(wire.StartConnected.GetWireIndex(wire))
etisserant@0:                     else:
etisserant@0:                         idx = right_elements.index(wire.StartConnected)
etisserant@0:                         right_index[idx] = min(right_index[idx], wire.StartConnected.GetWireIndex(wire))
etisserant@0:                     wire.SetSelectedSegment(None)
etisserant@0:                     wire.Clean()
etisserant@0:                     rung.SelectElement(wire)
etisserant@0:                     self.Wires.remove(wire)
etisserant@0:                     self.Elements.remove(wire)
etisserant@0:                 wires = []
etisserant@0:                 right_wires = []
etisserant@0:                 for i, left_element in enumerate(left_elements):
etisserant@0:                     wire = Wire(self)
etisserant@0:                     wires.append(wire)
etisserant@0:                     connectors["input"].Connect((wire, 0), False)
etisserant@0:                     left_element.InsertConnect(left_index[i], (wire, -1), False)
etisserant@0:                     wire.ConnectStartPoint(None, connectors["input"])
etisserant@0:                     wire.ConnectEndPoint(None, left_element)
etisserant@0:                 for i, right_element in enumerate(right_elements):
etisserant@0:                     wire = Wire(self)
etisserant@0:                     wires.append(wire)
etisserant@0:                     right_wires.append(wire)
etisserant@0:                     right_element.InsertConnect(right_index[i], (wire, 0), False)
etisserant@0:                     connectors["output"].Connect((wire, -1), False)
etisserant@0:                     wire.ConnectStartPoint(None, right_element)
etisserant@0:                     wire.ConnectEndPoint(None, connectors["output"])
etisserant@0:                 right_wires.reverse()
etisserant@0:                 for wire in wires:
etisserant@0:                     self.Wires.append(wire)
etisserant@0:                     self.Elements.append(wire)
etisserant@0:                     rung.SelectElement(wire)
etisserant@0:                 self.RefreshPosition(contact)
etisserant@0:                 if len(right_wires) > 1:
etisserant@0:                     group = Graphic_Group(self)
etisserant@0:                     group.SetSelected(False)
etisserant@0:                     for wire in right_wires:
etisserant@0:                         wire.SetSelectedSegment(-1)
etisserant@0:                         group.SelectElement(wire)
etisserant@0:                     self.SelectedElement = group
etisserant@0:                 else:
etisserant@0:                     right_wires[0].SetSelectedSegment(-1)
etisserant@0:                     self.SelectedElement = right_wires[0]
etisserant@0:                 rung.RefreshBoundingBox()
etisserant@0:                 new_bbox = rung.GetBoundingBox()
etisserant@0:                 self.RefreshRungs(new_bbox.height - old_bbox.height, rungindex + 1)
etisserant@0:                 self.Refresh()
etisserant@0: 
etisserant@0:     def AddBranch(self):
etisserant@0:         blocks = []
etisserant@0:         if self.SelectedElement in self.Blocks:
etisserant@0:             blocks = [self.SelectedElement]
etisserant@0:         elif self.SelectedElement not in self.Elements:
etisserant@0:             elements = self.SelectedElement.GetElements()
etisserant@0:             for element in elements:
etisserant@0:                 if isinstance(element, (LD_PowerRail, LD_Coil)):
etisserant@0:                     return
etisserant@0:                 blocks.append(element)
etisserant@0:         if len(blocks) > 0:
etisserant@0:             blocks_infos = []
etisserant@0:             left_elements = []
etisserant@0:             left_index = []
etisserant@0:             right_elements = []
etisserant@0:             right_index = []
etisserant@0:             for block in blocks:
etisserant@0:                 connectors = block.GetConnectors()
etisserant@0:                 block_infos = {"inputs":[],"outputs":[],"lefts":[],"rights":[]}
etisserant@0:                 if "inputs" in connectors:
etisserant@0:                     block_infos["inputs"] = connectors["inputs"]
etisserant@0:                 if "outputs" in connectors:
etisserant@0:                     block_infos["outputs"] = connectors["outputs"]
etisserant@0:                 if "input" in connectors:
etisserant@0:                     block_infos["inputs"] = [connectors["input"]]
etisserant@0:                 if "output" in connectors:
etisserant@0:                     block_infos["outputs"] = [connectors["output"]]
etisserant@0:                 for connector in block_infos["inputs"]:
etisserant@0:                     for wire, handle in connector.GetWires():
etisserant@0:                         found = False
etisserant@0:                         for infos in blocks_infos:
etisserant@0:                             if wire.EndConnected in infos["outputs"]:
etisserant@0:                                 for left_element in infos["lefts"]:
etisserant@0:                                     if left_element not in block_infos["lefts"]:
etisserant@0:                                         block_infos["lefts"].append(left_element)
etisserant@0:                                 found = True
etisserant@0:                         if not found and wire.EndConnected not in block_infos["lefts"]:
etisserant@0:                             block_infos["lefts"].append(wire.EndConnected)
etisserant@0:                             if wire.EndConnected not in left_elements:
etisserant@0:                                 left_elements.append(wire.EndConnected)
etisserant@0:                                 left_index.append(wire.EndConnected.GetWireIndex(wire))
etisserant@0:                             else:
etisserant@0:                                 index = left_elements.index(wire.EndConnected)
etisserant@0:                                 left_index[index] = max(left_index[index], wire.EndConnected.GetWireIndex(wire))
etisserant@0:                 for connector in block_infos["outputs"]:
etisserant@0:                     for wire, handle in connector.GetWires():
etisserant@0:                         found = False
etisserant@0:                         for infos in blocks_infos:
etisserant@0:                             if wire.StartConnected in infos["inputs"]:
etisserant@0:                                 for right_element in infos["rights"]:
etisserant@0:                                     if right_element not in block_infos["rights"]:
etisserant@0:                                         block_infos["rights"].append(right_element)
etisserant@0:                                 found = True
etisserant@0:                         if not found and wire.StartConnected not in block_infos["rights"]:
etisserant@0:                             block_infos["rights"].append(wire.StartConnected)
etisserant@0:                             if wire.StartConnected not in right_elements:
etisserant@0:                                 right_elements.append(wire.StartConnected)
etisserant@0:                                 right_index.append(wire.StartConnected.GetWireIndex(wire))
etisserant@0:                             else:
etisserant@0:                                 index = right_elements.index(wire.StartConnected)
etisserant@0:                                 right_index[index] = max(right_index[index], wire.StartConnected.GetWireIndex(wire))
etisserant@0:                 for connector in block_infos["inputs"]:
etisserant@0:                     for infos in blocks_infos:
etisserant@0:                         if connector in infos["rights"]:
etisserant@0:                             infos["rights"].remove(connector)
etisserant@0:                             if connector in right_elements:
etisserant@0:                                 index = right_elements.index(connector)
etisserant@0:                                 right_elements.pop(index)
etisserant@0:                                 right_index.pop(index)
etisserant@0:                             for right_element in block_infos["rights"]:
etisserant@0:                                 if right_element not in infos["rights"]:
etisserant@0:                                     infos["rights"].append(right_element)
etisserant@0:                 for connector in block_infos["outputs"]:
etisserant@0:                     for infos in blocks_infos:
etisserant@0:                         if connector in infos["lefts"]:
etisserant@0:                             infos["lefts"].remove(connector)
etisserant@0:                             if connector in left_elements:
etisserant@0:                                 index = left_elements.index(connector)
etisserant@0:                                 left_elements.pop(index)
etisserant@0:                                 left_index.pop(index)
etisserant@0:                             for left_element in block_infos["lefts"]:
etisserant@0:                                 if left_element not in infos["lefts"]:
etisserant@0:                                     infos["lefts"].append(left_element)
etisserant@0:                 blocks_infos.append(block_infos)
etisserant@0:             for infos in blocks_infos:
etisserant@0:                 left_elements = [element for element in infos["lefts"]]
etisserant@0:                 for left_element in left_elements:
etisserant@0:                     if isinstance(left_element.GetParentBlock(), LD_PowerRail):
etisserant@0:                         infos["lefts"].remove(left_element)
etisserant@0:                         if "LD_PowerRail" not in infos["lefts"]:
etisserant@0:                             infos["lefts"].append("LD_PowerRail")
etisserant@0:                 right_elements = [element for element in infos["rights"]]
etisserant@0:                 for right_element in right_elements:
etisserant@0:                     if isinstance(right_element.GetParentBlock(), LD_PowerRail):
etisserant@0:                         infos["rights"].remove(tight_element)
etisserant@0:                         if "LD_PowerRail" not in infos["rights"]:
etisserant@0:                             infos["rights"].append("LD_PowerRail")
etisserant@0:                 infos["lefts"].sort()
etisserant@0:                 infos["rights"].sort()
etisserant@0:             lefts = blocks_infos[0]["lefts"]
etisserant@0:             rights = blocks_infos[0]["rights"]
etisserant@0:             good = True
etisserant@0:             for infos in blocks_infos[1:]:
etisserant@0:                 good &= infos["lefts"] == lefts
etisserant@0:                 good &= infos["rights"] == rights
etisserant@0:             if good:
etisserant@0:                 rungindex = self.FindRung(blocks[0])
etisserant@0:                 rung = self.Rungs[rungindex]
etisserant@0:                 old_bbox = rung.GetBoundingBox()
etisserant@0:                 left_powerrail = True
etisserant@0:                 right_powerrail = True
etisserant@0:                 for element in left_elements:
etisserant@0:                     left_powerrail &= isinstance(element.GetParentBlock(), LD_PowerRail)
etisserant@0:                 for element in right_elements:
etisserant@0:                     right_powerrail &= isinstance(element.GetParentBlock(), LD_PowerRail)
etisserant@0:                 if not left_powerrail or not right_powerrail:
etisserant@0:                     if left_powerrail:
etisserant@0:                         powerrail = left_elements[0].GetParentBlock()
etisserant@0:                         index = 0
etisserant@0:                         for left_element in left_elements: 
etisserant@0:                             index = max(index, powerrail.GetConnectorIndex(left_element))
etisserant@0:                         if powerrail.IsNullConnector(index + 1):
etisserant@0:                             powerrail.DeleteConnector(index + 1)
etisserant@0:                         powerrail.InsertConnector(index + 1)
etisserant@0:                         powerrail.RefreshModel()
etisserant@0:                         connectors = powerrail.GetConnectors()
etisserant@0:                         for i, right_element in enumerate(right_elements):
etisserant@0:                             new_wire = Wire(self)
etisserant@0:                             right_element.InsertConnect(right_index[i] + 1, (new_wire, 0), False)
etisserant@0:                             connectors[index + 1].Connect((new_wire, -1), False)
etisserant@0:                             new_wire.ConnectStartPoint(None, right_element)
etisserant@0:                             new_wire.ConnectEndPoint(None, connectors[index + 1])
etisserant@0:                             self.Wires.append(new_wire)
etisserant@0:                             self.Elements.append(new_wire)
etisserant@0:                             rung.SelectElement(new_wire)
etisserant@0:                         right_elements.reverse()
etisserant@0:                     elif right_powerrail:
etisserant@0:                         pass
etisserant@0:                     else:
etisserant@0:                         left_elements.reverse()
etisserant@0:                         right_elements.reverse()
etisserant@0:                         wires = []
etisserant@0:                         for i, left_element in enumerate(left_elements):
etisserant@0:                             for j, right_element in enumerate(right_elements):
etisserant@0:                                 exist = False
etisserant@0:                                 for wire, handle in right_element.GetWires():
etisserant@0:                                     exist |= wire.EndConnected == left_element
etisserant@0:                                 if not exist:
etisserant@0:                                     new_wire = Wire(self)
etisserant@0:                                     wires.append(new_wire)
etisserant@0:                                     right_element.InsertConnect(right_index[j] + 1, (new_wire, 0), False)
etisserant@0:                                     left_element.InsertConnect(left_index[i] + 1, (new_wire, -1), False)
etisserant@0:                                     new_wire.ConnectStartPoint(None, right_element)
etisserant@0:                                     new_wire.ConnectEndPoint(None, left_element)
etisserant@0:                         wires.reverse()
etisserant@0:                         for wire in wires:
etisserant@0:                             self.Wires.append(wire)
etisserant@0:                             self.Elements.append(wire)
etisserant@0:                             rung.SelectElement(wire)
etisserant@0:                         right_elements.reverse()
etisserant@0:                 for block in blocks:
etisserant@0:                     self.RefreshPosition(block)
etisserant@0:                 for right_element in right_elements:
etisserant@0:                     self.RefreshPosition(right_element.GetParentBlock())
etisserant@0:                 self.SelectedElement.RefreshBoundingBox()
etisserant@0:                 rung.RefreshBoundingBox()
etisserant@0:                 new_bbox = rung.GetBoundingBox()
etisserant@0:                 self.RefreshRungs(new_bbox.height - old_bbox.height, rungindex + 1)
etisserant@0:                 self.Refresh()
etisserant@0: 
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: #                          Delete element functions
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: 
etisserant@0:     def DeleteContact(self, contact):
etisserant@0:         rungindex = self.FindRung(contact)
etisserant@0:         rung = self.Rungs[rungindex]
etisserant@0:         old_bbox = rung.GetBoundingBox()
etisserant@0:         connectors = contact.GetConnectors()
etisserant@0:         input_wires = [wire for wire, handle in connectors["input"].GetWires()]
etisserant@0:         output_wires = [wire for wire, handle in connectors["output"].GetWires()]
etisserant@0:         left_elements = [(wire.EndConnected, wire.EndConnected.GetWireIndex(wire)) for wire in input_wires]
etisserant@0:         right_elements = [(wire.StartConnected, wire.StartConnected.GetWireIndex(wire)) for wire in output_wires]
etisserant@0:         for wire in input_wires:
etisserant@0:             wire.Clean()
etisserant@0:             rung.SelectElement(wire)
etisserant@0:             self.Wires.remove(wire)
etisserant@0:             self.Elements.remove(wire)
etisserant@0:         for wire in output_wires:
etisserant@0:             wire.Clean()
etisserant@0:             rung.SelectElement(wire)
etisserant@0:             self.Wires.remove(wire)
etisserant@0:             self.Elements.remove(wire)
etisserant@0:         rung.SelectElement(contact)
etisserant@0:         contact.Clean()
etisserant@0:         left_elements.reverse()
etisserant@0:         right_elements.reverse()
etisserant@0:         powerrail = len(left_elements) == 1 and isinstance(left_elements[0][0].GetParentBlock(), LD_PowerRail)
etisserant@0:         for left_element, left_index in left_elements:
etisserant@0:             for right_element, right_index in right_elements:
etisserant@0:                 wire_removed = []
etisserant@0:                 for wire, handle in right_element.GetWires():
etisserant@0:                     if wire.EndConnected == left_element:
etisserant@0:                         wire_removed.append(wire)
etisserant@0:                     elif isinstance(wire.EndConnected.GetParentBlock(), LD_PowerRail) and powerrail:
etisserant@0:                         left_powerrail = wire.EndConnected.GetParentBlock()
etisserant@0:                         index = left_powerrail.GetConnectorIndex(wire.EndConnected)
etisserant@0:                         left_powerrail.DeleteConnector(index)
etisserant@0:                         wire_removed.append(wire)
etisserant@0:                 for wire in wire_removed:
etisserant@0:                     wire.Clean()
etisserant@0:                     self.Wires.remove(wire)
etisserant@0:                     self.Elements.remove(wire)
etisserant@0:                     rung.SelectElement(wire)
etisserant@0:         wires = []
etisserant@0:         for left_element, left_index in left_elements:
etisserant@0:             for right_element, right_index in right_elements:
etisserant@0:                 wire = Wire(self)
etisserant@0:                 wires.append(wire)
etisserant@0:                 right_element.InsertConnect(right_index, (wire, 0), False)
etisserant@0:                 left_element.InsertConnect(left_index, (wire, -1), False)
etisserant@0:                 wire.ConnectStartPoint(None, right_element)
etisserant@0:                 wire.ConnectEndPoint(None, left_element)
etisserant@0:         wires.reverse()
etisserant@0:         for wire in wires:
etisserant@0:             self.Wires.append(wire)
etisserant@0:             self.Elements.append(wire)
etisserant@0:             rung.SelectElement(wire)
etisserant@0:         right_elements.reverse()
etisserant@0:         for right_element, right_index in right_elements:
etisserant@0:             self.RefreshPosition(right_element.GetParentBlock())
etisserant@0:         self.Blocks.remove(contact)
etisserant@0:         self.Elements.remove(contact)
etisserant@0:         self.Controler.RemoveCurrentElementEditingInstance(contact.GetId())
etisserant@0:         rung.RefreshBoundingBox()
etisserant@0:         new_bbox = rung.GetBoundingBox()
etisserant@0:         self.RefreshRungs(new_bbox.height - old_bbox.height, rungindex + 1)
etisserant@0:         self.SelectedElement = None
etisserant@0: 
etisserant@0:     def DeleteCoil(self, coil):
etisserant@0:         rungindex = self.FindRung(coil)
etisserant@0:         rung = self.Rungs[rungindex]
etisserant@0:         bbox = rung.GetBoundingBox()
etisserant@0:         for element in rung.GetElements():
etisserant@0:             if element in self.Wires:
etisserant@0:                 element.Clean()
etisserant@0:                 self.Wires.remove(element)
etisserant@0:                 self.Elements.remove(element)
etisserant@0:         for element in rung.GetElements():
etisserant@0:             if element in self.Blocks:
etisserant@0:                 self.Controler.RemoveCurrentElementEditingInstance(element.GetId())
etisserant@0:                 self.Blocks.remove(element)
etisserant@0:                 self.Elements.remove(element)
etisserant@0:         self.Controler.RemoveCurrentElementEditingInstance(self.Comments[rungindex].GetId())
etisserant@0:         self.Elements.remove(self.Comments[rungindex])
etisserant@0:         self.Comments.pop(rungindex)
etisserant@0:         self.Rungs.pop(rungindex)
etisserant@0:         if rungindex < len(self.Rungs):
etisserant@0:             next_bbox = self.Rungs[rungindex].GetBoundingBox()
etisserant@0:             self.RefreshRungs(bbox.y - next_bbox.y, rungindex)
etisserant@0:         self.SelectedElement = None
etisserant@0: 
etisserant@0:     def DeleteWire(self, wire):
etisserant@0:         wires = []
etisserant@0:         left_elements = []
etisserant@0:         right_elements = []
etisserant@0:         if wire in self.Wires:
etisserant@0:             wires = [wire]
etisserant@0:         elif wire not in self.Elements:
etisserant@0:             for element in wire.GetElements():
etisserant@0:                 if element in self.Wires:
etisserant@0:                     wires.append(element)
etisserant@0:                 else:
etisserant@0:                     wires = []
etisserant@0:                     break
etisserant@0:         if len(wires) > 0:
etisserant@0:             rungindex = self.FindRung(wires[0])
etisserant@0:             rung = self.Rungs[rungindex]
etisserant@0:             old_bbox = rung.GetBoundingBox()
etisserant@0:             for wire in wires:
etisserant@0:                 connections = wire.GetSelectedSegmentConnections()
etisserant@0:                 left_block = wire.EndConnected.GetParentBlock()
etisserant@0:                 if wire.EndConnected not in left_elements:
etisserant@0:                     left_elements.append(wire.EndConnected)
etisserant@0:                 if wire.StartConnected not in right_elements:
etisserant@0:                     right_elements.append(wire.StartConnected)
etisserant@0:                 if connections == (False, False) or connections == (False, True) and isinstance(left_block, LD_PowerRail):
etisserant@0:                     wire.Clean()
etisserant@0:                     self.Wires.remove(wire)
etisserant@0:                     self.Elements.remove(wire)
etisserant@0:                     rung.SelectElement(wire)
etisserant@0:             for left_element in left_elements:
etisserant@0:                 left_block = left_element.GetParentBlock()
etisserant@0:                 if isinstance(left_block, LD_PowerRail):
etisserant@0:                     if len(left_element.GetWires()) == 0:
etisserant@0:                         index = left_block.GetConnectorIndex(left_element)
etisserant@0:                         left_block.DeleteConnector(index)
etisserant@0:                 else:
etisserant@0:                     connectors = left_block.GetConnectors()
etisserant@0:                     output_connectors = []
etisserant@0:                     if "outputs" in connectors:
etisserant@0:                         output_connectors = connectors["outputs"]
etisserant@0:                     if "output" in connectors:
etisserant@0:                         output_connectors = [connectors["output"]]
etisserant@0:                     for connector in output_connectors:
etisserant@0:                         for wire, handle in connector.GetWires():
etisserant@0:                             self.RefreshPosition(wire.StartConnected.GetParentBlock())
etisserant@0:             for right_element in right_elements:
etisserant@0:                 self.RefreshPosition(right_element.GetParentBlock())
etisserant@0:             rung.RefreshBoundingBox()
etisserant@0:             new_bbox = rung.GetBoundingBox()
etisserant@0:             self.RefreshRungs(new_bbox.height - old_bbox.height, rungindex + 1)
etisserant@0:             self.SelectedElement = None
etisserant@0: 
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: #                        Refresh element position functions
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: 
etisserant@0:     def RefreshPosition(self, element):
etisserant@0:         if isinstance(element, LD_PowerRail) and element.GetType() == LEFTRAIL:
etisserant@0:             element.RefreshModel()
etisserant@0:             return
etisserant@0:         connectors = element.GetConnectors()
etisserant@0:         input_connectors = []
etisserant@0:         output_connectors = []
etisserant@0:         if isinstance(element, LD_PowerRail) and element.GetType() == RIGHTRAIL:
etisserant@0:             input_connectors = connectors
etisserant@0:         else:
etisserant@0:             if "inputs" in connectors:
etisserant@0:                 input_connectors = connectors["inputs"]
etisserant@0:             if "outputs" in connectors:
etisserant@0:                 output_connectors = connectors["outputs"]
etisserant@0:             if "input" in connectors:
etisserant@0:                 input_connectors = [connectors["input"]]
etisserant@0:             if "output" in connectors:
etisserant@0:                 output_connectors = [connectors["output"]]
etisserant@0:         position = element.GetPosition()
etisserant@0:         minx = 0
etisserant@0:         onlyone = []
etisserant@0:         for connector in input_connectors:
etisserant@0:             onlyone.append(len(connector.GetWires()) == 1)
etisserant@0:             for wire, handle in connector.GetWires():
etisserant@0:                 onlyone[-1] &= len(wire.EndConnected.GetWires()) == 1
etisserant@0:                 leftblock = wire.EndConnected.GetParentBlock()
etisserant@0:                 pos = leftblock.GetPosition()
etisserant@0:                 size = leftblock.GetSize()
etisserant@0:                 minx = max(minx, pos[0] + size[0])
etisserant@0:         if isinstance(element, LD_Coil):
etisserant@0:             interval = LD_WIRECOIL_SIZE
etisserant@0:         else:
etisserant@0:             interval = LD_WIRE_SIZE
etisserant@0:         if False in onlyone:
etisserant@0:             interval += LD_WIRE_SIZE
etisserant@0:         movex = minx + interval - position[0]
etisserant@0:         element.Move(movex, 0)
etisserant@0:         for i, connector in enumerate(input_connectors):
etisserant@0:             startpoint = connector.GetPosition(False)
etisserant@0:             previous_blocks = []
etisserant@0:             block_list = []
etisserant@0:             start_offset = 0
etisserant@0:             if not onlyone[i]:
etisserant@0:                 middlepoint = minx + LD_WIRE_SIZE
etisserant@0:             for j, (wire, handle) in enumerate(connector.GetWires()):
etisserant@0:                 block = wire.EndConnected.GetParentBlock()
etisserant@0:                 endpoint = wire.EndConnected.GetPosition(False)
etisserant@0:                 if j == 0:
etisserant@0:                     if not onlyone[i] and wire.EndConnected.GetWireIndex(wire) > 0:
etisserant@0:                         start_offset = endpoint.y - startpoint.y
etisserant@0:                     offset = start_offset
etisserant@0:                 else:
etisserant@0:                     offset = start_offset + LD_LINE_SIZE * CalcBranchSize(previous_blocks, block)
etisserant@0:                 if block in block_list:
etisserant@0:                     wires = wire.EndConnected.GetWires()
etisserant@0:                     endmiddlepoint = wires[0][0].StartConnected.GetPosition(False)[0] - LD_WIRE_SIZE
etisserant@0:                     points = [startpoint, wxPoint(middlepoint, startpoint.y),
etisserant@0:                               wxPoint(middlepoint, startpoint.y + offset),
etisserant@0:                               wxPoint(endmiddlepoint, startpoint.y + offset),
etisserant@0:                               wxPoint(endmiddlepoint, endpoint.y), endpoint]
etisserant@0:                 else:
etisserant@0:                     if startpoint.y + offset != endpoint.y:
etisserant@0:                         if isinstance(block, LD_PowerRail):
etisserant@0:                             index = block.GetConnectorIndex(wire.EndConnected)
etisserant@0:                             if index:
etisserant@0:                                 diff = (startpoint.y - endpoint.y) / LD_LINE_SIZE
etisserant@0:                                 for k in xrange(abs(diff)):
etisserant@0:                                     if diff < 0:
etisserant@0:                                         block.DeleteConnector(index - 1 - k)
etisserant@0:                                     else:
etisserant@0:                                         block.InsertConnector(index + k, False)
etisserant@0:                         else:
etisserant@0:                             block.Move(0, startpoint.y + offset - endpoint.y)
etisserant@0:                             self.RefreshPosition(block)
etisserant@0:                         endpoint = wire.EndConnected.GetPosition(False)
etisserant@0:                     if not onlyone[i]:
etisserant@0:                         points = [startpoint, wxPoint(middlepoint, startpoint.y),
etisserant@0:                                   wxPoint(middlepoint, endpoint.y), endpoint]
etisserant@0:                     else:
etisserant@0:                         points = [startpoint, endpoint]
etisserant@0:                 wire.SetPoints(points)
etisserant@0:                 previous_blocks.append(block)
etisserant@0:                 ExtractNextBlocks(block, block_list)
etisserant@0:         element.RefreshModel(False)
etisserant@0:         for connector in output_connectors:
etisserant@0:             for wire, handle in connector.GetWires():
etisserant@0:                 self.RefreshPosition(wire.StartConnected.GetParentBlock())
etisserant@0:     
etisserant@0:     def RefreshRungs(self, movey, fromidx):
etisserant@0:         if movey != 0:
etisserant@0:             for i in xrange(fromidx, len(self.Rungs)):
etisserant@0:                 self.Comments[i].Move(0, movey)
etisserant@0:                 self.Comments[i].RefreshModel()
etisserant@0:                 self.Rungs[i].Move(0, movey)
etisserant@0:                 for element in self.Rungs[i].GetElements():
etisserant@0:                     if element in self.Blocks:
etisserant@0:                         self.RefreshPosition(element)
etisserant@0: 
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: #                          Edit element content functions
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: 
etisserant@0:     def EditContactContent(self, contact):
etisserant@0:         dialog = LDElementDialog(self.Parent, "contact")
etisserant@0:         varlist = []
etisserant@0:         vars = self.Controler.GetCurrentElementEditingInterfaceVars()
etisserant@0:         if vars:
etisserant@0:             for var in vars:
lbessard@2:                 if var["Class"] != "Output" and var["Type"] == "BOOL":
etisserant@0:                     varlist.append(var["Name"])
etisserant@0:         dialog.SetVariables(varlist)
etisserant@0:         dialog.SetValues({"name":contact.GetName(),"type":contact.GetType()})
etisserant@0:         if dialog.ShowModal() == wxID_OK:
etisserant@0:             values = dialog.GetValues()
etisserant@0:             contact.SetName(values["name"])
etisserant@0:             contact.SetType(values["type"])
etisserant@0:             contact.RefreshModel(False)
etisserant@0:             self.Refresh()
etisserant@0:         dialog.Destroy()
etisserant@0: 
etisserant@0:     def EditCoilContent(self, coil):
etisserant@0:         dialog = LDElementDialog(self.Parent, "coil")
etisserant@0:         varlist = []
etisserant@0:         vars = self.Controler.GetCurrentElementEditingInterfaceVars()
etisserant@0:         if vars:
etisserant@0:             for var in vars:
lbessard@2:                 if var["Class"] != "Input" and var["Type"] == "BOOL":
etisserant@0:                     varlist.append(var["Name"])
etisserant@0:         returntype = self.Controler.GetCurrentElementEditingInterfaceReturnType()
etisserant@0:         if returntype == "BOOL":
etisserant@0:             varlist.append(self.Controler.GetCurrentElementEditingName())
etisserant@0:         dialog.SetVariables(varlist)
etisserant@0:         dialog.SetValues({"name":coil.GetName(),"type":coil.GetType()})
etisserant@0:         if dialog.ShowModal() == wxID_OK:
etisserant@0:             values = dialog.GetValues()
etisserant@0:             coil.SetName(values["name"])
etisserant@0:             coil.SetType(values["type"])
etisserant@0:             coil.RefreshModel(False)
etisserant@0:             self.Refresh()
etisserant@0:         dialog.Destroy()
etisserant@0: 
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: #                        Edit Ladder Element Properties Dialog
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: 
etisserant@0: [wxID_LDELEMENTDIALOG, wxID_LDELEMENTDIALOGMAINPANEL, 
etisserant@0:  wxID_LDELEMENTDIALOGNAME, wxID_LDELEMENTDIALOGRADIOBUTTON1, 
etisserant@0:  wxID_LDELEMENTDIALOGRADIOBUTTON2, wxID_LDELEMENTDIALOGRADIOBUTTON3,
etisserant@0:  wxID_LDELEMENTDIALOGRADIOBUTTON4, wxID_LDELEMENTDIALOGPREVIEW,
etisserant@0:  wxID_LDELEMENTDIALOGSTATICTEXT1, wxID_LDELEMENTDIALOGSTATICTEXT2, 
etisserant@0:  wxID_LDELEMENTDIALOGSTATICTEXT3, 
etisserant@0: ] = [wx.NewId() for _init_ctrls in range(11)]
etisserant@0: 
etisserant@0: class LDElementDialog(wx.Dialog):
etisserant@0:     def _init_coll_flexGridSizer1_Items(self, parent):
etisserant@0:         # generated method, don't edit
etisserant@0: 
etisserant@0:         parent.AddWindow(self.MainPanel, 0, border=0, flag=0)
etisserant@0: 
etisserant@0:     def _init_sizers(self):
etisserant@0:         # generated method, don't edit
etisserant@0:         self.flexGridSizer1 = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=0)
etisserant@0: 
etisserant@0:         self._init_coll_flexGridSizer1_Items(self.flexGridSizer1)
etisserant@0: 
etisserant@0:         self.SetSizer(self.flexGridSizer1)
etisserant@0: 
etisserant@0:     def _init_ctrls(self, prnt, title, labels):
etisserant@0:         # generated method, don't edit
etisserant@0:         wx.Dialog.__init__(self, id=wxID_LDELEMENTDIALOG,
etisserant@0:               name='VariablePropertiesDialog', parent=prnt, pos=wx.Point(376, 223),
etisserant@0:               size=wx.Size(350, 260), style=wx.DEFAULT_DIALOG_STYLE,
etisserant@0:               title=title)
etisserant@0:         self.SetClientSize(wx.Size(350, 260))
etisserant@0: 
etisserant@0:         self.MainPanel = wx.Panel(id=wxID_LDELEMENTDIALOGMAINPANEL,
etisserant@0:               name='MainPanel', parent=self, pos=wx.Point(0, 0),
etisserant@0:               size=wx.Size(340, 200), style=wx.TAB_TRAVERSAL)
etisserant@0:         self.MainPanel.SetAutoLayout(True)
etisserant@0: 
etisserant@0:         self.staticText1 = wx.StaticText(id=wxID_LDELEMENTDIALOGSTATICTEXT1,
etisserant@0:               label='Type:', name='staticText1', parent=self.MainPanel,
etisserant@0:               pos=wx.Point(24, 24), size=wx.Size(70, 17), style=0)
etisserant@0: 
etisserant@0:         self.staticText2 = wx.StaticText(id=wxID_LDELEMENTDIALOGSTATICTEXT2,
etisserant@0:               label='Name:', name='staticText2', parent=self.MainPanel,
etisserant@0:               pos=wx.Point(24, 150), size=wx.Size(70, 17), style=0)
etisserant@0: 
etisserant@0:         self.staticText3 = wx.StaticText(id=wxID_LDELEMENTDIALOGSTATICTEXT3,
etisserant@0:               label='Preview:', name='staticText3', parent=self.MainPanel,
etisserant@0:               pos=wx.Point(174, 24), size=wx.Size(100, 17), style=0)
etisserant@0: 
etisserant@0:         self.radioButton1 = wx.RadioButton(id=wxID_LDELEMENTDIALOGRADIOBUTTON1,
etisserant@0:               label=labels[0], name='radioButton1', parent=self.MainPanel,
etisserant@0:               pos=wx.Point(24, 48), size=wx.Size(114, 24), style=0)
etisserant@0:         EVT_RADIOBUTTON(self, wxID_LDELEMENTDIALOGRADIOBUTTON1, self.OnTypeChanged)
etisserant@0:         self.radioButton1.SetValue(True)
etisserant@0: 
etisserant@0:         self.radioButton2 = wx.RadioButton(id=wxID_LDELEMENTDIALOGRADIOBUTTON2,
etisserant@0:               label=labels[1], name='radioButton2', parent=self.MainPanel, 
etisserant@0:               pos=wx.Point(24, 72), size=wx.Size(128, 24), style=0)
etisserant@0:         EVT_RADIOBUTTON(self, wxID_LDELEMENTDIALOGRADIOBUTTON2, self.OnTypeChanged)
etisserant@0: 
etisserant@0:         self.radioButton3 = wx.RadioButton(id=wxID_LDELEMENTDIALOGRADIOBUTTON3,
etisserant@0:               label=labels[2], name='radioButton3', parent=self.MainPanel,
etisserant@0:               pos=wx.Point(24, 96), size=wx.Size(114, 24), style=0)
etisserant@0:         EVT_RADIOBUTTON(self, wxID_LDELEMENTDIALOGRADIOBUTTON3, self.OnTypeChanged)
etisserant@0: 
etisserant@0:         self.radioButton4 = wx.RadioButton(id=wxID_LDELEMENTDIALOGRADIOBUTTON4,
etisserant@0:               label=labels[3], name='radioButton4', parent=self.MainPanel, 
etisserant@0:               pos=wx.Point(24, 120), size=wx.Size(128, 24), style=0)
etisserant@0:         EVT_RADIOBUTTON(self, wxID_LDELEMENTDIALOGRADIOBUTTON4, self.OnTypeChanged)
etisserant@0: 
etisserant@0:         self.Name = wx.Choice(id=wxID_LDELEMENTDIALOGNAME,
etisserant@0:               name='Name', parent=self.MainPanel, pos=wx.Point(24, 174),
etisserant@0:               size=wx.Size(145, 24), style=0)
etisserant@0:         EVT_CHOICE(self, wxID_LDELEMENTDIALOGNAME, self.OnNameChanged)
etisserant@0: 
etisserant@0:         self.Preview = wx.Panel(id=wxID_LDELEMENTDIALOGPREVIEW,
etisserant@0:               name='Preview', parent=self.MainPanel, pos=wx.Point(174, 48),
etisserant@0:               size=wx.Size(150, 150), style=wx.TAB_TRAVERSAL|wx.SIMPLE_BORDER)
etisserant@0:         self.Preview.SetBackgroundColour(wxColour(255,255,255))
etisserant@0: 
etisserant@0:         self._init_sizers()
etisserant@0: 
etisserant@0:     def __init__(self, parent, type):
etisserant@0:         self.Type = type
etisserant@0:         if type == "contact":
lbessard@2:             self._init_ctrls(parent, "Edit Contact Values", ['Normal','Negate','Rising Edge','Falling Edge'])
etisserant@0:             self.Element = LD_Contact(self.Preview, CONTACT_NORMAL, "")
etisserant@0:         elif type == "coil":
lbessard@2:             self._init_ctrls(parent, "Edit Coil Values", ['Normal','Negate','Set','Reset'])
etisserant@0:             self.Element = LD_Coil(self.Preview, COIL_NORMAL, "")
etisserant@0:         self.Element.SetPosition((150 - LD_ELEMENT_SIZE[0]) / 2, (150 - LD_ELEMENT_SIZE[1]) / 2)
etisserant@0: 
etisserant@0:         self.ButtonSizer = self.CreateButtonSizer(wxOK|wxCANCEL)
etisserant@0:         self.flexGridSizer1.Add(self.ButtonSizer, 1, wxALIGN_RIGHT)
etisserant@0: 
etisserant@0:         EVT_PAINT(self, self.OnPaint)
etisserant@0: 
etisserant@0:     def SetVariables(self, vars):
etisserant@0:         self.Name.Clear()
etisserant@0:         for name in vars:
etisserant@0:             self.Name.Append(name)
etisserant@0:         self.Name.Enable(self.Name.GetCount() > 0)
etisserant@0: 
etisserant@0:     def SetValues(self, values):
etisserant@0:         for name, value in values.items():
etisserant@0:             if name == "name":
etisserant@0:                 self.Element.SetName(value)
etisserant@0:                 self.Name.SetStringSelection(value)
etisserant@0:             elif name == "type":
etisserant@0:                 self.Element.SetType(value)
etisserant@0:                 if self.Type == "contact":
etisserant@0:                     if value == CONTACT_NORMAL:
etisserant@0:                         self.radioButton1.SetValue(True)
etisserant@0:                     elif value == CONTACT_REVERSE:
etisserant@0:                         self.radioButton2.SetValue(True)
etisserant@0:                     elif value == CONTACT_RISING:
etisserant@0:                         self.radioButton3.SetValue(True)
etisserant@0:                     elif value == CONTACT_FALLING:
etisserant@0:                         self.radioButton4.SetValue(True)
etisserant@0:                 elif self.Type == "coil":
etisserant@0:                     if value == COIL_NORMAL:
etisserant@0:                         self.radioButton1.SetValue(True)
etisserant@0:                     elif value == COIL_REVERSE:
etisserant@0:                         self.radioButton2.SetValue(True)
etisserant@0:                     elif value == COIL_SET:
etisserant@0:                         self.radioButton3.SetValue(True)
etisserant@0:                     elif value == COIL_RESET:
etisserant@0:                         self.radioButton4.SetValue(True)
etisserant@0: 
etisserant@0:     def GetValues(self):
etisserant@0:         values = {}
etisserant@0:         values["name"] = self.Element.GetName()
etisserant@0:         values["type"] = self.Element.GetType()
etisserant@0:         return values
etisserant@0: 
etisserant@0:     def OnTypeChanged(self, event):
etisserant@0:         if self.Type == "contact":
etisserant@0:             if self.radioButton1.GetValue():
etisserant@0:                 self.Element.SetType(CONTACT_NORMAL)
etisserant@0:             elif self.radioButton2.GetValue():
etisserant@0:                 self.Element.SetType(CONTACT_REVERSE)
etisserant@0:             elif self.radioButton3.GetValue():
etisserant@0:                 self.Element.SetType(CONTACT_RISING)
etisserant@0:             elif self.radioButton4.GetValue():
etisserant@0:                 self.Element.SetType(CONTACT_FALLING)
etisserant@0:         elif self.Type == "coil":
etisserant@0:             if self.radioButton1.GetValue():
etisserant@0:                 self.Element.SetType(COIL_NORMAL)
etisserant@0:             elif self.radioButton2.GetValue():
etisserant@0:                 self.Element.SetType(COIL_REVERSE)
etisserant@0:             elif self.radioButton3.GetValue():
etisserant@0:                 self.Element.SetType(COIL_SET)
etisserant@0:             elif self.radioButton4.GetValue():
etisserant@0:                 self.Element.SetType(COIL_RESET)
etisserant@0:         self.RefreshPreview()
etisserant@0:         event.Skip()
etisserant@0: 
etisserant@0:     def OnNameChanged(self, event):
etisserant@0:         self.Element.SetName(self.Name.GetStringSelection())
etisserant@0:         self.RefreshPreview()
etisserant@0:         event.Skip()
etisserant@0: 
etisserant@0:     def RefreshPreview(self):
etisserant@0:         dc = wxClientDC(self.Preview)
etisserant@0:         dc.Clear()
etisserant@0:         self.Element.Draw(dc)
etisserant@0: 
etisserant@0:     def OnPaint(self, event):
etisserant@0:         self.RefreshPreview()
etisserant@0:         event.Skip()