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

    30

from wx.lib.agw.advancedsplash import AdvancedSplash

    31

import tempfile

    32

import shutil

    33

import random

    34

import time

    35

    36

CWD = os.path.split(os.path.realpath(__file__))[0]

    37

    38

def Bpath(*args):

    39

    return os.path.join(CWD,*args)

    40

    41

if __name__ == '__main__':

    42

    def usage():

    43

        print "\nUsage of Beremiz.py :"

    44

        print "\n   %s [Projectpath] [Buildpath]\n"%sys.argv[0]

    45

    46

    try:

    47

        opts, args = getopt.getopt(sys.argv[1:], "hu:", ["help", "updatecheck="])

    48

    except getopt.GetoptError:

    49

        # print help information and exit:

    50

        usage()

    51

        sys.exit(2)

    52

    53

    for o, a in opts:

    54

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

    55

            usage()

    56

            sys.exit()

    57

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

    58

            updateinfo_url = a

    59

    60

    if len(args) > 2:

    61

        usage()

    62

        sys.exit()

    63

    elif len(args) == 1:

    64

        projectOpen = args[0]

    65

        buildpath = None

    66

    elif len(args) == 2:

    67

        projectOpen = args[0]

    68

        buildpath = args[1]

    69

    else:

    70

        projectOpen = None

    71

        buildpath = None

    72

    73

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

    75

    else :

    77

    78

    app = wx.PySimpleApp(redirect=BMZ_DBG)

    79

    app.SetAppName('beremiz')

    80

    wx.InitAllImageHandlers()

    81

    82

    # popup splash

    83

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

    84

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

    85

    splash=AdvancedSplash(None, bitmap=bmp)

    86

    wx.Yield()

    87

    88

    if updateinfo_url is not None:

    89

        updateinfo = "Fetching %s" % updateinfo_url

    90

        # warn for possible updates

    91

        def updateinfoproc():

    92

            global updateinfo

    93

            try :

    94

                import urllib2

    95

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

    96

            except :

    97

                updateinfo = "update info unavailable." 

    98

    99

        from threading import Thread

   100

        splash.SetText(text=updateinfo)

   101

        wx.Yield()

   102

        updateinfoThread = Thread(target=updateinfoproc)

   103

        updateinfoThread.start()

   104

        updateinfoThread.join(2)

   105

        splash.SetText(text=updateinfo)

   106

        wx.Yield()

   107

   108

# Import module for internationalization

   109

import gettext

   110

   111

# Get folder containing translation files

   112

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

   113

# Get the default language

   114

langid = wx.LANGUAGE_DEFAULT

   115

# Define translation domain (name of translation files)

   116

domain = "Beremiz"

   117

   118

# Define locale for wx

   119

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

   120

if loc is None:

   121

    test_loc = wx.Locale(langid)

   122

    test_loc.AddCatalogLookupPathPrefix(localedir)

   123

    if test_loc.AddCatalog(domain):

   124

        loc = wx.Locale(langid)

   125

    else:

   126

        loc = wx.Locale(wx.LANGUAGE_ENGLISH)

   127

    __builtin__.__dict__['loc'] = loc

   128

# Define location for searching translation files

   129

loc.AddCatalogLookupPathPrefix(localedir)

   130

# Define locale domain

   131

loc.AddCatalog(domain)

   132

   133

def unicode_translation(message):

   134

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

   135

   136

if __name__ == '__main__':

   137

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

   138

   139

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

   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

import TextCtrlAutoComplete, cPickle

   146

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

   147

from plugger import PluginsRoot, MATIEC_ERROR_MODEL

   148

from wxPopen import ProcessLogger

   149

   150

from docutils import *

   152

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

   153

   154

SCROLLBAR_UNIT = 10

   155

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

   156

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

   157

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

   158

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

   159

   160

if wx.Platform == '__WXMSW__':

   161

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

   162

              'mono' : 'Courier New',

   163

              'helv' : 'Arial',

   164

              'other': 'Comic Sans MS',

   165

              'size' : 16,

   166

             }

   167

else:

   168

    faces = { 'times': 'Times',

   169

              'mono' : 'Courier',

   170

              'helv' : 'Helvetica',

   171

              'other': 'new century schoolbook',

   172

              'size' : 18,

   173

             }

   174

   175

# Some helpers to tweak GenBitmapTextButtons

   176

# TODO: declare customized classes instead.

   177

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

   178

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

   179

   180

def make_genbitmaptogglebutton_flat(button):

   181

    button.GetBackgroundBrush = gen_mini_GetBackgroundBrush(button)

   182

    button.labelDelta = 0

   183

    button.SetBezelWidth(0)

   184

    button.SetUseFocusIndicator(False)

   185

   186

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

   187

import wx.lib.imageutils

   188

