package com.aot.engine.cards.movements;

import com.aot.engine.Color;
import com.aot.engine.board.Board;
import com.aot.engine.board.Square;
import java.util.List;
import java.util.Set;

public class LineAndDiagonalMovementsCard extends MovementsCard {

    public LineAndDiagonalMovementsCard(Board board, String name, int numberOfMovements, Color color) {
        super(board, name, numberOfMovements, color);
        setPossibleSquareGetter();
    }

    private void setPossibleSquareGetter() {
        probableSquaresGetter = lineAndDiagonalProbableSquaresGetter;
    }

    public LineAndDiagonalMovementsCard(Board board, String name, int numberOfMovements, Color color, List<Color> addtionalMovementsColor) {
        super(board, name, numberOfMovements, color, addtionalMovementsColor);
        setPossibleSquareGetter();
    }

    @Override
    public Set<String> getPossibleMovements(Square currentSquare) {
        return getLineAndDiagonalMovements(currentSquare);
    }

    @Override
    protected void resetPossibleSquareGetter() {
        probableSquaresGetter = lineProbableSquaresGetter;
    }

}

@@ -1054,41 +1054,16 @@
     
-probableSquaresGetter = lineProba
+setPossi
 bleS
@@ -1059,34 +1059,35 @@
 etPossibleSquare
-s
 Getter
+()
 ;%0A    %7D%0A%0A%7D%0A

correct resetPossibleSquareGetter.