also store the initial states within an epoch model

7 years ago · cf4ee1eb5f
2 changed files with 3 additions and 0 deletions
--- 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;
            }
            
--- 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 {