1

#!/usr/bin/env python

     2

# -*- coding: utf-8 -*-

     3

     4

# This file is part of Beremiz, a Integrated Development Environment for

     5

# programming IEC 61131-3 automates supporting plcopen standard and CanFestival.

     6

#

     7

# Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD

     8

#

     9

# See COPYING file for copyrights details.

    10

#

    11

# This program is free software; you can redistribute it and/or

    12

# modify it under the terms of the GNU General Public License

    13

# as published by the Free Software Foundation; either version 2

    14

# of the License, or (at your option) any later version.

    15

#

    16

# This program is distributed in the hope that it will be useful,

    17

# but WITHOUT ANY WARRANTY; without even the implied warranty of

    18

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

    19

# GNU General Public License for more details.

    20

#

    21

# You should have received a copy of the GNU General Public License

    22

# along with this program; if not, write to the Free Software

    23

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

    24

    25

    26

    27

    28

    29

import wx

    30

    31

from editors.Viewer import *

    32

from graphics.SFC_Objects import *

    33

from graphics.GraphicCommons import SELECTION_DIVERGENCE, \

    34

    SELECTION_CONVERGENCE, SIMULTANEOUS_DIVERGENCE, SIMULTANEOUS_CONVERGENCE, EAST, NORTH, WEST, SOUTH

    35

    36

SFC_Objects = (SFC_Step, SFC_ActionBlock, SFC_Transition, SFC_Divergence, SFC_Jump)

    37

    38

    39

