changed node stuff

1 files changed, 10 insertions, 10 deletions

diff --git a/python/alphaBeta.py b/python/alphaBeta.py
index 214ed23..b35a7f2 100644
--- a/python/alphaBeta.py
+++ b/python/alphaBeta.py
@@ -12,18 +12,18 @@ class GameController(object):
 class ABNode(object): # A class for Node related operations
     def __init__(self, state):
         self.Current = state
-        self.CurrentScore = 0
+        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]
+    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)
+        self.children[(r,c,o)].play_move(move)
+        self.children[(r,c,o)].update_score()
 
-    def Make(self, i, j, player): # Function for generating a child node
-        self.children[(i, j)] = ABNode(self.Current.Get_currentState())
-        mul = 1
-        if player:
-            mul *= -1
-        self.children[(i, j)].CurrentScore = (self.children[(i, j)].Current.action(i, j) * mul) + self.CurrentScore
-
-    def Populate(self, i, j, Child): # Function for adding a node
-        self.children[(i,j)] = Child
+    def Populate(self, r,c,o, Child): # Function for adding a node
+        self.children[(r,c,o)] = Child
 
     def Draw(self): # function for drawing the board
         self.Current.Draw_mat()