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

    26

updateinfo_url = None

    27

    28

import os, sys, getopt, wx

    29

import __builtin__

    30

from wx.lib.agw.advancedsplash import AdvancedSplash

    31

import tempfile

    32

import shutil

    33

import random

    34

import time

    35

from types import ListType

    36

    37

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

    38

    39

def Bpath(*args):

    40

    return os.path.join(CWD,*args)

    41

    42

if __name__ == '__main__':

    43

    def usage():

    44

        print "\nUsage of Beremiz.py :"

    45

        print "\n   %s [Projectpath] [Buildpath]\n"%sys.argv[0]

    46

    47

    try:

    48

        opts, args = getopt.getopt(sys.argv[1:], "hu:e:", ["help", "updatecheck=", "extend="])

    49

    except getopt.GetoptError:

    50

        # print help information and exit:

    51

        usage()

    52

        sys.exit(2)

    53

    54

    extensions=[]

    55

    56

    for o, a in opts:

    57

        if o in ("-h", "--help"):

    58

            usage()

    59

            sys.exit()

    60

        if o in ("-u", "--updatecheck"):

    61

            updateinfo_url = a

    62

        if o in ("-e", "--extend"):

    63

            extensions.append(a)

    64

    65

    if len(args) > 2:

    66

        usage()

    67

        sys.exit()

    68

    elif len(args) == 1:

    69

        projectOpen = args[0]

    70

        buildpath = None

    71

    elif len(args) == 2:

    72

        projectOpen = args[0]

    73

        buildpath = args[1]

    74

    else:

    75

        projectOpen = None

    76

        buildpath = None

    77

    78

    if os.path.exists("BEREMIZ_DEBUG"):

    79

        __builtin__.__dict__["BMZ_DBG"] = True

    80

    else :

    81

        __builtin__.__dict__["BMZ_DBG"] = False

    82

    83

    app = wx.PySimpleApp(redirect=BMZ_DBG)

    84

    app.SetAppName('beremiz')

    85

    wx.InitAllImageHandlers()

    86

    87

    # popup splash

    88

    bmp = wx.Image(Bpath("images", "splash.png")).ConvertToBitmap()

    89

    #splash=AdvancedSplash(None, bitmap=bmp, style=wx.SPLASH_CENTRE_ON_SCREEN, timeout=4000)

    90

    splash=AdvancedSplash(None, bitmap=bmp)

    91

    wx.Yield()

    92

    93

    if updateinfo_url is not None:

    94

        updateinfo = "Fetching %s" % updateinfo_url

    95

        # warn for possible updates

    96

        def updateinfoproc():

    97

            global updateinfo

    98

            try :

    99

                import urllib2

   100

                updateinfo = urllib2.urlopen(updateinfo_url,None).read()

   101

            except :

   102

                updateinfo = "update info unavailable." 

   103

   104

        from threading import Thread

   105

        splash.SetText(text=updateinfo)

   106

        wx.Yield()

   107

        updateinfoThread = Thread(target=updateinfoproc)

   108

        updateinfoThread.start()

   109

        updateinfoThread.join(2)

   110

        splash.SetText(text=updateinfo)

   111

        wx.Yield()

   112

   113

from util.TranslationCatalogs import AddCatalog

   114

from util.BitmapLibrary import AddBitmapFolder, GetBitmap

   115

   116

AddCatalog(os.path.join(CWD, "locale"))

   117

AddBitmapFolder(os.path.join(CWD, "images"))

   118

   119

if __name__ == '__main__':

   120

    # Import module for internationalization

   121

    import gettext

   122

   123

    __builtin__.__dict__['_'] = wx.GetTranslation

   124

   125

    # Load extensions

   126

    for extfilename in extensions:

   127

        extension_folder = os.path.split(os.path.realpath(extfilename))[0]

   128

        sys.path.append(extension_folder)

   129

        AddCatalog(os.path.join(extension_folder, "locale"))

   130

        AddBitmapFolder(os.path.join(extension_folder, "images"))

   131

        execfile(extfilename, locals())

   132

   133

import wx.lib.buttons, wx.lib.statbmp

   134

import cPickle

   135

import types, time, re, platform, time, traceback, commands

   136

   137

from docutil import OpenHtmlFrame

   138

from IDEFrame import IDEFrame, AppendMenu

   139

from IDEFrame import TITLE, EDITORTOOLBAR, FILEMENU, EDITMENU, DISPLAYMENU, PROJECTTREE, POUINSTANCEVARIABLESPANEL, LIBRARYTREE, SCALING, PAGETITLES 

   140

