X-Git-Url: https://git.auder.net/?a=blobdiff_plain;f=public%2Fjavascripts%2Fvariants%2FAtomic.js;h=2adb98842aeb14316249abdafdf6813ab046d8df;hb=f6dbe8e31a3260487664f1e0b50710b3f3efaf5f;hp=666c50b214406656bc04a258e6f75c448e5a2b75;hpb=388e4c401f05b1f6a4c54e33c9da9114969a53c0;p=vchess.git

diff --git a/public/javascripts/variants/Atomic.js b/public/javascripts/variants/Atomic.js
index 666c50b2..2adb9884 100644
--- a/public/javascripts/variants/Atomic.js
+++ b/public/javascripts/variants/Atomic.js
@@ -102,37 +102,30 @@ class AtomicRules extends ChessRules
 		}
 	}
 
-	underCheck(move)
+	underCheck(color)
 	{
-		const c = this.turn;
-		const oppCol = this.getOppCol(c);
-		this.play(move);
+		const oppCol = this.getOppCol(color);
 		let res = undefined;
 		// If our king disappeared, move is not valid
-		if (this.kingPos[c][0] < 0)
+		if (this.kingPos[color][0] < 0)
 			res = true;
 		// If opponent king disappeared, move is valid
 		else if (this.kingPos[oppCol][0] < 0)
 			res = false;
 		// Otherwise, if we remain under check, move is not valid
 		else
-			res = this.isAttacked(this.kingPos[c], [oppCol]);
-		this.undo(move);
+			res = this.isAttacked(this.kingPos[color], [oppCol]);
 		return res;
 	}
 
-	getCheckSquares(move)
+	getCheckSquares(color)
 	{
-		const c = this.getOppCol(this.turn);
-		// King might explode:
-		const saveKingPos = JSON.parse(JSON.stringify(this.kingPos[c]));
-		this.play(move);
 		let res = [ ];
-		if (this.kingPos[c][0] < 0)
-			res = [saveKingPos];
-		else if (this.isAttacked(this.kingPos[c], [this.getOppCol(c)]))
-			res = [ JSON.parse(JSON.stringify(this.kingPos[c])) ]
-		this.undo(move);
+		if (this.kingPos[color][0] >= 0 //king might have exploded
+			&& this.isAttacked(this.kingPos[color], [this.getOppCol(color)]))
+		{
+			res = [ JSON.parse(JSON.stringify(this.kingPos[color])) ]
+		}
 		return res;
 	}