StateBehavior.h

Go to the documentation of this file.

#pragma once
 
#include <cstdint>
#include <span>
 
#include "storm/generator/Choice.h"
 
namespace storm {
namespace generator {
 
template<typename ValueType, typename StateType = uint32_t>
class StateBehavior {
   public:
    StateBehavior();
 
    StateBehavior(StateBehavior const& other);
    StateBehavior(StateBehavior&& other) noexcept;
    StateBehavior& operator=(StateBehavior const& other);
    StateBehavior& operator=(StateBehavior&& other) noexcept;

    void clear();

    void clearChoices();

    void addChoice(Choice<ValueType, StateType>&& choice);

    Choice<ValueType, StateType>& startNewChoice(uint_fast64_t actionIndex = 0, bool markovian = false);

    void removeLastChoices(std::size_t numberOfChoices);

    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::span<Choice<ValueType, StateType> const> getChoices() const;

    std::span<Choice<ValueType, StateType>> getChoices();

    std::vector<ValueType> const& getStateRewards() const;

    std::vector<ValueType>& getStateRewards();

    std::size_t getNumberOfChoices() const;
 
   private:
    // The storage for the choices. Only the first numberOfChoices entries are considered to be part of this behavior; the remaining entries are kept for reuse.
    std::vector<Choice<ValueType, StateType>> choices;
 
    // The number of choices available in the state.
    std::size_t numberOfChoices;
 
    // 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