diff --git a/src/storm/models/symbolic/Model.cpp b/src/storm/models/symbolic/Model.cpp index 1ebf69f48..483b4337a 100644 --- a/src/storm/models/symbolic/Model.cpp +++ b/src/storm/models/symbolic/Model.cpp @@ -192,6 +192,11 @@ namespace storm { bool Model::hasRewardModel() const { return !this->rewardModels.empty(); } + + template + std::unordered_map::RewardModelType> const& Model::getRewardModels() const { + return this->rewardModels; + } template void Model::printModelInformationToStream(std::ostream& out) const { diff --git a/src/storm/models/symbolic/Model.h b/src/storm/models/symbolic/Model.h index 14c4edbc5..eecbdfdd6 100644 --- a/src/storm/models/symbolic/Model.h +++ b/src/storm/models/symbolic/Model.h @@ -246,6 +246,8 @@ namespace storm { * @return True iff the model has a reward model. */ bool hasRewardModel() const; + + std::unordered_map const& getRewardModels() const; /*! * Retrieves the number of reward models associated with this model.