1

#!/usr/bin/env python

     2

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

     3

     4

#This file is part of PLCOpenEditor, a library implementing an IEC 61131-3 editor

     5

#based on the plcopen standard. 

     6

#

     7

#Copyright (C): Edouard TISSERANT and Laurent BESSARD

     8

#

     9

#See COPYING file for copyrights details.

    10

#

    11

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

    12

#modify it under the terms of the GNU General Public

    13

#License as published by the Free Software Foundation; either

    14

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

    15

#

    16

#This library 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 GNU

    19

#Lesser General Public License for more details.

    20

#

    21

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

    22

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

    23

#Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

    24

    25

from wxPython.wx import *

    26

from wxPython.grid import *

    27

import wx

    28

    29

from graphics.FBD_Objects import *

    30

from graphics.LD_Objects import *

    31

from graphics.SFC_Objects import *

    32

    33

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

    34

#                          Create New Block Dialog

    35

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

    36

    37

[wxID_BLOCKPROPERTIESDIALOG, wxID_BLOCKPROPERTIESDIALOGMAINPANEL, 

    38

 wxID_BLOCKPROPERTIESDIALOGNAME, wxID_BLOCKPROPERTIESDIALOGTYPETREE, 

    39

 wxID_BLOCKPROPERTIESDIALOGTYPEDESC, wxID_BLOCKPROPERTIESDIALOGINPUTS, 

    40

 wxID_BLOCKPROPERTIESDIALOGPREVIEW, wxID_BLOCKPROPERTIESDIALOGSTATICTEXT1, 

    41

 wxID_BLOCKPROPERTIESDIALOGSTATICTEXT2, wxID_BLOCKPROPERTIESDIALOGSTATICTEXT3, 

    42

 wxID_BLOCKPROPERTIESDIALOGSTATICTEXT4, 

    43

] = [wx.NewId() for _init_ctrls in range(11)]

    44

    45

[CATEGORY, BLOCK] = range(2)

    46

    47

