Fix computer games for Apocalypse variant

[vchess.git] / client / src / variants / Threechecks.js

import { ChessRules } from "@/base_rules";

export class ThreechecksRules extends ChessRules {
  static IsGoodFlags(flags) {
    // 4 for castle + 2 for checks (0,1 or 2)
    return !!flags.match(/^[01]{4,4}[012]{2,2}$/);
  }

  setFlags(fenflags) {
    super.setFlags(fenflags); //castleFlags
    this.checkFlags = { w: 0, b: 0 };
    const flags = fenflags.substr(4); //skip first 4 digits, for castle
    for (let c of ["w", "b"]) {
      this.checkFlags[c] = parseInt(flags.charAt(c == "w" ? 0 : 1));
    }
  }

  aggregateFlags() {
    return [this.castleFlags, this.checkFlags];
  }

  disaggregateFlags(flags) {
    this.castleFlags = flags[0];
    this.checkFlags = flags[1];
  }

  getPpath(b) {
    // TODO: !!this.checkFlags condition for printDiagram, but clearly not good.
    // This is just a temporary fix.
    if (b[1] == 'k' && !!this.checkFlags && this.checkFlags[b[0]] > 0)
      return "Threechecks/" + b[0] + 'k_' + this.checkFlags[b[0]];
    return b;
  }

  postPlay(move) {
    super.postPlay(move);
    // Does this move give check?
    const oppCol = this.turn;
    if (this.underCheck(oppCol))
      this.checkFlags[oppCol]++;
  }

  getCurrentScore() {
    const color = this.turn;
    if (this.checkFlags[color] >= 3)
      return color == "w" ? "0-1" : "1-0";
    return super.getCurrentScore();
  }

  static GenRandInitFen(randomness) {
    // Add check flags (at 0)
    return ChessRules.GenRandInitFen(randomness).slice(0, -2) + "00";
  }

  getFlagsFen() {
    let fen = super.getFlagsFen();
    // Add check flags
    for (let c of ["w", "b"]) fen += this.checkFlags[c];
    return fen;
  }

  evalPosition() {
    const baseEval = super.evalPosition();
    // Take number of checks into account
    return baseEval/5 - this.checkFlags["w"] + this.checkFlags["b"];
  }
};