From 5e9c42f2afa312bca8a359f36187ed783b651cbe Mon Sep 17 00:00:00 2001
From: sjunges <sebastian.junges@rwth-aachen.de>
Date: Mon, 25 Jan 2016 15:32:19 +0100
Subject: [PATCH] intermediate commit

Former-commit-id: 6acb50ec6293fca2038a09a258796da635bf0f07
---
 src/models/sparse/Model.cpp | 3 +++
 src/models/sparse/Model.h   | 1 +
 src/storage/SparseMatrix.h  | 8 ++++++++
 3 files changed, 12 insertions(+)

diff --git a/src/models/sparse/Model.cpp b/src/models/sparse/Model.cpp
index 791062b54..a1e71c0b7 100644
--- a/src/models/sparse/Model.cpp
+++ b/src/models/sparse/Model.cpp
@@ -320,6 +320,9 @@ namespace storm {
                 return this->rewardModels;
             }
 
+            std::set<storm::Variable> getProbabilityParameters(Model<storm::RationalNumber> const& model) {
+                return storm::storage::getVariables(model.getTransitionMatrix());
+            }
             
             template class Model<double>;
             template class Model<float>;
diff --git a/src/models/sparse/Model.h b/src/models/sparse/Model.h
index 3d7776800..7d36ebca3 100644
--- a/src/models/sparse/Model.h
+++ b/src/models/sparse/Model.h
@@ -345,6 +345,7 @@ namespace storm {
                 boost::optional<std::vector<LabelSet>> choiceLabeling;
             };
             
+            std::set<storm::Variable> getProbabilityParameters(Model<storm::RationalFunction> const& model);
         } // namespace sparse
     } // namespace models
 } // namespace storm
diff --git a/src/storage/SparseMatrix.h b/src/storage/SparseMatrix.h
index 99eea1c06..21d675cab 100644
--- a/src/storage/SparseMatrix.h
+++ b/src/storage/SparseMatrix.h
@@ -969,6 +969,14 @@ namespace storm {
             std::vector<index_type> rowGroupIndices;
             
         };
+        
+        std::set<storm::Variable> getVariables(SparseMatrix<storm::RationalFunction> const& matrix) {
+            std::set<storm::Variable> result;
+            for(auto const& entry : matrix) {
+                entry.probability.gatherVariables(result);
+            }
+            return result;
+        }
 
     } // namespace storage
 } // namespace storm