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

    29

import __builtin__

    30

import tempfile

    31

import shutil

    32

import random

    33

import time

    34

from types import ListType

    35

    36

beremiz_dir = os.path.dirname(os.path.realpath(__file__))

    37

    43

from wx.lib.agw.advancedsplash import AdvancedSplash

    44

    45

def Bpath(*args):

    46

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

    47

    48

def ShowSplashScreen():

    49

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

    50

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

    51

    splash = AdvancedSplash(None, bitmap=bmp)

    52

    53

    # process all events

    54

    # even the events generated by splash themself during showing

    55

    for i in range(0,30):

    56

        wx.Yield()

    57

        time.sleep(0.01);

    58

    return splash

    59

    60

if __name__ == '__main__':

    61

    def usage():

    62

        print "\nUsage of Beremiz.py :"

    63

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

    64

    65

    try:

    66

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

    67

    except getopt.GetoptError:

    68

        # print help information and exit:

    69

        usage()

    70

        sys.exit(2)

    71

    72

    extensions=[]

    73

    74

    for o, a in opts:

    75

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

    76

            usage()

    77

            sys.exit()

    78

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

    79

            updateinfo_url = a

    80

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

    81

            extensions.append(a)

    82

    83

    if len(args) > 2:

    84

        usage()

    85

        sys.exit()

    86

    elif len(args) == 1:

    87

        projectOpen = args[0]

    88

        buildpath = None

    89

    elif len(args) == 2:

    90

        projectOpen = args[0]

    91

        buildpath = args[1]

    92

    else:

    93

        projectOpen = None

    94

        buildpath = None

    95

    96

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

    97

        __builtin__.__dict__["BMZ_DBG"] = True

    98

    else :

    99

        __builtin__.__dict__["BMZ_DBG"] = False

   100

   101

    if wx.VERSION >= (3, 0, 0):

   102

        app = wx.App(redirect=BMZ_DBG)

   103

    else:

   104

        app = wx.PySimpleApp(redirect=BMZ_DBG)

   105

   106

    app.SetAppName('beremiz')

   107

    if wx.VERSION < (3, 0, 0):

   108

	wx.InitAllImageHandlers()

   109

   110

    # popup splash

   111

    splash = ShowSplashScreen()

   112

   113

    if updateinfo_url is not None:

   114

        updateinfo = "Fetching %s" % updateinfo_url

   115

        # warn for possible updates

   116

        def updateinfoproc():

   117

            global updateinfo

   118

            try :

   119

                import urllib2

   120

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

   121

            except :

   122

                updateinfo = "update info unavailable."

   123

   124

        from threading import Thread

   125

        splash.SetText(text=updateinfo)

   126

        wx.Yield()

   127

        updateinfoThread = Thread(target=updateinfoproc)

   128

        updateinfoThread.start()

   129

        updateinfoThread.join(2)

   130

        splash.SetText(text=updateinfo)

   131

        wx.Yield()

   132

   133

    from util.misc import InstallLocalRessources

   134

    InstallLocalRessources(beremiz_dir)

   135

   136

    # Load extensions

   137

    for extfilename in extensions:

   138

        from util.TranslationCatalogs import AddCatalog

   139

        from util.BitmapLibrary import AddBitmapFolder

   140

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

   141

        sys.path.append(extension_folder)

   142

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

   143

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

   144

        execfile(extfilename, locals())

   145

   146

   147

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

   148

import cPickle

   149

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

   150

   151

from docutil import OpenHtmlFrame

   152

from editors.EditorPanel import EditorPanel

   153

from editors.Viewer import Viewer

   154

from editors.TextViewer import TextViewer

   155

from editors.ResourceEditor import ConfigurationEditor, ResourceEditor

   156

from editors.DataTypeEditor import DataTypeEditor

   157

from util.MiniTextControler import MiniTextControler

   158

from util.ProcessLogger import ProcessLogger

   159

from controls.LogViewer import LogViewer

   160

from controls.CustomStyledTextCtrl import CustomStyledTextCtrl

   161

   162

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

   163

from ProjectController import ProjectController, GetAddMenuItems, MATIEC_ERROR_MODEL, ITEM_CONFNODE

   164

   165

   166

MAX_RECENT_PROJECTS = 10

   167

   168

if wx.Platform == '__WXMSW__':

   169

    faces = {

   170

        'mono' : 'Courier New',

   171

        'size' : 8,

   172

    }

   173

else:

   174

    faces = {

   175

        'mono' : 'Courier',

   176

        'size' : 10,

   177

    }

   178

   179

