ContextWebGPU.h

#pragma once
 
#include "CommonWebGPU.h"
#include "../IBindGroups.h"
#include "../IContext.h"
#include "../Statistics.h"
 
#include <map>
#include <vector>
 
namespace Cogs {
  class GraphicsDeviceWebGPU;
  class EffectsWebGPU;
  struct BindGroupsWebGPU;
 
  // enum WGPUAnnotationType{
  //   ANNOTATION_TYPE_RENDER_PASS,
  //   ANNOTATION_TYPE_COMPUTE_PASS,
  //   ANNOTATION_TYPE_RENDER_BUNDLE,
  //   ANNOTATION_TYPE_COMMAND_ENCODER,
  // };
 
  class ContextWebGPU : public IContext {
  public:
    void initialize(GraphicsDeviceWebGPU *graphicsDevice);
 
    virtual const FrameStatistics& getLastFrameStatistics() override { return prevStats; }
    virtual const UploadStatistics& getLastUploadStatistics() override { return prevUploadStats; }
 
    virtual void frameStatisticsConfigure(bool enable) override { frameStatisticsEnabled = enable; }
 
    virtual void signal(FenceHandle /*fenceHandle*/) override;
 
    virtual void clearRenderTarget(const float* /*value*/) override;
    virtual void clearRenderTarget(const float** /*values*/, const int /*numvalues*/) override;
    virtual void clearDepth(const float /*depth*/ = 1.0f) override;
 
    virtual void pushCommandGroupAnnotation(const StringView & name) override;
    virtual void popCommandGroupAnnotation() override;
    virtual void setAnnotationMarker(const StringView & name) override;
 
    virtual void beginRenderPass(const RenderPassInfo &info) override;
    virtual void endRenderPass() override;
 
    virtual void setRenderTarget(const RenderTargetHandle handle, const DepthStencilHandle depthStencilHandle) override;
    virtual void setViewport(const float /*x*/, const float /*y*/, const float /*width*/, const float /*height*/) override;
    virtual void setScissor(const int /*x*/, const int /*y*/, const int /*width*/, const int /*height*/) override;
    virtual void setDepthStencilState(const DepthStencilStateHandle /*handle*/) override;
    virtual void setBlendState(const BlendStateHandle /*handle*/, const float* /*constant*/) override;
    virtual void setRasterizerState(const RasterizerStateHandle /*handle*/) override;
    void setDefaults();
    virtual void setEffect(EffectHandle /*handle*/) override;
 
    virtual void setTexture(const StringView& /*name*/, unsigned int /*unit*/, TextureHandle /*textureHandle*/) override;
    virtual void setTexture(const TextureBindingHandle /*textureBindingHandle*/, const TextureHandle /*textureHandle*/) override;
    virtual void setTexture(const StringView& /*name*/, TextureViewHandle /*textureViewHandle*/) override;
    virtual void setTexture(const TextureBindingHandle /*textureBindingHandle*/, TextureViewHandle /*textureViewHandle*/) override;
 
    virtual void setSamplerState(const StringView& /*name*/, unsigned int /*unit*/, SamplerStateHandle /*samplerStateHandle*/) override;
    virtual void setSamplerState(const SamplerStateBindingHandle /*samplerStateBindingHandle*/, const SamplerStateHandle /*samplerStateHandle*/) override;
 
    virtual void setInputLayout(const InputLayoutHandle /*inputLayoutHandle*/) override;
 
    virtual void setVertexBuffers(const VertexBufferHandle* /*vertexBufferHandles*/, const size_t /*count*/, const uint32_t* /*strides*/, const uint32_t* /*offsets*/) override;
    virtual void setVertexBuffers(const VertexBufferHandle* /*vertexBufferHandles*/, const size_t /*count*/) override;
 
    virtual void setIndexBuffer(IndexBufferHandle /*bufferHandle*/, uint32_t /*stride*/ = 4, uint32_t /*offset*/ = 0) override;
 
    virtual void setVertexArrayObject(VertexArrayObjectHandle /*vertexArrayObject*/) override;
 
    virtual void setBindGroup(BindGroupHandle bindGroupHandle, uint32_t groupIdx) override;
    virtual void setConstantBuffer(const StringView& /*name*/, const BufferHandle /*bufferHandle*/, const uint32_t /*offset*/ = 0, const uint32_t /*size*/ = ~0u) override;
    virtual void setConstantBuffer(const ConstantBufferBindingHandle /*bufferBindingHandle*/, const BufferHandle /*bufferHandle*/, const uint32_t /*offset*/ = 0, const uint32_t /*size*/ = ~0u) override;
    virtual void setBuffer(const StringView& /*name*/, BufferHandle /*bufferHandle*/) override;
    virtual void setBuffer(const BufferBindingHandle /*bufferBindingHandle*/, BufferHandle /*bufferHandle*/) override;
    virtual void setBufferCounter(BufferHandle /*bufferHandle*/, uint32_t /*value*/) override;
    virtual void setBufferCounter(BufferHandle /*bufferHandle*/, BufferHandle /*sourceBufferHandle*/) override;
 
    virtual void getBufferCounter(BufferHandle /*bufferHandle*/, BufferHandle /*destinationBufferHandle*/) override;
    virtual uint32_t getBufferCounter(BufferHandle /*bufferHandle*/) override { return 0; }
 
