From c9d5e2959a2a834dcdb0bd8cd18f67d72c6d871e Mon Sep 17 00:00:00 2001
From: Stefan Pranger <stefan.pranger@student.tugraz.at>
Date: Tue, 23 Feb 2021 17:48:22 +0100
Subject: [PATCH] let MDP helper return type store choiceValues

---
 .../helper/MDPModelCheckingHelperReturnType.h   | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/src/storm/modelchecker/prctl/helper/MDPModelCheckingHelperReturnType.h b/src/storm/modelchecker/prctl/helper/MDPModelCheckingHelperReturnType.h
index d40c5ade0..7a8c7b320 100644
--- a/src/storm/modelchecker/prctl/helper/MDPModelCheckingHelperReturnType.h
+++ b/src/storm/modelchecker/prctl/helper/MDPModelCheckingHelperReturnType.h
@@ -9,31 +9,34 @@ namespace storm {
     namespace storage {
         class BitVector;
     }
-    
+
     namespace modelchecker {
         namespace helper {
             template<typename ValueType>
             struct MDPSparseModelCheckingHelperReturnType {
-                
+
                 MDPSparseModelCheckingHelperReturnType(MDPSparseModelCheckingHelperReturnType const&) = delete;
                 MDPSparseModelCheckingHelperReturnType(MDPSparseModelCheckingHelperReturnType&&) = default;
-                
+
                 MDPSparseModelCheckingHelperReturnType(std::vector<ValueType>&& values, std::unique_ptr<storm::storage::Scheduler<ValueType>>&& scheduler = nullptr) : values(std::move(values)), scheduler(std::move(scheduler)) {
                     // Intentionally left empty.
                 }
-                
+
                 virtual ~MDPSparseModelCheckingHelperReturnType() {
                     // Intentionally left empty.
                 }
-                
+
                 // The values computed for the states.
                 std::vector<ValueType> values;
-                
+
+                // The values computed for the available choices.
+                std::vector<ValueType> choiceValues;
+
                 // A scheduler, if it was computed.
                 std::unique_ptr<storm::storage::Scheduler<ValueType>> scheduler;
             };
         }
-        
+
     }
 }