summaryrefslogtreecommitdiff
path: root/src/ch/epfl/maze/physical/ProbabilisticAnimal.java
blob: 6267500d1978e87f8122daca5158cd14459c0275 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
package ch.epfl.maze.physical;

import ch.epfl.maze.util.Direction;
import ch.epfl.maze.util.Vector2D;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Random;
import java.util.stream.Collectors;

/**
 * A probabilistic animal that use a random component in its decision making process.
 *
 * @author Pacien TRAN-GIRARD
 */
abstract public class ProbabilisticAnimal extends Animal {

    private final Random randomSource;
    protected Direction currentDirection;

    /**
     * Constructs a probabilistic animal with a starting position
     *
     * @param position Starting position of the probabilistic animal in the labyrinth
     */
    public ProbabilisticAnimal(Vector2D position) {
        super(position); // no pun intended
        this.randomSource = new Random();
        this.currentDirection = Direction.NONE;
    }

    /**
     * Excludes the given direction from possible choices.
     *
     * @param choices   An array of choices
     * @param toExclude The Direction to exclude
     * @return An array of smart choices
     */
    protected Direction[] excludeDirection(Direction[] choices, Direction toExclude) {
        return (new ArrayList<>(Arrays.asList(choices)))
                .stream()
                .filter(dir -> dir != toExclude)
                .collect(Collectors.toList())
                .toArray(new Direction[0]);
    }

    /**
     * Excludes the origin direction from possible choices.
     *
     * @param choices An array of choices
     * @return An array of smart choices
     */
    protected Direction[] excludeOrigin(Direction[] choices) {
        return this.excludeDirection(choices, this.currentDirection.reverse());
    }

    /**
     * Returns a random Direction from the given choices.
     *
     * @param choices An array of Direction
     * @return A random Direction taken from the given choices
     */
    protected Direction getRandomDirection(Direction[] choices) {
        return choices[randomSource.nextInt(choices.length)];
    }

    /**
     * Moves according to an improved version of a <i>random walk</i> : the
     * probabilistic animal does not directly retrace its steps if not forced.
     */
    @Override
    public Direction move(Direction[] choices) {
        if (choices.length == 0) return Direction.NONE;

        Direction[] smartChoices = choices.length > 1 ? this.excludeOrigin(choices) : choices;
        this.currentDirection = this.getRandomDirection(smartChoices);
        return this.currentDirection;
    }

}