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

    30

    31

import types

    32

    33

import time

    34

    35

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

    36

    37

from plugger import PluginsRoot

    38

    39

from wxPopen import wxPopen3

    40

    41

SCROLLBAR_UNIT = 10

    42

WINDOW_COLOUR = wx.Colour(240,240,240)

    44

    45

if wx.Platform == '__WXMSW__':

    46

    faces = { 'times': 'Times New Roman',

    47

              'mono' : 'Courier New',

    48

              'helv' : 'Arial',

    49

              'other': 'Comic Sans MS',

    50

              'size' : 16,

    51

             }

    52

else:

    53

    faces = { 'times': 'Times',

    54

              'mono' : 'Courier',

    55

              'helv' : 'Helvetica',

    56

              'other': 'new century schoolbook',

    57

              'size' : 18,

    58

             }

    59

    60

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

    61

    63

gen_mini_GetLabelSize = lambda obj:lambda:(wx.lib.buttons.GenBitmapTextButton._GetLabelSize(obj)[:-1] + (False,))

    64

gen_textbutton_GetLabelSize = lambda obj:lambda:(wx.lib.buttons.GenButton._GetLabelSize(obj)[:-1] + (False,))

    65

   106

   107

class LogPseudoFile:

   108

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

   109

    def __init__(self, output = None):

   110

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

   111

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

   112

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

   113

        self.default_style = None

   114

        self.output = output

   115

   116

    def writelines(self, l):

   117

        map(self.write, l)

   118

   119

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

   120

        if not style : style=self.black_white    

   121

        if self.default_style != style: 

   122

            self.output.SetDefaultStyle(style)

   123

            self.default_style = style

   124

        self.output.AppendText(s) 

   125

   126

    def write_warning(self, s):

   127

        self.write(s,self.red_white)

   128

   129

    def write_error(self, s):

   130

        self.write(s,self.red_yellow)

   131

   132

    def flush(self):

   133

        self.output.SetValue("")

   134

   135

    def isatty(self):

   136

        return false

   137

   138

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

   139

        self.errlen = 0

   140

        self.exitcode = None

   141

        self.outdata = ""

   142

        self.errdata = ""

   143

   144

        def output(v):

   145

            self.outdata += v

   146

            if not no_stdout:

   147

                self.write(v)

   148

   149

        def errors(v):

   150

            self.errdata += v

   151

            self.errlen += 1

   152

            if self.errlen > sz_limit:

   153

                p.kill()

   154

            self.write_warning(v)

   155

   156

        def fin(pid,ecode):

   157

            self.exitcode = ecode

   158

            if self.exitcode != 0:

   159

                self.write(Command + "\n")

   160

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

   161

   162

        def spin(p):

   163

            while not p.finished:

   164

                wx.Yield()

   165

                time.sleep(0.01)

   166

   167

        input = []

   168

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

   169

        if p.pid:

   170

        	spin(p)

   171

   172

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

   173

   174

