#!/usr/bin/python

import sys

import os

from Tkinter import *

from idlelib.Percolator import Percolator

from idlelib.ColorDelegator import ColorDelegator

from idlelib.textView import TextViewer

import turtle

import time

STARTUP = 1

READY = 2

RUNNING = 3

DONE = 4

EVENTDRIVEN = 5

menufont = ("Arial", 12, NORMAL)

btnfont = ("Arial", 12, 'bold')

txtfont = ('Lucida Console', 8, 'normal')

def getExampleEntries():

    cwd = os.getcwd()

    if "turtleDemo.py" not in os.listdir(cwd):

        print "Directory of turtleDemo must be current working directory!"

        print "But in your case this is", cwd

        sys.exit()

    entries1 = [entry for entry in os.listdir(cwd) if

                     entry.startswith("tdemo_") and

                     not entry.endswith(".pyc")]

    entries2 = []

    for entry in entries1:

        if entry.endswith(".py"):

            entries2.append(entry)

        else:

            path = os.path.join(cwd,entry)

            sys.path.append(path)

            subdir = [entry]

            scripts = [script for script in os.listdir(path) if

                            script.startswith("tdemo_") and

                            script.endswith(".py")]

            entries2.append(subdir+scripts)

    return entries2

def showDemoHelp():

    TextViewer(demo.root, "Help on turtleDemo", "demohelp.txt")

def showAboutDemo():

    TextViewer(demo.root, "About turtleDemo", "about_turtledemo.txt")

def showAboutTurtle():

    TextViewer(demo.root, "About the new turtle module", "about_turtle.txt")

