ComputeSystemVK.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/ComputeSystem.h>
 
#ifdef JPH_USE_VK
 
#include <Jolt/Compute/VK/ComputeQueueVK.h>
 
JPH_NAMESPACE_BEGIN
 
class JPH_EXPORT ComputeSystemVK : public ComputeSystem
{
public:
    JPH_OVERRIDE_NEW_DELETE
 
    // Initialize / shutdown the compute system
    bool                            Initialize(VkPhysicalDevice inPhysicalDevice, VkDevice inDevice, uint32 inComputeQueueIndex, ComputeSystemResult &outResult);
    void                            Shutdown();
 
    // See: ComputeSystem
    virtual ComputeShaderResult     CreateComputeShader(const char *inName, uint32 inGroupSizeX, uint32 inGroupSizeY, uint32 inGroupSizeZ) override;
    virtual ComputeBufferResult     CreateComputeBuffer(ComputeBuffer::EType inType, uint64 inSize, uint inStride, const void *inData = nullptr) override;
    virtual ComputeQueueResult      CreateComputeQueue() override;
 
    VkDevice                        GetDevice() const                                               { return mDevice; }
 
    virtual bool                    CreateBuffer(VkDeviceSize inSize, VkBufferUsageFlags inUsage, VkMemoryPropertyFlags inProperties, BufferVK &outBuffer) = 0;
    virtual void                    FreeBuffer(BufferVK &ioBuffer) = 0;
    virtual void *                  MapBuffer(BufferVK &ioBuffer) = 0;
    virtual void                    UnmapBuffer(BufferVK &ioBuffer) = 0;
 
protected:
    virtual bool                    InitializeMemory() = 0;
    virtual void                    ShutdownMemory() = 0;
 
    VkPhysicalDevice                mPhysicalDevice = VK_NULL_HANDLE;
    VkDevice                        mDevice = VK_NULL_HANDLE;
    uint32                          mComputeQueueIndex = 0;
    PFN_vkSetDebugUtilsObjectNameEXT mVkSetDebugUtilsObjectNameEXT = nullptr;
 
private:
    // Buffer that can be bound when we have no buffer
    BufferVK                        mDummyBuffer;
};
 
JPH_NAMESPACE_END
 
#endif // JPH_USE_VK