[ID_BEREMIZ, ID_BEREMIZMAINSPLITTER, 

   175

 ID_BEREMIZPLCCONFIG, ID_BEREMIZLOGCONSOLE, 

   176

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

   177

   178

[ID_BEREMIZFILEMENUITEMS0, ID_BEREMIZFILEMENUITEMS1, 

   179

 ID_BEREMIZFILEMENUITEMS2, ID_BEREMIZFILEMENUITEMS3, 

   180

 ID_BEREMIZFILEMENUITEMS5, ID_BEREMIZFILEMENUITEMS7, 

   181

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

   182

   183

[ID_BEREMIZEDITMENUITEMS0, ID_BEREMIZEDITMENUITEMS2, 

   184

 ID_BEREMIZEDITMENUITEMS3, 

   185

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

   186

   187

[ID_BEREMIZRUNMENUITEMS0, ID_BEREMIZRUNMENUITEMS2, 

   188

 ID_BEREMIZRUNMENUITEMS3, ID_BEREMIZRUNMENUITEMS5, 

   189

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

   190

   191

[ID_BEREMIZHELPMENUITEMS0, ID_BEREMIZHELPMENUITEMS1, 

   192

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

   193

   194

class Beremiz(wx.Frame):

   195

   196

    def _init_coll_FileMenu_Items(self, parent):

   197

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

   198

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

   199

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

   200

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

   201

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

   202

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

   203

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

   204

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

   205

        parent.AppendSeparator()

   206

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

   207

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

   208

        parent.AppendSeparator()

   209

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

   210

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

   211

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

   212

              id=ID_BEREMIZFILEMENUITEMS0)

   213

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

   214

              id=ID_BEREMIZFILEMENUITEMS1)

   215

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

   216

              id=ID_BEREMIZFILEMENUITEMS2)

   217

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

   218

              id=ID_BEREMIZFILEMENUITEMS3)

   219

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

   220

              id=ID_BEREMIZFILEMENUITEMS5)

   221

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

   222

              id=ID_BEREMIZFILEMENUITEMS7)

   223

   224

    def _init_coll_EditMenu_Items(self, parent):

   225

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

   226

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

   227

        parent.AppendSeparator()

   228

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

   229

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

   230

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

   231

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

   232

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

   233

              id=ID_BEREMIZEDITMENUITEMS0)

   234

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

   235

              id=ID_BEREMIZEDITMENUITEMS2)

   236

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

   237

              id=ID_BEREMIZEDITMENUITEMS3)

   238

   239

    def _init_coll_RunMenu_Items(self, parent):

   240

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

   241

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

   242

        parent.AppendSeparator()

   243

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

   244

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

   245

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

   246

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

   247

        parent.AppendSeparator()

   248

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

   249

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

   250

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

   251

              id=ID_BEREMIZRUNMENUITEMS0)

   252

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

   253

              id=ID_BEREMIZRUNMENUITEMS2)

   254

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

   255

              id=ID_BEREMIZRUNMENUITEMS3)

   256

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

   257

              id=ID_BEREMIZRUNMENUITEMS5)

   258

   259

    def _init_coll_HelpMenu_Items(self, parent):

   260

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

   261

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

   262

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

   263

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

   264

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

   265

              id=ID_BEREMIZHELPMENUITEMS0)

   266

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

   267

              id=ID_BEREMIZHELPMENUITEMS1)

   268

   269

    def _init_coll_menuBar1_Menus(self, parent):

   270

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

   271

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

   272

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

   273

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

   274

   275

    def _init_utils(self):

   276

        self.menuBar1 = wx.MenuBar()

   277

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

   278

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

   279

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

   280

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

   281

   282

        self._init_coll_menuBar1_Menus(self.menuBar1)

   283

        self._init_coll_FileMenu_Items(self.FileMenu)

   284

        #self._init_coll_EditMenu_Items(self.EditMenu)

   285

        #self._init_coll_RunMenu_Items(self.RunMenu)

   286

        self._init_coll_HelpMenu_Items(self.HelpMenu)

   287

   288

    def _init_coll_PLCConfigMainSizer_Items(self, parent):

   289

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

   290

        parent.AddSizer(self.PluginTreeSizer, 0, border=10, flag=wx.BOTTOM|wx.LEFT|wx.RIGHT)

   291

   292

    def _init_coll_PLCConfigMainSizer_Growables(self, parent):

   293

        parent.AddGrowableCol(0)

   294

        parent.AddGrowableRow(1)

   295

   296

    def _init_coll_PluginTreeSizer_Growables(self, parent):

   297

        parent.AddGrowableCol(0)

   298

        parent.AddGrowableCol(1)

   299

   300

    def _init_sizers(self):

   301

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

   302

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

   305

   306

        self._init_coll_PLCConfigMainSizer_Items(self.PLCConfigMainSizer)

   307

        self._init_coll_PLCConfigMainSizer_Growables(self.PLCConfigMainSizer)

   308

        self._init_coll_PluginTreeSizer_Growables(self.PluginTreeSizer)

   309

   310

        self.PLCConfig.SetSizer(self.PLCConfigMainSizer)

   311

   312

    def _init_ctrls(self, prnt):

   313

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

   314

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

   315

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

   316

        self._init_utils()

   317

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

   318

        self.SetMenuBar(self.menuBar1)

   319

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

   320

   321

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

   322

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

   323

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

   324

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

   325

            self.MainSplitter.SetNeedUpdating(True)

   326

            self.MainSplitter.SetMinimumPaneSize(1)

   327

   328

            parent = self.MainSplitter

   329

        else:

   330

            parent = self

   331

   332

        self.PLCConfig = wx.ScrolledWindow(id=ID_BEREMIZPLCCONFIG,

   333

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

   334

              size=wx.Size(-1, -1), style=wx.TAB_TRAVERSAL|wx.SUNKEN_BORDER|wx.HSCROLL|wx.VSCROLL)

   335

        self.PLCConfig.SetBackgroundColour(wx.WHITE)

   336

        self.PLCConfig.Bind(wx.EVT_SIZE, self.OnMoveWindow)

   337

   338

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

   339

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

   340

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

   341

   342

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

   343

            self.MainSplitter.SplitHorizontally(self.PLCConfig, self.LogConsole, -250)

   344

   345

        else:

   346

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

   347

            self.AUIManager.SetDockSizeConstraint(0.5, 0.5)

   348

   349

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

   350

   351

            self.AUIManager.AddPane(self.LogConsole, wx.aui.AuiPaneInfo().

   352

                Caption("Log Console").Bottom().Layer(1).

   353

                BestSize(wx.Size(800, 200)).CloseButton(False))

   354

   355

            self.AUIManager.Update()

   356

   357

        self._init_sizers()

   358

   359

    def __init__(self, parent, projectOpen):

   360

        self._init_ctrls(parent)

   361

   362

        self.Log = LogPseudoFile(self.LogConsole)

   363

   364

        self.PluginRoot = PluginsRoot(self)

   365

        self.DisableEvents = False

   366

   367

        self.PluginInfos = {}

   368

   369

        if projectOpen:

   370

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

   371

            self.RefreshPLCParams()

   372

            self.RefreshPluginTree()

   373

   374

        self.RefreshMainMenu()

   375

   376

    def OnMoveWindow(self, event):

   377

        self.GetBestSize()

   378

        self.RefreshScrollBars()

   379

        event.Skip()

   380

   381

    def OnFrameActivated(self, event):

   382

        if not event.GetActive():

   383

            self.PluginRoot.RefreshPluginsBlockLists()

   384

   385

    def RefreshMainMenu(self):

   386

        if self.menuBar1:

   387

            if self.PluginRoot.HasProjectOpened():

   388

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

   389

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

   390

                self.FileMenu.Enable(ID_BEREMIZFILEMENUITEMS2, True)

   391

                self.FileMenu.Enable(ID_BEREMIZFILEMENUITEMS3, True)

   392

                self.FileMenu.Enable(ID_BEREMIZFILEMENUITEMS5, True)

   393

            else:

   394

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

   395

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

   396

                self.FileMenu.Enable(ID_BEREMIZFILEMENUITEMS2, False)

   397

                self.FileMenu.Enable(ID_BEREMIZFILEMENUITEMS3, False)

   398

                self.FileMenu.Enable(ID_BEREMIZFILEMENUITEMS5, False)

   399

   400

    def RefreshScrollBars(self):

   401

        xstart, ystart = self.PLCConfig.GetViewStart()

   402

        window_size = self.PLCConfig.GetClientSize()

   403

        sizer = self.PLCConfig.GetSizer()

   404

        if sizer:

   405

            maxx, maxy = sizer.GetMinSize()

   406

            self.PLCConfig.SetScrollbars(SCROLLBAR_UNIT, SCROLLBAR_UNIT, 

   407

                maxx / SCROLLBAR_UNIT, maxy / SCROLLBAR_UNIT, xstart, ystart)

   408

   409

    def RefreshPLCParams(self):

   410

        self.ClearSizer(self.PLCParamsSizer)

   411

   412

        if self.PluginRoot.HasProjectOpened():

   413

            plcwindow = wx.Panel(self.PLCConfig, -1, size=wx.Size(-1, -1))

   415

            self.PLCParamsSizer.AddWindow(plcwindow, 0, border=0, flag=wx.GROW)

   416

   417

            plcwindowsizer = wx.BoxSizer(wx.HORIZONTAL)

   418

            plcwindow.SetSizer(plcwindowsizer)

   419

   420

            st = wx.StaticText(plcwindow, -1)

   421

            st.SetFont(wx.Font(faces["size"], wx.DEFAULT, wx.NORMAL, wx.BOLD, faceName = faces["helv"]))

   422

            st.SetLabel(self.PluginRoot.GetProjectName())

   423

            plcwindowsizer.AddWindow(st, 0, border=5, flag=wx.ALL|wx.ALIGN_CENTER)

   424

   433

            plcwindowmainsizer = wx.BoxSizer(wx.VERTICAL)

   434

            plcwindowsizer.AddSizer(plcwindowmainsizer, 0, border=5, flag=wx.ALL)

   435

   436

            plcwindowbuttonsizer = wx.BoxSizer(wx.HORIZONTAL)

   437

            plcwindowmainsizer.AddSizer(plcwindowbuttonsizer, 0, border=0, flag=wx.ALIGN_CENTER)

   438

   439

            msizer = self.GenerateMethodButtonSizer(self.PluginRoot, plcwindow)

   440

            plcwindowbuttonsizer.AddSizer(msizer, 0, border=0, flag=wx.GROW)

   441

   442

            paramswindow = wx.Panel(plcwindow, -1, size=wx.Size(-1, -1))

   444

            plcwindowbuttonsizer.AddWindow(paramswindow, 0, border=0, flag=0)

   445

   446

            psizer = wx.BoxSizer(wx.HORIZONTAL)

   447

            paramswindow.SetSizer(psizer)

   448

   449

            plugin_infos = self.PluginRoot.GetParamsAttributes()

   450

            self.RefreshSizerElement(paramswindow, psizer, self.PluginRoot, plugin_infos, None, False)

   451

   452

            paramswindow.Hide()

   453

   454

            minimizebutton_id = wx.NewId()

   455

            minimizebutton = wx.lib.buttons.GenBitmapToggleButton(id=minimizebutton_id, bitmap=wx.Bitmap(os.path.join(CWD, 'images', 'Maximize.png')),

   456

                  name='MinimizeButton', parent=plcwindow, pos=wx.Point(0, 0),

   457

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

   459

            minimizebutton.SetBitmapSelected(wx.Bitmap(os.path.join(CWD, 'images', 'Minimize.png')))

   460

            plcwindowbuttonsizer.AddWindow(minimizebutton, 0, border=5, flag=wx.ALL)

   461

   467

   468

            def togglewindow(event):

   469

                if minimizebutton.GetToggle():

   470

                    paramswindow.Show()

   471

                    msizer.SetCols(1)

   474

                else:

   475

                    paramswindow.Hide()

   476

                    msizer.SetCols(len(self.PluginRoot.PluginMethods))

   479

                self.PLCConfigMainSizer.Layout()

   480

                self.RefreshScrollBars()

   481

                event.Skip()

   482

            minimizebutton.Bind(wx.EVT_BUTTON, togglewindow, id=minimizebutton_id)

   483

   484

        self.PLCConfigMainSizer.Layout()

   485

        self.RefreshScrollBars()

   486