#
# Copyright (c) 2011 Tiziano Müller <*****@*****.**>
#
#
#
#

from driven_pendulum import DrivenPendulum
from numpy import pi
from numpy.random import randn
from matplotlib.pyplot import plot, show

pendulum = DrivenPendulum()

samples = 1
npoints = 100
gridsize = 10

#for p in randn(samples, 1):
#    print "calculating sample for p=", p
p = randn()
pendulum.setInitialConditions(0., 0., 1., p, 0.025, 1., 0.025)
x = [0.]*npoints
y = [0.]*npoints
for i in xrange(npoints):
    for j in xrange(gridsize):
        pendulum.evolve(2.*pi/gridsize)
    x[i], y[i] = pendulum.getPhaseSpacePoint()
plot(x, y)
show()
class App:
    def __init__(self, master, height, width):
        from Tkinter import Label, Entry, Canvas, Button
    
        Label(master, text="Initial conditions:").grid(row=0, column=0, columnspan=2)
        Label(master, text="q").grid(row=1, column=0)
        Label(master, text="p").grid(row=2, column=0)
        Label(master, text="t").grid(row=3, column=0)
        Label(master, text="v").grid(row=4, column=0)
        Label(master, text="b").grid(row=5, column=0)
        Label(master, text="l").grid(row=6, column=0)
        Label(master, text="g").grid(row=7, column=0)
        Button(master, text="Reset", command=self.resetInitialConditions).grid(row=8, column=0, columnspan=2)

        self.q = Entry(master)
        self.p = Entry(master)
        self.t = Entry(master)
        self.v = Entry(master)
        self.b = Entry(master)
        self.l = Entry(master)
        self.g = Entry(master)

        self.q.grid(row=1, column=1)
        self.p.grid(row=2, column=1)
        self.t.grid(row=3, column=1)
        self.v.grid(row=4, column=1)
        self.b.grid(row=5, column=1)
        self.l.grid(row=6, column=1)
        self.g.grid(row=7, column=1)

        self.canv = Canvas(master, height=height, width=width)
        self.canv.grid(row=0, column=2, rowspan=8)

        self.pendulum = DrivenPendulum()

        self.q.insert(0, 2.)
        self.p.insert(0, 0.)
        self.t.insert(0, 0.)
        self.v.insert(0, 2.)
        self.b.insert(0, 1.)
        self.l.insert(0, 1.5)
        self.g.insert(0, -10.)

        self.rad = 8
        self.scaleFactor = 50
        self.shiftPosition = array([width//2, height//2])

        self.axes = self.canv.create_line(tuple(self.shiftPosition) + tuple(self.shiftPosition) + tuple(self.shiftPosition))
        self.bob1 = self.canv.create_oval(self.getOvalCoords(self.shiftPosition), fill="blue")
        self.bob2 = self.canv.create_oval(self.getOvalCoords(self.shiftPosition), fill="red")

        self.resetInitialConditions()
        self.updatePendulum()

    def getOvalCoords(self, position):
        return (position[0]-self.rad, position[1]-self.rad, position[0]+self.rad, position[1]+self.rad)

    def updatePendulum(self):

        pos1 = (self.scaleFactor*self.pendulum.getInnerPosition()*array([1, -1])).astype(int) + self.shiftPosition
        pos2 = (self.scaleFactor*self.pendulum.getOuterPosition()*array([1, -1])).astype(int) + self.shiftPosition

        self.canv.coords(self.axes, tuple(self.shiftPosition) + tuple(pos1) + tuple(pos2))
        self.canv.coords(self.bob1, self.getOvalCoords(pos1))
        self.canv.coords(self.bob2, self.getOvalCoords(pos2))

        print self.pendulum.getPhaseSpacePoint()

        self.pendulum.evolve(0.05)
        self.canv.after(20, self.updatePendulum)

    def resetInitialConditions(self):
        self.pendulum.setInitialConditions(
                float(self.q.get()),
                float(self.t.get()),
                float(self.v.get()),
                float(self.p.get()),
                float(self.b.get()),
                float(self.l.get()),
                float(self.g.get()),
                )