From d3077f8fea850f6246773d9bff73387a2c8d7c6f Mon Sep 17 00:00:00 2001 From: Yorick Barbanneau Date: Fri, 22 Dec 2023 16:18:44 +0100 Subject: [PATCH] Simplify weight heuristic calculation --- src/classes/Heuristic.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/src/classes/Heuristic.py b/src/classes/Heuristic.py index 6d71790..86d449c 100644 --- a/src/classes/Heuristic.py +++ b/src/classes/Heuristic.py @@ -77,18 +77,17 @@ class WeightHeuristicEngine(HeuristicEngine): def get_weight(self, board, player): score = 0 - size = self.options['size'] - w = [[0 for _ in range(size)] for _ in range(size)] for pos_x in range(self.options['size']): for pos_y in range(self.options['size']): + # Get player at x,y p = board._board[pos_x][pos_y] + if p == player: score += self.weights[pos_x][pos_y] - w[pos_x][pos_y] = self.weights[pos_x][pos_y] - elif p != player and p != board._EMPTY: + elif p != board._EMPTY: score -= self.weights[pos_x][pos_y] - w[pos_x][pos_y] = -self.weights[pos_x][pos_y] + return score """