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

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

   141

sys.path.append(base_folder)

   142

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

   143

   144

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

   145

from util.TextCtrlAutoComplete import TextCtrlAutoComplete

   146

import cPickle

   147

from util.BrowseValuesLibraryDialog import BrowseValuesLibraryDialog

   148

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

   149

from ProjectController import ProjectController, MATIEC_ERROR_MODEL

   150

from util import MiniTextControler

   151

from util.ProcessLogger import ProcessLogger

   152

   154

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

   155

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

   156

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

   157

   158

SCROLLBAR_UNIT = 10

   159

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

   160

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

   161

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

   162

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

   163

   164

if wx.Platform == '__WXMSW__':

   165

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

   166

              'mono' : 'Courier New',

   167

              'helv' : 'Arial',

   168

              'other': 'Comic Sans MS',

   169

              'size' : 16,

   170

             }

   171

else:

   172

    faces = { 'times': 'Times',

   173

              'mono' : 'Courier',

   174

              'helv' : 'Helvetica',

   175

              'other': 'new century schoolbook',

   176

              'size' : 18,

   177

             }

   178

   179

MAX_RECENT_PROJECTS = 10

   180

   181

# Some helpers to tweak GenBitmapTextButtons

   182

# TODO: declare customized classes instead.

   183

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

   184

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

   185

   186

def make_genbitmaptogglebutton_flat(button):

   187

    button.GetBackgroundBrush = gen_mini_GetBackgroundBrush(button)

   188

    button.labelDelta = 0

   189

    button.SetBezelWidth(0)

   190

    button.SetUseFocusIndicator(False)

   191

   192

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

   193

import wx.lib.imageutils

   194

from wx.lib.imageutils import grayOut as old_grayOut

   195

def grayOut(anImage):

   196

    if anImage.HasAlpha():

   197

        AlphaData = anImage.GetAlphaData()

   198

    else :

   199

        AlphaData = None

   200

   201

    old_grayOut(anImage)

   202

   203

    if AlphaData is not None:

   204

        anImage.SetAlphaData(AlphaData)

   205

   206

wx.lib.imageutils.grayOut = grayOut

   207

   208

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

   209

    def _GetLabelSize(self):

   210

        """ used internally """

   211

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

   212

        if not self.bmpLabel:

   213

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

   214

   215

        w_bmp = self.bmpLabel.GetWidth()+2

   216

        h_bmp = self.bmpLabel.GetHeight()+2

   217

        height = h + h_bmp

   218

        if w_bmp > w:

   219

            width = w_bmp

   220

        else:

   221

            width = w

   222

        return width, height, False

   223

   224

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

   225

        bmp = self.bmpLabel

   226

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

   227

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

   228

                bmp = self.bmpDisabled

   229

            if self.bmpFocus and self.hasFocus:

   230

                bmp = self.bmpFocus

   231

            if self.bmpSelected and not self.up:

   232

                bmp = self.bmpSelected

   233

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

   234

            if not self.up:

   235

                dw = dy = self.labelDelta

   236

            hasMask = bmp.GetMask() != None

   237

        else:

   238

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

   239

   240

        dc.SetFont(self.GetFont())

   241

        if self.IsEnabled():

   242

            dc.SetTextForeground(self.GetForegroundColour())

   243

        else:

   244

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

   245

   246

        label = self.GetLabel()

   247

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

   248

        if not self.up:

   249

            dw = dy = self.labelDelta

   250

   251

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

   252

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

   253

        if bmp !=None:

   254

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

   255

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

   256

            pos_y += bh + 2

   257

   258

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

   259

   260

   261

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

   262

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

   263

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

   264

    def __init__(self, parent, ID, bitmapname,

   265

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

   266

                 style = 0,

   267

                 name = "genstatbmp"):

   268

   269

        bitmappath = Bpath( "images", bitmapname)

   270

        if os.path.isfile(bitmappath):

   271

            bitmap = wx.Bitmap(bitmappath)

   272

        else:

   273

            bitmap = None

   274

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

   275

                 pos, size,

   276

                 style,

   277

                 name)

   278

   279

    def OnPaint(self, event):

   280

        dc = wx.PaintDC(self)

   281

        colour = self.GetParent().GetBackgroundColour()

   282

        dc.SetPen(wx.Pen(colour))

   283

        dc.SetBrush(wx.Brush(colour ))

   284

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

   285

        if self._bitmap:

   286

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

   287

   288

   289

from threading import Lock,Timer,currentThread

   290

MainThread = currentThread().ident

   291

REFRESH_PERIOD = 0.1

   292

from time import time as gettime

   293

