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:

    49

    except getopt.GetoptError:

    50

        # print help information and exit:

    51

        usage()

    52

        sys.exit(2)

    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

    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

# Import module for internationalization

   114

import gettext

   115

   116

# Get folder containing translation files

   117

localedir = os.path.join(CWD,"locale")

   118

# Get the default language

   119

langid = wx.LANGUAGE_DEFAULT

   120

# Define translation domain (name of translation files)

   121

domain = "Beremiz"

   122

   123

# Define locale for wx

   124

loc = __builtin__.__dict__.get('loc', None)

   125

if loc is None:

   126

    test_loc = wx.Locale(langid)

   127

    test_loc.AddCatalogLookupPathPrefix(localedir)

   128

    if test_loc.AddCatalog(domain):

   129

        loc = wx.Locale(langid)

   130

    else:

   131

        loc = wx.Locale(wx.LANGUAGE_ENGLISH)

   132

    __builtin__.__dict__['loc'] = loc

   133

# Define location for searching translation files

   134

loc.AddCatalogLookupPathPrefix(localedir)

   135

# Define locale domain

   136

loc.AddCatalog(domain)

   137

   138

def unicode_translation(message):

   139

    return wx.GetTranslation(message).encode("utf-8")

   140

   141

if __name__ == '__main__':

   142

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

   143

   144

base_folder = os.path.split(sys.path[0])[0]

   145

sys.path.append(base_folder)

   146

sys.path.append(os.path.join(base_folder, "plcopeneditor"))

   147

   148

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

   149

from util.TextCtrlAutoComplete import TextCtrlAutoComplete

   150

import cPickle

   151

from util.BrowseValuesLibraryDialog import BrowseValuesLibraryDialog

   152

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

   153

from ProjectController import ProjectController, MATIEC_ERROR_MODEL

   154

from util.MiniTextControler import MiniTextControler

   155

from util.ProcessLogger import ProcessLogger

   156

   157

from docutil import OpenHtmlFrame

   159

from PLCOpenEditor import EditorPanel, Viewer, TextViewer, GraphicViewer, ResourceEditor, ConfigurationEditor, DataTypeEditor

   160

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

   161

   162

SCROLLBAR_UNIT = 10

   163

WINDOW_COLOUR = wx.Colour(240,240,240)

   164

TITLE_COLOUR = wx.Colour(200,200,220)

   165

CHANGED_TITLE_COLOUR = wx.Colour(220,200,220)

   166

CHANGED_WINDOW_COLOUR = wx.Colour(255,240,240)

   167

   168

if wx.Platform == '__WXMSW__':

   169

    faces = { 'times': 'Times New Roman',

   170

              'mono' : 'Courier New',

   171

              'helv' : 'Arial',

   172

              'other': 'Comic Sans MS',

   173

              'size' : 16,

   174

             }

   175

else:

   176

    faces = { 'times': 'Times',

   177

              'mono' : 'Courier',

   178

              'helv' : 'Helvetica',

   179

              'other': 'new century schoolbook',

   180

              'size' : 18,

   181

             }

   182

   183

MAX_RECENT_PROJECTS = 10

   184

   185

# Some helpers to tweak GenBitmapTextButtons

   186

# TODO: declare customized classes instead.

   187

gen_mini_GetBackgroundBrush = lambda obj:lambda dc: wx.Brush(obj.GetParent().GetBackgroundColour(), wx.SOLID)

   188

gen_textbutton_GetLabelSize = lambda obj:lambda:(wx.lib.buttons.GenButton._GetLabelSize(obj)[:-1] + (False,))

   189

   190

def make_genbitmaptogglebutton_flat(button):

   191

    button.GetBackgroundBrush = gen_mini_GetBackgroundBrush(button)

   192

    button.labelDelta = 0

   193

    button.SetBezelWidth(0)

   194

    button.SetUseFocusIndicator(False)

   195

   196

# Patch wx.lib.imageutils so that gray is supported on alpha images

   197

import wx.lib.imageutils

   198

from wx.lib.imageutils import grayOut as old_grayOut

   199

def grayOut(anImage):

   200

    if anImage.HasAlpha():

   201

        AlphaData = anImage.GetAlphaData()

   202

    else :

   203

        AlphaData = None

   204

   205

    old_grayOut(anImage)

   206

   207

    if AlphaData is not None:

   208

        anImage.SetAlphaData(AlphaData)

   209

   210

wx.lib.imageutils.grayOut = grayOut

   211

   212

