Laurent@1096: import re
lbessard@145:
laurent@630: import wx
laurent@630: import wx.grid
laurent@630: import wx.stc as stc
laurent@630: import wx.lib.buttons
laurent@630:
Laurent@1097: from plcopen.plcopen import TestTextElement
Laurent@1146: from plcopen.structures import TestIdentifier, IEC_KEYWORDS
laurent@738: from controls import CustomGrid, CustomTable
Laurent@1096: from editors.ConfTreeNodeEditor import ConfTreeNodeEditor
Laurent@814: from util.BitmapLibrary import GetBitmap
Laurent@1104: from controls.CustomStyledTextCtrl import CustomStyledTextCtrl, faces, GetCursorPos, NAVIGATION_KEYS
Laurent@1146: from controls.VariablePanel import VARIABLE_NAME_SUFFIX_MODEL
Laurent@1097: from graphics.GraphicCommons import ERROR_HIGHLIGHT, SEARCH_RESULT_HIGHLIGHT, REFRESH_HIGHLIGHT_PERIOD
lbessard@145:
Laurent@1104: [STC_CODE_ERROR, STC_CODE_SEARCH_RESULT,
Laurent@1104: STC_CODE_SECTION] = range(15, 18)
Laurent@1097:
Laurent@1097: HIGHLIGHT_TYPES = {
Laurent@1097: ERROR_HIGHLIGHT: STC_CODE_ERROR,
Laurent@1097: SEARCH_RESULT_HIGHLIGHT: STC_CODE_SEARCH_RESULT,
Laurent@1097: }
Laurent@1097:
Laurent@1161: EDGE_COLUMN = 80
Laurent@1161:
Laurent@1096: class CodeEditor(CustomStyledTextCtrl):
Laurent@1096:
Laurent@1096: KEYWORDS = []
Laurent@1096: COMMENT_HEADER = ""
Laurent@1096:
Laurent@1096: def __init__(self, parent, window, controler):
Laurent@1096: CustomStyledTextCtrl.__init__(self, parent, -1, wx.DefaultPosition,
Laurent@920: wx.Size(-1, 300), 0)
Laurent@1161:
lbessard@145: self.SetMarginType(1, stc.STC_MARGIN_NUMBER)
lbessard@145: self.SetMarginWidth(1, 25)
lbessard@145:
lbessard@145: self.SetProperty("fold", "1")
lbessard@145: self.SetProperty("tab.timmy.whinge.level", "1")
lbessard@145: self.SetMargins(0,0)
lbessard@145:
lbessard@145: self.SetViewWhiteSpace(False)
lbessard@145:
lbessard@145: self.SetEdgeMode(stc.STC_EDGE_BACKGROUND)
Laurent@1161: self.SetEdgeColumn(EDGE_COLUMN)
lbessard@145:
lbessard@145: # Setup a margin to hold fold markers
lbessard@145: self.SetMarginType(2, stc.STC_MARGIN_SYMBOL)
lbessard@145: self.SetMarginMask(2, stc.STC_MASK_FOLDERS)
lbessard@145: self.SetMarginSensitive(2, True)
lbessard@145: self.SetMarginWidth(2, 12)
lbessard@145:
Laurent@1096: # Like a flattened tree control using square headers
Laurent@1096: self.MarkerDefine(stc.STC_MARKNUM_FOLDEROPEN, stc.STC_MARK_BOXMINUS, "white", "#808080")
Laurent@1096: self.MarkerDefine(stc.STC_MARKNUM_FOLDER, stc.STC_MARK_BOXPLUS, "white", "#808080")
Laurent@1096: self.MarkerDefine(stc.STC_MARKNUM_FOLDERSUB, stc.STC_MARK_VLINE, "white", "#808080")
Laurent@1096: self.MarkerDefine(stc.STC_MARKNUM_FOLDERTAIL, stc.STC_MARK_LCORNER, "white", "#808080")
Laurent@1096: self.MarkerDefine(stc.STC_MARKNUM_FOLDEREND, stc.STC_MARK_BOXPLUSCONNECTED, "white", "#808080")
Laurent@1096: self.MarkerDefine(stc.STC_MARKNUM_FOLDEROPENMID, stc.STC_MARK_BOXMINUSCONNECTED, "white", "#808080")
Laurent@1096: self.MarkerDefine(stc.STC_MARKNUM_FOLDERMIDTAIL, stc.STC_MARK_TCORNER, "white", "#808080")
Laurent@1096:
lbessard@145: self.Bind(stc.EVT_STC_UPDATEUI, self.OnUpdateUI)
lbessard@145: self.Bind(stc.EVT_STC_MARGINCLICK, self.OnMarginClick)
lbessard@145: self.Bind(wx.EVT_KEY_DOWN, self.OnKeyPressed)
lbessard@145:
lbessard@145: # Make some styles, The lexer defines what each style is used for, we
lbessard@145: # just have to define what each style looks like. This set is adapted from
lbessard@145: # Scintilla sample property files.
lbessard@145:
lbessard@145: # Global default styles for all languages
lbessard@145: self.StyleSetSpec(stc.STC_STYLE_DEFAULT, "face:%(mono)s,size:%(size)d" % faces)
lbessard@145: self.StyleClearAll() # Reset all to be like the default
lbessard@145:
lbessard@145: # Global default styles for all languages
lbessard@145: self.StyleSetSpec(stc.STC_STYLE_DEFAULT, "face:%(mono)s,size:%(size)d" % faces)
Laurent@1066: self.StyleSetSpec(stc.STC_STYLE_LINENUMBER, "back:#C0C0C0,face:%(helv)s,size:%(size)d" % faces)
Laurent@1104: self.StyleSetSpec(stc.STC_STYLE_CONTROLCHAR, "face:%(mono)s" % faces)
lbessard@145: self.StyleSetSpec(stc.STC_STYLE_BRACELIGHT, "fore:#FFFFFF,back:#0000FF,bold")
lbessard@145: self.StyleSetSpec(stc.STC_STYLE_BRACEBAD, "fore:#000000,back:#FF0000,bold")
lbessard@145:
Laurent@1097: # Highlighting styles
Laurent@1097: self.StyleSetSpec(STC_CODE_ERROR, 'fore:#FF0000,back:#FFFF00,size:%(size)d' % faces)
Laurent@1097: self.StyleSetSpec(STC_CODE_SEARCH_RESULT, 'fore:#FFFFFF,back:#FFA500,size:%(size)d' % faces)
Laurent@1097:
Laurent@1104: # Section style
Laurent@1104: self.StyleSetSpec(STC_CODE_SECTION, 'fore:#808080,size:%(size)d')
Laurent@1104: self.StyleSetChangeable(STC_CODE_SECTION, False)
Laurent@1104:
lbessard@145: # Indentation size
Laurent@1161: self.SetTabWidth(4)
lbessard@145: self.SetUseTabs(0)
Laurent@1096:
Laurent@1096: self.SetCodeLexer()
Laurent@1096: self.SetKeyWords(0, " ".join(self.KEYWORDS))
Laurent@1096:
lbessard@145: self.Controler = controler
lbessard@145: self.ParentWindow = window
lbessard@145:
lbessard@145: self.DisableEvents = True
lbessard@145: self.CurrentAction = None
lbessard@145:
Laurent@1178: self.ResetSearchResults()
Laurent@1097:
Laurent@1097: self.RefreshHighlightsTimer = wx.Timer(self, -1)
Laurent@1097: self.Bind(wx.EVT_TIMER, self.OnRefreshHighlightsTimer, self.RefreshHighlightsTimer)
Laurent@1097:
Laurent@1096: self.SectionsComments = {}
Laurent@1124: for section in self.Controler.SECTIONS_NAMES:
Laurent@1110: section_comment = " %s section " % (section)
Laurent@1161: len_headers = EDGE_COLUMN - len(section_comment)
Laurent@1110: section_comment = self.COMMENT_HEADER * (len_headers / 2) + \
Laurent@1110: section_comment + \
Laurent@1110: self.COMMENT_HEADER * (len_headers - len_headers / 2)
Laurent@1110:
Laurent@1096: self.SectionsComments[section] = {
Laurent@1110: "comment": section_comment,
Laurent@1096: }
Laurent@1110:
Laurent@1124: for i, section in enumerate(self.Controler.SECTIONS_NAMES):
Laurent@1110: section_infos = self.SectionsComments[section]
Laurent@1124: if i + 1 < len(self.Controler.SECTIONS_NAMES):
Laurent@1124: section_end = self.SectionsComments[
Laurent@1124: self.Controler.SECTIONS_NAMES[i + 1]]["comment"]
Laurent@1110: else:
Laurent@1110: section_end = "$"
Laurent@1110: section_infos["pattern"] = re.compile(
Laurent@1110: section_infos["comment"] + "(.*)" +
Laurent@1110: section_end, re.DOTALL)
Laurent@1110:
lbessard@145: self.SetModEventMask(wx.stc.STC_MOD_BEFOREINSERT|wx.stc.STC_MOD_BEFOREDELETE)
lbessard@145:
Laurent@1096: self.Bind(wx.stc.EVT_STC_DO_DROP, self.OnDoDrop)
lbessard@145: self.Bind(wx.EVT_KILL_FOCUS, self.OnKillFocus)
Laurent@1096: self.Bind(wx.stc.EVT_STC_MODIFIED, self.OnModification)
Laurent@1096:
Laurent@1096: def SetCodeLexer(self):
Laurent@1096: pass
lbessard@145:
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:
lbessard@145: def OnModification(self, event):
lbessard@145: if not self.DisableEvents:
lbessard@145: mod_type = event.GetModificationType()
lbessard@145: if not (mod_type&wx.stc.STC_PERFORMED_UNDO or mod_type&wx.stc.STC_PERFORMED_REDO):
lbessard@145: if mod_type&wx.stc.STC_MOD_BEFOREINSERT:
lbessard@145: if self.CurrentAction == None:
lbessard@145: self.StartBuffering()
lbessard@145: elif self.CurrentAction[0] != "Add" or self.CurrentAction[1] != event.GetPosition() - 1:
lbessard@145: self.Controler.EndBuffering()
lbessard@145: self.StartBuffering()
lbessard@145: self.CurrentAction = ("Add", event.GetPosition())
laurent@658: wx.CallAfter(self.RefreshModel)
lbessard@145: elif mod_type&wx.stc.STC_MOD_BEFOREDELETE:
lbessard@145: if self.CurrentAction == None:
lbessard@145: self.StartBuffering()
lbessard@145: elif self.CurrentAction[0] != "Delete" or self.CurrentAction[1] != event.GetPosition() + 1:
lbessard@145: self.Controler.EndBuffering()
lbessard@145: self.StartBuffering()
lbessard@145: self.CurrentAction = ("Delete", event.GetPosition())
laurent@658: wx.CallAfter(self.RefreshModel)
Laurent@1104: wx.CallAfter(self.RefreshSectionStyling)
lbessard@145: event.Skip()
lbessard@145:
lbessard@145: def OnDoDrop(self, event):
Laurent@1138: try:
Laurent@1138: values = eval(event.GetDragText())
Laurent@1138: except:
Laurent@1138: values = event.GetDragText()
Laurent@1138: if isinstance(values, tuple):
Laurent@1138: message = None
Laurent@1138: if values[3] == self.Controler.GetCurrentLocation():
Laurent@1138: self.ResetBuffer()
Laurent@1138: event.SetDragText(values[0])
Laurent@1138: wx.CallAfter(self.RefreshModel)
Laurent@1138: else:
Laurent@1138: event.SetDragText("")
Laurent@1138: else:
Laurent@1138: self.ResetBuffer()
Laurent@1138: wx.CallAfter(self.RefreshModel)
lbessard@145: event.Skip()
lbessard@145:
lbessard@145: # Buffer the last model state
lbessard@145: def RefreshBuffer(self):
Laurent@1096: self.Controler.BufferCodeFile()
laurent@658: if self.ParentWindow is not None:
lbessard@145: self.ParentWindow.RefreshTitle()
Edouard@587: self.ParentWindow.RefreshFileMenu()
lbessard@145: self.ParentWindow.RefreshEditMenu()
laurent@630: self.ParentWindow.RefreshPageTitles()
lbessard@145:
lbessard@145: def StartBuffering(self):
lbessard@145: self.Controler.StartBuffering()
laurent@658: if self.ParentWindow is not None:
lbessard@145: self.ParentWindow.RefreshTitle()
Edouard@587: self.ParentWindow.RefreshFileMenu()
lbessard@145: self.ParentWindow.RefreshEditMenu()
laurent@630: self.ParentWindow.RefreshPageTitles()
lbessard@145:
lbessard@145: def ResetBuffer(self):
lbessard@145: if self.CurrentAction != None:
lbessard@145: self.Controler.EndBuffering()
lbessard@145: self.CurrentAction = None
lbessard@145:
Laurent@1096: def GetCodeText(self):
Laurent@1096: parts = self.Controler.GetTextParts()
Laurent@1096: text = ""
Laurent@1124: for section in self.Controler.SECTIONS_NAMES:
Laurent@1096: section_comments = self.SectionsComments[section]
Laurent@1110: text += section_comments["comment"]
Laurent@1117: if parts[section] == "":
Laurent@1104: text += "\n"
Laurent@1117: else:
Laurent@1117: if not parts[section].startswith("\n"):
Laurent@1117: text += "\n"
Laurent@1117: text += parts[section]
Laurent@1117: if not parts[section].endswith("\n"):
Laurent@1117: text += "\n"
Laurent@1096: return text
Laurent@1096:
Laurent@1101: def RefreshView(self, scroll_to_highlight=False):
lbessard@145: self.ResetBuffer()
lbessard@145: self.DisableEvents = True
lbessard@145: old_cursor_pos = self.GetCurrentPos()
Laurent@1060: line = self.GetFirstVisibleLine()
Laurent@1060: column = self.GetXOffset()
lbessard@145: old_text = self.GetText()
Laurent@1096: new_text = self.GetCodeText()
Laurent@1060: if old_text != new_text:
Laurent@1101: self.SetText(new_text)
Laurent@1060: new_cursor_pos = GetCursorPos(old_text, new_text)
Laurent@1060: self.LineScroll(column, line)
Laurent@1060: if new_cursor_pos != None:
Laurent@1060: self.GotoPos(new_cursor_pos)
Laurent@1060: else:
Laurent@1060: self.GotoPos(old_cursor_pos)
Laurent@1060: self.EmptyUndoBuffer()
lbessard@145: self.DisableEvents = False
lbessard@145:
Laurent@1104: self.RefreshSectionStyling()
Laurent@1104:
Laurent@1104: self.ShowHighlights()
Laurent@1178:
Laurent@1104: def RefreshSectionStyling(self):
lbessard@145: self.Colourise(0, -1)
Laurent@1097:
Laurent@1104: text = self.GetText()
Laurent@1104: for line in xrange(self.GetLineCount()):
Laurent@1104: self.SetLineState(line, 0)
Laurent@1104:
Laurent@1124: for section in self.Controler.SECTIONS_NAMES:
Laurent@1104: section_comments = self.SectionsComments[section]
Laurent@1110: start_pos = text.find(section_comments["comment"])
Laurent@1110: end_pos = start_pos + len(section_comments["comment"])
Laurent@1104: self.StartStyling(start_pos, 0xff)
Laurent@1110: self.SetStyling(end_pos - start_pos, STC_CODE_SECTION)
Laurent@1110: self.SetLineState(self.LineFromPosition(start_pos), 1)
Laurent@1110:
Laurent@1110: self.StartStyling(end_pos, 0x00)
Laurent@1110: self.SetStyling(len(self.GetText()) - end_pos, stc.STC_STYLE_DEFAULT)
lbessard@145:
laurent@630: def DoGetBestSize(self):
laurent@630: return self.ParentWindow.GetPanelBestSize()
laurent@630:
lbessard@145: def RefreshModel(self):
Laurent@1096: text = self.GetText()
Laurent@1096: parts = {}
Laurent@1124: for section in self.Controler.SECTIONS_NAMES:
Laurent@1096: section_comments = self.SectionsComments[section]
Laurent@1096: result = section_comments["pattern"].search(text)
Laurent@1096: if result is not None:
Laurent@1096: parts[section] = result.group(1)
Laurent@1096: else:
Laurent@1096: parts[section] = ""
Laurent@1096: self.Controler.SetTextParts(parts)
Laurent@1178: self.ResetSearchResults()
Laurent@1178:
lbessard@145: def OnKeyPressed(self, event):
lbessard@145: if self.CallTipActive():
lbessard@145: self.CallTipCancel()
lbessard@145: key = event.GetKeyCode()
Laurent@1124: current_pos = self.GetCurrentPos()
Laurent@1124: selected = self.GetSelection()
Laurent@1124: text_selected = selected[0] != selected[1]
Laurent@1124:
Laurent@1139: # Test if caret is before Windows like new line
Laurent@1139: text = self.GetText()
Laurent@1150: if current_pos < len(text) and ord(text[current_pos]) == 13:
Laurent@1139: newline_size = 2
Laurent@1139: else:
Laurent@1139: newline_size = 1
Laurent@1139:
Laurent@1124: # Disable to type any character in section header lines
Laurent@1117: if (self.GetLineState(self.LineFromPosition(current_pos)) and
Laurent@1124: not text_selected and
Laurent@1117: key not in NAVIGATION_KEYS + [
Laurent@1117: wx.WXK_RETURN,
Laurent@1117: wx.WXK_NUMPAD_ENTER]):
Laurent@1104: return
Laurent@1104:
Laurent@1124: # Disable to delete line between code and header lines
Laurent@1124: elif (self.GetCurLine()[0].strip() != "" and not text_selected and
Laurent@1124: (key == wx.WXK_BACK and
Laurent@1124: self.GetLineState(self.LineFromPosition(max(0, current_pos - 1))) or
Laurent@1124: key in [wx.WXK_DELETE, wx.WXK_NUMPAD_DELETE] and
Laurent@1139: self.GetLineState(self.LineFromPosition(min(len(text), current_pos + newline_size))))):
Laurent@1124: return
Laurent@1124:
Laurent@1104: elif key == 32 and event.ControlDown():
lbessard@145: pos = self.GetCurrentPos()
lbessard@145:
lbessard@145: # Tips
lbessard@145: if event.ShiftDown():
lbessard@145: pass
lbessard@145: # Code completion
lbessard@145: else:
lbessard@145: self.AutoCompSetIgnoreCase(False) # so this needs to match
Laurent@1136:
Laurent@1136: keywords = self.KEYWORDS + [var["Name"]
Laurent@1136: for var in self.Controler.GetVariables()]
Laurent@1136: keywords.sort()
Laurent@1136: self.AutoCompShow(0, " ".join(keywords))
lbessard@145: else:
lbessard@145: event.Skip()
lbessard@145:
lbessard@145: def OnKillFocus(self, event):
lbessard@145: self.AutoCompCancel()
lbessard@145: event.Skip()
lbessard@145:
Laurent@1126: def OnUpdateUI(self, event):
lbessard@145: # check for matching braces
lbessard@145: braceAtCaret = -1
lbessard@145: braceOpposite = -1
lbessard@145: charBefore = None
lbessard@145: caretPos = self.GetCurrentPos()
lbessard@145:
lbessard@145: if caretPos > 0:
lbessard@145: charBefore = self.GetCharAt(caretPos - 1)
lbessard@145: styleBefore = self.GetStyleAt(caretPos - 1)
lbessard@145:
lbessard@145: # check before
lbessard@145: if charBefore and chr(charBefore) in "[]{}()" and styleBefore == stc.STC_P_OPERATOR:
lbessard@145: braceAtCaret = caretPos - 1
lbessard@145:
lbessard@145: # check after
lbessard@145: if braceAtCaret < 0:
lbessard@145: charAfter = self.GetCharAt(caretPos)
lbessard@145: styleAfter = self.GetStyleAt(caretPos)
lbessard@145:
lbessard@145: if charAfter and chr(charAfter) in "[]{}()" and styleAfter == stc.STC_P_OPERATOR:
lbessard@145: braceAtCaret = caretPos
lbessard@145:
lbessard@145: if braceAtCaret >= 0:
lbessard@145: braceOpposite = self.BraceMatch(braceAtCaret)
lbessard@145:
lbessard@145: if braceAtCaret != -1 and braceOpposite == -1:
lbessard@145: self.BraceBadLight(braceAtCaret)
lbessard@145: else:
lbessard@145: self.BraceHighlight(braceAtCaret, braceOpposite)
Laurent@1146:
Laurent@1146: selected_text = self.GetSelectedText()
Laurent@1146: if selected_text:
Laurent@1146: self.ParentWindow.SetCopyBuffer(selected_text, True)
Laurent@1161:
Laurent@1126: def OnMarginClick(self, event):
lbessard@145: # fold and unfold as needed
lbessard@145: if evt.GetMargin() == 2:
lbessard@145: if evt.GetShift() and evt.GetControl():
lbessard@145: self.FoldAll()
lbessard@145: else:
lbessard@145: lineClicked = self.LineFromPosition(evt.GetPosition())
lbessard@145:
lbessard@145: if self.GetFoldLevel(lineClicked) & stc.STC_FOLDLEVELHEADERFLAG:
lbessard@145: if evt.GetShift():
lbessard@145: self.SetFoldExpanded(lineClicked, True)
lbessard@145: self.Expand(lineClicked, True, True, 1)
lbessard@145: elif evt.GetControl():
lbessard@145: if self.GetFoldExpanded(lineClicked):
lbessard@145: self.SetFoldExpanded(lineClicked, False)
lbessard@145: self.Expand(lineClicked, False, True, 0)
lbessard@145: else:
lbessard@145: self.SetFoldExpanded(lineClicked, True)
lbessard@145: self.Expand(lineClicked, True, True, 100)
lbessard@145: else:
lbessard@145: self.ToggleFold(lineClicked)
Laurent@1126: event.Skip()
lbessard@145:
lbessard@145: def FoldAll(self):
lbessard@145: lineCount = self.GetLineCount()
lbessard@145: expanding = True
lbessard@145:
lbessard@145: # find out if we are folding or unfolding
lbessard@145: for lineNum in range(lineCount):
lbessard@145: if self.GetFoldLevel(lineNum) & stc.STC_FOLDLEVELHEADERFLAG:
lbessard@145: expanding = not self.GetFoldExpanded(lineNum)
lbessard@145: break
lbessard@145:
lbessard@145: lineNum = 0
lbessard@145:
lbessard@145: while lineNum < lineCount:
lbessard@145: level = self.GetFoldLevel(lineNum)
lbessard@145: if level & stc.STC_FOLDLEVELHEADERFLAG and \
lbessard@145: (level & stc.STC_FOLDLEVELNUMBERMASK) == stc.STC_FOLDLEVELBASE:
lbessard@145:
lbessard@145: if expanding:
lbessard@145: self.SetFoldExpanded(lineNum, True)
lbessard@145: lineNum = self.Expand(lineNum, True)
lbessard@145: lineNum = lineNum - 1
lbessard@145: else:
lbessard@145: lastChild = self.GetLastChild(lineNum, -1)
lbessard@145: self.SetFoldExpanded(lineNum, False)
lbessard@145:
lbessard@145: if lastChild > lineNum:
lbessard@145: self.HideLines(lineNum+1, lastChild)
lbessard@145:
lbessard@145: lineNum = lineNum + 1
lbessard@145:
lbessard@145:
lbessard@145:
lbessard@145: def Expand(self, line, doExpand, force=False, visLevels=0, level=-1):
lbessard@145: lastChild = self.GetLastChild(line, level)
lbessard@145: line = line + 1
lbessard@145:
lbessard@145: while line <= lastChild:
lbessard@145: if force:
lbessard@145: if visLevels > 0:
lbessard@145: self.ShowLines(line, line)
lbessard@145: else:
lbessard@145: self.HideLines(line, line)
lbessard@145: else:
lbessard@145: if doExpand:
lbessard@145: self.ShowLines(line, line)
lbessard@145:
lbessard@145: if level == -1:
lbessard@145: level = self.GetFoldLevel(line)
lbessard@145:
lbessard@145: if level & stc.STC_FOLDLEVELHEADERFLAG:
lbessard@145: if force:
lbessard@145: if visLevels > 1:
lbessard@145: self.SetFoldExpanded(line, True)
lbessard@145: else:
lbessard@145: self.SetFoldExpanded(line, False)
lbessard@145:
lbessard@145: line = self.Expand(line, doExpand, force, visLevels-1)
lbessard@145:
lbessard@145: else:
lbessard@145: if doExpand and self.GetFoldExpanded(line):
lbessard@145: line = self.Expand(line, True, force, visLevels-1)
lbessard@145: else:
lbessard@145: line = self.Expand(line, False, force, visLevels-1)
lbessard@145: else:
lbessard@145: line = line + 1
lbessard@145:
lbessard@145: return line
lbessard@145:
laurent@637: def Cut(self):
laurent@637: self.ResetBuffer()
laurent@637: self.DisableEvents = True
laurent@637: self.CmdKeyExecute(wx.stc.STC_CMD_CUT)
laurent@637: self.DisableEvents = False
laurent@637: self.RefreshModel()
laurent@637: self.RefreshBuffer()
laurent@637:
laurent@637: def Copy(self):
laurent@637: self.CmdKeyExecute(wx.stc.STC_CMD_COPY)
Laurent@1124: self.ParentWindow.RefreshEditMenu()
laurent@637:
laurent@637: def Paste(self):
laurent@637: self.ResetBuffer()
laurent@637: self.DisableEvents = True
laurent@637: self.CmdKeyExecute(wx.stc.STC_CMD_PASTE)
laurent@637: self.DisableEvents = False
laurent@637: self.RefreshModel()
laurent@637: self.RefreshBuffer()
laurent@637:
Laurent@1097: def Find(self, direction, search_params):
Laurent@1097: if self.SearchParams != search_params:
Laurent@1097: self.ClearHighlights(SEARCH_RESULT_HIGHLIGHT)
Laurent@1097:
Laurent@1097: self.SearchParams = search_params
Laurent@1097: criteria = {
Laurent@1097: "raw_pattern": search_params["find_pattern"],
Laurent@1097: "pattern": re.compile(search_params["find_pattern"]),
Laurent@1097: "case_sensitive": search_params["case_sensitive"],
Laurent@1097: "regular_expression": search_params["regular_expression"],
Laurent@1097: "filter": "all"}
Laurent@1097:
Laurent@1097: self.SearchResults = [
Laurent@1097: (start, end, SEARCH_RESULT_HIGHLIGHT)
Laurent@1097: for start, end, text in
Laurent@1097: TestTextElement(self.GetText(), criteria)]
Laurent@1097: self.CurrentFindHighlight = None
Laurent@1097:
Laurent@1097: if len(self.SearchResults) > 0:
Laurent@1097: if self.CurrentFindHighlight is not None:
Laurent@1097: old_idx = self.SearchResults.index(self.CurrentFindHighlight)
Laurent@1097: if self.SearchParams["wrap"]:
Laurent@1097: idx = (old_idx + direction) % len(self.SearchResults)
Laurent@1097: else:
Laurent@1097: idx = max(0, min(old_idx + direction, len(self.SearchResults) - 1))
Laurent@1097: if idx != old_idx:
Laurent@1097: self.RemoveHighlight(*self.CurrentFindHighlight)
Laurent@1097: self.CurrentFindHighlight = self.SearchResults[idx]
Laurent@1097: self.AddHighlight(*self.CurrentFindHighlight)
Laurent@1097: else:
Laurent@1178: caret_pos = self.GetCurrentPos()
Laurent@1178: if self.SearchParams["wrap"]:
Laurent@1178: self.CurrentFindHighlight = self.SearchResults[0]
Laurent@1178: else:
Laurent@1178: self.CurrentFindHighlight = None
Laurent@1178: for start, end, highlight_type in self.SearchResults:
Laurent@1178: if start[0] == 0:
Laurent@1178: highlight_start_pos = start[1]
Laurent@1178: else:
Laurent@1178: highlight_start_pos = self.GetLineEndPosition(start[0] - 1) + start[1] + 1
Laurent@1178: if highlight_start_pos >= caret_pos:
Laurent@1178: self.CurrentFindHighlight = (start, end, highlight_type)
Laurent@1178: break
Laurent@1178: if self.CurrentFindHighlight is not None:
Laurent@1178: self.AddHighlight(*self.CurrentFindHighlight)
Laurent@1097:
Laurent@1101: self.ScrollToLine(self.CurrentFindHighlight[0][0])
Laurent@1101:
Laurent@1097: else:
Laurent@1097: if self.CurrentFindHighlight is not None:
Laurent@1097: self.RemoveHighlight(*self.CurrentFindHighlight)
Laurent@1097: self.CurrentFindHighlight = None
Laurent@1097:
Laurent@1097: #-------------------------------------------------------------------------------
Laurent@1097: # Highlights showing functions
Laurent@1097: #-------------------------------------------------------------------------------
Laurent@1097:
Laurent@1097: def OnRefreshHighlightsTimer(self, event):
Laurent@1101: self.RefreshView(True)
Laurent@1097: event.Skip()
Laurent@1097:
Laurent@1097: def ClearHighlights(self, highlight_type=None):
Laurent@1097: if highlight_type is None:
Laurent@1097: self.Highlights = []
Laurent@1097: else:
Laurent@1097: highlight_type = HIGHLIGHT_TYPES.get(highlight_type, None)
Laurent@1097: if highlight_type is not None:
Laurent@1097: self.Highlights = [(start, end, highlight) for (start, end, highlight) in self.Highlights if highlight != highlight_type]
Laurent@1097: self.RefreshView()
Laurent@1097:
Laurent@1097: def AddHighlight(self, start, end, highlight_type):
Laurent@1097: highlight_type = HIGHLIGHT_TYPES.get(highlight_type, None)
Laurent@1097: if highlight_type is not None:
Laurent@1097: self.Highlights.append((start, end, highlight_type))
Laurent@1097: self.GotoPos(self.PositionFromLine(start[0]) + start[1])
Laurent@1097: self.RefreshHighlightsTimer.Start(int(REFRESH_HIGHLIGHT_PERIOD * 1000), oneShot=True)
Laurent@1097: self.RefreshView()
Laurent@1097:
Laurent@1097: def RemoveHighlight(self, start, end, highlight_type):
Laurent@1097: highlight_type = HIGHLIGHT_TYPES.get(highlight_type, None)
Laurent@1097: if (highlight_type is not None and
Laurent@1097: (start, end, highlight_type) in self.Highlights):
Laurent@1097: self.Highlights.remove((start, end, highlight_type))
Laurent@1097: self.RefreshHighlightsTimer.Start(int(REFRESH_HIGHLIGHT_PERIOD * 1000), oneShot=True)
Laurent@1097:
Laurent@1097: def ShowHighlights(self):
Laurent@1097: for start, end, highlight_type in self.Highlights:
Laurent@1097: if start[0] == 0:
Laurent@1097: highlight_start_pos = start[1]
Laurent@1097: else:
Laurent@1097: highlight_start_pos = self.GetLineEndPosition(start[0] - 1) + start[1] + 1
Laurent@1097: if end[0] == 0:
Laurent@1178: highlight_end_pos = end[1] + 1
Laurent@1097: else:
Laurent@1097: highlight_end_pos = self.GetLineEndPosition(end[0] - 1) + end[1] + 2
Laurent@1097: self.StartStyling(highlight_start_pos, 0xff)
Laurent@1097: self.SetStyling(highlight_end_pos - highlight_start_pos, highlight_type)
Laurent@1178: self.StartStyling(highlight_end_pos, 0x00)
Laurent@1097: self.SetStyling(len(self.GetText()) - highlight_end_pos, stc.STC_STYLE_DEFAULT)
Laurent@1097:
lbessard@145:
lbessard@145: #-------------------------------------------------------------------------------
lbessard@145: # Helper for VariablesGrid values
lbessard@145: #-------------------------------------------------------------------------------
lbessard@145:
laurent@651: class VariablesTable(CustomTable):
laurent@651:
lbessard@145: def GetValue(self, row, col):
lbessard@145: if row < self.GetNumberRows():
lbessard@145: if col == 0:
lbessard@145: return row + 1
lbessard@145: else:
laurent@651: return str(self.data[row].get(self.GetColLabelValue(col, False), ""))
laurent@651:
lbessard@145: def _updateColAttrs(self, grid):
lbessard@145: """
lbessard@145: wxGrid -> update the column attributes to add the
lbessard@145: appropriate renderer given the column name.
lbessard@145:
lbessard@145: Otherwise default to the default renderer.
lbessard@145: """
lbessard@145:
lbessard@145: typelist = None
lbessard@145: accesslist = None
lbessard@145: for row in range(self.GetNumberRows()):
lbessard@145: for col in range(self.GetNumberCols()):
lbessard@145: editor = None
lbessard@145: renderer = None
laurent@665: colname = self.GetColLabelValue(col, False)
lbessard@145:
Laurent@1096: if colname in ["Name", "Initial"]:
lbessard@145: editor = wx.grid.GridCellTextEditor()
lbessard@145: elif colname == "Class":
lbessard@145: editor = wx.grid.GridCellChoiceEditor()
Edouard@603: editor.SetParameters("input,memory,output")
lbessard@145: elif colname == "Type":
lbessard@145: pass
lbessard@145: else:
lbessard@145: grid.SetReadOnly(row, col, True)
lbessard@145:
lbessard@145: grid.SetCellEditor(row, col, editor)
lbessard@145: grid.SetCellRenderer(row, col, renderer)
lbessard@145:
lbessard@145: grid.SetCellBackgroundColour(row, col, wx.WHITE)
laurent@651: self.ResizeRow(grid, row)
laurent@848:
lbessard@145:
lbessard@145: class VariablesEditor(wx.Panel):
lbessard@145:
lbessard@145: def __init__(self, parent, window, controler):
laurent@848: wx.Panel.__init__(self, parent, style=wx.TAB_TRAVERSAL)
laurent@848:
Laurent@1161: main_sizer = wx.FlexGridSizer(cols=2, hgap=0, rows=1, vgap=4)
Laurent@1161: main_sizer.AddGrowableCol(1)
Laurent@1161: main_sizer.AddGrowableRow(0)
Laurent@1161:
Laurent@1161: controls_sizer = wx.BoxSizer(wx.VERTICAL)
Laurent@1161: main_sizer.AddSizer(controls_sizer, border=5, flag=wx.ALL)
laurent@848:
laurent@848: for name, bitmap, help in [
laurent@848: ("AddVariableButton", "add_element", _("Add variable")),
laurent@848: ("DeleteVariableButton", "remove_element", _("Remove variable")),
laurent@848: ("UpVariableButton", "up", _("Move variable up")),
laurent@848: ("DownVariableButton", "down", _("Move variable down"))]:
laurent@848: button = wx.lib.buttons.GenBitmapButton(self, bitmap=GetBitmap(bitmap),
laurent@848: size=wx.Size(28, 28), style=wx.NO_BORDER)
laurent@848: button.SetToolTipString(help)
laurent@848: setattr(self, name, button)
Laurent@1161: controls_sizer.AddWindow(button, border=5, flag=wx.BOTTOM)
Laurent@1097:
Laurent@1152: self.VariablesGrid = CustomGrid(self, style=wx.VSCROLL)
Laurent@1097: self.VariablesGrid.Bind(wx.grid.EVT_GRID_CELL_CHANGE, self.OnVariablesGridCellChange)
Laurent@1097: self.VariablesGrid.Bind(wx.grid.EVT_GRID_CELL_LEFT_CLICK, self.OnVariablesGridCellLeftClick)
Laurent@1097: self.VariablesGrid.Bind(wx.grid.EVT_GRID_EDITOR_SHOWN, self.OnVariablesGridEditorShown)
Laurent@1097: main_sizer.AddWindow(self.VariablesGrid, flag=wx.GROW)
laurent@848:
laurent@848: self.SetSizer(main_sizer)
laurent@848:
lbessard@145: self.ParentWindow = window
lbessard@145: self.Controler = controler
lbessard@145:
Laurent@1146: self.VariablesDefaultValue = {"Name" : "", "Type" : "INT", "Initial": ""}
Laurent@1096: self.Table = VariablesTable(self, [], ["#", "Name", "Type", "Initial"])
lbessard@145: self.ColAlignements = [wx.ALIGN_RIGHT, wx.ALIGN_LEFT, wx.ALIGN_LEFT, wx.ALIGN_LEFT]
lbessard@145: self.ColSizes = [40, 200, 150, 150]
lbessard@145: self.VariablesGrid.SetTable(self.Table)
laurent@626: self.VariablesGrid.SetButtons({"Add": self.AddVariableButton,
laurent@626: "Delete": self.DeleteVariableButton,
laurent@626: "Up": self.UpVariableButton,
laurent@626: "Down": self.DownVariableButton})
laurent@626:
laurent@626: def _AddVariable(new_row):
Laurent@1146: if new_row > 0:
Laurent@1146: row_content = self.Table.data[new_row - 1].copy()
Laurent@1146: result = VARIABLE_NAME_SUFFIX_MODEL.search(row_content["Name"])
Laurent@1146: if result is not None:
Laurent@1146: name = row_content["Name"][:result.start(1)]
Laurent@1146: suffix = result.group(1)
Laurent@1146: if suffix != "":
Laurent@1146: start_idx = int(suffix)
Laurent@1146: else:
Laurent@1146: start_idx = 0
Laurent@1146: else:
Laurent@1146: name = row_content["Name"]
Laurent@1146: start_idx = 0
Laurent@1146: row_content["Name"] = self.Controler.GenerateNewName(
Laurent@1146: name + "%d", start_idx)
Laurent@1146: else:
Laurent@1146: row_content = self.VariablesDefaultValue.copy()
Laurent@1146: self.Table.InsertRow(new_row, row_content)
laurent@626: self.RefreshModel()
laurent@626: self.RefreshView()
laurent@626: return new_row
laurent@626: setattr(self.VariablesGrid, "_AddRow", _AddVariable)
laurent@626:
laurent@626: def _DeleteVariable(row):
laurent@626: self.Table.RemoveRow(row)
laurent@626: self.RefreshModel()
laurent@626: self.RefreshView()
laurent@626: setattr(self.VariablesGrid, "_DeleteRow", _DeleteVariable)
laurent@626:
laurent@626: def _MoveVariable(row, move):
laurent@626: new_row = self.Table.MoveRow(row, move)
laurent@626: if new_row != row:
laurent@626: self.RefreshModel()
laurent@626: self.RefreshView()
laurent@626: return new_row
laurent@626: setattr(self.VariablesGrid, "_MoveRow", _MoveVariable)
laurent@626:
lbessard@145: self.VariablesGrid.SetRowLabelSize(0)
lbessard@145: for col in range(self.Table.GetNumberCols()):
lbessard@145: attr = wx.grid.GridCellAttr()
lbessard@145: attr.SetAlignment(self.ColAlignements[col], wx.ALIGN_CENTRE)
lbessard@145: self.VariablesGrid.SetColAttr(col, attr)
lbessard@145: self.VariablesGrid.SetColSize(col, self.ColSizes[col])
lbessard@145: self.Table.ResetView(self.VariablesGrid)
lbessard@145:
lbessard@145: def RefreshModel(self):
lbessard@145: self.Controler.SetVariables(self.Table.GetData())
lbessard@145: self.RefreshBuffer()
lbessard@145:
lbessard@145: # Buffer the last model state
lbessard@145: def RefreshBuffer(self):
Laurent@1096: self.Controler.BufferCodeFile()
lbessard@145: self.ParentWindow.RefreshTitle()
Edouard@587: self.ParentWindow.RefreshFileMenu()
lbessard@145: self.ParentWindow.RefreshEditMenu()
laurent@630: self.ParentWindow.RefreshPageTitles()
lbessard@145:
lbessard@145: def RefreshView(self):
lbessard@145: self.Table.SetData(self.Controler.GetVariables())
lbessard@145: self.Table.ResetView(self.VariablesGrid)
laurent@626: self.VariablesGrid.RefreshButtons()
laurent@626:
laurent@630: def DoGetBestSize(self):
laurent@630: return self.ParentWindow.GetPanelBestSize()
laurent@630:
lbessard@145: def OnVariablesGridCellChange(self, event):
Laurent@1146: row, col = event.GetRow(), event.GetCol()
Laurent@1146: colname = self.Table.GetColLabelValue(col, False)
Laurent@1146: value = self.Table.GetValue(row, col)
Laurent@1146: message = None
Laurent@1146:
Laurent@1146: if colname == "Name" and value != "":
Laurent@1146: if not TestIdentifier(value):
Laurent@1146: message = _("\"%s\" is not a valid identifier!") % value
Laurent@1146: elif value.upper() in IEC_KEYWORDS:
Laurent@1146: message = _("\"%s\" is a keyword. It can't be used!") % value
Laurent@1146: elif value.upper() in [var["Name"].upper()
Laurent@1146: for var_row, var in enumerate(self.Table.data)
Laurent@1146: if var_row != row]:
Laurent@1146: message = _("A variable with \"%s\" as name already exists!") % value
Laurent@1146: else:
Laurent@1146: self.RefreshModel()
Laurent@1146: wx.CallAfter(self.RefreshView)
Laurent@1146: else:
Laurent@1146: self.RefreshModel()
Laurent@1146: wx.CallAfter(self.RefreshView)
Laurent@1146:
Laurent@1146: if message is not None:
Laurent@1146: dialog = wx.MessageDialog(self, message, _("Error"), wx.OK|wx.ICON_ERROR)
Laurent@1146: dialog.ShowModal()
Laurent@1146: dialog.Destroy()
Laurent@1146: event.Veto()
Laurent@1146: else:
Laurent@1146: event.Skip()
lbessard@145:
lbessard@145: def OnVariablesGridEditorShown(self, event):
lbessard@145: row, col = event.GetRow(), event.GetCol()
laurent@801: if self.Table.GetColLabelValue(col, False) == "Type":
lbessard@145: type_menu = wx.Menu(title='')
lbessard@145: base_menu = wx.Menu(title='')
lbessard@145: for base_type in self.Controler.GetBaseTypes():
lbessard@145: new_id = wx.NewId()
Laurent@1096: base_menu.Append(help='', id=new_id, kind=wx.ITEM_NORMAL, text=base_type)
lbessard@145: self.Bind(wx.EVT_MENU, self.GetVariableTypeFunction(base_type), id=new_id)
lbessard@145: type_menu.AppendMenu(wx.NewId(), "Base Types", base_menu)
lbessard@145: datatype_menu = wx.Menu(title='')
Laurent@1097: for datatype in self.Controler.GetDataTypes():
lbessard@145: new_id = wx.NewId()
Laurent@1096: datatype_menu.Append(help='', id=new_id, kind=wx.ITEM_NORMAL, text=datatype)
lbessard@145: self.Bind(wx.EVT_MENU, self.GetVariableTypeFunction(datatype), id=new_id)
lbessard@145: type_menu.AppendMenu(wx.NewId(), "User Data Types", datatype_menu)
lbessard@145: rect = self.VariablesGrid.BlockToDeviceRect((row, col), (row, col))
laurent@708:
lbessard@145: self.VariablesGrid.PopupMenuXY(type_menu, rect.x + rect.width, rect.y + self.VariablesGrid.GetColLabelSize())
laurent@708: type_menu.Destroy()
lbessard@145: event.Veto()
lbessard@145: else:
lbessard@145: event.Skip()
lbessard@145:
lbessard@145: def GetVariableTypeFunction(self, base_type):
lbessard@145: def VariableTypeFunction(event):
lbessard@145: row = self.VariablesGrid.GetGridCursorRow()
lbessard@145: self.Table.SetValueByName(row, "Type", base_type)
lbessard@145: self.Table.ResetView(self.VariablesGrid)
lbessard@145: self.RefreshModel()
lbessard@145: self.RefreshView()
lbessard@145: event.Skip()
lbessard@145: return VariableTypeFunction
lbessard@145:
lbessard@145: def OnVariablesGridCellLeftClick(self, event):
lbessard@145: if event.GetCol() == 0:
lbessard@145: row = event.GetRow()
lbessard@145: data_type = self.Table.GetValueByName(row, "Type")
laurent@401: var_name = self.Table.GetValueByName(row, "Name")
Laurent@1138: data = wx.TextDataObject(str((var_name, "Global", data_type,
Laurent@1138: self.Controler.GetCurrentLocation())))
lbessard@145: dragSource = wx.DropSource(self.VariablesGrid)
lbessard@145: dragSource.SetData(data)
lbessard@145: dragSource.DoDragDrop()
Laurent@874: return
lbessard@145: event.Skip()
lbessard@145:
lbessard@145:
lbessard@145: #-------------------------------------------------------------------------------
Laurent@1096: # CodeFileEditor Main Frame Class
lbessard@145: #-------------------------------------------------------------------------------
lbessard@145:
Laurent@1096: class CodeFileEditor(ConfTreeNodeEditor):
laurent@738:
Laurent@1138: CONFNODEEDITOR_TABS = []
Laurent@1138: CODE_EDITOR = None
Laurent@1138:
Laurent@1138: def _create_CodePanel(self, prnt):
Laurent@1138: self.CodeEditorPanel = wx.SplitterWindow(prnt)
Laurent@1138: self.CodeEditorPanel.SetMinimumPaneSize(1)
Laurent@1138:
Laurent@1138: self.VariablesPanel = VariablesEditor(self.CodeEditorPanel,
Laurent@1138: self.ParentWindow, self.Controler)
Laurent@1138:
Laurent@1138: if self.CODE_EDITOR is not None:
Laurent@1138: self.CodeEditor = self.CODE_EDITOR(self.CodeEditorPanel,
Laurent@1138: self.ParentWindow, self.Controler)
Laurent@1138:
Laurent@1138: self.CodeEditorPanel.SplitHorizontally(self.VariablesPanel,
Laurent@1138: self.CodeEditor, 150)
Laurent@1138: else:
Laurent@1138: self.CodeEditorPanel.Initialize(self.VariablesPanel)
Laurent@1138:
Laurent@1138: return self.CodeEditorPanel
laurent@738:
laurent@630: def __init__(self, parent, controler, window):
laurent@743: ConfTreeNodeEditor.__init__(self, parent, controler, window)
Laurent@1138:
Laurent@1138: wx.CallAfter(self.CodeEditorPanel.SetSashPosition, 150)
laurent@630:
laurent@630: def GetBufferState(self):
laurent@630: return self.Controler.GetBufferState()
laurent@630:
laurent@630: def Undo(self):
laurent@630: self.Controler.LoadPrevious()
laurent@630: self.RefreshView()
laurent@630:
laurent@630: def Redo(self):
laurent@630: self.Controler.LoadNext()
laurent@630: self.RefreshView()
laurent@630:
laurent@630: def RefreshView(self):
laurent@751: ConfTreeNodeEditor.RefreshView(self)
laurent@751:
Laurent@1096: self.VariablesPanel.RefreshView()
Laurent@1138: self.CodeEditor.RefreshView()
Laurent@1138:
Laurent@1138: def Find(self, direction, search_params):
Laurent@1138: self.CodeEditor.Find(direction, search_params)
Laurent@1138: