blob: 32677e1ea25e2d19451b173688c7241703b483c6 [file] [log] [blame]
Benjamin Peterson9cf41d02010-03-11 22:33:25 +00001#!/usr/bin/env python
Martin v. Löwis87184592008-06-04 06:29:55 +00002import sys
3import os
4
5from Tkinter import *
6from idlelib.Percolator import Percolator
7from idlelib.ColorDelegator import ColorDelegator
Terry Jan Reedyed6820d2014-06-22 02:32:26 -04008from idlelib.textView import view_file
Martin v. Löwis87184592008-06-04 06:29:55 +00009
10import turtle
11import time
12
Terry Jan Reedyed6820d2014-06-22 02:32:26 -040013demo_dir = os.getcwd()
14if "turtleDemo.py" not in os.listdir(demo_dir):
15 print "Directory of turtleDemo must be current working directory!"
16 print "But in your case this is", demo_dir
17 sys.exit()
18
Martin v. Löwis87184592008-06-04 06:29:55 +000019STARTUP = 1
20READY = 2
21RUNNING = 3
22DONE = 4
23EVENTDRIVEN = 5
24
25menufont = ("Arial", 12, NORMAL)
26btnfont = ("Arial", 12, 'bold')
27txtfont = ('Lucida Console', 8, 'normal')
28
29def getExampleEntries():
Terry Jan Reedyed6820d2014-06-22 02:32:26 -040030 entries1 = [entry for entry in os.listdir(demo_dir) if
Martin v. Löwis87184592008-06-04 06:29:55 +000031 entry.startswith("tdemo_") and
32 not entry.endswith(".pyc")]
33 entries2 = []
34 for entry in entries1:
35 if entry.endswith(".py"):
36 entries2.append(entry)
37 else:
Terry Jan Reedyed6820d2014-06-22 02:32:26 -040038 path = os.path.join(demo_dir, entry)
Martin v. Löwis87184592008-06-04 06:29:55 +000039 sys.path.append(path)
40 subdir = [entry]
41 scripts = [script for script in os.listdir(path) if
42 script.startswith("tdemo_") and
43 script.endswith(".py")]
44 entries2.append(subdir+scripts)
45 return entries2
46
Terry Jan Reedy8ac01bf2014-07-25 01:56:17 -040047help_entries = ( # (help_label, help_file)
48 ('Turtledemo help', "demohelp.txt"),
49 ('About turtledemo', "about_turtledemo.txt"),
50 ('About turtle module', "about_turtle.txt"),
51 )
Martin v. Löwis87184592008-06-04 06:29:55 +000052
53class DemoWindow(object):
54
Terry Jan Reedye2fbf4f2014-07-23 15:01:07 -040055 def __init__(self, filename=None):
Martin v. Löwis87184592008-06-04 06:29:55 +000056 self.root = root = turtle._root = Tk()
Terry Jan Reedye2fbf4f2014-07-23 15:01:07 -040057 root.title('Python turtle-graphics examples')
Martin v. Löwis87184592008-06-04 06:29:55 +000058 root.wm_protocol("WM_DELETE_WINDOW", self._destroy)
59
Terry Jan Reedye2fbf4f2014-07-23 15:01:07 -040060 root.grid_rowconfigure(1, weight=1)
61 root.grid_columnconfigure(0, weight=1)
62 root.grid_columnconfigure(1, minsize=90, weight=1)
63 root.grid_columnconfigure(2, minsize=90, weight=1)
64 root.grid_columnconfigure(3, minsize=90, weight=1)
Martin v. Löwis87184592008-06-04 06:29:55 +000065
Terry Jan Reedye2fbf4f2014-07-23 15:01:07 -040066 self.mBar = Frame(root, relief=RAISED, borderwidth=2)
Martin v. Löwis87184592008-06-04 06:29:55 +000067 self.ExamplesBtn = self.makeLoadDemoMenu()
68 self.OptionsBtn = self.makeHelpMenu()
Terry Jan Reedye2fbf4f2014-07-23 15:01:07 -040069 self.mBar.grid(row=0, columnspan=4, sticky='news')
Martin v. Löwis87184592008-06-04 06:29:55 +000070
Terry Jan Reedye2fbf4f2014-07-23 15:01:07 -040071 pane = PanedWindow(orient=HORIZONTAL, sashwidth=5,
72 sashrelief=SOLID, bg='#ddd')
73 pane.add(self.makeTextFrame(pane))
74 pane.add(self.makeGraphFrame(pane))
75 pane.grid(row=1, columnspan=4, sticky='news')
Martin v. Löwis87184592008-06-04 06:29:55 +000076
Terry Jan Reedye2fbf4f2014-07-23 15:01:07 -040077 self.output_lbl = Label(root, height= 1, text=" --- ", bg="#ddf",
78 font=("Arial", 16, 'normal'), borderwidth=2,
79 relief=RIDGE)
80 self.start_btn = Button(root, text=" START ", font=btnfont,
81 fg="white", disabledforeground = "#fed",
82 command=self.startDemo)
83 self.stop_btn = Button(root, text=" STOP ", font=btnfont,
84 fg="white", disabledforeground = "#fed",
85 command=self.stopIt)
86 self.clear_btn = Button(root, text=" CLEAR ", font=btnfont,
87 fg="white", disabledforeground="#fed",
88 command = self.clearCanvas)
89 self.output_lbl.grid(row=2, column=0, sticky='news', padx=(0,5))
90 self.start_btn.grid(row=2, column=1, sticky='ew')
91 self.stop_btn.grid(row=2, column=2, sticky='ew')
92 self.clear_btn.grid(row=2, column=3, sticky='ew')
Martin v. Löwis87184592008-06-04 06:29:55 +000093
Terry Jan Reedye2fbf4f2014-07-23 15:01:07 -040094 Percolator(self.text).insertfilter(ColorDelegator())
Martin v. Löwis87184592008-06-04 06:29:55 +000095 self.dirty = False
96 self.exitflag = False
97 if filename:
98 self.loadfile(filename)
99 self.configGUI(NORMAL, DISABLED, DISABLED, DISABLED,
100 "Choose example from menu", "black")
101 self.state = STARTUP
102
Terry Jan Reedye2fbf4f2014-07-23 15:01:07 -0400103
104 def onResize(self, event):
105 cwidth = self._canvas.winfo_width()
106 cheight = self._canvas.winfo_height()
107 self._canvas.xview_moveto(0.5*(self.canvwidth-cwidth)/self.canvwidth)
108 self._canvas.yview_moveto(0.5*(self.canvheight-cheight)/self.canvheight)
109
110 def makeTextFrame(self, root):
111 self.text_frame = text_frame = Frame(root)
112 self.text = text = Text(text_frame, name='text', padx=5,
113 wrap='none', width=45)
114
115 self.vbar = vbar = Scrollbar(text_frame, name='vbar')
116 vbar['command'] = text.yview
117 vbar.pack(side=LEFT, fill=Y)
118 self.hbar = hbar = Scrollbar(text_frame, name='hbar', orient=HORIZONTAL)
119 hbar['command'] = text.xview
120 hbar.pack(side=BOTTOM, fill=X)
121
122 text['font'] = txtfont
123 text['yscrollcommand'] = vbar.set
124 text['xscrollcommand'] = hbar.set
125 text.pack(side=LEFT, fill=BOTH, expand=1)
126 return text_frame
127
128 def makeGraphFrame(self, root):
129 turtle._Screen._root = root
130 self.canvwidth = 1000
131 self.canvheight = 800
132 turtle._Screen._canvas = self._canvas = canvas = turtle.ScrolledCanvas(
133 root, 800, 600, self.canvwidth, self.canvheight)
134 canvas.adjustScrolls()
135 canvas._rootwindow.bind('<Configure>', self.onResize)
136 canvas._canvas['borderwidth'] = 0
137
138 self.screen = _s_ = turtle.Screen()
139 turtle.TurtleScreen.__init__(_s_, _s_._canvas)
140 self.scanvas = _s_._canvas
141 turtle.RawTurtle.screens = [_s_]
142 return canvas
Martin v. Löwis87184592008-06-04 06:29:55 +0000143
144 def configGUI(self, menu, start, stop, clear, txt="", color="blue"):
145 self.ExamplesBtn.config(state=menu)
146
147 self.start_btn.config(state=start)
148 if start==NORMAL:
149 self.start_btn.config(bg="#d00")
150 else:
151 self.start_btn.config(bg="#fca")
152
153 self.stop_btn.config(state=stop)
154 if stop==NORMAL:
155 self.stop_btn.config(bg="#d00")
156 else:
157 self.stop_btn.config(bg="#fca")
158 self.clear_btn.config(state=clear)
159
160 self.clear_btn.config(state=clear)
161 if clear==NORMAL:
162 self.clear_btn.config(bg="#d00")
163 else:
164 self.clear_btn.config(bg="#fca")
165
166 self.output_lbl.config(text=txt, fg=color)
167
Martin v. Löwis87184592008-06-04 06:29:55 +0000168 def makeLoadDemoMenu(self):
Terry Jan Reedye2fbf4f2014-07-23 15:01:07 -0400169 CmdBtn = Menubutton(self.mBar, text='Examples',
170 underline=0, font=menufont)
Martin v. Löwis87184592008-06-04 06:29:55 +0000171 CmdBtn.pack(side=LEFT, padx="2m")
172 CmdBtn.menu = Menu(CmdBtn)
173
174 for entry in getExampleEntries():
175 def loadexample(x):
176 def emit():
177 self.loadfile(x)
178 return emit
179 if isinstance(entry,str):
Terry Jan Reedye2fbf4f2014-07-23 15:01:07 -0400180 CmdBtn.menu.add_command(label=entry[6:-3], underline=0,
181 font=menufont,
Martin v. Löwis87184592008-06-04 06:29:55 +0000182 command=loadexample(entry))
183 else:
184 _dir, entries = entry[0], entry[1:]
185 CmdBtn.menu.choices = Menu(CmdBtn.menu)
186 for e in entries:
Terry Jan Reedye2fbf4f2014-07-23 15:01:07 -0400187 CmdBtn.menu.choices.add_command(
188 label=e[6:-3], underline=0, font=menufont,
189 command = loadexample(os.path.join(_dir,e)))
190 CmdBtn.menu.add_cascade(
191 label=_dir[6:], menu = CmdBtn.menu.choices, font=menufont)
Martin v. Löwis87184592008-06-04 06:29:55 +0000192
193 CmdBtn['menu'] = CmdBtn.menu
194 return CmdBtn
195
Martin v. Löwis87184592008-06-04 06:29:55 +0000196 def makeHelpMenu(self):
197 CmdBtn = Menubutton(self.mBar, text='Help', underline=0, font = menufont)
198 CmdBtn.pack(side=LEFT, padx='2m')
199 CmdBtn.menu = Menu(CmdBtn)
200
Terry Jan Reedy8ac01bf2014-07-25 01:56:17 -0400201 for help_label, help_file in help_entries:
202 def show(help_label=help_label, help_file=help_file):
203 view_file(self.root, help_label, os.path.join(demo_dir, help_file))
204 CmdBtn.menu.add_command(label=help_label, font=menufont, command=show)
Martin v. Löwis87184592008-06-04 06:29:55 +0000205
206 CmdBtn['menu'] = CmdBtn.menu
207 return CmdBtn
208
209 def refreshCanvas(self):
210 if not self.dirty: return
211 self.screen.clear()
Martin v. Löwis87184592008-06-04 06:29:55 +0000212 self.dirty=False
213
214 def loadfile(self,filename):
215 self.refreshCanvas()
216 if os.path.exists(filename) and not os.path.isdir(filename):
217 # load and display file text
218 f = open(filename,'r')
219 chars = f.read()
220 f.close()
221 self.text.delete("1.0", "end")
222 self.text.insert("1.0",chars)
223 direc, fname = os.path.split(filename)
Georg Brandl574a8572010-05-24 21:37:54 +0000224 self.root.title(fname[6:-3]+" - a Python turtle graphics example")
Martin v. Löwis87184592008-06-04 06:29:55 +0000225 self.module = __import__(fname[:-3])
226 reload(self.module)
227 self.configGUI(NORMAL, NORMAL, DISABLED, DISABLED,
228 "Press start button", "red")
229 self.state = READY
230
231 def startDemo(self):
232 self.refreshCanvas()
233 self.dirty = True
234 turtle.TurtleScreen._RUNNING = True
235 self.configGUI(DISABLED, DISABLED, NORMAL, DISABLED,
236 "demo running...", "black")
237 self.screen.clear()
238 self.screen.mode("standard")
239 self.state = RUNNING
240
241 try:
242 result = self.module.main()
243 if result == "EVENTLOOP":
244 self.state = EVENTDRIVEN
245 else:
246 self.state = DONE
247 except turtle.Terminator:
248 self.state = DONE
249 result = "stopped!"
250 if self.state == DONE:
251 self.configGUI(NORMAL, NORMAL, DISABLED, NORMAL,
252 result)
253 elif self.state == EVENTDRIVEN:
254 self.exitflag = True
255 self.configGUI(DISABLED, DISABLED, NORMAL, DISABLED,
256 "use mouse/keys or STOP", "red")
257
258 def clearCanvas(self):
259 self.refreshCanvas()
260 self.scanvas.config(cursor="")
261 self.configGUI(NORMAL, NORMAL, DISABLED, DISABLED)
262
263 def stopIt(self):
264 if self.exitflag:
265 self.clearCanvas()
266 self.exitflag = False
267 self.configGUI(NORMAL, NORMAL, DISABLED, DISABLED,
268 "STOPPED!", "red")
269 turtle.TurtleScreen._RUNNING = False
Martin v. Löwis87184592008-06-04 06:29:55 +0000270 else:
271 turtle.TurtleScreen._RUNNING = False
Terry Jan Reedye2fbf4f2014-07-23 15:01:07 -0400272
273 def _destroy(self):
274 self.root.destroy()
Terry Jan Reedycbe6d242014-07-23 17:27:51 -0400275 #sys.exit()
276
277def main():
278 demo = DemoWindow()
279 demo.root.mainloop()
Martin v. Löwis87184592008-06-04 06:29:55 +0000280
281if __name__ == '__main__':
Terry Jan Reedycbe6d242014-07-23 17:27:51 -0400282 main()