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