Skip to content

Commit 88d0875

Browse files
author
David E. DeMarle
committed
Add a demonstration of creating an AMR data set
imported from Kitware's VTK course material
1 parent 6ac05f6 commit 88d0875

3 files changed

Lines changed: 221 additions & 0 deletions

File tree

‎CMakeLists.txt‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -70,6 +70,7 @@ endif()
7070
message(STATUS "VTKWikiExamples: Example executables are stored in ${CMAKE_RUNTIME_OUTPUT_DIRECTORY}")
7171

7272
ADD_SUBDIRECTORY(Cxx/Animation)
73+
ADD_SUBDIRECTORY(Cxx/CompositeData)
7374
IF(VTK_USE_MYSQL)
7475
ADD_SUBDIRECTORY(Cxx/Databases)
7576
ENDIF(VTK_USE_MYSQL)

‎Cxx/CompositeData/CMakeLists.txt‎

Lines changed: 38 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,38 @@
1+
PROJECT (${WIKI}CompositeData)
2+
3+
IF(NOT WikiExamples_BINARY_DIR)
4+
FIND_PACKAGE(VTK REQUIRED)
5+
IF(NOT VTK_USE_RENDERING)
6+
MESSAGE(FATAL_ERROR "Example ${PROJECT_NAME} requires VTK_USE_RENDERING.")
7+
ENDIF(NOT VTK_USE_RENDERING)
8+
INCLUDE(${VTK_USE_FILE})
9+
ENDIF(NOT WikiExamples_BINARY_DIR)
10+
11+
if("${VTK_MAJOR_VERSION}.${VTK_MINOR_VERSION}" LESS 5.8)
12+
SET(KIT_LIBS vtkFiltering vtkHybrid)
13+
else()
14+
SET(KIT_LIBS ${VTK_LIBRARIES})
15+
endif()
16+
#
17+
# Build all .cxx files in the directory
18+
FILE(GLOB ALL_FILES *.cxx)
19+
foreach(SOURCE_FILE ${ALL_FILES})
20+
STRING(REPLACE ".cxx" "" TMP ${SOURCE_FILE})
21+
STRING(REPLACE ${CMAKE_CURRENT_SOURCE_DIR}/ "" EXAMPLE ${TMP})
22+
ADD_EXECUTABLE(${WIKI}${EXAMPLE} ${EXECUTABLE_FLAG} ${EXAMPLE}.cxx)
23+
TARGET_LINK_LIBRARIES(${WIKI}${EXAMPLE} ${KIT_LIBS})
24+
endforeach(SOURCE_FILE)
25+
26+
if (BUILD_TESTING)
27+
# Testing
28+
SET(KIT CompositeData)
29+
SET(NEEDS_ARGS
30+
# none yet
31+
)
32+
33+
ADD_TEST(${KIT}-OverlappingAMR ${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/${KIT}CxxTests
34+
TestOverlappingAMR -E 35)
35+
36+
INCLUDE(${WikiExamples_SOURCE_DIR}/CMake/ExamplesTesting.cmake)
37+
38+
endif()

‎Cxx/CompositeData/OverlappingAMR.cxx‎

Lines changed: 182 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,182 @@
1+
//Demonstrates how to create and populate a VTK's Overlapping AMR Grid type Data
2+
3+
#include <vtkAMRBox.h>
4+
#include <vtkAMRUtilities.h>
5+
#include <vtkContourFilter.h>
6+
#include <vtkFloatArray.h>
7+
#include <vtkCompositeDataGeometryFilter.h>
8+
#include <vtkOutlineFilter.h>
9+
#include <vtkOverlappingAMR.h>
10+
#include <vtkPointData.h>
11+
#include <vtkPolyDataMapper.h>
12+
#include <vtkRenderer.h>
13+
#include <vtkRenderWindow.h>
14+
#include <vtkRenderWindowInteractor.h>
15+
#include <vtkSmartPointer.h>
16+
#include <vtkSphere.h>
17+
#include <vtkUniformGrid.h>
18+
19+
namespace
20+
{
21+
void MakeScalars(int dims[3], double origin[3], double spacing[3],
22+
vtkFloatArray* scalars)
23+
{
24+
// Implicit function used to compute scalars
25+
vtkSmartPointer<vtkSphere> sphere = vtkSmartPointer<vtkSphere>::New();
26+
sphere->SetRadius(3);
27+
sphere->SetCenter(5, 5, 5);
28+
29+
scalars->SetNumberOfTuples(dims[0]*dims[1]*dims[2]);
30+
for (int k=0; k<dims[2]; k++)
31+
{
32+
double z = origin[2] + spacing[2]*k;
33+
for (int j=0; j<dims[1]; j++)
34+
{
35+
double y = origin[1] + spacing[1]*j;
36+
for (int i=0; i<dims[0]; i++)
37+
{
38+
double x = origin[0] + spacing[0]*i;
39+
scalars->SetValue(
40+
k*dims[0]*dims[1] + j*dims[0] + i,
41+
sphere->EvaluateFunction(x, y, z));
42+
}
43+
}
44+
}
45+
}
46+
}
47+
48+
int main (int, char *[])
49+
{
50+
// Create and populate the AMR dataset
51+
// The dataset should look like
52+
// Level 0
53+
// uniform grid, dimensions 11, 11, 11, AMR box (0, 0, 0) - (9, 9, 9)
54+
// Level 1 - refinement ratio : 2
55+
// uniform grid, dimensions 11, 11, 11, AMR box (0, 0, 0) - (9, 9, 9)
56+
// uniform grid, dimensions 11, 11, 11, AMR box (10, 10, 10) - (19, 19, 19)
57+
// Use MakeScalars() above to fill the scalar arrays
58+
59+
vtkSmartPointer<vtkOverlappingAMR> amr =
60+
vtkSmartPointer<vtkOverlappingAMR>::New();
61+
int blocksPerLevel[] = { 1, 2 };
62+
amr->Initialize(2, blocksPerLevel);
63+
64+
double origin[3] = {0.0, 0.0, 0.0};
65+
double spacing[3] = {1.0, 1.0, 1.0};
66+
int dims[3] = {11, 11, 11};
67+
68+
vtkSmartPointer<vtkUniformGrid> ug1 = vtkSmartPointer<vtkUniformGrid>::New();
69+
// Geometry
70+
ug1->SetOrigin(origin);
71+
ug1->SetSpacing(spacing);
72+
ug1->SetDimensions(dims);
73+
74+
// Data
75+
vtkSmartPointer<vtkFloatArray> scalars =
76+
vtkSmartPointer<vtkFloatArray>::New();
77+
ug1->GetPointData()->SetScalars(scalars);
78+
MakeScalars(dims, origin, spacing, scalars);
79+
80+
int lo[3] = {0, 0, 0};
81+
int hi[3] = {9, 9, 9};
82+
vtkAMRBox box1(lo, hi);
83+
amr->SetAMRBox(0, 0, box1);
84+
amr->SetDataSet(0, 0, ug1);
85+
86+
double spacing2[3] = {0.5, 0.5, 0.5};
87+
88+
vtkSmartPointer<vtkUniformGrid> ug2 = vtkSmartPointer<vtkUniformGrid>::New();
89+
// Geometry
90+
ug2->SetOrigin(origin);
91+
ug2->SetSpacing(spacing2);
92+
ug2->SetDimensions(dims);
93+
94+
// Data
95+
scalars = vtkSmartPointer<vtkFloatArray>::New();
96+
ug2->GetPointData()->SetScalars(scalars);
97+
MakeScalars(dims, origin, spacing2, scalars);
98+
99+
int lo2[3] = {0, 0, 0};
100+
int hi2[3] = {9, 9, 9};
101+
vtkAMRBox box2(lo2, hi2);
102+
amr->SetAMRBox(1, 0, box2);
103+
amr->SetDataSet(1, 0, ug2);
104+
105+
double origin3[3] = {5, 5, 5};
106+
107+
vtkSmartPointer<vtkUniformGrid> ug3 = vtkSmartPointer<vtkUniformGrid>::New();
108+
// Geometry
109+
ug3->SetOrigin(origin3);
110+
ug3->SetSpacing(spacing2);
111+
ug3->SetDimensions(dims);
112+
113+
// Data
114+
scalars = vtkSmartPointer<vtkFloatArray>::New();
115+
ug3->GetPointData()->SetScalars(scalars);
116+
MakeScalars(dims, origin3, spacing2, scalars);
117+
118+
int lo3[3] = {10, 10, 10};
119+
int hi3[3] = {19, 19, 19};
120+
vtkAMRBox box3(lo3, hi3);
121+
amr->SetAMRBox(1, 1, box3);
122+
amr->SetDataSet(1, 1, ug3);
123+
124+
amr->SetRefinementRatio(0, 2);
125+
vtkAMRUtilities::BlankCells(amr);
126+
127+
// Render the amr data here
128+
vtkSmartPointer<vtkOutlineFilter> of =
129+
vtkSmartPointer<vtkOutlineFilter>::New();
130+
of->SetInputData(amr);
131+
132+
vtkSmartPointer<vtkCompositeDataGeometryFilter> geomFilter =
133+
vtkSmartPointer<vtkCompositeDataGeometryFilter>::New();
134+
geomFilter->SetInputConnection(of->GetOutputPort());
135+
136+
// Create an iso-surface - at 10
137+
vtkSmartPointer<vtkContourFilter> cf =
138+
vtkSmartPointer<vtkContourFilter>::New();
139+
cf->SetInputData(amr);
140+
cf->SetNumberOfContours(1);
141+
cf->SetValue(0, 10.0);
142+
143+
vtkSmartPointer<vtkCompositeDataGeometryFilter> geomFilter2 =
144+
vtkSmartPointer<vtkCompositeDataGeometryFilter>::New();
145+
geomFilter2->SetInputConnection(cf->GetOutputPort());
146+
147+
// create the render window, renderer, and interactor
148+
vtkSmartPointer<vtkRenderer> aren = vtkSmartPointer<vtkRenderer>::New();
149+
vtkSmartPointer<vtkRenderWindow> renWin =
150+
vtkSmartPointer<vtkRenderWindow>::New();
151+
renWin->AddRenderer(aren);
152+
153+
vtkSmartPointer<vtkRenderWindowInteractor> iren =
154+
vtkSmartPointer<vtkRenderWindowInteractor>::New();
155+
iren->SetRenderWindow(renWin);
156+
157+
// associate the geometry with a mapper and the mapper to an actor
158+
vtkSmartPointer<vtkPolyDataMapper> mapper =
159+
vtkSmartPointer<vtkPolyDataMapper>::New();
160+
161+
mapper->SetInputConnection(geomFilter->GetOutputPort());
162+
163+
vtkActor *actor1 = vtkActor::New();
164+
actor1->SetMapper(mapper);
165+
166+
// associate the geometry with a mapper and the mapper to an actor
167+
vtkSmartPointer<vtkPolyDataMapper> mapper2 =
168+
vtkSmartPointer<vtkPolyDataMapper>::New();
169+
170+
mapper2->SetInputConnection(geomFilter2->GetOutputPort());
171+
172+
vtkActor *actor2 = vtkActor::New();
173+
actor2->SetMapper(mapper2);
174+
175+
// add the actor to the renderer and start handling events
176+
aren->AddActor(actor1);
177+
aren->AddActor(actor2);
178+
renWin->Render();
179+
iren->Start();
180+
181+
return EXIT_SUCCESS;
182+
}

0 commit comments

Comments
 (0)