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
|
#include <vtkDataSetAttributes.h>
#include <vtkFloatArray.h>
#include <vtkMutableUndirectedGraph.h>
#include <vtkPoints.h>
#include <vtkSmartPointer.h>
#include "vtkGraphWeightEuclideanDistanceFilter.h"
template<typename T>
static bool FuzzyCompare(const T a, const T b);
int TestGraphWeightEuclideanDistanceFilter(int, char *[])
{
// Create a graph
vtkSmartPointer<vtkMutableUndirectedGraph> g =
vtkSmartPointer<vtkMutableUndirectedGraph>::New();
// Add 4 vertices to the graph
vtkIdType v1 = g->AddVertex();
vtkIdType v2 = g->AddVertex();
vtkIdType v3 = g->AddVertex();
vtkIdType v4 = g->AddVertex();
// Add 3 edges to the graph
g->AddEdge ( v1, v2 );
g->AddEdge ( v1, v3 );
g->AddEdge ( v1, v4 );
// Create 4 points - one for each vertex
vtkSmartPointer<vtkPoints> points =
vtkSmartPointer<vtkPoints>::New();
points->InsertNextPoint(0.0, 0.0, 0.0);
points->InsertNextPoint(1.0, 0.0, 0.0);
points->InsertNextPoint(0.0, 1.0, 0.0);
points->InsertNextPoint(0.0, 0.0, 2.0);
// Add the coordinates of the points to the graph
g->SetPoints(points);
vtkSmartPointer<vtkGraphWeightEuclideanDistanceFilter> weightFilter =
vtkSmartPointer<vtkGraphWeightEuclideanDistanceFilter>::New();
weightFilter->SetInputData(g);
weightFilter->Update();
vtkFloatArray* weights = vtkArrayDownCast<vtkFloatArray>(weightFilter->GetOutput()->GetEdgeData()->GetArray("Weights"));
std::vector<float> correctWeights;
correctWeights.push_back(1.0f);
correctWeights.push_back(1.0f);
correctWeights.push_back(2.0f);
for(vtkIdType i = 0; i < weights->GetNumberOfTuples(); ++i)
{
float w = weights->GetValue(i);
if(!FuzzyCompare<float>(w, correctWeights[i]))
{
std::cerr << "Weight " << i << " was " << w << " and should have been " << correctWeights[i] << std::endl;
return EXIT_FAILURE;
}
}
return EXIT_SUCCESS;
}
template<typename T>
static bool FuzzyCompare(const T a, const T b)
{
if(fabs(static_cast<float>(a) - static_cast<float>(b)) < 1e-4)
{
return true;
}
return false;
}
|