BeliefCollector.h

Go to the documentation of this file.

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

template<typename BeliefType>
class BeliefCollector {
   public:
    static constexpr bool HasConsecutiveIds = true;

    bool isEqual(BeliefId const& first, BeliefId const& second) const;

    BeliefId getNumberOfBeliefIds() const;

    BeliefType const& getBeliefFromId(BeliefId const& id) const;

    BeliefId getIdFromBelief(BeliefType const& belief) const;

    bool containsBelief(BeliefType const& belief) const;

    bool containsId(BeliefId const& id) const;

    BeliefId getIdOptional(BeliefType const& belief) const;

    BeliefId getIdOrAddBelief(BeliefType&& belief);

    BeliefId addBelief(BeliefType&& inputBelief);
 
   private:
    std::vector<BeliefType> gatheredBeliefs;
    // Resolve hash collisions by comparing the beliefs in gatheredBeliefs.
    std::vector<std::unordered_multimap<std::size_t, BeliefId>> beliefToIdMap;
};
 
}  // namespace storm::pomdp::beliefs