s444409/DSZI_Survival
s444409/DSZI_Survival
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add decision tree pause mode

Browse Source
This commit is contained in:
Michał Czekański 2020-05-24 15:57:49 +02:00
parent 2c2d865d7f
commit b8d2a4f379
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/game/Game.py
View File

@ -61,6 +61,12 @@ class Game:
elif argv[1] == "test": elif argv[1] == "test":
self.testRun(filesPath) self.testRun(filesPath)
elif argv[1] == "dt": elif argv[1] == "dt":
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) 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":
@ -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)