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
|
/*=========================================================================
Program: Visualization Toolkit
Module: ParallelBFS.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.
=========================================================================*/
#include <mpi.h>
#include "vtkEdgeListIterator.h"
#include "vtkGraphLayoutView.h"
#include "vtkInEdgeIterator.h"
#include "vtkInformation.h"
#include "vtkMPIController.h"
#include "vtkPBGLCollectGraph.h"
#include "vtkPBGLDistributedGraphHelper.h"
#include "vtkPBGLBreadthFirstSearch.h"
#include "vtkPBGLRandomGraphSource.h"
#include "vtkRenderer.h"
#include "vtkRenderWindow.h"
#include "vtkRenderWindowInteractor.h"
#include "vtkSmartPointer.h"
#include "vtkStreamingDemandDrivenPipeline.h"
#include "vtkUndirectedGraph.h"
#include "vtkViewTheme.h"
int main(int argc, char** argv)
{
MPI_Init(&argc, &argv);
vtkSmartPointer<vtkPBGLRandomGraphSource> source =
vtkSmartPointer<vtkPBGLRandomGraphSource>::New();
source->DirectedOff();
source->SetNumberOfVertices(100000);
source->SetNumberOfEdges(10000);
source->StartWithTreeOn();
vtkSmartPointer<vtkPBGLBreadthFirstSearch> bfs =
vtkSmartPointer<vtkPBGLBreadthFirstSearch>::New();
bfs->SetInputConnection(source->GetOutputPort());
vtkSmartPointer<vtkPBGLCollectGraph> collect =
vtkSmartPointer<vtkPBGLCollectGraph>::New();
collect->SetInputConnection(bfs->GetOutputPort());
collect->UpdateInformation();
// Setup pipeline request
vtkStreamingDemandDrivenPipeline* exec =
vtkStreamingDemandDrivenPipeline::SafeDownCast(collect->GetExecutive());
vtkSmartPointer<vtkMPIController> controller =
vtkSmartPointer<vtkMPIController>::New();
controller->Initialize(&argc, &argv, 1);
int rank = controller->GetLocalProcessId();
int procs = controller->GetNumberOfProcesses();
exec->SetUpdateNumberOfPieces(exec->GetOutputInformation(0), procs);
exec->SetUpdatePiece(exec->GetOutputInformation(0), rank);
collect->Update();
if (rank == 0)
{
vtkSmartPointer<vtkUndirectedGraph> g =
vtkSmartPointer<vtkUndirectedGraph>::New();
g->ShallowCopy(collect->GetOutput());
vtkSmartPointer<vtkGraphLayoutView> view =
vtkSmartPointer<vtkGraphLayoutView>::New();
vtkSmartPointer<vtkViewTheme> theme;
theme.TakeReference(vtkViewTheme::CreateMellowTheme());
view->ApplyViewTheme(theme);
view->SetRepresentationFromInput(g);
view->SetVertexColorArrayName("BFS");
view->ColorVerticesOn();
vtkRenderWindow* win = view->GetRenderWindow();
view->Update();
view->GetRenderer()->ResetCamera();
win->GetInteractor()->Initialize();
win->GetInteractor()->Start();
}
controller->Finalize();
return 0;
}
|
