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

    38

import wxversion

    39

wxversion.select('2.8')

    40

import wx

    41

from wx.lib.agw.advancedsplash import AdvancedSplash

    42

    43

def Bpath(*args):

    44

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

    45

    46

if __name__ == '__main__':

    47

    def usage():

    48

        print "\nUsage of Beremiz.py :"

    49

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

    50

    51

    try:

    52

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

    53

    except getopt.GetoptError:

    54

        # print help information and exit:

    55

        usage()

    56

        sys.exit(2)

    57

    58

    extensions=[]

    59

    60

    for o, a in opts:

    61

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

    62

            usage()

    63

            sys.exit()

    64

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

    65

            updateinfo_url = a

    66

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

    67

            extensions.append(a)

    68

    69

    if len(args) > 2:

    70

        usage()

    71

        sys.exit()

    72

    elif len(args) == 1:

    73

        projectOpen = args[0]

    74

        buildpath = None

    75

    elif len(args) == 2:

    76

        projectOpen = args[0]

    77

        buildpath = args[1]

    78

    else:

    79

        projectOpen = None

    80

        buildpath = None

    81

    82

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

    83

        __builtin__.__dict__["BMZ_DBG"] = True

    84

    else :

    85

        __builtin__.__dict__["BMZ_DBG"] = False

    86

    87

    app = wx.PySimpleApp(redirect=BMZ_DBG)

    88

    app.SetAppName('beremiz')

    89

    wx.InitAllImageHandlers()

    90

    91

    # popup splash

    92

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

    93

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

    94

    splash=AdvancedSplash(None, bitmap=bmp)

    95

    wx.Yield()

    96

    97

    if updateinfo_url is not None:

    98

        updateinfo = "Fetching %s" % updateinfo_url

    99

        # warn for possible updates

   100

        def updateinfoproc():

   101

            global updateinfo

   102

            try :

   103

                import urllib2

   104

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

   105

            except :

   106

                updateinfo = "update info unavailable."

   107

   108

        from threading import Thread

   109

        splash.SetText(text=updateinfo)

   110

        wx.Yield()

   111

        updateinfoThread = Thread(target=updateinfoproc)

   112

        updateinfoThread.start()

   113

        updateinfoThread.join(2)

   114

        splash.SetText(text=updateinfo)

   115

        wx.Yield()

   116

   117

    from util.misc import InstallLocalRessources

   118

    InstallLocalRessources(beremiz_dir)

   119

   120

    # Load extensions

   121

    for extfilename in extensions:

   122

        from util.TranslationCatalogs import AddCatalog

   123

        from util.BitmapLibrary import AddBitmapFolder

   124

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

   125

        sys.path.append(extension_folder)

   126

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

   127

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

   128

        execfile(extfilename, locals())

   129

   130

   131

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

   132

import cPickle

   133

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

   134

   135

from docutil import OpenHtmlFrame

   136

from editors.EditorPanel import EditorPanel

   137

from editors.Viewer import Viewer

   138

from editors.TextViewer import TextViewer

   139

from editors.ResourceEditor import ConfigurationEditor, ResourceEditor

   140

from editors.DataTypeEditor import DataTypeEditor

   141

from util.MiniTextControler import MiniTextControler

   142

from util.ProcessLogger import ProcessLogger

   143

from controls.LogViewer import LogViewer

   144

from controls.CustomStyledTextCtrl import CustomStyledTextCtrl

   145

   146

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

   147

from ProjectController import ProjectController, GetAddMenuItems, MATIEC_ERROR_MODEL, ITEM_CONFNODE

   148

   149

   150

MAX_RECENT_PROJECTS = 10

   151

   152

if wx.Platform == '__WXMSW__':

   153

    faces = {

   154

        'mono' : 'Courier New',

   155

        'size' : 8,

   156

    }

   157

else:

   158

    faces = {

   159

        'mono' : 'Courier',

   160

        'size' : 10,

   161

    }

   162

   163

from threading import Lock,Timer,currentThread

   164

MainThread = currentThread().ident

   165

REFRESH_PERIOD = 0.1

   166

from time import time as gettime

   167

class LogPseudoFile:

   168

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

   169

    def __init__(self, output, risecall):

   170

        self.red_white = 1

   171

        self.red_yellow = 2

   172

        self.black_white = wx.stc.STC_STYLE_DEFAULT

   173

        self.output = output

   174

        self.risecall = risecall

   175

        # to prevent rapid fire on rising log panel

   176

        self.rising_timer = 0

   177

        self.lock = Lock()

   178

        self.YieldLock = Lock()

   179

        self.RefreshLock = Lock()

   180

        self.TimerAccessLock = Lock()

   181

        self.stack = []

   182

        self.LastRefreshTime = gettime()

   183

        self.LastRefreshTimer = None

   184

   185

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

   186

        if self.lock.acquire():

   187

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

   188

            self.lock.release()

   189

            current_time = gettime()

   190

            self.TimerAccessLock.acquire()

   191

            if self.LastRefreshTimer:

   192

                self.LastRefreshTimer.cancel()

   193

                self.LastRefreshTimer=None

   194

            self.TimerAccessLock.release()

   195

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

   196

                self._should_write()

   197

            else:

   198

                self.TimerAccessLock.acquire()

   199

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

   200

                self.LastRefreshTimer.start()

   201

                self.TimerAccessLock.release()

   202

   203

    def _timer_expired(self):

   204

        if self.RefreshLock.acquire(False):

   205

            self._should_write()

   206

        else:

   207

            self.TimerAccessLock.acquire()

   208

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

   209

            self.LastRefreshTimer.start()

   210

            self.TimerAccessLock.release()

   211

   212

    def _should_write(self):

   213

        wx.CallAfter(self._write)

   214

        if MainThread == currentThread().ident:

   215

            app = wx.GetApp()

   216

            if app is not None:

   217

                if self.YieldLock.acquire(0):

   218

                    app.Yield()

   219

                    self.YieldLock.release()

   220

   221

    def _write(self):

   222

        if self.output :

   223

            self.output.Freeze()

   224

            self.lock.acquire()

   225

            for s, style in self.stack:

   226

                if style is None : style=self.black_white

   227

                if style != self.black_white:

   228

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

   229

   230

                # Temporary deactivate read only mode on StyledTextCtrl for

   231

                # adding text. It seems that text modifications, even

   232

                # programmatically, are disabled in StyledTextCtrl when read

   233

                # only is active

   234

                self.output.SetReadOnly(False)

   235

                self.output.AppendText(s)

   236

                self.output.SetReadOnly(True)

   237

   238

                if style != self.black_white:

   239

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

   240

            self.stack = []

   241

            self.lock.release()

   242

            self.output.Thaw()

   243

            self.LastRefreshTime = gettime()

   244

            try:

   245

                self.RefreshLock.release()

   246

            except:

   247

                pass

   248

            newtime = time.time()

   249

            if newtime - self.rising_timer > 1:

   250

                self.risecall(self.output)

   251

            self.rising_timer = newtime

   252

   253

    def write_warning(self, s):

   254

        self.write(s,self.red_white)

   255

   256

    def write_error(self, s):

   257

        self.write(s,self.red_yellow)

   258

   259

    def writeyield(self, s):

   260

        self.write(s)

   261

        wx.GetApp().Yield()

   262

   263

    def flush(self):

   264

        # Temporary deactivate read only mode on StyledTextCtrl for clearing

   265

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

   266

        # disabled in StyledTextCtrl when read only is active

   267

        self.output.SetReadOnly(False)

   268

        self.output.SetText("")

   269

        self.output.SetReadOnly(True)

   270

   271

    def isatty(self):

   272

        return False

   273

   274

ID_FILEMENURECENTPROJECTS = wx.NewId()

   275

   276

from IDEFrame import TITLE,\

   277

                     EDITORTOOLBAR,\

   278

                     FILEMENU,\

   279

                     EDITMENU,\

   280

                     DISPLAYMENU,\

   281

                     PROJECTTREE,\

   282

                     POUINSTANCEVARIABLESPANEL,\

   283

                     LIBRARYTREE,\

   284

                     SCALING,\

   285

                     PAGETITLES,\

   286

                     IDEFrame, AppendMenu,\

   287

                     EncodeFileSystemPath, DecodeFileSystemPath

   288

from util.BitmapLibrary import GetBitmap

   289

   290

