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@738: from controls import CustomGrid, CustomTable
Laurent@1096: from editors.ConfTreeNodeEditor import ConfTreeNodeEditor
Laurent@814: from util.BitmapLibrary import GetBitmap
Laurent@1091: from controls.CustomStyledTextCtrl import CustomStyledTextCtrl, faces, GetCursorPos
lbessard@145: 
Laurent@1096: SECTIONS_NAMES = ["Includes", "Globals", "Init",
Laurent@1096:                   "CleanUp", "Retrieve", "Publish"]
Laurent@1096: 
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)
lbessard@145:         self.StyleSetSpec(stc.STC_STYLE_CONTROLCHAR, "face:%(other)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:         
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@1096:         self.SectionsComments = {}
Laurent@1096:         for section in SECTIONS_NAMES:
Laurent@1096:             section_start_comment = "%s %s section\n" % (self.COMMENT_HEADER, section)
Laurent@1096:             section_end_comment = "\n%s End %s section\n\n" % (self.COMMENT_HEADER, section)
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)
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@1096:             text += parts[section]
Laurent@1096:             text += section_comments["end"]
Laurent@1096:         return text
Laurent@1096: 
lbessard@145:     def RefreshView(self):
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()
lbessard@145:         self.SetText(new_text)
Laurent@1060:         if old_text != 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:         
lbessard@145:         self.Colourise(0, -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()
lbessard@145: 
lbessard@145:         if 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: 
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@848:         main_sizer.AddGrowableRow(0)
laurent@848:         
Laurent@920:         self.VariablesGrid = CustomGrid(self, size=wx.Size(-1, 300), style=wx.VSCROLL)
laurent@848:         self.VariablesGrid.Bind(wx.grid.EVT_GRID_CELL_CHANGE, self.OnVariablesGridCellChange)
laurent@848:         self.VariablesGrid.Bind(wx.grid.EVT_GRID_CELL_LEFT_CLICK, self.OnVariablesGridCellLeftClick)
laurent@848:         self.VariablesGrid.Bind(wx.grid.EVT_GRID_EDITOR_SHOWN, self.OnVariablesGridEditorShown)
laurent@848:         main_sizer.AddWindow(self.VariablesGrid, flag=wx.GROW)
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@848:             controls_sizer.AddWindow(button, border=5, flag=wx.LEFT)
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@610:             for datatype in self.Controler.GetDataTypes(basetypes=False, only_locatables=True):
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: