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