From 3b6c2ff51fcbac8e91479be5f5a962188fc95138 Mon Sep 17 00:00:00 2001
From: Dave Parker <dave.x.parker@gmail.com>
Date: Tue, 9 Jul 2013 22:58:41 +0000
Subject: [PATCH] Generate strategy storage for expected rewards (explicit
 engine).

git-svn-id: https://www.prismmodelchecker.org/svn/prism/prism/trunk@7006 bbc10eb1-c90d-0410-af57-cb519fbb1720
---
 prism/src/explicit/MDPModelChecker.java | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/prism/src/explicit/MDPModelChecker.java b/prism/src/explicit/MDPModelChecker.java
index 32a7d257..18a25caf 100644
--- a/prism/src/explicit/MDPModelChecker.java
+++ b/prism/src/explicit/MDPModelChecker.java
@@ -239,6 +239,7 @@ public class MDPModelChecker extends ProbModelChecker
 
 		res = computeReachRewards((MDP) model, modelRewards, b, min);
 		rewards = StateValues.createFromDoubleArray(res.soln, model);
+		result.setStrategy(res.strat);
 
 		return rewards;
 	}
@@ -1268,6 +1269,10 @@ public class MDPModelChecker extends ProbModelChecker
 			throw new PrismException("Unknown MDP solution method " + mdpSolnMethod.fullName());
 		}
 
+		// Store strategy
+		if (genStrat) {
+			res.strat = new MDStrategyArray(mdp, strat);
+		}
 		// Export adversary
 		if (exportAdv) {
 			// Prune strategy