from turtle import *
from math import *
from random import *
from matplotlib import pyplot

#A=Arbre C=Cendre F=Feu V=Vide

N=50 # taille de la foret
FORET=[['V' for i in range(N+1+1)]]+[['V']+[choice(['A']) for i in range(N)]+['V'] for j in range(N)]+[['V' for i in range(N+1+1)]]  #Structure de la foret

FORET[1][randint(1,N)]='F' # Foyer
P=0.4 # Probabilité de propagation à un voisin

####### MODELISATION ####

def suivant(foret,p):
    s=[['V' for i in range(len(foret))] for j in range(len(foret))]
    for Y in range(1,len(foret)-1):
        for X in range(1,len(foret[Y])-1):

            if foret[Y][X]=='F':
                s[Y][X]='C'

            elif foret[Y][X]=='A':
                if foret[Y-1][X]=='F' or foret[Y+1][X]=='F' or foret[Y][X-1]=='F' or foret[Y][X+1]=='F':
                    if random()<p: # le voisin prend feu
                        s[Y][X]='F'
                    else:
                        s[Y][X]='A'
                else:
                    s[Y][X]='A'

            elif foret[Y][X]=='C':
                s[Y][X]='C'

            elif foret[Y][X]=='V':
                s[Y][X]='V'



    return s


def feu(foret):
    for Y in range(1,len(foret)-1):
        for X in range(1,len(foret[Y])-1):
            if foret[Y][X]=='F':
                return True

    return False

####### CALCULS ######

def degats(foret):
    c=0 #nombre d'arbres brûlés
    for Y in range(1,len(foret)-1):
        for X in range(1,len(foret[Y])-1):
            if foret[Y][X]=='C':
                c=c+1
    return c

def FEU(foret,p):
    while feu(foret) :
        foret=suivant(foret,p)

    return foret # foret après l'incendie


##### REPRESENTATION #########

def represente(pas):
    for i in range(int(1/pas)):
        p=i*pas
        f=degats(FEU(FORET,p))/N**2
        pyplot.plot(p,f,'*')
    pyplot.show()


if __name__ == '__main__':
    represente(0.05)