PhysicsReplicationCache.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
/*=============================================================================
    PhysicsReplicationCache.h 
    Cache physics state from PT on GT for actors using physics replication.
=============================================================================*/
 
#pragma once
 
#include "Chaos/SimCallbackObject.h"
#include "Engine/ReplicatedState.h"
 
class FPhysScene_Chaos;
class UPrimitiveComponent;
 
// -------------- Game Thread --------------
 
class FPhysicsReplicationCache
{
public:
    FPhysicsReplicationCache(FPhysScene_Chaos* InPhysicsScene);
    
    virtual ~FPhysicsReplicationCache();
 
    const FRigidBodyState* GetStateFromReplicationCache(UPrimitiveComponent* RootComponent, int32& OutSolverFrame);
 
    void RegisterForReplicationCache(UPrimitiveComponent* RootComponent);
    
    void UnregisterForReplicationCache(UPrimitiveComponent* RootComponent);
 
private:
    void ProcessAsyncOutput();
 
private:
    int32 SolverFrame = 0;
    TMap<Chaos::FConstPhysicsObjectHandle, FRigidBodyState> ReplicationCache_External;
    
    FPhysScene_Chaos* PhysicsScene;
    FDelegateHandle DelegateInjectInputs_External;
 
    class FPhysicsReplicationCacheAsync* AsyncPhysicsReplicationCache;
};
 
 
 
// -------------- Async Marshaling --------------
 
struct FPhysicsReplicationCacheAsyncInput : public Chaos::FSimCallbackInput
{
    mutable TArray<Chaos::FConstPhysicsObjectHandle> AccessedObjects;
    TArray<Chaos::FConstPhysicsObjectHandle> UnregisterObjects;
 
    void Reset()
    {
        AccessedObjects.Reset();
        UnregisterObjects.Reset();
    }
};
 
struct FPhysicsReplicationCacheAsyncOutput : public Chaos::FSimCallbackOutput
{
    int32 SolverFrame;
    TArray<Chaos::FConstPhysicsObjectHandle> ReplicationCache_Key_Marshal;
    TArray<FRigidBodyState> ReplicationCache_Value_Marshal;
 
    void Reset()
    {
        SolverFrame = 0;
    }
};
 
 
 
// -------------- Physics Thread --------------
 
struct FPhysicsReplicationCacheData
{
public:
    FPhysicsReplicationCacheData()
    { }
 
    void SetAccessTime(double Time) { AccessTime = Time; }
    const double GetAccessTime() const { return AccessTime; }
    FRigidBodyState& GetState() { return StateData; }
 
private:
    double AccessTime;
    FRigidBodyState StateData;
};
 
class FPhysicsReplicationCacheAsync : public Chaos::TSimCallbackObject<
    FPhysicsReplicationCacheAsyncInput,
    FPhysicsReplicationCacheAsyncOutput,
    Chaos::ESimCallbackOptions::Presimulate | Chaos::ESimCallbackOptions::PostSolve | Chaos::ESimCallbackOptions::PhysicsObjectUnregister>
{
    virtual void OnPreSimulate_Internal() override;
    virtual void OnPostSolve_Internal() override;
 
    virtual void OnPhysicsObjectUnregistered_Internal(Chaos::FConstPhysicsObjectHandle PhysicsObject) override;
 
    void ProcessAsyncInputs();
 
    void PopulateReplicationCache_Internal();
 
private:
    TMap<Chaos::FConstPhysicsObjectHandle, FPhysicsReplicationCacheData> ReplicationCache_Internal;
    bool bUpdateAfterRemoval = false;
};