#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Tue Oct  6 11:28:28 2020

@author: yvestalbourdel
"""

import numpy as np
import matplotlib.pyplot as plt
import time

D=0.00002345
L=1.0
T0=20.0
Tc=100.0
Tf=0.0
Nx=100
dx=L/Nx
dt=2.0
Nt=10000
K=(D*(dt)/((dx)**2))
print (np.floor(Nt*dt/(60)),'minutes environ')
x=np.arange(0,(Nx+2)*dx,dx)
T=np.concatenate(([Tc],[T0]*Nx,[Tf]))
dT=np.zeros(Nx+2)

for j in range(1,Nt):
    for i in range(1,Nx+1):
        # dT[i]=...............
    T+=dT
    if ((j-1)%600==0):
        plotlabel="t=%0.1f minutes" %((j)*dt/60)
        plt.plot(x,T,'.',label=plotlabel,color = plt.get_cmap('cool')(float((j)/Nt)))

plt.xlabel(u'$x(m)$', fontsize=10)
plt.ylabel(u'$T (°Celsius)$', fontsize=10, rotation=90)
plt.title(u'régime transitoire dans une barre de Fer entre deux thermostats')
plt.legend()
plt.show()