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

# Copyright (C) 2017: Andrey Skvortsov <andrej.skvortzov@gmail.com>

     9

#

    10

# See COPYING file for copyrights details.

    11

#

    12

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

    13

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

    14

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

    15

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

    16

#

    17

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

    18

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

    19

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

    20

# GNU General Public License for more details.

    21

#

    22

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

    23

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

    24

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

    25

    26

import wx

    27

    28

from graphics.GraphicCommons import CONNECTOR, CONTINUATION

    29

from graphics.FBD_Objects import FBD_Connector

    30

from BlockPreviewDialog import BlockPreviewDialog

    31

    32

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

    33

#                       Set Connection Parameters Dialog

    34

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

    35

    36

"""

    37

Class that implements a dialog for defining parameters of a connection graphic

    38

element

    39

"""

    40

    41

class ConnectionDialog(BlockPreviewDialog):

    42

    43

    def __init__(self, parent, controller, tagname, apply_button=False):

    44

        """

    45

        Constructor

    46

        @param parent: Parent wx.Window of dialog for modal

    47

        @param controller: Reference to project controller

    48

        @param tagname: Tagname of project POU edited

    49

        @param apply_button: Enable button for applying connector modification

    50

        to all connector having the same name in POU (default: False)

    51

        """

    52

        BlockPreviewDialog.__init__(self, parent, controller, tagname, 

    53

              title=_('Connection Properties'))

    54

    55

        # Init common sizers

    56

        self._init_sizers(2, 0, 7, 1, 0, None)

    57

    58

        # Create label for connection type

    59

        type_label = wx.StaticText(self, label=_('Type:'))

    60

        self.LeftGridSizer.AddWindow(type_label, flag=wx.GROW)

    61

    62

        # Create radio buttons for selecting connection type

    63

        self.TypeRadioButtons = {}

    64

        first = True

    65

        for type, label in [(CONNECTOR, _('Connector')),

    66

                            (CONTINUATION, _('Continuation'))]:

    67

            radio_button = wx.RadioButton(self, label=label, 

    68

                  style=(wx.RB_GROUP if first else 0))

    69

            radio_button.SetValue(first)

    70

            self.Bind(wx.EVT_RADIOBUTTON, self.OnTypeChanged, radio_button)

    71

            self.LeftGridSizer.AddWindow(radio_button, flag=wx.GROW)

    72

            self.TypeRadioButtons[type] = radio_button

    73

            first = False

    74

    75

        # Create label for connection name

    76

        name_label = wx.StaticText(self, label=_('Name:'))

    77

        self.LeftGridSizer.AddWindow(name_label, flag=wx.GROW)

    78

    79

        # Create text control for defining connection name

    80

        self.ConnectionName = wx.TextCtrl(self)

    81

        self.ConnectionName.SetMinSize(wx.Size(200,-1))

    82

        self.Bind(wx.EVT_TEXT, self.OnNameChanged, self.ConnectionName)

    83

        self.LeftGridSizer.AddWindow(self.ConnectionName, flag=wx.GROW)

    84

    85

        # Add preview panel and associated label to sizers

    86

        self.Preview.SetMinSize(wx.Size(-1,100))

    87

        self.LeftGridSizer.AddWindow(self.PreviewLabel, flag=wx.GROW)

    88

        self.LeftGridSizer.AddWindow(self.Preview, flag=wx.GROW)

    89

    90

        # Add buttons sizer to sizers

    91

        self.MainSizer.AddSizer(self.ButtonSizer, border=20, 

    92

              flag=wx.ALIGN_RIGHT|wx.BOTTOM|wx.LEFT|wx.RIGHT)

    93

        self.ColumnSizer.RemoveSizer(self.RightGridSizer)

    94

    95

        # Add button for applying connection name modification to all connection

    96

        # of POU

    97

        if apply_button:

    98

            self.ApplyToAllButton = wx.Button(self, label=_("Propagate Name"))

    99

            self.ApplyToAllButton.SetToolTipString(

   100

                _("Apply name modification to all continuations with the same name"))

   101

            self.Bind(wx.EVT_BUTTON, self.OnApplyToAll, self.ApplyToAllButton)

   102

            self.ButtonSizer.AddWindow(self.ApplyToAllButton, flag=wx.LEFT)

   103

        else:

   104

            self.ConnectionName.ChangeValue(

   105

                controller.GenerateNewName(

   106

                        tagname, None, "Connection%d", 0))

   107

        self.Fit()

   108

   109

        # Connector radio button is default control having keyboard focus

   110

        self.TypeRadioButtons[CONNECTOR].SetFocus()

   111

   112

    def GetConnectionType(self):

   113

        """

   114

        Return type selected for connection

   115

        @return: Type selected (CONNECTOR or CONTINUATION)

   116

        """

   117

        return (CONNECTOR

   118

                if self.TypeRadioButtons[CONNECTOR].GetValue()

   119

                else CONTINUATION)

   120

   121

    def SetValues(self, values):

   122

        """

   123

        Set default connection parameters

   124

        @param values: Connection parameters values

   125

        """

   126

        # For each parameters defined, set corresponding control value

   127

        for name, value in values.items():

   128

   129

            # Parameter is connection type

   130

            if name == "type":

   131

                self.TypeRadioButtons[value].SetValue(True)

   132

   133

            # Parameter is connection name

   134

            elif name == "name":

   135

                self.ConnectionName.SetValue(value)

   136

   137

        # Refresh preview panel

   138

        self.RefreshPreview()

   139

   140

    def GetValues(self):

   141

        """

   142

        Return connection parameters defined in dialog

   143

        @return: {parameter_name: parameter_value,...}

   144

        """

   145

        values = {

   146

            "type": self.GetConnectionType(),

   147

            "name": self.ConnectionName.GetValue()}

   148

        values["width"], values["height"] = self.Element.GetSize()

   149

        return values

   150

   151

    def TestConnectionName(self):

   152

        """

   153

        Test that connection name is valid

   154

        @return: True if connection name is valid

   155

        """

   156

        message = None

   157

   158

        # Get connection name typed by user

   159

        connection_name = self.ConnectionName.GetValue()

   160

   161

        # Test that a name have been defined

   162

        if connection_name == "":

   163

            message = _("Form isn't complete. Name must be filled!")

   164

   165

        # If an error have been identify, show error message dialog

   166

        if message is not None:

   167

            self.ShowErrorMessage(message)

   168

            # Test failed

   169

            return False

   170

   171

        # Return result of element name test

   172

        return self.TestElementName(connection_name)

   173

   174

    def OnOK(self, event):

   175

        """

   176

        Called when dialog OK button is pressed

   177

        Test if connection name is valid

   178

        @param event: wx.Event from OK button

   179

        """

   180

        # Close dialog if connection name is valid

   181

        if self.TestConnectionName():

   182

            self.EndModal(wx.ID_OK)

   183

   184

    def OnApplyToAll(self, event):

   185

        """

   186

        Called when Apply To All button is pressed

   187

        Test if connection name is valid

   188

        @param event: wx.Event from OK button

   189

        """

   190

        # Close dialog if connection name is valid

   191

        if self.TestConnectionName():

   192

            self.EndModal(wx.ID_YESTOALL)

   193

   194

    def OnTypeChanged(self, event):

   195

        """

   196

        Called when connection type changed

   197

        @param event: wx.RadioButtonEvent

   198

        """

   199

        self.RefreshPreview()

   200

        event.Skip()

   201

   202

    def OnNameChanged(self, event):

   203

        """

   204

        Called when connection name value changed

   205

        @param event: wx.TextEvent

   206

        """

   207

        self.RefreshPreview()

   208

        event.Skip()

   209

   210

    def RefreshPreview(self):

   211

        """

   212

        Refresh preview panel of graphic element

   213

        Override BlockPreviewDialog function

   214

        """

   215

        # Set graphic element displayed, creating a FBD connection element

   216

        self.Element = FBD_Connector(self.Preview, 

   217

                self.GetConnectionType(),

   218

                self.ConnectionName.GetValue())

   219

   220

        # Call BlockPreviewDialog function

   221

        BlockPreviewDialog.RefreshPreview(self)

   222