From 7e7e025900b58ef4486846dabfd7176ddfcf2f74 Mon Sep 17 00:00:00 2001
From: sjunges <sebastian.junges@rwth-aachen.de>
Date: Tue, 29 Sep 2015 16:07:33 +0200
Subject: [PATCH] modifying reward models

Former-commit-id: 401410cee6b55ba457dc2738328b1cd5d542a750
---
 src/models/sparse/NondeterministicModel.cpp | 20 ++++++++++++++++++++
 src/models/sparse/NondeterministicModel.h   |  8 +++++++-
 2 files changed, 27 insertions(+), 1 deletion(-)

diff --git a/src/models/sparse/NondeterministicModel.cpp b/src/models/sparse/NondeterministicModel.cpp
index 9cfbe75f9..d0b0a5a25 100644
--- a/src/models/sparse/NondeterministicModel.cpp
+++ b/src/models/sparse/NondeterministicModel.cpp
@@ -62,6 +62,24 @@ namespace storm {
                     }
                 }
             }
+
+            template<typename ValueType, typename RewardModelType>
+            template<typename T>
+            void NondeterministicModel<ValueType, RewardModelType>::modifyStateActionRewards(std::string const& modelName, std::map<uint_fast64_t, T> const& modifications) {
+                RewardModelType& rewardModel = this->rewardModel(modelName);
+                for(auto const& mod : modifications) {
+                    rewardModel.setStateActionReward(mod.first, mod.second);
+                }
+            };
+
+            template<typename ValueType, typename RewardModelType>
+            template<typename T>
+            void NondeterministicModel<ValueType, RewardModelType>::modifyStateRewards(std::string const& modelName, std::map<uint_fast64_t, T> const& modifications) {
+                RewardModelType& rewardModel = this->rewardModel(modelName);
+                for(auto const& mod : modifications) {
+                    rewardModel.setStateReward(mod.first, mod.second);
+                }
+            };
             
             template<typename ValueType, typename RewardModelType>
             void NondeterministicModel<ValueType, RewardModelType>::reduceToStateBasedRewards() {
@@ -152,6 +170,8 @@ namespace storm {
 
 #ifdef STORM_HAVE_CARL
             template class NondeterministicModel<double, storm::models::sparse::StandardRewardModel<storm::Interval>>;
+            template void NondeterministicModel<double, storm::models::sparse::StandardRewardModel<storm::Interval>>::modifyStateActionRewards(std::string const& modelName, std::map<uint_fast64_t, double> const& modifications);
+            template void NondeterministicModel<double, storm::models::sparse::StandardRewardModel<storm::Interval>>::modifyStateRewards(std::string const& modelName, std::map<uint_fast64_t, double> const& modifications);
 
             template class NondeterministicModel<storm::RationalFunction>;
 #endif
diff --git a/src/models/sparse/NondeterministicModel.h b/src/models/sparse/NondeterministicModel.h
index 5bacf6dd7..e236cbca7 100644
--- a/src/models/sparse/NondeterministicModel.h
+++ b/src/models/sparse/NondeterministicModel.h
@@ -81,7 +81,13 @@ namespace storm {
                  * @param modifications A mapping from state-action pairs to the their new reward values.
                  */
                 void modifyStateActionRewards(RewardModelType& rewardModel, std::map<std::pair<uint_fast64_t, LabelSet>, typename RewardModelType::ValueType> const& modifications) const;
-                
+
+                template<typename T>
+                void modifyStateActionRewards(std::string const& modelName, std::map<uint_fast64_t, T> const& modifications);
+                template<typename T>
+                void modifyStateRewards(std::string const& modelName, std::map<uint_fast64_t, T> const& modifications);
+
+
                 virtual void reduceToStateBasedRewards() override;
                 
                 virtual void printModelInformationToStream(std::ostream& out) const override;