diff --git a/src/storm/modelchecker/rpatl/helper/internal/GameViHelper.cpp b/src/storm/modelchecker/rpatl/helper/internal/GameViHelper.cpp index 2426268b5..96a2e0c81 100644 --- a/src/storm/modelchecker/rpatl/helper/internal/GameViHelper.cpp +++ b/src/storm/modelchecker/rpatl/helper/internal/GameViHelper.cpp @@ -175,6 +175,11 @@ namespace storm { _transitionMatrix = newTransitionMatrix; } + template + void GameViHelper::updateStatesOfCoaltion(storm::storage::BitVector newStatesOfCoaltion) { + _statesOfCoalition = newStatesOfCoaltion; + } + template std::vector const& GameViHelper::getProducedOptimalChoices() const { STORM_LOG_ASSERT(this->isProduceSchedulerSet(), "Trying to get the produced optimal choices although no scheduler was requested."); diff --git a/src/storm/modelchecker/rpatl/helper/internal/GameViHelper.h b/src/storm/modelchecker/rpatl/helper/internal/GameViHelper.h index 8f2c8143f..393f8a79a 100644 --- a/src/storm/modelchecker/rpatl/helper/internal/GameViHelper.h +++ b/src/storm/modelchecker/rpatl/helper/internal/GameViHelper.h @@ -44,10 +44,15 @@ namespace storm { bool isProduceSchedulerSet() const; /*! - * Changes the transitionMatrix of the gameViHelper to the given one. + * Changes the transitionMatrix to the given one. */ void updateTransitionMatrix(storm::storage::SparseMatrix newTransitionMatrix); + /*! + * Changes the statesOfCoalition to the given one. + */ + void updateStatesOfCoaltion(storm::storage::BitVector newStatesOfCoaltion); + storm::storage::Scheduler extractScheduler() const; void getChoiceValues(Environment const& env, std::vector const& x, std::vector& choiceValues);