Pandemonium 1 & 2, Stealthbomb 1 & 2
[vchess.git] / client / src / variants / Wildebeest.js
index a30e7f9..93353f9 100644 (file)
@@ -1,72 +1,70 @@
-import { ChessRules } from "@/base_rules";
+import { ChessRules, Move, PiPo } from "@/base_rules";
 import { ArrayFun } from "@/utils/array";
 import { sample, randInt } from "@/utils/alea";
 
-export const VariantRules = class  WildebeestRules extends ChessRules
-{
-  static getPpath(b)
-  {
-    return ([V.CAMEL,V.WILDEBEEST].includes(b[1]) ? "Wildebeest/" : "") + b;
-  }
+export class WildebeestRules extends ChessRules {
 
-  static get size() { return {x:10,y:11}; }
+  static get size() {
+    return { x: 10, y: 11 };
+  }
 
-  static get CAMEL() { return 'c'; }
-  static get WILDEBEEST() { return 'w'; }
+  static get CAMEL() {
+    return "c";
+  }
+  static get WILDEBEEST() {
+    return "w";
+  }
 
-  static get PIECES()
-  {
-    return ChessRules.PIECES.concat([V.CAMEL,V.WILDEBEEST]);
+  static get PIECES() {
+    return ChessRules.PIECES.concat([V.CAMEL, V.WILDEBEEST]);
   }
 
-  static get steps()
-  {
+  static get steps() {
     return Object.assign(
-      ChessRules.steps, //add camel moves:
-      {'c': [ [-3,-1],[-3,1],[-1,-3],[-1,3],[1,-3],[1,3],[3,-1],[3,1] ]}
+      {},
+      ChessRules.steps,
+      // Add camel moves:
+      {
+        c: [
+          [-3, -1],
+          [-3, 1],
+          [-1, -3],
+          [-1, 3],
+          [1, -3],
+          [1, 3],
+          [3, -1],
+          [3, 1]
+        ]
+      }
     );
   }
 
-  static IsGoodEnpassant(enpassant)
-  {
-    if (enpassant != "-")
-    {
-      const squares = enpassant.split(",");
-      if (squares.length > 2)
-        return false;
-      for (let sq of squares)
-      {
-        const ep = V.SquareToCoords(sq);
-        if (isNaN(ep.x) || !V.OnBoard(ep))
-          return false;
-      }
-    }
+  static IsGoodEnpassant(enpassant) {
+    if (enpassant != "-") return !!enpassant.match(/^([a-j][0-9]{1,2},?)+$/);
     return true;
   }
 
+  getPpath(b) {
+    return ([V.CAMEL, V.WILDEBEEST].includes(b[1]) ? "Wildebeest/" : "") + b;
+  }
+
   // There may be 2 enPassant squares (if pawn jump 3 squares)
-  getEnpassantFen()
-  {
+  getEnpassantFen() {
     const L = this.epSquares.length;
-    if (!this.epSquares[L-1])
-      return "-"; //no en-passant
+    if (!this.epSquares[L - 1]) return "-"; //no en-passant
     let res = "";
-    this.epSquares[L-1].forEach(sq => {
+    this.epSquares[L - 1].forEach(sq => {
       res += V.CoordsToSquare(sq) + ",";
     });
-    return res.slice(0,-1); //remove last comma
+    return res.slice(0, -1); //remove last comma
   }
 
   // En-passant after 2-sq or 3-sq jumps
-  getEpSquare(moveOrSquare)
-  {
-    if (!moveOrSquare)
-      return undefined;
-    if (typeof moveOrSquare === "string")
-    {
+  getEpSquare(moveOrSquare) {
+    if (!moveOrSquare) return undefined;
+    if (typeof moveOrSquare === "string") {
       const square = moveOrSquare;
-      if (square == "-")
-        return undefined;
+      if (square == "-") return undefined;
       let res = [];
       square.split(",").forEach(sq => {
         res.push(V.SquareToCoords(sq));
@@ -75,18 +73,19 @@ export const VariantRules = class  WildebeestRules extends ChessRules
     }
     // Argument is a move:
     const move = moveOrSquare;
-    const [sx,sy,ex] = [move.start.x,move.start.y,move.end.x];
-    if (this.getPiece(sx,sy) == V.PAWN && Math.abs(sx - ex) >= 2)
-    {
-      const step = (ex-sx) / Math.abs(ex-sx);
-      let res = [{
-        x: sx + step,
-        y: sy
-      }];
-      if (sx + 2*step != ex) //3-squares move
-      {
+    const [sx, sy, ex] = [move.start.x, move.start.y, move.end.x];
+    if (this.getPiece(sx, sy) == V.PAWN && Math.abs(sx - ex) >= 2) {
+      const step = (ex - sx) / Math.abs(ex - sx);
+      let res = [
+        {
+          x: sx + step,
+          y: sy
+        }
+      ];
+      if (sx + 2 * step != ex) {
+        // 3-squares move
         res.push({
-          x: sx + 2*step,
+          x: sx + 2 * step,
           y: sy
         });
       }
@@ -95,85 +94,83 @@ export const VariantRules = class  WildebeestRules extends ChessRules
     return undefined; //default
   }
 
-  getPotentialMovesFrom([x,y])
-  {
-    switch (this.getPiece(x,y))
-    {
+  getPotentialMovesFrom([x, y]) {
+    switch (this.getPiece(x, y)) {
       case V.CAMEL:
-        return this.getPotentialCamelMoves([x,y]);
+        return this.getPotentialCamelMoves([x, y]);
       case V.WILDEBEEST:
-        return this.getPotentialWildebeestMoves([x,y]);
+        return this.getPotentialWildebeestMoves([x, y]);
       default:
-        return super.getPotentialMovesFrom([x,y])
+        return super.getPotentialMovesFrom([x, y]);
     }
   }
 
   // Pawns jump 2 or 3 squares, and promote to queen or wildebeest
-  getPotentialPawnMoves([x,y])
-  {
+  getPotentialPawnMoves([x, y]) {
     const color = this.turn;
     let moves = [];
-    const [sizeX,sizeY] = [V.size.x,V.size.y];
-    const shiftX = (color == "w" ? -1 : 1);
-    const startRanks = (color == "w" ? [sizeX-2,sizeX-3] : [1,2]);
-    const lastRank = (color == "w" ? 0 : sizeX-1);
-    const finalPieces = x + shiftX == lastRank
-      ? [V.ROOK,V.KNIGHT,V.BISHOP,V.QUEEN]
-      : [V.PAWN];
-
-    if (this.board[x+shiftX][y] == V.EMPTY)
-    {
+    const [sizeX, sizeY] = [V.size.x, V.size.y];
+    const shiftX = color == "w" ? -1 : 1;
+    const startRanks = color == "w" ? [sizeX - 2, sizeX - 3] : [1, 2];
+    const lastRanks = color == "w" ? [0, 1] : [sizeX - 1, sizeX  -2];
+    let finalPieces = [V.PAWN];
+    if (x + shiftX == lastRanks[1])
+      Array.prototype.push.apply(finalPieces, [V.WILDEBEEST, V.QUEEN]);
+    else if (x + shiftX == lastRanks[0])
+      finalPieces = [V.WILDEBEEST, V.QUEEN];
+
+    if (this.board[x + shiftX][y] == V.EMPTY) {
       // One square forward
       for (let piece of finalPieces)
-        moves.push(this.getBasicMove([x,y], [x+shiftX,y], {c:color,p:piece}));
-      if (startRanks.includes(x))
-      {
-        if (this.board[x+2*shiftX][y] == V.EMPTY)
-        {
+        moves.push(
+          this.getBasicMove([x, y], [x + shiftX, y], { c: color, p: piece })
+        );
+      if (startRanks.includes(x)) {
+        if (this.board[x + 2 * shiftX][y] == V.EMPTY) {
           // Two squares jump
-          moves.push(this.getBasicMove([x,y], [x+2*shiftX,y]));
-          if (x==startRanks[0] && this.board[x+3*shiftX][y] == V.EMPTY)
-          {
+          moves.push(this.getBasicMove([x, y], [x + 2 * shiftX, y]));
+          if (x == startRanks[0] && this.board[x + 3 * shiftX][y] == V.EMPTY) {
             // Three squares jump
-            moves.push(this.getBasicMove([x,y], [x+3*shiftX,y]));
+            moves.push(this.getBasicMove([x, y], [x + 3 * shiftX, y]));
           }
         }
       }
     }
     // Captures
-    for (let shiftY of [-1,1])
-    {
-      if (y + shiftY >= 0 && y + shiftY < sizeY
-        && this.board[x+shiftX][y+shiftY] != V.EMPTY
-        && this.canTake([x,y], [x+shiftX,y+shiftY]))
-      {
-        for (let piece of finalPieces)
-        {
-          moves.push(this.getBasicMove([x,y], [x+shiftX,y+shiftY],
-            {c:color,p:piece}));
+    for (let shiftY of [-1, 1]) {
+      if (
+        y + shiftY >= 0 &&
+        y + shiftY < sizeY &&
+        this.board[x + shiftX][y + shiftY] != V.EMPTY &&
+        this.canTake([x, y], [x + shiftX, y + shiftY])
+      ) {
+        for (let piece of finalPieces) {
+          moves.push(
+            this.getBasicMove([x, y], [x + shiftX, y + shiftY], {
+              c: color,
+              p: piece
+            })
+          );
         }
       }
     }
 
     // En passant
     const Lep = this.epSquares.length;
-    const epSquare = this.epSquares[Lep-1];
-    if (!!epSquare)
-    {
-      for (let epsq of epSquare)
-      {
+    const epSquare = this.epSquares[Lep - 1];
+    if (!!epSquare) {
+      for (let epsq of epSquare) {
         // TODO: some redundant checks
-        if (epsq.x == x+shiftX && Math.abs(epsq.y - y) == 1)
-        {
-          var enpassantMove = this.getBasicMove([x,y], [epsq.x,epsq.y]);
+        if (epsq.x == x + shiftX && Math.abs(epsq.y - y) == 1) {
+          let enpassantMove = this.getBasicMove([x, y], [epsq.x, epsq.y]);
           // WARNING: the captured pawn may be diagonally behind us,
           // if it's a 3-squares jump and we take on 1st passing square
-          const px = (this.board[x][epsq.y] != V.EMPTY ? x : x - shiftX);
+          const px = this.board[x][epsq.y] != V.EMPTY ? x : x - shiftX;
           enpassantMove.vanish.push({
             x: px,
             y: epsq.y,
-            p: 'p',
-            c: this.getColor(px,epsq.y)
+            p: "p",
+            c: this.getColor(px, epsq.y)
           });
           moves.push(enpassantMove);
         }
@@ -183,76 +180,172 @@ export const VariantRules = class  WildebeestRules extends ChessRules
     return moves;
   }
 
-  // TODO: wildebeest castle
-
-  getPotentialCamelMoves(sq)
-  {
+  getPotentialCamelMoves(sq) {
     return this.getSlideNJumpMoves(sq, V.steps[V.CAMEL], "oneStep");
   }
 
-  getPotentialWildebeestMoves(sq)
-  {
+  getPotentialWildebeestMoves(sq) {
     return this.getSlideNJumpMoves(
-      sq, V.steps[V.KNIGHT].concat(V.steps[V.CAMEL]), "oneStep");
+      sq,
+      V.steps[V.KNIGHT].concat(V.steps[V.CAMEL]),
+      "oneStep"
+    );
   }
 
-  isAttacked(sq, colors)
-  {
-    return super.isAttacked(sq, colors)
-      || this.isAttackedByCamel(sq, colors)
-      || this.isAttackedByWildebeest(sq, colors);
+  getPPpath(m) {
+    if (
+      m.appear.length == 2 && m.vanish.length == 2 &&
+      Math.abs(m.end.y - m.start.y) == 1 &&
+      this.board[m.end.x][m.end.y] == V.EMPTY
+    ) {
+      // Castle, king moved by one square only, not directly onto rook
+      return "Wildebeest/castle";
+    }
+    return super.getPPpath(m);
   }
 
-  isAttackedByCamel(sq, colors)
-  {
-    return this.isAttackedBySlideNJump(sq, colors,
-      V.CAMEL, V.steps[V.CAMEL], "oneStep");
+  // Special Wildebeest castling rules:
+  getCastleMoves([x, y]) {
+    const c = this.getColor(x, y);
+    const oppCol = V.GetOppCol(c);
+    let moves = [];
+    let i = 0;
+    const castlingKing = this.board[x][y].charAt(1);
+    castlingCheck: for (
+      let castleSide = 0;
+      castleSide < 2;
+      castleSide++ //"large", then "small"
+    ) {
+      if (this.castleFlags[c][castleSide] >= V.size.y) continue;
+      // Rook and king are on initial position
+      const rookPos = this.castleFlags[c][castleSide];
+      const range = (castleSide == 0 ? [rookPos, y] : [y, rookPos]);
+
+      // King and rook must be connected:
+      for (let i = range[0] + 1; i <= range[1] - 1; i++) {
+        if (this.board[x][i] != V.EMPTY) continue castlingCheck;
+      }
+      const step = 2 * castleSide - 1;
+      // No attacks on the path of the king ?
+      for (let i = range[0]; i <= range[1]; i++) {
+        if (i != rookPos && this.isAttacked([x, i], oppCol))
+          continue castlingCheck;
+        if (
+          i != y &&
+          // Do not end in the corner, except if starting square is too near
+          (i > 0 || y == 1) &&
+          (i < V.size.y - 1 || y == V.size.y - 2)
+        ) {
+          // Found a possible castle move:
+          moves.push(
+            new Move({
+              appear: [
+                new PiPo({
+                  x: x,
+                  y: i,
+                  p: V.KING,
+                  c: c
+                }),
+                new PiPo({
+                  x: x,
+                  y: i - step,
+                  p: V.ROOK,
+                  c: c
+                })
+              ],
+              vanish: [
+                new PiPo({ x: x, y: y, p: V.KING, c: c }),
+                new PiPo({ x: x, y: rookPos, p: V.ROOK, c: c })
+              ]
+            })
+          );
+        }
+      }
+    }
+
+    return moves;
+  }
+
+  isAttacked(sq, color) {
+    return (
+      super.isAttacked(sq, color) ||
+      this.isAttackedByCamel(sq, color) ||
+      this.isAttackedByWildebeest(sq, color)
+    );
   }
 
-  isAttackedByWildebeest(sq, colors)
-  {
-    return this.isAttackedBySlideNJump(sq, colors, V.WILDEBEEST,
-      V.steps[V.KNIGHT].concat(V.steps[V.CAMEL]), "oneStep");
+  isAttackedByCamel(sq, color) {
+    return this.isAttackedBySlideNJump(
+      sq,
+      color,
+      V.CAMEL,
+      V.steps[V.CAMEL],
+      "oneStep"
+    );
   }
 
-  getCurrentScore()
-  {
-    if (this.atLeastOneMove()) // game not over
-      return "*";
+  isAttackedByWildebeest(sq, color) {
+    return this.isAttackedBySlideNJump(
+      sq,
+      color,
+      V.WILDEBEEST,
+      V.steps[V.KNIGHT].concat(V.steps[V.CAMEL]),
+      "oneStep"
+    );
+  }
 
+  getCurrentScore() {
+    if (this.atLeastOneMove()) return "*";
     // No valid move: game is lost (stalemate is a win)
-    return (this.turn == "w" ? "0-1" : "1-0");
+    return this.turn == "w" ? "0-1" : "1-0";
   }
 
   static get VALUES() {
     return Object.assign(
-      ChessRules.VALUES,
-      {'c': 3, 'w': 7} //experimental
+      { c: 3, w: 7 }, //experimental
+      ChessRules.VALUES
     );
   }
 
-  static get SEARCH_DEPTH() { return 2; }
+  static get SEARCH_DEPTH() {
+    return 2;
+  }
+
+  static GenRandInitFen(randomness) {
+    if (randomness == 0) {
+      return (
+        "rnccwkqbbnr/ppppppppppp/92/92/92/92/92/92/PPPPPPPPPPP/RNBBQKWCCNR " +
+        "w 0 akak -"
+      );
+    }
+
+    let pieces = { w: new Array(11), b: new Array(11) };
+    let flags = "";
+    for (let c of ["w", "b"]) {
+      if (c == 'b' && randomness == 1) {
+        pieces['b'] = pieces['w'];
+        flags += flags;
+        break;
+      }
 
-  static GenRandInitFen()
-  {
-    let pieces = { "w": new Array(10), "b": new Array(10) };
-    for (let c of ["w","b"])
-    {
       let positions = ArrayFun.range(11);
 
       // Get random squares for bishops + camels (different colors)
-      let randIndexes = sample(ArrayFun.range(6), 2)
-        .map(i => { return 2*i; });
+      let randIndexes = sample(ArrayFun.range(6), 2).map(i => {
+        return 2 * i;
+      });
       let bishop1Pos = positions[randIndexes[0]];
       let camel1Pos = positions[randIndexes[1]];
       // The second bishop (camel) must be on a square of different color
-      let randIndexes_tmp = sample(ArrayFun.range(5), 2)
-        .map(i => { return 2*i+1; });
+      let randIndexes_tmp = sample(ArrayFun.range(5), 2).map(i => {
+        return 2 * i + 1;
+      });
       let bishop2Pos = positions[randIndexes_tmp[0]];
       let camel2Pos = positions[randIndexes_tmp[1]];
-      for (let idx of randIndexes.concat(randIndexes_tmp)
-        .sort((a,b) => { return b-a; })) //largest indices first
-      {
+      for (let idx of randIndexes.concat(randIndexes_tmp).sort((a, b) => {
+        return b - a;
+      })) {
+        // Largest indices first
         positions.splice(idx, 1);
       }
 
@@ -276,21 +369,25 @@ export const VariantRules = class  WildebeestRules extends ChessRules
       let kingPos = positions[1];
       let rook2Pos = positions[2];
 
-      pieces[c][rook1Pos] = 'r';
-      pieces[c][knight1Pos] = 'n';
-      pieces[c][bishop1Pos] = 'b';
-      pieces[c][queenPos] = 'q';
-      pieces[c][camel1Pos] = 'c';
-      pieces[c][camel2Pos] = 'c';
-      pieces[c][wildebeestPos] = 'w';
-      pieces[c][kingPos] = 'k';
-      pieces[c][bishop2Pos] = 'b';
-      pieces[c][knight2Pos] = 'n';
-      pieces[c][rook2Pos] = 'r';
+      pieces[c][rook1Pos] = "r";
+      pieces[c][knight1Pos] = "n";
+      pieces[c][bishop1Pos] = "b";
+      pieces[c][queenPos] = "q";
+      pieces[c][camel1Pos] = "c";
+      pieces[c][camel2Pos] = "c";
+      pieces[c][wildebeestPos] = "w";
+      pieces[c][kingPos] = "k";
+      pieces[c][bishop2Pos] = "b";
+      pieces[c][knight2Pos] = "n";
+      pieces[c][rook2Pos] = "r";
+      flags += V.CoordToColumn(rook1Pos) + V.CoordToColumn(rook2Pos);
     }
-    return pieces["b"].join("") +
-      "/ppppppppppp/11/11/11/11/11/11/PPPPPPPPPPP/" +
+    return (
+      pieces["b"].join("") +
+      "/ppppppppppp/92/92/92/92/92/92/PPPPPPPPPPP/" +
       pieces["w"].join("").toUpperCase() +
-      " w 0 1111 -";
+      " w 0 " + flags + " -"
+    );
   }
-}
+
+};