From abb2b4a40353a1349a6183c60a966da8c7946ca2 Mon Sep 17 00:00:00 2001
From: Joachim Klein <klein@tcs.inf.tu-dresden.de>
Date: Tue, 5 Jul 2016 09:51:00 +0000
Subject: [PATCH] ProbModel: add resetStateRewards (analogous to
 resetTransRewards)

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

diff --git a/prism/src/prism/ProbModel.java b/prism/src/prism/ProbModel.java
index a3889871..5d968c50 100644
--- a/prism/src/prism/ProbModel.java
+++ b/prism/src/prism/ProbModel.java
@@ -473,6 +473,19 @@ public class ProbModel implements Model
 		this.transRewards[i] = transRewards;
 	}
 
+	/**
+	 * Reset state rewards DD for reward with index i.
+	 *
+	 * <br>[ STORES: stateRewards, DEREFS: <i>old state reward DD</i> ]
+	 */
+	public void resetStateRewards(int i, JDDNode stateRewards)
+	{
+		if (this.stateRewards[i] != null) {
+			JDD.Deref(this.stateRewards[i]);
+		}
+		this.stateRewards[i] = stateRewards;
+	}
+
 	// do reachability
 
 	public void doReachability()