DistributionWithReward.h

Go to the documentation of this file.

#pragma once
 
#include "storm/storage/Distribution.h"
 
#include "storm/utility/ConstantsComparator.h"
#include "storm/utility/constants.h"
 
namespace storm {
 
namespace storage {
 
template<typename ValueType, typename StateType = uint32_t>
class DistributionWithReward : public Distribution<ValueType, StateType> {
   public:
    DistributionWithReward(ValueType const& reward = storm::utility::zero<ValueType>());
 
    DistributionWithReward(DistributionWithReward const& other) = default;
    DistributionWithReward& operator=(DistributionWithReward const& other) = default;
    DistributionWithReward(DistributionWithReward&& other) = default;
    DistributionWithReward& operator=(DistributionWithReward&& other) = default;
 
    bool equals(DistributionWithReward<ValueType, StateType> const& other, storm::utility::ConstantsComparator<ValueType> const& comparator) const;
 
    bool less(DistributionWithReward<ValueType, StateType> const& other, storm::utility::ConstantsComparator<ValueType> const& comparator) const;
 
    void setReward(ValueType const& reward);
 
    ValueType const& getReward() const;
 
   private:
    ValueType reward;
};
 
}  // namespace storage
}  // namespace storm