vtkWebGPUComputePassInternals.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
 
#ifndef vtkWebGPUComputePassInternals_h
#define vtkWebGPUComputePassInternals_h
 
#include "Private/vtkWebGPUComputePassBufferStorageInternals.h"
#include "Private/vtkWebGPUComputePassTextureStorageInternals.h"
#include "Private/vtkWebGPUHandle.h"
#include "vtkObject.h"
#include "vtkSmartPointer.h"
#include "vtkWebGPUConfiguration.h"
#include "vtk_wgpu.h"
 
#include <unordered_map>
#include <unordered_set>
 
class vtkWebGPUComputePipeline;
 
VTK_ABI_NAMESPACE_BEGIN

class VTKRENDERINGWEBGPU_EXPORT vtkWebGPUComputePassInternals : public vtkObject
{
public:
  static vtkWebGPUComputePassInternals* New();
  vtkTypeMacro(vtkWebGPUComputePassInternals, vtkObject);
 
  void PrintSelf(ostream& os, vtkIndent indent) override;

  void SetParentPass(vtkWeakPointer<vtkWebGPUComputePass> parentPass);


  void SetWGPUConfiguration(vtkWebGPUConfiguration* config);
  vtkGetSmartPointerMacro(WGPUConfiguration, vtkWebGPUConfiguration);


  vtkWeakPointer<vtkWebGPUComputePipeline> GetAssociatedPipeline();
  void SetAssociatedPipeline(vtkWeakPointer<vtkWebGPUComputePipeline> associatedPipeline);

  WGPUBindGroupLayoutEntry CreateBindGroupLayoutEntry(
    uint32_t binding, vtkWebGPUComputeBuffer::BufferMode mode);

  WGPUBindGroupLayoutEntry CreateBindGroupLayoutEntry(uint32_t binding,
    vtkSmartPointer<vtkWebGPUComputeTexture> computeTexture,
    vtkSmartPointer<vtkWebGPUComputeTextureView> textureView);

  WGPUBindGroupLayoutEntry CreateBindGroupLayoutEntry(uint32_t binding,
    vtkSmartPointer<vtkWebGPUComputeTexture> computeTexture,
    WGPUTextureViewDimension textureViewDimension);

  WGPUBindGroupEntry CreateBindGroupEntry(
    WGPUBuffer buffer, uint32_t binding, vtkWebGPUComputeBuffer::BufferMode mode, uint32_t offset);

  WGPUBindGroupEntry CreateBindGroupEntry(uint32_t binding, WGPUTextureView textureView);

  void CreateShaderModule();

  void CreateBindGroupsAndLayouts();

  static vtkWebGPU::BindGroupLayout CreateBindGroupLayout(
    const WGPUDevice& device, const std::vector<WGPUBindGroupLayoutEntry>& layoutEntries);

  std::vector<WGPUBindGroupEntry> CreateBindGroupEntries(
    const std::vector<vtkWebGPUComputeBuffer*>& buffers);

  bool CheckBufferIndex(int bufferIndex, const std::string& callerFunctionName);

  bool CheckTextureIndex(int textureIndex, const std::string& callerFunctionName);

  bool CheckTextureViewIndex(int textureViewIndex, const std::string& callerFunctionName);

  bool CheckBufferCorrectness(vtkSmartPointer<vtkWebGPUComputeBuffer> buffer);

  void RecreateBufferBindGroup(int bufferIndex);

  void RegisterBufferToPipeline(
    vtkSmartPointer<vtkWebGPUComputeBuffer> buffer, vtkWebGPU::Buffer wgpuBuffer);


  bool GetRegisteredBufferFromPipeline(
    vtkSmartPointer<vtkWebGPUComputeBuffer> buffer, vtkWebGPU::Buffer& wgpuBuffer);
 
  bool GetRegisteredTextureFromPipeline(
    vtkSmartPointer<vtkWebGPUComputeTexture> texture, vtkWebGPU::Texture& wgpuTexture);

  vtkWebGPU::Buffer GetWGPUBuffer(std::size_t bufferIndex);

  void RegisterTextureToPipeline(
    vtkSmartPointer<vtkWebGPUComputeTexture> texture, vtkWebGPU::Texture wgpuTexture);

  void RecreateBuffer(int bufferIndex, vtkIdType newByteSize);

