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

    40

def Bpath(*args):

    41

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

    42

    43

if __name__ == '__main__':

    44

    def usage():

    45

        print "\nUsage of Beremiz.py :"

    46

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

    47

    48

    try:

    49

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

    50

    except getopt.GetoptError:

    51

        # print help information and exit:

    52

        usage()

    53

        sys.exit(2)

    54

    55

    extensions=[]

    56

    57

    for o, a in opts:

    58

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

    59

            usage()

    60

            sys.exit()

    61

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

    62

            updateinfo_url = a

    63

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

    64

            extensions.append(a)

    65

    66

    if len(args) > 2:

    67

        usage()

    68

        sys.exit()

    69

    elif len(args) == 1:

    70

        projectOpen = args[0]

    71

        buildpath = None

    72

    elif len(args) == 2:

    73

        projectOpen = args[0]

    74

        buildpath = args[1]

    75

    else:

    76

        projectOpen = None

    77

        buildpath = None

    78

    79

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

    80

        __builtin__.__dict__["BMZ_DBG"] = True

    81

    else :

    82

        __builtin__.__dict__["BMZ_DBG"] = False

    83

    84

    app = wx.PySimpleApp(redirect=BMZ_DBG)

    85

    app.SetAppName('beremiz')

    86

    wx.InitAllImageHandlers()

    87

    88

    # popup splash

    89

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

    90

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

    91

    splash=AdvancedSplash(None, bitmap=bmp)

    92

    wx.Yield()

    93

    94

    if updateinfo_url is not None:

    95

        updateinfo = "Fetching %s" % updateinfo_url

    96

        # warn for possible updates

    97

        def updateinfoproc():

    98

            global updateinfo

    99

            try :

   100

                import urllib2

   101

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

   102

            except :

   103

                updateinfo = "update info unavailable."

   104

   105

        from threading import Thread

   106

        splash.SetText(text=updateinfo)

   107

        wx.Yield()

   108

        updateinfoThread = Thread(target=updateinfoproc)

   109

        updateinfoThread.start()

   110

        updateinfoThread.join(2)

   111

        splash.SetText(text=updateinfo)

   112

        wx.Yield()

   113

   114

    from util.misc import InstallLocalRessources

   115

    InstallLocalRessources(CWD)

   116

   117

    # Load extensions

   118

    for extfilename in extensions:

   119

        from util.TranslationCatalogs import AddCatalog

   120

        from util.BitmapLibrary import AddBitmapFolder

   121

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

   122

        sys.path.append(extension_folder)

   123

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

   124

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

   125

        execfile(extfilename, locals())

   126

   127

   128

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

   129

import cPickle

   130

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

   131

   132

from docutil import OpenHtmlFrame

   133

from editors.EditorPanel import EditorPanel

   134

from editors.Viewer import Viewer

   135

from editors.TextViewer import TextViewer

   136

from editors.ResourceEditor import ConfigurationEditor, ResourceEditor

   137

from editors.DataTypeEditor import DataTypeEditor

   138

from util.MiniTextControler import MiniTextControler

   139

from util.ProcessLogger import ProcessLogger

   140

from controls.LogViewer import LogViewer

   141

from controls.CustomStyledTextCtrl import CustomStyledTextCtrl

   142

   143

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

   144

from ProjectController import ProjectController, GetAddMenuItems, MATIEC_ERROR_MODEL, ITEM_CONFNODE

   145

   146

   147

MAX_RECENT_PROJECTS = 10

   148

   149

if wx.Platform == '__WXMSW__':

   150

    faces = {

   151

        'mono' : 'Courier New',

   152

        'size' : 8,

   153

    }

   154

else:

   155

    faces = {

   156

        'mono' : 'Courier',

   157

        'size' : 10,

   158

    }

   159

   160

from threading import Lock,Timer,currentThread

   161

MainThread = currentThread().ident

   162

REFRESH_PERIOD = 0.1

   163

from time import time as gettime

   164