    virtual void draw(PrimitiveType /*primitiveType*/, const size_t /*startVertex*/, const size_t /*numVertexes*/) override;
    virtual void drawIndexed(PrimitiveType /*primitiveType*/, const size_t /*startIndex*/, const size_t /*numIndexes*/, const size_t /*startVertex*/ = 0) override;
    virtual void drawInstanced(PrimitiveType /*primitiveType*/, const size_t /*startVertex*/, const size_t /*numVertexes*/, const size_t /*startInstance*/, const size_t /*numInstances*/) override;
    virtual void drawInstancedIndexed(PrimitiveType /*primitiveType*/, const size_t /*startInstance*/, const size_t /*numInstances*/, const size_t /*startIndex*/, const size_t /*numIndexes*/) override;
 
    virtual void dispatchCompute(const unsigned int /*threadGroupsX*/, const unsigned int /*threadGroupsY*/, const unsigned int /*threadGroupsZ*/) override;
 
    virtual void readDepthBuffer(BufferHandle /*bufferHandle*/, int /*x*/, int /*y*/, int /*width*/, int /*height*/, Framebuffer /*framebuffer*/) override;
    virtual void readColorBuffer(BufferHandle /*bufferHandle*/, int /*x*/, int /*y*/, int /*width*/, int /*height*/, Framebuffer /*framebuffer*/) override;
 
    virtual void*map(BufferHandle /*bufferHandle*/, MapMode::EMapMode /*mapMode*/, uint32_t* /*stride*/ = nullptr) override;
    virtual void*map(TextureHandle /*textureHandle*/, MapMode::EMapMode /*accessMode*/, uint32_t* /*rowPitch*/, uint32_t* /*depthPitch*/) override;
    virtual void unmap(BufferHandle /*bufferHandle*/) override;
    virtual void unmap(TextureHandle /*textureHandle*/) override;
    virtual void updateBuffer(BufferHandle /*bufferHandle*/, const void* /*data*/, const size_t /*size*/) override;
 
    virtual void updateSubTexture(TextureHandle /*textureHandle*/, const size_t /*level*/, const void* /*data*/) override;
    virtual void updateSubBuffer(BufferHandle /*bufferHandle*/, const size_t /*offset*/, const size_t /*size*/, const void* /*data*/) override;
 
    virtual void resolveResource(TextureHandle /*source*/, TextureHandle /*destination*/) override;
    virtual void copyResource(BufferHandle /*destinationHandle*/, BufferHandle /*sourceHandle*/) override;
    virtual void copyResource(TextureHandle /*destinationHandle*/, TextureHandle /*sourceHandle*/) override;
    virtual void copyTexture(TextureHandle /*dstHandle*/, unsigned /*dstSub*/, unsigned /*dstX*/, unsigned /*dstY*/, unsigned /*dstZ*/, TextureHandle /*sourceHandle*/, unsigned /*srcSub*/) override;
 
    virtual void clearResource(BufferHandle /*destinationHandle*/, uint32_t* /*values*/) override;
    virtual void clearResource(BufferHandle /*destinationHandle*/, float* /*values*/) override;
 
  private:
    FrameStatistics prevStats{};
    FrameStatistics currStats{};
 
    UploadStatistics prevUploadStats{};
    UploadStatistics currUploadStats{};
 
 
    bool frameStatisticsEnabled = false;
    bool uploadStatisticsEnabled = true;
 
    // Invoked by draw call to account for submission.
    void frameStatisticsAccountDrawCall(size_t count, bool indexed);
 
    // Invoked by upload calls
    void uploadStatisticsBufferUpload(size_t size);
    void uploadStatisticsTextureUpload(size_t size);
 
    bool createOwnedBindGroup(uint32_t group);
 
    EffectHandle getCurrentEffect();
    EffectsWebGPU *getEffects();
    BindGroupsWebGPU *getBindGroups();
  public:
    void beginFrame();
 
    void beginRenderPassInt();
    void endRenderPassInt();
    void updateRenderPass();
    bool updateRenderPipeline(PrimitiveType primitiveType);
 
    void beginComputePass();
    void endComputePass();
    void updateComputePass();
    bool updateComputePipeline();
 
    void ClearBindGroups();
 
    GraphicsDeviceWebGPU *graphicsDevice = nullptr;
 
    WGPURenderPassEncoder renderPassEncoder = 0;
    WGPUComputePassEncoder computePassEncoder = 0;
 
    EffectHandle effect = {};
    InputLayoutHandle inputLayoutHandle = {};
    RasterizerStateHandle rasterizeStateHandle = {};
    DepthStencilStateHandle depthStencilStateHandle = {};
    BlendStateHandle blendStateHandle = {};
 
    RenderPipelineHandle currentRenderPipeline = {};
    ComputePipelineHandle currentComputePipeline = {};
    WGPUIndexFormat currentIndexFormat = WGPUIndexFormat_Uint32;
 
    bool inRenderPass = false;
    RenderTargetHandle renderTargetHandle = {};
    DepthStencilHandle depthStencilHandle = {};
    WGPUColor clearColor[8] = {};
    float clearDepthVal = 1.0f;
    bool do_clear_render_target = false;
    bool do_clear_depth = false;
    bool update_render_target = true;
 
    static constexpr uint32_t MAX_BIND_GROUPS = static_cast<uint32_t>(MaxBindGroups);
    BindGroupHandle bind_group_handles[MAX_BIND_GROUPS] { BindGroupHandle::NoHandle };
    bool bind_group_owned_by_context[MAX_BIND_GROUPS]{ false };
 
  };
}