From a75decca44bc63a3d0a766db02ad94f5eaecfc1f Mon Sep 17 00:00:00 2001
From: Nemo <me@captnemo.in>
Date: Sun, 4 Mar 2018 03:51:44 +0530
Subject: [PATCH] generate children

---
 gothok/game.py | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/gothok/game.py b/gothok/game.py
index bf2bb7e..c42ae1a 100644
--- a/gothok/game.py
+++ b/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):