class LogPseudoFile:

   294

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

   295

    def __init__(self, output, risecall):

   296

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

   297

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

   298

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

   299

        self.default_style = None

   300

        self.output = output

   301

        self.risecall = risecall

   302

        # to prevent rapid fire on rising log panel

   303

        self.rising_timer = 0

   304

        self.lock = Lock()

   305

        self.YieldLock = Lock()

   306

        self.RefreshLock = Lock()

   307

        self.stack = []

   308

        self.LastRefreshTime = gettime()

   309

        self.LastRefreshTimer = None

   310

   311

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

   312

        if self.lock.acquire():

   313

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

   314

            self.lock.release()

   315

            current_time = gettime()

   316

            if self.LastRefreshTimer:

   317

                self.LastRefreshTimer.cancel()

   318

                self.LastRefreshTimer=None

   319

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

   320

                self._should_write()

   321

            else:

   322

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

   323

                self.LastRefreshTimer.start()

   324

   325

    def _should_write(self):

   326

        wx.CallAfter(self._write)

   327

        if MainThread == currentThread().ident:

   328

            app = wx.GetApp()

   329

            if app is not None:

   330

                if self.YieldLock.acquire(0):

   331

                    app.Yield()

   332

                    self.YieldLock.release()

   333

   334

    def _write(self):

   335

        if self.output :

   336

            self.output.Freeze(); 

   337

            self.lock.acquire()

   338

            for s, style in self.stack:

   339

                if style is None : style=self.black_white

   340

                if self.default_style != style: 

   341

                    self.output.SetDefaultStyle(style)

   342

                    self.default_style = style

   343

                self.output.AppendText(s)

   344

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

   345

            self.stack = []

   346

            self.lock.release()

   347

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

   348

            self.output.Thaw()

   349

            self.LastRefreshTime = gettime()

   350

            try:

   351

                self.RefreshLock.release()

   352

            except:

   353

                pass

   354

            newtime = time.time()

   355

            if newtime - self.rising_timer > 1:

   356

                self.risecall()

   357

            self.rising_timer = newtime

   358

   359

    def write_warning(self, s):

   360

        self.write(s,self.red_white)

   361

   362

    def write_error(self, s):

   363

        self.write(s,self.red_yellow)

   364

   365

    def writeyield(self, s):

   366

        self.write(s)

   367

        wx.GetApp().Yield()

   368

   369

    def flush(self):

   370

        self.output.SetValue("")

   371

   372

    def isatty(self):

   373

        return false

   374

   375

[ID_BEREMIZ, ID_BEREMIZMAINSPLITTER, 

   376

 ID_BEREMIZPLCCONFIG, ID_BEREMIZLOGCONSOLE, 

   377

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

   378

   379

[ID_FILEMENURECENTPROJECTS,

   380

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

   381

   382

CONFNODEMENU_POSITION = 3

   383

   384

class Beremiz(IDEFrame):

   385

   386

    def _init_coll_MenuBar_Menus(self, parent):

   387

        IDEFrame._init_coll_MenuBar_Menus(self, parent)

   388

   389

        parent.Insert(pos=CONFNODEMENU_POSITION, 

   390

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

   391

   392

    def _init_utils(self):

   393

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

   394

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

   395

   396

        IDEFrame._init_utils(self)

   397

   398

    def _init_coll_FileMenu_Items(self, parent):

   399

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

   400

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

   401

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

   402

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

   403

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

   404

        parent.AppendSeparator()

   405

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

   406

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

   407

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

   408

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

   409

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

   410

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

   411

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

   412

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

   413

        parent.AppendSeparator()

   414

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

   415

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

   416

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

   417

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

   418

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

   419

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

   420

        parent.AppendSeparator()

   421

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

   422

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

   423

        parent.AppendSeparator()

   424

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

   425

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

   426

   427

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

   428

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

   429

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

   430

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

   431

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

   432

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

   433

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

   434

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

   435

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

   436

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

   437

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

   438

   439

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

   440

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

   441

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

   442

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

   443

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

   444

   445

    def _init_coll_HelpMenu_Items(self, parent):

   446

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

   447

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

   448

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

   449

   450

    def _init_coll_PLCConfigMainSizer_Items(self, parent):

   451

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

   452

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

   453

   454

    def _init_coll_PLCConfigMainSizer_Growables(self, parent):

   455

        parent.AddGrowableCol(0)

   456

        parent.AddGrowableRow(1)

   457

   458

    def _init_coll_ConfNodeTreeSizer_Growables(self, parent):

   459

        parent.AddGrowableCol(0)

   460

        parent.AddGrowableCol(1)

   461

   462

    def _init_beremiz_sizers(self):

   463

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

   464

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

   465

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

   466

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

   467

   468

        self._init_coll_PLCConfigMainSizer_Items(self.PLCConfigMainSizer)

   469

        self._init_coll_PLCConfigMainSizer_Growables(self.PLCConfigMainSizer)

   470

        self._init_coll_ConfNodeTreeSizer_Growables(self.ConfNodeTreeSizer)

   471

   472

        self.PLCConfig.SetSizer(self.PLCConfigMainSizer)

   473

   474

    def _init_ctrls(self, prnt):

   475

        IDEFrame._init_ctrls(self, prnt)

   476

   477

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

   478

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

   479

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

   480

                                ("Run",      wx.WXK_F5),

   481

                                ("Transfer", wx.WXK_F6),

   482

                                ("Connect",  wx.WXK_F7),

   483

                                ("Build",    wx.WXK_F11)]:

   484

            def OnMethodGen(obj,meth):

   485

                def OnMethod(evt):

   486

                    if obj.CTR is not None:

   487

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

   488

                    wx.CallAfter(self.RefreshAll)

   489

                return OnMethod

   490

            newid = wx.NewId()