Laurent@814: #!/usr/bin/env python
Laurent@814: # -*- coding: utf-8 -*-
Laurent@814: 
Laurent@814: #This file is part of PLCOpenEditor, a library implementing an IEC 61131-3 editor
Laurent@814: #based on the plcopen standard.
Laurent@814: #
Laurent@814: #Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD
Laurent@814: #
Laurent@814: #See COPYING file for copyrights details.
Laurent@814: #
Laurent@814: #This library is free software; you can redistribute it and/or
Laurent@814: #modify it under the terms of the GNU General Public
Laurent@814: #License as published by the Free Software Foundation; either
Laurent@814: #version 2.1 of the License, or (at your option) any later version.
Laurent@814: #
Laurent@814: #This library is distributed in the hope that it will be useful,
Laurent@814: #but WITHOUT ANY WARRANTY; without even the implied warranty of
Laurent@814: #MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
Laurent@814: #General Public License for more details.
Laurent@814: #
Laurent@814: #You should have received a copy of the GNU General Public
Laurent@814: #License along with this library; if not, write to the Free Software
Laurent@814: #Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
Laurent@814: 
Laurent@814: from types import *
Laurent@814: 
Laurent@814: import wx
Laurent@814: 
Laurent@814: from Viewer import *
Laurent@814: 
Laurent@814: class SFC_Viewer(Viewer):
Laurent@814:     
Laurent@814:     def __init__(self, parent, tagname, window, controler, debug = False, instancepath = ""):
Laurent@814:         Viewer.__init__(self, parent, tagname, window, controler, debug, instancepath)
Laurent@814:         self.CurrentLanguage = "SFC"
Laurent@814:     
Laurent@814:     def ConnectConnectors(self, start, end):
Laurent@814:         startpoint = [start.GetPosition(False), start.GetDirection()]
Laurent@814:         endpoint = [end.GetPosition(False), end.GetDirection()]
Laurent@814:         wire = Wire(self, startpoint, endpoint)
Laurent@814:         self.AddWire(wire)
Laurent@814:         start.Connect((wire, 0), False)
Laurent@814:         end.Connect((wire, -1), False)
Laurent@814:         wire.ConnectStartPoint(None, start)
Laurent@814:         wire.ConnectEndPoint(None, end)
Laurent@814:         return wire
Laurent@814:     
Laurent@814:     def CreateTransition(self, connector, next = None):
Laurent@814:         previous = connector.GetParentBlock()
Laurent@814:         id = self.GetNewId()
Laurent@814:         transition = SFC_Transition(self, "reference", "", 0, id)
Laurent@814:         pos = connector.GetPosition(False)
Laurent@814:         transition.SetPosition(pos.x, pos.y + SFC_WIRE_MIN_SIZE)
Laurent@814:         transition_connectors = transition.GetConnectors()
Laurent@814:         wire = self.ConnectConnectors(transition_connectors["input"], connector)
Laurent@814:         if isinstance(previous, SFC_Divergence):
Laurent@814:             previous.RefreshConnectedPosition(connector)
Laurent@814:         else:
Laurent@814:             previous.RefreshOutputPosition()
Laurent@814:         wire.SetPoints([wx.Point(pos.x, pos.y + GetWireSize(previous)), wx.Point(pos.x, pos.y)])
Laurent@814:         self.AddBlock(transition)
Laurent@814:         self.Controler.AddEditedElementTransition(self.TagName, id)
Laurent@814:         self.RefreshTransitionModel(transition)
Laurent@814:         if next:
Laurent@814:             wire = self.ConnectConnectors(next, transition_connectors["output"])
Laurent@814:             pos = transition_connectors["output"].GetPosition(False)
Laurent@814:             next_block = next.GetParentBlock()
Laurent@814:             next_pos = next.GetPosition(False)
Laurent@814:             transition.RefreshOutputPosition((0, pos.y + SFC_WIRE_MIN_SIZE - next_pos.y))
Laurent@814:             wire.SetPoints([wx.Point(pos.x, pos.y + SFC_WIRE_MIN_SIZE), wx.Point(pos.x, pos.y)])
Laurent@814:             if isinstance(next_block, SFC_Divergence):
Laurent@814:                 next_block.RefreshPosition()
Laurent@814:             transition.RefreshOutputModel(True)
Laurent@814:         return transition
Laurent@814:     
Laurent@814:     def RemoveTransition(self, transition):
Laurent@814:         connectors = transition.GetConnectors()
Laurent@814:         input_wires = connectors["input"].GetWires()
Laurent@814:         if len(input_wires) != 1:
Laurent@814:             return
Laurent@814:         input_wire = input_wires[0][0]
Laurent@814:         previous = input_wire.EndConnected
Laurent@814:         input_wire.Clean()
Laurent@814:         self.RemoveWire(input_wire)
Laurent@814:         output_wires = connectors["output"].GetWires()
Laurent@814:         if len(output_wires) != 1:
Laurent@814:             return
Laurent@814:         output_wire = output_wires[0][0]
Laurent@814:         next = output_wire.StartConnected
Laurent@814:         output_wire.Clean()
Laurent@814:         self.RemoveWire(output_wire)
Laurent@814:         transition.Clean()
Laurent@814:         self.RemoveBlock(transition)
Laurent@814:         self.Controler.RemoveEditedElementInstance(self.TagName, transition.GetId())
Laurent@814:         wire = self.ConnectConnectors(next, previous)
Laurent@814:         return wire
Laurent@814:     
Laurent@814:     def CreateStep(self, name, connector, next = None):
Laurent@814:         previous = connector.GetParentBlock()
Laurent@814:         id = self.GetNewId()
Laurent@814:         step = SFC_Step(self, name, False, id)
Laurent@814:         if next:
Laurent@814:             step.AddOutput()
Laurent@814:         min_width, min_height = step.GetMinSize()
Laurent@814:         pos = connector.GetPosition(False)
Laurent@814:         step.SetPosition(pos.x, pos.y + SFC_WIRE_MIN_SIZE)
Laurent@814:         step.SetSize(min_width, min_height)
Laurent@814:         step_connectors = step.GetConnectors()
Laurent@814:         wire = self.ConnectConnectors(step_connectors["input"], connector)
Laurent@814:         if isinstance(previous, SFC_Divergence):
Laurent@814:             previous.RefreshConnectedPosition(connector)
Laurent@814:         else:
Laurent@814:             previous.RefreshOutputPosition()
Laurent@814:         wire.SetPoints([wx.Point(pos.x, pos.y + GetWireSize(previous)), wx.Point(pos.x, pos.y)])
Laurent@814:         self.AddBlock(step)
Laurent@814:         self.Controler.AddEditedElementStep(self.TagName, id)
Laurent@814:         self.RefreshStepModel(step)
Laurent@814:         if next:
Laurent@814:             wire = self.ConnectConnectors(next, step_connectors["output"])
Laurent@814:             pos = step_connectors["output"].GetPosition(False)
Laurent@814:             next_block = next.GetParentBlock()
Laurent@814:             next_pos = next.GetPosition(False)
Laurent@814:             step.RefreshOutputPosition((0, pos.y + SFC_WIRE_MIN_SIZE - next_pos.y))
Laurent@814:             wire.SetPoints([wx.Point(pos.x, pos.y + SFC_WIRE_MIN_SIZE), wx.Point(pos.x, pos.y)])
Laurent@814:             if isinstance(next_block, SFC_Divergence):
Laurent@814:                 next_block.RefreshPosition()
Laurent@814:             step.RefreshOutputModel(True)
Laurent@814:         return step
Laurent@814: 
Laurent@814:     def RemoveStep(self, step):
Laurent@814:         connectors = step.GetConnectors()
Laurent@814:         if connectors["input"]:
Laurent@814:             input_wires = connectors["input"].GetWires()
Laurent@814:             if len(input_wires) != 1:
Laurent@814:                 return
Laurent@814:             input_wire = input_wires[0][0]
Laurent@814:             previous = input_wire.EndConnected
Laurent@814:             input_wire.Clean()
Laurent@814:             self.RemoveWire(input_wire)
Laurent@814:         else:
Laurent@814:             previous = None
Laurent@814:         if connectors["output"]:
Laurent@814:             output_wires = connectors["output"].GetWires()
Laurent@814:             if len(output_wires) != 1:
Laurent@814:                 return
Laurent@814:             output_wire = output_wires[0][0]
Laurent@814:             next = output_wire.StartConnected
Laurent@814:             output_wire.Clean()
Laurent@814:             self.RemoveWire(output_wire)
Laurent@814:         else:
Laurent@814:             next = None
Laurent@814:         action = step.GetActionConnected()
Laurent@814:         if action:
Laurent@814:             self.DeleteActionBlock(action.GetParentBlock())
Laurent@814:         step.Clean()
Laurent@814:         self.RemoveBlock(step)
Laurent@814:         self.Controler.RemoveEditedElementInstance(self.TagName, step.GetId())
Laurent@814:         if next and previous:
Laurent@814:             wire = self.ConnectConnectors(next, previous)
Laurent@814:             return wire
Laurent@814:         else:
Laurent@814:             return None
Laurent@814: 
Laurent@814: #-------------------------------------------------------------------------------
Laurent@814: #                          Mouse event functions
Laurent@814: #-------------------------------------------------------------------------------
Laurent@814: 
Laurent@814:     def OnViewerLeftDown(self, event):
Laurent@814:         if self.GetDrawingMode() == FREEDRAWING_MODE:
Laurent@814:             Viewer.OnViewerLeftDown(self, event)
Laurent@814:         elif self.Mode == MODE_SELECTION:
Laurent@814:             if event.ShiftDown() and not event.ControlDown() and self.SelectedElement is not None:
Laurent@814:                 element = self.FindElement(event, True)
Laurent@814:                 if element and not self.IsWire(element):
Laurent@814:                     if isinstance(self.SelectedElement, Graphic_Group):
Laurent@814:                         self.SelectedElement.SelectElement(element)
Laurent@814:                     else:
Laurent@814:                         group = Graphic_Group(self)
Laurent@814:                         self.SelectedElement.SetSelected(False)
Laurent@814:                         group.SelectElement(self.SelectedElement)
Laurent@814:                         group.SelectElement(element)
Laurent@814:                         self.SelectedElement = group
Laurent@814:                     elements = self.SelectedElement.GetElements()
Laurent@814:                     if len(elements) == 0:
Laurent@814:                         self.SelectedElement = element
Laurent@814:                     elif len(elements) == 1:
Laurent@814:                         self.SelectedElement = elements[0]
Laurent@814:                     self.SelectedElement.SetSelected(True)
Laurent@814:             else:
Laurent@814:                 element = self.FindElement(event)
Laurent@814:                 if self.SelectedElement and self.SelectedElement != element:
Laurent@814:                     if self.IsWire(self.SelectedElement):
Laurent@814:                         self.SelectedElement.SetSelectedSegment(None)
Laurent@814:                     else:
Laurent@814:                         self.SelectedElement.SetSelected(False)
Laurent@814:                     self.SelectedElement = None
Laurent@814:                 if element:
Laurent@814:                     self.SelectedElement = element
Laurent@814:                     self.SelectedElement.OnLeftDown(event, self.GetLogicalDC(), self.Scaling)
Laurent@814:                     self.SelectedElement.Refresh()
Laurent@814:                 else:
Laurent@814:                     self.rubberBand.Reset()
Laurent@814:                     self.rubberBand.OnLeftDown(event, self.GetLogicalDC(), self.Scaling)
Laurent@814:         elif self.Mode == MODE_COMMENT:
Laurent@814:             self.rubberBand.Reset()
Laurent@814:             self.rubberBand.OnLeftDown(event, self.GetLogicalDC(), self.Scaling)
Laurent@814:         event.Skip()
Laurent@814: 
Laurent@814:     def OnViewerLeftUp(self, event):
Laurent@814:         if self.GetDrawingMode() == FREEDRAWING_MODE:
Laurent@814:             Viewer.OnViewerLeftUp(self, event)
Laurent@814:         elif self.rubberBand.IsShown():
Laurent@814:             if self.Mode == MODE_SELECTION:
Laurent@814:                 elements = self.SearchElements(self.rubberBand.GetCurrentExtent())
Laurent@814:                 self.rubberBand.OnLeftUp(event, self.GetLogicalDC(), self.Scaling)
Laurent@814:                 if len(elements) > 0:
Laurent@814:                     self.SelectedElement = Graphic_Group(self)
Laurent@814:                     self.SelectedElement.SetElements(elements)
Laurent@814:                     self.SelectedElement.SetSelected(True)
Laurent@814:             elif self.Mode == MODE_COMMENT:
Laurent@814:                 bbox = self.rubberBand.GetCurrentExtent()
Laurent@814:                 self.rubberBand.OnLeftUp(event, self.GetLogicalDC(), self.Scaling)
Laurent@814:                 wx.CallAfter(self.AddComment, bbox)
Laurent@814:         elif self.Mode == MODE_INITIALSTEP:
Laurent@814:             wx.CallAfter(self.AddInitialStep, GetScaledEventPosition(event, self.GetLogicalDC(), self.Scaling))
Laurent@814:         elif self.Mode == MODE_SELECTION and self.SelectedElement:
Laurent@814:             if self.IsWire(self.SelectedElement):
Laurent@814:                 self.SelectedElement.SetSelectedSegment(0)
Laurent@814:             else:
Laurent@814:                 self.SelectedElement.OnLeftUp(event, self.GetLogicalDC(), self.Scaling)
Laurent@814:                 self.SelectedElement.Refresh()
Laurent@814:             wx.CallAfter(self.SetCurrentCursor, 0)
Laurent@814:         elif self.Mode == MODE_WIRE and self.SelectedElement:
Laurent@814:             self.SelectedElement.ResetPoints()
Laurent@814:             self.SelectedElement.OnMotion(event, self.GetLogicalDC(), self.Scaling)
Laurent@814:             self.SelectedElement.GeneratePoints()
Laurent@814:             self.SelectedElement.RefreshModel()
Laurent@814:             self.SelectedElement.SetSelected(True)
Laurent@814:         event.Skip()
Laurent@814:     
Laurent@814:     def OnViewerRightUp(self, event):
Laurent@814:         if self.GetDrawingMode() == FREEDRAWING_MODE:
Laurent@814:             Viewer.OnViewerRightUp(self, event)
Laurent@814:         else:
Laurent@814:             element = self.FindElement(event)
Laurent@814:             if element:
Laurent@814:                 if self.SelectedElement and self.SelectedElement != element:
Laurent@814:                     self.SelectedElement.SetSelected(False)
Laurent@814:                 self.SelectedElement = element
Laurent@814:                 if self.IsWire(self.SelectedElement):
Laurent@814:                     self.SelectedElement.SetSelectedSegment(0)
Laurent@814:                 else:
Laurent@814:                     self.SelectedElement.SetSelected(True)
Laurent@814:                     self.SelectedElement.OnRightUp(event, self.GetLogicalDC(), self.Scaling)
Laurent@814:                     self.SelectedElement.Refresh()
Laurent@814:                 wx.CallAfter(self.SetCurrentCursor, 0)
Laurent@814:         event.Skip()
Laurent@814:     
Laurent@814:     def OnViewerLeftDClick(self, event):
Laurent@814:         if self.GetDrawingMode() == FREEDRAWING_MODE:
Laurent@814:             Viewer.OnViewerLeftDClick(self, event)
Laurent@814:         elif self.Mode == MODE_SELECTION and self.SelectedElement:
Laurent@814:             self.SelectedElement.OnLeftDClick(event, self.GetLogicalDC(), self.Scaling)
Laurent@814:             self.Refresh(False)
Laurent@814:         event.Skip()
Laurent@814:     
Laurent@814:     def OnViewerMotion(self, event):
Laurent@814:         if self.GetDrawingMode() == FREEDRAWING_MODE:
Laurent@814:             Viewer.OnViewerMotion(self, event)
Laurent@814:         else:
Laurent@814:             if self.rubberBand.IsShown():
Laurent@814:                 self.rubberBand.OnMotion(event, self.GetLogicalDC(), self.Scaling)
Laurent@814:             elif self.Mode == MODE_SELECTION and self.SelectedElement:
Laurent@814:                 if not self.IsWire(self.SelectedElement) and not isinstance(self.SelectedElement, Graphic_Group):
Laurent@814:                     self.SelectedElement.OnMotion(event, self.GetLogicalDC(), self.Scaling)
Laurent@814:                     self.SelectedElement.Refresh()
Laurent@814:             elif self.Mode == MODE_WIRE and self.SelectedElement:
Laurent@814:                 self.SelectedElement.ResetPoints()
Laurent@814:                 self.SelectedElement.OnMotion(event, self.GetLogicalDC(), self.Scaling)
Laurent@814:                 self.SelectedElement.GeneratePoints()
Laurent@814:                 self.SelectedElement.Refresh()
Laurent@814:             self.UpdateScrollPos(event)
Laurent@814:         event.Skip()
Laurent@814: 
Laurent@814: #-------------------------------------------------------------------------------
Laurent@814: #                          Keyboard event functions
Laurent@814: #-------------------------------------------------------------------------------
Laurent@814: 
Laurent@814:     def OnChar(self, event):
Laurent@814:         if self.GetDrawingMode() == FREEDRAWING_MODE:
Laurent@814:             Viewer.OnChar(self, event)
Laurent@814:         else:
Laurent@814:             xpos, ypos = self.GetScrollPos(wx.HORIZONTAL), self.GetScrollPos(wx.VERTICAL)
Laurent@814:             xmax = self.GetScrollRange(wx.HORIZONTAL) - self.GetScrollThumb(wx.HORIZONTAL)
Laurent@814:             ymax = self.GetScrollRange(wx.VERTICAL) - self.GetScrollThumb(wx.VERTICAL)
Laurent@814:             keycode = event.GetKeyCode()
Laurent@814:             if self.Scaling:
Laurent@814:                 scaling = self.Scaling
Laurent@814:             else:
Laurent@814:                 scaling = (8, 8)
Laurent@814:             if keycode == wx.WXK_DELETE and self.SelectedElement:
Laurent@814:                 self.SelectedElement.Delete()
Laurent@814:                 self.SelectedElement = None
Laurent@814:                 self.RefreshBuffer()
Laurent@814:                 self.RefreshScrollBars()
Laurent@814:                 self.Refresh(False)
Laurent@814:             elif keycode == wx.WXK_LEFT:
Laurent@814:                 if event.ControlDown() and event.ShiftDown():
Laurent@814:                     self.Scroll(0, ypos)
Laurent@814:                 elif event.ControlDown():
Laurent@814:                     event.Skip()
Laurent@814:                 elif self.SelectedElement:
Laurent@814:                     self.SelectedElement.Move(-scaling[0], 0)
Laurent@814:                     self.SelectedElement.RefreshModel()
Laurent@814:                     self.RefreshBuffer()
Laurent@814:                     self.RefreshScrollBars()
Laurent@814:                     self.RefreshRect(self.GetScrolledRect(self.SelectedElement.GetRedrawRect(-scaling[0], 0)), False)
Laurent@814:             elif keycode == wx.WXK_RIGHT:
Laurent@814:                 if event.ControlDown() and event.ShiftDown():
Laurent@814:                     self.Scroll(xmax, ypos)
Laurent@814:                 elif event.ControlDown():
Laurent@814:                     event.Skip()
Laurent@814:                 elif self.SelectedElement:
Laurent@814:                     self.SelectedElement.Move(scaling[0], 0)
Laurent@814:                     self.SelectedElement.RefreshModel()
Laurent@814:                     self.RefreshBuffer()
Laurent@814:                     self.RefreshScrollBars()
Laurent@814:                     self.RefreshRect(self.GetScrolledRect(self.SelectedElement.GetRedrawRect(scaling[0], 0)), False)
Laurent@814:             elif keycode == wx.WXK_UP:
Laurent@814:                 if event.ControlDown() and event.ShiftDown():
Laurent@814:                     self.Scroll(xpos, 0)
Laurent@814:                 elif event.ControlDown():
Laurent@814:                     event.Skip()
Laurent@814:                 elif self.SelectedElement:
Laurent@814:                     self.SelectedElement.Move(0, -scaling[1])
Laurent@814:                     self.SelectedElement.RefreshModel()
Laurent@814:                     self.RefreshBuffer()
Laurent@814:                     self.RefreshScrollBars()
Laurent@814:                     self.RefreshRect(self.GetScrolledRect(self.SelectedElement.GetRedrawRect(0, -scaling[1])), False)
Laurent@814:             elif keycode == wx.WXK_DOWN:
Laurent@814:                 if event.ControlDown() and event.ShiftDown():
Laurent@814:                     self.Scroll(xpos, ymax)
Laurent@814:                 elif event.ControlDown():
Laurent@814:                     event.Skip()
Laurent@814:                 elif self.SelectedElement:
Laurent@814:                     self.SelectedElement.Move(0, scaling[1])
Laurent@814:                     self.SelectedElement.RefreshModel()
Laurent@814:                     self.RefreshBuffer()
Laurent@814:                     self.RefreshScrollBars()
Laurent@814:                     self.RefreshRect(self.GetScrolledRect(self.SelectedElement.GetRedrawRect(0, scaling[1])), False)
Laurent@814:             else:
Laurent@814:                 event.Skip()
Laurent@814:         
Laurent@814: #-------------------------------------------------------------------------------
Laurent@814: #                          Adding element functions
Laurent@814: #-------------------------------------------------------------------------------
Laurent@814: 
Laurent@814:     def AddInitialStep(self, pos):
Laurent@814:         dialog = SFCStepNameDialog(self.ParentWindow, _("Please enter step name"), _("Add a new initial step"), "", wx.OK|wx.CANCEL)
Laurent@814:         dialog.SetPouNames(self.Controler.GetProjectPouNames(self.Debug))
Laurent@1347:         dialog.SetVariables(self.Controler.GetEditedElementInterfaceVars(self.TagName, debug=self.Debug))
Laurent@814:         dialog.SetStepNames([block.GetName() for block in self.Blocks if isinstance(block, SFC_Step)])
Laurent@814:         if dialog.ShowModal() == wx.ID_OK:
Laurent@814:             id = self.GetNewId()
Laurent@814:             name = dialog.GetValue()
Laurent@814:             step = SFC_Step(self, name, True, id)
Laurent@814:             min_width, min_height = step.GetMinSize()
Laurent@814:             step.SetPosition(pos.x, pos.y)
Laurent@814:             width, height = step.GetSize()
Laurent@814:             step.SetSize(max(min_width, width), max(min_height, height))
Laurent@814:             self.AddBlock(step)
Laurent@814:             self.Controler.AddEditedElementStep(self.TagName, id)
Laurent@814:             self.RefreshStepModel(step)
Laurent@814:             self.RefreshBuffer()
Laurent@814:             self.RefreshScrollBars()
Laurent@814:             self.Refresh(False)
Laurent@814:         dialog.Destroy()
Laurent@814: 
Laurent@814:     def AddStep(self):
Laurent@814:         if self.SelectedElement in self.Wires or isinstance(self.SelectedElement, SFC_Step):
Laurent@814:             dialog = SFCStepNameDialog(self.ParentWindow, _("Add a new step"), _("Please enter step name"), "", wx.OK|wx.CANCEL)
Laurent@814:             dialog.SetPouNames(self.Controler.GetProjectPouNames(self.Debug))
Laurent@1347:             dialog.SetVariables(self.Controler.GetEditedElementInterfaceVars(self.TagName, debug=self.Debug))
Laurent@814:             dialog.SetStepNames([block.GetName() for block in self.Blocks if isinstance(block, SFC_Step)])
Laurent@814:             if dialog.ShowModal() == wx.ID_OK:
Laurent@814:                 name = dialog.GetValue()
Laurent@814:                 if self.IsWire(self.SelectedElement):
Laurent@814:                     self.SelectedElement.SetSelectedSegment(None)
Laurent@814:                     previous = self.SelectedElement.EndConnected
Laurent@814:                     next = self.SelectedElement.StartConnected
Laurent@814:                     self.SelectedElement.Clean()
Laurent@814:                     self.RemoveWire(self.SelectedElement)
Laurent@814:                 else:
Laurent@814:                     connectors = self.SelectedElement.GetConnectors()
Laurent@814:                     if connectors["output"]:
Laurent@814:                         previous = connectors["output"]
Laurent@814:                         wires = previous.GetWires()
Laurent@814:                         if len(wires) != 1:
Laurent@814:                             return
Laurent@814:                         wire = wires[0][0]
Laurent@814:                         next = wire.StartConnected
Laurent@814:                         wire.Clean()
Laurent@814:                         self.RemoveWire(wire)
Laurent@814:                     else:
Laurent@814:                         self.SelectedElement.AddOutput()
Laurent@814:                         connectors = self.SelectedElement.GetConnectors()
Laurent@814:                         self.RefreshStepModel(self.SelectedElement)
Laurent@814:                         previous = connectors["output"]
Laurent@814:                         next = None
Laurent@814:                 previous_block = previous.GetParentBlock()
Laurent@814:                 if isinstance(previous_block, SFC_Step) or isinstance(previous_block, SFC_Divergence) and previous_block.GetType() in [SELECTION_DIVERGENCE, SELECTION_CONVERGENCE]:
Laurent@814:                     transition = self.CreateTransition(previous)
Laurent@814:                     transition_connectors = transition.GetConnectors()
Laurent@814:                     step = self.CreateStep(name, transition_connectors["output"], next)
Laurent@814:                 else:
Laurent@814:                     step = self.CreateStep(name, previous)
Laurent@814:                     step.AddOutput()
Laurent@814:                     step.RefreshModel()
Laurent@814:                     step_connectors = step.GetConnectors()
Laurent@814:                     transition = self.CreateTransition(step_connectors["output"], next)
Laurent@814:                 if self.IsWire(self.SelectedElement):
Laurent@814:                     self.SelectedElement = wire
Laurent@814:                     self.SelectedElement.SetSelectedSegment(0)
Laurent@814:                 else:
Laurent@814:                     self.SelectedElement.SetSelected(False)
Laurent@814:                     self.SelectedElement = step
Laurent@814:                     self.SelectedElement.SetSelected(True)
Laurent@814:                 self.RefreshBuffer()
Laurent@814:                 self.RefreshScrollBars()
Laurent@814:                 self.Refresh(False)
Laurent@814:             dialog.Destroy()
Laurent@814:     
Laurent@814:     def AddStepAction(self):
Laurent@814:         if isinstance(self.SelectedElement, SFC_Step):
Laurent@814:             connectors = self.SelectedElement.GetConnectors()
Laurent@814:             if not connectors["action"]:
Laurent@814:                 dialog = ActionBlockDialog(self.ParentWindow)
Laurent@814:                 dialog.SetQualifierList(self.Controler.GetQualifierTypes())
Laurent@814:                 dialog.SetActionList(self.Controler.GetEditedElementActions(self.TagName, self.Debug))
Laurent@1347:                 dialog.SetVariableList(self.Controler.GetEditedElementInterfaceVars(self.TagName, debug=self.Debug))
Laurent@814:                 if dialog.ShowModal() == wx.ID_OK:
Laurent@814:                     actions = dialog.GetValues()
Laurent@814:                     self.SelectedElement.AddAction()
Laurent@814:                     self.RefreshStepModel(self.SelectedElement)
Laurent@814:                     connectors = self.SelectedElement.GetConnectors()
Laurent@814:                     pos = connectors["action"].GetPosition(False)
Laurent@814:                     id = self.GetNewId()
Laurent@814:                     actionblock = SFC_ActionBlock(self, [], id)
Laurent@814:                     actionblock.SetPosition(pos.x + SFC_WIRE_MIN_SIZE, pos.y - SFC_STEP_DEFAULT_SIZE[1] / 2)
Laurent@814:                     actionblock_connector = actionblock.GetConnector()
Laurent@814:                     wire = self.ConnectConnectors(actionblock_connector, connectors["action"])
Laurent@814:                     wire.SetPoints([wx.Point(pos.x + SFC_WIRE_MIN_SIZE, pos.y), wx.Point(pos.x, pos.y)])
Laurent@814:                     actionblock.SetActions(actions)
Laurent@814:                     self.AddBlock(actionblock)
Laurent@814:                     self.Controler.AddEditedElementActionBlock(self.TagName, id)
Laurent@814:                     self.RefreshActionBlockModel(actionblock)
Laurent@814:                     self.RefreshBuffer()
Laurent@814:                     self.RefreshScrollBars()
Laurent@814:                     self.Refresh(False)
Laurent@814:                 dialog.Destroy()
Laurent@814:     
Laurent@814:     def AddDivergence(self):
Laurent@814:         if self.SelectedElement in self.Wires or isinstance(self.SelectedElement, Graphic_Group) or isinstance(self.SelectedElement, SFC_Step):        
Laurent@814:             dialog = SFCDivergenceDialog(self.ParentWindow)
Laurent@814:             dialog.SetPreviewFont(self.GetFont())
Laurent@814:             if dialog.ShowModal() == wx.ID_OK:
Laurent@814:                 value = dialog.GetValues()
Laurent@814:                 if value["type"] == SELECTION_DIVERGENCE:
Laurent@814:                     if self.SelectedElement in self.Wires and isinstance(self.SelectedElement.EndConnected.GetParentBlock(), SFC_Step):
Laurent@814:                         self.SelectedElement.SetSelectedSegment(None)
Laurent@814:                         previous = self.SelectedElement.EndConnected
Laurent@814:                         next = self.SelectedElement.StartConnected
Laurent@814:                         self.SelectedElement.Clean()
Laurent@814:                         self.RemoveWire(self.SelectedElement)
Laurent@814:                         self.SelectedElement = None
Laurent@814:                     elif isinstance(self.SelectedElement, SFC_Step):
Laurent@814:                         connectors = self.SelectedElement.GetConnectors()
Laurent@814:                         if connectors["output"]:
Laurent@814:                             previous = connectors["output"]
Laurent@814:                             wires = previous.GetWires()
Laurent@814:                             if len(wires) != 1:
Laurent@814:                                 return
Laurent@814:                             wire = wires[0][0]
Laurent@814:                             next = wire.StartConnected
Laurent@814:                             wire.Clean()
Laurent@814:                             self.RemoveWire(wire)
Laurent@814:                         else:
Laurent@814:                             self.SelectedElement.AddOutput()
Laurent@814:                             connectors = self.SelectedElement.GetConnectors()
Laurent@814:                             self.RefreshStepModel(self.SelectedElement)
Laurent@814:                             previous = connectors["output"]
Laurent@814:                             next = None
Laurent@814:                     else:
Laurent@814:                         return
Laurent@814:                     id = self.GetNewId()
Laurent@814:                     divergence = SFC_Divergence(self, SELECTION_DIVERGENCE, value["number"], id)
Laurent@814:                     pos = previous.GetPosition(False)
Laurent@814:                     previous_block = previous.GetParentBlock()
Laurent@814:                     wire_size = GetWireSize(previous_block)
Laurent@814:                     divergence.SetPosition(pos.x, pos.y + wire_size)
Laurent@814:                     divergence_connectors = divergence.GetConnectors()
Laurent@814:                     wire = self.ConnectConnectors(divergence_connectors["inputs"][0], previous)
Laurent@814:                     previous_block.RefreshOutputPosition()
Laurent@814:                     wire.SetPoints([wx.Point(pos.x, pos.y + wire_size), wx.Point(pos.x, pos.y)])
Laurent@814:                     self.AddBlock(divergence)
Laurent@814:                     self.Controler.AddEditedElementDivergence(self.TagName, id, value["type"])
Laurent@814:                     self.RefreshDivergenceModel(divergence)
Laurent@814:                     for index, connector in enumerate(divergence_connectors["outputs"]):
Laurent@814:                         if next:
Laurent@814:                             wire = self.ConnectConnectors(next, connector)
Laurent@814:                             pos = connector.GetPosition(False)
Laurent@814:                             next_pos = next.GetPosition(False)
Laurent@814:                             next_block = next.GetParentBlock()
Laurent@814:                             divergence.RefreshOutputPosition((0, pos.y + SFC_WIRE_MIN_SIZE - next_pos.y))
Laurent@814:                             divergence.RefreshConnectedPosition(connector)
Laurent@814:                             wire.SetPoints([wx.Point(pos.x, pos.y + SFC_WIRE_MIN_SIZE), wx.Point(pos.x, pos.y)])
Laurent@814:                             next_block.RefreshModel()
Laurent@814:                             next = None
Laurent@814:                         else:
Laurent@814:                             transition = self.CreateTransition(connector)
Laurent@814:                             transition_connectors = transition.GetConnectors()
Laurent@814:                             step = self.CreateStep("Step", transition_connectors["output"])
Laurent@814:                 elif value["type"] == SIMULTANEOUS_DIVERGENCE:
Laurent@814:                     if self.SelectedElement in self.Wires and isinstance(self.SelectedElement.EndConnected.GetParentBlock(), SFC_Transition):
Laurent@814:                         self.SelectedElement.SetSelectedSegment(None)
Laurent@814:                         previous = self.SelectedElement.EndConnected
Laurent@814:                         next = self.SelectedElement.StartConnected
Laurent@814:                         self.SelectedElement.Clean()
Laurent@814:                         self.RemoveWire(self.SelectedElement)
Laurent@814:                         self.SelectedElement = None
Laurent@814:                     elif isinstance(self.SelectedElement, SFC_Step):
Laurent@814:                         connectors = self.SelectedElement.GetConnectors()
Laurent@814:                         if connectors["output"]:
Laurent@814:                             previous = connectors["output"]
Laurent@814:                             wires = previous.GetWires()
Laurent@814:                             if len(wires) != 1:
Laurent@814:                                 return
Laurent@814:                             wire = wires[0][0]
Laurent@814:                             next = wire.StartConnected
Laurent@814:                             wire.Clean()
Laurent@814:                             self.RemoveWire(wire)
Laurent@814:                         else:
Laurent@814:                             self.SelectedElement.AddOutput()
Laurent@814:                             connectors = self.SelectedElement.GetConnectors()
Laurent@814:                             self.RefreshStepModel(self.SelectedElement)
Laurent@814:                             previous = connectors["output"]
Laurent@814:                             next = None
Laurent@814:                         transition = self.CreateTransition(previous)
Laurent@814:                         transition_connectors = transition.GetConnectors()
Laurent@814:                         previous = transition_connectors["output"]
Laurent@814:                     else:
Laurent@814:                         return
Laurent@814:                     id = self.GetNewId()
Laurent@814:                     divergence = SFC_Divergence(self, SIMULTANEOUS_DIVERGENCE, value["number"], id)
Laurent@814:                     pos = previous.GetPosition(False)
Laurent@814:                     previous_block = previous.GetParentBlock()
Laurent@814:                     wire_size = GetWireSize(previous_block)
Laurent@814:                     divergence.SetPosition(pos.x, pos.y + wire_size)
Laurent@814:                     divergence_connectors = divergence.GetConnectors()
Laurent@814:                     wire = self.ConnectConnectors(divergence_connectors["inputs"][0], previous)
Laurent@814:                     previous_block.RefreshOutputPosition()
Laurent@814:                     wire.SetPoints([wx.Point(pos.x, pos.y + wire_size), wx.Point(pos.x, pos.y)])
Laurent@814:                     self.AddBlock(divergence)
Laurent@814:                     self.Controler.AddEditedElementDivergence(self.TagName, id, value["type"])
Laurent@814:                     self.RefreshDivergenceModel(divergence)
Laurent@814:                     for index, connector in enumerate(divergence_connectors["outputs"]):
Laurent@814:                         if next:
Laurent@814:                             wire = self.ConnectConnectors(next, connector)
Laurent@814:                             pos = connector.GetPosition(False)
Laurent@814:                             next_pos = next.GetPosition(False)
Laurent@814:                             next_block = next.GetParentBlock()
Laurent@814:                             divergence.RefreshOutputPosition((0, pos.y + SFC_WIRE_MIN_SIZE - next_pos.y))
Laurent@814:                             divergence.RefreshConnectedPosition(connector)
Laurent@814:                             wire.SetPoints([wx.Point(pos.x, pos.y + SFC_WIRE_MIN_SIZE), wx.Point(pos.x, pos.y)])
Laurent@814:                             next_block.RefreshModel()
Laurent@814:                             next = None
Laurent@814:                         else:
Laurent@814:                             step = self.CreateStep("Step", connector)
Laurent@814:                 elif isinstance(self.SelectedElement, Graphic_Group) and len(self.SelectedElement.GetElements()) > 1:
Laurent@814:                     next = None
Laurent@814:                     for element in self.SelectedElement.GetElements():
Laurent@814:                         connectors = element.GetConnectors()
Laurent@814:                         if not isinstance(element, SFC_Step) or connectors["output"] and next:
Laurent@814:                             return
Laurent@814:                         elif connectors["output"] and not next:
Laurent@814:                             wires = connectors["output"].GetWires()
Laurent@814:                             if len(wires) != 1:
Laurent@814:                                 return
Laurent@814:                             if value["type"] == SELECTION_CONVERGENCE:
Laurent@814:                                 transition = wires[0][0].StartConnected.GetParentBlock()
Laurent@814:                                 transition_connectors = transition.GetConnectors()
Laurent@814:                                 wires = transition_connectors["output"].GetWires()
Laurent@814:                                 if len(wires) != 1:
Laurent@814:                                     return
Laurent@814:                             wire = wires[0][0]
Laurent@814:                             next = wire.StartConnected
Laurent@814:                             wire.Clean()
Laurent@814:                             self.RemoveWire(wire)
Laurent@814:                     inputs = []
Laurent@814:                     for input in self.SelectedElement.GetElements():
Laurent@814:                         input_connectors = input.GetConnectors()
Laurent@814:                         if not input_connectors["output"]:
Laurent@814:                             input.AddOutput()
Laurent@814:                             input.RefreshModel()
Laurent@814:                             input_connectors = input.GetConnectors()
Laurent@814:                             if value["type"] == SELECTION_CONVERGENCE:
Laurent@814:                                 transition = self.CreateTransition(input_connectors["output"])
Laurent@814:                                 transition_connectors = transition.GetConnectors()
Laurent@814:                                 inputs.append(transition_connectors["output"])
Laurent@814:                             else:
Laurent@814:                                 inputs.append(input_connectors["output"])
Laurent@814:                         elif value["type"] == SELECTION_CONVERGENCE:
Laurent@814:                             wires = input_connectors["output"].GetWires()
Laurent@814:                             transition = wires[0][0].StartConnected.GetParentBlock()
Laurent@814:                             transition_connectors = transition.GetConnectors()
Laurent@814:                             inputs.append(transition_connectors["output"])
Laurent@814:                         else:
Laurent@814:                             inputs.append(input_connectors["output"])
Laurent@814:                     id = self.GetNewId()
Laurent@814:                     divergence = SFC_Divergence(self, value["type"], len(inputs), id)
Laurent@814:                     pos = inputs[0].GetPosition(False)
Laurent@814:                     divergence.SetPosition(pos.x, pos.y + SFC_WIRE_MIN_SIZE)
Laurent@814:                     divergence_connectors = divergence.GetConnectors()
Laurent@814:                     for i, input in enumerate(inputs):
Laurent@814:                         pos = input.GetPosition(False)
Laurent@814:                         wire = self.ConnectConnectors(divergence_connectors["inputs"][i], input)
Laurent@814:                         wire_size = GetWireSize(input)
Laurent@814:                         wire.SetPoints([wx.Point(pos.x, pos.y + wire_size), wx.Point(pos.x, pos.y)])
Laurent@814:                         input_block = input.GetParentBlock()
Laurent@814:                         input_block.RefreshOutputPosition()
Laurent@814:                     divergence.RefreshPosition()
Laurent@814:                     pos = divergence_connectors["outputs"][0].GetRelPosition()
Laurent@814:                     divergence.MoveConnector(divergence_connectors["outputs"][0], - pos.x)
Laurent@814:                     self.AddBlock(divergence)
Laurent@814:                     self.Controler.AddEditedElementDivergence(self.TagName, id, value["type"])
Laurent@814:                     self.RefreshDivergenceModel(divergence)
Laurent@814:                     if next:
Laurent@814:                         wire = self.ConnectConnectors(next, divergence_connectors["outputs"][0])
Laurent@814:                         pos = divergence_connectors["outputs"][0].GetPosition(False)
Laurent@814:                         next_pos = next.GetPosition(False)
Laurent@814:                         next_block = next.GetParentBlock()
Laurent@814:                         divergence.RefreshOutputPosition((0, pos.y + SFC_WIRE_MIN_SIZE - next_pos.y))
Laurent@814:                         divergence.RefreshConnectedPosition(divergence_connectors["outputs"][0])
Laurent@814:                         wire.SetPoints([wx.Point(pos.x, pos.y + SFC_WIRE_MIN_SIZE), wx.Point(pos.x, pos.y)])
Laurent@814:                         next_block.RefreshModel()
Laurent@814:                     else:
Laurent@814:                         if value["type"] == SELECTION_CONVERGENCE:
Laurent@814:                             previous = divergence_connectors["outputs"][0]
Laurent@814:                         else:
Laurent@814:                             transition = self.CreateTransition(divergence_connectors["outputs"][0])
Laurent@814:                             transition_connectors = transition.GetConnectors()
Laurent@814:                             previous = transition_connectors["output"]
Laurent@814:                         self.CreateStep("Step", previous)
Laurent@814:                 self.RefreshBuffer()
Laurent@814:                 self.RefreshScrollBars()
Laurent@814:                 self.Refresh(False)
Laurent@814:             dialog.Destroy()
Laurent@814:     
Laurent@814:     def AddDivergenceBranch(self, divergence):
Laurent@814:         if isinstance(divergence, SFC_Divergence):
Laurent@814:             if self.GetDrawingMode() == FREEDRAWING_MODE:
Laurent@814:                 divergence.AddBranch()
Laurent@814:                 self.RefreshDivergenceModel(divergence)
Laurent@814:             else:
Laurent@814:                 type = divergence.GetType()
Laurent@814:                 if type in [SELECTION_DIVERGENCE, SIMULTANEOUS_DIVERGENCE]:
Laurent@814:                     divergence.AddBranch()
Laurent@814:                     divergence_connectors = divergence.GetConnectors()
Laurent@814:                     if type == SELECTION_DIVERGENCE:
Laurent@814:                         transition = self.CreateTransition(divergence_connectors["outputs"][-1])
Laurent@814:                         transition_connectors = transition.GetConnectors()
Laurent@814:                         previous = transition_connectors["output"]
Laurent@814:                     else:
Laurent@814:                         previous = divergence_connectors["outputs"][-1]
Laurent@814:                     step = self.CreateStep("Step", previous)
Laurent@814:             self.RefreshBuffer()
Laurent@814:             self.RefreshScrollBars()
Laurent@814:             self.Refresh(False)
Laurent@814:     
Laurent@814:     def RemoveDivergenceBranch(self, divergence):
Laurent@814:         if isinstance(divergence, SFC_Divergence):
Laurent@814:             if self.GetDrawingMode() == FREEDRAWING_MODE:
Laurent@814:                 divergence.RemoveHandledBranch()
Laurent@814:                 self.RefreshDivergenceModel(divergence)
Laurent@814:                 self.RefreshBuffer()
Laurent@814:                 self.RefreshScrollBars()
Laurent@814:                 self.Refresh(False)
Laurent@814:     
Laurent@814:     def AddJump(self):
Laurent@814:         if isinstance(self.SelectedElement, SFC_Step) and not self.SelectedElement.Output:
Laurent@814:             choices = []
Laurent@814:             for block in self.Blocks:
Laurent@814:                 if isinstance(block, SFC_Step):
Laurent@814:                     choices.append(block.GetName())
Laurent@814:             dialog = wx.SingleChoiceDialog(self.ParentWindow, 
Laurent@814:                   _("Add a new jump"), _("Please choose a target"), 
Laurent@814:                   choices, wx.DEFAULT_DIALOG_STYLE|wx.OK|wx.CANCEL)
Laurent@814:             if dialog.ShowModal() == wx.ID_OK:
Laurent@814:                 value = dialog.GetStringSelection()
Laurent@814:                 self.SelectedElement.AddOutput()
Laurent@814:                 self.RefreshStepModel(self.SelectedElement)
Laurent@814:                 step_connectors = self.SelectedElement.GetConnectors()
Laurent@814:                 transition = self.CreateTransition(step_connectors["output"])
Laurent@814:                 transition_connectors = transition.GetConnectors()
Laurent@814:                 id = self.GetNewId()
Laurent@814:                 jump = SFC_Jump(self, value, id)
Laurent@814:                 pos = transition_connectors["output"].GetPosition(False)
Laurent@814:                 jump.SetPosition(pos.x, pos.y + SFC_WIRE_MIN_SIZE)
Laurent@814:                 self.AddBlock(jump)
Laurent@814:                 self.Controler.AddEditedElementJump(self.TagName, id)
Laurent@814:                 jump_connector = jump.GetConnector()
Laurent@814:                 wire = self.ConnectConnectors(jump_connector, transition_connectors["output"])
Laurent@814:                 transition.RefreshOutputPosition()
Laurent@814:                 wire.SetPoints([wx.Point(pos.x, pos.y + SFC_WIRE_MIN_SIZE), wx.Point(pos.x, pos.y)])
Laurent@814:                 self.RefreshJumpModel(jump)
Laurent@814:                 self.RefreshBuffer()
Laurent@814:                 self.RefreshScrollBars()
Laurent@814:                 self.Refresh(False)
Laurent@814:             dialog.Destroy()
Laurent@814: 
Laurent@814:     def EditStepContent(self, step):
Laurent@814:         if self.GetDrawingMode() == FREEDRAWING_MODE:
Laurent@814:             Viewer.EditStepContent(self, step)
Laurent@814:         else:
Laurent@814:             dialog = SFCStepNameDialog(self.ParentWindow, _("Edit step name"), _("Please enter step name"), step.GetName(), wx.OK|wx.CANCEL)
Laurent@814:             dialog.SetPouNames(self.Controler.GetProjectPouNames(self.Debug))
Laurent@1347:             dialog.SetVariables(self.Controler.GetEditedElementInterfaceVars(self.TagName, debug=self.Debug))
Laurent@814:             dialog.SetStepNames([block.GetName() for block in self.Blocks if isinstance(block, SFC_Step) and block.GetName() != step.GetName()])
Laurent@814:             if dialog.ShowModal() == wx.ID_OK:
Laurent@814:                 value = dialog.GetValue()
Laurent@814:                 step.SetName(value)
Laurent@814:                 min_size = step.GetMinSize()
Laurent@814:                 size = step.GetSize()
Laurent@814:                 step.UpdateSize(max(min_size[0], size[0]), max(min_size[1], size[1]))
Laurent@814:                 step.RefreshModel()
Laurent@814:                 self.RefreshBuffer()
Laurent@814:                 self.RefreshScrollBars()
Laurent@814:                 self.Refresh(False)
Laurent@814:             dialog.Destroy()
Laurent@814: 
Laurent@814: #-------------------------------------------------------------------------------
Laurent@814: #                          Delete element functions
Laurent@814: #-------------------------------------------------------------------------------
Laurent@814: 
Laurent@814:     def DeleteStep(self, step):
Laurent@814:         if self.GetDrawingMode() == FREEDRAWING_MODE:
Laurent@814:             Viewer.DeleteStep(self, step)
Laurent@814:         else:
Laurent@814:             step_connectors = step.GetConnectors()
Laurent@814:             if not step.GetInitial() or not step_connectors["output"]:
Laurent@814:                 previous = step.GetPreviousConnector()
Laurent@814:                 if previous:
Laurent@814:                     previous_block = previous.GetParentBlock()
Laurent@814:                 else:
Laurent@814:                     previous_block = None
Laurent@814:                 next = step.GetNextConnector()
Laurent@814:                 if next:
Laurent@814:                     next_block = next.GetParentBlock()
Laurent@814:                 else:
Laurent@814:                     next_block = None
Laurent@814:                 if isinstance(next_block, SFC_Transition):
Laurent@814:                     self.RemoveTransition(next_block)
Laurent@814:                     next = step.GetNextConnector()
Laurent@814:                     if next:
Laurent@814:                         next_block = next.GetParentBlock()
Laurent@814:                     else:
Laurent@814:                         next_block = None
Laurent@814:                 elif isinstance(previous_block, SFC_Transition):
Laurent@814:                     self.RemoveTransition(previous_block)
Laurent@814:                     previous = step.GetPreviousConnector()
Laurent@814:                     if previous:
Laurent@814:                         previous_block = previous.GetParentBlock()
Laurent@814:                     else:
Laurent@814:                         previous_block = None
Laurent@814:                 wire = self.RemoveStep(step)
Laurent@814:                 self.SelectedElement = None
Laurent@814:                 if next_block:
Laurent@814:                     if isinstance(next_block, SFC_Divergence) and next_block.GetType() == SIMULTANEOUS_CONVERGENCE and isinstance(previous_block, SFC_Divergence) and previous_block.GetType() == SIMULTANEOUS_DIVERGENCE:
Laurent@814:                         wire.Clean()
Laurent@814:                         self.RemoveWire(wire)
Laurent@814:                         next_block.RemoveBranch(next)
Laurent@814:                         if next_block.GetBranchNumber() < 2:
Laurent@814:                             self.DeleteDivergence(next_block)
Laurent@814:                         else:
Laurent@814:                             next_block.RefreshModel()
Laurent@814:                         previous_block.RemoveBranch(previous)
Laurent@814:                         if previous_block.GetBranchNumber() < 2:
Laurent@814:                             self.DeleteDivergence(previous_block)
Laurent@814:                         else:
Laurent@814:                             previous_block.RefreshModel()
Laurent@814:                     else:
Laurent@814:                         pos = previous.GetPosition(False)
Laurent@814:                         next_pos = next.GetPosition(False)
Laurent@814:                         wire_size = GetWireSize(previous_block)
Laurent@814:                         previous_block.RefreshOutputPosition((0, pos.y + wire_size - next_pos.y))
Laurent@814:                         wire.SetPoints([wx.Point(pos.x, pos.y + wire_size), wx.Point(pos.x, pos.y)])
Laurent@814:                         if isinstance(next_block, SFC_Divergence):
Laurent@814:                             next_block.RefreshPosition()
Laurent@814:                         previous_block.RefreshOutputModel(True)
Laurent@814:                 else:
Laurent@814:                     if isinstance(previous_block, SFC_Step):
Laurent@814:                         previous_block.RemoveOutput()
Laurent@814:                         self.RefreshStepModel(previous_block)
Laurent@814:                     elif isinstance(previous_block, SFC_Divergence):
Laurent@814:                         if previous_block.GetType() in [SELECTION_CONVERGENCE, SIMULTANEOUS_CONVERGENCE]:
Laurent@814:                             self.DeleteDivergence(previous_block)
Laurent@814:                         else:
Laurent@814:                             previous_block.RemoveBranch(previous)
Laurent@814:                             if previous_block.GetBranchNumber() < 2:
Laurent@814:                                 self.DeleteDivergence(previous_block)
Laurent@814:                             else:
Laurent@814:                                 self.RefreshDivergenceModel(previous_block)
Laurent@814:         
Laurent@814:     def DeleteTransition(self, transition):
Laurent@814:         if self.GetDrawingMode() == FREEDRAWING_MODE:
Laurent@814:             Viewer.DeleteTransition(self, transition)
Laurent@814:         else:
Laurent@814:             previous = transition.GetPreviousConnector()
Laurent@814:             previous_block = previous.GetParentBlock()
Laurent@814:             next = transition.GetNextConnector()
Laurent@814:             next_block = next.GetParentBlock()
Laurent@814:             if isinstance(previous_block, SFC_Divergence) and previous_block.GetType() == SELECTION_DIVERGENCE and isinstance(next_block, SFC_Divergence) and next_block.GetType() == SELECTION_CONVERGENCE:
Laurent@814:                 wires = previous.GetWires()
Laurent@814:                 if len(wires) != 1:
Laurent@814:                     return
Laurent@814:                 wire = wires[0][0]
Laurent@814:                 wire.Clean()
Laurent@814:                 self.RemoveWire(wire)
Laurent@814:                 wires = next.GetWires()
Laurent@814:                 if len(wires) != 1:
Laurent@814:                     return
Laurent@814:                 wire = wires[0][0]
Laurent@814:                 wire.Clean()
Laurent@814:                 self.RemoveWire(wire)
Laurent@814:                 transition.Clean()
Laurent@814:                 self.RemoveBlock(transition)
Laurent@814:                 self.Controler.RemoveEditedElementInstance(self.TagName, transition.GetId())
Laurent@814:                 previous_block.RemoveBranch(previous)
Laurent@814:                 if previous_block.GetBranchNumber() < 2:
Laurent@814:                     self.DeleteDivergence(previous_block)
Laurent@814:                 else:
Laurent@814:                     self.RefreshDivergenceModel(previous_block)
Laurent@814:                 next_block.RemoveBranch(next)
Laurent@814:                 if next_block.GetBranchNumber() < 2:
Laurent@814:                     self.DeleteDivergence(next_block)
Laurent@814:                 else:
Laurent@814:                     self.RefreshDivergenceModel(next_block)
Laurent@814:             
Laurent@814:     def DeleteDivergence(self, divergence):
Laurent@814:         if self.GetDrawingMode() == FREEDRAWING_MODE:
Laurent@814:             Viewer.DeleteDivergence(self, divergence)
Laurent@814:         else:
Laurent@814:             connectors = divergence.GetConnectors()
Laurent@814:             type = divergence.GetType()
Laurent@814:             if type in [SELECTION_CONVERGENCE, SIMULTANEOUS_CONVERGENCE]:
Laurent@814:                 wires = connectors["outputs"][0].GetWires()
Laurent@814:                 if len(wires) > 1:
Laurent@814:                     return
Laurent@814:                 elif len(wires) == 1:
Laurent@814:                     next = wires[0][0].StartConnected
Laurent@814:                     next_block = next.GetParentBlock()
Laurent@814:                     wire = wires[0][0]
Laurent@814:                     wire.Clean()
Laurent@814:                     self.RemoveWire(wire)
Laurent@814:                 else:
Laurent@814:                     next = None
Laurent@814:                     next_block = None
Laurent@814:                 for index, connector in enumerate(connectors["inputs"]):
Laurent@814:                     if next and index == 0:
Laurent@814:                         wires = connector.GetWires()
Laurent@814:                         wire = wires[0][0]
Laurent@814:                         previous = wires[0][0].EndConnected
Laurent@814:                         wire.Clean()
Laurent@814:                         self.RemoveWire(wire)
Laurent@814:                     else:
Laurent@814:                         if type == SELECTION_CONVERGENCE:
Laurent@814:                             wires = connector.GetWires()
Laurent@814:                             previous_block = wires[0][0].EndConnected.GetParentBlock()
Laurent@814:                             self.RemoveTransition(previous_block)
Laurent@814:                         wires = connector.GetWires()
Laurent@814:                         wire = wires[0][0]
Laurent@814:                         previous_connector = wire.EndConnected
Laurent@814:                         previous_block = previous_connector.GetParentBlock()
Laurent@814:                         wire.Clean()
Laurent@814:                         self.RemoveWire(wire)
Laurent@814:                         if isinstance(previous_block, SFC_Step):
Laurent@814:                             previous_block.RemoveOutput()
Laurent@814:                             self.RefreshStepModel(previous_block)
Laurent@814:                         elif isinstance(previous_block, SFC_Divergence):
Laurent@814:                             if previous_block.GetType() in [SELECTION_DIVERGENCE, SIMULTANEOUS_DIVERGENCE]:
Laurent@814:                                 previous_block.RemoveBranch(previous_connector)
Laurent@814:                                 if previous_block.GetBranchNumber() < 2:
Laurent@814:                                     self.DeleteDivergence(previous_block)
Laurent@814:                                 else:
Laurent@814:                                     self.RefreshDivergenceModel(previous_block)
Laurent@814:                             else:
Laurent@814:                                 self.DeleteDivergence(previous_block)
Laurent@814:                 divergence.Clean()
Laurent@814:                 self.RemoveBlock(divergence)
Laurent@814:                 self.Controler.RemoveEditedElementInstance(self.TagName, divergence.GetId())
Laurent@814:                 if next:
Laurent@814:                     wire = self.ConnectConnectors(next, previous)
Laurent@814:                     previous_block = previous.GetParentBlock()
Laurent@814:                     previous_pos = previous.GetPosition(False)
Laurent@814:                     next_pos = next.GetPosition(False)
Laurent@814:                     wire_size = GetWireSize(previous_block)
Laurent@814:                     previous_block.RefreshOutputPosition((0, previous_pos.y + wire_size - next_pos.y))
Laurent@814:                     wire.SetPoints([wx.Point(previous_pos.x, previous_pos.y + wire_size), 
Laurent@814:                         wx.Point(previous_pos.x, previous_pos.y)])
Laurent@814:                     if isinstance(next_block, SFC_Divergence):
Laurent@814:                         next_block.RefreshPosition()
Laurent@814:                     previous_block.RefreshOutputModel(True)
Laurent@814:             elif divergence.GetBranchNumber() == 1:
Laurent@814:                 wires = connectors["inputs"][0].GetWires()
Laurent@814:                 if len(wires) != 1:
Laurent@814:                     return
Laurent@814:                 wire = wires[0][0]
Laurent@814:                 previous = wire.EndConnected
Laurent@814:                 previous_block = previous.GetParentBlock()
Laurent@814:                 wire.Clean()
Laurent@814:                 self.RemoveWire(wire)
Laurent@814:                 wires = connectors["outputs"][0].GetWires()
Laurent@814:                 if len(wires) != 1:
Laurent@814:                     return
Laurent@814:                 wire = wires[0][0]
Laurent@814:                 next = wire.StartConnected
Laurent@814:                 next_block = next.GetParentBlock()
Laurent@814:                 wire.Clean()
Laurent@814:                 self.RemoveWire(wire)
Laurent@814:                 divergence.Clean()
Laurent@814:                 self.RemoveBlock(divergence)
Laurent@814:                 self.Controler.RemoveEditedElementInstance(self.TagName, divergence.GetId())
Laurent@814:                 wire = self.ConnectConnectors(next, previous)
Laurent@814:                 previous_pos = previous.GetPosition(False)
Laurent@814:                 next_pos = next.GetPosition(False)
Laurent@814:                 wire_size = GetWireSize(previous_block)
Laurent@814:                 previous_block.RefreshOutputPosition((previous_pos.x - next_pos.x, previous_pos.y + wire_size - next_pos.y))
Laurent@814:                 wire.SetPoints([wx.Point(previous_pos.x, previous_pos.y + wire_size), 
Laurent@814:                     wx.Point(previous_pos.x, previous_pos.y)])
Laurent@814:                 if isinstance(next_block, SFC_Divergence):
Laurent@814:                     next_block.RefreshPosition()
Laurent@814:                 previous_block.RefreshOutputModel(True)
Laurent@814:             
Laurent@814:     def DeleteJump(self, jump):
Laurent@814:         if self.GetDrawingMode() == FREEDRAWING_MODE:
Laurent@814:             Viewer.DeleteJump(self, jump)
Laurent@814:         else:
Laurent@814:             previous = jump.GetPreviousConnector()
Laurent@814:             previous_block = previous.GetParentBlock()
Laurent@814:             if isinstance(previous_block, SFC_Transition):
Laurent@814:                 self.RemoveTransition(previous_block)
Laurent@814:                 previous = jump.GetPreviousConnector()
Laurent@814:                 if previous:
Laurent@814:                     previous_block = previous.GetParentBlock()
Laurent@814:                 else:
Laurent@814:                     previous_block = None
Laurent@814:             wires = previous.GetWires()
Laurent@814:             if len(wires) != 1:
Laurent@814:                 return
Laurent@814:             wire = wires[0][0]
Laurent@814:             wire.Clean()
Laurent@814:             self.RemoveWire(wire)
Laurent@814:             jump.Clean()
Laurent@814:             self.RemoveBlock(jump)
Laurent@814:             self.Controler.RemoveEditedElementInstance(self.TagName, jump.GetId())
Laurent@814:             if isinstance(previous_block, SFC_Step):
Laurent@814:                 previous_block.RemoveOutput()
Laurent@814:                 self.RefreshStepModel(previous_block)
Laurent@814:             elif isinstance(previous_block, SFC_Divergence):
Laurent@814:                 if previous_block.GetType() in [SELECTION_CONVERGENCE, SIMULTANEOUS_CONVERGENCE]:
Laurent@814:                     self.DeleteDivergence(previous_block)
Laurent@814:                 else:
Laurent@814:                     previous_block.RemoveBranch(previous)
Laurent@814:                     if previous_block.GetBranchNumber() < 2:
Laurent@814:                         self.DeleteDivergence(previous_block)
Laurent@814:                     else:
Laurent@814:                         previous_block.RefreshModel()
Laurent@814:             
Laurent@814:     def DeleteActionBlock(self, actionblock):
Laurent@814:         if self.GetDrawingMode() == FREEDRAWING_MODE:
Laurent@814:             Viewer.DeleteActionBlock(self, actionblock)
Laurent@814:         else:
Laurent@814:             connector = actionblock.GetConnector()
Laurent@814:             wires = connector.GetWires()
Laurent@814:             if len(wires) != 1:
Laurent@814:                 return
Laurent@814:             wire = wires[0][0]
Laurent@814:             step = wire.EndConnected.GetParentBlock()
Laurent@814:             wire.Clean()
Laurent@814:             self.RemoveWire(wire)
Laurent@814:             actionblock.Clean()
Laurent@814:             self.RemoveBlock(actionblock)
Laurent@814:             self.Controler.RemoveEditedElementInstance(self.TagName, actionblock.GetId())
Laurent@814:             step.RemoveAction()
Laurent@814:             self.RefreshStepModel(step)
Laurent@814:             step.RefreshOutputPosition()
Laurent@814:             step.RefreshOutputModel(True)
Laurent@814:             
Laurent@814:     def DeleteWire(self, wire):
Laurent@814:         if self.GetDrawingMode() == FREEDRAWING_MODE:
Laurent@814:             Viewer.DeleteWire(self, wire)
Laurent@814: 
Laurent@814: #-------------------------------------------------------------------------------
Laurent@814: #                          Model update functions
Laurent@814: #-------------------------------------------------------------------------------
Laurent@814: 
Laurent@814:     def RefreshBlockModel(self, block):
Laurent@814:         blockid = block.GetId()
Laurent@814:         infos = {}
Laurent@814:         infos["type"] = block.GetType()
Laurent@814:         infos["name"] = block.GetName()
Laurent@814:         infos["x"], infos["y"] = block.GetPosition()
Laurent@814:         infos["width"], infos["height"] = block.GetSize()
Laurent@814:         infos["connectors"] = block.GetConnectors()
Laurent@814:         self.Controler.SetEditedElementBlockInfos(self.TagName, blockid, infos)
Laurent@814: