vtkOpenGLSurfaceProbeVolumeMapper.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 vtkOpenGLSurfaceProbeVolumeMapper_h
#define vtkOpenGLSurfaceProbeVolumeMapper_h
 
#include "vtkNew.h" // For vtkNew
#include "vtkOpenGLPolyDataMapper.h"
#include "vtkRenderingVolumeOpenGL2Module.h" // For export macro
 
VTK_ABI_NAMESPACE_BEGIN
class vtkOpenGLFramebufferObject;
class vtkOpenGLRenderWindow;
class vtkTextureObject;
class vtkVolumeTexture;
 
class VTKRENDERINGVOLUMEOPENGL2_EXPORT vtkOpenGLSurfaceProbeVolumeMapper
  : public vtkOpenGLPolyDataMapper
{
public:
  static vtkOpenGLSurfaceProbeVolumeMapper* New();
  vtkTypeMacro(vtkOpenGLSurfaceProbeVolumeMapper, vtkOpenGLPolyDataMapper);


  void SetProbeInputData(vtkPolyData* in);
  vtkPolyData* GetProbeInput();
  void SetProbeInputConnection(vtkAlgorithmOutput* algOutput);


  void SetSourceData(vtkImageData* in);
  vtkImageData* GetSource();
  void SetSourceConnection(vtkAlgorithmOutput* algOutput);


  vtkGetMacro(Window, double);
  vtkSetMacro(Window, double);
 
  vtkGetMacro(Level, double);
  vtkSetMacro(Level, double);


  enum class BlendModes : unsigned int
  {
    NONE = 0,
    MAX,
    MIN,
    AVERAGE
  };
 
  vtkGetEnumMacro(BlendMode, BlendModes);
  vtkSetEnumMacro(BlendMode, BlendModes);
  void SetBlendModeToNone() { this->SetBlendMode(BlendModes::NONE); }
  void SetBlendModeToMaximumIntensity() { this->SetBlendMode(BlendModes::MAX); }
  void SetBlendModeToMinimumIntensity() { this->SetBlendMode(BlendModes::MIN); }
  void SetBlendModeToAverageIntensity() { this->SetBlendMode(BlendModes::AVERAGE); }


  vtkGetMacro(BlendWidth, double);
  vtkSetMacro(BlendWidth, double);
 
  void RenderPiece(vtkRenderer* ren, vtkActor* act) override;
 
  void UpdateShaders(vtkOpenGLHelper& cellBO, vtkRenderer* ren, vtkActor* act) override;
 
protected:
  int FillInputPortInformation(int port, vtkInformation* info) override;
 
  virtual void ReplaceShaderPositionPass(vtkActor* act);
  virtual void ReplaceShaderProbePass(vtkActor* act);
  virtual void UpdateShadersProbePass(vtkOpenGLHelper& cellBO, vtkRenderer* ren);
 
private:
  vtkOpenGLSurfaceProbeVolumeMapper();
  ~vtkOpenGLSurfaceProbeVolumeMapper() override = default;
 
  void CreateTexture(vtkTextureObject*, vtkOpenGLRenderWindow*);
  void ReplaceActiveFBO(vtkRenderer*);
  void RestoreActiveFBO(vtkRenderer*);
 
  vtkNew<vtkOpenGLFramebufferObject> FBO;
 
  vtkNew<vtkTextureObject> PositionsTextureObject;
  vtkNew<vtkTextureObject> NormalsTextureObject;
  vtkNew<vtkVolumeTexture> VolumeTexture;
 
  vtkNew<vtkImageData> TransformedSource;
 
  // Internal pass type used for shader updates
  enum class PassTypes : unsigned int
  {
    DEFAULT = 0,
    POSITION_TEXTURE,
    PROBE
  };
  PassTypes CurrentPass = PassTypes::DEFAULT;
 
  // Window / level
  double Window = 1.0;
  double Level = 0.0;
 
  // Blend mode
  BlendModes BlendMode = BlendModes::NONE;
  double BlendWidth = 1.0;
 
  // Saved state
  bool SavedScissorTestState = false;
  bool SavedBlendState = false;
  int SavedViewport[4] = {};
};
 
VTK_ABI_NAMESPACE_END
#endif