Laurent@814: #!/usr/bin/env python
Laurent@814: # -*- coding: utf-8 -*-
Laurent@814: 
andrej@1571: # This file is part of Beremiz, a Integrated Development Environment for
andrej@1571: # programming IEC 61131-3 automates supporting plcopen standard and CanFestival.
Laurent@814: #
andrej@1571: # Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD
Laurent@814: #
andrej@1571: # See COPYING file for copyrights details.
Laurent@814: #
andrej@1571: # This program is free software; you can redistribute it and/or
andrej@1571: # modify it under the terms of the GNU General Public License
andrej@1571: # as published by the Free Software Foundation; either version 2
andrej@1571: # of the License, or (at your option) any later version.
Laurent@814: #
andrej@1571: # This program is distributed in the hope that it will be useful,
andrej@1571: # but WITHOUT ANY WARRANTY; without even the implied warranty of
andrej@1571: # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
andrej@1571: # GNU General Public License for more details.
Laurent@814: #
andrej@1571: # You should have received a copy of the GNU General Public License
andrej@1571: # along with this program; if not, write to the Free Software
andrej@1571: # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
Laurent@814: 
Laurent@814: import wx
Laurent@814: import time
Laurent@814: from types import *
Laurent@814: 
Laurent@814: from Viewer import *
Laurent@814: 
andrej@1736: 
Laurent@814: def ExtractNextBlocks(block, block_list):
Laurent@814:     current_list = [block]
Laurent@814:     while len(current_list) > 0:
Laurent@814:         next_list = []
Laurent@814:         for current in current_list:
Laurent@814:             connectors = current.GetConnectors()
Laurent@814:             input_connectors = []
Laurent@814:             if isinstance(current, LD_PowerRail) and current.GetType() == RIGHTRAIL:
Laurent@814:                 input_connectors = connectors
Laurent@814:             else:
Laurent@814:                 if "inputs" in connectors:
Laurent@814:                     input_connectors = connectors["inputs"]
Laurent@814:                 if "input" in connectors:
Laurent@814:                     input_connectors = [connectors["input"]]
Laurent@814:             for connector in input_connectors:
Laurent@814:                 for wire, handle in connector.GetWires():
Laurent@814:                     next = wire.EndConnected.GetParentBlock()
Laurent@814:                     if not isinstance(next, LD_PowerRail) and next not in block_list:
Laurent@814:                         block_list.append(next)
Laurent@814:                         next_list.append(next)
Laurent@814:         current_list = next_list
andrej@1730: 
andrej@1736: 
Laurent@814: def CalcBranchSize(elements, stops):
Laurent@814:     branch_size = 0
Laurent@814:     stop_list = stops
Laurent@814:     for stop in stops:
Laurent@814:         ExtractNextBlocks(stop, stop_list)
Laurent@814:     element_tree = {}
Laurent@814:     for element in elements:
Laurent@814:         if element not in element_tree:
andrej@1740:             element_tree[element] = {"parents": ["start"], "children": [], "weight": None}
Laurent@814:             GenerateTree(element, element_tree, stop_list)
Laurent@814:         elif element_tree[element]:
Laurent@814:             element_tree[element]["parents"].append("start")
andrej@1740:     remove_stops = {"start": [], "stop": []}
Laurent@814:     for element, values in element_tree.items():
Laurent@814:         if "stop" in values["children"]:
Laurent@814:             removed = []
Laurent@814:             for child in values["children"]:
Laurent@814:                 if child != "stop":
andrej@1779:                     # if child in elements:
andrej@1779:                     #     RemoveElement(child, element_tree)
andrej@1779:                     #     removed.append(child)
Laurent@814:                     if "start" in element_tree[child]["parents"]:
Laurent@814:                         if element not in remove_stops["stop"]:
Laurent@814:                             remove_stops["stop"].append(element)
Laurent@814:                         if child not in remove_stops["start"]:
Laurent@814:                             remove_stops["start"].append(child)
Laurent@814:             for child in removed:
Laurent@814:                 values["children"].remove(child)
Laurent@814:     for element in remove_stops["start"]:
Laurent@814:         element_tree[element]["parents"].remove("start")
Laurent@814:     for element in remove_stops["stop"]:
Laurent@814:         element_tree[element]["children"].remove("stop")
Laurent@814:     for element, values in element_tree.items():
Laurent@814:         if values and "stop" in values["children"]:
Laurent@814:             CalcWeight(element, element_tree)
Laurent@814:             if values["weight"]:
Laurent@814:                 branch_size += values["weight"]
Laurent@814:             else:
Laurent@814:                 return 1
Laurent@814:     return branch_size
Laurent@814: 
andrej@1736: 
Laurent@814: def RemoveElement(remove, element_tree):
Laurent@814:     if remove in element_tree and element_tree[remove]:
Laurent@814:         for child in element_tree[remove]["children"]:
Laurent@814:             if child != "stop":
Laurent@814:                 RemoveElement(child, element_tree)
Laurent@814:         element_tree.pop(remove)
andrej@1753: #        element_tree[remove] = None
Laurent@814: 
andrej@1736: 
Laurent@814: def GenerateTree(element, element_tree, stop_list):
Laurent@814:     if element in element_tree:
Laurent@814:         connectors = element.GetConnectors()
Laurent@814:         input_connectors = []
Laurent@814:         if isinstance(element, LD_PowerRail) and element.GetType() == RIGHTRAIL:
Laurent@814:             input_connectors = connectors
Laurent@814:         else:
Laurent@814:             if "inputs" in connectors:
Laurent@814:                 input_connectors = connectors["inputs"]
Laurent@814:             if "input" in connectors:
Laurent@814:                 input_connectors = [connectors["input"]]
Laurent@814:         for connector in input_connectors:
Laurent@814:             for wire, handle in connector.GetWires():
Laurent@814:                 next = wire.EndConnected.GetParentBlock()
Laurent@814:                 if isinstance(next, LD_PowerRail) and next.GetType() == LEFTRAIL or next in stop_list:
andrej@1779:                     # for remove in element_tree[element]["children"]:
andrej@1779:                     #     RemoveElement(remove, element_tree)
andrej@1779:                     # element_tree[element]["children"] = ["stop"]
Laurent@814:                     element_tree[element]["children"].append("stop")
andrej@1779:                 # elif element_tree[element]["children"] == ["stop"]:
andrej@1779:                 #     element_tree[next] = None
Laurent@814:                 elif next not in element_tree or element_tree[next]:
Laurent@814:                     element_tree[element]["children"].append(next)
Laurent@814:                     if next in element_tree:
Laurent@814:                         element_tree[next]["parents"].append(element)
Laurent@814:                     else:
andrej@1740:                         element_tree[next] = {"parents": [element], "children": [], "weight": None}
Laurent@814:                         GenerateTree(next, element_tree, stop_list)
Laurent@814: 
andrej@1736: 
Laurent@814: def CalcWeight(element, element_tree):
Laurent@814:     weight = 0
Laurent@814:     parts = None
Laurent@814:     if element in element_tree:
Laurent@814:         for parent in element_tree[element]["parents"]:
Laurent@814:             if parent == "start":
Laurent@814:                 weight += 1
Laurent@814:             elif parent in element_tree:
Laurent@814:                 if not parts:
Laurent@814:                     parts = len(element_tree[parent]["children"])
Laurent@814:                 else:
Laurent@814:                     parts = min(parts, len(element_tree[parent]["children"]))
Laurent@814:                 if not element_tree[parent]["weight"]:
Laurent@814:                     CalcWeight(parent, element_tree)
Laurent@814:                 if element_tree[parent]["weight"]:
Laurent@814:                     weight += element_tree[parent]["weight"]
Laurent@814:                 else:
Laurent@814:                     element_tree[element]["weight"] = None
Laurent@814:                     return
Laurent@814:             else:
Laurent@814:                 element_tree[element]["weight"] = None
Laurent@814:                 return
Laurent@814:         if not parts:
Laurent@814:             parts = 1
Laurent@814:         element_tree[element]["weight"] = max(1, weight / parts)
Laurent@814: 
Laurent@814: 
andrej@1782: # -------------------------------------------------------------------------------
Laurent@814: #                     Ladder Diagram Graphic elements Viewer class
andrej@1782: # -------------------------------------------------------------------------------
Laurent@814: 
Laurent@814: 
Laurent@814: class LD_Viewer(Viewer):
andrej@1736:     """
andrej@1736:     Class derived from Viewer class that implements a Viewer of Ladder Diagram
andrej@1736:     """
Laurent@814: 
andrej@1744:     def __init__(self, parent, tagname, window, controler, debug=False, instancepath=""):
Laurent@814:         Viewer.__init__(self, parent, tagname, window, controler, debug, instancepath)
Laurent@814:         self.Rungs = []
Laurent@814:         self.RungComments = []
Laurent@814:         self.CurrentLanguage = "LD"
Laurent@814: 
andrej@1782:     # -------------------------------------------------------------------------------
andrej@1782:     #                          Refresh functions
andrej@1782:     # -------------------------------------------------------------------------------
Laurent@814: 
Laurent@814:     def ResetView(self):
Laurent@814:         self.Rungs = []
Laurent@814:         self.RungComments = []
Laurent@814:         Viewer.ResetView(self)
Laurent@814: 
Laurent@814:     def RefreshView(self, variablepanel=True, selection=None):
Laurent@814:         Viewer.RefreshView(self, variablepanel, selection)
Laurent@1081:         if self.GetDrawingMode() != FREEDRAWING_MODE:
Laurent@1081:             for i, rung in enumerate(self.Rungs):
Laurent@1081:                 bbox = rung.GetBoundingBox()
Laurent@1081:                 if i < len(self.RungComments):
Laurent@1081:                     if self.RungComments[i]:
Laurent@1081:                         pos = self.RungComments[i].GetPosition()
Laurent@1081:                         if pos[1] > bbox.y:
Laurent@1081:                             self.RungComments.insert(i, None)
Laurent@1081:                 else:
Laurent@1081:                     self.RungComments.insert(i, None)
andrej@1730: 
Laurent@814:     def loadInstance(self, instance, ids, selection):
Laurent@814:         Viewer.loadInstance(self, instance, ids, selection)
Laurent@814:         if self.GetDrawingMode() != FREEDRAWING_MODE:
Laurent@814:             if instance["type"] == "leftPowerRail":
Laurent@814:                 element = self.FindElementById(instance["id"])
Laurent@814:                 rung = Graphic_Group(self)
Laurent@814:                 rung.SelectElement(element)
Laurent@814:                 self.Rungs.append(rung)
Laurent@814:             elif instance["type"] == "rightPowerRail":
Laurent@814:                 rungs = []
Laurent@814:                 for connector in instance["inputs"]:
Laurent@814:                     for link in connector["links"]:
Laurent@814:                         connected = self.FindElementById(link["refLocalId"])
Laurent@814:                         rung = self.FindRung(connected)
Laurent@814:                         if rung not in rungs:
Laurent@814:                             rungs.append(rung)
Laurent@814:                 if len(rungs) > 1:
andrej@1765:                     raise ValueError(
andrej@1765:                         _("Ladder element with id %d is on more than one rung.")
andrej@1765:                         % instance["id"])
andrej@1765: 
Laurent@814:                 element = self.FindElementById(instance["id"])
Laurent@814:                 element_connectors = element.GetConnectors()
Laurent@814:                 self.Rungs[rungs[0]].SelectElement(element)
Laurent@814:                 for connector in element_connectors["inputs"]:
Laurent@814:                     for wire, num in connector.GetWires():
Laurent@814:                         self.Rungs[rungs[0]].SelectElement(wire)
Laurent@814:                 wx.CallAfter(self.RefreshPosition, element)
Laurent@814:             elif instance["type"] in ["contact", "coil"]:
Laurent@814:                 rungs = []
Laurent@814:                 for link in instance["inputs"][0]["links"]:
Laurent@814:                     connected = self.FindElementById(link["refLocalId"])
Laurent@814:                     rung = self.FindRung(connected)
Laurent@814:                     if rung not in rungs:
Laurent@814:                         rungs.append(rung)
Laurent@814:                 if len(rungs) > 1:
andrej@1765:                     raise ValueError(
andrej@1765:                         _("Ladder element with id %d is on more than one rung.")
andrej@1765:                         % instance["id"])
andrej@1765: 
Laurent@814:                 element = self.FindElementById(instance["id"])
andrej@1730:                 element_connectors = element.GetConnectors()
Laurent@814:                 self.Rungs[rungs[0]].SelectElement(element)
Laurent@814:                 for wire, num in element_connectors["inputs"][0].GetWires():
Laurent@814:                     self.Rungs[rungs[0]].SelectElement(wire)
Laurent@814:                 wx.CallAfter(self.RefreshPosition, element)
Laurent@814:             elif instance["type"] == "comment":
Laurent@814:                 element = self.FindElementById(instance["id"])
Laurent@814:                 pos = element.GetPosition()
Laurent@814:                 i = 0
Laurent@814:                 inserted = False
andrej@1730:                 while i < len(self.RungComments) and not inserted:
Laurent@814:                     ipos = self.RungComments[i].GetPosition()
Laurent@814:                     if pos[1] < ipos[1]:
Laurent@814:                         self.RungComments.insert(i, element)
Laurent@814:                         inserted = True
Laurent@814:                     i += 1
Laurent@814:                 if not inserted:
Laurent@814:                     self.RungComments.append(element)
andrej@1730: 
andrej@1782:     # -------------------------------------------------------------------------------
andrej@1782:     #                          Search Element functions
andrej@1782:     # -------------------------------------------------------------------------------
Laurent@814: 
Laurent@814:     def FindRung(self, element):
Laurent@814:         for i, rung in enumerate(self.Rungs):
Laurent@814:             if rung.IsElementIn(element):
Laurent@814:                 return i
Laurent@814:         return None
Laurent@814: 
andrej@1744:     def FindElement(self, event, exclude_group=False, connectors=True):
Laurent@814:         if self.GetDrawingMode() == FREEDRAWING_MODE:
Laurent@814:             return Viewer.FindElement(self, event, exclude_group, connectors)
andrej@1730: 
Laurent@814:         dc = self.GetLogicalDC()
Laurent@814:         pos = event.GetLogicalPosition(dc)
Laurent@814:         if self.SelectedElement and not isinstance(self.SelectedElement, (Graphic_Group, Wire)):
Laurent@814:             if self.SelectedElement.HitTest(pos, connectors) or self.SelectedElement.TestHandle(pos) != (0, 0):
Laurent@814:                 return self.SelectedElement
Laurent@814:         elements = []
Laurent@814:         for element in self.GetElements(sort_wires=True):
Laurent@814:             if element.HitTest(pos, connectors) or element.TestHandle(event) != (0, 0):
Laurent@814:                 elements.append(element)
Laurent@814:         if len(elements) == 1:
Laurent@814:             return elements[0]
Laurent@814:         elif len(elements) > 1:
Laurent@814:             group = Graphic_Group(self)
Laurent@814:             for element in elements:
Laurent@814:                 if self.IsBlock(element):
Laurent@814:                     return element
Laurent@814:                 group.SelectElement(element)
Laurent@814:             return group
Laurent@814:         return None
Laurent@814: 
Laurent@814:     def SearchElements(self, bbox):
Laurent@890:         if self.GetDrawingMode() == FREEDRAWING_MODE:
Laurent@890:             return Viewer.SearchElements(self, bbox)
andrej@1730: 
Laurent@814:         elements = []
Laurent@814:         for element in self.Blocks.values() + self.Comments.values():
Laurent@814:             if element.IsInSelection(bbox):
Laurent@814:                 elements.append(element)
Laurent@814:         return elements
Laurent@814: 
andrej@1782:     # -------------------------------------------------------------------------------
andrej@1782:     #                          Mouse event functions
andrej@1782:     # -------------------------------------------------------------------------------
Laurent@814: 
Laurent@814:     def OnViewerLeftDown(self, event):
Laurent@814:         if self.GetDrawingMode() == FREEDRAWING_MODE:
Laurent@814:             Viewer.OnViewerLeftDown(self, event)
Laurent@814:         elif self.Mode == MODE_SELECTION:
Laurent@814:             element = self.FindElement(event)
Laurent@814:             if self.SelectedElement:
Laurent@814:                 if not isinstance(self.SelectedElement, Graphic_Group):
Laurent@814:                     if self.SelectedElement != element:
Laurent@814:                         if self.IsWire(self.SelectedElement):
Laurent@814:                             self.SelectedElement.SetSelectedSegment(None)
Laurent@814:                         else:
Laurent@814:                             self.SelectedElement.SetSelected(False)
Laurent@814:                     else:
Laurent@814:                         self.SelectedElement = None
Laurent@814:                 elif element and isinstance(element, Graphic_Group):
Laurent@814:                     if self.SelectedElement.GetElements() != element.GetElements():
Laurent@814:                         for elt in self.SelectedElement.GetElements():
Laurent@814:                             if self.IsWire(elt):
Laurent@814:                                 elt.SetSelectedSegment(None)
Laurent@814:                         self.SelectedElement.SetSelected(False)
Laurent@814:                         self.SelectedElement = None
Laurent@814:                 else:
Laurent@814:                     for elt in self.SelectedElement.GetElements():
Laurent@814:                         if self.IsWire(elt):
Laurent@814:                             elt.SetSelectedSegment(None)
Laurent@814:                     self.SelectedElement.SetSelected(False)
Laurent@814:                     self.SelectedElement = None
Laurent@814:             if element:
Laurent@814:                 self.SelectedElement = element
Laurent@814:                 self.SelectedElement.OnLeftDown(event, self.GetLogicalDC(), self.Scaling)
Laurent@814:                 self.SelectedElement.Refresh()
Laurent@814:             else:
Laurent@814:                 self.rubberBand.Reset()
Laurent@814:                 self.rubberBand.OnLeftDown(event, self.GetLogicalDC(), self.Scaling)
Laurent@814:         event.Skip()
Laurent@814: 
Laurent@814:     def OnViewerLeftUp(self, event):
Laurent@814:         if self.GetDrawingMode() == FREEDRAWING_MODE:
Laurent@814:             Viewer.OnViewerLeftUp(self, event)
Laurent@814:         elif self.rubberBand.IsShown():
Laurent@814:             if self.Mode == MODE_SELECTION:
Laurent@814:                 elements = self.SearchElements(self.rubberBand.GetCurrentExtent())
Laurent@814:                 self.rubberBand.OnLeftUp(event, self.GetLogicalDC(), self.Scaling)
Laurent@814:                 if len(elements) > 0:
Laurent@814:                     self.SelectedElement = Graphic_Group(self)
Laurent@814:                     self.SelectedElement.SetElements(elements)
Laurent@814:                     self.SelectedElement.SetSelected(True)
Laurent@814:         elif self.Mode == MODE_SELECTION and self.SelectedElement:
andrej@1730:             dc = self.GetLogicalDC()
Laurent@814:             if not isinstance(self.SelectedElement, Graphic_Group):
Laurent@814:                 if self.IsWire(self.SelectedElement):
Laurent@814:                     result = self.SelectedElement.TestSegment(event.GetLogicalPosition(dc), True)
Laurent@814:                     if result and result[1] in [EAST, WEST]:
Laurent@814:                         self.SelectedElement.SetSelectedSegment(result[0])
Laurent@814:                 else:
Laurent@814:                     self.SelectedElement.OnLeftUp(event, dc, self.Scaling)
Laurent@814:             else:
Laurent@814:                 for element in self.SelectedElement.GetElements():
Laurent@814:                     if self.IsWire(element):
Laurent@814:                         result = element.TestSegment(event.GetLogicalPosition(dc), True)
Laurent@814:                         if result and result[1] in [EAST, WEST]:
Laurent@814:                             element.SetSelectedSegment(result[0])
Laurent@814:                     else:
Laurent@814:                         element.OnLeftUp(event, dc, self.Scaling)
Laurent@814:             self.SelectedElement.Refresh()
Laurent@814:             wx.CallAfter(self.SetCurrentCursor, 0)
Laurent@814:         event.Skip()
Laurent@814: 
Laurent@814:     def OnViewerRightUp(self, event):
Laurent@814:         if self.GetDrawingMode() == FREEDRAWING_MODE:
Laurent@814:             Viewer.OnViewerRightUp(self, event)
Laurent@814:         else:
Laurent@814:             element = self.FindElement(event)
Laurent@814:             if element:
Laurent@814:                 if self.SelectedElement and self.SelectedElement != element:
Laurent@814:                     self.SelectedElement.SetSelected(False)
Laurent@814:                 self.SelectedElement = element
Laurent@814:                 if self.IsWire(self.SelectedElement):
Laurent@814:                     self.SelectedElement.SetSelectedSegment(0)
Laurent@814:                 else:
Laurent@814:                     self.SelectedElement.SetSelected(True)
Laurent@814:                     self.SelectedElement.OnRightUp(event, self.GetLogicalDC(), self.Scaling)
Laurent@814:                     self.SelectedElement.Refresh()
Laurent@814:                 wx.CallAfter(self.SetCurrentCursor, 0)
Laurent@814:         event.Skip()
Laurent@814: 
andrej@1782:     # -------------------------------------------------------------------------------
andrej@1782:     #                          Keyboard event functions
andrej@1782:     # -------------------------------------------------------------------------------
Laurent@814: 
Laurent@814:     def OnChar(self, event):
Laurent@814:         if self.GetDrawingMode() == FREEDRAWING_MODE:
Laurent@814:             Viewer.OnChar(self, event)
Laurent@814:         else:
Laurent@814:             xpos, ypos = self.GetScrollPos(wx.HORIZONTAL), self.GetScrollPos(wx.VERTICAL)
Laurent@814:             xmax = self.GetScrollRange(wx.HORIZONTAL) - self.GetScrollThumb(wx.HORIZONTAL)
Laurent@814:             ymax = self.GetScrollRange(wx.VERTICAL) - self.GetScrollThumb(wx.VERTICAL)
Laurent@814:             keycode = event.GetKeyCode()
Laurent@814:             if keycode == wx.WXK_DELETE and self.SelectedElement:
Laurent@814:                 if self.IsBlock(self.SelectedElement):
Laurent@814:                     self.SelectedElement.Delete()
Laurent@814:                 elif self.IsWire(self.SelectedElement):
Laurent@814:                     self.DeleteWire(self.SelectedElement)
Laurent@814:                 elif isinstance(self.SelectedElement, Graphic_Group):
Laurent@814:                     all_wires = True
Laurent@814:                     for element in self.SelectedElement.GetElements():
Laurent@814:                         all_wires &= self.IsWire(element)
Laurent@814:                     if all_wires:
Laurent@814:                         self.DeleteWire(self.SelectedElement)
Laurent@814:                     else:
Laurent@814:                         self.SelectedElement.Delete()
Laurent@814:                 self.RefreshBuffer()
Laurent@814:                 self.RefreshScrollBars()
Laurent@814:                 self.Refresh(False)
Laurent@814:             elif keycode == wx.WXK_LEFT:
Laurent@814:                 if event.ControlDown() and event.ShiftDown():
Laurent@814:                     self.Scroll(0, ypos)
Laurent@814:                 elif event.ControlDown():
Laurent@814:                     self.Scroll(max(0, xpos - 1), ypos)
Laurent@814:             elif keycode == wx.WXK_RIGHT:
Laurent@814:                 if event.ControlDown() and event.ShiftDown():
Laurent@814:                     self.Scroll(xmax, ypos)
Laurent@814:                 elif event.ControlDown():
Laurent@814:                     self.Scroll(min(xpos + 1, xmax), ypos)
Laurent@814:             elif keycode == wx.WXK_UP:
Laurent@814:                 if event.ControlDown() and event.ShiftDown():
Laurent@814:                     self.Scroll(xpos, 0)
Laurent@814:                 elif event.ControlDown():
Laurent@814:                     self.Scroll(xpos, max(0, ypos - 1))
Laurent@814:             elif keycode == wx.WXK_DOWN:
Laurent@814:                 if event.ControlDown() and event.ShiftDown():
Laurent@814:                     self.Scroll(xpos, ymax)
Laurent@814:                 elif event.ControlDown():
Laurent@814:                     self.Scroll(xpos, min(ypos + 1, ymax))
Laurent@814:             else:
Laurent@814:                 event.Skip()
Laurent@814: 
andrej@1782:     # -------------------------------------------------------------------------------
andrej@1782:     #                  Model adding functions from Drop Target
andrej@1782:     # -------------------------------------------------------------------------------
Laurent@814: 
Laurent@814:     def AddVariableBlock(self, x, y, scaling, var_class, var_name, var_type):
Laurent@814:         if var_type == "BOOL":
Laurent@814:             id = self.GetNewId()
Laurent@814:             if var_class == INPUT:
Laurent@814:                 contact = LD_Contact(self, CONTACT_NORMAL, var_name, id)
Laurent@814:                 width, height = contact.GetMinSize()
Laurent@814:                 if scaling is not None:
Laurent@814:                     x = round(float(x) / float(scaling[0])) * scaling[0]
Laurent@814:                     y = round(float(y) / float(scaling[1])) * scaling[1]
Laurent@814:                     width = round(float(width) / float(scaling[0]) + 0.5) * scaling[0]
Laurent@814:                     height = round(float(height) / float(scaling[1]) + 0.5) * scaling[1]
Laurent@814:                 contact.SetPosition(x, y)
Laurent@814:                 contact.SetSize(width, height)
Laurent@814:                 self.AddBlock(contact)
Laurent@814:                 self.Controler.AddEditedElementContact(self.GetTagName(), id)
Laurent@814:                 self.RefreshContactModel(contact)
Laurent@814:             else:
Laurent@814:                 coil = LD_Coil(self, COIL_NORMAL, var_name, id)
Laurent@814:                 width, height = coil.GetMinSize()
Laurent@814:                 if scaling is not None:
Laurent@814:                     x = round(float(x) / float(scaling[0])) * scaling[0]
Laurent@814:                     y = round(float(y) / float(scaling[1])) * scaling[1]
Laurent@814:                     width = round(float(width) / float(scaling[0]) + 0.5) * scaling[0]
Laurent@814:                     height = round(float(height) / float(scaling[1]) + 0.5) * scaling[1]
Laurent@814:                 coil.SetPosition(x, y)
Laurent@814:                 coil.SetSize(width, height)
Laurent@814:                 self.AddBlock(coil)
Laurent@814:                 self.Controler.AddEditedElementCoil(self.GetTagName(), id)
Laurent@814:                 self.RefreshCoilModel(coil)
Laurent@814:             self.RefreshBuffer()
Laurent@814:             self.RefreshScrollBars()
Laurent@814:             self.RefreshVisibleElements()
Laurent@814:             self.Refresh(False)
Laurent@814:         else:
Laurent@814:             Viewer.AddVariableBlock(self, x, y, scaling, var_class, var_name, var_type)
Laurent@814: 
andrej@1782:     # -------------------------------------------------------------------------------
andrej@1782:     #                          Adding element functions
andrej@1782:     # -------------------------------------------------------------------------------
Laurent@814: 
Laurent@814:     def AddLadderRung(self):
Laurent@814:         dialog = LDElementDialog(self.ParentWindow, self.Controler, "coil")
Laurent@814:         dialog.SetPreviewFont(self.GetFont())
Laurent@814:         varlist = []
Laurent@1347:         vars = self.Controler.GetEditedElementInterfaceVars(self.TagName, debug=self.Debug)
Laurent@814:         if vars:
Laurent@814:             for var in vars:
Laurent@1347:                 if var.Class != "Input" and var.Type == "BOOL":
Laurent@1347:                     varlist.append(var.Name)
Laurent@1347:         returntype = self.Controler.GetEditedElementInterfaceReturnType(self.TagName, debug=self.Debug)
Laurent@814:         if returntype == "BOOL":
Laurent@814:             varlist.append(self.Controler.GetEditedElementName(self.TagName))
Laurent@814:         dialog.SetVariables(varlist)
andrej@1740:         dialog.SetValues({"name": "", "type": COIL_NORMAL})
Laurent@814:         if dialog.ShowModal() == wx.ID_OK:
Laurent@814:             values = dialog.GetValues()
Laurent@814:             startx, starty = LD_OFFSET[0], 0
Laurent@814:             if len(self.Rungs) > 0:
Laurent@814:                 bbox = self.Rungs[-1].GetBoundingBox()
Laurent@814:                 starty = bbox.y + bbox.height
Laurent@814:             starty += LD_OFFSET[1]
Laurent@814:             rung = Graphic_Group(self)
andrej@1730: 
Laurent@814:             # Create comment
Laurent@814:             id = self.GetNewId()
Laurent@814:             comment = Comment(self, _("Comment"), id)
Laurent@814:             comment.SetPosition(startx, starty)
Laurent@814:             comment.SetSize(LD_COMMENT_DEFAULTSIZE[0], LD_COMMENT_DEFAULTSIZE[1])
Laurent@814:             self.AddComment(comment)
Laurent@814:             self.RungComments.append(comment)
Laurent@814:             self.Controler.AddEditedElementComment(self.TagName, id)
Laurent@814:             self.RefreshCommentModel(comment)
Laurent@814:             starty += LD_COMMENT_DEFAULTSIZE[1] + LD_OFFSET[1]
andrej@1730: 
Laurent@814:             # Create LeftPowerRail
Laurent@814:             id = self.GetNewId()
Laurent@814:             leftpowerrail = LD_PowerRail(self, LEFTRAIL, id)
Laurent@814:             leftpowerrail.SetPosition(startx, starty)
Laurent@814:             leftpowerrail_connectors = leftpowerrail.GetConnectors()
Laurent@814:             self.AddBlock(leftpowerrail)
Laurent@814:             rung.SelectElement(leftpowerrail)
Laurent@814:             self.Controler.AddEditedElementPowerRail(self.TagName, id, LEFTRAIL)
Laurent@814:             self.RefreshPowerRailModel(leftpowerrail)
andrej@1730: 
Laurent@814:             # Create Coil
Laurent@814:             id = self.GetNewId()
Laurent@814:             coil = LD_Coil(self, values["type"], values["name"], id)
Laurent@814:             coil.SetPosition(startx, starty + (LD_LINE_SIZE - LD_ELEMENT_SIZE[1]) / 2)
Laurent@814:             coil_connectors = coil.GetConnectors()
Laurent@814:             self.AddBlock(coil)
Laurent@814:             rung.SelectElement(coil)
Laurent@814:             self.Controler.AddEditedElementCoil(self.TagName, id)
andrej@1730: 
Laurent@814:             # Create Wire between LeftPowerRail and Coil
Laurent@814:             wire = Wire(self)
Laurent@814:             start_connector = coil_connectors["inputs"][0]
Laurent@814:             end_connector = leftpowerrail_connectors["outputs"][0]
Laurent@814:             start_connector.Connect((wire, 0), False)
Laurent@814:             end_connector.Connect((wire, -1), False)
Laurent@814:             wire.ConnectStartPoint(None, start_connector)
Laurent@814:             wire.ConnectEndPoint(None, end_connector)
Laurent@814:             self.AddWire(wire)
Laurent@814:             rung.SelectElement(wire)
andrej@1730: 
Laurent@814:             # Create RightPowerRail
Laurent@814:             id = self.GetNewId()
Laurent@814:             rightpowerrail = LD_PowerRail(self, RIGHTRAIL, id)
Laurent@814:             rightpowerrail.SetPosition(startx, starty)
Laurent@814:             rightpowerrail_connectors = rightpowerrail.GetConnectors()
Laurent@814:             self.AddBlock(rightpowerrail)
Laurent@814:             rung.SelectElement(rightpowerrail)
Laurent@814:             self.Controler.AddEditedElementPowerRail(self.TagName, id, RIGHTRAIL)
andrej@1730: 
Laurent@814:             # Create Wire between LeftPowerRail and Coil
Laurent@814:             wire = Wire(self)
Laurent@814:             start_connector = rightpowerrail_connectors["inputs"][0]
Laurent@814:             end_connector = coil_connectors["outputs"][0]
Laurent@814:             start_connector.Connect((wire, 0), False)
Laurent@814:             end_connector.Connect((wire, -1), False)
Laurent@814:             wire.ConnectStartPoint(None, start_connector)
Laurent@814:             wire.ConnectEndPoint(None, end_connector)
Laurent@814:             self.AddWire(wire)
Laurent@814:             rung.SelectElement(wire)
Laurent@814:             self.RefreshPosition(coil)
Laurent@814:             self.Rungs.append(rung)
Laurent@814:             self.RefreshBuffer()
Laurent@814:             self.RefreshScrollBars()
Laurent@814:             self.RefreshVisibleElements()
Laurent@814:             self.Refresh(False)
Laurent@814: 
Laurent@814:     def AddLadderContact(self):
Laurent@814:         wires = []
Laurent@814:         if self.IsWire(self.SelectedElement):
Laurent@814:             left_element = self.SelectedElement.EndConnected
Laurent@814:             if not isinstance(left_element.GetParentBlock(), LD_Coil):
Laurent@814:                 wires.append(self.SelectedElement)
andrej@1740:         elif self.SelectedElement and isinstance(self.SelectedElement, Graphic_Group):
Laurent@814:             if False not in [self.IsWire(element) for element in self.SelectedElement.GetElements()]:
Laurent@814:                 for element in self.SelectedElement.GetElements():
Laurent@814:                     wires.append(element)
Laurent@814:         if len(wires) > 0:
Laurent@814:             dialog = LDElementDialog(self.ParentWindow, self.Controler, "contact")
Laurent@814:             dialog.SetPreviewFont(self.GetFont())
Laurent@814:             varlist = []
Laurent@1347:             vars = self.Controler.GetEditedElementInterfaceVars(self.TagName, debug=self.Debug)
Laurent@814:             if vars:
Laurent@814:                 for var in vars:
Laurent@1347:                     if var.Class != "Output" and var.Type == "BOOL":
Laurent@1347:                         varlist.append(var.Name)
Laurent@814:             dialog.SetVariables(varlist)
andrej@1740:             dialog.SetValues({"name": "", "type": CONTACT_NORMAL})
Laurent@814:             if dialog.ShowModal() == wx.ID_OK:
Laurent@814:                 values = dialog.GetValues()
Laurent@814:                 points = wires[0].GetSelectedSegmentPoints()
Laurent@814:                 id = self.GetNewId()
Laurent@814:                 contact = LD_Contact(self, values["type"], values["name"], id)
Laurent@814:                 contact.SetPosition(0, points[0].y - (LD_ELEMENT_SIZE[1] + 1) / 2)
Laurent@814:                 self.AddBlock(contact)
Laurent@814:                 self.Controler.AddEditedElementContact(self.TagName, id)
Laurent@814:                 rungindex = self.FindRung(wires[0])
Laurent@814:                 rung = self.Rungs[rungindex]
Laurent@814:                 old_bbox = rung.GetBoundingBox()
Laurent@814:                 rung.SelectElement(contact)
Laurent@814:                 connectors = contact.GetConnectors()
Laurent@814:                 left_elements = []
Laurent@814:                 right_elements = []
Laurent@814:                 left_index = []
Laurent@814:                 right_index = []
Laurent@814:                 for wire in wires:
Laurent@814:                     if wire.EndConnected not in left_elements:
Laurent@814:                         left_elements.append(wire.EndConnected)
Laurent@814:                         left_index.append(wire.EndConnected.GetWireIndex(wire))
Laurent@814:                     else:
Laurent@814:                         idx = left_elements.index(wire.EndConnected)
Laurent@814:                         left_index[idx] = min(left_index[idx], wire.EndConnected.GetWireIndex(wire))
Laurent@814:                     if wire.StartConnected not in right_elements:
Laurent@814:                         right_elements.append(wire.StartConnected)
Laurent@814:                         right_index.append(wire.StartConnected.GetWireIndex(wire))
Laurent@814:                     else:
Laurent@814:                         idx = right_elements.index(wire.StartConnected)
Laurent@814:                         right_index[idx] = min(right_index[idx], wire.StartConnected.GetWireIndex(wire))
Laurent@814:                     wire.SetSelectedSegment(None)
Laurent@814:                     wire.Clean()
Laurent@814:                     rung.SelectElement(wire)
Laurent@814:                     self.RemoveWire(wire)
Laurent@814:                 wires = []
Laurent@814:                 right_wires = []
Laurent@814:                 for i, left_element in enumerate(left_elements):
Laurent@814:                     wire = Wire(self)
Laurent@814:                     wires.append(wire)
Laurent@814:                     connectors["inputs"][0].Connect((wire, 0), False)
Laurent@814:                     left_element.InsertConnect(left_index[i], (wire, -1), False)
Laurent@814:                     wire.ConnectStartPoint(None, connectors["inputs"][0])
Laurent@814:                     wire.ConnectEndPoint(None, left_element)
Laurent@814:                 for i, right_element in enumerate(right_elements):
Laurent@814:                     wire = Wire(self)
Laurent@814:                     wires.append(wire)
Laurent@814:                     right_wires.append(wire)
Laurent@814:                     right_element.InsertConnect(right_index[i], (wire, 0), False)
Laurent@814:                     connectors["outputs"][0].Connect((wire, -1), False)
Laurent@814:                     wire.ConnectStartPoint(None, right_element)
Laurent@814:                     wire.ConnectEndPoint(None, connectors["outputs"][0])
Laurent@814:                 right_wires.reverse()
Laurent@814:                 for wire in wires:
Laurent@814:                     self.AddWire(wire)
Laurent@814:                     rung.SelectElement(wire)
Laurent@814:                 self.RefreshPosition(contact)
Laurent@814:                 if len(right_wires) > 1:
Laurent@814:                     group = Graphic_Group(self)
Laurent@814:                     group.SetSelected(False)
Laurent@814:                     for wire in right_wires:
Laurent@814:                         wire.SetSelectedSegment(-1)
Laurent@814:                         group.SelectElement(wire)
Laurent@814:                     self.SelectedElement = group
Laurent@814:                 else:
Laurent@814:                     right_wires[0].SetSelectedSegment(-1)
Laurent@814:                     self.SelectedElement = right_wires[0]
Laurent@814:                 rung.RefreshBoundingBox()
Laurent@814:                 new_bbox = rung.GetBoundingBox()
Laurent@814:                 self.RefreshRungs(new_bbox.height - old_bbox.height, rungindex + 1)
Laurent@814:                 self.RefreshBuffer()
Laurent@814:                 self.RefreshScrollBars()
Laurent@814:                 self.RefreshVisibleElements()
Laurent@814:                 self.Refresh(False)
Laurent@814:         else:
andrej@1745:             message = wx.MessageDialog(self, _("You must select the wire where a contact should be added!"), _("Error"), wx.OK | wx.ICON_ERROR)
Laurent@814:             message.ShowModal()
Laurent@814:             message.Destroy()
Laurent@814: 
Laurent@814:     def AddLadderBranch(self):
Laurent@814:         blocks = []
Laurent@814:         if self.IsBlock(self.SelectedElement):
Laurent@814:             blocks = [self.SelectedElement]
Laurent@814:         elif isinstance(self.SelectedElement, Graphic_Group):
Laurent@814:             elements = self.SelectedElement.GetElements()
Laurent@814:             for element in elements:
Laurent@814:                 blocks.append(element)
Laurent@814:         if len(blocks) > 0:
Laurent@814:             blocks_infos = []
Laurent@814:             left_elements = []
Laurent@814:             left_index = []
Laurent@814:             right_elements = []
Laurent@814:             right_index = []
Laurent@814:             for block in blocks:
Laurent@814:                 connectors = block.GetConnectors()
andrej@1740:                 block_infos = {"lefts": [], "rights": []}
Laurent@814:                 block_infos.update(connectors)
Laurent@814:                 for connector in block_infos["inputs"]:
Laurent@814:                     for wire, handle in connector.GetWires():
Laurent@814:                         found = False
Laurent@814:                         for infos in blocks_infos:
Laurent@814:                             if wire.EndConnected in infos["outputs"]:
Laurent@814:                                 for left_element in infos["lefts"]:
Laurent@814:                                     if left_element not in block_infos["lefts"]:
Laurent@814:                                         block_infos["lefts"].append(left_element)
Laurent@814:                                 found = True
Laurent@814:                         if not found and wire.EndConnected not in block_infos["lefts"]:
Laurent@814:                             block_infos["lefts"].append(wire.EndConnected)
Laurent@814:                             if wire.EndConnected not in left_elements:
Laurent@814:                                 left_elements.append(wire.EndConnected)
Laurent@814:                                 left_index.append(wire.EndConnected.GetWireIndex(wire))
Laurent@814:                             else:
Laurent@814:                                 index = left_elements.index(wire.EndConnected)
Laurent@814:                                 left_index[index] = max(left_index[index], wire.EndConnected.GetWireIndex(wire))
Laurent@814:                 for connector in block_infos["outputs"]:
Laurent@814:                     for wire, handle in connector.GetWires():
Laurent@814:                         found = False
Laurent@814:                         for infos in blocks_infos:
Laurent@814:                             if wire.StartConnected in infos["inputs"]:
Laurent@814:                                 for right_element in infos["rights"]:
Laurent@814:                                     if right_element not in block_infos["rights"]:
Laurent@814:                                         block_infos["rights"].append(right_element)
Laurent@814:                                 found = True
Laurent@814:                         if not found and wire.StartConnected not in block_infos["rights"]:
Laurent@814:                             block_infos["rights"].append(wire.StartConnected)
Laurent@814:                             if wire.StartConnected not in right_elements:
Laurent@814:                                 right_elements.append(wire.StartConnected)
Laurent@814:                                 right_index.append(wire.StartConnected.GetWireIndex(wire))
Laurent@814:                             else:
Laurent@814:                                 index = right_elements.index(wire.StartConnected)
Laurent@814:                                 right_index[index] = max(right_index[index], wire.StartConnected.GetWireIndex(wire))
Laurent@814:                 for connector in block_infos["inputs"]:
Laurent@814:                     for infos in blocks_infos:
Laurent@814:                         if connector in infos["rights"]:
Laurent@814:                             infos["rights"].remove(connector)
Laurent@814:                             if connector in right_elements:
Laurent@814:                                 index = right_elements.index(connector)
Laurent@814:                                 right_elements.pop(index)
Laurent@814:                                 right_index.pop(index)
Laurent@814:                             for right_element in block_infos["rights"]:
Laurent@814:                                 if right_element not in infos["rights"]:
Laurent@814:                                     infos["rights"].append(right_element)
Laurent@814:                 for connector in block_infos["outputs"]:
Laurent@814:                     for infos in blocks_infos:
Laurent@814:                         if connector in infos["lefts"]:
Laurent@814:                             infos["lefts"].remove(connector)
Laurent@814:                             if connector in left_elements:
Laurent@814:                                 index = left_elements.index(connector)
Laurent@814:                                 left_elements.pop(index)
Laurent@814:                                 left_index.pop(index)
Laurent@814:                             for left_element in block_infos["lefts"]:
Laurent@814:                                 if left_element not in infos["lefts"]:
Laurent@814:                                     infos["lefts"].append(left_element)
Laurent@814:                 blocks_infos.append(block_infos)
Laurent@814:             for infos in blocks_infos:
Laurent@814:                 left_elements = [element for element in infos["lefts"]]
Laurent@814:                 for left_element in left_elements:
Laurent@814:                     if isinstance(left_element.GetParentBlock(), LD_PowerRail):
Laurent@814:                         infos["lefts"].remove(left_element)
Laurent@814:                         if "LD_PowerRail" not in infos["lefts"]:
Laurent@814:                             infos["lefts"].append("LD_PowerRail")
Laurent@814:                 right_elements = [element for element in infos["rights"]]
Laurent@814:                 for right_element in right_elements:
Laurent@814:                     if isinstance(right_element.GetParentBlock(), LD_PowerRail):
Laurent@814:                         infos["rights"].remove(right_element)
Laurent@814:                         if "LD_PowerRail" not in infos["rights"]:
Laurent@814:                             infos["rights"].append("LD_PowerRail")
Laurent@814:                 infos["lefts"].sort()
Laurent@814:                 infos["rights"].sort()
Laurent@814:             lefts = blocks_infos[0]["lefts"]
Laurent@814:             rights = blocks_infos[0]["rights"]
Laurent@814:             good = True
Laurent@814:             for infos in blocks_infos[1:]:
Laurent@814:                 good &= infos["lefts"] == lefts
Laurent@814:                 good &= infos["rights"] == rights
Laurent@814:             if good:
Laurent@814:                 rungindex = self.FindRung(blocks[0])
Laurent@814:                 rung = self.Rungs[rungindex]
Laurent@814:                 old_bbox = rung.GetBoundingBox()
Laurent@814:                 left_powerrail = True
Laurent@814:                 right_powerrail = True
Laurent@814:                 for element in left_elements:
Laurent@814:                     left_powerrail &= isinstance(element.GetParentBlock(), LD_PowerRail)
Laurent@814:                 for element in right_elements:
Laurent@814:                     right_powerrail &= isinstance(element.GetParentBlock(), LD_PowerRail)
Laurent@814:                 if not left_powerrail or not right_powerrail:
Laurent@814:                     wires = []
Laurent@814:                     if left_powerrail:
Laurent@814:                         powerrail = left_elements[0].GetParentBlock()
Laurent@814:                         index = 0
andrej@1730:                         for left_element in left_elements:
Laurent@814:                             index = max(index, powerrail.GetConnectorIndex(left_element))
Laurent@814:                         powerrail.InsertConnector(index + 1)
Laurent@814:                         powerrail.RefreshModel()
Laurent@814:                         connectors = powerrail.GetConnectors()
Laurent@814:                         right_elements.reverse()
Laurent@814:                         for i, right_element in enumerate(right_elements):
Laurent@814:                             new_wire = Wire(self)
Laurent@814:                             wires.append(new_wire)
Laurent@814:                             right_element.InsertConnect(right_index[i] + 1, (new_wire, 0), False)
Laurent@814:                             connectors["outputs"][index + 1].Connect((new_wire, -1), False)
Laurent@814:                             new_wire.ConnectStartPoint(None, right_element)
Laurent@814:                             new_wire.ConnectEndPoint(None, connectors["outputs"][index + 1])
Laurent@814:                         right_elements.reverse()
Laurent@814:                     elif right_powerrail:
Laurent@814:                         dialog = LDElementDialog(self.ParentWindow, self.Controleur, "coil")
Laurent@814:                         dialog.SetPreviewFont(self.GetFont())
Laurent@814:                         varlist = []
Laurent@1347:                         vars = self.Controler.GetEditedElementInterfaceVars(self.TagName, debug=self.Debug)
Laurent@814:                         if vars:
Laurent@814:                             for var in vars:
Laurent@1347:                                 if var.Class != "Input" and var.Type == "BOOL":
Laurent@1347:                                     varlist.append(var.Name)
Laurent@1347:                         returntype = self.Controler.GetEditedElementInterfaceReturnType(self.TagName, debug=self.Debug)
Laurent@814:                         if returntype == "BOOL":
Laurent@814:                             varlist.append(self.Controler.GetEditedElementName(self.TagName))
Laurent@814:                         dialog.SetVariables(varlist)
andrej@1740:                         dialog.SetValues({"name": "", "type": COIL_NORMAL})
Laurent@814:                         if dialog.ShowModal() == wx.ID_OK:
Laurent@814:                             values = dialog.GetValues()
Laurent@814:                             powerrail = right_elements[0].GetParentBlock()
Laurent@814:                             index = 0
andrej@1730:                             for right_element in right_elements:
Laurent@814:                                 index = max(index, powerrail.GetConnectorIndex(right_element))
Laurent@814:                             powerrail.InsertConnector(index + 1)
Laurent@814:                             powerrail.RefreshModel()
Laurent@814:                             connectors = powerrail.GetConnectors()
andrej@1730: 
Laurent@814:                             # Create Coil
Laurent@814:                             id = self.GetNewId()
Laurent@814:                             coil = LD_Coil(self, values["type"], values["name"], id)
Laurent@814:                             pos = blocks[0].GetPosition()
Laurent@814:                             coil.SetPosition(pos[0], pos[1] + LD_LINE_SIZE)
Laurent@814:                             self.AddBlock(coil)
Laurent@814:                             rung.SelectElement(coil)
Laurent@814:                             self.Controler.AddEditedElementCoil(self.TagName, id)
Laurent@814:                             coil_connectors = coil.GetConnectors()
andrej@1730: 
Laurent@814:                             # Create Wire between LeftPowerRail and Coil
Laurent@814:                             wire = Wire(self)
Laurent@814:                             connectors["inputs"][index + 1].Connect((wire, 0), False)
Laurent@814:                             coil_connectors["outputs"][0].Connect((wire, -1), False)
Laurent@814:                             wire.ConnectStartPoint(None, connectors["inputs"][index + 1])
Laurent@814:                             wire.ConnectEndPoint(None, coil_connectors["outputs"][0])
Laurent@814:                             self.AddWire(wire)
Laurent@814:                             rung.SelectElement(wire)
Laurent@814:                             left_elements.reverse()
andrej@1730: 
Laurent@814:                             for i, left_element in enumerate(left_elements):
Laurent@814:                                 # Create Wire between LeftPowerRail and Coil
Laurent@814:                                 new_wire = Wire(self)
Laurent@814:                                 wires.append(new_wire)
Laurent@814:                                 coil_connectors["inputs"][0].Connect((new_wire, 0), False)
Laurent@814:                                 left_element.InsertConnect(left_index[i] + 1, (new_wire, -1), False)
Laurent@814:                                 new_wire.ConnectStartPoint(None, coil_connectors["inputs"][0])
Laurent@814:                                 new_wire.ConnectEndPoint(None, left_element)
andrej@1730: 
Laurent@814:                             self.RefreshPosition(coil)
Laurent@814:                     else:
Laurent@814:                         left_elements.reverse()
Laurent@814:                         right_elements.reverse()
Laurent@814:                         for i, left_element in enumerate(left_elements):
Laurent@814:                             for j, right_element in enumerate(right_elements):
Laurent@814:                                 exist = False
Laurent@814:                                 for wire, handle in right_element.GetWires():
Laurent@814:                                     exist |= wire.EndConnected == left_element
Laurent@814:                                 if not exist:
Laurent@814:                                     new_wire = Wire(self)
Laurent@814:                                     wires.append(new_wire)
Laurent@814:                                     right_element.InsertConnect(right_index[j] + 1, (new_wire, 0), False)
Laurent@814:                                     left_element.InsertConnect(left_index[i] + 1, (new_wire, -1), False)
Laurent@814:                                     new_wire.ConnectStartPoint(None, right_element)
Laurent@814:                                     new_wire.ConnectEndPoint(None, left_element)
Laurent@814:                     wires.reverse()
Laurent@814:                     for wire in wires:
Laurent@814:                         self.AddWire(wire)
Laurent@814:                         rung.SelectElement(wire)
Laurent@814:                     right_elements.reverse()
Laurent@814:                 for block in blocks:
Laurent@814:                     self.RefreshPosition(block)
Laurent@814:                 for right_element in right_elements:
Laurent@814:                     self.RefreshPosition(right_element.GetParentBlock())
Laurent@814:                 self.SelectedElement.RefreshBoundingBox()
Laurent@814:                 rung.RefreshBoundingBox()
Laurent@814:                 new_bbox = rung.GetBoundingBox()
Laurent@814:                 self.RefreshRungs(new_bbox.height - old_bbox.height, rungindex + 1)
Laurent@814:                 self.RefreshBuffer()
Laurent@814:                 self.RefreshScrollBars()
Laurent@814:                 self.RefreshVisibleElements()
Laurent@814:                 self.Refresh(False)
Laurent@814:             else:
andrej@1745:                 message = wx.MessageDialog(self, _("The group of block must be coherent!"), _("Error"), wx.OK | wx.ICON_ERROR)
Laurent@814:                 message.ShowModal()
Laurent@814:                 message.Destroy()
Laurent@814:         else:
andrej@1745:             message = wx.MessageDialog(self, _("You must select the block or group of blocks around which a branch should be added!"), _("Error"), wx.OK | wx.ICON_ERROR)
Laurent@814:             message.ShowModal()
Laurent@814:             message.Destroy()
Laurent@814: 
Laurent@814:     def AddLadderBlock(self):
andrej@1745:         message = wx.MessageDialog(self, _("This option isn't available yet!"), _("Warning"), wx.OK | wx.ICON_EXCLAMATION)
Laurent@814:         message.ShowModal()
Laurent@814:         message.Destroy()
Laurent@814: 
andrej@1782:     # -------------------------------------------------------------------------------
andrej@1782:     #                          Delete element functions
andrej@1782:     # -------------------------------------------------------------------------------
Laurent@814: 
Laurent@814:     def DeleteContact(self, contact):
Laurent@814:         if self.GetDrawingMode() == FREEDRAWING_MODE:
Laurent@814:             Viewer.DeleteContact(self, contact)
Laurent@814:         else:
Laurent@814:             rungindex = self.FindRung(contact)
Laurent@814:             rung = self.Rungs[rungindex]
Laurent@814:             old_bbox = rung.GetBoundingBox()
Laurent@814:             connectors = contact.GetConnectors()
Laurent@814:             input_wires = [wire for wire, handle in connectors["inputs"][0].GetWires()]
Laurent@814:             output_wires = [wire for wire, handle in connectors["outputs"][0].GetWires()]
Laurent@814:             left_elements = [(wire.EndConnected, wire.EndConnected.GetWireIndex(wire)) for wire in input_wires]
Laurent@814:             right_elements = [(wire.StartConnected, wire.StartConnected.GetWireIndex(wire)) for wire in output_wires]
Laurent@814:             for wire in input_wires:
Laurent@814:                 wire.Clean()
Laurent@814:                 rung.SelectElement(wire)
Laurent@814:                 self.RemoveWire(wire)
Laurent@814:             for wire in output_wires:
Laurent@814:                 wire.Clean()
Laurent@814:                 rung.SelectElement(wire)
Laurent@814:                 self.RemoveWire(wire)
Laurent@814:             rung.SelectElement(contact)
Laurent@814:             contact.Clean()
Laurent@814:             left_elements.reverse()
Laurent@814:             right_elements.reverse()
Laurent@814:             powerrail = len(left_elements) == 1 and isinstance(left_elements[0][0].GetParentBlock(), LD_PowerRail)
Laurent@814:             for left_element, left_index in left_elements:
Laurent@814:                 for right_element, right_index in right_elements:
Laurent@814:                     wire_removed = []
Laurent@814:                     for wire, handle in right_element.GetWires():
Laurent@814:                         if wire.EndConnected == left_element:
Laurent@814:                             wire_removed.append(wire)
Laurent@814:                         elif isinstance(wire.EndConnected.GetParentBlock(), LD_PowerRail) and powerrail:
Laurent@814:                             left_powerrail = wire.EndConnected.GetParentBlock()
Laurent@814:                             index = left_powerrail.GetConnectorIndex(wire.EndConnected)
Laurent@814:                             left_powerrail.DeleteConnector(index)
Laurent@814:                             wire_removed.append(wire)
Laurent@814:                     for wire in wire_removed:
Laurent@814:                         wire.Clean()
Laurent@814:                         self.RemoveWire(wire)
Laurent@814:                         rung.SelectElement(wire)
Laurent@814:             wires = []
Laurent@814:             for left_element, left_index in left_elements:
Laurent@814:                 for right_element, right_index in right_elements:
Laurent@814:                     wire = Wire(self)
Laurent@814:                     wires.append(wire)
Laurent@814:                     right_element.InsertConnect(right_index, (wire, 0), False)
Laurent@814:                     left_element.InsertConnect(left_index, (wire, -1), False)
Laurent@814:                     wire.ConnectStartPoint(None, right_element)
Laurent@814:                     wire.ConnectEndPoint(None, left_element)
Laurent@814:             wires.reverse()
Laurent@814:             for wire in wires:
Laurent@814:                 self.AddWire(wire)
Laurent@814:                 rung.SelectElement(wire)
Laurent@814:             right_elements.reverse()
Laurent@814:             for right_element, right_index in right_elements:
Laurent@814:                 self.RefreshPosition(right_element.GetParentBlock())
Laurent@814:             self.RemoveBlock(contact)
Laurent@814:             self.Controler.RemoveEditedElementInstance(self.TagName, contact.GetId())
Laurent@814:             rung.RefreshBoundingBox()
Laurent@814:             new_bbox = rung.GetBoundingBox()
Laurent@814:             self.RefreshRungs(new_bbox.height - old_bbox.height, rungindex + 1)
Laurent@814:             self.SelectedElement = None
Laurent@814: 
Laurent@814:     def RecursiveDeletion(self, element, rung):
Laurent@814:         connectors = element.GetConnectors()
Laurent@814:         input_wires = [wire for wire, handle in connectors["inputs"][0].GetWires()]
Laurent@814:         left_elements = [wire.EndConnected for wire in input_wires]
Laurent@814:         rung.SelectElement(element)
Laurent@814:         element.Clean()
Laurent@814:         for wire in input_wires:
Laurent@814:             wire.Clean()
Laurent@814:             self.RemoveWire(wire)
Laurent@814:             rung.SelectElement(wire)
Laurent@814:         self.RemoveBlock(element)
Laurent@814:         self.Controler.RemoveEditedElementInstance(self.TagName, element.GetId())
Laurent@814:         for left_element in left_elements:
Laurent@814:             block = left_element.GetParentBlock()
Laurent@814:             if len(left_element.GetWires()) == 0:
Laurent@814:                 self.RecursiveDeletion(block, rung)
Laurent@814:             else:
Laurent@814:                 self.RefreshPosition(block)
Laurent@814: 
Laurent@814:     def DeleteCoil(self, coil):
Laurent@814:         if self.GetDrawingMode() == FREEDRAWING_MODE:
Laurent@814:             Viewer.DeleteContact(self, coil)
Laurent@814:         else:
Laurent@814:             rungindex = self.FindRung(coil)
Laurent@814:             rung = self.Rungs[rungindex]
Laurent@814:             old_bbox = rung.GetBoundingBox()
Laurent@814:             nbcoils = 0
Laurent@814:             for element in rung.GetElements():
Laurent@814:                 if isinstance(element, LD_Coil):
Laurent@814:                     nbcoils += 1
Laurent@814:             if nbcoils > 1:
Laurent@814:                 connectors = coil.GetConnectors()
Laurent@814:                 output_wires = [wire for wire, handle in connectors["outputs"][0].GetWires()]
Laurent@814:                 right_elements = [wire.StartConnected for wire in output_wires]
Laurent@814:                 for wire in output_wires:
Laurent@814:                     wire.Clean()
Laurent@814:                     self.Wires.remove(wire)
Laurent@814:                     self.Elements.remove(wire)
Laurent@814:                     rung.SelectElement(wire)
Laurent@814:                 for right_element in right_elements:
Laurent@814:                     right_block = right_element.GetParentBlock()
Laurent@814:                     if isinstance(right_block, LD_PowerRail):
Laurent@814:                         if len(right_element.GetWires()) == 0:
Laurent@814:                             index = right_block.GetConnectorIndex(right_element)
Laurent@814:                             right_block.DeleteConnector(index)
Laurent@814:                             powerrail_connectors = right_block.GetConnectors()
Laurent@814:                             for connector in powerrail_connectors["inputs"]:
Laurent@814:                                 for wire, handle in connector.GetWires():
Laurent@814:                                     block = wire.EndConnected.GetParentBlock()
Laurent@814:                                     endpoint = wire.EndConnected.GetPosition(False)
Laurent@814:                                     startpoint = connector.GetPosition(False)
Laurent@814:                                     block.Move(0, startpoint.y - endpoint.y)
Laurent@814:                                     self.RefreshPosition(block)
Laurent@814:                 self.RecursiveDeletion(coil, rung)
Laurent@814:             else:
Laurent@814:                 for element in rung.GetElements():
Laurent@814:                     if self.IsWire(element):
Laurent@814:                         element.Clean()
Laurent@814:                         self.RemoveWire(element)
Laurent@814:                 for element in rung.GetElements():
Laurent@814:                     if self.IsBlock(element):
Laurent@814:                         self.Controler.RemoveEditedElementInstance(self.TagName, element.GetId())
Laurent@814:                         self.RemoveBlock(element)
Laurent@814:                 self.Controler.RemoveEditedElementInstance(self.TagName, self.Comments[rungindex].GetId())
Laurent@814:                 self.RemoveComment(self.RungComments[rungindex])
Laurent@814:                 self.RungComments.pop(rungindex)
Laurent@814:                 self.Rungs.pop(rungindex)
Laurent@814:                 if rungindex < len(self.Rungs):
Laurent@814:                     next_bbox = self.Rungs[rungindex].GetBoundingBox()
Laurent@814:                     self.RefreshRungs(old_bbox.y - next_bbox.y, rungindex)
Laurent@814:             self.SelectedElement = None
Laurent@814: 
Laurent@814:     def DeleteWire(self, wire):
Laurent@814:         if self.GetDrawingMode() == FREEDRAWING_MODE:
Laurent@814:             Viewer.DeleteWire(self, wire)
Laurent@814:         else:
Laurent@814:             wires = []
Laurent@814:             left_elements = []
Laurent@814:             right_elements = []
Laurent@814:             if self.IsWire(wire):
Laurent@814:                 wires = [wire]
Laurent@814:             elif isinstance(wire, Graphic_Group):
Laurent@814:                 for element in wire.GetElements():
Laurent@814:                     if self.IsWire(element):
Laurent@814:                         wires.append(element)
Laurent@814:                     else:
Laurent@814:                         wires = []
Laurent@814:                         break
Laurent@814:             if len(wires) > 0:
Laurent@814:                 rungindex = self.FindRung(wires[0])
Laurent@814:                 rung = self.Rungs[rungindex]
Laurent@814:                 old_bbox = rung.GetBoundingBox()
Laurent@814:                 for wire in wires:
Laurent@814:                     connections = wire.GetSelectedSegmentConnections()
Laurent@814:                     left_block = wire.EndConnected.GetParentBlock()
Laurent@814:                     if wire.EndConnected not in left_elements:
Laurent@814:                         left_elements.append(wire.EndConnected)
Laurent@814:                     if wire.StartConnected not in right_elements:
Laurent@814:                         right_elements.append(wire.StartConnected)
Laurent@814:                     if connections == (False, False) or connections == (False, True) and isinstance(left_block, LD_PowerRail):
Laurent@814:                         wire.Clean()
Laurent@814:                         self.RemoveWire(wire)
Laurent@814:                         rung.SelectElement(wire)
Laurent@814:                 for left_element in left_elements:
Laurent@814:                     left_block = left_element.GetParentBlock()
Laurent@814:                     if isinstance(left_block, LD_PowerRail):
Laurent@814:                         if len(left_element.GetWires()) == 0:
Laurent@814:                             index = left_block.GetConnectorIndex(left_element)
Laurent@814:                             left_block.DeleteConnector(index)
Laurent@814:                     else:
Laurent@814:                         connectors = left_block.GetConnectors()
Laurent@814:                         for connector in connectors["outputs"]:
Laurent@814:                             for wire, handle in connector.GetWires():
Laurent@814:                                 self.RefreshPosition(wire.StartConnected.GetParentBlock())
Laurent@814:                 for right_element in right_elements:
Laurent@814:                     self.RefreshPosition(right_element.GetParentBlock())
Laurent@814:                 rung.RefreshBoundingBox()
Laurent@814:                 new_bbox = rung.GetBoundingBox()
Laurent@814:                 self.RefreshRungs(new_bbox.height - old_bbox.height, rungindex + 1)
Laurent@814:                 self.SelectedElement = None
Laurent@814: 
andrej@1782:     # -------------------------------------------------------------------------------
andrej@1782:     #                        Refresh element position functions
andrej@1782:     # -------------------------------------------------------------------------------
Laurent@814: 
Laurent@814:     def RefreshPosition(self, element, recursive=True):
Laurent@814:         # If element is LeftPowerRail, no need to update position
Laurent@814:         if isinstance(element, LD_PowerRail) and element.GetType() == LEFTRAIL:
Laurent@814:             element.RefreshModel()
Laurent@814:             return
andrej@1730: 
Laurent@814:         # Extract max position of the elements connected to input
Laurent@814:         connectors = element.GetConnectors()
Laurent@814:         position = element.GetPosition()
Laurent@814:         maxx = 0
Laurent@814:         onlyone = []
Laurent@814:         for connector in connectors["inputs"]:
Laurent@814:             onlyone.append(len(connector.GetWires()) == 1)
Laurent@814:             for wire, handle in connector.GetWires():
Laurent@814:                 onlyone[-1] &= len(wire.EndConnected.GetWires()) == 1
Laurent@814:                 leftblock = wire.EndConnected.GetParentBlock()
Laurent@814:                 pos = leftblock.GetPosition()
Laurent@814:                 size = leftblock.GetSize()
Laurent@814:                 maxx = max(maxx, pos[0] + size[0])
andrej@1730: 
Laurent@814:         # Refresh position of element
Laurent@814:         if isinstance(element, LD_Coil):
Laurent@814:             interval = LD_WIRECOIL_SIZE
Laurent@814:         else:
Laurent@814:             interval = LD_WIRE_SIZE
Laurent@814:         if False in onlyone:
Laurent@814:             interval += LD_WIRE_SIZE
Laurent@814:         movex = maxx + interval - position[0]
Laurent@814:         element.Move(movex, 0)
Laurent@814:         position = element.GetPosition()
andrej@1730: 
Laurent@814:         # Extract blocks connected to inputs
Laurent@814:         blocks = []
Laurent@814:         for i, connector in enumerate(connectors["inputs"]):
Laurent@814:             for j, (wire, handle) in enumerate(connector.GetWires()):
Laurent@814:                 blocks.append(wire.EndConnected.GetParentBlock())
andrej@1730: 
Laurent@814:         for i, connector in enumerate(connectors["inputs"]):
Laurent@814:             startpoint = connector.GetPosition(False)
Laurent@814:             previous_blocks = []
Laurent@814:             block_list = []
Laurent@814:             start_offset = 0
Laurent@814:             if not onlyone[i]:
Laurent@814:                 middlepoint = maxx + LD_WIRE_SIZE
Laurent@814:             for j, (wire, handle) in enumerate(connector.GetWires()):
Laurent@814:                 block = wire.EndConnected.GetParentBlock()
Laurent@814:                 if isinstance(element, LD_PowerRail):
Laurent@814:                     pos = block.GetPosition()
Laurent@814:                     size = leftblock.GetSize()
Laurent@814:                     movex = position[0] - LD_WIRE_SIZE - size[0] - pos[0]
Laurent@814:                     block.Move(movex, 0)
Laurent@814:                 endpoint = wire.EndConnected.GetPosition(False)
Laurent@814:                 if j == 0:
Laurent@814:                     if not onlyone[i] and wire.EndConnected.GetWireIndex(wire) > 0:
Laurent@814:                         start_offset = endpoint.y - startpoint.y
Laurent@814:                     offset = start_offset
Laurent@814:                 else:
Laurent@814:                     offset = start_offset + LD_LINE_SIZE * CalcBranchSize(previous_blocks, blocks)
Laurent@814:                 if block in block_list:
Laurent@814:                     wires = wire.EndConnected.GetWires()
Laurent@814:                     endmiddlepoint = wires[0][0].StartConnected.GetPosition(False)[0] - LD_WIRE_SIZE
Laurent@814:                     points = [startpoint, wx.Point(middlepoint, startpoint.y),
Laurent@814:                               wx.Point(middlepoint, startpoint.y + offset),
Laurent@814:                               wx.Point(endmiddlepoint, startpoint.y + offset),
Laurent@814:                               wx.Point(endmiddlepoint, endpoint.y), endpoint]
Laurent@814:                 else:
Laurent@814:                     if startpoint.y + offset != endpoint.y:
Laurent@814:                         if isinstance(element, LD_PowerRail):
Laurent@814:                             element.MoveConnector(connector, startpoint.y - endpoint.y)
Laurent@814:                         elif isinstance(block, LD_PowerRail):
Laurent@814:                             block.MoveConnector(wire.EndConnected, startpoint.y - endpoint.y)
Laurent@814:                         else:
Laurent@814:                             block.Move(0, startpoint.y + offset - endpoint.y)
Laurent@814:                             self.RefreshPosition(block, False)
Laurent@814:                         endpoint = wire.EndConnected.GetPosition(False)
Laurent@814:                     if not onlyone[i]:
Laurent@814:                         points = [startpoint, wx.Point(middlepoint, startpoint.y),
Laurent@814:                                   wx.Point(middlepoint, endpoint.y), endpoint]
Laurent@814:                     else:
Laurent@814:                         points = [startpoint, endpoint]
Laurent@814:                 wire.SetPoints(points, False)
Laurent@814:                 previous_blocks.append(block)
Laurent@814:                 blocks.remove(block)
Laurent@814:                 ExtractNextBlocks(block, block_list)
andrej@1730: 
Laurent@814:         element.RefreshModel(False)
Laurent@814:         if recursive:
Laurent@814:             for connector in connectors["outputs"]:
Laurent@814:                 for wire, handle in connector.GetWires():
Laurent@814:                     self.RefreshPosition(wire.StartConnected.GetParentBlock())
andrej@1730: 
Laurent@814:     def RefreshRungs(self, movey, fromidx):
Laurent@814:         if movey != 0:
Laurent@814:             for i in xrange(fromidx, len(self.Rungs)):
Laurent@814:                 self.RungComments[i].Move(0, movey)
Laurent@814:                 self.RungComments[i].RefreshModel()
Laurent@814:                 self.Rungs[i].Move(0, movey)
Laurent@814:                 for element in self.Rungs[i].GetElements():
Laurent@814:                     if self.IsBlock(element):
Laurent@814:                         self.RefreshPosition(element)
Laurent@814: 
andrej@1782:     # -------------------------------------------------------------------------------
andrej@1782:     #                          Edit element content functions
andrej@1782:     # -------------------------------------------------------------------------------
Laurent@814: 
Laurent@814:     def EditPowerRailContent(self, powerrail):
Laurent@814:         if self.GetDrawingMode() == FREEDRAWING_MODE:
Laurent@814:             Viewer.EditPowerRailContent(self, powerrail)