vtkParametricFunctionSource.h

Go to the documentation of this file.

/*=========================================================================

  Program:   Visualization Toolkit
  Module:    vtkParametricFunctionSource.h

  Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
  All rights reserved.
  See Copyright.txt or http://www.kitware.com/Copyright.htm for details.

     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notice for more information.

=========================================================================*/
#ifndef vtkParametricFunctionSource_h
#define vtkParametricFunctionSource_h

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

class vtkCellArray;
class vtkParametricFunction;

class VTKFILTERSSOURCES_EXPORT 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(void) {this->SetScalarMode(SCALAR_NONE);}
    void SetScalarModeToU(void) {this->SetScalarMode(SCALAR_U);}
    void SetScalarModeToV(void) {this->SetScalarMode(SCALAR_V);}
    void SetScalarModeToU0(void) {this->SetScalarMode(SCALAR_U0);}
    void SetScalarModeToV0(void) {this->SetScalarMode(SCALAR_V0);}
    void SetScalarModeToU0V0(void) {this->SetScalarMode(SCALAR_U0V0);}
    void SetScalarModeToModulus(void) {this->SetScalarMode(SCALAR_MODULUS);}
    void SetScalarModeToPhase(void) {this->SetScalarMode(SCALAR_PHASE);}
    void SetScalarModeToQuadrant(void) {this->SetScalarMode(SCALAR_QUADRANT);}
    void SetScalarModeToX(void) {this->SetScalarMode(SCALAR_X);}
    void SetScalarModeToY(void) {this->SetScalarMode(SCALAR_Y);}
    void SetScalarModeToZ(void) {this->SetScalarMode(SCALAR_Z);}
    void SetScalarModeToDistance(void) {this->SetScalarMode(SCALAR_DISTANCE);}
    void SetScalarModeToFunctionDefined(void)
    {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;

};

#endif