added updateStatesOfCoalition to GameViHelper

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.");

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);