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
|
/*=========================================================================
Program: Visualization Toolkit
Module: TestGPURayCastRenderToTexture.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.
=========================================================================*/
// Description
// Test the GPU volume mapper low level API to render color and depth buffers to
// the frame buffer.
#include "vtkActor.h"
#include "vtkCamera.h"
#include "vtkColorTransferFunction.h"
#include "vtkGPUVolumeRayCastMapper.h"
#include "vtkImageActor.h"
#include "vtkImageData.h"
#include "vtkImageMapper3D.h"
#include "vtkNew.h"
#include "vtkOpenGLGPUVolumeRayCastMapper.h"
#include "vtkPiecewiseFunction.h"
#include "vtkRegressionTestImage.h"
#include "vtkRenderer.h"
#include "vtkRenderWindow.h"
#include "vtkRenderWindowInteractor.h"
#include "vtkTesting.h"
#include "vtkTestUtilities.h"
#include "vtkVolume16Reader.h"
#include "vtkVolumeProperty.h"
int TestGPURayCastRenderToTexture(int argc, char *argv[])
{
cout << "CTEST_FULL_OUTPUT (Avoid ctest truncation of output)" << endl;
char* fname = vtkTestUtilities::ExpandDataFileName(argc, argv, "Data/headsq/quarter");
vtkNew<vtkVolume16Reader> reader;
reader->SetDataDimensions( 64, 64);
reader->SetDataByteOrderToLittleEndian();
reader->SetImageRange( 1, 93);
reader->SetDataSpacing( 3.2, 3.2, 1.5);
reader->SetFilePrefix( fname );
reader->SetDataMask( 0x7fff);
delete[] fname;
vtkNew<vtkGPUVolumeRayCastMapper> volumeMapper;
volumeMapper->SetInputConnection(reader->GetOutputPort());
volumeMapper->RenderToImageOn();
vtkNew<vtkColorTransferFunction> colorFunction;
colorFunction->AddRGBPoint(900.0, 198/255.0, 134/255.0, 66/255.0);
vtkNew<vtkPiecewiseFunction> scalarOpacity;
scalarOpacity->AddPoint(0, 0.0);
scalarOpacity->AddPoint(70, 0.0);
scalarOpacity->AddPoint(449, 0.0);
scalarOpacity->AddPoint(900, 0.15);
scalarOpacity->AddPoint(1120, 0.25);
scalarOpacity->AddPoint(1404, 0.35);
scalarOpacity->AddPoint(4095, 0.5);
vtkNew<vtkVolumeProperty> volumeProperty;
volumeProperty->ShadeOn();
volumeProperty->SetInterpolationType(VTK_LINEAR_INTERPOLATION);
volumeProperty->SetColor(colorFunction.GetPointer());
volumeProperty->SetScalarOpacity(scalarOpacity.GetPointer());
// Setup volume actor
vtkNew<vtkVolume> volume;
volume->SetMapper(volumeMapper.GetPointer());
volume->SetProperty(volumeProperty.GetPointer());
// Testing prefers image comparison with small images
vtkNew<vtkRenderWindow> renWin;
// Intentional odd and NPOT width/height
renWin->SetSize(401, 399);
vtkNew<vtkRenderer> ren;
renWin->AddRenderer(ren.GetPointer());
vtkNew<vtkRenderWindowInteractor> iren;
iren->SetRenderWindow(renWin.GetPointer());
ren->AddVolume(volume.GetPointer());
ren->GetActiveCamera()->Azimuth(90);
ren->GetActiveCamera()->Roll(90);
ren->GetActiveCamera()->Azimuth(-90);
ren->ResetCamera();
ren->GetActiveCamera()->Zoom(1.8);
renWin->Render();
vtkNew<vtkImageData> im;
// Get color texture as image
volumeMapper->GetColorImage(im.GetPointer());
ren->RemoveVolume(volume.GetPointer());
vtkNew<vtkImageActor> ia;
ia->GetMapper()->SetInputData(im.GetPointer());
ren->AddActor(ia.GetPointer());
ren->GetActiveCamera()->SetPosition(0, 0, -1);
ren->GetActiveCamera()->SetFocalPoint(0, 0, 1);
ren->GetActiveCamera()->SetViewUp(0, 1, 0);
ren->ResetCamera();
renWin->Render();
iren->Initialize();
int retVal = vtkRegressionTestImage( renWin.GetPointer() );
if( retVal == vtkRegressionTester::DO_INTERACTOR)
{
iren->Start();
}
return !retVal;
}
|
