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) 2007: 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

#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

import wx

    26

import wx.grid

    27

    28

from graphics import *

    29

    30

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

    31

#                          Create New Block Dialog

    32

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

    33

    34

[ID_BLOCKPROPERTIESDIALOG, ID_BLOCKPROPERTIESDIALOGNAME, 

    35

 ID_BLOCKPROPERTIESDIALOGTYPETREE, ID_BLOCKPROPERTIESDIALOGTYPEDESC, 

    36

 ID_BLOCKPROPERTIESDIALOGINPUTS, ID_BLOCKPROPERTIESDIALOGPREVIEW, 

    37

 ID_BLOCKPROPERTIESDIALOGSTATICTEXT1, ID_BLOCKPROPERTIESDIALOGSTATICTEXT2, 

    38

 ID_BLOCKPROPERTIESDIALOGSTATICTEXT3, ID_BLOCKPROPERTIESDIALOGSTATICTEXT4, 

    39

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

    40

    41

[CATEGORY, BLOCK] = range(2)

    42

    43

class BlockPropertiesDialog(wx.Dialog):

    44

    def _init_coll_flexGridSizer1_Items(self, parent):

    45

        parent.AddSizer(self.MainSizer, 0, border=20, flag=wx.GROW|wx.TOP|wx.LEFT|wx.RIGHT)

    46

        parent.AddSizer(self.ButtonSizer, 0, border=20, flag=wx.ALIGN_RIGHT|wx.BOTTOM|wx.LEFT|wx.RIGHT)

    47

    48

    def _init_coll_flexGridSizer1_Growables(self, parent):

    49

        parent.AddGrowableCol(0)

    50

        parent.AddGrowableRow(0)

    51

    52

    def _init_coll_MainSizer_Items(self, parent):

    53

        parent.AddSizer(self.LeftBoxSizer, 1, border=5, flag=wx.GROW|wx.RIGHT)

    54

        parent.AddSizer(self.RightGridSizer, 1, border=5, flag=wx.GROW|wx.LEFT)

    55

    56

    def _init_coll_LeftBoxSizer_Items(self, parent):

    57

        parent.AddWindow(self.TypeTree, 3, border=5, flag=wx.GROW|wx.BOTTOM)

    58

        parent.AddWindow(self.TypeDesc, 1, border=0, flag=wx.GROW)

    59

    60

    def _init_coll_RightGridSizer_Items(self, parent):

    61

        parent.AddSizer(self.RightUpGridSizer, 0, border=0, flag=wx.GROW)

    62

        parent.AddWindow(self.staticText4, 0, border=0, flag=wx.GROW)

    63

        parent.AddWindow(self.Preview, 0, border=0, flag=wx.GROW)

    64

    65

    def _init_coll_RightGridSizer_Growables(self, parent):

    66

        parent.AddGrowableCol(0)

    67

        parent.AddGrowableRow(2)

    68

    69

    def _init_coll_RightUpGridSizer_Items(self, parent):

    70

        parent.AddWindow(self.staticText2, 0, border=0, flag=wx.GROW|wx.ALIGN_BOTTOM)

    71

        parent.AddWindow(self.staticText3, 0, border=0, flag=wx.GROW|wx.ALIGN_BOTTOM)

    72

        parent.AddWindow(self.Name, 0, border=0, flag=wx.GROW)

    73

        parent.AddWindow(self.Inputs, 0, border=0, flag=wx.GROW)

    74

    75

    def _init_sizers(self):

    76

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

    77

        self.MainSizer = wx.BoxSizer(wx.HORIZONTAL)

    78

        self.LeftBoxSizer = wx.StaticBoxSizer(self.staticbox1, wx.VERTICAL)

    79

        self.RightGridSizer = wx.FlexGridSizer(cols=1, hgap=0, rows=3, vgap=5)

    80

        self.RightUpGridSizer = wx.GridSizer(cols=2, hgap=5, rows=2, vgap=5)

    81

    82

        self._init_coll_flexGridSizer1_Items(self.flexGridSizer1)

    83

        self._init_coll_flexGridSizer1_Growables(self.flexGridSizer1)

    84

        self._init_coll_MainSizer_Items(self.MainSizer)

    85

        self._init_coll_LeftBoxSizer_Items(self.LeftBoxSizer)

    86

        self._init_coll_RightGridSizer_Items(self.RightGridSizer)

    87

        self._init_coll_RightGridSizer_Growables(self.RightGridSizer)

    88

        self._init_coll_RightUpGridSizer_Items(self.RightUpGridSizer)

    89

    90

        self.SetSizer(self.flexGridSizer1)

    91

    92

    def _init_ctrls(self, prnt):

    93

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

    94

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

    95

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

    96

              title='Block Properties')

    97

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

    98

    99

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

   100

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

   101

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

   102

   103

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

   104

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

   105

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

   106

   107

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

   108

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

   109

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

   110

   111

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

   112

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

   113

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

   114

   115

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

   116

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

   117

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

   118

        self.Bind(wx.EVT_TREE_SEL_CHANGED, self.OnTypeTreeItemSelected,

   119

              id=ID_BLOCKPROPERTIESDIALOGTYPETREE)

   120

   121

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

   122

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

   123

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

   124

   125

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

   126

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

   127

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

   128

        self.Bind(wx.EVT_TEXT, self.OnNameChanged, id=ID_BLOCKPROPERTIESDIALOGNAME)

   129

   130

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

   131

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

   132

              size=wx.Size(0, 24), style=wx.SP_ARROW_KEYS, min=2, max=20)

   133

        self.Bind(wx.EVT_SPINCTRL, self.OnInputsChanged, id=ID_BLOCKPROPERTIESDIALOGINPUTS)

   134

   135

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

   136

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

   137

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

   138

        self.Preview.SetBackgroundColour(wx.Colour(255,255,255))

   139

        setattr(self.Preview, "GetDrawingMode", lambda:FREEDRAWING_MODE)

   140

   141

        self.ButtonSizer = self.CreateButtonSizer(wx.OK|wx.CANCEL|wx.CENTRE)

   142

        self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.ButtonSizer.GetAffirmativeButton().GetId())

   143

   144

        self.Bind(wx.EVT_PAINT, self.OnPaint)

   145

   146

        self._init_sizers()

   147

   148

    def __init__(self, parent):

   149

        self._init_ctrls(parent)

   150

        self.Name.SetValue("")

   151

        self.Name.Enable(False)

   152

        self.Inputs.Enable(False)

   153

        self.Block = None

   154

        self.MinBlockSize = None

   158

   159

    def FindTreeItem(self, root, name, inputs = None):

   160

        if root.IsOk():

   161

            pydata = self.TypeTree.GetPyData(root)

   162

            if inputs and "inputs" in pydata:

   163

                same_inputs = pydata["inputs"] == inputs

   164

            else:

   165

                same_inputs = True

   166

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

   167

                return root

   168

            else:

   169

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

   170

                while item.IsOk():

   171

                    result = self.FindTreeItem(item, name, inputs)

   172

                    if result:

   173

                        return result

   174

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

   175

        return None

   176

   177

    def OnOK(self, event):

   178

        error = []

   179

        selected = self.TypeTree.GetSelection()

   182

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

   183

            message = wx.MessageDialog(self, "Form isn't complete. Valid block type must be selected!", "Error", wx.OK|wx.ICON_ERROR)

   184

            message.ShowModal()

   185

            message.Destroy()

   187

            message = wx.MessageDialog(self, "Form isn't complete. Name must be filled!", "Error", wx.OK|wx.ICON_ERROR)

   188

            message.ShowModal()

   189

            message.Destroy()

   206

        else:

   207

            self.EndModal(wx.ID_OK)

   208

   209

    def SetBlockList(self, blocktypes):

   210

        root = self.TypeTree.AddRoot("")

   211

        self.TypeTree.SetPyData(root, {"type" : CATEGORY})

   212

        for category in blocktypes:

   213

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

   214

            self.TypeTree.SetPyData(category_item, {"type" : CATEGORY})

   215

            for blocktype in category["list"]:

   216

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

   217

                self.TypeTree.SetPyData(blocktype_item, {"type" : BLOCK, "inputs" : tuple([type for name, type, modifier in blocktype["inputs"]])})

   218

   219

    def SetMinBlockSize(self, size):

   220

        self.MinBlockSize = size

   221

   228

    def SetValues(self, values):

   229

        for name, value in values.items():

   230

            if name == "type":

   231

                inputs = None

   232

                if "inputs" in values:

   233

                    inputs = values["inputs"]

   234

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

   235

                if item:

   236

                    self.TypeTree.SelectItem(item)

   237

            elif name == "name":

   238

                self.Name.SetValue(value)

   239

            elif name == "extension":

   240

                self.Inputs.SetValue(value)

   241

        self.RefreshPreview()

   242

   243

    def GetValues(self):

   244

        values = {}

   245

        item = self.TypeTree.GetSelection()

   246

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

   247

        values["inputs"] = self.TypeTree.GetPyData(item)["inputs"]

   248

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

   249

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

   250

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

   251

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

   252

        return values

   253

   254

    def OnTypeTreeItemSelected(self, event):

   255

        self.Name.SetValue("")

   256

        selected = event.GetItem()

   257

        pydata = self.TypeTree.GetPyData(selected)

   258

        if pydata["type"] != CATEGORY:

   259

            blocktype = GetBlockType(self.TypeTree.GetItemText(selected), pydata["inputs"])

   260

            if blocktype:

   261

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

   262

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

   263

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

   264

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

   265

                wx.CallAfter(self.RefreshPreview)

   266

            else:

   267

                self.Name.Enable(False)

   268

                self.Inputs.Enable(False)

   269

                self.Inputs.SetValue(2)

   270

                self.TypeDesc.SetValue("")

   271

                wx.CallAfter(self.ErasePreview)

   272

        else:

   273

            self.Name.Enable(False)

   274

            self.Inputs.Enable(False)

   275

            self.Inputs.SetValue(2)

   276

            self.TypeDesc.SetValue("")

   277

            wx.CallAfter(self.ErasePreview)

   278

        event.Skip()

   279

   280

    def OnNameChanged(self, event):

   281

        if self.Name.IsEnabled():

   282

            self.RefreshPreview()

   283

        event.Skip()

   284

   285

    def OnInputsChanged(self, event):

   286

        if self.Inputs.IsEnabled():

   287

            self.RefreshPreview()

   288

        event.Skip()

   289

   290

    def ErasePreview(self):

   291

        dc = wx.ClientDC(self.Preview)

   292

        dc.Clear()

   293

        self.Block = None

   294

   295

    def RefreshPreview(self):

   296

        dc = wx.ClientDC(self.Preview)

   297

        dc.Clear()

   298

        item = self.TypeTree.GetSelection()

   299

        pydata = self.TypeTree.GetPyData(item)

   300

        if pydata["type"] == CATEGORY:

   301

            self.Block = None

   302

        else:

   303

            blocktype = self.TypeTree.GetItemText(item)

   304

            if blocktype:

   305

                self.Block = FBD_Block(self.Preview, blocktype, self.Name.GetValue(), extension = self.Inputs.GetValue(), inputs = pydata["inputs"])

   306

                width, height = self.MinBlockSize

   307

                min_width, min_height = self.Block.GetMinSize()

   308

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

   309

                self.Block.SetSize(width, height)

   310

                clientsize = self.Preview.GetClientSize()

   311

                x = (clientsize.width - width) / 2

   312

                y = (clientsize.height - height) / 2

   313

                self.Block.SetPosition(x, y)

   314

                self.Block.Draw(dc)

   315

            else:

   316

                self.Block = None

   317

   318

    def OnPaint(self, event):

   319

        if self.Block:

   320

            self.RefreshPreview()

   321

   322

   323

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

   324

#                          Create New Variable Dialog

   325

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

   326

   327

[ID_VARIABLEPROPERTIESDIALOG, ID_VARIABLEPROPERTIESDIALOGSPACER, 

   328

 ID_VARIABLEPROPERTIESDIALOGNAME, ID_VARIABLEPROPERTIESDIALOGCLASS, 

   329

 ID_VARIABLEPROPERTIESDIALOGPREVIEW, ID_VARIABLEPROPERTIESDIALOGEXPRESSION,

   330

 ID_VARIABLEPROPERTIESDIALOGSTATICTEXT1, ID_VARIABLEPROPERTIESDIALOGSTATICTEXT2,

   331

 ID_VARIABLEPROPERTIESDIALOGSTATICTEXT3, ID_VARIABLEPROPERTIESDIALOGSTATICTEXT4,

   332

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

   333

   334

class VariablePropertiesDialog(wx.Dialog):

   335

    def _init_coll_flexGridSizer1_Items(self, parent):

   336

        parent.AddSizer(self.MainSizer, 0, border=20, flag=wx.GROW|wx.TOP|wx.LEFT|wx.RIGHT)

   337

        parent.AddSizer(self.ButtonSizer, 0, border=20, flag=wx.ALIGN_RIGHT|wx.BOTTOM|wx.LEFT|wx.RIGHT)

   338

   339

    def _init_coll_flexGridSizer1_Growables(self, parent):

   340

        parent.AddGrowableCol(0)

   341

        parent.AddGrowableRow(0)

   342

   343

    def _init_coll_MainSizer_Items(self, parent):

   344

        parent.AddSizer(self.TopSizer, 0, border=0, flag=wx.GROW)

   345

        parent.AddWindow(self.staticText4, 0, border=0, flag=wx.GROW)

   346

        parent.AddWindow(self.Preview, 0, border=0, flag=wx.GROW)

   347

   348

    def _init_coll_MainSizer_Growables(self, parent):

   349

        parent.AddGrowableCol(0)

   350

        parent.AddGrowableRow(2)

   351

   352

    def _init_coll_TopSizer_Items(self, parent):

   353

        parent.AddSizer(self.LeftGridSizer, 1, border=5, flag=wx.GROW|wx.RIGHT)

   354

        parent.AddSizer(self.RightGridSizer, 1, border=5, flag=wx.GROW|wx.LEFT)

   355

   356

    def _init_coll_LeftGridSizer_Items(self, parent):

   357

        parent.AddWindow(self.staticText1, 0, border=0, flag=wx.GROW)

   358

        parent.AddWindow(self.Class, 0, border=0, flag=wx.GROW)

   359

        parent.AddWindow(self.Spacer, 0, border=0, flag=wx.GROW)

   360

        parent.AddWindow(self.staticText2, 0, border=0, flag=wx.GROW)

   361

        parent.AddWindow(self.Expression, 0, border=0, flag=wx.GROW)

   362

   363

    def _init_coll_LeftGridSizer_Growables(self, parent):

   364

        parent.AddGrowableCol(0)

   365

        parent.AddGrowableRow(2)

   366

   367

    def _init_coll_RightGridSizer_Items(self, parent):

   368

        parent.AddWindow(self.staticText3, 0, border=0, flag=wx.GROW)

   369

        parent.AddWindow(self.Name, 0, border=0, flag=wx.GROW)

   370

   371

    def _init_coll_RightGridSizer_Growables(self, parent):

   372

        parent.AddGrowableCol(0)

   373

        parent.AddGrowableRow(1)

   374

   375

    def _init_sizers(self):

   376

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

   377

        self.MainSizer = wx.FlexGridSizer(cols=1, hgap=0, rows=3, vgap=5)

   378

        self.TopSizer = wx.BoxSizer(wx.HORIZONTAL)

   379

        self.LeftGridSizer = wx.FlexGridSizer(cols=1, hgap=0, rows=5, vgap=5)

   380

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

   381

   382

        self._init_coll_flexGridSizer1_Items(self.flexGridSizer1)

   383

        self._init_coll_flexGridSizer1_Growables(self.flexGridSizer1)

   384

        self._init_coll_MainSizer_Items(self.MainSizer)

   385

        self._init_coll_MainSizer_Growables(self.MainSizer)

   386

        self._init_coll_TopSizer_Items(self.TopSizer)

   387

        self._init_coll_LeftGridSizer_Items(self.LeftGridSizer)

   388

        self._init_coll_LeftGridSizer_Growables(self.LeftGridSizer)

   389

        self._init_coll_RightGridSizer_Items(self.RightGridSizer)

   390

        self._init_coll_RightGridSizer_Growables(self.RightGridSizer)

   391

   392

        self.SetSizer(self.flexGridSizer1)

   393

   394

    def _init_ctrls(self, prnt):

   395

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

   396

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

   397

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

   398

              title='Variable Properties')

   399

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

   400

   401

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

   402

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

   403

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

   404

   405

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

   406

              label='Expression:', name='staticText2', parent=self,

   407

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

   408

   409

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

   410

              label='Name:', name='staticText3', parent=self,

   411

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

   412

   413

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

   414

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

   415

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

   416

   417

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

   418

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

   419

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

   420

        self.Bind(wx.EVT_CHOICE, self.OnClassChanged, id=ID_VARIABLEPROPERTIESDIALOGCLASS)

   421

   422

        self.Name = wx.ListBox(id=ID_VARIABLEPROPERTIESDIALOGNAME,

   423

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

   424

              size=wx.Size(0, 90), style=wx.LB_SINGLE)

   425

        self.Bind(wx.EVT_LISTBOX, self.OnNameChanged, id=ID_VARIABLEPROPERTIESDIALOGNAME)

   426

   427

        self.Expression = wx.TextCtrl(id=ID_VARIABLEPROPERTIESDIALOGEXPRESSION,

   428

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

   429

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

   430

        self.Bind(wx.EVT_TEXT, self.OnExpressionChanged, id=ID_VARIABLEPROPERTIESDIALOGEXPRESSION)

   431

   432

        self.Spacer = wx.Panel(id=ID_VARIABLEPROPERTIESDIALOGSPACER,

   433

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

   434

              size=wx.Size(0, 0), style=wx.TAB_TRAVERSAL)

   435

   436

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

   437

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

   438

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

   439

        self.Preview.SetBackgroundColour(wx.Colour(255,255,255))

   440

        setattr(self.Preview, "GetDrawingMode", lambda:FREEDRAWING_MODE)

   441

   442

        self.ButtonSizer = self.CreateButtonSizer(wx.OK|wx.CANCEL|wx.CENTRE)

   443

   444

        self.Bind(wx.EVT_PAINT, self.OnPaint)

   445

   446

        self._init_sizers()

   447

   448

    def __init__(self, parent):

   449

        self._init_ctrls(parent)

   450

        self.Variable = None

   451

        self.VarList = []

   452

        self.MinVariableSize = None

   453

        self.RefreshNameList()

   454

   455

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

   456

            self.Class.Append(choice)

   457

        self.Class.SetStringSelection("Input")

   458

   459

    def RefreshNameList(self):

   460

        selected = self.Name.GetStringSelection()

   461

        self.Name.Clear()

   462

        self.Name.Append("")

   463

        for name, var_type, value_type in self.VarList:

   464

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

   465

                self.Name.Append(name)

   466

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

   467

                self.Name.Append(name)

   468

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

   469

                self.Name.Append(name)

   470

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

   471

                self.Name.Append(name)

   472

        if self.Name.FindString(selected) != wx.NOT_FOUND:

   473

            self.Name.SetStringSelection(selected)

   474

        else:

   475

            self.Name.SetStringSelection("")

   476

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

   477

   478

    def SetMinVariableSize(self, size):

   479

        self.MinVariableSize = size

   480

   481

    def SetVariables(self, vars):

   482

        self.VarList = vars

   483

        self.RefreshNameList()

   484

   485

    def SetValues(self, values):

   486

        value_type = values.get("type", None)

   487

        value_name = values.get("name", None)

   488

        if value_type:

   489

            if value_type == INPUT:

   490

                self.Class.SetStringSelection("Input")

   491

            if value_type == OUTPUT:

   492

                self.Class.SetStringSelection("Output")

   493

            if value_type == INOUT:

   494

                self.Class.SetStringSelection("InOut")

   495

            self.RefreshNameList()

   496

        if value_name:

   497

            if self.Name.FindString(value_name) != wx.NOT_FOUND:

   498

                self.Name.SetStringSelection(value_name)

   499

                self.Expression.Enable(False)

   500