from threading import Lock,Timer,currentThread

   180

MainThread = currentThread().ident

   181

REFRESH_PERIOD = 0.1

   182

from time import time as gettime

   183

class LogPseudoFile:

   184

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

   185

    def __init__(self, output, risecall):

   186

        self.red_white = 1

   187

        self.red_yellow = 2

   188

        self.black_white = wx.stc.STC_STYLE_DEFAULT

   189

        self.output = output

   190

        self.risecall = risecall

   191

        # to prevent rapid fire on rising log panel

   192

        self.rising_timer = 0

   193

        self.lock = Lock()

   194

        self.YieldLock = Lock()

   195

        self.RefreshLock = Lock()

   196

        self.TimerAccessLock = Lock()

   197

        self.stack = []

   198

        self.LastRefreshTime = gettime()

   199

        self.LastRefreshTimer = None

   200

   201

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

   202

        if self.lock.acquire():

   203

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

   204

            self.lock.release()

   205

            current_time = gettime()

   206

            self.TimerAccessLock.acquire()

   207

            if self.LastRefreshTimer:

   208

                self.LastRefreshTimer.cancel()

   209

                self.LastRefreshTimer=None

   210

            self.TimerAccessLock.release()

   211

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

   212

                self._should_write()

   213

            else:

   214

                self.TimerAccessLock.acquire()

   215

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

   216

                self.LastRefreshTimer.start()

   217

                self.TimerAccessLock.release()

   218

   219

    def _timer_expired(self):

   220

        if self.RefreshLock.acquire(False):

   221

            self._should_write()

   222

        else:

   223

            self.TimerAccessLock.acquire()

   224

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

   225

            self.LastRefreshTimer.start()

   226

            self.TimerAccessLock.release()

   227

   228

    def _should_write(self):

   229

        wx.CallAfter(self._write)

   230

        if MainThread == currentThread().ident:

   231

            app = wx.GetApp()

   232

            if app is not None:

   233

                if self.YieldLock.acquire(0):

   234

                    app.Yield()

   235

                    self.YieldLock.release()

   236

   237

    def _write(self):

   238

        if self.output :

   239

            self.output.Freeze()

   240

            self.lock.acquire()

   241

            for s, style in self.stack:

   242

                if style is None : style=self.black_white

   243

                if style != self.black_white:

   244

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

   245

   246

                # Temporary deactivate read only mode on StyledTextCtrl for

   247

                # adding text. It seems that text modifications, even

   248

                # programmatically, are disabled in StyledTextCtrl when read

   249

                # only is active

   250

                self.output.SetReadOnly(False)

   251

                self.output.AppendText(s)

   252

                self.output.SetReadOnly(True)

   253

   254

                if style != self.black_white:

   255

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

   256

            self.stack = []

   257

            self.lock.release()

   258

            self.output.Thaw()

   259

            self.LastRefreshTime = gettime()

   260

            try:

   261

                self.RefreshLock.release()

   262

            except:

   263

                pass

   264

            newtime = time.time()

   265

            if newtime - self.rising_timer > 1:

   266

                self.risecall(self.output)

   267

            self.rising_timer = newtime

   268

   269

    def write_warning(self, s):

   270

        self.write(s,self.red_white)

   271

   272

    def write_error(self, s):

   273

        self.write(s,self.red_yellow)

   274

   275

    def writeyield(self, s):

   276

        self.write(s)

   277

        wx.GetApp().Yield()

   278

   279

    def flush(self):

   280

        # Temporary deactivate read only mode on StyledTextCtrl for clearing

   281

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

   282

        # disabled in StyledTextCtrl when read only is active

   283

        self.output.SetReadOnly(False)

   284

        self.output.SetText("")

   285

        self.output.SetReadOnly(True)

   286

   287

    def isatty(self):

   288

        return False

   289

   290

ID_FILEMENURECENTPROJECTS = wx.NewId()

   291

   292

from IDEFrame import TITLE,\

   293

                     EDITORTOOLBAR,\

   294

                     FILEMENU,\

   295

                     EDITMENU,\

   296

                     DISPLAYMENU,\

   297

                     PROJECTTREE,\

   298

                     POUINSTANCEVARIABLESPANEL,\

   299

                     LIBRARYTREE,\

   300

                     SCALING,\

   301

                     PAGETITLES,\

   302

                     IDEFrame, AppendMenu,\

   303

                     EncodeFileSystemPath, DecodeFileSystemPath

   304

