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
|
/*=========================================================================
Program: Visualization Toolkit
Module: TestGDALRasterReader.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 <vtkDataArray.h>
#include <vtkGDALRasterReader.h>
#include <vtkImageActor.h>
#include <vtkImageProperty.h>
#include <vtkLookupTable.h>
#include <vtkNew.h>
#include <vtkPointData.h>
#include <vtkRegressionTestImage.h>
#include <vtkRenderWindow.h>
#include <vtkRenderer.h>
#include <vtkRenderWindow.h>
#include <vtkRenderWindowInteractor.h>
#include <vtkUniformGrid.h>
#include <iostream>
// Main program
int TestGDALRasterPalette(int argc, char** argv)
{
if (argc < 3)
{
std::cerr << "Expected TestName -D InputFile.tif" << std::endl;
return -1;
}
std::string inputFileName(argv[2]);
// Create reader to read shape file.
vtkNew<vtkGDALRasterReader> reader;
reader->SetFileName(inputFileName.c_str());
reader->Update();
vtkUniformGrid *image = vtkUniformGrid::SafeDownCast(reader->GetOutput());
// Check that reader generated point scalars
if (image->GetPointData()->GetNumberOfArrays() < 1)
{
std::cerr << "ERROR: Missing point data scalars" << std::endl;
return 1;
}
if (image->GetPointData()->GetScalars()->GetSize() == 0)
{
std::cerr << "ERROR: Point data scalars empty" << std::endl;
return 1;
}
//image->GetPointData()->GetScalars()->Print(std::cout);
// Check that reader generated color table
vtkLookupTable *colorTable =
image->GetPointData()->GetScalars()->GetLookupTable();
if (!colorTable)
{
std::cerr << "ERROR: Missing color table" << std::endl;
return 1;
}
if (colorTable->GetNumberOfAvailableColors() != 256)
{
std::cerr << "ERROR: Color table does not have 256 colors."
<< " Instead has " << colorTable->GetNumberOfAvailableColors()
<< std::endl;
return 1;
}
//colorTable->Print(std::cout);
// Create a renderer and actor
vtkNew<vtkRenderer> renderer;
vtkNew<vtkImageActor> actor;
actor->SetInputData(reader->GetOutput());
actor->InterpolateOff();
//actor->GetProperty()->SetInterpolationTypeToNearest();
actor->GetProperty()->SetLookupTable(colorTable);
actor->GetProperty()->UseLookupTableScalarRangeOn();
renderer->AddActor(actor.GetPointer());
// Create a render window, and an interactor
vtkNew<vtkRenderWindow> renderWindow;
vtkNew<vtkRenderWindowInteractor> renderWindowInteractor;
renderWindow->AddRenderer(renderer.GetPointer());
renderWindowInteractor->SetRenderWindow(renderWindow.GetPointer());
//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.GetPointer());
if (retVal == vtkRegressionTester::DO_INTERACTOR)
{
renderWindowInteractor->Start();
}
return !retVal;
}
|
