File: vtkIntegrationGaussianStrategy.h

package info (click to toggle)
vtk9 9.5.2%2Bdfsg3-4
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 205,916 kB
  • sloc: cpp: 2,336,565; ansic: 327,116; python: 111,200; yacc: 4,104; java: 3,977; sh: 3,032; xml: 2,771; perl: 2,189; lex: 1,787; makefile: 178; javascript: 165; objc: 153; tcl: 59
file content (214 lines) | stat: -rw-r--r-- 10,098 bytes parent folder | download | duplicates (5)
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
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
/**
 * @class   vtkIntegrationGaussianStrategy
 * @brief   Gaussian Quadrature Integration Strategy
 *
 * This class inherits from `vtkIntegrationStrategy` and override
 * several function for specific cell types to use the Gaussian Quadrature
 * rule which enables the correct computation of higher order cells and non
 * simplicial shapes. Note that not all cells are properly handled by the
 * strategy, and the computation falls back to the linear strategy for those.
 * Reference to the quadrature method:
 *  https://en.wikipedia.org/wiki/Gaussian_quadrature
 * Reference for quadrature weights computation :
 *  https://www.mm.bme.hu/~gyebro/files/ans_help_v182/ans_thry/thy_et1.html
 * Note that we specifically implement the Gauss-Legendre quadrature
 */

#ifndef vtkIntegrationGaussianStrategy_h
#define vtkIntegrationGaussianStrategy_h

#include "vtkCellType.h"              // For VTKCellType
#include "vtkDataSet.h"               // For Intermediate
#include "vtkFiltersParallelModule.h" // For export macro
#include "vtkIntegrationStrategy.h"
#include "vtkSmartPointer.h" // For Intermediate

VTK_ABI_NAMESPACE_BEGIN
class vtkDataSet;
class vtkDataSetAttributes;
class vtkIdTypeArray;
class vtkInformationQuadratureSchemeDefinitionVectorKey;
class vtkIntegrateAttributesFieldList;
class vtkIntegrationLinearStrategy;
class vtkQuadratureSchemeDefinition;
class vtkUnstructuredGrid;

class VTKFILTERSPARALLEL_EXPORT vtkIntegrationGaussianStrategy : public vtkIntegrationStrategy
{
public:
  static vtkIntegrationGaussianStrategy* New();
  void PrintSelf(ostream& os, vtkIndent indent) override;
  vtkTypeMacro(vtkIntegrationGaussianStrategy, vtkIntegrationStrategy);

  ///@{
  /**
   * Integration functions operating on some cell types. The Gaussian strategy is fully implemented.
   */
  void IntegrateQuad(vtkDataSet* input, vtkUnstructuredGrid* output, vtkIdType cellId,
    vtkIdType pt1Id, vtkIdType pt2Id, vtkIdType pt3Id, vtkIdType pt4Id, double& sum,
    double sumCenter[3], vtkIntegrateAttributesFieldList& cellFieldList,
    vtkIntegrateAttributesFieldList& pointFieldList, int index) override;

  void IntegrateHexahedron(vtkDataSet* input, vtkUnstructuredGrid* output, vtkGenericCell* cell,
    vtkIdType cellId, vtkIdType numPts, const vtkIdType* cellPtIds, vtkIdList* cellPtIdsList,
    double& sum, double sumCenter[3], vtkIntegrateAttributesFieldList& cellFieldList,
    vtkIntegrateAttributesFieldList& pointFieldList, int index) override;

  void IntegrateWedge(vtkDataSet* input, vtkUnstructuredGrid* output, vtkGenericCell* cell,
    vtkIdType cellId, vtkIdType numPts, const vtkIdType* cellPtIds, vtkIdList* cellPtIdsList,
    double& sum, double sumCenter[3], vtkIntegrateAttributesFieldList& cellFieldList,
    vtkIntegrateAttributesFieldList& pointFieldList, int index) override;

  void IntegratePyramid(vtkDataSet* input, vtkUnstructuredGrid* output, vtkGenericCell* cell,
    vtkIdType cellId, vtkIdType numPts, const vtkIdType* cellPtIds, vtkIdList* cellPtIdsList,
    double& sum, double sumCenter[3], vtkIntegrateAttributesFieldList& cellFieldList,
    vtkIntegrateAttributesFieldList& pointFieldList, int index) override;
  ///@}

  ///@{
  /**
   * Integration functions operating on some cell types. The Gaussian strategy falls back to linear
   * strategy but is strictly equivalent.
   */
  void IntegratePolyLine(vtkDataSet* input, vtkUnstructuredGrid* output, vtkIdType cellId,
    vtkIdType numPts, const vtkIdType* cellPtIds, double& sum, double sumCenter[3],
    vtkIntegrateAttributesFieldList& cellFieldList, vtkIntegrateAttributesFieldList& pointFieldList,
    int index) override;

  void IntegrateTriangleStrip(vtkDataSet* input, vtkUnstructuredGrid* output, vtkIdType cellId,
    vtkIdType numPts, const vtkIdType* cellPtIds, double& sum, double sumCenter[3],
    vtkIntegrateAttributesFieldList& cellFieldList, vtkIntegrateAttributesFieldList& pointFieldList,
    int index) override;

  void IntegrateTriangle(vtkDataSet* input, vtkUnstructuredGrid* output, vtkIdType cellId,
    vtkIdType pt1Id, vtkIdType pt2Id, vtkIdType pt3Id, double& sum, double sumCenter[3],
    vtkIntegrateAttributesFieldList& cellFieldList, vtkIntegrateAttributesFieldList& pointFieldList,
    int index) override;

  void IntegrateTetrahedron(vtkDataSet* input, vtkUnstructuredGrid* output, vtkIdType cellId,
    vtkIdType pt1Id, vtkIdType pt2Id, vtkIdType pt3Id, vtkIdType pt4Id, double& sum,
    double sumCenter[3], vtkIntegrateAttributesFieldList& cellFieldList,
    vtkIntegrateAttributesFieldList& pointFieldList, int index) override;

  void IntegratePixel(vtkDataSet* input, vtkUnstructuredGrid* output, vtkIdType cellId,
    vtkIdType numPts, const vtkIdType* cellPtIds, double& sum, double sumCenter[3],
    vtkIntegrateAttributesFieldList& cellFieldList, vtkIntegrateAttributesFieldList& pointFieldList,
    int index) override;

  void IntegrateVoxel(vtkDataSet* input, vtkUnstructuredGrid* output, vtkIdType cellId,
    vtkIdType numPts, const vtkIdType* cellPtIds, double& sum, double sumCenter[3],
    vtkIntegrateAttributesFieldList& cellFieldList, vtkIntegrateAttributesFieldList& pointFieldList,
    int index) override;

  void IntegrateGeneral1DCell(vtkDataSet* input, vtkUnstructuredGrid* output, vtkIdType cellId,
    vtkIdType numPts, const vtkIdType* cellPtIds, double& sum, double sumCenter[3],
    vtkIntegrateAttributesFieldList& cellFieldList, vtkIntegrateAttributesFieldList& pointFieldList,
    int index) override;

  void IntegrateGeneral2DCell(vtkDataSet* input, vtkUnstructuredGrid* output, vtkIdType cellId,
    vtkIdType numPts, const vtkIdType* cellPtIds, double& sum, double sumCenter[3],
    vtkIntegrateAttributesFieldList& cellFieldList, vtkIntegrateAttributesFieldList& pointFieldList,
    int index) override;

  void IntegrateGeneral3DCell(vtkDataSet* input, vtkUnstructuredGrid* output, vtkIdType cellId,
    vtkIdType numPts, const vtkIdType* cellPtIds, double& sum, double sumCenter[3],
    vtkIntegrateAttributesFieldList& cellFieldList, vtkIntegrateAttributesFieldList& pointFieldList,
    int index) override;
  ///@}

  ///@{
  /**
   * Integration functions operating on some cell types. The Gaussian strategy is not implemented
   * and falls back to linear strategy.
   */
  void IntegratePolygon(vtkDataSet* input, vtkUnstructuredGrid* output, vtkIdType cellId,
    vtkIdType numPts, const vtkIdType* cellPtIds, double& sum, double sumCenter[3],
    vtkIntegrateAttributesFieldList& cellFieldList, vtkIntegrateAttributesFieldList& pointFieldList,
    int index) override;

