StateBehavior.h

Go to the documentation of this file.

#ifndef STORM_GENERATOR_PRISM_STATEBEHAVIOR_H_
#define STORM_GENERATOR_PRISM_STATEBEHAVIOR_H_
 
#include <cstdint>
 
#include "storm/generator/Choice.h"
 
namespace storm {
namespace generator {
 
template<typename ValueType, typename StateType = uint32_t>
class StateBehavior {
   public:
    StateBehavior();
 
    void addChoice(Choice<ValueType, StateType>&& choice);
 
    void addStateReward(ValueType const& stateReward);
 
    void addStateRewards(std::vector<ValueType>&& stateRewards);
 
    void setExpanded(bool newValue = true);
 
    bool wasExpanded() const;
 
    bool empty() const;
 
    typename std::vector<Choice<ValueType, StateType>>::const_iterator begin() const;
 
    typename std::vector<Choice<ValueType, StateType>>::const_iterator end() const;
 
    std::vector<Choice<ValueType, StateType>> const& getChoices() const;
 
    std::vector<Choice<ValueType, StateType>>& getChoices();
 
    std::vector<ValueType> const& getStateRewards() const;
 
    std::size_t getNumberOfChoices() const;
 
   private:
    // The choices available in the state.
    std::vector<Choice<ValueType, StateType>> choices;
 
    // The state rewards (under the different, selected reward models) of the state.
    std::vector<ValueType> stateRewards;
 
    // A flag indicating whether the state was actually expanded.
    bool expanded;
};
 
}  // namespace generator
}  // namespace storm
 
#endif /* STORM_GENERATOR_PRISM_STATEBEHAVIOR_H_ */