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

    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

    92

    app.SetAppName('beremiz')

    95

    96

    # popup splash

    97

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

    98

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

    99

    splash=AdvancedSplash(None, bitmap=bmp)

   100

    wx.Yield()

   101

   102

    if updateinfo_url is not None:

   103

        updateinfo = "Fetching %s" % updateinfo_url

   104

        # warn for possible updates

   105

        def updateinfoproc():

   106

            global updateinfo

   107

            try :

   108

                import urllib2

   109

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

   110

            except :

   111

                updateinfo = "update info unavailable."

   112

   113

        from threading import Thread

   114

        splash.SetText(text=updateinfo)

   115

        wx.Yield()

   116

        updateinfoThread = Thread(target=updateinfoproc)

   117

        updateinfoThread.start()

   118

        updateinfoThread.join(2)

   119

        splash.SetText(text=updateinfo)

   120

        wx.Yield()

   121

   122

    from util.misc import InstallLocalRessources

   123

    InstallLocalRessources(beremiz_dir)

   124

   125

    # Load extensions

   126

    for extfilename in extensions:

   127

        from util.TranslationCatalogs import AddCatalog

   128

        from util.BitmapLibrary import AddBitmapFolder

   129

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

   130

        sys.path.append(extension_folder)

   131

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

   132

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

   133

        execfile(extfilename, locals())

   134

   135

   136

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

   137

import cPickle

   138

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

   139

   140

from docutil import OpenHtmlFrame

   141

from editors.EditorPanel import EditorPanel

   142

from editors.Viewer import Viewer

   143

from editors.TextViewer import TextViewer

   144

from editors.ResourceEditor import ConfigurationEditor, ResourceEditor

   145

from editors.DataTypeEditor import DataTypeEditor

   146

from util.MiniTextControler import MiniTextControler

   147

from util.ProcessLogger import ProcessLogger

   148

from controls.LogViewer import LogViewer

   149

from controls.CustomStyledTextCtrl import CustomStyledTextCtrl

   150

   151

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

   152

from ProjectController import ProjectController, GetAddMenuItems, MATIEC_ERROR_MODEL, ITEM_CONFNODE

   153

   154

   155

MAX_RECENT_PROJECTS = 10

   156

   157

if wx.Platform == '__WXMSW__':

   158

    faces = {

   159

        'mono' : 'Courier New',

   160

        'size' : 8,

   161

    }

   162

else:

   163

    faces = {

   164

        'mono' : 'Courier',

   165

        'size' : 10,

   166

    }

   167

   168

from threading import Lock,Timer,currentThread

   169

MainThread = currentThread().ident

   170

REFRESH_PERIOD = 0.1

   171

from time import time as gettime

   172

