vtkFitToHeightMapFilter.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 vtkFitToHeightMapFilter_h
#define vtkFitToHeightMapFilter_h
 
#include "vtkFiltersModelingModule.h" // For export macro
#include "vtkPolyDataAlgorithm.h"
 
VTK_ABI_NAMESPACE_BEGIN
class vtkImageData;
 
class VTKFILTERSMODELING_EXPORT vtkFitToHeightMapFilter : public vtkPolyDataAlgorithm
{
public:
 
  static vtkFitToHeightMapFilter* New();
  vtkTypeMacro(vtkFitToHeightMapFilter, vtkPolyDataAlgorithm);
  void PrintSelf(ostream& os, vtkIndent indent) override;
 
  void SetHeightMapData(vtkImageData* idata);
 
 
  void SetHeightMapConnection(vtkAlgorithmOutput* algOutput);
 
 
  vtkImageData* GetHeightMap();
  vtkImageData* GetHeightMap(vtkInformationVector* sourceInfo);
 
  // Strategies to fit the polydata.
  enum FittingStrategy
  {
    POINT_PROJECTION = 0,
    POINT_MINIMUM_HEIGHT = 1,
    POINT_MAXIMUM_HEIGHT = 2,
    POINT_AVERAGE_HEIGHT = 3,
    CELL_MINIMUM_HEIGHT = 4,
    CELL_MAXIMUM_HEIGHT = 5,
    CELL_AVERAGE_HEIGHT = 6,
  };
 
 
  vtkSetMacro(FittingStrategy, int);
  vtkGetMacro(FittingStrategy, int);
  void SetFittingStrategyToPointProjection() { this->SetFittingStrategy(POINT_PROJECTION); }
  void SetFittingStrategyToPointMinimumHeight() { this->SetFittingStrategy(POINT_MINIMUM_HEIGHT); }
  void SetFittingStrategyToPointMaximumHeight() { this->SetFittingStrategy(POINT_MAXIMUM_HEIGHT); }
  void SetFittingStrategyToAverageHeight() { this->SetFittingStrategy(POINT_AVERAGE_HEIGHT); }
  void SetFittingStrategyToCellMinimumHeight() { this->SetFittingStrategy(CELL_MINIMUM_HEIGHT); }
  void SetFittingStrategyToCellMaximumHeight() { this->SetFittingStrategy(CELL_MAXIMUM_HEIGHT); }
  void SetFittingStrategyToCellAverageHeight() { this->SetFittingStrategy(CELL_AVERAGE_HEIGHT); }
 
 
  vtkSetMacro(UseHeightMapOffset, vtkTypeBool);
  vtkGetMacro(UseHeightMapOffset, vtkTypeBool);
  vtkBooleanMacro(UseHeightMapOffset, vtkTypeBool);
 
protected:
  vtkFitToHeightMapFilter();
  ~vtkFitToHeightMapFilter() override;
 
  int RequestData(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;
  int FillInputPortInformation(int, vtkInformation*) override;
 
  int FittingStrategy;
  vtkTypeBool UseHeightMapOffset;
  double Offset;
 
  void AdjustPoints(vtkPolyData* output, vtkIdType numCells, vtkPoints* newPts);
  void AdjustCells(
    vtkPolyData* output, vtkIdType numCells, double* cellHts, vtkPoints* inPts, vtkPoints* newPts);
 
private:
  vtkFitToHeightMapFilter(const vtkFitToHeightMapFilter&) = delete;
  void operator=(const vtkFitToHeightMapFilter&) = delete;
};
 
VTK_ABI_NAMESPACE_END
#endif