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 wx

    26

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

    27

    USE_AUI = True

    28

else:

    29

    USE_AUI = False

    30

    31

from plcopen.structures import *

    32

from PLCControler import ITEM_POU

    33

    34

from Dialogs import *

    35

    36

from types import TupleType

    37

    38

SCROLLBAR_UNIT = 10

    39

WINDOW_BORDER = 10

    40

SCROLL_ZONE = 10

    41

    42

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

    43

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

    44

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

    45

    else:

    46

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

    47

    48

if wx.Platform == '__WXMSW__':

    49

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

    50

              'mono' : 'Courier New',

    51

              'helv' : 'Arial',

    52

              'other': 'Comic Sans MS',

    53

              'size' : 20,

    54

             }

    55

else:

    56

    faces = { 'times': 'Times',

    57

              'mono' : 'Courier',

    58

              'helv' : 'Helvetica',

    59

              'other': 'new century schoolbook',

    60

              'size' : 20,

    61

             }

    62

    63

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

    64

#                       Graphic elements Viewer base class

    65

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

    66

    67

# ID Constants for alignment menu items

    68

[ID_VIEWERALIGNMENTMENUITEMS0, ID_VIEWERALIGNMENTMENUITEMS1,

    69

 ID_VIEWERALIGNMENTMENUITEMS2, ID_VIEWERALIGNMENTMENUITEMS4,

    70

 ID_VIEWERALIGNMENTMENUITEMS5, ID_VIEWERALIGNMENTMENUITEMS6,

    71

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

    72

    73

# ID Constants for contextual menu items

    74

[ID_VIEWERCONTEXTUALMENUITEMS0, ID_VIEWERCONTEXTUALMENUITEMS1,

    75

 ID_VIEWERCONTEXTUALMENUITEMS2, ID_VIEWERCONTEXTUALMENUITEMS3,

    76

 ID_VIEWERCONTEXTUALMENUITEMS5, ID_VIEWERCONTEXTUALMENUITEMS6,

    77

 ID_VIEWERCONTEXTUALMENUITEMS8, ID_VIEWERCONTEXTUALMENUITEMS9,

    78

 ID_VIEWERCONTEXTUALMENUITEMS11, ID_VIEWERCONTEXTUALMENUITEMS12,

    79

 ID_VIEWERCONTEXTUALMENUITEMS14, ID_VIEWERCONTEXTUALMENUITEMS16,

    80

 ID_VIEWERCONTEXTUALMENUITEMS17,

    81

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

    82

    83

    84

class ViewerDropTarget(wx.TextDropTarget):

    85

    86

    def __init__(self, parent):

    87

        wx.TextDropTarget.__init__(self)

    88

        self.ParentWindow = parent

    89

    90

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

    91

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

    92

        scaling = self.ParentWindow.Scaling

    93

        message = None

    94

        try:

    95

            values = eval(data)

    96

        except:

    97

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

    98

            values = None

    99

        if not isinstance(values, TupleType):

   100

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

   101

            values = None

   102

        if values is not None:

   103

            if values[1] == "debug":

   104

                pass

   105

            elif values[1] == "program":

   106

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

   107

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

   108

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

   109

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

   110

                if name == values[0]:

   111

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

   112

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

   113

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

   114

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

   115

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

   116

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

   117

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

   118

                else:

   119

                    blockname = values[2]

   120

                    if len(values) > 3:

   121

                        blockinputs = values[3]

   122

                    else:

   123

                        blockinputs = None

   124

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

   125

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

   126

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

   127

                            blockname = dialog.GetValue()

   128

                        else:

   129

                            return

   130

                        dialog.Destroy()

   131

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

   132

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

   133

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

   134

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

   135

                    else:

   136

                        id = self.ParentWindow.GetNewId()

   137

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

   138

                        width, height = block.GetMinSize()

   139

                        if scaling is not None:

   140

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

   141

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

   142

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

   143

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

   144

                        block.SetPosition(x, y)

   145

                        block.SetSize(width, height)

   146

                        self.ParentWindow.AddBlock(block)

   147

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

   148

                        self.ParentWindow.RefreshBlockModel(block)

   149

                        self.ParentWindow.RefreshBuffer()

   150

                        self.ParentWindow.RefreshScrollBars()

   151

                        self.ParentWindow.RefreshVisibleElements()

   152

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

   153

                        self.ParentWindow.Refresh(False)

   154

            elif values[1] != "location":

   155

                tagname = self.ParentWindow.GetTagName()

   156

                if values[3] == tagname:

   157

                    if values[1] == "Output":

   158

                        var_class = OUTPUT

   159

                    elif values[1] == "InOut":

   160

                        var_class = INPUT

   161

                    else:

   162

                        var_class = INPUT

   163

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

   164

                    if tree is not None:

   165

                        if len(tree) > 0:

   166

                            menu = wx.Menu(title='')

   167

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

   168

                            self.ParentWindow.PopupMenuXY(menu)

   169

                        else:

   170

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

   171

                    else:

   172

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

   173

                else:

   174

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

   175

        if message is not None:

   176

            wx.CallAfter(self.ShowMessage, message)

   177

   178

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

   180

            if base_path:

   181

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

   182

            else:

   183

                child_path = child_name

   186

                child_name += "[]"

   187

            new_id = wx.NewId()

   188

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

   189

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

   190

            if len(child_tree) > 0:

   191

                new_id = wx.NewId()

   192

                child_menu = wx.Menu(title='')

   193

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

   194

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

   195

   196

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

   197

        def AddVariableFunction(event):

   198

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

   199

        return AddVariableFunction

   200

   201

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

   202

        id = self.ParentWindow.GetNewId()

   203

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

   204

        width, height = variable.GetMinSize()

   205

        if scaling is not None:

   206

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

   207

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

   208

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

   209

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

   210

        variable.SetPosition(x, y)

   211

        variable.SetSize(width, height)

   212

        self.ParentWindow.AddBlock(variable)

   213

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

   214

        self.ParentWindow.RefreshVariableModel(variable)

   215

        self.ParentWindow.RefreshBuffer()

   216

        self.ParentWindow.RefreshScrollBars()

   217

        self.ParentWindow.RefreshVisibleElements()

   218

        self.ParentWindow.Refresh(False)

   219

   220

    def ShowMessage(self, message):

   221

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

   222

        message.ShowModal()

   223

        message.Destroy()

   224

   225

   226

"""

   227

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

   228

manipulating graphic elements

   229

"""

   230

   231

class Viewer(wx.ScrolledWindow):

   232

   233

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

   234

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

   235

            if id is not None:

   236

                event(self, id, function)

   237

            else:

   238

                event(self, function)

   239

   240

    # Create Alignment Menu items

   241

    def _init_coll_AlignmentMenu_Items(self, parent):

   242

        # Create menu items

   243

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

   244

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

   245

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

   246

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

   247

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

   248

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

   249

        parent.AppendSeparator()

   250

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

   251

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

   252

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

   253

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

   254

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

   255

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

   256

        # Link menu event to corresponding called functions

   257

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

   258

              id=ID_VIEWERALIGNMENTMENUITEMS0)

   259

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

   260

              id=ID_VIEWERALIGNMENTMENUITEMS1)

   261

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

   262

              id=ID_VIEWERALIGNMENTMENUITEMS2)

   263

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

   264

              id=ID_VIEWERALIGNMENTMENUITEMS4)

   265

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

   266

              id=ID_VIEWERALIGNMENTMENUITEMS5)

   267

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

   268

              id=ID_VIEWERALIGNMENTMENUITEMS6)

   269

   270

    # Create Contextual Menu items

   271

    def _init_coll_ContextualMenu_Items(self, parent):

   272

        # Create menu items

   273

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

   274

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

   275

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

   276

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

   277

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

   278

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

   279

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

   280

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

   281

        parent.AppendSeparator()

   282

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

   283

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

   284

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

   285

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

   286

        parent.AppendSeparator()

   287

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

   288

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

   289

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

   290

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

   291

        parent.AppendSeparator()

   292

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

   293

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

   294

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

   295

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

   296

        parent.AppendSeparator()

   297

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

   298

        parent.AppendSeparator()

   299

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

   300

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

   301

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

   302

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

   303

        # Link menu event to corresponding called functions

   304

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

   305

              id=ID_VIEWERCONTEXTUALMENUITEMS0)

   306

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

   307

              id=ID_VIEWERCONTEXTUALMENUITEMS1)

   308

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

   309

              id=ID_VIEWERCONTEXTUALMENUITEMS2)

   310

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

   311

              id=ID_VIEWERCONTEXTUALMENUITEMS3)

   312

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

   313

              id=ID_VIEWERCONTEXTUALMENUITEMS5)

   314

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

   315

              id=ID_VIEWERCONTEXTUALMENUITEMS6)

   316

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

   317

              id=ID_VIEWERCONTEXTUALMENUITEMS8)

   318

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

   319

              id=ID_VIEWERCONTEXTUALMENUITEMS9)

   320

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

   321

              id=ID_VIEWERCONTEXTUALMENUITEMS11)

   322

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

   323

              id=ID_VIEWERCONTEXTUALMENUITEMS12)

   324

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

   325

              id=ID_VIEWERCONTEXTUALMENUITEMS16)

   326

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

   327

              id=ID_VIEWERCONTEXTUALMENUITEMS17)

   328

   329

    # Create and initialize Contextual Menu

   330

    def _init_menus(self):

   331

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

   332

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

   333

   334

        self._init_coll_AlignmentMenu_Items(self.AlignmentMenu)

   335

        self._init_coll_ContextualMenu_Items(self.ContextualMenu)

   336

   337

    # Create a new Viewer

   338

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

   339

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

   340

            style=wx.HSCROLL | wx.VSCROLL)

   341

        self._init_menus()

   342

        # Adding a rubberband to Viewer

   343

        self.rubberBand = RubberBand(drawingSurface=self)

   344

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

   345

        self.SetBackgroundStyle(wx.BG_STYLE_CUSTOM)

   346

        self.ResetView()

   347

        self.Scaling = None

   348

        self.DrawGrid = True

   349

        self.GridBrush = wx.TRANSPARENT_BRUSH

   350

        self.PageSize = None

   351

        self.PagePen = wx.TRANSPARENT_PEN

   352

        self.DrawingWire = False

   353

        self.current_id = 0

   354

        self.TagName = tagname

   355

        self.Errors = []

   356

        self.Debug = debug

   357

        self.InstancePath = instancepath

   358

   359

        # Initialize Block, Wire and Comment numbers

   360

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

   361

   362

        # Initialize Viewer mode to Selection mode

   363

        self.Mode = MODE_SELECTION

   364

        self.SavedMode = False

   365

        self.CurrentLanguage = "FBD"

   366

   367

        self.ParentWindow = window

   368

        self.Controler = controler

   369

   370

        if not self.Debug:

   371

            self.SetDropTarget(ViewerDropTarget(self))

   372

   373

        dc = wx.ClientDC(self)

   374

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

   375

        dc.SetFont(font)

   376

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

   377

        while width > 260:

   378

            faces["size"] -= 1

   379

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

   380

            dc.SetFont(font)

   381

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

   382

        self.SetFont(font)

   383

   384

        self.ResetView()

   385

   386

        # Link Viewer event to corresponding methods

   387

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

   388

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

   389

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

   390

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

   391

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

   392

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

   393

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

   394

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

   395

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

   396

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

   397

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

   398

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

   399

   400

    def GetScrolledRect(self, rect):

   401

        rect.x, rect.y = self.CalcScrolledPosition(rect.x, rect.y)

   402

        return rect

   403

   404

    def GetScaling(self):

   405

        return self.Scaling

   406

   407

    def SetTagName(self, tagname):

   408

        self.TagName = tagname

   409

   410

    def GetTagName(self):

   411

        return self.TagName

   412

   413

    def GetInstancePath(self):

   414

        return self.InstancePath

   415

   416

    def IsViewing(self, tagname):

   417

        if self.Debug:

   418

            return self.InstancePath == tagname

   419

        else:

   420

            return self.TagName == tagname

   421

   422

    # Returns a new id

   423

    def GetNewId(self):

   424

        self.current_id += 1

   425

        return self.current_id

   426

   427

    # Destructor

   428

    def __del__(self):

   429

        self.Flush()

   430

        self.ResetView()

   431

   432

    def GetLogicalDC(self, buffered=False):

   433

        if buffered:

   434

            dc = wx.AutoBufferedPaintDC(self)

   435

        else:

   436

            dc = wx.ClientDC(self)

   437

        dc.SetFont(self.GetFont())

   438

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

   439

            self.DoPrepareDC(dc)

   440

        else:

   441

            self.PrepareDC(dc)

   442

        return dc

   443

   444

    def GetMiniFont(self):

   445

        font = self.GetFont()

   446

        return wx.Font(font.GetPointSize() * 0.75, wx.SWISS, wx.NORMAL, wx.NORMAL, faceName = faces["helv"])

   447

   448

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

   449

