ComputeQueueDX12.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
 
#ifdef JPH_USE_DX12
 
#include <Jolt/Compute/ComputeQueue.h>
#include <Jolt/Compute/DX12/ComputeShaderDX12.h>
#include <Jolt/Core/UnorderedSet.h>
 
JPH_NAMESPACE_BEGIN
 
class ComputeBufferDX12;
 
class JPH_EXPORT ComputeQueueDX12 final : public ComputeQueue
{
public:
    JPH_OVERRIDE_NEW_DELETE
 
    virtual                             ~ComputeQueueDX12() override;
 
    bool                                Initialize(ID3D12Device *inDevice, D3D12_COMMAND_LIST_TYPE inType, ComputeQueueResult &outResult);
 
    ID3D12GraphicsCommandList *         Start();
 
    // 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:
    void                                SyncCPUToGPU(const ComputeBufferDX12 *inBuffer);
 
    ComPtr<ID3D12CommandQueue>          mCommandQueue;                              
    ComPtr<ID3D12CommandAllocator>      mCommandAllocator;                          
    ComPtr<ID3D12GraphicsCommandList>   mCommandList;                               
    HANDLE                              mFenceEvent = INVALID_HANDLE_VALUE;         
    ComPtr<ID3D12Fence>                 mFence;                                     
    UINT64                              mFenceValue = 0;                            
    RefConst<ComputeShaderDX12>         mShader = nullptr;                          
    bool                                mIsStarted = false;                         
    bool                                mIsExecuting = false;                       
    UnorderedSet<RefConst<ComputeBuffer>> mUsedBuffers;                             
    Array<ComPtr<ID3D12Resource>>       mDelayedFreedBuffers;                       
};
 
JPH_NAMESPACE_END
 
#endif // JPH_USE_DX12