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

from controls.CustomStyledTextCtrl import CustomStyledTextCtrl

   151

   152

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

   153

from ProjectController import ProjectController, MATIEC_ERROR_MODEL, ITEM_CONFNODE

   154

   155

   156

MAX_RECENT_PROJECTS = 10

   157

   158

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

   159

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

   160

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

   161

    def __init__(self, parent, ID, bitmapname,

   162

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

   163

                 style = 0,

   164

                 name = "genstatbmp"):

   165

   166

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

   167

                 GetBitmap(bitmapname),

   168

                 pos, size,

   169

                 style,

   170

                 name)

   171

   172

    def OnPaint(self, event):

   173

        dc = wx.PaintDC(self)

   174

        colour = self.GetParent().GetBackgroundColour()

   175

        dc.SetPen(wx.Pen(colour))

   176

        dc.SetBrush(wx.Brush(colour ))

   177

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

   178

        if self._bitmap:

   179

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

   180

   181

if wx.Platform == '__WXMSW__':

   182

    faces = {

   183

        'mono' : 'Courier New',

   184

        'size' : 8,

   185

    }

   186

else:

   187

    faces = {

   188

        'mono' : 'Courier',

   189

        'size' : 10,

   190

    }

   191

   192

from threading import Lock,Timer,currentThread

   193

MainThread = currentThread().ident

   194

REFRESH_PERIOD = 0.1

   195

from time import time as gettime

   196

class LogPseudoFile:

   197

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

   198

    def __init__(self, output, risecall):

   199

        self.red_white = 1

   200

        self.red_yellow = 2

   201

        self.black_white = wx.stc.STC_STYLE_DEFAULT

   202

        self.output = output

   203

        self.risecall = risecall

   204

        # to prevent rapid fire on rising log panel

   205

        self.rising_timer = 0

   206

        self.lock = Lock()

   207

        self.YieldLock = Lock()

   208

        self.RefreshLock = Lock()

   209

        self.TimerAccessLock = Lock()

   210

        self.stack = []

   211

        self.LastRefreshTime = gettime()

   212

        self.LastRefreshTimer = None

   213

   214

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

   215

        if self.lock.acquire():

   216

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

   217

            self.lock.release()

   218

            current_time = gettime()

   219

            self.TimerAccessLock.acquire()

   220

            if self.LastRefreshTimer:

   221

                self.LastRefreshTimer.cancel()

   222

                self.LastRefreshTimer=None

   223

            self.TimerAccessLock.release()

   224

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

   225

                self._should_write()

   226

            else:

   227

                self.TimerAccessLock.acquire()

   228

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

   229

                self.LastRefreshTimer.start()

   230

                self.TimerAccessLock.release()

   231

   232

    def _timer_expired(self):

   233

        if self.RefreshLock.acquire(False):

   234

            self._should_write()

   235

        else:

   236

            self.TimerAccessLock.acquire()

   237

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

   238

            self.LastRefreshTimer.start()

   239

            self.TimerAccessLock.release()

   240

   241

    def _should_write(self):

   242

        wx.CallAfter(self._write)

   243

        if MainThread == currentThread().ident:

   244

            app = wx.GetApp()

   245

            if app is not None:

   246

                if self.YieldLock.acquire(0):

   247

                    app.Yield()

   248

                    self.YieldLock.release()

   249

   250

    def _write(self):

   251

        if self.output :

   252

            self.output.Freeze()

   253

            self.lock.acquire()

   254

            for s, style in self.stack:

   255

                if style is None : style=self.black_white

   256

                if style != self.black_white:

   257

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

   258

   259

                # Temporary deactivate read only mode on StyledTextCtrl for

   260

                # adding text. It seems that text modifications, even 

   261

                # programmatically, are disabled in StyledTextCtrl when read

   262

                # only is active 

   263

                self.output.SetReadOnly(False)

   264

                self.output.AppendText(s)

   265

                self.output.SetReadOnly(True)

   266

   267

                if style != self.black_white:

   268

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

   269

            self.stack = []

   270

            self.lock.release()

   271

            self.output.Thaw()

   272

            self.LastRefreshTime = gettime()

   273

            try:

   274

                self.RefreshLock.release()

   275

            except:

   276

                pass

   277

            newtime = time.time()

   278

            if newtime - self.rising_timer > 1:

   279

                self.risecall(self.output)

   280

            self.rising_timer = newtime

   281

   282

    def write_warning(self, s):

   283

        self.write(s,self.red_white)

   284

   285

    def write_error(self, s):

   286

        self.write(s,self.red_yellow)

   287

   288

    def writeyield(self, s):

   289

        self.write(s)

   290

        wx.GetApp().Yield()

   291

   292

    def flush(self):

   293

        # Temporary deactivate read only mode on StyledTextCtrl for clearing

   294

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

   295

        # disabled in StyledTextCtrl when read only is active 

   296

        self.output.SetReadOnly(False)

   297

        self.output.SetText("")

   298

        self.output.SetReadOnly(True)

   299

   300

    def isatty(self):

   301

        return false

   302

   303

