getRewardModel, private, added

Former-commit-id: 095b135758

src/models/sparse/Model.cpp

@@ -89,6 +89,12 @@ namespace storm {
                 return this->rewardModels.find(rewardModelName) != this->rewardModels.end();
             }
 
+            template<typename ValueType, typename RewardModelType>
+            RewardModelType& Model<ValueType, RewardModelType>::getRewardModel(std::string const& rewardModelName) {
+                assert(this->hasRewardModel(rewardModelName));
+                return this->rewardModels.find(rewardModelName)->second;
+            }
+
             template<typename ValueType, typename RewardModelType>
             RewardModelType const& Model<ValueType, RewardModelType>::getRewardModel(std::string const& rewardModelName) const {
                 auto it = this->rewardModels.find(rewardModelName);

src/models/sparse/Model.h

@@ -154,6 +154,7 @@ namespace storm {
                  */
                 RewardModelType const& getRewardModel(std::string const& rewardModelName) const;
 
+
                 /*!
                  * Retrieves the unique reward model, if there exists exactly one. Otherwise, an exception is thrown.
                  *
@@ -266,6 +267,8 @@ namespace storm {
                 virtual bool isSparseModel() const override;
                 
             protected:
+
+                RewardModelType & getRewardModel(std::string const& rewardModelName);
                 /*!
                  * Sets the transition matrix of the model.
                  *