diff --git a/src/storm/modelchecker/CheckTask.h b/src/storm/modelchecker/CheckTask.h index 45f2a9bc3..dd754c762 100644 --- a/src/storm/modelchecker/CheckTask.h +++ b/src/storm/modelchecker/CheckTask.h @@ -10,6 +10,7 @@ #include "storm/solver/OptimizationDirection.h" #include "storm/logic/ComparisonType.h" #include "storm/modelchecker/hints/ModelCheckerHint.h" +#include "storm/storage/SchedulerClass.h" #include "storm/exceptions/InvalidOperationException.h" @@ -222,6 +223,18 @@ namespace storm { return produceSchedulers; } + void setSchedulerRestriction(storm::storage::SchedulerClass const& schedulerClass) { + this->schedulerRestriction = schedulerClass; + } + + storm::storage::SchedulerClass const& getSchedulerRestriction() const { + return *this->schedulerRestriction; + } + + bool hasSchedulerRestriction() const { + return this->schedulerRestriction.is_initialized(); + } + /*! * sets a hint that might contain information that speeds up the modelchecking process (if supported by the model checker) */ @@ -288,6 +301,9 @@ namespace storm { // if this flag is set. bool produceSchedulers; + // Restricts the class of schedulers that are to be considered during analysis + boost::optional schedulerRestriction; + // A hint that might contain information that speeds up the modelchecking process (if supported by the model checker) std::shared_ptr hint; };