27

from graphics.GraphicCommons import CONNECTOR, CONTINUATION

    28

from graphics.FBD_Objects import FBD_Connector

    29

from BlockPreviewDialog import BlockPreviewDialog

    32

#                       Set Connection Parameters Dialog

    35

class ConnectionDialog(BlockPreviewDialog):

    37

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

    38

        """

    39

        Constructor

    40

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

    41

        @param controller: Reference to project controller

    42

        @param tagname: Tagname of project POU edited

    43

        @param apply_button: Enable button for applying connector modification

    44

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

    45

        """

    46

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

    49

        # Create dialog main sizer

    54

        # Create a sizer for dividing FBD connection parameters in two columns

    59

        # Create a sizer for left column

    65

        # Create label for connection type

    69

        # Create radio buttons for selecting connection type

    70

        self.ConnectionRadioButtons = {}

    71

        first = True

    72

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

    73

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

    74

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

    75

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

    76

            radio_button.SetValue(first)

    77

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

    78

            left_gridsizer.AddWindow(radio_button, flag=wx.GROW)

    79

            self.ConnectionRadioButtons[type] = radio_button

    80

            first = False

    81

    82

        # Create label for connection name

    86

        # Create text control for defining connection name

    91

        # Create a sizer for right column

    98

        # Add preview panel and associated label to sizers

    99

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

   102

        # Add buttons sizer to sizers

   103

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

   106

        # Add button for applying connection name modification to all connection

   107

        # of POU

   113

            self.ButtonSizer.AddWindow(self.ApplyToAllButton, border=10,

   114

                    flag=wx.LEFT)

   115

        else:

   116

            self.ConnectionName.ChangeValue(

   117

                controller.GenerateNewName(

   118

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

   122

        # Connector radio button is default control having keyboard focus

   123

        self.ConnectionRadioButtons[CONNECTOR].SetFocus()

   126

        """

   127

        Set default connection parameters

   128

        @param values: Connection parameters values

   129

        """

   130

        # For each parameters defined, set corresponding control value

   132

   133

            # Parameter is connection type

   135

                self.ConnectionRadioButtons[value].SetValue(True)

   136

   137

            # Parameter is connection name

   140

   141

        # Refresh preview panel

   145

        """

   146

        Return connection parameters defined in dialog

   147

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

   148

        """

   149

        values = {

   150

            "type": (CONNECTOR 

   151

                     if self.ConnectionRadioButtons[CONNECTOR].GetValue()

   152

                     else CONTINUATION),

   153

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

   154

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

   157

    def TestConnectionName(self):

   158

        """

   159

        Test that connection name is valid

   160

        @return: True if connection name is valid

   161

        """

   163

   164

        # Get connection name typed by user

   166

   167

        # Test that a name have been defined

   170

   171

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

   173

            self.ShowErrorMessage(message)

   174

            # Test failed

   176

   177

        # Return result of element name test

   178

        return self.TestElementName(connection_name)

   181

        """

   182

        Called when dialog OK button is pressed

   183

        Test if connection name is valid

   184

        @param event: wx.Event from OK button

   185

        """

   186

        # Close dialog if connection name is valid

   187

        if self.TestConnectionName():

   191

        """

   192

        Called when Apply To All button is pressed

   193

        Test if connection name is valid

   194

        @param event: wx.Event from OK button

   195

        """

   196

        # Close dialog if connection name is valid

   197

        if self.TestConnectionName():

   201

        """

   202

        Called when connection type changed

   203

        @param event: wx.RadioButtonEvent

   204

        """

   209

        """

   210

        Called when connection name value changed

   211

        @param event: wx.TextEvent

   212

        """

   217

        """

   218

        Refresh preview panel of graphic element

   219

        Override BlockPreviewDialog function

   220

        """

   221

        # Set graphic element displayed, creating a FBD connection element

   222

        self.Element = FBD_Connector(self.Preview, 

   223

                (CONNECTOR

   224

                 if self.ConnectionRadioButtons[CONNECTOR].GetValue()

   225

                 else CONTINUATION),

   226

                self.ConnectionName.GetValue())

   227

   228

        # Call BlockPreviewDialog function

   229

        BlockPreviewDialog.RefreshPreview(self)

   230