class LogPseudoFile:

   173

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

   174

    def __init__(self, output, risecall):

   175

        self.red_white = 1

   176

        self.red_yellow = 2

   177

        self.black_white = wx.stc.STC_STYLE_DEFAULT

   178

        self.output = output

   179

        self.risecall = risecall

   180

        # to prevent rapid fire on rising log panel

   181

        self.rising_timer = 0

   182

        self.lock = Lock()

   183

        self.YieldLock = Lock()

   184

        self.RefreshLock = Lock()

   185

        self.TimerAccessLock = Lock()

   186

        self.stack = []

   187

        self.LastRefreshTime = gettime()

   188

        self.LastRefreshTimer = None

   189

   190

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

   191

        if self.lock.acquire():

   192

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

   193

            self.lock.release()

   194

            current_time = gettime()

   195

            self.TimerAccessLock.acquire()

   196

            if self.LastRefreshTimer:

   197

                self.LastRefreshTimer.cancel()

   198

                self.LastRefreshTimer=None

   199

            self.TimerAccessLock.release()

   200

            if current_time - self.LastRefreshTime > REFRESH_PERIOD and 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 _timer_expired(self):

   209

        if self.RefreshLock.acquire(False):

   210

            self._should_write()

   211

        else:

   212

            self.TimerAccessLock.acquire()

   213

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

   214

            self.LastRefreshTimer.start()

   215

            self.TimerAccessLock.release()

   216

   217

    def _should_write(self):

   218

        wx.CallAfter(self._write)

   219

        if MainThread == currentThread().ident:

   220

            app = wx.GetApp()

   221

            if app is not None:

   222

                if self.YieldLock.acquire(0):

   223

                    app.Yield()

   224

                    self.YieldLock.release()

   225

   226

    def _write(self):

   227

        if self.output :

   228

            self.output.Freeze()

   229

            self.lock.acquire()

   230

            for s, style in self.stack:

   231

                if style is None : style=self.black_white

   232

                if style != self.black_white:

   233

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

   234

   235

                # Temporary deactivate read only mode on StyledTextCtrl for

   236

                # adding text. It seems that text modifications, even

   237

                # programmatically, are disabled in StyledTextCtrl when read

   238

                # only is active

   239

                self.output.SetReadOnly(False)

   240

                self.output.AppendText(s)

   241

                self.output.SetReadOnly(True)

   242

   243

                if style != self.black_white:

   244

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

   245

            self.stack = []

   246

            self.lock.release()

   247

            self.output.Thaw()

   248

            self.LastRefreshTime = gettime()

   249

            try:

   250

                self.RefreshLock.release()

   251

            except:

   252

                pass

   253

            newtime = time.time()

   254

            if newtime - self.rising_timer > 1:

   255

                self.risecall(self.output)

   256

            self.rising_timer = newtime

   257

   258

    def write_warning(self, s):

   259

        self.write(s,self.red_white)

   260

   261

    def write_error(self, s):

   262

        self.write(s,self.red_yellow)

   263

   264

    def writeyield(self, s):

   265

        self.write(s)

   266

        wx.GetApp().Yield()

   267

   268

    def flush(self):

   269

        # Temporary deactivate read only mode on StyledTextCtrl for clearing

   270

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

   271

        # disabled in StyledTextCtrl when read only is active

   272

        self.output.SetReadOnly(False)

   273

        self.output.SetText("")

   274

        self.output.SetReadOnly(True)

   275

   276

    def isatty(self):

   277

        return False

   278

   279

ID_FILEMENURECENTPROJECTS = wx.NewId()

   280

   281

from IDEFrame import TITLE,\

   282

                     EDITORTOOLBAR,\

   283

                     FILEMENU,\

   284

                     EDITMENU,\

   285

                     DISPLAYMENU,\

   286

                     PROJECTTREE,\

   287

                     POUINSTANCEVARIABLESPANEL,\

   288

                     LIBRARYTREE,\

   289

                     SCALING,\

   290

                     PAGETITLES,\

   291

                     IDEFrame, AppendMenu,\

   292

                     EncodeFileSystemPath, DecodeFileSystemPath

   293

from util.BitmapLibrary import GetBitmap

   294

   295

