DataflowSimulationContext.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "UObject/ObjectMacros.h"
#include "Dataflow/DataflowObjectInterface.h"
#include "Dataflow/DataflowSimulationProxy.h"
 
namespace UE::Dataflow
{
    template<typename Base = UE::Dataflow::FContextSingle>
    class TSimulationContext : public UE::Dataflow::TEngineContext<Base>
    {
    public:
        DATAFLOW_CONTEXT_INTERNAL(UE::Dataflow::TEngineContext<Base>, TSimulationContext);
 
        explicit TSimulationContext(const TObjectPtr<UObject>& InOwner)
                : Super(InOwner)
        {}
        
        virtual ~TSimulationContext() override {};
        
        void SetTimingInfos(const float DeltaSeconds, const float TimeSeconds) {DeltaTime = DeltaSeconds; SimulationTime = TimeSeconds;}
 
        float GetDeltaTime() const {return DeltaTime;}
        
        float GetSimulationTime() const {return SimulationTime;}
 
        template<typename ProxyType>
        void GetTypedProxies(TArray<ProxyType*>& FilteredProxies) const;
 
        DATAFLOWSIMULATION_API void GetSimulationProxies(const FString& ProxyType, const TArray<FString>& SimulationGroups, TArray<FDataflowSimulationProxy*>& FilteredProxies) const;
 
        DATAFLOWSIMULATION_API void AddSimulationProxy(const FString& ProxyType, FDataflowSimulationProxy* SimulationProxy);
        
        DATAFLOWSIMULATION_API void RemoveSimulationProxy(const FString& ProxyType, const FDataflowSimulationProxy* SimulationProxy);
 
        DATAFLOWSIMULATION_API void ResetSimulationProxies();
        
        DATAFLOWSIMULATION_API int32 NumSimulationProxies(const FString& ProxyType) const;
 
        DATAFLOWSIMULATION_API void RegisterProxyGroups();
 
        DATAFLOWSIMULATION_API void BuildGroupBits(const TArray<FString>& SimulationGroups, TBitArray<>& GroupBits) const;
 
        void PushIterationIndex()
        {
            IterationIndices.Push(0);
        }
 
        void PopIterationIndex()
        {
            IterationIndices.Pop();
        }
 
        void SetIterationIndex(const int32 IterationIndex)
        {
            IterationIndices.Last() = IterationIndex;
        }
 
        int32 GetIterationIndex()
        {
            return IterationIndices.Last();
        }
        
    private :
 
        float SimulationTime = 0.0f;
 
        float DeltaTime = 0.0f;
 
        TMap<FString, TSet<FDataflowSimulationProxy*>> SimulationProxies;
 
        TMap<FString, uint32> GroupIndices;
 
        TArray<int32> IterationIndices;
    };
 
    template<typename Base>
    template<typename ProxyType>
    void TSimulationContext<Base>::GetTypedProxies(TArray<ProxyType*>& FilteredProxies) const
    {
        if(const TSet<FDataflowSimulationProxy*>* TypedProxies = SimulationProxies.Find(ProxyType::StaticStruct()->GetName()))
        {
            for(FDataflowSimulationProxy* SimulationProxy : *TypedProxies)
            {
                if(SimulationProxy)
                {
                    FilteredProxies.Add(SimulationProxy->AsType<ProxyType>());
                }
            }
        }
    }
 
    using FDataflowSimulationContext = TSimulationContext<UE::Dataflow::FContextSingle>;
    using FDataflowSimulationContextThreaded =  TSimulationContext<UE::Dataflow::FContextThreaded>;
}