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