ComputeShaderVK.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/ComputeShader.h>
 
#ifdef JPH_USE_VK
 
#include <Jolt/Compute/VK/IncludeVK.h>
#include <Jolt/Core/UnorderedMap.h>
 
JPH_NAMESPACE_BEGIN
 
class ComputeSystemVK;

class JPH_EXPORT ComputeShaderVK : public ComputeShader
{
public:
    JPH_OVERRIDE_NEW_DELETE

                                        ComputeShaderVK(ComputeSystemVK *inComputeSystem, uint32 inGroupSizeX, uint32 inGroupSizeY, uint32 inGroupSizeZ) : ComputeShader(inGroupSizeX, inGroupSizeY, inGroupSizeZ), mComputeSystem(inComputeSystem) { }
    virtual                             ~ComputeShaderVK() override;

    bool                                Initialize(const Array<uint8> &inSPVCode, VkBuffer inDummyBuffer, ComputeShaderResult &outResult);

    uint32                              NameToBufferInfoIndex(const char *inName) const;

    VkPipeline                          GetPipeline() const                         { return mPipeline; }
    VkPipelineLayout                    GetPipelineLayout() const                   { return mPipelineLayout; }
    VkDescriptorSetLayout               GetDescriptorSetLayout() const              { return mDescriptorSetLayout; }
    const Array<VkDescriptorSetLayoutBinding> &GetLayoutBindings() const            { return mLayoutBindings; }
    const Array<VkDescriptorBufferInfo> &GetBufferInfos() const                     { return mBufferInfos; }
 
private:
    ComputeSystemVK *                   mComputeSystem;
    VkShaderModule                      mShaderModule = VK_NULL_HANDLE;
    VkPipelineLayout                    mPipelineLayout = VK_NULL_HANDLE;
    VkPipeline                          mPipeline = VK_NULL_HANDLE;
    VkDescriptorSetLayout               mDescriptorSetLayout = VK_NULL_HANDLE;
    Array<String>                       mBindingNames;                              
    UnorderedMap<string_view, uint32>   mNameToBufferInfoIndex;                     
    Array<VkDescriptorSetLayoutBinding> mLayoutBindings;
    Array<VkDescriptorBufferInfo>       mBufferInfos;
};
 
JPH_NAMESPACE_END
 
#endif // JPH_USE_VK