from scipy.integrate import odeint
import matplotlib.pyplot as plt
import numpy as np
import pylab as pl
import math

#plot f(x) vs. x
x = np.arange(-4.0,4.0,0.01)
fx = 4*x**2 - 16
zerox = 0.*x

#Create plot env
f = plt.figure()
plt.grid(True)
f.set_figwidth(8)
f.set_figheight(8)

#plot some stuff
plt.plot(x,fx,'g-',x,zerox,'b-')

plt.xlabel('x',fontsize = 15)
plt.ylabel('y', fontsize = 15)
plt.savefig("2024-09-09/ex1.png")

#Integrating stuff!
#Define the integration:
def dx_dt(xx,t):
    return[4.*xx[0]**2 -16.]

f1 = plt.figure()
f.set_figwidth(12)
f.set_figheight(12)

# forward integration in time, so we need a time span
ts = np.linspace(0,3,1000)
ic = 1
xs = odeint(dx_dt, ic, ts)
print(xs)

plt.plot(ts,xs)
plt.savefig("2024-09-09/ex2.png")
plt.show()