vtkAxisAlignedTransformFilter.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 vtkAxisAlignedTransformFilter_h
#define vtkAxisAlignedTransformFilter_h
 
#include "vtkDataObjectAlgorithm.h"
 
#include "vtkFiltersGeneralModule.h" // for export macro
 
VTK_ABI_NAMESPACE_BEGIN
class vtkHyperTree;
class vtkHyperTreeGrid;
class vtkHyperTreeGridNonOrientedCursor;
class vtkTransform;
class vtkImageData;
class vtkRectilinearGrid;
 
class VTKFILTERSGENERAL_EXPORT vtkAxisAlignedTransformFilter : public vtkDataObjectAlgorithm
{
public:
  vtkTypeMacro(vtkAxisAlignedTransformFilter, vtkDataObjectAlgorithm);
  void PrintSelf(ostream& os, vtkIndent indent) override;
  static vtkAxisAlignedTransformFilter* New();
 
  enum Axis : int
  {
    X = 0,
    Y = 1,
    Z = 2
  };
 
  enum Angle : int
  {
    ROT0 = 0,
    ROT90 = 1,
    ROT180 = 2,
    ROT270 = 3
  };
 
 
  vtkSetVector3Macro(Translation, double);
  vtkGetVectorMacro(Translation, double, 3);
 
 
  vtkSetVector3Macro(Scale, double);
  vtkGetVectorMacro(Scale, double, 3);
 
 
  vtkSetClampMacro(RotationAngle, int, ROT0, ROT270);
  vtkGetMacro(RotationAngle, int);
 
 
  vtkSetClampMacro(RotationAxis, int, X, Z);
  vtkGetMacro(RotationAxis, unsigned int);
 
protected:
  vtkAxisAlignedTransformFilter() = default;
  ~vtkAxisAlignedTransformFilter() override = default;
 
  vtkTypeBool ProcessRequest(vtkInformation* request, vtkInformationVector** inputVector,
    vtkInformationVector* outputVector) override;
 
  int RequestDataObject(vtkInformation* request, vtkInformationVector** inputVector,
    vtkInformationVector* outputVector) override;
 
  int RequestData(vtkInformation* request, vtkInformationVector** inputVector,
    vtkInformationVector* outputVector) override;
  int FillInputPortInformation(int port, vtkInformation* info) override;
 
private:
  vtkAxisAlignedTransformFilter(const vtkAxisAlignedTransformFilter&) = delete;
  void operator=(const vtkAxisAlignedTransformFilter&) = delete;
 
 
  bool ProcessGeneric(vtkDataObject* inputDataObject, vtkDataObject* outputDataObject);
  bool ProcessHTG(vtkHyperTreeGrid* inputHTG, vtkHyperTreeGrid* outputHTG, int R[3][3]);
  bool ProcessImageData(vtkImageData* inputID, vtkImageData* outputID, int R[3][3]);
  bool ProcessRectilinearGrid(
    vtkRectilinearGrid* inputRG, vtkRectilinearGrid* outputRG, int R[3][3]);
 
  bool Dispatch(vtkDataObject* inputDataObject, vtkDataObject* outputDataObject);
 
  void GetRotationMatrix(int axis, int rotation, int R[3][3]);
 
  int GetRotatedId(int id, int R[3][3], int newDims[3], int dims[3], int Tvec[3], bool transposed);
 
  void GetTransform(vtkTransform* transform);
 
  void ComputeCellScale(vtkDataArray* xCoords, vtkDataArray* yCoords, vtkDataArray* zCoords,
    int dims[3], double scales[3]);
 
  void ApplyCellScale(vtkHyperTreeGridNonOrientedCursor* cursor, double scales[3]);
 
  vtkAxisAlignedTransformFilter::Axis FindNormalAxis(int dims[3]);
 
  std::vector<unsigned int> ComputePermutation(
    unsigned int branchFactor, int axis, int normalAxis, int rotation, int dimension);
 
  void CopyAndRotate(vtkHyperTree* input, vtkHyperTree* output, vtkIdType inputIndex,
    vtkIdType outputIndex, const std::vector<unsigned int>& permutation, unsigned int depth,
    vtkHyperTreeGridNonOrientedCursor* cursor);
 
  vtkHyperTree* CreateNewRotatedHyperTree(
    vtkHyperTreeGrid* htg, vtkHyperTree* dest, const std::vector<unsigned int>& permutation);
 
  // Transformation parameters
  double Translation[3] = { 0.0, 0.0, 0.0 };
  double Scale[3] = { 1.0, 1.0, 1.0 };
  int RotationAngle = Angle::ROT0;
  int RotationAxis = Axis::X;
};
 
VTK_ABI_NAMESPACE_END
#endif