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
|
/*=========================================================================
Program: Visualization Toolkit
Module: $RCSfile: TestGraphLayoutView.cxx,v $
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 (c) Sandia Corporation
See Copyright.txt or http://www.paraview.org/HTML/Copyright.html for details.
----------------------------------------------------------------------------*/
#include "vtkCellData.h"
#include "vtkCommand.h"
#include "vtkDataRepresentation.h"
#include "vtkGraphLayoutView.h"
#include "vtkIdTypeArray.h"
#include "vtkRegressionTestImage.h"
#include "vtkRenderWindow.h"
#include "vtkRenderWindowInteractor.h"
#include "vtkStringToNumeric.h"
#include "vtkTestUtilities.h"
#include "vtkXMLTreeReader.h"
#include "vtkSmartPointer.h"
#define VTK_CREATE(type, name) \
vtkSmartPointer<type> name = vtkSmartPointer<type>::New()
int TestGraphLayoutView(int argc, char* argv[])
{
char* file = vtkTestUtilities::ExpandDataFileName(argc, argv,
"Data/treetest.xml");
VTK_CREATE(vtkXMLTreeReader, reader);
reader->SetFileName(file);
reader->SetMaskArrays(true);
reader->Update();
vtkTree* t = reader->GetOutput();
VTK_CREATE(vtkIdTypeArray, dist);
dist->SetName("distance");
for (vtkIdType i = 0; i < t->GetNumberOfEdges(); i++)
{
dist->InsertNextValue(i);
}
t->GetEdgeData()->AddArray(dist);
VTK_CREATE(vtkStringToNumeric, numeric);
numeric->SetInput(t);
// Graph layout view
VTK_CREATE(vtkRenderWindow, win);
VTK_CREATE(vtkRenderWindowInteractor, iren);
iren->SetRenderWindow(win);
VTK_CREATE(vtkGraphLayoutView, view);
view->SetLayoutStrategyToCircular();
view->SetVertexLabelArrayName("name");
view->VertexLabelVisibilityOn();
view->SetVertexColorArrayName("size");
view->ColorVerticesOn();
view->SetEdgeColorArrayName("distance");
view->ColorEdgesOn();
view->SetupRenderWindow(win);
view->AddRepresentationFromInputConnection(numeric->GetOutputPort());
int retVal = vtkRegressionTestImage(win);
if (retVal == vtkRegressionTester::DO_INTERACTOR)
{
iren->Initialize();
iren->Start();
retVal = vtkRegressionTester::PASSED;
}
delete [] file;
return !retVal;
}
|