# chapitre 5 : fichier à modifier

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits import mplot3d

# On définit la fonction f :

def f(x, y):
    return x**2+y**2
    
# On donne les bornes du domaine de définition :

xmin = -2
xmax = 2
ymin = -2
ymax = 2

# Préparation des tableaux
    
x = np.linspace(xmin, xmax, 150)
y = np.linspace(ymin, ymax, 150)
X, Y = np.meshgrid(x, y)
Z = f(X,Y)

# Surface représentant f

fig1 = plt.figure(figsize = (12,7.5))
ax3d = plt.axes(projection='3d')

surf = ax3d.plot_surface(X, Y, Z, rstride=5, cstride=5,cmap='gist_heat_r')

fig1.colorbar(surf, shrink=0.5, aspect=7)

ax3d.set_title("Surface représentant $f$", fontsize = 15)
#ax3d.set_zlim(0,4)
ax3d.set_xlabel('x', fontsize = 11)
ax3d.set_ylabel('y', fontsize = 11)
ax3d.set_zlabel('Z', fontsize = 11)

plt.savefig("Customized Surface Plot.png")
plt.show()

# Lignes de niveaux de f

fig2 = plt.figure(figsize = (7.8,7))
#contours = plt.contour(X,Y,Z, np.arange(0,2,0.05), cmap = 'gray')

contours = plt.contour(X,Y,Z, levels=80, cmap = 'gray')

plt.clabel(contours, inline = True, fontsize = 9)
plt.title("Lignes de niveau de $f$", fontsize=12)
plt.xlabel('x', fontsize=11)
plt.ylabel('y', fontsize=11)

plt.show()