SchedulerChoice.h

Go to the documentation of this file.

#pragma once
 
#include "storm/storage/Distribution.h"
#include "storm/utility/constants.h"
 
namespace storm {
namespace storage {
 
template<typename ValueType>
class SchedulerChoice {
   public:
    SchedulerChoice();
 
    SchedulerChoice(uint_fast64_t deterministicChoice);
 
    SchedulerChoice(storm::storage::Distribution<ValueType, uint_fast64_t> const& randomizedChoice);
 
    SchedulerChoice(storm::storage::Distribution<ValueType, uint_fast64_t>&& randomizedChoice);
 
    bool isDefined() const;
 
    bool isDeterministic() const;
 
    uint_fast64_t getDeterministicChoice() const;
 
    storm::storage::Distribution<ValueType, uint_fast64_t> const& getChoiceAsDistribution() const;
 
    template<typename NewValueType>
    SchedulerChoice<NewValueType> toValueType() const {
        storm::storage::Distribution<NewValueType, uint_fast64_t> newDistribution;
        for (auto const& stateValuePair : distribution) {
            newDistribution.addProbability(stateValuePair.first, storm::utility::convertNumber<NewValueType>(stateValuePair.second));
        }
        return SchedulerChoice<NewValueType>(std::move(newDistribution));
    }
 
   private:
    storm::storage::Distribution<ValueType, uint_fast64_t> distribution;
};
 
template<typename ValueType>
std::ostream& operator<<(std::ostream& out, SchedulerChoice<ValueType> const& schedulerChoice);
}  // namespace storage
}  // namespace storm