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

if wx.VERSION >= (2, 8, 0):

    30

    import wx.lib.customtreectrl as CT

    31

    32

import types

    33

    34

import time

    35

    36

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

    37

    38

from plugger import PluginsRoot

    39

    40

from wxPopen import wxPopen3

    41

    42

CWD = os.path.split(os.path.realpath(__file__))[0]

    43

    44

class LogPseudoFile:

    45

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

    46

    def __init__(self, output = None):

    47

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

    48

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

    49

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

    50

        self.default_style = None

    51

        self.output = output

    52

    53

    def writelines(self, l):

    54

        map(self.write, l)

    55

    56

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

    57

        if not style : style=self.black_white

    58

        if self.default_style != style: 

    59

            self.output.SetDefaultStyle(style)

    60

            self.default_style = style

    61

        self.output.AppendText(s) 

    62

    63

    def write_warning(self, s):

    64

        self.write(s,self.red_white)

    65

    66

    def write_error(self, s):

    67

        self.write(s,self.red_yellow)

    68

    69

    def flush(self):

    70

        self.output.SetValue("")

    71

    72

    def isatty(self):

    73

        return false

    74

    75

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

    76

        self.errlen = 0

    77

        self.exitcode = None

    78

        self.outdata = ""

    79

        self.errdata = ""

    80

    81

        def output(v):

    82

            self.outdata += v

    83

            if not no_stdout:

    84

                self.write(v)

    85

    86

        def errors(v):

    87

            self.errdata += v

    88

            self.errlen += 1

    89

            if self.errlen > sz_limit:

    90

                p.kill()

    91

            self.write_warning(v)

    92

    93

        def fin(pid,ecode):

    94

            self.exitcode = ecode

    95

            if self.exitcode != 0:

    96

                self.write(Command + "\n")

    97

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

    98

    99

        def spin(p):

   100

            while not p.finished:

   101

                wx.Yield()

   102

                time.sleep(0.01)

   103

   104

        input = []

   105

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

   106

        if p.pid:

   107

        	spin(p)

   108

   109

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

   110

   111

