vtkAxisAlignedReflectionFilter.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 vtkAxisAlignedReflectionFilter_h
#define vtkAxisAlignedReflectionFilter_h
 
#include "vtkCompositeDataSetAlgorithm.h"
#include "vtkFiltersGeneralModule.h" // for export macro
#include "vtkPlane.h"                // for vtkPlane class
#include "vtkSmartPointer.h"         // for smart pointer
 
VTK_ABI_NAMESPACE_BEGIN
class vtkDataSet;
class vtkDataSetAttributes;
class vtkExplicitStructuredGrid;
class vtkHyperTreeGrid;
class vtkImageData;
class vtkPartitionedDataSetCollection;
class vtkPolyData;
class vtkRectilinearGrid;
class vtkStructuredGrid;
 
class VTKFILTERSGENERAL_EXPORT vtkAxisAlignedReflectionFilter : public vtkCompositeDataSetAlgorithm
{
public:
  static vtkAxisAlignedReflectionFilter* New();
 
  vtkTypeMacro(vtkAxisAlignedReflectionFilter, vtkCompositeDataSetAlgorithm);
  void PrintSelf(ostream& os, vtkIndent indent) override;
 
  enum PlaneAxis
  {
    X_PLANE = 0,
    Y_PLANE = 1,
    Z_PLANE = 2
  };
 
  enum PlaneModes
  {
    PLANE = 0,
    X_MIN = 1,
    Y_MIN = 2,
    Z_MIN = 3,
    X_MAX = 4,
    Y_MAX = 5,
    Z_MAX = 6,
  };
 
 
  vtkSetSmartPointerMacro(ReflectionPlane, vtkPlane);
  vtkGetSmartPointerMacro(ReflectionPlane, vtkPlane);
 
 
  vtkSetClampMacro(PlaneMode, int, 0, 6);
  vtkGetMacro(PlaneMode, int);
  void SetPlaneModeToPlane() { this->SetPlaneMode(PLANE); }
  void SetPlaneModeToXMin() { this->SetPlaneMode(X_MIN); }
  void SetPlaneModeToYMin() { this->SetPlaneMode(Y_MIN); }
  void SetPlaneModeToZMin() { this->SetPlaneMode(Z_MIN); }
  void SetPlaneModeToXMax() { this->SetPlaneMode(X_MAX); }
  void SetPlaneModeToYMax() { this->SetPlaneMode(Y_MAX); }
  void SetPlaneModeToZMax() { this->SetPlaneMode(Z_MAX); }
 
 
  vtkSetMacro(CopyInput, bool);
  vtkGetMacro(CopyInput, bool);
  vtkBooleanMacro(CopyInput, bool);
 
 
  vtkSetMacro(ReflectAllInputArrays, bool);
  vtkGetMacro(ReflectAllInputArrays, bool);
  vtkBooleanMacro(ReflectAllInputArrays, bool);
 
  vtkMTimeType GetMTime() override;
 
protected:
  vtkAxisAlignedReflectionFilter() = default;
  ~vtkAxisAlignedReflectionFilter() override = default;
 
  int RequestDataObject(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;
  int RequestData(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;
  int FillInputPortInformation(int port, vtkInformation* info) override;
 
  virtual void ComputeBounds(vtkDataObject* input, double bounds[6]);
 
private:
  vtkAxisAlignedReflectionFilter(const vtkAxisAlignedReflectionFilter&) = delete;
  void operator=(const vtkAxisAlignedReflectionFilter&) = delete;
 
  void FindAndReflectArrays(vtkDataSet* input, vtkDataSet* output, int mirrorDir[3],
    int mirrorSymmetricTensorDir[6], int mirrorTensorDir[9]);
 
  void AddPartitionedDataSet(vtkPartitionedDataSetCollection* outputPDSC, vtkDataObject* dObj,
    vtkInformation* inputMetadata, int nodeId, bool isParentMultiblock, bool isInputCopy);
 
  bool ProcessComposite(vtkPartitionedDataSetCollection* outputPDSC, vtkCompositeDataSet* inputCD,
    double bounds[6], int inputNodeId, int reflectionNodeId);
  bool ProcessLeaf(
    vtkDataObject* inputDataObject, vtkDataObject* outputDataObject, double bounds[6]);
 
 
  void ProcessExplicitStructuredGrid(vtkExplicitStructuredGrid* input,
    vtkExplicitStructuredGrid* output, double constant[3], int mirrorDir[3],
    int mirrorSymmetricTensorDir[6], int mirrorTensorDir[9]);
  void ProcessHtg(vtkHyperTreeGrid* input, vtkHyperTreeGrid* output, int mirrorDir[3],
    int mirrorSymmetricTensorDir[6], int mirrorTensorDir[9]);
  void ProcessImageData(vtkImageData* input, vtkImageData* output, double constant[3],
    int mirrorDir[3], int mirrorSymmetricTensorDir[6], int mirrorTensorDir[9]);
  void ProcessPolyData(vtkPolyData* input, vtkPolyData* output, double constant[3],
    int mirrorDir[3], int mirrorSymmetricTensorDir[6], int mirrorTensorDir[9]);
  void ProcessRectilinearGrid(vtkRectilinearGrid* input, vtkRectilinearGrid* output,
    double constant[3], int mirrorDir[3], int mirrorSymmetricTensorDir[6], int mirrorTensorDir[9]);
  void ProcessStructuredGrid(vtkStructuredGrid* input, vtkStructuredGrid* output,
    double constant[3], int mirrorDir[3], int mirrorSymmetricTensorDir[6], int mirrorTensorDir[9]);
 
  bool CopyInput = true;
  bool ReflectAllInputArrays = false;
  int PlaneMode = PLANE;
  vtkSmartPointer<vtkPlane> ReflectionPlane;
 
  PlaneAxis PlaneAxisInternal = X_PLANE;
  double PlaneOriginInternal[3] = { 0.0, 0.0, 0.0 };
 
  // For naming purposes
  int InputCount = 0;
  int ReflectionCount = 0;
 
  int PartitionIndex = 0;
};
 
VTK_ABI_NAMESPACE_END
#endif