1

#!/usr/bin/env python

     2

# -*- coding: utf-8 -*-

     3

     4

#This file is part of PLCOpenEditor, a library implementing an IEC 61131-3 editor

     5

#based on the plcopen standard. 

     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

import math

    26

    27

import wx

    28

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

    29

    USE_AUI = True

    30

else:

    31

    USE_AUI = False

    32

    33

from plcopen.structures import *

    34

from PLCControler import ITEM_POU

    35

    36

from Dialogs import *

    37

    38

from types import TupleType

    39

    40

SCROLLBAR_UNIT = 10

    41

WINDOW_BORDER = 10

    42

SCROLL_ZONE = 10

    43

    44

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

    45

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

    46

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

    47

    else:

    48

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

    49

    50

if wx.Platform == '__WXMSW__':

    51

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

    52

              'mono' : 'Courier New',

    53

              'helv' : 'Arial',

    54

              'other': 'Comic Sans MS',

    55

              'size' : 20,

    56

             }

    57

else:

    58

    faces = { 'times': 'Times',

    59

              'mono' : 'Courier',

    60

              'helv' : 'Helvetica',

    61

              'other': 'new century schoolbook',

    62

              'size' : 20,

    63

             }

    64

    65

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

    66

    67

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

    68

#                       Graphic elements Viewer base class

    69

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

    70

    71

# ID Constants for alignment menu items

    72

[ID_VIEWERALIGNMENTMENUITEMS0, ID_VIEWERALIGNMENTMENUITEMS1,

    73

 ID_VIEWERALIGNMENTMENUITEMS2, ID_VIEWERALIGNMENTMENUITEMS4,

    74

 ID_VIEWERALIGNMENTMENUITEMS5, ID_VIEWERALIGNMENTMENUITEMS6,

    75

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

    76

    77

# ID Constants for contextual menu items

    78

[ID_VIEWERCONTEXTUALMENUITEMS0, ID_VIEWERCONTEXTUALMENUITEMS1,

    79

 ID_VIEWERCONTEXTUALMENUITEMS2, ID_VIEWERCONTEXTUALMENUITEMS3,

    80

 ID_VIEWERCONTEXTUALMENUITEMS5, ID_VIEWERCONTEXTUALMENUITEMS6,

    81

 ID_VIEWERCONTEXTUALMENUITEMS8, ID_VIEWERCONTEXTUALMENUITEMS9,

    82

 ID_VIEWERCONTEXTUALMENUITEMS11, ID_VIEWERCONTEXTUALMENUITEMS12,

    83

 ID_VIEWERCONTEXTUALMENUITEMS14, ID_VIEWERCONTEXTUALMENUITEMS16,

    84

 ID_VIEWERCONTEXTUALMENUITEMS17,

    85

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

    86

    87

    88

class ViewerDropTarget(wx.TextDropTarget):

    89

    90

    def __init__(self, parent):

    91

        wx.TextDropTarget.__init__(self)

    92

        self.ParentWindow = parent

    93

    94

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

    95

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

    96

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

    97

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

    98

        scaling = self.ParentWindow.Scaling

    99

        message = None

   100

        try:

   101

            values = eval(data)

   102

        except:

   103

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

   104

            values = None

   105

        if not isinstance(values, TupleType):

   106

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

   107

            values = None

   108

        if values is not None:

   109

            if values[1] == "debug":

   110

                pass

   111

            elif values[1] == "program":

   112

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

   113

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

   114

                name, type = self.ParentWindow.Controler.GetEditedElementType(self.ParentWindow.GetTagName(), self.ParentWindow.Debug)

   115

                words = self.ParentWindow.TagName.split("::")

   116

                if name == values[0]:

   117

                    message = "\"%s\" can't use itself!"%name

   118

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

   119

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

   120

                elif words[0] == "T" and values[1] != "function":

   121

                    message = "Function Blocks can't be used in Transitions!"

   122

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

   123

                    message = "\"%s\" is already used by \"%s\"!"%(name, values[0])

   124

                else:

   125

                    blockname = values[2]

   126

                    if len(values) > 3:

   127

                        blockinputs = values[3]

   128

                    else:

   129

                        blockinputs = None

   130

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

   131

                        dialog = wx.TextEntryDialog(self.ParentWindow.ParentWindow, "Block name", "Please enter a block name", "", wx.OK|wx.CANCEL|wx.CENTRE)

   132

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

   133

                            blockname = dialog.GetValue()

   134

                        else:

   135

                            return

   136

                        dialog.Destroy()

   137

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

   138

                        message = "\"%s\" pou already exists!"%blockname

   139

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

   140

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

   141

                    else:

   142

                        id = self.ParentWindow.GetNewId()

   143

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

   144

                        width, height = block.GetMinSize()

   145

                        if scaling is not None:

   146

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

   147

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

   148

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

   149

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

   150

                        block.SetPosition(x, y)

   151

                        block.SetSize(width, height)

   152

                        self.ParentWindow.AddBlock(block)

   153

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

   154

                        self.ParentWindow.RefreshBlockModel(block)

   155

                        self.ParentWindow.RefreshBuffer()

   156

                        self.ParentWindow.RefreshScrollBars()

   157

                        self.ParentWindow.RefreshVisibleElements()

   158

                        self.ParentWindow.ParentWindow.RefreshVariablePanel(self.ParentWindow.GetTagName())

   159

                        self.ParentWindow.Refresh(False)

   160

            elif values[1] != "location":

   161

                tagname = self.ParentWindow.GetTagName()

   162

                if values[3] == tagname:

   163

                    if values[1] == "Output":

   164

                        var_class = OUTPUT

   165

                    elif values[1] == "InOut":

   166

                        var_class = INPUT

   167

                    else:

   168

                        var_class = INPUT

   169

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

   170

                    if tree is not None:

   171

                        if len(tree[0]) > 0:

   172

                            menu = wx.Menu(title='')

   173

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

   174

                            self.ParentWindow.PopupMenuXY(menu)

   175

                        else:

   176

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

   177

                    else:

   178

                        message = "Unknown variable \"%s\" this POU!" % values[0]

   179

                else:

   180

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

   181

        if message is not None:

   182

            wx.CallAfter(self.ShowMessage, message)

   183

   184

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

   185

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

   186

            if base_path:

   187

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

   188

            else:

   189

                child_path = child_name

   190

            if len(child_dimensions) > 0:

   191

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

   192

                child_name += "[]"

   193

            new_id = wx.NewId()

   194

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

   195

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

   196

            if len(child_tree) > 0:

   197

                new_id = wx.NewId()

   198

                child_menu = wx.Menu(title='')

   199

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

   200

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

   201

   202

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

   203

        def AddVariableFunction(event):

   204

            self.AddParentVariableBlock(x, y, scaling, var_class, var_name, var_type)

   205

        return AddVariableFunction

   206

   207

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

   208

        id = self.ParentWindow.GetNewId()

   209

        variable = FBD_Variable(self.ParentWindow, var_class, var_name, var_type, id)

   210

        width, height = variable.GetMinSize()

   211

        if scaling is not None:

   212

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

   213

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

   214

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

   215

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

   216

        variable.SetPosition(x, y)

   217

        variable.SetSize(width, height)

   218

        self.ParentWindow.AddBlock(variable)

   219

        self.ParentWindow.Controler.AddEditedElementVariable(self.ParentWindow.GetTagName(), id, var_class)

   220

        self.ParentWindow.RefreshVariableModel(variable)

   221

        self.ParentWindow.RefreshBuffer()

   222

        self.ParentWindow.RefreshScrollBars()

   223

        self.ParentWindow.RefreshVisibleElements()

   224

        self.ParentWindow.Refresh(False)

   225

   226

    def ShowMessage(self, message):

   227

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

   228

        message.ShowModal()

   229

        message.Destroy()

   230

   231

   232

"""

   233

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

   234

manipulating graphic elements

   235

"""

   236

   237

class Viewer(wx.ScrolledWindow, DebugViewer):

   238

   239

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

   240

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

   241

            if id is not None:

   242

                event(self, id, function)

   243

            else:

   244

                event(self, function)

   245

   246

    # Create Alignment Menu items

   247

    def _init_coll_AlignmentMenu_Items(self, parent):

   248

        # Create menu items

   249

        AppendMenu(parent, help='', id=ID_VIEWERALIGNMENTMENUITEMS0,

   250

              kind=wx.ITEM_NORMAL, text=u'Left')

   251

        AppendMenu(parent, help='', id=ID_VIEWERALIGNMENTMENUITEMS1,

   252

              kind=wx.ITEM_NORMAL, text=u'Center')

   253

        AppendMenu(parent, help='', id=ID_VIEWERALIGNMENTMENUITEMS2,

   254

              kind=wx.ITEM_NORMAL, text=u'Right')

   255

        parent.AppendSeparator()

   256

        AppendMenu(parent, help='', id=ID_VIEWERALIGNMENTMENUITEMS4,

   257

              kind=wx.ITEM_NORMAL, text=u'Top')

   258

        AppendMenu(parent, help='', id=ID_VIEWERALIGNMENTMENUITEMS5,

   259

              kind=wx.ITEM_NORMAL, text=u'Middle')

   260

        AppendMenu(parent, help='', id=ID_VIEWERALIGNMENTMENUITEMS6,

   261

              kind=wx.ITEM_NORMAL, text=u'Bottom')

   262

        # Link menu event to corresponding called functions

   263

        self.Bind(wx.EVT_MENU, self.OnAlignLeftMenu,

   264

              id=ID_VIEWERALIGNMENTMENUITEMS0)

   265

        self.Bind(wx.EVT_MENU, self.OnAlignCenterMenu,

   266

              id=ID_VIEWERALIGNMENTMENUITEMS1)

   267

        self.Bind(wx.EVT_MENU, self.OnAlignRightMenu,

   268

              id=ID_VIEWERALIGNMENTMENUITEMS2)

   269

        self.Bind(wx.EVT_MENU, self.OnAlignTopMenu,

   270

              id=ID_VIEWERALIGNMENTMENUITEMS4)

   271

        self.Bind(wx.EVT_MENU, self.OnAlignMiddleMenu,

   272

              id=ID_VIEWERALIGNMENTMENUITEMS5)

   273

        self.Bind(wx.EVT_MENU, self.OnAlignBottomMenu,

   274

              id=ID_VIEWERALIGNMENTMENUITEMS6)

   275

   276

    # Create Contextual Menu items

   277

    def _init_coll_ContextualMenu_Items(self, parent):

   278

        # Create menu items

   279

        AppendMenu(parent, help='', id=ID_VIEWERCONTEXTUALMENUITEMS0,

   280

              kind=wx.ITEM_RADIO, text=u'No Modifier')

   281

        AppendMenu(parent, help='', id=ID_VIEWERCONTEXTUALMENUITEMS1,

   282

              kind=wx.ITEM_RADIO, text=u'Negated')

   283

        AppendMenu(parent, help='', id=ID_VIEWERCONTEXTUALMENUITEMS2,

   284

              kind=wx.ITEM_RADIO, text=u'Rising Edge')

   285

        AppendMenu(parent, help='', id=ID_VIEWERCONTEXTUALMENUITEMS3,

   286

              kind=wx.ITEM_RADIO, text=u'Falling Edge')

   287

        parent.AppendSeparator()

   288

        AppendMenu(parent, help='', id=ID_VIEWERCONTEXTUALMENUITEMS5,

   289

              kind=wx.ITEM_NORMAL, text=u'Add Wire Segment')

   290

        AppendMenu(parent, help='', id=ID_VIEWERCONTEXTUALMENUITEMS6,

   291

              kind=wx.ITEM_NORMAL, text=u'Delete Wire Segment')

   292

        parent.AppendSeparator()

   293

        AppendMenu(parent, help='', id=ID_VIEWERCONTEXTUALMENUITEMS8,

   294

              kind=wx.ITEM_NORMAL, text=u'Add Divergence Branch')

   295

        AppendMenu(parent, help='', id=ID_VIEWERCONTEXTUALMENUITEMS9,

   296

              kind=wx.ITEM_NORMAL, text=u'Delete Divergence Branch')

   297

        parent.AppendSeparator()

   298

        AppendMenu(parent, help='', id=ID_VIEWERCONTEXTUALMENUITEMS11,

   299

              kind=wx.ITEM_NORMAL, text=u'Clear Execution Order')

   300

        AppendMenu(parent, help='', id=ID_VIEWERCONTEXTUALMENUITEMS12,

   301

              kind=wx.ITEM_NORMAL, text=u'Reset Execution Order')

   302

        parent.AppendSeparator()

   303

        parent.AppendMenu(ID_VIEWERCONTEXTUALMENUITEMS14, "Alignment", self.AlignmentMenu)

   304

        parent.AppendSeparator()

   305

        AppendMenu(parent, help='', id=ID_VIEWERCONTEXTUALMENUITEMS16,

   306

              kind=wx.ITEM_NORMAL, text=u'Edit Block')

   307

        AppendMenu(parent, help='', id=ID_VIEWERCONTEXTUALMENUITEMS17,

   308

              kind=wx.ITEM_NORMAL, text=u'Delete')

   309

        # Link menu event to corresponding called functions

   310

        self.Bind(wx.EVT_MENU, self.OnNoModifierMenu,

   311

              id=ID_VIEWERCONTEXTUALMENUITEMS0)

   312

        self.Bind(wx.EVT_MENU, self.OnNegatedMenu,

   313

              id=ID_VIEWERCONTEXTUALMENUITEMS1)

   314

        self.Bind(wx.EVT_MENU, self.OnRisingEdgeMenu,

   315

              id=ID_VIEWERCONTEXTUALMENUITEMS2)

   316

        self.Bind(wx.EVT_MENU, self.OnFallingEdgeMenu,

   317

              id=ID_VIEWERCONTEXTUALMENUITEMS3)

   318

        self.Bind(wx.EVT_MENU, self.OnAddSegmentMenu,

   319

              id=ID_VIEWERCONTEXTUALMENUITEMS5)

   320

        self.Bind(wx.EVT_MENU, self.OnDeleteSegmentMenu,

   321

              id=ID_VIEWERCONTEXTUALMENUITEMS6)

   322

        self.Bind(wx.EVT_MENU, self.OnAddBranchMenu,

   323

              id=ID_VIEWERCONTEXTUALMENUITEMS8)

   324

        self.Bind(wx.EVT_MENU, self.OnDeleteBranchMenu,

   325

              id=ID_VIEWERCONTEXTUALMENUITEMS9)

   326

        self.Bind(wx.EVT_MENU, self.OnClearExecutionOrderMenu,

   327

              id=ID_VIEWERCONTEXTUALMENUITEMS11)

   328

        self.Bind(wx.EVT_MENU, self.OnResetExecutionOrderMenu,

   329

              id=ID_VIEWERCONTEXTUALMENUITEMS12)

   330

        self.Bind(wx.EVT_MENU, self.OnEditBlockMenu,

   331

              id=ID_VIEWERCONTEXTUALMENUITEMS16)

   332

        self.Bind(wx.EVT_MENU, self.OnDeleteMenu,

   333

              id=ID_VIEWERCONTEXTUALMENUITEMS17)

   334

   335

    # Create and initialize Contextual Menu

   336

    def _init_menus(self):

   337

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

   338

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

   339

   340

        self._init_coll_AlignmentMenu_Items(self.AlignmentMenu)

   341

        self._init_coll_ContextualMenu_Items(self.ContextualMenu)

   342

   343

    # Create a new Viewer

   344

    def __init__(self, parent, tagname, window, controler, debug = False, instancepath = ""):

   345

        wx.ScrolledWindow.__init__(self, parent, pos=wx.Point(0, 0), size=wx.Size(0, 0), 

   346

            style=wx.HSCROLL | wx.VSCROLL | wx.ALWAYS_SHOW_SB)

   347

        DebugViewer.__init__(self, controler, debug)

   348

        self._init_menus()

   349

        # Adding a rubberband to Viewer

   350

        self.rubberBand = RubberBand(drawingSurface=self)

   351

        self.SetBackgroundColour(wx.Colour(255,255,255))

   352

        self.SetBackgroundStyle(wx.BG_STYLE_CUSTOM)

   353

        self.ResetView()

   354

        self.Scaling = None

   355

        self.DrawGrid = True

   356

        self.GridBrush = wx.TRANSPARENT_BRUSH

   357

        self.PageSize = None

   358

        self.PagePen = wx.TRANSPARENT_PEN

   359

        self.DrawingWire = False

   360

        self.current_id = 0

   361

        self.TagName = tagname

   362

        self.Errors = []

   363

        self.InstancePath = instancepath

   364

        self.StartMousePos = None

   365

        self.StartScreenPos = None

   366

   367

        # Initialize Block, Wire and Comment numbers

   368

        self.block_id = self.wire_id = self.comment_id = 0

   369

   370

        # Initialize Viewer mode to Selection mode

   371

        self.Mode = MODE_SELECTION

   372

        self.SavedMode = False

   373

        self.CurrentLanguage = "FBD"

   374

   375

        self.ParentWindow = window

   376

        self.Controler = controler

   377

   378

        if not self.Debug:

   379

            self.SetDropTarget(ViewerDropTarget(self))

   380

   381

        self.NewDataRefreshRect = None

   382

   383

        dc = wx.ClientDC(self)

   384

        font = wx.Font(faces["size"], wx.SWISS, wx.NORMAL, wx.NORMAL, faceName = faces["mono"])

   385

        dc.SetFont(font)

   386

        width, height = dc.GetTextExtent("ABCDEFGHIJKLMNOPQRSTUVWXYZ")

   387

        while width > 260:

   388

            faces["size"] -= 1

   389

            font = wx.Font(faces["size"], wx.SWISS, wx.NORMAL, wx.NORMAL, faceName = faces["mono"])

   390

            dc.SetFont(font)

   391

            width, height = dc.GetTextExtent("ABCDEFGHIJKLMNOPQRSTUVWXYZ")

   392

        self.SetFont(font)

   393

        self.MiniTextDC = wx.MemoryDC()

   394

        self.MiniTextDC.SetFont(wx.Font(faces["size"] * 0.75, wx.SWISS, wx.NORMAL, wx.NORMAL, faceName = faces["helv"]))

   395

   396

        self.SetScale(len(ZOOM_FACTORS) / 2)

   397

   398

        self.ResetView()

   399

   400

        # Link Viewer event to corresponding methods

   401

        self.Bind(wx.EVT_PAINT, self.OnPaint)

   402

        self.Bind(wx.EVT_LEFT_DOWN, self.OnViewerLeftDown)

   403

        self.Bind(wx.EVT_LEFT_UP, self.OnViewerLeftUp)

   404

        self.Bind(wx.EVT_LEFT_DCLICK, self.OnViewerLeftDClick)

   405

        self.Bind(wx.EVT_RIGHT_DOWN, self.OnViewerRightDown)

   406

        self.Bind(wx.EVT_RIGHT_UP, self.OnViewerRightUp)

   407

        self.Bind(wx.EVT_MIDDLE_DOWN, self.OnViewerMiddleDown)

   408

        self.Bind(wx.EVT_MIDDLE_UP, self.OnViewerMiddleUp)

   409

        self.Bind(wx.EVT_LEAVE_WINDOW, self.OnLeaveViewer)

   410

        self.Bind(wx.EVT_MOTION, self.OnViewerMotion)

   411

        self.Bind(wx.EVT_CHAR, self.OnChar)

   412

        self.Bind(wx.EVT_SCROLLWIN, self.OnScrollWindow)

   413

        self.Bind(wx.EVT_SCROLLWIN_THUMBRELEASE, self.OnScrollStop)

   414

        self.Bind(wx.EVT_MOUSEWHEEL, self.OnMouseWheelWindow)

   415

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

   416

   417

    def GetScrolledRect(self, rect):

   418

        rect.x, rect.y = self.CalcScrolledPosition(int(rect.x * self.ViewScale[0]), 

   419

                                                   int(rect.y * self.ViewScale[1]))

   420

        rect.width = int(rect.width * self.ViewScale[0]) + 2

   421

        rect.height = int(rect.height * self.ViewScale[1]) + 2

   422

        return rect

   423

   424

    def GetScaling(self):

   425

        return self.Scaling

   426

   427

    def SetTagName(self, tagname):

   428

        self.TagName = tagname

   429

   430

    def GetTagName(self):

   431

        return self.TagName

   432

   433

    def GetInstancePath(self):

   434

        return self.InstancePath

   435

   436

    def IsViewing(self, tagname):

   437

        if self.Debug:

   438

            return self.InstancePath == tagname

   439

        else:

   440

            return self.TagName == tagname

   441

   442

    # Returns a new id

   443

    def GetNewId(self):

   444

        self.current_id += 1

   445

        return self.current_id

   446

   447

    # Destructor

   448

    def __del__(self):

   449

        DebugViewer.__del__(self)

   450

        self.Flush()

   451

        self.ResetView()

   452

   453

    def SetScale(self, scale_number, refresh=True):

   454

        self.CurrentScale = max(0, min(scale_number, len(ZOOM_FACTORS) - 1))

   455

        self.ViewScale = (ZOOM_FACTORS[self.CurrentScale], ZOOM_FACTORS[self.CurrentScale])

   456

        self.RefreshScaling(refresh)

   457

   458

    def GetScale(self):

   459

        return self.CurrentScale

   460

   461

    def GetLogicalDC(self, buffered=False):

   462

        if buffered:

   463

            bitmap = wx.EmptyBitmap(*self.GetClientSize())

   464

            dc = wx.MemoryDC(bitmap)

   465

        else:

   466

            dc = wx.ClientDC(self)

   467

        dc.SetFont(self.GetFont())

   468

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

   469

            self.DoPrepareDC(dc)

   470

        else:

   471

            self.PrepareDC(dc)

   472

        dc.SetUserScale(self.ViewScale[0], self.ViewScale[1])

   473

        return dc

   474

   475

    def GetMiniTextExtent(self, text):

   476

        return self.MiniTextDC.GetTextExtent(text)

   477

   478

    def GetMiniFont(self):

   479

        return self.MiniTextDC.GetFont()

   480

   481

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

   482

#                         Element management functions

   483

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

   484

   485

    def AddBlock(self, block):

   486

        self.block_id += 1

   487

        self.Blocks[block] = self.block_id

   488