class LogPseudoFile:

   165

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

   166

    def __init__(self, output, risecall):

   167

        self.red_white = 1

   168

        self.red_yellow = 2

   169

        self.black_white = wx.stc.STC_STYLE_DEFAULT

   170

        self.output = output

   171

        self.risecall = risecall

   172

        # to prevent rapid fire on rising log panel

   173

        self.rising_timer = 0

   174

        self.lock = Lock()

   175

        self.YieldLock = Lock()

   176

        self.RefreshLock = Lock()

   177

        self.TimerAccessLock = Lock()

   178

        self.stack = []

   179

        self.LastRefreshTime = gettime()

   180

        self.LastRefreshTimer = None

   181

   182

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

   183

        if self.lock.acquire():

   184

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

   185

            self.lock.release()

   186

            current_time = gettime()

   187

            self.TimerAccessLock.acquire()

   188

            if self.LastRefreshTimer:

   189

                self.LastRefreshTimer.cancel()

   190

                self.LastRefreshTimer=None

   191

            self.TimerAccessLock.release()

   192

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

   193

                self._should_write()

   194

            else:

   195

                self.TimerAccessLock.acquire()

   196

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

   197

                self.LastRefreshTimer.start()

   198

                self.TimerAccessLock.release()

   199

   200

    def _timer_expired(self):

   201

        if self.RefreshLock.acquire(False):

   202

            self._should_write()

   203

        else:

   204

            self.TimerAccessLock.acquire()

   205

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

   206

            self.LastRefreshTimer.start()

   207

            self.TimerAccessLock.release()

   208

   209

    def _should_write(self):

   210

        wx.CallAfter(self._write)

   211

        if MainThread == currentThread().ident:

   212

            app = wx.GetApp()

   213

            if app is not None:

   214

                if self.YieldLock.acquire(0):

   215

                    app.Yield()

   216

                    self.YieldLock.release()

   217

   218

    def _write(self):

   219

        if self.output :

   220

            self.output.Freeze()

   221

            self.lock.acquire()

   222

            for s, style in self.stack:

   223

                if style is None : style=self.black_white

   224

                if style != self.black_white:

   225

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

   226

   227

                # Temporary deactivate read only mode on StyledTextCtrl for

   228

                # adding text. It seems that text modifications, even

   229

                # programmatically, are disabled in StyledTextCtrl when read

   230

                # only is active

   231

                self.output.SetReadOnly(False)

   232

                self.output.AppendText(s)

   233

                self.output.SetReadOnly(True)

   234

   235

                if style != self.black_white:

   236

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

   237

            self.stack = []

   238

            self.lock.release()

   239

            self.output.Thaw()

   240

            self.LastRefreshTime = gettime()

   241

            try:

   242

                self.RefreshLock.release()

   243

            except:

   244

                pass

   245

            newtime = time.time()

   246

            if newtime - self.rising_timer > 1:

   247

                self.risecall(self.output)

   248

            self.rising_timer = newtime

   249

   250

    def write_warning(self, s):

   251

        self.write(s,self.red_white)

   252

   253

    def write_error(self, s):

   254

        self.write(s,self.red_yellow)

   255

   256

    def writeyield(self, s):

   257

        self.write(s)

   258

        wx.GetApp().Yield()

   259

   260

    def flush(self):

   261

        # Temporary deactivate read only mode on StyledTextCtrl for clearing

   262

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

   263

        # disabled in StyledTextCtrl when read only is active

   264

        self.output.SetReadOnly(False)

   265

        self.output.SetText("")

   266

        self.output.SetReadOnly(True)

   267

   268

    def isatty(self):

   269

        return False

   270

   271

ID_FILEMENURECENTPROJECTS = wx.NewId()

   272

   273

from IDEFrame import TITLE,\

   274

                     EDITORTOOLBAR,\

   275

                     FILEMENU,\

   276

                     EDITMENU,\

   277

                     DISPLAYMENU,\

   278

                     PROJECTTREE,\

   279

                     POUINSTANCEVARIABLESPANEL,\

   280

                     LIBRARYTREE,\

   281

                     SCALING,\

   282

                     PAGETITLES,\

   283

                     IDEFrame, AppendMenu,\

   284

                     EncodeFileSystemPath, DecodeFileSystemPath

   285

from util.BitmapLibrary import GetBitmap

   286

   287

