From 557695e82b0fc0de78713308cb45b7a0451923cb Mon Sep 17 00:00:00 2001
From: Dave Parker <dave.x.parker@gmail.com>
Date: Tue, 24 Nov 2015 18:56:39 +0000
Subject: [PATCH] Small refactor in SimulatorEngine.

git-svn-id: https://www.prismmodelchecker.org/svn/prism/prism/trunk@10946 bbc10eb1-c90d-0410-af57-cb519fbb1720
---
 prism/src/simulator/SimulatorEngine.java | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/prism/src/simulator/SimulatorEngine.java b/prism/src/simulator/SimulatorEngine.java
index 8fe84f9d..05639541 100644
--- a/prism/src/simulator/SimulatorEngine.java
+++ b/prism/src/simulator/SimulatorEngine.java
@@ -464,9 +464,7 @@ public class SimulatorEngine extends PrismComponent
 	 */
 	public void computeTransitionsForStep(int step) throws PrismException
 	{
-		updater.calculateTransitions(((PathFull) path).getState(step), transitionList);
-		transitionListBuilt = true;
-		transitionListState = new State(((PathFull) path).getState(step));
+		computeTransitionsForState(((PathFull) path).getState(step));
 	}
 
 	/**
@@ -474,7 +472,15 @@ public class SimulatorEngine extends PrismComponent
 	 */
 	public void computeTransitionsForCurrentState() throws PrismException
 	{
-		updater.calculateTransitions(path.getCurrentState(), transitionList);
+		computeTransitionsForState(path.getCurrentState());
+	}
+
+	/**
+	 * Re-compute the transition table for a particular state.
+	 */
+	private void computeTransitionsForState(State state) throws PrismException
+	{
+		updater.calculateTransitions(state, transitionList);
 		transitionListBuilt = true;
 		transitionListState = null;
 	}