ComputeBufferVK.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/ComputeBuffer.h>
 
#ifdef JPH_USE_VK
 
#include <Jolt/Compute/VK/BufferVK.h>
 
JPH_NAMESPACE_BEGIN
 
class ComputeSystemVK;

class JPH_EXPORT ComputeBufferVK final : public ComputeBuffer
{
public:
    JPH_OVERRIDE_NEW_DELETE

                                    ComputeBufferVK(ComputeSystemVK *inComputeSystem, EType inType, uint64 inSize, uint inStride);
    virtual                         ~ComputeBufferVK() override;
 
    bool                            Initialize(const void *inData);
 
    virtual ComputeBufferResult     CreateReadBackBuffer() const override;
 
    VkBuffer                        GetBufferCPU() const                                    { return mBufferCPU.mBuffer; }
    VkBuffer                        GetBufferGPU() const                                    { return mBufferGPU.mBuffer; }
    BufferVK                        ReleaseBufferCPU() const                                { BufferVK tmp = mBufferCPU; mBufferCPU = BufferVK(); return tmp; }
 
    void                            Barrier(VkCommandBuffer inCommandBuffer, VkPipelineStageFlags inToStage, VkAccessFlagBits inToFlags, bool inForce) const;
    bool                            SyncCPUToGPU(VkCommandBuffer inCommandBuffer) const;
 
private:
    virtual void *                  MapInternal(EMode inMode) override;
    virtual void                    UnmapInternal() override;
 
    ComputeSystemVK *               mComputeSystem;
    mutable BufferVK                mBufferCPU;
    BufferVK                        mBufferGPU;
    mutable bool                    mNeedsSync = false;                                     
    mutable VkAccessFlagBits        mAccessFlagBits = VK_ACCESS_SHADER_READ_BIT;            
    mutable VkPipelineStageFlags    mAccessStage = VK_PIPELINE_STAGE_COMPUTE_SHADER_BIT;    
};
 
JPH_NAMESPACE_END
 
#endif // JPH_USE_VK