GrailDebugger.h

#ifndef GRAIL_DEBUGGER_H
#define GRAIL_DEBUGGER_H
 
#include "../AIEntity.hh"
#include "../AIManager.hh"
#include "ITimestampProvider.h"
#include "IDebugSnapshotReceiver.h"
#include "ISnapshotGenerator.h"
#include "GrailStateSnapshotGenerator.h"
 
#include <vector>
#include <map>
#include <memory>
#include <unordered_set>
 
namespace grail
{
 class GrailDebugger : public IEntityChangeObserver, public IReasonerChangeObserver
 {
 public:
 GrailDebugger(const std::shared_ptr<ITimestampProvider>& timestampProvider);
 ~GrailDebugger() override = default;
 
 void AttachToManager(AIManager* manager, bool attachToAllEntities = true);
 
 void DetachFromManager(AIManager* manager);
 
 void AttachToEntity(AIEntity& entity);
 
 void DetachFromEntity(AIEntity& entity);
 
 void AddDebugSnapshotReceiver(const AIManager* manager, IDebugSnapshotReceiver* receiver);
 
 void RemoveDebugSnapshotReceiver(const AIManager* manager, IDebugSnapshotReceiver* receiver);
 
 void Update();
 
 // IReasonerChangeObserver interface
 void OnReasonerSet(AIEntity& entity, Reasoner* newReasoner) override;
 
 // IEntityChangeObserver interface
 void OnEntityAdded(const AIManager& manager, AIEntity& entity) override;
 void OnEntityRemoved(const AIManager& manager, size_t entityId) override;
 
 private:
 bool AttachToReasoner(AIEntity& entity);
 
 std::shared_ptr<ITimestampProvider> timestampProvider = nullptr;
 std::map<const AIManager*, std::vector<IDebugSnapshotReceiver*>> debugSnapshotReceivers{};
 
 std::map<size_t, std::pair<std::unique_ptr<ISnapshotGenerator>, std::unordered_set<const AIManager*>>> reasonerSnapshotGenerators{};
 std::map<const AIManager*, std::unique_ptr<GrailStateSnapshotGenerator>> grailStateSnapshotGenerators{};
 };
}
 
#endif