fixed node again

1 files changed, 11 insertions, 2 deletions

diff --git a/python/alphaBeta.py b/python/alphaBeta.py
index b35a7f2..be44c85 100644
--- a/python/alphaBeta.py
+++ b/python/alphaBeta.py
@@ -12,10 +12,19 @@ class GameController(object):
 class ABNode(object): # A class for Node related operations
     def __init__(self, state):
         self.Current = state
-        self.CurrentScore = self.Current.score[1]-self.Current.score[2]
+        if state.player == 1:
+            self.CurrentScore = self.Current.score[2]-self.Current.score[1]
+        elif state.player ==2:
+            self.CurrentScore = self.Current.score[1]-self.Current.score[2]
         self.children = {}
+
+
     def update_score(self):
-         self.CurrentScore = self.Current.score[1]-self.Current.score[2]
+        if self.Current.player == 2:
+            self.CurrentScore = self.Current.score[1]-self.Current.score[2]
+        elif self.Current.player == 1:
+            self.CurrentScore = self.Current.score[2]-self.Current.score[1]
+            
     def Make(self, r,c,o, player): # Function for generating a child node
         self.children[(r,c,o)] = ABNode(self.Current)
         move=(r,c,o)