From a27e76107050c091b7af076e18c9c46d6f7b099c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Steffen=20M=C3=A4rcker?= <maercker@tcs.inf.tu-dresden.de>
Date: Tue, 6 Mar 2018 13:43:39 +0100
Subject: [PATCH] Implement optimized #forEachTransition in DTMCSparse

---
 prism/src/explicit/DTMCSparse.java | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/prism/src/explicit/DTMCSparse.java b/prism/src/explicit/DTMCSparse.java
index bde9de8c..f457debb 100644
--- a/prism/src/explicit/DTMCSparse.java
+++ b/prism/src/explicit/DTMCSparse.java
@@ -226,6 +226,14 @@ public class DTMCSparse extends DTMCExplicit
 
 	//--- DTMC ---
 
+	@Override
+	public void forEachTransition(int state, TransitionConsumer consumer)
+	{
+		for (int col = rows[state], stop = rows[state+1]; col < stop; col++) {
+			consumer.accept(state, columns[col], probabilities[col]);
+		}
+	}
+
 	@Override
 	public int getNumTransitions(int state)
 	{