EffectsWebGPU.h

#pragma once
 
#include "CommonWebGPU.h"
 
#include "../Base/EffectsCommon.h"
 
namespace Cogs {
  class GraphicsDeviceWebGPU;
 
  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];
 
    size_t hash(size_t base)
    {
      base = Cogs::hash((size_t)vs_module, base);
      base = Cogs::hash((size_t)fs_module, base);
      base = Cogs::hash((size_t)cs_module, base);
      base = Cogs::hash(vs_entry.c_str(), base);
      base = Cogs::hash(fs_entry.c_str(), base);
      base = Cogs::hash(cs_entry.c_str(), base);
      return base;
    }
  };
 
  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;
  };
}