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.TranslationCatalogs import AddCatalog

   114

from util.BitmapLibrary import AddBitmapFolder, GetBitmap

   115

   116

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

   117

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

   118

   119

if __name__ == '__main__':

   120

    # Import module for internationalization

   121

    import gettext

   122

   123

    __builtin__.__dict__['_'] = wx.GetTranslation

   124

   125

    # Load extensions

   126

    for extfilename in extensions:

   127

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

   128

        sys.path.append(extension_folder)

   129

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

   130

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

   131

        execfile(extfilename, locals())

   132

   133

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

   134

import cPickle

   135

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

   136

   137

from docutil import OpenHtmlFrame

   138

from IDEFrame import IDEFrame, AppendMenu

   139

from IDEFrame import TITLE, EDITORTOOLBAR, FILEMENU, EDITMENU, DISPLAYMENU, PROJECTTREE, POUINSTANCEVARIABLESPANEL, LIBRARYTREE, SCALING, PAGETITLES 

   140

from IDEFrame import EncodeFileSystemPath, DecodeFileSystemPath

   141

from editors.EditorPanel import EditorPanel

   142

from editors.Viewer import Viewer

   143

from editors.TextViewer import TextViewer

   144

from editors.GraphicViewer import GraphicViewer

   145

from editors.ResourceEditor import ConfigurationEditor, ResourceEditor

   146

from editors.DataTypeEditor import DataTypeEditor

   147

from util.MiniTextControler import MiniTextControler

   148

from util.ProcessLogger import ProcessLogger

   149

from controls.LogViewer import LogViewer

   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, MATIEC_ERROR_MODEL, ITEM_CONFNODE

   153

   154

   155

MAX_RECENT_PROJECTS = 10

   156

   157

class GenStaticBitmap(wx.lib.statbmp.GenStaticBitmap):

   158

    """ Customized GenStaticBitmap, fix transparency redraw bug on wx2.8/win32, 

   159

    and accept image name as __init__ parameter, fail silently if file do not exist"""

   160

    def __init__(self, parent, ID, bitmapname,

   161

                 pos = wx.DefaultPosition, size = wx.DefaultSize,

   162

                 style = 0,

   163

                 name = "genstatbmp"):

   164

   165

        wx.lib.statbmp.GenStaticBitmap.__init__(self, parent, ID, 

   166

                 GetBitmap(bitmapname),

   167

                 pos, size,

   168

                 style,

   169

                 name)

   170

   171

    def OnPaint(self, event):

   172

        dc = wx.PaintDC(self)

   173

        colour = self.GetParent().GetBackgroundColour()

   174

        dc.SetPen(wx.Pen(colour))

   175

        dc.SetBrush(wx.Brush(colour ))

   176

        dc.DrawRectangle(0, 0, *dc.GetSizeTuple())

   177

        if self._bitmap:

   178

            dc.DrawBitmap(self._bitmap, 0, 0, True)

   179

   180

if wx.Platform == '__WXMSW__':

   181

    faces = {

   182

        'mono' : 'Courier New',

   183

        'size' : 8,

   184

    }

   185

else:

   186

    faces = {

   187

        'mono' : 'Courier',

   188

        'size' : 10,

   189

    }

   190

   191

from threading import Lock,Timer,currentThread

   192

MainThread = currentThread().ident

   193

REFRESH_PERIOD = 0.1

   194

from time import time as gettime

   195

