Commit ccf29199 authored by Kenneth Moreland's avatar Kenneth Moreland
Browse files

ENH: Add test for FastSplatter.

parent 41f765b8
......@@ -4,6 +4,7 @@ IF (VTK_USE_RENDERING AND VTK_USE_DISPLAY)
ImportExport.cxx
ImageWeightedSum.cxx
ImageAccumulate.cxx
FastSplatter.cxx
EXTRA_INCLUDE vtkTestDriver.h
)
ADD_EXECUTABLE(${KIT}CxxTests ${Tests})
......
/*
* Copyright 2004 Sandia Corporation.
* Under the terms of Contract DE-AC04-94AL85000, there is a non-exclusive
* license for use of this work by or on behalf of the
* U.S. Government. Redistribution and use in source and binary forms, with
* or without modification, are permitted provided that this Notice and any
* statement of authorship are reproduced on all copies.
*/
// Simple test of vtkFastSplatter
#include "vtkImageData.h"
#include "vtkImageShiftScale.h"
#include "vtkFastSplatter.h"
#include "vtkImageViewer2.h"
#include "vtkPoints.h"
#include "vtkPolyData.h"
#include "vtkRenderer.h"
#include "vtkRenderWindow.h"
#include "vtkRenderWindowInteractor.h"
#include "vtkRegressionTestImage.h"
#include "vtkSmartPointer.h"
#define VTK_CREATE(type, name) \
vtkSmartPointer<type> name = vtkSmartPointer<type>::New()
#include <math.h>
const int SPLAT_IMAGE_SIZE = 100;
int FastSplatter(int argc, char *argv[])
{
// For the purposes of this example we'll build the splat image by
// hand.
VTK_CREATE(vtkImageData, SplatImage);
SplatImage->SetScalarTypeToFloat();
SplatImage->SetNumberOfScalarComponents(1);
SplatImage->SetDimensions(SPLAT_IMAGE_SIZE, SPLAT_IMAGE_SIZE, 1);
SplatImage->AllocateScalars();
for (int i = 0; i < SPLAT_IMAGE_SIZE; ++i)
{
for (int j = 0; j < SPLAT_IMAGE_SIZE; ++j)
{
double xCoord = 1 - fabs( (i - SPLAT_IMAGE_SIZE/2)
/ (SPLAT_IMAGE_SIZE/2.0) );
double yCoord = 1 - fabs( (j - SPLAT_IMAGE_SIZE/2)
/ (SPLAT_IMAGE_SIZE/2.0) );
SplatImage->SetScalarComponentFromDouble(i, j, 0, 0,
xCoord * yCoord );
}
}
VTK_CREATE(vtkPolyData, SplatPoints);
VTK_CREATE(vtkPoints, Points);
Points->SetNumberOfPoints( 5 );
double point[3];
point[0] = 0;
point[1] = 0;
point[2] = 0;
Points->SetPoint( 0, point );
point[0] = 1;
point[1] = 1;
point[2] = 0;
Points->SetPoint( 1, point );
point[0] = -1;
point[1] = 1;
point[2] = 0;
Points->SetPoint( 2, point );
point[0] = 1;
point[1] = -1;
point[2] = 0;
Points->SetPoint( 3, point );
point[0] = -1;
point[1] = -1;
point[2] = 0;
Points->SetPoint( 4, point );
SplatPoints->SetPoints(Points);
VTK_CREATE(vtkFastSplatter, splatter);
splatter->SetInput( SplatPoints );
splatter->SetOutputDimensions( 2*SPLAT_IMAGE_SIZE,
2*SPLAT_IMAGE_SIZE,
1 );
splatter->SetInput(1, SplatImage );
// The image viewers and writers are only happy with unsigned char
// images. This will convert the floats into that format.
VTK_CREATE(vtkImageShiftScale, resultScale);
resultScale->SetOutputScalarTypeToUnsignedChar();
resultScale->SetShift(0);
resultScale->SetScale(255);
resultScale->SetInput( splatter->GetOutput() );
splatter->Update();
resultScale->Update();
// Set up a viewer for the image. vtkImageViewer and
// vtkImageViewer2 are convenient wrappers around vtkActor2D,
// vtkImageMapper, vtkRenderer, and vtkRenderWindow. All you need
// to supply is the interactor and hooray, Bob's your uncle.
VTK_CREATE(vtkImageViewer2, ImageViewer);
ImageViewer->SetInput( resultScale->GetOutput() );
ImageViewer->SetColorLevel(127);
ImageViewer->SetColorWindow(255);
VTK_CREATE(vtkRenderWindowInteractor, iren);
ImageViewer->SetupInteractor(iren);
ImageViewer->Render();
ImageViewer->GetRenderer()->ResetCamera();
int retVal = vtkRegressionTestImage(ImageViewer->GetRenderWindow());
if (retVal == vtkRegressionTester::DO_INTERACTOR)
{
ImageViewer->Render();
iren->Start();
retVal = vtkRegressionTester::PASSED;
}
return !retVal;
}
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment