/Demo/turtle/turtleDemo.py

  1#!/usr/bin/python
  2import sys
  3import os
  4
  5from Tkinter import *
  6from idlelib.Percolator import Percolator
  7from idlelib.ColorDelegator import ColorDelegator
  8from idlelib.textView import TextViewer
  9
 10import turtle
 11import time
 12
 13STARTUP = 1
 14READY = 2
 15RUNNING = 3
 16DONE = 4
 17EVENTDRIVEN = 5
 18
 19menufont = ("Arial", 12, NORMAL)
 20btnfont = ("Arial", 12, 'bold')
 21txtfont = ('Lucida Console', 8, 'normal')
 22
 23def getExampleEntries():
 24    cwd = os.getcwd()
 25    if "turtleDemo.py" not in os.listdir(cwd):
 26        print "Directory of turtleDemo must be current working directory!"
 27        print "But in your case this is", cwd
 28        sys.exit()
 29    entries1 = [entry for entry in os.listdir(cwd) if
 30                     entry.startswith("tdemo_") and
 31                     not entry.endswith(".pyc")]
 32    entries2 = []
 33    for entry in entries1:
 34        if entry.endswith(".py"):
 35            entries2.append(entry)
 36        else:
 37            path = os.path.join(cwd,entry)
 38            sys.path.append(path)
 39            subdir = [entry]
 40            scripts = [script for script in os.listdir(path) if
 41                            script.startswith("tdemo_") and
 42                            script.endswith(".py")]
 43            entries2.append(subdir+scripts)
 44    return entries2
 45
 46def showDemoHelp():
 47    TextViewer(demo.root, "Help on turtleDemo", "demohelp.txt")
 48
 49def showAboutDemo():
 50    TextViewer(demo.root, "About turtleDemo", "about_turtledemo.txt")
 51
 52def showAboutTurtle():
 53    TextViewer(demo.root, "About the new turtle module", "about_turtle.txt")
 54
 55class DemoWindow(object):
 56
 57    def __init__(self, filename=None):   #, root=None):
 58        self.root = root = turtle._root = Tk()
 59        root.wm_protocol("WM_DELETE_WINDOW", self._destroy)
 60
 61        #################
 62        self.mBar = Frame(root, relief=RAISED, borderwidth=2)
 63        self.mBar.pack(fill=X)
 64
 65        self.ExamplesBtn = self.makeLoadDemoMenu()
 66        self.OptionsBtn = self.makeHelpMenu()
 67        self.mBar.tk_menuBar(self.ExamplesBtn, self.OptionsBtn) #, QuitBtn)
 68
 69        root.title('Python turtle-graphics examples')
 70        #################
 71        self.left_frame = left_frame = Frame(root)
 72        self.text_frame = text_frame = Frame(left_frame)
 73        self.vbar = vbar =Scrollbar(text_frame, name='vbar')
 74        self.text = text = Text(text_frame,
 75                                name='text', padx=5, wrap='none',
 76                                width=45)
 77        vbar['command'] = text.yview
 78        vbar.pack(side=LEFT, fill=Y)
 79        #####################
 80        self.hbar = hbar =Scrollbar(text_frame, name='hbar', orient=HORIZONTAL)
 81        hbar['command'] = text.xview
 82        hbar.pack(side=BOTTOM, fill=X)
 83        #####################
 84        text['yscrollcommand'] = vbar.set
 85        text.config(font=txtfont)
 86        text.config(xscrollcommand=hbar.set)
 87        text.pack(side=LEFT, fill=Y, expand=1)
 88        #####################
 89        self.output_lbl = Label(left_frame, height= 1,text=" --- ", bg = "#ddf",
 90                                font = ("Arial", 16, 'normal'))
 91        self.output_lbl.pack(side=BOTTOM, expand=0, fill=X)
 92        #####################
 93        text_frame.pack(side=LEFT, fill=BOTH, expand=0)
 94        left_frame.pack(side=LEFT, fill=BOTH, expand=0)
 95        self.graph_frame = g_frame = Frame(root)
 96
 97        turtle._Screen._root = g_frame
 98        turtle._Screen._canvas = turtle.ScrolledCanvas(g_frame, 800, 600, 1000, 800)
 99        #xturtle.Screen._canvas.pack(expand=1, fill="both")
100        self.screen = _s_ = turtle.Screen()
101        turtle.TurtleScreen.__init__(_s_, _s_._canvas)
102        self.scanvas = _s_._canvas
103        #xturtle.RawTurtle.canvases = [self.scanvas]
104        turtle.RawTurtle.screens = [_s_]
105
106        self.scanvas.pack(side=TOP, fill=BOTH, expand=1)
107
108        self.btn_frame = btn_frame = Frame(g_frame, height=100)
109        self.start_btn = Button(btn_frame, text=" START ", font=btnfont, fg = "white",
110                                disabledforeground = "#fed", command=self.startDemo)
111        self.start_btn.pack(side=LEFT, fill=X, expand=1)
112        self.stop_btn = Button(btn_frame, text=" STOP ",  font=btnfont, fg = "white",
113                                disabledforeground = "#fed", command = self.stopIt)
114        self.stop_btn.pack(side=LEFT, fill=X, expand=1)
115        self.clear_btn = Button(btn_frame, text=" CLEAR ",  font=btnfont, fg = "white",
116                                disabledforeground = "#fed", command = self.clearCanvas)
117        self.clear_btn.pack(side=LEFT, fill=X, expand=1)
118
119        self.btn_frame.pack(side=TOP, fill=BOTH, expand=0)
120        self.graph_frame.pack(side=TOP, fill=BOTH, expand=1)
121
122        Percolator(text).insertfilter(ColorDelegator())
123        self.dirty = False
124        self.exitflag = False
125        if filename:
126            self.loadfile(filename)
127        self.configGUI(NORMAL, DISABLED, DISABLED, DISABLED,
128                       "Choose example from menu", "black")
129        self.state = STARTUP
130
131    def _destroy(self):
132        self.root.destroy()
133        sys.exit()
134
135    def configGUI(self, menu, start, stop, clear, txt="", color="blue"):
136        self.ExamplesBtn.config(state=menu)
137
138        self.start_btn.config(state=start)
139        if start==NORMAL:
140            self.start_btn.config(bg="#d00")
141        else:
142            self.start_btn.config(bg="#fca")
143
144        self.stop_btn.config(state=stop)
145        if stop==NORMAL:
146            self.stop_btn.config(bg="#d00")
147        else:
148            self.stop_btn.config(bg="#fca")
149        self.clear_btn.config(state=clear)
150
151        self.clear_btn.config(state=clear)
152        if clear==NORMAL:
153            self.clear_btn.config(bg="#d00")
154        else:
155            self.clear_btn.config(bg="#fca")
156
157        self.output_lbl.config(text=txt, fg=color)
158
159
160    def makeLoadDemoMenu(self):
161        CmdBtn = Menubutton(self.mBar, text='Examples', underline=0, font=menufont)
162        CmdBtn.pack(side=LEFT, padx="2m")
163        CmdBtn.menu = Menu(CmdBtn)
164
165        for entry in getExampleEntries():
166            def loadexample(x):
167                def emit():
168                    self.loadfile(x)
169                return emit
170            if isinstance(entry,str):
171                CmdBtn.menu.add_command(label=entry[6:-3], underline=0, font=menufont,
172                                        command=loadexample(entry))
173            else:
174                _dir, entries = entry[0], entry[1:]
175                CmdBtn.menu.choices = Menu(CmdBtn.menu)
176                for e in entries:
177                    CmdBtn.menu.choices.add_command(label=e[6:-3], underline=0, font=menufont,
178                              command = loadexample(os.path.join(_dir,e)))
179
180                CmdBtn.menu.add_cascade(label=_dir[6:],
181                                        menu = CmdBtn.menu.choices, font=menufont )
182
183        CmdBtn['menu'] = CmdBtn.menu
184        return CmdBtn
185
186
187    def makeHelpMenu(self):
188        CmdBtn = Menubutton(self.mBar, text='Help', underline=0, font = menufont)
189        CmdBtn.pack(side=LEFT, padx='2m')
190        CmdBtn.menu = Menu(CmdBtn)
191
192        CmdBtn.menu.add_command(label='About turtle.py', font=menufont, command=showAboutTurtle)
193        CmdBtn.menu.add_command(label='turtleDemo - Help', font=menufont, command=showDemoHelp)
194        CmdBtn.menu.add_command(label='About turtleDemo', font=menufont, command=showAboutDemo)
195
196        CmdBtn['menu'] = CmdBtn.menu
197        return CmdBtn
198
199    def refreshCanvas(self):
200        if not self.dirty: return
201        self.screen.clear()
202        #self.screen.mode("standard")
203        self.dirty=False
204
205    def loadfile(self,filename):
206        self.refreshCanvas()
207        if os.path.exists(filename) and not os.path.isdir(filename):
208            # load and display file text
209            f = open(filename,'r')
210            chars = f.read()
211            f.close()
212            self.text.delete("1.0", "end")
213            self.text.insert("1.0",chars)
214            direc, fname = os.path.split(filename)
215            self.root.title(fname[6:-3]+" - an xturtle example")
216            self.module = __import__(fname[:-3])
217            reload(self.module)
218            self.configGUI(NORMAL, NORMAL, DISABLED, DISABLED,
219                           "Press start button", "red")
220            self.state = READY
221
222    def startDemo(self):
223        self.refreshCanvas()
224        self.dirty = True
225        turtle.TurtleScreen._RUNNING = True
226        self.configGUI(DISABLED, DISABLED, NORMAL, DISABLED,
227                       "demo running...", "black")
228        self.screen.clear()
229        self.screen.mode("standard")
230        self.state = RUNNING
231
232        try:
233            result = self.module.main()
234            if result == "EVENTLOOP":
235                self.state = EVENTDRIVEN
236            else:
237                self.state = DONE
238        except turtle.Terminator:
239            self.state = DONE
240            result = "stopped!"
241        if self.state == DONE:
242            self.configGUI(NORMAL, NORMAL, DISABLED, NORMAL,
243                           result)
244        elif self.state == EVENTDRIVEN:
245            self.exitflag = True
246            self.configGUI(DISABLED, DISABLED, NORMAL, DISABLED,
247                           "use mouse/keys or STOP", "red")
248
249    def clearCanvas(self):
250        self.refreshCanvas()
251        self.scanvas.config(cursor="")
252        self.configGUI(NORMAL, NORMAL, DISABLED, DISABLED)
253
254    def stopIt(self):
255        if self.exitflag:
256            self.clearCanvas()
257            self.exitflag = False
258            self.configGUI(NORMAL, NORMAL, DISABLED, DISABLED,
259                           "STOPPED!", "red")
260            turtle.TurtleScreen._RUNNING = False
261            #print "stopIT: exitflag = True"
262        else:
263            turtle.TurtleScreen._RUNNING = False
264            #print "stopIt: exitflag = False"
265
266if __name__ == '__main__':
267    demo = DemoWindow()
268    RUN = True
269    while RUN:
270        try:
271            print "ENTERING mainloop"
272            demo.root.mainloop()
273        except AttributeError:
274            print "CRASH!!!- WAIT A MOMENT!"
275            time.sleep(0.3)
276            print "GOING ON .."
277            demo.refreshCanvas()
278##            time.sleep(1)
279        except:
280            RUN = FALSE