VTK/Examples/Cxx/DataStructures/KDTreeFindPointsWithinRadius

From KitwarePublic

Jump to: navigation, search

KDTreeFindPointsWithinRadius.cxx

#include <vtkSmartPointer.h>
#include <vtkPoints.h>
#include <vtkPointSource.h>
#include <vtkPolyData.h>
#include <vtkCellArray.h>
#include <vtkIdList.h>
#include <vtkKdTreePointLocator.h>
 
int main(int argc, char *argv[])
{
  //Create some random points
  vtkSmartPointer<vtkPointSource> pointSource = 
      vtkSmartPointer<vtkPointSource>::New();
  pointSource->SetNumberOfPoints(10);
  pointSource->Update();
 
  //Create the tree
  vtkSmartPointer<vtkKdTreePointLocator> pointTree = 
      vtkSmartPointer<vtkKdTreePointLocator>::New();
  pointTree->SetDataSet(pointSource->GetOutput());
  pointTree->BuildLocator();
 
  // Find the k closest points to (0,0,0)
  unsigned int k = 1;
  double testPoint[3] = {0.0, 0.0, 0.0};
  vtkSmartPointer<vtkIdList> result = 
      vtkSmartPointer<vtkIdList>::New();
 
  pointTree->FindPointsWithinRadius(1.0, testPoint, result);
 
  for(vtkIdType i = 0; i < k; i++)
    {
    vtkIdType point_ind = result->GetId(i);
    double p[3];
    pointSource->GetOutput()->GetPoint(point_ind, p);
    std::cout << "Closest point " << i << ": Point "
        << point_ind << ": (" << p[0] << ", " << p[1] << ", " << p[2] << ")" << std::endl;
    }
 
  return EXIT_SUCCESS;
}

CMakeLists.txt

cmake_minimum_required(VERSION 2.8)
 
PROJECT(KDTreeFindPointsWithinRadius.cxx)
 
find_package(VTK REQUIRED)
include(${VTK_USE_FILE})
 
if (APPLE)
  add_executable(KDTreeFindPointsWithinRadius.cxx MACOSX_BUNDLE KDTreeFindPointsWithinRadius.cxx.cxx)
else()
  add_executable(KDTreeFindPointsWithinRadius.cxx KDTreeFindPointsWithinRadius.cxx.cxx)
endif()
 
if(VTK_LIBRARIES)
  target_link_libraries(KDTreeFindPointsWithinRadius.cxx ${VTK_LIBRARIES})
else()
  target_link_libraries(KDTreeFindPointsWithinRadius.cxx vtkHybrid )
endif()
Personal tools