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 program is free software; you can redistribute it and/or

    12

# modify it under the terms of the GNU General Public License

    13

# as published by the Free Software Foundation; either version 2

    14

# of the License, or (at your option) any later version.

    15

#

    16

# This program 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

    19

# GNU General Public License for more details.

    20

#

    21

# You should have received a copy of the GNU General Public License

    22

# along with this program; if not, write to the Free Software

    23

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

    24

    25

import re

    26

import math

    27

from time import time as gettime

    28

from types import TupleType

    29

from threading import Lock

    30

    31

import wx

    32

    33

from plcopen.structures import *

    34

from PLCControler import ITEM_VAR_LOCAL, ITEM_POU, ITEM_PROGRAM, ITEM_FUNCTIONBLOCK

    35

from graphics.DebugDataConsumer import DebugDataConsumer

    36

    37

from dialogs import *

    38

from graphics import *

    39

from editors.DebugViewer import DebugViewer, REFRESH_PERIOD

    40

from EditorPanel import EditorPanel

    41

    42

SCROLLBAR_UNIT = 10

    43

WINDOW_BORDER = 10

    44

SCROLL_ZONE = 10

    45

    46

CURSORS = None

    47

SFC_Objects = (SFC_Step, SFC_ActionBlock, SFC_Transition, SFC_Divergence, SFC_Jump)

    48

    49

def ResetCursors():

    50

    global CURSORS

    51

    if CURSORS == None:

    52

        CURSORS = [wx.NullCursor,

    53

                   wx.StockCursor(wx.CURSOR_HAND),

    54

                   wx.StockCursor(wx.CURSOR_SIZENWSE),

    55

                   wx.StockCursor(wx.CURSOR_SIZENESW),

    56

                   wx.StockCursor(wx.CURSOR_SIZEWE),

    57

                   wx.StockCursor(wx.CURSOR_SIZENS)]

    58

    59

def AppendMenu(parent, help, id, kind, text):

    60

    if wx.VERSION >= (2, 6, 0):

    61

        parent.Append(help=help, id=id, kind=kind, text=text)

    62

    else:

    63

        parent.Append(helpString=help, id=id, kind=kind, item=text)

    64

    65

if wx.Platform == '__WXMSW__':

    66

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

    67

              'mono' : 'Courier New',

    68

              'helv' : 'Arial',

    69

              'other': 'Comic Sans MS',

    70

              'size' : 10,

    71

             }

    72

else:

    73

    faces = { 'times': 'Times',

    74

              'mono' : 'Courier',

    75

              'helv' : 'Helvetica',

    76

              'other': 'new century schoolbook',

    77

              'size' : 12,

    78

             }

    79

    80

if wx.Platform == '__WXMSW__':

    81

    MAX_ZOOMIN = 4

    82

else:

    83

    MAX_ZOOMIN = 7

    84

ZOOM_FACTORS = [math.sqrt(2) ** x for x in xrange(-6, MAX_ZOOMIN)]

    85

    86

def GetVariableCreationFunction(variable_type):

    87

    def variableCreationFunction(viewer, id, specific_values):

    88

        return FBD_Variable(viewer, variable_type,

    89

                                    specific_values.name,

    90

                                    specific_values.value_type,

    91

                                    id,

    92

                                    specific_values.execution_order)

    93

    return variableCreationFunction

    94

    95

def GetConnectorCreationFunction(connector_type):

    96

    def connectorCreationFunction(viewer, id, specific_values):

    97

        return FBD_Connector(viewer, connector_type,

    98

                                     specific_values.name, id)

    99

    return connectorCreationFunction

   100

   101

def commentCreationFunction(viewer, id, specific_values):

   102

    return Comment(viewer, specific_values.content, id)

   103

   104

def GetPowerRailCreationFunction(powerrail_type):

   105

    def powerRailCreationFunction(viewer, id, specific_values):

   106

        return LD_PowerRail(viewer, powerrail_type, id,

   107

                                    specific_values.connectors)

   108

    return powerRailCreationFunction

   109

   110

NEGATED_VALUE = lambda x: x if x is not None else False

   111

MODIFIER_VALUE = lambda x: x if x is not None else 'none'

   112

   113

CONTACT_TYPES = {(True, "none"): CONTACT_REVERSE,

   114

                 (False, "rising"): CONTACT_RISING,

   115

                 (False, "falling"): CONTACT_FALLING}

   116

   117

