Reenables another pin test.

aryann · Nov 11, 2024 · ba869b4 · ba869b4
1 parent a6606f2
commit ba869b4
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 14 deletions.
diff --git a/engine/src/moves.test.ts b/engine/src/moves.test.ts
@@ -1071,7 +1071,7 @@ describe("castling", () => {
     ]);
   });
 
-  it.skip("pinned pieces", () => {
+  it("pinned pieces", () => {
     const generator = make("K7/8/8/8/8/5Q2/8/7q");
 
     assert.sameDeepMembers(generator.generateMoves("f3"), [
@@ -1095,6 +1095,16 @@ describe("castling", () => {
         to: "b7",
         type: "normal",
       },
+      {
+        from: "f3",
+        to: "g2",
+        type: "normal",
+      },
+      {
+        from: "f3",
+        to: "h1",
+        type: "normal",
+      },
     ]);
   });
 });
diff --git a/engine/src/moves.ts b/engine/src/moves.ts
@@ -14,6 +14,7 @@ import {
   UP_LEFT,
   UP_RIGHT,
 } from "./offsets";
+import { pins, TAllowedMoves } from "./pins";
 import {
   getRank,
   getSide,
@@ -53,23 +54,21 @@ export class MoveGenerator {
     }
 
     // TODO(aryann): Remove this check once all test boards have kings on them.
-    // const side = getSide(piece);
-    // let pinnedPieces: Map<TSquare, TAllowedMoves> = new Map();
-    // try {
-    //   const kingSquare = this.board.getKingSquare(side);
-    //   pinnedPieces = pins(this.board, kingSquare);
-    // } catch (e) {}
+    const side = getSide(piece);
+    let pinnedPieces: Map<TSquare, TAllowedMoves> = new Map();
+    try {
+      const kingSquare = this.board.getKingSquare(side);
+      pinnedPieces = pins(this.board, kingSquare);
+    } catch (e) {}
 
     const moves: TMove[] = [];
 
     for (const move of this.generatePseudoLegalMoves(from)) {
-      // TODO(aryann): Enable this once the pinning logic works:
-      //
-      // const pinnedPiece = pinnedPieces.get(move.from);
-      // if (pinnedPiece && !pinnedPiece.includes(move.to)) {
-      //   // Pinned pieces cannot be moved.
-      //   continue;
-      // }
+      const pinnedPiece = pinnedPieces.get(move.from);
+      if (pinnedPiece && !pinnedPiece.includes(move.to)) {
+        // Pinned pieces cannot be moved.
+        continue;
+      }
 
       // TODO(aryann): If the king is currently in check, exclude all moves that
       // do not take the king out of check.