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
#include "vtkVolumeTexture.h"                // For VolumeTexture
#include "vtkWrappingHints.h"                // For VTK_MARSHALAUTO
 
VTK_ABI_NAMESPACE_BEGIN
class vtkOpenGLFramebufferObject;
class vtkOpenGLRenderWindow;
class vtkTextureObject;
 
class VTKRENDERINGVOLUMEOPENGL2_EXPORT VTK_MARSHALAUTO 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);

  vtkSetSmartPointerMacro(VolumeTexture, vtkVolumeTexture);


  double GetWindow() { return this->ScalarRange[1] - this->ScalarRange[0]; }
  void SetWindow(double window)
  {
    double level = this->GetLevel();
    this->SetScalarRange(level - 0.5 * window, level + 0.5 * window);
  }
 
  double GetLevel() { return 0.5 * (this->ScalarRange[0] + this->ScalarRange[1]); }
  void SetLevel(double level)
  {
    double window = this->GetWindow();
    this->SetScalarRange(level - 0.5 * window, level + 0.5 * window);
  }



  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;
 
  using vtkMapper::MapScalars;
  vtkUnsignedCharArray* MapScalars(vtkDataSet* input, double alpha, int& cellFlag) override;

  bool HasTranslucentPolygonalGeometry() override;

  void CreateDefaultLookupTable() 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;
 
  void CreateTexture(vtkTextureObject*, vtkOpenGLRenderWindow*);
  void ReplaceActiveFBO(vtkRenderer*);
  void RestoreActiveFBO(vtkRenderer*);
 
  vtkNew<vtkOpenGLFramebufferObject> FBO;
 
  vtkNew<vtkTextureObject> PositionsTextureObject;
  vtkNew<vtkTextureObject> NormalsTextureObject;
  vtkSmartPointer<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;
 
  // 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