[ID_BEREMIZ, ID_BEREMIZMAINSPLITTER, 

   304

 ID_BEREMIZPLCCONFIG, ID_BEREMIZLOGCONSOLE, 

   305

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

   306

   307

[ID_FILEMENURECENTPROJECTS,

   308

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

   309

   310

CONFNODEMENU_POSITION = 3

   311

   312

class Beremiz(IDEFrame):

   313

   314

    def _init_utils(self):

   315

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

   316

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

   317

   318

        IDEFrame._init_utils(self)

   319

   320

    def _init_coll_FileMenu_Items(self, parent):

   321

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

   322

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

   323

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

   324

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

   325

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

   326

        parent.AppendSeparator()

   327

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

   328

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

   329

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

   330

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

   331

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

   332

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

   333

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

   334

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

   335

        parent.AppendSeparator()

   336

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

   337

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

   338

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

   339

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

   340

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

   341

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

   342

        parent.AppendSeparator()

   343

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

   344

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

   345

   346

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

   347

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

   348

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

   349

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

   350

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

   351

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

   352

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

   353

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

   354

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

   355

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

   356

   357

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

   358

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

   359

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

   360

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

   361

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

   362

   363

    def _init_coll_AddMenu_Items(self, parent):

   364

        IDEFrame._init_coll_AddMenu_Items(self, parent, False)

   365

   366

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

   367

        #new_id = wx.NewId()

   368

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

   369

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

   370

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

   371

   372

        for name, XSDClass, help in ProjectController.CTNChildrenTypes:

   373

            new_id = wx.NewId()

   374

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

   375

                       kind=wx.ITEM_NORMAL, text=help)

   376

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

   377

   378

    def _init_coll_HelpMenu_Items(self, parent):

   379

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

   380

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

   381

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

   382

   383

    def _init_coll_ConnectionStatusBar_Fields(self, parent):

   384

        parent.SetFieldsCount(3)

   385

   386

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

   387

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

   388

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

   389

   390

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

   391

   392

    def _init_ctrls(self, prnt):

   393

        IDEFrame._init_ctrls(self, prnt)

   394

   395

        self.EditMenuSize = self.EditMenu.GetMenuItemCount()

   396

   397

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

   398

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

   399

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

   400

                                ("Run",      wx.WXK_F5),

   401

                                ("Transfer", wx.WXK_F6),

   402

                                ("Connect",  wx.WXK_F7),

   403

                                ("Build",    wx.WXK_F11)]:

   404

            def OnMethodGen(obj,meth):

   405

                def OnMethod(evt):

   406

                    if obj.CTR is not None:

   407

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

   408

                    wx.CallAfter(self.RefreshStatusToolBar)

   409

                return OnMethod

   410

            newid = wx.NewId()

   411

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

   412

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

   413

   414

        self.SetAcceleratorTable(wx.AcceleratorTable(accels))

   415

   416

        self.LogConsole = CustomStyledTextCtrl(id=ID_BEREMIZLOGCONSOLE,

   417

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

   418

                  size=wx.Size(0, 0))

   419

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

   420

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

   421

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

   422

        self.LogConsole.SetReadOnly(True)

   423

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

   424

   425

        # Define Log Console styles

   426

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

   427

        self.LogConsole.StyleClearAll()

   428

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

   429

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

   430

   431

        # Define Log Console markers

   432

        self.LogConsole.SetMarginSensitive(1, True)

   433

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

   434

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

   435

   436

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

   437

   438

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

   439

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

   440

   441

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

   442

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

   443

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

   444

   445

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

   446

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

   447

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

   448

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

   449

   450

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

   451

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

   452

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

   453

        StatusToolBar.Realize()

   454

        self.Panes["StatusToolBar"] = StatusToolBar

   455

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

   456

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

   457

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

   458

                  LeftDockable(False).RightDockable(False))

   459

   460

        self.AUIManager.Update()

   461

   462

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

   463

        self._init_coll_ConnectionStatusBar_Fields(self.ConnectionStatusBar)

   464

        self.SetStatusBar(self.ConnectionStatusBar)

   465

   466

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

   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 [

   482

            ("CONFIGURATION", LOCATION_CONFNODE),

   483

            ("RESOURCE",      LOCATION_MODULE),

   484

            ("PROGRAM",       LOCATION_GROUP),

   485