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@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@1097: from graphics.GraphicCommons import ERROR_HIGHLIGHT, SEARCH_RESULT_HIGHLIGHT, REFRESH_HIGHLIGHT_PERIOD lbessard@145: Laurent@1096: SECTIONS_NAMES = ["Includes", "Globals", "Init", Laurent@1096: "CleanUp", "Retrieve", "Publish"] Laurent@1096: 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@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) lbessard@145: lbessard@145: self.SetMarginType(1, stc.STC_MARGIN_NUMBER) lbessard@145: self.SetMarginWidth(1, 25) lbessard@145: lbessard@145: self.CmdKeyAssign(ord('B'), stc.STC_SCMOD_CTRL, stc.STC_CMD_ZOOMIN) lbessard@145: self.CmdKeyAssign(ord('N'), stc.STC_SCMOD_CTRL, stc.STC_CMD_ZOOMOUT) 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) lbessard@145: self.SetEdgeColumn(78) 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: # register some images for use in the AutoComplete box. lbessard@145: #self.RegisterImage(1, images.getSmilesBitmap()) lbessard@145: self.RegisterImage(1, lbessard@145: wx.ArtProvider.GetBitmap(wx.ART_DELETE, size=(16,16))) lbessard@145: self.RegisterImage(2, lbessard@145: wx.ArtProvider.GetBitmap(wx.ART_NEW, size=(16,16))) lbessard@145: self.RegisterImage(3, lbessard@145: wx.ArtProvider.GetBitmap(wx.ART_COPY, size=(16,16))) lbessard@145: lbessard@145: # Indentation size lbessard@145: self.SetTabWidth(2) 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@1097: self.Highlights = [] Laurent@1097: self.SearchParams = None Laurent@1097: self.SearchResults = None Laurent@1097: self.CurrentFindHighlight = None 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@1096: for section in SECTIONS_NAMES: Laurent@1104: section_start_comment = "%s %s section %s" % ( Laurent@1104: self.COMMENT_HEADER, section, self.COMMENT_HEADER) Laurent@1104: section_end_comment = "%s End %s section %s" % ( Laurent@1104: self.COMMENT_HEADER, section, self.COMMENT_HEADER) Laurent@1096: self.SectionsComments[section] = { Laurent@1096: "start": section_start_comment, Laurent@1096: "end": section_end_comment, Laurent@1096: "pattern": re.compile(section_start_comment + Laurent@1096: "(.*)" + Laurent@1096: section_end_comment, Laurent@1096: re.DOTALL) Laurent@1096: } Laurent@1096: 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: 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): lbessard@145: self.ResetBuffer() lbessard@145: 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@1096: for section in SECTIONS_NAMES: Laurent@1096: section_comments = self.SectionsComments[section] Laurent@1096: text += section_comments["start"] Laurent@1109: if not parts[section].startswith("\n") or parts[section] == "\n": Laurent@1104: text += "\n" Laurent@1096: text += parts[section] Laurent@1104: if not parts[section].endswith("\n"): Laurent@1104: text += "\n" Laurent@1096: text += section_comments["end"] Laurent@1104: if section != SECTIONS_NAMES[-1]: Laurent@1104: text += "\n\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@1104: 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@1104: last_styled_pos = None Laurent@1104: end_pos = None Laurent@1104: for section in SECTIONS_NAMES: Laurent@1104: section_comments = self.SectionsComments[section] Laurent@1104: start_pos = text.find(section_comments["start"]) Laurent@1104: end_pos = start_pos + len(section_comments["start"]) Laurent@1104: if last_styled_pos is None: Laurent@1104: last_styled_pos = start_pos Laurent@1104: self.StartStyling(start_pos, 0xff) Laurent@1104: self.SetStyling(end_pos - last_styled_pos, STC_CODE_SECTION) Laurent@1104: for line in xrange(self.LineFromPosition(last_styled_pos), Laurent@1104: self.LineFromPosition(end_pos) + 1): Laurent@1104: self.SetLineState(line, 1) Laurent@1104: start_pos = text.find(section_comments["end"]) Laurent@1104: self.StartStyling(start_pos, 0xff) Laurent@1104: last_styled_pos = start_pos Laurent@1104: end_pos = start_pos + len(section_comments["end"]) Laurent@1104: Laurent@1104: if last_styled_pos is not None and end_pos: Laurent@1104: self.SetStyling(end_pos - last_styled_pos, STC_CODE_SECTION) Laurent@1104: for line in xrange(self.LineFromPosition(last_styled_pos), Laurent@1104: self.LineFromPosition(end_pos) + 1): Laurent@1104: self.SetLineState(line, 1) 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@1096: for section in 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) lbessard@145: lbessard@145: def OnKeyPressed(self, event): lbessard@145: if self.CallTipActive(): lbessard@145: self.CallTipCancel() lbessard@145: key = event.GetKeyCode() Laurent@1104: Laurent@1104: if (self.GetLineState(self.LineFromPosition(self.GetSelection()[0])) and Laurent@1104: key not in NAVIGATION_KEYS): Laurent@1104: return Laurent@1104: 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 lbessard@145: lbessard@145: # Images are specified with a appended "?type" Laurent@1096: self.AutoCompShow(0, " ".join([word + "?1" for word in self.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: lbessard@145: def OnUpdateUI(self, evt): 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) lbessard@145: lbessard@145: def OnMarginClick(self, evt): 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) lbessard@145: 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@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@1097: self.CurrentFindHighlight = self.SearchResults[0] Laurent@1097: 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@1097: highlight_end_pos = end[1] - indent + 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@1097: self.StartStyling(highlight_start_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@848: main_sizer = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=4) laurent@848: main_sizer.AddGrowableCol(0) Laurent@1097: main_sizer.AddGrowableRow(1) laurent@848: laurent@848: controls_sizer = wx.BoxSizer(wx.HORIZONTAL) laurent@848: main_sizer.AddSizer(controls_sizer, border=5, flag=wx.TOP|wx.ALIGN_RIGHT) 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@1097: controls_sizer.AddWindow(button, border=5, flag=wx.RIGHT) Laurent@1097: Laurent@1097: self.VariablesGrid = CustomGrid(self, size=wx.Size(-1, 300), 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@1096: self.VariablesDefaultValue = {"Name" : "", "Type" : "", "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@626: self.Table.InsertRow(new_row, self.VariablesDefaultValue.copy()) 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): lbessard@145: self.RefreshModel() laurent@656: wx.CallAfter(self.RefreshView) lbessard@145: 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@1095: location = "_".join(map(lambda x:str(x), self.Controler.GetCurrentLocation())) Laurent@1095: data = wx.TextDataObject(str(("%s_%s" % (var_name, location), Laurent@1095: "Global", data_type, ""))) 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@920: CONFNODEEDITOR_TABS = [ Laurent@1096: (_("Variables"), "_create_VariablesPanel")] Laurent@1096: Laurent@1096: def _create_VariablesPanel(self, prnt): Laurent@1096: self.VariablesPanel = VariablesEditor(prnt, self.ParentWindow, self.Controler) Laurent@1096: Laurent@1096: return self.VariablesPanel laurent@738: laurent@630: def __init__(self, parent, controler, window): laurent@743: ConfTreeNodeEditor.__init__(self, parent, controler, window) 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@1096: