wxPython/demo/StyledTextCtrl_2.py

   1
   2 import  keyword
   3
   4 import  wx
   5 import  wx.stc  as  stc
   6
   7 import  images
   8
   9 #----------------------------------------------------------------------
  10
  11 demoText = """\
  12 ## This version of the editor has been set up to edit Python source
  13 ## code.  Here is a copy of wxPython/demo/Main.py to play with.
  14
  15
  16 """
  17
  18 #----------------------------------------------------------------------
  19
  20
  21 if wx.Platform == '__WXMSW__':
  22     faces = { 'times': 'Times New Roman',
  23               'mono' : 'Courier New',
  24               'helv' : 'Arial',
  25               'other': 'Comic Sans MS',
  26               'size' : 10,
  27               'size2': 8,
  28              }
  29 elif wx.Platform == '__WXMAC__':
  30     faces = { 'times': 'Times New Roman',
  31               'mono' : 'Courier New',
  32               'helv' : 'Arial',
  33               'other': 'Comic Sans MS',
  34               'size' : 12,
  35               'size2': 10,
  36              }
  37 else:
  38     faces = { 'times': 'Times',
  39               'mono' : 'Courier',
  40               'helv' : 'Helvetica',
  41               'other': 'new century schoolbook',
  42               'size' : 12,
  43               'size2': 10,
  44              }
  45
  46
  47 #----------------------------------------------------------------------
  48
  49 class PythonSTC(stc.StyledTextCtrl):
  50
  51     fold_symbols = 2
  52
  53     def __init__(self, parent, ID,
  54                  pos=wx.DefaultPosition, size=wx.DefaultSize,
  55                  style=0):
  56         stc.StyledTextCtrl.__init__(self, parent, ID, pos, size, style)
  57
  58         self.CmdKeyAssign(ord('B'), stc.STC_SCMOD_CTRL, stc.STC_CMD_ZOOMIN)
  59         self.CmdKeyAssign(ord('N'), stc.STC_SCMOD_CTRL, stc.STC_CMD_ZOOMOUT)
  60
  61         self.SetLexer(stc.STC_LEX_PYTHON)
  62         self.SetKeyWords(0, " ".join(keyword.kwlist))
  63
  64         self.SetProperty("fold", "1")
  65         self.SetProperty("tab.timmy.whinge.level", "1")
  66         self.SetMargins(0,0)
  67
  68         self.SetViewWhiteSpace(False)
  69         #self.SetBufferedDraw(False)
  70         #self.SetViewEOL(True)
  71         #self.SetEOLMode(stc.STC_EOL_CRLF)
  72         #self.SetUseAntiAliasing(True)
  73
  74         self.SetEdgeMode(stc.STC_EDGE_BACKGROUND)
  75         self.SetEdgeColumn(78)
  76
  77         # Setup a margin to hold fold markers
  78         #self.SetFoldFlags(16)  ###  WHAT IS THIS VALUE?  WHAT ARE THE OTHER FLAGS?  DOES IT MATTER?
  79         self.SetMarginType(2, stc.STC_MARGIN_SYMBOL)
  80         self.SetMarginMask(2, stc.STC_MASK_FOLDERS)
  81         self.SetMarginSensitive(2, True)
  82         self.SetMarginWidth(2, 12)
  83
  84         if self.fold_symbols == 0:
  85             # Arrow pointing right for contracted folders, arrow pointing down for expanded
  86             self.MarkerDefine(stc.STC_MARKNUM_FOLDEROPEN,    stc.STC_MARK_ARROWDOWN, "black", "black")
  87             self.MarkerDefine(stc.STC_MARKNUM_FOLDER,        stc.STC_MARK_ARROW, "black", "black")
  88             self.MarkerDefine(stc.STC_MARKNUM_FOLDERSUB,     stc.STC_MARK_EMPTY, "black", "black")
  89             self.MarkerDefine(stc.STC_MARKNUM_FOLDERTAIL,    stc.STC_MARK_EMPTY, "black", "black")
  90             self.MarkerDefine(stc.STC_MARKNUM_FOLDEREND,     stc.STC_MARK_EMPTY,     "white", "black")
  91             self.MarkerDefine(stc.STC_MARKNUM_FOLDEROPENMID, stc.STC_MARK_EMPTY,     "white", "black")
  92             self.MarkerDefine(stc.STC_MARKNUM_FOLDERMIDTAIL, stc.STC_MARK_EMPTY,     "white", "black")
  93
  94         elif self.fold_symbols == 1:
  95             # Plus for contracted folders, minus for expanded
  96             self.MarkerDefine(stc.STC_MARKNUM_FOLDEROPEN,    stc.STC_MARK_MINUS, "white", "black")
  97             self.MarkerDefine(stc.STC_MARKNUM_FOLDER,        stc.STC_MARK_PLUS,  "white", "black")
  98             self.MarkerDefine(stc.STC_MARKNUM_FOLDERSUB,     stc.STC_MARK_EMPTY, "white", "black")
  99             self.MarkerDefine(stc.STC_MARKNUM_FOLDERTAIL,    stc.STC_MARK_EMPTY, "white", "black")
 100             self.MarkerDefine(stc.STC_MARKNUM_FOLDEREND,     stc.STC_MARK_EMPTY, "white", "black")
 101             self.MarkerDefine(stc.STC_MARKNUM_FOLDEROPENMID, stc.STC_MARK_EMPTY, "white", "black")
 102             self.MarkerDefine(stc.STC_MARKNUM_FOLDERMIDTAIL, stc.STC_MARK_EMPTY, "white", "black")
 103
 104         elif self.fold_symbols == 2:
 105             # Like a flattened tree control using circular headers and curved joins
 106             self.MarkerDefine(stc.STC_MARKNUM_FOLDEROPEN,    stc.STC_MARK_CIRCLEMINUS,          "white", "#404040")
 107             self.MarkerDefine(stc.STC_MARKNUM_FOLDER,        stc.STC_MARK_CIRCLEPLUS,           "white", "#404040")
 108             self.MarkerDefine(stc.STC_MARKNUM_FOLDERSUB,     stc.STC_MARK_VLINE,                "white", "#404040")
 109             self.MarkerDefine(stc.STC_MARKNUM_FOLDERTAIL,    stc.STC_MARK_LCORNERCURVE,         "white", "#404040")
 110             self.MarkerDefine(stc.STC_MARKNUM_FOLDEREND,     stc.STC_MARK_CIRCLEPLUSCONNECTED,  "white", "#404040")
 111             self.MarkerDefine(stc.STC_MARKNUM_FOLDEROPENMID, stc.STC_MARK_CIRCLEMINUSCONNECTED, "white", "#404040")
 112             self.MarkerDefine(stc.STC_MARKNUM_FOLDERMIDTAIL, stc.STC_MARK_TCORNERCURVE,         "white", "#404040")
 113
 114         elif self.fold_symbols == 3:
 115             # Like a flattened tree control using square headers
 116             self.MarkerDefine(stc.STC_MARKNUM_FOLDEROPEN,    stc.STC_MARK_BOXMINUS,          "white", "#808080")
 117             self.MarkerDefine(stc.STC_MARKNUM_FOLDER,        stc.STC_MARK_BOXPLUS,           "white", "#808080")
 118             self.MarkerDefine(stc.STC_MARKNUM_FOLDERSUB,     stc.STC_MARK_VLINE,             "white", "#808080")
 119             self.MarkerDefine(stc.STC_MARKNUM_FOLDERTAIL,    stc.STC_MARK_LCORNER,           "white", "#808080")
 120             self.MarkerDefine(stc.STC_MARKNUM_FOLDEREND,     stc.STC_MARK_BOXPLUSCONNECTED,  "white", "#808080")
 121             self.MarkerDefine(stc.STC_MARKNUM_FOLDEROPENMID, stc.STC_MARK_BOXMINUSCONNECTED, "white", "#808080")
 122             self.MarkerDefine(stc.STC_MARKNUM_FOLDERMIDTAIL, stc.STC_MARK_TCORNER,           "white", "#808080")
 123
 124
 125         self.Bind(stc.EVT_STC_UPDATEUI, self.OnUpdateUI)
 126         self.Bind(stc.EVT_STC_MARGINCLICK, self.OnMarginClick)
 127         self.Bind(wx.EVT_KEY_DOWN, self.OnKeyPressed)
 128
 129         # Make some styles,  The lexer defines what each style is used for, we
 130         # just have to define what each style looks like.  This set is adapted from
 131         # Scintilla sample property files.
 132
 133         # Global default styles for all languages
 134         self.StyleSetSpec(stc.STC_STYLE_DEFAULT,     "face:%(helv)s,size:%(size)d" % faces)
 135         self.StyleClearAll()  # Reset all to be like the default
 136
 137         # Global default styles for all languages
 138         self.StyleSetSpec(stc.STC_STYLE_DEFAULT,     "face:%(helv)s,size:%(size)d" % faces)
 139         self.StyleSetSpec(stc.STC_STYLE_LINENUMBER,  "back:#C0C0C0,face:%(helv)s,size:%(size2)d" % faces)
 140         self.StyleSetSpec(stc.STC_STYLE_CONTROLCHAR, "face:%(other)s" % faces)
 141         self.StyleSetSpec(stc.STC_STYLE_BRACELIGHT,  "fore:#FFFFFF,back:#0000FF,bold")
 142         self.StyleSetSpec(stc.STC_STYLE_BRACEBAD,    "fore:#000000,back:#FF0000,bold")
 143
 144         # Python styles
 145         # Default
 146         self.StyleSetSpec(stc.STC_P_DEFAULT, "fore:#000000,face:%(helv)s,size:%(size)d" % faces)
 147         # Comments
 148         self.StyleSetSpec(stc.STC_P_COMMENTLINE, "fore:#007F00,face:%(other)s,size:%(size)d" % faces)
 149         # Number
 150         self.StyleSetSpec(stc.STC_P_NUMBER, "fore:#007F7F,size:%(size)d" % faces)
 151         # String
 152         self.StyleSetSpec(stc.STC_P_STRING, "fore:#7F007F,face:%(helv)s,size:%(size)d" % faces)
 153         # Single quoted string
 154         self.StyleSetSpec(stc.STC_P_CHARACTER, "fore:#7F007F,face:%(helv)s,size:%(size)d" % faces)
 155         # Keyword
 156         self.StyleSetSpec(stc.STC_P_WORD, "fore:#00007F,bold,size:%(size)d" % faces)
 157         # Triple quotes
 158         self.StyleSetSpec(stc.STC_P_TRIPLE, "fore:#7F0000,size:%(size)d" % faces)
 159         # Triple double quotes
 160         self.StyleSetSpec(stc.STC_P_TRIPLEDOUBLE, "fore:#7F0000,size:%(size)d" % faces)
 161         # Class name definition
 162         self.StyleSetSpec(stc.STC_P_CLASSNAME, "fore:#0000FF,bold,underline,size:%(size)d" % faces)
 163         # Function or method name definition
 164         self.StyleSetSpec(stc.STC_P_DEFNAME, "fore:#007F7F,bold,size:%(size)d" % faces)
 165         # Operators
 166         self.StyleSetSpec(stc.STC_P_OPERATOR, "bold,size:%(size)d" % faces)
 167         # Identifiers
 168         self.StyleSetSpec(stc.STC_P_IDENTIFIER, "fore:#000000,face:%(helv)s,size:%(size)d" % faces)
 169         # Comment-blocks
 170         self.StyleSetSpec(stc.STC_P_COMMENTBLOCK, "fore:#7F7F7F,size:%(size)d" % faces)
 171         # End of line where string is not closed
 172         self.StyleSetSpec(stc.STC_P_STRINGEOL, "fore:#000000,face:%(mono)s,back:#E0C0E0,eol,size:%(size)d" % faces)
 173
 174         self.SetCaretForeground("BLUE")
 175
 176
 177         # register some images for use in the AutoComplete box.
 178         self.RegisterImage(1, images.getSmilesBitmap())
 179         self.RegisterImage(2,
 180             wx.ArtProvider.GetBitmap(wx.ART_NEW, size=(16,16)))
 181         self.RegisterImage(3,
 182             wx.ArtProvider.GetBitmap(wx.ART_COPY, size=(16,16)))
 183
 184
 185     def OnKeyPressed(self, event):
 186         if self.CallTipActive():
 187             self.CallTipCancel()
 188         key = event.GetKeyCode()
 189
 190         if key == 32 and event.ControlDown():
 191             pos = self.GetCurrentPos()
 192
 193             # Tips
 194             if event.ShiftDown():
 195                 self.CallTipSetBackground("yellow")
 196                 self.CallTipShow(pos, 'lots of of text: blah, blah, blah\n\n'
 197                                  'show some suff, maybe parameters..\n\n'
 198                                  'fubar(param1, param2)')
 199             # Code completion
 200             else:
 201                 #lst = []
 202                 #for x in range(50000):
 203                 #    lst.append('%05d' % x)
 204                 #st = " ".join(lst)
 205                 #print len(st)
 206                 #self.AutoCompShow(0, st)
 207
 208                 kw = keyword.kwlist[:]
 209                 kw.append("zzzzzz?2")
 210                 kw.append("aaaaa?2")
 211                 kw.append("__init__?3")
 212                 kw.append("zzaaaaa?2")
 213                 kw.append("zzbaaaa?2")
 214                 kw.append("this_is_a_longer_value")
 215                 #kw.append("this_is_a_much_much_much_much_much_much_much_longer_value")
 216
 217                 kw.sort()  # Python sorts are case sensitive
 218                 self.AutoCompSetIgnoreCase(False)  # so this needs to match
 219
 220                 # Images are specified with a appended "?type"
 221                 for i in range(len(kw)):
 222                     if kw[i] in keyword.kwlist:
 223                         kw[i] = kw[i] + "?1"
 224
 225                 self.AutoCompShow(0, " ".join(kw))
 226         else:
 227             event.Skip()
 228
 229
 230     def OnUpdateUI(self, evt):
 231         # check for matching braces
 232         braceAtCaret = -1
 233         braceOpposite = -1
 234         charBefore = None
 235         caretPos = self.GetCurrentPos()
 236
 237         if caretPos > 0:
 238             charBefore = self.GetCharAt(caretPos - 1)
 239             styleBefore = self.GetStyleAt(caretPos - 1)
 240
 241         # check before
 242         if charBefore and chr(charBefore) in "[]{}()" and styleBefore == stc.STC_P_OPERATOR:
 243             braceAtCaret = caretPos - 1
 244
 245         # check after
 246         if braceAtCaret < 0:
 247             charAfter = self.GetCharAt(caretPos)
 248             styleAfter = self.GetStyleAt(caretPos)
 249
 250             if charAfter and chr(charAfter) in "[]{}()" and styleAfter == stc.STC_P_OPERATOR:
 251                 braceAtCaret = caretPos
 252
 253         if braceAtCaret >= 0:
 254             braceOpposite = self.BraceMatch(braceAtCaret)
 255
 256         if braceAtCaret != -1  and braceOpposite == -1:
 257             self.BraceBadLight(braceAtCaret)
 258         else:
 259             self.BraceHighlight(braceAtCaret, braceOpposite)
 260             #pt = self.PointFromPosition(braceOpposite)
 261             #self.Refresh(True, wxRect(pt.x, pt.y, 5,5))
 262             #print pt
 263             #self.Refresh(False)
 264
 265
 266     def OnMarginClick(self, evt):
 267         # fold and unfold as needed
 268         if evt.GetMargin() == 2:
 269             if evt.GetShift() and evt.GetControl():
 270                 self.FoldAll()
 271             else:
 272                 lineClicked = self.LineFromPosition(evt.GetPosition())
 273
 274                 if self.GetFoldLevel(lineClicked) & stc.STC_FOLDLEVELHEADERFLAG:
 275                     if evt.GetShift():
 276                         self.SetFoldExpanded(lineClicked, True)
 277                         self.Expand(lineClicked, True, True, 1)
 278                     elif evt.GetControl():
 279                         if self.GetFoldExpanded(lineClicked):
 280                             self.SetFoldExpanded(lineClicked, False)
 281                             self.Expand(lineClicked, False, True, 0)
 282                         else:
 283                             self.SetFoldExpanded(lineClicked, True)
 284                             self.Expand(lineClicked, True, True, 100)
 285                     else:
 286                         self.ToggleFold(lineClicked)
 287
 288
 289     def FoldAll(self):
 290         lineCount = self.GetLineCount()
 291         expanding = True
 292
 293         # find out if we are folding or unfolding
 294         for lineNum in range(lineCount):
 295             if self.GetFoldLevel(lineNum) & stc.STC_FOLDLEVELHEADERFLAG:
 296                 expanding = not self.GetFoldExpanded(lineNum)
 297                 break
 298
 299         lineNum = 0
 300
 301         while lineNum < lineCount:
 302             level = self.GetFoldLevel(lineNum)
 303             if level & stc.STC_FOLDLEVELHEADERFLAG and \
 304                (level & stc.STC_FOLDLEVELNUMBERMASK) == stc.STC_FOLDLEVELBASE:
 305
 306                 if expanding:
 307                     self.SetFoldExpanded(lineNum, True)
 308                     lineNum = self.Expand(lineNum, True)
 309                     lineNum = lineNum - 1
 310                 else:
 311                     lastChild = self.GetLastChild(lineNum, -1)
 312                     self.SetFoldExpanded(lineNum, False)
 313
 314                     if lastChild > lineNum:
 315                         self.HideLines(lineNum+1, lastChild)
 316
 317             lineNum = lineNum + 1
 318
 319
 320
 321     def Expand(self, line, doExpand, force=False, visLevels=0, level=-1):
 322         lastChild = self.GetLastChild(line, level)
 323         line = line + 1
 324
 325         while line <= lastChild:
 326             if force:
 327                 if visLevels > 0:
 328                     self.ShowLines(line, line)
 329                 else:
 330                     self.HideLines(line, line)
 331             else:
 332                 if doExpand:
 333                     self.ShowLines(line, line)
 334
 335             if level == -1:
 336                 level = self.GetFoldLevel(line)
 337
 338             if level & stc.STC_FOLDLEVELHEADERFLAG:
 339                 if force:
 340                     if visLevels > 1:
 341                         self.SetFoldExpanded(line, True)
 342                     else:
 343                         self.SetFoldExpanded(line, False)
 344
 345                     line = self.Expand(line, doExpand, force, visLevels-1)
 346
 347                 else:
 348                     if doExpand and self.GetFoldExpanded(line):
 349                         line = self.Expand(line, True, force, visLevels-1)
 350                     else:
 351                         line = self.Expand(line, False, force, visLevels-1)
 352             else:
 353                 line = line + 1
 354
 355         return line
 356
 357
 358 #----------------------------------------------------------------------
 359
 360 _USE_PANEL = 1
 361
 362 def runTest(frame, nb, log):
 363     if not _USE_PANEL:
 364         ed = p = PythonSTC(nb, -1)
 365     else:
 366         p = wx.Panel(nb, -1, style = wx.NO_FULL_REPAINT_ON_RESIZE)
 367         ed = PythonSTC(p, -1)
 368         s = wx.BoxSizer(wx.HORIZONTAL)
 369         s.Add(ed, 1, wx.EXPAND)
 370         p.SetSizer(s)
 371         p.SetAutoLayout(True)
 372
 373
 374     ed.SetText(demoText + open('Main.py').read())
 375     ed.EmptyUndoBuffer()
 376     ed.Colourise(0, -1)
 377
 378     # line numbers in the margin
 379     ed.SetMarginType(1, stc.STC_MARGIN_NUMBER)
 380     ed.SetMarginWidth(1, 25)
 381
 382     return p
 383
 384
 385
 386 #----------------------------------------------------------------------
 387
 388
 389 overview = """\
 390 <html><body>
 391 Once again, no docs yet.  <b>Sorry.</b>  But <a href="data/stc.h.html">this</a>
 392 and <a href="http://www.scintilla.org/ScintillaDoc.html">this</a> should
 393 be helpful.
 394 </body><html>
 395 """
 396
 397
 398 if __name__ == '__main__':
 399     import sys,os
 400     import run
 401     run.main(['', os.path.basename(sys.argv[0])] + sys.argv[1:])
 402
 403
 404
 405
 406
 407 #----------------------------------------------------------------------
 408 #----------------------------------------------------------------------
 409