NormalEstimation
VTKExamples/Cxx/Points/NormalEstimation
Code¶
NormalEstimation.cxx
#include <vtkSmartPointer.h> #include <vtkPCANormalEstimation.h> #include <vtkPointSource.h> #include <vtkSphereSource.h> #include <vtkArrowSource.h> #include <vtkGlyph3D.h> #include <vtkMath.h> #include <vtkPolyDataMapper.h> #include <vtkProperty.h> #include <vtkRenderWindow.h> #include <vtkRenderWindowInteractor.h> #include <vtkRenderer.h> #include <vtkCamera.h> static void MakeGlyphs(vtkPolyData *src, double size, vtkGlyph3D *glyph); int main (int, char *[]) { vtkMath::RandomSeed(4355412); // for test result consistency double radius = 1.0; vtkSmartPointer<vtkPointSource> points = vtkSmartPointer<vtkPointSource>::New(); points->SetNumberOfPoints(1000); points->SetRadius(radius); points->SetCenter(0.0, 0.0, 0.0); points->SetDistributionToShell(); int sampleSize = 10; vtkSmartPointer<vtkPCANormalEstimation> normals = vtkSmartPointer<vtkPCANormalEstimation>::New(); normals->SetInputConnection (points->GetOutputPort()); normals->SetSampleSize(sampleSize); normals->SetNormalOrientationToGraphTraversal(); normals->Update(); vtkSmartPointer<vtkGlyph3D> glyph3D = vtkSmartPointer<vtkGlyph3D>::New(); MakeGlyphs(normals->GetOutput(), radius * .2, glyph3D.GetPointer()); vtkSmartPointer<vtkPolyDataMapper> glyph3DMapper = vtkSmartPointer<vtkPolyDataMapper>::New(); glyph3DMapper->SetInputConnection(glyph3D->GetOutputPort()); vtkSmartPointer<vtkActor> glyph3DActor = vtkSmartPointer<vtkActor>::New(); glyph3DActor->SetMapper(glyph3DMapper); glyph3DActor->GetProperty()->SetColor(0.8900, 0.8100, 0.3400); vtkSmartPointer<vtkSphereSource> sphere = vtkSmartPointer<vtkSphereSource>::New(); sphere->SetRadius(1.0); sphere->SetThetaResolution(41); sphere->SetPhiResolution(21); vtkSmartPointer<vtkPolyDataMapper> sphereMapper = vtkSmartPointer<vtkPolyDataMapper>::New(); sphereMapper->SetInputConnection(sphere->GetOutputPort()); vtkSmartPointer<vtkActor> sphereActor = vtkSmartPointer<vtkActor>::New(); sphereActor->SetMapper(sphereMapper); sphereActor->GetProperty()->SetColor(1.0000, 0.4900, 0.2500); // Create graphics stuff // vtkSmartPointer<vtkRenderer> ren1 = vtkSmartPointer<vtkRenderer>::New(); ren1->SetBackground(.3, .4, .6); vtkSmartPointer<vtkRenderWindow> renWin = vtkSmartPointer<vtkRenderWindow>::New(); renWin->AddRenderer(ren1); renWin->SetSize(512,512); vtkSmartPointer<vtkRenderWindowInteractor> iren = vtkSmartPointer<vtkRenderWindowInteractor>::New(); iren->SetRenderWindow(renWin); // Add the actors to the renderer, set the background and size // ren1->AddActor(glyph3DActor); ren1->AddActor(sphereActor); // Generate an interesting view // ren1->ResetCamera(); ren1->GetActiveCamera()->Azimuth(120); ren1->GetActiveCamera()->Elevation(30); ren1->GetActiveCamera()->Dolly(1.0); ren1->ResetCameraClippingRange(); iren->Initialize(); iren->Start(); return EXIT_SUCCESS; } void MakeGlyphs(vtkPolyData *src, double size, vtkGlyph3D *glyph) { // Source for the glyph filter vtkSmartPointer<vtkArrowSource> arrow = vtkSmartPointer<vtkArrowSource>::New(); arrow->SetTipResolution(16); arrow->SetTipLength(.3); arrow->SetTipRadius(.1); glyph->SetSourceConnection(arrow->GetOutputPort()); glyph->SetInputData(src); glyph->SetVectorModeToUseNormal(); // glyph->SetColorModeToColorByVector(); glyph->SetScaleModeToScaleByVector(); glyph->SetScaleFactor(size); glyph->OrientOn(); glyph->Update(); }
CMakeLists.txt¶
cmake_minimum_required(VERSION 2.8) PROJECT(NormalEstimation) find_package(VTK REQUIRED) include(${VTK_USE_FILE}) add_executable(NormalEstimation MACOSX_BUNDLE NormalEstimation.cxx) target_link_libraries(NormalEstimation ${VTK_LIBRARIES})
Download and Build NormalEstimation¶
Danger
The generation of tar files has not been ported to the new VTKExamples. Some tarballs may be missing or out-of-date.
Click here to download NormalEstimation and its CMakeLists.txt file. Once the tarball NormalEstimation.tar has been downloaded and extracted,
cd NormalEstimation/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:
./NormalEstimation
WINDOWS USERS PLEASE NOTE: Be sure to add the VTK bin directory to your path. This will resolve the VTK dll's at run time.