From a4a8bc2e690e7afe0b3310d58d1a7462e1b66d47 Mon Sep 17 00:00:00 2001
From: TimQu <tim.quatmann@cs.rwth-aachen.de>
Date: Thu, 5 Oct 2017 14:49:42 +0200
Subject: [PATCH] accessing the epoch manager and the dimensions from the
 outside

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

diff --git a/src/storm/modelchecker/multiobjective/rewardbounded/MultiDimensionalRewardUnfolding.cpp b/src/storm/modelchecker/multiobjective/rewardbounded/MultiDimensionalRewardUnfolding.cpp
index 5db3147c2..98f71093e 100644
--- a/src/storm/modelchecker/multiobjective/rewardbounded/MultiDimensionalRewardUnfolding.cpp
+++ b/src/storm/modelchecker/multiobjective/rewardbounded/MultiDimensionalRewardUnfolding.cpp
@@ -586,6 +586,16 @@ namespace storm {
                 return getStateSolution(epoch, productModel->getInitialProductState(initialStateIndex, model.getInitialStates()));
             }
 
+            template<typename ValueType, bool SingleObjectiveMode>
+            EpochManager const& MultiDimensionalRewardUnfolding<ValueType, SingleObjectiveMode>::getEpochManager() const {
+                return epochManager;
+            }
+            
+            template<typename ValueType, bool SingleObjectiveMode>
+            Dimension<ValueType> const& MultiDimensionalRewardUnfolding<ValueType, SingleObjectiveMode>::getDimension(uint64_t dim) const {
+                return dimensions.at(dim);
+            }
+            
             template class MultiDimensionalRewardUnfolding<double, true>;
             template class MultiDimensionalRewardUnfolding<double, false>;
             template class MultiDimensionalRewardUnfolding<storm::RationalNumber, true>;
diff --git a/src/storm/modelchecker/multiobjective/rewardbounded/MultiDimensionalRewardUnfolding.h b/src/storm/modelchecker/multiobjective/rewardbounded/MultiDimensionalRewardUnfolding.h
index f9476c340..03f4afcd6 100644
--- a/src/storm/modelchecker/multiobjective/rewardbounded/MultiDimensionalRewardUnfolding.h
+++ b/src/storm/modelchecker/multiobjective/rewardbounded/MultiDimensionalRewardUnfolding.h
@@ -79,6 +79,8 @@ namespace storm {
                 SolutionType const& getInitialStateResult(Epoch const& epoch); // Assumes that the initial state is unique
                 SolutionType const& getInitialStateResult(Epoch const& epoch, uint64_t initialStateIndex);
                 
+                EpochManager const& getEpochManager() const;
+                Dimension<ValueType> const& getDimension(uint64_t dim) const;
                 
             private: