Laurent@814: #!/usr/bin/env python Laurent@814: # -*- coding: utf-8 -*- Laurent@814: Laurent@814: #This file is part of PLCOpenEditor, a library implementing an IEC 61131-3 editor Laurent@814: #based on the plcopen standard. Laurent@814: # Laurent@814: #Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD Laurent@814: # Laurent@814: #See COPYING file for copyrights details. Laurent@814: # Laurent@814: #This library is free software; you can redistribute it and/or Laurent@814: #modify it under the terms of the GNU General Public Laurent@814: #License as published by the Free Software Foundation; either Laurent@814: #version 2.1 of the License, or (at your option) any later version. Laurent@814: # Laurent@814: #This library is distributed in the hope that it will be useful, Laurent@814: #but WITHOUT ANY WARRANTY; without even the implied warranty of Laurent@814: #MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Laurent@814: #General Public License for more details. Laurent@814: # Laurent@814: #You should have received a copy of the GNU General Public Laurent@814: #License along with this library; if not, write to the Free Software Laurent@814: #Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA Laurent@814: Laurent@814: import re Laurent@814: from types import * Laurent@814: Laurent@814: import wx Laurent@814: import wx.stc Laurent@814: Laurent@814: from graphics.GraphicCommons import ERROR_HIGHLIGHT, SEARCH_RESULT_HIGHLIGHT, REFRESH_HIGHLIGHT_PERIOD Laurent@814: from plcopen.structures import ST_BLOCK_START_KEYWORDS, ST_BLOCK_END_KEYWORDS, IEC_BLOCK_START_KEYWORDS, IEC_BLOCK_END_KEYWORDS, LOCATIONDATATYPES Laurent@814: from EditorPanel import EditorPanel Laurent@1091: from controls.CustomStyledTextCtrl import CustomStyledTextCtrl, faces, GetCursorPos Laurent@814: Laurent@814: #------------------------------------------------------------------------------- Laurent@814: # Textual programs Viewer class Laurent@814: #------------------------------------------------------------------------------- Laurent@814: Laurent@814: Laurent@814: NEWLINE = "\n" Laurent@814: NUMBERS = [str(i) for i in xrange(10)] Laurent@814: LETTERS = ['_'] Laurent@814: for i in xrange(26): Laurent@814: LETTERS.append(chr(ord('a') + i)) Laurent@814: LETTERS.append(chr(ord('A') + i)) Laurent@814: Laurent@814: [STC_PLC_WORD, STC_PLC_COMMENT, STC_PLC_NUMBER, STC_PLC_STRING, Laurent@814: STC_PLC_VARIABLE, STC_PLC_PARAMETER, STC_PLC_FUNCTION, STC_PLC_JUMP, Laurent@814: STC_PLC_ERROR, STC_PLC_SEARCH_RESULT] = range(10) Laurent@871: [SPACE, WORD, NUMBER, STRING, WSTRING, COMMENT, PRAGMA, DPRAGMA] = range(8) Laurent@814: Laurent@814: [ID_TEXTVIEWER, ID_TEXTVIEWERTEXTCTRL, Laurent@814: ] = [wx.NewId() for _init_ctrls in range(2)] Laurent@814: Laurent@814: re_texts = {} Laurent@814: re_texts["letter"] = "[A-Za-z]" Laurent@814: re_texts["digit"] = "[0-9]" Laurent@814: re_texts["identifier"] = "((?:%(letter)s|(?:_(?:%(letter)s|%(digit)s)))(?:_?(?:%(letter)s|%(digit)s))*)"%re_texts Laurent@814: IDENTIFIER_MODEL = re.compile(re_texts["identifier"]) Laurent@814: LABEL_MODEL = re.compile("[ \t\n]%(identifier)s:[ \t\n]"%re_texts) Laurent@814: EXTENSIBLE_PARAMETER = re.compile("IN[1-9][0-9]*$") Laurent@814: Laurent@814: HIGHLIGHT_TYPES = { Laurent@814: ERROR_HIGHLIGHT: STC_PLC_ERROR, Laurent@814: SEARCH_RESULT_HIGHLIGHT: STC_PLC_SEARCH_RESULT, Laurent@814: } Laurent@814: Laurent@814: def LineStartswith(line, symbols): Laurent@814: return reduce(lambda x, y: x or y, map(lambda x: line.startswith(x), symbols), False) Laurent@814: Laurent@814: class TextViewer(EditorPanel): Laurent@814: Laurent@814: ID = ID_TEXTVIEWER Laurent@814: Laurent@814: if wx.VERSION < (2, 6, 0): Laurent@814: def Bind(self, event, function, id = None): Laurent@814: if id is not None: Laurent@814: event(self, id, function) Laurent@814: else: Laurent@814: event(self, function) Laurent@814: Laurent@814: def _init_Editor(self, prnt): Laurent@1091: self.Editor = CustomStyledTextCtrl(id=ID_TEXTVIEWERTEXTCTRL, Laurent@814: parent=prnt, name="TextViewer", size=wx.Size(0, 0), style=0) Laurent@814: self.Editor.ParentWindow = self Laurent@814: Laurent@814: self.Editor.CmdKeyAssign(ord('+'), wx.stc.STC_SCMOD_CTRL, wx.stc.STC_CMD_ZOOMIN) Laurent@814: self.Editor.CmdKeyAssign(ord('-'), wx.stc.STC_SCMOD_CTRL, wx.stc.STC_CMD_ZOOMOUT) Laurent@814: Laurent@814: self.Editor.SetViewWhiteSpace(False) Laurent@814: Laurent@814: self.Editor.SetLexer(wx.stc.STC_LEX_CONTAINER) Laurent@814: Laurent@814: # Global default styles for all languages Laurent@814: self.Editor.StyleSetSpec(wx.stc.STC_STYLE_DEFAULT, "face:%(mono)s,size:%(size)d" % faces) Laurent@814: self.Editor.StyleClearAll() # Reset all to be like the default Laurent@814: Laurent@814: self.Editor.StyleSetSpec(wx.stc.STC_STYLE_LINENUMBER, "back:#C0C0C0,size:%(size)d" % faces) Laurent@814: self.Editor.SetSelBackground(1, "#E0E0E0") Laurent@814: Laurent@814: # Highlighting styles Laurent@814: self.Editor.StyleSetSpec(STC_PLC_WORD, "fore:#00007F,bold,size:%(size)d" % faces) Laurent@814: self.Editor.StyleSetSpec(STC_PLC_VARIABLE, "fore:#7F0000,size:%(size)d" % faces) Laurent@814: self.Editor.StyleSetSpec(STC_PLC_PARAMETER, "fore:#7F007F,size:%(size)d" % faces) Laurent@814: self.Editor.StyleSetSpec(STC_PLC_FUNCTION, "fore:#7F7F00,size:%(size)d" % faces) Laurent@814: self.Editor.StyleSetSpec(STC_PLC_COMMENT, "fore:#7F7F7F,size:%(size)d" % faces) Laurent@814: self.Editor.StyleSetSpec(STC_PLC_NUMBER, "fore:#007F7F,size:%(size)d" % faces) Laurent@814: self.Editor.StyleSetSpec(STC_PLC_STRING, "fore:#007F00,size:%(size)d" % faces) Laurent@814: self.Editor.StyleSetSpec(STC_PLC_JUMP, "fore:#FF7FFF,size:%(size)d" % faces) Laurent@814: self.Editor.StyleSetSpec(STC_PLC_ERROR, "fore:#FF0000,back:#FFFF00,size:%(size)d" % faces) Laurent@814: self.Editor.StyleSetSpec(STC_PLC_SEARCH_RESULT, "fore:#FFFFFF,back:#FFA500,size:%(size)d" % faces) Laurent@814: Laurent@814: # Indicators styles Laurent@814: self.Editor.IndicatorSetStyle(0, wx.stc.STC_INDIC_SQUIGGLE) Laurent@814: if self.ParentWindow is not None and self.Controler is not None: Laurent@814: self.Editor.IndicatorSetForeground(0, wx.RED) Laurent@814: else: Laurent@814: self.Editor.IndicatorSetForeground(0, wx.WHITE) Laurent@814: Laurent@814: # Line numbers in the margin Laurent@814: self.Editor.SetMarginType(1, wx.stc.STC_MARGIN_NUMBER) Laurent@814: self.Editor.SetMarginWidth(1, 50) Laurent@814: Laurent@814: # Folding Laurent@814: self.Editor.MarkerDefine(wx.stc.STC_MARKNUM_FOLDEROPEN, wx.stc.STC_MARK_BOXMINUS, "white", "#808080") Laurent@814: self.Editor.MarkerDefine(wx.stc.STC_MARKNUM_FOLDER, wx.stc.STC_MARK_BOXPLUS, "white", "#808080") Laurent@814: self.Editor.MarkerDefine(wx.stc.STC_MARKNUM_FOLDERSUB, wx.stc.STC_MARK_VLINE, "white", "#808080") Laurent@814: self.Editor.MarkerDefine(wx.stc.STC_MARKNUM_FOLDERTAIL, wx.stc.STC_MARK_LCORNER, "white", "#808080") Laurent@814: self.Editor.MarkerDefine(wx.stc.STC_MARKNUM_FOLDEREND, wx.stc.STC_MARK_BOXPLUSCONNECTED, "white", "#808080") Laurent@814: self.Editor.MarkerDefine(wx.stc.STC_MARKNUM_FOLDEROPENMID, wx.stc.STC_MARK_BOXMINUSCONNECTED, "white", "#808080") Laurent@814: self.Editor.MarkerDefine(wx.stc.STC_MARKNUM_FOLDERMIDTAIL, wx.stc.STC_MARK_TCORNER, "white", "#808080") Laurent@814: Laurent@814: # Indentation size Laurent@814: self.Editor.SetTabWidth(2) Laurent@814: self.Editor.SetUseTabs(0) Laurent@814: Laurent@814: self.Editor.SetModEventMask(wx.stc.STC_MOD_BEFOREINSERT| Laurent@814: wx.stc.STC_MOD_BEFOREDELETE| Laurent@814: wx.stc.STC_PERFORMED_USER) Laurent@814: Laurent@814: self.Bind(wx.stc.EVT_STC_STYLENEEDED, self.OnStyleNeeded, id=ID_TEXTVIEWERTEXTCTRL) Laurent@814: self.Editor.Bind(wx.stc.EVT_STC_MARGINCLICK, self.OnMarginClick) Laurent@1126: self.Editor.Bind(wx.stc.EVT_STC_UPDATEUI, self.OnUpdateUI) Laurent@814: self.Editor.Bind(wx.EVT_KEY_DOWN, self.OnKeyDown) Laurent@814: if self.Controler is not None: Laurent@814: self.Editor.Bind(wx.EVT_KILL_FOCUS, self.OnKillFocus) Laurent@814: self.Bind(wx.stc.EVT_STC_DO_DROP, self.OnDoDrop, id=ID_TEXTVIEWERTEXTCTRL) Laurent@814: self.Bind(wx.stc.EVT_STC_MODIFIED, self.OnModification, id=ID_TEXTVIEWERTEXTCTRL) Laurent@814: Laurent@814: def __init__(self, parent, tagname, window, controler, debug = False, instancepath = ""): Laurent@814: if tagname != "" and controler is not None: Laurent@814: self.VARIABLE_PANEL_TYPE = controler.GetPouType(tagname.split("::")[1]) Laurent@814: Laurent@814: EditorPanel.__init__(self, parent, tagname, window, controler, debug) Laurent@814: Laurent@814: self.Keywords = [] Laurent@814: self.Variables = {} Laurent@814: self.Functions = {} Laurent@814: self.TypeNames = [] Laurent@814: self.Jumps = [] Laurent@814: self.EnumeratedValues = [] Laurent@814: self.DisableEvents = True Laurent@814: self.TextSyntax = None Laurent@814: self.CurrentAction = None Laurent@1178: Laurent@814: self.InstancePath = instancepath Laurent@814: self.ContextStack = [] Laurent@814: self.CallStack = [] Laurent@814: Laurent@1178: self.ResetSearchResults() Laurent@1178: Laurent@814: self.RefreshHighlightsTimer = wx.Timer(self, -1) Laurent@814: self.Bind(wx.EVT_TIMER, self.OnRefreshHighlightsTimer, self.RefreshHighlightsTimer) Laurent@814: Laurent@814: def __del__(self): Laurent@814: self.RefreshHighlightsTimer.Stop() Laurent@814: Laurent@814: def GetTitle(self): Laurent@814: if self.Debug or self.TagName == "": Laurent@814: if len(self.InstancePath) > 15: Laurent@814: return "..." + self.InstancePath[-12:] Laurent@814: return self.InstancePath Laurent@814: return EditorPanel.GetTitle(self) Laurent@814: Laurent@814: def GetInstancePath(self): Laurent@814: return self.InstancePath Laurent@814: Laurent@814: def IsViewing(self, tagname): Laurent@814: if self.Debug or self.TagName == "": Laurent@814: return self.InstancePath == tagname Laurent@814: else: Laurent@814: return self.TagName == tagname Laurent@814: Laurent@814: def GetText(self): Laurent@814: return self.Editor.GetText() Laurent@814: Laurent@814: def SetText(self, text): Laurent@814: self.Editor.SetText(text) Laurent@814: Laurent@814: def SelectAll(self): Laurent@814: self.Editor.SelectAll() Laurent@814: Laurent@814: def Colourise(self, start, end): Laurent@814: self.Editor.Colourise(start, end) Laurent@814: Laurent@814: def StartStyling(self, pos, mask): Laurent@814: self.Editor.StartStyling(pos, mask) Laurent@814: Laurent@814: def SetStyling(self, length, style): Laurent@814: self.Editor.SetStyling(length, style) Laurent@814: Laurent@814: def GetCurrentPos(self): Laurent@814: return self.Editor.GetCurrentPos() Laurent@814: Laurent@1178: def ResetSearchResults(self): Laurent@1178: self.Highlights = [] Laurent@1178: self.SearchParams = None Laurent@1178: self.SearchResults = None Laurent@1178: self.CurrentFindHighlight = None Laurent@1178: Laurent@814: def OnModification(self, event): Laurent@814: if not self.DisableEvents: Laurent@814: mod_type = event.GetModificationType() Laurent@814: if mod_type&wx.stc.STC_MOD_BEFOREINSERT: Laurent@814: if self.CurrentAction == None: Laurent@814: self.StartBuffering() Laurent@814: elif self.CurrentAction[0] != "Add" or self.CurrentAction[1] != event.GetPosition() - 1: Laurent@814: self.Controler.EndBuffering() Laurent@814: self.StartBuffering() Laurent@814: self.CurrentAction = ("Add", event.GetPosition()) Laurent@814: wx.CallAfter(self.RefreshModel) Laurent@814: elif mod_type&wx.stc.STC_MOD_BEFOREDELETE: Laurent@814: if self.CurrentAction == None: Laurent@814: self.StartBuffering() Laurent@814: elif self.CurrentAction[0] != "Delete" or self.CurrentAction[1] != event.GetPosition() + 1: Laurent@814: self.Controler.EndBuffering() Laurent@814: self.StartBuffering() Laurent@814: self.CurrentAction = ("Delete", event.GetPosition()) Laurent@814: wx.CallAfter(self.RefreshModel) Laurent@814: event.Skip() Laurent@814: Laurent@814: def OnDoDrop(self, event): Laurent@814: try: Laurent@814: values = eval(event.GetDragText()) Laurent@814: except: Laurent@814: values = event.GetDragText() Laurent@814: if isinstance(values, tuple): Laurent@814: message = None Laurent@814: if values[1] in ["program", "debug"]: Laurent@814: event.SetDragText("") Laurent@814: elif values[1] in ["functionBlock", "function"]: Laurent@814: blocktype = values[0] Laurent@814: blockname = values[2] Laurent@814: if len(values) > 3: Laurent@814: blockinputs = values[3] Laurent@814: else: Laurent@814: blockinputs = None Laurent@814: if values[1] != "function": Laurent@814: if blockname == "": Laurent@814: dialog = wx.TextEntryDialog(self.ParentWindow, "Block name", "Please enter a block name", "", wx.OK|wx.CANCEL|wx.CENTRE) Laurent@814: if dialog.ShowModal() == wx.ID_OK: Laurent@814: blockname = dialog.GetValue() Laurent@814: else: Laurent@814: return Laurent@814: dialog.Destroy() Laurent@814: if blockname.upper() in [name.upper() for name in self.Controler.GetProjectPouNames(self.Debug)]: Laurent@814: message = _("\"%s\" pou already exists!")%blockname Laurent@814: elif blockname.upper() in [name.upper() for name in self.Controler.GetEditedElementVariables(self.TagName, self.Debug)]: Laurent@814: message = _("\"%s\" element for this pou already exists!")%blockname Laurent@814: else: Laurent@814: self.Controler.AddEditedElementPouVar(self.TagName, values[0], blockname) Laurent@814: self.RefreshVariablePanel() Laurent@814: self.RefreshVariableTree() Laurent@814: blockinfo = self.Controler.GetBlockType(blocktype, blockinputs, self.Debug) Laurent@814: hint = ',\n '.join( Laurent@814: [ " " + fctdecl[0]+" := (*"+fctdecl[1]+"*)" for fctdecl in blockinfo["inputs"]] + Laurent@814: [ " " + fctdecl[0]+" => (*"+fctdecl[1]+"*)" for fctdecl in blockinfo["outputs"]]) Laurent@814: if values[1] == "function": Laurent@814: event.SetDragText(blocktype+"(\n "+hint+")") Laurent@814: else: Laurent@814: event.SetDragText(blockname+"(\n "+hint+")") Laurent@814: elif values[1] == "location": Laurent@814: pou_name, pou_type = self.Controler.GetEditedElementType(self.TagName, self.Debug) Laurent@814: if len(values) > 2 and pou_type == "program": Laurent@814: var_name = values[3] Laurent@814: if var_name.upper() in [name.upper() for name in self.Controler.GetProjectPouNames(self.Debug)]: Laurent@814: message = _("\"%s\" pou already exists!")%var_name Laurent@814: elif var_name.upper() in [name.upper() for name in self.Controler.GetEditedElementVariables(self.TagName, self.Debug)]: Laurent@814: message = _("\"%s\" element for this pou already exists!")%var_name Laurent@814: else: Laurent@814: location = values[0] Laurent@814: if not location.startswith("%"): Laurent@814: dialog = wx.SingleChoiceDialog(self.ParentWindow, Laurent@814: _("Select a variable class:"), _("Variable class"), Laurent@814: ["Input", "Output", "Memory"], Laurent@814: wx.DEFAULT_DIALOG_STYLE|wx.OK|wx.CANCEL) Laurent@814: if dialog.ShowModal() == wx.ID_OK: Laurent@814: selected = dialog.GetSelection() Laurent@814: else: Laurent@814: selected = None Laurent@814: dialog.Destroy() Laurent@814: if selected is None: Laurent@814: event.SetDragText("") Laurent@814: return Laurent@814: if selected == 0: Laurent@814: location = "%I" + location Laurent@814: elif selected == 1: Laurent@814: location = "%Q" + location Laurent@814: else: Laurent@814: location = "%M" + location Laurent@814: if values[2] is not None: Laurent@814: var_type = values[2] Laurent@814: else: Laurent@814: var_type = LOCATIONDATATYPES.get(location[2], ["BOOL"])[0] Laurent@814: self.Controler.AddEditedElementPouVar(self.TagName, var_type, var_name, location, values[4]) Laurent@814: self.RefreshVariablePanel() Laurent@814: self.RefreshVariableTree() Laurent@814: event.SetDragText(var_name) Laurent@814: else: Laurent@814: event.SetDragText("") Laurent@814: elif values[1] == "Global": Laurent@814: var_name = values[0] Laurent@814: if var_name.upper() in [name.upper() for name in self.Controler.GetProjectPouNames(self.Debug)]: Laurent@814: message = _("\"%s\" pou already exists!")%var_name Laurent@814: else: Laurent@814: if not var_name.upper() in [name.upper() for name in self.Controler.GetEditedElementVariables(self.TagName, self.Debug)]: Laurent@814: self.Controler.AddEditedElementPouExternalVar(self.TagName, values[2], var_name) Laurent@814: self.RefreshVariablePanel() Laurent@814: self.RefreshVariableTree() Laurent@814: event.SetDragText(var_name) Laurent@814: elif values[1] == "Constant": Laurent@814: event.SetDragText(values[0]) Laurent@814: elif values[3] == self.TagName: Laurent@814: self.ResetBuffer() Laurent@814: event.SetDragText(values[0]) Laurent@814: wx.CallAfter(self.RefreshModel) Laurent@814: else: Laurent@814: message = _("Variable don't belong to this POU!") Laurent@814: if message is not None: Laurent@814: dialog = wx.MessageDialog(self, message, _("Error"), wx.OK|wx.ICON_ERROR) Laurent@814: dialog.ShowModal() Laurent@814: dialog.Destroy() Laurent@814: event.SetDragText("") Laurent@814: event.Skip() Laurent@814: Laurent@814: def SetTextSyntax(self, syntax): Laurent@814: self.TextSyntax = syntax Laurent@814: if syntax in ["ST", "ALL"]: Laurent@814: self.Editor.SetMarginType(2, wx.stc.STC_MARGIN_SYMBOL) Laurent@814: self.Editor.SetMarginMask(2, wx.stc.STC_MASK_FOLDERS) Laurent@814: self.Editor.SetMarginSensitive(2, 1) Laurent@814: self.Editor.SetMarginWidth(2, 12) Laurent@814: if syntax == "ST": Laurent@814: self.BlockStartKeywords = ST_BLOCK_START_KEYWORDS Laurent@814: self.BlockEndKeywords = ST_BLOCK_START_KEYWORDS Laurent@814: else: Laurent@814: self.BlockStartKeywords = IEC_BLOCK_START_KEYWORDS Laurent@814: self.BlockEndKeywords = IEC_BLOCK_START_KEYWORDS Laurent@814: else: Laurent@814: self.BlockStartKeywords = [] Laurent@814: self.BlockEndKeywords = [] Laurent@814: Laurent@814: def SetKeywords(self, keywords): Laurent@814: self.Keywords = [keyword.upper() for keyword in keywords] Laurent@814: self.Colourise(0, -1) Laurent@814: Laurent@814: def RefreshJumpList(self): Laurent@814: if self.TextSyntax != "IL": Laurent@814: self.Jumps = [jump.upper() for jump in LABEL_MODEL.findall(self.GetText())] Laurent@814: self.Colourise(0, -1) Laurent@814: Laurent@814: # Buffer the last model state Laurent@814: def RefreshBuffer(self): Laurent@814: self.Controler.BufferProject() Laurent@814: if self.ParentWindow: Laurent@814: self.ParentWindow.RefreshTitle() Laurent@814: self.ParentWindow.RefreshFileMenu() Laurent@814: self.ParentWindow.RefreshEditMenu() Laurent@814: Laurent@814: def StartBuffering(self): Laurent@814: self.Controler.StartBuffering() Laurent@814: if self.ParentWindow: Laurent@814: self.ParentWindow.RefreshTitle() Laurent@814: self.ParentWindow.RefreshFileMenu() Laurent@814: self.ParentWindow.RefreshEditMenu() Laurent@814: Laurent@814: def ResetBuffer(self): Laurent@814: if self.CurrentAction != None: Laurent@814: self.Controler.EndBuffering() Laurent@814: self.CurrentAction = None Laurent@814: Laurent@814: def GetBufferState(self): Laurent@814: if not self.Debug and self.TextSyntax != "ALL": Laurent@814: return self.Controler.GetBufferState() Laurent@814: return False, False Laurent@814: Laurent@814: def Undo(self): Laurent@814: if not self.Debug and self.TextSyntax != "ALL": Laurent@814: self.Controler.LoadPrevious() Laurent@814: self.ParentWindow.CloseTabsWithoutModel() Laurent@814: Laurent@814: def Redo(self): Laurent@814: if not self.Debug and self.TextSyntax != "ALL": Laurent@814: self.Controler.LoadNext() Laurent@814: self.ParentWindow.CloseTabsWithoutModel() Laurent@814: Laurent@814: def HasNoModel(self): Laurent@814: if not self.Debug and self.TextSyntax != "ALL": Laurent@814: return self.Controler.GetEditedElement(self.TagName) is None Laurent@814: return False Laurent@814: Laurent@814: def RefreshView(self, variablepanel=True): Laurent@814: EditorPanel.RefreshView(self, variablepanel) Laurent@814: Laurent@814: if self.Controler is not None: Laurent@814: self.ResetBuffer() Laurent@814: self.DisableEvents = True Laurent@814: old_cursor_pos = self.GetCurrentPos() Laurent@1060: line = self.Editor.GetFirstVisibleLine() Laurent@1060: column = self.Editor.GetXOffset() Laurent@814: old_text = self.GetText() Laurent@814: new_text = self.Controler.GetEditedElementText(self.TagName, self.Debug) Laurent@1060: if old_text != new_text: Laurent@1060: self.SetText(new_text) Laurent@1060: new_cursor_pos = GetCursorPos(old_text, new_text) Laurent@1060: self.Editor.LineScroll(column, line) Laurent@1060: if new_cursor_pos != None: Laurent@1060: self.Editor.GotoPos(new_cursor_pos) Laurent@1060: else: Laurent@1060: self.Editor.GotoPos(old_cursor_pos) Laurent@1060: self.RefreshJumpList() Laurent@1060: self.Editor.EmptyUndoBuffer() Laurent@814: self.DisableEvents = False Laurent@814: Laurent@814: self.RefreshVariableTree() Laurent@814: Laurent@814: self.TypeNames = [typename.upper() for typename in self.Controler.GetDataTypes(self.TagName, True, self.Debug)] Laurent@814: self.EnumeratedValues = [value.upper() for value in self.Controler.GetEnumeratedDataValues()] Laurent@814: Laurent@814: self.Functions = {} Laurent@814: for category in self.Controler.GetBlockTypes(self.TagName, self.Debug): Laurent@814: for blocktype in category["list"]: Laurent@814: blockname = blocktype["name"].upper() Laurent@814: if blocktype["type"] == "function" and blockname not in self.Keywords and blockname not in self.Variables.keys(): Laurent@814: interface = dict([(name, {}) for name, type, modifier in blocktype["inputs"] + blocktype["outputs"] if name != '']) Laurent@814: for param in ["EN", "ENO"]: Laurent@814: if not interface.has_key(param): Laurent@814: interface[param] = {} Laurent@814: if self.Functions.has_key(blockname): Laurent@814: self.Functions[blockname]["interface"].update(interface) Laurent@814: self.Functions[blockname]["extensible"] |= blocktype["extensible"] Laurent@814: else: Laurent@814: self.Functions[blockname] = {"interface": interface, Laurent@814: "extensible": blocktype["extensible"]} Laurent@814: Laurent@814: self.Colourise(0, -1) Laurent@814: Laurent@814: def RefreshVariableTree(self): Laurent@814: words = self.TagName.split("::") Laurent@814: self.Variables = self.GenerateVariableTree([(variable["Name"], variable["Type"], variable["Tree"]) for variable in self.Controler.GetEditedElementInterfaceVars(self.TagName, self.Debug)]) Laurent@814: if self.Controler.GetEditedElementType(self.TagName, self.Debug)[1] == "function" or words[0] == "T" and self.TextSyntax == "IL": Laurent@814: return_type = self.Controler.GetEditedElementInterfaceReturnType(self.TagName, self.Debug) Laurent@814: if return_type is not None: Laurent@814: var_tree, var_dimension = self.Controler.GenerateVarTree(return_type, self.Debug) Laurent@814: self.Variables[words[-1].upper()] = self.GenerateVariableTree(var_tree) Laurent@814: else: Laurent@814: self.Variables[words[-1].upper()] = {} Laurent@814: Laurent@814: def GenerateVariableTree(self, list): Laurent@814: tree = {} Laurent@814: for var_name, var_type, (var_tree, var_dimension) in list: Laurent@814: tree[var_name.upper()] = self.GenerateVariableTree(var_tree) Laurent@814: return tree Laurent@814: Laurent@814: def IsValidVariable(self, name, context): Laurent@814: return context is not None and context.get(name, None) is not None Laurent@814: Laurent@814: def IsCallParameter(self, name, call): Laurent@814: if call is not None: Laurent@814: return (call["interface"].get(name.upper(), None) is not None or Laurent@814: call["extensible"] and EXTENSIBLE_PARAMETER.match(name.upper()) is not None) Laurent@814: return False Laurent@814: Laurent@814: def RefreshLineFolding(self, line_number): Laurent@814: if self.TextSyntax in ["ST", "ALL"]: Laurent@814: level = wx.stc.STC_FOLDLEVELBASE + self.Editor.GetLineIndentation(line_number) Laurent@814: line = self.Editor.GetLine(line_number).strip() Laurent@814: if line == "": Laurent@814: if line_number > 0: Laurent@814: if LineStartswith(self.Editor.GetLine(line_number - 1).strip(), self.BlockEndKeywords): Laurent@814: level = self.Editor.GetFoldLevel(self.Editor.GetFoldParent(line_number - 1)) & wx.stc.STC_FOLDLEVELNUMBERMASK Laurent@814: else: Laurent@814: level = self.Editor.GetFoldLevel(line_number - 1) & wx.stc.STC_FOLDLEVELNUMBERMASK Laurent@814: if level != wx.stc.STC_FOLDLEVELBASE: Laurent@814: level |= wx.stc.STC_FOLDLEVELWHITEFLAG Laurent@814: elif LineStartswith(line, self.BlockStartKeywords): Laurent@814: level |= wx.stc.STC_FOLDLEVELHEADERFLAG Laurent@814: elif LineStartswith(line, self.BlockEndKeywords): Laurent@814: if LineStartswith(self.Editor.GetLine(line_number - 1).strip(), self.BlockEndKeywords): Laurent@814: level = self.Editor.GetFoldLevel(self.Editor.GetFoldParent(line_number - 1)) & wx.stc.STC_FOLDLEVELNUMBERMASK Laurent@814: else: Laurent@814: level = self.Editor.GetFoldLevel(line_number - 1) & wx.stc.STC_FOLDLEVELNUMBERMASK Laurent@814: self.Editor.SetFoldLevel(line_number, level) Laurent@814: Laurent@814: def OnStyleNeeded(self, event): Laurent@814: self.TextChanged = True Laurent@814: line_number = self.Editor.LineFromPosition(self.Editor.GetEndStyled()) Laurent@814: if line_number == 0: Laurent@814: start_pos = last_styled_pos = 0 Laurent@814: else: Laurent@814: start_pos = last_styled_pos = self.Editor.GetLineEndPosition(line_number - 1) + 1 Laurent@814: self.RefreshLineFolding(line_number) Laurent@814: end_pos = event.GetPosition() Laurent@814: self.StartStyling(start_pos, 0xff) Laurent@814: Laurent@814: current_context = self.Variables Laurent@814: current_call = None Laurent@814: Laurent@814: current_pos = last_styled_pos Laurent@814: state = SPACE Laurent@814: line = "" Laurent@814: word = "" Laurent@814: while current_pos < end_pos: Laurent@814: char = chr(self.Editor.GetCharAt(current_pos)).upper() Laurent@814: line += char Laurent@814: if char == NEWLINE: Laurent@814: self.ContextStack = [] Laurent@814: current_context = self.Variables Laurent@814: if state == COMMENT: Laurent@814: self.SetStyling(current_pos - last_styled_pos + 1, STC_PLC_COMMENT) Laurent@814: elif state == NUMBER: Laurent@814: self.SetStyling(current_pos - last_styled_pos, STC_PLC_NUMBER) Laurent@814: elif state == WORD: Laurent@814: if word in self.Keywords or word in self.TypeNames: Laurent@814: self.SetStyling(current_pos - last_styled_pos, STC_PLC_WORD) Laurent@814: elif self.IsValidVariable(word, current_context): Laurent@814: self.SetStyling(current_pos - last_styled_pos, STC_PLC_VARIABLE) Laurent@814: elif self.IsCallParameter(word, current_call): Laurent@814: self.SetStyling(current_pos - last_styled_pos, STC_PLC_PARAMETER) Laurent@814: elif word in self.Functions: Laurent@814: self.SetStyling(current_pos - last_styled_pos, STC_PLC_FUNCTION) Laurent@814: elif self.TextSyntax == "IL" and word in self.Jumps: Laurent@814: self.SetStyling(current_pos - last_styled_pos, STC_PLC_JUMP) Laurent@814: elif word in self.EnumeratedValues: Laurent@814: self.SetStyling(current_pos - last_styled_pos, STC_PLC_NUMBER) Laurent@814: else: Laurent@814: self.SetStyling(current_pos - last_styled_pos, 31) Laurent@814: if word not in ["]", ")"] and (self.GetCurrentPos() < last_styled_pos or self.GetCurrentPos() > current_pos): Laurent@814: self.StartStyling(last_styled_pos, wx.stc.STC_INDICS_MASK) Laurent@814: self.SetStyling(current_pos - last_styled_pos, wx.stc.STC_INDIC0_MASK) Laurent@814: self.StartStyling(current_pos, 0xff) Laurent@814: else: Laurent@814: self.SetStyling(current_pos - last_styled_pos, 31) Laurent@814: last_styled_pos = current_pos Laurent@1108: if state != DPRAGMA: Laurent@1108: state = SPACE Laurent@814: line = "" Laurent@814: line_number += 1 Laurent@814: self.RefreshLineFolding(line_number) Laurent@814: elif line.endswith("(*") and state != COMMENT: Laurent@814: self.SetStyling(current_pos - last_styled_pos - 1, 31) Laurent@814: last_styled_pos = current_pos Laurent@814: if state == WORD: Laurent@814: current_context = self.Variables Laurent@814: state = COMMENT Laurent@871: elif line.endswith("{") and state not in [PRAGMA, DPRAGMA]: Laurent@858: self.SetStyling(current_pos - last_styled_pos, 31) Laurent@814: last_styled_pos = current_pos Laurent@814: if state == WORD: Laurent@814: current_context = self.Variables Laurent@814: state = PRAGMA Laurent@871: elif line.endswith("{{") and state == PRAGMA: Laurent@871: self.SetStyling(current_pos - last_styled_pos, 31) Laurent@871: last_styled_pos = current_pos Laurent@871: state = DPRAGMA Laurent@814: elif state == COMMENT: Laurent@814: if line.endswith("*)"): Laurent@814: self.SetStyling(current_pos - last_styled_pos + 2, STC_PLC_COMMENT) Laurent@814: last_styled_pos = current_pos + 1 Laurent@814: state = SPACE Laurent@814: elif state == PRAGMA: Laurent@814: if line.endswith("}"): Laurent@871: self.SetStyling(current_pos - last_styled_pos, 31) Laurent@871: last_styled_pos = current_pos Laurent@871: state = SPACE Laurent@871: elif state == DPRAGMA: Laurent@871: if line.endswith("}}"): Laurent@1108: self.SetStyling(current_pos - last_styled_pos + 1, 31) Laurent@814: last_styled_pos = current_pos + 1 Laurent@814: state = SPACE Laurent@814: elif (line.endswith("'") or line.endswith('"')) and state not in [STRING, WSTRING]: Laurent@814: self.SetStyling(current_pos - last_styled_pos, 31) Laurent@814: last_styled_pos = current_pos Laurent@814: if state == WORD: Laurent@814: current_context = self.Variables Laurent@814: if line.endswith("'"): Laurent@814: state = STRING Laurent@814: else: Laurent@814: state = WSTRING Laurent@814: elif state == STRING: Laurent@814: if line.endswith("'") and not line.endswith("$'"): Laurent@814: self.SetStyling(current_pos - last_styled_pos + 1, STC_PLC_STRING) Laurent@814: last_styled_pos = current_pos + 1 Laurent@814: state = SPACE Laurent@814: elif state == WSTRING: Laurent@814: if line.endswith('"') and not line.endswith('$"'): Laurent@814: self.SetStyling(current_pos - last_styled_pos + 1, STC_PLC_STRING) Laurent@814: last_styled_pos = current_pos + 1 Laurent@814: state = SPACE Laurent@814: elif char in LETTERS: Laurent@814: if state == NUMBER: Laurent@814: word = "#" Laurent@814: state = WORD Laurent@814: elif state == SPACE: Laurent@814: self.SetStyling(current_pos - last_styled_pos, 31) Laurent@814: word = char Laurent@814: last_styled_pos = current_pos Laurent@814: state = WORD Laurent@814: else: Laurent@814: word += char Laurent@814: elif char in NUMBERS or char == '.' and state != WORD: Laurent@814: if state == SPACE: Laurent@814: self.SetStyling(current_pos - last_styled_pos, 31) Laurent@814: last_styled_pos = current_pos Laurent@814: state = NUMBER Laurent@814: elif state == WORD and char != '.': Laurent@814: word += char Laurent@814: elif char == '(' and state == SPACE: Laurent@814: self.CallStack.append(current_call) Laurent@814: current_call = None Laurent@814: else: Laurent@814: if state == WORD: Laurent@814: if word in self.Keywords or word in self.TypeNames: Laurent@814: self.SetStyling(current_pos - last_styled_pos, STC_PLC_WORD) Laurent@814: elif self.IsValidVariable(word, current_context): Laurent@814: self.SetStyling(current_pos - last_styled_pos, STC_PLC_VARIABLE) Laurent@814: elif self.IsCallParameter(word, current_call): Laurent@814: self.SetStyling(current_pos - last_styled_pos, STC_PLC_PARAMETER) Laurent@814: elif word in self.Functions: Laurent@814: self.SetStyling(current_pos - last_styled_pos, STC_PLC_FUNCTION) Laurent@814: elif self.TextSyntax == "IL" and word in self.Jumps: Laurent@814: self.SetStyling(current_pos - last_styled_pos, STC_PLC_JUMP) Laurent@814: elif word in self.EnumeratedValues: Laurent@814: self.SetStyling(current_pos - last_styled_pos, STC_PLC_NUMBER) Laurent@814: else: Laurent@814: self.SetStyling(current_pos - last_styled_pos, 31) Laurent@814: if word not in ["]", ")"] and (self.GetCurrentPos() < last_styled_pos or self.GetCurrentPos() > current_pos): Laurent@814: self.StartStyling(last_styled_pos, wx.stc.STC_INDICS_MASK) Laurent@814: self.SetStyling(current_pos - last_styled_pos, wx.stc.STC_INDIC0_MASK) Laurent@814: self.StartStyling(current_pos, 0xff) Laurent@814: if char == '.': Laurent@814: if word != "]": Laurent@814: if current_context is not None: Laurent@814: current_context = current_context.get(word, None) Laurent@814: else: Laurent@814: current_context = None Laurent@814: elif char == '(': Laurent@814: self.CallStack.append(current_call) Laurent@814: current_call = self.Functions.get(word, None) Laurent@814: if current_call is None and self.IsValidVariable(word, current_context): Laurent@814: current_call = {"interface": current_context.get(word, {}), Laurent@814: "extensible": False} Laurent@814: current_context = self.Variables Laurent@814: else: Laurent@894: if char == '[' and current_context is not None: Laurent@814: self.ContextStack.append(current_context.get(word, None)) Laurent@814: current_context = self.Variables Laurent@814: Laurent@814: word = "" Laurent@814: last_styled_pos = current_pos Laurent@814: state = SPACE Laurent@814: elif state == NUMBER: Laurent@814: self.SetStyling(current_pos - last_styled_pos, STC_PLC_NUMBER) Laurent@814: last_styled_pos = current_pos Laurent@814: state = SPACE Laurent@814: if char == ']': Laurent@814: if len(self.ContextStack) > 0: Laurent@814: current_context = self.ContextStack.pop() Laurent@814: else: Laurent@814: current_context = self.Variables Laurent@814: word = char Laurent@814: state = WORD Laurent@814: elif char == ')': Laurent@814: current_context = self.Variables Laurent@814: if len(self.CallStack) > 0: Laurent@814: current_call = self.CallStack.pop() Laurent@814: else: Laurent@814: current_call = None Laurent@814: word = char Laurent@814: state = WORD Laurent@814: current_pos += 1 Laurent@814: if state == COMMENT: Laurent@814: self.SetStyling(current_pos - last_styled_pos + 2, STC_PLC_COMMENT) Laurent@814: elif state == NUMBER: Laurent@814: self.SetStyling(current_pos - last_styled_pos, STC_PLC_NUMBER) Laurent@814: elif state == WORD: Laurent@814: if word in self.Keywords or word in self.TypeNames: Laurent@814: self.SetStyling(current_pos - last_styled_pos, STC_PLC_WORD) Laurent@814: elif self.IsValidVariable(word, current_context): Laurent@814: self.SetStyling(current_pos - last_styled_pos, STC_PLC_VARIABLE) Laurent@814: elif self.IsCallParameter(word, current_call): Laurent@814: self.SetStyling(current_pos - last_styled_pos, STC_PLC_PARAMETER) Laurent@814: elif self.TextSyntax == "IL" and word in self.Functions: Laurent@814: self.SetStyling(current_pos - last_styled_pos, STC_PLC_FUNCTION) Laurent@814: elif word in self.Jumps: Laurent@814: self.SetStyling(current_pos - last_styled_pos, STC_PLC_JUMP) Laurent@814: elif word in self.EnumeratedValues: Laurent@814: self.SetStyling(current_pos - last_styled_pos, STC_PLC_NUMBER) Laurent@814: else: Laurent@814: self.SetStyling(current_pos - last_styled_pos, 31) Laurent@814: else: Laurent@814: self.SetStyling(current_pos - start_pos, 31) Laurent@814: self.ShowHighlights(start_pos, end_pos) Laurent@814: event.Skip() Laurent@814: Laurent@814: def OnMarginClick(self, event): Laurent@814: if event.GetMargin() == 2: Laurent@814: line = self.Editor.LineFromPosition(event.GetPosition()) Laurent@814: if self.Editor.GetFoldLevel(line) & wx.stc.STC_FOLDLEVELHEADERFLAG: Laurent@814: self.Editor.ToggleFold(line) Laurent@814: event.Skip() Laurent@1126: Laurent@1126: def OnUpdateUI(self, event): Laurent@1126: if self.ParentWindow: Laurent@1126: self.ParentWindow.SetCopyBuffer(self.Editor.GetSelectedText(), True) Laurent@1126: event.Skip() Laurent@1126: Laurent@814: def Cut(self): Laurent@814: self.ResetBuffer() Laurent@814: self.DisableEvents = True Laurent@814: self.Editor.CmdKeyExecute(wx.stc.STC_CMD_CUT) Laurent@814: self.DisableEvents = False Laurent@814: self.RefreshModel() Laurent@814: self.RefreshBuffer() Laurent@814: Laurent@814: def Copy(self): Laurent@814: self.Editor.CmdKeyExecute(wx.stc.STC_CMD_COPY) Laurent@1126: if self.ParentWindow: Laurent@1126: self.ParentWindow.RefreshEditMenu() Laurent@814: Laurent@814: def Paste(self): Laurent@814: self.ResetBuffer() Laurent@814: self.DisableEvents = True Laurent@814: self.Editor.CmdKeyExecute(wx.stc.STC_CMD_PASTE) Laurent@814: self.DisableEvents = False Laurent@814: self.RefreshModel() Laurent@814: self.RefreshBuffer() Laurent@814: Laurent@1057: def Search(self, criteria): Laurent@1057: return self.Controler.SearchInPou(self.TagName, criteria, self.Debug) Laurent@1057: Laurent@814: def Find(self, direction, search_params): Laurent@814: if self.SearchParams != search_params: Laurent@814: self.ClearHighlights(SEARCH_RESULT_HIGHLIGHT) Laurent@814: Laurent@814: self.SearchParams = search_params Laurent@814: criteria = { Laurent@814: "raw_pattern": search_params["find_pattern"], Laurent@814: "pattern": re.compile(search_params["find_pattern"]), Laurent@814: "case_sensitive": search_params["case_sensitive"], Laurent@814: "regular_expression": search_params["regular_expression"], Laurent@814: "filter": "all"} Laurent@814: Laurent@814: self.SearchResults = [ Laurent@814: (infos[1:], start, end, SEARCH_RESULT_HIGHLIGHT) Laurent@814: for infos, start, end, text in Laurent@1057: self.Search(criteria)] Laurent@1057: self.CurrentFindHighlight = None Laurent@814: Laurent@814: if len(self.SearchResults) > 0: Laurent@814: if self.CurrentFindHighlight is not None: Laurent@814: old_idx = self.SearchResults.index(self.CurrentFindHighlight) Laurent@814: if self.SearchParams["wrap"]: Laurent@814: idx = (old_idx + direction) % len(self.SearchResults) Laurent@814: else: Laurent@814: idx = max(0, min(old_idx + direction, len(self.SearchResults) - 1)) Laurent@814: if idx != old_idx: Laurent@814: self.RemoveHighlight(*self.CurrentFindHighlight) Laurent@814: self.CurrentFindHighlight = self.SearchResults[idx] Laurent@814: self.AddHighlight(*self.CurrentFindHighlight) Laurent@814: else: Laurent@814: self.CurrentFindHighlight = self.SearchResults[0] Laurent@814: self.AddHighlight(*self.CurrentFindHighlight) Laurent@814: Laurent@814: else: Laurent@814: if self.CurrentFindHighlight is not None: Laurent@814: self.RemoveHighlight(*self.CurrentFindHighlight) Laurent@814: self.CurrentFindHighlight = None Laurent@814: Laurent@814: def RefreshModel(self): Laurent@814: self.RefreshJumpList() Laurent@814: self.Controler.SetEditedElementText(self.TagName, self.GetText()) Laurent@1178: self.ResetSearchResults() Laurent@814: Laurent@814: def OnKeyDown(self, event): Laurent@814: if self.Controler is not None: Laurent@814: Laurent@814: if self.Editor.CallTipActive(): Laurent@814: self.Editor.CallTipCancel() Laurent@814: key = event.GetKeyCode() Laurent@814: key_handled = False Laurent@814: Laurent@814: line = self.Editor.GetCurrentLine() Laurent@814: if line == 0: Laurent@814: start_pos = 0 Laurent@814: else: Laurent@814: start_pos = self.Editor.GetLineEndPosition(line - 1) + 1 Laurent@814: end_pos = self.GetCurrentPos() Laurent@814: lineText = self.Editor.GetTextRange(start_pos, end_pos).replace("\t", " ") Laurent@814: Laurent@814: # Code completion Laurent@814: if key == wx.WXK_SPACE and event.ControlDown(): Laurent@814: Laurent@814: words = lineText.split(" ") Laurent@814: words = [word for i, word in enumerate(words) if word != '' or i == len(words) - 1] Laurent@814: Laurent@814: kw = [] Laurent@814: Laurent@814: if self.TextSyntax == "IL": Laurent@814: if len(words) == 1: Laurent@814: kw = self.Keywords Laurent@814: elif len(words) == 2: Laurent@814: if words[0].upper() in ["CAL", "CALC", "CALNC"]: Laurent@814: kw = self.Functions Laurent@814: elif words[0].upper() in ["JMP", "JMPC", "JMPNC"]: Laurent@814: kw = self.Jumps Laurent@814: else: Laurent@814: kw = self.Variables.keys() Laurent@814: else: Laurent@871: kw = self.Keywords + self.Variables.keys() + self.Functions.keys() Laurent@814: if len(kw) > 0: Laurent@814: if len(words[-1]) > 0: Laurent@814: kw = [keyword for keyword in kw if keyword.startswith(words[-1])] Laurent@814: kw.sort() Laurent@814: self.Editor.AutoCompSetIgnoreCase(True) Laurent@814: self.Editor.AutoCompShow(len(words[-1]), " ".join(kw)) Laurent@814: key_handled = True Laurent@814: elif key == wx.WXK_RETURN or key == wx.WXK_NUMPAD_ENTER: Laurent@814: if self.TextSyntax in ["ST", "ALL"]: Laurent@814: indent = self.Editor.GetLineIndentation(line) Laurent@814: if LineStartswith(lineText.strip(), self.BlockStartKeywords): Laurent@814: indent = (indent / 2 + 1) * 2 Laurent@814: self.Editor.AddText("\n" + " " * indent) Laurent@814: key_handled = True Laurent@814: elif key == wx.WXK_BACK: Laurent@814: if self.TextSyntax in ["ST", "ALL"]: Laurent@814: indent = self.Editor.GetLineIndentation(line) Laurent@814: if lineText.strip() == "" and indent > 0: Laurent@814: self.Editor.DelLineLeft() Laurent@814: self.Editor.AddText(" " * ((max(0, indent - 1) / 2) * 2)) Laurent@814: key_handled = True Laurent@814: if not key_handled: Laurent@814: event.Skip() Laurent@814: Laurent@814: def OnKillFocus(self, event): Laurent@814: self.Editor.AutoCompCancel() Laurent@814: event.Skip() Laurent@814: Laurent@814: #------------------------------------------------------------------------------- Laurent@814: # Highlights showing functions Laurent@814: #------------------------------------------------------------------------------- Laurent@814: Laurent@814: def OnRefreshHighlightsTimer(self, event): Laurent@814: self.RefreshView() Laurent@814: event.Skip() Laurent@814: Laurent@814: def ClearHighlights(self, highlight_type=None): Laurent@814: EditorPanel.ClearHighlights(self, highlight_type) Laurent@814: Laurent@814: if highlight_type is None: Laurent@814: self.Highlights = [] Laurent@814: else: Laurent@814: highlight_type = HIGHLIGHT_TYPES.get(highlight_type, None) Laurent@814: if highlight_type is not None: Laurent@814: self.Highlights = [(infos, start, end, highlight) for (infos, start, end, highlight) in self.Highlights if highlight != highlight_type] Laurent@814: self.RefreshView() Laurent@814: Laurent@814: def AddHighlight(self, infos, start, end, highlight_type): Laurent@814: EditorPanel.AddHighlight(self, infos, start, end, highlight_type) Laurent@814: Laurent@814: highlight_type = HIGHLIGHT_TYPES.get(highlight_type, None) Laurent@814: if infos[0] == "body" and highlight_type is not None: Laurent@814: self.Highlights.append((infos[1], start, end, highlight_type)) Laurent@814: self.Editor.GotoPos(self.Editor.PositionFromLine(start[0]) + start[1]) Laurent@814: self.RefreshHighlightsTimer.Start(int(REFRESH_HIGHLIGHT_PERIOD * 1000), oneShot=True) Laurent@814: Laurent@814: def RemoveHighlight(self, infos, start, end, highlight_type): Laurent@814: EditorPanel.RemoveHighlight(self, infos, start, end, highlight_type) Laurent@814: Laurent@814: highlight_type = HIGHLIGHT_TYPES.get(highlight_type, None) Laurent@814: if (infos[0] == "body" and highlight_type is not None and Laurent@814: (infos[1], start, end, highlight_type) in self.Highlights): Laurent@814: self.Highlights.remove((infos[1], start, end, highlight_type)) Laurent@814: self.RefreshHighlightsTimer.Start(int(REFRESH_HIGHLIGHT_PERIOD * 1000), oneShot=True) Laurent@814: Laurent@814: def ShowHighlights(self, start_pos, end_pos): Laurent@814: for indent, start, end, highlight_type in self.Highlights: Laurent@814: if start[0] == 0: Laurent@814: highlight_start_pos = start[1] - indent Laurent@814: else: Laurent@814: highlight_start_pos = self.Editor.GetLineEndPosition(start[0] - 1) + start[1] - indent + 1 Laurent@814: if end[0] == 0: Laurent@814: highlight_end_pos = end[1] - indent + 1 Laurent@814: else: Laurent@814: highlight_end_pos = self.Editor.GetLineEndPosition(end[0] - 1) + end[1] - indent + 2 Laurent@814: if highlight_start_pos < end_pos and highlight_end_pos > start_pos: Laurent@814: self.StartStyling(highlight_start_pos, 0xff) Laurent@814: self.SetStyling(highlight_end_pos - highlight_start_pos, highlight_type) Laurent@814: self.StartStyling(highlight_start_pos, 0x00) Laurent@814: self.SetStyling(len(self.Editor.GetText()) - highlight_end_pos, wx.stc.STC_STYLE_DEFAULT) Laurent@814: