scheduler restriction in multiobjective model checking environment

src/storm/environment/modelchecker/MultiObjectiveModelCheckerEnvironment.cpp

@@ -97,4 +97,20 @@ namespace storm {
     void MultiObjectiveModelCheckerEnvironment::unsetMaxSteps() {
         maxSteps = boost::none;
     }
+    
+    bool MultiObjectiveModelCheckerEnvironment::isSchedulerRestrictionSet() const {
+        return schedulerRestriction.is_initialized();
+    }
+    
+    storm::storage::SchedulerClass const& MultiObjectiveModelCheckerEnvironment::getSchedulerRestriction() const {
+        return schedulerRestriction.get();
+    }
+    
+    void MultiObjectiveModelCheckerEnvironment::setSchedulerRestriction(storm::storage::SchedulerClass const& value) {
+        schedulerRestriction = value;
+    }
+    
+    void MultiObjectiveModelCheckerEnvironment::unsetSchedulerRestriction() {
+        schedulerRestriction = boost::none;
+    }
 }

src/storm/environment/modelchecker/MultiObjectiveModelCheckerEnvironment.h

@@ -5,6 +5,7 @@
 #include "storm/environment/modelchecker/ModelCheckerEnvironment.h"
 #include "storm/modelchecker/multiobjective/MultiObjectiveModelCheckingMethod.h"
 #include "storm/adapters/RationalNumberAdapter.h"
+#include "storm/storage/SchedulerClass.h"
 
 namespace storm {
     
@@ -31,18 +32,22 @@ namespace storm {
         storm::RationalNumber const& getPrecision() const;
         void setPrecision(storm::RationalNumber const& value);
         
-        uint64_t const& getMaxSteps() const;
         bool isMaxStepsSet() const;
+        uint64_t const& getMaxSteps() const;
         void setMaxSteps(uint64_t const& value);
         void unsetMaxSteps();
         
+        bool isSchedulerRestrictionSet() const;
+        storm::storage::SchedulerClass const& getSchedulerRestriction() const;
+        void setSchedulerRestriction(storm::storage::SchedulerClass const& value);
+        void unsetSchedulerRestriction();
         
     private:
         storm::modelchecker::multiobjective::MultiObjectiveMethod method;
         boost::optional<std::string> plotPathUnderApprox, plotPathOverApprox, plotPathParetoPoints;
         storm::RationalNumber precision;
         boost::optional<uint64_t> maxSteps;
-        
+        boost::optional<storm::storage::SchedulerClass> schedulerRestriction;
     };
 }