From 8b29b71c467f4a0562ab84318e279fccae2cc0a2 Mon Sep 17 00:00:00 2001
From: TimQu <tim.quatmann@cs.rwth-aachen.de>
Date: Tue, 20 Jun 2017 13:40:22 +0200
Subject: [PATCH] added the option to build a model with state valuations to
 the api

---
 src/storm/api/builder.h | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/storm/api/builder.h b/src/storm/api/builder.h
index dab7ac5d5..b2ecc2626 100644
--- a/src/storm/api/builder.h
+++ b/src/storm/api/builder.h
@@ -61,7 +61,7 @@ namespace storm {
         }
 
         template<typename ValueType>
-        std::shared_ptr<storm::models::sparse::Model<ValueType>> buildSparseModel(storm::storage::SymbolicModelDescription const& model, std::vector<std::shared_ptr<storm::logic::Formula const>> const& formulas, bool buildChoiceLabels = false, bool buildChoiceOrigins = false) {
+        std::shared_ptr<storm::models::sparse::Model<ValueType>> buildSparseModel(storm::storage::SymbolicModelDescription const& model, std::vector<std::shared_ptr<storm::logic::Formula const>> const& formulas, bool buildChoiceLabels = false, bool buildChoiceOrigins = false, bool buildStateValuations = false) {
             storm::builder::BuilderOptions options(formulas);
 
             if (storm::settings::getModule<storm::settings::modules::IOSettings>().isBuildFullModelSet()) {
@@ -69,8 +69,9 @@ namespace storm {
                 options.setBuildAllRewardModels();
                 options.clearTerminalStates();
             }
-            options.setBuildChoiceOrigins(buildChoiceOrigins);
             options.setBuildChoiceLabels(buildChoiceLabels);
+            options.setBuildChoiceOrigins(buildChoiceOrigins);
+            options.setBuildStateValuations(buildStateValuations);
             
             if (storm::settings::getModule<storm::settings::modules::IOSettings>().isJitSet()) {
                 STORM_LOG_THROW(model.isJaniModel(), storm::exceptions::NotSupportedException, "Cannot use JIT-based model builder for non-JANI model.");