from wx.lib.imageutils import grayOut as old_grayOut

   189

def grayOut(anImage):

   190

    if anImage.HasAlpha():

   191

        AlphaData = anImage.GetAlphaData()

   192

    else :

   193

        AlphaData = None

   194

   195

    old_grayOut(anImage)

   196

   197

    if AlphaData is not None:

   198

        anImage.SetAlphaData(AlphaData)

   199

   200

wx.lib.imageutils.grayOut = grayOut

   201

   202

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

   203

    def _GetLabelSize(self):

   204

        """ used internally """

   205

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

   206

        if not self.bmpLabel:

   207

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

   208

   209

        w_bmp = self.bmpLabel.GetWidth()+2

   210

        h_bmp = self.bmpLabel.GetHeight()+2

   211

        height = h + h_bmp

   212

        if w_bmp > w:

   213

            width = w_bmp

   214

        else:

   215

            width = w

   216

        return width, height, False

   217

   218

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

   219

        bmp = self.bmpLabel

   220

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

   221

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

   222

                bmp = self.bmpDisabled

   223

            if self.bmpFocus and self.hasFocus:

   224

                bmp = self.bmpFocus

   225

            if self.bmpSelected and not self.up:

   226

                bmp = self.bmpSelected

   227

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

   228

            if not self.up:

   229

                dw = dy = self.labelDelta

   230

            hasMask = bmp.GetMask() != None

   231

        else:

   232

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

   233

   234

        dc.SetFont(self.GetFont())

   235

        if self.IsEnabled():

   236

            dc.SetTextForeground(self.GetForegroundColour())

   237

        else:

   238

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

   239

   240

        label = self.GetLabel()

   241

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

   242

        if not self.up:

   243

            dw = dy = self.labelDelta

   244

   245

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

   246

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

   247

        if bmp !=None:

   248

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

   249

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

   250

            pos_y += bh + 2

   251

   252

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

   253

   254

   255

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

   256

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

   257

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

   258

    def __init__(self, parent, ID, bitmapname,

   259

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

   260

                 style = 0,

   261

                 name = "genstatbmp"):

   262

   263

        bitmappath = Bpath( "images", bitmapname)

   264

        if os.path.isfile(bitmappath):

   265

            bitmap = wx.Bitmap(bitmappath)

   266

        else:

   267

            bitmap = None

   268

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

   269

                 pos, size,

   270

                 style,

   271

                 name)

   272

   273

    def OnPaint(self, event):

   274

        dc = wx.PaintDC(self)

   275

        colour = self.GetParent().GetBackgroundColour()

   276

        dc.SetPen(wx.Pen(colour))

   277

        dc.SetBrush(wx.Brush(colour ))

   278

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

   279

        if self._bitmap:

   280

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

   281

   282

   283

class LogPseudoFile:

   284

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

   285

    def __init__(self, output, risecall):

   286

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

   287

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

   288

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

   289

        self.default_style = None

   290

        self.output = output

   291

        self.risecall = risecall

   292

        # to prevent rapid fire on rising log panel

   293

        self.rising_timer = 0

   294

   295

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

   296

        if style is None : style=self.black_white

   297

        self.output.Freeze(); 

   298

        if self.default_style != style: 

   299

            self.output.SetDefaultStyle(style)

   300

            self.default_style = style

   301

        self.output.AppendText(s)

   302

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

   303

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

   304

        self.output.Thaw()

   305

        newtime = time.time()

   306

        if newtime - self.rising_timer > 1:

   307

            self.risecall()

   308

        self.rising_timer = newtime 

   309

   310

    def write_warning(self, s):

   311

        self.write(s,self.red_white)

   312

   313

    def write_error(self, s):

   314

        self.write(s,self.red_yellow)

   315

   316

    def writeyield(self, s):

   317

        self.write(s)

   318

        wx.GetApp().Yield()

   319

   320

    def flush(self):

   321

        self.output.SetValue("")

   322

   323

    def isatty(self):

   324

        return false

   325

   326

