一杯清酒邀明月
天下本无事,庸人扰之而烦耳。
 1 #include <vtkActor.h>
 2 #include <vtkFloatArray.h>
 3 #include <vtkLookupTable.h>
 4 #include <vtkPointData.h>
 5 #include <vtkPolyData.h>
 6 #include <vtkPolyDataMapper.h>
 7 #include <vtkRenderer.h>
 8 #include <vtkRenderWindow.h>
 9 #include <vtkRenderWindowInteractor.h>
10 #include <vtkSmartPointer.h>
11 #include <vtkSTLReader.h>
12 #include <vtkInteractorStyleTrackballCamera.h>
13  
14 #include <vtkAutoInit.h> 
15 VTK_MODULE_INIT(vtkRenderingOpenGL2);
16 VTK_MODULE_INIT(vtkInteractionStyle);
17 VTK_MODULE_INIT(vtkRenderingVolumeOpenGL2);
18 VTK_MODULE_INIT(vtkRenderingFreeType);
19  
20 int main(int, char* [])
21 {
22     // 加载一个J20的STL模型
23     vtkSmartPointer<vtkSTLReader> source = vtkSmartPointer<vtkSTLReader>::New();
24     source->SetFileName("C:\\Users\\wangjun\\3D Objects\\J20.stl");
25     source->Update();
26  
27     int numPts = source->GetOutput()->GetPoints()->GetNumberOfPoints();                    // 获取模型的顶点数量
28     vtkSmartPointer<vtkFloatArray> scalars = vtkSmartPointer<vtkFloatArray>::New();        // 创建存储顶点属性的float数组
29     scalars->SetNumberOfValues(numPts);
30     for (int i = 0; i < numPts; ++i)        // 为属性数组中的每个元素设置标量值(这个标量值可以当作颜色值)
31         scalars->SetValue(i, i);
32  
33     vtkSmartPointer<vtkPolyData> poly = vtkSmartPointer<vtkPolyData>::New();
34     poly->DeepCopy(source->GetOutput());
35     poly->GetPointData()->SetScalars(scalars);
36  
37     // 创建颜色查找表
38     vtkSmartPointer<vtkLookupTable> hueLut = vtkSmartPointer<vtkLookupTable>::New();
39     hueLut->SetNumberOfColors(numPts);        // 指定颜色查找表中有多少种颜色
40     hueLut->SetHueRange(0.67, 0.0);            // 设定HSV颜色范围,色调H取值范围为0°~360°,从红色开始按逆时针方向计算,红色为0°/0.0,绿色为120°/0.34,蓝色为240°/0.67
41     hueLut->Build();
42  
43     vtkSmartPointer<vtkPolyDataMapper> mapper = vtkSmartPointer<vtkPolyDataMapper>::New();
44     mapper->SetInputData(poly);
45     mapper->SetScalarRange(0, numPts);            // 设置标量值的范围
46     mapper->ScalarVisibilityOn();
47     //mapper->SetColorModeToMapScalars();        // 无论变量数据是何种类型,该方法都通过查询表对标量数据进行映射
48     mapper->SetColorModeToDefault();            // 默认的映射器行为,即把unsigned char类型的标量属性数据当作颜色值,不执行隐式。对于其他类型的标量数据,将通过查询表映射。
49     mapper->SetLookupTable(hueLut);
50  
51     vtkSmartPointer<vtkActor> actor = vtkSmartPointer<vtkActor>::New();
52     actor->SetMapper(mapper);
53  
54     vtkSmartPointer<vtkRenderer> renderer = vtkSmartPointer<vtkRenderer>::New();
55     renderer->GradientBackgroundOn();
56     renderer->SetBackground(1, 1, 1);
57     renderer->SetBackground2(0, 0, 0);
58  
59     vtkSmartPointer<vtkRenderWindow> renderWindow = vtkSmartPointer<vtkRenderWindow>::New();
60     renderWindow->AddRenderer(renderer);
61  
62     vtkSmartPointer<vtkRenderWindowInteractor> renderWindowInteractor = vtkSmartPointer<vtkRenderWindowInteractor>::New();
63     renderWindowInteractor->SetRenderWindow(renderWindow);
64     vtkSmartPointer<vtkInteractorStyleTrackballCamera> style = vtkSmartPointer<vtkInteractorStyleTrackballCamera>::New();
65     renderWindowInteractor->SetInteractorStyle(style);
66  
67     renderer->AddActor(actor);
68     renderWindow->SetSize(600, 600);
69     renderWindow->Render();
70     renderWindowInteractor->Start();
71  
72     return 0;
73 }

 

 

posted on 2020-11-24 15:46  一杯清酒邀明月  阅读(1612)  评论(0编辑  收藏  举报