StandardRewardModel.h

Go to the documentation of this file.

#pragma once
 
#include <boost/optional.hpp>
#include <set>
 
#include "storm/storage/dd/DdType.h"
 
namespace storm {
namespace dd {
template<storm::dd::DdType Type, typename ValueType>
class Add;
 
template<storm::dd::DdType Type>
class Bdd;
}  // namespace dd
 
namespace expressions {
class Variable;
}
 
namespace models {
namespace symbolic {
 
template<storm::dd::DdType Type, typename ValueType>
class StandardRewardModel {
   public:
    explicit StandardRewardModel(boost::optional<storm::dd::Add<Type, ValueType>> const& stateRewardVector,
                                 boost::optional<storm::dd::Add<Type, ValueType>> const& stateActionRewardVector,
                                 boost::optional<storm::dd::Add<Type, ValueType>> const& transitionRewardMatrix);
 
    bool empty() const;
 
    bool hasStateRewards() const;
 
    bool hasOnlyStateRewards() const;
 
    storm::dd::Add<Type, ValueType> const& getStateRewardVector() const;
 
    storm::dd::Add<Type, ValueType>& getStateRewardVector();
 
    boost::optional<storm::dd::Add<Type, ValueType>> const& getOptionalStateRewardVector() const;
 
    bool hasStateActionRewards() const;
 
    storm::dd::Add<Type, ValueType> const& getStateActionRewardVector() const;
 
    storm::dd::Add<Type, ValueType>& getStateActionRewardVector();
 
    boost::optional<storm::dd::Add<Type, ValueType>> const& getOptionalStateActionRewardVector() const;
 
    bool hasTransitionRewards() const;
 
    storm::dd::Add<Type, ValueType> const& getTransitionRewardMatrix() const;
 
    storm::dd::Add<Type, ValueType>& getTransitionRewardMatrix();
 
    boost::optional<storm::dd::Add<Type, ValueType>> const& getOptionalTransitionRewardMatrix() const;
 
    storm::dd::Add<Type, ValueType> getTotalRewardVector(storm::dd::Add<Type, ValueType> const& transitionMatrix,
                                                         std::set<storm::expressions::Variable> const& columnVariables) const;
 
    storm::dd::Add<Type, ValueType> getTotalRewardVector(storm::dd::Add<Type, ValueType> const& filterAdd,
                                                         storm::dd::Add<Type, ValueType> const& transitionMatrix,
                                                         std::set<storm::expressions::Variable> const& columnVariables) const;
 
    storm::dd::Add<Type, ValueType> getTotalRewardVector(storm::dd::Add<Type, ValueType> const& stateFilterAdd,
                                                         storm::dd::Add<Type, ValueType> const& choiceFilterAdd,
                                                         storm::dd::Add<Type, ValueType> const& transitionMatrix,
                                                         std::set<storm::expressions::Variable> const& columnVariables) const;
 
    storm::dd::Add<Type, ValueType> getTotalRewardVector(storm::dd::Add<Type, ValueType> const& transitionMatrix,
                                                         std::set<storm::expressions::Variable> const& columnVariables,
                                                         storm::dd::Add<Type, ValueType> const& weights, bool scaleTransAndActions) const;
 
    StandardRewardModel<Type, ValueType>& operator*=(storm::dd::Add<Type, ValueType> const& filter);
 
    StandardRewardModel<Type, ValueType> divideStateRewardVector(storm::dd::Add<Type, ValueType> const& divisor) const;
 
    void reduceToStateBasedRewards(storm::dd::Add<Type, ValueType> const& transitionMatrix, std::set<storm::expressions::Variable> const& rowVariables,
                                   std::set<storm::expressions::Variable> const& columnVariables, bool reduceToStateRewards);
 
    template<typename NewValueType>
    StandardRewardModel<Type, NewValueType> toValueType() const;
 
   private:
    // The state reward vector.
    boost::optional<storm::dd::Add<Type, ValueType>> optionalStateRewardVector;
 
    // A vector of state-action-based rewards.
    boost::optional<storm::dd::Add<Type, ValueType>> optionalStateActionRewardVector;
 
    // A matrix of transition rewards.
    boost::optional<storm::dd::Add<Type, ValueType>> optionalTransitionRewardMatrix;
};
 
}  // namespace symbolic
}  // namespace models
}  // namespace storm