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
|
/*=========================================================================
Program: Visualization Toolkit
Module: DistributedData.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.
=========================================================================*/
// Test of vtkProjectSphereFilter. It checks the output in here
// and doesn't compare to an image.
#include "vtkArrayCalculator.h"
#include "vtkCellData.h"
#include "vtkDataArray.h"
#include "vtkNew.h"
#include "vtkPointData.h"
#include "vtkPointDataToCellData.h"
#include "vtkProjectSphereFilter.h"
#include "vtkSphereSource.h"
#include "vtkTestUtilities.h"
namespace
{
// returns true for success. type is Point or Cell to give
// feedback for errors in the passed in array.
bool CheckFieldData(const char* type, vtkDataArray* array, int component,
double minValue, double maxValue)
{
double values[3];
for(vtkIdType i=0;i<array->GetNumberOfTuples();i++)
{
array->GetTuple(i, values);
for(int j=0;j<3;j++)
{
if(j == component)
{
if(values[j] < minValue || values[j] > maxValue)
{
vtkGenericWarningMacro("Array type " << type << " with name "
<< array->GetName() << " has bad value of " << values[j]
<< " but should be between " << minValue << " and " << maxValue);
return false;
}
}
else if(values[j] < -0.001 || values[j] > 0.001)
{
vtkGenericWarningMacro("Array type " << type << " with name " << array->GetName()
<< " should be 0 but has value of " << values[j]);
return false;
}
}
}
return true;
}
}
int TestProjectSphereFilter(int vtkNotUsed(argc), char* [])
{
int numberOfErrors = 0;
vtkNew<vtkSphereSource> sphere;
sphere->SetRadius(1);
sphere->SetCenter(0, 0, 0);
sphere->SetThetaResolution(50);
sphere->SetPhiResolution(50);
vtkNew<vtkArrayCalculator> calculator;
calculator->SetInputConnection(sphere->GetOutputPort());
calculator->SetResultArrayName("result");
calculator->SetFunction("-coordsY*iHat/sqrt(coordsY^2+coordsX^2)+coordsX*jHat/sqrt(coordsY^2+coordsX^2)");
calculator->SetAttributeModeToUsePointData();
calculator->AddCoordinateScalarVariable("coordsX", 0);
calculator->AddCoordinateScalarVariable("coordsY", 1);
vtkNew<vtkProjectSphereFilter> projectSphere;
projectSphere->SetCenter(0, 0, 0);
projectSphere->SetInputConnection(calculator->GetOutputPort());
vtkNew<vtkPointDataToCellData> pointToCell;
pointToCell->SetInputConnection(projectSphere->GetOutputPort());
pointToCell->PassPointDataOn();
pointToCell->Update();
vtkDataSet* grid = pointToCell->GetOutput();
if(grid->GetNumberOfPoints() != 2450)
{
vtkGenericWarningMacro(
"Wrong number of points. There are " << grid->GetNumberOfPoints() <<
" but should be 2450.");
numberOfErrors++;
}
if(grid->GetNumberOfCells() != 4700)
{
vtkGenericWarningMacro(
"Wrong number of cells. There are " << grid->GetNumberOfCells() <<
" but should be 4700.");
numberOfErrors++;
}
if(CheckFieldData("Point", grid->GetPointData()->GetArray("result"), 0, .99, 1.01)
== false)
{
numberOfErrors++;
}
if(CheckFieldData("Point", grid->GetPointData()->GetArray("Normals"), 2, .99, 1.01)
== false)
{
numberOfErrors++;
}
if(CheckFieldData("Cell", grid->GetCellData()->GetArray("Normals"), 2, .99, 1.01)
== false)
{
numberOfErrors++;
}
return numberOfErrors;
}
|