Fix Checkless and trackKingWrap()
[xogo.git] / base_rules.js
CommitLineData
9b760538
BA
1import {Random} from "/utils/alea.js";
2import {ArrayFun} from "/utils/array.js";
0adfbdb5 3import {FenUtil} from "/utils/setupPieces.js";
41534b92
BA
4import PiPo from "/utils/PiPo.js";
5import Move from "/utils/Move.js";
6
5f08c59b
BA
7// Helper class for move animation
8class TargetObj {
9
10 constructor(callOnComplete) {
11 this.value = 0;
12 this.target = 0;
13 this.callOnComplete = callOnComplete;
14 }
15 increment() {
16 if (++this.value == this.target)
17 this.callOnComplete();
18 }
19
20};
21
41534b92 22// NOTE: x coords: top to bottom (white perspective); y: left to right
cc2c7183 23// NOTE: ChessRules is aliased as window.C, and variants as window.V
41534b92
BA
24export default class ChessRules {
25
e5f93427 26 static get Aliases() {
3caec36f 27 return {'C': ChessRules};
e5f93427
BA
28 }
29
41534b92
BA
30 /////////////////////////
31 // VARIANT SPECIFICATIONS
32
33 // Some variants have specific options, like the number of pawns in Monster,
34 // or the board size for Pandemonium.
35 // Users can generally select a randomness level from 0 to 2.
36 static get Options() {
37 return {
41534b92
BA
38 select: [{
39 label: "Randomness",
40 variable: "randomness",
41 defaut: 0,
42 options: [
b4ae3ff6
BA
43 {label: "Deterministic", value: 0},
44 {label: "Symmetric random", value: 1},
45 {label: "Asymmetric random", value: 2}
41534b92
BA
46 ]
47 }],
437dfd42 48 input: [
f8b43ef7
BA
49 {
50 label: "Capture king",
437dfd42
BA
51 variable: "taking",
52 type: "checkbox",
53 defaut: false
f8b43ef7
BA
54 },
55 {
56 label: "Falling pawn",
437dfd42
BA
57 variable: "pawnfall",
58 type: "checkbox",
59 defaut: false
f8b43ef7
BA
60 }
61 ],
41534b92
BA
62 // Game modifiers (using "elementary variants"). Default: false
63 styles: [
64 "atomic",
65 "balance", //takes precedence over doublemove & progressive
66 "cannibal",
67 "capture",
68 "crazyhouse",
69 "cylinder", //ok with all
70 "dark",
71 "doublemove",
72 "madrasi",
73 "progressive", //(natural) priority over doublemove
74 "recycle",
75 "rifle",
76 "teleport",
77 "zen"
78 ]
79 };
80 }
81
c9ab0340
BA
82 get pawnPromotions() {
83 return ['q', 'r', 'n', 'b'];
41534b92
BA
84 }
85
86 // Some variants don't have flags:
87 get hasFlags() {
88 return true;
89 }
90 // Or castle
91 get hasCastle() {
92 return this.hasFlags;
93 }
94
95 // En-passant captures allowed?
96 get hasEnpassant() {
97 return true;
98 }
99
100 get hasReserve() {
101 return (
102 !!this.options["crazyhouse"] ||
103 (!!this.options["recycle"] && !this.options["teleport"])
104 );
105 }
24872b22
BA
106 // Some variants do not store reserve state (Align4, Chakart...)
107 get hasReserveFen() {
108 return this.hasReserve;
109 }
41534b92
BA
110
111 get noAnimate() {
112 return !!this.options["dark"];
113 }
114
6b9320bb
BA
115 // Some variants use only click information
116 get clickOnly() {
117 return false;
118 }
119
98d14451 120 // Some variants reveal moves only after both players played
126ffc70 121 get hideMoves() {
98d14451
BA
122 return false;
123 }
124
41534b92 125 // Some variants use click infos:
15106e82
BA
126 doClick(coords) {
127 if (typeof coords.x != "number")
b4ae3ff6 128 return null; //click on reserves
41534b92 129 if (
cc2c7183 130 this.options["teleport"] && this.subTurnTeleport == 2 &&
15106e82 131 this.board[coords.x][coords.y] == ""
41534b92 132 ) {
1a7c0492 133 let res = new Move({
41534b92
BA
134 start: {x: this.captured.x, y: this.captured.y},
135 appear: [
136 new PiPo({
15106e82
BA
137 x: coords.x,
138 y: coords.y,
ca8a3993 139 c: this.captured.c,
41534b92
BA
140 p: this.captured.p
141 })
142 ],
1a7c0492 143 vanish: []
41534b92 144 });
1a7c0492
BA
145 res.drag = {c: this.captured.c, p: this.captured.p};
146 return res;
41534b92
BA
147 }
148 return null;
149 }
150
151 ////////////////////
152 // COORDINATES UTILS
153
4bff03f5 154 // 3a --> {x:3, y:10}
41534b92 155 static SquareToCoords(sq) {
15106e82
BA
156 return ArrayFun.toObject(["x", "y"],
157 [0, 1].map(i => parseInt(sq[i], 36)));
41534b92
BA
158 }
159
4bff03f5 160 // {x:11, y:12} --> bc
15106e82
BA
161 static CoordsToSquare(cd) {
162 return Object.values(cd).map(c => c.toString(36)).join("");
41534b92
BA
163 }
164
15106e82
BA
165 coordsToId(cd) {
166 if (typeof cd.x == "number") {
167 return (
168 `${this.containerId}|sq-${cd.x.toString(36)}-${cd.y.toString(36)}`
169 );
170 }
41534b92 171 // Reserve :
15106e82 172 return `${this.containerId}|rsq-${cd.x}-${cd.y}`;
41534b92
BA
173 }
174
175 idToCoords(targetId) {
b4ae3ff6
BA
176 if (!targetId)
177 return null; //outside page, maybe...
41534b92
BA
178 const idParts = targetId.split('|'); //prefix|sq-2-3 (start at 0 => 3,4)
179 if (
180 idParts.length < 2 ||
181 idParts[0] != this.containerId ||
182 !idParts[1].match(/sq-[0-9a-zA-Z]-[0-9a-zA-Z]/)
183 ) {
184 return null;
185 }
186 const squares = idParts[1].split('-');
187 if (squares[0] == "sq")
15106e82
BA
188 return {x: parseInt(squares[1], 36), y: parseInt(squares[2], 36)};
189 // squares[0] == "rsq" : reserve, 'c' + 'p' (letters color & piece)
190 return {x: squares[1], y: squares[2]};
41534b92
BA
191 }
192
193 /////////////
194 // FEN UTILS
195
196 // Turn "wb" into "B" (for FEN)
197 board2fen(b) {
4bff03f5 198 return (b[0] == "w" ? b[1].toUpperCase() : b[1]);
41534b92
BA
199 }
200
201 // Turn "p" into "bp" (for board)
202 fen2board(f) {
4bff03f5 203 return (f.charCodeAt(0) <= 90 ? "w" + f.toLowerCase() : "b" + f);
41534b92
BA
204 }
205
41534b92 206 genRandInitFen(seed) {
f31de5e4
BA
207 Random.setSeed(seed); //may be unused
208 let baseFen = this.genRandInitBaseFen();
209 baseFen.o = Object.assign({init: true}, baseFen.o);
210 const parts = this.getPartFen(baseFen.o);
211 return (
dc10e429 212 baseFen.fen + " w 0" +
f31de5e4
BA
213 (Object.keys(parts).length > 0 ? (" " + JSON.stringify(parts)) : "")
214 );
215 }
216
217 // Setup the initial random-or-not (asymmetric-or-not) position
218 genRandInitBaseFen() {
0adfbdb5
BA
219 const s = FenUtil.setupPieces(
220 ['r', 'n', 'b', 'q', 'k', 'b', 'n', 'r'],
221 {
10c9010b 222 randomness: this.options["randomness"],
0adfbdb5 223 between: {p1: 'k', p2: 'r'},
7c038235
BA
224 diffCol: ['b'],
225 flags: ['r']
41534b92 226 }
0adfbdb5
BA
227 );
228 return {
7c038235
BA
229 fen: s.b.join("") + "/pppppppp/8/8/8/8/PPPPPPPP/" +
230 s.w.join("").toUpperCase(),
0adfbdb5
BA
231 o: {flags: s.flags}
232 };
41534b92
BA
233 }
234
235 // "Parse" FEN: just return untransformed string data
236 parseFen(fen) {
237 const fenParts = fen.split(" ");
238 let res = {
239 position: fenParts[0],
240 turn: fenParts[1],
241 movesCount: fenParts[2]
242 };
b4ae3ff6
BA
243 if (fenParts.length > 3)
244 res = Object.assign(res, JSON.parse(fenParts[3]));
41534b92
BA
245 return res;
246 }
247
248 // Return current fen (game state)
249 getFen() {
f31de5e4
BA
250 const parts = this.getPartFen({});
251 return (
252 this.getBaseFen() +
253 (Object.keys(parts).length > 0 ? (" " + JSON.stringify(parts)) : "")
41534b92 254 );
f31de5e4
BA
255 }
256
257 getBaseFen() {
258 return this.getPosition() + " " + this.turn + " " + this.movesCount;
259 }
260
261 getPartFen(o) {
262 let parts = {};
b4ae3ff6 263 if (this.hasFlags)
f31de5e4 264 parts["flags"] = o.init ? o.flags : this.getFlagsFen();
41534b92 265 if (this.hasEnpassant)
f31de5e4 266 parts["enpassant"] = o.init ? "-" : this.getEnpassantFen();
24872b22 267 if (this.hasReserveFen)
f31de5e4 268 parts["reserve"] = this.getReserveFen(o);
41534b92 269 if (this.options["crazyhouse"])
f31de5e4
BA
270 parts["ispawn"] = this.getIspawnFen(o);
271 return parts;
41534b92
BA
272 }
273
d621e620
BA
274 static FenEmptySquares(count) {
275 // if more than 9 consecutive free spaces, break the integer,
276 // otherwise FEN parsing will fail.
277 if (count <= 9)
278 return count;
279 // Most boards of size < 18:
280 if (count <= 18)
281 return "9" + (count - 9);
282 // Except Gomoku:
283 return "99" + (count - 18);
284 }
285
41534b92 286 // Position part of the FEN string
15106e82 287 getPosition() {
41534b92 288 let position = "";
55d3b31f 289 for (let i = 0; i < this.size.x; i++) {
41534b92 290 let emptyCount = 0;
55d3b31f 291 for (let j = 0; j < this.size.y; j++) {
b4ae3ff6
BA
292 if (this.board[i][j] == "")
293 emptyCount++;
41534b92
BA
294 else {
295 if (emptyCount > 0) {
296 // Add empty squares in-between
d621e620 297 position += C.FenEmptySquares(emptyCount);
41534b92
BA
298 emptyCount = 0;
299 }
300 position += this.board2fen(this.board[i][j]);
301 }
302 }
303 if (emptyCount > 0)
304 // "Flush remainder"
d621e620 305 position += C.FenEmptySquares(emptyCount);
55d3b31f 306 if (i < this.size.x - 1)
b4ae3ff6 307 position += "/"; //separate rows
41534b92
BA
308 }
309 return position;
310 }
311
41534b92
BA
312 // Flags part of the FEN string
313 getFlagsFen() {
314 return ["w", "b"].map(c => {
15106e82 315 return this.castleFlags[c].map(x => x.toString(36)).join("");
41534b92
BA
316 }).join("");
317 }
318
319 // Enpassant part of the FEN string
320 getEnpassantFen() {
b4ae3ff6 321 if (!this.epSquare)
f31de5e4 322 return "-";
cc2c7183 323 return C.CoordsToSquare(this.epSquare);
41534b92
BA
324 }
325
f31de5e4
BA
326 getReserveFen(o) {
327 if (o.init)
328 return "000000000000";
41534b92
BA
329 return (
330 ["w","b"].map(c => Object.values(this.reserve[c]).join("")).join("")
331 );
332 }
333
f31de5e4
BA
334 getIspawnFen(o) {
335 if (o.init)
336 // NOTE: cannot merge because this.ispawn doesn't exist yet
337 return "-";
15106e82
BA
338 const squares = Object.keys(this.ispawn);
339 if (squares.length == 0)
b4ae3ff6 340 return "-";
15106e82 341 return squares.join(",");
41534b92
BA
342 }
343
344 // Set flags from fen (castle: white a,h then black a,h)
345 setFlags(fenflags) {
346 this.castleFlags = {
15106e82
BA
347 w: [0, 1].map(i => parseInt(fenflags.charAt(i), 36)),
348 b: [2, 3].map(i => parseInt(fenflags.charAt(i), 36))
41534b92
BA
349 };
350 }
351
352 //////////////////
353 // INITIALIZATION
354
bc2bc396 355 constructor(o) {
41534b92 356 this.options = o.options;
535c464b
BA
357 // Fill missing options (always the case if random challenge)
358 (V.Options.select || []).concat(V.Options.input || []).forEach(opt => {
359 if (this.options[opt.variable] === undefined)
360 this.options[opt.variable] = opt.defaut;
361 });
f31de5e4
BA
362
363 // Some variables
41534b92 364 this.playerColor = o.color;
15106e82 365 this.afterPlay = o.afterPlay; //trigger some actions after playing a move
f31de5e4
BA
366 this.containerId = o.element;
367 this.isDiagram = o.diagram;
368 this.marks = o.marks;
41534b92 369
f31de5e4 370 // Initializations
b4ae3ff6
BA
371 if (!o.fen)
372 o.fen = this.genRandInitFen(o.seed);
5f08c59b 373 this.re_initFromFen(o.fen);
41534b92
BA
374 this.graphicalInit();
375 }
376
5f08c59b
BA
377 re_initFromFen(fen, oldBoard) {
378 const fenParsed = this.parseFen(fen);
379 this.board = oldBoard || this.getBoard(fenParsed.position);
380 this.turn = fenParsed.turn;
381 this.movesCount = parseInt(fenParsed.movesCount, 10);
382 this.setOtherVariables(fenParsed);
383 }
384
41534b92
BA
385 // Turn position fen into double array ["wb","wp","bk",...]
386 getBoard(position) {
387 const rows = position.split("/");
388 let board = ArrayFun.init(this.size.x, this.size.y, "");
389 for (let i = 0; i < rows.length; i++) {
390 let j = 0;
391 for (let indexInRow = 0; indexInRow < rows[i].length; indexInRow++) {
392 const character = rows[i][indexInRow];
393 const num = parseInt(character, 10);
394 // If num is a number, just shift j:
b4ae3ff6
BA
395 if (!isNaN(num))
396 j += num;
41534b92 397 // Else: something at position i,j
b4ae3ff6
BA
398 else
399 board[i][j++] = this.fen2board(character);
41534b92
BA
400 }
401 }
402 return board;
403 }
404
405 // Some additional variables from FEN (variant dependant)
406 setOtherVariables(fenParsed) {
407 // Set flags and enpassant:
b4ae3ff6
BA
408 if (this.hasFlags)
409 this.setFlags(fenParsed.flags);
41534b92
BA
410 if (this.hasEnpassant)
411 this.epSquare = this.getEpSquare(fenParsed.enpassant);
549ca151 412 if (this.hasReserve && !this.isDiagram)
b4ae3ff6
BA
413 this.initReserves(fenParsed.reserve);
414 if (this.options["crazyhouse"])
415 this.initIspawn(fenParsed.ispawn);
cc2c7183
BA
416 if (this.options["teleport"]) {
417 this.subTurnTeleport = 1;
418 this.captured = null;
419 }
41534b92 420 if (this.options["dark"]) {
41534b92 421 // Setup enlightened: squares reachable by player side
c9ab0340
BA
422 this.enlightened = ArrayFun.init(this.size.x, this.size.y, false);
423 this.updateEnlightened();
41534b92 424 }
5f08c59b
BA
425 this.subTurn = 1; //may be unused
426 if (!this.moveStack) //avoid resetting (unwanted)
427 this.moveStack = [];
41534b92
BA
428 }
429
ca8a3993 430 // ordering as in pieces() p,r,n,b,q,k
dc10e429
BA
431 initReserves(reserveStr, pieceArray) {
432 if (!pieceArray)
433 pieceArray = ['p', 'r', 'n', 'b', 'q', 'k'];
ca8a3993 434 const counts = reserveStr.split("").map(c => parseInt(c, 36));
dc10e429
BA
435 const L = pieceArray.length;
436 this.reserve = {
437 w: ArrayFun.toObject(pieceArray, counts.slice(0, L)),
438 b: ArrayFun.toObject(pieceArray, counts.slice(L, 2 * L))
439 };
41534b92
BA
440 }
441
442 initIspawn(ispawnStr) {
15106e82
BA
443 if (ispawnStr != "-")
444 this.ispawn = ArrayFun.toObject(ispawnStr.split(","), true);
b4ae3ff6
BA
445 else
446 this.ispawn = {};
41534b92
BA
447 }
448
ca8a3993
BA
449 ////////////////
450 // VISUAL UTILS
451
452 getPieceWidth(rwidth) {
453 return (rwidth / this.size.y);
454 }
455
456 getReserveSquareSize(rwidth, nbR) {
457 const sqSize = this.getPieceWidth(rwidth);
458 return Math.min(sqSize, rwidth / nbR);
459 }
460
461 getReserveNumId(color, piece) {
462 return `${this.containerId}|rnum-${color}${piece}`;
463 }
464
41534b92
BA
465 getNbReservePieces(color) {
466 return (
467 Object.values(this.reserve[color]).reduce(
468 (oldV,newV) => oldV + (newV > 0 ? 1 : 0), 0)
469 );
470 }
471
15106e82 472 getRankInReserve(c, p) {
ceeac4e8 473 const pieces = Object.keys(this.pieces(c, c, p));
15106e82
BA
474 const lastIndex = pieces.findIndex(pp => pp == p)
475 let toTest = pieces.slice(0, lastIndex);
476 return toTest.reduce(
477 (oldV,newV) => oldV + (this.reserve[c][newV] > 0 ? 1 : 0), 0);
478 }
479
9b760538 480 static AddClass_es(elt, class_es) {
3b641716
BA
481 if (!Array.isArray(class_es))
482 class_es = [class_es];
9b760538 483 class_es.forEach(cl => elt.classList.add(cl));
3b641716
BA
484 }
485
9b760538 486 static RemoveClass_es(elt, class_es) {
3b641716
BA
487 if (!Array.isArray(class_es))
488 class_es = [class_es];
9b760538 489 class_es.forEach(cl => elt.classList.remove(cl));
3b641716
BA
490 }
491
ca8a3993
BA
492 // Generally light square bottom-right
493 getSquareColorClass(x, y) {
494 return ((x+y) % 2 == 0 ? "light-square": "dark-square");
495 }
496
497 getMaxDistance(r) {
498 // Works for all rectangular boards:
499 return Math.sqrt(r.width ** 2 + r.height ** 2);
500 }
501
502 getDomPiece(x, y) {
503 return (typeof x == "string" ? this.r_pieces : this.g_pieces)[x][y];
504 }
505
506 //////////////////
507 // VISUAL METHODS
508
41534b92 509 graphicalInit() {
65c770d1
BA
510 const g_init = () => {
511 this.re_drawBoardElements();
549ca151 512 if (!this.isDiagram && !this.mouseListeners && !this.touchListeners)
5abaabb3 513 this.initMouseEvents();
65c770d1
BA
514 };
515 let container = document.getElementById(this.containerId);
549ca151
BA
516 this.windowResizeObs = new ResizeObserver(g_init);
517 this.windowResizeObs.observe(container);
41534b92
BA
518 }
519
520 re_drawBoardElements() {
521 const board = this.getSvgChessboard();
e7b64798
BA
522 const container = document.getElementById(this.containerId);
523 const rc = container.getBoundingClientRect();
524 let chessboard = container.querySelector(".chessboard");
3c61449b
BA
525 chessboard.innerHTML = "";
526 chessboard.insertAdjacentHTML('beforeend', board);
41534b92 527 // Compare window ratio width / height to aspectRatio:
e7b64798 528 const windowRatio = rc.width / rc.height;
41534b92 529 let cbWidth, cbHeight;
f55a0a67
BA
530 const vRatio = this.size.ratio || 1;
531 if (windowRatio <= vRatio) {
41534b92 532 // Limiting dimension is width:
e7b64798 533 cbWidth = Math.min(rc.width, 767);
f55a0a67 534 cbHeight = cbWidth / vRatio;
41534b92
BA
535 }
536 else {
537 // Limiting dimension is height:
e7b64798 538 cbHeight = Math.min(rc.height, 767);
f55a0a67 539 cbWidth = cbHeight * vRatio;
41534b92 540 }
549ca151 541 if (this.hasReserve && !this.isDiagram) {
41534b92
BA
542 const sqSize = cbWidth / this.size.y;
543 // NOTE: allocate space for reserves (up/down) even if they are empty
15106e82 544 // Cannot use getReserveSquareSize() here, but sqSize is an upper bound.
e7b64798
BA
545 if ((rc.height - cbHeight) / 2 < sqSize + 5) {
546 cbHeight = rc.height - 2 * (sqSize + 5);
f55a0a67 547 cbWidth = cbHeight * vRatio;
41534b92
BA
548 }
549 }
3c61449b
BA
550 chessboard.style.width = cbWidth + "px";
551 chessboard.style.height = cbHeight + "px";
41534b92 552 // Center chessboard:
e7b64798
BA
553 const spaceLeft = (rc.width - cbWidth) / 2,
554 spaceTop = (rc.height - cbHeight) / 2;
3c61449b
BA
555 chessboard.style.left = spaceLeft + "px";
556 chessboard.style.top = spaceTop + "px";
41534b92
BA
557 // Give sizes instead of recomputing them,
558 // because chessboard might not be drawn yet.
10c9010b 559 this.setupVisualPieces({
41534b92
BA
560 width: cbWidth,
561 height: cbHeight,
562 x: spaceLeft,
563 y: spaceTop
564 });
565 }
566
567 // Get SVG board (background, no pieces)
568 getSvgChessboard() {
41534b92
BA
569 const flipped = (this.playerColor == 'b');
570 let board = `
571 <svg
f55a0a67 572 viewBox="0 0 ${10*this.size.y} ${10*this.size.x}"
535c464b 573 class="chessboard_SVG">`;
728cb1e3
BA
574 for (let i=0; i < this.size.x; i++) {
575 for (let j=0; j < this.size.y; j++) {
9b760538
BA
576 if (!this.onBoard(i, j))
577 continue;
41534b92
BA
578 const ii = (flipped ? this.size.x - 1 - i : i);
579 const jj = (flipped ? this.size.y - 1 - j : j);
c7bf7b1b
BA
580 let classes = this.getSquareColorClass(ii, jj);
581 if (this.enlightened && !this.enlightened[ii][jj])
582 classes += " in-shadow";
41534b92 583 // NOTE: x / y reversed because coordinates system is reversed.
535c464b
BA
584 board += `
585 <rect
586 class="${classes}"
587 id="${this.coordsToId({x: ii, y: jj})}"
588 width="10"
589 height="10"
590 x="${10*j}"
591 y="${10*i}"
592 />`;
41534b92
BA
593 }
594 }
535c464b 595 board += "</svg>";
41534b92
BA
596 return board;
597 }
598
10c9010b 599 setupVisualPieces(r) {
3c61449b
BA
600 let chessboard =
601 document.getElementById(this.containerId).querySelector(".chessboard");
b4ae3ff6
BA
602 if (!r)
603 r = chessboard.getBoundingClientRect();
41534b92 604 const pieceWidth = this.getPieceWidth(r.width);
b2fc1259
BA
605 const addPiece = (i, j, arrName, classes) => {
606 this[arrName][i][j] = document.createElement("piece");
607 C.AddClass_es(this[arrName][i][j], classes);
608 this[arrName][i][j].style.width = pieceWidth + "px";
609 this[arrName][i][j].style.height = pieceWidth + "px";
610 let [ip, jp] = this.getPixelPosition(i, j, r);
611 // Translate coordinates to use chessboard as reference:
612 this[arrName][i][j].style.transform =
613 `translate(${ip - r.x}px,${jp - r.y}px)`;
614 chessboard.appendChild(this[arrName][i][j]);
615 };
616 const conditionalReset = (arrName) => {
617 if (this[arrName]) {
618 // Refreshing: delete old pieces first. This isn't necessary,
619 // but simpler (this method isn't called many times)
620 for (let i=0; i<this.size.x; i++) {
621 for (let j=0; j<this.size.y; j++) {
622 if (this[arrName][i][j]) {
623 this[arrName][i][j].remove();
624 this[arrName][i][j] = null;
625 }
626 }
627 }
628 }
629 else
630 this[arrName] = ArrayFun.init(this.size.x, this.size.y, null);
631 if (arrName == "d_pieces")
632 this.marks.forEach(([i, j]) => addPiece(i, j, arrName, "mark"));
633 };
634 if (this.marks)
635 conditionalReset("d_pieces");
636 conditionalReset("g_pieces");
41534b92
BA
637 for (let i=0; i < this.size.x; i++) {
638 for (let j=0; j < this.size.y; j++) {
c9ab0340 639 if (this.board[i][j] != "") {
41534b92 640 const color = this.getColor(i, j);
cc2c7183 641 const piece = this.getPiece(i, j);
b2fc1259 642 addPiece(i, j, "g_pieces", this.pieces(color, i, j)[piece]["class"]);
616a8d7a 643 this.g_pieces[i][j].classList.add(V.GetColorClass(color));
c9ab0340
BA
644 if (this.enlightened && !this.enlightened[i][j])
645 this.g_pieces[i][j].classList.add("hidden");
b2fc1259
BA
646 }
647 if (this.marks && this.d_pieces[i][j]) {
648 let classes = ["mark"];
649 if (this.board[i][j] != "")
650 classes.push("transparent");
651 addPiece(i, j, "d_pieces", classes);
41534b92
BA
652 }
653 }
654 }
549ca151 655 if (this.hasReserve && !this.isDiagram)
b4ae3ff6 656 this.re_drawReserve(['w', 'b'], r);
41534b92
BA
657 }
658
24872b22 659 // NOTE: assume this.reserve != null
41534b92
BA
660 re_drawReserve(colors, r) {
661 if (this.r_pieces) {
662 // Remove (old) reserve pieces
663 for (let c of colors) {
24872b22
BA
664 Object.keys(this.r_pieces[c]).forEach(p => {
665 this.r_pieces[c][p].remove();
666 delete this.r_pieces[c][p];
667 const numId = this.getReserveNumId(c, p);
668 document.getElementById(numId).remove();
41534b92 669 });
41534b92
BA
670 }
671 }
b4ae3ff6 672 else
9db5050a
BA
673 this.r_pieces = { w: {}, b: {} };
674 let container = document.getElementById(this.containerId);
b4ae3ff6 675 if (!r)
9db5050a 676 r = container.querySelector(".chessboard").getBoundingClientRect();
41534b92 677 for (let c of colors) {
24872b22
BA
678 let reservesDiv = document.getElementById("reserves_" + c);
679 if (reservesDiv)
680 reservesDiv.remove();
b4ae3ff6
BA
681 if (!this.reserve[c])
682 continue;
41534b92 683 const nbR = this.getNbReservePieces(c);
b4ae3ff6
BA
684 if (nbR == 0)
685 continue;
41534b92
BA
686 const sqResSize = this.getReserveSquareSize(r.width, nbR);
687 let ridx = 0;
688 const vShift = (c == this.playerColor ? r.height + 5 : -sqResSize - 5);
689 const [i0, j0] = [r.x, r.y + vShift];
690 let rcontainer = document.createElement("div");
691 rcontainer.id = "reserves_" + c;
692 rcontainer.classList.add("reserves");
693 rcontainer.style.left = i0 + "px";
694 rcontainer.style.top = j0 + "px";
1aa9054d
BA
695 // NOTE: +1 fix display bug on Firefox at least
696 rcontainer.style.width = (nbR * sqResSize + 1) + "px";
41534b92 697 rcontainer.style.height = sqResSize + "px";
9db5050a 698 container.appendChild(rcontainer);
41534b92 699 for (let p of Object.keys(this.reserve[c])) {
b4ae3ff6
BA
700 if (this.reserve[c][p] == 0)
701 continue;
41534b92 702 let r_cell = document.createElement("div");
15106e82 703 r_cell.id = this.coordsToId({x: c, y: p});
41534b92 704 r_cell.classList.add("reserve-cell");
1aa9054d
BA
705 r_cell.style.width = sqResSize + "px";
706 r_cell.style.height = sqResSize + "px";
41534b92
BA
707 rcontainer.appendChild(r_cell);
708 let piece = document.createElement("piece");
2159c239 709 C.AddClass_es(piece, this.pieces(c, c, p)[p]["class"]);
616a8d7a 710 piece.classList.add(V.GetColorClass(c));
41534b92
BA
711 piece.style.width = "100%";
712 piece.style.height = "100%";
713 this.r_pieces[c][p] = piece;
714 r_cell.appendChild(piece);
715 let number = document.createElement("div");
716 number.textContent = this.reserve[c][p];
717 number.classList.add("reserve-num");
718 number.id = this.getReserveNumId(c, p);
719 const fontSize = "1.3em";
720 number.style.fontSize = fontSize;
721 number.style.fontSize = fontSize;
722 r_cell.appendChild(number);
723 ridx++;
724 }
725 }
726 }
727
728 updateReserve(color, piece, count) {
55a15dcb 729 if (this.options["cannibal"] && C.CannibalKings[piece])
cc2c7183 730 piece = "k"; //capturing cannibal king: back to king form
41534b92
BA
731 const oldCount = this.reserve[color][piece];
732 this.reserve[color][piece] = count;
182ba661
BA
733 // Redrawing is much easier if count==0 (or undefined)
734 if ([oldCount, count].some(item => !item))
b4ae3ff6 735 this.re_drawReserve([color]);
41534b92
BA
736 else {
737 const numId = this.getReserveNumId(color, piece);
738 document.getElementById(numId).textContent = count;
739 }
740 }
741
c4e9bb92
BA
742 // Resize board: no need to destroy/recreate pieces
743 rescale(mode) {
e7b64798
BA
744 const container = document.getElementById(this.containerId);
745 let chessboard = container.querySelector(".chessboard");
746 const rc = container.getBoundingClientRect(),
747 r = chessboard.getBoundingClientRect();
c4e9bb92
BA
748 const multFact = (mode == "up" ? 1.05 : 0.95);
749 let [newWidth, newHeight] = [multFact * r.width, multFact * r.height];
535c464b 750 // Stay in window:
f55a0a67 751 const vRatio = this.size.ratio || 1;
e7b64798
BA
752 if (newWidth > rc.width) {
753 newWidth = rc.width;
f55a0a67 754 newHeight = newWidth / vRatio;
c4e9bb92 755 }
e7b64798
BA
756 if (newHeight > rc.height) {
757 newHeight = rc.height;
f55a0a67 758 newWidth = newHeight * vRatio;
c4e9bb92 759 }
535c464b
BA
760 chessboard.style.width = newWidth + "px";
761 chessboard.style.height = newHeight + "px";
e7b64798 762 const newX = (rc.width - newWidth) / 2;
3c61449b 763 chessboard.style.left = newX + "px";
e7b64798 764 const newY = (rc.height - newHeight) / 2;
3c61449b 765 chessboard.style.top = newY + "px";
9db5050a 766 const newR = {x: newX, y: newY, width: newWidth, height: newHeight};
c4e9bb92 767 const pieceWidth = this.getPieceWidth(newWidth);
d621e620
BA
768 // NOTE: next "if" for variants which use squares filling
769 // instead of "physical", moving pieces
770 if (this.g_pieces) {
c4e9bb92
BA
771 for (let i=0; i < this.size.x; i++) {
772 for (let j=0; j < this.size.y; j++) {
773 if (this.g_pieces[i][j]) {
d621e620 774 // NOTE: could also use CSS transform "scale"
c4e9bb92
BA
775 this.g_pieces[i][j].style.width = pieceWidth + "px";
776 this.g_pieces[i][j].style.height = pieceWidth + "px";
777 const [ip, jp] = this.getPixelPosition(i, j, newR);
d621e620 778 // Translate coordinates to use chessboard as reference:
c4e9bb92 779 this.g_pieces[i][j].style.transform =
d621e620
BA
780 `translate(${ip - newX}px,${jp - newY}px)`;
781 }
41534b92
BA
782 }
783 }
784 }
b2fc1259 785 if (this.hasReserve)
c4e9bb92 786 this.rescaleReserve(newR);
41534b92
BA
787 }
788
789 rescaleReserve(r) {
41534b92 790 for (let c of ['w','b']) {
b4ae3ff6
BA
791 if (!this.reserve[c])
792 continue;
41534b92 793 const nbR = this.getNbReservePieces(c);
b4ae3ff6
BA
794 if (nbR == 0)
795 continue;
41534b92
BA
796 // Resize container first
797 const sqResSize = this.getReserveSquareSize(r.width, nbR);
798 const vShift = (c == this.playerColor ? r.height + 5 : -sqResSize - 5);
799 const [i0, j0] = [r.x, r.y + vShift];
800 let rcontainer = document.getElementById("reserves_" + c);
801 rcontainer.style.left = i0 + "px";
802 rcontainer.style.top = j0 + "px";
1aa9054d 803 rcontainer.style.width = (nbR * sqResSize + 1) + "px";
41534b92
BA
804 rcontainer.style.height = sqResSize + "px";
805 // And then reserve cells:
806 const rpieceWidth = this.getReserveSquareSize(r.width, nbR);
807 Object.keys(this.reserve[c]).forEach(p => {
b4ae3ff6
BA
808 if (this.reserve[c][p] == 0)
809 return;
15106e82 810 let r_cell = document.getElementById(this.coordsToId({x: c, y: p}));
1aa9054d
BA
811 r_cell.style.width = sqResSize + "px";
812 r_cell.style.height = sqResSize + "px";
41534b92
BA
813 });
814 }
815 }
816
9db5050a 817 // Return the absolute pixel coordinates given current position.
41534b92
BA
818 // Our coordinate system differs from CSS one (x <--> y).
819 // We return here the CSS coordinates (more useful).
820 getPixelPosition(i, j, r) {
b4ae3ff6
BA
821 if (i < 0 || j < 0)
822 return [0, 0]; //piece vanishes
15106e82
BA
823 let x, y;
824 if (typeof i == "string") {
825 // Reserves: need to know the rank of piece
826 const nbR = this.getNbReservePieces(i);
827 const rsqSize = this.getReserveSquareSize(r.width, nbR);
828 x = this.getRankInReserve(i, j) * rsqSize;
829 y = (this.playerColor == i ? y = r.height + 5 : - 5 - rsqSize);
830 }
831 else {
832 const sqSize = r.width / this.size.y;
833 const flipped = (this.playerColor == 'b');
834 x = (flipped ? this.size.y - 1 - j : j) * sqSize;
835 y = (flipped ? this.size.x - 1 - i : i) * sqSize;
836 }
9db5050a 837 return [r.x + x, r.y + y];
41534b92
BA
838 }
839
840 initMouseEvents() {
9db5050a
BA
841 let container = document.getElementById(this.containerId);
842 let chessboard = container.querySelector(".chessboard");
41534b92
BA
843
844 const getOffset = e => {
3c61449b
BA
845 if (e.clientX)
846 // Mouse
847 return {x: e.clientX, y: e.clientY};
41534b92
BA
848 let touchLocation = null;
849 if (e.targetTouches && e.targetTouches.length >= 1)
850 // Touch screen, dragstart
851 touchLocation = e.targetTouches[0];
852 else if (e.changedTouches && e.changedTouches.length >= 1)
853 // Touch screen, dragend
854 touchLocation = e.changedTouches[0];
855 if (touchLocation)
11625344 856 return {x: touchLocation.clientX, y: touchLocation.clientY};
57b8015b 857 return {x: 0, y: 0}; //shouldn't reach here =)
41534b92
BA
858 }
859
860 const centerOnCursor = (piece, e) => {
15106e82 861 const centerShift = this.getPieceWidth(r.width) / 2;
41534b92 862 const offset = getOffset(e);
9db5050a
BA
863 piece.style.left = (offset.x - centerShift) + "px";
864 piece.style.top = (offset.y - centerShift) + "px";
41534b92
BA
865 }
866
867 let start = null,
868 r = null,
869 startPiece, curPiece = null,
15106e82 870 pieceWidth;
41534b92 871 const mousedown = (e) => {
cb17fed8 872 // Disable zoom on smartphones:
b4ae3ff6
BA
873 if (e.touches && e.touches.length > 1)
874 e.preventDefault();
3c61449b 875 r = chessboard.getBoundingClientRect();
15106e82
BA
876 pieceWidth = this.getPieceWidth(r.width);
877 const cd = this.idToCoords(e.target.id);
878 if (cd) {
879 const move = this.doClick(cd);
b4ae3ff6 880 if (move)
e7d409fc 881 this.buildMoveStack(move, r);
6b9320bb 882 else if (!this.clickOnly) {
15106e82
BA
883 const [x, y] = Object.values(cd);
884 if (typeof x != "number")
885 startPiece = this.r_pieces[x][y];
886 else
887 startPiece = this.g_pieces[x][y];
888 if (startPiece && this.canIplay(x, y)) {
41534b92 889 e.preventDefault();
15106e82 890 start = cd;
41534b92
BA
891 curPiece = startPiece.cloneNode();
892 curPiece.style.transform = "none";
893 curPiece.style.zIndex = 5;
15106e82
BA
894 curPiece.style.width = pieceWidth + "px";
895 curPiece.style.height = pieceWidth + "px";
41534b92 896 centerOnCursor(curPiece, e);
9db5050a 897 container.appendChild(curPiece);
41534b92 898 startPiece.style.opacity = "0.4";
3c61449b 899 chessboard.style.cursor = "none";
41534b92
BA
900 }
901 }
902 }
903 };
904
905 const mousemove = (e) => {
906 if (start) {
907 e.preventDefault();
908 centerOnCursor(curPiece, e);
909 }
11625344
BA
910 else if (e.changedTouches && e.changedTouches.length >= 1)
911 // Attempt to prevent horizontal swipe...
912 e.preventDefault();
41534b92
BA
913 };
914
915 const mouseup = (e) => {
b4ae3ff6
BA
916 if (!start)
917 return;
41534b92
BA
918 const [x, y] = [start.x, start.y];
919 start = null;
920 e.preventDefault();
3c61449b 921 chessboard.style.cursor = "pointer";
41534b92
BA
922 startPiece.style.opacity = "1";
923 const offset = getOffset(e);
924 const landingElt = document.elementFromPoint(offset.x, offset.y);
15106e82
BA
925 const cd =
926 (landingElt ? this.idToCoords(landingElt.id) : undefined);
927 if (cd) {
41534b92
BA
928 // NOTE: clearly suboptimal, but much easier, and not a big deal.
929 const potentialMoves = this.getPotentialMovesFrom([x, y])
15106e82 930 .filter(m => m.end.x == cd.x && m.end.y == cd.y);
41534b92 931 const moves = this.filterValid(potentialMoves);
b4ae3ff6
BA
932 if (moves.length >= 2)
933 this.showChoices(moves, r);
934 else if (moves.length == 1)
e7d409fc 935 this.buildMoveStack(moves[0], r);
41534b92
BA
936 }
937 curPiece.remove();
938 };
939
a5da6269
BA
940 const resize = (e) => this.rescale(e.deltaY < 0 ? "up" : "down");
941
41534b92 942 if ('onmousedown' in window) {
a5da6269
BA
943 this.mouseListeners = [
944 {type: "mousedown", listener: mousedown},
945 {type: "mousemove", listener: mousemove},
946 {type: "mouseup", listener: mouseup},
947 {type: "wheel", listener: resize}
948 ];
949 this.mouseListeners.forEach(ml => {
950 document.addEventListener(ml.type, ml.listener);
951 });
41534b92
BA
952 }
953 if ('ontouchstart' in window) {
a5da6269
BA
954 this.touchListeners = [
955 {type: "touchstart", listener: mousedown},
956 {type: "touchmove", listener: mousemove},
957 {type: "touchend", listener: mouseup}
958 ];
959 this.touchListeners.forEach(tl => {
960 // https://stackoverflow.com/a/42509310/12660887
961 document.addEventListener(tl.type, tl.listener, {passive: false});
962 });
41534b92 963 }
11625344 964 // TODO: onpointerdown/move/up ? See reveal.js /controllers/touch.js
41534b92
BA
965 }
966
7c038235 967 // NOTE: not called if isDiagram
a5da6269 968 removeListeners() {
549ca151
BA
969 let container = document.getElementById(this.containerId);
970 this.windowResizeObs.unobserve(container);
a5da6269
BA
971 if ('onmousedown' in window) {
972 this.mouseListeners.forEach(ml => {
973 document.removeEventListener(ml.type, ml.listener);
974 });
975 }
976 if ('ontouchstart' in window) {
977 this.touchListeners.forEach(tl => {
978 // https://stackoverflow.com/a/42509310/12660887
979 document.removeEventListener(tl.type, tl.listener);
980 });
981 }
982 }
983
41534b92
BA
984 showChoices(moves, r) {
985 let container = document.getElementById(this.containerId);
3c61449b 986 let chessboard = container.querySelector(".chessboard");
41534b92
BA
987 let choices = document.createElement("div");
988 choices.id = "choices";
a2bb7e06
BA
989 if (!r)
990 r = chessboard.getBoundingClientRect();
41534b92
BA
991 choices.style.width = r.width + "px";
992 choices.style.height = r.height + "px";
993 choices.style.left = r.x + "px";
994 choices.style.top = r.y + "px";
3c61449b
BA
995 chessboard.style.opacity = "0.5";
996 container.appendChild(choices);
15106e82 997 const squareWidth = r.width / this.size.y;
41534b92
BA
998 const firstUpLeft = (r.width - (moves.length * squareWidth)) / 2;
999 const firstUpTop = (r.height - squareWidth) / 2;
1000 const color = moves[0].appear[0].c;
1001 const callback = (m) => {
3c61449b
BA
1002 chessboard.style.opacity = "1";
1003 container.removeChild(choices);
e7d409fc 1004 this.buildMoveStack(m, r);
41534b92
BA
1005 }
1006 for (let i=0; i < moves.length; i++) {
1007 let choice = document.createElement("div");
1008 choice.classList.add("choice");
1009 choice.style.width = squareWidth + "px";
1010 choice.style.height = squareWidth + "px";
1011 choice.style.left = (firstUpLeft + i * squareWidth) + "px";
1012 choice.style.top = firstUpTop + "px";
1013 choice.style.backgroundColor = "lightyellow";
1014 choice.onclick = () => callback(moves[i]);
1015 const piece = document.createElement("piece");
3b641716 1016 const cdisp = moves[i].choice || moves[i].appear[0].p;
2159c239
BA
1017 C.AddClass_es(piece,
1018 this.pieces(color, moves[i].end.x, moves[i].end.y)[cdisp]["class"]);
616a8d7a 1019 piece.classList.add(V.GetColorClass(color));
41534b92
BA
1020 piece.style.width = "100%";
1021 piece.style.height = "100%";
1022 choice.appendChild(piece);
1023 choices.appendChild(choice);
1024 }
1025 }
1026
98d14451
BA
1027 displayMessage(elt, msg, classe_s, timeout) {
1028 if (elt)
1029 // Fixed element, e.g. for Dice Chess
1030 elt.innerHTML = msg;
1031 else {
1032 // Temporary div (Chakart, Apocalypse...)
1033 let divMsg = document.createElement("div");
1034 C.AddClass_es(divMsg, classe_s);
1035 divMsg.innerHTML = msg;
1036 let container = document.getElementById(this.containerId);
1037 container.appendChild(divMsg);
1038 setTimeout(() => container.removeChild(divMsg), timeout);
1039 }
1040 }
1041
ca8a3993
BA
1042 ////////////////
1043 // DARK METHODS
1044
1045 updateEnlightened() {
1046 this.oldEnlightened = this.enlightened;
1047 this.enlightened = ArrayFun.init(this.size.x, this.size.y, false);
1048 // Add pieces positions + all squares reachable by moves (includes Zen):
1049 for (let x=0; x<this.size.x; x++) {
1050 for (let y=0; y<this.size.y; y++) {
1051 if (this.board[x][y] != "" && this.getColor(x, y) == this.playerColor)
1052 {
1053 this.enlightened[x][y] = true;
1054 this.getPotentialMovesFrom([x, y]).forEach(m => {
1055 this.enlightened[m.end.x][m.end.y] = true;
1056 });
1057 }
1058 }
1059 }
1060 if (this.epSquare)
1061 this.enlightEnpassant();
1062 }
1063
1064 // Include square of the en-passant capturing square:
1065 enlightEnpassant() {
1066 // NOTE: shortcut, pawn has only one attack type, doesn't depend on square
ceeac4e8
BA
1067 // TODO: (0, 0) is wrong, would need to place an attacker here...
1068 const steps = this.pieces(this.playerColor, 0, 0)["p"].attack[0].steps;
ca8a3993
BA
1069 for (let step of steps) {
1070 const x = this.epSquare.x - step[0],
1071 y = this.getY(this.epSquare.y - step[1]);
1072 if (
1073 this.onBoard(x, y) &&
1074 this.getColor(x, y) == this.playerColor &&
1075 this.getPieceType(x, y) == "p"
1076 ) {
1077 this.enlightened[x][this.epSquare.y] = true;
1078 break;
1079 }
1080 }
1081 }
1082
1083 // Apply diff this.enlightened --> oldEnlightened on board
1084 graphUpdateEnlightened() {
1085 let chessboard =
1086 document.getElementById(this.containerId).querySelector(".chessboard");
1087 const r = chessboard.getBoundingClientRect();
1088 const pieceWidth = this.getPieceWidth(r.width);
1089 for (let x=0; x<this.size.x; x++) {
1090 for (let y=0; y<this.size.y; y++) {
1091 if (!this.enlightened[x][y] && this.oldEnlightened[x][y]) {
1092 let elt = document.getElementById(this.coordsToId({x: x, y: y}));
1093 elt.classList.add("in-shadow");
1094 if (this.g_pieces[x][y])
1095 this.g_pieces[x][y].classList.add("hidden");
1096 }
1097 else if (this.enlightened[x][y] && !this.oldEnlightened[x][y]) {
1098 let elt = document.getElementById(this.coordsToId({x: x, y: y}));
1099 elt.classList.remove("in-shadow");
1100 if (this.g_pieces[x][y])
1101 this.g_pieces[x][y].classList.remove("hidden");
1102 }
1103 }
1104 }
1105 }
1106
41534b92
BA
1107 //////////////
1108 // BASIC UTILS
1109
1110 get size() {
15106e82
BA
1111 return {
1112 x: 8,
1113 y: 8,
f55a0a67 1114 ratio: 1 //for rectangular board = y / x (optional, 1 = default)
15106e82 1115 };
41534b92
BA
1116 }
1117
ca8a3993 1118 // Color of thing on square (i,j). '' if square is empty
41534b92 1119 getColor(i, j) {
15106e82
BA
1120 if (typeof i == "string")
1121 return i; //reserves
41534b92
BA
1122 return this.board[i][j].charAt(0);
1123 }
1124
15106e82 1125 static GetColorClass(c) {
bc2bc396
BA
1126 if (c == 'w')
1127 return "white";
1128 if (c == 'b')
1129 return "black";
24872b22 1130 return "other-color"; //unidentified color
15106e82
BA
1131 }
1132
ca8a3993 1133 // Piece on i,j. '' if square is empty
41534b92 1134 getPiece(i, j) {
15106e82
BA
1135 if (typeof j == "string")
1136 return j; //reserves
41534b92
BA
1137 return this.board[i][j].charAt(1);
1138 }
1139
cc2c7183 1140 // Piece type on square (i,j)
6b9320bb 1141 getPieceType(x, y, p) {
65cf1690 1142 if (!p)
6b9320bb 1143 p = this.getPiece(x, y);
ceeac4e8 1144 return this.pieces(this.getColor(x, y), x, y)[p].moveas || p;
6b9320bb
BA
1145 }
1146
1147 isKing(x, y, p) {
1148 if (!p)
1149 p = this.getPiece(x, y);
1150 if (!this.options["cannibal"])
1151 return p == 'k';
1152 return !!C.CannibalKings[p];
cc2c7183
BA
1153 }
1154
ceeac4e8
BA
1155 static GetOppTurn(color) {
1156 return (color == 'w' ? 'b' : 'w');
1157 }
1158
1159 // Get opponent color(s): may differ from turn (e.g. Checkered)
1160 getOppCols(color) {
41534b92
BA
1161 return (color == "w" ? "b" : "w");
1162 }
1163
41534b92
BA
1164 // Is (x,y) on the chessboard?
1165 onBoard(x, y) {
b99ce1fb
BA
1166 return (x >= 0 && x < this.size.x &&
1167 y >= 0 && y < this.size.y);
41534b92
BA
1168 }
1169
15106e82 1170 // Am I allowed to move thing at square x,y ?
41534b92 1171 canIplay(x, y) {
0c44c676 1172 return (this.playerColor == this.turn && this.getColor(x, y) == this.turn);
41534b92
BA
1173 }
1174
1175 ////////////////////////
1176 // PIECES SPECIFICATIONS
1177
5006aaca
BA
1178 getPawnShift(color) {
1179 return (color == "w" ? -1 : 1);
1180 }
1181
c9ab0340 1182 pieces(color, x, y) {
5006aaca 1183 const pawnShift = this.getPawnShift(color);
9db5050a
BA
1184 // NOTE: jump 2 squares from first rank (pawns can be here sometimes)
1185 const initRank = ((color == 'w' && x >= 6) || (color == 'b' && x <= 1));
41534b92
BA
1186 return {
1187 'p': {
1188 "class": "pawn",
c9ab0340
BA
1189 moves: [
1190 {
1191 steps: [[pawnShift, 0]],
1192 range: (initRank ? 2 : 1)
1193 }
1194 ],
1195 attack: [
1196 {
1197 steps: [[pawnShift, 1], [pawnShift, -1]],
1198 range: 1
1199 }
1200 ]
41534b92 1201 },
41534b92
BA
1202 'r': {
1203 "class": "rook",
33b42748 1204 both: [
c9ab0340
BA
1205 {steps: [[0, 1], [0, -1], [1, 0], [-1, 0]]}
1206 ]
41534b92 1207 },
41534b92
BA
1208 'n': {
1209 "class": "knight",
33b42748 1210 both: [
c9ab0340
BA
1211 {
1212 steps: [
1213 [1, 2], [1, -2], [-1, 2], [-1, -2],
1214 [2, 1], [-2, 1], [2, -1], [-2, -1]
1215 ],
1216 range: 1
1217 }
1218 ]
41534b92 1219 },
41534b92
BA
1220 'b': {
1221 "class": "bishop",
33b42748 1222 both: [
c9ab0340
BA
1223 {steps: [[1, 1], [1, -1], [-1, 1], [-1, -1]]}
1224 ]
41534b92 1225 },
41534b92
BA
1226 'q': {
1227 "class": "queen",
33b42748 1228 both: [
c9ab0340
BA
1229 {
1230 steps: [
1231 [0, 1], [0, -1], [1, 0], [-1, 0],
1232 [1, 1], [1, -1], [-1, 1], [-1, -1]
1233 ]
1234 }
41534b92
BA
1235 ]
1236 },
41534b92
BA
1237 'k': {
1238 "class": "king",
33b42748 1239 both: [
c9ab0340
BA
1240 {
1241 steps: [
1242 [0, 1], [0, -1], [1, 0], [-1, 0],
1243 [1, 1], [1, -1], [-1, 1], [-1, -1]
1244 ],
1245 range: 1
1246 }
1247 ]
cc2c7183
BA
1248 },
1249 // Cannibal kings:
c9ab0340
BA
1250 '!': {"class": "king-pawn", moveas: "p"},
1251 '#': {"class": "king-rook", moveas: "r"},
1252 '$': {"class": "king-knight", moveas: "n"},
1253 '%': {"class": "king-bishop", moveas: "b"},
1254 '*': {"class": "king-queen", moveas: "q"}
41534b92
BA
1255 };
1256 }
1257
af9c9be3
BA
1258 // NOTE: using special symbols to not interfere with variants' pieces codes
1259 static get CannibalKings() {
1260 return {
1261 "!": "p",
1262 "#": "r",
1263 "$": "n",
1264 "%": "b",
1265 "*": "q",
1266 "k": "k"
1267 };
1268 }
1269
1270 static get CannibalKingCode() {
1271 return {
1272 "p": "!",
1273 "r": "#",
1274 "n": "$",
1275 "b": "%",
1276 "q": "*",
1277 "k": "k"
1278 };
1279 }
1280
1281 //////////////////////////
1282 // MOVES GENERATION UTILS
41534b92 1283
adf7c659
BA
1284 // For Cylinder: get Y coordinate
1285 getY(y) {
b4ae3ff6
BA
1286 if (!this.options["cylinder"])
1287 return y;
41534b92 1288 let res = y % this.size.y;
b4ae3ff6 1289 if (res < 0)
adf7c659 1290 res += this.size.y;
41534b92
BA
1291 return res;
1292 }
1293
ca8a3993
BA
1294 getSegments(curSeg, segStart, segEnd) {
1295 if (curSeg.length == 0)
1296 return undefined;
1297 let segments = JSON.parse(JSON.stringify(curSeg)); //not altering
1298 segments.push([[segStart[0], segStart[1]], [segEnd[0], segEnd[1]]]);
1299 return segments;
1300 }
1301
6b9320bb 1302 getStepSpec(color, x, y, piece) {
0e466aac 1303 let pieceType = piece;
33b42748 1304 let allSpecs = this.pieces(color, x, y);
0e466aac
BA
1305 if (!piece)
1306 pieceType = this.getPieceType(x, y);
1307 else if (allSpecs[piece].moveas)
1308 pieceType = allSpecs[piece].moveas;
33b42748
BA
1309 let res = allSpecs[pieceType];
1310 if (!res["both"])
1311 res.both = [];
1312 if (!res["moves"])
1313 res.moves = [];
1314 if (!res["attack"])
1315 res.attack = [];
1316 return res;
ca8a3993
BA
1317 }
1318
af9c9be3
BA
1319 // Can thing on square1 capture thing on square2?
1320 canTake([x1, y1], [x2, y2]) {
1321 return this.getColor(x1, y1) !== this.getColor(x2, y2);
1322 }
1323
de1421be 1324 // Teleport & Recycle. Assumption: color(x1,y1) == color(x2,y2)
727f2e55 1325 canSelfTake([x1, y1], [x2, y2]) {
de1421be 1326 return !this.isKing(x2, y2);
727f2e55
BA
1327 }
1328
af9c9be3
BA
1329 canStepOver(i, j, p) {
1330 // In some variants, objects on boards don't stop movement (Chakart)
1331 return this.board[i][j] == "";
1332 }
1333
ca8a3993
BA
1334 canDrop([c, p], [i, j]) {
1335 return (
1336 this.board[i][j] == "" &&
1337 (!this.enlightened || this.enlightened[i][j]) &&
1338 (
1339 p != "p" ||
1340 (c == 'w' && i < this.size.x - 1) ||
1341 (c == 'b' && i > 0)
1342 )
1343 );
1344 }
1345
af9c9be3
BA
1346 // For Madrasi:
1347 // (redefined in Baroque etc, where Madrasi condition doesn't make sense)
1348 isImmobilized([x, y]) {
1349 if (!this.options["madrasi"])
1350 return false;
1351 const color = this.getColor(x, y);
ceeac4e8 1352 const oppCols = this.getOppCols(color);
6b9320bb
BA
1353 const piece = this.getPieceType(x, y);
1354 const stepSpec = this.getStepSpec(color, x, y, piece);
33b42748 1355 const attacks = stepSpec.both.concat(stepSpec.attack);
af9c9be3
BA
1356 for (let a of attacks) {
1357 outerLoop: for (let step of a.steps) {
1358 let [i, j] = [x + step[0], y + step[1]];
1359 let stepCounter = 1;
1360 while (this.onBoard(i, j) && this.board[i][j] == "") {
1361 if (a.range <= stepCounter++)
1362 continue outerLoop;
1363 i += step[0];
1364 j = this.getY(j + step[1]);
1365 }
1366 if (
1367 this.onBoard(i, j) &&
ceeac4e8 1368 oppCols.includes(this.getColor(i, j)) &&
af9c9be3
BA
1369 this.getPieceType(i, j) == piece
1370 ) {
1371 return true;
1372 }
1373 }
1374 }
1375 return false;
1376 }
1377
41534b92
BA
1378 // Stop at the first capture found
1379 atLeastOneCapture(color) {
6b9320bb
BA
1380 const allowed = (sq1, sq2) => {
1381 return (
1382 // NOTE: canTake is reversed for Zen.
1383 // Generally ok because of the symmetry. TODO?
1384 this.canTake(sq1, sq2) &&
1385 this.filterValid(
1386 [this.getBasicMove(sq1, sq2)]).length >= 1
1387 );
af9c9be3
BA
1388 };
1389 for (let i=0; i<this.size.x; i++) {
1390 for (let j=0; j<this.size.y; j++) {
1391 if (this.getColor(i, j) == color) {
1392 if (
6b9320bb
BA
1393 (
1394 !this.options["zen"] &&
1395 this.findDestSquares(
1396 [i, j],
1397 {
1398 attackOnly: true,
1399 one: true,
1400 segments: this.options["cylinder"]
1401 },
1402 allowed
1403 )
1404 )
1405 ||
1406 (
1407 (
1408 this.options["zen"] &&
1409 this.findCapturesOn(
1410 [i, j],
1411 {
1412 one: true,
1413 segments: this.options["cylinder"]
1414 },
1415 allowed
1416 )
1417 )
1418 )
af9c9be3
BA
1419 ) {
1420 return true;
41534b92
BA
1421 }
1422 }
1423 }
1424 }
1425 return false;
1426 }
1427
af9c9be3 1428 compatibleStep([x1, y1], [x2, y2], step, range) {
6b9320bb 1429 const epsilon = 1e-7; //arbitrary small value
af9c9be3
BA
1430 let shifts = [0];
1431 if (this.options["cylinder"])
1432 Array.prototype.push.apply(shifts, [-this.size.y, this.size.y]);
1433 for (let sh of shifts) {
1434 const rx = (x2 - x1) / step[0],
1435 ry = (y2 + sh - y1) / step[1];
1436 if (
6b9320bb 1437 // Zero step but non-zero interval => impossible
af9c9be3 1438 (!Number.isFinite(rx) && !Number.isNaN(rx)) ||
6b9320bb
BA
1439 (!Number.isFinite(ry) && !Number.isNaN(ry)) ||
1440 // Negative number of step (impossible)
1441 (rx < 0 || ry < 0) ||
1442 // Not the same number of steps in both directions:
1443 (!Number.isNaN(rx) && !Number.isNaN(ry) && Math.abs(rx - ry) > epsilon)
af9c9be3
BA
1444 ) {
1445 continue;
1446 }
1447 let distance = (Number.isNaN(rx) ? ry : rx);
6b9320bb 1448 if (Math.abs(distance - Math.round(distance)) > epsilon)
af9c9be3
BA
1449 continue;
1450 distance = Math.round(distance); //in case of (numerical...)
6b9320bb 1451 if (!range || range >= distance)
af9c9be3
BA
1452 return true;
1453 }
1454 return false;
1455 }
1456
1457 ////////////////////
1458 // MOVES GENERATION
1459
41534b92
BA
1460 getDropMovesFrom([c, p]) {
1461 // NOTE: by design, this.reserve[c][p] >= 1 on user click
1a7c0492 1462 // (but not necessarily otherwise: atLeastOneMove() etc)
b4ae3ff6
BA
1463 if (this.reserve[c][p] == 0)
1464 return [];
41534b92
BA
1465 let moves = [];
1466 for (let i=0; i<this.size.x; i++) {
1467 for (let j=0; j<this.size.y; j++) {
9b760538 1468 if (this.onBoard(i, j) && this.canDrop([c, p], [i, j])) {
ca8a3993
BA
1469 let mv = new Move({
1470 start: {x: c, y: p},
1471 end: {x: i, y: j},
1472 appear: [new PiPo({x: i, y: j, c: c, p: p})],
1473 vanish: []
1474 });
1475 if (this.board[i][j] != "") {
1476 mv.vanish.push(new PiPo({
1477 x: i,
1478 y: j,
1479 c: this.getColor(i, j),
1480 p: this.getPiece(i, j)
1481 }));
1482 }
1483 moves.push(mv);
41534b92
BA
1484 }
1485 }
1486 }
1487 return moves;
1488 }
1489
1490 // All possible moves from selected square
de1421be 1491 // TODO: generalize usage if arg "color" (e.g. Checkered)
ca8a3993 1492 getPotentialMovesFrom([x, y], color) {
8b301184
BA
1493 if (this.subTurnTeleport == 2)
1494 return [];
ca8a3993
BA
1495 if (typeof x == "string")
1496 return this.getDropMovesFrom([x, y]);
1497 if (this.isImmobilized([x, y]))
b4ae3ff6 1498 return [];
ca8a3993
BA
1499 const piece = this.getPieceType(x, y);
1500 let moves = this.getPotentialMovesOf(piece, [x, y]);
1501 if (piece == "p" && this.hasEnpassant && this.epSquare)
1502 Array.prototype.push.apply(moves, this.getEnpassantCaptures([x, y]));
41534b92 1503 if (
a548cb4e 1504 this.isKing(0, 0, piece) && this.hasCastle &&
41534b92
BA
1505 this.castleFlags[color || this.turn].some(v => v < this.size.y)
1506 ) {
ca8a3993 1507 Array.prototype.push.apply(moves, this.getCastleMoves([x, y]));
41534b92
BA
1508 }
1509 return this.postProcessPotentialMoves(moves);
1510 }
1511
1512 postProcessPotentialMoves(moves) {
b4ae3ff6
BA
1513 if (moves.length == 0)
1514 return [];
41534b92 1515 const color = this.getColor(moves[0].start.x, moves[0].start.y);
ceeac4e8 1516 const oppCols = this.getOppCols(color);
41534b92 1517
6b9320bb 1518 if (this.options["capture"] && this.atLeastOneCapture(color))
ceeac4e8 1519 moves = this.capturePostProcess(moves, oppCols);
41534b92 1520
57b8015b 1521 if (this.options["atomic"])
ceeac4e8 1522 moves = this.atomicPostProcess(moves, color, oppCols);
cc2c7183 1523
c9ab0340
BA
1524 if (
1525 moves.length > 0 &&
1526 this.getPieceType(moves[0].start.x, moves[0].start.y) == "p"
1527 ) {
ceeac4e8 1528 moves = this.pawnPostProcess(moves, color, oppCols);
c9ab0340
BA
1529 }
1530
6b9320bb 1531 if (this.options["cannibal"] && this.options["rifle"])
cc2c7183 1532 // In this case a rifle-capture from last rank may promote a pawn
98d14451 1533 moves = this.riflePromotePostProcess(moves, color);
57b8015b
BA
1534
1535 return moves;
1536 }
1537
ceeac4e8 1538 capturePostProcess(moves, oppCols) {
57b8015b
BA
1539 // Filter out non-capturing moves (not using m.vanish because of
1540 // self captures of Recycle and Teleport).
1541 return moves.filter(m => {
1542 return (
1543 this.board[m.end.x][m.end.y] != "" &&
ceeac4e8 1544 oppCols.includes(this.getColor(m.end.x, m.end.y))
57b8015b
BA
1545 );
1546 });
1547 }
1548
ceeac4e8 1549 atomicPostProcess(moves, color, oppCols) {
57b8015b
BA
1550 moves.forEach(m => {
1551 if (
1552 this.board[m.end.x][m.end.y] != "" &&
ceeac4e8 1553 oppCols.includes(this.getColor(m.end.x, m.end.y))
57b8015b
BA
1554 ) {
1555 // Explosion!
1556 let steps = [
1557 [-1, -1],
1558 [-1, 0],
1559 [-1, 1],
1560 [0, -1],
1561 [0, 1],
1562 [1, -1],
1563 [1, 0],
1564 [1, 1]
1565 ];
e7d409fc
BA
1566 let mNext = new Move({
1567 start: m.end,
1568 end: m.end,
1569 appear: [],
1570 vanish: []
1571 });
57b8015b
BA
1572 for (let step of steps) {
1573 let x = m.end.x + step[0];
d262cff4 1574 let y = this.getY(m.end.y + step[1]);
57b8015b
BA
1575 if (
1576 this.onBoard(x, y) &&
1577 this.board[x][y] != "" &&
e7d409fc 1578 (x != m.start.x || y != m.start.y) &&
57b8015b
BA
1579 this.getPieceType(x, y) != "p"
1580 ) {
e7d409fc 1581 mNext.vanish.push(
57b8015b
BA
1582 new PiPo({
1583 p: this.getPiece(x, y),
1584 c: this.getColor(x, y),
1585 x: x,
1586 y: y
1587 })
1588 );
1589 }
1590 }
e7d409fc
BA
1591 if (!this.options["rifle"]) {
1592 // The moving piece also vanish
1593 mNext.vanish.unshift(
1594 new PiPo({
1595 x: m.end.x,
1596 y: m.end.y,
1597 c: color,
1598 p: this.getPiece(m.start.x, m.start.y)
1599 })
1600 );
1601 }
1602 m.next = mNext;
57b8015b
BA
1603 }
1604 });
98d14451 1605 return moves;
57b8015b
BA
1606 }
1607
ceeac4e8 1608 pawnPostProcess(moves, color, oppCols) {
57b8015b
BA
1609 let moreMoves = [];
1610 const lastRank = (color == "w" ? 0 : this.size.x - 1);
1611 const initPiece = this.getPiece(moves[0].start.x, moves[0].start.y);
1612 moves.forEach(m => {
57b8015b
BA
1613 const [x1, y1] = [m.start.x, m.start.y];
1614 const [x2, y2] = [m.end.x, m.end.y];
1615 const promotionOk = (
1616 x2 == lastRank &&
1617 (!this.options["rifle"] || this.board[x2][y2] == "")
1618 );
1619 if (!promotionOk)
1620 return; //nothing to do
8cc2f6d0
BA
1621 if (this.options["pawnfall"]) {
1622 m.appear.shift();
8cc2f6d0
BA
1623 return;
1624 }
9ac67217 1625 let finalPieces;
99ea2453
BA
1626 if (
1627 this.options["cannibal"] &&
1628 this.board[x2][y2] != "" &&
ceeac4e8 1629 oppCols.includes(this.getColor(x2, y2))
99ea2453
BA
1630 ) {
1631 finalPieces = [this.getPieceType(x2, y2)];
1632 }
1633 else
1634 finalPieces = this.pawnPromotions;
57b8015b
BA
1635 m.appear[0].p = finalPieces[0];
1636 if (initPiece == "!") //cannibal king-pawn
1637 m.appear[0].p = C.CannibalKingCode[finalPieces[0]];
1638 for (let i=1; i<finalPieces.length; i++) {
9ac67217 1639 let newMove = JSON.parse(JSON.stringify(m));
57b8015b 1640 const piece = finalPieces[i];
9ac67217 1641 m.appear[0].p = (initPiece != "!" ? piece : C.CannibalKingCode[piece]);
57b8015b
BA
1642 moreMoves.push(newMove);
1643 }
1644 });
98d14451 1645 return moves.concat(moreMoves);
57b8015b 1646 }
cc2c7183 1647
9db5050a 1648 riflePromotePostProcess(moves, color) {
57b8015b
BA
1649 const lastRank = (color == "w" ? 0 : this.size.x - 1);
1650 let newMoves = [];
1651 moves.forEach(m => {
1652 if (
1653 m.start.x == lastRank &&
1654 m.appear.length >= 1 &&
1655 m.appear[0].p == "p" &&
1656 m.appear[0].x == m.start.x &&
1657 m.appear[0].y == m.start.y
1658 ) {
57b8015b
BA
1659 m.appear[0].p = this.pawnPromotions[0];
1660 for (let i=1; i<this.pawnPromotions.length; i++) {
1661 let newMv = JSON.parse(JSON.stringify(m));
1662 newMv.appear[0].p = this.pawnSpecs.promotions[i];
1663 newMoves.push(newMv);
1664 }
1665 }
1666 });
98d14451 1667 return moves.concat(newMoves);
41534b92
BA
1668 }
1669
af9c9be3 1670 // Generic method to find possible moves of "sliding or jumping" pieces
6b9320bb 1671 getPotentialMovesOf(piece, [x, y]) {
41534b92 1672 const color = this.getColor(x, y);
6b9320bb 1673 const stepSpec = this.getStepSpec(color, x, y, piece);
af9c9be3 1674 let squares = [];
6b9320bb 1675 if (stepSpec.attack) {
af9c9be3
BA
1676 squares = this.findDestSquares(
1677 [x, y],
1678 {
1679 attackOnly: true,
6b9320bb
BA
1680 segments: this.options["cylinder"],
1681 stepSpec: stepSpec
af9c9be3 1682 },
6b9320bb 1683 ([i1, j1], [i2, j2]) => {
af9c9be3 1684 return (
6b9320bb
BA
1685 (!this.options["zen"] || this.isKing(i2, j2)) &&
1686 this.canTake([i1, j1], [i2, j2])
af9c9be3 1687 );
c9ab0340 1688 }
af9c9be3
BA
1689 );
1690 }
1691 const noSpecials = this.findDestSquares(
1692 [x, y],
1693 {
6b9320bb
BA
1694 moveOnly: !!stepSpec.attack || this.options["zen"],
1695 segments: this.options["cylinder"],
1696 stepSpec: stepSpec
1697 }
af9c9be3
BA
1698 );
1699 Array.prototype.push.apply(squares, noSpecials);
1700 if (this.options["zen"]) {
1701 let zenCaptures = this.findCapturesOn(
1702 [x, y],
6b9320bb
BA
1703 {}, //byCol: default is ok
1704 ([i1, j1], [i2, j2]) =>
1705 !this.isKing(i1, j1) && this.canTake([i2, j2], [i1, j1])
af9c9be3
BA
1706 );
1707 // Technical step: segments (if any) are reversed
1708 if (this.options["cylinder"]) {
1709 zenCaptures.forEach(z => {
ca8a3993 1710 z.segments = z.segments.reverse().map(s => s.reverse())
af9c9be3 1711 });
41534b92 1712 }
af9c9be3 1713 Array.prototype.push.apply(squares, zenCaptures);
41534b92 1714 }
af9c9be3
BA
1715 if (
1716 this.options["recycle"] ||
1717 (this.options["teleport"] && this.subTurnTeleport == 1)
1718 ) {
1719 const selfCaptures = this.findDestSquares(
1720 [x, y],
1721 {
1722 attackOnly: true,
6b9320bb
BA
1723 segments: this.options["cylinder"],
1724 stepSpec: stepSpec
af9c9be3 1725 },
de1421be
BA
1726 ([i1, j1], [i2, j2]) => {
1727 return (
1728 this.getColor(i2, j2) == color &&
1729 this.canSelfTake([i1, j1], [i2, j2])
1730 );
1731 }
af9c9be3
BA
1732 );
1733 Array.prototype.push.apply(squares, selfCaptures);
1734 }
1735 return squares.map(s => {
1736 let mv = this.getBasicMove([x, y], s.sq);
727f2e55 1737 if (this.options["cylinder"] && !!s.segments && s.segments.length >= 2)
af9c9be3
BA
1738 mv.segments = s.segments;
1739 return mv;
1740 });
3b641716
BA
1741 }
1742
af9c9be3
BA
1743 findDestSquares([x, y], o, allowed) {
1744 if (!allowed)
6b9320bb 1745 allowed = (sq1, sq2) => this.canTake(sq1, sq2);
65cf1690 1746 const apparentPiece = this.getPiece(x, y); //how it looks
af9c9be3
BA
1747 let res = [];
1748 // Next 3 for Cylinder mode: (unused if !o.segments)
adf7c659
BA
1749 let explored = {};
1750 let segments = [];
1751 let segStart = [];
af9c9be3
BA
1752 const addSquare = ([i, j]) => {
1753 let elt = {sq: [i, j]};
1754 if (o.segments)
727f2e55 1755 elt.segments = this.getSegments(segments, segStart, [i, j]);
af9c9be3 1756 res.push(elt);
adf7c659 1757 };
33b42748 1758 const exploreSteps = (stepArray, mode) => {
c9ab0340
BA
1759 for (let s of stepArray) {
1760 outerLoop: for (let step of s.steps) {
af9c9be3
BA
1761 if (o.segments) {
1762 segments = [];
1763 segStart = [x, y];
1764 }
d262cff4
BA
1765 let [i, j] = [x, y];
1766 let stepCounter = 0;
1767 while (
1768 this.onBoard(i, j) &&
65cf1690 1769 ((i == x && j == y) || this.canStepOver(i, j, apparentPiece))
d262cff4 1770 ) {
6b9320bb 1771 if (!explored[i + "." + j] && (i != x || j != y)) {
c9ab0340 1772 explored[i + "." + j] = true;
af9c9be3 1773 if (
6b9320bb
BA
1774 !o.captureTarget ||
1775 (o.captureTarget[0] == i && o.captureTarget[1] == j)
af9c9be3 1776 ) {
33b42748 1777 if (o.one && mode != "attack")
af9c9be3 1778 return true;
33b42748 1779 if (mode != "attack")
af9c9be3
BA
1780 addSquare(!o.captureTarget ? [i, j] : [x, y]);
1781 if (o.captureTarget)
1782 return res[0];
1783 }
c9ab0340
BA
1784 }
1785 if (s.range <= stepCounter++)
1786 continue outerLoop;
d262cff4 1787 const oldIJ = [i, j];
c9ab0340 1788 i += step[0];
adf7c659 1789 j = this.getY(j + step[1]);
af9c9be3 1790 if (o.segments && Math.abs(j - oldIJ[1]) > 1) {
d262cff4 1791 // Boundary between segments (cylinder mode)
adf7c659
BA
1792 segments.push([[segStart[0], segStart[1]], oldIJ]);
1793 segStart = [i, j];
d262cff4 1794 }
c9ab0340
BA
1795 }
1796 if (!this.onBoard(i, j))
1797 continue;
1798 const pieceIJ = this.getPieceType(i, j);
af9c9be3 1799 if (!explored[i + "." + j]) {
c9ab0340 1800 explored[i + "." + j] = true;
6b9320bb 1801 if (allowed([x, y], [i, j])) {
33b42748 1802 if (o.one && mode != "moves")
af9c9be3 1803 return true;
33b42748 1804 if (mode != "moves")
af9c9be3
BA
1805 addSquare(!o.captureTarget ? [i, j] : [x, y]);
1806 if (
1807 o.captureTarget &&
1808 o.captureTarget[0] == i && o.captureTarget[1] == j
1809 ) {
1810 return res[0];
1811 }
1812 }
c9ab0340
BA
1813 }
1814 }
41534b92 1815 }
6b9320bb 1816 return undefined; //default, but let's explicit it
c9ab0340 1817 };
af9c9be3 1818 if (o.captureTarget)
33b42748 1819 return exploreSteps(o.captureSteps, "attack");
af9c9be3 1820 else {
6b9320bb
BA
1821 const stepSpec =
1822 o.stepSpec || this.getStepSpec(this.getColor(x, y), x, y);
1823 let outOne = false;
33b42748
BA
1824 if (!o.attackOnly)
1825 outOne = exploreSteps(stepSpec.both.concat(stepSpec.moves), "moves");
1826 if (!outOne && !o.moveOnly)
1827 outOne = exploreSteps(stepSpec.both.concat(stepSpec.attack), "attack");
6b9320bb 1828 return (o.one ? outOne : res);
082e639a 1829 }
41534b92
BA
1830 }
1831
082e639a 1832 // Search for enemy (or not) pieces attacking [x, y]
af9c9be3 1833 findCapturesOn([x, y], o, allowed) {
af9c9be3 1834 if (!o.byCol)
ceeac4e8 1835 o.byCol = this.getOppCols(this.getColor(x, y) || this.turn);
ca8a3993 1836 let res = [];
c9ab0340
BA
1837 for (let i=0; i<this.size.x; i++) {
1838 for (let j=0; j<this.size.y; j++) {
af9c9be3 1839 const colIJ = this.getColor(i, j);
57b8015b
BA
1840 if (
1841 this.board[i][j] != "" &&
af9c9be3 1842 o.byCol.includes(colIJ) &&
57b8015b
BA
1843 !this.isImmobilized([i, j])
1844 ) {
65cf1690
BA
1845 const apparentPiece = this.getPiece(i, j);
1846 // Quick check: does this potential attacker target x,y ?
1847 if (this.canStepOver(x, y, apparentPiece))
1848 continue;
af9c9be3 1849 const stepSpec = this.getStepSpec(colIJ, i, j);
33b42748 1850 const attacks = stepSpec.attack.concat(stepSpec.both);
c9ab0340
BA
1851 for (let a of attacks) {
1852 for (let s of a.steps) {
1853 // Quick check: if step isn't compatible, don't even try
af9c9be3 1854 if (!this.compatibleStep([i, j], [x, y], s, a.range))
c9ab0340
BA
1855 continue;
1856 // Finally verify that nothing stand in-between
6b9320bb 1857 const out = this.findDestSquares(
af9c9be3
BA
1858 [i, j],
1859 {
1860 captureTarget: [x, y],
1861 captureSteps: [{steps: [s], range: a.range}],
b57b726c 1862 segments: o.segments
af9c9be3 1863 },
6b9320bb 1864 allowed
af9c9be3 1865 );
6b9320bb 1866 if (out) {
af9c9be3
BA
1867 if (o.one)
1868 return true;
6b9320bb 1869 res.push(out);
c9ab0340
BA
1870 }
1871 }
1872 }
41534b92 1873 }
c9ab0340
BA
1874 }
1875 }
6b9320bb 1876 return (o.one ? false : res);
57b8015b
BA
1877 }
1878
41534b92
BA
1879 // Build a regular move from its initial and destination squares.
1880 // tr: transformation
1881 getBasicMove([sx, sy], [ex, ey], tr) {
1882 const initColor = this.getColor(sx, sy);
cc2c7183 1883 const initPiece = this.getPiece(sx, sy);
41534b92
BA
1884 const destColor = (this.board[ex][ey] != "" ? this.getColor(ex, ey) : "");
1885 let mv = new Move({
1886 appear: [],
1887 vanish: [],
15106e82
BA
1888 start: {x: sx, y: sy},
1889 end: {x: ex, y: ey}
41534b92
BA
1890 });
1891 if (
1892 !this.options["rifle"] ||
1893 this.board[ex][ey] == "" ||
1894 destColor == initColor //Recycle, Teleport
1895 ) {
1896 mv.appear = [
1897 new PiPo({
1898 x: ex,
1899 y: ey,
1900 c: !!tr ? tr.c : initColor,
1901 p: !!tr ? tr.p : initPiece
1902 })
1903 ];
1904 mv.vanish = [
1905 new PiPo({
1906 x: sx,
1907 y: sy,
1908 c: initColor,
1909 p: initPiece
1910 })
1911 ];
1912 }
1913 if (this.board[ex][ey] != "") {
1914 mv.vanish.push(
1915 new PiPo({
1916 x: ex,
1917 y: ey,
1918 c: this.getColor(ex, ey),
cc2c7183 1919 p: this.getPiece(ex, ey)
41534b92
BA
1920 })
1921 );
41534b92 1922 if (this.options["cannibal"] && destColor != initColor) {
6b9320bb 1923 const lastIdx = mv.vanish.length - 1; //think "Rifle+Cannibal"
cc2c7183 1924 let trPiece = mv.vanish[lastIdx].p;
6b9320bb 1925 if (this.isKing(sx, sy))
cc2c7183 1926 trPiece = C.CannibalKingCode[trPiece];
b4ae3ff6
BA
1927 if (mv.appear.length >= 1)
1928 mv.appear[0].p = trPiece;
41534b92
BA
1929 else if (this.options["rifle"]) {
1930 mv.appear.unshift(
1931 new PiPo({
1932 x: sx,
1933 y: sy,
1934 c: initColor,
cc2c7183 1935 p: trPiece
41534b92
BA
1936 })
1937 );
1938 mv.vanish.unshift(
1939 new PiPo({
1940 x: sx,
1941 y: sy,
1942 c: initColor,
1943 p: initPiece
1944 })
1945 );
1946 }
1947 }
1948 }
1949 return mv;
1950 }
1951
1952 // En-passant square, if any
1953 getEpSquare(moveOrSquare) {
1954 if (typeof moveOrSquare === "string") {
1955 const square = moveOrSquare;
b4ae3ff6
BA
1956 if (square == "-")
1957 return undefined;
cc2c7183 1958 return C.SquareToCoords(square);
41534b92
BA
1959 }
1960 // Argument is a move:
1961 const move = moveOrSquare;
1962 const s = move.start,
1963 e = move.end;
1964 if (
1965 s.y == e.y &&
1966 Math.abs(s.x - e.x) == 2 &&
1967 // Next conditions for variants like Atomic or Rifle, Recycle...
65cf1690
BA
1968 (
1969 move.appear.length > 0 &&
ca8a3993 1970 this.getPieceType(0, 0, move.appear[0].p) == 'p'
65cf1690
BA
1971 )
1972 &&
1973 (
1974 move.vanish.length > 0 &&
ca8a3993 1975 this.getPieceType(0, 0, move.vanish[0].p) == 'p'
65cf1690 1976 )
41534b92
BA
1977 ) {
1978 return {
1979 x: (s.x + e.x) / 2,
1980 y: s.y
1981 };
1982 }
1983 return undefined; //default
1984 }
1985
1986 // Special case of en-passant captures: treated separately
c9ab0340 1987 getEnpassantCaptures([x, y]) {
41534b92 1988 const color = this.getColor(x, y);
c9ab0340 1989 const shiftX = (color == 'w' ? -1 : 1);
ceeac4e8 1990 const oppCols = this.getOppCols(color);
41534b92 1991 if (
ca8a3993 1992 this.epSquare &&
41534b92 1993 this.epSquare.x == x + shiftX &&
d262cff4 1994 Math.abs(this.getY(this.epSquare.y - y)) == 1 &&
ca8a3993
BA
1995 // Doublemove (and Progressive?) guards:
1996 this.board[this.epSquare.x][this.epSquare.y] == "" &&
ceeac4e8 1997 oppCols.includes(this.getColor(x, this.epSquare.y))
41534b92
BA
1998 ) {
1999 const [epx, epy] = [this.epSquare.x, this.epSquare.y];
b57b726c 2000 this.board[epx][epy] = this.board[x][this.epSquare.y];
ca8a3993 2001 let enpassantMove = this.getBasicMove([x, y], [epx, epy]);
41534b92
BA
2002 this.board[epx][epy] = "";
2003 const lastIdx = enpassantMove.vanish.length - 1; //think Rifle
2004 enpassantMove.vanish[lastIdx].x = x;
ca8a3993 2005 return [enpassantMove];
41534b92 2006 }
ca8a3993 2007 return [];
41534b92
BA
2008 }
2009
ca8a3993 2010 getCastleMoves([x, y], finalSquares, castleWith) {
41534b92
BA
2011 const c = this.getColor(x, y);
2012
2013 // Castling ?
ceeac4e8 2014 const oppCols = this.getOppCols(c);
41534b92
BA
2015 let moves = [];
2016 // King, then rook:
2017 finalSquares =
2018 finalSquares || [ [2, 3], [this.size.y - 2, this.size.y - 3] ];
cc2c7183 2019 const castlingKing = this.getPiece(x, y);
41534b92
BA
2020 castlingCheck: for (
2021 let castleSide = 0;
2022 castleSide < 2;
2023 castleSide++ //large, then small
2024 ) {
b4ae3ff6
BA
2025 if (this.castleFlags[c][castleSide] >= this.size.y)
2026 continue;
41534b92
BA
2027 // If this code is reached, rook and king are on initial position
2028
2029 // NOTE: in some variants this is not a rook
2030 const rookPos = this.castleFlags[c][castleSide];
cc2c7183 2031 const castlingPiece = this.getPiece(x, rookPos);
41534b92
BA
2032 if (
2033 this.board[x][rookPos] == "" ||
2034 this.getColor(x, rookPos) != c ||
ca8a3993 2035 (castleWith && !castleWith.includes(castlingPiece))
41534b92
BA
2036 ) {
2037 // Rook is not here, or changed color (see Benedict)
2038 continue;
2039 }
2040 // Nothing on the path of the king ? (and no checks)
2041 const finDist = finalSquares[castleSide][0] - y;
2042 let step = finDist / Math.max(1, Math.abs(finDist));
2043 let i = y;
2044 do {
2045 if (
ca8a3993
BA
2046 // NOTE: next weird test because underCheck() verification
2047 // will be executed in filterValid() later.
2048 (
2049 i != finalSquares[castleSide][0] &&
ceeac4e8 2050 this.underCheck([[x, i]], oppCols)
ca8a3993
BA
2051 )
2052 ||
41534b92
BA
2053 (
2054 this.board[x][i] != "" &&
2055 // NOTE: next check is enough, because of chessboard constraints
2056 (this.getColor(x, i) != c || ![rookPos, y].includes(i))
2057 )
2058 ) {
2059 continue castlingCheck;
2060 }
2061 i += step;
2062 } while (i != finalSquares[castleSide][0]);
2063 // Nothing on the path to the rook?
2064 step = (castleSide == 0 ? -1 : 1);
2065 for (i = y + step; i != rookPos; i += step) {
b4ae3ff6
BA
2066 if (this.board[x][i] != "")
2067 continue castlingCheck;
41534b92
BA
2068 }
2069
2070 // Nothing on final squares, except maybe king and castling rook?
2071 for (i = 0; i < 2; i++) {
2072 if (
2073 finalSquares[castleSide][i] != rookPos &&
2074 this.board[x][finalSquares[castleSide][i]] != "" &&
2075 (
2076 finalSquares[castleSide][i] != y ||
2077 this.getColor(x, finalSquares[castleSide][i]) != c
2078 )
2079 ) {
2080 continue castlingCheck;
2081 }
2082 }
2083
ca8a3993 2084 // If this code is reached, castle is potentially valid
41534b92
BA
2085 moves.push(
2086 new Move({
2087 appear: [
2088 new PiPo({
2089 x: x,
2090 y: finalSquares[castleSide][0],
2091 p: castlingKing,
2092 c: c
2093 }),
2094 new PiPo({
2095 x: x,
2096 y: finalSquares[castleSide][1],
2097 p: castlingPiece,
2098 c: c
2099 })
2100 ],
2101 vanish: [
2102 // King might be initially disguised (Titan...)
2103 new PiPo({ x: x, y: y, p: castlingKing, c: c }),
2104 new PiPo({ x: x, y: rookPos, p: castlingPiece, c: c })
2105 ],
2106 end:
2107 Math.abs(y - rookPos) <= 2
c9ab0340
BA
2108 ? {x: x, y: rookPos}
2109 : {x: x, y: y + 2 * (castleSide == 0 ? -1 : 1)}
41534b92
BA
2110 })
2111 );
2112 }
2113
2114 return moves;
2115 }
2116
2117 ////////////////////
2118 // MOVES VALIDATION
2119
5006aaca 2120 // Is piece (or square) at given position attacked by "oppCol(s)" ?
ceeac4e8 2121 underAttack([x, y], oppCols) {
6b9320bb
BA
2122 // An empty square is considered as king,
2123 // since it's used only in getCastleMoves (TODO?)
2124 const king = this.board[x][y] == "" || this.isKing(x, y);
af9c9be3
BA
2125 return (
2126 (
2127 (!this.options["zen"] || king) &&
6b9320bb
BA
2128 this.findCapturesOn(
2129 [x, y],
2130 {
ceeac4e8 2131 byCol: oppCols,
6b9320bb
BA
2132 segments: this.options["cylinder"],
2133 one: true
2134 }
2135 )
af9c9be3
BA
2136 )
2137 ||
2138 (
6b9320bb
BA
2139 (!!this.options["zen"] && !king) &&
2140 this.findDestSquares(
2141 [x, y],
2142 {
2143 attackOnly: true,
2144 segments: this.options["cylinder"],
2145 one: true
2146 },
ceeac4e8 2147 ([i1, j1], [i2, j2]) => oppCols.includes(this.getColor(i2, j2))
6b9320bb 2148 )
af9c9be3
BA
2149 )
2150 );
2151 }
2152
f3e90e30 2153 // Argument is (very generally) an array of squares (= arrays)
ceeac4e8 2154 underCheck(square_s, oppCols) {
b4ae3ff6
BA
2155 if (this.options["taking"] || this.options["dark"])
2156 return false;
ceeac4e8 2157 return square_s.some(sq => this.underAttack(sq, oppCols));
41534b92
BA
2158 }
2159
a548cb4e 2160 // Scan board for king(s)
41534b92 2161 searchKingPos(color) {
a548cb4e 2162 let res = [];
41534b92
BA
2163 for (let i=0; i < this.size.x; i++) {
2164 for (let j=0; j < this.size.y; j++) {
6b9320bb 2165 if (this.getColor(i, j) == color && this.isKing(i, j))
a548cb4e 2166 res.push([i, j]);
41534b92
BA
2167 }
2168 }
a548cb4e 2169 return res;
41534b92
BA
2170 }
2171
de1421be
BA
2172 // cb: callback returning a boolean (false if king missing)
2173 trackKingWrap(move, kingPos, cb) {
f57232cd
BA
2174 if (move.appear.length == 0 && move.vanish.length == 0)
2175 return true;
2176 const color =
2177 (move.vanish.length > 0 ? move.vanish[0].c : move.appear[0].c);
de1421be
BA
2178 let newKingPP = null,
2179 sqIdx = 0,
2180 res = true; //a priori valid
2181 const oldKingPP =
2182 move.vanish.find(v => this.isKing(0, 0, v.p) && v.c == color);
2183 if (oldKingPP) {
2184 // Search king in appear array:
2185 newKingPP =
2186 move.appear.find(a => this.isKing(0, 0, a.p) && a.c == color);
2187 if (newKingPP) {
2188 sqIdx = kingPos.findIndex(kp =>
2189 kp[0] == oldKingPP.x && kp[1] == oldKingPP.y);
2190 kingPos[sqIdx] = [newKingPP.x, newKingPP.y];
2191 }
2192 else
2193 res = false; //king vanished
2194 }
2195 res &&= cb(kingPos);
2196 if (oldKingPP && newKingPP)
2197 kingPos[sqIdx] = [oldKingPP.x, oldKingPP.y];
2198 return res;
2199 }
2200
af9c9be3 2201 // 'color' arg because some variants (e.g. Refusal) check opponent moves
f5435757 2202 filterValid(moves, color) {
f5435757
BA
2203 if (!color)
2204 color = this.turn;
ceeac4e8 2205 const oppCols = this.getOppCols(color);
a548cb4e 2206 let kingPos = this.searchKingPos(color);
41534b92 2207 return moves.filter(m => {
b57b726c 2208 this.playOnBoard(m);
de1421be
BA
2209 const res = this.trackKingWrap(m, kingPos, (kp) => {
2210 return !this.underCheck(kp, oppCols);
2211 });
b57b726c
BA
2212 this.undoOnBoard(m);
2213 return res;
41534b92
BA
2214 });
2215 }
2216
2217 /////////////////
2218 // MOVES PLAYING
2219
41534b92
BA
2220 // Apply a move on board
2221 playOnBoard(move) {
6997e386
BA
2222 for (let psq of move.vanish)
2223 this.board[psq.x][psq.y] = "";
2224 for (let psq of move.appear)
2225 this.board[psq.x][psq.y] = psq.c + psq.p;
41534b92
BA
2226 }
2227 // Un-apply the played move
2228 undoOnBoard(move) {
6997e386
BA
2229 for (let psq of move.appear)
2230 this.board[psq.x][psq.y] = "";
2231 for (let psq of move.vanish)
2232 this.board[psq.x][psq.y] = psq.c + psq.p;
41534b92
BA
2233 }
2234
2235 updateCastleFlags(move) {
2236 // Update castling flags if start or arrive from/at rook/king locations
2237 move.appear.concat(move.vanish).forEach(psq => {
6b9320bb 2238 if (this.isKing(0, 0, psq.p))
41534b92 2239 this.castleFlags[psq.c] = [this.size.y, this.size.y];
41534b92 2240 // NOTE: not "else if" because king can capture enemy rook...
cc2c7183 2241 let c = "";
b4ae3ff6
BA
2242 if (psq.x == 0)
2243 c = "b";
2244 else if (psq.x == this.size.x - 1)
2245 c = "w";
cc2c7183 2246 if (c != "") {
41534b92 2247 const fidx = this.castleFlags[c].findIndex(f => f == psq.y);
b4ae3ff6
BA
2248 if (fidx >= 0)
2249 this.castleFlags[c][fidx] = this.size.y;
41534b92
BA
2250 }
2251 });
2252 }
2253
2254 prePlay(move) {
2255 if (
99ea2453
BA
2256 this.hasCastle &&
2257 // If flags already off, no need to re-check:
ca8a3993
BA
2258 Object.values(this.castleFlags).some(cvals =>
2259 cvals.some(val => val < this.size.y))
41534b92 2260 ) {
99ea2453
BA
2261 this.updateCastleFlags(move);
2262 }
2263 if (this.options["crazyhouse"]) {
2264 move.vanish.forEach(v => {
2265 const square = C.CoordsToSquare({x: v.x, y: v.y});
2266 if (this.ispawn[square])
2267 delete this.ispawn[square];
2268 });
2269 if (move.appear.length > 0 && move.vanish.length > 0) {
2270 // Assumption: something is moving
2271 const initSquare = C.CoordsToSquare(move.start);
f429756d 2272 const destSquare = C.CoordsToSquare(move.end);
99ea2453
BA
2273 if (
2274 this.ispawn[initSquare] ||
ca8a3993 2275 (move.vanish[0].p == 'p' && move.appear[0].p != 'p')
41534b92 2276 ) {
f429756d
BA
2277 this.ispawn[destSquare] = true;
2278 }
2279 else if (
2280 this.ispawn[destSquare] &&
2281 this.getColor(move.end.x, move.end.y) != move.vanish[0].c
2282 ) {
ca8a3993 2283 move.vanish[1].p = 'p';
f429756d 2284 delete this.ispawn[destSquare];
41534b92
BA
2285 }
2286 }
2287 }
2288 const minSize = Math.min(move.appear.length, move.vanish.length);
0c44c676
BA
2289 if (
2290 this.hasReserve &&
2291 // Warning; atomic pawn removal isn't a capture
2292 (!this.options["atomic"] || !this.rempawn || this.movesCount >= 1)
2293 ) {
41534b92
BA
2294 const color = this.turn;
2295 for (let i=minSize; i<move.appear.length; i++) {
2296 // Something appears = dropped on board (some exceptions, Chakart...)
0c44c676
BA
2297 if (move.appear[i].c == color) {
2298 const piece = move.appear[i].p;
2299 this.updateReserve(color, piece, this.reserve[color][piece] - 1);
2300 }
41534b92
BA
2301 }
2302 for (let i=minSize; i<move.vanish.length; i++) {
2303 // Something vanish: add to reserve except if recycle & opponent
0c44c676
BA
2304 if (
2305 this.options["crazyhouse"] ||
2306 (this.options["recycle"] && move.vanish[i].c == color)
2307 ) {
2308 const piece = move.vanish[i].p;
41534b92 2309 this.updateReserve(color, piece, this.reserve[color][piece] + 1);
0c44c676 2310 }
41534b92
BA
2311 }
2312 }
2313 }
2314
2315 play(move) {
2316 this.prePlay(move);
b4ae3ff6
BA
2317 if (this.hasEnpassant)
2318 this.epSquare = this.getEpSquare(move);
41534b92
BA
2319 this.playOnBoard(move);
2320 this.postPlay(move);
2321 }
2322
2323 postPlay(move) {
b4ae3ff6 2324 if (this.options["dark"])
c9ab0340 2325 this.updateEnlightened();
41534b92
BA
2326 if (this.options["teleport"]) {
2327 if (
cc2c7183 2328 this.subTurnTeleport == 1 &&
41534b92 2329 move.vanish.length > move.appear.length &&
518bfb7a 2330 move.vanish[1].c == this.turn
41534b92
BA
2331 ) {
2332 const v = move.vanish[move.vanish.length - 1];
2333 this.captured = {x: v.x, y: v.y, c: v.c, p: v.p};
cc2c7183 2334 this.subTurnTeleport = 2;
41534b92
BA
2335 return;
2336 }
cc2c7183 2337 this.subTurnTeleport = 1;
41534b92
BA
2338 this.captured = null;
2339 }
182ba661
BA
2340 this.tryChangeTurn(move);
2341 }
2342
2343 tryChangeTurn(move) {
5f08c59b 2344 if (this.isLastMove(move)) {
bc97fdd1 2345 this.turn = C.GetOppTurn(this.turn);
5f08c59b
BA
2346 this.movesCount++;
2347 this.subTurn = 1;
41534b92 2348 }
af9c9be3
BA
2349 else if (!move.next)
2350 this.subTurn++;
5f08c59b
BA
2351 }
2352
2353 isLastMove(move) {
af9c9be3
BA
2354 if (move.next)
2355 return false;
2356 const color = this.turn;
ceeac4e8
BA
2357 const oppKingPos = this.searchKingPos(C.GetOppTurn(color));
2358 if (oppKingPos.length == 0 || this.underCheck(oppKingPos, [color]))
af9c9be3 2359 return true;
5f08c59b 2360 return (
af9c9be3
BA
2361 (
2362 !this.options["balance"] ||
6b9320bb
BA
2363 ![1, 2].includes(this.movesCount) ||
2364 this.subTurn == 2
af9c9be3
BA
2365 )
2366 &&
2367 (
2368 !this.options["doublemove"] ||
2369 this.movesCount == 0 ||
2370 this.subTurn == 2
2371 )
2372 &&
2373 (
2374 !this.options["progressive"] ||
2375 this.subTurn == this.movesCount + 1
2376 )
5f08c59b 2377 );
41534b92
BA
2378 }
2379
2380 // "Stop at the first move found"
2381 atLeastOneMove(color) {
41534b92
BA
2382 for (let i = 0; i < this.size.x; i++) {
2383 for (let j = 0; j < this.size.y; j++) {
2384 if (this.board[i][j] != "" && this.getColor(i, j) == color) {
cc2c7183
BA
2385 // NOTE: in fact searching for all potential moves from i,j.
2386 // I don't believe this is an issue, for now at least.
de1421be 2387 const moves = this.getPotentialMovesFrom([i, j], color);
b4ae3ff6
BA
2388 if (moves.some(m => this.filterValid([m]).length >= 1))
2389 return true;
41534b92
BA
2390 }
2391 }
2392 }
2393 if (this.hasReserve && this.reserve[color]) {
2394 for (let p of Object.keys(this.reserve[color])) {
2395 const moves = this.getDropMovesFrom([color, p]);
b4ae3ff6
BA
2396 if (moves.some(m => this.filterValid([m]).length >= 1))
2397 return true;
41534b92
BA
2398 }
2399 }
2400 return false;
2401 }
2402
2403 // What is the score ? (Interesting if game is over)
d6d0a46e
BA
2404 getCurrentScore(move_s) {
2405 const move = move_s[move_s.length - 1];
2406 // Shortcut in case the score was computed before:
2407 if (move.result)
2408 return move.result;
ceeac4e8 2409 const oppTurn = C.GetOppTurn(this.turn);
a548cb4e 2410 const kingPos = {
ceeac4e8
BA
2411 w: this.searchKingPos('w'),
2412 b: this.searchKingPos('b')
a548cb4e 2413 };
ceeac4e8 2414 if (kingPos[this.turn].length == 0 && kingPos[oppTurn].length == 0)
b4ae3ff6 2415 return "1/2";
ceeac4e8 2416 if (kingPos[this.turn].length == 0)
b57b726c 2417 return (this.turn == "w" ? "0-1" : "1-0");
ceeac4e8 2418 if (kingPos[oppTurn].length == 0)
b57b726c 2419 return (this.turn == "w" ? "1-0" : "0-1");
ceeac4e8 2420 if (this.atLeastOneMove(this.turn))
b4ae3ff6 2421 return "*";
41534b92 2422 // No valid move: stalemate or checkmate?
ceeac4e8 2423 if (!this.underCheck(kingPos[this.turn], this.getOppCols(this.turn)))
b4ae3ff6 2424 return "1/2";
41534b92 2425 // OK, checkmate
ceeac4e8 2426 return (this.turn == "w" ? "0-1" : "1-0");
41534b92
BA
2427 }
2428
41534b92
BA
2429 playVisual(move, r) {
2430 move.vanish.forEach(v => {
bc97fdd1
BA
2431 if (this.g_pieces[v.x][v.y]) //can be null (e.g. Apocalypse)
2432 this.g_pieces[v.x][v.y].remove();
c9ab0340 2433 this.g_pieces[v.x][v.y] = null;
41534b92 2434 });
3c61449b
BA
2435 let chessboard =
2436 document.getElementById(this.containerId).querySelector(".chessboard");
b4ae3ff6
BA
2437 if (!r)
2438 r = chessboard.getBoundingClientRect();
41534b92
BA
2439 const pieceWidth = this.getPieceWidth(r.width);
2440 move.appear.forEach(a => {
41534b92 2441 this.g_pieces[a.x][a.y] = document.createElement("piece");
2159c239
BA
2442 C.AddClass_es(this.g_pieces[a.x][a.y],
2443 this.pieces(a.c, a.x, a.y)[a.p]["class"]);
616a8d7a 2444 this.g_pieces[a.x][a.y].classList.add(V.GetColorClass(a.c));
41534b92
BA
2445 this.g_pieces[a.x][a.y].style.width = pieceWidth + "px";
2446 this.g_pieces[a.x][a.y].style.height = pieceWidth + "px";
2447 const [ip, jp] = this.getPixelPosition(a.x, a.y, r);
9db5050a
BA
2448 // Translate coordinates to use chessboard as reference:
2449 this.g_pieces[a.x][a.y].style.transform =
2450 `translate(${ip - r.x}px,${jp - r.y}px)`;
c9ab0340
BA
2451 if (this.enlightened && !this.enlightened[a.x][a.y])
2452 this.g_pieces[a.x][a.y].classList.add("hidden");
3c61449b 2453 chessboard.appendChild(this.g_pieces[a.x][a.y]);
41534b92 2454 });
c9ab0340
BA
2455 if (this.options["dark"])
2456 this.graphUpdateEnlightened();
41534b92
BA
2457 }
2458
5f08c59b 2459 // TODO: send stack receive stack, or allow incremental? (good/bad points)
e7d409fc 2460 buildMoveStack(move, r) {
5f08c59b
BA
2461 this.moveStack.push(move);
2462 this.computeNextMove(move);
b9877ed2
BA
2463 const then = () => {
2464 const newTurn = this.turn;
2465 if (this.moveStack.length == 1 && !this.hideMoves)
2466 this.playVisual(move, r);
2467 if (move.next) {
2468 this.gameState = {
2469 fen: this.getFen(),
2470 board: JSON.parse(JSON.stringify(this.board)) //easier
2471 };
2472 this.buildMoveStack(move.next, r);
5f08c59b
BA
2473 }
2474 else {
b9877ed2
BA
2475 if (this.moveStack.length == 1) {
2476 // Usual case (one normal move)
2477 this.afterPlay(this.moveStack, newTurn, {send: true, res: true});
2478 this.moveStack = [];
2479 }
2480 else {
2481 this.afterPlay(this.moveStack, newTurn, {send: true, res: false});
2482 this.re_initFromFen(this.gameState.fen, this.gameState.board);
2483 this.playReceivedMove(this.moveStack.slice(1), () => {
2484 this.afterPlay(this.moveStack, newTurn, {send: false, res: true});
2485 this.moveStack = [];
2486 });
2487 }
5f08c59b 2488 }
b9877ed2
BA
2489 };
2490 // If hiding moves, then they are revealed in play() with callback
2491 this.play(move, this.hideMoves ? then : null);
2492 if (!this.hideMoves)
2493 then();
41534b92
BA
2494 }
2495
5f08c59b
BA
2496 // Implemented in variants using (automatic) moveStack
2497 computeNextMove(move) {}
2498
5f08c59b
BA
2499 animateMoving(start, end, drag, segments, cb) {
2500 let initPiece = this.getDomPiece(start.x, start.y);
bc97fdd1
BA
2501 if (!initPiece) { //TODO: shouldn't occur!
2502 cb();
2503 return;
2504 }
5f08c59b 2505 // NOTE: cloning often not required, but light enough, and simpler
9db5050a
BA
2506 let movingPiece = initPiece.cloneNode();
2507 initPiece.style.opacity = "0";
2508 let container =
2509 document.getElementById(this.containerId)
2510 const r = container.querySelector(".chessboard").getBoundingClientRect();
5f08c59b 2511 if (typeof start.x == "string") {
082e639a
BA
2512 // Need to bound width/height (was 100% for reserve pieces)
2513 const pieceWidth = this.getPieceWidth(r.width);
2514 movingPiece.style.width = pieceWidth + "px";
2515 movingPiece.style.height = pieceWidth + "px";
2516 }
f55a0a67 2517 const maxDist = this.getMaxDistance(r);
5f08c59b
BA
2518 const apparentColor = this.getColor(start.x, start.y);
2519 const pieces = this.pieces(apparentColor, start.x, start.y);
2520 if (drag) {
2521 const startCode = this.getPiece(start.x, start.y);
3b641716 2522 C.RemoveClass_es(movingPiece, pieces[startCode]["class"]);
5f08c59b
BA
2523 C.AddClass_es(movingPiece, pieces[drag.p]["class"]);
2524 if (apparentColor != drag.c) {
616a8d7a
BA
2525 movingPiece.classList.remove(V.GetColorClass(apparentColor));
2526 movingPiece.classList.add(V.GetColorClass(drag.c));
41534b92 2527 }
41534b92 2528 }
9db5050a 2529 container.appendChild(movingPiece);
15106e82 2530 const animateSegment = (index, cb) => {
9db5050a 2531 // NOTE: move.drag could be generalized per-segment (usage?)
5f08c59b
BA
2532 const [i1, j1] = segments[index][0];
2533 const [i2, j2] = segments[index][1];
15106e82
BA
2534 const dep = this.getPixelPosition(i1, j1, r);
2535 const arr = this.getPixelPosition(i2, j2, r);
9db5050a
BA
2536 movingPiece.style.transitionDuration = "0s";
2537 movingPiece.style.transform = `translate(${dep[0]}px, ${dep[1]}px)`;
15106e82
BA
2538 const distance =
2539 Math.sqrt((arr[0] - dep[0]) ** 2 + (arr[1] - dep[1]) ** 2);
2540 const duration = 0.2 + (distance / maxDist) * 0.3;
9db5050a
BA
2541 // TODO: unclear why we need this new delay below:
2542 setTimeout(() => {
2543 movingPiece.style.transitionDuration = duration + "s";
adf7c659 2544 // movingPiece is child of container: no need to adjust coordinates
9db5050a
BA
2545 movingPiece.style.transform = `translate(${arr[0]}px, ${arr[1]}px)`;
2546 setTimeout(cb, duration * 1000);
2547 }, 50);
15106e82 2548 };
15106e82 2549 let index = 0;
635418a5 2550 const animateSegmentCallback = () => {
5f08c59b 2551 if (index < segments.length)
635418a5 2552 animateSegment(index++, animateSegmentCallback);
15106e82 2553 else {
9db5050a
BA
2554 movingPiece.remove();
2555 initPiece.style.opacity = "1";
5f08c59b 2556 cb();
15106e82 2557 }
635418a5
BA
2558 };
2559 animateSegmentCallback();
41534b92
BA
2560 }
2561
5f08c59b
BA
2562 // Input array of objects with at least fields x,y (e.g. PiPo)
2563 animateFading(arr, cb) {
2564 const animLength = 350; //TODO: 350ms? More? Less?
2565 arr.forEach(v => {
2566 let fadingPiece = this.getDomPiece(v.x, v.y);
2567 fadingPiece.style.transitionDuration = (animLength / 1000) + "s";
2568 fadingPiece.style.opacity = "0";
2569 });
2570 setTimeout(cb, animLength);
2571 }
2572
2573 animate(move, callback) {
2574 if (this.noAnimate || move.noAnimate) {
2575 callback();
2576 return;
2577 }
2578 let segments = move.segments;
2579 if (!segments)
2580 segments = [ [[move.start.x, move.start.y], [move.end.x, move.end.y]] ];
2581 let targetObj = new TargetObj(callback);
2582 if (move.start.x != move.end.x || move.start.y != move.end.y) {
2583 targetObj.target++;
2584 this.animateMoving(move.start, move.end, move.drag, segments,
2585 () => targetObj.increment());
2586 }
2587 if (move.vanish.length > move.appear.length) {
2588 const arr = move.vanish.slice(move.appear.length)
e7d409fc
BA
2589 // Ignore disappearing pieces hidden by some appearing ones:
2590 .filter(v => move.appear.every(a => a.x != v.x || a.y != v.y));
5f08c59b
BA
2591 if (arr.length > 0) {
2592 targetObj.target++;
2593 this.animateFading(arr, () => targetObj.increment());
2594 }
2595 }
462947f0
BA
2596 targetObj.target +=
2597 this.tryAnimateCastle(move, () => targetObj.increment());
5f08c59b
BA
2598 targetObj.target +=
2599 this.customAnimate(move, segments, () => targetObj.increment());
2600 if (targetObj.target == 0)
2601 callback();
2602 }
2603
462947f0
BA
2604 tryAnimateCastle(move, cb) {
2605 if (
2606 this.hasCastle &&
2607 move.vanish.length == 2 &&
2608 move.appear.length == 2 &&
2609 this.isKing(0, 0, move.vanish[0].p) &&
2610 this.isKing(0, 0, move.appear[0].p)
2611 ) {
2612 const start = {x: move.vanish[1].x, y: move.vanish[1].y},
2613 end = {x: move.appear[1].x, y: move.appear[1].y};
2614 const segments = [ [[start.x, start.y], [end.x, end.y]] ];
2615 this.animateMoving(start, end, null, segments, cb);
2616 return 1;
2617 }
2618 return 0;
2619 }
2620
5f08c59b
BA
2621 // Potential other animations (e.g. for Suction variant)
2622 customAnimate(move, segments, cb) {
2623 return 0; //nb of targets
2624 }
2625
126ffc70 2626 launchAnimation(moves, container, callback) {
98d14451 2627 if (this.hideMoves) {
b9877ed2
BA
2628 for (let i=0; i<moves.length; i++)
2629 // If hiding moves, they are revealed into play():
2630 this.play(moves[i], i == moves.length - 1 ? callback : () => {});
98d14451
BA
2631 return;
2632 }
2633 const r = container.querySelector(".chessboard").getBoundingClientRect();
2634 const animateRec = i => {
2635 this.animate(moves[i], () => {
2636 this.play(moves[i]);
2637 this.playVisual(moves[i], r);
2638 if (i < moves.length - 1)
2639 setTimeout(() => animateRec(i+1), 300);
2640 else
2641 callback();
2642 });
21e8e712 2643 };
98d14451
BA
2644 animateRec(0);
2645 }
2646
2647 playReceivedMove(moves, callback) {
e081c5eb
BA
2648 // Delay if user wasn't focused:
2649 const checkDisplayThenAnimate = (delay) => {
3c61449b 2650 if (container.style.display == "none") {
21e8e712
BA
2651 alert("New move! Let's go back to game...");
2652 document.getElementById("gameInfos").style.display = "none";
3c61449b 2653 container.style.display = "block";
126ffc70
BA
2654 setTimeout(
2655 () => this.launchAnimation(moves, container, callback),
2656 700
2657 );
2658 }
2659 else {
2660 setTimeout(
2661 () => this.launchAnimation(moves, container, callback),
2662 delay || 0
2663 );
21e8e712 2664 }
21e8e712 2665 };
3c61449b 2666 let container = document.getElementById(this.containerId);
016306e3
BA
2667 if (document.hidden) {
2668 document.onvisibilitychange = () => {
c9a20f4f 2669 // TODO here: page reload ?! (some issues if tab changed...)
016306e3 2670 document.onvisibilitychange = undefined;
e081c5eb 2671 checkDisplayThenAnimate(700);
fd31883b 2672 };
fd31883b 2673 }
b4ae3ff6
BA
2674 else
2675 checkDisplayThenAnimate();
41534b92
BA
2676 }
2677
2678};