class LogPseudoFile:

   196

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

   197

    def __init__(self, output, risecall):

   198

        self.red_white = 1

   199

        self.red_yellow = 2

   200

        self.black_white = wx.stc.STC_STYLE_DEFAULT

   201

        self.output = output

   202

        self.risecall = risecall

   203

        # to prevent rapid fire on rising log panel

   204

        self.rising_timer = 0

   205

        self.lock = Lock()

   206

        self.YieldLock = Lock()

   207

        self.RefreshLock = Lock()

   208

        self.TimerAccessLock = Lock()

   209

        self.stack = []

   210

        self.LastRefreshTime = gettime()

   211

        self.LastRefreshTimer = None

   212

   213

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

   214

        if self.lock.acquire():

   215

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

   216

            self.lock.release()

   217

            current_time = gettime()

   218

            self.TimerAccessLock.acquire()

   219

            if self.LastRefreshTimer:

   220

                self.LastRefreshTimer.cancel()

   221

                self.LastRefreshTimer=None

   222

            self.TimerAccessLock.release()

   223

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

   224

                self._should_write()

   225

            else:

   226

                self.TimerAccessLock.acquire()

   227

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

   228

                self.LastRefreshTimer.start()

   229

                self.TimerAccessLock.release()

   230

   231

    def _timer_expired(self):

   232

        if self.RefreshLock.acquire(False):

   233

            self._should_write()

   234

        else:

   235

            self.TimerAccessLock.acquire()

   236

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

   237

            self.LastRefreshTimer.start()

   238

            self.TimerAccessLock.release()

   239

   240

    def _should_write(self):

   241

        wx.CallAfter(self._write)

   242

        if MainThread == currentThread().ident:

   243

            app = wx.GetApp()

   244

            if app is not None:

   245

                if self.YieldLock.acquire(0):

   246

                    app.Yield()

   247

                    self.YieldLock.release()

   248

   249

    def _write(self):

   250

        if self.output :

   251

            self.output.Freeze()

   252

            self.lock.acquire()

   253

            for s, style in self.stack:

   254

                if style is None : style=self.black_white

   255

                if style != self.black_white:

   256

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

   257

   258

                # Temporary deactivate read only mode on StyledTextCtrl for adding text

   259

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

   260

                # in StyledTextCtrl when read only is active 

   261

                self.output.SetReadOnly(False)

   262

                self.output.AddText(s)

   263

                self.output.SetReadOnly(True)

   264

   265

                if style != self.black_white:

   266

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

   267

            self.stack = []

   268

            self.lock.release()

   269

            self.output.ScrollToLine(self.output.GetLineCount())

   270

            self.output.Thaw()

   271

            self.LastRefreshTime = gettime()

   272

            try:

   273

                self.RefreshLock.release()

   274

            except:

   275

                pass

   276

            newtime = time.time()

   277

            if newtime - self.rising_timer > 1:

   278

                self.risecall(self.output)

   279

            self.rising_timer = newtime

   280

   281

    def write_warning(self, s):

   282

        self.write(s,self.red_white)

   283

   284

    def write_error(self, s):

   285

        self.write(s,self.red_yellow)

   286

   287

    def writeyield(self, s):

   288

        self.write(s)

   289

        wx.GetApp().Yield()

   290

   291

    def flush(self):

   292

        # Temporary deactivate read only mode on StyledTextCtrl for clearing text

   293

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

   294

        # in StyledTextCtrl when read only is active 

   295

        self.output.SetReadOnly(False)

   296

        self.output.SetText("")

   297

        self.output.SetReadOnly(True)

   298

   299

    def isatty(self):

   300

        return false

   301

   302

[ID_BEREMIZ, ID_BEREMIZMAINSPLITTER, 

   303

 ID_BEREMIZPLCCONFIG, ID_BEREMIZLOGCONSOLE, 

   304

 ID_BEREMIZINSPECTOR] = [wx.NewId() for _init_ctrls in range(5)]

   305

   306

[ID_FILEMENURECENTPROJECTS,

   307

] = [wx.NewId() for _init_ctrls in range(1)]

   308

   309

CONFNODEMENU_POSITION = 3

   310

   311

