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
|
#include <vtkSmartPointer.h>
#include <vtkXMLPolyDataReader.h>
#include <vtkBandedPolyDataContourFilter.h>
#include <vtkFloatArray.h>
#include <vtkCellData.h>
#include <vtkPointData.h>
#include <vtkScalarsToColors.h>
#include <vtkLookupTable.h>
#include <vtkPolyDataMapper.h>
#include <vtkProperty.h>
#include <vtkActor.h>
#include <vtkRenderer.h>
#include <vtkRenderWindow.h>
#include <vtkRenderWindowInteractor.h>
#include <vector>
int main (int argc, char *argv[])
{
if (argc < 3)
{
std::cerr << "Usage: " << argv[0] << " InputPolyDataFile(.vtp) NumberOfContours" << std::endl;
return EXIT_FAILURE;
}
// Read the file
vtkSmartPointer<vtkXMLPolyDataReader> reader =
vtkSmartPointer<vtkXMLPolyDataReader>::New();
reader->SetFileName( argv[1] );
reader->Update(); // Update so that we can get the scalar range
double scalarRange[2];
reader->GetOutput()->GetPointData()->GetScalars()->GetRange(scalarRange);
// Check for a reasonable number of contours to avoid excessive
// computation. Here we arbitrarily pick an upper limit of 1000
int numberOfContours = atoi(argv[2]);
if (numberOfContours > 1000)
{
std::cout << "ERROR: the number of contours " << numberOfContours << " exceeds 1000" << std::endl;
return EXIT_FAILURE;
}
if (numberOfContours <= 0)
{
std::cout << "ERROR: the number of contours " << numberOfContours << " is <= 0" << std::endl;
return EXIT_FAILURE;
}
vtkSmartPointer<vtkBandedPolyDataContourFilter> bandedContours =
vtkSmartPointer<vtkBandedPolyDataContourFilter>::New();
bandedContours->SetInputConnection(reader->GetOutputPort());
bandedContours->SetScalarModeToValue();
bandedContours->GenerateContourEdgesOn();
bandedContours->GenerateValues(
numberOfContours, scalarRange[0], scalarRange[1]);
vtkSmartPointer<vtkLookupTable> lut =
vtkSmartPointer<vtkLookupTable>::New();
lut->SetNumberOfTableValues(numberOfContours + 1);
lut->Build();
vtkSmartPointer<vtkPolyDataMapper> contourMapper =
vtkSmartPointer<vtkPolyDataMapper>::New();
contourMapper->SetInputConnection(bandedContours->GetOutputPort());
contourMapper->SetScalarRange(scalarRange[0], scalarRange[1]);
contourMapper->SetScalarModeToUseCellData();
contourMapper->SetLookupTable(lut);
vtkSmartPointer<vtkActor> contourActor =
vtkSmartPointer<vtkActor>::New();
contourActor->SetMapper(contourMapper);
contourActor->GetProperty()->SetInterpolationToFlat();
vtkSmartPointer<vtkPolyDataMapper> contourLineMapper =
vtkSmartPointer<vtkPolyDataMapper>::New();
contourLineMapper->SetInputData(bandedContours->GetContourEdgesOutput());
contourLineMapper->SetScalarRange(scalarRange[0], scalarRange[1]);
contourLineMapper->ScalarVisibilityOff();
vtkSmartPointer<vtkActor> contourLineActor =
vtkSmartPointer<vtkActor>::New();
contourLineActor->SetMapper(contourLineMapper);
contourLineActor->GetProperty()->SetLineWidth(2);
// The usual renderer, render window and interactor
vtkSmartPointer<vtkRenderer> ren1 =
vtkSmartPointer<vtkRenderer>::New();
vtkSmartPointer<vtkRenderWindow> renWin =
vtkSmartPointer<vtkRenderWindow>::New();
vtkSmartPointer<vtkRenderWindowInteractor>
iren = vtkSmartPointer<vtkRenderWindowInteractor>::New();
ren1->SetBackground(.1, .2, .3);
renWin->AddRenderer(ren1);
iren->SetRenderWindow(renWin);
// Add the actors
ren1->AddActor(contourActor);
ren1->AddActor(contourLineActor);
// Begin interaction
renWin->Render();
iren->Start();
return EXIT_SUCCESS;
}
|
