vtkWebGPUSkybox.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 vtkWebGPUSkybox_h
#define vtkWebGPUSkybox_h

#include "vtkSkybox.h"

#include "vtkNew.h"                   // for ivars
#include "vtkRenderingWebGPUModule.h" // For export macro
#include "vtkWrappingHints.h"         // For VTK_MARSHALAUTO
#include "vtk_wgpu.h"                 // for webgpu

VTK_ABI_NAMESPACE_BEGIN
class vtkMatrix3x3;
class vtkOverrideAttribute;
class vtkWebGPUConfiguration;
class vtkWebGPURenderWindow;
class vtkWebGPURenderer;

class VTKRENDERINGWEBGPU_EXPORT VTK_MARSHALAUTO vtkWebGPUSkybox : public vtkSkybox
{
public:
  static vtkWebGPUSkybox* New();
  VTK_NEWINSTANCE
  static vtkOverrideAttribute* CreateOverrideAttributes();
  vtkTypeMacro(vtkWebGPUSkybox, vtkSkybox);
  void PrintSelf(ostream& os, vtkIndent indent) override;

  void Render(vtkRenderer* ren, vtkMapper* mapper) override;

  void ReleaseGraphicsResources(vtkWindow* window) override;

protected:
  vtkWebGPUSkybox();
  ~vtkWebGPUSkybox() override;

private:
  vtkWebGPUSkybox(const vtkWebGPUSkybox&) = delete;
  void operator=(const vtkWebGPUSkybox&) = delete;

  std::string BuildShaderSource();

  void CreatePipeline(vtkWebGPURenderWindow* wgpuRenderWindow);

  void CreateBindGroup(vtkWebGPUConfiguration* wgpuConfiguration);

  void UpdateUniformBuffer(vtkWebGPUConfiguration* wgpuConfiguration, vtkRenderer* ren);

  // Uniform block matching the WGSL struct layout
  struct SkyboxUniforms
  {
    float CameraPosition[4];              // vec4 (padded from vec3)
    float FloorPlane[4];                  // vec4
    float FloorRight[4];                  // vec4 (padded from vec3)
    float FloorFront[4];                  // vec4 (padded from vec3)
    float FloorTexCoordScale[2];          // vec2
    float LeftEye;                        // f32
    float ProjectionMode;                 // f32: 0=Cube, 1=Sphere, 2=StereoSphere, 3=Floor
    alignas(16) float RotationMatrix[12]; // mat3x3 in WGSL = 3 x vec4 (each row padded to vec4)
  };

  int LastProjection = -1;
  bool LastGammaCorrect = false;

  wgpu::RenderPipeline Pipeline;
  wgpu::BindGroupLayout BindGroupLayout;
  wgpu::BindGroup BindGroup;
  wgpu::Buffer UniformBuffer;
  wgpu::Buffer MatrixBuffer;
  wgpu::BindGroupLayout MatrixBindGroupLayout;
  wgpu::BindGroup MatrixBindGroup;
  std::string PipelineKey;

  vtkNew<vtkMatrix3x3> RotationMatrix;

  vtkMTimeType TextureBuildTime = 0;
};

#define vtkWebGPUSkybox_OVERRIDE_ATTRIBUTES vtkWebGPUSkybox::CreateOverrideAttributes()
VTK_ABI_NAMESPACE_END
#endif