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 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: if wx.VERSION >= (2, 8, 0): lbessard@121: import wx.aui lbessard@121: lbessard@121: class LD_MDIViewer(wx.aui.AuiMDIChildFrame): lbessard@121: def __init__(self, parent, tagname, window, controler): lbessard@121: wx.aui.AuiMDIChildFrame.__init__(self, parent, -1, title = "") lbessard@121: lbessard@121: sizer = wx.BoxSizer(wx.HORIZONTAL) lbessard@121: lbessard@121: self.Viewer = LD_Viewer(self, tagname, window, controler) lbessard@121: lbessard@121: sizer.AddWindow(self.Viewer, 1, border=0, flag=wx.GROW) lbessard@121: lbessard@121: self.SetSizer(sizer) lbessard@121: lbessard@121: def GetViewer(self): lbessard@121: return self.Viewer 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@121: def __init__(self, parent, tagname, window, controler): lbessard@121: Viewer.__init__(self, parent, tagname, window, controler) 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: etisserant@0: def RefreshView(self): etisserant@0: Viewer.RefreshView(self) 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: etisserant@0: def loadInstance(self, instance, ids): etisserant@0: Viewer.loadInstance(self, instance, ids) 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 = [] lbessard@80: for connector in instance["connectors"]: 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: lbessard@80: raise "ValueError", "Ladder element with id %d is on more than one rung."%instance["id"] lbessard@80: element = self.FindElementById(instance["id"]) lbessard@80: self.Rungs[rungs[0]].SelectElement(element) lbessard@80: for connector in element.GetConnectors(): lbessard@80: for wire, num in connector.GetWires(): lbessard@80: self.Rungs[rungs[0]].SelectElement(wire) lbessard@80: self.RefreshPosition(element) lbessard@80: elif instance["type"] in ["contact", "coil"]: lbessard@80: rungs = [] lbessard@80: for link in instance["connectors"]["input"]["links"]: etisserant@0: connected = self.FindElementById(link["refLocalId"]) etisserant@0: rung = self.FindRung(connected) etisserant@0: if rung not in rungs: etisserant@0: rungs.append(rung) lbessard@80: if len(rungs) > 1: lbessard@80: raise "ValueError", "Ladder element with id %d is on more than one rung."%instance["id"] lbessard@80: element = self.FindElementById(instance["id"]) lbessard@80: self.Rungs[rungs[0]].SelectElement(element) lbessard@80: for wire, num in element.GetConnectors()["input"].GetWires(): etisserant@0: self.Rungs[rungs[0]].SelectElement(wire) lbessard@67: 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) etisserant@0: etisserant@0: #------------------------------------------------------------------------------- etisserant@0: # Search Element functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: etisserant@0: def FindRung(self, element): etisserant@0: for i, rung in enumerate(self.Rungs): etisserant@0: if rung.IsElementIn(element): etisserant@0: return i etisserant@0: return None etisserant@0: lbessard@80: def FindElement(self, pos, exclude_group = False): lbessard@80: if self.GetDrawingMode() == FREEDRAWING_MODE: lbessard@80: return Viewer.FindElement(self, pos, exclude_group) lbessard@27: lbessard@80: if self.SelectedElement and not (exclude_group and isinstance(self.SelectedElement, Graphic_Group)): lbessard@80: if self.SelectedElement.HitTest(pos) 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): etisserant@0: if element.HitTest(pos) or element.TestHandle(pos) != (0, 0): etisserant@0: elements.append(element) etisserant@0: if len(elements) == 1: etisserant@0: return elements[0] etisserant@0: elif len(elements) > 1: etisserant@0: group = Graphic_Group(self) etisserant@0: for element in elements: 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): lbessard@27: if self.GetDrawingMode() == FREEDRAWING_MODE: lbessard@27: return Viewer.SearchElements(self, bbox) lbessard@27: etisserant@0: elements = [] etisserant@0: for element in self.Blocks: 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: lbessard@27: dc = self.GetLogicalDC() lbessard@27: pos = event.GetLogicalPosition(dc) etisserant@0: element = self.FindElement(pos) 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 lbessard@27: self.SelectedElement.OnLeftDown(event, dc, self.Scaling) lbessard@144: self.SelectedElement.Refresh() etisserant@0: else: etisserant@0: self.rubberBand.Reset() lbessard@27: self.rubberBand.OnLeftDown(event, dc, 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() lbessard@64: wx.CallAfter(self.SetCursor, wx.NullCursor) 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: lbessard@27: dc = self.GetLogicalDC() lbessard@27: pos = event.GetLogicalPosition(dc) lbessard@27: element = self.FindElement(pos) lbessard@27: if element: lbessard@27: if self.SelectedElement and self.SelectedElement != element: lbessard@27: self.SelectedElement.SetSelected(False) lbessard@27: self.SelectedElement = element lbessard@42: if self.IsWire(self.SelectedElement): lbessard@27: self.SelectedElement.SetSelectedSegment(0) lbessard@27: else: lbessard@27: self.SelectedElement.SetSelected(True) lbessard@27: self.SelectedElement.OnRightUp(event, dc, self.Scaling) lbessard@144: self.SelectedElement.Refresh() lbessard@64: wx.CallAfter(self.SetCursor, wx.NullCursor) lbessard@27: event.Skip() lbessard@27: lbessard@27: def OnViewerLeftDClick(self, event): lbessard@27: if self.GetDrawingMode() == FREEDRAWING_MODE: lbessard@27: Viewer.OnViewerLeftDClick(self, event) lbessard@27: elif self.Mode == MODE_SELECTION and self.SelectedElement: lbessard@27: self.SelectedElement.OnLeftDClick(event, self.GetLogicalDC(), self.Scaling) lbessard@144: self.SelectedElement.Refresh() etisserant@0: event.Skip() etisserant@0: lbessard@27: def OnViewerMotion(self, event): lbessard@27: if self.GetDrawingMode() == FREEDRAWING_MODE: lbessard@27: Viewer.OnViewerMotion(self, event) lbessard@42: elif self.rubberBand.IsShown(): lbessard@42: self.rubberBand.OnMotion(event, self.GetLogicalDC(), self.Scaling) 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@56: self.Refresh() 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: #------------------------------------------------------------------------------- etisserant@0: # Adding element functions etisserant@0: #------------------------------------------------------------------------------- etisserant@0: lbessard@42: def AddLadderRung(self): lbessard@90: dialog = LDElementDialog(self.ParentWindow, "coil") etisserant@0: varlist = [] lbessard@121: vars = self.Controler.GetEditedElementInterfaceVars(self.TagName) 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@121: returntype = self.Controler.GetEditedElementInterfaceReturnType(self.TagName) 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) etisserant@0: # Create comment etisserant@0: id = self.GetNewId() etisserant@0: comment = Comment(self, "Commentaire", id) etisserant@0: comment.SetPosition(startx, starty) etisserant@0: comment.SetSize(LD_COMMENT_DEFAULTSIZE[0], LD_COMMENT_DEFAULTSIZE[1]) 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] etisserant@0: # Create LeftPowerRail etisserant@0: id = self.GetNewId() etisserant@0: leftpowerrail = LD_PowerRail(self, LEFTRAIL, id) etisserant@0: leftpowerrail.SetPosition(startx, starty) lbessard@42: self.AddBlock(leftpowerrail) etisserant@0: rung.SelectElement(leftpowerrail) lbessard@121: self.Controler.AddEditedElementPowerRail(self.TagName, id, LEFTRAIL) etisserant@0: self.RefreshPowerRailModel(leftpowerrail) etisserant@0: # Create Coil etisserant@0: id = self.GetNewId() etisserant@0: coil = LD_Coil(self, values["type"], values["name"], id) etisserant@0: coil.SetPosition(startx, starty + (LD_LINE_SIZE - LD_ELEMENT_SIZE[1]) / 2) lbessard@42: self.AddBlock(coil) etisserant@0: rung.SelectElement(coil) lbessard@121: self.Controler.AddEditedElementCoil(self.TagName, id) etisserant@0: # Create Wire between LeftPowerRail and Coil etisserant@0: wire = Wire(self) etisserant@0: start_connector = coil.GetConnectors()["input"] etisserant@0: end_connector = leftpowerrail.GetConnectors()[0] etisserant@0: start_connector.Connect((wire, 0), False) etisserant@0: end_connector.Connect((wire, -1), False) etisserant@0: wire.ConnectStartPoint(None, start_connector) etisserant@0: wire.ConnectEndPoint(None, end_connector) lbessard@50: self.AddWire(wire) etisserant@0: rung.SelectElement(wire) etisserant@0: # Create RightPowerRail etisserant@0: id = self.GetNewId() etisserant@0: rightpowerrail = LD_PowerRail(self, RIGHTRAIL, id) etisserant@0: rightpowerrail.SetPosition(startx, starty) lbessard@42: self.AddBlock(rightpowerrail) etisserant@0: rung.SelectElement(rightpowerrail) lbessard@121: self.Controler.AddEditedElementPowerRail(self.TagName, id, RIGHTRAIL) etisserant@0: # Create Wire between LeftPowerRail and Coil etisserant@0: wire = Wire(self) etisserant@0: start_connector = rightpowerrail.GetConnectors()[0] etisserant@0: end_connector = coil.GetConnectors()["output"] etisserant@0: start_connector.Connect((wire, 0), False) etisserant@0: end_connector.Connect((wire, -1), False) etisserant@0: wire.ConnectStartPoint(None, start_connector) etisserant@0: wire.ConnectEndPoint(None, end_connector) 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() etisserant@0: self.Refresh() 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: lbessard@90: dialog = LDElementDialog(self.ParentWindow, "contact") etisserant@0: varlist = [] lbessard@121: vars = self.Controler.GetEditedElementInterfaceVars(self.TagName) 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) etisserant@0: connectors["input"].Connect((wire, 0), False) etisserant@0: left_element.InsertConnect(left_index[i], (wire, -1), False) etisserant@0: wire.ConnectStartPoint(None, connectors["input"]) etisserant@0: wire.ConnectEndPoint(None, left_element) etisserant@0: for i, right_element in enumerate(right_elements): etisserant@0: wire = Wire(self) etisserant@0: wires.append(wire) etisserant@0: right_wires.append(wire) etisserant@0: right_element.InsertConnect(right_index[i], (wire, 0), False) etisserant@0: connectors["output"].Connect((wire, -1), False) etisserant@0: wire.ConnectStartPoint(None, right_element) etisserant@0: wire.ConnectEndPoint(None, connectors["output"]) etisserant@0: right_wires.reverse() etisserant@0: for wire in wires: 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() etisserant@0: self.Refresh() lbessard@7: else: lbessard@64: 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() etisserant@0: block_infos = {"inputs":[],"outputs":[],"lefts":[],"rights":[]} etisserant@0: if "inputs" in connectors: etisserant@0: block_infos["inputs"] = connectors["inputs"] etisserant@0: if "outputs" in connectors: etisserant@0: block_infos["outputs"] = connectors["outputs"] etisserant@0: if "input" in connectors: etisserant@0: block_infos["inputs"] = [connectors["input"]] etisserant@0: if "output" in connectors: etisserant@0: block_infos["outputs"] = [connectors["output"]] etisserant@0: for connector in block_infos["inputs"]: etisserant@0: for wire, handle in connector.GetWires(): etisserant@0: found = False etisserant@0: for infos in blocks_infos: etisserant@0: if wire.EndConnected in infos["outputs"]: etisserant@0: for left_element in infos["lefts"]: etisserant@0: if left_element not in block_infos["lefts"]: etisserant@0: block_infos["lefts"].append(left_element) etisserant@0: found = True etisserant@0: if not found and wire.EndConnected not in block_infos["lefts"]: etisserant@0: block_infos["lefts"].append(wire.EndConnected) etisserant@0: if wire.EndConnected not in left_elements: etisserant@0: left_elements.append(wire.EndConnected) etisserant@0: left_index.append(wire.EndConnected.GetWireIndex(wire)) etisserant@0: else: etisserant@0: index = left_elements.index(wire.EndConnected) etisserant@0: left_index[index] = max(left_index[index], wire.EndConnected.GetWireIndex(wire)) etisserant@0: for connector in block_infos["outputs"]: etisserant@0: for wire, handle in connector.GetWires(): etisserant@0: found = False etisserant@0: for infos in blocks_infos: etisserant@0: if wire.StartConnected in infos["inputs"]: etisserant@0: for right_element in infos["rights"]: etisserant@0: if right_element not in block_infos["rights"]: etisserant@0: block_infos["rights"].append(right_element) etisserant@0: found = True etisserant@0: if not found and wire.StartConnected not in block_infos["rights"]: etisserant@0: block_infos["rights"].append(wire.StartConnected) etisserant@0: if wire.StartConnected not in right_elements: etisserant@0: right_elements.append(wire.StartConnected) etisserant@0: right_index.append(wire.StartConnected.GetWireIndex(wire)) etisserant@0: else: etisserant@0: index = right_elements.index(wire.StartConnected) etisserant@0: right_index[index] = max(right_index[index], wire.StartConnected.GetWireIndex(wire)) etisserant@0: for connector in block_infos["inputs"]: etisserant@0: for infos in blocks_infos: etisserant@0: if connector in infos["rights"]: etisserant@0: infos["rights"].remove(connector) etisserant@0: if connector in right_elements: etisserant@0: index = right_elements.index(connector) etisserant@0: right_elements.pop(index) etisserant@0: right_index.pop(index) etisserant@0: for right_element in block_infos["rights"]: etisserant@0: if right_element not in infos["rights"]: etisserant@0: infos["rights"].append(right_element) etisserant@0: for connector in block_infos["outputs"]: etisserant@0: for infos in blocks_infos: etisserant@0: if connector in infos["lefts"]: etisserant@0: infos["lefts"].remove(connector) etisserant@0: if connector in left_elements: etisserant@0: index = left_elements.index(connector) etisserant@0: left_elements.pop(index) etisserant@0: left_index.pop(index) etisserant@0: for left_element in block_infos["lefts"]: etisserant@0: if left_element not in infos["lefts"]: etisserant@0: infos["lefts"].append(left_element) etisserant@0: blocks_infos.append(block_infos) etisserant@0: for infos in blocks_infos: etisserant@0: left_elements = [element for element in infos["lefts"]] etisserant@0: for left_element in left_elements: etisserant@0: if isinstance(left_element.GetParentBlock(), LD_PowerRail): etisserant@0: infos["lefts"].remove(left_element) etisserant@0: if "LD_PowerRail" not in infos["lefts"]: etisserant@0: infos["lefts"].append("LD_PowerRail") etisserant@0: right_elements = [element for element in infos["rights"]] etisserant@0: for right_element in right_elements: etisserant@0: if isinstance(right_element.GetParentBlock(), LD_PowerRail): 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: if powerrail.IsNullConnector(index + 1): etisserant@0: powerrail.DeleteConnector(index + 1) etisserant@0: powerrail.InsertConnector(index + 1) etisserant@0: powerrail.RefreshModel() etisserant@0: connectors = powerrail.GetConnectors() 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) etisserant@0: connectors[index + 1].Connect((new_wire, -1), False) etisserant@0: new_wire.ConnectStartPoint(None, right_element) etisserant@0: new_wire.ConnectEndPoint(None, connectors[index + 1]) etisserant@0: right_elements.reverse() etisserant@0: elif right_powerrail: lbessard@90: dialog = LDElementDialog(self.ParentWindow, "coil") lbessard@7: varlist = [] lbessard@121: vars = self.Controler.GetEditedElementInterfaceVars(self.TagName) 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@121: returntype = self.Controler.GetEditedElementInterfaceReturnType(self.TagName) 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: if powerrail.IsNullConnector(index + 1): lbessard@7: powerrail.DeleteConnector(index + 1) lbessard@7: powerrail.InsertConnector(index + 1) lbessard@7: powerrail.RefreshModel() lbessard@7: connectors = powerrail.GetConnectors() 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() lbessard@7: # Create Wire between LeftPowerRail and Coil lbessard@7: wire = Wire(self) lbessard@7: connectors[index + 1].Connect((wire, 0), False) lbessard@7: coil_connectors["output"].Connect((wire, -1), False) lbessard@7: wire.ConnectStartPoint(None, connectors[index + 1]) lbessard@7: wire.ConnectEndPoint(None, coil_connectors["output"]) lbessard@42: self.AddWire(wire) lbessard@27: rung.SelectElement(wire) lbessard@27: left_elements.reverse() 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) lbessard@7: coil_connectors["input"].Connect((new_wire, 0), False) lbessard@7: left_element.InsertConnect(left_index[i] + 1, (new_wire, -1), False) lbessard@7: new_wire.ConnectStartPoint(None, coil_connectors["input"]) lbessard@7: new_wire.ConnectEndPoint(None, left_element) 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() etisserant@0: self.Refresh() lbessard@7: else: lbessard@64: 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: lbessard@64: 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): lbessard@64: 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() lbessard@27: input_wires = [wire for wire, handle in connectors["input"].GetWires()] lbessard@27: output_wires = [wire for wire, handle in connectors["output"].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() lbessard@8: input_wires = [wire for wire, handle in connectors["input"].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() lbessard@27: output_wires = [wire for wire, handle in connectors["output"].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() lbessard@27: for connector in powerrail_connectors: 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() lbessard@27: output_connectors = [] lbessard@27: if "outputs" in connectors: lbessard@27: output_connectors = connectors["outputs"] lbessard@27: if "output" in connectors: lbessard@27: output_connectors = [connectors["output"]] lbessard@27: for connector in output_connectors: 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): etisserant@0: if isinstance(element, LD_PowerRail) and element.GetType() == LEFTRAIL: etisserant@0: element.RefreshModel() etisserant@0: return etisserant@0: connectors = element.GetConnectors() etisserant@0: input_connectors = [] etisserant@0: output_connectors = [] etisserant@0: if isinstance(element, LD_PowerRail) and element.GetType() == RIGHTRAIL: etisserant@0: input_connectors = connectors lbessard@8: for i, connector in enumerate(input_connectors): lbessard@8: for j, (wire, handle) in enumerate(connector.GetWires()): lbessard@8: block = wire.EndConnected.GetParentBlock() lbessard@8: self.RefreshPosition(block, False) etisserant@0: else: etisserant@0: if "inputs" in connectors: etisserant@0: input_connectors = connectors["inputs"] etisserant@0: if "outputs" in connectors: etisserant@0: output_connectors = connectors["outputs"] etisserant@0: if "input" in connectors: etisserant@0: input_connectors = [connectors["input"]] etisserant@0: if "output" in connectors: etisserant@0: output_connectors = [connectors["output"]] etisserant@0: position = element.GetPosition() etisserant@0: minx = 0 etisserant@0: onlyone = [] etisserant@0: for connector in input_connectors: etisserant@0: onlyone.append(len(connector.GetWires()) == 1) etisserant@0: for wire, handle in connector.GetWires(): etisserant@0: onlyone[-1] &= len(wire.EndConnected.GetWires()) == 1 etisserant@0: leftblock = wire.EndConnected.GetParentBlock() etisserant@0: pos = leftblock.GetPosition() etisserant@0: size = leftblock.GetSize() etisserant@0: minx = max(minx, pos[0] + size[0]) etisserant@0: if isinstance(element, LD_Coil): etisserant@0: interval = LD_WIRECOIL_SIZE etisserant@0: else: etisserant@0: interval = LD_WIRE_SIZE etisserant@0: if False in onlyone: etisserant@0: interval += LD_WIRE_SIZE etisserant@0: movex = minx + interval - position[0] etisserant@0: element.Move(movex, 0) lbessard@8: position = element.GetPosition() lbessard@27: blocks = [] lbessard@27: for i, connector in enumerate(input_connectors): lbessard@27: for j, (wire, handle) in enumerate(connector.GetWires()): lbessard@27: blocks.append(wire.EndConnected.GetParentBlock()) etisserant@0: for i, connector in enumerate(input_connectors): etisserant@0: startpoint = connector.GetPosition(False) etisserant@0: previous_blocks = [] etisserant@0: block_list = [] etisserant@0: start_offset = 0 etisserant@0: if not onlyone[i]: etisserant@0: middlepoint = minx + LD_WIRE_SIZE etisserant@0: for j, (wire, handle) in enumerate(connector.GetWires()): etisserant@0: block = wire.EndConnected.GetParentBlock() 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): lbessard@8: diff = (startpoint.y - endpoint.y) / LD_LINE_SIZE lbessard@8: for k in xrange(abs(diff)): lbessard@8: if diff < 0: lbessard@8: element.DeleteConnector(i - 1 - k) lbessard@8: else: lbessard@8: element.InsertConnector(i + k, False) lbessard@8: elif isinstance(block, LD_PowerRail): etisserant@0: index = block.GetConnectorIndex(wire.EndConnected) etisserant@0: if index: etisserant@0: diff = (startpoint.y - endpoint.y) / LD_LINE_SIZE etisserant@0: for k in xrange(abs(diff)): etisserant@0: if diff < 0: etisserant@0: block.DeleteConnector(index - 1 - k) etisserant@0: else: etisserant@0: block.InsertConnector(index + k, False) etisserant@0: else: etisserant@0: block.Move(0, startpoint.y + offset - endpoint.y) 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] etisserant@0: wire.SetPoints(points) etisserant@0: previous_blocks.append(block) lbessard@27: blocks.remove(block) etisserant@0: ExtractNextBlocks(block, block_list) etisserant@0: element.RefreshModel(False) lbessard@8: if recursive: lbessard@8: for connector in output_connectors: 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: etisserant@0: def EditContactContent(self, contact): lbessard@28: if self.GetDrawingMode() == FREEDRAWING_MODE: lbessard@28: Viewer.EditContactContent(self, contact) lbessard@28: else: lbessard@90: dialog = LDElementDialog(self.ParentWindow, "contact") lbessard@28: varlist = [] lbessard@121: vars = self.Controler.GetEditedElementInterfaceVars(self.TagName) lbessard@28: if vars: lbessard@28: for var in vars: lbessard@28: if var["Class"] != "Output" and var["Type"] == "BOOL": lbessard@28: varlist.append(var["Name"]) lbessard@28: dialog.SetVariables(varlist) lbessard@28: dialog.SetValues({"name":contact.GetName(),"type":contact.GetType()}) lbessard@64: if dialog.ShowModal() == wx.ID_OK: lbessard@28: values = dialog.GetValues() lbessard@28: contact.SetName(values["name"]) lbessard@28: contact.SetType(values["type"]) lbessard@28: contact.RefreshModel(False) lbessard@56: self.RefreshBuffer() lbessard@56: self.RefreshScrollBars() lbessard@144: contact.Refresh() lbessard@28: dialog.Destroy() etisserant@0: etisserant@0: def EditCoilContent(self, coil): lbessard@28: if self.GetDrawingMode() == FREEDRAWING_MODE: lbessard@28: Viewer.EditCoilContent(self, coil) lbessard@28: else: lbessard@90: dialog = LDElementDialog(self.ParentWindow, "coil") lbessard@28: varlist = [] lbessard@121: vars = self.Controler.GetEditedElementInterfaceVars(self.TagName) lbessard@28: if vars: lbessard@28: for var in vars: lbessard@28: if var["Class"] != "Input" and var["Type"] == "BOOL": lbessard@28: varlist.append(var["Name"]) lbessard@121: returntype = self.Controler.GetEditedElementInterfaceReturnType(self.TagName) lbessard@28: if returntype == "BOOL": lbessard@121: varlist.append(self.Controler.GetEditedElementName(self.TagName)) lbessard@28: dialog.SetVariables(varlist) lbessard@28: dialog.SetValues({"name":coil.GetName(),"type":coil.GetType()}) lbessard@64: if dialog.ShowModal() == wx.ID_OK: lbessard@28: values = dialog.GetValues() lbessard@28: coil.SetName(values["name"]) lbessard@28: coil.SetType(values["type"]) lbessard@28: coil.RefreshModel(False) lbessard@56: self.RefreshBuffer() lbessard@56: self.RefreshScrollBars() lbessard@144: coil.Refresh() lbessard@28: dialog.Destroy() lbessard@28: lbessard@28: def EditPowerRailContent(self, powerrail): lbessard@28: if self.GetDrawingMode() == FREEDRAWING_MODE: lbessard@28: Viewer.EditPowerRailContent(self, powerrail) lbessard@28: lbessard@122: #------------------------------------------------------------------------------- lbessard@122: # Model update functions lbessard@122: #------------------------------------------------------------------------------- lbessard@122: lbessard@122: def RefreshBlockModel(self, block): lbessard@122: blockid = block.GetId() lbessard@122: infos = {} lbessard@122: infos["type"] = block.GetType() lbessard@122: infos["name"] = block.GetName() lbessard@122: infos["x"], infos["y"] = block.GetPosition() lbessard@122: infos["width"], infos["height"] = block.GetSize() lbessard@122: infos["connectors"] = block.GetConnectors() lbessard@122: self.Controler.SetEditedElementBlockInfos(self.TagName, blockid, infos)