import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline 
# De laatste regel zorgt dat de plots in het notebook komen.

t=0
dt = 0.001 
m = 0.1
g = -9.8
cw = 1
rho = 1.028 
A = 0.01 
v=0
z = 5

eindtijd = 1
tijden = [t] # de lijst 'tijden' met een begintijd is gemaakt
snelheden = [v] # en 'snelheden'
plaatsen = [z] # en 'plaatsen' ook
while t < eindtijd:
    # begin model
    t = t + dt
    Fz = m * g
    Fw = .5 * cw * rho * A * v ** 2
    a=(Fz + Fw) / m
    dv = a * dt
    v = v + dv
    dz = v * dt
    z = z + dz
    # eind model
    # begin van het vullen van de lijsten
    snelheden.append(v) # voegt een snelheid toe aan de lijst met snelheden.
    plaatsen.append(z) # voegt een plaats toe aan een lijst met plaatsen.
    tijden.append(t) # voegt de tijd toe aan een lijst met tijden.
    # eind van het vullen van lijsten

plt.figure(figsize=(10,10))
plt.plot(tijden, snelheden)
plt.title('De snelheid als functie van de tijd')
plt.xlabel('tijd [s]')
plt.ylabel('snelheid [m/s]')

plt.figure(figsize=(10,10))
plt.plot(tijden, plaatsen)
plt.title('De plaats als functie van de tijd')
plt.xlabel('tijd [s]')
plt.ylabel('hoogte [m]')

Ekin = 1 / 2 * m * np.array(snelheden) ** 2
Epot = -m * g * np.array(plaatsen)

plt.figure(figsize=(10,10))
plt.plot(tijden, Ekin, label = 'Kinethisch')
plt.plot(tijden, Epot, label = 'Potentieel')
plt.title('De energie als functie van de tijd')
plt.xlabel('Tijd [s]')
plt.ylabel('Energie [J]')
plt.legend()