vtkParametricFunctionSource.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 vtkParametricFunctionSource_h
#define vtkParametricFunctionSource_h

#include "vtkFiltersSourcesModule.h" // For export macro
#include "vtkPolyDataAlgorithm.h"
#include "vtkWrappingHints.h" // For VTK_MARSHALAUTO

VTK_ABI_NAMESPACE_BEGIN
class vtkCellArray;
class vtkParametricFunction;

class VTKFILTERSSOURCES_EXPORT VTK_MARSHALAUTO vtkParametricFunctionSource
  : public vtkPolyDataAlgorithm
{
public:
  vtkTypeMacro(vtkParametricFunctionSource, vtkPolyDataAlgorithm);
  void PrintSelf(ostream& os, vtkIndent indent) override;

  static vtkParametricFunctionSource* New();


  virtual void SetParametricFunction(vtkParametricFunction*);
  vtkGetObjectMacro(ParametricFunction, vtkParametricFunction);


  vtkSetClampMacro(UResolution, int, 2, VTK_INT_MAX);
  vtkGetMacro(UResolution, int);


  vtkSetClampMacro(VResolution, int, 2, VTK_INT_MAX);
  vtkGetMacro(VResolution, int);


  vtkSetClampMacro(WResolution, int, 2, VTK_INT_MAX);
  vtkGetMacro(WResolution, int);


  vtkBooleanMacro(GenerateTextureCoordinates, vtkTypeBool);
  vtkSetClampMacro(GenerateTextureCoordinates, vtkTypeBool, 0, 1);
  vtkGetMacro(GenerateTextureCoordinates, vtkTypeBool);


  vtkBooleanMacro(GenerateNormals, vtkTypeBool);
  vtkSetClampMacro(GenerateNormals, vtkTypeBool, 0, 1);
  vtkGetMacro(GenerateNormals, vtkTypeBool);

  enum SCALAR_MODE
  {
    SCALAR_NONE = 0,
    SCALAR_U,
    SCALAR_V,
    SCALAR_U0,
    SCALAR_V0,
    SCALAR_U0V0,
    SCALAR_MODULUS,
    SCALAR_PHASE,
    SCALAR_QUADRANT,
    SCALAR_X,
    SCALAR_Y,
    SCALAR_Z,
    SCALAR_DISTANCE,
    SCALAR_FUNCTION_DEFINED
  };


  vtkSetClampMacro(ScalarMode, int, SCALAR_NONE, SCALAR_FUNCTION_DEFINED);
  vtkGetMacro(ScalarMode, int);
  void SetScalarModeToNone() { this->SetScalarMode(SCALAR_NONE); }
  void SetScalarModeToU() { this->SetScalarMode(SCALAR_U); }
  void SetScalarModeToV() { this->SetScalarMode(SCALAR_V); }
  void SetScalarModeToU0() { this->SetScalarMode(SCALAR_U0); }
  void SetScalarModeToV0() { this->SetScalarMode(SCALAR_V0); }
  void SetScalarModeToU0V0() { this->SetScalarMode(SCALAR_U0V0); }
  void SetScalarModeToModulus() { this->SetScalarMode(SCALAR_MODULUS); }
  void SetScalarModeToPhase() { this->SetScalarMode(SCALAR_PHASE); }
  void SetScalarModeToQuadrant() { this->SetScalarMode(SCALAR_QUADRANT); }
  void SetScalarModeToX() { this->SetScalarMode(SCALAR_X); }
  void SetScalarModeToY() { this->SetScalarMode(SCALAR_Y); }
  void SetScalarModeToZ() { this->SetScalarMode(SCALAR_Z); }
  void SetScalarModeToDistance() { this->SetScalarMode(SCALAR_DISTANCE); }
  void SetScalarModeToFunctionDefined() { this->SetScalarMode(SCALAR_FUNCTION_DEFINED); }

  vtkMTimeType GetMTime() override;


  vtkSetMacro(OutputPointsPrecision, int);
  vtkGetMacro(OutputPointsPrecision, int);

protected:
  vtkParametricFunctionSource();
  ~vtkParametricFunctionSource() override;

  // Usual data generation method
  int RequestData(
    vtkInformation* info, vtkInformationVector** input, vtkInformationVector* output) override;

  // Variables
  vtkParametricFunction* ParametricFunction;

  int UResolution;
  int VResolution;
  int WResolution;
  vtkTypeBool GenerateTextureCoordinates;
  vtkTypeBool GenerateNormals;
  int ScalarMode;
  int OutputPointsPrecision;

private:
  // Create output depending on function dimension
  void Produce1DOutput(vtkInformationVector* output);
  void Produce2DOutput(vtkInformationVector* output);

  void MakeTriangles(vtkCellArray* strips, int PtsU, int PtsV);

  vtkParametricFunctionSource(const vtkParametricFunctionSource&) = delete;
  void operator=(const vtkParametricFunctionSource&) = delete;
};

VTK_ABI_NAMESPACE_END
#endif