VTK 实例52：曲率计算

#include "vtkAutoInit.h" 
//VTK_MODULE_INIT(vtkRenderingOpenGL2);
//VTK_MODULE_INIT(vtkInteractionStyle);

VTK_MODULE_INIT(vtkRenderingOpenGL2);
VTK_MODULE_INIT(vtkInteractionStyle);
VTK_MODULE_INIT(vtkRenderingFreeType);

#include <vtkSmartPointer.h>
#include <vtkPolyDataReader.h>
#include <vtkCurvatures.h>
#include <vtkLookupTable.h>
#include <vtkPolyDataMapper.h>
#include <vtkActor.h>
#include <vtkScalarBarActor.h>
#include <vtkPointData.h>
#include <vtkRenderer.h>
#include <vtkRenderWindow.h>
#include <vtkRenderWindowInteractor.h>

    int main()
    {
        vtkSmartPointer<vtkPolyDataReader> reader =
            vtkSmartPointer<vtkPolyDataReader>::New();
        reader->SetFileName("C:\\Users\\Administrator\\Desktop\\VTK2\\hellovtk\\vtk_图像处理学习\\第六章_图像处理\\data\\fran_cut.vtk");
        reader->Update();

        vtkSmartPointer<vtkCurvatures> curvaturesFilter =
            vtkSmartPointer<vtkCurvatures>::New();
        curvaturesFilter->SetInputConnection(reader->GetOutputPort());
        //curvaturesFilter->SetCurvatureTypeToMinimum(); //最小曲率
        curvaturesFilter->SetCurvatureTypeToMaximum();   //最大曲率
        //curvaturesFilter->SetCurvatureTypeToGaussian();//高斯曲率
        //curvaturesFilter->SetCurvatureTypeToMean();    //平均曲率
        curvaturesFilter->Update();

        double scalarRange[2];
        curvaturesFilter->GetOutput()->GetScalarRange(scalarRange);
        //建立查找表 做颜色映射
        vtkSmartPointer<vtkLookupTable> lut =
            vtkSmartPointer<vtkLookupTable>::New();
        lut->SetHueRange(0.0, 0.6);
        lut->SetAlphaRange(1.0, 1.0);
        lut->SetValueRange(1.0, 1.0);
        lut->SetSaturationRange(1.0, 1.0);
        lut->SetNumberOfTableValues(256);
        lut->SetRange(scalarRange);
        lut->Build();
        ///
        vtkSmartPointer<vtkPolyDataMapper> mapper =
            vtkSmartPointer<vtkPolyDataMapper>::New();
        mapper->SetInputData(curvaturesFilter->GetOutput());
        mapper->SetLookupTable(lut);
        mapper->SetScalarRange(scalarRange);

        vtkSmartPointer<vtkActor> actor =
            vtkSmartPointer<vtkActor>::New();
        actor->SetMapper(mapper);

        vtkSmartPointer<vtkScalarBarActor> scalarBar =
            vtkSmartPointer<vtkScalarBarActor>::New();
        scalarBar->SetLookupTable(mapper->GetLookupTable());
        scalarBar->SetTitle(curvaturesFilter->GetOutput()->GetPointData()->GetScalars()->GetName());
        scalarBar->SetNumberOfLabels(5); //设置5个标签

        vtkSmartPointer<vtkRenderer> render =
            vtkSmartPointer<vtkRenderer>::New();
        render->AddActor(actor);
        render->AddActor2D(scalarBar);
        render->SetBackground(0, 0, 0);

        vtkSmartPointer<vtkRenderWindow> rw =
            vtkSmartPointer<vtkRenderWindow>::New();
        rw->AddRenderer(render);
        rw->SetSize(640, 480);
        rw->SetWindowName("Calculating PolyData Curvature");

        vtkSmartPointer<vtkRenderWindowInteractor> rwi =
            vtkSmartPointer<vtkRenderWindowInteractor>::New();
        rwi->SetRenderWindow(rw);
        rwi->Initialize();
        rwi->Start();
        return 0;
    }