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):