class DemoWindow(object):

    def __init__(self, filename=None):   #, root=None):

        self.root = root = turtle._root = Tk()

        root.wm_protocol("WM_DELETE_WINDOW", self._destroy)

        #################

        self.mBar = Frame(root, relief=RAISED, borderwidth=2)

        self.mBar.pack(fill=X)

        self.ExamplesBtn = self.makeLoadDemoMenu()

        self.OptionsBtn = self.makeHelpMenu()

        self.mBar.tk_menuBar(self.ExamplesBtn, self.OptionsBtn) #, QuitBtn)

        root.title('Python turtle-graphics examples')

        #################

        self.left_frame = left_frame = Frame(root)

        self.text_frame = text_frame = Frame(left_frame)

        self.vbar = vbar =Scrollbar(text_frame, name='vbar')

        self.text = text = Text(text_frame,

                                name='text', padx=5, wrap='none',

                                width=45)

        vbar['command'] = text.yview

        vbar.pack(side=LEFT, fill=Y)

        #####################

        self.hbar = hbar =Scrollbar(text_frame, name='hbar', orient=HORIZONTAL)

        hbar['command'] = text.xview

        hbar.pack(side=BOTTOM, fill=X)

        #####################

        text['yscrollcommand'] = vbar.set

        text.config(font=txtfont)

        text.config(xscrollcommand=hbar.set)

        text.pack(side=LEFT, fill=Y, expand=1)

        #####################

        self.output_lbl = Label(left_frame, height= 1,text=" --- ", bg = "#ddf",

                                font = ("Arial", 16, 'normal'))

        self.output_lbl.pack(side=BOTTOM, expand=0, fill=X)

        #####################

        text_frame.pack(side=LEFT, fill=BOTH, expand=0)

        left_frame.pack(side=LEFT, fill=BOTH, expand=0)

        self.graph_frame = g_frame = Frame(root)

        turtle._Screen._root = g_frame

        turtle._Screen._canvas = turtle.ScrolledCanvas(g_frame, 800, 600, 1000, 800)

        #xturtle.Screen._canvas.pack(expand=1, fill="both")

        self.screen = _s_ = turtle.Screen()

        self.scanvas = _s_._canvas

        #xturtle.RawTurtle.canvases = [self.scanvas]

        turtle.RawTurtle.screens = [_s_]

        self.scanvas.pack(side=TOP, fill=BOTH, expand=1)

        self.btn_frame = btn_frame = Frame(g_frame, height=100)

        self.start_btn = Button(btn_frame, text=" START ", font=btnfont, fg = "white",

                                disabledforeground = "#fed", command=self.startDemo)

        self.start_btn.pack(side=LEFT, fill=X, expand=1)

        self.stop_btn = Button(btn_frame, text=" STOP ",  font=btnfont, fg = "white",

                                disabledforeground = "#fed", command = self.stopIt)

        self.stop_btn.pack(side=LEFT, fill=X, expand=1)

        self.clear_btn = Button(btn_frame, text=" CLEAR ",  font=btnfont, fg = "white",

                                disabledforeground = "#fed", command = self.clearCanvas)

        self.clear_btn.pack(side=LEFT, fill=X, expand=1)

        self.btn_frame.pack(side=TOP, fill=BOTH, expand=0)

        self.graph_frame.pack(side=TOP, fill=BOTH, expand=1)

        Percolator(text).insertfilter(ColorDelegator())

        self.dirty = False

        self.exitflag = False

        if filename:

            self.loadfile(filename)

        self.configGUI(NORMAL, DISABLED, DISABLED, DISABLED,

                       "Choose example from menu", "black")

        self.state = STARTUP

    def _destroy(self):

        self.root.destroy()

        sys.exit()

    def configGUI(self, menu, start, stop, clear, txt="", color="blue"):

        self.ExamplesBtn.config(state=menu)

        self.start_btn.config(state=start)

        if start==NORMAL:

            self.start_btn.config(bg="#d00")

        else:

            self.start_btn.config(bg="#fca")

        self.stop_btn.config(state=stop)

        if stop==NORMAL:

            self.stop_btn.config(bg="#d00")

        else:

            self.stop_btn.config(bg="#fca")

        self.clear_btn.config(state=clear)

        self.clear_btn.config(state=clear)

        if clear==NORMAL:

            self.clear_btn.config(bg="#d00")

        else:

            self.clear_btn.config(bg="#fca")

        self.output_lbl.config(text=txt, fg=color)

    def makeLoadDemoMenu(self):

        CmdBtn = Menubutton(self.mBar, text='Examples', underline=0, font=menufont)

        CmdBtn.pack(side=LEFT, padx="2m")

        CmdBtn.menu = Menu(CmdBtn)

        for entry in getExampleEntries():

            def loadexample(x):

                def emit():

                    self.loadfile(x)

                return emit

            if isinstance(entry,str):

                CmdBtn.menu.add_command(label=entry[6:-3], underline=0, font=menufont,

                                        command=loadexample(entry))

            else:

                _dir, entries = entry[0], entry[1:]

                CmdBtn.menu.choices = Menu(CmdBtn.menu)

                for e in entries:

                    CmdBtn.menu.choices.add_command(label=e[6:-3], underline=0, font=menufont,

                              command = loadexample(os.path.join(_dir,e)))

                CmdBtn.menu.add_cascade(label=_dir[6:],

                                        menu = CmdBtn.menu.choices, font=menufont )

        CmdBtn['menu'] = CmdBtn.menu

        return CmdBtn

    def makeHelpMenu(self):

        CmdBtn = Menubutton(self.mBar, text='Help', underline=0, font = menufont)

        CmdBtn.pack(side=LEFT, padx='2m')

        CmdBtn.menu = Menu(CmdBtn)

        CmdBtn.menu.add_command(label='About turtle.py', font=menufont, command=showAboutTurtle)

        CmdBtn.menu.add_command(label='turtleDemo - Help', font=menufont, command=showDemoHelp)

        CmdBtn.menu.add_command(label='About turtleDemo', font=menufont, command=showAboutDemo)

        CmdBtn['menu'] = CmdBtn.menu

        return CmdBtn

    def refreshCanvas(self):

        if not self.dirty: return

        self.screen.clear()

        #self.screen.mode("standard")

        self.dirty=False

    def loadfile(self,filename):

        self.refreshCanvas()

        if os.path.exists(filename) and not os.path.isdir(filename):

            # load and display file text

            f = open(filename,'r')

            chars = f.read()

            f.close()

            self.text.delete("1.0", "end")

            self.text.insert("1.0",chars)

            direc, fname = os.path.split(filename)

            self.root.title(fname[6:-3]+" - an xturtle example")

            self.module = __import__(fname[:-3])

            reload(self.module)

            self.configGUI(NORMAL, NORMAL, DISABLED, DISABLED,

                           "Press start button", "red")

            self.state = READY

    def startDemo(self):

        self.refreshCanvas()

        self.dirty = True

        turtle.TurtleScreen._RUNNING = True

        self.configGUI(DISABLED, DISABLED, NORMAL, DISABLED,

                       "demo running...", "black")

        self.screen.clear()

        self.screen.mode("standard")

        self.state = RUNNING

        try:

            result = self.module.main()

            if result == "EVENTLOOP":

                self.state = EVENTDRIVEN

            else:

                self.state = DONE

        except turtle.Terminator:

            self.state = DONE

            result = "stopped!"

        if self.state == DONE:

            self.configGUI(NORMAL, NORMAL, DISABLED, NORMAL,

                           result)

        elif self.state == EVENTDRIVEN:

            self.exitflag = True

            self.configGUI(DISABLED, DISABLED, NORMAL, DISABLED,

                           "use mouse/keys or STOP", "red")

    def clearCanvas(self):

        self.refreshCanvas()

        self.scanvas.config(cursor="")

        self.configGUI(NORMAL, NORMAL, DISABLED, DISABLED)

    def stopIt(self):

        if self.exitflag:

            self.clearCanvas()

            self.exitflag = False

            self.configGUI(NORMAL, NORMAL, DISABLED, DISABLED,

                           "STOPPED!", "red")

            turtle.TurtleScreen._RUNNING = False

            #print "stopIT: exitflag = True"

        else:

            turtle.TurtleScreen._RUNNING = False

            #print "stopIt: exitflag = False"

if __name__ == '__main__':

    demo = DemoWindow()

    RUN = True

    while RUN:

        try:

            print "ENTERING mainloop"

            demo.root.mainloop()

        except AttributeError:

            print "CRASH!!!- WAIT A MOMENT!"

            time.sleep(0.3)

            print "GOING ON .."

            demo.refreshCanvas()

##            time.sleep(1)

        except:

            RUN = FALSE