import numpy as np
import matplotlib.pyplot as plt

# Definerer konstanter
g = 9.81
k = 1.0
m = 1.0

# Definerer initalverdien
v0 = 0.

# Definerer tidsintervall og noen konstanter
tmax = 10.0
dt = 0.01
c = k/m
n = int(tmax/dt)

# Lager arrays for ? lagre resultater og setter startverdiene
a = np.zeros(n)
v = np.zeros(n)
t = np.zeros(n)
v[0] = v0
t[0] = 0

# Integrasjonsl?kka
i = 0
while i<n-1:
    a[i] = g - c*v[i]
    v[i+1] = v[i]+a[i]*dt
    t[i+1] = t[i]+dt
    i = i+1
    

# Plotter resultatene

plt.subplot(2,1,1)
plt.plot(t,v)
plt.xlabel('t [s]')
plt.ylabel('v [m/s]')

plt.subplot(2,1,2)
plt.plot(t,a)
plt.xlabel('t [s]')
plt.ylabel('a [m/s$^2$]')

plt.savefig('fallMedLuftmotstand.png')
plt.show()