8ea3c9ccffe941e7a49fbfbb653653e4f9ff7021
[vchess.git] / public / javascripts / variants / Berolina.js
1 class BerolinaRules extends ChessRules
2 {
3 // En-passant after 2-sq jump
4 getEpSquare(moveOrSquare)
5 {
6 if (!moveOrSquare)
7 return undefined;
8 if (typeof moveOrSquare === "string")
9 {
10 const square = moveOrSquare;
11 if (square == "-")
12 return undefined;
13 // Enemy pawn initial column must be given too:
14 let res = [];
15 const epParts = square.split(",");
16 res.push(V.SquareToCoords(epParts[0]));
17 res.push(V.ColumnToCoord(epParts[1]));
18 return res;
19 }
20 // Argument is a move:
21 const move = moveOrSquare;
22 const [sx,ex,sy] = [move.start.x,move.end.x,move.start.y];
23 if (this.getPiece(sx,sy) == V.PAWN && Math.abs(sx - ex) == 2)
24 {
25 return
26 [
27 {
28 x: (ex + sx)/2,
29 y: (move.end.y + sy)/2
30 },
31 move.end.y
32 ];
33 }
34 return undefined; //default
35 }
36
37 // Special pawns movements
38 getPotentialPawnMoves([x,y])
39 {
40 const color = this.turn;
41 let moves = [];
42 const [sizeX,sizeY] = [V.size.x,V.size.y];
43 const shiftX = (color == "w" ? -1 : 1);
44 const firstRank = (color == 'w' ? sizeX-1 : 0);
45 const startRank = (color == "w" ? sizeX-2 : 1);
46 const lastRank = (color == "w" ? 0 : sizeX-1);
47
48 if (x+shiftX >= 0 && x+shiftX < sizeX) //TODO: always true
49 {
50 const finalPieces = x + shiftX == lastRank
51 ? [V.ROOK,V.KNIGHT,V.BISHOP,V.QUEEN]
52 : [V.PAWN]
53 // One square diagonally
54 for (let shiftY of [-1,1])
55 {
56 if (this.board[x+shiftX][y+shiftY] == V.EMPTY)
57 {
58 for (let piece of finalPieces)
59 {
60 moves.push(this.getBasicMove([x,y], [x+shiftX,y+shiftY],
61 {c:color,p:piece}));
62 }
63 if (x == startRank && y+2*shiftY>=0 && y+2*shiftY<sizeY
64 && this.board[x+2*shiftX][y+2*shiftY] == V.EMPTY)
65 {
66 // Two squares jump
67 moves.push(this.getBasicMove([x,y], [x+2*shiftX,y+2*shiftY]));
68 }
69 }
70 }
71 // Capture
72 if (this.board[x+shiftX][y] != V.EMPTY
73 && this.canTake([x,y], [x+shiftX,y]))
74 {
75 for (let piece of finalPieces)
76 moves.push(this.getBasicMove([x,y], [x+shiftX,y], {c:color,p:piece}));
77 }
78 }
79
80 // En passant
81 const Lep = this.epSquares.length;
82 const epSquare = this.epSquares[Lep-1]; //always at least one element
83 if (!!epSquare && epSquare[0].x == x+shiftX && epSquare[0].y == y
84 && Math.abs(epSquare[1] - y) == 1)
85 {
86 let enpassantMove = this.getBasicMove([x,y], [x+shiftX,y]);
87 enpassantMove.vanish.push({
88 x: x,
89 y: epSquare[1],
90 p: 'p',
91 c: this.getColor(x,epSquare[1])
92 });
93 moves.push(enpassantMove);
94 }
95
96 return moves;
97 }
98
99 isAttackedByPawn([x,y], colors)
100 {
101 for (let c of colors)
102 {
103 let pawnShift = (c=="w" ? 1 : -1);
104 if (x+pawnShift>=0 && x+pawnShift<V.size.x)
105 {
106 if (this.getPiece(x+pawnShift,y)==V.PAWN
107 && this.getColor(x+pawnShift,y)==c)
108 {
109 return true;
110 }
111 }
112 }
113 return false;
114 }
115
116 getNotation(move)
117 {
118 const piece = this.getPiece(move.start.x, move.start.y);
119 if (piece == V.PAWN)
120 {
121 // Pawn move
122 const finalSquare = V.CoordsToSquare(move.end);
123 let notation = "";
124 if (move.vanish.length == 2) //capture
125 notation = "Px" + finalSquare;
126 else
127 {
128 // No capture: indicate the initial square for potential ambiguity
129 const startSquare = V.CoordsToSquare(move.start);
130 notation = startSquare + finalSquare;
131 }
132 if (move.appear[0].p != V.PAWN) //promotion
133 notation += "=" + move.appear[0].p.toUpperCase();
134 return notation;
135 }
136 return super.getNotation(move); //all other pieces are orthodox
137 }
138 }
139
140 const VariantRules = BerolinaRules;