From e847d71e1338a79c8852c0f09369fa93c31749a5 Mon Sep 17 00:00:00 2001
From: Sebastian Junges <sebastian.junges@rwth-aachen.de>
Date: Wed, 15 Feb 2017 15:45:44 +0100
Subject: [PATCH] SymbolicModel: getRewardModels.

---
 src/storm/models/symbolic/Model.cpp | 5 +++++
 src/storm/models/symbolic/Model.h   | 2 ++
 2 files changed, 7 insertions(+)

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<Type, ValueType>::hasRewardModel() const {
                 return !this->rewardModels.empty();
             }
+
+            template<storm::dd::DdType Type, typename ValueType>
+            std::unordered_map<std::string, typename Model<Type, ValueType>::RewardModelType> const& Model<Type, ValueType>::getRewardModels() const {
+                return this->rewardModels;
+            }
             
             template<storm::dd::DdType Type, typename ValueType>
             void Model<Type, ValueType>::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<std::string, RewardModelType> const& getRewardModels() const;
                 
                 /*!
                  * Retrieves the number of reward models associated with this model.