BeliefExplorationMatrix.h

Go to the documentation of this file.

#pragma once
 
#include <set>
#include <string>
#include <tuple>
#include <vector>
 
#include "storm-pomdp/beliefs/utility/types.h"
 
namespace storm::pomdp::beliefs {

template<typename ValueType, typename...>
struct BeliefExplorationTransition {
    ValueType probability;
    storm::pomdp::beliefs::BeliefId targetBelief;
};
 
template<typename ValueType, typename ExtraTransitionData>
struct BeliefExplorationTransition<ValueType, ExtraTransitionData> {
    ValueType probability;
    storm::pomdp::beliefs::BeliefId targetBelief;
    ExtraTransitionData data;
};
 
template<typename ValueType, typename FirstExtraTransitionData, typename... OtherExtraTransitionData>
struct BeliefExplorationTransition<ValueType, FirstExtraTransitionData, OtherExtraTransitionData...> {
    ValueType probability;
    storm::pomdp::beliefs::BeliefId targetBelief;
    std::tuple<FirstExtraTransitionData, OtherExtraTransitionData...> data;
};

template<typename ValueType, typename... ExtraTransitionData>
class BeliefExplorationMatrix {
   public:
    BeliefExplorationMatrix();

    void endCurrentRow();

    void endCurrentRowGroup();

    std::size_t rows() const;

    std::size_t groups() const;

    bool hasChoiceLabels() const;

    std::vector<BeliefExplorationTransition<ValueType, ExtraTransitionData...>> transitions;
    std::vector<uint64_t> rowIndications;
    std::vector<uint64_t> rowGroupIndices;
    std::vector<std::set<std::string>> choiceLabels;
};
 
}  // namespace storm::pomdp::beliefs