PBDRigidActiveParticlesBuffer.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "Chaos/Framework/MultiBufferResource.h"
#include "Chaos/ParticleHandle.h"
#include "Chaos/PBDRigidsEvolutionFwd.h"
 
namespace Chaos
{
    class FSingleParticlePhysicsProxy;
 
    struct FPBDRigidDirtyParticlesBufferOut
    {
        TArray<FSingleParticlePhysicsProxy*> DirtyGameThreadParticles;
        // Some particle types (clustered) only exist on the game thread, but we
        // still need to pull data over via their proxies.
        TSet<IPhysicsProxyBase*> PhysicsParticleProxies;
    };
 
 
    class FPBDRigidDirtyParticlesBuffer
    {
        friend class FPBDRigidDirtyParticlesBufferAccessor;
 
    public:
        CHAOS_API FPBDRigidDirtyParticlesBuffer(const Chaos::EMultiBufferMode& InBufferMode, bool bInSingleThreaded);
 
        CHAOS_API void CaptureSolverData(FPBDRigidsSolver* Solver);
 
        CHAOS_API void ReadLock();
        CHAOS_API void ReadUnlock();
        CHAOS_API void WriteLock();
        CHAOS_API void WriteUnlock();
    
    private:
        const FPBDRigidDirtyParticlesBufferOut* GetSolverOutData() const
        {
            return SolverDataOut->GetConsumerBuffer();
        }
 
        CHAOS_API void BufferPhysicsResults(FPBDRigidsSolver* Solver);
 
        void FlipDataOut()
        {
            SolverDataOut->FlipProducer();
        }
 
        Chaos::EMultiBufferMode BufferMode;
        FRWLock ResourceOutLock;
        bool bUseLock;
 
        // Physics thread to game thread
        TUniquePtr<IBufferResource<FPBDRigidDirtyParticlesBufferOut>> SolverDataOut;
    };
 
    class FPBDRigidDirtyParticlesBufferAccessor
    {
    public:
        FPBDRigidDirtyParticlesBufferAccessor(FPBDRigidDirtyParticlesBuffer* InManager) : Manager(InManager)
        {
            check(InManager);
            Manager->ReadLock();
        }
 
        const FPBDRigidDirtyParticlesBufferOut* GetSolverOutData() const
        {
            return Manager->GetSolverOutData();
        }
 
        ~FPBDRigidDirtyParticlesBufferAccessor()
        {
            Manager->ReadUnlock();
        }
 
    private:
        FPBDRigidDirtyParticlesBuffer* Manager;
    };
}