Properly define class constants and protect object attributes

1 files changed, 12 insertions, 4 deletions

diff --git a/src/ch/epfl/maze/physical/ProbabilisticAnimal.java b/src/ch/epfl/maze/physical/ProbabilisticAnimal.java
index 45c2175..ed26d23 100644
--- a/src/ch/epfl/maze/physical/ProbabilisticAnimal.java
+++ b/src/ch/epfl/maze/physical/ProbabilisticAnimal.java
@@ -15,8 +15,9 @@ import java.util.stream.Collectors;
  */
 abstract public class ProbabilisticAnimal extends Animal {
 
-    private final Random randomSource;
-    protected Direction currentDirection;
+    public static final Random RANDOM_SOURCE = new Random();
+
+    private Direction currentDirection;
 
     /**
      * Constructs a probabilistic animal with a starting position
@@ -25,10 +26,17 @@ abstract public class ProbabilisticAnimal extends Animal {
      */
     public ProbabilisticAnimal(Vector2D position) {
         super(position); // no pun intended
-        this.randomSource = new Random();
         this.currentDirection = Direction.NONE;
     }
 
+    protected Direction getCurrentDirection() {
+        return this.currentDirection;
+    }
+
+    protected void setCurrentDirection(Direction direction) {
+        this.currentDirection = direction;
+    }
+
     /**
      * Excludes the given direction from possible choices.
      *
@@ -61,7 +69,7 @@ abstract public class ProbabilisticAnimal extends Animal {
      * @return A random Direction taken from the given choices
      */
     protected Direction getRandomDirection(Direction[] choices) {
-        return choices[randomSource.nextInt(choices.length)];
+        return choices[RANDOM_SOURCE.nextInt(choices.length)];
     }
 
     /**