SnapshotGenerator.h

#ifndef GRAIL_SNAPSHOT_GENERATOR_H
#define GRAIL_SNAPSHOT_GENERATOR_H
 
#include "IDebugSnapshotReceiver.h"
#include "ISnapshotGenerator.h"
 
#include <vector>
#include <algorithm>
#include <memory>
 
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;
        ~SnapshotGenerator() override = default;
 
        void AddReceiver(IDebugSnapshotReceiver* receiver) override
        {
            if (receiver == nullptr)
                return;
            snapshotReceivers.push_back(receiver);
        }
 
        void RemoveReceiver(IDebugSnapshotReceiver* receiver) override
        {
            snapshotReceivers.erase(std::remove(snapshotReceivers.begin(), snapshotReceivers.end(), receiver));
        }
 
        void ProduceSnapshot() override
        {
            SnapshotType snapshot;
            if(TryProduceSnapshot(snapshot))
            {
                snapshot.absoluteTime = timestampProvider != nullptr ? timestampProvider->GetTimestamp() : 0.0f;
                for(auto receiver : snapshotReceivers)
                {
                    PassSnapshotToReceiver(*receiver, snapshot);
                }
            }
        }
 
        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