SchedulerTrackingHelper.h

Go to the documentation of this file.

#pragma once
#include <memory>
#include <vector>
 
#include "storm/solver/OptimizationDirection.h"
#include "storm/solver/helper/ValueIterationOperatorForward.h"
 
namespace storm::solver::helper {
 
template<typename ValueType, typename SolutionType = ValueType>
class SchedulerTrackingHelper {
   public:
    SchedulerTrackingHelper(std::shared_ptr<ValueIterationOperator<ValueType, false, SolutionType>> viOperator);
 
    bool computeScheduler(std::vector<SolutionType>& operandIn, std::vector<ValueType> const& offsets, storm::OptimizationDirection const& dir,
                          std::vector<uint64_t>& schedulerStorage, bool robust, std::vector<SolutionType>* operandOut = nullptr) const;
 
   private:
    template<storm::OptimizationDirection Dir, storm::OptimizationDirection RobustDir>
    bool computeScheduler(std::vector<SolutionType>& operandIn, std::vector<ValueType> const& offsets, std::vector<uint64_t>& schedulerStorage,
                          std::vector<SolutionType>* operandOut) const;
 
   private:
    std::shared_ptr<ValueIterationOperator<ValueType, false, SolutionType>> viOperator;
};
 
}  // namespace storm::solver::helper