Refactor Monkey A.I.

1 files changed, 41 insertions, 0 deletions

diff --git a/src/ch/epfl/maze/physical/stragegies/picker/CyclePicker.java b/src/ch/epfl/maze/physical/stragegies/picker/CyclePicker.java
new file mode 100644
index 0000000..e1a969f
--- /dev/null
+++ b/src/ch/epfl/maze/physical/stragegies/picker/CyclePicker.java
@@ -0,0 +1,41 @@
+package ch.epfl.maze.physical.stragegies.picker;
+
+import ch.epfl.maze.util.Direction;
+
+import java.util.Set;
+
+/**
+ * A decision maker cycling in each Direction starting from a given one.
+ *
+ * @author Pacien TRAN-GIRARD
+ */
+public interface CyclePicker extends BlindPicker {
+
+    /**
+     * Returns the cycle starting Direction.
+     *
+     * @return The starting Direction
+     */
+    Direction getStartingDirection();
+
+    /**
+     * Returns the Rotation Direction.
+     *
+     * @return The Rotation Direction
+     */
+    Direction getRotationDirection();
+
+    @Override
+    default Direction pick(Set<Direction> choices) {
+        if (choices.isEmpty()) return FALLBACK_DIRECTION;
+        if (this.getStartingDirection() == Direction.NONE) return FALLBACK_DIRECTION;
+        if (this.getRotationDirection() == Direction.NONE) return FALLBACK_DIRECTION;
+
+        Direction dir = this.getStartingDirection();
+        while (!choices.contains(dir))
+            dir = dir.unRelativeDirection(this.getRotationDirection());
+
+        return dir;
+    }
+
+}