GrailStateSnapshotGenerator.h

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