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:], "hue:", ["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

# 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

   154

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

   155

from util.TextCtrlAutoComplete import TextCtrlAutoComplete

   156

import cPickle

   157

from util.BrowseValuesLibraryDialog import BrowseValuesLibraryDialog

   158

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

   159

from ProjectController import ProjectController, MATIEC_ERROR_MODEL

   160

from util.MiniTextControler import MiniTextControler

   161

from util.ProcessLogger import ProcessLogger

   162

   163

from docutil import OpenHtmlFrame

   164

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

   165

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

   166

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

   167

   168

SCROLLBAR_UNIT = 10

   169

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

   170

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

   171

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

   172

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

   173

   174

if wx.Platform == '__WXMSW__':

   175

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

   176

              'mono' : 'Courier New',

   177

              'helv' : 'Arial',

   178

              'other': 'Comic Sans MS',

   179

              'size' : 16,

   180

             }

   181

else:

   182

    faces = { 'times': 'Times',

   183

              'mono' : 'Courier',

   184

              'helv' : 'Helvetica',

   185

              'other': 'new century schoolbook',

   186

              'size' : 18,

   187

             }

   188

   189

MAX_RECENT_PROJECTS = 10

   190

   191

# Some helpers to tweak GenBitmapTextButtons

   192

# TODO: declare customized classes instead.

   193

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

   194

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

   195

   196

def make_genbitmaptogglebutton_flat(button):

   197

    button.GetBackgroundBrush = gen_mini_GetBackgroundBrush(button)

   198

    button.labelDelta = 0

   199

    button.SetBezelWidth(0)

   200

    button.SetUseFocusIndicator(False)

   201

   202

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

   203

import wx.lib.imageutils

   204

from wx.lib.imageutils import grayOut as old_grayOut

   205

def grayOut(anImage):

   206

    if anImage.HasAlpha():

   207

        AlphaData = anImage.GetAlphaData()

   208

    else :

   209

        AlphaData = None

   210

   211

    old_grayOut(anImage)

   212

   213

    if AlphaData is not None:

   214

        anImage.SetAlphaData(AlphaData)

   215

   216

wx.lib.imageutils.grayOut = grayOut

   217

   218

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

   219

    def _GetLabelSize(self):

   220

        """ used internally """

   221

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

   222

        if not self.bmpLabel:

   223

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

   224

   225

        w_bmp = self.bmpLabel.GetWidth()+2

   226

        h_bmp = self.bmpLabel.GetHeight()+2

   227

        height = h + h_bmp

   228

        if w_bmp > w:

   229

            width = w_bmp

   230

        else:

   231

            width = w

   232

        return width, height, False

   233

   234

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

   235

        bmp = self.bmpLabel

   236

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

   237

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

   238

                bmp = self.bmpDisabled

   239

            if self.bmpFocus and self.hasFocus:

   240

                bmp = self.bmpFocus

   241

            if self.bmpSelected and not self.up:

   242

                bmp = self.bmpSelected

   243

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

   244

            if not self.up:

   245

                dw = dy = self.labelDelta

   246

            hasMask = bmp.GetMask() != None

   247

        else:

   248

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

   249

   250

        dc.SetFont(self.GetFont())

   251

        if self.IsEnabled():

   252

            dc.SetTextForeground(self.GetForegroundColour())

   253

        else:

   254

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

   255

   256

        label = self.GetLabel()

   257

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

   258

        if not self.up:

   259

            dw = dy = self.labelDelta

   260

   261

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

   262

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

   263

        if bmp !=None:

   264

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

   265

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

   266

            pos_y += bh + 2

   267

   268

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

   269

   270

   271

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

   272

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

   273

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

   274

    def __init__(self, parent, ID, bitmapname,

   275

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

   276

                 style = 0,

   277

                 name = "genstatbmp"):

   278

   279

        bitmappath = Bpath( "images", bitmapname)

   280

        if os.path.isfile(bitmappath):

   281

            bitmap = wx.Bitmap(bitmappath)

   282

        else:

   283

            bitmap = None

   284

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

   285

                 pos, size,

   286

                 style,

   287

                 name)

   288

   289

    def OnPaint(self, event):

   290

        dc = wx.PaintDC(self)

   291

        colour = self.GetParent().GetBackgroundColour()

   292

        dc.SetPen(wx.Pen(colour))

   293

        dc.SetBrush(wx.Brush(colour ))

   294

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

   295

        if self._bitmap:

   296

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

   297

   298

   299

from threading import Lock,Timer,currentThread

   300

MainThread = currentThread().ident

   301

REFRESH_PERIOD = 0.1

   302

from time import time as gettime

   303

