From 17d1898b2908f7b57072aab149f5f7535a0559ba Mon Sep 17 00:00:00 2001
From: ccremers <ccremers@4f7d145f-a0d4-0310-ae10-d96cc04433c1>
Date: Tue, 30 Jan 2007 11:01:17 +0000
Subject: [PATCH] - Factored out editor component.

---
 gui/Gui/Editor.py     | 82 +++++++++++++++++++++++++++++++++++++++++++
 gui/Gui/Mainwindow.py | 51 +++++----------------------
 2 files changed, 90 insertions(+), 43 deletions(-)
 create mode 100644 gui/Gui/Editor.py

diff --git a/gui/Gui/Editor.py b/gui/Gui/Editor.py
new file mode 100644
index 0000000..6a15fbc
--- /dev/null
+++ b/gui/Gui/Editor.py
@@ -0,0 +1,82 @@
+#!/usr/bin/python
+
+#---------------------------------------------------------------------------
+
+""" Import externals """
+import wx
+
+# Use Scintilla editor?
+useStc = True       # It looks nicer!
+#useStc = False      # It is sometimes buggy, claims the internet
+
+# Test Scintilla and if it fails, get rid of it
+if useStc:
+    try:
+        from wx.stc import *
+    except:
+        useStc = False
+
+#---------------------------------------------------------------------------
+
+""" Import scyther-gui components """
+
+#---------------------------------------------------------------------------
+
+""" Some constants """
+
+#---------------------------------------------------------------------------
+
+def selectEditor(parent):
+    """
+    Pick an editor (Scintilla or default) and return the object.
+    """
+    if useStc:
+        return EditorStc(parent)
+    else:
+        return EditorNormal(parent)
+
+#---------------------------------------------------------------------------
+
+class Editor(object):
+
+    def __init__(self, parent):
+        # Empty start
+        self.SetText("")
+
+#---------------------------------------------------------------------------
+
+class EditorNormal(Editor):
+
+    def __init__(self, parent):
+        self.control = wx.TextCtrl(parent, style=wx.TE_MULTILINE)
+
+        # Call parent
+        Editor.__init__(self,parent)
+
+    def GetText(self):
+        return self.control.GetValue()
+
+    def SetText(self, txt):
+        self.control.SetValue(txt)
+
+#---------------------------------------------------------------------------
+
+class EditorStc(Editor):
+
+    def __init__(self, parent):
+        # Scintilla layout with line numbers
+        self.control = StyledTextCtrl(parent)
+        self.control.SetMarginType(1, STC_MARGIN_NUMBER)
+        self.control.SetMarginWidth(1, 30)
+
+        # Call parent
+        Editor.__init__(self,parent)
+
+    def GetText(self):
+        return self.control.GetText()
+
+    def SetText(self, txt):
+        self.control.SetText(txt)
+
+#---------------------------------------------------------------------------
+
diff --git a/gui/Gui/Mainwindow.py b/gui/Gui/Mainwindow.py
index 0d0f78f..a96bea5 100644
--- a/gui/Gui/Mainwindow.py
+++ b/gui/Gui/Mainwindow.py
@@ -6,17 +6,6 @@
 import wx
 import os.path
 
-# Use Scintilla editor?
-useStc = True       # It looks nicer!
-#useStc = False      # It is sometimes buggy!
-
-# Test Scintilla and if it fails, get rid of it
-if useStc:
-    try:
-        import wx.stc as stc
-    except:
-        useStc = False
-
 #---------------------------------------------------------------------------
 
 """ Import scyther-gui components """
@@ -24,6 +13,7 @@ import Settingswindow
 import Scytherthread
 import Icon
 import About
+import Editor
 
 #---------------------------------------------------------------------------
 
@@ -86,24 +76,6 @@ class MainWindow(wx.Frame):
 
         self.firstCommand()
 
-    def GetText(self):
-        """
-        Interface for both types of editor
-        """
-        if useStc:
-            return self.control.GetText()
-        else:
-            return self.control.GetValue()
-
-    def SetText(self,txt):
-        """
-        Interface for both types of editor
-        """
-        if useStc:
-            self.control.SetText(txt)
-        else:
-            self.control.SetValue(txt)
-        
     def CreateInteriorWindowComponents(self):
         ''' Create "interior" window components. In this case it is just a
             simple multiline text control. '''
@@ -121,22 +93,15 @@ class MainWindow(wx.Frame):
 
         # Top: input
         self.top = wx.Notebook(self,-1)
-
-        if useStc:
-            # Scintilla layout with line numbers
-            self.control = stc.StyledTextCtrl(self.top)
-            self.control.SetMarginType(1, stc.STC_MARGIN_NUMBER)
-            self.control.SetMarginWidth(1, 30)
-        else:
-            # Simpler default
-            self.control = wx.TextCtrl(self.top, style=wx.TE_MULTILINE)
+        # Editor there
+        self.editor = Editor.selectEditor(self.top)
 
         if self.load:
             textfile = open(os.path.join(self.dirname, self.filename), 'r')
-            self.SetText(textfile.read())
+            self.editor.SetText(textfile.read())
             os.chdir(self.dirname)
             textfile.close()
-        self.top.AddPage(self.control,"Protocol description")
+        self.top.AddPage(self.editor.control,"Protocol description")
         self.settings = Settingswindow.SettingsWindow(self.top,self)
         self.top.AddPage(self.settings,"Settings")
 
@@ -227,14 +192,14 @@ class MainWindow(wx.Frame):
 
     def OnSave(self, event):
         textfile = open(os.path.join(self.dirname, self.filename), 'w')
-        textfile.write(self.GetText())
+        textfile.write(self.editor.GetText())
         textfile.close()
 
     def OnOpen(self, event):
         if self.askUserForFilename(style=wx.OPEN,
                                    **self.defaultFileDialogOptions()):
             textfile = open(os.path.join(self.dirname, self.filename), 'r')
-            self.SetText(textfile.read())
+            self.editor.SetText(textfile.read())
             textfile.close()
 
     def OnSaveAs(self, event):
@@ -244,7 +209,7 @@ class MainWindow(wx.Frame):
             os.chdir(self.dirname)
 
     def RunScyther(self, mode):
-        spdl = self.GetText()
+        spdl = self.editor.GetText()
         s = Scytherthread.ScytherRun(self,mode,spdl)
 
     def OnVerify(self, event):