Update genForet.py

genForet.py

@@ -7,29 +7,79 @@ Created on Fri Feb  1 11:30:02 2019
 import numpy as np
 import random
 #import tkinter as tk
-
+"""
+    Les états possibles de nos cases :
+        - 0 : pas d'arbre
+        - 1 : un arbre sain
+        - -1 : un arbre en feu
+        - -2 : un arbre mort
+"""
 #Fonctions agissant sur la matrice
 def genForet(n,p):
+    if n>=2:
         foret=np.full(shape=(n,n), fill_value=0)
         for i in range(0,foret.shape[0]):
             for j in range(0,foret.shape[1]):
                 if random.random() <= p:
                     foret[i,j]=1 
         return foret
+    else:
+        raise ValueError("n doit prendre une valeur de 2 au moins")
 
 def enflammer(foret,x,y): #Fonction mettant le feu aléatoirement à une case
     if x==-1:
         x=random.randint(0, foret.shape[0]-1)
     if y==-1:
         y=random.randint(0, foret.shape[1]-1)
-    foret[x,y]=-1
-    return foret
+    foretEnflammée = np.copy(foret)
+    foretEnflammée[x,y]=-1
+    return foretEnflammée
 
-def propagation(n, p=0.5, x=-1, y=-1):
-    F=[]
-    foret = genForet(n, p)
-    F.append(foret)
-    F.append(enflammer(foret,x,y))
+def propager(F,k,i,j):
+    if not i == 0 and not i == F[k].shape[0]-1 and not j == 0 and not j == F[k].shape[1]-1: #Si la case n'est pas sur la bordure 
+        print("La case {};{} va bruler les quatre autour d'elle".format(i,j))
+        if F[k][i-1,j] == 1: #Si la case au dessus est un arbre
+            F[k][i-1,j] = -1 #La case au dessus brûle
+        if F[k][i+1,j] == 1: #Si la case au dessous est un arbre
+            F[k][i+1,j] = -1 #La case en dessous brûle
+        if F[k][i,j-1] == 1: #Si la case à gauche est un arbre
+            F[k][i,j-1] = -1 #La case à gauche brûle
+        if F[k][i,j+1] ==1: #Si la case à droite est un arbre
+            F[k][i,j+1] = -1 #La case à droite brûle
+            
+    if i == 0:#La case est au bord supérieur
+        if F[k][i+1,j] == 1: #Si la case au dessous est un arbre
+            F[k][i+1,j] = -1 #La case en dessous brûle
+    if i == F[k].shape[0]-1: #La case est au bord inférieur
+        if F[k][i-1,j] == 1: #Si la case au dessus est un arbre
+            F[k][i-1,j] = -1 #La case au dessus brûle
+    if j == 0: #La case est au bord gauche
+        if F[k][i,j+1] ==1: #Si la case à droite est un arbre
+            F[k][i,j+1] = -1 #La case à droite brûle
+    if j == F[k].shape[1]-1: #La case est au bord droit
+        if F[k][i,j-1] == 1: #Si la case à gauche est un arbre
+            F[k][i,j-1] = -1 #La case à gauche brûle
+
+def feuDeForet(n, tours, p=0.5, x=-1, y=-1):
+    F=[genForet(n, p)]
+    print(F[0], "Instant initial")
+    F.append(enflammer(F[0],x,y)) #On stocke les deux premiers état de notre forêt
+    print(F[1], "Rang : 1 (Départ de feu)")
+    for k in range(2,tours): 
+        F.append(np.copy(F[k-1])) #On copie la forêt du tour précédent
+        for i in range(0,F[k].shape[0]):
+            for j in range(0,F[k].shape[1]):
+                
+                if F[k][i,j]==-1:#Si la case est en feu au rang k
+                    if F[k-1][i,j]==0: #Si cette case ne portait pas d'arbre au rang k-1 alors on propage le feu et la case s'éteint
+                        propager(F,k,i,j)
+                        F[k][i,j]=0
+                    elif F[k-2][i,j]==-1: #Si la case était en feu au tour précédent : propage et mort de l'arbre
+                        propager(F,k,i,j)
+                        F[k][i,j]=-2
+                    else: #La case a été mise en feu, elle propage le feu
+                        propager(F,k,i,j)
+        print(F[k], "Rang : {}".format(k))
     return F #Return temporaire