FireFighter.java

package model;

import util.Position;
import util.TargetStrategy;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;


public class FireFighter extends Extinguisher {


    private final TargetStrategy targetStrategy = new TargetStrategy();


    public FireFighter(Position position){
        super(position);
        element = ModelElement.FIREFIGHTER;
    }


    public List<Position> update(FirefighterBoard firefighterBoard){
        List<Position> modifiedPositions = new ArrayList<>();

        Position currentPosition = this.getPosition();
        Position newPosition = targetStrategy.neighborClosestToFire(currentPosition, firefighterBoard.getFires().keySet(), firefighterBoard.getNeighbors());

        this.setPosition(newPosition);
        modifiedPositions.add(currentPosition);
        modifiedPositions.add(newPosition);

        firefighterBoard.extinguish(newPosition);

        for (Position neighbor : firefighterBoard.getNeighbors().get(newPosition)) {
            firefighterBoard.extinguish(neighbor);
            modifiedPositions.add(neighbor);
        }

        return modifiedPositions;
    }

}