from util.BitmapLibrary import GetBitmap

   305

   306

class Beremiz(IDEFrame):

   307

   308

    def _init_utils(self):

   309

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

   310

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

   311

   312

        IDEFrame._init_utils(self)

   313

   314

    def _init_coll_FileMenu_Items(self, parent):

   315

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

   316

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

   317

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

   318

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

   319

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

   320

        parent.AppendSeparator()

   321

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

   322

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

   323

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

   324

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

   325

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

   326

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

   327

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

   328

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

   329

        parent.AppendSeparator()

   330

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

   331

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

   332

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

   333

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

   334

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

   335

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

   336

        parent.AppendSeparator()

   337

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

   338

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

   339

   340

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

   341

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

   342

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

   343

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

   344

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

   345

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

   346

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

   347

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

   348

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

   349

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

   350

   351

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

   352

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

   353

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

   354

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

   355

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

   356

   357

    def _RecursiveAddMenuItems(self, menu, items):

   358

        for name, text, help, children in items:

   359

            new_id = wx.NewId()

   360

            if len(children) > 0:

   361

                new_menu = wx.Menu(title='')

   362

                menu.AppendMenu(new_id, text, new_menu)

   363

                self._RecursiveAddMenuItems(new_menu, children)

   364

            else:

   365

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

   366

                       kind=wx.ITEM_NORMAL, text=text)

   367

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

   368

                          id=new_id)

   369

   370

    def _init_coll_AddMenu_Items(self, parent):

   371

        IDEFrame._init_coll_AddMenu_Items(self, parent, False)

   372

        self._RecursiveAddMenuItems(parent, GetAddMenuItems())

   373

   374

    def _init_coll_HelpMenu_Items(self, parent):

   375

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

   376

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

   377

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

   378

   379

    def _init_coll_ConnectionStatusBar_Fields(self, parent):

   380

        parent.SetFieldsCount(3)

   381

   382

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

   383

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

   384

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

   385

   386

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

   387

   388

    def _init_ctrls(self, prnt):

   389

        IDEFrame._init_ctrls(self, prnt)

   390

   391

        self.EditMenuSize = self.EditMenu.GetMenuItemCount()

   392

   393

        inspectorID = wx.NewId()

   394

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

   395

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

   396

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

   397

                                ("Run",      wx.WXK_F5),

   398

                                ("Transfer", wx.WXK_F6),

   399

                                ("Connect",  wx.WXK_F7),

   400

                                ("Build",    wx.WXK_F11)]:

   401

            def OnMethodGen(obj,meth):

   402

                def OnMethod(evt):

   403

                    if obj.CTR is not None:

   404

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

   405

                    wx.CallAfter(self.RefreshStatusToolBar)

   406

                return OnMethod

   407

            newid = wx.NewId()

   408

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

   409

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

   410

   411

        self.SetAcceleratorTable(wx.AcceleratorTable(accels))

   412

   413

        self.LogConsole = CustomStyledTextCtrl(

   414

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

   415

                  size=wx.Size(0, 0))

   416

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

   417

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

   418

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

   419

        self.LogConsole.SetReadOnly(True)

   420

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

   421

   422

        # Define Log Console styles

   423

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

   424

        self.LogConsole.StyleClearAll()

   425

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

   426

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

   427

   428

        # Define Log Console markers

   429

        self.LogConsole.SetMarginSensitive(1, True)

   430

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

   431

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

   432

   433

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

   434

   435

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

   436

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

   437

   438

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

   439

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

   440

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

   441

   442

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

   443

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

   444

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

   445

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

   446

   447

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

   448

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

   449

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

   450

        StatusToolBar.Realize()

   451

        self.Panes["StatusToolBar"] = StatusToolBar

   452

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

   453

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

   454

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

   455

                  LeftDockable(False).RightDockable(False))

   456

   457

        self.AUIManager.Update()

   458

   459

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

   460

        self._init_coll_ConnectionStatusBar_Fields(self.ConnectionStatusBar)

   461

        self.SetStatusBar(self.ConnectionStatusBar)

   462

   463

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

   464

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

   465

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

   466

   467

        IDEFrame.__init__(self, parent, debug)

   468

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

   469

   470

        self.local_runtime = None

   471

        self.runtime_port = None

   472

        self.local_runtime_tmpdir = None

   473

   474

        self.LastPanelSelected = None

   475

   476

        # Define Tree item icon list

   477

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

   478

        self.LocationImageDict = {}

   479

   480

        # Icons for location items

   481

        for imgname, itemtype in [