# -*- coding: utf-8 -*-
import numpy as np
import matplotlib.pyplot as plt

import matplotlib
matplotlib.rcParams['mathtext.fontset'] = 'stix'
matplotlib.rcParams['font.family'] = 'STIXGeneral'


#%% FONCTION
omega=0.5
A = 15
Delta=2.5
T0=10


def S(t): 
    return T0+A*np.exp(-0/Delta)*np.cos(omega*t)

def H(t):
    return T0+A*np.exp(-2/Delta)*np.cos(omega*t-2/Delta)
def K(t):
    return T0+A*np.exp(-8/Delta)*np.cos(omega*t-8/Delta)


#LISTE DES X
t = np.linspace(0,40,1000)

#LISTE DES Y
Y1 = H(t)
Y2 = S(t)
Y3 = K(t)

#%% GRAPHE

fig = plt.figure(figsize=(6,6))
ax = fig.add_subplot(111)

ax.set_xlim(0,40) #Limites de l'axe X
ax.set_ylim(-10,30) #Limites de l'axe Y

ax.set_xlabel(r'Temps (en sec)', fontsize = 20) #Titre axe X
ax.set_ylabel(r'Température (en °C)', fontsize = 20) #Titre axe Y


ax.plot(t,Y2, label=r'En surface', lw=2) #label : nom dans la légende, lw : épaisseur du trait

ax.plot(t,Y1, label=r'A 2m de profondeur', lw=2) #label : nom dans la légende, lw : épaisseur du trait

ax.plot(t,Y3, label=r'A 8m de profondeur', lw=2)

#%% OPTIONNEL

ax.grid(True) #Grille
ax.legend(loc=2 ,prop={'size':15}) #Legende