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.misc import InstallLocalRessources

   114

    InstallLocalRessources(CWD)

   115

   116

    # Load extensions

   117

    for extfilename in extensions:

   120

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

   121

        sys.path.append(extension_folder)

   122

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

   123

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

   124

        execfile(extfilename, locals())

   125

   126

   127

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

   128

import cPickle

   129

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

   130

   131

from docutil import OpenHtmlFrame

   132

from editors.EditorPanel import EditorPanel

   133

from editors.Viewer import Viewer

   134

from editors.TextViewer import TextViewer

   135

from editors.ResourceEditor import ConfigurationEditor, ResourceEditor

   136

from editors.DataTypeEditor import DataTypeEditor

   137

from util.MiniTextControler import MiniTextControler

   138

from util.ProcessLogger import ProcessLogger

   139

from controls.LogViewer import LogViewer

   140

from controls.CustomStyledTextCtrl import CustomStyledTextCtrl

   141

   142

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

   143

from ProjectController import ProjectController, GetAddMenuItems, MATIEC_ERROR_MODEL, ITEM_CONFNODE

   144

   145

   146

MAX_RECENT_PROJECTS = 10

   147

   148

if wx.Platform == '__WXMSW__':

   149

    faces = {

   150

        'mono' : 'Courier New',

   151

        'size' : 8,

   152

    }

   153

else:

   154

    faces = {

   155

        'mono' : 'Courier',

   156

        'size' : 10,

   157

    }

   158

   159

from threading import Lock,Timer,currentThread

   160

MainThread = currentThread().ident

   161

REFRESH_PERIOD = 0.1

   162

from time import time as gettime

   163

class LogPseudoFile:

   164

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

   165

    def __init__(self, output, risecall):

   166

        self.red_white = 1

   167

        self.red_yellow = 2

   168

        self.black_white = wx.stc.STC_STYLE_DEFAULT

   169

        self.output = output

   170

        self.risecall = risecall

   171

        # to prevent rapid fire on rising log panel

   172

        self.rising_timer = 0

   173

        self.lock = Lock()

   174

        self.YieldLock = Lock()

   175

        self.RefreshLock = Lock()

   176

        self.TimerAccessLock = Lock()

   177

        self.stack = []

   178

        self.LastRefreshTime = gettime()

   179

        self.LastRefreshTimer = None

   180

   181

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

   182

        if self.lock.acquire():

   183

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

   184

            self.lock.release()

   185

            current_time = gettime()

   186

            self.TimerAccessLock.acquire()

   187

            if self.LastRefreshTimer:

   188

                self.LastRefreshTimer.cancel()

   189

                self.LastRefreshTimer=None

   190

            self.TimerAccessLock.release()

   191

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

   192

                self._should_write()

   193

            else:

   194

                self.TimerAccessLock.acquire()

   195

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

   196

                self.LastRefreshTimer.start()

   197

                self.TimerAccessLock.release()

   198

   199

    def _timer_expired(self):

   200

        if self.RefreshLock.acquire(False):

   201

            self._should_write()

   202

        else:

   203

            self.TimerAccessLock.acquire()

   204

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

   205

            self.LastRefreshTimer.start()

   206

            self.TimerAccessLock.release()

   207

   208

    def _should_write(self):

   209

        wx.CallAfter(self._write)

   210

        if MainThread == currentThread().ident:

   211

            app = wx.GetApp()

   212

            if app is not None:

   213

                if self.YieldLock.acquire(0):

   214

                    app.Yield()

   215

                    self.YieldLock.release()

   216

   217

    def _write(self):

   218

        if self.output :

   219

            self.output.Freeze()

   220

            self.lock.acquire()

   221

            for s, style in self.stack:

   222

                if style is None : style=self.black_white

   223

                if style != self.black_white:

   224

                    self.output.StartStyling(self.output.GetLength(), 0xff)

   225

   226

                # Temporary deactivate read only mode on StyledTextCtrl for

   227

                # adding text. It seems that text modifications, even 

   228

                # programmatically, are disabled in StyledTextCtrl when read

   229

                # only is active 

   230

                self.output.SetReadOnly(False)

   231

                self.output.AppendText(s)

   232

                self.output.SetReadOnly(True)

   233

   234

                if style != self.black_white:

   235

                    self.output.SetStyling(len(s), style)

   236

            self.stack = []

   237

            self.lock.release()

   238

            self.output.Thaw()

   239

            self.LastRefreshTime = gettime()

   240

            try:

   241

                self.RefreshLock.release()

   242

            except:

   243

                pass

   244

            newtime = time.time()

   245

            if newtime - self.rising_timer > 1:

   246

                self.risecall(self.output)

   247

            self.rising_timer = newtime

   248

   249

    def write_warning(self, s):

   250

        self.write(s,self.red_white)

   251

   252

    def write_error(self, s):

   253

        self.write(s,self.red_yellow)

   254

   255

    def writeyield(self, s):

   256

        self.write(s)

   257

        wx.GetApp().Yield()

   258

   259

    def flush(self):

   260

        # Temporary deactivate read only mode on StyledTextCtrl for clearing

   261

        # text. It seems that text modifications, even programmatically, are

   262

        # disabled in StyledTextCtrl when read only is active 

   263

        self.output.SetReadOnly(False)

   264

        self.output.SetText("")

   265

        self.output.SetReadOnly(True)

   266

   267

    def isatty(self):

   268

        return False

   269

   270

