Add decision tree pause mode

This commit is contained in:
Michał Czekański 2020-05-24 15:57:49 +02:00
parent 2c2d865d7f
commit b8d2a4f379

View File

@ -61,7 +61,13 @@ class Game:
elif argv[1] == "test": elif argv[1] == "test":
self.testRun(filesPath) self.testRun(filesPath)
elif argv[1] == "dt": elif argv[1] == "dt":
self.dtRun(filesPath) if len(argv) >= 3:
if argv[2] == "-p":
print("Running Decision Tree in pause mode.")
self.dtRun(filesPath, True)
else:
print("Running Decision Tree.")
self.dtRun(filesPath)
elif argv[1] == "ga" and len(argv) >= 3: elif argv[1] == "ga" and len(argv) >= 3:
if len(argv) >= 4 and argv[3] == "-t": if len(argv) >= 4 and argv[3] == "-t":
print("Running Genetic Algorithm in multithreaded mode, iter = ", argv[2]) print("Running Genetic Algorithm in multithreaded mode, iter = ", argv[2])
@ -207,12 +213,13 @@ class Game:
# Flip the display # Flip the display
pygame.display.flip() pygame.display.flip()
def dtRun(self, filesPath): def dtRun(self, filesPath, pauseAfterDecision=False):
""" """
Runs game in decision tree mode. Runs game in decision tree mode.
In this mode user can only watch how player performs decisions with usage of decision tree. In this mode user can only watch how player performs decisions with usage of decision tree.
:param pauseAfterDecision: If pause mode is true, then simulation will be paused after each tree decision.
:param filesPath: :param filesPath:
""" """
self.running = True self.running = True
@ -279,6 +286,8 @@ class Game:
# Choose target for player using decision tree # Choose target for player using decision tree
if self.player.movementTarget is None: if self.player.movementTarget is None:
self.player.gotoToTarget(survivalDecisionTree.pickEntity(self.player, self.map), self.map) self.player.gotoToTarget(survivalDecisionTree.pickEntity(self.player, self.map), self.map)
if pauseAfterDecision:
pause = True
self.screen.ui.updateBarsBasedOnPlayerStats(self.player.statistics) self.screen.ui.updateBarsBasedOnPlayerStats(self.player.statistics)