vtkWebGPUComputePipeline.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 vtkWebGPUComputePipeline_h
#define vtkWebGPUComputePipeline_h

#include "vtkObject.h"
#include "vtkWebGPUComputePass.h"   // for the list of compute passes held by this pipeline
#include "vtkWebGPUConfiguration.h" // for requesting device / adapter
#include "vtk_wgpu.h"               // for webgpu

#include <unordered_map> // for the registered buffers / textures

VTK_ABI_NAMESPACE_BEGIN

class vtkWebGPURenderWindow;
class vtkWebGPURenderer;

class VTKRENDERINGWEBGPU_EXPORT vtkWebGPUComputePipeline : public vtkObject
{
public:
  vtkTypeMacro(vtkWebGPUComputePipeline, vtkObject);
  static vtkWebGPUComputePipeline* New();

  void PrintSelf(ostream& os, vtkIndent indent) override;


  vtkGetMacro(Label, std::string&);
  vtkSetMacro(Label, std::string);

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

  vtkSmartPointer<vtkWebGPUComputePass> CreateComputePass();

  const std::vector<vtkSmartPointer<vtkWebGPUComputePass>>& GetComputePasses() const;

  void DispatchAllPasses();

  void Update();

  void ReleaseResources();

protected:
  vtkWebGPUComputePipeline();
  ~vtkWebGPUComputePipeline() override;

private:
  friend class vtkWebGPUComputePassInternals;
  friend class vtkWebGPUCellToPrimitiveConverter;
  friend class vtkWebGPURenderWindow;
  friend class vtkWebGPURenderer;

  vtkWebGPUComputePipeline(const vtkWebGPUComputePipeline&) = delete;
  void operator=(const vtkWebGPUComputePipeline&) = delete;

  void RegisterBuffer(vtkSmartPointer<vtkWebGPUComputeBuffer> buffer, wgpu::Buffer wgpuBuffer);

  void RegisterTexture(vtkSmartPointer<vtkWebGPUComputeTexture> texture, wgpu::Texture wgpuTexture);


  bool GetRegisteredBuffer(
    vtkSmartPointer<vtkWebGPUComputeBuffer> buffer, wgpu::Buffer& wgpuBuffer);

  bool GetRegisteredTexture(
    vtkSmartPointer<vtkWebGPUComputeTexture> texture, wgpu::Texture& wgpuTexture);

  void EnsureConfigured();

  vtkSmartPointer<vtkWebGPUConfiguration> WGPUConfiguration = nullptr;

  std::vector<vtkSmartPointer<vtkWebGPUComputePass>> ComputePasses;

  std::unordered_map<vtkSmartPointer<vtkWebGPUComputeBuffer>, wgpu::Buffer> RegisteredBuffers;
  std::unordered_map<vtkSmartPointer<vtkWebGPUComputeTexture>, wgpu::Texture> RegisteredTextures;

  // Label used for debugging
  std::string Label = "WebGPU Compute Pipeline";
};

VTK_ABI_NAMESPACE_END

#endif