Update TODO
[vchess.git] / client / src / variants / Dobutsu.js
1 import { ChessRules, PiPo, Move } from "@/base_rules";
2 import { ArrayFun } from "@/utils/array";
3 import { sample, shuffle } from "@/utils/alea";
4
5 export class DobutsuRules extends ChessRules {
6
7 static get HasFlags() {
8 return false;
9 }
10
11 static get HasEnpassant() {
12 return false;
13 }
14
15 static get Monochrome() {
16 return true;
17 }
18
19 static IsGoodFen(fen) {
20 if (!ChessRules.IsGoodFen(fen)) return false;
21 const fenParsed = V.ParseFen(fen);
22 // 3) Check reserves
23 if (!fenParsed.reserve || !fenParsed.reserve.match(/^[0-9]{14,14}$/))
24 return false;
25 return true;
26 }
27
28 static ParseFen(fen) {
29 const fenParts = fen.split(" ");
30 return Object.assign(
31 ChessRules.ParseFen(fen),
32 { reserve: fenParts[3] }
33 );
34 }
35
36 static get ELEPHANT() {
37 return "e";
38 }
39 static get GIRAFFE() {
40 return "g";
41 }
42 static get HEN() {
43 return "h";
44 }
45
46 static get PIECES() {
47 return [
48 ChessRules.PAWN,
49 ChessRules.KING,
50 V.ELEPHANT,
51 V.GIRAFFE,
52 V.HEN
53 ];
54 }
55
56 getPpath(b, color, score, orientation) {
57 // 'i' for "inversed":
58 const suffix = (b[0] == orientation ? "" : "i");
59 return "Dobutsu/" + b + suffix;
60 }
61
62 getPPpath(m, orientation) {
63 return (
64 this.getPpath(
65 m.appear[0].c + m.appear[0].p,
66 null,
67 null,
68 orientation
69 )
70 );
71 }
72
73 static GenRandInitFen() {
74 return "gke/1p1/1P1/EKG w 0 00000000";
75 }
76
77 getFen() {
78 return super.getFen() + " " + this.getReserveFen();
79 }
80
81 getFenForRepeat() {
82 return super.getFenForRepeat() + "_" + this.getReserveFen();
83 }
84
85 getReserveFen() {
86 let counts = new Array(6);
87 for (let i = 0; i < V.RESERVE_PIECES.length; i++) {
88 counts[i] = this.reserve["w"][V.RESERVE_PIECES[i]];
89 counts[3 + i] = this.reserve["b"][V.RESERVE_PIECES[i]];
90 }
91 return counts.join("");
92 }
93
94 setOtherVariables(fen) {
95 super.setOtherVariables(fen);
96 // Also init reserves (used by the interface to show landable pieces)
97 const reserve =
98 V.ParseFen(fen).reserve.split("").map(x => parseInt(x, 10));
99 this.reserve = {
100 w: {
101 [V.PAWN]: reserve[0],
102 [V.ELEPHANT]: reserve[1],
103 [V.GIRAFFE]: reserve[2]
104 },
105 b: {
106 [V.PAWN]: reserve[3],
107 [V.ELEPHANT]: reserve[4],
108 [V.GIRAFFE]: reserve[5]
109 }
110 };
111 }
112
113 // Goal is to capture the king, easier to not track kings
114 scanKings() {}
115
116 getColor(i, j) {
117 if (i >= V.size.x) return i == V.size.x ? "w" : "b";
118 return this.board[i][j].charAt(0);
119 }
120
121 getPiece(i, j) {
122 if (i >= V.size.x) return V.RESERVE_PIECES[j];
123 return this.board[i][j].charAt(1);
124 }
125
126 static get size() {
127 return { x: 4, y: 3};
128 }
129
130 getReservePpath(index, color, orientation) {
131 return (
132 "Dobutsu/" + color + V.RESERVE_PIECES[index] +
133 (color != orientation ? 'i' : '')
134 );
135 }
136
137 // Ordering on reserve pieces
138 static get RESERVE_PIECES() {
139 return (
140 // No king, since the goal is to capture it
141 [V.PAWN, V.ELEPHANT, V.GIRAFFE]
142 );
143 }
144
145 getReserveMoves([x, y]) {
146 const color = this.turn;
147 const p = V.RESERVE_PIECES[y];
148 if (this.reserve[color][p] == 0) return [];
149 let moves = [];
150 for (let i = 0; i < V.size.x; i++) {
151 for (let j = 0; j < V.size.y; j++) {
152 if (this.board[i][j] == V.EMPTY) {
153 let mv = new Move({
154 appear: [
155 new PiPo({
156 x: i,
157 y: j,
158 c: color,
159 p: p
160 })
161 ],
162 vanish: [],
163 start: { x: x, y: y }, //a bit artificial...
164 end: { x: i, y: j }
165 });
166 moves.push(mv);
167 }
168 }
169 }
170 return moves;
171 }
172
173 getPotentialMovesFrom(sq) {
174 if (sq[0] >= V.size.x) {
175 // Reserves, outside of board: x == sizeX(+1)
176 return this.getReserveMoves(sq);
177 }
178 switch (this.getPiece(sq[0], sq[1])) {
179 case V.PAWN: return this.getPotentialPawnMoves(sq);
180 case V.HEN: return this.getPotentialHenMoves(sq);
181 case V.ELEPHANT: return this.getPotentialElephantMoves(sq);
182 case V.GIRAFFE: return this.getPotentialGiraffeMoves(sq);
183 case V.KING: return super.getPotentialKingMoves(sq);
184 }
185 return []; //never reached
186 }
187
188 getPotentialPawnMoves([x, y]) {
189 const c = this.turn;
190 const beforeLastRank = (c == 'w' ? 1 : 2);
191 const forward = (c == 'w' ? -1 : 1);
192 if (!V.OnBoard(x + forward, y)) return []; //stuck pawn
193 if (
194 this.board[x + forward][y] == V.EMPTY ||
195 this.getColor(x + forward, y) != c
196 ) {
197 const tr = (x == beforeLastRank ? { p: V.HEN, c: c } : null);
198 return [super.getBasicMove([x, y], [x + forward, y], tr)];
199 }
200 }
201
202 getPotentialHenMoves(sq) {
203 const c = this.turn;
204 const forward = (c == 'w' ? -1 : 1);
205 const steps = V.steps[V.ROOK].concat([[forward, 1], [forward, -1]]);
206 return super.getSlideNJumpMoves(sq, steps, "oneStep");
207 }
208
209 getPotentialElephantMoves(sq) {
210 return super.getSlideNJumpMoves(sq, V.steps[V.BISHOP], "oneStep");
211 }
212
213 getPotentialGiraffeMoves(sq) {
214 return super.getSlideNJumpMoves(sq, V.steps[V.ROOK], "oneStep");
215 }
216
217 getAllValidMoves() {
218 let moves = super.getAllPotentialMoves();
219 const color = this.turn;
220 for (let i = 0; i < V.RESERVE_PIECES.length; i++) {
221 moves = moves.concat(
222 this.getReserveMoves([V.size.x + (color == "w" ? 0 : 1), i])
223 );
224 }
225 return moves;
226 }
227
228 // Goal is to capture the king:
229 isAttacked() {
230 return false;
231 }
232 filterValid(moves) {
233 return moves;
234 }
235 getCheckSquares() {
236 return [];
237 }
238
239 static MayDecode(piece) {
240 if (piece == V.HEN) return V.PAWN;
241 return piece;
242 }
243
244 postPlay(move) {
245 const color = move.appear[0].c;
246 if (move.vanish.length == 0)
247 // Drop unpromoted piece:
248 this.reserve[color][move.appear[0].p]--;
249 else if (move.vanish.length == 2)
250 // May capture a promoted piece:
251 this.reserve[color][V.MayDecode(move.vanish[1].p)]++;
252 }
253
254 postUndo(move) {
255 const color = this.turn;
256 if (move.vanish.length == 0)
257 this.reserve[color][move.appear[0].p]++;
258 else if (move.vanish.length == 2)
259 this.reserve[color][V.MayDecode(move.vanish[1].p)]--;
260 }
261
262 getCurrentScore() {
263 const c = this.turn;
264 if (this.board.every(row => row.every(cell => cell != c + 'k')))
265 return (c == 'w' ? "0-1" : "1-0");
266 const oppCol = V.GetOppCol(c);
267 const oppLastRank = (c == 'w' ? 3 : 0);
268 for (let j=0; j < V.size.y; j++) {
269 if (this.board[oppLastRank][j] == oppCol + 'k')
270 return (oppCol == 'w' ? "1-0" : "0-1");
271 }
272 return "*";
273 }
274
275 static get SEARCH_DEPTH() {
276 return 4;
277 }
278
279 static get VALUES() {
280 // NOTE: very arbitrary
281 return {
282 p: 1,
283 h: 4,
284 g: 3,
285 e: 2,
286 k: 1000
287 }
288 }
289
290 evalPosition() {
291 let evaluation = super.evalPosition();
292 // Add reserves:
293 for (let i = 0; i < V.RESERVE_PIECES.length; i++) {
294 const p = V.RESERVE_PIECES[i];
295 evaluation += this.reserve["w"][p] * V.VALUES[p];
296 evaluation -= this.reserve["b"][p] * V.VALUES[p];
297 }
298 return evaluation;
299 }
300
301 getNotation(move) {
302 const finalSquare = V.CoordsToSquare(move.end);
303 if (move.vanish.length == 0) {
304 // Rebirth:
305 const piece = move.appear[0].p.toUpperCase();
306 return (piece != 'P' ? piece : "") + "@" + finalSquare;
307 }
308 const piece = move.vanish[0].p.toUpperCase();
309 return (
310 (piece != 'P' || move.vanish.length == 2 ? piece : "") +
311 (move.vanish.length == 2 ? "x" : "") +
312 finalSquare +
313 (
314 move.appear[0].p != move.vanish[0].p
315 ? "=" + move.appear[0].p.toUpperCase()
316 : ""
317 )
318 );
319 }
320
321 };