GrailStateSnapshotGenerator.h

#ifndef GRAIL_STATE_SNAPSHOT_GENERATOR_H
#define GRAIL_STATE_SNAPSHOT_GENERATOR_H
 
#include "SnapshotGenerator.h"
#include "../../GrailData/DebugInfo/GrailStateSnapshot.h"
#include "../AIManager.hh"
 
#include <set>
 
namespace grail
{
    class GrailStateSnapshotGenerator : public SnapshotGenerator<GrailStateSnapshot>
    {
    public:
        GrailStateSnapshotGenerator(AIManager& manager);
        ~GrailStateSnapshotGenerator() override = default;
 
        void AddNewTagToNextSnapshot(const std::string& tag);
        virtual void Release() override;
    protected:
        bool TryProduceSnapshot(GrailStateSnapshot& outSnapshot) override;
        void PassSnapshotToReceiver(IDebugSnapshotReceiver& receiver, GrailStateSnapshot& snapshot) override;
 
    private:
        std::vector<EntityState> GetEntityStates() const;
        std::vector<EntityGroupData> GetEntityGroupData() const;
        std::vector<BlackboardState> GetSharedBlackboardStates() const;
 
        std::set<std::string> tagsForNextSnapshot{};
        const AIManager& aiManager;
    };
}
 
#endif