def contactCreationFunction(viewer, id, specific_values):

   118

    contact_type = CONTACT_TYPES.get((NEGATED_VALUE(specific_values.negated),

   119

                                      MODIFIER_VALUE(specific_values.edge)),

   120

                                     CONTACT_NORMAL)

   121

    return LD_Contact(viewer, contact_type, specific_values.name, id)

   122

   123

COIL_TYPES = {(True, "none", "none"): COIL_REVERSE,

   124

              (False, "none", "set"): COIL_SET,

   125

              (False, "none", "reset"): COIL_RESET,

   126

              (False, "rising", "none"): COIL_RISING,

   127

              (False, "falling", "none"): COIL_FALLING}

   128

   129

def coilCreationFunction(viewer, id, specific_values):

   130

    coil_type = COIL_TYPES.get((NEGATED_VALUE(specific_values.negated),

   131

                                MODIFIER_VALUE(specific_values.edge),

   132

                                MODIFIER_VALUE(specific_values.storage)),

   133

                               COIL_NORMAL)

   134

    return LD_Coil(viewer, coil_type, specific_values.name, id)

   135

   136

def stepCreationFunction(viewer, id, specific_values):

   137

    step = SFC_Step(viewer, specific_values.name,

   138

                            specific_values.initial, id)

   139

    if specific_values.action is not None:

   140

        step.AddAction()

   141

        connector = step.GetActionConnector()

   142

        connector.SetPosition(wx.Point(*specific_values.action.position))

   143

    return step

   144

   145

def transitionCreationFunction(viewer, id, specific_values):

   146

    transition = SFC_Transition(viewer, specific_values.condition_type,

   147

                                        specific_values.condition,

   148

                                        specific_values.priority, id)

   149

    return transition

   150

   151

divergence_types = [SELECTION_DIVERGENCE,

   152

                    SELECTION_CONVERGENCE, SIMULTANEOUS_DIVERGENCE, SIMULTANEOUS_CONVERGENCE]

   153

   154

def GetDivergenceCreationFunction(divergence_type):

   155

    def divergenceCreationFunction(viewer, id, specific_values):

   156

        return SFC_Divergence(viewer, divergence_type,

   157

                                      specific_values.connectors, id)

   158

    return divergenceCreationFunction

   159

   160

def jumpCreationFunction(viewer, id, specific_values):

   161

    return SFC_Jump(viewer, specific_values.target, id)

   162

   163

def actionBlockCreationFunction(viewer, id, specific_values):

   164

    return SFC_ActionBlock(viewer, specific_values.actions, id)

   165

   166

ElementCreationFunctions = {

   167

    "input": GetVariableCreationFunction(INPUT),

   168

    "output": GetVariableCreationFunction(OUTPUT),

   169

    "inout": GetVariableCreationFunction(INOUT),

   170

    "connector": GetConnectorCreationFunction(CONNECTOR),

   171

    "continuation": GetConnectorCreationFunction(CONTINUATION),

   172

    "comment": commentCreationFunction,

   173

    "leftPowerRail": GetPowerRailCreationFunction(LEFTRAIL),

   174

    "rightPowerRail": GetPowerRailCreationFunction(RIGHTRAIL),

   175

    "contact": contactCreationFunction,

   176

    "coil": coilCreationFunction,

   177

    "step": stepCreationFunction,

   178

    "transition": transitionCreationFunction,

   179

    "selectionDivergence": GetDivergenceCreationFunction(SELECTION_DIVERGENCE),

   180

    "selectionConvergence": GetDivergenceCreationFunction(SELECTION_CONVERGENCE),

   181

    "simultaneousDivergence": GetDivergenceCreationFunction(SIMULTANEOUS_DIVERGENCE),

   182

    "simultaneousConvergence": GetDivergenceCreationFunction(SIMULTANEOUS_CONVERGENCE),

   183

    "jump": jumpCreationFunction,

   184

    "actionBlock": actionBlockCreationFunction,

   185

}

   186

   187

def sort_blocks(block_infos1, block_infos2):

   188

    x1, y1 = block_infos1[0].GetPosition()

   189

    x2, y2 = block_infos2[0].GetPosition()

   190

    if y1 == y2:

   191

        return cmp(x1, x2)

   192

    else:

   193

        return cmp(y1, y2)

   194

   195

#-------------------------------------------------------------------------------

   196

#                       Graphic elements Viewer base class

   197

#-------------------------------------------------------------------------------

   198

   199

