Use stream for explosion generation

1 files changed, 5 insertions, 6 deletions

diff --git a/src/ch/epfl/xblast/server/Bomb.java b/src/ch/epfl/xblast/server/Bomb.java
index 3b1203b..f32960f 100644
--- a/src/ch/epfl/xblast/server/Bomb.java
+++ b/src/ch/epfl/xblast/server/Bomb.java
@@ -6,9 +6,10 @@ import ch.epfl.xblast.Cell;
 import ch.epfl.xblast.Direction;
 import ch.epfl.xblast.PlayerID;
 
-import java.util.ArrayList;
 import java.util.List;
 import java.util.Objects;
+import java.util.stream.Collectors;
+import java.util.stream.Stream;
 
 /**
  * A Bomb.
@@ -106,11 +107,9 @@ public final class Bomb {
      * @return the explosion
      */
     public List<Sq<Sq<Cell>>> explosion() {
-        List<Sq<Sq<Cell>>> explosion = new ArrayList<>();
-        for (Direction dir : Direction.values()) {
-            explosion.add(this.explosionArmTowards(dir));
-        }
-        return explosion;
+        return Stream.of(Direction.values())
+                .map(this::explosionArmTowards)
+                .collect(Collectors.toList());
     }
 
 }