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);

  vtkSmartPointer<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