ComputeQueueCPU.h

Go to the documentation of this file.

// Jolt Physics Library (https://github.com/jrouwe/JoltPhysics)
// SPDX-FileCopyrightText: 2026 Jorrit Rouwe
// SPDX-License-Identifier: MIT
 
#pragma once
 
#include <Jolt/Compute/ComputeQueue.h>
 
#ifdef JPH_USE_CPU_COMPUTE
 
#include <Jolt/Compute/CPU/ComputeShaderCPU.h>
#include <Jolt/Core/UnorderedSet.h>
 
JPH_NAMESPACE_BEGIN
 
class JPH_EXPORT ComputeQueueCPU final : public ComputeQueue
{
public:
    JPH_OVERRIDE_NEW_DELETE
 
    virtual                             ~ComputeQueueCPU() override;
 
    // See: ComputeQueue
    virtual void                        SetShader(const ComputeShader *inShader) override;
    virtual void                        SetConstantBuffer(const char *inName, const ComputeBuffer *inBuffer) override;
    virtual void                        SetBuffer(const char *inName, const ComputeBuffer *inBuffer) override;
    virtual void                        SetRWBuffer(const char *inName, ComputeBuffer *inBuffer, EBarrier inBarrier = EBarrier::Yes) override;
    virtual void                        ScheduleReadback(ComputeBuffer *inDst, const ComputeBuffer *inSrc) override;
    virtual void                        Dispatch(uint inThreadGroupsX, uint inThreadGroupsY, uint inThreadGroupsZ) override;
    virtual void                        Execute() override;
    virtual void                        Wait() override;
 
private:
    RefConst<ComputeShaderCPU>          mShader = nullptr;                          
    ShaderWrapper *                     mWrapper = nullptr;                         
    UnorderedSet<RefConst<ComputeBuffer>> mUsedBuffers;                             
};
 
JPH_NAMESPACE_END
 
#endif // JPH_USE_CPU_COMPUTE