ComputeShader.h

浏览该文件的文档.

#pragma once
 
#include <array>
#include <cstdint>
 
namespace eve::gpgpu {
 
class GpuBuffer;
 
class ComputeShader {
public:
    static constexpr int kMaxBindings = 8;
    static constexpr int kMaxFloats = 32;
    static constexpr uint32_t kPushConstantBytes = uint32_t(kMaxFloats * sizeof(float));
 
    ComputeShader() = default;
    virtual ~ComputeShader() = default;
 
    ComputeShader(const ComputeShader &) = delete;
    ComputeShader &operator=(const ComputeShader &) = delete;
 
    virtual void bindBuffer(int binding, GpuBuffer *buffer) = 0;
    virtual GpuBuffer *getBoundBuffer(int binding) const = 0;
 
    virtual void setFloat(int index, float value) = 0;
    virtual float getFloat(int index) const = 0;
 
    virtual void clearBindings() = 0;
 
    const float *pushConstantData() const { return push_.data(); }
 
protected:
    std::array<float, kMaxFloats> push_{};
};
 
}  // namespace eve::gpgpu