[ID_BEREMIZ, ID_BEREMIZMAINSPLITTER, 

   112

 ID_BEREMIZSECONDSPLITTER, ID_BEREMIZLEFTPANEL, 

   113

 ID_BEREMIZPARAMSPANEL, ID_BEREMIZLOGCONSOLE, 

   114

 ID_BEREMIZPLUGINTREE, ID_BEREMIZPLUGINCHILDS, 

   115

 ID_BEREMIZADDBUTTON, ID_BEREMIZDELETEBUTTON,

   116

 ID_BEREMIZPARAMSSTATICBOX, ID_BEREMIZPARAMSENABLE, 

   117

 ID_BEREMIZPARAMSTARGETTYPE, ID_BEREMIZPARAMSIECCHANNEL, 

   118

 ID_BEREMIZPARAMSSTATICTEXT1, ID_BEREMIZPARAMSSTATICTEXT2, 

   119

 ID_BEREMIZPARAMSATTRIBUTESGRID,

   120

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

   121

   122

[ID_BEREMIZFILEMENUITEMS0, ID_BEREMIZFILEMENUITEMS1, 

   123

 ID_BEREMIZFILEMENUITEMS2, ID_BEREMIZFILEMENUITEMS3, 

   124

 ID_BEREMIZFILEMENUITEMS5, ID_BEREMIZFILEMENUITEMS7, 

   125

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

   126

   127

[ID_BEREMIZEDITMENUITEMS0, ID_BEREMIZEDITMENUITEMS2, 

   128

 ID_BEREMIZEDITMENUITEMS3, 

   129

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

   130

   131

[ID_BEREMIZRUNMENUITEMS0, ID_BEREMIZRUNMENUITEMS2, 

   132

 ID_BEREMIZRUNMENUITEMS3, ID_BEREMIZRUNMENUITEMS5, 

   133

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

   134

   135

[ID_BEREMIZHELPMENUITEMS0, ID_BEREMIZHELPMENUITEMS1, 

   136

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

   137

   138

class Beremiz(wx.Frame):

   139

   140

    def _init_coll_FileMenu_Items(self, parent):

   141

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

   142

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

   143

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

   144

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

   145

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

   146

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

   147

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

   148

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

   149

        parent.AppendSeparator()

   150

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

   151

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

   152

        parent.AppendSeparator()

   153

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

   154

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

   155

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

   156

              id=ID_BEREMIZFILEMENUITEMS0)

   157

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

   158

              id=ID_BEREMIZFILEMENUITEMS1)

   159

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

   160

              id=ID_BEREMIZFILEMENUITEMS2)

   161

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

   162

              id=ID_BEREMIZFILEMENUITEMS3)

   163

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

   164

              id=ID_BEREMIZFILEMENUITEMS5)

   165

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

   166

              id=ID_BEREMIZFILEMENUITEMS7)

   167

   168

    def _init_coll_EditMenu_Items(self, parent):

   169

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

   170

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

   171

        parent.AppendSeparator()

   172

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

   173

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

   174

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

   175

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

   176

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

   177

              id=ID_BEREMIZEDITMENUITEMS0)

   178

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

   179

              id=ID_BEREMIZEDITMENUITEMS2)

   180

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

   181

              id=ID_BEREMIZEDITMENUITEMS3)

   182

   183

    def _init_coll_RunMenu_Items(self, parent):

   184

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

   185

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

   186

        parent.AppendSeparator()

   187

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

   188

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

   189

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

   190

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

   191

        parent.AppendSeparator()

   192

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

   193

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

   194

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

   195

              id=ID_BEREMIZRUNMENUITEMS0)

   196

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

   197

              id=ID_BEREMIZRUNMENUITEMS2)

   198

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

   199

              id=ID_BEREMIZRUNMENUITEMS3)

   200

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

   201

              id=ID_BEREMIZRUNMENUITEMS5)

   202

   203

    def _init_coll_HelpMenu_Items(self, parent):

   204

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

   205

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

   206

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

   207

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

   208

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

   209

              id=ID_BEREMIZHELPMENUITEMS0)

   210

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

   211

              id=ID_BEREMIZHELPMENUITEMS1)

   212

   213

    def _init_coll_menuBar1_Menus(self, parent):

   214

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

   215

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

   216

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

   217

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

   218

   219

    def _init_utils(self):

   220

        self.menuBar1 = wx.MenuBar()

   221

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

   222

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

   223

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

   224

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

   225

   226

        self._init_coll_menuBar1_Menus(self.menuBar1)

   227

        self._init_coll_FileMenu_Items(self.FileMenu)

   228

        #self._init_coll_EditMenu_Items(self.EditMenu)

   229

        #self._init_coll_RunMenu_Items(self.RunMenu)

   230

        self._init_coll_HelpMenu_Items(self.HelpMenu)

   231

   232

    def _init_coll_LeftGridSizer_Items(self, parent):

   233

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

   234

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

   235

   236

    def _init_coll_LeftGridSizer_Growables(self, parent):

   237

        parent.AddGrowableCol(0)

   238

        parent.AddGrowableRow(0)

   239

   240

    def _init_coll_RightGridSizer_Items(self, parent):

   241

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

   242

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

   243

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

   244

        else:

   245

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

   246

   247

    def _init_coll_RightGridSizer_Growables(self, parent):

   248

        parent.AddGrowableCol(0)

   249

        parent.AddGrowableRow(1)

   250

   251

    def _init_coll_ButtonGridSizer_Items(self, parent):

   252

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

   253

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

   254

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

   255

   256

    def _init_coll_ButtonGridSizer_Growables(self, parent):

   257

        parent.AddGrowableCol(0)

   258

        parent.AddGrowableRow(0)

   259

   260

    def _init_sizers(self):

   261

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

   262

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

   263

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

   264

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

   265

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

   266

   267

        self._init_coll_LeftGridSizer_Growables(self.LeftGridSizer)

   268

        self._init_coll_LeftGridSizer_Items(self.LeftGridSizer)

   269

        self._init_coll_ButtonGridSizer_Growables(self.ButtonGridSizer)

   270

        self._init_coll_ButtonGridSizer_Items(self.ButtonGridSizer)

   271

        self._init_coll_RightGridSizer_Growables(self.RightGridSizer)

   272

        self._init_coll_RightGridSizer_Items(self.RightGridSizer)

   273

   274

        self.LeftPanel.SetSizer(self.LeftGridSizer)

   275

        self.RightPanel.SetSizer(self.RightGridSizer)

   276

        self.ParamsPanel.SetSizer(self.ParamsPanelMainSizer)

   277

   278

    def _init_ctrls(self, prnt):

   279

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

   280

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

   281

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

   282

        self._init_utils()

   283

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

   284

        self.SetMenuBar(self.menuBar1)

   285

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

   286

   287

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

   288

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

   289

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

   290

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

   291

            self.MainSplitter.SetNeedUpdating(True)

   292

            self.MainSplitter.SetMinimumPaneSize(1)

   293

   294

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

   295

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

   296

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

   297

   298

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

   299

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

   300

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

   301

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

   302

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

   303

                  id=ID_BEREMIZPLUGINTREE)

   304

   305

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

   306

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

   307

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

   308

   309

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

   310

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

   311

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

   312

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

   313

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

   314

                  id=ID_BEREMIZADDBUTTON)

   315

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

   316

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

   317

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

   318

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

   319

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

   320

                  id=ID_BEREMIZDELETEBUTTON)

   321

   322

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

   323

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

   324

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

   325

   326

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

   327

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

   328

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

   329

            self.SecondSplitter.SetNeedUpdating(True)

   330

            self.SecondSplitter.SetMinimumPaneSize(1)

   331

   332

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

   333

                  300)

   334

   335

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

   336

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

   337

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

   338

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

   339

   340

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

   341

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

   342

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

   343

   344

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

   345

                  -250)

   346

   347

            self._init_sizers()

   348

        else:

   349

            self.AUIManager = wx.aui.AuiManager(self)

   350

            self.AUIManager.SetDockSizeConstraint(0.5, 0.5)

   351

            self.Panes = {}

   352

   353

            self.PluginTree = CT.CustomTreeCtrl(id=ID_BEREMIZPLUGINTREE,

   354

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

   355

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

   356

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

   357

            self.PluginTree.Bind(wx.EVT_IDLE, self.OnPluginTreeIdle)

   358

            self.Bind(CT.EVT_TREE_SEL_CHANGED, self.OnPluginTreeItemSelected,

   359

                  id=ID_BEREMIZPLUGINTREE)

   360

            self.Bind(CT.EVT_TREE_ITEM_CHECKED, self.OnPluginTreeItemChecked,

   361

                  id=ID_BEREMIZPLUGINTREE)

   362

            self.Bind(wx.EVT_TREE_BEGIN_LABEL_EDIT, self.OnPluginTreeItemBeginEdit,

   363

                  id=ID_BEREMIZPLUGINTREE)

   364

            self.Bind(CT.EVT_TREE_END_LABEL_EDIT, self.OnPluginTreeItemEndEdit,

   365

                  id=ID_BEREMIZPLUGINTREE)

   366

            self.Bind(CT.EVT_TREE_ITEM_EXPANDED, self.OnPluginTreeItemExpanded,

   367

                  id=ID_BEREMIZPLUGINTREE)

   368

            self.Bind(CT.EVT_TREE_ITEM_COLLAPSED, self.OnPluginTreeItemCollapsed,

   369

                  id=ID_BEREMIZPLUGINTREE)

   370

   371

            self.AUIManager.AddPane(self.PluginTree, wx.aui.AuiPaneInfo().CenterPane())

   372

   373

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

   374

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

   375

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

   376

            self.AUIManager.AddPane(self.LogConsole, wx.aui.AuiPaneInfo().Caption("Log Console").Bottom().Layer(1).BestSize(wx.Size(800, 200)).CloseButton(False))

   377

   378

            self.AUIManager.Update()

   379

   380

    def __init__(self, parent, projectOpen):

   381

        self._init_ctrls(parent)

   382

   383

        self.Log = LogPseudoFile(self.LogConsole)

   384

   385

        self.PluginRoot = PluginsRoot(self)

   386

        self.DisableEvents = False

   387

   388

        if projectOpen:

   389

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

   390

            self.RefreshPluginTree()

   391

            self.RefreshPluginToolBar()

   392

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

   393

   394

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

   395

            self.RefreshPluginParams()

   396

            self.RefreshButtons()

   397

   398

        self.RefreshMainMenu()

   399

   400

    def OnFrameActivated(self, event):

   401

        if not event.GetActive():

   402

            self.PluginRoot.RefreshPluginsBlockLists()

   403

   404

    def RefreshButtons(self):

   405

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

   406

            if self.PluginRoot.HasProjectOpened():

   407

                self.PluginChilds.Enable(True)

   408

                self.AddButton.Enable(True)

   409

                self.DeleteButton.Enable(True)

   410

            else:

   411

                self.PluginChilds.Enable(False)

   412

                self.AddButton.Enable(False)

   413

                self.DeleteButton.Enable(False)

   414

   415

    def RefreshMainMenu(self):

   416

        if self.menuBar1:

   417

            if self.PluginRoot.HasProjectOpened():

   418

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

   419

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

   420

                self.FileMenu.Enable(ID_BEREMIZFILEMENUITEMS2, True)

   421

                self.FileMenu.Enable(ID_BEREMIZFILEMENUITEMS3, True)

   422

                self.FileMenu.Enable(ID_BEREMIZFILEMENUITEMS5, True)

   423

            else:

   424

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

   425

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

   426

                self.FileMenu.Enable(ID_BEREMIZFILEMENUITEMS2, False)

   427

                self.FileMenu.Enable(ID_BEREMIZFILEMENUITEMS3, False)

   428

                self.FileMenu.Enable(ID_BEREMIZFILEMENUITEMS5, False)

   429

   430

    def RefreshPluginTree(self):

   431

        self.DisableEvents = True

   432

        infos = self.PluginRoot.GetPlugInfos()

   433

        root = self.PluginTree.GetRootItem()

   434

        if root is None or not root.IsOk():

   435

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

   436

        last_selected = self.GetSelectedPluginName()

   437

        self.GenerateTreeBranch(root, infos, True)

   438

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

   439

        self.SelectedPluginByName(root, last_selected)

   440

   441

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

   442

            self.RefreshPluginParams()

   443

   444

        self.DisableEvents = False

   445

        self.MustRecalTreeSizes = True

   446

   447

    def SelectedPluginByName(self, root, name):

   448

        if name:

   449

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

   450

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

   451

            while item is not None and item.IsOk():

   452

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

   453

                    if len(toks)>1:

   454

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

   455

                    else:

   456

                        self.PluginTree.SelectItem(item, True)

   457

                        return True

   458

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

   459

        return False

   460

   461

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

   462

        to_delete = []

   463

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

   464

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

   465

   466

        if wx.VERSION >= (2, 8, 0):

   467

            plugin = self.GetSelectedPlugin(root)

   468

            if plugin is not None:

   469

                old_window = self.PluginTree.GetItemWindow(root)

   470

                if old_window is not None:

   471

                    old_window.GetSizer().Clear(True)

   472

                    old_window.Destroy()

   473

   474

                outside = wx.Panel(self.PluginTree, -1, size=wx.Size(-1, -1))

   475

                outside.SetBackgroundColour(wx.WHITE)

   476

   477

                insidesizer = wx.FlexGridSizer(cols=2,rows=1)

   478

                insidesizer.AddGrowableCol(1)

   479

   480

                outside.SetSizer(insidesizer)

   481

   482

                insidesizer.AddSpacer(-1)

   483

   484

                window = wx.Panel(outside, -1, size=wx.Size(-1, -1))

   485

                window.SetBackgroundColour(wx.Colour(250,250,255))

   486

                insidesizer.AddWindow(window, border=1, flag=wx.GROW|wx.ALL)

   487

   488

                tcsizer = wx.FlexGridSizer(cols=3, hgap=0, rows=1, vgap=0)

   489