class BlockPropertiesDialog(wx.Dialog):

    48

    def _init_coll_flexGridSizer1_Items(self, parent):

    49

        # generated method, don't edit

    50

    51

        parent.AddWindow(self.MainPanel, 0, border=0, flag=0)

    52

    53

    def _init_sizers(self):

    54

        # generated method, don't edit

    55

        self.flexGridSizer1 = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=0)

    56

    57

        self._init_coll_flexGridSizer1_Items(self.flexGridSizer1)

    58

    59

        self.SetSizer(self.flexGridSizer1)

    60

    61

    def _init_ctrls(self, prnt):

    62

        # generated method, don't edit

    63

        wx.Dialog.__init__(self, id=wxID_BLOCKPROPERTIESDIALOG,

    64

              name='BlockPropertiesDialog', parent=prnt, pos=wx.Point(376, 223),

    65

              size=wx.Size(600, 360), style=wx.DEFAULT_DIALOG_STYLE,

    66

              title='Block Properties')

    67

        self.SetClientSize(wx.Size(600, 360))

    68

    69

        self.MainPanel = wx.Panel(id=wxID_BLOCKPROPERTIESDIALOGMAINPANEL,

    70

              name='MainPanel', parent=self, pos=wx.Point(0, 0),

    71

              size=wx.Size(600, 320), style=wx.TAB_TRAVERSAL)

    72

        self.MainPanel.SetAutoLayout(True)

    73

    74

        self.staticbox1 = wx.StaticBox(id=wxID_BLOCKPROPERTIESDIALOGSTATICTEXT1,

    75

              label='Type:', name='staticBox1', parent=self.MainPanel,

    76

              pos=wx.Point(24, 24), size=wx.Size(245, 280), style=0)

    77

    78

        self.staticText2 = wx.StaticText(id=wxID_BLOCKPROPERTIESDIALOGSTATICTEXT2,

    79

              label='Name:', name='staticText2', parent=self.MainPanel,

    80

              pos=wx.Point(274, 24), size=wx.Size(70, 17), style=0)

    81

    82

        self.staticText3 = wx.StaticText(id=wxID_BLOCKPROPERTIESDIALOGSTATICTEXT2,

    83

              label='Inputs:', name='staticText4', parent=self.MainPanel,

    84

              pos=wx.Point(424, 24), size=wx.Size(70, 17), style=0)

    85

    86

        self.staticText4 = wx.StaticText(id=wxID_BLOCKPROPERTIESDIALOGSTATICTEXT4,

    87

              label='Preview:', name='staticText4', parent=self.MainPanel,

    88

              pos=wx.Point(274, 80), size=wx.Size(100, 17), style=0)

    89

    90

        self.TypeTree = wx.TreeCtrl(id=wxID_BLOCKPROPERTIESDIALOGTYPETREE,

    91

              name='TypeTree', parent=self.MainPanel, pos=wx.Point(34, 44),

    92

              size=wx.Size(225, 180), style=wx.TR_HAS_BUTTONS|wx.TR_HIDE_ROOT|wx.TR_SINGLE|wx.SUNKEN_BORDER)

    93

        EVT_TREE_SEL_CHANGED(self, wxID_BLOCKPROPERTIESDIALOGTYPETREE, self.OnTypeTreeItemSelected)

    94

    95

        self.TypeDesc = wx.TextCtrl(id=wxID_BLOCKPROPERTIESDIALOGTYPEDESC,

    96

              name='TypeDesc', parent=self.MainPanel, pos=wx.Point(34, 230),

    97

              size=wx.Size(225, 65), style=wx.TE_READONLY|wx.TE_MULTILINE)

    98

    99

        self.Name = wx.TextCtrl(id=wxID_BLOCKPROPERTIESDIALOGNAME, value='',

   100

              name='Name', parent=self.MainPanel, pos=wx.Point(274, 48),

   101

              size=wx.Size(145, 24), style=0)

   102

        EVT_TEXT(self, wxID_BLOCKPROPERTIESDIALOGNAME, self.OnNameChanged)

   103

   104

        self.Inputs = wx.SpinCtrl(id=wxID_BLOCKPROPERTIESDIALOGINPUTS,

   105

              name='Inputs', parent=self.MainPanel, pos=wx.Point(424, 48),

   106

              size=wx.Size(145, 24), style=wxSP_ARROW_KEYS, min=2, max=20)

   107

        EVT_SPINCTRL(self, wxID_BLOCKPROPERTIESDIALOGINPUTS, self.OnInputsChanged)

   108

   109

        self.Preview = wx.Panel(id=wxID_BLOCKPROPERTIESDIALOGPREVIEW,

   110

              name='Preview', parent=self.MainPanel, pos=wx.Point(274, 104),

   111

              size=wx.Size(300, 200), style=wx.TAB_TRAVERSAL|wx.SIMPLE_BORDER)

   112

        self.Preview.SetBackgroundColour(wxColour(255,255,255))

   113

   114

        self._init_sizers()

   115

   116

    def __init__(self, parent):

   117

        self._init_ctrls(parent)

   118

        self.ButtonSizer = self.CreateButtonSizer(wxOK|wxCANCEL|wxCENTRE)

   119

        self.flexGridSizer1.Add(self.ButtonSizer, 1, wxALIGN_RIGHT)

   120

        self.Name.SetValue("")

   121

        self.Name.Enable(False)

   122

        self.Inputs.Enable(False)

   123

        self.Block = None

   124

        self.MinBlockSize = None

   125

   126

        EVT_PAINT(self, self.OnPaint)

   127

        EVT_BUTTON(self, self.ButtonSizer.GetAffirmativeButton().GetId(), self.OnOK)

   128

   129

    def FindTreeItem(self, root, name):

   130

        if root.IsOk():

   131

            if self.TypeTree.GetItemText(root) == name:

   132

                return root

   133

            else:

   134

                item, root_cookie = self.TypeTree.GetFirstChild(root)

   135

                while item.IsOk():

   136

                    result = self.FindTreeItem(item, name)

   137

                    if result:

   138

                        return result

   139

                    item, root_cookie = self.TypeTree.GetNextChild(root, root_cookie)

   140

        return None

   141

   142

    def OnOK(self, event):

   143

        error = []

   144

        selected = self.TypeTree.GetSelection()

   145

        if not selected.IsOk() or self.TypeTree.GetItemParent(selected) == self.TypeTree.GetRootItem() or selected == self.TypeTree.GetRootItem():

   146

            message = wxMessageDialog(self, "Form isn't complete. Valid block type must be selected!", "Error", wxOK|wxICON_ERROR)

   147

            message.ShowModal()

   148

            message.Destroy()

   149

        elif self.Name.IsEnabled() and self.Name.GetValue() == "":

   150

            message = wxMessageDialog(self, "Form isn't complete. Name must be filled!", "Error", wxOK|wxICON_ERROR)

   151

            message.ShowModal()

   152

            message.Destroy()

   153

        else:

   154

            self.EndModal(wxID_OK)

   155

   156

    def SetBlockList(self, blocktypes):

   157

        root = self.TypeTree.AddRoot("")

   158

        self.TypeTree.SetPyData(root, CATEGORY)

   159

        for category in blocktypes:

   160

            category_item = self.TypeTree.AppendItem(root, category["name"])

   161

            self.TypeTree.SetPyData(category_item, CATEGORY)

   162

            for blocktype in category["list"]:

   163

                blocktype_item = self.TypeTree.AppendItem(category_item, blocktype["name"])

   164

                self.TypeTree.SetPyData(blocktype_item, BLOCK)

   165

   166

    def SetMinBlockSize(self, size):

   167

        self.MinBlockSize = size

   168

   169

    def SetValues(self, values):

   170

        for name, value in values.items():

   171

            if name == "type":

   172

                item = self.FindTreeItem(self.TypeTree.GetRootItem(), value)

   173

                if item:

   174

                    self.TypeTree.SelectItem(item)

   175

            elif name == "name":

   176

                self.Name.SetValue(value)

   177

            elif name == "extension":

   178

                self.Inputs.SetValue(value)

   179

        self.RefreshPreview()

   180

   181

    def GetValues(self):

   182

        values = {}

   183

        values["type"] = self.TypeTree.GetItemText(self.TypeTree.GetSelection())

   184

        if self.Name.GetValue() != "":

   185

            values["name"] = self.Name.GetValue()

   186

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

   187

        values["extension"] = self.Inputs.GetValue()

   188

        return values

   189

   190

    def OnTypeTreeItemSelected(self, event):

   191

        self.Name.SetValue("")

   192

        selected = event.GetItem()

   193

        if self.TypeTree.GetPyData(selected) != CATEGORY:

   194

            blocktype = GetBlockType(self.TypeTree.GetItemText(selected))

   195

            if blocktype:

   196

                self.Inputs.SetValue(len(blocktype["inputs"]))

   197

                self.Inputs.Enable(blocktype["extensible"])

   198

                self.Name.Enable(blocktype["type"] != "function")

   199

                self.TypeDesc.SetValue(blocktype["comment"])

   200

                wxCallAfter(self.RefreshPreview)

   201

            else:

   202

                self.Name.Enable(False)

   203

                self.Inputs.Enable(False)

   204

                self.Inputs.SetValue(2)

   205

                self.TypeDesc.SetValue("")

   206

                wxCallAfter(self.ErasePreview)

   207

        else:

   208

            self.Name.Enable(False)

   209

            self.Inputs.Enable(False)

   210

            self.Inputs.SetValue(2)

   211

            self.TypeDesc.SetValue("")

   212

            wxCallAfter(self.ErasePreview)

   213

        event.Skip()

   214

   215

    def OnNameChanged(self, event):

   216

        if self.Name.IsEnabled():

   217

            self.RefreshPreview()

   218

        event.Skip()

   219

   220

    def OnInputsChanged(self, event):

   221

        if self.Inputs.IsEnabled():

   222

            self.RefreshPreview()

   223

        event.Skip()

   224

   225

    def ErasePreview(self):

   226

        dc = wxClientDC(self.Preview)

   227

        dc.Clear()

   228

        self.Block = None

   229

   230

    def RefreshPreview(self):

   231

        dc = wxClientDC(self.Preview)

   232

        dc.Clear()

   233

        item = self.TypeTree.GetSelection()

   234

        if self.TypeTree.GetPyData(item) == CATEGORY:

   235

            self.Block = None

   236

        else:

   237

            blocktype = self.TypeTree.GetItemText(item)

   238

            if blocktype:

   239

                self.Block = FBD_Block(self.Preview, blocktype, self.Name.GetValue(), extension = self.Inputs.GetValue())

   240

                width, height = self.MinBlockSize

   241

                min_width, min_height = self.Block.GetMinSize()

   242

                width, height = max(min_width, width), max(min_height, height)

   243

                self.Block.SetSize(width, height)

   244

                clientsize = self.Preview.GetClientSize()

   245

                x = (clientsize.width - width) / 2

   246

                y = (clientsize.height - height) / 2

   247

                self.Block.SetPosition(x, y)

   248

                self.Block.Draw(dc)

   249

            else:

   250

                self.Block = None

   251

   252

    def OnPaint(self, event):

   253

        if self.Block:

   254

            self.RefreshPreview()

   255

   256

   257

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

   258