ID_FILEMENURECENTPROJECTS = wx.NewId()

   271

   272

from IDEFrame import TITLE,\

   273

                     EDITORTOOLBAR,\

   274

                     FILEMENU,\

   275

                     EDITMENU,\

   276

                     DISPLAYMENU,\

   277

                     PROJECTTREE,\

   278

                     POUINSTANCEVARIABLESPANEL,\

   279

                     LIBRARYTREE,\

   280

                     SCALING,\

   281

                     PAGETITLES,\

   282

                     IDEFrame, AppendMenu,\

   283

                     EncodeFileSystemPath, DecodeFileSystemPath

   284

from util.BitmapLibrary import GetBitmap

   285

   286

class Beremiz(IDEFrame):

   287

   288

    def _init_utils(self):

   289

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

   290

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

   291

   292

        IDEFrame._init_utils(self)

   293

   294

    def _init_coll_FileMenu_Items(self, parent):

   295

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

   296

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

   297

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

   298

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

   299

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

   300

        parent.AppendSeparator()

   301

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

   302

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

   303

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

   304

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

   305

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

   306

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

   307

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

   308

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

   309

        parent.AppendSeparator()

   310

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

   311

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

   312

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

   313

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

   314

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

   315

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

   316

        parent.AppendSeparator()

   317

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

   318

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

   319

   320

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

   321

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

   322

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

   323

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

   324

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

   325

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

   326

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

   327

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

   328

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

   329

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

   330

   331

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

   332

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

   333

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

   334

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

   335

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

   336

   337

    def _RecursiveAddMenuItems(self, menu, items):

   338

        for name, text, help, children in items:

   339

            new_id = wx.NewId()

   340

            if len(children) > 0:

   341

                new_menu = wx.Menu(title='')

   342

                menu.AppendMenu(new_id, text, new_menu)

   343

                self._RecursiveAddMenuItems(new_menu, children)

   344

            else:

   345

                AppendMenu(menu, help=help, id=new_id, 

   346

                       kind=wx.ITEM_NORMAL, text=text)

   347

                self.Bind(wx.EVT_MENU, self.GetAddConfNodeFunction(name), 

   348

                          id=new_id)    

   349

   350

    def _init_coll_AddMenu_Items(self, parent):

   351

        IDEFrame._init_coll_AddMenu_Items(self, parent, False)

   352

   353

        # Disable add resource until matiec is able to handle multiple ressource definition

   354

        #new_id = wx.NewId()

   355

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

   356

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

   357

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

   358

   359

        self._RecursiveAddMenuItems(parent, GetAddMenuItems())

   360

   361

    def _init_coll_HelpMenu_Items(self, parent):

   362

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

   363

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

   364

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

   365

   366

    def _init_coll_ConnectionStatusBar_Fields(self, parent):

   367

        parent.SetFieldsCount(3)

   368

   369

        parent.SetStatusText(number=0, text='')

   370

        parent.SetStatusText(number=1, text='')

   371

        parent.SetStatusText(number=2, text='')

   372

   373

        parent.SetStatusWidths([-1, 300, 200])

   374

   375

    def _init_ctrls(self, prnt):

   376

        IDEFrame._init_ctrls(self, prnt)

   377

   378

        self.EditMenuSize = self.EditMenu.GetMenuItemCount()

   379

   380

        inspectorID = wx.NewId()

   381

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

   382

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

   383

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

   384

                                ("Run",      wx.WXK_F5),

   385

                                ("Transfer", wx.WXK_F6),

   386

                                ("Connect",  wx.WXK_F7),

   387

                                ("Build",    wx.WXK_F11)]:

   388

            def OnMethodGen(obj,meth):

   389

                def OnMethod(evt):

   390

                    if obj.CTR is not None:

   391

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

   392

                    wx.CallAfter(self.RefreshStatusToolBar)

   393

                return OnMethod

   394

            newid = wx.NewId()

   395

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

   396

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

   397

   398

        self.SetAcceleratorTable(wx.AcceleratorTable(accels))

   399

   400

        self.LogConsole = CustomStyledTextCtrl(

   401

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

   402

                  size=wx.Size(0, 0))

   403

        self.LogConsole.Bind(wx.EVT_SET_FOCUS, self.OnLogConsoleFocusChanged)

   404

        self.LogConsole.Bind(wx.EVT_KILL_FOCUS, self.OnLogConsoleFocusChanged)

   405

        self.LogConsole.Bind(wx.stc.EVT_STC_UPDATEUI, self.OnLogConsoleUpdateUI)

   406

        self.LogConsole.SetReadOnly(True)

   407

        self.LogConsole.SetWrapMode(wx.stc.STC_WRAP_CHAR)

   408

   409

        # Define Log Console styles

   410

        self.LogConsole.StyleSetSpec(wx.stc.STC_STYLE_DEFAULT, "face:%(mono)s,size:%(size)d" % faces)

   411

        self.LogConsole.StyleClearAll()

   412

        self.LogConsole.StyleSetSpec(1, "face:%(mono)s,fore:#FF0000,size:%(size)d" % faces)

   413

        self.LogConsole.StyleSetSpec(2, "face:%(mono)s,fore:#FF0000,back:#FFFF00,size:%(size)d" % faces)

   414

   415

        # Define Log Console markers

   416

        self.LogConsole.SetMarginSensitive(1, True)

   417

        self.LogConsole.SetMarginType(1, wx.stc.STC_MARGIN_SYMBOL)

   418

        self.LogConsole.MarkerDefine(0, wx.stc.STC_MARK_CIRCLE, "BLACK", "RED")

   419

   420

        self.LogConsole.SetModEventMask(wx.stc.STC_MOD_INSERTTEXT)

   421

   422

        self.LogConsole.Bind(wx.stc.EVT_STC_MARGINCLICK, self.OnLogConsoleMarginClick)

   423

        self.LogConsole.Bind(wx.stc.EVT_STC_MODIFIED, self.OnLogConsoleModified)

   424

   425

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

   426

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

   427

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

   428

   429

        self.LogViewer = LogViewer(self.BottomNoteBook, self)

   430

        self.MainTabs["LogViewer"] = (self.LogViewer, _("PLC Log"))

   431

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

   432

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

   433

   434

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

   435

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

   436

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

   437

        StatusToolBar.Realize()

   438

        self.Panes["StatusToolBar"] = StatusToolBar

   439

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

   440

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

   441

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

   442

                  LeftDockable(False).RightDockable(False))

   443

   444

        self.AUIManager.Update()

   445

   446

        self.ConnectionStatusBar = wx.StatusBar(self, style=wx.ST_SIZEGRIP)

   447

        self._init_coll_ConnectionStatusBar_Fields(self.ConnectionStatusBar)

   448

        self.SetStatusBar(self.ConnectionStatusBar)

   449

   450

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

   451

        IDEFrame.__init__(self, parent, debug)

   452

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

   453

   454

        self.local_runtime = None

   455

        self.runtime_port = None

   456

        self.local_runtime_tmpdir = None

   457

   458

        self.LastPanelSelected = None

   459

   460

        # Define Tree item icon list

   461

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

   462

        self.LocationImageDict = {}

   463

   464

        # Icons for location items

   465

        for imgname, itemtype in [

   466

            ("CONFIGURATION", LOCATION_CONFNODE),

   467

            ("RESOURCE",      LOCATION_MODULE),

   468

            ("PROGRAM",       LOCATION_GROUP),

   469

            ("VAR_INPUT",     LOCATION_VAR_INPUT),

   470

            ("VAR_OUTPUT",    LOCATION_VAR_OUTPUT),

   471

            ("VAR_LOCAL",     LOCATION_VAR_MEMORY)]:

   472

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

   473

   474

        # Icons for other items

   475

        for imgname, itemtype in [

   476

            ("Extension", ITEM_CONFNODE)]:

   477

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

   478

   479

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

   480

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

   481

   482

        if projectOpen is not None:

   483

            projectOpen = DecodeFileSystemPath(projectOpen, False)

   484