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 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

__version__ = "$Revision$"

    26

    27

import wx

    28

import wx.lib.buttons

    29

    30

import types

    31

    32

import time

    33

    34

import os, re, platform, sys, time, traceback, getopt, commands

    35

    36

from plugger import PluginsRoot

    37

    38

from wxPopen import wxPopen3

    39

    40

class LogPseudoFile:

    41

    """ Base class for file like objects to facilitate StdOut for the Shell."""

    42

    def __init__(self, output = None):

    43

        self.red_white = wx.TextAttr("RED", "WHITE")

    44

        self.red_yellow = wx.TextAttr("RED", "YELLOW")

    45

        self.black_white = wx.TextAttr("BLACK", "WHITE")

    46

        self.default_style = None

    47

        self.output = output

    48

    49

    def writelines(self, l):

    50

        map(self.write, l)

    51

    52

    def write(self, s, style = None):

    53

        if not style : style=self.black_white

    54

        if self.default_style != style: 

    55

            self.output.SetDefaultStyle(style)

    56

            self.default_style = style

    57

        self.output.AppendText(s) 

    58

    59

    def write_warning(self, s):

    60

        self.write(s,self.red_white)

    61

    62

    def write_error(self, s):

    63

        self.write(s,self.red_yellow)

    64

    65

    def flush(self):

    66

        self.output.SetValue("")

    67

    68

    def isatty(self):

    69

        return false

    70

    71

    def LogCommand(self, Command, sz_limit = 100, no_stdout=False):

    72

        self.errlen = 0

    73

        self.exitcode = None

    74

        self.outdata = ""

    75

        self.errdata = ""

    76

    77

        def output(v):

    78

            self.outdata += v

    79

            if not no_stdout:

    80

                self.write(v)

    81

    82

        def errors(v):

    83

            self.errdata += v

    84

            self.errlen += 1

    85

            if self.errlen > sz_limit:

    86

                p.kill()

    87

            self.write_warning(v)

    88

    89

        def fin(pid,ecode):

    90

            self.exitcode = ecode

    91

            if self.exitcode != 0:

    92

                self.write(Command + "\n")

    93

                self.write_warning("exited with status %d (pid %d)\n"%(ecode,pid))

    94

    95

        def spin(p):

    96

            while not p.finished:

    97

                wx.Yield()

    98

                time.sleep(0.01)

    99

   100

        input = []

   101

        p = wxPopen3(Command, input, output, errors, fin, self.output)

   102

        spin(p)

   103

   104

        return (self.exitcode, self.outdata, self.errdata)

   105

   106