from IDEFrame import EncodeFileSystemPath, DecodeFileSystemPath

   141

from editors.EditorPanel import EditorPanel

   142

from editors.Viewer import Viewer

   143

from editors.TextViewer import TextViewer

   144

from editors.GraphicViewer import GraphicViewer

   145

from editors.ResourceEditor import ConfigurationEditor, ResourceEditor

   146

from editors.DataTypeEditor import DataTypeEditor

   147

from util.MiniTextControler import MiniTextControler

   148

from util.ProcessLogger import ProcessLogger

   149

   150

from PLCControler import LOCATION_CONFNODE, LOCATION_MODULE, LOCATION_GROUP, LOCATION_VAR_INPUT, LOCATION_VAR_OUTPUT, LOCATION_VAR_MEMORY, ITEM_PROJECT, ITEM_RESOURCE

   151

from ProjectController import ProjectController, MATIEC_ERROR_MODEL, ITEM_CONFNODE

   152

   153

   154

MAX_RECENT_PROJECTS = 10

   155

   156

class GenStaticBitmap(wx.lib.statbmp.GenStaticBitmap):

   157

    """ Customized GenStaticBitmap, fix transparency redraw bug on wx2.8/win32, 

   158

    and accept image name as __init__ parameter, fail silently if file do not exist"""

   159

    def __init__(self, parent, ID, bitmapname,

   160

                 pos = wx.DefaultPosition, size = wx.DefaultSize,

   161

                 style = 0,

   162

                 name = "genstatbmp"):

   163

   164

        wx.lib.statbmp.GenStaticBitmap.__init__(self, parent, ID, 

   165

                 GetBitmap(bitmapname),

   166

                 pos, size,

   167

                 style,

   168

                 name)

   169

   170

    def OnPaint(self, event):

   171

        dc = wx.PaintDC(self)

   172

        colour = self.GetParent().GetBackgroundColour()

   173

        dc.SetPen(wx.Pen(colour))

   174

        dc.SetBrush(wx.Brush(colour ))

   175

        dc.DrawRectangle(0, 0, *dc.GetSizeTuple())

   176

        if self._bitmap:

   177

            dc.DrawBitmap(self._bitmap, 0, 0, True)

   178

   179

   180

from threading import Lock,Timer,currentThread

   181

MainThread = currentThread().ident

   182

REFRESH_PERIOD = 0.1

   183

from time import time as gettime

   184

class LogPseudoFile:

   185

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

   186

    def __init__(self, output, risecall):

   187

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

   188

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

   189

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

   190

        self.default_style = None

   191

        self.output = output

   192

        self.risecall = risecall

   193

        # to prevent rapid fire on rising log panel

   194

        self.rising_timer = 0

   195

        self.lock = Lock()

   196

        self.YieldLock = Lock()

   197

        self.RefreshLock = Lock()

   198

        self.TimerAccessLock = Lock()

   199

        self.stack = []

   200

        self.LastRefreshTime = gettime()

   201

        self.LastRefreshTimer = None

   202

   203

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

   204

        if self.lock.acquire():

   205

            self.stack.append((s,style))

   206

            self.lock.release()

   207

            current_time = gettime()

   208

            self.TimerAccessLock.acquire()

   209

            if self.LastRefreshTimer:

   210

                self.LastRefreshTimer.cancel()

   211

                self.LastRefreshTimer=None

   212

            self.TimerAccessLock.release()

   213

            if current_time - self.LastRefreshTime > REFRESH_PERIOD and self.RefreshLock.acquire(False):

   214

                self._should_write()

   215

            else:

   216

                self.TimerAccessLock.acquire()

   217

                self.LastRefreshTimer = Timer(REFRESH_PERIOD, self._timer_expired)

   218

                self.LastRefreshTimer.start()

   219

                self.TimerAccessLock.release()

   220

   221

    def _timer_expired(self):

   222

        if self.RefreshLock.acquire(False):

   223

            self._should_write()

   224

        else:

   225

            self.TimerAccessLock.acquire()

   226

            self.LastRefreshTimer = Timer(REFRESH_PERIOD, self._timer_expired)

   227

            self.LastRefreshTimer.start()

   228

            self.TimerAccessLock.release()

   229

   230

    def _should_write(self):

   231

        wx.CallAfter(self._write)

   232

        if MainThread == currentThread().ident:

   233

            app = wx.GetApp()

   234

            if app is not None:

   235

                if self.YieldLock.acquire(0):

   236

                    app.Yield()

   237

                    self.YieldLock.release()

   238

   239

    def _write(self):

   240

        if self.output :

   241

            self.output.Freeze()

   242

            self.lock.acquire()

   243

            for s, style in self.stack:

   244

                if style is None : style=self.black_white

   245

                if self.default_style != style: 

   246

                    self.output.SetDefaultStyle(style)

   247

                    self.default_style = style

   248

                self.output.AppendText(s)

   249

                self.output.ScrollLines(s.count('\n')+1)

   250

            self.stack = []

   251

            self.lock.release()

   252

            self.output.ShowPosition(self.output.GetLastPosition())

   253

            self.output.Thaw()

   254

            self.LastRefreshTime = gettime()

   255

            try:

   256

                self.RefreshLock.release()

   257

            except:

   258

                pass

   259

            newtime = time.time()

   260

            if newtime - self.rising_timer > 1:

   261

                self.risecall()

   262

            self.rising_timer = newtime

   263

   264

    def write_warning(self, s):

   265

        self.write(s,self.red_white)

   266

   267

    def write_error(self, s):

   268

        self.write(s,self.red_yellow)

   269

   270

    def writeyield(self, s):

   271

        self.write(s)

   272

        wx.GetApp().Yield()

   273

   274

    def flush(self):

   275

        self.output.SetValue("")

   276

   277

    def isatty(self):

   278

        return false

   279

   280