class Beremiz(IDEFrame):

   288

   289

    def _init_utils(self):

   290

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

   291

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

   292

   293

        IDEFrame._init_utils(self)

   294

   295

    def _init_coll_FileMenu_Items(self, parent):

   296

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

   297

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

   298

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

   299

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

   300

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

   301

        parent.AppendSeparator()

   302

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

   303

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

   304

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

   305

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

   306

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

   307

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

   308

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

   309

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

   310

        parent.AppendSeparator()

   311

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

   312

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

   313

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

   314

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

   315

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

   316

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

   317

        parent.AppendSeparator()

   318

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

   319

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

   320

   321

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

   322

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

   323

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

   324

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

   325

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

   326

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

   327

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

   328

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

   329

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

   330

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

   331

   332

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

   333

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

   334

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

   335

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

   336

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

   337

   338

    def _RecursiveAddMenuItems(self, menu, items):

   339

        for name, text, help, children in items:

   340

            new_id = wx.NewId()

   341

            if len(children) > 0:

   342

                new_menu = wx.Menu(title='')

   343

                menu.AppendMenu(new_id, text, new_menu)

   344

                self._RecursiveAddMenuItems(new_menu, children)

   345

            else:

   346

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

   347

                       kind=wx.ITEM_NORMAL, text=text)

   348

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

   349

                          id=new_id)

   350

   351

    def _init_coll_AddMenu_Items(self, parent):

   352

        IDEFrame._init_coll_AddMenu_Items(self, parent, False)

   353

        self._RecursiveAddMenuItems(parent, GetAddMenuItems())

   354

   355

    def _init_coll_HelpMenu_Items(self, parent):

   356

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

   357

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

   358

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

   359

   360

    def _init_coll_ConnectionStatusBar_Fields(self, parent):

   361

        parent.SetFieldsCount(3)

   362

   363

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

   364

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

   365

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

   366

   367

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

   368

   369

    def _init_ctrls(self, prnt):

   370

        IDEFrame._init_ctrls(self, prnt)

   371

   372

        self.EditMenuSize = self.EditMenu.GetMenuItemCount()

   373

   374

        inspectorID = wx.NewId()

   375

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

   376

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

   377

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

   378

                                ("Run",      wx.WXK_F5),

   379

                                ("Transfer", wx.WXK_F6),

   380

                                ("Connect",  wx.WXK_F7),

   381

                                ("Build",    wx.WXK_F11)]:

   382

            def OnMethodGen(obj,meth):

   383

                def OnMethod(evt):

   384

                    if obj.CTR is not None:

   385

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

   386

                    wx.CallAfter(self.RefreshStatusToolBar)

   387

                return OnMethod

   388

            newid = wx.NewId()

   389

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

   390

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

   391

   392

        self.SetAcceleratorTable(wx.AcceleratorTable(accels))

   393

   394

        self.LogConsole = CustomStyledTextCtrl(

   395

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

   396

                  size=wx.Size(0, 0))

   397

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

   398

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

   399

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

   400

        self.LogConsole.SetReadOnly(True)

   401

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

   402

   403

        # Define Log Console styles

   404

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

   405

        self.LogConsole.StyleClearAll()

   406

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

   407

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

   408

   409

        # Define Log Console markers

   410

        self.LogConsole.SetMarginSensitive(1, True)

   411

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

   412

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

   413

   414

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

   415

   416

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

   417

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

   418

   419

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

   420

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

   421

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

   422

   423

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

   424

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

   425

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

   426

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

   427

   428

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

   429

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

   430

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

   431

        StatusToolBar.Realize()

   432

        self.Panes["StatusToolBar"] = StatusToolBar

   433

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

   434

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

   435

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

   436

                  LeftDockable(False).RightDockable(False))

   437

   438

        self.AUIManager.Update()

   439

   440

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

   441

        self._init_coll_ConnectionStatusBar_Fields(self.ConnectionStatusBar)

   442

        self.SetStatusBar(self.ConnectionStatusBar)

   443

   444

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

   445

        IDEFrame.__init__(self, parent, debug)

   446

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

   447

   448

        self.local_runtime = None

   449

        self.runtime_port = None

   450

        self.local_runtime_tmpdir = None

   451

   452

        self.LastPanelSelected = None

   453

   454

        # Define Tree item icon list

   455

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

   456

        self.LocationImageDict = {}

   457

   458

        # Icons for location items

   459

        for imgname, itemtype in [

   460

            ("CONFIGURATION", LOCATION_CONFNODE),

   461

            ("RESOURCE",      LOCATION_MODULE),

   462

            ("PROGRAM",       LOCATION_GROUP),

   463

            ("VAR_INPUT",     LOCATION_VAR_INPUT),

   464

            ("VAR_OUTPUT",    LOCATION_VAR_OUTPUT),

   465

            ("VAR_LOCAL",     LOCATION_VAR_MEMORY)]:

   466

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

   467

   468

        # Icons for other items

   469

        for imgname, itemtype in [

   470

            ("Extension", ITEM_CONFNODE)]:

   471

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

   472

   473

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

   474

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

   475

   476

        if projectOpen is not None:

   477

            projectOpen = DecodeFileSystemPath(projectOpen, False)

   478

   479

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

   480

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

   481

            self.Controler = self.CTR