client/src/variants/Grand.js

   1 import { ChessRules } from "@/base_rules";
   2 import { ArrayFun } from "@/utils/array";
   3 import { randInt } from "@/utils/alea";
   4
   5 // NOTE: initial setup differs from the original; see
   6 // https://www.chessvariants.com/large.dir/freeling.html
   7 export class GrandRules extends ChessRules {
   8   static IsGoodFen(fen) {
   9     if (!ChessRules.IsGoodFen(fen)) return false;
  10     const fenParsed = V.ParseFen(fen);
  11     // 5) Check captures
  12     if (!fenParsed.captured || !fenParsed.captured.match(/^[0-9]{14,14}$/))
  13       return false;
  14     return true;
  15   }
  16
  17   static IsGoodEnpassant(enpassant) {
  18     if (enpassant != "-") {
  19       const squares = enpassant.split(",");
  20       if (squares.length > 2) return false;
  21       for (let sq of squares) {
  22         const ep = V.SquareToCoords(sq);
  23         if (isNaN(ep.x) || !V.OnBoard(ep)) return false;
  24       }
  25     }
  26     return true;
  27   }
  28
  29   static ParseFen(fen) {
  30     const fenParts = fen.split(" ");
  31     return Object.assign(
  32       ChessRules.ParseFen(fen),
  33       { captured: fenParts[5] }
  34     );
  35   }
  36
  37   getPpath(b) {
  38     return ([V.MARSHALL, V.CARDINAL].includes(b[1]) ? "Grand/" : "") + b;
  39   }
  40
  41   getFen() {
  42     return super.getFen() + " " + this.getCapturedFen();
  43   }
  44
  45   getFenForRepeat() {
  46     return super.getFenForRepeat() + "_" + this.getCapturedFen();
  47   }
  48
  49   getCapturedFen() {
  50     let counts = [...Array(14).fill(0)];
  51     let i = 0;
  52     for (let j = 0; j < V.PIECES.length; j++) {
  53       if (V.PIECES[j] == V.KING)
  54         //no king captured
  55         continue;
  56       counts[i] = this.captured["w"][V.PIECES[i]];
  57       counts[7 + i] = this.captured["b"][V.PIECES[i]];
  58       i++;
  59     }
  60     return counts.join("");
  61   }
  62
  63   setOtherVariables(fen) {
  64     super.setOtherVariables(fen);
  65     const fenParsed = V.ParseFen(fen);
  66     // Initialize captured pieces' counts from FEN
  67     this.captured = {
  68       w: {
  69         [V.PAWN]: parseInt(fenParsed.captured[0]),
  70         [V.ROOK]: parseInt(fenParsed.captured[1]),
  71         [V.KNIGHT]: parseInt(fenParsed.captured[2]),
  72         [V.BISHOP]: parseInt(fenParsed.captured[3]),
  73         [V.QUEEN]: parseInt(fenParsed.captured[4]),
  74         [V.MARSHALL]: parseInt(fenParsed.captured[5]),
  75         [V.CARDINAL]: parseInt(fenParsed.captured[6])
  76       },
  77       b: {
  78         [V.PAWN]: parseInt(fenParsed.captured[7]),
  79         [V.ROOK]: parseInt(fenParsed.captured[8]),
  80         [V.KNIGHT]: parseInt(fenParsed.captured[9]),
  81         [V.BISHOP]: parseInt(fenParsed.captured[10]),
  82         [V.QUEEN]: parseInt(fenParsed.captured[11]),
  83         [V.MARSHALL]: parseInt(fenParsed.captured[12]),
  84         [V.CARDINAL]: parseInt(fenParsed.captured[13])
  85       }
  86     };
  87   }
  88
  89   static get size() {
  90     return { x: 10, y: 10 };
  91   }
  92
  93   // Rook + knight:
  94   static get MARSHALL() {
  95     return "m";
  96   }
  97
  98   // Bishop + knight
  99   static get CARDINAL() {
 100     return "c";
 101   }
 102
 103   static get PIECES() {
 104     return ChessRules.PIECES.concat([V.MARSHALL, V.CARDINAL]);
 105   }
 106
 107   // There may be 2 enPassant squares (if pawn jump 3 squares)
 108   getEnpassantFen() {
 109     const L = this.epSquares.length;
 110     if (!this.epSquares[L - 1]) return "-"; //no en-passant
 111     let res = "";
 112     this.epSquares[L - 1].forEach(sq => {
 113       res += V.CoordsToSquare(sq) + ",";
 114     });
 115     return res.slice(0, -1); //remove last comma
 116   }
 117
 118   // En-passant after 2-sq or 3-sq jumps
 119   getEpSquare(moveOrSquare) {
 120     if (!moveOrSquare) return undefined;
 121     if (typeof moveOrSquare === "string") {
 122       const square = moveOrSquare;
 123       if (square == "-") return undefined;
 124       let res = [];
 125       square.split(",").forEach(sq => {
 126         res.push(V.SquareToCoords(sq));
 127       });
 128       return res;
 129     }
 130     // Argument is a move:
 131     const move = moveOrSquare;
 132     const [sx, sy, ex] = [move.start.x, move.start.y, move.end.x];
 133     if (this.getPiece(sx, sy) == V.PAWN && Math.abs(sx - ex) >= 2) {
 134       const step = (ex - sx) / Math.abs(ex - sx);
 135       let res = [
 136         {
 137           x: sx + step,
 138           y: sy
 139         }
 140       ];
 141       if (sx + 2 * step != ex) {
 142         //3-squares move
 143         res.push({
 144           x: sx + 2 * step,
 145           y: sy
 146         });
 147       }
 148       return res;
 149     }
 150     return undefined; //default
 151   }
 152
 153   getPotentialMovesFrom([x, y]) {
 154     switch (this.getPiece(x, y)) {
 155       case V.MARSHALL:
 156         return this.getPotentialMarshallMoves([x, y]);
 157       case V.CARDINAL:
 158         return this.getPotentialCardinalMoves([x, y]);
 159       default:
 160         return super.getPotentialMovesFrom([x, y]);
 161     }
 162   }
 163
 164   // Special pawn rules: promotions to captured friendly pieces,
 165   // optional on ranks 8-9 and mandatory on rank 10.
 166   getPotentialPawnMoves([x, y]) {
 167     const color = this.turn;
 168     let moves = [];
 169     const [sizeX, sizeY] = [V.size.x, V.size.y];
 170     const shiftX = color == "w" ? -1 : 1;
 171     const startRanks = color == "w" ? [sizeX - 2, sizeX - 3] : [1, 2];
 172     const lastRanks =
 173       color == "w" ? [0, 1, 2] : [sizeX - 1, sizeX - 2, sizeX - 3];
 174     const promotionPieces = [
 175       V.ROOK,
 176       V.KNIGHT,
 177       V.BISHOP,
 178       V.QUEEN,
 179       V.MARSHALL,
 180       V.CARDINAL
 181     ];
 182
 183     // Always x+shiftX >= 0 && x+shiftX < sizeX, because no pawns on last rank
 184     let finalPieces = undefined;
 185     if (lastRanks.includes(x + shiftX)) {
 186       finalPieces = promotionPieces.filter(p => this.captured[color][p] > 0);
 187       if (x + shiftX != lastRanks[0]) finalPieces.push(V.PAWN);
 188     } else finalPieces = [V.PAWN];
 189     if (this.board[x + shiftX][y] == V.EMPTY) {
 190       // One square forward
 191       for (let piece of finalPieces)
 192         moves.push(
 193           this.getBasicMove([x, y], [x + shiftX, y], { c: color, p: piece })
 194         );
 195       if (startRanks.includes(x)) {
 196         if (this.board[x + 2 * shiftX][y] == V.EMPTY) {
 197           // Two squares jump
 198           moves.push(this.getBasicMove([x, y], [x + 2 * shiftX, y]));
 199           if (x == startRanks[0] && this.board[x + 3 * shiftX][y] == V.EMPTY) {
 200             // Three squares jump
 201             moves.push(this.getBasicMove([x, y], [x + 3 * shiftX, y]));
 202           }
 203         }
 204       }
 205     }
 206     // Captures
 207     for (let shiftY of [-1, 1]) {
 208       if (
 209         y + shiftY >= 0 &&
 210         y + shiftY < sizeY &&
 211         this.board[x + shiftX][y + shiftY] != V.EMPTY &&
 212         this.canTake([x, y], [x + shiftX, y + shiftY])
 213       ) {
 214         for (let piece of finalPieces) {
 215           moves.push(
 216             this.getBasicMove([x, y], [x + shiftX, y + shiftY], {
 217               c: color,
 218               p: piece
 219             })
 220           );
 221         }
 222       }
 223     }
 224
 225     // En passant
 226     const Lep = this.epSquares.length;
 227     const epSquare = this.epSquares[Lep - 1];
 228     if (epSquare) {
 229       for (let epsq of epSquare) {
 230         // TODO: some redundant checks
 231         if (epsq.x == x + shiftX && Math.abs(epsq.y - y) == 1) {
 232           let enpassantMove = this.getBasicMove([x, y], [epsq.x, epsq.y]);
 233           // WARNING: the captured pawn may be diagonally behind us,
 234           // if it's a 3-squares jump and we take on 1st passing square
 235           const px = this.board[x][epsq.y] != V.EMPTY ? x : x - shiftX;
 236           enpassantMove.vanish.push({
 237             x: px,
 238             y: epsq.y,
 239             p: "p",
 240             c: this.getColor(px, epsq.y)
 241           });
 242           moves.push(enpassantMove);
 243         }
 244       }
 245     }
 246
 247     return moves;
 248   }
 249
 250   // TODO: different castle?
 251
 252   getPotentialMarshallMoves(sq) {
 253     return this.getSlideNJumpMoves(sq, V.steps[V.ROOK]).concat(
 254       this.getSlideNJumpMoves(sq, V.steps[V.KNIGHT], "oneStep")
 255     );
 256   }
 257
 258   getPotentialCardinalMoves(sq) {
 259     return this.getSlideNJumpMoves(sq, V.steps[V.BISHOP]).concat(
 260       this.getSlideNJumpMoves(sq, V.steps[V.KNIGHT], "oneStep")
 261     );
 262   }
 263
 264   isAttacked(sq, color) {
 265     return (
 266       super.isAttacked(sq, color) ||
 267       this.isAttackedByMarshall(sq, color) ||
 268       this.isAttackedByCardinal(sq, color)
 269     );
 270   }
 271
 272   isAttackedByMarshall(sq, color) {
 273     return (
 274       this.isAttackedBySlideNJump(sq, color, V.MARSHALL, V.steps[V.ROOK]) ||
 275       this.isAttackedBySlideNJump(
 276         sq,
 277         color,
 278         V.MARSHALL,
 279         V.steps[V.KNIGHT],
 280         "oneStep"
 281       )
 282     );
 283   }
 284
 285   isAttackedByCardinal(sq, color) {
 286     return (
 287       this.isAttackedBySlideNJump(sq, color, V.CARDINAL, V.steps[V.BISHOP]) ||
 288       this.isAttackedBySlideNJump(
 289         sq,
 290         color,
 291         V.CARDINAL,
 292         V.steps[V.KNIGHT],
 293         "oneStep"
 294       )
 295     );
 296   }
 297
 298   postPlay(move) {
 299     super.postPlay(move);
 300     if (move.vanish.length == 2 && move.appear.length == 1) {
 301       // Capture: update this.captured
 302       this.captured[move.vanish[1].c][move.vanish[1].p]++;
 303     }
 304     if (move.vanish[0].p != move.appear[0].p) {
 305       // Promotion: update this.captured
 306       this.captured[move.vanish[0].c][move.appear[0].p]--;
 307     }
 308   }
 309
 310   postUndo(move) {
 311     super.postUndo(move);
 312     if (move.vanish.length == 2 && move.appear.length == 1)
 313       this.captured[move.vanish[1].c][move.vanish[1].p]--;
 314     if (move.vanish[0].p != move.appear[0].p)
 315       this.captured[move.vanish[0].c][move.appear[0].p]++;
 316   }
 317
 318   static get VALUES() {
 319     return Object.assign(
 320       { c: 5, m: 7 }, //experimental
 321       ChessRules.VALUES
 322     );
 323   }
 324
 325   static get SEARCH_DEPTH() {
 326     return 2;
 327   }
 328
 329   static GenRandInitFen(randomness) {
 330     if (randomness == 0) {
 331       return (
 332         "r8r/1nbqkmcbn1/pppppppppp/91/91/91/91/PPPPPPPPPP/1NBQKMCBN1/R8R " +
 333         // No castling in the official initial setup
 334         "w 0 zzzz - 00000000000000"
 335       );
 336     }
 337
 338     let pieces = { w: new Array(10), b: new Array(10) };
 339     let flags = "";
 340     // Shuffle pieces on first and last rank
 341     for (let c of ["w", "b"]) {
 342       if (c == 'b' && randomness == 1) {
 343         pieces['b'] = pieces['w'];
 344         flags += flags;
 345         break;
 346       }
 347
 348       let positions = ArrayFun.range(10);
 349
 350       // Get random squares for bishops
 351       let randIndex = 2 * randInt(5);
 352       let bishop1Pos = positions[randIndex];
 353       // The second bishop must be on a square of different color
 354       let randIndex_tmp = 2 * randInt(5) + 1;
 355       let bishop2Pos = positions[randIndex_tmp];
 356       // Remove chosen squares
 357       positions.splice(Math.max(randIndex, randIndex_tmp), 1);
 358       positions.splice(Math.min(randIndex, randIndex_tmp), 1);
 359
 360       // Get random squares for knights
 361       randIndex = randInt(8);
 362       let knight1Pos = positions[randIndex];
 363       positions.splice(randIndex, 1);
 364       randIndex = randInt(7);
 365       let knight2Pos = positions[randIndex];
 366       positions.splice(randIndex, 1);
 367
 368       // Get random square for queen
 369       randIndex = randInt(6);
 370       let queenPos = positions[randIndex];
 371       positions.splice(randIndex, 1);
 372
 373       // ...random square for marshall
 374       randIndex = randInt(5);
 375       let marshallPos = positions[randIndex];
 376       positions.splice(randIndex, 1);
 377
 378       // ...random square for cardinal
 379       randIndex = randInt(4);
 380       let cardinalPos = positions[randIndex];
 381       positions.splice(randIndex, 1);
 382
 383       // Rooks and king positions are now fixed,
 384       // because of the ordering rook-king-rook
 385       let rook1Pos = positions[0];
 386       let kingPos = positions[1];
 387       let rook2Pos = positions[2];
 388
 389       // Finally put the shuffled pieces in the board array
 390       pieces[c][rook1Pos] = "r";
 391       pieces[c][knight1Pos] = "n";
 392       pieces[c][bishop1Pos] = "b";
 393       pieces[c][queenPos] = "q";
 394       pieces[c][marshallPos] = "m";
 395       pieces[c][cardinalPos] = "c";
 396       pieces[c][kingPos] = "k";
 397       pieces[c][bishop2Pos] = "b";
 398       pieces[c][knight2Pos] = "n";
 399       pieces[c][rook2Pos] = "r";
 400       flags += V.CoordToColumn(rook1Pos) + V.CoordToColumn(rook2Pos);
 401     }
 402     return (
 403       pieces["b"].join("") +
 404       "/pppppppppp/91/91/91/91/91/91/PPPPPPPPPP/" +
 405       pieces["w"].join("").toUpperCase() +
 406       " w 0 " + flags + " - 00000000000000"
 407     );
 408   }
 409 };