ComputeQueueVK.h

Go to the documentation of this file.

// Jolt Physics Library (https://github.com/jrouwe/JoltPhysics)
// SPDX-FileCopyrightText: 2025 Jorrit Rouwe
// SPDX-License-Identifier: MIT
 
#pragma once
 
#include <Jolt/Compute/ComputeQueue.h>
 
#ifdef JPH_USE_VK
 
#include <Jolt/Compute/VK/ComputeShaderVK.h>
#include <Jolt/Compute/VK/BufferVK.h>
#include <Jolt/Core/UnorderedMap.h>
#include <Jolt/Core/UnorderedSet.h>
 
JPH_NAMESPACE_BEGIN
 
class ComputeSystemVK;
class ComputeBufferVK;
 
class JPH_EXPORT ComputeQueueVK final : public ComputeQueue
{
public:
    JPH_OVERRIDE_NEW_DELETE
 
    explicit                            ComputeQueueVK(ComputeSystemVK *inComputeSystem) : mComputeSystem(inComputeSystem) { }
    virtual                             ~ComputeQueueVK() override;
 
    bool                                Initialize(uint32 inComputeQueueIndex, ComputeQueueResult &outResult);
 
    // 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:
    bool                                BeginCommandBuffer();
 
    // Copy the CPU buffer to the GPU buffer if needed
    void                                SyncCPUToGPU(const ComputeBufferVK *inBuffer);
 
    ComputeSystemVK *                   mComputeSystem;
    VkQueue                             mQueue = VK_NULL_HANDLE;
    VkCommandPool                       mCommandPool = VK_NULL_HANDLE;
    VkDescriptorPool                    mDescriptorPool = VK_NULL_HANDLE;
    VkCommandBuffer                     mCommandBuffer = VK_NULL_HANDLE;
    bool                                mCommandBufferRecording = false;                
    VkFence                             mFence = VK_NULL_HANDLE;
    bool                                mIsExecuting = false;                           
    RefConst<ComputeShaderVK>           mShader;                                        
    Array<VkDescriptorBufferInfo>       mBufferInfos;                                   
    UnorderedSet<RefConst<ComputeBuffer>> mUsedBuffers;                                 
    Array<BufferVK>                     mDelayedFreedBuffers;                           
};
 
JPH_NAMESPACE_END
 
#endif // JPH_USE_VK