[ID_BEREMIZ, ID_BEREMIZMAINSPLITTER, 

   327

 ID_BEREMIZPLCCONFIG, ID_BEREMIZLOGCONSOLE, 

   328

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

   329

   330

class Beremiz(IDEFrame):

   331

   332

    def _init_coll_FileMenu_Items(self, parent):

   333

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

   334

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

   335

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

   336

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

   337

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

   338

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

   339

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

   340

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

   341

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

   342

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

   343

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

   344

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

   345

        parent.AppendSeparator()

   346

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

   347

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

   348

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

   349

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

   350

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

   351

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

   352

        parent.AppendSeparator()

   353

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

   354

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

   355

        parent.AppendSeparator()

   356

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

   357

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

   358

   359

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

   360

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

   361

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

   362

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

   363

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

   364

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

   365

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

   366

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

   367

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

   368

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

   369

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

   370

   371

    def _init_coll_HelpMenu_Items(self, parent):

   372

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

   373

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

   374

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

   375

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

   376

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

   377

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

   378

   379

    def _init_coll_PLCConfigMainSizer_Items(self, parent):

   380

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

   381

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

   382

   383

    def _init_coll_PLCConfigMainSizer_Growables(self, parent):

   384

        parent.AddGrowableCol(0)

   385

        parent.AddGrowableRow(1)

   386

   387

    def _init_coll_PluginTreeSizer_Growables(self, parent):

   388

        parent.AddGrowableCol(0)

   389

        parent.AddGrowableCol(1)

   390

   391

    def _init_beremiz_sizers(self):

   392

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

   393

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

   394

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

   395

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

   396

   397

        self._init_coll_PLCConfigMainSizer_Items(self.PLCConfigMainSizer)

   398

        self._init_coll_PLCConfigMainSizer_Growables(self.PLCConfigMainSizer)

   399

        self._init_coll_PluginTreeSizer_Growables(self.PluginTreeSizer)

   400

   401

        self.PLCConfig.SetSizer(self.PLCConfigMainSizer)

   402

   403

    def _init_ctrls(self, prnt):

   404

        IDEFrame._init_ctrls(self, prnt)

   405

   406

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

   407

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

   408

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

   409

                                ("Run",      wx.WXK_F5),

   410

                                ("Transfer", wx.WXK_F6),

   411

                                ("Connect",  wx.WXK_F7),

   412

                                ("Build",    wx.WXK_F11)]:

   413

            def OnMethodGen(obj,meth):

   414

                def OnMethod(evt):

   415

                    if obj.PluginRoot is not None:

   416

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

   417

                    wx.CallAfter(self.RefreshAll)

   418

                return OnMethod

   419

            newid = wx.NewId()

   420

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

   421

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

   422

   423

        self.SetAcceleratorTable(wx.AcceleratorTable(accels))

   424

   425

        self.PLCConfig = wx.ScrolledWindow(id=ID_BEREMIZPLCCONFIG,

   427

              size=wx.Size(-1, -1), style=wx.TAB_TRAVERSAL|wx.SUNKEN_BORDER|wx.HSCROLL|wx.VSCROLL)

   428

        self.PLCConfig.SetBackgroundColour(wx.WHITE)

   429

        self.PLCConfig.Bind(wx.EVT_LEFT_DOWN, self.OnPanelLeftDown)

   430

        self.PLCConfig.Bind(wx.EVT_SIZE, self.OnMoveWindow)

   431

        self.PLCConfig.Bind(wx.EVT_MOUSEWHEEL, self.OnPLCConfigScroll)

   432

        self.BottomNoteBook.InsertPage(0, self.PLCConfig, _("Topology"), True)

   433

   434

        self.LogConsole = wx.TextCtrl(id=ID_BEREMIZLOGCONSOLE, value='',

   435

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

   436

                  size=wx.Size(0, 0), style=wx.TE_MULTILINE|wx.TE_RICH2)

   437

        self.LogConsole.Bind(wx.EVT_LEFT_DCLICK, self.OnLogConsoleDClick)

   438

        self.BottomNoteBook.AddPage(self.LogConsole, _("Log Console"))

   441

   442

        self._init_beremiz_sizers()

   443

   444

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

   445

        IDEFrame.__init__(self, parent, debug)

   446

        self.Log = LogPseudoFile(self.LogConsole,self.RiseLogConsole)

   447

   448

        self.local_runtime = None

   449

        self.runtime_port = None

   450

        self.local_runtime_tmpdir = None

   451

   452

        self.DisableEvents = False

   453

        # Variable allowing disabling of PLCConfig scroll when Popup shown 

   454

        self.ScrollingEnabled = True

   455

   456

        self.PluginInfos = {}

   457

   475

        if projectOpen is not None and os.path.isdir(projectOpen):

   476

            self.PluginRoot = PluginsRoot(self, self.Log)

   477

            self.Controler = self.PluginRoot

   478

            result = self.PluginRoot.LoadProject(projectOpen, buildpath)

   479

            if not result:

   480

                self._Refresh(TYPESTREE, INSTANCESTREE, LIBRARYTREE)

   481

                self.RefreshAll()

   482

            else:

   483

                self.ResetView()

   484

                self.ShowErrorMessage(result)

   485

        else:

   486

            self.PluginRoot = plugin_root

   487

            self.Controler = plugin_root

   488

            if plugin_root is not None:

   489

                self._Refresh(TYPESTREE, INSTANCESTREE, LIBRARYTREE)

   490

                self.RefreshAll()

   491

        if self.EnableDebug:

   492

            self.DebugVariablePanel.SetDataProducer(self.PluginRoot)

   493