From 03ec53321ec1b82f9b02e48573dbe8570d039e46 Mon Sep 17 00:00:00 2001
From: lukpo <lukas.posch@student.tugraz.at>
Date: Mon, 9 Aug 2021 15:24:58 +0200
Subject: [PATCH] added updateStatesOfCoalition to GameViHelper

---
 .../modelchecker/rpatl/helper/internal/GameViHelper.cpp    | 5 +++++
 .../modelchecker/rpatl/helper/internal/GameViHelper.h      | 7 ++++++-
 2 files changed, 11 insertions(+), 1 deletion(-)

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