class Beremiz(IDEFrame):

   296

   297

    def _init_utils(self):

   298

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

   299

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

   300

   301

        IDEFrame._init_utils(self)

   302

   303

    def _init_coll_FileMenu_Items(self, parent):

   304

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

   305

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

   306

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

   307

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

   308

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

   309

        parent.AppendSeparator()

   310

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

   311

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

   312

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

   313

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

   314

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

   315

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

   316

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

   317

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

   318

        parent.AppendSeparator()

   319

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

   320

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

   321

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

   322

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

   323

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

   324

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

   325

        parent.AppendSeparator()

   326

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

   327

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

   328

   329

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

   330

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

   331

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

   332

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

   333

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

   334

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

   335

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

   336

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

   337

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

   338

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

   339

   340

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

   341

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

   342

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

   343

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

   344

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

   345

   346

    def _RecursiveAddMenuItems(self, menu, items):

   347

        for name, text, help, children in items:

   348

            new_id = wx.NewId()

   349

            if len(children) > 0:

   350

                new_menu = wx.Menu(title='')

   351

                menu.AppendMenu(new_id, text, new_menu)

   352

                self._RecursiveAddMenuItems(new_menu, children)

   353

            else:

   354

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

   355

                       kind=wx.ITEM_NORMAL, text=text)

   356

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

   357

                          id=new_id)

   358

   359

    def _init_coll_AddMenu_Items(self, parent):

   360

        IDEFrame._init_coll_AddMenu_Items(self, parent, False)

   361

        self._RecursiveAddMenuItems(parent, GetAddMenuItems())

   362

   363

    def _init_coll_HelpMenu_Items(self, parent):

   364

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

   365

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

   366

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

   367

   368

    def _init_coll_ConnectionStatusBar_Fields(self, parent):

   369

        parent.SetFieldsCount(3)

   370

   371

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

   372

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

   373

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

   374

   375

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

   376

   377

    def _init_ctrls(self, prnt):

   378

        IDEFrame._init_ctrls(self, prnt)

   379

   380

        self.EditMenuSize = self.EditMenu.GetMenuItemCount()

   381

   382

        inspectorID = wx.NewId()

   383

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

   384

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

   385

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

   386

                                ("Run",      wx.WXK_F5),

   387

                                ("Transfer", wx.WXK_F6),

   388

                                ("Connect",  wx.WXK_F7),

   389

                                ("Build",    wx.WXK_F11)]:

   390

            def OnMethodGen(obj,meth):

   391

                def OnMethod(evt):

   392

                    if obj.CTR is not None:

   393

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

   394

                    wx.CallAfter(self.RefreshStatusToolBar)

   395

                return OnMethod

   396

            newid = wx.NewId()

   397

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

   398

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

   399

   400

        self.SetAcceleratorTable(wx.AcceleratorTable(accels))

   401

   402

        self.LogConsole = CustomStyledTextCtrl(

   403

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

   404

                  size=wx.Size(0, 0))

   405

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

   406

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

   407

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

   408

        self.LogConsole.SetReadOnly(True)

   409

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

   410

   411

        # Define Log Console styles

   412

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

   413

        self.LogConsole.StyleClearAll()

   414

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

   415

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

   416

   417

        # Define Log Console markers

   418

        self.LogConsole.SetMarginSensitive(1, True)

   419

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

   420

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

   421

   422

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

   423

   424

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

   425

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

   426

   427

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

   428

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

   429

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

   430

   431

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

   432

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

   433

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

   434

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

   435

   436

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

   437

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

   438

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

   439

        StatusToolBar.Realize()

   440

        self.Panes["StatusToolBar"] = StatusToolBar

   441

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

   442

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

   443

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

   444

                  LeftDockable(False).RightDockable(False))

   445

   446

        self.AUIManager.Update()

   447

   448

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

   449

        self._init_coll_ConnectionStatusBar_Fields(self.ConnectionStatusBar)

   450

        self.SetStatusBar(self.ConnectionStatusBar)

   451

   452

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

   453

        IDEFrame.__init__(self, parent, debug)

   454

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

   455

   456

        self.local_runtime = None

   457

        self.runtime_port = None

   458

        self.local_runtime_tmpdir = None

   459

   460

        self.LastPanelSelected = None

   461

   462

        # Define Tree item icon list

   463

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

   464

        self.LocationImageDict = {}

   465

   466

        # Icons for location items

   467

        for imgname, itemtype in [

   468

            ("CONFIGURATION", LOCATION_CONFNODE),

   469

            ("RESOURCE",      LOCATION_MODULE),

   470

            ("PROGRAM",       LOCATION_GROUP),

   471

            ("VAR_INPUT",     LOCATION_VAR_INPUT),

   472

            ("VAR_OUTPUT",    LOCATION_VAR_OUTPUT),

   473

            ("VAR_LOCAL",     LOCATION_VAR_MEMORY)]:

   474

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

   475

   476

        # Icons for other items

   477

        for imgname, itemtype in [

   478

            ("Extension", ITEM_CONFNODE)]:

   479

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

   480

   481

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