class GenBitmapTextButton(wx.lib.buttons.GenBitmapTextButton):

   213

    def _GetLabelSize(self):

   214

        """ used internally """

   215

        w, h = self.GetTextExtent(self.GetLabel())

   216

        if not self.bmpLabel:

   217

            return w, h, False       # if there isn't a bitmap use the size of the text

   218

   219

        w_bmp = self.bmpLabel.GetWidth()+2

   220

        h_bmp = self.bmpLabel.GetHeight()+2

   221

        height = h + h_bmp

   222

        if w_bmp > w:

   223

            width = w_bmp

   224

        else:

   225

            width = w

   226

        return width, height, False

   227

   228

    def DrawLabel(self, dc, width, height, dw=0, dy=0):

   229

        bmp = self.bmpLabel

   230

        if bmp != None:     # if the bitmap is used

   231

            if self.bmpDisabled and not self.IsEnabled():

   232

                bmp = self.bmpDisabled

   233

            if self.bmpFocus and self.hasFocus:

   234

                bmp = self.bmpFocus

   235

            if self.bmpSelected and not self.up:

   236

                bmp = self.bmpSelected

   237

            bw,bh = bmp.GetWidth(), bmp.GetHeight()

   238

            if not self.up:

   239

                dw = dy = self.labelDelta

   240

            hasMask = bmp.GetMask() != None

   241

        else:

   242

            bw = bh = 0     # no bitmap -> size is zero

   243

   244

        dc.SetFont(self.GetFont())

   245

        if self.IsEnabled():

   246

            dc.SetTextForeground(self.GetForegroundColour())

   247

        else:

   248

            dc.SetTextForeground(wx.SystemSettings.GetColour(wx.SYS_COLOUR_GRAYTEXT))

   249

   250

        label = self.GetLabel()

   251

        tw, th = dc.GetTextExtent(label)        # size of text

   252

        if not self.up:

   253

            dw = dy = self.labelDelta

   254

   255

        pos_x = (width-bw)/2+dw      # adjust for bitmap and text to centre

   256

        pos_y = (height-bh-th)/2+dy

   257

        if bmp !=None:

   258

            dc.DrawBitmap(bmp, pos_x, pos_y, hasMask) # draw bitmap if available

   259

            pos_x = (width-tw)/2+dw      # adjust for bitmap and text to centre

   260

            pos_y += bh + 2

   261

   262

        dc.DrawText(label, pos_x, pos_y)      # draw the text

   263

   264

   265

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

   266

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

   267

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

   268

    def __init__(self, parent, ID, bitmapname,

   269

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

   270

                 style = 0,

   271

                 name = "genstatbmp"):

   272

   273

        bitmappath = Bpath( "images", bitmapname)

   274

        if os.path.isfile(bitmappath):

   275

            bitmap = wx.Bitmap(bitmappath)

   276

        else:

   277

            bitmap = None

   278

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

   279

                 pos, size,

   280

                 style,

   281

                 name)

   282

   283

    def OnPaint(self, event):

   284

        dc = wx.PaintDC(self)

   285

        colour = self.GetParent().GetBackgroundColour()

   286

        dc.SetPen(wx.Pen(colour))

   287

        dc.SetBrush(wx.Brush(colour ))

   288

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

   289

        if self._bitmap:

   290

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

   291

   292

   293

from threading import Lock,Timer,currentThread

   294

MainThread = currentThread().ident

   295

REFRESH_PERIOD = 0.1

   296

from time import time as gettime

   297

class LogPseudoFile:

   298

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

   299

    def __init__(self, output, risecall):

   300

        self.red_white = wx.TextAttr("RED", "WHITE")

   301

        self.red_yellow = wx.TextAttr("RED", "YELLOW")

   302

        self.black_white = wx.TextAttr("BLACK", "WHITE")

   303

        self.default_style = None

   304

        self.output = output

   305

        self.risecall = risecall

   306

        # to prevent rapid fire on rising log panel

   307

        self.rising_timer = 0

   308

        self.lock = Lock()

   309

        self.YieldLock = Lock()

   310

        self.RefreshLock = Lock()

   311

        self.stack = []

   312

        self.LastRefreshTime = gettime()

   313

        self.LastRefreshTimer = None

   314

   315

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

   316

        if self.lock.acquire():

   317

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

   318

            self.lock.release()

   319

            current_time = gettime()

   320

            if self.LastRefreshTimer:

   321

                self.LastRefreshTimer.cancel()

   322

                self.LastRefreshTimer=None

   323

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

   324

                self._should_write()

   325

            else:

   326

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

   327

                self.LastRefreshTimer.start()

   328

   329

    def _should_write(self):

   330

        wx.CallAfter(self._write)

   331

        if MainThread == currentThread().ident:

   332

            app = wx.GetApp()

   333

            if app is not None:

   334

                if self.YieldLock.acquire(0):

   335

                    app.Yield()

   336

                    self.YieldLock.release()

   337

   338

    def _write(self):

   339

        if self.output :

   340

            self.output.Freeze(); 

   341

            self.lock.acquire()

   342

            for s, style in self.stack:

   343

                if style is None : style=self.black_white

   344

                if self.default_style != style: 

   345

                    self.output.SetDefaultStyle(style)

   346

                    self.default_style = style

   347

                self.output.AppendText(s)

   348

                self.output.ScrollLines(s.count('\n')+1)

   349

            self.stack = []

   350

            self.lock.release()

   351

            self.output.ShowPosition(self.output.GetLastPosition())

   352

            self.output.Thaw()

   353

            self.LastRefreshTime = gettime()

   354

            try:

   355

                self.RefreshLock.release()

   356

            except:

   357

                pass

   358

            newtime = time.time()

   359

            if newtime - self.rising_timer > 1:

   360

                self.risecall()

   361

            self.rising_timer = newtime

   362

   363

    def write_warning(self, s):

   364

        self.write(s,self.red_white)

   365

   366

    def write_error(self, s):

   367

        self.write(s,self.red_yellow)

   368

   369

    def writeyield(self, s):

   370

        self.write(s)

   371

        wx.GetApp().Yield()

   372

   373

    def flush(self):

   374

        self.output.SetValue("")

   375

   376

    def isatty(self):

   377

        return false

   378

   379

[ID_BEREMIZ, ID_BEREMIZMAINSPLITTER, 

   380

 ID_BEREMIZPLCCONFIG, ID_BEREMIZLOGCONSOLE, 

   381

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

   382

   383

[ID_FILEMENURECENTPROJECTS,

   384

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

   385

   386

CONFNODEMENU_POSITION = 3

   387

   388

class Beremiz(IDEFrame):

   389

   390

    def _init_coll_MenuBar_Menus(self, parent):

   391

        IDEFrame._init_coll_MenuBar_Menus(self, parent)

   392

   393

        parent.Insert(pos=CONFNODEMENU_POSITION, 

   394

                      menu=self.ConfNodeMenu, title=_(u'&ConfNode'))

   395

   396

    def _init_utils(self):

   397

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

   398

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

   399

   400

        IDEFrame._init_utils(self)

   401

   402

    def _init_coll_FileMenu_Items(self, parent):

   403

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

   404

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

   405

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

   406

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

   407

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

   408

        parent.AppendSeparator()

   409

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

   410

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

   411

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

   412

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

   413

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

   414

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

   415

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

   416

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

   417

        parent.AppendSeparator()

   418

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

   419

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

   420

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

   421

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

   422

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

   423

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

   424

        parent.AppendSeparator()

   425

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

   426

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

   427

        parent.AppendSeparator()

   428

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

   429

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

   430

   431

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

   432

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

   433

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

   434

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

   435

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

   436

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

   437

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

   438

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

   439

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

   440

        self.Bind(wx.EVT_MENU, self.OnPropertiesMenu, id=wx.ID_PROPERTIES)

   441

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

   442

   443

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

   444

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

   445

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

   446

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

   447

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

   448

   449

    def _init_coll_HelpMenu_Items(self, parent):

   450

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

   451

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

   452

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

   453

   454

    def _init_coll_PLCConfigMainSizer_Items(self, parent):

   455

        parent.AddSizer(self.PLCParamsSizer, 0, border=10, flag=wx.GROW|wx.TOP|wx.LEFT|wx.RIGHT)

   456

        parent.AddSizer(self.ConfNodeTreeSizer, 0, border=10, flag=wx.BOTTOM|wx.LEFT|wx.RIGHT)

   457

   458

    def _init_coll_PLCConfigMainSizer_Growables(self, parent):

   459

        parent.AddGrowableCol(0)

   460

        parent.AddGrowableRow(1)

   461

   462

    def _init_coll_ConfNodeTreeSizer_Growables(self, parent):

   463

        parent.AddGrowableCol(0)

   464

        parent.AddGrowableCol(1)

   465

   466

    def _init_beremiz_sizers(self):

   467

        self.PLCConfigMainSizer = wx.FlexGridSizer(cols=1, hgap=2, rows=2, vgap=2)

   468

        self.PLCParamsSizer = wx.BoxSizer(wx.VERTICAL)

   469

        #self.ConfNodeTreeSizer = wx.FlexGridSizer(cols=3, hgap=0, rows=0, vgap=2)

   470

        self.ConfNodeTreeSizer = wx.FlexGridSizer(cols=2, hgap=0, rows=0, vgap=2)

   471

   472

        self._init_coll_PLCConfigMainSizer_Items(self.PLCConfigMainSizer)

   473

        self._init_coll_PLCConfigMainSizer_Growables(self.PLCConfigMainSizer)

   474

        self._init_coll_ConfNodeTreeSizer_Growables(self.ConfNodeTreeSizer)

   475

   476

        self.PLCConfig.SetSizer(self.PLCConfigMainSizer)

   477

   478

    def _init_ctrls(self, prnt):

   479

        IDEFrame._init_ctrls(self, prnt)

   480

   481

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

   482

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

   483

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

   484

                                ("Run",      wx.WXK_F5),

   485

                                ("Transfer", wx.WXK_F6),

   486

                                ("Connect",  wx.WXK_F7),

   487

                                ("Build",    wx.WXK_F11)]:

   488

            def OnMethodGen(obj,meth):

   489

                def OnMethod(evt):

   490

                    if obj.CTR is not None: