GrailDebugger.h

// Copyright QED Software 2023.
 
#ifndef GRAIL_DEBUGGER_H
#define GRAIL_DEBUGGER_H
 
#include "GrailStateSnapshotGenerator.h"
#include "IDebugSnapshotReceiver.h"
#include "ISnapshotGenerator.h"
#include "ITimestampProvider.h"
#include "../AIEntity.hh"
#include "../AIManager.hh"
 
#include <map>
#include <memory>
#include <unordered_set>
#include <vector>
 
namespace grail
{
 class GrailDebugger : public IEntityChangeObserver, public IReasonerChangeObserver
 {
 public:
 struct DebuggerData
 {
 std::unique_ptr<GrailDebugger> debugger;
 std::unique_ptr<class DebugInfoGenerator> debugInfoGenerator;
 };
 
 GrailDebugger(const std::shared_ptr<ITimestampProvider>& timestampProvider);
 virtual ~GrailDebugger() override = default;
 
 static DebuggerData CreateDefaultFor(const AIManager& aiManager,
 const std::shared_ptr<ITimestampProvider>& timestampProvider);
 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() const;
 
 // IReasonerChangeObserver interface
 virtual void OnReasonerSet(AIEntity& entity, Reasoner* newReasoner) override;
 
 // IEntityChangeObserver interface
 virtual void OnEntityAdded(const AIManager& manager, AIEntity& entity) override;
 virtual void OnEntityRemoved(const AIManager& manager, size_t entityId) override;
 
 private:
 void AttachToEntity(AIEntity& entity, bool verifyEntityRegistered);
 bool AttachToReasoner(AIEntity& entity, bool verifyEntityRegistered);
 
 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