ChessEngine/Rook.java at master · gmihalea/ChessEngine · GitHub

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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
import java.util.ArrayList;

public class Rook extends Piece {

	private int moved;

	public boolean hasMoved() {
		return !(moved == 0);
	}

	public void makeMove() {
		this.moved ++;
	}

	public void undoMove() {
		this.moved --;
	}

	public Rook(Square position, int color) {
		super(position, color);
	}

	@Override
	public ArrayList<Square> getValidSquares() {
		int row = this.getPosition().getNumber();
		int column = this.getPosition().getLetter();
		int oponentColor = (this.getColor() == PieceColor.WHITE) ? PieceColor.BLACK : PieceColor.WHITE;
		Square intermediate;
		ArrayList<Square> result = new ArrayList<Square>();

		/*
		 * Checks availability upwards
		 */

		for(int i=row; i<8; i++) {
			intermediate = Board.translate(column, i+1);
			if(intermediate.getPiece() == null) {
				result.add(intermediate);
			} else if (intermediate.getPiece().getColor() == oponentColor) {
				result.add(intermediate);
				break;
			} else
				break;
		}

		/*
		 * Checks availability downwards
		 */

		for(int i=row; i>1; i--) {
			intermediate = Board.translate(column, i-1);
			if(intermediate.getPiece() == null) {
				result.add(intermediate);
			} else if (intermediate.getPiece().getColor() == oponentColor){
				result.add(intermediate);
				break;
			} else
				break;
		}

		/*
		 * Checks availability sideways to left
		 */

		for(int i=column; i>Letter.A; i--) {
			intermediate = Board.translate(i-1, row);
			if(intermediate.getPiece() == null) {
				result.add(intermediate);
			} else if (intermediate.getPiece().getColor() == oponentColor){
				result.add(intermediate);
				break;
			} else
				break;
		}

		/*
		 * Checks availability sideways to right
		 */

		for(int i=column; i<Letter.H; i++) {
			intermediate = Board.translate(i+1, row);
			if(intermediate.getPiece() == null) {
				result.add(intermediate);
			} else if (intermediate.getPiece().getColor() == oponentColor){
				result.add(intermediate);
				break;
			} else
				break;
		}

		return result;
	}

}