etisserant@0: #!/usr/bin/env python
etisserant@0: # -*- coding: utf-8 -*-
etisserant@0: 
etisserant@0: #This file is part of PLCOpenEditor, a library implementing an IEC 61131-3 editor
etisserant@0: #based on the plcopen standard. 
etisserant@0: #
etisserant@0: #Copyright (C): Edouard TISSERANT and Laurent BESSARD
etisserant@0: #
etisserant@0: #See COPYING file for copyrights details.
etisserant@0: #
etisserant@0: #This library is free software; you can redistribute it and/or
etisserant@0: #modify it under the terms of the GNU Lesser General Public
etisserant@0: #License as published by the Free Software Foundation; either
etisserant@0: #version 2.1 of the License, or (at your option) any later version.
etisserant@0: #
etisserant@0: #This library is distributed in the hope that it will be useful,
etisserant@0: #but WITHOUT ANY WARRANTY; without even the implied warranty of
etisserant@0: #MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
etisserant@0: #Lesser General Public License for more details.
etisserant@0: #
etisserant@0: #You should have received a copy of the GNU Lesser General Public
etisserant@0: #License along with this library; if not, write to the Free Software
etisserant@0: #Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
etisserant@0: 
etisserant@0: from wxPython.wx import *
etisserant@0: import wx
etisserant@0: 
etisserant@0: from GraphicCommons import *
etisserant@0: from plcopen.structures import *
etisserant@0: 
etisserant@0: def GetWireSize(block):
etisserant@0:     if isinstance(block, SFC_Step):
etisserant@0:         return SFC_WIRE_MIN_SIZE + block.GetActionExtraLineNumber() * SFC_ACTION_MIN_SIZE[1]
etisserant@0:     else:
etisserant@0:         return SFC_WIRE_MIN_SIZE
etisserant@0: 
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: #                         Sequencial Function Chart Step
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: 
etisserant@0: """
etisserant@0: Class that implements the graphic representation of a step
etisserant@0: """
etisserant@0: 
etisserant@0: class SFC_Step(Graphic_Element):
etisserant@0:     
etisserant@0:     # Create a new step
etisserant@0:     def __init__(self, parent, name, initial = False, id = None):
etisserant@0:         Graphic_Element.__init__(self, parent)
etisserant@0:         self.Name = name
etisserant@0:         self.Initial = initial
etisserant@0:         self.Id = id
etisserant@0:         self.Size = wxSize(SFC_STEP_DEFAULT_SIZE[0], SFC_STEP_DEFAULT_SIZE[1])
etisserant@0:         # Create an input and output connector
etisserant@0:         if not self.Initial:
etisserant@0:             self.Input = Connector(self, "", "ANY", wxPoint(self.Size[0] / 2, 0), NORTH)
etisserant@0:         else:
etisserant@0:             self.Input = None
etisserant@0:         self.Output = None
etisserant@0:         self.Action = None
etisserant@0:     
etisserant@0:     # Destructor
etisserant@0:     def __del__(self):
etisserant@0:         self.Input = None
etisserant@0:         self.Output = None
etisserant@0:         self.Action = None
etisserant@0:     
etisserant@0:     # Delete this step by calling the appropriate method
etisserant@0:     def Delete(self):
etisserant@0:         self.Parent.DeleteStep(self)
etisserant@0:     
etisserant@0:     # Unconnect input and output
etisserant@0:     def Clean(self):
etisserant@0:         if self.Input:
etisserant@0:             self.Input.UnConnect()
etisserant@0:         if self.Output:
etisserant@0:             self.Output.UnConnect()
etisserant@0:         if self.Action:
etisserant@0:             self.Action.UnConnect()
etisserant@0:     
etisserant@0:     # Add output connector to step
etisserant@0:     def AddOutput(self):
etisserant@0:         if not self.Output:
etisserant@0:             self.Output = Connector(self, "", "ANY", wxPoint(self.Size[0] / 2, self.Size[1]), SOUTH)
etisserant@0:             self.RefreshBoundingBox()
etisserant@0:     
etisserant@0:     # Remove output connector from step
etisserant@0:     def RemoveOutput(self):
etisserant@0:         if self.Output:
etisserant@0:             self.Output.UnConnect()
etisserant@0:             self.Output = None
etisserant@0:             self.RefreshBoundingBox()
etisserant@0:     
etisserant@0:     # Add action connector to step
etisserant@0:     def AddAction(self):
etisserant@0:         if not self.Action:
etisserant@0:             self.Action = Connector(self, "", "ANY", wxPoint(self.Size[0], self.Size[1] / 2), EAST)
etisserant@0:             self.RefreshBoundingBox()
etisserant@0:     
etisserant@0:     # Remove action connector from step
etisserant@0:     def RemoveAction(self):
etisserant@0:         if self.Action:
etisserant@0:             self.Action.UnConnect()
etisserant@0:             self.Action = None
etisserant@0:             self.RefreshBoundingBox()
etisserant@0:     
etisserant@0:     # Refresh the step bounding box
etisserant@0:     def RefreshBoundingBox(self):
etisserant@0:         dc = wxClientDC(self.Parent)
etisserant@0:         # Calculate the bounding box size
etisserant@0:         if self.Action:
etisserant@0:             bbx_width = self.Size[0] + CONNECTOR_SIZE
etisserant@0:         else:
etisserant@0:             bbx_width = self.Size[0]
etisserant@0:         if self.Initial:
etisserant@0:             bbx_y = self.Pos.y
etisserant@0:             bbx_height = self.Size[1]
etisserant@0:             if self.Output:
etisserant@0:                 bbx_height += CONNECTOR_SIZE
etisserant@0:         else:
etisserant@0:             bbx_y = self.Pos.y - CONNECTOR_SIZE
etisserant@0:             bbx_height = self.Size[1] + CONNECTOR_SIZE
etisserant@0:             if self.Output:
etisserant@0:                 bbx_height += CONNECTOR_SIZE
etisserant@0:         #self.BoundingBox = wxRect(self.Pos.x, bbx_y, bbx_width + 1, bbx_height + 1)
etisserant@0:         self.BoundingBox = wxRect(self.Pos.x, self.Pos.y, self.Size[0] + 1, self.Size[1] + 1)
etisserant@0:         
etisserant@0:     # Refresh the positions of the step connectors
etisserant@0:     def RefreshConnectors(self):
etisserant@0:         # Update input position if it exists
etisserant@0:         if self.Input:
etisserant@0:             self.Input.SetPosition(wxPoint(self.Size[0] / 2, 0))
etisserant@0:         # Update output position
etisserant@0:         if self.Output:
etisserant@0:             self.Output.SetPosition(wxPoint(self.Size[0] / 2, self.Size[1]))
etisserant@0:         # Update action position if it exists
etisserant@0:         if self.Action:
etisserant@0:             self.Action.SetPosition(wxPoint(self.Size[0], self.Size[1] / 2))
etisserant@0:         self.RefreshConnected()
etisserant@0:     
etisserant@0:     # Refresh the position of wires connected to step
etisserant@0:     def RefreshConnected(self, exclude = []):
etisserant@0:         if self.Input:
etisserant@0:             self.Input.MoveConnected(exclude)
etisserant@0:         if self.Output:
etisserant@0:             self.Output.MoveConnected(exclude)
etisserant@0:         if self.Action:
etisserant@0:             self.Action.MoveConnected(exclude)
etisserant@0:     
etisserant@0:     # Returns the step connector that starts with the point given if it exists 
etisserant@0:     def GetConnector(self, position):
etisserant@0:         # Test input connector if it exists
etisserant@0:         if self.Input:
etisserant@0:             input_pos = self.Input.GetRelPosition()
etisserant@0:             if position.x == self.Pos.x + input_pos.x and position.y == self.Pos.y + input_pos.y:
etisserant@0:                 return self.Input
etisserant@0:         # Test output connector if it exists
etisserant@0:         if self.Output:
etisserant@0:             output_pos = self.Output.GetRelPosition()
etisserant@0:             if position.x == self.Pos.x + output_pos.x and position.y == self.Pos.y + output_pos.y:
etisserant@0:                 return self.Output
etisserant@0:         # Test action connector if it exists
etisserant@0:         if self.Action:
etisserant@0:             action_pos = self.Action.GetRelPosition()
etisserant@0:             if position.x == self.Pos.x + action_pos.x and position.y == self.Pos.y + action_pos.y:
etisserant@0:                 return self.Action
etisserant@0:         return None
etisserant@0:     
etisserant@0:     # Returns input and output step connectors 
etisserant@0:     def GetConnectors(self):
etisserant@0:         return {"input":self.Input,"output":self.Output,"action":self.Action}
etisserant@0:     
etisserant@0:     # Test if point given is on step input or output connector
etisserant@0:     def TestConnector(self, pt, exclude=True):
etisserant@0:         # Test input connector if it exists
etisserant@0:         if self.Input and self.Input.TestPoint(pt, exclude):
etisserant@0:             return self.Input
etisserant@0:         # Test output connector
etisserant@0:         if self.Output and self.Output.TestPoint(pt, exclude):
etisserant@0:             return self.Output
etisserant@0:         return None
etisserant@0: 
etisserant@0:     # Changes the step name
etisserant@0:     def SetName(self, name):
etisserant@0:         self.Name = name
etisserant@0: 
etisserant@0:     # Returns the step name
etisserant@0:     def GetName(self):
etisserant@0:         return self.Name
etisserant@0: 
etisserant@0:     # Returns the step initial property
etisserant@0:     def GetInitial(self):
etisserant@0:         return self.Initial
etisserant@0:     
etisserant@0:     # Returns the connector connected to input
etisserant@0:     def GetPreviousConnector(self):
etisserant@0:         if self.Input:
etisserant@0:             wires = self.Input.GetWires()
etisserant@0:             if len(wires) == 1:
etisserant@0:                 return wires[0][0].EndConnected
etisserant@0:         return None
etisserant@0:     
etisserant@0:     # Returns the connector connected to output
etisserant@0:     def GetNextConnector(self):
etisserant@0:         if self.Output:
etisserant@0:             wires = self.Output.GetWires()
etisserant@0:             if len(wires) == 1:
etisserant@0:                 return wires[0][0].StartConnected
etisserant@0:         return None
etisserant@0:     
etisserant@0:     # Returns the connector connected to action
etisserant@0:     def GetActionConnector(self):
etisserant@0:         if self.Action:
etisserant@0:             wires = self.Action.GetWires()
etisserant@0:             if len(wires) == 1:
etisserant@0:                 return wires[0][0].StartConnected
etisserant@0:         return None
etisserant@0:     
etisserant@0:     # Returns the number of action line
etisserant@0:     def GetActionExtraLineNumber(self):
etisserant@0:         if self.Action:
etisserant@0:             wires = self.Action.GetWires()
etisserant@0:             if len(wires) != 1:
etisserant@0:                 return 0
etisserant@0:             action_block = wires[0][0].StartConnected.GetParentBlock()
etisserant@0:             return max(0, action_block.GetLineNumber() - 1)
etisserant@0:         return 0
etisserant@0:     
etisserant@0:     # Returns the step minimum size
etisserant@0:     def GetMinSize(self):
etisserant@0:         dc = wxClientDC(self.Parent)
etisserant@0:         text_width, text_height = dc.GetTextExtent(self.Name)
etisserant@0:         if self.Initial:
etisserant@0:             return text_width + 14, text_height + 14
etisserant@0:         else:
etisserant@0:             return text_width + 10, text_height + 10
etisserant@0:     
etisserant@0:     # Updates the step size
etisserant@0:     def UpdateSize(self, width, height):
etisserant@0:         diffx = self.Size.GetWidth() / 2 - width / 2
etisserant@0:         diffy = height - self.Size.GetHeight()
etisserant@0:         self.Move(diffx, 0)
etisserant@0:         Graphic_Element.SetSize(self, width, height)
etisserant@0:         self.RefreshOutputPosition((0, diffy))
etisserant@0:     
etisserant@0:     # Align input element with this step
etisserant@0:     def RefreshInputPosition(self):
etisserant@0:         if self.Input:
etisserant@0:             current_pos = self.Input.GetPosition(False)
etisserant@0:             input = self.GetPreviousConnector()
etisserant@0:             if input:
etisserant@0:                 input_pos = input.GetPosition(False)
etisserant@0:                 diffx = current_pos.x - input_pos.x
etisserant@0:                 input_block = input.GetParentBlock()
etisserant@0:                 if isinstance(input_block, SFC_Divergence):
etisserant@0:                     input_block.MoveConnector(input, diffx)
etisserant@0:                 else:
etisserant@0:                     if isinstance(input_block, SFC_Step):
etisserant@0:                         input_block.MoveActionBlock((diffx, 0))
etisserant@0:                     input_block.Move(diffx, 0)
etisserant@0:                     input_block.RefreshInputPosition()
etisserant@0:     
etisserant@0:     # Align output element with this step
etisserant@0:     def RefreshOutputPosition(self, move = None):
etisserant@0:         if self.Output:
etisserant@0:             wires = self.Output.GetWires()
etisserant@0:             if len(wires) != 1:
etisserant@0:                 return
etisserant@0:             current_pos = self.Output.GetPosition(False)
etisserant@0:             output = wires[0][0].StartConnected
etisserant@0:             output_pos = output.GetPosition(False)
etisserant@0:             diffx = current_pos.x - output_pos.x
etisserant@0:             output_block = output.GetParentBlock()
etisserant@0:             wire_size = SFC_WIRE_MIN_SIZE + self.GetActionExtraLineNumber() * SFC_ACTION_MIN_SIZE[1]
etisserant@0:             diffy = wire_size - output_pos.y + current_pos.y
etisserant@0:             if diffy != 0:
etisserant@0:                 if isinstance(output_block, SFC_Step):
etisserant@0:                     output_block.MoveActionBlock((diffx, diffy))
etisserant@0:                 wires[0][0].SetPoints([wxPoint(current_pos.x, current_pos.y + wire_size),
etisserant@0:                     wxPoint(current_pos.x, current_pos.y)])
etisserant@0:                 if not isinstance(output_block, SFC_Divergence) or output_block.GetConnectors()["inputs"].index(output) == 0:
etisserant@0:                     output_block.Move(diffx, diffy, self.Parent.Wires)
etisserant@0:                     output_block.RefreshOutputPosition((diffx, diffy))
etisserant@0:                 else:
etisserant@0:                     output_block.RefreshPosition()
etisserant@0:             elif move:
etisserant@0:                 if isinstance(output_block, SFC_Step):
etisserant@0:                     output_block.MoveActionBlock(move)
etisserant@0:                 wires[0][0].Move(move[0], move[1], True)
etisserant@0:                 if not isinstance(output_block, SFC_Divergence) or output_block.GetConnectors()["inputs"].index(output) == 0:
etisserant@0:                     output_block.Move(move[0], move[1], self.Parent.Wires)
etisserant@0:                     output_block.RefreshOutputPosition(move)
etisserant@0:                 else:
etisserant@0:                     output_block.RefreshPosition()
etisserant@0:             elif isinstance(output_block, SFC_Divergence):
etisserant@0:                 output_block.MoveConnector(output, diffx)
etisserant@0:             else:
etisserant@0:                 if isinstance(output_block, SFC_Step):
etisserant@0:                     output_block.MoveActionBlock((diffx, 0))
etisserant@0:                 output_block.Move(diffx, 0)
etisserant@0:                 output_block.RefreshOutputPosition()
etisserant@0:     
etisserant@0:     # Refresh action element with this step
etisserant@0:     def MoveActionBlock(self, move):
etisserant@0:         if self.Action:
etisserant@0:             wires = self.Action.GetWires()
etisserant@0:             if len(wires) != 1:
etisserant@0:                 return
etisserant@0:             action_block = wires[0][0].StartConnected.GetParentBlock()
etisserant@0:             action_block.Move(move[0], move[1], self.Parent.Wires)
etisserant@0:             wires[0][0].Move(move[0], move[1], True)
etisserant@0:     
etisserant@0:     # Resize the divergence from position and size given
etisserant@0:     def Resize(self, x, y, width, height):
etisserant@0:         self.UpdateSize(width, height)
etisserant@0:     
etisserant@0:     # Method called when a LeftDClick event have been generated
etisserant@0:     def OnLeftDClick(self, event, scaling):
etisserant@0:         # Edit the step properties
etisserant@0:         self.Parent.EditStepContent(self)
etisserant@0:     
etisserant@0:     # Method called when a RightUp event have been generated
etisserant@0:     def OnRightUp(self, event, scaling):
etisserant@0:         # Popup the menu with special items for a step
etisserant@0:         self.Parent.PopupDefaultMenu()
etisserant@0:     
etisserant@0:     # Refreshes the step state according to move defined and handle selected
etisserant@0:     def ProcessDragging(self, movex, movey):
etisserant@0:         handle_type, handle = self.Handle
etisserant@0:         if handle_type == HANDLE_MOVE:
etisserant@0:             action_block = None
etisserant@0:             if self.Initial:
etisserant@0:                 self.MoveActionBlock((movex, movey))
etisserant@0:                 self.Move(movex, movey, self.Parent.Wires)
etisserant@0:                 self.RefreshOutputPosition((movex, movey))
etisserant@0:             else:
etisserant@0:                 self.MoveActionBlock((movex, 0))
etisserant@0:                 self.Move(movex, 0)
etisserant@0:                 self.RefreshInputPosition()
etisserant@0:                 self.RefreshOutputPosition()
etisserant@0:         else:
etisserant@0:             Graphic_Element.ProcessDragging(self, movex, movey)
etisserant@0:     
etisserant@0:     # Refresh input element model
etisserant@0:     def RefreshInputModel(self):
etisserant@0:         if self.Input:
etisserant@0:             input = self.GetPreviousConnector()
etisserant@0:             if input:                
etisserant@0:                 input_block = input.GetParentBlock()
etisserant@0:                 input_block.RefreshModel(False)
etisserant@0:                 if not isinstance(input_block, SFC_Divergence):
etisserant@0:                     input_block.RefreshInputModel()
etisserant@0:     
etisserant@0:     # Refresh output element model
etisserant@0:     def RefreshOutputModel(self, move=False):
etisserant@0:         if self.Output:
etisserant@0:             output = self.GetNextConnector()
etisserant@0:             if output:
etisserant@0:                 output_block = output.GetParentBlock()
etisserant@0:                 output_block.RefreshModel(False)
etisserant@0:                 if not isinstance(output_block, SFC_Divergence) or move:
etisserant@0:                     output_block.RefreshOutputModel(move)
etisserant@0:     
etisserant@0:     # Refreshes the step model
etisserant@0:     def RefreshModel(self, move=True):
etisserant@0:         self.Parent.RefreshStepModel(self)
etisserant@0:         if self.Action:
etisserant@0:             action = self.GetActionConnector()
etisserant@0:             if action:
etisserant@0:                 action_block = action.GetParentBlock()
etisserant@0:                 action_block.RefreshModel(False)
etisserant@0:         # If step has moved, refresh the model of wires connected to output
etisserant@0:         if move:
etisserant@0:             self.RefreshInputModel()
etisserant@0:             self.RefreshOutputModel(self.Initial)
etisserant@0:     
etisserant@0:     # Draws step
etisserant@0:     def Draw(self, dc):
etisserant@0:         dc.SetPen(wxBLACK_PEN)
etisserant@0:         dc.SetBrush(wxWHITE_BRUSH)
etisserant@0:         # Draw two rectangles for representing the step
etisserant@0:         dc.DrawRectangle(self.Pos.x, self.Pos.y, self.Size[0] + 1, self.Size[1] + 1)
etisserant@0:         if self.Initial:
etisserant@0:             dc.DrawRectangle(self.Pos.x + 2, self.Pos.y + 2, self.Size[0] - 3, self.Size[1] - 3)
etisserant@0:         # Draw step name
etisserant@0:         namewidth, nameheight = dc.GetTextExtent(self.Name)
etisserant@0:         dc.DrawText(self.Name, self.Pos.x + (self.Size[0] - namewidth) / 2,
etisserant@0:                     self.Pos.y + (self.Size[1] - nameheight) / 2)
etisserant@0:         # Draw input and output connectors
etisserant@0:         if self.Input:
etisserant@0:             self.Input.Draw(dc)
etisserant@0:         if self.Output:
etisserant@0:             self.Output.Draw(dc)
etisserant@0:         if self.Action:
etisserant@0:             self.Action.Draw(dc)
etisserant@0:         Graphic_Element.Draw(self, dc)
etisserant@0: 
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: #                       Sequencial Function Chart Transition
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: 
etisserant@0: """
etisserant@0: Class that implements the graphic representation of a transition
etisserant@0: """
etisserant@0: 
etisserant@0: class SFC_Transition(Graphic_Element):
etisserant@0:     
etisserant@0:     # Create a new transition
etisserant@0:     def __init__(self, parent, type = "reference", condition = "", id = None):
etisserant@0:         Graphic_Element.__init__(self, parent)
etisserant@0:         self.Type = type
etisserant@0:         self.Condition = condition
etisserant@0:         self.Id = id
etisserant@0:         self.Size = wxSize(SFC_TRANSITION_SIZE[0], SFC_TRANSITION_SIZE[1])
etisserant@0:         # Create an input and output connector
etisserant@0:         self.Input = Connector(self, "", "ANY", wxPoint(self.Size[0] / 2, 0), NORTH)
etisserant@0:         self.Output = Connector(self, "", "ANY", wxPoint(self.Size[0] / 2, self.Size[1]), SOUTH)
etisserant@0:     
etisserant@0:     # Destructor
etisserant@0:     def __del__(self):
etisserant@0:         self.Input = None
etisserant@0:         self.Output = None
etisserant@0:     
etisserant@0:     # Forbids to change the transition size
etisserant@0:     def SetSize(self, width, height):
etisserant@0:         pass
etisserant@0:     
etisserant@0:     # Forbids to resize the transition
etisserant@0:     def Resize(self, x, y, width, height):
etisserant@0:         pass
etisserant@0:     
etisserant@0:     # Delete this transition by calling the appropriate method
etisserant@0:     def Delete(self):
etisserant@0:         self.Parent.DeleteTransition(self)
etisserant@0:     
etisserant@0:     # Unconnect input and output
etisserant@0:     def Clean(self):
etisserant@0:         self.Input.UnConnect()
etisserant@0:         self.Output.UnConnect()
etisserant@0:     
etisserant@0:     # Refresh the transition bounding box
etisserant@0:     def RefreshBoundingBox(self):
etisserant@0:         dc = wxClientDC(self.Parent)
etisserant@0:         if self.Condition != "":
etisserant@0:             text_width, text_height = dc.GetTextExtent(self.Condition)
etisserant@0:         else:
etisserant@0:             text_width, text_height = dc.GetTextExtent("Transition")
etisserant@0:         # Calculate the bounding box size
etisserant@0:         bbx_width = self.Size[0] + CONNECTOR_SIZE + 2 + text_width
etisserant@0:         bbx_y = self.Pos.y - max(0, (text_height - self.Size[1]) / 2)
etisserant@0:         bbx_height = max(self.Size[1], text_height)
etisserant@0:         self.BoundingBox = wxRect(self.Pos.x, bbx_y, bbx_width + 1, bbx_height + 1)
etisserant@0:         
etisserant@0:     # Returns the connector connected to input
etisserant@0:     def GetPreviousConnector(self):
etisserant@0:         wires = self.Input.GetWires()
etisserant@0:         if len(wires) == 1:
etisserant@0:             return wires[0][0].EndConnected
etisserant@0:         return None
etisserant@0:     
etisserant@0:     # Returns the connector connected to output
etisserant@0:     def GetNextConnector(self):
etisserant@0:         wires = self.Output.GetWires()
etisserant@0:         if len(wires) == 1:
etisserant@0:             return wires[0][0].StartConnected
etisserant@0:         return None
etisserant@0:     
etisserant@0:     # Refresh the positions of the transition connectors
etisserant@0:     def RefreshConnectors(self):
etisserant@0:         # Update input position
etisserant@0:         self.Input.SetPosition(wxPoint(self.Size[0] / 2, 0))
etisserant@0:         # Update output position
etisserant@0:         self.Output.SetPosition(wxPoint(self.Size[0] / 2, self.Size[1]))
etisserant@0:         self.RefreshConnected()
etisserant@0:     
etisserant@0:     # Refresh the position of the wires connected to transition
etisserant@0:     def RefreshConnected(self, exclude = []):
etisserant@0:         self.Input.MoveConnected(exclude)
etisserant@0:         self.Output.MoveConnected(exclude)
etisserant@0:     
etisserant@0:     # Returns the transition connector that starts with the point given if it exists 
etisserant@0:     def GetConnector(self, position):
etisserant@0:         # Test input connector
etisserant@0:         input_pos = self.Input.GetRelPosition()
etisserant@0:         if position.x == self.Pos.x + input_pos.x and position.y == self.Pos.y + input_pos.y:
etisserant@0:             return self.Input
etisserant@0:         # Test output connector
etisserant@0:         output_pos = self.Output.GetRelPosition()
etisserant@0:         if position.x == self.Pos.x + output_pos.x and position.y == self.Pos.y + output_pos.y:
etisserant@0:             return self.Output
etisserant@0:         return None
etisserant@0:     
etisserant@0:     # Returns input and output transition connectors
etisserant@0:     def GetConnectors(self):
etisserant@0:         return {"input":self.Input,"output":self.Output}
etisserant@0:     
etisserant@0:     # Test if point given is on transition input or output connector
etisserant@0:     def TestConnector(self, pt, exclude=True):
etisserant@0:         # Test input connector
etisserant@0:         if self.Input.TestPoint(pt, exclude):
etisserant@0:             return self.Input
etisserant@0:         # Test output connector
etisserant@0:         if self.Output.TestPoint(pt, exclude):
etisserant@0:             return self.Output
etisserant@0:         return None
etisserant@0: 
etisserant@0:     # Changes the transition type
etisserant@0:     def SetType(self, type):
etisserant@0:         self.Type = type
etisserant@0:         
etisserant@0:     # Returns the transition type
etisserant@0:     def GetType(self):
etisserant@0:         return self.Type
etisserant@0: 
etisserant@0:     # Changes the transition condition
etisserant@0:     def SetCondition(self, condition):
etisserant@0:         self.Condition = condition
etisserant@0:         self.RefreshBoundingBox()
etisserant@0: 
etisserant@0:     # Returns the transition condition
etisserant@0:     def GetCondition(self):
etisserant@0:         return self.Condition
etisserant@0:         
etisserant@0:     # Returns the transition minimum size
etisserant@0:     def GetMinSize(self):
etisserant@0:         return SFC_TRANSITION_SIZE
etisserant@0:     
etisserant@0:     # Align input element with this step
etisserant@0:     def RefreshInputPosition(self):
etisserant@0:         wires = self.Input.GetWires()
etisserant@0:         current_pos = self.Input.GetPosition(False)
etisserant@0:         input = self.GetPreviousConnector()
etisserant@0:         if input:
etisserant@0:             input_pos = input.GetPosition(False)
etisserant@0:             diffx = current_pos.x - input_pos.x
etisserant@0:             input_block = input.GetParentBlock()
etisserant@0:             if isinstance(input_block, SFC_Divergence):
etisserant@0:                 input_block.MoveConnector(input, diffx)
etisserant@0:             else:
etisserant@0:                 if isinstance(input_block, SFC_Step):
etisserant@0:                     input_block.MoveActionBlock((diffx, 0))
etisserant@0:                 input_block.Move(diffx, 0)
etisserant@0:                 input_block.RefreshInputPosition()
etisserant@0:     
etisserant@0:     # Align output element with this step
etisserant@0:     def RefreshOutputPosition(self, move = None):
etisserant@0:         wires = self.Output.GetWires()
etisserant@0:         if len(wires) != 1:
etisserant@0:             return
etisserant@0:         current_pos = self.Output.GetPosition(False)
etisserant@0:         output = wires[0][0].StartConnected
etisserant@0:         output_pos = output.GetPosition(False)
etisserant@0:         diffx = current_pos.x - output_pos.x
etisserant@0:         output_block = output.GetParentBlock()
etisserant@0:         if move:
etisserant@0:             if isinstance(output_block, SFC_Step):
etisserant@0:                 output_block.MoveActionBlock(move)
etisserant@0:             wires[0][0].Move(move[0], move[1], True)
etisserant@0:             if not isinstance(output_block, SFC_Divergence) or output_block.GetConnectors()["inputs"].index(output) == 0:
etisserant@0:                 output_block.Move(move[0], move[1], self.Parent.Wires)
etisserant@0:                 output_block.RefreshOutputPosition(move)
etisserant@0:             else:
etisserant@0:                 output_block.RefreshPosition()
etisserant@0:         elif isinstance(output_block, SFC_Divergence):
etisserant@0:             output_block.MoveConnector(output, diffx)
etisserant@0:         else:
etisserant@0:             if isinstance(output_block, SFC_Step):
etisserant@0:                 output_block.MoveActionBlock((diffx, 0))
etisserant@0:             output_block.Move(diffx, 0)
etisserant@0:             output_block.RefreshOutputPosition()
etisserant@0:     
etisserant@0:     # Method called when a LeftDClick event have been generated
etisserant@0:     def OnLeftDClick(self, event, scaling):
etisserant@0:         # Edit the transition properties
etisserant@0:         self.Parent.EditTransitionContent(self)
etisserant@0:     
etisserant@0:     # Method called when a RightUp event have been generated
etisserant@0:     def OnRightUp(self, event, scaling):
etisserant@0:         # Popup the menu with special items for a step
etisserant@0:         self.Parent.PopupDefaultMenu()
etisserant@0:     
etisserant@0:     # Refreshes the transition state according to move defined and handle selected
etisserant@0:     def ProcessDragging(self, movex, movey):
etisserant@0:         self.Move(movex, 0)
etisserant@0:         self.RefreshInputPosition()
etisserant@0:         self.RefreshOutputPosition()
etisserant@0:     
etisserant@0:     # Refresh input element model
etisserant@0:     def RefreshInputModel(self):
etisserant@0:         input = self.GetPreviousConnector()
etisserant@0:         if input:
etisserant@0:             input_block = input.GetParentBlock()
etisserant@0:             input_block.RefreshModel(False)
etisserant@0:             if not isinstance(input_block, SFC_Divergence):
etisserant@0:                 input_block.RefreshInputModel()
etisserant@0:     
etisserant@0:     # Refresh output element model
etisserant@0:     def RefreshOutputModel(self, move=False):
etisserant@0:         output = self.GetNextConnector()
etisserant@0:         if output:
etisserant@0:             output_block = output.GetParentBlock()
etisserant@0:             output_block.RefreshModel(False)
etisserant@0:             if not isinstance(output_block, SFC_Divergence) or move:
etisserant@0:                 output_block.RefreshOutputModel(move)
etisserant@0:     
etisserant@0:     # Refreshes the transition model
etisserant@0:     def RefreshModel(self, move=True):
etisserant@0:         self.Parent.RefreshTransitionModel(self)
etisserant@0:         # If transition has moved, refresh the model of wires connected to output
etisserant@0:         if move:
etisserant@0:             self.RefreshInputModel()
etisserant@0:             self.RefreshOutputModel()
etisserant@0:     
etisserant@0:     # Draws transition
etisserant@0:     def Draw(self, dc):
etisserant@0:         dc.SetPen(wxBLACK_PEN)
etisserant@0:         dc.SetBrush(wxBLACK_BRUSH)
etisserant@0:         # Draw plain rectangle for representing the transition
etisserant@0:         dc.DrawRectangle(self.Pos.x, self.Pos.y, self.Size[0] + 1, self.Size[1] + 1)
etisserant@0:         # Draw transition condition
etisserant@0:         if self.Condition != "":
etisserant@0:             text_width, text_height = dc.GetTextExtent(self.Condition)
etisserant@0:             condition = self.Condition
etisserant@0:         else:
etisserant@0:             text_width, text_height = dc.GetTextExtent("Transition")
etisserant@0:             condition = "Transition"
etisserant@0:         dc.DrawText(condition, self.Pos.x + self.Size[0] + CONNECTOR_SIZE + 2,
etisserant@0:                     self.Pos.y + (self.Size[1] - text_height) / 2)
etisserant@0:         # Draw input and output connectors
etisserant@0:         self.Input.Draw(dc)
etisserant@0:         self.Output.Draw(dc)
etisserant@0:         Graphic_Element.Draw(self, dc)
etisserant@0: 
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: #                Sequencial Function Chart Divergence and Convergence
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: 
etisserant@0: """
etisserant@0: Class that implements the graphic representation of a divergence or convergence,
etisserant@0: selection or simultaneous
etisserant@0: """
etisserant@0: 
etisserant@0: class SFC_Divergence(Graphic_Element):
etisserant@0:     
etisserant@0:     # Create a new divergence
etisserant@0:     def __init__(self, parent, type, number = 2, id = None):
etisserant@0:         Graphic_Element.__init__(self, parent)
etisserant@0:         self.Type = type
etisserant@0:         self.Id = id
etisserant@0:         self.RealConnectors = None
etisserant@0:         number = max(2, number)
etisserant@0:         if self.Type in [SELECTION_DIVERGENCE, SELECTION_CONVERGENCE]:
etisserant@0:             self.Size = wxSize((number - 1) * SFC_DEFAULT_SEQUENCE_INTERVAL, 1)
etisserant@0:         elif self.Type in [SIMULTANEOUS_DIVERGENCE, SIMULTANEOUS_CONVERGENCE]:
etisserant@0:             self.Size = wxSize((number - 1) * SFC_DEFAULT_SEQUENCE_INTERVAL, 3)
etisserant@0:         # Create an input and output connector
etisserant@0:         if self.Type in [SELECTION_DIVERGENCE, SIMULTANEOUS_DIVERGENCE]:
etisserant@0:             self.Inputs = [Connector(self, "", "ANY", wxPoint(self.Size[0] / 2, 0), NORTH)]
etisserant@0:             self.Outputs = []
etisserant@0:             for i in xrange(number):
etisserant@0:                 self.Outputs.append(Connector(self, "", "ANY", wxPoint(i * SFC_DEFAULT_SEQUENCE_INTERVAL, self.Size[1]), SOUTH))
etisserant@0:         elif self.Type in [SELECTION_CONVERGENCE, SIMULTANEOUS_CONVERGENCE]:
etisserant@0:             self.Inputs = []
etisserant@0:             for i in xrange(number):
etisserant@0:                 self.Inputs.append(Connector(self, "", "ANY", wxPoint(i * SFC_DEFAULT_SEQUENCE_INTERVAL, 0), NORTH))
etisserant@0:             self.Outputs = [Connector(self, "", "ANY", wxPoint(self.Size[0] / 2, self.Size[1]), SOUTH)]
etisserant@0:     
etisserant@0:     # Destructor
etisserant@0:     def __del__(self):
etisserant@0:         self.Inputs = []
etisserant@0:         self.Outputs = []
etisserant@0:     
etisserant@0:     # Forbids to resize the divergence
etisserant@0:     def Resize(self, x, y, width, height):
etisserant@0:         pass
etisserant@0:     
etisserant@0:     # Delete this divergence by calling the appropriate method
etisserant@0:     def Delete(self):
etisserant@0:         self.Parent.DeleteDivergence(self)
etisserant@0:     
etisserant@0:     # Returns the divergence type
etisserant@0:     def GetType(self):
etisserant@0:         return self.Type
etisserant@0:     
etisserant@0:     # Unconnect input and output
etisserant@0:     def Clean(self):
etisserant@0:         for input in self.Inputs:
etisserant@0:             input.UnConnect()
etisserant@0:         for output in self.Outputs:
etisserant@0:             output.UnConnect()
etisserant@0:     
etisserant@0:     # Add a branch to the divergence
etisserant@0:     def AddBranch(self):
etisserant@0:         if self.Type in [SELECTION_DIVERGENCE, SIMULTANEOUS_DIVERGENCE]:
etisserant@0:             maxx = 0
etisserant@0:             for output in self.Outputs:
etisserant@0:                 pos = output.GetRelPosition()
etisserant@0:                 maxx = max(maxx, pos.x)
etisserant@0:             connector = Connector(self, "", "ANY", wxPoint(maxx + SFC_DEFAULT_SEQUENCE_INTERVAL, self.Size[1]), SOUTH)
etisserant@0:             self.Outputs.append(connector)
etisserant@0:             self.MoveConnector(connector, 0)
etisserant@0:         elif self.Type in [SELECTION_CONVERGENCE, SIMULTANEOUS_CONVERGENCE]:
etisserant@0:             maxx = 0
etisserant@0:             for input in self.Inputs:
etisserant@0:                 pos = input.GetRelPosition()
etisserant@0:                 maxx = max(maxx, pos.x)
etisserant@0:             connector = Connector(self, "", "ANY", wxPoint(maxx + SFC_DEFAULT_SEQUENCE_INTERVAL, 0), NORTH)
etisserant@0:             self.Inputs.append(connector)
etisserant@0:             self.MoveConnector(connector, SFC_DEFAULT_SEQUENCE_INTERVAL)
etisserant@0:     
etisserant@0:     # Remove a branch from the divergence
etisserant@0:     def RemoveBranch(self, connector):
etisserant@0:         if self.Type in [SELECTION_DIVERGENCE, SIMULTANEOUS_DIVERGENCE]:
etisserant@0:             if connector in self.Outputs:
etisserant@0:                 self.Outputs.remove(connector)
etisserant@0:                 self.MoveConnector(self.Outputs[0], 0)
etisserant@0:         elif self.Type in [SELECTION_CONVERGENCE, SIMULTANEOUS_CONVERGENCE]:
etisserant@0:             if connector in self.Inputs:
etisserant@0:                 self.Inputs.remove(connector)
etisserant@0:                 self.MoveConnector(self.Inputs[0], 0)
etisserant@0:     
etisserant@0:     # Return the number of branches for the divergence
etisserant@0:     def GetBranchNumber(self):
etisserant@0:         if self.Type in [SELECTION_DIVERGENCE, SIMULTANEOUS_DIVERGENCE]:
etisserant@0:             return len(self.Outputs)
etisserant@0:         elif self.Type in [SELECTION_CONVERGENCE, SIMULTANEOUS_CONVERGENCE]:
etisserant@0:             return len(self.Inputs)
etisserant@0:     
etisserant@0:     # Returns if the point given is in the bounding box
etisserant@0:     def HitTest(self, pt):
etisserant@0:         rect = self.BoundingBox
etisserant@0:         return rect.InsideXY(pt.x, pt.y) or self.TestConnector(pt, False) != None
etisserant@0:     
etisserant@0:     # Refresh the divergence bounding box
etisserant@0:     def RefreshBoundingBox(self):
etisserant@0:         if self.Type in [SELECTION_DIVERGENCE, SELECTION_CONVERGENCE]:
etisserant@0:             self.BoundingBox = wxRect(self.Pos.x, self.Pos.y - CONNECTOR_SIZE, 
etisserant@0:                 self.Size[0] + 1, self.Size[1] + CONNECTOR_SIZE * 2 + 1)
etisserant@0:         elif self.Type in [SIMULTANEOUS_DIVERGENCE, SIMULTANEOUS_CONVERGENCE]:
etisserant@0:             self.BoundingBox = wxRect(self.Pos.x - SFC_SIMULTANEOUS_SEQUENCE_EXTRA, self.Pos.y - CONNECTOR_SIZE, 
etisserant@0:                 self.Size[0] + 2 * SFC_SIMULTANEOUS_SEQUENCE_EXTRA + 1, self.Size[1] + CONNECTOR_SIZE * 2 + 1)
etisserant@0:     
etisserant@0:     # Refresh the position of wires connected to divergence
etisserant@0:     def RefreshConnected(self, exclude = []):
etisserant@0:         for input in self.Inputs:
etisserant@0:             input.MoveConnected(exclude)
etisserant@0:         for output in self.Outputs:
etisserant@0:             output.MoveConnected(exclude)
etisserant@0:     
etisserant@0:     # Moves the divergence connector given
etisserant@0:     def MoveConnector(self, connector, movex):
etisserant@0:         position = connector.GetRelPosition()
etisserant@0:         connector.SetPosition(wxPoint(position.x + movex, position.y))
etisserant@0:         minx = self.Size[0]
etisserant@0:         maxx = 0
etisserant@0:         for input in self.Inputs:
etisserant@0:             input_pos = input.GetRelPosition()
etisserant@0:             minx = min(minx, input_pos.x)
etisserant@0:             maxx = max(maxx, input_pos.x)
etisserant@0:         for output in self.Outputs:
etisserant@0:             output_pos = output.GetRelPosition()
etisserant@0:             minx = min(minx, output_pos.x)
etisserant@0:             maxx = max(maxx, output_pos.x)
etisserant@0:         if minx != 0:
etisserant@0:             for input in self.Inputs:
etisserant@0:                 input_pos = input.GetRelPosition()
etisserant@0:                 input.SetPosition(wxPoint(input_pos.x - minx, input_pos.y))
etisserant@0:             for output in self.Outputs:
etisserant@0:                 output_pos = output.GetRelPosition()
etisserant@0:                 output.SetPosition(wxPoint(output_pos.x - minx, output_pos.y))
etisserant@0:         self.Pos.x += minx
etisserant@0:         self.Size[0] = maxx - minx
etisserant@0:         connector.MoveConnected()
etisserant@0:         self.RefreshBoundingBox()
etisserant@0:     
etisserant@0:     # Returns the divergence connector that starts with the point given if it exists 
etisserant@0:     def GetConnector(self, position):
etisserant@0:         # Test input connector
etisserant@0:         for input in self.Inputs:
etisserant@0:             input_pos = input.GetPosition(False)
etisserant@0:             if position.x == input_pos.x and position.y == input_pos.y:
etisserant@0:                 return input
etisserant@0:         # Test output connector
etisserant@0:         for output in self.Outputs:
etisserant@0:             output_pos = output.GetPosition(False)
etisserant@0:             if position.x == output_pos.x and position.y == output_pos.y:
etisserant@0:                 return output
etisserant@0:         return None
etisserant@0:     
etisserant@0:     # Returns input and output divergence connectors 
etisserant@0:     def GetConnectors(self):
etisserant@0:         return {"inputs":self.Inputs,"outputs":self.Outputs}
etisserant@0:     
etisserant@0:     # Test if point given is on divergence input or output connector
etisserant@0:     def TestConnector(self, pt, exclude=True):
etisserant@0:         # Test input connector
etisserant@0:         for input in self.Inputs:
etisserant@0:             if input.TestPoint(pt, exclude):
etisserant@0:                 return input
etisserant@0:         # Test output connector
etisserant@0:         for output in self.Outputs:
etisserant@0:             if output.TestPoint(pt, exclude):
etisserant@0:                 return output
etisserant@0:         return None
etisserant@0:     
etisserant@0:     # Changes the divergence size
etisserant@0:     def SetSize(self, width, height):
etisserant@0:         for i, input in enumerate(self.Inputs):
etisserant@0:             position = input.GetRelPosition()
etisserant@0:             if self.RealConnectors:
etisserant@0:                 input.SetPosition(wxPoint(int(round(self.RealConnectors["Inputs"][i] * width)), position.y))
etisserant@0:             else:
etisserant@0:                 input.SetPosition(wxPoint(int(round(float(position.x)*float(width)/float(self.Size[0]))), position.y))
etisserant@0:             input.MoveConnected()
etisserant@0:         for i, output in enumerate(self.Outputs):
etisserant@0:             position = output.GetRelPosition()
etisserant@0:             if self.RealConnectors:
etisserant@0:                 output.SetPosition(wxPoint(int(round(self.RealConnectors["Outputs"][i] * width)), position.y))
etisserant@0:             else:
etisserant@0:                 output.SetPosition(wxPoint(int(round(float(position.x)*float(width)/float(self.Size[0]))), position.y))
etisserant@0:             output.MoveConnected()
etisserant@0:         self.Size = wxSize(width, self.Size[1])
etisserant@0:         self.RefreshBoundingBox()
etisserant@0:     
etisserant@0:     # Returns the divergence minimum size
etisserant@0:     def GetMinSize(self):
etisserant@0:         return 0, self.Size[1]
etisserant@0:     
etisserant@0:     # Refresh the position of the block connected to connector
etisserant@0:     def RefreshConnectedPosition(self, connector):
etisserant@0:         wires = connector.GetWires()
etisserant@0:         if len(wires) != 1:
etisserant@0:             return
etisserant@0:         current_pos = connector.GetPosition(False)
etisserant@0:         if connector in self.Inputs:
etisserant@0:             next = wires[0][0].EndConnected
etisserant@0:         else:
etisserant@0:             next = wires[0][0].StartConnected
etisserant@0:         next_pos = next.GetPosition(False)
etisserant@0:         diffx = current_pos.x - next_pos.x
etisserant@0:         next_block = next.GetParentBlock()
etisserant@0:         if isinstance(next_block, SFC_Divergence):
etisserant@0:             next_block.MoveConnector(next, diffx)
etisserant@0:         else:
etisserant@0:             next_block.Move(diffx, 0)
etisserant@0:             if connector in self.Inputs:
etisserant@0:                 next_block.RefreshInputPosition()
etisserant@0:             else:
etisserant@0:                 next_block.RefreshOutputPosition()
etisserant@0:     
etisserant@0:     # Refresh the position of this divergence
etisserant@0:     def RefreshPosition(self):
etisserant@0:         y = 0
etisserant@0:         for input in self.Inputs:
etisserant@0:             wires = input.GetWires()
etisserant@0:             if len(wires) != 1:
etisserant@0:                 return
etisserant@0:             previous = wires[0][0].EndConnected
etisserant@0:             previous_pos = previous.GetPosition(False)
etisserant@0:             y = max(y, previous_pos.y + GetWireSize(previous.GetParentBlock()))
etisserant@0:         diffy = y - self.Pos.y
etisserant@0:         if diffy != 0:
etisserant@0:             self.Move(0, diffy, self.Parent.Wires)
etisserant@0:             self.RefreshOutputPosition((0, diffy))
etisserant@0:         for input in self.Inputs:
etisserant@0:             input.MoveConnected()
etisserant@0:     
etisserant@0:     # Align output element with this divergence
etisserant@0:     def RefreshOutputPosition(self, move = None):
etisserant@0:         if move:
etisserant@0:             for output_connector in self.Outputs:
etisserant@0:                 wires = output_connector.GetWires()
etisserant@0:                 if len(wires) != 1:
etisserant@0:                     return
etisserant@0:                 current_pos = output_connector.GetPosition(False)
etisserant@0:                 output = wires[0][0].StartConnected
etisserant@0:                 output_pos = output.GetPosition(False)
etisserant@0:                 diffx = current_pos.x - output_pos.x
etisserant@0:                 output_block = output.GetParentBlock()
etisserant@0:                 if isinstance(output_block, SFC_Step):
etisserant@0:                     output_block.MoveActionBlock(move)
etisserant@0:                 wires[0][0].Move(move[0], move[1], True)
etisserant@0:                 if not isinstance(output_block, SFC_Divergence) or output_block.GetConnectors()["inputs"].index(output) == 0:
etisserant@0:                     output_block.Move(move[0], move[1], self.Parent.Wires)
etisserant@0:                     output_block.RefreshOutputPosition(move)
etisserant@0:     
etisserant@0:     # Method called when a LeftDown event have been generated
etisserant@0:     def OnLeftDown(self, event, scaling):
etisserant@0:         pos = GetScaledEventPosition(event, scaling)
etisserant@0:         # Test if a connector have been handled
etisserant@0:         connector = self.TestConnector(pos, False)
etisserant@0:         if connector:
etisserant@0:             self.Handle = (HANDLE_CONNECTOR, connector)
etisserant@0:             self.Parent.SetCursor(wxStockCursor(wxCURSOR_HAND))
etisserant@0:             self.Selected = False
etisserant@0:             # Initializes the last position
etisserant@0:             self.oldPos = GetScaledEventPosition(event, scaling)
etisserant@0:         else:
etisserant@0:             self.RealConnectors = {"Inputs":[],"Outputs":[]}
etisserant@0:             for input in self.Inputs:
etisserant@0:                 position = input.GetRelPosition()
etisserant@0:                 self.RealConnectors["Inputs"].append(float(position.x)/float(self.Size[0]))
etisserant@0:             for output in self.Outputs:
etisserant@0:                 position = output.GetRelPosition()
etisserant@0:                 self.RealConnectors["Outputs"].append(float(position.x)/float(self.Size[0]))
etisserant@0:             Graphic_Element.OnLeftDown(self, event, scaling)
etisserant@0:     
etisserant@0:     # Method called when a LeftUp event have been generated
etisserant@0:     def OnLeftUp(self, event, scaling):
etisserant@0:         self.RealConnectors = None
etisserant@0:         handle_type, handle = self.Handle
etisserant@0:         if handle_type == HANDLE_CONNECTOR:
etisserant@0:             wires = handle.GetWires()
etisserant@0:             if len(wires) != 1:
etisserant@0:                 return
etisserant@0:             if handle in self.Inputs:
etisserant@0:                 block = wires[0][0].EndConnected.GetParentBlock()
etisserant@0:             else:
etisserant@0:                 block = wires[0][0].StartConnected.GetParentBlock()
etisserant@0:             block.RefreshModel(False)
etisserant@0:             if not isinstance(block, SFC_Divergence):
etisserant@0:                 if handle in self.Inputs:
etisserant@0:                     block.RefreshInputModel()
etisserant@0:                 else:
etisserant@0:                     block.RefreshOutputModel()
etisserant@0:         Graphic_Element.OnLeftUp(self, event, scaling)
etisserant@0:     
etisserant@0:     # Method called when a RightUp event have been generated
etisserant@0:     def OnRightUp(self, event, scaling):
etisserant@0:         pos = GetScaledEventPosition(event, scaling)
etisserant@0:         # Popup the menu with special items for a block and a connector if one is handled
etisserant@0:         connector = self.TestConnector(pos, False)
etisserant@0:         if connector:
etisserant@0:             self.Handle = (HANDLE_CONNECTOR, connector)
etisserant@0:             self.Parent.PopupDivergenceMenu(True)
etisserant@0:         else:
etisserant@0:             # Popup the divergence menu without delete branch
etisserant@0:             self.Parent.PopupDivergenceMenu(False)
etisserant@0:     
etisserant@0:     # Refreshes the divergence state according to move defined and handle selected
etisserant@0:     def ProcessDragging(self, movex, movey):
etisserant@0:         handle_type, handle = self.Handle
etisserant@0:         # A connector has been handled
etisserant@0:         if handle_type == HANDLE_CONNECTOR:
etisserant@0:             self.MoveConnector(handle, movex)
etisserant@0:             self.RefreshConnectedPosition(handle)
etisserant@0:     
etisserant@0:     # Refresh output element model
etisserant@0:     def RefreshOutputModel(self, move=False):
etisserant@0:         if move:
etisserant@0:             for output in self.Outputs:
etisserant@0:                 wires = output.GetWires()
etisserant@0:                 if len(wires) != 1:
etisserant@0:                     return
etisserant@0:                 output_block = wires[0][0].StartConnected.GetParentBlock()
etisserant@0:                 output_block.RefreshModel(False)
etisserant@0:                 if not isinstance(output_block, SFC_Divergence) or move:
etisserant@0:                     output_block.RefreshOutputModel(move)
etisserant@0:     
etisserant@0:     # Refreshes the divergence model
etisserant@0:     def RefreshModel(self, move=True):
etisserant@0:         self.Parent.RefreshDivergenceModel(self)
etisserant@0:         # If divergence has moved, refresh the model of wires connected to outputs
etisserant@0:         if move:
etisserant@0:             self.RefreshOutputModel()
etisserant@0:     
etisserant@0:     # Draws divergence
etisserant@0:     def Draw(self, dc):
etisserant@0:         dc.SetPen(wxBLACK_PEN)
etisserant@0:         dc.SetBrush(wxBLACK_BRUSH)
etisserant@0:         # Draw plain rectangle for representing the divergence
etisserant@0:         if self.Type in [SELECTION_DIVERGENCE, SELECTION_CONVERGENCE]:
etisserant@0:             dc.DrawRectangle(self.Pos.x, self.Pos.y, self.Size[0] + 1, self.Size[1] + 1)
etisserant@0:         elif self.Type in [SIMULTANEOUS_DIVERGENCE, SIMULTANEOUS_CONVERGENCE]:
etisserant@0:             dc.DrawLine(self.Pos.x - SFC_SIMULTANEOUS_SEQUENCE_EXTRA, self.Pos.y, 
etisserant@0:                         self.Pos.x + self.Size[0] + SFC_SIMULTANEOUS_SEQUENCE_EXTRA + 1, self.Pos.y)
etisserant@0:             dc.DrawLine(self.Pos.x - SFC_SIMULTANEOUS_SEQUENCE_EXTRA, self.Pos.y + 3, 
etisserant@0:                         self.Pos.x + self.Size[0] + SFC_SIMULTANEOUS_SEQUENCE_EXTRA + 1, self.Pos.y + 3)
etisserant@0:         # Draw inputs and outputs connectors
etisserant@0:         for input in self.Inputs:
etisserant@0:             input.Draw(dc)
etisserant@0:         for output in self.Outputs:
etisserant@0:             output.Draw(dc)
etisserant@0:         Graphic_Element.Draw(self, dc)
etisserant@0: 
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: #                   Sequencial Function Chart Jump to Step
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: 
etisserant@0: """
etisserant@0: Class that implements the graphic representation of a jump to step
etisserant@0: """
etisserant@0: 
etisserant@0: class SFC_Jump(Graphic_Element):
etisserant@0:     
etisserant@0:     # Create a new jump
etisserant@0:     def __init__(self, parent, target, id = None):
etisserant@0:         Graphic_Element.__init__(self, parent)
etisserant@0:         self.Target = target
etisserant@0:         self.Id = id
etisserant@0:         self.Size = wxSize(SFC_JUMP_SIZE[0], SFC_JUMP_SIZE[1])
etisserant@0:         # Create an input and output connector
etisserant@0:         self.Input = Connector(self, "", "ANY", wxPoint(self.Size[0] / 2, 0), NORTH)
etisserant@0:         
etisserant@0:     # Destructor
etisserant@0:     def __del__(self):
etisserant@0:         self.Inputs = None
etisserant@0:     
etisserant@0:     # Forbids to change the jump size
etisserant@0:     def SetSize(self, width, height):
etisserant@0:         pass
etisserant@0:     
etisserant@0:     # Forbids to resize jump
etisserant@0:     def Resize(self, x, y, width, height):
etisserant@0:         pass
etisserant@0:     
etisserant@0:     # Delete this jump by calling the appropriate method
etisserant@0:     def Delete(self):
etisserant@0:         self.Parent.DeleteJump(self)
etisserant@0:     
etisserant@0:     # Unconnect input
etisserant@0:     def Clean(self):
etisserant@0:         if self.Input:
etisserant@0:             self.Input.UnConnect()
etisserant@0:     
etisserant@0:     # Refresh the jump bounding box
etisserant@0:     def RefreshBoundingBox(self):
etisserant@0:         dc = wxClientDC(self.Parent)
etisserant@0:         text_width, text_height = dc.GetTextExtent(self.Target)
etisserant@0:         # Calculate the bounding box size
etisserant@0:         bbx_width = self.Size[0] + 2 + text_width
etisserant@0:         self.BoundingBox = wxRect(self.Pos.x, self.Pos.y - CONNECTOR_SIZE, 
etisserant@0:                 bbx_width + 1, self.Size[1] + CONNECTOR_SIZE + 1)
etisserant@0:     
etisserant@0:     # Returns the connector connected to input
etisserant@0:     def GetPreviousConnector(self):
etisserant@0:         if self.Input:
etisserant@0:             wires = self.Input.GetWires()
etisserant@0:             if len(wires) == 1:
etisserant@0:                 return wires[0][0].EndConnected
etisserant@0:         return None
etisserant@0:     
etisserant@0:     # Refresh the position of wires connected to jump
etisserant@0:     def RefreshConnected(self, exclude = []):
etisserant@0:         if self.Input:
etisserant@0:             self.Input.MoveConnected(exclude)
etisserant@0:     
etisserant@0:     # Returns input jump connector 
etisserant@0:     def GetConnector(self, position = None):
etisserant@0:         return self.Input
etisserant@0:     
etisserant@0:     # Test if point given is on jump input connector
etisserant@0:     def TestConnector(self, pt, exclude = True):
etisserant@0:         # Test input connector
etisserant@0:         if self.Input and self.Input.TestPoint(pt, exclude):
etisserant@0:             return self.Input
etisserant@0:         return None
etisserant@0:     
etisserant@0:     # Changes the jump target
etisserant@0:     def SetTarget(self, target):
etisserant@0:         self.Target = target
etisserant@0:         self.RefreshBoundingBox()
etisserant@0:     
etisserant@0:     # Returns the jump target
etisserant@0:     def GetTarget(self):
etisserant@0:         return self.Target
etisserant@0:     
etisserant@0:     # Returns the jump minimum size
etisserant@0:     def GetMinSize(self):
etisserant@0:         return SFC_JUMP_SIZE
etisserant@0:     
etisserant@0:     # Align input element with this jump
etisserant@0:     def RefreshInputPosition(self):
etisserant@0:         if self.Input:
etisserant@0:             current_pos = self.Input.GetPosition(False)
etisserant@0:             input = self.GetPreviousConnector()
etisserant@0:             if input:
etisserant@0:                 input_pos = input.GetPosition(False)
etisserant@0:                 diffx = current_pos.x - input_pos.x
etisserant@0:                 input_block = input.GetParentBlock()
etisserant@0:                 if isinstance(input_block, SFC_Divergence):
etisserant@0:                     input_block.MoveConnector(input, diffx)
etisserant@0:                 else:
etisserant@0:                     if isinstance(input_block, SFC_Step):
etisserant@0:                         input_block.MoveActionBlock((diffx, 0))
etisserant@0:                     input_block.Move(diffx, 0)
etisserant@0:                     input_block.RefreshInputPosition()
etisserant@0:     
etisserant@0:     # Can't align output element, because there is no output
etisserant@0:     def RefreshOutputPosition(self, move = None):
etisserant@0:         pass
etisserant@0:     
etisserant@0:     # Method called when a LeftDClick event have been generated
etisserant@0:     def OnLeftDClick(self, event, scaling):
etisserant@0:         # Edit the jump properties
etisserant@0:         self.Parent.EditJumpContent(self)
etisserant@0:     
etisserant@0:     # Method called when a RightUp event have been generated
etisserant@0:     def OnRightUp(self, event, scaling):
etisserant@0:         # Popup the default menu
etisserant@0:         self.Parent.PopupDefaultMenu()
etisserant@0:     
etisserant@0:     # Refreshes the jump state according to move defined and handle selected
etisserant@0:     def ProcessDragging(self, movex, movey):
etisserant@0:         self.Move(movex, 0)
etisserant@0:         self.RefreshInputPosition()
etisserant@0:     
etisserant@0:     # Refresh input element model
etisserant@0:     def RefreshInputModel(self):
etisserant@0:         input = self.GetPreviousConnector()
etisserant@0:         if input:
etisserant@0:             input_block = input.GetParentBlock()
etisserant@0:             input_block.RefreshModel(False)
etisserant@0:             if not isinstance(input_block, SFC_Divergence):
etisserant@0:                 input_block.RefreshInputModel()
etisserant@0:     
etisserant@0:     # Refresh output element model
etisserant@0:     def RefreshOutputModel(self, move=False):
etisserant@0:         pass
etisserant@0:     
etisserant@0:     # Refreshes the jump model
etisserant@0:     def RefreshModel(self, move=True):
etisserant@0:         self.Parent.RefreshJumpModel(self)
etisserant@0:         if move:
etisserant@0:             self.RefreshInputModel()
etisserant@0:     
etisserant@0:     # Draws divergence
etisserant@0:     def Draw(self, dc):
etisserant@0:         dc.SetPen(wxBLACK_PEN)
etisserant@0:         dc.SetBrush(wxBLACK_BRUSH)
etisserant@0:         # Draw plain rectangle for representing the divergence
etisserant@0:         dc.DrawLine(self.Pos.x + self.Size[0] / 2, self.Pos.y, self.Pos.x + self.Size[0] / 2, self.Pos.y + self.Size[1])
etisserant@0:         points = [wxPoint(self.Pos.x, self.Pos.y),
etisserant@0:                   wxPoint(self.Pos.x + self.Size[0] / 2, self.Pos.y + self.Size[1] / 3),
etisserant@0:                   wxPoint(self.Pos.x + self.Size[0], self.Pos.y),
etisserant@0:                   wxPoint(self.Pos.x + self.Size[0] / 2, self.Pos.y + self.Size[1])]
etisserant@0:         dc.DrawPolygon(points)
etisserant@0:         text_width, text_height = dc.GetTextExtent(self.Target)
etisserant@0:         dc.DrawText(self.Target, self.Pos.x + self.Size[0] + 2,
etisserant@0:                     self.Pos.y + (self.Size[1] - text_height) / 2)
etisserant@0:         # Draw input connector
etisserant@0:         if self.Input:
etisserant@0:             self.Input.Draw(dc)
etisserant@0:         Graphic_Element.Draw(self, dc)
etisserant@0: 
etisserant@0: 
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: #                   Sequencial Function Chart Action Block
etisserant@0: #-------------------------------------------------------------------------------
etisserant@0: 
etisserant@0: """
etisserant@0: Class that implements the graphic representation of an action block
etisserant@0: """
etisserant@0: 
etisserant@0: class SFC_ActionBlock(Graphic_Element):
etisserant@0:     
etisserant@0:     # Create a new action block
etisserant@0:     def __init__(self, parent, actions = [], id = None):
etisserant@0:         Graphic_Element.__init__(self, parent)
etisserant@0:         self.Id = id
etisserant@0:         self.Size = wxSize(SFC_ACTION_MIN_SIZE[0], SFC_ACTION_MIN_SIZE[1])
etisserant@0:         # Create an input and output connector
etisserant@0:         self.Input = Connector(self, "", "ANY", wxPoint(0, SFC_ACTION_MIN_SIZE[1] / 2), WEST)
etisserant@0:         self.SetActions(actions)
etisserant@0:     
etisserant@0:     # Destructor
etisserant@0:     def __del__(self):
etisserant@0:         self.Input = None
etisserant@0:     
etisserant@0:     # Returns the number of action lines
etisserant@0:     def GetLineNumber(self):
etisserant@0:         return len(self.Actions)
etisserant@0:     
etisserant@0:     # Forbids to resize the action block
etisserant@0:     def Resize(self, x, y, width, height):
etisserant@0:         if x == 0:
etisserant@0:             self.SetSize(width, self.Size[1])
etisserant@0:     
etisserant@0:     # Delete this action block by calling the appropriate method
etisserant@0:     def Delete(self):
etisserant@0:         self.Parent.DeleteActionBlock(self)
etisserant@0:     
etisserant@0:     # Unconnect input and output
etisserant@0:     def Clean(self):
etisserant@0:         self.Input.UnConnect()
etisserant@0:         
etisserant@0:     # Refresh the action block bounding box
etisserant@0:     def RefreshBoundingBox(self):
etisserant@0:         self.BoundingBox = wxRect(self.Pos.x, self.Pos.y, self.Size[0], self.Size[1])
etisserant@0:     
etisserant@0:     # Refresh the position of wires connected to action block
etisserant@0:     def RefreshConnected(self, exclude = []):
etisserant@0:         self.Input.MoveConnected(exclude)
etisserant@0:     
etisserant@0:     # Returns input action block connector 
etisserant@0:     def GetConnector(self, position = None):
etisserant@0:         return self.Input
etisserant@0:     
etisserant@0:     # Test if point given is on action block input connector
etisserant@0:     def TestConnector(self, pt, exclude = True):
etisserant@0:         # Test input connector
etisserant@0:         if self.Input.TestPoint(pt, exclude):
etisserant@0:             return self.Input
etisserant@0:         return None
etisserant@0:     
etisserant@0:     # Changes the action block actions
etisserant@0:     def SetActions(self, actions):
etisserant@0:         dc = wxClientDC(self.Parent)
etisserant@0:         self.Actions = actions
etisserant@0:         self.ColSize = [0, 0, 0]
etisserant@0:         for action in self.Actions:
etisserant@0:             width, height = dc.GetTextExtent(action["qualifier"])
etisserant@0:             self.ColSize[0] = max(self.ColSize[0], width + 10)
etisserant@0:             if "duration" in action:
etisserant@0:                 width, height = dc.GetTextExtent("T#%s"%action["duration"])
etisserant@0:                 self.ColSize[0] = max(self.ColSize[0], width + 10)
etisserant@0:             width, height = dc.GetTextExtent(action["value"])
etisserant@0:             self.ColSize[1] = max(self.ColSize[1], width + 10)
etisserant@0:             if "indicator" in action and action["indicator"] != "":
etisserant@0:                 width, height = dc.GetTextExtent(action["indicator"])
etisserant@0:                 self.ColSize[2] = max(self.ColSize[2], width + 10)
etisserant@0:         self.Size = wxSize(max(self.ColSize[0] + self.ColSize[1] + self.ColSize[2],
etisserant@0:             self.Size[0]), len(self.Actions) * SFC_ACTION_MIN_SIZE[1])
etisserant@0:         self.RefreshBoundingBox()
etisserant@0:         if self.Input:
etisserant@0:             wires = self.Input.GetWires()
etisserant@0:             if len(wires) == 1:
etisserant@0:                 input_block = wires[0][0].EndConnected.GetParentBlock()
etisserant@0:                 input_block.RefreshOutputPosition()
etisserant@0:                 input_block.RefreshOutputModel(True)
etisserant@0:     
etisserant@0:     # Returns the action block actions
etisserant@0:     def GetActions(self):
etisserant@0:         return self.Actions
etisserant@0:     
etisserant@0:     # Returns the action block minimum size
etisserant@0:     def GetMinSize(self):
etisserant@0:         return SFC_ACTION_MIN_SIZE[0], len(self.Actions) * SFC_ACTION_MIN_SIZE[1]
etisserant@0:     
etisserant@0:     # Method called when a LeftDClick event have been generated
etisserant@0:     def OnLeftDClick(self, event, scaling):
etisserant@0:         # Edit the action block properties
etisserant@0:         self.Parent.EditActionBlockContent(self)
etisserant@0:     
etisserant@0:     # Refreshes the action block state according to move defined and handle selected
etisserant@0:     def ProcessDragging(self, movex, movey):
etisserant@0:         handle_type, handle = self.Handle
etisserant@0:         if handle_type == HANDLE_MOVE:
etisserant@0:             wires = self.Input.GetWires()
etisserant@0:             if len(wires) == 1:
etisserant@0:                 input_pos = wires[0][0].EndConnected.GetPosition(False)
etisserant@0:                 if self.Pos.x - input_pos.x + movex >= SFC_WIRE_MIN_SIZE:
etisserant@0:                     self.Move(movex, 0)
etisserant@0:         else:
etisserant@0:             Graphic_Element.ProcessDragging(self, movex, movey)
etisserant@0:     
etisserant@0:    # Refreshes the action block model
etisserant@0:     def RefreshModel(self, move=True):
etisserant@0:         self.Parent.RefreshActionBlockModel(self)
etisserant@0:     
etisserant@0:     # Draws divergence
etisserant@0:     def Draw(self, dc):
etisserant@0:         dc.SetPen(wxBLACK_PEN)
etisserant@0:         dc.SetBrush(wxWHITE_BRUSH)
etisserant@0:         colsize = [self.ColSize[0], self.Size[0] - self.ColSize[0] - self.ColSize[2], self.ColSize[2]]
etisserant@0:         # Draw plain rectangle for representing the action block
etisserant@0:         dc.DrawRectangle(self.Pos.x, self.Pos.y, self.Size[0] + 1, self.Size[1] + 1)
etisserant@0:         dc.DrawLine(self.Pos.x + colsize[0], self.Pos.y, 
etisserant@0:                 self.Pos.x + colsize[0], self.Pos.y + self.Size[1])
etisserant@0:         dc.DrawLine(self.Pos.x + colsize[0] + colsize[1], self.Pos.y, 
etisserant@0:                 self.Pos.x + colsize[0] + colsize[1], self.Pos.y + self.Size[1])
etisserant@0:         for i, action in enumerate(self.Actions):
etisserant@0:             if i != 0:
etisserant@0:                 dc.DrawLine(self.Pos.x, self.Pos.y + i * SFC_ACTION_MIN_SIZE[1], 
etisserant@0:                     self.Pos.x + self.Size[0], self.Pos.y + i * SFC_ACTION_MIN_SIZE[1])
etisserant@0:             text_width, text_height = dc.GetTextExtent(action["qualifier"])
etisserant@0:             if "duration" in action:
etisserant@0:                 dc.DrawText(action["qualifier"], self.Pos.x + (colsize[0] - text_width) / 2,
etisserant@0:                     self.Pos.y + i * SFC_ACTION_MIN_SIZE[1] + SFC_ACTION_MIN_SIZE[1] / 2 - text_height)
etisserant@0:                 text_width, text_height = dc.GetTextExtent("T#%s"%action["duration"])
etisserant@0:                 dc.DrawText("T#%s"%action["duration"], self.Pos.x + (colsize[0] - text_width) / 2,
etisserant@0:                     self.Pos.y + i * SFC_ACTION_MIN_SIZE[1] + SFC_ACTION_MIN_SIZE[1] / 2)
etisserant@0:             else:
etisserant@0:                 dc.DrawText(action["qualifier"], self.Pos.x + (colsize[0] - text_width) / 2,
etisserant@0:                         self.Pos.y + i * SFC_ACTION_MIN_SIZE[1] + (SFC_ACTION_MIN_SIZE[1] - text_height) / 2)
etisserant@0:             text_width, text_height = dc.GetTextExtent(action["value"])
etisserant@0:             dc.DrawText(action["value"], self.Pos.x + colsize[0] + (colsize[1] - text_width) / 2,
etisserant@0:                     self.Pos.y + i * SFC_ACTION_MIN_SIZE[1] + (SFC_ACTION_MIN_SIZE[1] - text_height) / 2)
etisserant@0:             if "indicator" in action:
etisserant@0:                 text_width, text_height = dc.GetTextExtent(action["indicator"])
etisserant@0:                 dc.DrawText(action["indicator"], self.Pos.x + colsize[0] + colsize[1] + (colsize[2] - text_width) / 2,
etisserant@0:                         self.Pos.y + i * SFC_ACTION_MIN_SIZE[1] + (SFC_ACTION_MIN_SIZE[1] - text_height) / 2)
etisserant@0:         # Draw input connector
etisserant@0:         self.Input.Draw(dc)
etisserant@0:         Graphic_Element.Draw(self, dc)