EffectsWebGPU.h

#pragma once
 
#include "CommonWebGPU.h"
 
#include "../Base/EffectsCommon.h"
 
namespace Cogs {
  class GraphicsDeviceWebGPU;
 
  enum WebGPUConstantBufferUsage {
    ShaderStage_None = 0x00000000,
    ShaderStage_Vertex = 0x00000001,
    ShaderStage_Fragment = 0x00000002,
    ShaderStage_Compute = 0x00000004,
    ShaderStage_Force32 = 0x7FFFFFFF
  }; // Copied from webgpu.h
 
  enum WebGPUConstantBufferType { UniformBuffer, Texture, Sampler };
 
  struct WebGPUConstantBufferBinding {
    size_t group;
    size_t nameHash;
    WGPUBindGroupLayoutEntry bg_ent;
  };
 
  struct SemanticSlotBinding {
    uint8_t semantic;
    uint8_t slot;
    uint8_t binding;
    WGPUVertexFormat format;
    size_t nameHash;
  };
 
  struct EffectWebGPU : public Effect
  {
    static const size_t maxTexUnits = 128;
    static const size_t maxConstantBuffers = 128;
    static const size_t maxVertexAttribs = 32;
    WGPUShaderModule vs_module;
    WGPUShaderModule fs_module;
    WGPUShaderModule cs_module;
    std::string vs_entry;
    std::string fs_entry;
    std::string cs_entry;
    std::string name;
    size_t num_bindings;
    size_t num_attribs;
    WebGPUConstantBufferBinding constantBufferBindings[maxConstantBuffers];
    SemanticSlotBinding semanticSlotBindings[maxVertexAttribs];
  };
 
  class EffectsWebGPU : public EffectsCommon
  {
  public:
    void initialize(GraphicsDeviceWebGPU *device, IBuffers * buffers);
 
    virtual void releaseEffect(EffectHandle effectHandle) override;
    virtual void releaseResources() override;
 
    virtual void annotate(EffectHandle /*handle*/, const StringView& /*name*/) override {}
    virtual void annotateVS(EffectHandle /*handle*/, const StringView& /*name*/) override {}
    virtual void annotateGS(EffectHandle /*handle*/, const StringView& /*name*/) override {}
    virtual void annotatePS(EffectHandle /*handle*/, const StringView& /*name*/) override {}
    virtual void annotateCS(EffectHandle /*handle*/, const StringView& /*name*/) override {}
 
    virtual EffectHandle loadComputeEffect(const StringView& /*fileName*/, EffectFlags::EEffectFlags /*effectFlags*/) override;
    virtual EffectHandle loadComputeEffect(const StringView& /*fileName*/, const PreprocessorDefinitions& /*defines*/, EffectFlags::EEffectFlags /*effectFlags*/) override;
 
    virtual EffectHandle load(const ProcessedContent&   /*vsSource*/,
                              const ProcessedContent&   /*hsSource*/,
                              const ProcessedContent&   /*dsSource*/,
                              const ProcessedContent&   /*gsSource*/,
                              const ProcessedContent&   /*psSource*/,
                              const StringView&         /*vsEntryPoint*/,
                              const StringView&         /*hsEntryPoint*/,
                              const StringView&         /*dsEntryPoint*/,
                              const StringView&         /*gsEntryPoint*/,
                              const StringView&         /*psEntryPoint*/,
                              const EffectDescription&  /*desc*/) override;
 
    // TODO fix this
    ConstantBufferBindingHandle getConstantBufferBinding(EffectHandle effectHandle, const StringView& name) override;
    BufferBindingHandle getBufferBinding(EffectHandle effectHandle, const StringView& name) override;
    void releaseBufferBinding(BufferBindingHandle /*bufferBindingHandle*/) override
    {
    }
    TextureBindingHandle getTextureBinding(EffectHandle effectHandle, const StringView& name, const unsigned int slot) override;
    SamplerStateBindingHandle getSamplerStateBinding(EffectHandle effectHandle, const StringView& /*name*/, const unsigned int slot) override;
 
    bool addConstantBufferBindings(EffectWebGPU& effect, const std::vector<Cogs::WebGPUConstantBufferBinding>& bindings);
    const WebGPUConstantBufferBinding* getConstantBufferBindings(EffectHandle effectHandle, size_t& num_bindings);
 
    GraphicsDeviceWebGPU *graphicsDevice = nullptr;
 
    SafeResourceMap<EffectHandle, EffectWebGPU> effects;
  };
}