class SFC_Viewer(Viewer):

    40

    41

    SFC_StandardRules = {

    42

        # The key of this dict is a block that user try to connect,

    43

        # and the value is a list of blocks, that can be connected with the current block

    44

        # and with directions of connection

    45

        "SFC_Step": [("SFC_ActionBlock", EAST),

    46

                     ("SFC_Transition", SOUTH),

    47

                     (SELECTION_DIVERGENCE, SOUTH),

    48

                     (SIMULTANEOUS_CONVERGENCE, SOUTH)],

    49

    50

        "SFC_ActionBlock": [("SFC_Step", EAST)],

    51

    52

        "SFC_Transition": [("SFC_Step", SOUTH),

    53

                           (SELECTION_CONVERGENCE, SOUTH),

    54

                           (SIMULTANEOUS_DIVERGENCE, SOUTH),

    55

                           ("SFC_Jump", SOUTH),

    56

                           ("FBD_Block", EAST),

    57

                           ("FBD_Variable", EAST),

    58

                           ("FBD_Connector", EAST),

    59

                           ("LD_Contact", EAST),

    60

                           ("LD_PowerRail", EAST),

    61

                           ("LD_Coil", EAST)],

    62

    63

        SELECTION_DIVERGENCE: [("SFC_Transition", SOUTH)],

    64

    65

        SELECTION_CONVERGENCE: [("SFC_Step", SOUTH),

    66

                                ("SFC_Jump", SOUTH)],

    67

    68

        SIMULTANEOUS_DIVERGENCE: [("SFC_Step", SOUTH)],

    69

    70

        SIMULTANEOUS_CONVERGENCE: [("SFC_Transition", SOUTH)],

    71

    72

        "SFC_Jump": [],

    73

    74

        "FBD_Block": [("SFC_Transition", WEST)],

    75

    76

        "FBD_Variable": [("SFC_Transition", WEST)],

    77

    78

        "FBD_Connector": [("SFC_Transition", WEST)],

    79

    80

        "LD_Contact": [("SFC_Transition", WEST)],

    81

    82

        "LD_PowerRail": [("SFC_Transition", WEST)],

    83

    84

        "LD_Coil": [("SFC_Transition", WEST)]

    85

    }

    86

    87

    def __init__(self, parent, tagname, window, controler, debug=False, instancepath=""):

    88

        Viewer.__init__(self, parent, tagname, window, controler, debug, instancepath)

    89

        self.CurrentLanguage = "SFC"

    90

    91

    def ConnectConnectors(self, start, end):

    92

        startpoint = [start.GetPosition(False), start.GetDirection()]

    93

        endpoint = [end.GetPosition(False), end.GetDirection()]

    94

        wire = Wire(self, startpoint, endpoint)

    95

        self.AddWire(wire)

    96

        start.Connect((wire, 0), False)

    97

        end.Connect((wire, -1), False)

    98

        wire.ConnectStartPoint(None, start)

    99

        wire.ConnectEndPoint(None, end)

   100

        return wire

   101

   102

    def CreateTransition(self, connector, next=None):

   103

        previous = connector.GetParentBlock()

   104

        id = self.GetNewId()

   105

        transition = SFC_Transition(self, "reference", "", 0, id)

   106

        pos = connector.GetPosition(False)

   107

        transition.SetPosition(pos.x, pos.y + SFC_WIRE_MIN_SIZE)

   108

        transition_connectors = transition.GetConnectors()

   109

        wire = self.ConnectConnectors(transition_connectors["input"], connector)

   110

        if isinstance(previous, SFC_Divergence):

   111

            previous.RefreshConnectedPosition(connector)

   112

        else:

   113

            previous.RefreshOutputPosition()

   114

        wire.SetPoints([wx.Point(pos.x, pos.y + GetWireSize(previous)), wx.Point(pos.x, pos.y)])

   115

        self.AddBlock(transition)

   116

        self.Controler.AddEditedElementTransition(self.TagName, id)

   117

        self.RefreshTransitionModel(transition)

   118

        if next:

   119

            wire = self.ConnectConnectors(next, transition_connectors["output"])

   120

            pos = transition_connectors["output"].GetPosition(False)

   121

            next_block = next.GetParentBlock()

   122

            next_pos = next.GetPosition(False)

   123

            transition.RefreshOutputPosition((0, pos.y + SFC_WIRE_MIN_SIZE - next_pos.y))

   124

            wire.SetPoints([wx.Point(pos.x, pos.y + SFC_WIRE_MIN_SIZE), wx.Point(pos.x, pos.y)])

   125

            if isinstance(next_block, SFC_Divergence):

   126

                next_block.RefreshPosition()

   127

            transition.RefreshOutputModel(True)

   128

        return transition

   129

   130

    def RemoveTransition(self, transition):

   131

        connectors = transition.GetConnectors()

   132

        input_wires = connectors["input"].GetWires()

   133

        if len(input_wires) != 1:

   134

            return

   135

        input_wire = input_wires[0][0]

   136

        previous = input_wire.EndConnected

   137

        input_wire.Clean()

   138

        self.RemoveWire(input_wire)

   139

        output_wires = connectors["output"].GetWires()

   140

        if len(output_wires) != 1:

   141

            return

   142

        output_wire = output_wires[0][0]

   143

        next = output_wire.StartConnected

   144

        output_wire.Clean()

   145

        self.RemoveWire(output_wire)

   146

        transition.Clean()

   147

        self.RemoveBlock(transition)

   148

        self.Controler.RemoveEditedElementInstance(self.TagName, transition.GetId())

   149

        wire = self.ConnectConnectors(next, previous)

   150

        return wire

   151

   152

    def CreateStep(self, name, connector, next=None):

   153

        previous = connector.GetParentBlock()

   154

        id = self.GetNewId()

   155

        step = SFC_Step(self, name, False, id)

   156

        if next:

   157

            step.AddOutput()

   158

        min_width, min_height = step.GetMinSize()

   159

        pos = connector.GetPosition(False)

   160

        step.SetPosition(pos.x, pos.y + SFC_WIRE_MIN_SIZE)

   161

        step.SetSize(min_width, min_height)

   162

        step_connectors = step.GetConnectors()

   163

        wire = self.ConnectConnectors(step_connectors["input"], connector)

   164

        if isinstance(previous, SFC_Divergence):

   165

            previous.RefreshConnectedPosition(connector)

   166

        else:

   167

            previous.RefreshOutputPosition()

   168

        wire.SetPoints([wx.Point(pos.x, pos.y + GetWireSize(previous)), wx.Point(pos.x, pos.y)])

   169

        self.AddBlock(step)

   170

        self.Controler.AddEditedElementStep(self.TagName, id)

   171

        self.RefreshStepModel(step)

   172

        if next:

   173

            wire = self.ConnectConnectors(next, step_connectors["output"])

   174

            pos = step_connectors["output"].GetPosition(False)

   175

            next_block = next.GetParentBlock()

   176

            next_pos = next.GetPosition(False)

   177

            step.RefreshOutputPosition((0, pos.y + SFC_WIRE_MIN_SIZE - next_pos.y))

   178

            wire.SetPoints([wx.Point(pos.x, pos.y + SFC_WIRE_MIN_SIZE), wx.Point(pos.x, pos.y)])

   179

            if isinstance(next_block, SFC_Divergence):

   180

                next_block.RefreshPosition()

   181

            step.RefreshOutputModel(True)

   182

        return step

   183

   184

    def RemoveStep(self, step):

   185

        connectors = step.GetConnectors()

   186

        if connectors["input"]:

   187

            input_wires = connectors["input"].GetWires()

   188

            if len(input_wires) != 1:

   189

                return

   190

            input_wire = input_wires[0][0]

   191

            previous = input_wire.EndConnected

   192

            input_wire.Clean()

   193

            self.RemoveWire(input_wire)

   194

        else:

   195

            previous = None

   196

        if connectors["output"]:

   197

            output_wires = connectors["output"].GetWires()

   198

            if len(output_wires) != 1:

   199

                return

   200

            output_wire = output_wires[0][0]

   201

            next = output_wire.StartConnected

   202

            output_wire.Clean()

   203

            self.RemoveWire(output_wire)

   204

        else:

   205

            next = None

   206

        action = step.GetActionConnected()

   207

        if action:

   208

            self.DeleteActionBlock(action.GetParentBlock())

   209

        step.Clean()

   210

        self.RemoveBlock(step)

   211

        self.Controler.RemoveEditedElementInstance(self.TagName, step.GetId())

   212

        if next and previous:

   213

            wire = self.ConnectConnectors(next, previous)

   214

            return wire

   215

        else:

   216

            return None

   217

   218

    # -------------------------------------------------------------------------------

   219

    #                          Mouse event functions

   220

    # -------------------------------------------------------------------------------

   221

   222

    def OnViewerLeftDown(self, event):

   223

        if self.GetDrawingMode() == FREEDRAWING_MODE:

   224

            Viewer.OnViewerLeftDown(self, event)

   225

        elif self.Mode == MODE_SELECTION:

   226

            if event.ShiftDown() and not event.ControlDown() and self.SelectedElement is not None:

   227

                element = self.FindElement(event, True)

   228

                if element and not self.IsWire(element):

   229

                    if isinstance(self.SelectedElement, Graphic_Group):

   230

                        self.SelectedElement.SelectElement(element)

   231

                    else:

   232

                        group = Graphic_Group(self)

   233

                        self.SelectedElement.SetSelected(False)

   234

                        group.SelectElement(self.SelectedElement)

   235

                        group.SelectElement(element)

   236

                        self.SelectedElement = group

   237

                    elements = self.SelectedElement.GetElements()

   238

                    if len(elements) == 0:

   239

                        self.SelectedElement = element

   240

                    elif len(elements) == 1:

   241

                        self.SelectedElement = elements[0]

   242

                    self.SelectedElement.SetSelected(True)

   243

            else:

   244

                element = self.FindElement(event)

   245

                if self.SelectedElement and self.SelectedElement != element:

   246

                    if self.IsWire(self.SelectedElement):

   247

                        self.SelectedElement.SetSelectedSegment(None)

   248

                    else:

   249

                        self.SelectedElement.SetSelected(False)

   250

                    self.SelectedElement = None

   251

                if element:

   252

                    self.SelectedElement = element

   253

                    self.SelectedElement.OnLeftDown(event, self.GetLogicalDC(), self.Scaling)

   254

                    self.SelectedElement.Refresh()

   255

                else:

   256

                    self.rubberBand.Reset()

   257

                    self.rubberBand.OnLeftDown(event, self.GetLogicalDC(), self.Scaling)

   258

        elif self.Mode == MODE_COMMENT:

   259

            self.rubberBand.Reset()

   260

            self.rubberBand.OnLeftDown(event, self.GetLogicalDC(), self.Scaling)

   261

        event.Skip()

   262

   263

    def OnViewerLeftUp(self, event):

   264

        if self.GetDrawingMode() == FREEDRAWING_MODE:

   265

            Viewer.OnViewerLeftUp(self, event)

   266

        elif self.rubberBand.IsShown():

   267

            if self.Mode == MODE_SELECTION:

   268

                elements = self.SearchElements(self.rubberBand.GetCurrentExtent())

   269

                self.rubberBand.OnLeftUp(event, self.GetLogicalDC(), self.Scaling)

   270

                if len(elements) > 0:

   271

                    self.SelectedElement = Graphic_Group(self)

   272

                    self.SelectedElement.SetElements(elements)

   273

                    self.SelectedElement.SetSelected(True)

   274

            elif self.Mode == MODE_COMMENT:

   275

                bbox = self.rubberBand.GetCurrentExtent()

   276

                self.rubberBand.OnLeftUp(event, self.GetLogicalDC(), self.Scaling)

   277

                wx.CallAfter(self.AddComment, bbox)

   278

        elif self.Mode == MODE_INITIALSTEP:

   279

            wx.CallAfter(self.AddInitialStep, GetScaledEventPosition(event, self.GetLogicalDC(), self.Scaling))

   280

        elif self.Mode == MODE_SELECTION and self.SelectedElement:

   281

            if self.IsWire(self.SelectedElement):

   282

                self.SelectedElement.SetSelectedSegment(0)

   283

            else:

   284

                self.SelectedElement.OnLeftUp(event, self.GetLogicalDC(), self.Scaling)

   285

                self.SelectedElement.Refresh()

   286

            wx.CallAfter(self.SetCurrentCursor, 0)

   287

        #

   288

        # FIXME:

   289

        # This code was forgotten by commit

   290

        # 9c74d00ce93e from plcopeneditor_history repository

   291

        # 'Last bugs on block and wire moving, resizing with cursor fixed'

   292

        #

   293

        # elif self.Mode == MODE_WIRE and self.SelectedElement:

   294

        #     self.SelectedElement.ResetPoints()

   295

        #     self.SelectedElement.OnMotion(event, self.GetLogicalDC(), self.Scaling)

   296

        #     self.SelectedElement.GeneratePoints()

   297

        #     self.SelectedElement.RefreshModel()

   298

        #     self.SelectedElement.SetSelected(True)

   299

        event.Skip()

   300

   301

    def OnViewerRightUp(self, event):

   302

        if self.GetDrawingMode() == FREEDRAWING_MODE:

   303

            Viewer.OnViewerRightUp(self, event)

   304

        else:

   305

            element = self.FindElement(event)

   306

            if element:

   307

                if self.SelectedElement and self.SelectedElement != element:

   308

                    self.SelectedElement.SetSelected(False)

   309

                self.SelectedElement = element

   310

                if self.IsWire(self.SelectedElement):

   311

                    self.SelectedElement.SetSelectedSegment(0)

   312

                else:

   313

                    self.SelectedElement.SetSelected(True)

   314

                    self.SelectedElement.OnRightUp(event, self.GetLogicalDC(), self.Scaling)

   315

                    self.SelectedElement.Refresh()

   316

                wx.CallAfter(self.SetCurrentCursor, 0)

   317

        event.Skip()

   318

   319

    def OnViewerLeftDClick(self, event):

   320

        if self.GetDrawingMode() == FREEDRAWING_MODE:

   321

            Viewer.OnViewerLeftDClick(self, event)

   322

        elif self.Mode == MODE_SELECTION and self.SelectedElement:

   323

            self.SelectedElement.OnLeftDClick(event, self.GetLogicalDC(), self.Scaling)

   324

            self.Refresh(False)

   325

        event.Skip()

   326

   327

    def OnViewerMotion(self, event):

   328

        if self.GetDrawingMode() == FREEDRAWING_MODE:

   329

            Viewer.OnViewerMotion(self, event)

   330

        else:

   331

            if self.rubberBand.IsShown():

   332

                self.rubberBand.OnMotion(event, self.GetLogicalDC(), self.Scaling)

   333

            elif self.Mode == MODE_SELECTION and self.SelectedElement:

   334

                if not self.IsWire(self.SelectedElement) and not isinstance(self.SelectedElement, Graphic_Group):

   335

                    self.SelectedElement.OnMotion(event, self.GetLogicalDC(), self.Scaling)

   336

                    self.SelectedElement.Refresh()

   337

            #

   338

            # FIXME:

   339

            # This code was forgotten by commit

   340

            # 9c74d00ce93e from plcopeneditor_history repository

   341

            # 'Last bugs on block and wire moving, resizing with cursor fixed'

   342

            #

   343

            # elif self.Mode == MODE_WIRE and self.SelectedElement:

   344

            #     self.SelectedElement.ResetPoints()

   345

            #     self.SelectedElement.OnMotion(event, self.GetLogicalDC(), self.Scaling)

   346

            #     self.SelectedElement.GeneratePoints()

   347

            #     self.SelectedElement.Refresh()

   348

            self.UpdateScrollPos(event)

   349

        event.Skip()

   350

   351

    def GetBlockName(self, block):

   352

        blockName = block.__class__.__name__

   353

        if blockName == "SFC_Divergence":

   354

            blockName = block.Type

   355

        return blockName

   356

   357

    # This method check the IEC 61131-3 compatibility between two SFC blocks

   358

    def BlockCompatibility(self, startblock=None, endblock=None, direction=None):

   359

        if startblock is not None and endblock is not None and \

   360

           (isinstance(startblock, SFC_Objects) or isinstance(endblock, SFC_Objects)):

   361

            # Full "SFC_StandardRules" table would be symmetrical and

   362

            # to avoid duplicate records and minimize the table only upper part is defined.

   363

            if direction == SOUTH or direction == EAST:

   364

                startblock, endblock = endblock, startblock

   365

            start = self.GetBlockName(startblock)

   366

            end = self.GetBlockName(endblock)

   367

            for val in self.SFC_StandardRules[start]:

   368

                if end in val:

   369

                    return True

   370

            return False

   371

        return True

   372

   373

    # -------------------------------------------------------------------------------

   374

    #                          Keyboard event functions

   375

    # -------------------------------------------------------------------------------

   376

   377

    def OnChar(self, event):

   378

        if self.GetDrawingMode() == FREEDRAWING_MODE:

   379

            Viewer.OnChar(self, event)

   380

        else:

   381

            xpos, ypos = self.GetScrollPos(wx.HORIZONTAL), self.GetScrollPos(wx.VERTICAL)

   382

            xmax = self.GetScrollRange(wx.HORIZONTAL) - self.GetScrollThumb(wx.HORIZONTAL)

   383

            ymax = self.GetScrollRange(wx.VERTICAL) - self.GetScrollThumb(wx.VERTICAL)

   384

            keycode = event.GetKeyCode()

   385

            if self.Scaling:

   386

                scaling = self.Scaling

   387

            else:

   388

                scaling = (8, 8)

   389

            if keycode == wx.WXK_DELETE and self.SelectedElement:

   390

                self.SelectedElement.Delete()

   391

                self.SelectedElement = None

   392

                self.RefreshBuffer()

   393

                self.RefreshScrollBars()

   394

                self.Refresh(False)

   395

            elif keycode == wx.WXK_LEFT:

   396

                if event.ControlDown() and event.ShiftDown():

   397

                    self.Scroll(0, ypos)

   398

                elif event.ControlDown():

   399

                    event.Skip()

   400

                elif self.SelectedElement:

   401

                    self.SelectedElement.Move(-scaling[0], 0)

   402

                    self.SelectedElement.RefreshModel()

   403

                    self.RefreshBuffer()

   404

                    self.RefreshScrollBars()

   405

                    self.RefreshRect(self.GetScrolledRect(self.SelectedElement.GetRedrawRect(-scaling[0], 0)), False)

   406

            elif keycode == wx.WXK_RIGHT:

   407

                if event.ControlDown() and event.ShiftDown():

   408

                    self.Scroll(xmax, ypos)

   409

                elif event.ControlDown():

   410

                    event.Skip()

   411

                elif self.SelectedElement:

   412

                    self.SelectedElement.Move(scaling[0], 0)

   413

                    self.SelectedElement.RefreshModel()

   414

                    self.RefreshBuffer()

   415

                    self.RefreshScrollBars()

   416

                    self.RefreshRect(self.GetScrolledRect(self.SelectedElement.GetRedrawRect(scaling[0], 0)), False)

   417

            elif keycode == wx.WXK_UP:

   418

                if event.ControlDown() and event.ShiftDown():

   419

                    self.Scroll(xpos, 0)

   420

                elif event.ControlDown():

   421

                    event.Skip()

   422

                elif self.SelectedElement:

   423

                    self.SelectedElement.Move(0, -scaling[1])

   424

                    self.SelectedElement.RefreshModel()

   425

                    self.RefreshBuffer()

   426

                    self.RefreshScrollBars()

   427

                    self.RefreshRect(self.GetScrolledRect(self.SelectedElement.GetRedrawRect(0, -scaling[1])), False)

   428

            elif keycode == wx.WXK_DOWN:

   429

                if event.ControlDown() and event.ShiftDown():

   430

                    self.Scroll(xpos, ymax)

   431

                elif event.ControlDown():

   432

                    event.Skip()

   433

                elif self.SelectedElement:

   434

                    self.SelectedElement.Move(0, scaling[1])

   435

                    self.SelectedElement.RefreshModel()

   436

                    self.RefreshBuffer()

   437

                    self.RefreshScrollBars()

   438

                    self.RefreshRect(self.GetScrolledRect(self.SelectedElement.GetRedrawRect(0, scaling[1])), False)

   439

            else:

   440

                event.Skip()

   441

   442

    # -------------------------------------------------------------------------------

   443

    #                          Adding element functions

   444

    # -------------------------------------------------------------------------------

   445

   446

    def AddInitialStep(self, pos):

   447

        dialog = SFCStepNameDialog(self.ParentWindow, _("Please enter step name"), _("Add a new initial step"), "", wx.OK | wx.CANCEL)

   448

        dialog.SetPouNames(self.Controler.GetProjectPouNames(self.Debug))

   449

        dialog.SetVariables(self.Controler.GetEditedElementInterfaceVars(self.TagName, debug=self.Debug))

   450

        dialog.SetStepNames([block.GetName() for block in self.Blocks if isinstance(block, SFC_Step)])

   451

        if dialog.ShowModal() == wx.ID_OK:

   452

            id = self.GetNewId()

   453

            name = dialog.GetValue()

   454

            step = SFC_Step(self, name, True, id)

   455

            min_width, min_height = step.GetMinSize()

   456

            step.SetPosition(pos.x, pos.y)

   457

            width, height = step.GetSize()

   458

            step.SetSize(max(min_width, width), max(min_height, height))

   459

            self.AddBlock(step)

   460

            self.Controler.AddEditedElementStep(self.TagName, id)

   461

            self.RefreshStepModel(step)

   462

            self.RefreshBuffer()

   463

            self.RefreshScrollBars()

   464

            self.Refresh(False)

   465

        dialog.Destroy()

   466

   467

    def AddStep(self):

   468

        if self.SelectedElement in self.Wires or isinstance(self.SelectedElement, SFC_Step):

   469

            dialog = SFCStepNameDialog(self.ParentWindow, _("Add a new step"), _("Please enter step name"), "", wx.OK | wx.CANCEL)

   470

            dialog.SetPouNames(self.Controler.GetProjectPouNames(self.Debug))

   471

            dialog.SetVariables(self.Controler.GetEditedElementInterfaceVars(self.TagName, debug=self.Debug))

   472

            dialog.SetStepNames([block.GetName() for block in self.Blocks if isinstance(block, SFC_Step)])

   473

            if dialog.ShowModal() == wx.ID_OK:

   474

                name = dialog.GetValue()

   475

                if self.IsWire(self.SelectedElement):

   476

                    self.SelectedElement.SetSelectedSegment(None)

   477

                    previous = self.SelectedElement.EndConnected

   478

                    next = self.SelectedElement.StartConnected

   479

                    self.SelectedElement.Clean()

   480

                    self.RemoveWire(self.SelectedElement)

   481

                else:

   482

                    connectors = self.SelectedElement.GetConnectors()

   483

                    if connectors["output"]:

   484

                        previous = connectors["output"]

   485

                        wires = previous.GetWires()

   486

                        if len(wires) != 1:

   487

                            return

   488

                        wire = wires[0][0]

   489

                        next = wire.StartConnected

   490

                        wire.Clean()

   491

                        self.RemoveWire(wire)

   492

                    else:

   493

                        self.SelectedElement.AddOutput()

   494

                        connectors = self.SelectedElement.GetConnectors()

   495

                        self.RefreshStepModel(self.SelectedElement)

   496

                        previous = connectors["output"]

   497

                        next = None

   498

                previous_block = previous.GetParentBlock()

   499

                if isinstance(previous_block, SFC_Step) or isinstance(previous_block, SFC_Divergence) and previous_block.GetType() in [SELECTION_DIVERGENCE, SELECTION_CONVERGENCE]:

   500

                    transition = self.CreateTransition(previous)

   501

                    transition_connectors = transition.GetConnectors()

   502

                    step = self.CreateStep(name, transition_connectors["output"], next)

   503

                else:

   504

                    step = self.CreateStep(name, previous)

   505

                    step.AddOutput()

   506

                    step.RefreshModel()

   507

                    step_connectors = step.GetConnectors()

   508

                    transition = self.CreateTransition(step_connectors["output"], next)

   509

                if self.IsWire(self.SelectedElement):

   510

                    self.SelectedElement = wire

   511

                    self.SelectedElement.SetSelectedSegment(0)

   512

                else:

   513

                    self.SelectedElement.SetSelected(False)

   514

                    self.SelectedElement = step

   515

                    self.SelectedElement.SetSelected(True)

   516