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
|
/*=========================================================================
Program: Visualization Toolkit
Module: TestPruneTreeFilter.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.
=========================================================================*/
/*-------------------------------------------------------------------------
Copyright 2008 Sandia Corporation.
Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
the U.S. Government retains certain rights in this software.
-------------------------------------------------------------------------*/
#include "vtkActor.h"
#include "vtkActor2D.h"
#include "vtkDataSetAttributes.h"
#include "vtkGraphLayout.h"
#include "vtkGraphMapper.h"
#include "vtkGraphToPolyData.h"
#include "vtkIdTypeArray.h"
#include "vtkLabeledDataMapper.h"
#include "vtkMutableDirectedGraph.h"
#include "vtkPruneTreeFilter.h"
#include "vtkRegressionTestImage.h"
#include "vtkRenderer.h"
#include "vtkRenderWindow.h"
#include "vtkRenderWindowInteractor.h"
#include "vtkSmartPointer.h"
#include "vtkTestUtilities.h"
#include "vtkTreeLayoutStrategy.h"
#define VTK_CREATE(type, name) \
vtkSmartPointer<type> name = vtkSmartPointer<type>::New()
int TestPruneTreeFilter(int argc, char* argv[])
{
VTK_CREATE(vtkMutableDirectedGraph, builder);
builder->AddVertex(); // 0
builder->AddChild(0); // 1
builder->AddChild(0); // 2
builder->AddChild(1); // 3
builder->AddChild(1); // 4
builder->AddChild(2); // 5
builder->AddChild(2); // 6
builder->AddChild(3); // 7
builder->AddChild(3); // 8
builder->AddChild(4); // 9
builder->AddChild(4); // 10
VTK_CREATE(vtkTree, tree);
tree->ShallowCopy(builder);
VTK_CREATE(vtkIdTypeArray, idArr);
idArr->SetName("id");
for (vtkIdType i = 0; i < 11; ++i)
{
idArr->InsertNextValue(i);
}
tree->GetVertexData()->AddArray(idArr);
VTK_CREATE(vtkPruneTreeFilter, prune);
prune->SetInputData(tree);
prune->SetParentVertex(2);
VTK_CREATE(vtkTreeLayoutStrategy, strategy);
VTK_CREATE(vtkGraphLayout, layout);
//layout->SetInput(tree);
layout->SetInputConnection(prune->GetOutputPort());
layout->SetLayoutStrategy(strategy);
VTK_CREATE(vtkGraphToPolyData, poly);
poly->SetInputConnection(layout->GetOutputPort());
VTK_CREATE(vtkLabeledDataMapper, labelMapper);
labelMapper->SetInputConnection(poly->GetOutputPort());
labelMapper->SetLabelModeToLabelFieldData();
labelMapper->SetInputArrayToProcess(0, 0, 0, vtkDataObject::FIELD_ASSOCIATION_POINTS, "id");
VTK_CREATE(vtkActor2D, labelActor);
labelActor->SetMapper(labelMapper);
VTK_CREATE(vtkGraphMapper, mapper);
mapper->SetInputConnection(layout->GetOutputPort());
VTK_CREATE(vtkActor, actor);
actor->SetMapper(mapper);
VTK_CREATE(vtkRenderer, ren);
ren->AddActor(actor);
ren->AddActor(labelActor);
VTK_CREATE(vtkRenderWindowInteractor, iren);
VTK_CREATE(vtkRenderWindow, win);
win->AddRenderer(ren);
win->SetInteractor(iren);
int retVal = vtkRegressionTestImage(win);
if (retVal == vtkRegressionTester::DO_INTERACTOR)
{
iren->Initialize();
iren->Start();
retVal = vtkRegressionTester::PASSED;
}
return !retVal;
}
|