[wxWidgets.git] / wxPython / samples / docview / DocViewDemo.py

#----------------------------------------------------------------------------
# Name:         DocViewDemo.py
# Purpose:      Port of the wxWindows docview sample classes
#
# Author:       Peter Yared
#
# Created:      8/1/03
# CVS-ID:       $Id$
# Copyright:    (c) 2003, 2004 ActiveGrid, Inc. (Port of wxWindows classes by Julian Smart et al)
# License:      wxWindows license
#----------------------------------------------------------------------------


#----------------------------------------------------------------------
# This isn't the most object oriented code (it is somewhat repetitive,
# but it is true to the wxWindows C++ demos.
#----------------------------------------------------------------------


import wx
import wx.lib.docview
_ = wx.GetTranslation


#----------------------------------------------------------------------
#----------------------------------------------------------------------
# TextEdit document and view classes

class TextEditDocument(wx.lib.docview.Document):


    def OnSaveDocument(self, filename):
        view = self.GetFirstView()
        if not view.GetTextSW().SaveFile(filename):
            return False
        self.Modify(False)
        self.SetDocumentSaved(True)
##        if wx.Platform == "__WXMAC__":
##            fn = wx.Filename(filename)
##            fn.MacSetDefaultTypeAndCreator()
        return True


    def OnOpenDocument(self, filename):
        view = self.GetFirstView()
        if not view.GetTextSW().LoadFile(filename):
            return False
        self.SetFilename(filename, True)
        self.Modify(False)
        self.UpdateAllViews()
        self._savedYet = True
        return True


    def IsModified(self):
        view = self.GetFirstView()
        if view:
            return wx.lib.docview.Document.IsModified(self) or (view.GetTextSW() and view.GetTextSW().IsModified())
        else:
            return wx.lib.docview.Document.IsModified(self)


    def Modify(self, mod):
        view = self.GetFirstView()
        wx.lib.docview.Document.Modify(self, mod)
        if not mod and view and view.GetTextSW():
            view.GetTextSW().DiscardEdits()


class TextEditView(wx.lib.docview.View):


    def OnCreate(self, doc, flags):
        flags = doc.GetDocumentManager().GetFlags()
        if flags & wx.lib.docview.DOC_SDI and doc.GetDocumentManager().GetMaxDocsOpen() == 1:
            self._frame = wx.GetApp().GetMainFrame()
            self.SetFrame(self._frame)
            sizer = wx.BoxSizer()
            self._textsw = MyTextWindow(self, self._frame, wx.DefaultPosition, wx.DefaultSize, wx.TE_MULTILINE)
            sizer.Add(self._textsw, 1, wx.EXPAND, 0)
            self._frame.SetSizer(sizer)
            self._frame.Layout()
            self.Activate(True)
            return True
        elif flags & wx.lib.docview.DOC_MDI:
            self._frame = wx.lib.docview.DocMDIChildFrame(doc, self, wx.GetApp().GetMainFrame(), -1, wx.GetApp().GetAppName(), (10, 10), (300, 300), wx.DEFAULT_FRAME_STYLE)
            self.SetFrame(self._frame)
            sizer = wx.BoxSizer()
            self._textsw = MyTextWindow(self, self._frame, wx.DefaultPosition, wx.DefaultSize, wx.TE_MULTILINE)
            sizer.Add(self._textsw, 1, wx.EXPAND, 0)
            self._frame.SetSizer(sizer)
            self._frame.Layout()
            self._frame.Show(True)
            self.Activate(True)
            return True
        else:  # flags & wx.lib.docview.DOC_SDI
            self._frame = wx.GetApp().CreateChildFrame(doc, self)
            width, height = self._frame.GetClientSize()
            self._textsw = MyTextWindow(self, self._frame, (0, 0), (width, height), wx.TE_MULTILINE)
            self._frame.SetTitle(_("TextEditView"))
            self._frame.Show(True)
            self.Activate(True)
            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:
            self._textsw.Undo()
            return True
        elif id == wx.ID_REDO:
            self._textsw.Redo()
            return True
        else:
            return wx.lib.docview.View.ProcessEvent(self, event)


    def GetTextSW(self):
        return self._textsw


    def OnDraw(self, dc):
        pass


    def OnClose(self, deleteWindow = True):
        if not self.GetDocument().Close():
            return False
        self.Activate(False)
        if deleteWindow:
            if self.GetDocument().GetDocumentManager().GetMaxDocsOpen() == 1 and self.GetDocument().GetDocumentManager().GetFlags() & wx.lib.docview.DOC_SDI:
                if self._textsw:
                    self._textsw.Destroy()
                return True
            else:
                self._frame.Destroy()
                return True
        return True


