generate children

gothok/game.py

@@ -112,10 +112,21 @@ class GameNode(object):
     def runSimulation(self):
         self.backPropagate(self.simulate())
 
+    def getChildren(self):
+        possibleMoves = self.state.getPossibleMoves()
+        children = []
+
+        for move in possibleMoves:
+            newState = self.state.playMove(move)
+            childNode = GameNode(newState, self.state)
+            children.append(childNode)
+
+        return children
+
     def chooseChild(self):
         # Define children nodes
         if(not self.children):
-            self.children = self.getChildren(self.state)
+            self.children = self.getChildren()
 
         # Run simulation on leaf nodes
         if(len(self.children) == 0):