From 5d80aa60d70cb1a42b2bf8606c04cc09a79ade35 Mon Sep 17 00:00:00 2001
From: TimQu <tim.quatmann@cs.rwth-aachen.de>
Date: Tue, 10 Oct 2017 17:31:32 +0200
Subject: [PATCH] extended pomdp settings

---
 .../settings/modules/POMDPSettings.cpp        | 35 ++++++++++++++++++-
 .../settings/modules/POMDPSettings.h          |  7 ++++
 2 files changed, 41 insertions(+), 1 deletion(-)

diff --git a/src/storm-pomdp-cli/settings/modules/POMDPSettings.cpp b/src/storm-pomdp-cli/settings/modules/POMDPSettings.cpp
index 249ef585a..04f31dfd6 100644
--- a/src/storm-pomdp-cli/settings/modules/POMDPSettings.cpp
+++ b/src/storm-pomdp-cli/settings/modules/POMDPSettings.cpp
@@ -16,10 +16,19 @@ namespace storm {
             
             const std::string POMDPSettings::moduleName = "pomdp";
             const std::string exportAsParametricModelOption = "parametric-drn";
-
+            const std::string qualitativeReductionOption = "qualitativereduction";
+            const std::string analyzeUniqueObservationsOption = "uniqueobservations";
+            const std::string mecReductionOption = "mecreduction";
+            const std::string selfloopReductionOption = "selfloopreduction";
+            const std::string memoryBoundOption = "memorybound";
 
             POMDPSettings::POMDPSettings() : ModuleSettings(moduleName) {
                 this->addOption(storm::settings::OptionBuilder(moduleName, exportAsParametricModelOption, false, "Export the parametric file.").addArgument(storm::settings::ArgumentBuilder::createStringArgument("filename", "The name of the file to which to write the model.").build()).build());
+                this->addOption(storm::settings::OptionBuilder(moduleName, qualitativeReductionOption, false, "Reduces the model size by performing qualitative analysis (E.g. merge states with prob. 1.").build());
+                this->addOption(storm::settings::OptionBuilder(moduleName, analyzeUniqueObservationsOption, false, "Computes the states with a unique observation").build());
+                this->addOption(storm::settings::OptionBuilder(moduleName, mecReductionOption, false, "Reduces the model size by analyzing maximal end components").build());
+                this->addOption(storm::settings::OptionBuilder(moduleName, selfloopReductionOption, false, "Reduces the model size by removing self loop actions").build());
+                this->addOption(storm::settings::OptionBuilder(moduleName, memoryBoundOption, false, "Sets the maximal number of allowed memory states (1 means memoryless schedulers).").addArgument(storm::settings::ArgumentBuilder::createUnsignedIntegerArgument("bound", "The maximal number of memory states.").setDefaultValueUnsignedInteger(1).addValidatorUnsignedInteger(storm::settings::ArgumentValidatorFactory::createUnsignedGreaterValidator(0)).build()).build());
 
             }
 
@@ -30,6 +39,30 @@ namespace storm {
             std::string POMDPSettings::getExportToParametricFilename() const {
                 return this->getOption(exportAsParametricModelOption).getArgumentByName("filename").getValueAsString();
             }
+            
+            bool POMDPSettings::isQualitativeReductionSet() const {
+                return this->getOption(qualitativeReductionOption).getHasOptionBeenSet();
+            }
+            
+            bool POMDPSettings::isAnalyzeUniqueObservationsSet() const {
+                return this->getOption(analyzeUniqueObservationsOption).getHasOptionBeenSet();
+            }
+            
+            bool POMDPSettings::isMecReductionSet() const {
+                return this->getOption(mecReductionOption).getHasOptionBeenSet();
+            }
+            
+            bool POMDPSettings::isSelfloopReductionSet() const {
+                return this->getOption(selfloopReductionOption).getHasOptionBeenSet();
+            }
+            
+            uint64_t POMDPSettings::getMemoryBound() const {
+                return this->getOption(memoryBoundOption).getArgumentByName("bound").getValueAsUnsignedInteger();
+            }
+            
+
+            
+            
 
             void POMDPSettings::finalize() {
             }
diff --git a/src/storm-pomdp-cli/settings/modules/POMDPSettings.h b/src/storm-pomdp-cli/settings/modules/POMDPSettings.h
index bbf5fa19f..8aed328f7 100644
--- a/src/storm-pomdp-cli/settings/modules/POMDPSettings.h
+++ b/src/storm-pomdp-cli/settings/modules/POMDPSettings.h
@@ -25,6 +25,13 @@ namespace storm {
                 bool isExportToParametricSet() const;
                 std::string getExportToParametricFilename() const;
                 
+                bool isQualitativeReductionSet() const;
+                bool isAnalyzeUniqueObservationsSet() const;
+                bool isMecReductionSet() const;
+                bool isSelfloopReductionSet() const;
+                uint64_t getMemoryBound() const;
+                
+                
                 bool check() const override;
                 void finalize() override;