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 126 127 128 129
|
/*=========================================================================
Program: Visualization Toolkit
Module: TestGPURayCastRenderDepthToImage2.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 depth buffer to texture
#include "vtkActor.h"
#include "vtkColorTransferFunction.h"
#include "vtkGPUVolumeRayCastMapper.h"
#include "vtkImageActor.h"
#include "vtkImageData.h"
#include "vtkImageMapToColors.h"
#include "vtkImageMapper3D.h"
#include "vtkLookupTable.h"
#include "vtkNew.h"
#include "vtkOpenGLGPUVolumeRayCastMapper.h"
#include "vtkPiecewiseFunction.h"
#include "vtkRTAnalyticSource.h"
#include "vtkRegressionTestImage.h"
#include "vtkRenderWindow.h"
#include "vtkRenderWindowInteractor.h"
#include "vtkRenderer.h"
#include "vtkTestUtilities.h"
#include "vtkVolumeProperty.h"
int TestGPURayCastRenderDepthToImage2(int argc, char *argv[])
{
cout << "CTEST_FULL_OUTPUT (Avoid ctest truncation of output)" << endl;
vtkNew<vtkGPUVolumeRayCastMapper> volumeMapper;
vtkNew<vtkRTAnalyticSource> waveletSource;
volumeMapper->SetInputConnection(waveletSource->GetOutputPort());
volumeMapper->RenderToImageOn();
volumeMapper->SetClampDepthToBackface(1);
vtkNew<vtkColorTransferFunction> colorFunction;
colorFunction->AddRGBPoint(37.35310363769531, 0.231373, 0.298039, 0.752941);
colorFunction->AddRGBPoint(157.0909652709961, 0.865003, 0.865003, 0.865003);
colorFunction->AddRGBPoint(276.8288269042969, 0.705882, 0.0156863, 0.14902);
float dataRange[2];
dataRange[0] = 37.3;
dataRange[1] = 276.8;
float halfSpread = (dataRange[1] - dataRange[0]) / 2.0;
float center = dataRange[0] + halfSpread;
vtkNew<vtkPiecewiseFunction> scalarOpacity;
scalarOpacity->RemoveAllPoints();
scalarOpacity->AddPoint(center, 0.0);
scalarOpacity->AddPoint(dataRange[1], 0.4);
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->ResetCamera();
renWin->Render();
vtkNew<vtkImageData> im;
// Get color texture as image
volumeMapper->GetColorImage(im.GetPointer());
// Get depth texture as image
volumeMapper->GetDepthImage(im.GetPointer());
// Create a grayscale lookup table
vtkNew<vtkLookupTable> lut;
lut->SetRange(0.0, 1.0);
lut->SetValueRange(0.0, 1.0);
lut->SetSaturationRange(0.0, 0.0);
lut->SetRampToLinear();
lut->Build();
// Map the pixel values of the image with the lookup table
vtkNew<vtkImageMapToColors> imageMap;
imageMap->SetInputData(im.GetPointer());
imageMap->SetLookupTable(lut.GetPointer());
// Render the image in the scene
vtkNew<vtkImageActor> ia;
ia->GetMapper()->SetInputConnection(imageMap->GetOutputPort());
ren->AddActor(ia.GetPointer());
ren->RemoveVolume(volume.GetPointer());
ren->ResetCamera();
renWin->Render();
iren->Initialize();
int retVal = vtkRegressionTestImage( renWin.GetPointer() );
if( retVal == vtkRegressionTester::DO_INTERACTOR)
{
iren->Start();
}
return !retVal;
}
|