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
|
// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
#include "vtkGDALVectorReader.h"
// VTK includes
#include <vtkActor.h>
#include <vtkCellData.h>
#include <vtkCompositePolyDataMapper.h>
#include <vtkDataSetAttributes.h>
#include <vtkDoubleArray.h>
#include <vtkLookupTable.h>
#include <vtkMapper.h>
#include <vtkMultiBlockDataSet.h>
#include <vtkNew.h>
#include <vtkPolyData.h>
#include <vtkPolyDataMapper.h>
#include <vtkProperty.h>
#include <vtkRegressionTestImage.h>
#include <vtkRenderWindow.h>
#include <vtkRenderWindowInteractor.h>
#include <vtkRenderer.h>
#include <vtkSmartPointer.h>
#include <vtkTestUtilities.h>
// C++ includes
#include <sstream>
// Main program
int TestGDALVectorReader(int argc, char* argv[])
{
const char* vectorFileName =
vtkTestUtilities::ExpandDataFileName(argc, argv, "Data/GIS/countries.shp");
// Create reader to read shape file.
vtkNew<vtkGDALVectorReader> reader;
reader->SetFileName(vectorFileName);
reader->AddFeatureIdsOn();
delete[] vectorFileName;
// Test layer information helpers
reader->UpdateInformation();
int nl = reader->GetNumberOfLayers();
for (int i = 0; i < nl; ++i)
{
reader->SetActiveLayer(i);
cout << "Layer " << i << " Type " << reader->GetActiveLayerType() << " FeatureCount "
<< reader->GetActiveLayerFeatureCount() << "\n";
}
reader->SetActiveLayer(0); // Read only layer 0, which is the only layer.
reader->Update();
// We need a renderer
vtkNew<vtkRenderer> renderer;
// Get the data
vtkSmartPointer<vtkMultiBlockDataSet> mbds = reader->GetOutput();
// Verify that feature IDs exist as a scalar (assuming first block exists)
if (mbds && mbds->GetNumberOfBlocks() > 0)
{
vtkPolyData* pd = vtkPolyData::SafeDownCast(mbds->GetBlock(0));
vtkCellData* cd = pd ? pd->GetCellData() : nullptr;
if (cd)
{
if (!cd->GetPedigreeIds())
{
cerr << "Unable to find pedigree IDs even though AddFeatureIds was ON\n";
return 1;
}
}
}
// Create scene
vtkNew<vtkActor> actor;
vtkNew<vtkCompositePolyDataMapper> mapper;
// Create an interesting lookup table
vtkNew<vtkLookupTable> lut;
lut->SetTableRange(1.0, 8.0);
lut->SetValueRange(0.6, 0.9);
lut->SetHueRange(0.0, 0.8);
lut->SetSaturationRange(0.0, 0.7);
lut->SetNumberOfColors(8);
lut->Build();
mapper->SetInputDataObject(mbds);
mapper->SelectColorArray("mapcolor8");
mapper->SetScalarModeToUseCellFieldData();
mapper->SetScalarVisibility(1);
mapper->UseLookupTableScalarRangeOn();
mapper->SetLookupTable(lut);
mapper->SetColorModeToMapScalars();
actor->SetMapper(mapper);
actor->GetProperty()->SetLineWidth(1.4f);
renderer->AddActor(actor);
// Create a render window, and an interactor
vtkNew<vtkRenderWindow> renderWindow;
vtkNew<vtkRenderWindowInteractor> renderWindowInteractor;
renderWindow->AddRenderer(renderer);
renderWindowInteractor->SetRenderWindow(renderWindow);
// Add the actor to the scene
renderer->SetBackground(1.0, 1.0, 1.0);
renderWindow->SetSize(400, 400);
renderWindow->Render();
renderer->ResetCamera();
renderWindow->Render();
int retVal = vtkRegressionTestImage(renderWindow);
if (retVal == vtkRegressionTester::DO_INTERACTOR)
{
renderWindowInteractor->Start();
}
return !retVal;
}
|