class MyTextWindow(wx.TextCtrl):


    def __init__(self, view, frame, pos, size, style):
        wx.TextCtrl.__init__(self, frame, -1, "", pos, size, style)
        self._view = view


#----------------------------------------------------------------------
#----------------------------------------------------------------------
# TextEdit Sample Application


class MyApp(wx.PySimpleApp):


    def OnInit(self):


        demoMode = wx.GetSingleChoiceIndex(_("Select the demo mode"),
                                           _("wxPython DocView Demo"),
                                           [_("SDI Single Document"), _("SDI"), _("MDI")])

        if demoMode == 0 or demoMode == 1:
            flags = wx.lib.docview.DOC_SDI
        elif demoMode == 2:
            flags = wx.lib.docview.DOC_MDI
        else:
            return False

        self.SetAppName(_("DocView Demo"))

        docManager = wx.lib.docview.DocManager(flags = flags)
        docManager.AssociateTemplate(wx.lib.docview.DocTemplate(docManager,
                                                         _("Text"),
                                                         "*.text;*.txt",
                                                         _("Text"),
                                                         _(".txt"),
                                                         _("Text Document"),
                                                         _("Text View"),
                                                         TextEditDocument,
                                                         TextEditView))
        #if wx.Platform == "__WXMAC__":
        #     wx.FileName.MacRegisterDefaultTypeAndCreator("txt", 'TEXT', 'WXMA')

        if demoMode == 0:
            docManager.SetMaxDocsOpen(1)

        if demoMode == 2:  # MDI
            self._frame = MyMDIFrame(docManager, None, -1, _("DocView Demo"), (0, 0), (500, 400), wx.DEFAULT_FRAME_STYLE)
        else:  # SDI
            self._frame = MyFrame(docManager, None, -1, _("DocView Demo"), (0, 0), (500, 400), wx.DEFAULT_FRAME_STYLE)

        fileMenu = wx.Menu()
        editMenu = None

        fileMenu.Append(wx.ID_NEW, _("&New..."))
        fileMenu.Append(wx.ID_OPEN, _("&Open..."))

        if demoMode == 2:  # MDI
            fileMenu.Append(wx.ID_CLOSE, _("&Close"))
            fileMenu.AppendSeparator()

        if demoMode == 0 or demoMode == 2:  # Single Doc or MDI
            fileMenu.Append(wx.ID_SAVE, _("&Save"))
            fileMenu.Append(wx.ID_SAVEAS, _("Save &As"))
            fileMenu.AppendSeparator()
            fileMenu.Append(wx.ID_PRINT, _("&Print"))
            fileMenu.Append(wx.ID_PRINT_SETUP, _("Page &Setup"))
            fileMenu.Append(wx.ID_PREVIEW, _("Print Pre&view"))

            editMenu = wx.Menu()
            editMenu.Append(wx.ID_UNDO, _("&Undo"))
            editMenu.Append(wx.ID_REDO, _("&Redo"))

            self._frame.editMenu = editMenu

        fileMenu.AppendSeparator()
        fileMenu.Append(wx.ID_EXIT, _("E&xit"))

        docManager.FileHistoryUseMenu(fileMenu)

        helpMenu = wx.Menu()
        helpMenu.Append(wx.ID_ABOUT, _("&About"))

        menuBar = wx.MenuBar()
        menuBar.Append(fileMenu, _("&File"))
        if editMenu:
            menuBar.Append(editMenu, _("&Edit"))
        menuBar.Append(helpMenu, _("&Help"))

        self._frame.SetMenuBar(menuBar)
        self._frame.Centre(wx.BOTH)
        self._frame.Show(True)

        self.SetTopWindow(self._frame)

        if demoMode == 0:  # Single doc
            docManager.OnFileNew(None)

        return True


    def GetMainFrame(self):
        return self._frame


    def GetDemoMode(self):
        return self._demoMode


    def CreateChildFrame(self, doc, view):
        subframe = wx.lib.docview.DocChildFrame(doc, view, self.GetMainFrame(), -1, wx.GetApp().GetAppName(), (10, 10), (300, 300), wx.DEFAULT_FRAME_STYLE)

        fileMenu = wx.Menu()
        fileMenu.Append(wx.ID_NEW, _("&New"))
        fileMenu.Append(wx.ID_OPEN, _("&Open"))
        fileMenu.Append(wx.ID_CLOSE, _("&Close"))
        fileMenu.AppendSeparator()
        fileMenu.Append(wx.ID_SAVE, _("&Save"))
        fileMenu.Append(wx.ID_SAVEAS, _("Save &As"))
        fileMenu.AppendSeparator()
        fileMenu.Append(wx.ID_PRINT, _("&Print"))
        fileMenu.Append(wx.ID_PRINT_SETUP, _("Page &Setup"))
        fileMenu.Append(wx.ID_PREVIEW, _("Print Pre&view"))

        editMenu = wx.Menu()
        editMenu.Append(wx.ID_UNDO, _("&Undo"))
        editMenu.Append(wx.ID_REDO, _("&Redo"))

        helpMenu = wx.Menu()
        helpMenu.Append(wx.ID_ABOUT, _("&About"))

        menuBar = wx.MenuBar()

        menuBar.Append(fileMenu, _("&File"))
        menuBar.Append(editMenu, _("&Edit"))
        menuBar.Append(helpMenu, _("&Help"))

        subframe.SetMenuBar(menuBar)
        subframe.Centre(wx.BOTH)
        return subframe


class MyFrame(wx.lib.docview.DocParentFrame):


    def __init__(self, manager, frame, id, title, pos = wx.DefaultPosition, size = wx.DefaultSize, style = wx.DEFAULT_FRAME_STYLE, name = "frame"):
        wx.lib.docview.DocParentFrame.__init__(self, manager, frame, id, title, pos, size, style, name)
        self._frame = frame
        wx.EVT_MENU(self, wx.ID_ABOUT, self.OnAbout)


    def OnAbout(self, event):
        wx.MessageBox(wx.GetApp().GetAppName(), _("About DocView"))


    def GetMainFrame(self):
        return self._frame


class MyMDIFrame(wx.lib.docview.DocMDIParentFrame):

    def __init__(self, manager, frame, id, title, pos = wx.DefaultPosition, size = wx.DefaultSize, style = wx.DEFAULT_FRAME_STYLE, name = "frame"):
        wx.lib.docview.DocMDIParentFrame.__init__(self, manager, frame, id, title, pos, size, style, name)
        self._frame = frame
        wx.EVT_MENU(self, wx.ID_ABOUT, self.OnAbout)


    def OnAbout(self, event):
        wx.MessageBox(wx.GetApp().GetAppName(), _("About DocView"))


    def GetMainFrame(self):
        return self._frame


