StateBehavior.h

Go to the documentation of this file.

#pragma once
 
#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