from __future__ import division
from visual.graph import *
scene.y = 400 # moves animation window down 400 pixels to make room for graph# Constants
pi = 3.14159
L0 = .08 # equilibrium length of spring (not stretched)
g = 0 # gravitational acceleration set to zero (not on earth)
k = 14 # insert the spring constant you found for the coiled spring
# Objects
# Ceiling to hang spring from.
# block to act as mass.
# Spring is represented by a cylinder.
scene.center = vector(0,-.1,0) # you may want to adjust this to improve display
ceiling = box(pos=vector(0,0,0), size=(.3,0.005,0.005)) # make the contact pointthe origin
block=box(pos=vector(0,-0.1,0), size=(.02,0.02,0.02), color=color.yellow)
# Using the positions of the block and ceiling set the cylinder to stretch from the ceiling to the block
spring = cylinder(pos=ceiling.pos, axis=block.pos, radius=.005)
# Initial values
block.m = 0.45 # insert the measured mass from coiled spring experiment.
block.v = vector(0,0,0) # the vector velocity assuming the block is initially stationary
block.p= block.m * block.v
block.pos=vector(0,-L0-0.05,0) # initial position of block 0.05m from equilibrium
# Setting the timestep and zeroing the cumulative time
deltat = .0001 # you should decrease this later to test if it is small enough
t = 0
W = 0
displacement=0
Kgraph = gcurve(color=color.cyan)
Ugraph = gcurve(color=color.yellow)
KplusUgraph = gcurve(color=color.red)
Wgraph = gcurve(color=color.green)
# Loop for repetitive calculations
scene.autoscale=0
while t < 4:
Fnet= -(((block.pos-vector(0,-L0,0))*k))-((block.p/block.m)*.2) #INSERT the force of the spring on the block
displacement=(mag(block.p)/block.m)*deltat
block.p= block.p+Fnet*deltat # updates the momentum
block.pos= block.pos+block.p/block.m*deltat # updates the position
spring.axis = block.pos #updates the spring axis so it stays on the block
t=t+deltat
pmag = mag(block.p) # or whatever you've called the block's momentum
K = (pmag**2)*.5/block.m #COMPLETE this for the kinetic energy of the
block
U = ((mag(block.pos)-L0)**2)*.5*k #COMPLETE this for the potential energy of the block-spring system (note no gravity)
W = W - displacement*(mag(block.p)/block.m)*.2
Kgraph.plot(pos=(t,K))
Ugraph.plot(pos=(t,U))
KplusUgraph.plot(pos=(t,K+U))
Wgraph.plot(pos=(t,W))