VTK/Examples/Cxx/Graphs/BreadthFirstDistance

From KitwarePublic

Jump to: navigation, search

This example shows how to get the distance from a vertex to the root vertex.

BreadthFirstDistance.cxx

#include <vtkSmartPointer.h>
 
#include <vtkBoostBreadthFirstSearch.h>
#include <vtkDataSetAttributes.h>
#include <vtkDoubleArray.h>
#include <vtkIntArray.h>
#include <vtkMutableUndirectedGraph.h>
#include <vtkTree.h>
 
int main(int, char *[])
{
  vtkSmartPointer<vtkMutableUndirectedGraph> g = 
    vtkSmartPointer<vtkMutableUndirectedGraph>::New();
 
  // Create 3 vertices
  vtkIdType v1 = g->AddVertex();
  vtkIdType v2 = g->AddVertex();
  vtkIdType v3 = g->AddVertex();
  vtkIdType v4 = g->AddVertex();
 
  // Create a graph
  g->AddEdge(v1, v2);
  g->AddEdge(v1, v3);
  g->AddEdge(v2, v4);
 
  vtkSmartPointer<vtkBoostBreadthFirstSearch> BFS = 
    vtkSmartPointer<vtkBoostBreadthFirstSearch>::New();
  BFS->SetOriginVertex(v1);
  BFS->SetInput(g);
  BFS->Update();
 
  vtkIntArray* level = vtkIntArray::SafeDownCast(
    BFS->GetOutput()->GetVertexData()->GetArray("BFS"));
  for(vtkIdType i = 0; i < level->GetNumberOfTuples(); i++)
    {
    std::cout << "id " << i << " : " << level->GetValue(i) << std::endl;
    }
  return EXIT_SUCCESS;
}

CMakeLists.txt

cmake_minimum_required(VERSION 2.6)
 
PROJECT(BreadthFirstSearch)
 
FIND_PACKAGE(VTK REQUIRED)
INCLUDE(${VTK_USE_FILE})
 
ADD_EXECUTABLE(BreadthFirstSearch BreadthFirstSearch.cpp)
TARGET_LINK_LIBRARIES(BreadthFirstSearch vtkHybrid vtkInfovis)
Retrieved from "http://www.vtk.org/Wiki/VTK/Examples/Cxx/Graphs/BreadthFirstDistance"
Personal tools