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
|
#include <iostream>
#include "FEAdaptor.h"
#include <vtkAMRBox.h>
#include <vtkCompositeDataIterator.h>
#include <vtkCPDataDescription.h>
#include <vtkCPInputDataDescription.h>
#include <vtkCPProcessor.h>
#include <vtkCPPythonScriptPipeline.h>
#include <vtkNew.h>
#include <vtkOverlappingAMR.h>
#include <vtkUniformGrid.h>
namespace
{
vtkCPProcessor* Processor = NULL;
vtkOverlappingAMR* VTKGrid;
void BuildVTKGrid()
{
if(VTKGrid == NULL)
{
// The grid structure isn't changing so we only build it
// the first time it's needed. If we needed the memory
// we could delete it and rebuild as necessary.
VTKGrid = vtkOverlappingAMR::New();
BuildVTKGrid();
}
int numberOfLevels = 3;
int blocksPerLevel[3] = {1, 1, 1};
VTKGrid->Initialize(numberOfLevels, blocksPerLevel);
VTKGrid->SetGridDescription(VTK_XYZ_GRID);
double origin[] = {0,0,0};
double level0Spacing[] = {4, 4, 4};
double level1Spacing[] = {2, 2, 2};
double level2Spacing[] = {1, 1, 1};
VTKGrid->SetOrigin(origin);
int level0Dims[] = {25, 25, 25};
vtkAMRBox level0Box(origin, level0Dims, level0Spacing, origin, VTK_XYZ_GRID);
int level1Dims[] = {20, 20, 20};
vtkAMRBox level1Box(origin, level1Dims, level1Spacing, origin, VTK_XYZ_GRID);
int level2Dims[] = {10, 10, 10};
vtkAMRBox level2Box(origin, level2Dims, level2Spacing, origin, VTK_XYZ_GRID);
VTKGrid->SetSpacing(0, level0Spacing);
VTKGrid->SetAMRBox(0, 0, level0Box);
VTKGrid->SetSpacing(1, level1Spacing);
VTKGrid->SetAMRBox(1, 0, level1Box);
VTKGrid->SetSpacing(2, level2Spacing);
VTKGrid->SetAMRBox(2, 0, level2Box);
VTKGrid->GenerateParentChildInformation();
// the highest level grid
vtkNew<vtkUniformGrid> level0Grid;
level0Grid->SetSpacing(level0Spacing);
level0Grid->SetOrigin(0, 0, 0);
level0Grid->SetExtent(0, 25, 0, 25, 0, 25);
VTKGrid->SetDataSet(0, 0, level0Grid.GetPointer());
// the mid-level grid
vtkNew<vtkUniformGrid> level1Grid0;
level1Grid0->SetSpacing(level1Spacing);
level1Grid0->SetExtent(0, 20, 0, 20, 0, 20);
VTKGrid->SetDataSet(1, 0, level1Grid0.GetPointer());
// the lowest level grid
vtkNew<vtkUniformGrid> level2Grid;
level2Grid->SetSpacing(level2Spacing);
level2Grid->SetExtent(0, 10, 0, 10, 0, 10);
VTKGrid->SetDataSet(2, 0, level2Grid.GetPointer());
}
}
namespace FEAdaptor
{
void Initialize(int numScripts, char* scripts[])
{
if(Processor == NULL)
{
Processor = vtkCPProcessor::New();
Processor->Initialize();
}
else
{
Processor->RemoveAllPipelines();
}
for(int i=1;i<numScripts;i++)
{
vtkNew<vtkCPPythonScriptPipeline> pipeline;
pipeline->Initialize(scripts[i]);
Processor->AddPipeline(pipeline.GetPointer());
}
}
void Finalize()
{
if(Processor)
{
Processor->Delete();
Processor = NULL;
}
if(VTKGrid)
{
VTKGrid->Delete();
VTKGrid = NULL;
}
}
void CoProcess(double time, unsigned int timeStep, bool lastTimeStep)
{
vtkNew<vtkCPDataDescription> dataDescription;
dataDescription->AddInput("input");
dataDescription->SetTimeData(time, timeStep);
if(lastTimeStep == true)
{
// assume that we want to all the pipelines to execute if it
// is the last time step.
dataDescription->ForceOutputOn();
}
if(Processor->RequestDataDescription(dataDescription.GetPointer()) != 0)
{
BuildVTKGrid();
dataDescription->GetInputDescriptionByName("input")->SetGrid(VTKGrid);
Processor->CoProcess(dataDescription.GetPointer());
}
}
} // end of Catalyst namespace
|
