PipelineStatesWebGPU.h

#pragma once
 
#include "CommonWebGPU.h"
 
#include "../Base/ResourceMap.h"
#include "../IPipelineStates.h"
 
#include <unordered_map>
 
namespace Cogs {
  class GraphicsDeviceWebGPU;
 
  struct RenderPipelineWebGPU
  {
    size_t hash;
    EffectHandle effect;
    InputLayoutHandle inputLayoutHandle;
    PrimitiveType primitiveType;
    RasterizerStateHandle rasterizeStateHandle;
    DepthStencilStateHandle depthStencilStateHandle;
    BlendStateHandle blendStateHandle;
    RenderTargetHandle renderTargetHandle;
    DepthStencilHandle depthStencilHandle;
 
    WGPURenderPipeline pipeline;
    WGPUPipelineLayout pipeline_layout;
    WGPUBindGroupLayout bind_group_layout;
  };
  struct ComputePipelineWebGPU
  {
    size_t hash;
    EffectHandle effect;
 
    WGPUComputePipeline pipeline;
    WGPUBindGroupLayout layout;
  };
 
  struct PipelineStatesWebGPU : public IPipelineStates {
 
    void initialize(GraphicsDeviceWebGPU *device);
 
    size_t renderPipelineHash(EffectHandle effect,
                              InputLayoutHandle inputLayoutHandle,
                              PrimitiveType primitiveType,
                              RasterizerStateHandle rasterizeStateHandle,
                              DepthStencilStateHandle depthStencilStateHandle,
                              BlendStateHandle blendStateHandle,
                              RenderTargetHandle renderTargetHandle,
                              DepthStencilHandle depthStencilHandle,
                              WGPUIndexFormat stripIndexFormat = WGPUIndexFormat_Undefined);
    RenderPipelineHandle loadRenderPipeline(EffectHandle effect,
                                            InputLayoutHandle inputLayoutHandle,
                                            PrimitiveType primitiveType,
                                            RasterizerStateHandle rasterizeStateHandle,
                                            DepthStencilStateHandle depthStencilStateHandle,
                                            BlendStateHandle blendStateHandle,
                                            RenderTargetHandle renderTargetHandle,
                                            DepthStencilHandle depthStencilHandle,
                                            WGPUIndexFormat stripIndexFormat = WGPUIndexFormat_Undefined);
    ComputePipelineHandle loadComputePipeline(EffectHandle effect);
 
    void releaseRenderPipeline(RenderPipelineHandle psoHandle) override;
    void releaseComputePipeline(ComputePipelineHandle psoHandle) override;
    void releaseResources() override;
 
    GraphicsDeviceWebGPU *graphics_device = nullptr;
 
    std::unordered_map<size_t, RenderPipelineHandle> renderPipelineHashMap;
    SafeResourceMap<RenderPipelineHandle, RenderPipelineWebGPU> renderPipeline;
 
    std::unordered_map<size_t, ComputePipelineHandle> computePipelineHashMap;
    SafeResourceMap<ComputePipelineHandle, ComputePipelineWebGPU> computePipeline;
  };
}