[ID_BEREMIZ, ID_BEREMIZMAINSPLITTER, 

   281

 ID_BEREMIZPLCCONFIG, ID_BEREMIZLOGCONSOLE, 

   282

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

   283

   284

[ID_FILEMENURECENTPROJECTS,

   285

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

   286

   287

CONFNODEMENU_POSITION = 3

   288

   289

class Beremiz(IDEFrame):

   290

   291

    def _init_utils(self):

   292

        self.ConfNodeMenu = wx.Menu(title='')

   293

        self.RecentProjectsMenu = wx.Menu(title='')

   294

   295

        IDEFrame._init_utils(self)

   296

   297

    def _init_coll_FileMenu_Items(self, parent):

   298

        AppendMenu(parent, help='', id=wx.ID_NEW,

   299

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

   300

        AppendMenu(parent, help='', id=wx.ID_OPEN,

   301

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

   302

        parent.AppendMenu(ID_FILEMENURECENTPROJECTS, _("&Recent Projects"), self.RecentProjectsMenu)

   303

        parent.AppendSeparator()

   304

        AppendMenu(parent, help='', id=wx.ID_SAVE,

   305

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

   306

        AppendMenu(parent, help='', id=wx.ID_SAVEAS,

   307

              kind=wx.ITEM_NORMAL, text=_(u'Save as') + '\tCTRL+SHIFT+S')

   308

        AppendMenu(parent, help='', id=wx.ID_CLOSE,

   309

              kind=wx.ITEM_NORMAL, text=_(u'Close Tab') + '\tCTRL+W')

   310

        AppendMenu(parent, help='', id=wx.ID_CLOSE_ALL,

   311

              kind=wx.ITEM_NORMAL, text=_(u'Close Project') + '\tCTRL+SHIFT+W')

   312

        parent.AppendSeparator()

   313

        AppendMenu(parent, help='', id=wx.ID_PAGE_SETUP,

   314

              kind=wx.ITEM_NORMAL, text=_(u'Page Setup') + '\tCTRL+ALT+P')

   315

        AppendMenu(parent, help='', id=wx.ID_PREVIEW,

   316

              kind=wx.ITEM_NORMAL, text=_(u'Preview') + '\tCTRL+SHIFT+P')

   317

        AppendMenu(parent, help='', id=wx.ID_PRINT,

   318

              kind=wx.ITEM_NORMAL, text=_(u'Print') + '\tCTRL+P')

   319

        parent.AppendSeparator()

   320

        AppendMenu(parent, help='', id=wx.ID_EXIT,

   321

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

   322

   323

        self.Bind(wx.EVT_MENU, self.OnNewProjectMenu, id=wx.ID_NEW)

   324

        self.Bind(wx.EVT_MENU, self.OnOpenProjectMenu, id=wx.ID_OPEN)

   325

        self.Bind(wx.EVT_MENU, self.OnSaveProjectMenu, id=wx.ID_SAVE)

   326

        self.Bind(wx.EVT_MENU, self.OnSaveProjectAsMenu, id=wx.ID_SAVEAS)

   327

        self.Bind(wx.EVT_MENU, self.OnCloseTabMenu, id=wx.ID_CLOSE)

   328

        self.Bind(wx.EVT_MENU, self.OnCloseProjectMenu, id=wx.ID_CLOSE_ALL)

   329

        self.Bind(wx.EVT_MENU, self.OnPageSetupMenu, id=wx.ID_PAGE_SETUP)

   330

        self.Bind(wx.EVT_MENU, self.OnPreviewMenu, id=wx.ID_PREVIEW)

   331

        self.Bind(wx.EVT_MENU, self.OnPrintMenu, id=wx.ID_PRINT)

   332

        self.Bind(wx.EVT_MENU, self.OnQuitMenu, id=wx.ID_EXIT)

   333

   334

        self.AddToMenuToolBar([(wx.ID_NEW, "new", _(u'New'), None),

   335

                               (wx.ID_OPEN, "open", _(u'Open'), None),

   336

                               (wx.ID_SAVE, "save", _(u'Save'), None),

   337

                               (wx.ID_SAVEAS, "saveas", _(u'Save As...'), None),

   338

                               (wx.ID_PRINT, "print", _(u'Print'), None)])

   339

   340

    def _init_coll_AddMenu_Items(self, parent):

   341

        IDEFrame._init_coll_AddMenu_Items(self, parent, False)

   342

        new_id = wx.NewId()

   343

        AppendMenu(parent, help='', id=new_id,

   344

                  kind=wx.ITEM_NORMAL, text=_(u'&Resource'))

   345

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

   346

        for name, XSDClass, help in ProjectController.CTNChildrenTypes:

   347

            new_id = wx.NewId()

   348

            AppendMenu(parent, help='', id=new_id, 

   349

                       kind=wx.ITEM_NORMAL, text=help)

   350

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

   351

   352

    def _init_coll_HelpMenu_Items(self, parent):

   353

        parent.Append(help='', id=wx.ID_ABOUT,

   354

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

   355

        self.Bind(wx.EVT_MENU, self.OnAboutMenu, id=wx.ID_ABOUT)

   356

   357

    def _init_ctrls(self, prnt):

   358

        IDEFrame._init_ctrls(self, prnt)

   359

   360

        self.EditMenuSize = self.EditMenu.GetMenuItemCount()

   361

   362

        self.Bind(wx.EVT_MENU, self.OnOpenWidgetInspector, id=ID_BEREMIZINSPECTOR)

   363

        accels = [wx.AcceleratorEntry(wx.ACCEL_CTRL|wx.ACCEL_ALT, ord('I'), ID_BEREMIZINSPECTOR)]

   364

        for method,shortcut in [("Stop",     wx.WXK_F4),

   365

                                ("Run",      wx.WXK_F5),

   366

                                ("Transfer", wx.WXK_F6),

   367

                                ("Connect",  wx.WXK_F7),

   368

                                ("Build",    wx.WXK_F11)]:

   369

            def OnMethodGen(obj,meth):

   370

                def OnMethod(evt):

   371

                    if obj.CTR is not None:

   372

                       obj.CTR.CallMethod('_'+meth)

   373

                    wx.CallAfter(self.RefreshStatusToolBar)

   374

                return OnMethod

   375

            newid = wx.NewId()

   376

            self.Bind(wx.EVT_MENU, OnMethodGen(self,method), id=newid)

   377

            accels += [wx.AcceleratorEntry(wx.ACCEL_NORMAL, shortcut,newid)]

   378

   379

        self.SetAcceleratorTable(wx.AcceleratorTable(accels))

   380

   381

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

   382

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

   383

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

   384

        self.LogConsole.Bind(wx.EVT_LEFT_DCLICK, self.OnLogConsoleDClick)

   385

        self.MainTabs["LogConsole"] = (self.LogConsole, _("Log Console"))

   386

        self.BottomNoteBook.AddPage(*self.MainTabs["LogConsole"])

   387

        #self.BottomNoteBook.Split(self.BottomNoteBook.GetPageIndex(self.LogConsole), wx.RIGHT)

   388

   389

        StatusToolBar = wx.ToolBar(self, -1, wx.DefaultPosition, wx.DefaultSize,

   390

                wx.TB_FLAT | wx.TB_NODIVIDER | wx.NO_BORDER)

   391

        StatusToolBar.SetToolBitmapSize(wx.Size(25, 25))

   392

        StatusToolBar.Realize()

   393

        self.Panes["StatusToolBar"] = StatusToolBar

   394

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

   395

                  Name("StatusToolBar").Caption(_("Status ToolBar")).

   396

                  ToolbarPane().Top().Position(1).

   397

                  LeftDockable(False).RightDockable(False))

   398

   399

        self.AUIManager.Update()

   400

   401

    def __init__(self, parent, projectOpen=None, buildpath=None, ctr=None, debug=True):

   402

        IDEFrame.__init__(self, parent, debug)

   403

        self.Log = LogPseudoFile(self.LogConsole,self.RiseLogConsole)

   404

   405

        self.local_runtime = None

   406

        self.runtime_port = None

   407

        self.local_runtime_tmpdir = None

   408

   409

        self.LastPanelSelected = None

   410

   411

        # Define Tree item icon list

   412

        self.LocationImageList = wx.ImageList(16, 16)

   413

        self.LocationImageDict = {}

   414

   415

        # Icons for location items

   416

        for imgname, itemtype in [

   417

            ("CONFIGURATION", LOCATION_CONFNODE),

   418

            ("RESOURCE",      LOCATION_MODULE),

   419

            ("PROGRAM",       LOCATION_GROUP),

   420

            ("VAR_INPUT",     LOCATION_VAR_INPUT),

   421

            ("VAR_OUTPUT",    LOCATION_VAR_OUTPUT),

   422

            ("VAR_LOCAL",     LOCATION_VAR_MEMORY)]:

   423

            self.LocationImageDict[itemtype] = self.LocationImageList.Add(GetBitmap(imgname))

   424

   425

        # Icons for other items

   426

        for imgname, itemtype in [

   427

            ("Extension", ITEM_CONFNODE)]:

   428

            self.TreeImageDict[itemtype] = self.TreeImageList.Add(GetBitmap(imgname))

   429

   430

        # Add beremiz's icon in top left corner of the frame

   431

        self.SetIcon(wx.Icon(Bpath("images", "brz.ico"), wx.BITMAP_TYPE_ICO))

   432

   433

        if projectOpen is not None:

   434

            projectOpen = DecodeFileSystemPath(projectOpen, False)

   435

   436

        if (self.EnableSaveProjectState() and ctr is None and 

   437

            projectOpen is None and self.Config.HasEntry("currenteditedproject")):

   438

            try:

   439

                projectOpen = DecodeFileSystemPath(self.Config.Read("currenteditedproject"))

   440

                if projectOpen == "":

   441

                    projectOpen = None

   442

            except:

   443

                projectOpen = None

   444

   445

        if projectOpen is not None and os.path.isdir(projectOpen):

   446

            self.CTR = ProjectController(self, self.Log)

   447

            self.Controler = self.CTR

   448

            result = self.CTR.LoadProject(projectOpen, buildpath)

   449

            if not result:

   450

                self.LibraryPanel.SetController(self.Controler)

   451

                self.ProjectTree.Enable(True)

   452

                self.PouInstanceVariablesPanel.SetController(self.Controler)

   453

                self.RefreshConfigRecentProjects(os.path.abspath(projectOpen))

   454

                self._Refresh(PROJECTTREE, POUINSTANCEVARIABLESPANEL, LIBRARYTREE)

   455

            else:

   456

                self.ResetView()

   457

                self.ShowErrorMessage(result)

   458

        else:

   459

            self.CTR = ctr

   460

            self.Controler = ctr

   461

            if ctr is not None:

   462

                self.LibraryPanel.SetController(self.Controler)

   463

                self.ProjectTree.Enable(True)

   464

                self.PouInstanceVariablesPanel.SetController(self.Controler)

   465

                self._Refresh(PROJECTTREE, POUINSTANCEVARIABLESPANEL, LIBRARYTREE)

   466

        if self.EnableDebug:

   467

            self.DebugVariablePanel.SetDataProducer(self.CTR)

   468

   469

        self.Bind(wx.EVT_CLOSE, self.OnCloseFrame)

   470

   471

        self._Refresh(TITLE, EDITORTOOLBAR, FILEMENU, EDITMENU, DISPLAYMENU)

   472

        self.RefreshAll()

   473

        self.LogConsole.SetFocus()

   474

   475

    def RiseLogConsole(self):

   476

        self.BottomNoteBook.SetSelection(self.BottomNoteBook.GetPageIndex(self.LogConsole))

   477

   478

    def RefreshTitle(self):

   479

        name = _("Beremiz")

   480

        if self.CTR is not None:

   481

            projectname = self.CTR.GetProjectName()

   482

            if self.CTR.ProjectTestModified():

   483

                projectname = "~%s~" % projectname

   484

            self.SetTitle("%s - %s" % (name, projectname))