wxPython/demo/wxStyledTextCtrl_2.py

   1
   2 from wxPython.wx import *
   3 from wxPython.stc import *
   4
   5 import keyword
   6
   7 #----------------------------------------------------------------------
   8
   9 demoText = """\
  10 ## This version of the editor has been set up to edit Python source
  11 ## code.  Here is a copy of wxPython/demo/Main.py to play with.
  12
  13
  14 """
  15
  16 #----------------------------------------------------------------------
  17
  18
  19 if wxPlatform == '__WXMSW__':
  20     faces = { 'times': 'Times New Roman',
  21               'mono' : 'Courier New',
  22               'helv' : 'Arial',
  23               'other': 'Comic Sans MS',
  24               'size' : 10,
  25               'size2': 8,
  26              }
  27 else:
  28     faces = { 'times': 'Times',
  29               'mono' : 'Courier',
  30               'helv' : 'Helvetica',
  31               'other': 'new century schoolbook',
  32               'size' : 13,
  33               'size2': 11,
  34              }
  35
  36
  37 #----------------------------------------------------------------------
  38
  39 class PythonSTC(wxStyledTextCtrl):
  40     def __init__(self, parent, ID):
  41         wxStyledTextCtrl.__init__(self, parent, ID)
  42
  43         self.CmdKeyAssign(ord('B'), wxSTC_SCMOD_CTRL, wxSTC_CMD_ZOOMIN)
  44         self.CmdKeyAssign(ord('N'), wxSTC_SCMOD_CTRL, wxSTC_CMD_ZOOMOUT)
  45
  46         self.SetLexer(wxSTC_LEX_PYTHON)
  47         self.SetKeyWords(0, string.join(keyword.kwlist))
  48
  49         self.SetProperty("fold", "1")
  50         self.SetProperty("tab.timmy.whinge.level", "1")
  51         self.SetMargins(0,0)
  52
  53         self.SetViewWhiteSpace(false)
  54         #self.SetBufferedDraw(false)
  55
  56         self.SetEdgeMode(wxSTC_EDGE_BACKGROUND)
  57         self.SetEdgeColumn(78)
  58
  59         # Setup a margin to hold fold markers
  60         #self.SetFoldFlags(16)  ###  WHAT IS THIS VALUE?  WHAT ARE THE OTHER FLAGS?  DOES IT MATTER?
  61         self.SetMarginType(2, wxSTC_MARGIN_SYMBOL)
  62         self.SetMarginMask(2, wxSTC_MASK_FOLDERS)
  63         self.SetMarginSensitive(2, true)
  64         self.SetMarginWidth(2, 15)
  65         self.MarkerDefine(wxSTC_MARKNUM_FOLDER, wxSTC_MARK_ARROW, "navy", "navy")
  66         self.MarkerDefine(wxSTC_MARKNUM_FOLDEROPEN, wxSTC_MARK_ARROWDOWN, "navy", "navy")
  67
  68
  69         EVT_STC_UPDATEUI(self,    ID, self.OnUpdateUI)
  70         EVT_STC_MARGINCLICK(self, ID, self.OnMarginClick)
  71
  72
  73         # Make some styles,  The lexer defines what each style is used for, we
  74         # just have to define what each style looks like.  This set is adapted from
  75         # Scintilla sample property files.
  76
  77         self.StyleClearAll()
  78
  79         # Global default styles for all languages
  80         self.StyleSetSpec(wxSTC_STYLE_DEFAULT,     "face:%(helv)s,size:%(size)d" % faces)
  81         self.StyleSetSpec(wxSTC_STYLE_LINENUMBER,  "back:#C0C0C0,face:%(helv)s,size:%(size2)d" % faces)
  82         self.StyleSetSpec(wxSTC_STYLE_CONTROLCHAR, "face:%(other)s" % faces)
  83         self.StyleSetSpec(wxSTC_STYLE_BRACELIGHT,  "fore:#FFFFFF,back:#0000FF,bold")
  84         self.StyleSetSpec(wxSTC_STYLE_BRACEBAD,    "fore:#000000,back:#FF0000,bold")
  85
  86         # Python styles
  87         # White space
  88         self.StyleSetSpec(wxSTC_P_DEFAULT, "fore:#808080,face:%(helv)s,size:%(size)d" % faces)
  89         # Comment
  90         self.StyleSetSpec(wxSTC_P_COMMENTLINE, "fore:#007F00,face:%(other)s,size:%(size)d" % faces)
  91         # Number
  92         self.StyleSetSpec(wxSTC_P_NUMBER, "fore:#007F7F,size:%(size)d" % faces)
  93         # String
  94         self.StyleSetSpec(wxSTC_P_STRING, "fore:#7F007F,italic,face:%(times)s,size:%(size)d" % faces)
  95         # Single quoted string
  96         self.StyleSetSpec(wxSTC_P_CHARACTER, "fore:#7F007F,italic,face:%(times)s,size:%(size)d" % faces)
  97         # Keyword
  98         self.StyleSetSpec(wxSTC_P_WORD, "fore:#00007F,bold,size:%(size)d" % faces)
  99         # Triple quotes
 100         self.StyleSetSpec(wxSTC_P_TRIPLE, "fore:#7F0000,size:%(size)d" % faces)
 101         # Triple double quotes
 102         self.StyleSetSpec(wxSTC_P_TRIPLEDOUBLE, "fore:#7F0000,size:%(size)d" % faces)
 103         # Class name definition
 104         self.StyleSetSpec(wxSTC_P_CLASSNAME, "fore:#0000FF,bold,underline,size:%(size)d" % faces)
 105         # Function or method name definition
 106         self.StyleSetSpec(wxSTC_P_DEFNAME, "fore:#007F7F,bold,size:%(size)d" % faces)
 107         # Operators
 108         self.StyleSetSpec(wxSTC_P_OPERATOR, "bold,size:%(size)d" % faces)
 109         # Identifiers
 110         self.StyleSetSpec(wxSTC_P_IDENTIFIER, "fore:#808080,face:%(helv)s,size:%(size)d" % faces)
 111         # Comment-blocks
 112         self.StyleSetSpec(wxSTC_P_COMMENTBLOCK, "fore:#7F7F7F,size:%(size)d" % faces)
 113         # End of line where string is not closed
 114         self.StyleSetSpec(wxSTC_P_STRINGEOL, "fore:#000000,face:%(mono)s,back:#E0C0E0,eol,size:%(size)d" % faces)
 115
 116
 117         self.SetCaretForeground("BLUE")
 118
 119         EVT_KEY_DOWN(self, self.OnKeyPressed)
 120
 121
 122     def OnKeyPressed(self, event):
 123         key = event.KeyCode()
 124         if key == 32 and event.ControlDown():
 125             pos = self.GetCurrentPos()
 126             # Tips
 127             if event.ShiftDown():
 128                 self.CallTipSetBackground("yellow")
 129                 self.CallTipShow(pos, 'param1, param2')
 130             # Code completion
 131             else:
 132                 #lst = []
 133                 #for x in range(50000):
 134                 #    lst.append('%05d' % x)
 135                 #st = string.join(lst)
 136                 #print len(st)
 137                 #self.AutoCompShow(0, st)
 138
 139                 kw = keyword.kwlist[:]
 140                 kw.append("zzzzzz")
 141                 kw.append("aaaaa")
 142                 kw.append("__init__")
 143                 kw.append("zzaaaaa")
 144                 kw.append("zzbaaaa")
 145                 kw.append("this_is_a_longer_value")
 146                 kw.append("this_is_a_much_much_much_much_much_much_much_longer_value")
 147
 148                 kw.sort()  # Python sorts are case sensitive
 149                 self.AutoCompSetIgnoreCase(false)  # so this needs to match
 150
 151                 self.AutoCompShow(0, string.join(kw))
 152         else:
 153             event.Skip()
 154
 155
 156     def OnUpdateUI(self, evt):
 157         # check for matching braces
 158         braceAtCaret = -1
 159         braceOpposite = -1
 160         charBefore = None
 161         caretPos = self.GetCurrentPos()
 162         if caretPos > 0:
 163             charBefore = self.GetCharAt(caretPos - 1)
 164             styleBefore = self.GetStyleAt(caretPos - 1)
 165
 166         # check before
 167         if charBefore and chr(charBefore) in "[]{}()" and styleBefore == wxSTC_P_OPERATOR:
 168             braceAtCaret = caretPos - 1
 169
 170         # check after
 171         if braceAtCaret < 0:
 172             charAfter = self.GetCharAt(caretPos)
 173             styleAfter = self.GetStyleAt(caretPos)
 174             if charAfter and chr(charAfter) in "[]{}()" and styleAfter == wxSTC_P_OPERATOR:
 175                 braceAtCaret = caretPos
 176
 177         if braceAtCaret >= 0:
 178             braceOpposite = self.BraceMatch(braceAtCaret)
 179
 180         if braceAtCaret != -1  and braceOpposite == -1:
 181             self.BraceBadLight(braceAtCaret)
 182         else:
 183             self.BraceHighlight(braceAtCaret, braceOpposite)
 184             #pt = self.PointFromPosition(braceOpposite)
 185             #self.Refresh(true, wxRect(pt.x, pt.y, 5,5))
 186             #print pt
 187             #self.Refresh(false)
 188
 189
 190     def OnMarginClick(self, evt):
 191         # fold and unfold as needed
 192         if evt.GetMargin() == 2:
 193             if evt.GetShift() and evt.GetControl():
 194                 self.FoldAll()
 195             else:
 196                 lineClicked = self.LineFromPosition(evt.GetPosition())
 197                 if self.GetFoldLevel(lineClicked) & wxSTC_FOLDLEVELHEADERFLAG:
 198                     if evt.GetShift():
 199                         self.SetFoldExpanded(lineClicked, true)
 200                         self.Expand(lineClicked, true, true, 1)
 201                     elif evt.GetControl():
 202                         if self.GetFoldExpanded(lineClicked):
 203                             self.SetFoldExpanded(lineClicked, false)
 204                             self.Expand(lineClicked, false, true, 0)
 205                         else:
 206                             self.SetFoldExpanded(lineClicked, true)
 207                             self.Expand(lineClicked, true, true, 100)
 208                     else:
 209                         self.ToggleFold(lineClicked)
 210
 211
 212     def FoldAll(self):
 213         lineCount = self.GetLineCount()
 214         expanding = true
 215
 216         # find out if we are folding or unfolding
 217         for lineNum in range(lineCount):
 218             if self.GetFoldLevel(lineNum) & wxSTC_FOLDLEVELHEADERFLAG:
 219                 expanding = not self.GetFoldExpanded(lineNum)
 220                 break;
 221
 222         lineNum = 0
 223         while lineNum < lineCount:
 224             level = self.GetFoldLevel(lineNum)
 225             if level & wxSTC_FOLDLEVELHEADERFLAG and \
 226                (level & wxSTC_FOLDLEVELNUMBERMASK) == wxSTC_FOLDLEVELBASE:
 227
 228                 if expanding:
 229                     self.SetFoldExpanded(lineNum, true)
 230                     lineNum = self.Expand(lineNum, true)
 231                     lineNum = lineNum - 1
 232                 else:
 233                     lastChild = self.GetLastChild(lineNum, -1)
 234                     self.SetFoldExpanded(lineNum, false)
 235                     if lastChild > lineNum:
 236                         self.HideLines(lineNum+1, lastChild)
 237
 238             lineNum = lineNum + 1
 239
 240
 241
 242     def Expand(self, line, doExpand, force=false, visLevels=0, level=-1):
 243         lastChild = self.GetLastChild(line, level)
 244         line = line + 1
 245         while line <= lastChild:
 246             if force:
 247                 if visLevels > 0:
 248                     self.ShowLines(line, line)
 249                 else:
 250                     self.HideLines(line, line)
 251             else:
 252                 if doExpand:
 253                     self.ShowLines(line, line)
 254
 255             if level == -1:
 256                 level = self.GetFoldLevel(line)
 257
 258             if level & wxSTC_FOLDLEVELHEADERFLAG:
 259                 if force:
 260                     if visLevels > 1:
 261                         self.SetFoldExpanded(line, true)
 262                     else:
 263                         self.SetFoldExpanded(line, false)
 264                     line = self.Expand(line, doExpand, force, visLevels-1)
 265
 266                 else:
 267                     if doExpand and self.GetFoldExpanded(line):
 268                         line = self.Expand(line, true, force, visLevels-1)
 269                     else:
 270                         line = self.Expand(line, false, force, visLevels-1)
 271             else:
 272                 line = line + 1;
 273
 274         return line
 275
 276
 277 #----------------------------------------------------------------------
 278
 279 _USE_PANEL = 1
 280
 281 def runTest(frame, nb, log):
 282     if not _USE_PANEL:
 283         ed = p = PythonSTC(nb, -1)
 284     else:
 285         p = wxPanel(nb, -1)
 286         ed = PythonSTC(p, -1)
 287         s = wxBoxSizer(wxHORIZONTAL)
 288         s.Add(ed, 1, wxEXPAND)
 289         p.SetSizer(s)
 290         p.SetAutoLayout(true)
 291
 292
 293     ed.SetText(demoText + open('Main.py').read())
 294     ed.EmptyUndoBuffer()
 295     ed.Colourise(0, -1)
 296
 297     # line numbers in the margin
 298     ed.SetMarginType(1, wxSTC_MARGIN_NUMBER)
 299     ed.SetMarginWidth(1, 25)
 300
 301     return p
 302
 303
 304
 305 #----------------------------------------------------------------------
 306
 307
 308 overview = """\
 309 <html><body>
 310 Once again, no docs yet.  <b>Sorry.</b>  But <a href="data/stc.h.html">this</a>
 311 and <a href="http://www.scintilla.org/ScintillaDoc.html">this</a> should
 312 be helpful.
 313 </body><html>
 314 """
 315
 316
 317 if __name__ == '__main__':
 318     import sys
 319     app = wxPySimpleApp()
 320     frame = wxFrame(None, -1, "Tester...", size=(640, 480))
 321     win = runTest(frame, frame, sys.stdout)
 322     frame.Show(true)
 323     app.MainLoop()
 324
 325
 326
 327
 328
 329
 330 #----------------------------------------------------------------------
 331 #----------------------------------------------------------------------
 332