-#----------------------------------------------------------------------------
-# Name: TextEditor.py
-# Purpose: Text Editor for pydocview
-#
-# Author: Peter Yared
-#
-# Created: 8/15/03
-# CVS-ID: $Id$
-# Copyright: (c) 2003-2005 ActiveGrid, Inc.
-# License: wxWindows License
-#----------------------------------------------------------------------------
-import wx
-import wx.lib.docview
-import wx.lib.pydocview
-import string
-import FindService
-_ = wx.GetTranslation
-
-class TextDocument(wx.lib.docview.Document):
-
- def __init__(self):
- wx.lib.docview.Document .__init__(self)
- self._inModify = False
-
-
- def SaveObject(self, fileObject):
- view = self.GetFirstView()
- val = view.GetTextCtrl().GetValue()
- if wx.USE_UNICODE:
- val = val.encode('utf-8')
- fileObject.write(val)
- return True
-
-
- def LoadObject(self, fileObject):
- view = self.GetFirstView()
- data = fileObject.read()
- if wx.USE_UNICODE:
- data = data.decode('utf-8')
- view.GetTextCtrl().SetValue(data)
- return True
-
-
- def IsModified(self):
- view = self.GetFirstView()
- if view and view.GetTextCtrl():
- return view.GetTextCtrl().IsModified()
- return False
-
-
- def Modify(self, modify):
- if self._inModify:
- return
- self._inModify = True
-
- view = self.GetFirstView()
- if not modify and view and view.GetTextCtrl():
- view.GetTextCtrl().DiscardEdits()
-
- wx.lib.docview.Document.Modify(self, modify) # this must called be after the DiscardEdits call above.
-
- self._inModify = False
-
-
-class TextView(wx.lib.docview.View):
-
-
- #----------------------------------------------------------------------------
- # Overridden methods
- #----------------------------------------------------------------------------
-
- def __init__(self):
- wx.lib.docview.View.__init__(self)
- self._textCtrl = None
- self._wordWrap = wx.ConfigBase_Get().ReadInt("TextEditorWordWrap", True)
-
-
- def OnCreate(self, doc, flags):
- frame = wx.GetApp().CreateDocumentFrame(self, doc, flags)
- sizer = wx.BoxSizer()
- font, color = self._GetFontAndColorFromConfig()
- self._textCtrl = self._BuildTextCtrl(frame, font, color = color)
- self._textCtrl.Bind(wx.EVT_TEXT, self.OnModify)
- sizer.Add(self._textCtrl, 1, wx.EXPAND, 0)
- frame.SetSizer(sizer)
- frame.Layout()
- frame.Show(True)
- self.Activate()
- return True
-
-
- def OnModify(self, event):
- self.GetDocument().Modify(True)
-
-
- def _BuildTextCtrl(self, parent, font, color = wx.BLACK, value = "", selection = [0, 0]):
- if self._wordWrap:
- wordWrapStyle = wx.TE_WORDWRAP
- else:
- wordWrapStyle = wx.TE_DONTWRAP
- textCtrl = wx.TextCtrl(parent, -1, pos = wx.DefaultPosition, size = parent.GetClientSize(), style = wx.TE_MULTILINE | wx.TE_RICH | wordWrapStyle)
- textCtrl.SetFont(font)
- textCtrl.SetForegroundColour(color)
- textCtrl.SetValue(value)
- return textCtrl
-
-
- def _GetFontAndColorFromConfig(self):
- font = wx.Font(10, wx.DEFAULT, wx.NORMAL, wx.NORMAL)
- config = wx.ConfigBase_Get()
- fontData = config.Read("TextEditorFont", "")
- if fontData:
- nativeFont = wx.NativeFontInfo()
- nativeFont.FromString(fontData)
- font.SetNativeFontInfo(nativeFont)
- color = wx.BLACK
- colorData = config.Read("TextEditorColor", "")
- if colorData:
- red = int("0x" + colorData[0:2], 16)
- green = int("0x" + colorData[2:4], 16)
- blue = int("0x" + colorData[4:6], 16)
- color = wx.Color(red, green, blue)
- return font, color
-
-
- def OnCreateCommandProcessor(self):
- # Don't create a command processor, it has its own
- pass
-
-
- def OnActivateView(self, activate, activeView, deactiveView):
- if activate and self._textCtrl:
- # In MDI mode just calling set focus doesn't work and in SDI mode using CallAfter causes an endless loop
- if self.GetDocumentManager().GetFlags() & wx.lib.docview.DOC_SDI:
- self._textCtrl.SetFocus()
- else:
- def SetFocusToTextCtrl():
- if self._textCtrl: # Need to make sure it is there in case we are in the closeall mode of the MDI window
- self._textCtrl.SetFocus()
- wx.CallAfter(SetFocusToTextCtrl)
-
-
- def OnUpdate(self, sender = None, hint = None):
- if wx.lib.docview.View.OnUpdate(self, sender, hint):
- return
-
- if hint == "Word Wrap":
- self.SetWordWrap(wx.ConfigBase_Get().ReadInt("TextEditorWordWrap", True))
- elif hint == "Font":
- font, color = self._GetFontAndColorFromConfig()
- self.SetFont(font, color)
-
-
- def OnClose(self, deleteWindow = True):
- if not wx.lib.docview.View.OnClose(self, deleteWindow):
- return False
- self.Activate(False)
- if deleteWindow and self.GetFrame():
- self.GetFrame().Destroy()
- return True
-
-
- # Since ProcessEvent is not virtual, we have to trap the relevant events using this pseudo-ProcessEvent instead of EVT_MENU
- def ProcessEvent(self, event):
- id = event.GetId()
- if id == wx.ID_UNDO:
- if not self._textCtrl:
- return False
- self._textCtrl.Undo()
- return True
- elif id == wx.ID_REDO:
- if not self._textCtrl:
- return False
- self._textCtrl.Redo()
- return True
- elif id == wx.ID_CUT:
- if not self._textCtrl:
- return False
- self._textCtrl.Cut()
- return True
- elif id == wx.ID_COPY:
- if not self._textCtrl:
- return False
- self._textCtrl.Copy()
- return True
- elif id == wx.ID_PASTE:
- if not self._textCtrl:
- return False
- self._textCtrl.Paste()
- return True
- elif id == wx.ID_CLEAR:
- if not self._textCtrl:
- return False
- self._textCtrl.Replace(self._textCtrl.GetSelection()[0], self._textCtrl.GetSelection()[1], '')
- return True
- elif id == wx.ID_SELECTALL:
- if not self._textCtrl:
- return False
- self._textCtrl.SetSelection(-1, -1)
- return True
- elif id == TextService.CHOOSE_FONT_ID:
- if not self._textCtrl:
- return False
- self.OnChooseFont(event)
- return True
- elif id == TextService.WORD_WRAP_ID:
- if not self._textCtrl:
- return False
- self.OnWordWrap(event)
- return True
- elif id == FindService.FindService.FIND_ID:
- self.OnFind()
- return True
- elif id == FindService.FindService.FIND_PREVIOUS_ID:
- self.DoFind(forceFindPrevious = True)
- return True
- elif id == FindService.FindService.FIND_NEXT_ID:
- self.DoFind(forceFindNext = True)
- return True
- elif id == FindService.FindService.REPLACE_ID:
- self.OnFind(replace = True)
- return True
- elif id == FindService.FindService.FINDONE_ID:
- self.DoFind()
- return True
- elif id == FindService.FindService.REPLACEONE_ID:
- self.DoFind(replace = True)
- return True
- elif id == FindService.FindService.REPLACEALL_ID:
- self.DoFind(replaceAll = True)
- return True
- elif id == FindService.FindService.GOTO_LINE_ID:
- self.OnGotoLine(event)
- return True
- else:
- return wx.lib.docview.View.ProcessEvent(self, event)
-
-
- def ProcessUpdateUIEvent(self, event):
- if not self._textCtrl:
- return False
-
- hasText = len(self._textCtrl.GetValue()) > 0
-
- id = event.GetId()
- if id == wx.ID_UNDO:
- event.Enable(self._textCtrl.CanUndo())
- return True
- elif id == wx.ID_REDO:
- event.Enable(self._textCtrl.CanRedo())
- return True
- if id == wx.ID_CUT:
- event.Enable(self._textCtrl.CanCut())
- return True
- elif id == wx.ID_COPY:
- event.Enable(self._textCtrl.CanCopy())
- return True
- elif id == wx.ID_PASTE:
- event.Enable(self._textCtrl.CanPaste())
- return True
- elif id == wx.ID_CLEAR:
- event.Enable(self._textCtrl.CanCopy())
- return True
- elif id == wx.ID_SELECTALL:
- event.Enable(hasText)
- return True
- elif id == TextService.CHOOSE_FONT_ID:
- event.Enable(True)
- return True
- elif id == TextService.WORD_WRAP_ID:
- event.Enable(True)
- return True
- elif id == FindService.FindService.FIND_ID:
- event.Enable(hasText)
- return True
- elif id == FindService.FindService.FIND_PREVIOUS_ID:
- event.Enable(hasText and
- self._FindServiceHasString() and
- self._textCtrl.GetSelection()[0] > 0)
- return True
- elif id == FindService.FindService.FIND_NEXT_ID:
- event.Enable(hasText and
- self._FindServiceHasString() and
- self._textCtrl.GetSelection()[0] < len(self._textCtrl.GetValue()))
- return True
- elif id == FindService.FindService.REPLACE_ID:
- event.Enable(hasText)
- return True
- elif id == FindService.FindService.GOTO_LINE_ID:
- event.Enable(True)
- return True
- else:
- return wx.lib.docview.View.ProcessUpdateUIEvent(self, event)
-
-
- #----------------------------------------------------------------------------
- # Methods for TextDocument to call
- #----------------------------------------------------------------------------
-
- def GetTextCtrl(self):
- return self._textCtrl
-
-
- #----------------------------------------------------------------------------
- # Format methods
- #----------------------------------------------------------------------------
-
- def OnChooseFont(self, event):
- data = wx.FontData()
- data.EnableEffects(True)
- data.SetInitialFont(self._textCtrl.GetFont())
- data.SetColour(self._textCtrl.GetForegroundColour())
- fontDialog = wx.FontDialog(self.GetFrame(), data)
- if fontDialog.ShowModal() == wx.ID_OK:
- data = fontDialog.GetFontData()
- self.SetFont(data.GetChosenFont(), data.GetColour())
- fontDialog.Destroy()
-
-
- def SetFont(self, font, color):
- self._textCtrl.SetFont(font)
- self._textCtrl.SetForegroundColour(color)
- self._textCtrl.Refresh()
- self._textCtrl.Layout()
-
-
- def OnWordWrap(self, event):
- self.SetWordWrap(not self.GetWordWrap())
-
-
- def GetWordWrap(self):
- return self._wordWrap
-
-
- def SetWordWrap(self, wordWrap = True):
- self._wordWrap = wordWrap
- temp = self._textCtrl
- self._textCtrl = self._BuildTextCtrl(temp.GetParent(),
- font = temp.GetFont(),
- color = temp.GetForegroundColour(),
- value = temp.GetValue(),
- selection = temp.GetSelection())
- self.GetDocument().Modify(temp.IsModified())
- temp.Destroy()
-
-
- #----------------------------------------------------------------------------
- # Find methods
- #----------------------------------------------------------------------------
-
- def OnFind(self, replace = False):
- findService = wx.GetApp().GetService(FindService.FindService)
- if findService:
- findService.ShowFindReplaceDialog(findString = self._textCtrl.GetStringSelection(), replace = replace)
-
-
- def DoFind(self, forceFindNext = False, forceFindPrevious = False, replace = False, replaceAll = False):
- findService = wx.GetApp().GetService(FindService.FindService)
- if not findService:
- return
- findString = findService.GetFindString()
- if len(findString) == 0:
- return -1
- replaceString = findService.GetReplaceString()
- flags = findService.GetFlags()
- startLoc, endLoc = self._textCtrl.GetSelection()
-
- wholeWord = flags & wx.FR_WHOLEWORD > 0
- matchCase = flags & wx.FR_MATCHCASE > 0
- regExp = flags & FindService.FindService.FR_REGEXP > 0
- down = flags & wx.FR_DOWN > 0
- wrap = flags & FindService.FindService.FR_WRAP > 0
-
- if forceFindPrevious: # this is from function keys, not dialog box
- down = False
- wrap = False # user would want to know they're at the end of file
- elif forceFindNext:
- down = True
- wrap = False # user would want to know they're at the end of file
-
- # On replace dialog operations, user is allowed to replace the currently highlighted text to determine if it should be replaced or not.
- # Typically, it is the text from a previous find operation, but we must check to see if it isn't, user may have moved the cursor or selected some other text accidentally.
- # If the text is a match, then replace it.
- if replace:
- result, start, end, replText = findService.DoFind(findString, replaceString, self._textCtrl.GetStringSelection(), 0, 0, True, matchCase, wholeWord, regExp, replace)
- if result > 0:
- self._textCtrl.Replace(startLoc, endLoc, replaceString)
- self.GetDocument().Modify(True)
- wx.GetApp().GetTopWindow().PushStatusText(_("1 occurrence of \"%s\" replaced") % findString)
- if down:
- startLoc += len(replText) # advance start location past replacement string to new text
- endLoc = startLoc
-
- text = self._textCtrl.GetValue()
- if wx.Platform == "__WXMSW__":
- text = string.replace(text, '\n', '\r\n')
-
- # Find the next matching text occurance or if it is a ReplaceAll, replace all occurances
- # Even if the user is Replacing, we should replace here, but only select the text and let the user replace it with the next Replace operation
- result, start, end, text = findService.DoFind(findString, replaceString, text, startLoc, endLoc, down, matchCase, wholeWord, regExp, False, replaceAll, wrap)
- if result > 0:
- self._textCtrl.SetValue(text)
- self.GetDocument().Modify(True)
- if result == 1:
- wx.GetApp().GetTopWindow().PushStatusText(_("1 occurrence of \"%s\" replaced") % findString)
- else:
- wx.GetApp().GetTopWindow().PushStatusText(_("%i occurrences of \"%s\" replaced") % (result, findString))
- elif result == 0:
- self._textCtrl.SetSelection(start, end)
- self._textCtrl.SetFocus()
- wx.GetApp().GetTopWindow().PushStatusText(_("Found \"%s\"") % findString)
- else:
- wx.GetApp().GetTopWindow().PushStatusText(_("Can't find \"%s\"") % findString)
-
-
- def _FindServiceHasString(self):
- findService = wx.GetApp().GetService(FindService.FindService)
- if not findService or not findService.GetFindString():
- return False
- return True
-
-
- def OnGotoLine(self, event):
- findService = wx.GetApp().GetService(FindService.FindService)
- if findService:
- line = findService.GetLineNumber(self.GetDocumentManager().FindSuitableParent())
- if line > -1:
- pos = self._textCtrl.XYToPosition(0, line - 1)
- self._textCtrl.SetSelection(pos, pos)
-
-
-class TextService(wx.lib.pydocview.DocService):
-
-
- WORD_WRAP_ID = wx.NewId()
- CHOOSE_FONT_ID = wx.NewId()
-
-
- def __init__(self):
- wx.lib.pydocview.DocService.__init__(self)
-
-
- def InstallControls(self, frame, menuBar = None, toolBar = None, statusBar = None, document = None):
- if document and document.GetDocumentTemplate().GetDocumentType() != TextDocument:
- return
- config = wx.ConfigBase_Get()
-
- formatMenuIndex = menuBar.FindMenu(_("&Format"))
- if formatMenuIndex > -1:
- formatMenu = menuBar.GetMenu(formatMenuIndex)
- else:
- formatMenu = wx.Menu()
- formatMenu = wx.Menu()
- if not menuBar.FindItemById(TextService.WORD_WRAP_ID):
- formatMenu.AppendCheckItem(TextService.WORD_WRAP_ID, _("Word Wrap"), _("Wraps text horizontally when checked"))
- formatMenu.Check(TextService.WORD_WRAP_ID, config.ReadInt("TextEditorWordWrap", True))
- wx.EVT_MENU(frame, TextService.WORD_WRAP_ID, frame.ProcessEvent)
- wx.EVT_UPDATE_UI(frame, TextService.WORD_WRAP_ID, frame.ProcessUpdateUIEvent)
- if not menuBar.FindItemById(TextService.CHOOSE_FONT_ID):
- formatMenu.Append(TextService.CHOOSE_FONT_ID, _("Font..."), _("Sets the font to use"))
- wx.EVT_MENU(frame, TextService.CHOOSE_FONT_ID, frame.ProcessEvent)
- wx.EVT_UPDATE_UI(frame, TextService.CHOOSE_FONT_ID, frame.ProcessUpdateUIEvent)
- if formatMenuIndex == -1:
- viewMenuIndex = menuBar.FindMenu(_("&View"))
- menuBar.Insert(viewMenuIndex + 1, formatMenu, _("&Format"))
-
-
- def ProcessUpdateUIEvent(self, event):
- id = event.GetId()
- if id == TextService.CHOOSE_FONT_ID:
- event.Enable(False)
- return True
- elif id == TextService.WORD_WRAP_ID:
- event.Enable(False)
- return True
- else:
- return False
-
-
-class TextOptionsPanel(wx.Panel):
-
-
- def __init__(self, parent, id):
- wx.Panel.__init__(self, parent, id)
- SPACE = 10
- HALF_SPACE = 5
- config = wx.ConfigBase_Get()
- self._textFont = wx.Font(10, wx.DEFAULT, wx.NORMAL, wx.NORMAL)
- fontData = config.Read("TextEditorFont", "")
- if fontData:
- nativeFont = wx.NativeFontInfo()
- nativeFont.FromString(fontData)
- self._textFont.SetNativeFontInfo(nativeFont)
- self._originalTextFont = self._textFont
- self._textColor = wx.BLACK
- colorData = config.Read("TextEditorColor", "")
- if colorData:
- red = int("0x" + colorData[0:2], 16)
- green = int("0x" + colorData[2:4], 16)
- blue = int("0x" + colorData[4:6], 16)
- self._textColor = wx.Color(red, green, blue)
- self._originalTextColor = self._textColor
- parent.AddPage(self, _("Text"))
- fontLabel = wx.StaticText(self, -1, _("Font:"))
- self._sampleTextCtrl = wx.TextCtrl(self, -1, "", size = (125, -1))
- self._sampleTextCtrl.SetEditable(False)
- chooseFontButton = wx.Button(self, -1, _("Choose Font..."))
- wx.EVT_BUTTON(self, chooseFontButton.GetId(), self.OnChooseFont)
- self._wordWrapCheckBox = wx.CheckBox(self, -1, _("Wrap words inside text area"))
- self._wordWrapCheckBox.SetValue(wx.ConfigBase_Get().ReadInt("TextEditorWordWrap", True))
- textPanelBorderSizer = wx.BoxSizer(wx.VERTICAL)
- textPanelSizer = wx.BoxSizer(wx.VERTICAL)
- textFontSizer = wx.BoxSizer(wx.HORIZONTAL)
- textFontSizer.Add(fontLabel, 0, wx.ALIGN_LEFT | wx.RIGHT | wx.TOP, HALF_SPACE)
- textFontSizer.Add(self._sampleTextCtrl, 0, wx.ALIGN_LEFT | wx.EXPAND | wx.RIGHT, HALF_SPACE)
- textFontSizer.Add(chooseFontButton, 0, wx.ALIGN_RIGHT | wx.LEFT, HALF_SPACE)
- textPanelSizer.Add(textFontSizer, 0, wx.ALL, HALF_SPACE)
- textPanelSizer.Add(self._wordWrapCheckBox, 0, wx.ALL, HALF_SPACE)
- textPanelBorderSizer.Add(textPanelSizer, 0, wx.ALL, SPACE)
- self.SetSizer(textPanelBorderSizer)
- self.UpdateSampleFont()
-
-
- def UpdateSampleFont(self):
- nativeFont = wx.NativeFontInfo()
- nativeFont.FromString(self._textFont.GetNativeFontInfoDesc())
- font = wx.NullFont
- font.SetNativeFontInfo(nativeFont)
- #font.SetPointSize(self._sampleTextCtrl.GetFont().GetPointSize()) # Use the standard point size
- self._sampleTextCtrl.SetFont(font)
- self._sampleTextCtrl.SetForegroundColour(self._textColor)
- self._sampleTextCtrl.SetValue(_("%d pt. %s") % (self._textFont.GetPointSize(), self._textFont.GetFaceName()))
- self._sampleTextCtrl.Refresh()
- self.Layout()
-
-
- def OnChooseFont(self, event):
- data = wx.FontData()
- data.EnableEffects(True)
- data.SetInitialFont(self._textFont)
- data.SetColour(self._textColor)
- fontDialog = wx.FontDialog(self, data)
- if fontDialog.ShowModal() == wx.ID_OK:
- data = fontDialog.GetFontData()
- self._textFont = data.GetChosenFont()
- self._textColor = data.GetColour()
- self.UpdateSampleFont()
- fontDialog.Destroy()
-
-
- def OnOK(self, optionsDialog):
- config = wx.ConfigBase_Get()
- doWordWrapUpdate = config.ReadInt("TextEditorWordWrap", True) != self._wordWrapCheckBox.GetValue()
- config.WriteInt("TextEditorWordWrap", self._wordWrapCheckBox.GetValue())
- doFontUpdate = self._originalTextFont != self._textFont or self._originalTextColor != self._textColor
- config.Write("TextEditorFont", self._textFont.GetNativeFontInfoDesc())
- config.Write("TextEditorColor", "%02x%02x%02x" % (self._textColor.Red(), self._textColor.Green(), self._textColor.Blue()))
- if doWordWrapUpdate or doFontUpdate:
- for document in optionsDialog.GetDocManager().GetDocuments():
- if document.GetDocumentTemplate().GetDocumentType() == TextDocument:
- if doWordWrapUpdate:
- document.UpdateAllViews(hint = "Word Wrap")
- if doFontUpdate:
- document.UpdateAllViews(hint = "Font")