diff --git a/src/main/java/model/FirefighterBoard.java b/src/main/java/model/FirefighterBoard.java index ce6d9e41b393468ecf7cfe510f7a5890243a950c..61a83b0bfe511c82f59de9055f2da8abcf4c6a4d 100644 --- a/src/main/java/model/FirefighterBoard.java +++ b/src/main/java/model/FirefighterBoard.java @@ -201,6 +201,9 @@ public class FirefighterBoard implements Board<List<ModelElement>> { for (Position firefighterPosition : firefighterPositions) { updatedElements.add(new Pair<>(firefighterPosition, ModelElement.FIREFIGHTER)); } + for (Position cloudPosition : cloudPositions) { + updatedElements.add(new Pair<>(cloudPosition, ModelElement.CLOUD)); + } return updatedElements; } diff --git a/src/main/java/view/FirefighterGrid.java b/src/main/java/view/FirefighterGrid.java index ab926abfa39a23b8fdf3fb818e03f25bd1f78c53..0df4d0972080f94f5614a5b60f8a4ab33f7e2eeb 100644 --- a/src/main/java/view/FirefighterGrid.java +++ b/src/main/java/view/FirefighterGrid.java @@ -43,6 +43,8 @@ public class FirefighterGrid extends Canvas { gc.setFill(Color.RED); // Fire is red } else if (element == ModelElement.FIREFIGHTER) { gc.setFill(Color.BLUE); // Firefighter is blue + } else if (element == ModelElement.CLOUD){ + gc.setFill(Color.YELLOW); } else { gc.setFill(Color.WHITE); // Empty space is white }