vtkShader.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 vtkShader_h
#define vtkShader_h

#include "vtkObject.h"
#include "vtkRenderingOpenGL2Module.h" // for export macro

#include <string> // For member variables.
#include <vector> // For member variables.

VTK_ABI_NAMESPACE_BEGIN
class VTKRENDERINGOPENGL2_EXPORT vtkShader : public vtkObject
{
public:
  static vtkShader* New();
  vtkTypeMacro(vtkShader, vtkObject);
  void PrintSelf(ostream& os, vtkIndent indent) override;

  enum Type
  {
    Vertex,         
    Fragment,       
    Geometry,       
    Compute,        
    TessControl,    
    TessEvaluation, 
    Unknown         
  };

  void SetType(Type type);

  Type GetType() const { return this->ShaderType; }

  void SetSource(const std::string& source);

  std::string GetSource() const { return this->Source; }

  std::string GetError() const { return this->Error; }

  int GetHandle() const { return this->Handle; }

  bool Compile();

  void Cleanup();

  static bool IsComputeShaderSupported();

  static bool IsTessellationShaderSupported();

  class ReplacementSpec
  {
  public:
    std::string OriginalValue;
    vtkShader::Type ShaderType;
    bool ReplaceFirst;
    bool operator<(const ReplacementSpec& v1) const
    {
      if (this->OriginalValue != v1.OriginalValue)
      {
        return this->OriginalValue < v1.OriginalValue;
      }
      if (this->ShaderType != v1.ShaderType)
      {
        return this->ShaderType < v1.ShaderType;
      }
      return (this->ReplaceFirst < v1.ReplaceFirst);
    }
    bool operator>(const ReplacementSpec& v1) const
    {
      if (this->OriginalValue != v1.OriginalValue)
      {
        return this->OriginalValue > v1.OriginalValue;
      }
      if (this->ShaderType != v1.ShaderType)
      {
        return this->ShaderType > v1.ShaderType;
      }
      return (this->ReplaceFirst > v1.ReplaceFirst);
    }
  };
  class ReplacementValue
  {
  public:
    std::string Replacement;
    bool ReplaceAll;
  };

protected:
  vtkShader();
  ~vtkShader() override;

  Type ShaderType;
  int Handle;
  bool Dirty;

  std::string Source;
  std::string Error;

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

VTK_ABI_NAMESPACE_END
#endif