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:", ["help", "updatecheck="])

    49

    except getopt.GetoptError:

    50

        # print help information and exit:

    51

        usage()

    52

        sys.exit(2)

    53

    54

    for o, a in opts:

    55

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

    56

            usage()

    57

            sys.exit()

    58

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

    59

            updateinfo_url = a

    60

    61

    if len(args) > 2:

    62

        usage()

    63

        sys.exit()

    64

    elif len(args) == 1:

    65

        projectOpen = args[0]

    66

        buildpath = None

    67

    elif len(args) == 2:

    68

        projectOpen = args[0]

    69

        buildpath = args[1]

    70

    else:

    71

        projectOpen = None

    72

        buildpath = None

    73

    74

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

    75

        __builtin__.__dict__["BMZ_DBG"] = True

    76

    else :

    77

        __builtin__.__dict__["BMZ_DBG"] = False

    78

    79

    app = wx.PySimpleApp(redirect=BMZ_DBG)

    80

    app.SetAppName('beremiz')

    81

    wx.InitAllImageHandlers()

    82

    83

    # popup splash

    84

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

    85

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

    86

    splash=AdvancedSplash(None, bitmap=bmp)

    87

    wx.Yield()

    88

    89

    if updateinfo_url is not None:

    90

        updateinfo = "Fetching %s" % updateinfo_url

    91

        # warn for possible updates

    92

        def updateinfoproc():

    93

            global updateinfo

    94

            try :

    95

                import urllib2

    96

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

    97

            except :

    98

                updateinfo = "update info unavailable." 

    99

   100

        from threading import Thread

   101

        splash.SetText(text=updateinfo)

   102

        wx.Yield()

   103

        updateinfoThread = Thread(target=updateinfoproc)

   104

        updateinfoThread.start()

   105

        updateinfoThread.join(2)

   106

        splash.SetText(text=updateinfo)

   107

        wx.Yield()

   108

   109

# Import module for internationalization

   110

import gettext

   111

   112

# Get folder containing translation files

   113

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

   114

# Get the default language

   115

langid = wx.LANGUAGE_DEFAULT

   116

# Define translation domain (name of translation files)

   117

domain = "Beremiz"

   118

   119

# Define locale for wx

   120

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

   121

if loc is None:

   122

    test_loc = wx.Locale(langid)

   123

    test_loc.AddCatalogLookupPathPrefix(localedir)

   124

    if test_loc.AddCatalog(domain):

   125

        loc = wx.Locale(langid)

   126

    else:

   127

        loc = wx.Locale(wx.LANGUAGE_ENGLISH)

   128

    __builtin__.__dict__['loc'] = loc

   129

# Define location for searching translation files

   130

loc.AddCatalogLookupPathPrefix(localedir)

   131

# Define locale domain

   132

loc.AddCatalog(domain)

   133

   134

def unicode_translation(message):

   135

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

   136

   137

if __name__ == '__main__':

   138

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

   139

   140

#Quick hack to be able to find Beremiz IEC tools. Should be config params.

   141

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

   142

sys.path.append(base_folder)

   143

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

   144

   145

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

   146

import TextCtrlAutoComplete, cPickle

   147

from BrowseValuesLibraryDialog import BrowseValuesLibraryDialog

   148

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

   149

from plugger import PluginsRoot, MATIEC_ERROR_MODEL

   150

from wxPopen import ProcessLogger

   151

   152

from docutils import *

   153

from PLCOpenEditor import IDEFrame, Viewer, AppendMenu, TITLE, EDITORTOOLBAR, FILEMENU, EDITMENU, DISPLAYMENU, TYPESTREE, INSTANCESTREE, LIBRARYTREE, SCALING, PAGETITLES, USE_AUI

   154

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

   155

   156

SCROLLBAR_UNIT = 10

   157

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

   158

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

   159

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

   160

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

   161

   162

if wx.Platform == '__WXMSW__':

   163

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

   164

              'mono' : 'Courier New',

   165

              'helv' : 'Arial',

   166

              'other': 'Comic Sans MS',

   167

              'size' : 16,

   168

             }

   169

else:

   170

    faces = { 'times': 'Times',

   171

              'mono' : 'Courier',

   172

              'helv' : 'Helvetica',

   173

              'other': 'new century schoolbook',

   174

              'size' : 18,

   175

             }

   176

   177

MAX_RECENT_PROJECTS = 10

   178

   179

# Some helpers to tweak GenBitmapTextButtons

   180

# TODO: declare customized classes instead.

   181

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

   182

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

   183

   184

