Fix Monochrome update
[vchess.git] / client / src / variants / Zen.js
CommitLineData
0c3fe8a6
BA
1import { ChessRules } from "@/base_rules";
2
32f6285e 3export class ZenRules extends ChessRules {
ffeaef85
BA
4 getEpSquare(moveOrSquare) {
5 if (!moveOrSquare) return undefined;
6 if (typeof moveOrSquare === "string") {
7 const square = moveOrSquare;
8 if (square == "-") return undefined;
9 return V.SquareToCoords(square);
10 }
11 const move = moveOrSquare;
12 const s = move.start,
13 e = move.end;
14 if (
15 // Exclude captures (of rooks for example)
16 move.vanish.length == 1 &&
17 s.y == e.y &&
18 Math.abs(s.x - e.x) == 2 &&
19 move.appear[0].p == V.PAWN
20 ) {
21 return {
22 x: (s.x + e.x) / 2,
23 y: s.y
24 };
25 }
26 return undefined;
6808d7a1 27 }
dac39588
BA
28
29 // TODO(?): some duplicated code in 2 next functions
6808d7a1 30 getSlideNJumpMoves([x, y], steps, oneStep) {
dac39588 31 let moves = [];
6808d7a1 32 outerLoop: for (let loop = 0; loop < steps.length; loop++) {
dac39588
BA
33 const step = steps[loop];
34 let i = x + step[0];
35 let j = y + step[1];
6808d7a1
BA
36 while (V.OnBoard(i, j) && this.board[i][j] == V.EMPTY) {
37 moves.push(this.getBasicMove([x, y], [i, j]));
38 if (oneStep) continue outerLoop;
dac39588
BA
39 i += step[0];
40 j += step[1];
41 }
42 // No capture check: handled elsewhere (next method)
43 }
44 return moves;
45 }
46
47 // follow steps from x,y until something is met.
48 // if met piece is opponent and same movement (asA): eat it!
6808d7a1
BA
49 findCaptures_aux([x, y], asA) {
50 const color = this.getColor(x, y);
9bd6786b 51 let moves = [];
6808d7a1
BA
52 const steps =
53 asA != V.PAWN
54 ? asA == V.QUEEN
55 ? V.steps[V.ROOK].concat(V.steps[V.BISHOP])
56 : V.steps[asA]
57 : color == "w"
58 ? [
59 [-1, -1],
60 [-1, 1]
61 ]
62 : [
63 [1, -1],
64 [1, 1]
65 ];
9bd6786b 66 const oneStep = [V.KNIGHT,V.PAWN].includes(asA); //we don't capture king
6808d7a1
BA
67 const lastRank = color == "w" ? 0 : V.size.x - 1;
68 const promotionPieces = [V.ROOK, V.KNIGHT, V.BISHOP, V.QUEEN];
8ca6042e 69 const oppCol = V.GetOppCol(color);
6808d7a1 70 outerLoop: for (let loop = 0; loop < steps.length; loop++) {
dac39588
BA
71 const step = steps[loop];
72 let i = x + step[0];
73 let j = y + step[1];
6808d7a1
BA
74 while (V.OnBoard(i, j) && this.board[i][j] == V.EMPTY) {
75 if (oneStep) continue outerLoop;
dac39588
BA
76 i += step[0];
77 j += step[1];
78 }
6808d7a1
BA
79 if (
80 V.OnBoard(i, j) &&
8ca6042e 81 this.getColor(i, j) == oppCol &&
6808d7a1
BA
82 this.getPiece(i, j) == asA
83 ) {
dac39588 84 // eat!
6808d7a1 85 if (this.getPiece(x, y) == V.PAWN && i == lastRank) {
dac39588
BA
86 // Special case of promotion:
87 promotionPieces.forEach(p => {
6808d7a1 88 moves.push(this.getBasicMove([x, y], [i, j], { c: color, p: p }));
dac39588 89 });
6808d7a1 90 } else {
dac39588 91 // All other cases
6808d7a1 92 moves.push(this.getBasicMove([x, y], [i, j]));
dac39588
BA
93 }
94 }
95 }
96 return moves;
97 }
98
99 // Find possible captures from a square: look in every direction!
6808d7a1 100 findCaptures(sq) {
dac39588 101 let moves = [];
dac39588
BA
102 Array.prototype.push.apply(moves, this.findCaptures_aux(sq, V.PAWN));
103 Array.prototype.push.apply(moves, this.findCaptures_aux(sq, V.ROOK));
104 Array.prototype.push.apply(moves, this.findCaptures_aux(sq, V.KNIGHT));
105 Array.prototype.push.apply(moves, this.findCaptures_aux(sq, V.BISHOP));
106 Array.prototype.push.apply(moves, this.findCaptures_aux(sq, V.QUEEN));
dac39588
BA
107 return moves;
108 }
109
e9b736ee
BA
110 canTake(sq1, sq2) {
111 return false; //captures handled separately
112 }
dac39588 113
8ca6042e
BA
114 getPotentialMovesFrom(sq) {
115 return super.getPotentialMovesFrom(sq).concat(this.findCaptures(sq));
dac39588
BA
116 }
117
6808d7a1 118 getNotation(move) {
dac39588 119 // Recognize special moves first
6808d7a1 120 if (move.appear.length == 2) {
dac39588 121 // castle
6808d7a1
BA
122 if (move.end.y < move.start.y) return "0-0-0";
123 return "0-0";
dac39588
BA
124 }
125
2c5d7b20 126 // Translate initial square (because pieces may fly unusually!)
dac39588
BA
127 const initialSquare = V.CoordsToSquare(move.start);
128
129 // Translate final square
130 const finalSquare = V.CoordsToSquare(move.end);
131
132 let notation = "";
133 const piece = this.getPiece(move.start.x, move.start.y);
6808d7a1 134 if (piece == V.PAWN) {
dac39588 135 // pawn move (TODO: enPassant indication)
8ca6042e 136 if (move.vanish.length == 2) {
dac39588
BA
137 // capture
138 notation = initialSquare + "x" + finalSquare;
8ca6042e 139 }
6808d7a1
BA
140 else notation = finalSquare;
141 if (piece != move.appear[0].p)
142 //promotion
dac39588 143 notation += "=" + move.appear[0].p.toUpperCase();
8ca6042e
BA
144 }
145 else {
dac39588
BA
146 // Piece movement
147 notation = piece.toUpperCase();
6808d7a1 148 if (move.vanish.length > 1) notation += initialSquare + "x";
dac39588
BA
149 notation += finalSquare;
150 }
151 return notation;
152 }
153
6808d7a1 154 static get VALUES() {
dac39588 155 return {
6808d7a1
BA
156 p: 1,
157 r: 3,
158 n: 2,
159 b: 2,
160 q: 5,
161 k: 1000
162 };
dac39588 163 }
6808d7a1 164};