# ID Constants for alignment menu items

   200

[ID_VIEWERALIGNMENTMENUITEMS0, ID_VIEWERALIGNMENTMENUITEMS1,

   201

 ID_VIEWERALIGNMENTMENUITEMS2, ID_VIEWERALIGNMENTMENUITEMS4,

   202

 ID_VIEWERALIGNMENTMENUITEMS5, ID_VIEWERALIGNMENTMENUITEMS6,

   203

] = [wx.NewId() for _init_coll_AlignmentMenu_Items in range(6)]

   204

   205

# ID Constants for contextual menu items

   206

[ID_VIEWERCONTEXTUALMENUITEMS0, ID_VIEWERCONTEXTUALMENUITEMS1,

   207

 ID_VIEWERCONTEXTUALMENUITEMS2, ID_VIEWERCONTEXTUALMENUITEMS3,

   208

 ID_VIEWERCONTEXTUALMENUITEMS5, ID_VIEWERCONTEXTUALMENUITEMS6,

   209

 ID_VIEWERCONTEXTUALMENUITEMS8, ID_VIEWERCONTEXTUALMENUITEMS9,

   210

 ID_VIEWERCONTEXTUALMENUITEMS11, ID_VIEWERCONTEXTUALMENUITEMS12,

   211

 ID_VIEWERCONTEXTUALMENUITEMS14, ID_VIEWERCONTEXTUALMENUITEMS16,

   212

 ID_VIEWERCONTEXTUALMENUITEMS17,

   213

] = [wx.NewId() for _init_coll_ContextualMenu_Items in range(13)]

   214

   215

   216