app = MyApp()
app.MainLoop()
Commit	Line	Data
d1dc2b32 RD	1	#----------------------------------------------------------------------------
	2	# Name: DocViewDemo.py
	3	# Purpose: Port of the wxWindows docview sample classes
	4	#
	5	# Author: Peter Yared
	6	#
	7	# Created: 8/1/03
	8	# CVS-ID: $Id$
	9	# Copyright: (c) 2003, 2004 ActiveGrid, Inc. (Port of wxWindows classes by Julian Smart et al)
	10	# License: wxWindows license
	11	#----------------------------------------------------------------------------
	12
	13
	14
	15	#----------------------------------------------------------------------
	16	# This isn't the most object oriented code (it is somewhat repetitive,
	17	# but it is true to the wxWindows C++ demos.
	18	#----------------------------------------------------------------------
	19
	20
	21	import wx
	22	import wx.lib.docview
	23	_ = wx.GetTranslation
	24
	25
	26	#----------------------------------------------------------------------
	27	#----------------------------------------------------------------------
	28	# TextEdit document and view classes
	29
	30	class TextEditDocument(wx.lib.docview.Document):
	31
	32
	33	def OnSaveDocument(self, filename):
	34	view = self.GetFirstView()
	35	if not view.GetTextSW().SaveFile(filename):
	36	return False
	37	self.Modify(False)
	38	self.SetDocumentSaved(True)
	39	## if wx.Platform == "__WXMAC__":
	40	## fn = wx.Filename(filename)
	41	## fn.MacSetDefaultTypeAndCreator()
	42	return True
	43
	44
	45	def OnOpenDocument(self, filename):
	46	view = self.GetFirstView()
	47	if not view.GetTextSW().LoadFile(filename):
	48	return False
	49	self.SetFilename(filename, True)
	50	self.Modify(False)
	51	self.UpdateAllViews()
	52	self._savedYet = True
	53	return True
	54
	55
	56	def IsModified(self):
	57	view = self.GetFirstView()
	58	if view:
	59	return wx.lib.docview.Document.IsModified(self) or (view.GetTextSW() and view.GetTextSW().IsModified())
	60	else:
	61	return wx.lib.docview.Document.IsModified(self)
	62
	63
	64	def Modify(self, mod):
65	view = self.GetFirstView()
66	wx.lib.docview.Document.Modify(self, mod)
67	if not mod and view and view.GetTextSW():
68	view.GetTextSW().DiscardEdits()
69
70
71	class TextEditView(wx.lib.docview.View):
72
73
74	def OnCreate(self, doc, flags):
75	flags = doc.GetDocumentManager().GetFlags()
76	if flags & wx.lib.docview.DOC_SDI and doc.GetDocumentManager().GetMaxDocsOpen() == 1:
77	self._frame = wx.GetApp().GetMainFrame()
78	self.SetFrame(self._frame)
79	sizer = wx.BoxSizer()
80	self._textsw = MyTextWindow(self, self._frame, wx.DefaultPosition, wx.DefaultSize, wx.TE_MULTILINE)
81	sizer.Add(self._textsw, 1, wx.EXPAND, 0)
82	self._frame.SetSizer(sizer)
83	self._frame.Layout()
84	self.Activate(True)
85	return True
86	elif flags & wx.lib.docview.DOC_MDI:
87	self._frame = wx.lib.docview.DocMDIChildFrame(doc, self, wx.GetApp().GetMainFrame(), -1, wx.GetApp().GetAppName(), (10, 10), (300, 300), wx.DEFAULT_FRAME_STYLE)
88	self.SetFrame(self._frame)
89	sizer = wx.BoxSizer()
90	self._textsw = MyTextWindow(self, self._frame, wx.DefaultPosition, wx.DefaultSize, wx.TE_MULTILINE)
91	sizer.Add(self._textsw, 1, wx.EXPAND, 0)
92	self._frame.SetSizer(sizer)
93	self._frame.Layout()
74b89458	94	self._frame.Show(True)
d1dc2b32 RD	95	self.Activate(True)
	96	return True
	97	else: # flags & wx.lib.docview.DOC_SDI
	98	self._frame = wx.GetApp().CreateChildFrame(doc, self)
	99	width, height = self._frame.GetClientSize()
	100	self._textsw = MyTextWindow(self, self._frame, (0, 0), (width, height), wx.TE_MULTILINE)
	101	self._frame.SetTitle(_("TextEditView"))
	102	self._frame.Show(True)
	103	self.Activate(True)
	104	return True
	105
	106
	107	# Since ProcessEvent is not virtual, we have to trap the relevant events using this pseudo-ProcessEvent instead of EVT_MENU
	108	def ProcessEvent(self, event):
	109	id = event.GetId()
	110	if id == wx.ID_UNDO:
	111	self._textsw.Undo()
	112	return True
	113	elif id == wx.ID_REDO:
	114	self._textsw.Redo()
	115	return True
	116	else:
	117	return wx.lib.docview.View.ProcessEvent(self, event)
	118
	119
	120	def GetTextSW(self):
	121	return self._textsw
	122
	123
	124	def OnDraw(self, dc):
	125	pass
	126
	127
	128	def OnClose(self, deleteWindow = True):
	129	if not self.GetDocument().Close():
	130	return False
	131	self.Activate(False)
	132	if deleteWindow:
	133	if self.GetDocument().GetDocumentManager().GetMaxDocsOpen() == 1 and self.GetDocument().GetDocumentManager().GetFlags() & wx.lib.docview.DOC_SDI:
	134	if self._textsw:
	135	self._textsw.Destroy()
	136	return True
	137	else:
	138	self._frame.Destroy()
	139	return True
	140	return True
	141
	142
	143	class MyTextWindow(wx.TextCtrl):
	144
	145
	146	def __init__(self, view, frame, pos, size, style):
	147	wx.TextCtrl.__init__(self, frame, -1, "", pos, size, style)
	148	self._view = view
	149
	150
	151
	152	#----------------------------------------------------------------------
	153	#----------------------------------------------------------------------
	154	# TextEdit Sample Application
	155
	156
	157	class MyApp(wx.PySimpleApp):
	158
