ExplorationQueue.h

Go to the documentation of this file.

#pragma once
 
#include <deque>
#include <unordered_set>
 
#include "storm-pomdp/beliefs/exploration/ExplorationQueueOrder.h"
#include "storm-pomdp/beliefs/utility/types.h"
 
namespace storm::pomdp::beliefs {
 
class ExplorationQueue {
   public:
    ExplorationQueue(ExplorationQueueOrder const order = ExplorationQueueOrder::Unordered);

    void changeOrder(ExplorationQueueOrder const newOrder);

    bool hasNext() const;

    bool push(BeliefId const id);

    BeliefId popNext();

    std::unordered_set<storm::pomdp::beliefs::BeliefId> getContents() const;
 
   private:
    ExplorationQueueOrder order;

    std::unordered_set<storm::pomdp::beliefs::BeliefId> contents;

    std::deque<storm::pomdp::beliefs::BeliefId> queue;
};
 
}  // namespace storm::pomdp::beliefs