  void IntegrateDefault(vtkDataSet* input, vtkUnstructuredGrid* output, vtkGenericCell* cell,
    vtkIdType cellId, vtkIdType numPts, vtkIdList* cellPtIds, double& sum, double sumCenter[3],
    vtkIntegrateAttributesFieldList& cellFieldList, vtkIntegrateAttributesFieldList& pointFieldList,
    int index) override;
  ///@}

  ///@{
  /**
   * Integrates on data arrays depending on the number of node of the shape.
   */
  void IntegrateData1(vtkDataSetAttributes* inda, vtkDataSetAttributes* outda, vtkIdType pt1Id,
    double k, vtkIntegrateAttributesFieldList& fieldlist, int fieldlist_index) override;

  void IntegrateData2(vtkDataSetAttributes* inda, vtkDataSetAttributes* outda, vtkIdType pt1Id,
    vtkIdType pt2Id, double k, vtkIntegrateAttributesFieldList& fieldlist,
    int fieldlist_index) override;

  void IntegrateData3(vtkDataSetAttributes* inda, vtkDataSetAttributes* outda, vtkIdType pt1Id,
    vtkIdType pt2Id, vtkIdType pt3Id, double k, vtkIntegrateAttributesFieldList& fieldlist,
    int fieldlist_index) override;

  void IntegrateData4(vtkDataSetAttributes* inda, vtkDataSetAttributes* outda, vtkIdType pt1Id,
    vtkIdType pt2Id, vtkIdType pt3Id, vtkIdType pt4Id, double k,
    vtkIntegrateAttributesFieldList& fieldlist, int fieldlist_index) override;
  ///@}

  /**
   * Setup necessary precomputations. In particular, interpolates data fields to quadratures points.
   */
  void Initialize(vtkDataSet* input) override;

protected:
  vtkIntegrationGaussianStrategy() = default;
  ~vtkIntegrationGaussianStrategy() override;

private:
  vtkIntegrationGaussianStrategy(const vtkIntegrationGaussianStrategy&) = delete;
  void operator=(const vtkIntegrationGaussianStrategy&) = delete;

  /**
   * Helper function to integrate using the Gaussian Quadrature on any cell
   */
  void IntegrateGaussian(vtkDataSet* input, vtkUnstructuredGrid* output, VTKCellType celltype,
    vtkIdType cellId, vtkIdType numPts, const vtkIdType* cellPtIds, vtkIdList* cellPtIdsList,
    double& sum, double sumCenter[3], vtkIntegrateAttributesFieldList& cellFieldList,
    vtkIntegrateAttributesFieldList& pointFieldList, int index);

  /**
   * Integrate over the point datas.
   */
  void IntegratePointDataGaussian(vtkDataSetAttributes* outda, vtkIdType cellId,
    int nQuadraturePoints, std::vector<double> partialArea,
    vtkIntegrateAttributesFieldList& fieldList, int index);

  ///@{
  /**
   * Helper functions for Jacobian determinant computation.
   */
  double ComputeJacobianDet2D(const double dN[8], const std::vector<std::array<double, 3>>& pts);
  double ComputeJacobianDet(
    const double* dN, const std::vector<std::array<double, 3>>& pts, int nNodes);
  ///@}

  ///@{
  /**
   * Helper function for the initialization.
   */
  void AddPointDataArray();
  void InitializeQuadratureOffsets();
  std::string GenerateUniqueArrayName(const std::string& baseName);
  vtkSmartPointer<vtkQuadratureSchemeDefinition> CreateQuadratureSchemeDefinition(int cellType);
  void InitializeQuadratureOffsetsArray(
    vtkInformation* info, vtkInformationQuadratureSchemeDefinitionVectorKey* key);
  void ComputeQuadratureInterpolation();
  ///@}

  vtkNew<vtkIntegrationLinearStrategy> LinearStrategy;
  vtkNew<vtkIdTypeArray> Offsets;
  vtkSmartPointer<vtkDataSet> Intermediate = nullptr;
  vtkQuadratureSchemeDefinition** CellDefinitionDictionnary = nullptr;
};

VTK_ABI_NAMESPACE_END
#endif