WebGPUShaderProgram.hpp

Go to the documentation of this file.

#pragma once
 
#include "../Interface/ShaderProgram.hpp"
 
#include <initializer_list>
#include <vector>
#include "WebGPU.hpp"
 
namespace Video {
 
class Shader;
class WebGPUShader;
class WebGPURenderer;
 
class WebGPUShaderProgram : public ShaderProgram {
  public:
 
    WebGPUShaderProgram(WGPUDevice device, std::initializer_list<const Shader*> shaders);
 
    ~WebGPUShaderProgram() final;
 
 
    const std::vector<const WebGPUShader*>& GetShaders() const;
 
 
    const WGPUBindGroupLayout* GetBindGroupLayouts() const;
 
 
    bool IsBindGroupLayoutEmpty(ShaderProgram::BindingType bindingType) const;
 
 
    bool HasUniformsBuffer() const;
 
 
    uint32_t GetPushConstantSize() const;
 
  private:
    WebGPUShaderProgram(const WebGPUShaderProgram& other) = delete;
 
    struct StorageBufferInfo {
        WGPUShaderStageFlags shaderStages;
        bool readWrite;
    };
 
    void AddMatrices(WGPUDevice device);
    void AddStorageBuffers(WGPUDevice device, const std::vector<StorageBufferInfo>& storageBufferInfos);
    void AddMaterial(WGPUDevice device, unsigned int count);
    void AddPushConstants(WGPUDevice device, unsigned int pushConstantCount, ShaderSource::ReflectionInfo::PushConstant* pushConstants);
    void AddUniforms(WGPUDevice device, bool hasUniformsBuffer);
 
    std::vector<const WebGPUShader*> shaders;
    // And extra bind group for emulated push constants.
    WGPUBindGroupLayout bindGroupLayouts[ShaderProgram::BindingType::BINDING_TYPES];
    bool bindGroupLayoutIsEmpty[ShaderProgram::BindingType::BINDING_TYPES];
 
    bool hasUniformsBuffer = false;
    uint32_t pushConstantSize = 0;
};
 
} // namespace Video