class Beremiz(IDEFrame):

   291

   292

    def _init_utils(self):

   293

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

   294

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

   295

   296

        IDEFrame._init_utils(self)

   297

   298

    def _init_coll_FileMenu_Items(self, parent):

   299

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

   300

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

   301

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

   302

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

   303

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

   304

        parent.AppendSeparator()

   305

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

   306

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

   307

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

   308

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

   309

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

   310

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

   311

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

   312

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

   313

        parent.AppendSeparator()

   314

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

   315

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

   316

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

   317

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

   318

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

   319

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

   320

        parent.AppendSeparator()

   321

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

   322

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

   323

   324

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

   325

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

   326

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

   327

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

   328

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

   329

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

   330

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

   331

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

   332

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

   333

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

   334

   335

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

   336

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

   337

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

   338

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

   339

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

   340

   341

    def _RecursiveAddMenuItems(self, menu, items):

   342

        for name, text, help, children in items:

   343

            new_id = wx.NewId()

   344

            if len(children) > 0:

   345

                new_menu = wx.Menu(title='')

   346

                menu.AppendMenu(new_id, text, new_menu)

   347

                self._RecursiveAddMenuItems(new_menu, children)

   348

            else:

   349

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

   350

                       kind=wx.ITEM_NORMAL, text=text)

   351

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

   352

                          id=new_id)

   353

   354

    def _init_coll_AddMenu_Items(self, parent):

   355

        IDEFrame._init_coll_AddMenu_Items(self, parent, False)

   356

        self._RecursiveAddMenuItems(parent, GetAddMenuItems())

   357

   358

    def _init_coll_HelpMenu_Items(self, parent):

   359

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

   360

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

   361

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

   362

   363

    def _init_coll_ConnectionStatusBar_Fields(self, parent):

   364

        parent.SetFieldsCount(3)

   365

   366

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

   367

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

   368

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

   369

   370

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

   371

   372

    def _init_ctrls(self, prnt):

   373

        IDEFrame._init_ctrls(self, prnt)

   374

   375

        self.EditMenuSize = self.EditMenu.GetMenuItemCount()

   376

   377

        inspectorID = wx.NewId()

   378

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

   379

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

   380

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

   381

                                ("Run",      wx.WXK_F5),

   382

                                ("Transfer", wx.WXK_F6),

   383

                                ("Connect",  wx.WXK_F7),

   384

                                ("Build",    wx.WXK_F11)]:

   385

            def OnMethodGen(obj,meth):

   386

                def OnMethod(evt):

   387

                    if obj.CTR is not None:

   388

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

   389

                    wx.CallAfter(self.RefreshStatusToolBar)

   390

                return OnMethod

   391

            newid = wx.NewId()

   392

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

   393

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

   394

   395

        self.SetAcceleratorTable(wx.AcceleratorTable(accels))

   396

   397

        self.LogConsole = CustomStyledTextCtrl(

   398

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

   399

                  size=wx.Size(0, 0))

   400

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

   401

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

   402

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

   403

        self.LogConsole.SetReadOnly(True)

   404

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

   405

   406

        # Define Log Console styles

   407

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

   408

        self.LogConsole.StyleClearAll()

   409

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

   410

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

   411

   412

        # Define Log Console markers

   413

        self.LogConsole.SetMarginSensitive(1, True)

   414

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

   415

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

   416

   417

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

   418

   419

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

   420

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

   421

   422

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

   423

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

   424

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

   425

   426

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

   427

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

   428

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

   429

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

   430

   431

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

   432

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

   433

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

   434

        StatusToolBar.Realize()

   435

        self.Panes["StatusToolBar"] = StatusToolBar

   436

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

   437

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

   438

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

   439

                  LeftDockable(False).RightDockable(False))

   440

   441

        self.AUIManager.Update()

   442

   443

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

   444

        self._init_coll_ConnectionStatusBar_Fields(self.ConnectionStatusBar)

   445

        self.SetStatusBar(self.ConnectionStatusBar)

   446

   447

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

   448

        IDEFrame.__init__(self, parent, debug)

   449

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

   450

   451

        self.local_runtime = None

   452

        self.runtime_port = None

   453

        self.local_runtime_tmpdir = None

   454

   455

        self.LastPanelSelected = None

   456

   457

        # Define Tree item icon list

   458

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

   459

        self.LocationImageDict = {}

   460

   461

        # Icons for location items

   462

        for imgname, itemtype in [

   463

            ("CONFIGURATION", LOCATION_CONFNODE),

   464

            ("RESOURCE",      LOCATION_MODULE),

   465

            ("PROGRAM",       LOCATION_GROUP),

   466

            ("VAR_INPUT",     LOCATION_VAR_INPUT),

   467

            ("VAR_OUTPUT",    LOCATION_VAR_OUTPUT),

   468

            ("VAR_LOCAL",     LOCATION_VAR_MEMORY)]:

   469

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

   470

   471

        # Icons for other items

   472

        for imgname, itemtype in [

   473

            ("Extension", ITEM_CONFNODE)]:

   474

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

   475

   476

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

   477

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

   478

   479

        if projectOpen is not None:

   480

            projectOpen = DecodeFileSystemPath(projectOpen, False)