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