Fix Weiqi

[xogo.git] / variants / Bario / class.js

import ChessRules from "/base_rules.js";
import PiPo from "/utils/PiPo.js";
import Move from "/utils/Move.js";

export default class BarioRules extends ChessRules {

  static get Options() {
    return {
      select: C.Options.select,
      input: C.Options.input,
      styles: [
        "atomic", "cannibal", "capture", "cylinder",
        "dark", "madrasi", "rifle", "teleport"
      ]
    };
  }

  // Does not really seem necessary (although the author mention it)
  // Instead, first move = pick a square for the king.
  get hasFlags() {
    return false;
  }
  get hasReserve() {
    return true;
  }

  pieces(color, x, y) {
    return Object.assign(
      {
        'u': {
          "class": "undefined",
          moves: []
        }
      },
      super.pieces(color, x, y)
    );
  }

  get onlyClick() {
    return this.movesCount <= 1;
  }

  // Initiate the game by choosing a square for the king:
  doClick(coords) {
    const color = this.turn;
    if (
      this.movesCount <= 1 &&
      (
        (color == 'w' && coords.x == this.size.x - 1) ||
        (color == 'b' && coords.x == 0)
      )
    ) {
      return new Move({
        appear: [ new PiPo({x: coords.x, y: coords.y, c: color, p: 'k' }) ],
        vanish: [ new PiPo({x: coords.x, y: coords.y, c: color, p: 'u' }) ]
      });
    }
    return null;
  }

  genRandInitBaseFen() {
    return {
      fen: "uuuuuuuu/pppppppp/8/8/8/8/PPPPPPPP/UUUUUUUU",
      o: {}
    }
  }

  getPartFen(o) {
    return Object.assign(
      {
        captureUndef: (o.init || !this.captureUndef)
          ? "-"
          : C.CoordsToSquare(this.captureUndef)
      },
      super.getPartFen(o)
    );
  }

  getReserveFen(o) {
    if (o.init)
      return "22212221";
    return (
      ["w","b"].map(c => Object.values(this.reserve[c]).join("")).join("")
    );
  }

  initReserves(reserveStr) {
    super.initReserves(reserveStr, ['r', 'n', 'b', 'q']);
  }

  setOtherVariables(fenParsed) {
    super.setOtherVariables(fenParsed);
    this.captureUndef = fenParsed.captureUndef == '-'
      ? null :
      C.SquareToCoords(fenParsed.captureUndef);
    this.definition = null;
  }

  canDrop([c, p], [i, j]) {
    switch (this.subTurn) {
      case 0:
        return i == this.captureUndef.x && j == this.captureUndef.y;
      case 1:
        return this.getPiece(i, j) == 'u' && c == this.getColor(i, j);
    }
    return false; //never reached
  }

  getPotentialMovesFrom([x, y]) {
    if (this.movesCount <= 1)
      return [];
    let moves = [];
    switch (this.subTurn) {
      case 0:
        if (typeof x == "string")
          moves = this.getDropMovesFrom([x, y]);
        // Empty move: just start + end
        moves.forEach(m => {m.vanish.pop(); m.appear.pop();});
        break;
      case 1:
        // Both normal move (from defined piece) and definition allowed
        if (typeof x == "string")
          moves = this.getDropMovesFrom([x, y]);
        else if (this.getPiece(x, y) != 'u')
          moves = super.getPotentialMovesFrom([x, y]);
        break;
      case 2:
        // We can only move the just-defined piece
        if (x == this.definition.x && y == this.definition.y)
          moves = super.getPotentialMovesFrom([x, y]);
        break;
    }
    return moves;
  }

  filterValid(moves) {
    if (this.movesCount <= 1 || this.subTurn == 0)
      return moves;
    let filterLater = [];
    if (this.subTurn == 1) {
      // Remove defining moves with un-movable def piece,
      // and separate compatible definitions.
      moves = moves.filter(m => {
        if (m.vanish.length >= 2 || m.vanish[0].p != 'u')
          return true;
        this.playOnBoard(m);
        const canMove = super.filterValid(
          super.getPotentialMovesFrom([m.end.x, m.end.y])).length >= 1;
        this.undoOnBoard(m);
        if (canMove)
          filterLater.push(m);
        return false;
      });
    }
    return super.filterValid(moves).concat(filterLater);
  }

  atLeastOneMove(color) {
    if (this.subTurn != 1)
      return true;
    return super.atLeastOneMove(color);
  }

