From 8fc54d7563b6a5f4fba918c684151446233af1fd Mon Sep 17 00:00:00 2001
From: b22015516 <mehdi.belhachemi@etu.univ-amu.fr>
Date: Fri, 8 Nov 2024 11:15:27 +0100
Subject: [PATCH] mehdi : modifiying initializeElements() method

---
 src/main/java/model/Fire.java             |  4 +++-
 src/main/java/model/FireFighter.java      |  5 +++--
 src/main/java/model/FirefighterBoard.java | 14 +++++++++-----
 3 files changed, 15 insertions(+), 8 deletions(-)

diff --git a/src/main/java/model/Fire.java b/src/main/java/model/Fire.java
index 16cc169..648fdf7 100644
--- a/src/main/java/model/Fire.java
+++ b/src/main/java/model/Fire.java
@@ -16,5 +16,7 @@ public class Fire {
         this.firePositions = postion;
     }
 
-
+    public Position getFirePosition() {
+        return firePositions;
+    }
 }
diff --git a/src/main/java/model/FireFighter.java b/src/main/java/model/FireFighter.java
index b377c42..96d9f7f 100644
--- a/src/main/java/model/FireFighter.java
+++ b/src/main/java/model/FireFighter.java
@@ -18,6 +18,7 @@ public class FireFighter {
         this.FireFighterPosition = postion;
     }
 
-
-
+    public Position getFireFighterPosition() {
+        return FireFighterPosition;
+    }
 }
diff --git a/src/main/java/model/FirefighterBoard.java b/src/main/java/model/FirefighterBoard.java
index 94e52d5..5627c7c 100644
--- a/src/main/java/model/FirefighterBoard.java
+++ b/src/main/java/model/FirefighterBoard.java
@@ -57,11 +57,15 @@ public class FirefighterBoard implements Board<List<ModelElement>> {
   @Override
   public List<ModelElement> getState(Position position) {
     List<ModelElement> result = new ArrayList<>();
-    for (Position firefighterPosition : firefighterPositions)
-      if (firefighterPosition.equals(position))
+    for (FireFighter firefighter : firefighter)
+      if (firefighter.getFireFighterPosition().equals(position))
         result.add(ModelElement.FIREFIGHTER);
-    if (firePositions.contains(position))
-      result.add(ModelElement.FIRE);
+
+      for(Fire fire : fire){
+        if(fire.getFirePosition().equals(position)){
+          result.add(ModelElement.FIRE);
+        }
+      }
     return result;
   }
 
@@ -86,7 +90,7 @@ public class FirefighterBoard implements Board<List<ModelElement>> {
     List<Position> modifiedPositions = new ArrayList<>();
     if (step % 2 == 0) {
       List<Position> newFirePositions = new ArrayList<>();
-      for (Position fire : firePositions) {
+      for (Fire fire : fires) {
         newFirePositions.addAll(neighbors.get(fire));
       }
       firePositions.addAll(newFirePositions);
-- 
GitLab