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
|
/*=========================================================================
Program: Visualization Toolkit
Module: TestBiQuadraticQuad.cxx
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.
=========================================================================*/
#include "vtkNew.h"
#include "vtkBiQuadraticQuad.h"
#include "vtkPointData.h"
#include "vtkCellArray.h"
#include "vtkDoubleArray.h"
#include "vtkProbeFilter.h"
#include "vtkUnstructuredGrid.h"
#include "vtkPolyData.h"
#include "vtkMathUtilities.h"
//----------------------------------------------------------------------------
int TestBiQuadraticQuad(int, char*[])
{
vtkNew<vtkPoints> points;
points->InsertNextPoint(0.0, 0.0, 0.0);
points->InsertNextPoint(1.0, 0.0, 0.0);
points->InsertNextPoint(1.0, 1.0, 0.0);
points->InsertNextPoint(0.0, 1.0, 0.0);
points->InsertNextPoint(0.5, 0.0, 0.0);
points->InsertNextPoint(1.0, 0.5, 0.0);
points->InsertNextPoint(0.5, 1.0, 0.0);
points->InsertNextPoint(0.0, 0.5, 0.0);
points->InsertNextPoint(0.5, 0.5, 0.0);
vtkNew<vtkBiQuadraticQuad> quad;
for (int i = 0; i < 9; ++i)
{
quad->GetPointIds()->SetId(i, i);
}
vtkNew<vtkCellArray> cellArray;
cellArray->InsertNextCell(quad.Get());
vtkNew<vtkDoubleArray> uArray;
uArray->SetName("u");
uArray->SetNumberOfComponents(1);
uArray->SetNumberOfTuples(9);
// set u(x, y) = x
for (int i=0; i<9; i++)
{
uArray->SetValue(i, points->GetPoint(i)[0]);
}
vtkNew<vtkUnstructuredGrid> grid;
grid->SetPoints(points.Get());
grid->SetCells(VTK_BIQUADRATIC_QUAD, cellArray.Get());
grid->GetPointData()->SetScalars(uArray.Get());
double probeX = 2.0 / 3.0;
double probeY = 0.25;
vtkNew<vtkPoints> probePoints;
probePoints->InsertNextPoint(probeX, probeY, 0.0);
vtkNew<vtkPolyData> probePolyData;
probePolyData->SetPoints(probePoints.Get());
vtkNew<vtkProbeFilter> prober;
prober->SetSourceData(grid.Get());
prober->SetInputData(probePolyData.Get());
prober->Update();
vtkDataArray* data = prober->GetOutput()->GetPointData()->GetScalars();
vtkDoubleArray* doubleData = vtkDoubleArray::SafeDownCast(data);
double interpolated(0.0);
if (doubleData)
{
interpolated = doubleData->GetComponent(0, 0);
}
else
{
cout << "Failed to downcast prober scalars." << endl;
}
if (!vtkMathUtilities::FuzzyCompare(interpolated, probeX, 1.0e-6))
{
cout << "Interpolated value of " << interpolated << " with probe value "
<< probeX << " difference of " << (interpolated - probeX) << endl;
return EXIT_FAILURE;
}
return EXIT_SUCCESS;
}
|
