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