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 re

    26

from types import *

    27

    28

import wx

    29

import wx.stc

    30

    31

from graphics.GraphicCommons import ERROR_HIGHLIGHT, SEARCH_RESULT_HIGHLIGHT, REFRESH_HIGHLIGHT_PERIOD

    32

from plcopen.structures import ST_BLOCK_START_KEYWORDS, ST_BLOCK_END_KEYWORDS, IEC_BLOCK_START_KEYWORDS, IEC_BLOCK_END_KEYWORDS, LOCATIONDATATYPES

    33

from EditorPanel import EditorPanel

    34

from controls.CustomStyledTextCtrl import CustomStyledTextCtrl, faces, GetCursorPos

    35

    36

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

    37

#                         Textual programs Viewer class

    38

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

    39

    40

    41

NEWLINE = "\n"

    42

NUMBERS = [str(i) for i in xrange(10)]

    43

LETTERS = ['_']

    44

for i in xrange(26):

    45

    LETTERS.append(chr(ord('a') + i))

    46

    LETTERS.append(chr(ord('A') + i))

    47

    48

[STC_PLC_WORD, STC_PLC_COMMENT, STC_PLC_NUMBER, STC_PLC_STRING, 

    49

 STC_PLC_VARIABLE, STC_PLC_PARAMETER, STC_PLC_FUNCTION, STC_PLC_JUMP, 

    50

 STC_PLC_ERROR, STC_PLC_SEARCH_RESULT] = range(10)

    51

[SPACE, WORD, NUMBER, STRING, WSTRING, COMMENT, PRAGMA, DPRAGMA] = range(8)

    52

    53

