Add Madrasi, fix Absorption, better pieces management
[xogo.git] / base_rules.js
CommitLineData
41534b92
BA
1import { Random } from "/utils/alea.js";
2import { ArrayFun } from "/utils/array.js";
3import PiPo from "/utils/PiPo.js";
4import Move from "/utils/Move.js";
5
6// NOTE: x coords: top to bottom (white perspective); y: left to right
cc2c7183 7// NOTE: ChessRules is aliased as window.C, and variants as window.V
41534b92
BA
8export default class ChessRules {
9
e5f93427 10 static get Aliases() {
3caec36f 11 return {'C': ChessRules};
e5f93427
BA
12 }
13
41534b92
BA
14 /////////////////////////
15 // VARIANT SPECIFICATIONS
16
17 // Some variants have specific options, like the number of pawns in Monster,
18 // or the board size for Pandemonium.
19 // Users can generally select a randomness level from 0 to 2.
20 static get Options() {
21 return {
41534b92
BA
22 select: [{
23 label: "Randomness",
24 variable: "randomness",
25 defaut: 0,
26 options: [
b4ae3ff6
BA
27 {label: "Deterministic", value: 0},
28 {label: "Symmetric random", value: 1},
29 {label: "Asymmetric random", value: 2}
41534b92
BA
30 ]
31 }],
f8b43ef7
BA
32 check: [
33 {
34 label: "Capture king",
35 defaut: false,
36 variable: "taking"
37 },
38 {
39 label: "Falling pawn",
40 defaut: false,
41 variable: "pawnfall"
42 }
43 ],
41534b92
BA
44 // Game modifiers (using "elementary variants"). Default: false
45 styles: [
46 "atomic",
47 "balance", //takes precedence over doublemove & progressive
48 "cannibal",
49 "capture",
50 "crazyhouse",
51 "cylinder", //ok with all
52 "dark",
53 "doublemove",
54 "madrasi",
55 "progressive", //(natural) priority over doublemove
56 "recycle",
57 "rifle",
58 "teleport",
59 "zen"
60 ]
61 };
62 }
63
c9ab0340
BA
64 get pawnPromotions() {
65 return ['q', 'r', 'n', 'b'];
41534b92
BA
66 }
67
68 // Some variants don't have flags:
69 get hasFlags() {
70 return true;
71 }
72 // Or castle
73 get hasCastle() {
74 return this.hasFlags;
75 }
76
77 // En-passant captures allowed?
78 get hasEnpassant() {
79 return true;
80 }
81
82 get hasReserve() {
83 return (
84 !!this.options["crazyhouse"] ||
85 (!!this.options["recycle"] && !this.options["teleport"])
86 );
87 }
88
89 get noAnimate() {
90 return !!this.options["dark"];
91 }
92
93 // Some variants use click infos:
94 doClick([x, y]) {
b4ae3ff6
BA
95 if (typeof x != "number")
96 return null; //click on reserves
41534b92 97 if (
cc2c7183 98 this.options["teleport"] && this.subTurnTeleport == 2 &&
41534b92
BA
99 this.board[x][y] == ""
100 ) {
101 return new Move({
102 start: {x: this.captured.x, y: this.captured.y},
103 appear: [
104 new PiPo({
105 x: x,
106 y: y,
107 c: this.captured.c, //this.turn,
108 p: this.captured.p
109 })
110 ],
111 vanish: []
112 });
113 }
114 return null;
115 }
116
117 ////////////////////
118 // COORDINATES UTILS
119
120 // 3 --> d (column number to letter)
121 static CoordToColumn(colnum) {
122 return String.fromCharCode(97 + colnum);
123 }
124
125 // d --> 3 (column letter to number)
126 static ColumnToCoord(columnStr) {
127 return columnStr.charCodeAt(0) - 97;
128 }
129
130 // 7 (numeric) --> 1 (str) [from black viewpoint].
131 static CoordToRow(rownum) {
132 return rownum;
133 }
134
135 // NOTE: wrong row index (1 should be 7 ...etc). But OK for the usage.
136 static RowToCoord(rownumStr) {
137 // NOTE: 30 is way more than enough (allow up to 29 rows on one character)
138 return parseInt(rownumStr, 30);
139 }
140
141 // a2 --> {x:2,y:0} (this is in fact a6)
142 static SquareToCoords(sq) {
143 return {
cc2c7183 144 x: C.RowToCoord(sq[1]),
41534b92 145 // NOTE: column is always one char => max 26 columns
cc2c7183 146 y: C.ColumnToCoord(sq[0])
41534b92
BA
147 };
148 }
149
150 // {x:0,y:4} --> e0 (should be e8)
151 static CoordsToSquare(coords) {
cc2c7183 152 return C.CoordToColumn(coords.y) + C.CoordToRow(coords.x);
41534b92
BA
153 }
154
155 coordsToId([x, y]) {
156 if (typeof x == "number")
157 return `${this.containerId}|sq-${x.toString(30)}-${y.toString(30)}`;
158 // Reserve :
159 return `${this.containerId}|rsq-${x}-${y}`;
160 }
161
162 idToCoords(targetId) {
b4ae3ff6
BA
163 if (!targetId)
164 return null; //outside page, maybe...
41534b92
BA
165 const idParts = targetId.split('|'); //prefix|sq-2-3 (start at 0 => 3,4)
166 if (
167 idParts.length < 2 ||
168 idParts[0] != this.containerId ||
169 !idParts[1].match(/sq-[0-9a-zA-Z]-[0-9a-zA-Z]/)
170 ) {
171 return null;
172 }
173 const squares = idParts[1].split('-');
174 if (squares[0] == "sq")
175 return [ parseInt(squares[1], 30), parseInt(squares[2], 30) ];
176 // squares[0] == "rsq" : reserve, 'c' + 'p' (letters)
177 return [squares[1], squares[2]];
178 }
179
180 /////////////
181 // FEN UTILS
182
183 // Turn "wb" into "B" (for FEN)
184 board2fen(b) {
185 return b[0] == "w" ? b[1].toUpperCase() : b[1];
186 }
187
188 // Turn "p" into "bp" (for board)
189 fen2board(f) {
190 return f.charCodeAt(0) <= 90 ? "w" + f.toLowerCase() : "b" + f;
191 }
192
193 // Setup the initial random-or-not (asymmetric-or-not) position
194 genRandInitFen(seed) {
195 Random.setSeed(seed);
196
197 let fen, flags = "0707";
cc2c7183 198 if (!this.options.randomness)
41534b92
BA
199 // Deterministic:
200 fen = "rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w 0";
201
202 else {
203 // Randomize
204 let pieces = { w: new Array(8), b: new Array(8) };
205 flags = "";
206 // Shuffle pieces on first (and last rank if randomness == 2)
207 for (let c of ["w", "b"]) {
208 if (c == 'b' && this.options.randomness == 1) {
209 pieces['b'] = pieces['w'];
210 flags += flags;
211 break;
212 }
213
214 let positions = ArrayFun.range(8);
215
216 // Get random squares for bishops
217 let randIndex = 2 * Random.randInt(4);
218 const bishop1Pos = positions[randIndex];
219 // The second bishop must be on a square of different color
220 let randIndex_tmp = 2 * Random.randInt(4) + 1;
221 const bishop2Pos = positions[randIndex_tmp];
222 // Remove chosen squares
223 positions.splice(Math.max(randIndex, randIndex_tmp), 1);
224 positions.splice(Math.min(randIndex, randIndex_tmp), 1);
225
226 // Get random squares for knights
227 randIndex = Random.randInt(6);
228 const knight1Pos = positions[randIndex];
229 positions.splice(randIndex, 1);
230 randIndex = Random.randInt(5);
231 const knight2Pos = positions[randIndex];
232 positions.splice(randIndex, 1);
233
234 // Get random square for queen
235 randIndex = Random.randInt(4);
236 const queenPos = positions[randIndex];
237 positions.splice(randIndex, 1);
238
239 // Rooks and king positions are now fixed,
240 // because of the ordering rook-king-rook
241 const rook1Pos = positions[0];
242 const kingPos = positions[1];
243 const rook2Pos = positions[2];
244
245 // Finally put the shuffled pieces in the board array
246 pieces[c][rook1Pos] = "r";
247 pieces[c][knight1Pos] = "n";
248 pieces[c][bishop1Pos] = "b";
249 pieces[c][queenPos] = "q";
250 pieces[c][kingPos] = "k";
251 pieces[c][bishop2Pos] = "b";
252 pieces[c][knight2Pos] = "n";
253 pieces[c][rook2Pos] = "r";
254 flags += rook1Pos.toString() + rook2Pos.toString();
255 }
256 fen = (
257 pieces["b"].join("") +
258 "/pppppppp/8/8/8/8/PPPPPPPP/" +
259 pieces["w"].join("").toUpperCase() +
260 " w 0"
261 );
262 }
263 // Add turn + flags + enpassant (+ reserve)
264 let parts = [];
b4ae3ff6
BA
265 if (this.hasFlags)
266 parts.push(`"flags":"${flags}"`);
267 if (this.hasEnpassant)
268 parts.push('"enpassant":"-"');
269 if (this.hasReserve)
270 parts.push('"reserve":"000000000000"');
271 if (this.options["crazyhouse"])
272 parts.push('"ispawn":"-"');
273 if (parts.length >= 1)
274 fen += " {" + parts.join(",") + "}";
41534b92
BA
275 return fen;
276 }
277
278 // "Parse" FEN: just return untransformed string data
279 parseFen(fen) {
280 const fenParts = fen.split(" ");
281 let res = {
282 position: fenParts[0],
283 turn: fenParts[1],
284 movesCount: fenParts[2]
285 };
b4ae3ff6
BA
286 if (fenParts.length > 3)
287 res = Object.assign(res, JSON.parse(fenParts[3]));
41534b92
BA
288 return res;
289 }
290
291 // Return current fen (game state)
292 getFen() {
293 let fen = (
294 this.getBaseFen() + " " +
295 this.getTurnFen() + " " +
296 this.movesCount
297 );
298 let parts = [];
b4ae3ff6
BA
299 if (this.hasFlags)
300 parts.push(`"flags":"${this.getFlagsFen()}"`);
41534b92
BA
301 if (this.hasEnpassant)
302 parts.push(`"enpassant":"${this.getEnpassantFen()}"`);
b4ae3ff6
BA
303 if (this.hasReserve)
304 parts.push(`"reserve":"${this.getReserveFen()}"`);
41534b92
BA
305 if (this.options["crazyhouse"])
306 parts.push(`"ispawn":"${this.getIspawnFen()}"`);
b4ae3ff6
BA
307 if (parts.length >= 1)
308 fen += " {" + parts.join(",") + "}";
41534b92
BA
309 return fen;
310 }
311
312 // Position part of the FEN string
313 getBaseFen() {
314 const format = (count) => {
315 // if more than 9 consecutive free spaces, break the integer,
316 // otherwise FEN parsing will fail.
b4ae3ff6
BA
317 if (count <= 9)
318 return count;
41534b92 319 // Most boards of size < 18:
b4ae3ff6
BA
320 if (count <= 18)
321 return "9" + (count - 9);
41534b92
BA
322 // Except Gomoku:
323 return "99" + (count - 18);
324 };
325 let position = "";
326 for (let i = 0; i < this.size.y; i++) {
327 let emptyCount = 0;
328 for (let j = 0; j < this.size.x; j++) {
b4ae3ff6
BA
329 if (this.board[i][j] == "")
330 emptyCount++;
41534b92
BA
331 else {
332 if (emptyCount > 0) {
333 // Add empty squares in-between
334 position += format(emptyCount);
335 emptyCount = 0;
336 }
337 position += this.board2fen(this.board[i][j]);
338 }
339 }
340 if (emptyCount > 0)
341 // "Flush remainder"
342 position += format(emptyCount);
b4ae3ff6
BA
343 if (i < this.size.y - 1)
344 position += "/"; //separate rows
41534b92
BA
345 }
346 return position;
347 }
348
349 getTurnFen() {
350 return this.turn;
351 }
352
353 // Flags part of the FEN string
354 getFlagsFen() {
355 return ["w", "b"].map(c => {
356 return this.castleFlags[c].map(x => x.toString(30)).join("");
357 }).join("");
358 }
359
360 // Enpassant part of the FEN string
361 getEnpassantFen() {
b4ae3ff6
BA
362 if (!this.epSquare)
363 return "-"; //no en-passant
cc2c7183 364 return C.CoordsToSquare(this.epSquare);
41534b92
BA
365 }
366
367 getReserveFen() {
368 return (
369 ["w","b"].map(c => Object.values(this.reserve[c]).join("")).join("")
370 );
371 }
372
373 getIspawnFen() {
374 const coords = Object.keys(this.ispawn);
b4ae3ff6
BA
375 if (coords.length == 0)
376 return "-";
f429756d 377 return coords.join(",");
41534b92
BA
378 }
379
380 // Set flags from fen (castle: white a,h then black a,h)
381 setFlags(fenflags) {
382 this.castleFlags = {
383 w: [0, 1].map(i => parseInt(fenflags.charAt(i), 30)),
384 b: [2, 3].map(i => parseInt(fenflags.charAt(i), 30))
385 };
386 }
387
388 //////////////////
389 // INITIALIZATION
390
41534b92
BA
391 constructor(o) {
392 this.options = o.options;
393 this.playerColor = o.color;
394 this.afterPlay = o.afterPlay;
395
c9ab0340 396 // Fen string fully describes the game state
b4ae3ff6
BA
397 if (!o.fen)
398 o.fen = this.genRandInitFen(o.seed);
41534b92
BA
399 const fenParsed = this.parseFen(o.fen);
400 this.board = this.getBoard(fenParsed.position);
401 this.turn = fenParsed.turn;
402 this.movesCount = parseInt(fenParsed.movesCount, 10);
403 this.setOtherVariables(fenParsed);
404
405 // Graphical (can use variables defined above)
406 this.containerId = o.element;
407 this.graphicalInit();
408 }
409
410 // Turn position fen into double array ["wb","wp","bk",...]
411 getBoard(position) {
412 const rows = position.split("/");
413 let board = ArrayFun.init(this.size.x, this.size.y, "");
414 for (let i = 0; i < rows.length; i++) {
415 let j = 0;
416 for (let indexInRow = 0; indexInRow < rows[i].length; indexInRow++) {
417 const character = rows[i][indexInRow];
418 const num = parseInt(character, 10);
419 // If num is a number, just shift j:
b4ae3ff6
BA
420 if (!isNaN(num))
421 j += num;
41534b92 422 // Else: something at position i,j
b4ae3ff6
BA
423 else
424 board[i][j++] = this.fen2board(character);
41534b92
BA
425 }
426 }
427 return board;
428 }
429
430 // Some additional variables from FEN (variant dependant)
431 setOtherVariables(fenParsed) {
432 // Set flags and enpassant:
b4ae3ff6
BA
433 if (this.hasFlags)
434 this.setFlags(fenParsed.flags);
41534b92
BA
435 if (this.hasEnpassant)
436 this.epSquare = this.getEpSquare(fenParsed.enpassant);
b4ae3ff6
BA
437 if (this.hasReserve)
438 this.initReserves(fenParsed.reserve);
439 if (this.options["crazyhouse"])
440 this.initIspawn(fenParsed.ispawn);
41534b92 441 this.subTurn = 1; //may be unused
cc2c7183
BA
442 if (this.options["teleport"]) {
443 this.subTurnTeleport = 1;
444 this.captured = null;
445 }
41534b92 446 if (this.options["dark"]) {
41534b92 447 // Setup enlightened: squares reachable by player side
c9ab0340
BA
448 this.enlightened = ArrayFun.init(this.size.x, this.size.y, false);
449 this.updateEnlightened();
41534b92
BA
450 }
451 }
452
c9ab0340
BA
453 updateEnlightened() {
454 this.oldEnlightened = this.enlightened;
455 this.enlightened = ArrayFun.init(this.size.x, this.size.y, false);
41534b92 456 // Add pieces positions + all squares reachable by moves (includes Zen):
41534b92
BA
457 for (let x=0; x<this.size.x; x++) {
458 for (let y=0; y<this.size.y; y++) {
459 if (this.board[x][y] != "" && this.getColor(x, y) == this.playerColor)
460 {
c9ab0340 461 this.enlightened[x][y] = true;
41534b92 462 this.getPotentialMovesFrom([x, y]).forEach(m => {
c9ab0340 463 this.enlightened[m.end.x][m.end.y] = true;
41534b92
BA
464 });
465 }
466 }
467 }
b4ae3ff6 468 if (this.epSquare)
c9ab0340 469 this.enlightEnpassant();
41534b92
BA
470 }
471
c9ab0340
BA
472 // Include square of the en-passant capturing square:
473 enlightEnpassant() {
474 // NOTE: shortcut, pawn has only one attack type, doesn't depend on square
475 const steps = this.pieces(this.playerColor)["p"].attack[0].steps;
41534b92
BA
476 for (let step of steps) {
477 const x = this.epSquare.x - step[0],
478 y = this.computeY(this.epSquare.y - step[1]);
479 if (
480 this.onBoard(x, y) &&
481 this.getColor(x, y) == this.playerColor &&
cc2c7183 482 this.getPieceType(x, y) == "p"
41534b92 483 ) {
c9ab0340 484 this.enlightened[x][this.epSquare.y] = true;
41534b92
BA
485 break;
486 }
487 }
488 }
489
c9ab0340
BA
490 // Apply diff this.enlightened --> oldEnlightened on board
491 graphUpdateEnlightened() {
3c61449b
BA
492 let chessboard =
493 document.getElementById(this.containerId).querySelector(".chessboard");
494 const r = chessboard.getBoundingClientRect();
41534b92
BA
495 const pieceWidth = this.getPieceWidth(r.width);
496 for (let x=0; x<this.size.x; x++) {
497 for (let y=0; y<this.size.y; y++) {
c9ab0340 498 if (!this.enlightened[x][y] && this.oldEnlightened[x][y]) {
41534b92 499 let elt = document.getElementById(this.coordsToId([x, y]));
2dba3fe1 500 elt.classList.add("in-shadow");
c9ab0340
BA
501 if (this.g_pieces[x][y])
502 this.g_pieces[x][y].classList.add("hidden");
41534b92 503 }
c9ab0340 504 else if (this.enlightened[x][y] && !this.oldEnlightened[x][y]) {
41534b92 505 let elt = document.getElementById(this.coordsToId([x, y]));
2dba3fe1 506 elt.classList.remove("in-shadow");
c9ab0340
BA
507 if (this.g_pieces[x][y])
508 this.g_pieces[x][y].classList.remove("hidden");
41534b92
BA
509 }
510 }
511 }
512 }
513
c9ab0340 514 // ordering as in pieces() p,r,n,b,q,k (+ count in base 30 if needed)
41534b92
BA
515 initReserves(reserveStr) {
516 const counts = reserveStr.split("").map(c => parseInt(c, 30));
517 this.reserve = { w: {}, b: {} };
c9ab0340
BA
518 const pieceName = ['p', 'r', 'n', 'b', 'q', 'k'];
519 const L = pieceName.length;
520 for (let i of ArrayFun.range(2 * L)) {
521 if (i < L)
b4ae3ff6
BA
522 this.reserve['w'][pieceName[i]] = counts[i];
523 else
c9ab0340 524 this.reserve['b'][pieceName[i-L]] = counts[i];
41534b92
BA
525 }
526 }
527
528 initIspawn(ispawnStr) {
529 if (ispawnStr != "-") {
cc2c7183 530 this.ispawn = ispawnStr.split(",").map(C.SquareToCoords)
41534b92
BA
531 .reduce((o, key) => ({ ...o, [key]: true}), {});
532 }
b4ae3ff6
BA
533 else
534 this.ispawn = {};
41534b92
BA
535 }
536
537 getNbReservePieces(color) {
538 return (
539 Object.values(this.reserve[color]).reduce(
540 (oldV,newV) => oldV + (newV > 0 ? 1 : 0), 0)
541 );
542 }
543
544 //////////////
545 // VISUAL PART
546
547 getPieceWidth(rwidth) {
548 return (rwidth / this.size.y);
549 }
550
551 getSquareWidth(rwidth) {
552 return this.getPieceWidth(rwidth);
553 }
554
555 getReserveSquareSize(rwidth, nbR) {
556 const sqSize = this.getSquareWidth(rwidth);
557 return Math.min(sqSize, rwidth / nbR);
558 }
559
560 getReserveNumId(color, piece) {
561 return `${this.containerId}|rnum-${color}${piece}`;
562 }
563
564 graphicalInit() {
565 // NOTE: not window.onresize = this.re_drawBoardElts because scope (this)
566 window.onresize = () => this.re_drawBoardElements();
567 this.re_drawBoardElements();
568 this.initMouseEvents();
3c61449b
BA
569 const chessboard =
570 document.getElementById(this.containerId).querySelector(".chessboard");
571 new ResizeObserver(this.rescale).observe(chessboard);
41534b92
BA
572 }
573
574 re_drawBoardElements() {
575 const board = this.getSvgChessboard();
cc2c7183 576 const oppCol = C.GetOppCol(this.playerColor);
3c61449b
BA
577 let chessboard =
578 document.getElementById(this.containerId).querySelector(".chessboard");
579 chessboard.innerHTML = "";
580 chessboard.insertAdjacentHTML('beforeend', board);
41534b92
BA
581 const aspectRatio = this.size.y / this.size.x;
582 // Compare window ratio width / height to aspectRatio:
583 const windowRatio = window.innerWidth / window.innerHeight;
584 let cbWidth, cbHeight;
585 if (windowRatio <= aspectRatio) {
586 // Limiting dimension is width:
587 cbWidth = Math.min(window.innerWidth, 767);
588 cbHeight = cbWidth / aspectRatio;
589 }
590 else {
591 // Limiting dimension is height:
592 cbHeight = Math.min(window.innerHeight, 767);
593 cbWidth = cbHeight * aspectRatio;
594 }
595 if (this.reserve) {
596 const sqSize = cbWidth / this.size.y;
597 // NOTE: allocate space for reserves (up/down) even if they are empty
598 if ((window.innerHeight - cbHeight) / 2 < sqSize + 5) {
599 cbHeight = window.innerHeight - 2 * (sqSize + 5);
600 cbWidth = cbHeight * aspectRatio;
601 }
602 }
3c61449b
BA
603 chessboard.style.width = cbWidth + "px";
604 chessboard.style.height = cbHeight + "px";
41534b92
BA
605 // Center chessboard:
606 const spaceLeft = (window.innerWidth - cbWidth) / 2,
607 spaceTop = (window.innerHeight - cbHeight) / 2;
3c61449b
BA
608 chessboard.style.left = spaceLeft + "px";
609 chessboard.style.top = spaceTop + "px";
41534b92
BA
610 // Give sizes instead of recomputing them,
611 // because chessboard might not be drawn yet.
612 this.setupPieces({
613 width: cbWidth,
614 height: cbHeight,
615 x: spaceLeft,
616 y: spaceTop
617 });
618 }
619
620 // Get SVG board (background, no pieces)
621 getSvgChessboard() {
622 const [sizeX, sizeY] = [this.size.x, this.size.y];
623 const flipped = (this.playerColor == 'b');
624 let board = `
625 <svg
626 viewBox="0 0 80 80"
627 version="1.1"
3c61449b 628 class="chessboard_SVG">
41534b92
BA
629 <g>`;
630 for (let i=0; i < sizeX; i++) {
631 for (let j=0; j < sizeY; j++) {
632 const ii = (flipped ? this.size.x - 1 - i : i);
633 const jj = (flipped ? this.size.y - 1 - j : j);
c7bf7b1b
BA
634 let classes = this.getSquareColorClass(ii, jj);
635 if (this.enlightened && !this.enlightened[ii][jj])
636 classes += " in-shadow";
41534b92 637 // NOTE: x / y reversed because coordinates system is reversed.
c7bf7b1b
BA
638 board += `<rect
639 class="${classes}"
41534b92
BA
640 id="${this.coordsToId([ii, jj])}"
641 width="10"
642 height="10"
643 x="${10*j}"
644 y="${10*i}" />`;
645 }
646 }
647 board += "</g></svg>";
648 return board;
649 }
650
cc2c7183 651 // Generally light square bottom-right
c7bf7b1b
BA
652 getSquareColorClass(i, j) {
653 return ((i+j) % 2 == 0 ? "light-square": "dark-square");
41534b92
BA
654 }
655
656 setupPieces(r) {
657 if (this.g_pieces) {
658 // Refreshing: delete old pieces first
659 for (let i=0; i<this.size.x; i++) {
660 for (let j=0; j<this.size.y; j++) {
661 if (this.g_pieces[i][j]) {
662 this.g_pieces[i][j].remove();
663 this.g_pieces[i][j] = null;
664 }
665 }
666 }
667 }
b4ae3ff6
BA
668 else
669 this.g_pieces = ArrayFun.init(this.size.x, this.size.y, null);
3c61449b
BA
670 let chessboard =
671 document.getElementById(this.containerId).querySelector(".chessboard");
b4ae3ff6
BA
672 if (!r)
673 r = chessboard.getBoundingClientRect();
41534b92
BA
674 const pieceWidth = this.getPieceWidth(r.width);
675 for (let i=0; i < this.size.x; i++) {
676 for (let j=0; j < this.size.y; j++) {
c9ab0340 677 if (this.board[i][j] != "") {
41534b92 678 const color = this.getColor(i, j);
cc2c7183 679 const piece = this.getPiece(i, j);
41534b92
BA
680 this.g_pieces[i][j] = document.createElement("piece");
681 this.g_pieces[i][j].classList.add(this.pieces()[piece]["class"]);
682 this.g_pieces[i][j].classList.add(color == "w" ? "white" : "black");
683 this.g_pieces[i][j].style.width = pieceWidth + "px";
684 this.g_pieces[i][j].style.height = pieceWidth + "px";
685 const [ip, jp] = this.getPixelPosition(i, j, r);
686 this.g_pieces[i][j].style.transform = `translate(${ip}px,${jp}px)`;
c9ab0340
BA
687 if (this.enlightened && !this.enlightened[i][j])
688 this.g_pieces[i][j].classList.add("hidden");
3c61449b 689 chessboard.appendChild(this.g_pieces[i][j]);
41534b92
BA
690 }
691 }
692 }
b4ae3ff6
BA
693 if (this.reserve)
694 this.re_drawReserve(['w', 'b'], r);
41534b92
BA
695 }
696
697 // NOTE: assume !!this.reserve
698 re_drawReserve(colors, r) {
699 if (this.r_pieces) {
700 // Remove (old) reserve pieces
701 for (let c of colors) {
b4ae3ff6
BA
702 if (!this.reserve[c])
703 continue;
41534b92
BA
704 Object.keys(this.reserve[c]).forEach(p => {
705 if (this.r_pieces[c][p]) {
706 this.r_pieces[c][p].remove();
707 delete this.r_pieces[c][p];
708 const numId = this.getReserveNumId(c, p);
709 document.getElementById(numId).remove();
710 }
711 });
712 let reservesDiv = document.getElementById("reserves_" + c);
b4ae3ff6
BA
713 if (reservesDiv)
714 reservesDiv.remove();
41534b92
BA
715 }
716 }
b4ae3ff6
BA
717 else
718 this.r_pieces = { 'w': {}, 'b': {} };
f429756d 719 let container = document.getElementById(this.containerId);
b4ae3ff6 720 if (!r)
f429756d 721 r = container.querySelector(".chessboard").getBoundingClientRect();
41534b92 722 for (let c of colors) {
b4ae3ff6
BA
723 if (!this.reserve[c])
724 continue;
41534b92 725 const nbR = this.getNbReservePieces(c);
b4ae3ff6
BA
726 if (nbR == 0)
727 continue;
41534b92
BA
728 const sqResSize = this.getReserveSquareSize(r.width, nbR);
729 let ridx = 0;
730 const vShift = (c == this.playerColor ? r.height + 5 : -sqResSize - 5);
731 const [i0, j0] = [r.x, r.y + vShift];
732 let rcontainer = document.createElement("div");
733 rcontainer.id = "reserves_" + c;
734 rcontainer.classList.add("reserves");
735 rcontainer.style.left = i0 + "px";
736 rcontainer.style.top = j0 + "px";
1aa9054d
BA
737 // NOTE: +1 fix display bug on Firefox at least
738 rcontainer.style.width = (nbR * sqResSize + 1) + "px";
41534b92 739 rcontainer.style.height = sqResSize + "px";
f429756d 740 container.appendChild(rcontainer);
41534b92 741 for (let p of Object.keys(this.reserve[c])) {
b4ae3ff6
BA
742 if (this.reserve[c][p] == 0)
743 continue;
41534b92
BA
744 let r_cell = document.createElement("div");
745 r_cell.id = this.coordsToId([c, p]);
746 r_cell.classList.add("reserve-cell");
1aa9054d
BA
747 r_cell.style.width = sqResSize + "px";
748 r_cell.style.height = sqResSize + "px";
41534b92
BA
749 rcontainer.appendChild(r_cell);
750 let piece = document.createElement("piece");
c9ab0340 751 const pieceSpec = this.pieces()[p];
41534b92
BA
752 piece.classList.add(pieceSpec["class"]);
753 piece.classList.add(c == 'w' ? "white" : "black");
754 piece.style.width = "100%";
755 piece.style.height = "100%";
756 this.r_pieces[c][p] = piece;
757 r_cell.appendChild(piece);
758 let number = document.createElement("div");
759 number.textContent = this.reserve[c][p];
760 number.classList.add("reserve-num");
761 number.id = this.getReserveNumId(c, p);
762 const fontSize = "1.3em";
763 number.style.fontSize = fontSize;
764 number.style.fontSize = fontSize;
765 r_cell.appendChild(number);
766 ridx++;
767 }
768 }
769 }
770
771 updateReserve(color, piece, count) {
55a15dcb 772 if (this.options["cannibal"] && C.CannibalKings[piece])
cc2c7183 773 piece = "k"; //capturing cannibal king: back to king form
41534b92
BA
774 const oldCount = this.reserve[color][piece];
775 this.reserve[color][piece] = count;
776 // Redrawing is much easier if count==0
b4ae3ff6
BA
777 if ([oldCount, count].includes(0))
778 this.re_drawReserve([color]);
41534b92
BA
779 else {
780 const numId = this.getReserveNumId(color, piece);
781 document.getElementById(numId).textContent = count;
782 }
783 }
784
785 // After resize event: no need to destroy/recreate pieces
786 rescale() {
3c61449b 787 const container = document.getElementById(this.containerId);
b4ae3ff6
BA
788 if (!container)
789 return; //useful at initial loading
3c61449b
BA
790 let chessboard = container.querySelector(".chessboard");
791 const r = chessboard.getBoundingClientRect();
41534b92
BA
792 const newRatio = r.width / r.height;
793 const aspectRatio = this.size.y / this.size.x;
794 let newWidth = r.width,
795 newHeight = r.height;
796 if (newRatio > aspectRatio) {
797 newWidth = r.height * aspectRatio;
3c61449b 798 chessboard.style.width = newWidth + "px";
41534b92
BA
799 }
800 else if (newRatio < aspectRatio) {
801 newHeight = r.width / aspectRatio;
3c61449b 802 chessboard.style.height = newHeight + "px";
41534b92
BA
803 }
804 const newX = (window.innerWidth - newWidth) / 2;
3c61449b 805 chessboard.style.left = newX + "px";
41534b92 806 const newY = (window.innerHeight - newHeight) / 2;
3c61449b 807 chessboard.style.top = newY + "px";
41534b92
BA
808 const newR = { x: newX, y: newY, width: newWidth, height: newHeight };
809 const pieceWidth = this.getPieceWidth(newWidth);
810 for (let i=0; i < this.size.x; i++) {
811 for (let j=0; j < this.size.y; j++) {
c9ab0340 812 if (this.g_pieces[i][j]) {
41534b92
BA
813 // NOTE: could also use CSS transform "scale"
814 this.g_pieces[i][j].style.width = pieceWidth + "px";
815 this.g_pieces[i][j].style.height = pieceWidth + "px";
816 const [ip, jp] = this.getPixelPosition(i, j, newR);
817 this.g_pieces[i][j].style.transform = `translate(${ip}px,${jp}px)`;
818 }
819 }
820 }
b4ae3ff6
BA
821 if (this.reserve)
822 this.rescaleReserve(newR);
41534b92
BA
823 }
824
825 rescaleReserve(r) {
41534b92 826 for (let c of ['w','b']) {
b4ae3ff6
BA
827 if (!this.reserve[c])
828 continue;
41534b92 829 const nbR = this.getNbReservePieces(c);
b4ae3ff6
BA
830 if (nbR == 0)
831 continue;
41534b92
BA
832 // Resize container first
833 const sqResSize = this.getReserveSquareSize(r.width, nbR);
834 const vShift = (c == this.playerColor ? r.height + 5 : -sqResSize - 5);
835 const [i0, j0] = [r.x, r.y + vShift];
836 let rcontainer = document.getElementById("reserves_" + c);
837 rcontainer.style.left = i0 + "px";
838 rcontainer.style.top = j0 + "px";
1aa9054d 839 rcontainer.style.width = (nbR * sqResSize + 1) + "px";
41534b92
BA
840 rcontainer.style.height = sqResSize + "px";
841 // And then reserve cells:
842 const rpieceWidth = this.getReserveSquareSize(r.width, nbR);
843 Object.keys(this.reserve[c]).forEach(p => {
b4ae3ff6
BA
844 if (this.reserve[c][p] == 0)
845 return;
41534b92 846 let r_cell = document.getElementById(this.coordsToId([c, p]));
1aa9054d
BA
847 r_cell.style.width = sqResSize + "px";
848 r_cell.style.height = sqResSize + "px";
41534b92
BA
849 });
850 }
851 }
852
3c61449b 853 // Return the absolute pixel coordinates (on board) given current position.
41534b92
BA
854 // Our coordinate system differs from CSS one (x <--> y).
855 // We return here the CSS coordinates (more useful).
856 getPixelPosition(i, j, r) {
857 const sqSize = this.getSquareWidth(r.width);
b4ae3ff6
BA
858 if (i < 0 || j < 0)
859 return [0, 0]; //piece vanishes
41534b92
BA
860 const flipped = (this.playerColor == 'b');
861 const x = (flipped ? this.size.y - 1 - j : j) * sqSize;
862 const y = (flipped ? this.size.x - 1 - i : i) * sqSize;
863 return [x, y];
864 }
865
866 initMouseEvents() {
3c61449b
BA
867 let chessboard =
868 document.getElementById(this.containerId).querySelector(".chessboard");
41534b92
BA
869
870 const getOffset = e => {
3c61449b
BA
871 if (e.clientX)
872 // Mouse
873 return {x: e.clientX, y: e.clientY};
41534b92
BA
874 let touchLocation = null;
875 if (e.targetTouches && e.targetTouches.length >= 1)
876 // Touch screen, dragstart
877 touchLocation = e.targetTouches[0];
878 else if (e.changedTouches && e.changedTouches.length >= 1)
879 // Touch screen, dragend
880 touchLocation = e.changedTouches[0];
881 if (touchLocation)
11625344 882 return {x: touchLocation.clientX, y: touchLocation.clientY};
57b8015b 883 return {x: 0, y: 0}; //shouldn't reach here =)
41534b92
BA
884 }
885
886 const centerOnCursor = (piece, e) => {
887 const centerShift = sqSize / 2;
888 const offset = getOffset(e);
3c61449b
BA
889 piece.style.left = (offset.x - r.x - centerShift) + "px";
890 piece.style.top = (offset.y - r.y - centerShift) + "px";
41534b92
BA
891 }
892
893 let start = null,
894 r = null,
895 startPiece, curPiece = null,
896 sqSize;
897 const mousedown = (e) => {
cb17fed8 898 // Disable zoom on smartphones:
b4ae3ff6
BA
899 if (e.touches && e.touches.length > 1)
900 e.preventDefault();
3c61449b 901 r = chessboard.getBoundingClientRect();
41534b92
BA
902 sqSize = this.getSquareWidth(r.width);
903 const square = this.idToCoords(e.target.id);
904 if (square) {
905 const [i, j] = square;
906 const move = this.doClick([i, j]);
b4ae3ff6
BA
907 if (move)
908 this.playPlusVisual(move);
41534b92 909 else {
b4ae3ff6
BA
910 if (typeof i != "number")
911 startPiece = this.r_pieces[i][j];
912 else if (this.g_pieces[i][j])
913 startPiece = this.g_pieces[i][j];
41534b92
BA
914 if (startPiece && this.canIplay(i, j)) {
915 e.preventDefault();
916 start = { x: i, y: j };
917 curPiece = startPiece.cloneNode();
918 curPiece.style.transform = "none";
919 curPiece.style.zIndex = 5;
920 curPiece.style.width = sqSize + "px";
921 curPiece.style.height = sqSize + "px";
922 centerOnCursor(curPiece, e);
3c61449b 923 chessboard.appendChild(curPiece);
41534b92 924 startPiece.style.opacity = "0.4";
3c61449b 925 chessboard.style.cursor = "none";
41534b92
BA
926 }
927 }
928 }
929 };
930
931 const mousemove = (e) => {
932 if (start) {
933 e.preventDefault();
934 centerOnCursor(curPiece, e);
935 }
11625344
BA
936 else if (e.changedTouches && e.changedTouches.length >= 1)
937 // Attempt to prevent horizontal swipe...
938 e.preventDefault();
41534b92
BA
939 };
940
941 const mouseup = (e) => {
3c61449b 942 const newR = chessboard.getBoundingClientRect();
41534b92
BA
943 if (newR.width != r.width || newR.height != r.height) {
944 this.rescale();
945 return;
946 }
b4ae3ff6
BA
947 if (!start)
948 return;
41534b92
BA
949 const [x, y] = [start.x, start.y];
950 start = null;
951 e.preventDefault();
3c61449b 952 chessboard.style.cursor = "pointer";
41534b92
BA
953 startPiece.style.opacity = "1";
954 const offset = getOffset(e);
955 const landingElt = document.elementFromPoint(offset.x, offset.y);
f429756d 956 const sq = landingElt ? this.idToCoords(landingElt.id) : undefined;
41534b92
BA
957 if (sq) {
958 const [i, j] = sq;
959 // NOTE: clearly suboptimal, but much easier, and not a big deal.
960 const potentialMoves = this.getPotentialMovesFrom([x, y])
961 .filter(m => m.end.x == i && m.end.y == j);
962 const moves = this.filterValid(potentialMoves);
b4ae3ff6
BA
963 if (moves.length >= 2)
964 this.showChoices(moves, r);
965 else if (moves.length == 1)
966 this.playPlusVisual(moves[0], r);
41534b92
BA
967 }
968 curPiece.remove();
969 };
970
971 if ('onmousedown' in window) {
972 document.addEventListener("mousedown", mousedown);
973 document.addEventListener("mousemove", mousemove);
974 document.addEventListener("mouseup", mouseup);
975 }
976 if ('ontouchstart' in window) {
cb17fed8
BA
977 // https://stackoverflow.com/a/42509310/12660887
978 document.addEventListener("touchstart", mousedown, {passive: false});
979 document.addEventListener("touchmove", mousemove, {passive: false});
980 document.addEventListener("touchend", mouseup, {passive: false});
41534b92 981 }
11625344 982 // TODO: onpointerdown/move/up ? See reveal.js /controllers/touch.js
41534b92
BA
983 }
984
985 showChoices(moves, r) {
986 let container = document.getElementById(this.containerId);
3c61449b 987 let chessboard = container.querySelector(".chessboard");
41534b92
BA
988 let choices = document.createElement("div");
989 choices.id = "choices";
990 choices.style.width = r.width + "px";
991 choices.style.height = r.height + "px";
992 choices.style.left = r.x + "px";
993 choices.style.top = r.y + "px";
3c61449b
BA
994 chessboard.style.opacity = "0.5";
995 container.appendChild(choices);
41534b92
BA
996 const squareWidth = this.getSquareWidth(r.width);
997 const firstUpLeft = (r.width - (moves.length * squareWidth)) / 2;
998 const firstUpTop = (r.height - squareWidth) / 2;
999 const color = moves[0].appear[0].c;
1000 const callback = (m) => {
3c61449b
BA
1001 chessboard.style.opacity = "1";
1002 container.removeChild(choices);
41534b92
BA
1003 this.playPlusVisual(m, r);
1004 }
1005 for (let i=0; i < moves.length; i++) {
1006 let choice = document.createElement("div");
1007 choice.classList.add("choice");
1008 choice.style.width = squareWidth + "px";
1009 choice.style.height = squareWidth + "px";
1010 choice.style.left = (firstUpLeft + i * squareWidth) + "px";
1011 choice.style.top = firstUpTop + "px";
1012 choice.style.backgroundColor = "lightyellow";
1013 choice.onclick = () => callback(moves[i]);
1014 const piece = document.createElement("piece");
c9ab0340 1015 const pieceSpec = this.pieces()[moves[i].appear[0].p];
41534b92
BA
1016 piece.classList.add(pieceSpec["class"]);
1017 piece.classList.add(color == 'w' ? "white" : "black");
1018 piece.style.width = "100%";
1019 piece.style.height = "100%";
1020 choice.appendChild(piece);
1021 choices.appendChild(choice);
1022 }
1023 }
1024
1025 //////////////
1026 // BASIC UTILS
1027
1028 get size() {
c9ab0340 1029 return {"x": 8, "y": 8};
41534b92
BA
1030 }
1031
1032 // Color of thing on square (i,j). 'undefined' if square is empty
1033 getColor(i, j) {
1034 return this.board[i][j].charAt(0);
1035 }
1036
cc2c7183 1037 // Assume square i,j isn't empty
41534b92
BA
1038 getPiece(i, j) {
1039 return this.board[i][j].charAt(1);
1040 }
1041
cc2c7183
BA
1042 // Piece type on square (i,j)
1043 getPieceType(i, j) {
1044 const p = this.board[i][j].charAt(1);
1045 return C.CannibalKings[p] || p; //a cannibal king move as...
1046 }
1047
41534b92
BA
1048 // Get opponent color
1049 static GetOppCol(color) {
1050 return (color == "w" ? "b" : "w");
1051 }
1052
c9ab0340 1053 // Can thing on square1 capture (no return) thing on square2?
41534b92 1054 canTake([x1, y1], [x2, y2]) {
c9ab0340 1055 return (this.getColor(x1, y1) !== this.getColor(x2, y2));
41534b92
BA
1056 }
1057
1058 // Is (x,y) on the chessboard?
1059 onBoard(x, y) {
b99ce1fb
BA
1060 return (x >= 0 && x < this.size.x &&
1061 y >= 0 && y < this.size.y);
41534b92
BA
1062 }
1063
c9ab0340 1064 // Used in interface
41534b92
BA
1065 canIplay(x, y) {
1066 return (
1067 this.playerColor == this.turn &&
1068 (
1069 (typeof x == "number" && this.getColor(x, y) == this.turn) ||
1070 (typeof x == "string" && x == this.turn) //reserve
1071 )
1072 );
1073 }
1074
1075 ////////////////////////
1076 // PIECES SPECIFICATIONS
1077
c9ab0340 1078 pieces(color, x, y) {
41534b92 1079 const pawnShift = (color == "w" ? -1 : 1);
c9ab0340 1080 const initRank = ((color == 'w' && x == 6) || (color == 'b' && x == 1));
41534b92
BA
1081 return {
1082 'p': {
1083 "class": "pawn",
c9ab0340
BA
1084 moves: [
1085 {
1086 steps: [[pawnShift, 0]],
1087 range: (initRank ? 2 : 1)
1088 }
1089 ],
1090 attack: [
1091 {
1092 steps: [[pawnShift, 1], [pawnShift, -1]],
1093 range: 1
1094 }
1095 ]
41534b92
BA
1096 },
1097 // rook
1098 'r': {
1099 "class": "rook",
c9ab0340
BA
1100 moves: [
1101 {steps: [[0, 1], [0, -1], [1, 0], [-1, 0]]}
1102 ]
41534b92
BA
1103 },
1104 // knight
1105 'n': {
1106 "class": "knight",
c9ab0340
BA
1107 moves: [
1108 {
1109 steps: [
1110 [1, 2], [1, -2], [-1, 2], [-1, -2],
1111 [2, 1], [-2, 1], [2, -1], [-2, -1]
1112 ],
1113 range: 1
1114 }
1115 ]
41534b92
BA
1116 },
1117 // bishop
1118 'b': {
1119 "class": "bishop",
c9ab0340
BA
1120 moves: [
1121 {steps: [[1, 1], [1, -1], [-1, 1], [-1, -1]]}
1122 ]
41534b92
BA
1123 },
1124 // queen
1125 'q': {
1126 "class": "queen",
c9ab0340
BA
1127 moves: [
1128 {
1129 steps: [
1130 [0, 1], [0, -1], [1, 0], [-1, 0],
1131 [1, 1], [1, -1], [-1, 1], [-1, -1]
1132 ]
1133 }
41534b92
BA
1134 ]
1135 },
1136 // king
1137 'k': {
1138 "class": "king",
c9ab0340
BA
1139 moves: [
1140 {
1141 steps: [
1142 [0, 1], [0, -1], [1, 0], [-1, 0],
1143 [1, 1], [1, -1], [-1, 1], [-1, -1]
1144 ],
1145 range: 1
1146 }
1147 ]
cc2c7183
BA
1148 },
1149 // Cannibal kings:
c9ab0340
BA
1150 '!': {"class": "king-pawn", moveas: "p"},
1151 '#': {"class": "king-rook", moveas: "r"},
1152 '$': {"class": "king-knight", moveas: "n"},
1153 '%': {"class": "king-bishop", moveas: "b"},
1154 '*': {"class": "king-queen", moveas: "q"}
41534b92
BA
1155 };
1156 }
1157
41534b92
BA
1158 ////////////////////
1159 // MOVES GENERATION
1160
1161 // For Cylinder: get Y coordinate
1162 computeY(y) {
b4ae3ff6
BA
1163 if (!this.options["cylinder"])
1164 return y;
41534b92 1165 let res = y % this.size.y;
b4ae3ff6
BA
1166 if (res < 0)
1167 res += this.size.y;
41534b92
BA
1168 return res;
1169 }
1170
1171 // Stop at the first capture found
1172 atLeastOneCapture(color) {
1173 color = color || this.turn;
cc2c7183 1174 const oppCol = C.GetOppCol(color);
41534b92
BA
1175 for (let i = 0; i < this.size.x; i++) {
1176 for (let j = 0; j < this.size.y; j++) {
1177 if (this.board[i][j] != "" && this.getColor(i, j) == color) {
c9ab0340
BA
1178 const allSpecs = this.pieces(color, i, j)
1179 let specs = allSpecs[this.getPieceType(i, j)];
1180 const attacks = specs.attack || specs.moves;
1181 for (let a of attacks) {
1182 outerLoop: for (let step of a.steps) {
1183 let [ii, jj] = [i + step[0], this.computeY(j + step[1])];
1184 let stepCounter = 1;
1185 while (this.onBoard(ii, jj) && this.board[ii][jj] == "") {
1186 if (a.range <= stepCounter++)
1187 continue outerLoop;
1188 ii += step[0];
1189 jj = this.computeY(jj + step[1]);
1190 }
1191 if (
1192 this.onBoard(ii, jj) &&
1193 this.getColor(ii, jj) == oppCol &&
1194 this.filterValid(
1195 [this.getBasicMove([i, j], [ii, jj])]
1196 ).length >= 1
1197 ) {
1198 return true;
1199 }
41534b92
BA
1200 }
1201 }
1202 }
1203 }
1204 }
1205 return false;
1206 }
1207
1208 getDropMovesFrom([c, p]) {
1209 // NOTE: by design, this.reserve[c][p] >= 1 on user click
1210 // (but not necessarily otherwise)
b4ae3ff6
BA
1211 if (this.reserve[c][p] == 0)
1212 return [];
41534b92
BA
1213 let moves = [];
1214 for (let i=0; i<this.size.x; i++) {
1215 for (let j=0; j<this.size.y; j++) {
41534b92
BA
1216 if (
1217 this.board[i][j] == "" &&
c9ab0340 1218 (!this.enlightened || this.enlightened[i][j]) &&
41534b92 1219 (
cc2c7183 1220 p != "p" ||
41534b92
BA
1221 (c == 'w' && i < this.size.x - 1) ||
1222 (c == 'b' && i > 0)
1223 )
1224 ) {
1225 moves.push(
1226 new Move({
1227 start: {x: c, y: p},
1228 end: {x: i, y: j},
1229 appear: [new PiPo({x: i, y: j, c: c, p: p})],
1230 vanish: []
1231 })
1232 );
1233 }
1234 }
1235 }
1236 return moves;
1237 }
1238
1239 // All possible moves from selected square
c7bf7b1b 1240 getPotentialMovesFrom(sq, color) {
b4ae3ff6
BA
1241 if (typeof sq[0] == "string")
1242 return this.getDropMovesFrom(sq);
57b8015b 1243 if (this.isImmobilized(sq))
b4ae3ff6 1244 return [];
cc2c7183 1245 const piece = this.getPieceType(sq[0], sq[1]);
c9ab0340
BA
1246 let moves = this.getPotentialMovesOf(piece, sq);
1247 if (
1248 piece == "p" &&
1249 this.hasEnpassant &&
1250 this.epSquare
1251 ) {
1252 Array.prototype.push.apply(moves, this.getEnpassantCaptures(sq));
1253 }
41534b92 1254 if (
cc2c7183 1255 piece == "k" &&
41534b92
BA
1256 this.hasCastle &&
1257 this.castleFlags[color || this.turn].some(v => v < this.size.y)
1258 ) {
1259 Array.prototype.push.apply(moves, this.getCastleMoves(sq));
1260 }
1261 return this.postProcessPotentialMoves(moves);
1262 }
1263
1264 postProcessPotentialMoves(moves) {
b4ae3ff6
BA
1265 if (moves.length == 0)
1266 return [];
41534b92 1267 const color = this.getColor(moves[0].start.x, moves[0].start.y);
cc2c7183 1268 const oppCol = C.GetOppCol(color);
41534b92 1269
57b8015b
BA
1270 if (this.options["capture"] && this.atLeastOneCapture())
1271 moves = this.capturePostProcess(moves, oppCol);
41534b92 1272
57b8015b
BA
1273 if (this.options["atomic"])
1274 this.atomicPostProcess(moves, oppCol);
cc2c7183 1275
c9ab0340
BA
1276 if (
1277 moves.length > 0 &&
1278 this.getPieceType(moves[0].start.x, moves[0].start.y) == "p"
1279 ) {
57b8015b 1280 this.pawnPostProcess(moves, color, oppCol);
c9ab0340
BA
1281 }
1282
cc2c7183
BA
1283 if (
1284 this.options["cannibal"] &&
57b8015b 1285 this.options["rifle"]
cc2c7183
BA
1286 ) {
1287 // In this case a rifle-capture from last rank may promote a pawn
57b8015b
BA
1288 this.riflePromotePostProcess(moves);
1289 }
1290
1291 return moves;
1292 }
1293
1294 capturePostProcess(moves, oppCol) {
1295 // Filter out non-capturing moves (not using m.vanish because of
1296 // self captures of Recycle and Teleport).
1297 return moves.filter(m => {
1298 return (
1299 this.board[m.end.x][m.end.y] != "" &&
1300 this.getColor(m.end.x, m.end.y) == oppCol
1301 );
1302 });
1303 }
1304
1305 atomicPostProcess(moves, oppCol) {
1306 moves.forEach(m => {
1307 if (
1308 this.board[m.end.x][m.end.y] != "" &&
1309 this.getColor(m.end.x, m.end.y) == oppCol
1310 ) {
1311 // Explosion!
1312 let steps = [
1313 [-1, -1],
1314 [-1, 0],
1315 [-1, 1],
1316 [0, -1],
1317 [0, 1],
1318 [1, -1],
1319 [1, 0],
1320 [1, 1]
1321 ];
1322 for (let step of steps) {
1323 let x = m.end.x + step[0];
1324 let y = this.computeY(m.end.y + step[1]);
1325 if (
1326 this.onBoard(x, y) &&
1327 this.board[x][y] != "" &&
1328 this.getPieceType(x, y) != "p"
1329 ) {
1330 m.vanish.push(
1331 new PiPo({
1332 p: this.getPiece(x, y),
1333 c: this.getColor(x, y),
1334 x: x,
1335 y: y
1336 })
1337 );
1338 }
1339 }
1340 if (!this.options["rifle"])
1341 m.appear.pop(); //nothin appears
1342 }
1343 });
1344 }
1345
1346 pawnPostProcess(moves, color, oppCol) {
1347 let moreMoves = [];
1348 const lastRank = (color == "w" ? 0 : this.size.x - 1);
1349 const initPiece = this.getPiece(moves[0].start.x, moves[0].start.y);
1350 moves.forEach(m => {
1351 let finalPieces = ["p"];
1352 const [x1, y1] = [m.start.x, m.start.y];
1353 const [x2, y2] = [m.end.x, m.end.y];
1354 const promotionOk = (
1355 x2 == lastRank &&
1356 (!this.options["rifle"] || this.board[x2][y2] == "")
1357 );
1358 if (!promotionOk)
1359 return; //nothing to do
1360 if (!this.options["pawnfall"]) {
cc2c7183 1361 if (
57b8015b
BA
1362 this.options["cannibal"] &&
1363 this.board[x2][y2] != "" &&
1364 this.getColor(x2, y2) == oppCol
cc2c7183 1365 ) {
57b8015b 1366 finalPieces = [this.getPieceType(x2, y2)];
cc2c7183 1367 }
57b8015b
BA
1368 else
1369 finalPieces = this.pawnPromotions;
1370 }
1371 m.appear[0].p = finalPieces[0];
1372 if (initPiece == "!") //cannibal king-pawn
1373 m.appear[0].p = C.CannibalKingCode[finalPieces[0]];
1374 for (let i=1; i<finalPieces.length; i++) {
1375 const piece = finalPieces[i];
1376 let tr = null;
1377 if (!this.options["pawnfall"]) {
1378 tr = {
1379 c: color,
1380 p: (initPiece != "!" ? piece : C.CannibalKingCode[piece])
1381 };
1382 }
1383 let newMove = this.getBasicMove([x1, y1], [x2, y2], tr);
1384 if (this.options["pawnfall"]) {
1385 newMove.appear.shift();
1386 newMove.pawnfall = true; //required in prePlay()
1387 }
1388 moreMoves.push(newMove);
1389 }
1390 });
1391 Array.prototype.push.apply(moves, moreMoves);
1392 }
cc2c7183 1393
57b8015b
BA
1394 riflePromotePostProcess(moves) {
1395 const lastRank = (color == "w" ? 0 : this.size.x - 1);
1396 let newMoves = [];
1397 moves.forEach(m => {
1398 if (
1399 m.start.x == lastRank &&
1400 m.appear.length >= 1 &&
1401 m.appear[0].p == "p" &&
1402 m.appear[0].x == m.start.x &&
1403 m.appear[0].y == m.start.y
1404 ) {
1405 const promotionPiece0 = this.pawnSpecs.promotions[0];
1406 m.appear[0].p = this.pawnPromotions[0];
1407 for (let i=1; i<this.pawnPromotions.length; i++) {
1408 let newMv = JSON.parse(JSON.stringify(m));
1409 newMv.appear[0].p = this.pawnSpecs.promotions[i];
1410 newMoves.push(newMv);
1411 }
1412 }
1413 });
1414 Array.prototype.push.apply(moves, newMoves);
41534b92
BA
1415 }
1416
b99ce1fb 1417 // NOTE: using special symbols to not interfere with variants' pieces codes
cc2c7183
BA
1418 static get CannibalKings() {
1419 return {
b99ce1fb
BA
1420 "!": "p",
1421 "#": "r",
1422 "$": "n",
1423 "%": "b",
1424 "*": "q"
cc2c7183
BA
1425 };
1426 }
1427
1428 static get CannibalKingCode() {
1429 return {
b99ce1fb
BA
1430 "p": "!",
1431 "r": "#",
1432 "n": "$",
1433 "b": "%",
1434 "q": "*",
cc2c7183
BA
1435 "k": "k"
1436 };
1437 }
1438
1439 isKing(symbol) {
1440 return (
1441 symbol == 'k' ||
1442 (this.options["cannibal"] && C.CannibalKings[symbol])
1443 );
1444 }
1445
41534b92
BA
1446 // For Madrasi:
1447 // (redefined in Baroque etc, where Madrasi condition doesn't make sense)
1448 isImmobilized([x, y]) {
57b8015b
BA
1449 if (!this.options["madrasi"])
1450 return false;
41534b92 1451 const color = this.getColor(x, y);
cc2c7183 1452 const oppCol = C.GetOppCol(color);
c9ab0340 1453 const piece = this.getPieceType(x, y); //ok not cannibal king
57b8015b 1454 const stepSpec = this.pieces(color, x, y)[piece];
c9ab0340
BA
1455 const attacks = stepSpec.attack || stepSpec.moves;
1456 for (let a of attacks) {
1457 outerLoop: for (let step of a.steps) {
1458 let [i, j] = [x + step[0], y + step[1]];
1459 let stepCounter = 1;
1460 while (this.onBoard(i, j) && this.board[i][j] == "") {
1461 if (a.range <= stepCounter++)
1462 continue outerLoop;
1463 i += step[0];
1464 j = this.computeY(j + step[1]);
1465 }
1466 if (
1467 this.onBoard(i, j) &&
1468 this.getColor(i, j) == oppCol &&
1469 this.getPieceType(i, j) == piece
1470 ) {
1471 return true;
1472 }
41534b92
BA
1473 }
1474 }
1475 return false;
1476 }
1477
1478 // Generic method to find possible moves of "sliding or jumping" pieces
1479 getPotentialMovesOf(piece, [x, y]) {
1480 const color = this.getColor(x, y);
c9ab0340 1481 const stepSpec = this.pieces(color, x, y)[piece];
41534b92
BA
1482 let moves = [];
1483 let explored = {}; //for Cylinder mode
c9ab0340
BA
1484
1485 const findAddMoves = (type, stepArray) => {
1486 for (let s of stepArray) {
1487 outerLoop: for (let step of s.steps) {
1488 let [i, j] = [x + step[0], this.computeY(y + step[1])];
1489 let stepCounter = 1;
1490 while (this.onBoard(i, j) && this.board[i][j] == "") {
1491 if (type != "attack" && !explored[i + "." + j]) {
1492 explored[i + "." + j] = true;
1493 moves.push(this.getBasicMove([x, y], [i, j]));
1494 }
1495 if (s.range <= stepCounter++)
1496 continue outerLoop;
1497 i += step[0];
1498 j = this.computeY(j + step[1]);
1499 }
1500 if (!this.onBoard(i, j))
1501 continue;
1502 const pieceIJ = this.getPieceType(i, j);
1503 if (
1504 type != "moveonly" &&
1505 !explored[i + "." + j] &&
1506 (
1507 !this.options["zen"] ||
1508 pieceIJ == "k"
1509 ) &&
1510 (
1511 this.canTake([x, y], [i, j]) ||
1512 (
1513 (this.options["recycle"] || this.options["teleport"]) &&
1514 pieceIJ != "k"
1515 )
1516 )
1517 ) {
1518 explored[i + "." + j] = true;
1519 moves.push(this.getBasicMove([x, y], [i, j]));
1520 }
1521 }
41534b92 1522 }
c9ab0340
BA
1523 };
1524
1525 const specialAttack = !!stepSpec.attack;
1526 if (specialAttack)
1527 findAddMoves("attack", stepSpec.attack);
1528 findAddMoves(specialAttack ? "moveonly" : "all", stepSpec.moves);
41534b92 1529 if (this.options["zen"])
c9ab0340 1530 Array.prototype.push.apply(moves, this.findCapturesOn([x, y], true));
41534b92
BA
1531 return moves;
1532 }
1533
c9ab0340 1534 findCapturesOn([x, y], zen) {
41534b92
BA
1535 let moves = [];
1536 // Find reverse captures (opponent takes)
1537 const color = this.getColor(x, y);
cc2c7183
BA
1538 const pieceType = this.getPieceType(x, y);
1539 const oppCol = C.GetOppCol(color);
c9ab0340
BA
1540 for (let i=0; i<this.size.x; i++) {
1541 for (let j=0; j<this.size.y; j++) {
57b8015b
BA
1542 if (
1543 this.board[i][j] != "" &&
1544 this.canTake([i, j], [x, y]) &&
1545 !this.isImmobilized([i, j])
1546 ) {
c9ab0340
BA
1547 const piece = this.getPieceType(i, j);
1548 if (zen && C.CannibalKingCode[piece])
1549 continue; //king not captured in this way
1550 const stepSpec = this.pieces(oppCol, i, j)[piece];
1551 const attacks = stepSpec.attack || stepSpec.moves;
1552 for (let a of attacks) {
1553 for (let s of a.steps) {
1554 // Quick check: if step isn't compatible, don't even try
57b8015b 1555 if (!C.CompatibleStep([i, j], [x, y], s, a.range))
c9ab0340
BA
1556 continue;
1557 // Finally verify that nothing stand in-between
1558 let [ii, jj] = [i + s[0], this.computeY(j + s[1])];
1559 let stepCounter = 1;
1560 while (this.onBoard(ii, jj) && this.board[ii][jj] == "") {
1561 ii += s[0];
1562 jj = this.computeY(jj + s[1]);
1563 }
1564 if (ii == x && jj == y) {
1565 moves.push(this.getBasicMove([x, y], [i, j]));
1566 if (!zen)
1567 return moves; //test for underCheck
1568 }
1569 }
1570 }
41534b92 1571 }
c9ab0340
BA
1572 }
1573 }
41534b92
BA
1574 return moves;
1575 }
1576
57b8015b
BA
1577 static CompatibleStep([x1, y1], [x2, y2], step, range) {
1578 const rx = (x2 - x1) / step[0],
1579 ry = (y2 - y1) / step[1];
1580 if (
1581 (!Number.isFinite(rx) && !Number.isNaN(rx)) ||
1582 (!Number.isFinite(ry) && !Number.isNaN(ry))
1583 ) {
1584 return false;
1585 }
1586 let distance = (Number.isNaN(rx) ? ry : rx);
1587 // TODO: 1e-7 here is totally arbitrary
1588 if (Math.abs(distance - Math.round(distance)) > 1e-7)
1589 return false;
1590 distance = Math.round(distance); //in case of (numerical...)
1591 if (range < distance)
1592 return false;
1593 return true;
1594 }
1595
41534b92
BA
1596 // Build a regular move from its initial and destination squares.
1597 // tr: transformation
1598 getBasicMove([sx, sy], [ex, ey], tr) {
1599 const initColor = this.getColor(sx, sy);
cc2c7183 1600 const initPiece = this.getPiece(sx, sy);
41534b92
BA
1601 const destColor = (this.board[ex][ey] != "" ? this.getColor(ex, ey) : "");
1602 let mv = new Move({
1603 appear: [],
1604 vanish: [],
1605 start: {x:sx, y:sy},
1606 end: {x:ex, y:ey}
1607 });
1608 if (
1609 !this.options["rifle"] ||
1610 this.board[ex][ey] == "" ||
1611 destColor == initColor //Recycle, Teleport
1612 ) {
1613 mv.appear = [
1614 new PiPo({
1615 x: ex,
1616 y: ey,
1617 c: !!tr ? tr.c : initColor,
1618 p: !!tr ? tr.p : initPiece
1619 })
1620 ];
1621 mv.vanish = [
1622 new PiPo({
1623 x: sx,
1624 y: sy,
1625 c: initColor,
1626 p: initPiece
1627 })
1628 ];
1629 }
1630 if (this.board[ex][ey] != "") {
1631 mv.vanish.push(
1632 new PiPo({
1633 x: ex,
1634 y: ey,
1635 c: this.getColor(ex, ey),
cc2c7183 1636 p: this.getPiece(ex, ey)
41534b92
BA
1637 })
1638 );
1639 if (this.options["rifle"])
1640 // Rifle captures are tricky in combination with Atomic etc,
1641 // so it's useful to mark the move :
1642 mv.capture = true;
1643 if (this.options["cannibal"] && destColor != initColor) {
1644 const lastIdx = mv.vanish.length - 1;
cc2c7183
BA
1645 let trPiece = mv.vanish[lastIdx].p;
1646 if (this.isKing(this.getPiece(sx, sy)))
1647 trPiece = C.CannibalKingCode[trPiece];
b4ae3ff6
BA
1648 if (mv.appear.length >= 1)
1649 mv.appear[0].p = trPiece;
41534b92
BA
1650 else if (this.options["rifle"]) {
1651 mv.appear.unshift(
1652 new PiPo({
1653 x: sx,
1654 y: sy,
1655 c: initColor,
cc2c7183 1656 p: trPiece
41534b92
BA
1657 })
1658 );
1659 mv.vanish.unshift(
1660 new PiPo({
1661 x: sx,
1662 y: sy,
1663 c: initColor,
1664 p: initPiece
1665 })
1666 );
1667 }
1668 }
1669 }
1670 return mv;
1671 }
1672
1673 // En-passant square, if any
1674 getEpSquare(moveOrSquare) {
1675 if (typeof moveOrSquare === "string") {
1676 const square = moveOrSquare;
b4ae3ff6
BA
1677 if (square == "-")
1678 return undefined;
cc2c7183 1679 return C.SquareToCoords(square);
41534b92
BA
1680 }
1681 // Argument is a move:
1682 const move = moveOrSquare;
1683 const s = move.start,
1684 e = move.end;
1685 if (
1686 s.y == e.y &&
1687 Math.abs(s.x - e.x) == 2 &&
1688 // Next conditions for variants like Atomic or Rifle, Recycle...
cc2c7183
BA
1689 (move.appear.length > 0 && move.appear[0].p == "p") &&
1690 (move.vanish.length > 0 && move.vanish[0].p == "p")
41534b92
BA
1691 ) {
1692 return {
1693 x: (s.x + e.x) / 2,
1694 y: s.y
1695 };
1696 }
1697 return undefined; //default
1698 }
1699
1700 // Special case of en-passant captures: treated separately
c9ab0340 1701 getEnpassantCaptures([x, y]) {
41534b92 1702 const color = this.getColor(x, y);
c9ab0340 1703 const shiftX = (color == 'w' ? -1 : 1);
cc2c7183 1704 const oppCol = C.GetOppCol(color);
41534b92
BA
1705 let enpassantMove = null;
1706 if (
1707 !!this.epSquare &&
1708 this.epSquare.x == x + shiftX &&
1709 Math.abs(this.computeY(this.epSquare.y - y)) == 1 &&
1710 this.getColor(x, this.epSquare.y) == oppCol //Doublemove guard...
1711 ) {
1712 const [epx, epy] = [this.epSquare.x, this.epSquare.y];
1713 this.board[epx][epy] = oppCol + "p";
1714 enpassantMove = this.getBasicMove([x, y], [epx, epy]);
1715 this.board[epx][epy] = "";
1716 const lastIdx = enpassantMove.vanish.length - 1; //think Rifle
1717 enpassantMove.vanish[lastIdx].x = x;
1718 }
1719 return !!enpassantMove ? [enpassantMove] : [];
1720 }
1721
41534b92
BA
1722 // "castleInCheck" arg to let some variants castle under check
1723 getCastleMoves([x, y], finalSquares, castleInCheck, castleWith) {
1724 const c = this.getColor(x, y);
1725
1726 // Castling ?
cc2c7183 1727 const oppCol = C.GetOppCol(c);
41534b92
BA
1728 let moves = [];
1729 // King, then rook:
1730 finalSquares =
1731 finalSquares || [ [2, 3], [this.size.y - 2, this.size.y - 3] ];
cc2c7183 1732 const castlingKing = this.getPiece(x, y);
41534b92
BA
1733 castlingCheck: for (
1734 let castleSide = 0;
1735 castleSide < 2;
1736 castleSide++ //large, then small
1737 ) {
b4ae3ff6
BA
1738 if (this.castleFlags[c][castleSide] >= this.size.y)
1739 continue;
41534b92
BA
1740 // If this code is reached, rook and king are on initial position
1741
1742 // NOTE: in some variants this is not a rook
1743 const rookPos = this.castleFlags[c][castleSide];
cc2c7183 1744 const castlingPiece = this.getPiece(x, rookPos);
41534b92
BA
1745 if (
1746 this.board[x][rookPos] == "" ||
1747 this.getColor(x, rookPos) != c ||
1748 (!!castleWith && !castleWith.includes(castlingPiece))
1749 ) {
1750 // Rook is not here, or changed color (see Benedict)
1751 continue;
1752 }
1753 // Nothing on the path of the king ? (and no checks)
1754 const finDist = finalSquares[castleSide][0] - y;
1755 let step = finDist / Math.max(1, Math.abs(finDist));
1756 let i = y;
1757 do {
1758 if (
1759 (!castleInCheck && this.underCheck([x, i], oppCol)) ||
1760 (
1761 this.board[x][i] != "" &&
1762 // NOTE: next check is enough, because of chessboard constraints
1763 (this.getColor(x, i) != c || ![rookPos, y].includes(i))
1764 )
1765 ) {
1766 continue castlingCheck;
1767 }
1768 i += step;
1769 } while (i != finalSquares[castleSide][0]);
1770 // Nothing on the path to the rook?
1771 step = (castleSide == 0 ? -1 : 1);
1772 for (i = y + step; i != rookPos; i += step) {
b4ae3ff6
BA
1773 if (this.board[x][i] != "")
1774 continue castlingCheck;
41534b92
BA
1775 }
1776
1777 // Nothing on final squares, except maybe king and castling rook?
1778 for (i = 0; i < 2; i++) {
1779 if (
1780 finalSquares[castleSide][i] != rookPos &&
1781 this.board[x][finalSquares[castleSide][i]] != "" &&
1782 (
1783 finalSquares[castleSide][i] != y ||
1784 this.getColor(x, finalSquares[castleSide][i]) != c
1785 )
1786 ) {
1787 continue castlingCheck;
1788 }
1789 }
1790
1791 // If this code is reached, castle is valid
1792 moves.push(
1793 new Move({
1794 appear: [
1795 new PiPo({
1796 x: x,
1797 y: finalSquares[castleSide][0],
1798 p: castlingKing,
1799 c: c
1800 }),
1801 new PiPo({
1802 x: x,
1803 y: finalSquares[castleSide][1],
1804 p: castlingPiece,
1805 c: c
1806 })
1807 ],
1808 vanish: [
1809 // King might be initially disguised (Titan...)
1810 new PiPo({ x: x, y: y, p: castlingKing, c: c }),
1811 new PiPo({ x: x, y: rookPos, p: castlingPiece, c: c })
1812 ],
1813 end:
1814 Math.abs(y - rookPos) <= 2
c9ab0340
BA
1815 ? {x: x, y: rookPos}
1816 : {x: x, y: y + 2 * (castleSide == 0 ? -1 : 1)}
41534b92
BA
1817 })
1818 );
1819 }
1820
1821 return moves;
1822 }
1823
1824 ////////////////////
1825 // MOVES VALIDATION
1826
1827 // Is (king at) given position under check by "color" ?
1828 underCheck([x, y], color) {
b4ae3ff6
BA
1829 if (this.options["taking"] || this.options["dark"])
1830 return false;
c9ab0340 1831 return (this.findCapturesOn([x, y]).length >= 1);
41534b92
BA
1832 }
1833
1834 // Stop at first king found (TODO: multi-kings)
1835 searchKingPos(color) {
1836 for (let i=0; i < this.size.x; i++) {
1837 for (let j=0; j < this.size.y; j++) {
cc2c7183
BA
1838 if (this.getColor(i, j) == color && this.isKing(this.getPiece(i, j)))
1839 return [i, j];
41534b92
BA
1840 }
1841 }
1842 return [-1, -1]; //king not found
1843 }
1844
1845 filterValid(moves) {
b4ae3ff6
BA
1846 if (moves.length == 0)
1847 return [];
41534b92 1848 const color = this.turn;
cc2c7183 1849 const oppCol = C.GetOppCol(color);
41534b92
BA
1850 if (this.options["balance"] && [1, 3].includes(this.movesCount)) {
1851 // Forbid moves either giving check or exploding opponent's king:
1852 const oppKingPos = this.searchKingPos(oppCol);
1853 moves = moves.filter(m => {
1854 if (
cc2c7183
BA
1855 m.vanish.some(v => v.c == oppCol && v.p == "k") &&
1856 m.appear.every(a => a.c != oppCol || a.p != "k")
41534b92
BA
1857 )
1858 return false;
1859 this.playOnBoard(m);
1860 const res = !this.underCheck(oppKingPos, color);
1861 this.undoOnBoard(m);
1862 return res;
1863 });
1864 }
b4ae3ff6
BA
1865 if (this.options["taking"] || this.options["dark"])
1866 return moves;
41534b92
BA
1867 const kingPos = this.searchKingPos(color);
1868 let filtered = {}; //avoid re-checking similar moves (promotions...)
1869 return moves.filter(m => {
1870 const key = m.start.x + m.start.y + '.' + m.end.x + m.end.y;
1871 if (!filtered[key]) {
1872 this.playOnBoard(m);
1873 let square = kingPos,
1874 res = true; //a priori valid
cc2c7183
BA
1875 if (m.vanish.some(v => {
1876 return (v.p == "k" || C.CannibalKings[v.p]) && v.c == color;
1877 })) {
41534b92
BA
1878 // Search king in appear array:
1879 const newKingIdx =
cc2c7183
BA
1880 m.appear.findIndex(a => {
1881 return (a.p == "k" || C.CannibalKings[a.p]) && a.c == color;
1882 });
41534b92
BA
1883 if (newKingIdx >= 0)
1884 square = [m.appear[newKingIdx].x, m.appear[newKingIdx].y];
b4ae3ff6
BA
1885 else
1886 res = false;
41534b92
BA
1887 }
1888 res &&= !this.underCheck(square, oppCol);
1889 this.undoOnBoard(m);
1890 filtered[key] = res;
1891 return res;
1892 }
1893 return filtered[key];
1894 });
1895 }
1896
1897 /////////////////
1898 // MOVES PLAYING
1899
1900 // Aggregate flags into one object
1901 aggregateFlags() {
1902 return this.castleFlags;
1903 }
1904
1905 // Reverse operation
1906 disaggregateFlags(flags) {
1907 this.castleFlags = flags;
1908 }
1909
1910 // Apply a move on board
1911 playOnBoard(move) {
1912 for (let psq of move.vanish) this.board[psq.x][psq.y] = "";
1913 for (let psq of move.appear) this.board[psq.x][psq.y] = psq.c + psq.p;
1914 }
1915 // Un-apply the played move
1916 undoOnBoard(move) {
1917 for (let psq of move.appear) this.board[psq.x][psq.y] = "";
1918 for (let psq of move.vanish) this.board[psq.x][psq.y] = psq.c + psq.p;
1919 }
1920
1921 updateCastleFlags(move) {
1922 // Update castling flags if start or arrive from/at rook/king locations
1923 move.appear.concat(move.vanish).forEach(psq => {
1924 if (
1925 this.board[psq.x][psq.y] != "" &&
cc2c7183 1926 this.getPieceType(psq.x, psq.y) == "k"
41534b92
BA
1927 ) {
1928 this.castleFlags[psq.c] = [this.size.y, this.size.y];
1929 }
1930 // NOTE: not "else if" because king can capture enemy rook...
cc2c7183 1931 let c = "";
b4ae3ff6
BA
1932 if (psq.x == 0)
1933 c = "b";
1934 else if (psq.x == this.size.x - 1)
1935 c = "w";
cc2c7183 1936 if (c != "") {
41534b92 1937 const fidx = this.castleFlags[c].findIndex(f => f == psq.y);
b4ae3ff6
BA
1938 if (fidx >= 0)
1939 this.castleFlags[c][fidx] = this.size.y;
41534b92
BA
1940 }
1941 });
1942 }
1943
1944 prePlay(move) {
1945 if (
1946 typeof move.start.x == "number" &&
cc2c7183 1947 (!this.options["teleport"] || this.subTurnTeleport == 1)
41534b92
BA
1948 ) {
1949 // OK, not a drop move
1950 if (
1951 this.hasCastle &&
1952 // If flags already off, no need to re-check:
1953 Object.keys(this.castleFlags).some(c => {
1954 return this.castleFlags[c].some(val => val < this.size.y)})
1955 ) {
1956 this.updateCastleFlags(move);
1957 }
cc2c7183 1958 const initSquare = C.CoordsToSquare(move.start);
41534b92
BA
1959 if (
1960 this.options["crazyhouse"] &&
1961 (!this.options["rifle"] || !move.capture)
1962 ) {
f429756d 1963 const destSquare = C.CoordsToSquare(move.end);
41534b92
BA
1964 if (this.ispawn[initSquare]) {
1965 delete this.ispawn[initSquare];
f429756d 1966 this.ispawn[destSquare] = true;
41534b92
BA
1967 }
1968 else if (
cc2c7183
BA
1969 move.vanish[0].p == "p" &&
1970 move.appear[0].p != "p"
41534b92 1971 ) {
f429756d
BA
1972 this.ispawn[destSquare] = true;
1973 }
1974 else if (
1975 this.ispawn[destSquare] &&
1976 this.getColor(move.end.x, move.end.y) != move.vanish[0].c
1977 ) {
1978 move.vanish[1].p = "p";
1979 delete this.ispawn[destSquare];
41534b92
BA
1980 }
1981 }
1982 }
1983 const minSize = Math.min(move.appear.length, move.vanish.length);
f8b43ef7 1984 if (this.hasReserve && !move.pawnfall) {
41534b92
BA
1985 const color = this.turn;
1986 for (let i=minSize; i<move.appear.length; i++) {
1987 // Something appears = dropped on board (some exceptions, Chakart...)
1988 const piece = move.appear[i].p;
1989 this.updateReserve(color, piece, this.reserve[color][piece] - 1);
1990 }
1991 for (let i=minSize; i<move.vanish.length; i++) {
1992 // Something vanish: add to reserve except if recycle & opponent
1993 const piece = move.vanish[i].p;
1994 if (this.options["crazyhouse"] || move.vanish[i].c == color)
1995 this.updateReserve(color, piece, this.reserve[color][piece] + 1);
1996 }
1997 }
1998 }
1999
2000 play(move) {
2001 this.prePlay(move);
b4ae3ff6
BA
2002 if (this.hasEnpassant)
2003 this.epSquare = this.getEpSquare(move);
41534b92
BA
2004 this.playOnBoard(move);
2005 this.postPlay(move);
2006 }
2007
2008 postPlay(move) {
2009 const color = this.turn;
cc2c7183 2010 const oppCol = C.GetOppCol(color);
b4ae3ff6 2011 if (this.options["dark"])
c9ab0340 2012 this.updateEnlightened();
41534b92
BA
2013 if (this.options["teleport"]) {
2014 if (
cc2c7183 2015 this.subTurnTeleport == 1 &&
41534b92
BA
2016 move.vanish.length > move.appear.length &&
2017 move.vanish[move.vanish.length - 1].c == color
2018 ) {
2019 const v = move.vanish[move.vanish.length - 1];
2020 this.captured = {x: v.x, y: v.y, c: v.c, p: v.p};
cc2c7183 2021 this.subTurnTeleport = 2;
41534b92
BA
2022 return;
2023 }
cc2c7183 2024 this.subTurnTeleport = 1;
41534b92
BA
2025 this.captured = null;
2026 }
2027 if (this.options["balance"]) {
b4ae3ff6
BA
2028 if (![1, 3].includes(this.movesCount))
2029 this.turn = oppCol;
41534b92
BA
2030 }
2031 else {
2032 if (
2033 (
2034 this.options["doublemove"] &&
2035 this.movesCount >= 1 &&
2036 this.subTurn == 1
2037 ) ||
2038 (this.options["progressive"] && this.subTurn <= this.movesCount)
2039 ) {
2040 const oppKingPos = this.searchKingPos(oppCol);
6f74b81a
BA
2041 if (
2042 oppKingPos[0] >= 0 &&
2043 (
2044 this.options["taking"] ||
2045 !this.underCheck(oppKingPos, color)
2046 )
2047 ) {
41534b92
BA
2048 this.subTurn++;
2049 return;
2050 }
2051 }
2052 this.turn = oppCol;
2053 }
2054 this.movesCount++;
2055 this.subTurn = 1;
2056 }
2057
2058 // "Stop at the first move found"
2059 atLeastOneMove(color) {
2060 color = color || this.turn;
2061 for (let i = 0; i < this.size.x; i++) {
2062 for (let j = 0; j < this.size.y; j++) {
2063 if (this.board[i][j] != "" && this.getColor(i, j) == color) {
cc2c7183
BA
2064 // NOTE: in fact searching for all potential moves from i,j.
2065 // I don't believe this is an issue, for now at least.
41534b92 2066 const moves = this.getPotentialMovesFrom([i, j]);
b4ae3ff6
BA
2067 if (moves.some(m => this.filterValid([m]).length >= 1))
2068 return true;
41534b92
BA
2069 }
2070 }
2071 }
2072 if (this.hasReserve && this.reserve[color]) {
2073 for (let p of Object.keys(this.reserve[color])) {
2074 const moves = this.getDropMovesFrom([color, p]);
b4ae3ff6
BA
2075 if (moves.some(m => this.filterValid([m]).length >= 1))
2076 return true;
41534b92
BA
2077 }
2078 }
2079 return false;
2080 }
2081
2082 // What is the score ? (Interesting if game is over)
2083 getCurrentScore(move) {
2084 const color = this.turn;
cc2c7183 2085 const oppCol = C.GetOppCol(color);
41534b92 2086 const kingPos = [this.searchKingPos(color), this.searchKingPos(oppCol)];
b4ae3ff6
BA
2087 if (kingPos[0][0] < 0 && kingPos[1][0] < 0)
2088 return "1/2";
2089 if (kingPos[0][0] < 0)
2090 return (color == "w" ? "0-1" : "1-0");
2091 if (kingPos[1][0] < 0)
2092 return (color == "w" ? "1-0" : "0-1");
2093 if (this.atLeastOneMove())
2094 return "*";
41534b92 2095 // No valid move: stalemate or checkmate?
c9ab0340 2096 if (!this.underCheck(kingPos[0], color))
b4ae3ff6 2097 return "1/2";
41534b92
BA
2098 // OK, checkmate
2099 return (color == "w" ? "0-1" : "1-0");
2100 }
2101
cc2c7183 2102 // NOTE: quite suboptimal for eg. Benedict (not a big deal I think)
41534b92
BA
2103 playVisual(move, r) {
2104 move.vanish.forEach(v => {
c9ab0340
BA
2105 // TODO: next "if" shouldn't be required
2106 if (this.g_pieces[v.x][v.y])
2107 this.g_pieces[v.x][v.y].remove();
2108 this.g_pieces[v.x][v.y] = null;
41534b92 2109 });
3c61449b
BA
2110 let chessboard =
2111 document.getElementById(this.containerId).querySelector(".chessboard");
b4ae3ff6
BA
2112 if (!r)
2113 r = chessboard.getBoundingClientRect();
41534b92
BA
2114 const pieceWidth = this.getPieceWidth(r.width);
2115 move.appear.forEach(a => {
41534b92
BA
2116 this.g_pieces[a.x][a.y] = document.createElement("piece");
2117 this.g_pieces[a.x][a.y].classList.add(this.pieces()[a.p]["class"]);
2118 this.g_pieces[a.x][a.y].classList.add(a.c == "w" ? "white" : "black");
2119 this.g_pieces[a.x][a.y].style.width = pieceWidth + "px";
2120 this.g_pieces[a.x][a.y].style.height = pieceWidth + "px";
2121 const [ip, jp] = this.getPixelPosition(a.x, a.y, r);
2122 this.g_pieces[a.x][a.y].style.transform = `translate(${ip}px,${jp}px)`;
c9ab0340
BA
2123 if (this.enlightened && !this.enlightened[a.x][a.y])
2124 this.g_pieces[a.x][a.y].classList.add("hidden");
3c61449b 2125 chessboard.appendChild(this.g_pieces[a.x][a.y]);
41534b92 2126 });
c9ab0340
BA
2127 if (this.options["dark"])
2128 this.graphUpdateEnlightened();
41534b92
BA
2129 }
2130
2131 playPlusVisual(move, r) {
41534b92 2132 this.play(move);
c9ab0340 2133 this.playVisual(move, r);
41534b92
BA
2134 this.afterPlay(move); //user method
2135 }
2136
2137 // Assumes reserve on top (usage case otherwise? TODO?)
2138 getReserveShift(c, p, r) {
2139 let nbR = 0,
2140 ridx = 0;
2141 for (let pi of Object.keys(this.reserve[c])) {
b4ae3ff6
BA
2142 if (this.reserve[c][pi] == 0)
2143 continue;
2144 if (pi == p)
2145 ridx = nbR;
41534b92
BA
2146 nbR++;
2147 }
2148 const rsqSize = this.getReserveSquareSize(r.width, nbR);
f8b43ef7 2149 return [-ridx * rsqSize, rsqSize]; //slightly inaccurate... TODO?
41534b92
BA
2150 }
2151
2152 animate(move, callback) {
2153 if (this.noAnimate) {
2154 callback();
2155 return;
2156 }
2157 const [i1, j1] = [move.start.x, move.start.y];
2158 const dropMove = (typeof i1 == "string");
2159 const startArray = (dropMove ? this.r_pieces : this.g_pieces);
2160 let startPiece = startArray[i1][j1];
e8b85c86
BA
2161 // TODO: next "if" shouldn't be required
2162 if (!startPiece) {
2163 callback();
2164 return;
2165 }
3c61449b
BA
2166 let chessboard =
2167 document.getElementById(this.containerId).querySelector(".chessboard");
41534b92
BA
2168 const clonePiece = (
2169 !dropMove &&
2170 this.options["rifle"] ||
cc2c7183 2171 (this.options["teleport"] && this.subTurnTeleport == 2)
41534b92
BA
2172 );
2173 if (clonePiece) {
2174 startPiece = startPiece.cloneNode();
b4ae3ff6
BA
2175 if (this.options["rifle"])
2176 startArray[i1][j1].style.opacity = "0";
cc2c7183 2177 if (this.options["teleport"] && this.subTurnTeleport == 2) {
41534b92
BA
2178 const pieces = this.pieces();
2179 const startCode = (dropMove ? j1 : this.getPiece(i1, j1));
2180 startPiece.classList.remove(pieces[startCode]["class"]);
2181 startPiece.classList.add(pieces[this.captured.p]["class"]);
2182 // Color: OK
2183 }
3c61449b 2184 chessboard.appendChild(startPiece);
41534b92
BA
2185 }
2186 const [i2, j2] = [move.end.x, move.end.y];
2187 let startCoords;
2188 if (dropMove) {
2189 startCoords = [
2190 i1 == this.playerColor ? this.size.x : 0,
2191 this.size.y / 2 //not trying to be accurate here... (TODO?)
2192 ];
2193 }
b4ae3ff6
BA
2194 else
2195 startCoords = [i1, j1];
3c61449b 2196 const r = chessboard.getBoundingClientRect();
41534b92
BA
2197 const arrival = this.getPixelPosition(i2, j2, r); //TODO: arrival on drop?
2198 let rs = [0, 0];
b4ae3ff6
BA
2199 if (dropMove)
2200 rs = this.getReserveShift(i1, j1, r);
41534b92
BA
2201 const distance =
2202 Math.sqrt((startCoords[0] - i2) ** 2 + (startCoords[1] - j2) ** 2);
2203 const maxDist = Math.sqrt((this.size.x - 1)** 2 + (this.size.y - 1) ** 2);
2204 const multFact = (distance - 1) / (maxDist - 1); //1 == minDist
2205 const duration = 0.2 + multFact * 0.3;
bc34b505 2206 const initTransform = startPiece.style.transform;
41534b92
BA
2207 startPiece.style.transform =
2208 `translate(${arrival[0] + rs[0]}px, ${arrival[1] + rs[1]}px)`;
2209 startPiece.style.transitionDuration = duration + "s";
2210 setTimeout(
2211 () => {
2212 if (clonePiece) {
b4ae3ff6
BA
2213 if (this.options["rifle"])
2214 startArray[i1][j1].style.opacity = "1";
41534b92
BA
2215 startPiece.remove();
2216 }
bc34b505
BA
2217 else {
2218 startPiece.style.transform = initTransform;
2219 startPiece.style.transitionDuration = "0s";
2220 }
41534b92
BA
2221 callback();
2222 },
2223 duration * 1000
2224 );
2225 }
2226
2227 playReceivedMove(moves, callback) {
21e8e712 2228 const launchAnimation = () => {
3c61449b 2229 const r = container.querySelector(".chessboard").getBoundingClientRect();
21e8e712
BA
2230 const animateRec = i => {
2231 this.animate(moves[i], () => {
21e8e712 2232 this.play(moves[i]);
57b8015b 2233 this.playVisual(moves[i], r);
b4ae3ff6
BA
2234 if (i < moves.length - 1)
2235 setTimeout(() => animateRec(i+1), 300);
2236 else
2237 callback();
21e8e712
BA
2238 });
2239 };
2240 animateRec(0);
2241 };
e081c5eb
BA
2242 // Delay if user wasn't focused:
2243 const checkDisplayThenAnimate = (delay) => {
3c61449b 2244 if (container.style.display == "none") {
21e8e712
BA
2245 alert("New move! Let's go back to game...");
2246 document.getElementById("gameInfos").style.display = "none";
3c61449b 2247 container.style.display = "block";
21e8e712
BA
2248 setTimeout(launchAnimation, 700);
2249 }
b4ae3ff6
BA
2250 else
2251 setTimeout(launchAnimation, delay || 0);
21e8e712 2252 };
3c61449b 2253 let container = document.getElementById(this.containerId);
016306e3
BA
2254 if (document.hidden) {
2255 document.onvisibilitychange = () => {
2256 document.onvisibilitychange = undefined;
e081c5eb 2257 checkDisplayThenAnimate(700);
fd31883b 2258 };
fd31883b 2259 }
b4ae3ff6
BA
2260 else
2261 checkDisplayThenAnimate();
41534b92
BA
2262 }
2263
2264};