Add Suction
[xogo.git] / variants / Chakart / class.js
1 import ChessRules from "/base_rules.js";
2 import GiveawayRules from "/variants/Giveaway/class.js";
3 import { ArrayFun } from "/utils/array.js";
4 import { Random } from "/utils/alea.js";
5 import PiPo from "/utils/PiPo.js";
6 import Move from "/utils/Move.js";
7
8 export default class ChakartRules extends ChessRules {
9
10 static get Options() {
11 return {
12 select: [
13 {
14 label: "Randomness",
15 variable: "randomness",
16 defaut: 2,
17 options: [
18 {label: "Deterministic", value: 0},
19 {label: "Symmetric random", value: 1},
20 {label: "Asymmetric random", value: 2}
21 ]
22 }
23 ],
24 styles: ["cylinder"]
25 };
26 }
27
28 get pawnPromotions() {
29 return ['q', 'r', 'n', 'b', 'k'];
30 }
31
32 get hasCastle() {
33 return false;
34 }
35 get hasEnpassant() {
36 return false;
37 }
38 get hasReserve() {
39 return true;
40 }
41 get hasReserveFen() {
42 return false;
43 }
44
45 static get IMMOBILIZE_CODE() {
46 return {
47 'p': 's',
48 'r': 'u',
49 'n': 'o',
50 'b': 'c',
51 'q': 't',
52 'k': 'l'
53 };
54 }
55
56 static get IMMOBILIZE_DECODE() {
57 return {
58 's': 'p',
59 'u': 'r',
60 'o': 'n',
61 'c': 'b',
62 't': 'q',
63 'l': 'k'
64 };
65 }
66
67 // Fictive color 'a', bomb banana mushroom egg
68 static get BOMB() {
69 return 'w'; //"Wario"
70 }
71 static get BANANA() {
72 return 'd'; //"Donkey"
73 }
74 static get EGG() {
75 return 'e';
76 }
77 static get MUSHROOM() {
78 return 'm';
79 }
80
81 static get EGG_SURPRISE() {
82 return [
83 "kingboo", "bowser", "daisy", "koopa",
84 "luigi", "waluigi", "toadette", "chomp"];
85 }
86
87 canIplay(x, y) {
88 if (
89 this.playerColor != this.turn ||
90 Object.keys(V.IMMOBILIZE_DECODE).includes(this.getPiece(x, y))
91 ) {
92 return false;
93 }
94 return this.egg == "kingboo" || this.getColor(x, y) == this.turn;
95 }
96
97 pieces(color, x, y) {
98 const specials = {
99 'i': {"class": "invisible"}, //queen
100 '?': {"class": "mystery"}, //...initial square
101 'e': {"class": "egg"},
102 'm': {"class": "mushroom"},
103 'd': {"class": "banana"},
104 'w': {"class": "bomb"},
105 'z': {"class": "remote-capture"}
106 };
107 const bowsered = {
108 's': {"class": ["immobilized", "pawn"]},
109 'u': {"class": ["immobilized", "rook"]},
110 'o': {"class": ["immobilized", "knight"]},
111 'c': {"class": ["immobilized", "bishop"]},
112 't': {"class": ["immobilized", "queen"]},
113 'l': {"class": ["immobilized", "king"]}
114 };
115 return Object.assign({}, specials, bowsered, super.pieces(color, x, y));
116 }
117
118 genRandInitFen(seed) {
119 const gr = new GiveawayRules(
120 {mode: "suicide", options: this.options, genFenOnly: true});
121 // Add Peach + mario flags
122 return gr.genRandInitFen(seed).slice(0, -17) + '{"flags":"1111"}';
123 }
124
125 fen2board(f) {
126 return (
127 f.charCodeAt() <= 90
128 ? "w" + f.toLowerCase()
129 : (['w', 'd', 'e', 'm'].includes(f) ? "a" : "b") + f
130 );
131 }
132
133 setFlags(fenflags) {
134 // King can send shell? Queen can be invisible?
135 this.powerFlags = {
136 w: {k: false, q: false},
137 b: {k: false, q: false}
138 };
139 for (let c of ['w', 'b']) {
140 for (let p of ['k', 'q']) {
141 this.powerFlags[c][p] =
142 fenflags.charAt((c == "w" ? 0 : 2) + (p == 'k' ? 0 : 1)) == "1";
143 }
144 }
145 }
146
147 aggregateFlags() {
148 return this.powerFlags;
149 }
150
151 disaggregateFlags(flags) {
152 this.powerFlags = flags;
153 }
154
155 getFlagsFen() {
156 return ['w', 'b'].map(c => {
157 return ['k', 'q'].map(p => this.powerFlags[c][p] ? "1" : "0").join("");
158 }).join("");
159 }
160
161 setOtherVariables(fenParsed) {
162 this.setFlags(fenParsed.flags);
163 this.reserve = {}; //to be filled later
164 this.egg = null;
165 this.moveStack = [];
166 // Change seed (after FEN generation!!)
167 // so that further calls differ between players:
168 Random.setSeed(Math.floor(10000 * Math.random()));
169 }
170
171 // For Toadette bonus
172 getDropMovesFrom([c, p]) {
173 if (typeof c != "string" || this.reserve[c][p] == 0)
174 return [];
175 let moves = [];
176 const start = (c == 'w' && p == 'p' ? 1 : 0);
177 const end = (c == 'b' && p == 'p' ? 7 : 8);
178 for (let i = start; i < end; i++) {
179 for (let j = 0; j < this.size.y; j++) {
180 const pieceIJ = this.getPiece(i, j);
181 const colIJ = this.getColor(i, j);
182 if (this.board[i][j] == "" || colIJ == 'a' || pieceIJ == 'i') {
183 let m = new Move({
184 start: {x: c, y: p},
185 appear: [new PiPo({x: i, y: j, c: c, p: p})],
186 vanish: []
187 });
188 // A drop move may remove a bonus (or hidden queen!)
189 if (this.board[i][j] != "")
190 m.vanish.push(new PiPo({x: i, y: j, c: colIJ, p: pieceIJ}));
191 moves.push(m);
192 }
193 }
194 }
195 return moves;
196 }
197
198 getPotentialMovesFrom([x, y]) {
199 let moves = [];
200 const piece = this.getPiece(x, y);
201 if (this.egg == "toadette")
202 moves = this.getDropMovesFrom([x, y]);
203 else if (this.egg == "kingboo") {
204 const color = this.turn;
205 const oppCol = C.GetOppCol(color);
206 // Only allow to swap (non-immobilized!) pieces
207 for (let i=0; i<this.size.x; i++) {
208 for (let j=0; j<this.size.y; j++) {
209 const colIJ = this.getColor(i, j);
210 const pieceIJ = this.getPiece(i, j);
211 if (
212 (i != x || j != y) &&
213 ['w', 'b'].includes(colIJ) &&
214 !Object.keys(V.IMMOBILIZE_DECODE).includes(pieceIJ) &&
215 // Next conditions = no pawn on last rank
216 (
217 piece != 'p' ||
218 (
219 (color != 'w' || i != 0) &&
220 (color != 'b' || i != this.size.x - 1)
221 )
222 )
223 &&
224 (
225 pieceIJ != 'p' ||
226 (
227 (colIJ != 'w' || x != 0) &&
228 (colIJ != 'b' || x != this.size.x - 1)
229 )
230 )
231 ) {
232 let m = this.getBasicMove([x, y], [i, j]);
233 m.appear.push(new PiPo({x: x, y: y, p: pieceIJ, c: colIJ}));
234 m.kingboo = true; //avoid some side effects (bananas/bombs)
235 moves.push(m);
236 }
237 }
238 }
239 }
240 else {
241 // Normal case (including bonus daisy)
242 switch (piece) {
243 case 'p':
244 moves = this.getPawnMovesFrom([x, y]); //apply promotions
245 break;
246 case 'q':
247 moves = this.getQueenMovesFrom([x, y]);
248 break;
249 case 'k':
250 moves = this.getKingMovesFrom([x, y]);
251 break;
252 case 'n':
253 moves = this.getKnightMovesFrom([x, y]);
254 break;
255 case 'b':
256 case 'r':
257 // Explicitely listing types to avoid moving immobilized piece
258 moves = super.getPotentialMovesOf(piece, [x, y]);
259 break;
260 }
261 }
262 return moves;
263 }
264
265 canStepOver(i, j) {
266 return (
267 this.board[i][j] == "" ||
268 ['i', V.EGG, V.MUSHROOM].includes(this.getPiece(i, j))
269 );
270 }
271
272 getPawnMovesFrom([x, y]) {
273 const color = this.turn;
274 const oppCol = C.GetOppCol(color);
275 const shiftX = (color == 'w' ? -1 : 1);
276 const firstRank = (color == "w" ? this.size.x - 1 : 0);
277 let moves = [];
278 const frontPiece = this.getPiece(x + shiftX, y);
279 if (
280 this.board[x + shiftX][y] == "" ||
281 this.getColor(x + shiftX, y) == 'a' ||
282 frontPiece == 'i'
283 ) {
284 moves.push(this.getBasicMove([x, y], [x + shiftX, y]));
285 if (
286 [firstRank, firstRank + shiftX].includes(x) &&
287 ![V.BANANA, V.BOMB].includes(frontPiece) &&
288 (
289 this.board[x + 2 * shiftX][y] == "" ||
290 this.getColor(x + 2 * shiftX, y) == 'a' ||
291 this.getPiece(x + 2 * shiftX, y) == 'i'
292 )
293 ) {
294 moves.push(this.getBasicMove([x, y], [x + 2 * shiftX, y]));
295 }
296 }
297 for (let shiftY of [-1, 1]) {
298 if (
299 y + shiftY >= 0 &&
300 y + shiftY < this.size.y &&
301 this.board[x + shiftX][y + shiftY] != "" &&
302 // Pawns cannot capture invisible queen this way!
303 this.getPiece(x + shiftX, y + shiftY) != 'i' &&
304 ['a', oppCol].includes(this.getColor(x + shiftX, y + shiftY))
305 ) {
306 moves.push(this.getBasicMove([x, y], [x + shiftX, y + shiftY]));
307 }
308 }
309 super.pawnPostProcess(moves, color, oppCol);
310 // Add mushroom on before-last square
311 moves.forEach(m => {
312 let revStep = [m.start.x - m.end.x, m.start.y - m.end.y];
313 for (let i of [0, 1])
314 revStep[i] = revStep[i] / Math.abs(revStep[i]) || 0;
315 const [blx, bly] = [m.end.x + revStep[0], m.end.y + revStep[1]];
316 m.appear.push(new PiPo({x: blx, y: bly, c: 'a', p: 'm'}));
317 if (blx != x && this.board[blx][bly] != "") {
318 m.vanish.push(new PiPo({
319 x: blx,
320 y: bly,
321 c: this.getColor(blx, bly),
322 p: this.getPiece(blx, bly)
323 }));
324 }
325 });
326 return moves;
327 }
328
329 getKnightMovesFrom([x, y]) {
330 // Add egg on initial square:
331 return this.getPotentialMovesOf('n', [x, y]).map(m => {
332 m.appear.push(new PiPo({p: "e", c: "a", x: x, y: y}));
333 return m;
334 });
335 }
336
337 getQueenMovesFrom(sq) {
338 const normalMoves = this.getPotentialMovesOf('q', sq);
339 // If flag allows it, add 'invisible movements'
340 let invisibleMoves = [];
341 if (this.powerFlags[this.turn]['q']) {
342 normalMoves.forEach(m => {
343 if (
344 m.appear.length == 1 &&
345 m.vanish.length == 1 &&
346 // Only simple non-capturing moves:
347 m.vanish[0].c != 'a'
348 ) {
349 let im = JSON.parse(JSON.stringify(m));
350 im.appear[0].p = 'i';
351 im.noAnimate = true;
352 invisibleMoves.push(im);
353 }
354 });
355 }
356 return normalMoves.concat(invisibleMoves);
357 }
358
359 getKingMovesFrom([x, y]) {
360 let moves = this.getPotentialMovesOf('k', [x, y]);
361 // If flag allows it, add 'remote shell captures'
362 if (this.powerFlags[this.turn]['k']) {
363 super.pieces()['k'].moves[0].steps.forEach(step => {
364 let [i, j] = [x + step[0], y + step[1]];
365 while (this.onBoard(i, j) && this.canStepOver(i, j)) {
366 i += step[0];
367 j += step[1];
368 }
369 if (this.onBoard(i, j)) {
370 const colIJ = this.getColor(i, j);
371 if (colIJ != this.turn) {
372 // May just destroy a bomb or banana:
373 let shellCapture = new Move({
374 start: {x: x, y: y},
375 end: {x: i, y: j},
376 appear: [],
377 vanish: [
378 new PiPo({x: i, y: j, c: colIJ, p: this.getPiece(i, j)})
379 ]
380 });
381 shellCapture.shell = true; //easier play()
382 shellCapture.choice = 'z'; //to display in showChoices()
383 moves.push(shellCapture);
384 }
385 }
386 });
387 }
388 return moves;
389 }
390
391 play(move) {
392 const color = this.turn;
393 const oppCol = C.GetOppCol(color);
394 if (
395 move.appear.length > 0 &&
396 move.appear[0].p == 'p' &&
397 (
398 (color == 'w' && move.end.x == 0) ||
399 (color == 'b' && move.end.x == this.size.x - 1)
400 )
401 ) {
402 // "Forgotten" promotion, which occurred after some effect
403 let moves = [move];
404 super.pawnPostProcess(moves, color, oppCol);
405 super.showChoices(moves);
406 return false;
407 }
408 if (!move.nextComputed) {
409 // Set potential random effects, so that play() is deterministic
410 // from opponent viewpoint:
411 const endPiece = this.getPiece(move.end.x, move.end.y);
412 switch (endPiece) {
413 case V.EGG:
414 move.egg = Random.sample(V.EGG_SURPRISE);
415 move.next = this.getEggEffect(move);
416 break;
417 case V.MUSHROOM:
418 move.next = this.getMushroomEffect(move);
419 break;
420 case V.BANANA:
421 case V.BOMB:
422 move.next = this.getBombBananaEffect(move, endPiece);
423 break;
424 }
425 if (!move.next && move.appear.length > 0 && !move.kingboo) {
426 const movingPiece = move.appear[0].p;
427 if (['b', 'r'].includes(movingPiece)) {
428 // Drop a banana or bomb:
429 const bs =
430 this.getRandomSquare([move.end.x, move.end.y],
431 movingPiece == 'r'
432 ? [[1, 1], [1, -1], [-1, 1], [-1, -1]]
433 : [[1, 0], [-1, 0], [0, 1], [0, -1]],
434 "freeSquare");
435 if (bs) {
436 move.appear.push(
437 new PiPo({
438 x: bs[0],
439 y: bs[1],
440 c: 'a',
441 p: movingPiece == 'r' ? 'd' : 'w'
442 })
443 );
444 if (this.board[bs[0]][bs[1]] != "") {
445 move.vanish.push(
446 new PiPo({
447 x: bs[0],
448 y: bs[1],
449 c: this.getColor(bs[0], bs[1]),
450 p: this.getPiece(bs[0], bs[1])
451 })
452 );
453 }
454 }
455 }
456 }
457 move.nextComputed = true;
458 }
459 this.egg = move.egg;
460 if (move.egg == "toadette") {
461 this.reserve = { w: {}, b: {} };
462 // Randomly select a piece in pawnPromotions
463 if (!move.toadette)
464 move.toadette = Random.sample(this.pawnPromotions);
465 this.reserve[color][move.toadette] = 1;
466 this.re_drawReserve([color]);
467 }
468 else if (Object.keys(this.reserve).length > 0) {
469 this.reserve = {};
470 this.re_drawReserve([color]);
471 }
472 if (move.shell)
473 this.powerFlags[color]['k'] = false;
474 else if (move.appear.length > 0 && move.appear[0].p == 'i') {
475 this.powerFlags[move.appear[0].c]['q'] = false;
476 if (color == this.playerColor) {
477 move.appear.push(
478 new PiPo({x: move.start.x, y: move.start.y, c: color, p: '?'}));
479 }
480 }
481 if (color == this.playerColor) {
482 // Look for an immobilized piece of my color: it can now move
483 for (let i=0; i<8; i++) {
484 for (let j=0; j<8; j++) {
485 if ((i != move.end.x || j != move.end.y) && this.board[i][j] != "") {
486 const piece = this.getPiece(i, j);
487 if (
488 this.getColor(i, j) == color &&
489 Object.keys(V.IMMOBILIZE_DECODE).includes(piece)
490 ) {
491 move.vanish.push(new PiPo({
492 x: i, y: j, c: color, p: piece
493 }));
494 move.appear.push(new PiPo({
495 x: i, y: j, c: color, p: V.IMMOBILIZE_DECODE[piece]
496 }));
497 }
498 }
499 }
500 }
501 // Also make opponent invisible queen visible again, if any
502 for (let i=0; i<8; i++) {
503 for (let j=0; j<8; j++) {
504 if (
505 this.board[i][j] != "" &&
506 this.getColor(i, j) == oppCol
507 ) {
508 const pieceIJ = this.getPiece(i, j);
509 if (pieceIJ == 'i') {
510 move.vanish.push(new PiPo({x: i, y: j, c: oppCol, p: 'i'}));
511 move.appear.push(new PiPo({x: i, y: j, c: oppCol, p: 'q'}));
512 }
513 else if (pieceIJ == '?')
514 move.vanish.push(new PiPo({x: i, y: j, c: oppCol, p: '?'}));
515 }
516 }
517 }
518 }
519 if (!move.next && !["daisy", "toadette", "kingboo"].includes(move.egg)) {
520 this.turn = oppCol;
521 this.movesCount++;
522 }
523 if (move.egg)
524 this.displayBonus(move);
525 this.playOnBoard(move);
526 this.nextMove = move.next;
527 return true;
528 }
529
530 // Helper to set and apply banana/bomb effect
531 getRandomSquare([x, y], steps, freeSquare) {
532 let validSteps = steps.filter(s => this.onBoard(x + s[0], y + s[1]));
533 if (freeSquare) {
534 // Square to put banana/bomb cannot be occupied by a piece
535 validSteps = validSteps.filter(s => {
536 return ["", 'a'].includes(this.getColor(x + s[0], y + s[1]))
537 });
538 }
539 if (validSteps.length == 0)
540 return null;
541 const step = validSteps[Random.randInt(validSteps.length)];
542 return [x + step[0], y + step[1]];
543 }
544
545 getEggEffect(move) {
546 const getRandomPiece = (c) => {
547 let bagOfPieces = [];
548 for (let i=0; i<this.size.x; i++) {
549 for (let j=0; j<this.size.y; j++) {
550 if (this.getColor(i, j) == c && this.getPiece(i, j) != 'k')
551 bagOfPieces.push([i, j]);
552 }
553 }
554 if (bagOfPieces.length >= 1)
555 return Random.sample(bagOfPieces);
556 return null;
557 };
558 const color = this.turn;
559 let em = null;
560 switch (move.egg) {
561 case "luigi":
562 case "waluigi":
563 // Change color of friendly or enemy piece, king excepted
564 const oldColor = (move.egg == "waluigi" ? color : C.GetOppCol(color));
565 const newColor = C.GetOppCol(oldColor);
566 const coords = getRandomPiece(oldColor);
567 if (coords) {
568 const piece = this.getPiece(coords[0], coords[1]);
569 em = new Move({
570 appear: [
571 new PiPo({x: coords[0], y: coords[1], c: newColor, p: piece})
572 ],
573 vanish: [
574 new PiPo({x: coords[0], y: coords[1], c: oldColor, p: piece})
575 ]
576 });
577 }
578 break;
579 case "bowser":
580 em = new Move({
581 appear: [
582 new PiPo({
583 x: move.end.x,
584 y: move.end.y,
585 c: color,
586 p: V.IMMOBILIZE_CODE[move.appear[0].p]
587 })
588 ],
589 vanish: [
590 new PiPo({
591 x: move.end.x,
592 y: move.end.y,
593 c: color,
594 p: move.appear[0].p
595 })
596 ]
597 });
598 break;
599 case "koopa":
600 // Reverse move
601 em = new Move({
602 appear: [
603 new PiPo({
604 x: move.start.x, y: move.start.y, c: color, p: move.appear[0].p
605 })
606 ],
607 vanish: [
608 new PiPo({
609 x: move.end.x, y: move.end.y, c: color, p: move.appear[0].p
610 })
611 ]
612 });
613 if (this.board[move.start.x][move.start.y] != "") {
614 // Pawn or knight let something on init square
615 em.vanish.push(new PiPo({
616 x: move.start.x,
617 y: move.start.y,
618 c: 'a',
619 p: this.getPiece(move.start.x, move.start.y)
620 }));
621 }
622 em.koopa = true; //to cancel mushroom effect
623 break;
624 case "chomp":
625 // Eat piece
626 em = new Move({
627 appear: [],
628 vanish: [
629 new PiPo({
630 x: move.end.x, y: move.end.y, c: color, p: move.appear[0].p
631 })
632 ],
633 end: {x: move.end.x, y: move.end.y}
634 });
635 break;
636 }
637 if (em && move.egg != "koopa")
638 em.noAnimate = true; //static move
639 return em;
640 }
641
642 getMushroomEffect(move) {
643 if (move.koopa || typeof move.start.x == "string")
644 return null;
645 let step = [move.end.x - move.start.x, move.end.y - move.start.y];
646 if ([0, 1].some(i => Math.abs(step[i]) >= 2 && Math.abs(step[1-i]) != 1)) {
647 // Slider, multi-squares: normalize step
648 for (let j of [0, 1])
649 step[j] = step[j] / Math.abs(step[j]) || 0;
650 }
651 const nextSquare = [move.end.x + step[0], move.end.y + step[1]];
652 const afterSquare =
653 [nextSquare[0] + step[0], nextSquare[1] + step[1]];
654 let nextMove = null;
655 if (this.onBoard(nextSquare[0], nextSquare[1])) {
656 this.playOnBoard(move); //HACK for getBasicMove()
657 nextMove = this.getBasicMove([move.end.x, move.end.y], nextSquare);
658 this.undoOnBoard(move);
659 }
660 return nextMove;
661 }
662
663 getBombBananaEffect(move, item) {
664 const steps = item == V.BANANA
665 ? [[1, 0], [-1, 0], [0, 1], [0, -1]]
666 : [[1, 1], [1, -1], [-1, 1], [-1, -1]];
667 const nextSquare = this.getRandomSquare([move.end.x, move.end.y], steps);
668 this.playOnBoard(move); //HACK for getBasicMove()
669 const res = this.getBasicMove([move.end.x, move.end.y], nextSquare);
670 this.undoOnBoard(move);
671 return res;
672 }
673
674 displayBonus(move) {
675 let divBonus = document.createElement("div");
676 divBonus.classList.add("bonus-text");
677 divBonus.innerHTML = move.egg;
678 let container = document.getElementById(this.containerId);
679 container.appendChild(divBonus);
680 setTimeout(() => container.removeChild(divBonus), 2000);
681 }
682
683 atLeastOneMove() {
684 return true;
685 }
686
687 filterValid(moves) {
688 return moves;
689 }
690
691 playPlusVisual(move, r) {
692 const nextLines = () => {
693 if (!this.play(move))
694 return;
695 this.moveStack.push(move);
696 this.playVisual(move, r);
697 if (this.nextMove)
698 this.playPlusVisual(this.nextMove, r);
699 else {
700 this.afterPlay(this.moveStack);
701 this.moveStack = [];
702 }
703 };
704 if (this.moveStack.length == 0)
705 nextLines();
706 else
707 this.animate(move, nextLines);
708 }
709
710 };