EffectsWebGPU.h

#pragma once
 
#include "CommonWebGPU.h"
 
#include "../Base/EffectsCommon.h"
 
#include <map>
#include <unordered_map>
#include "BindGroupsWebGPU.h"
 
namespace Cogs {
  class GraphicsDeviceWebGPU;
  struct BindGroupWebGPU;
 
  enum WebGPUConstantBufferType { UniformBuffer, Texture, Sampler };
 
  struct WebGPUConstantBufferBinding {
    size_t group =0;
    size_t nameHash =0;
    WGPUBindGroupLayoutEntry bg_ent;
    std::string name;
  };
 
  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];
    BindGroupSetDescription bindGroupSetDescription;
    size_t bindGroupDescHash[MaxBindGroups] = {};
 
    void updateBindGroupDescHashes()
    {
      for (size_t g = 0; g < MaxBindGroups; ++g) {
        bindGroupDescHash[g] = bindGroupSetDescription.groups[g].hash();
      }
    }
 
    size_t hash(size_t base) const
    {
      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;
 
    virtual const BindGroupDescription& getBindGroupDescription(EffectHandle effectHandle, uint32_t groupIndex) override;
 
    size_t getBindGroupDescHash(EffectHandle effectHandle, uint32_t groupIndex);
 
    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);
 
    uint32_t getNumBindGroups(EffectHandle effectHandle) override;
 
    GraphicsDeviceWebGPU *graphicsDevice = nullptr;
 
    SafeResourceMap<EffectHandle, EffectWebGPU> effects;
 
  private:
  };
}