vtk_cone_polygon_model_rendering_quickstart.py

1import vtk
2
3def main():
4    # 1. Create a polygonal model of a cone
5    cone_source = vtk.vtkConeSource()
6    cone_source.SetHeight(3.0)
7    cone_source.SetRadius(1.0)
8    cone_source.SetResolution(10)
9
10    # 2. Create a mapper to map the polygonal data to graphics primitives
11    cone_mapper = vtk.vtkPolyDataMapper()
12    cone_mapper.SetInputConnection(cone_source.GetOutputPort())
13
14    # 3. Create an actor to represent the cone in the scene
15    cone_actor = vtk.vtkActor()
16    cone_actor.SetMapper(cone_mapper)
17
18    # 4. Create a renderer and a render window
19    renderer = vtk.vtkRenderer()
20    render_window = vtk.vtkRenderWindow()
21    render_window.AddRenderer(renderer)
22    render_window.SetWindowName("VTK Python Quickstart")
23
24    # 5. Create a render window interactor to handle mouse/keyboard events
25    render_window_interactor = vtk.vtkRenderWindowInteractor()
26    render_window_interactor.SetRenderWindow(render_window)
27
28    # 6. Add the actor to the renderer and set background color
29    renderer.AddActor(cone_actor)
30    renderer.SetBackground(0.1, 0.2, 0.4)  # Dark Blue
31
32    # 7. Zoom in a bit and render
33    renderer.ResetCamera()
34    render_window.Render()
35
36    # 8. Start the event loop
37    render_window_interactor.Start()
38
39if __name__ == "__main__":
40    main()