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 <typename ValueType> + void GameViHelper<ValueType>::updateStatesOfCoaltion(storm::storage::BitVector newStatesOfCoaltion) { + _statesOfCoalition = newStatesOfCoaltion; + } + template <typename ValueType> std::vector<uint64_t> const& GameViHelper<ValueType>::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<ValueType> newTransitionMatrix); + /*! + * Changes the statesOfCoalition to the given one. + */ + void updateStatesOfCoaltion(storm::storage::BitVector newStatesOfCoaltion); + storm::storage::Scheduler<ValueType> extractScheduler() const; void getChoiceValues(Environment const& env, std::vector<ValueType> const& x, std::vector<ValueType>& choiceValues);