from visual import * # must import visual or vis first
from visual.graph import * # import graphing features
ygraph = gcurve(color=color.yellow)
xgraph = gcurve(color=color.cyan)
gd = gdisplay(x=0,y=-20,width=600, height=150)
h = 20
g = 9.81
m = 25
vball = vector(0,0,0)
pball = m*vball
ball = sphere(pos=vector(0,h,0), radius = .8,color = color.cyan)
ground = box(pos=vector(0,-0.05,-10),size = (5,0.5,10),color = color.green)
deltat = .01
t = 0
while ball.pos.y>.1:
rate(100)
Fgrav = vector(0,-m*g,0)
Fnet = Fgrav
pball = pball + Fnet * deltat
a = vector(0,-g,0)
vball = vball + a*deltat
ball.pos = ball.pos + vball*deltat
PE = m*g*ball.pos
KE = .5 * m * vball.y**2
t = t+deltat
ygraph.plot(pos=(t,ball.pos.y))
xgraph.plot(pos=(t,ball.pos.x))
#TE = PE + KE
print ""
print "Fall time =",t,"PE= ",PE,"KE= ",KE#,"Total Energy = ",TE
print ""
print ""