class ViewerDropTarget(wx.TextDropTarget):

   217

   218

    def __init__(self, parent):

   219

        wx.TextDropTarget.__init__(self)

   220

        self.ParentWindow = parent

   221

   222

    def OnDropText(self, x, y, data):

   223

        self.ParentWindow.Select()

   224

        tagname = self.ParentWindow.GetTagName()

   225

        pou_name, pou_type = self.ParentWindow.Controler.GetEditedElementType(tagname, self.ParentWindow.Debug)

   226

        x, y = self.ParentWindow.CalcUnscrolledPosition(x, y)

   227

        x = int(x / self.ParentWindow.ViewScale[0])

   228

        y = int(y / self.ParentWindow.ViewScale[1])

   229

        scaling = self.ParentWindow.Scaling

   230

        message = None

   231

        try:

   232

            values = eval(data)

   233

        except:

   234

            message = _("Invalid value \"%s\" for viewer block")%data

   235

            values = None

   236

        if not isinstance(values, TupleType):

   237

            message = _("Invalid value \"%s\" for viewer block")%data

   238

            values = None

   239

        if values is not None:

   240

            if values[1] == "debug":

   241

                pass

   242

            elif values[1] == "program":

   243

                message = _("Programs can't be used by other POUs!")

   244

            elif values[1] in ["function", "functionBlock"]:

   245

                words = tagname.split("::")

   246

                if pou_name == values[0]:

   247

                    message = _("\"%s\" can't use itself!")%pou_name

   248

                elif pou_type == "function" and values[1] != "function":

   249

                    message = _("Function Blocks can't be used in Functions!")

   250

                elif self.ParentWindow.Controler.PouIsUsedBy(pou_name, values[0], self.ParentWindow.Debug):

   251

                    message = _("\"{a1}\" is already used by \"{a2}\"!").format(a1 = pou_name, a2 = values[0])

   252

                else:

   253

                    blockname = values[2]

   254

                    if len(values) > 3:

   255

                        blockinputs = values[3]

   256

                    else:

   257

                        blockinputs = None

   258

                    if values[1] != "function" and blockname == "":

   259

                        blockname = self.ParentWindow.GenerateNewName(blocktype=values[0])

   260

                    if blockname.upper() in [name.upper() for name in self.ParentWindow.Controler.GetProjectPouNames(self.ParentWindow.Debug)]:

   261

                        message = _("\"%s\" pou already exists!")%blockname

   262

                    elif blockname.upper() in [name.upper() for name in self.ParentWindow.Controler.GetEditedElementVariables(tagname, self.ParentWindow.Debug)]:

   263

                        message = _("\"%s\" element for this pou already exists!")%blockname

   264

                    else:

   265

                        id = self.ParentWindow.GetNewId()

   266

                        block = FBD_Block(self.ParentWindow, values[0], blockname, id, inputs = blockinputs)

   267

                        width, height = block.GetMinSize()

   268

                        if scaling is not None:

   269

                            x = round(float(x) / float(scaling[0])) * scaling[0]

   270

                            y = round(float(y) / float(scaling[1])) * scaling[1]

   271

                            width = round(float(width) / float(scaling[0]) + 0.5) * scaling[0]

   272

                            height = round(float(height) / float(scaling[1]) + 0.5) * scaling[1]

   273

                        block.SetPosition(x, y)

   274

                        block.SetSize(width, height)

   275

                        self.ParentWindow.AddBlock(block)

   276

                        self.ParentWindow.Controler.AddEditedElementBlock(tagname, id, values[0], blockname)

   277

                        self.ParentWindow.RefreshBlockModel(block)

   278

                        self.ParentWindow.RefreshBuffer()

   279

                        self.ParentWindow.RefreshScrollBars()

   280

                        self.ParentWindow.RefreshVisibleElements()

   281

                        self.ParentWindow.RefreshVariablePanel()

   282

                        self.ParentWindow.ParentWindow.RefreshPouInstanceVariablesPanel()

   283

                        self.ParentWindow.Refresh(False)

   284

            elif values[1] == "location":

   285

                if pou_type == "program":

   286

                    location = values[0]

   287

                    if not location.startswith("%"):

   288

                        dialog = wx.SingleChoiceDialog(self.ParentWindow.ParentWindow,

   289

                              _("Select a variable class:"), _("Variable class"),

   290

                              [_("Input"), _("Output"), _("Memory")],

   291

                              wx.DEFAULT_DIALOG_STYLE|wx.OK|wx.CANCEL)

   292

                        if dialog.ShowModal() == wx.ID_OK:

   293

                            selected = dialog.GetSelection()

   294

                        else:

   295

                            selected = None

   296

                        dialog.Destroy()

   297

                        if selected is None:

   298

                            return

   299

                        if selected == 0:

   300

                            location = "%I" + location

   301

                        elif selected == 1:

   302

                            location = "%Q" + location

   303

                        else:

   304

                            location = "%M" + location

   305

                    var_name = values[3]

   306

                    dlg = wx.TextEntryDialog(

   307

                        self.ParentWindow.ParentWindow,

   308

                        _("Confirm or change variable name"),

   309

                        _('Variable Drop'), var_name)

   310

                    dlg.SetValue(var_name)

   311

                    var_name = dlg.GetValue() if dlg.ShowModal() == wx.ID_OK else None

   312

                    dlg.Destroy()

   313

                    if var_name is None:

   314

                        return

   315

                    elif var_name.upper() in [name.upper() for name in self.ParentWindow.Controler.GetProjectPouNames(self.ParentWindow.Debug)]:

   316

                        message = _("\"%s\" pou already exists!")%var_name

   317

                    elif not var_name.upper() in [name.upper() for name in self.ParentWindow.Controler.GetEditedElementVariables(tagname, self.ParentWindow.Debug)]:

   318

                        if location[1] == "Q":

   319

                            var_class = OUTPUT

   320

                        else:

   321

                            var_class = INPUT

   322

                        if values[2] is not None:

   323

                            var_type = values[2]

   324

                        else:

   325

                            var_type = LOCATIONDATATYPES.get(location[2], ["BOOL"])[0]

   326

                        self.ParentWindow.Controler.AddEditedElementPouVar(tagname, var_type, var_name, location=location, description=values[4])

   327

                        self.ParentWindow.RefreshVariablePanel()

   328

                        self.ParentWindow.ParentWindow.RefreshPouInstanceVariablesPanel()

   329

                        self.ParentWindow.AddVariableBlock(x, y, scaling, var_class, var_name, var_type)

   330

                    else:

   331

                        message = _("\"%s\" element for this pou already exists!")%var_name

   332

            elif values[1] == "NamedConstant":

   333

                if pou_type == "program":

   334

                    initval = values[0]

   335

                    var_name = values[3]

   336

                    dlg = wx.TextEntryDialog(

   337

                        self.ParentWindow.ParentWindow,

   338

                        _("Confirm or change variable name"),

   339

                        _('Variable Drop'), var_name)

   340

                    dlg.SetValue(var_name)

   341

                    var_name = dlg.GetValue() if dlg.ShowModal() == wx.ID_OK else None

   342

                    dlg.Destroy()

   343

                    if var_name is None:

   344

                        return

   345

                    elif var_name.upper() in [name.upper() for name in self.ParentWindow.Controler.GetProjectPouNames(self.ParentWindow.Debug)]:

   346

                        message = _("\"%s\" pou already exists!")%var_name

   347

                    elif not var_name.upper() in [name.upper() for name in self.ParentWindow.Controler.GetEditedElementVariables(tagname, self.ParentWindow.Debug)]:

   348

                        var_class = INPUT

   349

                        var_type = values[2]

   350

                        self.ParentWindow.Controler.AddEditedElementPouVar(tagname, var_type, var_name, description=values[4], initval=initval)

   351

                        self.ParentWindow.RefreshVariablePanel()

   352

                        self.ParentWindow.ParentWindow.RefreshPouInstanceVariablesPanel()

   353

                        self.ParentWindow.AddVariableBlock(x, y, scaling, var_class, var_name, var_type)

   354

                    else:

   355

                        message = _("\"%s\" element for this pou already exists!")%var_name

   356

            elif values[1] == "Global":

   357

                var_name = values[0]

   358

                dlg = wx.TextEntryDialog(

   359

                    self.ParentWindow.ParentWindow,

   360

                    _("Confirm or change variable name"),

   361

                    _('Variable Drop'), var_name)

   362

                dlg.SetValue(var_name)

   363

                var_name = dlg.GetValue() if dlg.ShowModal() == wx.ID_OK else None

   364

                dlg.Destroy()

   365

                if var_name is None:

   366

                    return

   367

                elif var_name.upper() in [name.upper() for name in self.ParentWindow.Controler.GetProjectPouNames(self.ParentWindow.Debug)]:

   368

                    message = _("\"%s\" pou already exists!")%var_name

   369

                elif not var_name.upper() in [name.upper() for name in self.ParentWindow.Controler.GetEditedElementVariables(tagname, self.ParentWindow.Debug)]:

   370

                    self.ParentWindow.Controler.AddEditedElementPouExternalVar(tagname, values[2], var_name)

   371

                    self.ParentWindow.RefreshVariablePanel()

   372

                    self.ParentWindow.ParentWindow.RefreshPouInstanceVariablesPanel()

   373

                    self.ParentWindow.AddVariableBlock(x, y, scaling, INPUT, var_name, values[2])

   374

                else:

   375

                    message = _("\"%s\" element for this pou already exists!")%var_name

   376

            elif values[1] == "Constant":

   377

                self.ParentWindow.AddVariableBlock(x, y, scaling, INPUT, values[0], None)

   378

            elif values[3] == tagname:

   379

                if values[1] == "Output":

   380

                    var_class = OUTPUT

   381

                elif values[1] == "InOut":

   382

                    var_class = INPUT

   383

                else:

   384

                    var_class = INPUT

   385

                tree = dict([(var.Name, var.Tree) for var in self.ParentWindow.Controler.GetEditedElementInterfaceVars(tagname, True, self.ParentWindow.Debug)]).get(values[0], None)

   386

                if tree is not None:

   387

                    if len(tree[0]) > 0:

   388

                        menu = wx.Menu(title='')

   389

                        self.GenerateTreeMenu(x, y, scaling, menu, "", var_class, [(values[0], values[2], tree)])

   390

                        self.ParentWindow.PopupMenuXY(menu)

   391

                    else:

   392

                        self.ParentWindow.AddVariableBlock(x, y, scaling, var_class, values[0], values[2])

   393

                else:

   394

                    message = _("Unknown variable \"%s\" for this POU!") % values[0]

   395

            else:

   396

                message = _("Variable don't belong to this POU!")

   397

        if message is not None:

   398

            wx.CallAfter(self.ShowMessage, message)

   399

   400

    def GenerateTreeMenu(self, x, y, scaling, menu, base_path, var_class, tree):

   401

        for child_name, child_type, (child_tree, child_dimensions) in tree:

   402

            if base_path:

   403

                child_path = "%s.%s" % (base_path, child_name)

   404

            else:

   405

                child_path = child_name

   406

            if len(child_dimensions) > 0:

   407

                child_path += "[%s]" % ",".join([str(dimension[0]) for dimension in child_dimensions])

   408

                child_name += "[]"

   409

            new_id = wx.NewId()

   410

            AppendMenu(menu, help='', id=new_id, kind=wx.ITEM_NORMAL, text=child_name)

   411

            self.ParentWindow.Bind(wx.EVT_MENU, self.GetAddVariableBlockFunction(x, y, scaling, var_class, child_path, child_type), id=new_id)

   412

            if len(child_tree) > 0:

   413

                new_id = wx.NewId()

   414

                child_menu = wx.Menu(title='')

   415

                self.GenerateTreeMenu(x, y, scaling, child_menu, child_path, var_class, child_tree)

   416

                menu.AppendMenu(new_id, "%s." % child_name, child_menu)

   417

   418

    def GetAddVariableBlockFunction(self, x, y, scaling, var_class, var_name, var_type):

   419

        def AddVariableFunction(event):

   420

            self.ParentWindow.AddVariableBlock(x, y, scaling, var_class, var_name, var_type)

   421

        return AddVariableFunction

   422

   423

    def ShowMessage(self, message):

   424

        message = wx.MessageDialog(self.ParentWindow, message, _("Error"), wx.OK|wx.ICON_ERROR)

   425

        message.ShowModal()

   426

        message.Destroy()

   427

   428

   429

   430

