DebugInfoGenerator.h

// Copyright QED Software 2023.
 
#pragma once
 
#include "IDebugSnapshotReceiver.h"
#include "ITimestampProvider.h"
#include "../../GrailData/DebugInfo/GrailDebugInfo.h"
 
#include <cmath>
#include <memory>
 
namespace grail
{
 class DebugInfoGenerator : public IDebugSnapshotReceiver
 {
 public:
 DebugInfoGenerator(const std::shared_ptr<ITimestampProvider>& timestampProvider);
 virtual ~DebugInfoGenerator() override;
 
 const data::GrailDebugInfo& GetDebugInfo() const;
 void ResetDebugInfo();
#ifndef BUILD_ARM
 void SaveToFile(const std::string& filePath, bool useBinaryFormat = true) const;
#endif
 virtual void ProcessSnapshot(data::GrailStateSnapshot& snapshot) override;
 virtual void ProcessSnapshot(data::UtilityReasonerSnapshot& snapshot) override;
 virtual void ProcessSnapshot(data::PlannerReasonerSnapshot& snapshot) override;
 virtual void ProcessSnapshot(data::SimulatedGameReasonerSnapshot& snapshot) override;
 
 private:
 template <typename SnapshotType>
 void ProcessNewSnapshot(SnapshotType& snapshot,
 std::vector<SnapshotType>& snapshotList)
 {
 debugInfo.endTime = std::fmax(snapshot.absoluteTime, debugInfo.endTime);
 snapshot.relativeTime = timestampProvider->GetTimestamp() - debugInfo.startTime;
 snapshotList.push_back(snapshot);
 }
 
 template <typename SnapshotType>
 void ProcessReasonerSnapshot(SnapshotType& reasonerSnapshot,
 std::map<size_t, std::vector<SnapshotType>>& listsPerEntity,
 size_t entityId)
 {
 auto iter = listsPerEntity.find(entityId);
 if(iter == listsPerEntity.end())
 {
 std::vector<SnapshotType> snapshots;
 ProcessNewSnapshot(reasonerSnapshot, snapshots);
 listsPerEntity[entityId] = std::move(snapshots);
 }
 else
 {
 ProcessNewSnapshot(reasonerSnapshot, iter->second);
 }
 }
 
 std::shared_ptr<ITimestampProvider> timestampProvider = nullptr;
 data::GrailDebugInfo debugInfo{};
 };
}