def make_genbitmaptogglebutton_flat(button):

   185

    button.GetBackgroundBrush = gen_mini_GetBackgroundBrush(button)

   186

    button.labelDelta = 0

   187

    button.SetBezelWidth(0)

   188

    button.SetUseFocusIndicator(False)

   189

   190

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

   191

import wx.lib.imageutils

   192

from wx.lib.imageutils import grayOut as old_grayOut

   193

def grayOut(anImage):

   194

    if anImage.HasAlpha():

   195

        AlphaData = anImage.GetAlphaData()

   196

    else :

   197

        AlphaData = None

   198

   199

    old_grayOut(anImage)

   200

   201

    if AlphaData is not None:

   202

        anImage.SetAlphaData(AlphaData)

   203

   204

wx.lib.imageutils.grayOut = grayOut

   205

   206

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

   207

    def _GetLabelSize(self):

   208

        """ used internally """

   209

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

   210

        if not self.bmpLabel:

   211

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

   212

   213

        w_bmp = self.bmpLabel.GetWidth()+2

   214

        h_bmp = self.bmpLabel.GetHeight()+2

   215

        height = h + h_bmp

   216

        if w_bmp > w:

   217

            width = w_bmp

   218

        else:

   219

            width = w

   220

        return width, height, False

   221

   222

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

   223

        bmp = self.bmpLabel

   224

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

   225

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

   226

                bmp = self.bmpDisabled

   227

            if self.bmpFocus and self.hasFocus:

   228

                bmp = self.bmpFocus

   229

            if self.bmpSelected and not self.up:

   230

                bmp = self.bmpSelected

   231

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

   232

            if not self.up:

   233

                dw = dy = self.labelDelta

   234

            hasMask = bmp.GetMask() != None

   235

        else:

   236

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

   237

   238

        dc.SetFont(self.GetFont())

   239

        if self.IsEnabled():

   240

            dc.SetTextForeground(self.GetForegroundColour())

   241

        else:

   242

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

   243

   244

        label = self.GetLabel()

   245

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

   246

        if not self.up:

   247

            dw = dy = self.labelDelta

   248

   249

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

   250

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

   251

        if bmp !=None:

   252

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

   253

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

   254

            pos_y += bh + 2

   255

   256

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

   257

   258

   259

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

   260

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

   261

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

   262

    def __init__(self, parent, ID, bitmapname,

   263

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

   264

                 style = 0,

   265

                 name = "genstatbmp"):

   266

   267

        bitmappath = Bpath( "images", bitmapname)

   268

        if os.path.isfile(bitmappath):

   269

            bitmap = wx.Bitmap(bitmappath)

   270

        else:

   271

            bitmap = None

   272

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

   273

                 pos, size,

   274

                 style,

   275

                 name)

   276

   277

    def OnPaint(self, event):

   278

        dc = wx.PaintDC(self)

   279

        colour = self.GetParent().GetBackgroundColour()

   280

        dc.SetPen(wx.Pen(colour))

   281

        dc.SetBrush(wx.Brush(colour ))

   282

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

   283

        if self._bitmap:

   284

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

   285

   286

   287

from threading import Lock,Timer,currentThread

   288

MainThread = currentThread().ident

   289

REFRESH_PERIOD = 0.1

   290

from time import time as gettime

   291

class LogPseudoFile:

   292

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

   293

    def __init__(self, output, risecall):

   294

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

   295

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

   296

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

   297

        self.default_style = None

   298

        self.output = output

   299

        self.risecall = risecall

   300

        # to prevent rapid fire on rising log panel

   301

        self.rising_timer = 0

   302

        self.lock = Lock()

   303

        self.YieldLock = Lock()

   304

        self.RefreshLock = Lock()

   305

        self.stack = []

   306

        self.LastRefreshTime = gettime()

   307

        self.LastRefreshTimer = None

   308

   309

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

   310

        if self.lock.acquire():

   311

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

   312

            self.lock.release()

   313

            current_time = gettime()

   314

            if self.LastRefreshTimer:

   315

                self.LastRefreshTimer.cancel()

   316

                self.LastRefreshTimer=None

   317

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

   318

                self._should_write()

   319

            else:

   320

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

   321

                self.LastRefreshTimer.start()

   322

   323

    def _should_write(self):

   324

        wx.CallAfter(self._write)

   325

        if MainThread == currentThread().ident:

   326

            app = wx.GetApp()

   327

            if app is not None:

   328

                if self.YieldLock.acquire(0):

   329

                    app.Yield()

   330

                    self.YieldLock.release()

   331

   332

    def _write(self):

   333

        if self.output :

   334

            self.output.Freeze(); 

   335

            self.lock.acquire()

   336

            for s, style in self.stack:

   337

                if style is None : style=self.black_white

   338

                if self.default_style != style: 

   339

                    self.output.SetDefaultStyle(style)

   340

                    self.default_style = style

   341

                self.output.AppendText(s)

   342

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

   343

            self.stack = []

   344

            self.lock.release()

   345

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

   346

            self.output.Thaw()

   347

            self.LastRefreshTime = gettime()

   348

            try:

   349

                self.RefreshLock.release()

   350

            except:

   351

                pass

   352

            newtime = time.time()

   353

            if newtime - self.rising_timer > 1:

   354

                self.risecall()

   355

            self.rising_timer = newtime

   356

   357

    def write_warning(self, s):

   358

        self.write(s,self.red_white)

   359

   360

    def write_error(self, s):

   361

        self.write(s,self.red_yellow)

   362

   363

    def writeyield(self, s):

   364

        self.write(s)

   365

        wx.GetApp().Yield()

   366

   367

    def flush(self):

   368

        self.output.SetValue("")

   369

   370

    def isatty(self):

   371

        return false

   372

   373

[ID_BEREMIZ, ID_BEREMIZMAINSPLITTER, 

   374

 ID_BEREMIZPLCCONFIG, ID_BEREMIZLOGCONSOLE, 

   375

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

   376

   377

[ID_FILEMENURECENTPROJECTS,

   378

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

   379

   380

PLUGINMENU_POSITION = 3

   381

   382

class Beremiz(IDEFrame):

   383

   384

    def _init_coll_MenuBar_Menus(self, parent):

   385

        IDEFrame._init_coll_MenuBar_Menus(self, parent)

   386

   387

        parent.Insert(pos=PLUGINMENU_POSITION, 

   388

                      menu=self.PluginMenu, title=_(u'&Plugin'))

   389

   390

    def _init_utils(self):

   391

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

   392

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

   393

   394

        IDEFrame._init_utils(self)

   395

   396

    def _init_coll_FileMenu_Items(self, parent):

   397

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

   398

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

   399

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

   400

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

   401

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

   402

        parent.AppendSeparator()

   403

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

   404

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

   405

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

   406

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

   407

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

   408

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

   409

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

   410

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

   411

        parent.AppendSeparator()

   412

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

   413

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

   414

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

   415

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

   416

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

   417

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

   418

        parent.AppendSeparator()

   419

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

   420

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

   421

        parent.AppendSeparator()

   422

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

   423

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

   424

   425

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

   426

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

   427

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

   428

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

   429

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

   430

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

   431

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

   432

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

   433

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

   434

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

   435

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

   436

   437

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

   438

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

   439

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

   440

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

   441

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

   442

   443

    def _init_coll_HelpMenu_Items(self, parent):

   444

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

   445

              kind=wx.ITEM_NORMAL, text=_(u'Beremiz\tF1'))

   446

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

   447

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

   448

        self.Bind(wx.EVT_MENU, self.OnBeremizMenu, id=wx.ID_HELP)

   449

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

   450

   451

    def _init_coll_PLCConfigMainSizer_Items(self, parent):

   452

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

   453

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

   454

   455

    def _init_coll_PLCConfigMainSizer_Growables(self, parent):

   456

        parent.AddGrowableCol(0)

   457

        parent.AddGrowableRow(1)

   458

   459

    def _init_coll_PluginTreeSizer_Growables(self, parent):

   460

        parent.AddGrowableCol(0)

   461

        parent.AddGrowableCol(1)

   462

   463

    def _init_beremiz_sizers(self):

   464

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

   465

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

   466

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

   467

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

   468

   469

        self._init_coll_PLCConfigMainSizer_Items(self.PLCConfigMainSizer)

   470

        self._init_coll_PLCConfigMainSizer_Growables(self.PLCConfigMainSizer)

   471

        self._init_coll_PluginTreeSizer_Growables(self.PluginTreeSizer)

   472

   473

        self.PLCConfig.SetSizer(self.PLCConfigMainSizer)

   474

   475

    def _init_ctrls(self, prnt):

   476

        IDEFrame._init_ctrls(self, prnt)

   477

   478

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

   479

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

   480

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

   481

                                ("Run",      wx.WXK_F5),

   482

                                ("Transfer", wx.WXK_F6),

   483

                                ("Connect",  wx.WXK_F7),

   484

                                ("Build",    wx.WXK_F11)]:

   485

            def OnMethodGen(obj,meth):

   486

                def OnMethod(evt):

   487

                    if obj.PluginRoot is not None:

   488

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

   489

                    wx.CallAfter(self.RefreshAll)

   490

                return OnMethod

   491

            newid = wx.NewId()