StateActionReward.h

Go to the documentation of this file.

#pragma once
 
#include <map>
 
#include "storm/storage/expressions/Expression.h"
#include "storm/storage/prism/LocatedInformation.h"
 
namespace storm {
namespace storage {
namespace expressions {
class Variable;
}
}  // namespace storage
}  // namespace storm
 
namespace storm {
namespace prism {
class StateActionReward : public LocatedInformation {
   public:
    StateActionReward(uint_fast64_t actionIndex, std::string const& actionName, storm::expressions::Expression const& statePredicateExpression,
                      storm::expressions::Expression const& rewardValueExpression, std::string const& filename = "", uint_fast64_t lineNumber = 0);
 
    // Create default implementations of constructors/assignment.
    StateActionReward() = default;
    StateActionReward(StateActionReward const& other) = default;
    StateActionReward& operator=(StateActionReward const& other) = default;
    StateActionReward(StateActionReward&& other) = default;
    StateActionReward& operator=(StateActionReward&& other) = default;

    std::string const& getActionName() const;

    uint_fast64_t getActionIndex() const;

    storm::expressions::Expression const& getStatePredicateExpression() const;

    storm::expressions::Expression const& getRewardValueExpression() const;

    bool isLabeled() const;

    StateActionReward substitute(std::map<storm::expressions::Variable, storm::expressions::Expression> const& substitution) const;
 
    friend std::ostream& operator<<(std::ostream& stream, StateActionReward const& stateActionReward);
 
   private:
    // The index of the action name.
    uint_fast64_t actionIndex;
 
    // The name of the command this transition-based reward is attached to.
    std::string actionName;
 
    // A flag that stores whether the transition reward has an action label.
    bool labeled;
 
    // A predicate that needs to be satisfied by states for the reward to be obtained (by taking
    // a corresponding command transition).
    storm::expressions::Expression statePredicateExpression;
 
    // The expression specifying the value of the reward obtained along the transitions.
    storm::expressions::Expression rewardValueExpression;
};
 
}  // namespace prism
}  // namespace storm