class Beremiz(IDEFrame):

   312

   313

    def _init_utils(self):

   314

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

   315

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

   316

   317

        IDEFrame._init_utils(self)

   318

   319

    def _init_coll_FileMenu_Items(self, parent):

   320

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

   321

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

   322

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

   323

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

   324

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

   325

        parent.AppendSeparator()

   326

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

   327

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

   328

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

   329

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

   330

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

   331

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

   332

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

   333

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

   334

        parent.AppendSeparator()

   335

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

   336

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

   337

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

   338

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

   339

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

   340

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

   341

        parent.AppendSeparator()

   342

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

   343

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

   344

   345

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

   346

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

   347

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

   348

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

   349

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

   350

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

   351

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

   352

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

   353

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

   354

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

   355

   356

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

   357

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

   358

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

   359

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

   360

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

   361

   362

    def _init_coll_AddMenu_Items(self, parent):

   363

        IDEFrame._init_coll_AddMenu_Items(self, parent, False)

   364

   365

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

   366

        #new_id = wx.NewId()

   367

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

   368

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

   369

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

   370

   371

        for name, XSDClass, help in ProjectController.CTNChildrenTypes:

   372

            new_id = wx.NewId()

   373

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

   374

                       kind=wx.ITEM_NORMAL, text=help)

   375

            self.Bind(wx.EVT_MENU, self.GetAddConfNodeFunction(name), id=new_id)

   376

   377

    def _init_coll_HelpMenu_Items(self, parent):

   378

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

   379

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

   380

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

   381

   382

    def _init_coll_ConnectionStatusBar_Fields(self, parent):

   383

        parent.SetFieldsCount(3)

   384

   385

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

   386

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

   387

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

   388

   389

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

   390

   391

    def _init_ctrls(self, prnt):

   392

        IDEFrame._init_ctrls(self, prnt)

   393

   394

        self.EditMenuSize = self.EditMenu.GetMenuItemCount()

   395

   396

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

   397

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

   398

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

   399

                                ("Run",      wx.WXK_F5),

   400

                                ("Transfer", wx.WXK_F6),

   401

                                ("Connect",  wx.WXK_F7),

   402

                                ("Build",    wx.WXK_F11)]:

   403

            def OnMethodGen(obj,meth):

   404

                def OnMethod(evt):

   405

                    if obj.CTR is not None:

   406

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

   407

                    wx.CallAfter(self.RefreshStatusToolBar)

   408

                return OnMethod

   409

            newid = wx.NewId()

   410

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

   411

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

   412

   413

        self.SetAcceleratorTable(wx.AcceleratorTable(accels))

   414

   415

        self.LogConsole = wx.stc.StyledTextCtrl(id=ID_BEREMIZLOGCONSOLE,

   416

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

   417

                  size=wx.Size(0, 0))

   418

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

   419

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

   420

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

   421

        self.LogConsole.SetReadOnly(True)

   422

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

   423

   424

        # Define Log Console styles

   425

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

   426

        self.LogConsole.StyleClearAll()

   427

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

   428

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

   429

   430

        # Define Log Console markers

   431

        self.LogConsole.SetMarginSensitive(1, True)

   432

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

   433

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

   434

   435

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

   436

   437

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

   438

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

   439

   440

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

   441

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

   442

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

   443

   444

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

   445

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

   446

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

   447

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

   448

   449

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

   450

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

   451

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

   452

        StatusToolBar.Realize()

   453

        self.Panes["StatusToolBar"] = StatusToolBar

   454

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

   455

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

   456

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

   457

                  LeftDockable(False).RightDockable(False))

   458

   459

        self.AUIManager.Update()

   460

   461

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

   462

        self._init_coll_ConnectionStatusBar_Fields(self.ConnectionStatusBar)

   463

        self.SetStatusBar(self.ConnectionStatusBar)

   464

   465

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

   466

        IDEFrame.__init__(self, parent, debug)

   467

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

   468

   469

        self.local_runtime = None

   470

        self.runtime_port = None

   471

        self.local_runtime_tmpdir = None

   472

   473

        self.LastPanelSelected = None

   474

   475

        # Define Tree item icon list

   476

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

   477

        self.LocationImageDict = {}

   478

   479

        # Icons for location items

   480

        for imgname, itemtype in [

   481

            ("CONFIGURATION", LOCATION_CONFNODE),

   482

            ("RESOURCE",      LOCATION_MODULE),

   483

            ("PROGRAM",       LOCATION_GROUP),

   484

            ("VAR_INPUT",     LOCATION_VAR_INPUT),

   485

            ("VAR_OUTPUT",    LOCATION_VAR_OUTPUT),