Fix SuctionChess, draft HiddenRules (unfinished)
[vchess.git] / client / src / variants / Dark.js
CommitLineData
0c3fe8a6 1import { ChessRules } from "@/base_rules";
6808d7a1 2import { ArrayFun } from "@/utils/array";
0c3fe8a6
BA
3import { randInt } from "@/utils/alea";
4
6808d7a1 5export const VariantRules = class DarkRules extends ChessRules {
20620465 6 // Analyse in Dark mode makes no sense
8477e53d 7 static get CanAnalyze() {
20620465
BA
8 return false;
9 }
10
11 // Moves are revealed only when game ends
12 static get ShowMoves() {
13 return "none";
14 }
15
6808d7a1 16 setOtherVariables(fen) {
dac39588 17 super.setOtherVariables(fen);
6808d7a1 18 const [sizeX, sizeY] = [V.size.x, V.size.y];
dac39588 19 this.enlightened = {
6808d7a1
BA
20 w: ArrayFun.init(sizeX, sizeY),
21 b: ArrayFun.init(sizeX, sizeY)
dac39588
BA
22 };
23 // Setup enlightened: squares reachable by each side
24 // (TODO: one side would be enough ?)
25 this.updateEnlightened();
26 }
375ecdd1 27
6808d7a1
BA
28 updateEnlightened() {
29 for (let i = 0; i < V.size.x; i++) {
30 for (let j = 0; j < V.size.y; j++) {
dac39588
BA
31 this.enlightened["w"][i][j] = false;
32 this.enlightened["b"][i][j] = false;
33 }
34 }
6808d7a1 35 const pawnShift = { w: -1, b: 1 };
dac39588 36 // Initialize with pieces positions (which are seen)
6808d7a1
BA
37 for (let i = 0; i < V.size.x; i++) {
38 for (let j = 0; j < V.size.y; j++) {
39 if (this.board[i][j] != V.EMPTY) {
40 const color = this.getColor(i, j);
dac39588
BA
41 this.enlightened[color][i][j] = true;
42 // Add potential squares visible by "impossible pawn capture"
6808d7a1
BA
43 if (this.getPiece(i, j) == V.PAWN) {
44 for (let shiftY of [-1, 1]) {
45 if (
46 V.OnBoard(i + pawnShift[color], j + shiftY) &&
47 this.board[i + pawnShift[color]][j + shiftY] == V.EMPTY
48 ) {
241bf8f2 49 this.enlightened[color][i + pawnShift[color]][j + shiftY] = true;
dac39588
BA
50 }
51 }
52 }
53 }
54 }
55 }
56 const currentTurn = this.turn;
57 this.turn = "w";
58 const movesWhite = this.getAllValidMoves();
59 this.turn = "b";
60 const movesBlack = this.getAllValidMoves();
61 this.turn = currentTurn;
62 for (let move of movesWhite)
63 this.enlightened["w"][move.end.x][move.end.y] = true;
64 for (let move of movesBlack)
65 this.enlightened["b"][move.end.x][move.end.y] = true;
66 }
375ecdd1 67
dac39588 68 // Has to be redefined to avoid an infinite loop
6808d7a1 69 getAllValidMoves() {
dac39588 70 const color = this.turn;
dac39588 71 let potentialMoves = [];
6808d7a1
BA
72 for (let i = 0; i < V.size.x; i++) {
73 for (let j = 0; j < V.size.y; j++) {
74 if (this.board[i][j] != V.EMPTY && this.getColor(i, j) == color)
75 Array.prototype.push.apply(
76 potentialMoves,
77 this.getPotentialMovesFrom([i, j])
78 );
dac39588
BA
79 }
80 }
81 return potentialMoves; //because there are no checks
82 }
f6dbe8e3 83
6808d7a1 84 getCheckSquares() {
dac39588
BA
85 return [];
86 }
375ecdd1 87
6808d7a1 88 updateVariables(move) {
dac39588 89 super.updateVariables(move);
241bf8f2
BA
90 if (move.vanish.length >= 2 && move.vanish[1].p == V.KING)
91 // We took opponent king (because if castle vanish[1] is a rook)
6808d7a1 92 this.kingPos[this.turn] = [-1, -1];
388e4c40 93
dac39588
BA
94 // Update lights for both colors:
95 this.updateEnlightened();
96 }
388e4c40 97
6808d7a1 98 unupdateVariables(move) {
dac39588
BA
99 super.unupdateVariables(move);
100 const c = move.vanish[0].c;
101 const oppCol = V.GetOppCol(c);
241bf8f2
BA
102 if (this.kingPos[oppCol][0] < 0)
103 // Last move took opponent's king:
104 this.kingPos[oppCol] = [move.vanish[1].x, move.vanish[1].y];
388e4c40 105
dac39588
BA
106 // Update lights for both colors:
107 this.updateEnlightened();
108 }
375ecdd1 109
6808d7a1 110 getCurrentScore() {
dac39588
BA
111 const color = this.turn;
112 const kp = this.kingPos[color];
6808d7a1 113 if (kp[0] < 0)
241bf8f2 114 // King disappeared
6808d7a1 115 return color == "w" ? "0-1" : "1-0";
241bf8f2
BA
116 // Assume that stalemate is impossible (I think so. Would need proof...)
117 return "*";
dac39588 118 }
375ecdd1 119
6808d7a1 120 static get THRESHOLD_MATE() {
dac39588
BA
121 return 500; //checkmates evals may be slightly below 1000
122 }
5915f720 123
dac39588 124 // In this special situation, we just look 1 half move ahead
6808d7a1 125 getComputerMove() {
dac39588
BA
126 const maxeval = V.INFINITY;
127 const color = this.turn;
128 const oppCol = V.GetOppCol(color);
6808d7a1 129 const pawnShift = color == "w" ? -1 : 1;
5915f720 130
dac39588
BA
131 // Do not cheat: the current enlightment is all we can see
132 const myLight = JSON.parse(JSON.stringify(this.enlightened[color]));
5915f720 133
dac39588
BA
134 // Can a slider on (i,j) apparently take my king?
135 // NOTE: inaccurate because assume yes if some squares are shadowed
6808d7a1 136 const sliderTake = ([i, j], piece) => {
dac39588
BA
137 const kp = this.kingPos[color];
138 let step = undefined;
6808d7a1
BA
139 if (piece == V.BISHOP) {
140 if (Math.abs(kp[0] - i) == Math.abs(kp[1] - j)) {
141 step = [
142 (i - kp[0]) / Math.abs(i - kp[0]),
143 (j - kp[1]) / Math.abs(j - kp[1])
dac39588
BA
144 ];
145 }
6808d7a1
BA
146 } else if (piece == V.ROOK) {
147 if (kp[0] == i) step = [0, (j - kp[1]) / Math.abs(j - kp[1])];
148 else if (kp[1] == j) step = [(i - kp[0]) / Math.abs(i - kp[0]), 0];
dac39588 149 }
6808d7a1 150 if (!step) return false;
dac39588
BA
151 // Check for obstacles
152 let obstacle = false;
153 for (
6808d7a1 154 let x = kp[0] + step[0], y = kp[1] + step[1];
dac39588 155 x != i && y != j;
6808d7a1
BA
156 x += step[0], y += step[1]
157 ) {
158 if (myLight[x][y] && this.board[x][y] != V.EMPTY) {
dac39588
BA
159 obstacle = true;
160 break;
161 }
162 }
6808d7a1 163 if (!obstacle) return true;
dac39588
BA
164 return false;
165 };
5915f720 166
dac39588
BA
167 // Do I see something which can take my king ?
168 const kingThreats = () => {
169 const kp = this.kingPos[color];
6808d7a1
BA
170 for (let i = 0; i < V.size.x; i++) {
171 for (let j = 0; j < V.size.y; j++) {
172 if (
173 myLight[i][j] &&
174 this.board[i][j] != V.EMPTY &&
175 this.getColor(i, j) != color
176 ) {
177 switch (this.getPiece(i, j)) {
dac39588 178 case V.PAWN:
6808d7a1 179 if (kp[0] + pawnShift == i && Math.abs(kp[1] - j) == 1)
dac39588
BA
180 return true;
181 break;
182 case V.KNIGHT:
6808d7a1
BA
183 if (
184 (Math.abs(kp[0] - i) == 2 && Math.abs(kp[1] - j) == 1) ||
185 (Math.abs(kp[0] - i) == 1 && Math.abs(kp[1] - j) == 2)
186 ) {
dac39588
BA
187 return true;
188 }
189 break;
190 case V.KING:
191 if (Math.abs(kp[0] - i) == 1 && Math.abs(kp[1] - j) == 1)
192 return true;
193 break;
194 case V.BISHOP:
6808d7a1 195 if (sliderTake([i, j], V.BISHOP)) return true;
dac39588
BA
196 break;
197 case V.ROOK:
6808d7a1 198 if (sliderTake([i, j], V.ROOK)) return true;
dac39588
BA
199 break;
200 case V.QUEEN:
6808d7a1 201 if (sliderTake([i, j], V.BISHOP) || sliderTake([i, j], V.ROOK))
dac39588
BA
202 return true;
203 break;
204 }
205 }
206 }
207 }
208 return false;
209 };
5915f720 210
dac39588 211 let moves = this.getAllValidMoves();
6808d7a1 212 for (let move of moves) {
dac39588 213 this.play(move);
6808d7a1 214 if (this.kingPos[oppCol][0] >= 0 && kingThreats()) {
dac39588
BA
215 // We didn't take opponent king, and our king will be captured: bad
216 move.eval = -maxeval;
217 }
218 this.undo(move);
4f518610 219
6808d7a1 220 if (move.eval) continue;
5915f720 221
dac39588 222 move.eval = 0; //a priori...
5915f720 223
dac39588 224 // Can I take something ? If yes, do it if it seems good...
6808d7a1
BA
225 if (move.vanish.length == 2 && move.vanish[1].c != color) {
226 //avoid castle
dac39588
BA
227 const myPieceVal = V.VALUES[move.appear[0].p];
228 const hisPieceVal = V.VALUES[move.vanish[1].p];
6808d7a1
BA
229 if (myPieceVal <= hisPieceVal) move.eval = hisPieceVal - myPieceVal + 2;
230 //favor captures
231 else {
dac39588
BA
232 // Taking a pawn with minor piece,
233 // or minor piece or pawn with a rook,
234 // or anything but a queen with a queen,
235 // or anything with a king.
236 // ==> Do it at random, although
237 // this is clearly inferior to what a human can deduce...
6808d7a1 238 move.eval = Math.random() < 0.5 ? 1 : -1;
dac39588
BA
239 }
240 }
241 }
5915f720 242
dac39588
BA
243 // TODO: also need to implement the case when an opponent piece (in light)
244 // is threatening something - maybe not the king, but e.g. pawn takes rook.
5915f720 245
6808d7a1 246 moves.sort((a, b) => b.eval - a.eval);
dac39588 247 let candidates = [0];
6808d7a1 248 for (let j = 1; j < moves.length && moves[j].eval == moves[0].eval; j++)
dac39588
BA
249 candidates.push(j);
250 return moves[candidates[randInt(candidates.length)]];
251 }
6808d7a1 252};