1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195
|
/*=========================================================================
Program: Visualization Toolkit
Module: vtkPeriodicFiler.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.
=========================================================================*/
/**
* @class vtkAngularPeriodicFilter
* @brief A filter to produce mapped angular periodic
* multiblock dataset from a single block, by rotation.
*
*
* Generate angular periodic dataset by rotating points, vectors and tensors
* data arrays from an original data array.
* The generated dataset is of the same type than the input (float or double).
* To compute the rotation this filter needs
* i) a number of periods, wich can be the maximum, i.e. a full period,
* ii) an angle, wich can be fetched from a field data array in radian or directly
* in degrees; iii) the axis (X, Y or Z) and the center of rotation.
* Point coordinates are transformed, as well as all vectors (3-components) and
* tensors (9 components) in points and cell data arrays.
* The generated multiblock will have the same tree architecture than the input,
* except transformed leaves are replaced by a vtkMultipieceDataSet.
* Supported input leaf dataset type are: vtkPolyData, vtkStructuredGrid
* and vtkUnstructuredGrid. Other data objects are rotated using the
* transform filter (at a high cost!).
*/
#ifndef vtkAngularPeriodicFilter_h
#define vtkAngularPeriodicFilter_h
#include "vtkFiltersParallelModule.h" // For export macro
#include "vtkPeriodicFilter.h"
class vtkDataSetAttributes;
class vtkMultiPieceDataSet;
class vtkPointSet;
#define VTK_ROTATION_MODE_DIRECT_ANGLE 0 // Use user-provided angle
#define VTK_ROTATION_MODE_ARRAY_VALUE 1 // Use array from input data as angle
class VTKFILTERSPARALLEL_EXPORT vtkAngularPeriodicFilter : public vtkPeriodicFilter
{
public:
static vtkAngularPeriodicFilter* New();
vtkTypeMacro(vtkAngularPeriodicFilter, vtkPeriodicFilter);
void PrintSelf(ostream& os, vtkIndent indent);
//@{
/**
* Set/Get whether the rotated array values should be computed
* on-the-fly (default), which is compute-intensive, or the arrays should be
* explicitly generated and stored, at the cost of using more memory.
*/
vtkSetMacro(ComputeRotationsOnTheFly, bool);
vtkGetMacro(ComputeRotationsOnTheFly, bool);
vtkBooleanMacro(ComputeRotationsOnTheFly, bool);
//@}
//@{
/**
* Set/Get The rotation mode.
* VTK_ROTATION_MODE_DIRECT_ANGLE to specifiy a angle value (default),
* VTK_ROTATION_MODE_ARRAY_VALUE to use value from an array in the input dataset.
*/
vtkSetClampMacro(RotationMode, int,
VTK_ROTATION_MODE_DIRECT_ANGLE, VTK_ROTATION_MODE_ARRAY_VALUE);
vtkGetMacro(RotationMode, int);
void SetRotationModeToDirectAngle()
{ this->SetRotationMode(VTK_ROTATION_MODE_DIRECT_ANGLE); }
void SetRotationModeToArrayValue()
{ this->SetRotationMode(VTK_ROTATION_MODE_ARRAY_VALUE); }
//@}
//@{
/**
* Set/Get Rotation angle, in degrees.
* Used only with VTK_ROTATION_MODE_DIRECT_ANGLE.
* Default is 180.
*/
vtkSetMacro(RotationAngle, double);
vtkGetMacro(RotationAngle, double);
//@}
//@{
/**
* Set/Get Name of array to get the angle from.
* Used only with VTK_ROTATION_MODE_ARRAY_VALUE.
*/
vtkSetStringMacro(RotationArrayName);
vtkGetStringMacro(RotationArrayName);
//@}
//@{
/**
* Set/Get Rotation Axis, 0 for X, 1 for Y, 2 for Z
*/
vtkSetClampMacro(RotationAxis, int, 0, 2);
vtkGetMacro(RotationAxis, int);
void SetRotationAxisToX();
void SetRotationAxisToY();
void SetRotationAxisToZ();
//@}
//@{
/**
* Set/Get Rotation Center
*/
vtkSetVector3Macro(Center, double);
vtkGetVector3Macro(Center, double);
//@}
protected:
vtkAngularPeriodicFilter();
~vtkAngularPeriodicFilter();
virtual int RequestData(vtkInformation *,
vtkInformationVector **,
vtkInformationVector *);
/**
* Create a transform copy of the provided data array
*/
vtkDataArray* TransformDataArray(vtkDataArray* inputArray,
double angle,
bool useCenter = true,
bool normalize = false);
/**
* Append a periodic piece to dataset, by computing rotated mesh and data
*/
void AppendPeriodicPiece(double angle, vtkIdType iPiece,
vtkDataObject* inputNode,
vtkMultiPieceDataSet* multiPiece);
/**
* Manually set the number of period on a specific leaf
*/
virtual void SetPeriodNumber(vtkCompositeDataIterator* loc,
vtkCompositeDataSet* output,
int nbPeriod);
/**
* Compute periodic pointset, rotating point, using provided angle
*/
void ComputePeriodicMesh(vtkPointSet* dataset, vtkPointSet* rotatedDataset,
double angle);
/**
* Compute periodic point/cell data, using provided angle
*/
void ComputeAngularPeriodicData(vtkDataSetAttributes* data,
vtkDataSetAttributes* rotatedData,
double angle);
/**
* Create a periodic data, leaf of the tree
*/
void CreatePeriodicDataSet(vtkCompositeDataIterator* loc,
vtkCompositeDataSet* output,
vtkCompositeDataSet* input);
/**
* Generate a name for a piece in the periodic dataset from the input dataset
*/
virtual void GeneratePieceName(vtkCompositeDataSet* input,
vtkCompositeDataIterator* inputLoc,
vtkMultiPieceDataSet* output,
vtkIdType outputId);
private:
vtkAngularPeriodicFilter(const vtkAngularPeriodicFilter&) VTK_DELETE_FUNCTION;
void operator=(const vtkAngularPeriodicFilter&) VTK_DELETE_FUNCTION;
bool ComputeRotationsOnTheFly;
int RotationMode;
char* RotationArrayName; // user-provided array name to use as angle, for ROTATION_MODE_ARRAY_VALUE
// Transform parameters
double RotationAngle;
int RotationAxis; // Axis to rotate around, 0 for X, 1 for Y, 2 for Z
double Center[3]; // Center of rotation
};
#endif
|