VulkanComputeShader.h

浏览该文件的文档.

#pragma once
 
#include "gpgpu/ComputeShader.h"
#include "vkbuilder.hpp"
 
#include <array>
#include <vector>
 
namespace eve::gpgpu {
 
class VulkanComputeShader final : public ComputeShader {
public:
    ~VulkanComputeShader() override;
 
    void bindBuffer(int binding, GpuBuffer *buffer) override;
    GpuBuffer *getBoundBuffer(int binding) const override;
    void setFloat(int index, float value) override;
    float getFloat(int index) const override;
    void clearBindings() override;
 
    void flushDescriptors(vkb::Device &device);
 
    void beginSequence();
    void endSequence();
 
    void releasePendingDescriptors(vkb::Device &device);
 
    vkb::Device *device_ = nullptr;
    vk::ShaderModule module_{};
    vk::DescriptorSetLayout setLayout_{};
    vk::PipelineLayout pipelineLayout_{};
    vk::Pipeline pipeline_{};
    vk::DescriptorPool descriptorPool_{};
    vk::DescriptorSet descriptorSet_{};
    vk::Buffer dummyBuffer_{};
    vk::DeviceMemory dummyMemory_{};
#if defined(VKB_ENABLE_VMA)
    VmaAllocation dummyVmaAllocation_ = VK_NULL_HANDLE;
#endif
 
    std::array<GpuBuffer *, kMaxBindings> bindings_{};
    bool descriptorsDirty_ = true;
 
    // Sets superseded by newer flushes; still referenced by pending command
    // buffers until the owning Sequence submits and waits.
    std::vector<vk::DescriptorSet> pendingSets_;
    uint32_t                       activeSequenceCount_ = 0;
};
 
}  // namespace eve::gpgpu