Add Hypnotic + Mesmer variants
[vchess.git] / client / src / variants / Hypnotic.js
1 import { ChessRules } from "@/base_rules";
2
3 export class HypnoticRules extends ChessRules {
4
5 static IsGoodFen(fen) {
6 if (!ChessRules.IsGoodFen(fen)) return false;
7 const fenParsed = V.ParseFen(fen);
8 // 5) Check arrival of last hypnotizing move (if any)
9 if (
10 !fenParsed.hSquare ||
11 (fenParsed.hSquare != "-" && !fenParsed.hSquare.match(/^[a-h][1-8]$/))
12 ) {
13 return false;
14 }
15 return true;
16 }
17
18 static ParseFen(fen) {
19 const fenParts = fen.split(" ");
20 return Object.assign(
21 { hSquare: fenParts[5] },
22 ChessRules.ParseFen(fen)
23 );
24 }
25
26 static GenRandInitFen(randomness) {
27 return ChessRules.GenRandInitFen(randomness) + " -";
28 }
29
30 setOtherVariables(fen) {
31 super.setOtherVariables(fen);
32 const parsedFen = V.ParseFen(fen);
33 this.hSquares = [
34 parsedFen.hSquare != "-"
35 ? V.SquareToCoords(parsedFen.hSquare)
36 : null
37 ];
38 }
39
40 getFen() {
41 const L = this.hSquares.length;
42 return (
43 super.getFen() + " " +
44 (!this.hSquares[L-1] ? "-" : V.CoordsToSquare(this.hSquares[L-1]))
45 );
46 }
47
48 canIplay(side) {
49 // Wrong, but sufficient approximation let's say
50 return this.turn == side;
51 }
52
53 canTake([x1, y1], [x2, y2]) {
54 const c = this.turn;
55 const c1 = this.getColor(x1, y1);
56 const c2 = this.getColor(x2, y2);
57 return (c == c1 && c1 != c2) || (c != c1 && c1 == c2);
58 }
59
60 getPotentialMovesFrom([x, y]) {
61 const L = this.hSquares.length;
62 const lh = this.hSquares[L-1];
63 if (!!lh && lh.x == x && lh.y == y) return [];
64 const c = this.getColor(x, y);
65 if (c == this.turn) return super.getPotentialMovesFrom([x, y]);
66 // Playing opponent's pieces: hypnotizing moves. Allowed?
67 if (!this.isAttacked([x, y], this.turn)) return [];
68 const moves =
69 this.getPiece(x, y) == V.KING
70 // No castling with enemy king (...yes, should eat it but...)
71 ? super.getSlideNJumpMoves(
72 [x, y], V.steps[V.ROOK].concat(V.steps[V.BISHOP]), "oneStep")
73 : super.getPotentialMovesFrom([x, y]);
74 return moves;
75 }
76
77 getEnpassantCaptures([x, y], shiftX) {
78 const Lep = this.epSquares.length;
79 const epSquare = this.epSquares[Lep - 1]; //always at least one element
80 let enpassantMove = null;
81 const c = this.getColor(x, y);
82 if (
83 !!epSquare &&
84 epSquare.x == x + shiftX &&
85 Math.abs(epSquare.y - y) == 1 &&
86 // Next conditions to avoid capturing self hypnotized pawns:
87 this.board[x][epSquare.y] != V.EMPTY &&
88 this.getColor(x, epSquare.y) != c //TODO: probably redundant
89 ) {
90 enpassantMove = this.getBasicMove([x, y], [epSquare.x, epSquare.y]);
91 enpassantMove.vanish.push({
92 x: x,
93 y: epSquare.y,
94 p: this.board[x][epSquare.y].charAt(1),
95 c: this.getColor(x, epSquare.y)
96 });
97 }
98 return !!enpassantMove ? [enpassantMove] : [];
99 }
100
101 // TODO: avoid following code duplication, by using getColor()
102 // instead of this.turn at the beginning of 2 next methods
103 addPawnMoves([x1, y1], [x2, y2], moves, promotions) {
104 let finalPieces = [V.PAWN];
105 const color = this.getColor(x1, y1);
106 const lastRank = (color == "w" ? 0 : V.size.x - 1);
107 if (x2 == lastRank) finalPieces = V.PawnSpecs.promotions;
108 let tr = null;
109 for (let piece of finalPieces) {
110 tr = (piece != V.PAWN ? { c: color, p: piece } : null);
111 moves.push(this.getBasicMove([x1, y1], [x2, y2], tr));
112 }
113 }
114
115 getPotentialPawnMoves([x, y], promotions) {
116 const color = this.getColor(x, y);
117 const [sizeX, sizeY] = [V.size.x, V.size.y];
118 const forward = (color == 'w' ? -1 : 1);
119
120 let moves = [];
121 if (x + forward >= 0 && x + forward < sizeX) {
122 if (this.board[x + forward][y] == V.EMPTY) {
123 this.addPawnMoves([x, y], [x + forward, y], moves, promotions);
124 if (
125 ((color == 'w' && x == 6) || (color == 'b' && x == 1)) &&
126 this.board[x + 2 * forward][y] == V.EMPTY
127 ) {
128 moves.push(this.getBasicMove([x, y], [x + 2 * forward, y]));
129 }
130 }
131 for (let shiftY of [-1, 1]) {
132 if (
133 y + shiftY >= 0 && y + shiftY < sizeY &&
134 this.board[x + forward][y + shiftY] != V.EMPTY &&
135 this.canTake([x, y], [x + forward, y + shiftY])
136 ) {
137 this.addPawnMoves(
138 [x, y], [x + forward, y + shiftY],
139 moves, promotions
140 );
141 }
142 }
143 }
144 Array.prototype.push.apply(moves,
145 this.getEnpassantCaptures([x, y], forward));
146 return moves;
147 }
148
149 postPlay(move) {
150 super.postPlay(move);
151 if (move.vanish[0].c == this.turn)
152 this.hSquares.push({ x: move.appear[0].x, y: move.appear[0].y });
153 else this.hSquares.push(null);
154 if (move.vanish.length == 2 && move.vanish[1].p == V.KING)
155 this.kingPos[move.vanish[1].c] = [-1, -1];
156 }
157 postUndo(move) {
158 super.postUndo(move);
159 this.hSquares.pop();
160 if (move.vanish.length == 2 && move.vanish[1].p == V.KING)
161 this.kingPos[move.vanish[1].c] = [move.vanish[1].x, move.vanish[1].y];
162 }
163
164 getCheckSquares() {
165 return [];
166 }
167 filterValid(moves) {
168 return moves;
169 }
170
171 getCurrentScore() {
172 const c = this.turn;
173 if (this.kingPos[c][0] < 0) return (c == 'w' ? "0-1" : "1-0");
174 return "*";
175 }
176
177 static get SEARCH_DEPTH() {
178 return 2;
179 }
180
181 };