Fix Monkey undefined behaviour in special cases

1 files changed, 15 insertions, 4 deletions

diff --git a/src/ch/epfl/maze/physical/zoo/Monkey.java b/src/ch/epfl/maze/physical/zoo/Monkey.java
index 8aea75d..a29b7d4 100644
--- a/src/ch/epfl/maze/physical/zoo/Monkey.java
+++ b/src/ch/epfl/maze/physical/zoo/Monkey.java
@@ -37,9 +37,6 @@ public class Monkey extends Animal {
      */
 
     private Direction followLeft(Direction[] choices) {
-        if (choices.length == 0) return Direction.NONE;
-        if (choices.length == 1) return choices[0];
-
         List<Direction> choiceList = new ArrayList<>(Arrays.asList(choices));
         Direction dir = this.currentDirection.rotateLeft();
         while (!choiceList.contains(dir)) dir = dir.rotateRight();
@@ -48,12 +45,26 @@ public class Monkey extends Animal {
     }
 
     /**
+     * Determines the strategy to follow depending on special cases.
+     *
+     * @param choices An array of possible directions
+     * @return The Direction to take
+     */
+    private Direction findDirection(Direction[] choices) {
+        if (choices.length == 0) return Direction.NONE;
+        if (this.currentDirection == Direction.NONE) return choices[0];
+        if (choices.length == 1) return choices[0];
+        if (choices.length > 3) return choices[0];
+        return this.followLeft(choices);
+    }
+
+    /**
      * Moves according to the relative left wall that the monkey has to follow.
      */
 
     @Override
     public Direction move(Direction[] choices) {
-        this.currentDirection = this.followLeft(choices);
+        this.currentDirection = this.findDirection(choices);
         return this.currentDirection;
     }