class DebugInstanceName(DebugDataConsumer):

   431

    VALUE_TRANSLATION = {True: _("Active"), False: _("Inactive")}

   432

   433

    def __init__(self, parent):

   434

        DebugDataConsumer.__init__(self)

   435

        self.Parent = parent

   436

        self.ActionLastState = None

   437

        self.ActionState = None

   438

        self.x_offset = 2

   439

        self.y_offset = 2

   440

   441

    def SetValue(self, value):

   442

        self.ActionState = value

   443

        if self.ActionState != self.ActionLastState:

   444

            self.ActionLastState = self.ActionState

   445

            wx.CallAfter(self.Parent.ElementNeedRefresh, self)

   446

   447

    def GetInstanceName(self):

   448

        return _("Debug: %s") % self.Parent.InstancePath

   449

   450

    def GetRedrawRect(self):

   451

        x, y = self.Parent.CalcUnscrolledPosition(self.x_offset, self.y_offset)        

   452

        dc = self.Parent.GetLogicalDC()

   453

        ipw, iph = dc.GetTextExtent(self.GetInstanceName())

   454

        vw, vh = 0, 0

   455

        for value in self.VALUE_TRANSLATION.itervalues():

   456

            w, h = dc.GetTextExtent(" (%s)" % value)

   457

            vw = max(vw, w)

   458

            vh = max(vh, h)

   459

        return wx.Rect(ipw + x, y, vw, vh)

   460

   461

    def Draw(self, dc):

   462

        scalex, scaley = dc.GetUserScale()

   463

        dc.SetUserScale(1, 1)

   464

        x, y = self.Parent.CalcUnscrolledPosition(self.x_offset, self.y_offset)

   465

   466

        text = self.GetInstanceName()

   467

        if self.ActionState is not None:

   468

            text += " ("

   469

   470

        dc.DrawText(text, x, y)

   471

        tw, th = dc.GetTextExtent(text)                    

   472

        if self.ActionState is not None:

   473

   474

            text = self.VALUE_TRANSLATION[self.ActionState]

   475

            if self.ActionState:

   476

                dc.SetTextForeground(wx.GREEN)

   477

            dc.DrawText(text, x + tw, y)

   478

            if self.ActionState:

   479

                dc.SetTextForeground(wx.BLACK)

   480

            tw = tw + dc.GetTextExtent(text)[0]

   481

   482

            text = ")"                    

   483

            dc.DrawText(text, x + tw, y)

   484

        dc.SetUserScale(scalex, scaley)

   485

   486

"""

   487

Class that implements a Viewer based on a wx.ScrolledWindow for drawing and

   488

manipulating graphic elements

   489

"""

   490

   491

class Viewer(EditorPanel, DebugViewer):

   492

   493

    if wx.VERSION < (2, 6, 0):

   494

        def Bind(self, event, function, id = None):

   495

            if id is not None:

   496

                event(self, id, function)

   497

            else:

   498

                event(self, function)

   499

   500

    # Add list of menu items to the given menu

   501

    def AddMenuItems(self, menu, items):

   502

        for item in items:

   503

            if item is None:

   504

                menu.AppendSeparator()

   505

            else:

   506

                id, kind, text, help, callback = item

   507

                AppendMenu(menu, help=help, id=id, kind=kind, text=text)

   508

                # Link menu event to corresponding called functions