fixed win

bananasmoothii · Oct 12, 2023 · 5e29964 · 5e29964
1 parent 7bbe29d
commit 5e29964
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/game/connect4.rs b/src/game/connect4.rs
@@ -359,6 +359,13 @@ impl Game for ConnectFour {
      * Scores are invalid if the line cannot be completed
      */
     fn get_score(&self, player: Self::Player) -> Self::Score {
+        if let Some(winner) = self.winner {
+            return if winner == player {
+                Self::Score::MAX
+            } else {
+                Self::Score::MIN
+            }
+        }
         let p1_score = self.calculate_score(self.p1_aligns2, self.p1_aligns3)
             - self.calculate_score(self.p2_aligns2, self.p2_aligns3);
         if player.get() == 1u8 {