spacepaste

Paste details

reply | raw

  1.  
  2. # Chapter 10: Creating Components and Extending Functionality
  3. # Recipe 6: StyledTextCtrl Custom Highlighting
  4. #
  5. import wx
  6. import wx.stc as st
  7. import itertools
  9. issl_table = ';+-?.#~'
  11. if wx.Platform == '__WXMSW__':
  12. # for windows OS
  13. faces = {
  14. 'times': 'Times New Roman',
  15. 'mono' : 'Courier New',
  16. # try temporary switch to mono
  17. 'helv' : 'Courier New',
  18. #'helv' : 'Arial',
  19. 'other': 'Comic Sans MS',
  20. 'size' : 10,
  21. 'size2': 8,
  22. }
  23. else:
  24. faces = {
  25. 'times': 'Times',
  26. 'mono' : 'Courier',
  27. 'helv' : 'Helvetica',
  28. 'other': 'new century schoolbook',
  29. 'size' : 12,
  30. 'size2': 10,
  31. }
  33. class MyMiniFrame(wx.MiniFrame):
  34. def __init__(
  35. self, parent, title, pos=wx.DefaultPosition, size=wx.DefaultSize,
  36. style=wx.DEFAULT_FRAME_STYLE
  37. ):
  39. wx.MiniFrame.__init__(self, parent, -1, title, pos, size, style)
  40. panel = wx.Panel(self, -1)
  42. button = wx.Button(panel, -1, "Close Me")
  43. button.SetPosition((15, 15))
  44. self.Bind(wx.EVT_BUTTON, self.OnCloseMe, button)
  45. self.Bind(wx.EVT_CLOSE, self.OnCloseWindow)
  47. def OnCloseMe(self, event):
  48. self.Close(True)
  50. def OnCloseWindow(self, event):
  51. print "OnCloseWindow"
  52. self.Destroy()
  54. class VowelLexer(object):
  55. """Simple lexer to highlight vowels"""
  56. # Define some style IDs
  57. STC_STYLE_VOWEL_DEFAULT, \
  58. STC_STYLE_VOWEL_KW = range(2)
  59. def __init__(self, lang):
  60. self.lang = lang
  63. def StyleText(self, event):
  64. """Handle the EVT_STC_STYLENEEDED event"""
  65. stc = event.GetEventObject()
  66. # Last correctly styled character
  67. last_styled_pos = stc.GetEndStyled()
  68. # Get styling range for this call
  69. line = stc.LineFromPosition(last_styled_pos)
  70. start_pos = stc.PositionFromLine(line)
  71. end_pos = event.GetPosition()
  73. word = ''
  74. pos = start_pos
  76. def displaystyle(lang, stcvar, stvar):
  77. # set other options ...
  78. stcvar.SetProperty("fold", "1")
  79. stcvar.SetMargins(0, 0)
  80. stcvar.SetViewWhiteSpace(False)
  81. stcvar.SetEdgeMode(stvar.STC_EDGE_BACKGROUND)
  82. stcvar.SetEdgeColumn(78)
  83. stcvar.SetCaretForeground("blue")
  85. # setup a margin to hold the fold markers
  86. stcvar.SetMarginType(2, stvar.STC_MARGIN_SYMBOL)
  87. stcvar.SetMarginMask(2, stvar.STC_MASK_FOLDERS)
  88. stcvar.SetMarginSensitive(2, True)
  89. stcvar.SetMarginWidth(2, 12)
  91. stcvar.SetMarginType(1, stvar.STC_MARGIN_NUMBER)
  92. stcvar.SetMarginWidth(1, 30)
  94. # fold markers use square headers
  95. stcvar.MarkerDefine(stvar.STC_MARKNUM_FOLDEROPEN,
  96. st.STC_MARK_BOXMINUS, "white", "#808080")
  97. stcvar.MarkerDefine(stvar.STC_MARKNUM_FOLDER,
  98. st.STC_MARK_BOXPLUS, "white", "#808080")
  99. stcvar.MarkerDefine(stvar.STC_MARKNUM_FOLDERSUB,
  100. st.STC_MARK_VLINE, "white", "#808080")
  101. stcvar.MarkerDefine(stvar.STC_MARKNUM_FOLDERTAIL,
  102. st.STC_MARK_LCORNER, "white", "#808080")
  103. stcvar.MarkerDefine(stvar.STC_MARKNUM_FOLDEREND,
  104. st.STC_MARK_BOXPLUSCONNECTED, "white", "#808080")
  105. stcvar.MarkerDefine(stvar.STC_MARKNUM_FOLDEROPENMID,
  106. st.STC_MARK_BOXMINUSCONNECTED, "white", "#808080")
  107. stcvar.MarkerDefine(stvar.STC_MARKNUM_FOLDERMIDTAIL,
  108. st.STC_MARK_TCORNER, "white", "#808080")
  111. # make some general styles ...
  112. # global default styles for all languages
  113. # set default font
  114. stcvar.StyleSetSpec(stvar.STC_STYLE_DEFAULT,
  115. "face:%(helv)s,size:%(size)d" % faces)
  116. # set default background color
  117. beige = '#F5F5DC'
  118. stcvar.StyleSetBackground(style=stvar.STC_STYLE_DEFAULT,
  119. back=beige)
  120. # reset all to be like the default
  121. stcvar.StyleClearAll()
  123. # more global default styles for all languages
  124. stcvar.StyleSetSpec(stvar.STC_STYLE_LINENUMBER, "back:#C0C0C0,face:%(helv)s,size:%(size2)d" % faces)
  125. stcvar.StyleSetSpec(stvar.STC_STYLE_CONTROLCHAR, "face:%(other)s" % faces)
  126. stcvar.StyleSetSpec(stvar.STC_STYLE_BRACELIGHT, "fore:#FFFFFF,back:#0000FF,bold")
  127. stcvar.StyleSetSpec(stvar.STC_STYLE_BRACEBAD, "fore:#000000,back:#FF0000,bold")
  128. stcvar.StyleSetSpec(stvar.STC_STYLE_INDENTGUIDE, "fore:#CDCDCD")
  130. # make the Python styles ...
  131. # default
  132. if lang == 'python':
  133. # Python styles
  134. # Default
  135. stcvar.StyleSetSpec(stvar.STC_P_DEFAULT, "fore:#000000,face:%(helv)s,size:%(size)d" % faces)
  136. # Comments
  137. stcvar.StyleSetSpec(stvar.STC_P_COMMENTLINE, "fore:#007F00,face:%(other)s,size:%(size)d" % faces)
  138. # Number
  139. stcvar.StyleSetSpec(stvar.STC_P_NUMBER, "fore:#007F7F,size:%(size)d" % faces)
  140. # String
  141. stcvar.StyleSetSpec(stvar.STC_P_STRING, "fore:#7F007F,face:%(helv)s,size:%(size)d" % faces)
  142. # Single quoted string
  143. stcvar.StyleSetSpec(stvar.STC_P_CHARACTER, "fore:#7F007F,face:%(helv)s,size:%(size)d" % faces)
  144. # Keyword
  145. stcvar.StyleSetSpec(stvar.STC_P_WORD, "fore:#00007F,bold,size:%(size)d" % faces)
  146. # Triple quotes
  147. stcvar.StyleSetSpec(stvar.STC_P_TRIPLE, "fore:#7F0000,size:%(size)d" % faces)
  148. # Triple double quotes
  149. stcvar.StyleSetSpec(stvar.STC_P_TRIPLEDOUBLE, "fore:#7F0000,size:%(size)d" % faces)
  150. # Class name definition
  151. stcvar.StyleSetSpec(stvar.STC_P_CLASSNAME, "fore:#0000FF,bold,underline,size:%(size)d" % faces)
  152. # Function or method name definition
  153. stcvar.StyleSetSpec(stvar.STC_P_DEFNAME, "fore:#007F7F,bold,size:%(size)d" % faces)
  154. # Operators
  155. stcvar.StyleSetSpec(stvar.STC_P_OPERATOR, "bold,size:%(size)d" % faces)
  156. # Identifiers
  157. stcvar.StyleSetSpec(stvar.STC_P_IDENTIFIER, "fore:#000000,face:%(helv)s,size:%(size)d" % faces)
  158. # Comment-blocks
  159. stcvar.StyleSetSpec(stvar.STC_P_COMMENTBLOCK, "fore:#7F7F7F,size:%(size)d" % faces)
  160. # End of line where string is not closed
  161. stcvar.StyleSetSpec(stvar.STC_P_STRINGEOL, "fore:#000000,face:%(mono)s, back:#E0C0E0,eol,size:%(size)d" % faces)
  164. if self.lang == 'python':
  165. pythonlist = [
  166. 'and',
  167. 'as',
  168. 'assert',
  169. 'break',
  170. 'class',
  171. 'continue',
  172. 'def',
  173. 'del',
  174. 'elif',
  175. 'else',
  176. 'except',
  177. 'exec',
  178. 'finally',
  179. 'for',
  180. 'from',
  181. 'global',
  182. 'if',
  183. 'import',
  184. 'in',
  185. 'is',
  186. 'lambda',
  187. 'not',
  188. 'or',
  189. 'pass',
  190. 'print',
  191. 'raise',
  192. 'return',
  193. 'try',
  194. 'while',
  195. 'with',
  196. 'yield',
  197. ]
  199. stc.SetLexer(st.STC_LEX_PYTHON)
  200. stc.SetKeyWords(0, " ".join(pythonlist))
  202. displaystyle(self.lang, stc, st)
  205. class MyPopupMenu(wx.Menu):
  207. def __init__(self, parent):
  208. super(MyPopupMenu, self).__init__()
  210. self.parent = parent
  212. mmi = wx.MenuItem(self, wx.NewId(), 'Remove')
  213. self.AppendItem(mmi)
  214. self.Bind(wx.EVT_MENU, self.OnMinimize, mmi)
  216. # cmi = wx.MenuItem(self, wx.NewId(), 'Close')
  217. # self.AppendItem(cmi)
  218. # self.Bind(wx.EVT_MENU, self.OnClose, cmi)
  221. def OnMinimize(self, e):
  222. StcFrame.panel.Autosizer.Remove(self)
  224. def OnClose(self, e):
  225. self.parent.Close()
  227. class STC(st.StyledTextCtrl):
  228. def __init__(self, *args, **kwargs):
  229. super(STC, self).__init__(*args, **kwargs)
  230. # def __init__(self, *args, **kwargs):
  231. # super(STC, self).__init__(*args, **kwargs)
  233. # Attributes
  234. self.custlex = None
  236. # Event Handlers
  237. self.Bind(wx.stc.EVT_STC_STYLENEEDED, self.OnStyle)
  238. self.InitUI()
  240. def InitUI(self):
  242. self.Bind(wx.EVT_RIGHT_DOWN, self.OnRightDown)
  244. self.SetSize((250, 200))
  245. # self.SetTitle('Context menu')
  246. self.Centre()
  247. self.Show(True)
  249. def OnRightDown(self, e):
  250. self.PopupMenu(MyPopupMenu(self), e.GetPosition())
  252. def OnStyle(self, event):
  253. # Delegate to custom lexer object if one exists
  254. if self.custlex:
  255. self.custlex.StyleText(event)
  256. else:
  257. event.Skip()
  259. def SetLexer(self, lexerid, lexer=None):
  260. """Overrides StyledTextCtrl.SetLexer
  261. Adds optional param to pass in custom container
  262. lexer object.
  263. """
  264. self.custlex = lexer
  265. super(STC, self).SetLexer(lexerid)
  268. class StyledTextApp(wx.App):
  269. def OnInit(self):
  270. self.frame = StcFrame(None, title="Custom Lexer")
  271. self.frame.Show()
  272. return True
  274. class TestPanel(wx.Panel):
  275. def __init__(self, parent):
  276. wx.Panel.__init__(self, parent, -1)
  278. # Attributes
  279. self.stc1 = STC(self)
  280. self.stc1.Text = """print 'this is calling python'
  282. def main():
  283. pass
  285. print 'man'
  287. print 'good'"""
  288. self.stc1.SetUseHorizontalScrollBar(False)
  289. self.stc1.SetUseVerticalScrollBar(False)
  291. style = VowelLexer.STC_STYLE_VOWEL_DEFAULT
  292. self.stc1.StyleSetSpec(style, "fore:#000000")
  293. style = VowelLexer.STC_STYLE_VOWEL_KW
  294. self.stc1.StyleSetSpec(style, "fore:#FF0000,bold")
  295. self.stc1.SetLexer(st.STC_LEX_CONTAINER,
  296. VowelLexer('python')) # python
  299. sizer = wx.BoxSizer(wx.VERTICAL)
  300. sizer.Add(self.stc1, 0, wx.EXPAND)
  301. self.SetSizer(sizer)
  302. self.SetInitialSize((300, 300))
  304. self.Autosizer = wx.FlexGridSizer(rows=1, cols=2, hgap = -190, vgap = -1)
  305. self.Autosizer.SetFlexibleDirection(wx.VERTICAL)
  307. labels = "P".split()
  308. for label in labels:
  309. t=wx.StaticText(self, label=label)
  310. if label == 'P':
  311. t.SetForegroundColour((255,0,255))
  312. e=self.stc1
  313. self.Autosizer.Add(t,0)
  314. self.Autosizer.Add(e,1, wx.EXPAND)
  315. self.Autosizer.AddGrowableCol(1, 1) # 2 col
  316. self.SetSizer(self.Autosizer)
  317. self.Fit()
  320. self.Bind(wx.EVT_SCROLLWIN, self.OnScroll1)
  321. self.Bind(wx.EVT_SCROLLWIN, self.OnScroll2)
  322. self.Bind(wx.EVT_SCROLLWIN, self.OnScroll3)
  324. def OnScroll2(self, event):
  325. clone = event.Clone()
  326. clone.SetId(wx.NewId())
  327. clone.SetEventObject(self.stc1)
  328. self.stc1.GetEventHandler().ProcessEvent(clone)
  329. event.Skip()
  331. def OnScroll1(self, event):
  332. clone = event.Clone()
  333. clone.SetId(wx.NewId())
  334. clone.SetEventObject(self.stc2)
  335. self.stc2.GetEventHandler().ProcessEvent(clone)
  336. event.Skip()
  338. def OnScroll3(self, event):
  339. clone = event.Clone()
  340. clone.SetId(wx.NewId())
  341. clone.SetEventObject(self.stc3)
  342. self.stc3.GetEventHandler().ProcessEvent(clone)
  343. event.Skip()
  345. class StcFrame(wx.Frame):
  346. """Main application window"""
  347. def __init__(self, parent, *args, **kwargs):
  348. super(StcFrame, self).__init__(parent,
  349. *args,
  350. **kwargs)
  352. self.panel = TestPanel(self)
  355. b = wx.Button(self, 40, "Fortran", (300,200), style=wx.NO_BORDER)
  356. b.SetToolTipString("This button has a style flag of wx.NO_BORDER.\nOn some platforms that will give it a flattened look.")
  357. self.Bind(wx.EVT_BUTTON, self.OnClick, b)
  364. def OnClick(self, event):
  365. t=wx.StaticText(self, label='F')
  366. e= STC(self)
  367. e.Text = """"""
  368. e.SetUseHorizontalScrollBar(False)
  369. e.SetUseVerticalScrollBar(False)
  371. self.SetSizer(self.panel.Autosizer)
  372. self.panel.Layout()
  374. if __name__ == '__main__':
  375. app = StyledTextApp(False)
  376. app.MainLoop()
  377.  
Powered by spacepaste. Paste removal.