class LogPseudoFile:

   304

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

   305

    def __init__(self, output, risecall):

   306

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

   307

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

   308

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

   309

        self.default_style = None

   310

        self.output = output

   311

        self.risecall = risecall

   312

        # to prevent rapid fire on rising log panel

   313

        self.rising_timer = 0

   314

        self.lock = Lock()

   315

        self.YieldLock = Lock()

   316

        self.RefreshLock = Lock()

   317

        self.stack = []

   318

        self.LastRefreshTime = gettime()

   319

        self.LastRefreshTimer = None

   320

   321

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

   322

        if self.lock.acquire():

   323

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

   324

            self.lock.release()

   325

            current_time = gettime()

   326

            if self.LastRefreshTimer:

   327

                self.LastRefreshTimer.cancel()

   328

                self.LastRefreshTimer=None

   329

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

   330

                self._should_write()

   331

            else:

   332

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

   333

                self.LastRefreshTimer.start()

   334

   335

    def _should_write(self):

   336

        wx.CallAfter(self._write)

   337

        if MainThread == currentThread().ident:

   338

            app = wx.GetApp()

   339

            if app is not None:

   340

                if self.YieldLock.acquire(0):

   341

                    app.Yield()

   342

                    self.YieldLock.release()

   343

   344

    def _write(self):

   345

        if self.output :

   346

            self.output.Freeze(); 

   347

            self.lock.acquire()

   348

            for s, style in self.stack:

   349

                if style is None : style=self.black_white

   350

                if self.default_style != style: 

   351

                    self.output.SetDefaultStyle(style)

   352

                    self.default_style = style

   353

                self.output.AppendText(s)

   354

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

   355

            self.stack = []

   356

            self.lock.release()

   357

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

   358

            self.output.Thaw()

   359

            self.LastRefreshTime = gettime()

   360

            try:

   361

                self.RefreshLock.release()

   362

            except:

   363

                pass

   364

            newtime = time.time()

   365

            if newtime - self.rising_timer > 1:

   366

                self.risecall()

   367

            self.rising_timer = newtime

   368

   369

    def write_warning(self, s):

   370

        self.write(s,self.red_white)

   371

   372

    def write_error(self, s):

   373

        self.write(s,self.red_yellow)

   374

   375

    def writeyield(self, s):

   376

        self.write(s)

   377

        wx.GetApp().Yield()

   378

   379

    def flush(self):

   380

        self.output.SetValue("")

   381

   382

    def isatty(self):

   383

        return false

   384

   385

[ID_BEREMIZ, ID_BEREMIZMAINSPLITTER, 

   386

 ID_BEREMIZPLCCONFIG, ID_BEREMIZLOGCONSOLE, 

   387

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

   388

   389

[ID_FILEMENURECENTPROJECTS,

   390

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

   391

   392

CONFNODEMENU_POSITION = 3

   393

   394

class Beremiz(IDEFrame):

   395

   396

    def _init_coll_MenuBar_Menus(self, parent):

   397

        IDEFrame._init_coll_MenuBar_Menus(self, parent)

   398

   399

        parent.Insert(pos=CONFNODEMENU_POSITION, 

   400

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

   401

   402

    def _init_utils(self):

   403

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

   404

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

   405

   406

        IDEFrame._init_utils(self)

   407

   408

    def _init_coll_FileMenu_Items(self, parent):

   409

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

   410

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

   411

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

   412

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

   413

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

   414

        parent.AppendSeparator()

   415

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

   416

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

   417

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

   418

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

   419

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

   420

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

   421

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

   422

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

   423

        parent.AppendSeparator()

   424

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

   425

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

   426

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

   427

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

   428

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

   429

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

   430

        parent.AppendSeparator()

   431

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

   432

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

   433

        parent.AppendSeparator()

   434

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

   435

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

   436

   437

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

   438

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

   439

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

   440

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

   441

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

   442

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

   443

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

   444

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

   445

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

   446

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

   447

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

   448

   449

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

   450

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

   451

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

   452

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

   453

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

   454

   455

    def _init_coll_HelpMenu_Items(self, parent):

   456

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

   457

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

   458

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

   459

   460

    def _init_coll_PLCConfigMainSizer_Items(self, parent):

   461

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

   462

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

   463

   464

    def _init_coll_PLCConfigMainSizer_Growables(self, parent):

   465

        parent.AddGrowableCol(0)

   466

        parent.AddGrowableRow(1)

   467

   468

    def _init_coll_ConfNodeTreeSizer_Growables(self, parent):

   469

        parent.AddGrowableCol(0)

   470

        parent.AddGrowableCol(1)

   471

   472

    def _init_beremiz_sizers(self):

   473

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

   474

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

   475

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

   476

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

   477

   478

        self._init_coll_PLCConfigMainSizer_Items(self.PLCConfigMainSizer)

   479

        self._init_coll_PLCConfigMainSizer_Growables(self.PLCConfigMainSizer)

   480

        self._init_coll_ConfNodeTreeSizer_Growables(self.ConfNodeTreeSizer)

   481

   482

        self.PLCConfig.SetSizer(self.PLCConfigMainSizer)

   483

   484

    def _init_ctrls(self, prnt):

   485

        IDEFrame._init_ctrls(self, prnt)

   486

   487

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

   488

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

   489

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

   490

                                ("Run",      wx.WXK_F5),