#                          Create New Variable Dialog

   259

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

   260

   261

[wxID_VARIABLEPROPERTIESDIALOG, wxID_VARIABLEPROPERTIESDIALOGMAINPANEL, 

   262

 wxID_VARIABLEPROPERTIESDIALOGNAME, wxID_VARIABLEPROPERTIESDIALOGCLASS, 

   263

 wxID_VARIABLEPROPERTIESDIALOGPREVIEW, wxID_VARIABLEPROPERTIESDIALOGSTATICTEXT1,

   264

 wxID_VARIABLEPROPERTIESDIALOGSTATICTEXT2, wxID_VARIABLEPROPERTIESDIALOGSTATICTEXT3, 

   265

] = [wx.NewId() for _init_ctrls in range(8)]

   266

   267

class VariablePropertiesDialog(wx.Dialog):

   268

    def _init_coll_flexGridSizer1_Items(self, parent):

   269

        # generated method, don't edit

   270

   271

        parent.AddWindow(self.MainPanel, 0, border=0, flag=0)

   272

   273

    def _init_sizers(self):

   274

        # generated method, don't edit

   275

        self.flexGridSizer1 = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=0)

   276

   277

        self._init_coll_flexGridSizer1_Items(self.flexGridSizer1)

   278

   279

        self.SetSizer(self.flexGridSizer1)

   280

   281

    def _init_ctrls(self, prnt):

   282

        # generated method, don't edit

   283

        wx.Dialog.__init__(self, id=wxID_VARIABLEPROPERTIESDIALOG,

   284

              name='VariablePropertiesDialog', parent=prnt, pos=wx.Point(376, 223),

   285

              size=wx.Size(400, 320), style=wx.DEFAULT_DIALOG_STYLE,

   286

              title='Variable Properties')

   287

        self.SetClientSize(wx.Size(400, 320))

   288

   289

        self.MainPanel = wx.Panel(id=wxID_VARIABLEPROPERTIESDIALOGMAINPANEL,

   290

              name='MainPanel', parent=self, pos=wx.Point(0, 0),

   291

              size=wx.Size(400, 280), style=wx.TAB_TRAVERSAL)

   292

        self.MainPanel.SetAutoLayout(True)

   293

   294

        self.staticText1 = wx.StaticText(id=wxID_VARIABLEPROPERTIESDIALOGSTATICTEXT1,

   295

              label='Class:', name='staticText1', parent=self.MainPanel,

   296

              pos=wx.Point(24, 24), size=wx.Size(70, 17), style=0)

   297

   298

        self.staticText2 = wx.StaticText(id=wxID_VARIABLEPROPERTIESDIALOGSTATICTEXT2,

   299

              label='Name:', name='staticText2', parent=self.MainPanel,

   300

              pos=wx.Point(204, 24), size=wx.Size(70, 17), style=0)

   301

   302

        self.staticText3 = wx.StaticText(id=wxID_VARIABLEPROPERTIESDIALOGSTATICTEXT3,

   303

              label='Preview:', name='staticText3', parent=self.MainPanel,

   304

              pos=wx.Point(24, 78), size=wx.Size(100, 17), style=0)

   305

   306

        self.Class = wx.Choice(id=wxID_VARIABLEPROPERTIESDIALOGCLASS,

   307

              name='Class', parent=self.MainPanel, pos=wx.Point(24, 48),

   308

              size=wx.Size(145, 24), style=0)

   309

        EVT_CHOICE(self, wxID_VARIABLEPROPERTIESDIALOGCLASS, self.OnClassChanged)

   310

   311

        self.Name = wx.Choice(id=wxID_VARIABLEPROPERTIESDIALOGNAME,

   312

              name='Name', parent=self.MainPanel, pos=wx.Point(204, 48),

   313

              size=wx.Size(145, 24), style=0)

   314

        EVT_CHOICE(self, wxID_VARIABLEPROPERTIESDIALOGNAME, self.OnNameChanged)

   315

   316

        self.Preview = wx.Panel(id=wxID_VARIABLEPROPERTIESDIALOGPREVIEW,

   317

              name='Preview', parent=self.MainPanel, pos=wx.Point(24, 104),

   318

              size=wx.Size(350, 150), style=wx.TAB_TRAVERSAL|wx.SIMPLE_BORDER)

   319

        self.Preview.SetBackgroundColour(wxColour(255,255,255))

   320

   321

        self._init_sizers()

   322

   323

    def __init__(self, parent):

   324

        self._init_ctrls(parent)

   325

        self.ButtonSizer = self.CreateButtonSizer(wxOK|wxCANCEL|wxCENTRE)

   326

        self.flexGridSizer1.Add(self.ButtonSizer, 1, wxALIGN_RIGHT)

   327

        self.Variable = None

   328

        self.VarList = []

   329

        self.MinVariableSize = None

   330

        self.RefreshNameList()

   331

   332

        for choice in ["Input", "Output", "InOut"]:

   333

            self.Class.Append(choice)

   334

        self.Class.SetStringSelection("Input")

   335

   336

        EVT_PAINT(self, self.OnPaint)

   337

   338

    def RefreshNameList(self):

   339

        selected = self.Name.GetStringSelection()

   340

        self.Name.Clear()

   341

        for name, var_type, value_type in self.VarList:

   342

            if var_type in ["Local","Temp"]:

   343

                self.Name.Append(name)

   344

            elif var_type == "Input" and self.Class.GetStringSelection() == "Input":

   345

                self.Name.Append(name)

   346

            elif var_type == "Output" and self.Class.GetStringSelection() == "Output":

   347

                self.Name.Append(name)

   348

            elif var_type == "InOut" and self.Class.GetStringSelection() == "InOut":

   349

                self.Name.Append(name)

   350

        if self.Name.FindString(selected) != wxNOT_FOUND:

   351

            self.Name.SetStringSelection(selected)

   352

        self.Name.Enable(self.Name.GetCount() > 0)

   353

   354

    def SetMinVariableSize(self, size):

   355

        self.MinVariableSize = size

   356

   357

    def SetVariables(self, vars):

   358

        self.VarList = vars

   359

        self.RefreshNameList()

   360

   361

    def SetValues(self, values):

   362

        for name, value in values.items():

   363

            if name == "type":

   364

                if value == INPUT:

   365

                    self.Class.SetStringSelection("Input")

   366

                if value == OUTPUT:

   367

                    self.Class.SetStringSelection("Output")

   368

                if value == INOUT:

   369

                    self.Class.SetStringSelection("InOut")

   370

            elif name == "name":

   371

                self.Name.SetStringSelection(value)

   372

        self.RefreshPreview()

   373

   374

    def GetValues(self):

   375

        values = {}

   376

        classtype = self.Class.GetStringSelection()

   377

        if classtype == "Input":

   378

            values["type"] = INPUT

   379

        elif classtype == "Output":

   380

            values["type"] = OUTPUT

   381

        elif classtype == "InOut":

   382

            values["type"] = INOUT

   383

        values["name"] = self.Name.GetStringSelection()

   384

        values["value_type"] = ""

   385

        for var_name, var_type, value_type in self.VarList:

   386

            if var_name == values["name"]:

   387

                values["value_type"] = value_type

   388

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

   389

        return values

   390

   391

    def OnClassChanged(self, event):

   392

        self.RefreshNameList()

   393

        self.RefreshPreview()

   394

        event.Skip()

   395

   396

    def OnNameChanged(self, event):

   397

        self.RefreshPreview()

   398

        event.Skip()

   399

   400

    def RefreshPreview(self):

   401

        dc = wxClientDC(self.Preview)

   402

        dc.Clear()

   403

        name = self.Name.GetStringSelection()

   404

        type = ""

   405

        for var_name, var_type, value_type in self.VarList:

   406

            if var_name == name:

   407

                type = value_type

   408

        classtype = self.Class.GetStringSelection()

   409

        if classtype == "Input":

   410

            self.Variable = FBD_Variable(self.Preview, INPUT, name, type)

   411

        elif classtype == "Output":

   412

            self.Variable = FBD_Variable(self.Preview, OUTPUT, name, type)

   413

        elif classtype == "InOut":

   414

            self.Variable = FBD_Variable(self.Preview, INOUT, name, type)

   415

        width, height = self.MinVariableSize

   416

        min_width, min_height = self.Variable.GetMinSize()

   417

        width, height = max(min_width, width), max(min_height, height)

   418

        self.Variable.SetSize(width, height)

   419

        clientsize = self.Preview.GetClientSize()

   420

        x = (clientsize.width - width) / 2

   421

        y = (clientsize.height - height) / 2

   422

        self.Variable.SetPosition(x, y)

   423

        self.Variable.Draw(dc)

   424

   425

    def OnPaint(self, event):

   426

        self.RefreshPreview()

   427

   428

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

   429

#                          Create New Connection Dialog

   430

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

   431

   432

[wxID_CONNECTIONPROPERTIESDIALOG, wxID_CONNECTIONPROPERTIESDIALOGMAINPANEL, 

   433

 wxID_CONNECTIONPROPERTIESDIALOGNAME, wxID_CONNECTIONPROPERTIESDIALOGRADIOBUTTON1, 

   434

 wxID_CONNECTIONPROPERTIESDIALOGRADIOBUTTON2, wxID_CONNECTIONPROPERTIESDIALOGPREVIEW,

   435

 wxID_CONNECTIONPROPERTIESDIALOGSTATICTEXT1, wxID_CONNECTIONPROPERTIESDIALOGSTATICTEXT2, 

   436

 wxID_CONNECTIONPROPERTIESDIALOGSTATICTEXT3, 

   437

] = [wx.NewId() for _init_ctrls in range(9)]

   438

   439

class ConnectionPropertiesDialog(wx.Dialog):

   440

    def _init_coll_flexGridSizer1_Items(self, parent):

   441

        # generated method, don't edit

   442

   443

        parent.AddWindow(self.MainPanel, 0, border=0, flag=0)

   444

   445

    def _init_sizers(self):

   446

        # generated method, don't edit

   447

        self.flexGridSizer1 = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=0)

   448

   449

        self._init_coll_flexGridSizer1_Items(self.flexGridSizer1)

   450

   451

        self.SetSizer(self.flexGridSizer1)

   452

   453

    def _init_ctrls(self, prnt):

   454

        # generated method, don't edit

   455

        wx.Dialog.__init__(self, id=wxID_CONNECTIONPROPERTIESDIALOG,

   456

              name='ConnectionPropertiesDialog', parent=prnt, pos=wx.Point(376, 223),

   457

              size=wx.Size(350, 220), style=wx.DEFAULT_DIALOG_STYLE,

   458

              title='Connection Properties')

   459

        self.SetClientSize(wx.Size(350, 220))

   460

   461

        self.MainPanel = wx.Panel(id=wxID_CONNECTIONPROPERTIESDIALOGMAINPANEL,

   462

              name='MainPanel', parent=self, pos=wx.Point(0, 0),

   463

              size=wx.Size(340, 360), style=wx.TAB_TRAVERSAL)

   464

        self.MainPanel.SetAutoLayout(True)

   465

   466

        self.staticText1 = wx.StaticText(id=wxID_CONNECTIONPROPERTIESDIALOGSTATICTEXT1,

   467

              label='Type:', name='staticText1', parent=self.MainPanel,

   468

              pos=wx.Point(24, 24), size=wx.Size(70, 17), style=0)

   469

   470

        self.staticText2 = wx.StaticText(id=wxID_CONNECTIONPROPERTIESDIALOGSTATICTEXT2,

   471

              label='Name:', name='staticText2', parent=self.MainPanel,

   472

              pos=wx.Point(24, 104), size=wx.Size(70, 17), style=0)

   473

   474

        self.staticText3 = wx.StaticText(id=wxID_CONNECTIONPROPERTIESDIALOGSTATICTEXT3,

   475

              label='Preview:', name='staticText3', parent=self.MainPanel,

   476

              pos=wx.Point(174, 24), size=wx.Size(100, 17), style=0)

   477

   478

        self.radioButton1 = wx.RadioButton(id=wxID_CONNECTIONPROPERTIESDIALOGRADIOBUTTON1,

   479

              label='Connector', name='radioButton1', parent=self.MainPanel,

   480

              pos=wx.Point(24, 48), size=wx.Size(114, 24), style=0)

   481

        EVT_RADIOBUTTON(self, wxID_CONNECTIONPROPERTIESDIALOGRADIOBUTTON1, self.OnTypeChanged)

   482

        self.radioButton1.SetValue(True)

   483

   484

        self.radioButton2 = wx.RadioButton(id=wxID_CONNECTIONPROPERTIESDIALOGRADIOBUTTON2,

   485

              label='Continuation', name='radioButton2', parent=self.MainPanel, 

   486

              pos=wx.Point(24, 72), size=wx.Size(128, 24), style=0)

   487

        EVT_RADIOBUTTON(self, wxID_CONNECTIONPROPERTIESDIALOGRADIOBUTTON2, self.OnTypeChanged)

   488

        self.radioButton2.SetValue(False)

   489

   490

        self.Name = wx.TextCtrl(id=wxID_CONNECTIONPROPERTIESDIALOGNAME,

   491

              name='Name', parent=self.MainPanel, pos=wx.Point(24, 130),

   492

              size=wx.Size(145, 24), style=0)

   493

        EVT_TEXT(self, wxID_CONNECTIONPROPERTIESDIALOGNAME, self.OnNameChanged)

   494

   495

        self.Preview = wx.Panel(id=wxID_CONNECTIONPROPERTIESDIALOGPREVIEW,

   496

              name='Preview', parent=self.MainPanel, pos=wx.Point(174, 48),

   497

              size=wx.Size(150, 100), style=wx.TAB_TRAVERSAL|wx.SIMPLE_BORDER)

   498

        self.Preview.SetBackgroundColour(wxColour(255,255,255))

   499

   500

        self._init_sizers()

   501

   502

    def __init__(self, parent):

   503

        self._init_ctrls(parent)

   504

        self.ButtonSizer = self.CreateButtonSizer(wxOK|wxCANCEL|wxCENTRE)

   505

        self.flexGridSizer1.Add(self.ButtonSizer, 1, wxALIGN_RIGHT)

   506

        self.Connection = None

   507

        self.MinConnectionSize = None

   508

   509

        EVT_PAINT(self, self.OnPaint)

   510

   511

    def SetMinConnectionSize(self, size):

   512

        self.MinConnectionSize = size

   513

   514

    def GetValues(self):

   515

        values = {}

   516

        if self.radioButton1.GetValue():

   517

            values["type"] = CONNECTOR

   518

        else:

   519

            values["type"] = CONTINUATION

   520

        values["name"] = self.Name.GetValue()

   521

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

   522

        return values

   523

   524

    def OnTypeChanged(self, event):

   525

        self.RefreshPreview()

   526

        event.Skip()

   527

   528

    def OnNameChanged(self, event):

   529

        self.RefreshPreview()

   530

        event.Skip()