Former-commit-id: 6acb50ec62
6acb50ec62
@ -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>;
@ -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
@ -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