  underCheck(square_s, oppCol) {
    if (super.underCheck(square_s, oppCol))
      return true;
    // Check potential specializations of undefined using reserve:
    const allAttacks = Array.prototype.concat.apply(
      ['r', 'n', 'b', 'q'].map(p => this.pieces()[p].moves[0]));
    const [x, y] = square_s[0];
    for (let i=0; i<this.size.x; i++) {
      for (let j=0; j<this.size.y; j++) {
        if (
          this.board[i][j] != "" &&
          this.getColor(i, j) == oppCol &&
          this.getPiece(i, j) == 'u'
        ) {
          for (let stepDef of allAttacks) {
            for (let s of stepDef.steps) {
              if (!super.compatibleStep([i, j], [x, y], s, stepDef.range))
                continue;
              if (
                super.findDestSquares(
                  [i, j],
                  {
                    captureTarget: [x, y],
                    captureSteps: [{steps: [s], range: stepDef.range}],
                    segments: false,
                    attackOnly: true,
                    one: false
                  }
                )
              ) {
                return true;
              }
            }
          }
        }
      }
    }
    return false;
  }

  postPlay(move) {
    const color = this.turn;
    if (this.movesCount <= 1 || move.reset || move.next) {
      if (!move.next)
        this.tryChangeTurn();
      return;
    }
    if (this.subTurn == 0)
      this.captureUndef = null; //already used
    const captureUndef = (
      move.vanish.length == 2 && //exclude subTurn == 0
      move.vanish[1].c != color &&
      move.vanish[1].p == 'u'
    );
    if (typeof move.start.x == "number" && !captureUndef)
      // Normal move (including Teleport)
      super.postPlay(move);
    else if (typeof move.start.x == "string") {
      super.updateReserve(
        color, move.start.y, this.reserve[color][move.start.y] - 1);
      if (move.vanish.length == 1 && move.vanish[0].p == 'u')
        this.definition = move.end;
      this.subTurn++;
    }
    else {
      this.subTurn = 0;
      this.captureUndef = move.end;
      this.tryChangeTurn(null, captureUndef);
    }
  }

  // NOTE: not "trying", the turn always change here (TODO?)
  tryChangeTurn(move, captureUndef) {
    this.definition = null;
    this.subTurn = captureUndef ? 0 : 1;
    this.turn = C.GetOppCol(this.turn);
    this.movesCount++;
  }

  computeNextMove(move) {
    if (
      !this.definition || this.playerColor != this.turn ||
      this.board.some(row => row.some(cell =>
        cell.charAt(0) == this.turn && cell.charAt(1) == 'u'))
    ) {
      return;
    }
    const variety = (c) => {
      return (
        [...new Set(
          Array.prototype.concat.apply([],
            this.board.map(row =>
              row.filter(cell =>
                cell.charAt(0) == c && !['p', 'k'].includes(cell.charAt(1))
              ).map(cell => cell.charAt(1))
            )
          )
        )].length >= 2
      );
    };
    let next = {start: move.end, end: move.end, vanish: [], appear: []};
    this.playOnBoard(move);
    const twoOrMorePieces = {w: variety('w'), b: variety('b')};
    const resetCols =
      Object.keys(twoOrMorePieces).filter(k => twoOrMorePieces[k]);
    if (resetCols.length >= 1) {
      for (let i=0; i<this.size.x; i++) {
        for (let j=0; j<this.size.y; j++) {
          const colIJ = this.getColor(i, j);
          const pieceIJ = this.getPiece(i, j);
          if (
            resetCols.includes(colIJ) &&
            this.board[i][j] != "" &&
            !['p', 'k', 'u'].includes(pieceIJ)
          ) {
            // NOTE: could also use a "flip" strategy similar to Benedict
            next.vanish.push(new PiPo({c: colIJ, p: pieceIJ, x: i, y: j}));
            next.appear.push(new PiPo({c: colIJ, p: 'u', x: i, y: j}));
            this.reserve[colIJ][pieceIJ]++;
          }
        }
      }
      super.re_drawReserve(resetCols);
    }
    this.undoOnBoard(move);
    if (next.vanish.length >= 1) {
      next.reset = true;
      move.next = next;
    }
  }

  isLastMove() {
    return true; //called only on normal moves (not Teleport)
  }

  getCurrentScore(move_s) {
    return (this.movesCount <= 2 ? "*" : super.getCurrentScore(move_s));
  }

};