From cf4ee1eb5f4dee5237738998bdc2748d79d164a9 Mon Sep 17 00:00:00 2001
From: TimQu <tim.quatmann@cs.rwth-aachen.de>
Date: Tue, 15 Aug 2017 11:21:22 +0200
Subject: [PATCH] also store the initial states within an epoch model

---
 .../rewardbounded/MultiDimensionalRewardUnfolding.cpp           | 2 ++
 .../rewardbounded/MultiDimensionalRewardUnfolding.h             | 1 +
 2 files changed, 3 insertions(+)

diff --git a/src/storm/modelchecker/multiobjective/rewardbounded/MultiDimensionalRewardUnfolding.cpp b/src/storm/modelchecker/multiobjective/rewardbounded/MultiDimensionalRewardUnfolding.cpp
index c6bdeeca7..ef22078cc 100644
--- a/src/storm/modelchecker/multiobjective/rewardbounded/MultiDimensionalRewardUnfolding.cpp
+++ b/src/storm/modelchecker/multiobjective/rewardbounded/MultiDimensionalRewardUnfolding.cpp
@@ -242,6 +242,8 @@ namespace storm {
                     }
                 }
                 
+                result->initialStates = modelMemoryProduct->getInitialStates();
+                
                 return result;
             }
             
diff --git a/src/storm/modelchecker/multiobjective/rewardbounded/MultiDimensionalRewardUnfolding.h b/src/storm/modelchecker/multiobjective/rewardbounded/MultiDimensionalRewardUnfolding.h
index c87846979..6caeb96b8 100644
--- a/src/storm/modelchecker/multiobjective/rewardbounded/MultiDimensionalRewardUnfolding.h
+++ b/src/storm/modelchecker/multiobjective/rewardbounded/MultiDimensionalRewardUnfolding.h
@@ -29,6 +29,7 @@ namespace storm {
                     std::vector<std::vector<ValueType>> objectiveRewards;
                     std::vector<storm::storage::BitVector> objectiveRewardFilter;
                     std::vector<storm::storage::BitVector> relevantStates;
+                    storm::storage::BitVector initialStates;
                 };
                 
                 struct EpochSolution {