ExtractVisibleCells

medimage

1. #include <vtkActor.h>
   
2. #include <vtkDataSetMapper.h>
   
3. #include <vtkExtractSelection.h>
   
4. #include <vtkHardwareSelector.h>
   
5. #include <vtkInteractorStyleTrackballCamera.h>
   
6. #include <vtkNamedColors.h>
   
7. #include <vtkNew.h>
   
8. #include <vtkObjectFactory.h>
   
9. #include <vtkPolyData.h>
   
10. #include <vtkPolyDataMapper.h>
    
11. #include <vtkProperty.h>
    
12. #include <vtkRenderWindow.h>
    
13. #include <vtkRenderWindowInteractor.h>
    
14. #include <vtkRenderer.h>
    
15. #include <vtkRendererCollection.h>
    
16. #include <vtkSelection.h>
    
17. #include <vtkSphereSource.h>
    
18. 
    
19. // Define interaction style
    
20. class KeyPressInteractorStyle : public vtkInteractorStyleTrackballCamera
    
21. {
    
22. public:
    
23.   vtkNew<vtkNamedColors> colors;
    
24.   static KeyPressInteractorStyle* New();
    
25.   vtkTypeMacro(KeyPressInteractorStyle, vtkInteractorStyleTrackballCamera);
    
26. 
    
27.   virtual void OnKeyPress() override
    
28.   {
    
29.     // Get the keypress
    
30.     std::string key = this->Interactor->GetKeySym();
    
31. 
    
32.     // "s" for "s"elect
    
33.     if (key.compare("s") == 0)
    
34.     {
    
35.       vtkNew<vtkHardwareSelector> selector;
    
36.       selector->SetRenderer(this->Interactor->GetRenderWindow()
    
37.                                 ->GetRenderers()
    
38.                                 ->GetFirstRenderer());
    
39.       int* temp = this->Interactor->GetRenderWindow()->GetSize();
    
40.       unsigned int windowSize[4];
    
41.       windowSize[0] = temp[2];
    
42.       windowSize[1] = temp[3];
    
43.       windowSize[2] = temp[0];
    
44.       windowSize[3] = temp[1];
    
45.       /*
    
46.         for(unsigned int i = 0; i < 4; i++)
    
47.           {
    
48.           windowSize[i] = temp[i];
    
49.           }
    
50.         */
    
51.       selector->SetArea(windowSize);
    
52.       selector->SetFieldAssociation(vtkDataObject::FIELD_ASSOCIATION_CELLS);
    
53.       vtkSelection* selection = selector->Select();
    
54.       std::cout << "Selection has " << selection->GetNumberOfNodes()
    
55.                 << " nodes." << std::endl;
    
56. 
    
57.       vtkNew<vtkExtractSelection> extractSelection;
    
58.       extractSelection->SetInputData(0, this->Data);
    
59.       extractSelection->SetInputData(1, selection);
    
60.       extractSelection->Update();
    
61. 
    
62.       vtkNew<vtkDataSetMapper> mapper;
    
63.       mapper->SetInputConnection(extractSelection->GetOutputPort());
    
64. 
    
65.       vtkNew<vtkActor> actor;
    
66.       actor->SetMapper(mapper);
    
67.       actor->GetProperty()->SetColor(colors->GetColor3d("Red").GetData());
    
68.       this->Renderer->AddActor(actor);
    
69.     }
    
70. 
    
71.     // Forward events
    
72.     vtkInteractorStyleTrackballCamera::OnKeyPress();
    
73.   }
    
74. 
    
75.   vtkPolyData* Data;
    
76.   vtkRenderer* Renderer;
    
77. };
    
78. vtkStandardNewMacro(KeyPressInteractorStyle);
    
79. 
    
80. int main(int, char*[])
    
81. {
    
82.   vtkNew<vtkNamedColors> colors;
    
83. 
    
84.   // Create a sphere
    
85.   vtkNew<vtkSphereSource> sphereSource;
    
86.   sphereSource->SetCenter(0.0, 0.0, 0.0);
    
87.   sphereSource->SetRadius(5.0);
    
88.   sphereSource->Update();
    
89. 
    
90.   // Create a mapper and actor
    
91.   vtkNew<vtkPolyDataMapper> mapper;
    
92.   mapper->SetInputConnection(sphereSource->GetOutputPort());
    
93. 
    
94.   vtkNew<vtkActor> actor;
    
95.   actor->SetMapper(mapper);
    
96.   actor->GetProperty()->SetColor(colors->GetColor3d("Tomato").GetData());
    
97. 
    
98.   // Create a renderer, render window, and interactor
    
99.   vtkNew<vtkRenderer> renderer;
    
100.   vtkNew<vtkRenderWindow> renderWindow;
     
101.   renderWindow->SetMultiSamples(0); // Turn off anti-aliasing
     
102.   renderWindow->AddRenderer(renderer);
     
103.   vtkNew<vtkRenderWindowInteractor> renderWindowInteractor;
     
104.   renderWindowInteractor->SetRenderWindow(renderWindow);
     
105. 
     
106.   vtkNew<KeyPressInteractorStyle> style;
     
107.   style->Renderer = renderer;
     
108.   renderWindowInteractor->SetInteractorStyle(style);
     
109.   style->SetCurrentRenderer(renderer);
     
110.   style->Data = sphereSource->GetOutput();
     
111. 
     
112.   // Add the actor to the scene
     
113.   renderer->AddActor(actor);
     
114.   renderer->SetBackground(colors->GetColor3d("Mint").GetData());
     
115. 
     
116.   // Render and interact
     
117.   renderWindow->SetWindowName("ExtractVisibleCells");
     
118.   renderWindow->Render();
     
119.   renderWindowInteractor->Start();
     
120. 
     
121.   return EXIT_SUCCESS;
     
122. }
     

复制代码

1. cmake_minimum_required(VERSION 3.8...3.12 FATAL_ERROR)
   
2. project(ExtractVisibleCells)
   
3. 
   
4. find_package(VTK)
   
5. if (NOT VTK_FOUND)
   
6.   message("Skipping example: ${VTK_NOT_FOUND_MESSAGE}")
   
7.   return ()
   
8. endif ()
   
9. 
   
10. set(Srcs ExtractVisibleCells.cxx)
    
11. 
    
12. add_executable(qtimageviewer
    
13.   ${Srcs})
    
14. target_link_libraries(qtimageviewer
    
15.   PRIVATE
    
16.     ${VTK_LIBRARIES})
    
17. vtk_module_autoinit(
    
18.   TARGETS qtimageviewer
    
19.   MODULES ${VTK_LIBRARIES})

复制代码

CMakeLists.txt