vtkOpenGLTextureComputeShaderNormalization.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 vtkOpenGLTextureComputeShaderNormalization_h
#define vtkOpenGLTextureComputeShaderNormalization_h

#include "vtkOpenGLTextureNormalizationHelper.h"
#include "vtkRenderingOpenGL2Module.h"
#include "vtkWrappingHints.h" // For VTK_MARSHALAUTO
#include "vtk_glad.h"

VTK_ABI_NAMESPACE_BEGIN
class vtkOpenGLRenderWindow;

#ifdef GL_COMPUTE_SHADER

class VTKRENDERINGOPENGL2_EXPORT VTK_MARSHALAUTO vtkOpenGLTextureComputeShaderNormalization
  : public vtkOpenGLTextureNormalizationHelper
{
public:
  static vtkOpenGLTextureComputeShaderNormalization* New();
  vtkTypeMacro(vtkOpenGLTextureComputeShaderNormalization, vtkOpenGLTextureNormalizationHelper);
  void PrintSelf(ostream& os, vtkIndent indent) override;

  void Initialize(vtkOpenGLRenderWindow* context);

  bool ConvertUShortToFloat(const void* sourceData, size_t numValues, int numComps,
    unsigned int targetTexture, unsigned int width, unsigned int height) override;

  bool ConvertShortToFloat(const void* sourceData, size_t numValues, int numComps,
    unsigned int targetTexture, unsigned int width, unsigned int height) override;

protected:
  vtkOpenGLTextureComputeShaderNormalization();
  ~vtkOpenGLTextureComputeShaderNormalization() override;

  vtkOpenGLRenderWindow* Context = nullptr;
  GLuint ComputeProgram = 0;
  GLuint IntermediateStorageTexture = 0;

  bool InitializeComputeProgram();
  bool UploadToIntermediateTexture(const void* sourceData, size_t numValues, int numComps,
    unsigned int width, unsigned int height, GLenum sourceFormat);
  bool RunNormalizationCompute(
    int numComps, unsigned int targetTexture, unsigned int width, unsigned int height);

private:
  vtkOpenGLTextureComputeShaderNormalization(
    const vtkOpenGLTextureComputeShaderNormalization&) = delete;
  void operator=(const vtkOpenGLTextureComputeShaderNormalization&) = delete;
};

#endif // GL_COMPUTE_SHADER

VTK_ABI_NAMESPACE_END
#endif // vtkOpenGLTextureComputeShaderNormalization_h