SnapshotGenerator.h

// Copyright QED Software 2023.
 
#ifndef GRAIL_SNAPSHOT_GENERATOR_H
#define GRAIL_SNAPSHOT_GENERATOR_H
 
#include "IDebugSnapshotReceiver.h"
#include "ISnapshotGenerator.h"
 
#include <algorithm>
#include <memory>
#include <vector>
 
namespace grail
{
 template <typename SnapshotType>
 class SnapshotGenerator : public ISnapshotGenerator
 {
 public:
 SnapshotGenerator() = default;
 SnapshotGenerator(const SnapshotGenerator&) = default;
 SnapshotGenerator(SnapshotGenerator&&) = default;
 SnapshotGenerator& operator=(const SnapshotGenerator&) = default;
 SnapshotGenerator& operator=(SnapshotGenerator&&) = default;
 virtual ~SnapshotGenerator() override = default;
 
 virtual void AddReceiver(IDebugSnapshotReceiver* receiver) override
 {
 if(receiver == nullptr)
 {
 return;
 }
 snapshotReceivers.push_back(receiver);
 }
 
 virtual void RemoveReceiver(IDebugSnapshotReceiver* receiver) override
 {
 snapshotReceivers.erase(std::remove(snapshotReceivers.begin(), snapshotReceivers.end(), receiver));
 }
 
 virtual void ProduceSnapshot() override
 {
 SnapshotType snapshot;
 if(TryProduceSnapshot(snapshot))
 {
 snapshot.absoluteTime = timestampProvider != nullptr ? timestampProvider->GetTimestamp() : 0.0f;
 for(auto receiver : snapshotReceivers)
 {
 PassSnapshotToReceiver(*receiver, snapshot);
 }
 }
 }
 
 virtual void SetTimestampProvider(const std::shared_ptr<ITimestampProvider>& provider) override
 {
 timestampProvider = provider;
 }
 
 protected:
 virtual bool TryProduceSnapshot(SnapshotType& outSnapshot) = 0;
 virtual void PassSnapshotToReceiver(IDebugSnapshotReceiver& receiver, SnapshotType& snapshot) = 0;
 
 private:
 std::vector<IDebugSnapshotReceiver*> snapshotReceivers{};
 std::shared_ptr<ITimestampProvider> timestampProvider = nullptr;
 };
}
 
#endif