Skip to content

ShortestPath

Repository source: ShortestPath

Description

Currently, the only way to find the shortest path on a graph is to convert the graph to a mesh (using vtkGraphToPolyData) and then use the shortest path on a mesh functionality of vtkDijkstraGraphGeodesicPath.

Question

If you have a question about this example, please use the VTK Discourse Forum

Code

ShortestPath.cxx

#include <vtkActor.h>
#include <vtkDijkstraGraphGeodesicPath.h>
#include <vtkGraphLayoutView.h>
#include <vtkGraphToPolyData.h>
#include <vtkMutableDirectedGraph.h>
#include <vtkNamedColors.h>
#include <vtkNew.h>
#include <vtkPolyDataMapper.h>
#include <vtkProperty.h>
#include <vtkRenderWindow.h>
#include <vtkRenderWindowInteractor.h>
#include <vtkRenderer.h>

/*
   O v0
  /|\
 / |5\
v1-v2-v3
  1  1

  Shortest path between v0 and v2 should be 5
*/

int main(int argc, char* argv[])
{
  vtkNew<vtkNamedColors> colors;

  vtkNew<vtkMutableDirectedGraph> graph;
  vtkIdType v0 = graph->AddVertex();
  vtkIdType v1 = graph->AddVertex();
  vtkIdType v2 = graph->AddVertex();
  vtkIdType v3 = graph->AddVertex();

  graph->AddEdge(v0, v1);
  graph->AddEdge(v0, v2);
  graph->AddEdge(v0, v3);
  graph->AddEdge(v1, v2);
  graph->AddEdge(v2, v3);

  // Associate physical locations with the vertices
  vtkNew<vtkPoints> points;
  points->InsertNextPoint(0.0, 0.0, 0.0);
  points->InsertNextPoint(-1.0, -5.0, 0.0);
  points->InsertNextPoint(0.0, -5.0, 0.0);
  points->InsertNextPoint(1.0, -5.0, 0.0);

  graph->SetPoints(points);

  // Convert the graph to a polydata
  vtkNew<vtkGraphToPolyData> graphToPolyData;
  graphToPolyData->SetInputData(graph);
  graphToPolyData->Update();

  vtkNew<vtkDijkstraGraphGeodesicPath> dijkstra;
  dijkstra->SetInputConnection(graphToPolyData->GetOutputPort());
  dijkstra->SetStartVertex(0);
  dijkstra->SetEndVertex(2);
  dijkstra->Update();

  // Create a mapper and actor
  vtkNew<vtkPolyDataMapper> pathMapper;
  pathMapper->SetInputConnection(dijkstra->GetOutputPort());

  vtkNew<vtkActor> pathActor;
  pathActor->SetMapper(pathMapper);
  pathActor->GetProperty()->SetColor(colors->GetColor3d("Tomato").GetData());
  pathActor->GetProperty()->SetLineWidth(4);

  // Create a mapper and actor
  vtkNew<vtkPolyDataMapper> mapper;
  mapper->SetInputConnection(graphToPolyData->GetOutputPort());

  vtkNew<vtkActor> actor;
  actor->SetMapper(mapper);
  pathActor->GetProperty()->SetColor(
      colors->GetColor3d("NavajoWhite").GetData());

  // Create a renderer, render window, and interactor
  vtkNew<vtkRenderer> renderer;
  vtkNew<vtkRenderWindow> renderWindow;
  renderWindow->AddRenderer(renderer);
  vtkNew<vtkRenderWindowInteractor> renderWindowInteractor;
  renderWindowInteractor->SetRenderWindow(renderWindow);

  // Add the actor to the scene
  renderer->AddActor(actor);
  renderer->AddActor(pathActor);
  renderer->SetBackground(colors->GetColor3d("DarkGreen").GetData());
  renderer->SetBackground2(colors->GetColor3d("ForestGreen").GetData());
  renderer->GradientBackgroundOn();
  renderWindow->SetWindowName("ShortestPath");

  // Render and interact
  renderWindow->Render();
  renderWindowInteractor->Start();

  return EXIT_SUCCESS;
}

CMakeLists.txt

cmake_minimum_required(VERSION 3.12 FATAL_ERROR)

project(ShortestPath)

find_package(VTK COMPONENTS 
  CommonColor
  CommonCore
  CommonDataModel
  FiltersModeling
  FiltersSources
  InteractionStyle
  RenderingContextOpenGL2
  RenderingCore
  RenderingFreeType
  RenderingGL2PSOpenGL2
  RenderingOpenGL2
  ViewsInfovis
)

if (NOT VTK_FOUND)
  message(FATAL_ERROR "ShortestPath: Unable to find the VTK build folder.")
endif()

# Prevent a "command line is too long" failure in Windows.
set(CMAKE_NINJA_FORCE_RESPONSE_FILE "ON" CACHE BOOL "Force Ninja to use response files.")
add_executable(ShortestPath MACOSX_BUNDLE ShortestPath.cxx )
  target_link_libraries(ShortestPath PRIVATE ${VTK_LIBRARIES}
)
# vtk_module_autoinit is needed
vtk_module_autoinit(
  TARGETS ShortestPath
  MODULES ${VTK_LIBRARIES}
)

Download and Build ShortestPath

Click here to download ShortestPath and its CMakeLists.txt file. Once the tarball ShortestPath.tar has been downloaded and extracted,

cd ShortestPath/build

If VTK is installed:

cmake ..

If VTK is not installed but compiled on your system, you will need to specify the path to your VTK build:

cmake -DVTK_DIR:PATH=/home/me/vtk_build ..

Build the project:

make

and run it:

./ShortestPath

WINDOWS USERS

Be sure to add the VTK bin directory to your path. This will resolve the VTK dll's at run time.