159
160	def OnInit(self):
161
162
163	demoMode = wx.GetSingleChoiceIndex(_("Select the demo mode"),
164	_("wxPython DocView Demo"),
165	[_("SDI Single Document"), _("SDI"), _("MDI")])
166
167	if demoMode == 0 or demoMode == 1:
168	flags = wx.lib.docview.DOC_SDI
169	elif demoMode == 2:
170	flags = wx.lib.docview.DOC_MDI
171	else:
172	return False
173
174	self.SetAppName(_("DocView Demo"))
175
176	docManager = wx.lib.docview.DocManager(flags = flags)
177	docManager.AssociateTemplate(wx.lib.docview.DocTemplate(docManager,
178	_("Text"),
179	".text;.txt",
180	_("Text"),
181	_(".txt"),
182	_("Text Document"),
183	_("Text View"),
184	TextEditDocument,
185	TextEditView))
186	#if wx.Platform == "__WXMAC__":
187	# wx.FileName.MacRegisterDefaultTypeAndCreator("txt", 'TEXT', 'WXMA')
188
189	if demoMode == 0:
190	docManager.SetMaxDocsOpen(1)
191
192	if demoMode == 2: # MDI
193	self._frame = MyMDIFrame(docManager, None, -1, _("DocView Demo"), (0, 0), (500, 400), wx.DEFAULT_FRAME_STYLE)
194	else: # SDI
195	self._frame = MyFrame(docManager, None, -1, _("DocView Demo"), (0, 0), (500, 400), wx.DEFAULT_FRAME_STYLE)
196
197	fileMenu = wx.Menu()
198	editMenu = None
199
200	fileMenu.Append(wx.ID_NEW, _("&New..."))
201	fileMenu.Append(wx.ID_OPEN, _("&Open..."))
202
203	if demoMode == 2: # MDI
204	fileMenu.Append(wx.ID_CLOSE, _("&Close"))
205	fileMenu.AppendSeparator()
206
207	if demoMode == 0 or demoMode == 2: # Single Doc or MDI
208	fileMenu.Append(wx.ID_SAVE, _("&Save"))
209	fileMenu.Append(wx.ID_SAVEAS, _("Save &As"))
210	fileMenu.AppendSeparator()
211	fileMenu.Append(wx.ID_PRINT, _("&Print"))
212	fileMenu.Append(wx.ID_PRINT_SETUP, _("Page &Setup"))
213	fileMenu.Append(wx.ID_PREVIEW, _("Print Pre&view"))
214
215	editMenu = wx.Menu()
216	editMenu.Append(wx.ID_UNDO, _("&Undo"))
217	editMenu.Append(wx.ID_REDO, _("&Redo"))
218
219	self._frame.editMenu = editMenu
220
221	fileMenu.AppendSeparator()
222	fileMenu.Append(wx.ID_EXIT, _("E&xit"))
223
224	docManager.FileHistoryUseMenu(fileMenu)
225
226	helpMenu = wx.Menu()
227	helpMenu.Append(wx.ID_ABOUT, _("&About"))
228
229	menuBar = wx.MenuBar()
230	menuBar.Append(fileMenu, _("&File"))
231	if editMenu:
232	menuBar.Append(editMenu, _("&Edit"))
233	menuBar.Append(helpMenu, _("&Help"))
234
235	self._frame.SetMenuBar(menuBar)
236	self._frame.Centre(wx.BOTH)
237	self._frame.Show(True)
238
239	self.SetTopWindow(self._frame)
240
241	if demoMode == 0: # Single doc
242	docManager.OnFileNew(None)
243
244	return True
245
246
247	def GetMainFrame(self):
248	return self._frame
249
250
251	def GetDemoMode(self):
252	return self._demoMode
253
254
255	def CreateChildFrame(self, doc, view):
256	subframe = wx.lib.docview.DocChildFrame(doc, view, self.GetMainFrame(), -1, wx.GetApp().GetAppName(), (10, 10), (300, 300), wx.DEFAULT_FRAME_STYLE)
257
258	fileMenu = wx.Menu()
259	fileMenu.Append(wx.ID_NEW, _("&New"))
260	fileMenu.Append(wx.ID_OPEN, _("&Open"))
261	fileMenu.Append(wx.ID_CLOSE, _("&Close"))
262	fileMenu.AppendSeparator()
263	fileMenu.Append(wx.ID_SAVE, _("&Save"))
264	fileMenu.Append(wx.ID_SAVEAS, _("Save &As"))
265	fileMenu.AppendSeparator()
266	fileMenu.Append(wx.ID_PRINT, _("&Print"))
267	fileMenu.Append(wx.ID_PRINT_SETUP, _("Page &Setup"))
268	fileMenu.Append(wx.ID_PREVIEW, _("Print Pre&view"))
269
270	editMenu = wx.Menu()
271	editMenu.Append(wx.ID_UNDO, _("&Undo"))
272	editMenu.Append(wx.ID_REDO, _("&Redo"))
273
274	helpMenu = wx.Menu()
275	helpMenu.Append(wx.ID_ABOUT, _("&About"))
276
277	menuBar = wx.MenuBar()
278
279	menuBar.Append(fileMenu, _("&File"))
280	menuBar.Append(editMenu, _("&Edit"))
281	menuBar.Append(helpMenu, _("&Help"))
282
283	subframe.SetMenuBar(menuBar)
284	subframe.Centre(wx.BOTH)
285	return subframe
286
287
288	class MyFrame(wx.lib.docview.DocParentFrame):
289
290
291	def __init__(self, manager, frame, id, title, pos = wx.DefaultPosition, size = wx.DefaultSize, style = wx.DEFAULT_FRAME_STYLE, name = "frame"):
292	wx.lib.docview.DocParentFrame.__init__(self, manager, frame, id, title, pos, size, style, name)
293	self._frame = frame
294	wx.EVT_MENU(self, wx.ID_ABOUT, self.OnAbout)
295
296
297	def OnAbout(self, event):
298	wx.MessageBox(wx.GetApp().GetAppName(), _("About DocView"))
299
300
301	def GetMainFrame(self):
302	return self._frame
303
304
305	class MyMDIFrame(wx.lib.docview.DocMDIParentFrame):
306
307	def __init__(self, manager, frame, id, title, pos = wx.DefaultPosition, size = wx.DefaultSize, style = wx.DEFAULT_FRAME_STYLE, name = "frame"):
308	wx.lib.docview.DocMDIParentFrame.__init__(self, manager, frame, id, title, pos, size, style, name)
309	self._frame = frame
310	wx.EVT_MENU(self, wx.ID_ABOUT, self.OnAbout)
311
312
313	def OnAbout(self, event):
314	wx.MessageBox(wx.GetApp().GetAppName(), _("About DocView"))
315
316
317	def GetMainFrame(self):
318	return self._frame
319
320
321	app = MyApp()
322	app.MainLoop()
323
324