[ID_BEREMIZ, ID_BEREMIZMAINSPLITTER, 

   107

 ID_BEREMIZSECONDSPLITTER, ID_BEREMIZLEFTPANEL, 

   108

 ID_BEREMIZPARAMSPANEL, ID_BEREMIZLOGCONSOLE, 

   109

 ID_BEREMIZPLUGINTREE, ID_BEREMIZPLUGINCHILDS, 

   110

 ID_BEREMIZADDBUTTON, ID_BEREMIZDELETEBUTTON,

   111

 ID_BEREMIZPARAMSSTATICBOX, ID_BEREMIZPARAMSENABLE, 

   112

 ID_BEREMIZPARAMSTARGETTYPE, ID_BEREMIZPARAMSIECCHANNEL, 

   113

 ID_BEREMIZPARAMSSTATICTEXT1, ID_BEREMIZPARAMSSTATICTEXT2, 

   114

 ID_BEREMIZPARAMSATTRIBUTESGRID,

   115

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

   116

   117

[ID_BEREMIZFILEMENUITEMS0, ID_BEREMIZFILEMENUITEMS1, 

   118

 ID_BEREMIZFILEMENUITEMS2, ID_BEREMIZFILEMENUITEMS3, 

   119

 ID_BEREMIZFILEMENUITEMS5, ID_BEREMIZFILEMENUITEMS7, 

   120

] = [wx.NewId() for _init_coll_FileMenu_Items in range(6)]

   121

   122

[ID_BEREMIZEDITMENUITEMS0, ID_BEREMIZEDITMENUITEMS2, 

   123

 ID_BEREMIZEDITMENUITEMS3, 

   124

] = [wx.NewId() for _init_coll_EditMenu_Items in range(3)]

   125

   126

[ID_BEREMIZRUNMENUITEMS0, ID_BEREMIZRUNMENUITEMS2, 

   127

 ID_BEREMIZRUNMENUITEMS3, ID_BEREMIZRUNMENUITEMS5, 

   128

] = [wx.NewId() for _init_coll_EditMenu_Items in range(4)]

   129

   130

[ID_BEREMIZHELPMENUITEMS0, ID_BEREMIZHELPMENUITEMS1, 

   131

] = [wx.NewId() for _init_coll_HelpMenu_Items in range(2)]

   132

   133

class Beremiz(wx.Frame):

   134

   135

    def _init_coll_FileMenu_Items(self, parent):

   136

        parent.Append(help='', id=ID_BEREMIZFILEMENUITEMS0,

   137

              kind=wx.ITEM_NORMAL, text=u'New\tCTRL+N')

   138

        parent.Append(help='', id=ID_BEREMIZFILEMENUITEMS1,

   139

              kind=wx.ITEM_NORMAL, text=u'Open\tCTRL+O')

   140

        parent.Append(help='', id=ID_BEREMIZFILEMENUITEMS2,

   141

              kind=wx.ITEM_NORMAL, text=u'Save\tCTRL+S')

   142

        parent.Append(help='', id=ID_BEREMIZFILEMENUITEMS3,

   143

              kind=wx.ITEM_NORMAL, text=u'Close Project')

   144

        parent.AppendSeparator()

   145

        parent.Append(help='', id=ID_BEREMIZFILEMENUITEMS5,

   146

              kind=wx.ITEM_NORMAL, text=u'Properties')

   147

        parent.AppendSeparator()

   148

        parent.Append(help='', id=ID_BEREMIZFILEMENUITEMS7,

   149

              kind=wx.ITEM_NORMAL, text=u'Quit\tCTRL+Q')

   150

        self.Bind(wx.EVT_MENU, self.OnNewProjectMenu,

   151

              id=ID_BEREMIZFILEMENUITEMS0)

   152

        self.Bind(wx.EVT_MENU, self.OnOpenProjectMenu,

   153

              id=ID_BEREMIZFILEMENUITEMS1)

   154

        self.Bind(wx.EVT_MENU, self.OnSaveProjectMenu,

   155

              id=ID_BEREMIZFILEMENUITEMS2)

   156

        self.Bind(wx.EVT_MENU, self.OnCloseProjectMenu,

   157

              id=ID_BEREMIZFILEMENUITEMS3)

   158

        self.Bind(wx.EVT_MENU, self.OnPropertiesMenu,

   159

              id=ID_BEREMIZFILEMENUITEMS5)

   160

        self.Bind(wx.EVT_MENU, self.OnQuitMenu,

   161

              id=ID_BEREMIZFILEMENUITEMS7)

   162

   163

    def _init_coll_EditMenu_Items(self, parent):

   164

        parent.Append(help='', id=ID_BEREMIZEDITMENUITEMS0,

   165

              kind=wx.ITEM_NORMAL, text=u'Edit PLC\tCTRL+R')

   166

        parent.AppendSeparator()

   167

        parent.Append(help='', id=ID_BEREMIZEDITMENUITEMS2,

   168

              kind=wx.ITEM_NORMAL, text=u'Add Plugin')

   169

        parent.Append(help='', id=ID_BEREMIZEDITMENUITEMS3,

   170

              kind=wx.ITEM_NORMAL, text=u'Delete Plugin')

   171

        self.Bind(wx.EVT_MENU, self.OnEditPLCMenu,

   172

              id=ID_BEREMIZEDITMENUITEMS0)

   173

        self.Bind(wx.EVT_MENU, self.OnAddMenu,

   174

              id=ID_BEREMIZEDITMENUITEMS2)

   175

        self.Bind(wx.EVT_MENU, self.OnDeleteMenu,

   176

              id=ID_BEREMIZEDITMENUITEMS3)

   177

   178

    def _init_coll_RunMenu_Items(self, parent):

   179

        parent.Append(help='', id=ID_BEREMIZRUNMENUITEMS0,

   180

              kind=wx.ITEM_NORMAL, text=u'Build\tCTRL+R')

   181

        parent.AppendSeparator()

   182

        parent.Append(help='', id=ID_BEREMIZRUNMENUITEMS2,

   183

              kind=wx.ITEM_NORMAL, text=u'Simulate')

   184

        parent.Append(help='', id=ID_BEREMIZRUNMENUITEMS3,

   185

              kind=wx.ITEM_NORMAL, text=u'Run')

   186

        parent.AppendSeparator()

   187

        parent.Append(help='', id=ID_BEREMIZRUNMENUITEMS5,

   188

              kind=wx.ITEM_NORMAL, text=u'Save Log')

   189

        self.Bind(wx.EVT_MENU, self.OnBuildMenu,

   190

              id=ID_BEREMIZRUNMENUITEMS0)

   191

        self.Bind(wx.EVT_MENU, self.OnSimulateMenu,

   192

              id=ID_BEREMIZRUNMENUITEMS2)

   193

        self.Bind(wx.EVT_MENU, self.OnRunMenu,

   194

              id=ID_BEREMIZRUNMENUITEMS3)

   195

        self.Bind(wx.EVT_MENU, self.OnSaveLogMenu,

   196

              id=ID_BEREMIZRUNMENUITEMS5)

   197

   198

    def _init_coll_HelpMenu_Items(self, parent):

   199

        parent.Append(help='', id=ID_BEREMIZHELPMENUITEMS0,

   200

              kind=wx.ITEM_NORMAL, text=u'Beremiz\tF1')

   201

        parent.Append(help='', id=ID_BEREMIZHELPMENUITEMS1,

   202

              kind=wx.ITEM_NORMAL, text=u'About')

   203

        self.Bind(wx.EVT_MENU, self.OnBeremizMenu,

   204

              id=ID_BEREMIZHELPMENUITEMS0)

   205

        self.Bind(wx.EVT_MENU, self.OnAboutMenu,

   206

              id=ID_BEREMIZHELPMENUITEMS1)

   207

   208

    def _init_coll_menuBar1_Menus(self, parent):

   209

        parent.Append(menu=self.FileMenu, title=u'File')

   210

        #parent.Append(menu=self.EditMenu, title=u'Edit')

   211

        #parent.Append(menu=self.RunMenu, title=u'Run')

   212

        parent.Append(menu=self.HelpMenu, title=u'Help')

   213

   214

    def _init_utils(self):

   215

        self.menuBar1 = wx.MenuBar()

   216

        self.FileMenu = wx.Menu(title=u'')

   217

        #self.EditMenu = wx.Menu(title=u'')

   218

        #self.RunMenu = wx.Menu(title=u'')

   219

        self.HelpMenu = wx.Menu(title=u'')

   220

   221

        self._init_coll_menuBar1_Menus(self.menuBar1)

   222

        self._init_coll_FileMenu_Items(self.FileMenu)

   223

        #self._init_coll_EditMenu_Items(self.EditMenu)

   224

        #self._init_coll_RunMenu_Items(self.RunMenu)

   225

        self._init_coll_HelpMenu_Items(self.HelpMenu)

   226

   227

    def _init_coll_LeftGridSizer_Items(self, parent):

   228

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

   229

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

   230

   231

    def _init_coll_LeftGridSizer_Growables(self, parent):

   232

        parent.AddGrowableCol(0)

   233

        parent.AddGrowableRow(0)

   234

   235

    def _init_coll_RightGridSizer_Items(self, parent):

   236

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

   237

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

   238

   239

    def _init_coll_RightGridSizer_Growables(self, parent):

   240

        parent.AddGrowableCol(0)

   241

        parent.AddGrowableRow(1)

   242

   243

    def _init_coll_ButtonGridSizer_Items(self, parent):

   244

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

   245

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

   246

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

   247

   248

    def _init_coll_ButtonGridSizer_Growables(self, parent):

   249

        parent.AddGrowableCol(0)

   250

        parent.AddGrowableRow(0)

   251

   252

    def _init_sizers(self):

   253

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

   254

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

   255

        self.ButtonGridSizer = wx.FlexGridSizer(cols=3, hgap=2, rows=1, vgap=2)

   256

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

   257

        self.ParamsPanelMainSizer = wx.BoxSizer(wx.VERTICAL)

   258

   259

        self._init_coll_LeftGridSizer_Growables(self.LeftGridSizer)

   260

        self._init_coll_LeftGridSizer_Items(self.LeftGridSizer)

   261

        self._init_coll_RightGridSizer_Growables(self.RightGridSizer)

   262

        self._init_coll_RightGridSizer_Items(self.RightGridSizer)

   263

        self._init_coll_ButtonGridSizer_Growables(self.ButtonGridSizer)

   264

        self._init_coll_ButtonGridSizer_Items(self.ButtonGridSizer)

   265

   266

        self.LeftPanel.SetSizer(self.LeftGridSizer)

   267

        self.RightPanel.SetSizer(self.RightGridSizer)

   268

        self.ParamsPanel.SetSizer(self.ParamsPanelMainSizer)

   269

   270

    def _init_ctrls(self, prnt):

   271

        wx.Frame.__init__(self, id=ID_BEREMIZ, name=u'Beremiz',

   272

              parent=prnt, pos=wx.Point(0, 0), size=wx.Size(1000, 600),

   273

              style=wx.DEFAULT_FRAME_STYLE|wx.CLIP_CHILDREN, title=u'Beremiz')

   274

        self._init_utils()

   275

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

   276

        self.SetMenuBar(self.menuBar1)

   277

        self.Bind(wx.EVT_ACTIVATE, self.OnFrameActivated)

   278

   279

        self.MainSplitter = wx.SplitterWindow(id=ID_BEREMIZMAINSPLITTER,

   280

              name='MainSplitter', parent=self, point=wx.Point(0, 0),

   281

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

   282

        self.MainSplitter.SetNeedUpdating(True)

   283

        self.MainSplitter.SetMinimumPaneSize(1)

   284

   285

        self.LeftPanel = wx.Panel(id=ID_BEREMIZLEFTPANEL, 

   286

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

   287

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

   288

   289

        self.PluginTree = wx.TreeCtrl(id=ID_BEREMIZPLUGINTREE,

   290

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

   291

              size=wx.Size(-1, -1), style=wx.TR_HAS_BUTTONS|wx.TR_SINGLE|wx.SUNKEN_BORDER)

   292

        self.PluginTree.Bind(wx.EVT_RIGHT_UP, self.OnPluginTreeRightUp)

   293

        self.Bind(wx.EVT_TREE_SEL_CHANGED, self.OnPluginTreeItemSelected,

   294

              id=ID_BEREMIZPLUGINTREE)

   295

   296

        self.PluginChilds = wx.Choice(id=ID_BEREMIZPLUGINCHILDS,

   297

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

   298

              size=wx.Size(-1, -1), style=0)

   299

   300

        if wx.VERSION < (2, 8, 0):

   301

            self.AddButton = wx.lib.buttons.GenBitmapButton(ID=ID_BEREMIZADDBUTTON, bitmap=wx.Bitmap(os.path.join('images', 'Add.png')),

   302

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

   303

                  size=wx.Size(32, 32), style=wx.NO_BORDER)

   304

        else:

   305

            self.AddButton = wx.lib.buttons.GenBitmapButton(id=ID_BEREMIZADDBUTTON, bitmap=wx.Bitmap(os.path.join('images', 'Add.png')),

   306

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

   307

                  size=wx.Size(32, 32), style=wx.NO_BORDER)

   308

   309

        self.AddButton.SetToolTipString("Add a plugin of the type selected")

   310

        self.AddButton.Bind(wx.EVT_BUTTON, self.OnAddButton,

   311

              id=ID_BEREMIZADDBUTTON)

   312

   313

        if wx.VERSION < (2, 8, 0):

   314

            self.DeleteButton = wx.lib.buttons.GenBitmapButton(ID=ID_BEREMIZDELETEBUTTON, bitmap=wx.Bitmap(os.path.join('images', 'Delete.png')),

   315

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

   316

                  size=wx.Size(32, 32), style=wx.NO_BORDER)

   317

        else:

   318

            self.DeleteButton = wx.lib.buttons.GenBitmapButton(id=ID_BEREMIZDELETEBUTTON, bitmap=wx.Bitmap(os.path.join('images', 'Delete.png')),

   319

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

   320

                  size=wx.Size(32, 32), style=wx.NO_BORDER)

   321

        self.DeleteButton.SetToolTipString("Delete the current selected plugin")

   322

        self.DeleteButton.Bind(wx.EVT_BUTTON, self.OnDeleteButton,

   323

              id=ID_BEREMIZDELETEBUTTON)

   324

   325

        self.RightPanel = wx.Panel(id=ID_BEREMIZLEFTPANEL, 

   326

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

   327

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

   328

   329

        self.SecondSplitter = wx.SplitterWindow(id=ID_BEREMIZSECONDSPLITTER,

   330

              name='SecondSplitter', parent=self.RightPanel, point=wx.Point(0, 0),

   331

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

   332

        self.SecondSplitter.SetNeedUpdating(True)

   333

        self.SecondSplitter.SetMinimumPaneSize(1)

   334

   335

        self.MainSplitter.SplitVertically(self.LeftPanel, self.RightPanel,

   336

              300)

   337

   338

        self.ParamsPanel = wx.ScrolledWindow(id=ID_BEREMIZPARAMSPANEL, 

   339

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

   340

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

   341

        self.ParamsPanel.SetScrollbars(10, 10, 0, 0, 0, 0)

   342

   343

        self.LogConsole = wx.TextCtrl(id=ID_BEREMIZLOGCONSOLE, value='',

   344

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

   345

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

   346

   347

        self.SecondSplitter.SplitHorizontally(self.ParamsPanel, self.LogConsole,

   348

              -250)

   349

   350

        self._init_sizers()

   351

   352

    def __init__(self, parent, projectOpen):

   353

        self._init_ctrls(parent)

   354

   355

        self.Log = LogPseudoFile(self.LogConsole)

   356

   357

        self.PluginRoot = PluginsRoot(self)

   358

   359

        if projectOpen:

   360

            self.PluginRoot.LoadProject(projectOpen, self.Log)

   361

            self.RefreshPluginTree()

   362

            self.PluginTree.SelectItem(self.PluginTree.GetRootItem())

   363

   364

        self.RefreshPluginParams()

   365

        self.RefreshButtons()

   366

        self.RefreshMainMenu()

   367

   368

    def OnFrameActivated(self, event):

   369

        if not event.GetActive():

   370

            self.PluginRoot.RefreshPluginsBlockLists()

   371

   372

    def RefreshButtons(self):

   373

        if self.PluginRoot.HasProjectOpened():

   374

            self.PluginChilds.Enable(True)

   375

            self.AddButton.Enable(True)

   376

            self.DeleteButton.Enable(True)

   377

        else:

   378

            self.PluginChilds.Enable(False)

   379

            self.AddButton.Enable(False)

   380

            self.DeleteButton.Enable(False)

   381

   382

    def RefreshMainMenu(self):

   383

        if self.menuBar1:

   384

            if self.PluginRoot.HasProjectOpened():

   385

##                self.menuBar1.EnableTop(1, True)

   386

##                self.menuBar1.EnableTop(2, True)

   387

                self.FileMenu.Enable(ID_BEREMIZFILEMENUITEMS2, True)

   388

                self.FileMenu.Enable(ID_BEREMIZFILEMENUITEMS3, True)

   389

                self.FileMenu.Enable(ID_BEREMIZFILEMENUITEMS5, True)

   390

            else:

   391

##                self.menuBar1.EnableTop(1, False)

   392

##                self.menuBar1.EnableTop(2, False)

   393

                self.FileMenu.Enable(ID_BEREMIZFILEMENUITEMS2, False)

   394

                self.FileMenu.Enable(ID_BEREMIZFILEMENUITEMS3, False)

   395

                self.FileMenu.Enable(ID_BEREMIZFILEMENUITEMS5, False)

   396

   397

    def RefreshPluginTree(self):

   398

        infos = self.PluginRoot.GetPlugInfos()

   399

        root = self.PluginTree.GetRootItem()

   400

        if not root.IsOk():

   401

            root = self.PluginTree.AddRoot(infos["name"])

   402

        last_selected = self.GetSelectedPluginName()

   403

        self.GenerateTreeBranch(root, infos, True)

   404

        self.PluginTree.Expand(self.PluginTree.GetRootItem())

   405

        self.SelectedPluginByName(root, last_selected)

   406

        self.RefreshPluginParams()

   407

   408

    def SelectedPluginByName(self, root, name):

   409

        if name:

   410

            toks = name.split('.',1)

   411

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

   412

            while item.IsOk():

   413

                if self.PluginTree.GetPyData(item) == toks[0]:

   414

                    if len(toks)>1:

   415

                        return self.SelectedPluginByName(item, toks[1])

   416

                    else:

   417

                        self.PluginTree.SelectItem(item, True)

   418

                        return True

   419

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

   420

        return False

   421

   422

    def GenerateTreeBranch(self, root, infos, first = False):

   423

        to_delete = []

   424

        self.PluginTree.SetItemText(root, infos["name"])

   425

        self.PluginTree.SetPyData(root, infos["type"])

   426

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

   427

        for values in infos["values"]:

   428

            if not item.IsOk():

   429

                item = self.PluginTree.AppendItem(root, "")

   430

                if wx.Platform != '__WXMSW__':

   431

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

   432

            self.GenerateTreeBranch(item, values)

   433

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

   434

        while item.IsOk():

   435

            to_delete.append(item)

   436

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

   437

        for item in to_delete:

   438

            self.PluginTree.Delete(item)

   439

   440

    def GetSelectedPluginName(self):

   441

        selected = self.PluginTree.GetSelection()

   442

        if not selected.IsOk():

   443

            return None

   444

        if selected == self.PluginTree.GetRootItem():

   445

            return ""

   446

        else:

   447

            name = self.PluginTree.GetPyData(selected)

   448

            item = self.PluginTree.GetItemParent(selected)

   449

            while item.IsOk() and item != self.PluginTree.GetRootItem():

   450

                name = "%s.%s"%(self.PluginTree.GetPyData(item), name)

   451

                item = self.PluginTree.GetItemParent(item)

   452

            return name

   453

   454

    def GetSelectedPlugin(self):

   455

        name = self.GetSelectedPluginName()

   456

        if name is None:

   457

            return None

   458

        elif name == "":

   459

            return self.PluginRoot

   460

        else:

   461

            return self.PluginRoot.GetChildByName(name)

   462

   463

    def OnPluginTreeItemSelected(self, event):

   464

        wx.CallAfter(self.RefreshPluginParams)

   465

        event.Skip()

   466

   467

    def _GetAddPluginFunction(self, name):

   468

        def OnPluginMenu(event):

   469

            self.AddPlugin(name)

   470

            event.Skip()

   471

        return OnPluginMenu

   472

   473

    def OnPluginTreeRightUp(self, event):

   474

        plugin = self.GetSelectedPlugin()

   475

        if plugin:

   476

            main_menu = wx.Menu(title='')

   477

            if len(plugin.PlugChildsTypes) > 0:

   478

                plugin_menu = wx.Menu(title='')

   479

                for name, XSDClass in self.GetSelectedPlugin().PlugChildsTypes:

   480

                    new_id = wx.NewId()

   481

                    plugin_menu.Append(help='', id=new_id, kind=wx.ITEM_NORMAL, text=name)

   482

                    self.Bind(wx.EVT_MENU, self._GetAddPluginFunction(name), id=new_id)

   483

                main_menu.AppendMenu(-1, "Add", plugin_menu, '')

   484

            new_id = wx.NewId()

   485

            main_menu.Append(help='', id=new_id, kind=wx.ITEM_NORMAL, text="Delete")

   486

            self.Bind(wx.EVT_MENU, self.OnDeleteButton, id=new_id)

   487

            rect = self.PluginTree.GetBoundingRect(self.PluginTree.GetSelection())

   488

            self.PluginTree.PopupMenuXY(main_menu, rect.x + rect.width, rect.y)

   489

        event.Skip()

   490