From d939ebe375a658a726b2c65cafc6f40598e9d817 Mon Sep 17 00:00:00 2001
From: Stefan Pranger <stefan.pranger@student.tugraz.at>
Date: Wed, 16 Dec 2020 11:19:45 +0100
Subject: [PATCH] smg model now stores the player action indices

---
 src/storm/models/sparse/Smg.cpp | 12 ++++++++----
 src/storm/models/sparse/Smg.h   |  5 +++++
 2 files changed, 13 insertions(+), 4 deletions(-)

diff --git a/src/storm/models/sparse/Smg.cpp b/src/storm/models/sparse/Smg.cpp
index e0419844c..e7c20c9fa 100644
--- a/src/storm/models/sparse/Smg.cpp
+++ b/src/storm/models/sparse/Smg.cpp
@@ -6,6 +6,7 @@
 #include "storm/adapters/RationalFunctionAdapter.h"
 
 #include "storm/models/sparse/StandardRewardModel.h"
+#include "storm/models/sparse/Mdp.h"
 
 namespace storm {
     namespace models {
@@ -29,14 +30,17 @@ namespace storm {
             Smg<ValueType, RewardModelType>::Smg(storm::storage::sparse::ModelComponents<ValueType, RewardModelType> const& components, ModelType type)
                     : NondeterministicModel<ValueType, RewardModelType>(type, components) {
                 assert(type == storm::models::ModelType::Smg);
-                // Intentionally left empty
             }
 
-           template <typename ValueType, typename RewardModelType>
+            template <typename ValueType, typename RewardModelType>
             Smg<ValueType, RewardModelType>::Smg(storm::storage::sparse::ModelComponents<ValueType, RewardModelType>&& components, ModelType type)
-                    : NondeterministicModel<ValueType, RewardModelType>(type, std::move(components)) {
+                    : NondeterministicModel<ValueType, RewardModelType>(type, std::move(components)), playerActionIndices(components.playerActionIndices.get()) {
                assert(type == storm::models::ModelType::Smg);
-                // Intentionally left empty
+            }
+
+            template <typename ValueType, typename RewardModelType>
+            std::vector<uint_fast32_t> Smg<ValueType, RewardModelType>::getPlayerActionIndices() const {
+                return playerActionIndices;
             }
 
             template class Smg<double>;
diff --git a/src/storm/models/sparse/Smg.h b/src/storm/models/sparse/Smg.h
index 663817418..2bf7146c8 100644
--- a/src/storm/models/sparse/Smg.h
+++ b/src/storm/models/sparse/Smg.h
@@ -48,6 +48,11 @@ namespace storm {
 
                 Smg(Smg<ValueType, RewardModelType>&& other) = default;
                 Smg& operator=(Smg<ValueType, RewardModelType>&& other) = default;
+
+                std::vector<uint_fast64_t> getPlayerActionIndices() const;
+
+            private:
+                std::vector<uint_fast64_t> playerActionIndices;
             };
 
         } // namespace sparse