  void RecreateTexture(int textureIndex);

  void RecreateTextureViews(int textureIndex);

  vtkWebGPU::TextureView CreateWebGPUTextureView(
    vtkSmartPointer<vtkWebGPUComputeTextureView> textureView, vtkWebGPU::Texture wgpuTexture);

  void UpdateWebGPUBuffer(
    vtkSmartPointer<vtkWebGPUComputeBuffer> buffer, vtkWebGPU::Buffer wgpuBuffer);

  void UpdateComputeTextureAndViews(
    vtkSmartPointer<vtkWebGPUComputeTexture> texture, vtkWebGPU::Texture newWgpuTexture);

  void RecreateTextureBindGroup(int textureIndex);

  void SetupRenderBuffer(vtkSmartPointer<vtkWebGPUComputeRenderBuffer> renderBuffer);

  void RecreateRenderTexture(vtkSmartPointer<vtkWebGPUComputeRenderTexture> renderTexture);

  void CreateWebGPUComputePipeline();

  vtkWebGPU::PipelineLayout CreateWebGPUComputePipelineLayout();

  WGPUCommandEncoder CreateCommandEncoder();

  WGPUComputePassEncoder CreateComputePassEncoder(const WGPUCommandEncoder& commandEncoder);

  void WebGPUDispatch(unsigned int groupsX, unsigned int groupsY, unsigned int groupsZ);

  void SubmitCommandEncoderToQueue(const WGPUCommandEncoder& commandEncoder);

  void ReleaseResources();
 
  // Compute pass whose internals this class represents
  vtkWeakPointer<vtkWebGPUComputePass> ParentPass;
 
protected:
  vtkWebGPUComputePassInternals() = default;
  ~vtkWebGPUComputePassInternals() override;
 
private:
  friend class vtkWebGPUComputePass;
  friend class vtkWebGPUComputePassBufferStorageInternals;
  friend class vtkWebGPUComputePassTextureStorageInternals;
  friend class vtkWebGPUComputePipeline;
  // For the mapper to be able to access the vtkWebGPU::Buffer objects for use in a render pipeline
  friend class vtkWebGPUPointCloudMapperInternals;
 
  vtkWebGPUComputePassInternals(const vtkWebGPUComputePassInternals&) = delete;
  void operator=(const vtkWebGPUComputePassInternals&) = delete;

  bool Initialized = false;
  // Whether or not the binds group / layouts have changed since they were last created and so they
  // need to be recreated. Defaults to true since the bind group / layouts are initially not created
  // so they are invalid.
  bool BindGroupOrLayoutsInvalidated = true;
 
  // Device of the compute pipeline this pass belongs to. Used to submit commands.
  vtkSmartPointer<vtkWebGPUConfiguration> WGPUConfiguration;
 
  // The compute pipeline this compute pass belongs to.
  vtkWeakPointer<vtkWebGPUComputePipeline> AssociatedPipeline;
 
  vtkWebGPU::ShaderModule ShaderModule;
  // List of the bind groups, used to set the bind groups of the compute pass at each dispatch
  std::vector<vtkWebGPU::BindGroup> BindGroups;
  // Maps a bind group index to to the list of bind group entries for this group. These
  // entries will be used at the creation of the bind groups
  std::unordered_map<int, std::vector<WGPUBindGroupEntry>> BindGroupEntries;
  std::vector<vtkWebGPU::BindGroupLayout> BindGroupLayouts;
  // Maps a bind group index to to the list of bind group layout entries for this group.
  // These layout entries will be used at the creation of the bind group layouts
  std::unordered_map<int, std::vector<WGPUBindGroupLayoutEntry>> BindGroupLayoutEntries;
  // WebGPU compute shader pipeline
  vtkWebGPU::ComputePipeline ComputePipeline;
 
  // Object responsible for the management (creation, re-creatioin, deletion, ...) of textures and
  // their texture views
  vtkSmartPointer<vtkWebGPUComputePassTextureStorageInternals> TextureStorage;
 
  // Object responsible for the management (creation, re-creatioin, deletion, ...) of buffers
  vtkSmartPointer<vtkWebGPUComputePassBufferStorageInternals> BufferStorage;
};
 
VTK_ABI_NAMESPACE_END
 
#endif