[ID_TEXTVIEWER, ID_TEXTVIEWERTEXTCTRL,

    54

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

    55

    56

re_texts = {}

    57

re_texts["letter"] = "[A-Za-z]"

    58

re_texts["digit"] = "[0-9]"

    59

re_texts["identifier"] = "((?:%(letter)s|(?:_(?:%(letter)s|%(digit)s)))(?:_?(?:%(letter)s|%(digit)s))*)"%re_texts

    60

IDENTIFIER_MODEL = re.compile(re_texts["identifier"])

    61

LABEL_MODEL = re.compile("[ \t\n]%(identifier)s:[ \t\n]"%re_texts)

    62

EXTENSIBLE_PARAMETER = re.compile("IN[1-9][0-9]*$")

    63

    64

HIGHLIGHT_TYPES = {

    65

    ERROR_HIGHLIGHT: STC_PLC_ERROR,

    66

    SEARCH_RESULT_HIGHLIGHT: STC_PLC_SEARCH_RESULT,

    67

}

    68

    69

def LineStartswith(line, symbols):

    70

    return reduce(lambda x, y: x or y, map(lambda x: line.startswith(x), symbols), False)

    71

    72

class TextViewer(EditorPanel):

    73

    74

    ID = ID_TEXTVIEWER

    75

    76

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

    77

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

    78

            if id is not None:

    79

                event(self, id, function)

    80

            else:

    81

                event(self, function)

    82

    83

    def _init_Editor(self, prnt):

    84

        self.Editor = CustomStyledTextCtrl(id=ID_TEXTVIEWERTEXTCTRL, 

    85

                parent=prnt, name="TextViewer", size=wx.Size(0, 0), style=0)

    86

        self.Editor.ParentWindow = self

    87

    88

        self.Editor.CmdKeyAssign(ord('+'), wx.stc.STC_SCMOD_CTRL, wx.stc.STC_CMD_ZOOMIN)

    89

        self.Editor.CmdKeyAssign(ord('-'), wx.stc.STC_SCMOD_CTRL, wx.stc.STC_CMD_ZOOMOUT)

    90

    91

        self.Editor.SetViewWhiteSpace(False)

    92

    93

        self.Editor.SetLexer(wx.stc.STC_LEX_CONTAINER)

    94

    95

        # Global default styles for all languages

    96

        self.Editor.StyleSetSpec(wx.stc.STC_STYLE_DEFAULT, "face:%(mono)s,size:%(size)d" % faces)

    97

        self.Editor.StyleClearAll()  # Reset all to be like the default

    98

    99

        self.Editor.StyleSetSpec(wx.stc.STC_STYLE_LINENUMBER,  "back:#C0C0C0,size:%(size)d" % faces)

   100

        self.Editor.SetSelBackground(1, "#E0E0E0")

   101

   102

        # Highlighting styles

   103

        self.Editor.StyleSetSpec(STC_PLC_WORD, "fore:#00007F,bold,size:%(size)d" % faces)

   104

        self.Editor.StyleSetSpec(STC_PLC_VARIABLE, "fore:#7F0000,size:%(size)d" % faces)

   105

        self.Editor.StyleSetSpec(STC_PLC_PARAMETER, "fore:#7F007F,size:%(size)d" % faces)

   106

        self.Editor.StyleSetSpec(STC_PLC_FUNCTION, "fore:#7F7F00,size:%(size)d" % faces)

   107

        self.Editor.StyleSetSpec(STC_PLC_COMMENT, "fore:#7F7F7F,size:%(size)d" % faces)

   108

        self.Editor.StyleSetSpec(STC_PLC_NUMBER, "fore:#007F7F,size:%(size)d" % faces)

   109

        self.Editor.StyleSetSpec(STC_PLC_STRING, "fore:#007F00,size:%(size)d" % faces)

   110

        self.Editor.StyleSetSpec(STC_PLC_JUMP, "fore:#FF7FFF,size:%(size)d" % faces)

   111

        self.Editor.StyleSetSpec(STC_PLC_ERROR, "fore:#FF0000,back:#FFFF00,size:%(size)d" % faces)

   112

        self.Editor.StyleSetSpec(STC_PLC_SEARCH_RESULT, "fore:#FFFFFF,back:#FFA500,size:%(size)d" % faces)

   113

   114

        # Indicators styles

   115

        self.Editor.IndicatorSetStyle(0, wx.stc.STC_INDIC_SQUIGGLE)

   116

        if self.ParentWindow is not None and self.Controler is not None:

   117

            self.Editor.IndicatorSetForeground(0, wx.RED)

   118

        else:

   119

            self.Editor.IndicatorSetForeground(0, wx.WHITE)

   120

   121

        # Line numbers in the margin

   122

        self.Editor.SetMarginType(1, wx.stc.STC_MARGIN_NUMBER)

   123

        self.Editor.SetMarginWidth(1, 50)

   124

   125

        # Folding

   126

        self.Editor.MarkerDefine(wx.stc.STC_MARKNUM_FOLDEROPEN,    wx.stc.STC_MARK_BOXMINUS,          "white", "#808080")

   127

        self.Editor.MarkerDefine(wx.stc.STC_MARKNUM_FOLDER,        wx.stc.STC_MARK_BOXPLUS,           "white", "#808080")

   128

        self.Editor.MarkerDefine(wx.stc.STC_MARKNUM_FOLDERSUB,     wx.stc.STC_MARK_VLINE,             "white", "#808080")

   129

        self.Editor.MarkerDefine(wx.stc.STC_MARKNUM_FOLDERTAIL,    wx.stc.STC_MARK_LCORNER,           "white", "#808080")

   130

        self.Editor.MarkerDefine(wx.stc.STC_MARKNUM_FOLDEREND,     wx.stc.STC_MARK_BOXPLUSCONNECTED,  "white", "#808080")

   131

        self.Editor.MarkerDefine(wx.stc.STC_MARKNUM_FOLDEROPENMID, wx.stc.STC_MARK_BOXMINUSCONNECTED, "white", "#808080")

   132

        self.Editor.MarkerDefine(wx.stc.STC_MARKNUM_FOLDERMIDTAIL, wx.stc.STC_MARK_TCORNER,           "white", "#808080")

   133

   134

        # Indentation size

   135

        self.Editor.SetTabWidth(2)

   136

        self.Editor.SetUseTabs(0)

   137

   138

        self.Editor.SetModEventMask(wx.stc.STC_MOD_BEFOREINSERT|

   139

                                    wx.stc.STC_MOD_BEFOREDELETE|

   140

                                    wx.stc.STC_PERFORMED_USER)

   141

   142

        self.Bind(wx.stc.EVT_STC_STYLENEEDED, self.OnStyleNeeded, id=ID_TEXTVIEWERTEXTCTRL)

   143

        self.Editor.Bind(wx.stc.EVT_STC_MARGINCLICK, self.OnMarginClick)

   144

        self.Editor.Bind(wx.stc.EVT_STC_UPDATEUI, self.OnUpdateUI)

   145

        self.Editor.Bind(wx.EVT_KEY_DOWN, self.OnKeyDown)

   146

        if self.Controler is not None:

   147

            self.Editor.Bind(wx.EVT_KILL_FOCUS, self.OnKillFocus)

   148

            self.Bind(wx.stc.EVT_STC_DO_DROP, self.OnDoDrop, id=ID_TEXTVIEWERTEXTCTRL)

   149

            self.Bind(wx.stc.EVT_STC_MODIFIED, self.OnModification, id=ID_TEXTVIEWERTEXTCTRL)

   150

   151

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

   152

        if tagname != "" and controler is not None:

   153

            self.VARIABLE_PANEL_TYPE = controler.GetPouType(tagname.split("::")[1])

   154

   155

        EditorPanel.__init__(self, parent, tagname, window, controler, debug)

   156

   157

        self.Keywords = []

   158

        self.Variables = {}

   159

        self.Functions = {}

   160

        self.TypeNames = []

   161

        self.Jumps = []

   162

        self.EnumeratedValues = []

   163

        self.DisableEvents = True

   164

        self.TextSyntax = None

   165

        self.CurrentAction = None

   166

   167

        self.InstancePath = instancepath

   168

        self.ContextStack = []

   169

        self.CallStack = []

   170

   171

        self.ResetSearchResults()

   172

   173

        self.RefreshHighlightsTimer = wx.Timer(self, -1)

   174

        self.Bind(wx.EVT_TIMER, self.OnRefreshHighlightsTimer, self.RefreshHighlightsTimer)

   175

   176

    def __del__(self):

   177

        self.RefreshHighlightsTimer.Stop()

   178

   179

    def GetTitle(self):

   180

        if self.Debug or self.TagName == "":

   181

            if len(self.InstancePath) > 15:

   182

                return "..." + self.InstancePath[-12:]

   183

            return self.InstancePath

   184

        return EditorPanel.GetTitle(self)

   185

   186

    def GetInstancePath(self):

   187

        return self.InstancePath

   188

   189

    def IsViewing(self, tagname):

   190

        if self.Debug or self.TagName == "":

   191

            return self.InstancePath == tagname

   192

        else:

   193

            return self.TagName == tagname

   194

   195

    def GetText(self):

   196

        return self.Editor.GetText()

   197

   198

    def SetText(self, text):

   199

        self.Editor.SetText(text)

   200

   201

    def SelectAll(self):

   202

        self.Editor.SelectAll()

   203

   204

    def Colourise(self, start, end):

   205

        self.Editor.Colourise(start, end)

   206

   207

    def StartStyling(self, pos, mask):

   208

        self.Editor.StartStyling(pos, mask)

   209

   210

    def SetStyling(self, length, style):

   211

        self.Editor.SetStyling(length, style)

   212

   213

    def GetCurrentPos(self):

   214

        return self.Editor.GetCurrentPos()

   215

   216

    def ResetSearchResults(self):

   217

        self.Highlights = []

   218

        self.SearchParams = None

   219

        self.SearchResults = None

   220

        self.CurrentFindHighlight = None

   221

   222

    def OnModification(self, event):

   223

        if not self.DisableEvents:

   224

            mod_type = event.GetModificationType()

   225

            if mod_type&wx.stc.STC_MOD_BEFOREINSERT:

   226

                if self.CurrentAction == None:

   227

                    self.StartBuffering()

   228

                elif self.CurrentAction[0] != "Add" or self.CurrentAction[1] != event.GetPosition() - 1:

   229

                    self.Controler.EndBuffering()

   230

                    self.StartBuffering()

   231

                self.CurrentAction = ("Add", event.GetPosition())

   232

                wx.CallAfter(self.RefreshModel)

   233

            elif mod_type&wx.stc.STC_MOD_BEFOREDELETE:

   234

                if self.CurrentAction == None:

   235

                    self.StartBuffering()

   236

                elif self.CurrentAction[0] != "Delete" or self.CurrentAction[1] != event.GetPosition() + 1:

   237

                    self.Controler.EndBuffering()

   238

                    self.StartBuffering()

   239

                self.CurrentAction = ("Delete", event.GetPosition())

   240

                wx.CallAfter(self.RefreshModel)

   241

        event.Skip()

   242

   243

    def OnDoDrop(self, event):

   244

        try:

   245

            values = eval(event.GetDragText())

   246

        except:

   247

            values = event.GetDragText()

   248

        if isinstance(values, tuple):

   249

            message = None

   250

            if values[1] in ["program", "debug"]:

   251

                event.SetDragText("")

   252

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

   253

                blocktype = values[0]

   254

                blockname = values[2]

   255

                if len(values) > 3:

   256

                    blockinputs = values[3]

   257

                else:

   258

                    blockinputs = None

   259

                if values[1] != "function": 

   260

                    if blockname == "":

   261

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

   262

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

   263

                            blockname = dialog.GetValue()

   264

                        else:

   265

                            return

   266

                        dialog.Destroy()

   267

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

   268

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

   269

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

   270

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

   271

                    else:

   272

                        self.Controler.AddEditedElementPouVar(self.TagName, values[0], blockname)

   273

                        self.RefreshVariablePanel()

   274

                        self.RefreshVariableTree()

   275

                blockinfo = self.Controler.GetBlockType(blocktype, blockinputs, self.Debug)

   276

                hint = ',\n    '.join(

   277

                            [ " " + fctdecl[0]+" := (*"+fctdecl[1]+"*)" for fctdecl in blockinfo["inputs"]] +

   278

                            [ " " + fctdecl[0]+" => (*"+fctdecl[1]+"*)" for fctdecl in blockinfo["outputs"]])

   279

                if values[1] == "function":

   280

                    event.SetDragText(blocktype+"(\n    "+hint+")")

   281

                else:

   282

                    event.SetDragText(blockname+"(\n    "+hint+")")

   283

            elif values[1] == "location":

   284

                pou_name, pou_type = self.Controler.GetEditedElementType(self.TagName, self.Debug)

   285

                if len(values) > 2 and pou_type == "program":

   286

                    var_name = values[3]

   287

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

   288

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

   289

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

   290

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

   291

                    else:

   292

                        location = values[0]

   293

                        if not location.startswith("%"):

   294

                            dialog = wx.SingleChoiceDialog(self.ParentWindow, 

   295

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

   296

                                  ["Input", "Output", "Memory"], 

   297

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

   298

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

   299

                                selected = dialog.GetSelection()

   300

                            else:

   301

                                selected = None

   302

                            dialog.Destroy()

   303

                            if selected is None:

   304

                                event.SetDragText("")

   305

                                return

   306

                            if selected == 0:

   307

                                location = "%I" + location

   308

                            elif selected == 1:

   309

                                location = "%Q" + location

   310

                            else:

   311

                                location = "%M" + location

   312

                        if values[2] is not None:

   313

                            var_type = values[2]

   314

                        else:

   315

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

   316

                        self.Controler.AddEditedElementPouVar(self.TagName, var_type, var_name, location, values[4])

   317

                        self.RefreshVariablePanel()

   318

                        self.RefreshVariableTree()

   319

                        event.SetDragText(var_name)

   320

                else:

   321

                    event.SetDragText("")

   322

            elif values[1] == "Global":

   323

                var_name = values[0]

   324

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

   325

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

   326

                else:

   327

                    if not var_name.upper() in [name.upper() for name in self.Controler.GetEditedElementVariables(self.TagName, self.Debug)]:

   328

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

   329

                        self.RefreshVariablePanel()

   330

                        self.RefreshVariableTree()

   331

                    event.SetDragText(var_name)

   332

            elif values[1] == "Constant":

   333

                event.SetDragText(values[0])

   334

            elif values[3] == self.TagName:

   335

                self.ResetBuffer()

   336

                event.SetDragText(values[0])

   337

                wx.CallAfter(self.RefreshModel)

   338

            else:

   339

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

   340

            if message is not None:

   341

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

   342

                dialog.ShowModal()

   343

                dialog.Destroy()

   344

                event.SetDragText("")

   345

        event.Skip()

   346

   347

    def SetTextSyntax(self, syntax):

   348

        self.TextSyntax = syntax

   349

        if syntax in ["ST", "ALL"]:

   350

            self.Editor.SetMarginType(2, wx.stc.STC_MARGIN_SYMBOL)

   351

            self.Editor.SetMarginMask(2, wx.stc.STC_MASK_FOLDERS)

   352

            self.Editor.SetMarginSensitive(2, 1)

   353

            self.Editor.SetMarginWidth(2, 12)

   354

            if syntax == "ST":

   355

                self.BlockStartKeywords = ST_BLOCK_START_KEYWORDS

   356

                self.BlockEndKeywords = ST_BLOCK_START_KEYWORDS

   357

            else:

   358

                self.BlockStartKeywords = IEC_BLOCK_START_KEYWORDS

   359

                self.BlockEndKeywords = IEC_BLOCK_START_KEYWORDS

   360

        else:

   361

            self.BlockStartKeywords = []

   362

            self.BlockEndKeywords = []

   363

   364

    def SetKeywords(self, keywords):

   365

        self.Keywords = [keyword.upper() for keyword in keywords]

   366

        self.Colourise(0, -1)

   367

   368

    def RefreshJumpList(self):

   369

        if self.TextSyntax != "IL":

   370

            self.Jumps = [jump.upper() for jump in LABEL_MODEL.findall(self.GetText())]

   371

            self.Colourise(0, -1)

   372

   373

    # Buffer the last model state

   374

    def RefreshBuffer(self):

   375

        self.Controler.BufferProject()

   376

        if self.ParentWindow:

   377

            self.ParentWindow.RefreshTitle()

   378

            self.ParentWindow.RefreshFileMenu()

   379

            self.ParentWindow.RefreshEditMenu()

   380

   381

    def StartBuffering(self):

   382

        self.Controler.StartBuffering()

   383

        if self.ParentWindow:

   384

            self.ParentWindow.RefreshTitle()

   385

            self.ParentWindow.RefreshFileMenu()

   386

            self.ParentWindow.RefreshEditMenu()

   387

   388

    def ResetBuffer(self):

   389

        if self.CurrentAction != None:

   390

            self.Controler.EndBuffering()

   391

            self.CurrentAction = None

   392

   393

    def GetBufferState(self):

   394

        if not self.Debug and self.TextSyntax != "ALL":

   395

            return self.Controler.GetBufferState()

   396

        return False, False

   397

   398

    def Undo(self):

   399

        if not self.Debug and self.TextSyntax != "ALL":

   400

            self.Controler.LoadPrevious()

   401

            self.ParentWindow.CloseTabsWithoutModel()

   402

   403

    def Redo(self):

   404

        if not self.Debug and self.TextSyntax != "ALL":

   405

            self.Controler.LoadNext()

   406

            self.ParentWindow.CloseTabsWithoutModel()

   407

   408

    def HasNoModel(self):

   409

        if not self.Debug and self.TextSyntax != "ALL":

   410

            return self.Controler.GetEditedElement(self.TagName) is None

   411

        return False

   412

   413

    def RefreshView(self, variablepanel=True):

   414

        EditorPanel.RefreshView(self, variablepanel)

   415

   416

        if self.Controler is not None:

   417

            self.ResetBuffer()

   418

            self.DisableEvents = True

   419

            old_cursor_pos = self.GetCurrentPos()

   420

            line = self.Editor.GetFirstVisibleLine()

   421

            column = self.Editor.GetXOffset()

   422

            old_text = self.GetText()

   423

            new_text = self.Controler.GetEditedElementText(self.TagName, self.Debug)

   424

            if old_text != new_text:

   425

                self.SetText(new_text)

   426

                new_cursor_pos = GetCursorPos(old_text, new_text)

   427

                self.Editor.LineScroll(column, line)

   428

                if new_cursor_pos != None:

   429

                    self.Editor.GotoPos(new_cursor_pos)

   430

                else:

   431

                    self.Editor.GotoPos(old_cursor_pos)

   432

                self.RefreshJumpList()

   433

                self.Editor.EmptyUndoBuffer()

   434

            self.DisableEvents = False

   435

   436

            self.RefreshVariableTree()

   437

   438

            self.TypeNames = [typename.upper() for typename in self.Controler.GetDataTypes(self.TagName, True, self.Debug)]

   439

            self.EnumeratedValues = [value.upper() for value in self.Controler.GetEnumeratedDataValues()]

   440

   441

            self.Functions = {}

   442

            for category in self.Controler.GetBlockTypes(self.TagName, self.Debug):

   443

                for blocktype in category["list"]:

   444

                    blockname = blocktype["name"].upper()

   445

                    if blocktype["type"] == "function" and blockname not in self.Keywords and blockname not in self.Variables.keys():

   446

                        interface = dict([(name, {}) for name, type, modifier in blocktype["inputs"] + blocktype["outputs"] if name != ''])

   447

                        for param in ["EN", "ENO"]:

   448

                            if not interface.has_key(param):

   449

                                interface[param] = {}

   450

                        if self.Functions.has_key(blockname):

   451

                            self.Functions[blockname]["interface"].update(interface)

   452

                            self.Functions[blockname]["extensible"] |= blocktype["extensible"]

   453

                        else:

   454

                            self.Functions[blockname] = {"interface": interface,

   455

                                                         "extensible": blocktype["extensible"]}

   456

   457

        self.Colourise(0, -1)

   458

   459

    def RefreshVariableTree(self):

   460

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

   461

        self.Variables = self.GenerateVariableTree([(variable["Name"], variable["Type"], variable["Tree"]) for variable in self.Controler.GetEditedElementInterfaceVars(self.TagName, self.Debug)])

   462

        if self.Controler.GetEditedElementType(self.TagName, self.Debug)[1] == "function" or words[0] == "T" and self.TextSyntax == "IL":

   463

            return_type = self.Controler.GetEditedElementInterfaceReturnType(self.TagName, self.Debug)

   464

            if return_type is not None:

   465

                var_tree, var_dimension = self.Controler.GenerateVarTree(return_type, self.Debug)

   466

                self.Variables[words[-1].upper()] = self.GenerateVariableTree(var_tree)

   467

            else:

   468

                self.Variables[words[-1].upper()] = {}

   469

   470

    def GenerateVariableTree(self, list):

   471

        tree = {}

   472

        for var_name, var_type, (var_tree, var_dimension) in list:

   473

            tree[var_name.upper()] = self.GenerateVariableTree(var_tree)

   474

        return tree

   475

   476

    def IsValidVariable(self, name, context):

   477

        return context is not None and context.get(name, None) is not None

   478

   479

    def IsCallParameter(self, name, call):

   480

        if call is not None:

   481

            return (call["interface"].get(name.upper(), None) is not None or 

   482

                    call["extensible"] and EXTENSIBLE_PARAMETER.match(name.upper()) is not None)

   483

        return False

   484

   485

    def RefreshLineFolding(self, line_number):

   486

        if self.TextSyntax in ["ST", "ALL"]:

   487

            level = wx.stc.STC_FOLDLEVELBASE + self.Editor.GetLineIndentation(line_number)

   488

            line = self.Editor.GetLine(line_number).strip()

   489

            if line == "":

   490

                if line_number > 0:

   491

                    if LineStartswith(self.Editor.GetLine(line_number - 1).strip(), self.BlockEndKeywords):

   492

                        level = self.Editor.GetFoldLevel(self.Editor.GetFoldParent(line_number - 1)) & wx.stc.STC_FOLDLEVELNUMBERMASK

   493

                    else:

   494

                        level = self.Editor.GetFoldLevel(line_number - 1) & wx.stc.STC_FOLDLEVELNUMBERMASK

   495

                if level != wx.stc.STC_FOLDLEVELBASE:

   496

                    level |=  wx.stc.STC_FOLDLEVELWHITEFLAG 

   497

            elif LineStartswith(line, self.BlockStartKeywords):

   498

                level |= wx.stc.STC_FOLDLEVELHEADERFLAG

   499

            elif LineStartswith(line, self.BlockEndKeywords):

   500

                if LineStartswith(self.Editor.GetLine(line_number - 1).strip(), self.BlockEndKeywords):

   501

                    level = self.Editor.GetFoldLevel(self.Editor.GetFoldParent(line_number - 1)) & wx.stc.STC_FOLDLEVELNUMBERMASK

   502

                else:

   503

                    level = self.Editor.GetFoldLevel(line_number - 1) & wx.stc.STC_FOLDLEVELNUMBERMASK

   504

            self.Editor.SetFoldLevel(line_number, level)

   505

   506

    def OnStyleNeeded(self, event):

   507

        self.TextChanged = True

   508

        line_number = self.Editor.LineFromPosition(self.Editor.GetEndStyled())

   509

        if line_number == 0:

   510

            start_pos = last_styled_pos = 0

   511

        else:

   512

            start_pos = last_styled_pos = self.Editor.GetLineEndPosition(line_number - 1) + 1

   513

        self.RefreshLineFolding(line_number)

   514

        end_pos = event.GetPosition()

   515

        self.StartStyling(start_pos, 0xff)

   516

   517

        current_context = self.Variables

   518

        current_call = None

   519

   520

        current_pos = last_styled_pos

   521

        state = SPACE

   522

        line = ""

   523

        word = ""

   524

        while current_pos < end_pos:

   525

            char = chr(self.Editor.GetCharAt(current_pos)).upper()

   526

            line += char

   527