#                         Element management functions

   450

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

   451

   452

    def AddBlock(self, block):

   453

        self.block_id += 1

   454

        self.Blocks[block] = self.block_id

   455

   456

    def AddWire(self, wire):

   457

        self.wire_id += 1

   458

        self.Wires[wire] = self.wire_id

   459

   460

    def AddComment(self, comment):

   461

        self.comment_id += 1

   462

        self.Comments[comment] = self.comment_id

   463

   464

    def IsBlock(self, block):

   465

        return self.Blocks.get(block, False)

   466

   467

    def IsWire(self, wire):

   468

        return self.Wires.get(wire, False)

   469

   470

    def IsComment(self, comment):

   471

        return self.Comments.get(comment, False)

   472

   473

    def RemoveBlock(self, block):

   474

        self.Blocks.pop(block)

   475

   476

    def RemoveWire(self, wire):

   477

        self.Wires.pop(wire)

   478

   479

    def RemoveComment(self, comment):

   480

        self.Comments.pop(comment)

   481

   482

    def GetElements(self, sort_blocks=False, sort_wires=False, sort_comments=False):

   483

        blocks = self.Blocks.keys()

   484

        wires = self.Wires.keys()

   485

        comments = self.Comments.keys()

   486

        if sort_blocks:

   487

            blocks.sort(lambda x, y: cmp(self.Blocks[x], self.Blocks[y]))

   488

        if sort_wires:

   489

            wires.sort(lambda x, y: cmp(self.Wires[x], self.Wires[y]))

   490

        if sort_comments: